xref: /petsc/src/mat/impls/aij/seq/aijfact.c (revision e0527adcb6c7dd108cbb2b5ae92e7d7d2565f59c)
1b377110cSBarry Smith 
2c6db04a5SJed Brown #include <../src/mat/impls/aij/seq/aij.h>
3c6db04a5SJed Brown #include <../src/mat/impls/sbaij/seq/sbaij.h>
4c6db04a5SJed Brown #include <petscbt.h>
5c6db04a5SJed Brown #include <../src/mat/utils/freespace.h>
63b2fbd54SBarry Smith 
7f6a26d55SMatthew Knepley EXTERN_C_BEGIN
84a2ae208SSatish Balay #undef __FUNCT__
9a7a54a73SBarry Smith #define __FUNCT__ "MatGetOrdering_Flow_SeqAIJ"
108fa24674SBarry Smith /*
118fa24674SBarry Smith       Computes an ordering to get most of the large numerical values in the lower triangular part of the matrix
12312e372aSBarry Smith 
13312e372aSBarry Smith       This code does not work and is not called anywhere. It would be registered with MatOrderingRegisterAll()
148fa24674SBarry Smith */
15a7a54a73SBarry Smith PetscErrorCode MatGetOrdering_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;
22ace3abfcSBarry Smith   PetscBool         *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 
40ace3abfcSBarry Smith   ierr = PetscMalloc(n*sizeof(PetscBool),&done);CHKERRQ(ierr);
41ace3abfcSBarry Smith   ierr = PetscMemzero(done,n*sizeof(PetscBool));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   }
8670b3c8c7SBarry Smith   ierr = ISCreateGeneral(PETSC_COMM_SELF,n,order,PETSC_COPY_VALUES,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"
98ace3abfcSBarry Smith PetscErrorCode MatGetFactorAvailable_seqaij_petsc(Mat A,MatFactorType ftype,PetscBool  *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;
121*e0527adcSJed Brown     ierr = MatSetBlockSizes(*B,A->rmap->bs,A->cmap->bs);CHKERRQ(ierr);
122b24902e0SBarry Smith   } else if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) {
123b24902e0SBarry Smith     ierr = MatSetType(*B,MATSEQSBAIJ);CHKERRQ(ierr);
124b24902e0SBarry Smith     ierr = MatSeqSBAIJSetPreallocation(*B,1,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
12535233d99SShri Abhyankar     (*B)->ops->iccfactorsymbolic      = MatICCFactorSymbolic_SeqAIJ;
12635233d99SShri Abhyankar     (*B)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqAIJ;
127e32f2f54SBarry Smith   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Factor type not supported");
128d5f3da31SBarry Smith   (*B)->factortype = ftype;
129b24902e0SBarry Smith   PetscFunctionReturn(0);
130b24902e0SBarry Smith }
131e631078cSBarry Smith EXTERN_C_END
132b24902e0SBarry Smith 
133b24902e0SBarry Smith #undef __FUNCT__
134ad04f41aSHong Zhang #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ_inplace"
135ad04f41aSHong Zhang PetscErrorCode MatLUFactorSymbolic_SeqAIJ_inplace(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
136289bc588SBarry Smith {
137416022c9SBarry Smith   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
138289bc588SBarry Smith   IS                 isicol;
1396849ba73SBarry Smith   PetscErrorCode     ierr;
1405d0c19d7SBarry Smith   const PetscInt     *r,*ic;
1415d0c19d7SBarry Smith   PetscInt           i,n=A->rmap->n,*ai=a->i,*aj=a->j;
1421393bc97SHong Zhang   PetscInt           *bi,*bj,*ajtmp;
1431393bc97SHong Zhang   PetscInt           *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im;
1449e046878SBarry Smith   PetscReal          f;
1454875ba38SHong Zhang   PetscInt           nlnk,*lnk,k,**bi_ptr;
146a1a86e44SBarry Smith   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
1471393bc97SHong Zhang   PetscBT            lnkbt;
148289bc588SBarry Smith 
1493a40ed3dSBarry Smith   PetscFunctionBegin;
150e32f2f54SBarry Smith   if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"matrix must be square");
1514c49b128SBarry Smith   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
152ac121b3dSBarry Smith   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
153ac121b3dSBarry Smith   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
154289bc588SBarry Smith 
155289bc588SBarry Smith   /* get new row pointers */
1561393bc97SHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
1571393bc97SHong Zhang   bi[0] = 0;
1581393bc97SHong Zhang 
1591393bc97SHong Zhang   /* bdiag is location of diagonal in factor */
1601393bc97SHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
1611393bc97SHong Zhang   bdiag[0] = 0;
1621393bc97SHong Zhang 
1631393bc97SHong Zhang   /* linked list for storing column indices of the active row */
1641393bc97SHong Zhang   nlnk = n + 1;
1651393bc97SHong Zhang   ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
1661393bc97SHong Zhang 
16735baf27eSSatish Balay   ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr);
1681393bc97SHong Zhang 
1691393bc97SHong Zhang   /* initial FreeSpace size is f*(ai[n]+1) */
170d3d32019SBarry Smith   f = info->fill;
171a1a86e44SBarry Smith   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr);
1721393bc97SHong Zhang   current_space = free_space;
173289bc588SBarry Smith 
174289bc588SBarry Smith   for (i=0; i<n; i++) {
1751393bc97SHong Zhang     /* copy previous fill into linked list */
176289bc588SBarry Smith     nzi = 0;
1771393bc97SHong Zhang     nnz = ai[r[i]+1] - ai[r[i]];
178e32f2f54SBarry 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);
1791393bc97SHong Zhang     ajtmp = aj + ai[r[i]];
180afcc9904SHong Zhang     ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
1811393bc97SHong Zhang     nzi += nlnk;
1821393bc97SHong Zhang 
1831393bc97SHong Zhang     /* add pivot rows into linked list */
1841393bc97SHong Zhang     row = lnk[n];
1851393bc97SHong Zhang     while (row < i) {
186d1170560SHong Zhang       nzbd    = bdiag[row] - bi[row] + 1; /* num of entries in the row with column index <= row */
187d1170560SHong Zhang       ajtmp   = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */
188d1170560SHong Zhang       ierr = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr);
1891393bc97SHong Zhang       nzi += nlnk;
1901393bc97SHong Zhang       row  = lnk[row];
191289bc588SBarry Smith     }
1921393bc97SHong Zhang     bi[i+1] = bi[i] + nzi;
1931393bc97SHong Zhang     im[i]   = nzi;
1941393bc97SHong Zhang 
1951393bc97SHong Zhang     /* mark bdiag */
1961393bc97SHong Zhang     nzbd = 0;
1971393bc97SHong Zhang     nnz  = nzi;
1981393bc97SHong Zhang     k    = lnk[n];
1991393bc97SHong Zhang     while (nnz-- && k < i){
2001393bc97SHong Zhang       nzbd++;
2011393bc97SHong Zhang       k = lnk[k];
2021393bc97SHong Zhang     }
2031393bc97SHong Zhang     bdiag[i] = bi[i] + nzbd;
2041393bc97SHong Zhang 
2051393bc97SHong Zhang     /* if free space is not available, make more free space */
2061393bc97SHong Zhang     if (current_space->local_remaining<nzi) {
2071393bc97SHong Zhang       nnz = (n - i)*nzi; /* estimated and max additional space needed */
208a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
2091393bc97SHong Zhang       reallocs++;
2101393bc97SHong Zhang     }
2111393bc97SHong Zhang 
2121393bc97SHong Zhang     /* copy data into free space, then initialize lnk */
2131393bc97SHong Zhang     ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
2141393bc97SHong Zhang     bi_ptr[i] = current_space->array;
2151393bc97SHong Zhang     current_space->array           += nzi;
2161393bc97SHong Zhang     current_space->local_used      += nzi;
2171393bc97SHong Zhang     current_space->local_remaining -= nzi;
218289bc588SBarry Smith   }
2196cf91177SBarry Smith #if defined(PETSC_USE_INFO)
220464e8d44SSatish Balay   if (ai[n] != 0) {
2211393bc97SHong Zhang     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
222ae15b995SBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr);
223ae15b995SBarry Smith     ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
224ae15b995SBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr);
225ae15b995SBarry Smith     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
22605bf559cSSatish Balay   } else {
227ae15b995SBarry Smith     ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr);
22805bf559cSSatish Balay   }
22963ba0a88SBarry Smith #endif
2302fb3ed76SBarry Smith 
231898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
232898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
2331987afe7SBarry Smith 
2341393bc97SHong Zhang   /* destroy list of free space and other temporary array(s) */
2351393bc97SHong Zhang   ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
236a1a86e44SBarry Smith   ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr);
2371393bc97SHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
23835baf27eSSatish Balay   ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr);
239289bc588SBarry Smith 
240289bc588SBarry Smith   /* put together the new matrix */
241719d5645SBarry Smith   ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
242719d5645SBarry Smith   ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr);
243719d5645SBarry Smith   b    = (Mat_SeqAIJ*)(B)->data;
244e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
245e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
2467c922b88SBarry Smith   b->singlemalloc = PETSC_FALSE;
2471393bc97SHong Zhang   ierr          = PetscMalloc((bi[n]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);
2481393bc97SHong Zhang   b->j          = bj;
2491393bc97SHong Zhang   b->i          = bi;
2501393bc97SHong Zhang   b->diag       = bdiag;
251416022c9SBarry Smith   b->ilen       = 0;
252416022c9SBarry Smith   b->imax       = 0;
253416022c9SBarry Smith   b->row        = isrow;
254416022c9SBarry Smith   b->col        = iscol;
255c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
256c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
25782bf6240SBarry Smith   b->icol       = isicol;
25887828ca2SBarry Smith   ierr          = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
2591393bc97SHong Zhang 
2601393bc97SHong Zhang   /* In b structure:  Free imax, ilen, old a, old j.  Allocate solve_work, new a, new j */
261719d5645SBarry Smith   ierr = PetscLogObjectMemory(B,(bi[n]-n)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
2621393bc97SHong Zhang   b->maxnz = b->nz = bi[n] ;
2637fd98bd6SLois Curfman McInnes 
264d5f3da31SBarry Smith   (B)->factortype            = MAT_FACTOR_LU;
265719d5645SBarry Smith   (B)->info.factor_mallocs   = reallocs;
266719d5645SBarry Smith   (B)->info.fill_ratio_given = f;
267703deb49SSatish Balay 
2689f7d0b68SHong Zhang   if (ai[n]) {
269719d5645SBarry Smith     (B)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]);
270eea2913aSSatish Balay   } else {
271719d5645SBarry Smith     (B)->info.fill_ratio_needed = 0.0;
272eea2913aSSatish Balay   }
273ad04f41aSHong Zhang   (B)->ops->lufactornumeric  = MatLUFactorNumeric_SeqAIJ_inplace;
274f2344125SBarry Smith   if (a->inode.size) {
275f2344125SBarry Smith     (B)->ops->lufactornumeric  = MatLUFactorNumeric_SeqAIJ_Inode_inplace;
276f2344125SBarry Smith   }
2773a40ed3dSBarry Smith   PetscFunctionReturn(0);
278289bc588SBarry Smith }
2791393bc97SHong Zhang 
280ce3d78c0SShri Abhyankar #undef __FUNCT__
28135233d99SShri Abhyankar #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ"
28235233d99SShri Abhyankar PetscErrorCode MatLUFactorSymbolic_SeqAIJ(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
283ce3d78c0SShri Abhyankar {
284ce3d78c0SShri Abhyankar   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
285ce3d78c0SShri Abhyankar   IS                 isicol;
286ce3d78c0SShri Abhyankar   PetscErrorCode     ierr;
287ce3d78c0SShri Abhyankar   const PetscInt     *r,*ic;
288ce3d78c0SShri Abhyankar   PetscInt           i,n=A->rmap->n,*ai=a->i,*aj=a->j;
289ce3d78c0SShri Abhyankar   PetscInt           *bi,*bj,*ajtmp;
290ce3d78c0SShri Abhyankar   PetscInt           *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im;
291ce3d78c0SShri Abhyankar   PetscReal          f;
292ce3d78c0SShri Abhyankar   PetscInt           nlnk,*lnk,k,**bi_ptr;
293ce3d78c0SShri Abhyankar   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
294ce3d78c0SShri Abhyankar   PetscBT            lnkbt;
295ce3d78c0SShri Abhyankar 
296ce3d78c0SShri Abhyankar   PetscFunctionBegin;
2973898ab1dSSatish Balay   /* Uncomment the oldatastruct part only while testing new data structure for MatSolve() */
298a4be958cSHong Zhang   /*
2991a26ff17SHong Zhang   PetscBool          olddatastruct=PETSC_FALSE;
300acfcf0e5SJed Brown   ierr = PetscOptionsGetBool(PETSC_NULL,"-lu_old",&olddatastruct,PETSC_NULL);CHKERRQ(ierr);
301628f99d7SShri Abhyankar   if(olddatastruct){
302628f99d7SShri Abhyankar     ierr = MatLUFactorSymbolic_SeqAIJ_inplace(B,A,isrow,iscol,info);CHKERRQ(ierr);
303628f99d7SShri Abhyankar     PetscFunctionReturn(0);
304628f99d7SShri Abhyankar   }
305a4be958cSHong Zhang   */
306e32f2f54SBarry Smith   if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"matrix must be square");
307ce3d78c0SShri Abhyankar   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
308ce3d78c0SShri Abhyankar   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
309ce3d78c0SShri Abhyankar   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
310ce3d78c0SShri Abhyankar 
3111bfa06eaSJed Brown   /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */
3121bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
3131bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
314a6df321fSShri Abhyankar   bi[0] = bdiag[0] = 0;
3159b48462bSShri Abhyankar 
3169b48462bSShri Abhyankar   /* linked list for storing column indices of the active row */
3179b48462bSShri Abhyankar   nlnk = n + 1;
3189b48462bSShri Abhyankar   ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
3199b48462bSShri Abhyankar 
3209b48462bSShri Abhyankar   ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr);
3219b48462bSShri Abhyankar 
3229b48462bSShri Abhyankar   /* initial FreeSpace size is f*(ai[n]+1) */
3239b48462bSShri Abhyankar   f = info->fill;
3249b48462bSShri Abhyankar   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr);
3259b48462bSShri Abhyankar   current_space = free_space;
3269b48462bSShri Abhyankar 
3279b48462bSShri Abhyankar   for (i=0; i<n; i++) {
3289b48462bSShri Abhyankar     /* copy previous fill into linked list */
3299b48462bSShri Abhyankar     nzi = 0;
3309b48462bSShri Abhyankar     nnz = ai[r[i]+1] - ai[r[i]];
331e32f2f54SBarry 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);
3329b48462bSShri Abhyankar     ajtmp = aj + ai[r[i]];
3339b48462bSShri Abhyankar     ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
3349b48462bSShri Abhyankar     nzi += nlnk;
3359b48462bSShri Abhyankar 
3369b48462bSShri Abhyankar     /* add pivot rows into linked list */
3379b48462bSShri Abhyankar     row = lnk[n];
3389b48462bSShri Abhyankar     while (row < i){
3399b48462bSShri Abhyankar       nzbd  = bdiag[row] + 1; /* num of entries in the row with column index <= row */
3409b48462bSShri Abhyankar       ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */
3419b48462bSShri Abhyankar       ierr  = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr);
3429b48462bSShri Abhyankar       nzi  += nlnk;
3439b48462bSShri Abhyankar       row   = lnk[row];
3449b48462bSShri Abhyankar     }
3459b48462bSShri Abhyankar     bi[i+1] = bi[i] + nzi;
3469b48462bSShri Abhyankar     im[i]   = nzi;
3479b48462bSShri Abhyankar 
3489b48462bSShri Abhyankar     /* mark bdiag */
3499b48462bSShri Abhyankar     nzbd = 0;
3509b48462bSShri Abhyankar     nnz  = nzi;
3519b48462bSShri Abhyankar     k    = lnk[n];
3529b48462bSShri Abhyankar     while (nnz-- && k < i){
3539b48462bSShri Abhyankar       nzbd++;
3549b48462bSShri Abhyankar       k = lnk[k];
3559b48462bSShri Abhyankar     }
3569b48462bSShri Abhyankar     bdiag[i] = nzbd; /* note: bdiag[i] = nnzL as input for PetscFreeSpaceContiguous_LU() */
3579b48462bSShri Abhyankar 
3589b48462bSShri Abhyankar     /* if free space is not available, make more free space */
3599b48462bSShri Abhyankar     if (current_space->local_remaining<nzi) {
3609b48462bSShri Abhyankar       nnz = 2*(n - i)*nzi; /* estimated and max additional space needed */
3619b48462bSShri Abhyankar       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
3629b48462bSShri Abhyankar       reallocs++;
3639b48462bSShri Abhyankar     }
3649b48462bSShri Abhyankar 
3659b48462bSShri Abhyankar     /* copy data into free space, then initialize lnk */
3669b48462bSShri Abhyankar     ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
3679b48462bSShri Abhyankar     bi_ptr[i] = current_space->array;
3689b48462bSShri Abhyankar     current_space->array           += nzi;
3699b48462bSShri Abhyankar     current_space->local_used      += nzi;
3709b48462bSShri Abhyankar     current_space->local_remaining -= nzi;
3719b48462bSShri Abhyankar   }
3729b48462bSShri Abhyankar 
3739b48462bSShri Abhyankar   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
3749b48462bSShri Abhyankar   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
3759b48462bSShri Abhyankar 
3769263d837SHong Zhang   /*   copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */
3779b48462bSShri Abhyankar   ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
3782ce24eb6SHong Zhang   ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr);
3799b48462bSShri Abhyankar   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
3809b48462bSShri Abhyankar   ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr);
3819b48462bSShri Abhyankar 
3829b48462bSShri Abhyankar   /* put together the new matrix */
3839b48462bSShri Abhyankar   ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
3849b48462bSShri Abhyankar   ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr);
3859b48462bSShri Abhyankar   b    = (Mat_SeqAIJ*)(B)->data;
3869b48462bSShri Abhyankar   b->free_a       = PETSC_TRUE;
3879b48462bSShri Abhyankar   b->free_ij      = PETSC_TRUE;
3889b48462bSShri Abhyankar   b->singlemalloc = PETSC_FALSE;
3899b48462bSShri Abhyankar   ierr = PetscMalloc((bdiag[0]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);
3909b48462bSShri Abhyankar   b->j          = bj;
3919b48462bSShri Abhyankar   b->i          = bi;
3929b48462bSShri Abhyankar   b->diag       = bdiag;
3939b48462bSShri Abhyankar   b->ilen       = 0;
3949b48462bSShri Abhyankar   b->imax       = 0;
3959b48462bSShri Abhyankar   b->row        = isrow;
3969b48462bSShri Abhyankar   b->col        = iscol;
3979b48462bSShri Abhyankar   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
3989b48462bSShri Abhyankar   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
3999b48462bSShri Abhyankar   b->icol       = isicol;
4009b48462bSShri Abhyankar   ierr          = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
4019b48462bSShri Abhyankar 
4029b48462bSShri Abhyankar   /* In b structure:  Free imax, ilen, old a, old j.  Allocate solve_work, new a, new j */
4039b48462bSShri Abhyankar   ierr = PetscLogObjectMemory(B,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
4049b48462bSShri Abhyankar   b->maxnz = b->nz = bdiag[0]+1;
405d5f3da31SBarry Smith   B->factortype            = MAT_FACTOR_LU;
406f284f12aSHong Zhang   B->info.factor_mallocs   = reallocs;
407f284f12aSHong Zhang   B->info.fill_ratio_given = f;
4089b48462bSShri Abhyankar 
4099b48462bSShri Abhyankar   if (ai[n]) {
410f284f12aSHong Zhang     B->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]);
4119b48462bSShri Abhyankar   } else {
412f284f12aSHong Zhang     B->info.fill_ratio_needed = 0.0;
4139b48462bSShri Abhyankar   }
4149263d837SHong Zhang #if defined(PETSC_USE_INFO)
4159263d837SHong Zhang   if (ai[n] != 0) {
4169263d837SHong Zhang     PetscReal af = B->info.fill_ratio_needed;
4179263d837SHong Zhang     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr);
4189263d837SHong Zhang     ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
4199263d837SHong Zhang     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr);
4209263d837SHong Zhang     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
4219263d837SHong Zhang   } else {
4229263d837SHong Zhang     ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr);
4239263d837SHong Zhang   }
4249263d837SHong Zhang #endif
42535233d99SShri Abhyankar   B->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ;
426f2344125SBarry Smith   if (a->inode.size) {
427d3ac4fa3SBarry Smith     B->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode;
428d3ac4fa3SBarry Smith   }
4299b48462bSShri Abhyankar   PetscFunctionReturn(0);
4309b48462bSShri Abhyankar }
4319b48462bSShri Abhyankar 
4325b5bc046SBarry Smith /*
4335b5bc046SBarry Smith     Trouble in factorization, should we dump the original matrix?
4345b5bc046SBarry Smith */
4355b5bc046SBarry Smith #undef __FUNCT__
4365b5bc046SBarry Smith #define __FUNCT__ "MatFactorDumpMatrix"
4375b5bc046SBarry Smith PetscErrorCode MatFactorDumpMatrix(Mat A)
4385b5bc046SBarry Smith {
4395b5bc046SBarry Smith   PetscErrorCode ierr;
440ace3abfcSBarry Smith   PetscBool      flg = PETSC_FALSE;
4415b5bc046SBarry Smith 
4425b5bc046SBarry Smith   PetscFunctionBegin;
443acfcf0e5SJed Brown   ierr = PetscOptionsGetBool(PETSC_NULL,"-mat_factor_dump_on_error",&flg,PETSC_NULL);CHKERRQ(ierr);
4445b5bc046SBarry Smith   if (flg) {
4455b5bc046SBarry Smith     PetscViewer viewer;
4465b5bc046SBarry Smith     char        filename[PETSC_MAX_PATH_LEN];
4475b5bc046SBarry Smith 
4485b5bc046SBarry Smith     ierr = PetscSNPrintf(filename,PETSC_MAX_PATH_LEN,"matrix_factor_error.%d",PetscGlobalRank);CHKERRQ(ierr);
4497adad957SLisandro Dalcin     ierr = PetscViewerBinaryOpen(((PetscObject)A)->comm,filename,FILE_MODE_WRITE,&viewer);CHKERRQ(ierr);
4505b5bc046SBarry Smith     ierr = MatView(A,viewer);CHKERRQ(ierr);
4516bf464f9SBarry Smith     ierr = PetscViewerDestroy(&viewer);CHKERRQ(ierr);
4525b5bc046SBarry Smith   }
4535b5bc046SBarry Smith   PetscFunctionReturn(0);
4545b5bc046SBarry Smith }
4555b5bc046SBarry Smith 
456c9c72f8fSHong Zhang #undef __FUNCT__
45735233d99SShri Abhyankar #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ"
45835233d99SShri Abhyankar PetscErrorCode MatLUFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info)
459c9c72f8fSHong Zhang {
460c9c72f8fSHong Zhang   Mat              C=B;
461c9c72f8fSHong Zhang   Mat_SeqAIJ       *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data;
462c9c72f8fSHong Zhang   IS               isrow = b->row,isicol = b->icol;
463c9c72f8fSHong Zhang   PetscErrorCode   ierr;
464c9c72f8fSHong Zhang   const PetscInt   *r,*ic,*ics;
465bbd65245SShri Abhyankar   const PetscInt   n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j,*bdiag=b->diag;
466bbd65245SShri Abhyankar   PetscInt         i,j,k,nz,nzL,row,*pj;
467bbd65245SShri Abhyankar   const PetscInt   *ajtmp,*bjtmp;
468bbd65245SShri Abhyankar   MatScalar        *rtmp,*pc,multiplier,*pv;
469bbd65245SShri Abhyankar   const  MatScalar *aa=a->a,*v;
470ace3abfcSBarry Smith   PetscBool        row_identity,col_identity;
471d90ac83dSHong Zhang   FactorShiftCtx   sctx;
4724f81c4b7SBarry Smith   const PetscInt   *ddiag;
473d4ad06f7SHong Zhang   PetscReal        rs;
474d4ad06f7SHong Zhang   MatScalar        d;
475d4ad06f7SHong Zhang 
476c9c72f8fSHong Zhang   PetscFunctionBegin;
477d6e5152cSHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
478d90ac83dSHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
479d4ad06f7SHong Zhang 
480f4db908eSBarry Smith   if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
481d4ad06f7SHong Zhang     ddiag          = a->diag;
482d4ad06f7SHong Zhang     sctx.shift_top = info->zeropivot;
483d4ad06f7SHong Zhang     for (i=0; i<n; i++) {
484d4ad06f7SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
485d4ad06f7SHong Zhang       d  = (aa)[ddiag[i]];
486d4ad06f7SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
487d4ad06f7SHong Zhang       v  = aa+ai[i];
488d4ad06f7SHong Zhang       nz = ai[i+1] - ai[i];
489d4ad06f7SHong Zhang       for (j=0; j<nz; j++)
490d4ad06f7SHong Zhang 	rs += PetscAbsScalar(v[j]);
491d4ad06f7SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
492d4ad06f7SHong Zhang     }
493d4ad06f7SHong Zhang     sctx.shift_top   *= 1.1;
494d4ad06f7SHong Zhang     sctx.nshift_max   = 5;
495d4ad06f7SHong Zhang     sctx.shift_lo     = 0.;
496d4ad06f7SHong Zhang     sctx.shift_hi     = 1.;
497d4ad06f7SHong Zhang   }
498d4ad06f7SHong Zhang 
499c9c72f8fSHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
500c9c72f8fSHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
501c9c72f8fSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr);
502c9c72f8fSHong Zhang   ics  = ic;
503c9c72f8fSHong Zhang 
504d4ad06f7SHong Zhang   do {
50507b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
506c9c72f8fSHong Zhang     for (i=0; i<n; i++){
507c9c72f8fSHong Zhang       /* zero rtmp */
508c9c72f8fSHong Zhang       /* L part */
509c9c72f8fSHong Zhang       nz    = bi[i+1] - bi[i];
510c9c72f8fSHong Zhang       bjtmp = bj + bi[i];
511c9c72f8fSHong Zhang       for  (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
512c9c72f8fSHong Zhang 
513c9c72f8fSHong Zhang       /* U part */
51462fbd6afSShri Abhyankar       nz = bdiag[i]-bdiag[i+1];
51562fbd6afSShri Abhyankar       bjtmp = bj + bdiag[i+1]+1;
516813a1f2bSShri Abhyankar       for  (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
517813a1f2bSShri Abhyankar 
518813a1f2bSShri Abhyankar       /* load in initial (unfactored row) */
519813a1f2bSShri Abhyankar       nz    = ai[r[i]+1] - ai[r[i]];
520813a1f2bSShri Abhyankar       ajtmp = aj + ai[r[i]];
521813a1f2bSShri Abhyankar       v     = aa + ai[r[i]];
522813a1f2bSShri Abhyankar       for (j=0; j<nz; j++) {
523813a1f2bSShri Abhyankar         rtmp[ics[ajtmp[j]]] = v[j];
524813a1f2bSShri Abhyankar       }
525813a1f2bSShri Abhyankar       /* ZeropivotApply() */
52698a93161SHong Zhang       rtmp[i] += sctx.shift_amount;  /* shift the diagonal of the matrix */
527813a1f2bSShri Abhyankar 
528813a1f2bSShri Abhyankar       /* elimination */
529813a1f2bSShri Abhyankar       bjtmp = bj + bi[i];
530813a1f2bSShri Abhyankar       row   = *bjtmp++;
531813a1f2bSShri Abhyankar       nzL   = bi[i+1] - bi[i];
532f268cba8SShri Abhyankar       for(k=0; k < nzL;k++) {
533813a1f2bSShri Abhyankar         pc = rtmp + row;
534813a1f2bSShri Abhyankar         if (*pc != 0.0) {
535813a1f2bSShri Abhyankar           pv         = b->a + bdiag[row];
536813a1f2bSShri Abhyankar           multiplier = *pc * (*pv);
537813a1f2bSShri Abhyankar           *pc        = multiplier;
53862fbd6afSShri Abhyankar           pj = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */
53962fbd6afSShri Abhyankar 	  pv = b->a + bdiag[row+1]+1;
54062fbd6afSShri Abhyankar 	  nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */
541813a1f2bSShri Abhyankar           for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j];
542813a1f2bSShri Abhyankar           ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr);
543813a1f2bSShri Abhyankar         }
544f268cba8SShri Abhyankar         row = *bjtmp++;
545813a1f2bSShri Abhyankar       }
546813a1f2bSShri Abhyankar 
547813a1f2bSShri Abhyankar       /* finished row so stick it into b->a */
548813a1f2bSShri Abhyankar       rs = 0.0;
549813a1f2bSShri Abhyankar       /* L part */
550813a1f2bSShri Abhyankar       pv   = b->a + bi[i] ;
551813a1f2bSShri Abhyankar       pj   = b->j + bi[i] ;
552813a1f2bSShri Abhyankar       nz   = bi[i+1] - bi[i];
553813a1f2bSShri Abhyankar       for (j=0; j<nz; j++) {
554813a1f2bSShri Abhyankar         pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]);
555813a1f2bSShri Abhyankar       }
556813a1f2bSShri Abhyankar 
557813a1f2bSShri Abhyankar       /* U part */
55862fbd6afSShri Abhyankar       pv = b->a + bdiag[i+1]+1;
55962fbd6afSShri Abhyankar       pj = b->j + bdiag[i+1]+1;
56062fbd6afSShri Abhyankar       nz = bdiag[i] - bdiag[i+1]-1;
561813a1f2bSShri Abhyankar       for (j=0; j<nz; j++) {
562813a1f2bSShri Abhyankar         pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]);
563813a1f2bSShri Abhyankar       }
564813a1f2bSShri Abhyankar 
565813a1f2bSShri Abhyankar       sctx.rs  = rs;
566813a1f2bSShri Abhyankar       sctx.pv  = rtmp[i];
567d0660788SBarry Smith       ierr = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr);
56807b50cabSHong Zhang       if(sctx.newshift) break; /* break for-loop */
56907b50cabSHong Zhang       rtmp[i] = sctx.pv; /* sctx.pv might be updated in the case of MAT_SHIFT_INBLOCKS */
570813a1f2bSShri Abhyankar 
571813a1f2bSShri Abhyankar       /* Mark diagonal and invert diagonal for simplier triangular solves */
572813a1f2bSShri Abhyankar       pv  = b->a + bdiag[i];
573813a1f2bSShri Abhyankar       *pv = 1.0/rtmp[i];
574813a1f2bSShri Abhyankar 
575813a1f2bSShri Abhyankar     } /* endof for (i=0; i<n; i++){ */
576813a1f2bSShri Abhyankar 
5778ff23777SHong Zhang     /* MatPivotRefine() */
57807b50cabSHong Zhang     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max){
579813a1f2bSShri Abhyankar       /*
580813a1f2bSShri Abhyankar        * if no shift in this attempt & shifting & started shifting & can refine,
581813a1f2bSShri Abhyankar        * then try lower shift
582813a1f2bSShri Abhyankar        */
583813a1f2bSShri Abhyankar       sctx.shift_hi       = sctx.shift_fraction;
584813a1f2bSShri Abhyankar       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
585813a1f2bSShri Abhyankar       sctx.shift_amount   = sctx.shift_fraction * sctx.shift_top;
58607b50cabSHong Zhang       sctx.newshift       = PETSC_TRUE;
587813a1f2bSShri Abhyankar       sctx.nshift++;
588813a1f2bSShri Abhyankar     }
58907b50cabSHong Zhang   } while (sctx.newshift);
590813a1f2bSShri Abhyankar 
591813a1f2bSShri Abhyankar   ierr = PetscFree(rtmp);CHKERRQ(ierr);
592813a1f2bSShri Abhyankar   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
593813a1f2bSShri Abhyankar   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
594d3ac4fa3SBarry Smith 
595813a1f2bSShri Abhyankar   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
596813a1f2bSShri Abhyankar   ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr);
597813a1f2bSShri Abhyankar   if (row_identity && col_identity) {
59835233d99SShri Abhyankar     C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering;
599813a1f2bSShri Abhyankar   } else {
60035233d99SShri Abhyankar     C->ops->solve = MatSolve_SeqAIJ;
601813a1f2bSShri Abhyankar   }
60235233d99SShri Abhyankar   C->ops->solveadd           = MatSolveAdd_SeqAIJ;
60335233d99SShri Abhyankar   C->ops->solvetranspose     = MatSolveTranspose_SeqAIJ;
60435233d99SShri Abhyankar   C->ops->solvetransposeadd  = MatSolveTransposeAdd_SeqAIJ;
60535233d99SShri Abhyankar   C->ops->matsolve           = MatMatSolve_SeqAIJ;
606813a1f2bSShri Abhyankar   C->assembled    = PETSC_TRUE;
607813a1f2bSShri Abhyankar   C->preallocated = PETSC_TRUE;
608813a1f2bSShri Abhyankar   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
609813a1f2bSShri Abhyankar 
61014d2772eSHong Zhang   /* MatShiftView(A,info,&sctx) */
611813a1f2bSShri Abhyankar   if (sctx.nshift){
612f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
613813a1f2bSShri 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);
614f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
615813a1f2bSShri Abhyankar       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
616f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_INBLOCKS){
617915743fcSHong Zhang       ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %G\n",sctx.nshift,info->shiftamount);CHKERRQ(ierr);
618813a1f2bSShri Abhyankar     }
619813a1f2bSShri Abhyankar   }
620d3ac4fa3SBarry Smith   ierr = Mat_CheckInode_FactorLU(C,PETSC_FALSE);CHKERRQ(ierr);
621813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
622813a1f2bSShri Abhyankar }
623813a1f2bSShri Abhyankar 
6244a2ae208SSatish Balay #undef __FUNCT__
625ad04f41aSHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_inplace"
626ad04f41aSHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ_inplace(Mat B,Mat A,const MatFactorInfo *info)
627289bc588SBarry Smith {
628719d5645SBarry Smith   Mat             C=B;
629416022c9SBarry Smith   Mat_SeqAIJ      *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data;
63082bf6240SBarry Smith   IS              isrow = b->row,isicol = b->icol;
6316849ba73SBarry Smith   PetscErrorCode  ierr;
6325d0c19d7SBarry Smith   const PetscInt   *r,*ic,*ics;
633d278edc7SBarry Smith   PetscInt        nz,row,i,j,n=A->rmap->n,diag;
634d278edc7SBarry Smith   const PetscInt  *ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j;
635d278edc7SBarry Smith   const PetscInt  *ajtmp,*bjtmp,*diag_offset = b->diag,*pj;
636d278edc7SBarry Smith   MatScalar       *pv,*rtmp,*pc,multiplier,d;
637d278edc7SBarry Smith   const MatScalar *v,*aa=a->a;
63833f9893dSHong Zhang   PetscReal       rs=0.0;
639d90ac83dSHong Zhang   FactorShiftCtx  sctx;
6404f81c4b7SBarry Smith   const PetscInt  *ddiag;
641ace3abfcSBarry Smith   PetscBool       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 {
67207b50cabSHong Zhang     sctx.newshift = 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];
715d0660788SBarry Smith       ierr = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr);
71607b50cabSHong Zhang       if (sctx.newshift) break;
717504a3fadSHong Zhang       pv[diag] = sctx.pv;
71835aab85fSBarry Smith     }
719d2276718SHong Zhang 
72007b50cabSHong Zhang     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && 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;
72807b50cabSHong Zhang       sctx.newshift       = PETSC_TRUE;
729d2276718SHong Zhang       sctx.nshift++;
7306cc28720Svictorle     }
73107b50cabSHong Zhang   } while (sctx.newshift);
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;
793504a3fadSHong Zhang   const  MatScalar *aa=a->a,*vtmp;
794137fb511SHong Zhang 
795137fb511SHong Zhang   PetscFunctionBegin;
796e32f2f54SBarry Smith   if (A != B) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"input and output matrix must have same address");
797504a3fadSHong Zhang 
798504a3fadSHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
799504a3fadSHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
800504a3fadSHong Zhang 
801504a3fadSHong Zhang   if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
802504a3fadSHong Zhang     const PetscInt   *ddiag = a->diag;
803504a3fadSHong Zhang     sctx.shift_top = info->zeropivot;
804504a3fadSHong Zhang     for (i=0; i<n; i++) {
805504a3fadSHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
806504a3fadSHong Zhang       d  = (aa)[ddiag[i]];
807504a3fadSHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
808504a3fadSHong Zhang       vtmp  = aa+ai[i];
809504a3fadSHong Zhang       nz = ai[i+1] - ai[i];
810504a3fadSHong Zhang       for (j=0; j<nz; j++)
811504a3fadSHong Zhang 	rs += PetscAbsScalar(vtmp[j]);
812504a3fadSHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
813504a3fadSHong Zhang     }
814504a3fadSHong Zhang     sctx.shift_top   *= 1.1;
815504a3fadSHong Zhang     sctx.nshift_max   = 5;
816504a3fadSHong Zhang     sctx.shift_lo     = 0.;
817504a3fadSHong Zhang     sctx.shift_hi     = 1.;
818504a3fadSHong Zhang   }
819504a3fadSHong Zhang 
820137fb511SHong Zhang   ierr  = ISGetIndices(isrow,&r);CHKERRQ(ierr);
821137fb511SHong Zhang   ierr  = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
822137fb511SHong Zhang   ierr  = PetscMalloc((n+1)*sizeof(PetscScalar),&rtmp);CHKERRQ(ierr);
823137fb511SHong Zhang   ierr  = PetscMemzero(rtmp,(n+1)*sizeof(PetscScalar));CHKERRQ(ierr);
824b051339dSHong Zhang   ics = ic;
825137fb511SHong Zhang 
826504a3fadSHong Zhang #if defined(MV)
82775567043SBarry Smith   sctx.shift_top      = 0.;
828e60cf9a0SBarry Smith   sctx.nshift_max     = 0;
82975567043SBarry Smith   sctx.shift_lo       = 0.;
83075567043SBarry Smith   sctx.shift_hi       = 0.;
83175567043SBarry Smith   sctx.shift_fraction = 0.;
832137fb511SHong Zhang 
833f4db908eSBarry Smith   if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
83475567043SBarry Smith     sctx.shift_top = 0.;
835137fb511SHong Zhang     for (i=0; i<n; i++) {
836137fb511SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
837b051339dSHong Zhang       d  = (a->a)[diag[i]];
838137fb511SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
839b051339dSHong Zhang       v  = a->a+ai[i];
840b051339dSHong Zhang       nz = ai[i+1] - ai[i];
841137fb511SHong Zhang       for (j=0; j<nz; j++)
842137fb511SHong Zhang 	rs += PetscAbsScalar(v[j]);
843137fb511SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
844137fb511SHong Zhang     }
845137fb511SHong Zhang     if (sctx.shift_top < info->zeropivot) sctx.shift_top = info->zeropivot;
846137fb511SHong Zhang     sctx.shift_top    *= 1.1;
847137fb511SHong Zhang     sctx.nshift_max   = 5;
848137fb511SHong Zhang     sctx.shift_lo     = 0.;
849137fb511SHong Zhang     sctx.shift_hi     = 1.;
850137fb511SHong Zhang   }
851137fb511SHong Zhang 
85275567043SBarry Smith   sctx.shift_amount = 0.;
853137fb511SHong Zhang   sctx.nshift       = 0;
854504a3fadSHong Zhang #endif
855504a3fadSHong Zhang 
856137fb511SHong Zhang   do {
85707b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
858137fb511SHong Zhang     for (i=0; i<n; i++){
859b051339dSHong Zhang       /* load in initial unfactored row */
860b051339dSHong Zhang       nz    = ai[r[i]+1] - ai[r[i]];
861b051339dSHong Zhang       ajtmp = aj + ai[r[i]];
862b051339dSHong Zhang       v     = a->a + ai[r[i]];
863137fb511SHong Zhang       /* sort permuted ajtmp and values v accordingly */
864b051339dSHong Zhang       for (j=0; j<nz; j++) ajtmp[j] = ics[ajtmp[j]];
865137fb511SHong Zhang       ierr = PetscSortIntWithScalarArray(nz,ajtmp,v);CHKERRQ(ierr);
866137fb511SHong Zhang 
867b051339dSHong Zhang       diag[r[i]] = ai[r[i]];
868137fb511SHong Zhang       for (j=0; j<nz; j++) {
869137fb511SHong Zhang         rtmp[ajtmp[j]] = v[j];
870b051339dSHong Zhang         if (ajtmp[j] < i) diag[r[i]]++; /* update a->diag */
871137fb511SHong Zhang       }
872137fb511SHong Zhang       rtmp[r[i]] += sctx.shift_amount; /* shift the diagonal of the matrix */
873137fb511SHong Zhang 
874b051339dSHong Zhang       row = *ajtmp++;
875137fb511SHong Zhang       while  (row < i) {
876137fb511SHong Zhang         pc = rtmp + row;
877137fb511SHong Zhang         if (*pc != 0.0) {
878b051339dSHong Zhang           pv         = a->a + diag[r[row]];
879b051339dSHong Zhang           pj         = aj + diag[r[row]] + 1;
880137fb511SHong Zhang 
881137fb511SHong Zhang           multiplier = *pc / *pv++;
882137fb511SHong Zhang           *pc        = multiplier;
883b051339dSHong Zhang           nz         = ai[r[row]+1] - diag[r[row]] - 1;
884b051339dSHong Zhang           for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j];
885dc0b31edSSatish Balay           ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr);
886137fb511SHong Zhang         }
887b051339dSHong Zhang         row = *ajtmp++;
888137fb511SHong Zhang       }
889b051339dSHong Zhang       /* finished row so overwrite it onto a->a */
890b051339dSHong Zhang       pv   = a->a + ai[r[i]] ;
891b051339dSHong Zhang       pj   = aj + ai[r[i]] ;
892b051339dSHong Zhang       nz   = ai[r[i]+1] - ai[r[i]];
893b051339dSHong Zhang       nbdiag = diag[r[i]] - ai[r[i]]; /* num of entries before the diagonal */
894137fb511SHong Zhang 
895137fb511SHong Zhang       rs   = 0.0;
896137fb511SHong Zhang       for (j=0; j<nz; j++) {
897b051339dSHong Zhang         pv[j] = rtmp[pj[j]];
898b051339dSHong Zhang         if (j != nbdiag) rs += PetscAbsScalar(pv[j]);
899137fb511SHong Zhang       }
900137fb511SHong Zhang 
901137fb511SHong Zhang       sctx.rs  = rs;
902b051339dSHong Zhang       sctx.pv  = pv[nbdiag];
903d0660788SBarry Smith       ierr = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr);
90407b50cabSHong Zhang       if (sctx.newshift) break;
905504a3fadSHong Zhang       pv[nbdiag] = sctx.pv;
906137fb511SHong Zhang     }
907137fb511SHong Zhang 
90807b50cabSHong Zhang     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) {
909137fb511SHong Zhang       /*
910137fb511SHong Zhang        * if no shift in this attempt & shifting & started shifting & can refine,
911137fb511SHong Zhang        * then try lower shift
912137fb511SHong Zhang        */
9130481f469SBarry Smith       sctx.shift_hi        = sctx.shift_fraction;
9140481f469SBarry Smith       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
9150481f469SBarry Smith       sctx.shift_amount    = sctx.shift_fraction * sctx.shift_top;
91607b50cabSHong Zhang       sctx.newshift         = PETSC_TRUE;
917137fb511SHong Zhang       sctx.nshift++;
918137fb511SHong Zhang     }
91907b50cabSHong Zhang   } while (sctx.newshift);
920137fb511SHong Zhang 
921137fb511SHong Zhang   /* invert diagonal entries for simplier triangular solves */
922137fb511SHong Zhang   for (i=0; i<n; i++) {
923b051339dSHong Zhang     a->a[diag[r[i]]] = 1.0/a->a[diag[r[i]]];
924137fb511SHong Zhang   }
925137fb511SHong Zhang 
926137fb511SHong Zhang   ierr = PetscFree(rtmp);CHKERRQ(ierr);
927137fb511SHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
928137fb511SHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
929b051339dSHong Zhang   A->ops->solve             = MatSolve_SeqAIJ_InplaceWithPerm;
930ad04f41aSHong Zhang   A->ops->solveadd          = MatSolveAdd_SeqAIJ_inplace;
931ad04f41aSHong Zhang   A->ops->solvetranspose    = MatSolveTranspose_SeqAIJ_inplace;
932ad04f41aSHong Zhang   A->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ_inplace;
933b051339dSHong Zhang   A->assembled = PETSC_TRUE;
9345c9eb25fSBarry Smith   A->preallocated = PETSC_TRUE;
935d0f46423SBarry Smith   ierr = PetscLogFlops(A->cmap->n);CHKERRQ(ierr);
936137fb511SHong Zhang   if (sctx.nshift){
937f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
9380481f469SBarry 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);
939f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
940e738e422SBarry Smith       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
941137fb511SHong Zhang     }
942137fb511SHong Zhang   }
943137fb511SHong Zhang   PetscFunctionReturn(0);
944137fb511SHong Zhang }
945137fb511SHong Zhang 
9460a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
9474a2ae208SSatish Balay #undef __FUNCT__
9484a2ae208SSatish Balay #define __FUNCT__ "MatLUFactor_SeqAIJ"
9490481f469SBarry Smith PetscErrorCode MatLUFactor_SeqAIJ(Mat A,IS row,IS col,const MatFactorInfo *info)
950da3a660dSBarry Smith {
951dfbe8321SBarry Smith   PetscErrorCode ierr;
952416022c9SBarry Smith   Mat            C;
953416022c9SBarry Smith 
9543a40ed3dSBarry Smith   PetscFunctionBegin;
9552692d6eeSBarry Smith   ierr = MatGetFactor(A,MATSOLVERPETSC,MAT_FACTOR_LU,&C);CHKERRQ(ierr);
956719d5645SBarry Smith   ierr = MatLUFactorSymbolic(C,A,row,col,info);CHKERRQ(ierr);
957719d5645SBarry Smith   ierr = MatLUFactorNumeric(C,A,info);CHKERRQ(ierr);
958db4efbfdSBarry Smith   A->ops->solve            = C->ops->solve;
959db4efbfdSBarry Smith   A->ops->solvetranspose   = C->ops->solvetranspose;
960eb6b5d47SBarry Smith   ierr = MatHeaderMerge(A,C);CHKERRQ(ierr);
96152e6d16bSBarry Smith   ierr = PetscLogObjectParent(A,((Mat_SeqAIJ*)(A->data))->icol);CHKERRQ(ierr);
9623a40ed3dSBarry Smith   PetscFunctionReturn(0);
963da3a660dSBarry Smith }
9640a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
9651d098868SHong Zhang 
966889a8dedSBarry Smith 
9674a2ae208SSatish Balay #undef __FUNCT__
968ad04f41aSHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_inplace"
969ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_inplace(Mat A,Vec bb,Vec xx)
9708c37ef55SBarry Smith {
971416022c9SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
972416022c9SBarry Smith   IS                iscol = a->col,isrow = a->row;
9736849ba73SBarry Smith   PetscErrorCode    ierr;
9745d0c19d7SBarry Smith   PetscInt          i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
9755d0c19d7SBarry Smith   PetscInt          nz;
9765d0c19d7SBarry Smith   const PetscInt    *rout,*cout,*r,*c;
977dd6ea824SBarry Smith   PetscScalar       *x,*tmp,*tmps,sum;
978d9fead3dSBarry Smith   const PetscScalar *b;
979dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
9808c37ef55SBarry Smith 
9813a40ed3dSBarry Smith   PetscFunctionBegin;
9823a40ed3dSBarry Smith   if (!n) PetscFunctionReturn(0);
98391bf9adeSBarry Smith 
9843649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
9851ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
986416022c9SBarry Smith   tmp  = a->solve_work;
9878c37ef55SBarry Smith 
9883b2fbd54SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
9893b2fbd54SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
9908c37ef55SBarry Smith 
9918c37ef55SBarry Smith   /* forward solve the lower triangular */
9928c37ef55SBarry Smith   tmp[0] = b[*r++];
993010a20caSHong Zhang   tmps   = tmp;
9948c37ef55SBarry Smith   for (i=1; i<n; i++) {
995010a20caSHong Zhang     v   = aa + ai[i] ;
996010a20caSHong Zhang     vi  = aj + ai[i] ;
99753e7425aSBarry Smith     nz  = a->diag[i] - ai[i];
99853e7425aSBarry Smith     sum = b[*r++];
999003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
10008c37ef55SBarry Smith     tmp[i] = sum;
10018c37ef55SBarry Smith   }
10028c37ef55SBarry Smith 
10038c37ef55SBarry Smith   /* backward solve the upper triangular */
10048c37ef55SBarry Smith   for (i=n-1; i>=0; i--){
1005010a20caSHong Zhang     v   = aa + a->diag[i] + 1;
1006010a20caSHong Zhang     vi  = aj + a->diag[i] + 1;
1007416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
10088c37ef55SBarry Smith     sum = tmp[i];
1009003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
1010010a20caSHong Zhang     x[*c--] = tmp[i] = sum*aa[a->diag[i]];
10118c37ef55SBarry Smith   }
10128c37ef55SBarry Smith 
10133b2fbd54SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
10143b2fbd54SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
10153649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
10161ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1017dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
10183a40ed3dSBarry Smith   PetscFunctionReturn(0);
10198c37ef55SBarry Smith }
1020026e39d0SSatish Balay 
10212bebee5dSHong Zhang #undef __FUNCT__
1022ad04f41aSHong Zhang #define __FUNCT__ "MatMatSolve_SeqAIJ_inplace"
1023ad04f41aSHong Zhang PetscErrorCode MatMatSolve_SeqAIJ_inplace(Mat A,Mat B,Mat X)
10242bebee5dSHong Zhang {
10252bebee5dSHong Zhang   Mat_SeqAIJ      *a = (Mat_SeqAIJ*)A->data;
10262bebee5dSHong Zhang   IS              iscol = a->col,isrow = a->row;
10272bebee5dSHong Zhang   PetscErrorCode  ierr;
10285d0c19d7SBarry Smith   PetscInt        i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
10295d0c19d7SBarry Smith   PetscInt        nz,neq;
10305d0c19d7SBarry Smith   const PetscInt  *rout,*cout,*r,*c;
1031dd6ea824SBarry Smith   PetscScalar     *x,*b,*tmp,*tmps,sum;
1032dd6ea824SBarry Smith   const MatScalar *aa = a->a,*v;
1033ace3abfcSBarry Smith   PetscBool       bisdense,xisdense;
10342bebee5dSHong Zhang 
10352bebee5dSHong Zhang   PetscFunctionBegin;
10362bebee5dSHong Zhang   if (!n) PetscFunctionReturn(0);
10372bebee5dSHong Zhang 
1038251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr);
1039e32f2f54SBarry Smith   if (!bisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix");
1040251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr);
1041e32f2f54SBarry Smith   if (!xisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix");
1042db4efbfdSBarry Smith 
10432bebee5dSHong Zhang   ierr = MatGetArray(B,&b);CHKERRQ(ierr);
10442bebee5dSHong Zhang   ierr = MatGetArray(X,&x);CHKERRQ(ierr);
10452bebee5dSHong Zhang 
10462bebee5dSHong Zhang   tmp  = a->solve_work;
10472bebee5dSHong Zhang   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
10482bebee5dSHong Zhang   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
10492bebee5dSHong Zhang 
1050a36b22ccSSatish Balay   for (neq=0; neq<B->cmap->n; neq++){
10512bebee5dSHong Zhang     /* forward solve the lower triangular */
10522bebee5dSHong Zhang     tmp[0] = b[r[0]];
10532bebee5dSHong Zhang     tmps   = tmp;
10542bebee5dSHong Zhang     for (i=1; i<n; i++) {
10552bebee5dSHong Zhang       v   = aa + ai[i] ;
10562bebee5dSHong Zhang       vi  = aj + ai[i] ;
10572bebee5dSHong Zhang       nz  = a->diag[i] - ai[i];
10582bebee5dSHong Zhang       sum = b[r[i]];
1059003131ecSBarry Smith       PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
10602bebee5dSHong Zhang       tmp[i] = sum;
10612bebee5dSHong Zhang     }
10622bebee5dSHong Zhang     /* backward solve the upper triangular */
10632bebee5dSHong Zhang     for (i=n-1; i>=0; i--){
10642bebee5dSHong Zhang       v   = aa + a->diag[i] + 1;
10652bebee5dSHong Zhang       vi  = aj + a->diag[i] + 1;
10662bebee5dSHong Zhang       nz  = ai[i+1] - a->diag[i] - 1;
10672bebee5dSHong Zhang       sum = tmp[i];
1068003131ecSBarry Smith       PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
10692bebee5dSHong Zhang       x[c[i]] = tmp[i] = sum*aa[a->diag[i]];
10702bebee5dSHong Zhang     }
10712bebee5dSHong Zhang 
10722bebee5dSHong Zhang     b += n;
10732bebee5dSHong Zhang     x += n;
10742bebee5dSHong Zhang   }
10752bebee5dSHong Zhang   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
10762bebee5dSHong Zhang   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
10772bebee5dSHong Zhang   ierr = MatRestoreArray(B,&b);CHKERRQ(ierr);
10782bebee5dSHong Zhang   ierr = MatRestoreArray(X,&x);CHKERRQ(ierr);
1079dc0b31edSSatish Balay   ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr);
10802bebee5dSHong Zhang   PetscFunctionReturn(0);
10812bebee5dSHong Zhang }
10822bebee5dSHong Zhang 
1083137fb511SHong Zhang #undef __FUNCT__
108435233d99SShri Abhyankar #define __FUNCT__ "MatMatSolve_SeqAIJ"
108535233d99SShri Abhyankar PetscErrorCode MatMatSolve_SeqAIJ(Mat A,Mat B,Mat X)
10869bd0847aSShri Abhyankar {
10879bd0847aSShri Abhyankar   Mat_SeqAIJ      *a = (Mat_SeqAIJ*)A->data;
10889bd0847aSShri Abhyankar   IS              iscol = a->col,isrow = a->row;
10899bd0847aSShri Abhyankar   PetscErrorCode  ierr;
10909bd0847aSShri Abhyankar   PetscInt        i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j,*adiag = a->diag;
10919bd0847aSShri Abhyankar   PetscInt        nz,neq;
10929bd0847aSShri Abhyankar   const PetscInt  *rout,*cout,*r,*c;
10939bd0847aSShri Abhyankar   PetscScalar     *x,*b,*tmp,sum;
10949bd0847aSShri Abhyankar   const MatScalar *aa = a->a,*v;
1095ace3abfcSBarry Smith   PetscBool       bisdense,xisdense;
10969bd0847aSShri Abhyankar 
10979bd0847aSShri Abhyankar   PetscFunctionBegin;
10989bd0847aSShri Abhyankar   if (!n) PetscFunctionReturn(0);
10999bd0847aSShri Abhyankar 
1100251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr);
1101e32f2f54SBarry Smith   if (!bisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix");
1102251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr);
1103e32f2f54SBarry Smith   if (!xisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix");
11049bd0847aSShri Abhyankar 
11059bd0847aSShri Abhyankar   ierr = MatGetArray(B,&b);CHKERRQ(ierr);
11069bd0847aSShri Abhyankar   ierr = MatGetArray(X,&x);CHKERRQ(ierr);
11079bd0847aSShri Abhyankar 
11089bd0847aSShri Abhyankar   tmp  = a->solve_work;
11099bd0847aSShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
11109bd0847aSShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
11119bd0847aSShri Abhyankar 
11129bd0847aSShri Abhyankar   for (neq=0; neq<B->cmap->n; neq++){
11139bd0847aSShri Abhyankar     /* forward solve the lower triangular */
11149bd0847aSShri Abhyankar     tmp[0] = b[r[0]];
11159bd0847aSShri Abhyankar     v      = aa;
11169bd0847aSShri Abhyankar     vi     = aj;
11179bd0847aSShri Abhyankar     for (i=1; i<n; i++) {
11189bd0847aSShri Abhyankar       nz  = ai[i+1] - ai[i];
11199bd0847aSShri Abhyankar       sum = b[r[i]];
11209bd0847aSShri Abhyankar       PetscSparseDenseMinusDot(sum,tmp,v,vi,nz);
11219bd0847aSShri Abhyankar       tmp[i] = sum;
11229bd0847aSShri Abhyankar       v += nz; vi += nz;
11239bd0847aSShri Abhyankar     }
11249bd0847aSShri Abhyankar 
11259bd0847aSShri Abhyankar     /* backward solve the upper triangular */
11269bd0847aSShri Abhyankar     for (i=n-1; i>=0; i--){
11279bd0847aSShri Abhyankar       v   = aa + adiag[i+1]+1;
11289bd0847aSShri Abhyankar       vi  = aj + adiag[i+1]+1;
11299bd0847aSShri Abhyankar       nz  = adiag[i]-adiag[i+1]-1;
11309bd0847aSShri Abhyankar       sum = tmp[i];
11319bd0847aSShri Abhyankar       PetscSparseDenseMinusDot(sum,tmp,v,vi,nz);
11329bd0847aSShri Abhyankar       x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */
11339bd0847aSShri Abhyankar     }
11349bd0847aSShri Abhyankar 
11359bd0847aSShri Abhyankar     b += n;
11369bd0847aSShri Abhyankar     x += n;
11379bd0847aSShri Abhyankar   }
11389bd0847aSShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
11399bd0847aSShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
11409bd0847aSShri Abhyankar   ierr = MatRestoreArray(B,&b);CHKERRQ(ierr);
11419bd0847aSShri Abhyankar   ierr = MatRestoreArray(X,&x);CHKERRQ(ierr);
11429bd0847aSShri Abhyankar   ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr);
11439bd0847aSShri Abhyankar   PetscFunctionReturn(0);
11449bd0847aSShri Abhyankar }
11459bd0847aSShri Abhyankar 
11469bd0847aSShri Abhyankar #undef __FUNCT__
1147137fb511SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_InplaceWithPerm"
1148137fb511SHong Zhang PetscErrorCode MatSolve_SeqAIJ_InplaceWithPerm(Mat A,Vec bb,Vec xx)
1149137fb511SHong Zhang {
1150137fb511SHong Zhang   Mat_SeqAIJ      *a = (Mat_SeqAIJ*)A->data;
1151137fb511SHong Zhang   IS              iscol = a->col,isrow = a->row;
1152137fb511SHong Zhang   PetscErrorCode  ierr;
11535d0c19d7SBarry Smith   const PetscInt  *r,*c,*rout,*cout;
11545d0c19d7SBarry Smith   PetscInt        i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
11555d0c19d7SBarry Smith   PetscInt        nz,row;
1156dd6ea824SBarry Smith   PetscScalar     *x,*b,*tmp,*tmps,sum;
1157dd6ea824SBarry Smith   const MatScalar *aa = a->a,*v;
1158137fb511SHong Zhang 
1159137fb511SHong Zhang   PetscFunctionBegin;
1160137fb511SHong Zhang   if (!n) PetscFunctionReturn(0);
1161137fb511SHong Zhang 
1162137fb511SHong Zhang   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
1163137fb511SHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1164137fb511SHong Zhang   tmp  = a->solve_work;
1165137fb511SHong Zhang 
1166137fb511SHong Zhang   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
1167137fb511SHong Zhang   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
1168137fb511SHong Zhang 
1169137fb511SHong Zhang   /* forward solve the lower triangular */
1170137fb511SHong Zhang   tmp[0] = b[*r++];
1171137fb511SHong Zhang   tmps   = tmp;
1172137fb511SHong Zhang   for (row=1; row<n; row++) {
1173137fb511SHong Zhang     i   = rout[row]; /* permuted row */
1174137fb511SHong Zhang     v   = aa + ai[i] ;
1175137fb511SHong Zhang     vi  = aj + ai[i] ;
1176137fb511SHong Zhang     nz  = a->diag[i] - ai[i];
1177137fb511SHong Zhang     sum = b[*r++];
1178003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
1179137fb511SHong Zhang     tmp[row] = sum;
1180137fb511SHong Zhang   }
1181137fb511SHong Zhang 
1182137fb511SHong Zhang   /* backward solve the upper triangular */
1183137fb511SHong Zhang   for (row=n-1; row>=0; row--){
1184137fb511SHong Zhang     i   = rout[row]; /* permuted row */
1185137fb511SHong Zhang     v   = aa + a->diag[i] + 1;
1186137fb511SHong Zhang     vi  = aj + a->diag[i] + 1;
1187137fb511SHong Zhang     nz  = ai[i+1] - a->diag[i] - 1;
1188137fb511SHong Zhang     sum = tmp[row];
1189003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
1190137fb511SHong Zhang     x[*c--] = tmp[row] = sum*aa[a->diag[i]];
1191137fb511SHong Zhang   }
1192137fb511SHong Zhang 
1193137fb511SHong Zhang   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
1194137fb511SHong Zhang   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
1195137fb511SHong Zhang   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
1196137fb511SHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1197dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
1198137fb511SHong Zhang   PetscFunctionReturn(0);
1199137fb511SHong Zhang }
1200137fb511SHong Zhang 
1201930ae53cSBarry Smith /* ----------------------------------------------------------- */
1202c6db04a5SJed Brown #include <../src/mat/impls/aij/seq/ftn-kernels/fsolve.h>
12034a2ae208SSatish Balay #undef __FUNCT__
1204ad04f41aSHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering_inplace"
1205ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering_inplace(Mat A,Vec bb,Vec xx)
1206930ae53cSBarry Smith {
1207930ae53cSBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
12086849ba73SBarry Smith   PetscErrorCode    ierr;
1209d0f46423SBarry Smith   PetscInt          n = A->rmap->n;
1210b377110cSBarry Smith   const PetscInt    *ai = a->i,*aj = a->j,*adiag = a->diag;
1211356650c2SBarry Smith   PetscScalar       *x;
1212356650c2SBarry Smith   const PetscScalar *b;
1213dd6ea824SBarry Smith   const MatScalar   *aa = a->a;
1214aa482453SBarry Smith #if !defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ)
1215356650c2SBarry Smith   PetscInt          adiag_i,i,nz,ai_i;
1216b377110cSBarry Smith   const PetscInt    *vi;
1217dd6ea824SBarry Smith   const MatScalar   *v;
1218dd6ea824SBarry Smith   PetscScalar       sum;
1219d85afc42SSatish Balay #endif
1220930ae53cSBarry Smith 
12213a40ed3dSBarry Smith   PetscFunctionBegin;
12223a40ed3dSBarry Smith   if (!n) PetscFunctionReturn(0);
1223930ae53cSBarry Smith 
12243649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
12251ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1226930ae53cSBarry Smith 
1227aa482453SBarry Smith #if defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ)
12281c4feecaSSatish Balay   fortransolveaij_(&n,x,ai,aj,adiag,aa,b);
12291c4feecaSSatish Balay #else
1230930ae53cSBarry Smith   /* forward solve the lower triangular */
1231930ae53cSBarry Smith   x[0] = b[0];
1232930ae53cSBarry Smith   for (i=1; i<n; i++) {
1233930ae53cSBarry Smith     ai_i = ai[i];
1234930ae53cSBarry Smith     v    = aa + ai_i;
1235930ae53cSBarry Smith     vi   = aj + ai_i;
1236930ae53cSBarry Smith     nz   = adiag[i] - ai_i;
1237930ae53cSBarry Smith     sum  = b[i];
1238003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
1239930ae53cSBarry Smith     x[i] = sum;
1240930ae53cSBarry Smith   }
1241930ae53cSBarry Smith 
1242930ae53cSBarry Smith   /* backward solve the upper triangular */
1243930ae53cSBarry Smith   for (i=n-1; i>=0; i--){
1244930ae53cSBarry Smith     adiag_i = adiag[i];
1245d708749eSBarry Smith     v       = aa + adiag_i + 1;
1246d708749eSBarry Smith     vi      = aj + adiag_i + 1;
1247930ae53cSBarry Smith     nz      = ai[i+1] - adiag_i - 1;
1248930ae53cSBarry Smith     sum     = x[i];
1249003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
1250930ae53cSBarry Smith     x[i]    = sum*aa[adiag_i];
1251930ae53cSBarry Smith   }
12521c4feecaSSatish Balay #endif
1253dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
12543649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
12551ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
12563a40ed3dSBarry Smith   PetscFunctionReturn(0);
1257930ae53cSBarry Smith }
1258930ae53cSBarry Smith 
12594a2ae208SSatish Balay #undef __FUNCT__
1260ad04f41aSHong Zhang #define __FUNCT__ "MatSolveAdd_SeqAIJ_inplace"
1261ad04f41aSHong Zhang PetscErrorCode MatSolveAdd_SeqAIJ_inplace(Mat A,Vec bb,Vec yy,Vec xx)
1262da3a660dSBarry Smith {
1263416022c9SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
1264416022c9SBarry Smith   IS                iscol = a->col,isrow = a->row;
12656849ba73SBarry Smith   PetscErrorCode    ierr;
126604fbf559SBarry Smith   PetscInt          i, n = A->rmap->n,j;
12675d0c19d7SBarry Smith   PetscInt          nz;
126804fbf559SBarry Smith   const PetscInt    *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j;
126904fbf559SBarry Smith   PetscScalar       *x,*tmp,sum;
127004fbf559SBarry Smith   const PetscScalar *b;
1271dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
1272da3a660dSBarry Smith 
12733a40ed3dSBarry Smith   PetscFunctionBegin;
127478b31e54SBarry Smith   if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);}
1275da3a660dSBarry Smith 
12763649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
12771ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1278416022c9SBarry Smith   tmp  = a->solve_work;
1279da3a660dSBarry Smith 
12803b2fbd54SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
12813b2fbd54SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
1282da3a660dSBarry Smith 
1283da3a660dSBarry Smith   /* forward solve the lower triangular */
1284da3a660dSBarry Smith   tmp[0] = b[*r++];
1285da3a660dSBarry Smith   for (i=1; i<n; i++) {
1286010a20caSHong Zhang     v   = aa + ai[i] ;
1287010a20caSHong Zhang     vi  = aj + ai[i] ;
1288416022c9SBarry Smith     nz  = a->diag[i] - ai[i];
1289da3a660dSBarry Smith     sum = b[*r++];
129004fbf559SBarry Smith     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
1291da3a660dSBarry Smith     tmp[i] = sum;
1292da3a660dSBarry Smith   }
1293da3a660dSBarry Smith 
1294da3a660dSBarry Smith   /* backward solve the upper triangular */
1295da3a660dSBarry Smith   for (i=n-1; i>=0; i--){
1296010a20caSHong Zhang     v   = aa + a->diag[i] + 1;
1297010a20caSHong Zhang     vi  = aj + a->diag[i] + 1;
1298416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
1299da3a660dSBarry Smith     sum = tmp[i];
130004fbf559SBarry Smith     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
1301010a20caSHong Zhang     tmp[i] = sum*aa[a->diag[i]];
1302da3a660dSBarry Smith     x[*c--] += tmp[i];
1303da3a660dSBarry Smith   }
1304da3a660dSBarry Smith 
13053b2fbd54SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
13063b2fbd54SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
13073649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
13081ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1309dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
1310898183ecSLois Curfman McInnes 
13113a40ed3dSBarry Smith   PetscFunctionReturn(0);
1312da3a660dSBarry Smith }
131304fbf559SBarry Smith 
13144a2ae208SSatish Balay #undef __FUNCT__
131535233d99SShri Abhyankar #define __FUNCT__ "MatSolveAdd_SeqAIJ"
131635233d99SShri Abhyankar PetscErrorCode MatSolveAdd_SeqAIJ(Mat A,Vec bb,Vec yy,Vec xx)
13173c0119dfSShri Abhyankar {
13183c0119dfSShri Abhyankar   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
13193c0119dfSShri Abhyankar   IS                iscol = a->col,isrow = a->row;
13203c0119dfSShri Abhyankar   PetscErrorCode    ierr;
13213c0119dfSShri Abhyankar   PetscInt          i, n = A->rmap->n,j;
13223c0119dfSShri Abhyankar   PetscInt          nz;
13233c0119dfSShri Abhyankar   const PetscInt    *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j,*adiag = a->diag;
13243c0119dfSShri Abhyankar   PetscScalar       *x,*tmp,sum;
13253c0119dfSShri Abhyankar   const PetscScalar *b;
13263c0119dfSShri Abhyankar   const MatScalar   *aa = a->a,*v;
13273c0119dfSShri Abhyankar 
13283c0119dfSShri Abhyankar   PetscFunctionBegin;
13293c0119dfSShri Abhyankar   if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);}
13303c0119dfSShri Abhyankar 
13313649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
13323c0119dfSShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
13333c0119dfSShri Abhyankar   tmp  = a->solve_work;
13343c0119dfSShri Abhyankar 
13353c0119dfSShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
13363c0119dfSShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
13373c0119dfSShri Abhyankar 
13383c0119dfSShri Abhyankar   /* forward solve the lower triangular */
13393c0119dfSShri Abhyankar   tmp[0] = b[r[0]];
13403c0119dfSShri Abhyankar   v      = aa;
13413c0119dfSShri Abhyankar   vi     = aj;
13423c0119dfSShri Abhyankar   for (i=1; i<n; i++) {
13433c0119dfSShri Abhyankar     nz  = ai[i+1] - ai[i];
13443c0119dfSShri Abhyankar     sum = b[r[i]];
13453c0119dfSShri Abhyankar     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
13463c0119dfSShri Abhyankar     tmp[i] = sum;
13473c0119dfSShri Abhyankar     v += nz; vi += nz;
13483c0119dfSShri Abhyankar   }
13493c0119dfSShri Abhyankar 
13503c0119dfSShri Abhyankar   /* backward solve the upper triangular */
13513c0119dfSShri Abhyankar   v  = aa + adiag[n-1];
13523c0119dfSShri Abhyankar   vi = aj + adiag[n-1];
13533c0119dfSShri Abhyankar   for (i=n-1; i>=0; i--){
13543c0119dfSShri Abhyankar     nz  = adiag[i] - adiag[i+1] - 1;
13553c0119dfSShri Abhyankar     sum = tmp[i];
13563c0119dfSShri Abhyankar     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
13573c0119dfSShri Abhyankar     tmp[i] = sum*v[nz];
13583c0119dfSShri Abhyankar     x[c[i]] += tmp[i];
13593c0119dfSShri Abhyankar     v += nz+1; vi += nz+1;
13603c0119dfSShri Abhyankar   }
13613c0119dfSShri Abhyankar 
13623c0119dfSShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
13633c0119dfSShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
13643649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
13653c0119dfSShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
13663c0119dfSShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
13673c0119dfSShri Abhyankar 
13683c0119dfSShri Abhyankar   PetscFunctionReturn(0);
13693c0119dfSShri Abhyankar }
13703c0119dfSShri Abhyankar 
13713c0119dfSShri Abhyankar #undef __FUNCT__
1372ad04f41aSHong Zhang #define __FUNCT__ "MatSolveTranspose_SeqAIJ_inplace"
1373ad04f41aSHong Zhang PetscErrorCode MatSolveTranspose_SeqAIJ_inplace(Mat A,Vec bb,Vec xx)
1374da3a660dSBarry Smith {
1375416022c9SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
13762235e667SBarry Smith   IS                iscol = a->col,isrow = a->row;
13776849ba73SBarry Smith   PetscErrorCode    ierr;
137804fbf559SBarry Smith   const PetscInt    *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi;
137904fbf559SBarry Smith   PetscInt          i,n = A->rmap->n,j;
138004fbf559SBarry Smith   PetscInt          nz;
138104fbf559SBarry Smith   PetscScalar       *x,*tmp,s1;
1382dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
138304fbf559SBarry Smith   const PetscScalar *b;
1384da3a660dSBarry Smith 
13853a40ed3dSBarry Smith   PetscFunctionBegin;
13863649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
13871ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1388416022c9SBarry Smith   tmp  = a->solve_work;
1389da3a660dSBarry Smith 
13902235e667SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
13912235e667SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
1392da3a660dSBarry Smith 
1393da3a660dSBarry Smith   /* copy the b into temp work space according to permutation */
13942235e667SBarry Smith   for (i=0; i<n; i++) tmp[i] = b[c[i]];
1395da3a660dSBarry Smith 
1396da3a660dSBarry Smith   /* forward solve the U^T */
1397da3a660dSBarry Smith   for (i=0; i<n; i++) {
1398010a20caSHong Zhang     v   = aa + diag[i] ;
1399010a20caSHong Zhang     vi  = aj + diag[i] + 1;
1400f1af5d2fSBarry Smith     nz  = ai[i+1] - diag[i] - 1;
1401c38d4ed2SBarry Smith     s1  = tmp[i];
1402ef66eb69SBarry Smith     s1 *= (*v++);  /* multiply by inverse of diagonal entry */
140304fbf559SBarry Smith     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1404c38d4ed2SBarry Smith     tmp[i] = s1;
1405da3a660dSBarry Smith   }
1406da3a660dSBarry Smith 
1407da3a660dSBarry Smith   /* backward solve the L^T */
1408da3a660dSBarry Smith   for (i=n-1; i>=0; i--){
1409010a20caSHong Zhang     v   = aa + diag[i] - 1 ;
1410010a20caSHong Zhang     vi  = aj + diag[i] - 1 ;
1411f1af5d2fSBarry Smith     nz  = diag[i] - ai[i];
1412f1af5d2fSBarry Smith     s1  = tmp[i];
141304fbf559SBarry Smith     for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j];
1414da3a660dSBarry Smith   }
1415da3a660dSBarry Smith 
1416da3a660dSBarry Smith   /* copy tmp into x according to permutation */
1417591294e4SBarry Smith   for (i=0; i<n; i++) x[r[i]] = tmp[i];
1418da3a660dSBarry Smith 
14192235e667SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
14202235e667SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
14213649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
14221ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1423da3a660dSBarry Smith 
1424dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr);
14253a40ed3dSBarry Smith   PetscFunctionReturn(0);
1426da3a660dSBarry Smith }
1427da3a660dSBarry Smith 
1428d1fa9404SShri Abhyankar #undef __FUNCT__
142935233d99SShri Abhyankar #define __FUNCT__ "MatSolveTranspose_SeqAIJ"
143035233d99SShri Abhyankar PetscErrorCode MatSolveTranspose_SeqAIJ(Mat A,Vec bb,Vec xx)
1431d1fa9404SShri Abhyankar {
1432d1fa9404SShri Abhyankar   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
1433d1fa9404SShri Abhyankar   IS                iscol = a->col,isrow = a->row;
1434d1fa9404SShri Abhyankar   PetscErrorCode    ierr;
1435d1fa9404SShri Abhyankar   const PetscInt    *rout,*cout,*r,*c,*adiag = a->diag,*ai = a->i,*aj = a->j,*vi;
1436d1fa9404SShri Abhyankar   PetscInt          i,n = A->rmap->n,j;
1437d1fa9404SShri Abhyankar   PetscInt          nz;
1438d1fa9404SShri Abhyankar   PetscScalar       *x,*tmp,s1;
1439d1fa9404SShri Abhyankar   const MatScalar   *aa = a->a,*v;
1440d1fa9404SShri Abhyankar   const PetscScalar *b;
1441d1fa9404SShri Abhyankar 
1442d1fa9404SShri Abhyankar   PetscFunctionBegin;
14433649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
1444d1fa9404SShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1445d1fa9404SShri Abhyankar   tmp  = a->solve_work;
1446d1fa9404SShri Abhyankar 
1447d1fa9404SShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
1448d1fa9404SShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
1449d1fa9404SShri Abhyankar 
1450d1fa9404SShri Abhyankar   /* copy the b into temp work space according to permutation */
1451d1fa9404SShri Abhyankar   for (i=0; i<n; i++) tmp[i] = b[c[i]];
1452d1fa9404SShri Abhyankar 
1453d1fa9404SShri Abhyankar   /* forward solve the U^T */
1454d1fa9404SShri Abhyankar   for (i=0; i<n; i++) {
1455d1fa9404SShri Abhyankar     v   = aa + adiag[i+1] + 1;
1456d1fa9404SShri Abhyankar     vi  = aj + adiag[i+1] + 1;
1457d1fa9404SShri Abhyankar     nz  = adiag[i] - adiag[i+1] - 1;
1458d1fa9404SShri Abhyankar     s1  = tmp[i];
1459d1fa9404SShri Abhyankar     s1 *= v[nz];  /* multiply by inverse of diagonal entry */
1460d1fa9404SShri Abhyankar     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1461d1fa9404SShri Abhyankar     tmp[i] = s1;
1462d1fa9404SShri Abhyankar   }
1463d1fa9404SShri Abhyankar 
1464d1fa9404SShri Abhyankar   /* backward solve the L^T */
1465d1fa9404SShri Abhyankar   for (i=n-1; i>=0; i--){
1466d1fa9404SShri Abhyankar     v   = aa + ai[i];
1467d1fa9404SShri Abhyankar     vi  = aj + ai[i];
1468d1fa9404SShri Abhyankar     nz  = ai[i+1] - ai[i];
1469d1fa9404SShri Abhyankar     s1  = tmp[i];
1470d1fa9404SShri Abhyankar     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1471d1fa9404SShri Abhyankar   }
1472d1fa9404SShri Abhyankar 
1473d1fa9404SShri Abhyankar   /* copy tmp into x according to permutation */
1474d1fa9404SShri Abhyankar   for (i=0; i<n; i++) x[r[i]] = tmp[i];
1475d1fa9404SShri Abhyankar 
1476d1fa9404SShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
1477d1fa9404SShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
14783649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
1479d1fa9404SShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1480d1fa9404SShri Abhyankar 
1481d1fa9404SShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr);
1482d1fa9404SShri Abhyankar   PetscFunctionReturn(0);
1483d1fa9404SShri Abhyankar }
1484d1fa9404SShri Abhyankar 
14854a2ae208SSatish Balay #undef __FUNCT__
1486ad04f41aSHong Zhang #define __FUNCT__ "MatSolveTransposeAdd_SeqAIJ_inplace"
1487ad04f41aSHong Zhang PetscErrorCode MatSolveTransposeAdd_SeqAIJ_inplace(Mat A,Vec bb,Vec zz,Vec xx)
1488da3a660dSBarry Smith {
1489416022c9SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
14902235e667SBarry Smith   IS                iscol = a->col,isrow = a->row;
14916849ba73SBarry Smith   PetscErrorCode    ierr;
149204fbf559SBarry Smith   const PetscInt    *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi;
149304fbf559SBarry Smith   PetscInt          i,n = A->rmap->n,j;
149404fbf559SBarry Smith   PetscInt          nz;
149504fbf559SBarry Smith   PetscScalar       *x,*tmp,s1;
1496dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
149704fbf559SBarry Smith   const PetscScalar *b;
14986abc6512SBarry Smith 
14993a40ed3dSBarry Smith   PetscFunctionBegin;
150023bc6035SMatthew Knepley   if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);}
15013649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
15021ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1503416022c9SBarry Smith   tmp  = a->solve_work;
15046abc6512SBarry Smith 
15052235e667SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
15062235e667SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
15076abc6512SBarry Smith 
15086abc6512SBarry Smith   /* copy the b into temp work space according to permutation */
15092235e667SBarry Smith   for (i=0; i<n; i++) tmp[i] = b[c[i]];
15106abc6512SBarry Smith 
15116abc6512SBarry Smith   /* forward solve the U^T */
15126abc6512SBarry Smith   for (i=0; i<n; i++) {
1513010a20caSHong Zhang     v   = aa + diag[i] ;
1514010a20caSHong Zhang     vi  = aj + diag[i] + 1;
1515f1af5d2fSBarry Smith     nz  = ai[i+1] - diag[i] - 1;
151604fbf559SBarry Smith     s1  = tmp[i];
151704fbf559SBarry Smith     s1 *= (*v++);  /* multiply by inverse of diagonal entry */
151804fbf559SBarry Smith     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
151904fbf559SBarry Smith     tmp[i] = s1;
15206abc6512SBarry Smith   }
15216abc6512SBarry Smith 
15226abc6512SBarry Smith   /* backward solve the L^T */
15236abc6512SBarry Smith   for (i=n-1; i>=0; i--){
1524010a20caSHong Zhang     v   = aa + diag[i] - 1 ;
1525010a20caSHong Zhang     vi  = aj + diag[i] - 1 ;
1526f1af5d2fSBarry Smith     nz  = diag[i] - ai[i];
152704fbf559SBarry Smith     s1  = tmp[i];
152804fbf559SBarry Smith     for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j];
15296abc6512SBarry Smith   }
15306abc6512SBarry Smith 
15316abc6512SBarry Smith   /* copy tmp into x according to permutation */
15326abc6512SBarry Smith   for (i=0; i<n; i++) x[r[i]] += tmp[i];
15336abc6512SBarry Smith 
15342235e667SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
15352235e667SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
15363649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
15371ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
15386abc6512SBarry Smith 
153904fbf559SBarry Smith   ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr);
15403a40ed3dSBarry Smith   PetscFunctionReturn(0);
1541da3a660dSBarry Smith }
154204fbf559SBarry Smith 
1543533e6011SShri Abhyankar #undef __FUNCT__
154435233d99SShri Abhyankar #define __FUNCT__ "MatSolveTransposeAdd_SeqAIJ"
154535233d99SShri Abhyankar PetscErrorCode MatSolveTransposeAdd_SeqAIJ(Mat A,Vec bb,Vec zz,Vec xx)
1546533e6011SShri Abhyankar {
1547533e6011SShri Abhyankar   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
1548533e6011SShri Abhyankar   IS                iscol = a->col,isrow = a->row;
1549533e6011SShri Abhyankar   PetscErrorCode    ierr;
1550533e6011SShri Abhyankar   const PetscInt    *rout,*cout,*r,*c,*adiag = a->diag,*ai = a->i,*aj = a->j,*vi;
1551533e6011SShri Abhyankar   PetscInt          i,n = A->rmap->n,j;
1552533e6011SShri Abhyankar   PetscInt          nz;
1553533e6011SShri Abhyankar   PetscScalar       *x,*tmp,s1;
1554533e6011SShri Abhyankar   const MatScalar   *aa = a->a,*v;
1555533e6011SShri Abhyankar   const PetscScalar *b;
1556533e6011SShri Abhyankar 
1557533e6011SShri Abhyankar   PetscFunctionBegin;
1558533e6011SShri Abhyankar   if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);}
15593649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
1560533e6011SShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1561533e6011SShri Abhyankar   tmp  = a->solve_work;
1562533e6011SShri Abhyankar 
1563533e6011SShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
1564533e6011SShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
1565533e6011SShri Abhyankar 
1566533e6011SShri Abhyankar   /* copy the b into temp work space according to permutation */
1567533e6011SShri Abhyankar   for (i=0; i<n; i++) tmp[i] = b[c[i]];
1568533e6011SShri Abhyankar 
1569533e6011SShri Abhyankar   /* forward solve the U^T */
1570533e6011SShri Abhyankar   for (i=0; i<n; i++) {
1571533e6011SShri Abhyankar     v   = aa + adiag[i+1] + 1;
1572533e6011SShri Abhyankar     vi  = aj + adiag[i+1] + 1;
1573533e6011SShri Abhyankar     nz  = adiag[i] - adiag[i+1] - 1;
1574533e6011SShri Abhyankar     s1  = tmp[i];
1575533e6011SShri Abhyankar     s1 *= v[nz];  /* multiply by inverse of diagonal entry */
1576533e6011SShri Abhyankar     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1577533e6011SShri Abhyankar     tmp[i] = s1;
1578533e6011SShri Abhyankar   }
1579533e6011SShri Abhyankar 
1580533e6011SShri Abhyankar 
1581533e6011SShri Abhyankar   /* backward solve the L^T */
1582533e6011SShri Abhyankar   for (i=n-1; i>=0; i--){
1583533e6011SShri Abhyankar     v   = aa + ai[i] ;
1584533e6011SShri Abhyankar     vi  = aj + ai[i];
1585533e6011SShri Abhyankar     nz  = ai[i+1] - ai[i];
1586533e6011SShri Abhyankar     s1  = tmp[i];
1587c5e3b2a3SShri Abhyankar     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1588533e6011SShri Abhyankar   }
1589533e6011SShri Abhyankar 
1590533e6011SShri Abhyankar   /* copy tmp into x according to permutation */
1591533e6011SShri Abhyankar   for (i=0; i<n; i++) x[r[i]] += tmp[i];
1592533e6011SShri Abhyankar 
1593533e6011SShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
1594533e6011SShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
15953649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
1596533e6011SShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1597533e6011SShri Abhyankar 
1598533e6011SShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr);
1599533e6011SShri Abhyankar   PetscFunctionReturn(0);
1600533e6011SShri Abhyankar }
1601533e6011SShri Abhyankar 
16029e25ed09SBarry Smith /* ----------------------------------------------------------------*/
1603b47f5a65SHong Zhang 
160409573ac7SBarry Smith extern PetscErrorCode MatDuplicateNoCreate_SeqAIJ(Mat,Mat,MatDuplicateOption,PetscBool );
160508480c60SBarry Smith 
16068fc3a347SHong Zhang /*
16078a76ed4fSHong Zhang    ilu() under revised new data structure.
1608813a1f2bSShri Abhyankar    Factored arrays bj and ba are stored as
1609813a1f2bSShri Abhyankar      L(0,:), L(1,:), ...,L(n-1,:),  U(n-1,:),...,U(i,:),U(i-1,:),...,U(0,:)
1610813a1f2bSShri Abhyankar 
1611813a1f2bSShri Abhyankar    bi=fact->i is an array of size n+1, in which
1612813a1f2bSShri Abhyankar    bi+
1613baabb860SHong Zhang      bi[i]:  points to 1st entry of L(i,:),i=0,...,n-1
1614baabb860SHong Zhang      bi[n]:  points to L(n-1,n-1)+1
1615baabb860SHong Zhang 
1616813a1f2bSShri Abhyankar   bdiag=fact->diag is an array of size n+1,in which
1617baabb860SHong Zhang      bdiag[i]: points to diagonal of U(i,:), i=0,...,n-1
1618a24f213cSHong Zhang      bdiag[n]: points to entry of U(n-1,0)-1
1619813a1f2bSShri Abhyankar 
1620813a1f2bSShri Abhyankar    U(i,:) contains bdiag[i] as its last entry, i.e.,
1621813a1f2bSShri Abhyankar     U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i])
1622813a1f2bSShri Abhyankar */
1623813a1f2bSShri Abhyankar #undef __FUNCT__
162435233d99SShri Abhyankar #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_ilu0"
162535233d99SShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ_ilu0(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
1626813a1f2bSShri Abhyankar {
1627813a1f2bSShri Abhyankar 
1628813a1f2bSShri Abhyankar   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
1629813a1f2bSShri Abhyankar   PetscErrorCode     ierr;
1630bbd65245SShri Abhyankar   const PetscInt     n=A->rmap->n,*ai=a->i,*aj,*adiag=a->diag;
1631bbd65245SShri Abhyankar   PetscInt           i,j,k=0,nz,*bi,*bj,*bdiag;
1632ace3abfcSBarry Smith   PetscBool          missing;
16331df811f5SHong Zhang   IS                 isicol;
1634813a1f2bSShri Abhyankar 
1635813a1f2bSShri Abhyankar   PetscFunctionBegin;
1636e32f2f54SBarry 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);
16371df811f5SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr);
1638e32f2f54SBarry Smith   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i);
16391df811f5SHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
16401df811f5SHong Zhang 
1641813a1f2bSShri Abhyankar   ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_FALSE);CHKERRQ(ierr);
1642813a1f2bSShri Abhyankar   b    = (Mat_SeqAIJ*)(fact)->data;
1643813a1f2bSShri Abhyankar 
1644813a1f2bSShri Abhyankar   /* allocate matrix arrays for new data structure */
1645813a1f2bSShri Abhyankar   ierr = PetscMalloc3(ai[n]+1,PetscScalar,&b->a,ai[n]+1,PetscInt,&b->j,n+1,PetscInt,&b->i);CHKERRQ(ierr);
1646813a1f2bSShri Abhyankar   ierr = PetscLogObjectMemory(fact,ai[n]*(sizeof(PetscScalar)+sizeof(PetscInt))+(n+1)*sizeof(PetscInt));CHKERRQ(ierr);
1647813a1f2bSShri Abhyankar   b->singlemalloc = PETSC_TRUE;
1648813a1f2bSShri Abhyankar   if (!b->diag){
1649813a1f2bSShri Abhyankar     ierr = PetscMalloc((n+1)*sizeof(PetscInt),&b->diag);CHKERRQ(ierr);
1650813a1f2bSShri Abhyankar     ierr = PetscLogObjectMemory(fact,(n+1)*sizeof(PetscInt));CHKERRQ(ierr);
1651813a1f2bSShri Abhyankar   }
1652813a1f2bSShri Abhyankar   bdiag = b->diag;
1653813a1f2bSShri Abhyankar 
1654813a1f2bSShri Abhyankar   if (n > 0) {
1655813a1f2bSShri Abhyankar     ierr = PetscMemzero(b->a,(ai[n])*sizeof(MatScalar));CHKERRQ(ierr);
1656813a1f2bSShri Abhyankar   }
1657813a1f2bSShri Abhyankar 
1658813a1f2bSShri Abhyankar   /* set bi and bj with new data structure */
1659813a1f2bSShri Abhyankar   bi = b->i;
1660813a1f2bSShri Abhyankar   bj = b->j;
1661813a1f2bSShri Abhyankar 
1662813a1f2bSShri Abhyankar   /* L part */
1663e60cf9a0SBarry Smith   bi[0] = 0;
1664813a1f2bSShri Abhyankar   for (i=0; i<n; i++){
1665813a1f2bSShri Abhyankar     nz = adiag[i] - ai[i];
1666813a1f2bSShri Abhyankar     bi[i+1] = bi[i] + nz;
1667813a1f2bSShri Abhyankar     aj = a->j + ai[i];
1668813a1f2bSShri Abhyankar     for (j=0; j<nz; j++){
1669bbd65245SShri Abhyankar       /*   *bj = aj[j]; bj++; */
1670bbd65245SShri Abhyankar       bj[k++] = aj[j];
1671813a1f2bSShri Abhyankar     }
1672813a1f2bSShri Abhyankar   }
1673813a1f2bSShri Abhyankar 
1674813a1f2bSShri Abhyankar   /* U part */
167562fbd6afSShri Abhyankar   bdiag[n] = bi[n]-1;
1676813a1f2bSShri Abhyankar   for (i=n-1; i>=0; i--){
1677813a1f2bSShri Abhyankar     nz = ai[i+1] - adiag[i] - 1;
1678813a1f2bSShri Abhyankar     aj = a->j + adiag[i] + 1;
1679813a1f2bSShri Abhyankar     for (j=0; j<nz; j++){
1680bbd65245SShri Abhyankar       /*      *bj = aj[j]; bj++; */
1681bbd65245SShri Abhyankar       bj[k++] = aj[j];
1682813a1f2bSShri Abhyankar     }
1683813a1f2bSShri Abhyankar     /* diag[i] */
1684bbd65245SShri Abhyankar     /*    *bj = i; bj++; */
1685bbd65245SShri Abhyankar     bj[k++] = i;
1686a24f213cSHong Zhang     bdiag[i] = bdiag[i+1] + nz + 1;
1687813a1f2bSShri Abhyankar   }
16881df811f5SHong Zhang 
1689d5f3da31SBarry Smith   fact->factortype             = MAT_FACTOR_ILU;
16901df811f5SHong Zhang   fact->info.factor_mallocs    = 0;
16911df811f5SHong Zhang   fact->info.fill_ratio_given  = info->fill;
16921df811f5SHong Zhang   fact->info.fill_ratio_needed = 1.0;
169335233d99SShri Abhyankar   fact->ops->lufactornumeric   = MatLUFactorNumeric_SeqAIJ;
16941df811f5SHong Zhang 
16951df811f5SHong Zhang   b       = (Mat_SeqAIJ*)(fact)->data;
16961df811f5SHong Zhang   b->row  = isrow;
16971df811f5SHong Zhang   b->col  = iscol;
16981df811f5SHong Zhang   b->icol = isicol;
16991df811f5SHong Zhang   ierr    = PetscMalloc((fact->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
17001df811f5SHong Zhang   ierr    = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
17011df811f5SHong Zhang   ierr    = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
1702813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
1703813a1f2bSShri Abhyankar }
1704813a1f2bSShri Abhyankar 
1705813a1f2bSShri Abhyankar #undef __FUNCT__
170635233d99SShri Abhyankar #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ"
170735233d99SShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
1708813a1f2bSShri Abhyankar {
1709813a1f2bSShri Abhyankar   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
1710813a1f2bSShri Abhyankar   IS                 isicol;
1711813a1f2bSShri Abhyankar   PetscErrorCode     ierr;
1712813a1f2bSShri Abhyankar   const PetscInt     *r,*ic;
17131df811f5SHong Zhang   PetscInt           n=A->rmap->n,*ai=a->i,*aj=a->j;
1714813a1f2bSShri Abhyankar   PetscInt           *bi,*cols,nnz,*cols_lvl;
1715813a1f2bSShri Abhyankar   PetscInt           *bdiag,prow,fm,nzbd,reallocs=0,dcount=0;
1716813a1f2bSShri Abhyankar   PetscInt           i,levels,diagonal_fill;
1717ace3abfcSBarry Smith   PetscBool          col_identity,row_identity;
1718813a1f2bSShri Abhyankar   PetscReal          f;
1719813a1f2bSShri Abhyankar   PetscInt           nlnk,*lnk,*lnk_lvl=PETSC_NULL;
1720813a1f2bSShri Abhyankar   PetscBT            lnkbt;
1721813a1f2bSShri Abhyankar   PetscInt           nzi,*bj,**bj_ptr,**bjlvl_ptr;
1722813a1f2bSShri Abhyankar   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
1723813a1f2bSShri Abhyankar   PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL;
1724813a1f2bSShri Abhyankar 
1725813a1f2bSShri Abhyankar   PetscFunctionBegin;
17263898ab1dSSatish Balay   /* Uncomment the old data struct part only while testing new data structure for MatSolve() */
1727124a3351SHong Zhang   /*
1728ace3abfcSBarry Smith   PetscBool          olddatastruct=PETSC_FALSE;
1729acfcf0e5SJed Brown   ierr = PetscOptionsGetBool(PETSC_NULL,"-ilu_old",&olddatastruct,PETSC_NULL);CHKERRQ(ierr);
173035233d99SShri Abhyankar   if(olddatastruct){
1731ad04f41aSHong Zhang     ierr = MatILUFactorSymbolic_SeqAIJ_inplace(fact,A,isrow,iscol,info);CHKERRQ(ierr);
1732ad04f41aSHong Zhang     PetscFunctionReturn(0);
1733ad04f41aSHong Zhang   }
1734124a3351SHong Zhang   */
1735ad04f41aSHong Zhang 
1736813a1f2bSShri Abhyankar   levels = (PetscInt)info->levels;
1737813a1f2bSShri Abhyankar   ierr   = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
1738813a1f2bSShri Abhyankar   ierr   = ISIdentity(iscol,&col_identity);CHKERRQ(ierr);
1739813a1f2bSShri Abhyankar 
1740813a1f2bSShri Abhyankar   if (!levels && row_identity && col_identity) {
1741813a1f2bSShri Abhyankar     /* special case: ilu(0) with natural ordering */
174235233d99SShri Abhyankar     ierr = MatILUFactorSymbolic_SeqAIJ_ilu0(fact,A,isrow,iscol,info);CHKERRQ(ierr);
1743124a3351SHong Zhang     if (a->inode.size) {
1744124a3351SHong Zhang       fact->ops->lufactornumeric  = MatLUFactorNumeric_SeqAIJ_Inode;
1745124a3351SHong Zhang     }
1746813a1f2bSShri Abhyankar     PetscFunctionReturn(0);
1747813a1f2bSShri Abhyankar   }
1748813a1f2bSShri Abhyankar 
1749e32f2f54SBarry 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);
17501df811f5SHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
1751813a1f2bSShri Abhyankar   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
1752813a1f2bSShri Abhyankar   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
1753813a1f2bSShri Abhyankar 
17541bfa06eaSJed Brown   /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */
17551bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
17561bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
1757e60cf9a0SBarry Smith   bi[0] = bdiag[0] = 0;
1758813a1f2bSShri Abhyankar 
17590e83c824SBarry Smith   ierr = PetscMalloc2(n,PetscInt*,&bj_ptr,n,PetscInt*,&bjlvl_ptr);CHKERRQ(ierr);
1760813a1f2bSShri Abhyankar 
1761813a1f2bSShri Abhyankar   /* create a linked list for storing column indices of the active row */
1762813a1f2bSShri Abhyankar   nlnk = n + 1;
1763813a1f2bSShri Abhyankar   ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
1764813a1f2bSShri Abhyankar 
1765813a1f2bSShri Abhyankar   /* initial FreeSpace size is f*(ai[n]+1) */
17661df811f5SHong Zhang   f             = info->fill;
17671df811f5SHong Zhang   diagonal_fill = (PetscInt)info->diagonal_fill;
1768813a1f2bSShri Abhyankar   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr);
1769813a1f2bSShri Abhyankar   current_space = free_space;
1770813a1f2bSShri Abhyankar   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr);
1771813a1f2bSShri Abhyankar   current_space_lvl = free_space_lvl;
1772813a1f2bSShri Abhyankar 
1773813a1f2bSShri Abhyankar   for (i=0; i<n; i++) {
1774813a1f2bSShri Abhyankar     nzi = 0;
1775813a1f2bSShri Abhyankar     /* copy current row into linked list */
1776813a1f2bSShri Abhyankar     nnz  = ai[r[i]+1] - ai[r[i]];
1777e32f2f54SBarry 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);
1778813a1f2bSShri Abhyankar     cols = aj + ai[r[i]];
1779813a1f2bSShri Abhyankar     lnk[i] = -1; /* marker to indicate if diagonal exists */
1780813a1f2bSShri Abhyankar     ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
1781813a1f2bSShri Abhyankar     nzi += nlnk;
1782813a1f2bSShri Abhyankar 
1783813a1f2bSShri Abhyankar     /* make sure diagonal entry is included */
1784813a1f2bSShri Abhyankar     if (diagonal_fill && lnk[i] == -1) {
1785813a1f2bSShri Abhyankar       fm = n;
1786813a1f2bSShri Abhyankar       while (lnk[fm] < i) fm = lnk[fm];
1787813a1f2bSShri Abhyankar       lnk[i]     = lnk[fm]; /* insert diagonal into linked list */
1788813a1f2bSShri Abhyankar       lnk[fm]    = i;
1789813a1f2bSShri Abhyankar       lnk_lvl[i] = 0;
1790813a1f2bSShri Abhyankar       nzi++; dcount++;
1791813a1f2bSShri Abhyankar     }
1792813a1f2bSShri Abhyankar 
1793813a1f2bSShri Abhyankar     /* add pivot rows into the active row */
1794813a1f2bSShri Abhyankar     nzbd = 0;
1795813a1f2bSShri Abhyankar     prow = lnk[n];
1796813a1f2bSShri Abhyankar     while (prow < i) {
1797813a1f2bSShri Abhyankar       nnz      = bdiag[prow];
1798813a1f2bSShri Abhyankar       cols     = bj_ptr[prow] + nnz + 1;
1799813a1f2bSShri Abhyankar       cols_lvl = bjlvl_ptr[prow] + nnz + 1;
1800813a1f2bSShri Abhyankar       nnz      = bi[prow+1] - bi[prow] - nnz - 1;
1801813a1f2bSShri Abhyankar       ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr);
1802813a1f2bSShri Abhyankar       nzi += nlnk;
1803813a1f2bSShri Abhyankar       prow = lnk[prow];
1804813a1f2bSShri Abhyankar       nzbd++;
1805813a1f2bSShri Abhyankar     }
1806813a1f2bSShri Abhyankar     bdiag[i] = nzbd;
1807813a1f2bSShri Abhyankar     bi[i+1]  = bi[i] + nzi;
1808813a1f2bSShri Abhyankar 
1809813a1f2bSShri Abhyankar     /* if free space is not available, make more free space */
1810813a1f2bSShri Abhyankar     if (current_space->local_remaining<nzi) {
1811813a1f2bSShri Abhyankar       nnz = 2*nzi*(n - i); /* estimated and max additional space needed */
1812813a1f2bSShri Abhyankar       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
1813813a1f2bSShri Abhyankar       ierr = PetscFreeSpaceGet(nnz,&current_space_lvl);CHKERRQ(ierr);
1814813a1f2bSShri Abhyankar       reallocs++;
1815813a1f2bSShri Abhyankar     }
1816813a1f2bSShri Abhyankar 
1817813a1f2bSShri Abhyankar     /* copy data into free_space and free_space_lvl, then initialize lnk */
1818813a1f2bSShri Abhyankar     ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
1819813a1f2bSShri Abhyankar     bj_ptr[i]    = current_space->array;
1820813a1f2bSShri Abhyankar     bjlvl_ptr[i] = current_space_lvl->array;
1821813a1f2bSShri Abhyankar 
1822813a1f2bSShri Abhyankar     /* make sure the active row i has diagonal entry */
182365e19b50SBarry 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);
1824813a1f2bSShri Abhyankar 
1825813a1f2bSShri Abhyankar     current_space->array           += nzi;
1826813a1f2bSShri Abhyankar     current_space->local_used      += nzi;
1827813a1f2bSShri Abhyankar     current_space->local_remaining -= nzi;
1828813a1f2bSShri Abhyankar     current_space_lvl->array           += nzi;
1829813a1f2bSShri Abhyankar     current_space_lvl->local_used      += nzi;
1830813a1f2bSShri Abhyankar     current_space_lvl->local_remaining -= nzi;
1831813a1f2bSShri Abhyankar   }
1832813a1f2bSShri Abhyankar 
1833813a1f2bSShri Abhyankar   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
1834813a1f2bSShri Abhyankar   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
1835813a1f2bSShri Abhyankar 
1836813a1f2bSShri Abhyankar   /* copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */
18379263d837SHong Zhang   ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
18382ce24eb6SHong Zhang   ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr);
1839813a1f2bSShri Abhyankar 
1840813a1f2bSShri Abhyankar   ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
1841813a1f2bSShri Abhyankar   ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
18420e83c824SBarry Smith   ierr = PetscFree2(bj_ptr,bjlvl_ptr);CHKERRQ(ierr);
1843813a1f2bSShri Abhyankar 
1844813a1f2bSShri Abhyankar #if defined(PETSC_USE_INFO)
1845813a1f2bSShri Abhyankar   {
1846aef85c9fSShri Abhyankar     PetscReal af = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]);
1847813a1f2bSShri Abhyankar     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr);
1848813a1f2bSShri Abhyankar     ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
1849813a1f2bSShri Abhyankar     ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr);
1850813a1f2bSShri Abhyankar     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
1851813a1f2bSShri Abhyankar     if (diagonal_fill) {
1852813a1f2bSShri Abhyankar       ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr);
1853813a1f2bSShri Abhyankar     }
1854813a1f2bSShri Abhyankar   }
1855813a1f2bSShri Abhyankar #endif
1856813a1f2bSShri Abhyankar 
1857813a1f2bSShri Abhyankar   /* put together the new matrix */
1858813a1f2bSShri Abhyankar   ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
1859813a1f2bSShri Abhyankar   ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr);
1860813a1f2bSShri Abhyankar   b = (Mat_SeqAIJ*)(fact)->data;
1861813a1f2bSShri Abhyankar   b->free_a       = PETSC_TRUE;
1862813a1f2bSShri Abhyankar   b->free_ij      = PETSC_TRUE;
1863813a1f2bSShri Abhyankar   b->singlemalloc = PETSC_FALSE;
1864f268cba8SShri Abhyankar   ierr = PetscMalloc((bdiag[0]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);
1865813a1f2bSShri Abhyankar   b->j          = bj;
1866813a1f2bSShri Abhyankar   b->i          = bi;
1867813a1f2bSShri Abhyankar   b->diag       = bdiag;
1868813a1f2bSShri Abhyankar   b->ilen       = 0;
1869813a1f2bSShri Abhyankar   b->imax       = 0;
1870813a1f2bSShri Abhyankar   b->row        = isrow;
1871813a1f2bSShri Abhyankar   b->col        = iscol;
1872813a1f2bSShri Abhyankar   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
1873813a1f2bSShri Abhyankar   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
1874813a1f2bSShri Abhyankar   b->icol       = isicol;
1875813a1f2bSShri Abhyankar   ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
1876813a1f2bSShri Abhyankar   /* In b structure:  Free imax, ilen, old a, old j.
1877813a1f2bSShri Abhyankar      Allocate bdiag, solve_work, new a, new j */
1878f268cba8SShri Abhyankar   ierr = PetscLogObjectMemory(fact,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
1879f268cba8SShri Abhyankar   b->maxnz = b->nz = bdiag[0]+1;
1880813a1f2bSShri Abhyankar   (fact)->info.factor_mallocs    = reallocs;
1881813a1f2bSShri Abhyankar   (fact)->info.fill_ratio_given  = f;
1882f268cba8SShri Abhyankar   (fact)->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]);
188335233d99SShri Abhyankar   (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ;
1884f2344125SBarry Smith   if (a->inode.size) {
1885f2344125SBarry Smith     (fact)->ops->lufactornumeric  = MatLUFactorNumeric_SeqAIJ_Inode;
1886f2344125SBarry Smith   }
1887813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
1888813a1f2bSShri Abhyankar }
1889813a1f2bSShri Abhyankar 
18906516239fSHong Zhang #undef __FUNCT__
1891ad04f41aSHong Zhang #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_inplace"
1892ad04f41aSHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
18936516239fSHong Zhang {
18946516239fSHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
18956516239fSHong Zhang   IS                 isicol;
18966516239fSHong Zhang   PetscErrorCode     ierr;
18976516239fSHong Zhang   const PetscInt     *r,*ic;
18986516239fSHong Zhang   PetscInt           n=A->rmap->n,*ai=a->i,*aj=a->j,d;
18996516239fSHong Zhang   PetscInt           *bi,*cols,nnz,*cols_lvl;
19006516239fSHong Zhang   PetscInt           *bdiag,prow,fm,nzbd,reallocs=0,dcount=0;
19016516239fSHong Zhang   PetscInt           i,levels,diagonal_fill;
1902ace3abfcSBarry Smith   PetscBool          col_identity,row_identity;
19036516239fSHong Zhang   PetscReal          f;
19046516239fSHong Zhang   PetscInt           nlnk,*lnk,*lnk_lvl=PETSC_NULL;
19056516239fSHong Zhang   PetscBT            lnkbt;
19066516239fSHong Zhang   PetscInt           nzi,*bj,**bj_ptr,**bjlvl_ptr;
19076516239fSHong Zhang   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
19086516239fSHong Zhang   PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL;
1909ace3abfcSBarry Smith   PetscBool          missing;
19106516239fSHong Zhang 
19116516239fSHong Zhang   PetscFunctionBegin;
1912e32f2f54SBarry 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);
19136516239fSHong Zhang   f             = info->fill;
19146516239fSHong Zhang   levels        = (PetscInt)info->levels;
19156516239fSHong Zhang   diagonal_fill = (PetscInt)info->diagonal_fill;
19166516239fSHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
19176516239fSHong Zhang 
19186516239fSHong Zhang   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
19196516239fSHong Zhang   ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr);
19206516239fSHong Zhang   if (!levels && row_identity && col_identity) { /* special case: ilu(0) with natural ordering */
1921f77e22a1SHong Zhang     ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_TRUE);CHKERRQ(ierr);
1922ad04f41aSHong Zhang     (fact)->ops->lufactornumeric =  MatLUFactorNumeric_SeqAIJ_inplace;
1923f2344125SBarry Smith     if (a->inode.size) {
1924f2344125SBarry Smith       (fact)->ops->lufactornumeric  = MatLUFactorNumeric_SeqAIJ_Inode_inplace;
1925f2344125SBarry Smith     }
1926d5f3da31SBarry Smith     fact->factortype = MAT_FACTOR_ILU;
1927719d5645SBarry Smith     (fact)->info.factor_mallocs    = 0;
1928719d5645SBarry Smith     (fact)->info.fill_ratio_given  = info->fill;
1929719d5645SBarry Smith     (fact)->info.fill_ratio_needed = 1.0;
1930719d5645SBarry Smith     b               = (Mat_SeqAIJ*)(fact)->data;
19318ef3462fSBarry Smith     ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr);
1932e32f2f54SBarry Smith     if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d);
193308480c60SBarry Smith     b->row              = isrow;
193408480c60SBarry Smith     b->col              = iscol;
193582bf6240SBarry Smith     b->icol             = isicol;
1936719d5645SBarry Smith     ierr                = PetscMalloc(((fact)->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
1937c38d4ed2SBarry Smith     ierr                = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
1938c38d4ed2SBarry Smith     ierr                = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
19393a40ed3dSBarry Smith     PetscFunctionReturn(0);
194008480c60SBarry Smith   }
194108480c60SBarry Smith 
1942898183ecSLois Curfman McInnes   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
1943898183ecSLois Curfman McInnes   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
19449e25ed09SBarry Smith 
19451bfa06eaSJed Brown   /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */
19461bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
19471bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
1948a6df321fSShri Abhyankar   bi[0] = bdiag[0] = 0;
19496cf997b0SBarry Smith 
19500e83c824SBarry Smith   ierr = PetscMalloc2(n,PetscInt*,&bj_ptr,n,PetscInt*,&bjlvl_ptr);CHKERRQ(ierr);
19515a8e39fbSHong Zhang 
19525a8e39fbSHong Zhang   /* create a linked list for storing column indices of the active row */
19535a8e39fbSHong Zhang   nlnk = n + 1;
19545a8e39fbSHong Zhang   ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
19555a8e39fbSHong Zhang 
19565a8e39fbSHong Zhang   /* initial FreeSpace size is f*(ai[n]+1) */
1957a1a86e44SBarry Smith   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr);
19585a8e39fbSHong Zhang   current_space = free_space;
1959a1a86e44SBarry Smith   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr);
19605a8e39fbSHong Zhang   current_space_lvl = free_space_lvl;
19615a8e39fbSHong Zhang 
19625a8e39fbSHong Zhang   for (i=0; i<n; i++) {
19635a8e39fbSHong Zhang     nzi = 0;
19646cf997b0SBarry Smith     /* copy current row into linked list */
19655a8e39fbSHong Zhang     nnz  = ai[r[i]+1] - ai[r[i]];
1966e32f2f54SBarry 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);
19675a8e39fbSHong Zhang     cols = aj + ai[r[i]];
19685a8e39fbSHong Zhang     lnk[i] = -1; /* marker to indicate if diagonal exists */
19695a8e39fbSHong Zhang     ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
19705a8e39fbSHong Zhang     nzi += nlnk;
19716cf997b0SBarry Smith 
19726cf997b0SBarry Smith     /* make sure diagonal entry is included */
19735a8e39fbSHong Zhang     if (diagonal_fill && lnk[i] == -1) {
19746cf997b0SBarry Smith       fm = n;
19755a8e39fbSHong Zhang       while (lnk[fm] < i) fm = lnk[fm];
19765a8e39fbSHong Zhang       lnk[i]     = lnk[fm]; /* insert diagonal into linked list */
19775a8e39fbSHong Zhang       lnk[fm]    = i;
19785a8e39fbSHong Zhang       lnk_lvl[i] = 0;
19795a8e39fbSHong Zhang       nzi++; dcount++;
19806cf997b0SBarry Smith     }
19816cf997b0SBarry Smith 
19825a8e39fbSHong Zhang     /* add pivot rows into the active row */
19835a8e39fbSHong Zhang     nzbd = 0;
19845a8e39fbSHong Zhang     prow = lnk[n];
19855a8e39fbSHong Zhang     while (prow < i) {
19865a8e39fbSHong Zhang       nnz      = bdiag[prow];
19875a8e39fbSHong Zhang       cols     = bj_ptr[prow] + nnz + 1;
19885a8e39fbSHong Zhang       cols_lvl = bjlvl_ptr[prow] + nnz + 1;
19895a8e39fbSHong Zhang       nnz      = bi[prow+1] - bi[prow] - nnz - 1;
19905a8e39fbSHong Zhang       ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr);
19915a8e39fbSHong Zhang       nzi += nlnk;
19925a8e39fbSHong Zhang       prow = lnk[prow];
19935a8e39fbSHong Zhang       nzbd++;
199456beaf04SBarry Smith     }
19955a8e39fbSHong Zhang     bdiag[i] = nzbd;
19965a8e39fbSHong Zhang     bi[i+1]  = bi[i] + nzi;
1997ecf371e4SBarry Smith 
19985a8e39fbSHong Zhang     /* if free space is not available, make more free space */
19995a8e39fbSHong Zhang     if (current_space->local_remaining<nzi) {
20005a8e39fbSHong Zhang       nnz = nzi*(n - i); /* estimated and max additional space needed */
2001a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
2002a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(nnz,&current_space_lvl);CHKERRQ(ierr);
20035a8e39fbSHong Zhang       reallocs++;
20045a8e39fbSHong Zhang     }
2005ecf371e4SBarry Smith 
20065a8e39fbSHong Zhang     /* copy data into free_space and free_space_lvl, then initialize lnk */
20075a8e39fbSHong Zhang     ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
20085a8e39fbSHong Zhang     bj_ptr[i]    = current_space->array;
20095a8e39fbSHong Zhang     bjlvl_ptr[i] = current_space_lvl->array;
20105a8e39fbSHong Zhang 
20115a8e39fbSHong Zhang     /* make sure the active row i has diagonal entry */
201265e19b50SBarry 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);
20135a8e39fbSHong Zhang 
20145a8e39fbSHong Zhang     current_space->array           += nzi;
20155a8e39fbSHong Zhang     current_space->local_used      += nzi;
20165a8e39fbSHong Zhang     current_space->local_remaining -= nzi;
20175a8e39fbSHong Zhang     current_space_lvl->array           += nzi;
20185a8e39fbSHong Zhang     current_space_lvl->local_used      += nzi;
20195a8e39fbSHong Zhang     current_space_lvl->local_remaining -= nzi;
20209e25ed09SBarry Smith   }
20215a8e39fbSHong Zhang 
2022898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
2023898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
20245a8e39fbSHong Zhang 
20255a8e39fbSHong Zhang   /* destroy list of free space and other temporary arrays */
20265a8e39fbSHong Zhang   ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
20276516239fSHong Zhang   ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); /* copy free_space -> bj */
20285a8e39fbSHong Zhang   ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
2029a1a86e44SBarry Smith   ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
20300e83c824SBarry Smith   ierr = PetscFree2(bj_ptr,bjlvl_ptr);CHKERRQ(ierr);
20319e25ed09SBarry Smith 
20326cf91177SBarry Smith #if defined(PETSC_USE_INFO)
2033f64065bbSBarry Smith   {
20345a8e39fbSHong Zhang     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
2035ae15b995SBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr);
2036ae15b995SBarry Smith     ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
2037ae15b995SBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr);
2038ae15b995SBarry Smith     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
2039335d9088SBarry Smith     if (diagonal_fill) {
2040ae15b995SBarry Smith       ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr);
2041335d9088SBarry Smith     }
2042f64065bbSBarry Smith   }
204363ba0a88SBarry Smith #endif
2044bbb6d6a8SBarry Smith 
20459e25ed09SBarry Smith   /* put together the new matrix */
204607dbf95fSBarry Smith   ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
2047719d5645SBarry Smith   ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr);
2048719d5645SBarry Smith   b = (Mat_SeqAIJ*)(fact)->data;
2049e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
2050e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
20517c922b88SBarry Smith   b->singlemalloc = PETSC_FALSE;
20520e83c824SBarry Smith   ierr = PetscMalloc(bi[n]*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);
20535a8e39fbSHong Zhang   b->j          = bj;
20545a8e39fbSHong Zhang   b->i          = bi;
20555a8e39fbSHong Zhang   for (i=0; i<n; i++) bdiag[i] += bi[i];
20565a8e39fbSHong Zhang   b->diag       = bdiag;
2057416022c9SBarry Smith   b->ilen       = 0;
2058416022c9SBarry Smith   b->imax       = 0;
2059416022c9SBarry Smith   b->row        = isrow;
2060416022c9SBarry Smith   b->col        = iscol;
2061c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
2062c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
206382bf6240SBarry Smith   b->icol       = isicol;
206487828ca2SBarry Smith   ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
2065416022c9SBarry Smith   /* In b structure:  Free imax, ilen, old a, old j.
20665a8e39fbSHong Zhang      Allocate bdiag, solve_work, new a, new j */
2067719d5645SBarry Smith   ierr = PetscLogObjectMemory(fact,(bi[n]-n) * (sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
20685a8e39fbSHong Zhang   b->maxnz             = b->nz = bi[n] ;
2069719d5645SBarry Smith   (fact)->info.factor_mallocs    = reallocs;
2070719d5645SBarry Smith   (fact)->info.fill_ratio_given  = f;
2071719d5645SBarry Smith   (fact)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]);
2072ad04f41aSHong Zhang   (fact)->ops->lufactornumeric =  MatLUFactorNumeric_SeqAIJ_inplace;
2073f2344125SBarry Smith   if (a->inode.size) {
2074f2344125SBarry Smith     (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode_inplace;
2075d3ac4fa3SBarry Smith   }
20763a40ed3dSBarry Smith   PetscFunctionReturn(0);
20779e25ed09SBarry Smith }
2078d5d45c9bSBarry Smith 
2079a6175056SHong Zhang #undef __FUNCT__
208035233d99SShri Abhyankar #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ"
208135233d99SShri Abhyankar PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info)
20825f44c854SHong Zhang {
20835f44c854SHong Zhang   Mat            C = B;
20845f44c854SHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
20855f44c854SHong Zhang   Mat_SeqSBAIJ   *b=(Mat_SeqSBAIJ*)C->data;
20865f44c854SHong Zhang   IS             ip=b->row,iip = b->icol;
20875f44c854SHong Zhang   PetscErrorCode ierr;
20885f44c854SHong Zhang   const PetscInt *rip,*riip;
20895f44c854SHong Zhang   PetscInt       i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bdiag=b->diag,*bjtmp;
20905f44c854SHong Zhang   PetscInt       *ai=a->i,*aj=a->j;
20915f44c854SHong Zhang   PetscInt       k,jmin,jmax,*c2r,*il,col,nexti,ili,nz;
20925f44c854SHong Zhang   MatScalar      *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi;
2093ace3abfcSBarry Smith   PetscBool      perm_identity;
2094d90ac83dSHong Zhang   FactorShiftCtx sctx;
209598a93161SHong Zhang   PetscReal      rs;
209698a93161SHong Zhang   MatScalar      d,*v;
209798a93161SHong Zhang 
20985f44c854SHong Zhang   PetscFunctionBegin;
209998a93161SHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
2100d90ac83dSHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
210198a93161SHong Zhang 
2102f4db908eSBarry Smith   if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
210398a93161SHong Zhang     sctx.shift_top = info->zeropivot;
210498a93161SHong Zhang     for (i=0; i<mbs; i++) {
210598a93161SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
210698a93161SHong Zhang       d  = (aa)[a->diag[i]];
210798a93161SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
210898a93161SHong Zhang       v  = aa+ai[i];
210998a93161SHong Zhang       nz = ai[i+1] - ai[i];
211098a93161SHong Zhang       for (j=0; j<nz; j++)
211198a93161SHong Zhang 	rs += PetscAbsScalar(v[j]);
211298a93161SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
211398a93161SHong Zhang     }
211498a93161SHong Zhang     sctx.shift_top   *= 1.1;
211598a93161SHong Zhang     sctx.nshift_max   = 5;
211698a93161SHong Zhang     sctx.shift_lo     = 0.;
211798a93161SHong Zhang     sctx.shift_hi     = 1.;
211898a93161SHong Zhang   }
21195f44c854SHong Zhang 
21205f44c854SHong Zhang   ierr  = ISGetIndices(ip,&rip);CHKERRQ(ierr);
21215f44c854SHong Zhang   ierr  = ISGetIndices(iip,&riip);CHKERRQ(ierr);
21225f44c854SHong Zhang 
21235f44c854SHong Zhang   /* allocate working arrays
21245f44c854SHong Zhang      c2r: linked list, keep track of pivot rows for a given column. c2r[col]: head of the list for a given col
21255f44c854SHong 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
21265f44c854SHong Zhang   */
21270e83c824SBarry Smith   ierr = PetscMalloc3(mbs,MatScalar,&rtmp,mbs,PetscInt,&il,mbs,PetscInt,&c2r);CHKERRQ(ierr);
21285f44c854SHong Zhang 
212998a93161SHong Zhang   do {
213007b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
213198a93161SHong Zhang 
2132c5546cabSHong Zhang     for (i=0; i<mbs; i++) c2r[i] = mbs;
21335f44c854SHong Zhang     il[0] = 0;
21345f44c854SHong Zhang 
21355f44c854SHong Zhang     for (k = 0; k<mbs; k++){
21365f44c854SHong Zhang       /* zero rtmp */
21375f44c854SHong Zhang       nz = bi[k+1] - bi[k];
21385f44c854SHong Zhang       bjtmp = bj + bi[k];
21395f44c854SHong Zhang       for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
21405f44c854SHong Zhang 
21415f44c854SHong Zhang       /* load in initial unfactored row */
21425f44c854SHong Zhang       bval = ba + bi[k];
21435f44c854SHong Zhang       jmin = ai[rip[k]]; jmax = ai[rip[k]+1];
21445f44c854SHong Zhang       for (j = jmin; j < jmax; j++){
21455f44c854SHong Zhang         col = riip[aj[j]];
21465f44c854SHong Zhang         if (col >= k){ /* only take upper triangular entry */
21475f44c854SHong Zhang           rtmp[col] = aa[j];
21485f44c854SHong Zhang           *bval++   = 0.0; /* for in-place factorization */
21495f44c854SHong Zhang         }
21505f44c854SHong Zhang       }
215198a93161SHong Zhang       /* shift the diagonal of the matrix: ZeropivotApply() */
215298a93161SHong Zhang       rtmp[k] += sctx.shift_amount;  /* shift the diagonal of the matrix */
21535f44c854SHong Zhang 
21545f44c854SHong Zhang       /* modify k-th row by adding in those rows i with U(i,k)!=0 */
21555f44c854SHong Zhang       dk = rtmp[k];
21565f44c854SHong Zhang       i  = c2r[k]; /* first row to be added to k_th row  */
21575f44c854SHong Zhang 
21585f44c854SHong Zhang       while (i < k){
21595f44c854SHong Zhang         nexti = c2r[i]; /* next row to be added to k_th row */
21605f44c854SHong Zhang 
21615f44c854SHong Zhang         /* compute multiplier, update diag(k) and U(i,k) */
21625f44c854SHong Zhang         ili   = il[i];  /* index of first nonzero element in U(i,k:bms-1) */
21635f44c854SHong Zhang         uikdi = - ba[ili]*ba[bdiag[i]];  /* diagonal(k) */
21645f44c854SHong Zhang         dk   += uikdi*ba[ili]; /* update diag[k] */
21655f44c854SHong Zhang         ba[ili] = uikdi; /* -U(i,k) */
21665f44c854SHong Zhang 
21675f44c854SHong Zhang         /* add multiple of row i to k-th row */
21685f44c854SHong Zhang         jmin = ili + 1; jmax = bi[i+1];
21695f44c854SHong Zhang         if (jmin < jmax){
21705f44c854SHong Zhang           for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j];
21715f44c854SHong Zhang           /* update il and c2r for row i */
21725f44c854SHong Zhang           il[i] = jmin;
21735f44c854SHong Zhang           j = bj[jmin]; c2r[i] = c2r[j]; c2r[j] = i;
21745f44c854SHong Zhang         }
21755f44c854SHong Zhang         i = nexti;
21765f44c854SHong Zhang       }
21775f44c854SHong Zhang 
21785f44c854SHong Zhang       /* copy data into U(k,:) */
217998a93161SHong Zhang       rs   = 0.0;
21805f44c854SHong Zhang       jmin = bi[k]; jmax = bi[k+1]-1;
21815f44c854SHong Zhang       if (jmin < jmax) {
21825f44c854SHong Zhang         for (j=jmin; j<jmax; j++){
218398a93161SHong Zhang           col = bj[j]; ba[j] = rtmp[col]; rs += PetscAbsScalar(ba[j]);
21845f44c854SHong Zhang         }
21855f44c854SHong Zhang         /* add the k-th row into il and c2r */
21865f44c854SHong Zhang         il[k] = jmin;
21875f44c854SHong Zhang         i = bj[jmin]; c2r[k] = c2r[i]; c2r[i] = k;
21885f44c854SHong Zhang       }
218998a93161SHong Zhang 
219098a93161SHong Zhang       /* MatPivotCheck() */
219198a93161SHong Zhang       sctx.rs  = rs;
219298a93161SHong Zhang       sctx.pv  = dk;
2193d0660788SBarry Smith       ierr = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr);
219407b50cabSHong Zhang       if(sctx.newshift) break;
219598a93161SHong Zhang       dk = sctx.pv;
219698a93161SHong Zhang 
219798a93161SHong Zhang       ba[bdiag[k]] = 1.0/dk; /* U(k,k) */
219898a93161SHong Zhang     }
219907b50cabSHong Zhang   } while (sctx.newshift);
22005f44c854SHong Zhang 
22010e83c824SBarry Smith   ierr = PetscFree3(rtmp,il,c2r);CHKERRQ(ierr);
22025f44c854SHong Zhang   ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr);
22035f44c854SHong Zhang   ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr);
22045f44c854SHong Zhang 
22055f44c854SHong Zhang   ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr);
22065f44c854SHong Zhang   if (perm_identity){
220735233d99SShri Abhyankar     B->ops->solve           = MatSolve_SeqSBAIJ_1_NaturalOrdering;
220835233d99SShri Abhyankar     B->ops->solvetranspose  = MatSolve_SeqSBAIJ_1_NaturalOrdering;
22092169352eSHong Zhang     B->ops->forwardsolve    = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering;
22102169352eSHong Zhang     B->ops->backwardsolve   = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering;
22115f44c854SHong Zhang   } else {
221235233d99SShri Abhyankar     B->ops->solve           = MatSolve_SeqSBAIJ_1;
221335233d99SShri Abhyankar     B->ops->solvetranspose  = MatSolve_SeqSBAIJ_1;
221435233d99SShri Abhyankar     B->ops->forwardsolve    = MatForwardSolve_SeqSBAIJ_1;
221535233d99SShri Abhyankar     B->ops->backwardsolve   = MatBackwardSolve_SeqSBAIJ_1;
22165f44c854SHong Zhang   }
22175f44c854SHong Zhang 
22185f44c854SHong Zhang   C->assembled    = PETSC_TRUE;
22195f44c854SHong Zhang   C->preallocated = PETSC_TRUE;
22205f44c854SHong Zhang   ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr);
222198a93161SHong Zhang 
222298a93161SHong Zhang   /* MatPivotView() */
222398a93161SHong Zhang   if (sctx.nshift){
2224f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
222598a93161SHong 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);
2226f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
222798a93161SHong Zhang       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
2228f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_INBLOCKS){
2229b2a402fcSHong Zhang       ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %G\n",sctx.nshift,info->shiftamount);CHKERRQ(ierr);
223098a93161SHong Zhang     }
223198a93161SHong Zhang   }
22325f44c854SHong Zhang   PetscFunctionReturn(0);
22335f44c854SHong Zhang }
22345f44c854SHong Zhang 
22355f44c854SHong Zhang #undef __FUNCT__
22360a3c351aSHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ_inplace"
22370a3c351aSHong Zhang PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ_inplace(Mat B,Mat A,const MatFactorInfo *info)
2238a6175056SHong Zhang {
2239719d5645SBarry Smith   Mat            C = B;
22400968510aSHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
22412ed133dbSHong Zhang   Mat_SeqSBAIJ   *b=(Mat_SeqSBAIJ*)C->data;
22429bfd6278SHong Zhang   IS             ip=b->row,iip = b->icol;
22432ed133dbSHong Zhang   PetscErrorCode ierr;
22445d0c19d7SBarry Smith   const PetscInt *rip,*riip;
2245c5546cabSHong Zhang   PetscInt       i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bcol,*bjtmp;
22462ed133dbSHong Zhang   PetscInt       *ai=a->i,*aj=a->j;
22472ed133dbSHong Zhang   PetscInt       k,jmin,jmax,*jl,*il,col,nexti,ili,nz;
2248622e2028SHong Zhang   MatScalar      *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi;
2249ace3abfcSBarry Smith   PetscBool      perm_identity;
22500e95ead3SHong Zhang   FactorShiftCtx sctx;
22510e95ead3SHong Zhang   PetscReal      rs;
22520e95ead3SHong Zhang   MatScalar      d,*v;
2253d5d45c9bSBarry Smith 
2254a6175056SHong Zhang   PetscFunctionBegin;
22550e95ead3SHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
22560e95ead3SHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
22570e95ead3SHong Zhang 
22580e95ead3SHong Zhang   if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
22590e95ead3SHong Zhang     sctx.shift_top = info->zeropivot;
22600e95ead3SHong Zhang     for (i=0; i<mbs; i++) {
22610e95ead3SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
22620e95ead3SHong Zhang       d  = (aa)[a->diag[i]];
22630e95ead3SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
22640e95ead3SHong Zhang       v  = aa+ai[i];
22650e95ead3SHong Zhang       nz = ai[i+1] - ai[i];
22660e95ead3SHong Zhang       for (j=0; j<nz; j++)
22670e95ead3SHong Zhang 	rs += PetscAbsScalar(v[j]);
22680e95ead3SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
22690e95ead3SHong Zhang     }
22700e95ead3SHong Zhang     sctx.shift_top   *= 1.1;
22710e95ead3SHong Zhang     sctx.nshift_max   = 5;
22720e95ead3SHong Zhang     sctx.shift_lo     = 0.;
22730e95ead3SHong Zhang     sctx.shift_hi     = 1.;
22740e95ead3SHong Zhang   }
2275ee45ca4aSHong Zhang 
22762ed133dbSHong Zhang   ierr  = ISGetIndices(ip,&rip);CHKERRQ(ierr);
22779bfd6278SHong Zhang   ierr  = ISGetIndices(iip,&riip);CHKERRQ(ierr);
22782ed133dbSHong Zhang 
22792ed133dbSHong Zhang   /* initialization */
22800e83c824SBarry Smith   ierr = PetscMalloc3(mbs,MatScalar,&rtmp,mbs,PetscInt,&il,mbs,PetscInt,&jl);CHKERRQ(ierr);
22810e95ead3SHong Zhang 
22822ed133dbSHong Zhang   do {
228307b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
22840e95ead3SHong Zhang 
2285c5546cabSHong Zhang     for (i=0; i<mbs; i++) jl[i] = mbs;
2286c5546cabSHong Zhang     il[0] = 0;
22872ed133dbSHong Zhang 
22882ed133dbSHong Zhang     for (k = 0; k<mbs; k++){
2289c5546cabSHong Zhang       /* zero rtmp */
2290c5546cabSHong Zhang       nz = bi[k+1] - bi[k];
2291c5546cabSHong Zhang       bjtmp = bj + bi[k];
2292c5546cabSHong Zhang       for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
2293c5546cabSHong Zhang 
2294c05c3958SHong Zhang       bval = ba + bi[k];
22952ed133dbSHong Zhang       /* initialize k-th row by the perm[k]-th row of A */
22962ed133dbSHong Zhang       jmin = ai[rip[k]]; jmax = ai[rip[k]+1];
22972ed133dbSHong Zhang       for (j = jmin; j < jmax; j++){
22989bfd6278SHong Zhang         col = riip[aj[j]];
22992ed133dbSHong Zhang         if (col >= k){ /* only take upper triangular entry */
23002ed133dbSHong Zhang           rtmp[col] = aa[j];
2301c05c3958SHong Zhang           *bval++  = 0.0; /* for in-place factorization */
23022ed133dbSHong Zhang         }
23032ed133dbSHong Zhang       }
230439e3d630SHong Zhang       /* shift the diagonal of the matrix */
2305540703acSHong Zhang       if (sctx.nshift) rtmp[k] += sctx.shift_amount;
23062ed133dbSHong Zhang 
23072ed133dbSHong Zhang       /* modify k-th row by adding in those rows i with U(i,k)!=0 */
23082ed133dbSHong Zhang       dk = rtmp[k];
23092ed133dbSHong Zhang       i = jl[k]; /* first row to be added to k_th row  */
23102ed133dbSHong Zhang 
23112ed133dbSHong Zhang       while (i < k){
23122ed133dbSHong Zhang         nexti = jl[i]; /* next row to be added to k_th row */
23132ed133dbSHong Zhang 
23142ed133dbSHong Zhang         /* compute multiplier, update diag(k) and U(i,k) */
23152ed133dbSHong Zhang         ili = il[i];  /* index of first nonzero element in U(i,k:bms-1) */
23162ed133dbSHong Zhang         uikdi = - ba[ili]*ba[bi[i]];  /* diagonal(k) */
23172ed133dbSHong Zhang         dk += uikdi*ba[ili];
23182ed133dbSHong Zhang         ba[ili] = uikdi; /* -U(i,k) */
23192ed133dbSHong Zhang 
23202ed133dbSHong Zhang         /* add multiple of row i to k-th row */
23212ed133dbSHong Zhang         jmin = ili + 1; jmax = bi[i+1];
23222ed133dbSHong Zhang         if (jmin < jmax){
23232ed133dbSHong Zhang           for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j];
23242ed133dbSHong Zhang           /* update il and jl for row i */
23252ed133dbSHong Zhang           il[i] = jmin;
23262ed133dbSHong Zhang           j = bj[jmin]; jl[i] = jl[j]; jl[j] = i;
23272ed133dbSHong Zhang         }
23282ed133dbSHong Zhang         i = nexti;
23292ed133dbSHong Zhang       }
23302ed133dbSHong Zhang 
23310697b6caSHong Zhang       /* shift the diagonals when zero pivot is detected */
23320697b6caSHong Zhang       /* compute rs=sum of abs(off-diagonal) */
23330697b6caSHong Zhang       rs   = 0.0;
23343c9e8598SHong Zhang       jmin = bi[k]+1;
23353c9e8598SHong Zhang       nz   = bi[k+1] - jmin;
23363c9e8598SHong Zhang       bcol = bj + jmin;
233704fbf559SBarry Smith       for (j=0; j<nz; j++) {
233804fbf559SBarry Smith         rs += PetscAbsScalar(rtmp[bcol[j]]);
23393c9e8598SHong Zhang       }
2340540703acSHong Zhang 
2341540703acSHong Zhang       sctx.rs = rs;
2342540703acSHong Zhang       sctx.pv = dk;
2343d0660788SBarry Smith       ierr = MatPivotCheck(A,info,&sctx,k);CHKERRQ(ierr);
234407b50cabSHong Zhang       if (sctx.newshift) break;
23450e95ead3SHong Zhang       dk = sctx.pv;
23463c9e8598SHong Zhang 
23473c9e8598SHong Zhang       /* copy data into U(k,:) */
234839e3d630SHong Zhang       ba[bi[k]] = 1.0/dk; /* U(k,k) */
234939e3d630SHong Zhang       jmin = bi[k]+1; jmax = bi[k+1];
235039e3d630SHong Zhang       if (jmin < jmax) {
235139e3d630SHong Zhang         for (j=jmin; j<jmax; j++){
2352c6ae9bfcSHong Zhang           col = bj[j]; ba[j] = rtmp[col];
23533c9e8598SHong Zhang         }
235439e3d630SHong Zhang         /* add the k-th row into il and jl */
23553c9e8598SHong Zhang         il[k] = jmin;
23563c9e8598SHong Zhang         i = bj[jmin]; jl[k] = jl[i]; jl[i] = k;
23573c9e8598SHong Zhang       }
23583c9e8598SHong Zhang     }
235907b50cabSHong Zhang   } while (sctx.newshift);
23600e95ead3SHong Zhang 
23610e83c824SBarry Smith   ierr = PetscFree3(rtmp,il,jl);CHKERRQ(ierr);
236239e3d630SHong Zhang   ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr);
23639bfd6278SHong Zhang   ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr);
2364db4efbfdSBarry Smith 
2365db4efbfdSBarry Smith   ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr);
2366db4efbfdSBarry Smith   if (perm_identity){
23670a3c351aSHong Zhang     B->ops->solve           = MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace;
23680a3c351aSHong Zhang     B->ops->solvetranspose  = MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace;
23690a3c351aSHong Zhang     B->ops->forwardsolve    = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace;
23700a3c351aSHong Zhang     B->ops->backwardsolve   = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace;
2371db4efbfdSBarry Smith   } else {
23720a3c351aSHong Zhang     B->ops->solve           = MatSolve_SeqSBAIJ_1_inplace;
23730a3c351aSHong Zhang     B->ops->solvetranspose  = MatSolve_SeqSBAIJ_1_inplace;
23740a3c351aSHong Zhang     B->ops->forwardsolve    = MatForwardSolve_SeqSBAIJ_1_inplace;
23750a3c351aSHong Zhang     B->ops->backwardsolve   = MatBackwardSolve_SeqSBAIJ_1_inplace;
2376db4efbfdSBarry Smith   }
2377db4efbfdSBarry Smith 
23783c9e8598SHong Zhang   C->assembled    = PETSC_TRUE;
23793c9e8598SHong Zhang   C->preallocated = PETSC_TRUE;
2380d0f46423SBarry Smith   ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr);
2381540703acSHong Zhang   if (sctx.nshift){
2382f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
23831e2582c4SBarry Smith       ierr = PetscInfo2(A,"number of shiftnz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
2384f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
23851e2582c4SBarry Smith       ierr = PetscInfo2(A,"number of shiftpd tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
23860697b6caSHong Zhang     }
23873c9e8598SHong Zhang   }
23883c9e8598SHong Zhang   PetscFunctionReturn(0);
23893c9e8598SHong Zhang }
2390a6175056SHong Zhang 
23918a76ed4fSHong Zhang /*
23928a76ed4fSHong Zhang    icc() under revised new data structure.
23938a76ed4fSHong Zhang    Factored arrays bj and ba are stored as
23948a76ed4fSHong Zhang      U(0,:),...,U(i,:),U(n-1,:)
23958a76ed4fSHong Zhang 
23968a76ed4fSHong Zhang    ui=fact->i is an array of size n+1, in which
23978a76ed4fSHong Zhang    ui+
23988a76ed4fSHong Zhang      ui[i]:  points to 1st entry of U(i,:),i=0,...,n-1
23998a76ed4fSHong Zhang      ui[n]:  points to U(n-1,n-1)+1
24008a76ed4fSHong Zhang 
24018a76ed4fSHong Zhang   udiag=fact->diag is an array of size n,in which
24028a76ed4fSHong Zhang      udiag[i]: points to diagonal of U(i,:), i=0,...,n-1
24038a76ed4fSHong Zhang 
24048a76ed4fSHong Zhang    U(i,:) contains udiag[i] as its last entry, i.e.,
24058a76ed4fSHong Zhang     U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i])
24068a76ed4fSHong Zhang */
24078a76ed4fSHong Zhang 
2408a6175056SHong Zhang #undef __FUNCT__
240935233d99SShri Abhyankar #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ"
241035233d99SShri Abhyankar PetscErrorCode MatICCFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
2411a6175056SHong Zhang {
24120968510aSHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
2413ed59401aSHong Zhang   Mat_SeqSBAIJ       *b;
2414dfbe8321SBarry Smith   PetscErrorCode     ierr;
2415ace3abfcSBarry Smith   PetscBool          perm_identity,missing;
24165f44c854SHong Zhang   PetscInt           reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag;
24175d0c19d7SBarry Smith   const PetscInt     *rip,*riip;
2418ed59401aSHong Zhang   PetscInt           jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow;
241958ebbce7SBarry Smith   PetscInt           nlnk,*lnk,*lnk_lvl=PETSC_NULL,d;
24205a8e39fbSHong Zhang   PetscInt           ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr;
2421ed59401aSHong Zhang   PetscReal          fill=info->fill,levels=info->levels;
2422a1a86e44SBarry Smith   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
2423a1a86e44SBarry Smith   PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL;
24247a48dd6fSHong Zhang   PetscBT            lnkbt;
2425b635d36bSHong Zhang   IS                 iperm;
2426a6175056SHong Zhang 
2427a6175056SHong Zhang   PetscFunctionBegin;
2428e32f2f54SBarry 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);
242958ebbce7SBarry Smith   ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr);
2430e32f2f54SBarry Smith   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d);
2431653a6975SHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
2432b635d36bSHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
24337a48dd6fSHong Zhang 
243439e3d630SHong Zhang   ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr);
24355f44c854SHong Zhang   ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr);
243639e3d630SHong Zhang   ui[0] = 0;
243739e3d630SHong Zhang 
24388a76ed4fSHong Zhang   /* ICC(0) without matrix ordering: simply rearrange column indices */
2439622e2028SHong Zhang   if (!levels && perm_identity) {
24405f44c854SHong Zhang     for (i=0; i<am; i++) {
2441c5546cabSHong Zhang       ncols    = ai[i+1] - a->diag[i];
2442c5546cabSHong Zhang       ui[i+1]  = ui[i] + ncols;
2443c5546cabSHong Zhang       udiag[i] = ui[i+1] - 1; /* points to the last entry of U(i,:) */
2444dc1e2a3fSHong Zhang     }
2445dc1e2a3fSHong Zhang     ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
2446dc1e2a3fSHong Zhang     cols = uj;
2447dc1e2a3fSHong Zhang     for (i=0; i<am; i++) {
24485f44c854SHong Zhang       aj    = a->j + a->diag[i] + 1; /* 1st entry of U(i,:) without diagonal */
2449dc1e2a3fSHong Zhang       ncols = ai[i+1] - a->diag[i] -1;
24505f44c854SHong Zhang       for (j=0; j<ncols; j++) *cols++ = aj[j];
24515f44c854SHong Zhang       *cols++ = i; /* diagoanl is located as the last entry of U(i,:) */
24525f44c854SHong Zhang     }
24535f44c854SHong Zhang   } else { /* case: levels>0 || (levels=0 && !perm_identity) */
24545f44c854SHong Zhang     ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
24555f44c854SHong Zhang     ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
24565f44c854SHong Zhang 
24575f44c854SHong Zhang     /* initialization */
24585f44c854SHong Zhang     ierr  = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr);
24595f44c854SHong Zhang 
24605f44c854SHong Zhang     /* jl: linked list for storing indices of the pivot rows
24615f44c854SHong Zhang        il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
24620e83c824SBarry Smith     ierr = PetscMalloc4(am,PetscInt*,&uj_ptr,am,PetscInt*,&uj_lvl_ptr,am,PetscInt,&jl,am,PetscInt,&il);CHKERRQ(ierr);
24635f44c854SHong Zhang     for (i=0; i<am; i++){
24645f44c854SHong Zhang       jl[i] = am; il[i] = 0;
24655f44c854SHong Zhang     }
24665f44c854SHong Zhang 
24675f44c854SHong Zhang     /* create and initialize a linked list for storing column indices of the active row k */
24685f44c854SHong Zhang     nlnk = am + 1;
24695f44c854SHong Zhang     ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
24705f44c854SHong Zhang 
247195b5ac22SHong Zhang     /* initial FreeSpace size is fill*(ai[am]+am)/2 */
247295b5ac22SHong Zhang     ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+am)/2),&free_space);CHKERRQ(ierr);
24735f44c854SHong Zhang     current_space = free_space;
247495b5ac22SHong Zhang     ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+am)/2),&free_space_lvl);CHKERRQ(ierr);
24755f44c854SHong Zhang     current_space_lvl = free_space_lvl;
24765f44c854SHong Zhang 
24775f44c854SHong Zhang     for (k=0; k<am; k++){  /* for each active row k */
24785f44c854SHong Zhang       /* initialize lnk by the column indices of row rip[k] of A */
24795f44c854SHong Zhang       nzk   = 0;
24805f44c854SHong Zhang       ncols = ai[rip[k]+1] - ai[rip[k]];
2481e32f2f54SBarry 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);
24825f44c854SHong Zhang       ncols_upper = 0;
24835f44c854SHong Zhang       for (j=0; j<ncols; j++){
24845f44c854SHong Zhang         i = *(aj + ai[rip[k]] + j); /* unpermuted column index */
24855f44c854SHong Zhang         if (riip[i] >= k){ /* only take upper triangular entry */
24865f44c854SHong Zhang           ajtmp[ncols_upper] = i;
24875f44c854SHong Zhang           ncols_upper++;
24885f44c854SHong Zhang         }
24895f44c854SHong Zhang       }
24905f44c854SHong Zhang       ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
24915f44c854SHong Zhang       nzk += nlnk;
24925f44c854SHong Zhang 
24935f44c854SHong Zhang       /* update lnk by computing fill-in for each pivot row to be merged in */
24945f44c854SHong Zhang       prow = jl[k]; /* 1st pivot row */
24955f44c854SHong Zhang 
24965f44c854SHong Zhang       while (prow < k){
24975f44c854SHong Zhang         nextprow = jl[prow];
24985f44c854SHong Zhang 
24995f44c854SHong Zhang         /* merge prow into k-th row */
25005f44c854SHong Zhang         jmin = il[prow] + 1;  /* index of the 2nd nzero entry in U(prow,k:am-1) */
25015f44c854SHong Zhang         jmax = ui[prow+1];
25025f44c854SHong Zhang         ncols = jmax-jmin;
25035f44c854SHong Zhang         i     = jmin - ui[prow];
25045f44c854SHong Zhang         cols  = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */
25055f44c854SHong Zhang         uj    = uj_lvl_ptr[prow] + i; /* levels of cols */
25065f44c854SHong Zhang         j     = *(uj - 1);
25075f44c854SHong Zhang         ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr);
25085f44c854SHong Zhang         nzk += nlnk;
25095f44c854SHong Zhang 
25105f44c854SHong Zhang         /* update il and jl for prow */
25115f44c854SHong Zhang         if (jmin < jmax){
25125f44c854SHong Zhang           il[prow] = jmin;
25135f44c854SHong Zhang           j = *cols; jl[prow] = jl[j]; jl[j] = prow;
25145f44c854SHong Zhang         }
25155f44c854SHong Zhang         prow = nextprow;
25165f44c854SHong Zhang       }
25175f44c854SHong Zhang 
25185f44c854SHong Zhang       /* if free space is not available, make more free space */
25195f44c854SHong Zhang       if (current_space->local_remaining<nzk) {
25205f44c854SHong Zhang         i  = am - k + 1; /* num of unfactored rows */
2521c6ae9bfcSHong Zhang         i *= PetscMin(nzk, i-1); /* i*nzk, i*(i-1): estimated and max additional space needed */
25225f44c854SHong Zhang         ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
25235f44c854SHong Zhang         ierr = PetscFreeSpaceGet(i,&current_space_lvl);CHKERRQ(ierr);
25245f44c854SHong Zhang         reallocs++;
25255f44c854SHong Zhang       }
25265f44c854SHong Zhang 
25275f44c854SHong Zhang       /* copy data into free_space and free_space_lvl, then initialize lnk */
2528e32f2f54SBarry Smith       if (nzk == 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k);
25295f44c854SHong Zhang       ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
25305f44c854SHong Zhang 
25315f44c854SHong Zhang       /* add the k-th row into il and jl */
25325f44c854SHong Zhang       if (nzk > 1){
25335f44c854SHong Zhang         i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
25345f44c854SHong Zhang         jl[k] = jl[i]; jl[i] = k;
25355f44c854SHong Zhang         il[k] = ui[k] + 1;
25365f44c854SHong Zhang       }
25375f44c854SHong Zhang       uj_ptr[k]     = current_space->array;
25385f44c854SHong Zhang       uj_lvl_ptr[k] = current_space_lvl->array;
25395f44c854SHong Zhang 
25405f44c854SHong Zhang       current_space->array           += nzk;
25415f44c854SHong Zhang       current_space->local_used      += nzk;
25425f44c854SHong Zhang       current_space->local_remaining -= nzk;
25435f44c854SHong Zhang 
25445f44c854SHong Zhang       current_space_lvl->array           += nzk;
25455f44c854SHong Zhang       current_space_lvl->local_used      += nzk;
25465f44c854SHong Zhang       current_space_lvl->local_remaining -= nzk;
25475f44c854SHong Zhang 
25485f44c854SHong Zhang       ui[k+1] = ui[k] + nzk;
25495f44c854SHong Zhang     }
25505f44c854SHong Zhang 
25515f44c854SHong Zhang     ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
25525f44c854SHong Zhang     ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
25530e83c824SBarry Smith     ierr = PetscFree4(uj_ptr,uj_lvl_ptr,jl,il);CHKERRQ(ierr);
25545f44c854SHong Zhang     ierr = PetscFree(ajtmp);CHKERRQ(ierr);
25555f44c854SHong Zhang 
25569263d837SHong Zhang     /* copy free_space into uj and free free_space; set ui, uj, udiag in new datastructure; */
25575f44c854SHong Zhang     ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
255835233d99SShri Abhyankar     ierr = PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag);CHKERRQ(ierr); /* store matrix factor  */
25595f44c854SHong Zhang     ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
25605f44c854SHong Zhang     ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
25615f44c854SHong Zhang 
25625f44c854SHong Zhang   } /* end of case: levels>0 || (levels=0 && !perm_identity) */
25635f44c854SHong Zhang 
25645f44c854SHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
25655f44c854SHong Zhang   b    = (Mat_SeqSBAIJ*)(fact)->data;
25665f44c854SHong Zhang   b->singlemalloc = PETSC_FALSE;
25675f44c854SHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr);
25685f44c854SHong Zhang   b->j    = uj;
25695f44c854SHong Zhang   b->i    = ui;
25705f44c854SHong Zhang   b->diag = udiag;
25717f53bb6cSHong Zhang   b->free_diag = PETSC_TRUE;
25725f44c854SHong Zhang   b->ilen = 0;
25735f44c854SHong Zhang   b->imax = 0;
25745f44c854SHong Zhang   b->row  = perm;
25755f44c854SHong Zhang   b->col  = perm;
25765f44c854SHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
25775f44c854SHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
25785f44c854SHong Zhang   b->icol = iperm;
25795f44c854SHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
25805f44c854SHong Zhang   ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
2581d595f711SHong Zhang   ierr = PetscLogObjectMemory(fact,ui[am]*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
25825f44c854SHong Zhang   b->maxnz   = b->nz = ui[am];
25835f44c854SHong Zhang   b->free_a  = PETSC_TRUE;
25845f44c854SHong Zhang   b->free_ij = PETSC_TRUE;
25855f44c854SHong Zhang 
2586f284f12aSHong Zhang   fact->info.factor_mallocs   = reallocs;
2587f284f12aSHong Zhang   fact->info.fill_ratio_given = fill;
25885f44c854SHong Zhang   if (ai[am] != 0) {
25896a69fef8SHong Zhang     /* nonzeros in lower triangular part of A (including diagonals) = (ai[am]+am)/2 */
259095b5ac22SHong Zhang     fact->info.fill_ratio_needed = ((PetscReal)2*ui[am])/(ai[am]+am);
25915f44c854SHong Zhang   } else {
2592f284f12aSHong Zhang     fact->info.fill_ratio_needed = 0.0;
25935f44c854SHong Zhang   }
25949263d837SHong Zhang #if defined(PETSC_USE_INFO)
25959263d837SHong Zhang     if (ai[am] != 0) {
25969263d837SHong Zhang       PetscReal af = fact->info.fill_ratio_needed;
25979263d837SHong Zhang       ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr);
25989263d837SHong Zhang       ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
25999263d837SHong Zhang       ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr);
26009263d837SHong Zhang     } else {
26019263d837SHong Zhang       ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr);
26029263d837SHong Zhang     }
26039263d837SHong Zhang #endif
260435233d99SShri Abhyankar   fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ;
26055f44c854SHong Zhang   PetscFunctionReturn(0);
26065f44c854SHong Zhang }
26075f44c854SHong Zhang 
26085f44c854SHong Zhang #undef __FUNCT__
2609ad04f41aSHong Zhang #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ_inplace"
2610ad04f41aSHong Zhang PetscErrorCode MatICCFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
26115f44c854SHong Zhang {
26125f44c854SHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
26135f44c854SHong Zhang   Mat_SeqSBAIJ       *b;
26145f44c854SHong Zhang   PetscErrorCode     ierr;
2615ace3abfcSBarry Smith   PetscBool          perm_identity,missing;
26165f44c854SHong Zhang   PetscInt           reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag;
26175f44c854SHong Zhang   const PetscInt     *rip,*riip;
26185f44c854SHong Zhang   PetscInt           jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow;
26195f44c854SHong Zhang   PetscInt           nlnk,*lnk,*lnk_lvl=PETSC_NULL,d;
26205f44c854SHong Zhang   PetscInt           ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr;
26215f44c854SHong Zhang   PetscReal          fill=info->fill,levels=info->levels;
26225f44c854SHong Zhang   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
26235f44c854SHong Zhang   PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL;
26245f44c854SHong Zhang   PetscBT            lnkbt;
26255f44c854SHong Zhang   IS                 iperm;
26265f44c854SHong Zhang 
26275f44c854SHong Zhang   PetscFunctionBegin;
2628e32f2f54SBarry 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);
26295f44c854SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr);
2630e32f2f54SBarry Smith   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d);
26315f44c854SHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
26325f44c854SHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
26335f44c854SHong Zhang 
26345f44c854SHong Zhang   ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr);
26355f44c854SHong Zhang   ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr);
26365f44c854SHong Zhang   ui[0] = 0;
26375f44c854SHong Zhang 
26385f44c854SHong Zhang   /* ICC(0) without matrix ordering: simply copies fill pattern */
26395f44c854SHong Zhang   if (!levels && perm_identity) {
26405f44c854SHong Zhang 
26415f44c854SHong Zhang     for (i=0; i<am; i++) {
26425f44c854SHong Zhang       ui[i+1]  = ui[i] + ai[i+1] - a->diag[i];
26435f44c854SHong Zhang       udiag[i] = ui[i];
2644ed59401aSHong Zhang     }
264539e3d630SHong Zhang     ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
264639e3d630SHong Zhang     cols = uj;
2647ed59401aSHong Zhang     for (i=0; i<am; i++) {
2648ed59401aSHong Zhang       aj    = a->j + a->diag[i];
264939e3d630SHong Zhang       ncols = ui[i+1] - ui[i];
265039e3d630SHong Zhang       for (j=0; j<ncols; j++) *cols++ = *aj++;
2651ed59401aSHong Zhang     }
265239e3d630SHong Zhang   } else { /* case: levels>0 || (levels=0 && !perm_identity) */
2653b635d36bSHong Zhang     ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
2654b635d36bSHong Zhang     ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
2655b635d36bSHong Zhang 
26567a48dd6fSHong Zhang     /* initialization */
26575a8e39fbSHong Zhang     ierr  = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr);
26587a48dd6fSHong Zhang 
26597a48dd6fSHong Zhang     /* jl: linked list for storing indices of the pivot rows
26607a48dd6fSHong Zhang        il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
26610e83c824SBarry Smith     ierr = PetscMalloc4(am,PetscInt*,&uj_ptr,am,PetscInt*,&uj_lvl_ptr,am,PetscInt,&jl,am,PetscInt,&il);CHKERRQ(ierr);
26627a48dd6fSHong Zhang     for (i=0; i<am; i++){
26637a48dd6fSHong Zhang       jl[i] = am; il[i] = 0;
26647a48dd6fSHong Zhang     }
26657a48dd6fSHong Zhang 
26667a48dd6fSHong Zhang     /* create and initialize a linked list for storing column indices of the active row k */
26677a48dd6fSHong Zhang     nlnk = am + 1;
26682ed133dbSHong Zhang     ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
26697a48dd6fSHong Zhang 
26707a48dd6fSHong Zhang     /* initial FreeSpace size is fill*(ai[am]+1) */
2671a1a86e44SBarry Smith     ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr);
26727a48dd6fSHong Zhang     current_space = free_space;
2673a1a86e44SBarry Smith     ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space_lvl);CHKERRQ(ierr);
26747a48dd6fSHong Zhang     current_space_lvl = free_space_lvl;
26757a48dd6fSHong Zhang 
26767a48dd6fSHong Zhang     for (k=0; k<am; k++){  /* for each active row k */
26777a48dd6fSHong Zhang       /* initialize lnk by the column indices of row rip[k] of A */
26787a48dd6fSHong Zhang       nzk   = 0;
2679622e2028SHong Zhang       ncols = ai[rip[k]+1] - ai[rip[k]];
2680e32f2f54SBarry 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);
2681622e2028SHong Zhang       ncols_upper = 0;
2682622e2028SHong Zhang       for (j=0; j<ncols; j++){
2683b635d36bSHong Zhang         i = *(aj + ai[rip[k]] + j); /* unpermuted column index */
2684b635d36bSHong Zhang         if (riip[i] >= k){ /* only take upper triangular entry */
26855a8e39fbSHong Zhang           ajtmp[ncols_upper] = i;
2686622e2028SHong Zhang           ncols_upper++;
2687622e2028SHong Zhang         }
2688622e2028SHong Zhang       }
2689b635d36bSHong Zhang       ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
26907a48dd6fSHong Zhang       nzk += nlnk;
26917a48dd6fSHong Zhang 
26927a48dd6fSHong Zhang       /* update lnk by computing fill-in for each pivot row to be merged in */
26937a48dd6fSHong Zhang       prow = jl[k]; /* 1st pivot row */
26947a48dd6fSHong Zhang 
26957a48dd6fSHong Zhang       while (prow < k){
26967a48dd6fSHong Zhang         nextprow = jl[prow];
26977a48dd6fSHong Zhang 
26987a48dd6fSHong Zhang         /* merge prow into k-th row */
26997a48dd6fSHong Zhang         jmin = il[prow] + 1;  /* index of the 2nd nzero entry in U(prow,k:am-1) */
27007a48dd6fSHong Zhang         jmax = ui[prow+1];
27017a48dd6fSHong Zhang         ncols = jmax-jmin;
2702ed59401aSHong Zhang         i     = jmin - ui[prow];
2703ed59401aSHong Zhang         cols  = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */
27045a8e39fbSHong Zhang         uj    = uj_lvl_ptr[prow] + i; /* levels of cols */
27055a8e39fbSHong Zhang         j     = *(uj - 1);
27065a8e39fbSHong Zhang         ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr);
27077a48dd6fSHong Zhang         nzk += nlnk;
27087a48dd6fSHong Zhang 
27097a48dd6fSHong Zhang         /* update il and jl for prow */
27107a48dd6fSHong Zhang         if (jmin < jmax){
27117a48dd6fSHong Zhang           il[prow] = jmin;
27127a48dd6fSHong Zhang           j = *cols; jl[prow] = jl[j]; jl[j] = prow;
27137a48dd6fSHong Zhang         }
27147a48dd6fSHong Zhang         prow = nextprow;
27157a48dd6fSHong Zhang       }
27167a48dd6fSHong Zhang 
27177a48dd6fSHong Zhang       /* if free space is not available, make more free space */
27187a48dd6fSHong Zhang       if (current_space->local_remaining<nzk) {
27197a48dd6fSHong Zhang         i = am - k + 1; /* num of unfactored rows */
2720fd3f55cbSMatthew Knepley         i *= PetscMin(nzk, (i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */
2721a1a86e44SBarry Smith         ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
2722a1a86e44SBarry Smith         ierr = PetscFreeSpaceGet(i,&current_space_lvl);CHKERRQ(ierr);
27237a48dd6fSHong Zhang         reallocs++;
27247a48dd6fSHong Zhang       }
27257a48dd6fSHong Zhang 
27267a48dd6fSHong Zhang       /* copy data into free_space and free_space_lvl, then initialize lnk */
272765e19b50SBarry Smith       if (!nzk) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k);
27282ed133dbSHong Zhang       ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
27297a48dd6fSHong Zhang 
27307a48dd6fSHong Zhang       /* add the k-th row into il and jl */
273139e3d630SHong Zhang       if (nzk > 1){
27327a48dd6fSHong Zhang         i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
27337a48dd6fSHong Zhang         jl[k] = jl[i]; jl[i] = k;
27347a48dd6fSHong Zhang         il[k] = ui[k] + 1;
27357a48dd6fSHong Zhang       }
27367a48dd6fSHong Zhang       uj_ptr[k]     = current_space->array;
27377a48dd6fSHong Zhang       uj_lvl_ptr[k] = current_space_lvl->array;
27387a48dd6fSHong Zhang 
27397a48dd6fSHong Zhang       current_space->array           += nzk;
27407a48dd6fSHong Zhang       current_space->local_used      += nzk;
27417a48dd6fSHong Zhang       current_space->local_remaining -= nzk;
27427a48dd6fSHong Zhang 
27437a48dd6fSHong Zhang       current_space_lvl->array           += nzk;
27447a48dd6fSHong Zhang       current_space_lvl->local_used      += nzk;
27457a48dd6fSHong Zhang       current_space_lvl->local_remaining -= nzk;
27467a48dd6fSHong Zhang 
27477a48dd6fSHong Zhang       ui[k+1] = ui[k] + nzk;
27487a48dd6fSHong Zhang     }
27497a48dd6fSHong Zhang 
27506cf91177SBarry Smith #if defined(PETSC_USE_INFO)
27517a48dd6fSHong Zhang     if (ai[am] != 0) {
275239e3d630SHong Zhang       PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]);
2753ae15b995SBarry Smith       ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr);
2754ae15b995SBarry Smith       ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
2755ae15b995SBarry Smith       ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr);
27567a48dd6fSHong Zhang     } else {
2757ae15b995SBarry Smith       ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr);
27587a48dd6fSHong Zhang     }
275963ba0a88SBarry Smith #endif
27607a48dd6fSHong Zhang 
27617a48dd6fSHong Zhang     ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
2762b635d36bSHong Zhang     ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
27630e83c824SBarry Smith     ierr = PetscFree4(uj_ptr,uj_lvl_ptr,jl,il);CHKERRQ(ierr);
27645a8e39fbSHong Zhang     ierr = PetscFree(ajtmp);CHKERRQ(ierr);
27657a48dd6fSHong Zhang 
27667a48dd6fSHong Zhang     /* destroy list of free space and other temporary array(s) */
27677a48dd6fSHong Zhang     ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
2768a1a86e44SBarry Smith     ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr);
27692ed133dbSHong Zhang     ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
2770a1a86e44SBarry Smith     ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
27717a48dd6fSHong Zhang 
277239e3d630SHong Zhang   } /* end of case: levels>0 || (levels=0 && !perm_identity) */
277339e3d630SHong Zhang 
27747a48dd6fSHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
27757a48dd6fSHong Zhang 
2776f284f12aSHong Zhang   b    = (Mat_SeqSBAIJ*)fact->data;
27777a48dd6fSHong Zhang   b->singlemalloc = PETSC_FALSE;
27787a48dd6fSHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr);
27797a48dd6fSHong Zhang   b->j    = uj;
27807a48dd6fSHong Zhang   b->i    = ui;
27815f44c854SHong Zhang   b->diag = udiag;
27827f53bb6cSHong Zhang   b->free_diag = PETSC_TRUE;
27837a48dd6fSHong Zhang   b->ilen = 0;
27847a48dd6fSHong Zhang   b->imax = 0;
27857a48dd6fSHong Zhang   b->row  = perm;
2786b635d36bSHong Zhang   b->col  = perm;
2787b635d36bSHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
2788b635d36bSHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
2789b635d36bSHong Zhang   b->icol = iperm;
27907a48dd6fSHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
27917a48dd6fSHong Zhang   ierr    = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
2792f284f12aSHong Zhang   ierr = PetscLogObjectMemory(fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
27937a48dd6fSHong Zhang   b->maxnz   = b->nz = ui[am];
2794e6b907acSBarry Smith   b->free_a  = PETSC_TRUE;
2795e6b907acSBarry Smith   b->free_ij = PETSC_TRUE;
27967a48dd6fSHong Zhang 
2797f284f12aSHong Zhang   fact->info.factor_mallocs    = reallocs;
2798f284f12aSHong Zhang   fact->info.fill_ratio_given  = fill;
27997a48dd6fSHong Zhang   if (ai[am] != 0) {
2800f284f12aSHong Zhang     fact->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]);
28017a48dd6fSHong Zhang   } else {
2802f284f12aSHong Zhang     fact->info.fill_ratio_needed = 0.0;
28037a48dd6fSHong Zhang   }
28040a3c351aSHong Zhang   fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_inplace;
2805a6175056SHong Zhang   PetscFunctionReturn(0);
2806a6175056SHong Zhang }
2807d5d45c9bSBarry Smith 
2808177d4faaSHong Zhang #undef __FUNCT__
2809177d4faaSHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_SeqAIJ"
281035233d99SShri Abhyankar PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
28110be760fbSHong Zhang {
28120be760fbSHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
28130be760fbSHong Zhang   Mat_SeqSBAIJ       *b;
28140be760fbSHong Zhang   PetscErrorCode     ierr;
2815ace3abfcSBarry Smith   PetscBool          perm_identity;
28160be760fbSHong Zhang   PetscReal          fill = info->fill;
28170be760fbSHong Zhang   const PetscInt     *rip,*riip;
28180be760fbSHong Zhang   PetscInt           i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow;
28190be760fbSHong Zhang   PetscInt           *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow;
28200be760fbSHong Zhang   PetscInt           nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr,*udiag;
28210be760fbSHong Zhang   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
28220be760fbSHong Zhang   PetscBT            lnkbt;
28230be760fbSHong Zhang   IS                 iperm;
28240be760fbSHong Zhang 
28250be760fbSHong Zhang   PetscFunctionBegin;
2826e32f2f54SBarry 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);
28270be760fbSHong Zhang   /* check whether perm is the identity mapping */
28280be760fbSHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
28290be760fbSHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
28300be760fbSHong Zhang   ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
28310be760fbSHong Zhang   ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
28320be760fbSHong Zhang 
28330be760fbSHong Zhang   /* initialization */
28340be760fbSHong Zhang   ierr  = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr);
28350be760fbSHong Zhang   ierr  = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr);
28360be760fbSHong Zhang   ui[0] = 0;
28370be760fbSHong Zhang 
28380be760fbSHong Zhang   /* jl: linked list for storing indices of the pivot rows
28390be760fbSHong Zhang      il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
28400be760fbSHong Zhang   ierr = PetscMalloc4(am,PetscInt*,&ui_ptr,am,PetscInt,&jl,am,PetscInt,&il,am,PetscInt,&cols);CHKERRQ(ierr);
28410be760fbSHong Zhang   for (i=0; i<am; i++){
28420be760fbSHong Zhang     jl[i] = am; il[i] = 0;
28430be760fbSHong Zhang   }
28440be760fbSHong Zhang 
28450be760fbSHong Zhang   /* create and initialize a linked list for storing column indices of the active row k */
28460be760fbSHong Zhang   nlnk = am + 1;
28470be760fbSHong Zhang   ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
28480be760fbSHong Zhang 
284995b5ac22SHong Zhang   /* initial FreeSpace size is fill*(ai[am]+am)/2 */
285095b5ac22SHong Zhang   ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+am)/2),&free_space);CHKERRQ(ierr);
28510be760fbSHong Zhang   current_space = free_space;
28520be760fbSHong Zhang 
28530be760fbSHong Zhang   for (k=0; k<am; k++){  /* for each active row k */
28540be760fbSHong Zhang     /* initialize lnk by the column indices of row rip[k] of A */
28550be760fbSHong Zhang     nzk   = 0;
28560be760fbSHong Zhang     ncols = ai[rip[k]+1] - ai[rip[k]];
2857e32f2f54SBarry 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);
28580be760fbSHong Zhang     ncols_upper = 0;
28590be760fbSHong Zhang     for (j=0; j<ncols; j++){
28600be760fbSHong Zhang       i = riip[*(aj + ai[rip[k]] + j)];
28610be760fbSHong Zhang       if (i >= k){ /* only take upper triangular entry */
28620be760fbSHong Zhang         cols[ncols_upper] = i;
28630be760fbSHong Zhang         ncols_upper++;
28640be760fbSHong Zhang       }
28650be760fbSHong Zhang     }
28660be760fbSHong Zhang     ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
28670be760fbSHong Zhang     nzk += nlnk;
28680be760fbSHong Zhang 
28690be760fbSHong Zhang     /* update lnk by computing fill-in for each pivot row to be merged in */
28700be760fbSHong Zhang     prow = jl[k]; /* 1st pivot row */
28710be760fbSHong Zhang 
28720be760fbSHong Zhang     while (prow < k){
28730be760fbSHong Zhang       nextprow = jl[prow];
28740be760fbSHong Zhang       /* merge prow into k-th row */
28750be760fbSHong Zhang       jmin = il[prow] + 1;  /* index of the 2nd nzero entry in U(prow,k:am-1) */
28760be760fbSHong Zhang       jmax = ui[prow+1];
28770be760fbSHong Zhang       ncols = jmax-jmin;
28780be760fbSHong Zhang       uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */
28790be760fbSHong Zhang       ierr = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
28800be760fbSHong Zhang       nzk += nlnk;
28810be760fbSHong Zhang 
28820be760fbSHong Zhang       /* update il and jl for prow */
28830be760fbSHong Zhang       if (jmin < jmax){
28840be760fbSHong Zhang         il[prow] = jmin;
28850be760fbSHong Zhang         j = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow;
28860be760fbSHong Zhang       }
28870be760fbSHong Zhang       prow = nextprow;
28880be760fbSHong Zhang     }
28890be760fbSHong Zhang 
28900be760fbSHong Zhang     /* if free space is not available, make more free space */
28910be760fbSHong Zhang     if (current_space->local_remaining<nzk) {
28920be760fbSHong Zhang       i  = am - k + 1; /* num of unfactored rows */
2893c6ae9bfcSHong Zhang       i *= PetscMin(nzk,i-1); /* i*nzk, i*(i-1): estimated and max additional space needed */
28940be760fbSHong Zhang       ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
28950be760fbSHong Zhang       reallocs++;
28960be760fbSHong Zhang     }
28970be760fbSHong Zhang 
28980be760fbSHong Zhang     /* copy data into free space, then initialize lnk */
28990be760fbSHong Zhang     ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
29000be760fbSHong Zhang 
29010be760fbSHong Zhang     /* add the k-th row into il and jl */
29027b056e98SHong Zhang     if (nzk > 1){
29030be760fbSHong Zhang       i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
29040be760fbSHong Zhang       jl[k] = jl[i]; jl[i] = k;
29050be760fbSHong Zhang       il[k] = ui[k] + 1;
29060be760fbSHong Zhang     }
29070be760fbSHong Zhang     ui_ptr[k] = current_space->array;
29080be760fbSHong Zhang     current_space->array           += nzk;
29090be760fbSHong Zhang     current_space->local_used      += nzk;
29100be760fbSHong Zhang     current_space->local_remaining -= nzk;
29110be760fbSHong Zhang 
29120be760fbSHong Zhang     ui[k+1] = ui[k] + nzk;
29130be760fbSHong Zhang   }
29140be760fbSHong Zhang 
29150be760fbSHong Zhang   ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
29160be760fbSHong Zhang   ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
29170be760fbSHong Zhang   ierr = PetscFree4(ui_ptr,jl,il,cols);CHKERRQ(ierr);
29180be760fbSHong Zhang 
29199263d837SHong Zhang   /* copy free_space into uj and free free_space; set ui, uj, udiag in new datastructure; */
29200be760fbSHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
292135233d99SShri Abhyankar   ierr = PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag);CHKERRQ(ierr); /* store matrix factor */
29220be760fbSHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
29230be760fbSHong Zhang 
29240be760fbSHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
29250be760fbSHong Zhang 
2926f284f12aSHong Zhang   b = (Mat_SeqSBAIJ*)fact->data;
29270be760fbSHong Zhang   b->singlemalloc = PETSC_FALSE;
29280be760fbSHong Zhang   b->free_a       = PETSC_TRUE;
29290be760fbSHong Zhang   b->free_ij      = PETSC_TRUE;
29300be760fbSHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr);
29310be760fbSHong Zhang   b->j    = uj;
29320be760fbSHong Zhang   b->i    = ui;
29330be760fbSHong Zhang   b->diag = udiag;
29340be760fbSHong Zhang   b->free_diag = PETSC_TRUE;
29350be760fbSHong Zhang   b->ilen = 0;
29360be760fbSHong Zhang   b->imax = 0;
29370be760fbSHong Zhang   b->row  = perm;
29380be760fbSHong Zhang   b->col  = perm;
29390be760fbSHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
29400be760fbSHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
29410be760fbSHong Zhang   b->icol = iperm;
29420be760fbSHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
29430be760fbSHong Zhang   ierr    = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
29440860e08cSHong Zhang   ierr    = PetscLogObjectMemory(fact,ui[am]*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
29450be760fbSHong Zhang   b->maxnz = b->nz = ui[am];
29460be760fbSHong Zhang 
2947f284f12aSHong Zhang   fact->info.factor_mallocs    = reallocs;
2948f284f12aSHong Zhang   fact->info.fill_ratio_given  = fill;
29490be760fbSHong Zhang   if (ai[am] != 0) {
29506a69fef8SHong Zhang     /* nonzeros in lower triangular part of A (including diagonals) = (ai[am]+am)/2 */
295195b5ac22SHong Zhang     fact->info.fill_ratio_needed = ((PetscReal)2*ui[am])/(ai[am]+am);
29520be760fbSHong Zhang   } else {
2953f284f12aSHong Zhang     fact->info.fill_ratio_needed = 0.0;
29540be760fbSHong Zhang   }
29559263d837SHong Zhang #if defined(PETSC_USE_INFO)
29569263d837SHong Zhang   if (ai[am] != 0) {
29579263d837SHong Zhang     PetscReal af = fact->info.fill_ratio_needed;
29589263d837SHong Zhang     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr);
29599263d837SHong Zhang     ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
29609263d837SHong Zhang     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr);
29619263d837SHong Zhang   } else {
29629263d837SHong Zhang      ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr);
29639263d837SHong Zhang   }
29649263d837SHong Zhang #endif
296535233d99SShri Abhyankar   fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ;
29660be760fbSHong Zhang   PetscFunctionReturn(0);
29670be760fbSHong Zhang }
29680be760fbSHong Zhang 
2969f76d2b81SHong Zhang #undef __FUNCT__
2970ad04f41aSHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_SeqAIJ_inplace"
2971ad04f41aSHong Zhang PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
2972f76d2b81SHong Zhang {
2973f76d2b81SHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
297410c27e3dSHong Zhang   Mat_SeqSBAIJ       *b;
2975dfbe8321SBarry Smith   PetscErrorCode     ierr;
2976ace3abfcSBarry Smith   PetscBool          perm_identity;
297710c27e3dSHong Zhang   PetscReal          fill = info->fill;
29785d0c19d7SBarry Smith   const PetscInt     *rip,*riip;
29795d0c19d7SBarry Smith   PetscInt           i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow;
298010c27e3dSHong Zhang   PetscInt           *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow;
29812ed133dbSHong Zhang   PetscInt           nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr;
2982a1a86e44SBarry Smith   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
298310c27e3dSHong Zhang   PetscBT            lnkbt;
29842ed133dbSHong Zhang   IS                 iperm;
2985f76d2b81SHong Zhang 
2986f76d2b81SHong Zhang   PetscFunctionBegin;
2987e32f2f54SBarry 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);
29882ed133dbSHong Zhang   /* check whether perm is the identity mapping */
2989f76d2b81SHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
29902ed133dbSHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
29912ed133dbSHong Zhang   ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
29929bfd6278SHong Zhang   ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
299310c27e3dSHong Zhang 
299410c27e3dSHong Zhang   /* initialization */
29951393bc97SHong Zhang   ierr  = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr);
299610c27e3dSHong Zhang   ui[0] = 0;
299710c27e3dSHong Zhang 
299810c27e3dSHong Zhang   /* jl: linked list for storing indices of the pivot rows
29991393bc97SHong Zhang      il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
30000e83c824SBarry Smith   ierr = PetscMalloc4(am,PetscInt*,&ui_ptr,am,PetscInt,&jl,am,PetscInt,&il,am,PetscInt,&cols);CHKERRQ(ierr);
30011393bc97SHong Zhang   for (i=0; i<am; i++){
30021393bc97SHong Zhang     jl[i] = am; il[i] = 0;
3003f76d2b81SHong Zhang   }
300410c27e3dSHong Zhang 
300510c27e3dSHong Zhang   /* create and initialize a linked list for storing column indices of the active row k */
30061393bc97SHong Zhang   nlnk = am + 1;
30071393bc97SHong Zhang   ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
300810c27e3dSHong Zhang 
30091393bc97SHong Zhang   /* initial FreeSpace size is fill*(ai[am]+1) */
3010a1a86e44SBarry Smith   ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr);
301110c27e3dSHong Zhang   current_space = free_space;
301210c27e3dSHong Zhang 
30131393bc97SHong Zhang   for (k=0; k<am; k++){  /* for each active row k */
301410c27e3dSHong Zhang     /* initialize lnk by the column indices of row rip[k] of A */
301510c27e3dSHong Zhang     nzk   = 0;
30162ed133dbSHong Zhang     ncols = ai[rip[k]+1] - ai[rip[k]];
3017e32f2f54SBarry 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);
30182ed133dbSHong Zhang     ncols_upper = 0;
30192ed133dbSHong Zhang     for (j=0; j<ncols; j++){
30209bfd6278SHong Zhang       i = riip[*(aj + ai[rip[k]] + j)];
30212ed133dbSHong Zhang       if (i >= k){ /* only take upper triangular entry */
30222ed133dbSHong Zhang         cols[ncols_upper] = i;
30232ed133dbSHong Zhang         ncols_upper++;
30242ed133dbSHong Zhang       }
30252ed133dbSHong Zhang     }
30261393bc97SHong Zhang     ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
302710c27e3dSHong Zhang     nzk += nlnk;
302810c27e3dSHong Zhang 
302910c27e3dSHong Zhang     /* update lnk by computing fill-in for each pivot row to be merged in */
303010c27e3dSHong Zhang     prow = jl[k]; /* 1st pivot row */
303110c27e3dSHong Zhang 
303210c27e3dSHong Zhang     while (prow < k){
303310c27e3dSHong Zhang       nextprow = jl[prow];
303410c27e3dSHong Zhang       /* merge prow into k-th row */
30351393bc97SHong Zhang       jmin = il[prow] + 1;  /* index of the 2nd nzero entry in U(prow,k:am-1) */
303610c27e3dSHong Zhang       jmax = ui[prow+1];
303710c27e3dSHong Zhang       ncols = jmax-jmin;
30381393bc97SHong Zhang       uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */
30395a8e39fbSHong Zhang       ierr = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
304010c27e3dSHong Zhang       nzk += nlnk;
304110c27e3dSHong Zhang 
304210c27e3dSHong Zhang       /* update il and jl for prow */
304310c27e3dSHong Zhang       if (jmin < jmax){
304410c27e3dSHong Zhang         il[prow] = jmin;
30452ed133dbSHong Zhang         j = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow;
304610c27e3dSHong Zhang       }
304710c27e3dSHong Zhang       prow = nextprow;
304810c27e3dSHong Zhang     }
304910c27e3dSHong Zhang 
305010c27e3dSHong Zhang     /* if free space is not available, make more free space */
305110c27e3dSHong Zhang     if (current_space->local_remaining<nzk) {
30521393bc97SHong Zhang       i = am - k + 1; /* num of unfactored rows */
305310c27e3dSHong Zhang       i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */
3054a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
305510c27e3dSHong Zhang       reallocs++;
305610c27e3dSHong Zhang     }
305710c27e3dSHong Zhang 
305810c27e3dSHong Zhang     /* copy data into free space, then initialize lnk */
30591393bc97SHong Zhang     ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
306010c27e3dSHong Zhang 
306110c27e3dSHong Zhang     /* add the k-th row into il and jl */
306210c27e3dSHong Zhang     if (nzk-1 > 0){
30631393bc97SHong Zhang       i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
306410c27e3dSHong Zhang       jl[k] = jl[i]; jl[i] = k;
306510c27e3dSHong Zhang       il[k] = ui[k] + 1;
306610c27e3dSHong Zhang     }
30672ed133dbSHong Zhang     ui_ptr[k] = current_space->array;
306810c27e3dSHong Zhang     current_space->array           += nzk;
306910c27e3dSHong Zhang     current_space->local_used      += nzk;
307010c27e3dSHong Zhang     current_space->local_remaining -= nzk;
307110c27e3dSHong Zhang 
307210c27e3dSHong Zhang     ui[k+1] = ui[k] + nzk;
307310c27e3dSHong Zhang   }
307410c27e3dSHong Zhang 
30756cf91177SBarry Smith #if defined(PETSC_USE_INFO)
30761393bc97SHong Zhang   if (ai[am] != 0) {
30771393bc97SHong Zhang     PetscReal af = (PetscReal)(ui[am])/((PetscReal)ai[am]);
3078ae15b995SBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr);
3079ae15b995SBarry Smith     ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
3080ae15b995SBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr);
308110c27e3dSHong Zhang   } else {
3082ae15b995SBarry Smith      ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr);
308310c27e3dSHong Zhang   }
308463ba0a88SBarry Smith #endif
308510c27e3dSHong Zhang 
308610c27e3dSHong Zhang   ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
30879bfd6278SHong Zhang   ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
30880e83c824SBarry Smith   ierr = PetscFree4(ui_ptr,jl,il,cols);CHKERRQ(ierr);
308910c27e3dSHong Zhang 
309010c27e3dSHong Zhang   /* destroy list of free space and other temporary array(s) */
30911393bc97SHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
3092a1a86e44SBarry Smith   ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr);
309310c27e3dSHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
309410c27e3dSHong Zhang 
309510c27e3dSHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
309610c27e3dSHong Zhang 
3097f284f12aSHong Zhang   b = (Mat_SeqSBAIJ*)fact->data;
309810c27e3dSHong Zhang   b->singlemalloc = PETSC_FALSE;
3099e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
3100e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
31011393bc97SHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr);
310210c27e3dSHong Zhang   b->j    = uj;
310310c27e3dSHong Zhang   b->i    = ui;
310410c27e3dSHong Zhang   b->diag = 0;
310510c27e3dSHong Zhang   b->ilen = 0;
310610c27e3dSHong Zhang   b->imax = 0;
310710c27e3dSHong Zhang   b->row  = perm;
31089bfd6278SHong Zhang   b->col  = perm;
31099bfd6278SHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
31109bfd6278SHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
31119bfd6278SHong Zhang   b->icol = iperm;
311210c27e3dSHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
31131393bc97SHong Zhang   ierr    = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
3114719d5645SBarry Smith   ierr    = PetscLogObjectMemory(fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
31151393bc97SHong Zhang   b->maxnz = b->nz = ui[am];
311610c27e3dSHong Zhang 
3117f284f12aSHong Zhang   fact->info.factor_mallocs    = reallocs;
3118f284f12aSHong Zhang   fact->info.fill_ratio_given  = fill;
31191393bc97SHong Zhang   if (ai[am] != 0) {
3120f284f12aSHong Zhang     fact->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]);
312110c27e3dSHong Zhang   } else {
3122f284f12aSHong Zhang     fact->info.fill_ratio_needed = 0.0;
312310c27e3dSHong Zhang   }
31240a3c351aSHong Zhang   fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_inplace;
3125f76d2b81SHong Zhang   PetscFunctionReturn(0);
3126f76d2b81SHong Zhang }
3127599ef60dSHong Zhang 
31281d098868SHong Zhang #undef __FUNCT__
312935233d99SShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering"
313035233d99SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering(Mat A,Vec bb,Vec xx)
3131813a1f2bSShri Abhyankar {
3132813a1f2bSShri Abhyankar   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
3133813a1f2bSShri Abhyankar   PetscErrorCode    ierr;
3134813a1f2bSShri Abhyankar   PetscInt          n = A->rmap->n;
3135813a1f2bSShri Abhyankar   const PetscInt    *ai = a->i,*aj = a->j,*adiag = a->diag,*vi;
3136813a1f2bSShri Abhyankar   PetscScalar       *x,sum;
3137813a1f2bSShri Abhyankar   const PetscScalar *b;
3138813a1f2bSShri Abhyankar   const MatScalar   *aa = a->a,*v;
3139813a1f2bSShri Abhyankar   PetscInt          i,nz;
3140813a1f2bSShri Abhyankar 
3141813a1f2bSShri Abhyankar   PetscFunctionBegin;
3142813a1f2bSShri Abhyankar   if (!n) PetscFunctionReturn(0);
3143813a1f2bSShri Abhyankar 
31443649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
3145813a1f2bSShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
3146813a1f2bSShri Abhyankar 
3147813a1f2bSShri Abhyankar   /* forward solve the lower triangular */
3148813a1f2bSShri Abhyankar   x[0] = b[0];
3149813a1f2bSShri Abhyankar   v    = aa;
3150813a1f2bSShri Abhyankar   vi   = aj;
3151813a1f2bSShri Abhyankar   for (i=1; i<n; i++) {
3152813a1f2bSShri Abhyankar     nz  = ai[i+1] - ai[i];
3153813a1f2bSShri Abhyankar     sum = b[i];
3154813a1f2bSShri Abhyankar     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
3155813a1f2bSShri Abhyankar     v  += nz;
3156813a1f2bSShri Abhyankar     vi += nz;
3157813a1f2bSShri Abhyankar     x[i] = sum;
3158813a1f2bSShri Abhyankar   }
3159813a1f2bSShri Abhyankar 
3160813a1f2bSShri Abhyankar   /* backward solve the upper triangular */
316162fbd6afSShri Abhyankar   for (i=n-1; i>=0; i--){
31623c0119dfSShri Abhyankar     v   = aa + adiag[i+1] + 1;
31633c0119dfSShri Abhyankar     vi  = aj + adiag[i+1] + 1;
3164813a1f2bSShri Abhyankar     nz = adiag[i] - adiag[i+1]-1;
3165813a1f2bSShri Abhyankar     sum = x[i];
3166813a1f2bSShri Abhyankar     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
31673c0119dfSShri Abhyankar     x[i] = sum*v[nz]; /* x[i]=aa[adiag[i]]*sum; v++; */
3168813a1f2bSShri Abhyankar   }
3169813a1f2bSShri Abhyankar 
3170813a1f2bSShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
31713649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
3172813a1f2bSShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
3173813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
3174813a1f2bSShri Abhyankar }
3175813a1f2bSShri Abhyankar 
31761d098868SHong Zhang #undef __FUNCT__
317735233d99SShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ"
317835233d99SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ(Mat A,Vec bb,Vec xx)
3179f268cba8SShri Abhyankar {
3180f268cba8SShri Abhyankar   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
3181f268cba8SShri Abhyankar   IS                iscol = a->col,isrow = a->row;
3182f268cba8SShri Abhyankar   PetscErrorCode    ierr;
3183f268cba8SShri Abhyankar   PetscInt          i,n=A->rmap->n,*vi,*ai=a->i,*aj=a->j,*adiag = a->diag,nz;
3184f268cba8SShri Abhyankar   const PetscInt    *rout,*cout,*r,*c;
3185f268cba8SShri Abhyankar   PetscScalar       *x,*tmp,sum;
3186f268cba8SShri Abhyankar   const PetscScalar *b;
3187f268cba8SShri Abhyankar   const MatScalar   *aa = a->a,*v;
3188f268cba8SShri Abhyankar 
3189f268cba8SShri Abhyankar   PetscFunctionBegin;
3190f268cba8SShri Abhyankar   if (!n) PetscFunctionReturn(0);
3191f268cba8SShri Abhyankar 
31923649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
3193f268cba8SShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
3194f268cba8SShri Abhyankar   tmp  = a->solve_work;
3195f268cba8SShri Abhyankar 
3196f268cba8SShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
3197f268cba8SShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
3198f268cba8SShri Abhyankar 
3199f268cba8SShri Abhyankar   /* forward solve the lower triangular */
3200f268cba8SShri Abhyankar   tmp[0] = b[r[0]];
3201f268cba8SShri Abhyankar   v      = aa;
3202f268cba8SShri Abhyankar   vi     = aj;
3203f268cba8SShri Abhyankar   for (i=1; i<n; i++) {
3204f268cba8SShri Abhyankar     nz  = ai[i+1] - ai[i];
3205f268cba8SShri Abhyankar     sum = b[r[i]];
3206f268cba8SShri Abhyankar     PetscSparseDenseMinusDot(sum,tmp,v,vi,nz);
3207f268cba8SShri Abhyankar     tmp[i] = sum;
3208f268cba8SShri Abhyankar     v += nz; vi += nz;
3209f268cba8SShri Abhyankar   }
3210f268cba8SShri Abhyankar 
3211f268cba8SShri Abhyankar   /* backward solve the upper triangular */
3212f268cba8SShri Abhyankar   for (i=n-1; i>=0; i--){
32133c0119dfSShri Abhyankar     v   = aa + adiag[i+1]+1;
32143c0119dfSShri Abhyankar     vi  = aj + adiag[i+1]+1;
3215f268cba8SShri Abhyankar     nz  = adiag[i]-adiag[i+1]-1;
3216f268cba8SShri Abhyankar     sum = tmp[i];
3217f268cba8SShri Abhyankar     PetscSparseDenseMinusDot(sum,tmp,v,vi,nz);
3218f268cba8SShri Abhyankar     x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */
3219f268cba8SShri Abhyankar   }
3220f268cba8SShri Abhyankar 
3221f268cba8SShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
3222f268cba8SShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
32233649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
3224f268cba8SShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
3225f268cba8SShri Abhyankar   ierr = PetscLogFlops(2*a->nz - A->cmap->n);CHKERRQ(ierr);
3226f268cba8SShri Abhyankar   PetscFunctionReturn(0);
3227f268cba8SShri Abhyankar }
3228f268cba8SShri Abhyankar 
3229f268cba8SShri Abhyankar #undef __FUNCT__
32301d098868SHong Zhang #define __FUNCT__ "MatILUDTFactor_SeqAIJ"
3231fe97e370SBarry Smith /*
3232fe97e370SBarry 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
3233fe97e370SBarry Smith */
32341d098868SHong Zhang PetscErrorCode MatILUDTFactor_SeqAIJ(Mat A,IS isrow,IS iscol,const MatFactorInfo *info,Mat *fact)
32351d098868SHong Zhang {
32361d098868SHong Zhang   Mat                B = *fact;
3237599ef60dSHong Zhang   Mat_SeqAIJ         *a=(Mat_SeqAIJ*)A->data,*b;
3238599ef60dSHong Zhang   IS                 isicol;
3239599ef60dSHong Zhang   PetscErrorCode     ierr;
32401d098868SHong Zhang   const PetscInt     *r,*ic;
32411d098868SHong Zhang   PetscInt           i,n=A->rmap->n,*ai=a->i,*aj=a->j,*ajtmp,*adiag;
3242dc3a2fd3SHong Zhang   PetscInt           *bi,*bj,*bdiag,*bdiag_rev;
3243f61a948aSLisandro Dalcin   PetscInt           row,nzi,nzi_bl,nzi_bu,*im,nzi_al,nzi_au;
32441d098868SHong Zhang   PetscInt           nlnk,*lnk;
32451d098868SHong Zhang   PetscBT            lnkbt;
3246ace3abfcSBarry Smith   PetscBool          row_identity,icol_identity;
32478aee2decSHong Zhang   MatScalar          *aatmp,*pv,*batmp,*ba,*rtmp,*pc,multiplier,*vtmp,diag_tmp;
32481d098868SHong Zhang   const PetscInt     *ics;
32491d098868SHong Zhang   PetscInt           j,nz,*pj,*bjtmp,k,ncut,*jtmp;
3250182b8fbaSHong Zhang   PetscReal          dt=info->dt,dtcol=info->dtcol,shift=info->shiftamount;
3251f61a948aSLisandro Dalcin   PetscInt           dtcount=(PetscInt)info->dtcount,nnz_max;
3252ace3abfcSBarry Smith   PetscBool          missing;
3253599ef60dSHong Zhang 
3254599ef60dSHong Zhang   PetscFunctionBegin;
3255f61a948aSLisandro Dalcin 
3256f61a948aSLisandro Dalcin   if (dt      == PETSC_DEFAULT) dt      = 0.005;
3257f61a948aSLisandro Dalcin   if (dtcol   == PETSC_DEFAULT) dtcol   = 0.01; /* XXX unused! */
3258f61a948aSLisandro Dalcin   if (dtcount == PETSC_DEFAULT) dtcount = (PetscInt)(1.5*a->rmax);
3259f61a948aSLisandro Dalcin 
32601d098868SHong Zhang   /* ------- symbolic factorization, can be reused ---------*/
32611d098868SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr);
3262e32f2f54SBarry Smith   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i);
32631d098868SHong Zhang   adiag=a->diag;
3264599ef60dSHong Zhang 
3265599ef60dSHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
3266599ef60dSHong Zhang 
3267599ef60dSHong Zhang   /* bdiag is location of diagonal in factor */
32681bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);     /* becomes b->diag */
32691bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag_rev);CHKERRQ(ierr); /* temporary */
3270599ef60dSHong Zhang 
32711d098868SHong Zhang   /* allocate row pointers bi */
32728fc3a347SHong Zhang   ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
32731d098868SHong Zhang 
32741d098868SHong Zhang   /* allocate bj and ba; max num of nonzero entries is (ai[n]+2*n*dtcount+2) */
3275393d3a68SHong Zhang   if (dtcount > n-1) dtcount = n-1; /* diagonal is excluded */
32761d098868SHong Zhang   nnz_max  = ai[n]+2*n*dtcount+2;
32778fc3a347SHong Zhang 
32788fc3a347SHong Zhang   ierr = PetscMalloc((nnz_max+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
32798fc3a347SHong Zhang   ierr = PetscMalloc((nnz_max+1)*sizeof(MatScalar),&ba);CHKERRQ(ierr);
3280599ef60dSHong Zhang 
3281599ef60dSHong Zhang   /* put together the new matrix */
3282599ef60dSHong Zhang   ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
3283599ef60dSHong Zhang   ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr);
3284f284f12aSHong Zhang   b    = (Mat_SeqAIJ*)B->data;
3285599ef60dSHong Zhang   b->free_a       = PETSC_TRUE;
3286599ef60dSHong Zhang   b->free_ij      = PETSC_TRUE;
3287599ef60dSHong Zhang   b->singlemalloc = PETSC_FALSE;
3288599ef60dSHong Zhang   b->a          = ba;
3289599ef60dSHong Zhang   b->j          = bj;
3290599ef60dSHong Zhang   b->i          = bi;
3291599ef60dSHong Zhang   b->diag       = bdiag;
3292599ef60dSHong Zhang   b->ilen       = 0;
3293599ef60dSHong Zhang   b->imax       = 0;
3294599ef60dSHong Zhang   b->row        = isrow;
3295599ef60dSHong Zhang   b->col        = iscol;
3296599ef60dSHong Zhang   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
3297599ef60dSHong Zhang   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
3298599ef60dSHong Zhang   b->icol       = isicol;
3299599ef60dSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
3300599ef60dSHong Zhang 
33011d098868SHong Zhang   ierr = PetscLogObjectMemory(B,nnz_max*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
33021d098868SHong Zhang   b->maxnz = nnz_max;
3303599ef60dSHong Zhang 
3304d5f3da31SBarry Smith   B->factortype            = MAT_FACTOR_ILUDT;
3305f284f12aSHong Zhang   B->info.factor_mallocs   = 0;
3306f284f12aSHong Zhang   B->info.fill_ratio_given = ((PetscReal)nnz_max)/((PetscReal)ai[n]);
33071d098868SHong Zhang   CHKMEMQ;
33081d098868SHong Zhang   /* ------- end of symbolic factorization ---------*/
3309599ef60dSHong Zhang 
3310599ef60dSHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
3311599ef60dSHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
3312599ef60dSHong Zhang   ics  = ic;
3313599ef60dSHong Zhang 
3314599ef60dSHong Zhang   /* linked list for storing column indices of the active row */
3315599ef60dSHong Zhang   nlnk = n + 1;
3316599ef60dSHong Zhang   ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
33171d098868SHong Zhang 
33181d098868SHong Zhang   /* im: used by PetscLLAddSortedLU(); jtmp: working array for column indices of active row */
33190e83c824SBarry Smith   ierr = PetscMalloc2(n,PetscInt,&im,n,PetscInt,&jtmp);CHKERRQ(ierr);
33201d098868SHong Zhang   /* rtmp, vtmp: working arrays for sparse and contiguous row entries of active row */
33210e83c824SBarry Smith   ierr = PetscMalloc2(n,MatScalar,&rtmp,n,MatScalar,&vtmp);CHKERRQ(ierr);
33220e83c824SBarry Smith   ierr = PetscMemzero(rtmp,n*sizeof(MatScalar));CHKERRQ(ierr);
3323599ef60dSHong Zhang 
3324599ef60dSHong Zhang   bi[0]    = 0;
33258fc3a347SHong Zhang   bdiag[0] = nnz_max-1; /* location of diag[0] in factor B */
3326dc3a2fd3SHong Zhang   bdiag_rev[n] = bdiag[0];
33278fc3a347SHong Zhang   bi[2*n+1] = bdiag[0]+1; /* endof bj and ba array */
3328599ef60dSHong Zhang   for (i=0; i<n; i++) {
3329599ef60dSHong Zhang     /* copy initial fill into linked list */
3330599ef60dSHong Zhang     nzi = 0; /* nonzeros for active row i */
33318369b28dSHong Zhang     nzi = ai[r[i]+1] - ai[r[i]];
3332e32f2f54SBarry 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);
33338369b28dSHong Zhang     nzi_al = adiag[r[i]] - ai[r[i]];
33348369b28dSHong Zhang     nzi_au = ai[r[i]+1] - adiag[r[i]] -1;
3335599ef60dSHong Zhang     ajtmp = aj + ai[r[i]];
33368369b28dSHong Zhang     ierr = PetscLLAddPerm(nzi,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
3337599ef60dSHong Zhang 
3338599ef60dSHong Zhang     /* load in initial (unfactored row) */
33391d098868SHong Zhang     aatmp = a->a + ai[r[i]];
33408369b28dSHong Zhang     for (j=0; j<nzi; j++) {
33411d098868SHong Zhang       rtmp[ics[*ajtmp++]] = *aatmp++;
3342599ef60dSHong Zhang     }
3343599ef60dSHong Zhang 
3344599ef60dSHong Zhang     /* add pivot rows into linked list */
3345599ef60dSHong Zhang     row = lnk[n];
3346599ef60dSHong Zhang     while (row < i ) {
33471d098868SHong Zhang       nzi_bl = bi[row+1] - bi[row] + 1;
33481d098868SHong Zhang       bjtmp = bj + bdiag[row+1]+1; /* points to 1st column next to the diagonal in U */
33491d098868SHong Zhang       ierr  = PetscLLAddSortedLU(bjtmp,row,nlnk,lnk,lnkbt,i,nzi_bl,im);CHKERRQ(ierr);
3350599ef60dSHong Zhang       nzi  += nlnk;
3351599ef60dSHong Zhang       row   = lnk[row];
3352599ef60dSHong Zhang     }
3353599ef60dSHong Zhang 
33548369b28dSHong Zhang     /* copy data from lnk into jtmp, then initialize lnk */
33558369b28dSHong Zhang     ierr = PetscLLClean(n,n,nzi,lnk,jtmp,lnkbt);CHKERRQ(ierr);
3356599ef60dSHong Zhang 
3357599ef60dSHong Zhang     /* numerical factorization */
33588369b28dSHong Zhang     bjtmp = jtmp;
3359599ef60dSHong Zhang     row   = *bjtmp++; /* 1st pivot row */
3360599ef60dSHong Zhang     while  ( row < i ) {
3361599ef60dSHong Zhang       pc         = rtmp + row;
33623c2a7987SHong Zhang       pv         = ba + bdiag[row]; /* 1./(diag of the pivot row) */
33633c2a7987SHong Zhang       multiplier = (*pc) * (*pv);
3364599ef60dSHong Zhang       *pc        = multiplier;
33653c2a7987SHong Zhang       if (PetscAbsScalar(*pc) > dt){ /* apply tolerance dropping rule */
33661d098868SHong Zhang         pj         = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */
33671d098868SHong Zhang         pv         = ba + bdiag[row+1] + 1;
33681d098868SHong Zhang         /* if (multiplier < -1.0 or multiplier >1.0) printf("row/prow %d, %d, multiplier %g\n",i,row,multiplier); */
33691d098868SHong Zhang         nz         = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */
33701d098868SHong Zhang         for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++);
3371dc0b31edSSatish Balay         ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr);
3372599ef60dSHong Zhang       }
3373599ef60dSHong Zhang       row = *bjtmp++;
3374599ef60dSHong Zhang     }
3375599ef60dSHong Zhang 
33768369b28dSHong Zhang     /* copy sparse rtmp into contiguous vtmp; separate L and U part */
33778aee2decSHong Zhang     diag_tmp = rtmp[i];  /* save diagonal value - may not needed?? */
33788369b28dSHong Zhang     nzi_bl = 0; j = 0;
33798369b28dSHong Zhang     while (jtmp[j] < i){ /* Note: jtmp is sorted */
33808aee2decSHong Zhang       vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0;
33818369b28dSHong Zhang       nzi_bl++; j++;
33828369b28dSHong Zhang     }
33838369b28dSHong Zhang     nzi_bu = nzi - nzi_bl -1;
33848369b28dSHong Zhang     while (j < nzi){
33858aee2decSHong Zhang       vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0;
33868369b28dSHong Zhang       j++;
33878369b28dSHong Zhang     }
33888369b28dSHong Zhang 
3389599ef60dSHong Zhang     bjtmp = bj + bi[i];
3390599ef60dSHong Zhang     batmp = ba + bi[i];
33918369b28dSHong Zhang     /* apply level dropping rule to L part */
33928369b28dSHong Zhang     ncut = nzi_al + dtcount;
33938369b28dSHong Zhang     if (ncut < nzi_bl){
33948369b28dSHong Zhang       ierr = PetscSortSplit(ncut,nzi_bl,vtmp,jtmp);CHKERRQ(ierr);
33958369b28dSHong Zhang       ierr = PetscSortIntWithScalarArray(ncut,jtmp,vtmp);CHKERRQ(ierr);
3396599ef60dSHong Zhang     } else {
33978369b28dSHong Zhang       ncut = nzi_bl;
33988369b28dSHong Zhang     }
33998369b28dSHong Zhang     for (j=0; j<ncut; j++){
34008369b28dSHong Zhang       bjtmp[j] = jtmp[j];
34018369b28dSHong Zhang       batmp[j] = vtmp[j];
34026da515a1SHong Zhang       /* printf(" (%d,%g),",bjtmp[j],batmp[j]); */
34038369b28dSHong Zhang     }
34041d098868SHong Zhang     bi[i+1] = bi[i] + ncut;
34058369b28dSHong Zhang     nzi = ncut + 1;
34068369b28dSHong Zhang 
34078369b28dSHong Zhang     /* apply level dropping rule to U part */
34088369b28dSHong Zhang     ncut = nzi_au + dtcount;
34098369b28dSHong Zhang     if (ncut < nzi_bu){
34108369b28dSHong Zhang       ierr = PetscSortSplit(ncut,nzi_bu,vtmp+nzi_bl+1,jtmp+nzi_bl+1);CHKERRQ(ierr);
34118369b28dSHong Zhang       ierr = PetscSortIntWithScalarArray(ncut,jtmp+nzi_bl+1,vtmp+nzi_bl+1);CHKERRQ(ierr);
34128369b28dSHong Zhang     } else {
34138369b28dSHong Zhang       ncut = nzi_bu;
34148369b28dSHong Zhang     }
34158369b28dSHong Zhang     nzi += ncut;
34161d098868SHong Zhang 
34171d098868SHong Zhang     /* mark bdiagonal */
34181d098868SHong Zhang     bdiag[i+1]       = bdiag[i] - (ncut + 1);
3419dc3a2fd3SHong Zhang     bdiag_rev[n-i-1] = bdiag[i+1];
34208fc3a347SHong Zhang     bi[2*n - i]      = bi[2*n - i +1] - (ncut + 1);
34211d098868SHong Zhang     bjtmp = bj + bdiag[i];
34221d098868SHong Zhang     batmp = ba + bdiag[i];
34231d098868SHong Zhang     *bjtmp = i;
34248aee2decSHong Zhang     *batmp = diag_tmp; /* rtmp[i]; */
3425c9c72f8fSHong Zhang     if (*batmp == 0.0) {
3426c9c72f8fSHong Zhang       *batmp = dt+shift;
3427bd1bc851SBarry Smith       /* printf(" row %d add shift %g\n",i,shift); */
3428c9c72f8fSHong Zhang     }
3429b78a477dSHong Zhang     *batmp = 1.0/(*batmp); /* invert diagonal entries for simplier triangular solves */
34306da515a1SHong Zhang     /* printf(" (%d,%g),",*bjtmp,*batmp); */
34311d098868SHong Zhang 
34321d098868SHong Zhang     bjtmp = bj + bdiag[i+1]+1;
34331d098868SHong Zhang     batmp = ba + bdiag[i+1]+1;
34348369b28dSHong Zhang     for (k=0; k<ncut; k++){
34351d098868SHong Zhang       bjtmp[k] = jtmp[nzi_bl+1+k];
34361d098868SHong Zhang       batmp[k] = vtmp[nzi_bl+1+k];
34376da515a1SHong Zhang       /* printf(" (%d,%g),",bjtmp[k],batmp[k]); */
3438599ef60dSHong Zhang     }
34396da515a1SHong Zhang     /* printf("\n"); */
3440599ef60dSHong Zhang 
3441599ef60dSHong Zhang     im[i]   = nzi; /* used by PetscLLAddSortedLU() */
3442599ef60dSHong Zhang     /*
34431d098868SHong Zhang     printf("row %d: bi %d, bdiag %d\n",i,bi[i],bdiag[i]);
34448369b28dSHong Zhang     printf(" ----------------------------\n");
3445599ef60dSHong Zhang     */
34468369b28dSHong Zhang   } /* for (i=0; i<n; i++) */
34471d098868SHong Zhang   /* printf("end of L %d, beginning of U %d\n",bi[n],bdiag[n]); */
3448e32f2f54SBarry 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]);
3449599ef60dSHong Zhang 
3450599ef60dSHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
3451599ef60dSHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
3452599ef60dSHong Zhang 
3453599ef60dSHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
34540e83c824SBarry Smith   ierr = PetscFree2(im,jtmp);CHKERRQ(ierr);
34550e83c824SBarry Smith   ierr = PetscFree2(rtmp,vtmp);CHKERRQ(ierr);
34561bfa06eaSJed Brown   ierr = PetscFree(bdiag_rev);CHKERRQ(ierr);
3457599ef60dSHong Zhang 
3458599ef60dSHong Zhang   ierr = PetscLogFlops(B->cmap->n);CHKERRQ(ierr);
34591d098868SHong Zhang   b->maxnz = b->nz = bi[n] + bdiag[0] - bdiag[n];
3460599ef60dSHong Zhang 
3461599ef60dSHong Zhang   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
3462599ef60dSHong Zhang   ierr = ISIdentity(isicol,&icol_identity);CHKERRQ(ierr);
3463599ef60dSHong Zhang   if (row_identity && icol_identity) {
346435233d99SShri Abhyankar     B->ops->solve = MatSolve_SeqAIJ_NaturalOrdering;
3465599ef60dSHong Zhang   } else {
346635233d99SShri Abhyankar     B->ops->solve = MatSolve_SeqAIJ;
3467599ef60dSHong Zhang   }
3468599ef60dSHong Zhang 
34691d098868SHong Zhang   B->ops->solveadd          = 0;
34701d098868SHong Zhang   B->ops->solvetranspose    = 0;
34711d098868SHong Zhang   B->ops->solvetransposeadd = 0;
34721d098868SHong Zhang   B->ops->matsolve          = 0;
3473599ef60dSHong Zhang   B->assembled              = PETSC_TRUE;
3474599ef60dSHong Zhang   B->preallocated           = PETSC_TRUE;
3475599ef60dSHong Zhang   PetscFunctionReturn(0);
3476599ef60dSHong Zhang }
3477599ef60dSHong Zhang 
34783c2a7987SHong Zhang /* a wraper of MatILUDTFactor_SeqAIJ() */
34793c2a7987SHong Zhang #undef __FUNCT__
34803c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorSymbolic_SeqAIJ"
3481fe97e370SBarry Smith /*
3482fe97e370SBarry 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
3483fe97e370SBarry Smith */
3484fe97e370SBarry Smith 
34857087cfbeSBarry Smith PetscErrorCode  MatILUDTFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS row,IS col,const MatFactorInfo *info)
34863c2a7987SHong Zhang {
34873c2a7987SHong Zhang   PetscErrorCode     ierr;
34883c2a7987SHong Zhang 
34893c2a7987SHong Zhang   PetscFunctionBegin;
34903c2a7987SHong Zhang   ierr = MatILUDTFactor_SeqAIJ(A,row,col,info,&fact);CHKERRQ(ierr);
34913c2a7987SHong Zhang   PetscFunctionReturn(0);
34923c2a7987SHong Zhang }
34933c2a7987SHong Zhang 
34943c2a7987SHong Zhang /*
34953c2a7987SHong Zhang    same as MatLUFactorNumeric_SeqAIJ(), except using contiguous array matrix factors
34963c2a7987SHong Zhang    - intend to replace existing MatLUFactorNumeric_SeqAIJ()
34973c2a7987SHong Zhang */
34983c2a7987SHong Zhang #undef __FUNCT__
34993c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorNumeric_SeqAIJ"
3500fe97e370SBarry Smith /*
3501fe97e370SBarry 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
3502fe97e370SBarry Smith */
3503fe97e370SBarry Smith 
35047087cfbeSBarry Smith PetscErrorCode  MatILUDTFactorNumeric_SeqAIJ(Mat fact,Mat A,const MatFactorInfo *info)
35053c2a7987SHong Zhang {
35063c2a7987SHong Zhang   Mat            C=fact;
35073c2a7987SHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data;
35083c2a7987SHong Zhang   IS             isrow = b->row,isicol = b->icol;
35093c2a7987SHong Zhang   PetscErrorCode ierr;
35103c2a7987SHong Zhang   const PetscInt *r,*ic,*ics;
3511b78a477dSHong Zhang   PetscInt       i,j,k,n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j;
3512b78a477dSHong Zhang   PetscInt       *ajtmp,*bjtmp,nz,nzl,nzu,row,*bdiag = b->diag,*pj;
35133c2a7987SHong Zhang   MatScalar      *rtmp,*pc,multiplier,*v,*pv,*aa=a->a;
3514182b8fbaSHong Zhang   PetscReal      dt=info->dt,shift=info->shiftamount;
3515ace3abfcSBarry Smith   PetscBool      row_identity, col_identity;
35163c2a7987SHong Zhang 
35173c2a7987SHong Zhang   PetscFunctionBegin;
35183c2a7987SHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
35193c2a7987SHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
3520b78a477dSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr);
35213c2a7987SHong Zhang   ics  = ic;
35223c2a7987SHong Zhang 
35233c2a7987SHong Zhang   for (i=0; i<n; i++){
3524b78a477dSHong Zhang     /* initialize rtmp array */
3525b78a477dSHong Zhang     nzl   = bi[i+1] - bi[i];       /* num of nozeros in L(i,:) */
35263c2a7987SHong Zhang     bjtmp = bj + bi[i];
3527b78a477dSHong Zhang     for  (j=0; j<nzl; j++) rtmp[*bjtmp++] = 0.0;
3528b78a477dSHong Zhang     rtmp[i] = 0.0;
3529b78a477dSHong Zhang     nzu   = bdiag[i] - bdiag[i+1]; /* num of nozeros in U(i,:) */
3530b78a477dSHong Zhang     bjtmp = bj + bdiag[i+1] + 1;
3531b78a477dSHong Zhang     for  (j=0; j<nzu; j++) rtmp[*bjtmp++] = 0.0;
35323c2a7987SHong Zhang 
3533b78a477dSHong Zhang     /* load in initial unfactored row of A */
35346da515a1SHong Zhang     /* printf("row %d\n",i); */
35353c2a7987SHong Zhang     nz    = ai[r[i]+1] - ai[r[i]];
35363c2a7987SHong Zhang     ajtmp = aj + ai[r[i]];
35373c2a7987SHong Zhang     v     = aa + ai[r[i]];
35383c2a7987SHong Zhang     for (j=0; j<nz; j++) {
3539b78a477dSHong Zhang       rtmp[ics[*ajtmp++]] = v[j];
35406da515a1SHong Zhang       /* printf(" (%d,%g),",ics[ajtmp[j]],rtmp[ics[ajtmp[j]]]); */
35413c2a7987SHong Zhang     }
35426da515a1SHong Zhang     /* printf("\n"); */
35433c2a7987SHong Zhang 
3544b78a477dSHong Zhang     /* numerical factorization */
3545b78a477dSHong Zhang     bjtmp = bj + bi[i]; /* point to 1st entry of L(i,:) */
3546b78a477dSHong Zhang     nzl   = bi[i+1] - bi[i]; /* num of entries in L(i,:) */
3547b78a477dSHong Zhang     k = 0;
3548b78a477dSHong Zhang     while (k < nzl){
35493c2a7987SHong Zhang       row   = *bjtmp++;
35506da515a1SHong Zhang       /* printf("  prow %d\n",row); */
35513c2a7987SHong Zhang       pc         = rtmp + row;
3552b78a477dSHong Zhang       pv         = b->a + bdiag[row]; /* 1./(diag of the pivot row) */
3553b78a477dSHong Zhang       multiplier = (*pc) * (*pv);
35543c2a7987SHong Zhang       *pc        = multiplier;
3555b78a477dSHong Zhang       if (PetscAbsScalar(multiplier) > dt){
3556b78a477dSHong Zhang         pj         = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */
3557b78a477dSHong Zhang         pv         = b->a + bdiag[row+1] + 1;
3558b78a477dSHong Zhang         nz         = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */
3559b78a477dSHong Zhang         for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++);
35606da515a1SHong Zhang         /* ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); */
35613c2a7987SHong Zhang       }
3562b78a477dSHong Zhang       k++;
35633c2a7987SHong Zhang     }
35643c2a7987SHong Zhang 
3565b78a477dSHong Zhang     /* finished row so stick it into b->a */
3566b78a477dSHong Zhang     /* L-part */
3567b78a477dSHong Zhang     pv = b->a + bi[i] ;
3568b78a477dSHong Zhang     pj = bj + bi[i] ;
3569b78a477dSHong Zhang     nzl = bi[i+1] - bi[i];
3570b78a477dSHong Zhang     for (j=0; j<nzl; j++) {
3571b78a477dSHong Zhang       pv[j] = rtmp[pj[j]];
35726da515a1SHong Zhang       /* printf(" (%d,%g),",pj[j],pv[j]); */
35733c2a7987SHong Zhang     }
3574b78a477dSHong Zhang 
3575b78a477dSHong Zhang     /* diagonal: invert diagonal entries for simplier triangular solves */
3576b78a477dSHong Zhang     if (rtmp[i] == 0.0) rtmp[i] = dt+shift;
3577b78a477dSHong Zhang     b->a[bdiag[i]] = 1.0/rtmp[i];
35786da515a1SHong Zhang     /* printf(" (%d,%g),",i,b->a[bdiag[i]]); */
3579b78a477dSHong Zhang 
3580b78a477dSHong Zhang     /* U-part */
3581b78a477dSHong Zhang     pv = b->a + bdiag[i+1] + 1;
3582b78a477dSHong Zhang     pj = bj + bdiag[i+1] + 1;
3583b78a477dSHong Zhang     nzu = bdiag[i] - bdiag[i+1] - 1;
3584b78a477dSHong Zhang     for (j=0; j<nzu; j++) {
3585b78a477dSHong Zhang       pv[j] = rtmp[pj[j]];
35866da515a1SHong Zhang       /* printf(" (%d,%g),",pj[j],pv[j]); */
3587b78a477dSHong Zhang     }
35886da515a1SHong Zhang     /* printf("\n"); */
3589b78a477dSHong Zhang   }
3590b78a477dSHong Zhang 
35913c2a7987SHong Zhang   ierr = PetscFree(rtmp);CHKERRQ(ierr);
35923c2a7987SHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
35933c2a7987SHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
3594b78a477dSHong Zhang 
35953c2a7987SHong Zhang   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
35963c2a7987SHong Zhang   ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr);
35973c2a7987SHong Zhang   if (row_identity && col_identity) {
359835233d99SShri Abhyankar     C->ops->solve   = MatSolve_SeqAIJ_NaturalOrdering;
35993c2a7987SHong Zhang   } else {
360035233d99SShri Abhyankar     C->ops->solve   = MatSolve_SeqAIJ;
36013c2a7987SHong Zhang   }
3602b78a477dSHong Zhang   C->ops->solveadd           = 0;
3603b78a477dSHong Zhang   C->ops->solvetranspose     = 0;
3604b78a477dSHong Zhang   C->ops->solvetransposeadd  = 0;
3605b78a477dSHong Zhang   C->ops->matsolve           = 0;
36063c2a7987SHong Zhang   C->assembled    = PETSC_TRUE;
36073c2a7987SHong Zhang   C->preallocated = PETSC_TRUE;
36083c2a7987SHong Zhang   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
36093c2a7987SHong Zhang   PetscFunctionReturn(0);
36103c2a7987SHong Zhang }
3611