xref: /petsc/src/mat/impls/aij/seq/aijfact.c (revision 1f9e874a3bdfbe6ede17348c2aef342cadaa5746)
1*1f9e874aSBarry Smith /*$Id: aijfact.c,v 1.135 1999/12/16 23:32:24 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 
2386bacbd2SBarry Smith #if !defined(PETSC_USE_COMPLEX) && defined(PETSC_HAVE_SAADILUDT)
2486bacbd2SBarry Smith 
2586bacbd2SBarry Smith #if defined(PETSC_HAVE_FORTRAN_CAPS)
2686bacbd2SBarry Smith #define dperm_  DPERM
2786bacbd2SBarry Smith #define ilutp_  ILUTP
2886bacbd2SBarry Smith #define ilu0_   ILU0
2986bacbd2SBarry Smith #define msrcsr_ MSRCSR
3086bacbd2SBarry Smith #elif !defined(PETSC_HAVE_FORTRAN_UNDERSCORE)
3186bacbd2SBarry Smith #define dperm_  dperm
3286bacbd2SBarry Smith #define ilutp_  ilutp
3386bacbd2SBarry Smith #define ilu0_   ilu0
3486bacbd2SBarry Smith #define msrcsr_ msrcsr
3586bacbd2SBarry Smith #endif
3686bacbd2SBarry Smith 
3786bacbd2SBarry Smith EXTERN_C_BEGIN
3886bacbd2SBarry Smith extern void dperm_(int*,double*,int*,int*,double*,int*,int*,int*,int*,int*);
3986bacbd2SBarry Smith extern void ilutp_(int*,double*,int*,int*,int*,double*,double*,int*,double*,int*,int*,int*,double*,int*,int*,int*);
4086bacbd2SBarry Smith extern void msrcsr_(int*,double*,int*,double*,int*,int*,double*,int*);
4186bacbd2SBarry Smith extern void ilu0_(int*,double*,int*,int*,double*,int*,int*,int*,int *);
4286bacbd2SBarry Smith EXTERN_C_END
4386bacbd2SBarry Smith 
4486bacbd2SBarry Smith #undef __FUNC__
4586bacbd2SBarry Smith #define __FUNC__ "MatILUDTFactor_SeqAIJ"
4686bacbd2SBarry Smith   /* ------------------------------------------------------------
4786bacbd2SBarry Smith 
4886bacbd2SBarry Smith           This interface was contribed by Tony Caola
4986bacbd2SBarry Smith 
5086bacbd2SBarry Smith      This routine is an interface to the pivoting drop-tolerance
5186bacbd2SBarry Smith      ILU routine written by Yousef Saad (saad@cs.umn.edu).  It
5286bacbd2SBarry Smith      was inspired by the non-pivoting iludt written by David
5386bacbd2SBarry Smith      Hysom (hysom@cs.odu.edu).
5486bacbd2SBarry Smith 
5586bacbd2SBarry Smith      Thanks to Prof. Saad, Dr. Hysom, and Dr. Smith for their
5686bacbd2SBarry Smith      help in getting this routine ironed out.
5786bacbd2SBarry Smith 
5886bacbd2SBarry Smith      As of right now, there are a couple of things that could
5986bacbd2SBarry Smith      be, uh, better.
6086bacbd2SBarry Smith 
6186bacbd2SBarry Smith      1 - Since Saad's routine is Fortran based, memory cannot be
6286bacbd2SBarry Smith      malloc'd.  I was trying to get the expected fill from the
6386bacbd2SBarry Smith      preconditioner and use this number as the multiplier in
6486bacbd2SBarry Smith      the equation for jmax, below, but couldn't figure it out.
6586bacbd2SBarry Smith      Anyway, perhaps a better solution is to run SPARSKIT through
6686bacbd2SBarry Smith      f2c and incorporate mallocs(), but I want to graduate so I'll
6786bacbd2SBarry Smith      just rebuild Petsc. . .
6886bacbd2SBarry Smith 
6986bacbd2SBarry Smith      shift = 1, ishift = 0, for indices start at 1
7086bacbd2SBarry Smith      shift = 0, ishift = 1, for indices starting at 0
7186bacbd2SBarry Smith      ------------------------------------------------------------
7286bacbd2SBarry Smith   */
7386bacbd2SBarry Smith 
7486bacbd2SBarry Smith int MatILUDTFactor_SeqAIJ(Mat A,MatILUInfo *info,IS isrow,IS iscol,Mat *fact)
7586bacbd2SBarry Smith {
7686bacbd2SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data, *b;
7707d2056aSBarry Smith   IS         iscolf, isicol, isirow;
7886bacbd2SBarry Smith   PetscTruth reorder;
7986bacbd2SBarry Smith   int        *c,*r,*ic,*ir, ierr, i, n = a->m;
8086bacbd2SBarry Smith   int        *old_i = a->i, *old_j = a->j, *new_i, *old_i2, *old_j2,*new_j;
81313ae024SBarry Smith   int        *ordcol, *iwk,*iperm, *jw;
8286bacbd2SBarry Smith   int        ishift = !a->indexshift,shift = -a->indexshift;
83b19713cbSBarry Smith   int        jmax,lfill,job,*o_i,*o_j;
84b19713cbSBarry Smith   Scalar     *old_a = a->a, *w, *new_a, *old_a2, *wk,permtol=0.0,*o_a;
8586bacbd2SBarry Smith 
8686bacbd2SBarry Smith   PetscFunctionBegin;
8786bacbd2SBarry Smith 
8886bacbd2SBarry Smith   if (info->dt == PETSC_DEFAULT)      info->dt      = .005;
8986bacbd2SBarry Smith   if (info->dtcount == PETSC_DEFAULT) info->dtcount = (int) (1.5*a->rmax);
9086bacbd2SBarry Smith   if (info->dtcol == PETSC_DEFAULT)   info->dtcol   = .01;
9186bacbd2SBarry Smith   if (info->fill == PETSC_DEFAULT)    info->fill    = (n*info->dtcount)/a->nz;
9286bacbd2SBarry Smith   lfill   = info->dtcount/2;
9386bacbd2SBarry Smith   jmax    = info->fill*a->nz;
9486bacbd2SBarry Smith   permtol = info->dtcol;
9586bacbd2SBarry Smith 
9686bacbd2SBarry Smith 
9786bacbd2SBarry Smith   /* ------------------------------------------------------------
9886bacbd2SBarry Smith      If reorder=.TRUE., then the original matrix has to be
9986bacbd2SBarry Smith      reordered to reflect the user selected ordering scheme, and
10086bacbd2SBarry Smith      then de-reordered so it is in it's original format.
10186bacbd2SBarry Smith      Because Saad's dperm() is NOT in place, we have to copy
10286bacbd2SBarry Smith      the original matrix and allocate more storage. . .
10386bacbd2SBarry Smith      ------------------------------------------------------------
10486bacbd2SBarry Smith   */
10586bacbd2SBarry Smith 
10686bacbd2SBarry Smith   /* set reorder to true if either isrow or iscol is not identity */
10786bacbd2SBarry Smith   ierr = ISIdentity(isrow,&reorder);CHKERRQ(ierr);
10886bacbd2SBarry Smith   if (reorder) {ierr = ISIdentity(iscol,&reorder);CHKERRQ(ierr);}
10986bacbd2SBarry Smith   reorder = PetscNot(reorder);
11086bacbd2SBarry Smith 
11186bacbd2SBarry Smith 
11286bacbd2SBarry Smith   /* storage for ilu factor */
11386bacbd2SBarry Smith   new_i = (int *)    PetscMalloc((n+1)*sizeof(int));   CHKPTRQ(new_i);
11486bacbd2SBarry Smith   new_j = (int *)    PetscMalloc(jmax*sizeof(int));    CHKPTRQ(new_j);
11586bacbd2SBarry Smith   new_a = (Scalar *) PetscMalloc(jmax*sizeof(Scalar)); CHKPTRQ(new_a);
11686bacbd2SBarry Smith 
11786bacbd2SBarry Smith   ordcol = (int *) PetscMalloc(n*sizeof(int)); CHKPTRQ(ordcol);
11886bacbd2SBarry Smith 
11986bacbd2SBarry Smith   /* ------------------------------------------------------------
12086bacbd2SBarry Smith      Make sure that everything is Fortran formatted (1-Based)
12186bacbd2SBarry Smith      ------------------------------------------------------------
12286bacbd2SBarry Smith   */
123b19713cbSBarry Smith   if (ishift) {
124b19713cbSBarry Smith     for (i=old_i[0];i<old_i[n];i++) {
125b19713cbSBarry Smith       old_j[i]++;
12686bacbd2SBarry Smith     }
127b19713cbSBarry Smith     for(i=0;i<n+1;i++) {
128b19713cbSBarry Smith       old_i[i]++;
129b19713cbSBarry Smith     };
13086bacbd2SBarry Smith   };
13186bacbd2SBarry Smith 
13286bacbd2SBarry Smith   if (reorder) {
133c0c2c81eSBarry Smith     ierr = ISGetIndices(iscol,&c);           CHKERRQ(ierr);
134c0c2c81eSBarry Smith     ierr = ISGetIndices(isrow,&r);           CHKERRQ(ierr);
135c0c2c81eSBarry Smith     for(i=0;i<n;i++) {
136c0c2c81eSBarry Smith       r[i]  = r[i]+1;
137c0c2c81eSBarry Smith       c[i]  = c[i]+1;
138c0c2c81eSBarry Smith     }
139313ae024SBarry Smith     old_i2 = (int *) PetscMalloc((n+1)*sizeof(int)); CHKPTRQ(old_i2);
140313ae024SBarry Smith     old_j2 = (int *) PetscMalloc((old_i[n]-old_i[0]+1)*sizeof(int)); CHKPTRQ(old_j2);
141313ae024SBarry Smith     old_a2 = (Scalar *) PetscMalloc((old_i[n]-old_i[0]+1)*sizeof(Scalar));CHKPTRQ(old_a2);
142313ae024SBarry Smith     job = 3; dperm_(&n,old_a,old_j,old_i,old_a2,old_j2,old_i2,r,c,&job);
143c0c2c81eSBarry Smith     for (i=0;i<n;i++) {
144c0c2c81eSBarry Smith       r[i]  = r[i]-1;
145c0c2c81eSBarry Smith       c[i]  = c[i]-1;
146c0c2c81eSBarry Smith     }
147c0c2c81eSBarry Smith     ierr = ISRestoreIndices(iscol,&c); CHKERRQ(ierr);
148c0c2c81eSBarry Smith     ierr = ISRestoreIndices(isrow,&r); CHKERRQ(ierr);
149b19713cbSBarry Smith     o_a = old_a2;
150b19713cbSBarry Smith     o_j = old_j2;
151b19713cbSBarry Smith     o_i = old_i2;
152b19713cbSBarry Smith   } else {
153b19713cbSBarry Smith     o_a = old_a;
154b19713cbSBarry Smith     o_j = old_j;
155b19713cbSBarry Smith     o_i = old_i;
15686bacbd2SBarry Smith   }
15786bacbd2SBarry Smith 
15886bacbd2SBarry Smith   /* ------------------------------------------------------------
15986bacbd2SBarry Smith      Call Saad's ilutp() routine to generate the factorization
16086bacbd2SBarry Smith      ------------------------------------------------------------
16186bacbd2SBarry Smith   */
16286bacbd2SBarry Smith 
16386bacbd2SBarry Smith   iperm   = (int *)    PetscMalloc(2*n*sizeof(int)); CHKPTRQ(iperm);
16486bacbd2SBarry Smith   jw      = (int *)    PetscMalloc(2*n*sizeof(int)); CHKPTRQ(jw);
16586bacbd2SBarry Smith   w       = (Scalar *) PetscMalloc(n*sizeof(Scalar)); CHKPTRQ(w);
16686bacbd2SBarry Smith 
167b19713cbSBarry Smith   ilutp_(&n,o_a,o_j,o_i,&lfill,&info->dt,&permtol,&n,new_a,new_j,new_i,&jmax,w,jw,iperm,&ierr);
16886bacbd2SBarry Smith   if (ierr) {
16986bacbd2SBarry Smith     switch (ierr) {
17086bacbd2SBarry Smith       case -3: SETERRQ1(1,1,"ilutp(), matrix U overflows, need larger info->fill value %d",jmax);
17186bacbd2SBarry Smith                break;
17286bacbd2SBarry Smith       case -2: SETERRQ1(1,1,"ilutp(), matrix L overflows, need larger info->fill value %d",jmax);
17386bacbd2SBarry Smith                break;
17486bacbd2SBarry Smith       case -5: SETERRQ(1,1,"ilutp(), zero row encountered");
17586bacbd2SBarry Smith                break;
17686bacbd2SBarry Smith       case -1: SETERRQ(1,1,"ilutp(), input matrix may be wrong");
17786bacbd2SBarry Smith                break;
17886bacbd2SBarry Smith       case -4: SETERRQ1(1,1,"ilutp(), illegal info->fill value %d",jmax);
17986bacbd2SBarry Smith                break;
18086bacbd2SBarry Smith       default: SETERRQ1(1,1,"ilutp(), zero pivot detected on row %d",ierr);
18186bacbd2SBarry Smith     }
18286bacbd2SBarry Smith   }
18386bacbd2SBarry Smith 
18486bacbd2SBarry Smith   ierr = PetscFree(w);CHKERRQ(ierr);
18586bacbd2SBarry Smith   ierr = PetscFree(jw);CHKERRQ(ierr);
18686bacbd2SBarry Smith 
18786bacbd2SBarry Smith   /* ------------------------------------------------------------
18886bacbd2SBarry Smith      Saad's routine gives the result in Modified Sparse Row (msr)
18986bacbd2SBarry Smith      Convert to Compressed Sparse Row format (csr)
19086bacbd2SBarry Smith      ------------------------------------------------------------
19186bacbd2SBarry Smith   */
19286bacbd2SBarry Smith 
19386bacbd2SBarry Smith   wk  = (Scalar *)    PetscMalloc(n*sizeof(Scalar)); CHKPTRQ(wk);
19486bacbd2SBarry Smith   iwk = (int *) PetscMalloc((n+1)*sizeof(int)); CHKPTRQ(iwk);
19586bacbd2SBarry Smith 
19686bacbd2SBarry Smith   msrcsr_(&n,new_a,new_j,new_a,new_j,new_i,wk,iwk);
19786bacbd2SBarry Smith 
19886bacbd2SBarry Smith   ierr = PetscFree(iwk);CHKERRQ(ierr);
19986bacbd2SBarry Smith   ierr = PetscFree(wk);CHKERRQ(ierr);
20086bacbd2SBarry Smith 
20186bacbd2SBarry Smith   if (reorder) {
20286bacbd2SBarry Smith     ierr = PetscFree(old_a2);CHKERRQ(ierr);
20386bacbd2SBarry Smith     ierr = PetscFree(old_j2);CHKERRQ(ierr);
20486bacbd2SBarry Smith     ierr = PetscFree(old_i2);CHKERRQ(ierr);
205313ae024SBarry Smith   } else {
206b19713cbSBarry Smith     /* fix permutation of old_j that the factorization introduced */
207b19713cbSBarry Smith     for (i=old_i[0]; i<=old_i[n]; i++) {
208b19713cbSBarry Smith       old_j[i-1] = iperm[old_j[i-1]-1];
209b19713cbSBarry Smith     }
210b19713cbSBarry Smith   }
211b19713cbSBarry Smith 
212b801d0f9SBarry Smith   /* get rid of the shift to indices starting at 1 */
213b19713cbSBarry Smith   if (ishift) {
21486bacbd2SBarry Smith     for (i=0; i<n+1; i++) {
215b19713cbSBarry Smith       old_i[i]--;
216b19713cbSBarry Smith     }
217b19713cbSBarry Smith     for (i=old_i[0];i<old_i[n];i++) {
218b19713cbSBarry Smith       old_j[i]--;
219b19713cbSBarry Smith     }
22086bacbd2SBarry Smith   }
22186bacbd2SBarry Smith 
222b19713cbSBarry Smith   /* Make the factored matrix 0-based */
223b19713cbSBarry Smith   if (ishift) {
22486bacbd2SBarry Smith     for (i=0; i<n+1; i++) {
225b19713cbSBarry Smith       new_i[i]--;
226b19713cbSBarry Smith     }
227b19713cbSBarry Smith     for (i=new_i[0];i<new_i[n];i++) {
228b19713cbSBarry Smith       new_j[i]--;
229b19713cbSBarry Smith     }
23086bacbd2SBarry Smith   }
23186bacbd2SBarry Smith 
23286bacbd2SBarry Smith   /*-- due to the pivoting, we need to reorder iscol to correctly --*/
23386bacbd2SBarry Smith   /*-- permute the right-hand-side and solution vectors           --*/
234a3a5ab83SBarry Smith   ierr = ISInvertPermutation(iscol,&isicol); CHKERRQ(ierr);
235a3a5ab83SBarry Smith   ierr = ISInvertPermutation(isrow,&isirow); CHKERRQ(ierr);
236c0c2c81eSBarry Smith   ierr = ISGetIndices(isicol,&ic);          CHKERRQ(ierr);
23786bacbd2SBarry Smith   for(i=0; i<n; i++) {
23886bacbd2SBarry Smith     ordcol[i] = ic[iperm[i]-1];
23986bacbd2SBarry Smith   };
24086bacbd2SBarry Smith   ierr = ISRestoreIndices(isicol,&ic); CHKERRQ(ierr);
24107d2056aSBarry Smith   ierr = ISDestroy(isicol);CHKERRQ(ierr);
24286bacbd2SBarry Smith 
243c0c2c81eSBarry Smith   ierr = PetscFree(iperm);CHKERRQ(ierr);
244c0c2c81eSBarry Smith 
24586bacbd2SBarry Smith   ierr = ISCreateGeneral(PETSC_COMM_SELF, n, ordcol, &iscolf);
24607d2056aSBarry Smith   ierr = PetscFree(ordcol);CHKERRQ(ierr);
24786bacbd2SBarry Smith 
24886bacbd2SBarry Smith   /*----- put together the new matrix -----*/
24986bacbd2SBarry Smith 
25086bacbd2SBarry Smith   ierr = MatCreateSeqAIJ(A->comm,n,n,0,PETSC_NULL,fact); CHKERRQ(ierr);
25186bacbd2SBarry Smith   (*fact)->factor    = FACTOR_LU;
25286bacbd2SBarry Smith   (*fact)->assembled = PETSC_TRUE;
25386bacbd2SBarry Smith 
25486bacbd2SBarry Smith   b = (Mat_SeqAIJ *) (*fact)->data;
25586bacbd2SBarry Smith   ierr = PetscFree(b->imax);CHKERRQ(ierr);
25686bacbd2SBarry Smith   b->sorted        = PETSC_FALSE;
25707d2056aSBarry Smith   b->singlemalloc  = PETSC_FALSE;
258b19713cbSBarry Smith   /* the next line frees the default space generated by the MatCreate() */
25986bacbd2SBarry Smith   ierr             = PetscFree(b->a);CHKERRQ(ierr);
26086bacbd2SBarry Smith   ierr             = PetscFree(b->ilen);CHKERRQ(ierr);
26186bacbd2SBarry Smith   b->a             = new_a;
26286bacbd2SBarry Smith   b->j             = new_j;
26386bacbd2SBarry Smith   b->i             = new_i;
26486bacbd2SBarry Smith   b->ilen          = 0;
26586bacbd2SBarry Smith   b->imax          = 0;
266*1f9e874aSBarry Smith   /*  I am not sure why these are the inverses of the row and column permutations; but the other way is NO GOOD */
267313ae024SBarry Smith   b->row           = isirow;
26886bacbd2SBarry Smith   b->col           = iscolf;
26986bacbd2SBarry Smith   b->solve_work    =  (Scalar *) PetscMalloc( (n+1)*sizeof(Scalar));CHKPTRQ(b->solve_work);
27086bacbd2SBarry Smith   b->maxnz = b->nz = new_i[n];
27186bacbd2SBarry Smith   b->indexshift    = a->indexshift;
27286bacbd2SBarry Smith   ierr = MatMarkDiagonal_SeqAIJ(*fact);CHKERRQ(ierr);
27386bacbd2SBarry Smith   (*fact)->info.factor_mallocs = 0;
27486bacbd2SBarry Smith 
27586bacbd2SBarry Smith   ierr = MatMarkDiagonal_SeqAIJ(A);CHKERRQ(ierr);
27686bacbd2SBarry Smith 
27786bacbd2SBarry Smith   /* check out for identical nodes. If found, use inode functions */
27886bacbd2SBarry Smith   ierr = Mat_AIJ_CheckInode(*fact);CHKERRQ(ierr);
27986bacbd2SBarry Smith 
28086bacbd2SBarry Smith   PetscFunctionReturn(0);
28186bacbd2SBarry Smith }
28286bacbd2SBarry Smith #else
28386bacbd2SBarry Smith #undef __FUNC__
28486bacbd2SBarry Smith #define __FUNC__ "MatILUDTFactor_SeqAIJ"
285c3b2863dSBarry Smith int MatILUDTFactor_SeqAIJ(Mat A,MatILUInfo *info,IS isrow,IS iscol,Mat *fact)
28686bacbd2SBarry Smith {
28786bacbd2SBarry Smith   PetscFunctionBegin;
28886bacbd2SBarry Smith   SETERRQ(1,1,"You must install Saad's ILUDT to use this");
28986bacbd2SBarry Smith }
29086bacbd2SBarry Smith #endif
29186bacbd2SBarry Smith 
292289bc588SBarry Smith /*
293289bc588SBarry Smith     Factorization code for AIJ format.
294289bc588SBarry Smith */
2955615d1e5SSatish Balay #undef __FUNC__
2965615d1e5SSatish Balay #define __FUNC__ "MatLUFactorSymbolic_SeqAIJ"
297416022c9SBarry Smith int MatLUFactorSymbolic_SeqAIJ(Mat A,IS isrow,IS iscol,double f,Mat *B)
298289bc588SBarry Smith {
299416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data, *b;
300289bc588SBarry Smith   IS         isicol;
301416022c9SBarry Smith   int        *r,*ic, ierr, i, n = a->m, *ai = a->i, *aj = a->j;
302416022c9SBarry Smith   int        *ainew,*ajnew, jmax,*fill, *ajtmp, nz,shift = a->indexshift;
30391bf9adeSBarry Smith   int        *idnew, idx, row,m,fm, nnz, nzi, realloc = 0,nzbd,*im;
304289bc588SBarry Smith 
3053a40ed3dSBarry Smith   PetscFunctionBegin;
306d3cbd50cSLois Curfman McInnes   PetscValidHeaderSpecific(isrow,IS_COOKIE);
307d3cbd50cSLois Curfman McInnes   PetscValidHeaderSpecific(iscol,IS_COOKIE);
308f1af5d2fSBarry Smith   if (A->M != A->N) SETERRQ(PETSC_ERR_ARG_WRONG,0,"matrix must be square");
3093b2fbd54SBarry Smith 
31078b31e54SBarry Smith   ierr = ISInvertPermutation(iscol,&isicol);CHKERRQ(ierr);
311ac121b3dSBarry Smith   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
312ac121b3dSBarry Smith   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
313289bc588SBarry Smith 
314289bc588SBarry Smith   /* get new row pointers */
3150452661fSBarry Smith   ainew    = (int *) PetscMalloc( (n+1)*sizeof(int) );CHKPTRQ(ainew);
316dbb450caSBarry Smith   ainew[0] = -shift;
317289bc588SBarry Smith   /* don't know how many column pointers are needed so estimate */
318dbb450caSBarry Smith   jmax  = (int) (f*ai[n]+(!shift));
3190452661fSBarry Smith   ajnew = (int *) PetscMalloc( (jmax)*sizeof(int) );CHKPTRQ(ajnew);
320289bc588SBarry Smith   /* fill is a linked list of nonzeros in active row */
3210452661fSBarry Smith   fill = (int *) PetscMalloc( (2*n+1)*sizeof(int));CHKPTRQ(fill);
3222fb3ed76SBarry Smith   im   = fill + n + 1;
323289bc588SBarry Smith   /* idnew is location of diagonal in factor */
3240452661fSBarry Smith   idnew    = (int *) PetscMalloc( (n+1)*sizeof(int));CHKPTRQ(idnew);
325dbb450caSBarry Smith   idnew[0] = -shift;
326289bc588SBarry Smith 
327289bc588SBarry Smith   for ( i=0; i<n; i++ ) {
328289bc588SBarry Smith     /* first copy previous fill into linked list */
329289bc588SBarry Smith     nnz     = nz    = ai[r[i]+1] - ai[r[i]];
3306b96ef82SSatish Balay     if (!nz) SETERRQ(PETSC_ERR_MAT_LU_ZRPVT,1,"Empty row in matrix");
331dbb450caSBarry Smith     ajtmp   = aj + ai[r[i]] + shift;
332289bc588SBarry Smith     fill[n] = n;
333289bc588SBarry Smith     while (nz--) {
334289bc588SBarry Smith       fm  = n;
335dbb450caSBarry Smith       idx = ic[*ajtmp++ + shift];
336289bc588SBarry Smith       do {
337289bc588SBarry Smith         m  = fm;
338289bc588SBarry Smith         fm = fill[m];
339289bc588SBarry Smith       } while (fm < idx);
340289bc588SBarry Smith       fill[m]   = idx;
341289bc588SBarry Smith       fill[idx] = fm;
342289bc588SBarry Smith     }
343289bc588SBarry Smith     row = fill[n];
344289bc588SBarry Smith     while ( row < i ) {
345dbb450caSBarry Smith       ajtmp = ajnew + idnew[row] + (!shift);
3462fb3ed76SBarry Smith       nzbd  = 1 + idnew[row] - ainew[row];
3472fb3ed76SBarry Smith       nz    = im[row] - nzbd;
348289bc588SBarry Smith       fm    = row;
3492fb3ed76SBarry Smith       while (nz-- > 0) {
350dbb450caSBarry Smith         idx = *ajtmp++ + shift;
3512fb3ed76SBarry Smith         nzbd++;
3522fb3ed76SBarry Smith         if (idx == i) im[row] = nzbd;
353289bc588SBarry Smith         do {
354289bc588SBarry Smith           m  = fm;
355289bc588SBarry Smith           fm = fill[m];
356289bc588SBarry Smith         } while (fm < idx);
357289bc588SBarry Smith         if (fm != idx) {
358289bc588SBarry Smith           fill[m]   = idx;
359289bc588SBarry Smith           fill[idx] = fm;
360289bc588SBarry Smith           fm        = idx;
361289bc588SBarry Smith           nnz++;
362289bc588SBarry Smith         }
363289bc588SBarry Smith       }
364289bc588SBarry Smith       row = fill[row];
365289bc588SBarry Smith     }
366289bc588SBarry Smith     /* copy new filled row into permanent storage */
367289bc588SBarry Smith     ainew[i+1] = ainew[i] + nnz;
368496e697eSBarry Smith     if (ainew[i+1] > jmax) {
369ecf371e4SBarry Smith 
370ecf371e4SBarry Smith       /* estimate how much additional space we will need */
371ecf371e4SBarry Smith       /* use the strategy suggested by David Hysom <hysom@perch-t.icase.edu> */
372ecf371e4SBarry Smith       /* just double the memory each time */
373ecf371e4SBarry Smith       int maxadd = jmax;
374ecf371e4SBarry Smith       /* maxadd = (int) ((f*(ai[n]+(!shift))*(n-i+5))/n); */
375bbb6d6a8SBarry Smith       if (maxadd < nnz) maxadd = (n-i)*(nnz+1);
3762fb3ed76SBarry Smith       jmax += maxadd;
377ecf371e4SBarry Smith 
378ecf371e4SBarry Smith       /* allocate a longer ajnew */
3790452661fSBarry Smith       ajtmp = (int *) PetscMalloc( jmax*sizeof(int) );CHKPTRQ(ajtmp);
380549d3d68SSatish Balay       ierr  = PetscMemcpy(ajtmp,ajnew,(ainew[i]+shift)*sizeof(int));CHKERRQ(ierr);
381606d414cSSatish Balay       ierr  = PetscFree(ajnew);CHKERRQ(ierr);
382289bc588SBarry Smith       ajnew = ajtmp;
3832fb3ed76SBarry Smith       realloc++; /* count how many times we realloc */
384289bc588SBarry Smith     }
385dbb450caSBarry Smith     ajtmp = ajnew + ainew[i] + shift;
386289bc588SBarry Smith     fm    = fill[n];
387289bc588SBarry Smith     nzi   = 0;
3882fb3ed76SBarry Smith     im[i] = nnz;
389289bc588SBarry Smith     while (nnz--) {
390289bc588SBarry Smith       if (fm < i) nzi++;
391dbb450caSBarry Smith       *ajtmp++ = fm - shift;
392289bc588SBarry Smith       fm       = fill[fm];
393289bc588SBarry Smith     }
394289bc588SBarry Smith     idnew[i] = ainew[i] + nzi;
395289bc588SBarry Smith   }
396464e8d44SSatish Balay   if (ai[n] != 0) {
397464e8d44SSatish Balay     double af = ((double)ainew[n])/((double)ai[n]);
398c38d4ed2SBarry Smith     PLogInfo(A,"MatLUFactorSymbolic_SeqAIJ:Reallocs %d Fill ratio:given %g needed %g\n",realloc,f,af);
399981c4779SBarry Smith     PLogInfo(A,"MatLUFactorSymbolic_SeqAIJ:Run with -pc_lu_fill %g or use \n",af);
400981c4779SBarry Smith     PLogInfo(A,"MatLUFactorSymbolic_SeqAIJ:PCLUSetFill(pc,%g);\n",af);
401981c4779SBarry Smith     PLogInfo(A,"MatLUFactorSymbolic_SeqAIJ:for best performance.\n");
40205bf559cSSatish Balay   } else {
403981c4779SBarry Smith     PLogInfo(A,"MatLUFactorSymbolic_SeqAIJ: Empty matrix\n");
40405bf559cSSatish Balay   }
4052fb3ed76SBarry Smith 
406898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
407898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
4081987afe7SBarry Smith 
409606d414cSSatish Balay   ierr = PetscFree(fill);CHKERRQ(ierr);
410289bc588SBarry Smith 
411289bc588SBarry Smith   /* put together the new matrix */
412b4fd4287SBarry Smith   ierr = MatCreateSeqAIJ(A->comm,n,n,0,PETSC_NULL,B);CHKERRQ(ierr);
4131987afe7SBarry Smith   PLogObjectParent(*B,isicol);
414416022c9SBarry Smith   b = (Mat_SeqAIJ *) (*B)->data;
415606d414cSSatish Balay   ierr = PetscFree(b->imax);CHKERRQ(ierr);
4167c922b88SBarry Smith   b->singlemalloc = PETSC_FALSE;
417e8d4e0b9SBarry Smith   /* the next line frees the default space generated by the Create() */
418606d414cSSatish Balay   ierr = PetscFree(b->a);CHKERRQ(ierr);
419606d414cSSatish Balay   ierr = PetscFree(b->ilen);CHKERRQ(ierr);
42091bf9adeSBarry Smith   b->a          = (Scalar *) PetscMalloc((ainew[n]+shift+1)*sizeof(Scalar));CHKPTRQ(b->a);
421416022c9SBarry Smith   b->j          = ajnew;
422416022c9SBarry Smith   b->i          = ainew;
423416022c9SBarry Smith   b->diag       = idnew;
424416022c9SBarry Smith   b->ilen       = 0;
425416022c9SBarry Smith   b->imax       = 0;
426416022c9SBarry Smith   b->row        = isrow;
427416022c9SBarry Smith   b->col        = iscol;
428c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
429c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
43082bf6240SBarry Smith   b->icol       = isicol;
43191bf9adeSBarry Smith   b->solve_work = (Scalar *) PetscMalloc( (n+1)*sizeof(Scalar));CHKPTRQ(b->solve_work);
432416022c9SBarry Smith   /* In b structure:  Free imax, ilen, old a, old j.
4337fd98bd6SLois Curfman McInnes      Allocate idnew, solve_work, new a, new j */
434416022c9SBarry Smith   PLogObjectMemory(*B,(ainew[n]+shift-n)*(sizeof(int)+sizeof(Scalar)));
435416022c9SBarry Smith   b->maxnz = b->nz = ainew[n] + shift;
4367fd98bd6SLois Curfman McInnes 
437e93ccc4dSBarry Smith   (*B)->factor                 =  FACTOR_LU;;
438ae068f58SLois Curfman McInnes   (*B)->info.factor_mallocs    = realloc;
439ae068f58SLois Curfman McInnes   (*B)->info.fill_ratio_given  = f;
440703deb49SSatish Balay   (*B)->ops->lufactornumeric   =  A->ops->lufactornumeric; /* Use Inode variant if A has inodes */
441703deb49SSatish Balay 
442e93ccc4dSBarry Smith   if (ai[n] != 0) {
443e93ccc4dSBarry Smith     (*B)->info.fill_ratio_needed = ((double)ainew[n])/((double)ai[n]);
444eea2913aSSatish Balay   } else {
445eea2913aSSatish Balay     (*B)->info.fill_ratio_needed = 0.0;
446eea2913aSSatish Balay   }
4473a40ed3dSBarry Smith   PetscFunctionReturn(0);
448289bc588SBarry Smith }
4490a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
450184914b5SBarry Smith extern int Mat_AIJ_CheckInode(Mat);
45141c01911SSatish Balay 
4525615d1e5SSatish Balay #undef __FUNC__
4535615d1e5SSatish Balay #define __FUNC__ "MatLUFactorNumeric_SeqAIJ"
454416022c9SBarry Smith int MatLUFactorNumeric_SeqAIJ(Mat A,Mat *B)
455289bc588SBarry Smith {
456416022c9SBarry Smith   Mat        C = *B;
457416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data, *b = (Mat_SeqAIJ *)C->data;
45882bf6240SBarry Smith   IS         isrow = b->row, isicol = b->icol;
459416022c9SBarry Smith   int        *r,*ic, ierr, i, j, n = a->m, *ai = b->i, *aj = b->j;
4601987afe7SBarry Smith   int        *ajtmpold, *ajtmp, nz, row, *ics, shift = a->indexshift;
461f2582111SSatish Balay   int        *diag_offset = b->diag,diag,k;
46235aab85fSBarry Smith   int        preserve_row_sums = (int) a->ilu_preserve_row_sums;
4633a40ed3dSBarry Smith   register   int    *pj;
4648ecf7165SBarry Smith   Scalar     *rtmp,*v, *pc, multiplier,sum,inner_sum,*rowsums = 0;
46535aab85fSBarry Smith   double     ssum;
46635aab85fSBarry Smith   register   Scalar *pv, *rtmps,*u_values;
467289bc588SBarry Smith 
4683a40ed3dSBarry Smith   PetscFunctionBegin;
46982bf6240SBarry Smith 
47078b31e54SBarry Smith   ierr  = ISGetIndices(isrow,&r);CHKERRQ(ierr);
47178b31e54SBarry Smith   ierr  = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
4720452661fSBarry Smith   rtmp  = (Scalar *) PetscMalloc( (n+1)*sizeof(Scalar) );CHKPTRQ(rtmp);
473549d3d68SSatish Balay   ierr  = PetscMemzero(rtmp,(n+1)*sizeof(Scalar));CHKERRQ(ierr);
4740cf60383SBarry Smith   rtmps = rtmp + shift; ics = ic + shift;
475289bc588SBarry Smith 
4766cf997b0SBarry Smith   /* precalculate row sums */
47735aab85fSBarry Smith   if (preserve_row_sums) {
47835aab85fSBarry Smith     rowsums = (Scalar *) PetscMalloc( n*sizeof(Scalar) );CHKPTRQ(rowsums);
47935aab85fSBarry Smith     for ( i=0; i<n; i++ ) {
48035aab85fSBarry Smith       nz  = a->i[r[i]+1] - a->i[r[i]];
48135aab85fSBarry Smith       v   = a->a + a->i[r[i]] + shift;
48235aab85fSBarry Smith       sum = 0.0;
48335aab85fSBarry Smith       for ( j=0; j<nz; j++ ) sum += v[j];
48435aab85fSBarry Smith       rowsums[i] = sum;
48535aab85fSBarry Smith     }
48635aab85fSBarry Smith   }
48735aab85fSBarry Smith 
488289bc588SBarry Smith   for ( i=0; i<n; i++ ) {
489289bc588SBarry Smith     nz    = ai[i+1] - ai[i];
490dbb450caSBarry Smith     ajtmp = aj + ai[i] + shift;
49148e94559SBarry Smith     for  ( j=0; j<nz; j++ ) rtmps[ajtmp[j]] = 0.0;
492289bc588SBarry Smith 
493289bc588SBarry Smith     /* load in initial (unfactored row) */
494416022c9SBarry Smith     nz       = a->i[r[i]+1] - a->i[r[i]];
495416022c9SBarry Smith     ajtmpold = a->j + a->i[r[i]] + shift;
496416022c9SBarry Smith     v        = a->a + a->i[r[i]] + shift;
4970cf60383SBarry Smith     for ( j=0; j<nz; j++ ) rtmp[ics[ajtmpold[j]]] =  v[j];
498289bc588SBarry Smith 
499dbb450caSBarry Smith     row = *ajtmp++ + shift;
500f2582111SSatish Balay     while  (row < i ) {
5018c37ef55SBarry Smith       pc = rtmp + row;
5028c37ef55SBarry Smith       if (*pc != 0.0) {
5031987afe7SBarry Smith         pv         = b->a + diag_offset[row] + shift;
5041987afe7SBarry Smith         pj         = b->j + diag_offset[row] + (!shift);
50535aab85fSBarry Smith         multiplier = *pc / *pv++;
5068c37ef55SBarry Smith         *pc        = multiplier;
507f2582111SSatish Balay         nz         = ai[row+1] - diag_offset[row] - 1;
508f2582111SSatish Balay         for (j=0; j<nz; j++) rtmps[pj[j]] -= multiplier * pv[j];
509f2582111SSatish Balay         PLogFlops(2*nz);
510289bc588SBarry Smith       }
511dbb450caSBarry Smith       row = *ajtmp++ + shift;
512289bc588SBarry Smith     }
513416022c9SBarry Smith     /* finished row so stick it into b->a */
514416022c9SBarry Smith     pv = b->a + ai[i] + shift;
515416022c9SBarry Smith     pj = b->j + ai[i] + shift;
5168c37ef55SBarry Smith     nz = ai[i+1] - ai[i];
517416022c9SBarry Smith     for ( j=0; j<nz; j++ ) {pv[j] = rtmps[pj[j]];}
51835aab85fSBarry Smith     diag = diag_offset[i] - ai[i];
51935aab85fSBarry Smith     /*
52035aab85fSBarry Smith           Possibly adjust diagonal entry on current row to force
52135aab85fSBarry Smith         LU matrix to have same row sum as initial matrix.
52235aab85fSBarry Smith     */
523d708749eSBarry Smith     if (pv[diag] == 0.0) {
5246cf997b0SBarry Smith       SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,0,"Zero pivot row %d",i);
525d708749eSBarry Smith     }
52635aab85fSBarry Smith     if (preserve_row_sums) {
52735aab85fSBarry Smith       pj  = b->j + ai[i] + shift;
52835aab85fSBarry Smith       sum = rowsums[i];
52935aab85fSBarry Smith       for ( j=0; j<diag; j++ ) {
53035aab85fSBarry Smith         u_values  = b->a + diag_offset[pj[j]] + shift;
53135aab85fSBarry Smith         nz        = ai[pj[j]+1] - diag_offset[pj[j]];
53235aab85fSBarry Smith         inner_sum = 0.0;
53335aab85fSBarry Smith         for ( k=0; k<nz; k++ ) {
53435aab85fSBarry Smith           inner_sum += u_values[k];
53535aab85fSBarry Smith         }
53635aab85fSBarry Smith         sum -= pv[j]*inner_sum;
53735aab85fSBarry Smith 
53835aab85fSBarry Smith       }
53935aab85fSBarry Smith       nz       = ai[i+1] - diag_offset[i] - 1;
54035aab85fSBarry Smith       u_values = b->a + diag_offset[i] + 1 + shift;
54135aab85fSBarry Smith       for ( k=0; k<nz; k++ ) {
54235aab85fSBarry Smith         sum -= u_values[k];
54335aab85fSBarry Smith       }
54435aab85fSBarry Smith       ssum = PetscAbsScalar(sum/pv[diag]);
54535aab85fSBarry Smith       if (ssum < 1000. && ssum > .001) pv[diag] = sum;
54635aab85fSBarry Smith     }
54735aab85fSBarry Smith     /* check pivot entry for current row */
5488c37ef55SBarry Smith   }
5490f11f9aeSSatish Balay 
55035aab85fSBarry Smith   /* invert diagonal entries for simplier triangular solves */
55135aab85fSBarry Smith   for ( i=0; i<n; i++ ) {
55235aab85fSBarry Smith     b->a[diag_offset[i]+shift] = 1.0/b->a[diag_offset[i]+shift];
55335aab85fSBarry Smith   }
55435aab85fSBarry Smith 
555606d414cSSatish Balay   if (preserve_row_sums) {ierr = PetscFree(rowsums);CHKERRQ(ierr);}
556606d414cSSatish Balay   ierr = PetscFree(rtmp);CHKERRQ(ierr);
55778b31e54SBarry Smith   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
55878b31e54SBarry Smith   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
559416022c9SBarry Smith   C->factor = FACTOR_LU;
56041c01911SSatish Balay   ierr = Mat_AIJ_CheckInode(C);CHKERRQ(ierr);
561c456f294SBarry Smith   C->assembled = PETSC_TRUE;
562416022c9SBarry Smith   PLogFlops(b->n);
5633a40ed3dSBarry Smith   PetscFunctionReturn(0);
564289bc588SBarry Smith }
5650a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
5665615d1e5SSatish Balay #undef __FUNC__
5675615d1e5SSatish Balay #define __FUNC__ "MatLUFactor_SeqAIJ"
568416022c9SBarry Smith int MatLUFactor_SeqAIJ(Mat A,IS row,IS col,double f)
569da3a660dSBarry Smith {
570416022c9SBarry Smith   Mat_SeqAIJ     *mat = (Mat_SeqAIJ *) A->data;
57186bacbd2SBarry Smith   int            ierr,refct;
572416022c9SBarry Smith   Mat            C;
573f830108cSBarry Smith   PetscOps       *Abops;
5740a6ffc59SBarry Smith   MatOps         Aops;
575416022c9SBarry Smith 
5763a40ed3dSBarry Smith   PetscFunctionBegin;
577f2582111SSatish Balay   ierr = MatLUFactorSymbolic(A,row,col,f,&C);CHKERRQ(ierr);
578f2582111SSatish Balay   ierr = MatLUFactorNumeric(A,&C);CHKERRQ(ierr);
579da3a660dSBarry Smith 
580da3a660dSBarry Smith   /* free all the data structures from mat */
581606d414cSSatish Balay   ierr = PetscFree(mat->a);CHKERRQ(ierr);
582606d414cSSatish Balay   if (!mat->singlemalloc) {
583606d414cSSatish Balay     ierr = PetscFree(mat->i);CHKERRQ(ierr);
584606d414cSSatish Balay     ierr = PetscFree(mat->j);CHKERRQ(ierr);
585606d414cSSatish Balay   }
586606d414cSSatish Balay   if (mat->diag) {ierr = PetscFree(mat->diag);CHKERRQ(ierr);}
587606d414cSSatish Balay   if (mat->ilen) {ierr = PetscFree(mat->ilen);CHKERRQ(ierr);}
588606d414cSSatish Balay   if (mat->imax) {ierr = PetscFree(mat->imax);CHKERRQ(ierr);}
589606d414cSSatish Balay   if (mat->solve_work) {ierr = PetscFree(mat->solve_work);CHKERRQ(ierr);}
590606d414cSSatish Balay   if (mat->inode.size) {ierr = PetscFree(mat->inode.size);CHKERRQ(ierr);}
5910f0aacb9SBarry Smith   if (mat->icol) {ierr = ISDestroy(mat->icol);CHKERRQ(ierr);}
592606d414cSSatish Balay   ierr = PetscFree(mat);CHKERRQ(ierr);
593da3a660dSBarry Smith 
59417642b18SBarry Smith   ierr = MapDestroy(A->rmap);CHKERRQ(ierr);
59517642b18SBarry Smith   ierr = MapDestroy(A->cmap);CHKERRQ(ierr);
59617642b18SBarry Smith 
597f830108cSBarry Smith   /*
598f830108cSBarry Smith        This is horrible, horrible code. We need to keep the
599f830108cSBarry Smith     A pointers for the bops and ops but copy everything
600f830108cSBarry Smith     else from C.
601f830108cSBarry Smith   */
602f830108cSBarry Smith   Abops = A->bops;
603f830108cSBarry Smith   Aops  = A->ops;
60486bacbd2SBarry Smith   refct = A->refct;
605549d3d68SSatish Balay   ierr  = PetscMemcpy(A,C,sizeof(struct _p_Mat));CHKERRQ(ierr);
606451c4af8SSatish Balay   mat   = (Mat_SeqAIJ *) A->data;
607451c4af8SSatish Balay   PLogObjectParent(A,mat->icol);
608451c4af8SSatish Balay 
609f830108cSBarry Smith   A->bops  = Abops;
610f830108cSBarry Smith   A->ops   = Aops;
611bef8e0ddSBarry Smith   A->qlist = 0;
61286bacbd2SBarry Smith   A->refct = refct;
613c38d4ed2SBarry Smith   /* copy over the type_name and name */
614c38d4ed2SBarry Smith   ierr     = PetscStrallocpy(C->type_name,&A->type_name);CHKERRQ(ierr);
615c38d4ed2SBarry Smith   ierr     = PetscStrallocpy(C->name,&A->name);CHKERRQ(ierr);
616f830108cSBarry Smith 
6170452661fSBarry Smith   PetscHeaderDestroy(C);
618c38d4ed2SBarry Smith 
6193a40ed3dSBarry Smith   PetscFunctionReturn(0);
620da3a660dSBarry Smith }
6210a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
6225615d1e5SSatish Balay #undef __FUNC__
6235615d1e5SSatish Balay #define __FUNC__ "MatSolve_SeqAIJ"
624416022c9SBarry Smith int MatSolve_SeqAIJ(Mat A,Vec bb, Vec xx)
6258c37ef55SBarry Smith {
626416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
627416022c9SBarry Smith   IS         iscol = a->col, isrow = a->row;
628416022c9SBarry Smith   int        *r,*c, ierr, i,  n = a->m, *vi, *ai = a->i, *aj = a->j;
6293b2fbd54SBarry Smith   int        nz,shift = a->indexshift,*rout,*cout;
630416022c9SBarry Smith   Scalar     *x,*b,*tmp, *tmps, *aa = a->a, sum, *v;
6318c37ef55SBarry Smith 
6323a40ed3dSBarry Smith   PetscFunctionBegin;
6333a40ed3dSBarry Smith   if (!n) PetscFunctionReturn(0);
63491bf9adeSBarry Smith 
635e1311b90SBarry Smith   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
636e1311b90SBarry Smith   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
637416022c9SBarry Smith   tmp  = a->solve_work;
6388c37ef55SBarry Smith 
6393b2fbd54SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
6403b2fbd54SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
6418c37ef55SBarry Smith 
6428c37ef55SBarry Smith   /* forward solve the lower triangular */
6438c37ef55SBarry Smith   tmp[0] = b[*r++];
6440cf60383SBarry Smith   tmps   = tmp + shift;
6458c37ef55SBarry Smith   for ( i=1; i<n; i++ ) {
646dbb450caSBarry Smith     v   = aa + ai[i] + shift;
647dbb450caSBarry Smith     vi  = aj + ai[i] + shift;
648416022c9SBarry Smith     nz  = a->diag[i] - ai[i];
6498c37ef55SBarry Smith     sum = b[*r++];
6500cf60383SBarry Smith     while (nz--) sum -= *v++ * tmps[*vi++];
6518c37ef55SBarry Smith     tmp[i] = sum;
6528c37ef55SBarry Smith   }
6538c37ef55SBarry Smith 
6548c37ef55SBarry Smith   /* backward solve the upper triangular */
6558c37ef55SBarry Smith   for ( i=n-1; i>=0; i-- ){
656416022c9SBarry Smith     v   = aa + a->diag[i] + (!shift);
657416022c9SBarry Smith     vi  = aj + a->diag[i] + (!shift);
658416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
6598c37ef55SBarry Smith     sum = tmp[i];
6600cf60383SBarry Smith     while (nz--) sum -= *v++ * tmps[*vi++];
661416022c9SBarry Smith     x[*c--] = tmp[i] = sum*aa[a->diag[i]+shift];
6628c37ef55SBarry Smith   }
6638c37ef55SBarry Smith 
6643b2fbd54SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
6653b2fbd54SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
666cb5b572fSBarry Smith   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
667cb5b572fSBarry Smith   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
668416022c9SBarry Smith   PLogFlops(2*a->nz - a->n);
6693a40ed3dSBarry Smith   PetscFunctionReturn(0);
6708c37ef55SBarry Smith }
671026e39d0SSatish Balay 
672930ae53cSBarry Smith /* ----------------------------------------------------------- */
673930ae53cSBarry Smith #undef __FUNC__
674930ae53cSBarry Smith #define __FUNC__ "MatSolve_SeqAIJ_NaturalOrdering"
675930ae53cSBarry Smith int MatSolve_SeqAIJ_NaturalOrdering(Mat A,Vec bb, Vec xx)
676930ae53cSBarry Smith {
677930ae53cSBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
678d85afc42SSatish Balay   int        n = a->m, *ai = a->i, *aj = a->j, *adiag = a->diag,ierr;
679d85afc42SSatish Balay   Scalar     *x,*b, *aa = a->a, sum;
680aa482453SBarry Smith #if !defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ)
681d85afc42SSatish Balay   int        adiag_i,i,*vi,nz,ai_i;
682d85afc42SSatish Balay   Scalar     *v;
683d85afc42SSatish Balay #endif
684930ae53cSBarry Smith 
6853a40ed3dSBarry Smith   PetscFunctionBegin;
6863a40ed3dSBarry Smith   if (!n) PetscFunctionReturn(0);
687930ae53cSBarry Smith   if (a->indexshift) {
6883a40ed3dSBarry Smith      ierr = MatSolve_SeqAIJ(A,bb,xx);CHKERRQ(ierr);
6893a40ed3dSBarry Smith      PetscFunctionReturn(0);
690930ae53cSBarry Smith   }
691930ae53cSBarry Smith 
692e1311b90SBarry Smith   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
693e1311b90SBarry Smith   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
694930ae53cSBarry Smith 
695aa482453SBarry Smith #if defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ)
6961c4feecaSSatish Balay   fortransolveaij_(&n,x,ai,aj,adiag,aa,b);
6971c4feecaSSatish Balay #else
698930ae53cSBarry Smith   /* forward solve the lower triangular */
699930ae53cSBarry Smith   x[0] = b[0];
700930ae53cSBarry Smith   for ( i=1; i<n; i++ ) {
701930ae53cSBarry Smith     ai_i = ai[i];
702930ae53cSBarry Smith     v    = aa + ai_i;
703930ae53cSBarry Smith     vi   = aj + ai_i;
704930ae53cSBarry Smith     nz   = adiag[i] - ai_i;
705930ae53cSBarry Smith     sum  = b[i];
706930ae53cSBarry Smith     while (nz--) sum -= *v++ * x[*vi++];
707930ae53cSBarry Smith     x[i] = sum;
708930ae53cSBarry Smith   }
709930ae53cSBarry Smith 
710930ae53cSBarry Smith   /* backward solve the upper triangular */
711930ae53cSBarry Smith   for ( i=n-1; i>=0; i-- ){
712930ae53cSBarry Smith     adiag_i = adiag[i];
713d708749eSBarry Smith     v       = aa + adiag_i + 1;
714d708749eSBarry Smith     vi      = aj + adiag_i + 1;
715930ae53cSBarry Smith     nz      = ai[i+1] - adiag_i - 1;
716930ae53cSBarry Smith     sum     = x[i];
717930ae53cSBarry Smith     while (nz--) sum -= *v++ * x[*vi++];
718930ae53cSBarry Smith     x[i]    = sum*aa[adiag_i];
719930ae53cSBarry Smith   }
7201c4feecaSSatish Balay #endif
721930ae53cSBarry Smith   PLogFlops(2*a->nz - a->n);
722cb5b572fSBarry Smith   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
723cb5b572fSBarry Smith   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
7243a40ed3dSBarry Smith   PetscFunctionReturn(0);
725930ae53cSBarry Smith }
726930ae53cSBarry Smith 
7275615d1e5SSatish Balay #undef __FUNC__
7285615d1e5SSatish Balay #define __FUNC__ "MatSolveAdd_SeqAIJ"
729416022c9SBarry Smith int MatSolveAdd_SeqAIJ(Mat A,Vec bb, Vec yy, Vec xx)
730da3a660dSBarry Smith {
731416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
732416022c9SBarry Smith   IS         iscol = a->col, isrow = a->row;
733416022c9SBarry Smith   int        *r,*c, ierr, i,  n = a->m, *vi, *ai = a->i, *aj = a->j;
7343b2fbd54SBarry Smith   int        nz, shift = a->indexshift,*rout,*cout;
735416022c9SBarry Smith   Scalar     *x,*b,*tmp, *aa = a->a, sum, *v;
736da3a660dSBarry Smith 
7373a40ed3dSBarry Smith   PetscFunctionBegin;
73878b31e54SBarry Smith   if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);}
739da3a660dSBarry Smith 
740e1311b90SBarry Smith   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
741e1311b90SBarry Smith   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
742416022c9SBarry Smith   tmp  = a->solve_work;
743da3a660dSBarry Smith 
7443b2fbd54SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
7453b2fbd54SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
746da3a660dSBarry Smith 
747da3a660dSBarry Smith   /* forward solve the lower triangular */
748da3a660dSBarry Smith   tmp[0] = b[*r++];
749da3a660dSBarry Smith   for ( i=1; i<n; i++ ) {
750dbb450caSBarry Smith     v   = aa + ai[i] + shift;
751dbb450caSBarry Smith     vi  = aj + ai[i] + shift;
752416022c9SBarry Smith     nz  = a->diag[i] - ai[i];
753da3a660dSBarry Smith     sum = b[*r++];
754dbb450caSBarry Smith     while (nz--) sum -= *v++ * tmp[*vi++ + shift];
755da3a660dSBarry Smith     tmp[i] = sum;
756da3a660dSBarry Smith   }
757da3a660dSBarry Smith 
758da3a660dSBarry Smith   /* backward solve the upper triangular */
759da3a660dSBarry Smith   for ( i=n-1; i>=0; i-- ){
760416022c9SBarry Smith     v   = aa + a->diag[i] + (!shift);
761416022c9SBarry Smith     vi  = aj + a->diag[i] + (!shift);
762416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
763da3a660dSBarry Smith     sum = tmp[i];
764dbb450caSBarry Smith     while (nz--) sum -= *v++ * tmp[*vi++ + shift];
765416022c9SBarry Smith     tmp[i] = sum*aa[a->diag[i]+shift];
766da3a660dSBarry Smith     x[*c--] += tmp[i];
767da3a660dSBarry Smith   }
768da3a660dSBarry Smith 
7693b2fbd54SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
7703b2fbd54SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
771cb5b572fSBarry Smith   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
772cb5b572fSBarry Smith   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
773416022c9SBarry Smith   PLogFlops(2*a->nz);
774898183ecSLois Curfman McInnes 
7753a40ed3dSBarry Smith   PetscFunctionReturn(0);
776da3a660dSBarry Smith }
777da3a660dSBarry Smith /* -------------------------------------------------------------------*/
7785615d1e5SSatish Balay #undef __FUNC__
7797c922b88SBarry Smith #define __FUNC__ "MatSolveTranspose_SeqAIJ"
7807c922b88SBarry Smith int MatSolveTranspose_SeqAIJ(Mat A,Vec bb, Vec xx)
781da3a660dSBarry Smith {
782416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
7832235e667SBarry Smith   IS         iscol = a->col, isrow = a->row;
784416022c9SBarry Smith   int        *r,*c, ierr, i, n = a->m, *vi, *ai = a->i, *aj = a->j;
785f1af5d2fSBarry Smith   int        nz,shift = a->indexshift,*rout,*cout, *diag = a->diag;
786f1af5d2fSBarry Smith   Scalar     *x,*b,*tmp, *aa = a->a, *v, s1;
787da3a660dSBarry Smith 
7883a40ed3dSBarry Smith   PetscFunctionBegin;
789e1311b90SBarry Smith   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
790e1311b90SBarry Smith   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
791416022c9SBarry Smith   tmp  = a->solve_work;
792da3a660dSBarry Smith 
7932235e667SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
7942235e667SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
795da3a660dSBarry Smith 
796da3a660dSBarry Smith   /* copy the b into temp work space according to permutation */
7972235e667SBarry Smith   for ( i=0; i<n; i++ ) tmp[i] = b[c[i]];
798da3a660dSBarry Smith 
799da3a660dSBarry Smith   /* forward solve the U^T */
800da3a660dSBarry Smith   for ( i=0; i<n; i++ ) {
801f1af5d2fSBarry Smith     v   = aa + diag[i] + shift;
802f1af5d2fSBarry Smith     vi  = aj + diag[i] + (!shift);
803f1af5d2fSBarry Smith     nz  = ai[i+1] - diag[i] - 1;
804c38d4ed2SBarry Smith     s1  = tmp[i];
805c38d4ed2SBarry Smith     s1 *= *(v++);  /* multiply by inverse of diagonal entry */
806da3a660dSBarry Smith     while (nz--) {
807f1af5d2fSBarry Smith       tmp[*vi++ + shift] -= (*v++)*s1;
808da3a660dSBarry Smith     }
809c38d4ed2SBarry Smith     tmp[i] = s1;
810da3a660dSBarry Smith   }
811da3a660dSBarry Smith 
812da3a660dSBarry Smith   /* backward solve the L^T */
813da3a660dSBarry Smith   for ( i=n-1; i>=0; i-- ){
814f1af5d2fSBarry Smith     v   = aa + diag[i] - 1 + shift;
815f1af5d2fSBarry Smith     vi  = aj + diag[i] - 1 + shift;
816f1af5d2fSBarry Smith     nz  = diag[i] - ai[i];
817f1af5d2fSBarry Smith     s1  = tmp[i];
818da3a660dSBarry Smith     while (nz--) {
819f1af5d2fSBarry Smith       tmp[*vi-- + shift] -= (*v--)*s1;
820da3a660dSBarry Smith     }
821da3a660dSBarry Smith   }
822da3a660dSBarry Smith 
823da3a660dSBarry Smith   /* copy tmp into x according to permutation */
824da3a660dSBarry Smith   for ( i=0; i<n; i++ ) x[r[i]] = tmp[i];
825da3a660dSBarry Smith 
8262235e667SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
8272235e667SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
828cb5b572fSBarry Smith   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
829cb5b572fSBarry Smith   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
830da3a660dSBarry Smith 
831416022c9SBarry Smith   PLogFlops(2*a->nz-a->n);
8323a40ed3dSBarry Smith   PetscFunctionReturn(0);
833da3a660dSBarry Smith }
834da3a660dSBarry Smith 
8355615d1e5SSatish Balay #undef __FUNC__
8367c922b88SBarry Smith #define __FUNC__ "MatSolveTransposeAdd_SeqAIJ"
8377c922b88SBarry Smith int MatSolveTransposeAdd_SeqAIJ(Mat A,Vec bb, Vec zz,Vec xx)
838da3a660dSBarry Smith {
839416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
8402235e667SBarry Smith   IS         iscol = a->col, isrow = a->row;
841416022c9SBarry Smith   int        *r,*c, ierr, i, n = a->m, *vi, *ai = a->i, *aj = a->j;
842f1af5d2fSBarry Smith   int        nz,shift = a->indexshift, *rout, *cout, *diag = a->diag;
843416022c9SBarry Smith   Scalar     *x,*b,*tmp, *aa = a->a, *v;
8446abc6512SBarry Smith 
8453a40ed3dSBarry Smith   PetscFunctionBegin;
8466abc6512SBarry Smith   if (zz != xx) VecCopy(zz,xx);
8476abc6512SBarry Smith 
848e1311b90SBarry Smith   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
849e1311b90SBarry Smith   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
850416022c9SBarry Smith   tmp = a->solve_work;
8516abc6512SBarry Smith 
8522235e667SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
8532235e667SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
8546abc6512SBarry Smith 
8556abc6512SBarry Smith   /* copy the b into temp work space according to permutation */
8562235e667SBarry Smith   for ( i=0; i<n; i++ ) tmp[i] = b[c[i]];
8576abc6512SBarry Smith 
8586abc6512SBarry Smith   /* forward solve the U^T */
8596abc6512SBarry Smith   for ( i=0; i<n; i++ ) {
860f1af5d2fSBarry Smith     v   = aa + diag[i] + shift;
861f1af5d2fSBarry Smith     vi  = aj + diag[i] + (!shift);
862f1af5d2fSBarry Smith     nz  = ai[i+1] - diag[i] - 1;
8636abc6512SBarry Smith     tmp[i] *= *v++;
8646abc6512SBarry Smith     while (nz--) {
865dbb450caSBarry Smith       tmp[*vi++ + shift] -= (*v++)*tmp[i];
8666abc6512SBarry Smith     }
8676abc6512SBarry Smith   }
8686abc6512SBarry Smith 
8696abc6512SBarry Smith   /* backward solve the L^T */
8706abc6512SBarry Smith   for ( i=n-1; i>=0; i-- ){
871f1af5d2fSBarry Smith     v   = aa + diag[i] - 1 + shift;
872f1af5d2fSBarry Smith     vi  = aj + diag[i] - 1 + shift;
873f1af5d2fSBarry Smith     nz  = diag[i] - ai[i];
8746abc6512SBarry Smith     while (nz--) {
875dbb450caSBarry Smith       tmp[*vi-- + shift] -= (*v--)*tmp[i];
8766abc6512SBarry Smith     }
8776abc6512SBarry Smith   }
8786abc6512SBarry Smith 
8796abc6512SBarry Smith   /* copy tmp into x according to permutation */
8806abc6512SBarry Smith   for ( i=0; i<n; i++ ) x[r[i]] += tmp[i];
8816abc6512SBarry Smith 
8822235e667SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
8832235e667SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
884cb5b572fSBarry Smith   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
885cb5b572fSBarry Smith   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
8866abc6512SBarry Smith 
887416022c9SBarry Smith   PLogFlops(2*a->nz);
8883a40ed3dSBarry Smith   PetscFunctionReturn(0);
889da3a660dSBarry Smith }
8909e25ed09SBarry Smith /* ----------------------------------------------------------------*/
8917c922b88SBarry Smith extern int MatMissingDiagonal_SeqAIJ(Mat);
89208480c60SBarry Smith 
8935615d1e5SSatish Balay #undef __FUNC__
8945615d1e5SSatish Balay #define __FUNC__ "MatILUFactorSymbolic_SeqAIJ"
8956cf997b0SBarry Smith int MatILUFactorSymbolic_SeqAIJ(Mat A,IS isrow,IS iscol,MatILUInfo *info,Mat *fact)
8969e25ed09SBarry Smith {
897416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data, *b;
8989e25ed09SBarry Smith   IS         isicol;
899416022c9SBarry Smith   int        *r,*ic, ierr, prow, n = a->m, *ai = a->i, *aj = a->j;
90056beaf04SBarry Smith   int        *ainew,*ajnew, jmax,*fill, *xi, nz, *im,*ajfill,*flev;
901335d9088SBarry Smith   int        *dloc, idx, row,m,fm, nzf, nzi,len,  realloc = 0, dcount = 0;
9026cf997b0SBarry Smith   int        incrlev,nnz,i,shift = a->indexshift,levels,diagonal_fill;
90377c4ece6SBarry Smith   PetscTruth col_identity, row_identity;
9046cf997b0SBarry Smith   double     f;
9059e25ed09SBarry Smith 
9063a40ed3dSBarry Smith   PetscFunctionBegin;
9076cf997b0SBarry Smith   if (info) {
9086cf997b0SBarry Smith     f             = info->fill;
9090cd17293SBarry Smith     levels        = (int) info->levels;
9100cd17293SBarry Smith     diagonal_fill = (int) info->diagonal_fill;
9116cf997b0SBarry Smith   } else {
9126cf997b0SBarry Smith     f             = 1.0;
9136cf997b0SBarry Smith     levels        = 0;
9146cf997b0SBarry Smith     diagonal_fill = 0;
9156cf997b0SBarry Smith   }
91682bf6240SBarry Smith   ierr = ISInvertPermutation(iscol,&isicol);CHKERRQ(ierr);
91782bf6240SBarry Smith 
91808480c60SBarry Smith   /* special case that simply copies fill pattern */
919be0abb6dSBarry Smith   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
920be0abb6dSBarry Smith   ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr);
92186bacbd2SBarry Smith   if (!levels && row_identity && col_identity) {
9222e8a6d31SBarry Smith     ierr = MatDuplicate_SeqAIJ(A,MAT_DO_NOT_COPY_VALUES,fact);CHKERRQ(ierr);
92308480c60SBarry Smith     (*fact)->factor = FACTOR_LU;
92408480c60SBarry Smith     b               = (Mat_SeqAIJ *) (*fact)->data;
92508480c60SBarry Smith     if (!b->diag) {
9267c922b88SBarry Smith       ierr = MatMarkDiagonal_SeqAIJ(*fact);CHKERRQ(ierr);
92708480c60SBarry Smith     }
9287c922b88SBarry Smith     ierr = MatMissingDiagonal_SeqAIJ(*fact);CHKERRQ(ierr);
92908480c60SBarry Smith     b->row              = isrow;
93008480c60SBarry Smith     b->col              = iscol;
93182bf6240SBarry Smith     b->icol             = isicol;
9320452661fSBarry Smith     b->solve_work       = (Scalar *) PetscMalloc((b->m+1)*sizeof(Scalar));CHKPTRQ(b->solve_work);
933f830108cSBarry Smith     (*fact)->ops->solve = MatSolve_SeqAIJ_NaturalOrdering;
934c38d4ed2SBarry Smith     ierr                = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
935c38d4ed2SBarry Smith     ierr                = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
9363a40ed3dSBarry Smith     PetscFunctionReturn(0);
93708480c60SBarry Smith   }
93808480c60SBarry Smith 
939898183ecSLois Curfman McInnes   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
940898183ecSLois Curfman McInnes   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
9419e25ed09SBarry Smith 
9429e25ed09SBarry Smith   /* get new row pointers */
9430452661fSBarry Smith   ainew = (int *) PetscMalloc( (n+1)*sizeof(int) );CHKPTRQ(ainew);
944dbb450caSBarry Smith   ainew[0] = -shift;
9459e25ed09SBarry Smith   /* don't know how many column pointers are needed so estimate */
946dbb450caSBarry Smith   jmax = (int) (f*(ai[n]+!shift));
9470452661fSBarry Smith   ajnew = (int *) PetscMalloc( (jmax)*sizeof(int) );CHKPTRQ(ajnew);
9489e25ed09SBarry Smith   /* ajfill is level of fill for each fill entry */
9490452661fSBarry Smith   ajfill = (int *) PetscMalloc( (jmax)*sizeof(int) );CHKPTRQ(ajfill);
9509e25ed09SBarry Smith   /* fill is a linked list of nonzeros in active row */
9510452661fSBarry Smith   fill = (int *) PetscMalloc( (n+1)*sizeof(int));CHKPTRQ(fill);
95256beaf04SBarry Smith   /* im is level for each filled value */
9530452661fSBarry Smith   im = (int *) PetscMalloc( (n+1)*sizeof(int));CHKPTRQ(im);
95456beaf04SBarry Smith   /* dloc is location of diagonal in factor */
9550452661fSBarry Smith   dloc = (int *) PetscMalloc( (n+1)*sizeof(int));CHKPTRQ(dloc);
95656beaf04SBarry Smith   dloc[0]  = 0;
95756beaf04SBarry Smith   for ( prow=0; prow<n; prow++ ) {
9586cf997b0SBarry Smith 
9596cf997b0SBarry Smith     /* copy current row into linked list */
96056beaf04SBarry Smith     nzf     = nz  = ai[r[prow]+1] - ai[r[prow]];
961385f7a74SSatish Balay     if (!nz) SETERRQ(PETSC_ERR_MAT_LU_ZRPVT,1,"Empty row in matrix");
962dbb450caSBarry Smith     xi      = aj + ai[r[prow]] + shift;
9639e25ed09SBarry Smith     fill[n]    = n;
964435faa5fSBarry Smith     fill[prow] = -1; /* marker to indicate if diagonal exists */
9659e25ed09SBarry Smith     while (nz--) {
9669e25ed09SBarry Smith       fm  = n;
967dbb450caSBarry Smith       idx = ic[*xi++ + shift];
9689e25ed09SBarry Smith       do {
9699e25ed09SBarry Smith         m  = fm;
9709e25ed09SBarry Smith         fm = fill[m];
9719e25ed09SBarry Smith       } while (fm < idx);
9729e25ed09SBarry Smith       fill[m]   = idx;
9739e25ed09SBarry Smith       fill[idx] = fm;
97456beaf04SBarry Smith       im[idx]   = 0;
9759e25ed09SBarry Smith     }
9766cf997b0SBarry Smith 
9776cf997b0SBarry Smith     /* make sure diagonal entry is included */
978435faa5fSBarry Smith     if (diagonal_fill && fill[prow] == -1) {
9796cf997b0SBarry Smith       fm = n;
980435faa5fSBarry Smith       while (fill[fm] < prow) fm = fill[fm];
981435faa5fSBarry Smith       fill[prow] = fill[fm]; /* insert diagonal into linked list */
982435faa5fSBarry Smith       fill[fm]   = prow;
9836cf997b0SBarry Smith       im[prow]   = 0;
9846cf997b0SBarry Smith       nzf++;
985335d9088SBarry Smith       dcount++;
9866cf997b0SBarry Smith     }
9876cf997b0SBarry Smith 
98856beaf04SBarry Smith     nzi = 0;
9899e25ed09SBarry Smith     row = fill[n];
99056beaf04SBarry Smith     while ( row < prow ) {
99156beaf04SBarry Smith       incrlev = im[row] + 1;
99256beaf04SBarry Smith       nz      = dloc[row];
9936cf997b0SBarry Smith       xi      = ajnew  + ainew[row] + shift + nz + 1;
994dbb450caSBarry Smith       flev    = ajfill + ainew[row] + shift + nz + 1;
995416022c9SBarry Smith       nnz     = ainew[row+1] - ainew[row] - nz - 1;
99656beaf04SBarry Smith       fm      = row;
99756beaf04SBarry Smith       while (nnz-- > 0) {
998dbb450caSBarry Smith         idx = *xi++ + shift;
99956beaf04SBarry Smith         if (*flev + incrlev > levels) {
100056beaf04SBarry Smith           flev++;
100156beaf04SBarry Smith           continue;
100256beaf04SBarry Smith         }
100356beaf04SBarry Smith         do {
10049e25ed09SBarry Smith           m  = fm;
10059e25ed09SBarry Smith           fm = fill[m];
100656beaf04SBarry Smith         } while (fm < idx);
10079e25ed09SBarry Smith         if (fm != idx) {
100856beaf04SBarry Smith           im[idx]   = *flev + incrlev;
10099e25ed09SBarry Smith           fill[m]   = idx;
10109e25ed09SBarry Smith           fill[idx] = fm;
10119e25ed09SBarry Smith           fm        = idx;
101256beaf04SBarry Smith           nzf++;
1013ecf371e4SBarry Smith         } else {
101456beaf04SBarry Smith           if (im[idx] > *flev + incrlev) im[idx] = *flev+incrlev;
10159e25ed09SBarry Smith         }
101656beaf04SBarry Smith         flev++;
10179e25ed09SBarry Smith       }
10189e25ed09SBarry Smith       row = fill[row];
101956beaf04SBarry Smith       nzi++;
10209e25ed09SBarry Smith     }
10219e25ed09SBarry Smith     /* copy new filled row into permanent storage */
102256beaf04SBarry Smith     ainew[prow+1] = ainew[prow] + nzf;
1023d7e8b826SBarry Smith     if (ainew[prow+1] > jmax-shift) {
1024ecf371e4SBarry Smith 
1025ecf371e4SBarry Smith       /* estimate how much additional space we will need */
1026ecf371e4SBarry Smith       /* use the strategy suggested by David Hysom <hysom@perch-t.icase.edu> */
1027ecf371e4SBarry Smith       /* just double the memory each time */
1028ecf371e4SBarry Smith       /*  maxadd = (int) ((f*(ai[n]+!shift)*(n-prow+5))/n); */
1029ecf371e4SBarry Smith       int maxadd = jmax;
103056beaf04SBarry Smith       if (maxadd < nzf) maxadd = (n-prow)*(nzf+1);
1031bbb6d6a8SBarry Smith       jmax += maxadd;
1032ecf371e4SBarry Smith 
1033ecf371e4SBarry Smith       /* allocate a longer ajnew and ajfill */
10340452661fSBarry Smith       xi     = (int *) PetscMalloc( jmax*sizeof(int) );CHKPTRQ(xi);
1035549d3d68SSatish Balay       ierr   = PetscMemcpy(xi,ajnew,(ainew[prow]+shift)*sizeof(int));CHKERRQ(ierr);
1036606d414cSSatish Balay       ierr = PetscFree(ajnew);CHKERRQ(ierr);
103756beaf04SBarry Smith       ajnew  = xi;
10380452661fSBarry Smith       xi     = (int *) PetscMalloc( jmax*sizeof(int) );CHKPTRQ(xi);
1039549d3d68SSatish Balay       ierr   = PetscMemcpy(xi,ajfill,(ainew[prow]+shift)*sizeof(int));CHKERRQ(ierr);
1040606d414cSSatish Balay       ierr = PetscFree(ajfill);CHKERRQ(ierr);
104156beaf04SBarry Smith       ajfill = xi;
1042ecf371e4SBarry Smith       realloc++; /* count how many times we realloc */
10439e25ed09SBarry Smith     }
1044dbb450caSBarry Smith     xi          = ajnew + ainew[prow] + shift;
1045dbb450caSBarry Smith     flev        = ajfill + ainew[prow] + shift;
104656beaf04SBarry Smith     dloc[prow]  = nzi;
10479e25ed09SBarry Smith     fm          = fill[n];
104856beaf04SBarry Smith     while (nzf--) {
1049dbb450caSBarry Smith       *xi++   = fm - shift;
105056beaf04SBarry Smith       *flev++ = im[fm];
10519e25ed09SBarry Smith       fm      = fill[fm];
10529e25ed09SBarry Smith     }
10536cf997b0SBarry Smith     /* make sure row has diagonal entry */
10546cf997b0SBarry Smith     if (ajnew[ainew[prow]+shift+dloc[prow]]+shift != prow) {
10556cf997b0SBarry Smith       SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,1,"Row %d has missing diagonal in factored matrix\n\
10566cf997b0SBarry Smith     try running with -pc_ilu_nonzeros_along_diagonal or -pc_ilu_diagonal_fill",prow);
10576cf997b0SBarry Smith     }
10589e25ed09SBarry Smith   }
1059606d414cSSatish Balay   ierr = PetscFree(ajfill); CHKERRQ(ierr);
1060898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
1061898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
1062606d414cSSatish Balay   ierr = PetscFree(fill);CHKERRQ(ierr);
1063606d414cSSatish Balay   ierr = PetscFree(im);CHKERRQ(ierr);
10649e25ed09SBarry Smith 
1065f64065bbSBarry Smith   {
1066464e8d44SSatish Balay     double af = ((double)ainew[n])/((double)ai[n]);
1067c38d4ed2SBarry Smith     PLogInfo(A,"MatILUFactorSymbolic_SeqAIJ:Reallocs %d Fill ratio:given %g needed %g\n",realloc,f,af);
1068981c4779SBarry Smith     PLogInfo(A,"MatILUFactorSymbolic_SeqAIJ:Run with -pc_ilu_fill %g or use \n",af);
1069981c4779SBarry Smith     PLogInfo(A,"MatILUFactorSymbolic_SeqAIJ:PCILUSetFill(pc,%g);\n",af);
1070981c4779SBarry Smith     PLogInfo(A,"MatILUFactorSymbolic_SeqAIJ:for best performance.\n");
1071335d9088SBarry Smith     if (diagonal_fill) {
1072335d9088SBarry Smith       PLogInfo(A,"MatILUFactorSymbolic_SeqAIJ:Detected and replace %d missing diagonals",dcount);
1073335d9088SBarry Smith     }
1074f64065bbSBarry Smith   }
1075bbb6d6a8SBarry Smith 
10769e25ed09SBarry Smith   /* put together the new matrix */
1077b4fd4287SBarry Smith   ierr = MatCreateSeqAIJ(A->comm,n,n,0,PETSC_NULL,fact);CHKERRQ(ierr);
1078fa6007c9SSatish Balay   PLogObjectParent(*fact,isicol);
1079416022c9SBarry Smith   b = (Mat_SeqAIJ *) (*fact)->data;
1080606d414cSSatish Balay   ierr = PetscFree(b->imax);CHKERRQ(ierr);
10817c922b88SBarry Smith   b->singlemalloc = PETSC_FALSE;
1082dbb450caSBarry Smith   len = (ainew[n] + shift)*sizeof(Scalar);
10839e25ed09SBarry Smith   /* the next line frees the default space generated by the Create() */
1084606d414cSSatish Balay   ierr = PetscFree(b->a);CHKERRQ(ierr);
1085606d414cSSatish Balay   ierr = PetscFree(b->ilen);CHKERRQ(ierr);
10866b96ef82SSatish Balay   b->a          = (Scalar *) PetscMalloc( len+1 );CHKPTRQ(b->a);
1087416022c9SBarry Smith   b->j          = ajnew;
1088416022c9SBarry Smith   b->i          = ainew;
108956beaf04SBarry Smith   for ( i=0; i<n; i++ ) dloc[i] += ainew[i];
1090416022c9SBarry Smith   b->diag       = dloc;
1091416022c9SBarry Smith   b->ilen       = 0;
1092416022c9SBarry Smith   b->imax       = 0;
1093416022c9SBarry Smith   b->row        = isrow;
1094416022c9SBarry Smith   b->col        = iscol;
1095c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
1096c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
109782bf6240SBarry Smith   b->icol       = isicol;
109882bf6240SBarry Smith   b->solve_work = (Scalar *) PetscMalloc( (n+1)*sizeof(Scalar));CHKPTRQ(b->solve_work);
1099416022c9SBarry Smith   /* In b structure:  Free imax, ilen, old a, old j.
110056beaf04SBarry Smith      Allocate dloc, solve_work, new a, new j */
1101dbb450caSBarry Smith   PLogObjectMemory(*fact,(ainew[n]+shift-n) * (sizeof(int)+sizeof(Scalar)));
1102416022c9SBarry Smith   b->maxnz          = b->nz = ainew[n] + shift;
11039e25ed09SBarry Smith   (*fact)->factor   = FACTOR_LU;
1104ae068f58SLois Curfman McInnes 
1105ae068f58SLois Curfman McInnes   (*fact)->info.factor_mallocs    = realloc;
1106ae068f58SLois Curfman McInnes   (*fact)->info.fill_ratio_given  = f;
1107ae068f58SLois Curfman McInnes   (*fact)->info.fill_ratio_needed = ((double)ainew[n])/((double)ai[prow]);
1108e93ccc4dSBarry Smith   (*fact)->factor                 =  FACTOR_LU;;
1109ae068f58SLois Curfman McInnes 
11103a40ed3dSBarry Smith   PetscFunctionReturn(0);
11119e25ed09SBarry Smith }
1112d5d45c9bSBarry Smith 
1113d5d45c9bSBarry Smith 
1114d5d45c9bSBarry Smith 
1115d5d45c9bSBarry Smith 
1116