xref: /petsc/src/mat/impls/aij/seq/aij.c (revision 9ea0dfa2fdacae7aa8b0f0a24607bca34aba59e3)
117ab2063SBarry Smith #ifndef lint
2*9ea0dfa2SSatish Balay static char vcid[] = "$Id: aij.c,v 1.215 1997/04/10 00:02:38 bsmith Exp balay $";
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;
186c2653b3dSLois Curfman McInnes       else if (nonew == -1) SETERRQ(1,1,"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 
19217ab2063SBarry Smith         /* malloc new storage space */
193416022c9SBarry Smith         len     = new_nz*(sizeof(int)+sizeof(Scalar))+(a->m+1)*sizeof(int);
1940452661fSBarry Smith         new_a   = (Scalar *) PetscMalloc( len ); CHKPTRQ(new_a);
19517ab2063SBarry Smith         new_j   = (int *) (new_a + new_nz);
19617ab2063SBarry Smith         new_i   = new_j + new_nz;
19717ab2063SBarry Smith 
19817ab2063SBarry Smith         /* copy over old data into new slots */
19917ab2063SBarry Smith         for ( ii=0; ii<row+1; ii++ ) {new_i[ii] = ai[ii];}
200416022c9SBarry Smith         for ( ii=row+1; ii<a->m+1; ii++ ) {new_i[ii] = ai[ii]+CHUNKSIZE;}
201416022c9SBarry Smith         PetscMemcpy(new_j,aj,(ai[row]+nrow+shift)*sizeof(int));
202416022c9SBarry Smith         len = (new_nz - CHUNKSIZE - ai[row] - nrow - shift);
203416022c9SBarry Smith         PetscMemcpy(new_j+ai[row]+shift+nrow+CHUNKSIZE,aj+ai[row]+shift+nrow,
20417ab2063SBarry Smith                                                            len*sizeof(int));
205416022c9SBarry Smith         PetscMemcpy(new_a,aa,(ai[row]+nrow+shift)*sizeof(Scalar));
206416022c9SBarry Smith         PetscMemcpy(new_a+ai[row]+shift+nrow+CHUNKSIZE,aa+ai[row]+shift+nrow,
20717ab2063SBarry Smith                                                            len*sizeof(Scalar));
20817ab2063SBarry Smith         /* free up old matrix storage */
2090452661fSBarry Smith         PetscFree(a->a);
2100452661fSBarry Smith         if (!a->singlemalloc) {PetscFree(a->i);PetscFree(a->j);}
211416022c9SBarry Smith         aa = a->a = new_a; ai = a->i = new_i; aj = a->j = new_j;
212416022c9SBarry Smith         a->singlemalloc = 1;
21317ab2063SBarry Smith 
21417ab2063SBarry Smith         rp   = aj + ai[row] + shift; ap = aa + ai[row] + shift;
215416022c9SBarry Smith         rmax = imax[row] = imax[row] + CHUNKSIZE;
216416022c9SBarry Smith         PLogObjectMemory(A,CHUNKSIZE*(sizeof(int) + sizeof(Scalar)));
217416022c9SBarry Smith         a->maxnz += CHUNKSIZE;
218b810aeb4SBarry Smith         a->reallocs++;
21917ab2063SBarry Smith       }
220416022c9SBarry Smith       N = nrow++ - 1; a->nz++;
221416022c9SBarry Smith       /* shift up all the later entries in this row */
222416022c9SBarry Smith       for ( ii=N; ii>=i; ii-- ) {
22317ab2063SBarry Smith         rp[ii+1] = rp[ii];
22417ab2063SBarry Smith         ap[ii+1] = ap[ii];
22517ab2063SBarry Smith       }
22617ab2063SBarry Smith       rp[i] = col;
22717ab2063SBarry Smith       ap[i] = value;
22817ab2063SBarry Smith       noinsert:;
229416022c9SBarry Smith       low = i + 1;
23017ab2063SBarry Smith     }
23117ab2063SBarry Smith     ailen[row] = nrow;
23217ab2063SBarry Smith   }
23317ab2063SBarry Smith   return 0;
23417ab2063SBarry Smith }
23517ab2063SBarry Smith 
2365615d1e5SSatish Balay #undef __FUNC__
2375615d1e5SSatish Balay #define __FUNC__ "MatGetValues_SeqAIJ"
2388f6be9afSLois Curfman McInnes int MatGetValues_SeqAIJ(Mat A,int m,int *im,int n,int *in,Scalar *v)
2397eb43aa7SLois Curfman McInnes {
2407eb43aa7SLois Curfman McInnes   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
241b49de8d1SLois Curfman McInnes   int        *rp, k, low, high, t, row, nrow, i, col, l, *aj = a->j;
2427eb43aa7SLois Curfman McInnes   int        *ai = a->i, *ailen = a->ilen, shift = a->indexshift;
2437eb43aa7SLois Curfman McInnes   Scalar     *ap, *aa = a->a, zero = 0.0;
2447eb43aa7SLois Curfman McInnes 
2457eb43aa7SLois Curfman McInnes   for ( k=0; k<m; k++ ) { /* loop over rows */
2467eb43aa7SLois Curfman McInnes     row  = im[k];
247e3372554SBarry Smith     if (row < 0) SETERRQ(1,0,"Negative row");
248e3372554SBarry Smith     if (row >= a->m) SETERRQ(1,0,"Row too large");
2497eb43aa7SLois Curfman McInnes     rp   = aj + ai[row] + shift; ap = aa + ai[row] + shift;
2507eb43aa7SLois Curfman McInnes     nrow = ailen[row];
2517eb43aa7SLois Curfman McInnes     for ( l=0; l<n; l++ ) { /* loop over columns */
252e3372554SBarry Smith       if (in[l] < 0) SETERRQ(1,0,"Negative column");
253e3372554SBarry Smith       if (in[l] >= a->n) SETERRQ(1,0,"Column too large");
2547eb43aa7SLois Curfman McInnes       col = in[l] - shift;
2557eb43aa7SLois Curfman McInnes       high = nrow; low = 0; /* assume unsorted */
2567eb43aa7SLois Curfman McInnes       while (high-low > 5) {
2577eb43aa7SLois Curfman McInnes         t = (low+high)/2;
2587eb43aa7SLois Curfman McInnes         if (rp[t] > col) high = t;
2597eb43aa7SLois Curfman McInnes         else             low  = t;
2607eb43aa7SLois Curfman McInnes       }
2617eb43aa7SLois Curfman McInnes       for ( i=low; i<high; i++ ) {
2627eb43aa7SLois Curfman McInnes         if (rp[i] > col) break;
2637eb43aa7SLois Curfman McInnes         if (rp[i] == col) {
264b49de8d1SLois Curfman McInnes           *v++ = ap[i];
2657eb43aa7SLois Curfman McInnes           goto finished;
2667eb43aa7SLois Curfman McInnes         }
2677eb43aa7SLois Curfman McInnes       }
268b49de8d1SLois Curfman McInnes       *v++ = zero;
2697eb43aa7SLois Curfman McInnes       finished:;
2707eb43aa7SLois Curfman McInnes     }
2717eb43aa7SLois Curfman McInnes   }
2727eb43aa7SLois Curfman McInnes   return 0;
2737eb43aa7SLois Curfman McInnes }
2747eb43aa7SLois Curfman McInnes 
27517ab2063SBarry Smith 
2765615d1e5SSatish Balay #undef __FUNC__
2775615d1e5SSatish Balay #define __FUNC__ "MatView_SeqAIJ_Binary"
2788f6be9afSLois Curfman McInnes extern int MatView_SeqAIJ_Binary(Mat A,Viewer viewer)
27917ab2063SBarry Smith {
280416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
281416022c9SBarry Smith   int        i, fd, *col_lens, ierr;
28217ab2063SBarry Smith 
28390ace30eSBarry Smith   ierr = ViewerBinaryGetDescriptor(viewer,&fd); CHKERRQ(ierr);
2840452661fSBarry Smith   col_lens = (int *) PetscMalloc( (4+a->m)*sizeof(int) ); CHKPTRQ(col_lens);
285416022c9SBarry Smith   col_lens[0] = MAT_COOKIE;
286416022c9SBarry Smith   col_lens[1] = a->m;
287416022c9SBarry Smith   col_lens[2] = a->n;
288416022c9SBarry Smith   col_lens[3] = a->nz;
289416022c9SBarry Smith 
290416022c9SBarry Smith   /* store lengths of each row and write (including header) to file */
291416022c9SBarry Smith   for ( i=0; i<a->m; i++ ) {
292416022c9SBarry Smith     col_lens[4+i] = a->i[i+1] - a->i[i];
29317ab2063SBarry Smith   }
29477c4ece6SBarry Smith   ierr = PetscBinaryWrite(fd,col_lens,4+a->m,BINARY_INT,1); CHKERRQ(ierr);
2950452661fSBarry Smith   PetscFree(col_lens);
296416022c9SBarry Smith 
297416022c9SBarry Smith   /* store column indices (zero start index) */
298416022c9SBarry Smith   if (a->indexshift) {
299416022c9SBarry Smith     for ( i=0; i<a->nz; i++ ) a->j[i]--;
30017ab2063SBarry Smith   }
30177c4ece6SBarry Smith   ierr = PetscBinaryWrite(fd,a->j,a->nz,BINARY_INT,0); CHKERRQ(ierr);
302416022c9SBarry Smith   if (a->indexshift) {
303416022c9SBarry Smith     for ( i=0; i<a->nz; i++ ) a->j[i]++;
30417ab2063SBarry Smith   }
305416022c9SBarry Smith 
306416022c9SBarry Smith   /* store nonzero values */
30777c4ece6SBarry Smith   ierr = PetscBinaryWrite(fd,a->a,a->nz,BINARY_SCALAR,0); CHKERRQ(ierr);
30817ab2063SBarry Smith   return 0;
30917ab2063SBarry Smith }
310416022c9SBarry Smith 
3115615d1e5SSatish Balay #undef __FUNC__
3125615d1e5SSatish Balay #define __FUNC__ "MatView_SeqAIJ_ASCII"
3138f6be9afSLois Curfman McInnes extern int MatView_SeqAIJ_ASCII(Mat A,Viewer viewer)
314416022c9SBarry Smith {
315416022c9SBarry Smith   Mat_SeqAIJ  *a = (Mat_SeqAIJ *) A->data;
316496e697eSBarry Smith   int         ierr, i,j, m = a->m, shift = a->indexshift, format, flg1,flg2;
31717ab2063SBarry Smith   FILE        *fd;
31817ab2063SBarry Smith   char        *outputname;
31917ab2063SBarry Smith 
32090ace30eSBarry Smith   ierr = ViewerASCIIGetPointer(viewer,&fd); CHKERRQ(ierr);
321416022c9SBarry Smith   ierr = ViewerFileGetOutputname_Private(viewer,&outputname); CHKERRQ(ierr);
32290ace30eSBarry Smith   ierr = ViewerGetFormat(viewer,&format);
323a93ec695SBarry Smith   if (format == VIEWER_FORMAT_ASCII_INFO) {
32495e01e2fSLois Curfman McInnes     return 0;
32595e01e2fSLois Curfman McInnes   }
326a93ec695SBarry Smith   else if (format == VIEWER_FORMAT_ASCII_INFO_LONG) {
327496e697eSBarry Smith     ierr = OptionsHasName(PETSC_NULL,"-mat_aij_no_inode",&flg1); CHKERRQ(ierr);
328496e697eSBarry Smith     ierr = OptionsHasName(PETSC_NULL,"-mat_no_unroll",&flg2); CHKERRQ(ierr);
329496e697eSBarry Smith     if (flg1 || flg2) fprintf(fd,"  not using I-node routines\n");
33095e01e2fSLois Curfman McInnes     else     fprintf(fd,"  using I-node routines: found %d nodes, limit used is %d\n",
33195e01e2fSLois Curfman McInnes         a->inode.node_count,a->inode.limit);
33217ab2063SBarry Smith   }
333a93ec695SBarry Smith   else if (format == VIEWER_FORMAT_ASCII_MATLAB) {
334416022c9SBarry Smith     fprintf(fd,"%% Size = %d %d \n",m,a->n);
3354e220ebcSLois Curfman McInnes     fprintf(fd,"%% Nonzeros = %d \n",a->nz);
3364e220ebcSLois Curfman McInnes     fprintf(fd,"zzz = zeros(%d,3);\n",a->nz);
33717ab2063SBarry Smith     fprintf(fd,"zzz = [\n");
33817ab2063SBarry Smith 
33917ab2063SBarry Smith     for (i=0; i<m; i++) {
340416022c9SBarry Smith       for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
34117ab2063SBarry Smith #if defined(PETSC_COMPLEX)
3426945ee14SBarry Smith         fprintf(fd,"%d %d  %18.16e + %18.16e i \n",i+1,a->j[j]+!shift,real(a->a[j]),
343416022c9SBarry Smith                    imag(a->a[j]));
34417ab2063SBarry Smith #else
3457a743949SBarry Smith         fprintf(fd,"%d %d  %18.16e\n", i+1, a->j[j]+!shift, a->a[j]);
34617ab2063SBarry Smith #endif
34717ab2063SBarry Smith       }
34817ab2063SBarry Smith     }
34917ab2063SBarry Smith     fprintf(fd,"];\n %s = spconvert(zzz);\n",outputname);
35017ab2063SBarry Smith   }
351a93ec695SBarry Smith   else if (format == VIEWER_FORMAT_ASCII_COMMON) {
35244cd7ae7SLois Curfman McInnes     for ( i=0; i<m; i++ ) {
35344cd7ae7SLois Curfman McInnes       fprintf(fd,"row %d:",i);
35444cd7ae7SLois Curfman McInnes       for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
35544cd7ae7SLois Curfman McInnes #if defined(PETSC_COMPLEX)
35644cd7ae7SLois Curfman McInnes         if (imag(a->a[j]) != 0.0 && real(a->a[j]) != 0.0)
35744cd7ae7SLois Curfman McInnes           fprintf(fd," %d %g + %g i",a->j[j]+shift,real(a->a[j]),imag(a->a[j]));
35844cd7ae7SLois Curfman McInnes         else if (real(a->a[j]) != 0.0)
35944cd7ae7SLois Curfman McInnes           fprintf(fd," %d %g ",a->j[j]+shift,real(a->a[j]));
36044cd7ae7SLois Curfman McInnes #else
36144cd7ae7SLois Curfman McInnes         if (a->a[j] != 0.0) fprintf(fd," %d %g ",a->j[j]+shift,a->a[j]);
36244cd7ae7SLois Curfman McInnes #endif
36344cd7ae7SLois Curfman McInnes       }
36444cd7ae7SLois Curfman McInnes       fprintf(fd,"\n");
36544cd7ae7SLois Curfman McInnes     }
36644cd7ae7SLois Curfman McInnes   }
36717ab2063SBarry Smith   else {
36817ab2063SBarry Smith     for ( i=0; i<m; i++ ) {
36917ab2063SBarry Smith       fprintf(fd,"row %d:",i);
370416022c9SBarry Smith       for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
37117ab2063SBarry Smith #if defined(PETSC_COMPLEX)
372416022c9SBarry Smith         if (imag(a->a[j]) != 0.0) {
373416022c9SBarry Smith           fprintf(fd," %d %g + %g i",a->j[j]+shift,real(a->a[j]),imag(a->a[j]));
37417ab2063SBarry Smith         }
37517ab2063SBarry Smith         else {
376416022c9SBarry Smith           fprintf(fd," %d %g ",a->j[j]+shift,real(a->a[j]));
37717ab2063SBarry Smith         }
37817ab2063SBarry Smith #else
379416022c9SBarry Smith         fprintf(fd," %d %g ",a->j[j]+shift,a->a[j]);
38017ab2063SBarry Smith #endif
38117ab2063SBarry Smith       }
38217ab2063SBarry Smith       fprintf(fd,"\n");
38317ab2063SBarry Smith     }
38417ab2063SBarry Smith   }
38517ab2063SBarry Smith   fflush(fd);
386416022c9SBarry Smith   return 0;
387416022c9SBarry Smith }
388416022c9SBarry Smith 
3895615d1e5SSatish Balay #undef __FUNC__
3905615d1e5SSatish Balay #define __FUNC__ "MatView_SeqAIJ_Draw"
3918f6be9afSLois Curfman McInnes extern int MatView_SeqAIJ_Draw(Mat A,Viewer viewer)
392416022c9SBarry Smith {
393416022c9SBarry Smith   Mat_SeqAIJ  *a = (Mat_SeqAIJ *) A->data;
394cddf8d76SBarry Smith   int         ierr, i,j, m = a->m, shift = a->indexshift,pause,color;
3950513a670SBarry Smith   int         format;
39694a9d846SBarry Smith   double      xl,yl,xr,yr,w,h,xc,yc,scale = 1.0,x_l,x_r,y_l,y_r,maxv = 0.0;
397bcd2baecSBarry Smith   Draw        draw;
398cddf8d76SBarry Smith   DrawButton  button;
39919bcc07fSBarry Smith   PetscTruth  isnull;
400cddf8d76SBarry Smith 
4010513a670SBarry Smith   ierr = ViewerDrawGetDraw(viewer,&draw); CHKERRQ(ierr);
4020513a670SBarry Smith   ierr = DrawClear(draw); CHKERRQ(ierr);
4030513a670SBarry Smith   ierr = ViewerGetFormat(viewer,&format); CHKERRQ(ierr);
40419bcc07fSBarry Smith   ierr = DrawIsNull(draw,&isnull); CHKERRQ(ierr); if (isnull) return 0;
40519bcc07fSBarry Smith 
406416022c9SBarry Smith   xr  = a->n; yr = a->m; h = yr/10.0; w = xr/10.0;
407416022c9SBarry Smith   xr += w;    yr += h;  xl = -w;     yl = -h;
408416022c9SBarry Smith   ierr = DrawSetCoordinates(draw,xl,yl,xr,yr); CHKERRQ(ierr);
409416022c9SBarry Smith   /* loop over matrix elements drawing boxes */
4100513a670SBarry Smith 
4110513a670SBarry Smith   if (format != VIEWER_FORMAT_DRAW_CONTOUR) {
4120513a670SBarry Smith     /* Blue for negative, Cyan for zero and  Red for positive */
413cddf8d76SBarry Smith     color = DRAW_BLUE;
414416022c9SBarry Smith     for ( i=0; i<m; i++ ) {
415cddf8d76SBarry Smith       y_l = m - i - 1.0; y_r = y_l + 1.0;
416416022c9SBarry Smith       for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
417cddf8d76SBarry Smith         x_l = a->j[j] + shift; x_r = x_l + 1.0;
418cddf8d76SBarry Smith #if defined(PETSC_COMPLEX)
419cddf8d76SBarry Smith         if (real(a->a[j]) >=  0.) continue;
420cddf8d76SBarry Smith #else
421cddf8d76SBarry Smith         if (a->a[j] >=  0.) continue;
422cddf8d76SBarry Smith #endif
423cddf8d76SBarry Smith         DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);
424cddf8d76SBarry Smith       }
425cddf8d76SBarry Smith     }
426cddf8d76SBarry Smith     color = DRAW_CYAN;
427cddf8d76SBarry Smith     for ( i=0; i<m; i++ ) {
428cddf8d76SBarry Smith       y_l = m - i - 1.0; y_r = y_l + 1.0;
429cddf8d76SBarry Smith       for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
430cddf8d76SBarry Smith         x_l = a->j[j] + shift; x_r = x_l + 1.0;
431cddf8d76SBarry Smith         if (a->a[j] !=  0.) continue;
432cddf8d76SBarry Smith         DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);
433cddf8d76SBarry Smith       }
434cddf8d76SBarry Smith     }
435cddf8d76SBarry Smith     color = DRAW_RED;
436cddf8d76SBarry Smith     for ( i=0; i<m; i++ ) {
437cddf8d76SBarry Smith       y_l = m - i - 1.0; y_r = y_l + 1.0;
438cddf8d76SBarry Smith       for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
439cddf8d76SBarry Smith         x_l = a->j[j] + shift; x_r = x_l + 1.0;
440cddf8d76SBarry Smith #if defined(PETSC_COMPLEX)
441cddf8d76SBarry Smith         if (real(a->a[j]) <=  0.) continue;
442cddf8d76SBarry Smith #else
443cddf8d76SBarry Smith         if (a->a[j] <=  0.) continue;
444cddf8d76SBarry Smith #endif
445cddf8d76SBarry Smith         DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);
446416022c9SBarry Smith       }
447416022c9SBarry Smith     }
4480513a670SBarry Smith   } else {
4490513a670SBarry Smith     /* use contour shading to indicate magnitude of values */
4500513a670SBarry Smith     /* first determine max of all nonzero values */
4510513a670SBarry Smith     int    nz = a->nz,count;
4520513a670SBarry Smith     Draw   popup;
4530513a670SBarry Smith 
4540513a670SBarry Smith     for ( i=0; i<nz; i++ ) {
4550513a670SBarry Smith       if (PetscAbsScalar(a->a[i]) > maxv) maxv = PetscAbsScalar(a->a[i]);
4560513a670SBarry Smith     }
4570513a670SBarry Smith     ierr = DrawCreatePopUp(draw,&popup); CHKERRQ(ierr);
4580513a670SBarry Smith     ierr = DrawScalePopup(popup,0.0,maxv); CHKERRQ(ierr);
4590513a670SBarry Smith     count = 0;
4600513a670SBarry Smith     for ( i=0; i<m; i++ ) {
4610513a670SBarry Smith       y_l = m - i - 1.0; y_r = y_l + 1.0;
4620513a670SBarry Smith       for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
4630513a670SBarry Smith         x_l = a->j[j] + shift; x_r = x_l + 1.0;
4640513a670SBarry Smith         color = 32 + (int) ((200.0 - 32.0)*PetscAbsScalar(a->a[count])/maxv);
4650513a670SBarry Smith         DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);
4660513a670SBarry Smith         count++;
4670513a670SBarry Smith       }
4680513a670SBarry Smith     }
4690513a670SBarry Smith   }
470416022c9SBarry Smith   DrawFlush(draw);
471cddf8d76SBarry Smith   DrawGetPause(draw,&pause);
472cddf8d76SBarry Smith   if (pause >= 0) { PetscSleep(pause); return 0;}
473cddf8d76SBarry Smith 
474cddf8d76SBarry Smith   /* allow the matrix to zoom or shrink */
4756945ee14SBarry Smith   ierr = DrawCheckResizedWindow(draw);
476cddf8d76SBarry Smith   ierr = DrawGetMouseButton(draw,&button,&xc,&yc,0,0);
477cddf8d76SBarry Smith   while (button != BUTTON_RIGHT) {
478cddf8d76SBarry Smith     DrawClear(draw);
479cddf8d76SBarry Smith     if (button == BUTTON_LEFT) scale = .5;
480cddf8d76SBarry Smith     else if (button == BUTTON_CENTER) scale = 2.;
481cddf8d76SBarry Smith     xl = scale*(xl + w - xc) + xc - w*scale;
482cddf8d76SBarry Smith     xr = scale*(xr - w - xc) + xc + w*scale;
483cddf8d76SBarry Smith     yl = scale*(yl + h - yc) + yc - h*scale;
484cddf8d76SBarry Smith     yr = scale*(yr - h - yc) + yc + h*scale;
485cddf8d76SBarry Smith     w *= scale; h *= scale;
486cddf8d76SBarry Smith     ierr = DrawSetCoordinates(draw,xl,yl,xr,yr); CHKERRQ(ierr);
4870513a670SBarry Smith     if (format != VIEWER_FORMAT_DRAW_CONTOUR) {
4880513a670SBarry Smith       /* Blue for negative, Cyan for zero and  Red for positive */
489cddf8d76SBarry Smith       color = DRAW_BLUE;
490cddf8d76SBarry Smith       for ( i=0; i<m; i++ ) {
491cddf8d76SBarry Smith         y_l = m - i - 1.0; y_r = y_l + 1.0;
492cddf8d76SBarry Smith         for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
493cddf8d76SBarry Smith           x_l = a->j[j] + shift; x_r = x_l + 1.0;
494cddf8d76SBarry Smith #if defined(PETSC_COMPLEX)
495cddf8d76SBarry Smith           if (real(a->a[j]) >=  0.) continue;
496cddf8d76SBarry Smith #else
497cddf8d76SBarry Smith           if (a->a[j] >=  0.) continue;
498cddf8d76SBarry Smith #endif
499cddf8d76SBarry Smith           DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);
500cddf8d76SBarry Smith         }
501cddf8d76SBarry Smith       }
502cddf8d76SBarry Smith       color = DRAW_CYAN;
503cddf8d76SBarry Smith       for ( i=0; i<m; i++ ) {
504cddf8d76SBarry Smith         y_l = m - i - 1.0; y_r = y_l + 1.0;
505cddf8d76SBarry Smith         for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
506cddf8d76SBarry Smith           x_l = a->j[j] + shift; x_r = x_l + 1.0;
507cddf8d76SBarry Smith           if (a->a[j] !=  0.) continue;
508cddf8d76SBarry Smith           DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);
509cddf8d76SBarry Smith         }
510cddf8d76SBarry Smith       }
511cddf8d76SBarry Smith       color = DRAW_RED;
512cddf8d76SBarry Smith       for ( i=0; i<m; i++ ) {
513cddf8d76SBarry Smith         y_l = m - i - 1.0; y_r = y_l + 1.0;
514cddf8d76SBarry Smith         for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
515cddf8d76SBarry Smith           x_l = a->j[j] + shift; x_r = x_l + 1.0;
516cddf8d76SBarry Smith #if defined(PETSC_COMPLEX)
517cddf8d76SBarry Smith           if (real(a->a[j]) <=  0.) continue;
518cddf8d76SBarry Smith #else
519cddf8d76SBarry Smith           if (a->a[j] <=  0.) continue;
520cddf8d76SBarry Smith #endif
521cddf8d76SBarry Smith           DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);
522cddf8d76SBarry Smith         }
523cddf8d76SBarry Smith       }
5240513a670SBarry Smith     } else {
5250513a670SBarry Smith       /* use contour shading to indicate magnitude of values */
5260513a670SBarry Smith       int count = 0;
5270513a670SBarry Smith       for ( i=0; i<m; i++ ) {
5280513a670SBarry Smith         y_l = m - i - 1.0; y_r = y_l + 1.0;
5290513a670SBarry Smith         for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
5300513a670SBarry Smith           x_l = a->j[j] + shift; x_r = x_l + 1.0;
5310513a670SBarry Smith           color = 32 + (int) ((200.0 - 32.0)*PetscAbsScalar(a->a[count])/maxv);
5320513a670SBarry Smith           DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);
5330513a670SBarry Smith           count++;
5340513a670SBarry Smith         }
5350513a670SBarry Smith       }
5360513a670SBarry Smith     }
5370513a670SBarry Smith 
5386945ee14SBarry Smith     ierr = DrawCheckResizedWindow(draw);
539cddf8d76SBarry Smith     ierr = DrawGetMouseButton(draw,&button,&xc,&yc,0,0);
540cddf8d76SBarry Smith   }
541416022c9SBarry Smith   return 0;
542416022c9SBarry Smith }
543416022c9SBarry Smith 
5445615d1e5SSatish Balay #undef __FUNC__
545c22c1629SBarry Smith #define __FUNC__ "MatView_SeqAIJ" /* ADIC Ignore */
5468f6be9afSLois Curfman McInnes int MatView_SeqAIJ(PetscObject obj,Viewer viewer)
547416022c9SBarry Smith {
548416022c9SBarry Smith   Mat         A = (Mat) obj;
549416022c9SBarry Smith   Mat_SeqAIJ  *a = (Mat_SeqAIJ*) A->data;
550bcd2baecSBarry Smith   ViewerType  vtype;
551bcd2baecSBarry Smith   int         ierr;
552416022c9SBarry Smith 
553bcd2baecSBarry Smith   ierr = ViewerGetType(viewer,&vtype); CHKERRQ(ierr);
554bcd2baecSBarry Smith   if (vtype == MATLAB_VIEWER) {
555416022c9SBarry Smith     return ViewerMatlabPutSparse_Private(viewer,a->m,a->n,a->nz,a->a,a->i,a->j);
556416022c9SBarry Smith   }
557bcd2baecSBarry Smith   else if (vtype == ASCII_FILE_VIEWER || vtype == ASCII_FILES_VIEWER){
558416022c9SBarry Smith     return MatView_SeqAIJ_ASCII(A,viewer);
559416022c9SBarry Smith   }
560bcd2baecSBarry Smith   else if (vtype == BINARY_FILE_VIEWER) {
561416022c9SBarry Smith     return MatView_SeqAIJ_Binary(A,viewer);
562416022c9SBarry Smith   }
563bcd2baecSBarry Smith   else if (vtype == DRAW_VIEWER) {
564bcd2baecSBarry Smith     return MatView_SeqAIJ_Draw(A,viewer);
56517ab2063SBarry Smith   }
56617ab2063SBarry Smith   return 0;
56717ab2063SBarry Smith }
56819bcc07fSBarry Smith 
569c456f294SBarry Smith extern int Mat_AIJ_CheckInode(Mat);
5705615d1e5SSatish Balay #undef __FUNC__
5715615d1e5SSatish Balay #define __FUNC__ "MatAssemblyEnd_SeqAIJ"
5728f6be9afSLois Curfman McInnes int MatAssemblyEnd_SeqAIJ(Mat A,MatAssemblyType mode)
57317ab2063SBarry Smith {
574416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
57541c01911SSatish Balay   int        fshift = 0,i,j,*ai = a->i, *aj = a->j, *imax = a->imax,ierr;
57694a9d846SBarry Smith   int        m = a->m, *ip, N, *ailen = a->ilen,shift = a->indexshift,rmax;
577416022c9SBarry Smith   Scalar     *aa = a->a, *ap;
57817ab2063SBarry Smith 
5796d4a8577SBarry Smith   if (mode == MAT_FLUSH_ASSEMBLY) return 0;
58017ab2063SBarry Smith 
58194a9d846SBarry Smith   rmax = ailen[0]; /* determine row with most nonzeros */
58217ab2063SBarry Smith   for ( i=1; i<m; i++ ) {
583416022c9SBarry Smith     /* move each row back by the amount of empty slots (fshift) before it*/
58417ab2063SBarry Smith     fshift += imax[i-1] - ailen[i-1];
58594a9d846SBarry Smith     rmax   = PetscMax(rmax,ailen[i]);
58617ab2063SBarry Smith     if (fshift) {
587416022c9SBarry Smith       ip = aj + ai[i] + shift; ap = aa + ai[i] + shift;
58817ab2063SBarry Smith       N = ailen[i];
58917ab2063SBarry Smith       for ( j=0; j<N; j++ ) {
59017ab2063SBarry Smith         ip[j-fshift] = ip[j];
59117ab2063SBarry Smith         ap[j-fshift] = ap[j];
59217ab2063SBarry Smith       }
59317ab2063SBarry Smith     }
59417ab2063SBarry Smith     ai[i] = ai[i-1] + ailen[i-1];
59517ab2063SBarry Smith   }
59617ab2063SBarry Smith   if (m) {
59717ab2063SBarry Smith     fshift += imax[m-1] - ailen[m-1];
59817ab2063SBarry Smith     ai[m] = ai[m-1] + ailen[m-1];
59917ab2063SBarry Smith   }
60017ab2063SBarry Smith   /* reset ilen and imax for each row */
60117ab2063SBarry Smith   for ( i=0; i<m; i++ ) {
60217ab2063SBarry Smith     ailen[i] = imax[i] = ai[i+1] - ai[i];
60317ab2063SBarry Smith   }
604416022c9SBarry Smith   a->nz = ai[m] + shift;
60517ab2063SBarry Smith 
60617ab2063SBarry Smith   /* diagonals may have moved, so kill the diagonal pointers */
607416022c9SBarry Smith   if (fshift && a->diag) {
6080452661fSBarry Smith     PetscFree(a->diag);
609416022c9SBarry Smith     PLogObjectMemory(A,-(m+1)*sizeof(int));
610416022c9SBarry Smith     a->diag = 0;
61117ab2063SBarry Smith   }
6124e220ebcSLois Curfman McInnes   PLogInfo(A,"MatAssemblyEnd_SeqAIJ:Matrix size: %d X %d; storage space: %d unneeded, %d used\n",
6134e220ebcSLois Curfman McInnes            m,a->n,fshift,a->nz);
6144e220ebcSLois Curfman McInnes   PLogInfo(A,"MatAssemblyEnd_SeqAIJ:Number of mallocs during MatSetValues is %d\n",
615b810aeb4SBarry Smith            a->reallocs);
61694a9d846SBarry Smith   PLogInfo(A,"MatAssemblyEnd_SeqAIJ:Most nonzeros in any row is %d\n",rmax);
617dd5f02e7SSatish Balay   a->reallocs          = 0;
6184e220ebcSLois Curfman McInnes   A->info.nz_unneeded  = (double)fshift;
6194e220ebcSLois Curfman McInnes 
62076dd722bSSatish Balay   /* check out for identical nodes. If found, use inode functions */
62141c01911SSatish Balay   ierr = Mat_AIJ_CheckInode(A); CHKERRQ(ierr);
62217ab2063SBarry Smith   return 0;
62317ab2063SBarry Smith }
62417ab2063SBarry Smith 
6255615d1e5SSatish Balay #undef __FUNC__
6265615d1e5SSatish Balay #define __FUNC__ "MatZeroEntries_SeqAIJ"
6278f6be9afSLois Curfman McInnes int MatZeroEntries_SeqAIJ(Mat A)
62817ab2063SBarry Smith {
629416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
630cddf8d76SBarry Smith   PetscMemzero(a->a,(a->i[a->m]+a->indexshift)*sizeof(Scalar));
63117ab2063SBarry Smith   return 0;
63217ab2063SBarry Smith }
633416022c9SBarry Smith 
6345615d1e5SSatish Balay #undef __FUNC__
6355615d1e5SSatish Balay #define __FUNC__ "MatDestroy_SeqAIJ"
63617ab2063SBarry Smith int MatDestroy_SeqAIJ(PetscObject obj)
63717ab2063SBarry Smith {
638416022c9SBarry Smith   Mat        A  = (Mat) obj;
639416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
64090f02eecSBarry Smith   int        ierr;
641d5d45c9bSBarry Smith 
64217ab2063SBarry Smith #if defined(PETSC_LOG)
643416022c9SBarry Smith   PLogObjectState(obj,"Rows=%d, Cols=%d, NZ=%d",a->m,a->n,a->nz);
64417ab2063SBarry Smith #endif
6450452661fSBarry Smith   PetscFree(a->a);
6460452661fSBarry Smith   if (!a->singlemalloc) { PetscFree(a->i); PetscFree(a->j);}
6470452661fSBarry Smith   if (a->diag) PetscFree(a->diag);
6480452661fSBarry Smith   if (a->ilen) PetscFree(a->ilen);
6490452661fSBarry Smith   if (a->imax) PetscFree(a->imax);
6500452661fSBarry Smith   if (a->solve_work) PetscFree(a->solve_work);
65176dd722bSSatish Balay   if (a->inode.size) PetscFree(a->inode.size);
6520452661fSBarry Smith   PetscFree(a);
65390f02eecSBarry Smith   if (A->mapping) {
65490f02eecSBarry Smith     ierr = ISLocalToGlobalMappingDestroy(A->mapping); CHKERRQ(ierr);
65590f02eecSBarry Smith   }
656f2655603SLois Curfman McInnes   PLogObjectDestroy(A);
657f2655603SLois Curfman McInnes   PetscHeaderDestroy(A);
65817ab2063SBarry Smith   return 0;
65917ab2063SBarry Smith }
66017ab2063SBarry Smith 
6615615d1e5SSatish Balay #undef __FUNC__
6625615d1e5SSatish Balay #define __FUNC__ "MatCompress_SeqAIJ"
6638f6be9afSLois Curfman McInnes int MatCompress_SeqAIJ(Mat A)
66417ab2063SBarry Smith {
66517ab2063SBarry Smith   return 0;
66617ab2063SBarry Smith }
66717ab2063SBarry Smith 
6685615d1e5SSatish Balay #undef __FUNC__
6695615d1e5SSatish Balay #define __FUNC__ "MatSetOption_SeqAIJ"
6708f6be9afSLois Curfman McInnes int MatSetOption_SeqAIJ(Mat A,MatOption op)
67117ab2063SBarry Smith {
672416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
6736d4a8577SBarry Smith   if      (op == MAT_ROW_ORIENTED)               a->roworiented = 1;
6746d4a8577SBarry Smith   else if (op == MAT_COLUMN_ORIENTED)            a->roworiented = 0;
6756d4a8577SBarry Smith   else if (op == MAT_COLUMNS_SORTED)             a->sorted      = 1;
676219d9a1aSLois Curfman McInnes   else if (op == MAT_COLUMNS_UNSORTED)           a->sorted      = 0;
6776d4a8577SBarry Smith   else if (op == MAT_NO_NEW_NONZERO_LOCATIONS)   a->nonew       = 1;
678c2653b3dSLois Curfman McInnes   else if (op == MAT_NEW_NONZERO_LOCATION_ERROR) a->nonew       = -1;
6796d4a8577SBarry Smith   else if (op == MAT_YES_NEW_NONZERO_LOCATIONS)  a->nonew       = 0;
6806d4a8577SBarry Smith   else if (op == MAT_ROWS_SORTED ||
681219d9a1aSLois Curfman McInnes            op == MAT_ROWS_UNSORTED ||
6826d4a8577SBarry Smith            op == MAT_SYMMETRIC ||
6836d4a8577SBarry Smith            op == MAT_STRUCTURALLY_SYMMETRIC ||
68490f02eecSBarry Smith            op == MAT_YES_NEW_DIAGONALS ||
6852b362799SSatish Balay            op == MAT_IGNORE_OFF_PROC_ENTRIES)
68694a424c1SBarry Smith     PLogInfo(A,"Info:MatSetOption_SeqAIJ:Option ignored\n");
6876d4a8577SBarry Smith   else if (op == MAT_NO_NEW_DIAGONALS)
688e3372554SBarry Smith     {SETERRQ(PETSC_ERR_SUP,0,"MAT_NO_NEW_DIAGONALS");}
6896d4a8577SBarry Smith   else if (op == MAT_INODE_LIMIT_1)            a->inode.limit  = 1;
6906d4a8577SBarry Smith   else if (op == MAT_INODE_LIMIT_2)            a->inode.limit  = 2;
6916d4a8577SBarry Smith   else if (op == MAT_INODE_LIMIT_3)            a->inode.limit  = 3;
6926d4a8577SBarry Smith   else if (op == MAT_INODE_LIMIT_4)            a->inode.limit  = 4;
6936d4a8577SBarry Smith   else if (op == MAT_INODE_LIMIT_5)            a->inode.limit  = 5;
694e2f28af5SBarry Smith   else
695e3372554SBarry Smith     {SETERRQ(PETSC_ERR_SUP,0,"unknown option");}
69617ab2063SBarry Smith   return 0;
69717ab2063SBarry Smith }
69817ab2063SBarry Smith 
6995615d1e5SSatish Balay #undef __FUNC__
7005615d1e5SSatish Balay #define __FUNC__ "MatGetDiagonal_SeqAIJ"
7018f6be9afSLois Curfman McInnes int MatGetDiagonal_SeqAIJ(Mat A,Vec v)
70217ab2063SBarry Smith {
703416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
704416022c9SBarry Smith   int        i,j, n,shift = a->indexshift;
70517ab2063SBarry Smith   Scalar     *x, zero = 0.0;
70617ab2063SBarry Smith 
70717ab2063SBarry Smith   VecSet(&zero,v);
70890f02eecSBarry Smith   VecGetArray_Fast(v,x); VecGetLocalSize(v,&n);
709e3372554SBarry Smith   if (n != a->m) SETERRQ(1,0,"Nonconforming matrix and vector");
710416022c9SBarry Smith   for ( i=0; i<a->m; i++ ) {
711416022c9SBarry Smith     for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
712416022c9SBarry Smith       if (a->j[j]+shift == i) {
713416022c9SBarry Smith         x[i] = a->a[j];
71417ab2063SBarry Smith         break;
71517ab2063SBarry Smith       }
71617ab2063SBarry Smith     }
71717ab2063SBarry Smith   }
71817ab2063SBarry Smith   return 0;
71917ab2063SBarry Smith }
72017ab2063SBarry Smith 
72117ab2063SBarry Smith /* -------------------------------------------------------*/
72217ab2063SBarry Smith /* Should check that shapes of vectors and matrices match */
72317ab2063SBarry Smith /* -------------------------------------------------------*/
7245615d1e5SSatish Balay #undef __FUNC__
7255615d1e5SSatish Balay #define __FUNC__ "MatMultTrans_SeqAIJ"
72644cd7ae7SLois Curfman McInnes int MatMultTrans_SeqAIJ(Mat A,Vec xx,Vec yy)
72717ab2063SBarry Smith {
728416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
72917ab2063SBarry Smith   Scalar     *x, *y, *v, alpha;
730416022c9SBarry Smith   int        m = a->m, n, i, *idx, shift = a->indexshift;
73117ab2063SBarry Smith 
73290f02eecSBarry Smith   VecGetArray_Fast(xx,x); VecGetArray_Fast(yy,y);
733cddf8d76SBarry Smith   PetscMemzero(y,a->n*sizeof(Scalar));
73417ab2063SBarry Smith   y = y + shift; /* shift for Fortran start by 1 indexing */
73517ab2063SBarry Smith   for ( i=0; i<m; i++ ) {
736416022c9SBarry Smith     idx   = a->j + a->i[i] + shift;
737416022c9SBarry Smith     v     = a->a + a->i[i] + shift;
738416022c9SBarry Smith     n     = a->i[i+1] - a->i[i];
73917ab2063SBarry Smith     alpha = x[i];
74017ab2063SBarry Smith     while (n-->0) {y[*idx++] += alpha * *v++;}
74117ab2063SBarry Smith   }
742416022c9SBarry Smith   PLogFlops(2*a->nz - a->n);
74317ab2063SBarry Smith   return 0;
74417ab2063SBarry Smith }
745d5d45c9bSBarry Smith 
7465615d1e5SSatish Balay #undef __FUNC__
7475615d1e5SSatish Balay #define __FUNC__ "MatMultTransAdd_SeqAIJ"
74844cd7ae7SLois Curfman McInnes int MatMultTransAdd_SeqAIJ(Mat A,Vec xx,Vec zz,Vec yy)
74917ab2063SBarry Smith {
750416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
75117ab2063SBarry Smith   Scalar     *x, *y, *v, alpha;
752416022c9SBarry Smith   int        m = a->m, n, i, *idx,shift = a->indexshift;
75317ab2063SBarry Smith 
75490f02eecSBarry Smith   VecGetArray_Fast(xx,x); VecGetArray_Fast(yy,y);
75517ab2063SBarry Smith   if (zz != yy) VecCopy(zz,yy);
75617ab2063SBarry Smith   y = y + shift; /* shift for Fortran start by 1 indexing */
75717ab2063SBarry Smith   for ( i=0; i<m; i++ ) {
758416022c9SBarry Smith     idx   = a->j + a->i[i] + shift;
759416022c9SBarry Smith     v     = a->a + a->i[i] + shift;
760416022c9SBarry Smith     n     = a->i[i+1] - a->i[i];
76117ab2063SBarry Smith     alpha = x[i];
76217ab2063SBarry Smith     while (n-->0) {y[*idx++] += alpha * *v++;}
76317ab2063SBarry Smith   }
76490f02eecSBarry Smith   PLogFlops(2*a->nz);
76517ab2063SBarry Smith   return 0;
76617ab2063SBarry Smith }
76717ab2063SBarry Smith 
7685615d1e5SSatish Balay #undef __FUNC__
7695615d1e5SSatish Balay #define __FUNC__ "MatMult_SeqAIJ"
77044cd7ae7SLois Curfman McInnes int MatMult_SeqAIJ(Mat A,Vec xx,Vec yy)
77117ab2063SBarry Smith {
772416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
77317ab2063SBarry Smith   Scalar     *x, *y, *v, sum;
774*9ea0dfa2SSatish Balay   int        m = a->m, n, i, *idx, shift = a->indexshift,*ii,jrow,j;
77517ab2063SBarry Smith 
77690f02eecSBarry Smith   VecGetArray_Fast(xx,x); VecGetArray_Fast(yy,y);
77717ab2063SBarry Smith   x    = x + shift; /* shift for Fortran start by 1 indexing */
778416022c9SBarry Smith   idx  = a->j;
779416022c9SBarry Smith   v    = a->a;
780416022c9SBarry Smith   ii   = a->i;
78117ab2063SBarry Smith   for ( i=0; i<m; i++ ) {
782*9ea0dfa2SSatish Balay     jrow = ii[i];
783*9ea0dfa2SSatish Balay     n    = ii[i+1] - jrow;
78417ab2063SBarry Smith     sum  = 0.0;
785*9ea0dfa2SSatish Balay     /* while (n--) sum += *v++ * x[*idx++]; */
786*9ea0dfa2SSatish Balay     for ( j=0; j<n; j++) {
787*9ea0dfa2SSatish Balay       sum += v[jrow]*x[idx[jrow]]; jrow++;
788*9ea0dfa2SSatish Balay      }
78917ab2063SBarry Smith     y[i] = sum;
79017ab2063SBarry Smith   }
791416022c9SBarry Smith   PLogFlops(2*a->nz - m);
79217ab2063SBarry Smith   return 0;
79317ab2063SBarry Smith }
79417ab2063SBarry Smith 
7955615d1e5SSatish Balay #undef __FUNC__
7965615d1e5SSatish Balay #define __FUNC__ "MatMultAdd_SeqAIJ"
79744cd7ae7SLois Curfman McInnes int MatMultAdd_SeqAIJ(Mat A,Vec xx,Vec yy,Vec zz)
79817ab2063SBarry Smith {
799416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
80017ab2063SBarry Smith   Scalar     *x, *y, *z, *v, sum;
801*9ea0dfa2SSatish Balay   int        m = a->m, n, i, *idx, shift = a->indexshift,*ii,jrow,j;
802*9ea0dfa2SSatish Balay 
80317ab2063SBarry Smith 
80490f02eecSBarry Smith   VecGetArray_Fast(xx,x); VecGetArray_Fast(yy,y); VecGetArray_Fast(zz,z);
80517ab2063SBarry Smith   x    = x + shift; /* shift for Fortran start by 1 indexing */
806cddf8d76SBarry Smith   idx  = a->j;
807cddf8d76SBarry Smith   v    = a->a;
808cddf8d76SBarry Smith   ii   = a->i;
80917ab2063SBarry Smith   for ( i=0; i<m; i++ ) {
810*9ea0dfa2SSatish Balay     jrow = ii[i];
811*9ea0dfa2SSatish Balay     n    = ii[i+1] - jrow;
81217ab2063SBarry Smith     sum  = y[i];
813*9ea0dfa2SSatish Balay     /* while (n--) sum += *v++ * x[*idx++]; */
814*9ea0dfa2SSatish Balay     for ( j=0; j<n; j++) {
815*9ea0dfa2SSatish Balay       sum += v[jrow]*x[idx[jrow]]; jrow++;
816*9ea0dfa2SSatish Balay      }
81717ab2063SBarry Smith     z[i] = sum;
81817ab2063SBarry Smith   }
819416022c9SBarry Smith   PLogFlops(2*a->nz);
82017ab2063SBarry Smith   return 0;
82117ab2063SBarry Smith }
82217ab2063SBarry Smith 
82317ab2063SBarry Smith /*
82417ab2063SBarry Smith      Adds diagonal pointers to sparse matrix structure.
82517ab2063SBarry Smith */
82617ab2063SBarry Smith 
8275615d1e5SSatish Balay #undef __FUNC__
8285615d1e5SSatish Balay #define __FUNC__ "MatMarkDiag_SeqAIJ"
829416022c9SBarry Smith int MatMarkDiag_SeqAIJ(Mat A)
83017ab2063SBarry Smith {
831416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
832416022c9SBarry Smith   int        i,j, *diag, m = a->m,shift = a->indexshift;
83317ab2063SBarry Smith 
8340452661fSBarry Smith   diag = (int *) PetscMalloc( (m+1)*sizeof(int)); CHKPTRQ(diag);
835416022c9SBarry Smith   PLogObjectMemory(A,(m+1)*sizeof(int));
836416022c9SBarry Smith   for ( i=0; i<a->m; i++ ) {
837416022c9SBarry Smith     for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
838416022c9SBarry Smith       if (a->j[j]+shift == i) {
83917ab2063SBarry Smith         diag[i] = j - shift;
84017ab2063SBarry Smith         break;
84117ab2063SBarry Smith       }
84217ab2063SBarry Smith     }
84317ab2063SBarry Smith   }
844416022c9SBarry Smith   a->diag = diag;
84517ab2063SBarry Smith   return 0;
84617ab2063SBarry Smith }
84717ab2063SBarry Smith 
8485615d1e5SSatish Balay #undef __FUNC__
8495615d1e5SSatish Balay #define __FUNC__ "MatRelax_SeqAIJ"
85044cd7ae7SLois Curfman McInnes int MatRelax_SeqAIJ(Mat A,Vec bb,double omega,MatSORType flag,
85117ab2063SBarry Smith                            double fshift,int its,Vec xx)
85217ab2063SBarry Smith {
853416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
854416022c9SBarry Smith   Scalar     *x, *b, *bs,  d, *xs, sum, *v = a->a,*t,scale,*ts, *xb;
855d5d45c9bSBarry Smith   int        ierr, *idx, *diag,n = a->n, m = a->m, i, shift = a->indexshift;
85617ab2063SBarry Smith 
85790f02eecSBarry Smith   VecGetArray_Fast(xx,x); VecGetArray_Fast(bb,b);
858416022c9SBarry Smith   if (!a->diag) {if ((ierr = MatMarkDiag_SeqAIJ(A))) return ierr;}
859416022c9SBarry Smith   diag = a->diag;
860416022c9SBarry Smith   xs   = x + shift; /* shifted by one for index start of a or a->j*/
86117ab2063SBarry Smith   if (flag == SOR_APPLY_UPPER) {
86217ab2063SBarry Smith    /* apply ( U + D/omega) to the vector */
86317ab2063SBarry Smith     bs = b + shift;
86417ab2063SBarry Smith     for ( i=0; i<m; i++ ) {
865416022c9SBarry Smith         d    = fshift + a->a[diag[i] + shift];
866416022c9SBarry Smith         n    = a->i[i+1] - diag[i] - 1;
867416022c9SBarry Smith         idx  = a->j + diag[i] + (!shift);
868416022c9SBarry Smith         v    = a->a + diag[i] + (!shift);
86917ab2063SBarry Smith         sum  = b[i]*d/omega;
87017ab2063SBarry Smith         SPARSEDENSEDOT(sum,bs,v,idx,n);
87117ab2063SBarry Smith         x[i] = sum;
87217ab2063SBarry Smith     }
87317ab2063SBarry Smith     return 0;
87417ab2063SBarry Smith   }
87517ab2063SBarry Smith   if (flag == SOR_APPLY_LOWER) {
876e3372554SBarry Smith     SETERRQ(1,0,"SOR_APPLY_LOWER is not done");
87717ab2063SBarry Smith   }
878416022c9SBarry Smith   else if (flag & SOR_EISENSTAT) {
87917ab2063SBarry Smith     /* Let  A = L + U + D; where L is lower trianglar,
88017ab2063SBarry Smith     U is upper triangular, E is diagonal; This routine applies
88117ab2063SBarry Smith 
88217ab2063SBarry Smith             (L + E)^{-1} A (U + E)^{-1}
88317ab2063SBarry Smith 
88417ab2063SBarry Smith     to a vector efficiently using Eisenstat's trick. This is for
88517ab2063SBarry Smith     the case of SSOR preconditioner, so E is D/omega where omega
88617ab2063SBarry Smith     is the relaxation factor.
88717ab2063SBarry Smith     */
8880452661fSBarry Smith     t = (Scalar *) PetscMalloc( m*sizeof(Scalar) ); CHKPTRQ(t);
88917ab2063SBarry Smith     scale = (2.0/omega) - 1.0;
89017ab2063SBarry Smith 
89117ab2063SBarry Smith     /*  x = (E + U)^{-1} b */
89217ab2063SBarry Smith     for ( i=m-1; i>=0; i-- ) {
893416022c9SBarry Smith       d    = fshift + a->a[diag[i] + shift];
894416022c9SBarry Smith       n    = a->i[i+1] - diag[i] - 1;
895416022c9SBarry Smith       idx  = a->j + diag[i] + (!shift);
896416022c9SBarry Smith       v    = a->a + diag[i] + (!shift);
89717ab2063SBarry Smith       sum  = b[i];
89817ab2063SBarry Smith       SPARSEDENSEMDOT(sum,xs,v,idx,n);
89917ab2063SBarry Smith       x[i] = omega*(sum/d);
90017ab2063SBarry Smith     }
90117ab2063SBarry Smith 
90217ab2063SBarry Smith     /*  t = b - (2*E - D)x */
903416022c9SBarry Smith     v = a->a;
90417ab2063SBarry Smith     for ( i=0; i<m; i++ ) { t[i] = b[i] - scale*(v[*diag++ + shift])*x[i]; }
90517ab2063SBarry Smith 
90617ab2063SBarry Smith     /*  t = (E + L)^{-1}t */
907416022c9SBarry Smith     ts = t + shift; /* shifted by one for index start of a or a->j*/
908416022c9SBarry Smith     diag = a->diag;
90917ab2063SBarry Smith     for ( i=0; i<m; i++ ) {
910416022c9SBarry Smith       d    = fshift + a->a[diag[i]+shift];
911416022c9SBarry Smith       n    = diag[i] - a->i[i];
912416022c9SBarry Smith       idx  = a->j + a->i[i] + shift;
913416022c9SBarry Smith       v    = a->a + a->i[i] + shift;
91417ab2063SBarry Smith       sum  = t[i];
91517ab2063SBarry Smith       SPARSEDENSEMDOT(sum,ts,v,idx,n);
91617ab2063SBarry Smith       t[i] = omega*(sum/d);
91717ab2063SBarry Smith     }
91817ab2063SBarry Smith 
91917ab2063SBarry Smith     /*  x = x + t */
92017ab2063SBarry Smith     for ( i=0; i<m; i++ ) { x[i] += t[i]; }
9210452661fSBarry Smith     PetscFree(t);
92217ab2063SBarry Smith     return 0;
92317ab2063SBarry Smith   }
92417ab2063SBarry Smith   if (flag & SOR_ZERO_INITIAL_GUESS) {
92517ab2063SBarry Smith     if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP){
92617ab2063SBarry Smith       for ( i=0; i<m; i++ ) {
927416022c9SBarry Smith         d    = fshift + a->a[diag[i]+shift];
928416022c9SBarry Smith         n    = diag[i] - a->i[i];
929416022c9SBarry Smith         idx  = a->j + a->i[i] + shift;
930416022c9SBarry Smith         v    = a->a + a->i[i] + shift;
93117ab2063SBarry Smith         sum  = b[i];
93217ab2063SBarry Smith         SPARSEDENSEMDOT(sum,xs,v,idx,n);
93317ab2063SBarry Smith         x[i] = omega*(sum/d);
93417ab2063SBarry Smith       }
93517ab2063SBarry Smith       xb = x;
93617ab2063SBarry Smith     }
93717ab2063SBarry Smith     else xb = b;
93817ab2063SBarry Smith     if ((flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP) &&
93917ab2063SBarry Smith         (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP)) {
94017ab2063SBarry Smith       for ( i=0; i<m; i++ ) {
941416022c9SBarry Smith         x[i] *= a->a[diag[i]+shift];
94217ab2063SBarry Smith       }
94317ab2063SBarry Smith     }
94417ab2063SBarry Smith     if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP){
94517ab2063SBarry Smith       for ( i=m-1; i>=0; i-- ) {
946416022c9SBarry Smith         d    = fshift + a->a[diag[i] + shift];
947416022c9SBarry Smith         n    = a->i[i+1] - diag[i] - 1;
948416022c9SBarry Smith         idx  = a->j + diag[i] + (!shift);
949416022c9SBarry Smith         v    = a->a + diag[i] + (!shift);
95017ab2063SBarry Smith         sum  = xb[i];
95117ab2063SBarry Smith         SPARSEDENSEMDOT(sum,xs,v,idx,n);
95217ab2063SBarry Smith         x[i] = omega*(sum/d);
95317ab2063SBarry Smith       }
95417ab2063SBarry Smith     }
95517ab2063SBarry Smith     its--;
95617ab2063SBarry Smith   }
95717ab2063SBarry Smith   while (its--) {
95817ab2063SBarry Smith     if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP){
95917ab2063SBarry Smith       for ( i=0; i<m; i++ ) {
960416022c9SBarry Smith         d    = fshift + a->a[diag[i]+shift];
961416022c9SBarry Smith         n    = a->i[i+1] - a->i[i];
962416022c9SBarry Smith         idx  = a->j + a->i[i] + shift;
963416022c9SBarry Smith         v    = a->a + a->i[i] + shift;
96417ab2063SBarry Smith         sum  = b[i];
96517ab2063SBarry Smith         SPARSEDENSEMDOT(sum,xs,v,idx,n);
9667e33a6baSBarry Smith         x[i] = (1. - omega)*x[i] + omega*(sum + a->a[diag[i]+shift]*x[i])/d;
96717ab2063SBarry Smith       }
96817ab2063SBarry Smith     }
96917ab2063SBarry Smith     if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP){
97017ab2063SBarry Smith       for ( i=m-1; i>=0; i-- ) {
971416022c9SBarry Smith         d    = fshift + a->a[diag[i] + shift];
972416022c9SBarry Smith         n    = a->i[i+1] - a->i[i];
973416022c9SBarry Smith         idx  = a->j + a->i[i] + shift;
974416022c9SBarry Smith         v    = a->a + a->i[i] + shift;
97517ab2063SBarry Smith         sum  = b[i];
97617ab2063SBarry Smith         SPARSEDENSEMDOT(sum,xs,v,idx,n);
9777e33a6baSBarry Smith         x[i] = (1. - omega)*x[i] + omega*(sum + a->a[diag[i]+shift]*x[i])/d;
97817ab2063SBarry Smith       }
97917ab2063SBarry Smith     }
98017ab2063SBarry Smith   }
98117ab2063SBarry Smith   return 0;
98217ab2063SBarry Smith }
98317ab2063SBarry Smith 
9845615d1e5SSatish Balay #undef __FUNC__
9855615d1e5SSatish Balay #define __FUNC__ "MatGetInfo_SeqAIJ"
9868f6be9afSLois Curfman McInnes int MatGetInfo_SeqAIJ(Mat A,MatInfoType flag,MatInfo *info)
98717ab2063SBarry Smith {
988416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
9894e220ebcSLois Curfman McInnes 
9904e220ebcSLois Curfman McInnes   info->rows_global    = (double)a->m;
9914e220ebcSLois Curfman McInnes   info->columns_global = (double)a->n;
9924e220ebcSLois Curfman McInnes   info->rows_local     = (double)a->m;
9934e220ebcSLois Curfman McInnes   info->columns_local  = (double)a->n;
9944e220ebcSLois Curfman McInnes   info->block_size     = 1.0;
9954e220ebcSLois Curfman McInnes   info->nz_allocated   = (double)a->maxnz;
9964e220ebcSLois Curfman McInnes   info->nz_used        = (double)a->nz;
9974e220ebcSLois Curfman McInnes   info->nz_unneeded    = (double)(a->maxnz - a->nz);
9984e220ebcSLois Curfman McInnes   /*  if (info->nz_unneeded != A->info.nz_unneeded)
9994e220ebcSLois Curfman McInnes     printf("space descrepancy: maxnz-nz = %d, nz_unneeded = %d\n",(int)info->nz_unneeded,(int)A->info.nz_unneeded); */
10004e220ebcSLois Curfman McInnes   info->assemblies     = (double)A->num_ass;
10014e220ebcSLois Curfman McInnes   info->mallocs        = (double)a->reallocs;
10024e220ebcSLois Curfman McInnes   info->memory         = A->mem;
10034e220ebcSLois Curfman McInnes   if (A->factor) {
10044e220ebcSLois Curfman McInnes     info->fill_ratio_given  = A->info.fill_ratio_given;
10054e220ebcSLois Curfman McInnes     info->fill_ratio_needed = A->info.fill_ratio_needed;
10064e220ebcSLois Curfman McInnes     info->factor_mallocs    = A->info.factor_mallocs;
10074e220ebcSLois Curfman McInnes   } else {
10084e220ebcSLois Curfman McInnes     info->fill_ratio_given  = 0;
10094e220ebcSLois Curfman McInnes     info->fill_ratio_needed = 0;
10104e220ebcSLois Curfman McInnes     info->factor_mallocs    = 0;
10114e220ebcSLois Curfman McInnes   }
101217ab2063SBarry Smith   return 0;
101317ab2063SBarry Smith }
101417ab2063SBarry Smith 
101517ab2063SBarry Smith extern int MatLUFactorSymbolic_SeqAIJ(Mat,IS,IS,double,Mat*);
101617ab2063SBarry Smith extern int MatLUFactorNumeric_SeqAIJ(Mat,Mat*);
101717ab2063SBarry Smith extern int MatLUFactor_SeqAIJ(Mat,IS,IS,double);
101817ab2063SBarry Smith extern int MatSolve_SeqAIJ(Mat,Vec,Vec);
101917ab2063SBarry Smith extern int MatSolveAdd_SeqAIJ(Mat,Vec,Vec,Vec);
102017ab2063SBarry Smith extern int MatSolveTrans_SeqAIJ(Mat,Vec,Vec);
102117ab2063SBarry Smith extern int MatSolveTransAdd_SeqAIJ(Mat,Vec,Vec,Vec);
102217ab2063SBarry Smith 
10235615d1e5SSatish Balay #undef __FUNC__
10245615d1e5SSatish Balay #define __FUNC__ "MatZeroRows_SeqAIJ"
10258f6be9afSLois Curfman McInnes int MatZeroRows_SeqAIJ(Mat A,IS is,Scalar *diag)
102617ab2063SBarry Smith {
1027416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
1028416022c9SBarry Smith   int         i,ierr,N, *rows,m = a->m - 1,shift = a->indexshift;
102917ab2063SBarry Smith 
103077c4ece6SBarry Smith   ierr = ISGetSize(is,&N); CHKERRQ(ierr);
103117ab2063SBarry Smith   ierr = ISGetIndices(is,&rows); CHKERRQ(ierr);
103217ab2063SBarry Smith   if (diag) {
103317ab2063SBarry Smith     for ( i=0; i<N; i++ ) {
1034e3372554SBarry Smith       if (rows[i] < 0 || rows[i] > m) SETERRQ(1,0,"row out of range");
1035416022c9SBarry Smith       if (a->ilen[rows[i]] > 0) { /* in case row was completely empty */
1036416022c9SBarry Smith         a->ilen[rows[i]] = 1;
1037416022c9SBarry Smith         a->a[a->i[rows[i]]+shift] = *diag;
1038416022c9SBarry Smith         a->j[a->i[rows[i]]+shift] = rows[i]+shift;
103917ab2063SBarry Smith       }
104017ab2063SBarry Smith       else {
104117ab2063SBarry Smith         ierr = MatSetValues_SeqAIJ(A,1,&rows[i],1,&rows[i],diag,INSERT_VALUES);
104217ab2063SBarry Smith         CHKERRQ(ierr);
104317ab2063SBarry Smith       }
104417ab2063SBarry Smith     }
104517ab2063SBarry Smith   }
104617ab2063SBarry Smith   else {
104717ab2063SBarry Smith     for ( i=0; i<N; i++ ) {
1048e3372554SBarry Smith       if (rows[i] < 0 || rows[i] > m) SETERRQ(1,0,"row out of range");
1049416022c9SBarry Smith       a->ilen[rows[i]] = 0;
105017ab2063SBarry Smith     }
105117ab2063SBarry Smith   }
105217ab2063SBarry Smith   ISRestoreIndices(is,&rows);
105343a90d84SBarry Smith   ierr = MatAssemblyEnd_SeqAIJ(A,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr);
105417ab2063SBarry Smith   return 0;
105517ab2063SBarry Smith }
105617ab2063SBarry Smith 
10575615d1e5SSatish Balay #undef __FUNC__
10585615d1e5SSatish Balay #define __FUNC__ "MatGetSize_SeqAIJ"
10598f6be9afSLois Curfman McInnes int MatGetSize_SeqAIJ(Mat A,int *m,int *n)
106017ab2063SBarry Smith {
1061416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
1062416022c9SBarry Smith   *m = a->m; *n = a->n;
106317ab2063SBarry Smith   return 0;
106417ab2063SBarry Smith }
106517ab2063SBarry Smith 
10665615d1e5SSatish Balay #undef __FUNC__
10675615d1e5SSatish Balay #define __FUNC__ "MatGetOwnershipRange_SeqAIJ"
10688f6be9afSLois Curfman McInnes int MatGetOwnershipRange_SeqAIJ(Mat A,int *m,int *n)
106917ab2063SBarry Smith {
1070416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
1071416022c9SBarry Smith   *m = 0; *n = a->m;
107217ab2063SBarry Smith   return 0;
107317ab2063SBarry Smith }
1074026e39d0SSatish Balay 
10755615d1e5SSatish Balay #undef __FUNC__
10765615d1e5SSatish Balay #define __FUNC__ "MatGetRow_SeqAIJ"
10774e093b46SBarry Smith int MatGetRow_SeqAIJ(Mat A,int row,int *nz,int **idx,Scalar **v)
107817ab2063SBarry Smith {
1079416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
1080c456f294SBarry Smith   int        *itmp,i,shift = a->indexshift;
108117ab2063SBarry Smith 
1082e3372554SBarry Smith   if (row < 0 || row >= a->m) SETERRQ(1,0,"Row out of range");
108317ab2063SBarry Smith 
1084416022c9SBarry Smith   *nz = a->i[row+1] - a->i[row];
1085416022c9SBarry Smith   if (v) *v = a->a + a->i[row] + shift;
108617ab2063SBarry Smith   if (idx) {
1087416022c9SBarry Smith     itmp = a->j + a->i[row] + shift;
10884e093b46SBarry Smith     if (*nz && shift) {
10890452661fSBarry Smith       *idx = (int *) PetscMalloc( (*nz)*sizeof(int) ); CHKPTRQ(*idx);
109017ab2063SBarry Smith       for ( i=0; i<(*nz); i++ ) {(*idx)[i] = itmp[i] + shift;}
10914e093b46SBarry Smith     } else if (*nz) {
10924e093b46SBarry Smith       *idx = itmp;
109317ab2063SBarry Smith     }
109417ab2063SBarry Smith     else *idx = 0;
109517ab2063SBarry Smith   }
109617ab2063SBarry Smith   return 0;
109717ab2063SBarry Smith }
109817ab2063SBarry Smith 
10995615d1e5SSatish Balay #undef __FUNC__
11005615d1e5SSatish Balay #define __FUNC__ "MatRestoreRow_SeqAIJ"
11014e093b46SBarry Smith int MatRestoreRow_SeqAIJ(Mat A,int row,int *nz,int **idx,Scalar **v)
110217ab2063SBarry Smith {
11034e093b46SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
11044e093b46SBarry Smith   if (idx) {if (*idx && a->indexshift) PetscFree(*idx);}
110517ab2063SBarry Smith   return 0;
110617ab2063SBarry Smith }
110717ab2063SBarry Smith 
11085615d1e5SSatish Balay #undef __FUNC__
11095615d1e5SSatish Balay #define __FUNC__ "MatNorm_SeqAIJ"
11108f6be9afSLois Curfman McInnes int MatNorm_SeqAIJ(Mat A,NormType type,double *norm)
111117ab2063SBarry Smith {
1112416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
1113416022c9SBarry Smith   Scalar     *v = a->a;
111417ab2063SBarry Smith   double     sum = 0.0;
1115416022c9SBarry Smith   int        i, j,shift = a->indexshift;
111617ab2063SBarry Smith 
111717ab2063SBarry Smith   if (type == NORM_FROBENIUS) {
1118416022c9SBarry Smith     for (i=0; i<a->nz; i++ ) {
111917ab2063SBarry Smith #if defined(PETSC_COMPLEX)
112017ab2063SBarry Smith       sum += real(conj(*v)*(*v)); v++;
112117ab2063SBarry Smith #else
112217ab2063SBarry Smith       sum += (*v)*(*v); v++;
112317ab2063SBarry Smith #endif
112417ab2063SBarry Smith     }
112517ab2063SBarry Smith     *norm = sqrt(sum);
112617ab2063SBarry Smith   }
112717ab2063SBarry Smith   else if (type == NORM_1) {
112817ab2063SBarry Smith     double *tmp;
1129416022c9SBarry Smith     int    *jj = a->j;
113066963ce1SSatish Balay     tmp = (double *) PetscMalloc( (a->n+1)*sizeof(double) ); CHKPTRQ(tmp);
1131cddf8d76SBarry Smith     PetscMemzero(tmp,a->n*sizeof(double));
113217ab2063SBarry Smith     *norm = 0.0;
1133416022c9SBarry Smith     for ( j=0; j<a->nz; j++ ) {
1134a2744918SBarry Smith         tmp[*jj++ + shift] += PetscAbsScalar(*v);  v++;
113517ab2063SBarry Smith     }
1136416022c9SBarry Smith     for ( j=0; j<a->n; j++ ) {
113717ab2063SBarry Smith       if (tmp[j] > *norm) *norm = tmp[j];
113817ab2063SBarry Smith     }
11390452661fSBarry Smith     PetscFree(tmp);
114017ab2063SBarry Smith   }
114117ab2063SBarry Smith   else if (type == NORM_INFINITY) {
114217ab2063SBarry Smith     *norm = 0.0;
1143416022c9SBarry Smith     for ( j=0; j<a->m; j++ ) {
1144416022c9SBarry Smith       v = a->a + a->i[j] + shift;
114517ab2063SBarry Smith       sum = 0.0;
1146416022c9SBarry Smith       for ( i=0; i<a->i[j+1]-a->i[j]; i++ ) {
1147cddf8d76SBarry Smith         sum += PetscAbsScalar(*v); v++;
114817ab2063SBarry Smith       }
114917ab2063SBarry Smith       if (sum > *norm) *norm = sum;
115017ab2063SBarry Smith     }
115117ab2063SBarry Smith   }
115217ab2063SBarry Smith   else {
1153e3372554SBarry Smith     SETERRQ(1,0,"No support for two norm yet");
115417ab2063SBarry Smith   }
115517ab2063SBarry Smith   return 0;
115617ab2063SBarry Smith }
115717ab2063SBarry Smith 
11585615d1e5SSatish Balay #undef __FUNC__
11595615d1e5SSatish Balay #define __FUNC__ "MatTranspose_SeqAIJ"
11608f6be9afSLois Curfman McInnes int MatTranspose_SeqAIJ(Mat A,Mat *B)
116117ab2063SBarry Smith {
1162416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
1163416022c9SBarry Smith   Mat        C;
1164416022c9SBarry Smith   int        i, ierr, *aj = a->j, *ai = a->i, m = a->m, len, *col;
1165416022c9SBarry Smith   int        shift = a->indexshift;
1166d5d45c9bSBarry Smith   Scalar     *array = a->a;
116717ab2063SBarry Smith 
11683638b69dSLois Curfman McInnes   if (B == PETSC_NULL && m != a->n)
1169e3372554SBarry Smith     SETERRQ(1,0,"Square matrix only for in-place");
11700452661fSBarry Smith   col = (int *) PetscMalloc((1+a->n)*sizeof(int)); CHKPTRQ(col);
1171cddf8d76SBarry Smith   PetscMemzero(col,(1+a->n)*sizeof(int));
117217ab2063SBarry Smith   if (shift) {
117317ab2063SBarry Smith     for ( i=0; i<ai[m]-1; i++ ) aj[i] -= 1;
117417ab2063SBarry Smith   }
117517ab2063SBarry Smith   for ( i=0; i<ai[m]+shift; i++ ) col[aj[i]] += 1;
1176416022c9SBarry Smith   ierr = MatCreateSeqAIJ(A->comm,a->n,m,0,col,&C); CHKERRQ(ierr);
11770452661fSBarry Smith   PetscFree(col);
117817ab2063SBarry Smith   for ( i=0; i<m; i++ ) {
117917ab2063SBarry Smith     len = ai[i+1]-ai[i];
1180416022c9SBarry Smith     ierr = MatSetValues(C,len,aj,1,&i,array,INSERT_VALUES); CHKERRQ(ierr);
118117ab2063SBarry Smith     array += len; aj += len;
118217ab2063SBarry Smith   }
118317ab2063SBarry Smith   if (shift) {
118417ab2063SBarry Smith     for ( i=0; i<ai[m]-1; i++ ) aj[i] += 1;
118517ab2063SBarry Smith   }
118617ab2063SBarry Smith 
11876d4a8577SBarry Smith   ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr);
11886d4a8577SBarry Smith   ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr);
118917ab2063SBarry Smith 
11903638b69dSLois Curfman McInnes   if (B != PETSC_NULL) {
1191416022c9SBarry Smith     *B = C;
119217ab2063SBarry Smith   } else {
1193416022c9SBarry Smith     /* This isn't really an in-place transpose */
11940452661fSBarry Smith     PetscFree(a->a);
11950452661fSBarry Smith     if (!a->singlemalloc) {PetscFree(a->i); PetscFree(a->j);}
11960452661fSBarry Smith     if (a->diag) PetscFree(a->diag);
11970452661fSBarry Smith     if (a->ilen) PetscFree(a->ilen);
11980452661fSBarry Smith     if (a->imax) PetscFree(a->imax);
11990452661fSBarry Smith     if (a->solve_work) PetscFree(a->solve_work);
12001073c447SSatish Balay     if (a->inode.size) PetscFree(a->inode.size);
12010452661fSBarry Smith     PetscFree(a);
1202416022c9SBarry Smith     PetscMemcpy(A,C,sizeof(struct _Mat));
12030452661fSBarry Smith     PetscHeaderDestroy(C);
120417ab2063SBarry Smith   }
120517ab2063SBarry Smith   return 0;
120617ab2063SBarry Smith }
120717ab2063SBarry Smith 
12085615d1e5SSatish Balay #undef __FUNC__
12095615d1e5SSatish Balay #define __FUNC__ "MatDiagonalScale_SeqAIJ"
12108f6be9afSLois Curfman McInnes int MatDiagonalScale_SeqAIJ(Mat A,Vec ll,Vec rr)
121117ab2063SBarry Smith {
1212416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
121317ab2063SBarry Smith   Scalar     *l,*r,x,*v;
1214d5d45c9bSBarry Smith   int        i,j,m = a->m, n = a->n, M, nz = a->nz, *jj,shift = a->indexshift;
121517ab2063SBarry Smith 
121617ab2063SBarry Smith   if (ll) {
12173ea7c6a1SSatish Balay     /* The local size is used so that VecMPI can be passed to this routine
12183ea7c6a1SSatish Balay        by MatDiagonalScale_MPIAIJ */
12199b1297e1SSatish Balay     VecGetLocalSize_Fast(ll,m);
1220e3372554SBarry Smith     if (m != a->m) SETERRQ(1,0,"Left scaling vector wrong length");
122190f02eecSBarry Smith     VecGetArray_Fast(ll,l);
1222416022c9SBarry Smith     v = a->a;
122317ab2063SBarry Smith     for ( i=0; i<m; i++ ) {
122417ab2063SBarry Smith       x = l[i];
1225416022c9SBarry Smith       M = a->i[i+1] - a->i[i];
122617ab2063SBarry Smith       for ( j=0; j<M; j++ ) { (*v++) *= x;}
122717ab2063SBarry Smith     }
122844cd7ae7SLois Curfman McInnes     PLogFlops(nz);
122917ab2063SBarry Smith   }
123017ab2063SBarry Smith   if (rr) {
12319b1297e1SSatish Balay     VecGetLocalSize_Fast(rr,n);
1232e3372554SBarry Smith     if (n != a->n) SETERRQ(1,0,"Right scaling vector wrong length");
123390f02eecSBarry Smith     VecGetArray_Fast(rr,r);
1234416022c9SBarry Smith     v = a->a; jj = a->j;
123517ab2063SBarry Smith     for ( i=0; i<nz; i++ ) {
123617ab2063SBarry Smith       (*v++) *= r[*jj++ + shift];
123717ab2063SBarry Smith     }
123844cd7ae7SLois Curfman McInnes     PLogFlops(nz);
123917ab2063SBarry Smith   }
124017ab2063SBarry Smith   return 0;
124117ab2063SBarry Smith }
124217ab2063SBarry Smith 
12435615d1e5SSatish Balay #undef __FUNC__
12445615d1e5SSatish Balay #define __FUNC__ "MatGetSubMatrix_SeqAIJ"
12458f6be9afSLois Curfman McInnes int MatGetSubMatrix_SeqAIJ(Mat A,IS isrow,IS iscol,MatGetSubMatrixCall scall,Mat *B)
124617ab2063SBarry Smith {
1247db02288aSLois Curfman McInnes   Mat_SeqAIJ   *a = (Mat_SeqAIJ *) A->data,*c;
124802834360SBarry Smith   int          nznew, *smap, i, k, kstart, kend, ierr, oldcols = a->n,*lens;
124999141d43SSatish Balay   int          row,mat_i,*mat_j,tcol,first,step,*mat_ilen;
1250a2744918SBarry Smith   register int sum,lensi;
125199141d43SSatish Balay   int          *irow, *icol, nrows, ncols, shift = a->indexshift,*ssmap;
125299141d43SSatish Balay   int          *starts,*j_new,*i_new,*aj = a->j, *ai = a->i,ii,*ailen = a->ilen;
125399141d43SSatish Balay   Scalar       *a_new,*mat_a;
1254416022c9SBarry Smith   Mat          C;
125517ab2063SBarry Smith 
1256b48a1e75SSatish Balay   ierr = ISSorted(isrow,(PetscTruth*)&i);
1257e3372554SBarry Smith   if (!i) SETERRQ(1,0,"ISrow is not sorted");
125899141d43SSatish Balay   ierr = ISSorted(iscol,(PetscTruth*)&i);
1259e3372554SBarry Smith   if (!i) SETERRQ(1,0,"IScol is not sorted");
126099141d43SSatish Balay 
126117ab2063SBarry Smith   ierr = ISGetIndices(isrow,&irow); CHKERRQ(ierr);
126217ab2063SBarry Smith   ierr = ISGetSize(isrow,&nrows); CHKERRQ(ierr);
126317ab2063SBarry Smith   ierr = ISGetSize(iscol,&ncols); CHKERRQ(ierr);
126417ab2063SBarry Smith 
12657264ac53SSatish Balay   if (ISStrideGetInfo(iscol,&first,&step) && step == 1) { /* no need to sort */
126602834360SBarry Smith     /* special case of contiguous rows */
126757faeb66SBarry Smith     lens   = (int *) PetscMalloc((ncols+nrows+1)*sizeof(int)); CHKPTRQ(lens);
126802834360SBarry Smith     starts = lens + ncols;
126902834360SBarry Smith     /* loop over new rows determining lens and starting points */
127002834360SBarry Smith     for (i=0; i<nrows; i++) {
1271a2744918SBarry Smith       kstart  = ai[irow[i]]+shift;
1272a2744918SBarry Smith       kend    = kstart + ailen[irow[i]];
127302834360SBarry Smith       for ( k=kstart; k<kend; k++ ) {
1274d8ced48eSBarry Smith         if (aj[k]+shift >= first) {
127502834360SBarry Smith           starts[i] = k;
127602834360SBarry Smith           break;
127702834360SBarry Smith 	}
127802834360SBarry Smith       }
1279a2744918SBarry Smith       sum = 0;
128002834360SBarry Smith       while (k < kend) {
1281d8ced48eSBarry Smith         if (aj[k++]+shift >= first+ncols) break;
1282a2744918SBarry Smith         sum++;
128302834360SBarry Smith       }
1284a2744918SBarry Smith       lens[i] = sum;
128502834360SBarry Smith     }
128602834360SBarry Smith     /* create submatrix */
1287cddf8d76SBarry Smith     if (scall == MAT_REUSE_MATRIX) {
128808480c60SBarry Smith       int n_cols,n_rows;
128908480c60SBarry Smith       ierr = MatGetSize(*B,&n_rows,&n_cols); CHKERRQ(ierr);
1290e3372554SBarry Smith       if (n_rows != nrows || n_cols != ncols) SETERRQ(1,0,"");
1291d8ced48eSBarry Smith       ierr = MatZeroEntries(*B); CHKERRQ(ierr);
129208480c60SBarry Smith       C = *B;
129308480c60SBarry Smith     }
129408480c60SBarry Smith     else {
129502834360SBarry Smith       ierr = MatCreateSeqAIJ(A->comm,nrows,ncols,0,lens,&C);CHKERRQ(ierr);
129608480c60SBarry Smith     }
1297db02288aSLois Curfman McInnes     c = (Mat_SeqAIJ*) C->data;
1298db02288aSLois Curfman McInnes 
129902834360SBarry Smith     /* loop over rows inserting into submatrix */
1300db02288aSLois Curfman McInnes     a_new    = c->a;
1301db02288aSLois Curfman McInnes     j_new    = c->j;
1302db02288aSLois Curfman McInnes     i_new    = c->i;
1303db02288aSLois Curfman McInnes     i_new[0] = -shift;
130402834360SBarry Smith     for (i=0; i<nrows; i++) {
1305a2744918SBarry Smith       ii    = starts[i];
1306a2744918SBarry Smith       lensi = lens[i];
1307a2744918SBarry Smith       for ( k=0; k<lensi; k++ ) {
1308a2744918SBarry Smith         *j_new++ = aj[ii+k] - first;
130902834360SBarry Smith       }
1310a2744918SBarry Smith       PetscMemcpy(a_new,a->a + starts[i],lensi*sizeof(Scalar));
1311a2744918SBarry Smith       a_new      += lensi;
1312a2744918SBarry Smith       i_new[i+1]  = i_new[i] + lensi;
1313a2744918SBarry Smith       c->ilen[i]  = lensi;
131402834360SBarry Smith     }
13150452661fSBarry Smith     PetscFree(lens);
131602834360SBarry Smith   }
131702834360SBarry Smith   else {
131802834360SBarry Smith     ierr = ISGetIndices(iscol,&icol); CHKERRQ(ierr);
13190452661fSBarry Smith     smap  = (int *) PetscMalloc((1+oldcols)*sizeof(int)); CHKPTRQ(smap);
132002834360SBarry Smith     ssmap = smap + shift;
132199141d43SSatish Balay     lens  = (int *) PetscMalloc((1+nrows)*sizeof(int)); CHKPTRQ(lens);
1322cddf8d76SBarry Smith     PetscMemzero(smap,oldcols*sizeof(int));
132317ab2063SBarry Smith     for ( i=0; i<ncols; i++ ) smap[icol[i]] = i+1;
132402834360SBarry Smith     /* determine lens of each row */
132502834360SBarry Smith     for (i=0; i<nrows; i++) {
1326d8ced48eSBarry Smith       kstart  = ai[irow[i]]+shift;
132702834360SBarry Smith       kend    = kstart + a->ilen[irow[i]];
132802834360SBarry Smith       lens[i] = 0;
132902834360SBarry Smith       for ( k=kstart; k<kend; k++ ) {
1330d8ced48eSBarry Smith         if (ssmap[aj[k]]) {
133102834360SBarry Smith           lens[i]++;
133202834360SBarry Smith         }
133302834360SBarry Smith       }
133402834360SBarry Smith     }
133517ab2063SBarry Smith     /* Create and fill new matrix */
1336a2744918SBarry Smith     if (scall == MAT_REUSE_MATRIX) {
133799141d43SSatish Balay       c = (Mat_SeqAIJ *)((*B)->data);
133899141d43SSatish Balay 
1339e3372554SBarry Smith       if (c->m  != nrows || c->n != ncols) SETERRQ(1,0,"");
134099141d43SSatish Balay       if (PetscMemcmp(c->ilen,lens, c->m *sizeof(int))) {
1341e3372554SBarry Smith         SETERRQ(1,0,"Cannot reuse matrix. wrong no of nonzeros");
134299141d43SSatish Balay       }
134399141d43SSatish Balay       PetscMemzero(c->ilen,c->m*sizeof(int));
134408480c60SBarry Smith       C = *B;
134508480c60SBarry Smith     }
134608480c60SBarry Smith     else {
134702834360SBarry Smith       ierr = MatCreateSeqAIJ(A->comm,nrows,ncols,0,lens,&C);CHKERRQ(ierr);
134808480c60SBarry Smith     }
134999141d43SSatish Balay     c = (Mat_SeqAIJ *)(C->data);
135017ab2063SBarry Smith     for (i=0; i<nrows; i++) {
135199141d43SSatish Balay       row    = irow[i];
135217ab2063SBarry Smith       nznew  = 0;
135399141d43SSatish Balay       kstart = ai[row]+shift;
135499141d43SSatish Balay       kend   = kstart + a->ilen[row];
135599141d43SSatish Balay       mat_i  = c->i[i]+shift;
135699141d43SSatish Balay       mat_j  = c->j + mat_i;
135799141d43SSatish Balay       mat_a  = c->a + mat_i;
135899141d43SSatish Balay       mat_ilen = c->ilen + i;
135917ab2063SBarry Smith       for ( k=kstart; k<kend; k++ ) {
136099141d43SSatish Balay         if ((tcol=ssmap[a->j[k]])) {
136199141d43SSatish Balay           *mat_j++ = tcol - (!shift);
136299141d43SSatish Balay           *mat_a++ = a->a[k];
136399141d43SSatish Balay           (*mat_ilen)++;
136499141d43SSatish Balay 
136517ab2063SBarry Smith         }
136617ab2063SBarry Smith       }
136717ab2063SBarry Smith     }
136802834360SBarry Smith     /* Free work space */
136902834360SBarry Smith     ierr = ISRestoreIndices(iscol,&icol); CHKERRQ(ierr);
137099141d43SSatish Balay     PetscFree(smap); PetscFree(lens);
137102834360SBarry Smith   }
13726d4a8577SBarry Smith   ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr);
13736d4a8577SBarry Smith   ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr);
137417ab2063SBarry Smith 
137517ab2063SBarry Smith   ierr = ISRestoreIndices(isrow,&irow); CHKERRQ(ierr);
1376416022c9SBarry Smith   *B = C;
137717ab2063SBarry Smith   return 0;
137817ab2063SBarry Smith }
137917ab2063SBarry Smith 
1380a871dcd8SBarry Smith /*
138163b91edcSBarry Smith      note: This can only work for identity for row and col. It would
138263b91edcSBarry Smith    be good to check this and otherwise generate an error.
1383a871dcd8SBarry Smith */
13845615d1e5SSatish Balay #undef __FUNC__
13855615d1e5SSatish Balay #define __FUNC__ "MatILUFactor_SeqAIJ"
13868f6be9afSLois Curfman McInnes int MatILUFactor_SeqAIJ(Mat inA,IS row,IS col,double efill,int fill)
1387a871dcd8SBarry Smith {
138863b91edcSBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) inA->data;
138908480c60SBarry Smith   int        ierr;
139063b91edcSBarry Smith   Mat        outA;
139163b91edcSBarry Smith 
1392e3372554SBarry Smith   if (fill != 0) SETERRQ(1,0,"Only fill=0 supported");
1393a871dcd8SBarry Smith 
139463b91edcSBarry Smith   outA          = inA;
139563b91edcSBarry Smith   inA->factor   = FACTOR_LU;
139663b91edcSBarry Smith   a->row        = row;
139763b91edcSBarry Smith   a->col        = col;
139863b91edcSBarry Smith 
139994a9d846SBarry Smith   if (!a->solve_work) { /* this matrix may have been factored before */
14000452661fSBarry Smith     a->solve_work = (Scalar *) PetscMalloc( (a->m+1)*sizeof(Scalar)); CHKPTRQ(a->solve_work);
140194a9d846SBarry Smith   }
140263b91edcSBarry Smith 
140308480c60SBarry Smith   if (!a->diag) {
140408480c60SBarry Smith     ierr = MatMarkDiag_SeqAIJ(inA); CHKERRQ(ierr);
140563b91edcSBarry Smith   }
140663b91edcSBarry Smith   ierr = MatLUFactorNumeric_SeqAIJ(inA,&outA); CHKERRQ(ierr);
1407a871dcd8SBarry Smith   return 0;
1408a871dcd8SBarry Smith }
1409a871dcd8SBarry Smith 
1410f0b747eeSBarry Smith #include "pinclude/plapack.h"
14115615d1e5SSatish Balay #undef __FUNC__
14125615d1e5SSatish Balay #define __FUNC__ "MatScale_SeqAIJ"
14138f6be9afSLois Curfman McInnes int MatScale_SeqAIJ(Scalar *alpha,Mat inA)
1414f0b747eeSBarry Smith {
1415f0b747eeSBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) inA->data;
1416f0b747eeSBarry Smith   int        one = 1;
1417f0b747eeSBarry Smith   BLscal_( &a->nz, alpha, a->a, &one );
1418f0b747eeSBarry Smith   PLogFlops(a->nz);
1419f0b747eeSBarry Smith   return 0;
1420f0b747eeSBarry Smith }
1421f0b747eeSBarry Smith 
14225615d1e5SSatish Balay #undef __FUNC__
14235615d1e5SSatish Balay #define __FUNC__ "MatGetSubMatrices_SeqAIJ"
14248f6be9afSLois Curfman McInnes int MatGetSubMatrices_SeqAIJ(Mat A,int n, IS *irow,IS *icol,MatGetSubMatrixCall scall,
1425cddf8d76SBarry Smith                                     Mat **B)
1426cddf8d76SBarry Smith {
1427cddf8d76SBarry Smith   int ierr,i;
1428cddf8d76SBarry Smith 
1429cddf8d76SBarry Smith   if (scall == MAT_INITIAL_MATRIX) {
14300452661fSBarry Smith     *B = (Mat *) PetscMalloc( (n+1)*sizeof(Mat) ); CHKPTRQ(*B);
1431cddf8d76SBarry Smith   }
1432cddf8d76SBarry Smith 
1433cddf8d76SBarry Smith   for ( i=0; i<n; i++ ) {
1434905e6a2fSBarry Smith     ierr = MatGetSubMatrix_SeqAIJ(A,irow[i],icol[i],scall,&(*B)[i]);CHKERRQ(ierr);
1435cddf8d76SBarry Smith   }
1436cddf8d76SBarry Smith   return 0;
1437cddf8d76SBarry Smith }
1438cddf8d76SBarry Smith 
14395615d1e5SSatish Balay #undef __FUNC__
14405615d1e5SSatish Balay #define __FUNC__ "MatGetBlockSize_SeqAIJ"
14418f6be9afSLois Curfman McInnes int MatGetBlockSize_SeqAIJ(Mat A, int *bs)
14425a838052SSatish Balay {
14435a838052SSatish Balay   *bs = 1;
14445a838052SSatish Balay   return 0;
14455a838052SSatish Balay }
14465a838052SSatish Balay 
14475615d1e5SSatish Balay #undef __FUNC__
14485615d1e5SSatish Balay #define __FUNC__ "MatIncreaseOverlap_SeqAIJ"
14498f6be9afSLois Curfman McInnes int MatIncreaseOverlap_SeqAIJ(Mat A, int is_max, IS *is, int ov)
14504dcbc457SBarry Smith {
1451e4d965acSSatish Balay   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
145206763907SSatish Balay   int        shift, row, i,j,k,l,m,n, *idx,ierr, *nidx, isz, val;
14538a047759SSatish Balay   int        start, end, *ai, *aj;
145406763907SSatish Balay   char       *table;
14558a047759SSatish Balay   shift = a->indexshift;
1456e4d965acSSatish Balay   m     = a->m;
1457e4d965acSSatish Balay   ai    = a->i;
14588a047759SSatish Balay   aj    = a->j+shift;
14598a047759SSatish Balay 
1460e3372554SBarry Smith   if (ov < 0)  SETERRQ(1,0,"illegal overlap value used");
146106763907SSatish Balay 
146206763907SSatish Balay   table = (char *) PetscMalloc((m/BITSPERBYTE +1)*sizeof(char)); CHKPTRQ(table);
146306763907SSatish Balay   nidx  = (int *) PetscMalloc((m+1)*sizeof(int)); CHKPTRQ(nidx);
146406763907SSatish Balay 
1465e4d965acSSatish Balay   for ( i=0; i<is_max; i++ ) {
1466b97fc60eSLois Curfman McInnes     /* Initialize the two local arrays */
1467e4d965acSSatish Balay     isz  = 0;
146806763907SSatish Balay     PetscMemzero(table,(m/BITSPERBYTE +1)*sizeof(char));
1469e4d965acSSatish Balay 
1470e4d965acSSatish Balay     /* Extract the indices, assume there can be duplicate entries */
14714dcbc457SBarry Smith     ierr = ISGetIndices(is[i],&idx);  CHKERRQ(ierr);
147277c4ece6SBarry Smith     ierr = ISGetSize(is[i],&n);  CHKERRQ(ierr);
1473e4d965acSSatish Balay 
1474dd097bc3SLois Curfman McInnes     /* Enter these into the temp arrays. I.e., mark table[row], enter row into new index */
1475e4d965acSSatish Balay     for ( j=0; j<n ; ++j){
147606763907SSatish Balay       if(!BT_LOOKUP(table, idx[j])) { nidx[isz++] = idx[j];}
14774dcbc457SBarry Smith     }
147806763907SSatish Balay     ierr = ISRestoreIndices(is[i],&idx);  CHKERRQ(ierr);
147906763907SSatish Balay     ierr = ISDestroy(is[i]); CHKERRQ(ierr);
1480e4d965acSSatish Balay 
148104a348a9SBarry Smith     k = 0;
148204a348a9SBarry Smith     for ( j=0; j<ov; j++){ /* for each overlap */
148304a348a9SBarry Smith       n = isz;
148406763907SSatish Balay       for ( ; k<n ; k++){ /* do only those rows in nidx[k], which are not done yet */
1485e4d965acSSatish Balay         row   = nidx[k];
1486e4d965acSSatish Balay         start = ai[row];
1487e4d965acSSatish Balay         end   = ai[row+1];
148804a348a9SBarry Smith         for ( l = start; l<end ; l++){
14898a047759SSatish Balay           val = aj[l] + shift;
149006763907SSatish Balay           if (!BT_LOOKUP(table,val)) {nidx[isz++] = val;}
1491e4d965acSSatish Balay         }
1492e4d965acSSatish Balay       }
1493e4d965acSSatish Balay     }
1494029af93fSBarry Smith     ierr = ISCreateGeneral(PETSC_COMM_SELF, isz, nidx, (is+i)); CHKERRQ(ierr);
1495e4d965acSSatish Balay   }
149604a348a9SBarry Smith   PetscFree(table);
149706763907SSatish Balay   PetscFree(nidx);
1498e4d965acSSatish Balay   return 0;
14994dcbc457SBarry Smith }
150017ab2063SBarry Smith 
15010513a670SBarry Smith /* -------------------------------------------------------------- */
15020513a670SBarry Smith #undef __FUNC__
15030513a670SBarry Smith #define __FUNC__ "MatPermute_SeqAIJ"
15040513a670SBarry Smith int MatPermute_SeqAIJ(Mat A, IS rowp, IS colp, Mat *B)
15050513a670SBarry Smith {
15060513a670SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
15070513a670SBarry Smith   Scalar     *vwork;
15080513a670SBarry Smith   int        i, ierr, nz, m = a->m, n = a->n, *cwork;
15090513a670SBarry Smith   int        *row,*col,*cnew,j,*lens;
15100513a670SBarry Smith 
15110513a670SBarry Smith   ierr = ISGetIndices(rowp,&row); CHKERRQ(ierr);
15120513a670SBarry Smith   ierr = ISGetIndices(colp,&col); CHKERRQ(ierr);
15130513a670SBarry Smith 
15140513a670SBarry Smith   /* determine lengths of permuted rows */
15150513a670SBarry Smith   lens = (int *) PetscMalloc( (m+1)*sizeof(int) ); CHKPTRQ(lens);
15160513a670SBarry Smith   for (i=0; i<m; i++ ) {
15170513a670SBarry Smith     lens[row[i]] = a->i[i+1] - a->i[i];
15180513a670SBarry Smith   }
15190513a670SBarry Smith   ierr = MatCreateSeqAIJ(A->comm,m,n,0,lens,B);CHKERRQ(ierr);
15200513a670SBarry Smith   PetscFree(lens);
15210513a670SBarry Smith 
15220513a670SBarry Smith   cnew = (int *) PetscMalloc( n*sizeof(int) ); CHKPTRQ(cnew);
15230513a670SBarry Smith   for (i=0; i<m; i++) {
15240513a670SBarry Smith     ierr = MatGetRow(A,i,&nz,&cwork,&vwork); CHKERRQ(ierr);
15250513a670SBarry Smith     for (j=0; j<nz; j++ ) { cnew[j] = col[cwork[j]];}
15260513a670SBarry Smith     ierr = MatSetValues(*B,1,&row[i],nz,cnew,vwork,INSERT_VALUES); CHKERRQ(ierr);
15270513a670SBarry Smith     ierr = MatRestoreRow(A,i,&nz,&cwork,&vwork); CHKERRQ(ierr);
15280513a670SBarry Smith   }
15290513a670SBarry Smith   PetscFree(cnew);
15300513a670SBarry Smith   ierr = MatAssemblyBegin(*B,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr);
15310513a670SBarry Smith   ierr = MatAssemblyEnd(*B,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr);
15320513a670SBarry Smith   ierr = ISRestoreIndices(rowp,&row); CHKERRQ(ierr);
15330513a670SBarry Smith   ierr = ISRestoreIndices(colp,&col); CHKERRQ(ierr);
15340513a670SBarry Smith   return 0;
15350513a670SBarry Smith }
15360513a670SBarry Smith 
15375615d1e5SSatish Balay #undef __FUNC__
15385615d1e5SSatish Balay #define __FUNC__ "MatPrintHelp_SeqAIJ"
1539682d7d0cSBarry Smith int MatPrintHelp_SeqAIJ(Mat A)
1540682d7d0cSBarry Smith {
1541682d7d0cSBarry Smith   static int called = 0;
1542682d7d0cSBarry Smith   MPI_Comm   comm = A->comm;
1543682d7d0cSBarry Smith 
1544682d7d0cSBarry Smith   if (called) return 0; else called = 1;
154577c4ece6SBarry Smith   PetscPrintf(comm," Options for MATSEQAIJ and MATMPIAIJ matrix formats (the defaults):\n");
15460f665d81SLois Curfman McInnes   PetscPrintf(comm,"  -mat_lu_pivotthreshold <threshold>: Set pivoting threshold\n");
15470f665d81SLois Curfman McInnes   PetscPrintf(comm,"  -mat_aij_oneindex: internal indices begin at 1 instead of the default 0.\n");
15480f665d81SLois Curfman McInnes   PetscPrintf(comm,"  -mat_aij_no_inode: Do not use inodes\n");
15490f665d81SLois Curfman McInnes   PetscPrintf(comm,"  -mat_aij_inode_limit <limit>: Set inode limit (max limit=5)\n");
1550682d7d0cSBarry Smith #if defined(HAVE_ESSL)
15510f665d81SLois Curfman McInnes   PetscPrintf(comm,"  -mat_aij_essl: Use IBM sparse LU factorization and solve.\n");
1552682d7d0cSBarry Smith #endif
1553682d7d0cSBarry Smith   return 0;
1554682d7d0cSBarry Smith }
15558f6be9afSLois Curfman McInnes extern int MatEqual_SeqAIJ(Mat A,Mat B, PetscTruth* flg);
1556a93ec695SBarry Smith extern int MatFDColoringCreate_SeqAIJ(Mat,ISColoring,MatFDColoring);
1557a93ec695SBarry Smith extern int MatColoringPatch_SeqAIJ(Mat,int,int *,ISColoring *);
1558a93ec695SBarry Smith 
1559682d7d0cSBarry Smith /* -------------------------------------------------------------------*/
156017ab2063SBarry Smith static struct _MatOps MatOps = {MatSetValues_SeqAIJ,
156117ab2063SBarry Smith        MatGetRow_SeqAIJ,MatRestoreRow_SeqAIJ,
1562416022c9SBarry Smith        MatMult_SeqAIJ,MatMultAdd_SeqAIJ,
1563416022c9SBarry Smith        MatMultTrans_SeqAIJ,MatMultTransAdd_SeqAIJ,
156417ab2063SBarry Smith        MatSolve_SeqAIJ,MatSolveAdd_SeqAIJ,
156517ab2063SBarry Smith        MatSolveTrans_SeqAIJ,MatSolveTransAdd_SeqAIJ,
156617ab2063SBarry Smith        MatLUFactor_SeqAIJ,0,
156717ab2063SBarry Smith        MatRelax_SeqAIJ,
156817ab2063SBarry Smith        MatTranspose_SeqAIJ,
15697264ac53SSatish Balay        MatGetInfo_SeqAIJ,MatEqual_SeqAIJ,
1570f0b747eeSBarry Smith        MatGetDiagonal_SeqAIJ,MatDiagonalScale_SeqAIJ,MatNorm_SeqAIJ,
157117ab2063SBarry Smith        0,MatAssemblyEnd_SeqAIJ,
157217ab2063SBarry Smith        MatCompress_SeqAIJ,
157317ab2063SBarry Smith        MatSetOption_SeqAIJ,MatZeroEntries_SeqAIJ,MatZeroRows_SeqAIJ,
157417ab2063SBarry Smith        MatLUFactorSymbolic_SeqAIJ,MatLUFactorNumeric_SeqAIJ,0,0,
157517ab2063SBarry Smith        MatGetSize_SeqAIJ,MatGetSize_SeqAIJ,MatGetOwnershipRange_SeqAIJ,
157617ab2063SBarry Smith        MatILUFactorSymbolic_SeqAIJ,0,
157794a9d846SBarry Smith        0,0,
15783d1612f7SBarry Smith        MatConvertSameType_SeqAIJ,0,0,
1579cddf8d76SBarry Smith        MatILUFactor_SeqAIJ,0,0,
15807eb43aa7SLois Curfman McInnes        MatGetSubMatrices_SeqAIJ,MatIncreaseOverlap_SeqAIJ,
1581682d7d0cSBarry Smith        MatGetValues_SeqAIJ,0,
1582f0b747eeSBarry Smith        MatPrintHelp_SeqAIJ,
15835a838052SSatish Balay        MatScale_SeqAIJ,0,0,
15846945ee14SBarry Smith        MatILUDTFactor_SeqAIJ,
15856945ee14SBarry Smith        MatGetBlockSize_SeqAIJ,
15863b2fbd54SBarry Smith        MatGetRowIJ_SeqAIJ,
15873b2fbd54SBarry Smith        MatRestoreRowIJ_SeqAIJ,
15883b2fbd54SBarry Smith        MatGetColumnIJ_SeqAIJ,
1589a93ec695SBarry Smith        MatRestoreColumnIJ_SeqAIJ,
1590a93ec695SBarry Smith        MatFDColoringCreate_SeqAIJ,
15910513a670SBarry Smith        MatColoringPatch_SeqAIJ,
15920513a670SBarry Smith        0,
15930513a670SBarry Smith        MatPermute_SeqAIJ};
159417ab2063SBarry Smith 
159517ab2063SBarry Smith extern int MatUseSuperLU_SeqAIJ(Mat);
159617ab2063SBarry Smith extern int MatUseEssl_SeqAIJ(Mat);
159717ab2063SBarry Smith extern int MatUseDXML_SeqAIJ(Mat);
159817ab2063SBarry Smith 
15995615d1e5SSatish Balay #undef __FUNC__
16005615d1e5SSatish Balay #define __FUNC__ "MatCreateSeqAIJ"
160117ab2063SBarry Smith /*@C
1602682d7d0cSBarry Smith    MatCreateSeqAIJ - Creates a sparse matrix in AIJ (compressed row) format
16030d15e28bSLois Curfman McInnes    (the default parallel PETSc format).  For good matrix assembly performance
16046e62573dSLois Curfman McInnes    the user should preallocate the matrix storage by setting the parameter nz
16052bd5e0b2SLois Curfman McInnes    (or the array nzz).  By setting these parameters accurately, performance
16062bd5e0b2SLois Curfman McInnes    during matrix assembly can be increased by more than a factor of 50.
160717ab2063SBarry Smith 
160817ab2063SBarry Smith    Input Parameters:
1609029af93fSBarry Smith .  comm - MPI communicator, set to PETSC_COMM_SELF
161017ab2063SBarry Smith .  m - number of rows
161117ab2063SBarry Smith .  n - number of columns
161217ab2063SBarry Smith .  nz - number of nonzeros per row (same for all rows)
16132bd5e0b2SLois Curfman McInnes .  nzz - array containing the number of nonzeros in the various rows
16142bd5e0b2SLois Curfman McInnes          (possibly different for each row) or PETSC_NULL
161517ab2063SBarry Smith 
161617ab2063SBarry Smith    Output Parameter:
1617416022c9SBarry Smith .  A - the matrix
161817ab2063SBarry Smith 
161917ab2063SBarry Smith    Notes:
162017ab2063SBarry Smith    The AIJ format (also called the Yale sparse matrix format or
162117ab2063SBarry Smith    compressed row storage), is fully compatible with standard Fortran 77
16220002213bSLois Curfman McInnes    storage.  That is, the stored row and column indices can begin at
162344cd7ae7SLois Curfman McInnes    either one (as in Fortran) or zero.  See the users' manual for details.
162417ab2063SBarry Smith 
162517ab2063SBarry Smith    Specify the preallocated storage with either nz or nnz (not both).
1626a40aa06bSLois Curfman McInnes    Set nz=PETSC_DEFAULT and nnz=PETSC_NULL for PETSc to control dynamic memory
16273d323bbdSBarry Smith    allocation.  For large problems you MUST preallocate memory or you
16286da5968aSLois Curfman McInnes    will get TERRIBLE performance, see the users' manual chapter on matrices.
162917ab2063SBarry Smith 
1630682d7d0cSBarry Smith    By default, this format uses inodes (identical nodes) when possible, to
1631682d7d0cSBarry Smith    improve numerical efficiency of Matrix vector products and solves. We
1632682d7d0cSBarry Smith    search for consecutive rows with the same nonzero structure, thereby
16336c7ebb05SLois Curfman McInnes    reusing matrix information to achieve increased efficiency.
16346c7ebb05SLois Curfman McInnes 
16356c7ebb05SLois Curfman McInnes    Options Database Keys:
16366c7ebb05SLois Curfman McInnes $    -mat_aij_no_inode  - Do not use inodes
16376e62573dSLois Curfman McInnes $    -mat_aij_inode_limit <limit> - Set inode limit.
16386e62573dSLois Curfman McInnes $        (max limit=5)
16396e62573dSLois Curfman McInnes $    -mat_aij_oneindex - Internally use indexing starting at 1
16406e62573dSLois Curfman McInnes $        rather than 0.  Note: When calling MatSetValues(),
16416e62573dSLois Curfman McInnes $        the user still MUST index entries starting at 0!
164217ab2063SBarry Smith 
164317ab2063SBarry Smith .seealso: MatCreate(), MatCreateMPIAIJ(), MatSetValues()
164417ab2063SBarry Smith @*/
1645416022c9SBarry Smith int MatCreateSeqAIJ(MPI_Comm comm,int m,int n,int nz,int *nnz, Mat *A)
164617ab2063SBarry Smith {
1647416022c9SBarry Smith   Mat        B;
1648416022c9SBarry Smith   Mat_SeqAIJ *b;
16496945ee14SBarry Smith   int        i, len, ierr, flg,size;
16506945ee14SBarry Smith 
16516945ee14SBarry Smith   MPI_Comm_size(comm,&size);
1652e3372554SBarry Smith   if (size > 1) SETERRQ(1,0,"Comm must be of size 1");
1653d5d45c9bSBarry Smith 
1654416022c9SBarry Smith   *A                  = 0;
16550452661fSBarry Smith   PetscHeaderCreate(B,_Mat,MAT_COOKIE,MATSEQAIJ,comm);
1656416022c9SBarry Smith   PLogObjectCreate(B);
16570452661fSBarry Smith   B->data             = (void *) (b = PetscNew(Mat_SeqAIJ)); CHKPTRQ(b);
165844cd7ae7SLois Curfman McInnes   PetscMemzero(b,sizeof(Mat_SeqAIJ));
1659416022c9SBarry Smith   PetscMemcpy(&B->ops,&MatOps,sizeof(struct _MatOps));
1660416022c9SBarry Smith   B->destroy          = MatDestroy_SeqAIJ;
1661416022c9SBarry Smith   B->view             = MatView_SeqAIJ;
1662416022c9SBarry Smith   B->factor           = 0;
1663416022c9SBarry Smith   B->lupivotthreshold = 1.0;
166490f02eecSBarry Smith   B->mapping          = 0;
16657a743949SBarry Smith   ierr = OptionsGetDouble(PETSC_NULL,"-mat_lu_pivotthreshold",&B->lupivotthreshold,
166669957df2SSatish Balay                           &flg); CHKERRQ(ierr);
16677a743949SBarry Smith   b->ilu_preserve_row_sums = PETSC_FALSE;
16687a743949SBarry Smith   ierr = OptionsHasName(PETSC_NULL,"-pc_ilu_preserve_row_sums",
16697a743949SBarry Smith                         (int*) &b->ilu_preserve_row_sums); CHKERRQ(ierr);
1670416022c9SBarry Smith   b->row              = 0;
1671416022c9SBarry Smith   b->col              = 0;
1672416022c9SBarry Smith   b->indexshift       = 0;
1673b810aeb4SBarry Smith   b->reallocs         = 0;
167469957df2SSatish Balay   ierr = OptionsHasName(PETSC_NULL,"-mat_aij_oneindex", &flg); CHKERRQ(ierr);
167569957df2SSatish Balay   if (flg) b->indexshift = -1;
167617ab2063SBarry Smith 
167744cd7ae7SLois Curfman McInnes   b->m = m; B->m = m; B->M = m;
167844cd7ae7SLois Curfman McInnes   b->n = n; B->n = n; B->N = n;
16790452661fSBarry Smith   b->imax = (int *) PetscMalloc( (m+1)*sizeof(int) ); CHKPTRQ(b->imax);
1680b4fd4287SBarry Smith   if (nnz == PETSC_NULL) {
16817b8455f0SLois Curfman McInnes     if (nz == PETSC_DEFAULT) nz = 10;
16827b8455f0SLois Curfman McInnes     else if (nz <= 0)        nz = 1;
1683416022c9SBarry Smith     for ( i=0; i<m; i++ ) b->imax[i] = nz;
168417ab2063SBarry Smith     nz = nz*m;
168517ab2063SBarry Smith   }
168617ab2063SBarry Smith   else {
168717ab2063SBarry Smith     nz = 0;
1688416022c9SBarry Smith     for ( i=0; i<m; i++ ) {b->imax[i] = nnz[i]; nz += nnz[i];}
168917ab2063SBarry Smith   }
169017ab2063SBarry Smith 
169117ab2063SBarry Smith   /* allocate the matrix space */
1692416022c9SBarry Smith   len     = nz*(sizeof(int) + sizeof(Scalar)) + (b->m+1)*sizeof(int);
16930452661fSBarry Smith   b->a  = (Scalar *) PetscMalloc( len ); CHKPTRQ(b->a);
1694416022c9SBarry Smith   b->j  = (int *) (b->a + nz);
1695cddf8d76SBarry Smith   PetscMemzero(b->j,nz*sizeof(int));
1696416022c9SBarry Smith   b->i  = b->j + nz;
1697416022c9SBarry Smith   b->singlemalloc = 1;
169817ab2063SBarry Smith 
1699416022c9SBarry Smith   b->i[0] = -b->indexshift;
170017ab2063SBarry Smith   for (i=1; i<m+1; i++) {
1701416022c9SBarry Smith     b->i[i] = b->i[i-1] + b->imax[i-1];
170217ab2063SBarry Smith   }
170317ab2063SBarry Smith 
1704416022c9SBarry Smith   /* b->ilen will count nonzeros in each row so far. */
17050452661fSBarry Smith   b->ilen = (int *) PetscMalloc((m+1)*sizeof(int));
1706416022c9SBarry Smith   PLogObjectMemory(B,len+2*(m+1)*sizeof(int)+sizeof(struct _Mat)+sizeof(Mat_SeqAIJ));
1707416022c9SBarry Smith   for ( i=0; i<b->m; i++ ) { b->ilen[i] = 0;}
170817ab2063SBarry Smith 
1709416022c9SBarry Smith   b->nz               = 0;
1710416022c9SBarry Smith   b->maxnz            = nz;
1711416022c9SBarry Smith   b->sorted           = 0;
1712416022c9SBarry Smith   b->roworiented      = 1;
1713416022c9SBarry Smith   b->nonew            = 0;
1714416022c9SBarry Smith   b->diag             = 0;
1715416022c9SBarry Smith   b->solve_work       = 0;
171671bd300dSLois Curfman McInnes   b->spptr            = 0;
1717754ec7b1SSatish Balay   b->inode.node_count = 0;
1718754ec7b1SSatish Balay   b->inode.size       = 0;
17196c7ebb05SLois Curfman McInnes   b->inode.limit      = 5;
17206c7ebb05SLois Curfman McInnes   b->inode.max_limit  = 5;
17214e220ebcSLois Curfman McInnes   B->info.nz_unneeded = (double)b->maxnz;
172217ab2063SBarry Smith 
1723416022c9SBarry Smith   *A = B;
17244e220ebcSLois Curfman McInnes 
17254b14c69eSBarry Smith   /*  SuperLU is not currently supported through PETSc */
17264b14c69eSBarry Smith #if defined(HAVE_SUPERLU)
172769957df2SSatish Balay   ierr = OptionsHasName(PETSC_NULL,"-mat_aij_superlu", &flg); CHKERRQ(ierr);
172869957df2SSatish Balay   if (flg) { ierr = MatUseSuperLU_SeqAIJ(B); CHKERRQ(ierr); }
17294b14c69eSBarry Smith #endif
173069957df2SSatish Balay   ierr = OptionsHasName(PETSC_NULL,"-mat_aij_essl", &flg); CHKERRQ(ierr);
173169957df2SSatish Balay   if (flg) { ierr = MatUseEssl_SeqAIJ(B); CHKERRQ(ierr); }
173269957df2SSatish Balay   ierr = OptionsHasName(PETSC_NULL,"-mat_aij_dxml", &flg); CHKERRQ(ierr);
173369957df2SSatish Balay   if (flg) {
1734e3372554SBarry Smith     if (!b->indexshift) SETERRQ(1,0,"need -mat_aij_oneindex with -mat_aij_dxml");
1735416022c9SBarry Smith     ierr = MatUseDXML_SeqAIJ(B); CHKERRQ(ierr);
173617ab2063SBarry Smith   }
173769957df2SSatish Balay   ierr = OptionsHasName(PETSC_NULL,"-help", &flg); CHKERRQ(ierr);
173869957df2SSatish Balay   if (flg) {ierr = MatPrintHelp(B); CHKERRQ(ierr); }
173917ab2063SBarry Smith   return 0;
174017ab2063SBarry Smith }
174117ab2063SBarry Smith 
17425615d1e5SSatish Balay #undef __FUNC__
17435615d1e5SSatish Balay #define __FUNC__ "MatConvertSameType_SeqAIJ"
17443d1612f7SBarry Smith int MatConvertSameType_SeqAIJ(Mat A,Mat *B,int cpvalues)
174517ab2063SBarry Smith {
1746416022c9SBarry Smith   Mat        C;
1747416022c9SBarry Smith   Mat_SeqAIJ *c,*a = (Mat_SeqAIJ *) A->data;
174808480c60SBarry Smith   int        i,len, m = a->m,shift = a->indexshift;
174917ab2063SBarry Smith 
17504043dd9cSLois Curfman McInnes   *B = 0;
17510452661fSBarry Smith   PetscHeaderCreate(C,_Mat,MAT_COOKIE,MATSEQAIJ,A->comm);
1752416022c9SBarry Smith   PLogObjectCreate(C);
17530452661fSBarry Smith   C->data       = (void *) (c = PetscNew(Mat_SeqAIJ)); CHKPTRQ(c);
175441c01911SSatish Balay   PetscMemcpy(&C->ops,&A->ops,sizeof(struct _MatOps));
1755416022c9SBarry Smith   C->destroy    = MatDestroy_SeqAIJ;
1756416022c9SBarry Smith   C->view       = MatView_SeqAIJ;
1757416022c9SBarry Smith   C->factor     = A->factor;
1758416022c9SBarry Smith   c->row        = 0;
1759416022c9SBarry Smith   c->col        = 0;
1760416022c9SBarry Smith   c->indexshift = shift;
1761c456f294SBarry Smith   C->assembled  = PETSC_TRUE;
176217ab2063SBarry Smith 
176344cd7ae7SLois Curfman McInnes   c->m = C->m   = a->m;
176444cd7ae7SLois Curfman McInnes   c->n = C->n   = a->n;
176544cd7ae7SLois Curfman McInnes   C->M          = a->m;
176644cd7ae7SLois Curfman McInnes   C->N          = a->n;
176717ab2063SBarry Smith 
17680452661fSBarry Smith   c->imax       = (int *) PetscMalloc((m+1)*sizeof(int)); CHKPTRQ(c->imax);
17690452661fSBarry Smith   c->ilen       = (int *) PetscMalloc((m+1)*sizeof(int)); CHKPTRQ(c->ilen);
177017ab2063SBarry Smith   for ( i=0; i<m; i++ ) {
1771416022c9SBarry Smith     c->imax[i] = a->imax[i];
1772416022c9SBarry Smith     c->ilen[i] = a->ilen[i];
177317ab2063SBarry Smith   }
177417ab2063SBarry Smith 
177517ab2063SBarry Smith   /* allocate the matrix space */
1776416022c9SBarry Smith   c->singlemalloc = 1;
1777416022c9SBarry Smith   len     = (m+1)*sizeof(int)+(a->i[m])*(sizeof(Scalar)+sizeof(int));
17780452661fSBarry Smith   c->a  = (Scalar *) PetscMalloc( len ); CHKPTRQ(c->a);
1779416022c9SBarry Smith   c->j  = (int *) (c->a + a->i[m] + shift);
1780416022c9SBarry Smith   c->i  = c->j + a->i[m] + shift;
1781416022c9SBarry Smith   PetscMemcpy(c->i,a->i,(m+1)*sizeof(int));
178217ab2063SBarry Smith   if (m > 0) {
1783416022c9SBarry Smith     PetscMemcpy(c->j,a->j,(a->i[m]+shift)*sizeof(int));
178408480c60SBarry Smith     if (cpvalues == COPY_VALUES) {
1785416022c9SBarry Smith       PetscMemcpy(c->a,a->a,(a->i[m]+shift)*sizeof(Scalar));
178617ab2063SBarry Smith     }
178708480c60SBarry Smith   }
178817ab2063SBarry Smith 
1789416022c9SBarry Smith   PLogObjectMemory(C,len+2*(m+1)*sizeof(int)+sizeof(struct _Mat)+sizeof(Mat_SeqAIJ));
1790416022c9SBarry Smith   c->sorted      = a->sorted;
1791416022c9SBarry Smith   c->roworiented = a->roworiented;
1792416022c9SBarry Smith   c->nonew       = a->nonew;
17937a743949SBarry Smith   c->ilu_preserve_row_sums = a->ilu_preserve_row_sums;
179417ab2063SBarry Smith 
1795416022c9SBarry Smith   if (a->diag) {
17960452661fSBarry Smith     c->diag = (int *) PetscMalloc( (m+1)*sizeof(int) ); CHKPTRQ(c->diag);
1797416022c9SBarry Smith     PLogObjectMemory(C,(m+1)*sizeof(int));
179817ab2063SBarry Smith     for ( i=0; i<m; i++ ) {
1799416022c9SBarry Smith       c->diag[i] = a->diag[i];
180017ab2063SBarry Smith     }
180117ab2063SBarry Smith   }
1802416022c9SBarry Smith   else c->diag          = 0;
18036c7ebb05SLois Curfman McInnes   c->inode.limit        = a->inode.limit;
18046c7ebb05SLois Curfman McInnes   c->inode.max_limit    = a->inode.max_limit;
1805754ec7b1SSatish Balay   if (a->inode.size){
1806daed632aSSatish Balay     c->inode.size       = (int *) PetscMalloc( (m+1)*sizeof(int) ); CHKPTRQ(c->inode.size);
1807754ec7b1SSatish Balay     c->inode.node_count = a->inode.node_count;
1808daed632aSSatish Balay     PetscMemcpy( c->inode.size, a->inode.size, (m+1)*sizeof(int));
1809754ec7b1SSatish Balay   } else {
1810754ec7b1SSatish Balay     c->inode.size       = 0;
1811754ec7b1SSatish Balay     c->inode.node_count = 0;
1812754ec7b1SSatish Balay   }
1813416022c9SBarry Smith   c->nz                 = a->nz;
1814416022c9SBarry Smith   c->maxnz              = a->maxnz;
1815416022c9SBarry Smith   c->solve_work         = 0;
181676dd722bSSatish Balay   c->spptr              = 0;      /* Dangerous -I'm throwing away a->spptr */
1817754ec7b1SSatish Balay 
1818416022c9SBarry Smith   *B = C;
181917ab2063SBarry Smith   return 0;
182017ab2063SBarry Smith }
182117ab2063SBarry Smith 
18225615d1e5SSatish Balay #undef __FUNC__
18235615d1e5SSatish Balay #define __FUNC__ "MatLoad_SeqAIJ"
182419bcc07fSBarry Smith int MatLoad_SeqAIJ(Viewer viewer,MatType type,Mat *A)
182517ab2063SBarry Smith {
1826416022c9SBarry Smith   Mat_SeqAIJ   *a;
1827416022c9SBarry Smith   Mat          B;
182817699dbbSLois Curfman McInnes   int          i, nz, ierr, fd, header[4],size,*rowlengths = 0,M,N,shift;
1829bcd2baecSBarry Smith   MPI_Comm     comm;
183017ab2063SBarry Smith 
183119bcc07fSBarry Smith   PetscObjectGetComm((PetscObject) viewer,&comm);
183217699dbbSLois Curfman McInnes   MPI_Comm_size(comm,&size);
1833e3372554SBarry Smith   if (size > 1) SETERRQ(1,0,"view must have one processor");
183490ace30eSBarry Smith   ierr = ViewerBinaryGetDescriptor(viewer,&fd); CHKERRQ(ierr);
183577c4ece6SBarry Smith   ierr = PetscBinaryRead(fd,header,4,BINARY_INT); CHKERRQ(ierr);
1836e3372554SBarry Smith   if (header[0] != MAT_COOKIE) SETERRQ(1,0,"not matrix object in file");
183717ab2063SBarry Smith   M = header[1]; N = header[2]; nz = header[3];
183817ab2063SBarry Smith 
183917ab2063SBarry Smith   /* read in row lengths */
18400452661fSBarry Smith   rowlengths = (int*) PetscMalloc( M*sizeof(int) ); CHKPTRQ(rowlengths);
184177c4ece6SBarry Smith   ierr = PetscBinaryRead(fd,rowlengths,M,BINARY_INT); CHKERRQ(ierr);
184217ab2063SBarry Smith 
184317ab2063SBarry Smith   /* create our matrix */
1844416022c9SBarry Smith   ierr = MatCreateSeqAIJ(comm,M,N,0,rowlengths,A); CHKERRQ(ierr);
1845416022c9SBarry Smith   B = *A;
1846416022c9SBarry Smith   a = (Mat_SeqAIJ *) B->data;
1847416022c9SBarry Smith   shift = a->indexshift;
184817ab2063SBarry Smith 
184917ab2063SBarry Smith   /* read in column indices and adjust for Fortran indexing*/
185077c4ece6SBarry Smith   ierr = PetscBinaryRead(fd,a->j,nz,BINARY_INT); CHKERRQ(ierr);
185117ab2063SBarry Smith   if (shift) {
185217ab2063SBarry Smith     for ( i=0; i<nz; i++ ) {
1853416022c9SBarry Smith       a->j[i] += 1;
185417ab2063SBarry Smith     }
185517ab2063SBarry Smith   }
185617ab2063SBarry Smith 
185717ab2063SBarry Smith   /* read in nonzero values */
185877c4ece6SBarry Smith   ierr = PetscBinaryRead(fd,a->a,nz,BINARY_SCALAR); CHKERRQ(ierr);
185917ab2063SBarry Smith 
186017ab2063SBarry Smith   /* set matrix "i" values */
1861416022c9SBarry Smith   a->i[0] = -shift;
186217ab2063SBarry Smith   for ( i=1; i<= M; i++ ) {
1863416022c9SBarry Smith     a->i[i]      = a->i[i-1] + rowlengths[i-1];
1864416022c9SBarry Smith     a->ilen[i-1] = rowlengths[i-1];
186517ab2063SBarry Smith   }
18660452661fSBarry Smith   PetscFree(rowlengths);
186717ab2063SBarry Smith 
18686d4a8577SBarry Smith   ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr);
18696d4a8577SBarry Smith   ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr);
187017ab2063SBarry Smith   return 0;
187117ab2063SBarry Smith }
187217ab2063SBarry Smith 
18735615d1e5SSatish Balay #undef __FUNC__
18745615d1e5SSatish Balay #define __FUNC__ "MatEqual_SeqAIJ"
18758f6be9afSLois Curfman McInnes int MatEqual_SeqAIJ(Mat A,Mat B, PetscTruth* flg)
18767264ac53SSatish Balay {
18777264ac53SSatish Balay   Mat_SeqAIJ *a = (Mat_SeqAIJ *)A->data, *b = (Mat_SeqAIJ *)B->data;
18787264ac53SSatish Balay 
1879e3372554SBarry Smith   if (B->type !=MATSEQAIJ)SETERRQ(1,0,"Matrices must be same type");
18807264ac53SSatish Balay 
18817264ac53SSatish Balay   /* If the  matrix dimensions are not equal, or no of nonzeros or shift */
18827264ac53SSatish Balay   if ((a->m != b->m ) || (a->n !=b->n) ||( a->nz != b->nz)||
1883bcd2baecSBarry Smith       (a->indexshift != b->indexshift)) {
188477c4ece6SBarry Smith     *flg = PETSC_FALSE; return 0;
1885bcd2baecSBarry Smith   }
18867264ac53SSatish Balay 
18877264ac53SSatish Balay   /* if the a->i are the same */
1888bcd2baecSBarry Smith   if (PetscMemcmp(a->i,b->i, (a->n+1)*sizeof(int))) {
188977c4ece6SBarry Smith     *flg = PETSC_FALSE; return 0;
18907264ac53SSatish Balay   }
18917264ac53SSatish Balay 
18927264ac53SSatish Balay   /* if a->j are the same */
1893bcd2baecSBarry Smith   if (PetscMemcmp(a->j, b->j, (a->nz)*sizeof(int))) {
189477c4ece6SBarry Smith     *flg = PETSC_FALSE; return 0;
1895bcd2baecSBarry Smith   }
1896bcd2baecSBarry Smith 
1897bcd2baecSBarry Smith   /* if a->a are the same */
189819bcc07fSBarry Smith   if (PetscMemcmp(a->a, b->a, (a->nz)*sizeof(Scalar))) {
189977c4ece6SBarry Smith     *flg = PETSC_FALSE; return 0;
19007264ac53SSatish Balay   }
190177c4ece6SBarry Smith   *flg = PETSC_TRUE;
19027264ac53SSatish Balay   return 0;
19037264ac53SSatish Balay 
19047264ac53SSatish Balay }
1905