xref: /petsc/src/mat/impls/aij/seq/aij.c (revision 3a40ed3dce77c081171d005ae1a6ff4bb9d13b6f)
1a5eb4965SSatish Balay #ifdef PETSC_RCS_HEADER
2*3a40ed3dSBarry Smith static char vcid[] = "$Id: aij.c,v 1.239 1997/10/06 16:23:56 balay Exp bsmith $";
317ab2063SBarry Smith #endif
417ab2063SBarry Smith 
5d5d45c9bSBarry Smith /*
63369ce9aSBarry Smith     Defines the basic matrix operations for the AIJ (compressed row)
7d5d45c9bSBarry Smith   matrix storage format.
8d5d45c9bSBarry Smith */
93369ce9aSBarry Smith 
103369ce9aSBarry Smith #include "pinclude/pviewer.h"
113369ce9aSBarry Smith #include "sys.h"
1270f55243SBarry Smith #include "src/mat/impls/aij/seq/aij.h"
13f5eb4b81SSatish Balay #include "src/vec/vecimpl.h"
14f5eb4b81SSatish Balay #include "src/inline/spops.h"
158d195f9aSBarry Smith #include "src/inline/dot.h"
16f5eb4b81SSatish Balay #include "src/inline/bitarray.h"
1717ab2063SBarry Smith 
18a2ce50c7SBarry Smith /*
19a2ce50c7SBarry Smith     Basic AIJ format ILU based on drop tolerance
20a2ce50c7SBarry Smith */
215615d1e5SSatish Balay #undef __FUNC__
225615d1e5SSatish Balay #define __FUNC__ "MatILUDTFactor_SeqAIJ"
23a2ce50c7SBarry Smith int MatILUDTFactor_SeqAIJ(Mat A,double dt,int maxnz,IS row,IS col,Mat *fact)
24a2ce50c7SBarry Smith {
25a2ce50c7SBarry Smith   /* Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; */
26a2ce50c7SBarry Smith   int        ierr = 1;
27a2ce50c7SBarry Smith 
28*3a40ed3dSBarry Smith   PetscFunctionBegin;
29e3372554SBarry Smith   SETERRQ(ierr,0,"Not implemented");
30a2ce50c7SBarry Smith }
31a2ce50c7SBarry Smith 
32bcd2baecSBarry Smith extern int MatToSymmetricIJ_SeqAIJ(int,int*,int*,int,int,int**,int**);
3317ab2063SBarry Smith 
345615d1e5SSatish Balay #undef __FUNC__
355615d1e5SSatish Balay #define __FUNC__ "MatGetRowIJ_SeqAIJ"
368f6be9afSLois Curfman McInnes int MatGetRowIJ_SeqAIJ(Mat A,int oshift,PetscTruth symmetric,int *m,int **ia,int **ja,
376945ee14SBarry Smith                            PetscTruth *done)
3817ab2063SBarry Smith {
39416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
406945ee14SBarry Smith   int        ierr,i,ishift;
4117ab2063SBarry Smith 
42*3a40ed3dSBarry Smith   PetscFunctionBegin;
4331625ec5SSatish Balay   *m     = A->m;
44*3a40ed3dSBarry Smith   if (!ia) PetscFunctionReturn(0);
456945ee14SBarry Smith   ishift = a->indexshift;
466945ee14SBarry Smith   if (symmetric) {
4731625ec5SSatish Balay     ierr = MatToSymmetricIJ_SeqAIJ(a->m,a->i,a->j,ishift,oshift,ia,ja); CHKERRQ(ierr);
486945ee14SBarry Smith   } else if (oshift == 0 && ishift == -1) {
4931625ec5SSatish Balay     int nz = a->i[a->m];
503b2fbd54SBarry Smith     /* malloc space and  subtract 1 from i and j indices */
5131625ec5SSatish Balay     *ia = (int *) PetscMalloc( (a->m+1)*sizeof(int) ); CHKPTRQ(*ia);
523b2fbd54SBarry Smith     *ja = (int *) PetscMalloc( (nz+1)*sizeof(int) ); CHKPTRQ(*ja);
533b2fbd54SBarry Smith     for ( i=0; i<nz; i++ ) (*ja)[i] = a->j[i] - 1;
5431625ec5SSatish Balay     for ( i=0; i<a->m+1; i++ ) (*ia)[i] = a->i[i] - 1;
556945ee14SBarry Smith   } else if (oshift == 1 && ishift == 0) {
5631625ec5SSatish Balay     int nz = a->i[a->m] + 1;
573b2fbd54SBarry Smith     /* malloc space and  add 1 to i and j indices */
5831625ec5SSatish Balay     *ia = (int *) PetscMalloc( (a->m+1)*sizeof(int) ); CHKPTRQ(*ia);
593b2fbd54SBarry Smith     *ja = (int *) PetscMalloc( (nz+1)*sizeof(int) ); CHKPTRQ(*ja);
603b2fbd54SBarry Smith     for ( i=0; i<nz; i++ ) (*ja)[i] = a->j[i] + 1;
6131625ec5SSatish Balay     for ( i=0; i<a->m+1; i++ ) (*ia)[i] = a->i[i] + 1;
626945ee14SBarry Smith   } else {
636945ee14SBarry Smith     *ia = a->i; *ja = a->j;
64a2ce50c7SBarry Smith   }
65a2ce50c7SBarry Smith 
66*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
67a2744918SBarry Smith }
68a2744918SBarry Smith 
695615d1e5SSatish Balay #undef __FUNC__
705615d1e5SSatish Balay #define __FUNC__ "MatRestoreRowIJ_SeqAIJ"
718f6be9afSLois Curfman McInnes int MatRestoreRowIJ_SeqAIJ(Mat A,int oshift,PetscTruth symmetric,int *n,int **ia,int **ja,
726945ee14SBarry Smith                                PetscTruth *done)
736945ee14SBarry Smith {
746945ee14SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
753b2fbd54SBarry Smith   int        ishift = a->indexshift;
766945ee14SBarry Smith 
77*3a40ed3dSBarry Smith   PetscFunctionBegin;
78*3a40ed3dSBarry Smith   if (!ia) PetscFunctionReturn(0);
793b2fbd54SBarry Smith   if (symmetric || (oshift == 0 && ishift == -1) || (oshift == 1 && ishift == 0)) {
806945ee14SBarry Smith     PetscFree(*ia);
816945ee14SBarry Smith     PetscFree(*ja);
82bcd2baecSBarry Smith   }
83*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
8417ab2063SBarry Smith }
8517ab2063SBarry Smith 
865615d1e5SSatish Balay #undef __FUNC__
875615d1e5SSatish Balay #define __FUNC__ "MatGetColumnIJ_SeqAIJ"
8843a90d84SBarry Smith int MatGetColumnIJ_SeqAIJ(Mat A,int oshift,PetscTruth symmetric,int *nn,int **ia,int **ja,
893b2fbd54SBarry Smith                            PetscTruth *done)
903b2fbd54SBarry Smith {
913b2fbd54SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
92a93ec695SBarry Smith   int        ierr,i,ishift = a->indexshift,*collengths,*cia,*cja,n = A->n,m = A->m;
93a93ec695SBarry Smith   int        nz = a->i[m]+ishift,row,*jj,mr,col;
943b2fbd54SBarry Smith 
95*3a40ed3dSBarry Smith   PetscFunctionBegin;
963b2fbd54SBarry Smith   *nn     = A->n;
97*3a40ed3dSBarry Smith   if (!ia) PetscFunctionReturn(0);
983b2fbd54SBarry Smith   if (symmetric) {
99179192dfSSatish Balay     ierr = MatToSymmetricIJ_SeqAIJ(a->m,a->i,a->j,ishift,oshift,ia,ja); CHKERRQ(ierr);
1003b2fbd54SBarry Smith   } else {
10161d2ded1SBarry Smith     collengths = (int *) PetscMalloc( (n+1)*sizeof(int) ); CHKPTRQ(collengths);
1023b2fbd54SBarry Smith     PetscMemzero(collengths,n*sizeof(int));
1033b2fbd54SBarry Smith     cia        = (int *) PetscMalloc( (n+1)*sizeof(int) ); CHKPTRQ(cia);
104a93ec695SBarry Smith     cja        = (int *) PetscMalloc( (nz+1)*sizeof(int) ); CHKPTRQ(cja);
1053b2fbd54SBarry Smith     jj = a->j;
1063b2fbd54SBarry Smith     for ( i=0; i<nz; i++ ) {
1073b2fbd54SBarry Smith       collengths[jj[i] + ishift]++;
1083b2fbd54SBarry Smith     }
1093b2fbd54SBarry Smith     cia[0] = oshift;
1103b2fbd54SBarry Smith     for ( i=0; i<n; i++) {
1113b2fbd54SBarry Smith       cia[i+1] = cia[i] + collengths[i];
1123b2fbd54SBarry Smith     }
1133b2fbd54SBarry Smith     PetscMemzero(collengths,n*sizeof(int));
1143b2fbd54SBarry Smith     jj = a->j;
115a93ec695SBarry Smith     for ( row=0; row<m; row++ ) {
116a93ec695SBarry Smith       mr = a->i[row+1] - a->i[row];
117a93ec695SBarry Smith       for ( i=0; i<mr; i++ ) {
1183b2fbd54SBarry Smith         col = *jj++ + ishift;
1193b2fbd54SBarry Smith         cja[cia[col] + collengths[col]++ - oshift] = row + oshift;
1203b2fbd54SBarry Smith       }
1213b2fbd54SBarry Smith     }
1223b2fbd54SBarry Smith     PetscFree(collengths);
1233b2fbd54SBarry Smith     *ia = cia; *ja = cja;
1243b2fbd54SBarry Smith   }
1253b2fbd54SBarry Smith 
126*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
1273b2fbd54SBarry Smith }
1283b2fbd54SBarry Smith 
1295615d1e5SSatish Balay #undef __FUNC__
1305615d1e5SSatish Balay #define __FUNC__ "MatRestoreColumnIJ_SeqAIJ"
13143a90d84SBarry Smith int MatRestoreColumnIJ_SeqAIJ(Mat A,int oshift,PetscTruth symmetric,int *n,int **ia,
1323b2fbd54SBarry Smith                                      int **ja,PetscTruth *done)
1333b2fbd54SBarry Smith {
134*3a40ed3dSBarry Smith   PetscFunctionBegin;
135*3a40ed3dSBarry Smith   if (!ia) PetscFunctionReturn(0);
1363b2fbd54SBarry Smith 
1373b2fbd54SBarry Smith   PetscFree(*ia);
1383b2fbd54SBarry Smith   PetscFree(*ja);
1393b2fbd54SBarry Smith 
140*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
1413b2fbd54SBarry Smith }
1423b2fbd54SBarry Smith 
143227d817aSBarry Smith #define CHUNKSIZE   15
14417ab2063SBarry Smith 
1455615d1e5SSatish Balay #undef __FUNC__
1465615d1e5SSatish Balay #define __FUNC__ "MatSetValues_SeqAIJ"
14761d2ded1SBarry Smith int MatSetValues_SeqAIJ(Mat A,int m,int *im,int n,int *in,Scalar *v,InsertMode is)
14817ab2063SBarry Smith {
149416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
150416022c9SBarry Smith   int        *rp,k,low,high,t,ii,row,nrow,i,col,l,rmax, N, sorted = a->sorted;
1514b0e389bSBarry Smith   int        *imax = a->imax, *ai = a->i, *ailen = a->ilen,roworiented = a->roworiented;
152d5d45c9bSBarry Smith   int        *aj = a->j, nonew = a->nonew,shift = a->indexshift;
153416022c9SBarry Smith   Scalar     *ap,value, *aa = a->a;
15417ab2063SBarry Smith 
155*3a40ed3dSBarry Smith   PetscFunctionBegin;
15617ab2063SBarry Smith   for ( k=0; k<m; k++ ) { /* loop over added rows */
157416022c9SBarry Smith     row  = im[k];
158*3a40ed3dSBarry Smith #if defined(USE_PETSC_BOPT_g)
159e3372554SBarry Smith     if (row < 0) SETERRQ(1,0,"Negative row");
160e3372554SBarry Smith     if (row >= a->m) SETERRQ(1,0,"Row too large");
1613b2fbd54SBarry Smith #endif
16217ab2063SBarry Smith     rp   = aj + ai[row] + shift; ap = aa + ai[row] + shift;
16317ab2063SBarry Smith     rmax = imax[row]; nrow = ailen[row];
164416022c9SBarry Smith     low = 0;
16517ab2063SBarry Smith     for ( l=0; l<n; l++ ) { /* loop over added columns */
166*3a40ed3dSBarry Smith #if defined(USE_PETSC_BOPT_g)
167e3372554SBarry Smith       if (in[l] < 0) SETERRQ(1,0,"Negative column");
168e3372554SBarry Smith       if (in[l] >= a->n) SETERRQ(1,0,"Column too large");
1693b2fbd54SBarry Smith #endif
1704b0e389bSBarry Smith       col = in[l] - shift;
1714b0e389bSBarry Smith       if (roworiented) {
1724b0e389bSBarry Smith         value = *v++;
1734b0e389bSBarry Smith       }
1744b0e389bSBarry Smith       else {
1754b0e389bSBarry Smith         value = v[k + l*m];
1764b0e389bSBarry Smith       }
177416022c9SBarry Smith       if (!sorted) low = 0; high = nrow;
178416022c9SBarry Smith       while (high-low > 5) {
179416022c9SBarry Smith         t = (low+high)/2;
180416022c9SBarry Smith         if (rp[t] > col) high = t;
181416022c9SBarry Smith         else             low  = t;
18217ab2063SBarry Smith       }
183416022c9SBarry Smith       for ( i=low; i<high; i++ ) {
18417ab2063SBarry Smith         if (rp[i] > col) break;
18517ab2063SBarry Smith         if (rp[i] == col) {
186416022c9SBarry Smith           if (is == ADD_VALUES) ap[i] += value;
18717ab2063SBarry Smith           else                  ap[i] = value;
18817ab2063SBarry Smith           goto noinsert;
18917ab2063SBarry Smith         }
19017ab2063SBarry Smith       }
191c2653b3dSLois Curfman McInnes       if (nonew == 1) goto noinsert;
19211ebbc71SLois Curfman McInnes       else if (nonew == -1) SETERRQ(1,0,"Inserting a new nonzero in the matrix");
19317ab2063SBarry Smith       if (nrow >= rmax) {
19417ab2063SBarry Smith         /* there is no extra room in row, therefore enlarge */
195416022c9SBarry Smith         int    new_nz = ai[a->m] + CHUNKSIZE,len,*new_i,*new_j;
19617ab2063SBarry Smith         Scalar *new_a;
19717ab2063SBarry Smith 
19811ebbc71SLois Curfman McInnes         if (nonew == -2) SETERRQ(1,0,"Inserting a new nonzero in the matrix");
19996854ed6SLois Curfman McInnes 
20017ab2063SBarry Smith         /* malloc new storage space */
201416022c9SBarry Smith         len     = new_nz*(sizeof(int)+sizeof(Scalar))+(a->m+1)*sizeof(int);
2020452661fSBarry Smith         new_a   = (Scalar *) PetscMalloc( len ); CHKPTRQ(new_a);
20317ab2063SBarry Smith         new_j   = (int *) (new_a + new_nz);
20417ab2063SBarry Smith         new_i   = new_j + new_nz;
20517ab2063SBarry Smith 
20617ab2063SBarry Smith         /* copy over old data into new slots */
20717ab2063SBarry Smith         for ( ii=0; ii<row+1; ii++ ) {new_i[ii] = ai[ii];}
208416022c9SBarry Smith         for ( ii=row+1; ii<a->m+1; ii++ ) {new_i[ii] = ai[ii]+CHUNKSIZE;}
209416022c9SBarry Smith         PetscMemcpy(new_j,aj,(ai[row]+nrow+shift)*sizeof(int));
210416022c9SBarry Smith         len = (new_nz - CHUNKSIZE - ai[row] - nrow - shift);
211416022c9SBarry Smith         PetscMemcpy(new_j+ai[row]+shift+nrow+CHUNKSIZE,aj+ai[row]+shift+nrow,
21217ab2063SBarry Smith                                                            len*sizeof(int));
213416022c9SBarry Smith         PetscMemcpy(new_a,aa,(ai[row]+nrow+shift)*sizeof(Scalar));
214416022c9SBarry Smith         PetscMemcpy(new_a+ai[row]+shift+nrow+CHUNKSIZE,aa+ai[row]+shift+nrow,
21517ab2063SBarry Smith                                                            len*sizeof(Scalar));
21617ab2063SBarry Smith         /* free up old matrix storage */
2170452661fSBarry Smith         PetscFree(a->a);
2180452661fSBarry Smith         if (!a->singlemalloc) {PetscFree(a->i);PetscFree(a->j);}
219416022c9SBarry Smith         aa = a->a = new_a; ai = a->i = new_i; aj = a->j = new_j;
220416022c9SBarry Smith         a->singlemalloc = 1;
22117ab2063SBarry Smith 
22217ab2063SBarry Smith         rp   = aj + ai[row] + shift; ap = aa + ai[row] + shift;
223416022c9SBarry Smith         rmax = imax[row] = imax[row] + CHUNKSIZE;
224416022c9SBarry Smith         PLogObjectMemory(A,CHUNKSIZE*(sizeof(int) + sizeof(Scalar)));
225416022c9SBarry Smith         a->maxnz += CHUNKSIZE;
226b810aeb4SBarry Smith         a->reallocs++;
22717ab2063SBarry Smith       }
228416022c9SBarry Smith       N = nrow++ - 1; a->nz++;
229416022c9SBarry Smith       /* shift up all the later entries in this row */
230416022c9SBarry Smith       for ( ii=N; ii>=i; ii-- ) {
23117ab2063SBarry Smith         rp[ii+1] = rp[ii];
23217ab2063SBarry Smith         ap[ii+1] = ap[ii];
23317ab2063SBarry Smith       }
23417ab2063SBarry Smith       rp[i] = col;
23517ab2063SBarry Smith       ap[i] = value;
23617ab2063SBarry Smith       noinsert:;
237416022c9SBarry Smith       low = i + 1;
23817ab2063SBarry Smith     }
23917ab2063SBarry Smith     ailen[row] = nrow;
24017ab2063SBarry Smith   }
241*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
24217ab2063SBarry Smith }
24317ab2063SBarry Smith 
2445615d1e5SSatish Balay #undef __FUNC__
2455615d1e5SSatish Balay #define __FUNC__ "MatGetValues_SeqAIJ"
2468f6be9afSLois Curfman McInnes int MatGetValues_SeqAIJ(Mat A,int m,int *im,int n,int *in,Scalar *v)
2477eb43aa7SLois Curfman McInnes {
2487eb43aa7SLois Curfman McInnes   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
249b49de8d1SLois Curfman McInnes   int        *rp, k, low, high, t, row, nrow, i, col, l, *aj = a->j;
2507eb43aa7SLois Curfman McInnes   int        *ai = a->i, *ailen = a->ilen, shift = a->indexshift;
2517eb43aa7SLois Curfman McInnes   Scalar     *ap, *aa = a->a, zero = 0.0;
2527eb43aa7SLois Curfman McInnes 
253*3a40ed3dSBarry Smith   PetscFunctionBegin;
2547eb43aa7SLois Curfman McInnes   for ( k=0; k<m; k++ ) { /* loop over rows */
2557eb43aa7SLois Curfman McInnes     row  = im[k];
256e3372554SBarry Smith     if (row < 0) SETERRQ(1,0,"Negative row");
257e3372554SBarry Smith     if (row >= a->m) SETERRQ(1,0,"Row too large");
2587eb43aa7SLois Curfman McInnes     rp   = aj + ai[row] + shift; ap = aa + ai[row] + shift;
2597eb43aa7SLois Curfman McInnes     nrow = ailen[row];
2607eb43aa7SLois Curfman McInnes     for ( l=0; l<n; l++ ) { /* loop over columns */
261e3372554SBarry Smith       if (in[l] < 0) SETERRQ(1,0,"Negative column");
262e3372554SBarry Smith       if (in[l] >= a->n) SETERRQ(1,0,"Column too large");
2637eb43aa7SLois Curfman McInnes       col = in[l] - shift;
2647eb43aa7SLois Curfman McInnes       high = nrow; low = 0; /* assume unsorted */
2657eb43aa7SLois Curfman McInnes       while (high-low > 5) {
2667eb43aa7SLois Curfman McInnes         t = (low+high)/2;
2677eb43aa7SLois Curfman McInnes         if (rp[t] > col) high = t;
2687eb43aa7SLois Curfman McInnes         else             low  = t;
2697eb43aa7SLois Curfman McInnes       }
2707eb43aa7SLois Curfman McInnes       for ( i=low; i<high; i++ ) {
2717eb43aa7SLois Curfman McInnes         if (rp[i] > col) break;
2727eb43aa7SLois Curfman McInnes         if (rp[i] == col) {
273b49de8d1SLois Curfman McInnes           *v++ = ap[i];
2747eb43aa7SLois Curfman McInnes           goto finished;
2757eb43aa7SLois Curfman McInnes         }
2767eb43aa7SLois Curfman McInnes       }
277b49de8d1SLois Curfman McInnes       *v++ = zero;
2787eb43aa7SLois Curfman McInnes       finished:;
2797eb43aa7SLois Curfman McInnes     }
2807eb43aa7SLois Curfman McInnes   }
281*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
2827eb43aa7SLois Curfman McInnes }
2837eb43aa7SLois Curfman McInnes 
28417ab2063SBarry Smith 
2855615d1e5SSatish Balay #undef __FUNC__
2865615d1e5SSatish Balay #define __FUNC__ "MatView_SeqAIJ_Binary"
2878f6be9afSLois Curfman McInnes extern int MatView_SeqAIJ_Binary(Mat A,Viewer viewer)
28817ab2063SBarry Smith {
289416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
290416022c9SBarry Smith   int        i, fd, *col_lens, ierr;
29117ab2063SBarry Smith 
292*3a40ed3dSBarry Smith   PetscFunctionBegin;
29390ace30eSBarry Smith   ierr = ViewerBinaryGetDescriptor(viewer,&fd); CHKERRQ(ierr);
2940452661fSBarry Smith   col_lens = (int *) PetscMalloc( (4+a->m)*sizeof(int) ); CHKPTRQ(col_lens);
295416022c9SBarry Smith   col_lens[0] = MAT_COOKIE;
296416022c9SBarry Smith   col_lens[1] = a->m;
297416022c9SBarry Smith   col_lens[2] = a->n;
298416022c9SBarry Smith   col_lens[3] = a->nz;
299416022c9SBarry Smith 
300416022c9SBarry Smith   /* store lengths of each row and write (including header) to file */
301416022c9SBarry Smith   for ( i=0; i<a->m; i++ ) {
302416022c9SBarry Smith     col_lens[4+i] = a->i[i+1] - a->i[i];
30317ab2063SBarry Smith   }
3040752156aSBarry Smith   ierr = PetscBinaryWrite(fd,col_lens,4+a->m,PETSC_INT,1); CHKERRQ(ierr);
3050452661fSBarry Smith   PetscFree(col_lens);
306416022c9SBarry Smith 
307416022c9SBarry Smith   /* store column indices (zero start index) */
308416022c9SBarry Smith   if (a->indexshift) {
309416022c9SBarry Smith     for ( i=0; i<a->nz; i++ ) a->j[i]--;
31017ab2063SBarry Smith   }
3110752156aSBarry Smith   ierr = PetscBinaryWrite(fd,a->j,a->nz,PETSC_INT,0); CHKERRQ(ierr);
312416022c9SBarry Smith   if (a->indexshift) {
313416022c9SBarry Smith     for ( i=0; i<a->nz; i++ ) a->j[i]++;
31417ab2063SBarry Smith   }
315416022c9SBarry Smith 
316416022c9SBarry Smith   /* store nonzero values */
3170752156aSBarry Smith   ierr = PetscBinaryWrite(fd,a->a,a->nz,PETSC_SCALAR,0); CHKERRQ(ierr);
318*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
31917ab2063SBarry Smith }
320416022c9SBarry Smith 
3215615d1e5SSatish Balay #undef __FUNC__
3225615d1e5SSatish Balay #define __FUNC__ "MatView_SeqAIJ_ASCII"
3238f6be9afSLois Curfman McInnes extern int MatView_SeqAIJ_ASCII(Mat A,Viewer viewer)
324416022c9SBarry Smith {
325416022c9SBarry Smith   Mat_SeqAIJ  *a = (Mat_SeqAIJ *) A->data;
326496e697eSBarry Smith   int         ierr, i,j, m = a->m, shift = a->indexshift, format, flg1,flg2;
32717ab2063SBarry Smith   FILE        *fd;
32817ab2063SBarry Smith   char        *outputname;
32917ab2063SBarry Smith 
330*3a40ed3dSBarry Smith   PetscFunctionBegin;
33190ace30eSBarry Smith   ierr = ViewerASCIIGetPointer(viewer,&fd); CHKERRQ(ierr);
332416022c9SBarry Smith   ierr = ViewerFileGetOutputname_Private(viewer,&outputname); CHKERRQ(ierr);
33390ace30eSBarry Smith   ierr = ViewerGetFormat(viewer,&format);
334a93ec695SBarry Smith   if (format == VIEWER_FORMAT_ASCII_INFO) {
335*3a40ed3dSBarry Smith     PetscFunctionReturn(0);
336*3a40ed3dSBarry Smith   } else if (format == VIEWER_FORMAT_ASCII_INFO_LONG) {
337496e697eSBarry Smith     ierr = OptionsHasName(PETSC_NULL,"-mat_aij_no_inode",&flg1); CHKERRQ(ierr);
338496e697eSBarry Smith     ierr = OptionsHasName(PETSC_NULL,"-mat_no_unroll",&flg2); CHKERRQ(ierr);
339496e697eSBarry Smith     if (flg1 || flg2) fprintf(fd,"  not using I-node routines\n");
34095e01e2fSLois Curfman McInnes     else     fprintf(fd,"  using I-node routines: found %d nodes, limit used is %d\n",
34195e01e2fSLois Curfman McInnes         a->inode.node_count,a->inode.limit);
342*3a40ed3dSBarry Smith   } else if (format == VIEWER_FORMAT_ASCII_MATLAB) {
343d00d2cf4SBarry Smith     int nofinalvalue = 0;
344d00d2cf4SBarry Smith     if ((a->i[m] == a->i[m-1]) || (a->j[a->nz-1] != a->n-!shift)) {
345d00d2cf4SBarry Smith       nofinalvalue = 1;
346d00d2cf4SBarry Smith     }
347416022c9SBarry Smith     fprintf(fd,"%% Size = %d %d \n",m,a->n);
3484e220ebcSLois Curfman McInnes     fprintf(fd,"%% Nonzeros = %d \n",a->nz);
349d00d2cf4SBarry Smith     fprintf(fd,"zzz = zeros(%d,3);\n",a->nz+nofinalvalue);
35017ab2063SBarry Smith     fprintf(fd,"zzz = [\n");
35117ab2063SBarry Smith 
35217ab2063SBarry Smith     for (i=0; i<m; i++) {
353416022c9SBarry Smith       for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
354*3a40ed3dSBarry Smith #if defined(USE_PETSC_COMPLEX)
3556945ee14SBarry Smith         fprintf(fd,"%d %d  %18.16e + %18.16e i \n",i+1,a->j[j]+!shift,real(a->a[j]),
356416022c9SBarry Smith                    imag(a->a[j]));
35717ab2063SBarry Smith #else
3587a743949SBarry Smith         fprintf(fd,"%d %d  %18.16e\n", i+1, a->j[j]+!shift, a->a[j]);
35917ab2063SBarry Smith #endif
36017ab2063SBarry Smith       }
36117ab2063SBarry Smith     }
362d00d2cf4SBarry Smith     if (nofinalvalue) {
363d00d2cf4SBarry Smith       fprintf(fd,"%d %d  %18.16e\n", m, a->n, 0.0);
364d00d2cf4SBarry Smith     }
36517ab2063SBarry Smith     fprintf(fd,"];\n %s = spconvert(zzz);\n",outputname);
366*3a40ed3dSBarry Smith   } else if (format == VIEWER_FORMAT_ASCII_COMMON) {
36744cd7ae7SLois Curfman McInnes     for ( i=0; i<m; i++ ) {
36844cd7ae7SLois Curfman McInnes       fprintf(fd,"row %d:",i);
36944cd7ae7SLois Curfman McInnes       for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
370*3a40ed3dSBarry Smith #if defined(USE_PETSC_COMPLEX)
371766eeae4SLois Curfman McInnes         if (imag(a->a[j]) > 0.0 && real(a->a[j]) != 0.0)
37244cd7ae7SLois Curfman McInnes           fprintf(fd," %d %g + %g i",a->j[j]+shift,real(a->a[j]),imag(a->a[j]));
373766eeae4SLois Curfman McInnes         else if (imag(a->a[j]) < 0.0 && real(a->a[j]) != 0.0)
374766eeae4SLois Curfman McInnes           fprintf(fd," %d %g - %g i",a->j[j]+shift,real(a->a[j]),-imag(a->a[j]));
37544cd7ae7SLois Curfman McInnes         else if (real(a->a[j]) != 0.0)
37644cd7ae7SLois Curfman McInnes           fprintf(fd," %d %g ",a->j[j]+shift,real(a->a[j]));
37744cd7ae7SLois Curfman McInnes #else
37844cd7ae7SLois Curfman McInnes         if (a->a[j] != 0.0) fprintf(fd," %d %g ",a->j[j]+shift,a->a[j]);
37944cd7ae7SLois Curfman McInnes #endif
38044cd7ae7SLois Curfman McInnes       }
38144cd7ae7SLois Curfman McInnes       fprintf(fd,"\n");
38244cd7ae7SLois Curfman McInnes     }
38344cd7ae7SLois Curfman McInnes   }
384496be53dSLois Curfman McInnes   else if (format == VIEWER_FORMAT_ASCII_SYMMODU) {
385496be53dSLois Curfman McInnes     int nzd=0, fshift=1, *sptr;
3862e44a96cSLois Curfman McInnes     sptr = (int *) PetscMalloc( (m+1)*sizeof(int) ); CHKPTRQ(sptr);
387496be53dSLois Curfman McInnes     for ( i=0; i<m; i++ ) {
388496be53dSLois Curfman McInnes       sptr[i] = nzd+1;
389496be53dSLois Curfman McInnes       for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
390496be53dSLois Curfman McInnes         if (a->j[j] >= i) {
391*3a40ed3dSBarry Smith #if defined(USE_PETSC_COMPLEX)
392496be53dSLois Curfman McInnes           if (imag(a->a[j]) != 0.0 || real(a->a[j]) != 0.0) nzd++;
393496be53dSLois Curfman McInnes #else
394496be53dSLois Curfman McInnes           if (a->a[j] != 0.0) nzd++;
395496be53dSLois Curfman McInnes #endif
396496be53dSLois Curfman McInnes         }
397496be53dSLois Curfman McInnes       }
398496be53dSLois Curfman McInnes     }
3992e44a96cSLois Curfman McInnes     sptr[m] = nzd+1;
400496be53dSLois Curfman McInnes     fprintf(fd," %d %d\n\n",m,nzd);
4012e44a96cSLois Curfman McInnes     for ( i=0; i<m+1; i+=6 ) {
4022e44a96cSLois Curfman McInnes       if (i+4<m) fprintf(fd," %d %d %d %d %d %d\n",sptr[i],sptr[i+1],sptr[i+2],sptr[i+3],sptr[i+4],sptr[i+5]);
4032e44a96cSLois Curfman McInnes       else if (i+3<m) fprintf(fd," %d %d %d %d %d\n",sptr[i],sptr[i+1],sptr[i+2],sptr[i+3],sptr[i+4]);
4042e44a96cSLois Curfman McInnes       else if (i+2<m) fprintf(fd," %d %d %d %d\n",sptr[i],sptr[i+1],sptr[i+2],sptr[i+3]);
4052e44a96cSLois Curfman McInnes       else if (i+1<m) fprintf(fd," %d %d %d\n",sptr[i],sptr[i+1],sptr[i+2]);
4062e44a96cSLois Curfman McInnes       else if (i<m)   fprintf(fd," %d %d\n",sptr[i],sptr[i+1]);
4077272d637SLois Curfman McInnes       else            fprintf(fd," %d\n",sptr[i]);
408496be53dSLois Curfman McInnes     }
409496be53dSLois Curfman McInnes     fprintf(fd,"\n");
410496be53dSLois Curfman McInnes     PetscFree(sptr);
411496be53dSLois Curfman McInnes     for ( i=0; i<m; i++ ) {
412496be53dSLois Curfman McInnes       for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
413496be53dSLois Curfman McInnes         if (a->j[j] >= i) fprintf(fd," %d ",a->j[j]+fshift);
414496be53dSLois Curfman McInnes       }
415496be53dSLois Curfman McInnes       fprintf(fd,"\n");
416496be53dSLois Curfman McInnes     }
417496be53dSLois Curfman McInnes     fprintf(fd,"\n");
418496be53dSLois Curfman McInnes     for ( i=0; i<m; i++ ) {
419496be53dSLois Curfman McInnes       for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
420496be53dSLois Curfman McInnes         if (a->j[j] >= i) {
421*3a40ed3dSBarry Smith #if defined(USE_PETSC_COMPLEX)
422496be53dSLois Curfman McInnes           if (imag(a->a[j]) != 0.0 || real(a->a[j]) != 0.0)
423496be53dSLois Curfman McInnes             fprintf(fd," %18.16e %18.16e ",real(a->a[j]),imag(a->a[j]));
424496be53dSLois Curfman McInnes #else
425496be53dSLois Curfman McInnes           if (a->a[j] != 0.0) fprintf(fd," %18.16e ",a->a[j]);
426496be53dSLois Curfman McInnes #endif
427496be53dSLois Curfman McInnes         }
428496be53dSLois Curfman McInnes       }
429496be53dSLois Curfman McInnes       fprintf(fd,"\n");
430496be53dSLois Curfman McInnes     }
431*3a40ed3dSBarry Smith   } else {
43217ab2063SBarry Smith     for ( i=0; i<m; i++ ) {
43317ab2063SBarry Smith       fprintf(fd,"row %d:",i);
434416022c9SBarry Smith       for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
435*3a40ed3dSBarry Smith #if defined(USE_PETSC_COMPLEX)
436766eeae4SLois Curfman McInnes         if (imag(a->a[j]) > 0.0) {
437416022c9SBarry Smith           fprintf(fd," %d %g + %g i",a->j[j]+shift,real(a->a[j]),imag(a->a[j]));
438766eeae4SLois Curfman McInnes         } else if (imag(a->a[j]) < 0.0) {
439766eeae4SLois Curfman McInnes           fprintf(fd," %d %g - %g i",a->j[j]+shift,real(a->a[j]),-imag(a->a[j]));
440*3a40ed3dSBarry Smith         } else {
441416022c9SBarry Smith           fprintf(fd," %d %g ",a->j[j]+shift,real(a->a[j]));
44217ab2063SBarry Smith         }
44317ab2063SBarry Smith #else
444416022c9SBarry Smith         fprintf(fd," %d %g ",a->j[j]+shift,a->a[j]);
44517ab2063SBarry Smith #endif
44617ab2063SBarry Smith       }
44717ab2063SBarry Smith       fprintf(fd,"\n");
44817ab2063SBarry Smith     }
44917ab2063SBarry Smith   }
45017ab2063SBarry Smith   fflush(fd);
451*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
452416022c9SBarry Smith }
453416022c9SBarry Smith 
4545615d1e5SSatish Balay #undef __FUNC__
4555615d1e5SSatish Balay #define __FUNC__ "MatView_SeqAIJ_Draw"
4568f6be9afSLois Curfman McInnes extern int MatView_SeqAIJ_Draw(Mat A,Viewer viewer)
457416022c9SBarry Smith {
458416022c9SBarry Smith   Mat_SeqAIJ  *a = (Mat_SeqAIJ *) A->data;
459cddf8d76SBarry Smith   int         ierr, i,j, m = a->m, shift = a->indexshift,pause,color;
4600513a670SBarry Smith   int         format;
46194a9d846SBarry Smith   double      xl,yl,xr,yr,w,h,xc,yc,scale = 1.0,x_l,x_r,y_l,y_r,maxv = 0.0;
462bcd2baecSBarry Smith   Draw        draw;
463cddf8d76SBarry Smith   DrawButton  button;
46419bcc07fSBarry Smith   PetscTruth  isnull;
465cddf8d76SBarry Smith 
466*3a40ed3dSBarry Smith   PetscFunctionBegin;
4670513a670SBarry Smith   ierr = ViewerDrawGetDraw(viewer,&draw); CHKERRQ(ierr);
4680513a670SBarry Smith   ierr = DrawClear(draw); CHKERRQ(ierr);
4690513a670SBarry Smith   ierr = ViewerGetFormat(viewer,&format); CHKERRQ(ierr);
470*3a40ed3dSBarry Smith   ierr = DrawIsNull(draw,&isnull); CHKERRQ(ierr); if (isnull) PetscFunctionReturn(0);
47119bcc07fSBarry Smith 
472416022c9SBarry Smith   xr  = a->n; yr = a->m; h = yr/10.0; w = xr/10.0;
473416022c9SBarry Smith   xr += w;    yr += h;  xl = -w;     yl = -h;
474416022c9SBarry Smith   ierr = DrawSetCoordinates(draw,xl,yl,xr,yr); CHKERRQ(ierr);
475416022c9SBarry Smith   /* loop over matrix elements drawing boxes */
4760513a670SBarry Smith 
4770513a670SBarry Smith   if (format != VIEWER_FORMAT_DRAW_CONTOUR) {
4780513a670SBarry Smith     /* Blue for negative, Cyan for zero and  Red for positive */
479cddf8d76SBarry Smith     color = DRAW_BLUE;
480416022c9SBarry Smith     for ( i=0; i<m; i++ ) {
481cddf8d76SBarry Smith       y_l = m - i - 1.0; y_r = y_l + 1.0;
482416022c9SBarry Smith       for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
483cddf8d76SBarry Smith         x_l = a->j[j] + shift; x_r = x_l + 1.0;
484*3a40ed3dSBarry Smith #if defined(USE_PETSC_COMPLEX)
485cddf8d76SBarry Smith         if (real(a->a[j]) >=  0.) continue;
486cddf8d76SBarry Smith #else
487cddf8d76SBarry Smith         if (a->a[j] >=  0.) continue;
488cddf8d76SBarry Smith #endif
489cddf8d76SBarry Smith         DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);
490cddf8d76SBarry Smith       }
491cddf8d76SBarry Smith     }
492cddf8d76SBarry Smith     color = DRAW_CYAN;
493cddf8d76SBarry Smith     for ( i=0; i<m; i++ ) {
494cddf8d76SBarry Smith       y_l = m - i - 1.0; y_r = y_l + 1.0;
495cddf8d76SBarry Smith       for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
496cddf8d76SBarry Smith         x_l = a->j[j] + shift; x_r = x_l + 1.0;
497cddf8d76SBarry Smith         if (a->a[j] !=  0.) continue;
498cddf8d76SBarry Smith         DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);
499cddf8d76SBarry Smith       }
500cddf8d76SBarry Smith     }
501cddf8d76SBarry Smith     color = DRAW_RED;
502cddf8d76SBarry Smith     for ( i=0; i<m; i++ ) {
503cddf8d76SBarry Smith       y_l = m - i - 1.0; y_r = y_l + 1.0;
504cddf8d76SBarry Smith       for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
505cddf8d76SBarry Smith         x_l = a->j[j] + shift; x_r = x_l + 1.0;
506*3a40ed3dSBarry Smith #if defined(USE_PETSC_COMPLEX)
507cddf8d76SBarry Smith         if (real(a->a[j]) <=  0.) continue;
508cddf8d76SBarry Smith #else
509cddf8d76SBarry Smith         if (a->a[j] <=  0.) continue;
510cddf8d76SBarry Smith #endif
511cddf8d76SBarry Smith         DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);
512416022c9SBarry Smith       }
513416022c9SBarry Smith     }
5140513a670SBarry Smith   } else {
5150513a670SBarry Smith     /* use contour shading to indicate magnitude of values */
5160513a670SBarry Smith     /* first determine max of all nonzero values */
5170513a670SBarry Smith     int    nz = a->nz,count;
5180513a670SBarry Smith     Draw   popup;
5190513a670SBarry Smith 
5200513a670SBarry Smith     for ( i=0; i<nz; i++ ) {
5210513a670SBarry Smith       if (PetscAbsScalar(a->a[i]) > maxv) maxv = PetscAbsScalar(a->a[i]);
5220513a670SBarry Smith     }
5230513a670SBarry Smith     ierr = DrawCreatePopUp(draw,&popup); CHKERRQ(ierr);
5240513a670SBarry Smith     ierr = DrawScalePopup(popup,0.0,maxv); CHKERRQ(ierr);
5250513a670SBarry Smith     count = 0;
5260513a670SBarry Smith     for ( i=0; i<m; i++ ) {
5270513a670SBarry Smith       y_l = m - i - 1.0; y_r = y_l + 1.0;
5280513a670SBarry Smith       for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
5290513a670SBarry Smith         x_l = a->j[j] + shift; x_r = x_l + 1.0;
5300513a670SBarry Smith         color = 32 + (int) ((200.0 - 32.0)*PetscAbsScalar(a->a[count])/maxv);
5310513a670SBarry Smith         DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);
5320513a670SBarry Smith         count++;
5330513a670SBarry Smith       }
5340513a670SBarry Smith     }
5350513a670SBarry Smith   }
536416022c9SBarry Smith   DrawFlush(draw);
537cddf8d76SBarry Smith   DrawGetPause(draw,&pause);
538*3a40ed3dSBarry Smith   if (pause >= 0) { PetscSleep(pause); PetscFunctionReturn(0);}
539cddf8d76SBarry Smith 
540cddf8d76SBarry Smith   /* allow the matrix to zoom or shrink */
5416945ee14SBarry Smith   ierr = DrawCheckResizedWindow(draw);
542cddf8d76SBarry Smith   ierr = DrawGetMouseButton(draw,&button,&xc,&yc,0,0);
543cddf8d76SBarry Smith   while (button != BUTTON_RIGHT) {
544cddf8d76SBarry Smith     DrawClear(draw);
545cddf8d76SBarry Smith     if (button == BUTTON_LEFT) scale = .5;
546cddf8d76SBarry Smith     else if (button == BUTTON_CENTER) scale = 2.;
547cddf8d76SBarry Smith     xl = scale*(xl + w - xc) + xc - w*scale;
548cddf8d76SBarry Smith     xr = scale*(xr - w - xc) + xc + w*scale;
549cddf8d76SBarry Smith     yl = scale*(yl + h - yc) + yc - h*scale;
550cddf8d76SBarry Smith     yr = scale*(yr - h - yc) + yc + h*scale;
551cddf8d76SBarry Smith     w *= scale; h *= scale;
552cddf8d76SBarry Smith     ierr = DrawSetCoordinates(draw,xl,yl,xr,yr); CHKERRQ(ierr);
5530513a670SBarry Smith     if (format != VIEWER_FORMAT_DRAW_CONTOUR) {
5540513a670SBarry Smith       /* Blue for negative, Cyan for zero and  Red for positive */
555cddf8d76SBarry Smith       color = DRAW_BLUE;
556cddf8d76SBarry Smith       for ( i=0; i<m; i++ ) {
557cddf8d76SBarry Smith         y_l = m - i - 1.0; y_r = y_l + 1.0;
558cddf8d76SBarry Smith         for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
559cddf8d76SBarry Smith           x_l = a->j[j] + shift; x_r = x_l + 1.0;
560*3a40ed3dSBarry Smith #if defined(USE_PETSC_COMPLEX)
561cddf8d76SBarry Smith           if (real(a->a[j]) >=  0.) continue;
562cddf8d76SBarry Smith #else
563cddf8d76SBarry Smith           if (a->a[j] >=  0.) continue;
564cddf8d76SBarry Smith #endif
565cddf8d76SBarry Smith           DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);
566cddf8d76SBarry Smith         }
567cddf8d76SBarry Smith       }
568cddf8d76SBarry Smith       color = DRAW_CYAN;
569cddf8d76SBarry Smith       for ( i=0; i<m; i++ ) {
570cddf8d76SBarry Smith         y_l = m - i - 1.0; y_r = y_l + 1.0;
571cddf8d76SBarry Smith         for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
572cddf8d76SBarry Smith           x_l = a->j[j] + shift; x_r = x_l + 1.0;
573cddf8d76SBarry Smith           if (a->a[j] !=  0.) continue;
574cddf8d76SBarry Smith           DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);
575cddf8d76SBarry Smith         }
576cddf8d76SBarry Smith       }
577cddf8d76SBarry Smith       color = DRAW_RED;
578cddf8d76SBarry Smith       for ( i=0; i<m; i++ ) {
579cddf8d76SBarry Smith         y_l = m - i - 1.0; y_r = y_l + 1.0;
580cddf8d76SBarry Smith         for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
581cddf8d76SBarry Smith           x_l = a->j[j] + shift; x_r = x_l + 1.0;
582*3a40ed3dSBarry Smith #if defined(USE_PETSC_COMPLEX)
583cddf8d76SBarry Smith           if (real(a->a[j]) <=  0.) continue;
584cddf8d76SBarry Smith #else
585cddf8d76SBarry Smith           if (a->a[j] <=  0.) continue;
586cddf8d76SBarry Smith #endif
587cddf8d76SBarry Smith           DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);
588cddf8d76SBarry Smith         }
589cddf8d76SBarry Smith       }
5900513a670SBarry Smith     } else {
5910513a670SBarry Smith       /* use contour shading to indicate magnitude of values */
5920513a670SBarry Smith       int count = 0;
5930513a670SBarry Smith       for ( i=0; i<m; i++ ) {
5940513a670SBarry Smith         y_l = m - i - 1.0; y_r = y_l + 1.0;
5950513a670SBarry Smith         for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
5960513a670SBarry Smith           x_l = a->j[j] + shift; x_r = x_l + 1.0;
5970513a670SBarry Smith           color = 32 + (int) ((200.0 - 32.0)*PetscAbsScalar(a->a[count])/maxv);
598*3a40ed3dSBarry Smith           DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color); CHKERRQ(ierr);
5990513a670SBarry Smith           count++;
6000513a670SBarry Smith         }
6010513a670SBarry Smith       }
6020513a670SBarry Smith     }
6030513a670SBarry Smith 
604*3a40ed3dSBarry Smith     ierr = DrawCheckResizedWindow(draw); CHKERRQ(ierr);
605*3a40ed3dSBarry Smith     ierr = DrawGetMouseButton(draw,&button,&xc,&yc,0,0);  CHKERRQ(ierr);
606cddf8d76SBarry Smith   }
607*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
608416022c9SBarry Smith }
609416022c9SBarry Smith 
6105615d1e5SSatish Balay #undef __FUNC__
611d4bb536fSBarry Smith #define __FUNC__ "MatView_SeqAIJ"
6128f6be9afSLois Curfman McInnes int MatView_SeqAIJ(PetscObject obj,Viewer viewer)
613416022c9SBarry Smith {
614416022c9SBarry Smith   Mat         A = (Mat) obj;
615416022c9SBarry Smith   Mat_SeqAIJ  *a = (Mat_SeqAIJ*) A->data;
616bcd2baecSBarry Smith   ViewerType  vtype;
617bcd2baecSBarry Smith   int         ierr;
618416022c9SBarry Smith 
619*3a40ed3dSBarry Smith   PetscFunctionBegin;
620bcd2baecSBarry Smith   ierr = ViewerGetType(viewer,&vtype); CHKERRQ(ierr);
621bcd2baecSBarry Smith   if (vtype == MATLAB_VIEWER) {
622*3a40ed3dSBarry Smith     ierr = ViewerMatlabPutSparse_Private(viewer,a->m,a->n,a->nz,a->a,a->i,a->j);  CHKERRQ(ierr);
623416022c9SBarry Smith   }
624bcd2baecSBarry Smith   else if (vtype == ASCII_FILE_VIEWER || vtype == ASCII_FILES_VIEWER){
625*3a40ed3dSBarry Smith     ierr = MatView_SeqAIJ_ASCII(A,viewer); CHKERRQ(ierr);
626416022c9SBarry Smith   }
627bcd2baecSBarry Smith   else if (vtype == BINARY_FILE_VIEWER) {
628*3a40ed3dSBarry Smith     ierr = MatView_SeqAIJ_Binary(A,viewer); CHKERRQ(ierr);
629416022c9SBarry Smith   }
630bcd2baecSBarry Smith   else if (vtype == DRAW_VIEWER) {
631*3a40ed3dSBarry Smith     ierr = MatView_SeqAIJ_Draw(A,viewer); CHKERRQ(ierr);
63217ab2063SBarry Smith   }
633*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
63417ab2063SBarry Smith }
63519bcc07fSBarry Smith 
636c456f294SBarry Smith extern int Mat_AIJ_CheckInode(Mat);
6375615d1e5SSatish Balay #undef __FUNC__
6385615d1e5SSatish Balay #define __FUNC__ "MatAssemblyEnd_SeqAIJ"
6398f6be9afSLois Curfman McInnes int MatAssemblyEnd_SeqAIJ(Mat A,MatAssemblyType mode)
64017ab2063SBarry Smith {
641416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
64241c01911SSatish Balay   int        fshift = 0,i,j,*ai = a->i, *aj = a->j, *imax = a->imax,ierr;
64343ee02c3SBarry Smith   int        m = a->m, *ip, N, *ailen = a->ilen,shift = a->indexshift,rmax = 0;
644416022c9SBarry Smith   Scalar     *aa = a->a, *ap;
64517ab2063SBarry Smith 
646*3a40ed3dSBarry Smith   PetscFunctionBegin;
647*3a40ed3dSBarry Smith   if (mode == MAT_FLUSH_ASSEMBLY) PetscFunctionReturn(0);
64817ab2063SBarry Smith 
64943ee02c3SBarry Smith   if (m) rmax = ailen[0]; /* determine row with most nonzeros */
65017ab2063SBarry Smith   for ( i=1; i<m; i++ ) {
651416022c9SBarry Smith     /* move each row back by the amount of empty slots (fshift) before it*/
65217ab2063SBarry Smith     fshift += imax[i-1] - ailen[i-1];
65394a9d846SBarry Smith     rmax   = PetscMax(rmax,ailen[i]);
65417ab2063SBarry Smith     if (fshift) {
655416022c9SBarry Smith       ip = aj + ai[i] + shift; ap = aa + ai[i] + shift;
65617ab2063SBarry Smith       N = ailen[i];
65717ab2063SBarry Smith       for ( j=0; j<N; j++ ) {
65817ab2063SBarry Smith         ip[j-fshift] = ip[j];
65917ab2063SBarry Smith         ap[j-fshift] = ap[j];
66017ab2063SBarry Smith       }
66117ab2063SBarry Smith     }
66217ab2063SBarry Smith     ai[i] = ai[i-1] + ailen[i-1];
66317ab2063SBarry Smith   }
66417ab2063SBarry Smith   if (m) {
66517ab2063SBarry Smith     fshift += imax[m-1] - ailen[m-1];
66617ab2063SBarry Smith     ai[m] = ai[m-1] + ailen[m-1];
66717ab2063SBarry Smith   }
66817ab2063SBarry Smith   /* reset ilen and imax for each row */
66917ab2063SBarry Smith   for ( i=0; i<m; i++ ) {
67017ab2063SBarry Smith     ailen[i] = imax[i] = ai[i+1] - ai[i];
67117ab2063SBarry Smith   }
672416022c9SBarry Smith   a->nz = ai[m] + shift;
67317ab2063SBarry Smith 
67417ab2063SBarry Smith   /* diagonals may have moved, so kill the diagonal pointers */
675416022c9SBarry Smith   if (fshift && a->diag) {
6760452661fSBarry Smith     PetscFree(a->diag);
677416022c9SBarry Smith     PLogObjectMemory(A,-(m+1)*sizeof(int));
678416022c9SBarry Smith     a->diag = 0;
67917ab2063SBarry Smith   }
6804e220ebcSLois Curfman McInnes   PLogInfo(A,"MatAssemblyEnd_SeqAIJ:Matrix size: %d X %d; storage space: %d unneeded, %d used\n",
6814e220ebcSLois Curfman McInnes            m,a->n,fshift,a->nz);
6824e220ebcSLois Curfman McInnes   PLogInfo(A,"MatAssemblyEnd_SeqAIJ:Number of mallocs during MatSetValues is %d\n",
683b810aeb4SBarry Smith            a->reallocs);
68494a9d846SBarry Smith   PLogInfo(A,"MatAssemblyEnd_SeqAIJ:Most nonzeros in any row is %d\n",rmax);
685dd5f02e7SSatish Balay   a->reallocs          = 0;
6864e220ebcSLois Curfman McInnes   A->info.nz_unneeded  = (double)fshift;
6874e220ebcSLois Curfman McInnes 
68876dd722bSSatish Balay   /* check out for identical nodes. If found, use inode functions */
68941c01911SSatish Balay   ierr = Mat_AIJ_CheckInode(A); CHKERRQ(ierr);
690*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
69117ab2063SBarry Smith }
69217ab2063SBarry Smith 
6935615d1e5SSatish Balay #undef __FUNC__
6945615d1e5SSatish Balay #define __FUNC__ "MatZeroEntries_SeqAIJ"
6958f6be9afSLois Curfman McInnes int MatZeroEntries_SeqAIJ(Mat A)
69617ab2063SBarry Smith {
697416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
698*3a40ed3dSBarry Smith 
699*3a40ed3dSBarry Smith   PetscFunctionBegin;
700cddf8d76SBarry Smith   PetscMemzero(a->a,(a->i[a->m]+a->indexshift)*sizeof(Scalar));
701*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
70217ab2063SBarry Smith }
703416022c9SBarry Smith 
7045615d1e5SSatish Balay #undef __FUNC__
7055615d1e5SSatish Balay #define __FUNC__ "MatDestroy_SeqAIJ"
70617ab2063SBarry Smith int MatDestroy_SeqAIJ(PetscObject obj)
70717ab2063SBarry Smith {
708416022c9SBarry Smith   Mat        A  = (Mat) obj;
709416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
710d5d45c9bSBarry Smith 
711*3a40ed3dSBarry Smith   PetscFunctionBegin;
712*3a40ed3dSBarry Smith #if defined(USE_PETSC_LOG)
713416022c9SBarry Smith   PLogObjectState(obj,"Rows=%d, Cols=%d, NZ=%d",a->m,a->n,a->nz);
71417ab2063SBarry Smith #endif
7150452661fSBarry Smith   PetscFree(a->a);
7160452661fSBarry Smith   if (!a->singlemalloc) { PetscFree(a->i); PetscFree(a->j);}
7170452661fSBarry Smith   if (a->diag) PetscFree(a->diag);
7180452661fSBarry Smith   if (a->ilen) PetscFree(a->ilen);
7190452661fSBarry Smith   if (a->imax) PetscFree(a->imax);
7200452661fSBarry Smith   if (a->solve_work) PetscFree(a->solve_work);
72176dd722bSSatish Balay   if (a->inode.size) PetscFree(a->inode.size);
7220452661fSBarry Smith   PetscFree(a);
723eed86810SBarry Smith 
724f2655603SLois Curfman McInnes   PLogObjectDestroy(A);
725f2655603SLois Curfman McInnes   PetscHeaderDestroy(A);
726*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
72717ab2063SBarry Smith }
72817ab2063SBarry Smith 
7295615d1e5SSatish Balay #undef __FUNC__
7305615d1e5SSatish Balay #define __FUNC__ "MatCompress_SeqAIJ"
7318f6be9afSLois Curfman McInnes int MatCompress_SeqAIJ(Mat A)
73217ab2063SBarry Smith {
733*3a40ed3dSBarry Smith   PetscFunctionBegin;
734*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
73517ab2063SBarry Smith }
73617ab2063SBarry Smith 
7375615d1e5SSatish Balay #undef __FUNC__
7385615d1e5SSatish Balay #define __FUNC__ "MatSetOption_SeqAIJ"
7398f6be9afSLois Curfman McInnes int MatSetOption_SeqAIJ(Mat A,MatOption op)
74017ab2063SBarry Smith {
741416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
742*3a40ed3dSBarry Smith 
743*3a40ed3dSBarry Smith   PetscFunctionBegin;
7446d4a8577SBarry Smith   if      (op == MAT_ROW_ORIENTED)                 a->roworiented = 1;
7456d4a8577SBarry Smith   else if (op == MAT_COLUMN_ORIENTED)              a->roworiented = 0;
7466d4a8577SBarry Smith   else if (op == MAT_COLUMNS_SORTED)               a->sorted      = 1;
747219d9a1aSLois Curfman McInnes   else if (op == MAT_COLUMNS_UNSORTED)             a->sorted      = 0;
7486d4a8577SBarry Smith   else if (op == MAT_NO_NEW_NONZERO_LOCATIONS)     a->nonew       = 1;
749c2653b3dSLois Curfman McInnes   else if (op == MAT_NEW_NONZERO_LOCATION_ERROR)   a->nonew       = -1;
75096854ed6SLois Curfman McInnes   else if (op == MAT_NEW_NONZERO_ALLOCATION_ERROR) a->nonew       = -2;
7516d4a8577SBarry Smith   else if (op == MAT_YES_NEW_NONZERO_LOCATIONS)    a->nonew       = 0;
7526d4a8577SBarry Smith   else if (op == MAT_ROWS_SORTED ||
753219d9a1aSLois Curfman McInnes            op == MAT_ROWS_UNSORTED ||
7546d4a8577SBarry Smith            op == MAT_SYMMETRIC ||
7556d4a8577SBarry Smith            op == MAT_STRUCTURALLY_SYMMETRIC ||
75690f02eecSBarry Smith            op == MAT_YES_NEW_DIAGONALS ||
7572b362799SSatish Balay            op == MAT_IGNORE_OFF_PROC_ENTRIES)
75894a424c1SBarry Smith     PLogInfo(A,"Info:MatSetOption_SeqAIJ:Option ignored\n");
759*3a40ed3dSBarry Smith   else if (op == MAT_NO_NEW_DIAGONALS) {
760*3a40ed3dSBarry Smith     SETERRQ(PETSC_ERR_SUP,0,"MAT_NO_NEW_DIAGONALS");
761*3a40ed3dSBarry Smith   } else if (op == MAT_INODE_LIMIT_1)            a->inode.limit  = 1;
7626d4a8577SBarry Smith   else if (op == MAT_INODE_LIMIT_2)            a->inode.limit  = 2;
7636d4a8577SBarry Smith   else if (op == MAT_INODE_LIMIT_3)            a->inode.limit  = 3;
7646d4a8577SBarry Smith   else if (op == MAT_INODE_LIMIT_4)            a->inode.limit  = 4;
7656d4a8577SBarry Smith   else if (op == MAT_INODE_LIMIT_5)            a->inode.limit  = 5;
766*3a40ed3dSBarry Smith   else SETERRQ(PETSC_ERR_SUP,0,"unknown option");
767*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
76817ab2063SBarry Smith }
76917ab2063SBarry Smith 
7705615d1e5SSatish Balay #undef __FUNC__
7715615d1e5SSatish Balay #define __FUNC__ "MatGetDiagonal_SeqAIJ"
7728f6be9afSLois Curfman McInnes int MatGetDiagonal_SeqAIJ(Mat A,Vec v)
77317ab2063SBarry Smith {
774416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
775*3a40ed3dSBarry Smith   int        i,j, n,shift = a->indexshift,ierr;
77617ab2063SBarry Smith   Scalar     *x, zero = 0.0;
77717ab2063SBarry Smith 
778*3a40ed3dSBarry Smith   PetscFunctionBegin;
779*3a40ed3dSBarry Smith   ierr = VecSet(&zero,v);CHKERRQ(ierr);
78090f02eecSBarry Smith   VecGetArray_Fast(v,x); VecGetLocalSize(v,&n);
781e3372554SBarry Smith   if (n != a->m) SETERRQ(1,0,"Nonconforming matrix and vector");
782416022c9SBarry Smith   for ( i=0; i<a->m; i++ ) {
783416022c9SBarry Smith     for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
784416022c9SBarry Smith       if (a->j[j]+shift == i) {
785416022c9SBarry Smith         x[i] = a->a[j];
78617ab2063SBarry Smith         break;
78717ab2063SBarry Smith       }
78817ab2063SBarry Smith     }
78917ab2063SBarry Smith   }
790*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
79117ab2063SBarry Smith }
79217ab2063SBarry Smith 
79317ab2063SBarry Smith /* -------------------------------------------------------*/
79417ab2063SBarry Smith /* Should check that shapes of vectors and matrices match */
79517ab2063SBarry Smith /* -------------------------------------------------------*/
7965615d1e5SSatish Balay #undef __FUNC__
7975615d1e5SSatish Balay #define __FUNC__ "MatMultTrans_SeqAIJ"
79844cd7ae7SLois Curfman McInnes int MatMultTrans_SeqAIJ(Mat A,Vec xx,Vec yy)
79917ab2063SBarry Smith {
800416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
80117ab2063SBarry Smith   Scalar     *x, *y, *v, alpha;
802416022c9SBarry Smith   int        m = a->m, n, i, *idx, shift = a->indexshift;
80317ab2063SBarry Smith 
804*3a40ed3dSBarry Smith   PetscFunctionBegin;
80590f02eecSBarry Smith   VecGetArray_Fast(xx,x); VecGetArray_Fast(yy,y);
806cddf8d76SBarry Smith   PetscMemzero(y,a->n*sizeof(Scalar));
80717ab2063SBarry Smith   y = y + shift; /* shift for Fortran start by 1 indexing */
80817ab2063SBarry Smith   for ( i=0; i<m; i++ ) {
809416022c9SBarry Smith     idx   = a->j + a->i[i] + shift;
810416022c9SBarry Smith     v     = a->a + a->i[i] + shift;
811416022c9SBarry Smith     n     = a->i[i+1] - a->i[i];
81217ab2063SBarry Smith     alpha = x[i];
81317ab2063SBarry Smith     while (n-->0) {y[*idx++] += alpha * *v++;}
81417ab2063SBarry Smith   }
815416022c9SBarry Smith   PLogFlops(2*a->nz - a->n);
816*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
81717ab2063SBarry Smith }
818d5d45c9bSBarry Smith 
8195615d1e5SSatish Balay #undef __FUNC__
8205615d1e5SSatish Balay #define __FUNC__ "MatMultTransAdd_SeqAIJ"
82144cd7ae7SLois Curfman McInnes int MatMultTransAdd_SeqAIJ(Mat A,Vec xx,Vec zz,Vec yy)
82217ab2063SBarry Smith {
823416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
82417ab2063SBarry Smith   Scalar     *x, *y, *v, alpha;
825416022c9SBarry Smith   int        m = a->m, n, i, *idx,shift = a->indexshift;
82617ab2063SBarry Smith 
827*3a40ed3dSBarry Smith   PetscFunctionBegin;
82890f02eecSBarry Smith   VecGetArray_Fast(xx,x); VecGetArray_Fast(yy,y);
82917ab2063SBarry Smith   if (zz != yy) VecCopy(zz,yy);
83017ab2063SBarry Smith   y = y + shift; /* shift for Fortran start by 1 indexing */
83117ab2063SBarry Smith   for ( i=0; i<m; i++ ) {
832416022c9SBarry Smith     idx   = a->j + a->i[i] + shift;
833416022c9SBarry Smith     v     = a->a + a->i[i] + shift;
834416022c9SBarry Smith     n     = a->i[i+1] - a->i[i];
83517ab2063SBarry Smith     alpha = x[i];
83617ab2063SBarry Smith     while (n-->0) {y[*idx++] += alpha * *v++;}
83717ab2063SBarry Smith   }
83890f02eecSBarry Smith   PLogFlops(2*a->nz);
839*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
84017ab2063SBarry Smith }
84117ab2063SBarry Smith 
8425615d1e5SSatish Balay #undef __FUNC__
8435615d1e5SSatish Balay #define __FUNC__ "MatMult_SeqAIJ"
84444cd7ae7SLois Curfman McInnes int MatMult_SeqAIJ(Mat A,Vec xx,Vec yy)
84517ab2063SBarry Smith {
846416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
84717ab2063SBarry Smith   Scalar     *x, *y, *v, sum;
8489ea0dfa2SSatish Balay   int        m = a->m, n, i, *idx, shift = a->indexshift,*ii,jrow,j;
84917ab2063SBarry Smith 
850*3a40ed3dSBarry Smith   PetscFunctionBegin;
85190f02eecSBarry Smith   VecGetArray_Fast(xx,x); VecGetArray_Fast(yy,y);
85217ab2063SBarry Smith   x    = x + shift; /* shift for Fortran start by 1 indexing */
853416022c9SBarry Smith   idx  = a->j;
8549b88f4a7SBarry Smith   v    = a->a + shift; /* shift for Fortran start by 1 indexing */
855416022c9SBarry Smith   ii   = a->i;
8568d195f9aSBarry Smith #if defined(USE_FORTRAN_KERNELS)
8578d195f9aSBarry Smith   fortranmultaij_(&m,x,ii,idx,v,y);
8588d195f9aSBarry Smith #else
85917ab2063SBarry Smith   for ( i=0; i<m; i++ ) {
8609ea0dfa2SSatish Balay     jrow = ii[i];
8619ea0dfa2SSatish Balay     n    = ii[i+1] - jrow;
86217ab2063SBarry Smith     sum  = 0.0;
8639ea0dfa2SSatish Balay     for ( j=0; j<n; j++) {
8649ea0dfa2SSatish Balay       sum += v[jrow]*x[idx[jrow]]; jrow++;
8659ea0dfa2SSatish Balay      }
86617ab2063SBarry Smith     y[i] = sum;
86717ab2063SBarry Smith   }
8688d195f9aSBarry Smith #endif
869416022c9SBarry Smith   PLogFlops(2*a->nz - m);
870*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
87117ab2063SBarry Smith }
87217ab2063SBarry Smith 
8735615d1e5SSatish Balay #undef __FUNC__
8745615d1e5SSatish Balay #define __FUNC__ "MatMultAdd_SeqAIJ"
87544cd7ae7SLois Curfman McInnes int MatMultAdd_SeqAIJ(Mat A,Vec xx,Vec yy,Vec zz)
87617ab2063SBarry Smith {
877416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
87817ab2063SBarry Smith   Scalar     *x, *y, *z, *v, sum;
8799ea0dfa2SSatish Balay   int        m = a->m, n, i, *idx, shift = a->indexshift,*ii,jrow,j;
8809ea0dfa2SSatish Balay 
88117ab2063SBarry Smith 
882*3a40ed3dSBarry Smith   PetscFunctionBegin;
88390f02eecSBarry Smith   VecGetArray_Fast(xx,x); VecGetArray_Fast(yy,y); VecGetArray_Fast(zz,z);
88417ab2063SBarry Smith   x    = x + shift; /* shift for Fortran start by 1 indexing */
885cddf8d76SBarry Smith   idx  = a->j;
8869b88f4a7SBarry Smith   v    = a->a + shift; /* shift for Fortran start by 1 indexing */
887cddf8d76SBarry Smith   ii   = a->i;
88802ab625aSSatish Balay #if defined(USE_FORTRAN_KERNELS)
88902ab625aSSatish Balay   fortranmultaddaij_(&m,x,ii,idx,v,y,z);
89002ab625aSSatish Balay #else
89117ab2063SBarry Smith   for ( i=0; i<m; i++ ) {
8929ea0dfa2SSatish Balay     jrow = ii[i];
8939ea0dfa2SSatish Balay     n    = ii[i+1] - jrow;
89417ab2063SBarry Smith     sum  = y[i];
8959ea0dfa2SSatish Balay     for ( j=0; j<n; j++) {
8969ea0dfa2SSatish Balay       sum += v[jrow]*x[idx[jrow]]; jrow++;
8979ea0dfa2SSatish Balay      }
89817ab2063SBarry Smith     z[i] = sum;
89917ab2063SBarry Smith   }
90002ab625aSSatish Balay #endif
901416022c9SBarry Smith   PLogFlops(2*a->nz);
902*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
90317ab2063SBarry Smith }
90417ab2063SBarry Smith 
90517ab2063SBarry Smith /*
90617ab2063SBarry Smith      Adds diagonal pointers to sparse matrix structure.
90717ab2063SBarry Smith */
90817ab2063SBarry Smith 
9095615d1e5SSatish Balay #undef __FUNC__
9105615d1e5SSatish Balay #define __FUNC__ "MatMarkDiag_SeqAIJ"
911416022c9SBarry Smith int MatMarkDiag_SeqAIJ(Mat A)
91217ab2063SBarry Smith {
913416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
914416022c9SBarry Smith   int        i,j, *diag, m = a->m,shift = a->indexshift;
91517ab2063SBarry Smith 
916*3a40ed3dSBarry Smith   PetscFunctionBegin;
9170452661fSBarry Smith   diag = (int *) PetscMalloc( (m+1)*sizeof(int)); CHKPTRQ(diag);
918416022c9SBarry Smith   PLogObjectMemory(A,(m+1)*sizeof(int));
919416022c9SBarry Smith   for ( i=0; i<a->m; i++ ) {
920416022c9SBarry Smith     for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
921416022c9SBarry Smith       if (a->j[j]+shift == i) {
92217ab2063SBarry Smith         diag[i] = j - shift;
92317ab2063SBarry Smith         break;
92417ab2063SBarry Smith       }
92517ab2063SBarry Smith     }
92617ab2063SBarry Smith   }
927416022c9SBarry Smith   a->diag = diag;
928*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
92917ab2063SBarry Smith }
93017ab2063SBarry Smith 
9315615d1e5SSatish Balay #undef __FUNC__
9325615d1e5SSatish Balay #define __FUNC__ "MatRelax_SeqAIJ"
93344cd7ae7SLois Curfman McInnes int MatRelax_SeqAIJ(Mat A,Vec bb,double omega,MatSORType flag,
93417ab2063SBarry Smith                            double fshift,int its,Vec xx)
93517ab2063SBarry Smith {
936416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
937416022c9SBarry Smith   Scalar     *x, *b, *bs,  d, *xs, sum, *v = a->a,*t,scale,*ts, *xb;
938d5d45c9bSBarry Smith   int        ierr, *idx, *diag,n = a->n, m = a->m, i, shift = a->indexshift;
93917ab2063SBarry Smith 
940*3a40ed3dSBarry Smith   PetscFunctionBegin;
94190f02eecSBarry Smith   VecGetArray_Fast(xx,x); VecGetArray_Fast(bb,b);
942d00d2cf4SBarry Smith   if (!a->diag) {ierr = MatMarkDiag_SeqAIJ(A);CHKERRQ(ierr);}
943416022c9SBarry Smith   diag = a->diag;
944416022c9SBarry Smith   xs   = x + shift; /* shifted by one for index start of a or a->j*/
94517ab2063SBarry Smith   if (flag == SOR_APPLY_UPPER) {
94617ab2063SBarry Smith    /* apply ( U + D/omega) to the vector */
94717ab2063SBarry Smith     bs = b + shift;
94817ab2063SBarry Smith     for ( i=0; i<m; i++ ) {
949416022c9SBarry Smith         d    = fshift + a->a[diag[i] + shift];
950416022c9SBarry Smith         n    = a->i[i+1] - diag[i] - 1;
951416022c9SBarry Smith         idx  = a->j + diag[i] + (!shift);
952416022c9SBarry Smith         v    = a->a + diag[i] + (!shift);
95317ab2063SBarry Smith         sum  = b[i]*d/omega;
95417ab2063SBarry Smith         SPARSEDENSEDOT(sum,bs,v,idx,n);
95517ab2063SBarry Smith         x[i] = sum;
95617ab2063SBarry Smith     }
957*3a40ed3dSBarry Smith     PetscFunctionReturn(0);
95817ab2063SBarry Smith   }
95917ab2063SBarry Smith   if (flag == SOR_APPLY_LOWER) {
960e3372554SBarry Smith     SETERRQ(1,0,"SOR_APPLY_LOWER is not done");
961*3a40ed3dSBarry Smith   } else if (flag & SOR_EISENSTAT) {
96217ab2063SBarry Smith     /* Let  A = L + U + D; where L is lower trianglar,
96317ab2063SBarry Smith     U is upper triangular, E is diagonal; This routine applies
96417ab2063SBarry Smith 
96517ab2063SBarry Smith             (L + E)^{-1} A (U + E)^{-1}
96617ab2063SBarry Smith 
96717ab2063SBarry Smith     to a vector efficiently using Eisenstat's trick. This is for
96817ab2063SBarry Smith     the case of SSOR preconditioner, so E is D/omega where omega
96917ab2063SBarry Smith     is the relaxation factor.
97017ab2063SBarry Smith     */
9710452661fSBarry Smith     t = (Scalar *) PetscMalloc( m*sizeof(Scalar) ); CHKPTRQ(t);
97217ab2063SBarry Smith     scale = (2.0/omega) - 1.0;
97317ab2063SBarry Smith 
97417ab2063SBarry Smith     /*  x = (E + U)^{-1} b */
97517ab2063SBarry Smith     for ( i=m-1; i>=0; i-- ) {
976416022c9SBarry Smith       d    = fshift + a->a[diag[i] + shift];
977416022c9SBarry Smith       n    = a->i[i+1] - diag[i] - 1;
978416022c9SBarry Smith       idx  = a->j + diag[i] + (!shift);
979416022c9SBarry Smith       v    = a->a + diag[i] + (!shift);
98017ab2063SBarry Smith       sum  = b[i];
98117ab2063SBarry Smith       SPARSEDENSEMDOT(sum,xs,v,idx,n);
98217ab2063SBarry Smith       x[i] = omega*(sum/d);
98317ab2063SBarry Smith     }
98417ab2063SBarry Smith 
98517ab2063SBarry Smith     /*  t = b - (2*E - D)x */
986416022c9SBarry Smith     v = a->a;
98717ab2063SBarry Smith     for ( i=0; i<m; i++ ) { t[i] = b[i] - scale*(v[*diag++ + shift])*x[i]; }
98817ab2063SBarry Smith 
98917ab2063SBarry Smith     /*  t = (E + L)^{-1}t */
990416022c9SBarry Smith     ts = t + shift; /* shifted by one for index start of a or a->j*/
991416022c9SBarry Smith     diag = a->diag;
99217ab2063SBarry Smith     for ( i=0; i<m; i++ ) {
993416022c9SBarry Smith       d    = fshift + a->a[diag[i]+shift];
994416022c9SBarry Smith       n    = diag[i] - a->i[i];
995416022c9SBarry Smith       idx  = a->j + a->i[i] + shift;
996416022c9SBarry Smith       v    = a->a + a->i[i] + shift;
99717ab2063SBarry Smith       sum  = t[i];
99817ab2063SBarry Smith       SPARSEDENSEMDOT(sum,ts,v,idx,n);
99917ab2063SBarry Smith       t[i] = omega*(sum/d);
100017ab2063SBarry Smith     }
100117ab2063SBarry Smith 
100217ab2063SBarry Smith     /*  x = x + t */
100317ab2063SBarry Smith     for ( i=0; i<m; i++ ) { x[i] += t[i]; }
10040452661fSBarry Smith     PetscFree(t);
1005*3a40ed3dSBarry Smith     PetscFunctionReturn(0);
100617ab2063SBarry Smith   }
100717ab2063SBarry Smith   if (flag & SOR_ZERO_INITIAL_GUESS) {
100817ab2063SBarry Smith     if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP){
100917ab2063SBarry Smith       for ( i=0; i<m; i++ ) {
1010416022c9SBarry Smith         d    = fshift + a->a[diag[i]+shift];
1011416022c9SBarry Smith         n    = diag[i] - a->i[i];
1012416022c9SBarry Smith         idx  = a->j + a->i[i] + shift;
1013416022c9SBarry Smith         v    = a->a + a->i[i] + shift;
101417ab2063SBarry Smith         sum  = b[i];
101517ab2063SBarry Smith         SPARSEDENSEMDOT(sum,xs,v,idx,n);
101617ab2063SBarry Smith         x[i] = omega*(sum/d);
101717ab2063SBarry Smith       }
101817ab2063SBarry Smith       xb = x;
1019*3a40ed3dSBarry Smith     } else xb = b;
102017ab2063SBarry Smith     if ((flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP) &&
102117ab2063SBarry Smith         (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP)) {
102217ab2063SBarry Smith       for ( i=0; i<m; i++ ) {
1023416022c9SBarry Smith         x[i] *= a->a[diag[i]+shift];
102417ab2063SBarry Smith       }
102517ab2063SBarry Smith     }
102617ab2063SBarry Smith     if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP){
102717ab2063SBarry Smith       for ( i=m-1; i>=0; i-- ) {
1028416022c9SBarry Smith         d    = fshift + a->a[diag[i] + shift];
1029416022c9SBarry Smith         n    = a->i[i+1] - diag[i] - 1;
1030416022c9SBarry Smith         idx  = a->j + diag[i] + (!shift);
1031416022c9SBarry Smith         v    = a->a + diag[i] + (!shift);
103217ab2063SBarry Smith         sum  = xb[i];
103317ab2063SBarry Smith         SPARSEDENSEMDOT(sum,xs,v,idx,n);
103417ab2063SBarry Smith         x[i] = omega*(sum/d);
103517ab2063SBarry Smith       }
103617ab2063SBarry Smith     }
103717ab2063SBarry Smith     its--;
103817ab2063SBarry Smith   }
103917ab2063SBarry Smith   while (its--) {
104017ab2063SBarry Smith     if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP){
104117ab2063SBarry Smith       for ( i=0; i<m; i++ ) {
1042416022c9SBarry Smith         d    = fshift + a->a[diag[i]+shift];
1043416022c9SBarry Smith         n    = a->i[i+1] - a->i[i];
1044416022c9SBarry Smith         idx  = a->j + a->i[i] + shift;
1045416022c9SBarry Smith         v    = a->a + a->i[i] + shift;
104617ab2063SBarry Smith         sum  = b[i];
104717ab2063SBarry Smith         SPARSEDENSEMDOT(sum,xs,v,idx,n);
10487e33a6baSBarry Smith         x[i] = (1. - omega)*x[i] + omega*(sum + a->a[diag[i]+shift]*x[i])/d;
104917ab2063SBarry Smith       }
105017ab2063SBarry Smith     }
105117ab2063SBarry Smith     if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP){
105217ab2063SBarry Smith       for ( i=m-1; i>=0; i-- ) {
1053416022c9SBarry Smith         d    = fshift + a->a[diag[i] + shift];
1054416022c9SBarry Smith         n    = a->i[i+1] - a->i[i];
1055416022c9SBarry Smith         idx  = a->j + a->i[i] + shift;
1056416022c9SBarry Smith         v    = a->a + a->i[i] + shift;
105717ab2063SBarry Smith         sum  = b[i];
105817ab2063SBarry Smith         SPARSEDENSEMDOT(sum,xs,v,idx,n);
10597e33a6baSBarry Smith         x[i] = (1. - omega)*x[i] + omega*(sum + a->a[diag[i]+shift]*x[i])/d;
106017ab2063SBarry Smith       }
106117ab2063SBarry Smith     }
106217ab2063SBarry Smith   }
1063*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
106417ab2063SBarry Smith }
106517ab2063SBarry Smith 
10665615d1e5SSatish Balay #undef __FUNC__
10675615d1e5SSatish Balay #define __FUNC__ "MatGetInfo_SeqAIJ"
10688f6be9afSLois Curfman McInnes int MatGetInfo_SeqAIJ(Mat A,MatInfoType flag,MatInfo *info)
106917ab2063SBarry Smith {
1070416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
10714e220ebcSLois Curfman McInnes 
1072*3a40ed3dSBarry Smith   PetscFunctionBegin;
10734e220ebcSLois Curfman McInnes   info->rows_global    = (double)a->m;
10744e220ebcSLois Curfman McInnes   info->columns_global = (double)a->n;
10754e220ebcSLois Curfman McInnes   info->rows_local     = (double)a->m;
10764e220ebcSLois Curfman McInnes   info->columns_local  = (double)a->n;
10774e220ebcSLois Curfman McInnes   info->block_size     = 1.0;
10784e220ebcSLois Curfman McInnes   info->nz_allocated   = (double)a->maxnz;
10794e220ebcSLois Curfman McInnes   info->nz_used        = (double)a->nz;
10804e220ebcSLois Curfman McInnes   info->nz_unneeded    = (double)(a->maxnz - a->nz);
10814e220ebcSLois Curfman McInnes   /*  if (info->nz_unneeded != A->info.nz_unneeded)
10824e220ebcSLois Curfman McInnes     printf("space descrepancy: maxnz-nz = %d, nz_unneeded = %d\n",(int)info->nz_unneeded,(int)A->info.nz_unneeded); */
10834e220ebcSLois Curfman McInnes   info->assemblies     = (double)A->num_ass;
10844e220ebcSLois Curfman McInnes   info->mallocs        = (double)a->reallocs;
10854e220ebcSLois Curfman McInnes   info->memory         = A->mem;
10864e220ebcSLois Curfman McInnes   if (A->factor) {
10874e220ebcSLois Curfman McInnes     info->fill_ratio_given  = A->info.fill_ratio_given;
10884e220ebcSLois Curfman McInnes     info->fill_ratio_needed = A->info.fill_ratio_needed;
10894e220ebcSLois Curfman McInnes     info->factor_mallocs    = A->info.factor_mallocs;
10904e220ebcSLois Curfman McInnes   } else {
10914e220ebcSLois Curfman McInnes     info->fill_ratio_given  = 0;
10924e220ebcSLois Curfman McInnes     info->fill_ratio_needed = 0;
10934e220ebcSLois Curfman McInnes     info->factor_mallocs    = 0;
10944e220ebcSLois Curfman McInnes   }
1095*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
109617ab2063SBarry Smith }
109717ab2063SBarry Smith 
109817ab2063SBarry Smith extern int MatLUFactorSymbolic_SeqAIJ(Mat,IS,IS,double,Mat*);
109917ab2063SBarry Smith extern int MatLUFactorNumeric_SeqAIJ(Mat,Mat*);
110017ab2063SBarry Smith extern int MatLUFactor_SeqAIJ(Mat,IS,IS,double);
110117ab2063SBarry Smith extern int MatSolve_SeqAIJ(Mat,Vec,Vec);
110217ab2063SBarry Smith extern int MatSolveAdd_SeqAIJ(Mat,Vec,Vec,Vec);
110317ab2063SBarry Smith extern int MatSolveTrans_SeqAIJ(Mat,Vec,Vec);
110417ab2063SBarry Smith extern int MatSolveTransAdd_SeqAIJ(Mat,Vec,Vec,Vec);
110517ab2063SBarry Smith 
11065615d1e5SSatish Balay #undef __FUNC__
11075615d1e5SSatish Balay #define __FUNC__ "MatZeroRows_SeqAIJ"
11088f6be9afSLois Curfman McInnes int MatZeroRows_SeqAIJ(Mat A,IS is,Scalar *diag)
110917ab2063SBarry Smith {
1110416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
1111416022c9SBarry Smith   int         i,ierr,N, *rows,m = a->m - 1,shift = a->indexshift;
111217ab2063SBarry Smith 
1113*3a40ed3dSBarry Smith   PetscFunctionBegin;
111477c4ece6SBarry Smith   ierr = ISGetSize(is,&N); CHKERRQ(ierr);
111517ab2063SBarry Smith   ierr = ISGetIndices(is,&rows); CHKERRQ(ierr);
111617ab2063SBarry Smith   if (diag) {
111717ab2063SBarry Smith     for ( i=0; i<N; i++ ) {
1118e3372554SBarry Smith       if (rows[i] < 0 || rows[i] > m) SETERRQ(1,0,"row out of range");
1119416022c9SBarry Smith       if (a->ilen[rows[i]] > 0) { /* in case row was completely empty */
1120416022c9SBarry Smith         a->ilen[rows[i]] = 1;
1121416022c9SBarry Smith         a->a[a->i[rows[i]]+shift] = *diag;
1122416022c9SBarry Smith         a->j[a->i[rows[i]]+shift] = rows[i]+shift;
1123*3a40ed3dSBarry Smith       } else {
112417ab2063SBarry Smith         ierr = MatSetValues_SeqAIJ(A,1,&rows[i],1,&rows[i],diag,INSERT_VALUES);
112517ab2063SBarry Smith         CHKERRQ(ierr);
112617ab2063SBarry Smith       }
112717ab2063SBarry Smith     }
1128*3a40ed3dSBarry Smith   } else {
112917ab2063SBarry Smith     for ( i=0; i<N; i++ ) {
1130e3372554SBarry Smith       if (rows[i] < 0 || rows[i] > m) SETERRQ(1,0,"row out of range");
1131416022c9SBarry Smith       a->ilen[rows[i]] = 0;
113217ab2063SBarry Smith     }
113317ab2063SBarry Smith   }
113417ab2063SBarry Smith   ISRestoreIndices(is,&rows);
113543a90d84SBarry Smith   ierr = MatAssemblyEnd_SeqAIJ(A,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr);
1136*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
113717ab2063SBarry Smith }
113817ab2063SBarry Smith 
11395615d1e5SSatish Balay #undef __FUNC__
11405615d1e5SSatish Balay #define __FUNC__ "MatGetSize_SeqAIJ"
11418f6be9afSLois Curfman McInnes int MatGetSize_SeqAIJ(Mat A,int *m,int *n)
114217ab2063SBarry Smith {
1143416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
1144*3a40ed3dSBarry Smith 
1145*3a40ed3dSBarry Smith   PetscFunctionBegin;
11460752156aSBarry Smith   if (m) *m = a->m;
11470752156aSBarry Smith   if (n) *n = a->n;
1148*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
114917ab2063SBarry Smith }
115017ab2063SBarry Smith 
11515615d1e5SSatish Balay #undef __FUNC__
11525615d1e5SSatish Balay #define __FUNC__ "MatGetOwnershipRange_SeqAIJ"
11538f6be9afSLois Curfman McInnes int MatGetOwnershipRange_SeqAIJ(Mat A,int *m,int *n)
115417ab2063SBarry Smith {
1155416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
1156*3a40ed3dSBarry Smith 
1157*3a40ed3dSBarry Smith   PetscFunctionBegin;
1158416022c9SBarry Smith   *m = 0; *n = a->m;
1159*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
116017ab2063SBarry Smith }
1161026e39d0SSatish Balay 
11625615d1e5SSatish Balay #undef __FUNC__
11635615d1e5SSatish Balay #define __FUNC__ "MatGetRow_SeqAIJ"
11644e093b46SBarry Smith int MatGetRow_SeqAIJ(Mat A,int row,int *nz,int **idx,Scalar **v)
116517ab2063SBarry Smith {
1166416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
1167c456f294SBarry Smith   int        *itmp,i,shift = a->indexshift;
116817ab2063SBarry Smith 
1169*3a40ed3dSBarry Smith   PetscFunctionBegin;
1170e3372554SBarry Smith   if (row < 0 || row >= a->m) SETERRQ(1,0,"Row out of range");
117117ab2063SBarry Smith 
1172416022c9SBarry Smith   *nz = a->i[row+1] - a->i[row];
1173416022c9SBarry Smith   if (v) *v = a->a + a->i[row] + shift;
117417ab2063SBarry Smith   if (idx) {
1175416022c9SBarry Smith     itmp = a->j + a->i[row] + shift;
11764e093b46SBarry Smith     if (*nz && shift) {
11770452661fSBarry Smith       *idx = (int *) PetscMalloc( (*nz)*sizeof(int) ); CHKPTRQ(*idx);
117817ab2063SBarry Smith       for ( i=0; i<(*nz); i++ ) {(*idx)[i] = itmp[i] + shift;}
11794e093b46SBarry Smith     } else if (*nz) {
11804e093b46SBarry Smith       *idx = itmp;
118117ab2063SBarry Smith     }
118217ab2063SBarry Smith     else *idx = 0;
118317ab2063SBarry Smith   }
1184*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
118517ab2063SBarry Smith }
118617ab2063SBarry Smith 
11875615d1e5SSatish Balay #undef __FUNC__
11885615d1e5SSatish Balay #define __FUNC__ "MatRestoreRow_SeqAIJ"
11894e093b46SBarry Smith int MatRestoreRow_SeqAIJ(Mat A,int row,int *nz,int **idx,Scalar **v)
119017ab2063SBarry Smith {
11914e093b46SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
1192*3a40ed3dSBarry Smith 
1193*3a40ed3dSBarry Smith   PetscFunctionBegin;
11944e093b46SBarry Smith   if (idx) {if (*idx && a->indexshift) PetscFree(*idx);}
1195*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
119617ab2063SBarry Smith }
119717ab2063SBarry Smith 
11985615d1e5SSatish Balay #undef __FUNC__
11995615d1e5SSatish Balay #define __FUNC__ "MatNorm_SeqAIJ"
12008f6be9afSLois Curfman McInnes int MatNorm_SeqAIJ(Mat A,NormType type,double *norm)
120117ab2063SBarry Smith {
1202416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
1203416022c9SBarry Smith   Scalar     *v = a->a;
120417ab2063SBarry Smith   double     sum = 0.0;
1205416022c9SBarry Smith   int        i, j,shift = a->indexshift;
120617ab2063SBarry Smith 
1207*3a40ed3dSBarry Smith   PetscFunctionBegin;
120817ab2063SBarry Smith   if (type == NORM_FROBENIUS) {
1209416022c9SBarry Smith     for (i=0; i<a->nz; i++ ) {
1210*3a40ed3dSBarry Smith #if defined(USE_PETSC_COMPLEX)
121117ab2063SBarry Smith       sum += real(conj(*v)*(*v)); v++;
121217ab2063SBarry Smith #else
121317ab2063SBarry Smith       sum += (*v)*(*v); v++;
121417ab2063SBarry Smith #endif
121517ab2063SBarry Smith     }
121617ab2063SBarry Smith     *norm = sqrt(sum);
1217*3a40ed3dSBarry Smith   } else if (type == NORM_1) {
121817ab2063SBarry Smith     double *tmp;
1219416022c9SBarry Smith     int    *jj = a->j;
122066963ce1SSatish Balay     tmp = (double *) PetscMalloc( (a->n+1)*sizeof(double) ); CHKPTRQ(tmp);
1221cddf8d76SBarry Smith     PetscMemzero(tmp,a->n*sizeof(double));
122217ab2063SBarry Smith     *norm = 0.0;
1223416022c9SBarry Smith     for ( j=0; j<a->nz; j++ ) {
1224a2744918SBarry Smith         tmp[*jj++ + shift] += PetscAbsScalar(*v);  v++;
122517ab2063SBarry Smith     }
1226416022c9SBarry Smith     for ( j=0; j<a->n; j++ ) {
122717ab2063SBarry Smith       if (tmp[j] > *norm) *norm = tmp[j];
122817ab2063SBarry Smith     }
12290452661fSBarry Smith     PetscFree(tmp);
1230*3a40ed3dSBarry Smith   } else if (type == NORM_INFINITY) {
123117ab2063SBarry Smith     *norm = 0.0;
1232416022c9SBarry Smith     for ( j=0; j<a->m; j++ ) {
1233416022c9SBarry Smith       v = a->a + a->i[j] + shift;
123417ab2063SBarry Smith       sum = 0.0;
1235416022c9SBarry Smith       for ( i=0; i<a->i[j+1]-a->i[j]; i++ ) {
1236cddf8d76SBarry Smith         sum += PetscAbsScalar(*v); v++;
123717ab2063SBarry Smith       }
123817ab2063SBarry Smith       if (sum > *norm) *norm = sum;
123917ab2063SBarry Smith     }
1240*3a40ed3dSBarry Smith   } else {
1241e3372554SBarry Smith     SETERRQ(1,0,"No support for two norm yet");
124217ab2063SBarry Smith   }
1243*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
124417ab2063SBarry Smith }
124517ab2063SBarry Smith 
12465615d1e5SSatish Balay #undef __FUNC__
12475615d1e5SSatish Balay #define __FUNC__ "MatTranspose_SeqAIJ"
12488f6be9afSLois Curfman McInnes int MatTranspose_SeqAIJ(Mat A,Mat *B)
124917ab2063SBarry Smith {
1250416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
1251416022c9SBarry Smith   Mat        C;
1252416022c9SBarry Smith   int        i, ierr, *aj = a->j, *ai = a->i, m = a->m, len, *col;
1253416022c9SBarry Smith   int        shift = a->indexshift;
1254d5d45c9bSBarry Smith   Scalar     *array = a->a;
125517ab2063SBarry Smith 
1256*3a40ed3dSBarry Smith   PetscFunctionBegin;
1257*3a40ed3dSBarry Smith   if (B == PETSC_NULL && m != a->n) SETERRQ(1,0,"Square matrix only for in-place");
12580452661fSBarry Smith   col = (int *) PetscMalloc((1+a->n)*sizeof(int)); CHKPTRQ(col);
1259cddf8d76SBarry Smith   PetscMemzero(col,(1+a->n)*sizeof(int));
126017ab2063SBarry Smith   if (shift) {
126117ab2063SBarry Smith     for ( i=0; i<ai[m]-1; i++ ) aj[i] -= 1;
126217ab2063SBarry Smith   }
126317ab2063SBarry Smith   for ( i=0; i<ai[m]+shift; i++ ) col[aj[i]] += 1;
1264416022c9SBarry Smith   ierr = MatCreateSeqAIJ(A->comm,a->n,m,0,col,&C); CHKERRQ(ierr);
12650452661fSBarry Smith   PetscFree(col);
126617ab2063SBarry Smith   for ( i=0; i<m; i++ ) {
126717ab2063SBarry Smith     len = ai[i+1]-ai[i];
1268416022c9SBarry Smith     ierr = MatSetValues(C,len,aj,1,&i,array,INSERT_VALUES); CHKERRQ(ierr);
126917ab2063SBarry Smith     array += len; aj += len;
127017ab2063SBarry Smith   }
127117ab2063SBarry Smith   if (shift) {
127217ab2063SBarry Smith     for ( i=0; i<ai[m]-1; i++ ) aj[i] += 1;
127317ab2063SBarry Smith   }
127417ab2063SBarry Smith 
12756d4a8577SBarry Smith   ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr);
12766d4a8577SBarry Smith   ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr);
127717ab2063SBarry Smith 
12783638b69dSLois Curfman McInnes   if (B != PETSC_NULL) {
1279416022c9SBarry Smith     *B = C;
128017ab2063SBarry Smith   } else {
1281416022c9SBarry Smith     /* This isn't really an in-place transpose */
12820452661fSBarry Smith     PetscFree(a->a);
12830452661fSBarry Smith     if (!a->singlemalloc) {PetscFree(a->i); PetscFree(a->j);}
12840452661fSBarry Smith     if (a->diag) PetscFree(a->diag);
12850452661fSBarry Smith     if (a->ilen) PetscFree(a->ilen);
12860452661fSBarry Smith     if (a->imax) PetscFree(a->imax);
12870452661fSBarry Smith     if (a->solve_work) PetscFree(a->solve_work);
12881073c447SSatish Balay     if (a->inode.size) PetscFree(a->inode.size);
12890452661fSBarry Smith     PetscFree(a);
1290f09e8eb9SSatish Balay     PetscMemcpy(A,C,sizeof(struct _p_Mat));
12910452661fSBarry Smith     PetscHeaderDestroy(C);
129217ab2063SBarry Smith   }
1293*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
129417ab2063SBarry Smith }
129517ab2063SBarry Smith 
12965615d1e5SSatish Balay #undef __FUNC__
12975615d1e5SSatish Balay #define __FUNC__ "MatDiagonalScale_SeqAIJ"
12988f6be9afSLois Curfman McInnes int MatDiagonalScale_SeqAIJ(Mat A,Vec ll,Vec rr)
129917ab2063SBarry Smith {
1300416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
130117ab2063SBarry Smith   Scalar     *l,*r,x,*v;
1302d5d45c9bSBarry Smith   int        i,j,m = a->m, n = a->n, M, nz = a->nz, *jj,shift = a->indexshift;
130317ab2063SBarry Smith 
1304*3a40ed3dSBarry Smith   PetscFunctionBegin;
130517ab2063SBarry Smith   if (ll) {
13063ea7c6a1SSatish Balay     /* The local size is used so that VecMPI can be passed to this routine
13073ea7c6a1SSatish Balay        by MatDiagonalScale_MPIAIJ */
13089b1297e1SSatish Balay     VecGetLocalSize_Fast(ll,m);
1309e3372554SBarry Smith     if (m != a->m) SETERRQ(1,0,"Left scaling vector wrong length");
131090f02eecSBarry Smith     VecGetArray_Fast(ll,l);
1311416022c9SBarry Smith     v = a->a;
131217ab2063SBarry Smith     for ( i=0; i<m; i++ ) {
131317ab2063SBarry Smith       x = l[i];
1314416022c9SBarry Smith       M = a->i[i+1] - a->i[i];
131517ab2063SBarry Smith       for ( j=0; j<M; j++ ) { (*v++) *= x;}
131617ab2063SBarry Smith     }
131744cd7ae7SLois Curfman McInnes     PLogFlops(nz);
131817ab2063SBarry Smith   }
131917ab2063SBarry Smith   if (rr) {
13209b1297e1SSatish Balay     VecGetLocalSize_Fast(rr,n);
1321e3372554SBarry Smith     if (n != a->n) SETERRQ(1,0,"Right scaling vector wrong length");
132290f02eecSBarry Smith     VecGetArray_Fast(rr,r);
1323416022c9SBarry Smith     v = a->a; jj = a->j;
132417ab2063SBarry Smith     for ( i=0; i<nz; i++ ) {
132517ab2063SBarry Smith       (*v++) *= r[*jj++ + shift];
132617ab2063SBarry Smith     }
132744cd7ae7SLois Curfman McInnes     PLogFlops(nz);
132817ab2063SBarry Smith   }
1329*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
133017ab2063SBarry Smith }
133117ab2063SBarry Smith 
13325615d1e5SSatish Balay #undef __FUNC__
13335615d1e5SSatish Balay #define __FUNC__ "MatGetSubMatrix_SeqAIJ"
13348f6be9afSLois Curfman McInnes int MatGetSubMatrix_SeqAIJ(Mat A,IS isrow,IS iscol,MatGetSubMatrixCall scall,Mat *B)
133517ab2063SBarry Smith {
1336db02288aSLois Curfman McInnes   Mat_SeqAIJ   *a = (Mat_SeqAIJ *) A->data,*c;
133702834360SBarry Smith   int          nznew, *smap, i, k, kstart, kend, ierr, oldcols = a->n,*lens;
133899141d43SSatish Balay   int          row,mat_i,*mat_j,tcol,first,step,*mat_ilen;
1339a2744918SBarry Smith   register int sum,lensi;
134099141d43SSatish Balay   int          *irow, *icol, nrows, ncols, shift = a->indexshift,*ssmap;
134199141d43SSatish Balay   int          *starts,*j_new,*i_new,*aj = a->j, *ai = a->i,ii,*ailen = a->ilen;
134299141d43SSatish Balay   Scalar       *a_new,*mat_a;
1343416022c9SBarry Smith   Mat          C;
134417ab2063SBarry Smith 
1345*3a40ed3dSBarry Smith   PetscFunctionBegin;
1346b48a1e75SSatish Balay   ierr = ISSorted(isrow,(PetscTruth*)&i);
1347e3372554SBarry Smith   if (!i) SETERRQ(1,0,"ISrow is not sorted");
134899141d43SSatish Balay   ierr = ISSorted(iscol,(PetscTruth*)&i);
1349e3372554SBarry Smith   if (!i) SETERRQ(1,0,"IScol is not sorted");
135099141d43SSatish Balay 
135117ab2063SBarry Smith   ierr = ISGetIndices(isrow,&irow); CHKERRQ(ierr);
135217ab2063SBarry Smith   ierr = ISGetSize(isrow,&nrows); CHKERRQ(ierr);
135317ab2063SBarry Smith   ierr = ISGetSize(iscol,&ncols); CHKERRQ(ierr);
135417ab2063SBarry Smith 
13557264ac53SSatish Balay   if (ISStrideGetInfo(iscol,&first,&step) && step == 1) { /* no need to sort */
135602834360SBarry Smith     /* special case of contiguous rows */
135757faeb66SBarry Smith     lens   = (int *) PetscMalloc((ncols+nrows+1)*sizeof(int)); CHKPTRQ(lens);
135802834360SBarry Smith     starts = lens + ncols;
135902834360SBarry Smith     /* loop over new rows determining lens and starting points */
136002834360SBarry Smith     for (i=0; i<nrows; i++) {
1361a2744918SBarry Smith       kstart  = ai[irow[i]]+shift;
1362a2744918SBarry Smith       kend    = kstart + ailen[irow[i]];
136302834360SBarry Smith       for ( k=kstart; k<kend; k++ ) {
1364d8ced48eSBarry Smith         if (aj[k]+shift >= first) {
136502834360SBarry Smith           starts[i] = k;
136602834360SBarry Smith           break;
136702834360SBarry Smith 	}
136802834360SBarry Smith       }
1369a2744918SBarry Smith       sum = 0;
137002834360SBarry Smith       while (k < kend) {
1371d8ced48eSBarry Smith         if (aj[k++]+shift >= first+ncols) break;
1372a2744918SBarry Smith         sum++;
137302834360SBarry Smith       }
1374a2744918SBarry Smith       lens[i] = sum;
137502834360SBarry Smith     }
137602834360SBarry Smith     /* create submatrix */
1377cddf8d76SBarry Smith     if (scall == MAT_REUSE_MATRIX) {
137808480c60SBarry Smith       int n_cols,n_rows;
137908480c60SBarry Smith       ierr = MatGetSize(*B,&n_rows,&n_cols); CHKERRQ(ierr);
1380e3372554SBarry Smith       if (n_rows != nrows || n_cols != ncols) SETERRQ(1,0,"");
1381d8ced48eSBarry Smith       ierr = MatZeroEntries(*B); CHKERRQ(ierr);
138208480c60SBarry Smith       C = *B;
1383*3a40ed3dSBarry Smith     } else {
138402834360SBarry Smith       ierr = MatCreateSeqAIJ(A->comm,nrows,ncols,0,lens,&C);CHKERRQ(ierr);
138508480c60SBarry Smith     }
1386db02288aSLois Curfman McInnes     c = (Mat_SeqAIJ*) C->data;
1387db02288aSLois Curfman McInnes 
138802834360SBarry Smith     /* loop over rows inserting into submatrix */
1389db02288aSLois Curfman McInnes     a_new    = c->a;
1390db02288aSLois Curfman McInnes     j_new    = c->j;
1391db02288aSLois Curfman McInnes     i_new    = c->i;
1392db02288aSLois Curfman McInnes     i_new[0] = -shift;
139302834360SBarry Smith     for (i=0; i<nrows; i++) {
1394a2744918SBarry Smith       ii    = starts[i];
1395a2744918SBarry Smith       lensi = lens[i];
1396a2744918SBarry Smith       for ( k=0; k<lensi; k++ ) {
1397a2744918SBarry Smith         *j_new++ = aj[ii+k] - first;
139802834360SBarry Smith       }
1399a2744918SBarry Smith       PetscMemcpy(a_new,a->a + starts[i],lensi*sizeof(Scalar));
1400a2744918SBarry Smith       a_new      += lensi;
1401a2744918SBarry Smith       i_new[i+1]  = i_new[i] + lensi;
1402a2744918SBarry Smith       c->ilen[i]  = lensi;
140302834360SBarry Smith     }
14040452661fSBarry Smith     PetscFree(lens);
1405*3a40ed3dSBarry Smith   } else {
140602834360SBarry Smith     ierr = ISGetIndices(iscol,&icol); CHKERRQ(ierr);
14070452661fSBarry Smith     smap  = (int *) PetscMalloc((1+oldcols)*sizeof(int)); CHKPTRQ(smap);
140802834360SBarry Smith     ssmap = smap + shift;
140999141d43SSatish Balay     lens  = (int *) PetscMalloc((1+nrows)*sizeof(int)); CHKPTRQ(lens);
1410cddf8d76SBarry Smith     PetscMemzero(smap,oldcols*sizeof(int));
141117ab2063SBarry Smith     for ( i=0; i<ncols; i++ ) smap[icol[i]] = i+1;
141202834360SBarry Smith     /* determine lens of each row */
141302834360SBarry Smith     for (i=0; i<nrows; i++) {
1414d8ced48eSBarry Smith       kstart  = ai[irow[i]]+shift;
141502834360SBarry Smith       kend    = kstart + a->ilen[irow[i]];
141602834360SBarry Smith       lens[i] = 0;
141702834360SBarry Smith       for ( k=kstart; k<kend; k++ ) {
1418d8ced48eSBarry Smith         if (ssmap[aj[k]]) {
141902834360SBarry Smith           lens[i]++;
142002834360SBarry Smith         }
142102834360SBarry Smith       }
142202834360SBarry Smith     }
142317ab2063SBarry Smith     /* Create and fill new matrix */
1424a2744918SBarry Smith     if (scall == MAT_REUSE_MATRIX) {
142599141d43SSatish Balay       c = (Mat_SeqAIJ *)((*B)->data);
142699141d43SSatish Balay 
1427e3372554SBarry Smith       if (c->m  != nrows || c->n != ncols) SETERRQ(1,0,"");
142899141d43SSatish Balay       if (PetscMemcmp(c->ilen,lens, c->m *sizeof(int))) {
1429e3372554SBarry Smith         SETERRQ(1,0,"Cannot reuse matrix. wrong no of nonzeros");
143099141d43SSatish Balay       }
143199141d43SSatish Balay       PetscMemzero(c->ilen,c->m*sizeof(int));
143208480c60SBarry Smith       C = *B;
1433*3a40ed3dSBarry Smith     } else {
143402834360SBarry Smith       ierr = MatCreateSeqAIJ(A->comm,nrows,ncols,0,lens,&C);CHKERRQ(ierr);
143508480c60SBarry Smith     }
143699141d43SSatish Balay     c = (Mat_SeqAIJ *)(C->data);
143717ab2063SBarry Smith     for (i=0; i<nrows; i++) {
143899141d43SSatish Balay       row    = irow[i];
143917ab2063SBarry Smith       nznew  = 0;
144099141d43SSatish Balay       kstart = ai[row]+shift;
144199141d43SSatish Balay       kend   = kstart + a->ilen[row];
144299141d43SSatish Balay       mat_i  = c->i[i]+shift;
144399141d43SSatish Balay       mat_j  = c->j + mat_i;
144499141d43SSatish Balay       mat_a  = c->a + mat_i;
144599141d43SSatish Balay       mat_ilen = c->ilen + i;
144617ab2063SBarry Smith       for ( k=kstart; k<kend; k++ ) {
144799141d43SSatish Balay         if ((tcol=ssmap[a->j[k]])) {
144899141d43SSatish Balay           *mat_j++ = tcol - (!shift);
144999141d43SSatish Balay           *mat_a++ = a->a[k];
145099141d43SSatish Balay           (*mat_ilen)++;
145199141d43SSatish Balay 
145217ab2063SBarry Smith         }
145317ab2063SBarry Smith       }
145417ab2063SBarry Smith     }
145502834360SBarry Smith     /* Free work space */
145602834360SBarry Smith     ierr = ISRestoreIndices(iscol,&icol); CHKERRQ(ierr);
145799141d43SSatish Balay     PetscFree(smap); PetscFree(lens);
145802834360SBarry Smith   }
14596d4a8577SBarry Smith   ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr);
14606d4a8577SBarry Smith   ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr);
146117ab2063SBarry Smith 
146217ab2063SBarry Smith   ierr = ISRestoreIndices(isrow,&irow); CHKERRQ(ierr);
1463416022c9SBarry Smith   *B = C;
1464*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
146517ab2063SBarry Smith }
146617ab2063SBarry Smith 
1467a871dcd8SBarry Smith /*
146863b91edcSBarry Smith      note: This can only work for identity for row and col. It would
146963b91edcSBarry Smith    be good to check this and otherwise generate an error.
1470a871dcd8SBarry Smith */
14715615d1e5SSatish Balay #undef __FUNC__
14725615d1e5SSatish Balay #define __FUNC__ "MatILUFactor_SeqAIJ"
14738f6be9afSLois Curfman McInnes int MatILUFactor_SeqAIJ(Mat inA,IS row,IS col,double efill,int fill)
1474a871dcd8SBarry Smith {
147563b91edcSBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) inA->data;
147608480c60SBarry Smith   int        ierr;
147763b91edcSBarry Smith   Mat        outA;
147863b91edcSBarry Smith 
1479*3a40ed3dSBarry Smith   PetscFunctionBegin;
1480e3372554SBarry Smith   if (fill != 0) SETERRQ(1,0,"Only fill=0 supported");
1481a871dcd8SBarry Smith 
148263b91edcSBarry Smith   outA          = inA;
148363b91edcSBarry Smith   inA->factor   = FACTOR_LU;
148463b91edcSBarry Smith   a->row        = row;
148563b91edcSBarry Smith   a->col        = col;
148663b91edcSBarry Smith 
148794a9d846SBarry Smith   if (!a->solve_work) { /* this matrix may have been factored before */
14880452661fSBarry Smith     a->solve_work = (Scalar *) PetscMalloc( (a->m+1)*sizeof(Scalar)); CHKPTRQ(a->solve_work);
148994a9d846SBarry Smith   }
149063b91edcSBarry Smith 
149108480c60SBarry Smith   if (!a->diag) {
149208480c60SBarry Smith     ierr = MatMarkDiag_SeqAIJ(inA); CHKERRQ(ierr);
149363b91edcSBarry Smith   }
149463b91edcSBarry Smith   ierr = MatLUFactorNumeric_SeqAIJ(inA,&outA); CHKERRQ(ierr);
1495*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
1496a871dcd8SBarry Smith }
1497a871dcd8SBarry Smith 
1498f0b747eeSBarry Smith #include "pinclude/plapack.h"
14995615d1e5SSatish Balay #undef __FUNC__
15005615d1e5SSatish Balay #define __FUNC__ "MatScale_SeqAIJ"
15018f6be9afSLois Curfman McInnes int MatScale_SeqAIJ(Scalar *alpha,Mat inA)
1502f0b747eeSBarry Smith {
1503f0b747eeSBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) inA->data;
1504f0b747eeSBarry Smith   int        one = 1;
1505*3a40ed3dSBarry Smith 
1506*3a40ed3dSBarry Smith   PetscFunctionBegin;
1507f0b747eeSBarry Smith   BLscal_( &a->nz, alpha, a->a, &one );
1508f0b747eeSBarry Smith   PLogFlops(a->nz);
1509*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
1510f0b747eeSBarry Smith }
1511f0b747eeSBarry Smith 
15125615d1e5SSatish Balay #undef __FUNC__
15135615d1e5SSatish Balay #define __FUNC__ "MatGetSubMatrices_SeqAIJ"
15148f6be9afSLois Curfman McInnes int MatGetSubMatrices_SeqAIJ(Mat A,int n, IS *irow,IS *icol,MatGetSubMatrixCall scall,
1515cddf8d76SBarry Smith                                     Mat **B)
1516cddf8d76SBarry Smith {
1517cddf8d76SBarry Smith   int ierr,i;
1518cddf8d76SBarry Smith 
1519*3a40ed3dSBarry Smith   PetscFunctionBegin;
1520cddf8d76SBarry Smith   if (scall == MAT_INITIAL_MATRIX) {
15210452661fSBarry Smith     *B = (Mat *) PetscMalloc( (n+1)*sizeof(Mat) ); CHKPTRQ(*B);
1522cddf8d76SBarry Smith   }
1523cddf8d76SBarry Smith 
1524cddf8d76SBarry Smith   for ( i=0; i<n; i++ ) {
1525905e6a2fSBarry Smith     ierr = MatGetSubMatrix_SeqAIJ(A,irow[i],icol[i],scall,&(*B)[i]);CHKERRQ(ierr);
1526cddf8d76SBarry Smith   }
1527*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
1528cddf8d76SBarry Smith }
1529cddf8d76SBarry Smith 
15305615d1e5SSatish Balay #undef __FUNC__
15315615d1e5SSatish Balay #define __FUNC__ "MatGetBlockSize_SeqAIJ"
15328f6be9afSLois Curfman McInnes int MatGetBlockSize_SeqAIJ(Mat A, int *bs)
15335a838052SSatish Balay {
15345a838052SSatish Balay   *bs = 1;
1535*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
15365a838052SSatish Balay }
15375a838052SSatish Balay 
15385615d1e5SSatish Balay #undef __FUNC__
15395615d1e5SSatish Balay #define __FUNC__ "MatIncreaseOverlap_SeqAIJ"
15408f6be9afSLois Curfman McInnes int MatIncreaseOverlap_SeqAIJ(Mat A, int is_max, IS *is, int ov)
15414dcbc457SBarry Smith {
1542e4d965acSSatish Balay   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
154306763907SSatish Balay   int        shift, row, i,j,k,l,m,n, *idx,ierr, *nidx, isz, val;
15448a047759SSatish Balay   int        start, end, *ai, *aj;
1545bbd702dbSSatish Balay   BT         table;
1546bbd702dbSSatish Balay 
1547*3a40ed3dSBarry Smith   PetscFunctionBegin;
15488a047759SSatish Balay   shift = a->indexshift;
1549e4d965acSSatish Balay   m     = a->m;
1550e4d965acSSatish Balay   ai    = a->i;
15518a047759SSatish Balay   aj    = a->j+shift;
15528a047759SSatish Balay 
1553e3372554SBarry Smith   if (ov < 0)  SETERRQ(1,0,"illegal overlap value used");
155406763907SSatish Balay 
155506763907SSatish Balay   nidx  = (int *) PetscMalloc((m+1)*sizeof(int)); CHKPTRQ(nidx);
1556bbd702dbSSatish Balay   ierr  = BTCreate(m,table); CHKERRQ(ierr);
155706763907SSatish Balay 
1558e4d965acSSatish Balay   for ( i=0; i<is_max; i++ ) {
1559b97fc60eSLois Curfman McInnes     /* Initialize the two local arrays */
1560e4d965acSSatish Balay     isz  = 0;
1561bbd702dbSSatish Balay     BTMemzero(m,table);
1562e4d965acSSatish Balay 
1563e4d965acSSatish Balay     /* Extract the indices, assume there can be duplicate entries */
15644dcbc457SBarry Smith     ierr = ISGetIndices(is[i],&idx);  CHKERRQ(ierr);
156577c4ece6SBarry Smith     ierr = ISGetSize(is[i],&n);  CHKERRQ(ierr);
1566e4d965acSSatish Balay 
1567dd097bc3SLois Curfman McInnes     /* Enter these into the temp arrays. I.e., mark table[row], enter row into new index */
1568e4d965acSSatish Balay     for ( j=0; j<n ; ++j){
1569bbd702dbSSatish Balay       if(!BTLookupSet(table, idx[j])) { nidx[isz++] = idx[j];}
15704dcbc457SBarry Smith     }
157106763907SSatish Balay     ierr = ISRestoreIndices(is[i],&idx);  CHKERRQ(ierr);
157206763907SSatish Balay     ierr = ISDestroy(is[i]); CHKERRQ(ierr);
1573e4d965acSSatish Balay 
157404a348a9SBarry Smith     k = 0;
157504a348a9SBarry Smith     for ( j=0; j<ov; j++){ /* for each overlap */
157604a348a9SBarry Smith       n = isz;
157706763907SSatish Balay       for ( ; k<n ; k++){ /* do only those rows in nidx[k], which are not done yet */
1578e4d965acSSatish Balay         row   = nidx[k];
1579e4d965acSSatish Balay         start = ai[row];
1580e4d965acSSatish Balay         end   = ai[row+1];
158104a348a9SBarry Smith         for ( l = start; l<end ; l++){
15828a047759SSatish Balay           val = aj[l] + shift;
15832a8f2162SSatish Balay           if (!BTLookupSet(table,val)) {nidx[isz++] = val;}
1584e4d965acSSatish Balay         }
1585e4d965acSSatish Balay       }
1586e4d965acSSatish Balay     }
1587029af93fSBarry Smith     ierr = ISCreateGeneral(PETSC_COMM_SELF, isz, nidx, (is+i)); CHKERRQ(ierr);
1588e4d965acSSatish Balay   }
1589bbd702dbSSatish Balay   BTDestroy(table);
159006763907SSatish Balay   PetscFree(nidx);
1591*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
15924dcbc457SBarry Smith }
159317ab2063SBarry Smith 
15940513a670SBarry Smith /* -------------------------------------------------------------- */
15950513a670SBarry Smith #undef __FUNC__
15960513a670SBarry Smith #define __FUNC__ "MatPermute_SeqAIJ"
15970513a670SBarry Smith int MatPermute_SeqAIJ(Mat A, IS rowp, IS colp, Mat *B)
15980513a670SBarry Smith {
15990513a670SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
16000513a670SBarry Smith   Scalar     *vwork;
16010513a670SBarry Smith   int        i, ierr, nz, m = a->m, n = a->n, *cwork;
16020513a670SBarry Smith   int        *row,*col,*cnew,j,*lens;
160356cd22aeSBarry Smith   IS         icolp,irowp;
16040513a670SBarry Smith 
1605*3a40ed3dSBarry Smith   PetscFunctionBegin;
160656cd22aeSBarry Smith   ierr = ISInvertPermutation(rowp,&irowp); CHKERRQ(ierr);
160756cd22aeSBarry Smith   ierr = ISGetIndices(irowp,&row); CHKERRQ(ierr);
160856cd22aeSBarry Smith   ierr = ISInvertPermutation(colp,&icolp); CHKERRQ(ierr);
160956cd22aeSBarry Smith   ierr = ISGetIndices(icolp,&col); CHKERRQ(ierr);
16100513a670SBarry Smith 
16110513a670SBarry Smith   /* determine lengths of permuted rows */
16120513a670SBarry Smith   lens = (int *) PetscMalloc( (m+1)*sizeof(int) ); CHKPTRQ(lens);
16130513a670SBarry Smith   for (i=0; i<m; i++ ) {
16140513a670SBarry Smith     lens[row[i]] = a->i[i+1] - a->i[i];
16150513a670SBarry Smith   }
16160513a670SBarry Smith   ierr = MatCreateSeqAIJ(A->comm,m,n,0,lens,B);CHKERRQ(ierr);
16170513a670SBarry Smith   PetscFree(lens);
16180513a670SBarry Smith 
16190513a670SBarry Smith   cnew = (int *) PetscMalloc( n*sizeof(int) ); CHKPTRQ(cnew);
16200513a670SBarry Smith   for (i=0; i<m; i++) {
16210513a670SBarry Smith     ierr = MatGetRow(A,i,&nz,&cwork,&vwork); CHKERRQ(ierr);
16220513a670SBarry Smith     for (j=0; j<nz; j++ ) { cnew[j] = col[cwork[j]];}
16230513a670SBarry Smith     ierr = MatSetValues(*B,1,&row[i],nz,cnew,vwork,INSERT_VALUES); CHKERRQ(ierr);
16240513a670SBarry Smith     ierr = MatRestoreRow(A,i,&nz,&cwork,&vwork); CHKERRQ(ierr);
16250513a670SBarry Smith   }
16260513a670SBarry Smith   PetscFree(cnew);
16270513a670SBarry Smith   ierr = MatAssemblyBegin(*B,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr);
16280513a670SBarry Smith   ierr = MatAssemblyEnd(*B,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr);
162956cd22aeSBarry Smith   ierr = ISRestoreIndices(irowp,&row); CHKERRQ(ierr);
163056cd22aeSBarry Smith   ierr = ISRestoreIndices(icolp,&col); CHKERRQ(ierr);
163156cd22aeSBarry Smith   ierr = ISDestroy(irowp); CHKERRQ(ierr);
163256cd22aeSBarry Smith   ierr = ISDestroy(icolp); CHKERRQ(ierr);
1633*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
16340513a670SBarry Smith }
16350513a670SBarry Smith 
16365615d1e5SSatish Balay #undef __FUNC__
16375615d1e5SSatish Balay #define __FUNC__ "MatPrintHelp_SeqAIJ"
1638682d7d0cSBarry Smith int MatPrintHelp_SeqAIJ(Mat A)
1639682d7d0cSBarry Smith {
1640682d7d0cSBarry Smith   static int called = 0;
1641682d7d0cSBarry Smith   MPI_Comm   comm = A->comm;
1642682d7d0cSBarry Smith 
1643*3a40ed3dSBarry Smith   PetscFunctionBegin;
1644*3a40ed3dSBarry Smith   if (called) {PetscFunctionReturn(0);} else called = 1;
164577c4ece6SBarry Smith   PetscPrintf(comm," Options for MATSEQAIJ and MATMPIAIJ matrix formats (the defaults):\n");
16460f665d81SLois Curfman McInnes   PetscPrintf(comm,"  -mat_lu_pivotthreshold <threshold>: Set pivoting threshold\n");
16470f665d81SLois Curfman McInnes   PetscPrintf(comm,"  -mat_aij_oneindex: internal indices begin at 1 instead of the default 0.\n");
16480f665d81SLois Curfman McInnes   PetscPrintf(comm,"  -mat_aij_no_inode: Do not use inodes\n");
16490f665d81SLois Curfman McInnes   PetscPrintf(comm,"  -mat_aij_inode_limit <limit>: Set inode limit (max limit=5)\n");
1650682d7d0cSBarry Smith #if defined(HAVE_ESSL)
16510f665d81SLois Curfman McInnes   PetscPrintf(comm,"  -mat_aij_essl: Use IBM sparse LU factorization and solve.\n");
1652682d7d0cSBarry Smith #endif
1653*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
1654682d7d0cSBarry Smith }
16558f6be9afSLois Curfman McInnes extern int MatEqual_SeqAIJ(Mat A,Mat B, PetscTruth* flg);
1656a93ec695SBarry Smith extern int MatFDColoringCreate_SeqAIJ(Mat,ISColoring,MatFDColoring);
1657a93ec695SBarry Smith extern int MatColoringPatch_SeqAIJ(Mat,int,int *,ISColoring *);
1658a93ec695SBarry Smith 
1659682d7d0cSBarry Smith /* -------------------------------------------------------------------*/
166017ab2063SBarry Smith static struct _MatOps MatOps = {MatSetValues_SeqAIJ,
166117ab2063SBarry Smith        MatGetRow_SeqAIJ,MatRestoreRow_SeqAIJ,
1662416022c9SBarry Smith        MatMult_SeqAIJ,MatMultAdd_SeqAIJ,
1663416022c9SBarry Smith        MatMultTrans_SeqAIJ,MatMultTransAdd_SeqAIJ,
166417ab2063SBarry Smith        MatSolve_SeqAIJ,MatSolveAdd_SeqAIJ,
166517ab2063SBarry Smith        MatSolveTrans_SeqAIJ,MatSolveTransAdd_SeqAIJ,
166617ab2063SBarry Smith        MatLUFactor_SeqAIJ,0,
166717ab2063SBarry Smith        MatRelax_SeqAIJ,
166817ab2063SBarry Smith        MatTranspose_SeqAIJ,
16697264ac53SSatish Balay        MatGetInfo_SeqAIJ,MatEqual_SeqAIJ,
1670f0b747eeSBarry Smith        MatGetDiagonal_SeqAIJ,MatDiagonalScale_SeqAIJ,MatNorm_SeqAIJ,
167117ab2063SBarry Smith        0,MatAssemblyEnd_SeqAIJ,
167217ab2063SBarry Smith        MatCompress_SeqAIJ,
167317ab2063SBarry Smith        MatSetOption_SeqAIJ,MatZeroEntries_SeqAIJ,MatZeroRows_SeqAIJ,
167417ab2063SBarry Smith        MatLUFactorSymbolic_SeqAIJ,MatLUFactorNumeric_SeqAIJ,0,0,
167517ab2063SBarry Smith        MatGetSize_SeqAIJ,MatGetSize_SeqAIJ,MatGetOwnershipRange_SeqAIJ,
167617ab2063SBarry Smith        MatILUFactorSymbolic_SeqAIJ,0,
167794a9d846SBarry Smith        0,0,
16783d1612f7SBarry Smith        MatConvertSameType_SeqAIJ,0,0,
1679cddf8d76SBarry Smith        MatILUFactor_SeqAIJ,0,0,
16807eb43aa7SLois Curfman McInnes        MatGetSubMatrices_SeqAIJ,MatIncreaseOverlap_SeqAIJ,
1681682d7d0cSBarry Smith        MatGetValues_SeqAIJ,0,
1682f0b747eeSBarry Smith        MatPrintHelp_SeqAIJ,
16835a838052SSatish Balay        MatScale_SeqAIJ,0,0,
16846945ee14SBarry Smith        MatILUDTFactor_SeqAIJ,
16856945ee14SBarry Smith        MatGetBlockSize_SeqAIJ,
16863b2fbd54SBarry Smith        MatGetRowIJ_SeqAIJ,
16873b2fbd54SBarry Smith        MatRestoreRowIJ_SeqAIJ,
16883b2fbd54SBarry Smith        MatGetColumnIJ_SeqAIJ,
1689a93ec695SBarry Smith        MatRestoreColumnIJ_SeqAIJ,
1690a93ec695SBarry Smith        MatFDColoringCreate_SeqAIJ,
16910513a670SBarry Smith        MatColoringPatch_SeqAIJ,
16920513a670SBarry Smith        0,
16930513a670SBarry Smith        MatPermute_SeqAIJ};
169417ab2063SBarry Smith 
169517ab2063SBarry Smith extern int MatUseSuperLU_SeqAIJ(Mat);
169617ab2063SBarry Smith extern int MatUseEssl_SeqAIJ(Mat);
169717ab2063SBarry Smith extern int MatUseDXML_SeqAIJ(Mat);
169817ab2063SBarry Smith 
16995615d1e5SSatish Balay #undef __FUNC__
17005615d1e5SSatish Balay #define __FUNC__ "MatCreateSeqAIJ"
170117ab2063SBarry Smith /*@C
1702682d7d0cSBarry Smith    MatCreateSeqAIJ - Creates a sparse matrix in AIJ (compressed row) format
17030d15e28bSLois Curfman McInnes    (the default parallel PETSc format).  For good matrix assembly performance
17046e62573dSLois Curfman McInnes    the user should preallocate the matrix storage by setting the parameter nz
17052bd5e0b2SLois Curfman McInnes    (or the array nzz).  By setting these parameters accurately, performance
17062bd5e0b2SLois Curfman McInnes    during matrix assembly can be increased by more than a factor of 50.
170717ab2063SBarry Smith 
170817ab2063SBarry Smith    Input Parameters:
1709029af93fSBarry Smith .  comm - MPI communicator, set to PETSC_COMM_SELF
171017ab2063SBarry Smith .  m - number of rows
171117ab2063SBarry Smith .  n - number of columns
171217ab2063SBarry Smith .  nz - number of nonzeros per row (same for all rows)
17132bd5e0b2SLois Curfman McInnes .  nzz - array containing the number of nonzeros in the various rows
17142bd5e0b2SLois Curfman McInnes          (possibly different for each row) or PETSC_NULL
171517ab2063SBarry Smith 
171617ab2063SBarry Smith    Output Parameter:
1717416022c9SBarry Smith .  A - the matrix
171817ab2063SBarry Smith 
171917ab2063SBarry Smith    Notes:
172017ab2063SBarry Smith    The AIJ format (also called the Yale sparse matrix format or
172117ab2063SBarry Smith    compressed row storage), is fully compatible with standard Fortran 77
17220002213bSLois Curfman McInnes    storage.  That is, the stored row and column indices can begin at
172344cd7ae7SLois Curfman McInnes    either one (as in Fortran) or zero.  See the users' manual for details.
172417ab2063SBarry Smith 
172517ab2063SBarry Smith    Specify the preallocated storage with either nz or nnz (not both).
1726a40aa06bSLois Curfman McInnes    Set nz=PETSC_DEFAULT and nnz=PETSC_NULL for PETSc to control dynamic memory
17273d323bbdSBarry Smith    allocation.  For large problems you MUST preallocate memory or you
17286da5968aSLois Curfman McInnes    will get TERRIBLE performance, see the users' manual chapter on matrices.
172917ab2063SBarry Smith 
1730682d7d0cSBarry Smith    By default, this format uses inodes (identical nodes) when possible, to
17314fca80b9SLois Curfman McInnes    improve numerical efficiency of matrix-vector products and solves. We
1732682d7d0cSBarry Smith    search for consecutive rows with the same nonzero structure, thereby
17336c7ebb05SLois Curfman McInnes    reusing matrix information to achieve increased efficiency.
17346c7ebb05SLois Curfman McInnes 
17356c7ebb05SLois Curfman McInnes    Options Database Keys:
17366c7ebb05SLois Curfman McInnes $    -mat_aij_no_inode  - Do not use inodes
17376e62573dSLois Curfman McInnes $    -mat_aij_inode_limit <limit> - Set inode limit.
17386e62573dSLois Curfman McInnes $        (max limit=5)
17396e62573dSLois Curfman McInnes $    -mat_aij_oneindex - Internally use indexing starting at 1
17406e62573dSLois Curfman McInnes $        rather than 0.  Note: When calling MatSetValues(),
17416e62573dSLois Curfman McInnes $        the user still MUST index entries starting at 0!
174217ab2063SBarry Smith 
174317ab2063SBarry Smith .seealso: MatCreate(), MatCreateMPIAIJ(), MatSetValues()
174417ab2063SBarry Smith @*/
1745416022c9SBarry Smith int MatCreateSeqAIJ(MPI_Comm comm,int m,int n,int nz,int *nnz, Mat *A)
174617ab2063SBarry Smith {
1747416022c9SBarry Smith   Mat        B;
1748416022c9SBarry Smith   Mat_SeqAIJ *b;
17496945ee14SBarry Smith   int        i, len, ierr, flg,size;
17506945ee14SBarry Smith 
1751*3a40ed3dSBarry Smith   PetscFunctionBegin;
17526945ee14SBarry Smith   MPI_Comm_size(comm,&size);
1753e3372554SBarry Smith   if (size > 1) SETERRQ(1,0,"Comm must be of size 1");
1754d5d45c9bSBarry Smith 
1755416022c9SBarry Smith   *A                  = 0;
1756d4bb536fSBarry Smith   PetscHeaderCreate(B,_p_Mat,MAT_COOKIE,MATSEQAIJ,comm,MatDestroy,MatView);
1757416022c9SBarry Smith   PLogObjectCreate(B);
17580452661fSBarry Smith   B->data             = (void *) (b = PetscNew(Mat_SeqAIJ)); CHKPTRQ(b);
175944cd7ae7SLois Curfman McInnes   PetscMemzero(b,sizeof(Mat_SeqAIJ));
1760416022c9SBarry Smith   PetscMemcpy(&B->ops,&MatOps,sizeof(struct _MatOps));
1761416022c9SBarry Smith   B->destroy          = MatDestroy_SeqAIJ;
1762416022c9SBarry Smith   B->view             = MatView_SeqAIJ;
1763416022c9SBarry Smith   B->factor           = 0;
1764416022c9SBarry Smith   B->lupivotthreshold = 1.0;
176590f02eecSBarry Smith   B->mapping          = 0;
17667a743949SBarry Smith   ierr = OptionsGetDouble(PETSC_NULL,"-mat_lu_pivotthreshold",&B->lupivotthreshold,
176769957df2SSatish Balay                           &flg); CHKERRQ(ierr);
17687a743949SBarry Smith   b->ilu_preserve_row_sums = PETSC_FALSE;
17697a743949SBarry Smith   ierr = OptionsHasName(PETSC_NULL,"-pc_ilu_preserve_row_sums",
17707a743949SBarry Smith                         (int*) &b->ilu_preserve_row_sums); CHKERRQ(ierr);
1771416022c9SBarry Smith   b->row              = 0;
1772416022c9SBarry Smith   b->col              = 0;
1773416022c9SBarry Smith   b->indexshift       = 0;
1774b810aeb4SBarry Smith   b->reallocs         = 0;
177569957df2SSatish Balay   ierr = OptionsHasName(PETSC_NULL,"-mat_aij_oneindex", &flg); CHKERRQ(ierr);
177669957df2SSatish Balay   if (flg) b->indexshift = -1;
177717ab2063SBarry Smith 
177844cd7ae7SLois Curfman McInnes   b->m = m; B->m = m; B->M = m;
177944cd7ae7SLois Curfman McInnes   b->n = n; B->n = n; B->N = n;
17800452661fSBarry Smith   b->imax = (int *) PetscMalloc( (m+1)*sizeof(int) ); CHKPTRQ(b->imax);
1781b4fd4287SBarry Smith   if (nnz == PETSC_NULL) {
17827b8455f0SLois Curfman McInnes     if (nz == PETSC_DEFAULT) nz = 10;
17837b8455f0SLois Curfman McInnes     else if (nz <= 0)        nz = 1;
1784416022c9SBarry Smith     for ( i=0; i<m; i++ ) b->imax[i] = nz;
178517ab2063SBarry Smith     nz = nz*m;
1786*3a40ed3dSBarry Smith   } else {
178717ab2063SBarry Smith     nz = 0;
1788416022c9SBarry Smith     for ( i=0; i<m; i++ ) {b->imax[i] = nnz[i]; nz += nnz[i];}
178917ab2063SBarry Smith   }
179017ab2063SBarry Smith 
179117ab2063SBarry Smith   /* allocate the matrix space */
1792416022c9SBarry Smith   len     = nz*(sizeof(int) + sizeof(Scalar)) + (b->m+1)*sizeof(int);
17930452661fSBarry Smith   b->a  = (Scalar *) PetscMalloc( len ); CHKPTRQ(b->a);
1794416022c9SBarry Smith   b->j  = (int *) (b->a + nz);
1795cddf8d76SBarry Smith   PetscMemzero(b->j,nz*sizeof(int));
1796416022c9SBarry Smith   b->i  = b->j + nz;
1797416022c9SBarry Smith   b->singlemalloc = 1;
179817ab2063SBarry Smith 
1799416022c9SBarry Smith   b->i[0] = -b->indexshift;
180017ab2063SBarry Smith   for (i=1; i<m+1; i++) {
1801416022c9SBarry Smith     b->i[i] = b->i[i-1] + b->imax[i-1];
180217ab2063SBarry Smith   }
180317ab2063SBarry Smith 
1804416022c9SBarry Smith   /* b->ilen will count nonzeros in each row so far. */
18050452661fSBarry Smith   b->ilen = (int *) PetscMalloc((m+1)*sizeof(int));
1806f09e8eb9SSatish Balay   PLogObjectMemory(B,len+2*(m+1)*sizeof(int)+sizeof(struct _p_Mat)+sizeof(Mat_SeqAIJ));
1807416022c9SBarry Smith   for ( i=0; i<b->m; i++ ) { b->ilen[i] = 0;}
180817ab2063SBarry Smith 
1809416022c9SBarry Smith   b->nz               = 0;
1810416022c9SBarry Smith   b->maxnz            = nz;
1811416022c9SBarry Smith   b->sorted           = 0;
1812416022c9SBarry Smith   b->roworiented      = 1;
1813416022c9SBarry Smith   b->nonew            = 0;
1814416022c9SBarry Smith   b->diag             = 0;
1815416022c9SBarry Smith   b->solve_work       = 0;
181671bd300dSLois Curfman McInnes   b->spptr            = 0;
1817754ec7b1SSatish Balay   b->inode.node_count = 0;
1818754ec7b1SSatish Balay   b->inode.size       = 0;
18196c7ebb05SLois Curfman McInnes   b->inode.limit      = 5;
18206c7ebb05SLois Curfman McInnes   b->inode.max_limit  = 5;
18214e220ebcSLois Curfman McInnes   B->info.nz_unneeded = (double)b->maxnz;
182217ab2063SBarry Smith 
1823416022c9SBarry Smith   *A = B;
18244e220ebcSLois Curfman McInnes 
18254b14c69eSBarry Smith   /*  SuperLU is not currently supported through PETSc */
18264b14c69eSBarry Smith #if defined(HAVE_SUPERLU)
182769957df2SSatish Balay   ierr = OptionsHasName(PETSC_NULL,"-mat_aij_superlu", &flg); CHKERRQ(ierr);
182869957df2SSatish Balay   if (flg) { ierr = MatUseSuperLU_SeqAIJ(B); CHKERRQ(ierr); }
18294b14c69eSBarry Smith #endif
183069957df2SSatish Balay   ierr = OptionsHasName(PETSC_NULL,"-mat_aij_essl", &flg); CHKERRQ(ierr);
183169957df2SSatish Balay   if (flg) { ierr = MatUseEssl_SeqAIJ(B); CHKERRQ(ierr); }
183269957df2SSatish Balay   ierr = OptionsHasName(PETSC_NULL,"-mat_aij_dxml", &flg); CHKERRQ(ierr);
183369957df2SSatish Balay   if (flg) {
1834e3372554SBarry Smith     if (!b->indexshift) SETERRQ(1,0,"need -mat_aij_oneindex with -mat_aij_dxml");
1835416022c9SBarry Smith     ierr = MatUseDXML_SeqAIJ(B); CHKERRQ(ierr);
183617ab2063SBarry Smith   }
183769957df2SSatish Balay   ierr = OptionsHasName(PETSC_NULL,"-help", &flg); CHKERRQ(ierr);
183869957df2SSatish Balay   if (flg) {ierr = MatPrintHelp(B); CHKERRQ(ierr); }
1839*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
184017ab2063SBarry Smith }
184117ab2063SBarry Smith 
18425615d1e5SSatish Balay #undef __FUNC__
18435615d1e5SSatish Balay #define __FUNC__ "MatConvertSameType_SeqAIJ"
18443d1612f7SBarry Smith int MatConvertSameType_SeqAIJ(Mat A,Mat *B,int cpvalues)
184517ab2063SBarry Smith {
1846416022c9SBarry Smith   Mat        C;
1847416022c9SBarry Smith   Mat_SeqAIJ *c,*a = (Mat_SeqAIJ *) A->data;
184808480c60SBarry Smith   int        i,len, m = a->m,shift = a->indexshift;
184917ab2063SBarry Smith 
1850*3a40ed3dSBarry Smith   PetscFunctionBegin;
18514043dd9cSLois Curfman McInnes   *B = 0;
1852d4bb536fSBarry Smith   PetscHeaderCreate(C,_p_Mat,MAT_COOKIE,MATSEQAIJ,A->comm,MatDestroy,MatView);
1853416022c9SBarry Smith   PLogObjectCreate(C);
18540452661fSBarry Smith   C->data       = (void *) (c = PetscNew(Mat_SeqAIJ)); CHKPTRQ(c);
185541c01911SSatish Balay   PetscMemcpy(&C->ops,&A->ops,sizeof(struct _MatOps));
1856416022c9SBarry Smith   C->destroy    = MatDestroy_SeqAIJ;
1857416022c9SBarry Smith   C->view       = MatView_SeqAIJ;
1858416022c9SBarry Smith   C->factor     = A->factor;
1859416022c9SBarry Smith   c->row        = 0;
1860416022c9SBarry Smith   c->col        = 0;
1861416022c9SBarry Smith   c->indexshift = shift;
1862c456f294SBarry Smith   C->assembled  = PETSC_TRUE;
186317ab2063SBarry Smith 
186444cd7ae7SLois Curfman McInnes   c->m = C->m   = a->m;
186544cd7ae7SLois Curfman McInnes   c->n = C->n   = a->n;
186644cd7ae7SLois Curfman McInnes   C->M          = a->m;
186744cd7ae7SLois Curfman McInnes   C->N          = a->n;
186817ab2063SBarry Smith 
18690452661fSBarry Smith   c->imax       = (int *) PetscMalloc((m+1)*sizeof(int)); CHKPTRQ(c->imax);
18700452661fSBarry Smith   c->ilen       = (int *) PetscMalloc((m+1)*sizeof(int)); CHKPTRQ(c->ilen);
187117ab2063SBarry Smith   for ( i=0; i<m; i++ ) {
1872416022c9SBarry Smith     c->imax[i] = a->imax[i];
1873416022c9SBarry Smith     c->ilen[i] = a->ilen[i];
187417ab2063SBarry Smith   }
187517ab2063SBarry Smith 
187617ab2063SBarry Smith   /* allocate the matrix space */
1877416022c9SBarry Smith   c->singlemalloc = 1;
1878416022c9SBarry Smith   len     = (m+1)*sizeof(int)+(a->i[m])*(sizeof(Scalar)+sizeof(int));
18790452661fSBarry Smith   c->a  = (Scalar *) PetscMalloc( len ); CHKPTRQ(c->a);
1880416022c9SBarry Smith   c->j  = (int *) (c->a + a->i[m] + shift);
1881416022c9SBarry Smith   c->i  = c->j + a->i[m] + shift;
1882416022c9SBarry Smith   PetscMemcpy(c->i,a->i,(m+1)*sizeof(int));
188317ab2063SBarry Smith   if (m > 0) {
1884416022c9SBarry Smith     PetscMemcpy(c->j,a->j,(a->i[m]+shift)*sizeof(int));
188508480c60SBarry Smith     if (cpvalues == COPY_VALUES) {
1886416022c9SBarry Smith       PetscMemcpy(c->a,a->a,(a->i[m]+shift)*sizeof(Scalar));
188717ab2063SBarry Smith     }
188808480c60SBarry Smith   }
188917ab2063SBarry Smith 
1890f09e8eb9SSatish Balay   PLogObjectMemory(C,len+2*(m+1)*sizeof(int)+sizeof(struct _p_Mat)+sizeof(Mat_SeqAIJ));
1891416022c9SBarry Smith   c->sorted      = a->sorted;
1892416022c9SBarry Smith   c->roworiented = a->roworiented;
1893416022c9SBarry Smith   c->nonew       = a->nonew;
18947a743949SBarry Smith   c->ilu_preserve_row_sums = a->ilu_preserve_row_sums;
189517ab2063SBarry Smith 
1896416022c9SBarry Smith   if (a->diag) {
18970452661fSBarry Smith     c->diag = (int *) PetscMalloc( (m+1)*sizeof(int) ); CHKPTRQ(c->diag);
1898416022c9SBarry Smith     PLogObjectMemory(C,(m+1)*sizeof(int));
189917ab2063SBarry Smith     for ( i=0; i<m; i++ ) {
1900416022c9SBarry Smith       c->diag[i] = a->diag[i];
190117ab2063SBarry Smith     }
1902*3a40ed3dSBarry Smith   } else c->diag          = 0;
19036c7ebb05SLois Curfman McInnes   c->inode.limit        = a->inode.limit;
19046c7ebb05SLois Curfman McInnes   c->inode.max_limit    = a->inode.max_limit;
1905754ec7b1SSatish Balay   if (a->inode.size){
1906daed632aSSatish Balay     c->inode.size       = (int *) PetscMalloc( (m+1)*sizeof(int) ); CHKPTRQ(c->inode.size);
1907754ec7b1SSatish Balay     c->inode.node_count = a->inode.node_count;
1908daed632aSSatish Balay     PetscMemcpy( c->inode.size, a->inode.size, (m+1)*sizeof(int));
1909754ec7b1SSatish Balay   } else {
1910754ec7b1SSatish Balay     c->inode.size       = 0;
1911754ec7b1SSatish Balay     c->inode.node_count = 0;
1912754ec7b1SSatish Balay   }
1913416022c9SBarry Smith   c->nz                 = a->nz;
1914416022c9SBarry Smith   c->maxnz              = a->maxnz;
1915416022c9SBarry Smith   c->solve_work         = 0;
191676dd722bSSatish Balay   c->spptr              = 0;      /* Dangerous -I'm throwing away a->spptr */
1917754ec7b1SSatish Balay 
1918416022c9SBarry Smith   *B = C;
1919*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
192017ab2063SBarry Smith }
192117ab2063SBarry Smith 
19225615d1e5SSatish Balay #undef __FUNC__
19235615d1e5SSatish Balay #define __FUNC__ "MatLoad_SeqAIJ"
192419bcc07fSBarry Smith int MatLoad_SeqAIJ(Viewer viewer,MatType type,Mat *A)
192517ab2063SBarry Smith {
1926416022c9SBarry Smith   Mat_SeqAIJ   *a;
1927416022c9SBarry Smith   Mat          B;
192817699dbbSLois Curfman McInnes   int          i, nz, ierr, fd, header[4],size,*rowlengths = 0,M,N,shift;
1929bcd2baecSBarry Smith   MPI_Comm     comm;
193017ab2063SBarry Smith 
1931*3a40ed3dSBarry Smith   PetscFunctionBegin;
193219bcc07fSBarry Smith   PetscObjectGetComm((PetscObject) viewer,&comm);
193317699dbbSLois Curfman McInnes   MPI_Comm_size(comm,&size);
1934e3372554SBarry Smith   if (size > 1) SETERRQ(1,0,"view must have one processor");
193590ace30eSBarry Smith   ierr = ViewerBinaryGetDescriptor(viewer,&fd); CHKERRQ(ierr);
19360752156aSBarry Smith   ierr = PetscBinaryRead(fd,header,4,PETSC_INT); CHKERRQ(ierr);
1937e3372554SBarry Smith   if (header[0] != MAT_COOKIE) SETERRQ(1,0,"not matrix object in file");
193817ab2063SBarry Smith   M = header[1]; N = header[2]; nz = header[3];
193917ab2063SBarry Smith 
194017ab2063SBarry Smith   /* read in row lengths */
19410452661fSBarry Smith   rowlengths = (int*) PetscMalloc( M*sizeof(int) ); CHKPTRQ(rowlengths);
19420752156aSBarry Smith   ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT); CHKERRQ(ierr);
194317ab2063SBarry Smith 
194417ab2063SBarry Smith   /* create our matrix */
1945416022c9SBarry Smith   ierr = MatCreateSeqAIJ(comm,M,N,0,rowlengths,A); CHKERRQ(ierr);
1946416022c9SBarry Smith   B = *A;
1947416022c9SBarry Smith   a = (Mat_SeqAIJ *) B->data;
1948416022c9SBarry Smith   shift = a->indexshift;
194917ab2063SBarry Smith 
195017ab2063SBarry Smith   /* read in column indices and adjust for Fortran indexing*/
19510752156aSBarry Smith   ierr = PetscBinaryRead(fd,a->j,nz,PETSC_INT); CHKERRQ(ierr);
195217ab2063SBarry Smith   if (shift) {
195317ab2063SBarry Smith     for ( i=0; i<nz; i++ ) {
1954416022c9SBarry Smith       a->j[i] += 1;
195517ab2063SBarry Smith     }
195617ab2063SBarry Smith   }
195717ab2063SBarry Smith 
195817ab2063SBarry Smith   /* read in nonzero values */
19590752156aSBarry Smith   ierr = PetscBinaryRead(fd,a->a,nz,PETSC_SCALAR); CHKERRQ(ierr);
196017ab2063SBarry Smith 
196117ab2063SBarry Smith   /* set matrix "i" values */
1962416022c9SBarry Smith   a->i[0] = -shift;
196317ab2063SBarry Smith   for ( i=1; i<= M; i++ ) {
1964416022c9SBarry Smith     a->i[i]      = a->i[i-1] + rowlengths[i-1];
1965416022c9SBarry Smith     a->ilen[i-1] = rowlengths[i-1];
196617ab2063SBarry Smith   }
19670452661fSBarry Smith   PetscFree(rowlengths);
196817ab2063SBarry Smith 
19696d4a8577SBarry Smith   ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr);
19706d4a8577SBarry Smith   ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr);
1971*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
197217ab2063SBarry Smith }
197317ab2063SBarry Smith 
19745615d1e5SSatish Balay #undef __FUNC__
19755615d1e5SSatish Balay #define __FUNC__ "MatEqual_SeqAIJ"
19768f6be9afSLois Curfman McInnes int MatEqual_SeqAIJ(Mat A,Mat B, PetscTruth* flg)
19777264ac53SSatish Balay {
19787264ac53SSatish Balay   Mat_SeqAIJ *a = (Mat_SeqAIJ *)A->data, *b = (Mat_SeqAIJ *)B->data;
19797264ac53SSatish Balay 
1980*3a40ed3dSBarry Smith   PetscFunctionBegin;
1981e3372554SBarry Smith   if (B->type !=MATSEQAIJ)SETERRQ(1,0,"Matrices must be same type");
19827264ac53SSatish Balay 
19837264ac53SSatish Balay   /* If the  matrix dimensions are not equal, or no of nonzeros or shift */
19847264ac53SSatish Balay   if ((a->m != b->m ) || (a->n !=b->n) ||( a->nz != b->nz)||
1985bcd2baecSBarry Smith       (a->indexshift != b->indexshift)) {
1986*3a40ed3dSBarry Smith     *flg = PETSC_FALSE; PetscFunctionReturn(0);
1987bcd2baecSBarry Smith   }
19887264ac53SSatish Balay 
19897264ac53SSatish Balay   /* if the a->i are the same */
19908108c231SLois Curfman McInnes   if (PetscMemcmp(a->i,b->i,(a->m+1)*sizeof(int))) {
1991*3a40ed3dSBarry Smith     *flg = PETSC_FALSE; PetscFunctionReturn(0);
19927264ac53SSatish Balay   }
19937264ac53SSatish Balay 
19947264ac53SSatish Balay   /* if a->j are the same */
1995bcd2baecSBarry Smith   if (PetscMemcmp(a->j, b->j, (a->nz)*sizeof(int))) {
1996*3a40ed3dSBarry Smith     *flg = PETSC_FALSE; PetscFunctionReturn(0);
1997bcd2baecSBarry Smith   }
1998bcd2baecSBarry Smith 
1999bcd2baecSBarry Smith   /* if a->a are the same */
200019bcc07fSBarry Smith   if (PetscMemcmp(a->a, b->a, (a->nz)*sizeof(Scalar))) {
2001*3a40ed3dSBarry Smith     *flg = PETSC_FALSE; PetscFunctionReturn(0);
20027264ac53SSatish Balay   }
200377c4ece6SBarry Smith   *flg = PETSC_TRUE;
2004*3a40ed3dSBarry Smith   PetscFunctionReturn(0);
20057264ac53SSatish Balay 
20067264ac53SSatish Balay }
2007