Annotation of OpenXM_contrib/pari/src/basemath/alglin1.c, Revision 1.1.1.1
1.1 maekawa 1: /********************************************************************/
2: /** **/
3: /** LINEAR ALGEBRA **/
4: /** (first part) **/
5: /** **/
6: /********************************************************************/
7: /* $Id: alglin1.c,v 1.1.1.1 1999/09/16 13:47:15 karim Exp $ */
8: #include "pari.h"
9:
10: /*******************************************************************/
11: /* */
12: /* TRANSPOSE */
13: /* */
14: /*******************************************************************/
15:
16: GEN
17: gtrans(GEN x)
18: {
19: long i,j,lx,dx, tx=typ(x);
20: GEN y,p1;
21:
22: if (! is_matvec_t(tx)) err(typeer,"gtrans");
23: switch(tx)
24: {
25: case t_VEC:
26: y=gcopy(x); settyp(y,t_COL); break;
27:
28: case t_COL:
29: y=gcopy(x); settyp(y,t_VEC); break;
30:
31: case t_MAT:
32: lx=lg(x); if (lx==1) return cgetg(1,t_MAT);
33: dx=lg(x[1]); y=cgetg(dx,tx);
34: for (i=1; i<dx; i++)
35: {
36: p1=cgetg(lx,t_COL); y[i]=(long)p1;
37: for (j=1; j<lx; j++) p1[j]=lcopy(gcoeff(x,i,j));
38: }
39: break;
40:
41: default: y=gcopy(x); break;
42: }
43: return y;
44: }
45:
46: /*******************************************************************/
47: /* */
48: /* CONCATENATION & EXTRACTION */
49: /* */
50: /*******************************************************************/
51:
52: static GEN
53: strconcat(GEN x, GEN y)
54: {
55: long flx=0,fly=0,l;
56: char *sx,*sy,*str;
57:
58: if (typ(x)==t_STR) sx = GSTR(x); else { flx=1; sx = GENtostr(x); }
59: if (typ(y)==t_STR) sy = GSTR(y); else { fly=1; sy = GENtostr(y); }
60: l = strlen(sx) + strlen(sy) + 1;
61: l = (l+BYTES_IN_LONG) >> TWOPOTBYTES_IN_LONG;
62: x = cgetg(l+1, t_STR); str = GSTR(x);
63: strcpy(str,sx);
64: strcat(str,sy);
65: if (flx) free(sx);
66: if (fly) free(sy);
67: return x;
68: }
69:
70: GEN
71: concatsp(GEN x, GEN y)
72: {
73: long tx=typ(x),ty=typ(y),lx=lg(x),ly=lg(y),i;
74: GEN z,p1;
75:
76: if (tx==t_LIST || ty==t_LIST) return listconcat(x,y);
77: if (tx==t_STR || ty==t_STR) return strconcat(x,y);
78:
79: if (tx==t_MAT && lx==1)
80: {
81: if (ty!=t_VEC || ly==1) return gtomat(y);
82: err(concater);
83: }
84: if (ty==t_MAT && ly==1)
85: {
86: if (tx!=t_VEC || lx==1) return gtomat(x);
87: err(concater);
88: }
89:
90: if (! is_matvec_t(tx))
91: {
92: if (! is_matvec_t(ty))
93: {
94: z=cgetg(3,t_VEC); z[1]=(long)x; z[2]=(long)y;
95: return z;
96: }
97: z=cgetg(ly+1,ty);
98: if (ty != t_MAT) p1 = x;
99: else
100: {
101: if (lg(y[1])!=2) err(concater);
102: p1=cgetg(2,t_COL); p1[1]=(long)x;
103: }
104: for (i=2; i<=ly; i++) z[i]=y[i-1];
105: z[1]=(long)p1; return z;
106: }
107: if (! is_matvec_t(ty))
108: {
109: z=cgetg(lx+1,tx);
110: if (tx != t_MAT) p1 = y;
111: else
112: {
113: if (lg(x[1])!=2) err(concater);
114: p1=cgetg(2,t_COL); p1[1]=(long)y;
115: }
116: for (i=1; i<lx; i++) z[i]=x[i];
117: z[lx]=(long)p1; return z;
118: }
119:
120: if (tx == ty)
121: {
122: if (tx == t_MAT && lg(x[1]) != lg(y[1])) err(concater);
123: z=cgetg(lx+ly-1,tx);
124: for (i=1; i<lx; i++) z[i]=x[i];
125: for (i=1; i<ly; i++) z[lx+i-1]=y[i];
126: return z;
127: }
128:
129: switch(tx)
130: {
131: case t_VEC:
132: switch(ty)
133: {
134: case t_COL:
135: if (lx<=2) return (lx==1)? y: concatsp((GEN) x[1],y);
136: if (ly>=3) break;
137: return (ly==1)? x: concatsp(x,(GEN) y[1]);
138: case t_MAT:
139: z=cgetg(ly,ty); if (lx != ly) break;
140: for (i=1; i<ly; i++) z[i]=(long)concatsp((GEN) x[i],(GEN) y[i]);
141: return z;
142: }
143: break;
144:
145: case t_COL:
146: switch(ty)
147: {
148: case t_VEC:
149: if (lx<=2) return (lx==1)? y: concatsp((GEN) x[1],y);
150: if (ly>=3) break;
151: return (ly==1)? x: concatsp(x,(GEN) y[1]);
152: case t_MAT:
153: if (lx != lg(y[1])) break;
154: z=cgetg(ly+1,ty); z[1]=(long)x;
155: for (i=2; i<=ly; i++) z[i]=y[i-1];
156: return z;
157: }
158: break;
159:
160: case t_MAT:
161: switch(ty)
162: {
163: case t_VEC:
164: z=cgetg(lx,tx); if (ly != lx) break;
165: for (i=1; i<lx; i++) z[i]=(long)concatsp((GEN) x[i],(GEN) y[i]);
166: return z;
167: case t_COL:
168: if (ly != lg(x[1])) break;
169: z=cgetg(lx+1,tx); z[lx]=(long)y;
170: for (i=1; i<lx; i++) z[i]=x[i];
171: return z;
172: }
173: break;
174: }
175: err(concater);
176: return NULL; /* not reached */
177: }
178:
179: GEN
180: concat(GEN x, GEN y)
181: {
182: long tx = typ(x), lx,ty,ly,i;
183: GEN z,p1;
184:
185: if (!y)
186: {
187: long av = avma, tetpil;
188: if (tx == t_LIST)
189: { lx = lgef(x); i = 2; }
190: else if (tx == t_VEC)
191: { lx = lg(x); i = 1; }
192: else err(concater);
193: if (i>=lx) err(talker,"trying to concat elements of an empty vector");
194: y = (GEN)x[i++];
195: for (; i<lx; i++) y = concatsp(y, (GEN)x[i]);
196: tetpil = avma; return gerepile(av,tetpil,gcopy(y));
197: }
198: ty = typ(y);
199: if (tx==t_LIST || ty==t_LIST) return listconcat(x,y);
200: if (tx==t_STR || ty==t_STR) return strconcat(x,y);
201: lx=lg(x); ly=lg(y);
202:
203: if (tx==t_MAT && lx==1)
204: {
205: if (ty!=t_VEC || ly==1) return gtomat(y);
206: err(concater);
207: }
208: if (ty==t_MAT && ly==1)
209: {
210: if (tx!=t_VEC || lx==1) return gtomat(x);
211: err(concater);
212: }
213:
214: if (! is_matvec_t(tx))
215: {
216: if (! is_matvec_t(ty))
217: {
218: z=cgetg(3,t_VEC); z[1]=lcopy(x); z[2]=lcopy(y);
219: return z;
220: }
221: z=cgetg(ly+1,ty);
222: if (ty != t_MAT) p1 = gcopy(x);
223: else
224: {
225: if (lg(y[1])!=2) err(concater);
226: p1=cgetg(2,t_COL); p1[1]=lcopy(x);
227: }
228: for (i=2; i<=ly; i++) z[i]=lcopy((GEN) y[i-1]);
229: z[1]=(long)p1; return z;
230: }
231: if (! is_matvec_t(ty))
232: {
233: z=cgetg(lx+1,tx);
234: if (tx != t_MAT) p1 = gcopy(y);
235: else
236: {
237: if (lg(x[1])!=2) err(concater);
238: p1=cgetg(2,t_COL); p1[1]=lcopy(y);
239: }
240: for (i=1; i<lx; i++) z[i]=lcopy((GEN) x[i]);
241: z[lx]=(long)p1; return z;
242: }
243:
244: if (tx == ty)
245: {
246: if (tx == t_MAT && lg(x[1]) != lg(y[1])) err(concater);
247: z=cgetg(lx+ly-1,tx);
248: for (i=1; i<lx; i++) z[i]=lcopy((GEN) x[i]);
249: for (i=1; i<ly; i++) z[lx+i-1]=lcopy((GEN) y[i]);
250: return z;
251: }
252:
253: switch(tx)
254: {
255: case t_VEC:
256: switch(ty)
257: {
258: case t_COL:
259: if (lx<=2) return (lx==1)? gcopy(y): concat((GEN) x[1],y);
260: if (ly>=3) break;
261: return (ly==1)? gcopy(x): concat(x,(GEN) y[1]);
262: case t_MAT:
263: z=cgetg(ly,ty); if (lx != ly) break;
264: for (i=1; i<ly; i++) z[i]=lconcat((GEN) x[i],(GEN) y[i]);
265: return z;
266: }
267: break;
268:
269: case t_COL:
270: switch(ty)
271: {
272: case t_VEC:
273: if (lx<=2) return (lx==1)? gcopy(y): concat((GEN) x[1],y);
274: if (ly>=3) break;
275: return (ly==1)? gcopy(x): concat(x,(GEN) y[1]);
276: case t_MAT:
277: if (lx != lg(y[1])) break;
278: z=cgetg(ly+1,ty); z[1]=lcopy(x);
279: for (i=2; i<=ly; i++) z[i]=lcopy((GEN) y[i-1]);
280: return z;
281: }
282: break;
283:
284: case t_MAT:
285: switch(ty)
286: {
287: case t_VEC:
288: z=cgetg(lx,tx); if (ly != lx) break;
289: for (i=1; i<lx; i++) z[i]=lconcat((GEN) x[i],(GEN) y[i]);
290: return z;
291: case t_COL:
292: if (ly != lg(x[1])) break;
293: z=cgetg(lx+1,tx); z[lx]=lcopy(y);
294: for (i=1; i<lx; i++) z[i]=lcopy((GEN) x[i]);
295: return z;
296: }
297: break;
298: }
299: err(concater);
300: return NULL; /* not reached */
301: }
302:
303: static long
304: str_to_long(char *s, char **pt)
305: {
306: long a = atol(s);
307: while (isspace((int)*s)) s++;
308: if (*s == '-' || *s == '+') s++;
309: while (isdigit((int)*s) || isspace((int)*s)) s++;
310: *pt = s; return a;
311: }
312:
313: static int
314: get_range(char *s, long *a, long *b, long *compl, long lx)
315: {
316: long max = lx - 1;
317:
318: *a = 1; *b = max;
319: if (*s == '^') { *compl = 1; s++; } else *compl = 0;
320: if (*s == 0) return 0;
321: if (*s != '.')
322: {
323: *a = str_to_long(s, &s);
324: if (*a < 0) *a += lx;
325: if (*a<1 || *a>max) return 0;
326: }
327: if (*s == '.')
328: {
329: s++; if (*s != '.') return 0;
330: do s++; while (isspace((int)*s));
331: if (*s)
332: {
333: *b = str_to_long(s, &s);
334: if (*b < 0) *b += lx;
335: if (*b<1 || *b>max || *s) return 0;
336: }
337: return 1;
338: }
339: if (*s) return 0;
340: *b = *a; return 1;
341: }
342:
343: GEN
344: extract(GEN x, GEN l)
345: {
346: long av,i,j, tl = typ(l), tx = typ(x), lx = lg(x);
347: GEN y;
348:
349: if (! is_matvec_t(tx)) err(typeer,"extract");
350: if (tl==t_INT)
351: {
352: /* extract components of x as per the bits of mask l */
353: if (!signe(l)) return cgetg(1,tx);
354: av=avma; y = (GEN) gpmalloc(lx*sizeof(long));
355: i = j = 1; while (!mpodd(l)) { l=shifti(l,-1); i++; }
356: while (signe(l) && i<lx)
357: {
358: if (mod2(l)) y[j++] = x[i];
359: i++; l=shifti(l,-1);
360: }
361: if (signe(l)) err(talker,"mask too large in vecextract");
362: y[0] = evaltyp(tx) | evallg(j);
363: avma=av; x = gcopy(y); free(y); return x;
364: }
365: if (tl==t_STR)
366: {
367: char *s = GSTR(l);
368: long first, last, compl;
369: if (! get_range(s, &first, &last, &compl, lx))
370: err(talker, "incorrect range in extract");
371: if (lx == 1) return gcopy(x);
372: if (compl)
373: {
374: if (first <= last)
375: {
376: y = cgetg(lx - (last - first + 1),tx);
377: for (j=1; j<first; j++) y[j] = lcopy((GEN)x[j]);
378: for (i=last+1; i<lx; i++,j++) y[j] = lcopy((GEN)x[i]);
379: }
380: else
381: {
382: y = cgetg(lx - (first - last + 1),tx);
383: for (j=1,i=lx-1; i>first; i--,j++) y[j] = lcopy((GEN)x[i]);
384: for (i=last-1; i>0; i--,j++) y[j] = lcopy((GEN)x[i]);
385: }
386: }
387: else
388: {
389: if (first <= last)
390: {
391: y = cgetg(last-first+2,tx);
392: for (i=first,j=1; i<=last; i++,j++) y[j] = lcopy((GEN)x[i]);
393: }
394: else
395: {
396: y = cgetg(first-last+2,tx);
397: for (i=first,j=1; i>=last; i--,j++) y[j] = lcopy((GEN)x[i]);
398: }
399: }
400: return y;
401: }
402:
403: if (is_vec_t(tl))
404: {
405: long ll=lg(l); y=cgetg(ll,tx);
406: for (i=1; i<ll; i++)
407: {
408: j = itos((GEN) l[i]);
409: if (j>=lx || j<=0) err(talker,"no such component in vecextract");
410: y[i] = lcopy((GEN) x[j]);
411: }
412: return y;
413: }
414: if (tl == t_VECSMALL)
415: {
416: long ll=lg(l); y=cgetg(ll,tx);
417: for (i=1; i<ll; i++)
418: {
419: j = l[i];
420: if (j>=lx || j<=0) err(talker,"no such component in vecextract");
421: y[i] = lcopy((GEN) x[j]);
422: }
423: return y;
424: }
425: err(talker,"incorrect mask in vecextract");
426: return NULL; /* not reached */
427: }
428:
429: GEN
430: matextract(GEN x, GEN l1, GEN l2)
431: {
432: long av = avma, tetpil;
433:
434: if (typ(x)!=t_MAT) err(typeer,"matextract");
435: x = extract(gtrans(extract(x,l2)),l1); tetpil=avma;
436: return gerepile(av,tetpil, gtrans(x));
437: }
438:
439: GEN
440: extract0(GEN x, GEN l1, GEN l2)
441: {
442: if (! l2) return extract(x,l1);
443: return matextract(x,l1,l2);
444: }
445:
446: /*******************************************************************/
447: /* */
448: /* SCALAR-MATRIX OPERATIONS */
449: /* */
450: /*******************************************************************/
451:
452: /* create the square nxn matrix equal to z*Id */
453: static GEN
454: gscalmat_proto(GEN z, GEN myzero, long n, int flag)
455: {
456: long i,j;
457: GEN y = cgetg(n+1,t_MAT);
458: if (n < 0) err(talker,"negative size in scalmat");
459: if (flag) z = (flag==1)? stoi((long)z): gcopy(z);
460: for (i=1; i<=n; i++)
461: {
462: y[i]=lgetg(n+1,t_COL);
463: for (j=1; j<=n; j++)
464: coeff(y,j,i) = (i==j)? (long)z: (long)myzero;
465: }
466: return y;
467: }
468:
469: GEN
470: gscalmat(GEN x, long n) { return gscalmat_proto(x,gzero,n,2); }
471:
472: GEN
473: gscalsmat(long x, long n) { return gscalmat_proto((GEN)x,gzero,n,1); }
474:
475: GEN
476: idmat(long n) { return gscalmat_proto(gun,gzero,n,0); }
477:
478: GEN
479: idmat_intern(long n,GEN myun,GEN z) { return gscalmat_proto(myun,z,n,0); }
480:
481: GEN
482: gscalcol_proto(GEN z, GEN myzero, long n)
483: {
484: GEN y = cgetg(n+1,t_COL);
485: long i;
486:
487: if (n)
488: {
489: y[1]=(long)z;
490: for (i=2; i<=n; i++) y[i]=(long)myzero;
491: }
492: return y;
493: }
494:
495: GEN
496: zerocol(long n)
497: {
498: GEN y = cgetg(n+1,t_COL);
499: long i;
500: for (i=1; i<=n; i++) y[i]=zero;
501: return y;
502: }
503:
504: GEN
505: gscalcol(GEN x, long n) { return gscalcol_proto(gcopy(x),gzero,n); }
506:
507: GEN
508: gscalcol_i(GEN x, long n) { return gscalcol_proto(x,gzero,n); }
509:
510: GEN
511: gtomat(GEN x)
512: {
513: long tx,lx,i;
514: GEN y,p1;
515:
516: if (!x) return cgetg(1, t_MAT);
517: tx = typ(x);
518: if (! is_matvec_t(tx))
519: {
520: y=cgetg(2,t_MAT); p1=cgetg(2,t_COL); y[1]=(long)p1;
521: p1[1]=lcopy(x); return y;
522: }
523: switch(tx)
524: {
525: case t_VEC:
526: lx=lg(x); y=cgetg(lx,t_MAT);
527: for (i=1; i<lx; i++)
528: {
529: p1=cgetg(2,t_COL); y[i]=(long)p1;
530: p1[1]=lcopy((GEN) x[i]);
531: }
532: break;
533: case t_COL:
534: y=cgetg(2,t_MAT); y[1]=lcopy(x); break;
535: case t_MAT:
536: y=gcopy(x); break;
537: }
538: return y;
539: }
540:
541: long
542: isdiagonal(GEN x)
543: {
544: long nco,i,j;
545:
546: if (typ(x)!=t_MAT) err(typeer,"isdiagonal");
547: nco=lg(x)-1; if (!nco) return 1;
548: if (nco != lg(x[1])-1) return 0;
549:
550: for (j=1; j<=nco; j++)
551: {
552: GEN *col = (GEN*) x[j];
553: for (i=1; i<=nco; i++)
554: if (i!=j && !gcmp0(col[i])) return 0;
555: }
556: return 1;
557: }
558:
559: /* create the diagonal matrix, whose diagonal is given by x */
560: GEN
561: diagonal(GEN x)
562: {
563: long i,j,lx,tx=typ(x);
564: GEN y,p1;
565:
566: if (! is_matvec_t(tx)) return gscalmat(x,1);
567: if (tx==t_MAT)
568: {
569: if (isdiagonal(x)) return gcopy(x);
570: err(talker,"incorrect object in diagonal");
571: }
572: lx=lg(x); y=cgetg(lx,t_MAT);
573: for (j=1; j<lx; j++)
574: {
575: p1=cgetg(lx,t_COL); y[j]=(long)p1;
576: for (i=1; i<lx; i++)
577: p1[i] = (i==j)? lcopy((GEN) x[i]): zero;
578: }
579: return y;
580: }
581:
582: /* compute m*diagonal(d) */
583: GEN
584: matmuldiagonal(GEN m, GEN d)
585: {
586: long j=typ(d),lx=lg(m);
587: GEN y;
588:
589: if (typ(m)!=t_MAT) err(typeer,"matmuldiagonal");
590: if (! is_vec_t(j) || lg(d)!=lx)
591: err(talker,"incorrect vector in matmuldiagonal");
592: y=cgetg(lx,t_MAT);
593: for (j=1; j<lx; j++) y[j] = lmul((GEN) d[j],(GEN) m[j]);
594: return y;
595: }
596:
597: /* compute m*n assuming the result is a diagonal matrix */
598: GEN
599: matmultodiagonal(GEN m, GEN n)
600: {
601: long lx,i,j;
602: GEN s,y;
603:
604: if (typ(m)!=t_MAT || typ(n)!=t_MAT) err(typeer,"matmultodiagonal");
605: lx=lg(n); y=idmat(lx-1);
606: if (lx == 1)
607: { if (lg(m) != 1) err(consister,"matmultodiagonal"); }
608: else
609: { if (lg(m) != lg(n[1])) err(consister,"matmultodiagonal"); }
610: for (i=1; i<lx; i++)
611: {
612: s = gzero;
613: for (j=1; j<lx; j++)
614: s = gadd(s,gmul(gcoeff(m,i,j),gcoeff(n,j,i)));
615: coeff(y,i,i) = (long)s;
616: }
617: return y;
618: }
619:
620: /* [m[1,1], ..., m[l,l]] */
621: GEN
622: mattodiagonal(GEN m)
623: {
624: long i, lx = lg(m);
625: GEN y = cgetg(lx,t_VEC);
626:
627: if (typ(m)!=t_MAT) err(typeer,"mattodiagonal");
628: if (lx == 1) return y;
629: for (i=1; i<lx; i++) y[i] = lcopy(gcoeff(m,i,i));
630: return y;
631: }
632:
633: /*******************************************************************/
634: /* */
635: /* ADDITION SCALAR + MATRIX */
636: /* */
637: /*******************************************************************/
638:
639: /* create the square matrix x*Id + y */
640: GEN
641: gaddmat(GEN x, GEN y)
642: {
643: long ly,dy,i,j;
644: GEN z;
645:
646: ly=lg(y); if (ly==1) err(gadderf,"Scalar","t_MAT");
647: dy=lg(y[1]);
648: if (typ(y)!=t_MAT || ly!=dy) err(mattype1,"gaddmat");
649: z=cgetg(ly,t_MAT);
650: for (i=1; i<ly; i++)
651: {
652: z[i]=lgetg(dy,t_COL);
653: for (j=1; j<dy; j++)
654: coeff(z,j,i) = i==j? ladd(x,gcoeff(y,j,i)): lcopy(gcoeff(y,j,i));
655: }
656: return z;
657: }
658:
659: /*******************************************************************/
660: /* */
661: /* Solve A*X=B (Gauss pivot) */
662: /* */
663: /*******************************************************************/
664: #define swap(x,y) { long _t=x; x=y; y=_t; }
665:
666: /* Assume x is a non-empty matrix. Return 0 if maximal pivot should not be
667: * used, and the matrix precision (min real precision of coeffs) otherwise.
668: */
669: static long
670: matprec(GEN x)
671: {
672: long tx,i,j,l, k = VERYBIGINT, lx = lg(x), ly = lg(x[1]);
673: GEN p1;
674: for (i=1; i<lx; i++)
675: for (j=1; j<ly; j++)
676: {
677: p1 = gmael(x,i,j); tx = typ(p1);
678: if (!is_scalar_t(tx)) return 0;
679: l = precision(p1); if (l && l<k) k = l;
680: }
681: return (k==VERYBIGINT)? 0: k;
682: }
683:
684: /* As above, returning 1 if the precision would be non-zero, 0 otherwise */
685: static long
686: use_maximal_pivot(GEN x)
687: {
688: long tx,i,j, lx = lg(x), ly = lg(x[1]);
689: GEN p1;
690: for (i=1; i<lx; i++)
691: for (j=1; j<ly; j++)
692: {
693: p1 = gmael(x,i,j); tx = typ(p1);
694: if (!is_scalar_t(tx)) return 0;
695: if (precision(p1)) return 1;
696: }
697: return 0;
698: }
699:
700: static GEN
701: check_b(GEN b, long nbli)
702: {
703: GEN col;
704: if (!b) return idmat(nbli);
705: b = dummycopy(b);
706: col = (typ(b) == t_MAT)? (GEN)b[1]: b;
707: if (nbli == lg(col)-1) return b;
708: err(talker,"incompatible matrix dimensions in gauss");
709: return NULL; /* not reached */
710: }
711:
712: GEN
713: gauss_get_col(GEN a, GEN b, GEN p, long nbli)
714: {
715: GEN m, u=cgetg(nbli+1,t_COL);
716: long i,j;
717:
718: u[nbli] = ldiv((GEN) b[nbli],p);
719: for (i=nbli-1; i>0; i--)
720: {
721: m = gneg_i((GEN)b[i]);
722: for (j=i+1; j<=nbli; j++)
723: m = gadd(m, gmul(gcoeff(a,i,j),(GEN) u[j]));
724: u[i] = ldiv(gneg_i(m), gcoeff(a,i,i));
725: }
726: return u;
727: }
728:
729: /* Gauss pivot.
730: * Compute a^(-1)*b, where nblig(a) = nbcol(a) = nblig(b).
731: * b is a matrix or column vector, NULL meaning: take the identity matrix
732: * Be careful, if a or b is empty, the result is the empty matrix...
733: */
734: GEN
735: gauss(GEN a, GEN b)
736: {
737: long inexact,ismat,nbli,nbco,i,j,k,av,tetpil,lim;
738: GEN p,m,u;
739: /* nbli: nb lines of b = nb columns of a */
740: /* nbco: nb columns of b (if matrix) */
741:
742: if (typ(a)!=t_MAT) err(mattype1,"gauss");
743: if (b && typ(b)!=t_COL && typ(b)!=t_MAT) err(typeer,"gauss");
744: if (lg(a) == 1)
745: {
746: if (b && lg(b)!=1) err(consister,"gauss");
747: if (DEBUGLEVEL)
748: err(warner,"in Gauss lg(a)=%ld lg(b)=%ld",lg(a),b?lg(b):-1);
749: return cgetg(1,t_MAT);
750: }
751: av=avma; lim=stack_lim(av,1);
752: nbli = lg(a)-1; if (nbli!=lg(a[1])-1) err(mattype1,"gauss");
753: a = dummycopy(a);
754: b = check_b(b,nbli);
755: nbco = lg(b)-1;
756: inexact = use_maximal_pivot(a);
757: ismat = (typ(b)==t_MAT);
758: if(DEBUGLEVEL>4)
759: fprintferr("Entering gauss with inexact=%ld ismat=%ld\n",inexact,ismat);
760:
761: for (i=1; i<nbli; i++)
762: {
763: /* k is the line where we find the pivot */
764: p=gcoeff(a,i,i); k=i;
765: if (inexact) /* maximal pivot */
766: {
767: long e, ex = gexpo(p);
768: for (j=i+1; j<=nbli; j++)
769: {
770: e = gexpo(gcoeff(a,j,i));
771: if (e > ex) { ex=e; k=j; }
772: }
773: if (gcmp0(gcoeff(a,k,i))) err(matinv1);
774: }
775: else if (gcmp0(p)) /* first non-zero pivot */
776: {
777: do k++; while (k<=nbli && gcmp0(gcoeff(a,k,i)));
778: if (k>nbli) err(matinv1);
779: }
780:
781: /* if (k!=i), exchange the lines s.t. k = i */
782: if (k != i)
783: {
784: for (j=i; j<=nbli; j++) swap(coeff(a,i,j), coeff(a,k,j));
785: if (ismat)
786: {
787: for (j=1; j<=nbco; j++) swap(coeff(b,i,j), coeff(b,k,j));
788: }
789: else
790: swap(b[i],b[k]);
791: p = gcoeff(a,i,i);
792: }
793:
794: for (k=i+1; k<=nbli; k++)
795: {
796: m=gcoeff(a,k,i);
797: if (!gcmp0(m))
798: {
799: m = gneg_i(gdiv(m,p));
800: for (j=i+1; j<=nbli; j++)
801: {
802: u = gmul(m,gcoeff(a,i,j));
803: coeff(a,k,j) = ladd(gcoeff(a,k,j),u);
804: }
805: if (ismat) for (j=1; j<=nbco; j++)
806: {
807: u = gmul(m,gcoeff(b,i,j));
808: coeff(b,k,j) = ladd(gcoeff(b,k,j),u);
809: }
810: else
811: {
812: u = gmul(m,(GEN) b[i]);
813: b[k] = ladd((GEN) b[k],u);
814: }
815: }
816: }
817: if (low_stack(lim, stack_lim(av,1)))
818: {
819: GEN *gptr[2];
820: if(DEBUGMEM>1) err(warnmem,"gauss. i=%ld",i);
821: gptr[0]=&a; gptr[1]=&b;
822: gerepilemany(av,gptr,2);
823: }
824: }
825:
826: if(DEBUGLEVEL>4) fprintferr("Solving the triangular system\n");
827: p=gcoeff(a,nbli,nbli);
828: if (!inexact && gcmp0(p)) err(matinv1);
829: if (!ismat) u = gauss_get_col(a,b,p,nbli);
830: else
831: {
832: long av1 = avma;
833: lim = stack_lim(av1,1); u=cgetg(nbco+1,t_MAT);
834: for (j=2; j<=nbco; j++) u[j] = zero; /* dummy */
835: for (j=1; j<=nbco; j++)
836: {
837: u[j] = (long)gauss_get_col(a,(GEN)b[j],p,nbli);
838: if (low_stack(lim, stack_lim(av1,1)))
839: {
840: if(DEBUGMEM>1) err(warnmem,"gauss[2]. j=%ld", j);
841: tetpil=avma; u = gerepile(av1,tetpil,gcopy(u));
842: }
843: }
844: }
845: tetpil=avma; return gerepile(av,tetpil,gcopy(u));
846: }
847:
848: /* x a matrix with integer coefficients. Return a multiple of the determinant
849: * of the lattice generated by the columns of x (to be used with hnfmod)
850: */
851: GEN
852: detint(GEN x)
853: {
854: GEN pass,c,v,det1,piv,pivprec,vi,p1;
855: long i,j,k,rg,n,m,m1,av=avma,av1,lim;
856:
857: if (typ(x)!=t_MAT) err(typeer,"detint");
858: n=lg(x)-1; if (!n) return gun;
859: m1=lg(x[1]); m=m1-1; lim=stack_lim(av,1);
860: c=new_chunk(m1); for (k=1; k<=m; k++) c[k]=0;
861: av1=avma; pass=cgetg(m1,t_MAT);
862: for (j=1; j<=m; j++)
863: {
864: p1=cgetg(m1,t_COL); pass[j]=(long)p1;
865: for (i=1; i<=m; i++) p1[i]=zero;
866: }
867: v=cgetg(m1,t_COL);
868: det1=gzero; piv=pivprec=gun;
869: for (rg=0,k=1; k<=n; k++)
870: {
871: long t = 0;
872: for (i=1; i<=m; i++)
873: if (!c[i])
874: {
875: vi=mulii(piv,gcoeff(x,i,k));
876: for (j=1; j<=m; j++)
877: if (c[j]) vi=addii(vi,mulii(gcoeff(pass,i,j),gcoeff(x,j,k)));
878: v[i]=(long)vi; if (!t && signe(vi)) t=i;
879: }
880: if (t)
881: {
882: if (rg == m-1)
883: { det1=mppgcd((GEN)v[t],det1); c[t]=0; }
884: else
885: {
886: rg++; pivprec = piv; piv=(GEN)v[t]; c[t]=k;
887: for (i=1; i<=m; i++)
888: if (!c[i])
889: {
890: GEN p2 = negi((GEN)v[i]);
891: for (j=1; j<=m; j++)
892: if (c[j] && j!=t)
893: {
894: p1 = addii(mulii(gcoeff(pass,i,j), piv),
895: mulii(gcoeff(pass,t,j), p2));
896: if (rg>1) p1 = divii(p1,pivprec);
897: coeff(pass,i,j) = (long)p1;
898: }
899: coeff(pass,i,t) = (long)p2;
900: }
901: }
902: }
903: if (low_stack(lim, stack_lim(av,1)))
904: {
905: GEN *gptr[5];
906: if(DEBUGMEM>1) err(warnmem,"detint. k=%ld",k);
907: gptr[0]=&det1; gptr[1]=ϖ gptr[2]=&pivprec;
908: gptr[3]=&pass; gptr[4]=&v; gerepilemany(av1,gptr,5);
909: }
910: }
911: return gerepileupto(av, absi(det1));
912: }
913:
914: static void
915: gerepile_gauss_keep(GEN x, long m, long n, long k, long t, long av)
916: {
917: long tetpil = avma,dec,u,A,i;
918:
919: if (DEBUGMEM > 1) err(warnmem,"gauss_pivot_keep. k=%ld, n=%ld",k,n);
920: for (u=t+1; u<=m; u++) copyifstack(coeff(x,u,k), coeff(x,u,k));
921: for (i=k+1; i<=n; i++)
922: for (u=1; u<=m; u++) copyifstack(coeff(x,u,i), coeff(x,u,i));
923:
924: (void)gerepile(av,tetpil,NULL); dec = av-tetpil;
925: for (u=t+1; u<=m; u++)
926: {
927: A=coeff(x,u,k);
928: if (A<av && A>=bot) coeff(x,u,k)+=dec;
929: }
930: for (i=k+1; i<=n; i++)
931: for (u=1; u<=m; u++)
932: {
933: A=coeff(x,u,i);
934: if (A<av && A>=bot) coeff(x,u,i)+=dec;
935: }
936: }
937:
938: static void
939: gerepile_gauss_keep_mod_p(GEN x, GEN p, long m, long n, long k, long t, long av)
940: {
941: long tetpil = avma,dec,u,A,i;
942:
943: if (DEBUGMEM > 1) err(warnmem,"gauss_pivot_keep. k=%ld, n=%ld",k,n);
944: for (u=t+1; u<=m; u++)
945: if (isonstack(coeff(x,u,k))) coeff(x,u,k) = lmodii(gcoeff(x,u,k),p);
946: for (i=k+1; i<=n; i++)
947: for (u=1; u<=m; u++)
948: if (isonstack(coeff(x,u,i))) coeff(x,u,i) = lmodii(gcoeff(x,u,i),p);
949:
950: (void)gerepile(av,tetpil,NULL); dec = av-tetpil;
951: for (u=t+1; u<=m; u++)
952: {
953: A=coeff(x,u,k);
954: if (A<av && A>=bot) coeff(x,u,k)+=dec;
955: }
956: for (i=k+1; i<=n; i++)
957: for (u=1; u<=m; u++)
958: {
959: A=coeff(x,u,i);
960: if (A<av && A>=bot) coeff(x,u,i)+=dec;
961: }
962: }
963:
964: /* special gerepile for huge matrices */
965:
966: static void
967: gerepile_gauss(GEN x,long m,long n,long k,long t,long av, long j, GEN c)
968: {
969: long tetpil = avma,dec,u,A,i;
970:
971: if (DEBUGMEM > 1) err(warnmem,"gauss_pivot. k=%ld, n=%ld",k,n);
972: for (u=t+1; u<=m; u++)
973: if (u==j || !c[u]) copyifstack(coeff(x,u,k), coeff(x,u,k));
974: for (u=1; u<=m; u++)
975: if (u==j || !c[u])
976: for (i=k+1; i<=n; i++) copyifstack(coeff(x,u,i), coeff(x,u,i));
977:
978: (void)gerepile(av,tetpil,NULL); dec = av-tetpil;
979: for (u=t+1; u<=m; u++)
980: if (u==j || !c[u])
981: {
982: A=coeff(x,u,k);
983: if (A<av && A>=bot) coeff(x,u,k)+=dec;
984: }
985: for (u=1; u<=m; u++)
986: if (u==j || !c[u])
987: for (i=k+1; i<=n; i++)
988: {
989: A=coeff(x,u,i);
990: if (A<av && A>=bot) coeff(x,u,i)+=dec;
991: }
992: }
993:
994: /*******************************************************************/
995: /* */
996: /* KERNEL of an m x n matrix */
997: /* return n - rk(x) linearly independant vectors */
998: /* */
999: /*******************************************************************/
1000:
1001: /* x has INTEGER coefficients */
1002: GEN
1003: keri(GEN x)
1004: {
1005: GEN c,d,y,p,pp;
1006: long i,j,k,r,t,n,m,av,av0,tetpil,lim;
1007:
1008: if (typ(x)!=t_MAT) err(typeer,"keri");
1009: n=lg(x)-1; if (!n) return cgetg(1,t_MAT);
1010:
1011: av0=avma; m=lg(x[1])-1; r=0;
1012: pp=cgetg(n+1,t_COL);
1013: x=dummycopy(x); p=gun;
1014: c=new_chunk(m+1); for (k=1; k<=m; k++) c[k]=0;
1015: d=new_chunk(n+1); av=avma; lim=stack_lim(av,1);
1016: for (k=1; k<=n; k++)
1017: {
1018: j=1;
1019: while (j<=m && (c[j] || !signe(gcoeff(x,j,k))) ) j++;
1020: if (j>m)
1021: {
1022: r++; d[k]=0;
1023: for(j=1; j<k; j++)
1024: if (d[j]) coeff(x,d[j],k) = lclone(gcoeff(x,d[j],k));
1025: pp[k]=lclone(p);
1026: }
1027: else
1028: {
1029: GEN p0 = p;
1030: long av1;
1031:
1032: c[j]=k; d[k]=j; p = gcoeff(x,j,k);
1033:
1034: for (t=1; t<=m; t++)
1035: if (t!=j)
1036: {
1037: GEN q=gcoeff(x,t,k), p1,p2;
1038: for (i=k+1; i<=n; i++)
1039: {
1040: av1=avma; (void)new_chunk(lgefint(p0));
1041: p1=mulii(q,gcoeff(x,j,i));
1042: p2=mulii(p,gcoeff(x,t,i));
1043: p1=subii(p2,p1); avma=av1;
1044: coeff(x,t,i) = ldivii(p1,p0);
1045: }
1046: if (low_stack(lim, stack_lim(av,1)))
1047: {
1048: p1 = gclone(p);
1049: gerepile_gauss_keep(x,m,n,k,t,av);
1050: p = gcopy(p1); gunclone(p1);
1051: }
1052: }
1053: }
1054: }
1055: if (!r) { avma=av0; y=cgetg(1,t_MAT); return y; }
1056:
1057: /* non trivial kernel */
1058: tetpil=avma; y=cgetg(r+1,t_MAT);
1059: for (j=k=1; j<=r; j++,k++)
1060: {
1061: p = cgetg(n+1, t_COL);
1062: y[j]=(long)p; while (d[k]) k++;
1063: for (i=1; i<k; i++)
1064: if (d[i])
1065: {
1066: c=gcoeff(x,d[i],k);
1067: p[i] = (long) forcecopy(c); gunclone(c);
1068: }
1069: else
1070: p[i] = zero;
1071: p[k]=lnegi((GEN)pp[k]); gunclone((GEN)pp[k]);
1072: for (i=k+1; i<=n; i++) p[i]=zero;
1073: }
1074: return gerepile(av0,tetpil,y);
1075: }
1076:
1077: GEN
1078: deplin(GEN x)
1079: {
1080: long i,j,k,t,nc,nl, av=avma;
1081: GEN y,q,c,l,d;
1082:
1083: if (typ(x) != t_MAT) err(typeer,"deplin");
1084: nc=lg(x)-1; if (!nc) err(talker,"empty matrix in deplin");
1085: nl=lg(x[1])-1;
1086: c=new_chunk(nl+1);
1087: l=new_chunk(nc+1);
1088: d=cgetg(nl+1,t_VEC);
1089: for (i=1; i<=nl; i++) { d[i]=un; c[i]=0; }
1090: k=1; t=1;
1091: while (t<=nl && k<=nc)
1092: {
1093: for (j=1; j<k; j++)
1094: for (i=1; i<=nl; i++)
1095: if (i!=l[j])
1096: coeff(x,i,k)=lsub(gmul((GEN) d[j],gcoeff(x,i,k)),
1097: gmul(gcoeff(x,i,j),gcoeff(x,l[j],k)));
1098: t=1;
1099: while ( t<=nl && (c[t] || gcmp0(gcoeff(x,t,k))) ) t++;
1100: if (t<=nl)
1101: {
1102: d[k]=coeff(x,t,k);
1103: c[t]=k; l[k++]=t;
1104: }
1105: }
1106: if (k>nc)
1107: {
1108: avma=av; y=cgetg(nc+1,t_COL);
1109: for (j=1; j<=nc; j++) y[j]=zero;
1110: return y;
1111: }
1112: y=cgetg(nc+1,t_COL);
1113: y[1]=(k>1)? coeff(x,l[1],k): un;
1114: for (q=gun,j=2; j<k; j++)
1115: {
1116: q=gmul(q,(GEN) d[j-1]);
1117: y[j]=lmul(gcoeff(x,l[j],k),q);
1118: }
1119: if (k>1) y[k]=lneg(gmul(q,(GEN) d[k-1]));
1120: for (j=k+1; j<=nc; j++) y[j]=zero;
1121: d=content(y); t=avma;
1122: return gerepile(av,t,gdiv(y,d));
1123: }
1124:
1125: /*******************************************************************/
1126: /* */
1127: /* GAUSS REDUCTION OF MATRICES (m lines x n cols) */
1128: /* (kernel, image, complementary image, rank) */
1129: /* */
1130: /*******************************************************************/
1131: static long gauss_ex;
1132: static int (*gauss_is_zero)(GEN);
1133:
1134: static int
1135: real0(GEN x)
1136: {
1137: return gcmp0(x) || (gexpo(x) < gauss_ex);
1138: }
1139:
1140: static void
1141: gauss_get_prec(GEN x, long prec)
1142: {
1143: long pr = matprec(x);
1144:
1145: if (!pr) { gauss_is_zero = &gcmp0; return; }
1146: if (pr > prec) prec = pr;
1147:
1148: gauss_ex = - (long)(0.85 * bit_accuracy(prec));
1149: gauss_is_zero = &real0;
1150: }
1151:
1152: /* return the transform of x under a standard Gauss pivot. r = dim ker(x).
1153: * d[k] contains the index of the first non-zero pivot in column k
1154: */
1155: static GEN
1156: gauss_pivot_keep(GEN x, long prec, GEN *dd, long *rr)
1157: {
1158: GEN c,d,p,mun;
1159: long i,j,k,r,t,n,m,av,lim;
1160:
1161: if (typ(x)!=t_MAT) err(typeer,"gauss_pivot");
1162: n=lg(x)-1; if (!n) { *dd=NULL; *rr=0; return cgetg(1,t_MAT); }
1163:
1164: gauss_get_prec(x,prec); m=lg(x[1])-1; r=0;
1165: x=dummycopy(x); mun=negi(gun);
1166: c=new_chunk(m+1); for (k=1; k<=m; k++) c[k]=0;
1167: d=(GEN)gpmalloc((n+1)*sizeof(long));
1168: av=avma; lim=stack_lim(av,1);
1169: for (k=1; k<=n; k++)
1170: {
1171: j=1; while (j<=m && (c[j] || gauss_is_zero(gcoeff(x,j,k)))) j++;
1172: if (j>m)
1173: {
1174: r++; d[k]=0;
1175: for(j=1; j<k; j++)
1176: if (d[j]) coeff(x,d[j],k) = lclone(gcoeff(x,d[j],k));
1177: }
1178: else
1179: {
1180: c[j]=k; d[k]=j; p = gdiv(mun,gcoeff(x,j,k));
1181: coeff(x,j,k) = (long)mun;
1182: for (i=k+1; i<=n; i++)
1183: coeff(x,j,i) = lmul(p,gcoeff(x,j,i));
1184: for (t=1; t<=m; t++)
1185: if (t!=j)
1186: {
1187: p=gcoeff(x,t,k); coeff(x,t,k)=zero;
1188: for (i=k+1; i<=n; i++)
1189: coeff(x,t,i) = ladd(gcoeff(x,t,i),gmul(p,gcoeff(x,j,i)));
1190: if (low_stack(lim, stack_lim(av,1)))
1191: gerepile_gauss_keep(x,m,n,k,t,av);
1192: }
1193: }
1194: }
1195: *dd=d; *rr=r; return x;
1196: }
1197:
1198: /* r = dim ker(x).
1199: * d[k] contains the index of the first non-zero pivot in column k
1200: */
1201: static void
1202: gauss_pivot(GEN x, long prec, GEN *dd, long *rr)
1203: {
1204: GEN c,d,mun,p;
1205: long i,j,k,r,t,n,m,av,lim;
1206:
1207: if (typ(x)!=t_MAT) err(typeer,"gauss_pivot");
1208: n=lg(x)-1; if (!n) { *dd=NULL; *rr=0; return; }
1209:
1210: gauss_get_prec(x,prec); m=lg(x[1])-1; r=0;
1211: x=dummycopy(x); mun=negi(gun);
1212: c=new_chunk(m+1); for (k=1; k<=m; k++) c[k]=0;
1213: d=(GEN)gpmalloc((n+1)*sizeof(long)); av=avma; lim=stack_lim(av,1);
1214: for (k=1; k<=n; k++)
1215: {
1216: j=1; while (j<=m && (c[j] || gauss_is_zero(gcoeff(x,j,k)))) j++;
1217: if (j>m) { r++; d[k]=0; }
1218: else
1219: {
1220: c[j]=k; d[k]=j; p = gdiv(mun,gcoeff(x,j,k));
1221: for (i=k+1; i<=n; i++)
1222: coeff(x,j,i) = lmul(p,gcoeff(x,j,i));
1223:
1224: for (t=1; t<=m; t++)
1225: if (!c[t]) /* no pivot on that line yet */
1226: {
1227: p=gcoeff(x,t,k); coeff(x,t,k)=zero;
1228: for (i=k+1; i<=n; i++)
1229: coeff(x,t,i) = ladd(gcoeff(x,t,i), gmul(p,gcoeff(x,j,i)));
1230: if (low_stack(lim, stack_lim(av,1)))
1231: gerepile_gauss(x,m,n,k,t,av,j,c);
1232: }
1233: for (i=k; i<=n; i++) coeff(x,j,i) = zero; /* dummy */
1234: }
1235: }
1236: *dd=d; *rr=r;
1237: }
1238:
1239: static GEN
1240: ker0(GEN x, long prec)
1241: {
1242: GEN d,y;
1243: long i,j,k,r,n, av = avma, tetpil;
1244:
1245: x=gauss_pivot_keep(x,prec,&d,&r);
1246: if (!r)
1247: {
1248: avma=av; if (d) free(d);
1249: return cgetg(1,t_MAT);
1250: }
1251: n = lg(x)-1; tetpil=avma; y=cgetg(r+1,t_MAT);
1252: for (j=k=1; j<=r; j++,k++)
1253: {
1254: GEN p = cgetg(n+1,t_COL);
1255:
1256: y[j]=(long)p; while (d[k]) k++;
1257: for (i=1; i<k; i++)
1258: if (d[i])
1259: {
1260: GEN p1=gcoeff(x,d[i],k);
1261: p[i] = (long)forcecopy(p1); gunclone(p1);
1262: }
1263: else
1264: p[i] = zero;
1265: p[k]=un; for (i=k+1; i<=n; i++) p[i]=zero;
1266: }
1267: free(d); return gerepile(av,tetpil,y);
1268: }
1269:
1270: GEN
1271: ker(GEN x) /* Programme pour types exacts */
1272: {
1273: return ker0(x,0);
1274: }
1275:
1276: GEN
1277: matker0(GEN x,long flag)
1278: {
1279: return flag? keri(x): ker(x);
1280: }
1281:
1282: static GEN
1283: image0(GEN x, long prec)
1284: {
1285: GEN d,y;
1286: long j,k,r, av = avma;
1287:
1288: gauss_pivot(x,prec,&d,&r);
1289:
1290: /* r = dim ker(x) */
1291: if (!r) { avma=av; if (d) free(d); return gcopy(x); }
1292:
1293: /* r = dim Im(x) */
1294: r = lg(x)-1 - r; avma=av;
1295: y=cgetg(r+1,t_MAT);
1296: for (j=k=1; j<=r; k++)
1297: if (d[k]) y[j++] = lcopy((GEN)x[k]);
1298: free(d); return y;
1299: }
1300:
1301: GEN
1302: image(GEN x) /* Programme pour types exacts */
1303: {
1304: return image0(x,0);
1305: }
1306:
1307: GEN
1308: imagereel(GEN x, long prec) /* Programme pour types inexacts */
1309: {
1310: return image0(x,prec);
1311: }
1312:
1313: static GEN
1314: imagecompl0(GEN x, long prec)
1315: {
1316: GEN d,y;
1317: long j,i,r,av = avma;
1318:
1319: gauss_pivot(x,prec,&d,&r);
1320: avma=av; y=cgetg(r+1,t_VEC);
1321: for (i=j=1; j<=r; i++)
1322: if (!d[i]) y[j++]=lstoi(i);
1323: if (d) free(d); return y;
1324: }
1325:
1326: /* for hnfspec: imagecompl(trans(x)) + image(trans(x)) */
1327: static GEN
1328: imagecomplspec(GEN x, long *nlze)
1329: {
1330: GEN d,y;
1331: long i,j,k,l,r,av = avma;
1332:
1333: x = gtrans(x); l = lg(x);
1334: gauss_pivot(x,0,&d,&r);
1335: avma=av; y = cgetg(l,t_VECSMALL);
1336: for (i=j=1, k=r+1; i<l; i++)
1337: if (d[i]) y[k++]=i; else y[j++]=i;
1338: *nlze = r;
1339: if (d) free(d); return y;
1340: }
1341:
1342: GEN
1343: imagecompl(GEN x) /* Programme pour types exacts */
1344: {
1345: return imagecompl0(x,0);
1346: }
1347:
1348: static GEN
1349: sinverseimage(GEN mat, GEN y)
1350: {
1351: long av=avma,tetpil,i, nbcol = lg(mat);
1352: GEN col,p1 = cgetg(nbcol+1,t_MAT);
1353:
1354: if (nbcol==1) return NULL;
1355: if (lg(y) != lg(mat[1])) err(consister,"inverseimage");
1356:
1357: p1[nbcol] = (long)y;
1358: for (i=1; i<nbcol; i++) p1[i]=mat[i];
1359: p1 = ker(p1); i=lg(p1)-1;
1360: if (!i) return NULL;
1361:
1362: col = (GEN)p1[i]; p1 = (GEN) col[nbcol];
1363: if (gcmp0(p1)) return NULL;
1364:
1365: p1 = gneg_i(p1); setlg(col,nbcol); tetpil=avma;
1366: return gerepile(av,tetpil, gdiv(col, p1));
1367: }
1368:
1369: /* Calcule l'image reciproque de v par m */
1370: GEN
1371: inverseimage(GEN m,GEN v)
1372: {
1373: long av=avma,j,lv,tv=typ(v);
1374: GEN y,p1;
1375:
1376: if (typ(m)!=t_MAT) err(typeer,"inverseimage");
1377: if (tv==t_COL)
1378: {
1379: p1 = sinverseimage(m,v);
1380: if (p1) return p1;
1381: avma = av; return cgetg(1,t_MAT);
1382: }
1383: if (tv!=t_MAT) err(typeer,"inverseimage");
1384:
1385: lv=lg(v)-1; y=cgetg(lv+1,t_MAT);
1386: for (j=1; j<=lv; j++)
1387: {
1388: p1 = sinverseimage(m,(GEN)v[j]);
1389: if (!p1) { avma = av; return cgetg(1,t_MAT); }
1390: y[j] = (long)p1;
1391: }
1392: return y;
1393: }
1394:
1395: /* x is an n x k matrix, rank(x) = k <= n. Return an invertible n x n matrix
1396: * whose first k columns are given by x. If rank(x)<k, the result may be wrong
1397: */
1398: GEN
1399: suppl_intern(GEN x, GEN myid)
1400: {
1401: long av = avma, lx = lg(x), n,i,j;
1402: GEN y,p1;
1403: stackzone *zone;
1404:
1405: if (typ(x) != t_MAT) err(typeer,"suppl");
1406: if (lx==1) err(talker,"empty matrix in suppl");
1407: n=lg(x[1]); if (lx>n) err(suppler2);
1408:
1409: zone = switch_stack(NULL, n*n);
1410: switch_stack(zone,1);
1411: y = myid? dummycopy(myid): idmat(n-1);
1412: switch_stack(zone,0);
1413: for (i=1; i<lx; i++)
1414: {
1415: p1=gauss(y,(GEN)x[i]); j=i;
1416: while (j<n && gcmp0((GEN)p1[j])) j++;
1417: if (j>=n) err(suppler2);
1418: p1=(GEN)y[i]; y[i]=x[i]; if (i!=j) y[j]=(long)p1;
1419: }
1420: avma = av; y = gcopy(y);
1421: free(zone); return y;
1422: }
1423:
1424: GEN
1425: suppl(GEN x)
1426: {
1427: return suppl_intern(x,NULL);
1428: }
1429:
1430: GEN
1431: image2(GEN x)
1432: {
1433: long av=avma,tetpil,k,n,i;
1434: GEN p1,p2;
1435:
1436: if (typ(x)!=t_MAT) err(typeer,"image2");
1437: k=lg(x)-1; if (!k) return gcopy(x);
1438: n=lg(x[1])-1; p1=ker(x); k=lg(p1)-1;
1439: if (k) { p1=suppl(p1); n=lg(p1)-1; }
1440: else p1=idmat(n);
1441:
1442: tetpil=avma; p2=cgetg(n-k+1,t_MAT);
1443: for (i=k+1; i<=n; i++) p2[i-k]=lmul(x,(GEN) p1[i]);
1444: return gerepile(av,tetpil,p2);
1445: }
1446:
1447: GEN
1448: matimage0(GEN x,long flag)
1449: {
1450: switch(flag)
1451: {
1452: case 0: return image(x);
1453: case 1: return image2(x);
1454: default: err(flagerr,"matimage");
1455: }
1456: return NULL; /* not reached */
1457: }
1458:
1459: long
1460: rank(GEN x)
1461: {
1462: long av = avma, r;
1463: GEN d;
1464:
1465: gauss_pivot(x,0,&d,&r);
1466: /* yield r = dim ker(x) */
1467:
1468: avma=av; if (d) free(d);
1469: return lg(x)-1 - r;
1470: }
1471:
1472: GEN
1473: indexrank(GEN x)
1474: {
1475: long av = avma, i,j,n,r;
1476: GEN res,d,p1,p2;
1477:
1478: /* yield r = dim ker(x) */
1479: gauss_pivot(x,0,&d,&r);
1480:
1481: /* now r = dim Im(x) */
1482: n = lg(x)-1; r = n - r;
1483:
1484: avma=av; res=cgetg(3,t_VEC);
1485: p1=cgetg(r+1,t_VEC); res[1]=(long)p1;
1486: p2=cgetg(r+1,t_VEC); res[2]=(long)p2;
1487: if (d)
1488: {
1489: for (i=0,j=1; j<=n; j++)
1490: if (d[j]) { i++; p1[i]=d[j]; p2[i]=j; }
1491: free(d);
1492: qsort(p1+1,r,sizeof(long),(QSCOMP)pari_compare_long);
1493: }
1494: for (i=1;i<=r;i++) { p1[i]=lstoi(p1[i]); p2[i]=lstoi(p2[i]); }
1495: return res;
1496: }
1497:
1498: /*******************************************************************/
1499: /* */
1500: /* LINEAR ALGEBRA MODULO P */
1501: /* */
1502: /*******************************************************************/
1503: #ifdef LONG_IS_64BIT
1504: # define MASK (0x7fffffff00000000UL)
1505: #else
1506: # define MASK (0x7fff0000UL)
1507: #endif
1508: static GEN
1509: ker_mod_p_small(GEN x, GEN pp, long nontriv)
1510: {
1511: GEN y,c,d;
1512: long a,i,j,k,r,t,n,m,av0,tetpil, p = pp[2], piv;
1513:
1514: n = lg(x)-1;
1515: m=lg(x[1])-1; r=0; av0 = avma;
1516: x = dummycopy(x);
1517: for (i=1; i<=n; i++)
1518: {
1519: GEN p1 = (GEN)x[i];
1520: for (j=1; j<=m; j++) p1[j] = itos((GEN)p1[j]);
1521: }
1522: c=new_chunk(m+1); for (k=1; k<=m; k++) c[k]=0;
1523: d=new_chunk(n+1);
1524: for (k=1; k<=n; k++)
1525: {
1526: for (j=1; j<=m; j++)
1527: if (!c[j])
1528: {
1529: a = coeff(x,j,k) % p;
1530: if (a) break;
1531: }
1532: if (j>m)
1533: {
1534: if (nontriv) { avma=av0; return NULL; }
1535: r++; d[k]=0;
1536: }
1537: else
1538: {
1539: c[j]=k; d[k]=j;
1540: {
1541: long av1 = avma;
1542: GEN p1 = mpinvmod(stoi(a), pp);
1543: piv = -itos(p1); avma = av1;
1544: }
1545: coeff(x,j,k) = -1;
1546: for (i=k+1; i<=n; i++)
1547: coeff(x,j,i) = (piv * coeff(x,j,i)) % p;
1548: for (t=1; t<=m; t++)
1549: if (t!=j)
1550: {
1551: piv = coeff(x,t,k) % p;
1552: if (piv)
1553: {
1554: coeff(x,t,k) = 0;
1555: for (i=k+1; i<=n; i++)
1556: {
1557: a = coeff(x,t,i) + piv * coeff(x,j,i);
1558: if (a & MASK) a %= p;
1559: coeff(x,t,i) = a;
1560: }
1561: }
1562: }
1563: }
1564: }
1565:
1566: tetpil=avma; y=cgetg(r+1,t_MAT);
1567: for (j=k=1; j<=r; j++,k++)
1568: {
1569: GEN c = cgetg(n+1,t_COL);
1570:
1571: y[j]=(long)c; while (d[k]) k++;
1572: for (i=1; i<k; i++)
1573: if (d[i])
1574: {
1575: long a = coeff(x,d[i],k) % p;
1576: if (a < 0) a += p;
1577: c[i] = lstoi(a);
1578: }
1579: else
1580: c[i] = zero;
1581: c[k]=un; for (i=k+1; i<=n; i++) c[i]=zero;
1582: }
1583: return gerepile(av0,tetpil,y);
1584: }
1585:
1586: static GEN
1587: ker_mod_p_i(GEN x, GEN p, long nontriv)
1588: {
1589: GEN y,c,d,piv,mun;
1590: long i,j,k,r,t,n,m,av0,av,lim,tetpil;
1591:
1592: if (typ(x)!=t_MAT) err(typeer,"ker_mod_p");
1593: n=lg(x)-1; if (!n) return cgetg(1,t_MAT);
1594: if (!is_bigint(p) && p[2] < (MAXHALFULONG>>1))
1595: return ker_mod_p_small(x, p, nontriv);
1596:
1597: m=lg(x[1])-1; r=0; av0 = avma;
1598: x=dummycopy(x); mun=negi(gun);
1599: c=new_chunk(m+1); for (k=1; k<=m; k++) c[k]=0;
1600: d=new_chunk(n+1);
1601: av=avma; lim=stack_lim(av,1);
1602: for (k=1; k<=n; k++)
1603: {
1604: for (j=1; j<=m; j++)
1605: if (!c[j])
1606: {
1607: coeff(x,j,k) = lmodii(gcoeff(x,j,k), p);
1608: if (signe(coeff(x,j,k))) break;
1609: }
1610: if (j>m)
1611: {
1612: if (nontriv) { avma = av0; return NULL; }
1613: r++; d[k]=0;
1614: for(j=1; j<k; j++)
1615: if (d[j]) coeff(x,d[j],k) = lclone(gcoeff(x,d[j],k));
1616: }
1617: else
1618: {
1619: c[j]=k; d[k]=j; piv = negi(mpinvmod(gcoeff(x,j,k), p));
1620: coeff(x,j,k) = (long)mun;
1621: for (i=k+1; i<=n; i++)
1622: coeff(x,j,i) = lmodii(mulii(piv,gcoeff(x,j,i)), p);
1623: for (t=1; t<=m; t++)
1624: if (t!=j)
1625: {
1626: piv = modii(gcoeff(x,t,k), p);
1627: if (signe(piv))
1628: {
1629: coeff(x,t,k)=zero;
1630: for (i=k+1; i<=n; i++)
1631: coeff(x,t,i) = laddii(gcoeff(x,t,i),mulii(piv,gcoeff(x,j,i)));
1632: if (low_stack(lim, stack_lim(av,1)))
1633: gerepile_gauss_keep_mod_p(x,p,m,n,k,t,av);
1634: }
1635: }
1636: }
1637: }
1638:
1639: tetpil=avma; y=cgetg(r+1,t_MAT);
1640: for (j=k=1; j<=r; j++,k++)
1641: {
1642: GEN c = cgetg(n+1,t_COL);
1643:
1644: y[j]=(long)c; while (d[k]) k++;
1645: for (i=1; i<k; i++)
1646: if (d[i])
1647: {
1648: GEN p1=gcoeff(x,d[i],k);
1649: c[i] = lmodii(p1, p); gunclone(p1);
1650: }
1651: else
1652: c[i] = zero;
1653: c[k]=un; for (i=k+1; i<=n; i++) c[i]=zero;
1654: }
1655: return gerepile(av0,tetpil,y);
1656: }
1657:
1658: static void
1659: gauss_pivot_mod_p(GEN x, GEN p, GEN *dd, long *rr)
1660: {
1661: GEN c,d,piv;
1662: long i,j,k,r,t,n,m,av,lim;
1663:
1664: if (typ(x)!=t_MAT) err(typeer,"gauss_pivot_mod_p");
1665: n=lg(x)-1; if (!n) { *dd=NULL; *rr=0; return; }
1666:
1667: m=lg(x[1])-1; r=0;
1668: x=dummycopy(x);
1669: c=new_chunk(m+1); for (k=1; k<=m; k++) c[k]=0;
1670: d=(GEN)gpmalloc((n+1)*sizeof(long)); av=avma; lim=stack_lim(av,1);
1671: for (k=1; k<=n; k++)
1672: {
1673: for (j=1; j<=m; j++)
1674: if (!c[j])
1675: {
1676: coeff(x,j,k) = lmodii(gcoeff(x,j,k), p);
1677: if (signe(coeff(x,j,k))) break;
1678: }
1679: if (j>m) { r++; d[k]=0; }
1680: else
1681: {
1682: c[j]=k; d[k]=j; piv = negi(mpinvmod(gcoeff(x,j,k), p));
1683: for (i=k+1; i<=n; i++)
1684: coeff(x,j,i) = lmodii(mulii(piv,gcoeff(x,j,i)), p);
1685: for (t=1; t<=m; t++)
1686: if (!c[t]) /* no pivot on that line yet */
1687: {
1688: piv=gcoeff(x,t,k);
1689: if (signe(piv))
1690: {
1691: coeff(x,t,k)=zero;
1692: for (i=k+1; i<=n; i++)
1693: coeff(x,t,i) = laddii(gcoeff(x,t,i), mulii(piv,gcoeff(x,j,i)));
1694: if (low_stack(lim, stack_lim(av,1)))
1695: gerepile_gauss(x,m,n,k,t,av,j,c);
1696: }
1697: }
1698: for (i=k; i<=n; i++) coeff(x,j,i) = zero; /* dummy */
1699: }
1700: }
1701: *dd=d; *rr=r;
1702: }
1703:
1704: GEN
1705: ker_mod_p(GEN x, GEN p)
1706: {
1707: return ker_mod_p_i(x, p, 0);
1708: }
1709:
1710: int
1711: ker_trivial_mod_p(GEN x, GEN p)
1712: {
1713: return ker_mod_p_i(x, p, 1)==NULL;
1714: }
1715:
1716: GEN
1717: image_mod_p(GEN x, GEN p)
1718: {
1719: GEN d,y;
1720: long j,k,r, av = avma;
1721:
1722: gauss_pivot_mod_p(x,p,&d,&r);
1723:
1724: /* r = dim ker(x) */
1725: if (!r) { avma=av; if (d) free(d); return gcopy(x); }
1726:
1727: /* r = dim Im(x) */
1728: r = lg(x)-1 - r; avma=av;
1729: y=cgetg(r+1,t_MAT);
1730: for (j=k=1; j<=r; k++)
1731: if (d[k]) y[j++] = lcopy((GEN)x[k]);
1732: free(d); return y;
1733: }
1734:
1735: /*******************************************************************/
1736: /* */
1737: /* EIGENVECTORS */
1738: /* (independent eigenvectors, sorted by increasing eigenvalue) */
1739: /* */
1740: /*******************************************************************/
1741:
1742: GEN
1743: eigen(GEN x, long prec)
1744: {
1745: GEN y,z,rr,p,ssesp,r1,r2,r3;
1746: long i,k,l,ly,av,tetpil,nbrac,ex, n = lg(x);
1747:
1748: if (typ(x)!=t_MAT) err(typeer,"eigen");
1749: if (n != 1 && n != lg(x[1])) err(mattype1,"eigen");
1750: if (n<=2) return gcopy(x);
1751:
1752: av=avma; ex = 16 - bit_accuracy(prec);
1753: y=cgetg(n,t_MAT); z=dummycopy(x);
1754: p=caradj(x,0,NULL); rr=roots(p,prec); nbrac=lg(rr)-1;
1755: for (i=1; i<=nbrac; i++)
1756: {
1757: GEN p1 = (GEN)rr[i];
1758: if (!signe(p1[2])) rr[i]=p1[1];
1759: }
1760: ly=1; k=1; r2=(GEN)rr[1];
1761: for(;;)
1762: {
1763: r3 = ground(r2); if (gexpo(gsub(r2,r3)) < ex) r2 = r3;
1764: for (i=1; i<n; i++)
1765: coeff(z,i,i) = lsub(gcoeff(x,i,i),r2);
1766: ssesp=ker0(z,prec); l=lg(ssesp);
1767: if (l == 1)
1768: err(talker, "precision too low in eigen");
1769: for (i=1; i<l; ) y[ly++]=ssesp[i++]; /* done with this eigenspace */
1770:
1771: r1=r2; /* try to find a different eigenvalue */
1772: do
1773: {
1774: if (k==nbrac)
1775: {
1776: tetpil=avma; setlg(y,ly); /* x may not be diagonalizable */
1777: return gerepile(av,tetpil,gcopy(y));
1778: }
1779: k++; r2=(GEN)rr[k];
1780: }
1781: while (gexpo(gsub(r1,r2)) < ex);
1782: }
1783: }
1784:
1785: /*******************************************************************/
1786: /* */
1787: /* DETERMINANT */
1788: /* */
1789: /*******************************************************************/
1790:
1791: GEN
1792: det0(GEN a,long flag)
1793: {
1794: switch(flag)
1795: {
1796: case 0: return det(a);
1797: case 1: return det2(a);
1798: default: err(flagerr,"matdet");
1799: }
1800: return NULL; /* not reached */
1801: }
1802:
1803: /* Exact types: choose the first non-zero pivot. Otherwise: maximal pivot */
1804: static GEN
1805: det_simple_gauss(GEN a, long inexact)
1806: {
1807: long i,j,k,av,av1,s, nbco = lg(a)-1;
1808: GEN x,p;
1809:
1810: av=avma; s=1; x=gun; a=dummycopy(a);
1811: for (i=1; i<nbco; i++)
1812: {
1813: p=gcoeff(a,i,i); k=i;
1814: if (inexact)
1815: {
1816: long e, ex = gexpo(p);
1817: GEN p1;
1818:
1819: for (j=i+1; j<=nbco; j++)
1820: {
1821: e = gexpo(gcoeff(a,i,j));
1822: if (e > ex) { ex=e; k=j; }
1823: }
1824: p1 = gcoeff(a,i,k);
1825: if (gcmp0(p1)) return gerepileupto(av, gcopy(p1));
1826: }
1827: else if (gcmp0(p))
1828: {
1829: do k++; while(k<=nbco && gcmp0(gcoeff(a,i,k)));
1830: if (k>nbco) return gerepileupto(av, gcopy(p));
1831: }
1832: if (k != i)
1833: {
1834: swap(a[i],a[k]); s = -s;
1835: p = gcoeff(a,i,i);
1836: }
1837:
1838: x = gmul(x,p);
1839: for (k=i+1; k<=nbco; k++)
1840: {
1841: GEN m = gcoeff(a,i,k);
1842: if (!gcmp0(m))
1843: {
1844: m = gneg_i(gdiv(m,p));
1845: for (j=i+1; j<=nbco; j++)
1846: coeff(a,j,k) = ladd(gcoeff(a,j,k), gmul(m,gcoeff(a,j,i)));
1847: }
1848: }
1849: }
1850: if (s<0) x = gneg_i(x);
1851: av1=avma; return gerepile(av,av1,gmul(x,gcoeff(a,nbco,nbco)));
1852: }
1853:
1854: /* a has integer entries, N = P^n */
1855: GEN
1856: det_mod_P_n(GEN a, GEN N, GEN P)
1857: {
1858: long va,i,j,k,s, av = avma, nbco = lg(a)-1;
1859: GEN x,p;
1860:
1861: s=1; va=0; x=gun; a=dummycopy(a);
1862: for (i=1; i<nbco; i++)
1863: {
1864: long fl = 0;
1865: for(;;)
1866: {
1867: for (k=i; k<=nbco; k++)
1868: {
1869: p=gcoeff(a,i,k);
1870: if (signe(p))
1871: {
1872: fl = 1;
1873: if (resii(p,P) != gzero) break;
1874: }
1875: }
1876: if (k <= nbco) break;
1877: va++; N = divii(N, P);
1878: if (!fl || is_pm1(N)) { avma=av; return gzero; }
1879: for (k=i; k<=nbco; k++) coeff(a,i,k) = ldivii(gcoeff(a,i,k), P);
1880: }
1881: if (k != i) { swap(a[i],a[k]); s = -s; }
1882:
1883: x = gmul(x,p); p = mpinvmod(p,N);
1884: for (k=i+1; k<=nbco; k++)
1885: {
1886: GEN m = resii(gcoeff(a,i,k), N);
1887: coeff(a,i,k) = zero;
1888: if (signe(m))
1889: {
1890: m = negi(resii(mulii(m,p), N));
1891: for (j=i+1; j<=nbco; j++)
1892: coeff(a,j,k) = lresii(addii(gcoeff(a,j,k),
1893: mulii(m,gcoeff(a,j,i))), N);
1894: }
1895: }
1896: }
1897: if (s<0) x = negi(x);
1898: x = resii(mulii(x,gcoeff(a,nbco,nbco)), N);
1899: return gerepileuptoint(av, mulii(x, gpowgs(P,va)));
1900: }
1901:
1902: GEN
1903: det2(GEN a)
1904: {
1905: long nbco = lg(a)-1;
1906: if (typ(a)!=t_MAT) err(mattype1,"det2");
1907: if (!nbco) return gun;
1908: if (nbco != lg(a[1])-1) err(mattype1,"det2");
1909: return det_simple_gauss(a,use_maximal_pivot(a));
1910: }
1911:
1912: /* determinant in a ring A: all computations are done within A
1913: * (Gauss-Bareiss algorithm)
1914: */
1915: GEN
1916: det(GEN a)
1917: {
1918: long nbco = lg(a)-1,i,j,k,av,s;
1919: GEN p,pprec;
1920:
1921: if (typ(a)!=t_MAT) err(mattype1,"det");
1922: if (!nbco) return gun;
1923: if (nbco != lg(a[1])-1) err(mattype1,"det");
1924: if (use_maximal_pivot(a)) return det_simple_gauss(a,1);
1925:
1926: av=avma; a=dummycopy(a); s=1;
1927: if (DEBUGLEVEL > 7) timer2();
1928: for (pprec=gun,i=1; i<nbco; i++,pprec=p)
1929: {
1930: GEN *ci, *ck, m, p1;
1931: int diveuc = (gcmp1(pprec)==0);
1932:
1933: p = gcoeff(a,i,i);
1934: if (gcmp0(p))
1935: {
1936: k=i+1; while (k<=nbco && gcmp0(gcoeff(a,i,k))) k++;
1937: if (k>nbco) return gerepileupto(av, gcopy(p));
1938: swap(a[k], a[i]); s = -s;
1939: p=gcoeff(a,i,i);
1940: }
1941: ci = (GEN*)a[i];
1942: for (k=i+1; k<=nbco; k++)
1943: {
1944: ck = (GEN*)a[k]; m = (GEN)ck[i];
1945: if (gcmp0(m))
1946: {
1947: if (gcmp1(p))
1948: {
1949: if (!gcmp1(pprec))
1950: a[k] = (long)gdivexact((GEN)a[k], pprec);
1951: }
1952: else
1953: for (j=i+1; j<=nbco; j++)
1954: {
1955: p1 = gmul(p,ck[j]);
1956: if (diveuc) p1 = gdivexact(p1,pprec);
1957: ck[j] = p1;
1958: }
1959: }
1960: else
1961: {
1962: m = gneg_i(m);
1963: for (j=i+1; j<=nbco; j++)
1964: {
1965: p1 = gadd(gmul(p,ck[j]), gmul(m,ci[j]));
1966: if (diveuc) p1 = gdivexact(p1,pprec);
1967: ck[j] = p1;
1968: }
1969: }
1970: }
1971: if (DEBUGLEVEL > 7) msgtimer("det, col %ld / %ld",i,nbco-1);
1972: }
1973: p = gcoeff(a,nbco,nbco);
1974: if (s < 0) p = gneg(p); else if (nbco==1) p = gcopy(p);
1975: return gerepileupto(av, p);
1976: }
1977:
1978: /*******************************************************************/
1979: /* */
1980: /* SPECIAL HNF (FOR INTERNAL USE !!!) */
1981: /* */
1982: /*******************************************************************/
1983: GEN lincomb_integral(GEN u, GEN v, GEN X, GEN Y);
1984: GEN vconcat(GEN Q1, GEN Q2);
1985:
1986: static int
1987: count(long **mat, long row, long len, long *firstnonzero)
1988: {
1989: int j, n=0;
1990:
1991: for (j=1; j<=len; j++)
1992: {
1993: long p = mat[j][row];
1994: if (p)
1995: {
1996: if (labs(p)!=1) return -1;
1997: n++; *firstnonzero=j;
1998: }
1999: }
2000: return n;
2001: }
2002:
2003: static int
2004: count2(long **mat, long row, long len)
2005: {
2006: int j;
2007: for (j=len; j; j--)
2008: if (labs(mat[j][row]) == 1) return j;
2009: return 0;
2010: }
2011:
2012: static GEN
2013: hnffinal(GEN matgen,GEN perm,GEN* ptdep,GEN* ptB,GEN* ptC)
2014: {
2015: GEN p1,p2,U,H,Hnew,Bnew,Cnew,diagH1;
2016: GEN B = *ptB, C = *ptC, dep = *ptdep, depnew;
2017: long av,i,j,k,s,i1,j1,lim,zc;
2018: long co = lg(C);
2019: long col = lg(matgen)-1;
2020: long lnz, nlze, lig;
2021:
2022: if (col == 0) return matgen;
2023: lnz = lg(matgen[1])-1; /* was called lnz-1 - nr in hnfspec */
2024: nlze = lg(dep[1])-1;
2025: lig = nlze + lnz;
2026: if (DEBUGLEVEL>5)
2027: {
2028: fprintferr("Entering hnffinal:\n");
2029: if (DEBUGLEVEL>6)
2030: {
2031: if (nlze) fprintferr("dep = %Z\n",dep);
2032: fprintferr("mit = %Z\n",matgen);
2033: fprintferr("B = %Z\n",B);
2034: }
2035: }
2036: /* [LLLKERIM]
2037: u1u2=lllkerim(matgen); u1=(GEN)u1u2[1]; u2=(GEN)u1u2[2];
2038: if (DEBUGLEVEL>6) fprintferr("lllkerim done\n");
2039: if (lg(u2)<=lnz)
2040: err(talker,"matrix not of maximal rank in hermite spec");
2041: p1=gmul(matgen,u2);
2042: detmat=absi(det(p1));
2043: if (DEBUGLEVEL>6) fprintferr("det done\n");
2044: H=hnfmod(p1,detmat);
2045: if (DEBUGLEVEL>6) fprintferr("hnfmod done\n");
2046: p2=gmul(u1,lllint(u1));
2047: if (DEBUGLEVEL>6) fprintferr("lllint done\n");
2048: p3=gmul(u2,gauss(p1,H));
2049: if (DEBUGLEVEL>6) fprintferr("gauss done\n");
2050: U=cgetg(col+1,t_MAT);
2051: for (j=1; j<lg(p2); j++) U[j]=p2[j];
2052: for (j=lg(p2); j<=col; j++) U[j]=p3[j+1-lg(p2)]; */
2053:
2054: /* [HNFHAVAS]
2055:
2056: p2=hnfhavas(matgen); p1=(GEN)p2[1]; U=(GEN)p2[2]; p5=(GEN)p2[3];
2057: if (DEBUGLEVEL>6) fprintferr("hnfhavas done\n");
2058: for (i=1; i < lg(p1) && gcmp0(p1[i]); i++);
2059: i1=i-1;
2060: u1=cgetg(i,t_MAT); for (j=1; j<i; j++) u1[j]=U[j];
2061: H=cgetg(j1=lg(p1)-i1,t_MAT); for (j=1; j<j1; j++) H[j]=p1[i1+j];
2062: p2=cgetg(lg(p5),t_VEC);
2063: for (i=1; i<lg(p5); i++) p2[i]=lstoi(perm[nlze+itos(p5[i])]);
2064: for (i=1; i<lg(p5); i++) perm[nlze+i]=itos(p2[i]);
2065: p2=u1;
2066: p1=cgetg(j1,t_MAT); for (j=1; j<j1; j++) p1[j]=U[i1+j];
2067: Bnew=cgetg(co-col,t_MAT);
2068: for (j=1; j<co-col; j++)
2069: {
2070: p3=cgetg(lig+1,t_COL); Bnew[j]=(long)p3;
2071: for (i=1; i<=nlze; i++) p3[i]=coeff(B,i,j);
2072: for (; i<=lig; i++) p3[i]=coeff(B,nlze+itos(p5[i-nlze]),j);
2073: }
2074: B=Bnew; */
2075:
2076: /* [HNFBATUT] */
2077: p1 = hnfall(matgen);
2078: H = (GEN)p1[1]; /* lnz x lnz */
2079: U = (GEN)p1[2]; /* col x col */
2080: /* Only keep the part above the H (above the 0s is 0 since the dep rows
2081: * are dependant from the ones in matgen) */
2082: zc = col - lnz; /* # of 0 columns, correspond to units */
2083: if (nlze) { dep = gmul(dep,U); dep += zc; }
2084:
2085: diagH1 = new_chunk(lnz+1); /* diagH1[i] = 0 iff H[i,i] != 1 (set later) */
2086:
2087: av = avma; lim = stack_lim(av,1);
2088: Cnew = cgetg(co,t_MAT);
2089: setlg(C, col+1);
2090: p1 = gmul(C,U); setlg(C, co);
2091: for (j=1; j<=col; j++) Cnew[j] = p1[j];
2092: for ( ; j<co ; j++) Cnew[j] = C[j];
2093: if (DEBUGLEVEL>5) fprintferr(" hnfall done\n");
2094:
2095: /* Clean up B using new H */
2096: for (s=0,i=lnz; i; i--)
2097: {
2098: GEN h = gcoeff(H,i,i);
2099: if ( (diagH1[i] = gcmp1(h)) ) { h = NULL; s++; }
2100: for (j=col+1; j<co; j++)
2101: {
2102: GEN z = (GEN)B[j-col];
2103: p1 = (GEN)z[i+nlze]; if (h) p1 = gdivent(p1,h);
2104: for (k=1; k<=nlze; k++)
2105: z[k] = lsubii((GEN)z[k], mulii(p1, gcoeff(dep,k,i)));
2106: for ( ; k<=lig; k++)
2107: z[k] = lsubii((GEN)z[k], mulii(p1, gcoeff(H,k-nlze,i)));
2108: Cnew[j] = lsub((GEN)Cnew[j], gmul(p1, (GEN)Cnew[i+zc]));
2109: }
2110: if (low_stack(lim, stack_lim(av,1)))
2111: {
2112: GEN *gptr[2]; gptr[0]=&Cnew; gptr[1]=&B;
2113: if(DEBUGMEM>1) err(warnmem,"hnffinal, i = %ld",i);
2114: gerepilemany(av,gptr,2);
2115: }
2116: }
2117: p1 = cgetg(lnz+1,t_VEC); p2 = perm + nlze;
2118: for (i1=0, j1=lnz-s, i=1; i<=lnz; i++) /* push the 1 rows down */
2119: if (diagH1[i])
2120: p1[++j1] = p2[i];
2121: else
2122: p2[++i1] = p2[i];
2123: for (i=i1+1; i<=lnz; i++) p2[i] = p1[i];
2124: if (DEBUGLEVEL>5) fprintferr(" first pass in hnffinal done\n");
2125:
2126: /* s = # extra redundant generators taken from H
2127: * zc col-s co zc = col lnz
2128: * [ 0 |dep | ] i = lnze + lnz - s = lig - s
2129: * nlze [--------| B' ]
2130: * [ 0 | H' | ] H' = H minus the s rows with a 1 on diagonal
2131: * i [--------|-----] lig-s (= "1-rows")
2132: * [ 0 | Id ]
2133: * [ | ] li */
2134: lig -= s; col -= s; lnz -= s;
2135: Hnew = cgetg(lnz+1,t_MAT);
2136: if (nlze) depnew = cgetg(lnz+1,t_MAT);
2137: Bnew = cgetg(co-col,t_MAT);
2138: C = dummycopy(Cnew);
2139: for (j=1,i1=j1=0; j<=lnz+s; j++)
2140: {
2141: GEN z = (GEN)H[j];
2142: if (diagH1[j])
2143: { /* hit exactly s times */
2144: i1++; p1 = cgetg(lig+1,t_COL); Bnew[i1] = (long)p1;
2145: C[i1+col] = Cnew[j+zc];
2146: for (i=1; i<=nlze; i++) p1[i] = coeff(dep,i,j);
2147: p1 += nlze;
2148: }
2149: else
2150: {
2151: j1++; p1 = cgetg(lnz+1,t_COL); Hnew[j1] = (long)p1;
2152: C[j1+zc] = Cnew[j+zc];
2153: if (nlze) depnew[j1] = dep[j];
2154: }
2155: for (i=k=1; k<=lnz; i++)
2156: if (!diagH1[i]) p1[k++] = z[i];
2157: }
2158: for (j=s+1; j<co-col; j++)
2159: {
2160: GEN z = (GEN)B[j-s];
2161: p1 = cgetg(lig+1,t_COL); Bnew[j] = (long)p1;
2162: for (i=1; i<=nlze; i++) p1[i] = z[i];
2163: z += nlze; p1 += nlze;
2164: for (i=k=1; k<=lnz; i++)
2165: if (!diagH1[i]) p1[k++] = z[i];
2166: }
2167: if (DEBUGLEVEL>5)
2168: {
2169: fprintferr("Leaving hnffinal\n");
2170: if (DEBUGLEVEL>6)
2171: {
2172: if (nlze) fprintferr("dep = %Z\n",depnew);
2173: fprintferr("mit = %Z\n",Hnew); outerr(Hnew);
2174: fprintferr("B = %Z\n",Bnew);
2175: fprintferr("C = %Z\n",C);
2176: }
2177: }
2178: if (nlze) *ptdep = depnew;
2179: *ptC = C;
2180: *ptB = Bnew; return Hnew;
2181: }
2182:
2183: /* for debugging */
2184: static void
2185: p_mat(long **mat, long *perm, long k0)
2186: {
2187: long av=avma, i,j;
2188: GEN p1, matj, matgen;
2189: long co = lg(mat);
2190: long li = lg(perm);
2191:
2192: fprintferr("Permutation: %Z\n",perm);
2193: matgen = cgetg(co,t_MAT);
2194: for (j=1; j<co; j++)
2195: {
2196: p1 = cgetg(li-k0,t_COL); matgen[j]=(long)p1;
2197: p1 -= k0; matj = mat[j];
2198: for (i=k0+1; i<li; i++) p1[i] = lstoi(matj[perm[i]]);
2199: }
2200: if (DEBUGLEVEL > 6) fprintferr("matgen = %Z\n",matgen);
2201: avma=av;
2202: }
2203:
2204: #define gswap(x,y) { long *_t=x; x=y; y=_t; }
2205:
2206: /* HNF reduce a relation matrix (column operations + row permutation)
2207: ** Input:
2208: ** mat = (li-1) x (co-1) matrix of long
2209: ** C = r x (co-1) matrix of GEN
2210: ** perm= permutation vector (length li-1), indexing the rows of mat: easier
2211: ** to maintain perm than to copy rows. For columns we can do it directly
2212: ** using e.g. swap(mat[i], mat[j])
2213: ** k0 = integer. The k0 first lines of mat are dense, the others are sparse.
2214: ** Output: cf ASCII art in the function body
2215: **
2216: ** row permutations applied to perm
2217: ** column operations applied to C.
2218: **/
2219: GEN
2220: hnfspec(long** mat, GEN perm, GEN* ptdep, GEN* ptB, GEN* ptC, long k0)
2221: {
2222: long av=avma,av2,*p,i,j,k,lk0,col,lig,*matj;
2223: long n,s,t,lim,nlze,lnz,nr;
2224: GEN p1,p2,matb,matbnew,vmax,matt,T,extramat;
2225: GEN B,H,dep,permpro;
2226: GEN *gptr[4];
2227: long co = lg(mat);
2228: long li = lg(perm); /* = lg(mat[1]) */
2229: int updateT = 1;
2230:
2231: if (DEBUGLEVEL>5)
2232: {
2233: fprintferr("Entering hnfspec\n");
2234: p_mat(mat,perm,0);
2235: }
2236: matt = cgetg(co,t_MAT); /* dense part of mat (top) */
2237: for (j=1; j<co; j++)
2238: {
2239: p1=cgetg(k0+1,t_COL); matt[j]=(long)p1; matj = mat[j];
2240: for (i=1; i<=k0; i++) p1[i] = lstoi(matj[perm[i]]);
2241: }
2242: vmax = cgetg(co,t_VECSMALL);
2243: av2 = avma; lim = stack_lim(av2,1);
2244:
2245: i=lig=li-1; col=co-1; lk0=k0;
2246: if (k0 || (lg(*ptC) > 1 && lg((*ptC)[1]) > 1)) T = idmat(col);
2247: else
2248: { /* dummy ! */
2249: GEN z = cgetg(1,t_COL);
2250: T = cgetg(co, t_MAT); updateT = 0;
2251: for (j=1; j<co; j++) T[j] = (long)z;
2252: }
2253: /* Look for lines with a single non0 entry, equal to ±1 */
2254: while (i > lk0)
2255: switch( count(mat,perm[i],col,&n) )
2256: {
2257: case 0: /* move zero lines between k0+1 and lk0 */
2258: lk0++; swap(perm[i], perm[lk0]);
2259: i=lig; continue;
2260:
2261: case 1: /* move trivial generator between lig+1 and li */
2262: swap(perm[i], perm[lig]);
2263: swap(T[n], T[col]);
2264: gswap(mat[n], mat[col]); p = mat[col];
2265: if (p[perm[lig]] < 0) /* = -1 */
2266: { /* convert relation -g = 0 to g = 0 */
2267: for (i=lk0+1; i<lig; i++) p[perm[i]] = -p[perm[i]];
2268: if (updateT)
2269: {
2270: p1 = (GEN)T[col];
2271: for (i=1; ; i++)
2272: if (signe((GEN)p1[i])) { p1[i] = lnegi((GEN)p1[i]); break; }
2273: }
2274: }
2275: lig--; col--; i=lig; continue;
2276:
2277: default: i--;
2278: }
2279: if (DEBUGLEVEL>5)
2280: {
2281: fprintferr(" after phase1:\n");
2282: p_mat(mat,perm,0);
2283: }
2284:
2285: #define absmax(s,z) {long _z = labs(z); if (_z > s) s = _z;}
2286:
2287: #if 0 /* TODO: check, and put back in */
2288: /* Get rid of all lines containing only 0 and ± 1, keeping track of column
2289: * operations in T. Leave the rows 1..lk0 alone [up to k0, coeff
2290: * explosion, between k0+1 and lk0, row is 0]
2291: */
2292: s = 0;
2293: while (lig > lk0 && s < (HIGHBIT>>1))
2294: {
2295: for (i=lig; i>lk0; i--)
2296: if (count(mat,perm[i],col,&n) >= 0) break;
2297: if (i == lk0) break;
2298:
2299: /* only 0, ±1 entries, at least 2 of them non-zero */
2300: swap(perm[i], perm[lig]);
2301: swap(T[n], T[col]); p1 = (GEN)T[col];
2302: gswap(mat[n], mat[col]); p = mat[col];
2303: if (p[perm[lig]] < 0)
2304: {
2305: for (i=lk0+1; i<=lig; i++) p[perm[i]] = -p[perm[i]];
2306: T[col] = lneg(p1);
2307: }
2308: for (j=1; j<n; j++)
2309: {
2310: matj = mat[j];
2311: if (! (t = matj[perm[lig]]) ) continue;
2312: if (t == 1)
2313: { /* t = 1 */
2314: for (i=lk0+1; i<=lig; i++)
2315: absmax(s, matj[perm[i]] -= p[perm[i]]);
2316: T[j] = lsub((GEN)T[j], p1);
2317: }
2318: else
2319: { /* t = -1 */
2320: for (i=lk0+1; i<=lig; i++)
2321: absmax(s, matj[perm[i]] += p[perm[i]]);
2322: T[j] = ladd((GEN)T[j], p1);
2323: }
2324: }
2325: lig--; col--;
2326: if (low_stack(lim, stack_lim(av2,1)))
2327: {
2328: if(DEBUGMEM>1) err(warnmem,"hnfspec[1]");
2329: T = gerepileupto(av2, gcopy(T));
2330: }
2331: }
2332: #endif
2333: /* As above with lines containing a ±1 (no other assumption).
2334: * Stop when single precision becomes dangerous */
2335: for (j=1; j<=col; j++)
2336: {
2337: matj = mat[j];
2338: for (s=0, i=lk0+1; i<=lig; i++) absmax(s, matj[i]);
2339: vmax[j] = s;
2340: }
2341: while (lig > lk0)
2342: {
2343: for (i=lig; i>lk0; i--)
2344: if ( (n = count2(mat,perm[i],col)) ) break;
2345: if (i == lk0) break;
2346:
2347: swap(perm[i], perm[lig]);
2348: swap(vmax[n], vmax[col]);
2349: gswap(mat[n], mat[col]); p = mat[col];
2350: swap(T[n], T[col]); p1 = (GEN)T[col];
2351: if (p[perm[lig]] < 0)
2352: {
2353: for (i=lk0+1; i<=lig; i++) p[perm[i]] = -p[perm[i]];
2354: p1 = gneg(p1); T[col] = (long)p1;
2355: }
2356: for (j=1; j<col; j++)
2357: {
2358: matj = mat[j];
2359: if (! (t = matj[perm[lig]]) ) continue;
2360: if (vmax[col] && labs(t) >= (HIGHBIT-vmax[j]) / vmax[col]) goto END2;
2361:
2362: for (s=0, i=lk0+1; i<=lig; i++)
2363: absmax(s, matj[perm[i]] -= t*p[perm[i]]);
2364: vmax[j] = s;
2365: T[j] = (long)lincomb_integral(gun,stoi(-t), (GEN)T[j],p1);
2366: }
2367: lig--; col--;
2368: if (low_stack(lim, stack_lim(av2,1)))
2369: {
2370: if(DEBUGMEM>1) err(warnmem,"hnfspec[2]");
2371: T = gerepileupto(av2,gcopy(T));
2372: }
2373: }
2374:
2375: END2: /* clean up mat: remove everything to the right of the 1s on diagonal */
2376: /* go multiprecision first */
2377: matb = cgetg(co,t_MAT); /* bottom part (complement of matt) */
2378: for (j=1; j<co; j++)
2379: {
2380: p1 = cgetg(li-k0,t_COL); matb[j] = (long)p1;
2381: p1 -= k0; matj = mat[j];
2382: for (i=k0+1; i<li; i++) p1[i] = lstoi(matj[perm[i]]);
2383: }
2384: if (DEBUGLEVEL>5)
2385: {
2386: fprintferr(" after phase2:\n");
2387: p_mat(mat,perm,k0);
2388: }
2389: for (i=li-2; i>lig; i--)
2390: {
2391: long i1, i0 = i - k0, k = i + co-li;
2392: GEN Bk = (GEN)matb[k];
2393: GEN Tk = (GEN)T[k];
2394: for (j=k+1; j<co; j++)
2395: {
2396: p1=(GEN)matb[j]; p2=(GEN)p1[i0];
2397: if (! (s=signe(p2)) ) continue;
2398:
2399: p1[i0] = zero;
2400: if (is_pm1(p2))
2401: {
2402: if (s > 0)
2403: { /* p2 = 1 */
2404: for (i1=1; i1<i0; i1++)
2405: p1[i1] = lsubii((GEN)p1[i1], (GEN)Bk[i1]);
2406: T[j] = lsub((GEN)T[j], Tk);
2407: }
2408: else
2409: { /* p2 = -1 */
2410: for (i1=1; i1<i0; i1++)
2411: p1[i1] = laddii((GEN)p1[i1], (GEN)Bk[i1]);
2412: T[j] = ladd((GEN)T[j], Tk);
2413: }
2414: }
2415: else
2416: {
2417: for (i1=1; i1<i0; i1++)
2418: p1[i1] = lsubii((GEN)p1[i1], mulii(p2,(GEN) Bk[i1]));
2419: T[j] = (long)lincomb_integral(gun,negi(p2), (GEN)T[j],Tk);
2420: }
2421: }
2422: if (low_stack(lim, stack_lim(av2,1)))
2423: {
2424: if(DEBUGMEM>1) err(warnmem,"hnfspec[3], i = %ld", i);
2425: for (j=1; j<co; j++) setlg(matb[j], i0+1); /* bottom can be forgotten */
2426: gptr[0]=&T; gptr[1]=&matb; gerepilemany(av2,gptr,2);
2427: }
2428: }
2429: gptr[0]=&T; gptr[1]=&matb; gerepilemany(av2,gptr,2);
2430: if (DEBUGLEVEL>5)
2431: {
2432: fprintferr(" matb cleaned up (using Id block)\n");
2433: if (DEBUGLEVEL>6) outerr(matb);
2434: }
2435:
2436: nlze = lk0 - k0; /* # of 0 rows */
2437: lnz = lig-nlze+1; /* 1 + # of non-0 rows (!= 0...0 1 0 ... 0) */
2438: if (updateT) matt = gmul(matt,T); /* update top rows */
2439: extramat = cgetg(col+1,t_MAT); /* = new C minus the 0 rows */
2440: for (j=1; j<=col; j++)
2441: {
2442: GEN z = (GEN)matt[j];
2443: GEN t = ((GEN)matb[j]) + nlze - k0;
2444: p2=cgetg(lnz,t_COL); extramat[j]=(long)p2;
2445: for (i=1; i<=k0; i++) p2[i] = z[i]; /* top k0 rows */
2446: for ( ; i<lnz; i++) p2[i] = t[i]; /* other non-0 rows */
2447: }
2448: permpro = imagecomplspec(extramat, &nr); /* lnz = lg(permpro) */
2449:
2450: if (nlze)
2451: { /* put the nlze 0 rows (trivial generators) at the top */
2452: p1 = new_chunk(lk0+1);
2453: for (i=1; i<=nlze; i++) p1[i] = perm[i + k0];
2454: for ( ; i<=lk0; i++) p1[i] = perm[i - nlze];
2455: for (i=1; i<=lk0; i++) perm[i] = p1[i];
2456: }
2457: /* sort other rows according to permpro (nr redundant generators first) */
2458: p1 = new_chunk(lnz); p2 = perm + nlze;
2459: for (i=1; i<lnz; i++) p1[i] = p2[permpro[i]];
2460: for (i=1; i<lnz; i++) p2[i] = p1[i];
2461: /* perm indexes the rows of mat
2462: * |_0__|__redund__|__dense__|__too big__|_____done______|
2463: * 0 nlze lig li
2464: * \___nr___/ \___k0__/
2465: * \____________lnz ______________/
2466: *
2467: * col co
2468: * [dep | ]
2469: * i0 [--------| B ] (i0 = nlze + nr)
2470: * [matbnew | ] matbnew has maximal rank = lnz-1 - nr
2471: * mat = [--------|-----] lig
2472: * [ 0 | Id ]
2473: * [ | ] li */
2474:
2475: matbnew = cgetg(col+1,t_MAT); /* dense+toobig, maximal rank. For hnffinal */
2476: dep = cgetg(col+1,t_MAT); /* rows dependant from the ones in matbnew */
2477: for (j=1; j<=col; j++)
2478: {
2479: GEN z = (GEN)extramat[j];
2480: p1 = cgetg(nlze+nr+1,t_COL); dep[j]=(long)p1;
2481: p2 = cgetg(lnz-nr,t_COL); matbnew[j]=(long)p2;
2482: for (i=1; i<=nlze; i++) p1[i]=zero;
2483: p1 += nlze; for (i=1; i<=nr; i++) p1[i] = z[permpro[i]];
2484: p2 -= nr; for ( ; i<lnz; i++) p2[i] = z[permpro[i]];
2485: }
2486:
2487: /* redundant generators in terms of the genuine generators
2488: * (x_i) = - (g_i) B */
2489: B = cgetg(co-col,t_MAT);
2490: for (j=col+1; j<co; j++)
2491: {
2492: GEN y = (GEN)matt[j];
2493: GEN z = (GEN)matb[j];
2494: p1=cgetg(lig+1,t_COL); B[j-col]=(long)p1;
2495: for (i=1; i<=nlze; i++) p1[i] = z[i];
2496: p1 += nlze; z += nlze-k0;
2497: for (k=1; k<lnz; k++)
2498: {
2499: i = permpro[k];
2500: p1[k] = (i <= k0)? y[i]: z[i];
2501: }
2502: }
2503: if (updateT) *ptC = gmul(*ptC,T);
2504: *ptdep = dep;
2505: *ptB = B;
2506: H = hnffinal(matbnew,perm,ptdep,ptB,ptC);
2507: gptr[0]=ptC;
2508: gptr[1]=ptdep;
2509: gptr[2]=ptB;
2510: gptr[3]=&H; gerepilemany(av,gptr,4);
2511: if (DEBUGLEVEL)
2512: msgtimer("hnfspec [%ld x %ld] --> [%ld x %ld]",li-1,co-1, lig-1,col-1);
2513: return H;
2514: }
2515:
2516: /* HNF reduce x, apply same transforms to C */
2517: GEN
2518: mathnfspec(GEN x, GEN *ptperm, GEN *ptdep, GEN *ptB, GEN *ptC)
2519: {
2520: long i,j,ly,lx = lg(x);
2521: GEN p1,p2,z,perm;
2522: if (lx == 1) return gcopy(x);
2523: ly = lg(x[1]);
2524: z = cgetg(lx,t_MAT);
2525: perm = cgetg(ly,t_VECSMALL); *ptperm = perm;
2526: for (i=1; i<ly; i++) perm[i] = i;
2527: for (i=1; i<lx; i++)
2528: {
2529: p1 = cgetg(ly,t_COL); z[i] = (long)p1;
2530: p2 = (GEN)x[i];
2531: for (j=1; j<ly; j++) p1[j] = itos((GEN)p2[j]);
2532: }
2533: /* [ dep | ]
2534: * [-----| B ]
2535: * [ H | ]
2536: * [-----|-----]
2537: * [ 0 | Id ] */
2538: return hnfspec((long**)z,perm, ptdep, ptB, ptC, 0);
2539: }
2540:
2541: /* add new relations to a matrix treated by hnfspec (extramat / extraC) */
2542: GEN
2543: hnfadd(GEN H, GEN perm, GEN* ptdep, GEN* ptB, GEN* ptC, /* cf hnfspec */
2544: GEN extramat,GEN extraC)
2545: {
2546: GEN p1,p2,p3,matb,extratop,Cnew,permpro;
2547: GEN B=*ptB, C=*ptC, dep=*ptdep, *gptr[4];
2548: long av = avma, i,j,lextra,colnew;
2549: long li = lg(perm);
2550: long co = lg(C);
2551: long lB = lg(B);
2552: long lig = li - lB;
2553: long col = co - lB;
2554: long lH = lg(H)-1;
2555: long nlze = lH? lg(dep[1])-1: lg(B[1])-1;
2556:
2557: if (DEBUGLEVEL>5)
2558: {
2559: fprintferr("Entering hnfadd:\n");
2560: if (DEBUGLEVEL>6) fprintferr("extramat = %Z\n",extramat);
2561: }
2562: /* col co
2563: * [ 0 |dep | ]
2564: * nlze [--------| B ]
2565: * [ 0 | H | ]
2566: * [--------|-----] lig
2567: * [ 0 | Id ]
2568: * [ | ] li */
2569: lextra = lg(extramat)-1;
2570: extratop = cgetg(lextra+1,t_MAT); /* [1..lig] part (top) */
2571: p2 = cgetg(lextra+1,t_MAT); /* bottom */
2572: for (j=1; j<=lextra; j++)
2573: {
2574: GEN z = (GEN)extramat[j];
2575: p1=cgetg(lig+1,t_COL); extratop[j] = (long)p1;
2576: for (i=1; i<=lig; i++) p1[i] = z[i];
2577: p1=cgetg(lB,t_COL); p2[j] = (long)p1;
2578: p1 -= lig;
2579: for ( ; i<li; i++) p1[i] = z[i];
2580: }
2581: if (li-1 != lig)
2582: { /* zero out bottom part, using the Id block */
2583: GEN A = cgetg(lB,t_MAT);
2584: for (j=1; j<lB; j++) A[j] = C[j+col];
2585: extraC = gsub(extraC, gmul(A,p2));
2586: extratop = gsub(extratop,gmul(B,p2));
2587: }
2588:
2589: colnew = lH + lextra;
2590: extramat = cgetg(colnew+1,t_MAT);
2591: Cnew = cgetg(lB+colnew,t_MAT);
2592: for (j=1; j<=lextra; j++)
2593: {
2594: extramat[j] = extratop[j];
2595: Cnew[j] = extraC[j];
2596: }
2597: for ( ; j<=colnew; j++)
2598: {
2599: p1 = cgetg(lig+1,t_COL); extramat[j] = (long)p1;
2600: p2 = (GEN)dep[j-lextra]; for (i=1; i<=nlze; i++) p1[i] = p2[i];
2601: p2 = (GEN) H[j-lextra]; for ( ; i<=lig ; i++) p1[i] = p2[i-nlze];
2602: }
2603: for (j=lextra+1; j<lB+colnew; j++)
2604: Cnew[j] = C[j-lextra+col-lH];
2605: if (DEBUGLEVEL>5)
2606: {
2607: fprintferr(" 1st phase done\n");
2608: if (DEBUGLEVEL>6) fprintferr("extramat = %Z\n",extramat);
2609: }
2610: permpro = imagecomplspec(extramat, &nlze);
2611: p1 = new_chunk(lig+1);
2612: for (i=1; i<=lig; i++) p1[i] = perm[permpro[i]];
2613: for (i=1; i<=lig; i++) perm[i] = p1[i];
2614:
2615: matb = cgetg(colnew+1,t_MAT);
2616: dep = cgetg(colnew+1,t_MAT);
2617: for (j=1; j<=colnew; j++)
2618: {
2619: GEN z = (GEN)extramat[j];
2620: p1=cgetg(nlze+1,t_COL); dep[j]=(long)p1;
2621: p2=cgetg(lig+1-nlze,t_COL); matb[j]=(long)p2;
2622: p2 -= nlze;
2623: for (i=1; i<=nlze; i++) p1[i] = z[permpro[i]];
2624: for ( ; i<= lig; i++) p2[i] = z[permpro[i]];
2625: }
2626: p3 = cgetg(lB,t_MAT);
2627: for (j=1; j<lB; j++)
2628: {
2629: p2 = (GEN)B[j];
2630: p1 = cgetg(lig+1,t_COL); p3[j] = (long)p1;
2631: for (i=1; i<=lig; i++) p1[i] = p2[permpro[i]];
2632: }
2633: B = p3;
2634: if (DEBUGLEVEL>5) fprintferr(" 2nd phase done\n");
2635: *ptdep = dep;
2636: *ptB = B;
2637: H = hnffinal(matb,perm,ptdep,ptB,&Cnew);
2638: p1 = cgetg(co+lextra,t_MAT);
2639: for (j=1; j <= col-lH; j++) p1[j] = C[j];
2640: C = Cnew - (col-lH);
2641: for ( ; j < co+lextra; j++) p1[j] = C[j];
2642:
2643: gptr[0]=ptC; *ptC=p1;
2644: gptr[1]=ptdep;
2645: gptr[2]=ptB;
2646: gptr[3]=&H; gerepilemany(av,gptr,4);
2647: if (DEBUGLEVEL)
2648: {
2649: if (DEBUGLEVEL>7)
2650: {
2651: fprintferr("mit = %Z\n",H);
2652: fprintferr("C = %Z\n",p1);
2653: }
2654: msgtimer("hnfadd (%d)",lextra);
2655: }
2656: return H;
2657: }
2658:
2659: /* return a solution of congruence system sum M_{ i,j } X_j = Y_i mod D_i
2660: * If ptu1 != NULL, put in *ptu1 a Z-basis of the homogeneous system
2661: */
2662: static GEN
2663: gaussmoduloall(GEN M, GEN D, GEN Y, GEN *ptu1)
2664: {
2665: long n,m,i,j,lM,av=avma,tetpil;
2666: GEN p1,delta,H,U,u1,u2,x;
2667:
2668: if (typ(M)!=t_MAT) err(typeer,"gaussmodulo");
2669: lM = lg(M); m = lM-1;
2670: if (!m)
2671: {
2672: if ((typ(Y)!=t_INT && lg(Y)!=1)
2673: || (typ(D)!=t_INT && lg(D)!=1)) err(consister,"gaussmodulo");
2674: return gzero;
2675: }
2676: n = lg(M[1])-1;
2677: switch(typ(D))
2678: {
2679: case t_VEC:
2680: case t_COL: delta=diagonal(D); break;
2681: case t_INT: delta=gscalmat(D,n); break;
2682: default: err(typeer,"gaussmodulo");
2683: }
2684: if (typ(Y) == t_INT)
2685: {
2686: p1 = cgetg(n+1,t_COL);
2687: for (i=1; i<=n; i++) p1[i]=(long)Y;
2688: Y = p1;
2689: }
2690: p1 = hnfall(concatsp(M,delta));
2691: H = (GEN)p1[1]; U = (GEN)p1[2];
2692: Y = gauss(H,Y);
2693: if (!gcmp1(denom(Y))) return gzero;
2694: u1 = cgetg(m+1,t_MAT);
2695: u2 = cgetg(n+1,t_MAT);
2696: for (j=1; j<=m; j++)
2697: {
2698: p1 = (GEN)U[j]; setlg(p1,lM);
2699: u1[j] = (long)p1;
2700: }
2701: U += m;
2702: for (j=1; j<=n; j++)
2703: {
2704: p1 = (GEN)U[j]; setlg(p1,lM);
2705: u2[j] = (long)p1;
2706: }
2707: x = gmul(u2,Y);
2708: tetpil=avma; x=lllreducemodmatrix(x,u1);
2709: if (!ptu1) x = gerepile(av,tetpil,x);
2710: else
2711: {
2712: GEN *gptr[2];
2713: *ptu1=gcopy(u1); gptr[0]=ptu1; gptr[1]=&x;
2714: gerepilemanysp(av,tetpil,gptr,2);
2715: }
2716: return x;
2717: }
2718:
2719: GEN
2720: matsolvemod0(GEN M, GEN D, GEN Y, long flag)
2721: {
2722: long av;
2723: GEN p1,y;
2724:
2725: if (!flag) return gaussmoduloall(M,D,Y,NULL);
2726:
2727: av=avma; y = cgetg(3,t_VEC);
2728: p1 = gaussmoduloall(M,D,Y, (GEN*)y+2);
2729: if (p1==gzero) { avma=av; return gzero; }
2730: y[1] = (long)p1; return y;
2731: }
2732:
2733: GEN
2734: gaussmodulo2(GEN M, GEN D, GEN Y)
2735: {
2736: return matsolvemod0(M,D,Y,1);
2737: }
2738:
2739: GEN
2740: gaussmodulo(GEN M, GEN D, GEN Y)
2741: {
2742: return matsolvemod0(M,D,Y,0);
2743: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>