version 1.2, 2015/08/06 09:15:32 |
version 1.13, 2016/08/01 01:35:01 |
|
|
/* $OpenXM: OpenXM/src/ox_pari/ox_pari.c,v 1.1 2015/08/04 05:24:44 noro Exp $ */ |
/* $OpenXM: OpenXM/src/ox_pari/ox_pari.c,v 1.12 2015/08/21 01:20:30 noro Exp $ */ |
|
|
#include <stdio.h> |
#include "ox_pari.h" |
#include <stdlib.h> |
|
#include <string.h> |
|
#include "pari/pari.h" |
|
#include "gmp.h" |
|
#include "gmp-impl.h" |
|
#include "ox_toolkit.h" |
|
OXFILE *fd_rw; |
OXFILE *fd_rw; |
|
|
static int stack_size = 0; |
static int stack_size = 0; |
Line 15 static cmo **stack = NULL; |
|
Line 10 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_list *GEN_to_cmo_list(GEN z); |
|
GEN cmo_to_GEN(cmo *c); |
|
GEN cmo_zz_to_GEN(cmo_zz *c); |
|
|
|
#define INIT_S_SIZE 2048 |
#define INIT_S_SIZE 2048 |
#define EXT_S_SIZE 2048 |
#define EXT_S_SIZE 2048 |
|
|
|
void *gc_realloc(void *p,size_t osize,size_t nsize) |
|
{ |
|
return (void *)GC_realloc(p,nsize); |
|
} |
|
|
|
void gc_free(void *p,size_t size) |
|
{ |
|
GC_free(p); |
|
} |
|
|
|
void init_gc() |
|
{ |
|
GC_INIT(); |
|
mp_set_memory_functions(GC_malloc,gc_realloc,gc_free); |
|
} |
|
|
void init_pari() |
void init_pari() |
{ |
{ |
pari_init(paristack,2); |
pari_init(paristack,2); |
|
|
|
|
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 |
|
|
|
|
int sm_mathcap() |
int sm_mathcap() |
{ |
{ |
mathcap_init(OX_PARI_VERSION, ID_STRING, "ox_pari", NULL, NULL); |
mathcap_init(OX_PARI_VERSION, ID_STRING, "ox_pari", NULL, NULL); |
push((cmo*)oxf_cmo_mathcap(fd_rw)); |
push((cmo*)oxf_cmo_mathcap(fd_rw)); |
return 0; |
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; |
|
} |
|
|
|
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_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: |
|
case CMO_ZZ: /* int */ |
|
return cmo_zz_to_GEN((cmo_zz *)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 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)); |
|
} |
|
} |
} |
|
|
|
|
#define PARI_MAX_AC 64 |
|
|
|
int sm_executeFunction() |
int sm_executeFunction() |
{ |
{ |
|
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; |
|
unsigned long prec; |
|
char buf[BUFSIZ]; |
|
|
cmo_string *func = (cmo_string *)pop(); |
if ( setjmp(GP_DATA->env) ) { |
if(func->tag != CMO_STRING) { |
sprintf(buf,"sm_executeFunction : an error occured in PARI."); |
printf("sm_executeFunction : func->tag is not CMO_STRING");fflush(stdout); |
push((cmo*)make_error2(buf)); |
push((cmo*)make_error2(0)); |
return -1; |
return -1; |
} |
} |
cmo_string *func = (cmo_string *)pop(); |
|
if(func->tag != CMO_STRING) { |
|
sprintf(buf,"sm_executeFunction : func->tag=%d is not CMO_STRING",func->tag); |
|
push((cmo*)make_error2(buf)); |
|
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, "factor") == 0) { |
if( strcmp( func->s, "exit" ) == 0 ) |
z = cmo_to_GEN(av[0]); |
exit(0); |
m = Z_factor(z); |
|
ret = GEN_to_cmo(m); |
parif =search_parif(func->s); |
|
if ( !parif ) { |
|
sprintf(buf,"%s : not implemented",func->s); |
|
push((cmo*)make_error2(buf)); |
|
return -1; |
|
} else if ( parif->type == 0 ) { |
|
/* one long int variable */ |
|
int a = cmo_to_int(av[0]); |
|
a = (int)(parif->f)(a); |
|
ret = (cmo *)new_cmo_int32(a); |
push(ret); |
push(ret); |
return 0; |
return 0; |
} else if(strcmp(func->s, "nextprime") == 0) { |
} else if ( parif->type == 1 ) { |
|
/* one number/poly/matrix argument possibly with prec */ |
|
av0 = avma; |
z = cmo_to_GEN(av[0]); |
z = cmo_to_GEN(av[0]); |
m = nextprime(z); |
prec = ac==2 ? cmo_to_int(av[1])*3.32193/32+3 : precreal; |
|
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,prec); |
ret = GEN_to_cmo(m); |
ret = GEN_to_cmo(m); |
|
avma = av0; |
push(ret); |
push(ret); |
return 0; |
return 0; |
} else if(strcmp(func->s, "det") == 0) { |
} else if ( parif->type == 2 ) { |
|
/* one number/poly/matrix argument with flag=0 */ |
|
av0 = avma; |
z = cmo_to_GEN(av[0]); |
z = cmo_to_GEN(av[0]); |
m = det(z); |
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); |
ret = GEN_to_cmo(m); |
|
avma = av0; |
push(ret); |
push(ret); |
return 0; |
return 0; |
} else if( strcmp( func->s, "exit" ) == 0 ){ |
} else { |
pop(); |
sprintf(buf,"%s : not implemented",func->s); |
exit(0); |
push((cmo*)make_error2(buf)); |
return 0; |
return -1; |
} else { |
} |
push((cmo*)make_error2(0)); |
|
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() |
{ |
{ |
GC_INIT(); |
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(); |
} |
} |
} |
} |