xref: /petsc/src/mat/impls/aij/seq/aij.c (revision eed8681046e740876c451765b135de56ba78c790)
117ab2063SBarry Smith #ifndef lint
2*eed86810SBarry Smith static char vcid[] = "$Id: aij.c,v 1.219 1997/05/23 18:38:26 balay Exp bsmith $";
317ab2063SBarry Smith #endif
417ab2063SBarry Smith 
5d5d45c9bSBarry Smith /*
63369ce9aSBarry Smith     Defines the basic matrix operations for the AIJ (compressed row)
7d5d45c9bSBarry Smith   matrix storage format.
8d5d45c9bSBarry Smith */
93369ce9aSBarry Smith 
103369ce9aSBarry Smith #include "pinclude/pviewer.h"
113369ce9aSBarry Smith #include "sys.h"
1270f55243SBarry Smith #include "src/mat/impls/aij/seq/aij.h"
13f5eb4b81SSatish Balay #include "src/vec/vecimpl.h"
14f5eb4b81SSatish Balay #include "src/inline/spops.h"
15e4d965acSSatish Balay #include "petsc.h"
16f5eb4b81SSatish Balay #include "src/inline/bitarray.h"
1717ab2063SBarry Smith 
18a2ce50c7SBarry Smith /*
19a2ce50c7SBarry Smith     Basic AIJ format ILU based on drop tolerance
20a2ce50c7SBarry Smith */
215615d1e5SSatish Balay #undef __FUNC__
225615d1e5SSatish Balay #define __FUNC__ "MatILUDTFactor_SeqAIJ"
23a2ce50c7SBarry Smith int MatILUDTFactor_SeqAIJ(Mat A,double dt,int maxnz,IS row,IS col,Mat *fact)
24a2ce50c7SBarry Smith {
25a2ce50c7SBarry Smith   /* Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; */
26a2ce50c7SBarry Smith   int        ierr = 1;
27a2ce50c7SBarry Smith 
28e3372554SBarry Smith   SETERRQ(ierr,0,"Not implemented");
29a2ce50c7SBarry Smith }
30a2ce50c7SBarry Smith 
31bcd2baecSBarry Smith extern int MatToSymmetricIJ_SeqAIJ(int,int*,int*,int,int,int**,int**);
3217ab2063SBarry Smith 
335615d1e5SSatish Balay #undef __FUNC__
345615d1e5SSatish Balay #define __FUNC__ "MatGetRowIJ_SeqAIJ"
358f6be9afSLois Curfman McInnes int MatGetRowIJ_SeqAIJ(Mat A,int oshift,PetscTruth symmetric,int *m,int **ia,int **ja,
366945ee14SBarry Smith                            PetscTruth *done)
3717ab2063SBarry Smith {
38416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
396945ee14SBarry Smith   int        ierr,i,ishift;
4017ab2063SBarry Smith 
4131625ec5SSatish Balay   *m     = A->m;
426945ee14SBarry Smith   if (!ia) return 0;
436945ee14SBarry Smith   ishift = a->indexshift;
446945ee14SBarry Smith   if (symmetric) {
4531625ec5SSatish Balay     ierr = MatToSymmetricIJ_SeqAIJ(a->m,a->i,a->j,ishift,oshift,ia,ja); CHKERRQ(ierr);
466945ee14SBarry Smith   } else if (oshift == 0 && ishift == -1) {
4731625ec5SSatish Balay     int nz = a->i[a->m];
483b2fbd54SBarry Smith     /* malloc space and  subtract 1 from i and j indices */
4931625ec5SSatish Balay     *ia = (int *) PetscMalloc( (a->m+1)*sizeof(int) ); CHKPTRQ(*ia);
503b2fbd54SBarry Smith     *ja = (int *) PetscMalloc( (nz+1)*sizeof(int) ); CHKPTRQ(*ja);
513b2fbd54SBarry Smith     for ( i=0; i<nz; i++ ) (*ja)[i] = a->j[i] - 1;
5231625ec5SSatish Balay     for ( i=0; i<a->m+1; i++ ) (*ia)[i] = a->i[i] - 1;
536945ee14SBarry Smith   } else if (oshift == 1 && ishift == 0) {
5431625ec5SSatish Balay     int nz = a->i[a->m] + 1;
553b2fbd54SBarry Smith     /* malloc space and  add 1 to i and j indices */
5631625ec5SSatish Balay     *ia = (int *) PetscMalloc( (a->m+1)*sizeof(int) ); CHKPTRQ(*ia);
573b2fbd54SBarry Smith     *ja = (int *) PetscMalloc( (nz+1)*sizeof(int) ); CHKPTRQ(*ja);
583b2fbd54SBarry Smith     for ( i=0; i<nz; i++ ) (*ja)[i] = a->j[i] + 1;
5931625ec5SSatish Balay     for ( i=0; i<a->m+1; i++ ) (*ia)[i] = a->i[i] + 1;
606945ee14SBarry Smith   } else {
616945ee14SBarry Smith     *ia = a->i; *ja = a->j;
62a2ce50c7SBarry Smith   }
63a2ce50c7SBarry Smith 
64a2744918SBarry Smith   return 0;
65a2744918SBarry Smith }
66a2744918SBarry Smith 
675615d1e5SSatish Balay #undef __FUNC__
685615d1e5SSatish Balay #define __FUNC__ "MatRestoreRowIJ_SeqAIJ"
698f6be9afSLois Curfman McInnes int MatRestoreRowIJ_SeqAIJ(Mat A,int oshift,PetscTruth symmetric,int *n,int **ia,int **ja,
706945ee14SBarry Smith                                PetscTruth *done)
716945ee14SBarry Smith {
726945ee14SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
733b2fbd54SBarry Smith   int        ishift = a->indexshift;
746945ee14SBarry Smith 
756945ee14SBarry Smith   if (!ia) return 0;
763b2fbd54SBarry Smith   if (symmetric || (oshift == 0 && ishift == -1) || (oshift == 1 && ishift == 0)) {
776945ee14SBarry Smith     PetscFree(*ia);
786945ee14SBarry Smith     PetscFree(*ja);
79bcd2baecSBarry Smith   }
8017ab2063SBarry Smith   return 0;
8117ab2063SBarry Smith }
8217ab2063SBarry Smith 
835615d1e5SSatish Balay #undef __FUNC__
845615d1e5SSatish Balay #define __FUNC__ "MatGetColumnIJ_SeqAIJ"
8543a90d84SBarry Smith int MatGetColumnIJ_SeqAIJ(Mat A,int oshift,PetscTruth symmetric,int *nn,int **ia,int **ja,
863b2fbd54SBarry Smith                            PetscTruth *done)
873b2fbd54SBarry Smith {
883b2fbd54SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
89a93ec695SBarry Smith   int        ierr,i,ishift = a->indexshift,*collengths,*cia,*cja,n = A->n,m = A->m;
90a93ec695SBarry Smith   int        nz = a->i[m]+ishift,row,*jj,mr,col;
913b2fbd54SBarry Smith 
923b2fbd54SBarry Smith   *nn     = A->n;
933b2fbd54SBarry Smith   if (!ia) return 0;
943b2fbd54SBarry Smith   if (symmetric) {
95179192dfSSatish Balay     ierr = MatToSymmetricIJ_SeqAIJ(a->m,a->i,a->j,ishift,oshift,ia,ja); CHKERRQ(ierr);
963b2fbd54SBarry Smith   } else {
9761d2ded1SBarry Smith     collengths = (int *) PetscMalloc( (n+1)*sizeof(int) ); CHKPTRQ(collengths);
983b2fbd54SBarry Smith     PetscMemzero(collengths,n*sizeof(int));
993b2fbd54SBarry Smith     cia        = (int *) PetscMalloc( (n+1)*sizeof(int) ); CHKPTRQ(cia);
100a93ec695SBarry Smith     cja        = (int *) PetscMalloc( (nz+1)*sizeof(int) ); CHKPTRQ(cja);
1013b2fbd54SBarry Smith     jj = a->j;
1023b2fbd54SBarry Smith     for ( i=0; i<nz; i++ ) {
1033b2fbd54SBarry Smith       collengths[jj[i] + ishift]++;
1043b2fbd54SBarry Smith     }
1053b2fbd54SBarry Smith     cia[0] = oshift;
1063b2fbd54SBarry Smith     for ( i=0; i<n; i++) {
1073b2fbd54SBarry Smith       cia[i+1] = cia[i] + collengths[i];
1083b2fbd54SBarry Smith     }
1093b2fbd54SBarry Smith     PetscMemzero(collengths,n*sizeof(int));
1103b2fbd54SBarry Smith     jj = a->j;
111a93ec695SBarry Smith     for ( row=0; row<m; row++ ) {
112a93ec695SBarry Smith       mr = a->i[row+1] - a->i[row];
113a93ec695SBarry Smith       for ( i=0; i<mr; i++ ) {
1143b2fbd54SBarry Smith         col = *jj++ + ishift;
1153b2fbd54SBarry Smith         cja[cia[col] + collengths[col]++ - oshift] = row + oshift;
1163b2fbd54SBarry Smith       }
1173b2fbd54SBarry Smith     }
1183b2fbd54SBarry Smith     PetscFree(collengths);
1193b2fbd54SBarry Smith     *ia = cia; *ja = cja;
1203b2fbd54SBarry Smith   }
1213b2fbd54SBarry Smith 
1223b2fbd54SBarry Smith   return 0;
1233b2fbd54SBarry Smith }
1243b2fbd54SBarry Smith 
1255615d1e5SSatish Balay #undef __FUNC__
1265615d1e5SSatish Balay #define __FUNC__ "MatRestoreColumnIJ_SeqAIJ"
12743a90d84SBarry Smith int MatRestoreColumnIJ_SeqAIJ(Mat A,int oshift,PetscTruth symmetric,int *n,int **ia,
1283b2fbd54SBarry Smith                                      int **ja,PetscTruth *done)
1293b2fbd54SBarry Smith {
1303b2fbd54SBarry Smith   if (!ia) return 0;
1313b2fbd54SBarry Smith 
1323b2fbd54SBarry Smith   PetscFree(*ia);
1333b2fbd54SBarry Smith   PetscFree(*ja);
1343b2fbd54SBarry Smith 
1353b2fbd54SBarry Smith   return 0;
1363b2fbd54SBarry Smith }
1373b2fbd54SBarry Smith 
138227d817aSBarry Smith #define CHUNKSIZE   15
13917ab2063SBarry Smith 
1405615d1e5SSatish Balay #undef __FUNC__
1415615d1e5SSatish Balay #define __FUNC__ "MatSetValues_SeqAIJ"
14261d2ded1SBarry Smith int MatSetValues_SeqAIJ(Mat A,int m,int *im,int n,int *in,Scalar *v,InsertMode is)
14317ab2063SBarry Smith {
144416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
145416022c9SBarry Smith   int        *rp,k,low,high,t,ii,row,nrow,i,col,l,rmax, N, sorted = a->sorted;
1464b0e389bSBarry Smith   int        *imax = a->imax, *ai = a->i, *ailen = a->ilen,roworiented = a->roworiented;
147d5d45c9bSBarry Smith   int        *aj = a->j, nonew = a->nonew,shift = a->indexshift;
148416022c9SBarry Smith   Scalar     *ap,value, *aa = a->a;
14917ab2063SBarry Smith 
15017ab2063SBarry Smith   for ( k=0; k<m; k++ ) { /* loop over added rows */
151416022c9SBarry Smith     row  = im[k];
1523b2fbd54SBarry Smith #if defined(PETSC_BOPT_g)
153e3372554SBarry Smith     if (row < 0) SETERRQ(1,0,"Negative row");
154e3372554SBarry Smith     if (row >= a->m) SETERRQ(1,0,"Row too large");
1553b2fbd54SBarry Smith #endif
15617ab2063SBarry Smith     rp   = aj + ai[row] + shift; ap = aa + ai[row] + shift;
15717ab2063SBarry Smith     rmax = imax[row]; nrow = ailen[row];
158416022c9SBarry Smith     low = 0;
15917ab2063SBarry Smith     for ( l=0; l<n; l++ ) { /* loop over added columns */
1603b2fbd54SBarry Smith #if defined(PETSC_BOPT_g)
161e3372554SBarry Smith       if (in[l] < 0) SETERRQ(1,0,"Negative column");
162e3372554SBarry Smith       if (in[l] >= a->n) SETERRQ(1,0,"Column too large");
1633b2fbd54SBarry Smith #endif
1644b0e389bSBarry Smith       col = in[l] - shift;
1654b0e389bSBarry Smith       if (roworiented) {
1664b0e389bSBarry Smith         value = *v++;
1674b0e389bSBarry Smith       }
1684b0e389bSBarry Smith       else {
1694b0e389bSBarry Smith         value = v[k + l*m];
1704b0e389bSBarry Smith       }
171416022c9SBarry Smith       if (!sorted) low = 0; high = nrow;
172416022c9SBarry Smith       while (high-low > 5) {
173416022c9SBarry Smith         t = (low+high)/2;
174416022c9SBarry Smith         if (rp[t] > col) high = t;
175416022c9SBarry Smith         else             low  = t;
17617ab2063SBarry Smith       }
177416022c9SBarry Smith       for ( i=low; i<high; i++ ) {
17817ab2063SBarry Smith         if (rp[i] > col) break;
17917ab2063SBarry Smith         if (rp[i] == col) {
180416022c9SBarry Smith           if (is == ADD_VALUES) ap[i] += value;
18117ab2063SBarry Smith           else                  ap[i] = value;
18217ab2063SBarry Smith           goto noinsert;
18317ab2063SBarry Smith         }
18417ab2063SBarry Smith       }
185c2653b3dSLois Curfman McInnes       if (nonew == 1) goto noinsert;
18611ebbc71SLois Curfman McInnes       else if (nonew == -1) SETERRQ(1,0,"Inserting a new nonzero in the matrix");
18717ab2063SBarry Smith       if (nrow >= rmax) {
18817ab2063SBarry Smith         /* there is no extra room in row, therefore enlarge */
189416022c9SBarry Smith         int    new_nz = ai[a->m] + CHUNKSIZE,len,*new_i,*new_j;
19017ab2063SBarry Smith         Scalar *new_a;
19117ab2063SBarry Smith 
19211ebbc71SLois Curfman McInnes         if (nonew == -2) SETERRQ(1,0,"Inserting a new nonzero in the matrix");
19396854ed6SLois Curfman McInnes 
19417ab2063SBarry Smith         /* malloc new storage space */
195416022c9SBarry Smith         len     = new_nz*(sizeof(int)+sizeof(Scalar))+(a->m+1)*sizeof(int);
1960452661fSBarry Smith         new_a   = (Scalar *) PetscMalloc( len ); CHKPTRQ(new_a);
19717ab2063SBarry Smith         new_j   = (int *) (new_a + new_nz);
19817ab2063SBarry Smith         new_i   = new_j + new_nz;
19917ab2063SBarry Smith 
20017ab2063SBarry Smith         /* copy over old data into new slots */
20117ab2063SBarry Smith         for ( ii=0; ii<row+1; ii++ ) {new_i[ii] = ai[ii];}
202416022c9SBarry Smith         for ( ii=row+1; ii<a->m+1; ii++ ) {new_i[ii] = ai[ii]+CHUNKSIZE;}
203416022c9SBarry Smith         PetscMemcpy(new_j,aj,(ai[row]+nrow+shift)*sizeof(int));
204416022c9SBarry Smith         len = (new_nz - CHUNKSIZE - ai[row] - nrow - shift);
205416022c9SBarry Smith         PetscMemcpy(new_j+ai[row]+shift+nrow+CHUNKSIZE,aj+ai[row]+shift+nrow,
20617ab2063SBarry Smith                                                            len*sizeof(int));
207416022c9SBarry Smith         PetscMemcpy(new_a,aa,(ai[row]+nrow+shift)*sizeof(Scalar));
208416022c9SBarry Smith         PetscMemcpy(new_a+ai[row]+shift+nrow+CHUNKSIZE,aa+ai[row]+shift+nrow,
20917ab2063SBarry Smith                                                            len*sizeof(Scalar));
21017ab2063SBarry Smith         /* free up old matrix storage */
2110452661fSBarry Smith         PetscFree(a->a);
2120452661fSBarry Smith         if (!a->singlemalloc) {PetscFree(a->i);PetscFree(a->j);}
213416022c9SBarry Smith         aa = a->a = new_a; ai = a->i = new_i; aj = a->j = new_j;
214416022c9SBarry Smith         a->singlemalloc = 1;
21517ab2063SBarry Smith 
21617ab2063SBarry Smith         rp   = aj + ai[row] + shift; ap = aa + ai[row] + shift;
217416022c9SBarry Smith         rmax = imax[row] = imax[row] + CHUNKSIZE;
218416022c9SBarry Smith         PLogObjectMemory(A,CHUNKSIZE*(sizeof(int) + sizeof(Scalar)));
219416022c9SBarry Smith         a->maxnz += CHUNKSIZE;
220b810aeb4SBarry Smith         a->reallocs++;
22117ab2063SBarry Smith       }
222416022c9SBarry Smith       N = nrow++ - 1; a->nz++;
223416022c9SBarry Smith       /* shift up all the later entries in this row */
224416022c9SBarry Smith       for ( ii=N; ii>=i; ii-- ) {
22517ab2063SBarry Smith         rp[ii+1] = rp[ii];
22617ab2063SBarry Smith         ap[ii+1] = ap[ii];
22717ab2063SBarry Smith       }
22817ab2063SBarry Smith       rp[i] = col;
22917ab2063SBarry Smith       ap[i] = value;
23017ab2063SBarry Smith       noinsert:;
231416022c9SBarry Smith       low = i + 1;
23217ab2063SBarry Smith     }
23317ab2063SBarry Smith     ailen[row] = nrow;
23417ab2063SBarry Smith   }
23517ab2063SBarry Smith   return 0;
23617ab2063SBarry Smith }
23717ab2063SBarry Smith 
2385615d1e5SSatish Balay #undef __FUNC__
2395615d1e5SSatish Balay #define __FUNC__ "MatGetValues_SeqAIJ"
2408f6be9afSLois Curfman McInnes int MatGetValues_SeqAIJ(Mat A,int m,int *im,int n,int *in,Scalar *v)
2417eb43aa7SLois Curfman McInnes {
2427eb43aa7SLois Curfman McInnes   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
243b49de8d1SLois Curfman McInnes   int        *rp, k, low, high, t, row, nrow, i, col, l, *aj = a->j;
2447eb43aa7SLois Curfman McInnes   int        *ai = a->i, *ailen = a->ilen, shift = a->indexshift;
2457eb43aa7SLois Curfman McInnes   Scalar     *ap, *aa = a->a, zero = 0.0;
2467eb43aa7SLois Curfman McInnes 
2477eb43aa7SLois Curfman McInnes   for ( k=0; k<m; k++ ) { /* loop over rows */
2487eb43aa7SLois Curfman McInnes     row  = im[k];
249e3372554SBarry Smith     if (row < 0) SETERRQ(1,0,"Negative row");
250e3372554SBarry Smith     if (row >= a->m) SETERRQ(1,0,"Row too large");
2517eb43aa7SLois Curfman McInnes     rp   = aj + ai[row] + shift; ap = aa + ai[row] + shift;
2527eb43aa7SLois Curfman McInnes     nrow = ailen[row];
2537eb43aa7SLois Curfman McInnes     for ( l=0; l<n; l++ ) { /* loop over columns */
254e3372554SBarry Smith       if (in[l] < 0) SETERRQ(1,0,"Negative column");
255e3372554SBarry Smith       if (in[l] >= a->n) SETERRQ(1,0,"Column too large");
2567eb43aa7SLois Curfman McInnes       col = in[l] - shift;
2577eb43aa7SLois Curfman McInnes       high = nrow; low = 0; /* assume unsorted */
2587eb43aa7SLois Curfman McInnes       while (high-low > 5) {
2597eb43aa7SLois Curfman McInnes         t = (low+high)/2;
2607eb43aa7SLois Curfman McInnes         if (rp[t] > col) high = t;
2617eb43aa7SLois Curfman McInnes         else             low  = t;
2627eb43aa7SLois Curfman McInnes       }
2637eb43aa7SLois Curfman McInnes       for ( i=low; i<high; i++ ) {
2647eb43aa7SLois Curfman McInnes         if (rp[i] > col) break;
2657eb43aa7SLois Curfman McInnes         if (rp[i] == col) {
266b49de8d1SLois Curfman McInnes           *v++ = ap[i];
2677eb43aa7SLois Curfman McInnes           goto finished;
2687eb43aa7SLois Curfman McInnes         }
2697eb43aa7SLois Curfman McInnes       }
270b49de8d1SLois Curfman McInnes       *v++ = zero;
2717eb43aa7SLois Curfman McInnes       finished:;
2727eb43aa7SLois Curfman McInnes     }
2737eb43aa7SLois Curfman McInnes   }
2747eb43aa7SLois Curfman McInnes   return 0;
2757eb43aa7SLois Curfman McInnes }
2767eb43aa7SLois Curfman McInnes 
27717ab2063SBarry Smith 
2785615d1e5SSatish Balay #undef __FUNC__
2795615d1e5SSatish Balay #define __FUNC__ "MatView_SeqAIJ_Binary"
2808f6be9afSLois Curfman McInnes extern int MatView_SeqAIJ_Binary(Mat A,Viewer viewer)
28117ab2063SBarry Smith {
282416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
283416022c9SBarry Smith   int        i, fd, *col_lens, ierr;
28417ab2063SBarry Smith 
28590ace30eSBarry Smith   ierr = ViewerBinaryGetDescriptor(viewer,&fd); CHKERRQ(ierr);
2860452661fSBarry Smith   col_lens = (int *) PetscMalloc( (4+a->m)*sizeof(int) ); CHKPTRQ(col_lens);
287416022c9SBarry Smith   col_lens[0] = MAT_COOKIE;
288416022c9SBarry Smith   col_lens[1] = a->m;
289416022c9SBarry Smith   col_lens[2] = a->n;
290416022c9SBarry Smith   col_lens[3] = a->nz;
291416022c9SBarry Smith 
292416022c9SBarry Smith   /* store lengths of each row and write (including header) to file */
293416022c9SBarry Smith   for ( i=0; i<a->m; i++ ) {
294416022c9SBarry Smith     col_lens[4+i] = a->i[i+1] - a->i[i];
29517ab2063SBarry Smith   }
29677c4ece6SBarry Smith   ierr = PetscBinaryWrite(fd,col_lens,4+a->m,BINARY_INT,1); CHKERRQ(ierr);
2970452661fSBarry Smith   PetscFree(col_lens);
298416022c9SBarry Smith 
299416022c9SBarry Smith   /* store column indices (zero start index) */
300416022c9SBarry Smith   if (a->indexshift) {
301416022c9SBarry Smith     for ( i=0; i<a->nz; i++ ) a->j[i]--;
30217ab2063SBarry Smith   }
30377c4ece6SBarry Smith   ierr = PetscBinaryWrite(fd,a->j,a->nz,BINARY_INT,0); CHKERRQ(ierr);
304416022c9SBarry Smith   if (a->indexshift) {
305416022c9SBarry Smith     for ( i=0; i<a->nz; i++ ) a->j[i]++;
30617ab2063SBarry Smith   }
307416022c9SBarry Smith 
308416022c9SBarry Smith   /* store nonzero values */
30977c4ece6SBarry Smith   ierr = PetscBinaryWrite(fd,a->a,a->nz,BINARY_SCALAR,0); CHKERRQ(ierr);
31017ab2063SBarry Smith   return 0;
31117ab2063SBarry Smith }
312416022c9SBarry Smith 
3135615d1e5SSatish Balay #undef __FUNC__
3145615d1e5SSatish Balay #define __FUNC__ "MatView_SeqAIJ_ASCII"
3158f6be9afSLois Curfman McInnes extern int MatView_SeqAIJ_ASCII(Mat A,Viewer viewer)
316416022c9SBarry Smith {
317416022c9SBarry Smith   Mat_SeqAIJ  *a = (Mat_SeqAIJ *) A->data;
318496e697eSBarry Smith   int         ierr, i,j, m = a->m, shift = a->indexshift, format, flg1,flg2;
31917ab2063SBarry Smith   FILE        *fd;
32017ab2063SBarry Smith   char        *outputname;
32117ab2063SBarry Smith 
32290ace30eSBarry Smith   ierr = ViewerASCIIGetPointer(viewer,&fd); CHKERRQ(ierr);
323416022c9SBarry Smith   ierr = ViewerFileGetOutputname_Private(viewer,&outputname); CHKERRQ(ierr);
32490ace30eSBarry Smith   ierr = ViewerGetFormat(viewer,&format);
325a93ec695SBarry Smith   if (format == VIEWER_FORMAT_ASCII_INFO) {
32695e01e2fSLois Curfman McInnes     return 0;
32795e01e2fSLois Curfman McInnes   }
328a93ec695SBarry Smith   else if (format == VIEWER_FORMAT_ASCII_INFO_LONG) {
329496e697eSBarry Smith     ierr = OptionsHasName(PETSC_NULL,"-mat_aij_no_inode",&flg1); CHKERRQ(ierr);
330496e697eSBarry Smith     ierr = OptionsHasName(PETSC_NULL,"-mat_no_unroll",&flg2); CHKERRQ(ierr);
331496e697eSBarry Smith     if (flg1 || flg2) fprintf(fd,"  not using I-node routines\n");
33295e01e2fSLois Curfman McInnes     else     fprintf(fd,"  using I-node routines: found %d nodes, limit used is %d\n",
33395e01e2fSLois Curfman McInnes         a->inode.node_count,a->inode.limit);
33417ab2063SBarry Smith   }
335a93ec695SBarry Smith   else if (format == VIEWER_FORMAT_ASCII_MATLAB) {
336416022c9SBarry Smith     fprintf(fd,"%% Size = %d %d \n",m,a->n);
3374e220ebcSLois Curfman McInnes     fprintf(fd,"%% Nonzeros = %d \n",a->nz);
3384e220ebcSLois Curfman McInnes     fprintf(fd,"zzz = zeros(%d,3);\n",a->nz);
33917ab2063SBarry Smith     fprintf(fd,"zzz = [\n");
34017ab2063SBarry Smith 
34117ab2063SBarry Smith     for (i=0; i<m; i++) {
342416022c9SBarry Smith       for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
34317ab2063SBarry Smith #if defined(PETSC_COMPLEX)
3446945ee14SBarry Smith         fprintf(fd,"%d %d  %18.16e + %18.16e i \n",i+1,a->j[j]+!shift,real(a->a[j]),
345416022c9SBarry Smith                    imag(a->a[j]));
34617ab2063SBarry Smith #else
3477a743949SBarry Smith         fprintf(fd,"%d %d  %18.16e\n", i+1, a->j[j]+!shift, a->a[j]);
34817ab2063SBarry Smith #endif
34917ab2063SBarry Smith       }
35017ab2063SBarry Smith     }
35117ab2063SBarry Smith     fprintf(fd,"];\n %s = spconvert(zzz);\n",outputname);
35217ab2063SBarry Smith   }
353a93ec695SBarry Smith   else if (format == VIEWER_FORMAT_ASCII_COMMON) {
35444cd7ae7SLois Curfman McInnes     for ( i=0; i<m; i++ ) {
35544cd7ae7SLois Curfman McInnes       fprintf(fd,"row %d:",i);
35644cd7ae7SLois Curfman McInnes       for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
35744cd7ae7SLois Curfman McInnes #if defined(PETSC_COMPLEX)
35844cd7ae7SLois Curfman McInnes         if (imag(a->a[j]) != 0.0 && real(a->a[j]) != 0.0)
35944cd7ae7SLois Curfman McInnes           fprintf(fd," %d %g + %g i",a->j[j]+shift,real(a->a[j]),imag(a->a[j]));
36044cd7ae7SLois Curfman McInnes         else if (real(a->a[j]) != 0.0)
36144cd7ae7SLois Curfman McInnes           fprintf(fd," %d %g ",a->j[j]+shift,real(a->a[j]));
36244cd7ae7SLois Curfman McInnes #else
36344cd7ae7SLois Curfman McInnes         if (a->a[j] != 0.0) fprintf(fd," %d %g ",a->j[j]+shift,a->a[j]);
36444cd7ae7SLois Curfman McInnes #endif
36544cd7ae7SLois Curfman McInnes       }
36644cd7ae7SLois Curfman McInnes       fprintf(fd,"\n");
36744cd7ae7SLois Curfman McInnes     }
36844cd7ae7SLois Curfman McInnes   }
36917ab2063SBarry Smith   else {
37017ab2063SBarry Smith     for ( i=0; i<m; i++ ) {
37117ab2063SBarry Smith       fprintf(fd,"row %d:",i);
372416022c9SBarry Smith       for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
37317ab2063SBarry Smith #if defined(PETSC_COMPLEX)
374416022c9SBarry Smith         if (imag(a->a[j]) != 0.0) {
375416022c9SBarry Smith           fprintf(fd," %d %g + %g i",a->j[j]+shift,real(a->a[j]),imag(a->a[j]));
37617ab2063SBarry Smith         }
37717ab2063SBarry Smith         else {
378416022c9SBarry Smith           fprintf(fd," %d %g ",a->j[j]+shift,real(a->a[j]));
37917ab2063SBarry Smith         }
38017ab2063SBarry Smith #else
381416022c9SBarry Smith         fprintf(fd," %d %g ",a->j[j]+shift,a->a[j]);
38217ab2063SBarry Smith #endif
38317ab2063SBarry Smith       }
38417ab2063SBarry Smith       fprintf(fd,"\n");
38517ab2063SBarry Smith     }
38617ab2063SBarry Smith   }
38717ab2063SBarry Smith   fflush(fd);
388416022c9SBarry Smith   return 0;
389416022c9SBarry Smith }
390416022c9SBarry Smith 
3915615d1e5SSatish Balay #undef __FUNC__
3925615d1e5SSatish Balay #define __FUNC__ "MatView_SeqAIJ_Draw"
3938f6be9afSLois Curfman McInnes extern int MatView_SeqAIJ_Draw(Mat A,Viewer viewer)
394416022c9SBarry Smith {
395416022c9SBarry Smith   Mat_SeqAIJ  *a = (Mat_SeqAIJ *) A->data;
396cddf8d76SBarry Smith   int         ierr, i,j, m = a->m, shift = a->indexshift,pause,color;
3970513a670SBarry Smith   int         format;
39894a9d846SBarry Smith   double      xl,yl,xr,yr,w,h,xc,yc,scale = 1.0,x_l,x_r,y_l,y_r,maxv = 0.0;
399bcd2baecSBarry Smith   Draw        draw;
400cddf8d76SBarry Smith   DrawButton  button;
40119bcc07fSBarry Smith   PetscTruth  isnull;
402cddf8d76SBarry Smith 
4030513a670SBarry Smith   ierr = ViewerDrawGetDraw(viewer,&draw); CHKERRQ(ierr);
4040513a670SBarry Smith   ierr = DrawClear(draw); CHKERRQ(ierr);
4050513a670SBarry Smith   ierr = ViewerGetFormat(viewer,&format); CHKERRQ(ierr);
40619bcc07fSBarry Smith   ierr = DrawIsNull(draw,&isnull); CHKERRQ(ierr); if (isnull) return 0;
40719bcc07fSBarry Smith 
408416022c9SBarry Smith   xr  = a->n; yr = a->m; h = yr/10.0; w = xr/10.0;
409416022c9SBarry Smith   xr += w;    yr += h;  xl = -w;     yl = -h;
410416022c9SBarry Smith   ierr = DrawSetCoordinates(draw,xl,yl,xr,yr); CHKERRQ(ierr);
411416022c9SBarry Smith   /* loop over matrix elements drawing boxes */
4120513a670SBarry Smith 
4130513a670SBarry Smith   if (format != VIEWER_FORMAT_DRAW_CONTOUR) {
4140513a670SBarry Smith     /* Blue for negative, Cyan for zero and  Red for positive */
415cddf8d76SBarry Smith     color = DRAW_BLUE;
416416022c9SBarry Smith     for ( i=0; i<m; i++ ) {
417cddf8d76SBarry Smith       y_l = m - i - 1.0; y_r = y_l + 1.0;
418416022c9SBarry Smith       for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
419cddf8d76SBarry Smith         x_l = a->j[j] + shift; x_r = x_l + 1.0;
420cddf8d76SBarry Smith #if defined(PETSC_COMPLEX)
421cddf8d76SBarry Smith         if (real(a->a[j]) >=  0.) continue;
422cddf8d76SBarry Smith #else
423cddf8d76SBarry Smith         if (a->a[j] >=  0.) continue;
424cddf8d76SBarry Smith #endif
425cddf8d76SBarry Smith         DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);
426cddf8d76SBarry Smith       }
427cddf8d76SBarry Smith     }
428cddf8d76SBarry Smith     color = DRAW_CYAN;
429cddf8d76SBarry Smith     for ( i=0; i<m; i++ ) {
430cddf8d76SBarry Smith       y_l = m - i - 1.0; y_r = y_l + 1.0;
431cddf8d76SBarry Smith       for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
432cddf8d76SBarry Smith         x_l = a->j[j] + shift; x_r = x_l + 1.0;
433cddf8d76SBarry Smith         if (a->a[j] !=  0.) continue;
434cddf8d76SBarry Smith         DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);
435cddf8d76SBarry Smith       }
436cddf8d76SBarry Smith     }
437cddf8d76SBarry Smith     color = DRAW_RED;
438cddf8d76SBarry Smith     for ( i=0; i<m; i++ ) {
439cddf8d76SBarry Smith       y_l = m - i - 1.0; y_r = y_l + 1.0;
440cddf8d76SBarry Smith       for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
441cddf8d76SBarry Smith         x_l = a->j[j] + shift; x_r = x_l + 1.0;
442cddf8d76SBarry Smith #if defined(PETSC_COMPLEX)
443cddf8d76SBarry Smith         if (real(a->a[j]) <=  0.) continue;
444cddf8d76SBarry Smith #else
445cddf8d76SBarry Smith         if (a->a[j] <=  0.) continue;
446cddf8d76SBarry Smith #endif
447cddf8d76SBarry Smith         DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);
448416022c9SBarry Smith       }
449416022c9SBarry Smith     }
4500513a670SBarry Smith   } else {
4510513a670SBarry Smith     /* use contour shading to indicate magnitude of values */
4520513a670SBarry Smith     /* first determine max of all nonzero values */
4530513a670SBarry Smith     int    nz = a->nz,count;
4540513a670SBarry Smith     Draw   popup;
4550513a670SBarry Smith 
4560513a670SBarry Smith     for ( i=0; i<nz; i++ ) {
4570513a670SBarry Smith       if (PetscAbsScalar(a->a[i]) > maxv) maxv = PetscAbsScalar(a->a[i]);
4580513a670SBarry Smith     }
4590513a670SBarry Smith     ierr = DrawCreatePopUp(draw,&popup); CHKERRQ(ierr);
4600513a670SBarry Smith     ierr = DrawScalePopup(popup,0.0,maxv); CHKERRQ(ierr);
4610513a670SBarry Smith     count = 0;
4620513a670SBarry Smith     for ( i=0; i<m; i++ ) {
4630513a670SBarry Smith       y_l = m - i - 1.0; y_r = y_l + 1.0;
4640513a670SBarry Smith       for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
4650513a670SBarry Smith         x_l = a->j[j] + shift; x_r = x_l + 1.0;
4660513a670SBarry Smith         color = 32 + (int) ((200.0 - 32.0)*PetscAbsScalar(a->a[count])/maxv);
4670513a670SBarry Smith         DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);
4680513a670SBarry Smith         count++;
4690513a670SBarry Smith       }
4700513a670SBarry Smith     }
4710513a670SBarry Smith   }
472416022c9SBarry Smith   DrawFlush(draw);
473cddf8d76SBarry Smith   DrawGetPause(draw,&pause);
474cddf8d76SBarry Smith   if (pause >= 0) { PetscSleep(pause); return 0;}
475cddf8d76SBarry Smith 
476cddf8d76SBarry Smith   /* allow the matrix to zoom or shrink */
4776945ee14SBarry Smith   ierr = DrawCheckResizedWindow(draw);
478cddf8d76SBarry Smith   ierr = DrawGetMouseButton(draw,&button,&xc,&yc,0,0);
479cddf8d76SBarry Smith   while (button != BUTTON_RIGHT) {
480cddf8d76SBarry Smith     DrawClear(draw);
481cddf8d76SBarry Smith     if (button == BUTTON_LEFT) scale = .5;
482cddf8d76SBarry Smith     else if (button == BUTTON_CENTER) scale = 2.;
483cddf8d76SBarry Smith     xl = scale*(xl + w - xc) + xc - w*scale;
484cddf8d76SBarry Smith     xr = scale*(xr - w - xc) + xc + w*scale;
485cddf8d76SBarry Smith     yl = scale*(yl + h - yc) + yc - h*scale;
486cddf8d76SBarry Smith     yr = scale*(yr - h - yc) + yc + h*scale;
487cddf8d76SBarry Smith     w *= scale; h *= scale;
488cddf8d76SBarry Smith     ierr = DrawSetCoordinates(draw,xl,yl,xr,yr); CHKERRQ(ierr);
4890513a670SBarry Smith     if (format != VIEWER_FORMAT_DRAW_CONTOUR) {
4900513a670SBarry Smith       /* Blue for negative, Cyan for zero and  Red for positive */
491cddf8d76SBarry Smith       color = DRAW_BLUE;
492cddf8d76SBarry Smith       for ( i=0; i<m; i++ ) {
493cddf8d76SBarry Smith         y_l = m - i - 1.0; y_r = y_l + 1.0;
494cddf8d76SBarry Smith         for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
495cddf8d76SBarry Smith           x_l = a->j[j] + shift; x_r = x_l + 1.0;
496cddf8d76SBarry Smith #if defined(PETSC_COMPLEX)
497cddf8d76SBarry Smith           if (real(a->a[j]) >=  0.) continue;
498cddf8d76SBarry Smith #else
499cddf8d76SBarry Smith           if (a->a[j] >=  0.) continue;
500cddf8d76SBarry Smith #endif
501cddf8d76SBarry Smith           DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);
502cddf8d76SBarry Smith         }
503cddf8d76SBarry Smith       }
504cddf8d76SBarry Smith       color = DRAW_CYAN;
505cddf8d76SBarry Smith       for ( i=0; i<m; i++ ) {
506cddf8d76SBarry Smith         y_l = m - i - 1.0; y_r = y_l + 1.0;
507cddf8d76SBarry Smith         for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
508cddf8d76SBarry Smith           x_l = a->j[j] + shift; x_r = x_l + 1.0;
509cddf8d76SBarry Smith           if (a->a[j] !=  0.) continue;
510cddf8d76SBarry Smith           DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);
511cddf8d76SBarry Smith         }
512cddf8d76SBarry Smith       }
513cddf8d76SBarry Smith       color = DRAW_RED;
514cddf8d76SBarry Smith       for ( i=0; i<m; i++ ) {
515cddf8d76SBarry Smith         y_l = m - i - 1.0; y_r = y_l + 1.0;
516cddf8d76SBarry Smith         for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
517cddf8d76SBarry Smith           x_l = a->j[j] + shift; x_r = x_l + 1.0;
518cddf8d76SBarry Smith #if defined(PETSC_COMPLEX)
519cddf8d76SBarry Smith           if (real(a->a[j]) <=  0.) continue;
520cddf8d76SBarry Smith #else
521cddf8d76SBarry Smith           if (a->a[j] <=  0.) continue;
522cddf8d76SBarry Smith #endif
523cddf8d76SBarry Smith           DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);
524cddf8d76SBarry Smith         }
525cddf8d76SBarry Smith       }
5260513a670SBarry Smith     } else {
5270513a670SBarry Smith       /* use contour shading to indicate magnitude of values */
5280513a670SBarry Smith       int count = 0;
5290513a670SBarry Smith       for ( i=0; i<m; i++ ) {
5300513a670SBarry Smith         y_l = m - i - 1.0; y_r = y_l + 1.0;
5310513a670SBarry Smith         for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
5320513a670SBarry Smith           x_l = a->j[j] + shift; x_r = x_l + 1.0;
5330513a670SBarry Smith           color = 32 + (int) ((200.0 - 32.0)*PetscAbsScalar(a->a[count])/maxv);
5340513a670SBarry Smith           DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);
5350513a670SBarry Smith           count++;
5360513a670SBarry Smith         }
5370513a670SBarry Smith       }
5380513a670SBarry Smith     }
5390513a670SBarry Smith 
5406945ee14SBarry Smith     ierr = DrawCheckResizedWindow(draw);
541cddf8d76SBarry Smith     ierr = DrawGetMouseButton(draw,&button,&xc,&yc,0,0);
542cddf8d76SBarry Smith   }
543416022c9SBarry Smith   return 0;
544416022c9SBarry Smith }
545416022c9SBarry Smith 
5465615d1e5SSatish Balay #undef __FUNC__
547c22c1629SBarry Smith #define __FUNC__ "MatView_SeqAIJ" /* ADIC Ignore */
5488f6be9afSLois Curfman McInnes int MatView_SeqAIJ(PetscObject obj,Viewer viewer)
549416022c9SBarry Smith {
550416022c9SBarry Smith   Mat         A = (Mat) obj;
551416022c9SBarry Smith   Mat_SeqAIJ  *a = (Mat_SeqAIJ*) A->data;
552bcd2baecSBarry Smith   ViewerType  vtype;
553bcd2baecSBarry Smith   int         ierr;
554416022c9SBarry Smith 
555bcd2baecSBarry Smith   ierr = ViewerGetType(viewer,&vtype); CHKERRQ(ierr);
556bcd2baecSBarry Smith   if (vtype == MATLAB_VIEWER) {
557416022c9SBarry Smith     return ViewerMatlabPutSparse_Private(viewer,a->m,a->n,a->nz,a->a,a->i,a->j);
558416022c9SBarry Smith   }
559bcd2baecSBarry Smith   else if (vtype == ASCII_FILE_VIEWER || vtype == ASCII_FILES_VIEWER){
560416022c9SBarry Smith     return MatView_SeqAIJ_ASCII(A,viewer);
561416022c9SBarry Smith   }
562bcd2baecSBarry Smith   else if (vtype == BINARY_FILE_VIEWER) {
563416022c9SBarry Smith     return MatView_SeqAIJ_Binary(A,viewer);
564416022c9SBarry Smith   }
565bcd2baecSBarry Smith   else if (vtype == DRAW_VIEWER) {
566bcd2baecSBarry Smith     return MatView_SeqAIJ_Draw(A,viewer);
56717ab2063SBarry Smith   }
56817ab2063SBarry Smith   return 0;
56917ab2063SBarry Smith }
57019bcc07fSBarry Smith 
571c456f294SBarry Smith extern int Mat_AIJ_CheckInode(Mat);
5725615d1e5SSatish Balay #undef __FUNC__
5735615d1e5SSatish Balay #define __FUNC__ "MatAssemblyEnd_SeqAIJ"
5748f6be9afSLois Curfman McInnes int MatAssemblyEnd_SeqAIJ(Mat A,MatAssemblyType mode)
57517ab2063SBarry Smith {
576416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
57741c01911SSatish Balay   int        fshift = 0,i,j,*ai = a->i, *aj = a->j, *imax = a->imax,ierr;
57894a9d846SBarry Smith   int        m = a->m, *ip, N, *ailen = a->ilen,shift = a->indexshift,rmax;
579416022c9SBarry Smith   Scalar     *aa = a->a, *ap;
58017ab2063SBarry Smith 
5816d4a8577SBarry Smith   if (mode == MAT_FLUSH_ASSEMBLY) return 0;
58217ab2063SBarry Smith 
58394a9d846SBarry Smith   rmax = ailen[0]; /* determine row with most nonzeros */
58417ab2063SBarry Smith   for ( i=1; i<m; i++ ) {
585416022c9SBarry Smith     /* move each row back by the amount of empty slots (fshift) before it*/
58617ab2063SBarry Smith     fshift += imax[i-1] - ailen[i-1];
58794a9d846SBarry Smith     rmax   = PetscMax(rmax,ailen[i]);
58817ab2063SBarry Smith     if (fshift) {
589416022c9SBarry Smith       ip = aj + ai[i] + shift; ap = aa + ai[i] + shift;
59017ab2063SBarry Smith       N = ailen[i];
59117ab2063SBarry Smith       for ( j=0; j<N; j++ ) {
59217ab2063SBarry Smith         ip[j-fshift] = ip[j];
59317ab2063SBarry Smith         ap[j-fshift] = ap[j];
59417ab2063SBarry Smith       }
59517ab2063SBarry Smith     }
59617ab2063SBarry Smith     ai[i] = ai[i-1] + ailen[i-1];
59717ab2063SBarry Smith   }
59817ab2063SBarry Smith   if (m) {
59917ab2063SBarry Smith     fshift += imax[m-1] - ailen[m-1];
60017ab2063SBarry Smith     ai[m] = ai[m-1] + ailen[m-1];
60117ab2063SBarry Smith   }
60217ab2063SBarry Smith   /* reset ilen and imax for each row */
60317ab2063SBarry Smith   for ( i=0; i<m; i++ ) {
60417ab2063SBarry Smith     ailen[i] = imax[i] = ai[i+1] - ai[i];
60517ab2063SBarry Smith   }
606416022c9SBarry Smith   a->nz = ai[m] + shift;
60717ab2063SBarry Smith 
60817ab2063SBarry Smith   /* diagonals may have moved, so kill the diagonal pointers */
609416022c9SBarry Smith   if (fshift && a->diag) {
6100452661fSBarry Smith     PetscFree(a->diag);
611416022c9SBarry Smith     PLogObjectMemory(A,-(m+1)*sizeof(int));
612416022c9SBarry Smith     a->diag = 0;
61317ab2063SBarry Smith   }
6144e220ebcSLois Curfman McInnes   PLogInfo(A,"MatAssemblyEnd_SeqAIJ:Matrix size: %d X %d; storage space: %d unneeded, %d used\n",
6154e220ebcSLois Curfman McInnes            m,a->n,fshift,a->nz);
6164e220ebcSLois Curfman McInnes   PLogInfo(A,"MatAssemblyEnd_SeqAIJ:Number of mallocs during MatSetValues is %d\n",
617b810aeb4SBarry Smith            a->reallocs);
61894a9d846SBarry Smith   PLogInfo(A,"MatAssemblyEnd_SeqAIJ:Most nonzeros in any row is %d\n",rmax);
619dd5f02e7SSatish Balay   a->reallocs          = 0;
6204e220ebcSLois Curfman McInnes   A->info.nz_unneeded  = (double)fshift;
6214e220ebcSLois Curfman McInnes 
62276dd722bSSatish Balay   /* check out for identical nodes. If found, use inode functions */
62341c01911SSatish Balay   ierr = Mat_AIJ_CheckInode(A); CHKERRQ(ierr);
62417ab2063SBarry Smith   return 0;
62517ab2063SBarry Smith }
62617ab2063SBarry Smith 
6275615d1e5SSatish Balay #undef __FUNC__
6285615d1e5SSatish Balay #define __FUNC__ "MatZeroEntries_SeqAIJ"
6298f6be9afSLois Curfman McInnes int MatZeroEntries_SeqAIJ(Mat A)
63017ab2063SBarry Smith {
631416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
632cddf8d76SBarry Smith   PetscMemzero(a->a,(a->i[a->m]+a->indexshift)*sizeof(Scalar));
63317ab2063SBarry Smith   return 0;
63417ab2063SBarry Smith }
635416022c9SBarry Smith 
6365615d1e5SSatish Balay #undef __FUNC__
6375615d1e5SSatish Balay #define __FUNC__ "MatDestroy_SeqAIJ"
63817ab2063SBarry Smith int MatDestroy_SeqAIJ(PetscObject obj)
63917ab2063SBarry Smith {
640416022c9SBarry Smith   Mat        A  = (Mat) obj;
641416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
64290f02eecSBarry Smith   int        ierr;
643d5d45c9bSBarry Smith 
64417ab2063SBarry Smith #if defined(PETSC_LOG)
645416022c9SBarry Smith   PLogObjectState(obj,"Rows=%d, Cols=%d, NZ=%d",a->m,a->n,a->nz);
64617ab2063SBarry Smith #endif
6470452661fSBarry Smith   PetscFree(a->a);
6480452661fSBarry Smith   if (!a->singlemalloc) { PetscFree(a->i); PetscFree(a->j);}
6490452661fSBarry Smith   if (a->diag) PetscFree(a->diag);
6500452661fSBarry Smith   if (a->ilen) PetscFree(a->ilen);
6510452661fSBarry Smith   if (a->imax) PetscFree(a->imax);
6520452661fSBarry Smith   if (a->solve_work) PetscFree(a->solve_work);
65376dd722bSSatish Balay   if (a->inode.size) PetscFree(a->inode.size);
6540452661fSBarry Smith   PetscFree(a);
655*eed86810SBarry 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;
67996854ed6SLois Curfman McInnes   else if (op == MAT_NEW_NONZERO_ALLOCATION_ERROR) a->nonew       = -2;
6806d4a8577SBarry Smith   else if (op == MAT_YES_NEW_NONZERO_LOCATIONS)    a->nonew       = 0;
6816d4a8577SBarry Smith   else if (op == MAT_ROWS_SORTED ||
682219d9a1aSLois Curfman McInnes            op == MAT_ROWS_UNSORTED ||
6836d4a8577SBarry Smith            op == MAT_SYMMETRIC ||
6846d4a8577SBarry Smith            op == MAT_STRUCTURALLY_SYMMETRIC ||
68590f02eecSBarry Smith            op == MAT_YES_NEW_DIAGONALS ||
6862b362799SSatish Balay            op == MAT_IGNORE_OFF_PROC_ENTRIES)
68794a424c1SBarry Smith     PLogInfo(A,"Info:MatSetOption_SeqAIJ:Option ignored\n");
6886d4a8577SBarry Smith   else if (op == MAT_NO_NEW_DIAGONALS)
689e3372554SBarry Smith     {SETERRQ(PETSC_ERR_SUP,0,"MAT_NO_NEW_DIAGONALS");}
6906d4a8577SBarry Smith   else if (op == MAT_INODE_LIMIT_1)            a->inode.limit  = 1;
6916d4a8577SBarry Smith   else if (op == MAT_INODE_LIMIT_2)            a->inode.limit  = 2;
6926d4a8577SBarry Smith   else if (op == MAT_INODE_LIMIT_3)            a->inode.limit  = 3;
6936d4a8577SBarry Smith   else if (op == MAT_INODE_LIMIT_4)            a->inode.limit  = 4;
6946d4a8577SBarry Smith   else if (op == MAT_INODE_LIMIT_5)            a->inode.limit  = 5;
695e2f28af5SBarry Smith   else
696e3372554SBarry Smith     {SETERRQ(PETSC_ERR_SUP,0,"unknown option");}
69717ab2063SBarry Smith   return 0;
69817ab2063SBarry Smith }
69917ab2063SBarry Smith 
7005615d1e5SSatish Balay #undef __FUNC__
7015615d1e5SSatish Balay #define __FUNC__ "MatGetDiagonal_SeqAIJ"
7028f6be9afSLois Curfman McInnes int MatGetDiagonal_SeqAIJ(Mat A,Vec v)
70317ab2063SBarry Smith {
704416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
705416022c9SBarry Smith   int        i,j, n,shift = a->indexshift;
70617ab2063SBarry Smith   Scalar     *x, zero = 0.0;
70717ab2063SBarry Smith 
70817ab2063SBarry Smith   VecSet(&zero,v);
70990f02eecSBarry Smith   VecGetArray_Fast(v,x); VecGetLocalSize(v,&n);
710e3372554SBarry Smith   if (n != a->m) SETERRQ(1,0,"Nonconforming matrix and vector");
711416022c9SBarry Smith   for ( i=0; i<a->m; i++ ) {
712416022c9SBarry Smith     for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
713416022c9SBarry Smith       if (a->j[j]+shift == i) {
714416022c9SBarry Smith         x[i] = a->a[j];
71517ab2063SBarry Smith         break;
71617ab2063SBarry Smith       }
71717ab2063SBarry Smith     }
71817ab2063SBarry Smith   }
71917ab2063SBarry Smith   return 0;
72017ab2063SBarry Smith }
72117ab2063SBarry Smith 
72217ab2063SBarry Smith /* -------------------------------------------------------*/
72317ab2063SBarry Smith /* Should check that shapes of vectors and matrices match */
72417ab2063SBarry Smith /* -------------------------------------------------------*/
7255615d1e5SSatish Balay #undef __FUNC__
7265615d1e5SSatish Balay #define __FUNC__ "MatMultTrans_SeqAIJ"
72744cd7ae7SLois Curfman McInnes int MatMultTrans_SeqAIJ(Mat A,Vec xx,Vec yy)
72817ab2063SBarry Smith {
729416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
73017ab2063SBarry Smith   Scalar     *x, *y, *v, alpha;
731416022c9SBarry Smith   int        m = a->m, n, i, *idx, shift = a->indexshift;
73217ab2063SBarry Smith 
73390f02eecSBarry Smith   VecGetArray_Fast(xx,x); VecGetArray_Fast(yy,y);
734cddf8d76SBarry Smith   PetscMemzero(y,a->n*sizeof(Scalar));
73517ab2063SBarry Smith   y = y + shift; /* shift for Fortran start by 1 indexing */
73617ab2063SBarry Smith   for ( i=0; i<m; i++ ) {
737416022c9SBarry Smith     idx   = a->j + a->i[i] + shift;
738416022c9SBarry Smith     v     = a->a + a->i[i] + shift;
739416022c9SBarry Smith     n     = a->i[i+1] - a->i[i];
74017ab2063SBarry Smith     alpha = x[i];
74117ab2063SBarry Smith     while (n-->0) {y[*idx++] += alpha * *v++;}
74217ab2063SBarry Smith   }
743416022c9SBarry Smith   PLogFlops(2*a->nz - a->n);
74417ab2063SBarry Smith   return 0;
74517ab2063SBarry Smith }
746d5d45c9bSBarry Smith 
7475615d1e5SSatish Balay #undef __FUNC__
7485615d1e5SSatish Balay #define __FUNC__ "MatMultTransAdd_SeqAIJ"
74944cd7ae7SLois Curfman McInnes int MatMultTransAdd_SeqAIJ(Mat A,Vec xx,Vec zz,Vec yy)
75017ab2063SBarry Smith {
751416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
75217ab2063SBarry Smith   Scalar     *x, *y, *v, alpha;
753416022c9SBarry Smith   int        m = a->m, n, i, *idx,shift = a->indexshift;
75417ab2063SBarry Smith 
75590f02eecSBarry Smith   VecGetArray_Fast(xx,x); VecGetArray_Fast(yy,y);
75617ab2063SBarry Smith   if (zz != yy) VecCopy(zz,yy);
75717ab2063SBarry Smith   y = y + shift; /* shift for Fortran start by 1 indexing */
75817ab2063SBarry Smith   for ( i=0; i<m; i++ ) {
759416022c9SBarry Smith     idx   = a->j + a->i[i] + shift;
760416022c9SBarry Smith     v     = a->a + a->i[i] + shift;
761416022c9SBarry Smith     n     = a->i[i+1] - a->i[i];
76217ab2063SBarry Smith     alpha = x[i];
76317ab2063SBarry Smith     while (n-->0) {y[*idx++] += alpha * *v++;}
76417ab2063SBarry Smith   }
76590f02eecSBarry Smith   PLogFlops(2*a->nz);
76617ab2063SBarry Smith   return 0;
76717ab2063SBarry Smith }
76817ab2063SBarry Smith 
7695615d1e5SSatish Balay #undef __FUNC__
7705615d1e5SSatish Balay #define __FUNC__ "MatMult_SeqAIJ"
77144cd7ae7SLois Curfman McInnes int MatMult_SeqAIJ(Mat A,Vec xx,Vec yy)
77217ab2063SBarry Smith {
773416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
77417ab2063SBarry Smith   Scalar     *x, *y, *v, sum;
7759ea0dfa2SSatish Balay   int        m = a->m, n, i, *idx, shift = a->indexshift,*ii,jrow,j;
77617ab2063SBarry Smith 
77790f02eecSBarry Smith   VecGetArray_Fast(xx,x); VecGetArray_Fast(yy,y);
77817ab2063SBarry Smith   x    = x + shift; /* shift for Fortran start by 1 indexing */
779416022c9SBarry Smith   idx  = a->j;
780416022c9SBarry Smith   v    = a->a;
781416022c9SBarry Smith   ii   = a->i;
78217ab2063SBarry Smith   for ( i=0; i<m; i++ ) {
7839ea0dfa2SSatish Balay     jrow = ii[i];
7849ea0dfa2SSatish Balay     n    = ii[i+1] - jrow;
78517ab2063SBarry Smith     sum  = 0.0;
7869ea0dfa2SSatish Balay     /* while (n--) sum += *v++ * x[*idx++]; */
7879ea0dfa2SSatish Balay     for ( j=0; j<n; j++) {
7889ea0dfa2SSatish Balay       sum += v[jrow]*x[idx[jrow]]; jrow++;
7899ea0dfa2SSatish Balay      }
79017ab2063SBarry Smith     y[i] = sum;
79117ab2063SBarry Smith   }
792416022c9SBarry Smith   PLogFlops(2*a->nz - m);
79317ab2063SBarry Smith   return 0;
79417ab2063SBarry Smith }
79517ab2063SBarry Smith 
7965615d1e5SSatish Balay #undef __FUNC__
7975615d1e5SSatish Balay #define __FUNC__ "MatMultAdd_SeqAIJ"
79844cd7ae7SLois Curfman McInnes int MatMultAdd_SeqAIJ(Mat A,Vec xx,Vec yy,Vec zz)
79917ab2063SBarry Smith {
800416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
80117ab2063SBarry Smith   Scalar     *x, *y, *z, *v, sum;
8029ea0dfa2SSatish Balay   int        m = a->m, n, i, *idx, shift = a->indexshift,*ii,jrow,j;
8039ea0dfa2SSatish Balay 
80417ab2063SBarry Smith 
80590f02eecSBarry Smith   VecGetArray_Fast(xx,x); VecGetArray_Fast(yy,y); VecGetArray_Fast(zz,z);
80617ab2063SBarry Smith   x    = x + shift; /* shift for Fortran start by 1 indexing */
807cddf8d76SBarry Smith   idx  = a->j;
808cddf8d76SBarry Smith   v    = a->a;
809cddf8d76SBarry Smith   ii   = a->i;
81017ab2063SBarry Smith   for ( i=0; i<m; i++ ) {
8119ea0dfa2SSatish Balay     jrow = ii[i];
8129ea0dfa2SSatish Balay     n    = ii[i+1] - jrow;
81317ab2063SBarry Smith     sum  = y[i];
8149ea0dfa2SSatish Balay     /* while (n--) sum += *v++ * x[*idx++]; */
8159ea0dfa2SSatish Balay     for ( j=0; j<n; j++) {
8169ea0dfa2SSatish Balay       sum += v[jrow]*x[idx[jrow]]; jrow++;
8179ea0dfa2SSatish Balay      }
81817ab2063SBarry Smith     z[i] = sum;
81917ab2063SBarry Smith   }
820416022c9SBarry Smith   PLogFlops(2*a->nz);
82117ab2063SBarry Smith   return 0;
82217ab2063SBarry Smith }
82317ab2063SBarry Smith 
82417ab2063SBarry Smith /*
82517ab2063SBarry Smith      Adds diagonal pointers to sparse matrix structure.
82617ab2063SBarry Smith */
82717ab2063SBarry Smith 
8285615d1e5SSatish Balay #undef __FUNC__
8295615d1e5SSatish Balay #define __FUNC__ "MatMarkDiag_SeqAIJ"
830416022c9SBarry Smith int MatMarkDiag_SeqAIJ(Mat A)
83117ab2063SBarry Smith {
832416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
833416022c9SBarry Smith   int        i,j, *diag, m = a->m,shift = a->indexshift;
83417ab2063SBarry Smith 
8350452661fSBarry Smith   diag = (int *) PetscMalloc( (m+1)*sizeof(int)); CHKPTRQ(diag);
836416022c9SBarry Smith   PLogObjectMemory(A,(m+1)*sizeof(int));
837416022c9SBarry Smith   for ( i=0; i<a->m; i++ ) {
838416022c9SBarry Smith     for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) {
839416022c9SBarry Smith       if (a->j[j]+shift == i) {
84017ab2063SBarry Smith         diag[i] = j - shift;
84117ab2063SBarry Smith         break;
84217ab2063SBarry Smith       }
84317ab2063SBarry Smith     }
84417ab2063SBarry Smith   }
845416022c9SBarry Smith   a->diag = diag;
84617ab2063SBarry Smith   return 0;
84717ab2063SBarry Smith }
84817ab2063SBarry Smith 
8495615d1e5SSatish Balay #undef __FUNC__
8505615d1e5SSatish Balay #define __FUNC__ "MatRelax_SeqAIJ"
85144cd7ae7SLois Curfman McInnes int MatRelax_SeqAIJ(Mat A,Vec bb,double omega,MatSORType flag,
85217ab2063SBarry Smith                            double fshift,int its,Vec xx)
85317ab2063SBarry Smith {
854416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
855416022c9SBarry Smith   Scalar     *x, *b, *bs,  d, *xs, sum, *v = a->a,*t,scale,*ts, *xb;
856d5d45c9bSBarry Smith   int        ierr, *idx, *diag,n = a->n, m = a->m, i, shift = a->indexshift;
85717ab2063SBarry Smith 
85890f02eecSBarry Smith   VecGetArray_Fast(xx,x); VecGetArray_Fast(bb,b);
859416022c9SBarry Smith   if (!a->diag) {if ((ierr = MatMarkDiag_SeqAIJ(A))) return ierr;}
860416022c9SBarry Smith   diag = a->diag;
861416022c9SBarry Smith   xs   = x + shift; /* shifted by one for index start of a or a->j*/
86217ab2063SBarry Smith   if (flag == SOR_APPLY_UPPER) {
86317ab2063SBarry Smith    /* apply ( U + D/omega) to the vector */
86417ab2063SBarry Smith     bs = b + shift;
86517ab2063SBarry Smith     for ( i=0; i<m; i++ ) {
866416022c9SBarry Smith         d    = fshift + a->a[diag[i] + shift];
867416022c9SBarry Smith         n    = a->i[i+1] - diag[i] - 1;
868416022c9SBarry Smith         idx  = a->j + diag[i] + (!shift);
869416022c9SBarry Smith         v    = a->a + diag[i] + (!shift);
87017ab2063SBarry Smith         sum  = b[i]*d/omega;
87117ab2063SBarry Smith         SPARSEDENSEDOT(sum,bs,v,idx,n);
87217ab2063SBarry Smith         x[i] = sum;
87317ab2063SBarry Smith     }
87417ab2063SBarry Smith     return 0;
87517ab2063SBarry Smith   }
87617ab2063SBarry Smith   if (flag == SOR_APPLY_LOWER) {
877e3372554SBarry Smith     SETERRQ(1,0,"SOR_APPLY_LOWER is not done");
87817ab2063SBarry Smith   }
879416022c9SBarry Smith   else if (flag & SOR_EISENSTAT) {
88017ab2063SBarry Smith     /* Let  A = L + U + D; where L is lower trianglar,
88117ab2063SBarry Smith     U is upper triangular, E is diagonal; This routine applies
88217ab2063SBarry Smith 
88317ab2063SBarry Smith             (L + E)^{-1} A (U + E)^{-1}
88417ab2063SBarry Smith 
88517ab2063SBarry Smith     to a vector efficiently using Eisenstat's trick. This is for
88617ab2063SBarry Smith     the case of SSOR preconditioner, so E is D/omega where omega
88717ab2063SBarry Smith     is the relaxation factor.
88817ab2063SBarry Smith     */
8890452661fSBarry Smith     t = (Scalar *) PetscMalloc( m*sizeof(Scalar) ); CHKPTRQ(t);
89017ab2063SBarry Smith     scale = (2.0/omega) - 1.0;
89117ab2063SBarry Smith 
89217ab2063SBarry Smith     /*  x = (E + U)^{-1} b */
89317ab2063SBarry Smith     for ( i=m-1; i>=0; i-- ) {
894416022c9SBarry Smith       d    = fshift + a->a[diag[i] + shift];
895416022c9SBarry Smith       n    = a->i[i+1] - diag[i] - 1;
896416022c9SBarry Smith       idx  = a->j + diag[i] + (!shift);
897416022c9SBarry Smith       v    = a->a + diag[i] + (!shift);
89817ab2063SBarry Smith       sum  = b[i];
89917ab2063SBarry Smith       SPARSEDENSEMDOT(sum,xs,v,idx,n);
90017ab2063SBarry Smith       x[i] = omega*(sum/d);
90117ab2063SBarry Smith     }
90217ab2063SBarry Smith 
90317ab2063SBarry Smith     /*  t = b - (2*E - D)x */
904416022c9SBarry Smith     v = a->a;
90517ab2063SBarry Smith     for ( i=0; i<m; i++ ) { t[i] = b[i] - scale*(v[*diag++ + shift])*x[i]; }
90617ab2063SBarry Smith 
90717ab2063SBarry Smith     /*  t = (E + L)^{-1}t */
908416022c9SBarry Smith     ts = t + shift; /* shifted by one for index start of a or a->j*/
909416022c9SBarry Smith     diag = a->diag;
91017ab2063SBarry Smith     for ( i=0; i<m; i++ ) {
911416022c9SBarry Smith       d    = fshift + a->a[diag[i]+shift];
912416022c9SBarry Smith       n    = diag[i] - a->i[i];
913416022c9SBarry Smith       idx  = a->j + a->i[i] + shift;
914416022c9SBarry Smith       v    = a->a + a->i[i] + shift;
91517ab2063SBarry Smith       sum  = t[i];
91617ab2063SBarry Smith       SPARSEDENSEMDOT(sum,ts,v,idx,n);
91717ab2063SBarry Smith       t[i] = omega*(sum/d);
91817ab2063SBarry Smith     }
91917ab2063SBarry Smith 
92017ab2063SBarry Smith     /*  x = x + t */
92117ab2063SBarry Smith     for ( i=0; i<m; i++ ) { x[i] += t[i]; }
9220452661fSBarry Smith     PetscFree(t);
92317ab2063SBarry Smith     return 0;
92417ab2063SBarry Smith   }
92517ab2063SBarry Smith   if (flag & SOR_ZERO_INITIAL_GUESS) {
92617ab2063SBarry Smith     if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP){
92717ab2063SBarry Smith       for ( i=0; i<m; i++ ) {
928416022c9SBarry Smith         d    = fshift + a->a[diag[i]+shift];
929416022c9SBarry Smith         n    = diag[i] - a->i[i];
930416022c9SBarry Smith         idx  = a->j + a->i[i] + shift;
931416022c9SBarry Smith         v    = a->a + a->i[i] + shift;
93217ab2063SBarry Smith         sum  = b[i];
93317ab2063SBarry Smith         SPARSEDENSEMDOT(sum,xs,v,idx,n);
93417ab2063SBarry Smith         x[i] = omega*(sum/d);
93517ab2063SBarry Smith       }
93617ab2063SBarry Smith       xb = x;
93717ab2063SBarry Smith     }
93817ab2063SBarry Smith     else xb = b;
93917ab2063SBarry Smith     if ((flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP) &&
94017ab2063SBarry Smith         (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP)) {
94117ab2063SBarry Smith       for ( i=0; i<m; i++ ) {
942416022c9SBarry Smith         x[i] *= a->a[diag[i]+shift];
94317ab2063SBarry Smith       }
94417ab2063SBarry Smith     }
94517ab2063SBarry Smith     if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP){
94617ab2063SBarry Smith       for ( i=m-1; i>=0; i-- ) {
947416022c9SBarry Smith         d    = fshift + a->a[diag[i] + shift];
948416022c9SBarry Smith         n    = a->i[i+1] - diag[i] - 1;
949416022c9SBarry Smith         idx  = a->j + diag[i] + (!shift);
950416022c9SBarry Smith         v    = a->a + diag[i] + (!shift);
95117ab2063SBarry Smith         sum  = xb[i];
95217ab2063SBarry Smith         SPARSEDENSEMDOT(sum,xs,v,idx,n);
95317ab2063SBarry Smith         x[i] = omega*(sum/d);
95417ab2063SBarry Smith       }
95517ab2063SBarry Smith     }
95617ab2063SBarry Smith     its--;
95717ab2063SBarry Smith   }
95817ab2063SBarry Smith   while (its--) {
95917ab2063SBarry Smith     if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP){
96017ab2063SBarry Smith       for ( i=0; i<m; i++ ) {
961416022c9SBarry Smith         d    = fshift + a->a[diag[i]+shift];
962416022c9SBarry Smith         n    = a->i[i+1] - a->i[i];
963416022c9SBarry Smith         idx  = a->j + a->i[i] + shift;
964416022c9SBarry Smith         v    = a->a + a->i[i] + shift;
96517ab2063SBarry Smith         sum  = b[i];
96617ab2063SBarry Smith         SPARSEDENSEMDOT(sum,xs,v,idx,n);
9677e33a6baSBarry Smith         x[i] = (1. - omega)*x[i] + omega*(sum + a->a[diag[i]+shift]*x[i])/d;
96817ab2063SBarry Smith       }
96917ab2063SBarry Smith     }
97017ab2063SBarry Smith     if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP){
97117ab2063SBarry Smith       for ( i=m-1; i>=0; i-- ) {
972416022c9SBarry Smith         d    = fshift + a->a[diag[i] + shift];
973416022c9SBarry Smith         n    = a->i[i+1] - a->i[i];
974416022c9SBarry Smith         idx  = a->j + a->i[i] + shift;
975416022c9SBarry Smith         v    = a->a + a->i[i] + shift;
97617ab2063SBarry Smith         sum  = b[i];
97717ab2063SBarry Smith         SPARSEDENSEMDOT(sum,xs,v,idx,n);
9787e33a6baSBarry Smith         x[i] = (1. - omega)*x[i] + omega*(sum + a->a[diag[i]+shift]*x[i])/d;
97917ab2063SBarry Smith       }
98017ab2063SBarry Smith     }
98117ab2063SBarry Smith   }
98217ab2063SBarry Smith   return 0;
98317ab2063SBarry Smith }
98417ab2063SBarry Smith 
9855615d1e5SSatish Balay #undef __FUNC__
9865615d1e5SSatish Balay #define __FUNC__ "MatGetInfo_SeqAIJ"
9878f6be9afSLois Curfman McInnes int MatGetInfo_SeqAIJ(Mat A,MatInfoType flag,MatInfo *info)
98817ab2063SBarry Smith {
989416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
9904e220ebcSLois Curfman McInnes 
9914e220ebcSLois Curfman McInnes   info->rows_global    = (double)a->m;
9924e220ebcSLois Curfman McInnes   info->columns_global = (double)a->n;
9934e220ebcSLois Curfman McInnes   info->rows_local     = (double)a->m;
9944e220ebcSLois Curfman McInnes   info->columns_local  = (double)a->n;
9954e220ebcSLois Curfman McInnes   info->block_size     = 1.0;
9964e220ebcSLois Curfman McInnes   info->nz_allocated   = (double)a->maxnz;
9974e220ebcSLois Curfman McInnes   info->nz_used        = (double)a->nz;
9984e220ebcSLois Curfman McInnes   info->nz_unneeded    = (double)(a->maxnz - a->nz);
9994e220ebcSLois Curfman McInnes   /*  if (info->nz_unneeded != A->info.nz_unneeded)
10004e220ebcSLois Curfman McInnes     printf("space descrepancy: maxnz-nz = %d, nz_unneeded = %d\n",(int)info->nz_unneeded,(int)A->info.nz_unneeded); */
10014e220ebcSLois Curfman McInnes   info->assemblies     = (double)A->num_ass;
10024e220ebcSLois Curfman McInnes   info->mallocs        = (double)a->reallocs;
10034e220ebcSLois Curfman McInnes   info->memory         = A->mem;
10044e220ebcSLois Curfman McInnes   if (A->factor) {
10054e220ebcSLois Curfman McInnes     info->fill_ratio_given  = A->info.fill_ratio_given;
10064e220ebcSLois Curfman McInnes     info->fill_ratio_needed = A->info.fill_ratio_needed;
10074e220ebcSLois Curfman McInnes     info->factor_mallocs    = A->info.factor_mallocs;
10084e220ebcSLois Curfman McInnes   } else {
10094e220ebcSLois Curfman McInnes     info->fill_ratio_given  = 0;
10104e220ebcSLois Curfman McInnes     info->fill_ratio_needed = 0;
10114e220ebcSLois Curfman McInnes     info->factor_mallocs    = 0;
10124e220ebcSLois Curfman McInnes   }
101317ab2063SBarry Smith   return 0;
101417ab2063SBarry Smith }
101517ab2063SBarry Smith 
101617ab2063SBarry Smith extern int MatLUFactorSymbolic_SeqAIJ(Mat,IS,IS,double,Mat*);
101717ab2063SBarry Smith extern int MatLUFactorNumeric_SeqAIJ(Mat,Mat*);
101817ab2063SBarry Smith extern int MatLUFactor_SeqAIJ(Mat,IS,IS,double);
101917ab2063SBarry Smith extern int MatSolve_SeqAIJ(Mat,Vec,Vec);
102017ab2063SBarry Smith extern int MatSolveAdd_SeqAIJ(Mat,Vec,Vec,Vec);
102117ab2063SBarry Smith extern int MatSolveTrans_SeqAIJ(Mat,Vec,Vec);
102217ab2063SBarry Smith extern int MatSolveTransAdd_SeqAIJ(Mat,Vec,Vec,Vec);
102317ab2063SBarry Smith 
10245615d1e5SSatish Balay #undef __FUNC__
10255615d1e5SSatish Balay #define __FUNC__ "MatZeroRows_SeqAIJ"
10268f6be9afSLois Curfman McInnes int MatZeroRows_SeqAIJ(Mat A,IS is,Scalar *diag)
102717ab2063SBarry Smith {
1028416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
1029416022c9SBarry Smith   int         i,ierr,N, *rows,m = a->m - 1,shift = a->indexshift;
103017ab2063SBarry Smith 
103177c4ece6SBarry Smith   ierr = ISGetSize(is,&N); CHKERRQ(ierr);
103217ab2063SBarry Smith   ierr = ISGetIndices(is,&rows); CHKERRQ(ierr);
103317ab2063SBarry Smith   if (diag) {
103417ab2063SBarry Smith     for ( i=0; i<N; i++ ) {
1035e3372554SBarry Smith       if (rows[i] < 0 || rows[i] > m) SETERRQ(1,0,"row out of range");
1036416022c9SBarry Smith       if (a->ilen[rows[i]] > 0) { /* in case row was completely empty */
1037416022c9SBarry Smith         a->ilen[rows[i]] = 1;
1038416022c9SBarry Smith         a->a[a->i[rows[i]]+shift] = *diag;
1039416022c9SBarry Smith         a->j[a->i[rows[i]]+shift] = rows[i]+shift;
104017ab2063SBarry Smith       }
104117ab2063SBarry Smith       else {
104217ab2063SBarry Smith         ierr = MatSetValues_SeqAIJ(A,1,&rows[i],1,&rows[i],diag,INSERT_VALUES);
104317ab2063SBarry Smith         CHKERRQ(ierr);
104417ab2063SBarry Smith       }
104517ab2063SBarry Smith     }
104617ab2063SBarry Smith   }
104717ab2063SBarry Smith   else {
104817ab2063SBarry Smith     for ( i=0; i<N; i++ ) {
1049e3372554SBarry Smith       if (rows[i] < 0 || rows[i] > m) SETERRQ(1,0,"row out of range");
1050416022c9SBarry Smith       a->ilen[rows[i]] = 0;
105117ab2063SBarry Smith     }
105217ab2063SBarry Smith   }
105317ab2063SBarry Smith   ISRestoreIndices(is,&rows);
105443a90d84SBarry Smith   ierr = MatAssemblyEnd_SeqAIJ(A,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr);
105517ab2063SBarry Smith   return 0;
105617ab2063SBarry Smith }
105717ab2063SBarry Smith 
10585615d1e5SSatish Balay #undef __FUNC__
10595615d1e5SSatish Balay #define __FUNC__ "MatGetSize_SeqAIJ"
10608f6be9afSLois Curfman McInnes int MatGetSize_SeqAIJ(Mat A,int *m,int *n)
106117ab2063SBarry Smith {
1062416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
1063416022c9SBarry Smith   *m = a->m; *n = a->n;
106417ab2063SBarry Smith   return 0;
106517ab2063SBarry Smith }
106617ab2063SBarry Smith 
10675615d1e5SSatish Balay #undef __FUNC__
10685615d1e5SSatish Balay #define __FUNC__ "MatGetOwnershipRange_SeqAIJ"
10698f6be9afSLois Curfman McInnes int MatGetOwnershipRange_SeqAIJ(Mat A,int *m,int *n)
107017ab2063SBarry Smith {
1071416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
1072416022c9SBarry Smith   *m = 0; *n = a->m;
107317ab2063SBarry Smith   return 0;
107417ab2063SBarry Smith }
1075026e39d0SSatish Balay 
10765615d1e5SSatish Balay #undef __FUNC__
10775615d1e5SSatish Balay #define __FUNC__ "MatGetRow_SeqAIJ"
10784e093b46SBarry Smith int MatGetRow_SeqAIJ(Mat A,int row,int *nz,int **idx,Scalar **v)
107917ab2063SBarry Smith {
1080416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
1081c456f294SBarry Smith   int        *itmp,i,shift = a->indexshift;
108217ab2063SBarry Smith 
1083e3372554SBarry Smith   if (row < 0 || row >= a->m) SETERRQ(1,0,"Row out of range");
108417ab2063SBarry Smith 
1085416022c9SBarry Smith   *nz = a->i[row+1] - a->i[row];
1086416022c9SBarry Smith   if (v) *v = a->a + a->i[row] + shift;
108717ab2063SBarry Smith   if (idx) {
1088416022c9SBarry Smith     itmp = a->j + a->i[row] + shift;
10894e093b46SBarry Smith     if (*nz && shift) {
10900452661fSBarry Smith       *idx = (int *) PetscMalloc( (*nz)*sizeof(int) ); CHKPTRQ(*idx);
109117ab2063SBarry Smith       for ( i=0; i<(*nz); i++ ) {(*idx)[i] = itmp[i] + shift;}
10924e093b46SBarry Smith     } else if (*nz) {
10934e093b46SBarry Smith       *idx = itmp;
109417ab2063SBarry Smith     }
109517ab2063SBarry Smith     else *idx = 0;
109617ab2063SBarry Smith   }
109717ab2063SBarry Smith   return 0;
109817ab2063SBarry Smith }
109917ab2063SBarry Smith 
11005615d1e5SSatish Balay #undef __FUNC__
11015615d1e5SSatish Balay #define __FUNC__ "MatRestoreRow_SeqAIJ"
11024e093b46SBarry Smith int MatRestoreRow_SeqAIJ(Mat A,int row,int *nz,int **idx,Scalar **v)
110317ab2063SBarry Smith {
11044e093b46SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
11054e093b46SBarry Smith   if (idx) {if (*idx && a->indexshift) PetscFree(*idx);}
110617ab2063SBarry Smith   return 0;
110717ab2063SBarry Smith }
110817ab2063SBarry Smith 
11095615d1e5SSatish Balay #undef __FUNC__
11105615d1e5SSatish Balay #define __FUNC__ "MatNorm_SeqAIJ"
11118f6be9afSLois Curfman McInnes int MatNorm_SeqAIJ(Mat A,NormType type,double *norm)
111217ab2063SBarry Smith {
1113416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
1114416022c9SBarry Smith   Scalar     *v = a->a;
111517ab2063SBarry Smith   double     sum = 0.0;
1116416022c9SBarry Smith   int        i, j,shift = a->indexshift;
111717ab2063SBarry Smith 
111817ab2063SBarry Smith   if (type == NORM_FROBENIUS) {
1119416022c9SBarry Smith     for (i=0; i<a->nz; i++ ) {
112017ab2063SBarry Smith #if defined(PETSC_COMPLEX)
112117ab2063SBarry Smith       sum += real(conj(*v)*(*v)); v++;
112217ab2063SBarry Smith #else
112317ab2063SBarry Smith       sum += (*v)*(*v); v++;
112417ab2063SBarry Smith #endif
112517ab2063SBarry Smith     }
112617ab2063SBarry Smith     *norm = sqrt(sum);
112717ab2063SBarry Smith   }
112817ab2063SBarry Smith   else if (type == NORM_1) {
112917ab2063SBarry Smith     double *tmp;
1130416022c9SBarry Smith     int    *jj = a->j;
113166963ce1SSatish Balay     tmp = (double *) PetscMalloc( (a->n+1)*sizeof(double) ); CHKPTRQ(tmp);
1132cddf8d76SBarry Smith     PetscMemzero(tmp,a->n*sizeof(double));
113317ab2063SBarry Smith     *norm = 0.0;
1134416022c9SBarry Smith     for ( j=0; j<a->nz; j++ ) {
1135a2744918SBarry Smith         tmp[*jj++ + shift] += PetscAbsScalar(*v);  v++;
113617ab2063SBarry Smith     }
1137416022c9SBarry Smith     for ( j=0; j<a->n; j++ ) {
113817ab2063SBarry Smith       if (tmp[j] > *norm) *norm = tmp[j];
113917ab2063SBarry Smith     }
11400452661fSBarry Smith     PetscFree(tmp);
114117ab2063SBarry Smith   }
114217ab2063SBarry Smith   else if (type == NORM_INFINITY) {
114317ab2063SBarry Smith     *norm = 0.0;
1144416022c9SBarry Smith     for ( j=0; j<a->m; j++ ) {
1145416022c9SBarry Smith       v = a->a + a->i[j] + shift;
114617ab2063SBarry Smith       sum = 0.0;
1147416022c9SBarry Smith       for ( i=0; i<a->i[j+1]-a->i[j]; i++ ) {
1148cddf8d76SBarry Smith         sum += PetscAbsScalar(*v); v++;
114917ab2063SBarry Smith       }
115017ab2063SBarry Smith       if (sum > *norm) *norm = sum;
115117ab2063SBarry Smith     }
115217ab2063SBarry Smith   }
115317ab2063SBarry Smith   else {
1154e3372554SBarry Smith     SETERRQ(1,0,"No support for two norm yet");
115517ab2063SBarry Smith   }
115617ab2063SBarry Smith   return 0;
115717ab2063SBarry Smith }
115817ab2063SBarry Smith 
11595615d1e5SSatish Balay #undef __FUNC__
11605615d1e5SSatish Balay #define __FUNC__ "MatTranspose_SeqAIJ"
11618f6be9afSLois Curfman McInnes int MatTranspose_SeqAIJ(Mat A,Mat *B)
116217ab2063SBarry Smith {
1163416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
1164416022c9SBarry Smith   Mat        C;
1165416022c9SBarry Smith   int        i, ierr, *aj = a->j, *ai = a->i, m = a->m, len, *col;
1166416022c9SBarry Smith   int        shift = a->indexshift;
1167d5d45c9bSBarry Smith   Scalar     *array = a->a;
116817ab2063SBarry Smith 
11693638b69dSLois Curfman McInnes   if (B == PETSC_NULL && m != a->n)
1170e3372554SBarry Smith     SETERRQ(1,0,"Square matrix only for in-place");
11710452661fSBarry Smith   col = (int *) PetscMalloc((1+a->n)*sizeof(int)); CHKPTRQ(col);
1172cddf8d76SBarry Smith   PetscMemzero(col,(1+a->n)*sizeof(int));
117317ab2063SBarry Smith   if (shift) {
117417ab2063SBarry Smith     for ( i=0; i<ai[m]-1; i++ ) aj[i] -= 1;
117517ab2063SBarry Smith   }
117617ab2063SBarry Smith   for ( i=0; i<ai[m]+shift; i++ ) col[aj[i]] += 1;
1177416022c9SBarry Smith   ierr = MatCreateSeqAIJ(A->comm,a->n,m,0,col,&C); CHKERRQ(ierr);
11780452661fSBarry Smith   PetscFree(col);
117917ab2063SBarry Smith   for ( i=0; i<m; i++ ) {
118017ab2063SBarry Smith     len = ai[i+1]-ai[i];
1181416022c9SBarry Smith     ierr = MatSetValues(C,len,aj,1,&i,array,INSERT_VALUES); CHKERRQ(ierr);
118217ab2063SBarry Smith     array += len; aj += len;
118317ab2063SBarry Smith   }
118417ab2063SBarry Smith   if (shift) {
118517ab2063SBarry Smith     for ( i=0; i<ai[m]-1; i++ ) aj[i] += 1;
118617ab2063SBarry Smith   }
118717ab2063SBarry Smith 
11886d4a8577SBarry Smith   ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr);
11896d4a8577SBarry Smith   ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr);
119017ab2063SBarry Smith 
11913638b69dSLois Curfman McInnes   if (B != PETSC_NULL) {
1192416022c9SBarry Smith     *B = C;
119317ab2063SBarry Smith   } else {
1194416022c9SBarry Smith     /* This isn't really an in-place transpose */
11950452661fSBarry Smith     PetscFree(a->a);
11960452661fSBarry Smith     if (!a->singlemalloc) {PetscFree(a->i); PetscFree(a->j);}
11970452661fSBarry Smith     if (a->diag) PetscFree(a->diag);
11980452661fSBarry Smith     if (a->ilen) PetscFree(a->ilen);
11990452661fSBarry Smith     if (a->imax) PetscFree(a->imax);
12000452661fSBarry Smith     if (a->solve_work) PetscFree(a->solve_work);
12011073c447SSatish Balay     if (a->inode.size) PetscFree(a->inode.size);
12020452661fSBarry Smith     PetscFree(a);
1203f09e8eb9SSatish Balay     PetscMemcpy(A,C,sizeof(struct _p_Mat));
12040452661fSBarry Smith     PetscHeaderDestroy(C);
120517ab2063SBarry Smith   }
120617ab2063SBarry Smith   return 0;
120717ab2063SBarry Smith }
120817ab2063SBarry Smith 
12095615d1e5SSatish Balay #undef __FUNC__
12105615d1e5SSatish Balay #define __FUNC__ "MatDiagonalScale_SeqAIJ"
12118f6be9afSLois Curfman McInnes int MatDiagonalScale_SeqAIJ(Mat A,Vec ll,Vec rr)
121217ab2063SBarry Smith {
1213416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
121417ab2063SBarry Smith   Scalar     *l,*r,x,*v;
1215d5d45c9bSBarry Smith   int        i,j,m = a->m, n = a->n, M, nz = a->nz, *jj,shift = a->indexshift;
121617ab2063SBarry Smith 
121717ab2063SBarry Smith   if (ll) {
12183ea7c6a1SSatish Balay     /* The local size is used so that VecMPI can be passed to this routine
12193ea7c6a1SSatish Balay        by MatDiagonalScale_MPIAIJ */
12209b1297e1SSatish Balay     VecGetLocalSize_Fast(ll,m);
1221e3372554SBarry Smith     if (m != a->m) SETERRQ(1,0,"Left scaling vector wrong length");
122290f02eecSBarry Smith     VecGetArray_Fast(ll,l);
1223416022c9SBarry Smith     v = a->a;
122417ab2063SBarry Smith     for ( i=0; i<m; i++ ) {
122517ab2063SBarry Smith       x = l[i];
1226416022c9SBarry Smith       M = a->i[i+1] - a->i[i];
122717ab2063SBarry Smith       for ( j=0; j<M; j++ ) { (*v++) *= x;}
122817ab2063SBarry Smith     }
122944cd7ae7SLois Curfman McInnes     PLogFlops(nz);
123017ab2063SBarry Smith   }
123117ab2063SBarry Smith   if (rr) {
12329b1297e1SSatish Balay     VecGetLocalSize_Fast(rr,n);
1233e3372554SBarry Smith     if (n != a->n) SETERRQ(1,0,"Right scaling vector wrong length");
123490f02eecSBarry Smith     VecGetArray_Fast(rr,r);
1235416022c9SBarry Smith     v = a->a; jj = a->j;
123617ab2063SBarry Smith     for ( i=0; i<nz; i++ ) {
123717ab2063SBarry Smith       (*v++) *= r[*jj++ + shift];
123817ab2063SBarry Smith     }
123944cd7ae7SLois Curfman McInnes     PLogFlops(nz);
124017ab2063SBarry Smith   }
124117ab2063SBarry Smith   return 0;
124217ab2063SBarry Smith }
124317ab2063SBarry Smith 
12445615d1e5SSatish Balay #undef __FUNC__
12455615d1e5SSatish Balay #define __FUNC__ "MatGetSubMatrix_SeqAIJ"
12468f6be9afSLois Curfman McInnes int MatGetSubMatrix_SeqAIJ(Mat A,IS isrow,IS iscol,MatGetSubMatrixCall scall,Mat *B)
124717ab2063SBarry Smith {
1248db02288aSLois Curfman McInnes   Mat_SeqAIJ   *a = (Mat_SeqAIJ *) A->data,*c;
124902834360SBarry Smith   int          nznew, *smap, i, k, kstart, kend, ierr, oldcols = a->n,*lens;
125099141d43SSatish Balay   int          row,mat_i,*mat_j,tcol,first,step,*mat_ilen;
1251a2744918SBarry Smith   register int sum,lensi;
125299141d43SSatish Balay   int          *irow, *icol, nrows, ncols, shift = a->indexshift,*ssmap;
125399141d43SSatish Balay   int          *starts,*j_new,*i_new,*aj = a->j, *ai = a->i,ii,*ailen = a->ilen;
125499141d43SSatish Balay   Scalar       *a_new,*mat_a;
1255416022c9SBarry Smith   Mat          C;
125617ab2063SBarry Smith 
1257b48a1e75SSatish Balay   ierr = ISSorted(isrow,(PetscTruth*)&i);
1258e3372554SBarry Smith   if (!i) SETERRQ(1,0,"ISrow is not sorted");
125999141d43SSatish Balay   ierr = ISSorted(iscol,(PetscTruth*)&i);
1260e3372554SBarry Smith   if (!i) SETERRQ(1,0,"IScol is not sorted");
126199141d43SSatish Balay 
126217ab2063SBarry Smith   ierr = ISGetIndices(isrow,&irow); CHKERRQ(ierr);
126317ab2063SBarry Smith   ierr = ISGetSize(isrow,&nrows); CHKERRQ(ierr);
126417ab2063SBarry Smith   ierr = ISGetSize(iscol,&ncols); CHKERRQ(ierr);
126517ab2063SBarry Smith 
12667264ac53SSatish Balay   if (ISStrideGetInfo(iscol,&first,&step) && step == 1) { /* no need to sort */
126702834360SBarry Smith     /* special case of contiguous rows */
126857faeb66SBarry Smith     lens   = (int *) PetscMalloc((ncols+nrows+1)*sizeof(int)); CHKPTRQ(lens);
126902834360SBarry Smith     starts = lens + ncols;
127002834360SBarry Smith     /* loop over new rows determining lens and starting points */
127102834360SBarry Smith     for (i=0; i<nrows; i++) {
1272a2744918SBarry Smith       kstart  = ai[irow[i]]+shift;
1273a2744918SBarry Smith       kend    = kstart + ailen[irow[i]];
127402834360SBarry Smith       for ( k=kstart; k<kend; k++ ) {
1275d8ced48eSBarry Smith         if (aj[k]+shift >= first) {
127602834360SBarry Smith           starts[i] = k;
127702834360SBarry Smith           break;
127802834360SBarry Smith 	}
127902834360SBarry Smith       }
1280a2744918SBarry Smith       sum = 0;
128102834360SBarry Smith       while (k < kend) {
1282d8ced48eSBarry Smith         if (aj[k++]+shift >= first+ncols) break;
1283a2744918SBarry Smith         sum++;
128402834360SBarry Smith       }
1285a2744918SBarry Smith       lens[i] = sum;
128602834360SBarry Smith     }
128702834360SBarry Smith     /* create submatrix */
1288cddf8d76SBarry Smith     if (scall == MAT_REUSE_MATRIX) {
128908480c60SBarry Smith       int n_cols,n_rows;
129008480c60SBarry Smith       ierr = MatGetSize(*B,&n_rows,&n_cols); CHKERRQ(ierr);
1291e3372554SBarry Smith       if (n_rows != nrows || n_cols != ncols) SETERRQ(1,0,"");
1292d8ced48eSBarry Smith       ierr = MatZeroEntries(*B); CHKERRQ(ierr);
129308480c60SBarry Smith       C = *B;
129408480c60SBarry Smith     }
129508480c60SBarry Smith     else {
129602834360SBarry Smith       ierr = MatCreateSeqAIJ(A->comm,nrows,ncols,0,lens,&C);CHKERRQ(ierr);
129708480c60SBarry Smith     }
1298db02288aSLois Curfman McInnes     c = (Mat_SeqAIJ*) C->data;
1299db02288aSLois Curfman McInnes 
130002834360SBarry Smith     /* loop over rows inserting into submatrix */
1301db02288aSLois Curfman McInnes     a_new    = c->a;
1302db02288aSLois Curfman McInnes     j_new    = c->j;
1303db02288aSLois Curfman McInnes     i_new    = c->i;
1304db02288aSLois Curfman McInnes     i_new[0] = -shift;
130502834360SBarry Smith     for (i=0; i<nrows; i++) {
1306a2744918SBarry Smith       ii    = starts[i];
1307a2744918SBarry Smith       lensi = lens[i];
1308a2744918SBarry Smith       for ( k=0; k<lensi; k++ ) {
1309a2744918SBarry Smith         *j_new++ = aj[ii+k] - first;
131002834360SBarry Smith       }
1311a2744918SBarry Smith       PetscMemcpy(a_new,a->a + starts[i],lensi*sizeof(Scalar));
1312a2744918SBarry Smith       a_new      += lensi;
1313a2744918SBarry Smith       i_new[i+1]  = i_new[i] + lensi;
1314a2744918SBarry Smith       c->ilen[i]  = lensi;
131502834360SBarry Smith     }
13160452661fSBarry Smith     PetscFree(lens);
131702834360SBarry Smith   }
131802834360SBarry Smith   else {
131902834360SBarry Smith     ierr = ISGetIndices(iscol,&icol); CHKERRQ(ierr);
13200452661fSBarry Smith     smap  = (int *) PetscMalloc((1+oldcols)*sizeof(int)); CHKPTRQ(smap);
132102834360SBarry Smith     ssmap = smap + shift;
132299141d43SSatish Balay     lens  = (int *) PetscMalloc((1+nrows)*sizeof(int)); CHKPTRQ(lens);
1323cddf8d76SBarry Smith     PetscMemzero(smap,oldcols*sizeof(int));
132417ab2063SBarry Smith     for ( i=0; i<ncols; i++ ) smap[icol[i]] = i+1;
132502834360SBarry Smith     /* determine lens of each row */
132602834360SBarry Smith     for (i=0; i<nrows; i++) {
1327d8ced48eSBarry Smith       kstart  = ai[irow[i]]+shift;
132802834360SBarry Smith       kend    = kstart + a->ilen[irow[i]];
132902834360SBarry Smith       lens[i] = 0;
133002834360SBarry Smith       for ( k=kstart; k<kend; k++ ) {
1331d8ced48eSBarry Smith         if (ssmap[aj[k]]) {
133202834360SBarry Smith           lens[i]++;
133302834360SBarry Smith         }
133402834360SBarry Smith       }
133502834360SBarry Smith     }
133617ab2063SBarry Smith     /* Create and fill new matrix */
1337a2744918SBarry Smith     if (scall == MAT_REUSE_MATRIX) {
133899141d43SSatish Balay       c = (Mat_SeqAIJ *)((*B)->data);
133999141d43SSatish Balay 
1340e3372554SBarry Smith       if (c->m  != nrows || c->n != ncols) SETERRQ(1,0,"");
134199141d43SSatish Balay       if (PetscMemcmp(c->ilen,lens, c->m *sizeof(int))) {
1342e3372554SBarry Smith         SETERRQ(1,0,"Cannot reuse matrix. wrong no of nonzeros");
134399141d43SSatish Balay       }
134499141d43SSatish Balay       PetscMemzero(c->ilen,c->m*sizeof(int));
134508480c60SBarry Smith       C = *B;
134608480c60SBarry Smith     }
134708480c60SBarry Smith     else {
134802834360SBarry Smith       ierr = MatCreateSeqAIJ(A->comm,nrows,ncols,0,lens,&C);CHKERRQ(ierr);
134908480c60SBarry Smith     }
135099141d43SSatish Balay     c = (Mat_SeqAIJ *)(C->data);
135117ab2063SBarry Smith     for (i=0; i<nrows; i++) {
135299141d43SSatish Balay       row    = irow[i];
135317ab2063SBarry Smith       nznew  = 0;
135499141d43SSatish Balay       kstart = ai[row]+shift;
135599141d43SSatish Balay       kend   = kstart + a->ilen[row];
135699141d43SSatish Balay       mat_i  = c->i[i]+shift;
135799141d43SSatish Balay       mat_j  = c->j + mat_i;
135899141d43SSatish Balay       mat_a  = c->a + mat_i;
135999141d43SSatish Balay       mat_ilen = c->ilen + i;
136017ab2063SBarry Smith       for ( k=kstart; k<kend; k++ ) {
136199141d43SSatish Balay         if ((tcol=ssmap[a->j[k]])) {
136299141d43SSatish Balay           *mat_j++ = tcol - (!shift);
136399141d43SSatish Balay           *mat_a++ = a->a[k];
136499141d43SSatish Balay           (*mat_ilen)++;
136599141d43SSatish Balay 
136617ab2063SBarry Smith         }
136717ab2063SBarry Smith       }
136817ab2063SBarry Smith     }
136902834360SBarry Smith     /* Free work space */
137002834360SBarry Smith     ierr = ISRestoreIndices(iscol,&icol); CHKERRQ(ierr);
137199141d43SSatish Balay     PetscFree(smap); PetscFree(lens);
137202834360SBarry Smith   }
13736d4a8577SBarry Smith   ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr);
13746d4a8577SBarry Smith   ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr);
137517ab2063SBarry Smith 
137617ab2063SBarry Smith   ierr = ISRestoreIndices(isrow,&irow); CHKERRQ(ierr);
1377416022c9SBarry Smith   *B = C;
137817ab2063SBarry Smith   return 0;
137917ab2063SBarry Smith }
138017ab2063SBarry Smith 
1381a871dcd8SBarry Smith /*
138263b91edcSBarry Smith      note: This can only work for identity for row and col. It would
138363b91edcSBarry Smith    be good to check this and otherwise generate an error.
1384a871dcd8SBarry Smith */
13855615d1e5SSatish Balay #undef __FUNC__
13865615d1e5SSatish Balay #define __FUNC__ "MatILUFactor_SeqAIJ"
13878f6be9afSLois Curfman McInnes int MatILUFactor_SeqAIJ(Mat inA,IS row,IS col,double efill,int fill)
1388a871dcd8SBarry Smith {
138963b91edcSBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) inA->data;
139008480c60SBarry Smith   int        ierr;
139163b91edcSBarry Smith   Mat        outA;
139263b91edcSBarry Smith 
1393e3372554SBarry Smith   if (fill != 0) SETERRQ(1,0,"Only fill=0 supported");
1394a871dcd8SBarry Smith 
139563b91edcSBarry Smith   outA          = inA;
139663b91edcSBarry Smith   inA->factor   = FACTOR_LU;
139763b91edcSBarry Smith   a->row        = row;
139863b91edcSBarry Smith   a->col        = col;
139963b91edcSBarry Smith 
140094a9d846SBarry Smith   if (!a->solve_work) { /* this matrix may have been factored before */
14010452661fSBarry Smith     a->solve_work = (Scalar *) PetscMalloc( (a->m+1)*sizeof(Scalar)); CHKPTRQ(a->solve_work);
140294a9d846SBarry Smith   }
140363b91edcSBarry Smith 
140408480c60SBarry Smith   if (!a->diag) {
140508480c60SBarry Smith     ierr = MatMarkDiag_SeqAIJ(inA); CHKERRQ(ierr);
140663b91edcSBarry Smith   }
140763b91edcSBarry Smith   ierr = MatLUFactorNumeric_SeqAIJ(inA,&outA); CHKERRQ(ierr);
1408a871dcd8SBarry Smith   return 0;
1409a871dcd8SBarry Smith }
1410a871dcd8SBarry Smith 
1411f0b747eeSBarry Smith #include "pinclude/plapack.h"
14125615d1e5SSatish Balay #undef __FUNC__
14135615d1e5SSatish Balay #define __FUNC__ "MatScale_SeqAIJ"
14148f6be9afSLois Curfman McInnes int MatScale_SeqAIJ(Scalar *alpha,Mat inA)
1415f0b747eeSBarry Smith {
1416f0b747eeSBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) inA->data;
1417f0b747eeSBarry Smith   int        one = 1;
1418f0b747eeSBarry Smith   BLscal_( &a->nz, alpha, a->a, &one );
1419f0b747eeSBarry Smith   PLogFlops(a->nz);
1420f0b747eeSBarry Smith   return 0;
1421f0b747eeSBarry Smith }
1422f0b747eeSBarry Smith 
14235615d1e5SSatish Balay #undef __FUNC__
14245615d1e5SSatish Balay #define __FUNC__ "MatGetSubMatrices_SeqAIJ"
14258f6be9afSLois Curfman McInnes int MatGetSubMatrices_SeqAIJ(Mat A,int n, IS *irow,IS *icol,MatGetSubMatrixCall scall,
1426cddf8d76SBarry Smith                                     Mat **B)
1427cddf8d76SBarry Smith {
1428cddf8d76SBarry Smith   int ierr,i;
1429cddf8d76SBarry Smith 
1430cddf8d76SBarry Smith   if (scall == MAT_INITIAL_MATRIX) {
14310452661fSBarry Smith     *B = (Mat *) PetscMalloc( (n+1)*sizeof(Mat) ); CHKPTRQ(*B);
1432cddf8d76SBarry Smith   }
1433cddf8d76SBarry Smith 
1434cddf8d76SBarry Smith   for ( i=0; i<n; i++ ) {
1435905e6a2fSBarry Smith     ierr = MatGetSubMatrix_SeqAIJ(A,irow[i],icol[i],scall,&(*B)[i]);CHKERRQ(ierr);
1436cddf8d76SBarry Smith   }
1437cddf8d76SBarry Smith   return 0;
1438cddf8d76SBarry Smith }
1439cddf8d76SBarry Smith 
14405615d1e5SSatish Balay #undef __FUNC__
14415615d1e5SSatish Balay #define __FUNC__ "MatGetBlockSize_SeqAIJ"
14428f6be9afSLois Curfman McInnes int MatGetBlockSize_SeqAIJ(Mat A, int *bs)
14435a838052SSatish Balay {
14445a838052SSatish Balay   *bs = 1;
14455a838052SSatish Balay   return 0;
14465a838052SSatish Balay }
14475a838052SSatish Balay 
14485615d1e5SSatish Balay #undef __FUNC__
14495615d1e5SSatish Balay #define __FUNC__ "MatIncreaseOverlap_SeqAIJ"
14508f6be9afSLois Curfman McInnes int MatIncreaseOverlap_SeqAIJ(Mat A, int is_max, IS *is, int ov)
14514dcbc457SBarry Smith {
1452e4d965acSSatish Balay   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
145306763907SSatish Balay   int        shift, row, i,j,k,l,m,n, *idx,ierr, *nidx, isz, val;
14548a047759SSatish Balay   int        start, end, *ai, *aj;
145506763907SSatish Balay   char       *table;
14568a047759SSatish Balay   shift = a->indexshift;
1457e4d965acSSatish Balay   m     = a->m;
1458e4d965acSSatish Balay   ai    = a->i;
14598a047759SSatish Balay   aj    = a->j+shift;
14608a047759SSatish Balay 
1461e3372554SBarry Smith   if (ov < 0)  SETERRQ(1,0,"illegal overlap value used");
146206763907SSatish Balay 
146306763907SSatish Balay   table = (char *) PetscMalloc((m/BITSPERBYTE +1)*sizeof(char)); CHKPTRQ(table);
146406763907SSatish Balay   nidx  = (int *) PetscMalloc((m+1)*sizeof(int)); CHKPTRQ(nidx);
146506763907SSatish Balay 
1466e4d965acSSatish Balay   for ( i=0; i<is_max; i++ ) {
1467b97fc60eSLois Curfman McInnes     /* Initialize the two local arrays */
1468e4d965acSSatish Balay     isz  = 0;
146906763907SSatish Balay     PetscMemzero(table,(m/BITSPERBYTE +1)*sizeof(char));
1470e4d965acSSatish Balay 
1471e4d965acSSatish Balay     /* Extract the indices, assume there can be duplicate entries */
14724dcbc457SBarry Smith     ierr = ISGetIndices(is[i],&idx);  CHKERRQ(ierr);
147377c4ece6SBarry Smith     ierr = ISGetSize(is[i],&n);  CHKERRQ(ierr);
1474e4d965acSSatish Balay 
1475dd097bc3SLois Curfman McInnes     /* Enter these into the temp arrays. I.e., mark table[row], enter row into new index */
1476e4d965acSSatish Balay     for ( j=0; j<n ; ++j){
147706763907SSatish Balay       if(!BT_LOOKUP(table, idx[j])) { nidx[isz++] = idx[j];}
14784dcbc457SBarry Smith     }
147906763907SSatish Balay     ierr = ISRestoreIndices(is[i],&idx);  CHKERRQ(ierr);
148006763907SSatish Balay     ierr = ISDestroy(is[i]); CHKERRQ(ierr);
1481e4d965acSSatish Balay 
148204a348a9SBarry Smith     k = 0;
148304a348a9SBarry Smith     for ( j=0; j<ov; j++){ /* for each overlap */
148404a348a9SBarry Smith       n = isz;
148506763907SSatish Balay       for ( ; k<n ; k++){ /* do only those rows in nidx[k], which are not done yet */
1486e4d965acSSatish Balay         row   = nidx[k];
1487e4d965acSSatish Balay         start = ai[row];
1488e4d965acSSatish Balay         end   = ai[row+1];
148904a348a9SBarry Smith         for ( l = start; l<end ; l++){
14908a047759SSatish Balay           val = aj[l] + shift;
149106763907SSatish Balay           if (!BT_LOOKUP(table,val)) {nidx[isz++] = val;}
1492e4d965acSSatish Balay         }
1493e4d965acSSatish Balay       }
1494e4d965acSSatish Balay     }
1495029af93fSBarry Smith     ierr = ISCreateGeneral(PETSC_COMM_SELF, isz, nidx, (is+i)); CHKERRQ(ierr);
1496e4d965acSSatish Balay   }
149704a348a9SBarry Smith   PetscFree(table);
149806763907SSatish Balay   PetscFree(nidx);
1499e4d965acSSatish Balay   return 0;
15004dcbc457SBarry Smith }
150117ab2063SBarry Smith 
15020513a670SBarry Smith /* -------------------------------------------------------------- */
15030513a670SBarry Smith #undef __FUNC__
15040513a670SBarry Smith #define __FUNC__ "MatPermute_SeqAIJ"
15050513a670SBarry Smith int MatPermute_SeqAIJ(Mat A, IS rowp, IS colp, Mat *B)
15060513a670SBarry Smith {
15070513a670SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
15080513a670SBarry Smith   Scalar     *vwork;
15090513a670SBarry Smith   int        i, ierr, nz, m = a->m, n = a->n, *cwork;
15100513a670SBarry Smith   int        *row,*col,*cnew,j,*lens;
15110513a670SBarry Smith 
15120513a670SBarry Smith   ierr = ISGetIndices(rowp,&row); CHKERRQ(ierr);
15130513a670SBarry Smith   ierr = ISGetIndices(colp,&col); CHKERRQ(ierr);
15140513a670SBarry Smith 
15150513a670SBarry Smith   /* determine lengths of permuted rows */
15160513a670SBarry Smith   lens = (int *) PetscMalloc( (m+1)*sizeof(int) ); CHKPTRQ(lens);
15170513a670SBarry Smith   for (i=0; i<m; i++ ) {
15180513a670SBarry Smith     lens[row[i]] = a->i[i+1] - a->i[i];
15190513a670SBarry Smith   }
15200513a670SBarry Smith   ierr = MatCreateSeqAIJ(A->comm,m,n,0,lens,B);CHKERRQ(ierr);
15210513a670SBarry Smith   PetscFree(lens);
15220513a670SBarry Smith 
15230513a670SBarry Smith   cnew = (int *) PetscMalloc( n*sizeof(int) ); CHKPTRQ(cnew);
15240513a670SBarry Smith   for (i=0; i<m; i++) {
15250513a670SBarry Smith     ierr = MatGetRow(A,i,&nz,&cwork,&vwork); CHKERRQ(ierr);
15260513a670SBarry Smith     for (j=0; j<nz; j++ ) { cnew[j] = col[cwork[j]];}
15270513a670SBarry Smith     ierr = MatSetValues(*B,1,&row[i],nz,cnew,vwork,INSERT_VALUES); CHKERRQ(ierr);
15280513a670SBarry Smith     ierr = MatRestoreRow(A,i,&nz,&cwork,&vwork); CHKERRQ(ierr);
15290513a670SBarry Smith   }
15300513a670SBarry Smith   PetscFree(cnew);
15310513a670SBarry Smith   ierr = MatAssemblyBegin(*B,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr);
15320513a670SBarry Smith   ierr = MatAssemblyEnd(*B,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr);
15330513a670SBarry Smith   ierr = ISRestoreIndices(rowp,&row); CHKERRQ(ierr);
15340513a670SBarry Smith   ierr = ISRestoreIndices(colp,&col); CHKERRQ(ierr);
15350513a670SBarry Smith   return 0;
15360513a670SBarry Smith }
15370513a670SBarry Smith 
15385615d1e5SSatish Balay #undef __FUNC__
15395615d1e5SSatish Balay #define __FUNC__ "MatPrintHelp_SeqAIJ"
1540682d7d0cSBarry Smith int MatPrintHelp_SeqAIJ(Mat A)
1541682d7d0cSBarry Smith {
1542682d7d0cSBarry Smith   static int called = 0;
1543682d7d0cSBarry Smith   MPI_Comm   comm = A->comm;
1544682d7d0cSBarry Smith 
1545682d7d0cSBarry Smith   if (called) return 0; else called = 1;
154677c4ece6SBarry Smith   PetscPrintf(comm," Options for MATSEQAIJ and MATMPIAIJ matrix formats (the defaults):\n");
15470f665d81SLois Curfman McInnes   PetscPrintf(comm,"  -mat_lu_pivotthreshold <threshold>: Set pivoting threshold\n");
15480f665d81SLois Curfman McInnes   PetscPrintf(comm,"  -mat_aij_oneindex: internal indices begin at 1 instead of the default 0.\n");
15490f665d81SLois Curfman McInnes   PetscPrintf(comm,"  -mat_aij_no_inode: Do not use inodes\n");
15500f665d81SLois Curfman McInnes   PetscPrintf(comm,"  -mat_aij_inode_limit <limit>: Set inode limit (max limit=5)\n");
1551682d7d0cSBarry Smith #if defined(HAVE_ESSL)
15520f665d81SLois Curfman McInnes   PetscPrintf(comm,"  -mat_aij_essl: Use IBM sparse LU factorization and solve.\n");
1553682d7d0cSBarry Smith #endif
1554682d7d0cSBarry Smith   return 0;
1555682d7d0cSBarry Smith }
15568f6be9afSLois Curfman McInnes extern int MatEqual_SeqAIJ(Mat A,Mat B, PetscTruth* flg);
1557a93ec695SBarry Smith extern int MatFDColoringCreate_SeqAIJ(Mat,ISColoring,MatFDColoring);
1558a93ec695SBarry Smith extern int MatColoringPatch_SeqAIJ(Mat,int,int *,ISColoring *);
1559a93ec695SBarry Smith 
1560682d7d0cSBarry Smith /* -------------------------------------------------------------------*/
156117ab2063SBarry Smith static struct _MatOps MatOps = {MatSetValues_SeqAIJ,
156217ab2063SBarry Smith        MatGetRow_SeqAIJ,MatRestoreRow_SeqAIJ,
1563416022c9SBarry Smith        MatMult_SeqAIJ,MatMultAdd_SeqAIJ,
1564416022c9SBarry Smith        MatMultTrans_SeqAIJ,MatMultTransAdd_SeqAIJ,
156517ab2063SBarry Smith        MatSolve_SeqAIJ,MatSolveAdd_SeqAIJ,
156617ab2063SBarry Smith        MatSolveTrans_SeqAIJ,MatSolveTransAdd_SeqAIJ,
156717ab2063SBarry Smith        MatLUFactor_SeqAIJ,0,
156817ab2063SBarry Smith        MatRelax_SeqAIJ,
156917ab2063SBarry Smith        MatTranspose_SeqAIJ,
15707264ac53SSatish Balay        MatGetInfo_SeqAIJ,MatEqual_SeqAIJ,
1571f0b747eeSBarry Smith        MatGetDiagonal_SeqAIJ,MatDiagonalScale_SeqAIJ,MatNorm_SeqAIJ,
157217ab2063SBarry Smith        0,MatAssemblyEnd_SeqAIJ,
157317ab2063SBarry Smith        MatCompress_SeqAIJ,
157417ab2063SBarry Smith        MatSetOption_SeqAIJ,MatZeroEntries_SeqAIJ,MatZeroRows_SeqAIJ,
157517ab2063SBarry Smith        MatLUFactorSymbolic_SeqAIJ,MatLUFactorNumeric_SeqAIJ,0,0,
157617ab2063SBarry Smith        MatGetSize_SeqAIJ,MatGetSize_SeqAIJ,MatGetOwnershipRange_SeqAIJ,
157717ab2063SBarry Smith        MatILUFactorSymbolic_SeqAIJ,0,
157894a9d846SBarry Smith        0,0,
15793d1612f7SBarry Smith        MatConvertSameType_SeqAIJ,0,0,
1580cddf8d76SBarry Smith        MatILUFactor_SeqAIJ,0,0,
15817eb43aa7SLois Curfman McInnes        MatGetSubMatrices_SeqAIJ,MatIncreaseOverlap_SeqAIJ,
1582682d7d0cSBarry Smith        MatGetValues_SeqAIJ,0,
1583f0b747eeSBarry Smith        MatPrintHelp_SeqAIJ,
15845a838052SSatish Balay        MatScale_SeqAIJ,0,0,
15856945ee14SBarry Smith        MatILUDTFactor_SeqAIJ,
15866945ee14SBarry Smith        MatGetBlockSize_SeqAIJ,
15873b2fbd54SBarry Smith        MatGetRowIJ_SeqAIJ,
15883b2fbd54SBarry Smith        MatRestoreRowIJ_SeqAIJ,
15893b2fbd54SBarry Smith        MatGetColumnIJ_SeqAIJ,
1590a93ec695SBarry Smith        MatRestoreColumnIJ_SeqAIJ,
1591a93ec695SBarry Smith        MatFDColoringCreate_SeqAIJ,
15920513a670SBarry Smith        MatColoringPatch_SeqAIJ,
15930513a670SBarry Smith        0,
15940513a670SBarry Smith        MatPermute_SeqAIJ};
159517ab2063SBarry Smith 
159617ab2063SBarry Smith extern int MatUseSuperLU_SeqAIJ(Mat);
159717ab2063SBarry Smith extern int MatUseEssl_SeqAIJ(Mat);
159817ab2063SBarry Smith extern int MatUseDXML_SeqAIJ(Mat);
159917ab2063SBarry Smith 
16005615d1e5SSatish Balay #undef __FUNC__
16015615d1e5SSatish Balay #define __FUNC__ "MatCreateSeqAIJ"
160217ab2063SBarry Smith /*@C
1603682d7d0cSBarry Smith    MatCreateSeqAIJ - Creates a sparse matrix in AIJ (compressed row) format
16040d15e28bSLois Curfman McInnes    (the default parallel PETSc format).  For good matrix assembly performance
16056e62573dSLois Curfman McInnes    the user should preallocate the matrix storage by setting the parameter nz
16062bd5e0b2SLois Curfman McInnes    (or the array nzz).  By setting these parameters accurately, performance
16072bd5e0b2SLois Curfman McInnes    during matrix assembly can be increased by more than a factor of 50.
160817ab2063SBarry Smith 
160917ab2063SBarry Smith    Input Parameters:
1610029af93fSBarry Smith .  comm - MPI communicator, set to PETSC_COMM_SELF
161117ab2063SBarry Smith .  m - number of rows
161217ab2063SBarry Smith .  n - number of columns
161317ab2063SBarry Smith .  nz - number of nonzeros per row (same for all rows)
16142bd5e0b2SLois Curfman McInnes .  nzz - array containing the number of nonzeros in the various rows
16152bd5e0b2SLois Curfman McInnes          (possibly different for each row) or PETSC_NULL
161617ab2063SBarry Smith 
161717ab2063SBarry Smith    Output Parameter:
1618416022c9SBarry Smith .  A - the matrix
161917ab2063SBarry Smith 
162017ab2063SBarry Smith    Notes:
162117ab2063SBarry Smith    The AIJ format (also called the Yale sparse matrix format or
162217ab2063SBarry Smith    compressed row storage), is fully compatible with standard Fortran 77
16230002213bSLois Curfman McInnes    storage.  That is, the stored row and column indices can begin at
162444cd7ae7SLois Curfman McInnes    either one (as in Fortran) or zero.  See the users' manual for details.
162517ab2063SBarry Smith 
162617ab2063SBarry Smith    Specify the preallocated storage with either nz or nnz (not both).
1627a40aa06bSLois Curfman McInnes    Set nz=PETSC_DEFAULT and nnz=PETSC_NULL for PETSc to control dynamic memory
16283d323bbdSBarry Smith    allocation.  For large problems you MUST preallocate memory or you
16296da5968aSLois Curfman McInnes    will get TERRIBLE performance, see the users' manual chapter on matrices.
163017ab2063SBarry Smith 
1631682d7d0cSBarry Smith    By default, this format uses inodes (identical nodes) when possible, to
1632682d7d0cSBarry Smith    improve numerical efficiency of Matrix vector products and solves. We
1633682d7d0cSBarry Smith    search for consecutive rows with the same nonzero structure, thereby
16346c7ebb05SLois Curfman McInnes    reusing matrix information to achieve increased efficiency.
16356c7ebb05SLois Curfman McInnes 
16366c7ebb05SLois Curfman McInnes    Options Database Keys:
16376c7ebb05SLois Curfman McInnes $    -mat_aij_no_inode  - Do not use inodes
16386e62573dSLois Curfman McInnes $    -mat_aij_inode_limit <limit> - Set inode limit.
16396e62573dSLois Curfman McInnes $        (max limit=5)
16406e62573dSLois Curfman McInnes $    -mat_aij_oneindex - Internally use indexing starting at 1
16416e62573dSLois Curfman McInnes $        rather than 0.  Note: When calling MatSetValues(),
16426e62573dSLois Curfman McInnes $        the user still MUST index entries starting at 0!
164317ab2063SBarry Smith 
164417ab2063SBarry Smith .seealso: MatCreate(), MatCreateMPIAIJ(), MatSetValues()
164517ab2063SBarry Smith @*/
1646416022c9SBarry Smith int MatCreateSeqAIJ(MPI_Comm comm,int m,int n,int nz,int *nnz, Mat *A)
164717ab2063SBarry Smith {
1648416022c9SBarry Smith   Mat        B;
1649416022c9SBarry Smith   Mat_SeqAIJ *b;
16506945ee14SBarry Smith   int        i, len, ierr, flg,size;
16516945ee14SBarry Smith 
16526945ee14SBarry Smith   MPI_Comm_size(comm,&size);
1653e3372554SBarry Smith   if (size > 1) SETERRQ(1,0,"Comm must be of size 1");
1654d5d45c9bSBarry Smith 
1655416022c9SBarry Smith   *A                  = 0;
1656f09e8eb9SSatish Balay   PetscHeaderCreate(B,_p_Mat,MAT_COOKIE,MATSEQAIJ,comm);
1657416022c9SBarry Smith   PLogObjectCreate(B);
16580452661fSBarry Smith   B->data             = (void *) (b = PetscNew(Mat_SeqAIJ)); CHKPTRQ(b);
165944cd7ae7SLois Curfman McInnes   PetscMemzero(b,sizeof(Mat_SeqAIJ));
1660416022c9SBarry Smith   PetscMemcpy(&B->ops,&MatOps,sizeof(struct _MatOps));
1661416022c9SBarry Smith   B->destroy          = MatDestroy_SeqAIJ;
1662416022c9SBarry Smith   B->view             = MatView_SeqAIJ;
1663416022c9SBarry Smith   B->factor           = 0;
1664416022c9SBarry Smith   B->lupivotthreshold = 1.0;
166590f02eecSBarry Smith   B->mapping          = 0;
16667a743949SBarry Smith   ierr = OptionsGetDouble(PETSC_NULL,"-mat_lu_pivotthreshold",&B->lupivotthreshold,
166769957df2SSatish Balay                           &flg); CHKERRQ(ierr);
16687a743949SBarry Smith   b->ilu_preserve_row_sums = PETSC_FALSE;
16697a743949SBarry Smith   ierr = OptionsHasName(PETSC_NULL,"-pc_ilu_preserve_row_sums",
16707a743949SBarry Smith                         (int*) &b->ilu_preserve_row_sums); CHKERRQ(ierr);
1671416022c9SBarry Smith   b->row              = 0;
1672416022c9SBarry Smith   b->col              = 0;
1673416022c9SBarry Smith   b->indexshift       = 0;
1674b810aeb4SBarry Smith   b->reallocs         = 0;
167569957df2SSatish Balay   ierr = OptionsHasName(PETSC_NULL,"-mat_aij_oneindex", &flg); CHKERRQ(ierr);
167669957df2SSatish Balay   if (flg) b->indexshift = -1;
167717ab2063SBarry Smith 
167844cd7ae7SLois Curfman McInnes   b->m = m; B->m = m; B->M = m;
167944cd7ae7SLois Curfman McInnes   b->n = n; B->n = n; B->N = n;
16800452661fSBarry Smith   b->imax = (int *) PetscMalloc( (m+1)*sizeof(int) ); CHKPTRQ(b->imax);
1681b4fd4287SBarry Smith   if (nnz == PETSC_NULL) {
16827b8455f0SLois Curfman McInnes     if (nz == PETSC_DEFAULT) nz = 10;
16837b8455f0SLois Curfman McInnes     else if (nz <= 0)        nz = 1;
1684416022c9SBarry Smith     for ( i=0; i<m; i++ ) b->imax[i] = nz;
168517ab2063SBarry Smith     nz = nz*m;
168617ab2063SBarry Smith   }
168717ab2063SBarry Smith   else {
168817ab2063SBarry Smith     nz = 0;
1689416022c9SBarry Smith     for ( i=0; i<m; i++ ) {b->imax[i] = nnz[i]; nz += nnz[i];}
169017ab2063SBarry Smith   }
169117ab2063SBarry Smith 
169217ab2063SBarry Smith   /* allocate the matrix space */
1693416022c9SBarry Smith   len     = nz*(sizeof(int) + sizeof(Scalar)) + (b->m+1)*sizeof(int);
16940452661fSBarry Smith   b->a  = (Scalar *) PetscMalloc( len ); CHKPTRQ(b->a);
1695416022c9SBarry Smith   b->j  = (int *) (b->a + nz);
1696cddf8d76SBarry Smith   PetscMemzero(b->j,nz*sizeof(int));
1697416022c9SBarry Smith   b->i  = b->j + nz;
1698416022c9SBarry Smith   b->singlemalloc = 1;
169917ab2063SBarry Smith 
1700416022c9SBarry Smith   b->i[0] = -b->indexshift;
170117ab2063SBarry Smith   for (i=1; i<m+1; i++) {
1702416022c9SBarry Smith     b->i[i] = b->i[i-1] + b->imax[i-1];
170317ab2063SBarry Smith   }
170417ab2063SBarry Smith 
1705416022c9SBarry Smith   /* b->ilen will count nonzeros in each row so far. */
17060452661fSBarry Smith   b->ilen = (int *) PetscMalloc((m+1)*sizeof(int));
1707f09e8eb9SSatish Balay   PLogObjectMemory(B,len+2*(m+1)*sizeof(int)+sizeof(struct _p_Mat)+sizeof(Mat_SeqAIJ));
1708416022c9SBarry Smith   for ( i=0; i<b->m; i++ ) { b->ilen[i] = 0;}
170917ab2063SBarry Smith 
1710416022c9SBarry Smith   b->nz               = 0;
1711416022c9SBarry Smith   b->maxnz            = nz;
1712416022c9SBarry Smith   b->sorted           = 0;
1713416022c9SBarry Smith   b->roworiented      = 1;
1714416022c9SBarry Smith   b->nonew            = 0;
1715416022c9SBarry Smith   b->diag             = 0;
1716416022c9SBarry Smith   b->solve_work       = 0;
171771bd300dSLois Curfman McInnes   b->spptr            = 0;
1718754ec7b1SSatish Balay   b->inode.node_count = 0;
1719754ec7b1SSatish Balay   b->inode.size       = 0;
17206c7ebb05SLois Curfman McInnes   b->inode.limit      = 5;
17216c7ebb05SLois Curfman McInnes   b->inode.max_limit  = 5;
17224e220ebcSLois Curfman McInnes   B->info.nz_unneeded = (double)b->maxnz;
172317ab2063SBarry Smith 
1724416022c9SBarry Smith   *A = B;
17254e220ebcSLois Curfman McInnes 
17264b14c69eSBarry Smith   /*  SuperLU is not currently supported through PETSc */
17274b14c69eSBarry Smith #if defined(HAVE_SUPERLU)
172869957df2SSatish Balay   ierr = OptionsHasName(PETSC_NULL,"-mat_aij_superlu", &flg); CHKERRQ(ierr);
172969957df2SSatish Balay   if (flg) { ierr = MatUseSuperLU_SeqAIJ(B); CHKERRQ(ierr); }
17304b14c69eSBarry Smith #endif
173169957df2SSatish Balay   ierr = OptionsHasName(PETSC_NULL,"-mat_aij_essl", &flg); CHKERRQ(ierr);
173269957df2SSatish Balay   if (flg) { ierr = MatUseEssl_SeqAIJ(B); CHKERRQ(ierr); }
173369957df2SSatish Balay   ierr = OptionsHasName(PETSC_NULL,"-mat_aij_dxml", &flg); CHKERRQ(ierr);
173469957df2SSatish Balay   if (flg) {
1735e3372554SBarry Smith     if (!b->indexshift) SETERRQ(1,0,"need -mat_aij_oneindex with -mat_aij_dxml");
1736416022c9SBarry Smith     ierr = MatUseDXML_SeqAIJ(B); CHKERRQ(ierr);
173717ab2063SBarry Smith   }
173869957df2SSatish Balay   ierr = OptionsHasName(PETSC_NULL,"-help", &flg); CHKERRQ(ierr);
173969957df2SSatish Balay   if (flg) {ierr = MatPrintHelp(B); CHKERRQ(ierr); }
174017ab2063SBarry Smith   return 0;
174117ab2063SBarry Smith }
174217ab2063SBarry Smith 
17435615d1e5SSatish Balay #undef __FUNC__
17445615d1e5SSatish Balay #define __FUNC__ "MatConvertSameType_SeqAIJ"
17453d1612f7SBarry Smith int MatConvertSameType_SeqAIJ(Mat A,Mat *B,int cpvalues)
174617ab2063SBarry Smith {
1747416022c9SBarry Smith   Mat        C;
1748416022c9SBarry Smith   Mat_SeqAIJ *c,*a = (Mat_SeqAIJ *) A->data;
174908480c60SBarry Smith   int        i,len, m = a->m,shift = a->indexshift;
175017ab2063SBarry Smith 
17514043dd9cSLois Curfman McInnes   *B = 0;
1752f09e8eb9SSatish Balay   PetscHeaderCreate(C,_p_Mat,MAT_COOKIE,MATSEQAIJ,A->comm);
1753416022c9SBarry Smith   PLogObjectCreate(C);
17540452661fSBarry Smith   C->data       = (void *) (c = PetscNew(Mat_SeqAIJ)); CHKPTRQ(c);
175541c01911SSatish Balay   PetscMemcpy(&C->ops,&A->ops,sizeof(struct _MatOps));
1756416022c9SBarry Smith   C->destroy    = MatDestroy_SeqAIJ;
1757416022c9SBarry Smith   C->view       = MatView_SeqAIJ;
1758416022c9SBarry Smith   C->factor     = A->factor;
1759416022c9SBarry Smith   c->row        = 0;
1760416022c9SBarry Smith   c->col        = 0;
1761416022c9SBarry Smith   c->indexshift = shift;
1762c456f294SBarry Smith   C->assembled  = PETSC_TRUE;
176317ab2063SBarry Smith 
176444cd7ae7SLois Curfman McInnes   c->m = C->m   = a->m;
176544cd7ae7SLois Curfman McInnes   c->n = C->n   = a->n;
176644cd7ae7SLois Curfman McInnes   C->M          = a->m;
176744cd7ae7SLois Curfman McInnes   C->N          = a->n;
176817ab2063SBarry Smith 
17690452661fSBarry Smith   c->imax       = (int *) PetscMalloc((m+1)*sizeof(int)); CHKPTRQ(c->imax);
17700452661fSBarry Smith   c->ilen       = (int *) PetscMalloc((m+1)*sizeof(int)); CHKPTRQ(c->ilen);
177117ab2063SBarry Smith   for ( i=0; i<m; i++ ) {
1772416022c9SBarry Smith     c->imax[i] = a->imax[i];
1773416022c9SBarry Smith     c->ilen[i] = a->ilen[i];
177417ab2063SBarry Smith   }
177517ab2063SBarry Smith 
177617ab2063SBarry Smith   /* allocate the matrix space */
1777416022c9SBarry Smith   c->singlemalloc = 1;
1778416022c9SBarry Smith   len     = (m+1)*sizeof(int)+(a->i[m])*(sizeof(Scalar)+sizeof(int));
17790452661fSBarry Smith   c->a  = (Scalar *) PetscMalloc( len ); CHKPTRQ(c->a);
1780416022c9SBarry Smith   c->j  = (int *) (c->a + a->i[m] + shift);
1781416022c9SBarry Smith   c->i  = c->j + a->i[m] + shift;
1782416022c9SBarry Smith   PetscMemcpy(c->i,a->i,(m+1)*sizeof(int));
178317ab2063SBarry Smith   if (m > 0) {
1784416022c9SBarry Smith     PetscMemcpy(c->j,a->j,(a->i[m]+shift)*sizeof(int));
178508480c60SBarry Smith     if (cpvalues == COPY_VALUES) {
1786416022c9SBarry Smith       PetscMemcpy(c->a,a->a,(a->i[m]+shift)*sizeof(Scalar));
178717ab2063SBarry Smith     }
178808480c60SBarry Smith   }
178917ab2063SBarry Smith 
1790f09e8eb9SSatish Balay   PLogObjectMemory(C,len+2*(m+1)*sizeof(int)+sizeof(struct _p_Mat)+sizeof(Mat_SeqAIJ));
1791416022c9SBarry Smith   c->sorted      = a->sorted;
1792416022c9SBarry Smith   c->roworiented = a->roworiented;
1793416022c9SBarry Smith   c->nonew       = a->nonew;
17947a743949SBarry Smith   c->ilu_preserve_row_sums = a->ilu_preserve_row_sums;
179517ab2063SBarry Smith 
1796416022c9SBarry Smith   if (a->diag) {
17970452661fSBarry Smith     c->diag = (int *) PetscMalloc( (m+1)*sizeof(int) ); CHKPTRQ(c->diag);
1798416022c9SBarry Smith     PLogObjectMemory(C,(m+1)*sizeof(int));
179917ab2063SBarry Smith     for ( i=0; i<m; i++ ) {
1800416022c9SBarry Smith       c->diag[i] = a->diag[i];
180117ab2063SBarry Smith     }
180217ab2063SBarry Smith   }
1803416022c9SBarry Smith   else c->diag          = 0;
18046c7ebb05SLois Curfman McInnes   c->inode.limit        = a->inode.limit;
18056c7ebb05SLois Curfman McInnes   c->inode.max_limit    = a->inode.max_limit;
1806754ec7b1SSatish Balay   if (a->inode.size){
1807daed632aSSatish Balay     c->inode.size       = (int *) PetscMalloc( (m+1)*sizeof(int) ); CHKPTRQ(c->inode.size);
1808754ec7b1SSatish Balay     c->inode.node_count = a->inode.node_count;
1809daed632aSSatish Balay     PetscMemcpy( c->inode.size, a->inode.size, (m+1)*sizeof(int));
1810754ec7b1SSatish Balay   } else {
1811754ec7b1SSatish Balay     c->inode.size       = 0;
1812754ec7b1SSatish Balay     c->inode.node_count = 0;
1813754ec7b1SSatish Balay   }
1814416022c9SBarry Smith   c->nz                 = a->nz;
1815416022c9SBarry Smith   c->maxnz              = a->maxnz;
1816416022c9SBarry Smith   c->solve_work         = 0;
181776dd722bSSatish Balay   c->spptr              = 0;      /* Dangerous -I'm throwing away a->spptr */
1818754ec7b1SSatish Balay 
1819416022c9SBarry Smith   *B = C;
182017ab2063SBarry Smith   return 0;
182117ab2063SBarry Smith }
182217ab2063SBarry Smith 
18235615d1e5SSatish Balay #undef __FUNC__
18245615d1e5SSatish Balay #define __FUNC__ "MatLoad_SeqAIJ"
182519bcc07fSBarry Smith int MatLoad_SeqAIJ(Viewer viewer,MatType type,Mat *A)
182617ab2063SBarry Smith {
1827416022c9SBarry Smith   Mat_SeqAIJ   *a;
1828416022c9SBarry Smith   Mat          B;
182917699dbbSLois Curfman McInnes   int          i, nz, ierr, fd, header[4],size,*rowlengths = 0,M,N,shift;
1830bcd2baecSBarry Smith   MPI_Comm     comm;
183117ab2063SBarry Smith 
183219bcc07fSBarry Smith   PetscObjectGetComm((PetscObject) viewer,&comm);
183317699dbbSLois Curfman McInnes   MPI_Comm_size(comm,&size);
1834e3372554SBarry Smith   if (size > 1) SETERRQ(1,0,"view must have one processor");
183590ace30eSBarry Smith   ierr = ViewerBinaryGetDescriptor(viewer,&fd); CHKERRQ(ierr);
183677c4ece6SBarry Smith   ierr = PetscBinaryRead(fd,header,4,BINARY_INT); CHKERRQ(ierr);
1837e3372554SBarry Smith   if (header[0] != MAT_COOKIE) SETERRQ(1,0,"not matrix object in file");
183817ab2063SBarry Smith   M = header[1]; N = header[2]; nz = header[3];
183917ab2063SBarry Smith 
184017ab2063SBarry Smith   /* read in row lengths */
18410452661fSBarry Smith   rowlengths = (int*) PetscMalloc( M*sizeof(int) ); CHKPTRQ(rowlengths);
184277c4ece6SBarry Smith   ierr = PetscBinaryRead(fd,rowlengths,M,BINARY_INT); CHKERRQ(ierr);
184317ab2063SBarry Smith 
184417ab2063SBarry Smith   /* create our matrix */
1845416022c9SBarry Smith   ierr = MatCreateSeqAIJ(comm,M,N,0,rowlengths,A); CHKERRQ(ierr);
1846416022c9SBarry Smith   B = *A;
1847416022c9SBarry Smith   a = (Mat_SeqAIJ *) B->data;
1848416022c9SBarry Smith   shift = a->indexshift;
184917ab2063SBarry Smith 
185017ab2063SBarry Smith   /* read in column indices and adjust for Fortran indexing*/
185177c4ece6SBarry Smith   ierr = PetscBinaryRead(fd,a->j,nz,BINARY_INT); CHKERRQ(ierr);
185217ab2063SBarry Smith   if (shift) {
185317ab2063SBarry Smith     for ( i=0; i<nz; i++ ) {
1854416022c9SBarry Smith       a->j[i] += 1;
185517ab2063SBarry Smith     }
185617ab2063SBarry Smith   }
185717ab2063SBarry Smith 
185817ab2063SBarry Smith   /* read in nonzero values */
185977c4ece6SBarry Smith   ierr = PetscBinaryRead(fd,a->a,nz,BINARY_SCALAR); CHKERRQ(ierr);
186017ab2063SBarry Smith 
186117ab2063SBarry Smith   /* set matrix "i" values */
1862416022c9SBarry Smith   a->i[0] = -shift;
186317ab2063SBarry Smith   for ( i=1; i<= M; i++ ) {
1864416022c9SBarry Smith     a->i[i]      = a->i[i-1] + rowlengths[i-1];
1865416022c9SBarry Smith     a->ilen[i-1] = rowlengths[i-1];
186617ab2063SBarry Smith   }
18670452661fSBarry Smith   PetscFree(rowlengths);
186817ab2063SBarry Smith 
18696d4a8577SBarry Smith   ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr);
18706d4a8577SBarry Smith   ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr);
187117ab2063SBarry Smith   return 0;
187217ab2063SBarry Smith }
187317ab2063SBarry Smith 
18745615d1e5SSatish Balay #undef __FUNC__
18755615d1e5SSatish Balay #define __FUNC__ "MatEqual_SeqAIJ"
18768f6be9afSLois Curfman McInnes int MatEqual_SeqAIJ(Mat A,Mat B, PetscTruth* flg)
18777264ac53SSatish Balay {
18787264ac53SSatish Balay   Mat_SeqAIJ *a = (Mat_SeqAIJ *)A->data, *b = (Mat_SeqAIJ *)B->data;
18797264ac53SSatish Balay 
1880e3372554SBarry Smith   if (B->type !=MATSEQAIJ)SETERRQ(1,0,"Matrices must be same type");
18817264ac53SSatish Balay 
18827264ac53SSatish Balay   /* If the  matrix dimensions are not equal, or no of nonzeros or shift */
18837264ac53SSatish Balay   if ((a->m != b->m ) || (a->n !=b->n) ||( a->nz != b->nz)||
1884bcd2baecSBarry Smith       (a->indexshift != b->indexshift)) {
188577c4ece6SBarry Smith     *flg = PETSC_FALSE; return 0;
1886bcd2baecSBarry Smith   }
18877264ac53SSatish Balay 
18887264ac53SSatish Balay   /* if the a->i are the same */
1889bcd2baecSBarry Smith   if (PetscMemcmp(a->i,b->i, (a->n+1)*sizeof(int))) {
189077c4ece6SBarry Smith     *flg = PETSC_FALSE; return 0;
18917264ac53SSatish Balay   }
18927264ac53SSatish Balay 
18937264ac53SSatish Balay   /* if a->j are the same */
1894bcd2baecSBarry Smith   if (PetscMemcmp(a->j, b->j, (a->nz)*sizeof(int))) {
189577c4ece6SBarry Smith     *flg = PETSC_FALSE; return 0;
1896bcd2baecSBarry Smith   }
1897bcd2baecSBarry Smith 
1898bcd2baecSBarry Smith   /* if a->a are the same */
189919bcc07fSBarry Smith   if (PetscMemcmp(a->a, b->a, (a->nz)*sizeof(Scalar))) {
190077c4ece6SBarry Smith     *flg = PETSC_FALSE; return 0;
19017264ac53SSatish Balay   }
190277c4ece6SBarry Smith   *flg = PETSC_TRUE;
19037264ac53SSatish Balay   return 0;
19047264ac53SSatish Balay 
19057264ac53SSatish Balay }
1906