xref: /petsc/src/mat/impls/aij/seq/aijfact.c (revision 4da77479a31d4174cbdddbb52d31fbfbd1ffb3e6)
1*4da77479SBarry Smith /*$Id: aijfact.c,v 1.146 2000/03/14 17:29:40 bsmith Exp bsmith $*/
2289bc588SBarry Smith 
370f55243SBarry Smith #include "src/mat/impls/aij/seq/aij.h"
490f02eecSBarry Smith #include "src/vec/vecimpl.h"
51c4feecaSSatish Balay #include "src/inline/dot.h"
63b2fbd54SBarry Smith 
75615d1e5SSatish Balay #undef __FUNC__
891e9ee9fSBarry Smith #define __FUNC__ "MatOrdering_Flow_SeqAIJ"
991e9ee9fSBarry Smith int MatOrdering_Flow_SeqAIJ(Mat mat,MatOrderingType type,IS *irow,IS *icol)
103b2fbd54SBarry Smith {
113a40ed3dSBarry Smith   PetscFunctionBegin;
123a40ed3dSBarry Smith 
13e3372554SBarry Smith   SETERRQ(PETSC_ERR_SUP,0,"Code not written");
14aa482453SBarry Smith #if !defined(PETSC_USE_DEBUG)
15d64ed03dSBarry Smith   PetscFunctionReturn(0);
16d64ed03dSBarry Smith #endif
173b2fbd54SBarry Smith }
183b2fbd54SBarry Smith 
1986bacbd2SBarry Smith 
2086bacbd2SBarry Smith extern int MatMarkDiagonal_SeqAIJ(Mat);
2186bacbd2SBarry Smith extern int Mat_AIJ_CheckInode(Mat);
2286bacbd2SBarry Smith 
235bd2ddc7SBarry Smith extern int SPARSEKIT2dperm(int*,Scalar*,int*,int*,Scalar*,int*,int*,int*,int*,int*);
24329f5518SBarry Smith extern int SPARSEKIT2ilutp(int*,Scalar*,int*,int*,int*,PetscReal*,PetscReal*,int*,Scalar*,int*,int*,int*,Scalar*,int*,int*,int*);
255bd2ddc7SBarry Smith extern int SPARSEKIT2msrcsr(int*,Scalar*,int*,Scalar*,int*,int*,Scalar*,int*);
2686bacbd2SBarry Smith 
2786bacbd2SBarry Smith #undef __FUNC__
2886bacbd2SBarry Smith #define __FUNC__ "MatILUDTFactor_SeqAIJ"
2986bacbd2SBarry Smith   /* ------------------------------------------------------------
3086bacbd2SBarry Smith 
3186bacbd2SBarry Smith           This interface was contribed by Tony Caola
3286bacbd2SBarry Smith 
3386bacbd2SBarry Smith      This routine is an interface to the pivoting drop-tolerance
345bd2ddc7SBarry Smith      ILU routine written by Yousef Saad (saad@cs.umn.edu) as part of
355bd2ddc7SBarry Smith      SPARSEKIT2.
365bd2ddc7SBarry Smith 
375bd2ddc7SBarry Smith      The SPARSEKIT2 routines used here are covered by the GNU
385bd2ddc7SBarry Smith      copyright; see the file gnu in this directory.
3986bacbd2SBarry Smith 
4086bacbd2SBarry Smith      Thanks to Prof. Saad, Dr. Hysom, and Dr. Smith for their
4186bacbd2SBarry Smith      help in getting this routine ironed out.
4286bacbd2SBarry Smith 
435bd2ddc7SBarry Smith      The major drawback to this routine is that if info->fill is
445bd2ddc7SBarry Smith      not large enough it fails rather than allocating more space;
455bd2ddc7SBarry Smith      this can be fixed by hacking/improving the f2c version of
465bd2ddc7SBarry Smith      Yousef Saad's code.
4786bacbd2SBarry Smith 
485bd2ddc7SBarry Smith      ishift = 0, for indices start at 1
495bd2ddc7SBarry Smith      ishift = 1, for indices starting at 0
5086bacbd2SBarry Smith      ------------------------------------------------------------
5186bacbd2SBarry Smith   */
5286bacbd2SBarry Smith 
5386bacbd2SBarry Smith int MatILUDTFactor_SeqAIJ(Mat A,MatILUInfo *info,IS isrow,IS iscol,Mat *fact)
5486bacbd2SBarry Smith {
5586bacbd2SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b;
5607d2056aSBarry Smith   IS         iscolf,isicol,isirow;
5786bacbd2SBarry Smith   PetscTruth reorder;
585bd2ddc7SBarry Smith   int        *c,*r,*ic,ierr,i,n = a->m;
59329f5518SBarry Smith   int        *old_i = a->i,*old_j = a->j,*new_i,*old_i2 = 0,*old_j2 = 0,*new_j;
60313ae024SBarry Smith   int        *ordcol,*iwk,*iperm,*jw;
615bd2ddc7SBarry Smith   int        ishift = !a->indexshift;
62b19713cbSBarry Smith   int        jmax,lfill,job,*o_i,*o_j;
63329f5518SBarry Smith   Scalar     *old_a = a->a,*w,*new_a,*old_a2 = 0,*wk,*o_a;
64329f5518SBarry Smith   PetscReal  permtol,af;
6586bacbd2SBarry Smith 
6686bacbd2SBarry Smith   PetscFunctionBegin;
6786bacbd2SBarry Smith 
6886bacbd2SBarry Smith   if (info->dt == PETSC_DEFAULT)      info->dt      = .005;
6986bacbd2SBarry Smith   if (info->dtcount == PETSC_DEFAULT) info->dtcount = (int)(1.5*a->rmax);
7086bacbd2SBarry Smith   if (info->dtcol == PETSC_DEFAULT)   info->dtcol   = .01;
716faa4630SBarry Smith   if (info->fill == PETSC_DEFAULT)    info->fill    = ((double)(n*info->dtcount))/a->nz;
726faa4630SBarry Smith   lfill   = (int)(info->dtcount/2.0);
736faa4630SBarry Smith   jmax    = (int)(info->fill*a->nz);
7486bacbd2SBarry Smith   permtol = info->dtcol;
7586bacbd2SBarry Smith 
7686bacbd2SBarry Smith 
7786bacbd2SBarry Smith   /* ------------------------------------------------------------
7886bacbd2SBarry Smith      If reorder=.TRUE., then the original matrix has to be
7986bacbd2SBarry Smith      reordered to reflect the user selected ordering scheme, and
8086bacbd2SBarry Smith      then de-reordered so it is in it's original format.
8186bacbd2SBarry Smith      Because Saad's dperm() is NOT in place, we have to copy
8286bacbd2SBarry Smith      the original matrix and allocate more storage. . .
8386bacbd2SBarry Smith      ------------------------------------------------------------
8486bacbd2SBarry Smith   */
8586bacbd2SBarry Smith 
8686bacbd2SBarry Smith   /* set reorder to true if either isrow or iscol is not identity */
8786bacbd2SBarry Smith   ierr = ISIdentity(isrow,&reorder);CHKERRQ(ierr);
8886bacbd2SBarry Smith   if (reorder) {ierr = ISIdentity(iscol,&reorder);CHKERRQ(ierr);}
8986bacbd2SBarry Smith   reorder = PetscNot(reorder);
9086bacbd2SBarry Smith 
9186bacbd2SBarry Smith 
9286bacbd2SBarry Smith   /* storage for ilu factor */
9386bacbd2SBarry Smith   new_i = (int*)   PetscMalloc((n+1)*sizeof(int));CHKPTRQ(new_i);
9486bacbd2SBarry Smith   new_j = (int*)   PetscMalloc(jmax*sizeof(int));CHKPTRQ(new_j);
9586bacbd2SBarry Smith   new_a = (Scalar*)PetscMalloc(jmax*sizeof(Scalar));CHKPTRQ(new_a);
9686bacbd2SBarry Smith 
9786bacbd2SBarry Smith   ordcol = (int*)PetscMalloc(n*sizeof(int));CHKPTRQ(ordcol);
9886bacbd2SBarry Smith 
9986bacbd2SBarry Smith   /* ------------------------------------------------------------
10086bacbd2SBarry Smith      Make sure that everything is Fortran formatted (1-Based)
10186bacbd2SBarry Smith      ------------------------------------------------------------
10286bacbd2SBarry Smith   */
103b19713cbSBarry Smith   if (ishift) {
104b19713cbSBarry Smith     for (i=old_i[0];i<old_i[n];i++) {
105b19713cbSBarry Smith       old_j[i]++;
10686bacbd2SBarry Smith     }
107b19713cbSBarry Smith     for(i=0;i<n+1;i++) {
108b19713cbSBarry Smith       old_i[i]++;
109b19713cbSBarry Smith     };
11086bacbd2SBarry Smith   };
11186bacbd2SBarry Smith 
11286bacbd2SBarry Smith   if (reorder) {
113c0c2c81eSBarry Smith     ierr = ISGetIndices(iscol,&c);           CHKERRQ(ierr);
114c0c2c81eSBarry Smith     ierr = ISGetIndices(isrow,&r);           CHKERRQ(ierr);
115c0c2c81eSBarry Smith     for(i=0;i<n;i++) {
116c0c2c81eSBarry Smith       r[i]  = r[i]+1;
117c0c2c81eSBarry Smith       c[i]  = c[i]+1;
118c0c2c81eSBarry Smith     }
119313ae024SBarry Smith     old_i2 = (int*)PetscMalloc((n+1)*sizeof(int));CHKPTRQ(old_i2);
120313ae024SBarry Smith     old_j2 = (int*)PetscMalloc((old_i[n]-old_i[0]+1)*sizeof(int));CHKPTRQ(old_j2);
121313ae024SBarry Smith     old_a2 = (Scalar*)PetscMalloc((old_i[n]-old_i[0]+1)*sizeof(Scalar));CHKPTRQ(old_a2);
1225bd2ddc7SBarry Smith     job = 3; SPARSEKIT2dperm(&n,old_a,old_j,old_i,old_a2,old_j2,old_i2,r,c,&job);
123c0c2c81eSBarry Smith     for (i=0;i<n;i++) {
124c0c2c81eSBarry Smith       r[i]  = r[i]-1;
125c0c2c81eSBarry Smith       c[i]  = c[i]-1;
126c0c2c81eSBarry Smith     }
127c0c2c81eSBarry Smith     ierr = ISRestoreIndices(iscol,&c);CHKERRQ(ierr);
128c0c2c81eSBarry Smith     ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
129b19713cbSBarry Smith     o_a = old_a2;
130b19713cbSBarry Smith     o_j = old_j2;
131b19713cbSBarry Smith     o_i = old_i2;
132b19713cbSBarry Smith   } else {
133b19713cbSBarry Smith     o_a = old_a;
134b19713cbSBarry Smith     o_j = old_j;
135b19713cbSBarry Smith     o_i = old_i;
13686bacbd2SBarry Smith   }
13786bacbd2SBarry Smith 
13886bacbd2SBarry Smith   /* ------------------------------------------------------------
13986bacbd2SBarry Smith      Call Saad's ilutp() routine to generate the factorization
14086bacbd2SBarry Smith      ------------------------------------------------------------
14186bacbd2SBarry Smith   */
14286bacbd2SBarry Smith 
14386bacbd2SBarry Smith   iperm   = (int*)   PetscMalloc(2*n*sizeof(int));CHKPTRQ(iperm);
14486bacbd2SBarry Smith   jw      = (int*)   PetscMalloc(2*n*sizeof(int));CHKPTRQ(jw);
14586bacbd2SBarry Smith   w       = (Scalar*)PetscMalloc(n*sizeof(Scalar));CHKPTRQ(w);
14686bacbd2SBarry Smith 
1475bd2ddc7SBarry Smith   SPARSEKIT2ilutp(&n,o_a,o_j,o_i,&lfill,&info->dt,&permtol,&n,new_a,new_j,new_i,&jmax,w,jw,iperm,&ierr);
14886bacbd2SBarry Smith   if (ierr) {
14986bacbd2SBarry Smith     switch (ierr) {
1500511f09bSBarry Smith       case -3: SETERRQ2(1,1,"ilutp(), matrix U overflows, need larger info->fill current fill %g space allocated %d",info->fill,jmax);
1510511f09bSBarry Smith       case -2: SETERRQ2(1,1,"ilutp(), matrix L overflows, need larger info->fill current fill %g space allocated %d",info->fill,jmax);
15286bacbd2SBarry Smith       case -5: SETERRQ(1,1,"ilutp(), zero row encountered");
15386bacbd2SBarry Smith       case -1: SETERRQ(1,1,"ilutp(), input matrix may be wrong");
15486bacbd2SBarry Smith       case -4: SETERRQ1(1,1,"ilutp(), illegal info->fill value %d",jmax);
15586bacbd2SBarry Smith       default: SETERRQ1(1,1,"ilutp(), zero pivot detected on row %d",ierr);
15686bacbd2SBarry Smith     }
15786bacbd2SBarry Smith   }
15886bacbd2SBarry Smith 
15986bacbd2SBarry Smith   ierr = PetscFree(w);CHKERRQ(ierr);
16086bacbd2SBarry Smith   ierr = PetscFree(jw);CHKERRQ(ierr);
16186bacbd2SBarry Smith 
16286bacbd2SBarry Smith   /* ------------------------------------------------------------
16386bacbd2SBarry Smith      Saad's routine gives the result in Modified Sparse Row (msr)
16486bacbd2SBarry Smith      Convert to Compressed Sparse Row format (csr)
16586bacbd2SBarry Smith      ------------------------------------------------------------
16686bacbd2SBarry Smith   */
16786bacbd2SBarry Smith 
16886bacbd2SBarry Smith   wk  = (Scalar*)PetscMalloc(n*sizeof(Scalar));CHKPTRQ(wk);
16986bacbd2SBarry Smith   iwk = (int*)   PetscMalloc((n+1)*sizeof(int));CHKPTRQ(iwk);
17086bacbd2SBarry Smith 
1715bd2ddc7SBarry Smith   SPARSEKIT2msrcsr(&n,new_a,new_j,new_a,new_j,new_i,wk,iwk);
17286bacbd2SBarry Smith 
17386bacbd2SBarry Smith   ierr = PetscFree(iwk);CHKERRQ(ierr);
17486bacbd2SBarry Smith   ierr = PetscFree(wk);CHKERRQ(ierr);
17586bacbd2SBarry Smith 
17686bacbd2SBarry Smith   if (reorder) {
17786bacbd2SBarry Smith     ierr = PetscFree(old_a2);CHKERRQ(ierr);
17886bacbd2SBarry Smith     ierr = PetscFree(old_j2);CHKERRQ(ierr);
17986bacbd2SBarry Smith     ierr = PetscFree(old_i2);CHKERRQ(ierr);
180313ae024SBarry Smith   } else {
181b19713cbSBarry Smith     /* fix permutation of old_j that the factorization introduced */
182f170005cSBarry Smith     for (i=old_i[0]; i<old_i[n]; i++) {
183b19713cbSBarry Smith       old_j[i-1] = iperm[old_j[i-1]-1];
184b19713cbSBarry Smith     }
185b19713cbSBarry Smith   }
186b19713cbSBarry Smith 
187b801d0f9SBarry Smith   /* get rid of the shift to indices starting at 1 */
188b19713cbSBarry Smith   if (ishift) {
18986bacbd2SBarry Smith     for (i=0; i<n+1; i++) {
190b19713cbSBarry Smith       old_i[i]--;
191b19713cbSBarry Smith     }
192b19713cbSBarry Smith     for (i=old_i[0];i<old_i[n];i++) {
193b19713cbSBarry Smith       old_j[i]--;
194b19713cbSBarry Smith     }
19586bacbd2SBarry Smith   }
19686bacbd2SBarry Smith 
197b19713cbSBarry Smith   /* Make the factored matrix 0-based */
198b19713cbSBarry Smith   if (ishift) {
19986bacbd2SBarry Smith     for (i=0; i<n+1; i++) {
200b19713cbSBarry Smith       new_i[i]--;
201b19713cbSBarry Smith     }
202b19713cbSBarry Smith     for (i=new_i[0];i<new_i[n];i++) {
203b19713cbSBarry Smith       new_j[i]--;
204b19713cbSBarry Smith     }
20586bacbd2SBarry Smith   }
20686bacbd2SBarry Smith 
20786bacbd2SBarry Smith   /*-- due to the pivoting, we need to reorder iscol to correctly --*/
20886bacbd2SBarry Smith   /*-- permute the right-hand-side and solution vectors           --*/
2094c49b128SBarry Smith   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
2104c49b128SBarry Smith   ierr = ISInvertPermutation(isrow,PETSC_DECIDE,&isirow);CHKERRQ(ierr);
211c0c2c81eSBarry Smith   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
21286bacbd2SBarry Smith   for(i=0; i<n; i++) {
21386bacbd2SBarry Smith     ordcol[i] = ic[iperm[i]-1];
21486bacbd2SBarry Smith   };
21586bacbd2SBarry Smith   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
21607d2056aSBarry Smith   ierr = ISDestroy(isicol);CHKERRQ(ierr);
21786bacbd2SBarry Smith 
218c0c2c81eSBarry Smith   ierr = PetscFree(iperm);CHKERRQ(ierr);
219c0c2c81eSBarry Smith 
220329f5518SBarry Smith   ierr = ISCreateGeneral(PETSC_COMM_SELF,n,ordcol,&iscolf);CHKERRQ(ierr);
22107d2056aSBarry Smith   ierr = PetscFree(ordcol);CHKERRQ(ierr);
22286bacbd2SBarry Smith 
22386bacbd2SBarry Smith   /*----- put together the new matrix -----*/
22486bacbd2SBarry Smith 
22586bacbd2SBarry Smith   ierr = MatCreateSeqAIJ(A->comm,n,n,0,PETSC_NULL,fact);CHKERRQ(ierr);
22686bacbd2SBarry Smith   (*fact)->factor    = FACTOR_LU;
22786bacbd2SBarry Smith   (*fact)->assembled = PETSC_TRUE;
22886bacbd2SBarry Smith 
22986bacbd2SBarry Smith   b = (Mat_SeqAIJ*)(*fact)->data;
23086bacbd2SBarry Smith   ierr = PetscFree(b->imax);CHKERRQ(ierr);
23186bacbd2SBarry Smith   b->sorted        = PETSC_FALSE;
23207d2056aSBarry Smith   b->singlemalloc  = PETSC_FALSE;
233b19713cbSBarry Smith   /* the next line frees the default space generated by the MatCreate() */
23486bacbd2SBarry Smith   ierr             = PetscFree(b->a);CHKERRQ(ierr);
23586bacbd2SBarry Smith   ierr             = PetscFree(b->ilen);CHKERRQ(ierr);
23686bacbd2SBarry Smith   b->a             = new_a;
23786bacbd2SBarry Smith   b->j             = new_j;
23886bacbd2SBarry Smith   b->i             = new_i;
23986bacbd2SBarry Smith   b->ilen          = 0;
24086bacbd2SBarry Smith   b->imax          = 0;
2411f9e874aSBarry Smith   /*  I am not sure why these are the inverses of the row and column permutations; but the other way is NO GOOD */
242313ae024SBarry Smith   b->row           = isirow;
24386bacbd2SBarry Smith   b->col           = iscolf;
24486bacbd2SBarry Smith   b->solve_work    =  (Scalar*)PetscMalloc((n+1)*sizeof(Scalar));CHKPTRQ(b->solve_work);
24586bacbd2SBarry Smith   b->maxnz = b->nz = new_i[n];
24686bacbd2SBarry Smith   b->indexshift    = a->indexshift;
24786bacbd2SBarry Smith   ierr = MatMarkDiagonal_SeqAIJ(*fact);CHKERRQ(ierr);
24886bacbd2SBarry Smith   (*fact)->info.factor_mallocs = 0;
24986bacbd2SBarry Smith 
25086bacbd2SBarry Smith   ierr = MatMarkDiagonal_SeqAIJ(A);CHKERRQ(ierr);
25186bacbd2SBarry Smith 
25286bacbd2SBarry Smith   /* check out for identical nodes. If found, use inode functions */
25386bacbd2SBarry Smith   ierr = Mat_AIJ_CheckInode(*fact);CHKERRQ(ierr);
25486bacbd2SBarry Smith 
255431e710aSBarry Smith   af = ((double)b->nz)/((double)a->nz) + .001;
256431e710aSBarry Smith   PLogInfo(A,"MatILUDTFactor_SeqAIJ:Fill ratio:given %g needed %g\n",info->fill,af);
257431e710aSBarry Smith   PLogInfo(A,"MatILUDTFactor_SeqAIJ:Run with -pc_ilu_fill %g or use \n",af);
258431e710aSBarry Smith   PLogInfo(A,"MatILUDTFactor_SeqAIJ:PCILUSetFill(pc,%g);\n",af);
259431e710aSBarry Smith   PLogInfo(A,"MatILUDTFactor_SeqAIJ:for best performance.\n");
260431e710aSBarry Smith 
26186bacbd2SBarry Smith   PetscFunctionReturn(0);
26286bacbd2SBarry Smith }
26386bacbd2SBarry Smith 
264289bc588SBarry Smith /*
265289bc588SBarry Smith     Factorization code for AIJ format.
266289bc588SBarry Smith */
2675615d1e5SSatish Balay #undef __FUNC__
2685615d1e5SSatish Balay #define __FUNC__ "MatLUFactorSymbolic_SeqAIJ"
269329f5518SBarry Smith int MatLUFactorSymbolic_SeqAIJ(Mat A,IS isrow,IS iscol,PetscReal f,Mat *B)
270289bc588SBarry Smith {
271416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b;
272289bc588SBarry Smith   IS         isicol;
273416022c9SBarry Smith   int        *r,*ic,ierr,i,n = a->m,*ai = a->i,*aj = a->j;
274416022c9SBarry Smith   int        *ainew,*ajnew,jmax,*fill,*ajtmp,nz,shift = a->indexshift;
27591bf9adeSBarry Smith   int        *idnew,idx,row,m,fm,nnz,nzi,realloc = 0,nzbd,*im;
276289bc588SBarry Smith 
2773a40ed3dSBarry Smith   PetscFunctionBegin;
278d3cbd50cSLois Curfman McInnes   PetscValidHeaderSpecific(isrow,IS_COOKIE);
279d3cbd50cSLois Curfman McInnes   PetscValidHeaderSpecific(iscol,IS_COOKIE);
280f1af5d2fSBarry Smith   if (A->M != A->N) SETERRQ(PETSC_ERR_ARG_WRONG,0,"matrix must be square");
2813b2fbd54SBarry Smith 
2829863cb4fSBarry Smith   if (!isrow) {
2839863cb4fSBarry Smith     ierr = ISCreateStride(PETSC_COMM_SELF,A->M,0,1,&isrow);CHKERRQ(ierr);
2849863cb4fSBarry Smith   }
2859863cb4fSBarry Smith   if (!iscol) {
2869863cb4fSBarry Smith     ierr = ISCreateStride(PETSC_COMM_SELF,A->M,0,1,&iscol);CHKERRQ(ierr);
2879863cb4fSBarry Smith   }
2889863cb4fSBarry Smith 
2894c49b128SBarry Smith   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
290ac121b3dSBarry Smith   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
291ac121b3dSBarry Smith   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
292289bc588SBarry Smith 
293289bc588SBarry Smith   /* get new row pointers */
2940452661fSBarry Smith   ainew    = (int*)PetscMalloc((n+1)*sizeof(int));CHKPTRQ(ainew);
295dbb450caSBarry Smith   ainew[0] = -shift;
296289bc588SBarry Smith   /* don't know how many column pointers are needed so estimate */
297dbb450caSBarry Smith   jmax  = (int)(f*ai[n]+(!shift));
2980452661fSBarry Smith   ajnew = (int*)PetscMalloc((jmax)*sizeof(int));CHKPTRQ(ajnew);
299289bc588SBarry Smith   /* fill is a linked list of nonzeros in active row */
3000452661fSBarry Smith   fill = (int*)PetscMalloc((2*n+1)*sizeof(int));CHKPTRQ(fill);
3012fb3ed76SBarry Smith   im   = fill + n + 1;
302289bc588SBarry Smith   /* idnew is location of diagonal in factor */
3030452661fSBarry Smith   idnew    = (int*)PetscMalloc((n+1)*sizeof(int));CHKPTRQ(idnew);
304dbb450caSBarry Smith   idnew[0] = -shift;
305289bc588SBarry Smith 
306289bc588SBarry Smith   for (i=0; i<n; i++) {
307289bc588SBarry Smith     /* first copy previous fill into linked list */
308289bc588SBarry Smith     nnz     = nz    = ai[r[i]+1] - ai[r[i]];
3096b96ef82SSatish Balay     if (!nz) SETERRQ(PETSC_ERR_MAT_LU_ZRPVT,1,"Empty row in matrix");
310dbb450caSBarry Smith     ajtmp   = aj + ai[r[i]] + shift;
311289bc588SBarry Smith     fill[n] = n;
312289bc588SBarry Smith     while (nz--) {
313289bc588SBarry Smith       fm  = n;
314dbb450caSBarry Smith       idx = ic[*ajtmp++ + shift];
315289bc588SBarry Smith       do {
316289bc588SBarry Smith         m  = fm;
317289bc588SBarry Smith         fm = fill[m];
318289bc588SBarry Smith       } while (fm < idx);
319289bc588SBarry Smith       fill[m]   = idx;
320289bc588SBarry Smith       fill[idx] = fm;
321289bc588SBarry Smith     }
322289bc588SBarry Smith     row = fill[n];
323289bc588SBarry Smith     while (row < i) {
324dbb450caSBarry Smith       ajtmp = ajnew + idnew[row] + (!shift);
3252fb3ed76SBarry Smith       nzbd  = 1 + idnew[row] - ainew[row];
3262fb3ed76SBarry Smith       nz    = im[row] - nzbd;
327289bc588SBarry Smith       fm    = row;
3282fb3ed76SBarry Smith       while (nz-- > 0) {
329dbb450caSBarry Smith         idx = *ajtmp++ + shift;
3302fb3ed76SBarry Smith         nzbd++;
3312fb3ed76SBarry Smith         if (idx == i) im[row] = nzbd;
332289bc588SBarry Smith         do {
333289bc588SBarry Smith           m  = fm;
334289bc588SBarry Smith           fm = fill[m];
335289bc588SBarry Smith         } while (fm < idx);
336289bc588SBarry Smith         if (fm != idx) {
337289bc588SBarry Smith           fill[m]   = idx;
338289bc588SBarry Smith           fill[idx] = fm;
339289bc588SBarry Smith           fm        = idx;
340289bc588SBarry Smith           nnz++;
341289bc588SBarry Smith         }
342289bc588SBarry Smith       }
343289bc588SBarry Smith       row = fill[row];
344289bc588SBarry Smith     }
345289bc588SBarry Smith     /* copy new filled row into permanent storage */
346289bc588SBarry Smith     ainew[i+1] = ainew[i] + nnz;
347496e697eSBarry Smith     if (ainew[i+1] > jmax) {
348ecf371e4SBarry Smith 
349ecf371e4SBarry Smith       /* estimate how much additional space we will need */
350ecf371e4SBarry Smith       /* use the strategy suggested by David Hysom <hysom@perch-t.icase.edu> */
351ecf371e4SBarry Smith       /* just double the memory each time */
352ecf371e4SBarry Smith       int maxadd = jmax;
353ecf371e4SBarry Smith       /* maxadd = (int)((f*(ai[n]+(!shift))*(n-i+5))/n); */
354bbb6d6a8SBarry Smith       if (maxadd < nnz) maxadd = (n-i)*(nnz+1);
3552fb3ed76SBarry Smith       jmax += maxadd;
356ecf371e4SBarry Smith 
357ecf371e4SBarry Smith       /* allocate a longer ajnew */
3580452661fSBarry Smith       ajtmp = (int*)PetscMalloc(jmax*sizeof(int));CHKPTRQ(ajtmp);
359549d3d68SSatish Balay       ierr  = PetscMemcpy(ajtmp,ajnew,(ainew[i]+shift)*sizeof(int));CHKERRQ(ierr);
360606d414cSSatish Balay       ierr  = PetscFree(ajnew);CHKERRQ(ierr);
361289bc588SBarry Smith       ajnew = ajtmp;
3622fb3ed76SBarry Smith       realloc++; /* count how many times we realloc */
363289bc588SBarry Smith     }
364dbb450caSBarry Smith     ajtmp = ajnew + ainew[i] + shift;
365289bc588SBarry Smith     fm    = fill[n];
366289bc588SBarry Smith     nzi   = 0;
3672fb3ed76SBarry Smith     im[i] = nnz;
368289bc588SBarry Smith     while (nnz--) {
369289bc588SBarry Smith       if (fm < i) nzi++;
370dbb450caSBarry Smith       *ajtmp++ = fm - shift;
371289bc588SBarry Smith       fm       = fill[fm];
372289bc588SBarry Smith     }
373289bc588SBarry Smith     idnew[i] = ainew[i] + nzi;
374289bc588SBarry Smith   }
375464e8d44SSatish Balay   if (ai[n] != 0) {
376329f5518SBarry Smith     PetscReal af = ((PetscReal)ainew[n])/((PetscReal)ai[n]);
377c38d4ed2SBarry Smith     PLogInfo(A,"MatLUFactorSymbolic_SeqAIJ:Reallocs %d Fill ratio:given %g needed %g\n",realloc,f,af);
378981c4779SBarry Smith     PLogInfo(A,"MatLUFactorSymbolic_SeqAIJ:Run with -pc_lu_fill %g or use \n",af);
379981c4779SBarry Smith     PLogInfo(A,"MatLUFactorSymbolic_SeqAIJ:PCLUSetFill(pc,%g);\n",af);
380981c4779SBarry Smith     PLogInfo(A,"MatLUFactorSymbolic_SeqAIJ:for best performance.\n");
38105bf559cSSatish Balay   } else {
382981c4779SBarry Smith     PLogInfo(A,"MatLUFactorSymbolic_SeqAIJ: Empty matrix\n");
38305bf559cSSatish Balay   }
3842fb3ed76SBarry Smith 
385898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
386898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
3871987afe7SBarry Smith 
388606d414cSSatish Balay   ierr = PetscFree(fill);CHKERRQ(ierr);
389289bc588SBarry Smith 
390289bc588SBarry Smith   /* put together the new matrix */
391b4fd4287SBarry Smith   ierr = MatCreateSeqAIJ(A->comm,n,n,0,PETSC_NULL,B);CHKERRQ(ierr);
3921987afe7SBarry Smith   PLogObjectParent(*B,isicol);
393416022c9SBarry Smith   b = (Mat_SeqAIJ*)(*B)->data;
394606d414cSSatish Balay   ierr = PetscFree(b->imax);CHKERRQ(ierr);
3957c922b88SBarry Smith   b->singlemalloc = PETSC_FALSE;
396e8d4e0b9SBarry Smith   /* the next line frees the default space generated by the Create() */
397606d414cSSatish Balay   ierr = PetscFree(b->a);CHKERRQ(ierr);
398606d414cSSatish Balay   ierr = PetscFree(b->ilen);CHKERRQ(ierr);
39991bf9adeSBarry Smith   b->a          = (Scalar*)PetscMalloc((ainew[n]+shift+1)*sizeof(Scalar));CHKPTRQ(b->a);
400416022c9SBarry Smith   b->j          = ajnew;
401416022c9SBarry Smith   b->i          = ainew;
402416022c9SBarry Smith   b->diag       = idnew;
403416022c9SBarry Smith   b->ilen       = 0;
404416022c9SBarry Smith   b->imax       = 0;
405416022c9SBarry Smith   b->row        = isrow;
406416022c9SBarry Smith   b->col        = iscol;
407c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
408c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
40982bf6240SBarry Smith   b->icol       = isicol;
41091bf9adeSBarry Smith   b->solve_work = (Scalar*)PetscMalloc((n+1)*sizeof(Scalar));CHKPTRQ(b->solve_work);
411416022c9SBarry Smith   /* In b structure:  Free imax, ilen, old a, old j.
4127fd98bd6SLois Curfman McInnes      Allocate idnew, solve_work, new a, new j */
413416022c9SBarry Smith   PLogObjectMemory(*B,(ainew[n]+shift-n)*(sizeof(int)+sizeof(Scalar)));
414416022c9SBarry Smith   b->maxnz = b->nz = ainew[n] + shift;
4157fd98bd6SLois Curfman McInnes 
416329f5518SBarry Smith   (*B)->factor                 =  FACTOR_LU;
417ae068f58SLois Curfman McInnes   (*B)->info.factor_mallocs    = realloc;
418ae068f58SLois Curfman McInnes   (*B)->info.fill_ratio_given  = f;
4197dda7e10SBarry Smith   (*B)->ops->lufactornumeric   =  A->ops->lufactornumeric; /* Use Inode variant ONLY if A has inodes */
420703deb49SSatish Balay 
421e93ccc4dSBarry Smith   if (ai[n] != 0) {
422329f5518SBarry Smith     (*B)->info.fill_ratio_needed = ((PetscReal)ainew[n])/((PetscReal)ai[n]);
423eea2913aSSatish Balay   } else {
424eea2913aSSatish Balay     (*B)->info.fill_ratio_needed = 0.0;
425eea2913aSSatish Balay   }
4263a40ed3dSBarry Smith   PetscFunctionReturn(0);
427289bc588SBarry Smith }
4280a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
429184914b5SBarry Smith extern int Mat_AIJ_CheckInode(Mat);
43041c01911SSatish Balay 
4315615d1e5SSatish Balay #undef __FUNC__
4325615d1e5SSatish Balay #define __FUNC__ "MatLUFactorNumeric_SeqAIJ"
433416022c9SBarry Smith int MatLUFactorNumeric_SeqAIJ(Mat A,Mat *B)
434289bc588SBarry Smith {
435416022c9SBarry Smith   Mat        C = *B;
436416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b = (Mat_SeqAIJ *)C->data;
43782bf6240SBarry Smith   IS         isrow = b->row,isicol = b->icol;
438416022c9SBarry Smith   int        *r,*ic,ierr,i,j,n = a->m,*ai = b->i,*aj = b->j;
4391987afe7SBarry Smith   int        *ajtmpold,*ajtmp,nz,row,*ics,shift = a->indexshift;
440f2582111SSatish Balay   int        *diag_offset = b->diag,diag,k;
441329f5518SBarry Smith   int        preserve_row_sums = (int)a->ilu_preserve_row_sums,*pj;
442329f5518SBarry Smith   Scalar     *rtmp,*v,*pc,multiplier,sum,inner_sum,*rowsums = 0,*pv,*rtmps,*u_values;
443329f5518SBarry Smith   PetscReal  ssum;
444289bc588SBarry Smith 
4453a40ed3dSBarry Smith   PetscFunctionBegin;
44682bf6240SBarry Smith 
44778b31e54SBarry Smith   ierr  = ISGetIndices(isrow,&r);CHKERRQ(ierr);
44878b31e54SBarry Smith   ierr  = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
4490452661fSBarry Smith   rtmp  = (Scalar*)PetscMalloc((n+1)*sizeof(Scalar));CHKPTRQ(rtmp);
450549d3d68SSatish Balay   ierr  = PetscMemzero(rtmp,(n+1)*sizeof(Scalar));CHKERRQ(ierr);
4510cf60383SBarry Smith   rtmps = rtmp + shift; ics = ic + shift;
452289bc588SBarry Smith 
4536cf997b0SBarry Smith   /* precalculate row sums */
45435aab85fSBarry Smith   if (preserve_row_sums) {
45535aab85fSBarry Smith     rowsums = (Scalar*)PetscMalloc(n*sizeof(Scalar));CHKPTRQ(rowsums);
45635aab85fSBarry Smith     for (i=0; i<n; i++) {
45735aab85fSBarry Smith       nz  = a->i[r[i]+1] - a->i[r[i]];
45835aab85fSBarry Smith       v   = a->a + a->i[r[i]] + shift;
45935aab85fSBarry Smith       sum = 0.0;
46035aab85fSBarry Smith       for (j=0; j<nz; j++) sum += v[j];
46135aab85fSBarry Smith       rowsums[i] = sum;
46235aab85fSBarry Smith     }
46335aab85fSBarry Smith   }
46435aab85fSBarry Smith 
465289bc588SBarry Smith   for (i=0; i<n; i++) {
466289bc588SBarry Smith     nz    = ai[i+1] - ai[i];
467dbb450caSBarry Smith     ajtmp = aj + ai[i] + shift;
46848e94559SBarry Smith     for  (j=0; j<nz; j++) rtmps[ajtmp[j]] = 0.0;
469289bc588SBarry Smith 
470289bc588SBarry Smith     /* load in initial (unfactored row) */
471416022c9SBarry Smith     nz       = a->i[r[i]+1] - a->i[r[i]];
472416022c9SBarry Smith     ajtmpold = a->j + a->i[r[i]] + shift;
473416022c9SBarry Smith     v        = a->a + a->i[r[i]] + shift;
4740cf60383SBarry Smith     for (j=0; j<nz; j++) rtmp[ics[ajtmpold[j]]] =  v[j];
475289bc588SBarry Smith 
476dbb450caSBarry Smith     row = *ajtmp++ + shift;
477f2582111SSatish Balay     while  (row < i) {
4788c37ef55SBarry Smith       pc = rtmp + row;
4798c37ef55SBarry Smith       if (*pc != 0.0) {
4801987afe7SBarry Smith         pv         = b->a + diag_offset[row] + shift;
4811987afe7SBarry Smith         pj         = b->j + diag_offset[row] + (!shift);
48235aab85fSBarry Smith         multiplier = *pc / *pv++;
4838c37ef55SBarry Smith         *pc        = multiplier;
484f2582111SSatish Balay         nz         = ai[row+1] - diag_offset[row] - 1;
485f2582111SSatish Balay         for (j=0; j<nz; j++) rtmps[pj[j]] -= multiplier * pv[j];
486f2582111SSatish Balay         PLogFlops(2*nz);
487289bc588SBarry Smith       }
488dbb450caSBarry Smith       row = *ajtmp++ + shift;
489289bc588SBarry Smith     }
490416022c9SBarry Smith     /* finished row so stick it into b->a */
491416022c9SBarry Smith     pv = b->a + ai[i] + shift;
492416022c9SBarry Smith     pj = b->j + ai[i] + shift;
4938c37ef55SBarry Smith     nz = ai[i+1] - ai[i];
494416022c9SBarry Smith     for (j=0; j<nz; j++) {pv[j] = rtmps[pj[j]];}
49535aab85fSBarry Smith     diag = diag_offset[i] - ai[i];
49635aab85fSBarry Smith     /*
49735aab85fSBarry Smith           Possibly adjust diagonal entry on current row to force
49835aab85fSBarry Smith         LU matrix to have same row sum as initial matrix.
49935aab85fSBarry Smith     */
500d708749eSBarry Smith     if (pv[diag] == 0.0) {
5016cf997b0SBarry Smith       SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,0,"Zero pivot row %d",i);
502d708749eSBarry Smith     }
50335aab85fSBarry Smith     if (preserve_row_sums) {
50435aab85fSBarry Smith       pj  = b->j + ai[i] + shift;
50535aab85fSBarry Smith       sum = rowsums[i];
50635aab85fSBarry Smith       for (j=0; j<diag; j++) {
50735aab85fSBarry Smith         u_values  = b->a + diag_offset[pj[j]] + shift;
50835aab85fSBarry Smith         nz        = ai[pj[j]+1] - diag_offset[pj[j]];
50935aab85fSBarry Smith         inner_sum = 0.0;
51035aab85fSBarry Smith         for (k=0; k<nz; k++) {
51135aab85fSBarry Smith           inner_sum += u_values[k];
51235aab85fSBarry Smith         }
51335aab85fSBarry Smith         sum -= pv[j]*inner_sum;
51435aab85fSBarry Smith 
51535aab85fSBarry Smith       }
51635aab85fSBarry Smith       nz       = ai[i+1] - diag_offset[i] - 1;
51735aab85fSBarry Smith       u_values = b->a + diag_offset[i] + 1 + shift;
51835aab85fSBarry Smith       for (k=0; k<nz; k++) {
51935aab85fSBarry Smith         sum -= u_values[k];
52035aab85fSBarry Smith       }
52135aab85fSBarry Smith       ssum = PetscAbsScalar(sum/pv[diag]);
52235aab85fSBarry Smith       if (ssum < 1000. && ssum > .001) pv[diag] = sum;
52335aab85fSBarry Smith     }
52435aab85fSBarry Smith     /* check pivot entry for current row */
5258c37ef55SBarry Smith   }
5260f11f9aeSSatish Balay 
52735aab85fSBarry Smith   /* invert diagonal entries for simplier triangular solves */
52835aab85fSBarry Smith   for (i=0; i<n; i++) {
52935aab85fSBarry Smith     b->a[diag_offset[i]+shift] = 1.0/b->a[diag_offset[i]+shift];
53035aab85fSBarry Smith   }
53135aab85fSBarry Smith 
532606d414cSSatish Balay   if (preserve_row_sums) {ierr = PetscFree(rowsums);CHKERRQ(ierr);}
533606d414cSSatish Balay   ierr = PetscFree(rtmp);CHKERRQ(ierr);
53478b31e54SBarry Smith   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
53578b31e54SBarry Smith   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
536416022c9SBarry Smith   C->factor = FACTOR_LU;
53741c01911SSatish Balay   ierr = Mat_AIJ_CheckInode(C);CHKERRQ(ierr);
5387dda7e10SBarry Smith   (*B)->ops->lufactornumeric   =  A->ops->lufactornumeric; /* Use Inode variant ONLY if A has inodes */
539c456f294SBarry Smith   C->assembled = PETSC_TRUE;
540416022c9SBarry Smith   PLogFlops(b->n);
5413a40ed3dSBarry Smith   PetscFunctionReturn(0);
542289bc588SBarry Smith }
5430a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
5445615d1e5SSatish Balay #undef __FUNC__
5455615d1e5SSatish Balay #define __FUNC__ "MatLUFactor_SeqAIJ"
546329f5518SBarry Smith int MatLUFactor_SeqAIJ(Mat A,IS row,IS col,PetscReal f)
547da3a660dSBarry Smith {
548416022c9SBarry Smith   Mat_SeqAIJ     *mat = (Mat_SeqAIJ*)A->data;
54986bacbd2SBarry Smith   int            ierr,refct;
550416022c9SBarry Smith   Mat            C;
551f830108cSBarry Smith   PetscOps       *Abops;
5520a6ffc59SBarry Smith   MatOps         Aops;
553416022c9SBarry Smith 
5543a40ed3dSBarry Smith   PetscFunctionBegin;
555f2582111SSatish Balay   ierr = MatLUFactorSymbolic(A,row,col,f,&C);CHKERRQ(ierr);
556f2582111SSatish Balay   ierr = MatLUFactorNumeric(A,&C);CHKERRQ(ierr);
557da3a660dSBarry Smith 
558da3a660dSBarry Smith   /* free all the data structures from mat */
559606d414cSSatish Balay   ierr = PetscFree(mat->a);CHKERRQ(ierr);
560606d414cSSatish Balay   if (!mat->singlemalloc) {
561606d414cSSatish Balay     ierr = PetscFree(mat->i);CHKERRQ(ierr);
562606d414cSSatish Balay     ierr = PetscFree(mat->j);CHKERRQ(ierr);
563606d414cSSatish Balay   }
564606d414cSSatish Balay   if (mat->diag) {ierr = PetscFree(mat->diag);CHKERRQ(ierr);}
565606d414cSSatish Balay   if (mat->ilen) {ierr = PetscFree(mat->ilen);CHKERRQ(ierr);}
566606d414cSSatish Balay   if (mat->imax) {ierr = PetscFree(mat->imax);CHKERRQ(ierr);}
567606d414cSSatish Balay   if (mat->solve_work) {ierr = PetscFree(mat->solve_work);CHKERRQ(ierr);}
568606d414cSSatish Balay   if (mat->inode.size) {ierr = PetscFree(mat->inode.size);CHKERRQ(ierr);}
5690f0aacb9SBarry Smith   if (mat->icol) {ierr = ISDestroy(mat->icol);CHKERRQ(ierr);}
570606d414cSSatish Balay   ierr = PetscFree(mat);CHKERRQ(ierr);
571da3a660dSBarry Smith 
57217642b18SBarry Smith   ierr = MapDestroy(A->rmap);CHKERRQ(ierr);
57317642b18SBarry Smith   ierr = MapDestroy(A->cmap);CHKERRQ(ierr);
57417642b18SBarry Smith 
575f830108cSBarry Smith   /*
576f830108cSBarry Smith        This is horrible, horrible code. We need to keep the
577f830108cSBarry Smith     A pointers for the bops and ops but copy everything
578f830108cSBarry Smith     else from C.
579f830108cSBarry Smith   */
580f830108cSBarry Smith   Abops = A->bops;
581f830108cSBarry Smith   Aops  = A->ops;
58286bacbd2SBarry Smith   refct = A->refct;
583549d3d68SSatish Balay   ierr  = PetscMemcpy(A,C,sizeof(struct _p_Mat));CHKERRQ(ierr);
584451c4af8SSatish Balay   mat   = (Mat_SeqAIJ*)A->data;
585451c4af8SSatish Balay   PLogObjectParent(A,mat->icol);
586451c4af8SSatish Balay 
587f830108cSBarry Smith   A->bops  = Abops;
588f830108cSBarry Smith   A->ops   = Aops;
589bef8e0ddSBarry Smith   A->qlist = 0;
59086bacbd2SBarry Smith   A->refct = refct;
591c38d4ed2SBarry Smith   /* copy over the type_name and name */
592c38d4ed2SBarry Smith   ierr     = PetscStrallocpy(C->type_name,&A->type_name);CHKERRQ(ierr);
593c38d4ed2SBarry Smith   ierr     = PetscStrallocpy(C->name,&A->name);CHKERRQ(ierr);
594f830108cSBarry Smith 
5950452661fSBarry Smith   PetscHeaderDestroy(C);
596c38d4ed2SBarry Smith 
5973a40ed3dSBarry Smith   PetscFunctionReturn(0);
598da3a660dSBarry Smith }
5990a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
6005615d1e5SSatish Balay #undef __FUNC__
6015615d1e5SSatish Balay #define __FUNC__ "MatSolve_SeqAIJ"
602416022c9SBarry Smith int MatSolve_SeqAIJ(Mat A,Vec bb,Vec xx)
6038c37ef55SBarry Smith {
604416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data;
605416022c9SBarry Smith   IS         iscol = a->col,isrow = a->row;
606416022c9SBarry Smith   int        *r,*c,ierr,i, n = a->m,*vi,*ai = a->i,*aj = a->j;
6073b2fbd54SBarry Smith   int        nz,shift = a->indexshift,*rout,*cout;
608416022c9SBarry Smith   Scalar     *x,*b,*tmp,*tmps,*aa = a->a,sum,*v;
6098c37ef55SBarry Smith 
6103a40ed3dSBarry Smith   PetscFunctionBegin;
6113a40ed3dSBarry Smith   if (!n) PetscFunctionReturn(0);
61291bf9adeSBarry Smith 
613e1311b90SBarry Smith   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
614e1311b90SBarry Smith   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
615416022c9SBarry Smith   tmp  = a->solve_work;
6168c37ef55SBarry Smith 
6173b2fbd54SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
6183b2fbd54SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
6198c37ef55SBarry Smith 
6208c37ef55SBarry Smith   /* forward solve the lower triangular */
6218c37ef55SBarry Smith   tmp[0] = b[*r++];
6220cf60383SBarry Smith   tmps   = tmp + shift;
6238c37ef55SBarry Smith   for (i=1; i<n; i++) {
624dbb450caSBarry Smith     v   = aa + ai[i] + shift;
625dbb450caSBarry Smith     vi  = aj + ai[i] + shift;
626416022c9SBarry Smith     nz  = a->diag[i] - ai[i];
6278c37ef55SBarry Smith     sum = b[*r++];
6280cf60383SBarry Smith     while (nz--) sum -= *v++ * tmps[*vi++];
6298c37ef55SBarry Smith     tmp[i] = sum;
6308c37ef55SBarry Smith   }
6318c37ef55SBarry Smith 
6328c37ef55SBarry Smith   /* backward solve the upper triangular */
6338c37ef55SBarry Smith   for (i=n-1; i>=0; i--){
634416022c9SBarry Smith     v   = aa + a->diag[i] + (!shift);
635416022c9SBarry Smith     vi  = aj + a->diag[i] + (!shift);
636416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
6378c37ef55SBarry Smith     sum = tmp[i];
6380cf60383SBarry Smith     while (nz--) sum -= *v++ * tmps[*vi++];
639416022c9SBarry Smith     x[*c--] = tmp[i] = sum*aa[a->diag[i]+shift];
6408c37ef55SBarry Smith   }
6418c37ef55SBarry Smith 
6423b2fbd54SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
6433b2fbd54SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
644cb5b572fSBarry Smith   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
645cb5b572fSBarry Smith   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
646416022c9SBarry Smith   PLogFlops(2*a->nz - a->n);
6473a40ed3dSBarry Smith   PetscFunctionReturn(0);
6488c37ef55SBarry Smith }
649026e39d0SSatish Balay 
650930ae53cSBarry Smith /* ----------------------------------------------------------- */
651930ae53cSBarry Smith #undef __FUNC__
652930ae53cSBarry Smith #define __FUNC__ "MatSolve_SeqAIJ_NaturalOrdering"
653930ae53cSBarry Smith int MatSolve_SeqAIJ_NaturalOrdering(Mat A,Vec bb,Vec xx)
654930ae53cSBarry Smith {
655930ae53cSBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data;
656d85afc42SSatish Balay   int        n = a->m,*ai = a->i,*aj = a->j,*adiag = a->diag,ierr;
657d85afc42SSatish Balay   Scalar     *x,*b,*aa = a->a,sum;
658aa482453SBarry Smith #if !defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ)
659d85afc42SSatish Balay   int        adiag_i,i,*vi,nz,ai_i;
660d85afc42SSatish Balay   Scalar     *v;
661d85afc42SSatish Balay #endif
662930ae53cSBarry Smith 
6633a40ed3dSBarry Smith   PetscFunctionBegin;
6643a40ed3dSBarry Smith   if (!n) PetscFunctionReturn(0);
665930ae53cSBarry Smith   if (a->indexshift) {
6663a40ed3dSBarry Smith      ierr = MatSolve_SeqAIJ(A,bb,xx);CHKERRQ(ierr);
6673a40ed3dSBarry Smith      PetscFunctionReturn(0);
668930ae53cSBarry Smith   }
669930ae53cSBarry Smith 
670e1311b90SBarry Smith   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
671e1311b90SBarry Smith   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
672930ae53cSBarry Smith 
673aa482453SBarry Smith #if defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ)
6741c4feecaSSatish Balay   fortransolveaij_(&n,x,ai,aj,adiag,aa,b);
6751c4feecaSSatish Balay #else
676930ae53cSBarry Smith   /* forward solve the lower triangular */
677930ae53cSBarry Smith   x[0] = b[0];
678930ae53cSBarry Smith   for (i=1; i<n; i++) {
679930ae53cSBarry Smith     ai_i = ai[i];
680930ae53cSBarry Smith     v    = aa + ai_i;
681930ae53cSBarry Smith     vi   = aj + ai_i;
682930ae53cSBarry Smith     nz   = adiag[i] - ai_i;
683930ae53cSBarry Smith     sum  = b[i];
684930ae53cSBarry Smith     while (nz--) sum -= *v++ * x[*vi++];
685930ae53cSBarry Smith     x[i] = sum;
686930ae53cSBarry Smith   }
687930ae53cSBarry Smith 
688930ae53cSBarry Smith   /* backward solve the upper triangular */
689930ae53cSBarry Smith   for (i=n-1; i>=0; i--){
690930ae53cSBarry Smith     adiag_i = adiag[i];
691d708749eSBarry Smith     v       = aa + adiag_i + 1;
692d708749eSBarry Smith     vi      = aj + adiag_i + 1;
693930ae53cSBarry Smith     nz      = ai[i+1] - adiag_i - 1;
694930ae53cSBarry Smith     sum     = x[i];
695930ae53cSBarry Smith     while (nz--) sum -= *v++ * x[*vi++];
696930ae53cSBarry Smith     x[i]    = sum*aa[adiag_i];
697930ae53cSBarry Smith   }
6981c4feecaSSatish Balay #endif
699930ae53cSBarry Smith   PLogFlops(2*a->nz - a->n);
700cb5b572fSBarry Smith   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
701cb5b572fSBarry Smith   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
7023a40ed3dSBarry Smith   PetscFunctionReturn(0);
703930ae53cSBarry Smith }
704930ae53cSBarry Smith 
7055615d1e5SSatish Balay #undef __FUNC__
7065615d1e5SSatish Balay #define __FUNC__ "MatSolveAdd_SeqAIJ"
707416022c9SBarry Smith int MatSolveAdd_SeqAIJ(Mat A,Vec bb,Vec yy,Vec xx)
708da3a660dSBarry Smith {
709416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data;
710416022c9SBarry Smith   IS         iscol = a->col,isrow = a->row;
711416022c9SBarry Smith   int        *r,*c,ierr,i, n = a->m,*vi,*ai = a->i,*aj = a->j;
7123b2fbd54SBarry Smith   int        nz,shift = a->indexshift,*rout,*cout;
713416022c9SBarry Smith   Scalar     *x,*b,*tmp,*aa = a->a,sum,*v;
714da3a660dSBarry Smith 
7153a40ed3dSBarry Smith   PetscFunctionBegin;
71678b31e54SBarry Smith   if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);}
717da3a660dSBarry Smith 
718e1311b90SBarry Smith   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
719e1311b90SBarry Smith   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
720416022c9SBarry Smith   tmp  = a->solve_work;
721da3a660dSBarry Smith 
7223b2fbd54SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
7233b2fbd54SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
724da3a660dSBarry Smith 
725da3a660dSBarry Smith   /* forward solve the lower triangular */
726da3a660dSBarry Smith   tmp[0] = b[*r++];
727da3a660dSBarry Smith   for (i=1; i<n; i++) {
728dbb450caSBarry Smith     v   = aa + ai[i] + shift;
729dbb450caSBarry Smith     vi  = aj + ai[i] + shift;
730416022c9SBarry Smith     nz  = a->diag[i] - ai[i];
731da3a660dSBarry Smith     sum = b[*r++];
732dbb450caSBarry Smith     while (nz--) sum -= *v++ * tmp[*vi++ + shift];
733da3a660dSBarry Smith     tmp[i] = sum;
734da3a660dSBarry Smith   }
735da3a660dSBarry Smith 
736da3a660dSBarry Smith   /* backward solve the upper triangular */
737da3a660dSBarry Smith   for (i=n-1; i>=0; i--){
738416022c9SBarry Smith     v   = aa + a->diag[i] + (!shift);
739416022c9SBarry Smith     vi  = aj + a->diag[i] + (!shift);
740416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
741da3a660dSBarry Smith     sum = tmp[i];
742dbb450caSBarry Smith     while (nz--) sum -= *v++ * tmp[*vi++ + shift];
743416022c9SBarry Smith     tmp[i] = sum*aa[a->diag[i]+shift];
744da3a660dSBarry Smith     x[*c--] += tmp[i];
745da3a660dSBarry Smith   }
746da3a660dSBarry Smith 
7473b2fbd54SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
7483b2fbd54SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
749cb5b572fSBarry Smith   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
750cb5b572fSBarry Smith   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
751416022c9SBarry Smith   PLogFlops(2*a->nz);
752898183ecSLois Curfman McInnes 
7533a40ed3dSBarry Smith   PetscFunctionReturn(0);
754da3a660dSBarry Smith }
755da3a660dSBarry Smith /* -------------------------------------------------------------------*/
7565615d1e5SSatish Balay #undef __FUNC__
7577c922b88SBarry Smith #define __FUNC__ "MatSolveTranspose_SeqAIJ"
7587c922b88SBarry Smith int MatSolveTranspose_SeqAIJ(Mat A,Vec bb,Vec xx)
759da3a660dSBarry Smith {
760416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data;
7612235e667SBarry Smith   IS         iscol = a->col,isrow = a->row;
762416022c9SBarry Smith   int        *r,*c,ierr,i,n = a->m,*vi,*ai = a->i,*aj = a->j;
763f1af5d2fSBarry Smith   int        nz,shift = a->indexshift,*rout,*cout,*diag = a->diag;
764f1af5d2fSBarry Smith   Scalar     *x,*b,*tmp,*aa = a->a,*v,s1;
765da3a660dSBarry Smith 
7663a40ed3dSBarry Smith   PetscFunctionBegin;
767e1311b90SBarry Smith   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
768e1311b90SBarry Smith   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
769416022c9SBarry Smith   tmp  = a->solve_work;
770da3a660dSBarry Smith 
7712235e667SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
7722235e667SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
773da3a660dSBarry Smith 
774da3a660dSBarry Smith   /* copy the b into temp work space according to permutation */
7752235e667SBarry Smith   for (i=0; i<n; i++) tmp[i] = b[c[i]];
776da3a660dSBarry Smith 
777da3a660dSBarry Smith   /* forward solve the U^T */
778da3a660dSBarry Smith   for (i=0; i<n; i++) {
779f1af5d2fSBarry Smith     v   = aa + diag[i] + shift;
780f1af5d2fSBarry Smith     vi  = aj + diag[i] + (!shift);
781f1af5d2fSBarry Smith     nz  = ai[i+1] - diag[i] - 1;
782c38d4ed2SBarry Smith     s1  = tmp[i];
783c38d4ed2SBarry Smith     s1 *= *(v++);  /* multiply by inverse of diagonal entry */
784da3a660dSBarry Smith     while (nz--) {
785f1af5d2fSBarry Smith       tmp[*vi++ + shift] -= (*v++)*s1;
786da3a660dSBarry Smith     }
787c38d4ed2SBarry Smith     tmp[i] = s1;
788da3a660dSBarry Smith   }
789da3a660dSBarry Smith 
790da3a660dSBarry Smith   /* backward solve the L^T */
791da3a660dSBarry Smith   for (i=n-1; i>=0; i--){
792f1af5d2fSBarry Smith     v   = aa + diag[i] - 1 + shift;
793f1af5d2fSBarry Smith     vi  = aj + diag[i] - 1 + shift;
794f1af5d2fSBarry Smith     nz  = diag[i] - ai[i];
795f1af5d2fSBarry Smith     s1  = tmp[i];
796da3a660dSBarry Smith     while (nz--) {
797f1af5d2fSBarry Smith       tmp[*vi-- + shift] -= (*v--)*s1;
798da3a660dSBarry Smith     }
799da3a660dSBarry Smith   }
800da3a660dSBarry Smith 
801da3a660dSBarry Smith   /* copy tmp into x according to permutation */
802da3a660dSBarry Smith   for (i=0; i<n; i++) x[r[i]] = tmp[i];
803da3a660dSBarry Smith 
8042235e667SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
8052235e667SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
806cb5b572fSBarry Smith   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
807cb5b572fSBarry Smith   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
808da3a660dSBarry Smith 
809416022c9SBarry Smith   PLogFlops(2*a->nz-a->n);
8103a40ed3dSBarry Smith   PetscFunctionReturn(0);
811da3a660dSBarry Smith }
812da3a660dSBarry Smith 
8135615d1e5SSatish Balay #undef __FUNC__
8147c922b88SBarry Smith #define __FUNC__ "MatSolveTransposeAdd_SeqAIJ"
8157c922b88SBarry Smith int MatSolveTransposeAdd_SeqAIJ(Mat A,Vec bb,Vec zz,Vec xx)
816da3a660dSBarry Smith {
817416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data;
8182235e667SBarry Smith   IS         iscol = a->col,isrow = a->row;
819416022c9SBarry Smith   int        *r,*c,ierr,i,n = a->m,*vi,*ai = a->i,*aj = a->j;
820f1af5d2fSBarry Smith   int        nz,shift = a->indexshift,*rout,*cout,*diag = a->diag;
821416022c9SBarry Smith   Scalar     *x,*b,*tmp,*aa = a->a,*v;
8226abc6512SBarry Smith 
8233a40ed3dSBarry Smith   PetscFunctionBegin;
8246abc6512SBarry Smith   if (zz != xx) VecCopy(zz,xx);
8256abc6512SBarry Smith 
826e1311b90SBarry Smith   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
827e1311b90SBarry Smith   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
828416022c9SBarry Smith   tmp = a->solve_work;
8296abc6512SBarry Smith 
8302235e667SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
8312235e667SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
8326abc6512SBarry Smith 
8336abc6512SBarry Smith   /* copy the b into temp work space according to permutation */
8342235e667SBarry Smith   for (i=0; i<n; i++) tmp[i] = b[c[i]];
8356abc6512SBarry Smith 
8366abc6512SBarry Smith   /* forward solve the U^T */
8376abc6512SBarry Smith   for (i=0; i<n; i++) {
838f1af5d2fSBarry Smith     v   = aa + diag[i] + shift;
839f1af5d2fSBarry Smith     vi  = aj + diag[i] + (!shift);
840f1af5d2fSBarry Smith     nz  = ai[i+1] - diag[i] - 1;
8416abc6512SBarry Smith     tmp[i] *= *v++;
8426abc6512SBarry Smith     while (nz--) {
843dbb450caSBarry Smith       tmp[*vi++ + shift] -= (*v++)*tmp[i];
8446abc6512SBarry Smith     }
8456abc6512SBarry Smith   }
8466abc6512SBarry Smith 
8476abc6512SBarry Smith   /* backward solve the L^T */
8486abc6512SBarry Smith   for (i=n-1; i>=0; i--){
849f1af5d2fSBarry Smith     v   = aa + diag[i] - 1 + shift;
850f1af5d2fSBarry Smith     vi  = aj + diag[i] - 1 + shift;
851f1af5d2fSBarry Smith     nz  = diag[i] - ai[i];
8526abc6512SBarry Smith     while (nz--) {
853dbb450caSBarry Smith       tmp[*vi-- + shift] -= (*v--)*tmp[i];
8546abc6512SBarry Smith     }
8556abc6512SBarry Smith   }
8566abc6512SBarry Smith 
8576abc6512SBarry Smith   /* copy tmp into x according to permutation */
8586abc6512SBarry Smith   for (i=0; i<n; i++) x[r[i]] += tmp[i];
8596abc6512SBarry Smith 
8602235e667SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
8612235e667SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
862cb5b572fSBarry Smith   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
863cb5b572fSBarry Smith   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
8646abc6512SBarry Smith 
865416022c9SBarry Smith   PLogFlops(2*a->nz);
8663a40ed3dSBarry Smith   PetscFunctionReturn(0);
867da3a660dSBarry Smith }
8689e25ed09SBarry Smith /* ----------------------------------------------------------------*/
8697c922b88SBarry Smith extern int MatMissingDiagonal_SeqAIJ(Mat);
87008480c60SBarry Smith 
8715615d1e5SSatish Balay #undef __FUNC__
8725615d1e5SSatish Balay #define __FUNC__ "MatILUFactorSymbolic_SeqAIJ"
8736cf997b0SBarry Smith int MatILUFactorSymbolic_SeqAIJ(Mat A,IS isrow,IS iscol,MatILUInfo *info,Mat *fact)
8749e25ed09SBarry Smith {
875416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b;
8769e25ed09SBarry Smith   IS         isicol;
877416022c9SBarry Smith   int        *r,*ic,ierr,prow,n = a->m,*ai = a->i,*aj = a->j;
87856beaf04SBarry Smith   int        *ainew,*ajnew,jmax,*fill,*xi,nz,*im,*ajfill,*flev;
879335d9088SBarry Smith   int        *dloc,idx,row,m,fm,nzf,nzi,len, realloc = 0,dcount = 0;
8806cf997b0SBarry Smith   int        incrlev,nnz,i,shift = a->indexshift,levels,diagonal_fill;
88177c4ece6SBarry Smith   PetscTruth col_identity,row_identity;
882329f5518SBarry Smith   PetscReal  f;
8839e25ed09SBarry Smith 
8843a40ed3dSBarry Smith   PetscFunctionBegin;
8856cf997b0SBarry Smith   if (info) {
8866cf997b0SBarry Smith     f             = info->fill;
8870cd17293SBarry Smith     levels        = (int)info->levels;
8880cd17293SBarry Smith     diagonal_fill = (int)info->diagonal_fill;
8896cf997b0SBarry Smith   } else {
8906cf997b0SBarry Smith     f             = 1.0;
8916cf997b0SBarry Smith     levels        = 0;
8926cf997b0SBarry Smith     diagonal_fill = 0;
8936cf997b0SBarry Smith   }
894*4da77479SBarry Smith   if (!isrow) {
895*4da77479SBarry Smith     ierr = ISCreateStride(PETSC_COMM_SELF,A->M,0,1,&isrow);CHKERRQ(ierr);
896*4da77479SBarry Smith   }
897*4da77479SBarry Smith   if (!iscol) {
898*4da77479SBarry Smith     ierr = ISCreateStride(PETSC_COMM_SELF,A->M,0,1,&iscol);CHKERRQ(ierr);
899*4da77479SBarry Smith   }
900*4da77479SBarry Smith 
9014c49b128SBarry Smith   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
90282bf6240SBarry Smith 
90308480c60SBarry Smith   /* special case that simply copies fill pattern */
904be0abb6dSBarry Smith   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
905be0abb6dSBarry Smith   ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr);
90686bacbd2SBarry Smith   if (!levels && row_identity && col_identity) {
9072e8a6d31SBarry Smith     ierr = MatDuplicate_SeqAIJ(A,MAT_DO_NOT_COPY_VALUES,fact);CHKERRQ(ierr);
90808480c60SBarry Smith     (*fact)->factor = FACTOR_LU;
90908480c60SBarry Smith     b               = (Mat_SeqAIJ*)(*fact)->data;
91008480c60SBarry Smith     if (!b->diag) {
9117c922b88SBarry Smith       ierr = MatMarkDiagonal_SeqAIJ(*fact);CHKERRQ(ierr);
91208480c60SBarry Smith     }
9137c922b88SBarry Smith     ierr = MatMissingDiagonal_SeqAIJ(*fact);CHKERRQ(ierr);
91408480c60SBarry Smith     b->row              = isrow;
91508480c60SBarry Smith     b->col              = iscol;
91682bf6240SBarry Smith     b->icol             = isicol;
9170452661fSBarry Smith     b->solve_work       = (Scalar*)PetscMalloc((b->m+1)*sizeof(Scalar));CHKPTRQ(b->solve_work);
918f830108cSBarry Smith     (*fact)->ops->solve = MatSolve_SeqAIJ_NaturalOrdering;
919c38d4ed2SBarry Smith     ierr                = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
920c38d4ed2SBarry Smith     ierr                = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
9213a40ed3dSBarry Smith     PetscFunctionReturn(0);
92208480c60SBarry Smith   }
92308480c60SBarry Smith 
924898183ecSLois Curfman McInnes   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
925898183ecSLois Curfman McInnes   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
9269e25ed09SBarry Smith 
9279e25ed09SBarry Smith   /* get new row pointers */
9280452661fSBarry Smith   ainew = (int*)PetscMalloc((n+1)*sizeof(int));CHKPTRQ(ainew);
929dbb450caSBarry Smith   ainew[0] = -shift;
9309e25ed09SBarry Smith   /* don't know how many column pointers are needed so estimate */
931dbb450caSBarry Smith   jmax = (int)(f*(ai[n]+!shift));
9320452661fSBarry Smith   ajnew = (int*)PetscMalloc((jmax)*sizeof(int));CHKPTRQ(ajnew);
9339e25ed09SBarry Smith   /* ajfill is level of fill for each fill entry */
9340452661fSBarry Smith   ajfill = (int*)PetscMalloc((jmax)*sizeof(int));CHKPTRQ(ajfill);
9359e25ed09SBarry Smith   /* fill is a linked list of nonzeros in active row */
9360452661fSBarry Smith   fill = (int*)PetscMalloc((n+1)*sizeof(int));CHKPTRQ(fill);
93756beaf04SBarry Smith   /* im is level for each filled value */
9380452661fSBarry Smith   im = (int*)PetscMalloc((n+1)*sizeof(int));CHKPTRQ(im);
93956beaf04SBarry Smith   /* dloc is location of diagonal in factor */
9400452661fSBarry Smith   dloc = (int*)PetscMalloc((n+1)*sizeof(int));CHKPTRQ(dloc);
94156beaf04SBarry Smith   dloc[0]  = 0;
94256beaf04SBarry Smith   for (prow=0; prow<n; prow++) {
9436cf997b0SBarry Smith 
9446cf997b0SBarry Smith     /* copy current row into linked list */
94556beaf04SBarry Smith     nzf     = nz  = ai[r[prow]+1] - ai[r[prow]];
946385f7a74SSatish Balay     if (!nz) SETERRQ(PETSC_ERR_MAT_LU_ZRPVT,1,"Empty row in matrix");
947dbb450caSBarry Smith     xi      = aj + ai[r[prow]] + shift;
9489e25ed09SBarry Smith     fill[n]    = n;
949435faa5fSBarry Smith     fill[prow] = -1; /* marker to indicate if diagonal exists */
9509e25ed09SBarry Smith     while (nz--) {
9519e25ed09SBarry Smith       fm  = n;
952dbb450caSBarry Smith       idx = ic[*xi++ + shift];
9539e25ed09SBarry Smith       do {
9549e25ed09SBarry Smith         m  = fm;
9559e25ed09SBarry Smith         fm = fill[m];
9569e25ed09SBarry Smith       } while (fm < idx);
9579e25ed09SBarry Smith       fill[m]   = idx;
9589e25ed09SBarry Smith       fill[idx] = fm;
95956beaf04SBarry Smith       im[idx]   = 0;
9609e25ed09SBarry Smith     }
9616cf997b0SBarry Smith 
9626cf997b0SBarry Smith     /* make sure diagonal entry is included */
963435faa5fSBarry Smith     if (diagonal_fill && fill[prow] == -1) {
9646cf997b0SBarry Smith       fm = n;
965435faa5fSBarry Smith       while (fill[fm] < prow) fm = fill[fm];
966435faa5fSBarry Smith       fill[prow] = fill[fm]; /* insert diagonal into linked list */
967435faa5fSBarry Smith       fill[fm]   = prow;
9686cf997b0SBarry Smith       im[prow]   = 0;
9696cf997b0SBarry Smith       nzf++;
970335d9088SBarry Smith       dcount++;
9716cf997b0SBarry Smith     }
9726cf997b0SBarry Smith 
97356beaf04SBarry Smith     nzi = 0;
9749e25ed09SBarry Smith     row = fill[n];
97556beaf04SBarry Smith     while (row < prow) {
97656beaf04SBarry Smith       incrlev = im[row] + 1;
97756beaf04SBarry Smith       nz      = dloc[row];
9786cf997b0SBarry Smith       xi      = ajnew  + ainew[row] + shift + nz + 1;
979dbb450caSBarry Smith       flev    = ajfill + ainew[row] + shift + nz + 1;
980416022c9SBarry Smith       nnz     = ainew[row+1] - ainew[row] - nz - 1;
98156beaf04SBarry Smith       fm      = row;
98256beaf04SBarry Smith       while (nnz-- > 0) {
983dbb450caSBarry Smith         idx = *xi++ + shift;
98456beaf04SBarry Smith         if (*flev + incrlev > levels) {
98556beaf04SBarry Smith           flev++;
98656beaf04SBarry Smith           continue;
98756beaf04SBarry Smith         }
98856beaf04SBarry Smith         do {
9899e25ed09SBarry Smith           m  = fm;
9909e25ed09SBarry Smith           fm = fill[m];
99156beaf04SBarry Smith         } while (fm < idx);
9929e25ed09SBarry Smith         if (fm != idx) {
99356beaf04SBarry Smith           im[idx]   = *flev + incrlev;
9949e25ed09SBarry Smith           fill[m]   = idx;
9959e25ed09SBarry Smith           fill[idx] = fm;
9969e25ed09SBarry Smith           fm        = idx;
99756beaf04SBarry Smith           nzf++;
998ecf371e4SBarry Smith         } else {
99956beaf04SBarry Smith           if (im[idx] > *flev + incrlev) im[idx] = *flev+incrlev;
10009e25ed09SBarry Smith         }
100156beaf04SBarry Smith         flev++;
10029e25ed09SBarry Smith       }
10039e25ed09SBarry Smith       row = fill[row];
100456beaf04SBarry Smith       nzi++;
10059e25ed09SBarry Smith     }
10069e25ed09SBarry Smith     /* copy new filled row into permanent storage */
100756beaf04SBarry Smith     ainew[prow+1] = ainew[prow] + nzf;
1008d7e8b826SBarry Smith     if (ainew[prow+1] > jmax-shift) {
1009ecf371e4SBarry Smith 
1010ecf371e4SBarry Smith       /* estimate how much additional space we will need */
1011ecf371e4SBarry Smith       /* use the strategy suggested by David Hysom <hysom@perch-t.icase.edu> */
1012ecf371e4SBarry Smith       /* just double the memory each time */
1013ecf371e4SBarry Smith       /*  maxadd = (int)((f*(ai[n]+!shift)*(n-prow+5))/n); */
1014ecf371e4SBarry Smith       int maxadd = jmax;
101556beaf04SBarry Smith       if (maxadd < nzf) maxadd = (n-prow)*(nzf+1);
1016bbb6d6a8SBarry Smith       jmax += maxadd;
1017ecf371e4SBarry Smith 
1018ecf371e4SBarry Smith       /* allocate a longer ajnew and ajfill */
10190452661fSBarry Smith       xi     = (int*)PetscMalloc(jmax*sizeof(int));CHKPTRQ(xi);
1020549d3d68SSatish Balay       ierr   = PetscMemcpy(xi,ajnew,(ainew[prow]+shift)*sizeof(int));CHKERRQ(ierr);
1021606d414cSSatish Balay       ierr = PetscFree(ajnew);CHKERRQ(ierr);
102256beaf04SBarry Smith       ajnew  = xi;
10230452661fSBarry Smith       xi     = (int*)PetscMalloc(jmax*sizeof(int));CHKPTRQ(xi);
1024549d3d68SSatish Balay       ierr   = PetscMemcpy(xi,ajfill,(ainew[prow]+shift)*sizeof(int));CHKERRQ(ierr);
1025606d414cSSatish Balay       ierr = PetscFree(ajfill);CHKERRQ(ierr);
102656beaf04SBarry Smith       ajfill = xi;
1027ecf371e4SBarry Smith       realloc++; /* count how many times we realloc */
10289e25ed09SBarry Smith     }
1029dbb450caSBarry Smith     xi          = ajnew + ainew[prow] + shift;
1030dbb450caSBarry Smith     flev        = ajfill + ainew[prow] + shift;
103156beaf04SBarry Smith     dloc[prow]  = nzi;
10329e25ed09SBarry Smith     fm          = fill[n];
103356beaf04SBarry Smith     while (nzf--) {
1034dbb450caSBarry Smith       *xi++   = fm - shift;
103556beaf04SBarry Smith       *flev++ = im[fm];
10369e25ed09SBarry Smith       fm      = fill[fm];
10379e25ed09SBarry Smith     }
10386cf997b0SBarry Smith     /* make sure row has diagonal entry */
10396cf997b0SBarry Smith     if (ajnew[ainew[prow]+shift+dloc[prow]]+shift != prow) {
10406cf997b0SBarry Smith       SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,1,"Row %d has missing diagonal in factored matrix\n\
10416cf997b0SBarry Smith     try running with -pc_ilu_nonzeros_along_diagonal or -pc_ilu_diagonal_fill",prow);
10426cf997b0SBarry Smith     }
10439e25ed09SBarry Smith   }
1044606d414cSSatish Balay   ierr = PetscFree(ajfill);CHKERRQ(ierr);
1045898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
1046898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
1047606d414cSSatish Balay   ierr = PetscFree(fill);CHKERRQ(ierr);
1048606d414cSSatish Balay   ierr = PetscFree(im);CHKERRQ(ierr);
10499e25ed09SBarry Smith 
1050f64065bbSBarry Smith   {
1051329f5518SBarry Smith     PetscReal af = ((PetscReal)ainew[n])/((PetscReal)ai[n]);
1052c38d4ed2SBarry Smith     PLogInfo(A,"MatILUFactorSymbolic_SeqAIJ:Reallocs %d Fill ratio:given %g needed %g\n",realloc,f,af);
1053981c4779SBarry Smith     PLogInfo(A,"MatILUFactorSymbolic_SeqAIJ:Run with -pc_ilu_fill %g or use \n",af);
1054981c4779SBarry Smith     PLogInfo(A,"MatILUFactorSymbolic_SeqAIJ:PCILUSetFill(pc,%g);\n",af);
1055981c4779SBarry Smith     PLogInfo(A,"MatILUFactorSymbolic_SeqAIJ:for best performance.\n");
1056335d9088SBarry Smith     if (diagonal_fill) {
1057335d9088SBarry Smith       PLogInfo(A,"MatILUFactorSymbolic_SeqAIJ:Detected and replace %d missing diagonals",dcount);
1058335d9088SBarry Smith     }
1059f64065bbSBarry Smith   }
1060bbb6d6a8SBarry Smith 
10619e25ed09SBarry Smith   /* put together the new matrix */
1062b4fd4287SBarry Smith   ierr = MatCreateSeqAIJ(A->comm,n,n,0,PETSC_NULL,fact);CHKERRQ(ierr);
1063fa6007c9SSatish Balay   PLogObjectParent(*fact,isicol);
1064416022c9SBarry Smith   b = (Mat_SeqAIJ*)(*fact)->data;
1065606d414cSSatish Balay   ierr = PetscFree(b->imax);CHKERRQ(ierr);
10667c922b88SBarry Smith   b->singlemalloc = PETSC_FALSE;
1067dbb450caSBarry Smith   len = (ainew[n] + shift)*sizeof(Scalar);
10689e25ed09SBarry Smith   /* the next line frees the default space generated by the Create() */
1069606d414cSSatish Balay   ierr = PetscFree(b->a);CHKERRQ(ierr);
1070606d414cSSatish Balay   ierr = PetscFree(b->ilen);CHKERRQ(ierr);
10716b96ef82SSatish Balay   b->a          = (Scalar*)PetscMalloc(len+1);CHKPTRQ(b->a);
1072416022c9SBarry Smith   b->j          = ajnew;
1073416022c9SBarry Smith   b->i          = ainew;
107456beaf04SBarry Smith   for (i=0; i<n; i++) dloc[i] += ainew[i];
1075416022c9SBarry Smith   b->diag       = dloc;
1076416022c9SBarry Smith   b->ilen       = 0;
1077416022c9SBarry Smith   b->imax       = 0;
1078416022c9SBarry Smith   b->row        = isrow;
1079416022c9SBarry Smith   b->col        = iscol;
1080c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
1081c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
108282bf6240SBarry Smith   b->icol       = isicol;
108382bf6240SBarry Smith   b->solve_work = (Scalar*)PetscMalloc((n+1)*sizeof(Scalar));CHKPTRQ(b->solve_work);
1084416022c9SBarry Smith   /* In b structure:  Free imax, ilen, old a, old j.
108556beaf04SBarry Smith      Allocate dloc, solve_work, new a, new j */
1086dbb450caSBarry Smith   PLogObjectMemory(*fact,(ainew[n]+shift-n) * (sizeof(int)+sizeof(Scalar)));
1087416022c9SBarry Smith   b->maxnz          = b->nz = ainew[n] + shift;
10889e25ed09SBarry Smith   (*fact)->factor   = FACTOR_LU;
1089ae068f58SLois Curfman McInnes 
1090ae068f58SLois Curfman McInnes   (*fact)->info.factor_mallocs    = realloc;
1091ae068f58SLois Curfman McInnes   (*fact)->info.fill_ratio_given  = f;
1092329f5518SBarry Smith   (*fact)->info.fill_ratio_needed = ((PetscReal)ainew[n])/((PetscReal)ai[prow]);
1093329f5518SBarry Smith   (*fact)->factor                 =  FACTOR_LU;
1094ae068f58SLois Curfman McInnes 
10953a40ed3dSBarry Smith   PetscFunctionReturn(0);
10969e25ed09SBarry Smith }
1097d5d45c9bSBarry Smith 
1098d5d45c9bSBarry Smith 
1099d5d45c9bSBarry Smith 
1100d5d45c9bSBarry Smith 
1101