xref: /petsc/src/mat/impls/aij/seq/aijfact.c (revision db4efbfdc2bb749ba21a5c202e01785163a71462)
1be1d678aSKris Buschelman #define PETSCMAT_DLL
2289bc588SBarry Smith 
370f55243SBarry Smith #include "src/mat/impls/aij/seq/aij.h"
41c4feecaSSatish Balay #include "src/inline/dot.h"
5ed480e8bSBarry Smith #include "src/inline/spops.h"
61393bc97SHong Zhang #include "petscbt.h"
71393bc97SHong Zhang #include "src/mat/utils/freespace.h"
83b2fbd54SBarry Smith 
94a2ae208SSatish Balay #undef __FUNCT__
104a2ae208SSatish Balay #define __FUNCT__ "MatOrdering_Flow_SeqAIJ"
11dfbe8321SBarry Smith PetscErrorCode MatOrdering_Flow_SeqAIJ(Mat mat,const MatOrderingType type,IS *irow,IS *icol)
123b2fbd54SBarry Smith {
133a40ed3dSBarry Smith   PetscFunctionBegin;
143a40ed3dSBarry Smith 
1529bbc08cSBarry Smith   SETERRQ(PETSC_ERR_SUP,"Code not written");
16aa482453SBarry Smith #if !defined(PETSC_USE_DEBUG)
17d64ed03dSBarry Smith   PetscFunctionReturn(0);
18d64ed03dSBarry Smith #endif
193b2fbd54SBarry Smith }
203b2fbd54SBarry Smith 
2186bacbd2SBarry Smith 
22e34fafa9SBarry Smith #if !defined(PETSC_AVOID_GNUCOPYRIGHT_CODE)
23a77337e4SBarry Smith EXTERN PetscErrorCode SPARSEKIT2dperm(PetscInt*,MatScalar*,PetscInt*,PetscInt*,MatScalar*,PetscInt*,PetscInt*,PetscInt*,PetscInt*,PetscInt*);
24a77337e4SBarry Smith EXTERN PetscErrorCode SPARSEKIT2ilutp(PetscInt*,MatScalar*,PetscInt*,PetscInt*,PetscInt*,PetscReal,PetscReal*,PetscInt*,MatScalar*,PetscInt*,PetscInt*,PetscInt*,MatScalar*,PetscInt*,PetscInt*,PetscErrorCode*);
25a77337e4SBarry Smith EXTERN PetscErrorCode SPARSEKIT2msrcsr(PetscInt*,MatScalar*,PetscInt*,MatScalar*,PetscInt*,PetscInt*,MatScalar*,PetscInt*);
26e34fafa9SBarry Smith #endif
2786bacbd2SBarry Smith 
284a2ae208SSatish Balay #undef __FUNCT__
294a2ae208SSatish Balay #define __FUNCT__ "MatILUDTFactor_SeqAIJ"
3086bacbd2SBarry Smith   /* ------------------------------------------------------------
3186bacbd2SBarry Smith 
3286bacbd2SBarry Smith           This interface was contribed by Tony Caola
3386bacbd2SBarry Smith 
3486bacbd2SBarry Smith      This routine is an interface to the pivoting drop-tolerance
355bd2ddc7SBarry Smith      ILU routine written by Yousef Saad (saad@cs.umn.edu) as part of
365bd2ddc7SBarry Smith      SPARSEKIT2.
375bd2ddc7SBarry Smith 
385bd2ddc7SBarry Smith      The SPARSEKIT2 routines used here are covered by the GNU
395bd2ddc7SBarry Smith      copyright; see the file gnu in this directory.
4086bacbd2SBarry Smith 
4186bacbd2SBarry Smith      Thanks to Prof. Saad, Dr. Hysom, and Dr. Smith for their
4286bacbd2SBarry Smith      help in getting this routine ironed out.
4386bacbd2SBarry Smith 
445bd2ddc7SBarry Smith      The major drawback to this routine is that if info->fill is
455bd2ddc7SBarry Smith      not large enough it fails rather than allocating more space;
465bd2ddc7SBarry Smith      this can be fixed by hacking/improving the f2c version of
475bd2ddc7SBarry Smith      Yousef Saad's code.
4886bacbd2SBarry Smith 
4986bacbd2SBarry Smith      ------------------------------------------------------------
5086bacbd2SBarry Smith */
517529efd4SKris Buschelman PetscErrorCode MatILUDTFactor_SeqAIJ(Mat A,IS isrow,IS iscol,MatFactorInfo *info,Mat *fact)
5286bacbd2SBarry Smith {
5360ec86bdSBarry Smith #if defined(PETSC_AVOID_GNUCOPYRIGHT_CODE)
5460ec86bdSBarry Smith   PetscFunctionBegin;
55e005ede5SBarry Smith   SETERRQ(PETSC_ERR_SUP_SYS,"This distribution does not include GNU Copyright code\n\
5660ec86bdSBarry Smith   You can obtain the drop tolerance routines by installing PETSc from\n\
5760ec86bdSBarry Smith   www.mcs.anl.gov/petsc\n");
5860ec86bdSBarry Smith #else
5986bacbd2SBarry Smith   Mat_SeqAIJ     *a = (Mat_SeqAIJ*)A->data,*b;
6007d2056aSBarry Smith   IS             iscolf,isicol,isirow;
6186bacbd2SBarry Smith   PetscTruth     reorder;
629cc1f4e3SBarry Smith   PetscErrorCode ierr,sierr;
63d0f46423SBarry Smith   PetscInt       *c,*r,*ic,i,n = A->rmap->n;
6438baddfdSBarry Smith   PetscInt       *old_i = a->i,*old_j = a->j,*new_i,*old_i2 = 0,*old_j2 = 0,*new_j;
6538baddfdSBarry Smith   PetscInt       *ordcol,*iwk,*iperm,*jw;
6638baddfdSBarry Smith   PetscInt       jmax,lfill,job,*o_i,*o_j;
6754f21887SBarry Smith   MatScalar      *old_a = a->a,*w,*new_a,*old_a2 = 0,*wk,*o_a;
6854a8161fSBarry Smith   PetscReal      af;
6986bacbd2SBarry Smith 
7086bacbd2SBarry Smith   PetscFunctionBegin;
7186bacbd2SBarry Smith 
7286bacbd2SBarry Smith   if (info->dt == PETSC_DEFAULT)      info->dt      = .005;
7338baddfdSBarry Smith   if (info->dtcount == PETSC_DEFAULT) info->dtcount = (PetscInt)(1.5*a->rmax);
7486bacbd2SBarry Smith   if (info->dtcol == PETSC_DEFAULT)   info->dtcol   = .01;
7533259db3SBarry Smith   if (info->fill == PETSC_DEFAULT)    info->fill    = ((double)(n*(info->dtcount+1)))/a->nz;
7638baddfdSBarry Smith   lfill   = (PetscInt)(info->dtcount/2.0);
7738baddfdSBarry Smith   jmax    = (PetscInt)(info->fill*a->nz);
7886bacbd2SBarry Smith 
7986bacbd2SBarry Smith 
8086bacbd2SBarry Smith   /* ------------------------------------------------------------
8186bacbd2SBarry Smith      If reorder=.TRUE., then the original matrix has to be
8286bacbd2SBarry Smith      reordered to reflect the user selected ordering scheme, and
8386bacbd2SBarry Smith      then de-reordered so it is in it's original format.
8486bacbd2SBarry Smith      Because Saad's dperm() is NOT in place, we have to copy
8586bacbd2SBarry Smith      the original matrix and allocate more storage. . .
8686bacbd2SBarry Smith      ------------------------------------------------------------
8786bacbd2SBarry Smith   */
8886bacbd2SBarry Smith 
8986bacbd2SBarry Smith   /* set reorder to true if either isrow or iscol is not identity */
9086bacbd2SBarry Smith   ierr = ISIdentity(isrow,&reorder);CHKERRQ(ierr);
9186bacbd2SBarry Smith   if (reorder) {ierr = ISIdentity(iscol,&reorder);CHKERRQ(ierr);}
9286bacbd2SBarry Smith   reorder = PetscNot(reorder);
9386bacbd2SBarry Smith 
9486bacbd2SBarry Smith 
9586bacbd2SBarry Smith   /* storage for ilu factor */
9638baddfdSBarry Smith   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&new_i);CHKERRQ(ierr);
9738baddfdSBarry Smith   ierr = PetscMalloc(jmax*sizeof(PetscInt),&new_j);CHKERRQ(ierr);
98a77337e4SBarry Smith   ierr = PetscMalloc(jmax*sizeof(MatScalar),&new_a);CHKERRQ(ierr);
9938baddfdSBarry Smith   ierr = PetscMalloc(n*sizeof(PetscInt),&ordcol);CHKERRQ(ierr);
10086bacbd2SBarry Smith 
10186bacbd2SBarry Smith   /* ------------------------------------------------------------
10286bacbd2SBarry Smith      Make sure that everything is Fortran formatted (1-Based)
10386bacbd2SBarry Smith      ------------------------------------------------------------
10486bacbd2SBarry Smith   */
105b19713cbSBarry Smith   for (i=old_i[0];i<old_i[n];i++) {
106b19713cbSBarry Smith     old_j[i]++;
10786bacbd2SBarry Smith   }
108b19713cbSBarry Smith   for(i=0;i<n+1;i++) {
109b19713cbSBarry Smith     old_i[i]++;
110b19713cbSBarry Smith   };
111010a20caSHong Zhang 
11286bacbd2SBarry Smith 
11386bacbd2SBarry Smith   if (reorder) {
114c0c2c81eSBarry Smith     ierr = ISGetIndices(iscol,&c);CHKERRQ(ierr);
115c0c2c81eSBarry Smith     ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
116c0c2c81eSBarry Smith     for(i=0;i<n;i++) {
117c0c2c81eSBarry Smith       r[i]  = r[i]+1;
118c0c2c81eSBarry Smith       c[i]  = c[i]+1;
119c0c2c81eSBarry Smith     }
12038baddfdSBarry Smith     ierr = PetscMalloc((n+1)*sizeof(PetscInt),&old_i2);CHKERRQ(ierr);
12138baddfdSBarry Smith     ierr = PetscMalloc((old_i[n]-old_i[0]+1)*sizeof(PetscInt),&old_j2);CHKERRQ(ierr);
122a77337e4SBarry Smith     ierr = PetscMalloc((old_i[n]-old_i[0]+1)*sizeof(MatScalar),&old_a2);CHKERRQ(ierr);
1235bd2ddc7SBarry Smith     job  = 3; SPARSEKIT2dperm(&n,old_a,old_j,old_i,old_a2,old_j2,old_i2,r,c,&job);
124c0c2c81eSBarry Smith     for (i=0;i<n;i++) {
125c0c2c81eSBarry Smith       r[i]  = r[i]-1;
126c0c2c81eSBarry Smith       c[i]  = c[i]-1;
127c0c2c81eSBarry Smith     }
128c0c2c81eSBarry Smith     ierr = ISRestoreIndices(iscol,&c);CHKERRQ(ierr);
129c0c2c81eSBarry Smith     ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
130b19713cbSBarry Smith     o_a = old_a2;
131b19713cbSBarry Smith     o_j = old_j2;
132b19713cbSBarry Smith     o_i = old_i2;
133b19713cbSBarry Smith   } else {
134b19713cbSBarry Smith     o_a = old_a;
135b19713cbSBarry Smith     o_j = old_j;
136b19713cbSBarry Smith     o_i = old_i;
13786bacbd2SBarry Smith   }
13886bacbd2SBarry Smith 
13986bacbd2SBarry Smith   /* ------------------------------------------------------------
14086bacbd2SBarry Smith      Call Saad's ilutp() routine to generate the factorization
14186bacbd2SBarry Smith      ------------------------------------------------------------
14286bacbd2SBarry Smith   */
14386bacbd2SBarry Smith 
14438baddfdSBarry Smith   ierr = PetscMalloc(2*n*sizeof(PetscInt),&iperm);CHKERRQ(ierr);
14538baddfdSBarry Smith   ierr = PetscMalloc(2*n*sizeof(PetscInt),&jw);CHKERRQ(ierr);
14687828ca2SBarry Smith   ierr = PetscMalloc(n*sizeof(PetscScalar),&w);CHKERRQ(ierr);
14786bacbd2SBarry Smith 
14854a8161fSBarry Smith   SPARSEKIT2ilutp(&n,o_a,o_j,o_i,&lfill,(PetscReal)info->dt,&info->dtcol,&n,new_a,new_j,new_i,&jmax,w,jw,iperm,&sierr);
1496849ba73SBarry Smith   if (sierr) {
1506849ba73SBarry Smith     switch (sierr) {
151a83599f4SBarry Smith       case -3: SETERRQ2(PETSC_ERR_LIB,"ilutp(), matrix U overflows, need larger info->fill current fill %G space allocated %D",info->fill,jmax);
152a83599f4SBarry Smith       case -2: SETERRQ2(PETSC_ERR_LIB,"ilutp(), matrix L overflows, need larger info->fill current fill %G space allocated %D",info->fill,jmax);
153e005ede5SBarry Smith       case -5: SETERRQ(PETSC_ERR_LIB,"ilutp(), zero row encountered");
154e005ede5SBarry Smith       case -1: SETERRQ(PETSC_ERR_LIB,"ilutp(), input matrix may be wrong");
15577431f27SBarry Smith       case -4: SETERRQ1(PETSC_ERR_LIB,"ilutp(), illegal info->fill value %D",jmax);
15677431f27SBarry Smith       default: SETERRQ1(PETSC_ERR_LIB,"ilutp(), zero pivot detected on row %D",sierr);
15786bacbd2SBarry Smith     }
15886bacbd2SBarry Smith   }
15986bacbd2SBarry Smith 
16086bacbd2SBarry Smith   ierr = PetscFree(w);CHKERRQ(ierr);
16186bacbd2SBarry Smith   ierr = PetscFree(jw);CHKERRQ(ierr);
16286bacbd2SBarry Smith 
16386bacbd2SBarry Smith   /* ------------------------------------------------------------
16486bacbd2SBarry Smith      Saad's routine gives the result in Modified Sparse Row (msr)
16586bacbd2SBarry Smith      Convert to Compressed Sparse Row format (csr)
16686bacbd2SBarry Smith      ------------------------------------------------------------
16786bacbd2SBarry Smith   */
16886bacbd2SBarry Smith 
16987828ca2SBarry Smith   ierr = PetscMalloc(n*sizeof(PetscScalar),&wk);CHKERRQ(ierr);
17038baddfdSBarry Smith   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&iwk);CHKERRQ(ierr);
17186bacbd2SBarry Smith 
1725bd2ddc7SBarry Smith   SPARSEKIT2msrcsr(&n,new_a,new_j,new_a,new_j,new_i,wk,iwk);
17386bacbd2SBarry Smith 
17486bacbd2SBarry Smith   ierr = PetscFree(iwk);CHKERRQ(ierr);
17586bacbd2SBarry Smith   ierr = PetscFree(wk);CHKERRQ(ierr);
17686bacbd2SBarry Smith 
17786bacbd2SBarry Smith   if (reorder) {
17886bacbd2SBarry Smith     ierr = PetscFree(old_a2);CHKERRQ(ierr);
17986bacbd2SBarry Smith     ierr = PetscFree(old_j2);CHKERRQ(ierr);
18086bacbd2SBarry Smith     ierr = PetscFree(old_i2);CHKERRQ(ierr);
181313ae024SBarry Smith   } else {
182b19713cbSBarry Smith     /* fix permutation of old_j that the factorization introduced */
183f170005cSBarry Smith     for (i=old_i[0]; i<old_i[n]; i++) {
184b19713cbSBarry Smith       old_j[i-1] = iperm[old_j[i-1]-1];
185b19713cbSBarry Smith     }
186b19713cbSBarry Smith   }
187b19713cbSBarry Smith 
188b801d0f9SBarry Smith   /* get rid of the shift to indices starting at 1 */
18986bacbd2SBarry Smith   for (i=0; i<n+1; i++) {
190b19713cbSBarry Smith     old_i[i]--;
191b19713cbSBarry Smith   }
192b19713cbSBarry Smith   for (i=old_i[0];i<old_i[n];i++) {
193b19713cbSBarry Smith     old_j[i]--;
194b19713cbSBarry Smith   }
19586bacbd2SBarry Smith 
196b19713cbSBarry Smith   /* Make the factored matrix 0-based */
19786bacbd2SBarry Smith   for (i=0; i<n+1; i++) {
198b19713cbSBarry Smith     new_i[i]--;
199b19713cbSBarry Smith   }
200b19713cbSBarry Smith   for (i=new_i[0];i<new_i[n];i++) {
201b19713cbSBarry Smith     new_j[i]--;
202b19713cbSBarry Smith   }
20386bacbd2SBarry Smith 
20486bacbd2SBarry Smith   /*-- due to the pivoting, we need to reorder iscol to correctly --*/
20586bacbd2SBarry Smith   /*-- permute the right-hand-side and solution vectors           --*/
2064c49b128SBarry Smith   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
2074c49b128SBarry Smith   ierr = ISInvertPermutation(isrow,PETSC_DECIDE,&isirow);CHKERRQ(ierr);
208c0c2c81eSBarry Smith   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
20986bacbd2SBarry Smith   for(i=0; i<n; i++) {
21086bacbd2SBarry Smith     ordcol[i] = ic[iperm[i]-1];
21186bacbd2SBarry Smith   };
21286bacbd2SBarry Smith   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
21307d2056aSBarry Smith   ierr = ISDestroy(isicol);CHKERRQ(ierr);
21486bacbd2SBarry Smith 
215c0c2c81eSBarry Smith   ierr = PetscFree(iperm);CHKERRQ(ierr);
216c0c2c81eSBarry Smith 
217329f5518SBarry Smith   ierr = ISCreateGeneral(PETSC_COMM_SELF,n,ordcol,&iscolf);CHKERRQ(ierr);
21807d2056aSBarry Smith   ierr = PetscFree(ordcol);CHKERRQ(ierr);
21986bacbd2SBarry Smith 
22086bacbd2SBarry Smith   /*----- put together the new matrix -----*/
22186bacbd2SBarry Smith 
2227adad957SLisandro Dalcin   ierr = MatCreate(((PetscObject)A)->comm,fact);CHKERRQ(ierr);
223f69a0ea3SMatthew Knepley   ierr = MatSetSizes(*fact,n,n,n,n);CHKERRQ(ierr);
2247adad957SLisandro Dalcin   ierr = MatSetType(*fact,((PetscObject)A)->type_name);CHKERRQ(ierr);
225ab93d7beSBarry Smith   ierr = MatSeqAIJSetPreallocation_SeqAIJ(*fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
2265c9eb25fSBarry Smith   (*fact)->factor    = MAT_FACTOR_LU;
22786bacbd2SBarry Smith   (*fact)->assembled = PETSC_TRUE;
22886bacbd2SBarry Smith 
22986bacbd2SBarry Smith   b = (Mat_SeqAIJ*)(*fact)->data;
230e6b907acSBarry Smith   b->free_a        = PETSC_TRUE;
231e6b907acSBarry Smith   b->free_ij       = PETSC_TRUE;
23207d2056aSBarry Smith   b->singlemalloc  = PETSC_FALSE;
23386bacbd2SBarry Smith   b->a             = new_a;
23486bacbd2SBarry Smith   b->j             = new_j;
23586bacbd2SBarry Smith   b->i             = new_i;
23686bacbd2SBarry Smith   b->ilen          = 0;
23786bacbd2SBarry Smith   b->imax          = 0;
2381f9e874aSBarry Smith   /*  I am not sure why these are the inverses of the row and column permutations; but the other way is NO GOOD */
239313ae024SBarry Smith   b->row           = isirow;
24086bacbd2SBarry Smith   b->col           = iscolf;
24187828ca2SBarry Smith   ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
24286bacbd2SBarry Smith   b->maxnz = b->nz = new_i[n];
24386bacbd2SBarry Smith   ierr = MatMarkDiagonal_SeqAIJ(*fact);CHKERRQ(ierr);
24486bacbd2SBarry Smith   (*fact)->info.factor_mallocs = 0;
24586bacbd2SBarry Smith 
246431e710aSBarry Smith   af = ((double)b->nz)/((double)a->nz) + .001;
247ae15b995SBarry Smith   ierr = PetscInfo2(A,"Fill ratio:given %G needed %G\n",info->fill,af);CHKERRQ(ierr);
248ae15b995SBarry Smith   ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
249ae15b995SBarry Smith   ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr);
250ae15b995SBarry Smith   ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
251431e710aSBarry Smith 
2527529efd4SKris Buschelman   ierr = MatILUDTFactor_Inode(A,isrow,iscol,info,fact);CHKERRQ(ierr);
25371c2f376SKris Buschelman 
25486bacbd2SBarry Smith   PetscFunctionReturn(0);
25560ec86bdSBarry Smith #endif
25686bacbd2SBarry Smith }
25786bacbd2SBarry Smith 
258e631078cSBarry Smith EXTERN_C_BEGIN
2594a2ae208SSatish Balay #undef __FUNCT__
260*db4efbfdSBarry Smith #define __FUNCT__ "MatGetFactorAvailable_seqaij_petsc"
261*db4efbfdSBarry Smith PetscErrorCode MatGetFactorAvailable_seqaij_petsc(Mat A,MatFactorType ftype,PetscTruth *flg)
262*db4efbfdSBarry Smith {
263*db4efbfdSBarry Smith   PetscFunctionBegin;
264*db4efbfdSBarry Smith   *flg = PETSC_TRUE;
265*db4efbfdSBarry Smith   PetscFunctionReturn(0);
266*db4efbfdSBarry Smith }
267*db4efbfdSBarry Smith EXTERN_C_END
268*db4efbfdSBarry Smith 
269*db4efbfdSBarry Smith EXTERN_C_BEGIN
270*db4efbfdSBarry Smith #undef __FUNCT__
271b24902e0SBarry Smith #define __FUNCT__ "MatGetFactor_seqaij_petsc"
272b24902e0SBarry Smith PetscErrorCode MatGetFactor_seqaij_petsc(Mat A,MatFactorType ftype,Mat *B)
273b24902e0SBarry Smith {
274d0f46423SBarry Smith   PetscInt           n = A->rmap->n;
275b24902e0SBarry Smith   PetscErrorCode     ierr;
276b24902e0SBarry Smith 
277b24902e0SBarry Smith   PetscFunctionBegin;
278b24902e0SBarry Smith   ierr = MatCreate(((PetscObject)A)->comm,B);CHKERRQ(ierr);
279b24902e0SBarry Smith   ierr = MatSetSizes(*B,n,n,n,n);CHKERRQ(ierr);
280b24902e0SBarry Smith   if (ftype == MAT_FACTOR_LU || ftype == MAT_FACTOR_ILU) {
281b24902e0SBarry Smith     ierr = MatSetType(*B,MATSEQAIJ);CHKERRQ(ierr);
282*db4efbfdSBarry Smith     (*B)->ops->lufactorsymbolic = MatLUFactorSymbolic_SeqAIJ;
283*db4efbfdSBarry Smith     (*B)->ops->ilufactorsymbolic= MatILUFactorSymbolic_SeqAIJ;
284b24902e0SBarry Smith   } else if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) {
285b24902e0SBarry Smith     ierr = MatSetType(*B,MATSEQSBAIJ);CHKERRQ(ierr);
286b24902e0SBarry Smith     ierr = MatSeqSBAIJSetPreallocation(*B,1,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
2875c9eb25fSBarry Smith     (*B)->ops->iccfactorsymbolic      = MatICCFactorSymbolic_SeqAIJ;
2885c9eb25fSBarry Smith     (*B)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqAIJ;
289b24902e0SBarry Smith   } else SETERRQ(PETSC_ERR_SUP,"Factor type not supported");
290*db4efbfdSBarry Smith   (*B)->factor = ftype;
291b24902e0SBarry Smith   PetscFunctionReturn(0);
292b24902e0SBarry Smith }
293e631078cSBarry Smith EXTERN_C_END
294b24902e0SBarry Smith 
295b24902e0SBarry Smith #undef __FUNCT__
296b9617806SBarry Smith #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ"
297dfbe8321SBarry Smith PetscErrorCode MatLUFactorSymbolic_SeqAIJ(Mat A,IS isrow,IS iscol,MatFactorInfo *info,Mat *B)
298289bc588SBarry Smith {
299416022c9SBarry Smith   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
300289bc588SBarry Smith   IS                 isicol;
3016849ba73SBarry Smith   PetscErrorCode     ierr;
302d0f46423SBarry Smith   PetscInt           *r,*ic,i,n=A->rmap->n,*ai=a->i,*aj=a->j;
3031393bc97SHong Zhang   PetscInt           *bi,*bj,*ajtmp;
3041393bc97SHong Zhang   PetscInt           *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im;
3059e046878SBarry Smith   PetscReal          f;
3064875ba38SHong Zhang   PetscInt           nlnk,*lnk,k,**bi_ptr;
307a1a86e44SBarry Smith   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
3081393bc97SHong Zhang   PetscBT            lnkbt;
309289bc588SBarry Smith 
3103a40ed3dSBarry Smith   PetscFunctionBegin;
311d0f46423SBarry Smith   if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_ERR_ARG_WRONG,"matrix must be square");
3124c49b128SBarry Smith   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
313ac121b3dSBarry Smith   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
314ac121b3dSBarry Smith   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
315289bc588SBarry Smith 
316289bc588SBarry Smith   /* get new row pointers */
3171393bc97SHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
3181393bc97SHong Zhang   bi[0] = 0;
3191393bc97SHong Zhang 
3201393bc97SHong Zhang   /* bdiag is location of diagonal in factor */
3211393bc97SHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
3221393bc97SHong Zhang   bdiag[0] = 0;
3231393bc97SHong Zhang 
3241393bc97SHong Zhang   /* linked list for storing column indices of the active row */
3251393bc97SHong Zhang   nlnk = n + 1;
3261393bc97SHong Zhang   ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
3271393bc97SHong Zhang 
32835baf27eSSatish Balay   ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr);
3291393bc97SHong Zhang 
3301393bc97SHong Zhang   /* initial FreeSpace size is f*(ai[n]+1) */
331d3d32019SBarry Smith   f = info->fill;
332a1a86e44SBarry Smith   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr);
3331393bc97SHong Zhang   current_space = free_space;
334289bc588SBarry Smith 
335289bc588SBarry Smith   for (i=0; i<n; i++) {
3361393bc97SHong Zhang     /* copy previous fill into linked list */
337289bc588SBarry Smith     nzi = 0;
3381393bc97SHong Zhang     nnz = ai[r[i]+1] - ai[r[i]];
3391393bc97SHong Zhang     if (!nnz) SETERRQ(PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix");
3401393bc97SHong Zhang     ajtmp = aj + ai[r[i]];
341afcc9904SHong Zhang     ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
3421393bc97SHong Zhang     nzi += nlnk;
3431393bc97SHong Zhang 
3441393bc97SHong Zhang     /* add pivot rows into linked list */
3451393bc97SHong Zhang     row = lnk[n];
3461393bc97SHong Zhang     while (row < i) {
347d1170560SHong Zhang       nzbd    = bdiag[row] - bi[row] + 1; /* num of entries in the row with column index <= row */
348d1170560SHong Zhang       ajtmp   = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */
349d1170560SHong Zhang       ierr = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr);
3501393bc97SHong Zhang       nzi += nlnk;
3511393bc97SHong Zhang       row  = lnk[row];
352289bc588SBarry Smith     }
3531393bc97SHong Zhang     bi[i+1] = bi[i] + nzi;
3541393bc97SHong Zhang     im[i]   = nzi;
3551393bc97SHong Zhang 
3561393bc97SHong Zhang     /* mark bdiag */
3571393bc97SHong Zhang     nzbd = 0;
3581393bc97SHong Zhang     nnz  = nzi;
3591393bc97SHong Zhang     k    = lnk[n];
3601393bc97SHong Zhang     while (nnz-- && k < i){
3611393bc97SHong Zhang       nzbd++;
3621393bc97SHong Zhang       k = lnk[k];
3631393bc97SHong Zhang     }
3641393bc97SHong Zhang     bdiag[i] = bi[i] + nzbd;
3651393bc97SHong Zhang 
3661393bc97SHong Zhang     /* if free space is not available, make more free space */
3671393bc97SHong Zhang     if (current_space->local_remaining<nzi) {
3681393bc97SHong Zhang       nnz = (n - i)*nzi; /* estimated and max additional space needed */
369a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
3701393bc97SHong Zhang       reallocs++;
3711393bc97SHong Zhang     }
3721393bc97SHong Zhang 
3731393bc97SHong Zhang     /* copy data into free space, then initialize lnk */
3741393bc97SHong Zhang     ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
3751393bc97SHong Zhang     bi_ptr[i] = current_space->array;
3761393bc97SHong Zhang     current_space->array           += nzi;
3771393bc97SHong Zhang     current_space->local_used      += nzi;
3781393bc97SHong Zhang     current_space->local_remaining -= nzi;
379289bc588SBarry Smith   }
3806cf91177SBarry Smith #if defined(PETSC_USE_INFO)
381464e8d44SSatish Balay   if (ai[n] != 0) {
3821393bc97SHong Zhang     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
383ae15b995SBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr);
384ae15b995SBarry Smith     ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
385ae15b995SBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr);
386ae15b995SBarry Smith     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
38705bf559cSSatish Balay   } else {
388ae15b995SBarry Smith     ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr);
38905bf559cSSatish Balay   }
39063ba0a88SBarry Smith #endif
3912fb3ed76SBarry Smith 
392898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
393898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
3941987afe7SBarry Smith 
3951393bc97SHong Zhang   /* destroy list of free space and other temporary array(s) */
3961393bc97SHong Zhang   ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
397a1a86e44SBarry Smith   ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr);
3981393bc97SHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
39935baf27eSSatish Balay   ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr);
400289bc588SBarry Smith 
401289bc588SBarry Smith   /* put together the new matrix */
4025c9eb25fSBarry Smith   ierr = MatSeqAIJSetPreallocation_SeqAIJ(*B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
40352e6d16bSBarry Smith   ierr = PetscLogObjectParent(*B,isicol);CHKERRQ(ierr);
404416022c9SBarry Smith   b    = (Mat_SeqAIJ*)(*B)->data;
405e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
406e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
4077c922b88SBarry Smith   b->singlemalloc = PETSC_FALSE;
4081393bc97SHong Zhang   ierr          = PetscMalloc((bi[n]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);
4091393bc97SHong Zhang   b->j          = bj;
4101393bc97SHong Zhang   b->i          = bi;
4111393bc97SHong Zhang   b->diag       = bdiag;
412416022c9SBarry Smith   b->ilen       = 0;
413416022c9SBarry Smith   b->imax       = 0;
414416022c9SBarry Smith   b->row        = isrow;
415416022c9SBarry Smith   b->col        = iscol;
416c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
417c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
41882bf6240SBarry Smith   b->icol       = isicol;
41987828ca2SBarry Smith   ierr          = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
4201393bc97SHong Zhang 
4211393bc97SHong Zhang   /* In b structure:  Free imax, ilen, old a, old j.  Allocate solve_work, new a, new j */
4221393bc97SHong Zhang   ierr = PetscLogObjectMemory(*B,(bi[n]-n)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
4231393bc97SHong Zhang   b->maxnz = b->nz = bi[n] ;
4247fd98bd6SLois Curfman McInnes 
4255c9eb25fSBarry Smith   (*B)->factor                 =  MAT_FACTOR_LU;
426418422e8SSatish Balay   (*B)->info.factor_mallocs    = reallocs;
427ae068f58SLois Curfman McInnes   (*B)->info.fill_ratio_given  = f;
428703deb49SSatish Balay 
429e93ccc4dSBarry Smith   if (ai[n] != 0) {
4301393bc97SHong Zhang     (*B)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]);
431eea2913aSSatish Balay   } else {
432eea2913aSSatish Balay     (*B)->info.fill_ratio_needed = 0.0;
433eea2913aSSatish Balay   }
434*db4efbfdSBarry Smith   (*B)->ops->lufactornumeric  = MatLUFactorNumeric_SeqAIJ;
435*db4efbfdSBarry Smith   (*B)->ops->solve            = MatSolve_SeqAIJ;
436*db4efbfdSBarry Smith   (*B)->ops->solvetranspose   = MatSolveTranspose_SeqAIJ;
437*db4efbfdSBarry Smith   /* switch to inodes if appropriate */
4384846f1f5SKris Buschelman   ierr = MatLUFactorSymbolic_Inode(A,isrow,iscol,info,B);CHKERRQ(ierr);
4393a40ed3dSBarry Smith   PetscFunctionReturn(0);
440289bc588SBarry Smith }
4411393bc97SHong Zhang 
4425b5bc046SBarry Smith /*
4435b5bc046SBarry Smith     Trouble in factorization, should we dump the original matrix?
4445b5bc046SBarry Smith */
4455b5bc046SBarry Smith #undef __FUNCT__
4465b5bc046SBarry Smith #define __FUNCT__ "MatFactorDumpMatrix"
4475b5bc046SBarry Smith PetscErrorCode MatFactorDumpMatrix(Mat A)
4485b5bc046SBarry Smith {
4495b5bc046SBarry Smith   PetscErrorCode ierr;
4505b5bc046SBarry Smith   PetscTruth     flg;
4515b5bc046SBarry Smith 
4525b5bc046SBarry Smith   PetscFunctionBegin;
4535b5bc046SBarry Smith   ierr = PetscOptionsHasName(PETSC_NULL,"-mat_factor_dump_on_error",&flg);CHKERRQ(ierr);
4545b5bc046SBarry Smith   if (flg) {
4555b5bc046SBarry Smith     PetscViewer viewer;
4565b5bc046SBarry Smith     char        filename[PETSC_MAX_PATH_LEN];
4575b5bc046SBarry Smith 
4585b5bc046SBarry Smith     ierr = PetscSNPrintf(filename,PETSC_MAX_PATH_LEN,"matrix_factor_error.%d",PetscGlobalRank);CHKERRQ(ierr);
4597adad957SLisandro Dalcin     ierr = PetscViewerBinaryOpen(((PetscObject)A)->comm,filename,FILE_MODE_WRITE,&viewer);CHKERRQ(ierr);
4605b5bc046SBarry Smith     ierr = MatView(A,viewer);CHKERRQ(ierr);
4615b5bc046SBarry Smith     ierr = PetscViewerDestroy(viewer);CHKERRQ(ierr);
4625b5bc046SBarry Smith   }
4635b5bc046SBarry Smith   PetscFunctionReturn(0);
4645b5bc046SBarry Smith }
4655b5bc046SBarry Smith 
466*db4efbfdSBarry Smith extern PetscErrorCode MatSolve_Inode(Mat,Vec,Vec);
467*db4efbfdSBarry Smith 
4680a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
4694a2ae208SSatish Balay #undef __FUNCT__
4704a2ae208SSatish Balay #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ"
471af281ebdSHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ(Mat A,MatFactorInfo *info,Mat *B)
472289bc588SBarry Smith {
473416022c9SBarry Smith   Mat            C=*B;
474416022c9SBarry Smith   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data;
47582bf6240SBarry Smith   IS             isrow = b->row,isicol = b->icol;
4766849ba73SBarry Smith   PetscErrorCode ierr;
477d0f46423SBarry Smith   PetscInt       *r,*ic,i,j,n=A->rmap->n,*bi=b->i,*bj=b->j;
478b3bf805bSHong Zhang   PetscInt       *ajtmp,*bjtmp,nz,row,*ics;
479d2276718SHong Zhang   PetscInt       *diag_offset = b->diag,diag,*pj;
48054f21887SBarry Smith   PetscScalar    *rtmp,*pc,multiplier,*rtmps;
48154f21887SBarry Smith   MatScalar      *v,*pv;
4826a7c8fc2SHong Zhang   PetscScalar    d;
4836a7c8fc2SHong Zhang   PetscReal      rs;
484b3bf805bSHong Zhang   LUShift_Ctx    sctx;
48542898933SHong Zhang   PetscInt       newshift,*ddiag;
486289bc588SBarry Smith 
4873a40ed3dSBarry Smith   PetscFunctionBegin;
48878b31e54SBarry Smith   ierr  = ISGetIndices(isrow,&r);CHKERRQ(ierr);
48978b31e54SBarry Smith   ierr  = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
49087828ca2SBarry Smith   ierr  = PetscMalloc((n+1)*sizeof(PetscScalar),&rtmp);CHKERRQ(ierr);
49187828ca2SBarry Smith   ierr  = PetscMemzero(rtmp,(n+1)*sizeof(PetscScalar));CHKERRQ(ierr);
492010a20caSHong Zhang   rtmps = rtmp; ics = ic;
493289bc588SBarry Smith 
494ada7143aSSatish Balay   sctx.shift_top  = 0;
495ada7143aSSatish Balay   sctx.nshift_max = 0;
496ada7143aSSatish Balay   sctx.shift_lo   = 0;
497ada7143aSSatish Balay   sctx.shift_hi   = 0;
498ada7143aSSatish Balay 
499118f5789SBarry Smith   /* if both shift schemes are chosen by user, only use info->shiftpd */
500118f5789SBarry Smith   if (info->shiftpd && info->shiftnz) info->shiftnz = 0.0;
5011a890ab1SHong Zhang   if (info->shiftpd) { /* set sctx.shift_top=max{rs} */
5029deb78dcSHong Zhang     PetscInt *aai = a->i;
50342898933SHong Zhang     ddiag          = a->diag;
504118f5789SBarry Smith     sctx.shift_top = 0;
5056cc28720Svictorle     for (i=0; i<n; i++) {
5069f95998fSHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
5079f95998fSHong Zhang       d  = (a->a)[ddiag[i]];
5081a890ab1SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
509010a20caSHong Zhang       v  = a->a+aai[i];
510e24cfe64SHong Zhang       nz = aai[i+1] - aai[i];
511e24cfe64SHong Zhang       for (j=0; j<nz; j++)
5121a890ab1SHong Zhang 	rs += PetscAbsScalar(v[j]);
5131a890ab1SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
5146cc28720Svictorle     }
515c3af1dc1SHong Zhang     if (sctx.shift_top < info->zeropivot) sctx.shift_top = info->zeropivot;
516118f5789SBarry Smith     sctx.shift_top    *= 1.1;
517d2276718SHong Zhang     sctx.nshift_max   = 5;
518d2276718SHong Zhang     sctx.shift_lo     = 0.;
519d2276718SHong Zhang     sctx.shift_hi     = 1.;
520a0a356efSHong Zhang   }
521a0a356efSHong Zhang 
522a0a356efSHong Zhang   sctx.shift_amount = 0;
523a0a356efSHong Zhang   sctx.nshift       = 0;
524085256b3SBarry Smith   do {
525d2276718SHong Zhang     sctx.lushift = PETSC_FALSE;
526289bc588SBarry Smith     for (i=0; i<n; i++){
527b3bf805bSHong Zhang       nz    = bi[i+1] - bi[i];
528b3bf805bSHong Zhang       bjtmp = bj + bi[i];
529b3bf805bSHong Zhang       for  (j=0; j<nz; j++) rtmps[bjtmp[j]] = 0.0;
530289bc588SBarry Smith 
531289bc588SBarry Smith       /* load in initial (unfactored row) */
532416022c9SBarry Smith       nz    = a->i[r[i]+1] - a->i[r[i]];
533b3bf805bSHong Zhang       ajtmp = a->j + a->i[r[i]];
534010a20caSHong Zhang       v     = a->a + a->i[r[i]];
535085256b3SBarry Smith       for (j=0; j<nz; j++) {
536b3bf805bSHong Zhang         rtmp[ics[ajtmp[j]]] = v[j];
537085256b3SBarry Smith       }
538d2276718SHong Zhang       rtmp[ics[r[i]]] += sctx.shift_amount; /* shift the diagonal of the matrix */
539289bc588SBarry Smith 
540b3bf805bSHong Zhang       row = *bjtmp++;
541f2582111SSatish Balay       while  (row < i) {
5428c37ef55SBarry Smith         pc = rtmp + row;
5438c37ef55SBarry Smith         if (*pc != 0.0) {
544010a20caSHong Zhang           pv         = b->a + diag_offset[row];
545010a20caSHong Zhang           pj         = b->j + diag_offset[row] + 1;
54635aab85fSBarry Smith           multiplier = *pc / *pv++;
5478c37ef55SBarry Smith           *pc        = multiplier;
548b3bf805bSHong Zhang           nz         = bi[row+1] - diag_offset[row] - 1;
549f2582111SSatish Balay           for (j=0; j<nz; j++) rtmps[pj[j]] -= multiplier * pv[j];
550efee365bSSatish Balay           ierr = PetscLogFlops(2*nz);CHKERRQ(ierr);
551289bc588SBarry Smith         }
552b3bf805bSHong Zhang         row = *bjtmp++;
553289bc588SBarry Smith       }
554416022c9SBarry Smith       /* finished row so stick it into b->a */
555b3bf805bSHong Zhang       pv   = b->a + bi[i] ;
556b3bf805bSHong Zhang       pj   = b->j + bi[i] ;
557b3bf805bSHong Zhang       nz   = bi[i+1] - bi[i];
558b3bf805bSHong Zhang       diag = diag_offset[i] - bi[i];
559d3d32019SBarry Smith       rs   = 0.0;
560d3d32019SBarry Smith       for (j=0; j<nz; j++) {
561d3d32019SBarry Smith         pv[j] = rtmps[pj[j]];
562d3d32019SBarry Smith         if (j != diag) rs += PetscAbsScalar(pv[j]);
563d3d32019SBarry Smith       }
564d2276718SHong Zhang 
565b3bf805bSHong Zhang       /* 9/13/02 Victor Eijkhout suggested scaling zeropivot by rs for matrices with funny scalings */
566d2276718SHong Zhang       sctx.rs  = rs;
567d2276718SHong Zhang       sctx.pv  = pv[diag];
568c6b1f410SHong Zhang       ierr = MatLUCheckShift_inline(info,sctx,i,newshift);CHKERRQ(ierr);
5695b5bc046SBarry Smith       if (newshift == 1) break;
57035aab85fSBarry Smith     }
571d2276718SHong Zhang 
572118f5789SBarry Smith     if (info->shiftpd && !sctx.lushift && info->shift_fraction>0 && sctx.nshift<sctx.nshift_max) {
5736cc28720Svictorle       /*
5746c037d1bSvictorle        * if no shift in this attempt & shifting & started shifting & can refine,
5756cc28720Svictorle        * then try lower shift
5766cc28720Svictorle        */
577d2276718SHong Zhang       sctx.shift_hi        = info->shift_fraction;
578d2276718SHong Zhang       info->shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
579d2276718SHong Zhang       sctx.shift_amount    = info->shift_fraction * sctx.shift_top;
580d2276718SHong Zhang       sctx.lushift         = PETSC_TRUE;
581d2276718SHong Zhang       sctx.nshift++;
5826cc28720Svictorle     }
583d2276718SHong Zhang   } while (sctx.lushift);
5840f11f9aeSSatish Balay 
58535aab85fSBarry Smith   /* invert diagonal entries for simplier triangular solves */
58635aab85fSBarry Smith   for (i=0; i<n; i++) {
587010a20caSHong Zhang     b->a[diag_offset[i]] = 1.0/b->a[diag_offset[i]];
58835aab85fSBarry Smith   }
589606d414cSSatish Balay   ierr = PetscFree(rtmp);CHKERRQ(ierr);
59078b31e54SBarry Smith   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
59178b31e54SBarry Smith   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
592*db4efbfdSBarry Smith   if (b->inode.use) {
593*db4efbfdSBarry Smith     C->ops->solve   = MatSolve_Inode;
594*db4efbfdSBarry Smith   } else {
595*db4efbfdSBarry Smith     C->ops->solve   = MatSolve_SeqAIJ;
596*db4efbfdSBarry Smith   }
597*db4efbfdSBarry Smith   C->ops->solveadd           = MatSolveAdd_SeqAIJ;
598*db4efbfdSBarry Smith   C->ops->solvetranspose     = MatSolveTranspose_SeqAIJ;
599*db4efbfdSBarry Smith   C->ops->solvetransposeadd  = MatSolveTransposeAdd_SeqAIJ;
600c456f294SBarry Smith   C->assembled    = PETSC_TRUE;
6015c9eb25fSBarry Smith   C->preallocated = PETSC_TRUE;
602d0f46423SBarry Smith   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
603d2276718SHong Zhang   if (sctx.nshift){
604118f5789SBarry Smith     if (info->shiftnz) {
6051e2582c4SBarry Smith       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
606118f5789SBarry Smith     } else if (info->shiftpd) {
6071e2582c4SBarry Smith       ierr = PetscInfo4(A,"number of shift_pd tries %D, shift_amount %G, diagonal shifted up by %e fraction top_value %e\n",sctx.nshift,sctx.shift_amount,info->shift_fraction,sctx.shift_top);CHKERRQ(ierr);
6086cc28720Svictorle     }
6090697b6caSHong Zhang   }
6103a40ed3dSBarry Smith   PetscFunctionReturn(0);
611289bc588SBarry Smith }
6120889b5dcSvictorle 
613137fb511SHong Zhang /*
614137fb511SHong Zhang    This routine implements inplace ILU(0) with row or/and column permutations.
615137fb511SHong Zhang    Input:
616137fb511SHong Zhang      A - original matrix
617137fb511SHong Zhang    Output;
618137fb511SHong Zhang      A - a->i (rowptr) is same as original rowptr, but factored i-the row is stored in rowperm[i]
619137fb511SHong Zhang          a->j (col index) is permuted by the inverse of colperm, then sorted
620137fb511SHong Zhang          a->a reordered accordingly with a->j
621137fb511SHong Zhang          a->diag (ptr to diagonal elements) is updated.
622137fb511SHong Zhang */
623137fb511SHong Zhang #undef __FUNCT__
624137fb511SHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_InplaceWithPerm"
625137fb511SHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ_InplaceWithPerm(Mat A,MatFactorInfo *info,Mat *B)
626137fb511SHong Zhang {
627b051339dSHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
628b051339dSHong Zhang   IS             isrow = a->row,isicol = a->icol;
629137fb511SHong Zhang   PetscErrorCode ierr;
630d0f46423SBarry Smith   PetscInt       *r,*ic,i,j,n=A->rmap->n,*ai=a->i,*aj=a->j;
631b051339dSHong Zhang   PetscInt       *ajtmp,nz,row,*ics;
632b051339dSHong Zhang   PetscInt       *diag = a->diag,nbdiag,*pj;
633a77337e4SBarry Smith   PetscScalar    *rtmp,*pc,multiplier,d;
634a77337e4SBarry Smith   MatScalar      *v,*pv;
635137fb511SHong Zhang   PetscReal      rs;
636137fb511SHong Zhang   LUShift_Ctx    sctx;
637b051339dSHong Zhang   PetscInt       newshift;
638137fb511SHong Zhang 
639137fb511SHong Zhang   PetscFunctionBegin;
640b051339dSHong Zhang   if (A != *B) SETERRQ(PETSC_ERR_ARG_INCOMP,"input and output matrix must have same address");
641137fb511SHong Zhang   ierr  = ISGetIndices(isrow,&r);CHKERRQ(ierr);
642137fb511SHong Zhang   ierr  = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
643137fb511SHong Zhang   ierr  = PetscMalloc((n+1)*sizeof(PetscScalar),&rtmp);CHKERRQ(ierr);
644137fb511SHong Zhang   ierr  = PetscMemzero(rtmp,(n+1)*sizeof(PetscScalar));CHKERRQ(ierr);
645b051339dSHong Zhang   ics = ic;
646137fb511SHong Zhang 
647137fb511SHong Zhang   sctx.shift_top  = 0;
648137fb511SHong Zhang   sctx.nshift_max = 0;
649137fb511SHong Zhang   sctx.shift_lo   = 0;
650137fb511SHong Zhang   sctx.shift_hi   = 0;
651137fb511SHong Zhang 
652137fb511SHong Zhang   /* if both shift schemes are chosen by user, only use info->shiftpd */
653137fb511SHong Zhang   if (info->shiftpd && info->shiftnz) info->shiftnz = 0.0;
654137fb511SHong Zhang   if (info->shiftpd) { /* set sctx.shift_top=max{rs} */
655137fb511SHong Zhang     sctx.shift_top = 0;
656137fb511SHong Zhang     for (i=0; i<n; i++) {
657137fb511SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
658b051339dSHong Zhang       d  = (a->a)[diag[i]];
659137fb511SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
660b051339dSHong Zhang       v  = a->a+ai[i];
661b051339dSHong Zhang       nz = ai[i+1] - ai[i];
662137fb511SHong Zhang       for (j=0; j<nz; j++)
663137fb511SHong Zhang 	rs += PetscAbsScalar(v[j]);
664137fb511SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
665137fb511SHong Zhang     }
666137fb511SHong Zhang     if (sctx.shift_top < info->zeropivot) sctx.shift_top = info->zeropivot;
667137fb511SHong Zhang     sctx.shift_top    *= 1.1;
668137fb511SHong Zhang     sctx.nshift_max   = 5;
669137fb511SHong Zhang     sctx.shift_lo     = 0.;
670137fb511SHong Zhang     sctx.shift_hi     = 1.;
671137fb511SHong Zhang   }
672137fb511SHong Zhang 
673137fb511SHong Zhang   sctx.shift_amount = 0;
674137fb511SHong Zhang   sctx.nshift       = 0;
675137fb511SHong Zhang   do {
676137fb511SHong Zhang     sctx.lushift = PETSC_FALSE;
677137fb511SHong Zhang     for (i=0; i<n; i++){
678b051339dSHong Zhang       /* load in initial unfactored row */
679b051339dSHong Zhang       nz    = ai[r[i]+1] - ai[r[i]];
680b051339dSHong Zhang       ajtmp = aj + ai[r[i]];
681b051339dSHong Zhang       v     = a->a + ai[r[i]];
682137fb511SHong Zhang       /* sort permuted ajtmp and values v accordingly */
683b051339dSHong Zhang       for (j=0; j<nz; j++) ajtmp[j] = ics[ajtmp[j]];
684137fb511SHong Zhang       ierr = PetscSortIntWithScalarArray(nz,ajtmp,v);CHKERRQ(ierr);
685137fb511SHong Zhang 
686b051339dSHong Zhang       diag[r[i]] = ai[r[i]];
687137fb511SHong Zhang       for (j=0; j<nz; j++) {
688137fb511SHong Zhang         rtmp[ajtmp[j]] = v[j];
689b051339dSHong Zhang         if (ajtmp[j] < i) diag[r[i]]++; /* update a->diag */
690137fb511SHong Zhang       }
691137fb511SHong Zhang       rtmp[r[i]] += sctx.shift_amount; /* shift the diagonal of the matrix */
692137fb511SHong Zhang 
693b051339dSHong Zhang       row = *ajtmp++;
694137fb511SHong Zhang       while  (row < i) {
695137fb511SHong Zhang         pc = rtmp + row;
696137fb511SHong Zhang         if (*pc != 0.0) {
697b051339dSHong Zhang           pv         = a->a + diag[r[row]];
698b051339dSHong Zhang           pj         = aj + diag[r[row]] + 1;
699137fb511SHong Zhang 
700137fb511SHong Zhang           multiplier = *pc / *pv++;
701137fb511SHong Zhang           *pc        = multiplier;
702b051339dSHong Zhang           nz         = ai[r[row]+1] - diag[r[row]] - 1;
703b051339dSHong Zhang           for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j];
704137fb511SHong Zhang           ierr = PetscLogFlops(2*nz);CHKERRQ(ierr);
705137fb511SHong Zhang         }
706b051339dSHong Zhang         row = *ajtmp++;
707137fb511SHong Zhang       }
708b051339dSHong Zhang       /* finished row so overwrite it onto a->a */
709b051339dSHong Zhang       pv   = a->a + ai[r[i]] ;
710b051339dSHong Zhang       pj   = aj + ai[r[i]] ;
711b051339dSHong Zhang       nz   = ai[r[i]+1] - ai[r[i]];
712b051339dSHong Zhang       nbdiag = diag[r[i]] - ai[r[i]]; /* num of entries before the diagonal */
713137fb511SHong Zhang 
714137fb511SHong Zhang       rs   = 0.0;
715137fb511SHong Zhang       for (j=0; j<nz; j++) {
716b051339dSHong Zhang         pv[j] = rtmp[pj[j]];
717b051339dSHong Zhang         if (j != nbdiag) rs += PetscAbsScalar(pv[j]);
718137fb511SHong Zhang       }
719137fb511SHong Zhang 
720137fb511SHong Zhang       /* 9/13/02 Victor Eijkhout suggested scaling zeropivot by rs for matrices with funny scalings */
721137fb511SHong Zhang       sctx.rs  = rs;
722b051339dSHong Zhang       sctx.pv  = pv[nbdiag];
723c6b1f410SHong Zhang       ierr = MatLUCheckShift_inline(info,sctx,i,newshift);CHKERRQ(ierr);
724137fb511SHong Zhang       if (newshift == 1) break;
725137fb511SHong Zhang     }
726137fb511SHong Zhang 
727137fb511SHong Zhang     if (info->shiftpd && !sctx.lushift && info->shift_fraction>0 && sctx.nshift<sctx.nshift_max) {
728137fb511SHong Zhang       /*
729137fb511SHong Zhang        * if no shift in this attempt & shifting & started shifting & can refine,
730137fb511SHong Zhang        * then try lower shift
731137fb511SHong Zhang        */
732137fb511SHong Zhang       sctx.shift_hi        = info->shift_fraction;
733137fb511SHong Zhang       info->shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
734137fb511SHong Zhang       sctx.shift_amount    = info->shift_fraction * sctx.shift_top;
735137fb511SHong Zhang       sctx.lushift         = PETSC_TRUE;
736137fb511SHong Zhang       sctx.nshift++;
737137fb511SHong Zhang     }
738137fb511SHong Zhang   } while (sctx.lushift);
739137fb511SHong Zhang 
740137fb511SHong Zhang   /* invert diagonal entries for simplier triangular solves */
741137fb511SHong Zhang   for (i=0; i<n; i++) {
742b051339dSHong Zhang     a->a[diag[r[i]]] = 1.0/a->a[diag[r[i]]];
743137fb511SHong Zhang   }
744137fb511SHong Zhang 
745137fb511SHong Zhang   ierr = PetscFree(rtmp);CHKERRQ(ierr);
746137fb511SHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
747137fb511SHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
748b051339dSHong Zhang   A->ops->solve             = MatSolve_SeqAIJ_InplaceWithPerm;
749*db4efbfdSBarry Smith   A->ops->solveadd          = MatSolveAdd_SeqAIJ;
750*db4efbfdSBarry Smith   A->ops->solvetranspose    = MatSolveTranspose_SeqAIJ;
751*db4efbfdSBarry Smith   A->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ;
752b051339dSHong Zhang   A->assembled = PETSC_TRUE;
7535c9eb25fSBarry Smith   A->preallocated = PETSC_TRUE;
754d0f46423SBarry Smith   ierr = PetscLogFlops(A->cmap->n);CHKERRQ(ierr);
755137fb511SHong Zhang   if (sctx.nshift){
756137fb511SHong Zhang     if (info->shiftnz) {
7571e2582c4SBarry Smith       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
758137fb511SHong Zhang     } else if (info->shiftpd) {
7591e2582c4SBarry Smith       ierr = PetscInfo4(A,"number of shift_pd tries %D, shift_amount %G, diagonal shifted up by %e fraction top_value %e\n",sctx.nshift,sctx.shift_amount,info->shift_fraction,sctx.shift_top);CHKERRQ(ierr);
760137fb511SHong Zhang     }
761137fb511SHong Zhang   }
762137fb511SHong Zhang   PetscFunctionReturn(0);
763137fb511SHong Zhang }
764137fb511SHong Zhang 
7650a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
7664a2ae208SSatish Balay #undef __FUNCT__
7674a2ae208SSatish Balay #define __FUNCT__ "MatLUFactor_SeqAIJ"
768dfbe8321SBarry Smith PetscErrorCode MatLUFactor_SeqAIJ(Mat A,IS row,IS col,MatFactorInfo *info)
769da3a660dSBarry Smith {
770dfbe8321SBarry Smith   PetscErrorCode ierr;
771416022c9SBarry Smith   Mat            C;
772416022c9SBarry Smith 
7733a40ed3dSBarry Smith   PetscFunctionBegin;
77443244d56SBarry Smith   ierr = MatGetFactor(A,MAT_SOLVER_PETSC,MAT_FACTOR_LU,&C);CHKERRQ(ierr);
7759e046878SBarry Smith   ierr = MatLUFactorSymbolic(A,row,col,info,&C);CHKERRQ(ierr);
776af281ebdSHong Zhang   ierr = MatLUFactorNumeric(A,info,&C);CHKERRQ(ierr);
777*db4efbfdSBarry Smith   A->ops->solve            = C->ops->solve;
778*db4efbfdSBarry Smith   A->ops->solvetranspose   = C->ops->solvetranspose;
779273d9f13SBarry Smith   ierr = MatHeaderCopy(A,C);CHKERRQ(ierr);
78052e6d16bSBarry Smith   ierr = PetscLogObjectParent(A,((Mat_SeqAIJ*)(A->data))->icol);CHKERRQ(ierr);
7813a40ed3dSBarry Smith   PetscFunctionReturn(0);
782da3a660dSBarry Smith }
7830a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
7844a2ae208SSatish Balay #undef __FUNCT__
7854a2ae208SSatish Balay #define __FUNCT__ "MatSolve_SeqAIJ"
786dfbe8321SBarry Smith PetscErrorCode MatSolve_SeqAIJ(Mat A,Vec bb,Vec xx)
7878c37ef55SBarry Smith {
788416022c9SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
789416022c9SBarry Smith   IS                iscol = a->col,isrow = a->row;
7906849ba73SBarry Smith   PetscErrorCode    ierr;
791d0f46423SBarry Smith   PetscInt          *r,*c,i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
79238baddfdSBarry Smith   PetscInt          nz,*rout,*cout;
793dd6ea824SBarry Smith   PetscScalar       *x,*tmp,*tmps,sum;
794d9fead3dSBarry Smith   const PetscScalar *b;
795dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
7968c37ef55SBarry Smith 
7973a40ed3dSBarry Smith   PetscFunctionBegin;
7983a40ed3dSBarry Smith   if (!n) PetscFunctionReturn(0);
79991bf9adeSBarry Smith 
800d9fead3dSBarry Smith   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
8011ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
802416022c9SBarry Smith   tmp  = a->solve_work;
8038c37ef55SBarry Smith 
8043b2fbd54SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
8053b2fbd54SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
8068c37ef55SBarry Smith 
8078c37ef55SBarry Smith   /* forward solve the lower triangular */
8088c37ef55SBarry Smith   tmp[0] = b[*r++];
809010a20caSHong Zhang   tmps   = tmp;
8108c37ef55SBarry Smith   for (i=1; i<n; i++) {
811010a20caSHong Zhang     v   = aa + ai[i] ;
812010a20caSHong Zhang     vi  = aj + ai[i] ;
81353e7425aSBarry Smith     nz  = a->diag[i] - ai[i];
81453e7425aSBarry Smith     sum = b[*r++];
815ed480e8bSBarry Smith     SPARSEDENSEMDOT(sum,tmps,v,vi,nz);
8168c37ef55SBarry Smith     tmp[i] = sum;
8178c37ef55SBarry Smith   }
8188c37ef55SBarry Smith 
8198c37ef55SBarry Smith   /* backward solve the upper triangular */
8208c37ef55SBarry Smith   for (i=n-1; i>=0; i--){
821010a20caSHong Zhang     v   = aa + a->diag[i] + 1;
822010a20caSHong Zhang     vi  = aj + a->diag[i] + 1;
823416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
8248c37ef55SBarry Smith     sum = tmp[i];
825ed480e8bSBarry Smith     SPARSEDENSEMDOT(sum,tmps,v,vi,nz);
826010a20caSHong Zhang     x[*c--] = tmp[i] = sum*aa[a->diag[i]];
8278c37ef55SBarry Smith   }
8288c37ef55SBarry Smith 
8293b2fbd54SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
8303b2fbd54SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
831d9fead3dSBarry Smith   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
8321ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
833d0f46423SBarry Smith   ierr = PetscLogFlops(2*a->nz - A->cmap->n);CHKERRQ(ierr);
8343a40ed3dSBarry Smith   PetscFunctionReturn(0);
8358c37ef55SBarry Smith }
836026e39d0SSatish Balay 
8372bebee5dSHong Zhang #undef __FUNCT__
8382bebee5dSHong Zhang #define __FUNCT__ "MatMatSolve_SeqAIJ"
8392bebee5dSHong Zhang PetscErrorCode MatMatSolve_SeqAIJ(Mat A,Mat B,Mat X)
8402bebee5dSHong Zhang {
8412bebee5dSHong Zhang   Mat_SeqAIJ      *a = (Mat_SeqAIJ*)A->data;
8422bebee5dSHong Zhang   IS              iscol = a->col,isrow = a->row;
8432bebee5dSHong Zhang   PetscErrorCode  ierr;
844d0f46423SBarry Smith   PetscInt        *r,*c,i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
8452bebee5dSHong Zhang   PetscInt        nz,*rout,*cout,neq;
846dd6ea824SBarry Smith   PetscScalar     *x,*b,*tmp,*tmps,sum;
847dd6ea824SBarry Smith   const MatScalar *aa = a->a,*v;
848*db4efbfdSBarry Smith   PetscTruth      bisdense,xisdense;
8492bebee5dSHong Zhang 
8502bebee5dSHong Zhang   PetscFunctionBegin;
8512bebee5dSHong Zhang   if (!n) PetscFunctionReturn(0);
8522bebee5dSHong Zhang 
853*db4efbfdSBarry Smith   ierr = PetscTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr);
854*db4efbfdSBarry Smith   if (!bisdense) SETERRQ(PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix");
855*db4efbfdSBarry Smith   ierr = PetscTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr);
856*db4efbfdSBarry Smith   if (!xisdense) SETERRQ(PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix");
857*db4efbfdSBarry Smith 
8582bebee5dSHong Zhang   ierr = MatGetArray(B,&b);CHKERRQ(ierr);
8592bebee5dSHong Zhang   ierr = MatGetArray(X,&x);CHKERRQ(ierr);
8602bebee5dSHong Zhang 
8612bebee5dSHong Zhang   tmp  = a->solve_work;
8622bebee5dSHong Zhang   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
8632bebee5dSHong Zhang   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
8642bebee5dSHong Zhang 
8652bebee5dSHong Zhang   for (neq=0; neq<n; neq++){
8662bebee5dSHong Zhang     /* forward solve the lower triangular */
8672bebee5dSHong Zhang     tmp[0] = b[r[0]];
8682bebee5dSHong Zhang     tmps   = tmp;
8692bebee5dSHong Zhang     for (i=1; i<n; i++) {
8702bebee5dSHong Zhang       v   = aa + ai[i] ;
8712bebee5dSHong Zhang       vi  = aj + ai[i] ;
8722bebee5dSHong Zhang       nz  = a->diag[i] - ai[i];
8732bebee5dSHong Zhang       sum = b[r[i]];
8742bebee5dSHong Zhang       SPARSEDENSEMDOT(sum,tmps,v,vi,nz);
8752bebee5dSHong Zhang       tmp[i] = sum;
8762bebee5dSHong Zhang     }
8772bebee5dSHong Zhang     /* backward solve the upper triangular */
8782bebee5dSHong Zhang     for (i=n-1; i>=0; i--){
8792bebee5dSHong Zhang       v   = aa + a->diag[i] + 1;
8802bebee5dSHong Zhang       vi  = aj + a->diag[i] + 1;
8812bebee5dSHong Zhang       nz  = ai[i+1] - a->diag[i] - 1;
8822bebee5dSHong Zhang       sum = tmp[i];
8832bebee5dSHong Zhang       SPARSEDENSEMDOT(sum,tmps,v,vi,nz);
8842bebee5dSHong Zhang       x[c[i]] = tmp[i] = sum*aa[a->diag[i]];
8852bebee5dSHong Zhang     }
8862bebee5dSHong Zhang 
8872bebee5dSHong Zhang     b += n;
8882bebee5dSHong Zhang     x += n;
8892bebee5dSHong Zhang   }
8902bebee5dSHong Zhang   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
8912bebee5dSHong Zhang   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
8922bebee5dSHong Zhang   ierr = MatRestoreArray(B,&b);CHKERRQ(ierr);
8932bebee5dSHong Zhang   ierr = MatRestoreArray(X,&x);CHKERRQ(ierr);
8942bebee5dSHong Zhang   ierr = PetscLogFlops(n*(2*a->nz - n));CHKERRQ(ierr);
8952bebee5dSHong Zhang   PetscFunctionReturn(0);
8962bebee5dSHong Zhang }
8972bebee5dSHong Zhang 
898137fb511SHong Zhang #undef __FUNCT__
899137fb511SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_InplaceWithPerm"
900137fb511SHong Zhang PetscErrorCode MatSolve_SeqAIJ_InplaceWithPerm(Mat A,Vec bb,Vec xx)
901137fb511SHong Zhang {
902137fb511SHong Zhang   Mat_SeqAIJ      *a = (Mat_SeqAIJ*)A->data;
903137fb511SHong Zhang   IS              iscol = a->col,isrow = a->row;
904137fb511SHong Zhang   PetscErrorCode  ierr;
905d0f46423SBarry Smith   PetscInt        *r,*c,i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
906137fb511SHong Zhang   PetscInt        nz,*rout,*cout,row;
907dd6ea824SBarry Smith   PetscScalar     *x,*b,*tmp,*tmps,sum;
908dd6ea824SBarry Smith   const MatScalar *aa = a->a,*v;
909137fb511SHong Zhang 
910137fb511SHong Zhang   PetscFunctionBegin;
911137fb511SHong Zhang   if (!n) PetscFunctionReturn(0);
912137fb511SHong Zhang 
913137fb511SHong Zhang   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
914137fb511SHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
915137fb511SHong Zhang   tmp  = a->solve_work;
916137fb511SHong Zhang 
917137fb511SHong Zhang   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
918137fb511SHong Zhang   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
919137fb511SHong Zhang 
920137fb511SHong Zhang   /* forward solve the lower triangular */
921137fb511SHong Zhang   tmp[0] = b[*r++];
922137fb511SHong Zhang   tmps   = tmp;
923137fb511SHong Zhang   for (row=1; row<n; row++) {
924137fb511SHong Zhang     i   = rout[row]; /* permuted row */
925137fb511SHong Zhang     v   = aa + ai[i] ;
926137fb511SHong Zhang     vi  = aj + ai[i] ;
927137fb511SHong Zhang     nz  = a->diag[i] - ai[i];
928137fb511SHong Zhang     sum = b[*r++];
929137fb511SHong Zhang     SPARSEDENSEMDOT(sum,tmps,v,vi,nz);
930137fb511SHong Zhang     tmp[row] = sum;
931137fb511SHong Zhang   }
932137fb511SHong Zhang 
933137fb511SHong Zhang   /* backward solve the upper triangular */
934137fb511SHong Zhang   for (row=n-1; row>=0; row--){
935137fb511SHong Zhang     i   = rout[row]; /* permuted row */
936137fb511SHong Zhang     v   = aa + a->diag[i] + 1;
937137fb511SHong Zhang     vi  = aj + a->diag[i] + 1;
938137fb511SHong Zhang     nz  = ai[i+1] - a->diag[i] - 1;
939137fb511SHong Zhang     sum = tmp[row];
940137fb511SHong Zhang     SPARSEDENSEMDOT(sum,tmps,v,vi,nz);
941137fb511SHong Zhang     x[*c--] = tmp[row] = sum*aa[a->diag[i]];
942137fb511SHong Zhang   }
943137fb511SHong Zhang 
944137fb511SHong Zhang   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
945137fb511SHong Zhang   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
946137fb511SHong Zhang   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
947137fb511SHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
948d0f46423SBarry Smith   ierr = PetscLogFlops(2*a->nz - A->cmap->n);CHKERRQ(ierr);
949137fb511SHong Zhang   PetscFunctionReturn(0);
950137fb511SHong Zhang }
951137fb511SHong Zhang 
952930ae53cSBarry Smith /* ----------------------------------------------------------- */
9534a2ae208SSatish Balay #undef __FUNCT__
9544a2ae208SSatish Balay #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering"
955dfbe8321SBarry Smith PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering(Mat A,Vec bb,Vec xx)
956930ae53cSBarry Smith {
957930ae53cSBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
9586849ba73SBarry Smith   PetscErrorCode    ierr;
959d0f46423SBarry Smith   PetscInt          n = A->rmap->n;
960356650c2SBarry Smith   const PetscInt    *ai = a->i,*aj = a->j,*adiag = a->diag,*vi;
961356650c2SBarry Smith   PetscScalar       *x;
962356650c2SBarry Smith   const PetscScalar *b;
963dd6ea824SBarry Smith   const MatScalar   *aa = a->a;
964aa482453SBarry Smith #if !defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ)
965356650c2SBarry Smith   PetscInt          adiag_i,i,nz,ai_i;
966dd6ea824SBarry Smith   const MatScalar   *v;
967dd6ea824SBarry Smith   PetscScalar       sum;
968d85afc42SSatish Balay #endif
969930ae53cSBarry Smith 
9703a40ed3dSBarry Smith   PetscFunctionBegin;
9713a40ed3dSBarry Smith   if (!n) PetscFunctionReturn(0);
972930ae53cSBarry Smith 
973356650c2SBarry Smith   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
9741ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
975930ae53cSBarry Smith 
976aa482453SBarry Smith #if defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ)
9771c4feecaSSatish Balay   fortransolveaij_(&n,x,ai,aj,adiag,aa,b);
9781c4feecaSSatish Balay #else
979930ae53cSBarry Smith   /* forward solve the lower triangular */
980930ae53cSBarry Smith   x[0] = b[0];
981930ae53cSBarry Smith   for (i=1; i<n; i++) {
982930ae53cSBarry Smith     ai_i = ai[i];
983930ae53cSBarry Smith     v    = aa + ai_i;
984930ae53cSBarry Smith     vi   = aj + ai_i;
985930ae53cSBarry Smith     nz   = adiag[i] - ai_i;
986930ae53cSBarry Smith     sum  = b[i];
987930ae53cSBarry Smith     while (nz--) sum -= *v++ * x[*vi++];
988930ae53cSBarry Smith     x[i] = sum;
989930ae53cSBarry Smith   }
990930ae53cSBarry Smith 
991930ae53cSBarry Smith   /* backward solve the upper triangular */
992930ae53cSBarry Smith   for (i=n-1; i>=0; i--){
993930ae53cSBarry Smith     adiag_i = adiag[i];
994d708749eSBarry Smith     v       = aa + adiag_i + 1;
995d708749eSBarry Smith     vi      = aj + adiag_i + 1;
996930ae53cSBarry Smith     nz      = ai[i+1] - adiag_i - 1;
997930ae53cSBarry Smith     sum     = x[i];
998930ae53cSBarry Smith     while (nz--) sum -= *v++ * x[*vi++];
999930ae53cSBarry Smith     x[i]    = sum*aa[adiag_i];
1000930ae53cSBarry Smith   }
10011c4feecaSSatish Balay #endif
1002d0f46423SBarry Smith   ierr = PetscLogFlops(2*a->nz - A->cmap->n);CHKERRQ(ierr);
1003356650c2SBarry Smith   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
10041ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
10053a40ed3dSBarry Smith   PetscFunctionReturn(0);
1006930ae53cSBarry Smith }
1007930ae53cSBarry Smith 
10084a2ae208SSatish Balay #undef __FUNCT__
10094a2ae208SSatish Balay #define __FUNCT__ "MatSolveAdd_SeqAIJ"
1010dfbe8321SBarry Smith PetscErrorCode MatSolveAdd_SeqAIJ(Mat A,Vec bb,Vec yy,Vec xx)
1011da3a660dSBarry Smith {
1012416022c9SBarry Smith   Mat_SeqAIJ      *a = (Mat_SeqAIJ*)A->data;
1013416022c9SBarry Smith   IS              iscol = a->col,isrow = a->row;
10146849ba73SBarry Smith   PetscErrorCode  ierr;
1015d0f46423SBarry Smith   PetscInt        *r,*c,i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
101638baddfdSBarry Smith   PetscInt        nz,*rout,*cout;
1017dd6ea824SBarry Smith   PetscScalar     *x,*b,*tmp,sum;
1018dd6ea824SBarry Smith   const MatScalar *aa = a->a,*v;
1019da3a660dSBarry Smith 
10203a40ed3dSBarry Smith   PetscFunctionBegin;
102178b31e54SBarry Smith   if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);}
1022da3a660dSBarry Smith 
10231ebc52fbSHong Zhang   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
10241ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1025416022c9SBarry Smith   tmp  = a->solve_work;
1026da3a660dSBarry Smith 
10273b2fbd54SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
10283b2fbd54SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
1029da3a660dSBarry Smith 
1030da3a660dSBarry Smith   /* forward solve the lower triangular */
1031da3a660dSBarry Smith   tmp[0] = b[*r++];
1032da3a660dSBarry Smith   for (i=1; i<n; i++) {
1033010a20caSHong Zhang     v   = aa + ai[i] ;
1034010a20caSHong Zhang     vi  = aj + ai[i] ;
1035416022c9SBarry Smith     nz  = a->diag[i] - ai[i];
1036da3a660dSBarry Smith     sum = b[*r++];
1037010a20caSHong Zhang     while (nz--) sum -= *v++ * tmp[*vi++ ];
1038da3a660dSBarry Smith     tmp[i] = sum;
1039da3a660dSBarry Smith   }
1040da3a660dSBarry Smith 
1041da3a660dSBarry Smith   /* backward solve the upper triangular */
1042da3a660dSBarry Smith   for (i=n-1; i>=0; i--){
1043010a20caSHong Zhang     v   = aa + a->diag[i] + 1;
1044010a20caSHong Zhang     vi  = aj + a->diag[i] + 1;
1045416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
1046da3a660dSBarry Smith     sum = tmp[i];
1047010a20caSHong Zhang     while (nz--) sum -= *v++ * tmp[*vi++ ];
1048010a20caSHong Zhang     tmp[i] = sum*aa[a->diag[i]];
1049da3a660dSBarry Smith     x[*c--] += tmp[i];
1050da3a660dSBarry Smith   }
1051da3a660dSBarry Smith 
10523b2fbd54SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
10533b2fbd54SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
10541ebc52fbSHong Zhang   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
10551ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1056efee365bSSatish Balay   ierr = PetscLogFlops(2*a->nz);CHKERRQ(ierr);
1057898183ecSLois Curfman McInnes 
10583a40ed3dSBarry Smith   PetscFunctionReturn(0);
1059da3a660dSBarry Smith }
1060da3a660dSBarry Smith /* -------------------------------------------------------------------*/
10614a2ae208SSatish Balay #undef __FUNCT__
10624a2ae208SSatish Balay #define __FUNCT__ "MatSolveTranspose_SeqAIJ"
1063dfbe8321SBarry Smith PetscErrorCode MatSolveTranspose_SeqAIJ(Mat A,Vec bb,Vec xx)
1064da3a660dSBarry Smith {
1065416022c9SBarry Smith   Mat_SeqAIJ      *a = (Mat_SeqAIJ*)A->data;
10662235e667SBarry Smith   IS              iscol = a->col,isrow = a->row;
10676849ba73SBarry Smith   PetscErrorCode  ierr;
1068d0f46423SBarry Smith   PetscInt        *r,*c,i,n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
106938baddfdSBarry Smith   PetscInt        nz,*rout,*cout,*diag = a->diag;
1070dd6ea824SBarry Smith   PetscScalar     *x,*b,*tmp,s1;
1071dd6ea824SBarry Smith   const MatScalar *aa = a->a,*v;
1072da3a660dSBarry Smith 
10733a40ed3dSBarry Smith   PetscFunctionBegin;
10741ebc52fbSHong Zhang   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
10751ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1076416022c9SBarry Smith   tmp  = a->solve_work;
1077da3a660dSBarry Smith 
10782235e667SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
10792235e667SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
1080da3a660dSBarry Smith 
1081da3a660dSBarry Smith   /* copy the b into temp work space according to permutation */
10822235e667SBarry Smith   for (i=0; i<n; i++) tmp[i] = b[c[i]];
1083da3a660dSBarry Smith 
1084da3a660dSBarry Smith   /* forward solve the U^T */
1085da3a660dSBarry Smith   for (i=0; i<n; i++) {
1086010a20caSHong Zhang     v   = aa + diag[i] ;
1087010a20caSHong Zhang     vi  = aj + diag[i] + 1;
1088f1af5d2fSBarry Smith     nz  = ai[i+1] - diag[i] - 1;
1089c38d4ed2SBarry Smith     s1  = tmp[i];
1090ef66eb69SBarry Smith     s1 *= (*v++);  /* multiply by inverse of diagonal entry */
1091da3a660dSBarry Smith     while (nz--) {
1092010a20caSHong Zhang       tmp[*vi++ ] -= (*v++)*s1;
1093da3a660dSBarry Smith     }
1094c38d4ed2SBarry Smith     tmp[i] = s1;
1095da3a660dSBarry Smith   }
1096da3a660dSBarry Smith 
1097da3a660dSBarry Smith   /* backward solve the L^T */
1098da3a660dSBarry Smith   for (i=n-1; i>=0; i--){
1099010a20caSHong Zhang     v   = aa + diag[i] - 1 ;
1100010a20caSHong Zhang     vi  = aj + diag[i] - 1 ;
1101f1af5d2fSBarry Smith     nz  = diag[i] - ai[i];
1102f1af5d2fSBarry Smith     s1  = tmp[i];
1103da3a660dSBarry Smith     while (nz--) {
1104010a20caSHong Zhang       tmp[*vi-- ] -= (*v--)*s1;
1105da3a660dSBarry Smith     }
1106da3a660dSBarry Smith   }
1107da3a660dSBarry Smith 
1108da3a660dSBarry Smith   /* copy tmp into x according to permutation */
1109da3a660dSBarry Smith   for (i=0; i<n; i++) x[r[i]] = tmp[i];
1110da3a660dSBarry Smith 
11112235e667SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
11122235e667SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
11131ebc52fbSHong Zhang   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
11141ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1115da3a660dSBarry Smith 
1116d0f46423SBarry Smith   ierr = PetscLogFlops(2*a->nz-A->cmap->n);CHKERRQ(ierr);
11173a40ed3dSBarry Smith   PetscFunctionReturn(0);
1118da3a660dSBarry Smith }
1119da3a660dSBarry Smith 
11204a2ae208SSatish Balay #undef __FUNCT__
11214a2ae208SSatish Balay #define __FUNCT__ "MatSolveTransposeAdd_SeqAIJ"
1122dfbe8321SBarry Smith PetscErrorCode MatSolveTransposeAdd_SeqAIJ(Mat A,Vec bb,Vec zz,Vec xx)
1123da3a660dSBarry Smith {
1124416022c9SBarry Smith   Mat_SeqAIJ      *a = (Mat_SeqAIJ*)A->data;
11252235e667SBarry Smith   IS              iscol = a->col,isrow = a->row;
11266849ba73SBarry Smith   PetscErrorCode  ierr;
1127d0f46423SBarry Smith   PetscInt        *r,*c,i,n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
112838baddfdSBarry Smith   PetscInt        nz,*rout,*cout,*diag = a->diag;
1129dd6ea824SBarry Smith   PetscScalar     *x,*b,*tmp;
1130dd6ea824SBarry Smith   const MatScalar *aa = a->a,*v;
11316abc6512SBarry Smith 
11323a40ed3dSBarry Smith   PetscFunctionBegin;
113323bc6035SMatthew Knepley   if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);}
11346abc6512SBarry Smith 
11351ebc52fbSHong Zhang   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
11361ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1137416022c9SBarry Smith   tmp = a->solve_work;
11386abc6512SBarry Smith 
11392235e667SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
11402235e667SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
11416abc6512SBarry Smith 
11426abc6512SBarry Smith   /* copy the b into temp work space according to permutation */
11432235e667SBarry Smith   for (i=0; i<n; i++) tmp[i] = b[c[i]];
11446abc6512SBarry Smith 
11456abc6512SBarry Smith   /* forward solve the U^T */
11466abc6512SBarry Smith   for (i=0; i<n; i++) {
1147010a20caSHong Zhang     v   = aa + diag[i] ;
1148010a20caSHong Zhang     vi  = aj + diag[i] + 1;
1149f1af5d2fSBarry Smith     nz  = ai[i+1] - diag[i] - 1;
11506abc6512SBarry Smith     tmp[i] *= *v++;
11516abc6512SBarry Smith     while (nz--) {
1152010a20caSHong Zhang       tmp[*vi++ ] -= (*v++)*tmp[i];
11536abc6512SBarry Smith     }
11546abc6512SBarry Smith   }
11556abc6512SBarry Smith 
11566abc6512SBarry Smith   /* backward solve the L^T */
11576abc6512SBarry Smith   for (i=n-1; i>=0; i--){
1158010a20caSHong Zhang     v   = aa + diag[i] - 1 ;
1159010a20caSHong Zhang     vi  = aj + diag[i] - 1 ;
1160f1af5d2fSBarry Smith     nz  = diag[i] - ai[i];
11616abc6512SBarry Smith     while (nz--) {
1162010a20caSHong Zhang       tmp[*vi-- ] -= (*v--)*tmp[i];
11636abc6512SBarry Smith     }
11646abc6512SBarry Smith   }
11656abc6512SBarry Smith 
11666abc6512SBarry Smith   /* copy tmp into x according to permutation */
11676abc6512SBarry Smith   for (i=0; i<n; i++) x[r[i]] += tmp[i];
11686abc6512SBarry Smith 
11692235e667SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
11702235e667SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
11711ebc52fbSHong Zhang   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
11721ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
11736abc6512SBarry Smith 
1174efee365bSSatish Balay   ierr = PetscLogFlops(2*a->nz);CHKERRQ(ierr);
11753a40ed3dSBarry Smith   PetscFunctionReturn(0);
1176da3a660dSBarry Smith }
11779e25ed09SBarry Smith /* ----------------------------------------------------------------*/
11783c5fc038SBarry Smith EXTERN PetscErrorCode Mat_CheckInode(Mat,PetscTruth);
1179b24902e0SBarry Smith EXTERN PetscErrorCode MatDuplicateNoCreate_SeqAIJ(Mat,MatDuplicateOption,Mat*);
118008480c60SBarry Smith 
11814a2ae208SSatish Balay #undef __FUNCT__
11824a2ae208SSatish Balay #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ"
1183dfbe8321SBarry Smith PetscErrorCode MatILUFactorSymbolic_SeqAIJ(Mat A,IS isrow,IS iscol,MatFactorInfo *info,Mat *fact)
11849e25ed09SBarry Smith {
1185416022c9SBarry Smith   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
11869e25ed09SBarry Smith   IS                 isicol;
11876849ba73SBarry Smith   PetscErrorCode     ierr;
1188d0f46423SBarry Smith   PetscInt           *r,*ic,n=A->rmap->n,*ai=a->i,*aj=a->j,d;
11895a8e39fbSHong Zhang   PetscInt           *bi,*cols,nnz,*cols_lvl;
11905a8e39fbSHong Zhang   PetscInt           *bdiag,prow,fm,nzbd,len, reallocs=0,dcount=0;
11915a8e39fbSHong Zhang   PetscInt           i,levels,diagonal_fill;
119277c4ece6SBarry Smith   PetscTruth         col_identity,row_identity;
1193329f5518SBarry Smith   PetscReal          f;
11945a8e39fbSHong Zhang   PetscInt           nlnk,*lnk,*lnk_lvl=PETSC_NULL;
11955a8e39fbSHong Zhang   PetscBT            lnkbt;
11965a8e39fbSHong Zhang   PetscInt           nzi,*bj,**bj_ptr,**bjlvl_ptr;
1197a1a86e44SBarry Smith   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
1198a1a86e44SBarry Smith   PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL;
119909f38230SBarry Smith   PetscTruth         missing;
12009e25ed09SBarry Smith 
12013a40ed3dSBarry Smith   PetscFunctionBegin;
1202d0f46423SBarry Smith   if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n);
12036cf997b0SBarry Smith   f             = info->fill;
120438baddfdSBarry Smith   levels        = (PetscInt)info->levels;
120538baddfdSBarry Smith   diagonal_fill = (PetscInt)info->diagonal_fill;
12064c49b128SBarry Smith   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
120782bf6240SBarry Smith 
120808480c60SBarry Smith   /* special case that simply copies fill pattern */
1209be0abb6dSBarry Smith   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
1210be0abb6dSBarry Smith   ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr);
121186bacbd2SBarry Smith   if (!levels && row_identity && col_identity) {
1212b24902e0SBarry Smith     ierr = MatDuplicateNoCreate_SeqAIJ(A,MAT_DO_NOT_COPY_VALUES,fact);CHKERRQ(ierr);
12135c9eb25fSBarry Smith     (*fact)->factor                 = MAT_FACTOR_LU;
1214f3a39becSBarry Smith     (*fact)->info.factor_mallocs    = 0;
1215f3a39becSBarry Smith     (*fact)->info.fill_ratio_given  = info->fill;
1216f3a39becSBarry Smith     (*fact)->info.fill_ratio_needed = 1.0;
121708480c60SBarry Smith     b               = (Mat_SeqAIJ*)(*fact)->data;
12188ef3462fSBarry Smith     ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr);
121909f38230SBarry Smith     if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d);
122008480c60SBarry Smith     b->row              = isrow;
122108480c60SBarry Smith     b->col              = iscol;
122282bf6240SBarry Smith     b->icol             = isicol;
1223d0f46423SBarry Smith     ierr                = PetscMalloc(((*fact)->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
1224c38d4ed2SBarry Smith     ierr                = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
1225c38d4ed2SBarry Smith     ierr                = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
1226*db4efbfdSBarry Smith     (*fact)->ops->lufactornumeric =  MatLUFactorNumeric_SeqAIJ;
1227*db4efbfdSBarry Smith     ierr = MatILUFactorSymbolic_Inode(A,isrow,iscol,info,fact);CHKERRQ(ierr);
1228*db4efbfdSBarry Smith     (*fact)->ops->solve = MatSolve_SeqAIJ_NaturalOrdering;
12293a40ed3dSBarry Smith     PetscFunctionReturn(0);
123008480c60SBarry Smith   }
123108480c60SBarry Smith 
1232898183ecSLois Curfman McInnes   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
1233898183ecSLois Curfman McInnes   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
12349e25ed09SBarry Smith 
12359e25ed09SBarry Smith   /* get new row pointers */
12365a8e39fbSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
12375a8e39fbSHong Zhang   bi[0] = 0;
12385a8e39fbSHong Zhang   /* bdiag is location of diagonal in factor */
12395a8e39fbSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
12405a8e39fbSHong Zhang   bdiag[0]  = 0;
12416cf997b0SBarry Smith 
12425a8e39fbSHong Zhang   ierr = PetscMalloc((2*n+1)*sizeof(PetscInt**),&bj_ptr);CHKERRQ(ierr);
12435a8e39fbSHong Zhang   bjlvl_ptr = (PetscInt**)(bj_ptr + n);
12445a8e39fbSHong Zhang 
12455a8e39fbSHong Zhang   /* create a linked list for storing column indices of the active row */
12465a8e39fbSHong Zhang   nlnk = n + 1;
12475a8e39fbSHong Zhang   ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
12485a8e39fbSHong Zhang 
12495a8e39fbSHong Zhang   /* initial FreeSpace size is f*(ai[n]+1) */
1250a1a86e44SBarry Smith   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr);
12515a8e39fbSHong Zhang   current_space = free_space;
1252a1a86e44SBarry Smith   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr);
12535a8e39fbSHong Zhang   current_space_lvl = free_space_lvl;
12545a8e39fbSHong Zhang 
12555a8e39fbSHong Zhang   for (i=0; i<n; i++) {
12565a8e39fbSHong Zhang     nzi = 0;
12576cf997b0SBarry Smith     /* copy current row into linked list */
12585a8e39fbSHong Zhang     nnz  = ai[r[i]+1] - ai[r[i]];
12595a8e39fbSHong Zhang     if (!nnz) SETERRQ(PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix");
12605a8e39fbSHong Zhang     cols = aj + ai[r[i]];
12615a8e39fbSHong Zhang     lnk[i] = -1; /* marker to indicate if diagonal exists */
12625a8e39fbSHong Zhang     ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
12635a8e39fbSHong Zhang     nzi += nlnk;
12646cf997b0SBarry Smith 
12656cf997b0SBarry Smith     /* make sure diagonal entry is included */
12665a8e39fbSHong Zhang     if (diagonal_fill && lnk[i] == -1) {
12676cf997b0SBarry Smith       fm = n;
12685a8e39fbSHong Zhang       while (lnk[fm] < i) fm = lnk[fm];
12695a8e39fbSHong Zhang       lnk[i]     = lnk[fm]; /* insert diagonal into linked list */
12705a8e39fbSHong Zhang       lnk[fm]    = i;
12715a8e39fbSHong Zhang       lnk_lvl[i] = 0;
12725a8e39fbSHong Zhang       nzi++; dcount++;
12736cf997b0SBarry Smith     }
12746cf997b0SBarry Smith 
12755a8e39fbSHong Zhang     /* add pivot rows into the active row */
12765a8e39fbSHong Zhang     nzbd = 0;
12775a8e39fbSHong Zhang     prow = lnk[n];
12785a8e39fbSHong Zhang     while (prow < i) {
12795a8e39fbSHong Zhang       nnz      = bdiag[prow];
12805a8e39fbSHong Zhang       cols     = bj_ptr[prow] + nnz + 1;
12815a8e39fbSHong Zhang       cols_lvl = bjlvl_ptr[prow] + nnz + 1;
12825a8e39fbSHong Zhang       nnz      = bi[prow+1] - bi[prow] - nnz - 1;
12835a8e39fbSHong Zhang       ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr);
12845a8e39fbSHong Zhang       nzi += nlnk;
12855a8e39fbSHong Zhang       prow = lnk[prow];
12865a8e39fbSHong Zhang       nzbd++;
128756beaf04SBarry Smith     }
12885a8e39fbSHong Zhang     bdiag[i] = nzbd;
12895a8e39fbSHong Zhang     bi[i+1]  = bi[i] + nzi;
1290ecf371e4SBarry Smith 
12915a8e39fbSHong Zhang     /* if free space is not available, make more free space */
12925a8e39fbSHong Zhang     if (current_space->local_remaining<nzi) {
12935a8e39fbSHong Zhang       nnz = nzi*(n - i); /* estimated and max additional space needed */
1294a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
1295a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(nnz,&current_space_lvl);CHKERRQ(ierr);
12965a8e39fbSHong Zhang       reallocs++;
12975a8e39fbSHong Zhang     }
1298ecf371e4SBarry Smith 
12995a8e39fbSHong Zhang     /* copy data into free_space and free_space_lvl, then initialize lnk */
13005a8e39fbSHong Zhang     ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
13015a8e39fbSHong Zhang     bj_ptr[i]    = current_space->array;
13025a8e39fbSHong Zhang     bjlvl_ptr[i] = current_space_lvl->array;
13035a8e39fbSHong Zhang 
13045a8e39fbSHong Zhang     /* make sure the active row i has diagonal entry */
13055a8e39fbSHong Zhang     if (*(bj_ptr[i]+bdiag[i]) != i) {
130677431f27SBarry Smith       SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,"Row %D has missing diagonal in factored matrix\n\
1307d7ee0231SBarry Smith     try running with -pc_factor_nonzeros_along_diagonal or -pc_factor_diagonal_fill",i);
13086cf997b0SBarry Smith     }
13095a8e39fbSHong Zhang 
13105a8e39fbSHong Zhang     current_space->array           += nzi;
13115a8e39fbSHong Zhang     current_space->local_used      += nzi;
13125a8e39fbSHong Zhang     current_space->local_remaining -= nzi;
13135a8e39fbSHong Zhang     current_space_lvl->array           += nzi;
13145a8e39fbSHong Zhang     current_space_lvl->local_used      += nzi;
13155a8e39fbSHong Zhang     current_space_lvl->local_remaining -= nzi;
13169e25ed09SBarry Smith   }
13175a8e39fbSHong Zhang 
1318898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
1319898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
13205a8e39fbSHong Zhang 
13215a8e39fbSHong Zhang   /* destroy list of free space and other temporary arrays */
13225a8e39fbSHong Zhang   ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
1323a1a86e44SBarry Smith   ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr);
13245a8e39fbSHong Zhang   ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
1325a1a86e44SBarry Smith   ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
13265a8e39fbSHong Zhang   ierr = PetscFree(bj_ptr);CHKERRQ(ierr);
13279e25ed09SBarry Smith 
13286cf91177SBarry Smith #if defined(PETSC_USE_INFO)
1329f64065bbSBarry Smith   {
13305a8e39fbSHong Zhang     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
1331ae15b995SBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr);
1332ae15b995SBarry Smith     ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
1333ae15b995SBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr);
1334ae15b995SBarry Smith     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
1335335d9088SBarry Smith     if (diagonal_fill) {
1336ae15b995SBarry Smith       ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr);
1337335d9088SBarry Smith     }
1338f64065bbSBarry Smith   }
133963ba0a88SBarry Smith #endif
1340bbb6d6a8SBarry Smith 
13419e25ed09SBarry Smith   /* put together the new matrix */
134252e6d16bSBarry Smith   ierr = PetscLogObjectParent(*fact,isicol);CHKERRQ(ierr);
1343416022c9SBarry Smith   b = (Mat_SeqAIJ*)(*fact)->data;
1344e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
1345e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
13467c922b88SBarry Smith   b->singlemalloc = PETSC_FALSE;
13475a8e39fbSHong Zhang   len = (bi[n] )*sizeof(PetscScalar);
1348b0a32e0cSBarry Smith   ierr = PetscMalloc(len+1,&b->a);CHKERRQ(ierr);
13495a8e39fbSHong Zhang   b->j          = bj;
13505a8e39fbSHong Zhang   b->i          = bi;
13515a8e39fbSHong Zhang   for (i=0; i<n; i++) bdiag[i] += bi[i];
13525a8e39fbSHong Zhang   b->diag       = bdiag;
1353416022c9SBarry Smith   b->ilen       = 0;
1354416022c9SBarry Smith   b->imax       = 0;
1355416022c9SBarry Smith   b->row        = isrow;
1356416022c9SBarry Smith   b->col        = iscol;
1357c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
1358c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
135982bf6240SBarry Smith   b->icol       = isicol;
136087828ca2SBarry Smith   ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
1361416022c9SBarry Smith   /* In b structure:  Free imax, ilen, old a, old j.
13625a8e39fbSHong Zhang      Allocate bdiag, solve_work, new a, new j */
13635a8e39fbSHong Zhang   ierr = PetscLogObjectMemory(*fact,(bi[n]-n) * (sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
13645a8e39fbSHong Zhang   b->maxnz             = b->nz = bi[n] ;
13655c9eb25fSBarry Smith   (*fact)->factor = MAT_FACTOR_LU;
1366418422e8SSatish Balay   (*fact)->info.factor_mallocs    = reallocs;
1367ae068f58SLois Curfman McInnes   (*fact)->info.fill_ratio_given  = f;
13685a8e39fbSHong Zhang   (*fact)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]);
1369*db4efbfdSBarry Smith   (*fact)->ops->lufactornumeric =  MatLUFactorNumeric_SeqAIJ;
137054e71613SHong Zhang   ierr = MatILUFactorSymbolic_Inode(A,isrow,iscol,info,fact);CHKERRQ(ierr);
13713a40ed3dSBarry Smith   PetscFunctionReturn(0);
13729e25ed09SBarry Smith }
1373d5d45c9bSBarry Smith 
13743c9e8598SHong Zhang #include "src/mat/impls/sbaij/seq/sbaij.h"
1375a6175056SHong Zhang #undef __FUNCT__
1376a6175056SHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ"
1377af281ebdSHong Zhang PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ(Mat A,MatFactorInfo *info,Mat *B)
1378a6175056SHong Zhang {
13792ed133dbSHong Zhang   Mat            C = *B;
13800968510aSHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
13812ed133dbSHong Zhang   Mat_SeqSBAIJ   *b=(Mat_SeqSBAIJ*)C->data;
13829bfd6278SHong Zhang   IS             ip=b->row,iip = b->icol;
13832ed133dbSHong Zhang   PetscErrorCode ierr;
1384d0f46423SBarry Smith   PetscInt       *rip,*riip,i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bcol;
13852ed133dbSHong Zhang   PetscInt       *ai=a->i,*aj=a->j;
13862ed133dbSHong Zhang   PetscInt       k,jmin,jmax,*jl,*il,col,nexti,ili,nz;
1387622e2028SHong Zhang   MatScalar      *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi;
1388540703acSHong Zhang   PetscReal      zeropivot,rs,shiftnz;
1389fbf22428SSatish Balay   PetscReal      shiftpd;
1390540703acSHong Zhang   ChShift_Ctx    sctx;
1391540703acSHong Zhang   PetscInt       newshift;
1392*db4efbfdSBarry Smith   PetscTruth     perm_identity;
1393d5d45c9bSBarry Smith 
1394a6175056SHong Zhang   PetscFunctionBegin;
1395117f1891Stmunson 
1396540703acSHong Zhang   shiftnz   = info->shiftnz;
1397540703acSHong Zhang   shiftpd   = info->shiftpd;
1398ee45ca4aSHong Zhang   zeropivot = info->zeropivot;
1399ee45ca4aSHong Zhang 
14002ed133dbSHong Zhang   ierr  = ISGetIndices(ip,&rip);CHKERRQ(ierr);
14019bfd6278SHong Zhang   ierr  = ISGetIndices(iip,&riip);CHKERRQ(ierr);
14022ed133dbSHong Zhang 
14032ed133dbSHong Zhang   /* initialization */
14042ed133dbSHong Zhang   nz   = (2*mbs+1)*sizeof(PetscInt)+mbs*sizeof(MatScalar);
14052ed133dbSHong Zhang   ierr = PetscMalloc(nz,&il);CHKERRQ(ierr);
14062ed133dbSHong Zhang   jl   = il + mbs;
14072ed133dbSHong Zhang   rtmp = (MatScalar*)(jl + mbs);
14082ed133dbSHong Zhang 
1409540703acSHong Zhang   sctx.shift_amount = 0;
1410540703acSHong Zhang   sctx.nshift       = 0;
14112ed133dbSHong Zhang   do {
1412540703acSHong Zhang     sctx.chshift = PETSC_FALSE;
14132ed133dbSHong Zhang     for (i=0; i<mbs; i++) {
14142ed133dbSHong Zhang       rtmp[i] = 0.0; jl[i] = mbs; il[0] = 0;
1415a6175056SHong Zhang     }
14162ed133dbSHong Zhang 
14172ed133dbSHong Zhang     for (k = 0; k<mbs; k++){
1418c05c3958SHong Zhang       bval = ba + bi[k];
14192ed133dbSHong Zhang       /* initialize k-th row by the perm[k]-th row of A */
14202ed133dbSHong Zhang       jmin = ai[rip[k]]; jmax = ai[rip[k]+1];
14212ed133dbSHong Zhang       for (j = jmin; j < jmax; j++){
14229bfd6278SHong Zhang         col = riip[aj[j]];
14232ed133dbSHong Zhang         if (col >= k){ /* only take upper triangular entry */
14242ed133dbSHong Zhang           rtmp[col] = aa[j];
1425c05c3958SHong Zhang           *bval++  = 0.0; /* for in-place factorization */
14262ed133dbSHong Zhang         }
14272ed133dbSHong Zhang       }
142839e3d630SHong Zhang       /* shift the diagonal of the matrix */
1429540703acSHong Zhang       if (sctx.nshift) rtmp[k] += sctx.shift_amount;
14302ed133dbSHong Zhang 
14312ed133dbSHong Zhang       /* modify k-th row by adding in those rows i with U(i,k)!=0 */
14322ed133dbSHong Zhang       dk = rtmp[k];
14332ed133dbSHong Zhang       i = jl[k]; /* first row to be added to k_th row  */
14342ed133dbSHong Zhang 
14352ed133dbSHong Zhang       while (i < k){
14362ed133dbSHong Zhang         nexti = jl[i]; /* next row to be added to k_th row */
14372ed133dbSHong Zhang 
14382ed133dbSHong Zhang         /* compute multiplier, update diag(k) and U(i,k) */
14392ed133dbSHong Zhang         ili = il[i];  /* index of first nonzero element in U(i,k:bms-1) */
14402ed133dbSHong Zhang         uikdi = - ba[ili]*ba[bi[i]];  /* diagonal(k) */
14412ed133dbSHong Zhang         dk += uikdi*ba[ili];
14422ed133dbSHong Zhang         ba[ili] = uikdi; /* -U(i,k) */
14432ed133dbSHong Zhang 
14442ed133dbSHong Zhang         /* add multiple of row i to k-th row */
14452ed133dbSHong Zhang         jmin = ili + 1; jmax = bi[i+1];
14462ed133dbSHong Zhang         if (jmin < jmax){
14472ed133dbSHong Zhang           for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j];
14482ed133dbSHong Zhang           /* update il and jl for row i */
14492ed133dbSHong Zhang           il[i] = jmin;
14502ed133dbSHong Zhang           j = bj[jmin]; jl[i] = jl[j]; jl[j] = i;
14512ed133dbSHong Zhang         }
14522ed133dbSHong Zhang         i = nexti;
14532ed133dbSHong Zhang       }
14542ed133dbSHong Zhang 
14550697b6caSHong Zhang       /* shift the diagonals when zero pivot is detected */
14560697b6caSHong Zhang       /* compute rs=sum of abs(off-diagonal) */
14570697b6caSHong Zhang       rs   = 0.0;
14583c9e8598SHong Zhang       jmin = bi[k]+1;
14593c9e8598SHong Zhang       nz   = bi[k+1] - jmin;
14603c9e8598SHong Zhang       bcol = bj + jmin;
14613c9e8598SHong Zhang       while (nz--){
146289f845c8SHong Zhang         rs += PetscAbsScalar(rtmp[*bcol]);
146389f845c8SHong Zhang         bcol++;
14643c9e8598SHong Zhang       }
1465540703acSHong Zhang 
1466540703acSHong Zhang       sctx.rs = rs;
1467540703acSHong Zhang       sctx.pv = dk;
14685b5bc046SBarry Smith       ierr = MatCholeskyCheckShift_inline(info,sctx,k,newshift);CHKERRQ(ierr);
1469117f1891Stmunson 
1470117f1891Stmunson       if (newshift == 1) {
1471117f1891Stmunson         if (!sctx.shift_amount) {
1472117f1891Stmunson           sctx.shift_amount = 1e-5;
1473117f1891Stmunson         }
1474117f1891Stmunson         break;
1475117f1891Stmunson       }
14763c9e8598SHong Zhang 
14773c9e8598SHong Zhang       /* copy data into U(k,:) */
147839e3d630SHong Zhang       ba[bi[k]] = 1.0/dk; /* U(k,k) */
147939e3d630SHong Zhang       jmin = bi[k]+1; jmax = bi[k+1];
148039e3d630SHong Zhang       if (jmin < jmax) {
148139e3d630SHong Zhang         for (j=jmin; j<jmax; j++){
148239e3d630SHong Zhang           col = bj[j]; ba[j] = rtmp[col]; rtmp[col] = 0.0;
14833c9e8598SHong Zhang         }
148439e3d630SHong Zhang         /* add the k-th row into il and jl */
14853c9e8598SHong Zhang         il[k] = jmin;
14863c9e8598SHong Zhang         i = bj[jmin]; jl[k] = jl[i]; jl[i] = k;
14873c9e8598SHong Zhang       }
14883c9e8598SHong Zhang     }
1489540703acSHong Zhang   } while (sctx.chshift);
14903c9e8598SHong Zhang   ierr = PetscFree(il);CHKERRQ(ierr);
14913c9e8598SHong Zhang 
149239e3d630SHong Zhang   ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr);
14939bfd6278SHong Zhang   ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr);
1494*db4efbfdSBarry Smith 
1495*db4efbfdSBarry Smith   ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr);
1496*db4efbfdSBarry Smith   if (perm_identity){
1497*db4efbfdSBarry Smith     (*B)->ops->solve           = MatSolve_SeqSBAIJ_1_NaturalOrdering;
1498*db4efbfdSBarry Smith     (*B)->ops->solvetranspose  = MatSolve_SeqSBAIJ_1_NaturalOrdering;
1499*db4efbfdSBarry Smith     (*B)->ops->forwardsolve    = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering;
1500*db4efbfdSBarry Smith     (*B)->ops->backwardsolve   = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering;
1501*db4efbfdSBarry Smith   } else {
1502*db4efbfdSBarry Smith     (*B)->ops->solve           = MatSolve_SeqSBAIJ_1;
1503*db4efbfdSBarry Smith     (*B)->ops->solvetranspose  = MatSolve_SeqSBAIJ_1;
1504*db4efbfdSBarry Smith     (*B)->ops->forwardsolve    = MatForwardSolve_SeqSBAIJ_1;
1505*db4efbfdSBarry Smith     (*B)->ops->backwardsolve   = MatBackwardSolve_SeqSBAIJ_1;
1506*db4efbfdSBarry Smith   }
1507*db4efbfdSBarry Smith 
15083c9e8598SHong Zhang   C->assembled    = PETSC_TRUE;
15093c9e8598SHong Zhang   C->preallocated = PETSC_TRUE;
1510d0f46423SBarry Smith   ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr);
1511540703acSHong Zhang   if (sctx.nshift){
1512540703acSHong Zhang     if (shiftnz) {
15131e2582c4SBarry Smith       ierr = PetscInfo2(A,"number of shiftnz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
1514540703acSHong Zhang     } else if (shiftpd) {
15151e2582c4SBarry Smith       ierr = PetscInfo2(A,"number of shiftpd tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
15160697b6caSHong Zhang     }
15173c9e8598SHong Zhang   }
15183c9e8598SHong Zhang   PetscFunctionReturn(0);
15193c9e8598SHong Zhang }
1520a6175056SHong Zhang 
1521a6175056SHong Zhang #undef __FUNCT__
1522a6175056SHong Zhang #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ"
1523dfbe8321SBarry Smith PetscErrorCode MatICCFactorSymbolic_SeqAIJ(Mat A,IS perm,MatFactorInfo *info,Mat *fact)
1524a6175056SHong Zhang {
15250968510aSHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
1526ed59401aSHong Zhang   Mat_SeqSBAIJ       *b;
1527dfbe8321SBarry Smith   PetscErrorCode     ierr;
152858ebbce7SBarry Smith   PetscTruth         perm_identity,missing;
1529d0f46423SBarry Smith   PetscInt           reallocs=0,*rip,*riip,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui;
1530ed59401aSHong Zhang   PetscInt           jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow;
153158ebbce7SBarry Smith   PetscInt           nlnk,*lnk,*lnk_lvl=PETSC_NULL,d;
15325a8e39fbSHong Zhang   PetscInt           ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr;
1533ed59401aSHong Zhang   PetscReal          fill=info->fill,levels=info->levels;
1534a1a86e44SBarry Smith   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
1535a1a86e44SBarry Smith   PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL;
15367a48dd6fSHong Zhang   PetscBT            lnkbt;
1537b635d36bSHong Zhang   IS                 iperm;
1538a6175056SHong Zhang 
1539a6175056SHong Zhang   PetscFunctionBegin;
1540d0f46423SBarry Smith   if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n);
154158ebbce7SBarry Smith   ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr);
154258ebbce7SBarry Smith   if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d);
1543653a6975SHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
1544b635d36bSHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
15457a48dd6fSHong Zhang 
154639e3d630SHong Zhang   ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr);
154739e3d630SHong Zhang   ui[0] = 0;
154839e3d630SHong Zhang 
1549b635d36bSHong Zhang   /* ICC(0) without matrix ordering: simply copies fill pattern */
1550622e2028SHong Zhang   if (!levels && perm_identity) {
155158ebbce7SBarry Smith 
1552ed59401aSHong Zhang     for (i=0; i<am; i++) {
155339e3d630SHong Zhang       ui[i+1] = ui[i] + ai[i+1] - a->diag[i];
1554ed59401aSHong Zhang     }
155539e3d630SHong Zhang     ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
155639e3d630SHong Zhang     cols = uj;
1557ed59401aSHong Zhang     for (i=0; i<am; i++) {
1558ed59401aSHong Zhang       aj    = a->j + a->diag[i];
155939e3d630SHong Zhang       ncols = ui[i+1] - ui[i];
156039e3d630SHong Zhang       for (j=0; j<ncols; j++) *cols++ = *aj++;
1561ed59401aSHong Zhang     }
156239e3d630SHong Zhang   } else { /* case: levels>0 || (levels=0 && !perm_identity) */
1563b635d36bSHong Zhang     ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
1564b635d36bSHong Zhang     ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
1565b635d36bSHong Zhang 
15667a48dd6fSHong Zhang     /* initialization */
15675a8e39fbSHong Zhang     ierr  = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr);
15687a48dd6fSHong Zhang 
15697a48dd6fSHong Zhang     /* jl: linked list for storing indices of the pivot rows
15707a48dd6fSHong Zhang        il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
15711393bc97SHong Zhang     ierr = PetscMalloc((2*am+1)*sizeof(PetscInt)+2*am*sizeof(PetscInt**),&jl);CHKERRQ(ierr);
15727a48dd6fSHong Zhang     il         = jl + am;
15737a48dd6fSHong Zhang     uj_ptr     = (PetscInt**)(il + am);
15747a48dd6fSHong Zhang     uj_lvl_ptr = (PetscInt**)(uj_ptr + am);
15757a48dd6fSHong Zhang     for (i=0; i<am; i++){
15767a48dd6fSHong Zhang       jl[i] = am; il[i] = 0;
15777a48dd6fSHong Zhang     }
15787a48dd6fSHong Zhang 
15797a48dd6fSHong Zhang     /* create and initialize a linked list for storing column indices of the active row k */
15807a48dd6fSHong Zhang     nlnk = am + 1;
15812ed133dbSHong Zhang     ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
15827a48dd6fSHong Zhang 
15837a48dd6fSHong Zhang     /* initial FreeSpace size is fill*(ai[am]+1) */
1584a1a86e44SBarry Smith     ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr);
15857a48dd6fSHong Zhang     current_space = free_space;
1586a1a86e44SBarry Smith     ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space_lvl);CHKERRQ(ierr);
15877a48dd6fSHong Zhang     current_space_lvl = free_space_lvl;
15887a48dd6fSHong Zhang 
15897a48dd6fSHong Zhang     for (k=0; k<am; k++){  /* for each active row k */
15907a48dd6fSHong Zhang       /* initialize lnk by the column indices of row rip[k] of A */
15917a48dd6fSHong Zhang       nzk   = 0;
1592622e2028SHong Zhang       ncols = ai[rip[k]+1] - ai[rip[k]];
1593b635d36bSHong Zhang       if (!ncols) SETERRQ(PETSC_ERR_MAT_CH_ZRPVT,"Empty row in matrix");
1594622e2028SHong Zhang       ncols_upper = 0;
1595622e2028SHong Zhang       for (j=0; j<ncols; j++){
1596b635d36bSHong Zhang         i = *(aj + ai[rip[k]] + j); /* unpermuted column index */
1597b635d36bSHong Zhang         if (riip[i] >= k){ /* only take upper triangular entry */
15985a8e39fbSHong Zhang           ajtmp[ncols_upper] = i;
1599622e2028SHong Zhang           ncols_upper++;
1600622e2028SHong Zhang         }
1601622e2028SHong Zhang       }
1602b635d36bSHong Zhang       ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
16037a48dd6fSHong Zhang       nzk += nlnk;
16047a48dd6fSHong Zhang 
16057a48dd6fSHong Zhang       /* update lnk by computing fill-in for each pivot row to be merged in */
16067a48dd6fSHong Zhang       prow = jl[k]; /* 1st pivot row */
16077a48dd6fSHong Zhang 
16087a48dd6fSHong Zhang       while (prow < k){
16097a48dd6fSHong Zhang         nextprow = jl[prow];
16107a48dd6fSHong Zhang 
16117a48dd6fSHong Zhang         /* merge prow into k-th row */
16127a48dd6fSHong Zhang         jmin = il[prow] + 1;  /* index of the 2nd nzero entry in U(prow,k:am-1) */
16137a48dd6fSHong Zhang         jmax = ui[prow+1];
16147a48dd6fSHong Zhang         ncols = jmax-jmin;
1615ed59401aSHong Zhang         i     = jmin - ui[prow];
1616ed59401aSHong Zhang         cols  = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */
16175a8e39fbSHong Zhang         uj    = uj_lvl_ptr[prow] + i; /* levels of cols */
16185a8e39fbSHong Zhang         j     = *(uj - 1);
16195a8e39fbSHong Zhang         ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr);
16207a48dd6fSHong Zhang         nzk += nlnk;
16217a48dd6fSHong Zhang 
16227a48dd6fSHong Zhang         /* update il and jl for prow */
16237a48dd6fSHong Zhang         if (jmin < jmax){
16247a48dd6fSHong Zhang           il[prow] = jmin;
16257a48dd6fSHong Zhang           j = *cols; jl[prow] = jl[j]; jl[j] = prow;
16267a48dd6fSHong Zhang         }
16277a48dd6fSHong Zhang         prow = nextprow;
16287a48dd6fSHong Zhang       }
16297a48dd6fSHong Zhang 
16307a48dd6fSHong Zhang       /* if free space is not available, make more free space */
16317a48dd6fSHong Zhang       if (current_space->local_remaining<nzk) {
16327a48dd6fSHong Zhang         i = am - k + 1; /* num of unfactored rows */
16337a48dd6fSHong Zhang         i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */
1634a1a86e44SBarry Smith         ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
1635a1a86e44SBarry Smith         ierr = PetscFreeSpaceGet(i,&current_space_lvl);CHKERRQ(ierr);
16367a48dd6fSHong Zhang         reallocs++;
16377a48dd6fSHong Zhang       }
16387a48dd6fSHong Zhang 
16397a48dd6fSHong Zhang       /* copy data into free_space and free_space_lvl, then initialize lnk */
1640b635d36bSHong Zhang       if (nzk == 0) SETERRQ1(PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k);
16412ed133dbSHong Zhang       ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
16427a48dd6fSHong Zhang 
16437a48dd6fSHong Zhang       /* add the k-th row into il and jl */
164439e3d630SHong Zhang       if (nzk > 1){
16457a48dd6fSHong Zhang         i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
16467a48dd6fSHong Zhang         jl[k] = jl[i]; jl[i] = k;
16477a48dd6fSHong Zhang         il[k] = ui[k] + 1;
16487a48dd6fSHong Zhang       }
16497a48dd6fSHong Zhang       uj_ptr[k]     = current_space->array;
16507a48dd6fSHong Zhang       uj_lvl_ptr[k] = current_space_lvl->array;
16517a48dd6fSHong Zhang 
16527a48dd6fSHong Zhang       current_space->array           += nzk;
16537a48dd6fSHong Zhang       current_space->local_used      += nzk;
16547a48dd6fSHong Zhang       current_space->local_remaining -= nzk;
16557a48dd6fSHong Zhang 
16567a48dd6fSHong Zhang       current_space_lvl->array           += nzk;
16577a48dd6fSHong Zhang       current_space_lvl->local_used      += nzk;
16587a48dd6fSHong Zhang       current_space_lvl->local_remaining -= nzk;
16597a48dd6fSHong Zhang 
16607a48dd6fSHong Zhang       ui[k+1] = ui[k] + nzk;
16617a48dd6fSHong Zhang     }
16627a48dd6fSHong Zhang 
16636cf91177SBarry Smith #if defined(PETSC_USE_INFO)
16647a48dd6fSHong Zhang     if (ai[am] != 0) {
166539e3d630SHong Zhang       PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]);
1666ae15b995SBarry Smith       ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr);
1667ae15b995SBarry Smith       ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
1668ae15b995SBarry Smith       ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr);
16697a48dd6fSHong Zhang     } else {
1670ae15b995SBarry Smith       ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr);
16717a48dd6fSHong Zhang     }
167263ba0a88SBarry Smith #endif
16737a48dd6fSHong Zhang 
16747a48dd6fSHong Zhang     ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
1675b635d36bSHong Zhang     ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
16767a48dd6fSHong Zhang     ierr = PetscFree(jl);CHKERRQ(ierr);
16775a8e39fbSHong Zhang     ierr = PetscFree(ajtmp);CHKERRQ(ierr);
16787a48dd6fSHong Zhang 
16797a48dd6fSHong Zhang     /* destroy list of free space and other temporary array(s) */
16807a48dd6fSHong Zhang     ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
1681a1a86e44SBarry Smith     ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr);
16822ed133dbSHong Zhang     ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
1683a1a86e44SBarry Smith     ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
16847a48dd6fSHong Zhang 
168539e3d630SHong Zhang   } /* end of case: levels>0 || (levels=0 && !perm_identity) */
168639e3d630SHong Zhang 
16877a48dd6fSHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
16887a48dd6fSHong Zhang 
1689b24902e0SBarry Smith   b    = (Mat_SeqSBAIJ*)(*fact)->data;
16907a48dd6fSHong Zhang   b->singlemalloc = PETSC_FALSE;
16917a48dd6fSHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr);
16927a48dd6fSHong Zhang   b->j    = uj;
16937a48dd6fSHong Zhang   b->i    = ui;
16947a48dd6fSHong Zhang   b->diag = 0;
16957a48dd6fSHong Zhang   b->ilen = 0;
16967a48dd6fSHong Zhang   b->imax = 0;
16977a48dd6fSHong Zhang   b->row  = perm;
1698b635d36bSHong Zhang   b->col  = perm;
1699b635d36bSHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
1700b635d36bSHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
1701b635d36bSHong Zhang   b->icol = iperm;
17027a48dd6fSHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
17037a48dd6fSHong Zhang   ierr    = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
1704b24902e0SBarry Smith   ierr = PetscLogObjectMemory((*fact),(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
17057a48dd6fSHong Zhang   b->maxnz   = b->nz = ui[am];
1706e6b907acSBarry Smith   b->free_a  = PETSC_TRUE;
1707e6b907acSBarry Smith   b->free_ij = PETSC_TRUE;
17087a48dd6fSHong Zhang 
1709b24902e0SBarry Smith   (*fact)->info.factor_mallocs    = reallocs;
1710b24902e0SBarry Smith   (*fact)->info.fill_ratio_given  = fill;
17117a48dd6fSHong Zhang   if (ai[am] != 0) {
1712b24902e0SBarry Smith     (*fact)->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]);
17137a48dd6fSHong Zhang   } else {
1714b24902e0SBarry Smith     (*fact)->info.fill_ratio_needed = 0.0;
17157a48dd6fSHong Zhang   }
171639e3d630SHong Zhang   (*fact)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ;
1717a6175056SHong Zhang   PetscFunctionReturn(0);
1718a6175056SHong Zhang }
1719d5d45c9bSBarry Smith 
1720f76d2b81SHong Zhang #undef __FUNCT__
1721f76d2b81SHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_SeqAIJ"
1722dfbe8321SBarry Smith PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ(Mat A,IS perm,MatFactorInfo *info,Mat *fact)
1723f76d2b81SHong Zhang {
1724f76d2b81SHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
172510c27e3dSHong Zhang   Mat_SeqSBAIJ       *b;
1726dfbe8321SBarry Smith   PetscErrorCode     ierr;
1727f76d2b81SHong Zhang   PetscTruth         perm_identity;
172810c27e3dSHong Zhang   PetscReal          fill = info->fill;
1729d0f46423SBarry Smith   PetscInt           *rip,*riip,i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow;
173010c27e3dSHong Zhang   PetscInt           *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow;
17312ed133dbSHong Zhang   PetscInt           nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr;
1732a1a86e44SBarry Smith   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
173310c27e3dSHong Zhang   PetscBT            lnkbt;
17342ed133dbSHong Zhang   IS                 iperm;
1735f76d2b81SHong Zhang 
1736f76d2b81SHong Zhang   PetscFunctionBegin;
1737d0f46423SBarry Smith   if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n);
17382ed133dbSHong Zhang   /* check whether perm is the identity mapping */
1739f76d2b81SHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
17402ed133dbSHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
17412ed133dbSHong Zhang   ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
17429bfd6278SHong Zhang   ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
174310c27e3dSHong Zhang 
174410c27e3dSHong Zhang   /* initialization */
17451393bc97SHong Zhang   ierr  = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr);
174610c27e3dSHong Zhang   ui[0] = 0;
174710c27e3dSHong Zhang 
174810c27e3dSHong Zhang   /* jl: linked list for storing indices of the pivot rows
17491393bc97SHong Zhang      il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
17501393bc97SHong Zhang   ierr = PetscMalloc((3*am+1)*sizeof(PetscInt)+am*sizeof(PetscInt**),&jl);CHKERRQ(ierr);
17511393bc97SHong Zhang   il     = jl + am;
17521393bc97SHong Zhang   cols   = il + am;
17531393bc97SHong Zhang   ui_ptr = (PetscInt**)(cols + am);
17541393bc97SHong Zhang   for (i=0; i<am; i++){
17551393bc97SHong Zhang     jl[i] = am; il[i] = 0;
1756f76d2b81SHong Zhang   }
175710c27e3dSHong Zhang 
175810c27e3dSHong Zhang   /* create and initialize a linked list for storing column indices of the active row k */
17591393bc97SHong Zhang   nlnk = am + 1;
17601393bc97SHong Zhang   ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
176110c27e3dSHong Zhang 
17621393bc97SHong Zhang   /* initial FreeSpace size is fill*(ai[am]+1) */
1763a1a86e44SBarry Smith   ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr);
176410c27e3dSHong Zhang   current_space = free_space;
176510c27e3dSHong Zhang 
17661393bc97SHong Zhang   for (k=0; k<am; k++){  /* for each active row k */
176710c27e3dSHong Zhang     /* initialize lnk by the column indices of row rip[k] of A */
176810c27e3dSHong Zhang     nzk   = 0;
17692ed133dbSHong Zhang     ncols = ai[rip[k]+1] - ai[rip[k]];
17709bfd6278SHong Zhang     if (!ncols) SETERRQ(PETSC_ERR_MAT_CH_ZRPVT,"Empty row in matrix");
17712ed133dbSHong Zhang     ncols_upper = 0;
17722ed133dbSHong Zhang     for (j=0; j<ncols; j++){
17739bfd6278SHong Zhang       i = riip[*(aj + ai[rip[k]] + j)];
17742ed133dbSHong Zhang       if (i >= k){ /* only take upper triangular entry */
17752ed133dbSHong Zhang         cols[ncols_upper] = i;
17762ed133dbSHong Zhang         ncols_upper++;
17772ed133dbSHong Zhang       }
17782ed133dbSHong Zhang     }
17791393bc97SHong Zhang     ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
178010c27e3dSHong Zhang     nzk += nlnk;
178110c27e3dSHong Zhang 
178210c27e3dSHong Zhang     /* update lnk by computing fill-in for each pivot row to be merged in */
178310c27e3dSHong Zhang     prow = jl[k]; /* 1st pivot row */
178410c27e3dSHong Zhang 
178510c27e3dSHong Zhang     while (prow < k){
178610c27e3dSHong Zhang       nextprow = jl[prow];
178710c27e3dSHong Zhang       /* merge prow into k-th row */
17881393bc97SHong Zhang       jmin = il[prow] + 1;  /* index of the 2nd nzero entry in U(prow,k:am-1) */
178910c27e3dSHong Zhang       jmax = ui[prow+1];
179010c27e3dSHong Zhang       ncols = jmax-jmin;
17911393bc97SHong Zhang       uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */
17925a8e39fbSHong Zhang       ierr = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
179310c27e3dSHong Zhang       nzk += nlnk;
179410c27e3dSHong Zhang 
179510c27e3dSHong Zhang       /* update il and jl for prow */
179610c27e3dSHong Zhang       if (jmin < jmax){
179710c27e3dSHong Zhang         il[prow] = jmin;
17982ed133dbSHong Zhang         j = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow;
179910c27e3dSHong Zhang       }
180010c27e3dSHong Zhang       prow = nextprow;
180110c27e3dSHong Zhang     }
180210c27e3dSHong Zhang 
180310c27e3dSHong Zhang     /* if free space is not available, make more free space */
180410c27e3dSHong Zhang     if (current_space->local_remaining<nzk) {
18051393bc97SHong Zhang       i = am - k + 1; /* num of unfactored rows */
180610c27e3dSHong Zhang       i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */
1807a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
180810c27e3dSHong Zhang       reallocs++;
180910c27e3dSHong Zhang     }
181010c27e3dSHong Zhang 
181110c27e3dSHong Zhang     /* copy data into free space, then initialize lnk */
18121393bc97SHong Zhang     ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
181310c27e3dSHong Zhang 
181410c27e3dSHong Zhang     /* add the k-th row into il and jl */
181510c27e3dSHong Zhang     if (nzk-1 > 0){
18161393bc97SHong Zhang       i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
181710c27e3dSHong Zhang       jl[k] = jl[i]; jl[i] = k;
181810c27e3dSHong Zhang       il[k] = ui[k] + 1;
181910c27e3dSHong Zhang     }
18202ed133dbSHong Zhang     ui_ptr[k] = current_space->array;
182110c27e3dSHong Zhang     current_space->array           += nzk;
182210c27e3dSHong Zhang     current_space->local_used      += nzk;
182310c27e3dSHong Zhang     current_space->local_remaining -= nzk;
182410c27e3dSHong Zhang 
182510c27e3dSHong Zhang     ui[k+1] = ui[k] + nzk;
182610c27e3dSHong Zhang   }
182710c27e3dSHong Zhang 
18286cf91177SBarry Smith #if defined(PETSC_USE_INFO)
18291393bc97SHong Zhang   if (ai[am] != 0) {
18301393bc97SHong Zhang     PetscReal af = (PetscReal)(ui[am])/((PetscReal)ai[am]);
1831ae15b995SBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr);
1832ae15b995SBarry Smith     ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
1833ae15b995SBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr);
183410c27e3dSHong Zhang   } else {
1835ae15b995SBarry Smith      ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr);
183610c27e3dSHong Zhang   }
183763ba0a88SBarry Smith #endif
183810c27e3dSHong Zhang 
183910c27e3dSHong Zhang   ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
18409bfd6278SHong Zhang   ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
184110c27e3dSHong Zhang   ierr = PetscFree(jl);CHKERRQ(ierr);
184210c27e3dSHong Zhang 
184310c27e3dSHong Zhang   /* destroy list of free space and other temporary array(s) */
18441393bc97SHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
1845a1a86e44SBarry Smith   ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr);
184610c27e3dSHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
184710c27e3dSHong Zhang 
184810c27e3dSHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
184910c27e3dSHong Zhang 
1850b24902e0SBarry Smith   b = (Mat_SeqSBAIJ*)(*fact)->data;
185110c27e3dSHong Zhang   b->singlemalloc = PETSC_FALSE;
1852e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
1853e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
18541393bc97SHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr);
185510c27e3dSHong Zhang   b->j    = uj;
185610c27e3dSHong Zhang   b->i    = ui;
185710c27e3dSHong Zhang   b->diag = 0;
185810c27e3dSHong Zhang   b->ilen = 0;
185910c27e3dSHong Zhang   b->imax = 0;
186010c27e3dSHong Zhang   b->row  = perm;
18619bfd6278SHong Zhang   b->col  = perm;
18629bfd6278SHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
18639bfd6278SHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
18649bfd6278SHong Zhang   b->icol = iperm;
186510c27e3dSHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
18661393bc97SHong Zhang   ierr    = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
1867b24902e0SBarry Smith   ierr    = PetscLogObjectMemory(*fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
18681393bc97SHong Zhang   b->maxnz = b->nz = ui[am];
186910c27e3dSHong Zhang 
1870b24902e0SBarry Smith   (*fact)->info.factor_mallocs    = reallocs;
1871b24902e0SBarry Smith   (*fact)->info.fill_ratio_given  = fill;
18721393bc97SHong Zhang   if (ai[am] != 0) {
1873b24902e0SBarry Smith     (*fact)->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]);
187410c27e3dSHong Zhang   } else {
1875b24902e0SBarry Smith     (*fact)->info.fill_ratio_needed = 0.0;
187610c27e3dSHong Zhang   }
187739e3d630SHong Zhang   (*fact)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ;
1878f76d2b81SHong Zhang   PetscFunctionReturn(0);
1879f76d2b81SHong Zhang }
1880