xref: /petsc/src/mat/impls/aij/seq/aijfact.c (revision 0e95ead3fae88491e7be508fb32a9a35027f9909)
1be1d678aSKris Buschelman #define PETSCMAT_DLL
2289bc588SBarry Smith 
3b377110cSBarry Smith 
47c4f633dSBarry Smith #include "../src/mat/impls/aij/seq/aij.h"
5783ef271SHong Zhang #include "../src/mat/impls/sbaij/seq/sbaij.h"
61393bc97SHong Zhang #include "petscbt.h"
77c4f633dSBarry Smith #include "../src/mat/utils/freespace.h"
83b2fbd54SBarry Smith 
9f6a26d55SMatthew Knepley EXTERN_C_BEGIN
104a2ae208SSatish Balay #undef __FUNCT__
114a2ae208SSatish Balay #define __FUNCT__ "MatOrdering_Flow_SeqAIJ"
128fa24674SBarry Smith /*
138fa24674SBarry Smith       Computes an ordering to get most of the large numerical values in the lower triangular part of the matrix
148fa24674SBarry Smith */
15dfbe8321SBarry Smith PetscErrorCode MatOrdering_Flow_SeqAIJ(Mat mat,const MatOrderingType type,IS *irow,IS *icol)
163b2fbd54SBarry Smith {
178fa24674SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)mat->data;
188fa24674SBarry Smith   PetscErrorCode    ierr;
19889a8dedSBarry Smith   PetscInt          i,j,jj,k, kk,n = mat->rmap->n, current = 0, newcurrent = 0,*order;
208fa24674SBarry Smith   const PetscInt    *ai = a->i, *aj = a->j;
218fa24674SBarry Smith   const PetscScalar *aa = a->a;
228fa24674SBarry Smith   PetscTruth        *done;
23889a8dedSBarry Smith   PetscReal         best,past = 0,future;
243a40ed3dSBarry Smith 
258fa24674SBarry Smith   PetscFunctionBegin;
268fa24674SBarry Smith   /* pick initial row */
278fa24674SBarry Smith   best = -1;
288fa24674SBarry Smith   for (i=0; i<n; i++) {
2975567043SBarry Smith     future = 0.0;
308fa24674SBarry Smith     for (j=ai[i]; j<ai[i+1]; j++) {
318fa24674SBarry Smith       if (aj[j] != i) future  += PetscAbsScalar(aa[j]); else past = PetscAbsScalar(aa[j]);
328fa24674SBarry Smith     }
338fa24674SBarry Smith     if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */
348fa24674SBarry Smith     if (past/future > best) {
358fa24674SBarry Smith       best = past/future;
368fa24674SBarry Smith       current = i;
378fa24674SBarry Smith     }
388fa24674SBarry Smith   }
398fa24674SBarry Smith 
408fa24674SBarry Smith   ierr = PetscMalloc(n*sizeof(PetscTruth),&done);CHKERRQ(ierr);
418fa24674SBarry Smith   ierr = PetscMemzero(done,n*sizeof(PetscTruth));CHKERRQ(ierr);
420e83c824SBarry Smith   ierr = PetscMalloc(n*sizeof(PetscInt),&order);CHKERRQ(ierr);
438fa24674SBarry Smith   order[0] = current;
448fa24674SBarry Smith   for (i=0; i<n-1; i++) {
458fa24674SBarry Smith     done[current] = PETSC_TRUE;
468fa24674SBarry Smith     best          = -1;
478fa24674SBarry Smith     /* loop over all neighbors of current pivot */
488fa24674SBarry Smith     for (j=ai[current]; j<ai[current+1]; j++) {
498fa24674SBarry Smith       jj = aj[j];
508fa24674SBarry Smith       if (done[jj]) continue;
518fa24674SBarry Smith       /* loop over columns of potential next row computing weights for below and above diagonal */
528fa24674SBarry Smith       past = future = 0.0;
538fa24674SBarry Smith       for (k=ai[jj]; k<ai[jj+1]; k++) {
548fa24674SBarry Smith         kk = aj[k];
558fa24674SBarry Smith         if (done[kk]) past += PetscAbsScalar(aa[k]);
568fa24674SBarry Smith         else if (kk != jj) future  += PetscAbsScalar(aa[k]);
578fa24674SBarry Smith       }
588fa24674SBarry Smith       if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */
598fa24674SBarry Smith       if (past/future > best) {
608fa24674SBarry Smith         best = past/future;
618fa24674SBarry Smith         newcurrent = jj;
628fa24674SBarry Smith       }
638fa24674SBarry Smith     }
648fa24674SBarry Smith     if (best == -1) { /* no neighbors to select from so select best of all that remain */
658fa24674SBarry Smith       best = -1;
668fa24674SBarry Smith       for (k=0; k<n; k++) {
678fa24674SBarry Smith         if (done[k]) continue;
6875567043SBarry Smith         future = 0.0;
6975567043SBarry Smith         past   = 0.0;
708fa24674SBarry Smith         for (j=ai[k]; j<ai[k+1]; j++) {
718fa24674SBarry Smith           kk = aj[j];
728fa24674SBarry Smith           if (done[kk]) past += PetscAbsScalar(aa[j]);
738fa24674SBarry Smith           else if (kk != k) future  += PetscAbsScalar(aa[j]);
748fa24674SBarry Smith         }
758fa24674SBarry Smith         if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */
768fa24674SBarry Smith         if (past/future > best) {
778fa24674SBarry Smith           best = past/future;
788fa24674SBarry Smith           newcurrent = k;
798fa24674SBarry Smith         }
808fa24674SBarry Smith       }
818fa24674SBarry Smith     }
82e32f2f54SBarry Smith     if (current == newcurrent) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"newcurrent cannot be current");
838fa24674SBarry Smith     current = newcurrent;
848fa24674SBarry Smith     order[i+1] = current;
858fa24674SBarry Smith   }
868fa24674SBarry Smith   ierr = ISCreateGeneral(PETSC_COMM_SELF,n,order,irow);CHKERRQ(ierr);
878fa24674SBarry Smith   *icol = *irow;
888fa24674SBarry Smith   ierr = PetscObjectReference((PetscObject)*irow);CHKERRQ(ierr);
898fa24674SBarry Smith   ierr = PetscFree(done);CHKERRQ(ierr);
908fa24674SBarry Smith   ierr = PetscFree(order);CHKERRQ(ierr);
91d64ed03dSBarry Smith   PetscFunctionReturn(0);
923b2fbd54SBarry Smith }
93f6a26d55SMatthew Knepley EXTERN_C_END
943b2fbd54SBarry Smith 
95e631078cSBarry Smith EXTERN_C_BEGIN
964a2ae208SSatish Balay #undef __FUNCT__
97db4efbfdSBarry Smith #define __FUNCT__ "MatGetFactorAvailable_seqaij_petsc"
98db4efbfdSBarry Smith PetscErrorCode MatGetFactorAvailable_seqaij_petsc(Mat A,MatFactorType ftype,PetscTruth *flg)
99db4efbfdSBarry Smith {
100db4efbfdSBarry Smith   PetscFunctionBegin;
101db4efbfdSBarry Smith   *flg = PETSC_TRUE;
102db4efbfdSBarry Smith   PetscFunctionReturn(0);
103db4efbfdSBarry Smith }
104db4efbfdSBarry Smith EXTERN_C_END
105db4efbfdSBarry Smith 
106db4efbfdSBarry Smith EXTERN_C_BEGIN
107db4efbfdSBarry Smith #undef __FUNCT__
108b24902e0SBarry Smith #define __FUNCT__ "MatGetFactor_seqaij_petsc"
109b24902e0SBarry Smith PetscErrorCode MatGetFactor_seqaij_petsc(Mat A,MatFactorType ftype,Mat *B)
110b24902e0SBarry Smith {
111d0f46423SBarry Smith   PetscInt           n = A->rmap->n;
112b24902e0SBarry Smith   PetscErrorCode     ierr;
113b24902e0SBarry Smith 
114b24902e0SBarry Smith   PetscFunctionBegin;
115b24902e0SBarry Smith   ierr = MatCreate(((PetscObject)A)->comm,B);CHKERRQ(ierr);
116b24902e0SBarry Smith   ierr = MatSetSizes(*B,n,n,n,n);CHKERRQ(ierr);
117599ef60dSHong Zhang   if (ftype == MAT_FACTOR_LU || ftype == MAT_FACTOR_ILU || ftype == MAT_FACTOR_ILUDT){
118b24902e0SBarry Smith     ierr = MatSetType(*B,MATSEQAIJ);CHKERRQ(ierr);
11935233d99SShri Abhyankar     (*B)->ops->ilufactorsymbolic = MatILUFactorSymbolic_SeqAIJ;
12035233d99SShri Abhyankar     (*B)->ops->lufactorsymbolic  = MatLUFactorSymbolic_SeqAIJ;
121b24902e0SBarry Smith   } else if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) {
122b24902e0SBarry Smith     ierr = MatSetType(*B,MATSEQSBAIJ);CHKERRQ(ierr);
123b24902e0SBarry Smith     ierr = MatSeqSBAIJSetPreallocation(*B,1,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
12435233d99SShri Abhyankar     (*B)->ops->iccfactorsymbolic      = MatICCFactorSymbolic_SeqAIJ;
12535233d99SShri Abhyankar     (*B)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqAIJ;
126e32f2f54SBarry Smith   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Factor type not supported");
127d5f3da31SBarry Smith   (*B)->factortype = ftype;
128b24902e0SBarry Smith   PetscFunctionReturn(0);
129b24902e0SBarry Smith }
130e631078cSBarry Smith EXTERN_C_END
131b24902e0SBarry Smith 
132b24902e0SBarry Smith #undef __FUNCT__
133ad04f41aSHong Zhang #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ_inplace"
134ad04f41aSHong Zhang PetscErrorCode MatLUFactorSymbolic_SeqAIJ_inplace(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
135289bc588SBarry Smith {
136416022c9SBarry Smith   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
137289bc588SBarry Smith   IS                 isicol;
1386849ba73SBarry Smith   PetscErrorCode     ierr;
1395d0c19d7SBarry Smith   const PetscInt     *r,*ic;
1405d0c19d7SBarry Smith   PetscInt           i,n=A->rmap->n,*ai=a->i,*aj=a->j;
1411393bc97SHong Zhang   PetscInt           *bi,*bj,*ajtmp;
1421393bc97SHong Zhang   PetscInt           *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im;
1439e046878SBarry Smith   PetscReal          f;
1444875ba38SHong Zhang   PetscInt           nlnk,*lnk,k,**bi_ptr;
145a1a86e44SBarry Smith   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
1461393bc97SHong Zhang   PetscBT            lnkbt;
147289bc588SBarry Smith 
1483a40ed3dSBarry Smith   PetscFunctionBegin;
149e32f2f54SBarry Smith   if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"matrix must be square");
1504c49b128SBarry Smith   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
151ac121b3dSBarry Smith   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
152ac121b3dSBarry Smith   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
153289bc588SBarry Smith 
154289bc588SBarry Smith   /* get new row pointers */
1551393bc97SHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
1561393bc97SHong Zhang   bi[0] = 0;
1571393bc97SHong Zhang 
1581393bc97SHong Zhang   /* bdiag is location of diagonal in factor */
1591393bc97SHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
1601393bc97SHong Zhang   bdiag[0] = 0;
1611393bc97SHong Zhang 
1621393bc97SHong Zhang   /* linked list for storing column indices of the active row */
1631393bc97SHong Zhang   nlnk = n + 1;
1641393bc97SHong Zhang   ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
1651393bc97SHong Zhang 
16635baf27eSSatish Balay   ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr);
1671393bc97SHong Zhang 
1681393bc97SHong Zhang   /* initial FreeSpace size is f*(ai[n]+1) */
169d3d32019SBarry Smith   f = info->fill;
170a1a86e44SBarry Smith   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr);
1711393bc97SHong Zhang   current_space = free_space;
172289bc588SBarry Smith 
173289bc588SBarry Smith   for (i=0; i<n; i++) {
1741393bc97SHong Zhang     /* copy previous fill into linked list */
175289bc588SBarry Smith     nzi = 0;
1761393bc97SHong Zhang     nnz = ai[r[i]+1] - ai[r[i]];
177e32f2f54SBarry Smith     if (!nnz) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i);
1781393bc97SHong Zhang     ajtmp = aj + ai[r[i]];
179afcc9904SHong Zhang     ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
1801393bc97SHong Zhang     nzi += nlnk;
1811393bc97SHong Zhang 
1821393bc97SHong Zhang     /* add pivot rows into linked list */
1831393bc97SHong Zhang     row = lnk[n];
1841393bc97SHong Zhang     while (row < i) {
185d1170560SHong Zhang       nzbd    = bdiag[row] - bi[row] + 1; /* num of entries in the row with column index <= row */
186d1170560SHong Zhang       ajtmp   = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */
187d1170560SHong Zhang       ierr = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr);
1881393bc97SHong Zhang       nzi += nlnk;
1891393bc97SHong Zhang       row  = lnk[row];
190289bc588SBarry Smith     }
1911393bc97SHong Zhang     bi[i+1] = bi[i] + nzi;
1921393bc97SHong Zhang     im[i]   = nzi;
1931393bc97SHong Zhang 
1941393bc97SHong Zhang     /* mark bdiag */
1951393bc97SHong Zhang     nzbd = 0;
1961393bc97SHong Zhang     nnz  = nzi;
1971393bc97SHong Zhang     k    = lnk[n];
1981393bc97SHong Zhang     while (nnz-- && k < i){
1991393bc97SHong Zhang       nzbd++;
2001393bc97SHong Zhang       k = lnk[k];
2011393bc97SHong Zhang     }
2021393bc97SHong Zhang     bdiag[i] = bi[i] + nzbd;
2031393bc97SHong Zhang 
2041393bc97SHong Zhang     /* if free space is not available, make more free space */
2051393bc97SHong Zhang     if (current_space->local_remaining<nzi) {
2061393bc97SHong Zhang       nnz = (n - i)*nzi; /* estimated and max additional space needed */
207a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
2081393bc97SHong Zhang       reallocs++;
2091393bc97SHong Zhang     }
2101393bc97SHong Zhang 
2111393bc97SHong Zhang     /* copy data into free space, then initialize lnk */
2121393bc97SHong Zhang     ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
2131393bc97SHong Zhang     bi_ptr[i] = current_space->array;
2141393bc97SHong Zhang     current_space->array           += nzi;
2151393bc97SHong Zhang     current_space->local_used      += nzi;
2161393bc97SHong Zhang     current_space->local_remaining -= nzi;
217289bc588SBarry Smith   }
2186cf91177SBarry Smith #if defined(PETSC_USE_INFO)
219464e8d44SSatish Balay   if (ai[n] != 0) {
2201393bc97SHong Zhang     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
221ae15b995SBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr);
222ae15b995SBarry Smith     ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
223ae15b995SBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr);
224ae15b995SBarry Smith     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
22505bf559cSSatish Balay   } else {
226ae15b995SBarry Smith     ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr);
22705bf559cSSatish Balay   }
22863ba0a88SBarry Smith #endif
2292fb3ed76SBarry Smith 
230898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
231898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
2321987afe7SBarry Smith 
2331393bc97SHong Zhang   /* destroy list of free space and other temporary array(s) */
2341393bc97SHong Zhang   ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
235a1a86e44SBarry Smith   ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr);
2361393bc97SHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
23735baf27eSSatish Balay   ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr);
238289bc588SBarry Smith 
239289bc588SBarry Smith   /* put together the new matrix */
240719d5645SBarry Smith   ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
241719d5645SBarry Smith   ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr);
242719d5645SBarry Smith   b    = (Mat_SeqAIJ*)(B)->data;
243e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
244e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
2457c922b88SBarry Smith   b->singlemalloc = PETSC_FALSE;
2461393bc97SHong Zhang   ierr          = PetscMalloc((bi[n]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);
2471393bc97SHong Zhang   b->j          = bj;
2481393bc97SHong Zhang   b->i          = bi;
2491393bc97SHong Zhang   b->diag       = bdiag;
250416022c9SBarry Smith   b->ilen       = 0;
251416022c9SBarry Smith   b->imax       = 0;
252416022c9SBarry Smith   b->row        = isrow;
253416022c9SBarry Smith   b->col        = iscol;
254c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
255c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
25682bf6240SBarry Smith   b->icol       = isicol;
25787828ca2SBarry Smith   ierr          = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
2581393bc97SHong Zhang 
2591393bc97SHong Zhang   /* In b structure:  Free imax, ilen, old a, old j.  Allocate solve_work, new a, new j */
260719d5645SBarry Smith   ierr = PetscLogObjectMemory(B,(bi[n]-n)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
2611393bc97SHong Zhang   b->maxnz = b->nz = bi[n] ;
2627fd98bd6SLois Curfman McInnes 
263d5f3da31SBarry Smith   (B)->factortype            = MAT_FACTOR_LU;
264719d5645SBarry Smith   (B)->info.factor_mallocs   = reallocs;
265719d5645SBarry Smith   (B)->info.fill_ratio_given = f;
266703deb49SSatish Balay 
2679f7d0b68SHong Zhang   if (ai[n]) {
268719d5645SBarry Smith     (B)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]);
269eea2913aSSatish Balay   } else {
270719d5645SBarry Smith     (B)->info.fill_ratio_needed = 0.0;
271eea2913aSSatish Balay   }
272ad04f41aSHong Zhang   (B)->ops->lufactornumeric  = MatLUFactorNumeric_SeqAIJ_inplace;
273f2344125SBarry Smith   if (a->inode.size) {
274f2344125SBarry Smith     (B)->ops->lufactornumeric  = MatLUFactorNumeric_SeqAIJ_Inode_inplace;
275f2344125SBarry Smith   }
2763a40ed3dSBarry Smith   PetscFunctionReturn(0);
277289bc588SBarry Smith }
2781393bc97SHong Zhang 
279ce3d78c0SShri Abhyankar #undef __FUNCT__
28035233d99SShri Abhyankar #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ"
28135233d99SShri Abhyankar PetscErrorCode MatLUFactorSymbolic_SeqAIJ(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
282ce3d78c0SShri Abhyankar {
283ce3d78c0SShri Abhyankar   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
284ce3d78c0SShri Abhyankar   IS                 isicol;
285ce3d78c0SShri Abhyankar   PetscErrorCode     ierr;
286ce3d78c0SShri Abhyankar   const PetscInt     *r,*ic;
287ce3d78c0SShri Abhyankar   PetscInt           i,n=A->rmap->n,*ai=a->i,*aj=a->j;
288ce3d78c0SShri Abhyankar   PetscInt           *bi,*bj,*ajtmp;
289ce3d78c0SShri Abhyankar   PetscInt           *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im;
290ce3d78c0SShri Abhyankar   PetscReal          f;
291ce3d78c0SShri Abhyankar   PetscInt           nlnk,*lnk,k,**bi_ptr;
292ce3d78c0SShri Abhyankar   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
293ce3d78c0SShri Abhyankar   PetscBT            lnkbt;
2943898ab1dSSatish Balay   PetscTruth         olddatastruct=PETSC_FALSE;
295ce3d78c0SShri Abhyankar 
296ce3d78c0SShri Abhyankar   PetscFunctionBegin;
2973898ab1dSSatish Balay   /* Uncomment the oldatastruct part only while testing new data structure for MatSolve() */
298280e8513SShri Abhyankar   ierr = PetscOptionsGetTruth(PETSC_NULL,"-lu_old",&olddatastruct,PETSC_NULL);CHKERRQ(ierr);
299628f99d7SShri Abhyankar   if(olddatastruct){
300628f99d7SShri Abhyankar     ierr = MatLUFactorSymbolic_SeqAIJ_inplace(B,A,isrow,iscol,info);CHKERRQ(ierr);
301628f99d7SShri Abhyankar     PetscFunctionReturn(0);
302628f99d7SShri Abhyankar   }
303628f99d7SShri Abhyankar 
304e32f2f54SBarry Smith   if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"matrix must be square");
305ce3d78c0SShri Abhyankar   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
306ce3d78c0SShri Abhyankar   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
307ce3d78c0SShri Abhyankar   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
308ce3d78c0SShri Abhyankar 
3091bfa06eaSJed Brown   /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */
3101bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
3111bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
312a6df321fSShri Abhyankar   bi[0] = bdiag[0] = 0;
3139b48462bSShri Abhyankar 
3149b48462bSShri Abhyankar   /* linked list for storing column indices of the active row */
3159b48462bSShri Abhyankar   nlnk = n + 1;
3169b48462bSShri Abhyankar   ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
3179b48462bSShri Abhyankar 
3189b48462bSShri Abhyankar   ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr);
3199b48462bSShri Abhyankar 
3209b48462bSShri Abhyankar   /* initial FreeSpace size is f*(ai[n]+1) */
3219b48462bSShri Abhyankar   f = info->fill;
3229b48462bSShri Abhyankar   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr);
3239b48462bSShri Abhyankar   current_space = free_space;
3249b48462bSShri Abhyankar 
3259b48462bSShri Abhyankar   for (i=0; i<n; i++) {
3269b48462bSShri Abhyankar     /* copy previous fill into linked list */
3279b48462bSShri Abhyankar     nzi = 0;
3289b48462bSShri Abhyankar     nnz = ai[r[i]+1] - ai[r[i]];
329e32f2f54SBarry Smith     if (!nnz) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i);
3309b48462bSShri Abhyankar     ajtmp = aj + ai[r[i]];
3319b48462bSShri Abhyankar     ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
3329b48462bSShri Abhyankar     nzi += nlnk;
3339b48462bSShri Abhyankar 
3349b48462bSShri Abhyankar     /* add pivot rows into linked list */
3359b48462bSShri Abhyankar     row = lnk[n];
3369b48462bSShri Abhyankar     while (row < i){
3379b48462bSShri Abhyankar       nzbd  = bdiag[row] + 1; /* num of entries in the row with column index <= row */
3389b48462bSShri Abhyankar       ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */
3399b48462bSShri Abhyankar       ierr  = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr);
3409b48462bSShri Abhyankar       nzi  += nlnk;
3419b48462bSShri Abhyankar       row   = lnk[row];
3429b48462bSShri Abhyankar     }
3439b48462bSShri Abhyankar     bi[i+1] = bi[i] + nzi;
3449b48462bSShri Abhyankar     im[i]   = nzi;
3459b48462bSShri Abhyankar 
3469b48462bSShri Abhyankar     /* mark bdiag */
3479b48462bSShri Abhyankar     nzbd = 0;
3489b48462bSShri Abhyankar     nnz  = nzi;
3499b48462bSShri Abhyankar     k    = lnk[n];
3509b48462bSShri Abhyankar     while (nnz-- && k < i){
3519b48462bSShri Abhyankar       nzbd++;
3529b48462bSShri Abhyankar       k = lnk[k];
3539b48462bSShri Abhyankar     }
3549b48462bSShri Abhyankar     bdiag[i] = nzbd; /* note: bdiag[i] = nnzL as input for PetscFreeSpaceContiguous_LU() */
3559b48462bSShri Abhyankar 
3569b48462bSShri Abhyankar     /* if free space is not available, make more free space */
3579b48462bSShri Abhyankar     if (current_space->local_remaining<nzi) {
3589b48462bSShri Abhyankar       nnz = 2*(n - i)*nzi; /* estimated and max additional space needed */
3599b48462bSShri Abhyankar       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
3609b48462bSShri Abhyankar       reallocs++;
3619b48462bSShri Abhyankar     }
3629b48462bSShri Abhyankar 
3639b48462bSShri Abhyankar     /* copy data into free space, then initialize lnk */
3649b48462bSShri Abhyankar     ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
3659b48462bSShri Abhyankar     bi_ptr[i] = current_space->array;
3669b48462bSShri Abhyankar     current_space->array           += nzi;
3679b48462bSShri Abhyankar     current_space->local_used      += nzi;
3689b48462bSShri Abhyankar     current_space->local_remaining -= nzi;
3699b48462bSShri Abhyankar   }
3709b48462bSShri Abhyankar #if defined(PETSC_USE_INFO)
3719b48462bSShri Abhyankar   if (ai[n] != 0) {
3729b48462bSShri Abhyankar     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
3739b48462bSShri Abhyankar     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr);
3749b48462bSShri Abhyankar     ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
3759b48462bSShri Abhyankar     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr);
3769b48462bSShri Abhyankar     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
3779b48462bSShri Abhyankar   } else {
3789b48462bSShri Abhyankar     ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr);
3799b48462bSShri Abhyankar   }
3809b48462bSShri Abhyankar #endif
3819b48462bSShri Abhyankar 
3829b48462bSShri Abhyankar   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
3839b48462bSShri Abhyankar   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
3849b48462bSShri Abhyankar 
3859b48462bSShri Abhyankar   /* destroy list of free space and other temporary array(s) */
3869b48462bSShri Abhyankar   ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
3872ce24eb6SHong Zhang   ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr);
3889b48462bSShri Abhyankar   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
3899b48462bSShri Abhyankar   ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr);
3909b48462bSShri Abhyankar 
3919b48462bSShri Abhyankar   /* put together the new matrix */
3929b48462bSShri Abhyankar   ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
3939b48462bSShri Abhyankar   ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr);
3949b48462bSShri Abhyankar   b    = (Mat_SeqAIJ*)(B)->data;
3959b48462bSShri Abhyankar   b->free_a       = PETSC_TRUE;
3969b48462bSShri Abhyankar   b->free_ij      = PETSC_TRUE;
3979b48462bSShri Abhyankar   b->singlemalloc = PETSC_FALSE;
3989b48462bSShri Abhyankar   ierr = PetscMalloc((bdiag[0]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);
3999b48462bSShri Abhyankar   b->j          = bj;
4009b48462bSShri Abhyankar   b->i          = bi;
4019b48462bSShri Abhyankar   b->diag       = bdiag;
4029b48462bSShri Abhyankar   b->ilen       = 0;
4039b48462bSShri Abhyankar   b->imax       = 0;
4049b48462bSShri Abhyankar   b->row        = isrow;
4059b48462bSShri Abhyankar   b->col        = iscol;
4069b48462bSShri Abhyankar   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
4079b48462bSShri Abhyankar   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
4089b48462bSShri Abhyankar   b->icol       = isicol;
4099b48462bSShri Abhyankar   ierr          = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
4109b48462bSShri Abhyankar 
4119b48462bSShri Abhyankar   /* In b structure:  Free imax, ilen, old a, old j.  Allocate solve_work, new a, new j */
4129b48462bSShri Abhyankar   ierr = PetscLogObjectMemory(B,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
4139b48462bSShri Abhyankar   b->maxnz = b->nz = bdiag[0]+1;
414d5f3da31SBarry Smith   B->factortype            = MAT_FACTOR_LU;
415f284f12aSHong Zhang   B->info.factor_mallocs   = reallocs;
416f284f12aSHong Zhang   B->info.fill_ratio_given = f;
4179b48462bSShri Abhyankar 
4189b48462bSShri Abhyankar   if (ai[n]) {
419f284f12aSHong Zhang     B->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]);
4209b48462bSShri Abhyankar   } else {
421f284f12aSHong Zhang     B->info.fill_ratio_needed = 0.0;
4229b48462bSShri Abhyankar   }
42335233d99SShri Abhyankar   B->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ;
424f2344125SBarry Smith   if (a->inode.size) {
425d3ac4fa3SBarry Smith     B->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode;
426d3ac4fa3SBarry Smith   }
4279b48462bSShri Abhyankar   PetscFunctionReturn(0);
4289b48462bSShri Abhyankar }
4299b48462bSShri Abhyankar 
4305b5bc046SBarry Smith /*
4315b5bc046SBarry Smith     Trouble in factorization, should we dump the original matrix?
4325b5bc046SBarry Smith */
4335b5bc046SBarry Smith #undef __FUNCT__
4345b5bc046SBarry Smith #define __FUNCT__ "MatFactorDumpMatrix"
4355b5bc046SBarry Smith PetscErrorCode MatFactorDumpMatrix(Mat A)
4365b5bc046SBarry Smith {
4375b5bc046SBarry Smith   PetscErrorCode ierr;
43890d69ab7SBarry Smith   PetscTruth     flg = PETSC_FALSE;
4395b5bc046SBarry Smith 
4405b5bc046SBarry Smith   PetscFunctionBegin;
44190d69ab7SBarry Smith   ierr = PetscOptionsGetTruth(PETSC_NULL,"-mat_factor_dump_on_error",&flg,PETSC_NULL);CHKERRQ(ierr);
4425b5bc046SBarry Smith   if (flg) {
4435b5bc046SBarry Smith     PetscViewer viewer;
4445b5bc046SBarry Smith     char        filename[PETSC_MAX_PATH_LEN];
4455b5bc046SBarry Smith 
4465b5bc046SBarry Smith     ierr = PetscSNPrintf(filename,PETSC_MAX_PATH_LEN,"matrix_factor_error.%d",PetscGlobalRank);CHKERRQ(ierr);
4477adad957SLisandro Dalcin     ierr = PetscViewerBinaryOpen(((PetscObject)A)->comm,filename,FILE_MODE_WRITE,&viewer);CHKERRQ(ierr);
4485b5bc046SBarry Smith     ierr = MatView(A,viewer);CHKERRQ(ierr);
4495b5bc046SBarry Smith     ierr = PetscViewerDestroy(viewer);CHKERRQ(ierr);
4505b5bc046SBarry Smith   }
4515b5bc046SBarry Smith   PetscFunctionReturn(0);
4525b5bc046SBarry Smith }
4535b5bc046SBarry Smith 
454c9c72f8fSHong Zhang #undef __FUNCT__
45535233d99SShri Abhyankar #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ"
45635233d99SShri Abhyankar PetscErrorCode MatLUFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info)
457c9c72f8fSHong Zhang {
458c9c72f8fSHong Zhang   Mat              C=B;
459c9c72f8fSHong Zhang   Mat_SeqAIJ       *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data;
460c9c72f8fSHong Zhang   IS               isrow = b->row,isicol = b->icol;
461c9c72f8fSHong Zhang   PetscErrorCode   ierr;
462c9c72f8fSHong Zhang   const PetscInt   *r,*ic,*ics;
463bbd65245SShri Abhyankar   const PetscInt   n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j,*bdiag=b->diag;
464bbd65245SShri Abhyankar   PetscInt         i,j,k,nz,nzL,row,*pj;
465bbd65245SShri Abhyankar   const PetscInt   *ajtmp,*bjtmp;
466bbd65245SShri Abhyankar   MatScalar        *rtmp,*pc,multiplier,*pv;
467bbd65245SShri Abhyankar   const  MatScalar *aa=a->a,*v;
468c9c72f8fSHong Zhang   PetscTruth       row_identity,col_identity;
469d90ac83dSHong Zhang   FactorShiftCtx   sctx;
4704f81c4b7SBarry Smith   const PetscInt   *ddiag;
471d4ad06f7SHong Zhang   PetscReal        rs;
472d4ad06f7SHong Zhang   MatScalar        d;
473eca579c0SMatthew G Knepley   PetscInt         newshift = 0;
474d4ad06f7SHong Zhang 
475c9c72f8fSHong Zhang   PetscFunctionBegin;
476d6e5152cSHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
477d90ac83dSHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
478d4ad06f7SHong Zhang 
479f4db908eSBarry Smith   if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
480d4ad06f7SHong Zhang     ddiag          = a->diag;
481d4ad06f7SHong Zhang     sctx.shift_top = info->zeropivot;
482d4ad06f7SHong Zhang     for (i=0; i<n; i++) {
483d4ad06f7SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
484d4ad06f7SHong Zhang       d  = (aa)[ddiag[i]];
485d4ad06f7SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
486d4ad06f7SHong Zhang       v  = aa+ai[i];
487d4ad06f7SHong Zhang       nz = ai[i+1] - ai[i];
488d4ad06f7SHong Zhang       for (j=0; j<nz; j++)
489d4ad06f7SHong Zhang 	rs += PetscAbsScalar(v[j]);
490d4ad06f7SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
491d4ad06f7SHong Zhang     }
492d4ad06f7SHong Zhang     sctx.shift_top   *= 1.1;
493d4ad06f7SHong Zhang     sctx.nshift_max   = 5;
494d4ad06f7SHong Zhang     sctx.shift_lo     = 0.;
495d4ad06f7SHong Zhang     sctx.shift_hi     = 1.;
496d4ad06f7SHong Zhang   }
497d4ad06f7SHong Zhang 
498c9c72f8fSHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
499c9c72f8fSHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
500c9c72f8fSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr);
501c9c72f8fSHong Zhang   ics  = ic;
502c9c72f8fSHong Zhang 
503d4ad06f7SHong Zhang   do {
504d90ac83dSHong Zhang     sctx.useshift = PETSC_FALSE;
505c9c72f8fSHong Zhang     for (i=0; i<n; i++){
506c9c72f8fSHong Zhang       /* zero rtmp */
507c9c72f8fSHong Zhang       /* L part */
508c9c72f8fSHong Zhang       nz    = bi[i+1] - bi[i];
509c9c72f8fSHong Zhang       bjtmp = bj + bi[i];
510c9c72f8fSHong Zhang       for  (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
511c9c72f8fSHong Zhang 
512c9c72f8fSHong Zhang       /* U part */
51362fbd6afSShri Abhyankar       nz = bdiag[i]-bdiag[i+1];
51462fbd6afSShri Abhyankar       bjtmp = bj + bdiag[i+1]+1;
515813a1f2bSShri Abhyankar       for  (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
516813a1f2bSShri Abhyankar 
517813a1f2bSShri Abhyankar       /* load in initial (unfactored row) */
518813a1f2bSShri Abhyankar       nz    = ai[r[i]+1] - ai[r[i]];
519813a1f2bSShri Abhyankar       ajtmp = aj + ai[r[i]];
520813a1f2bSShri Abhyankar       v     = aa + ai[r[i]];
521813a1f2bSShri Abhyankar       for (j=0; j<nz; j++) {
522813a1f2bSShri Abhyankar         rtmp[ics[ajtmp[j]]] = v[j];
523813a1f2bSShri Abhyankar       }
524813a1f2bSShri Abhyankar       /* ZeropivotApply() */
52598a93161SHong Zhang       rtmp[i] += sctx.shift_amount;  /* shift the diagonal of the matrix */
526813a1f2bSShri Abhyankar 
527813a1f2bSShri Abhyankar       /* elimination */
528813a1f2bSShri Abhyankar       bjtmp = bj + bi[i];
529813a1f2bSShri Abhyankar       row   = *bjtmp++;
530813a1f2bSShri Abhyankar       nzL   = bi[i+1] - bi[i];
531f268cba8SShri Abhyankar       for(k=0; k < nzL;k++) {
532813a1f2bSShri Abhyankar         pc = rtmp + row;
533813a1f2bSShri Abhyankar         if (*pc != 0.0) {
534813a1f2bSShri Abhyankar           pv         = b->a + bdiag[row];
535813a1f2bSShri Abhyankar           multiplier = *pc * (*pv);
536813a1f2bSShri Abhyankar           *pc        = multiplier;
53762fbd6afSShri Abhyankar           pj = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */
53862fbd6afSShri Abhyankar 	  pv = b->a + bdiag[row+1]+1;
53962fbd6afSShri Abhyankar 	  nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */
540813a1f2bSShri Abhyankar           for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j];
541813a1f2bSShri Abhyankar           ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr);
542813a1f2bSShri Abhyankar         }
543f268cba8SShri Abhyankar         row = *bjtmp++;
544813a1f2bSShri Abhyankar       }
545813a1f2bSShri Abhyankar 
546813a1f2bSShri Abhyankar       /* finished row so stick it into b->a */
547813a1f2bSShri Abhyankar       rs = 0.0;
548813a1f2bSShri Abhyankar       /* L part */
549813a1f2bSShri Abhyankar       pv   = b->a + bi[i] ;
550813a1f2bSShri Abhyankar       pj   = b->j + bi[i] ;
551813a1f2bSShri Abhyankar       nz   = bi[i+1] - bi[i];
552813a1f2bSShri Abhyankar       for (j=0; j<nz; j++) {
553813a1f2bSShri Abhyankar         pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]);
554813a1f2bSShri Abhyankar       }
555813a1f2bSShri Abhyankar 
556813a1f2bSShri Abhyankar       /* U part */
55762fbd6afSShri Abhyankar       pv = b->a + bdiag[i+1]+1;
55862fbd6afSShri Abhyankar       pj = b->j + bdiag[i+1]+1;
55962fbd6afSShri Abhyankar       nz = bdiag[i] - bdiag[i+1]-1;
560813a1f2bSShri Abhyankar       for (j=0; j<nz; j++) {
561813a1f2bSShri Abhyankar         pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]);
562813a1f2bSShri Abhyankar       }
563813a1f2bSShri Abhyankar 
564813a1f2bSShri Abhyankar       sctx.rs  = rs;
565813a1f2bSShri Abhyankar       sctx.pv  = rtmp[i];
5669f522e40SShri Abhyankar       ierr = MatPivotCheck(info,&sctx,i,&newshift);CHKERRQ(ierr);
5679f522e40SShri Abhyankar       if(newshift == 1) break;
5688ff23777SHong Zhang       rtmp[i] = sctx.pv;
569813a1f2bSShri Abhyankar 
570813a1f2bSShri Abhyankar       /* Mark diagonal and invert diagonal for simplier triangular solves */
571813a1f2bSShri Abhyankar       pv  = b->a + bdiag[i];
572813a1f2bSShri Abhyankar       *pv = 1.0/rtmp[i];
573813a1f2bSShri Abhyankar 
574813a1f2bSShri Abhyankar     } /* endof for (i=0; i<n; i++){ */
575813a1f2bSShri Abhyankar 
5768ff23777SHong Zhang     /* MatPivotRefine() */
577f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.useshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max){
578813a1f2bSShri Abhyankar       /*
579813a1f2bSShri Abhyankar        * if no shift in this attempt & shifting & started shifting & can refine,
580813a1f2bSShri Abhyankar        * then try lower shift
581813a1f2bSShri Abhyankar        */
582813a1f2bSShri Abhyankar       sctx.shift_hi       = sctx.shift_fraction;
583813a1f2bSShri Abhyankar       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
584813a1f2bSShri Abhyankar       sctx.shift_amount   = sctx.shift_fraction * sctx.shift_top;
585d90ac83dSHong Zhang       sctx.useshift        = PETSC_TRUE;
586813a1f2bSShri Abhyankar       sctx.nshift++;
587813a1f2bSShri Abhyankar     }
588d90ac83dSHong Zhang   } while (sctx.useshift);
589813a1f2bSShri Abhyankar 
590813a1f2bSShri Abhyankar   ierr = PetscFree(rtmp);CHKERRQ(ierr);
591813a1f2bSShri Abhyankar   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
592813a1f2bSShri Abhyankar   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
593d3ac4fa3SBarry Smith 
594813a1f2bSShri Abhyankar   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
595813a1f2bSShri Abhyankar   ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr);
596813a1f2bSShri Abhyankar   if (row_identity && col_identity) {
59735233d99SShri Abhyankar     C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering;
598813a1f2bSShri Abhyankar   } else {
59935233d99SShri Abhyankar     C->ops->solve = MatSolve_SeqAIJ;
600813a1f2bSShri Abhyankar   }
60135233d99SShri Abhyankar   C->ops->solveadd           = MatSolveAdd_SeqAIJ;
60235233d99SShri Abhyankar   C->ops->solvetranspose     = MatSolveTranspose_SeqAIJ;
60335233d99SShri Abhyankar   C->ops->solvetransposeadd  = MatSolveTransposeAdd_SeqAIJ;
60435233d99SShri Abhyankar   C->ops->matsolve           = MatMatSolve_SeqAIJ;
605813a1f2bSShri Abhyankar   C->assembled    = PETSC_TRUE;
606813a1f2bSShri Abhyankar   C->preallocated = PETSC_TRUE;
607813a1f2bSShri Abhyankar   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
608813a1f2bSShri Abhyankar 
60914d2772eSHong Zhang   /* MatShiftView(A,info,&sctx) */
610813a1f2bSShri Abhyankar   if (sctx.nshift){
611f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
612813a1f2bSShri Abhyankar       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,sctx.shift_fraction,sctx.shift_top);CHKERRQ(ierr);
613f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
614813a1f2bSShri Abhyankar       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
615f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_INBLOCKS){
616915743fcSHong Zhang       ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %G\n",sctx.nshift,info->shiftamount);CHKERRQ(ierr);
617813a1f2bSShri Abhyankar     }
618813a1f2bSShri Abhyankar   }
619d3ac4fa3SBarry Smith   ierr = Mat_CheckInode_FactorLU(C,PETSC_FALSE);CHKERRQ(ierr);
620813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
621813a1f2bSShri Abhyankar }
622813a1f2bSShri Abhyankar 
6234a2ae208SSatish Balay #undef __FUNCT__
624ad04f41aSHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_inplace"
625ad04f41aSHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ_inplace(Mat B,Mat A,const MatFactorInfo *info)
626289bc588SBarry Smith {
627719d5645SBarry Smith   Mat             C=B;
628416022c9SBarry Smith   Mat_SeqAIJ      *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data;
62982bf6240SBarry Smith   IS              isrow = b->row,isicol = b->icol;
6306849ba73SBarry Smith   PetscErrorCode  ierr;
6315d0c19d7SBarry Smith   const PetscInt   *r,*ic,*ics;
632d278edc7SBarry Smith   PetscInt        nz,row,i,j,n=A->rmap->n,diag;
633d278edc7SBarry Smith   const PetscInt  *ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j;
634d278edc7SBarry Smith   const PetscInt  *ajtmp,*bjtmp,*diag_offset = b->diag,*pj;
635d278edc7SBarry Smith   MatScalar       *pv,*rtmp,*pc,multiplier,d;
636d278edc7SBarry Smith   const MatScalar *v,*aa=a->a;
63733f9893dSHong Zhang   PetscReal       rs=0.0;
638d90ac83dSHong Zhang   FactorShiftCtx  sctx;
639daa17b54SHong Zhang   PetscInt        newshift;
6404f81c4b7SBarry Smith   const PetscInt  *ddiag;
641d3ac4fa3SBarry Smith   PetscTruth      row_identity, col_identity;
642289bc588SBarry Smith 
6433a40ed3dSBarry Smith   PetscFunctionBegin;
644504a3fadSHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
645504a3fadSHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
646ada7143aSSatish Balay 
647f4db908eSBarry Smith   if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
64842898933SHong Zhang     ddiag          = a->diag;
6499f7d0b68SHong Zhang     sctx.shift_top = info->zeropivot;
6506cc28720Svictorle     for (i=0; i<n; i++) {
6519f95998fSHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
6529f7d0b68SHong Zhang       d  = (aa)[ddiag[i]];
6531a890ab1SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
6549f7d0b68SHong Zhang       v  = aa+ai[i];
6559f7d0b68SHong Zhang       nz = ai[i+1] - ai[i];
656e24cfe64SHong Zhang       for (j=0; j<nz; j++)
6571a890ab1SHong Zhang 	rs += PetscAbsScalar(v[j]);
6581a890ab1SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
6596cc28720Svictorle     }
660118f5789SBarry Smith     sctx.shift_top   *= 1.1;
661d2276718SHong Zhang     sctx.nshift_max   = 5;
662d2276718SHong Zhang     sctx.shift_lo     = 0.;
663d2276718SHong Zhang     sctx.shift_hi     = 1.;
664a0a356efSHong Zhang   }
665a0a356efSHong Zhang 
666504a3fadSHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
667504a3fadSHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
668504a3fadSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr);
669504a3fadSHong Zhang   ics  = ic;
670504a3fadSHong Zhang 
671085256b3SBarry Smith   do {
672d90ac83dSHong Zhang     sctx.useshift = PETSC_FALSE;
673289bc588SBarry Smith     for (i=0; i<n; i++){
674b3bf805bSHong Zhang       nz    = bi[i+1] - bi[i];
675b3bf805bSHong Zhang       bjtmp = bj + bi[i];
6769f7d0b68SHong Zhang       for  (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
677289bc588SBarry Smith 
678289bc588SBarry Smith       /* load in initial (unfactored row) */
6799f7d0b68SHong Zhang       nz    = ai[r[i]+1] - ai[r[i]];
6809f7d0b68SHong Zhang       ajtmp = aj + ai[r[i]];
6819f7d0b68SHong Zhang       v     = aa + ai[r[i]];
682085256b3SBarry Smith       for (j=0; j<nz; j++) {
683b3bf805bSHong Zhang         rtmp[ics[ajtmp[j]]] = v[j];
684085256b3SBarry Smith       }
685d2276718SHong Zhang       rtmp[ics[r[i]]] += sctx.shift_amount; /* shift the diagonal of the matrix */
686289bc588SBarry Smith 
687b3bf805bSHong Zhang       row = *bjtmp++;
688f2582111SSatish Balay       while  (row < i) {
6898c37ef55SBarry Smith         pc = rtmp + row;
6908c37ef55SBarry Smith         if (*pc != 0.0) {
691010a20caSHong Zhang           pv         = b->a + diag_offset[row];
692010a20caSHong Zhang           pj         = b->j + diag_offset[row] + 1;
69335aab85fSBarry Smith           multiplier = *pc / *pv++;
6948c37ef55SBarry Smith           *pc        = multiplier;
695b3bf805bSHong Zhang           nz         = bi[row+1] - diag_offset[row] - 1;
6969f7d0b68SHong Zhang           for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j];
697dc0b31edSSatish Balay           ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr);
698289bc588SBarry Smith         }
699b3bf805bSHong Zhang         row = *bjtmp++;
700289bc588SBarry Smith       }
701416022c9SBarry Smith       /* finished row so stick it into b->a */
702b3bf805bSHong Zhang       pv   = b->a + bi[i] ;
703b3bf805bSHong Zhang       pj   = b->j + bi[i] ;
704b3bf805bSHong Zhang       nz   = bi[i+1] - bi[i];
705b3bf805bSHong Zhang       diag = diag_offset[i] - bi[i];
706e57ff13aSBarry Smith       rs   = 0.0;
707d3d32019SBarry Smith       for (j=0; j<nz; j++) {
7089f7d0b68SHong Zhang         pv[j] = rtmp[pj[j]];
7099f7d0b68SHong Zhang         rs   += PetscAbsScalar(pv[j]);
710d3d32019SBarry Smith       }
711e57ff13aSBarry Smith       rs   -= PetscAbsScalar(pv[diag]);
712d2276718SHong Zhang 
713d2276718SHong Zhang       sctx.rs  = rs;
714d2276718SHong Zhang       sctx.pv  = pv[diag];
715504a3fadSHong Zhang       ierr = MatPivotCheck(info,&sctx,i,&newshift);CHKERRQ(ierr);
7165b5bc046SBarry Smith       if (newshift == 1) break;
717504a3fadSHong Zhang       pv[diag] = sctx.pv;
71835aab85fSBarry Smith     }
719d2276718SHong Zhang 
720f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.useshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) {
7216cc28720Svictorle       /*
7226c037d1bSvictorle        * if no shift in this attempt & shifting & started shifting & can refine,
7236cc28720Svictorle        * then try lower shift
7246cc28720Svictorle        */
7250481f469SBarry Smith       sctx.shift_hi       = sctx.shift_fraction;
7260481f469SBarry Smith       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
7270481f469SBarry Smith       sctx.shift_amount   = sctx.shift_fraction * sctx.shift_top;
728d90ac83dSHong Zhang       sctx.useshift        = PETSC_TRUE;
729d2276718SHong Zhang       sctx.nshift++;
7306cc28720Svictorle     }
731d90ac83dSHong Zhang   } while (sctx.useshift);
7320f11f9aeSSatish Balay 
73335aab85fSBarry Smith   /* invert diagonal entries for simplier triangular solves */
73435aab85fSBarry Smith   for (i=0; i<n; i++) {
735010a20caSHong Zhang     b->a[diag_offset[i]] = 1.0/b->a[diag_offset[i]];
73635aab85fSBarry Smith   }
737606d414cSSatish Balay   ierr = PetscFree(rtmp);CHKERRQ(ierr);
73878b31e54SBarry Smith   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
73978b31e54SBarry Smith   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
740d3ac4fa3SBarry Smith 
7418d8c7c43SBarry Smith   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
7428d8c7c43SBarry Smith   ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr);
7438d8c7c43SBarry Smith   if (row_identity && col_identity) {
744ad04f41aSHong Zhang     C->ops->solve   = MatSolve_SeqAIJ_NaturalOrdering_inplace;
7458d8c7c43SBarry Smith   } else {
746ad04f41aSHong Zhang     C->ops->solve   = MatSolve_SeqAIJ_inplace;
747db4efbfdSBarry Smith   }
748ad04f41aSHong Zhang   C->ops->solveadd           = MatSolveAdd_SeqAIJ_inplace;
749ad04f41aSHong Zhang   C->ops->solvetranspose     = MatSolveTranspose_SeqAIJ_inplace;
750ad04f41aSHong Zhang   C->ops->solvetransposeadd  = MatSolveTransposeAdd_SeqAIJ_inplace;
751ad04f41aSHong Zhang   C->ops->matsolve           = MatMatSolve_SeqAIJ_inplace;
752c456f294SBarry Smith   C->assembled    = PETSC_TRUE;
7535c9eb25fSBarry Smith   C->preallocated = PETSC_TRUE;
754d0f46423SBarry Smith   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
755d2276718SHong Zhang   if (sctx.nshift){
756f4db908eSBarry Smith      if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
7570481f469SBarry 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,sctx.shift_fraction,sctx.shift_top);CHKERRQ(ierr);
758f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
759e738e422SBarry Smith       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
7606cc28720Svictorle     }
7610697b6caSHong Zhang   }
762d3ac4fa3SBarry Smith   (C)->ops->solve            = MatSolve_SeqAIJ_inplace;
763d3ac4fa3SBarry Smith   (C)->ops->solvetranspose   = MatSolveTranspose_SeqAIJ_inplace;
764d3ac4fa3SBarry Smith   ierr = Mat_CheckInode(C,PETSC_FALSE);CHKERRQ(ierr);
7653a40ed3dSBarry Smith   PetscFunctionReturn(0);
766289bc588SBarry Smith }
7670889b5dcSvictorle 
768137fb511SHong Zhang /*
769137fb511SHong Zhang    This routine implements inplace ILU(0) with row or/and column permutations.
770137fb511SHong Zhang    Input:
771137fb511SHong Zhang      A - original matrix
772137fb511SHong Zhang    Output;
773137fb511SHong Zhang      A - a->i (rowptr) is same as original rowptr, but factored i-the row is stored in rowperm[i]
774137fb511SHong Zhang          a->j (col index) is permuted by the inverse of colperm, then sorted
775137fb511SHong Zhang          a->a reordered accordingly with a->j
776137fb511SHong Zhang          a->diag (ptr to diagonal elements) is updated.
777137fb511SHong Zhang */
778137fb511SHong Zhang #undef __FUNCT__
779137fb511SHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_InplaceWithPerm"
7800481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_SeqAIJ_InplaceWithPerm(Mat B,Mat A,const MatFactorInfo *info)
781137fb511SHong Zhang {
782b051339dSHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
783b051339dSHong Zhang   IS             isrow = a->row,isicol = a->icol;
784137fb511SHong Zhang   PetscErrorCode ierr;
7855d0c19d7SBarry Smith   const PetscInt *r,*ic,*ics;
7865d0c19d7SBarry Smith   PetscInt       i,j,n=A->rmap->n,*ai=a->i,*aj=a->j;
7875d0c19d7SBarry Smith   PetscInt       *ajtmp,nz,row;
788b051339dSHong Zhang   PetscInt       *diag = a->diag,nbdiag,*pj;
789a77337e4SBarry Smith   PetscScalar    *rtmp,*pc,multiplier,d;
790504a3fadSHong Zhang   MatScalar      *pv,*v;
791137fb511SHong Zhang   PetscReal      rs;
792d90ac83dSHong Zhang   FactorShiftCtx sctx;
793eca579c0SMatthew G Knepley   PetscInt       newshift = 0;
794504a3fadSHong Zhang   const  MatScalar *aa=a->a,*vtmp;
795137fb511SHong Zhang 
796137fb511SHong Zhang   PetscFunctionBegin;
797e32f2f54SBarry Smith   if (A != B) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"input and output matrix must have same address");
798504a3fadSHong Zhang 
799504a3fadSHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
800504a3fadSHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
801504a3fadSHong Zhang 
802504a3fadSHong Zhang   if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
803504a3fadSHong Zhang     const PetscInt   *ddiag = a->diag;
804504a3fadSHong Zhang     sctx.shift_top = info->zeropivot;
805504a3fadSHong Zhang     for (i=0; i<n; i++) {
806504a3fadSHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
807504a3fadSHong Zhang       d  = (aa)[ddiag[i]];
808504a3fadSHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
809504a3fadSHong Zhang       vtmp  = aa+ai[i];
810504a3fadSHong Zhang       nz = ai[i+1] - ai[i];
811504a3fadSHong Zhang       for (j=0; j<nz; j++)
812504a3fadSHong Zhang 	rs += PetscAbsScalar(vtmp[j]);
813504a3fadSHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
814504a3fadSHong Zhang     }
815504a3fadSHong Zhang     sctx.shift_top   *= 1.1;
816504a3fadSHong Zhang     sctx.nshift_max   = 5;
817504a3fadSHong Zhang     sctx.shift_lo     = 0.;
818504a3fadSHong Zhang     sctx.shift_hi     = 1.;
819504a3fadSHong Zhang   }
820504a3fadSHong Zhang 
821137fb511SHong Zhang   ierr  = ISGetIndices(isrow,&r);CHKERRQ(ierr);
822137fb511SHong Zhang   ierr  = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
823137fb511SHong Zhang   ierr  = PetscMalloc((n+1)*sizeof(PetscScalar),&rtmp);CHKERRQ(ierr);
824137fb511SHong Zhang   ierr  = PetscMemzero(rtmp,(n+1)*sizeof(PetscScalar));CHKERRQ(ierr);
825b051339dSHong Zhang   ics = ic;
826137fb511SHong Zhang 
827504a3fadSHong Zhang #if defined(MV)
82875567043SBarry Smith   sctx.shift_top      = 0.;
829e60cf9a0SBarry Smith   sctx.nshift_max     = 0;
83075567043SBarry Smith   sctx.shift_lo       = 0.;
83175567043SBarry Smith   sctx.shift_hi       = 0.;
83275567043SBarry Smith   sctx.shift_fraction = 0.;
833137fb511SHong Zhang 
834f4db908eSBarry Smith   if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
83575567043SBarry Smith     sctx.shift_top = 0.;
836137fb511SHong Zhang     for (i=0; i<n; i++) {
837137fb511SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
838b051339dSHong Zhang       d  = (a->a)[diag[i]];
839137fb511SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
840b051339dSHong Zhang       v  = a->a+ai[i];
841b051339dSHong Zhang       nz = ai[i+1] - ai[i];
842137fb511SHong Zhang       for (j=0; j<nz; j++)
843137fb511SHong Zhang 	rs += PetscAbsScalar(v[j]);
844137fb511SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
845137fb511SHong Zhang     }
846137fb511SHong Zhang     if (sctx.shift_top < info->zeropivot) sctx.shift_top = info->zeropivot;
847137fb511SHong Zhang     sctx.shift_top    *= 1.1;
848137fb511SHong Zhang     sctx.nshift_max   = 5;
849137fb511SHong Zhang     sctx.shift_lo     = 0.;
850137fb511SHong Zhang     sctx.shift_hi     = 1.;
851137fb511SHong Zhang   }
852137fb511SHong Zhang 
85375567043SBarry Smith   sctx.shift_amount = 0.;
854137fb511SHong Zhang   sctx.nshift       = 0;
855504a3fadSHong Zhang #endif
856504a3fadSHong Zhang 
857137fb511SHong Zhang   do {
858d90ac83dSHong Zhang     sctx.useshift = PETSC_FALSE;
859137fb511SHong Zhang     for (i=0; i<n; i++){
860b051339dSHong Zhang       /* load in initial unfactored row */
861b051339dSHong Zhang       nz    = ai[r[i]+1] - ai[r[i]];
862b051339dSHong Zhang       ajtmp = aj + ai[r[i]];
863b051339dSHong Zhang       v     = a->a + ai[r[i]];
864137fb511SHong Zhang       /* sort permuted ajtmp and values v accordingly */
865b051339dSHong Zhang       for (j=0; j<nz; j++) ajtmp[j] = ics[ajtmp[j]];
866137fb511SHong Zhang       ierr = PetscSortIntWithScalarArray(nz,ajtmp,v);CHKERRQ(ierr);
867137fb511SHong Zhang 
868b051339dSHong Zhang       diag[r[i]] = ai[r[i]];
869137fb511SHong Zhang       for (j=0; j<nz; j++) {
870137fb511SHong Zhang         rtmp[ajtmp[j]] = v[j];
871b051339dSHong Zhang         if (ajtmp[j] < i) diag[r[i]]++; /* update a->diag */
872137fb511SHong Zhang       }
873137fb511SHong Zhang       rtmp[r[i]] += sctx.shift_amount; /* shift the diagonal of the matrix */
874137fb511SHong Zhang 
875b051339dSHong Zhang       row = *ajtmp++;
876137fb511SHong Zhang       while  (row < i) {
877137fb511SHong Zhang         pc = rtmp + row;
878137fb511SHong Zhang         if (*pc != 0.0) {
879b051339dSHong Zhang           pv         = a->a + diag[r[row]];
880b051339dSHong Zhang           pj         = aj + diag[r[row]] + 1;
881137fb511SHong Zhang 
882137fb511SHong Zhang           multiplier = *pc / *pv++;
883137fb511SHong Zhang           *pc        = multiplier;
884b051339dSHong Zhang           nz         = ai[r[row]+1] - diag[r[row]] - 1;
885b051339dSHong Zhang           for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j];
886dc0b31edSSatish Balay           ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr);
887137fb511SHong Zhang         }
888b051339dSHong Zhang         row = *ajtmp++;
889137fb511SHong Zhang       }
890b051339dSHong Zhang       /* finished row so overwrite it onto a->a */
891b051339dSHong Zhang       pv   = a->a + ai[r[i]] ;
892b051339dSHong Zhang       pj   = aj + ai[r[i]] ;
893b051339dSHong Zhang       nz   = ai[r[i]+1] - ai[r[i]];
894b051339dSHong Zhang       nbdiag = diag[r[i]] - ai[r[i]]; /* num of entries before the diagonal */
895137fb511SHong Zhang 
896137fb511SHong Zhang       rs   = 0.0;
897137fb511SHong Zhang       for (j=0; j<nz; j++) {
898b051339dSHong Zhang         pv[j] = rtmp[pj[j]];
899b051339dSHong Zhang         if (j != nbdiag) rs += PetscAbsScalar(pv[j]);
900137fb511SHong Zhang       }
901137fb511SHong Zhang 
902137fb511SHong Zhang       sctx.rs  = rs;
903b051339dSHong Zhang       sctx.pv  = pv[nbdiag];
904504a3fadSHong Zhang       ierr = MatPivotCheck(info,&sctx,i,&newshift);CHKERRQ(ierr);
905137fb511SHong Zhang       if (newshift == 1) break;
906504a3fadSHong Zhang       pv[nbdiag] = sctx.pv;
907137fb511SHong Zhang     }
908137fb511SHong Zhang 
909f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.useshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) {
910137fb511SHong Zhang       /*
911137fb511SHong Zhang        * if no shift in this attempt & shifting & started shifting & can refine,
912137fb511SHong Zhang        * then try lower shift
913137fb511SHong Zhang        */
9140481f469SBarry Smith       sctx.shift_hi        = sctx.shift_fraction;
9150481f469SBarry Smith       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
9160481f469SBarry Smith       sctx.shift_amount    = sctx.shift_fraction * sctx.shift_top;
917d90ac83dSHong Zhang       sctx.useshift         = PETSC_TRUE;
918137fb511SHong Zhang       sctx.nshift++;
919137fb511SHong Zhang     }
920d90ac83dSHong Zhang   } while (sctx.useshift);
921137fb511SHong Zhang 
922137fb511SHong Zhang   /* invert diagonal entries for simplier triangular solves */
923137fb511SHong Zhang   for (i=0; i<n; i++) {
924b051339dSHong Zhang     a->a[diag[r[i]]] = 1.0/a->a[diag[r[i]]];
925137fb511SHong Zhang   }
926137fb511SHong Zhang 
927137fb511SHong Zhang   ierr = PetscFree(rtmp);CHKERRQ(ierr);
928137fb511SHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
929137fb511SHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
930b051339dSHong Zhang   A->ops->solve             = MatSolve_SeqAIJ_InplaceWithPerm;
931ad04f41aSHong Zhang   A->ops->solveadd          = MatSolveAdd_SeqAIJ_inplace;
932ad04f41aSHong Zhang   A->ops->solvetranspose    = MatSolveTranspose_SeqAIJ_inplace;
933ad04f41aSHong Zhang   A->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ_inplace;
934b051339dSHong Zhang   A->assembled = PETSC_TRUE;
9355c9eb25fSBarry Smith   A->preallocated = PETSC_TRUE;
936d0f46423SBarry Smith   ierr = PetscLogFlops(A->cmap->n);CHKERRQ(ierr);
937137fb511SHong Zhang   if (sctx.nshift){
938f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
9390481f469SBarry 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,sctx.shift_fraction,sctx.shift_top);CHKERRQ(ierr);
940f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
941e738e422SBarry Smith       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
942137fb511SHong Zhang     }
943137fb511SHong Zhang   }
944137fb511SHong Zhang   PetscFunctionReturn(0);
945137fb511SHong Zhang }
946137fb511SHong Zhang 
9470a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
9484a2ae208SSatish Balay #undef __FUNCT__
9494a2ae208SSatish Balay #define __FUNCT__ "MatLUFactor_SeqAIJ"
9500481f469SBarry Smith PetscErrorCode MatLUFactor_SeqAIJ(Mat A,IS row,IS col,const MatFactorInfo *info)
951da3a660dSBarry Smith {
952dfbe8321SBarry Smith   PetscErrorCode ierr;
953416022c9SBarry Smith   Mat            C;
954416022c9SBarry Smith 
9553a40ed3dSBarry Smith   PetscFunctionBegin;
95643244d56SBarry Smith   ierr = MatGetFactor(A,MAT_SOLVER_PETSC,MAT_FACTOR_LU,&C);CHKERRQ(ierr);
957719d5645SBarry Smith   ierr = MatLUFactorSymbolic(C,A,row,col,info);CHKERRQ(ierr);
958719d5645SBarry Smith   ierr = MatLUFactorNumeric(C,A,info);CHKERRQ(ierr);
959db4efbfdSBarry Smith   A->ops->solve            = C->ops->solve;
960db4efbfdSBarry Smith   A->ops->solvetranspose   = C->ops->solvetranspose;
961273d9f13SBarry Smith   ierr = MatHeaderCopy(A,C);CHKERRQ(ierr);
96252e6d16bSBarry Smith   ierr = PetscLogObjectParent(A,((Mat_SeqAIJ*)(A->data))->icol);CHKERRQ(ierr);
9633a40ed3dSBarry Smith   PetscFunctionReturn(0);
964da3a660dSBarry Smith }
9650a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
9661d098868SHong Zhang 
967889a8dedSBarry Smith 
9684a2ae208SSatish Balay #undef __FUNCT__
969ad04f41aSHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_inplace"
970ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_inplace(Mat A,Vec bb,Vec xx)
9718c37ef55SBarry Smith {
972416022c9SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
973416022c9SBarry Smith   IS                iscol = a->col,isrow = a->row;
9746849ba73SBarry Smith   PetscErrorCode    ierr;
9755d0c19d7SBarry Smith   PetscInt          i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
9765d0c19d7SBarry Smith   PetscInt          nz;
9775d0c19d7SBarry Smith   const PetscInt    *rout,*cout,*r,*c;
978dd6ea824SBarry Smith   PetscScalar       *x,*tmp,*tmps,sum;
979d9fead3dSBarry Smith   const PetscScalar *b;
980dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
9818c37ef55SBarry Smith 
9823a40ed3dSBarry Smith   PetscFunctionBegin;
9833a40ed3dSBarry Smith   if (!n) PetscFunctionReturn(0);
98491bf9adeSBarry Smith 
985d9fead3dSBarry Smith   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
9861ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
987416022c9SBarry Smith   tmp  = a->solve_work;
9888c37ef55SBarry Smith 
9893b2fbd54SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
9903b2fbd54SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
9918c37ef55SBarry Smith 
9928c37ef55SBarry Smith   /* forward solve the lower triangular */
9938c37ef55SBarry Smith   tmp[0] = b[*r++];
994010a20caSHong Zhang   tmps   = tmp;
9958c37ef55SBarry Smith   for (i=1; i<n; i++) {
996010a20caSHong Zhang     v   = aa + ai[i] ;
997010a20caSHong Zhang     vi  = aj + ai[i] ;
99853e7425aSBarry Smith     nz  = a->diag[i] - ai[i];
99953e7425aSBarry Smith     sum = b[*r++];
1000003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
10018c37ef55SBarry Smith     tmp[i] = sum;
10028c37ef55SBarry Smith   }
10038c37ef55SBarry Smith 
10048c37ef55SBarry Smith   /* backward solve the upper triangular */
10058c37ef55SBarry Smith   for (i=n-1; i>=0; i--){
1006010a20caSHong Zhang     v   = aa + a->diag[i] + 1;
1007010a20caSHong Zhang     vi  = aj + a->diag[i] + 1;
1008416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
10098c37ef55SBarry Smith     sum = tmp[i];
1010003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
1011010a20caSHong Zhang     x[*c--] = tmp[i] = sum*aa[a->diag[i]];
10128c37ef55SBarry Smith   }
10138c37ef55SBarry Smith 
10143b2fbd54SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
10153b2fbd54SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
1016d9fead3dSBarry Smith   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
10171ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1018dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
10193a40ed3dSBarry Smith   PetscFunctionReturn(0);
10208c37ef55SBarry Smith }
1021026e39d0SSatish Balay 
10222bebee5dSHong Zhang #undef __FUNCT__
1023ad04f41aSHong Zhang #define __FUNCT__ "MatMatSolve_SeqAIJ_inplace"
1024ad04f41aSHong Zhang PetscErrorCode MatMatSolve_SeqAIJ_inplace(Mat A,Mat B,Mat X)
10252bebee5dSHong Zhang {
10262bebee5dSHong Zhang   Mat_SeqAIJ      *a = (Mat_SeqAIJ*)A->data;
10272bebee5dSHong Zhang   IS              iscol = a->col,isrow = a->row;
10282bebee5dSHong Zhang   PetscErrorCode  ierr;
10295d0c19d7SBarry Smith   PetscInt        i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
10305d0c19d7SBarry Smith   PetscInt        nz,neq;
10315d0c19d7SBarry Smith   const PetscInt  *rout,*cout,*r,*c;
1032dd6ea824SBarry Smith   PetscScalar     *x,*b,*tmp,*tmps,sum;
1033dd6ea824SBarry Smith   const MatScalar *aa = a->a,*v;
1034db4efbfdSBarry Smith   PetscTruth      bisdense,xisdense;
10352bebee5dSHong Zhang 
10362bebee5dSHong Zhang   PetscFunctionBegin;
10372bebee5dSHong Zhang   if (!n) PetscFunctionReturn(0);
10382bebee5dSHong Zhang 
1039db4efbfdSBarry Smith   ierr = PetscTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr);
1040e32f2f54SBarry Smith   if (!bisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix");
1041db4efbfdSBarry Smith   ierr = PetscTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr);
1042e32f2f54SBarry Smith   if (!xisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix");
1043db4efbfdSBarry Smith 
10442bebee5dSHong Zhang   ierr = MatGetArray(B,&b);CHKERRQ(ierr);
10452bebee5dSHong Zhang   ierr = MatGetArray(X,&x);CHKERRQ(ierr);
10462bebee5dSHong Zhang 
10472bebee5dSHong Zhang   tmp  = a->solve_work;
10482bebee5dSHong Zhang   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
10492bebee5dSHong Zhang   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
10502bebee5dSHong Zhang 
1051a36b22ccSSatish Balay   for (neq=0; neq<B->cmap->n; neq++){
10522bebee5dSHong Zhang     /* forward solve the lower triangular */
10532bebee5dSHong Zhang     tmp[0] = b[r[0]];
10542bebee5dSHong Zhang     tmps   = tmp;
10552bebee5dSHong Zhang     for (i=1; i<n; i++) {
10562bebee5dSHong Zhang       v   = aa + ai[i] ;
10572bebee5dSHong Zhang       vi  = aj + ai[i] ;
10582bebee5dSHong Zhang       nz  = a->diag[i] - ai[i];
10592bebee5dSHong Zhang       sum = b[r[i]];
1060003131ecSBarry Smith       PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
10612bebee5dSHong Zhang       tmp[i] = sum;
10622bebee5dSHong Zhang     }
10632bebee5dSHong Zhang     /* backward solve the upper triangular */
10642bebee5dSHong Zhang     for (i=n-1; i>=0; i--){
10652bebee5dSHong Zhang       v   = aa + a->diag[i] + 1;
10662bebee5dSHong Zhang       vi  = aj + a->diag[i] + 1;
10672bebee5dSHong Zhang       nz  = ai[i+1] - a->diag[i] - 1;
10682bebee5dSHong Zhang       sum = tmp[i];
1069003131ecSBarry Smith       PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
10702bebee5dSHong Zhang       x[c[i]] = tmp[i] = sum*aa[a->diag[i]];
10712bebee5dSHong Zhang     }
10722bebee5dSHong Zhang 
10732bebee5dSHong Zhang     b += n;
10742bebee5dSHong Zhang     x += n;
10752bebee5dSHong Zhang   }
10762bebee5dSHong Zhang   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
10772bebee5dSHong Zhang   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
10782bebee5dSHong Zhang   ierr = MatRestoreArray(B,&b);CHKERRQ(ierr);
10792bebee5dSHong Zhang   ierr = MatRestoreArray(X,&x);CHKERRQ(ierr);
1080dc0b31edSSatish Balay   ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr);
10812bebee5dSHong Zhang   PetscFunctionReturn(0);
10822bebee5dSHong Zhang }
10832bebee5dSHong Zhang 
1084137fb511SHong Zhang #undef __FUNCT__
108535233d99SShri Abhyankar #define __FUNCT__ "MatMatSolve_SeqAIJ"
108635233d99SShri Abhyankar PetscErrorCode MatMatSolve_SeqAIJ(Mat A,Mat B,Mat X)
10879bd0847aSShri Abhyankar {
10889bd0847aSShri Abhyankar   Mat_SeqAIJ      *a = (Mat_SeqAIJ*)A->data;
10899bd0847aSShri Abhyankar   IS              iscol = a->col,isrow = a->row;
10909bd0847aSShri Abhyankar   PetscErrorCode  ierr;
10919bd0847aSShri Abhyankar   PetscInt        i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j,*adiag = a->diag;
10929bd0847aSShri Abhyankar   PetscInt        nz,neq;
10939bd0847aSShri Abhyankar   const PetscInt  *rout,*cout,*r,*c;
10949bd0847aSShri Abhyankar   PetscScalar     *x,*b,*tmp,sum;
10959bd0847aSShri Abhyankar   const MatScalar *aa = a->a,*v;
10969bd0847aSShri Abhyankar   PetscTruth      bisdense,xisdense;
10979bd0847aSShri Abhyankar 
10989bd0847aSShri Abhyankar   PetscFunctionBegin;
10999bd0847aSShri Abhyankar   if (!n) PetscFunctionReturn(0);
11009bd0847aSShri Abhyankar 
11019bd0847aSShri Abhyankar   ierr = PetscTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr);
1102e32f2f54SBarry Smith   if (!bisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix");
11039bd0847aSShri Abhyankar   ierr = PetscTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr);
1104e32f2f54SBarry Smith   if (!xisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix");
11059bd0847aSShri Abhyankar 
11069bd0847aSShri Abhyankar   ierr = MatGetArray(B,&b);CHKERRQ(ierr);
11079bd0847aSShri Abhyankar   ierr = MatGetArray(X,&x);CHKERRQ(ierr);
11089bd0847aSShri Abhyankar 
11099bd0847aSShri Abhyankar   tmp  = a->solve_work;
11109bd0847aSShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
11119bd0847aSShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
11129bd0847aSShri Abhyankar 
11139bd0847aSShri Abhyankar   for (neq=0; neq<B->cmap->n; neq++){
11149bd0847aSShri Abhyankar     /* forward solve the lower triangular */
11159bd0847aSShri Abhyankar     tmp[0] = b[r[0]];
11169bd0847aSShri Abhyankar     v      = aa;
11179bd0847aSShri Abhyankar     vi     = aj;
11189bd0847aSShri Abhyankar     for (i=1; i<n; i++) {
11199bd0847aSShri Abhyankar       nz  = ai[i+1] - ai[i];
11209bd0847aSShri Abhyankar       sum = b[r[i]];
11219bd0847aSShri Abhyankar       PetscSparseDenseMinusDot(sum,tmp,v,vi,nz);
11229bd0847aSShri Abhyankar       tmp[i] = sum;
11239bd0847aSShri Abhyankar       v += nz; vi += nz;
11249bd0847aSShri Abhyankar     }
11259bd0847aSShri Abhyankar 
11269bd0847aSShri Abhyankar     /* backward solve the upper triangular */
11279bd0847aSShri Abhyankar     for (i=n-1; i>=0; i--){
11289bd0847aSShri Abhyankar       v   = aa + adiag[i+1]+1;
11299bd0847aSShri Abhyankar       vi  = aj + adiag[i+1]+1;
11309bd0847aSShri Abhyankar       nz  = adiag[i]-adiag[i+1]-1;
11319bd0847aSShri Abhyankar       sum = tmp[i];
11329bd0847aSShri Abhyankar       PetscSparseDenseMinusDot(sum,tmp,v,vi,nz);
11339bd0847aSShri Abhyankar       x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */
11349bd0847aSShri Abhyankar     }
11359bd0847aSShri Abhyankar 
11369bd0847aSShri Abhyankar     b += n;
11379bd0847aSShri Abhyankar     x += n;
11389bd0847aSShri Abhyankar   }
11399bd0847aSShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
11409bd0847aSShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
11419bd0847aSShri Abhyankar   ierr = MatRestoreArray(B,&b);CHKERRQ(ierr);
11429bd0847aSShri Abhyankar   ierr = MatRestoreArray(X,&x);CHKERRQ(ierr);
11439bd0847aSShri Abhyankar   ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr);
11449bd0847aSShri Abhyankar   PetscFunctionReturn(0);
11459bd0847aSShri Abhyankar }
11469bd0847aSShri Abhyankar 
11479bd0847aSShri Abhyankar #undef __FUNCT__
1148137fb511SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_InplaceWithPerm"
1149137fb511SHong Zhang PetscErrorCode MatSolve_SeqAIJ_InplaceWithPerm(Mat A,Vec bb,Vec xx)
1150137fb511SHong Zhang {
1151137fb511SHong Zhang   Mat_SeqAIJ      *a = (Mat_SeqAIJ*)A->data;
1152137fb511SHong Zhang   IS              iscol = a->col,isrow = a->row;
1153137fb511SHong Zhang   PetscErrorCode  ierr;
11545d0c19d7SBarry Smith   const PetscInt  *r,*c,*rout,*cout;
11555d0c19d7SBarry Smith   PetscInt        i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
11565d0c19d7SBarry Smith   PetscInt        nz,row;
1157dd6ea824SBarry Smith   PetscScalar     *x,*b,*tmp,*tmps,sum;
1158dd6ea824SBarry Smith   const MatScalar *aa = a->a,*v;
1159137fb511SHong Zhang 
1160137fb511SHong Zhang   PetscFunctionBegin;
1161137fb511SHong Zhang   if (!n) PetscFunctionReturn(0);
1162137fb511SHong Zhang 
1163137fb511SHong Zhang   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
1164137fb511SHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1165137fb511SHong Zhang   tmp  = a->solve_work;
1166137fb511SHong Zhang 
1167137fb511SHong Zhang   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
1168137fb511SHong Zhang   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
1169137fb511SHong Zhang 
1170137fb511SHong Zhang   /* forward solve the lower triangular */
1171137fb511SHong Zhang   tmp[0] = b[*r++];
1172137fb511SHong Zhang   tmps   = tmp;
1173137fb511SHong Zhang   for (row=1; row<n; row++) {
1174137fb511SHong Zhang     i   = rout[row]; /* permuted row */
1175137fb511SHong Zhang     v   = aa + ai[i] ;
1176137fb511SHong Zhang     vi  = aj + ai[i] ;
1177137fb511SHong Zhang     nz  = a->diag[i] - ai[i];
1178137fb511SHong Zhang     sum = b[*r++];
1179003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
1180137fb511SHong Zhang     tmp[row] = sum;
1181137fb511SHong Zhang   }
1182137fb511SHong Zhang 
1183137fb511SHong Zhang   /* backward solve the upper triangular */
1184137fb511SHong Zhang   for (row=n-1; row>=0; row--){
1185137fb511SHong Zhang     i   = rout[row]; /* permuted row */
1186137fb511SHong Zhang     v   = aa + a->diag[i] + 1;
1187137fb511SHong Zhang     vi  = aj + a->diag[i] + 1;
1188137fb511SHong Zhang     nz  = ai[i+1] - a->diag[i] - 1;
1189137fb511SHong Zhang     sum = tmp[row];
1190003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
1191137fb511SHong Zhang     x[*c--] = tmp[row] = sum*aa[a->diag[i]];
1192137fb511SHong Zhang   }
1193137fb511SHong Zhang 
1194137fb511SHong Zhang   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
1195137fb511SHong Zhang   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
1196137fb511SHong Zhang   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
1197137fb511SHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1198dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
1199137fb511SHong Zhang   PetscFunctionReturn(0);
1200137fb511SHong Zhang }
1201137fb511SHong Zhang 
1202930ae53cSBarry Smith /* ----------------------------------------------------------- */
1203a4005a5dSBarry Smith #include "../src/mat/impls/aij/seq/ftn-kernels/fsolve.h"
12044a2ae208SSatish Balay #undef __FUNCT__
1205ad04f41aSHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering_inplace"
1206ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering_inplace(Mat A,Vec bb,Vec xx)
1207930ae53cSBarry Smith {
1208930ae53cSBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
12096849ba73SBarry Smith   PetscErrorCode    ierr;
1210d0f46423SBarry Smith   PetscInt          n = A->rmap->n;
1211b377110cSBarry Smith   const PetscInt    *ai = a->i,*aj = a->j,*adiag = a->diag;
1212356650c2SBarry Smith   PetscScalar       *x;
1213356650c2SBarry Smith   const PetscScalar *b;
1214dd6ea824SBarry Smith   const MatScalar   *aa = a->a;
1215aa482453SBarry Smith #if !defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ)
1216356650c2SBarry Smith   PetscInt          adiag_i,i,nz,ai_i;
1217b377110cSBarry Smith   const PetscInt    *vi;
1218dd6ea824SBarry Smith   const MatScalar   *v;
1219dd6ea824SBarry Smith   PetscScalar       sum;
1220d85afc42SSatish Balay #endif
1221930ae53cSBarry Smith 
12223a40ed3dSBarry Smith   PetscFunctionBegin;
12233a40ed3dSBarry Smith   if (!n) PetscFunctionReturn(0);
1224930ae53cSBarry Smith 
1225356650c2SBarry Smith   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
12261ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1227930ae53cSBarry Smith 
1228aa482453SBarry Smith #if defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ)
12291c4feecaSSatish Balay   fortransolveaij_(&n,x,ai,aj,adiag,aa,b);
12301c4feecaSSatish Balay #else
1231930ae53cSBarry Smith   /* forward solve the lower triangular */
1232930ae53cSBarry Smith   x[0] = b[0];
1233930ae53cSBarry Smith   for (i=1; i<n; i++) {
1234930ae53cSBarry Smith     ai_i = ai[i];
1235930ae53cSBarry Smith     v    = aa + ai_i;
1236930ae53cSBarry Smith     vi   = aj + ai_i;
1237930ae53cSBarry Smith     nz   = adiag[i] - ai_i;
1238930ae53cSBarry Smith     sum  = b[i];
1239003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
1240930ae53cSBarry Smith     x[i] = sum;
1241930ae53cSBarry Smith   }
1242930ae53cSBarry Smith 
1243930ae53cSBarry Smith   /* backward solve the upper triangular */
1244930ae53cSBarry Smith   for (i=n-1; i>=0; i--){
1245930ae53cSBarry Smith     adiag_i = adiag[i];
1246d708749eSBarry Smith     v       = aa + adiag_i + 1;
1247d708749eSBarry Smith     vi      = aj + adiag_i + 1;
1248930ae53cSBarry Smith     nz      = ai[i+1] - adiag_i - 1;
1249930ae53cSBarry Smith     sum     = x[i];
1250003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
1251930ae53cSBarry Smith     x[i]    = sum*aa[adiag_i];
1252930ae53cSBarry Smith   }
12531c4feecaSSatish Balay #endif
1254dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
1255356650c2SBarry Smith   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
12561ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
12573a40ed3dSBarry Smith   PetscFunctionReturn(0);
1258930ae53cSBarry Smith }
1259930ae53cSBarry Smith 
12604a2ae208SSatish Balay #undef __FUNCT__
1261ad04f41aSHong Zhang #define __FUNCT__ "MatSolveAdd_SeqAIJ_inplace"
1262ad04f41aSHong Zhang PetscErrorCode MatSolveAdd_SeqAIJ_inplace(Mat A,Vec bb,Vec yy,Vec xx)
1263da3a660dSBarry Smith {
1264416022c9SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
1265416022c9SBarry Smith   IS                iscol = a->col,isrow = a->row;
12666849ba73SBarry Smith   PetscErrorCode    ierr;
126704fbf559SBarry Smith   PetscInt          i, n = A->rmap->n,j;
12685d0c19d7SBarry Smith   PetscInt          nz;
126904fbf559SBarry Smith   const PetscInt    *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j;
127004fbf559SBarry Smith   PetscScalar       *x,*tmp,sum;
127104fbf559SBarry Smith   const PetscScalar *b;
1272dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
1273da3a660dSBarry Smith 
12743a40ed3dSBarry Smith   PetscFunctionBegin;
127578b31e54SBarry Smith   if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);}
1276da3a660dSBarry Smith 
127704fbf559SBarry Smith   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
12781ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1279416022c9SBarry Smith   tmp  = a->solve_work;
1280da3a660dSBarry Smith 
12813b2fbd54SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
12823b2fbd54SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
1283da3a660dSBarry Smith 
1284da3a660dSBarry Smith   /* forward solve the lower triangular */
1285da3a660dSBarry Smith   tmp[0] = b[*r++];
1286da3a660dSBarry Smith   for (i=1; i<n; i++) {
1287010a20caSHong Zhang     v   = aa + ai[i] ;
1288010a20caSHong Zhang     vi  = aj + ai[i] ;
1289416022c9SBarry Smith     nz  = a->diag[i] - ai[i];
1290da3a660dSBarry Smith     sum = b[*r++];
129104fbf559SBarry Smith     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
1292da3a660dSBarry Smith     tmp[i] = sum;
1293da3a660dSBarry Smith   }
1294da3a660dSBarry Smith 
1295da3a660dSBarry Smith   /* backward solve the upper triangular */
1296da3a660dSBarry Smith   for (i=n-1; i>=0; i--){
1297010a20caSHong Zhang     v   = aa + a->diag[i] + 1;
1298010a20caSHong Zhang     vi  = aj + a->diag[i] + 1;
1299416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
1300da3a660dSBarry Smith     sum = tmp[i];
130104fbf559SBarry Smith     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
1302010a20caSHong Zhang     tmp[i] = sum*aa[a->diag[i]];
1303da3a660dSBarry Smith     x[*c--] += tmp[i];
1304da3a660dSBarry Smith   }
1305da3a660dSBarry Smith 
13063b2fbd54SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
13073b2fbd54SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
130804fbf559SBarry Smith   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
13091ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1310dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
1311898183ecSLois Curfman McInnes 
13123a40ed3dSBarry Smith   PetscFunctionReturn(0);
1313da3a660dSBarry Smith }
131404fbf559SBarry Smith 
13154a2ae208SSatish Balay #undef __FUNCT__
131635233d99SShri Abhyankar #define __FUNCT__ "MatSolveAdd_SeqAIJ"
131735233d99SShri Abhyankar PetscErrorCode MatSolveAdd_SeqAIJ(Mat A,Vec bb,Vec yy,Vec xx)
13183c0119dfSShri Abhyankar {
13193c0119dfSShri Abhyankar   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
13203c0119dfSShri Abhyankar   IS                iscol = a->col,isrow = a->row;
13213c0119dfSShri Abhyankar   PetscErrorCode    ierr;
13223c0119dfSShri Abhyankar   PetscInt          i, n = A->rmap->n,j;
13233c0119dfSShri Abhyankar   PetscInt          nz;
13243c0119dfSShri Abhyankar   const PetscInt    *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j,*adiag = a->diag;
13253c0119dfSShri Abhyankar   PetscScalar       *x,*tmp,sum;
13263c0119dfSShri Abhyankar   const PetscScalar *b;
13273c0119dfSShri Abhyankar   const MatScalar   *aa = a->a,*v;
13283c0119dfSShri Abhyankar 
13293c0119dfSShri Abhyankar   PetscFunctionBegin;
13303c0119dfSShri Abhyankar   if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);}
13313c0119dfSShri Abhyankar 
13323c0119dfSShri Abhyankar   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
13333c0119dfSShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
13343c0119dfSShri Abhyankar   tmp  = a->solve_work;
13353c0119dfSShri Abhyankar 
13363c0119dfSShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
13373c0119dfSShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
13383c0119dfSShri Abhyankar 
13393c0119dfSShri Abhyankar   /* forward solve the lower triangular */
13403c0119dfSShri Abhyankar   tmp[0] = b[r[0]];
13413c0119dfSShri Abhyankar   v      = aa;
13423c0119dfSShri Abhyankar   vi     = aj;
13433c0119dfSShri Abhyankar   for (i=1; i<n; i++) {
13443c0119dfSShri Abhyankar     nz  = ai[i+1] - ai[i];
13453c0119dfSShri Abhyankar     sum = b[r[i]];
13463c0119dfSShri Abhyankar     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
13473c0119dfSShri Abhyankar     tmp[i] = sum;
13483c0119dfSShri Abhyankar     v += nz; vi += nz;
13493c0119dfSShri Abhyankar   }
13503c0119dfSShri Abhyankar 
13513c0119dfSShri Abhyankar   /* backward solve the upper triangular */
13523c0119dfSShri Abhyankar   v  = aa + adiag[n-1];
13533c0119dfSShri Abhyankar   vi = aj + adiag[n-1];
13543c0119dfSShri Abhyankar   for (i=n-1; i>=0; i--){
13553c0119dfSShri Abhyankar     nz  = adiag[i] - adiag[i+1] - 1;
13563c0119dfSShri Abhyankar     sum = tmp[i];
13573c0119dfSShri Abhyankar     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
13583c0119dfSShri Abhyankar     tmp[i] = sum*v[nz];
13593c0119dfSShri Abhyankar     x[c[i]] += tmp[i];
13603c0119dfSShri Abhyankar     v += nz+1; vi += nz+1;
13613c0119dfSShri Abhyankar   }
13623c0119dfSShri Abhyankar 
13633c0119dfSShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
13643c0119dfSShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
13653c0119dfSShri Abhyankar   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
13663c0119dfSShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
13673c0119dfSShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
13683c0119dfSShri Abhyankar 
13693c0119dfSShri Abhyankar   PetscFunctionReturn(0);
13703c0119dfSShri Abhyankar }
13713c0119dfSShri Abhyankar 
13723c0119dfSShri Abhyankar #undef __FUNCT__
1373ad04f41aSHong Zhang #define __FUNCT__ "MatSolveTranspose_SeqAIJ_inplace"
1374ad04f41aSHong Zhang PetscErrorCode MatSolveTranspose_SeqAIJ_inplace(Mat A,Vec bb,Vec xx)
1375da3a660dSBarry Smith {
1376416022c9SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
13772235e667SBarry Smith   IS                iscol = a->col,isrow = a->row;
13786849ba73SBarry Smith   PetscErrorCode    ierr;
137904fbf559SBarry Smith   const PetscInt    *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi;
138004fbf559SBarry Smith   PetscInt          i,n = A->rmap->n,j;
138104fbf559SBarry Smith   PetscInt          nz;
138204fbf559SBarry Smith   PetscScalar       *x,*tmp,s1;
1383dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
138404fbf559SBarry Smith   const PetscScalar *b;
1385da3a660dSBarry Smith 
13863a40ed3dSBarry Smith   PetscFunctionBegin;
138704fbf559SBarry Smith   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
13881ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1389416022c9SBarry Smith   tmp  = a->solve_work;
1390da3a660dSBarry Smith 
13912235e667SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
13922235e667SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
1393da3a660dSBarry Smith 
1394da3a660dSBarry Smith   /* copy the b into temp work space according to permutation */
13952235e667SBarry Smith   for (i=0; i<n; i++) tmp[i] = b[c[i]];
1396da3a660dSBarry Smith 
1397da3a660dSBarry Smith   /* forward solve the U^T */
1398da3a660dSBarry Smith   for (i=0; i<n; i++) {
1399010a20caSHong Zhang     v   = aa + diag[i] ;
1400010a20caSHong Zhang     vi  = aj + diag[i] + 1;
1401f1af5d2fSBarry Smith     nz  = ai[i+1] - diag[i] - 1;
1402c38d4ed2SBarry Smith     s1  = tmp[i];
1403ef66eb69SBarry Smith     s1 *= (*v++);  /* multiply by inverse of diagonal entry */
140404fbf559SBarry Smith     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1405c38d4ed2SBarry Smith     tmp[i] = s1;
1406da3a660dSBarry Smith   }
1407da3a660dSBarry Smith 
1408da3a660dSBarry Smith   /* backward solve the L^T */
1409da3a660dSBarry Smith   for (i=n-1; i>=0; i--){
1410010a20caSHong Zhang     v   = aa + diag[i] - 1 ;
1411010a20caSHong Zhang     vi  = aj + diag[i] - 1 ;
1412f1af5d2fSBarry Smith     nz  = diag[i] - ai[i];
1413f1af5d2fSBarry Smith     s1  = tmp[i];
141404fbf559SBarry Smith     for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j];
1415da3a660dSBarry Smith   }
1416da3a660dSBarry Smith 
1417da3a660dSBarry Smith   /* copy tmp into x according to permutation */
1418591294e4SBarry Smith   for (i=0; i<n; i++) x[r[i]] = tmp[i];
1419da3a660dSBarry Smith 
14202235e667SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
14212235e667SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
142204fbf559SBarry Smith   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
14231ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1424da3a660dSBarry Smith 
1425dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr);
14263a40ed3dSBarry Smith   PetscFunctionReturn(0);
1427da3a660dSBarry Smith }
1428da3a660dSBarry Smith 
1429d1fa9404SShri Abhyankar #undef __FUNCT__
143035233d99SShri Abhyankar #define __FUNCT__ "MatSolveTranspose_SeqAIJ"
143135233d99SShri Abhyankar PetscErrorCode MatSolveTranspose_SeqAIJ(Mat A,Vec bb,Vec xx)
1432d1fa9404SShri Abhyankar {
1433d1fa9404SShri Abhyankar   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
1434d1fa9404SShri Abhyankar   IS                iscol = a->col,isrow = a->row;
1435d1fa9404SShri Abhyankar   PetscErrorCode    ierr;
1436d1fa9404SShri Abhyankar   const PetscInt    *rout,*cout,*r,*c,*adiag = a->diag,*ai = a->i,*aj = a->j,*vi;
1437d1fa9404SShri Abhyankar   PetscInt          i,n = A->rmap->n,j;
1438d1fa9404SShri Abhyankar   PetscInt          nz;
1439d1fa9404SShri Abhyankar   PetscScalar       *x,*tmp,s1;
1440d1fa9404SShri Abhyankar   const MatScalar   *aa = a->a,*v;
1441d1fa9404SShri Abhyankar   const PetscScalar *b;
1442d1fa9404SShri Abhyankar 
1443d1fa9404SShri Abhyankar   PetscFunctionBegin;
1444d1fa9404SShri Abhyankar   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
1445d1fa9404SShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1446d1fa9404SShri Abhyankar   tmp  = a->solve_work;
1447d1fa9404SShri Abhyankar 
1448d1fa9404SShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
1449d1fa9404SShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
1450d1fa9404SShri Abhyankar 
1451d1fa9404SShri Abhyankar   /* copy the b into temp work space according to permutation */
1452d1fa9404SShri Abhyankar   for (i=0; i<n; i++) tmp[i] = b[c[i]];
1453d1fa9404SShri Abhyankar 
1454d1fa9404SShri Abhyankar   /* forward solve the U^T */
1455d1fa9404SShri Abhyankar   for (i=0; i<n; i++) {
1456d1fa9404SShri Abhyankar     v   = aa + adiag[i+1] + 1;
1457d1fa9404SShri Abhyankar     vi  = aj + adiag[i+1] + 1;
1458d1fa9404SShri Abhyankar     nz  = adiag[i] - adiag[i+1] - 1;
1459d1fa9404SShri Abhyankar     s1  = tmp[i];
1460d1fa9404SShri Abhyankar     s1 *= v[nz];  /* multiply by inverse of diagonal entry */
1461d1fa9404SShri Abhyankar     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1462d1fa9404SShri Abhyankar     tmp[i] = s1;
1463d1fa9404SShri Abhyankar   }
1464d1fa9404SShri Abhyankar 
1465d1fa9404SShri Abhyankar   /* backward solve the L^T */
1466d1fa9404SShri Abhyankar   for (i=n-1; i>=0; i--){
1467d1fa9404SShri Abhyankar     v   = aa + ai[i];
1468d1fa9404SShri Abhyankar     vi  = aj + ai[i];
1469d1fa9404SShri Abhyankar     nz  = ai[i+1] - ai[i];
1470d1fa9404SShri Abhyankar     s1  = tmp[i];
1471d1fa9404SShri Abhyankar     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1472d1fa9404SShri Abhyankar   }
1473d1fa9404SShri Abhyankar 
1474d1fa9404SShri Abhyankar   /* copy tmp into x according to permutation */
1475d1fa9404SShri Abhyankar   for (i=0; i<n; i++) x[r[i]] = tmp[i];
1476d1fa9404SShri Abhyankar 
1477d1fa9404SShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
1478d1fa9404SShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
1479d1fa9404SShri Abhyankar   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
1480d1fa9404SShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1481d1fa9404SShri Abhyankar 
1482d1fa9404SShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr);
1483d1fa9404SShri Abhyankar   PetscFunctionReturn(0);
1484d1fa9404SShri Abhyankar }
1485d1fa9404SShri Abhyankar 
14864a2ae208SSatish Balay #undef __FUNCT__
1487ad04f41aSHong Zhang #define __FUNCT__ "MatSolveTransposeAdd_SeqAIJ_inplace"
1488ad04f41aSHong Zhang PetscErrorCode MatSolveTransposeAdd_SeqAIJ_inplace(Mat A,Vec bb,Vec zz,Vec xx)
1489da3a660dSBarry Smith {
1490416022c9SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
14912235e667SBarry Smith   IS                iscol = a->col,isrow = a->row;
14926849ba73SBarry Smith   PetscErrorCode    ierr;
149304fbf559SBarry Smith   const PetscInt    *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi;
149404fbf559SBarry Smith   PetscInt          i,n = A->rmap->n,j;
149504fbf559SBarry Smith   PetscInt          nz;
149604fbf559SBarry Smith   PetscScalar       *x,*tmp,s1;
1497dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
149804fbf559SBarry Smith   const PetscScalar *b;
14996abc6512SBarry Smith 
15003a40ed3dSBarry Smith   PetscFunctionBegin;
150123bc6035SMatthew Knepley   if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);}
150204fbf559SBarry Smith   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
15031ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1504416022c9SBarry Smith   tmp  = a->solve_work;
15056abc6512SBarry Smith 
15062235e667SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
15072235e667SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
15086abc6512SBarry Smith 
15096abc6512SBarry Smith   /* copy the b into temp work space according to permutation */
15102235e667SBarry Smith   for (i=0; i<n; i++) tmp[i] = b[c[i]];
15116abc6512SBarry Smith 
15126abc6512SBarry Smith   /* forward solve the U^T */
15136abc6512SBarry Smith   for (i=0; i<n; i++) {
1514010a20caSHong Zhang     v   = aa + diag[i] ;
1515010a20caSHong Zhang     vi  = aj + diag[i] + 1;
1516f1af5d2fSBarry Smith     nz  = ai[i+1] - diag[i] - 1;
151704fbf559SBarry Smith     s1  = tmp[i];
151804fbf559SBarry Smith     s1 *= (*v++);  /* multiply by inverse of diagonal entry */
151904fbf559SBarry Smith     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
152004fbf559SBarry Smith     tmp[i] = s1;
15216abc6512SBarry Smith   }
15226abc6512SBarry Smith 
15236abc6512SBarry Smith   /* backward solve the L^T */
15246abc6512SBarry Smith   for (i=n-1; i>=0; i--){
1525010a20caSHong Zhang     v   = aa + diag[i] - 1 ;
1526010a20caSHong Zhang     vi  = aj + diag[i] - 1 ;
1527f1af5d2fSBarry Smith     nz  = diag[i] - ai[i];
152804fbf559SBarry Smith     s1  = tmp[i];
152904fbf559SBarry Smith     for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j];
15306abc6512SBarry Smith   }
15316abc6512SBarry Smith 
15326abc6512SBarry Smith   /* copy tmp into x according to permutation */
15336abc6512SBarry Smith   for (i=0; i<n; i++) x[r[i]] += tmp[i];
15346abc6512SBarry Smith 
15352235e667SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
15362235e667SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
153704fbf559SBarry Smith   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
15381ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
15396abc6512SBarry Smith 
154004fbf559SBarry Smith   ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr);
15413a40ed3dSBarry Smith   PetscFunctionReturn(0);
1542da3a660dSBarry Smith }
154304fbf559SBarry Smith 
1544533e6011SShri Abhyankar #undef __FUNCT__
154535233d99SShri Abhyankar #define __FUNCT__ "MatSolveTransposeAdd_SeqAIJ"
154635233d99SShri Abhyankar PetscErrorCode MatSolveTransposeAdd_SeqAIJ(Mat A,Vec bb,Vec zz,Vec xx)
1547533e6011SShri Abhyankar {
1548533e6011SShri Abhyankar   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
1549533e6011SShri Abhyankar   IS                iscol = a->col,isrow = a->row;
1550533e6011SShri Abhyankar   PetscErrorCode    ierr;
1551533e6011SShri Abhyankar   const PetscInt    *rout,*cout,*r,*c,*adiag = a->diag,*ai = a->i,*aj = a->j,*vi;
1552533e6011SShri Abhyankar   PetscInt          i,n = A->rmap->n,j;
1553533e6011SShri Abhyankar   PetscInt          nz;
1554533e6011SShri Abhyankar   PetscScalar       *x,*tmp,s1;
1555533e6011SShri Abhyankar   const MatScalar   *aa = a->a,*v;
1556533e6011SShri Abhyankar   const PetscScalar *b;
1557533e6011SShri Abhyankar 
1558533e6011SShri Abhyankar   PetscFunctionBegin;
1559533e6011SShri Abhyankar   if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);}
1560533e6011SShri Abhyankar   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
1561533e6011SShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1562533e6011SShri Abhyankar   tmp  = a->solve_work;
1563533e6011SShri Abhyankar 
1564533e6011SShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
1565533e6011SShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
1566533e6011SShri Abhyankar 
1567533e6011SShri Abhyankar   /* copy the b into temp work space according to permutation */
1568533e6011SShri Abhyankar   for (i=0; i<n; i++) tmp[i] = b[c[i]];
1569533e6011SShri Abhyankar 
1570533e6011SShri Abhyankar   /* forward solve the U^T */
1571533e6011SShri Abhyankar   for (i=0; i<n; i++) {
1572533e6011SShri Abhyankar     v   = aa + adiag[i+1] + 1;
1573533e6011SShri Abhyankar     vi  = aj + adiag[i+1] + 1;
1574533e6011SShri Abhyankar     nz  = adiag[i] - adiag[i+1] - 1;
1575533e6011SShri Abhyankar     s1  = tmp[i];
1576533e6011SShri Abhyankar     s1 *= v[nz];  /* multiply by inverse of diagonal entry */
1577533e6011SShri Abhyankar     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1578533e6011SShri Abhyankar     tmp[i] = s1;
1579533e6011SShri Abhyankar   }
1580533e6011SShri Abhyankar 
1581533e6011SShri Abhyankar 
1582533e6011SShri Abhyankar   /* backward solve the L^T */
1583533e6011SShri Abhyankar   for (i=n-1; i>=0; i--){
1584533e6011SShri Abhyankar     v   = aa + ai[i] ;
1585533e6011SShri Abhyankar     vi  = aj + ai[i];
1586533e6011SShri Abhyankar     nz  = ai[i+1] - ai[i];
1587533e6011SShri Abhyankar     s1  = tmp[i];
1588c5e3b2a3SShri Abhyankar     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1589533e6011SShri Abhyankar   }
1590533e6011SShri Abhyankar 
1591533e6011SShri Abhyankar   /* copy tmp into x according to permutation */
1592533e6011SShri Abhyankar   for (i=0; i<n; i++) x[r[i]] += tmp[i];
1593533e6011SShri Abhyankar 
1594533e6011SShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
1595533e6011SShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
1596533e6011SShri Abhyankar   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
1597533e6011SShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1598533e6011SShri Abhyankar 
1599533e6011SShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr);
1600533e6011SShri Abhyankar   PetscFunctionReturn(0);
1601533e6011SShri Abhyankar }
1602533e6011SShri Abhyankar 
16039e25ed09SBarry Smith /* ----------------------------------------------------------------*/
1604b47f5a65SHong Zhang 
1605f77e22a1SHong Zhang EXTERN PetscErrorCode MatDuplicateNoCreate_SeqAIJ(Mat,Mat,MatDuplicateOption,PetscTruth);
160608480c60SBarry Smith 
16078fc3a347SHong Zhang /*
16088a76ed4fSHong Zhang    ilu() under revised new data structure.
1609813a1f2bSShri Abhyankar    Factored arrays bj and ba are stored as
1610813a1f2bSShri Abhyankar      L(0,:), L(1,:), ...,L(n-1,:),  U(n-1,:),...,U(i,:),U(i-1,:),...,U(0,:)
1611813a1f2bSShri Abhyankar 
1612813a1f2bSShri Abhyankar    bi=fact->i is an array of size n+1, in which
1613813a1f2bSShri Abhyankar    bi+
1614baabb860SHong Zhang      bi[i]:  points to 1st entry of L(i,:),i=0,...,n-1
1615baabb860SHong Zhang      bi[n]:  points to L(n-1,n-1)+1
1616baabb860SHong Zhang 
1617813a1f2bSShri Abhyankar   bdiag=fact->diag is an array of size n+1,in which
1618baabb860SHong Zhang      bdiag[i]: points to diagonal of U(i,:), i=0,...,n-1
1619a24f213cSHong Zhang      bdiag[n]: points to entry of U(n-1,0)-1
1620813a1f2bSShri Abhyankar 
1621813a1f2bSShri Abhyankar    U(i,:) contains bdiag[i] as its last entry, i.e.,
1622813a1f2bSShri Abhyankar     U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i])
1623813a1f2bSShri Abhyankar */
1624813a1f2bSShri Abhyankar #undef __FUNCT__
162535233d99SShri Abhyankar #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_ilu0"
162635233d99SShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ_ilu0(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
1627813a1f2bSShri Abhyankar {
1628813a1f2bSShri Abhyankar 
1629813a1f2bSShri Abhyankar   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
1630813a1f2bSShri Abhyankar   PetscErrorCode     ierr;
1631bbd65245SShri Abhyankar   const PetscInt     n=A->rmap->n,*ai=a->i,*aj,*adiag=a->diag;
1632bbd65245SShri Abhyankar   PetscInt           i,j,k=0,nz,*bi,*bj,*bdiag;
16331df811f5SHong Zhang   PetscTruth         missing;
16341df811f5SHong Zhang   IS                 isicol;
1635813a1f2bSShri Abhyankar 
1636813a1f2bSShri Abhyankar   PetscFunctionBegin;
1637e32f2f54SBarry Smith   if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n);
16381df811f5SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr);
1639e32f2f54SBarry Smith   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i);
16401df811f5SHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
16411df811f5SHong Zhang 
1642813a1f2bSShri Abhyankar   ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_FALSE);CHKERRQ(ierr);
1643813a1f2bSShri Abhyankar   b    = (Mat_SeqAIJ*)(fact)->data;
1644813a1f2bSShri Abhyankar 
1645813a1f2bSShri Abhyankar   /* allocate matrix arrays for new data structure */
1646813a1f2bSShri Abhyankar   ierr = PetscMalloc3(ai[n]+1,PetscScalar,&b->a,ai[n]+1,PetscInt,&b->j,n+1,PetscInt,&b->i);CHKERRQ(ierr);
1647813a1f2bSShri Abhyankar   ierr = PetscLogObjectMemory(fact,ai[n]*(sizeof(PetscScalar)+sizeof(PetscInt))+(n+1)*sizeof(PetscInt));CHKERRQ(ierr);
1648813a1f2bSShri Abhyankar   b->singlemalloc = PETSC_TRUE;
1649813a1f2bSShri Abhyankar   if (!b->diag){
1650813a1f2bSShri Abhyankar     ierr = PetscMalloc((n+1)*sizeof(PetscInt),&b->diag);CHKERRQ(ierr);
1651813a1f2bSShri Abhyankar     ierr = PetscLogObjectMemory(fact,(n+1)*sizeof(PetscInt));CHKERRQ(ierr);
1652813a1f2bSShri Abhyankar   }
1653813a1f2bSShri Abhyankar   bdiag = b->diag;
1654813a1f2bSShri Abhyankar 
1655813a1f2bSShri Abhyankar   if (n > 0) {
1656813a1f2bSShri Abhyankar     ierr = PetscMemzero(b->a,(ai[n])*sizeof(MatScalar));CHKERRQ(ierr);
1657813a1f2bSShri Abhyankar   }
1658813a1f2bSShri Abhyankar 
1659813a1f2bSShri Abhyankar   /* set bi and bj with new data structure */
1660813a1f2bSShri Abhyankar   bi = b->i;
1661813a1f2bSShri Abhyankar   bj = b->j;
1662813a1f2bSShri Abhyankar 
1663813a1f2bSShri Abhyankar   /* L part */
1664e60cf9a0SBarry Smith   bi[0] = 0;
1665813a1f2bSShri Abhyankar   for (i=0; i<n; i++){
1666813a1f2bSShri Abhyankar     nz = adiag[i] - ai[i];
1667813a1f2bSShri Abhyankar     bi[i+1] = bi[i] + nz;
1668813a1f2bSShri Abhyankar     aj = a->j + ai[i];
1669813a1f2bSShri Abhyankar     for (j=0; j<nz; j++){
1670bbd65245SShri Abhyankar       /*   *bj = aj[j]; bj++; */
1671bbd65245SShri Abhyankar       bj[k++] = aj[j];
1672813a1f2bSShri Abhyankar     }
1673813a1f2bSShri Abhyankar   }
1674813a1f2bSShri Abhyankar 
1675813a1f2bSShri Abhyankar   /* U part */
167662fbd6afSShri Abhyankar   bdiag[n] = bi[n]-1;
1677813a1f2bSShri Abhyankar   for (i=n-1; i>=0; i--){
1678813a1f2bSShri Abhyankar     nz = ai[i+1] - adiag[i] - 1;
1679813a1f2bSShri Abhyankar     aj = a->j + adiag[i] + 1;
1680813a1f2bSShri Abhyankar     for (j=0; j<nz; j++){
1681bbd65245SShri Abhyankar       /*      *bj = aj[j]; bj++; */
1682bbd65245SShri Abhyankar       bj[k++] = aj[j];
1683813a1f2bSShri Abhyankar     }
1684813a1f2bSShri Abhyankar     /* diag[i] */
1685bbd65245SShri Abhyankar     /*    *bj = i; bj++; */
1686bbd65245SShri Abhyankar     bj[k++] = i;
1687a24f213cSHong Zhang     bdiag[i] = bdiag[i+1] + nz + 1;
1688813a1f2bSShri Abhyankar   }
16891df811f5SHong Zhang 
1690d5f3da31SBarry Smith   fact->factortype             = MAT_FACTOR_ILU;
16911df811f5SHong Zhang   fact->info.factor_mallocs    = 0;
16921df811f5SHong Zhang   fact->info.fill_ratio_given  = info->fill;
16931df811f5SHong Zhang   fact->info.fill_ratio_needed = 1.0;
169435233d99SShri Abhyankar   fact->ops->lufactornumeric   = MatLUFactorNumeric_SeqAIJ;
16951df811f5SHong Zhang 
16961df811f5SHong Zhang   b       = (Mat_SeqAIJ*)(fact)->data;
16971df811f5SHong Zhang   b->row  = isrow;
16981df811f5SHong Zhang   b->col  = iscol;
16991df811f5SHong Zhang   b->icol = isicol;
17001df811f5SHong Zhang   ierr    = PetscMalloc((fact->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
17011df811f5SHong Zhang   ierr    = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
17021df811f5SHong Zhang   ierr    = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
1703813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
1704813a1f2bSShri Abhyankar }
1705813a1f2bSShri Abhyankar 
1706813a1f2bSShri Abhyankar #undef __FUNCT__
170735233d99SShri Abhyankar #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ"
170835233d99SShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
1709813a1f2bSShri Abhyankar {
1710813a1f2bSShri Abhyankar   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
1711813a1f2bSShri Abhyankar   IS                 isicol;
1712813a1f2bSShri Abhyankar   PetscErrorCode     ierr;
1713813a1f2bSShri Abhyankar   const PetscInt     *r,*ic;
17141df811f5SHong Zhang   PetscInt           n=A->rmap->n,*ai=a->i,*aj=a->j;
1715813a1f2bSShri Abhyankar   PetscInt           *bi,*cols,nnz,*cols_lvl;
1716813a1f2bSShri Abhyankar   PetscInt           *bdiag,prow,fm,nzbd,reallocs=0,dcount=0;
1717813a1f2bSShri Abhyankar   PetscInt           i,levels,diagonal_fill;
1718813a1f2bSShri Abhyankar   PetscTruth         col_identity,row_identity;
1719813a1f2bSShri Abhyankar   PetscReal          f;
1720813a1f2bSShri Abhyankar   PetscInt           nlnk,*lnk,*lnk_lvl=PETSC_NULL;
1721813a1f2bSShri Abhyankar   PetscBT            lnkbt;
1722813a1f2bSShri Abhyankar   PetscInt           nzi,*bj,**bj_ptr,**bjlvl_ptr;
1723813a1f2bSShri Abhyankar   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
1724813a1f2bSShri Abhyankar   PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL;
1725813a1f2bSShri Abhyankar 
1726813a1f2bSShri Abhyankar   PetscFunctionBegin;
17273898ab1dSSatish Balay   /* Uncomment the old data struct part only while testing new data structure for MatSolve() */
1728124a3351SHong Zhang   /*
1729124a3351SHong Zhang   PetscTruth         olddatastruct=PETSC_FALSE;
173035233d99SShri Abhyankar   ierr = PetscOptionsGetTruth(PETSC_NULL,"-ilu_old",&olddatastruct,PETSC_NULL);CHKERRQ(ierr);
173135233d99SShri Abhyankar   if(olddatastruct){
1732ad04f41aSHong Zhang     ierr = MatILUFactorSymbolic_SeqAIJ_inplace(fact,A,isrow,iscol,info);CHKERRQ(ierr);
1733ad04f41aSHong Zhang     PetscFunctionReturn(0);
1734ad04f41aSHong Zhang   }
1735124a3351SHong Zhang   */
1736ad04f41aSHong Zhang 
1737813a1f2bSShri Abhyankar   levels = (PetscInt)info->levels;
1738813a1f2bSShri Abhyankar   ierr   = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
1739813a1f2bSShri Abhyankar   ierr   = ISIdentity(iscol,&col_identity);CHKERRQ(ierr);
1740813a1f2bSShri Abhyankar 
1741813a1f2bSShri Abhyankar   if (!levels && row_identity && col_identity) {
1742813a1f2bSShri Abhyankar     /* special case: ilu(0) with natural ordering */
174335233d99SShri Abhyankar     ierr = MatILUFactorSymbolic_SeqAIJ_ilu0(fact,A,isrow,iscol,info);CHKERRQ(ierr);
1744124a3351SHong Zhang     if (a->inode.size) {
1745124a3351SHong Zhang       fact->ops->lufactornumeric  = MatLUFactorNumeric_SeqAIJ_Inode;
1746124a3351SHong Zhang     }
1747813a1f2bSShri Abhyankar     PetscFunctionReturn(0);
1748813a1f2bSShri Abhyankar   }
1749813a1f2bSShri Abhyankar 
1750e32f2f54SBarry Smith   if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n);
17511df811f5SHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
1752813a1f2bSShri Abhyankar   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
1753813a1f2bSShri Abhyankar   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
1754813a1f2bSShri Abhyankar 
17551bfa06eaSJed Brown   /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */
17561bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
17571bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
1758e60cf9a0SBarry Smith   bi[0] = bdiag[0] = 0;
1759813a1f2bSShri Abhyankar 
17600e83c824SBarry Smith   ierr = PetscMalloc2(n,PetscInt*,&bj_ptr,n,PetscInt*,&bjlvl_ptr);CHKERRQ(ierr);
1761813a1f2bSShri Abhyankar 
1762813a1f2bSShri Abhyankar   /* create a linked list for storing column indices of the active row */
1763813a1f2bSShri Abhyankar   nlnk = n + 1;
1764813a1f2bSShri Abhyankar   ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
1765813a1f2bSShri Abhyankar 
1766813a1f2bSShri Abhyankar   /* initial FreeSpace size is f*(ai[n]+1) */
17671df811f5SHong Zhang   f             = info->fill;
17681df811f5SHong Zhang   diagonal_fill = (PetscInt)info->diagonal_fill;
1769813a1f2bSShri Abhyankar   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr);
1770813a1f2bSShri Abhyankar   current_space = free_space;
1771813a1f2bSShri Abhyankar   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr);
1772813a1f2bSShri Abhyankar   current_space_lvl = free_space_lvl;
1773813a1f2bSShri Abhyankar 
1774813a1f2bSShri Abhyankar   for (i=0; i<n; i++) {
1775813a1f2bSShri Abhyankar     nzi = 0;
1776813a1f2bSShri Abhyankar     /* copy current row into linked list */
1777813a1f2bSShri Abhyankar     nnz  = ai[r[i]+1] - ai[r[i]];
1778e32f2f54SBarry Smith     if (!nnz) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i);
1779813a1f2bSShri Abhyankar     cols = aj + ai[r[i]];
1780813a1f2bSShri Abhyankar     lnk[i] = -1; /* marker to indicate if diagonal exists */
1781813a1f2bSShri Abhyankar     ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
1782813a1f2bSShri Abhyankar     nzi += nlnk;
1783813a1f2bSShri Abhyankar 
1784813a1f2bSShri Abhyankar     /* make sure diagonal entry is included */
1785813a1f2bSShri Abhyankar     if (diagonal_fill && lnk[i] == -1) {
1786813a1f2bSShri Abhyankar       fm = n;
1787813a1f2bSShri Abhyankar       while (lnk[fm] < i) fm = lnk[fm];
1788813a1f2bSShri Abhyankar       lnk[i]     = lnk[fm]; /* insert diagonal into linked list */
1789813a1f2bSShri Abhyankar       lnk[fm]    = i;
1790813a1f2bSShri Abhyankar       lnk_lvl[i] = 0;
1791813a1f2bSShri Abhyankar       nzi++; dcount++;
1792813a1f2bSShri Abhyankar     }
1793813a1f2bSShri Abhyankar 
1794813a1f2bSShri Abhyankar     /* add pivot rows into the active row */
1795813a1f2bSShri Abhyankar     nzbd = 0;
1796813a1f2bSShri Abhyankar     prow = lnk[n];
1797813a1f2bSShri Abhyankar     while (prow < i) {
1798813a1f2bSShri Abhyankar       nnz      = bdiag[prow];
1799813a1f2bSShri Abhyankar       cols     = bj_ptr[prow] + nnz + 1;
1800813a1f2bSShri Abhyankar       cols_lvl = bjlvl_ptr[prow] + nnz + 1;
1801813a1f2bSShri Abhyankar       nnz      = bi[prow+1] - bi[prow] - nnz - 1;
1802813a1f2bSShri Abhyankar       ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr);
1803813a1f2bSShri Abhyankar       nzi += nlnk;
1804813a1f2bSShri Abhyankar       prow = lnk[prow];
1805813a1f2bSShri Abhyankar       nzbd++;
1806813a1f2bSShri Abhyankar     }
1807813a1f2bSShri Abhyankar     bdiag[i] = nzbd;
1808813a1f2bSShri Abhyankar     bi[i+1]  = bi[i] + nzi;
1809813a1f2bSShri Abhyankar 
1810813a1f2bSShri Abhyankar     /* if free space is not available, make more free space */
1811813a1f2bSShri Abhyankar     if (current_space->local_remaining<nzi) {
1812813a1f2bSShri Abhyankar       nnz = 2*nzi*(n - i); /* estimated and max additional space needed */
1813813a1f2bSShri Abhyankar       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
1814813a1f2bSShri Abhyankar       ierr = PetscFreeSpaceGet(nnz,&current_space_lvl);CHKERRQ(ierr);
1815813a1f2bSShri Abhyankar       reallocs++;
1816813a1f2bSShri Abhyankar     }
1817813a1f2bSShri Abhyankar 
1818813a1f2bSShri Abhyankar     /* copy data into free_space and free_space_lvl, then initialize lnk */
1819813a1f2bSShri Abhyankar     ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
1820813a1f2bSShri Abhyankar     bj_ptr[i]    = current_space->array;
1821813a1f2bSShri Abhyankar     bjlvl_ptr[i] = current_space_lvl->array;
1822813a1f2bSShri Abhyankar 
1823813a1f2bSShri Abhyankar     /* make sure the active row i has diagonal entry */
182465e19b50SBarry Smith     if (*(bj_ptr[i]+bdiag[i]) != i) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Row %D has missing diagonal in factored matrix\ntry running with -pc_factor_nonzeros_along_diagonal or -pc_factor_diagonal_fill",i);
1825813a1f2bSShri Abhyankar 
1826813a1f2bSShri Abhyankar     current_space->array           += nzi;
1827813a1f2bSShri Abhyankar     current_space->local_used      += nzi;
1828813a1f2bSShri Abhyankar     current_space->local_remaining -= nzi;
1829813a1f2bSShri Abhyankar     current_space_lvl->array           += nzi;
1830813a1f2bSShri Abhyankar     current_space_lvl->local_used      += nzi;
1831813a1f2bSShri Abhyankar     current_space_lvl->local_remaining -= nzi;
1832813a1f2bSShri Abhyankar   }
1833813a1f2bSShri Abhyankar 
1834813a1f2bSShri Abhyankar   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
1835813a1f2bSShri Abhyankar   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
1836813a1f2bSShri Abhyankar 
1837813a1f2bSShri Abhyankar   /* destroy list of free space and other temporary arrays */
1838813a1f2bSShri Abhyankar   ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
1839813a1f2bSShri Abhyankar 
1840813a1f2bSShri Abhyankar   /* copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */
18412ce24eb6SHong Zhang   ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr);
1842813a1f2bSShri Abhyankar 
1843813a1f2bSShri Abhyankar   ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
1844813a1f2bSShri Abhyankar   ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
18450e83c824SBarry Smith   ierr = PetscFree2(bj_ptr,bjlvl_ptr);CHKERRQ(ierr);
1846813a1f2bSShri Abhyankar 
1847813a1f2bSShri Abhyankar #if defined(PETSC_USE_INFO)
1848813a1f2bSShri Abhyankar   {
1849813a1f2bSShri Abhyankar     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
1850813a1f2bSShri Abhyankar     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr);
1851813a1f2bSShri Abhyankar     ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
1852813a1f2bSShri Abhyankar     ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr);
1853813a1f2bSShri Abhyankar     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
1854813a1f2bSShri Abhyankar     if (diagonal_fill) {
1855813a1f2bSShri Abhyankar       ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr);
1856813a1f2bSShri Abhyankar     }
1857813a1f2bSShri Abhyankar   }
1858813a1f2bSShri Abhyankar #endif
1859813a1f2bSShri Abhyankar 
1860813a1f2bSShri Abhyankar   /* put together the new matrix */
1861813a1f2bSShri Abhyankar   ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
1862813a1f2bSShri Abhyankar   ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr);
1863813a1f2bSShri Abhyankar   b = (Mat_SeqAIJ*)(fact)->data;
1864813a1f2bSShri Abhyankar   b->free_a       = PETSC_TRUE;
1865813a1f2bSShri Abhyankar   b->free_ij      = PETSC_TRUE;
1866813a1f2bSShri Abhyankar   b->singlemalloc = PETSC_FALSE;
1867f268cba8SShri Abhyankar   ierr = PetscMalloc((bdiag[0]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);
1868813a1f2bSShri Abhyankar   b->j          = bj;
1869813a1f2bSShri Abhyankar   b->i          = bi;
1870813a1f2bSShri Abhyankar   b->diag       = bdiag;
1871813a1f2bSShri Abhyankar   b->ilen       = 0;
1872813a1f2bSShri Abhyankar   b->imax       = 0;
1873813a1f2bSShri Abhyankar   b->row        = isrow;
1874813a1f2bSShri Abhyankar   b->col        = iscol;
1875813a1f2bSShri Abhyankar   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
1876813a1f2bSShri Abhyankar   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
1877813a1f2bSShri Abhyankar   b->icol       = isicol;
1878813a1f2bSShri Abhyankar   ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
1879813a1f2bSShri Abhyankar   /* In b structure:  Free imax, ilen, old a, old j.
1880813a1f2bSShri Abhyankar      Allocate bdiag, solve_work, new a, new j */
1881f268cba8SShri Abhyankar   ierr = PetscLogObjectMemory(fact,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
1882f268cba8SShri Abhyankar   b->maxnz = b->nz = bdiag[0]+1;
1883813a1f2bSShri Abhyankar   (fact)->info.factor_mallocs    = reallocs;
1884813a1f2bSShri Abhyankar   (fact)->info.fill_ratio_given  = f;
1885f268cba8SShri Abhyankar   (fact)->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]);
188635233d99SShri Abhyankar   (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ;
1887f2344125SBarry Smith   if (a->inode.size) {
1888f2344125SBarry Smith     (fact)->ops->lufactornumeric  = MatLUFactorNumeric_SeqAIJ_Inode;
1889f2344125SBarry Smith   }
1890813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
1891813a1f2bSShri Abhyankar }
1892813a1f2bSShri Abhyankar 
18936516239fSHong Zhang #undef __FUNCT__
1894ad04f41aSHong Zhang #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_inplace"
1895ad04f41aSHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
18966516239fSHong Zhang {
18976516239fSHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
18986516239fSHong Zhang   IS                 isicol;
18996516239fSHong Zhang   PetscErrorCode     ierr;
19006516239fSHong Zhang   const PetscInt     *r,*ic;
19016516239fSHong Zhang   PetscInt           n=A->rmap->n,*ai=a->i,*aj=a->j,d;
19026516239fSHong Zhang   PetscInt           *bi,*cols,nnz,*cols_lvl;
19036516239fSHong Zhang   PetscInt           *bdiag,prow,fm,nzbd,reallocs=0,dcount=0;
19046516239fSHong Zhang   PetscInt           i,levels,diagonal_fill;
19056516239fSHong Zhang   PetscTruth         col_identity,row_identity;
19066516239fSHong Zhang   PetscReal          f;
19076516239fSHong Zhang   PetscInt           nlnk,*lnk,*lnk_lvl=PETSC_NULL;
19086516239fSHong Zhang   PetscBT            lnkbt;
19096516239fSHong Zhang   PetscInt           nzi,*bj,**bj_ptr,**bjlvl_ptr;
19106516239fSHong Zhang   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
19116516239fSHong Zhang   PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL;
19126516239fSHong Zhang   PetscTruth         missing;
19136516239fSHong Zhang 
19146516239fSHong Zhang   PetscFunctionBegin;
1915e32f2f54SBarry Smith   if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n);
19166516239fSHong Zhang   f             = info->fill;
19176516239fSHong Zhang   levels        = (PetscInt)info->levels;
19186516239fSHong Zhang   diagonal_fill = (PetscInt)info->diagonal_fill;
19196516239fSHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
19206516239fSHong Zhang 
19216516239fSHong Zhang   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
19226516239fSHong Zhang   ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr);
19236516239fSHong Zhang   if (!levels && row_identity && col_identity) { /* special case: ilu(0) with natural ordering */
1924f77e22a1SHong Zhang     ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_TRUE);CHKERRQ(ierr);
1925ad04f41aSHong Zhang     (fact)->ops->lufactornumeric =  MatLUFactorNumeric_SeqAIJ_inplace;
1926f2344125SBarry Smith     if (a->inode.size) {
1927f2344125SBarry Smith       (fact)->ops->lufactornumeric  = MatLUFactorNumeric_SeqAIJ_Inode_inplace;
1928f2344125SBarry Smith     }
1929d5f3da31SBarry Smith     fact->factortype = MAT_FACTOR_ILU;
1930719d5645SBarry Smith     (fact)->info.factor_mallocs    = 0;
1931719d5645SBarry Smith     (fact)->info.fill_ratio_given  = info->fill;
1932719d5645SBarry Smith     (fact)->info.fill_ratio_needed = 1.0;
1933719d5645SBarry Smith     b               = (Mat_SeqAIJ*)(fact)->data;
19348ef3462fSBarry Smith     ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr);
1935e32f2f54SBarry Smith     if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d);
193608480c60SBarry Smith     b->row              = isrow;
193708480c60SBarry Smith     b->col              = iscol;
193882bf6240SBarry Smith     b->icol             = isicol;
1939719d5645SBarry Smith     ierr                = PetscMalloc(((fact)->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
1940c38d4ed2SBarry Smith     ierr                = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
1941c38d4ed2SBarry Smith     ierr                = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
19423a40ed3dSBarry Smith     PetscFunctionReturn(0);
194308480c60SBarry Smith   }
194408480c60SBarry Smith 
1945898183ecSLois Curfman McInnes   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
1946898183ecSLois Curfman McInnes   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
19479e25ed09SBarry Smith 
19481bfa06eaSJed Brown   /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */
19491bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
19501bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
1951a6df321fSShri Abhyankar   bi[0] = bdiag[0] = 0;
19526cf997b0SBarry Smith 
19530e83c824SBarry Smith   ierr = PetscMalloc2(n,PetscInt*,&bj_ptr,n,PetscInt*,&bjlvl_ptr);CHKERRQ(ierr);
19545a8e39fbSHong Zhang 
19555a8e39fbSHong Zhang   /* create a linked list for storing column indices of the active row */
19565a8e39fbSHong Zhang   nlnk = n + 1;
19575a8e39fbSHong Zhang   ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
19585a8e39fbSHong Zhang 
19595a8e39fbSHong Zhang   /* initial FreeSpace size is f*(ai[n]+1) */
1960a1a86e44SBarry Smith   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr);
19615a8e39fbSHong Zhang   current_space = free_space;
1962a1a86e44SBarry Smith   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr);
19635a8e39fbSHong Zhang   current_space_lvl = free_space_lvl;
19645a8e39fbSHong Zhang 
19655a8e39fbSHong Zhang   for (i=0; i<n; i++) {
19665a8e39fbSHong Zhang     nzi = 0;
19676cf997b0SBarry Smith     /* copy current row into linked list */
19685a8e39fbSHong Zhang     nnz  = ai[r[i]+1] - ai[r[i]];
1969e32f2f54SBarry Smith     if (!nnz) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i);
19705a8e39fbSHong Zhang     cols = aj + ai[r[i]];
19715a8e39fbSHong Zhang     lnk[i] = -1; /* marker to indicate if diagonal exists */
19725a8e39fbSHong Zhang     ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
19735a8e39fbSHong Zhang     nzi += nlnk;
19746cf997b0SBarry Smith 
19756cf997b0SBarry Smith     /* make sure diagonal entry is included */
19765a8e39fbSHong Zhang     if (diagonal_fill && lnk[i] == -1) {
19776cf997b0SBarry Smith       fm = n;
19785a8e39fbSHong Zhang       while (lnk[fm] < i) fm = lnk[fm];
19795a8e39fbSHong Zhang       lnk[i]     = lnk[fm]; /* insert diagonal into linked list */
19805a8e39fbSHong Zhang       lnk[fm]    = i;
19815a8e39fbSHong Zhang       lnk_lvl[i] = 0;
19825a8e39fbSHong Zhang       nzi++; dcount++;
19836cf997b0SBarry Smith     }
19846cf997b0SBarry Smith 
19855a8e39fbSHong Zhang     /* add pivot rows into the active row */
19865a8e39fbSHong Zhang     nzbd = 0;
19875a8e39fbSHong Zhang     prow = lnk[n];
19885a8e39fbSHong Zhang     while (prow < i) {
19895a8e39fbSHong Zhang       nnz      = bdiag[prow];
19905a8e39fbSHong Zhang       cols     = bj_ptr[prow] + nnz + 1;
19915a8e39fbSHong Zhang       cols_lvl = bjlvl_ptr[prow] + nnz + 1;
19925a8e39fbSHong Zhang       nnz      = bi[prow+1] - bi[prow] - nnz - 1;
19935a8e39fbSHong Zhang       ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr);
19945a8e39fbSHong Zhang       nzi += nlnk;
19955a8e39fbSHong Zhang       prow = lnk[prow];
19965a8e39fbSHong Zhang       nzbd++;
199756beaf04SBarry Smith     }
19985a8e39fbSHong Zhang     bdiag[i] = nzbd;
19995a8e39fbSHong Zhang     bi[i+1]  = bi[i] + nzi;
2000ecf371e4SBarry Smith 
20015a8e39fbSHong Zhang     /* if free space is not available, make more free space */
20025a8e39fbSHong Zhang     if (current_space->local_remaining<nzi) {
20035a8e39fbSHong Zhang       nnz = nzi*(n - i); /* estimated and max additional space needed */
2004a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
2005a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(nnz,&current_space_lvl);CHKERRQ(ierr);
20065a8e39fbSHong Zhang       reallocs++;
20075a8e39fbSHong Zhang     }
2008ecf371e4SBarry Smith 
20095a8e39fbSHong Zhang     /* copy data into free_space and free_space_lvl, then initialize lnk */
20105a8e39fbSHong Zhang     ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
20115a8e39fbSHong Zhang     bj_ptr[i]    = current_space->array;
20125a8e39fbSHong Zhang     bjlvl_ptr[i] = current_space_lvl->array;
20135a8e39fbSHong Zhang 
20145a8e39fbSHong Zhang     /* make sure the active row i has diagonal entry */
201565e19b50SBarry Smith     if (*(bj_ptr[i]+bdiag[i]) != i) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Row %D has missing diagonal in factored matrix\ntry running with -pc_factor_nonzeros_along_diagonal or -pc_factor_diagonal_fill",i);
20165a8e39fbSHong Zhang 
20175a8e39fbSHong Zhang     current_space->array           += nzi;
20185a8e39fbSHong Zhang     current_space->local_used      += nzi;
20195a8e39fbSHong Zhang     current_space->local_remaining -= nzi;
20205a8e39fbSHong Zhang     current_space_lvl->array           += nzi;
20215a8e39fbSHong Zhang     current_space_lvl->local_used      += nzi;
20225a8e39fbSHong Zhang     current_space_lvl->local_remaining -= nzi;
20239e25ed09SBarry Smith   }
20245a8e39fbSHong Zhang 
2025898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
2026898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
20275a8e39fbSHong Zhang 
20285a8e39fbSHong Zhang   /* destroy list of free space and other temporary arrays */
20295a8e39fbSHong Zhang   ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
20306516239fSHong Zhang   ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); /* copy free_space -> bj */
20315a8e39fbSHong Zhang   ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
2032a1a86e44SBarry Smith   ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
20330e83c824SBarry Smith   ierr = PetscFree2(bj_ptr,bjlvl_ptr);CHKERRQ(ierr);
20349e25ed09SBarry Smith 
20356cf91177SBarry Smith #if defined(PETSC_USE_INFO)
2036f64065bbSBarry Smith   {
20375a8e39fbSHong Zhang     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
2038ae15b995SBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr);
2039ae15b995SBarry Smith     ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
2040ae15b995SBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr);
2041ae15b995SBarry Smith     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
2042335d9088SBarry Smith     if (diagonal_fill) {
2043ae15b995SBarry Smith       ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr);
2044335d9088SBarry Smith     }
2045f64065bbSBarry Smith   }
204663ba0a88SBarry Smith #endif
2047bbb6d6a8SBarry Smith 
20489e25ed09SBarry Smith   /* put together the new matrix */
204907dbf95fSBarry Smith   ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
2050719d5645SBarry Smith   ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr);
2051719d5645SBarry Smith   b = (Mat_SeqAIJ*)(fact)->data;
2052e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
2053e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
20547c922b88SBarry Smith   b->singlemalloc = PETSC_FALSE;
20550e83c824SBarry Smith   ierr = PetscMalloc(bi[n]*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);
20565a8e39fbSHong Zhang   b->j          = bj;
20575a8e39fbSHong Zhang   b->i          = bi;
20585a8e39fbSHong Zhang   for (i=0; i<n; i++) bdiag[i] += bi[i];
20595a8e39fbSHong Zhang   b->diag       = bdiag;
2060416022c9SBarry Smith   b->ilen       = 0;
2061416022c9SBarry Smith   b->imax       = 0;
2062416022c9SBarry Smith   b->row        = isrow;
2063416022c9SBarry Smith   b->col        = iscol;
2064c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
2065c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
206682bf6240SBarry Smith   b->icol       = isicol;
206787828ca2SBarry Smith   ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
2068416022c9SBarry Smith   /* In b structure:  Free imax, ilen, old a, old j.
20695a8e39fbSHong Zhang      Allocate bdiag, solve_work, new a, new j */
2070719d5645SBarry Smith   ierr = PetscLogObjectMemory(fact,(bi[n]-n) * (sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
20715a8e39fbSHong Zhang   b->maxnz             = b->nz = bi[n] ;
2072719d5645SBarry Smith   (fact)->info.factor_mallocs    = reallocs;
2073719d5645SBarry Smith   (fact)->info.fill_ratio_given  = f;
2074719d5645SBarry Smith   (fact)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]);
2075ad04f41aSHong Zhang   (fact)->ops->lufactornumeric =  MatLUFactorNumeric_SeqAIJ_inplace;
2076f2344125SBarry Smith   if (a->inode.size) {
2077f2344125SBarry Smith     (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode_inplace;
2078d3ac4fa3SBarry Smith   }
20793a40ed3dSBarry Smith   PetscFunctionReturn(0);
20809e25ed09SBarry Smith }
2081d5d45c9bSBarry Smith 
2082a6175056SHong Zhang #undef __FUNCT__
208335233d99SShri Abhyankar #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ"
208435233d99SShri Abhyankar PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info)
20855f44c854SHong Zhang {
20865f44c854SHong Zhang   Mat            C = B;
20875f44c854SHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
20885f44c854SHong Zhang   Mat_SeqSBAIJ   *b=(Mat_SeqSBAIJ*)C->data;
20895f44c854SHong Zhang   IS             ip=b->row,iip = b->icol;
20905f44c854SHong Zhang   PetscErrorCode ierr;
20915f44c854SHong Zhang   const PetscInt *rip,*riip;
20925f44c854SHong Zhang   PetscInt       i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bdiag=b->diag,*bjtmp;
20935f44c854SHong Zhang   PetscInt       *ai=a->i,*aj=a->j;
20945f44c854SHong Zhang   PetscInt       k,jmin,jmax,*c2r,*il,col,nexti,ili,nz;
20955f44c854SHong Zhang   MatScalar      *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi;
20965f44c854SHong Zhang   PetscTruth     perm_identity;
2097eca579c0SMatthew G Knepley   PetscInt       newshift = 0;
20985f44c854SHong Zhang 
2099d90ac83dSHong Zhang   FactorShiftCtx sctx;
210098a93161SHong Zhang   PetscReal      rs;
210198a93161SHong Zhang   MatScalar      d,*v;
210298a93161SHong Zhang 
21035f44c854SHong Zhang   PetscFunctionBegin;
210498a93161SHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
2105d90ac83dSHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
210698a93161SHong Zhang 
2107f4db908eSBarry Smith   if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
210898a93161SHong Zhang     sctx.shift_top = info->zeropivot;
210998a93161SHong Zhang     for (i=0; i<mbs; i++) {
211098a93161SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
211198a93161SHong Zhang       d  = (aa)[a->diag[i]];
211298a93161SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
211398a93161SHong Zhang       v  = aa+ai[i];
211498a93161SHong Zhang       nz = ai[i+1] - ai[i];
211598a93161SHong Zhang       for (j=0; j<nz; j++)
211698a93161SHong Zhang 	rs += PetscAbsScalar(v[j]);
211798a93161SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
211898a93161SHong Zhang     }
211998a93161SHong Zhang     sctx.shift_top   *= 1.1;
212098a93161SHong Zhang     sctx.nshift_max   = 5;
212198a93161SHong Zhang     sctx.shift_lo     = 0.;
212298a93161SHong Zhang     sctx.shift_hi     = 1.;
212398a93161SHong Zhang   }
21245f44c854SHong Zhang 
21255f44c854SHong Zhang   ierr  = ISGetIndices(ip,&rip);CHKERRQ(ierr);
21265f44c854SHong Zhang   ierr  = ISGetIndices(iip,&riip);CHKERRQ(ierr);
21275f44c854SHong Zhang 
21285f44c854SHong Zhang   /* allocate working arrays
21295f44c854SHong Zhang      c2r: linked list, keep track of pivot rows for a given column. c2r[col]: head of the list for a given col
21305f44c854SHong Zhang      il:  for active k row, il[i] gives the index of the 1st nonzero entry in U[i,k:n-1] in bj and ba arrays
21315f44c854SHong Zhang   */
21320e83c824SBarry Smith   ierr = PetscMalloc3(mbs,MatScalar,&rtmp,mbs,PetscInt,&il,mbs,PetscInt,&c2r);CHKERRQ(ierr);
21335f44c854SHong Zhang 
213498a93161SHong Zhang   do {
2135d90ac83dSHong Zhang     sctx.useshift = PETSC_FALSE;
213698a93161SHong Zhang 
2137c5546cabSHong Zhang     for (i=0; i<mbs; i++) c2r[i] = mbs;
21385f44c854SHong Zhang     il[0] = 0;
21395f44c854SHong Zhang 
21405f44c854SHong Zhang     for (k = 0; k<mbs; k++){
21415f44c854SHong Zhang       /* zero rtmp */
21425f44c854SHong Zhang       nz = bi[k+1] - bi[k];
21435f44c854SHong Zhang       bjtmp = bj + bi[k];
21445f44c854SHong Zhang       for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
21455f44c854SHong Zhang 
21465f44c854SHong Zhang       /* load in initial unfactored row */
21475f44c854SHong Zhang       bval = ba + bi[k];
21485f44c854SHong Zhang       jmin = ai[rip[k]]; jmax = ai[rip[k]+1];
21495f44c854SHong Zhang       for (j = jmin; j < jmax; j++){
21505f44c854SHong Zhang         col = riip[aj[j]];
21515f44c854SHong Zhang         if (col >= k){ /* only take upper triangular entry */
21525f44c854SHong Zhang           rtmp[col] = aa[j];
21535f44c854SHong Zhang           *bval++   = 0.0; /* for in-place factorization */
21545f44c854SHong Zhang         }
21555f44c854SHong Zhang       }
215698a93161SHong Zhang       /* shift the diagonal of the matrix: ZeropivotApply() */
215798a93161SHong Zhang       rtmp[k] += sctx.shift_amount;  /* shift the diagonal of the matrix */
21585f44c854SHong Zhang 
21595f44c854SHong Zhang       /* modify k-th row by adding in those rows i with U(i,k)!=0 */
21605f44c854SHong Zhang       dk = rtmp[k];
21615f44c854SHong Zhang       i  = c2r[k]; /* first row to be added to k_th row  */
21625f44c854SHong Zhang 
21635f44c854SHong Zhang       while (i < k){
21645f44c854SHong Zhang         nexti = c2r[i]; /* next row to be added to k_th row */
21655f44c854SHong Zhang 
21665f44c854SHong Zhang         /* compute multiplier, update diag(k) and U(i,k) */
21675f44c854SHong Zhang         ili   = il[i];  /* index of first nonzero element in U(i,k:bms-1) */
21685f44c854SHong Zhang         uikdi = - ba[ili]*ba[bdiag[i]];  /* diagonal(k) */
21695f44c854SHong Zhang         dk   += uikdi*ba[ili]; /* update diag[k] */
21705f44c854SHong Zhang         ba[ili] = uikdi; /* -U(i,k) */
21715f44c854SHong Zhang 
21725f44c854SHong Zhang         /* add multiple of row i to k-th row */
21735f44c854SHong Zhang         jmin = ili + 1; jmax = bi[i+1];
21745f44c854SHong Zhang         if (jmin < jmax){
21755f44c854SHong Zhang           for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j];
21765f44c854SHong Zhang           /* update il and c2r for row i */
21775f44c854SHong Zhang           il[i] = jmin;
21785f44c854SHong Zhang           j = bj[jmin]; c2r[i] = c2r[j]; c2r[j] = i;
21795f44c854SHong Zhang         }
21805f44c854SHong Zhang         i = nexti;
21815f44c854SHong Zhang       }
21825f44c854SHong Zhang 
21835f44c854SHong Zhang       /* copy data into U(k,:) */
218498a93161SHong Zhang       rs   = 0.0;
21855f44c854SHong Zhang       jmin = bi[k]; jmax = bi[k+1]-1;
21865f44c854SHong Zhang       if (jmin < jmax) {
21875f44c854SHong Zhang         for (j=jmin; j<jmax; j++){
218898a93161SHong Zhang           col = bj[j]; ba[j] = rtmp[col]; rs += PetscAbsScalar(ba[j]);
21895f44c854SHong Zhang         }
21905f44c854SHong Zhang         /* add the k-th row into il and c2r */
21915f44c854SHong Zhang         il[k] = jmin;
21925f44c854SHong Zhang         i = bj[jmin]; c2r[k] = c2r[i]; c2r[i] = k;
21935f44c854SHong Zhang       }
219498a93161SHong Zhang 
219598a93161SHong Zhang       /* MatPivotCheck() */
219698a93161SHong Zhang       sctx.rs  = rs;
219798a93161SHong Zhang       sctx.pv  = dk;
21989f522e40SShri Abhyankar       ierr = MatPivotCheck(info,&sctx,i,&newshift);CHKERRQ(ierr);
21999f522e40SShri Abhyankar       if(newshift == 1) break;
220098a93161SHong Zhang       dk = sctx.pv;
220198a93161SHong Zhang 
220298a93161SHong Zhang       ba[bdiag[k]] = 1.0/dk; /* U(k,k) */
220398a93161SHong Zhang     }
2204d90ac83dSHong Zhang   } while (sctx.useshift);
22055f44c854SHong Zhang 
22060e83c824SBarry Smith   ierr = PetscFree3(rtmp,il,c2r);CHKERRQ(ierr);
22075f44c854SHong Zhang   ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr);
22085f44c854SHong Zhang   ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr);
22095f44c854SHong Zhang 
22105f44c854SHong Zhang   ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr);
22115f44c854SHong Zhang   if (perm_identity){
221235233d99SShri Abhyankar     B->ops->solve           = MatSolve_SeqSBAIJ_1_NaturalOrdering;
221335233d99SShri Abhyankar     B->ops->solvetranspose  = MatSolve_SeqSBAIJ_1_NaturalOrdering;
22142169352eSHong Zhang     B->ops->forwardsolve    = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering;
22152169352eSHong Zhang     B->ops->backwardsolve   = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering;
22165f44c854SHong Zhang   } else {
221735233d99SShri Abhyankar     B->ops->solve           = MatSolve_SeqSBAIJ_1;
221835233d99SShri Abhyankar     B->ops->solvetranspose  = MatSolve_SeqSBAIJ_1;
221935233d99SShri Abhyankar     B->ops->forwardsolve    = MatForwardSolve_SeqSBAIJ_1;
222035233d99SShri Abhyankar     B->ops->backwardsolve   = MatBackwardSolve_SeqSBAIJ_1;
22215f44c854SHong Zhang   }
22225f44c854SHong Zhang 
22235f44c854SHong Zhang   C->assembled    = PETSC_TRUE;
22245f44c854SHong Zhang   C->preallocated = PETSC_TRUE;
22255f44c854SHong Zhang   ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr);
222698a93161SHong Zhang 
222798a93161SHong Zhang   /* MatPivotView() */
222898a93161SHong Zhang   if (sctx.nshift){
2229f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
223098a93161SHong Zhang       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,sctx.shift_fraction,sctx.shift_top);CHKERRQ(ierr);
2231f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
223298a93161SHong Zhang       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
2233f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_INBLOCKS){
2234b2a402fcSHong Zhang       ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %G\n",sctx.nshift,info->shiftamount);CHKERRQ(ierr);
223598a93161SHong Zhang     }
223698a93161SHong Zhang   }
22375f44c854SHong Zhang   PetscFunctionReturn(0);
22385f44c854SHong Zhang }
22395f44c854SHong Zhang 
22405f44c854SHong Zhang #undef __FUNCT__
22410a3c351aSHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ_inplace"
22420a3c351aSHong Zhang PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ_inplace(Mat B,Mat A,const MatFactorInfo *info)
2243a6175056SHong Zhang {
2244719d5645SBarry Smith   Mat            C = B;
22450968510aSHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
22462ed133dbSHong Zhang   Mat_SeqSBAIJ   *b=(Mat_SeqSBAIJ*)C->data;
22479bfd6278SHong Zhang   IS             ip=b->row,iip = b->icol;
22482ed133dbSHong Zhang   PetscErrorCode ierr;
22495d0c19d7SBarry Smith   const PetscInt *rip,*riip;
2250c5546cabSHong Zhang   PetscInt       i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bcol,*bjtmp;
22512ed133dbSHong Zhang   PetscInt       *ai=a->i,*aj=a->j;
22522ed133dbSHong Zhang   PetscInt       k,jmin,jmax,*jl,*il,col,nexti,ili,nz;
2253622e2028SHong Zhang   MatScalar      *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi;
2254540703acSHong Zhang   PetscInt       newshift;
2255db4efbfdSBarry Smith   PetscTruth     perm_identity;
2256*0e95ead3SHong Zhang   FactorShiftCtx sctx;
2257*0e95ead3SHong Zhang   PetscReal      rs;
2258*0e95ead3SHong Zhang   MatScalar      d,*v;
2259d5d45c9bSBarry Smith 
2260a6175056SHong Zhang   PetscFunctionBegin;
2261*0e95ead3SHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
2262*0e95ead3SHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
2263*0e95ead3SHong Zhang 
2264*0e95ead3SHong Zhang   if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
2265*0e95ead3SHong Zhang     sctx.shift_top = info->zeropivot;
2266*0e95ead3SHong Zhang     for (i=0; i<mbs; i++) {
2267*0e95ead3SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
2268*0e95ead3SHong Zhang       d  = (aa)[a->diag[i]];
2269*0e95ead3SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
2270*0e95ead3SHong Zhang       v  = aa+ai[i];
2271*0e95ead3SHong Zhang       nz = ai[i+1] - ai[i];
2272*0e95ead3SHong Zhang       for (j=0; j<nz; j++)
2273*0e95ead3SHong Zhang 	rs += PetscAbsScalar(v[j]);
2274*0e95ead3SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
2275*0e95ead3SHong Zhang     }
2276*0e95ead3SHong Zhang     sctx.shift_top   *= 1.1;
2277*0e95ead3SHong Zhang     sctx.nshift_max   = 5;
2278*0e95ead3SHong Zhang     sctx.shift_lo     = 0.;
2279*0e95ead3SHong Zhang     sctx.shift_hi     = 1.;
2280*0e95ead3SHong Zhang   }
2281ee45ca4aSHong Zhang 
22822ed133dbSHong Zhang   ierr  = ISGetIndices(ip,&rip);CHKERRQ(ierr);
22839bfd6278SHong Zhang   ierr  = ISGetIndices(iip,&riip);CHKERRQ(ierr);
22842ed133dbSHong Zhang 
22852ed133dbSHong Zhang   /* initialization */
22860e83c824SBarry Smith   ierr = PetscMalloc3(mbs,MatScalar,&rtmp,mbs,PetscInt,&il,mbs,PetscInt,&jl);CHKERRQ(ierr);
2287*0e95ead3SHong Zhang 
22882ed133dbSHong Zhang   do {
2289*0e95ead3SHong Zhang     sctx.useshift = PETSC_FALSE;
2290*0e95ead3SHong Zhang 
2291c5546cabSHong Zhang     for (i=0; i<mbs; i++) jl[i] = mbs;
2292c5546cabSHong Zhang     il[0] = 0;
22932ed133dbSHong Zhang 
22942ed133dbSHong Zhang     for (k = 0; k<mbs; k++){
2295c5546cabSHong Zhang       /* zero rtmp */
2296c5546cabSHong Zhang       nz = bi[k+1] - bi[k];
2297c5546cabSHong Zhang       bjtmp = bj + bi[k];
2298c5546cabSHong Zhang       for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
2299c5546cabSHong Zhang 
2300c05c3958SHong Zhang       bval = ba + bi[k];
23012ed133dbSHong Zhang       /* initialize k-th row by the perm[k]-th row of A */
23022ed133dbSHong Zhang       jmin = ai[rip[k]]; jmax = ai[rip[k]+1];
23032ed133dbSHong Zhang       for (j = jmin; j < jmax; j++){
23049bfd6278SHong Zhang         col = riip[aj[j]];
23052ed133dbSHong Zhang         if (col >= k){ /* only take upper triangular entry */
23062ed133dbSHong Zhang           rtmp[col] = aa[j];
2307c05c3958SHong Zhang           *bval++  = 0.0; /* for in-place factorization */
23082ed133dbSHong Zhang         }
23092ed133dbSHong Zhang       }
231039e3d630SHong Zhang       /* shift the diagonal of the matrix */
2311540703acSHong Zhang       if (sctx.nshift) rtmp[k] += sctx.shift_amount;
23122ed133dbSHong Zhang 
23132ed133dbSHong Zhang       /* modify k-th row by adding in those rows i with U(i,k)!=0 */
23142ed133dbSHong Zhang       dk = rtmp[k];
23152ed133dbSHong Zhang       i = jl[k]; /* first row to be added to k_th row  */
23162ed133dbSHong Zhang 
23172ed133dbSHong Zhang       while (i < k){
23182ed133dbSHong Zhang         nexti = jl[i]; /* next row to be added to k_th row */
23192ed133dbSHong Zhang 
23202ed133dbSHong Zhang         /* compute multiplier, update diag(k) and U(i,k) */
23212ed133dbSHong Zhang         ili = il[i];  /* index of first nonzero element in U(i,k:bms-1) */
23222ed133dbSHong Zhang         uikdi = - ba[ili]*ba[bi[i]];  /* diagonal(k) */
23232ed133dbSHong Zhang         dk += uikdi*ba[ili];
23242ed133dbSHong Zhang         ba[ili] = uikdi; /* -U(i,k) */
23252ed133dbSHong Zhang 
23262ed133dbSHong Zhang         /* add multiple of row i to k-th row */
23272ed133dbSHong Zhang         jmin = ili + 1; jmax = bi[i+1];
23282ed133dbSHong Zhang         if (jmin < jmax){
23292ed133dbSHong Zhang           for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j];
23302ed133dbSHong Zhang           /* update il and jl for row i */
23312ed133dbSHong Zhang           il[i] = jmin;
23322ed133dbSHong Zhang           j = bj[jmin]; jl[i] = jl[j]; jl[j] = i;
23332ed133dbSHong Zhang         }
23342ed133dbSHong Zhang         i = nexti;
23352ed133dbSHong Zhang       }
23362ed133dbSHong Zhang 
23370697b6caSHong Zhang       /* shift the diagonals when zero pivot is detected */
23380697b6caSHong Zhang       /* compute rs=sum of abs(off-diagonal) */
23390697b6caSHong Zhang       rs   = 0.0;
23403c9e8598SHong Zhang       jmin = bi[k]+1;
23413c9e8598SHong Zhang       nz   = bi[k+1] - jmin;
23423c9e8598SHong Zhang       bcol = bj + jmin;
234304fbf559SBarry Smith       for (j=0; j<nz; j++) {
234404fbf559SBarry Smith         rs += PetscAbsScalar(rtmp[bcol[j]]);
23453c9e8598SHong Zhang       }
2346540703acSHong Zhang 
2347540703acSHong Zhang       sctx.rs = rs;
2348540703acSHong Zhang       sctx.pv = dk;
2349*0e95ead3SHong Zhang       ierr = MatPivotCheck(info,&sctx,k,&newshift);CHKERRQ(ierr);
2350*0e95ead3SHong Zhang       if (newshift == 1) break;
2351*0e95ead3SHong Zhang       dk = sctx.pv;
23523c9e8598SHong Zhang 
23533c9e8598SHong Zhang       /* copy data into U(k,:) */
235439e3d630SHong Zhang       ba[bi[k]] = 1.0/dk; /* U(k,k) */
235539e3d630SHong Zhang       jmin = bi[k]+1; jmax = bi[k+1];
235639e3d630SHong Zhang       if (jmin < jmax) {
235739e3d630SHong Zhang         for (j=jmin; j<jmax; j++){
2358c6ae9bfcSHong Zhang           col = bj[j]; ba[j] = rtmp[col];
23593c9e8598SHong Zhang         }
236039e3d630SHong Zhang         /* add the k-th row into il and jl */
23613c9e8598SHong Zhang         il[k] = jmin;
23623c9e8598SHong Zhang         i = bj[jmin]; jl[k] = jl[i]; jl[i] = k;
23633c9e8598SHong Zhang       }
23643c9e8598SHong Zhang     }
2365*0e95ead3SHong Zhang   } while (sctx.useshift);
2366*0e95ead3SHong Zhang 
23670e83c824SBarry Smith   ierr = PetscFree3(rtmp,il,jl);CHKERRQ(ierr);
236839e3d630SHong Zhang   ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr);
23699bfd6278SHong Zhang   ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr);
2370db4efbfdSBarry Smith 
2371db4efbfdSBarry Smith   ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr);
2372db4efbfdSBarry Smith   if (perm_identity){
23730a3c351aSHong Zhang     B->ops->solve           = MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace;
23740a3c351aSHong Zhang     B->ops->solvetranspose  = MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace;
23750a3c351aSHong Zhang     B->ops->forwardsolve    = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace;
23760a3c351aSHong Zhang     B->ops->backwardsolve   = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace;
2377db4efbfdSBarry Smith   } else {
23780a3c351aSHong Zhang     B->ops->solve           = MatSolve_SeqSBAIJ_1_inplace;
23790a3c351aSHong Zhang     B->ops->solvetranspose  = MatSolve_SeqSBAIJ_1_inplace;
23800a3c351aSHong Zhang     B->ops->forwardsolve    = MatForwardSolve_SeqSBAIJ_1_inplace;
23810a3c351aSHong Zhang     B->ops->backwardsolve   = MatBackwardSolve_SeqSBAIJ_1_inplace;
2382db4efbfdSBarry Smith   }
2383db4efbfdSBarry Smith 
23843c9e8598SHong Zhang   C->assembled    = PETSC_TRUE;
23853c9e8598SHong Zhang   C->preallocated = PETSC_TRUE;
2386d0f46423SBarry Smith   ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr);
2387540703acSHong Zhang   if (sctx.nshift){
2388f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
23891e2582c4SBarry Smith       ierr = PetscInfo2(A,"number of shiftnz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
2390f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
23911e2582c4SBarry Smith       ierr = PetscInfo2(A,"number of shiftpd tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
23920697b6caSHong Zhang     }
23933c9e8598SHong Zhang   }
23943c9e8598SHong Zhang   PetscFunctionReturn(0);
23953c9e8598SHong Zhang }
2396a6175056SHong Zhang 
23978a76ed4fSHong Zhang /*
23988a76ed4fSHong Zhang    icc() under revised new data structure.
23998a76ed4fSHong Zhang    Factored arrays bj and ba are stored as
24008a76ed4fSHong Zhang      U(0,:),...,U(i,:),U(n-1,:)
24018a76ed4fSHong Zhang 
24028a76ed4fSHong Zhang    ui=fact->i is an array of size n+1, in which
24038a76ed4fSHong Zhang    ui+
24048a76ed4fSHong Zhang      ui[i]:  points to 1st entry of U(i,:),i=0,...,n-1
24058a76ed4fSHong Zhang      ui[n]:  points to U(n-1,n-1)+1
24068a76ed4fSHong Zhang 
24078a76ed4fSHong Zhang   udiag=fact->diag is an array of size n,in which
24088a76ed4fSHong Zhang      udiag[i]: points to diagonal of U(i,:), i=0,...,n-1
24098a76ed4fSHong Zhang 
24108a76ed4fSHong Zhang    U(i,:) contains udiag[i] as its last entry, i.e.,
24118a76ed4fSHong Zhang     U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i])
24128a76ed4fSHong Zhang */
24138a76ed4fSHong Zhang 
2414a6175056SHong Zhang #undef __FUNCT__
241535233d99SShri Abhyankar #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ"
241635233d99SShri Abhyankar PetscErrorCode MatICCFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
2417a6175056SHong Zhang {
24180968510aSHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
2419ed59401aSHong Zhang   Mat_SeqSBAIJ       *b;
2420dfbe8321SBarry Smith   PetscErrorCode     ierr;
242158ebbce7SBarry Smith   PetscTruth         perm_identity,missing;
24225f44c854SHong Zhang   PetscInt           reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag;
24235d0c19d7SBarry Smith   const PetscInt     *rip,*riip;
2424ed59401aSHong Zhang   PetscInt           jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow;
242558ebbce7SBarry Smith   PetscInt           nlnk,*lnk,*lnk_lvl=PETSC_NULL,d;
24265a8e39fbSHong Zhang   PetscInt           ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr;
2427ed59401aSHong Zhang   PetscReal          fill=info->fill,levels=info->levels;
2428a1a86e44SBarry Smith   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
2429a1a86e44SBarry Smith   PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL;
24307a48dd6fSHong Zhang   PetscBT            lnkbt;
2431b635d36bSHong Zhang   IS                 iperm;
2432a6175056SHong Zhang 
2433a6175056SHong Zhang   PetscFunctionBegin;
2434e32f2f54SBarry Smith   if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n);
243558ebbce7SBarry Smith   ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr);
2436e32f2f54SBarry Smith   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d);
2437653a6975SHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
2438b635d36bSHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
24397a48dd6fSHong Zhang 
244039e3d630SHong Zhang   ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr);
24415f44c854SHong Zhang   ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr);
244239e3d630SHong Zhang   ui[0] = 0;
244339e3d630SHong Zhang 
24448a76ed4fSHong Zhang   /* ICC(0) without matrix ordering: simply rearrange column indices */
2445622e2028SHong Zhang   if (!levels && perm_identity) {
24465f44c854SHong Zhang     for (i=0; i<am; i++) {
2447c5546cabSHong Zhang       ncols    = ai[i+1] - a->diag[i];
2448c5546cabSHong Zhang       ui[i+1]  = ui[i] + ncols;
2449c5546cabSHong Zhang       udiag[i] = ui[i+1] - 1; /* points to the last entry of U(i,:) */
2450dc1e2a3fSHong Zhang     }
2451dc1e2a3fSHong Zhang     ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
2452dc1e2a3fSHong Zhang     cols = uj;
2453dc1e2a3fSHong Zhang     for (i=0; i<am; i++) {
24545f44c854SHong Zhang       aj    = a->j + a->diag[i] + 1; /* 1st entry of U(i,:) without diagonal */
2455dc1e2a3fSHong Zhang       ncols = ai[i+1] - a->diag[i] -1;
24565f44c854SHong Zhang       for (j=0; j<ncols; j++) *cols++ = aj[j];
24575f44c854SHong Zhang       *cols++ = i; /* diagoanl is located as the last entry of U(i,:) */
24585f44c854SHong Zhang     }
24595f44c854SHong Zhang   } else { /* case: levels>0 || (levels=0 && !perm_identity) */
24605f44c854SHong Zhang     ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
24615f44c854SHong Zhang     ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
24625f44c854SHong Zhang 
24635f44c854SHong Zhang     /* initialization */
24645f44c854SHong Zhang     ierr  = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr);
24655f44c854SHong Zhang 
24665f44c854SHong Zhang     /* jl: linked list for storing indices of the pivot rows
24675f44c854SHong Zhang        il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
24680e83c824SBarry Smith     ierr = PetscMalloc4(am,PetscInt*,&uj_ptr,am,PetscInt*,&uj_lvl_ptr,am,PetscInt,&jl,am,PetscInt,&il);CHKERRQ(ierr);
24695f44c854SHong Zhang     for (i=0; i<am; i++){
24705f44c854SHong Zhang       jl[i] = am; il[i] = 0;
24715f44c854SHong Zhang     }
24725f44c854SHong Zhang 
24735f44c854SHong Zhang     /* create and initialize a linked list for storing column indices of the active row k */
24745f44c854SHong Zhang     nlnk = am + 1;
24755f44c854SHong Zhang     ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
24765f44c854SHong Zhang 
24775f44c854SHong Zhang     /* initial FreeSpace size is fill*(ai[am]+1) */
24785f44c854SHong Zhang     ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr);
24795f44c854SHong Zhang     current_space = free_space;
24805f44c854SHong Zhang     ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space_lvl);CHKERRQ(ierr);
24815f44c854SHong Zhang     current_space_lvl = free_space_lvl;
24825f44c854SHong Zhang 
24835f44c854SHong Zhang     for (k=0; k<am; k++){  /* for each active row k */
24845f44c854SHong Zhang       /* initialize lnk by the column indices of row rip[k] of A */
24855f44c854SHong Zhang       nzk   = 0;
24865f44c854SHong Zhang       ncols = ai[rip[k]+1] - ai[rip[k]];
2487e32f2f54SBarry Smith       if (!ncols) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_MAT_CH_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",rip[k],k);
24885f44c854SHong Zhang       ncols_upper = 0;
24895f44c854SHong Zhang       for (j=0; j<ncols; j++){
24905f44c854SHong Zhang         i = *(aj + ai[rip[k]] + j); /* unpermuted column index */
24915f44c854SHong Zhang         if (riip[i] >= k){ /* only take upper triangular entry */
24925f44c854SHong Zhang           ajtmp[ncols_upper] = i;
24935f44c854SHong Zhang           ncols_upper++;
24945f44c854SHong Zhang         }
24955f44c854SHong Zhang       }
24965f44c854SHong Zhang       ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
24975f44c854SHong Zhang       nzk += nlnk;
24985f44c854SHong Zhang 
24995f44c854SHong Zhang       /* update lnk by computing fill-in for each pivot row to be merged in */
25005f44c854SHong Zhang       prow = jl[k]; /* 1st pivot row */
25015f44c854SHong Zhang 
25025f44c854SHong Zhang       while (prow < k){
25035f44c854SHong Zhang         nextprow = jl[prow];
25045f44c854SHong Zhang 
25055f44c854SHong Zhang         /* merge prow into k-th row */
25065f44c854SHong Zhang         jmin = il[prow] + 1;  /* index of the 2nd nzero entry in U(prow,k:am-1) */
25075f44c854SHong Zhang         jmax = ui[prow+1];
25085f44c854SHong Zhang         ncols = jmax-jmin;
25095f44c854SHong Zhang         i     = jmin - ui[prow];
25105f44c854SHong Zhang         cols  = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */
25115f44c854SHong Zhang         uj    = uj_lvl_ptr[prow] + i; /* levels of cols */
25125f44c854SHong Zhang         j     = *(uj - 1);
25135f44c854SHong Zhang         ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr);
25145f44c854SHong Zhang         nzk += nlnk;
25155f44c854SHong Zhang 
25165f44c854SHong Zhang         /* update il and jl for prow */
25175f44c854SHong Zhang         if (jmin < jmax){
25185f44c854SHong Zhang           il[prow] = jmin;
25195f44c854SHong Zhang           j = *cols; jl[prow] = jl[j]; jl[j] = prow;
25205f44c854SHong Zhang         }
25215f44c854SHong Zhang         prow = nextprow;
25225f44c854SHong Zhang       }
25235f44c854SHong Zhang 
25245f44c854SHong Zhang       /* if free space is not available, make more free space */
25255f44c854SHong Zhang       if (current_space->local_remaining<nzk) {
25265f44c854SHong Zhang         i  = am - k + 1; /* num of unfactored rows */
2527c6ae9bfcSHong Zhang         i *= PetscMin(nzk, i-1); /* i*nzk, i*(i-1): estimated and max additional space needed */
25285f44c854SHong Zhang         ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
25295f44c854SHong Zhang         ierr = PetscFreeSpaceGet(i,&current_space_lvl);CHKERRQ(ierr);
25305f44c854SHong Zhang         reallocs++;
25315f44c854SHong Zhang       }
25325f44c854SHong Zhang 
25335f44c854SHong Zhang       /* copy data into free_space and free_space_lvl, then initialize lnk */
2534e32f2f54SBarry Smith       if (nzk == 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k);
25355f44c854SHong Zhang       ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
25365f44c854SHong Zhang 
25375f44c854SHong Zhang       /* add the k-th row into il and jl */
25385f44c854SHong Zhang       if (nzk > 1){
25395f44c854SHong Zhang         i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
25405f44c854SHong Zhang         jl[k] = jl[i]; jl[i] = k;
25415f44c854SHong Zhang         il[k] = ui[k] + 1;
25425f44c854SHong Zhang       }
25435f44c854SHong Zhang       uj_ptr[k]     = current_space->array;
25445f44c854SHong Zhang       uj_lvl_ptr[k] = current_space_lvl->array;
25455f44c854SHong Zhang 
25465f44c854SHong Zhang       current_space->array           += nzk;
25475f44c854SHong Zhang       current_space->local_used      += nzk;
25485f44c854SHong Zhang       current_space->local_remaining -= nzk;
25495f44c854SHong Zhang 
25505f44c854SHong Zhang       current_space_lvl->array           += nzk;
25515f44c854SHong Zhang       current_space_lvl->local_used      += nzk;
25525f44c854SHong Zhang       current_space_lvl->local_remaining -= nzk;
25535f44c854SHong Zhang 
25545f44c854SHong Zhang       ui[k+1] = ui[k] + nzk;
25555f44c854SHong Zhang     }
25565f44c854SHong Zhang 
25575f44c854SHong Zhang #if defined(PETSC_USE_INFO)
25585f44c854SHong Zhang     if (ai[am] != 0) {
25595f44c854SHong Zhang       PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]);
25605f44c854SHong Zhang       ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr);
25615f44c854SHong Zhang       ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
25625f44c854SHong Zhang       ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr);
25635f44c854SHong Zhang     } else {
25645f44c854SHong Zhang       ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr);
25655f44c854SHong Zhang     }
25665f44c854SHong Zhang #endif
25675f44c854SHong Zhang 
25685f44c854SHong Zhang     ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
25695f44c854SHong Zhang     ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
25700e83c824SBarry Smith     ierr = PetscFree4(uj_ptr,uj_lvl_ptr,jl,il);CHKERRQ(ierr);
25715f44c854SHong Zhang     ierr = PetscFree(ajtmp);CHKERRQ(ierr);
25725f44c854SHong Zhang 
25735f44c854SHong Zhang     /* destroy list of free space and other temporary array(s) */
25745f44c854SHong Zhang     ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
257535233d99SShri Abhyankar     ierr = PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag);CHKERRQ(ierr); /* store matrix factor  */
25765f44c854SHong Zhang     ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
25775f44c854SHong Zhang     ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
25785f44c854SHong Zhang 
25795f44c854SHong Zhang   } /* end of case: levels>0 || (levels=0 && !perm_identity) */
25805f44c854SHong Zhang 
25815f44c854SHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
25825f44c854SHong Zhang   b    = (Mat_SeqSBAIJ*)(fact)->data;
25835f44c854SHong Zhang   b->singlemalloc = PETSC_FALSE;
25845f44c854SHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr);
25855f44c854SHong Zhang   b->j    = uj;
25865f44c854SHong Zhang   b->i    = ui;
25875f44c854SHong Zhang   b->diag = udiag;
25887f53bb6cSHong Zhang   b->free_diag = PETSC_TRUE;
25895f44c854SHong Zhang   b->ilen = 0;
25905f44c854SHong Zhang   b->imax = 0;
25915f44c854SHong Zhang   b->row  = perm;
25925f44c854SHong Zhang   b->col  = perm;
25935f44c854SHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
25945f44c854SHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
25955f44c854SHong Zhang   b->icol = iperm;
25965f44c854SHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
25975f44c854SHong Zhang   ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
2598d595f711SHong Zhang   ierr = PetscLogObjectMemory(fact,ui[am]*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
25995f44c854SHong Zhang   b->maxnz   = b->nz = ui[am];
26005f44c854SHong Zhang   b->free_a  = PETSC_TRUE;
26015f44c854SHong Zhang   b->free_ij = PETSC_TRUE;
26025f44c854SHong Zhang 
2603f284f12aSHong Zhang   fact->info.factor_mallocs    = reallocs;
2604f284f12aSHong Zhang   fact->info.fill_ratio_given  = fill;
26055f44c854SHong Zhang   if (ai[am] != 0) {
2606f284f12aSHong Zhang     fact->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]);
26075f44c854SHong Zhang   } else {
2608f284f12aSHong Zhang     fact->info.fill_ratio_needed = 0.0;
26095f44c854SHong Zhang   }
261035233d99SShri Abhyankar   fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ;
26115f44c854SHong Zhang   PetscFunctionReturn(0);
26125f44c854SHong Zhang }
26135f44c854SHong Zhang 
26145f44c854SHong Zhang #undef __FUNCT__
2615ad04f41aSHong Zhang #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ_inplace"
2616ad04f41aSHong Zhang PetscErrorCode MatICCFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
26175f44c854SHong Zhang {
26185f44c854SHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
26195f44c854SHong Zhang   Mat_SeqSBAIJ       *b;
26205f44c854SHong Zhang   PetscErrorCode     ierr;
26215f44c854SHong Zhang   PetscTruth         perm_identity,missing;
26225f44c854SHong Zhang   PetscInt           reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag;
26235f44c854SHong Zhang   const PetscInt     *rip,*riip;
26245f44c854SHong Zhang   PetscInt           jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow;
26255f44c854SHong Zhang   PetscInt           nlnk,*lnk,*lnk_lvl=PETSC_NULL,d;
26265f44c854SHong Zhang   PetscInt           ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr;
26275f44c854SHong Zhang   PetscReal          fill=info->fill,levels=info->levels;
26285f44c854SHong Zhang   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
26295f44c854SHong Zhang   PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL;
26305f44c854SHong Zhang   PetscBT            lnkbt;
26315f44c854SHong Zhang   IS                 iperm;
26325f44c854SHong Zhang 
26335f44c854SHong Zhang   PetscFunctionBegin;
2634e32f2f54SBarry Smith   if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n);
26355f44c854SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr);
2636e32f2f54SBarry Smith   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d);
26375f44c854SHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
26385f44c854SHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
26395f44c854SHong Zhang 
26405f44c854SHong Zhang   ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr);
26415f44c854SHong Zhang   ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr);
26425f44c854SHong Zhang   ui[0] = 0;
26435f44c854SHong Zhang 
26445f44c854SHong Zhang   /* ICC(0) without matrix ordering: simply copies fill pattern */
26455f44c854SHong Zhang   if (!levels && perm_identity) {
26465f44c854SHong Zhang 
26475f44c854SHong Zhang     for (i=0; i<am; i++) {
26485f44c854SHong Zhang       ui[i+1]  = ui[i] + ai[i+1] - a->diag[i];
26495f44c854SHong Zhang       udiag[i] = ui[i];
2650ed59401aSHong Zhang     }
265139e3d630SHong Zhang     ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
265239e3d630SHong Zhang     cols = uj;
2653ed59401aSHong Zhang     for (i=0; i<am; i++) {
2654ed59401aSHong Zhang       aj    = a->j + a->diag[i];
265539e3d630SHong Zhang       ncols = ui[i+1] - ui[i];
265639e3d630SHong Zhang       for (j=0; j<ncols; j++) *cols++ = *aj++;
2657ed59401aSHong Zhang     }
265839e3d630SHong Zhang   } else { /* case: levels>0 || (levels=0 && !perm_identity) */
2659b635d36bSHong Zhang     ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
2660b635d36bSHong Zhang     ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
2661b635d36bSHong Zhang 
26627a48dd6fSHong Zhang     /* initialization */
26635a8e39fbSHong Zhang     ierr  = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr);
26647a48dd6fSHong Zhang 
26657a48dd6fSHong Zhang     /* jl: linked list for storing indices of the pivot rows
26667a48dd6fSHong Zhang        il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
26670e83c824SBarry Smith     ierr = PetscMalloc4(am,PetscInt*,&uj_ptr,am,PetscInt*,&uj_lvl_ptr,am,PetscInt,&jl,am,PetscInt,&il);CHKERRQ(ierr);
26687a48dd6fSHong Zhang     for (i=0; i<am; i++){
26697a48dd6fSHong Zhang       jl[i] = am; il[i] = 0;
26707a48dd6fSHong Zhang     }
26717a48dd6fSHong Zhang 
26727a48dd6fSHong Zhang     /* create and initialize a linked list for storing column indices of the active row k */
26737a48dd6fSHong Zhang     nlnk = am + 1;
26742ed133dbSHong Zhang     ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
26757a48dd6fSHong Zhang 
26767a48dd6fSHong Zhang     /* initial FreeSpace size is fill*(ai[am]+1) */
2677a1a86e44SBarry Smith     ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr);
26787a48dd6fSHong Zhang     current_space = free_space;
2679a1a86e44SBarry Smith     ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space_lvl);CHKERRQ(ierr);
26807a48dd6fSHong Zhang     current_space_lvl = free_space_lvl;
26817a48dd6fSHong Zhang 
26827a48dd6fSHong Zhang     for (k=0; k<am; k++){  /* for each active row k */
26837a48dd6fSHong Zhang       /* initialize lnk by the column indices of row rip[k] of A */
26847a48dd6fSHong Zhang       nzk   = 0;
2685622e2028SHong Zhang       ncols = ai[rip[k]+1] - ai[rip[k]];
2686e32f2f54SBarry Smith       if (!ncols) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_MAT_CH_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",rip[k],k);
2687622e2028SHong Zhang       ncols_upper = 0;
2688622e2028SHong Zhang       for (j=0; j<ncols; j++){
2689b635d36bSHong Zhang         i = *(aj + ai[rip[k]] + j); /* unpermuted column index */
2690b635d36bSHong Zhang         if (riip[i] >= k){ /* only take upper triangular entry */
26915a8e39fbSHong Zhang           ajtmp[ncols_upper] = i;
2692622e2028SHong Zhang           ncols_upper++;
2693622e2028SHong Zhang         }
2694622e2028SHong Zhang       }
2695b635d36bSHong Zhang       ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
26967a48dd6fSHong Zhang       nzk += nlnk;
26977a48dd6fSHong Zhang 
26987a48dd6fSHong Zhang       /* update lnk by computing fill-in for each pivot row to be merged in */
26997a48dd6fSHong Zhang       prow = jl[k]; /* 1st pivot row */
27007a48dd6fSHong Zhang 
27017a48dd6fSHong Zhang       while (prow < k){
27027a48dd6fSHong Zhang         nextprow = jl[prow];
27037a48dd6fSHong Zhang 
27047a48dd6fSHong Zhang         /* merge prow into k-th row */
27057a48dd6fSHong Zhang         jmin = il[prow] + 1;  /* index of the 2nd nzero entry in U(prow,k:am-1) */
27067a48dd6fSHong Zhang         jmax = ui[prow+1];
27077a48dd6fSHong Zhang         ncols = jmax-jmin;
2708ed59401aSHong Zhang         i     = jmin - ui[prow];
2709ed59401aSHong Zhang         cols  = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */
27105a8e39fbSHong Zhang         uj    = uj_lvl_ptr[prow] + i; /* levels of cols */
27115a8e39fbSHong Zhang         j     = *(uj - 1);
27125a8e39fbSHong Zhang         ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr);
27137a48dd6fSHong Zhang         nzk += nlnk;
27147a48dd6fSHong Zhang 
27157a48dd6fSHong Zhang         /* update il and jl for prow */
27167a48dd6fSHong Zhang         if (jmin < jmax){
27177a48dd6fSHong Zhang           il[prow] = jmin;
27187a48dd6fSHong Zhang           j = *cols; jl[prow] = jl[j]; jl[j] = prow;
27197a48dd6fSHong Zhang         }
27207a48dd6fSHong Zhang         prow = nextprow;
27217a48dd6fSHong Zhang       }
27227a48dd6fSHong Zhang 
27237a48dd6fSHong Zhang       /* if free space is not available, make more free space */
27247a48dd6fSHong Zhang       if (current_space->local_remaining<nzk) {
27257a48dd6fSHong Zhang         i = am - k + 1; /* num of unfactored rows */
2726fd3f55cbSMatthew Knepley         i *= PetscMin(nzk, (i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */
2727a1a86e44SBarry Smith         ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
2728a1a86e44SBarry Smith         ierr = PetscFreeSpaceGet(i,&current_space_lvl);CHKERRQ(ierr);
27297a48dd6fSHong Zhang         reallocs++;
27307a48dd6fSHong Zhang       }
27317a48dd6fSHong Zhang 
27327a48dd6fSHong Zhang       /* copy data into free_space and free_space_lvl, then initialize lnk */
273365e19b50SBarry Smith       if (!nzk) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k);
27342ed133dbSHong Zhang       ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
27357a48dd6fSHong Zhang 
27367a48dd6fSHong Zhang       /* add the k-th row into il and jl */
273739e3d630SHong Zhang       if (nzk > 1){
27387a48dd6fSHong Zhang         i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
27397a48dd6fSHong Zhang         jl[k] = jl[i]; jl[i] = k;
27407a48dd6fSHong Zhang         il[k] = ui[k] + 1;
27417a48dd6fSHong Zhang       }
27427a48dd6fSHong Zhang       uj_ptr[k]     = current_space->array;
27437a48dd6fSHong Zhang       uj_lvl_ptr[k] = current_space_lvl->array;
27447a48dd6fSHong Zhang 
27457a48dd6fSHong Zhang       current_space->array           += nzk;
27467a48dd6fSHong Zhang       current_space->local_used      += nzk;
27477a48dd6fSHong Zhang       current_space->local_remaining -= nzk;
27487a48dd6fSHong Zhang 
27497a48dd6fSHong Zhang       current_space_lvl->array           += nzk;
27507a48dd6fSHong Zhang       current_space_lvl->local_used      += nzk;
27517a48dd6fSHong Zhang       current_space_lvl->local_remaining -= nzk;
27527a48dd6fSHong Zhang 
27537a48dd6fSHong Zhang       ui[k+1] = ui[k] + nzk;
27547a48dd6fSHong Zhang     }
27557a48dd6fSHong Zhang 
27566cf91177SBarry Smith #if defined(PETSC_USE_INFO)
27577a48dd6fSHong Zhang     if (ai[am] != 0) {
275839e3d630SHong Zhang       PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]);
2759ae15b995SBarry Smith       ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr);
2760ae15b995SBarry Smith       ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
2761ae15b995SBarry Smith       ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr);
27627a48dd6fSHong Zhang     } else {
2763ae15b995SBarry Smith       ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr);
27647a48dd6fSHong Zhang     }
276563ba0a88SBarry Smith #endif
27667a48dd6fSHong Zhang 
27677a48dd6fSHong Zhang     ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
2768b635d36bSHong Zhang     ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
27690e83c824SBarry Smith     ierr = PetscFree4(uj_ptr,uj_lvl_ptr,jl,il);CHKERRQ(ierr);
27705a8e39fbSHong Zhang     ierr = PetscFree(ajtmp);CHKERRQ(ierr);
27717a48dd6fSHong Zhang 
27727a48dd6fSHong Zhang     /* destroy list of free space and other temporary array(s) */
27737a48dd6fSHong Zhang     ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
2774a1a86e44SBarry Smith     ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr);
27752ed133dbSHong Zhang     ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
2776a1a86e44SBarry Smith     ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
27777a48dd6fSHong Zhang 
277839e3d630SHong Zhang   } /* end of case: levels>0 || (levels=0 && !perm_identity) */
277939e3d630SHong Zhang 
27807a48dd6fSHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
27817a48dd6fSHong Zhang 
2782f284f12aSHong Zhang   b    = (Mat_SeqSBAIJ*)fact->data;
27837a48dd6fSHong Zhang   b->singlemalloc = PETSC_FALSE;
27847a48dd6fSHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr);
27857a48dd6fSHong Zhang   b->j    = uj;
27867a48dd6fSHong Zhang   b->i    = ui;
27875f44c854SHong Zhang   b->diag = udiag;
27887f53bb6cSHong Zhang   b->free_diag = PETSC_TRUE;
27897a48dd6fSHong Zhang   b->ilen = 0;
27907a48dd6fSHong Zhang   b->imax = 0;
27917a48dd6fSHong Zhang   b->row  = perm;
2792b635d36bSHong Zhang   b->col  = perm;
2793b635d36bSHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
2794b635d36bSHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
2795b635d36bSHong Zhang   b->icol = iperm;
27967a48dd6fSHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
27977a48dd6fSHong Zhang   ierr    = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
2798f284f12aSHong Zhang   ierr = PetscLogObjectMemory(fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
27997a48dd6fSHong Zhang   b->maxnz   = b->nz = ui[am];
2800e6b907acSBarry Smith   b->free_a  = PETSC_TRUE;
2801e6b907acSBarry Smith   b->free_ij = PETSC_TRUE;
28027a48dd6fSHong Zhang 
2803f284f12aSHong Zhang   fact->info.factor_mallocs    = reallocs;
2804f284f12aSHong Zhang   fact->info.fill_ratio_given  = fill;
28057a48dd6fSHong Zhang   if (ai[am] != 0) {
2806f284f12aSHong Zhang     fact->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]);
28077a48dd6fSHong Zhang   } else {
2808f284f12aSHong Zhang     fact->info.fill_ratio_needed = 0.0;
28097a48dd6fSHong Zhang   }
28100a3c351aSHong Zhang   fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_inplace;
2811a6175056SHong Zhang   PetscFunctionReturn(0);
2812a6175056SHong Zhang }
2813d5d45c9bSBarry Smith 
281435233d99SShri Abhyankar PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
28150be760fbSHong Zhang {
28160be760fbSHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
28170be760fbSHong Zhang   Mat_SeqSBAIJ       *b;
28180be760fbSHong Zhang   PetscErrorCode     ierr;
28190be760fbSHong Zhang   PetscTruth         perm_identity;
28200be760fbSHong Zhang   PetscReal          fill = info->fill;
28210be760fbSHong Zhang   const PetscInt     *rip,*riip;
28220be760fbSHong Zhang   PetscInt           i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow;
28230be760fbSHong Zhang   PetscInt           *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow;
28240be760fbSHong Zhang   PetscInt           nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr,*udiag;
28250be760fbSHong Zhang   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
28260be760fbSHong Zhang   PetscBT            lnkbt;
28270be760fbSHong Zhang   IS                 iperm;
28280be760fbSHong Zhang 
28290be760fbSHong Zhang   PetscFunctionBegin;
2830e32f2f54SBarry Smith   if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n);
28310be760fbSHong Zhang   /* check whether perm is the identity mapping */
28320be760fbSHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
28330be760fbSHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
28340be760fbSHong Zhang   ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
28350be760fbSHong Zhang   ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
28360be760fbSHong Zhang 
28370be760fbSHong Zhang   /* initialization */
28380be760fbSHong Zhang   ierr  = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr);
28390be760fbSHong Zhang   ierr  = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr);
28400be760fbSHong Zhang   ui[0] = 0;
28410be760fbSHong Zhang 
28420be760fbSHong Zhang   /* jl: linked list for storing indices of the pivot rows
28430be760fbSHong Zhang      il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
28440be760fbSHong Zhang   ierr = PetscMalloc4(am,PetscInt*,&ui_ptr,am,PetscInt,&jl,am,PetscInt,&il,am,PetscInt,&cols);CHKERRQ(ierr);
28450be760fbSHong Zhang   for (i=0; i<am; i++){
28460be760fbSHong Zhang     jl[i] = am; il[i] = 0;
28470be760fbSHong Zhang   }
28480be760fbSHong Zhang 
28490be760fbSHong Zhang   /* create and initialize a linked list for storing column indices of the active row k */
28500be760fbSHong Zhang   nlnk = am + 1;
28510be760fbSHong Zhang   ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
28520be760fbSHong Zhang 
28530be760fbSHong Zhang   /* initial FreeSpace size is fill*(ai[am]+1) */
28540be760fbSHong Zhang   ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr);
28550be760fbSHong Zhang   current_space = free_space;
28560be760fbSHong Zhang 
28570be760fbSHong Zhang   for (k=0; k<am; k++){  /* for each active row k */
28580be760fbSHong Zhang     /* initialize lnk by the column indices of row rip[k] of A */
28590be760fbSHong Zhang     nzk   = 0;
28600be760fbSHong Zhang     ncols = ai[rip[k]+1] - ai[rip[k]];
2861e32f2f54SBarry Smith     if (!ncols) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_MAT_CH_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",rip[k],k);
28620be760fbSHong Zhang     ncols_upper = 0;
28630be760fbSHong Zhang     for (j=0; j<ncols; j++){
28640be760fbSHong Zhang       i = riip[*(aj + ai[rip[k]] + j)];
28650be760fbSHong Zhang       if (i >= k){ /* only take upper triangular entry */
28660be760fbSHong Zhang         cols[ncols_upper] = i;
28670be760fbSHong Zhang         ncols_upper++;
28680be760fbSHong Zhang       }
28690be760fbSHong Zhang     }
28700be760fbSHong Zhang     ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
28710be760fbSHong Zhang     nzk += nlnk;
28720be760fbSHong Zhang 
28730be760fbSHong Zhang     /* update lnk by computing fill-in for each pivot row to be merged in */
28740be760fbSHong Zhang     prow = jl[k]; /* 1st pivot row */
28750be760fbSHong Zhang 
28760be760fbSHong Zhang     while (prow < k){
28770be760fbSHong Zhang       nextprow = jl[prow];
28780be760fbSHong Zhang       /* merge prow into k-th row */
28790be760fbSHong Zhang       jmin = il[prow] + 1;  /* index of the 2nd nzero entry in U(prow,k:am-1) */
28800be760fbSHong Zhang       jmax = ui[prow+1];
28810be760fbSHong Zhang       ncols = jmax-jmin;
28820be760fbSHong Zhang       uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */
28830be760fbSHong Zhang       ierr = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
28840be760fbSHong Zhang       nzk += nlnk;
28850be760fbSHong Zhang 
28860be760fbSHong Zhang       /* update il and jl for prow */
28870be760fbSHong Zhang       if (jmin < jmax){
28880be760fbSHong Zhang         il[prow] = jmin;
28890be760fbSHong Zhang         j = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow;
28900be760fbSHong Zhang       }
28910be760fbSHong Zhang       prow = nextprow;
28920be760fbSHong Zhang     }
28930be760fbSHong Zhang 
28940be760fbSHong Zhang     /* if free space is not available, make more free space */
28950be760fbSHong Zhang     if (current_space->local_remaining<nzk) {
28960be760fbSHong Zhang       i  = am - k + 1; /* num of unfactored rows */
2897c6ae9bfcSHong Zhang       i *= PetscMin(nzk,i-1); /* i*nzk, i*(i-1): estimated and max additional space needed */
28980be760fbSHong Zhang       ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
28990be760fbSHong Zhang       reallocs++;
29000be760fbSHong Zhang     }
29010be760fbSHong Zhang 
29020be760fbSHong Zhang     /* copy data into free space, then initialize lnk */
29030be760fbSHong Zhang     ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
29040be760fbSHong Zhang 
29050be760fbSHong Zhang     /* add the k-th row into il and jl */
29067b056e98SHong Zhang     if (nzk > 1){
29070be760fbSHong Zhang       i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
29080be760fbSHong Zhang       jl[k] = jl[i]; jl[i] = k;
29090be760fbSHong Zhang       il[k] = ui[k] + 1;
29100be760fbSHong Zhang     }
29110be760fbSHong Zhang     ui_ptr[k] = current_space->array;
29120be760fbSHong Zhang     current_space->array           += nzk;
29130be760fbSHong Zhang     current_space->local_used      += nzk;
29140be760fbSHong Zhang     current_space->local_remaining -= nzk;
29150be760fbSHong Zhang 
29160be760fbSHong Zhang     ui[k+1] = ui[k] + nzk;
29170be760fbSHong Zhang   }
29180be760fbSHong Zhang 
29190be760fbSHong Zhang #if defined(PETSC_USE_INFO)
29200be760fbSHong Zhang   if (ai[am] != 0) {
29210be760fbSHong Zhang     PetscReal af = (PetscReal)(ui[am])/((PetscReal)ai[am]);
29220be760fbSHong Zhang     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr);
29230be760fbSHong Zhang     ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
29240be760fbSHong Zhang     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr);
29250be760fbSHong Zhang   } else {
29260be760fbSHong Zhang      ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr);
29270be760fbSHong Zhang   }
29280be760fbSHong Zhang #endif
29290be760fbSHong Zhang 
29300be760fbSHong Zhang   ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
29310be760fbSHong Zhang   ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
29320be760fbSHong Zhang   ierr = PetscFree4(ui_ptr,jl,il,cols);CHKERRQ(ierr);
29330be760fbSHong Zhang 
29340be760fbSHong Zhang   /* destroy list of free space and other temporary array(s) */
29350be760fbSHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
293635233d99SShri Abhyankar   ierr = PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag);CHKERRQ(ierr); /* store matrix factor */
29370be760fbSHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
29380be760fbSHong Zhang 
29390be760fbSHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
29400be760fbSHong Zhang 
2941f284f12aSHong Zhang   b = (Mat_SeqSBAIJ*)fact->data;
29420be760fbSHong Zhang   b->singlemalloc = PETSC_FALSE;
29430be760fbSHong Zhang   b->free_a       = PETSC_TRUE;
29440be760fbSHong Zhang   b->free_ij      = PETSC_TRUE;
29450be760fbSHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr);
29460be760fbSHong Zhang   b->j    = uj;
29470be760fbSHong Zhang   b->i    = ui;
29480be760fbSHong Zhang   b->diag = udiag;
29490be760fbSHong Zhang   b->free_diag = PETSC_TRUE;
29500be760fbSHong Zhang   b->ilen = 0;
29510be760fbSHong Zhang   b->imax = 0;
29520be760fbSHong Zhang   b->row  = perm;
29530be760fbSHong Zhang   b->col  = perm;
29540be760fbSHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
29550be760fbSHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
29560be760fbSHong Zhang   b->icol = iperm;
29570be760fbSHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
29580be760fbSHong Zhang   ierr    = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
29590860e08cSHong Zhang   ierr    = PetscLogObjectMemory(fact,ui[am]*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
29600be760fbSHong Zhang   b->maxnz = b->nz = ui[am];
29610be760fbSHong Zhang 
2962f284f12aSHong Zhang   fact->info.factor_mallocs    = reallocs;
2963f284f12aSHong Zhang   fact->info.fill_ratio_given  = fill;
29640be760fbSHong Zhang   if (ai[am] != 0) {
2965f284f12aSHong Zhang     fact->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]);
29660be760fbSHong Zhang   } else {
2967f284f12aSHong Zhang     fact->info.fill_ratio_needed = 0.0;
29680be760fbSHong Zhang   }
296935233d99SShri Abhyankar   fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ;
29700be760fbSHong Zhang   PetscFunctionReturn(0);
29710be760fbSHong Zhang }
29720be760fbSHong Zhang 
2973f76d2b81SHong Zhang #undef __FUNCT__
2974ad04f41aSHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_SeqAIJ_inplace"
2975ad04f41aSHong Zhang PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
2976f76d2b81SHong Zhang {
2977f76d2b81SHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
297810c27e3dSHong Zhang   Mat_SeqSBAIJ       *b;
2979dfbe8321SBarry Smith   PetscErrorCode     ierr;
2980f76d2b81SHong Zhang   PetscTruth         perm_identity;
298110c27e3dSHong Zhang   PetscReal          fill = info->fill;
29825d0c19d7SBarry Smith   const PetscInt     *rip,*riip;
29835d0c19d7SBarry Smith   PetscInt           i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow;
298410c27e3dSHong Zhang   PetscInt           *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow;
29852ed133dbSHong Zhang   PetscInt           nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr;
2986a1a86e44SBarry Smith   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
298710c27e3dSHong Zhang   PetscBT            lnkbt;
29882ed133dbSHong Zhang   IS                 iperm;
2989f76d2b81SHong Zhang 
2990f76d2b81SHong Zhang   PetscFunctionBegin;
2991e32f2f54SBarry Smith   if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n);
29922ed133dbSHong Zhang   /* check whether perm is the identity mapping */
2993f76d2b81SHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
29942ed133dbSHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
29952ed133dbSHong Zhang   ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
29969bfd6278SHong Zhang   ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
299710c27e3dSHong Zhang 
299810c27e3dSHong Zhang   /* initialization */
29991393bc97SHong Zhang   ierr  = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr);
300010c27e3dSHong Zhang   ui[0] = 0;
300110c27e3dSHong Zhang 
300210c27e3dSHong Zhang   /* jl: linked list for storing indices of the pivot rows
30031393bc97SHong Zhang      il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
30040e83c824SBarry Smith   ierr = PetscMalloc4(am,PetscInt*,&ui_ptr,am,PetscInt,&jl,am,PetscInt,&il,am,PetscInt,&cols);CHKERRQ(ierr);
30051393bc97SHong Zhang   for (i=0; i<am; i++){
30061393bc97SHong Zhang     jl[i] = am; il[i] = 0;
3007f76d2b81SHong Zhang   }
300810c27e3dSHong Zhang 
300910c27e3dSHong Zhang   /* create and initialize a linked list for storing column indices of the active row k */
30101393bc97SHong Zhang   nlnk = am + 1;
30111393bc97SHong Zhang   ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
301210c27e3dSHong Zhang 
30131393bc97SHong Zhang   /* initial FreeSpace size is fill*(ai[am]+1) */
3014a1a86e44SBarry Smith   ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr);
301510c27e3dSHong Zhang   current_space = free_space;
301610c27e3dSHong Zhang 
30171393bc97SHong Zhang   for (k=0; k<am; k++){  /* for each active row k */
301810c27e3dSHong Zhang     /* initialize lnk by the column indices of row rip[k] of A */
301910c27e3dSHong Zhang     nzk   = 0;
30202ed133dbSHong Zhang     ncols = ai[rip[k]+1] - ai[rip[k]];
3021e32f2f54SBarry Smith     if (!ncols) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_MAT_CH_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",rip[k],k);
30222ed133dbSHong Zhang     ncols_upper = 0;
30232ed133dbSHong Zhang     for (j=0; j<ncols; j++){
30249bfd6278SHong Zhang       i = riip[*(aj + ai[rip[k]] + j)];
30252ed133dbSHong Zhang       if (i >= k){ /* only take upper triangular entry */
30262ed133dbSHong Zhang         cols[ncols_upper] = i;
30272ed133dbSHong Zhang         ncols_upper++;
30282ed133dbSHong Zhang       }
30292ed133dbSHong Zhang     }
30301393bc97SHong Zhang     ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
303110c27e3dSHong Zhang     nzk += nlnk;
303210c27e3dSHong Zhang 
303310c27e3dSHong Zhang     /* update lnk by computing fill-in for each pivot row to be merged in */
303410c27e3dSHong Zhang     prow = jl[k]; /* 1st pivot row */
303510c27e3dSHong Zhang 
303610c27e3dSHong Zhang     while (prow < k){
303710c27e3dSHong Zhang       nextprow = jl[prow];
303810c27e3dSHong Zhang       /* merge prow into k-th row */
30391393bc97SHong Zhang       jmin = il[prow] + 1;  /* index of the 2nd nzero entry in U(prow,k:am-1) */
304010c27e3dSHong Zhang       jmax = ui[prow+1];
304110c27e3dSHong Zhang       ncols = jmax-jmin;
30421393bc97SHong Zhang       uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */
30435a8e39fbSHong Zhang       ierr = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
304410c27e3dSHong Zhang       nzk += nlnk;
304510c27e3dSHong Zhang 
304610c27e3dSHong Zhang       /* update il and jl for prow */
304710c27e3dSHong Zhang       if (jmin < jmax){
304810c27e3dSHong Zhang         il[prow] = jmin;
30492ed133dbSHong Zhang         j = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow;
305010c27e3dSHong Zhang       }
305110c27e3dSHong Zhang       prow = nextprow;
305210c27e3dSHong Zhang     }
305310c27e3dSHong Zhang 
305410c27e3dSHong Zhang     /* if free space is not available, make more free space */
305510c27e3dSHong Zhang     if (current_space->local_remaining<nzk) {
30561393bc97SHong Zhang       i = am - k + 1; /* num of unfactored rows */
305710c27e3dSHong Zhang       i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */
3058a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
305910c27e3dSHong Zhang       reallocs++;
306010c27e3dSHong Zhang     }
306110c27e3dSHong Zhang 
306210c27e3dSHong Zhang     /* copy data into free space, then initialize lnk */
30631393bc97SHong Zhang     ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
306410c27e3dSHong Zhang 
306510c27e3dSHong Zhang     /* add the k-th row into il and jl */
306610c27e3dSHong Zhang     if (nzk-1 > 0){
30671393bc97SHong Zhang       i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
306810c27e3dSHong Zhang       jl[k] = jl[i]; jl[i] = k;
306910c27e3dSHong Zhang       il[k] = ui[k] + 1;
307010c27e3dSHong Zhang     }
30712ed133dbSHong Zhang     ui_ptr[k] = current_space->array;
307210c27e3dSHong Zhang     current_space->array           += nzk;
307310c27e3dSHong Zhang     current_space->local_used      += nzk;
307410c27e3dSHong Zhang     current_space->local_remaining -= nzk;
307510c27e3dSHong Zhang 
307610c27e3dSHong Zhang     ui[k+1] = ui[k] + nzk;
307710c27e3dSHong Zhang   }
307810c27e3dSHong Zhang 
30796cf91177SBarry Smith #if defined(PETSC_USE_INFO)
30801393bc97SHong Zhang   if (ai[am] != 0) {
30811393bc97SHong Zhang     PetscReal af = (PetscReal)(ui[am])/((PetscReal)ai[am]);
3082ae15b995SBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr);
3083ae15b995SBarry Smith     ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
3084ae15b995SBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr);
308510c27e3dSHong Zhang   } else {
3086ae15b995SBarry Smith      ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr);
308710c27e3dSHong Zhang   }
308863ba0a88SBarry Smith #endif
308910c27e3dSHong Zhang 
309010c27e3dSHong Zhang   ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
30919bfd6278SHong Zhang   ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
30920e83c824SBarry Smith   ierr = PetscFree4(ui_ptr,jl,il,cols);CHKERRQ(ierr);
309310c27e3dSHong Zhang 
309410c27e3dSHong Zhang   /* destroy list of free space and other temporary array(s) */
30951393bc97SHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
3096a1a86e44SBarry Smith   ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr);
309710c27e3dSHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
309810c27e3dSHong Zhang 
309910c27e3dSHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
310010c27e3dSHong Zhang 
3101f284f12aSHong Zhang   b = (Mat_SeqSBAIJ*)fact->data;
310210c27e3dSHong Zhang   b->singlemalloc = PETSC_FALSE;
3103e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
3104e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
31051393bc97SHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr);
310610c27e3dSHong Zhang   b->j    = uj;
310710c27e3dSHong Zhang   b->i    = ui;
310810c27e3dSHong Zhang   b->diag = 0;
310910c27e3dSHong Zhang   b->ilen = 0;
311010c27e3dSHong Zhang   b->imax = 0;
311110c27e3dSHong Zhang   b->row  = perm;
31129bfd6278SHong Zhang   b->col  = perm;
31139bfd6278SHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
31149bfd6278SHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
31159bfd6278SHong Zhang   b->icol = iperm;
311610c27e3dSHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
31171393bc97SHong Zhang   ierr    = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
3118719d5645SBarry Smith   ierr    = PetscLogObjectMemory(fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
31191393bc97SHong Zhang   b->maxnz = b->nz = ui[am];
312010c27e3dSHong Zhang 
3121f284f12aSHong Zhang   fact->info.factor_mallocs    = reallocs;
3122f284f12aSHong Zhang   fact->info.fill_ratio_given  = fill;
31231393bc97SHong Zhang   if (ai[am] != 0) {
3124f284f12aSHong Zhang     fact->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]);
312510c27e3dSHong Zhang   } else {
3126f284f12aSHong Zhang     fact->info.fill_ratio_needed = 0.0;
312710c27e3dSHong Zhang   }
31280a3c351aSHong Zhang   fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_inplace;
3129f76d2b81SHong Zhang   PetscFunctionReturn(0);
3130f76d2b81SHong Zhang }
3131599ef60dSHong Zhang 
31321d098868SHong Zhang #undef __FUNCT__
313335233d99SShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering"
313435233d99SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering(Mat A,Vec bb,Vec xx)
3135813a1f2bSShri Abhyankar {
3136813a1f2bSShri Abhyankar   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
3137813a1f2bSShri Abhyankar   PetscErrorCode    ierr;
3138813a1f2bSShri Abhyankar   PetscInt          n = A->rmap->n;
3139813a1f2bSShri Abhyankar   const PetscInt    *ai = a->i,*aj = a->j,*adiag = a->diag,*vi;
3140813a1f2bSShri Abhyankar   PetscScalar       *x,sum;
3141813a1f2bSShri Abhyankar   const PetscScalar *b;
3142813a1f2bSShri Abhyankar   const MatScalar   *aa = a->a,*v;
3143813a1f2bSShri Abhyankar   PetscInt          i,nz;
3144813a1f2bSShri Abhyankar 
3145813a1f2bSShri Abhyankar   PetscFunctionBegin;
3146813a1f2bSShri Abhyankar   if (!n) PetscFunctionReturn(0);
3147813a1f2bSShri Abhyankar 
3148813a1f2bSShri Abhyankar   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
3149813a1f2bSShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
3150813a1f2bSShri Abhyankar 
3151813a1f2bSShri Abhyankar   /* forward solve the lower triangular */
3152813a1f2bSShri Abhyankar   x[0] = b[0];
3153813a1f2bSShri Abhyankar   v    = aa;
3154813a1f2bSShri Abhyankar   vi   = aj;
3155813a1f2bSShri Abhyankar   for (i=1; i<n; i++) {
3156813a1f2bSShri Abhyankar     nz  = ai[i+1] - ai[i];
3157813a1f2bSShri Abhyankar     sum = b[i];
3158813a1f2bSShri Abhyankar     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
3159813a1f2bSShri Abhyankar     v  += nz;
3160813a1f2bSShri Abhyankar     vi += nz;
3161813a1f2bSShri Abhyankar     x[i] = sum;
3162813a1f2bSShri Abhyankar   }
3163813a1f2bSShri Abhyankar 
3164813a1f2bSShri Abhyankar   /* backward solve the upper triangular */
316562fbd6afSShri Abhyankar   for (i=n-1; i>=0; i--){
31663c0119dfSShri Abhyankar     v   = aa + adiag[i+1] + 1;
31673c0119dfSShri Abhyankar     vi  = aj + adiag[i+1] + 1;
3168813a1f2bSShri Abhyankar     nz = adiag[i] - adiag[i+1]-1;
3169813a1f2bSShri Abhyankar     sum = x[i];
3170813a1f2bSShri Abhyankar     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
31713c0119dfSShri Abhyankar     x[i] = sum*v[nz]; /* x[i]=aa[adiag[i]]*sum; v++; */
3172813a1f2bSShri Abhyankar   }
3173813a1f2bSShri Abhyankar 
3174813a1f2bSShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
3175813a1f2bSShri Abhyankar   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
3176813a1f2bSShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
3177813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
3178813a1f2bSShri Abhyankar }
3179813a1f2bSShri Abhyankar 
31801d098868SHong Zhang #undef __FUNCT__
318135233d99SShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ"
318235233d99SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ(Mat A,Vec bb,Vec xx)
3183f268cba8SShri Abhyankar {
3184f268cba8SShri Abhyankar   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
3185f268cba8SShri Abhyankar   IS                iscol = a->col,isrow = a->row;
3186f268cba8SShri Abhyankar   PetscErrorCode    ierr;
3187f268cba8SShri Abhyankar   PetscInt          i,n=A->rmap->n,*vi,*ai=a->i,*aj=a->j,*adiag = a->diag,nz;
3188f268cba8SShri Abhyankar   const PetscInt    *rout,*cout,*r,*c;
3189f268cba8SShri Abhyankar   PetscScalar       *x,*tmp,sum;
3190f268cba8SShri Abhyankar   const PetscScalar *b;
3191f268cba8SShri Abhyankar   const MatScalar   *aa = a->a,*v;
3192f268cba8SShri Abhyankar 
3193f268cba8SShri Abhyankar   PetscFunctionBegin;
3194f268cba8SShri Abhyankar   if (!n) PetscFunctionReturn(0);
3195f268cba8SShri Abhyankar 
3196f268cba8SShri Abhyankar   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
3197f268cba8SShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
3198f268cba8SShri Abhyankar   tmp  = a->solve_work;
3199f268cba8SShri Abhyankar 
3200f268cba8SShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
3201f268cba8SShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
3202f268cba8SShri Abhyankar 
3203f268cba8SShri Abhyankar   /* forward solve the lower triangular */
3204f268cba8SShri Abhyankar   tmp[0] = b[r[0]];
3205f268cba8SShri Abhyankar   v      = aa;
3206f268cba8SShri Abhyankar   vi     = aj;
3207f268cba8SShri Abhyankar   for (i=1; i<n; i++) {
3208f268cba8SShri Abhyankar     nz  = ai[i+1] - ai[i];
3209f268cba8SShri Abhyankar     sum = b[r[i]];
3210f268cba8SShri Abhyankar     PetscSparseDenseMinusDot(sum,tmp,v,vi,nz);
3211f268cba8SShri Abhyankar     tmp[i] = sum;
3212f268cba8SShri Abhyankar     v += nz; vi += nz;
3213f268cba8SShri Abhyankar   }
3214f268cba8SShri Abhyankar 
3215f268cba8SShri Abhyankar   /* backward solve the upper triangular */
3216f268cba8SShri Abhyankar   for (i=n-1; i>=0; i--){
32173c0119dfSShri Abhyankar     v   = aa + adiag[i+1]+1;
32183c0119dfSShri Abhyankar     vi  = aj + adiag[i+1]+1;
3219f268cba8SShri Abhyankar     nz  = adiag[i]-adiag[i+1]-1;
3220f268cba8SShri Abhyankar     sum = tmp[i];
3221f268cba8SShri Abhyankar     PetscSparseDenseMinusDot(sum,tmp,v,vi,nz);
3222f268cba8SShri Abhyankar     x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */
3223f268cba8SShri Abhyankar   }
3224f268cba8SShri Abhyankar 
3225f268cba8SShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
3226f268cba8SShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
3227f268cba8SShri Abhyankar   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
3228f268cba8SShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
3229f268cba8SShri Abhyankar   ierr = PetscLogFlops(2*a->nz - A->cmap->n);CHKERRQ(ierr);
3230f268cba8SShri Abhyankar   PetscFunctionReturn(0);
3231f268cba8SShri Abhyankar }
3232f268cba8SShri Abhyankar 
3233f268cba8SShri Abhyankar #undef __FUNCT__
32341d098868SHong Zhang #define __FUNCT__ "MatILUDTFactor_SeqAIJ"
3235fe97e370SBarry Smith /*
3236fe97e370SBarry Smith     This will get a new name and become a varient of MatILUFactor_SeqAIJ() there is no longer seperate functions in the matrix function table for dt factors
3237fe97e370SBarry Smith */
32381d098868SHong Zhang PetscErrorCode MatILUDTFactor_SeqAIJ(Mat A,IS isrow,IS iscol,const MatFactorInfo *info,Mat *fact)
32391d098868SHong Zhang {
32401d098868SHong Zhang   Mat                B = *fact;
3241599ef60dSHong Zhang   Mat_SeqAIJ         *a=(Mat_SeqAIJ*)A->data,*b;
3242599ef60dSHong Zhang   IS                 isicol;
3243599ef60dSHong Zhang   PetscErrorCode     ierr;
32441d098868SHong Zhang   const PetscInt     *r,*ic;
32451d098868SHong Zhang   PetscInt           i,n=A->rmap->n,*ai=a->i,*aj=a->j,*ajtmp,*adiag;
3246dc3a2fd3SHong Zhang   PetscInt           *bi,*bj,*bdiag,*bdiag_rev;
3247f61a948aSLisandro Dalcin   PetscInt           row,nzi,nzi_bl,nzi_bu,*im,nzi_al,nzi_au;
32481d098868SHong Zhang   PetscInt           nlnk,*lnk;
32491d098868SHong Zhang   PetscBT            lnkbt;
32501d098868SHong Zhang   PetscTruth         row_identity,icol_identity,both_identity;
32518aee2decSHong Zhang   MatScalar          *aatmp,*pv,*batmp,*ba,*rtmp,*pc,multiplier,*vtmp,diag_tmp;
32521d098868SHong Zhang   const PetscInt     *ics;
32531d098868SHong Zhang   PetscInt           j,nz,*pj,*bjtmp,k,ncut,*jtmp;
3254182b8fbaSHong Zhang   PetscReal          dt=info->dt,dtcol=info->dtcol,shift=info->shiftamount;
3255f61a948aSLisandro Dalcin   PetscInt           dtcount=(PetscInt)info->dtcount,nnz_max;
3256599ef60dSHong Zhang   PetscTruth         missing;
3257599ef60dSHong Zhang 
3258599ef60dSHong Zhang   PetscFunctionBegin;
3259f61a948aSLisandro Dalcin 
3260f61a948aSLisandro Dalcin   if (dt      == PETSC_DEFAULT) dt      = 0.005;
3261f61a948aSLisandro Dalcin   if (dtcol   == PETSC_DEFAULT) dtcol   = 0.01; /* XXX unused! */
3262f61a948aSLisandro Dalcin   if (dtcount == PETSC_DEFAULT) dtcount = (PetscInt)(1.5*a->rmax);
3263f61a948aSLisandro Dalcin 
32641d098868SHong Zhang   /* ------- symbolic factorization, can be reused ---------*/
32651d098868SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr);
3266e32f2f54SBarry Smith   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i);
32671d098868SHong Zhang   adiag=a->diag;
3268599ef60dSHong Zhang 
3269599ef60dSHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
3270599ef60dSHong Zhang 
3271599ef60dSHong Zhang   /* bdiag is location of diagonal in factor */
32721bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);     /* becomes b->diag */
32731bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag_rev);CHKERRQ(ierr); /* temporary */
3274599ef60dSHong Zhang 
32751d098868SHong Zhang   /* allocate row pointers bi */
32768fc3a347SHong Zhang   ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
32771d098868SHong Zhang 
32781d098868SHong Zhang   /* allocate bj and ba; max num of nonzero entries is (ai[n]+2*n*dtcount+2) */
3279393d3a68SHong Zhang   if (dtcount > n-1) dtcount = n-1; /* diagonal is excluded */
32801d098868SHong Zhang   nnz_max  = ai[n]+2*n*dtcount+2;
32818fc3a347SHong Zhang 
32828fc3a347SHong Zhang   ierr = PetscMalloc((nnz_max+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
32838fc3a347SHong Zhang   ierr = PetscMalloc((nnz_max+1)*sizeof(MatScalar),&ba);CHKERRQ(ierr);
3284599ef60dSHong Zhang 
3285599ef60dSHong Zhang   /* put together the new matrix */
3286599ef60dSHong Zhang   ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
3287599ef60dSHong Zhang   ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr);
3288f284f12aSHong Zhang   b    = (Mat_SeqAIJ*)B->data;
3289599ef60dSHong Zhang   b->free_a       = PETSC_TRUE;
3290599ef60dSHong Zhang   b->free_ij      = PETSC_TRUE;
3291599ef60dSHong Zhang   b->singlemalloc = PETSC_FALSE;
3292599ef60dSHong Zhang   b->a          = ba;
3293599ef60dSHong Zhang   b->j          = bj;
3294599ef60dSHong Zhang   b->i          = bi;
3295599ef60dSHong Zhang   b->diag       = bdiag;
3296599ef60dSHong Zhang   b->ilen       = 0;
3297599ef60dSHong Zhang   b->imax       = 0;
3298599ef60dSHong Zhang   b->row        = isrow;
3299599ef60dSHong Zhang   b->col        = iscol;
3300599ef60dSHong Zhang   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
3301599ef60dSHong Zhang   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
3302599ef60dSHong Zhang   b->icol       = isicol;
3303599ef60dSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
3304599ef60dSHong Zhang 
33051d098868SHong Zhang   ierr = PetscLogObjectMemory(B,nnz_max*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
33061d098868SHong Zhang   b->maxnz = nnz_max;
3307599ef60dSHong Zhang 
3308d5f3da31SBarry Smith   B->factortype            = MAT_FACTOR_ILUDT;
3309f284f12aSHong Zhang   B->info.factor_mallocs   = 0;
3310f284f12aSHong Zhang   B->info.fill_ratio_given = ((PetscReal)nnz_max)/((PetscReal)ai[n]);
33111d098868SHong Zhang   CHKMEMQ;
33121d098868SHong Zhang   /* ------- end of symbolic factorization ---------*/
3313599ef60dSHong Zhang 
3314599ef60dSHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
3315599ef60dSHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
3316599ef60dSHong Zhang   ics  = ic;
3317599ef60dSHong Zhang 
3318599ef60dSHong Zhang   /* linked list for storing column indices of the active row */
3319599ef60dSHong Zhang   nlnk = n + 1;
3320599ef60dSHong Zhang   ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
33211d098868SHong Zhang 
33221d098868SHong Zhang   /* im: used by PetscLLAddSortedLU(); jtmp: working array for column indices of active row */
33230e83c824SBarry Smith   ierr = PetscMalloc2(n,PetscInt,&im,n,PetscInt,&jtmp);CHKERRQ(ierr);
33241d098868SHong Zhang   /* rtmp, vtmp: working arrays for sparse and contiguous row entries of active row */
33250e83c824SBarry Smith   ierr = PetscMalloc2(n,MatScalar,&rtmp,n,MatScalar,&vtmp);CHKERRQ(ierr);
33260e83c824SBarry Smith   ierr = PetscMemzero(rtmp,n*sizeof(MatScalar));CHKERRQ(ierr);
3327599ef60dSHong Zhang 
3328599ef60dSHong Zhang   bi[0]    = 0;
33298fc3a347SHong Zhang   bdiag[0] = nnz_max-1; /* location of diag[0] in factor B */
3330dc3a2fd3SHong Zhang   bdiag_rev[n] = bdiag[0];
33318fc3a347SHong Zhang   bi[2*n+1] = bdiag[0]+1; /* endof bj and ba array */
3332599ef60dSHong Zhang   for (i=0; i<n; i++) {
3333599ef60dSHong Zhang     /* copy initial fill into linked list */
3334599ef60dSHong Zhang     nzi = 0; /* nonzeros for active row i */
33358369b28dSHong Zhang     nzi = ai[r[i]+1] - ai[r[i]];
3336e32f2f54SBarry Smith     if (!nzi) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i);
33378369b28dSHong Zhang     nzi_al = adiag[r[i]] - ai[r[i]];
33388369b28dSHong Zhang     nzi_au = ai[r[i]+1] - adiag[r[i]] -1;
3339599ef60dSHong Zhang     ajtmp = aj + ai[r[i]];
33408369b28dSHong Zhang     ierr = PetscLLAddPerm(nzi,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
3341599ef60dSHong Zhang 
3342599ef60dSHong Zhang     /* load in initial (unfactored row) */
33431d098868SHong Zhang     aatmp = a->a + ai[r[i]];
33448369b28dSHong Zhang     for (j=0; j<nzi; j++) {
33451d098868SHong Zhang       rtmp[ics[*ajtmp++]] = *aatmp++;
3346599ef60dSHong Zhang     }
3347599ef60dSHong Zhang 
3348599ef60dSHong Zhang     /* add pivot rows into linked list */
3349599ef60dSHong Zhang     row = lnk[n];
3350599ef60dSHong Zhang     while (row < i ) {
33511d098868SHong Zhang       nzi_bl = bi[row+1] - bi[row] + 1;
33521d098868SHong Zhang       bjtmp = bj + bdiag[row+1]+1; /* points to 1st column next to the diagonal in U */
33531d098868SHong Zhang       ierr  = PetscLLAddSortedLU(bjtmp,row,nlnk,lnk,lnkbt,i,nzi_bl,im);CHKERRQ(ierr);
3354599ef60dSHong Zhang       nzi  += nlnk;
3355599ef60dSHong Zhang       row   = lnk[row];
3356599ef60dSHong Zhang     }
3357599ef60dSHong Zhang 
33588369b28dSHong Zhang     /* copy data from lnk into jtmp, then initialize lnk */
33598369b28dSHong Zhang     ierr = PetscLLClean(n,n,nzi,lnk,jtmp,lnkbt);CHKERRQ(ierr);
3360599ef60dSHong Zhang 
3361599ef60dSHong Zhang     /* numerical factorization */
33628369b28dSHong Zhang     bjtmp = jtmp;
3363599ef60dSHong Zhang     row   = *bjtmp++; /* 1st pivot row */
3364599ef60dSHong Zhang     while  ( row < i ) {
3365599ef60dSHong Zhang       pc         = rtmp + row;
33663c2a7987SHong Zhang       pv         = ba + bdiag[row]; /* 1./(diag of the pivot row) */
33673c2a7987SHong Zhang       multiplier = (*pc) * (*pv);
3368599ef60dSHong Zhang       *pc        = multiplier;
33693c2a7987SHong Zhang       if (PetscAbsScalar(*pc) > dt){ /* apply tolerance dropping rule */
33701d098868SHong Zhang         pj         = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */
33711d098868SHong Zhang         pv         = ba + bdiag[row+1] + 1;
33721d098868SHong Zhang         /* if (multiplier < -1.0 or multiplier >1.0) printf("row/prow %d, %d, multiplier %g\n",i,row,multiplier); */
33731d098868SHong Zhang         nz         = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */
33741d098868SHong Zhang         for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++);
3375dc0b31edSSatish Balay         ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr);
3376599ef60dSHong Zhang       }
3377599ef60dSHong Zhang       row = *bjtmp++;
3378599ef60dSHong Zhang     }
3379599ef60dSHong Zhang 
33808369b28dSHong Zhang     /* copy sparse rtmp into contiguous vtmp; separate L and U part */
33818aee2decSHong Zhang     diag_tmp = rtmp[i];  /* save diagonal value - may not needed?? */
33828369b28dSHong Zhang     nzi_bl = 0; j = 0;
33838369b28dSHong Zhang     while (jtmp[j] < i){ /* Note: jtmp is sorted */
33848aee2decSHong Zhang       vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0;
33858369b28dSHong Zhang       nzi_bl++; j++;
33868369b28dSHong Zhang     }
33878369b28dSHong Zhang     nzi_bu = nzi - nzi_bl -1;
33888369b28dSHong Zhang     while (j < nzi){
33898aee2decSHong Zhang       vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0;
33908369b28dSHong Zhang       j++;
33918369b28dSHong Zhang     }
33928369b28dSHong Zhang 
3393599ef60dSHong Zhang     bjtmp = bj + bi[i];
3394599ef60dSHong Zhang     batmp = ba + bi[i];
33958369b28dSHong Zhang     /* apply level dropping rule to L part */
33968369b28dSHong Zhang     ncut = nzi_al + dtcount;
33978369b28dSHong Zhang     if (ncut < nzi_bl){
33988369b28dSHong Zhang       ierr = PetscSortSplit(ncut,nzi_bl,vtmp,jtmp);CHKERRQ(ierr);
33998369b28dSHong Zhang       ierr = PetscSortIntWithScalarArray(ncut,jtmp,vtmp);CHKERRQ(ierr);
3400599ef60dSHong Zhang     } else {
34018369b28dSHong Zhang       ncut = nzi_bl;
34028369b28dSHong Zhang     }
34038369b28dSHong Zhang     for (j=0; j<ncut; j++){
34048369b28dSHong Zhang       bjtmp[j] = jtmp[j];
34058369b28dSHong Zhang       batmp[j] = vtmp[j];
34066da515a1SHong Zhang       /* printf(" (%d,%g),",bjtmp[j],batmp[j]); */
34078369b28dSHong Zhang     }
34081d098868SHong Zhang     bi[i+1] = bi[i] + ncut;
34098369b28dSHong Zhang     nzi = ncut + 1;
34108369b28dSHong Zhang 
34118369b28dSHong Zhang     /* apply level dropping rule to U part */
34128369b28dSHong Zhang     ncut = nzi_au + dtcount;
34138369b28dSHong Zhang     if (ncut < nzi_bu){
34148369b28dSHong Zhang       ierr = PetscSortSplit(ncut,nzi_bu,vtmp+nzi_bl+1,jtmp+nzi_bl+1);CHKERRQ(ierr);
34158369b28dSHong Zhang       ierr = PetscSortIntWithScalarArray(ncut,jtmp+nzi_bl+1,vtmp+nzi_bl+1);CHKERRQ(ierr);
34168369b28dSHong Zhang     } else {
34178369b28dSHong Zhang       ncut = nzi_bu;
34188369b28dSHong Zhang     }
34198369b28dSHong Zhang     nzi += ncut;
34201d098868SHong Zhang 
34211d098868SHong Zhang     /* mark bdiagonal */
34221d098868SHong Zhang     bdiag[i+1]       = bdiag[i] - (ncut + 1);
3423dc3a2fd3SHong Zhang     bdiag_rev[n-i-1] = bdiag[i+1];
34248fc3a347SHong Zhang     bi[2*n - i]      = bi[2*n - i +1] - (ncut + 1);
34251d098868SHong Zhang     bjtmp = bj + bdiag[i];
34261d098868SHong Zhang     batmp = ba + bdiag[i];
34271d098868SHong Zhang     *bjtmp = i;
34288aee2decSHong Zhang     *batmp = diag_tmp; /* rtmp[i]; */
3429c9c72f8fSHong Zhang     if (*batmp == 0.0) {
3430c9c72f8fSHong Zhang       *batmp = dt+shift;
3431bd1bc851SBarry Smith       /* printf(" row %d add shift %g\n",i,shift); */
3432c9c72f8fSHong Zhang     }
3433b78a477dSHong Zhang     *batmp = 1.0/(*batmp); /* invert diagonal entries for simplier triangular solves */
34346da515a1SHong Zhang     /* printf(" (%d,%g),",*bjtmp,*batmp); */
34351d098868SHong Zhang 
34361d098868SHong Zhang     bjtmp = bj + bdiag[i+1]+1;
34371d098868SHong Zhang     batmp = ba + bdiag[i+1]+1;
34388369b28dSHong Zhang     for (k=0; k<ncut; k++){
34391d098868SHong Zhang       bjtmp[k] = jtmp[nzi_bl+1+k];
34401d098868SHong Zhang       batmp[k] = vtmp[nzi_bl+1+k];
34416da515a1SHong Zhang       /* printf(" (%d,%g),",bjtmp[k],batmp[k]); */
3442599ef60dSHong Zhang     }
34436da515a1SHong Zhang     /* printf("\n"); */
3444599ef60dSHong Zhang 
3445599ef60dSHong Zhang     im[i]   = nzi; /* used by PetscLLAddSortedLU() */
3446599ef60dSHong Zhang     /*
34471d098868SHong Zhang     printf("row %d: bi %d, bdiag %d\n",i,bi[i],bdiag[i]);
34488369b28dSHong Zhang     printf(" ----------------------------\n");
3449599ef60dSHong Zhang     */
34508369b28dSHong Zhang   } /* for (i=0; i<n; i++) */
34511d098868SHong Zhang   /* printf("end of L %d, beginning of U %d\n",bi[n],bdiag[n]); */
3452e32f2f54SBarry Smith   if (bi[n] >= bdiag[n]) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"end of L array %d cannot >= the beginning of U array %d",bi[n],bdiag[n]);
3453599ef60dSHong Zhang 
3454599ef60dSHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
3455599ef60dSHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
3456599ef60dSHong Zhang 
3457599ef60dSHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
34580e83c824SBarry Smith   ierr = PetscFree2(im,jtmp);CHKERRQ(ierr);
34590e83c824SBarry Smith   ierr = PetscFree2(rtmp,vtmp);CHKERRQ(ierr);
34601bfa06eaSJed Brown   ierr = PetscFree(bdiag_rev);CHKERRQ(ierr);
3461599ef60dSHong Zhang 
3462599ef60dSHong Zhang   ierr = PetscLogFlops(B->cmap->n);CHKERRQ(ierr);
34631d098868SHong Zhang   b->maxnz = b->nz = bi[n] + bdiag[0] - bdiag[n];
3464599ef60dSHong Zhang 
3465599ef60dSHong Zhang   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
3466599ef60dSHong Zhang   ierr = ISIdentity(isicol,&icol_identity);CHKERRQ(ierr);
3467599ef60dSHong Zhang   both_identity = (PetscTruth) (row_identity && icol_identity);
3468599ef60dSHong Zhang   if (row_identity && icol_identity) {
346935233d99SShri Abhyankar     B->ops->solve = MatSolve_SeqAIJ_NaturalOrdering;
3470599ef60dSHong Zhang   } else {
347135233d99SShri Abhyankar     B->ops->solve = MatSolve_SeqAIJ;
3472599ef60dSHong Zhang   }
3473599ef60dSHong Zhang 
34741d098868SHong Zhang   B->ops->solveadd          = 0;
34751d098868SHong Zhang   B->ops->solvetranspose    = 0;
34761d098868SHong Zhang   B->ops->solvetransposeadd = 0;
34771d098868SHong Zhang   B->ops->matsolve          = 0;
3478599ef60dSHong Zhang   B->assembled              = PETSC_TRUE;
3479599ef60dSHong Zhang   B->preallocated           = PETSC_TRUE;
3480599ef60dSHong Zhang   PetscFunctionReturn(0);
3481599ef60dSHong Zhang }
3482599ef60dSHong Zhang 
34833c2a7987SHong Zhang /* a wraper of MatILUDTFactor_SeqAIJ() */
34843c2a7987SHong Zhang #undef __FUNCT__
34853c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorSymbolic_SeqAIJ"
3486fe97e370SBarry Smith /*
3487fe97e370SBarry Smith     This will get a new name and become a varient of MatILUFactor_SeqAIJ() there is no longer seperate functions in the matrix function table for dt factors
3488fe97e370SBarry Smith */
3489fe97e370SBarry Smith 
34903c2a7987SHong Zhang PetscErrorCode PETSCMAT_DLLEXPORT MatILUDTFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS row,IS col,const MatFactorInfo *info)
34913c2a7987SHong Zhang {
34923c2a7987SHong Zhang   PetscErrorCode     ierr;
34933c2a7987SHong Zhang 
34943c2a7987SHong Zhang   PetscFunctionBegin;
34953c2a7987SHong Zhang   ierr = MatILUDTFactor_SeqAIJ(A,row,col,info,&fact);CHKERRQ(ierr);
34963c2a7987SHong Zhang   PetscFunctionReturn(0);
34973c2a7987SHong Zhang }
34983c2a7987SHong Zhang 
34993c2a7987SHong Zhang /*
35003c2a7987SHong Zhang    same as MatLUFactorNumeric_SeqAIJ(), except using contiguous array matrix factors
35013c2a7987SHong Zhang    - intend to replace existing MatLUFactorNumeric_SeqAIJ()
35023c2a7987SHong Zhang */
35033c2a7987SHong Zhang #undef __FUNCT__
35043c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorNumeric_SeqAIJ"
3505fe97e370SBarry Smith /*
3506fe97e370SBarry Smith     This will get a new name and become a varient of MatILUFactor_SeqAIJ() there is no longer seperate functions in the matrix function table for dt factors
3507fe97e370SBarry Smith */
3508fe97e370SBarry Smith 
35093c2a7987SHong Zhang PetscErrorCode PETSCMAT_DLLEXPORT MatILUDTFactorNumeric_SeqAIJ(Mat fact,Mat A,const MatFactorInfo *info)
35103c2a7987SHong Zhang {
35113c2a7987SHong Zhang   Mat            C=fact;
35123c2a7987SHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data;
35133c2a7987SHong Zhang   IS             isrow = b->row,isicol = b->icol;
35143c2a7987SHong Zhang   PetscErrorCode ierr;
35153c2a7987SHong Zhang   const PetscInt *r,*ic,*ics;
3516b78a477dSHong Zhang   PetscInt       i,j,k,n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j;
3517b78a477dSHong Zhang   PetscInt       *ajtmp,*bjtmp,nz,nzl,nzu,row,*bdiag = b->diag,*pj;
35183c2a7987SHong Zhang   MatScalar      *rtmp,*pc,multiplier,*v,*pv,*aa=a->a;
3519182b8fbaSHong Zhang   PetscReal      dt=info->dt,shift=info->shiftamount;
3520b78a477dSHong Zhang   PetscTruth     row_identity, col_identity;
35213c2a7987SHong Zhang 
35223c2a7987SHong Zhang   PetscFunctionBegin;
35233c2a7987SHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
35243c2a7987SHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
3525b78a477dSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr);
35263c2a7987SHong Zhang   ics  = ic;
35273c2a7987SHong Zhang 
35283c2a7987SHong Zhang   for (i=0; i<n; i++){
3529b78a477dSHong Zhang     /* initialize rtmp array */
3530b78a477dSHong Zhang     nzl   = bi[i+1] - bi[i];       /* num of nozeros in L(i,:) */
35313c2a7987SHong Zhang     bjtmp = bj + bi[i];
3532b78a477dSHong Zhang     for  (j=0; j<nzl; j++) rtmp[*bjtmp++] = 0.0;
3533b78a477dSHong Zhang     rtmp[i] = 0.0;
3534b78a477dSHong Zhang     nzu   = bdiag[i] - bdiag[i+1]; /* num of nozeros in U(i,:) */
3535b78a477dSHong Zhang     bjtmp = bj + bdiag[i+1] + 1;
3536b78a477dSHong Zhang     for  (j=0; j<nzu; j++) rtmp[*bjtmp++] = 0.0;
35373c2a7987SHong Zhang 
3538b78a477dSHong Zhang     /* load in initial unfactored row of A */
35396da515a1SHong Zhang     /* printf("row %d\n",i); */
35403c2a7987SHong Zhang     nz    = ai[r[i]+1] - ai[r[i]];
35413c2a7987SHong Zhang     ajtmp = aj + ai[r[i]];
35423c2a7987SHong Zhang     v     = aa + ai[r[i]];
35433c2a7987SHong Zhang     for (j=0; j<nz; j++) {
3544b78a477dSHong Zhang       rtmp[ics[*ajtmp++]] = v[j];
35456da515a1SHong Zhang       /* printf(" (%d,%g),",ics[ajtmp[j]],rtmp[ics[ajtmp[j]]]); */
35463c2a7987SHong Zhang     }
35476da515a1SHong Zhang     /* printf("\n"); */
35483c2a7987SHong Zhang 
3549b78a477dSHong Zhang     /* numerical factorization */
3550b78a477dSHong Zhang     bjtmp = bj + bi[i]; /* point to 1st entry of L(i,:) */
3551b78a477dSHong Zhang     nzl   = bi[i+1] - bi[i]; /* num of entries in L(i,:) */
3552b78a477dSHong Zhang     k = 0;
3553b78a477dSHong Zhang     while (k < nzl){
35543c2a7987SHong Zhang       row   = *bjtmp++;
35556da515a1SHong Zhang       /* printf("  prow %d\n",row); */
35563c2a7987SHong Zhang       pc         = rtmp + row;
3557b78a477dSHong Zhang       pv         = b->a + bdiag[row]; /* 1./(diag of the pivot row) */
3558b78a477dSHong Zhang       multiplier = (*pc) * (*pv);
35593c2a7987SHong Zhang       *pc        = multiplier;
3560b78a477dSHong Zhang       if (PetscAbsScalar(multiplier) > dt){
3561b78a477dSHong Zhang         pj         = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */
3562b78a477dSHong Zhang         pv         = b->a + bdiag[row+1] + 1;
3563b78a477dSHong Zhang         nz         = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */
3564b78a477dSHong Zhang         for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++);
35656da515a1SHong Zhang         /* ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); */
35663c2a7987SHong Zhang       }
3567b78a477dSHong Zhang       k++;
35683c2a7987SHong Zhang     }
35693c2a7987SHong Zhang 
3570b78a477dSHong Zhang     /* finished row so stick it into b->a */
3571b78a477dSHong Zhang     /* L-part */
3572b78a477dSHong Zhang     pv = b->a + bi[i] ;
3573b78a477dSHong Zhang     pj = bj + bi[i] ;
3574b78a477dSHong Zhang     nzl = bi[i+1] - bi[i];
3575b78a477dSHong Zhang     for (j=0; j<nzl; j++) {
3576b78a477dSHong Zhang       pv[j] = rtmp[pj[j]];
35776da515a1SHong Zhang       /* printf(" (%d,%g),",pj[j],pv[j]); */
35783c2a7987SHong Zhang     }
3579b78a477dSHong Zhang 
3580b78a477dSHong Zhang     /* diagonal: invert diagonal entries for simplier triangular solves */
3581b78a477dSHong Zhang     if (rtmp[i] == 0.0) rtmp[i] = dt+shift;
3582b78a477dSHong Zhang     b->a[bdiag[i]] = 1.0/rtmp[i];
35836da515a1SHong Zhang     /* printf(" (%d,%g),",i,b->a[bdiag[i]]); */
3584b78a477dSHong Zhang 
3585b78a477dSHong Zhang     /* U-part */
3586b78a477dSHong Zhang     pv = b->a + bdiag[i+1] + 1;
3587b78a477dSHong Zhang     pj = bj + bdiag[i+1] + 1;
3588b78a477dSHong Zhang     nzu = bdiag[i] - bdiag[i+1] - 1;
3589b78a477dSHong Zhang     for (j=0; j<nzu; j++) {
3590b78a477dSHong Zhang       pv[j] = rtmp[pj[j]];
35916da515a1SHong Zhang       /* printf(" (%d,%g),",pj[j],pv[j]); */
3592b78a477dSHong Zhang     }
35936da515a1SHong Zhang     /* printf("\n"); */
3594b78a477dSHong Zhang   }
3595b78a477dSHong Zhang 
35963c2a7987SHong Zhang   ierr = PetscFree(rtmp);CHKERRQ(ierr);
35973c2a7987SHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
35983c2a7987SHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
3599b78a477dSHong Zhang 
36003c2a7987SHong Zhang   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
36013c2a7987SHong Zhang   ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr);
36023c2a7987SHong Zhang   if (row_identity && col_identity) {
360335233d99SShri Abhyankar     C->ops->solve   = MatSolve_SeqAIJ_NaturalOrdering;
36043c2a7987SHong Zhang   } else {
360535233d99SShri Abhyankar     C->ops->solve   = MatSolve_SeqAIJ;
36063c2a7987SHong Zhang   }
3607b78a477dSHong Zhang   C->ops->solveadd           = 0;
3608b78a477dSHong Zhang   C->ops->solvetranspose     = 0;
3609b78a477dSHong Zhang   C->ops->solvetransposeadd  = 0;
3610b78a477dSHong Zhang   C->ops->matsolve           = 0;
36113c2a7987SHong Zhang   C->assembled    = PETSC_TRUE;
36123c2a7987SHong Zhang   C->preallocated = PETSC_TRUE;
36133c2a7987SHong Zhang   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
36143c2a7987SHong Zhang   PetscFunctionReturn(0);
36153c2a7987SHong Zhang }
3616