xref: /petsc/src/mat/impls/aij/seq/aijfact.c (revision 9cc1f4e33c149b29186aeebaa37298fc478f8043)
1289bc588SBarry Smith 
270f55243SBarry Smith #include "src/mat/impls/aij/seq/aij.h"
31c4feecaSSatish Balay #include "src/inline/dot.h"
4ed480e8bSBarry Smith #include "src/inline/spops.h"
53b2fbd54SBarry Smith 
64a2ae208SSatish Balay #undef __FUNCT__
74a2ae208SSatish Balay #define __FUNCT__ "MatOrdering_Flow_SeqAIJ"
8dfbe8321SBarry Smith PetscErrorCode MatOrdering_Flow_SeqAIJ(Mat mat,const MatOrderingType type,IS *irow,IS *icol)
93b2fbd54SBarry Smith {
103a40ed3dSBarry Smith   PetscFunctionBegin;
113a40ed3dSBarry Smith 
1229bbc08cSBarry Smith   SETERRQ(PETSC_ERR_SUP,"Code not written");
13aa482453SBarry Smith #if !defined(PETSC_USE_DEBUG)
14d64ed03dSBarry Smith   PetscFunctionReturn(0);
15d64ed03dSBarry Smith #endif
163b2fbd54SBarry Smith }
173b2fbd54SBarry Smith 
1886bacbd2SBarry Smith 
19dfbe8321SBarry Smith EXTERN PetscErrorCode MatMarkDiagonal_SeqAIJ(Mat);
20dfbe8321SBarry Smith EXTERN PetscErrorCode Mat_AIJ_CheckInode(Mat,PetscTruth);
2186bacbd2SBarry Smith 
2238baddfdSBarry Smith EXTERN PetscErrorCode SPARSEKIT2dperm(PetscInt*,PetscScalar*,PetscInt*,PetscInt*,PetscScalar*,PetscInt*,PetscInt*,PetscInt*,PetscInt*,PetscInt*);
23*9cc1f4e3SBarry Smith EXTERN PetscErrorCode SPARSEKIT2ilutp(PetscInt*,PetscScalar*,PetscInt*,PetscInt*,PetscInt*,PetscReal,PetscReal*,PetscInt*,PetscScalar*,PetscInt*,PetscInt*,PetscInt*,PetscScalar*,PetscInt*,PetscInt*,PetscErrorCode*);
2438baddfdSBarry Smith EXTERN PetscErrorCode SPARSEKIT2msrcsr(PetscInt*,PetscScalar*,PetscInt*,PetscScalar*,PetscInt*,PetscInt*,PetscScalar*,PetscInt*);
2586bacbd2SBarry Smith 
264a2ae208SSatish Balay #undef __FUNCT__
274a2ae208SSatish Balay #define __FUNCT__ "MatILUDTFactor_SeqAIJ"
2886bacbd2SBarry Smith   /* ------------------------------------------------------------
2986bacbd2SBarry Smith 
3086bacbd2SBarry Smith           This interface was contribed by Tony Caola
3186bacbd2SBarry Smith 
3286bacbd2SBarry Smith      This routine is an interface to the pivoting drop-tolerance
335bd2ddc7SBarry Smith      ILU routine written by Yousef Saad (saad@cs.umn.edu) as part of
345bd2ddc7SBarry Smith      SPARSEKIT2.
355bd2ddc7SBarry Smith 
365bd2ddc7SBarry Smith      The SPARSEKIT2 routines used here are covered by the GNU
375bd2ddc7SBarry Smith      copyright; see the file gnu in this directory.
3886bacbd2SBarry Smith 
3986bacbd2SBarry Smith      Thanks to Prof. Saad, Dr. Hysom, and Dr. Smith for their
4086bacbd2SBarry Smith      help in getting this routine ironed out.
4186bacbd2SBarry Smith 
425bd2ddc7SBarry Smith      The major drawback to this routine is that if info->fill is
435bd2ddc7SBarry Smith      not large enough it fails rather than allocating more space;
445bd2ddc7SBarry Smith      this can be fixed by hacking/improving the f2c version of
455bd2ddc7SBarry Smith      Yousef Saad's code.
4686bacbd2SBarry Smith 
4786bacbd2SBarry Smith      ------------------------------------------------------------
4886bacbd2SBarry Smith */
49dfbe8321SBarry Smith PetscErrorCode MatILUDTFactor_SeqAIJ(Mat A,MatFactorInfo *info,IS isrow,IS iscol,Mat *fact)
5086bacbd2SBarry Smith {
5160ec86bdSBarry Smith #if defined(PETSC_AVOID_GNUCOPYRIGHT_CODE)
5260ec86bdSBarry Smith   PetscFunctionBegin;
53e005ede5SBarry Smith   SETERRQ(PETSC_ERR_SUP_SYS,"This distribution does not include GNU Copyright code\n\
5460ec86bdSBarry Smith   You can obtain the drop tolerance routines by installing PETSc from\n\
5560ec86bdSBarry Smith   www.mcs.anl.gov/petsc\n");
5660ec86bdSBarry Smith #else
5786bacbd2SBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data,*b;
5807d2056aSBarry Smith   IS             iscolf,isicol,isirow;
5986bacbd2SBarry Smith   PetscTruth     reorder;
60*9cc1f4e3SBarry Smith   PetscErrorCode ierr,sierr;
61*9cc1f4e3SBarry Smith   PetscInt       *c,*r,*ic,i,n = A->m;
6238baddfdSBarry Smith   PetscInt       *old_i = a->i,*old_j = a->j,*new_i,*old_i2 = 0,*old_j2 = 0,*new_j;
6338baddfdSBarry Smith   PetscInt       *ordcol,*iwk,*iperm,*jw;
6438baddfdSBarry Smith   PetscInt       jmax,lfill,job,*o_i,*o_j;
6587828ca2SBarry Smith   PetscScalar    *old_a = a->a,*w,*new_a,*old_a2 = 0,*wk,*o_a;
66329f5518SBarry Smith   PetscReal      permtol,af;
6786bacbd2SBarry Smith 
6886bacbd2SBarry Smith   PetscFunctionBegin;
6986bacbd2SBarry Smith 
7086bacbd2SBarry Smith   if (info->dt == PETSC_DEFAULT)      info->dt      = .005;
7138baddfdSBarry Smith   if (info->dtcount == PETSC_DEFAULT) info->dtcount = (PetscInt)(1.5*a->rmax);
7286bacbd2SBarry Smith   if (info->dtcol == PETSC_DEFAULT)   info->dtcol   = .01;
7333259db3SBarry Smith   if (info->fill == PETSC_DEFAULT)    info->fill    = ((double)(n*(info->dtcount+1)))/a->nz;
7438baddfdSBarry Smith   lfill   = (PetscInt)(info->dtcount/2.0);
7538baddfdSBarry Smith   jmax    = (PetscInt)(info->fill*a->nz);
7686bacbd2SBarry Smith   permtol = info->dtcol;
7786bacbd2SBarry Smith 
7886bacbd2SBarry Smith 
7986bacbd2SBarry Smith   /* ------------------------------------------------------------
8086bacbd2SBarry Smith      If reorder=.TRUE., then the original matrix has to be
8186bacbd2SBarry Smith      reordered to reflect the user selected ordering scheme, and
8286bacbd2SBarry Smith      then de-reordered so it is in it's original format.
8386bacbd2SBarry Smith      Because Saad's dperm() is NOT in place, we have to copy
8486bacbd2SBarry Smith      the original matrix and allocate more storage. . .
8586bacbd2SBarry Smith      ------------------------------------------------------------
8686bacbd2SBarry Smith   */
8786bacbd2SBarry Smith 
8886bacbd2SBarry Smith   /* set reorder to true if either isrow or iscol is not identity */
8986bacbd2SBarry Smith   ierr = ISIdentity(isrow,&reorder);CHKERRQ(ierr);
9086bacbd2SBarry Smith   if (reorder) {ierr = ISIdentity(iscol,&reorder);CHKERRQ(ierr);}
9186bacbd2SBarry Smith   reorder = PetscNot(reorder);
9286bacbd2SBarry Smith 
9386bacbd2SBarry Smith 
9486bacbd2SBarry Smith   /* storage for ilu factor */
9538baddfdSBarry Smith   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&new_i);CHKERRQ(ierr);
9638baddfdSBarry Smith   ierr = PetscMalloc(jmax*sizeof(PetscInt),&new_j);CHKERRQ(ierr);
9787828ca2SBarry Smith   ierr = PetscMalloc(jmax*sizeof(PetscScalar),&new_a);CHKERRQ(ierr);
9838baddfdSBarry Smith   ierr = PetscMalloc(n*sizeof(PetscInt),&ordcol);CHKERRQ(ierr);
9986bacbd2SBarry Smith 
10086bacbd2SBarry Smith   /* ------------------------------------------------------------
10186bacbd2SBarry Smith      Make sure that everything is Fortran formatted (1-Based)
10286bacbd2SBarry Smith      ------------------------------------------------------------
10386bacbd2SBarry Smith   */
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   };
110010a20caSHong Zhang 
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     }
11938baddfdSBarry Smith     ierr = PetscMalloc((n+1)*sizeof(PetscInt),&old_i2);CHKERRQ(ierr);
12038baddfdSBarry Smith     ierr = PetscMalloc((old_i[n]-old_i[0]+1)*sizeof(PetscInt),&old_j2);CHKERRQ(ierr);
12187828ca2SBarry Smith     ierr = PetscMalloc((old_i[n]-old_i[0]+1)*sizeof(PetscScalar),&old_a2);CHKERRQ(ierr);
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 
14338baddfdSBarry Smith   ierr = PetscMalloc(2*n*sizeof(PetscInt),&iperm);CHKERRQ(ierr);
14438baddfdSBarry Smith   ierr = PetscMalloc(2*n*sizeof(PetscInt),&jw);CHKERRQ(ierr);
14587828ca2SBarry Smith   ierr = PetscMalloc(n*sizeof(PetscScalar),&w);CHKERRQ(ierr);
14686bacbd2SBarry Smith 
1476849ba73SBarry Smith   SPARSEKIT2ilutp(&n,o_a,o_j,o_i,&lfill,(PetscReal)info->dt,&permtol,&n,new_a,new_j,new_i,&jmax,w,jw,iperm,&sierr);
1486849ba73SBarry Smith   if (sierr) {
1496849ba73SBarry Smith     switch (sierr) {
15077431f27SBarry Smith       case -3: SETERRQ2(PETSC_ERR_LIB,"ilutp(), matrix U overflows, need larger info->fill current fill %g space allocated %D",info->fill,jmax);
15177431f27SBarry Smith       case -2: SETERRQ2(PETSC_ERR_LIB,"ilutp(), matrix L overflows, need larger info->fill current fill %g space allocated %D",info->fill,jmax);
152e005ede5SBarry Smith       case -5: SETERRQ(PETSC_ERR_LIB,"ilutp(), zero row encountered");
153e005ede5SBarry Smith       case -1: SETERRQ(PETSC_ERR_LIB,"ilutp(), input matrix may be wrong");
15477431f27SBarry Smith       case -4: SETERRQ1(PETSC_ERR_LIB,"ilutp(), illegal info->fill value %D",jmax);
15577431f27SBarry Smith       default: SETERRQ1(PETSC_ERR_LIB,"ilutp(), zero pivot detected on row %D",sierr);
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 
16887828ca2SBarry Smith   ierr = PetscMalloc(n*sizeof(PetscScalar),&wk);CHKERRQ(ierr);
16938baddfdSBarry Smith   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&iwk);CHKERRQ(ierr);
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 */
18886bacbd2SBarry Smith   for (i=0; i<n+1; i++) {
189b19713cbSBarry Smith     old_i[i]--;
190b19713cbSBarry Smith   }
191b19713cbSBarry Smith   for (i=old_i[0];i<old_i[n];i++) {
192b19713cbSBarry Smith     old_j[i]--;
193b19713cbSBarry Smith   }
19486bacbd2SBarry Smith 
195b19713cbSBarry Smith   /* Make the factored matrix 0-based */
19686bacbd2SBarry Smith   for (i=0; i<n+1; i++) {
197b19713cbSBarry Smith     new_i[i]--;
198b19713cbSBarry Smith   }
199b19713cbSBarry Smith   for (i=new_i[0];i<new_i[n];i++) {
200b19713cbSBarry Smith     new_j[i]--;
201b19713cbSBarry Smith   }
20286bacbd2SBarry Smith 
20386bacbd2SBarry Smith   /*-- due to the pivoting, we need to reorder iscol to correctly --*/
20486bacbd2SBarry Smith   /*-- permute the right-hand-side and solution vectors           --*/
2054c49b128SBarry Smith   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
2064c49b128SBarry Smith   ierr = ISInvertPermutation(isrow,PETSC_DECIDE,&isirow);CHKERRQ(ierr);
207c0c2c81eSBarry Smith   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
20886bacbd2SBarry Smith   for(i=0; i<n; i++) {
20986bacbd2SBarry Smith     ordcol[i] = ic[iperm[i]-1];
21086bacbd2SBarry Smith   };
21186bacbd2SBarry Smith   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
21207d2056aSBarry Smith   ierr = ISDestroy(isicol);CHKERRQ(ierr);
21386bacbd2SBarry Smith 
214c0c2c81eSBarry Smith   ierr = PetscFree(iperm);CHKERRQ(ierr);
215c0c2c81eSBarry Smith 
216329f5518SBarry Smith   ierr = ISCreateGeneral(PETSC_COMM_SELF,n,ordcol,&iscolf);CHKERRQ(ierr);
21707d2056aSBarry Smith   ierr = PetscFree(ordcol);CHKERRQ(ierr);
21886bacbd2SBarry Smith 
21986bacbd2SBarry Smith   /*----- put together the new matrix -----*/
22086bacbd2SBarry Smith 
221f204ca49SKris Buschelman   ierr = MatCreate(A->comm,n,n,n,n,fact);CHKERRQ(ierr);
222f204ca49SKris Buschelman   ierr = MatSetType(*fact,A->type_name);CHKERRQ(ierr);
223f204ca49SKris Buschelman   ierr = MatSeqAIJSetPreallocation(*fact,0,PETSC_NULL);CHKERRQ(ierr);
22486bacbd2SBarry Smith   (*fact)->factor    = FACTOR_LU;
22586bacbd2SBarry Smith   (*fact)->assembled = PETSC_TRUE;
22686bacbd2SBarry Smith 
22786bacbd2SBarry Smith   b = (Mat_SeqAIJ*)(*fact)->data;
22886bacbd2SBarry Smith   ierr = PetscFree(b->imax);CHKERRQ(ierr);
22986bacbd2SBarry Smith   b->sorted        = PETSC_FALSE;
23007d2056aSBarry Smith   b->singlemalloc  = PETSC_FALSE;
231b19713cbSBarry Smith   /* the next line frees the default space generated by the MatCreate() */
23286bacbd2SBarry Smith   ierr             = PetscFree(b->a);CHKERRQ(ierr);
23386bacbd2SBarry Smith   ierr             = PetscFree(b->ilen);CHKERRQ(ierr);
23486bacbd2SBarry Smith   b->a             = new_a;
23586bacbd2SBarry Smith   b->j             = new_j;
23686bacbd2SBarry Smith   b->i             = new_i;
23786bacbd2SBarry Smith   b->ilen          = 0;
23886bacbd2SBarry Smith   b->imax          = 0;
2391f9e874aSBarry Smith   /*  I am not sure why these are the inverses of the row and column permutations; but the other way is NO GOOD */
240313ae024SBarry Smith   b->row           = isirow;
24186bacbd2SBarry Smith   b->col           = iscolf;
24287828ca2SBarry Smith   ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
24386bacbd2SBarry Smith   b->maxnz = b->nz = new_i[n];
24486bacbd2SBarry Smith   ierr = MatMarkDiagonal_SeqAIJ(*fact);CHKERRQ(ierr);
24586bacbd2SBarry Smith   (*fact)->info.factor_mallocs = 0;
24686bacbd2SBarry Smith 
24786bacbd2SBarry Smith   ierr = MatMarkDiagonal_SeqAIJ(A);CHKERRQ(ierr);
24886bacbd2SBarry Smith 
24986bacbd2SBarry Smith   /* check out for identical nodes. If found, use inode functions */
2503a7fca6bSBarry Smith   ierr = Mat_AIJ_CheckInode(*fact,PETSC_FALSE);CHKERRQ(ierr);
25186bacbd2SBarry Smith 
252431e710aSBarry Smith   af = ((double)b->nz)/((double)a->nz) + .001;
253b0a32e0cSBarry Smith   PetscLogInfo(A,"MatILUDTFactor_SeqAIJ:Fill ratio:given %g needed %g\n",info->fill,af);
254b0a32e0cSBarry Smith   PetscLogInfo(A,"MatILUDTFactor_SeqAIJ:Run with -pc_ilu_fill %g or use \n",af);
255b0a32e0cSBarry Smith   PetscLogInfo(A,"MatILUDTFactor_SeqAIJ:PCILUSetFill(pc,%g);\n",af);
256b0a32e0cSBarry Smith   PetscLogInfo(A,"MatILUDTFactor_SeqAIJ:for best performance.\n");
257431e710aSBarry Smith 
25886bacbd2SBarry Smith   PetscFunctionReturn(0);
25960ec86bdSBarry Smith #endif
26086bacbd2SBarry Smith }
26186bacbd2SBarry Smith 
262289bc588SBarry Smith /*
263289bc588SBarry Smith     Factorization code for AIJ format.
264289bc588SBarry Smith */
2654a2ae208SSatish Balay #undef __FUNCT__
266b9617806SBarry Smith #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ"
267dfbe8321SBarry Smith PetscErrorCode MatLUFactorSymbolic_SeqAIJ(Mat A,IS isrow,IS iscol,MatFactorInfo *info,Mat *B)
268289bc588SBarry Smith {
269416022c9SBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data,*b;
270289bc588SBarry Smith   IS             isicol;
2716849ba73SBarry Smith   PetscErrorCode ierr;
27238baddfdSBarry Smith   PetscInt       *r,*ic,i,n = A->m,*ai = a->i,*aj = a->j;
27338baddfdSBarry Smith   PetscInt       *ainew,*ajnew,jmax,*fill,*ajtmp,nz;
274418422e8SSatish Balay   PetscInt       *idnew,idx,row,m,fm,nnz,nzi,reallocs = 0,nzbd,*im;
2759e046878SBarry Smith   PetscReal      f;
276289bc588SBarry Smith 
2773a40ed3dSBarry Smith   PetscFunctionBegin;
27829bbc08cSBarry Smith   if (A->M != A->N) SETERRQ(PETSC_ERR_ARG_WRONG,"matrix must be square");
2794c49b128SBarry Smith   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
280ac121b3dSBarry Smith   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
281ac121b3dSBarry Smith   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
282289bc588SBarry Smith 
283289bc588SBarry Smith   /* get new row pointers */
28438baddfdSBarry Smith   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&ainew);CHKERRQ(ierr);
285010a20caSHong Zhang   ainew[0] = 0;
286289bc588SBarry Smith   /* don't know how many column pointers are needed so estimate */
287d3d32019SBarry Smith   f = info->fill;
28838baddfdSBarry Smith   jmax  = (PetscInt)(f*ai[n]+1);
28938baddfdSBarry Smith   ierr = PetscMalloc((jmax)*sizeof(PetscInt),&ajnew);CHKERRQ(ierr);
290289bc588SBarry Smith   /* fill is a linked list of nonzeros in active row */
29138baddfdSBarry Smith   ierr = PetscMalloc((2*n+1)*sizeof(PetscInt),&fill);CHKERRQ(ierr);
2922fb3ed76SBarry Smith   im   = fill + n + 1;
293289bc588SBarry Smith   /* idnew is location of diagonal in factor */
29438baddfdSBarry Smith   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&idnew);CHKERRQ(ierr);
295010a20caSHong Zhang   idnew[0] = 0;
296289bc588SBarry Smith 
297289bc588SBarry Smith   for (i=0; i<n; i++) {
298289bc588SBarry Smith     /* first copy previous fill into linked list */
299289bc588SBarry Smith     nnz     = nz    = ai[r[i]+1] - ai[r[i]];
30029bbc08cSBarry Smith     if (!nz) SETERRQ(PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix");
301010a20caSHong Zhang     ajtmp   = aj + ai[r[i]];
302289bc588SBarry Smith     fill[n] = n;
303289bc588SBarry Smith     while (nz--) {
304289bc588SBarry Smith       fm  = n;
305010a20caSHong Zhang       idx = ic[*ajtmp++];
306289bc588SBarry Smith       do {
307289bc588SBarry Smith         m  = fm;
308289bc588SBarry Smith         fm = fill[m];
309289bc588SBarry Smith       } while (fm < idx);
310289bc588SBarry Smith       fill[m]   = idx;
311289bc588SBarry Smith       fill[idx] = fm;
312289bc588SBarry Smith     }
313289bc588SBarry Smith     row = fill[n];
314289bc588SBarry Smith     while (row < i) {
315010a20caSHong Zhang       ajtmp = ajnew + idnew[row] + 1;
3162fb3ed76SBarry Smith       nzbd  = 1 + idnew[row] - ainew[row];
3172fb3ed76SBarry Smith       nz    = im[row] - nzbd;
318289bc588SBarry Smith       fm    = row;
3192fb3ed76SBarry Smith       while (nz-- > 0) {
320010a20caSHong Zhang         idx = *ajtmp++ ;
3212fb3ed76SBarry Smith         nzbd++;
3222fb3ed76SBarry Smith         if (idx == i) im[row] = nzbd;
323289bc588SBarry Smith         do {
324289bc588SBarry Smith           m  = fm;
325289bc588SBarry Smith           fm = fill[m];
326289bc588SBarry Smith         } while (fm < idx);
327289bc588SBarry Smith         if (fm != idx) {
328289bc588SBarry Smith           fill[m]   = idx;
329289bc588SBarry Smith           fill[idx] = fm;
330289bc588SBarry Smith           fm        = idx;
331289bc588SBarry Smith           nnz++;
332289bc588SBarry Smith         }
333289bc588SBarry Smith       }
334289bc588SBarry Smith       row = fill[row];
335289bc588SBarry Smith     }
336289bc588SBarry Smith     /* copy new filled row into permanent storage */
337289bc588SBarry Smith     ainew[i+1] = ainew[i] + nnz;
338496e697eSBarry Smith     if (ainew[i+1] > jmax) {
339ecf371e4SBarry Smith 
340ecf371e4SBarry Smith       /* estimate how much additional space we will need */
341ecf371e4SBarry Smith       /* use the strategy suggested by David Hysom <hysom@perch-t.icase.edu> */
342ecf371e4SBarry Smith       /* just double the memory each time */
34338baddfdSBarry Smith       PetscInt maxadd = jmax;
344ecf371e4SBarry Smith       /* maxadd = (int)((f*(ai[n]+(!shift))*(n-i+5))/n); */
345bbb6d6a8SBarry Smith       if (maxadd < nnz) maxadd = (n-i)*(nnz+1);
3462fb3ed76SBarry Smith       jmax += maxadd;
347ecf371e4SBarry Smith 
348ecf371e4SBarry Smith       /* allocate a longer ajnew */
34938baddfdSBarry Smith       ierr  = PetscMalloc(jmax*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr);
35038baddfdSBarry Smith       ierr  = PetscMemcpy(ajtmp,ajnew,(ainew[i])*sizeof(PetscInt));CHKERRQ(ierr);
351606d414cSSatish Balay       ierr  = PetscFree(ajnew);CHKERRQ(ierr);
352289bc588SBarry Smith       ajnew = ajtmp;
353418422e8SSatish Balay       reallocs++; /* count how many times we realloc */
354289bc588SBarry Smith     }
355010a20caSHong Zhang     ajtmp = ajnew + ainew[i];
356289bc588SBarry Smith     fm    = fill[n];
357289bc588SBarry Smith     nzi   = 0;
3582fb3ed76SBarry Smith     im[i] = nnz;
359289bc588SBarry Smith     while (nnz--) {
360289bc588SBarry Smith       if (fm < i) nzi++;
361010a20caSHong Zhang       *ajtmp++ = fm ;
362289bc588SBarry Smith       fm       = fill[fm];
363289bc588SBarry Smith     }
364289bc588SBarry Smith     idnew[i] = ainew[i] + nzi;
365289bc588SBarry Smith   }
366464e8d44SSatish Balay   if (ai[n] != 0) {
367329f5518SBarry Smith     PetscReal af = ((PetscReal)ainew[n])/((PetscReal)ai[n]);
368418422e8SSatish Balay     PetscLogInfo(A,"MatLUFactorSymbolic_SeqAIJ:Reallocs %D Fill ratio:given %g needed %g\n",reallocs,f,af);
369b0a32e0cSBarry Smith     PetscLogInfo(A,"MatLUFactorSymbolic_SeqAIJ:Run with -pc_lu_fill %g or use \n",af);
370b0a32e0cSBarry Smith     PetscLogInfo(A,"MatLUFactorSymbolic_SeqAIJ:PCLUSetFill(pc,%g);\n",af);
371b0a32e0cSBarry Smith     PetscLogInfo(A,"MatLUFactorSymbolic_SeqAIJ:for best performance.\n");
37205bf559cSSatish Balay   } else {
373b0a32e0cSBarry Smith     PetscLogInfo(A,"MatLUFactorSymbolic_SeqAIJ: Empty matrix\n");
37405bf559cSSatish Balay   }
3752fb3ed76SBarry Smith 
376898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
377898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
3781987afe7SBarry Smith 
379606d414cSSatish Balay   ierr = PetscFree(fill);CHKERRQ(ierr);
380289bc588SBarry Smith 
381289bc588SBarry Smith   /* put together the new matrix */
382f204ca49SKris Buschelman   ierr = MatCreate(A->comm,n,n,n,n,B);CHKERRQ(ierr);
383f204ca49SKris Buschelman   ierr = MatSetType(*B,A->type_name);CHKERRQ(ierr);
384f204ca49SKris Buschelman   ierr = MatSeqAIJSetPreallocation(*B,0,PETSC_NULL);CHKERRQ(ierr);
385b0a32e0cSBarry Smith   PetscLogObjectParent(*B,isicol);
386416022c9SBarry Smith   b    = (Mat_SeqAIJ*)(*B)->data;
387606d414cSSatish Balay   ierr = PetscFree(b->imax);CHKERRQ(ierr);
3887c922b88SBarry Smith   b->singlemalloc = PETSC_FALSE;
389e8d4e0b9SBarry Smith   /* the next line frees the default space generated by the Create() */
390606d414cSSatish Balay   ierr = PetscFree(b->a);CHKERRQ(ierr);
391606d414cSSatish Balay   ierr = PetscFree(b->ilen);CHKERRQ(ierr);
392010a20caSHong Zhang   ierr          = PetscMalloc((ainew[n]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);
393416022c9SBarry Smith   b->j          = ajnew;
394416022c9SBarry Smith   b->i          = ainew;
395416022c9SBarry Smith   b->diag       = idnew;
396416022c9SBarry Smith   b->ilen       = 0;
397416022c9SBarry Smith   b->imax       = 0;
398416022c9SBarry Smith   b->row        = isrow;
399416022c9SBarry Smith   b->col        = iscol;
400085256b3SBarry Smith   b->lu_damping        = info->damping;
40187828ca2SBarry Smith   b->lu_zeropivot      = info->zeropivot;
4022cea7109SBarry Smith   b->lu_shift          = info->shift;
4032cea7109SBarry Smith   b->lu_shift_fraction = info->shift_fraction;
404c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
405c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
40682bf6240SBarry Smith   b->icol       = isicol;
40787828ca2SBarry Smith   ierr          = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
408416022c9SBarry Smith   /* In b structure:  Free imax, ilen, old a, old j.
4097fd98bd6SLois Curfman McInnes      Allocate idnew, solve_work, new a, new j */
41038baddfdSBarry Smith   PetscLogObjectMemory(*B,(ainew[n]-n)*(sizeof(PetscInt)+sizeof(PetscScalar)));
411010a20caSHong Zhang   b->maxnz = b->nz = ainew[n] ;
4127fd98bd6SLois Curfman McInnes 
413329f5518SBarry Smith   (*B)->factor                 =  FACTOR_LU;
414418422e8SSatish Balay   (*B)->info.factor_mallocs    = reallocs;
415ae068f58SLois Curfman McInnes   (*B)->info.fill_ratio_given  = f;
4163a7fca6bSBarry Smith   ierr = Mat_AIJ_CheckInode(*B,PETSC_FALSE);CHKERRQ(ierr);
4177dda7e10SBarry Smith   (*B)->ops->lufactornumeric   =  A->ops->lufactornumeric; /* Use Inode variant ONLY if A has inodes */
418703deb49SSatish Balay 
419e93ccc4dSBarry Smith   if (ai[n] != 0) {
420329f5518SBarry Smith     (*B)->info.fill_ratio_needed = ((PetscReal)ainew[n])/((PetscReal)ai[n]);
421eea2913aSSatish Balay   } else {
422eea2913aSSatish Balay     (*B)->info.fill_ratio_needed = 0.0;
423eea2913aSSatish Balay   }
4243a40ed3dSBarry Smith   PetscFunctionReturn(0);
425289bc588SBarry Smith }
4260a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
427dfbe8321SBarry Smith EXTERN PetscErrorCode Mat_AIJ_CheckInode(Mat,PetscTruth);
42841c01911SSatish Balay 
4294a2ae208SSatish Balay #undef __FUNCT__
4304a2ae208SSatish Balay #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ"
431dfbe8321SBarry Smith PetscErrorCode MatLUFactorNumeric_SeqAIJ(Mat A,Mat *B)
432289bc588SBarry Smith {
433416022c9SBarry Smith   Mat            C = *B;
434416022c9SBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data,*b = (Mat_SeqAIJ *)C->data;
43582bf6240SBarry Smith   IS             isrow = b->row,isicol = b->icol;
4366849ba73SBarry Smith   PetscErrorCode ierr;
43738baddfdSBarry Smith   PetscInt       *r,*ic,i,j,n = A->m,*ai = b->i,*aj = b->j;
43838baddfdSBarry Smith   PetscInt       *ajtmpold,*ajtmp,nz,row,*ics;
43938baddfdSBarry Smith   PetscInt       *diag_offset = b->diag,diag,*pj,ndamp = 0, nshift=0;
44087828ca2SBarry Smith   PetscScalar    *rtmp,*v,*pc,multiplier,*pv,*rtmps;
4410cf777b8SBarry Smith   PetscReal      damping = b->lu_damping, zeropivot = b->lu_zeropivot,rs,d;
4420cf777b8SBarry Smith   PetscReal      row_shift,shift_fraction,shift_amount,shift_lo=0., shift_hi=1., shift_top=0.;
4430cf777b8SBarry Smith   PetscTruth     damp,lushift;
444289bc588SBarry Smith 
4453a40ed3dSBarry Smith   PetscFunctionBegin;
44678b31e54SBarry Smith   ierr  = ISGetIndices(isrow,&r);CHKERRQ(ierr);
44778b31e54SBarry Smith   ierr  = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
44887828ca2SBarry Smith   ierr  = PetscMalloc((n+1)*sizeof(PetscScalar),&rtmp);CHKERRQ(ierr);
44987828ca2SBarry Smith   ierr  = PetscMemzero(rtmp,(n+1)*sizeof(PetscScalar));CHKERRQ(ierr);
450010a20caSHong Zhang   rtmps = rtmp; ics = ic;
451289bc588SBarry Smith 
4526cc28720Svictorle   if (!a->diag) {
4530cf777b8SBarry Smith     ierr = MatMarkDiagonal_SeqAIJ(A);CHKERRQ(ierr);
4540cf777b8SBarry Smith   }
4550cf777b8SBarry Smith 
4560cf777b8SBarry Smith   if (b->lu_shift) { /* set max shift */
45738baddfdSBarry Smith     PetscInt *aai = a->i,*ddiag = a->diag;
4580cf777b8SBarry Smith     shift_top = 0;
4596cc28720Svictorle     for (i=0; i<n; i++) {
460010a20caSHong Zhang       d = PetscAbsScalar((a->a)[ddiag[i]]);
4616cc28720Svictorle       /* calculate amt of shift needed for this row */
4626c037d1bSvictorle       if (d<=0) {
4633aeef889SHong Zhang         row_shift = 0;
4643aeef889SHong Zhang       } else {
4653aeef889SHong Zhang         row_shift = -2*d;
4663aeef889SHong Zhang       }
467010a20caSHong Zhang       v = a->a+aai[i];
4688a2e2d9cSvictorle       for (j=0; j<aai[i+1]-aai[i]; j++)
4696cc28720Svictorle 	row_shift += PetscAbsScalar(v[j]);
4706cc28720Svictorle       if (row_shift>shift_top) shift_top = row_shift;
4716cc28720Svictorle     }
4726cc28720Svictorle   }
4736cc28720Svictorle 
4746cc28720Svictorle   shift_fraction = 0; shift_amount = 0;
475085256b3SBarry Smith   do {
4768a5eb818SBarry Smith     damp    = PETSC_FALSE;
4776cc28720Svictorle     lushift = PETSC_FALSE;
478289bc588SBarry Smith     for (i=0; i<n; i++) {
479289bc588SBarry Smith       nz    = ai[i+1] - ai[i];
480010a20caSHong Zhang       ajtmp = aj + ai[i];
48148e94559SBarry Smith       for  (j=0; j<nz; j++) rtmps[ajtmp[j]] = 0.0;
482289bc588SBarry Smith 
483289bc588SBarry Smith       /* load in initial (unfactored row) */
484416022c9SBarry Smith       nz       = a->i[r[i]+1] - a->i[r[i]];
485010a20caSHong Zhang       ajtmpold = a->j + a->i[r[i]];
486010a20caSHong Zhang       v        = a->a + a->i[r[i]];
487085256b3SBarry Smith       for (j=0; j<nz; j++) {
488085256b3SBarry Smith         rtmp[ics[ajtmpold[j]]] = v[j];
489085256b3SBarry Smith       }
490e2dd4fc4Svictorle       rtmp[ics[r[i]]] += damping + shift_amount; /* damp the diagonal of the matrix */
491289bc588SBarry Smith 
492010a20caSHong Zhang       row = *ajtmp++ ;
493f2582111SSatish Balay       while  (row < i) {
4948c37ef55SBarry Smith         pc = rtmp + row;
4958c37ef55SBarry Smith         if (*pc != 0.0) {
496010a20caSHong Zhang           pv         = b->a + diag_offset[row];
497010a20caSHong Zhang           pj         = b->j + diag_offset[row] + 1;
49835aab85fSBarry Smith           multiplier = *pc / *pv++;
4998c37ef55SBarry Smith           *pc        = multiplier;
500f2582111SSatish Balay           nz         = ai[row+1] - diag_offset[row] - 1;
501f2582111SSatish Balay           for (j=0; j<nz; j++) rtmps[pj[j]] -= multiplier * pv[j];
502b0a32e0cSBarry Smith           PetscLogFlops(2*nz);
503289bc588SBarry Smith         }
504010a20caSHong Zhang         row = *ajtmp++;
505289bc588SBarry Smith       }
506416022c9SBarry Smith       /* finished row so stick it into b->a */
507010a20caSHong Zhang       pv   = b->a + ai[i] ;
508010a20caSHong Zhang       pj   = b->j + ai[i] ;
5098c37ef55SBarry Smith       nz   = ai[i+1] - ai[i];
51035aab85fSBarry Smith       diag = diag_offset[i] - ai[i];
511d3d32019SBarry Smith       /* 9/13/02 Victor Eijkhout suggested scaling zeropivot by rs for matrices with funny scalings */
512d3d32019SBarry Smith       rs   = 0.0;
513d3d32019SBarry Smith       for (j=0; j<nz; j++) {
514d3d32019SBarry Smith         pv[j] = rtmps[pj[j]];
515d3d32019SBarry Smith         if (j != diag) rs += PetscAbsScalar(pv[j]);
516d3d32019SBarry Smith       }
5176cc28720Svictorle #define MAX_NSHIFT 5
518b2da817dSvictorle       if (PetscRealPart(pv[diag]) <= zeropivot*rs && b->lu_shift) {
5196cc28720Svictorle 	if (nshift>MAX_NSHIFT) {
520e005ede5SBarry Smith 	  SETERRQ(PETSC_ERR_CONV_FAILED,"Unable to determine shift to enforce positive definite preconditioner");
5216cc28720Svictorle 	} else if (nshift==MAX_NSHIFT) {
5226cc28720Svictorle 	  shift_fraction = shift_hi;
5236c037d1bSvictorle 	  lushift      = PETSC_FALSE;
5246cc28720Svictorle 	} else {
5256cc28720Svictorle 	  shift_lo = shift_fraction; shift_fraction = (shift_hi+shift_lo)/2.;
5266c037d1bSvictorle 	  lushift      = PETSC_TRUE;
5276cc28720Svictorle 	}
5286cc28720Svictorle 	shift_amount = shift_fraction * shift_top;
5290cf777b8SBarry Smith         nshift++;
5300cf777b8SBarry Smith         break;
5316cc28720Svictorle       }
5329e29f15eSvictorle       if (PetscAbsScalar(pv[diag]) <= zeropivot*rs) {
533d3d32019SBarry Smith         if (damping) {
5348a5eb818SBarry Smith           if (ndamp) damping *= 2.0;
535085256b3SBarry Smith           damp = PETSC_TRUE;
536085256b3SBarry Smith           ndamp++;
537085256b3SBarry Smith           break;
538085256b3SBarry Smith         } else {
53977431f27SBarry Smith           SETERRQ4(PETSC_ERR_MAT_LU_ZRPVT,"Zero pivot row %D value %g tolerance %g * rs %g",i,PetscAbsScalar(pv[diag]),zeropivot,rs);
540d708749eSBarry Smith         }
54135aab85fSBarry Smith       }
54235aab85fSBarry Smith     }
5436cc28720Svictorle     if (!lushift && b->lu_shift && shift_fraction>0 && nshift<MAX_NSHIFT) {
5446cc28720Svictorle       /*
5456c037d1bSvictorle        * if no shift in this attempt & shifting & started shifting & can refine,
5466cc28720Svictorle        * then try lower shift
5476cc28720Svictorle        */
5480cf777b8SBarry Smith       shift_hi       = shift_fraction;
5490cf777b8SBarry Smith       shift_fraction = (shift_hi+shift_lo)/2.;
5506cc28720Svictorle       shift_amount   = shift_fraction * shift_top;
5510cf777b8SBarry Smith       lushift        = PETSC_TRUE;
5520cf777b8SBarry Smith       nshift++;
5536cc28720Svictorle     }
5546cc28720Svictorle   } while (damp || lushift);
5550f11f9aeSSatish Balay 
55635aab85fSBarry Smith   /* invert diagonal entries for simplier triangular solves */
55735aab85fSBarry Smith   for (i=0; i<n; i++) {
558010a20caSHong Zhang     b->a[diag_offset[i]] = 1.0/b->a[diag_offset[i]];
55935aab85fSBarry Smith   }
56035aab85fSBarry Smith 
561606d414cSSatish Balay   ierr = PetscFree(rtmp);CHKERRQ(ierr);
56278b31e54SBarry Smith   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
56378b31e54SBarry Smith   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
564416022c9SBarry Smith   C->factor = FACTOR_LU;
5657dda7e10SBarry Smith   (*B)->ops->lufactornumeric   =  A->ops->lufactornumeric; /* Use Inode variant ONLY if A has inodes */
566c456f294SBarry Smith   C->assembled = PETSC_TRUE;
567b0a32e0cSBarry Smith   PetscLogFlops(C->n);
568d3d32019SBarry Smith   if (ndamp) {
56977431f27SBarry Smith     PetscLogInfo(0,"MatLUFactorNumerical_SeqAIJ: number of damping tries %D damping value %g\n",ndamp,damping);
570085256b3SBarry Smith   }
5716cc28720Svictorle   if (nshift) {
5726cc28720Svictorle     b->lu_shift_fraction = shift_fraction;
57377431f27SBarry Smith     PetscLogInfo(0,"MatLUFactorNumerical_SeqAIJ: diagonal shifted up by %e fraction top_value %e number shifts %D\n",shift_fraction,shift_top,nshift);
5746cc28720Svictorle   }
5753a40ed3dSBarry Smith   PetscFunctionReturn(0);
576289bc588SBarry Smith }
5770889b5dcSvictorle 
5780889b5dcSvictorle #undef __FUNCT__
5790889b5dcSvictorle #define __FUNCT__ "MatUsePETSc_SeqAIJ"
580dfbe8321SBarry Smith PetscErrorCode MatUsePETSc_SeqAIJ(Mat A)
5810889b5dcSvictorle {
5820889b5dcSvictorle   PetscFunctionBegin;
5830889b5dcSvictorle   A->ops->lufactorsymbolic = MatLUFactorSymbolic_SeqAIJ;
5840889b5dcSvictorle   A->ops->lufactornumeric  = MatLUFactorNumeric_SeqAIJ;
5850889b5dcSvictorle   PetscFunctionReturn(0);
5860889b5dcSvictorle }
5870889b5dcSvictorle 
5880889b5dcSvictorle 
5890a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
5904a2ae208SSatish Balay #undef __FUNCT__
5914a2ae208SSatish Balay #define __FUNCT__ "MatLUFactor_SeqAIJ"
592dfbe8321SBarry Smith PetscErrorCode MatLUFactor_SeqAIJ(Mat A,IS row,IS col,MatFactorInfo *info)
593da3a660dSBarry Smith {
594dfbe8321SBarry Smith   PetscErrorCode ierr;
595416022c9SBarry Smith   Mat            C;
596416022c9SBarry Smith 
5973a40ed3dSBarry Smith   PetscFunctionBegin;
5989e046878SBarry Smith   ierr = MatLUFactorSymbolic(A,row,col,info,&C);CHKERRQ(ierr);
599f2582111SSatish Balay   ierr = MatLUFactorNumeric(A,&C);CHKERRQ(ierr);
600273d9f13SBarry Smith   ierr = MatHeaderCopy(A,C);CHKERRQ(ierr);
601440f4c53SSatish Balay   PetscLogObjectParent(A,((Mat_SeqAIJ*)(A->data))->icol);
6023a40ed3dSBarry Smith   PetscFunctionReturn(0);
603da3a660dSBarry Smith }
6040a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
6054a2ae208SSatish Balay #undef __FUNCT__
6064a2ae208SSatish Balay #define __FUNCT__ "MatSolve_SeqAIJ"
607dfbe8321SBarry Smith PetscErrorCode MatSolve_SeqAIJ(Mat A,Vec bb,Vec xx)
6088c37ef55SBarry Smith {
609416022c9SBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
610416022c9SBarry Smith   IS             iscol = a->col,isrow = a->row;
6116849ba73SBarry Smith   PetscErrorCode ierr;
61238baddfdSBarry Smith   PetscInt       *r,*c,i, n = A->m,*vi,*ai = a->i,*aj = a->j;
61338baddfdSBarry Smith   PetscInt       nz,*rout,*cout;
61487828ca2SBarry Smith   PetscScalar    *x,*b,*tmp,*tmps,*aa = a->a,sum,*v;
6158c37ef55SBarry Smith 
6163a40ed3dSBarry Smith   PetscFunctionBegin;
6173a40ed3dSBarry Smith   if (!n) PetscFunctionReturn(0);
61891bf9adeSBarry Smith 
6191ebc52fbSHong Zhang   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
6201ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
621416022c9SBarry Smith   tmp  = a->solve_work;
6228c37ef55SBarry Smith 
6233b2fbd54SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
6243b2fbd54SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
6258c37ef55SBarry Smith 
6268c37ef55SBarry Smith   /* forward solve the lower triangular */
6278c37ef55SBarry Smith   tmp[0] = b[*r++];
628010a20caSHong Zhang   tmps   = tmp;
6298c37ef55SBarry Smith   for (i=1; i<n; i++) {
630010a20caSHong Zhang     v   = aa + ai[i] ;
631010a20caSHong Zhang     vi  = aj + ai[i] ;
63253e7425aSBarry Smith     nz  = a->diag[i] - ai[i];
63353e7425aSBarry Smith     sum = b[*r++];
634ed480e8bSBarry Smith     SPARSEDENSEMDOT(sum,tmps,v,vi,nz);
6358c37ef55SBarry Smith     tmp[i] = sum;
6368c37ef55SBarry Smith   }
6378c37ef55SBarry Smith 
6388c37ef55SBarry Smith   /* backward solve the upper triangular */
6398c37ef55SBarry Smith   for (i=n-1; i>=0; i--){
640010a20caSHong Zhang     v   = aa + a->diag[i] + 1;
641010a20caSHong Zhang     vi  = aj + a->diag[i] + 1;
642416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
6438c37ef55SBarry Smith     sum = tmp[i];
644ed480e8bSBarry Smith     SPARSEDENSEMDOT(sum,tmps,v,vi,nz);
645010a20caSHong Zhang     x[*c--] = tmp[i] = sum*aa[a->diag[i]];
6468c37ef55SBarry Smith   }
6478c37ef55SBarry Smith 
6483b2fbd54SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
6493b2fbd54SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
6501ebc52fbSHong Zhang   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
6511ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
652b0a32e0cSBarry Smith   PetscLogFlops(2*a->nz - A->n);
6533a40ed3dSBarry Smith   PetscFunctionReturn(0);
6548c37ef55SBarry Smith }
655026e39d0SSatish Balay 
656930ae53cSBarry Smith /* ----------------------------------------------------------- */
6574a2ae208SSatish Balay #undef __FUNCT__
6584a2ae208SSatish Balay #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering"
659dfbe8321SBarry Smith PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering(Mat A,Vec bb,Vec xx)
660930ae53cSBarry Smith {
661930ae53cSBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
6626849ba73SBarry Smith   PetscErrorCode ierr;
66338baddfdSBarry Smith   PetscInt       n = A->m,*ai = a->i,*aj = a->j,*adiag = a->diag;
664362ced78SSatish Balay   PetscScalar    *x,*b,*aa = a->a;
665aa482453SBarry Smith #if !defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ)
66638baddfdSBarry Smith   PetscInt       adiag_i,i,*vi,nz,ai_i;
667362ced78SSatish Balay   PetscScalar    *v,sum;
668d85afc42SSatish Balay #endif
669930ae53cSBarry Smith 
6703a40ed3dSBarry Smith   PetscFunctionBegin;
6713a40ed3dSBarry Smith   if (!n) PetscFunctionReturn(0);
672930ae53cSBarry Smith 
6731ebc52fbSHong Zhang   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
6741ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
675930ae53cSBarry Smith 
676aa482453SBarry Smith #if defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ)
6771c4feecaSSatish Balay   fortransolveaij_(&n,x,ai,aj,adiag,aa,b);
6781c4feecaSSatish Balay #else
679930ae53cSBarry Smith   /* forward solve the lower triangular */
680930ae53cSBarry Smith   x[0] = b[0];
681930ae53cSBarry Smith   for (i=1; i<n; i++) {
682930ae53cSBarry Smith     ai_i = ai[i];
683930ae53cSBarry Smith     v    = aa + ai_i;
684930ae53cSBarry Smith     vi   = aj + ai_i;
685930ae53cSBarry Smith     nz   = adiag[i] - ai_i;
686930ae53cSBarry Smith     sum  = b[i];
687930ae53cSBarry Smith     while (nz--) sum -= *v++ * x[*vi++];
688930ae53cSBarry Smith     x[i] = sum;
689930ae53cSBarry Smith   }
690930ae53cSBarry Smith 
691930ae53cSBarry Smith   /* backward solve the upper triangular */
692930ae53cSBarry Smith   for (i=n-1; i>=0; i--){
693930ae53cSBarry Smith     adiag_i = adiag[i];
694d708749eSBarry Smith     v       = aa + adiag_i + 1;
695d708749eSBarry Smith     vi      = aj + adiag_i + 1;
696930ae53cSBarry Smith     nz      = ai[i+1] - adiag_i - 1;
697930ae53cSBarry Smith     sum     = x[i];
698930ae53cSBarry Smith     while (nz--) sum -= *v++ * x[*vi++];
699930ae53cSBarry Smith     x[i]    = sum*aa[adiag_i];
700930ae53cSBarry Smith   }
7011c4feecaSSatish Balay #endif
702b0a32e0cSBarry Smith   PetscLogFlops(2*a->nz - A->n);
7031ebc52fbSHong Zhang   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
7041ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
7053a40ed3dSBarry Smith   PetscFunctionReturn(0);
706930ae53cSBarry Smith }
707930ae53cSBarry Smith 
7084a2ae208SSatish Balay #undef __FUNCT__
7094a2ae208SSatish Balay #define __FUNCT__ "MatSolveAdd_SeqAIJ"
710dfbe8321SBarry Smith PetscErrorCode MatSolveAdd_SeqAIJ(Mat A,Vec bb,Vec yy,Vec xx)
711da3a660dSBarry Smith {
712416022c9SBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
713416022c9SBarry Smith   IS             iscol = a->col,isrow = a->row;
7146849ba73SBarry Smith   PetscErrorCode ierr;
71538baddfdSBarry Smith   PetscInt       *r,*c,i, n = A->m,*vi,*ai = a->i,*aj = a->j;
71638baddfdSBarry Smith   PetscInt       nz,*rout,*cout;
71787828ca2SBarry Smith   PetscScalar    *x,*b,*tmp,*aa = a->a,sum,*v;
718da3a660dSBarry Smith 
7193a40ed3dSBarry Smith   PetscFunctionBegin;
72078b31e54SBarry Smith   if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);}
721da3a660dSBarry Smith 
7221ebc52fbSHong Zhang   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
7231ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
724416022c9SBarry Smith   tmp  = a->solve_work;
725da3a660dSBarry Smith 
7263b2fbd54SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
7273b2fbd54SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
728da3a660dSBarry Smith 
729da3a660dSBarry Smith   /* forward solve the lower triangular */
730da3a660dSBarry Smith   tmp[0] = b[*r++];
731da3a660dSBarry Smith   for (i=1; i<n; i++) {
732010a20caSHong Zhang     v   = aa + ai[i] ;
733010a20caSHong Zhang     vi  = aj + ai[i] ;
734416022c9SBarry Smith     nz  = a->diag[i] - ai[i];
735da3a660dSBarry Smith     sum = b[*r++];
736010a20caSHong Zhang     while (nz--) sum -= *v++ * tmp[*vi++ ];
737da3a660dSBarry Smith     tmp[i] = sum;
738da3a660dSBarry Smith   }
739da3a660dSBarry Smith 
740da3a660dSBarry Smith   /* backward solve the upper triangular */
741da3a660dSBarry Smith   for (i=n-1; i>=0; i--){
742010a20caSHong Zhang     v   = aa + a->diag[i] + 1;
743010a20caSHong Zhang     vi  = aj + a->diag[i] + 1;
744416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
745da3a660dSBarry Smith     sum = tmp[i];
746010a20caSHong Zhang     while (nz--) sum -= *v++ * tmp[*vi++ ];
747010a20caSHong Zhang     tmp[i] = sum*aa[a->diag[i]];
748da3a660dSBarry Smith     x[*c--] += tmp[i];
749da3a660dSBarry Smith   }
750da3a660dSBarry Smith 
7513b2fbd54SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
7523b2fbd54SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
7531ebc52fbSHong Zhang   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
7541ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
755b0a32e0cSBarry Smith   PetscLogFlops(2*a->nz);
756898183ecSLois Curfman McInnes 
7573a40ed3dSBarry Smith   PetscFunctionReturn(0);
758da3a660dSBarry Smith }
759da3a660dSBarry Smith /* -------------------------------------------------------------------*/
7604a2ae208SSatish Balay #undef __FUNCT__
7614a2ae208SSatish Balay #define __FUNCT__ "MatSolveTranspose_SeqAIJ"
762dfbe8321SBarry Smith PetscErrorCode MatSolveTranspose_SeqAIJ(Mat A,Vec bb,Vec xx)
763da3a660dSBarry Smith {
764416022c9SBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
7652235e667SBarry Smith   IS             iscol = a->col,isrow = a->row;
7666849ba73SBarry Smith   PetscErrorCode ierr;
76738baddfdSBarry Smith   PetscInt       *r,*c,i,n = A->m,*vi,*ai = a->i,*aj = a->j;
76838baddfdSBarry Smith   PetscInt       nz,*rout,*cout,*diag = a->diag;
76987828ca2SBarry Smith   PetscScalar    *x,*b,*tmp,*aa = a->a,*v,s1;
770da3a660dSBarry Smith 
7713a40ed3dSBarry Smith   PetscFunctionBegin;
7721ebc52fbSHong Zhang   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
7731ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
774416022c9SBarry Smith   tmp  = a->solve_work;
775da3a660dSBarry Smith 
7762235e667SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
7772235e667SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
778da3a660dSBarry Smith 
779da3a660dSBarry Smith   /* copy the b into temp work space according to permutation */
7802235e667SBarry Smith   for (i=0; i<n; i++) tmp[i] = b[c[i]];
781da3a660dSBarry Smith 
782da3a660dSBarry Smith   /* forward solve the U^T */
783da3a660dSBarry Smith   for (i=0; i<n; i++) {
784010a20caSHong Zhang     v   = aa + diag[i] ;
785010a20caSHong Zhang     vi  = aj + diag[i] + 1;
786f1af5d2fSBarry Smith     nz  = ai[i+1] - diag[i] - 1;
787c38d4ed2SBarry Smith     s1  = tmp[i];
788ef66eb69SBarry Smith     s1 *= (*v++);  /* multiply by inverse of diagonal entry */
789da3a660dSBarry Smith     while (nz--) {
790010a20caSHong Zhang       tmp[*vi++ ] -= (*v++)*s1;
791da3a660dSBarry Smith     }
792c38d4ed2SBarry Smith     tmp[i] = s1;
793da3a660dSBarry Smith   }
794da3a660dSBarry Smith 
795da3a660dSBarry Smith   /* backward solve the L^T */
796da3a660dSBarry Smith   for (i=n-1; i>=0; i--){
797010a20caSHong Zhang     v   = aa + diag[i] - 1 ;
798010a20caSHong Zhang     vi  = aj + diag[i] - 1 ;
799f1af5d2fSBarry Smith     nz  = diag[i] - ai[i];
800f1af5d2fSBarry Smith     s1  = tmp[i];
801da3a660dSBarry Smith     while (nz--) {
802010a20caSHong Zhang       tmp[*vi-- ] -= (*v--)*s1;
803da3a660dSBarry Smith     }
804da3a660dSBarry Smith   }
805da3a660dSBarry Smith 
806da3a660dSBarry Smith   /* copy tmp into x according to permutation */
807da3a660dSBarry Smith   for (i=0; i<n; i++) x[r[i]] = tmp[i];
808da3a660dSBarry Smith 
8092235e667SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
8102235e667SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
8111ebc52fbSHong Zhang   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
8121ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
813da3a660dSBarry Smith 
814b0a32e0cSBarry Smith   PetscLogFlops(2*a->nz-A->n);
8153a40ed3dSBarry Smith   PetscFunctionReturn(0);
816da3a660dSBarry Smith }
817da3a660dSBarry Smith 
8184a2ae208SSatish Balay #undef __FUNCT__
8194a2ae208SSatish Balay #define __FUNCT__ "MatSolveTransposeAdd_SeqAIJ"
820dfbe8321SBarry Smith PetscErrorCode MatSolveTransposeAdd_SeqAIJ(Mat A,Vec bb,Vec zz,Vec xx)
821da3a660dSBarry Smith {
822416022c9SBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
8232235e667SBarry Smith   IS             iscol = a->col,isrow = a->row;
8246849ba73SBarry Smith   PetscErrorCode ierr;
82538baddfdSBarry Smith   PetscInt       *r,*c,i,n = A->m,*vi,*ai = a->i,*aj = a->j;
82638baddfdSBarry Smith   PetscInt       nz,*rout,*cout,*diag = a->diag;
82787828ca2SBarry Smith   PetscScalar    *x,*b,*tmp,*aa = a->a,*v;
8286abc6512SBarry Smith 
8293a40ed3dSBarry Smith   PetscFunctionBegin;
83023bc6035SMatthew Knepley   if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);}
8316abc6512SBarry Smith 
8321ebc52fbSHong Zhang   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
8331ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
834416022c9SBarry Smith   tmp = a->solve_work;
8356abc6512SBarry Smith 
8362235e667SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
8372235e667SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
8386abc6512SBarry Smith 
8396abc6512SBarry Smith   /* copy the b into temp work space according to permutation */
8402235e667SBarry Smith   for (i=0; i<n; i++) tmp[i] = b[c[i]];
8416abc6512SBarry Smith 
8426abc6512SBarry Smith   /* forward solve the U^T */
8436abc6512SBarry Smith   for (i=0; i<n; i++) {
844010a20caSHong Zhang     v   = aa + diag[i] ;
845010a20caSHong Zhang     vi  = aj + diag[i] + 1;
846f1af5d2fSBarry Smith     nz  = ai[i+1] - diag[i] - 1;
8476abc6512SBarry Smith     tmp[i] *= *v++;
8486abc6512SBarry Smith     while (nz--) {
849010a20caSHong Zhang       tmp[*vi++ ] -= (*v++)*tmp[i];
8506abc6512SBarry Smith     }
8516abc6512SBarry Smith   }
8526abc6512SBarry Smith 
8536abc6512SBarry Smith   /* backward solve the L^T */
8546abc6512SBarry Smith   for (i=n-1; i>=0; i--){
855010a20caSHong Zhang     v   = aa + diag[i] - 1 ;
856010a20caSHong Zhang     vi  = aj + diag[i] - 1 ;
857f1af5d2fSBarry Smith     nz  = diag[i] - ai[i];
8586abc6512SBarry Smith     while (nz--) {
859010a20caSHong Zhang       tmp[*vi-- ] -= (*v--)*tmp[i];
8606abc6512SBarry Smith     }
8616abc6512SBarry Smith   }
8626abc6512SBarry Smith 
8636abc6512SBarry Smith   /* copy tmp into x according to permutation */
8646abc6512SBarry Smith   for (i=0; i<n; i++) x[r[i]] += tmp[i];
8656abc6512SBarry Smith 
8662235e667SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
8672235e667SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
8681ebc52fbSHong Zhang   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
8691ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
8706abc6512SBarry Smith 
871b0a32e0cSBarry Smith   PetscLogFlops(2*a->nz);
8723a40ed3dSBarry Smith   PetscFunctionReturn(0);
873da3a660dSBarry Smith }
8749e25ed09SBarry Smith /* ----------------------------------------------------------------*/
875dfbe8321SBarry Smith EXTERN PetscErrorCode MatMissingDiagonal_SeqAIJ(Mat);
87608480c60SBarry Smith 
8774a2ae208SSatish Balay #undef __FUNCT__
8784a2ae208SSatish Balay #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ"
879dfbe8321SBarry Smith PetscErrorCode MatILUFactorSymbolic_SeqAIJ(Mat A,IS isrow,IS iscol,MatFactorInfo *info,Mat *fact)
8809e25ed09SBarry Smith {
881416022c9SBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data,*b;
8829e25ed09SBarry Smith   IS             isicol;
8836849ba73SBarry Smith   PetscErrorCode ierr;
88438baddfdSBarry Smith   PetscInt       *r,*ic,prow,n = A->m,*ai = a->i,*aj = a->j;
88538baddfdSBarry Smith   PetscInt       *ainew,*ajnew,jmax,*fill,*xi,nz,*im,*ajfill,*flev;
886418422e8SSatish Balay   PetscInt       *dloc,idx,row,m,fm,nzf,nzi,len, reallocs = 0,dcount = 0;
88738baddfdSBarry Smith   PetscInt       incrlev,nnz,i,levels,diagonal_fill;
88877c4ece6SBarry Smith   PetscTruth     col_identity,row_identity;
889329f5518SBarry Smith   PetscReal      f;
8909e25ed09SBarry Smith 
8913a40ed3dSBarry Smith   PetscFunctionBegin;
8926cf997b0SBarry Smith   f             = info->fill;
89338baddfdSBarry Smith   levels        = (PetscInt)info->levels;
89438baddfdSBarry Smith   diagonal_fill = (PetscInt)info->diagonal_fill;
8954c49b128SBarry Smith   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
89682bf6240SBarry Smith 
89708480c60SBarry Smith   /* special case that simply copies fill pattern */
898be0abb6dSBarry Smith   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
899be0abb6dSBarry Smith   ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr);
90086bacbd2SBarry Smith   if (!levels && row_identity && col_identity) {
9012e8a6d31SBarry Smith     ierr = MatDuplicate_SeqAIJ(A,MAT_DO_NOT_COPY_VALUES,fact);CHKERRQ(ierr);
90208480c60SBarry Smith     (*fact)->factor = FACTOR_LU;
90308480c60SBarry Smith     b               = (Mat_SeqAIJ*)(*fact)->data;
90408480c60SBarry Smith     if (!b->diag) {
9057c922b88SBarry Smith       ierr = MatMarkDiagonal_SeqAIJ(*fact);CHKERRQ(ierr);
90608480c60SBarry Smith     }
9077c922b88SBarry Smith     ierr = MatMissingDiagonal_SeqAIJ(*fact);CHKERRQ(ierr);
90808480c60SBarry Smith     b->row              = isrow;
90908480c60SBarry Smith     b->col              = iscol;
91082bf6240SBarry Smith     b->icol             = isicol;
911a2e34c3dSBarry Smith     b->lu_damping       = info->damping;
91287828ca2SBarry Smith     b->lu_zeropivot     = info->zeropivot;
9132cea7109SBarry Smith     b->lu_shift         = info->shift;
9142cea7109SBarry Smith     b->lu_shift_fraction= info->shift_fraction;
91587828ca2SBarry Smith     ierr                = PetscMalloc(((*fact)->m+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
916f830108cSBarry Smith     (*fact)->ops->solve = MatSolve_SeqAIJ_NaturalOrdering;
917c38d4ed2SBarry Smith     ierr                = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
918c38d4ed2SBarry Smith     ierr                = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
9193a40ed3dSBarry Smith     PetscFunctionReturn(0);
92008480c60SBarry Smith   }
92108480c60SBarry Smith 
922898183ecSLois Curfman McInnes   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
923898183ecSLois Curfman McInnes   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
9249e25ed09SBarry Smith 
9259e25ed09SBarry Smith   /* get new row pointers */
92638baddfdSBarry Smith   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&ainew);CHKERRQ(ierr);
927010a20caSHong Zhang   ainew[0] = 0;
9289e25ed09SBarry Smith   /* don't know how many column pointers are needed so estimate */
92938baddfdSBarry Smith   jmax = (PetscInt)(f*(ai[n]+1));
93038baddfdSBarry Smith   ierr = PetscMalloc((jmax)*sizeof(PetscInt),&ajnew);CHKERRQ(ierr);
9319e25ed09SBarry Smith   /* ajfill is level of fill for each fill entry */
93238baddfdSBarry Smith   ierr = PetscMalloc((jmax)*sizeof(PetscInt),&ajfill);CHKERRQ(ierr);
9339e25ed09SBarry Smith   /* fill is a linked list of nonzeros in active row */
93438baddfdSBarry Smith   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&fill);CHKERRQ(ierr);
93556beaf04SBarry Smith   /* im is level for each filled value */
93638baddfdSBarry Smith   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&im);CHKERRQ(ierr);
93756beaf04SBarry Smith   /* dloc is location of diagonal in factor */
93838baddfdSBarry Smith   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&dloc);CHKERRQ(ierr);
93956beaf04SBarry Smith   dloc[0]  = 0;
94056beaf04SBarry Smith   for (prow=0; prow<n; prow++) {
9416cf997b0SBarry Smith 
9426cf997b0SBarry Smith     /* copy current row into linked list */
94356beaf04SBarry Smith     nzf     = nz  = ai[r[prow]+1] - ai[r[prow]];
94429bbc08cSBarry Smith     if (!nz) SETERRQ(PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix");
945010a20caSHong Zhang     xi      = aj + ai[r[prow]] ;
9469e25ed09SBarry Smith     fill[n]    = n;
947435faa5fSBarry Smith     fill[prow] = -1; /* marker to indicate if diagonal exists */
9489e25ed09SBarry Smith     while (nz--) {
9499e25ed09SBarry Smith       fm  = n;
950010a20caSHong Zhang       idx = ic[*xi++ ];
9519e25ed09SBarry Smith       do {
9529e25ed09SBarry Smith         m  = fm;
9539e25ed09SBarry Smith         fm = fill[m];
9549e25ed09SBarry Smith       } while (fm < idx);
9559e25ed09SBarry Smith       fill[m]   = idx;
9569e25ed09SBarry Smith       fill[idx] = fm;
95756beaf04SBarry Smith       im[idx]   = 0;
9589e25ed09SBarry Smith     }
9596cf997b0SBarry Smith 
9606cf997b0SBarry Smith     /* make sure diagonal entry is included */
961435faa5fSBarry Smith     if (diagonal_fill && fill[prow] == -1) {
9626cf997b0SBarry Smith       fm = n;
963435faa5fSBarry Smith       while (fill[fm] < prow) fm = fill[fm];
964435faa5fSBarry Smith       fill[prow] = fill[fm]; /* insert diagonal into linked list */
965435faa5fSBarry Smith       fill[fm]   = prow;
9666cf997b0SBarry Smith       im[prow]   = 0;
9676cf997b0SBarry Smith       nzf++;
968335d9088SBarry Smith       dcount++;
9696cf997b0SBarry Smith     }
9706cf997b0SBarry Smith 
97156beaf04SBarry Smith     nzi = 0;
9729e25ed09SBarry Smith     row = fill[n];
97356beaf04SBarry Smith     while (row < prow) {
97456beaf04SBarry Smith       incrlev = im[row] + 1;
97556beaf04SBarry Smith       nz      = dloc[row];
976010a20caSHong Zhang       xi      = ajnew  + ainew[row]  + nz + 1;
977010a20caSHong Zhang       flev    = ajfill + ainew[row]  + nz + 1;
978416022c9SBarry Smith       nnz     = ainew[row+1] - ainew[row] - nz - 1;
97956beaf04SBarry Smith       fm      = row;
98056beaf04SBarry Smith       while (nnz-- > 0) {
981010a20caSHong Zhang         idx = *xi++ ;
98256beaf04SBarry Smith         if (*flev + incrlev > levels) {
98356beaf04SBarry Smith           flev++;
98456beaf04SBarry Smith           continue;
98556beaf04SBarry Smith         }
98656beaf04SBarry Smith         do {
9879e25ed09SBarry Smith           m  = fm;
9889e25ed09SBarry Smith           fm = fill[m];
98956beaf04SBarry Smith         } while (fm < idx);
9909e25ed09SBarry Smith         if (fm != idx) {
99156beaf04SBarry Smith           im[idx]   = *flev + incrlev;
9929e25ed09SBarry Smith           fill[m]   = idx;
9939e25ed09SBarry Smith           fill[idx] = fm;
9949e25ed09SBarry Smith           fm        = idx;
99556beaf04SBarry Smith           nzf++;
996ecf371e4SBarry Smith         } else {
99756beaf04SBarry Smith           if (im[idx] > *flev + incrlev) im[idx] = *flev+incrlev;
9989e25ed09SBarry Smith         }
99956beaf04SBarry Smith         flev++;
10009e25ed09SBarry Smith       }
10019e25ed09SBarry Smith       row = fill[row];
100256beaf04SBarry Smith       nzi++;
10039e25ed09SBarry Smith     }
10049e25ed09SBarry Smith     /* copy new filled row into permanent storage */
100556beaf04SBarry Smith     ainew[prow+1] = ainew[prow] + nzf;
1006010a20caSHong Zhang     if (ainew[prow+1] > jmax) {
1007ecf371e4SBarry Smith 
1008ecf371e4SBarry Smith       /* estimate how much additional space we will need */
1009ecf371e4SBarry Smith       /* use the strategy suggested by David Hysom <hysom@perch-t.icase.edu> */
1010ecf371e4SBarry Smith       /* just double the memory each time */
101138baddfdSBarry Smith       /*  maxadd = (PetscInt)((f*(ai[n]+!shift)*(n-prow+5))/n); */
101238baddfdSBarry Smith       PetscInt maxadd = jmax;
101356beaf04SBarry Smith       if (maxadd < nzf) maxadd = (n-prow)*(nzf+1);
1014bbb6d6a8SBarry Smith       jmax += maxadd;
1015ecf371e4SBarry Smith 
1016ecf371e4SBarry Smith       /* allocate a longer ajnew and ajfill */
101738baddfdSBarry Smith       ierr   = PetscMalloc(jmax*sizeof(PetscInt),&xi);CHKERRQ(ierr);
101838baddfdSBarry Smith       ierr   = PetscMemcpy(xi,ajnew,(ainew[prow])*sizeof(PetscInt));CHKERRQ(ierr);
1019606d414cSSatish Balay       ierr   = PetscFree(ajnew);CHKERRQ(ierr);
102056beaf04SBarry Smith       ajnew  = xi;
102138baddfdSBarry Smith       ierr   = PetscMalloc(jmax*sizeof(PetscInt),&xi);CHKERRQ(ierr);
102238baddfdSBarry Smith       ierr   = PetscMemcpy(xi,ajfill,(ainew[prow])*sizeof(PetscInt));CHKERRQ(ierr);
1023606d414cSSatish Balay       ierr   = PetscFree(ajfill);CHKERRQ(ierr);
102456beaf04SBarry Smith       ajfill = xi;
1025418422e8SSatish Balay       reallocs++; /* count how many times we realloc */
10269e25ed09SBarry Smith     }
1027010a20caSHong Zhang     xi          = ajnew + ainew[prow] ;
1028010a20caSHong Zhang     flev        = ajfill + ainew[prow] ;
102956beaf04SBarry Smith     dloc[prow]  = nzi;
10309e25ed09SBarry Smith     fm          = fill[n];
103156beaf04SBarry Smith     while (nzf--) {
1032010a20caSHong Zhang       *xi++   = fm ;
103356beaf04SBarry Smith       *flev++ = im[fm];
10349e25ed09SBarry Smith       fm      = fill[fm];
10359e25ed09SBarry Smith     }
10366cf997b0SBarry Smith     /* make sure row has diagonal entry */
1037010a20caSHong Zhang     if (ajnew[ainew[prow]+dloc[prow]] != prow) {
103877431f27SBarry Smith       SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,"Row %D has missing diagonal in factored matrix\n\
10396cf997b0SBarry Smith     try running with -pc_ilu_nonzeros_along_diagonal or -pc_ilu_diagonal_fill",prow);
10406cf997b0SBarry Smith     }
10419e25ed09SBarry Smith   }
1042606d414cSSatish Balay   ierr = PetscFree(ajfill);CHKERRQ(ierr);
1043898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
1044898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
1045606d414cSSatish Balay   ierr = PetscFree(fill);CHKERRQ(ierr);
1046606d414cSSatish Balay   ierr = PetscFree(im);CHKERRQ(ierr);
10479e25ed09SBarry Smith 
1048f64065bbSBarry Smith   {
1049329f5518SBarry Smith     PetscReal af = ((PetscReal)ainew[n])/((PetscReal)ai[n]);
1050418422e8SSatish Balay     PetscLogInfo(A,"MatILUFactorSymbolic_SeqAIJ:Reallocs %D Fill ratio:given %g needed %g\n",reallocs,f,af);
1051c0d6ac57SBarry Smith     PetscLogInfo(A,"MatILUFactorSymbolic_SeqAIJ:Run with -[sub_]pc_ilu_fill %g or use \n",af);
1052c0d6ac57SBarry Smith     PetscLogInfo(A,"MatILUFactorSymbolic_SeqAIJ:PCILUSetFill([sub]pc,%g);\n",af);
1053b0a32e0cSBarry Smith     PetscLogInfo(A,"MatILUFactorSymbolic_SeqAIJ:for best performance.\n");
1054335d9088SBarry Smith     if (diagonal_fill) {
105577431f27SBarry Smith       PetscLogInfo(A,"MatILUFactorSymbolic_SeqAIJ:Detected and replaced %D missing diagonals",dcount);
1056335d9088SBarry Smith     }
1057f64065bbSBarry Smith   }
1058bbb6d6a8SBarry Smith 
10599e25ed09SBarry Smith   /* put together the new matrix */
1060f204ca49SKris Buschelman   ierr = MatCreate(A->comm,n,n,n,n,fact);CHKERRQ(ierr);
1061f204ca49SKris Buschelman   ierr = MatSetType(*fact,A->type_name);CHKERRQ(ierr);
1062f204ca49SKris Buschelman   ierr = MatSeqAIJSetPreallocation(*fact,0,PETSC_NULL);CHKERRQ(ierr);
1063b0a32e0cSBarry Smith   PetscLogObjectParent(*fact,isicol);
1064416022c9SBarry Smith   b = (Mat_SeqAIJ*)(*fact)->data;
1065606d414cSSatish Balay   ierr = PetscFree(b->imax);CHKERRQ(ierr);
10667c922b88SBarry Smith   b->singlemalloc = PETSC_FALSE;
1067010a20caSHong Zhang   len = (ainew[n] )*sizeof(PetscScalar);
10689e25ed09SBarry Smith   /* the next line frees the default space generated by the Create() */
1069606d414cSSatish Balay   ierr = PetscFree(b->a);CHKERRQ(ierr);
1070606d414cSSatish Balay   ierr = PetscFree(b->ilen);CHKERRQ(ierr);
1071b0a32e0cSBarry Smith   ierr = PetscMalloc(len+1,&b->a);CHKERRQ(ierr);
1072416022c9SBarry Smith   b->j          = ajnew;
1073416022c9SBarry Smith   b->i          = ainew;
107456beaf04SBarry Smith   for (i=0; i<n; i++) dloc[i] += ainew[i];
1075416022c9SBarry Smith   b->diag       = dloc;
1076416022c9SBarry Smith   b->ilen       = 0;
1077416022c9SBarry Smith   b->imax       = 0;
1078416022c9SBarry Smith   b->row        = isrow;
1079416022c9SBarry Smith   b->col        = iscol;
1080c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
1081c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
108282bf6240SBarry Smith   b->icol       = isicol;
108387828ca2SBarry Smith   ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
1084416022c9SBarry Smith   /* In b structure:  Free imax, ilen, old a, old j.
108556beaf04SBarry Smith      Allocate dloc, solve_work, new a, new j */
108638baddfdSBarry Smith   PetscLogObjectMemory(*fact,(ainew[n]-n) * (sizeof(PetscInt)+sizeof(PetscScalar)));
1087010a20caSHong Zhang   b->maxnz             = b->nz = ainew[n] ;
1088a2e34c3dSBarry Smith   b->lu_damping        = info->damping;
10892cea7109SBarry Smith   b->lu_shift          = info->shift;
10902cea7109SBarry Smith   b->lu_shift_fraction = info->shift_fraction;
109187828ca2SBarry Smith   b->lu_zeropivot = info->zeropivot;
10929e25ed09SBarry Smith   (*fact)->factor = FACTOR_LU;
10933a7fca6bSBarry Smith   ierr = Mat_AIJ_CheckInode(*fact,PETSC_FALSE);CHKERRQ(ierr);
10943a7fca6bSBarry Smith   (*fact)->ops->lufactornumeric =  A->ops->lufactornumeric; /* Use Inode variant ONLY if A has inodes */
1095ae068f58SLois Curfman McInnes 
1096418422e8SSatish Balay   (*fact)->info.factor_mallocs    = reallocs;
1097ae068f58SLois Curfman McInnes   (*fact)->info.fill_ratio_given  = f;
1098329f5518SBarry Smith   (*fact)->info.fill_ratio_needed = ((PetscReal)ainew[n])/((PetscReal)ai[prow]);
10993a40ed3dSBarry Smith   PetscFunctionReturn(0);
11009e25ed09SBarry Smith }
1101d5d45c9bSBarry Smith 
1102a6175056SHong Zhang #undef __FUNCT__
1103a6175056SHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ"
1104dfbe8321SBarry Smith PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ(Mat A,Mat *fact)
1105a6175056SHong Zhang {
11060968510aSHong Zhang   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
110779416396SBarry Smith   PetscErrorCode ierr,(*f)(Mat,Mat*);
1108d5d45c9bSBarry Smith 
1109a6175056SHong Zhang   PetscFunctionBegin;
11100968510aSHong Zhang   if (!a->sbaijMat){
11110968510aSHong Zhang     ierr = MatConvert(A,MATSEQSBAIJ,&a->sbaijMat);CHKERRQ(ierr);
1112a6175056SHong Zhang   }
1113d07a9264SSatish Balay   ierr = PetscObjectQueryFunction((PetscObject) *fact,"MatCholeskyFactorNumeric",(void (**)(void))&f);CHKERRQ(ierr);
111479416396SBarry Smith   ierr = (*f)(a->sbaijMat,fact);CHKERRQ(ierr);
11150968510aSHong Zhang   ierr = MatDestroy(a->sbaijMat);CHKERRQ(ierr);
1116064503c5SHong Zhang   a->sbaijMat = PETSC_NULL;
1117a6175056SHong Zhang   PetscFunctionReturn(0);
1118a6175056SHong Zhang }
1119a6175056SHong Zhang 
1120a6175056SHong Zhang #undef __FUNCT__
1121a6175056SHong Zhang #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ"
1122dfbe8321SBarry Smith PetscErrorCode MatICCFactorSymbolic_SeqAIJ(Mat A,IS perm,MatFactorInfo *info,Mat *fact)
1123a6175056SHong Zhang {
11240968510aSHong Zhang   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
1125dfbe8321SBarry Smith   PetscErrorCode ierr;
1126653a6975SHong Zhang   PetscTruth     perm_identity;
1127a6175056SHong Zhang 
1128a6175056SHong Zhang   PetscFunctionBegin;
1129653a6975SHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
1130653a6975SHong Zhang   if (!perm_identity){
1131e005ede5SBarry Smith     SETERRQ(PETSC_ERR_SUP,"Non-identity permutation is not supported yet");
1132653a6975SHong Zhang   }
11330968510aSHong Zhang   if (!a->sbaijMat){
11340968510aSHong Zhang     ierr = MatConvert(A,MATSEQSBAIJ,&a->sbaijMat);CHKERRQ(ierr);
11350968510aSHong Zhang   }
1136a6175056SHong Zhang 
1137b00f7748SHong Zhang   ierr = MatICCFactorSymbolic(a->sbaijMat,perm,info,fact);CHKERRQ(ierr);
1138d07a9264SSatish Balay   ierr = PetscObjectComposeFunction((PetscObject) *fact,"MatCholeskyFactorNumeric","dummyname",(FCNVOID)(*fact)->ops->choleskyfactornumeric);CHKERRQ(ierr);
1139653a6975SHong Zhang   (*fact)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ;
1140653a6975SHong Zhang 
1141a6175056SHong Zhang   PetscFunctionReturn(0);
1142a6175056SHong Zhang }
1143d5d45c9bSBarry Smith 
1144f76d2b81SHong Zhang #undef __FUNCT__
1145f76d2b81SHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_SeqAIJ"
1146dfbe8321SBarry Smith PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ(Mat A,IS perm,MatFactorInfo *info,Mat *fact)
1147f76d2b81SHong Zhang {
1148f76d2b81SHong Zhang   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data;
1149dfbe8321SBarry Smith   PetscErrorCode ierr;
1150f76d2b81SHong Zhang   PetscTruth     perm_identity;
1151f76d2b81SHong Zhang 
1152f76d2b81SHong Zhang   PetscFunctionBegin;
1153f76d2b81SHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
1154f76d2b81SHong Zhang   if (!perm_identity){
1155e005ede5SBarry Smith     SETERRQ(PETSC_ERR_SUP,"Non-identity permutation is not supported yet");
1156f76d2b81SHong Zhang   }
1157f76d2b81SHong Zhang   if (!a->sbaijMat){
1158f76d2b81SHong Zhang     ierr = MatConvert(A,MATSEQSBAIJ,&a->sbaijMat);CHKERRQ(ierr);
1159f76d2b81SHong Zhang   }
1160f76d2b81SHong Zhang   ierr = MatCholeskyFactorSymbolic(a->sbaijMat,perm,info,fact);CHKERRQ(ierr);
1161d07a9264SSatish Balay   ierr = PetscObjectComposeFunction((PetscObject) *fact,"MatCholeskyFactorNumeric","dummyname",(FCNVOID)(*fact)->ops->choleskyfactornumeric);CHKERRQ(ierr);
1162f76d2b81SHong Zhang   (*fact)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ;
1163f76d2b81SHong Zhang   PetscFunctionReturn(0);
1164f76d2b81SHong Zhang }
1165