version 1.3, 2015/08/17 05:18:35 |
version 1.17, 2018/06/04 06:39:26 |
|
|
/* $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.16 2018/03/27 07:05:17 noro Exp $ */ |
|
|
#include <stdio.h> |
#include <signal.h> |
#include <stdlib.h> |
#include "ox_pari.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 26 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); |
} |
} |
|
|
|
|
|
|
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) |
|
|
|
|
int sm_mathcap() |
int sm_mathcap() |
{ |
{ |
mathcap_init(OX_PARI_VERSION, ID_STRING, "ox_pari", NULL, NULL); |
#if 0 |
push((cmo*)oxf_cmo_mathcap(fd_rw)); |
char *opts[] = {"no_ox_reset", NULL}; |
return 0; |
mathcap_init2(OX_PARI_VERSION, ID_STRING, "ox_pari", NULL, NULL, opts); |
|
#else |
|
mathcap_init2(OX_PARI_VERSION, ID_STRING, "ox_pari", NULL, NULL, NULL); |
|
#endif |
|
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]; |
|
int status; |
|
char *err; |
|
|
if ( setjmp(GP_DATA->env) ) { |
if ( (status = setjmp(GP_DATA->env)) != 0 ) { |
printf("sm_executeFunction : an error occured.\n");fflush(stdout); |
err = errmessage[status]; |
push((cmo*)make_error2(0)); |
if ( status == errpile ) { |
return -1; |
sprintf(buf,"%s\nIncrease PARI stack by pari(allocatemem,size).",err); |
|
init_pari(); |
|
} else if ( strlen(err) != 0 ) |
|
sprintf(buf,"An error occured in PARI :%s",err); |
|
else |
|
sprintf(buf,"An error occured in PARI."); |
|
push((cmo*)make_error2(buf)); |
|
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; |
} |
} |
|
|
|
jmp_buf ox_env; |
|
|
|
void usr1_handler(int sig) |
|
{ |
|
longjmp(ox_env,1); |
|
} |
|
|
int main() |
int main() |
{ |
{ |
init_gc(); |
if ( setjmp(ox_env) ) { |
ox_stderr_init(stderr); |
fprintf(stderr,"resetting libpari and sending OX_SYNC_BALL..."); |
initialize_stack(); |
initialize_stack(); |
init_pari(); |
init_pari(); |
|
send_ox_tag(fd_rw,OX_SYNC_BALL); |
|
fprintf(stderr,"done\n"); |
|
} else { |
|
init_gc(); |
|
ox_stderr_init(stderr); |
|
initialize_stack(); |
|
init_pari(); |
|
|
|
fprintf(stderr,"ox_pari\n"); |
|
|
fprintf(stderr,"ox_pari\n"); |
fd_rw = oxf_open(3); |
|
oxf_determine_byteorder_server(fd_rw); |
|
} |
|
|
fd_rw = oxf_open(3); |
#if defined(__CYGWIN__) |
oxf_determine_byteorder_server(fd_rw); |
void *mysignal(int sig,void (*handler)(int m)); |
|
mysignal(SIGUSR1,usr1_handler); |
|
#else |
|
signal(SIGUSR1,usr1_handler); |
|
#endif |
|
|
while(1){ |
while(1){ |
receive(); |
receive(); |
} |
} |
} |
} |