=================================================================== RCS file: /home/cvs/OpenXM_contrib2/asir2000/engine/dist.c,v retrieving revision 1.31 retrieving revision 1.37 diff -u -p -r1.31 -r1.37 --- OpenXM_contrib2/asir2000/engine/dist.c 2004/05/14 06:02:54 1.31 +++ OpenXM_contrib2/asir2000/engine/dist.c 2005/11/26 01:28:12 1.37 @@ -45,7 +45,7 @@ * DEVELOPER SHALL HAVE NO LIABILITY IN CONNECTION WITH THE USE, * PERFORMANCE OR NON-PERFORMANCE OF THE SOFTWARE. * - * $OpenXM: OpenXM_contrib2/asir2000/engine/dist.c,v 1.30 2004/04/14 07:27:41 ohara Exp $ + * $OpenXM: OpenXM_contrib2/asir2000/engine/dist.c,v 1.36 2005/11/25 07:18:32 noro Exp $ */ #include "ca.h" @@ -596,10 +596,13 @@ void subd(VL vl,DP p1,DP p2,DP *pr) void chsgnd(DP p,DP *pr) { MP m,mr,mr0; + Obj r; if ( !p ) *pr = 0; - else { + else if ( OID(p) <= O_R ) { + chsgnr((Obj)p,&r); *pr = (DP)r; + } else { for ( mr0 = 0, m = BDY(p); m; m = NEXT(m) ) { NEXTMP(mr0,mr); chsgnp(C(m),&C(mr)); mr->dl = m->dl; } @@ -1774,4 +1777,570 @@ int compdv(VL vl,DPV p1,DPV p2) return t; return 0; } +} + +int ni_next(int *a,int n) +{ + int i,j,k,kj; + + /* find the first nonzero a[j] */ + for ( j = 0; j < n && a[j] == 0; j++ ); + /* find the first zero a[k] after a[j] */ + for ( k = j; k < n && a[k] == 1; k++ ); + if ( k == n ) return 0; + /* a[0] = 0, ... , a[j-1] = 0, a[j] = 1, ..., a[k-1] = 1, a[k] = 0 */ + /* a[0] = 1,..., a[k-j-2] = 1, a[k-j-1] = 0, ..., a[k-1] = 0, a[k] = 1 */ + kj = k-j-1; + for ( i = 0; i < kj; i++ ) a[i] = 1; + for ( ; i < k; i++ ) a[i] = 0; + a[k] = 1; + return 1; +} + +int comp_nbm(NBM a,NBM b) +{ + int d,i,w; + int *ab,*bb; + + if ( a->d > b->d ) return 1; + else if ( a->d < b->d ) return -1; + else { + d = a->d; ab = a->b; bb = b->b; + w = (d+31)/32; + for ( i = 0; i < w; i++ ) + if ( ab[i] > bb[i] ) return 1; + else if ( ab[i] < bb[i] ) return -1; + return 0; + } +} + +NBM mul_nbm(NBM a,NBM b) +{ + int ad,bd,d,i,j; + int *ab,*bb,*mb; + NBM m; + Q c,c1; + NODE r; + NBP u; + + ad = a->d; bd = b->d; ab = a->b; bb = b->b; + d = ad + bd; + NEWNBM(m); NEWNBMBDY(m,d); + m->d = d; mulq(a->c,b->c,&m->c); mb = m->b; + j = 0; + for ( i = 0; i < ad; i++, j++ ) + if ( NBM_GET(ab,i) ) NBM_SET(mb,j); + else NBM_CLR(mb,j); + for ( i = 0; i < bd; i++, j++ ) + if ( NBM_GET(bb,i) ) NBM_SET(mb,j); + else NBM_CLR(mb,j); + return m; +} + +NBP nbmtonbp(NBM m) +{ + NODE n; + NBP u; + + MKNODE(n,m,0); + MKNBP(u,n); + return u; +} + +/* a=c*x*rest -> a0= x*rest, ah=x, ar=rest */ + +Q separate_nbm(NBM a,NBP *a0,NBP *ah,NBP *ar) +{ + int i,d1; + NBM t; + + if ( !a->d ) error("separate_nbm : invalid argument"); + + NEWNBM(t); t->d = a->d; t->b = a->b; t->c = ONE; + *a0 = nbmtonbp(t); + + NEWNBM(t); NEWNBMBDY(t,1); t->d = 1; t->c = ONE; + if ( NBM_GET(a->b,0) ) NBM_SET(t->b,0); + else NBM_CLR(t->b,0); + *ah = nbmtonbp(t); + + d1 = a->d-1; + NEWNBM(t); NEWNBMBDY(t,d1); t->d = d1; t->c = ONE; + for ( i = 0; i < d1; i++ ) { + if ( NBM_GET(a->b,i+1) ) NBM_SET(t->b,i); + else NBM_CLR(t->b,i); + } + *ar = nbmtonbp(t); + return a->c; +} + +NBP make_xky(int k) +{ + int k1,i; + NBM t; + + NEWNBM(t); NEWNBMBDY(t,k); t->d = k; t->c = ONE; + k1 = k-1; + for ( i = 0; i < k1; i++ ) NBM_SET(t->b,i); + NBM_CLR(t->b,i); + return nbmtonbp(t); +} + +/* a=c*x^(k-1)*y*rest -> a0= x^(k-1)*y*rest, ah=x^(k-1)*y, ar=rest */ + +Q separate_xky_nbm(NBM a,NBP *a0,NBP *ah,NBP *ar) +{ + int i,d1,k,k1; + NBM t; + + if ( !a->d ) + error("separate_nbm : invalid argument"); + for ( i = 0; i < a->d && NBM_GET(a->b,i); i++ ); + if ( i == a->d ) + error("separate_nbm : invalid argument"); + k1 = i; + k = i+1; + + NEWNBM(t); t->d = a->d; t->b = a->b; t->c = ONE; + *a0 = nbmtonbp(t); + + NEWNBM(t); NEWNBMBDY(t,k); t->d = k; t->c = ONE; + for ( i = 0; i < k1; i++ ) NBM_SET(t->b,i); + NBM_CLR(t->b,i); + *ah = nbmtonbp(t); + + d1 = a->d-k; + NEWNBM(t); NEWNBMBDY(t,d1); t->d = d1; t->c = ONE; + for ( i = 0; i < d1; i++ ) { + if ( NBM_GET(a->b,i+k) ) NBM_SET(t->b,i); + else NBM_CLR(t->b,i); + } + *ar = nbmtonbp(t); + + return a->c; +} +#if 0 +NBP shuffle_mul_nbm(NBM a,NBM b) +{ + int ad,bd,d,i,ai,bi,bit,s; + int *ab,*bb,*wmb,*w; + NBM wm,tm; + Q c,c1; + NODE r,t,t1,p; + NBP u; + + ad = a->d; bd = b->d; ab = a->b; bb = b->b; + d = ad + bd; + w = (int *)ALLOCA(d*sizeof(int)); + NEWNBM(wm); NEWNBMBDY(wm,d); wmb = wm->b; + for ( i = 0; i < ad; i++ ) w[i] = 1; + for ( ; i < d; i++ ) w[i] = 0; + mulq(a->c,b->c,&c); + r = 0; + do { + wm->d = d; wm->c = c; + ai = 0; bi = 0; + for ( i = 0; i < d; i++ ) { + if ( w[i] ) { bit = NBM_GET(ab,ai); ai++; } + else { bit = NBM_GET(bb,bi); bi++; } + if ( bit ) NBM_SET(wmb,i); + else NBM_CLR(wmb,i); + } + for ( p = 0, t = r; t; p = t, t = NEXT(t) ) { + tm = (NBM)BDY(t); + s = comp_nbm(tm,wm); + if ( s < 0 ) { + /* insert */ + MKNODE(t1,wm,t); + if ( !p ) r = t1; + else NEXT(p) = t1; + NEWNBM(wm); NEWNBMBDY(wm,d); wmb = wm->b; + break; + } else if ( s == 0 ) { + /* add coefs */ + addq(tm->c,c,&c1); + if ( c1 ) tm->c = c1; + else NEXT(p) = NEXT(t); + break; + } + } + if ( !t ) { + /* append */ + MKNODE(t1,wm,t); + if ( !p ) r = t1; + else NEXT(p) = t1; + NEWNBM(wm); NEWNBMBDY(wm,d); wmb = wm->b; + } + } while ( ni_next(w,d) ); + MKNBP(u,r); + return u; +} +#else +void shuffle_mulnbp(VL vl,NBP p1,NBP p2, NBP *rp); +void harmonic_mulnbp(VL vl,NBP p1,NBP p2, NBP *rp); + +NBP shuffle_mul_nbm(NBM a,NBM b) +{ + NBP u,a0,ah,ar,b0,bh,br,a1,b1,t; + Q ac,bc,c; + + if ( !a->d || !b->d ) + u = nbmtonbp(mul_nbm(a,b)); + else { + ac = separate_nbm(a,&a0,&ah,&ar); + bc = separate_nbm(b,&b0,&bh,&br); + mulq(ac,bc,&c); + shuffle_mulnbp(CO,ar,b0,&t); mulnbp(CO,ah,t,&a1); + shuffle_mulnbp(CO,a0,br,&t); mulnbp(CO,bh,t,&b1); + addnbp(CO,a1,b1,&t); mulnbp(CO,(NBP)c,t,&u); + } + return u; +} +#endif + +int nbmtoxky(NBM a,int *b) +{ + int d,i,j,k; + int *p; + + d = a->d; p = a->b; + for ( i = j = 0, k = 1; i < d; i++ ) { + if ( !NBM_GET(p,i) ) { + b[j++] = k; + k = 1; + } else k++; + } + return j; +} + +#if 0 +NBP harmonic_mul_nbm(NBM a,NBM b) +{ + int da,db,d,la,lb,lmax,lmin,l,lab,la1,lb1,lab1; + int i,j,k,ia,ib,s; + int *wa,*wb,*w,*wab,*wa1,*wmb; + Q c,c1; + NBM wm,tm; + NODE r,t1,t,p; + NBP u; + + da = a->d; db = b->d; d = da+db; + wa = (int *)ALLOCA(da*sizeof(int)); + wb = (int *)ALLOCA(db*sizeof(int)); + la = nbmtoxky(a,wa); + lb = nbmtoxky(b,wb); + mulq(a->c,b->c,&c); + /* wa[0],..,wa[la-1] <-> x^wa[0]y x^wa[1]y .. */ + /* lmax : total length */ + lmax = la+lb; + lmin = la>lb?la:lb; + w = (int *)ALLOCA(lmax*sizeof(int)); + /* position of a+b */ + wab = (int *)ALLOCA(lmax*sizeof(int)); + /* position of a */ + wa1 = (int *)ALLOCA(lmax*sizeof(int)); + NEWNBM(wm); NEWNBMBDY(wm,d); wmb = wm->b; + for ( l = lmin, r = 0; l <= lmax; l++ ) { + lab = lmax - l; + la1 = la - lab; + lb1 = lb - lab; + lab1 = l-lab; + /* partion l into three parts: a, b, a+b */ + /* initialize wab */ + for ( i = 0; i < lab; i++ ) wab[i] = 1; + for ( ; i < l; i++ ) wab[i] = 0; + do { + /* initialize wa1 */ + for ( i = 0; i < la1; i++ ) wa1[i] = 1; + for ( ; i < lab1; i++ ) wa1[i] = 0; + do { + ia = 0; ib = 0; + for ( i = j = 0; i < l; i++ ) + if ( wab[i] ) w[i] = wa[ia++]+wb[ib++]; + else if ( wa1[j++] ) w[i] = wa[ia++]; + else w[i] = wb[ib++]; + for ( i = j = 0; i < l; i++ ) { + for ( k = w[i]-1; k > 0; k--, j++ ) NBM_SET(wmb,j); + NBM_CLR(wmb,j); j++; + } + wm->d = j; wm->c = c; + for ( p = 0, t = r; t; p = t, t = NEXT(t) ) { + tm = (NBM)BDY(t); + s = comp_nbm(tm,wm); + if ( s < 0 ) { + /* insert */ + MKNODE(t1,wm,t); + if ( !p ) r = t1; + else NEXT(p) = t1; + NEWNBM(wm); NEWNBMBDY(wm,d); wmb = wm->b; + break; + } else if ( s == 0 ) { + /* add coefs */ + addq(tm->c,c,&c1); + if ( c1 ) tm->c = c1; + else NEXT(p) = NEXT(t); + break; + } + } + if ( !t ) { + /* append */ + MKNODE(t1,wm,t); + if ( !p ) r = t1; + else NEXT(p) = t1; + NEWNBM(wm); NEWNBMBDY(wm,d); wmb = wm->b; + } + } while ( ni_next(wa1,lab1) ); + } while ( ni_next(wab,l) ); + } + MKNBP(u,r); + return u; +} +#else +NBP harmonic_mul_nbm(NBM a,NBM b) +{ + NBP u,a0,ah,ar,b0,bh,br,a1,b1,t,s,abk,ab1; + Q ac,bc,c; + + if ( !a->d || !b->d ) + u = nbmtonbp(mul_nbm(a,b)); + else { + mulq(a->c,b->c,&c); + ac = separate_xky_nbm(a,&a0,&ah,&ar); + bc = separate_xky_nbm(b,&b0,&bh,&br); + mulq(ac,bc,&c); + harmonic_mulnbp(CO,ar,b0,&t); mulnbp(CO,ah,t,&a1); + harmonic_mulnbp(CO,a0,br,&t); mulnbp(CO,bh,t,&b1); + abk = make_xky(((NBM)BDY(BDY(ah)))->d+((NBM)BDY(BDY(bh)))->d); + harmonic_mulnbp(CO,ar,br,&t); mulnbp(CO,abk,t,&ab1); + addnbp(CO,a1,b1,&t); addnbp(CO,t,ab1,&s); mulnbp(CO,(NBP)c,s,&u); + } + return u; + +} +#endif + +void addnbp(VL vl,NBP p1,NBP p2, NBP *rp) +{ + NODE b1,b2,br,br0; + NBM m1,m2,m; + Q c; + + if ( !p1 ) + *rp = p2; + else if ( !p2 ) + *rp = p1; + else { + for ( b1 = BDY(p1), b2 = BDY(p2), br0 = 0; b1 && b2; ) { + m1 = (NBM)BDY(b1); m2 = (NBM)BDY(b2); + switch ( comp_nbm(m1,m2) ) { + case 0: + addq(m1->c,m2->c,&c); + if ( c ) { + NEXTNODE(br0,br); + NEWNBM(m); m->d = m1->d; m->c = c; m->b = m1->b; + BDY(br) = (pointer)m; + } + b1 = NEXT(b1); b2 = NEXT(b2); break; + case 1: + NEXTNODE(br0,br); BDY(br) = BDY(b1); + b1 = NEXT(b1); break; + case -1: + NEXTNODE(br0,br); BDY(br) = BDY(b2); + b2 = NEXT(b2); break; + } + } + if ( !br0 ) + if ( b1 ) + br0 = b1; + else if ( b2 ) + br0 = b2; + else { + *rp = 0; + return; + } + else if ( b1 ) + NEXT(br) = b1; + else if ( b2 ) + NEXT(br) = b2; + else + NEXT(br) = 0; + MKNBP(*rp,br0); + } +} + +void subnbp(VL vl,NBP p1,NBP p2, NBP *rp) +{ + NBP t; + + chsgnnbp(p2,&t); + addnbp(vl,p1,t,rp); +} + +void chsgnnbp(NBP p,NBP *rp) +{ + NODE r0,r,b; + NBM m,m1; + + for ( r0 = 0, b = BDY(p); b; b = NEXT(b) ) { + NEXTNODE(r0,r); + m = (NBM)BDY(b); + NEWNBM(m1); m1->d = m->d; m1->b = m->b; chsgnq(m->c,&m1->c); + BDY(r) = m1; + } + if ( r0 ) NEXT(r) = 0; + MKNBP(*rp,r0); +} + +void mulnbmnbp(VL vl,NBM m,NBP p, NBP *rp); +void mulnbpnbm(VL vl,NBP p,NBM m, NBP *rp); + +void mulnbp(VL vl,NBP p1,NBP p2, NBP *rp) +{ + NODE b,n; + NBP r,t,s; + NBM m; + + if ( !p1 || !p2 ) { + *rp = 0; return; + } + if ( OID(p1) != O_NBP ) { + if ( !NUM(p1) || !RATN(p1) ) + error("mulnbp : invalid argument"); + NEWNBM(m); m->d = 0; m->b = 0; m->c = (Q)p1; + MKNODE(n,m,0); MKNBP(p1,n); + } + if ( OID(p2) != O_NBP ) { + if ( !NUM(p2) || !RATN(p2) ) + error("mulnbp : invalid argument"); + NEWNBM(m); m->d = 0; m->b = 0; m->c = (Q)p2; + MKNODE(n,m,0); MKNBP(p2,n); + } + if ( length(BDY(p1)) < length(BDY(p2)) ) { + for ( r = 0, b = BDY(p1); b; b = NEXT(b) ) { + mulnbmnbp(vl,(NBM)BDY(b),p2,&t); + addnbp(vl,r,t,&s); r = s; + } + *rp = r; + } else { + for ( r = 0, b = BDY(p2); b; b = NEXT(b) ) { + mulnbpnbm(vl,p1,(NBM)BDY(b),&t); + addnbp(vl,r,t,&s); r = s; + } + *rp = r; + } +} + +void mulnbmnbp(VL vl,NBM m,NBP p, NBP *rp) +{ + NODE b,r0,r; + + if ( !p ) *rp = 0; + else { + for ( r0 = 0, b = BDY(p); b; b = NEXT(b) ) { + NEXTNODE(r0,r); + BDY(r) = mul_nbm(m,(NBM)BDY(b)); + } + if ( r0 ) NEXT(r) = 0; + MKNBP(*rp,r0); + } +} + +void mulnbpnbm(VL vl,NBP p,NBM m, NBP *rp) +{ + NODE b,r0,r; + + if ( !p ) *rp = 0; + else { + for ( r0 = 0, b = BDY(p); b; b = NEXT(b) ) { + NEXTNODE(r0,r); + BDY(r) = mul_nbm((NBM)BDY(b),m); + } + if ( r0 ) NEXT(r) = 0; + MKNBP(*rp,r0); + } +} + +void pwrnbp(VL vl,NBP a,Q q,NBP *c) +{ + int t; + NBP a1,a2; + N n1; + Q q1; + NBM m; + NODE r; + + if ( !q ) { + NEWNBM(m); m->d = 0; m->c = ONE; m->b = 0; + MKNODE(r,m,0); MKNBP(*c,r); + } else if ( !a ) + *c = 0; + else if ( UNIQ(q) ) + *c = a; + else { + t = divin(NM(q),2,&n1); NTOQ(n1,1,q1); + pwrnbp(vl,a,q1,&a1); + mulnbp(vl,a1,a1,&a2); + if ( t ) + mulnbp(vl,a2,a,c); + else + *c = a2; + } +} + +void shuffle_mulnbp(VL vl,NBP p1,NBP p2, NBP *rp) +{ + NODE b1,b2,n; + NBP r,t,s; + NBM m; + + if ( !p1 || !p2 ) { + *rp = 0; return; + } + if ( OID(p1) != O_NBP ) { + if ( !NUM(p1) || !RATN(p1) ) + error("shuffle_mulnbp : invalid argument"); + NEWNBM(m); m->d = 0; m->b = 0; m->c = (Q)p1; + MKNODE(n,m,0); MKNBP(p1,n); + } + if ( OID(p2) != O_NBP ) { + if ( !NUM(p2) || !RATN(p2) ) + error("shuffle_mulnbp : invalid argument"); + NEWNBM(m); m->d = 0; m->b = 0; m->c = (Q)p2; + MKNODE(n,m,0); MKNBP(p2,n); + } + for ( r = 0, b1 = BDY(p1); b1; b1 = NEXT(b1) ) + for ( m = BDY(b1), b2 = BDY(p2); b2; b2 = NEXT(b2) ) { + t = shuffle_mul_nbm(m,(NBM)BDY(b2)); + addnbp(vl,r,t,&s); r = s; + } + *rp = r; +} + +void harmonic_mulnbp(VL vl,NBP p1,NBP p2, NBP *rp) +{ + NODE b1,b2,n; + NBP r,t,s; + NBM m; + + if ( !p1 || !p2 ) { + *rp = 0; return; + } + if ( OID(p1) != O_NBP ) { + if ( !NUM(p1) || !RATN(p1) ) + error("harmonic_mulnbp : invalid argument"); + NEWNBM(m); m->d = 0; m->b = 0; m->c = (Q)p1; + MKNODE(n,m,0); MKNBP(p1,n); + } + if ( OID(p2) != O_NBP ) { + if ( !NUM(p2) || !RATN(p2) ) + error("harmonic_mulnbp : invalid argument"); + NEWNBM(m); m->d = 0; m->b = 0; m->c = (Q)p2; + MKNODE(n,m,0); MKNBP(p2,n); + } + for ( r = 0, b1 = BDY(p1); b1; b1 = NEXT(b1) ) + for ( m = BDY(b1), b2 = BDY(p2); b2; b2 = NEXT(b2) ) { + t = harmonic_mul_nbm(m,(NBM)BDY(b2)); + addnbp(vl,r,t,&s); r = s; + } + *rp = r; }