=================================================================== RCS file: /home/cvs/OpenXM_contrib2/asir2018/engine/Q.c,v retrieving revision 1.2 retrieving revision 1.5 diff -u -p -r1.2 -r1.5 --- OpenXM_contrib2/asir2018/engine/Q.c 2018/09/21 07:06:51 1.2 +++ OpenXM_contrib2/asir2018/engine/Q.c 2018/09/28 08:20:28 1.5 @@ -25,12 +25,22 @@ void gc_free(void *p,size_t size) void init_gmpq() { - mp_set_memory_functions(Risa_GC_malloc,gc_realloc,gc_free); + mp_set_memory_functions(Risa_GC_malloc_atomic,gc_realloc,gc_free); mpz_init(ONEMPZ); mpz_set_ui(ONEMPZ,1); MPZTOZ(ONEMPZ,ONE); gmp_randinit_default(GMP_RAND); } +void pmat(Z **a,int row,int col) +{ + int i,j; + + for ( i = 0; i < row; i++, printf("\n") ) + for ( j = 0; j < col; j++, printf(" ") ) + printexpr(CO,a[i][j]); + printf("\n"); +} + Z utoz(unsigned int u) { mpz_t z; @@ -161,7 +171,7 @@ void mulz(Z n1,Z n2,Z *nr) void muladdtoz(Z n1,Z n2,Z *nr) { -#if 1 +#if 0 Z t; if ( n1 && n2 ) { @@ -171,7 +181,7 @@ void muladdtoz(Z n1,Z n2,Z *nr) mpz_addmul(BDY(*nr),BDY(n1),BDY(n2)); if ( !mpz_sgn(BDY(*nr)) ) *nr = 0; - } + } #else Z t,s; @@ -183,6 +193,7 @@ void muladdtoz(Z n1,Z n2,Z *nr) void mul1addtoz(Z n1,long u,Z *nr) { +#if 0 Z t; if ( n1 && u ) { @@ -196,6 +207,11 @@ void mul1addtoz(Z n1,long u,Z *nr) if ( !mpz_sgn(BDY(*nr)) ) *nr = 0; } +#else + Z t,s; + + mul1z(n1,u,&t); addz(*nr,t,&s); *nr = s; +#endif } void mul1z(Z n1,long n2,Z *nr) @@ -337,7 +353,7 @@ void pwrz(Z n1,Z n,Z *nr) } else if ( !smallz(n) ) { error("exponent too big."); *nr = 0; } else if ( n1->z && mpz_sgn(BDY((Z)n))>0 ) { - mpz_init(z); mpz_pow_ui(z,BDY(n1),QTOS(n)); MPZTOZ(z,*nr); + mpz_init(z); mpz_pow_ui(z,BDY(n1),ZTOS(n)); MPZTOZ(z,*nr); } else { MPZTOMPQ(BDY(n1),q); MPQTOQ(q,r); pwrq(r,(Q)n,&p); *nr = (Z)p; @@ -437,7 +453,7 @@ void gcdvz_estimate(VECT v,Z *q) else addz(s,b[i],&u); s = u; } - for ( i = 0, t = 0; i < n; i++ ) { + for ( t = 0; i < n; i++ ) { if ( b[i] && mpz_sgn(BDY(b[i]))<0 ) subz(t,b[i],&u); else addz(t,b[i],&u); t = u; @@ -445,6 +461,49 @@ void gcdvz_estimate(VECT v,Z *q) gcdz(s,t,q); } +void gcdv_mpz_estimate(mpz_t g,mpz_t *b,int n) +{ + int m,m2,i,j; + mpz_t s,t; + + mpz_init(g); + for ( i = 0, m = 0; i < n; i++ ) + if ( mpz_sgn(b[i]) ) m++; + if ( !m ) { + mpz_set_ui(g,0); + return; + } + if ( m == 1 ) { + for ( i = 0, m = 0; i < n; i++ ) + if ( mpz_sgn(b[i]) ) break; + if ( mpz_sgn(b[i])<0 ) mpz_neg(g,b[i]); + else mpz_set(g,b[i]); + return ; + } + m2 = m/2; + mpz_init_set_ui(s,0); + for ( i = j = 0; j < m2; i++ ) { + if ( mpz_sgn(b[i]) ) { + if ( mpz_sgn(b[i])<0 ) + mpz_sub(s,s,b[i]); + else + mpz_add(s,s,b[i]); + j++; + } + } + mpz_init_set_ui(t,0); + for ( ; i < n; i++ ) { + if ( mpz_sgn(b[i]) ) { + if ( mpz_sgn(b[i])<0 ) + mpz_sub(t,t,b[i]); + else + mpz_add(t,t,b[i]); + } + } + mpz_gcd(g,s,t); +} + + void factorialz(unsigned int n,Z *nr) { mpz_t a; @@ -589,7 +648,7 @@ void pwrq(Q n1,Q n,Q *nr) } else if ( !smallz((Z)n) ) { error("exponent too big."); *nr = 0; } else { - e = QTOS(n); + e = ZTOS(n); if ( e < 0 ) { e = -e; if ( n1->z ) { @@ -646,8 +705,8 @@ void mkbc(int n,Z *t) Z c,d,iq; for ( t[0] = ONE, i = 1; i <= n/2; i++ ) { - STOQ(n-i+1,c); mulz(t[i-1],c,&d); - STOQ(i,iq); divsz(d,iq,&t[i]); + STOZ(n-i+1,c); mulz(t[i-1],c,&d); + STOZ(i,iq); divsz(d,iq,&t[i]); } for ( ; i <= n; i++ ) t[i] = t[n-i]; @@ -1310,7 +1369,7 @@ void isqrtz(Z a,Z *r) else { k = z_bits((Q)a); /* a <= 2^k-1 */ bshiftz(ONE,-((k>>1)+(k&1)),&x); /* a <= x^2 */ - STOQ(2,two); + STOZ(2,two); while ( 1 ) { pwrz(x,two,&t); if ( cmpz(t,a) <= 0 ) { @@ -1421,19 +1480,20 @@ int generic_gauss_elim_hensel(MAT mat,MAT *nmmat,Z *dn int *rind,*cind; int count; int ret; - struct oEGT eg_mul,eg_inv,eg_intrat,eg_check,tmp0,tmp1; + struct oEGT eg_mul1,eg_mul2,tmp0,tmp1,tmp2; int period; int *wx,*ptr; int wxsize,nsize; Z wn; Z wq; +init_eg(&eg_mul1); init_eg(&eg_mul2); a0 = (Z **)mat->body; row = mat->row; col = mat->col; w = (int **)almat(row,col); for ( ind = 0; ; ind++ ) { md = get_lprime(ind); - STOQ(md,mdq); + STOZ(md,mdq); for ( i = 0; i < row; i++ ) for ( j = 0, ai = a0[i], wi = w[i]; j < col; j++ ) wi[j] = remqi((Q)ai[j],md); @@ -1478,40 +1538,54 @@ int generic_gauss_elim_hensel(MAT mat,MAT *nmmat,Z *dn MKMAT(xmat,rank,ri); x = (Z **)(xmat)->body; MKMAT(*nmmat,rank,ri); nm = (Z **)(*nmmat)->body; wc = (int **)almat(rank,ri); - for ( i = 0; i < rank; i++ ) - wc[i] = w[i]+rank; *rindp = rind = (int *)MALLOC_ATOMIC(rank*sizeof(int)); *cindp = cind = (int *)MALLOC_ATOMIC((ri)*sizeof(int)); period = F4_INTRAT_PERIOD; for ( q = ONE, count = 0; ; ) { + /* check Ax=B mod q */ if ( DP_Print > 3 ) fprintf(stderr,"o"); /* wc = b mod md */ for ( i = 0; i < rank; i++ ) - for ( j = 0, bi = b[i], wi = wc[i]; j < ri; j++ ) { + for ( j = 0, bi = b[i], wi = wc[i]; j < ri; j++ ) wi[j] = remqi((Q)bi[j],md); - if ( wi[j] && sgnz(bi[j]) < 0 ) - wi[j] = md-wi[j]; - } - /* wc = A^(-1)wc; wc is normalized */ - solve_by_lu_mod(w,rank,md,wc,ri,1); + /* wc = A^(-1)wc; wc is not normalized */ + solve_by_lu_mod(w,rank,md,wc,ri,0); /* x += q*wc */ +get_eg(&tmp0); for ( i = 0; i < rank; i++ ) for ( j = 0, wi = wc[i]; j < ri; j++ ) mul1addtoz(q,wi[j],&x[i][j]); - /* b =(A*wc+b)/md */ + /* b =(b-A*wc)/md */ +get_eg(&tmp1); add_eg(&eg_mul1,&tmp0,&tmp1); for ( i = 0; i < rank; i++ ) for ( j = 0; j < ri; j++ ) { - u = b[i][j]; - for ( k = 0; k < rank; k++ ) mul1addtoz(a[i][k],wc[k][j],&u); + mpz_t uz; + + if ( b[i][j] ) + mpz_init_set(uz,BDY(b[i][j])); + else + mpz_init_set_ui(uz,0); + for ( k = 0; k < rank; k++ ) { + if ( a[i][k] && wc[k][j] ) { + if ( wc[k][j] < 0 ) + mpz_addmul_ui(uz,BDY(a[i][k]),-wc[k][j]); + else + mpz_submul_ui(uz,BDY(a[i][k]),wc[k][j]); + } + } + MPZTOZ(uz,u); divsz(u,mdq,&b[i][j]); } +get_eg(&tmp2); add_eg(&eg_mul2,&tmp1,&tmp2); count++; /* q = q*md */ mulz(q,mdq,&u); q = u; if ( count == period ) { ret = intmtoratm(xmat,q,*nmmat,dn); if ( ret ) { + print_eg("MUL1",&eg_mul1); + print_eg("MUL2",&eg_mul2); for ( j = k = l = 0; j < col; j++ ) if ( cinfo[j] ) rind[k++] = j; @@ -1567,7 +1641,7 @@ int generic_gauss_elim_hensel_dalg(MAT mat,DP *mb,MAT w = (int **)almat(row,col); for ( ind = 0; ; ind++ ) { md = get_lprime(ind); - STOQ(md,mdq); + STOZ(md,mdq); for ( i = 0; i < row; i++ ) for ( j = 0, ai = a0[i], wi = w[i]; j < col; j++ ) wi[j] = remqi((Q)ai[j],md); @@ -1593,14 +1667,14 @@ int generic_gauss_elim_hensel_dalg(MAT mat,DP *mb,MAT a = (Z **)almat_pointer(rank,rank); /* lhs mat */ MKMAT(bmat,rank,col-rank); b = (Z **)bmat->body; /* lhs mat */ for ( j = li = ri = 0; j < col; j++ ) - if ( cinfo[j] ) { + if ( cinfo[j] > 0 ) { /* the column is in lhs */ for ( i = 0; i < rank; i++ ) { w[i][li] = w[i][j]; a[i][li] = a0[rinfo[i]][j]; } li++; - } else { + } else if ( !cinfo[j] ) { /* the column is in rhs */ for ( i = 0; i < rank; i++ ) b[i][ri] = a0[rinfo[i]][j]; @@ -1623,8 +1697,6 @@ int generic_gauss_elim_hensel_dalg(MAT mat,DP *mb,MAT MKMAT(xmat,rank,ri); x = (Z **)(xmat)->body; MKMAT(*nmmat,rank,ri); nm = (Z **)(*nmmat)->body; wc = (int **)almat(rank,ri); - for ( i = 0; i < rank; i++ ) - wc[i] = w[i]+rank; *rindp = rind = (int *)MALLOC_ATOMIC(rank*sizeof(int)); *cindp = cind = (int *)MALLOC_ATOMIC((ri)*sizeof(int)); @@ -1634,21 +1706,31 @@ int generic_gauss_elim_hensel_dalg(MAT mat,DP *mb,MAT fprintf(stderr,"o"); /* wc = b mod md */ for ( i = 0; i < rank; i++ ) - for ( j = 0, bi = b[i], wi = wc[i]; j < ri; j++ ) { + for ( j = 0, bi = b[i], wi = wc[i]; j < ri; j++ ) wi[j] = remqi((Q)bi[j],md); - if ( wi[j] && sgnz(bi[j]) < 0 ) - wi[j] = md-wi[j]; - } /* wc = A^(-1)wc; wc is normalized */ solve_by_lu_mod(w,rank,md,wc,ri,1); /* x += q*wc */ for ( i = 0; i < rank; i++ ) for ( j = 0, wi = wc[i]; j < ri; j++ ) mul1addtoz(q,wi[j],&x[i][j]); - /* b =(A*wc+b)/md */ + /* b =(b-A*wc)/md */ for ( i = 0; i < rank; i++ ) for ( j = 0; j < ri; j++ ) { - u = b[i][j]; - for ( k = 0; k < rank; k++ ) mul1addtoz(a[i][k],wc[k][j],&u); + mpz_t uz; + + if ( b[i][j] ) + mpz_init_set(uz,BDY(b[i][j])); + else + mpz_init_set_ui(uz,0); + for ( k = 0; k < rank; k++ ) { + if ( a[i][k] && wc[k][j] ) { + if ( wc[k][j] < 0 ) + mpz_addmul_ui(uz,BDY(a[i][k]),-wc[k][j]); + else + mpz_submul_ui(uz,BDY(a[i][k]),wc[k][j]); + } + } + MPZTOZ(uz,u); divsz(u,mdq,&b[i][j]); } count++;