xref: /petsc/src/mat/impls/aij/seq/aijfact.c (revision 9e046878677d6d1698ae34b565a09b19a637cbe3)
1*9e046878SBarry Smith /*$Id: aijfact.c,v 1.150 2000/05/10 16:40:36 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__
8b2863d3aSBarry Smith #define __FUNC__ /*<a name=""></a>*/"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 
20ca44d042SBarry Smith EXTERN int MatMarkDiagonal_SeqAIJ(Mat);
21ca44d042SBarry Smith EXTERN int Mat_AIJ_CheckInode(Mat);
2286bacbd2SBarry Smith 
23ca44d042SBarry Smith EXTERN int SPARSEKIT2dperm(int*,Scalar*,int*,int*,Scalar*,int*,int*,int*,int*,int*);
24ca44d042SBarry Smith EXTERN int SPARSEKIT2ilutp(int*,Scalar*,int*,int*,int*,PetscReal*,PetscReal*,int*,Scalar*,int*,int*,int*,Scalar*,int*,int*,int*);
25ca44d042SBarry Smith EXTERN int SPARSEKIT2msrcsr(int*,Scalar*,int*,Scalar*,int*,int*,Scalar*,int*);
2686bacbd2SBarry Smith 
2786bacbd2SBarry Smith #undef __FUNC__
28b2863d3aSBarry Smith #define __FUNC__ /*<a name=""></a>*/"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__
268*9e046878SBarry Smith #define __FUNC__ /*<a name="MatLUFactorSymbolic_SeqAIJ""></a>*/"MatLUFactorSymbolic_SeqAIJ"
269*9e046878SBarry Smith int MatLUFactorSymbolic_SeqAIJ(Mat A,IS isrow,IS iscol,MatLUInfo *info,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;
276*9e046878SBarry Smith   PetscReal  f;
277289bc588SBarry Smith 
2783a40ed3dSBarry Smith   PetscFunctionBegin;
279d3cbd50cSLois Curfman McInnes   PetscValidHeaderSpecific(isrow,IS_COOKIE);
280d3cbd50cSLois Curfman McInnes   PetscValidHeaderSpecific(iscol,IS_COOKIE);
281f1af5d2fSBarry Smith   if (A->M != A->N) SETERRQ(PETSC_ERR_ARG_WRONG,0,"matrix must be square");
2823b2fbd54SBarry Smith 
2839863cb4fSBarry Smith   if (!isrow) {
2849863cb4fSBarry Smith     ierr = ISCreateStride(PETSC_COMM_SELF,A->M,0,1,&isrow);CHKERRQ(ierr);
2859863cb4fSBarry Smith   }
2869863cb4fSBarry Smith   if (!iscol) {
2879863cb4fSBarry Smith     ierr = ISCreateStride(PETSC_COMM_SELF,A->M,0,1,&iscol);CHKERRQ(ierr);
2889863cb4fSBarry Smith   }
2899863cb4fSBarry Smith 
2904c49b128SBarry Smith   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
291ac121b3dSBarry Smith   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
292ac121b3dSBarry Smith   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
293289bc588SBarry Smith 
294289bc588SBarry Smith   /* get new row pointers */
2950452661fSBarry Smith   ainew    = (int*)PetscMalloc((n+1)*sizeof(int));CHKPTRQ(ainew);
296dbb450caSBarry Smith   ainew[0] = -shift;
297289bc588SBarry Smith   /* don't know how many column pointers are needed so estimate */
298*9e046878SBarry Smith   if (info) f = info->fill; else f = 1.0;
299dbb450caSBarry Smith   jmax  = (int)(f*ai[n]+(!shift));
3000452661fSBarry Smith   ajnew = (int*)PetscMalloc((jmax)*sizeof(int));CHKPTRQ(ajnew);
301289bc588SBarry Smith   /* fill is a linked list of nonzeros in active row */
3020452661fSBarry Smith   fill = (int*)PetscMalloc((2*n+1)*sizeof(int));CHKPTRQ(fill);
3032fb3ed76SBarry Smith   im   = fill + n + 1;
304289bc588SBarry Smith   /* idnew is location of diagonal in factor */
3050452661fSBarry Smith   idnew    = (int*)PetscMalloc((n+1)*sizeof(int));CHKPTRQ(idnew);
306dbb450caSBarry Smith   idnew[0] = -shift;
307289bc588SBarry Smith 
308289bc588SBarry Smith   for (i=0; i<n; i++) {
309289bc588SBarry Smith     /* first copy previous fill into linked list */
310289bc588SBarry Smith     nnz     = nz    = ai[r[i]+1] - ai[r[i]];
3116b96ef82SSatish Balay     if (!nz) SETERRQ(PETSC_ERR_MAT_LU_ZRPVT,1,"Empty row in matrix");
312dbb450caSBarry Smith     ajtmp   = aj + ai[r[i]] + shift;
313289bc588SBarry Smith     fill[n] = n;
314289bc588SBarry Smith     while (nz--) {
315289bc588SBarry Smith       fm  = n;
316dbb450caSBarry Smith       idx = ic[*ajtmp++ + shift];
317289bc588SBarry Smith       do {
318289bc588SBarry Smith         m  = fm;
319289bc588SBarry Smith         fm = fill[m];
320289bc588SBarry Smith       } while (fm < idx);
321289bc588SBarry Smith       fill[m]   = idx;
322289bc588SBarry Smith       fill[idx] = fm;
323289bc588SBarry Smith     }
324289bc588SBarry Smith     row = fill[n];
325289bc588SBarry Smith     while (row < i) {
326dbb450caSBarry Smith       ajtmp = ajnew + idnew[row] + (!shift);
3272fb3ed76SBarry Smith       nzbd  = 1 + idnew[row] - ainew[row];
3282fb3ed76SBarry Smith       nz    = im[row] - nzbd;
329289bc588SBarry Smith       fm    = row;
3302fb3ed76SBarry Smith       while (nz-- > 0) {
331dbb450caSBarry Smith         idx = *ajtmp++ + shift;
3322fb3ed76SBarry Smith         nzbd++;
3332fb3ed76SBarry Smith         if (idx == i) im[row] = nzbd;
334289bc588SBarry Smith         do {
335289bc588SBarry Smith           m  = fm;
336289bc588SBarry Smith           fm = fill[m];
337289bc588SBarry Smith         } while (fm < idx);
338289bc588SBarry Smith         if (fm != idx) {
339289bc588SBarry Smith           fill[m]   = idx;
340289bc588SBarry Smith           fill[idx] = fm;
341289bc588SBarry Smith           fm        = idx;
342289bc588SBarry Smith           nnz++;
343289bc588SBarry Smith         }
344289bc588SBarry Smith       }
345289bc588SBarry Smith       row = fill[row];
346289bc588SBarry Smith     }
347289bc588SBarry Smith     /* copy new filled row into permanent storage */
348289bc588SBarry Smith     ainew[i+1] = ainew[i] + nnz;
349496e697eSBarry Smith     if (ainew[i+1] > jmax) {
350ecf371e4SBarry Smith 
351ecf371e4SBarry Smith       /* estimate how much additional space we will need */
352ecf371e4SBarry Smith       /* use the strategy suggested by David Hysom <hysom@perch-t.icase.edu> */
353ecf371e4SBarry Smith       /* just double the memory each time */
354ecf371e4SBarry Smith       int maxadd = jmax;
355ecf371e4SBarry Smith       /* maxadd = (int)((f*(ai[n]+(!shift))*(n-i+5))/n); */
356bbb6d6a8SBarry Smith       if (maxadd < nnz) maxadd = (n-i)*(nnz+1);
3572fb3ed76SBarry Smith       jmax += maxadd;
358ecf371e4SBarry Smith 
359ecf371e4SBarry Smith       /* allocate a longer ajnew */
3600452661fSBarry Smith       ajtmp = (int*)PetscMalloc(jmax*sizeof(int));CHKPTRQ(ajtmp);
361549d3d68SSatish Balay       ierr  = PetscMemcpy(ajtmp,ajnew,(ainew[i]+shift)*sizeof(int));CHKERRQ(ierr);
362606d414cSSatish Balay       ierr  = PetscFree(ajnew);CHKERRQ(ierr);
363289bc588SBarry Smith       ajnew = ajtmp;
3642fb3ed76SBarry Smith       realloc++; /* count how many times we realloc */
365289bc588SBarry Smith     }
366dbb450caSBarry Smith     ajtmp = ajnew + ainew[i] + shift;
367289bc588SBarry Smith     fm    = fill[n];
368289bc588SBarry Smith     nzi   = 0;
3692fb3ed76SBarry Smith     im[i] = nnz;
370289bc588SBarry Smith     while (nnz--) {
371289bc588SBarry Smith       if (fm < i) nzi++;
372dbb450caSBarry Smith       *ajtmp++ = fm - shift;
373289bc588SBarry Smith       fm       = fill[fm];
374289bc588SBarry Smith     }
375289bc588SBarry Smith     idnew[i] = ainew[i] + nzi;
376289bc588SBarry Smith   }
377464e8d44SSatish Balay   if (ai[n] != 0) {
378329f5518SBarry Smith     PetscReal af = ((PetscReal)ainew[n])/((PetscReal)ai[n]);
379c38d4ed2SBarry Smith     PLogInfo(A,"MatLUFactorSymbolic_SeqAIJ:Reallocs %d Fill ratio:given %g needed %g\n",realloc,f,af);
380981c4779SBarry Smith     PLogInfo(A,"MatLUFactorSymbolic_SeqAIJ:Run with -pc_lu_fill %g or use \n",af);
381981c4779SBarry Smith     PLogInfo(A,"MatLUFactorSymbolic_SeqAIJ:PCLUSetFill(pc,%g);\n",af);
382981c4779SBarry Smith     PLogInfo(A,"MatLUFactorSymbolic_SeqAIJ:for best performance.\n");
38305bf559cSSatish Balay   } else {
384981c4779SBarry Smith     PLogInfo(A,"MatLUFactorSymbolic_SeqAIJ: Empty matrix\n");
38505bf559cSSatish Balay   }
3862fb3ed76SBarry Smith 
387898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
388898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
3891987afe7SBarry Smith 
390606d414cSSatish Balay   ierr = PetscFree(fill);CHKERRQ(ierr);
391289bc588SBarry Smith 
392289bc588SBarry Smith   /* put together the new matrix */
393b4fd4287SBarry Smith   ierr = MatCreateSeqAIJ(A->comm,n,n,0,PETSC_NULL,B);CHKERRQ(ierr);
3941987afe7SBarry Smith   PLogObjectParent(*B,isicol);
395416022c9SBarry Smith   b = (Mat_SeqAIJ*)(*B)->data;
396606d414cSSatish Balay   ierr = PetscFree(b->imax);CHKERRQ(ierr);
3977c922b88SBarry Smith   b->singlemalloc = PETSC_FALSE;
398e8d4e0b9SBarry Smith   /* the next line frees the default space generated by the Create() */
399606d414cSSatish Balay   ierr = PetscFree(b->a);CHKERRQ(ierr);
400606d414cSSatish Balay   ierr = PetscFree(b->ilen);CHKERRQ(ierr);
40191bf9adeSBarry Smith   b->a          = (Scalar*)PetscMalloc((ainew[n]+shift+1)*sizeof(Scalar));CHKPTRQ(b->a);
402416022c9SBarry Smith   b->j          = ajnew;
403416022c9SBarry Smith   b->i          = ainew;
404416022c9SBarry Smith   b->diag       = idnew;
405416022c9SBarry Smith   b->ilen       = 0;
406416022c9SBarry Smith   b->imax       = 0;
407416022c9SBarry Smith   b->row        = isrow;
408416022c9SBarry Smith   b->col        = iscol;
409c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
410c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
41182bf6240SBarry Smith   b->icol       = isicol;
41291bf9adeSBarry Smith   b->solve_work = (Scalar*)PetscMalloc((n+1)*sizeof(Scalar));CHKPTRQ(b->solve_work);
413416022c9SBarry Smith   /* In b structure:  Free imax, ilen, old a, old j.
4147fd98bd6SLois Curfman McInnes      Allocate idnew, solve_work, new a, new j */
415416022c9SBarry Smith   PLogObjectMemory(*B,(ainew[n]+shift-n)*(sizeof(int)+sizeof(Scalar)));
416416022c9SBarry Smith   b->maxnz = b->nz = ainew[n] + shift;
4177fd98bd6SLois Curfman McInnes 
418329f5518SBarry Smith   (*B)->factor                 =  FACTOR_LU;
419ae068f58SLois Curfman McInnes   (*B)->info.factor_mallocs    = realloc;
420ae068f58SLois Curfman McInnes   (*B)->info.fill_ratio_given  = f;
4217dda7e10SBarry Smith   (*B)->ops->lufactornumeric   =  A->ops->lufactornumeric; /* Use Inode variant ONLY if A has inodes */
422703deb49SSatish Balay 
423e93ccc4dSBarry Smith   if (ai[n] != 0) {
424329f5518SBarry Smith     (*B)->info.fill_ratio_needed = ((PetscReal)ainew[n])/((PetscReal)ai[n]);
425eea2913aSSatish Balay   } else {
426eea2913aSSatish Balay     (*B)->info.fill_ratio_needed = 0.0;
427eea2913aSSatish Balay   }
4283a40ed3dSBarry Smith   PetscFunctionReturn(0);
429289bc588SBarry Smith }
4300a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
431ca44d042SBarry Smith EXTERN int Mat_AIJ_CheckInode(Mat);
43241c01911SSatish Balay 
4335615d1e5SSatish Balay #undef __FUNC__
434*9e046878SBarry Smith #define __FUNC__ /*<a name="MatLUFactorNumeric_SeqAIJ"></a>*/"MatLUFactorNumeric_SeqAIJ"
435416022c9SBarry Smith int MatLUFactorNumeric_SeqAIJ(Mat A,Mat *B)
436289bc588SBarry Smith {
437416022c9SBarry Smith   Mat        C = *B;
438416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b = (Mat_SeqAIJ *)C->data;
43982bf6240SBarry Smith   IS         isrow = b->row,isicol = b->icol;
440416022c9SBarry Smith   int        *r,*ic,ierr,i,j,n = a->m,*ai = b->i,*aj = b->j;
4411987afe7SBarry Smith   int        *ajtmpold,*ajtmp,nz,row,*ics,shift = a->indexshift;
442f2582111SSatish Balay   int        *diag_offset = b->diag,diag,k;
443329f5518SBarry Smith   int        preserve_row_sums = (int)a->ilu_preserve_row_sums,*pj;
444329f5518SBarry Smith   Scalar     *rtmp,*v,*pc,multiplier,sum,inner_sum,*rowsums = 0,*pv,*rtmps,*u_values;
445329f5518SBarry Smith   PetscReal  ssum;
446289bc588SBarry Smith 
4473a40ed3dSBarry Smith   PetscFunctionBegin;
44882bf6240SBarry Smith 
44978b31e54SBarry Smith   ierr  = ISGetIndices(isrow,&r);CHKERRQ(ierr);
45078b31e54SBarry Smith   ierr  = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
4510452661fSBarry Smith   rtmp  = (Scalar*)PetscMalloc((n+1)*sizeof(Scalar));CHKPTRQ(rtmp);
452549d3d68SSatish Balay   ierr  = PetscMemzero(rtmp,(n+1)*sizeof(Scalar));CHKERRQ(ierr);
4530cf60383SBarry Smith   rtmps = rtmp + shift; ics = ic + shift;
454289bc588SBarry Smith 
4556cf997b0SBarry Smith   /* precalculate row sums */
45635aab85fSBarry Smith   if (preserve_row_sums) {
45735aab85fSBarry Smith     rowsums = (Scalar*)PetscMalloc(n*sizeof(Scalar));CHKPTRQ(rowsums);
45835aab85fSBarry Smith     for (i=0; i<n; i++) {
45935aab85fSBarry Smith       nz  = a->i[r[i]+1] - a->i[r[i]];
46035aab85fSBarry Smith       v   = a->a + a->i[r[i]] + shift;
46135aab85fSBarry Smith       sum = 0.0;
46235aab85fSBarry Smith       for (j=0; j<nz; j++) sum += v[j];
46335aab85fSBarry Smith       rowsums[i] = sum;
46435aab85fSBarry Smith     }
46535aab85fSBarry Smith   }
46635aab85fSBarry Smith 
467289bc588SBarry Smith   for (i=0; i<n; i++) {
468289bc588SBarry Smith     nz    = ai[i+1] - ai[i];
469dbb450caSBarry Smith     ajtmp = aj + ai[i] + shift;
47048e94559SBarry Smith     for  (j=0; j<nz; j++) rtmps[ajtmp[j]] = 0.0;
471289bc588SBarry Smith 
472289bc588SBarry Smith     /* load in initial (unfactored row) */
473416022c9SBarry Smith     nz       = a->i[r[i]+1] - a->i[r[i]];
474416022c9SBarry Smith     ajtmpold = a->j + a->i[r[i]] + shift;
475416022c9SBarry Smith     v        = a->a + a->i[r[i]] + shift;
4760cf60383SBarry Smith     for (j=0; j<nz; j++) rtmp[ics[ajtmpold[j]]] =  v[j];
477289bc588SBarry Smith 
478dbb450caSBarry Smith     row = *ajtmp++ + shift;
479f2582111SSatish Balay     while  (row < i) {
4808c37ef55SBarry Smith       pc = rtmp + row;
4818c37ef55SBarry Smith       if (*pc != 0.0) {
4821987afe7SBarry Smith         pv         = b->a + diag_offset[row] + shift;
4831987afe7SBarry Smith         pj         = b->j + diag_offset[row] + (!shift);
48435aab85fSBarry Smith         multiplier = *pc / *pv++;
4858c37ef55SBarry Smith         *pc        = multiplier;
486f2582111SSatish Balay         nz         = ai[row+1] - diag_offset[row] - 1;
487f2582111SSatish Balay         for (j=0; j<nz; j++) rtmps[pj[j]] -= multiplier * pv[j];
488f2582111SSatish Balay         PLogFlops(2*nz);
489289bc588SBarry Smith       }
490dbb450caSBarry Smith       row = *ajtmp++ + shift;
491289bc588SBarry Smith     }
492416022c9SBarry Smith     /* finished row so stick it into b->a */
493416022c9SBarry Smith     pv = b->a + ai[i] + shift;
494416022c9SBarry Smith     pj = b->j + ai[i] + shift;
4958c37ef55SBarry Smith     nz = ai[i+1] - ai[i];
496416022c9SBarry Smith     for (j=0; j<nz; j++) {pv[j] = rtmps[pj[j]];}
49735aab85fSBarry Smith     diag = diag_offset[i] - ai[i];
49835aab85fSBarry Smith     /*
49935aab85fSBarry Smith           Possibly adjust diagonal entry on current row to force
50035aab85fSBarry Smith         LU matrix to have same row sum as initial matrix.
50135aab85fSBarry Smith     */
502d708749eSBarry Smith     if (pv[diag] == 0.0) {
5036cf997b0SBarry Smith       SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,0,"Zero pivot row %d",i);
504d708749eSBarry Smith     }
50535aab85fSBarry Smith     if (preserve_row_sums) {
50635aab85fSBarry Smith       pj  = b->j + ai[i] + shift;
50735aab85fSBarry Smith       sum = rowsums[i];
50835aab85fSBarry Smith       for (j=0; j<diag; j++) {
50935aab85fSBarry Smith         u_values  = b->a + diag_offset[pj[j]] + shift;
51035aab85fSBarry Smith         nz        = ai[pj[j]+1] - diag_offset[pj[j]];
51135aab85fSBarry Smith         inner_sum = 0.0;
51235aab85fSBarry Smith         for (k=0; k<nz; k++) {
51335aab85fSBarry Smith           inner_sum += u_values[k];
51435aab85fSBarry Smith         }
51535aab85fSBarry Smith         sum -= pv[j]*inner_sum;
51635aab85fSBarry Smith 
51735aab85fSBarry Smith       }
51835aab85fSBarry Smith       nz       = ai[i+1] - diag_offset[i] - 1;
51935aab85fSBarry Smith       u_values = b->a + diag_offset[i] + 1 + shift;
52035aab85fSBarry Smith       for (k=0; k<nz; k++) {
52135aab85fSBarry Smith         sum -= u_values[k];
52235aab85fSBarry Smith       }
52335aab85fSBarry Smith       ssum = PetscAbsScalar(sum/pv[diag]);
52435aab85fSBarry Smith       if (ssum < 1000. && ssum > .001) pv[diag] = sum;
52535aab85fSBarry Smith     }
52635aab85fSBarry Smith     /* check pivot entry for current row */
5278c37ef55SBarry Smith   }
5280f11f9aeSSatish Balay 
52935aab85fSBarry Smith   /* invert diagonal entries for simplier triangular solves */
53035aab85fSBarry Smith   for (i=0; i<n; i++) {
53135aab85fSBarry Smith     b->a[diag_offset[i]+shift] = 1.0/b->a[diag_offset[i]+shift];
53235aab85fSBarry Smith   }
53335aab85fSBarry Smith 
534606d414cSSatish Balay   if (preserve_row_sums) {ierr = PetscFree(rowsums);CHKERRQ(ierr);}
535606d414cSSatish Balay   ierr = PetscFree(rtmp);CHKERRQ(ierr);
53678b31e54SBarry Smith   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
53778b31e54SBarry Smith   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
538416022c9SBarry Smith   C->factor = FACTOR_LU;
53941c01911SSatish Balay   ierr = Mat_AIJ_CheckInode(C);CHKERRQ(ierr);
5407dda7e10SBarry Smith   (*B)->ops->lufactornumeric   =  A->ops->lufactornumeric; /* Use Inode variant ONLY if A has inodes */
541c456f294SBarry Smith   C->assembled = PETSC_TRUE;
542416022c9SBarry Smith   PLogFlops(b->n);
5433a40ed3dSBarry Smith   PetscFunctionReturn(0);
544289bc588SBarry Smith }
5450a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
5465615d1e5SSatish Balay #undef __FUNC__
547*9e046878SBarry Smith #define __FUNC__ /*<a name="MatLUFactor_SeqAIJ"></a>*/"MatLUFactor_SeqAIJ"
548*9e046878SBarry Smith int MatLUFactor_SeqAIJ(Mat A,IS row,IS col,MatLUInfo *info)
549da3a660dSBarry Smith {
550416022c9SBarry Smith   Mat_SeqAIJ     *mat = (Mat_SeqAIJ*)A->data;
55186bacbd2SBarry Smith   int            ierr,refct;
552416022c9SBarry Smith   Mat            C;
553f830108cSBarry Smith   PetscOps       *Abops;
5540a6ffc59SBarry Smith   MatOps         Aops;
555416022c9SBarry Smith 
5563a40ed3dSBarry Smith   PetscFunctionBegin;
557*9e046878SBarry Smith   ierr = MatLUFactorSymbolic(A,row,col,info,&C);CHKERRQ(ierr);
558f2582111SSatish Balay   ierr = MatLUFactorNumeric(A,&C);CHKERRQ(ierr);
559da3a660dSBarry Smith 
560da3a660dSBarry Smith   /* free all the data structures from mat */
561606d414cSSatish Balay   ierr = PetscFree(mat->a);CHKERRQ(ierr);
562606d414cSSatish Balay   if (!mat->singlemalloc) {
563606d414cSSatish Balay     ierr = PetscFree(mat->i);CHKERRQ(ierr);
564606d414cSSatish Balay     ierr = PetscFree(mat->j);CHKERRQ(ierr);
565606d414cSSatish Balay   }
566606d414cSSatish Balay   if (mat->diag) {ierr = PetscFree(mat->diag);CHKERRQ(ierr);}
567606d414cSSatish Balay   if (mat->ilen) {ierr = PetscFree(mat->ilen);CHKERRQ(ierr);}
568606d414cSSatish Balay   if (mat->imax) {ierr = PetscFree(mat->imax);CHKERRQ(ierr);}
569606d414cSSatish Balay   if (mat->solve_work) {ierr = PetscFree(mat->solve_work);CHKERRQ(ierr);}
570606d414cSSatish Balay   if (mat->inode.size) {ierr = PetscFree(mat->inode.size);CHKERRQ(ierr);}
5710f0aacb9SBarry Smith   if (mat->icol) {ierr = ISDestroy(mat->icol);CHKERRQ(ierr);}
572606d414cSSatish Balay   ierr = PetscFree(mat);CHKERRQ(ierr);
573da3a660dSBarry Smith 
57417642b18SBarry Smith   ierr = MapDestroy(A->rmap);CHKERRQ(ierr);
57517642b18SBarry Smith   ierr = MapDestroy(A->cmap);CHKERRQ(ierr);
57617642b18SBarry Smith 
577f830108cSBarry Smith   /*
578f830108cSBarry Smith        This is horrible, horrible code. We need to keep the
579f830108cSBarry Smith     A pointers for the bops and ops but copy everything
580f830108cSBarry Smith     else from C.
581f830108cSBarry Smith   */
582f830108cSBarry Smith   Abops = A->bops;
583f830108cSBarry Smith   Aops  = A->ops;
58486bacbd2SBarry Smith   refct = A->refct;
585549d3d68SSatish Balay   ierr  = PetscMemcpy(A,C,sizeof(struct _p_Mat));CHKERRQ(ierr);
586451c4af8SSatish Balay   mat   = (Mat_SeqAIJ*)A->data;
587451c4af8SSatish Balay   PLogObjectParent(A,mat->icol);
588451c4af8SSatish Balay 
589f830108cSBarry Smith   A->bops  = Abops;
590f830108cSBarry Smith   A->ops   = Aops;
591bef8e0ddSBarry Smith   A->qlist = 0;
59286bacbd2SBarry Smith   A->refct = refct;
593c38d4ed2SBarry Smith   /* copy over the type_name and name */
594c38d4ed2SBarry Smith   ierr     = PetscStrallocpy(C->type_name,&A->type_name);CHKERRQ(ierr);
595c38d4ed2SBarry Smith   ierr     = PetscStrallocpy(C->name,&A->name);CHKERRQ(ierr);
596f830108cSBarry Smith 
5970452661fSBarry Smith   PetscHeaderDestroy(C);
598c38d4ed2SBarry Smith 
5993a40ed3dSBarry Smith   PetscFunctionReturn(0);
600da3a660dSBarry Smith }
6010a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
6025615d1e5SSatish Balay #undef __FUNC__
603b2863d3aSBarry Smith #define __FUNC__ /*<a name=""></a>*/"MatSolve_SeqAIJ"
604416022c9SBarry Smith int MatSolve_SeqAIJ(Mat A,Vec bb,Vec xx)
6058c37ef55SBarry Smith {
606416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data;
607416022c9SBarry Smith   IS         iscol = a->col,isrow = a->row;
608416022c9SBarry Smith   int        *r,*c,ierr,i, n = a->m,*vi,*ai = a->i,*aj = a->j;
6093b2fbd54SBarry Smith   int        nz,shift = a->indexshift,*rout,*cout;
610416022c9SBarry Smith   Scalar     *x,*b,*tmp,*tmps,*aa = a->a,sum,*v;
6118c37ef55SBarry Smith 
6123a40ed3dSBarry Smith   PetscFunctionBegin;
6133a40ed3dSBarry Smith   if (!n) PetscFunctionReturn(0);
61491bf9adeSBarry Smith 
615e1311b90SBarry Smith   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
616e1311b90SBarry Smith   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
617416022c9SBarry Smith   tmp  = a->solve_work;
6188c37ef55SBarry Smith 
6193b2fbd54SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
6203b2fbd54SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
6218c37ef55SBarry Smith 
6228c37ef55SBarry Smith   /* forward solve the lower triangular */
6238c37ef55SBarry Smith   tmp[0] = b[*r++];
6240cf60383SBarry Smith   tmps   = tmp + shift;
6258c37ef55SBarry Smith   for (i=1; i<n; i++) {
626dbb450caSBarry Smith     v   = aa + ai[i] + shift;
627dbb450caSBarry Smith     vi  = aj + ai[i] + shift;
628416022c9SBarry Smith     nz  = a->diag[i] - ai[i];
6298c37ef55SBarry Smith     sum = b[*r++];
6300cf60383SBarry Smith     while (nz--) sum -= *v++ * tmps[*vi++];
6318c37ef55SBarry Smith     tmp[i] = sum;
6328c37ef55SBarry Smith   }
6338c37ef55SBarry Smith 
6348c37ef55SBarry Smith   /* backward solve the upper triangular */
6358c37ef55SBarry Smith   for (i=n-1; i>=0; i--){
636416022c9SBarry Smith     v   = aa + a->diag[i] + (!shift);
637416022c9SBarry Smith     vi  = aj + a->diag[i] + (!shift);
638416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
6398c37ef55SBarry Smith     sum = tmp[i];
6400cf60383SBarry Smith     while (nz--) sum -= *v++ * tmps[*vi++];
641416022c9SBarry Smith     x[*c--] = tmp[i] = sum*aa[a->diag[i]+shift];
6428c37ef55SBarry Smith   }
6438c37ef55SBarry Smith 
6443b2fbd54SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
6453b2fbd54SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
646cb5b572fSBarry Smith   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
647cb5b572fSBarry Smith   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
648416022c9SBarry Smith   PLogFlops(2*a->nz - a->n);
6493a40ed3dSBarry Smith   PetscFunctionReturn(0);
6508c37ef55SBarry Smith }
651026e39d0SSatish Balay 
652930ae53cSBarry Smith /* ----------------------------------------------------------- */
653930ae53cSBarry Smith #undef __FUNC__
654b2863d3aSBarry Smith #define __FUNC__ /*<a name=""></a>*/"MatSolve_SeqAIJ_NaturalOrdering"
655930ae53cSBarry Smith int MatSolve_SeqAIJ_NaturalOrdering(Mat A,Vec bb,Vec xx)
656930ae53cSBarry Smith {
657930ae53cSBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data;
658d85afc42SSatish Balay   int        n = a->m,*ai = a->i,*aj = a->j,*adiag = a->diag,ierr;
659d85afc42SSatish Balay   Scalar     *x,*b,*aa = a->a,sum;
660aa482453SBarry Smith #if !defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ)
661d85afc42SSatish Balay   int        adiag_i,i,*vi,nz,ai_i;
662d85afc42SSatish Balay   Scalar     *v;
663d85afc42SSatish Balay #endif
664930ae53cSBarry Smith 
6653a40ed3dSBarry Smith   PetscFunctionBegin;
6663a40ed3dSBarry Smith   if (!n) PetscFunctionReturn(0);
667930ae53cSBarry Smith   if (a->indexshift) {
6683a40ed3dSBarry Smith      ierr = MatSolve_SeqAIJ(A,bb,xx);CHKERRQ(ierr);
6693a40ed3dSBarry Smith      PetscFunctionReturn(0);
670930ae53cSBarry Smith   }
671930ae53cSBarry Smith 
672e1311b90SBarry Smith   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
673e1311b90SBarry Smith   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
674930ae53cSBarry Smith 
675aa482453SBarry Smith #if defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ)
6761c4feecaSSatish Balay   fortransolveaij_(&n,x,ai,aj,adiag,aa,b);
6771c4feecaSSatish Balay #else
678930ae53cSBarry Smith   /* forward solve the lower triangular */
679930ae53cSBarry Smith   x[0] = b[0];
680930ae53cSBarry Smith   for (i=1; i<n; i++) {
681930ae53cSBarry Smith     ai_i = ai[i];
682930ae53cSBarry Smith     v    = aa + ai_i;
683930ae53cSBarry Smith     vi   = aj + ai_i;
684930ae53cSBarry Smith     nz   = adiag[i] - ai_i;
685930ae53cSBarry Smith     sum  = b[i];
686930ae53cSBarry Smith     while (nz--) sum -= *v++ * x[*vi++];
687930ae53cSBarry Smith     x[i] = sum;
688930ae53cSBarry Smith   }
689930ae53cSBarry Smith 
690930ae53cSBarry Smith   /* backward solve the upper triangular */
691930ae53cSBarry Smith   for (i=n-1; i>=0; i--){
692930ae53cSBarry Smith     adiag_i = adiag[i];
693d708749eSBarry Smith     v       = aa + adiag_i + 1;
694d708749eSBarry Smith     vi      = aj + adiag_i + 1;
695930ae53cSBarry Smith     nz      = ai[i+1] - adiag_i - 1;
696930ae53cSBarry Smith     sum     = x[i];
697930ae53cSBarry Smith     while (nz--) sum -= *v++ * x[*vi++];
698930ae53cSBarry Smith     x[i]    = sum*aa[adiag_i];
699930ae53cSBarry Smith   }
7001c4feecaSSatish Balay #endif
701930ae53cSBarry Smith   PLogFlops(2*a->nz - a->n);
702cb5b572fSBarry Smith   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
703cb5b572fSBarry Smith   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
7043a40ed3dSBarry Smith   PetscFunctionReturn(0);
705930ae53cSBarry Smith }
706930ae53cSBarry Smith 
7075615d1e5SSatish Balay #undef __FUNC__
708b2863d3aSBarry Smith #define __FUNC__ /*<a name=""></a>*/"MatSolveAdd_SeqAIJ"
709416022c9SBarry Smith int MatSolveAdd_SeqAIJ(Mat A,Vec bb,Vec yy,Vec xx)
710da3a660dSBarry Smith {
711416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data;
712416022c9SBarry Smith   IS         iscol = a->col,isrow = a->row;
713416022c9SBarry Smith   int        *r,*c,ierr,i, n = a->m,*vi,*ai = a->i,*aj = a->j;
7143b2fbd54SBarry Smith   int        nz,shift = a->indexshift,*rout,*cout;
715416022c9SBarry Smith   Scalar     *x,*b,*tmp,*aa = a->a,sum,*v;
716da3a660dSBarry Smith 
7173a40ed3dSBarry Smith   PetscFunctionBegin;
71878b31e54SBarry Smith   if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);}
719da3a660dSBarry Smith 
720e1311b90SBarry Smith   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
721e1311b90SBarry Smith   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
722416022c9SBarry Smith   tmp  = a->solve_work;
723da3a660dSBarry Smith 
7243b2fbd54SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
7253b2fbd54SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
726da3a660dSBarry Smith 
727da3a660dSBarry Smith   /* forward solve the lower triangular */
728da3a660dSBarry Smith   tmp[0] = b[*r++];
729da3a660dSBarry Smith   for (i=1; i<n; i++) {
730dbb450caSBarry Smith     v   = aa + ai[i] + shift;
731dbb450caSBarry Smith     vi  = aj + ai[i] + shift;
732416022c9SBarry Smith     nz  = a->diag[i] - ai[i];
733da3a660dSBarry Smith     sum = b[*r++];
734dbb450caSBarry Smith     while (nz--) sum -= *v++ * tmp[*vi++ + shift];
735da3a660dSBarry Smith     tmp[i] = sum;
736da3a660dSBarry Smith   }
737da3a660dSBarry Smith 
738da3a660dSBarry Smith   /* backward solve the upper triangular */
739da3a660dSBarry Smith   for (i=n-1; i>=0; i--){
740416022c9SBarry Smith     v   = aa + a->diag[i] + (!shift);
741416022c9SBarry Smith     vi  = aj + a->diag[i] + (!shift);
742416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
743da3a660dSBarry Smith     sum = tmp[i];
744dbb450caSBarry Smith     while (nz--) sum -= *v++ * tmp[*vi++ + shift];
745416022c9SBarry Smith     tmp[i] = sum*aa[a->diag[i]+shift];
746da3a660dSBarry Smith     x[*c--] += tmp[i];
747da3a660dSBarry Smith   }
748da3a660dSBarry Smith 
7493b2fbd54SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
7503b2fbd54SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
751cb5b572fSBarry Smith   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
752cb5b572fSBarry Smith   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
753416022c9SBarry Smith   PLogFlops(2*a->nz);
754898183ecSLois Curfman McInnes 
7553a40ed3dSBarry Smith   PetscFunctionReturn(0);
756da3a660dSBarry Smith }
757da3a660dSBarry Smith /* -------------------------------------------------------------------*/
7585615d1e5SSatish Balay #undef __FUNC__
759b2863d3aSBarry Smith #define __FUNC__ /*<a name=""></a>*/"MatSolveTranspose_SeqAIJ"
7607c922b88SBarry Smith int MatSolveTranspose_SeqAIJ(Mat A,Vec bb,Vec xx)
761da3a660dSBarry Smith {
762416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data;
7632235e667SBarry Smith   IS         iscol = a->col,isrow = a->row;
764416022c9SBarry Smith   int        *r,*c,ierr,i,n = a->m,*vi,*ai = a->i,*aj = a->j;
765f1af5d2fSBarry Smith   int        nz,shift = a->indexshift,*rout,*cout,*diag = a->diag;
766f1af5d2fSBarry Smith   Scalar     *x,*b,*tmp,*aa = a->a,*v,s1;
767da3a660dSBarry Smith 
7683a40ed3dSBarry Smith   PetscFunctionBegin;
769e1311b90SBarry Smith   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
770e1311b90SBarry Smith   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
771416022c9SBarry Smith   tmp  = a->solve_work;
772da3a660dSBarry Smith 
7732235e667SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
7742235e667SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
775da3a660dSBarry Smith 
776da3a660dSBarry Smith   /* copy the b into temp work space according to permutation */
7772235e667SBarry Smith   for (i=0; i<n; i++) tmp[i] = b[c[i]];
778da3a660dSBarry Smith 
779da3a660dSBarry Smith   /* forward solve the U^T */
780da3a660dSBarry Smith   for (i=0; i<n; i++) {
781f1af5d2fSBarry Smith     v   = aa + diag[i] + shift;
782f1af5d2fSBarry Smith     vi  = aj + diag[i] + (!shift);
783f1af5d2fSBarry Smith     nz  = ai[i+1] - diag[i] - 1;
784c38d4ed2SBarry Smith     s1  = tmp[i];
785c38d4ed2SBarry Smith     s1 *= *(v++);  /* multiply by inverse of diagonal entry */
786da3a660dSBarry Smith     while (nz--) {
787f1af5d2fSBarry Smith       tmp[*vi++ + shift] -= (*v++)*s1;
788da3a660dSBarry Smith     }
789c38d4ed2SBarry Smith     tmp[i] = s1;
790da3a660dSBarry Smith   }
791da3a660dSBarry Smith 
792da3a660dSBarry Smith   /* backward solve the L^T */
793da3a660dSBarry Smith   for (i=n-1; i>=0; i--){
794f1af5d2fSBarry Smith     v   = aa + diag[i] - 1 + shift;
795f1af5d2fSBarry Smith     vi  = aj + diag[i] - 1 + shift;
796f1af5d2fSBarry Smith     nz  = diag[i] - ai[i];
797f1af5d2fSBarry Smith     s1  = tmp[i];
798da3a660dSBarry Smith     while (nz--) {
799f1af5d2fSBarry Smith       tmp[*vi-- + shift] -= (*v--)*s1;
800da3a660dSBarry Smith     }
801da3a660dSBarry Smith   }
802da3a660dSBarry Smith 
803da3a660dSBarry Smith   /* copy tmp into x according to permutation */
804da3a660dSBarry Smith   for (i=0; i<n; i++) x[r[i]] = tmp[i];
805da3a660dSBarry Smith 
8062235e667SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
8072235e667SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
808cb5b572fSBarry Smith   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
809cb5b572fSBarry Smith   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
810da3a660dSBarry Smith 
811416022c9SBarry Smith   PLogFlops(2*a->nz-a->n);
8123a40ed3dSBarry Smith   PetscFunctionReturn(0);
813da3a660dSBarry Smith }
814da3a660dSBarry Smith 
8155615d1e5SSatish Balay #undef __FUNC__
816b2863d3aSBarry Smith #define __FUNC__ /*<a name=""></a>*/"MatSolveTransposeAdd_SeqAIJ"
8177c922b88SBarry Smith int MatSolveTransposeAdd_SeqAIJ(Mat A,Vec bb,Vec zz,Vec xx)
818da3a660dSBarry Smith {
819416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data;
8202235e667SBarry Smith   IS         iscol = a->col,isrow = a->row;
821416022c9SBarry Smith   int        *r,*c,ierr,i,n = a->m,*vi,*ai = a->i,*aj = a->j;
822f1af5d2fSBarry Smith   int        nz,shift = a->indexshift,*rout,*cout,*diag = a->diag;
823416022c9SBarry Smith   Scalar     *x,*b,*tmp,*aa = a->a,*v;
8246abc6512SBarry Smith 
8253a40ed3dSBarry Smith   PetscFunctionBegin;
8266abc6512SBarry Smith   if (zz != xx) VecCopy(zz,xx);
8276abc6512SBarry Smith 
828e1311b90SBarry Smith   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
829e1311b90SBarry Smith   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
830416022c9SBarry Smith   tmp = a->solve_work;
8316abc6512SBarry Smith 
8322235e667SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
8332235e667SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
8346abc6512SBarry Smith 
8356abc6512SBarry Smith   /* copy the b into temp work space according to permutation */
8362235e667SBarry Smith   for (i=0; i<n; i++) tmp[i] = b[c[i]];
8376abc6512SBarry Smith 
8386abc6512SBarry Smith   /* forward solve the U^T */
8396abc6512SBarry Smith   for (i=0; i<n; i++) {
840f1af5d2fSBarry Smith     v   = aa + diag[i] + shift;
841f1af5d2fSBarry Smith     vi  = aj + diag[i] + (!shift);
842f1af5d2fSBarry Smith     nz  = ai[i+1] - diag[i] - 1;
8436abc6512SBarry Smith     tmp[i] *= *v++;
8446abc6512SBarry Smith     while (nz--) {
845dbb450caSBarry Smith       tmp[*vi++ + shift] -= (*v++)*tmp[i];
8466abc6512SBarry Smith     }
8476abc6512SBarry Smith   }
8486abc6512SBarry Smith 
8496abc6512SBarry Smith   /* backward solve the L^T */
8506abc6512SBarry Smith   for (i=n-1; i>=0; i--){
851f1af5d2fSBarry Smith     v   = aa + diag[i] - 1 + shift;
852f1af5d2fSBarry Smith     vi  = aj + diag[i] - 1 + shift;
853f1af5d2fSBarry Smith     nz  = diag[i] - ai[i];
8546abc6512SBarry Smith     while (nz--) {
855dbb450caSBarry Smith       tmp[*vi-- + shift] -= (*v--)*tmp[i];
8566abc6512SBarry Smith     }
8576abc6512SBarry Smith   }
8586abc6512SBarry Smith 
8596abc6512SBarry Smith   /* copy tmp into x according to permutation */
8606abc6512SBarry Smith   for (i=0; i<n; i++) x[r[i]] += tmp[i];
8616abc6512SBarry Smith 
8622235e667SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
8632235e667SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
864cb5b572fSBarry Smith   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
865cb5b572fSBarry Smith   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
8666abc6512SBarry Smith 
867416022c9SBarry Smith   PLogFlops(2*a->nz);
8683a40ed3dSBarry Smith   PetscFunctionReturn(0);
869da3a660dSBarry Smith }
8709e25ed09SBarry Smith /* ----------------------------------------------------------------*/
871ca44d042SBarry Smith EXTERN int MatMissingDiagonal_SeqAIJ(Mat);
87208480c60SBarry Smith 
8735615d1e5SSatish Balay #undef __FUNC__
874b2863d3aSBarry Smith #define __FUNC__ /*<a name=""></a>*/"MatILUFactorSymbolic_SeqAIJ"
8756cf997b0SBarry Smith int MatILUFactorSymbolic_SeqAIJ(Mat A,IS isrow,IS iscol,MatILUInfo *info,Mat *fact)
8769e25ed09SBarry Smith {
877416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ*)A->data,*b;
8789e25ed09SBarry Smith   IS         isicol;
879416022c9SBarry Smith   int        *r,*ic,ierr,prow,n = a->m,*ai = a->i,*aj = a->j;
88056beaf04SBarry Smith   int        *ainew,*ajnew,jmax,*fill,*xi,nz,*im,*ajfill,*flev;
881335d9088SBarry Smith   int        *dloc,idx,row,m,fm,nzf,nzi,len, realloc = 0,dcount = 0;
8826cf997b0SBarry Smith   int        incrlev,nnz,i,shift = a->indexshift,levels,diagonal_fill;
88377c4ece6SBarry Smith   PetscTruth col_identity,row_identity;
884329f5518SBarry Smith   PetscReal  f;
8859e25ed09SBarry Smith 
8863a40ed3dSBarry Smith   PetscFunctionBegin;
8876cf997b0SBarry Smith   if (info) {
8886cf997b0SBarry Smith     f             = info->fill;
8890cd17293SBarry Smith     levels        = (int)info->levels;
8900cd17293SBarry Smith     diagonal_fill = (int)info->diagonal_fill;
8916cf997b0SBarry Smith   } else {
8926cf997b0SBarry Smith     f             = 1.0;
8936cf997b0SBarry Smith     levels        = 0;
8946cf997b0SBarry Smith     diagonal_fill = 0;
8956cf997b0SBarry Smith   }
8964da77479SBarry Smith   if (!isrow) {
8974da77479SBarry Smith     ierr = ISCreateStride(PETSC_COMM_SELF,A->M,0,1,&isrow);CHKERRQ(ierr);
8984da77479SBarry Smith   }
8994da77479SBarry Smith   if (!iscol) {
9004da77479SBarry Smith     ierr = ISCreateStride(PETSC_COMM_SELF,A->M,0,1,&iscol);CHKERRQ(ierr);
9014da77479SBarry Smith   }
9024da77479SBarry Smith 
9034c49b128SBarry Smith   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
90482bf6240SBarry Smith 
90508480c60SBarry Smith   /* special case that simply copies fill pattern */
906be0abb6dSBarry Smith   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
907be0abb6dSBarry Smith   ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr);
90886bacbd2SBarry Smith   if (!levels && row_identity && col_identity) {
9092e8a6d31SBarry Smith     ierr = MatDuplicate_SeqAIJ(A,MAT_DO_NOT_COPY_VALUES,fact);CHKERRQ(ierr);
91008480c60SBarry Smith     (*fact)->factor = FACTOR_LU;
91108480c60SBarry Smith     b               = (Mat_SeqAIJ*)(*fact)->data;
91208480c60SBarry Smith     if (!b->diag) {
9137c922b88SBarry Smith       ierr = MatMarkDiagonal_SeqAIJ(*fact);CHKERRQ(ierr);
91408480c60SBarry Smith     }
9157c922b88SBarry Smith     ierr = MatMissingDiagonal_SeqAIJ(*fact);CHKERRQ(ierr);
91608480c60SBarry Smith     b->row              = isrow;
91708480c60SBarry Smith     b->col              = iscol;
91882bf6240SBarry Smith     b->icol             = isicol;
9190452661fSBarry Smith     b->solve_work       = (Scalar*)PetscMalloc((b->m+1)*sizeof(Scalar));CHKPTRQ(b->solve_work);
920f830108cSBarry Smith     (*fact)->ops->solve = MatSolve_SeqAIJ_NaturalOrdering;
921c38d4ed2SBarry Smith     ierr                = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
922c38d4ed2SBarry Smith     ierr                = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
9233a40ed3dSBarry Smith     PetscFunctionReturn(0);
92408480c60SBarry Smith   }
92508480c60SBarry Smith 
926898183ecSLois Curfman McInnes   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
927898183ecSLois Curfman McInnes   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
9289e25ed09SBarry Smith 
9299e25ed09SBarry Smith   /* get new row pointers */
9300452661fSBarry Smith   ainew = (int*)PetscMalloc((n+1)*sizeof(int));CHKPTRQ(ainew);
931dbb450caSBarry Smith   ainew[0] = -shift;
9329e25ed09SBarry Smith   /* don't know how many column pointers are needed so estimate */
933dbb450caSBarry Smith   jmax = (int)(f*(ai[n]+!shift));
9340452661fSBarry Smith   ajnew = (int*)PetscMalloc((jmax)*sizeof(int));CHKPTRQ(ajnew);
9359e25ed09SBarry Smith   /* ajfill is level of fill for each fill entry */
9360452661fSBarry Smith   ajfill = (int*)PetscMalloc((jmax)*sizeof(int));CHKPTRQ(ajfill);
9379e25ed09SBarry Smith   /* fill is a linked list of nonzeros in active row */
9380452661fSBarry Smith   fill = (int*)PetscMalloc((n+1)*sizeof(int));CHKPTRQ(fill);
93956beaf04SBarry Smith   /* im is level for each filled value */
9400452661fSBarry Smith   im = (int*)PetscMalloc((n+1)*sizeof(int));CHKPTRQ(im);
94156beaf04SBarry Smith   /* dloc is location of diagonal in factor */
9420452661fSBarry Smith   dloc = (int*)PetscMalloc((n+1)*sizeof(int));CHKPTRQ(dloc);
94356beaf04SBarry Smith   dloc[0]  = 0;
94456beaf04SBarry Smith   for (prow=0; prow<n; prow++) {
9456cf997b0SBarry Smith 
9466cf997b0SBarry Smith     /* copy current row into linked list */
94756beaf04SBarry Smith     nzf     = nz  = ai[r[prow]+1] - ai[r[prow]];
948385f7a74SSatish Balay     if (!nz) SETERRQ(PETSC_ERR_MAT_LU_ZRPVT,1,"Empty row in matrix");
949dbb450caSBarry Smith     xi      = aj + ai[r[prow]] + shift;
9509e25ed09SBarry Smith     fill[n]    = n;
951435faa5fSBarry Smith     fill[prow] = -1; /* marker to indicate if diagonal exists */
9529e25ed09SBarry Smith     while (nz--) {
9539e25ed09SBarry Smith       fm  = n;
954dbb450caSBarry Smith       idx = ic[*xi++ + shift];
9559e25ed09SBarry Smith       do {
9569e25ed09SBarry Smith         m  = fm;
9579e25ed09SBarry Smith         fm = fill[m];
9589e25ed09SBarry Smith       } while (fm < idx);
9599e25ed09SBarry Smith       fill[m]   = idx;
9609e25ed09SBarry Smith       fill[idx] = fm;
96156beaf04SBarry Smith       im[idx]   = 0;
9629e25ed09SBarry Smith     }
9636cf997b0SBarry Smith 
9646cf997b0SBarry Smith     /* make sure diagonal entry is included */
965435faa5fSBarry Smith     if (diagonal_fill && fill[prow] == -1) {
9666cf997b0SBarry Smith       fm = n;
967435faa5fSBarry Smith       while (fill[fm] < prow) fm = fill[fm];
968435faa5fSBarry Smith       fill[prow] = fill[fm]; /* insert diagonal into linked list */
969435faa5fSBarry Smith       fill[fm]   = prow;
9706cf997b0SBarry Smith       im[prow]   = 0;
9716cf997b0SBarry Smith       nzf++;
972335d9088SBarry Smith       dcount++;
9736cf997b0SBarry Smith     }
9746cf997b0SBarry Smith 
97556beaf04SBarry Smith     nzi = 0;
9769e25ed09SBarry Smith     row = fill[n];
97756beaf04SBarry Smith     while (row < prow) {
97856beaf04SBarry Smith       incrlev = im[row] + 1;
97956beaf04SBarry Smith       nz      = dloc[row];
9806cf997b0SBarry Smith       xi      = ajnew  + ainew[row] + shift + nz + 1;
981dbb450caSBarry Smith       flev    = ajfill + ainew[row] + shift + nz + 1;
982416022c9SBarry Smith       nnz     = ainew[row+1] - ainew[row] - nz - 1;
98356beaf04SBarry Smith       fm      = row;
98456beaf04SBarry Smith       while (nnz-- > 0) {
985dbb450caSBarry Smith         idx = *xi++ + shift;
98656beaf04SBarry Smith         if (*flev + incrlev > levels) {
98756beaf04SBarry Smith           flev++;
98856beaf04SBarry Smith           continue;
98956beaf04SBarry Smith         }
99056beaf04SBarry Smith         do {
9919e25ed09SBarry Smith           m  = fm;
9929e25ed09SBarry Smith           fm = fill[m];
99356beaf04SBarry Smith         } while (fm < idx);
9949e25ed09SBarry Smith         if (fm != idx) {
99556beaf04SBarry Smith           im[idx]   = *flev + incrlev;
9969e25ed09SBarry Smith           fill[m]   = idx;
9979e25ed09SBarry Smith           fill[idx] = fm;
9989e25ed09SBarry Smith           fm        = idx;
99956beaf04SBarry Smith           nzf++;
1000ecf371e4SBarry Smith         } else {
100156beaf04SBarry Smith           if (im[idx] > *flev + incrlev) im[idx] = *flev+incrlev;
10029e25ed09SBarry Smith         }
100356beaf04SBarry Smith         flev++;
10049e25ed09SBarry Smith       }
10059e25ed09SBarry Smith       row = fill[row];
100656beaf04SBarry Smith       nzi++;
10079e25ed09SBarry Smith     }
10089e25ed09SBarry Smith     /* copy new filled row into permanent storage */
100956beaf04SBarry Smith     ainew[prow+1] = ainew[prow] + nzf;
1010d7e8b826SBarry Smith     if (ainew[prow+1] > jmax-shift) {
1011ecf371e4SBarry Smith 
1012ecf371e4SBarry Smith       /* estimate how much additional space we will need */
1013ecf371e4SBarry Smith       /* use the strategy suggested by David Hysom <hysom@perch-t.icase.edu> */
1014ecf371e4SBarry Smith       /* just double the memory each time */
1015ecf371e4SBarry Smith       /*  maxadd = (int)((f*(ai[n]+!shift)*(n-prow+5))/n); */
1016ecf371e4SBarry Smith       int maxadd = jmax;
101756beaf04SBarry Smith       if (maxadd < nzf) maxadd = (n-prow)*(nzf+1);
1018bbb6d6a8SBarry Smith       jmax += maxadd;
1019ecf371e4SBarry Smith 
1020ecf371e4SBarry Smith       /* allocate a longer ajnew and ajfill */
10210452661fSBarry Smith       xi     = (int*)PetscMalloc(jmax*sizeof(int));CHKPTRQ(xi);
1022549d3d68SSatish Balay       ierr   = PetscMemcpy(xi,ajnew,(ainew[prow]+shift)*sizeof(int));CHKERRQ(ierr);
1023606d414cSSatish Balay       ierr = PetscFree(ajnew);CHKERRQ(ierr);
102456beaf04SBarry Smith       ajnew  = xi;
10250452661fSBarry Smith       xi     = (int*)PetscMalloc(jmax*sizeof(int));CHKPTRQ(xi);
1026549d3d68SSatish Balay       ierr   = PetscMemcpy(xi,ajfill,(ainew[prow]+shift)*sizeof(int));CHKERRQ(ierr);
1027606d414cSSatish Balay       ierr = PetscFree(ajfill);CHKERRQ(ierr);
102856beaf04SBarry Smith       ajfill = xi;
1029ecf371e4SBarry Smith       realloc++; /* count how many times we realloc */
10309e25ed09SBarry Smith     }
1031dbb450caSBarry Smith     xi          = ajnew + ainew[prow] + shift;
1032dbb450caSBarry Smith     flev        = ajfill + ainew[prow] + shift;
103356beaf04SBarry Smith     dloc[prow]  = nzi;
10349e25ed09SBarry Smith     fm          = fill[n];
103556beaf04SBarry Smith     while (nzf--) {
1036dbb450caSBarry Smith       *xi++   = fm - shift;
103756beaf04SBarry Smith       *flev++ = im[fm];
10389e25ed09SBarry Smith       fm      = fill[fm];
10399e25ed09SBarry Smith     }
10406cf997b0SBarry Smith     /* make sure row has diagonal entry */
10416cf997b0SBarry Smith     if (ajnew[ainew[prow]+shift+dloc[prow]]+shift != prow) {
10426cf997b0SBarry Smith       SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,1,"Row %d has missing diagonal in factored matrix\n\
10436cf997b0SBarry Smith     try running with -pc_ilu_nonzeros_along_diagonal or -pc_ilu_diagonal_fill",prow);
10446cf997b0SBarry Smith     }
10459e25ed09SBarry Smith   }
1046606d414cSSatish Balay   ierr = PetscFree(ajfill);CHKERRQ(ierr);
1047898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
1048898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
1049606d414cSSatish Balay   ierr = PetscFree(fill);CHKERRQ(ierr);
1050606d414cSSatish Balay   ierr = PetscFree(im);CHKERRQ(ierr);
10519e25ed09SBarry Smith 
1052f64065bbSBarry Smith   {
1053329f5518SBarry Smith     PetscReal af = ((PetscReal)ainew[n])/((PetscReal)ai[n]);
1054c38d4ed2SBarry Smith     PLogInfo(A,"MatILUFactorSymbolic_SeqAIJ:Reallocs %d Fill ratio:given %g needed %g\n",realloc,f,af);
1055981c4779SBarry Smith     PLogInfo(A,"MatILUFactorSymbolic_SeqAIJ:Run with -pc_ilu_fill %g or use \n",af);
1056981c4779SBarry Smith     PLogInfo(A,"MatILUFactorSymbolic_SeqAIJ:PCILUSetFill(pc,%g);\n",af);
1057981c4779SBarry Smith     PLogInfo(A,"MatILUFactorSymbolic_SeqAIJ:for best performance.\n");
1058335d9088SBarry Smith     if (diagonal_fill) {
1059335d9088SBarry Smith       PLogInfo(A,"MatILUFactorSymbolic_SeqAIJ:Detected and replace %d missing diagonals",dcount);
1060335d9088SBarry Smith     }
1061f64065bbSBarry Smith   }
1062bbb6d6a8SBarry Smith 
10639e25ed09SBarry Smith   /* put together the new matrix */
1064b4fd4287SBarry Smith   ierr = MatCreateSeqAIJ(A->comm,n,n,0,PETSC_NULL,fact);CHKERRQ(ierr);
1065fa6007c9SSatish Balay   PLogObjectParent(*fact,isicol);
1066416022c9SBarry Smith   b = (Mat_SeqAIJ*)(*fact)->data;
1067606d414cSSatish Balay   ierr = PetscFree(b->imax);CHKERRQ(ierr);
10687c922b88SBarry Smith   b->singlemalloc = PETSC_FALSE;
1069dbb450caSBarry Smith   len = (ainew[n] + shift)*sizeof(Scalar);
10709e25ed09SBarry Smith   /* the next line frees the default space generated by the Create() */
1071606d414cSSatish Balay   ierr = PetscFree(b->a);CHKERRQ(ierr);
1072606d414cSSatish Balay   ierr = PetscFree(b->ilen);CHKERRQ(ierr);
10736b96ef82SSatish Balay   b->a          = (Scalar*)PetscMalloc(len+1);CHKPTRQ(b->a);
1074416022c9SBarry Smith   b->j          = ajnew;
1075416022c9SBarry Smith   b->i          = ainew;
107656beaf04SBarry Smith   for (i=0; i<n; i++) dloc[i] += ainew[i];
1077416022c9SBarry Smith   b->diag       = dloc;
1078416022c9SBarry Smith   b->ilen       = 0;
1079416022c9SBarry Smith   b->imax       = 0;
1080416022c9SBarry Smith   b->row        = isrow;
1081416022c9SBarry Smith   b->col        = iscol;
1082c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
1083c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
108482bf6240SBarry Smith   b->icol       = isicol;
108582bf6240SBarry Smith   b->solve_work = (Scalar*)PetscMalloc((n+1)*sizeof(Scalar));CHKPTRQ(b->solve_work);
1086416022c9SBarry Smith   /* In b structure:  Free imax, ilen, old a, old j.
108756beaf04SBarry Smith      Allocate dloc, solve_work, new a, new j */
1088dbb450caSBarry Smith   PLogObjectMemory(*fact,(ainew[n]+shift-n) * (sizeof(int)+sizeof(Scalar)));
1089416022c9SBarry Smith   b->maxnz          = b->nz = ainew[n] + shift;
10909e25ed09SBarry Smith   (*fact)->factor   = FACTOR_LU;
1091ae068f58SLois Curfman McInnes 
1092ae068f58SLois Curfman McInnes   (*fact)->info.factor_mallocs    = realloc;
1093ae068f58SLois Curfman McInnes   (*fact)->info.fill_ratio_given  = f;
1094329f5518SBarry Smith   (*fact)->info.fill_ratio_needed = ((PetscReal)ainew[n])/((PetscReal)ai[prow]);
1095329f5518SBarry Smith   (*fact)->factor                 =  FACTOR_LU;
1096ae068f58SLois Curfman McInnes 
10973a40ed3dSBarry Smith   PetscFunctionReturn(0);
10989e25ed09SBarry Smith }
1099d5d45c9bSBarry Smith 
1100d5d45c9bSBarry Smith 
1101d5d45c9bSBarry Smith 
1102d5d45c9bSBarry Smith 
1103