[BACK]Return to ox_pari.c CVS log [TXT][DIR] Up to [local] / OpenXM / src / ox_pari

Diff for /OpenXM/src/ox_pari/ox_pari.c between version 1.3 and 1.14

version 1.3, 2015/08/17 05:18:35 version 1.14, 2016/08/23 03:03:26
Line 1 
Line 1 
 /*      $OpenXM: OpenXM/src/ox_pari/ox_pari.c,v 1.2 2015/08/06 09:15:32 noro Exp $      */  /*  $OpenXM: OpenXM/src/ox_pari/ox_pari.c,v 1.13 2016/08/01 01:35:01 noro Exp $  */
   
 #include <stdio.h>  #include "ox_pari.h"
 #include <stdlib.h>  
 #include <string.h>  
 #include "pari/pari.h"  
 #include "pari/paripriv.h"  
 #include "gmp.h"  
 #include "gmp-impl.h"  
 #include "mpfr.h"  
 #include "ox_toolkit.h"  
 OXFILE *fd_rw;  OXFILE *fd_rw;
   
 #define MPFR_PREC(x)      ((x)->_mpfr_prec)  
 #define MPFR_EXP(x)       ((x)->_mpfr_exp)  
 #define MPFR_MANT(x)      ((x)->_mpfr_d)  
 #define MPFR_LAST_LIMB(x) ((MPFR_PREC (x) - 1) / GMP_NUMB_BITS)  
 #define MPFR_LIMB_SIZE(x) (MPFR_LAST_LIMB (x) + 1)  
   
 static int stack_size = 0;  static int stack_size = 0;
 static int stack_pointer = 0;  static int stack_pointer = 0;
 static cmo **stack = NULL;  static cmo **stack = NULL;
 extern int debug_print;  extern int debug_print;
 long paristack=10000000;  long paristack=10000000;
   
 void init_pari(void);  
 cmo *GEN_to_cmo(GEN z);  
 cmo_zz *GEN_to_cmo_zz(GEN z);  
 cmo_bf *GEN_to_cmo_bf(GEN z);  
 cmo_list *GEN_to_cmo_list(GEN z);  
 GEN cmo_to_GEN(cmo *c);  
 GEN cmo_zz_to_GEN(cmo_zz *c);  
 GEN cmo_bf_to_GEN(cmo_bf *c);  
   
 #define INIT_S_SIZE 2048  #define INIT_S_SIZE 2048
 #define EXT_S_SIZE  2048  #define EXT_S_SIZE  2048
   
Line 47  void gc_free(void *p,size_t size)
Line 25  void gc_free(void *p,size_t size)
   
 void init_gc()  void init_gc()
 {  {
         GC_INIT();    GC_INIT();
   mp_set_memory_functions(GC_malloc,gc_realloc,gc_free);    mp_set_memory_functions(GC_malloc,gc_realloc,gc_free);
 }  }
   
Line 58  void init_pari()
Line 36  void init_pari()
   
 int initialize_stack()  int initialize_stack()
 {  {
         stack_pointer = 0;    stack_pointer = 0;
         stack_size = INIT_S_SIZE;     stack_size = INIT_S_SIZE;
         stack = MALLOC(stack_size*sizeof(cmo*));    stack = MALLOC(stack_size*sizeof(cmo*));
         return 0;    return 0;
 }  }
   
 static int extend_stack()  static int extend_stack()
 {  {
         int size2 = stack_size + EXT_S_SIZE;    int size2 = stack_size + EXT_S_SIZE;
         cmo **stack2 = MALLOC(size2*sizeof(cmo*));    cmo **stack2 = MALLOC(size2*sizeof(cmo*));
         memcpy(stack2, stack, stack_size*sizeof(cmo *));    memcpy(stack2, stack, stack_size*sizeof(cmo *));
         free(stack);    free(stack);
         stack = stack2;    stack = stack2;
         stack_size = size2;    stack_size = size2;
         return 0;    return 0;
 }  }
   
 int push(cmo* m)  int push(cmo* m)
 {  {
         stack[stack_pointer] = m;    stack[stack_pointer] = m;
         stack_pointer++;    stack_pointer++;
         if(stack_pointer >= stack_size) {    if(stack_pointer >= stack_size) {
                 extend_stack();      extend_stack();
         }    }
         return 0;    return 0;
 }  }
   
 cmo* pop()  cmo* pop()
 {  {
         if(stack_pointer > 0) {    if(stack_pointer > 0) {
                 stack_pointer--;      stack_pointer--;
                 return stack[stack_pointer];      return stack[stack_pointer];
         }    }
         return new_cmo_null();    return new_cmo_null();
 }  }
   
 void pops(int n)  void pops(int n)
 {  {
         stack_pointer -= n;    stack_pointer -= n;
         if(stack_pointer < 0) {    if(stack_pointer < 0) {
                 stack_pointer = 0;      stack_pointer = 0;
         }    }
 }  }
   
 #define OX_PARI_VERSION 20150731  #define OX_PARI_VERSION 20150731
