xref: /petsc/src/mat/impls/aij/seq/aij.c (revision 496be53dd62c57fbc6a2cb3b3336b3fd5b5c31b5)
117ab2063SBarry Smith #ifndef lint
2*496be53dSLois Curfman McInnes static char vcid[] = "$Id: aij.c,v 1.222 1997/06/18 01:44:15 curfman Exp curfman $";
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"
15e4d965acSSatish Balay #include "petsc.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 
28e3372554SBarry Smith   SETERRQ(ierr,0,"Not implemented");
29a2ce50c7SBarry Smith }
30a2ce50c7SBarry Smith 
31bcd2baecSBarry Smith extern int MatToSymmetricIJ_SeqAIJ(int,int*,int*,int,int,int**,int**);
3217ab2063SBarry Smith 
335615d1e5SSatish Balay #undef __FUNC__
345615d1e5SSatish Balay #define __FUNC__ "MatGetRowIJ_SeqAIJ"
358f6be9afSLois Curfman McInnes int MatGetRowIJ_SeqAIJ(Mat A,int oshift,PetscTruth symmetric,int *m,int **ia,int **ja,
366945ee14SBarry Smith                            PetscTruth *done)
3717ab2063SBarry Smith {
38416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
396945ee14SBarry Smith   int        ierr,i,ishift;
4017ab2063SBarry Smith 
4131625ec5SSatish Balay   *m     = A->m;
426945ee14SBarry Smith   if (!ia) return 0;
436945ee14SBarry Smith   ishift = a->indexshift;
446945ee14SBarry Smith   if (symmetric) {
4531625ec5SSatish Balay     ierr = MatToSymmetricIJ_SeqAIJ(a->m,a->i,a->j,ishift,oshift,ia,ja); CHKERRQ(ierr);
466945ee14SBarry Smith   } else if (oshift == 0 && ishift == -1) {
4731625ec5SSatish Balay     int nz = a->i[a->m];
483b2fbd54SBarry Smith     /* malloc space and  subtract 1 from i and j indices */
4931625ec5SSatish Balay     *ia = (int *) PetscMalloc( (a->m+1)*sizeof(int) ); CHKPTRQ(*ia);
503b2fbd54SBarry Smith     *ja = (int *) PetscMalloc( (nz+1)*sizeof(int) ); CHKPTRQ(*ja);
513b2fbd54SBarry Smith     for ( i=0; i<nz; i++ ) (*ja)[i] = a->j[i] - 1;
5231625ec5SSatish Balay     for ( i=0; i<a->m+1; i++ ) (*ia)[i] = a->i[i] - 1;
536945ee14SBarry Smith   } else if (oshift == 1 && ishift == 0) {
5431625ec5SSatish Balay     int nz = a->i[a->m] + 1;
553b2fbd54SBarry Smith     /* malloc space and  add 1 to i and j indices */
5631625ec5SSatish Balay     *ia = (int *) PetscMalloc( (a->m+1)*sizeof(int) ); CHKPTRQ(*ia);
573b2fbd54SBarry Smith     *ja = (int *) PetscMalloc( (nz+1)*sizeof(int) ); CHKPTRQ(*ja);
583b2fbd54SBarry Smith     for ( i=0; i<nz; i++ ) (*ja)[i] = a->j[i] + 1;
5931625ec5SSatish Balay     for ( i=0; i<a->m+1; i++ ) (*ia)[i] = a->i[i] + 1;
606945ee14SBarry Smith   } else {
616945ee14SBarry Smith     *ia = a->i; *ja = a->j;
62a2ce50c7SBarry Smith   }
63a2ce50c7SBarry Smith 
64a2744918SBarry Smith   return 0;
65a2744918SBarry Smith }
66a2744918SBarry Smith 
675615d1e5SSatish Balay #undef __FUNC__
685615d1e5SSatish Balay #define __FUNC__ "MatRestoreRowIJ_SeqAIJ"
698f6be9afSLois Curfman McInnes int MatRestoreRowIJ_SeqAIJ(Mat A,int oshift,PetscTruth symmetric,int *n,int **ia,int **ja,
706945ee14SBarry Smith                                PetscTruth *done)
716945ee14SBarry Smith {
726945ee14SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
733b2fbd54SBarry Smith   int        ishift = a->indexshift;
746945ee14SBarry Smith 
756945ee14SBarry Smith   if (!ia) return 0;
763b2fbd54SBarry Smith   if (symmetric || (oshift == 0 && ishift == -1) || (oshift == 1 && ishift == 0)) {
776945ee14SBarry Smith     PetscFree(*ia);
786945ee14SBarry Smith     PetscFree(*ja);
79bcd2baecSBarry Smith   }
8017ab2063SBarry Smith   return 0;
8117ab2063SBarry Smith }
8217ab2063SBarry Smith 
835615d1e5SSatish Balay #undef __FUNC__
845615d1e5SSatish Balay #define __FUNC__ "MatGetColumnIJ_SeqAIJ"
8543a90d84SBarry Smith int MatGetColumnIJ_SeqAIJ(Mat A,int oshift,PetscTruth symmetric,int *nn,int **ia,int **ja,
863b2fbd54SBarry Smith                            PetscTruth *done)
873b2fbd54SBarry Smith {
883b2fbd54SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
89a93ec695SBarry Smith   int        ierr,i,ishift = a->indexshift,*collengths,*cia,*cja,n = A->n,m = A->m;
90a93ec695SBarry Smith   int        nz = a->i[m]+ishift,row,*jj,mr,col;
913b2fbd54SBarry Smith 
923b2fbd54SBarry Smith   *nn     = A->n;
933b2fbd54SBarry Smith   if (!ia) return 0;
943b2fbd54SBarry Smith   if (symmetric) {
95179192dfSSatish Balay     ierr = MatToSymmetricIJ_SeqAIJ(a->m,a->i,a->j,ishift,oshift,ia,ja); CHKERRQ(ierr);
963b2fbd54SBarry Smith   } else {
9761d2ded1SBarry Smith     collengths = (int *) PetscMalloc( (n+1)*sizeof(int) ); CHKPTRQ(collengths);
983b2fbd54SBarry Smith     PetscMemzero(collengths,n*sizeof(int));
993b2fbd54SBarry Smith     cia        = (int *) PetscMalloc( (n+1)*sizeof(int) ); CHKPTRQ(cia);
100a93ec695SBarry Smith     cja        = (int *) PetscMalloc( (nz+1)*sizeof(int) ); CHKPTRQ(cja);
1013b2fbd54SBarry Smith     jj = a->j;
1023b2fbd54SBarry Smith     for ( i=0; i<nz; i++ ) {
1033b2fbd54SBarry Smith       collengths[jj[i] + ishift]++;
1043b2fbd54SBarry Smith     }
1053b2fbd54SBarry Smith     cia[0] = oshift;
1063b2fbd54SBarry Smith     for ( i=0; i<n; i++) {
1073b2fbd54SBarry Smith       cia[i+1] = cia[i] + collengths[i];
1083b2fbd54SBarry Smith     }
1093b2fbd54SBarry Smith     PetscMemzero(collengths,n*sizeof(int));
1103b2fbd54SBarry Smith     jj = a->j;
111a93ec695SBarry Smith     for ( row=0; row<m; row++ ) {
112a93ec695SBarry Smith       mr = a->i[row+1] - a->i[row];
113a93ec695SBarry Smith       for ( i=0; i<mr; i++ ) {
1143b2fbd54SBarry Smith         col = *jj++ + ishift;
1153b2fbd54SBarry Smith         cja[cia[col] + collengths[col]++ - oshift] = row + oshift;
1163b2fbd54SBarry Smith       }
1173b2fbd54SBarry Smith     }
1183b2fbd54SBarry Smith     PetscFree(collengths);
1193b2fbd54SBarry Smith     *ia = cia; *ja = cja;
1203b2fbd54SBarry Smith   }
1213b2fbd54SBarry Smith 
1223b2fbd54SBarry Smith   return 0;
1233b2fbd54SBarry Smith }
1243b2fbd54SBarry Smith 
1255615d1e5SSatish Balay #undef __FUNC__
1265615d1e5SSatish Balay #define __FUNC__ "MatRestoreColumnIJ_SeqAIJ"
12743a90d84SBarry Smith int MatRestoreColumnIJ_SeqAIJ(Mat A,int oshift,PetscTruth symmetric,int *n,int **ia,
1283b2fbd54SBarry Smith                                      int **ja,PetscTruth *done)
1293b2fbd54SBarry Smith {
1303b2fbd54SBarry Smith   if (!ia) return 0;
1313b2fbd54SBarry Smith 
1323b2fbd54SBarry Smith   PetscFree(*ia);
1333b2fbd54SBarry Smith   PetscFree(*ja);
1343b2fbd54SBarry Smith 
1353b2fbd54SBarry Smith   return 0;
1363b2fbd54SBarry Smith }
1373b2fbd54SBarry Smith 
138227d817aSBarry Smith #define CHUNKSIZE   15
13917ab2063SBarry Smith 
1405615d1e5SSatish Balay #undef __FUNC__
1415615d1e5SSatish Balay #define __FUNC__ "MatSetValues_SeqAIJ"
14261d2ded1SBarry Smith int MatSetValues_SeqAIJ(Mat A,int m,int *im,int n,int *in,Scalar *v,InsertMode is)
14317ab2063SBarry Smith {
144416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
145416022c9SBarry Smith   int        *rp,k,low,high,t,ii,row,nrow,i,col,l,rmax, N, sorted = a->sorted;
1464b0e389bSBarry Smith   int        *imax = a->imax, *ai = a->i, *ailen = a->ilen,roworiented = a->roworiented;
147d5d45c9bSBarry Smith   int        *aj = a->j, nonew = a->nonew,shift = a->indexshift;
148416022c9SBarry Smith   Scalar     *ap,value, *aa = a->a;
14917ab2063SBarry Smith 
15017ab2063SBarry Smith   for ( k=0; k<m; k++ ) { /* loop over added rows */
151416022c9SBarry Smith     row  = im[k];
1523b2fbd54SBarry Smith #if defined(PETSC_BOPT_g)
153e3372554SBarry Smith     if (row < 0) SETERRQ(1,0,"Negative row");
154e3372554SBarry Smith     if (row >= a->m) SETERRQ(1,0,"Row too large");
1553b2fbd54SBarry Smith #endif
15617ab2063SBarry Smith     rp   = aj + ai[row] + shift; ap = aa + ai[row] + shift;
15717ab2063SBarry Smith     rmax = imax[row]; nrow = ailen[row];
158416022c9SBarry Smith     low = 0;
15917ab2063SBarry Smith     for ( l=0; l<n; l++ ) { /* loop over added columns */
1603b2fbd54SBarry Smith #if defined(PETSC_BOPT_g)
161e3372554SBarry Smith       if (in[l] < 0) SETERRQ(1,0,"Negative column");
162e3372554SBarry Smith       if (in[l] >= a->n) SETERRQ(1,0,"Column too large");
1633b2fbd54SBarry Smith #endif
1644b0e389bSBarry Smith       col = in[l] - shift;
1654b0e389bSBarry Smith       if (roworiented) {
1664b0e389bSBarry Smith         value = *v++;
1674b0e389bSBarry Smith       }
1684b0e389bSBarry Smith       else {
1694b0e389bSBarry Smith         value = v[k + l*m];
1704b0e389bSBarry Smith       }
171416022c9SBarry Smith       if (!sorted) low = 0; high = nrow;
172416022c9SBarry Smith       while (high-low > 5) {
173416022c9SBarry Smith         t = (low+high)/2;
174416022c9SBarry Smith         if (rp[t] > col) high = t;
175416022c9SBarry Smith         else             low  = t;
17617ab2063SBarry Smith       }
177416022c9SBarry Smith       for ( i=low; i<high; i++ ) {
17817ab2063SBarry Smith         if (rp[i] > col) break;
17917ab2063SBarry Smith         if (rp[i] == col) {
180416022c9SBarry Smith           if (is == ADD_VALUES) ap[i] += value;
18117ab2063SBarry Smith           else                  ap[i] = value;
18217ab2063SBarry Smith           goto noinsert;
18317ab2063SBarry Smith         }
18417ab2063SBarry Smith       }
185c2653b3dSLois Curfman McInnes       if (nonew == 1) goto noinsert;
18611ebbc71SLois Curfman McInnes       else if (nonew == -1) SETERRQ(1,0,"Inserting a new nonzero in the matrix");
18717ab2063SBarry Smith       if (nrow >= rmax) {
18817ab2063SBarry Smith         /* there is no extra room in row, therefore enlarge */
189416022c9SBarry Smith         int    new_nz = ai[a->m] + CHUNKSIZE,len,*new_i,*new_j;
19017ab2063SBarry Smith         Scalar *new_a;
19117ab2063SBarry Smith 
19211ebbc71SLois Curfman McInnes         if (nonew == -2) SETERRQ(1,0,"Inserting a new nonzero in the matrix");
19396854ed6SLois Curfman McInnes 
19417ab2063SBarry Smith         /* malloc new storage space */
195416022c9SBarry Smith         len     = new_nz*(sizeof(int)+sizeof(Scalar))+(a->m+1)*sizeof(int);
1960452661fSBarry Smith         new_a   = (Scalar *) PetscMalloc( len ); CHKPTRQ(new_a);
19717ab2063SBarry Smith         new_j   = (int *) (new_a + new_nz);
19817ab2063SBarry Smith         new_i   = new_j + new_nz;
19917ab2063SBarry Smith 
20017ab2063SBarry Smith         /* copy over old data into new slots */
20117ab2063SBarry Smith         for ( ii=0; ii<row+1; ii++ ) {new_i[ii] = ai[ii];}
202416022c9SBarry Smith         for ( ii=row+1; ii<a->m+1; ii++ ) {new_i[ii] = ai[ii]+CHUNKSIZE;}
203416022c9SBarry Smith         PetscMemcpy(new_j,aj,(ai[row]+nrow+shift)*sizeof(int));
204416022c9SBarry Smith         len = (new_nz - CHUNKSIZE - ai[row] - nrow - shift);
205416022c9SBarry Smith         PetscMemcpy(new_j+ai[row]+shift+nrow+CHUNKSIZE,aj+ai[row]+shift+nrow,
20617ab2063SBarry Smith                                                            len*sizeof(int));
207416022c9SBarry Smith         PetscMemcpy(new_a,aa,(ai[row]+nrow+shift)*sizeof(Scalar));
208416022c9SBarry Smith         PetscMemcpy(new_a+ai[row]+shift+nrow+CHUNKSIZE,aa+ai[row]+shift+nrow,
20917ab2063SBarry Smith                                                            len*sizeof(Scalar));
21017ab2063SBarry Smith         /* free up old matrix storage */
2110452661fSBarry Smith         PetscFree(a->a);
2120452661fSBarry Smith         if (!a->singlemalloc) {PetscFree(a->i);PetscFree(a->j);}
213416022c9SBarry Smith         aa = a->a = new_a; ai = a->i = new_i; aj = a->j = new_j;
214416022c9SBarry Smith         a->singlemalloc = 1;
21517ab2063SBarry Smith 
21617ab2063SBarry Smith         rp   = aj + ai[row] + shift; ap = aa + ai[row] + shift;
217416022c9SBarry Smith         rmax = imax[row] = imax[row] + CHUNKSIZE;
218416022c9SBarry Smith         PLogObjectMemory(A,CHUNKSIZE*(sizeof(int) + sizeof(Scalar)));
219416022c9SBarry Smith         a->maxnz += CHUNKSIZE;
220b810aeb4SBarry Smith         a->reallocs++;
22117ab2063SBarry Smith       }
222416022c9SBarry Smith       N = nrow++ - 1; a->nz++;
223416022c9SBarry Smith       /* shift up all the later entries in this row */
224416022c9SBarry Smith       for ( ii=N; ii>=i; ii-- ) {
22517ab2063SBarry Smith         rp[ii+1] = rp[ii];
22617ab2063SBarry Smith         ap[ii+1] = ap[ii];
22717ab2063SBarry Smith       }
22817ab2063SBarry Smith       rp[i] = col;
22917ab2063SBarry Smith       ap[i] = value;
23017ab2063SBarry Smith       noinsert:;
231416022c9SBarry Smith       low = i + 1;
23217ab2063SBarry Smith     }
23317ab2063SBarry Smith     ailen[row] = nrow;
23417ab2063SBarry Smith   }
23517ab2063SBarry Smith   return 0;
23617ab2063SBarry Smith }
23717ab2063SBarry Smith 
2385615d1e5SSatish Balay #undef __FUNC__
2395615d1e5SSatish Balay #define __FUNC__ "MatGetValues_SeqAIJ"
2408f6be9afSLois Curfman McInnes int MatGetValues_SeqAIJ(Mat A,int m,int *im,int n,int *in,Scalar *v)
2417eb43aa7SLois Curfman McInnes {
2427eb43aa7SLois Curfman McInnes   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
243b49de8d1SLois Curfman McInnes   int        *rp, k, low, high, t, row, nrow, i, col, l, *aj = a->j;
2447eb43aa7SLois Curfman McInnes   int        *ai = a->i, *ailen = a->ilen, shift = a->indexshift;
2457eb43aa7SLois Curfman McInnes   Scalar     *ap, *aa = a->a, zero = 0.0;
2467eb43aa7SLois Curfman McInnes 
2477eb43aa7SLois Curfman McInnes   for ( k=0; k<m; k++ ) { /* loop over rows */
2487eb43aa7SLois Curfman McInnes     row  = im[k];
249e3372554SBarry Smith     if (row < 0) SETERRQ(1,0,"Negative row");
250e3372554SBarry Smith     if (row >= a->m) SETERRQ(1,0,"Row too large");
2517eb43aa7SLois Curfman McInnes     rp   = aj + ai[row] + shift; ap = aa + ai[row] + shift;
2527eb43aa7SLois Curfman McInnes     nrow = ailen[row];
2537eb43aa7SLois Curfman McInnes     for ( l=0; l<n; l++ ) { /* loop over columns */
254e3372554SBarry Smith       if (in[l] < 0) SETERRQ(1,0,"Negative column");
255e3372554SBarry Smith       if (in[l] >= a->n) SETERRQ(1,0,"Column too large");
2567eb43aa7SLois Curfman McInnes       col = in[l] - shift;
2577eb43aa7SLois Curfman McInnes       high = nrow; low = 0; /* assume unsorted */
2587eb43aa7SLois Curfman McInnes       while (high-low > 5) {
2597eb43aa7SLois Curfman McInnes         t = (low+high)/2;
2607eb43aa7SLois Curfman McInnes         if (rp[t] > col) high = t;
2617eb43aa7SLois Curfman McInnes         else             low  = t;
2627eb43aa7SLois Curfman McInnes       }
2637eb43aa7SLois Curfman McInnes       for ( i=low; i<high; i++ ) {
2647eb43aa7SLois Curfman McInnes         if (rp[i] > col) break;
2657eb43aa7SLois Curfman McInnes         if (rp[i] == col) {
266b49de8d1SLois Curfman McInnes           *v++ = ap[i];
2677eb43aa7SLois Curfman McInnes           goto finished;
2687eb43aa7SLois Curfman McInnes         }
2697eb43aa7SLois Curfman McInnes       }
270b49de8d1SLois Curfman McInnes       *v++ = zero;
2717eb43aa7SLois Curfman McInnes       finished:;
2727eb43aa7SLois Curfman McInnes     }
2737eb43aa7SLois Curfman McInnes   }
2747eb43aa7SLois Curfman McInnes   return 0;
2757eb43aa7SLois Curfman McInnes }
2767eb43aa7SLois Curfman McInnes 
27717ab2063SBarry Smith 
2785615d1e5SSatish Balay #undef __FUNC__
2795615d1e5SSatish Balay #define __FUNC__ "MatView_SeqAIJ_Binary"
2808f6be9afSLois Curfman McInnes extern int MatView_SeqAIJ_Binary(Mat A,Viewer viewer)
28117ab2063SBarry Smith {
282416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
283416022c9SBarry Smith   int        i, fd, *col_lens, ierr;
28417ab2063SBarry Smith 
28590ace30eSBarry Smith   ierr = ViewerBinaryGetDescriptor(viewer,&fd); CHKERRQ(ierr);
2860452661fSBarry Smith   col_lens = (int *) PetscMalloc( (4+a->m)*sizeof(int) ); CHKPTRQ(col_lens);
287416022c9SBarry Smith   col_lens[0] = MAT_COOKIE;
288416022c9SBarry Smith   col_lens[1] = a->m;
289416022c9SBarry Smith   col_lens[2] = a->n;
290416022c9SBarry Smith   col_lens[3] = a->nz;
291416022c9SBarry Smith 
292416022c9SBarry Smith   /* store lengths of each row and write (including header) to file */
293416022c9SBarry Smith   for ( i=0; i<a->m; i++ ) {
294416022c9SBarry Smith     col_lens[4+i] = a->i[i+1] - a->i[i];
29517ab2063SBarry Smith   }
29677c4ece6SBarry Smith   ierr = PetscBinaryWrite(fd,col_lens,4+a->m,BINARY_INT,1); CHKERRQ(ierr);
2970452661fSBarry Smith   PetscFree(col_lens);
298416022c9SBarry Smith 
299416022c9SBarry Smith   /* store column indices (zero start index) */
300416022c9SBarry Smith   if (a->indexshift) {
301416022c9SBarry Smith     for ( i=0; i<a->nz; i++ ) a->j[i]--;
30217ab2063SBarry Smith   }
30377c4ece6SBarry Smith   ierr = PetscBinaryWrite(fd,a->j,a->nz,BINARY_INT,0); CHKERRQ(ierr);
304416022c9SBarry Smith   if (a->indexshift) {
305416022c9SBarry Smith     for ( i=0; i<a->nz; i++ ) a->j[i]++;
30617ab2063SBarry Smith   }
307416022c9SBarry Smith 
308416022c9SBarry Smith   /* store nonzero values */
30977c4ece6SBarry Smith   ierr = PetscBinaryWrite(fd,a->a,a->nz,BINARY_SCALAR,0); CHKERRQ(ierr);
31017ab2063SBarry Smith   return 0;
31117ab2063SBarry Smith }
312416022c9SBarry Smith 
3135615d1e5SSatish Balay #undef __FUNC__
3145615d1e5SSatish Balay #define __FUNC__ "MatView_SeqAIJ_ASCII"
3158f6be9afSLois Curfman McInnes extern int MatView_SeqAIJ_ASCII(Mat A,Viewer viewer)
316416022c9SBarry Smith {
317416022c9SBarry Smith   Mat_SeqAIJ  *a = (Mat_SeqAIJ *) A->data;
318496e697eSBarry Smith   int         ierr, i,j, m = a->m, shift = a->indexshift, format, flg1,flg2;
31917ab2063SBarry Smith   FILE        *fd;
32017ab2063SBarry Smith   char        *outputname;
32117ab2063SBarry Smith 
32290ace30eSBarry Smith   ierr = ViewerASCIIGetPointer(viewer,&fd); CHKERRQ(ierr);
323416022c9SBarry Smith   ierr = ViewerFileGetOutputname_Private(viewer,&outputname); CHKERRQ(ierr);
32490ace30eSBarry Smith   ierr = ViewerGetFormat(viewer,&format);
325a93ec695SBarry Smith   if (format == VIEWER_FORMAT_ASCII_INFO) {
32695e01e2fSLois Curfman McInnes     return 0;
32795e01e2fSLois Curfman McInnes   }
328a93ec695SBarry Smith   else if (format == VIEWER_FORMAT_ASCII_INFO_LONG) {
329496e697eSBarry Smith     ierr = OptionsHasName(PETSC_NULL,"-mat_aij_no_inode",&flg1); CHKERRQ(ierr);
330496e697eSBarry Smith     ierr = OptionsHasName(PETSC_NULL,"-mat_no_unroll",&flg2); CHKERRQ(ierr);
331496e697eSBarry Smith     if (flg1 || flg2) fprintf(fd,"  not using I-node routines\n");
33295e01e2fSLois Curfman McInnes     else     fprintf(fd,"  using I-node routines: found %d nodes, limit used is %d\n",
33395e01e2fSLois Curfman McInnes         a->inode.node_count,a->inode.limit);
33417ab2063SBarry Smith   }
335a93ec695SBarry Smith   else if (format == VIEWER_FORMAT_ASCII_MATLAB) {
336416022c9SBarry Smith     fprintf(fd,"%% Size = %d %d \n",m,a->n);
3374e220ebcSLois Curfman McInnes     fprintf(fd,"%% Nonzeros = %d \n",a->nz);
3384e220ebcSLois Curfman McInnes     fprintf(fd,"zzz = zeros(%d,3);\n",a->nz);
33917ab2063SBarry Smith     fprintf(fd,"zzz = [\n");
34017ab2063SBarry Smith 
34117ab2063SBarry Smith     for (i=0; i<m; i++) {
342416022c9SBarry Smith       for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
34317ab2063SBarry Smith #if defined(PETSC_COMPLEX)
3446945ee14SBarry Smith         fprintf(fd,"%d %d  %18.16e + %18.16e i \n",i+1,a->j[j]+!shift,real(a->a[j]),
345416022c9SBarry Smith                    imag(a->a[j]));
34617ab2063SBarry Smith #else
3477a743949SBarry Smith         fprintf(fd,"%d %d  %18.16e\n", i+1, a->j[j]+!shift, a->a[j]);
34817ab2063SBarry Smith #endif
34917ab2063SBarry Smith       }
35017ab2063SBarry Smith     }
35117ab2063SBarry Smith     fprintf(fd,"];\n %s = spconvert(zzz);\n",outputname);
35217ab2063SBarry Smith   }
353a93ec695SBarry Smith   else if (format == VIEWER_FORMAT_ASCII_COMMON) {
35444cd7ae7SLois Curfman McInnes     for ( i=0; i<m; i++ ) {
35544cd7ae7SLois Curfman McInnes       fprintf(fd,"row %d:",i);
35644cd7ae7SLois Curfman McInnes       for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
35744cd7ae7SLois Curfman McInnes #if defined(PETSC_COMPLEX)
358766eeae4SLois Curfman McInnes         if (imag(a->a[j]) > 0.0 && real(a->a[j]) != 0.0)
35944cd7ae7SLois Curfman McInnes           fprintf(fd," %d %g + %g i",a->j[j]+shift,real(a->a[j]),imag(a->a[j]));
360766eeae4SLois Curfman McInnes         else if (imag(a->a[j]) < 0.0 && real(a->a[j]) != 0.0)
361766eeae4SLois Curfman McInnes           fprintf(fd," %d %g - %g i",a->j[j]+shift,real(a->a[j]),-imag(a->a[j]));
36244cd7ae7SLois Curfman McInnes         else if (real(a->a[j]) != 0.0)
36344cd7ae7SLois Curfman McInnes           fprintf(fd," %d %g ",a->j[j]+shift,real(a->a[j]));
36444cd7ae7SLois Curfman McInnes #else
36544cd7ae7SLois Curfman McInnes         if (a->a[j] != 0.0) fprintf(fd," %d %g ",a->j[j]+shift,a->a[j]);
36644cd7ae7SLois Curfman McInnes #endif
36744cd7ae7SLois Curfman McInnes       }
36844cd7ae7SLois Curfman McInnes       fprintf(fd,"\n");
36944cd7ae7SLois Curfman McInnes     }
37044cd7ae7SLois Curfman McInnes   }
371*496be53dSLois Curfman McInnes   else if (format == VIEWER_FORMAT_ASCII_SYMMODU) {
372*496be53dSLois Curfman McInnes     int nzd=0, fshift=1, *sptr;
373*496be53dSLois Curfman McInnes     sptr = (int *) PetscMalloc( (m)*sizeof(int) ); CHKPTRQ(sptr);
374*496be53dSLois Curfman McInnes     for ( i=0; i<m; i++ ) {
375*496be53dSLois Curfman McInnes       sptr[i] = nzd+1;
376*496be53dSLois Curfman McInnes       for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
377*496be53dSLois Curfman McInnes         if (a->j[j] >= i) {
378*496be53dSLois Curfman McInnes #if defined(PETSC_COMPLEX)
379*496be53dSLois Curfman McInnes           if (imag(a->a[j]) != 0.0 || real(a->a[j]) != 0.0) nzd++;
380*496be53dSLois Curfman McInnes #else
381*496be53dSLois Curfman McInnes           if (a->a[j] != 0.0) nzd++;
382*496be53dSLois Curfman McInnes #endif
383*496be53dSLois Curfman McInnes         }
384*496be53dSLois Curfman McInnes       }
385*496be53dSLois Curfman McInnes     }
386*496be53dSLois Curfman McInnes     fprintf(fd," %d %d\n\n",m,nzd);
387*496be53dSLois Curfman McInnes     for ( i=0; i<m; i+=6 ) {
388*496be53dSLois Curfman McInnes       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]);
389*496be53dSLois Curfman McInnes     }
390*496be53dSLois Curfman McInnes     fprintf(fd,"\n");
391*496be53dSLois Curfman McInnes     PetscFree(sptr);
392*496be53dSLois Curfman McInnes     for ( i=0; i<m; i++ ) {
393*496be53dSLois Curfman McInnes       for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
394*496be53dSLois Curfman McInnes         if (a->j[j] >= i) fprintf(fd," %d ",a->j[j]+fshift);
395*496be53dSLois Curfman McInnes       }
396*496be53dSLois Curfman McInnes       fprintf(fd,"\n");
397*496be53dSLois Curfman McInnes     }
398*496be53dSLois Curfman McInnes     fprintf(fd,"\n");
399*496be53dSLois Curfman McInnes     for ( i=0; i<m; i++ ) {
400*496be53dSLois Curfman McInnes       for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
401*496be53dSLois Curfman McInnes         if (a->j[j] >= i) {
402*496be53dSLois Curfman McInnes #if defined(PETSC_COMPLEX)
403*496be53dSLois Curfman McInnes           if (imag(a->a[j]) != 0.0 || real(a->a[j]) != 0.0)
404*496be53dSLois Curfman McInnes             fprintf(fd," %18.16e %18.16e ",real(a->a[j]),imag(a->a[j]));
405*496be53dSLois Curfman McInnes #else
406*496be53dSLois Curfman McInnes           if (a->a[j] != 0.0) fprintf(fd," %18.16e ",a->a[j]);
407*496be53dSLois Curfman McInnes #endif
408*496be53dSLois Curfman McInnes         }
409*496be53dSLois Curfman McInnes       }
410*496be53dSLois Curfman McInnes       fprintf(fd,"\n");
411*496be53dSLois Curfman McInnes     }
412*496be53dSLois Curfman McInnes   }
41317ab2063SBarry Smith   else {
41417ab2063SBarry Smith     for ( i=0; i<m; i++ ) {
41517ab2063SBarry Smith       fprintf(fd,"row %d:",i);
416416022c9SBarry Smith       for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
41717ab2063SBarry Smith #if defined(PETSC_COMPLEX)
418766eeae4SLois Curfman McInnes         if (imag(a->a[j]) > 0.0) {
419416022c9SBarry Smith           fprintf(fd," %d %g + %g i",a->j[j]+shift,real(a->a[j]),imag(a->a[j]));
420766eeae4SLois Curfman McInnes         } else if (imag(a->a[j]) < 0.0) {
421766eeae4SLois Curfman McInnes           fprintf(fd," %d %g - %g i",a->j[j]+shift,real(a->a[j]),-imag(a->a[j]));
42217ab2063SBarry Smith         }
42317ab2063SBarry Smith         else {
424416022c9SBarry Smith           fprintf(fd," %d %g ",a->j[j]+shift,real(a->a[j]));
42517ab2063SBarry Smith         }
42617ab2063SBarry Smith #else
427416022c9SBarry Smith         fprintf(fd," %d %g ",a->j[j]+shift,a->a[j]);
42817ab2063SBarry Smith #endif
42917ab2063SBarry Smith       }
43017ab2063SBarry Smith       fprintf(fd,"\n");
43117ab2063SBarry Smith     }
43217ab2063SBarry Smith   }
43317ab2063SBarry Smith   fflush(fd);
434416022c9SBarry Smith   return 0;
435416022c9SBarry Smith }
436416022c9SBarry Smith 
4375615d1e5SSatish Balay #undef __FUNC__
4385615d1e5SSatish Balay #define __FUNC__ "MatView_SeqAIJ_Draw"
4398f6be9afSLois Curfman McInnes extern int MatView_SeqAIJ_Draw(Mat A,Viewer viewer)
440416022c9SBarry Smith {
441416022c9SBarry Smith   Mat_SeqAIJ  *a = (Mat_SeqAIJ *) A->data;
442cddf8d76SBarry Smith   int         ierr, i,j, m = a->m, shift = a->indexshift,pause,color;
4430513a670SBarry Smith   int         format;
44494a9d846SBarry Smith   double      xl,yl,xr,yr,w,h,xc,yc,scale = 1.0,x_l,x_r,y_l,y_r,maxv = 0.0;
445bcd2baecSBarry Smith   Draw        draw;
446cddf8d76SBarry Smith   DrawButton  button;
44719bcc07fSBarry Smith   PetscTruth  isnull;
448cddf8d76SBarry Smith 
4490513a670SBarry Smith   ierr = ViewerDrawGetDraw(viewer,&draw); CHKERRQ(ierr);
4500513a670SBarry Smith   ierr = DrawClear(draw); CHKERRQ(ierr);
4510513a670SBarry Smith   ierr = ViewerGetFormat(viewer,&format); CHKERRQ(ierr);
45219bcc07fSBarry Smith   ierr = DrawIsNull(draw,&isnull); CHKERRQ(ierr); if (isnull) return 0;
45319bcc07fSBarry Smith 
454416022c9SBarry Smith   xr  = a->n; yr = a->m; h = yr/10.0; w = xr/10.0;
455416022c9SBarry Smith   xr += w;    yr += h;  xl = -w;     yl = -h;
456416022c9SBarry Smith   ierr = DrawSetCoordinates(draw,xl,yl,xr,yr); CHKERRQ(ierr);
457416022c9SBarry Smith   /* loop over matrix elements drawing boxes */
4580513a670SBarry Smith 
4590513a670SBarry Smith   if (format != VIEWER_FORMAT_DRAW_CONTOUR) {
4600513a670SBarry Smith     /* Blue for negative, Cyan for zero and  Red for positive */
461cddf8d76SBarry Smith     color = DRAW_BLUE;
462416022c9SBarry Smith     for ( i=0; i<m; i++ ) {
463cddf8d76SBarry Smith       y_l = m - i - 1.0; y_r = y_l + 1.0;
464416022c9SBarry Smith       for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
465cddf8d76SBarry Smith         x_l = a->j[j] + shift; x_r = x_l + 1.0;
466cddf8d76SBarry Smith #if defined(PETSC_COMPLEX)
467cddf8d76SBarry Smith         if (real(a->a[j]) >=  0.) continue;
468cddf8d76SBarry Smith #else
469cddf8d76SBarry Smith         if (a->a[j] >=  0.) continue;
470cddf8d76SBarry Smith #endif
471cddf8d76SBarry Smith         DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);
472cddf8d76SBarry Smith       }
473cddf8d76SBarry Smith     }
474cddf8d76SBarry Smith     color = DRAW_CYAN;
475cddf8d76SBarry Smith     for ( i=0; i<m; i++ ) {
476cddf8d76SBarry Smith       y_l = m - i - 1.0; y_r = y_l + 1.0;
477cddf8d76SBarry Smith       for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
478cddf8d76SBarry Smith         x_l = a->j[j] + shift; x_r = x_l + 1.0;
479cddf8d76SBarry Smith         if (a->a[j] !=  0.) continue;
480cddf8d76SBarry Smith         DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);
481cddf8d76SBarry Smith       }
482cddf8d76SBarry Smith     }
483cddf8d76SBarry Smith     color = DRAW_RED;
484cddf8d76SBarry Smith     for ( i=0; i<m; i++ ) {
485cddf8d76SBarry Smith       y_l = m - i - 1.0; y_r = y_l + 1.0;
486cddf8d76SBarry Smith       for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
487cddf8d76SBarry Smith         x_l = a->j[j] + shift; x_r = x_l + 1.0;
488cddf8d76SBarry Smith #if defined(PETSC_COMPLEX)
489cddf8d76SBarry Smith         if (real(a->a[j]) <=  0.) continue;
490cddf8d76SBarry Smith #else
491cddf8d76SBarry Smith         if (a->a[j] <=  0.) continue;
492cddf8d76SBarry Smith #endif
493cddf8d76SBarry Smith         DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);
494416022c9SBarry Smith       }
495416022c9SBarry Smith     }
4960513a670SBarry Smith   } else {
4970513a670SBarry Smith     /* use contour shading to indicate magnitude of values */
4980513a670SBarry Smith     /* first determine max of all nonzero values */
4990513a670SBarry Smith     int    nz = a->nz,count;
5000513a670SBarry Smith     Draw   popup;
5010513a670SBarry Smith 
5020513a670SBarry Smith     for ( i=0; i<nz; i++ ) {
5030513a670SBarry Smith       if (PetscAbsScalar(a->a[i]) > maxv) maxv = PetscAbsScalar(a->a[i]);
5040513a670SBarry Smith     }
5050513a670SBarry Smith     ierr = DrawCreatePopUp(draw,&popup); CHKERRQ(ierr);
5060513a670SBarry Smith     ierr = DrawScalePopup(popup,0.0,maxv); CHKERRQ(ierr);
5070513a670SBarry Smith     count = 0;
5080513a670SBarry Smith     for ( i=0; i<m; i++ ) {
5090513a670SBarry Smith       y_l = m - i - 1.0; y_r = y_l + 1.0;
5100513a670SBarry Smith       for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
5110513a670SBarry Smith         x_l = a->j[j] + shift; x_r = x_l + 1.0;
5120513a670SBarry Smith         color = 32 + (int) ((200.0 - 32.0)*PetscAbsScalar(a->a[count])/maxv);
5130513a670SBarry Smith         DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);
5140513a670SBarry Smith         count++;
5150513a670SBarry Smith       }
5160513a670SBarry Smith     }
5170513a670SBarry Smith   }
518416022c9SBarry Smith   DrawFlush(draw);
519cddf8d76SBarry Smith   DrawGetPause(draw,&pause);
520cddf8d76SBarry Smith   if (pause >= 0) { PetscSleep(pause); return 0;}
521cddf8d76SBarry Smith 
522cddf8d76SBarry Smith   /* allow the matrix to zoom or shrink */
5236945ee14SBarry Smith   ierr = DrawCheckResizedWindow(draw);
524cddf8d76SBarry Smith   ierr = DrawGetMouseButton(draw,&button,&xc,&yc,0,0);
525cddf8d76SBarry Smith   while (button != BUTTON_RIGHT) {
526cddf8d76SBarry Smith     DrawClear(draw);
527cddf8d76SBarry Smith     if (button == BUTTON_LEFT) scale = .5;
528cddf8d76SBarry Smith     else if (button == BUTTON_CENTER) scale = 2.;
529cddf8d76SBarry Smith     xl = scale*(xl + w - xc) + xc - w*scale;
530cddf8d76SBarry Smith     xr = scale*(xr - w - xc) + xc + w*scale;
531cddf8d76SBarry Smith     yl = scale*(yl + h - yc) + yc - h*scale;
532cddf8d76SBarry Smith     yr = scale*(yr - h - yc) + yc + h*scale;
533cddf8d76SBarry Smith     w *= scale; h *= scale;
534cddf8d76SBarry Smith     ierr = DrawSetCoordinates(draw,xl,yl,xr,yr); CHKERRQ(ierr);
5350513a670SBarry Smith     if (format != VIEWER_FORMAT_DRAW_CONTOUR) {
5360513a670SBarry Smith       /* Blue for negative, Cyan for zero and  Red for positive */
537cddf8d76SBarry Smith       color = DRAW_BLUE;
538cddf8d76SBarry Smith       for ( i=0; i<m; i++ ) {
539cddf8d76SBarry Smith         y_l = m - i - 1.0; y_r = y_l + 1.0;
540cddf8d76SBarry Smith         for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
541cddf8d76SBarry Smith           x_l = a->j[j] + shift; x_r = x_l + 1.0;
542cddf8d76SBarry Smith #if defined(PETSC_COMPLEX)
543cddf8d76SBarry Smith           if (real(a->a[j]) >=  0.) continue;
544cddf8d76SBarry Smith #else
545cddf8d76SBarry Smith           if (a->a[j] >=  0.) continue;
546cddf8d76SBarry Smith #endif
547cddf8d76SBarry Smith           DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);
548cddf8d76SBarry Smith         }
549cddf8d76SBarry Smith       }
550cddf8d76SBarry Smith       color = DRAW_CYAN;
551cddf8d76SBarry Smith       for ( i=0; i<m; i++ ) {
552cddf8d76SBarry Smith         y_l = m - i - 1.0; y_r = y_l + 1.0;
553cddf8d76SBarry Smith         for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
554cddf8d76SBarry Smith           x_l = a->j[j] + shift; x_r = x_l + 1.0;
555cddf8d76SBarry Smith           if (a->a[j] !=  0.) continue;
556cddf8d76SBarry Smith           DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);
557cddf8d76SBarry Smith         }
558cddf8d76SBarry Smith       }
559cddf8d76SBarry Smith       color = DRAW_RED;
560cddf8d76SBarry Smith       for ( i=0; i<m; i++ ) {
561cddf8d76SBarry Smith         y_l = m - i - 1.0; y_r = y_l + 1.0;
562cddf8d76SBarry Smith         for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
563cddf8d76SBarry Smith           x_l = a->j[j] + shift; x_r = x_l + 1.0;
564cddf8d76SBarry Smith #if defined(PETSC_COMPLEX)
565cddf8d76SBarry Smith           if (real(a->a[j]) <=  0.) continue;
566cddf8d76SBarry Smith #else
567cddf8d76SBarry Smith           if (a->a[j] <=  0.) continue;
568cddf8d76SBarry Smith #endif
569cddf8d76SBarry Smith           DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);
570cddf8d76SBarry Smith         }
571cddf8d76SBarry Smith       }
5720513a670SBarry Smith     } else {
5730513a670SBarry Smith       /* use contour shading to indicate magnitude of values */
5740513a670SBarry Smith       int count = 0;
5750513a670SBarry Smith       for ( i=0; i<m; i++ ) {
5760513a670SBarry Smith         y_l = m - i - 1.0; y_r = y_l + 1.0;
5770513a670SBarry Smith         for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
5780513a670SBarry Smith           x_l = a->j[j] + shift; x_r = x_l + 1.0;
5790513a670SBarry Smith           color = 32 + (int) ((200.0 - 32.0)*PetscAbsScalar(a->a[count])/maxv);
5800513a670SBarry Smith           DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);
5810513a670SBarry Smith           count++;
5820513a670SBarry Smith         }
5830513a670SBarry Smith       }
5840513a670SBarry Smith     }
5850513a670SBarry Smith 
5866945ee14SBarry Smith     ierr = DrawCheckResizedWindow(draw);
587cddf8d76SBarry Smith     ierr = DrawGetMouseButton(draw,&button,&xc,&yc,0,0);
588cddf8d76SBarry Smith   }
589416022c9SBarry Smith   return 0;
590416022c9SBarry Smith }
591416022c9SBarry Smith 
5925615d1e5SSatish Balay #undef __FUNC__
593c22c1629SBarry Smith #define __FUNC__ "MatView_SeqAIJ" /* ADIC Ignore */
5948f6be9afSLois Curfman McInnes int MatView_SeqAIJ(PetscObject obj,Viewer viewer)
595416022c9SBarry Smith {
596416022c9SBarry Smith   Mat         A = (Mat) obj;
597416022c9SBarry Smith   Mat_SeqAIJ  *a = (Mat_SeqAIJ*) A->data;
598bcd2baecSBarry Smith   ViewerType  vtype;
599bcd2baecSBarry Smith   int         ierr;
600416022c9SBarry Smith 
601bcd2baecSBarry Smith   ierr = ViewerGetType(viewer,&vtype); CHKERRQ(ierr);
602bcd2baecSBarry Smith   if (vtype == MATLAB_VIEWER) {
603416022c9SBarry Smith     return ViewerMatlabPutSparse_Private(viewer,a->m,a->n,a->nz,a->a,a->i,a->j);
604416022c9SBarry Smith   }
605bcd2baecSBarry Smith   else if (vtype == ASCII_FILE_VIEWER || vtype == ASCII_FILES_VIEWER){
606416022c9SBarry Smith     return MatView_SeqAIJ_ASCII(A,viewer);
607416022c9SBarry Smith   }
608bcd2baecSBarry Smith   else if (vtype == BINARY_FILE_VIEWER) {
609416022c9SBarry Smith     return MatView_SeqAIJ_Binary(A,viewer);
610416022c9SBarry Smith   }
611bcd2baecSBarry Smith   else if (vtype == DRAW_VIEWER) {
612bcd2baecSBarry Smith     return MatView_SeqAIJ_Draw(A,viewer);
61317ab2063SBarry Smith   }
61417ab2063SBarry Smith   return 0;
61517ab2063SBarry Smith }
61619bcc07fSBarry Smith 
617c456f294SBarry Smith extern int Mat_AIJ_CheckInode(Mat);
6185615d1e5SSatish Balay #undef __FUNC__
6195615d1e5SSatish Balay #define __FUNC__ "MatAssemblyEnd_SeqAIJ"
6208f6be9afSLois Curfman McInnes int MatAssemblyEnd_SeqAIJ(Mat A,MatAssemblyType mode)
62117ab2063SBarry Smith {
622416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
62341c01911SSatish Balay   int        fshift = 0,i,j,*ai = a->i, *aj = a->j, *imax = a->imax,ierr;
62494a9d846SBarry Smith   int        m = a->m, *ip, N, *ailen = a->ilen,shift = a->indexshift,rmax;
625416022c9SBarry Smith   Scalar     *aa = a->a, *ap;
62617ab2063SBarry Smith 
6276d4a8577SBarry Smith   if (mode == MAT_FLUSH_ASSEMBLY) return 0;
62817ab2063SBarry Smith 
62994a9d846SBarry Smith   rmax = ailen[0]; /* determine row with most nonzeros */
63017ab2063SBarry Smith   for ( i=1; i<m; i++ ) {
631416022c9SBarry Smith     /* move each row back by the amount of empty slots (fshift) before it*/
63217ab2063SBarry Smith     fshift += imax[i-1] - ailen[i-1];
63394a9d846SBarry Smith     rmax   = PetscMax(rmax,ailen[i]);
63417ab2063SBarry Smith     if (fshift) {
635416022c9SBarry Smith       ip = aj + ai[i] + shift; ap = aa + ai[i] + shift;
63617ab2063SBarry Smith       N = ailen[i];
63717ab2063SBarry Smith       for ( j=0; j<N; j++ ) {
63817ab2063SBarry Smith         ip[j-fshift] = ip[j];
63917ab2063SBarry Smith         ap[j-fshift] = ap[j];
64017ab2063SBarry Smith       }
64117ab2063SBarry Smith     }
64217ab2063SBarry Smith     ai[i] = ai[i-1] + ailen[i-1];
64317ab2063SBarry Smith   }
64417ab2063SBarry Smith   if (m) {
64517ab2063SBarry Smith     fshift += imax[m-1] - ailen[m-1];
64617ab2063SBarry Smith     ai[m] = ai[m-1] + ailen[m-1];
64717ab2063SBarry Smith   }
64817ab2063SBarry Smith   /* reset ilen and imax for each row */
64917ab2063SBarry Smith   for ( i=0; i<m; i++ ) {
65017ab2063SBarry Smith     ailen[i] = imax[i] = ai[i+1] - ai[i];
65117ab2063SBarry Smith   }
652416022c9SBarry Smith   a->nz = ai[m] + shift;
65317ab2063SBarry Smith 
65417ab2063SBarry Smith   /* diagonals may have moved, so kill the diagonal pointers */
655416022c9SBarry Smith   if (fshift && a->diag) {
6560452661fSBarry Smith     PetscFree(a->diag);
657416022c9SBarry Smith     PLogObjectMemory(A,-(m+1)*sizeof(int));
658416022c9SBarry Smith     a->diag = 0;
65917ab2063SBarry Smith   }
6604e220ebcSLois Curfman McInnes   PLogInfo(A,"MatAssemblyEnd_SeqAIJ:Matrix size: %d X %d; storage space: %d unneeded, %d used\n",
6614e220ebcSLois Curfman McInnes            m,a->n,fshift,a->nz);
6624e220ebcSLois Curfman McInnes   PLogInfo(A,"MatAssemblyEnd_SeqAIJ:Number of mallocs during MatSetValues is %d\n",
663b810aeb4SBarry Smith            a->reallocs);
66494a9d846SBarry Smith   PLogInfo(A,"MatAssemblyEnd_SeqAIJ:Most nonzeros in any row is %d\n",rmax);
665dd5f02e7SSatish Balay   a->reallocs          = 0;
6664e220ebcSLois Curfman McInnes   A->info.nz_unneeded  = (double)fshift;
6674e220ebcSLois Curfman McInnes 
66876dd722bSSatish Balay   /* check out for identical nodes. If found, use inode functions */
66941c01911SSatish Balay   ierr = Mat_AIJ_CheckInode(A); CHKERRQ(ierr);
67017ab2063SBarry Smith   return 0;
67117ab2063SBarry Smith }
67217ab2063SBarry Smith 
6735615d1e5SSatish Balay #undef __FUNC__
6745615d1e5SSatish Balay #define __FUNC__ "MatZeroEntries_SeqAIJ"
6758f6be9afSLois Curfman McInnes int MatZeroEntries_SeqAIJ(Mat A)
67617ab2063SBarry Smith {
677416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
678cddf8d76SBarry Smith   PetscMemzero(a->a,(a->i[a->m]+a->indexshift)*sizeof(Scalar));
67917ab2063SBarry Smith   return 0;
68017ab2063SBarry Smith }
681416022c9SBarry Smith 
6825615d1e5SSatish Balay #undef __FUNC__
6835615d1e5SSatish Balay #define __FUNC__ "MatDestroy_SeqAIJ"
68417ab2063SBarry Smith int MatDestroy_SeqAIJ(PetscObject obj)
68517ab2063SBarry Smith {
686416022c9SBarry Smith   Mat        A  = (Mat) obj;
687416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
688d5d45c9bSBarry Smith 
68917ab2063SBarry Smith #if defined(PETSC_LOG)
690416022c9SBarry Smith   PLogObjectState(obj,"Rows=%d, Cols=%d, NZ=%d",a->m,a->n,a->nz);
69117ab2063SBarry Smith #endif
6920452661fSBarry Smith   PetscFree(a->a);
6930452661fSBarry Smith   if (!a->singlemalloc) { PetscFree(a->i); PetscFree(a->j);}
6940452661fSBarry Smith   if (a->diag) PetscFree(a->diag);
6950452661fSBarry Smith   if (a->ilen) PetscFree(a->ilen);
6960452661fSBarry Smith   if (a->imax) PetscFree(a->imax);
6970452661fSBarry Smith   if (a->solve_work) PetscFree(a->solve_work);
69876dd722bSSatish Balay   if (a->inode.size) PetscFree(a->inode.size);
6990452661fSBarry Smith   PetscFree(a);
700eed86810SBarry Smith 
701f2655603SLois Curfman McInnes   PLogObjectDestroy(A);
702f2655603SLois Curfman McInnes   PetscHeaderDestroy(A);
70317ab2063SBarry Smith   return 0;
70417ab2063SBarry Smith }
70517ab2063SBarry Smith 
7065615d1e5SSatish Balay #undef __FUNC__
7075615d1e5SSatish Balay #define __FUNC__ "MatCompress_SeqAIJ"
7088f6be9afSLois Curfman McInnes int MatCompress_SeqAIJ(Mat A)
70917ab2063SBarry Smith {
71017ab2063SBarry Smith   return 0;
71117ab2063SBarry Smith }
71217ab2063SBarry Smith 
7135615d1e5SSatish Balay #undef __FUNC__
7145615d1e5SSatish Balay #define __FUNC__ "MatSetOption_SeqAIJ"
7158f6be9afSLois Curfman McInnes int MatSetOption_SeqAIJ(Mat A,MatOption op)
71617ab2063SBarry Smith {
717416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
7186d4a8577SBarry Smith   if      (op == MAT_ROW_ORIENTED)                 a->roworiented = 1;
7196d4a8577SBarry Smith   else if (op == MAT_COLUMN_ORIENTED)              a->roworiented = 0;
7206d4a8577SBarry Smith   else if (op == MAT_COLUMNS_SORTED)               a->sorted      = 1;
721219d9a1aSLois Curfman McInnes   else if (op == MAT_COLUMNS_UNSORTED)             a->sorted      = 0;
7226d4a8577SBarry Smith   else if (op == MAT_NO_NEW_NONZERO_LOCATIONS)     a->nonew       = 1;
723c2653b3dSLois Curfman McInnes   else if (op == MAT_NEW_NONZERO_LOCATION_ERROR)   a->nonew       = -1;
72496854ed6SLois Curfman McInnes   else if (op == MAT_NEW_NONZERO_ALLOCATION_ERROR) a->nonew       = -2;
7256d4a8577SBarry Smith   else if (op == MAT_YES_NEW_NONZERO_LOCATIONS)    a->nonew       = 0;
7266d4a8577SBarry Smith   else if (op == MAT_ROWS_SORTED ||
727219d9a1aSLois Curfman McInnes            op == MAT_ROWS_UNSORTED ||
7286d4a8577SBarry Smith            op == MAT_SYMMETRIC ||
7296d4a8577SBarry Smith            op == MAT_STRUCTURALLY_SYMMETRIC ||
73090f02eecSBarry Smith            op == MAT_YES_NEW_DIAGONALS ||
7312b362799SSatish Balay            op == MAT_IGNORE_OFF_PROC_ENTRIES)
73294a424c1SBarry Smith     PLogInfo(A,"Info:MatSetOption_SeqAIJ:Option ignored\n");
7336d4a8577SBarry Smith   else if (op == MAT_NO_NEW_DIAGONALS)
734e3372554SBarry Smith     {SETERRQ(PETSC_ERR_SUP,0,"MAT_NO_NEW_DIAGONALS");}
7356d4a8577SBarry Smith   else if (op == MAT_INODE_LIMIT_1)            a->inode.limit  = 1;
7366d4a8577SBarry Smith   else if (op == MAT_INODE_LIMIT_2)            a->inode.limit  = 2;
7376d4a8577SBarry Smith   else if (op == MAT_INODE_LIMIT_3)            a->inode.limit  = 3;
7386d4a8577SBarry Smith   else if (op == MAT_INODE_LIMIT_4)            a->inode.limit  = 4;
7396d4a8577SBarry Smith   else if (op == MAT_INODE_LIMIT_5)            a->inode.limit  = 5;
740e2f28af5SBarry Smith   else
741e3372554SBarry Smith     {SETERRQ(PETSC_ERR_SUP,0,"unknown option");}
74217ab2063SBarry Smith   return 0;
74317ab2063SBarry Smith }
74417ab2063SBarry Smith 
7455615d1e5SSatish Balay #undef __FUNC__
7465615d1e5SSatish Balay #define __FUNC__ "MatGetDiagonal_SeqAIJ"
7478f6be9afSLois Curfman McInnes int MatGetDiagonal_SeqAIJ(Mat A,Vec v)
74817ab2063SBarry Smith {
749416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
750416022c9SBarry Smith   int        i,j, n,shift = a->indexshift;
75117ab2063SBarry Smith   Scalar     *x, zero = 0.0;
75217ab2063SBarry Smith 
75317ab2063SBarry Smith   VecSet(&zero,v);
75490f02eecSBarry Smith   VecGetArray_Fast(v,x); VecGetLocalSize(v,&n);
755e3372554SBarry Smith   if (n != a->m) SETERRQ(1,0,"Nonconforming matrix and vector");
756416022c9SBarry Smith   for ( i=0; i<a->m; i++ ) {
757416022c9SBarry Smith     for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
758416022c9SBarry Smith       if (a->j[j]+shift == i) {
759416022c9SBarry Smith         x[i] = a->a[j];
76017ab2063SBarry Smith         break;
76117ab2063SBarry Smith       }
76217ab2063SBarry Smith     }
76317ab2063SBarry Smith   }
76417ab2063SBarry Smith   return 0;
76517ab2063SBarry Smith }
76617ab2063SBarry Smith 
76717ab2063SBarry Smith /* -------------------------------------------------------*/
76817ab2063SBarry Smith /* Should check that shapes of vectors and matrices match */
76917ab2063SBarry Smith /* -------------------------------------------------------*/
7705615d1e5SSatish Balay #undef __FUNC__
7715615d1e5SSatish Balay #define __FUNC__ "MatMultTrans_SeqAIJ"
77244cd7ae7SLois Curfman McInnes int MatMultTrans_SeqAIJ(Mat A,Vec xx,Vec yy)
77317ab2063SBarry Smith {
774416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
77517ab2063SBarry Smith   Scalar     *x, *y, *v, alpha;
776416022c9SBarry Smith   int        m = a->m, n, i, *idx, shift = a->indexshift;
77717ab2063SBarry Smith 
77890f02eecSBarry Smith   VecGetArray_Fast(xx,x); VecGetArray_Fast(yy,y);
779cddf8d76SBarry Smith   PetscMemzero(y,a->n*sizeof(Scalar));
78017ab2063SBarry Smith   y = y + shift; /* shift for Fortran start by 1 indexing */
78117ab2063SBarry Smith   for ( i=0; i<m; i++ ) {
782416022c9SBarry Smith     idx   = a->j + a->i[i] + shift;
783416022c9SBarry Smith     v     = a->a + a->i[i] + shift;
784416022c9SBarry Smith     n     = a->i[i+1] - a->i[i];
78517ab2063SBarry Smith     alpha = x[i];
78617ab2063SBarry Smith     while (n-->0) {y[*idx++] += alpha * *v++;}
78717ab2063SBarry Smith   }
788416022c9SBarry Smith   PLogFlops(2*a->nz - a->n);
78917ab2063SBarry Smith   return 0;
79017ab2063SBarry Smith }
791d5d45c9bSBarry Smith 
7925615d1e5SSatish Balay #undef __FUNC__
7935615d1e5SSatish Balay #define __FUNC__ "MatMultTransAdd_SeqAIJ"
79444cd7ae7SLois Curfman McInnes int MatMultTransAdd_SeqAIJ(Mat A,Vec xx,Vec zz,Vec yy)
79517ab2063SBarry Smith {
796416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
79717ab2063SBarry Smith   Scalar     *x, *y, *v, alpha;
798416022c9SBarry Smith   int        m = a->m, n, i, *idx,shift = a->indexshift;
79917ab2063SBarry Smith 
80090f02eecSBarry Smith   VecGetArray_Fast(xx,x); VecGetArray_Fast(yy,y);
80117ab2063SBarry Smith   if (zz != yy) VecCopy(zz,yy);
80217ab2063SBarry Smith   y = y + shift; /* shift for Fortran start by 1 indexing */
80317ab2063SBarry Smith   for ( i=0; i<m; i++ ) {
804416022c9SBarry Smith     idx   = a->j + a->i[i] + shift;
805416022c9SBarry Smith     v     = a->a + a->i[i] + shift;
806416022c9SBarry Smith     n     = a->i[i+1] - a->i[i];
80717ab2063SBarry Smith     alpha = x[i];
80817ab2063SBarry Smith     while (n-->0) {y[*idx++] += alpha * *v++;}
80917ab2063SBarry Smith   }
81090f02eecSBarry Smith   PLogFlops(2*a->nz);
81117ab2063SBarry Smith   return 0;
81217ab2063SBarry Smith }
81317ab2063SBarry Smith 
8145615d1e5SSatish Balay #undef __FUNC__
8155615d1e5SSatish Balay #define __FUNC__ "MatMult_SeqAIJ"
81644cd7ae7SLois Curfman McInnes int MatMult_SeqAIJ(Mat A,Vec xx,Vec yy)
81717ab2063SBarry Smith {
818416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
81917ab2063SBarry Smith   Scalar     *x, *y, *v, sum;
8209ea0dfa2SSatish Balay   int        m = a->m, n, i, *idx, shift = a->indexshift,*ii,jrow,j;
82117ab2063SBarry Smith 
82290f02eecSBarry Smith   VecGetArray_Fast(xx,x); VecGetArray_Fast(yy,y);
82317ab2063SBarry Smith   x    = x + shift; /* shift for Fortran start by 1 indexing */
824416022c9SBarry Smith   idx  = a->j;
825416022c9SBarry Smith   v    = a->a;
826416022c9SBarry Smith   ii   = a->i;
82717ab2063SBarry Smith   for ( i=0; i<m; i++ ) {
8289ea0dfa2SSatish Balay     jrow = ii[i];
8299ea0dfa2SSatish Balay     n    = ii[i+1] - jrow;
83017ab2063SBarry Smith     sum  = 0.0;
8319ea0dfa2SSatish Balay     /* while (n--) sum += *v++ * x[*idx++]; */
8329ea0dfa2SSatish Balay     for ( j=0; j<n; j++) {
8339ea0dfa2SSatish Balay       sum += v[jrow]*x[idx[jrow]]; jrow++;
8349ea0dfa2SSatish Balay      }
83517ab2063SBarry Smith     y[i] = sum;
83617ab2063SBarry Smith   }
837416022c9SBarry Smith   PLogFlops(2*a->nz - m);
83817ab2063SBarry Smith   return 0;
83917ab2063SBarry Smith }
84017ab2063SBarry Smith 
8415615d1e5SSatish Balay #undef __FUNC__
8425615d1e5SSatish Balay #define __FUNC__ "MatMultAdd_SeqAIJ"
84344cd7ae7SLois Curfman McInnes int MatMultAdd_SeqAIJ(Mat A,Vec xx,Vec yy,Vec zz)
84417ab2063SBarry Smith {
845416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
84617ab2063SBarry Smith   Scalar     *x, *y, *z, *v, sum;
8479ea0dfa2SSatish Balay   int        m = a->m, n, i, *idx, shift = a->indexshift,*ii,jrow,j;
8489ea0dfa2SSatish Balay 
84917ab2063SBarry Smith 
85090f02eecSBarry Smith   VecGetArray_Fast(xx,x); VecGetArray_Fast(yy,y); VecGetArray_Fast(zz,z);
85117ab2063SBarry Smith   x    = x + shift; /* shift for Fortran start by 1 indexing */
852cddf8d76SBarry Smith   idx  = a->j;
853cddf8d76SBarry Smith   v    = a->a;
854cddf8d76SBarry Smith   ii   = a->i;
85517ab2063SBarry Smith   for ( i=0; i<m; i++ ) {
8569ea0dfa2SSatish Balay     jrow = ii[i];
8579ea0dfa2SSatish Balay     n    = ii[i+1] - jrow;
85817ab2063SBarry Smith     sum  = y[i];
8599ea0dfa2SSatish Balay     /* while (n--) sum += *v++ * x[*idx++]; */
8609ea0dfa2SSatish Balay     for ( j=0; j<n; j++) {
8619ea0dfa2SSatish Balay       sum += v[jrow]*x[idx[jrow]]; jrow++;
8629ea0dfa2SSatish Balay      }
86317ab2063SBarry Smith     z[i] = sum;
86417ab2063SBarry Smith   }
865416022c9SBarry Smith   PLogFlops(2*a->nz);
86617ab2063SBarry Smith   return 0;
86717ab2063SBarry Smith }
86817ab2063SBarry Smith 
86917ab2063SBarry Smith /*
87017ab2063SBarry Smith      Adds diagonal pointers to sparse matrix structure.
87117ab2063SBarry Smith */
87217ab2063SBarry Smith 
8735615d1e5SSatish Balay #undef __FUNC__
8745615d1e5SSatish Balay #define __FUNC__ "MatMarkDiag_SeqAIJ"
875416022c9SBarry Smith int MatMarkDiag_SeqAIJ(Mat A)
87617ab2063SBarry Smith {
877416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
878416022c9SBarry Smith   int        i,j, *diag, m = a->m,shift = a->indexshift;
87917ab2063SBarry Smith 
8800452661fSBarry Smith   diag = (int *) PetscMalloc( (m+1)*sizeof(int)); CHKPTRQ(diag);
881416022c9SBarry Smith   PLogObjectMemory(A,(m+1)*sizeof(int));
882416022c9SBarry Smith   for ( i=0; i<a->m; i++ ) {
883416022c9SBarry Smith     for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
884416022c9SBarry Smith       if (a->j[j]+shift == i) {
88517ab2063SBarry Smith         diag[i] = j - shift;
88617ab2063SBarry Smith         break;
88717ab2063SBarry Smith       }
88817ab2063SBarry Smith     }
88917ab2063SBarry Smith   }
890416022c9SBarry Smith   a->diag = diag;
89117ab2063SBarry Smith   return 0;
89217ab2063SBarry Smith }
89317ab2063SBarry Smith 
8945615d1e5SSatish Balay #undef __FUNC__
8955615d1e5SSatish Balay #define __FUNC__ "MatRelax_SeqAIJ"
89644cd7ae7SLois Curfman McInnes int MatRelax_SeqAIJ(Mat A,Vec bb,double omega,MatSORType flag,
89717ab2063SBarry Smith                            double fshift,int its,Vec xx)
89817ab2063SBarry Smith {
899416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
900416022c9SBarry Smith   Scalar     *x, *b, *bs,  d, *xs, sum, *v = a->a,*t,scale,*ts, *xb;
901d5d45c9bSBarry Smith   int        ierr, *idx, *diag,n = a->n, m = a->m, i, shift = a->indexshift;
90217ab2063SBarry Smith 
90390f02eecSBarry Smith   VecGetArray_Fast(xx,x); VecGetArray_Fast(bb,b);
904416022c9SBarry Smith   if (!a->diag) {if ((ierr = MatMarkDiag_SeqAIJ(A))) return ierr;}
905416022c9SBarry Smith   diag = a->diag;
906416022c9SBarry Smith   xs   = x + shift; /* shifted by one for index start of a or a->j*/
90717ab2063SBarry Smith   if (flag == SOR_APPLY_UPPER) {
90817ab2063SBarry Smith    /* apply ( U + D/omega) to the vector */
90917ab2063SBarry Smith     bs = b + shift;
91017ab2063SBarry Smith     for ( i=0; i<m; i++ ) {
911416022c9SBarry Smith         d    = fshift + a->a[diag[i] + shift];
912416022c9SBarry Smith         n    = a->i[i+1] - diag[i] - 1;
913416022c9SBarry Smith         idx  = a->j + diag[i] + (!shift);
914416022c9SBarry Smith         v    = a->a + diag[i] + (!shift);
91517ab2063SBarry Smith         sum  = b[i]*d/omega;
91617ab2063SBarry Smith         SPARSEDENSEDOT(sum,bs,v,idx,n);
91717ab2063SBarry Smith         x[i] = sum;
91817ab2063SBarry Smith     }
91917ab2063SBarry Smith     return 0;
92017ab2063SBarry Smith   }
92117ab2063SBarry Smith   if (flag == SOR_APPLY_LOWER) {
922e3372554SBarry Smith     SETERRQ(1,0,"SOR_APPLY_LOWER is not done");
92317ab2063SBarry Smith   }
924416022c9SBarry Smith   else if (flag & SOR_EISENSTAT) {
92517ab2063SBarry Smith     /* Let  A = L + U + D; where L is lower trianglar,
92617ab2063SBarry Smith     U is upper triangular, E is diagonal; This routine applies
92717ab2063SBarry Smith 
92817ab2063SBarry Smith             (L + E)^{-1} A (U + E)^{-1}
92917ab2063SBarry Smith 
93017ab2063SBarry Smith     to a vector efficiently using Eisenstat's trick. This is for
93117ab2063SBarry Smith     the case of SSOR preconditioner, so E is D/omega where omega
93217ab2063SBarry Smith     is the relaxation factor.
93317ab2063SBarry Smith     */
9340452661fSBarry Smith     t = (Scalar *) PetscMalloc( m*sizeof(Scalar) ); CHKPTRQ(t);
93517ab2063SBarry Smith     scale = (2.0/omega) - 1.0;
93617ab2063SBarry Smith 
93717ab2063SBarry Smith     /*  x = (E + U)^{-1} b */
93817ab2063SBarry Smith     for ( i=m-1; i>=0; i-- ) {
939416022c9SBarry Smith       d    = fshift + a->a[diag[i] + shift];
940416022c9SBarry Smith       n    = a->i[i+1] - diag[i] - 1;
941416022c9SBarry Smith       idx  = a->j + diag[i] + (!shift);
942416022c9SBarry Smith       v    = a->a + diag[i] + (!shift);
94317ab2063SBarry Smith       sum  = b[i];
94417ab2063SBarry Smith       SPARSEDENSEMDOT(sum,xs,v,idx,n);
94517ab2063SBarry Smith       x[i] = omega*(sum/d);
94617ab2063SBarry Smith     }
94717ab2063SBarry Smith 
94817ab2063SBarry Smith     /*  t = b - (2*E - D)x */
949416022c9SBarry Smith     v = a->a;
95017ab2063SBarry Smith     for ( i=0; i<m; i++ ) { t[i] = b[i] - scale*(v[*diag++ + shift])*x[i]; }
95117ab2063SBarry Smith 
95217ab2063SBarry Smith     /*  t = (E + L)^{-1}t */
953416022c9SBarry Smith     ts = t + shift; /* shifted by one for index start of a or a->j*/
954416022c9SBarry Smith     diag = a->diag;
95517ab2063SBarry Smith     for ( i=0; i<m; i++ ) {
956416022c9SBarry Smith       d    = fshift + a->a[diag[i]+shift];
957416022c9SBarry Smith       n    = diag[i] - a->i[i];
958416022c9SBarry Smith       idx  = a->j + a->i[i] + shift;
959416022c9SBarry Smith       v    = a->a + a->i[i] + shift;
96017ab2063SBarry Smith       sum  = t[i];
96117ab2063SBarry Smith       SPARSEDENSEMDOT(sum,ts,v,idx,n);
96217ab2063SBarry Smith       t[i] = omega*(sum/d);
96317ab2063SBarry Smith     }
96417ab2063SBarry Smith 
96517ab2063SBarry Smith     /*  x = x + t */
96617ab2063SBarry Smith     for ( i=0; i<m; i++ ) { x[i] += t[i]; }
9670452661fSBarry Smith     PetscFree(t);
96817ab2063SBarry Smith     return 0;
96917ab2063SBarry Smith   }
97017ab2063SBarry Smith   if (flag & SOR_ZERO_INITIAL_GUESS) {
97117ab2063SBarry Smith     if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP){
97217ab2063SBarry Smith       for ( i=0; i<m; i++ ) {
973416022c9SBarry Smith         d    = fshift + a->a[diag[i]+shift];
974416022c9SBarry Smith         n    = diag[i] - a->i[i];
975416022c9SBarry Smith         idx  = a->j + a->i[i] + shift;
976416022c9SBarry Smith         v    = a->a + a->i[i] + shift;
97717ab2063SBarry Smith         sum  = b[i];
97817ab2063SBarry Smith         SPARSEDENSEMDOT(sum,xs,v,idx,n);
97917ab2063SBarry Smith         x[i] = omega*(sum/d);
98017ab2063SBarry Smith       }
98117ab2063SBarry Smith       xb = x;
98217ab2063SBarry Smith     }
98317ab2063SBarry Smith     else xb = b;
98417ab2063SBarry Smith     if ((flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP) &&
98517ab2063SBarry Smith         (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP)) {
98617ab2063SBarry Smith       for ( i=0; i<m; i++ ) {
987416022c9SBarry Smith         x[i] *= a->a[diag[i]+shift];
98817ab2063SBarry Smith       }
98917ab2063SBarry Smith     }
99017ab2063SBarry Smith     if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP){
99117ab2063SBarry Smith       for ( i=m-1; i>=0; i-- ) {
992416022c9SBarry Smith         d    = fshift + a->a[diag[i] + shift];
993416022c9SBarry Smith         n    = a->i[i+1] - diag[i] - 1;
994416022c9SBarry Smith         idx  = a->j + diag[i] + (!shift);
995416022c9SBarry Smith         v    = a->a + diag[i] + (!shift);
99617ab2063SBarry Smith         sum  = xb[i];
99717ab2063SBarry Smith         SPARSEDENSEMDOT(sum,xs,v,idx,n);
99817ab2063SBarry Smith         x[i] = omega*(sum/d);
99917ab2063SBarry Smith       }
100017ab2063SBarry Smith     }
100117ab2063SBarry Smith     its--;
100217ab2063SBarry Smith   }
100317ab2063SBarry Smith   while (its--) {
100417ab2063SBarry Smith     if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP){
100517ab2063SBarry Smith       for ( i=0; i<m; i++ ) {
1006416022c9SBarry Smith         d    = fshift + a->a[diag[i]+shift];
1007416022c9SBarry Smith         n    = a->i[i+1] - a->i[i];
1008416022c9SBarry Smith         idx  = a->j + a->i[i] + shift;
1009416022c9SBarry Smith         v    = a->a + a->i[i] + shift;
101017ab2063SBarry Smith         sum  = b[i];
101117ab2063SBarry Smith         SPARSEDENSEMDOT(sum,xs,v,idx,n);
10127e33a6baSBarry Smith         x[i] = (1. - omega)*x[i] + omega*(sum + a->a[diag[i]+shift]*x[i])/d;
101317ab2063SBarry Smith       }
101417ab2063SBarry Smith     }
101517ab2063SBarry Smith     if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP){
101617ab2063SBarry Smith       for ( i=m-1; i>=0; i-- ) {
1017416022c9SBarry Smith         d    = fshift + a->a[diag[i] + shift];
1018416022c9SBarry Smith         n    = a->i[i+1] - a->i[i];
1019416022c9SBarry Smith         idx  = a->j + a->i[i] + shift;
1020416022c9SBarry Smith         v    = a->a + a->i[i] + shift;
102117ab2063SBarry Smith         sum  = b[i];
102217ab2063SBarry Smith         SPARSEDENSEMDOT(sum,xs,v,idx,n);
10237e33a6baSBarry Smith         x[i] = (1. - omega)*x[i] + omega*(sum + a->a[diag[i]+shift]*x[i])/d;
102417ab2063SBarry Smith       }
102517ab2063SBarry Smith     }
102617ab2063SBarry Smith   }
102717ab2063SBarry Smith   return 0;
102817ab2063SBarry Smith }
102917ab2063SBarry Smith 
10305615d1e5SSatish Balay #undef __FUNC__
10315615d1e5SSatish Balay #define __FUNC__ "MatGetInfo_SeqAIJ"
10328f6be9afSLois Curfman McInnes int MatGetInfo_SeqAIJ(Mat A,MatInfoType flag,MatInfo *info)
103317ab2063SBarry Smith {
1034416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
10354e220ebcSLois Curfman McInnes 
10364e220ebcSLois Curfman McInnes   info->rows_global    = (double)a->m;
10374e220ebcSLois Curfman McInnes   info->columns_global = (double)a->n;
10384e220ebcSLois Curfman McInnes   info->rows_local     = (double)a->m;
10394e220ebcSLois Curfman McInnes   info->columns_local  = (double)a->n;
10404e220ebcSLois Curfman McInnes   info->block_size     = 1.0;
10414e220ebcSLois Curfman McInnes   info->nz_allocated   = (double)a->maxnz;
10424e220ebcSLois Curfman McInnes   info->nz_used        = (double)a->nz;
10434e220ebcSLois Curfman McInnes   info->nz_unneeded    = (double)(a->maxnz - a->nz);
10444e220ebcSLois Curfman McInnes   /*  if (info->nz_unneeded != A->info.nz_unneeded)
10454e220ebcSLois Curfman McInnes     printf("space descrepancy: maxnz-nz = %d, nz_unneeded = %d\n",(int)info->nz_unneeded,(int)A->info.nz_unneeded); */
10464e220ebcSLois Curfman McInnes   info->assemblies     = (double)A->num_ass;
10474e220ebcSLois Curfman McInnes   info->mallocs        = (double)a->reallocs;
10484e220ebcSLois Curfman McInnes   info->memory         = A->mem;
10494e220ebcSLois Curfman McInnes   if (A->factor) {
10504e220ebcSLois Curfman McInnes     info->fill_ratio_given  = A->info.fill_ratio_given;
10514e220ebcSLois Curfman McInnes     info->fill_ratio_needed = A->info.fill_ratio_needed;
10524e220ebcSLois Curfman McInnes     info->factor_mallocs    = A->info.factor_mallocs;
10534e220ebcSLois Curfman McInnes   } else {
10544e220ebcSLois Curfman McInnes     info->fill_ratio_given  = 0;
10554e220ebcSLois Curfman McInnes     info->fill_ratio_needed = 0;
10564e220ebcSLois Curfman McInnes     info->factor_mallocs    = 0;
10574e220ebcSLois Curfman McInnes   }
105817ab2063SBarry Smith   return 0;
105917ab2063SBarry Smith }
106017ab2063SBarry Smith 
106117ab2063SBarry Smith extern int MatLUFactorSymbolic_SeqAIJ(Mat,IS,IS,double,Mat*);
106217ab2063SBarry Smith extern int MatLUFactorNumeric_SeqAIJ(Mat,Mat*);
106317ab2063SBarry Smith extern int MatLUFactor_SeqAIJ(Mat,IS,IS,double);
106417ab2063SBarry Smith extern int MatSolve_SeqAIJ(Mat,Vec,Vec);
106517ab2063SBarry Smith extern int MatSolveAdd_SeqAIJ(Mat,Vec,Vec,Vec);
106617ab2063SBarry Smith extern int MatSolveTrans_SeqAIJ(Mat,Vec,Vec);
106717ab2063SBarry Smith extern int MatSolveTransAdd_SeqAIJ(Mat,Vec,Vec,Vec);
106817ab2063SBarry Smith 
10695615d1e5SSatish Balay #undef __FUNC__
10705615d1e5SSatish Balay #define __FUNC__ "MatZeroRows_SeqAIJ"
10718f6be9afSLois Curfman McInnes int MatZeroRows_SeqAIJ(Mat A,IS is,Scalar *diag)
107217ab2063SBarry Smith {
1073416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
1074416022c9SBarry Smith   int         i,ierr,N, *rows,m = a->m - 1,shift = a->indexshift;
107517ab2063SBarry Smith 
107677c4ece6SBarry Smith   ierr = ISGetSize(is,&N); CHKERRQ(ierr);
107717ab2063SBarry Smith   ierr = ISGetIndices(is,&rows); CHKERRQ(ierr);
107817ab2063SBarry Smith   if (diag) {
107917ab2063SBarry Smith     for ( i=0; i<N; i++ ) {
1080e3372554SBarry Smith       if (rows[i] < 0 || rows[i] > m) SETERRQ(1,0,"row out of range");
1081416022c9SBarry Smith       if (a->ilen[rows[i]] > 0) { /* in case row was completely empty */
1082416022c9SBarry Smith         a->ilen[rows[i]] = 1;
1083416022c9SBarry Smith         a->a[a->i[rows[i]]+shift] = *diag;
1084416022c9SBarry Smith         a->j[a->i[rows[i]]+shift] = rows[i]+shift;
108517ab2063SBarry Smith       }
108617ab2063SBarry Smith       else {
108717ab2063SBarry Smith         ierr = MatSetValues_SeqAIJ(A,1,&rows[i],1,&rows[i],diag,INSERT_VALUES);
108817ab2063SBarry Smith         CHKERRQ(ierr);
108917ab2063SBarry Smith       }
109017ab2063SBarry Smith     }
109117ab2063SBarry Smith   }
109217ab2063SBarry Smith   else {
109317ab2063SBarry Smith     for ( i=0; i<N; i++ ) {
1094e3372554SBarry Smith       if (rows[i] < 0 || rows[i] > m) SETERRQ(1,0,"row out of range");
1095416022c9SBarry Smith       a->ilen[rows[i]] = 0;
109617ab2063SBarry Smith     }
109717ab2063SBarry Smith   }
109817ab2063SBarry Smith   ISRestoreIndices(is,&rows);
109943a90d84SBarry Smith   ierr = MatAssemblyEnd_SeqAIJ(A,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr);
110017ab2063SBarry Smith   return 0;
110117ab2063SBarry Smith }
110217ab2063SBarry Smith 
11035615d1e5SSatish Balay #undef __FUNC__
11045615d1e5SSatish Balay #define __FUNC__ "MatGetSize_SeqAIJ"
11058f6be9afSLois Curfman McInnes int MatGetSize_SeqAIJ(Mat A,int *m,int *n)
110617ab2063SBarry Smith {
1107416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
1108416022c9SBarry Smith   *m = a->m; *n = a->n;
110917ab2063SBarry Smith   return 0;
111017ab2063SBarry Smith }
111117ab2063SBarry Smith 
11125615d1e5SSatish Balay #undef __FUNC__
11135615d1e5SSatish Balay #define __FUNC__ "MatGetOwnershipRange_SeqAIJ"
11148f6be9afSLois Curfman McInnes int MatGetOwnershipRange_SeqAIJ(Mat A,int *m,int *n)
111517ab2063SBarry Smith {
1116416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
1117416022c9SBarry Smith   *m = 0; *n = a->m;
111817ab2063SBarry Smith   return 0;
111917ab2063SBarry Smith }
1120026e39d0SSatish Balay 
11215615d1e5SSatish Balay #undef __FUNC__
11225615d1e5SSatish Balay #define __FUNC__ "MatGetRow_SeqAIJ"
11234e093b46SBarry Smith int MatGetRow_SeqAIJ(Mat A,int row,int *nz,int **idx,Scalar **v)
112417ab2063SBarry Smith {
1125416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
1126c456f294SBarry Smith   int        *itmp,i,shift = a->indexshift;
112717ab2063SBarry Smith 
1128e3372554SBarry Smith   if (row < 0 || row >= a->m) SETERRQ(1,0,"Row out of range");
112917ab2063SBarry Smith 
1130416022c9SBarry Smith   *nz = a->i[row+1] - a->i[row];
1131416022c9SBarry Smith   if (v) *v = a->a + a->i[row] + shift;
113217ab2063SBarry Smith   if (idx) {
1133416022c9SBarry Smith     itmp = a->j + a->i[row] + shift;
11344e093b46SBarry Smith     if (*nz && shift) {
11350452661fSBarry Smith       *idx = (int *) PetscMalloc( (*nz)*sizeof(int) ); CHKPTRQ(*idx);
113617ab2063SBarry Smith       for ( i=0; i<(*nz); i++ ) {(*idx)[i] = itmp[i] + shift;}
11374e093b46SBarry Smith     } else if (*nz) {
11384e093b46SBarry Smith       *idx = itmp;
113917ab2063SBarry Smith     }
114017ab2063SBarry Smith     else *idx = 0;
114117ab2063SBarry Smith   }
114217ab2063SBarry Smith   return 0;
114317ab2063SBarry Smith }
114417ab2063SBarry Smith 
11455615d1e5SSatish Balay #undef __FUNC__
11465615d1e5SSatish Balay #define __FUNC__ "MatRestoreRow_SeqAIJ"
11474e093b46SBarry Smith int MatRestoreRow_SeqAIJ(Mat A,int row,int *nz,int **idx,Scalar **v)
114817ab2063SBarry Smith {
11494e093b46SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
11504e093b46SBarry Smith   if (idx) {if (*idx && a->indexshift) PetscFree(*idx);}
115117ab2063SBarry Smith   return 0;
115217ab2063SBarry Smith }
115317ab2063SBarry Smith 
11545615d1e5SSatish Balay #undef __FUNC__
11555615d1e5SSatish Balay #define __FUNC__ "MatNorm_SeqAIJ"
11568f6be9afSLois Curfman McInnes int MatNorm_SeqAIJ(Mat A,NormType type,double *norm)
115717ab2063SBarry Smith {
1158416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
1159416022c9SBarry Smith   Scalar     *v = a->a;
116017ab2063SBarry Smith   double     sum = 0.0;
1161416022c9SBarry Smith   int        i, j,shift = a->indexshift;
116217ab2063SBarry Smith 
116317ab2063SBarry Smith   if (type == NORM_FROBENIUS) {
1164416022c9SBarry Smith     for (i=0; i<a->nz; i++ ) {
116517ab2063SBarry Smith #if defined(PETSC_COMPLEX)
116617ab2063SBarry Smith       sum += real(conj(*v)*(*v)); v++;
116717ab2063SBarry Smith #else
116817ab2063SBarry Smith       sum += (*v)*(*v); v++;
116917ab2063SBarry Smith #endif
117017ab2063SBarry Smith     }
117117ab2063SBarry Smith     *norm = sqrt(sum);
117217ab2063SBarry Smith   }
117317ab2063SBarry Smith   else if (type == NORM_1) {
117417ab2063SBarry Smith     double *tmp;
1175416022c9SBarry Smith     int    *jj = a->j;
117666963ce1SSatish Balay     tmp = (double *) PetscMalloc( (a->n+1)*sizeof(double) ); CHKPTRQ(tmp);
1177cddf8d76SBarry Smith     PetscMemzero(tmp,a->n*sizeof(double));
117817ab2063SBarry Smith     *norm = 0.0;
1179416022c9SBarry Smith     for ( j=0; j<a->nz; j++ ) {
1180a2744918SBarry Smith         tmp[*jj++ + shift] += PetscAbsScalar(*v);  v++;
118117ab2063SBarry Smith     }
1182416022c9SBarry Smith     for ( j=0; j<a->n; j++ ) {
118317ab2063SBarry Smith       if (tmp[j] > *norm) *norm = tmp[j];
118417ab2063SBarry Smith     }
11850452661fSBarry Smith     PetscFree(tmp);
118617ab2063SBarry Smith   }
118717ab2063SBarry Smith   else if (type == NORM_INFINITY) {
118817ab2063SBarry Smith     *norm = 0.0;
1189416022c9SBarry Smith     for ( j=0; j<a->m; j++ ) {
1190416022c9SBarry Smith       v = a->a + a->i[j] + shift;
119117ab2063SBarry Smith       sum = 0.0;
1192416022c9SBarry Smith       for ( i=0; i<a->i[j+1]-a->i[j]; i++ ) {
1193cddf8d76SBarry Smith         sum += PetscAbsScalar(*v); v++;
119417ab2063SBarry Smith       }
119517ab2063SBarry Smith       if (sum > *norm) *norm = sum;
119617ab2063SBarry Smith     }
119717ab2063SBarry Smith   }
119817ab2063SBarry Smith   else {
1199e3372554SBarry Smith     SETERRQ(1,0,"No support for two norm yet");
120017ab2063SBarry Smith   }
120117ab2063SBarry Smith   return 0;
120217ab2063SBarry Smith }
120317ab2063SBarry Smith 
12045615d1e5SSatish Balay #undef __FUNC__
12055615d1e5SSatish Balay #define __FUNC__ "MatTranspose_SeqAIJ"
12068f6be9afSLois Curfman McInnes int MatTranspose_SeqAIJ(Mat A,Mat *B)
120717ab2063SBarry Smith {
1208416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
1209416022c9SBarry Smith   Mat        C;
1210416022c9SBarry Smith   int        i, ierr, *aj = a->j, *ai = a->i, m = a->m, len, *col;
1211416022c9SBarry Smith   int        shift = a->indexshift;
1212d5d45c9bSBarry Smith   Scalar     *array = a->a;
121317ab2063SBarry Smith 
12143638b69dSLois Curfman McInnes   if (B == PETSC_NULL && m != a->n)
1215e3372554SBarry Smith     SETERRQ(1,0,"Square matrix only for in-place");
12160452661fSBarry Smith   col = (int *) PetscMalloc((1+a->n)*sizeof(int)); CHKPTRQ(col);
1217cddf8d76SBarry Smith   PetscMemzero(col,(1+a->n)*sizeof(int));
121817ab2063SBarry Smith   if (shift) {
121917ab2063SBarry Smith     for ( i=0; i<ai[m]-1; i++ ) aj[i] -= 1;
122017ab2063SBarry Smith   }
122117ab2063SBarry Smith   for ( i=0; i<ai[m]+shift; i++ ) col[aj[i]] += 1;
1222416022c9SBarry Smith   ierr = MatCreateSeqAIJ(A->comm,a->n,m,0,col,&C); CHKERRQ(ierr);
12230452661fSBarry Smith   PetscFree(col);
122417ab2063SBarry Smith   for ( i=0; i<m; i++ ) {
122517ab2063SBarry Smith     len = ai[i+1]-ai[i];
1226416022c9SBarry Smith     ierr = MatSetValues(C,len,aj,1,&i,array,INSERT_VALUES); CHKERRQ(ierr);
122717ab2063SBarry Smith     array += len; aj += len;
122817ab2063SBarry Smith   }
122917ab2063SBarry Smith   if (shift) {
123017ab2063SBarry Smith     for ( i=0; i<ai[m]-1; i++ ) aj[i] += 1;
123117ab2063SBarry Smith   }
123217ab2063SBarry Smith 
12336d4a8577SBarry Smith   ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr);
12346d4a8577SBarry Smith   ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr);
123517ab2063SBarry Smith 
12363638b69dSLois Curfman McInnes   if (B != PETSC_NULL) {
1237416022c9SBarry Smith     *B = C;
123817ab2063SBarry Smith   } else {
1239416022c9SBarry Smith     /* This isn't really an in-place transpose */
12400452661fSBarry Smith     PetscFree(a->a);
12410452661fSBarry Smith     if (!a->singlemalloc) {PetscFree(a->i); PetscFree(a->j);}
12420452661fSBarry Smith     if (a->diag) PetscFree(a->diag);
12430452661fSBarry Smith     if (a->ilen) PetscFree(a->ilen);
12440452661fSBarry Smith     if (a->imax) PetscFree(a->imax);
12450452661fSBarry Smith     if (a->solve_work) PetscFree(a->solve_work);
12461073c447SSatish Balay     if (a->inode.size) PetscFree(a->inode.size);
12470452661fSBarry Smith     PetscFree(a);
1248f09e8eb9SSatish Balay     PetscMemcpy(A,C,sizeof(struct _p_Mat));
12490452661fSBarry Smith     PetscHeaderDestroy(C);
125017ab2063SBarry Smith   }
125117ab2063SBarry Smith   return 0;
125217ab2063SBarry Smith }
125317ab2063SBarry Smith 
12545615d1e5SSatish Balay #undef __FUNC__
12555615d1e5SSatish Balay #define __FUNC__ "MatDiagonalScale_SeqAIJ"
12568f6be9afSLois Curfman McInnes int MatDiagonalScale_SeqAIJ(Mat A,Vec ll,Vec rr)
125717ab2063SBarry Smith {
1258416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
125917ab2063SBarry Smith   Scalar     *l,*r,x,*v;
1260d5d45c9bSBarry Smith   int        i,j,m = a->m, n = a->n, M, nz = a->nz, *jj,shift = a->indexshift;
126117ab2063SBarry Smith 
126217ab2063SBarry Smith   if (ll) {
12633ea7c6a1SSatish Balay     /* The local size is used so that VecMPI can be passed to this routine
12643ea7c6a1SSatish Balay        by MatDiagonalScale_MPIAIJ */
12659b1297e1SSatish Balay     VecGetLocalSize_Fast(ll,m);
1266e3372554SBarry Smith     if (m != a->m) SETERRQ(1,0,"Left scaling vector wrong length");
126790f02eecSBarry Smith     VecGetArray_Fast(ll,l);
1268416022c9SBarry Smith     v = a->a;
126917ab2063SBarry Smith     for ( i=0; i<m; i++ ) {
127017ab2063SBarry Smith       x = l[i];
1271416022c9SBarry Smith       M = a->i[i+1] - a->i[i];
127217ab2063SBarry Smith       for ( j=0; j<M; j++ ) { (*v++) *= x;}
127317ab2063SBarry Smith     }
127444cd7ae7SLois Curfman McInnes     PLogFlops(nz);
127517ab2063SBarry Smith   }
127617ab2063SBarry Smith   if (rr) {
12779b1297e1SSatish Balay     VecGetLocalSize_Fast(rr,n);
1278e3372554SBarry Smith     if (n != a->n) SETERRQ(1,0,"Right scaling vector wrong length");
127990f02eecSBarry Smith     VecGetArray_Fast(rr,r);
1280416022c9SBarry Smith     v = a->a; jj = a->j;
128117ab2063SBarry Smith     for ( i=0; i<nz; i++ ) {
128217ab2063SBarry Smith       (*v++) *= r[*jj++ + shift];
128317ab2063SBarry Smith     }
128444cd7ae7SLois Curfman McInnes     PLogFlops(nz);
128517ab2063SBarry Smith   }
128617ab2063SBarry Smith   return 0;
128717ab2063SBarry Smith }
128817ab2063SBarry Smith 
12895615d1e5SSatish Balay #undef __FUNC__
12905615d1e5SSatish Balay #define __FUNC__ "MatGetSubMatrix_SeqAIJ"
12918f6be9afSLois Curfman McInnes int MatGetSubMatrix_SeqAIJ(Mat A,IS isrow,IS iscol,MatGetSubMatrixCall scall,Mat *B)
129217ab2063SBarry Smith {
1293db02288aSLois Curfman McInnes   Mat_SeqAIJ   *a = (Mat_SeqAIJ *) A->data,*c;
129402834360SBarry Smith   int          nznew, *smap, i, k, kstart, kend, ierr, oldcols = a->n,*lens;
129599141d43SSatish Balay   int          row,mat_i,*mat_j,tcol,first,step,*mat_ilen;
1296a2744918SBarry Smith   register int sum,lensi;
129799141d43SSatish Balay   int          *irow, *icol, nrows, ncols, shift = a->indexshift,*ssmap;
129899141d43SSatish Balay   int          *starts,*j_new,*i_new,*aj = a->j, *ai = a->i,ii,*ailen = a->ilen;
129999141d43SSatish Balay   Scalar       *a_new,*mat_a;
1300416022c9SBarry Smith   Mat          C;
130117ab2063SBarry Smith 
1302b48a1e75SSatish Balay   ierr = ISSorted(isrow,(PetscTruth*)&i);
1303e3372554SBarry Smith   if (!i) SETERRQ(1,0,"ISrow is not sorted");
130499141d43SSatish Balay   ierr = ISSorted(iscol,(PetscTruth*)&i);
1305e3372554SBarry Smith   if (!i) SETERRQ(1,0,"IScol is not sorted");
130699141d43SSatish Balay 
130717ab2063SBarry Smith   ierr = ISGetIndices(isrow,&irow); CHKERRQ(ierr);
130817ab2063SBarry Smith   ierr = ISGetSize(isrow,&nrows); CHKERRQ(ierr);
130917ab2063SBarry Smith   ierr = ISGetSize(iscol,&ncols); CHKERRQ(ierr);
131017ab2063SBarry Smith 
13117264ac53SSatish Balay   if (ISStrideGetInfo(iscol,&first,&step) && step == 1) { /* no need to sort */
131202834360SBarry Smith     /* special case of contiguous rows */
131357faeb66SBarry Smith     lens   = (int *) PetscMalloc((ncols+nrows+1)*sizeof(int)); CHKPTRQ(lens);
131402834360SBarry Smith     starts = lens + ncols;
131502834360SBarry Smith     /* loop over new rows determining lens and starting points */
131602834360SBarry Smith     for (i=0; i<nrows; i++) {
1317a2744918SBarry Smith       kstart  = ai[irow[i]]+shift;
1318a2744918SBarry Smith       kend    = kstart + ailen[irow[i]];
131902834360SBarry Smith       for ( k=kstart; k<kend; k++ ) {
1320d8ced48eSBarry Smith         if (aj[k]+shift >= first) {
132102834360SBarry Smith           starts[i] = k;
132202834360SBarry Smith           break;
132302834360SBarry Smith 	}
132402834360SBarry Smith       }
1325a2744918SBarry Smith       sum = 0;
132602834360SBarry Smith       while (k < kend) {
1327d8ced48eSBarry Smith         if (aj[k++]+shift >= first+ncols) break;
1328a2744918SBarry Smith         sum++;
132902834360SBarry Smith       }
1330a2744918SBarry Smith       lens[i] = sum;
133102834360SBarry Smith     }
133202834360SBarry Smith     /* create submatrix */
1333cddf8d76SBarry Smith     if (scall == MAT_REUSE_MATRIX) {
133408480c60SBarry Smith       int n_cols,n_rows;
133508480c60SBarry Smith       ierr = MatGetSize(*B,&n_rows,&n_cols); CHKERRQ(ierr);
1336e3372554SBarry Smith       if (n_rows != nrows || n_cols != ncols) SETERRQ(1,0,"");
1337d8ced48eSBarry Smith       ierr = MatZeroEntries(*B); CHKERRQ(ierr);
133808480c60SBarry Smith       C = *B;
133908480c60SBarry Smith     }
134008480c60SBarry Smith     else {
134102834360SBarry Smith       ierr = MatCreateSeqAIJ(A->comm,nrows,ncols,0,lens,&C);CHKERRQ(ierr);
134208480c60SBarry Smith     }
1343db02288aSLois Curfman McInnes     c = (Mat_SeqAIJ*) C->data;
1344db02288aSLois Curfman McInnes 
134502834360SBarry Smith     /* loop over rows inserting into submatrix */
1346db02288aSLois Curfman McInnes     a_new    = c->a;
1347db02288aSLois Curfman McInnes     j_new    = c->j;
1348db02288aSLois Curfman McInnes     i_new    = c->i;
1349db02288aSLois Curfman McInnes     i_new[0] = -shift;
135002834360SBarry Smith     for (i=0; i<nrows; i++) {
1351a2744918SBarry Smith       ii    = starts[i];
1352a2744918SBarry Smith       lensi = lens[i];
1353a2744918SBarry Smith       for ( k=0; k<lensi; k++ ) {
1354a2744918SBarry Smith         *j_new++ = aj[ii+k] - first;
135502834360SBarry Smith       }
1356a2744918SBarry Smith       PetscMemcpy(a_new,a->a + starts[i],lensi*sizeof(Scalar));
1357a2744918SBarry Smith       a_new      += lensi;
1358a2744918SBarry Smith       i_new[i+1]  = i_new[i] + lensi;
1359a2744918SBarry Smith       c->ilen[i]  = lensi;
136002834360SBarry Smith     }
13610452661fSBarry Smith     PetscFree(lens);
136202834360SBarry Smith   }
136302834360SBarry Smith   else {
136402834360SBarry Smith     ierr = ISGetIndices(iscol,&icol); CHKERRQ(ierr);
13650452661fSBarry Smith     smap  = (int *) PetscMalloc((1+oldcols)*sizeof(int)); CHKPTRQ(smap);
136602834360SBarry Smith     ssmap = smap + shift;
136799141d43SSatish Balay     lens  = (int *) PetscMalloc((1+nrows)*sizeof(int)); CHKPTRQ(lens);
1368cddf8d76SBarry Smith     PetscMemzero(smap,oldcols*sizeof(int));
136917ab2063SBarry Smith     for ( i=0; i<ncols; i++ ) smap[icol[i]] = i+1;
137002834360SBarry Smith     /* determine lens of each row */
137102834360SBarry Smith     for (i=0; i<nrows; i++) {
1372d8ced48eSBarry Smith       kstart  = ai[irow[i]]+shift;
137302834360SBarry Smith       kend    = kstart + a->ilen[irow[i]];
137402834360SBarry Smith       lens[i] = 0;
137502834360SBarry Smith       for ( k=kstart; k<kend; k++ ) {
1376d8ced48eSBarry Smith         if (ssmap[aj[k]]) {
137702834360SBarry Smith           lens[i]++;
137802834360SBarry Smith         }
137902834360SBarry Smith       }
138002834360SBarry Smith     }
138117ab2063SBarry Smith     /* Create and fill new matrix */
1382a2744918SBarry Smith     if (scall == MAT_REUSE_MATRIX) {
138399141d43SSatish Balay       c = (Mat_SeqAIJ *)((*B)->data);
138499141d43SSatish Balay 
1385e3372554SBarry Smith       if (c->m  != nrows || c->n != ncols) SETERRQ(1,0,"");
138699141d43SSatish Balay       if (PetscMemcmp(c->ilen,lens, c->m *sizeof(int))) {
1387e3372554SBarry Smith         SETERRQ(1,0,"Cannot reuse matrix. wrong no of nonzeros");
138899141d43SSatish Balay       }
138999141d43SSatish Balay       PetscMemzero(c->ilen,c->m*sizeof(int));
139008480c60SBarry Smith       C = *B;
139108480c60SBarry Smith     }
139208480c60SBarry Smith     else {
139302834360SBarry Smith       ierr = MatCreateSeqAIJ(A->comm,nrows,ncols,0,lens,&C);CHKERRQ(ierr);
139408480c60SBarry Smith     }
139599141d43SSatish Balay     c = (Mat_SeqAIJ *)(C->data);
139617ab2063SBarry Smith     for (i=0; i<nrows; i++) {
139799141d43SSatish Balay       row    = irow[i];
139817ab2063SBarry Smith       nznew  = 0;
139999141d43SSatish Balay       kstart = ai[row]+shift;
140099141d43SSatish Balay       kend   = kstart + a->ilen[row];
140199141d43SSatish Balay       mat_i  = c->i[i]+shift;
140299141d43SSatish Balay       mat_j  = c->j + mat_i;
140399141d43SSatish Balay       mat_a  = c->a + mat_i;
140499141d43SSatish Balay       mat_ilen = c->ilen + i;
140517ab2063SBarry Smith       for ( k=kstart; k<kend; k++ ) {
140699141d43SSatish Balay         if ((tcol=ssmap[a->j[k]])) {
140799141d43SSatish Balay           *mat_j++ = tcol - (!shift);
140899141d43SSatish Balay           *mat_a++ = a->a[k];
140999141d43SSatish Balay           (*mat_ilen)++;
141099141d43SSatish Balay 
141117ab2063SBarry Smith         }
141217ab2063SBarry Smith       }
141317ab2063SBarry Smith     }
141402834360SBarry Smith     /* Free work space */
141502834360SBarry Smith     ierr = ISRestoreIndices(iscol,&icol); CHKERRQ(ierr);
141699141d43SSatish Balay     PetscFree(smap); PetscFree(lens);
141702834360SBarry Smith   }
14186d4a8577SBarry Smith   ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr);
14196d4a8577SBarry Smith   ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr);
142017ab2063SBarry Smith 
142117ab2063SBarry Smith   ierr = ISRestoreIndices(isrow,&irow); CHKERRQ(ierr);
1422416022c9SBarry Smith   *B = C;
142317ab2063SBarry Smith   return 0;
142417ab2063SBarry Smith }
142517ab2063SBarry Smith 
1426a871dcd8SBarry Smith /*
142763b91edcSBarry Smith      note: This can only work for identity for row and col. It would
142863b91edcSBarry Smith    be good to check this and otherwise generate an error.
1429a871dcd8SBarry Smith */
14305615d1e5SSatish Balay #undef __FUNC__
14315615d1e5SSatish Balay #define __FUNC__ "MatILUFactor_SeqAIJ"
14328f6be9afSLois Curfman McInnes int MatILUFactor_SeqAIJ(Mat inA,IS row,IS col,double efill,int fill)
1433a871dcd8SBarry Smith {
143463b91edcSBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) inA->data;
143508480c60SBarry Smith   int        ierr;
143663b91edcSBarry Smith   Mat        outA;
143763b91edcSBarry Smith 
1438e3372554SBarry Smith   if (fill != 0) SETERRQ(1,0,"Only fill=0 supported");
1439a871dcd8SBarry Smith 
144063b91edcSBarry Smith   outA          = inA;
144163b91edcSBarry Smith   inA->factor   = FACTOR_LU;
144263b91edcSBarry Smith   a->row        = row;
144363b91edcSBarry Smith   a->col        = col;
144463b91edcSBarry Smith 
144594a9d846SBarry Smith   if (!a->solve_work) { /* this matrix may have been factored before */
14460452661fSBarry Smith     a->solve_work = (Scalar *) PetscMalloc( (a->m+1)*sizeof(Scalar)); CHKPTRQ(a->solve_work);
144794a9d846SBarry Smith   }
144863b91edcSBarry Smith 
144908480c60SBarry Smith   if (!a->diag) {
145008480c60SBarry Smith     ierr = MatMarkDiag_SeqAIJ(inA); CHKERRQ(ierr);
145163b91edcSBarry Smith   }
145263b91edcSBarry Smith   ierr = MatLUFactorNumeric_SeqAIJ(inA,&outA); CHKERRQ(ierr);
1453a871dcd8SBarry Smith   return 0;
1454a871dcd8SBarry Smith }
1455a871dcd8SBarry Smith 
1456f0b747eeSBarry Smith #include "pinclude/plapack.h"
14575615d1e5SSatish Balay #undef __FUNC__
14585615d1e5SSatish Balay #define __FUNC__ "MatScale_SeqAIJ"
14598f6be9afSLois Curfman McInnes int MatScale_SeqAIJ(Scalar *alpha,Mat inA)
1460f0b747eeSBarry Smith {
1461f0b747eeSBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) inA->data;
1462f0b747eeSBarry Smith   int        one = 1;
1463f0b747eeSBarry Smith   BLscal_( &a->nz, alpha, a->a, &one );
1464f0b747eeSBarry Smith   PLogFlops(a->nz);
1465f0b747eeSBarry Smith   return 0;
1466f0b747eeSBarry Smith }
1467f0b747eeSBarry Smith 
14685615d1e5SSatish Balay #undef __FUNC__
14695615d1e5SSatish Balay #define __FUNC__ "MatGetSubMatrices_SeqAIJ"
14708f6be9afSLois Curfman McInnes int MatGetSubMatrices_SeqAIJ(Mat A,int n, IS *irow,IS *icol,MatGetSubMatrixCall scall,
1471cddf8d76SBarry Smith                                     Mat **B)
1472cddf8d76SBarry Smith {
1473cddf8d76SBarry Smith   int ierr,i;
1474cddf8d76SBarry Smith 
1475cddf8d76SBarry Smith   if (scall == MAT_INITIAL_MATRIX) {
14760452661fSBarry Smith     *B = (Mat *) PetscMalloc( (n+1)*sizeof(Mat) ); CHKPTRQ(*B);
1477cddf8d76SBarry Smith   }
1478cddf8d76SBarry Smith 
1479cddf8d76SBarry Smith   for ( i=0; i<n; i++ ) {
1480905e6a2fSBarry Smith     ierr = MatGetSubMatrix_SeqAIJ(A,irow[i],icol[i],scall,&(*B)[i]);CHKERRQ(ierr);
1481cddf8d76SBarry Smith   }
1482cddf8d76SBarry Smith   return 0;
1483cddf8d76SBarry Smith }
1484cddf8d76SBarry Smith 
14855615d1e5SSatish Balay #undef __FUNC__
14865615d1e5SSatish Balay #define __FUNC__ "MatGetBlockSize_SeqAIJ"
14878f6be9afSLois Curfman McInnes int MatGetBlockSize_SeqAIJ(Mat A, int *bs)
14885a838052SSatish Balay {
14895a838052SSatish Balay   *bs = 1;
14905a838052SSatish Balay   return 0;
14915a838052SSatish Balay }
14925a838052SSatish Balay 
14935615d1e5SSatish Balay #undef __FUNC__
14945615d1e5SSatish Balay #define __FUNC__ "MatIncreaseOverlap_SeqAIJ"
14958f6be9afSLois Curfman McInnes int MatIncreaseOverlap_SeqAIJ(Mat A, int is_max, IS *is, int ov)
14964dcbc457SBarry Smith {
1497e4d965acSSatish Balay   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
149806763907SSatish Balay   int        shift, row, i,j,k,l,m,n, *idx,ierr, *nidx, isz, val;
14998a047759SSatish Balay   int        start, end, *ai, *aj;
150006763907SSatish Balay   char       *table;
15018a047759SSatish Balay   shift = a->indexshift;
1502e4d965acSSatish Balay   m     = a->m;
1503e4d965acSSatish Balay   ai    = a->i;
15048a047759SSatish Balay   aj    = a->j+shift;
15058a047759SSatish Balay 
1506e3372554SBarry Smith   if (ov < 0)  SETERRQ(1,0,"illegal overlap value used");
150706763907SSatish Balay 
150806763907SSatish Balay   table = (char *) PetscMalloc((m/BITSPERBYTE +1)*sizeof(char)); CHKPTRQ(table);
150906763907SSatish Balay   nidx  = (int *) PetscMalloc((m+1)*sizeof(int)); CHKPTRQ(nidx);
151006763907SSatish Balay 
1511e4d965acSSatish Balay   for ( i=0; i<is_max; i++ ) {
1512b97fc60eSLois Curfman McInnes     /* Initialize the two local arrays */
1513e4d965acSSatish Balay     isz  = 0;
151406763907SSatish Balay     PetscMemzero(table,(m/BITSPERBYTE +1)*sizeof(char));
1515e4d965acSSatish Balay 
1516e4d965acSSatish Balay     /* Extract the indices, assume there can be duplicate entries */
15174dcbc457SBarry Smith     ierr = ISGetIndices(is[i],&idx);  CHKERRQ(ierr);
151877c4ece6SBarry Smith     ierr = ISGetSize(is[i],&n);  CHKERRQ(ierr);
1519e4d965acSSatish Balay 
1520dd097bc3SLois Curfman McInnes     /* Enter these into the temp arrays. I.e., mark table[row], enter row into new index */
1521e4d965acSSatish Balay     for ( j=0; j<n ; ++j){
152206763907SSatish Balay       if(!BT_LOOKUP(table, idx[j])) { nidx[isz++] = idx[j];}
15234dcbc457SBarry Smith     }
152406763907SSatish Balay     ierr = ISRestoreIndices(is[i],&idx);  CHKERRQ(ierr);
152506763907SSatish Balay     ierr = ISDestroy(is[i]); CHKERRQ(ierr);
1526e4d965acSSatish Balay 
152704a348a9SBarry Smith     k = 0;
152804a348a9SBarry Smith     for ( j=0; j<ov; j++){ /* for each overlap */
152904a348a9SBarry Smith       n = isz;
153006763907SSatish Balay       for ( ; k<n ; k++){ /* do only those rows in nidx[k], which are not done yet */
1531e4d965acSSatish Balay         row   = nidx[k];
1532e4d965acSSatish Balay         start = ai[row];
1533e4d965acSSatish Balay         end   = ai[row+1];
153404a348a9SBarry Smith         for ( l = start; l<end ; l++){
15358a047759SSatish Balay           val = aj[l] + shift;
153606763907SSatish Balay           if (!BT_LOOKUP(table,val)) {nidx[isz++] = val;}
1537e4d965acSSatish Balay         }
1538e4d965acSSatish Balay       }
1539e4d965acSSatish Balay     }
1540029af93fSBarry Smith     ierr = ISCreateGeneral(PETSC_COMM_SELF, isz, nidx, (is+i)); CHKERRQ(ierr);
1541e4d965acSSatish Balay   }
154204a348a9SBarry Smith   PetscFree(table);
154306763907SSatish Balay   PetscFree(nidx);
1544e4d965acSSatish Balay   return 0;
15454dcbc457SBarry Smith }
154617ab2063SBarry Smith 
15470513a670SBarry Smith /* -------------------------------------------------------------- */
15480513a670SBarry Smith #undef __FUNC__
15490513a670SBarry Smith #define __FUNC__ "MatPermute_SeqAIJ"
15500513a670SBarry Smith int MatPermute_SeqAIJ(Mat A, IS rowp, IS colp, Mat *B)
15510513a670SBarry Smith {
15520513a670SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
15530513a670SBarry Smith   Scalar     *vwork;
15540513a670SBarry Smith   int        i, ierr, nz, m = a->m, n = a->n, *cwork;
15550513a670SBarry Smith   int        *row,*col,*cnew,j,*lens;
15560513a670SBarry Smith 
15570513a670SBarry Smith   ierr = ISGetIndices(rowp,&row); CHKERRQ(ierr);
15580513a670SBarry Smith   ierr = ISGetIndices(colp,&col); CHKERRQ(ierr);
15590513a670SBarry Smith 
15600513a670SBarry Smith   /* determine lengths of permuted rows */
15610513a670SBarry Smith   lens = (int *) PetscMalloc( (m+1)*sizeof(int) ); CHKPTRQ(lens);
15620513a670SBarry Smith   for (i=0; i<m; i++ ) {
15630513a670SBarry Smith     lens[row[i]] = a->i[i+1] - a->i[i];
15640513a670SBarry Smith   }
15650513a670SBarry Smith   ierr = MatCreateSeqAIJ(A->comm,m,n,0,lens,B);CHKERRQ(ierr);
15660513a670SBarry Smith   PetscFree(lens);
15670513a670SBarry Smith 
15680513a670SBarry Smith   cnew = (int *) PetscMalloc( n*sizeof(int) ); CHKPTRQ(cnew);
15690513a670SBarry Smith   for (i=0; i<m; i++) {
15700513a670SBarry Smith     ierr = MatGetRow(A,i,&nz,&cwork,&vwork); CHKERRQ(ierr);
15710513a670SBarry Smith     for (j=0; j<nz; j++ ) { cnew[j] = col[cwork[j]];}
15720513a670SBarry Smith     ierr = MatSetValues(*B,1,&row[i],nz,cnew,vwork,INSERT_VALUES); CHKERRQ(ierr);
15730513a670SBarry Smith     ierr = MatRestoreRow(A,i,&nz,&cwork,&vwork); CHKERRQ(ierr);
15740513a670SBarry Smith   }
15750513a670SBarry Smith   PetscFree(cnew);
15760513a670SBarry Smith   ierr = MatAssemblyBegin(*B,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr);
15770513a670SBarry Smith   ierr = MatAssemblyEnd(*B,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr);
15780513a670SBarry Smith   ierr = ISRestoreIndices(rowp,&row); CHKERRQ(ierr);
15790513a670SBarry Smith   ierr = ISRestoreIndices(colp,&col); CHKERRQ(ierr);
15800513a670SBarry Smith   return 0;
15810513a670SBarry Smith }
15820513a670SBarry Smith 
15835615d1e5SSatish Balay #undef __FUNC__
15845615d1e5SSatish Balay #define __FUNC__ "MatPrintHelp_SeqAIJ"
1585682d7d0cSBarry Smith int MatPrintHelp_SeqAIJ(Mat A)
1586682d7d0cSBarry Smith {
1587682d7d0cSBarry Smith   static int called = 0;
1588682d7d0cSBarry Smith   MPI_Comm   comm = A->comm;
1589682d7d0cSBarry Smith 
1590682d7d0cSBarry Smith   if (called) return 0; else called = 1;
159177c4ece6SBarry Smith   PetscPrintf(comm," Options for MATSEQAIJ and MATMPIAIJ matrix formats (the defaults):\n");
15920f665d81SLois Curfman McInnes   PetscPrintf(comm,"  -mat_lu_pivotthreshold <threshold>: Set pivoting threshold\n");
15930f665d81SLois Curfman McInnes   PetscPrintf(comm,"  -mat_aij_oneindex: internal indices begin at 1 instead of the default 0.\n");
15940f665d81SLois Curfman McInnes   PetscPrintf(comm,"  -mat_aij_no_inode: Do not use inodes\n");
15950f665d81SLois Curfman McInnes   PetscPrintf(comm,"  -mat_aij_inode_limit <limit>: Set inode limit (max limit=5)\n");
1596682d7d0cSBarry Smith #if defined(HAVE_ESSL)
15970f665d81SLois Curfman McInnes   PetscPrintf(comm,"  -mat_aij_essl: Use IBM sparse LU factorization and solve.\n");
1598682d7d0cSBarry Smith #endif
1599682d7d0cSBarry Smith   return 0;
1600682d7d0cSBarry Smith }
16018f6be9afSLois Curfman McInnes extern int MatEqual_SeqAIJ(Mat A,Mat B, PetscTruth* flg);
1602a93ec695SBarry Smith extern int MatFDColoringCreate_SeqAIJ(Mat,ISColoring,MatFDColoring);
1603a93ec695SBarry Smith extern int MatColoringPatch_SeqAIJ(Mat,int,int *,ISColoring *);
1604a93ec695SBarry Smith 
1605682d7d0cSBarry Smith /* -------------------------------------------------------------------*/
160617ab2063SBarry Smith static struct _MatOps MatOps = {MatSetValues_SeqAIJ,
160717ab2063SBarry Smith        MatGetRow_SeqAIJ,MatRestoreRow_SeqAIJ,
1608416022c9SBarry Smith        MatMult_SeqAIJ,MatMultAdd_SeqAIJ,
1609416022c9SBarry Smith        MatMultTrans_SeqAIJ,MatMultTransAdd_SeqAIJ,
161017ab2063SBarry Smith        MatSolve_SeqAIJ,MatSolveAdd_SeqAIJ,
161117ab2063SBarry Smith        MatSolveTrans_SeqAIJ,MatSolveTransAdd_SeqAIJ,
161217ab2063SBarry Smith        MatLUFactor_SeqAIJ,0,
161317ab2063SBarry Smith        MatRelax_SeqAIJ,
161417ab2063SBarry Smith        MatTranspose_SeqAIJ,
16157264ac53SSatish Balay        MatGetInfo_SeqAIJ,MatEqual_SeqAIJ,
1616f0b747eeSBarry Smith        MatGetDiagonal_SeqAIJ,MatDiagonalScale_SeqAIJ,MatNorm_SeqAIJ,
161717ab2063SBarry Smith        0,MatAssemblyEnd_SeqAIJ,
161817ab2063SBarry Smith        MatCompress_SeqAIJ,
161917ab2063SBarry Smith        MatSetOption_SeqAIJ,MatZeroEntries_SeqAIJ,MatZeroRows_SeqAIJ,
162017ab2063SBarry Smith        MatLUFactorSymbolic_SeqAIJ,MatLUFactorNumeric_SeqAIJ,0,0,
162117ab2063SBarry Smith        MatGetSize_SeqAIJ,MatGetSize_SeqAIJ,MatGetOwnershipRange_SeqAIJ,
162217ab2063SBarry Smith        MatILUFactorSymbolic_SeqAIJ,0,
162394a9d846SBarry Smith        0,0,
16243d1612f7SBarry Smith        MatConvertSameType_SeqAIJ,0,0,
1625cddf8d76SBarry Smith        MatILUFactor_SeqAIJ,0,0,
16267eb43aa7SLois Curfman McInnes        MatGetSubMatrices_SeqAIJ,MatIncreaseOverlap_SeqAIJ,
1627682d7d0cSBarry Smith        MatGetValues_SeqAIJ,0,
1628f0b747eeSBarry Smith        MatPrintHelp_SeqAIJ,
16295a838052SSatish Balay        MatScale_SeqAIJ,0,0,
16306945ee14SBarry Smith        MatILUDTFactor_SeqAIJ,
16316945ee14SBarry Smith        MatGetBlockSize_SeqAIJ,
16323b2fbd54SBarry Smith        MatGetRowIJ_SeqAIJ,
16333b2fbd54SBarry Smith        MatRestoreRowIJ_SeqAIJ,
16343b2fbd54SBarry Smith        MatGetColumnIJ_SeqAIJ,
1635a93ec695SBarry Smith        MatRestoreColumnIJ_SeqAIJ,
1636a93ec695SBarry Smith        MatFDColoringCreate_SeqAIJ,
16370513a670SBarry Smith        MatColoringPatch_SeqAIJ,
16380513a670SBarry Smith        0,
16390513a670SBarry Smith        MatPermute_SeqAIJ};
164017ab2063SBarry Smith 
164117ab2063SBarry Smith extern int MatUseSuperLU_SeqAIJ(Mat);
164217ab2063SBarry Smith extern int MatUseEssl_SeqAIJ(Mat);
164317ab2063SBarry Smith extern int MatUseDXML_SeqAIJ(Mat);
164417ab2063SBarry Smith 
16455615d1e5SSatish Balay #undef __FUNC__
16465615d1e5SSatish Balay #define __FUNC__ "MatCreateSeqAIJ"
164717ab2063SBarry Smith /*@C
1648682d7d0cSBarry Smith    MatCreateSeqAIJ - Creates a sparse matrix in AIJ (compressed row) format
16490d15e28bSLois Curfman McInnes    (the default parallel PETSc format).  For good matrix assembly performance
16506e62573dSLois Curfman McInnes    the user should preallocate the matrix storage by setting the parameter nz
16512bd5e0b2SLois Curfman McInnes    (or the array nzz).  By setting these parameters accurately, performance
16522bd5e0b2SLois Curfman McInnes    during matrix assembly can be increased by more than a factor of 50.
165317ab2063SBarry Smith 
165417ab2063SBarry Smith    Input Parameters:
1655029af93fSBarry Smith .  comm - MPI communicator, set to PETSC_COMM_SELF
165617ab2063SBarry Smith .  m - number of rows
165717ab2063SBarry Smith .  n - number of columns
165817ab2063SBarry Smith .  nz - number of nonzeros per row (same for all rows)
16592bd5e0b2SLois Curfman McInnes .  nzz - array containing the number of nonzeros in the various rows
16602bd5e0b2SLois Curfman McInnes          (possibly different for each row) or PETSC_NULL
166117ab2063SBarry Smith 
166217ab2063SBarry Smith    Output Parameter:
1663416022c9SBarry Smith .  A - the matrix
166417ab2063SBarry Smith 
166517ab2063SBarry Smith    Notes:
166617ab2063SBarry Smith    The AIJ format (also called the Yale sparse matrix format or
166717ab2063SBarry Smith    compressed row storage), is fully compatible with standard Fortran 77
16680002213bSLois Curfman McInnes    storage.  That is, the stored row and column indices can begin at
166944cd7ae7SLois Curfman McInnes    either one (as in Fortran) or zero.  See the users' manual for details.
167017ab2063SBarry Smith 
167117ab2063SBarry Smith    Specify the preallocated storage with either nz or nnz (not both).
1672a40aa06bSLois Curfman McInnes    Set nz=PETSC_DEFAULT and nnz=PETSC_NULL for PETSc to control dynamic memory
16733d323bbdSBarry Smith    allocation.  For large problems you MUST preallocate memory or you
16746da5968aSLois Curfman McInnes    will get TERRIBLE performance, see the users' manual chapter on matrices.
167517ab2063SBarry Smith 
1676682d7d0cSBarry Smith    By default, this format uses inodes (identical nodes) when possible, to
1677682d7d0cSBarry Smith    improve numerical efficiency of Matrix vector products and solves. We
1678682d7d0cSBarry Smith    search for consecutive rows with the same nonzero structure, thereby
16796c7ebb05SLois Curfman McInnes    reusing matrix information to achieve increased efficiency.
16806c7ebb05SLois Curfman McInnes 
16816c7ebb05SLois Curfman McInnes    Options Database Keys:
16826c7ebb05SLois Curfman McInnes $    -mat_aij_no_inode  - Do not use inodes
16836e62573dSLois Curfman McInnes $    -mat_aij_inode_limit <limit> - Set inode limit.
16846e62573dSLois Curfman McInnes $        (max limit=5)
16856e62573dSLois Curfman McInnes $    -mat_aij_oneindex - Internally use indexing starting at 1
16866e62573dSLois Curfman McInnes $        rather than 0.  Note: When calling MatSetValues(),
16876e62573dSLois Curfman McInnes $        the user still MUST index entries starting at 0!
168817ab2063SBarry Smith 
168917ab2063SBarry Smith .seealso: MatCreate(), MatCreateMPIAIJ(), MatSetValues()
169017ab2063SBarry Smith @*/
1691416022c9SBarry Smith int MatCreateSeqAIJ(MPI_Comm comm,int m,int n,int nz,int *nnz, Mat *A)
169217ab2063SBarry Smith {
1693416022c9SBarry Smith   Mat        B;
1694416022c9SBarry Smith   Mat_SeqAIJ *b;
16956945ee14SBarry Smith   int        i, len, ierr, flg,size;
16966945ee14SBarry Smith 
16976945ee14SBarry Smith   MPI_Comm_size(comm,&size);
1698e3372554SBarry Smith   if (size > 1) SETERRQ(1,0,"Comm must be of size 1");
1699d5d45c9bSBarry Smith 
1700416022c9SBarry Smith   *A                  = 0;
1701f09e8eb9SSatish Balay   PetscHeaderCreate(B,_p_Mat,MAT_COOKIE,MATSEQAIJ,comm);
1702416022c9SBarry Smith   PLogObjectCreate(B);
17030452661fSBarry Smith   B->data             = (void *) (b = PetscNew(Mat_SeqAIJ)); CHKPTRQ(b);
170444cd7ae7SLois Curfman McInnes   PetscMemzero(b,sizeof(Mat_SeqAIJ));
1705416022c9SBarry Smith   PetscMemcpy(&B->ops,&MatOps,sizeof(struct _MatOps));
1706416022c9SBarry Smith   B->destroy          = MatDestroy_SeqAIJ;
1707416022c9SBarry Smith   B->view             = MatView_SeqAIJ;
1708416022c9SBarry Smith   B->factor           = 0;
1709416022c9SBarry Smith   B->lupivotthreshold = 1.0;
171090f02eecSBarry Smith   B->mapping          = 0;
17117a743949SBarry Smith   ierr = OptionsGetDouble(PETSC_NULL,"-mat_lu_pivotthreshold",&B->lupivotthreshold,
171269957df2SSatish Balay                           &flg); CHKERRQ(ierr);
17137a743949SBarry Smith   b->ilu_preserve_row_sums = PETSC_FALSE;
17147a743949SBarry Smith   ierr = OptionsHasName(PETSC_NULL,"-pc_ilu_preserve_row_sums",
17157a743949SBarry Smith                         (int*) &b->ilu_preserve_row_sums); CHKERRQ(ierr);
1716416022c9SBarry Smith   b->row              = 0;
1717416022c9SBarry Smith   b->col              = 0;
1718416022c9SBarry Smith   b->indexshift       = 0;
1719b810aeb4SBarry Smith   b->reallocs         = 0;
172069957df2SSatish Balay   ierr = OptionsHasName(PETSC_NULL,"-mat_aij_oneindex", &flg); CHKERRQ(ierr);
172169957df2SSatish Balay   if (flg) b->indexshift = -1;
172217ab2063SBarry Smith 
172344cd7ae7SLois Curfman McInnes   b->m = m; B->m = m; B->M = m;
172444cd7ae7SLois Curfman McInnes   b->n = n; B->n = n; B->N = n;
17250452661fSBarry Smith   b->imax = (int *) PetscMalloc( (m+1)*sizeof(int) ); CHKPTRQ(b->imax);
1726b4fd4287SBarry Smith   if (nnz == PETSC_NULL) {
17277b8455f0SLois Curfman McInnes     if (nz == PETSC_DEFAULT) nz = 10;
17287b8455f0SLois Curfman McInnes     else if (nz <= 0)        nz = 1;
1729416022c9SBarry Smith     for ( i=0; i<m; i++ ) b->imax[i] = nz;
173017ab2063SBarry Smith     nz = nz*m;
173117ab2063SBarry Smith   }
173217ab2063SBarry Smith   else {
173317ab2063SBarry Smith     nz = 0;
1734416022c9SBarry Smith     for ( i=0; i<m; i++ ) {b->imax[i] = nnz[i]; nz += nnz[i];}
173517ab2063SBarry Smith   }
173617ab2063SBarry Smith 
173717ab2063SBarry Smith   /* allocate the matrix space */
1738416022c9SBarry Smith   len     = nz*(sizeof(int) + sizeof(Scalar)) + (b->m+1)*sizeof(int);
17390452661fSBarry Smith   b->a  = (Scalar *) PetscMalloc( len ); CHKPTRQ(b->a);
1740416022c9SBarry Smith   b->j  = (int *) (b->a + nz);
1741cddf8d76SBarry Smith   PetscMemzero(b->j,nz*sizeof(int));
1742416022c9SBarry Smith   b->i  = b->j + nz;
1743416022c9SBarry Smith   b->singlemalloc = 1;
174417ab2063SBarry Smith 
1745416022c9SBarry Smith   b->i[0] = -b->indexshift;
174617ab2063SBarry Smith   for (i=1; i<m+1; i++) {
1747416022c9SBarry Smith     b->i[i] = b->i[i-1] + b->imax[i-1];
174817ab2063SBarry Smith   }
174917ab2063SBarry Smith 
1750416022c9SBarry Smith   /* b->ilen will count nonzeros in each row so far. */
17510452661fSBarry Smith   b->ilen = (int *) PetscMalloc((m+1)*sizeof(int));
1752f09e8eb9SSatish Balay   PLogObjectMemory(B,len+2*(m+1)*sizeof(int)+sizeof(struct _p_Mat)+sizeof(Mat_SeqAIJ));
1753416022c9SBarry Smith   for ( i=0; i<b->m; i++ ) { b->ilen[i] = 0;}
175417ab2063SBarry Smith 
1755416022c9SBarry Smith   b->nz               = 0;
1756416022c9SBarry Smith   b->maxnz            = nz;
1757416022c9SBarry Smith   b->sorted           = 0;
1758416022c9SBarry Smith   b->roworiented      = 1;
1759416022c9SBarry Smith   b->nonew            = 0;
1760416022c9SBarry Smith   b->diag             = 0;
1761416022c9SBarry Smith   b->solve_work       = 0;
176271bd300dSLois Curfman McInnes   b->spptr            = 0;
1763754ec7b1SSatish Balay   b->inode.node_count = 0;
1764754ec7b1SSatish Balay   b->inode.size       = 0;
17656c7ebb05SLois Curfman McInnes   b->inode.limit      = 5;
17666c7ebb05SLois Curfman McInnes   b->inode.max_limit  = 5;
17674e220ebcSLois Curfman McInnes   B->info.nz_unneeded = (double)b->maxnz;
176817ab2063SBarry Smith 
1769416022c9SBarry Smith   *A = B;
17704e220ebcSLois Curfman McInnes 
17714b14c69eSBarry Smith   /*  SuperLU is not currently supported through PETSc */
17724b14c69eSBarry Smith #if defined(HAVE_SUPERLU)
177369957df2SSatish Balay   ierr = OptionsHasName(PETSC_NULL,"-mat_aij_superlu", &flg); CHKERRQ(ierr);
177469957df2SSatish Balay   if (flg) { ierr = MatUseSuperLU_SeqAIJ(B); CHKERRQ(ierr); }
17754b14c69eSBarry Smith #endif
177669957df2SSatish Balay   ierr = OptionsHasName(PETSC_NULL,"-mat_aij_essl", &flg); CHKERRQ(ierr);
177769957df2SSatish Balay   if (flg) { ierr = MatUseEssl_SeqAIJ(B); CHKERRQ(ierr); }
177869957df2SSatish Balay   ierr = OptionsHasName(PETSC_NULL,"-mat_aij_dxml", &flg); CHKERRQ(ierr);
177969957df2SSatish Balay   if (flg) {
1780e3372554SBarry Smith     if (!b->indexshift) SETERRQ(1,0,"need -mat_aij_oneindex with -mat_aij_dxml");
1781416022c9SBarry Smith     ierr = MatUseDXML_SeqAIJ(B); CHKERRQ(ierr);
178217ab2063SBarry Smith   }
178369957df2SSatish Balay   ierr = OptionsHasName(PETSC_NULL,"-help", &flg); CHKERRQ(ierr);
178469957df2SSatish Balay   if (flg) {ierr = MatPrintHelp(B); CHKERRQ(ierr); }
178517ab2063SBarry Smith   return 0;
178617ab2063SBarry Smith }
178717ab2063SBarry Smith 
17885615d1e5SSatish Balay #undef __FUNC__
17895615d1e5SSatish Balay #define __FUNC__ "MatConvertSameType_SeqAIJ"
17903d1612f7SBarry Smith int MatConvertSameType_SeqAIJ(Mat A,Mat *B,int cpvalues)
179117ab2063SBarry Smith {
1792416022c9SBarry Smith   Mat        C;
1793416022c9SBarry Smith   Mat_SeqAIJ *c,*a = (Mat_SeqAIJ *) A->data;
179408480c60SBarry Smith   int        i,len, m = a->m,shift = a->indexshift;
179517ab2063SBarry Smith 
17964043dd9cSLois Curfman McInnes   *B = 0;
1797f09e8eb9SSatish Balay   PetscHeaderCreate(C,_p_Mat,MAT_COOKIE,MATSEQAIJ,A->comm);
1798416022c9SBarry Smith   PLogObjectCreate(C);
17990452661fSBarry Smith   C->data       = (void *) (c = PetscNew(Mat_SeqAIJ)); CHKPTRQ(c);
180041c01911SSatish Balay   PetscMemcpy(&C->ops,&A->ops,sizeof(struct _MatOps));
1801416022c9SBarry Smith   C->destroy    = MatDestroy_SeqAIJ;
1802416022c9SBarry Smith   C->view       = MatView_SeqAIJ;
1803416022c9SBarry Smith   C->factor     = A->factor;
1804416022c9SBarry Smith   c->row        = 0;
1805416022c9SBarry Smith   c->col        = 0;
1806416022c9SBarry Smith   c->indexshift = shift;
1807c456f294SBarry Smith   C->assembled  = PETSC_TRUE;
180817ab2063SBarry Smith 
180944cd7ae7SLois Curfman McInnes   c->m = C->m   = a->m;
181044cd7ae7SLois Curfman McInnes   c->n = C->n   = a->n;
181144cd7ae7SLois Curfman McInnes   C->M          = a->m;
181244cd7ae7SLois Curfman McInnes   C->N          = a->n;
181317ab2063SBarry Smith 
18140452661fSBarry Smith   c->imax       = (int *) PetscMalloc((m+1)*sizeof(int)); CHKPTRQ(c->imax);
18150452661fSBarry Smith   c->ilen       = (int *) PetscMalloc((m+1)*sizeof(int)); CHKPTRQ(c->ilen);
181617ab2063SBarry Smith   for ( i=0; i<m; i++ ) {
1817416022c9SBarry Smith     c->imax[i] = a->imax[i];
1818416022c9SBarry Smith     c->ilen[i] = a->ilen[i];
181917ab2063SBarry Smith   }
182017ab2063SBarry Smith 
182117ab2063SBarry Smith   /* allocate the matrix space */
1822416022c9SBarry Smith   c->singlemalloc = 1;
1823416022c9SBarry Smith   len     = (m+1)*sizeof(int)+(a->i[m])*(sizeof(Scalar)+sizeof(int));
18240452661fSBarry Smith   c->a  = (Scalar *) PetscMalloc( len ); CHKPTRQ(c->a);
1825416022c9SBarry Smith   c->j  = (int *) (c->a + a->i[m] + shift);
1826416022c9SBarry Smith   c->i  = c->j + a->i[m] + shift;
1827416022c9SBarry Smith   PetscMemcpy(c->i,a->i,(m+1)*sizeof(int));
182817ab2063SBarry Smith   if (m > 0) {
1829416022c9SBarry Smith     PetscMemcpy(c->j,a->j,(a->i[m]+shift)*sizeof(int));
183008480c60SBarry Smith     if (cpvalues == COPY_VALUES) {
1831416022c9SBarry Smith       PetscMemcpy(c->a,a->a,(a->i[m]+shift)*sizeof(Scalar));
183217ab2063SBarry Smith     }
183308480c60SBarry Smith   }
183417ab2063SBarry Smith 
1835f09e8eb9SSatish Balay   PLogObjectMemory(C,len+2*(m+1)*sizeof(int)+sizeof(struct _p_Mat)+sizeof(Mat_SeqAIJ));
1836416022c9SBarry Smith   c->sorted      = a->sorted;
1837416022c9SBarry Smith   c->roworiented = a->roworiented;
1838416022c9SBarry Smith   c->nonew       = a->nonew;
18397a743949SBarry Smith   c->ilu_preserve_row_sums = a->ilu_preserve_row_sums;
184017ab2063SBarry Smith 
1841416022c9SBarry Smith   if (a->diag) {
18420452661fSBarry Smith     c->diag = (int *) PetscMalloc( (m+1)*sizeof(int) ); CHKPTRQ(c->diag);
1843416022c9SBarry Smith     PLogObjectMemory(C,(m+1)*sizeof(int));
184417ab2063SBarry Smith     for ( i=0; i<m; i++ ) {
1845416022c9SBarry Smith       c->diag[i] = a->diag[i];
184617ab2063SBarry Smith     }
184717ab2063SBarry Smith   }
1848416022c9SBarry Smith   else c->diag          = 0;
18496c7ebb05SLois Curfman McInnes   c->inode.limit        = a->inode.limit;
18506c7ebb05SLois Curfman McInnes   c->inode.max_limit    = a->inode.max_limit;
1851754ec7b1SSatish Balay   if (a->inode.size){
1852daed632aSSatish Balay     c->inode.size       = (int *) PetscMalloc( (m+1)*sizeof(int) ); CHKPTRQ(c->inode.size);
1853754ec7b1SSatish Balay     c->inode.node_count = a->inode.node_count;
1854daed632aSSatish Balay     PetscMemcpy( c->inode.size, a->inode.size, (m+1)*sizeof(int));
1855754ec7b1SSatish Balay   } else {
1856754ec7b1SSatish Balay     c->inode.size       = 0;
1857754ec7b1SSatish Balay     c->inode.node_count = 0;
1858754ec7b1SSatish Balay   }
1859416022c9SBarry Smith   c->nz                 = a->nz;
1860416022c9SBarry Smith   c->maxnz              = a->maxnz;
1861416022c9SBarry Smith   c->solve_work         = 0;
186276dd722bSSatish Balay   c->spptr              = 0;      /* Dangerous -I'm throwing away a->spptr */
1863754ec7b1SSatish Balay 
1864416022c9SBarry Smith   *B = C;
186517ab2063SBarry Smith   return 0;
186617ab2063SBarry Smith }
186717ab2063SBarry Smith 
18685615d1e5SSatish Balay #undef __FUNC__
18695615d1e5SSatish Balay #define __FUNC__ "MatLoad_SeqAIJ"
187019bcc07fSBarry Smith int MatLoad_SeqAIJ(Viewer viewer,MatType type,Mat *A)
187117ab2063SBarry Smith {
1872416022c9SBarry Smith   Mat_SeqAIJ   *a;
1873416022c9SBarry Smith   Mat          B;
187417699dbbSLois Curfman McInnes   int          i, nz, ierr, fd, header[4],size,*rowlengths = 0,M,N,shift;
1875bcd2baecSBarry Smith   MPI_Comm     comm;
187617ab2063SBarry Smith 
187719bcc07fSBarry Smith   PetscObjectGetComm((PetscObject) viewer,&comm);
187817699dbbSLois Curfman McInnes   MPI_Comm_size(comm,&size);
1879e3372554SBarry Smith   if (size > 1) SETERRQ(1,0,"view must have one processor");
188090ace30eSBarry Smith   ierr = ViewerBinaryGetDescriptor(viewer,&fd); CHKERRQ(ierr);
188177c4ece6SBarry Smith   ierr = PetscBinaryRead(fd,header,4,BINARY_INT); CHKERRQ(ierr);
1882e3372554SBarry Smith   if (header[0] != MAT_COOKIE) SETERRQ(1,0,"not matrix object in file");
188317ab2063SBarry Smith   M = header[1]; N = header[2]; nz = header[3];
188417ab2063SBarry Smith 
188517ab2063SBarry Smith   /* read in row lengths */
18860452661fSBarry Smith   rowlengths = (int*) PetscMalloc( M*sizeof(int) ); CHKPTRQ(rowlengths);
188777c4ece6SBarry Smith   ierr = PetscBinaryRead(fd,rowlengths,M,BINARY_INT); CHKERRQ(ierr);
188817ab2063SBarry Smith 
188917ab2063SBarry Smith   /* create our matrix */
1890416022c9SBarry Smith   ierr = MatCreateSeqAIJ(comm,M,N,0,rowlengths,A); CHKERRQ(ierr);
1891416022c9SBarry Smith   B = *A;
1892416022c9SBarry Smith   a = (Mat_SeqAIJ *) B->data;
1893416022c9SBarry Smith   shift = a->indexshift;
189417ab2063SBarry Smith 
189517ab2063SBarry Smith   /* read in column indices and adjust for Fortran indexing*/
189677c4ece6SBarry Smith   ierr = PetscBinaryRead(fd,a->j,nz,BINARY_INT); CHKERRQ(ierr);
189717ab2063SBarry Smith   if (shift) {
189817ab2063SBarry Smith     for ( i=0; i<nz; i++ ) {
1899416022c9SBarry Smith       a->j[i] += 1;
190017ab2063SBarry Smith     }
190117ab2063SBarry Smith   }
190217ab2063SBarry Smith 
190317ab2063SBarry Smith   /* read in nonzero values */
190477c4ece6SBarry Smith   ierr = PetscBinaryRead(fd,a->a,nz,BINARY_SCALAR); CHKERRQ(ierr);
190517ab2063SBarry Smith 
190617ab2063SBarry Smith   /* set matrix "i" values */
1907416022c9SBarry Smith   a->i[0] = -shift;
190817ab2063SBarry Smith   for ( i=1; i<= M; i++ ) {
1909416022c9SBarry Smith     a->i[i]      = a->i[i-1] + rowlengths[i-1];
1910416022c9SBarry Smith     a->ilen[i-1] = rowlengths[i-1];
191117ab2063SBarry Smith   }
19120452661fSBarry Smith   PetscFree(rowlengths);
191317ab2063SBarry Smith 
19146d4a8577SBarry Smith   ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr);
19156d4a8577SBarry Smith   ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr);
191617ab2063SBarry Smith   return 0;
191717ab2063SBarry Smith }
191817ab2063SBarry Smith 
19195615d1e5SSatish Balay #undef __FUNC__
19205615d1e5SSatish Balay #define __FUNC__ "MatEqual_SeqAIJ"
19218f6be9afSLois Curfman McInnes int MatEqual_SeqAIJ(Mat A,Mat B, PetscTruth* flg)
19227264ac53SSatish Balay {
19237264ac53SSatish Balay   Mat_SeqAIJ *a = (Mat_SeqAIJ *)A->data, *b = (Mat_SeqAIJ *)B->data;
19247264ac53SSatish Balay 
1925e3372554SBarry Smith   if (B->type !=MATSEQAIJ)SETERRQ(1,0,"Matrices must be same type");
19267264ac53SSatish Balay 
19277264ac53SSatish Balay   /* If the  matrix dimensions are not equal, or no of nonzeros or shift */
19287264ac53SSatish Balay   if ((a->m != b->m ) || (a->n !=b->n) ||( a->nz != b->nz)||
1929bcd2baecSBarry Smith       (a->indexshift != b->indexshift)) {
193077c4ece6SBarry Smith     *flg = PETSC_FALSE; return 0;
1931bcd2baecSBarry Smith   }
19327264ac53SSatish Balay 
19337264ac53SSatish Balay   /* if the a->i are the same */
1934bcd2baecSBarry Smith   if (PetscMemcmp(a->i,b->i, (a->n+1)*sizeof(int))) {
193577c4ece6SBarry Smith     *flg = PETSC_FALSE; return 0;
19367264ac53SSatish Balay   }
19377264ac53SSatish Balay 
19387264ac53SSatish Balay   /* if a->j are the same */
1939bcd2baecSBarry Smith   if (PetscMemcmp(a->j, b->j, (a->nz)*sizeof(int))) {
194077c4ece6SBarry Smith     *flg = PETSC_FALSE; return 0;
1941bcd2baecSBarry Smith   }
1942bcd2baecSBarry Smith 
1943bcd2baecSBarry Smith   /* if a->a are the same */
194419bcc07fSBarry Smith   if (PetscMemcmp(a->a, b->a, (a->nz)*sizeof(Scalar))) {
194577c4ece6SBarry Smith     *flg = PETSC_FALSE; return 0;
19467264ac53SSatish Balay   }
194777c4ece6SBarry Smith   *flg = PETSC_TRUE;
19487264ac53SSatish Balay   return 0;
19497264ac53SSatish Balay 
19507264ac53SSatish Balay }
1951