Line 107  void pops(int n)
Line 85  void pops(int n)
   
 int sm_mathcap()  int sm_mathcap()
 {  {
         mathcap_init(OX_PARI_VERSION, ID_STRING, "ox_pari", NULL, NULL);    char *opts[] = {"no_ox_reset", NULL};
         push((cmo*)oxf_cmo_mathcap(fd_rw));    mathcap_init2(OX_PARI_VERSION, ID_STRING, "ox_pari", NULL, NULL, opts);
         return 0;    push((cmo*)oxf_cmo_mathcap(fd_rw));
     return 0;
 }  }
   
 int sm_popCMO()  int sm_popCMO()
 {  {
         cmo* m = pop();    cmo* m = pop();
   
         if(m != NULL) {    if(m != NULL) {
                 send_ox_cmo(fd_rw, m);      send_ox_cmo(fd_rw, m);
                 return 0;      return 0;
         }    }
         return SM_popCMO;    return SM_popCMO;
 }  }
   
 cmo_error2 *make_error2(int code)  cmo_error2 *make_error2(char *message)
 {  {
         return (cmo_error2 *) new_cmo_int32(code);    return new_cmo_error2((cmo *)new_cmo_string(message));
 }  }
   
 int get_i()  int get_i()
 {  {
         cmo *c = pop();    cmo *c = pop();
         if(c->tag == CMO_INT32) {    if(c->tag == CMO_INT32) {
                 return ((cmo_int32 *)c)->i;      return ((cmo_int32 *)c)->i;
         }else if(c->tag == CMO_ZZ) {    }else if(c->tag == CMO_ZZ) {
                 return mpz_get_si(((cmo_zz *)c)->mpz);      return mpz_get_si(((cmo_zz *)c)->mpz);
         }    }
         make_error2(-1);    make_error2("get_i : invalid object");
         return 0;    return 0;
 }  }
   
 char *get_str()  char *get_str()
 {  {
         cmo *c = pop();    cmo *c = pop();
         if(c->tag == CMO_STRING) {    if(c->tag == CMO_STRING) {
                 return ((cmo_string *)c)->s;      return ((cmo_string *)c)->s;
         }  
         make_error2(-1);  
         return "";  
 }  
   
 int cmo2int(cmo *c)  
 {  
         if(c->tag == CMO_INT32) {  
                 return ((cmo_int32 *)c)->i;  
         }else if(c->tag == CMO_ZZ) {  
                 return mpz_get_si(((cmo_zz *)c)->mpz);  
         } else if(c->tag == CMO_NULL){  
                 return 0;  
         }  
   
         return 0;  
 }  
   
 GEN cmo_zz_to_GEN(cmo_zz *c)  
 {  
   mpz_ptr mpz;  
   GEN z;  
   long *ptr;  
   int j,sgn,len;  
   
   mpz = c->mpz;  
   sgn = mpz_sgn(mpz);  
   len = ABSIZ(mpz);  
   ptr = (long *)PTR(mpz);  
   z = cgeti(len+2);  
   for ( j = 0; j < len; j++ )  
     z[len-j+1] = ptr[j];  
   setsigne(z,sgn);  
   setlgefint(z,lg(z));  
   return z;  
 }  
   
 GEN cmo_bf_to_GEN(cmo_bf *c)  
 {  
   mpfr_ptr mpfr;  
   GEN z;  
   int sgn,len,j;  
   long exp;  
   long *ptr;  
   
   mpfr = c->mpfr;  
   sgn = MPFR_SIGN(mpfr);  
   exp = MPFR_EXP(mpfr)-1;  
   len = MPFR_LIMB_SIZE(mpfr);  
   ptr = (long *)MPFR_MANT(mpfr);  
   z = cgetr(len+2);  
   for ( j = 0; j < len; j++ )  
     z[len-j+1] = ptr[j];  
   z[1] = evalsigne(sgn)|evalexpo(exp);  
   setsigne(z,sgn);  
   return z;  
 }  
   
 cmo_zz *GEN_to_cmo_zz(GEN z)  
 {  
   cmo_zz *c;  
   
   c = new_cmo_zz();  
   mpz_import(c->mpz,lgef(z)-2,1,sizeof(long),0,0,&z[2]);  
   if ( signe(z) < 0 )  
     mpz_neg(c->mpz,c->mpz);  
   return c;  
 }  
   
 cmo_bf *GEN_to_cmo_bf(GEN z)  
 {  
   cmo_bf *c;  
   int len,prec,j;  
   long *ptr;  
   
   c = new_cmo_bf();  
   len = lg(z)-2;  
   prec = len*sizeof(long)*8;  
   mpfr_init2(c->mpfr,prec);  
   ptr = (long *)MPFR_MANT(c->mpfr);  
   for ( j = 0; j < len; j++ )  
     ptr[j] = z[len-j+1];  
   MPFR_EXP(c->mpfr) = (long long)(expo(z)+1);  
   MPFR_SIGN(c->mpfr) = gsigne(z);  
   return c;  
 }  
   
   
 cmo_list *GEN_to_cmo_list(GEN z)  
 {  
   cmo_list *c;  
   cmo *ob;  
   int i,len;  
   
   c = new_cmo_list();  
   len = lg(z)-1;  
   for ( i = 1; i <= len; i++ ) {  
     ob = GEN_to_cmo((GEN)z[i]);  
     c = list_append(c,ob);  
   }    }
   return c;    make_error2("get_str : invalid object");
     return "";
 }  }
   
   int ismatrix(GEN z)
 GEN cmo_to_GEN(cmo *c)  
 {  {
   switch ( c->tag ) {    int len,col,i;
   case CMO_ZERO:  
     return gen_0;  
   case CMO_ZZ: /* int */  
     return cmo_zz_to_GEN((cmo_zz *)c);  
   case CMO_BIGFLOAT: /* bigfloat */  
     return cmo_bf_to_GEN((cmo_bf *)c);  
   default:  
     return 0;  
   }  
 }  
   
 cmo *GEN_to_cmo(GEN z)    if ( typ(z) != t_VEC ) return 0;
 {    if ( typ((GEN)z[1]) != t_VEC ) return 0;
   if ( gcmp0(z) )    len = lg(z); col = lg((GEN)z[1]);
     return new_cmo_zero();    for ( i = 2; i < len; i++ )
   switch ( typ(z) ) {      if ( lg((GEN)z[i]) != col ) return 0;
   case 1: /* int */    return 1;
     return (cmo *)GEN_to_cmo_zz(z);  
   case 2: /* bigfloat */  
     return (cmo *)GEN_to_cmo_bf(z);  
   case 17: case 18: /* vector */  
     return (cmo *)GEN_to_cmo_list(z);  
   case 19: /* matrix */  
     return (cmo *)GEN_to_cmo_list(shallowtrans(z));  
   default:  
     return (cmo *)make_error2(typ(z));  
   }  
 }  }
   
 struct parif {  
   char *name;  
   int type;  
   GEN (*f)();  
 } parif_tab[] = {  
   {"erfc",1,gerfc},  
   {"factor",1,Z_factor},  
   {"isqrt",1,racine},  
   {"nextprime",1,nextprime},  
   {"det",1,det},  
   {"allocatemem",0,(GEN (*)())allocatemoremem},  
 };  
   
 #define PARI_MAX_AC 64  
   
 struct parif *search_parif(char *name)  
 {  
   int tablen,i;  
   
   tablen = sizeof(parif_tab)/sizeof(struct parif);  
   for ( i = 0; i < tablen; i++ ) {  
     if ( !strcmp(parif_tab[i].name,name) )  
       return &parif_tab[i];  
   }  
   return 0;  
 }  
   
 int sm_executeFunction()  int sm_executeFunction()
 {  {
   long ltop,lbot;    pari_sp av0;
   int ac,i;    int ac,i;
   cmo_int32 *c;    cmo_int32 *c;
   cmo *av[PARI_MAX_AC];    cmo *av[PARI_MAX_AC];
   cmo *ret;    cmo *ret;
   GEN z,m;    GEN z,m;
   struct parif *parif;    struct parif *parif;
     unsigned long prec;
     char buf[BUFSIZ];
   
   if ( setjmp(GP_DATA->env) ) {    if ( setjmp(GP_DATA->env) ) {
                 printf("sm_executeFunction : an error occured.\n");fflush(stdout);      sprintf(buf,"sm_executeFunction : an error occured in PARI.");
                 push((cmo*)make_error2(0));      push((cmo*)make_error2(buf));
                 return -1;      return -1;
   }    }
         cmo_string *func = (cmo_string *)pop();    cmo_string *func = (cmo_string *)pop();
         if(func->tag != CMO_STRING) {    if(func->tag != CMO_STRING) {
                 printf("sm_executeFunction : func->tag is not CMO_STRING");fflush(stdout);      sprintf(buf,"sm_executeFunction : func->tag=%d is not CMO_STRING",func->tag);
                 push((cmo*)make_error2(0));      push((cmo*)make_error2(buf));
                 return -1;      return -1;
         }    }
   
         c = (cmo_int32 *)pop();    c = (cmo_int32 *)pop();
   ac = c->i;    ac = c->i;
   if ( ac > PARI_MAX_AC ) {    if ( ac > PARI_MAX_AC ) {
                 push((cmo*)make_error2(0));      push((cmo*)make_error2("sm_executeFunction : too many arguments"));
                 return -1;      return -1;
   }    }
   for ( i = 0; i < ac; i++ ) {    for ( i = 0; i < ac; i++ ) {
     av[i] = (cmo *)pop();      av[i] = (cmo *)pop();
     fprintf(stderr,"arg%d:",i);  //    fprintf(stderr,"arg%d:",i);
     print_cmo(av[i]);  //    print_cmo(av[i]);
     fprintf(stderr,"\n");  //    fprintf(stderr,"\n");
   }    }
         if( strcmp( func->s, "exit" ) == 0 )    if( strcmp( func->s, "exit" ) == 0 )
                 exit(0);      exit(0);
   
   parif =search_parif(func->s);    parif =search_parif(func->s);
   if ( !parif ) {    if ( !parif ) {
                 push((cmo*)make_error2(0));      sprintf(buf,"%s : not implemented",func->s);
                 return -1;      push((cmo*)make_error2(buf));
       return -1;
  } else if ( parif->type == 0 ) {   } else if ( parif->type == 0 ) {
     /* one long int variable */      /* one long int variable */
     int a = cmo_to_int(av[0]);      int a = cmo_to_int(av[0]);
     a = (int)(*parif->f)(a);      a = (int)(parif->f)(a);
     ret = (cmo *)new_cmo_int32(a);      ret = (cmo *)new_cmo_int32(a);
     push(ret);      push(ret);
                 return 0;      return 0;
   } else if ( parif->type == 1 ) {    } else if ( parif->type == 1 ) {
     /* one variable possibly with prec */      /* one number/poly/matrix argument possibly with prec */
     unsigned long prec;      av0 = avma;
   
     ltop = avma;  
     z = cmo_to_GEN(av[0]);      z = cmo_to_GEN(av[0]);
     if ( ac == 2 ) {      prec = ac==2 ? cmo_to_int(av[1])*3.32193/32+3 : precreal;
       prec = cmo_to_int(av[1])*3.32193/32+3;      if ( ismatrix(z) ) {
     } else        int i,len;
       prec = precreal;        len = lg(z);
         for ( i = 1; i < len; i++ )
           settyp(z[i],t_COL);
         settyp(z,t_MAT);
         z = shallowtrans(z);
       }
       printf("input : "); output(z);
     m = (*parif->f)(z,prec);      m = (*parif->f)(z,prec);
     lbot = avma;  
     ret = GEN_to_cmo(m);      ret = GEN_to_cmo(m);
    // gerepile(ltop,lbot,0);      avma = av0;
     push(ret);      push(ret);
                 return 0;      return 0;
     } else if ( parif->type == 2 ) {
       /* one number/poly/matrix argument with flag=0 */
       av0 = avma;
       z = cmo_to_GEN(av[0]);
       if ( ismatrix(z) ) {
         int i,len;
         len = lg(z);
         for ( i = 1; i < len; i++ )
           settyp(z[i],t_COL);
         settyp(z,t_MAT);
         z = shallowtrans(z);
       }
       printf("input : "); output(z);
       m = (*parif->f)(z,0);
       ret = GEN_to_cmo(m);
       avma = av0;
       push(ret);
       return 0;
   } else {    } else {
                 push((cmo*)make_error2(0));      sprintf(buf,"%s : not implemented",func->s);
                 return -1;      push((cmo*)make_error2(buf));
       return -1;
   }    }
 }  }
   
 int receive_and_execute_sm_command()  int receive_and_execute_sm_command()
 {  {
         int code = receive_int32(fd_rw);    int code = receive_int32(fd_rw);
         switch(code) {    switch(code) {
         case SM_popCMO:    case SM_popCMO:
                 sm_popCMO();      sm_popCMO();
                 break;      break;
         case SM_executeFunction:    case SM_executeFunction:
                 sm_executeFunction();      sm_executeFunction();
                 break;      break;
         case SM_mathcap:    case SM_mathcap:
                 sm_mathcap();      sm_mathcap();
                 break;      break;
         case SM_setMathCap:    case SM_setMathCap:
                 pop();      pop();
                 break;      break;
         default:    case SM_shutdown:
                 printf("receive_and_execute_sm_command : code=%d\n",code);fflush(stdout);      exit(0);
                 break;      break;
         }    default:
         return 0;      printf("receive_and_execute_sm_command : code=%d\n",code);fflush(stdout);
       break;
     }
     return 0;
 }  }
   
 int receive()  int receive()
 {  {
         int tag;    int tag;
   
         tag = receive_ox_tag(fd_rw);    tag = receive_ox_tag(fd_rw);
         switch(tag) {    switch(tag) {
         case OX_DATA:    case OX_DATA:
                 printf("receive : ox_data %d\n",tag);fflush(stdout);      printf("receive : ox_data %d\n",tag);fflush(stdout);
                 push(receive_cmo(fd_rw));      push(receive_cmo(fd_rw));
                 break;      break;
         case OX_COMMAND:    case OX_COMMAND:
                 printf("receive : ox_command %d\n",tag);fflush(stdout);      printf("receive : ox_command %d\n",tag);fflush(stdout);
                 receive_and_execute_sm_command();      receive_and_execute_sm_command();
                 break;      break;
         default:    default:
                 printf("receive : tag=%d\n",tag);fflush(stdout);      printf("receive : tag=%d\n",tag);fflush(stdout);
         }    }
         return 0;    return 0;
 }  }
   
 int main()  int main()
 {  {
   init_gc();    init_gc();
         ox_stderr_init(stderr);    ox_stderr_init(stderr);
         initialize_stack();    initialize_stack();
         init_pari();    init_pari();
   
         fprintf(stderr,"ox_pari\n");    fprintf(stderr,"ox_pari\n");
   
         fd_rw = oxf_open(3);    fd_rw = oxf_open(3);
         oxf_determine_byteorder_server(fd_rw);    oxf_determine_byteorder_server(fd_rw);
   
         while(1){    while(1){
                 receive();      receive();
         }    }
 }  }
Legend:
Removed from v.1.3  
changed lines
  Added in v.1.14
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>