xref: /petsc/src/mat/impls/aij/seq/aijfact.c (revision 1a26ff17b5d9e461fafa0690dabf2fd6ce0cfbd5)
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;
121b24902e0SBarry Smith   } else if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) {
122b24902e0SBarry Smith     ierr = MatSetType(*B,MATSEQSBAIJ);CHKERRQ(ierr);
123b24902e0SBarry Smith     ierr = MatSeqSBAIJSetPreallocation(*B,1,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
12435233d99SShri Abhyankar     (*B)->ops->iccfactorsymbolic      = MatICCFactorSymbolic_SeqAIJ;
12535233d99SShri Abhyankar     (*B)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqAIJ;
126e32f2f54SBarry Smith   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Factor type not supported");
127d5f3da31SBarry Smith   (*B)->factortype = ftype;
128b24902e0SBarry Smith   PetscFunctionReturn(0);
129b24902e0SBarry Smith }
130e631078cSBarry Smith EXTERN_C_END
131b24902e0SBarry Smith 
132b24902e0SBarry Smith #undef __FUNCT__
133ad04f41aSHong Zhang #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ_inplace"
134ad04f41aSHong Zhang PetscErrorCode MatLUFactorSymbolic_SeqAIJ_inplace(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
135289bc588SBarry Smith {
136416022c9SBarry Smith   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
137289bc588SBarry Smith   IS                 isicol;
1386849ba73SBarry Smith   PetscErrorCode     ierr;
1395d0c19d7SBarry Smith   const PetscInt     *r,*ic;
1405d0c19d7SBarry Smith   PetscInt           i,n=A->rmap->n,*ai=a->i,*aj=a->j;
1411393bc97SHong Zhang   PetscInt           *bi,*bj,*ajtmp;
1421393bc97SHong Zhang   PetscInt           *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im;
1439e046878SBarry Smith   PetscReal          f;
1444875ba38SHong Zhang   PetscInt           nlnk,*lnk,k,**bi_ptr;
145a1a86e44SBarry Smith   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
1461393bc97SHong Zhang   PetscBT            lnkbt;
147289bc588SBarry Smith 
1483a40ed3dSBarry Smith   PetscFunctionBegin;
149e32f2f54SBarry Smith   if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"matrix must be square");
1504c49b128SBarry Smith   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
151ac121b3dSBarry Smith   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
152ac121b3dSBarry Smith   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
153289bc588SBarry Smith 
154289bc588SBarry Smith   /* get new row pointers */
1551393bc97SHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
1561393bc97SHong Zhang   bi[0] = 0;
1571393bc97SHong Zhang 
1581393bc97SHong Zhang   /* bdiag is location of diagonal in factor */
1591393bc97SHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
1601393bc97SHong Zhang   bdiag[0] = 0;
1611393bc97SHong Zhang 
1621393bc97SHong Zhang   /* linked list for storing column indices of the active row */
1631393bc97SHong Zhang   nlnk = n + 1;
1641393bc97SHong Zhang   ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
1651393bc97SHong Zhang 
16635baf27eSSatish Balay   ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr);
1671393bc97SHong Zhang 
1681393bc97SHong Zhang   /* initial FreeSpace size is f*(ai[n]+1) */
169d3d32019SBarry Smith   f = info->fill;
170a1a86e44SBarry Smith   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr);
1711393bc97SHong Zhang   current_space = free_space;
172289bc588SBarry Smith 
173289bc588SBarry Smith   for (i=0; i<n; i++) {
1741393bc97SHong Zhang     /* copy previous fill into linked list */
175289bc588SBarry Smith     nzi = 0;
1761393bc97SHong Zhang     nnz = ai[r[i]+1] - ai[r[i]];
177e32f2f54SBarry Smith     if (!nnz) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i);
1781393bc97SHong Zhang     ajtmp = aj + ai[r[i]];
179afcc9904SHong Zhang     ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
1801393bc97SHong Zhang     nzi += nlnk;
1811393bc97SHong Zhang 
1821393bc97SHong Zhang     /* add pivot rows into linked list */
1831393bc97SHong Zhang     row = lnk[n];
1841393bc97SHong Zhang     while (row < i) {
185d1170560SHong Zhang       nzbd    = bdiag[row] - bi[row] + 1; /* num of entries in the row with column index <= row */
186d1170560SHong Zhang       ajtmp   = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */
187d1170560SHong Zhang       ierr = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr);
1881393bc97SHong Zhang       nzi += nlnk;
1891393bc97SHong Zhang       row  = lnk[row];
190289bc588SBarry Smith     }
1911393bc97SHong Zhang     bi[i+1] = bi[i] + nzi;
1921393bc97SHong Zhang     im[i]   = nzi;
1931393bc97SHong Zhang 
1941393bc97SHong Zhang     /* mark bdiag */
1951393bc97SHong Zhang     nzbd = 0;
1961393bc97SHong Zhang     nnz  = nzi;
1971393bc97SHong Zhang     k    = lnk[n];
1981393bc97SHong Zhang     while (nnz-- && k < i){
1991393bc97SHong Zhang       nzbd++;
2001393bc97SHong Zhang       k = lnk[k];
2011393bc97SHong Zhang     }
2021393bc97SHong Zhang     bdiag[i] = bi[i] + nzbd;
2031393bc97SHong Zhang 
2041393bc97SHong Zhang     /* if free space is not available, make more free space */
2051393bc97SHong Zhang     if (current_space->local_remaining<nzi) {
2061393bc97SHong Zhang       nnz = (n - i)*nzi; /* estimated and max additional space needed */
207a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
2081393bc97SHong Zhang       reallocs++;
2091393bc97SHong Zhang     }
2101393bc97SHong Zhang 
2111393bc97SHong Zhang     /* copy data into free space, then initialize lnk */
2121393bc97SHong Zhang     ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
2131393bc97SHong Zhang     bi_ptr[i] = current_space->array;
2141393bc97SHong Zhang     current_space->array           += nzi;
2151393bc97SHong Zhang     current_space->local_used      += nzi;
2161393bc97SHong Zhang     current_space->local_remaining -= nzi;
217289bc588SBarry Smith   }
2186cf91177SBarry Smith #if defined(PETSC_USE_INFO)
219464e8d44SSatish Balay   if (ai[n] != 0) {
2201393bc97SHong Zhang     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
221ae15b995SBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr);
222ae15b995SBarry Smith     ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
223ae15b995SBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr);
224ae15b995SBarry Smith     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
22505bf559cSSatish Balay   } else {
226ae15b995SBarry Smith     ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr);
22705bf559cSSatish Balay   }
22863ba0a88SBarry Smith #endif
2292fb3ed76SBarry Smith 
230898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
231898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
2321987afe7SBarry Smith 
2331393bc97SHong Zhang   /* destroy list of free space and other temporary array(s) */
2341393bc97SHong Zhang   ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
235a1a86e44SBarry Smith   ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr);
2361393bc97SHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
23735baf27eSSatish Balay   ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr);
238289bc588SBarry Smith 
239289bc588SBarry Smith   /* put together the new matrix */
240719d5645SBarry Smith   ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
241719d5645SBarry Smith   ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr);
242719d5645SBarry Smith   b    = (Mat_SeqAIJ*)(B)->data;
243e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
244e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
2457c922b88SBarry Smith   b->singlemalloc = PETSC_FALSE;
2461393bc97SHong Zhang   ierr          = PetscMalloc((bi[n]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);
2471393bc97SHong Zhang   b->j          = bj;
2481393bc97SHong Zhang   b->i          = bi;
2491393bc97SHong Zhang   b->diag       = bdiag;
250416022c9SBarry Smith   b->ilen       = 0;
251416022c9SBarry Smith   b->imax       = 0;
252416022c9SBarry Smith   b->row        = isrow;
253416022c9SBarry Smith   b->col        = iscol;
254c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
255c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
25682bf6240SBarry Smith   b->icol       = isicol;
25787828ca2SBarry Smith   ierr          = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
2581393bc97SHong Zhang 
2591393bc97SHong Zhang   /* In b structure:  Free imax, ilen, old a, old j.  Allocate solve_work, new a, new j */
260719d5645SBarry Smith   ierr = PetscLogObjectMemory(B,(bi[n]-n)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
2611393bc97SHong Zhang   b->maxnz = b->nz = bi[n] ;
2627fd98bd6SLois Curfman McInnes 
263d5f3da31SBarry Smith   (B)->factortype            = MAT_FACTOR_LU;
264719d5645SBarry Smith   (B)->info.factor_mallocs   = reallocs;
265719d5645SBarry Smith   (B)->info.fill_ratio_given = f;
266703deb49SSatish Balay 
2679f7d0b68SHong Zhang   if (ai[n]) {
268719d5645SBarry Smith     (B)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]);
269eea2913aSSatish Balay   } else {
270719d5645SBarry Smith     (B)->info.fill_ratio_needed = 0.0;
271eea2913aSSatish Balay   }
272ad04f41aSHong Zhang   (B)->ops->lufactornumeric  = MatLUFactorNumeric_SeqAIJ_inplace;
273f2344125SBarry Smith   if (a->inode.size) {
274f2344125SBarry Smith     (B)->ops->lufactornumeric  = MatLUFactorNumeric_SeqAIJ_Inode_inplace;
275f2344125SBarry Smith   }
2763a40ed3dSBarry Smith   PetscFunctionReturn(0);
277289bc588SBarry Smith }
2781393bc97SHong Zhang 
279ce3d78c0SShri Abhyankar #undef __FUNCT__
28035233d99SShri Abhyankar #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ"
28135233d99SShri Abhyankar PetscErrorCode MatLUFactorSymbolic_SeqAIJ(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
282ce3d78c0SShri Abhyankar {
283ce3d78c0SShri Abhyankar   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
284ce3d78c0SShri Abhyankar   IS                 isicol;
285ce3d78c0SShri Abhyankar   PetscErrorCode     ierr;
286ce3d78c0SShri Abhyankar   const PetscInt     *r,*ic;
287ce3d78c0SShri Abhyankar   PetscInt           i,n=A->rmap->n,*ai=a->i,*aj=a->j;
288ce3d78c0SShri Abhyankar   PetscInt           *bi,*bj,*ajtmp;
289ce3d78c0SShri Abhyankar   PetscInt           *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im;
290ce3d78c0SShri Abhyankar   PetscReal          f;
291ce3d78c0SShri Abhyankar   PetscInt           nlnk,*lnk,k,**bi_ptr;
292ce3d78c0SShri Abhyankar   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
293ce3d78c0SShri Abhyankar   PetscBT            lnkbt;
294ce3d78c0SShri Abhyankar 
295ce3d78c0SShri Abhyankar   PetscFunctionBegin;
2963898ab1dSSatish Balay   /* Uncomment the oldatastruct part only while testing new data structure for MatSolve() */
297a4be958cSHong Zhang   /*
298*1a26ff17SHong Zhang   PetscBool          olddatastruct=PETSC_FALSE;
299acfcf0e5SJed Brown   ierr = PetscOptionsGetBool(PETSC_NULL,"-lu_old",&olddatastruct,PETSC_NULL);CHKERRQ(ierr);
300628f99d7SShri Abhyankar   if(olddatastruct){
301628f99d7SShri Abhyankar     ierr = MatLUFactorSymbolic_SeqAIJ_inplace(B,A,isrow,iscol,info);CHKERRQ(ierr);
302628f99d7SShri Abhyankar     PetscFunctionReturn(0);
303628f99d7SShri Abhyankar   }
304a4be958cSHong Zhang   */
305e32f2f54SBarry Smith   if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"matrix must be square");
306ce3d78c0SShri Abhyankar   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
307ce3d78c0SShri Abhyankar   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
308ce3d78c0SShri Abhyankar   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
309ce3d78c0SShri Abhyankar 
3101bfa06eaSJed Brown   /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */
3111bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
3121bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
313a6df321fSShri Abhyankar   bi[0] = bdiag[0] = 0;
3149b48462bSShri Abhyankar 
3159b48462bSShri Abhyankar   /* linked list for storing column indices of the active row */
3169b48462bSShri Abhyankar   nlnk = n + 1;
3179b48462bSShri Abhyankar   ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
3189b48462bSShri Abhyankar 
3199b48462bSShri Abhyankar   ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr);
3209b48462bSShri Abhyankar 
3219b48462bSShri Abhyankar   /* initial FreeSpace size is f*(ai[n]+1) */
3229b48462bSShri Abhyankar   f = info->fill;
3239b48462bSShri Abhyankar   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr);
3249b48462bSShri Abhyankar   current_space = free_space;
3259b48462bSShri Abhyankar 
3269b48462bSShri Abhyankar   for (i=0; i<n; i++) {
3279b48462bSShri Abhyankar     /* copy previous fill into linked list */
3289b48462bSShri Abhyankar     nzi = 0;
3299b48462bSShri Abhyankar     nnz = ai[r[i]+1] - ai[r[i]];
330e32f2f54SBarry 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);
3319b48462bSShri Abhyankar     ajtmp = aj + ai[r[i]];
3329b48462bSShri Abhyankar     ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
3339b48462bSShri Abhyankar     nzi += nlnk;
3349b48462bSShri Abhyankar 
3359b48462bSShri Abhyankar     /* add pivot rows into linked list */
3369b48462bSShri Abhyankar     row = lnk[n];
3379b48462bSShri Abhyankar     while (row < i){
3389b48462bSShri Abhyankar       nzbd  = bdiag[row] + 1; /* num of entries in the row with column index <= row */
3399b48462bSShri Abhyankar       ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */
3409b48462bSShri Abhyankar       ierr  = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr);
3419b48462bSShri Abhyankar       nzi  += nlnk;
3429b48462bSShri Abhyankar       row   = lnk[row];
3439b48462bSShri Abhyankar     }
3449b48462bSShri Abhyankar     bi[i+1] = bi[i] + nzi;
3459b48462bSShri Abhyankar     im[i]   = nzi;
3469b48462bSShri Abhyankar 
3479b48462bSShri Abhyankar     /* mark bdiag */
3489b48462bSShri Abhyankar     nzbd = 0;
3499b48462bSShri Abhyankar     nnz  = nzi;
3509b48462bSShri Abhyankar     k    = lnk[n];
3519b48462bSShri Abhyankar     while (nnz-- && k < i){
3529b48462bSShri Abhyankar       nzbd++;
3539b48462bSShri Abhyankar       k = lnk[k];
3549b48462bSShri Abhyankar     }
3559b48462bSShri Abhyankar     bdiag[i] = nzbd; /* note: bdiag[i] = nnzL as input for PetscFreeSpaceContiguous_LU() */
3569b48462bSShri Abhyankar 
3579b48462bSShri Abhyankar     /* if free space is not available, make more free space */
3589b48462bSShri Abhyankar     if (current_space->local_remaining<nzi) {
3599b48462bSShri Abhyankar       nnz = 2*(n - i)*nzi; /* estimated and max additional space needed */
3609b48462bSShri Abhyankar       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
3619b48462bSShri Abhyankar       reallocs++;
3629b48462bSShri Abhyankar     }
3639b48462bSShri Abhyankar 
3649b48462bSShri Abhyankar     /* copy data into free space, then initialize lnk */
3659b48462bSShri Abhyankar     ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
3669b48462bSShri Abhyankar     bi_ptr[i] = current_space->array;
3679b48462bSShri Abhyankar     current_space->array           += nzi;
3689b48462bSShri Abhyankar     current_space->local_used      += nzi;
3699b48462bSShri Abhyankar     current_space->local_remaining -= nzi;
3709b48462bSShri Abhyankar   }
3719b48462bSShri Abhyankar 
3729b48462bSShri Abhyankar   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
3739b48462bSShri Abhyankar   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
3749b48462bSShri Abhyankar 
3759263d837SHong Zhang   /*   copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */
3769b48462bSShri Abhyankar   ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
3772ce24eb6SHong Zhang   ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr);
3789b48462bSShri Abhyankar   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
3799b48462bSShri Abhyankar   ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr);
3809b48462bSShri Abhyankar 
3819b48462bSShri Abhyankar   /* put together the new matrix */
3829b48462bSShri Abhyankar   ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
3839b48462bSShri Abhyankar   ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr);
3849b48462bSShri Abhyankar   b    = (Mat_SeqAIJ*)(B)->data;
3859b48462bSShri Abhyankar   b->free_a       = PETSC_TRUE;
3869b48462bSShri Abhyankar   b->free_ij      = PETSC_TRUE;
3879b48462bSShri Abhyankar   b->singlemalloc = PETSC_FALSE;
3889b48462bSShri Abhyankar   ierr = PetscMalloc((bdiag[0]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);
3899b48462bSShri Abhyankar   b->j          = bj;
3909b48462bSShri Abhyankar   b->i          = bi;
3919b48462bSShri Abhyankar   b->diag       = bdiag;
3929b48462bSShri Abhyankar   b->ilen       = 0;
3939b48462bSShri Abhyankar   b->imax       = 0;
3949b48462bSShri Abhyankar   b->row        = isrow;
3959b48462bSShri Abhyankar   b->col        = iscol;
3969b48462bSShri Abhyankar   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
3979b48462bSShri Abhyankar   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
3989b48462bSShri Abhyankar   b->icol       = isicol;
3999b48462bSShri Abhyankar   ierr          = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
4009b48462bSShri Abhyankar 
4019b48462bSShri Abhyankar   /* In b structure:  Free imax, ilen, old a, old j.  Allocate solve_work, new a, new j */
4029b48462bSShri Abhyankar   ierr = PetscLogObjectMemory(B,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
4039b48462bSShri Abhyankar   b->maxnz = b->nz = bdiag[0]+1;
404d5f3da31SBarry Smith   B->factortype            = MAT_FACTOR_LU;
405f284f12aSHong Zhang   B->info.factor_mallocs   = reallocs;
406f284f12aSHong Zhang   B->info.fill_ratio_given = f;
4079b48462bSShri Abhyankar 
4089b48462bSShri Abhyankar   if (ai[n]) {
409f284f12aSHong Zhang     B->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]);
4109b48462bSShri Abhyankar   } else {
411f284f12aSHong Zhang     B->info.fill_ratio_needed = 0.0;
4129b48462bSShri Abhyankar   }
4139263d837SHong Zhang #if defined(PETSC_USE_INFO)
4149263d837SHong Zhang   if (ai[n] != 0) {
4159263d837SHong Zhang     PetscReal af = B->info.fill_ratio_needed;
4169263d837SHong Zhang     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr);
4179263d837SHong Zhang     ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
4189263d837SHong Zhang     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr);
4199263d837SHong Zhang     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
4209263d837SHong Zhang   } else {
4219263d837SHong Zhang     ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr);
4229263d837SHong Zhang   }
4239263d837SHong Zhang #endif
42435233d99SShri Abhyankar   B->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ;
425f2344125SBarry Smith   if (a->inode.size) {
426d3ac4fa3SBarry Smith     B->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode;
427d3ac4fa3SBarry Smith   }
4289b48462bSShri Abhyankar   PetscFunctionReturn(0);
4299b48462bSShri Abhyankar }
4309b48462bSShri Abhyankar 
4315b5bc046SBarry Smith /*
4325b5bc046SBarry Smith     Trouble in factorization, should we dump the original matrix?
4335b5bc046SBarry Smith */
4345b5bc046SBarry Smith #undef __FUNCT__
4355b5bc046SBarry Smith #define __FUNCT__ "MatFactorDumpMatrix"
4365b5bc046SBarry Smith PetscErrorCode MatFactorDumpMatrix(Mat A)
4375b5bc046SBarry Smith {
4385b5bc046SBarry Smith   PetscErrorCode ierr;
439ace3abfcSBarry Smith   PetscBool      flg = PETSC_FALSE;
4405b5bc046SBarry Smith 
4415b5bc046SBarry Smith   PetscFunctionBegin;
442acfcf0e5SJed Brown   ierr = PetscOptionsGetBool(PETSC_NULL,"-mat_factor_dump_on_error",&flg,PETSC_NULL);CHKERRQ(ierr);
4435b5bc046SBarry Smith   if (flg) {
4445b5bc046SBarry Smith     PetscViewer viewer;
4455b5bc046SBarry Smith     char        filename[PETSC_MAX_PATH_LEN];
4465b5bc046SBarry Smith 
4475b5bc046SBarry Smith     ierr = PetscSNPrintf(filename,PETSC_MAX_PATH_LEN,"matrix_factor_error.%d",PetscGlobalRank);CHKERRQ(ierr);
4487adad957SLisandro Dalcin     ierr = PetscViewerBinaryOpen(((PetscObject)A)->comm,filename,FILE_MODE_WRITE,&viewer);CHKERRQ(ierr);
4495b5bc046SBarry Smith     ierr = MatView(A,viewer);CHKERRQ(ierr);
4506bf464f9SBarry Smith     ierr = PetscViewerDestroy(&viewer);CHKERRQ(ierr);
4515b5bc046SBarry Smith   }
4525b5bc046SBarry Smith   PetscFunctionReturn(0);
4535b5bc046SBarry Smith }
4545b5bc046SBarry Smith 
455c9c72f8fSHong Zhang #undef __FUNCT__
45635233d99SShri Abhyankar #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ"
45735233d99SShri Abhyankar PetscErrorCode MatLUFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info)
458c9c72f8fSHong Zhang {
459c9c72f8fSHong Zhang   Mat              C=B;
460c9c72f8fSHong Zhang   Mat_SeqAIJ       *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data;
461c9c72f8fSHong Zhang   IS               isrow = b->row,isicol = b->icol;
462c9c72f8fSHong Zhang   PetscErrorCode   ierr;
463c9c72f8fSHong Zhang   const PetscInt   *r,*ic,*ics;
464bbd65245SShri Abhyankar   const PetscInt   n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j,*bdiag=b->diag;
465bbd65245SShri Abhyankar   PetscInt         i,j,k,nz,nzL,row,*pj;
466bbd65245SShri Abhyankar   const PetscInt   *ajtmp,*bjtmp;
467bbd65245SShri Abhyankar   MatScalar        *rtmp,*pc,multiplier,*pv;
468bbd65245SShri Abhyankar   const  MatScalar *aa=a->a,*v;
469ace3abfcSBarry Smith   PetscBool        row_identity,col_identity;
470d90ac83dSHong Zhang   FactorShiftCtx   sctx;
4714f81c4b7SBarry Smith   const PetscInt   *ddiag;
472d4ad06f7SHong Zhang   PetscReal        rs;
473d4ad06f7SHong Zhang   MatScalar        d;
474d4ad06f7SHong Zhang 
475c9c72f8fSHong Zhang   PetscFunctionBegin;
476d6e5152cSHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
477d90ac83dSHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
478d4ad06f7SHong Zhang 
479f4db908eSBarry Smith   if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
480d4ad06f7SHong Zhang     ddiag          = a->diag;
481d4ad06f7SHong Zhang     sctx.shift_top = info->zeropivot;
482d4ad06f7SHong Zhang     for (i=0; i<n; i++) {
483d4ad06f7SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
484d4ad06f7SHong Zhang       d  = (aa)[ddiag[i]];
485d4ad06f7SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
486d4ad06f7SHong Zhang       v  = aa+ai[i];
487d4ad06f7SHong Zhang       nz = ai[i+1] - ai[i];
488d4ad06f7SHong Zhang       for (j=0; j<nz; j++)
489d4ad06f7SHong Zhang 	rs += PetscAbsScalar(v[j]);
490d4ad06f7SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
491d4ad06f7SHong Zhang     }
492d4ad06f7SHong Zhang     sctx.shift_top   *= 1.1;
493d4ad06f7SHong Zhang     sctx.nshift_max   = 5;
494d4ad06f7SHong Zhang     sctx.shift_lo     = 0.;
495d4ad06f7SHong Zhang     sctx.shift_hi     = 1.;
496d4ad06f7SHong Zhang   }
497d4ad06f7SHong Zhang 
498c9c72f8fSHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
499c9c72f8fSHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
500c9c72f8fSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr);
501c9c72f8fSHong Zhang   ics  = ic;
502c9c72f8fSHong Zhang 
503d4ad06f7SHong Zhang   do {
50407b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
505c9c72f8fSHong Zhang     for (i=0; i<n; i++){
506c9c72f8fSHong Zhang       /* zero rtmp */
507c9c72f8fSHong Zhang       /* L part */
508c9c72f8fSHong Zhang       nz    = bi[i+1] - bi[i];
509c9c72f8fSHong Zhang       bjtmp = bj + bi[i];
510c9c72f8fSHong Zhang       for  (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
511c9c72f8fSHong Zhang 
512c9c72f8fSHong Zhang       /* U part */
51362fbd6afSShri Abhyankar       nz = bdiag[i]-bdiag[i+1];
51462fbd6afSShri Abhyankar       bjtmp = bj + bdiag[i+1]+1;
515813a1f2bSShri Abhyankar       for  (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
516813a1f2bSShri Abhyankar 
517813a1f2bSShri Abhyankar       /* load in initial (unfactored row) */
518813a1f2bSShri Abhyankar       nz    = ai[r[i]+1] - ai[r[i]];
519813a1f2bSShri Abhyankar       ajtmp = aj + ai[r[i]];
520813a1f2bSShri Abhyankar       v     = aa + ai[r[i]];
521813a1f2bSShri Abhyankar       for (j=0; j<nz; j++) {
522813a1f2bSShri Abhyankar         rtmp[ics[ajtmp[j]]] = v[j];
523813a1f2bSShri Abhyankar       }
524813a1f2bSShri Abhyankar       /* ZeropivotApply() */
52598a93161SHong Zhang       rtmp[i] += sctx.shift_amount;  /* shift the diagonal of the matrix */
526813a1f2bSShri Abhyankar 
527813a1f2bSShri Abhyankar       /* elimination */
528813a1f2bSShri Abhyankar       bjtmp = bj + bi[i];
529813a1f2bSShri Abhyankar       row   = *bjtmp++;
530813a1f2bSShri Abhyankar       nzL   = bi[i+1] - bi[i];
531f268cba8SShri Abhyankar       for(k=0; k < nzL;k++) {
532813a1f2bSShri Abhyankar         pc = rtmp + row;
533813a1f2bSShri Abhyankar         if (*pc != 0.0) {
534813a1f2bSShri Abhyankar           pv         = b->a + bdiag[row];
535813a1f2bSShri Abhyankar           multiplier = *pc * (*pv);
536813a1f2bSShri Abhyankar           *pc        = multiplier;
53762fbd6afSShri Abhyankar           pj = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */
53862fbd6afSShri Abhyankar 	  pv = b->a + bdiag[row+1]+1;
53962fbd6afSShri Abhyankar 	  nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */
540813a1f2bSShri Abhyankar           for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j];
541813a1f2bSShri Abhyankar           ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr);
542813a1f2bSShri Abhyankar         }
543f268cba8SShri Abhyankar         row = *bjtmp++;
544813a1f2bSShri Abhyankar       }
545813a1f2bSShri Abhyankar 
546813a1f2bSShri Abhyankar       /* finished row so stick it into b->a */
547813a1f2bSShri Abhyankar       rs = 0.0;
548813a1f2bSShri Abhyankar       /* L part */
549813a1f2bSShri Abhyankar       pv   = b->a + bi[i] ;
550813a1f2bSShri Abhyankar       pj   = b->j + bi[i] ;
551813a1f2bSShri Abhyankar       nz   = bi[i+1] - bi[i];
552813a1f2bSShri Abhyankar       for (j=0; j<nz; j++) {
553813a1f2bSShri Abhyankar         pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]);
554813a1f2bSShri Abhyankar       }
555813a1f2bSShri Abhyankar 
556813a1f2bSShri Abhyankar       /* U part */
55762fbd6afSShri Abhyankar       pv = b->a + bdiag[i+1]+1;
55862fbd6afSShri Abhyankar       pj = b->j + bdiag[i+1]+1;
55962fbd6afSShri Abhyankar       nz = bdiag[i] - bdiag[i+1]-1;
560813a1f2bSShri Abhyankar       for (j=0; j<nz; j++) {
561813a1f2bSShri Abhyankar         pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]);
562813a1f2bSShri Abhyankar       }
563813a1f2bSShri Abhyankar 
564813a1f2bSShri Abhyankar       sctx.rs  = rs;
565813a1f2bSShri Abhyankar       sctx.pv  = rtmp[i];
566d0660788SBarry Smith       ierr = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr);
56707b50cabSHong Zhang       if(sctx.newshift) break; /* break for-loop */
56807b50cabSHong Zhang       rtmp[i] = sctx.pv; /* sctx.pv might be updated in the case of MAT_SHIFT_INBLOCKS */
569813a1f2bSShri Abhyankar 
570813a1f2bSShri Abhyankar       /* Mark diagonal and invert diagonal for simplier triangular solves */
571813a1f2bSShri Abhyankar       pv  = b->a + bdiag[i];
572813a1f2bSShri Abhyankar       *pv = 1.0/rtmp[i];
573813a1f2bSShri Abhyankar 
574813a1f2bSShri Abhyankar     } /* endof for (i=0; i<n; i++){ */
575813a1f2bSShri Abhyankar 
5768ff23777SHong Zhang     /* MatPivotRefine() */
57707b50cabSHong Zhang     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max){
578813a1f2bSShri Abhyankar       /*
579813a1f2bSShri Abhyankar        * if no shift in this attempt & shifting & started shifting & can refine,
580813a1f2bSShri Abhyankar        * then try lower shift
581813a1f2bSShri Abhyankar        */
582813a1f2bSShri Abhyankar       sctx.shift_hi       = sctx.shift_fraction;
583813a1f2bSShri Abhyankar       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
584813a1f2bSShri Abhyankar       sctx.shift_amount   = sctx.shift_fraction * sctx.shift_top;
58507b50cabSHong Zhang       sctx.newshift       = PETSC_TRUE;
586813a1f2bSShri Abhyankar       sctx.nshift++;
587813a1f2bSShri Abhyankar     }
58807b50cabSHong Zhang   } while (sctx.newshift);
589813a1f2bSShri Abhyankar 
590813a1f2bSShri Abhyankar   ierr = PetscFree(rtmp);CHKERRQ(ierr);
591813a1f2bSShri Abhyankar   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
592813a1f2bSShri Abhyankar   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
593d3ac4fa3SBarry Smith 
594813a1f2bSShri Abhyankar   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
595813a1f2bSShri Abhyankar   ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr);
596813a1f2bSShri Abhyankar   if (row_identity && col_identity) {
59735233d99SShri Abhyankar     C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering;
598813a1f2bSShri Abhyankar   } else {
59935233d99SShri Abhyankar     C->ops->solve = MatSolve_SeqAIJ;
600813a1f2bSShri Abhyankar   }
60135233d99SShri Abhyankar   C->ops->solveadd           = MatSolveAdd_SeqAIJ;
60235233d99SShri Abhyankar   C->ops->solvetranspose     = MatSolveTranspose_SeqAIJ;
60335233d99SShri Abhyankar   C->ops->solvetransposeadd  = MatSolveTransposeAdd_SeqAIJ;
60435233d99SShri Abhyankar   C->ops->matsolve           = MatMatSolve_SeqAIJ;
605813a1f2bSShri Abhyankar   C->assembled    = PETSC_TRUE;
606813a1f2bSShri Abhyankar   C->preallocated = PETSC_TRUE;
607813a1f2bSShri Abhyankar   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
608813a1f2bSShri Abhyankar 
60914d2772eSHong Zhang   /* MatShiftView(A,info,&sctx) */
610813a1f2bSShri Abhyankar   if (sctx.nshift){
611f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
612813a1f2bSShri Abhyankar       ierr = PetscInfo4(A,"number of shift_pd tries %D, shift_amount %G, diagonal shifted up by %e fraction top_value %e\n",sctx.nshift,sctx.shift_amount,sctx.shift_fraction,sctx.shift_top);CHKERRQ(ierr);
613f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
614813a1f2bSShri Abhyankar       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
615f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_INBLOCKS){
616915743fcSHong Zhang       ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %G\n",sctx.nshift,info->shiftamount);CHKERRQ(ierr);
617813a1f2bSShri Abhyankar     }
618813a1f2bSShri Abhyankar   }
619d3ac4fa3SBarry Smith   ierr = Mat_CheckInode_FactorLU(C,PETSC_FALSE);CHKERRQ(ierr);
620813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
621813a1f2bSShri Abhyankar }
622813a1f2bSShri Abhyankar 
6234a2ae208SSatish Balay #undef __FUNCT__
624ad04f41aSHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_inplace"
625ad04f41aSHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ_inplace(Mat B,Mat A,const MatFactorInfo *info)
626289bc588SBarry Smith {
627719d5645SBarry Smith   Mat             C=B;
628416022c9SBarry Smith   Mat_SeqAIJ      *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data;
62982bf6240SBarry Smith   IS              isrow = b->row,isicol = b->icol;
6306849ba73SBarry Smith   PetscErrorCode  ierr;
6315d0c19d7SBarry Smith   const PetscInt   *r,*ic,*ics;
632d278edc7SBarry Smith   PetscInt        nz,row,i,j,n=A->rmap->n,diag;
633d278edc7SBarry Smith   const PetscInt  *ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j;
634d278edc7SBarry Smith   const PetscInt  *ajtmp,*bjtmp,*diag_offset = b->diag,*pj;
635d278edc7SBarry Smith   MatScalar       *pv,*rtmp,*pc,multiplier,d;
636d278edc7SBarry Smith   const MatScalar *v,*aa=a->a;
63733f9893dSHong Zhang   PetscReal       rs=0.0;
638d90ac83dSHong Zhang   FactorShiftCtx  sctx;
6394f81c4b7SBarry Smith   const PetscInt  *ddiag;
640ace3abfcSBarry Smith   PetscBool       row_identity, col_identity;
641289bc588SBarry Smith 
6423a40ed3dSBarry Smith   PetscFunctionBegin;
643504a3fadSHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
644504a3fadSHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
645ada7143aSSatish Balay 
646f4db908eSBarry Smith   if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
64742898933SHong Zhang     ddiag          = a->diag;
6489f7d0b68SHong Zhang     sctx.shift_top = info->zeropivot;
6496cc28720Svictorle     for (i=0; i<n; i++) {
6509f95998fSHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
6519f7d0b68SHong Zhang       d  = (aa)[ddiag[i]];
6521a890ab1SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
6539f7d0b68SHong Zhang       v  = aa+ai[i];
6549f7d0b68SHong Zhang       nz = ai[i+1] - ai[i];
655e24cfe64SHong Zhang       for (j=0; j<nz; j++)
6561a890ab1SHong Zhang 	rs += PetscAbsScalar(v[j]);
6571a890ab1SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
6586cc28720Svictorle     }
659118f5789SBarry Smith     sctx.shift_top   *= 1.1;
660d2276718SHong Zhang     sctx.nshift_max   = 5;
661d2276718SHong Zhang     sctx.shift_lo     = 0.;
662d2276718SHong Zhang     sctx.shift_hi     = 1.;
663a0a356efSHong Zhang   }
664a0a356efSHong Zhang 
665504a3fadSHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
666504a3fadSHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
667504a3fadSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr);
668504a3fadSHong Zhang   ics  = ic;
669504a3fadSHong Zhang 
670085256b3SBarry Smith   do {
67107b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
672289bc588SBarry Smith     for (i=0; i<n; i++){
673b3bf805bSHong Zhang       nz    = bi[i+1] - bi[i];
674b3bf805bSHong Zhang       bjtmp = bj + bi[i];
6759f7d0b68SHong Zhang       for  (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
676289bc588SBarry Smith 
677289bc588SBarry Smith       /* load in initial (unfactored row) */
6789f7d0b68SHong Zhang       nz    = ai[r[i]+1] - ai[r[i]];
6799f7d0b68SHong Zhang       ajtmp = aj + ai[r[i]];
6809f7d0b68SHong Zhang       v     = aa + ai[r[i]];
681085256b3SBarry Smith       for (j=0; j<nz; j++) {
682b3bf805bSHong Zhang         rtmp[ics[ajtmp[j]]] = v[j];
683085256b3SBarry Smith       }
684d2276718SHong Zhang       rtmp[ics[r[i]]] += sctx.shift_amount; /* shift the diagonal of the matrix */
685289bc588SBarry Smith 
686b3bf805bSHong Zhang       row = *bjtmp++;
687f2582111SSatish Balay       while  (row < i) {
6888c37ef55SBarry Smith         pc = rtmp + row;
6898c37ef55SBarry Smith         if (*pc != 0.0) {
690010a20caSHong Zhang           pv         = b->a + diag_offset[row];
691010a20caSHong Zhang           pj         = b->j + diag_offset[row] + 1;
69235aab85fSBarry Smith           multiplier = *pc / *pv++;
6938c37ef55SBarry Smith           *pc        = multiplier;
694b3bf805bSHong Zhang           nz         = bi[row+1] - diag_offset[row] - 1;
6959f7d0b68SHong Zhang           for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j];
696dc0b31edSSatish Balay           ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr);
697289bc588SBarry Smith         }
698b3bf805bSHong Zhang         row = *bjtmp++;
699289bc588SBarry Smith       }
700416022c9SBarry Smith       /* finished row so stick it into b->a */
701b3bf805bSHong Zhang       pv   = b->a + bi[i] ;
702b3bf805bSHong Zhang       pj   = b->j + bi[i] ;
703b3bf805bSHong Zhang       nz   = bi[i+1] - bi[i];
704b3bf805bSHong Zhang       diag = diag_offset[i] - bi[i];
705e57ff13aSBarry Smith       rs   = 0.0;
706d3d32019SBarry Smith       for (j=0; j<nz; j++) {
7079f7d0b68SHong Zhang         pv[j] = rtmp[pj[j]];
7089f7d0b68SHong Zhang         rs   += PetscAbsScalar(pv[j]);
709d3d32019SBarry Smith       }
710e57ff13aSBarry Smith       rs   -= PetscAbsScalar(pv[diag]);
711d2276718SHong Zhang 
712d2276718SHong Zhang       sctx.rs  = rs;
713d2276718SHong Zhang       sctx.pv  = pv[diag];
714d0660788SBarry Smith       ierr = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr);
71507b50cabSHong Zhang       if (sctx.newshift) break;
716504a3fadSHong Zhang       pv[diag] = sctx.pv;
71735aab85fSBarry Smith     }
718d2276718SHong Zhang 
71907b50cabSHong Zhang     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) {
7206cc28720Svictorle       /*
7216c037d1bSvictorle        * if no shift in this attempt & shifting & started shifting & can refine,
7226cc28720Svictorle        * then try lower shift
7236cc28720Svictorle        */
7240481f469SBarry Smith       sctx.shift_hi       = sctx.shift_fraction;
7250481f469SBarry Smith       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
7260481f469SBarry Smith       sctx.shift_amount   = sctx.shift_fraction * sctx.shift_top;
72707b50cabSHong Zhang       sctx.newshift       = PETSC_TRUE;
728d2276718SHong Zhang       sctx.nshift++;
7296cc28720Svictorle     }
73007b50cabSHong Zhang   } while (sctx.newshift);
7310f11f9aeSSatish Balay 
73235aab85fSBarry Smith   /* invert diagonal entries for simplier triangular solves */
73335aab85fSBarry Smith   for (i=0; i<n; i++) {
734010a20caSHong Zhang     b->a[diag_offset[i]] = 1.0/b->a[diag_offset[i]];
73535aab85fSBarry Smith   }
736606d414cSSatish Balay   ierr = PetscFree(rtmp);CHKERRQ(ierr);
73778b31e54SBarry Smith   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
73878b31e54SBarry Smith   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
739d3ac4fa3SBarry Smith 
7408d8c7c43SBarry Smith   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
7418d8c7c43SBarry Smith   ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr);
7428d8c7c43SBarry Smith   if (row_identity && col_identity) {
743ad04f41aSHong Zhang     C->ops->solve   = MatSolve_SeqAIJ_NaturalOrdering_inplace;
7448d8c7c43SBarry Smith   } else {
745ad04f41aSHong Zhang     C->ops->solve   = MatSolve_SeqAIJ_inplace;
746db4efbfdSBarry Smith   }
747ad04f41aSHong Zhang   C->ops->solveadd           = MatSolveAdd_SeqAIJ_inplace;
748ad04f41aSHong Zhang   C->ops->solvetranspose     = MatSolveTranspose_SeqAIJ_inplace;
749ad04f41aSHong Zhang   C->ops->solvetransposeadd  = MatSolveTransposeAdd_SeqAIJ_inplace;
750ad04f41aSHong Zhang   C->ops->matsolve           = MatMatSolve_SeqAIJ_inplace;
751c456f294SBarry Smith   C->assembled    = PETSC_TRUE;
7525c9eb25fSBarry Smith   C->preallocated = PETSC_TRUE;
753d0f46423SBarry Smith   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
754d2276718SHong Zhang   if (sctx.nshift){
755f4db908eSBarry Smith      if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
7560481f469SBarry 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);
757f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
758e738e422SBarry Smith       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
7596cc28720Svictorle     }
7600697b6caSHong Zhang   }
761d3ac4fa3SBarry Smith   (C)->ops->solve            = MatSolve_SeqAIJ_inplace;
762d3ac4fa3SBarry Smith   (C)->ops->solvetranspose   = MatSolveTranspose_SeqAIJ_inplace;
763d3ac4fa3SBarry Smith   ierr = Mat_CheckInode(C,PETSC_FALSE);CHKERRQ(ierr);
7643a40ed3dSBarry Smith   PetscFunctionReturn(0);
765289bc588SBarry Smith }
7660889b5dcSvictorle 
767137fb511SHong Zhang /*
768137fb511SHong Zhang    This routine implements inplace ILU(0) with row or/and column permutations.
769137fb511SHong Zhang    Input:
770137fb511SHong Zhang      A - original matrix
771137fb511SHong Zhang    Output;
772137fb511SHong Zhang      A - a->i (rowptr) is same as original rowptr, but factored i-the row is stored in rowperm[i]
773137fb511SHong Zhang          a->j (col index) is permuted by the inverse of colperm, then sorted
774137fb511SHong Zhang          a->a reordered accordingly with a->j
775137fb511SHong Zhang          a->diag (ptr to diagonal elements) is updated.
776137fb511SHong Zhang */
777137fb511SHong Zhang #undef __FUNCT__
778137fb511SHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_InplaceWithPerm"
7790481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_SeqAIJ_InplaceWithPerm(Mat B,Mat A,const MatFactorInfo *info)
780137fb511SHong Zhang {
781b051339dSHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
782b051339dSHong Zhang   IS             isrow = a->row,isicol = a->icol;
783137fb511SHong Zhang   PetscErrorCode ierr;
7845d0c19d7SBarry Smith   const PetscInt *r,*ic,*ics;
7855d0c19d7SBarry Smith   PetscInt       i,j,n=A->rmap->n,*ai=a->i,*aj=a->j;
7865d0c19d7SBarry Smith   PetscInt       *ajtmp,nz,row;
787b051339dSHong Zhang   PetscInt       *diag = a->diag,nbdiag,*pj;
788a77337e4SBarry Smith   PetscScalar    *rtmp,*pc,multiplier,d;
789504a3fadSHong Zhang   MatScalar      *pv,*v;
790137fb511SHong Zhang   PetscReal      rs;
791d90ac83dSHong Zhang   FactorShiftCtx sctx;
792504a3fadSHong Zhang   const  MatScalar *aa=a->a,*vtmp;
793137fb511SHong Zhang 
794137fb511SHong Zhang   PetscFunctionBegin;
795e32f2f54SBarry Smith   if (A != B) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"input and output matrix must have same address");
796504a3fadSHong Zhang 
797504a3fadSHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
798504a3fadSHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
799504a3fadSHong Zhang 
800504a3fadSHong Zhang   if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
801504a3fadSHong Zhang     const PetscInt   *ddiag = a->diag;
802504a3fadSHong Zhang     sctx.shift_top = info->zeropivot;
803504a3fadSHong Zhang     for (i=0; i<n; i++) {
804504a3fadSHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
805504a3fadSHong Zhang       d  = (aa)[ddiag[i]];
806504a3fadSHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
807504a3fadSHong Zhang       vtmp  = aa+ai[i];
808504a3fadSHong Zhang       nz = ai[i+1] - ai[i];
809504a3fadSHong Zhang       for (j=0; j<nz; j++)
810504a3fadSHong Zhang 	rs += PetscAbsScalar(vtmp[j]);
811504a3fadSHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
812504a3fadSHong Zhang     }
813504a3fadSHong Zhang     sctx.shift_top   *= 1.1;
814504a3fadSHong Zhang     sctx.nshift_max   = 5;
815504a3fadSHong Zhang     sctx.shift_lo     = 0.;
816504a3fadSHong Zhang     sctx.shift_hi     = 1.;
817504a3fadSHong Zhang   }
818504a3fadSHong Zhang 
819137fb511SHong Zhang   ierr  = ISGetIndices(isrow,&r);CHKERRQ(ierr);
820137fb511SHong Zhang   ierr  = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
821137fb511SHong Zhang   ierr  = PetscMalloc((n+1)*sizeof(PetscScalar),&rtmp);CHKERRQ(ierr);
822137fb511SHong Zhang   ierr  = PetscMemzero(rtmp,(n+1)*sizeof(PetscScalar));CHKERRQ(ierr);
823b051339dSHong Zhang   ics = ic;
824137fb511SHong Zhang 
825504a3fadSHong Zhang #if defined(MV)
82675567043SBarry Smith   sctx.shift_top      = 0.;
827e60cf9a0SBarry Smith   sctx.nshift_max     = 0;
82875567043SBarry Smith   sctx.shift_lo       = 0.;
82975567043SBarry Smith   sctx.shift_hi       = 0.;
83075567043SBarry Smith   sctx.shift_fraction = 0.;
831137fb511SHong Zhang 
832f4db908eSBarry Smith   if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
83375567043SBarry Smith     sctx.shift_top = 0.;
834137fb511SHong Zhang     for (i=0; i<n; i++) {
835137fb511SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
836b051339dSHong Zhang       d  = (a->a)[diag[i]];
837137fb511SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
838b051339dSHong Zhang       v  = a->a+ai[i];
839b051339dSHong Zhang       nz = ai[i+1] - ai[i];
840137fb511SHong Zhang       for (j=0; j<nz; j++)
841137fb511SHong Zhang 	rs += PetscAbsScalar(v[j]);
842137fb511SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
843137fb511SHong Zhang     }
844137fb511SHong Zhang     if (sctx.shift_top < info->zeropivot) sctx.shift_top = info->zeropivot;
845137fb511SHong Zhang     sctx.shift_top    *= 1.1;
846137fb511SHong Zhang     sctx.nshift_max   = 5;
847137fb511SHong Zhang     sctx.shift_lo     = 0.;
848137fb511SHong Zhang     sctx.shift_hi     = 1.;
849137fb511SHong Zhang   }
850137fb511SHong Zhang 
85175567043SBarry Smith   sctx.shift_amount = 0.;
852137fb511SHong Zhang   sctx.nshift       = 0;
853504a3fadSHong Zhang #endif
854504a3fadSHong Zhang 
855137fb511SHong Zhang   do {
85607b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
857137fb511SHong Zhang     for (i=0; i<n; i++){
858b051339dSHong Zhang       /* load in initial unfactored row */
859b051339dSHong Zhang       nz    = ai[r[i]+1] - ai[r[i]];
860b051339dSHong Zhang       ajtmp = aj + ai[r[i]];
861b051339dSHong Zhang       v     = a->a + ai[r[i]];
862137fb511SHong Zhang       /* sort permuted ajtmp and values v accordingly */
863b051339dSHong Zhang       for (j=0; j<nz; j++) ajtmp[j] = ics[ajtmp[j]];
864137fb511SHong Zhang       ierr = PetscSortIntWithScalarArray(nz,ajtmp,v);CHKERRQ(ierr);
865137fb511SHong Zhang 
866b051339dSHong Zhang       diag[r[i]] = ai[r[i]];
867137fb511SHong Zhang       for (j=0; j<nz; j++) {
868137fb511SHong Zhang         rtmp[ajtmp[j]] = v[j];
869b051339dSHong Zhang         if (ajtmp[j] < i) diag[r[i]]++; /* update a->diag */
870137fb511SHong Zhang       }
871137fb511SHong Zhang       rtmp[r[i]] += sctx.shift_amount; /* shift the diagonal of the matrix */
872137fb511SHong Zhang 
873b051339dSHong Zhang       row = *ajtmp++;
874137fb511SHong Zhang       while  (row < i) {
875137fb511SHong Zhang         pc = rtmp + row;
876137fb511SHong Zhang         if (*pc != 0.0) {
877b051339dSHong Zhang           pv         = a->a + diag[r[row]];
878b051339dSHong Zhang           pj         = aj + diag[r[row]] + 1;
879137fb511SHong Zhang 
880137fb511SHong Zhang           multiplier = *pc / *pv++;
881137fb511SHong Zhang           *pc        = multiplier;
882b051339dSHong Zhang           nz         = ai[r[row]+1] - diag[r[row]] - 1;
883b051339dSHong Zhang           for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j];
884dc0b31edSSatish Balay           ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr);
885137fb511SHong Zhang         }
886b051339dSHong Zhang         row = *ajtmp++;
887137fb511SHong Zhang       }
888b051339dSHong Zhang       /* finished row so overwrite it onto a->a */
889b051339dSHong Zhang       pv   = a->a + ai[r[i]] ;
890b051339dSHong Zhang       pj   = aj + ai[r[i]] ;
891b051339dSHong Zhang       nz   = ai[r[i]+1] - ai[r[i]];
892b051339dSHong Zhang       nbdiag = diag[r[i]] - ai[r[i]]; /* num of entries before the diagonal */
893137fb511SHong Zhang 
894137fb511SHong Zhang       rs   = 0.0;
895137fb511SHong Zhang       for (j=0; j<nz; j++) {
896b051339dSHong Zhang         pv[j] = rtmp[pj[j]];
897b051339dSHong Zhang         if (j != nbdiag) rs += PetscAbsScalar(pv[j]);
898137fb511SHong Zhang       }
899137fb511SHong Zhang 
900137fb511SHong Zhang       sctx.rs  = rs;
901b051339dSHong Zhang       sctx.pv  = pv[nbdiag];
902d0660788SBarry Smith       ierr = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr);
90307b50cabSHong Zhang       if (sctx.newshift) break;
904504a3fadSHong Zhang       pv[nbdiag] = sctx.pv;
905137fb511SHong Zhang     }
906137fb511SHong Zhang 
90707b50cabSHong Zhang     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) {
908137fb511SHong Zhang       /*
909137fb511SHong Zhang        * if no shift in this attempt & shifting & started shifting & can refine,
910137fb511SHong Zhang        * then try lower shift
911137fb511SHong Zhang        */
9120481f469SBarry Smith       sctx.shift_hi        = sctx.shift_fraction;
9130481f469SBarry Smith       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
9140481f469SBarry Smith       sctx.shift_amount    = sctx.shift_fraction * sctx.shift_top;
91507b50cabSHong Zhang       sctx.newshift         = PETSC_TRUE;
916137fb511SHong Zhang       sctx.nshift++;
917137fb511SHong Zhang     }
91807b50cabSHong Zhang   } while (sctx.newshift);
919137fb511SHong Zhang 
920137fb511SHong Zhang   /* invert diagonal entries for simplier triangular solves */
921137fb511SHong Zhang   for (i=0; i<n; i++) {
922b051339dSHong Zhang     a->a[diag[r[i]]] = 1.0/a->a[diag[r[i]]];
923137fb511SHong Zhang   }
924137fb511SHong Zhang 
925137fb511SHong Zhang   ierr = PetscFree(rtmp);CHKERRQ(ierr);
926137fb511SHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
927137fb511SHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
928b051339dSHong Zhang   A->ops->solve             = MatSolve_SeqAIJ_InplaceWithPerm;
929ad04f41aSHong Zhang   A->ops->solveadd          = MatSolveAdd_SeqAIJ_inplace;
930ad04f41aSHong Zhang   A->ops->solvetranspose    = MatSolveTranspose_SeqAIJ_inplace;
931ad04f41aSHong Zhang   A->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ_inplace;
932b051339dSHong Zhang   A->assembled = PETSC_TRUE;
9335c9eb25fSBarry Smith   A->preallocated = PETSC_TRUE;
934d0f46423SBarry Smith   ierr = PetscLogFlops(A->cmap->n);CHKERRQ(ierr);
935137fb511SHong Zhang   if (sctx.nshift){
936f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
9370481f469SBarry 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);
938f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
939e738e422SBarry Smith       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
940137fb511SHong Zhang     }
941137fb511SHong Zhang   }
942137fb511SHong Zhang   PetscFunctionReturn(0);
943137fb511SHong Zhang }
944137fb511SHong Zhang 
9450a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
9464a2ae208SSatish Balay #undef __FUNCT__
9474a2ae208SSatish Balay #define __FUNCT__ "MatLUFactor_SeqAIJ"
9480481f469SBarry Smith PetscErrorCode MatLUFactor_SeqAIJ(Mat A,IS row,IS col,const MatFactorInfo *info)
949da3a660dSBarry Smith {
950dfbe8321SBarry Smith   PetscErrorCode ierr;
951416022c9SBarry Smith   Mat            C;
952416022c9SBarry Smith 
9533a40ed3dSBarry Smith   PetscFunctionBegin;
9542692d6eeSBarry Smith   ierr = MatGetFactor(A,MATSOLVERPETSC,MAT_FACTOR_LU,&C);CHKERRQ(ierr);
955719d5645SBarry Smith   ierr = MatLUFactorSymbolic(C,A,row,col,info);CHKERRQ(ierr);
956719d5645SBarry Smith   ierr = MatLUFactorNumeric(C,A,info);CHKERRQ(ierr);
957db4efbfdSBarry Smith   A->ops->solve            = C->ops->solve;
958db4efbfdSBarry Smith   A->ops->solvetranspose   = C->ops->solvetranspose;
959eb6b5d47SBarry Smith   ierr = MatHeaderMerge(A,C);CHKERRQ(ierr);
96052e6d16bSBarry Smith   ierr = PetscLogObjectParent(A,((Mat_SeqAIJ*)(A->data))->icol);CHKERRQ(ierr);
9613a40ed3dSBarry Smith   PetscFunctionReturn(0);
962da3a660dSBarry Smith }
9630a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
9641d098868SHong Zhang 
965889a8dedSBarry Smith 
9664a2ae208SSatish Balay #undef __FUNCT__
967ad04f41aSHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_inplace"
968ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_inplace(Mat A,Vec bb,Vec xx)
9698c37ef55SBarry Smith {
970416022c9SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
971416022c9SBarry Smith   IS                iscol = a->col,isrow = a->row;
9726849ba73SBarry Smith   PetscErrorCode    ierr;
9735d0c19d7SBarry Smith   PetscInt          i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
9745d0c19d7SBarry Smith   PetscInt          nz;
9755d0c19d7SBarry Smith   const PetscInt    *rout,*cout,*r,*c;
976dd6ea824SBarry Smith   PetscScalar       *x,*tmp,*tmps,sum;
977d9fead3dSBarry Smith   const PetscScalar *b;
978dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
9798c37ef55SBarry Smith 
9803a40ed3dSBarry Smith   PetscFunctionBegin;
9813a40ed3dSBarry Smith   if (!n) PetscFunctionReturn(0);
98291bf9adeSBarry Smith 
9833649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
9841ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
985416022c9SBarry Smith   tmp  = a->solve_work;
9868c37ef55SBarry Smith 
9873b2fbd54SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
9883b2fbd54SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
9898c37ef55SBarry Smith 
9908c37ef55SBarry Smith   /* forward solve the lower triangular */
9918c37ef55SBarry Smith   tmp[0] = b[*r++];
992010a20caSHong Zhang   tmps   = tmp;
9938c37ef55SBarry Smith   for (i=1; i<n; i++) {
994010a20caSHong Zhang     v   = aa + ai[i] ;
995010a20caSHong Zhang     vi  = aj + ai[i] ;
99653e7425aSBarry Smith     nz  = a->diag[i] - ai[i];
99753e7425aSBarry Smith     sum = b[*r++];
998003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
9998c37ef55SBarry Smith     tmp[i] = sum;
10008c37ef55SBarry Smith   }
10018c37ef55SBarry Smith 
10028c37ef55SBarry Smith   /* backward solve the upper triangular */
10038c37ef55SBarry Smith   for (i=n-1; i>=0; i--){
1004010a20caSHong Zhang     v   = aa + a->diag[i] + 1;
1005010a20caSHong Zhang     vi  = aj + a->diag[i] + 1;
1006416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
10078c37ef55SBarry Smith     sum = tmp[i];
1008003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
1009010a20caSHong Zhang     x[*c--] = tmp[i] = sum*aa[a->diag[i]];
10108c37ef55SBarry Smith   }
10118c37ef55SBarry Smith 
10123b2fbd54SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
10133b2fbd54SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
10143649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
10151ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1016dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
10173a40ed3dSBarry Smith   PetscFunctionReturn(0);
10188c37ef55SBarry Smith }
1019026e39d0SSatish Balay 
10202bebee5dSHong Zhang #undef __FUNCT__
1021ad04f41aSHong Zhang #define __FUNCT__ "MatMatSolve_SeqAIJ_inplace"
1022ad04f41aSHong Zhang PetscErrorCode MatMatSolve_SeqAIJ_inplace(Mat A,Mat B,Mat X)
10232bebee5dSHong Zhang {
10242bebee5dSHong Zhang   Mat_SeqAIJ      *a = (Mat_SeqAIJ*)A->data;
10252bebee5dSHong Zhang   IS              iscol = a->col,isrow = a->row;
10262bebee5dSHong Zhang   PetscErrorCode  ierr;
10275d0c19d7SBarry Smith   PetscInt        i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
10285d0c19d7SBarry Smith   PetscInt        nz,neq;
10295d0c19d7SBarry Smith   const PetscInt  *rout,*cout,*r,*c;
1030dd6ea824SBarry Smith   PetscScalar     *x,*b,*tmp,*tmps,sum;
1031dd6ea824SBarry Smith   const MatScalar *aa = a->a,*v;
1032ace3abfcSBarry Smith   PetscBool       bisdense,xisdense;
10332bebee5dSHong Zhang 
10342bebee5dSHong Zhang   PetscFunctionBegin;
10352bebee5dSHong Zhang   if (!n) PetscFunctionReturn(0);
10362bebee5dSHong Zhang 
1037db4efbfdSBarry Smith   ierr = PetscTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr);
1038e32f2f54SBarry Smith   if (!bisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix");
1039db4efbfdSBarry Smith   ierr = PetscTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr);
1040e32f2f54SBarry Smith   if (!xisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix");
1041db4efbfdSBarry Smith 
10422bebee5dSHong Zhang   ierr = MatGetArray(B,&b);CHKERRQ(ierr);
10432bebee5dSHong Zhang   ierr = MatGetArray(X,&x);CHKERRQ(ierr);
10442bebee5dSHong Zhang 
10452bebee5dSHong Zhang   tmp  = a->solve_work;
10462bebee5dSHong Zhang   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
10472bebee5dSHong Zhang   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
10482bebee5dSHong Zhang 
1049a36b22ccSSatish Balay   for (neq=0; neq<B->cmap->n; neq++){
10502bebee5dSHong Zhang     /* forward solve the lower triangular */
10512bebee5dSHong Zhang     tmp[0] = b[r[0]];
10522bebee5dSHong Zhang     tmps   = tmp;
10532bebee5dSHong Zhang     for (i=1; i<n; i++) {
10542bebee5dSHong Zhang       v   = aa + ai[i] ;
10552bebee5dSHong Zhang       vi  = aj + ai[i] ;
10562bebee5dSHong Zhang       nz  = a->diag[i] - ai[i];
10572bebee5dSHong Zhang       sum = b[r[i]];
1058003131ecSBarry Smith       PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
10592bebee5dSHong Zhang       tmp[i] = sum;
10602bebee5dSHong Zhang     }
10612bebee5dSHong Zhang     /* backward solve the upper triangular */
10622bebee5dSHong Zhang     for (i=n-1; i>=0; i--){
10632bebee5dSHong Zhang       v   = aa + a->diag[i] + 1;
10642bebee5dSHong Zhang       vi  = aj + a->diag[i] + 1;
10652bebee5dSHong Zhang       nz  = ai[i+1] - a->diag[i] - 1;
10662bebee5dSHong Zhang       sum = tmp[i];
1067003131ecSBarry Smith       PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
10682bebee5dSHong Zhang       x[c[i]] = tmp[i] = sum*aa[a->diag[i]];
10692bebee5dSHong Zhang     }
10702bebee5dSHong Zhang 
10712bebee5dSHong Zhang     b += n;
10722bebee5dSHong Zhang     x += n;
10732bebee5dSHong Zhang   }
10742bebee5dSHong Zhang   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
10752bebee5dSHong Zhang   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
10762bebee5dSHong Zhang   ierr = MatRestoreArray(B,&b);CHKERRQ(ierr);
10772bebee5dSHong Zhang   ierr = MatRestoreArray(X,&x);CHKERRQ(ierr);
1078dc0b31edSSatish Balay   ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr);
10792bebee5dSHong Zhang   PetscFunctionReturn(0);
10802bebee5dSHong Zhang }
10812bebee5dSHong Zhang 
1082137fb511SHong Zhang #undef __FUNCT__
108335233d99SShri Abhyankar #define __FUNCT__ "MatMatSolve_SeqAIJ"
108435233d99SShri Abhyankar PetscErrorCode MatMatSolve_SeqAIJ(Mat A,Mat B,Mat X)
10859bd0847aSShri Abhyankar {
10869bd0847aSShri Abhyankar   Mat_SeqAIJ      *a = (Mat_SeqAIJ*)A->data;
10879bd0847aSShri Abhyankar   IS              iscol = a->col,isrow = a->row;
10889bd0847aSShri Abhyankar   PetscErrorCode  ierr;
10899bd0847aSShri Abhyankar   PetscInt        i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j,*adiag = a->diag;
10909bd0847aSShri Abhyankar   PetscInt        nz,neq;
10919bd0847aSShri Abhyankar   const PetscInt  *rout,*cout,*r,*c;
10929bd0847aSShri Abhyankar   PetscScalar     *x,*b,*tmp,sum;
10939bd0847aSShri Abhyankar   const MatScalar *aa = a->a,*v;
1094ace3abfcSBarry Smith   PetscBool       bisdense,xisdense;
10959bd0847aSShri Abhyankar 
10969bd0847aSShri Abhyankar   PetscFunctionBegin;
10979bd0847aSShri Abhyankar   if (!n) PetscFunctionReturn(0);
10989bd0847aSShri Abhyankar 
10999bd0847aSShri Abhyankar   ierr = PetscTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr);
1100e32f2f54SBarry Smith   if (!bisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix");
11019bd0847aSShri Abhyankar   ierr = PetscTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr);
1102e32f2f54SBarry Smith   if (!xisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix");
11039bd0847aSShri Abhyankar 
11049bd0847aSShri Abhyankar   ierr = MatGetArray(B,&b);CHKERRQ(ierr);
11059bd0847aSShri Abhyankar   ierr = MatGetArray(X,&x);CHKERRQ(ierr);
11069bd0847aSShri Abhyankar 
11079bd0847aSShri Abhyankar   tmp  = a->solve_work;
11089bd0847aSShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
11099bd0847aSShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
11109bd0847aSShri Abhyankar 
11119bd0847aSShri Abhyankar   for (neq=0; neq<B->cmap->n; neq++){
11129bd0847aSShri Abhyankar     /* forward solve the lower triangular */
11139bd0847aSShri Abhyankar     tmp[0] = b[r[0]];
11149bd0847aSShri Abhyankar     v      = aa;
11159bd0847aSShri Abhyankar     vi     = aj;
11169bd0847aSShri Abhyankar     for (i=1; i<n; i++) {
11179bd0847aSShri Abhyankar       nz  = ai[i+1] - ai[i];
11189bd0847aSShri Abhyankar       sum = b[r[i]];
11199bd0847aSShri Abhyankar       PetscSparseDenseMinusDot(sum,tmp,v,vi,nz);
11209bd0847aSShri Abhyankar       tmp[i] = sum;
11219bd0847aSShri Abhyankar       v += nz; vi += nz;
11229bd0847aSShri Abhyankar     }
11239bd0847aSShri Abhyankar 
11249bd0847aSShri Abhyankar     /* backward solve the upper triangular */
11259bd0847aSShri Abhyankar     for (i=n-1; i>=0; i--){
11269bd0847aSShri Abhyankar       v   = aa + adiag[i+1]+1;
11279bd0847aSShri Abhyankar       vi  = aj + adiag[i+1]+1;
11289bd0847aSShri Abhyankar       nz  = adiag[i]-adiag[i+1]-1;
11299bd0847aSShri Abhyankar       sum = tmp[i];
11309bd0847aSShri Abhyankar       PetscSparseDenseMinusDot(sum,tmp,v,vi,nz);
11319bd0847aSShri Abhyankar       x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */
11329bd0847aSShri Abhyankar     }
11339bd0847aSShri Abhyankar 
11349bd0847aSShri Abhyankar     b += n;
11359bd0847aSShri Abhyankar     x += n;
11369bd0847aSShri Abhyankar   }
11379bd0847aSShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
11389bd0847aSShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
11399bd0847aSShri Abhyankar   ierr = MatRestoreArray(B,&b);CHKERRQ(ierr);
11409bd0847aSShri Abhyankar   ierr = MatRestoreArray(X,&x);CHKERRQ(ierr);
11419bd0847aSShri Abhyankar   ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr);
11429bd0847aSShri Abhyankar   PetscFunctionReturn(0);
11439bd0847aSShri Abhyankar }
11449bd0847aSShri Abhyankar 
11459bd0847aSShri Abhyankar #undef __FUNCT__
1146137fb511SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_InplaceWithPerm"
1147137fb511SHong Zhang PetscErrorCode MatSolve_SeqAIJ_InplaceWithPerm(Mat A,Vec bb,Vec xx)
1148137fb511SHong Zhang {
1149137fb511SHong Zhang   Mat_SeqAIJ      *a = (Mat_SeqAIJ*)A->data;
1150137fb511SHong Zhang   IS              iscol = a->col,isrow = a->row;
1151137fb511SHong Zhang   PetscErrorCode  ierr;
11525d0c19d7SBarry Smith   const PetscInt  *r,*c,*rout,*cout;
11535d0c19d7SBarry Smith   PetscInt        i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
11545d0c19d7SBarry Smith   PetscInt        nz,row;
1155dd6ea824SBarry Smith   PetscScalar     *x,*b,*tmp,*tmps,sum;
1156dd6ea824SBarry Smith   const MatScalar *aa = a->a,*v;
1157137fb511SHong Zhang 
1158137fb511SHong Zhang   PetscFunctionBegin;
1159137fb511SHong Zhang   if (!n) PetscFunctionReturn(0);
1160137fb511SHong Zhang 
1161137fb511SHong Zhang   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
1162137fb511SHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1163137fb511SHong Zhang   tmp  = a->solve_work;
1164137fb511SHong Zhang 
1165137fb511SHong Zhang   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
1166137fb511SHong Zhang   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
1167137fb511SHong Zhang 
1168137fb511SHong Zhang   /* forward solve the lower triangular */
1169137fb511SHong Zhang   tmp[0] = b[*r++];
1170137fb511SHong Zhang   tmps   = tmp;
1171137fb511SHong Zhang   for (row=1; row<n; row++) {
1172137fb511SHong Zhang     i   = rout[row]; /* permuted row */
1173137fb511SHong Zhang     v   = aa + ai[i] ;
1174137fb511SHong Zhang     vi  = aj + ai[i] ;
1175137fb511SHong Zhang     nz  = a->diag[i] - ai[i];
1176137fb511SHong Zhang     sum = b[*r++];
1177003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
1178137fb511SHong Zhang     tmp[row] = sum;
1179137fb511SHong Zhang   }
1180137fb511SHong Zhang 
1181137fb511SHong Zhang   /* backward solve the upper triangular */
1182137fb511SHong Zhang   for (row=n-1; row>=0; row--){
1183137fb511SHong Zhang     i   = rout[row]; /* permuted row */
1184137fb511SHong Zhang     v   = aa + a->diag[i] + 1;
1185137fb511SHong Zhang     vi  = aj + a->diag[i] + 1;
1186137fb511SHong Zhang     nz  = ai[i+1] - a->diag[i] - 1;
1187137fb511SHong Zhang     sum = tmp[row];
1188003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
1189137fb511SHong Zhang     x[*c--] = tmp[row] = sum*aa[a->diag[i]];
1190137fb511SHong Zhang   }
1191137fb511SHong Zhang 
1192137fb511SHong Zhang   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
1193137fb511SHong Zhang   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
1194137fb511SHong Zhang   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
1195137fb511SHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1196dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
1197137fb511SHong Zhang   PetscFunctionReturn(0);
1198137fb511SHong Zhang }
1199137fb511SHong Zhang 
1200930ae53cSBarry Smith /* ----------------------------------------------------------- */
1201c6db04a5SJed Brown #include <../src/mat/impls/aij/seq/ftn-kernels/fsolve.h>
12024a2ae208SSatish Balay #undef __FUNCT__
1203ad04f41aSHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering_inplace"
1204ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering_inplace(Mat A,Vec bb,Vec xx)
1205930ae53cSBarry Smith {
1206930ae53cSBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
12076849ba73SBarry Smith   PetscErrorCode    ierr;
1208d0f46423SBarry Smith   PetscInt          n = A->rmap->n;
1209b377110cSBarry Smith   const PetscInt    *ai = a->i,*aj = a->j,*adiag = a->diag;
1210356650c2SBarry Smith   PetscScalar       *x;
1211356650c2SBarry Smith   const PetscScalar *b;
1212dd6ea824SBarry Smith   const MatScalar   *aa = a->a;
1213aa482453SBarry Smith #if !defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ)
1214356650c2SBarry Smith   PetscInt          adiag_i,i,nz,ai_i;
1215b377110cSBarry Smith   const PetscInt    *vi;
1216dd6ea824SBarry Smith   const MatScalar   *v;
1217dd6ea824SBarry Smith   PetscScalar       sum;
1218d85afc42SSatish Balay #endif
1219930ae53cSBarry Smith 
12203a40ed3dSBarry Smith   PetscFunctionBegin;
12213a40ed3dSBarry Smith   if (!n) PetscFunctionReturn(0);
1222930ae53cSBarry Smith 
12233649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
12241ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1225930ae53cSBarry Smith 
1226aa482453SBarry Smith #if defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ)
12271c4feecaSSatish Balay   fortransolveaij_(&n,x,ai,aj,adiag,aa,b);
12281c4feecaSSatish Balay #else
1229930ae53cSBarry Smith   /* forward solve the lower triangular */
1230930ae53cSBarry Smith   x[0] = b[0];
1231930ae53cSBarry Smith   for (i=1; i<n; i++) {
1232930ae53cSBarry Smith     ai_i = ai[i];
1233930ae53cSBarry Smith     v    = aa + ai_i;
1234930ae53cSBarry Smith     vi   = aj + ai_i;
1235930ae53cSBarry Smith     nz   = adiag[i] - ai_i;
1236930ae53cSBarry Smith     sum  = b[i];
1237003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
1238930ae53cSBarry Smith     x[i] = sum;
1239930ae53cSBarry Smith   }
1240930ae53cSBarry Smith 
1241930ae53cSBarry Smith   /* backward solve the upper triangular */
1242930ae53cSBarry Smith   for (i=n-1; i>=0; i--){
1243930ae53cSBarry Smith     adiag_i = adiag[i];
1244d708749eSBarry Smith     v       = aa + adiag_i + 1;
1245d708749eSBarry Smith     vi      = aj + adiag_i + 1;
1246930ae53cSBarry Smith     nz      = ai[i+1] - adiag_i - 1;
1247930ae53cSBarry Smith     sum     = x[i];
1248003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
1249930ae53cSBarry Smith     x[i]    = sum*aa[adiag_i];
1250930ae53cSBarry Smith   }
12511c4feecaSSatish Balay #endif
1252dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
12533649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
12541ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
12553a40ed3dSBarry Smith   PetscFunctionReturn(0);
1256930ae53cSBarry Smith }
1257930ae53cSBarry Smith 
12584a2ae208SSatish Balay #undef __FUNCT__
1259ad04f41aSHong Zhang #define __FUNCT__ "MatSolveAdd_SeqAIJ_inplace"
1260ad04f41aSHong Zhang PetscErrorCode MatSolveAdd_SeqAIJ_inplace(Mat A,Vec bb,Vec yy,Vec xx)
1261da3a660dSBarry Smith {
1262416022c9SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
1263416022c9SBarry Smith   IS                iscol = a->col,isrow = a->row;
12646849ba73SBarry Smith   PetscErrorCode    ierr;
126504fbf559SBarry Smith   PetscInt          i, n = A->rmap->n,j;
12665d0c19d7SBarry Smith   PetscInt          nz;
126704fbf559SBarry Smith   const PetscInt    *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j;
126804fbf559SBarry Smith   PetscScalar       *x,*tmp,sum;
126904fbf559SBarry Smith   const PetscScalar *b;
1270dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
1271da3a660dSBarry Smith 
12723a40ed3dSBarry Smith   PetscFunctionBegin;
127378b31e54SBarry Smith   if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);}
1274da3a660dSBarry Smith 
12753649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
12761ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1277416022c9SBarry Smith   tmp  = a->solve_work;
1278da3a660dSBarry Smith 
12793b2fbd54SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
12803b2fbd54SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
1281da3a660dSBarry Smith 
1282da3a660dSBarry Smith   /* forward solve the lower triangular */
1283da3a660dSBarry Smith   tmp[0] = b[*r++];
1284da3a660dSBarry Smith   for (i=1; i<n; i++) {
1285010a20caSHong Zhang     v   = aa + ai[i] ;
1286010a20caSHong Zhang     vi  = aj + ai[i] ;
1287416022c9SBarry Smith     nz  = a->diag[i] - ai[i];
1288da3a660dSBarry Smith     sum = b[*r++];
128904fbf559SBarry Smith     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
1290da3a660dSBarry Smith     tmp[i] = sum;
1291da3a660dSBarry Smith   }
1292da3a660dSBarry Smith 
1293da3a660dSBarry Smith   /* backward solve the upper triangular */
1294da3a660dSBarry Smith   for (i=n-1; i>=0; i--){
1295010a20caSHong Zhang     v   = aa + a->diag[i] + 1;
1296010a20caSHong Zhang     vi  = aj + a->diag[i] + 1;
1297416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
1298da3a660dSBarry Smith     sum = tmp[i];
129904fbf559SBarry Smith     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
1300010a20caSHong Zhang     tmp[i] = sum*aa[a->diag[i]];
1301da3a660dSBarry Smith     x[*c--] += tmp[i];
1302da3a660dSBarry Smith   }
1303da3a660dSBarry Smith 
13043b2fbd54SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
13053b2fbd54SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
13063649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
13071ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1308dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
1309898183ecSLois Curfman McInnes 
13103a40ed3dSBarry Smith   PetscFunctionReturn(0);
1311da3a660dSBarry Smith }
131204fbf559SBarry Smith 
13134a2ae208SSatish Balay #undef __FUNCT__
131435233d99SShri Abhyankar #define __FUNCT__ "MatSolveAdd_SeqAIJ"
131535233d99SShri Abhyankar PetscErrorCode MatSolveAdd_SeqAIJ(Mat A,Vec bb,Vec yy,Vec xx)
13163c0119dfSShri Abhyankar {
13173c0119dfSShri Abhyankar   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
13183c0119dfSShri Abhyankar   IS                iscol = a->col,isrow = a->row;
13193c0119dfSShri Abhyankar   PetscErrorCode    ierr;
13203c0119dfSShri Abhyankar   PetscInt          i, n = A->rmap->n,j;
13213c0119dfSShri Abhyankar   PetscInt          nz;
13223c0119dfSShri Abhyankar   const PetscInt    *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j,*adiag = a->diag;
13233c0119dfSShri Abhyankar   PetscScalar       *x,*tmp,sum;
13243c0119dfSShri Abhyankar   const PetscScalar *b;
13253c0119dfSShri Abhyankar   const MatScalar   *aa = a->a,*v;
13263c0119dfSShri Abhyankar 
13273c0119dfSShri Abhyankar   PetscFunctionBegin;
13283c0119dfSShri Abhyankar   if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);}
13293c0119dfSShri Abhyankar 
13303649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
13313c0119dfSShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
13323c0119dfSShri Abhyankar   tmp  = a->solve_work;
13333c0119dfSShri Abhyankar 
13343c0119dfSShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
13353c0119dfSShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
13363c0119dfSShri Abhyankar 
13373c0119dfSShri Abhyankar   /* forward solve the lower triangular */
13383c0119dfSShri Abhyankar   tmp[0] = b[r[0]];
13393c0119dfSShri Abhyankar   v      = aa;
13403c0119dfSShri Abhyankar   vi     = aj;
13413c0119dfSShri Abhyankar   for (i=1; i<n; i++) {
13423c0119dfSShri Abhyankar     nz  = ai[i+1] - ai[i];
13433c0119dfSShri Abhyankar     sum = b[r[i]];
13443c0119dfSShri Abhyankar     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
13453c0119dfSShri Abhyankar     tmp[i] = sum;
13463c0119dfSShri Abhyankar     v += nz; vi += nz;
13473c0119dfSShri Abhyankar   }
13483c0119dfSShri Abhyankar 
13493c0119dfSShri Abhyankar   /* backward solve the upper triangular */
13503c0119dfSShri Abhyankar   v  = aa + adiag[n-1];
13513c0119dfSShri Abhyankar   vi = aj + adiag[n-1];
13523c0119dfSShri Abhyankar   for (i=n-1; i>=0; i--){
13533c0119dfSShri Abhyankar     nz  = adiag[i] - adiag[i+1] - 1;
13543c0119dfSShri Abhyankar     sum = tmp[i];
13553c0119dfSShri Abhyankar     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
13563c0119dfSShri Abhyankar     tmp[i] = sum*v[nz];
13573c0119dfSShri Abhyankar     x[c[i]] += tmp[i];
13583c0119dfSShri Abhyankar     v += nz+1; vi += nz+1;
13593c0119dfSShri Abhyankar   }
13603c0119dfSShri Abhyankar 
13613c0119dfSShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
13623c0119dfSShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
13633649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
13643c0119dfSShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
13653c0119dfSShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
13663c0119dfSShri Abhyankar 
13673c0119dfSShri Abhyankar   PetscFunctionReturn(0);
13683c0119dfSShri Abhyankar }
13693c0119dfSShri Abhyankar 
13703c0119dfSShri Abhyankar #undef __FUNCT__
1371ad04f41aSHong Zhang #define __FUNCT__ "MatSolveTranspose_SeqAIJ_inplace"
1372ad04f41aSHong Zhang PetscErrorCode MatSolveTranspose_SeqAIJ_inplace(Mat A,Vec bb,Vec xx)
1373da3a660dSBarry Smith {
1374416022c9SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
13752235e667SBarry Smith   IS                iscol = a->col,isrow = a->row;
13766849ba73SBarry Smith   PetscErrorCode    ierr;
137704fbf559SBarry Smith   const PetscInt    *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi;
137804fbf559SBarry Smith   PetscInt          i,n = A->rmap->n,j;
137904fbf559SBarry Smith   PetscInt          nz;
138004fbf559SBarry Smith   PetscScalar       *x,*tmp,s1;
1381dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
138204fbf559SBarry Smith   const PetscScalar *b;
1383da3a660dSBarry Smith 
13843a40ed3dSBarry Smith   PetscFunctionBegin;
13853649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
13861ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1387416022c9SBarry Smith   tmp  = a->solve_work;
1388da3a660dSBarry Smith 
13892235e667SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
13902235e667SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
1391da3a660dSBarry Smith 
1392da3a660dSBarry Smith   /* copy the b into temp work space according to permutation */
13932235e667SBarry Smith   for (i=0; i<n; i++) tmp[i] = b[c[i]];
1394da3a660dSBarry Smith 
1395da3a660dSBarry Smith   /* forward solve the U^T */
1396da3a660dSBarry Smith   for (i=0; i<n; i++) {
1397010a20caSHong Zhang     v   = aa + diag[i] ;
1398010a20caSHong Zhang     vi  = aj + diag[i] + 1;
1399f1af5d2fSBarry Smith     nz  = ai[i+1] - diag[i] - 1;
1400c38d4ed2SBarry Smith     s1  = tmp[i];
1401ef66eb69SBarry Smith     s1 *= (*v++);  /* multiply by inverse of diagonal entry */
140204fbf559SBarry Smith     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1403c38d4ed2SBarry Smith     tmp[i] = s1;
1404da3a660dSBarry Smith   }
1405da3a660dSBarry Smith 
1406da3a660dSBarry Smith   /* backward solve the L^T */
1407da3a660dSBarry Smith   for (i=n-1; i>=0; i--){
1408010a20caSHong Zhang     v   = aa + diag[i] - 1 ;
1409010a20caSHong Zhang     vi  = aj + diag[i] - 1 ;
1410f1af5d2fSBarry Smith     nz  = diag[i] - ai[i];
1411f1af5d2fSBarry Smith     s1  = tmp[i];
141204fbf559SBarry Smith     for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j];
1413da3a660dSBarry Smith   }
1414da3a660dSBarry Smith 
1415da3a660dSBarry Smith   /* copy tmp into x according to permutation */
1416591294e4SBarry Smith   for (i=0; i<n; i++) x[r[i]] = tmp[i];
1417da3a660dSBarry Smith 
14182235e667SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
14192235e667SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
14203649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
14211ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1422da3a660dSBarry Smith 
1423dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr);
14243a40ed3dSBarry Smith   PetscFunctionReturn(0);
1425da3a660dSBarry Smith }
1426da3a660dSBarry Smith 
1427d1fa9404SShri Abhyankar #undef __FUNCT__
142835233d99SShri Abhyankar #define __FUNCT__ "MatSolveTranspose_SeqAIJ"
142935233d99SShri Abhyankar PetscErrorCode MatSolveTranspose_SeqAIJ(Mat A,Vec bb,Vec xx)
1430d1fa9404SShri Abhyankar {
1431d1fa9404SShri Abhyankar   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
1432d1fa9404SShri Abhyankar   IS                iscol = a->col,isrow = a->row;
1433d1fa9404SShri Abhyankar   PetscErrorCode    ierr;
1434d1fa9404SShri Abhyankar   const PetscInt    *rout,*cout,*r,*c,*adiag = a->diag,*ai = a->i,*aj = a->j,*vi;
1435d1fa9404SShri Abhyankar   PetscInt          i,n = A->rmap->n,j;
1436d1fa9404SShri Abhyankar   PetscInt          nz;
1437d1fa9404SShri Abhyankar   PetscScalar       *x,*tmp,s1;
1438d1fa9404SShri Abhyankar   const MatScalar   *aa = a->a,*v;
1439d1fa9404SShri Abhyankar   const PetscScalar *b;
1440d1fa9404SShri Abhyankar 
1441d1fa9404SShri Abhyankar   PetscFunctionBegin;
14423649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
1443d1fa9404SShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1444d1fa9404SShri Abhyankar   tmp  = a->solve_work;
1445d1fa9404SShri Abhyankar 
1446d1fa9404SShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
1447d1fa9404SShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
1448d1fa9404SShri Abhyankar 
1449d1fa9404SShri Abhyankar   /* copy the b into temp work space according to permutation */
1450d1fa9404SShri Abhyankar   for (i=0; i<n; i++) tmp[i] = b[c[i]];
1451d1fa9404SShri Abhyankar 
1452d1fa9404SShri Abhyankar   /* forward solve the U^T */
1453d1fa9404SShri Abhyankar   for (i=0; i<n; i++) {
1454d1fa9404SShri Abhyankar     v   = aa + adiag[i+1] + 1;
1455d1fa9404SShri Abhyankar     vi  = aj + adiag[i+1] + 1;
1456d1fa9404SShri Abhyankar     nz  = adiag[i] - adiag[i+1] - 1;
1457d1fa9404SShri Abhyankar     s1  = tmp[i];
1458d1fa9404SShri Abhyankar     s1 *= v[nz];  /* multiply by inverse of diagonal entry */
1459d1fa9404SShri Abhyankar     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1460d1fa9404SShri Abhyankar     tmp[i] = s1;
1461d1fa9404SShri Abhyankar   }
1462d1fa9404SShri Abhyankar 
1463d1fa9404SShri Abhyankar   /* backward solve the L^T */
1464d1fa9404SShri Abhyankar   for (i=n-1; i>=0; i--){
1465d1fa9404SShri Abhyankar     v   = aa + ai[i];
1466d1fa9404SShri Abhyankar     vi  = aj + ai[i];
1467d1fa9404SShri Abhyankar     nz  = ai[i+1] - ai[i];
1468d1fa9404SShri Abhyankar     s1  = tmp[i];
1469d1fa9404SShri Abhyankar     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1470d1fa9404SShri Abhyankar   }
1471d1fa9404SShri Abhyankar 
1472d1fa9404SShri Abhyankar   /* copy tmp into x according to permutation */
1473d1fa9404SShri Abhyankar   for (i=0; i<n; i++) x[r[i]] = tmp[i];
1474d1fa9404SShri Abhyankar 
1475d1fa9404SShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
1476d1fa9404SShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
14773649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
1478d1fa9404SShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1479d1fa9404SShri Abhyankar 
1480d1fa9404SShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr);
1481d1fa9404SShri Abhyankar   PetscFunctionReturn(0);
1482d1fa9404SShri Abhyankar }
1483d1fa9404SShri Abhyankar 
14844a2ae208SSatish Balay #undef __FUNCT__
1485ad04f41aSHong Zhang #define __FUNCT__ "MatSolveTransposeAdd_SeqAIJ_inplace"
1486ad04f41aSHong Zhang PetscErrorCode MatSolveTransposeAdd_SeqAIJ_inplace(Mat A,Vec bb,Vec zz,Vec xx)
1487da3a660dSBarry Smith {
1488416022c9SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
14892235e667SBarry Smith   IS                iscol = a->col,isrow = a->row;
14906849ba73SBarry Smith   PetscErrorCode    ierr;
149104fbf559SBarry Smith   const PetscInt    *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi;
149204fbf559SBarry Smith   PetscInt          i,n = A->rmap->n,j;
149304fbf559SBarry Smith   PetscInt          nz;
149404fbf559SBarry Smith   PetscScalar       *x,*tmp,s1;
1495dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
149604fbf559SBarry Smith   const PetscScalar *b;
14976abc6512SBarry Smith 
14983a40ed3dSBarry Smith   PetscFunctionBegin;
149923bc6035SMatthew Knepley   if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);}
15003649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
15011ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1502416022c9SBarry Smith   tmp  = a->solve_work;
15036abc6512SBarry Smith 
15042235e667SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
15052235e667SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
15066abc6512SBarry Smith 
15076abc6512SBarry Smith   /* copy the b into temp work space according to permutation */
15082235e667SBarry Smith   for (i=0; i<n; i++) tmp[i] = b[c[i]];
15096abc6512SBarry Smith 
15106abc6512SBarry Smith   /* forward solve the U^T */
15116abc6512SBarry Smith   for (i=0; i<n; i++) {
1512010a20caSHong Zhang     v   = aa + diag[i] ;
1513010a20caSHong Zhang     vi  = aj + diag[i] + 1;
1514f1af5d2fSBarry Smith     nz  = ai[i+1] - diag[i] - 1;
151504fbf559SBarry Smith     s1  = tmp[i];
151604fbf559SBarry Smith     s1 *= (*v++);  /* multiply by inverse of diagonal entry */
151704fbf559SBarry Smith     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
151804fbf559SBarry Smith     tmp[i] = s1;
15196abc6512SBarry Smith   }
15206abc6512SBarry Smith 
15216abc6512SBarry Smith   /* backward solve the L^T */
15226abc6512SBarry Smith   for (i=n-1; i>=0; i--){
1523010a20caSHong Zhang     v   = aa + diag[i] - 1 ;
1524010a20caSHong Zhang     vi  = aj + diag[i] - 1 ;
1525f1af5d2fSBarry Smith     nz  = diag[i] - ai[i];
152604fbf559SBarry Smith     s1  = tmp[i];
152704fbf559SBarry Smith     for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j];
15286abc6512SBarry Smith   }
15296abc6512SBarry Smith 
15306abc6512SBarry Smith   /* copy tmp into x according to permutation */
15316abc6512SBarry Smith   for (i=0; i<n; i++) x[r[i]] += tmp[i];
15326abc6512SBarry Smith 
15332235e667SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
15342235e667SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
15353649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
15361ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
15376abc6512SBarry Smith 
153804fbf559SBarry Smith   ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr);
15393a40ed3dSBarry Smith   PetscFunctionReturn(0);
1540da3a660dSBarry Smith }
154104fbf559SBarry Smith 
1542533e6011SShri Abhyankar #undef __FUNCT__
154335233d99SShri Abhyankar #define __FUNCT__ "MatSolveTransposeAdd_SeqAIJ"
154435233d99SShri Abhyankar PetscErrorCode MatSolveTransposeAdd_SeqAIJ(Mat A,Vec bb,Vec zz,Vec xx)
1545533e6011SShri Abhyankar {
1546533e6011SShri Abhyankar   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
1547533e6011SShri Abhyankar   IS                iscol = a->col,isrow = a->row;
1548533e6011SShri Abhyankar   PetscErrorCode    ierr;
1549533e6011SShri Abhyankar   const PetscInt    *rout,*cout,*r,*c,*adiag = a->diag,*ai = a->i,*aj = a->j,*vi;
1550533e6011SShri Abhyankar   PetscInt          i,n = A->rmap->n,j;
1551533e6011SShri Abhyankar   PetscInt          nz;
1552533e6011SShri Abhyankar   PetscScalar       *x,*tmp,s1;
1553533e6011SShri Abhyankar   const MatScalar   *aa = a->a,*v;
1554533e6011SShri Abhyankar   const PetscScalar *b;
1555533e6011SShri Abhyankar 
1556533e6011SShri Abhyankar   PetscFunctionBegin;
1557533e6011SShri Abhyankar   if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);}
15583649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
1559533e6011SShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1560533e6011SShri Abhyankar   tmp  = a->solve_work;
1561533e6011SShri Abhyankar 
1562533e6011SShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
1563533e6011SShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
1564533e6011SShri Abhyankar 
1565533e6011SShri Abhyankar   /* copy the b into temp work space according to permutation */
1566533e6011SShri Abhyankar   for (i=0; i<n; i++) tmp[i] = b[c[i]];
1567533e6011SShri Abhyankar 
1568533e6011SShri Abhyankar   /* forward solve the U^T */
1569533e6011SShri Abhyankar   for (i=0; i<n; i++) {
1570533e6011SShri Abhyankar     v   = aa + adiag[i+1] + 1;
1571533e6011SShri Abhyankar     vi  = aj + adiag[i+1] + 1;
1572533e6011SShri Abhyankar     nz  = adiag[i] - adiag[i+1] - 1;
1573533e6011SShri Abhyankar     s1  = tmp[i];
1574533e6011SShri Abhyankar     s1 *= v[nz];  /* multiply by inverse of diagonal entry */
1575533e6011SShri Abhyankar     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1576533e6011SShri Abhyankar     tmp[i] = s1;
1577533e6011SShri Abhyankar   }
1578533e6011SShri Abhyankar 
1579533e6011SShri Abhyankar 
1580533e6011SShri Abhyankar   /* backward solve the L^T */
1581533e6011SShri Abhyankar   for (i=n-1; i>=0; i--){
1582533e6011SShri Abhyankar     v   = aa + ai[i] ;
1583533e6011SShri Abhyankar     vi  = aj + ai[i];
1584533e6011SShri Abhyankar     nz  = ai[i+1] - ai[i];
1585533e6011SShri Abhyankar     s1  = tmp[i];
1586c5e3b2a3SShri Abhyankar     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1587533e6011SShri Abhyankar   }
1588533e6011SShri Abhyankar 
1589533e6011SShri Abhyankar   /* copy tmp into x according to permutation */
1590533e6011SShri Abhyankar   for (i=0; i<n; i++) x[r[i]] += tmp[i];
1591533e6011SShri Abhyankar 
1592533e6011SShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
1593533e6011SShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
15943649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
1595533e6011SShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1596533e6011SShri Abhyankar 
1597533e6011SShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr);
1598533e6011SShri Abhyankar   PetscFunctionReturn(0);
1599533e6011SShri Abhyankar }
1600533e6011SShri Abhyankar 
16019e25ed09SBarry Smith /* ----------------------------------------------------------------*/
1602b47f5a65SHong Zhang 
160309573ac7SBarry Smith extern PetscErrorCode MatDuplicateNoCreate_SeqAIJ(Mat,Mat,MatDuplicateOption,PetscBool );
160408480c60SBarry Smith 
16058fc3a347SHong Zhang /*
16068a76ed4fSHong Zhang    ilu() under revised new data structure.
1607813a1f2bSShri Abhyankar    Factored arrays bj and ba are stored as
1608813a1f2bSShri Abhyankar      L(0,:), L(1,:), ...,L(n-1,:),  U(n-1,:),...,U(i,:),U(i-1,:),...,U(0,:)
1609813a1f2bSShri Abhyankar 
1610813a1f2bSShri Abhyankar    bi=fact->i is an array of size n+1, in which
1611813a1f2bSShri Abhyankar    bi+
1612baabb860SHong Zhang      bi[i]:  points to 1st entry of L(i,:),i=0,...,n-1
1613baabb860SHong Zhang      bi[n]:  points to L(n-1,n-1)+1
1614baabb860SHong Zhang 
1615813a1f2bSShri Abhyankar   bdiag=fact->diag is an array of size n+1,in which
1616baabb860SHong Zhang      bdiag[i]: points to diagonal of U(i,:), i=0,...,n-1
1617a24f213cSHong Zhang      bdiag[n]: points to entry of U(n-1,0)-1
1618813a1f2bSShri Abhyankar 
1619813a1f2bSShri Abhyankar    U(i,:) contains bdiag[i] as its last entry, i.e.,
1620813a1f2bSShri Abhyankar     U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i])
1621813a1f2bSShri Abhyankar */
1622813a1f2bSShri Abhyankar #undef __FUNCT__
162335233d99SShri Abhyankar #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_ilu0"
162435233d99SShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ_ilu0(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
1625813a1f2bSShri Abhyankar {
1626813a1f2bSShri Abhyankar 
1627813a1f2bSShri Abhyankar   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
1628813a1f2bSShri Abhyankar   PetscErrorCode     ierr;
1629bbd65245SShri Abhyankar   const PetscInt     n=A->rmap->n,*ai=a->i,*aj,*adiag=a->diag;
1630bbd65245SShri Abhyankar   PetscInt           i,j,k=0,nz,*bi,*bj,*bdiag;
1631ace3abfcSBarry Smith   PetscBool          missing;
16321df811f5SHong Zhang   IS                 isicol;
1633813a1f2bSShri Abhyankar 
1634813a1f2bSShri Abhyankar   PetscFunctionBegin;
1635e32f2f54SBarry 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);
16361df811f5SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr);
1637e32f2f54SBarry Smith   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i);
16381df811f5SHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
16391df811f5SHong Zhang 
1640813a1f2bSShri Abhyankar   ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_FALSE);CHKERRQ(ierr);
1641813a1f2bSShri Abhyankar   b    = (Mat_SeqAIJ*)(fact)->data;
1642813a1f2bSShri Abhyankar 
1643813a1f2bSShri Abhyankar   /* allocate matrix arrays for new data structure */
1644813a1f2bSShri Abhyankar   ierr = PetscMalloc3(ai[n]+1,PetscScalar,&b->a,ai[n]+1,PetscInt,&b->j,n+1,PetscInt,&b->i);CHKERRQ(ierr);
1645813a1f2bSShri Abhyankar   ierr = PetscLogObjectMemory(fact,ai[n]*(sizeof(PetscScalar)+sizeof(PetscInt))+(n+1)*sizeof(PetscInt));CHKERRQ(ierr);
1646813a1f2bSShri Abhyankar   b->singlemalloc = PETSC_TRUE;
1647813a1f2bSShri Abhyankar   if (!b->diag){
1648813a1f2bSShri Abhyankar     ierr = PetscMalloc((n+1)*sizeof(PetscInt),&b->diag);CHKERRQ(ierr);
1649813a1f2bSShri Abhyankar     ierr = PetscLogObjectMemory(fact,(n+1)*sizeof(PetscInt));CHKERRQ(ierr);
1650813a1f2bSShri Abhyankar   }
1651813a1f2bSShri Abhyankar   bdiag = b->diag;
1652813a1f2bSShri Abhyankar 
1653813a1f2bSShri Abhyankar   if (n > 0) {
1654813a1f2bSShri Abhyankar     ierr = PetscMemzero(b->a,(ai[n])*sizeof(MatScalar));CHKERRQ(ierr);
1655813a1f2bSShri Abhyankar   }
1656813a1f2bSShri Abhyankar 
1657813a1f2bSShri Abhyankar   /* set bi and bj with new data structure */
1658813a1f2bSShri Abhyankar   bi = b->i;
1659813a1f2bSShri Abhyankar   bj = b->j;
1660813a1f2bSShri Abhyankar 
1661813a1f2bSShri Abhyankar   /* L part */
1662e60cf9a0SBarry Smith   bi[0] = 0;
1663813a1f2bSShri Abhyankar   for (i=0; i<n; i++){
1664813a1f2bSShri Abhyankar     nz = adiag[i] - ai[i];
1665813a1f2bSShri Abhyankar     bi[i+1] = bi[i] + nz;
1666813a1f2bSShri Abhyankar     aj = a->j + ai[i];
1667813a1f2bSShri Abhyankar     for (j=0; j<nz; j++){
1668bbd65245SShri Abhyankar       /*   *bj = aj[j]; bj++; */
1669bbd65245SShri Abhyankar       bj[k++] = aj[j];
1670813a1f2bSShri Abhyankar     }
1671813a1f2bSShri Abhyankar   }
1672813a1f2bSShri Abhyankar 
1673813a1f2bSShri Abhyankar   /* U part */
167462fbd6afSShri Abhyankar   bdiag[n] = bi[n]-1;
1675813a1f2bSShri Abhyankar   for (i=n-1; i>=0; i--){
1676813a1f2bSShri Abhyankar     nz = ai[i+1] - adiag[i] - 1;
1677813a1f2bSShri Abhyankar     aj = a->j + adiag[i] + 1;
1678813a1f2bSShri Abhyankar     for (j=0; j<nz; j++){
1679bbd65245SShri Abhyankar       /*      *bj = aj[j]; bj++; */
1680bbd65245SShri Abhyankar       bj[k++] = aj[j];
1681813a1f2bSShri Abhyankar     }
1682813a1f2bSShri Abhyankar     /* diag[i] */
1683bbd65245SShri Abhyankar     /*    *bj = i; bj++; */
1684bbd65245SShri Abhyankar     bj[k++] = i;
1685a24f213cSHong Zhang     bdiag[i] = bdiag[i+1] + nz + 1;
1686813a1f2bSShri Abhyankar   }
16871df811f5SHong Zhang 
1688d5f3da31SBarry Smith   fact->factortype             = MAT_FACTOR_ILU;
16891df811f5SHong Zhang   fact->info.factor_mallocs    = 0;
16901df811f5SHong Zhang   fact->info.fill_ratio_given  = info->fill;
16911df811f5SHong Zhang   fact->info.fill_ratio_needed = 1.0;
169235233d99SShri Abhyankar   fact->ops->lufactornumeric   = MatLUFactorNumeric_SeqAIJ;
16931df811f5SHong Zhang 
16941df811f5SHong Zhang   b       = (Mat_SeqAIJ*)(fact)->data;
16951df811f5SHong Zhang   b->row  = isrow;
16961df811f5SHong Zhang   b->col  = iscol;
16971df811f5SHong Zhang   b->icol = isicol;
16981df811f5SHong Zhang   ierr    = PetscMalloc((fact->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
16991df811f5SHong Zhang   ierr    = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
17001df811f5SHong Zhang   ierr    = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
1701813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
1702813a1f2bSShri Abhyankar }
1703813a1f2bSShri Abhyankar 
1704813a1f2bSShri Abhyankar #undef __FUNCT__
170535233d99SShri Abhyankar #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ"
170635233d99SShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
1707813a1f2bSShri Abhyankar {
1708813a1f2bSShri Abhyankar   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
1709813a1f2bSShri Abhyankar   IS                 isicol;
1710813a1f2bSShri Abhyankar   PetscErrorCode     ierr;
1711813a1f2bSShri Abhyankar   const PetscInt     *r,*ic;
17121df811f5SHong Zhang   PetscInt           n=A->rmap->n,*ai=a->i,*aj=a->j;
1713813a1f2bSShri Abhyankar   PetscInt           *bi,*cols,nnz,*cols_lvl;
1714813a1f2bSShri Abhyankar   PetscInt           *bdiag,prow,fm,nzbd,reallocs=0,dcount=0;
1715813a1f2bSShri Abhyankar   PetscInt           i,levels,diagonal_fill;
1716ace3abfcSBarry Smith   PetscBool          col_identity,row_identity;
1717813a1f2bSShri Abhyankar   PetscReal          f;
1718813a1f2bSShri Abhyankar   PetscInt           nlnk,*lnk,*lnk_lvl=PETSC_NULL;
1719813a1f2bSShri Abhyankar   PetscBT            lnkbt;
1720813a1f2bSShri Abhyankar   PetscInt           nzi,*bj,**bj_ptr,**bjlvl_ptr;
1721813a1f2bSShri Abhyankar   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
1722813a1f2bSShri Abhyankar   PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL;
1723813a1f2bSShri Abhyankar 
1724813a1f2bSShri Abhyankar   PetscFunctionBegin;
17253898ab1dSSatish Balay   /* Uncomment the old data struct part only while testing new data structure for MatSolve() */
1726124a3351SHong Zhang   /*
1727ace3abfcSBarry Smith   PetscBool          olddatastruct=PETSC_FALSE;
1728acfcf0e5SJed Brown   ierr = PetscOptionsGetBool(PETSC_NULL,"-ilu_old",&olddatastruct,PETSC_NULL);CHKERRQ(ierr);
172935233d99SShri Abhyankar   if(olddatastruct){
1730ad04f41aSHong Zhang     ierr = MatILUFactorSymbolic_SeqAIJ_inplace(fact,A,isrow,iscol,info);CHKERRQ(ierr);
1731ad04f41aSHong Zhang     PetscFunctionReturn(0);
1732ad04f41aSHong Zhang   }
1733124a3351SHong Zhang   */
1734ad04f41aSHong Zhang 
1735813a1f2bSShri Abhyankar   levels = (PetscInt)info->levels;
1736813a1f2bSShri Abhyankar   ierr   = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
1737813a1f2bSShri Abhyankar   ierr   = ISIdentity(iscol,&col_identity);CHKERRQ(ierr);
1738813a1f2bSShri Abhyankar 
1739813a1f2bSShri Abhyankar   if (!levels && row_identity && col_identity) {
1740813a1f2bSShri Abhyankar     /* special case: ilu(0) with natural ordering */
174135233d99SShri Abhyankar     ierr = MatILUFactorSymbolic_SeqAIJ_ilu0(fact,A,isrow,iscol,info);CHKERRQ(ierr);
1742124a3351SHong Zhang     if (a->inode.size) {
1743124a3351SHong Zhang       fact->ops->lufactornumeric  = MatLUFactorNumeric_SeqAIJ_Inode;
1744124a3351SHong Zhang     }
1745813a1f2bSShri Abhyankar     PetscFunctionReturn(0);
1746813a1f2bSShri Abhyankar   }
1747813a1f2bSShri Abhyankar 
1748e32f2f54SBarry 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);
17491df811f5SHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
1750813a1f2bSShri Abhyankar   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
1751813a1f2bSShri Abhyankar   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
1752813a1f2bSShri Abhyankar 
17531bfa06eaSJed Brown   /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */
17541bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
17551bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
1756e60cf9a0SBarry Smith   bi[0] = bdiag[0] = 0;
1757813a1f2bSShri Abhyankar 
17580e83c824SBarry Smith   ierr = PetscMalloc2(n,PetscInt*,&bj_ptr,n,PetscInt*,&bjlvl_ptr);CHKERRQ(ierr);
1759813a1f2bSShri Abhyankar 
1760813a1f2bSShri Abhyankar   /* create a linked list for storing column indices of the active row */
1761813a1f2bSShri Abhyankar   nlnk = n + 1;
1762813a1f2bSShri Abhyankar   ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
1763813a1f2bSShri Abhyankar 
1764813a1f2bSShri Abhyankar   /* initial FreeSpace size is f*(ai[n]+1) */
17651df811f5SHong Zhang   f             = info->fill;
17661df811f5SHong Zhang   diagonal_fill = (PetscInt)info->diagonal_fill;
1767813a1f2bSShri Abhyankar   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr);
1768813a1f2bSShri Abhyankar   current_space = free_space;
1769813a1f2bSShri Abhyankar   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr);
1770813a1f2bSShri Abhyankar   current_space_lvl = free_space_lvl;
1771813a1f2bSShri Abhyankar 
1772813a1f2bSShri Abhyankar   for (i=0; i<n; i++) {
1773813a1f2bSShri Abhyankar     nzi = 0;
1774813a1f2bSShri Abhyankar     /* copy current row into linked list */
1775813a1f2bSShri Abhyankar     nnz  = ai[r[i]+1] - ai[r[i]];
1776e32f2f54SBarry 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);
1777813a1f2bSShri Abhyankar     cols = aj + ai[r[i]];
1778813a1f2bSShri Abhyankar     lnk[i] = -1; /* marker to indicate if diagonal exists */
1779813a1f2bSShri Abhyankar     ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
1780813a1f2bSShri Abhyankar     nzi += nlnk;
1781813a1f2bSShri Abhyankar 
1782813a1f2bSShri Abhyankar     /* make sure diagonal entry is included */
1783813a1f2bSShri Abhyankar     if (diagonal_fill && lnk[i] == -1) {
1784813a1f2bSShri Abhyankar       fm = n;
1785813a1f2bSShri Abhyankar       while (lnk[fm] < i) fm = lnk[fm];
1786813a1f2bSShri Abhyankar       lnk[i]     = lnk[fm]; /* insert diagonal into linked list */
1787813a1f2bSShri Abhyankar       lnk[fm]    = i;
1788813a1f2bSShri Abhyankar       lnk_lvl[i] = 0;
1789813a1f2bSShri Abhyankar       nzi++; dcount++;
1790813a1f2bSShri Abhyankar     }
1791813a1f2bSShri Abhyankar 
1792813a1f2bSShri Abhyankar     /* add pivot rows into the active row */
1793813a1f2bSShri Abhyankar     nzbd = 0;
1794813a1f2bSShri Abhyankar     prow = lnk[n];
1795813a1f2bSShri Abhyankar     while (prow < i) {
1796813a1f2bSShri Abhyankar       nnz      = bdiag[prow];
1797813a1f2bSShri Abhyankar       cols     = bj_ptr[prow] + nnz + 1;
1798813a1f2bSShri Abhyankar       cols_lvl = bjlvl_ptr[prow] + nnz + 1;
1799813a1f2bSShri Abhyankar       nnz      = bi[prow+1] - bi[prow] - nnz - 1;
1800813a1f2bSShri Abhyankar       ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr);
1801813a1f2bSShri Abhyankar       nzi += nlnk;
1802813a1f2bSShri Abhyankar       prow = lnk[prow];
1803813a1f2bSShri Abhyankar       nzbd++;
1804813a1f2bSShri Abhyankar     }
1805813a1f2bSShri Abhyankar     bdiag[i] = nzbd;
1806813a1f2bSShri Abhyankar     bi[i+1]  = bi[i] + nzi;
1807813a1f2bSShri Abhyankar 
1808813a1f2bSShri Abhyankar     /* if free space is not available, make more free space */
1809813a1f2bSShri Abhyankar     if (current_space->local_remaining<nzi) {
1810813a1f2bSShri Abhyankar       nnz = 2*nzi*(n - i); /* estimated and max additional space needed */
1811813a1f2bSShri Abhyankar       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
1812813a1f2bSShri Abhyankar       ierr = PetscFreeSpaceGet(nnz,&current_space_lvl);CHKERRQ(ierr);
1813813a1f2bSShri Abhyankar       reallocs++;
1814813a1f2bSShri Abhyankar     }
1815813a1f2bSShri Abhyankar 
1816813a1f2bSShri Abhyankar     /* copy data into free_space and free_space_lvl, then initialize lnk */
1817813a1f2bSShri Abhyankar     ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
1818813a1f2bSShri Abhyankar     bj_ptr[i]    = current_space->array;
1819813a1f2bSShri Abhyankar     bjlvl_ptr[i] = current_space_lvl->array;
1820813a1f2bSShri Abhyankar 
1821813a1f2bSShri Abhyankar     /* make sure the active row i has diagonal entry */
182265e19b50SBarry 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);
1823813a1f2bSShri Abhyankar 
1824813a1f2bSShri Abhyankar     current_space->array           += nzi;
1825813a1f2bSShri Abhyankar     current_space->local_used      += nzi;
1826813a1f2bSShri Abhyankar     current_space->local_remaining -= nzi;
1827813a1f2bSShri Abhyankar     current_space_lvl->array           += nzi;
1828813a1f2bSShri Abhyankar     current_space_lvl->local_used      += nzi;
1829813a1f2bSShri Abhyankar     current_space_lvl->local_remaining -= nzi;
1830813a1f2bSShri Abhyankar   }
1831813a1f2bSShri Abhyankar 
1832813a1f2bSShri Abhyankar   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
1833813a1f2bSShri Abhyankar   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
1834813a1f2bSShri Abhyankar 
1835813a1f2bSShri Abhyankar   /* copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */
18369263d837SHong Zhang   ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
18372ce24eb6SHong Zhang   ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr);
1838813a1f2bSShri Abhyankar 
1839813a1f2bSShri Abhyankar   ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
1840813a1f2bSShri Abhyankar   ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
18410e83c824SBarry Smith   ierr = PetscFree2(bj_ptr,bjlvl_ptr);CHKERRQ(ierr);
1842813a1f2bSShri Abhyankar 
1843813a1f2bSShri Abhyankar #if defined(PETSC_USE_INFO)
1844813a1f2bSShri Abhyankar   {
1845aef85c9fSShri Abhyankar     PetscReal af = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]);
1846813a1f2bSShri Abhyankar     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr);
1847813a1f2bSShri Abhyankar     ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
1848813a1f2bSShri Abhyankar     ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr);
1849813a1f2bSShri Abhyankar     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
1850813a1f2bSShri Abhyankar     if (diagonal_fill) {
1851813a1f2bSShri Abhyankar       ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr);
1852813a1f2bSShri Abhyankar     }
1853813a1f2bSShri Abhyankar   }
1854813a1f2bSShri Abhyankar #endif
1855813a1f2bSShri Abhyankar 
1856813a1f2bSShri Abhyankar   /* put together the new matrix */
1857813a1f2bSShri Abhyankar   ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
1858813a1f2bSShri Abhyankar   ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr);
1859813a1f2bSShri Abhyankar   b = (Mat_SeqAIJ*)(fact)->data;
1860813a1f2bSShri Abhyankar   b->free_a       = PETSC_TRUE;
1861813a1f2bSShri Abhyankar   b->free_ij      = PETSC_TRUE;
1862813a1f2bSShri Abhyankar   b->singlemalloc = PETSC_FALSE;
1863f268cba8SShri Abhyankar   ierr = PetscMalloc((bdiag[0]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);
1864813a1f2bSShri Abhyankar   b->j          = bj;
1865813a1f2bSShri Abhyankar   b->i          = bi;
1866813a1f2bSShri Abhyankar   b->diag       = bdiag;
1867813a1f2bSShri Abhyankar   b->ilen       = 0;
1868813a1f2bSShri Abhyankar   b->imax       = 0;
1869813a1f2bSShri Abhyankar   b->row        = isrow;
1870813a1f2bSShri Abhyankar   b->col        = iscol;
1871813a1f2bSShri Abhyankar   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
1872813a1f2bSShri Abhyankar   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
1873813a1f2bSShri Abhyankar   b->icol       = isicol;
1874813a1f2bSShri Abhyankar   ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
1875813a1f2bSShri Abhyankar   /* In b structure:  Free imax, ilen, old a, old j.
1876813a1f2bSShri Abhyankar      Allocate bdiag, solve_work, new a, new j */
1877f268cba8SShri Abhyankar   ierr = PetscLogObjectMemory(fact,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
1878f268cba8SShri Abhyankar   b->maxnz = b->nz = bdiag[0]+1;
1879813a1f2bSShri Abhyankar   (fact)->info.factor_mallocs    = reallocs;
1880813a1f2bSShri Abhyankar   (fact)->info.fill_ratio_given  = f;
1881f268cba8SShri Abhyankar   (fact)->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]);
188235233d99SShri Abhyankar   (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ;
1883f2344125SBarry Smith   if (a->inode.size) {
1884f2344125SBarry Smith     (fact)->ops->lufactornumeric  = MatLUFactorNumeric_SeqAIJ_Inode;
1885f2344125SBarry Smith   }
1886813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
1887813a1f2bSShri Abhyankar }
1888813a1f2bSShri Abhyankar 
18896516239fSHong Zhang #undef __FUNCT__
1890ad04f41aSHong Zhang #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_inplace"
1891ad04f41aSHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
18926516239fSHong Zhang {
18936516239fSHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
18946516239fSHong Zhang   IS                 isicol;
18956516239fSHong Zhang   PetscErrorCode     ierr;
18966516239fSHong Zhang   const PetscInt     *r,*ic;
18976516239fSHong Zhang   PetscInt           n=A->rmap->n,*ai=a->i,*aj=a->j,d;
18986516239fSHong Zhang   PetscInt           *bi,*cols,nnz,*cols_lvl;
18996516239fSHong Zhang   PetscInt           *bdiag,prow,fm,nzbd,reallocs=0,dcount=0;
19006516239fSHong Zhang   PetscInt           i,levels,diagonal_fill;
1901ace3abfcSBarry Smith   PetscBool          col_identity,row_identity;
19026516239fSHong Zhang   PetscReal          f;
19036516239fSHong Zhang   PetscInt           nlnk,*lnk,*lnk_lvl=PETSC_NULL;
19046516239fSHong Zhang   PetscBT            lnkbt;
19056516239fSHong Zhang   PetscInt           nzi,*bj,**bj_ptr,**bjlvl_ptr;
19066516239fSHong Zhang   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
19076516239fSHong Zhang   PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL;
1908ace3abfcSBarry Smith   PetscBool          missing;
19096516239fSHong Zhang 
19106516239fSHong Zhang   PetscFunctionBegin;
1911e32f2f54SBarry 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);
19126516239fSHong Zhang   f             = info->fill;
19136516239fSHong Zhang   levels        = (PetscInt)info->levels;
19146516239fSHong Zhang   diagonal_fill = (PetscInt)info->diagonal_fill;
19156516239fSHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
19166516239fSHong Zhang 
19176516239fSHong Zhang   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
19186516239fSHong Zhang   ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr);
19196516239fSHong Zhang   if (!levels && row_identity && col_identity) { /* special case: ilu(0) with natural ordering */
1920f77e22a1SHong Zhang     ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_TRUE);CHKERRQ(ierr);
1921ad04f41aSHong Zhang     (fact)->ops->lufactornumeric =  MatLUFactorNumeric_SeqAIJ_inplace;
1922f2344125SBarry Smith     if (a->inode.size) {
1923f2344125SBarry Smith       (fact)->ops->lufactornumeric  = MatLUFactorNumeric_SeqAIJ_Inode_inplace;
1924f2344125SBarry Smith     }
1925d5f3da31SBarry Smith     fact->factortype = MAT_FACTOR_ILU;
1926719d5645SBarry Smith     (fact)->info.factor_mallocs    = 0;
1927719d5645SBarry Smith     (fact)->info.fill_ratio_given  = info->fill;
1928719d5645SBarry Smith     (fact)->info.fill_ratio_needed = 1.0;
1929719d5645SBarry Smith     b               = (Mat_SeqAIJ*)(fact)->data;
19308ef3462fSBarry Smith     ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr);
1931e32f2f54SBarry Smith     if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d);
193208480c60SBarry Smith     b->row              = isrow;
193308480c60SBarry Smith     b->col              = iscol;
193482bf6240SBarry Smith     b->icol             = isicol;
1935719d5645SBarry Smith     ierr                = PetscMalloc(((fact)->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
1936c38d4ed2SBarry Smith     ierr                = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
1937c38d4ed2SBarry Smith     ierr                = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
19383a40ed3dSBarry Smith     PetscFunctionReturn(0);
193908480c60SBarry Smith   }
194008480c60SBarry Smith 
1941898183ecSLois Curfman McInnes   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
1942898183ecSLois Curfman McInnes   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
19439e25ed09SBarry Smith 
19441bfa06eaSJed Brown   /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */
19451bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
19461bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
1947a6df321fSShri Abhyankar   bi[0] = bdiag[0] = 0;
19486cf997b0SBarry Smith 
19490e83c824SBarry Smith   ierr = PetscMalloc2(n,PetscInt*,&bj_ptr,n,PetscInt*,&bjlvl_ptr);CHKERRQ(ierr);
19505a8e39fbSHong Zhang 
19515a8e39fbSHong Zhang   /* create a linked list for storing column indices of the active row */
19525a8e39fbSHong Zhang   nlnk = n + 1;
19535a8e39fbSHong Zhang   ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
19545a8e39fbSHong Zhang 
19555a8e39fbSHong Zhang   /* initial FreeSpace size is f*(ai[n]+1) */
1956a1a86e44SBarry Smith   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr);
19575a8e39fbSHong Zhang   current_space = free_space;
1958a1a86e44SBarry Smith   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr);
19595a8e39fbSHong Zhang   current_space_lvl = free_space_lvl;
19605a8e39fbSHong Zhang 
19615a8e39fbSHong Zhang   for (i=0; i<n; i++) {
19625a8e39fbSHong Zhang     nzi = 0;
19636cf997b0SBarry Smith     /* copy current row into linked list */
19645a8e39fbSHong Zhang     nnz  = ai[r[i]+1] - ai[r[i]];
1965e32f2f54SBarry 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);
19665a8e39fbSHong Zhang     cols = aj + ai[r[i]];
19675a8e39fbSHong Zhang     lnk[i] = -1; /* marker to indicate if diagonal exists */
19685a8e39fbSHong Zhang     ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
19695a8e39fbSHong Zhang     nzi += nlnk;
19706cf997b0SBarry Smith 
19716cf997b0SBarry Smith     /* make sure diagonal entry is included */
19725a8e39fbSHong Zhang     if (diagonal_fill && lnk[i] == -1) {
19736cf997b0SBarry Smith       fm = n;
19745a8e39fbSHong Zhang       while (lnk[fm] < i) fm = lnk[fm];
19755a8e39fbSHong Zhang       lnk[i]     = lnk[fm]; /* insert diagonal into linked list */
19765a8e39fbSHong Zhang       lnk[fm]    = i;
19775a8e39fbSHong Zhang       lnk_lvl[i] = 0;
19785a8e39fbSHong Zhang       nzi++; dcount++;
19796cf997b0SBarry Smith     }
19806cf997b0SBarry Smith 
19815a8e39fbSHong Zhang     /* add pivot rows into the active row */
19825a8e39fbSHong Zhang     nzbd = 0;
19835a8e39fbSHong Zhang     prow = lnk[n];
19845a8e39fbSHong Zhang     while (prow < i) {
19855a8e39fbSHong Zhang       nnz      = bdiag[prow];
19865a8e39fbSHong Zhang       cols     = bj_ptr[prow] + nnz + 1;
19875a8e39fbSHong Zhang       cols_lvl = bjlvl_ptr[prow] + nnz + 1;
19885a8e39fbSHong Zhang       nnz      = bi[prow+1] - bi[prow] - nnz - 1;
19895a8e39fbSHong Zhang       ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr);
19905a8e39fbSHong Zhang       nzi += nlnk;
19915a8e39fbSHong Zhang       prow = lnk[prow];
19925a8e39fbSHong Zhang       nzbd++;
199356beaf04SBarry Smith     }
19945a8e39fbSHong Zhang     bdiag[i] = nzbd;
19955a8e39fbSHong Zhang     bi[i+1]  = bi[i] + nzi;
1996ecf371e4SBarry Smith 
19975a8e39fbSHong Zhang     /* if free space is not available, make more free space */
19985a8e39fbSHong Zhang     if (current_space->local_remaining<nzi) {
19995a8e39fbSHong Zhang       nnz = nzi*(n - i); /* estimated and max additional space needed */
2000a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
2001a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(nnz,&current_space_lvl);CHKERRQ(ierr);
20025a8e39fbSHong Zhang       reallocs++;
20035a8e39fbSHong Zhang     }
2004ecf371e4SBarry Smith 
20055a8e39fbSHong Zhang     /* copy data into free_space and free_space_lvl, then initialize lnk */
20065a8e39fbSHong Zhang     ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
20075a8e39fbSHong Zhang     bj_ptr[i]    = current_space->array;
20085a8e39fbSHong Zhang     bjlvl_ptr[i] = current_space_lvl->array;
20095a8e39fbSHong Zhang 
20105a8e39fbSHong Zhang     /* make sure the active row i has diagonal entry */
201165e19b50SBarry 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);
20125a8e39fbSHong Zhang 
20135a8e39fbSHong Zhang     current_space->array           += nzi;
20145a8e39fbSHong Zhang     current_space->local_used      += nzi;
20155a8e39fbSHong Zhang     current_space->local_remaining -= nzi;
20165a8e39fbSHong Zhang     current_space_lvl->array           += nzi;
20175a8e39fbSHong Zhang     current_space_lvl->local_used      += nzi;
20185a8e39fbSHong Zhang     current_space_lvl->local_remaining -= nzi;
20199e25ed09SBarry Smith   }
20205a8e39fbSHong Zhang 
2021898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
2022898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
20235a8e39fbSHong Zhang 
20245a8e39fbSHong Zhang   /* destroy list of free space and other temporary arrays */
20255a8e39fbSHong Zhang   ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
20266516239fSHong Zhang   ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); /* copy free_space -> bj */
20275a8e39fbSHong Zhang   ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
2028a1a86e44SBarry Smith   ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
20290e83c824SBarry Smith   ierr = PetscFree2(bj_ptr,bjlvl_ptr);CHKERRQ(ierr);
20309e25ed09SBarry Smith 
20316cf91177SBarry Smith #if defined(PETSC_USE_INFO)
2032f64065bbSBarry Smith   {
20335a8e39fbSHong Zhang     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
2034ae15b995SBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr);
2035ae15b995SBarry Smith     ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
2036ae15b995SBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr);
2037ae15b995SBarry Smith     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
2038335d9088SBarry Smith     if (diagonal_fill) {
2039ae15b995SBarry Smith       ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr);
2040335d9088SBarry Smith     }
2041f64065bbSBarry Smith   }
204263ba0a88SBarry Smith #endif
2043bbb6d6a8SBarry Smith 
20449e25ed09SBarry Smith   /* put together the new matrix */
204507dbf95fSBarry Smith   ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
2046719d5645SBarry Smith   ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr);
2047719d5645SBarry Smith   b = (Mat_SeqAIJ*)(fact)->data;
2048e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
2049e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
20507c922b88SBarry Smith   b->singlemalloc = PETSC_FALSE;
20510e83c824SBarry Smith   ierr = PetscMalloc(bi[n]*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);
20525a8e39fbSHong Zhang   b->j          = bj;
20535a8e39fbSHong Zhang   b->i          = bi;
20545a8e39fbSHong Zhang   for (i=0; i<n; i++) bdiag[i] += bi[i];
20555a8e39fbSHong Zhang   b->diag       = bdiag;
2056416022c9SBarry Smith   b->ilen       = 0;
2057416022c9SBarry Smith   b->imax       = 0;
2058416022c9SBarry Smith   b->row        = isrow;
2059416022c9SBarry Smith   b->col        = iscol;
2060c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
2061c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
206282bf6240SBarry Smith   b->icol       = isicol;
206387828ca2SBarry Smith   ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
2064416022c9SBarry Smith   /* In b structure:  Free imax, ilen, old a, old j.
20655a8e39fbSHong Zhang      Allocate bdiag, solve_work, new a, new j */
2066719d5645SBarry Smith   ierr = PetscLogObjectMemory(fact,(bi[n]-n) * (sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
20675a8e39fbSHong Zhang   b->maxnz             = b->nz = bi[n] ;
2068719d5645SBarry Smith   (fact)->info.factor_mallocs    = reallocs;
2069719d5645SBarry Smith   (fact)->info.fill_ratio_given  = f;
2070719d5645SBarry Smith   (fact)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]);
2071ad04f41aSHong Zhang   (fact)->ops->lufactornumeric =  MatLUFactorNumeric_SeqAIJ_inplace;
2072f2344125SBarry Smith   if (a->inode.size) {
2073f2344125SBarry Smith     (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode_inplace;
2074d3ac4fa3SBarry Smith   }
20753a40ed3dSBarry Smith   PetscFunctionReturn(0);
20769e25ed09SBarry Smith }
2077d5d45c9bSBarry Smith 
2078a6175056SHong Zhang #undef __FUNCT__
207935233d99SShri Abhyankar #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ"
208035233d99SShri Abhyankar PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info)
20815f44c854SHong Zhang {
20825f44c854SHong Zhang   Mat            C = B;
20835f44c854SHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
20845f44c854SHong Zhang   Mat_SeqSBAIJ   *b=(Mat_SeqSBAIJ*)C->data;
20855f44c854SHong Zhang   IS             ip=b->row,iip = b->icol;
20865f44c854SHong Zhang   PetscErrorCode ierr;
20875f44c854SHong Zhang   const PetscInt *rip,*riip;
20885f44c854SHong Zhang   PetscInt       i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bdiag=b->diag,*bjtmp;
20895f44c854SHong Zhang   PetscInt       *ai=a->i,*aj=a->j;
20905f44c854SHong Zhang   PetscInt       k,jmin,jmax,*c2r,*il,col,nexti,ili,nz;
20915f44c854SHong Zhang   MatScalar      *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi;
2092ace3abfcSBarry Smith   PetscBool      perm_identity;
2093d90ac83dSHong Zhang   FactorShiftCtx sctx;
209498a93161SHong Zhang   PetscReal      rs;
209598a93161SHong Zhang   MatScalar      d,*v;
209698a93161SHong Zhang 
20975f44c854SHong Zhang   PetscFunctionBegin;
209898a93161SHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
2099d90ac83dSHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
210098a93161SHong Zhang 
2101f4db908eSBarry Smith   if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
210298a93161SHong Zhang     sctx.shift_top = info->zeropivot;
210398a93161SHong Zhang     for (i=0; i<mbs; i++) {
210498a93161SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
210598a93161SHong Zhang       d  = (aa)[a->diag[i]];
210698a93161SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
210798a93161SHong Zhang       v  = aa+ai[i];
210898a93161SHong Zhang       nz = ai[i+1] - ai[i];
210998a93161SHong Zhang       for (j=0; j<nz; j++)
211098a93161SHong Zhang 	rs += PetscAbsScalar(v[j]);
211198a93161SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
211298a93161SHong Zhang     }
211398a93161SHong Zhang     sctx.shift_top   *= 1.1;
211498a93161SHong Zhang     sctx.nshift_max   = 5;
211598a93161SHong Zhang     sctx.shift_lo     = 0.;
211698a93161SHong Zhang     sctx.shift_hi     = 1.;
211798a93161SHong Zhang   }
21185f44c854SHong Zhang 
21195f44c854SHong Zhang   ierr  = ISGetIndices(ip,&rip);CHKERRQ(ierr);
21205f44c854SHong Zhang   ierr  = ISGetIndices(iip,&riip);CHKERRQ(ierr);
21215f44c854SHong Zhang 
21225f44c854SHong Zhang   /* allocate working arrays
21235f44c854SHong Zhang      c2r: linked list, keep track of pivot rows for a given column. c2r[col]: head of the list for a given col
21245f44c854SHong 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
21255f44c854SHong Zhang   */
21260e83c824SBarry Smith   ierr = PetscMalloc3(mbs,MatScalar,&rtmp,mbs,PetscInt,&il,mbs,PetscInt,&c2r);CHKERRQ(ierr);
21275f44c854SHong Zhang 
212898a93161SHong Zhang   do {
212907b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
213098a93161SHong Zhang 
2131c5546cabSHong Zhang     for (i=0; i<mbs; i++) c2r[i] = mbs;
21325f44c854SHong Zhang     il[0] = 0;
21335f44c854SHong Zhang 
21345f44c854SHong Zhang     for (k = 0; k<mbs; k++){
21355f44c854SHong Zhang       /* zero rtmp */
21365f44c854SHong Zhang       nz = bi[k+1] - bi[k];
21375f44c854SHong Zhang       bjtmp = bj + bi[k];
21385f44c854SHong Zhang       for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
21395f44c854SHong Zhang 
21405f44c854SHong Zhang       /* load in initial unfactored row */
21415f44c854SHong Zhang       bval = ba + bi[k];
21425f44c854SHong Zhang       jmin = ai[rip[k]]; jmax = ai[rip[k]+1];
21435f44c854SHong Zhang       for (j = jmin; j < jmax; j++){
21445f44c854SHong Zhang         col = riip[aj[j]];
21455f44c854SHong Zhang         if (col >= k){ /* only take upper triangular entry */
21465f44c854SHong Zhang           rtmp[col] = aa[j];
21475f44c854SHong Zhang           *bval++   = 0.0; /* for in-place factorization */
21485f44c854SHong Zhang         }
21495f44c854SHong Zhang       }
215098a93161SHong Zhang       /* shift the diagonal of the matrix: ZeropivotApply() */
215198a93161SHong Zhang       rtmp[k] += sctx.shift_amount;  /* shift the diagonal of the matrix */
21525f44c854SHong Zhang 
21535f44c854SHong Zhang       /* modify k-th row by adding in those rows i with U(i,k)!=0 */
21545f44c854SHong Zhang       dk = rtmp[k];
21555f44c854SHong Zhang       i  = c2r[k]; /* first row to be added to k_th row  */
21565f44c854SHong Zhang 
21575f44c854SHong Zhang       while (i < k){
21585f44c854SHong Zhang         nexti = c2r[i]; /* next row to be added to k_th row */
21595f44c854SHong Zhang 
21605f44c854SHong Zhang         /* compute multiplier, update diag(k) and U(i,k) */
21615f44c854SHong Zhang         ili   = il[i];  /* index of first nonzero element in U(i,k:bms-1) */
21625f44c854SHong Zhang         uikdi = - ba[ili]*ba[bdiag[i]];  /* diagonal(k) */
21635f44c854SHong Zhang         dk   += uikdi*ba[ili]; /* update diag[k] */
21645f44c854SHong Zhang         ba[ili] = uikdi; /* -U(i,k) */
21655f44c854SHong Zhang 
21665f44c854SHong Zhang         /* add multiple of row i to k-th row */
21675f44c854SHong Zhang         jmin = ili + 1; jmax = bi[i+1];
21685f44c854SHong Zhang         if (jmin < jmax){
21695f44c854SHong Zhang           for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j];
21705f44c854SHong Zhang           /* update il and c2r for row i */
21715f44c854SHong Zhang           il[i] = jmin;
21725f44c854SHong Zhang           j = bj[jmin]; c2r[i] = c2r[j]; c2r[j] = i;
21735f44c854SHong Zhang         }
21745f44c854SHong Zhang         i = nexti;
21755f44c854SHong Zhang       }
21765f44c854SHong Zhang 
21775f44c854SHong Zhang       /* copy data into U(k,:) */
217898a93161SHong Zhang       rs   = 0.0;
21795f44c854SHong Zhang       jmin = bi[k]; jmax = bi[k+1]-1;
21805f44c854SHong Zhang       if (jmin < jmax) {
21815f44c854SHong Zhang         for (j=jmin; j<jmax; j++){
218298a93161SHong Zhang           col = bj[j]; ba[j] = rtmp[col]; rs += PetscAbsScalar(ba[j]);
21835f44c854SHong Zhang         }
21845f44c854SHong Zhang         /* add the k-th row into il and c2r */
21855f44c854SHong Zhang         il[k] = jmin;
21865f44c854SHong Zhang         i = bj[jmin]; c2r[k] = c2r[i]; c2r[i] = k;
21875f44c854SHong Zhang       }
218898a93161SHong Zhang 
218998a93161SHong Zhang       /* MatPivotCheck() */
219098a93161SHong Zhang       sctx.rs  = rs;
219198a93161SHong Zhang       sctx.pv  = dk;
2192d0660788SBarry Smith       ierr = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr);
219307b50cabSHong Zhang       if(sctx.newshift) break;
219498a93161SHong Zhang       dk = sctx.pv;
219598a93161SHong Zhang 
219698a93161SHong Zhang       ba[bdiag[k]] = 1.0/dk; /* U(k,k) */
219798a93161SHong Zhang     }
219807b50cabSHong Zhang   } while (sctx.newshift);
21995f44c854SHong Zhang 
22000e83c824SBarry Smith   ierr = PetscFree3(rtmp,il,c2r);CHKERRQ(ierr);
22015f44c854SHong Zhang   ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr);
22025f44c854SHong Zhang   ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr);
22035f44c854SHong Zhang 
22045f44c854SHong Zhang   ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr);
22055f44c854SHong Zhang   if (perm_identity){
220635233d99SShri Abhyankar     B->ops->solve           = MatSolve_SeqSBAIJ_1_NaturalOrdering;
220735233d99SShri Abhyankar     B->ops->solvetranspose  = MatSolve_SeqSBAIJ_1_NaturalOrdering;
22082169352eSHong Zhang     B->ops->forwardsolve    = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering;
22092169352eSHong Zhang     B->ops->backwardsolve   = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering;
22105f44c854SHong Zhang   } else {
221135233d99SShri Abhyankar     B->ops->solve           = MatSolve_SeqSBAIJ_1;
221235233d99SShri Abhyankar     B->ops->solvetranspose  = MatSolve_SeqSBAIJ_1;
221335233d99SShri Abhyankar     B->ops->forwardsolve    = MatForwardSolve_SeqSBAIJ_1;
221435233d99SShri Abhyankar     B->ops->backwardsolve   = MatBackwardSolve_SeqSBAIJ_1;
22155f44c854SHong Zhang   }
22165f44c854SHong Zhang 
22175f44c854SHong Zhang   C->assembled    = PETSC_TRUE;
22185f44c854SHong Zhang   C->preallocated = PETSC_TRUE;
22195f44c854SHong Zhang   ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr);
222098a93161SHong Zhang 
222198a93161SHong Zhang   /* MatPivotView() */
222298a93161SHong Zhang   if (sctx.nshift){
2223f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
222498a93161SHong 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);
2225f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
222698a93161SHong Zhang       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
2227f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_INBLOCKS){
2228b2a402fcSHong Zhang       ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %G\n",sctx.nshift,info->shiftamount);CHKERRQ(ierr);
222998a93161SHong Zhang     }
223098a93161SHong Zhang   }
22315f44c854SHong Zhang   PetscFunctionReturn(0);
22325f44c854SHong Zhang }
22335f44c854SHong Zhang 
22345f44c854SHong Zhang #undef __FUNCT__
22350a3c351aSHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ_inplace"
22360a3c351aSHong Zhang PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ_inplace(Mat B,Mat A,const MatFactorInfo *info)
2237a6175056SHong Zhang {
2238719d5645SBarry Smith   Mat            C = B;
22390968510aSHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
22402ed133dbSHong Zhang   Mat_SeqSBAIJ   *b=(Mat_SeqSBAIJ*)C->data;
22419bfd6278SHong Zhang   IS             ip=b->row,iip = b->icol;
22422ed133dbSHong Zhang   PetscErrorCode ierr;
22435d0c19d7SBarry Smith   const PetscInt *rip,*riip;
2244c5546cabSHong Zhang   PetscInt       i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bcol,*bjtmp;
22452ed133dbSHong Zhang   PetscInt       *ai=a->i,*aj=a->j;
22462ed133dbSHong Zhang   PetscInt       k,jmin,jmax,*jl,*il,col,nexti,ili,nz;
2247622e2028SHong Zhang   MatScalar      *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi;
2248ace3abfcSBarry Smith   PetscBool      perm_identity;
22490e95ead3SHong Zhang   FactorShiftCtx sctx;
22500e95ead3SHong Zhang   PetscReal      rs;
22510e95ead3SHong Zhang   MatScalar      d,*v;
2252d5d45c9bSBarry Smith 
2253a6175056SHong Zhang   PetscFunctionBegin;
22540e95ead3SHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
22550e95ead3SHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
22560e95ead3SHong Zhang 
22570e95ead3SHong Zhang   if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
22580e95ead3SHong Zhang     sctx.shift_top = info->zeropivot;
22590e95ead3SHong Zhang     for (i=0; i<mbs; i++) {
22600e95ead3SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
22610e95ead3SHong Zhang       d  = (aa)[a->diag[i]];
22620e95ead3SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
22630e95ead3SHong Zhang       v  = aa+ai[i];
22640e95ead3SHong Zhang       nz = ai[i+1] - ai[i];
22650e95ead3SHong Zhang       for (j=0; j<nz; j++)
22660e95ead3SHong Zhang 	rs += PetscAbsScalar(v[j]);
22670e95ead3SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
22680e95ead3SHong Zhang     }
22690e95ead3SHong Zhang     sctx.shift_top   *= 1.1;
22700e95ead3SHong Zhang     sctx.nshift_max   = 5;
22710e95ead3SHong Zhang     sctx.shift_lo     = 0.;
22720e95ead3SHong Zhang     sctx.shift_hi     = 1.;
22730e95ead3SHong Zhang   }
2274ee45ca4aSHong Zhang 
22752ed133dbSHong Zhang   ierr  = ISGetIndices(ip,&rip);CHKERRQ(ierr);
22769bfd6278SHong Zhang   ierr  = ISGetIndices(iip,&riip);CHKERRQ(ierr);
22772ed133dbSHong Zhang 
22782ed133dbSHong Zhang   /* initialization */
22790e83c824SBarry Smith   ierr = PetscMalloc3(mbs,MatScalar,&rtmp,mbs,PetscInt,&il,mbs,PetscInt,&jl);CHKERRQ(ierr);
22800e95ead3SHong Zhang 
22812ed133dbSHong Zhang   do {
228207b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
22830e95ead3SHong Zhang 
2284c5546cabSHong Zhang     for (i=0; i<mbs; i++) jl[i] = mbs;
2285c5546cabSHong Zhang     il[0] = 0;
22862ed133dbSHong Zhang 
22872ed133dbSHong Zhang     for (k = 0; k<mbs; k++){
2288c5546cabSHong Zhang       /* zero rtmp */
2289c5546cabSHong Zhang       nz = bi[k+1] - bi[k];
2290c5546cabSHong Zhang       bjtmp = bj + bi[k];
2291c5546cabSHong Zhang       for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
2292c5546cabSHong Zhang 
2293c05c3958SHong Zhang       bval = ba + bi[k];
22942ed133dbSHong Zhang       /* initialize k-th row by the perm[k]-th row of A */
22952ed133dbSHong Zhang       jmin = ai[rip[k]]; jmax = ai[rip[k]+1];
22962ed133dbSHong Zhang       for (j = jmin; j < jmax; j++){
22979bfd6278SHong Zhang         col = riip[aj[j]];
22982ed133dbSHong Zhang         if (col >= k){ /* only take upper triangular entry */
22992ed133dbSHong Zhang           rtmp[col] = aa[j];
2300c05c3958SHong Zhang           *bval++  = 0.0; /* for in-place factorization */
23012ed133dbSHong Zhang         }
23022ed133dbSHong Zhang       }
230339e3d630SHong Zhang       /* shift the diagonal of the matrix */
2304540703acSHong Zhang       if (sctx.nshift) rtmp[k] += sctx.shift_amount;
23052ed133dbSHong Zhang 
23062ed133dbSHong Zhang       /* modify k-th row by adding in those rows i with U(i,k)!=0 */
23072ed133dbSHong Zhang       dk = rtmp[k];
23082ed133dbSHong Zhang       i = jl[k]; /* first row to be added to k_th row  */
23092ed133dbSHong Zhang 
23102ed133dbSHong Zhang       while (i < k){
23112ed133dbSHong Zhang         nexti = jl[i]; /* next row to be added to k_th row */
23122ed133dbSHong Zhang 
23132ed133dbSHong Zhang         /* compute multiplier, update diag(k) and U(i,k) */
23142ed133dbSHong Zhang         ili = il[i];  /* index of first nonzero element in U(i,k:bms-1) */
23152ed133dbSHong Zhang         uikdi = - ba[ili]*ba[bi[i]];  /* diagonal(k) */
23162ed133dbSHong Zhang         dk += uikdi*ba[ili];
23172ed133dbSHong Zhang         ba[ili] = uikdi; /* -U(i,k) */
23182ed133dbSHong Zhang 
23192ed133dbSHong Zhang         /* add multiple of row i to k-th row */
23202ed133dbSHong Zhang         jmin = ili + 1; jmax = bi[i+1];
23212ed133dbSHong Zhang         if (jmin < jmax){
23222ed133dbSHong Zhang           for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j];
23232ed133dbSHong Zhang           /* update il and jl for row i */
23242ed133dbSHong Zhang           il[i] = jmin;
23252ed133dbSHong Zhang           j = bj[jmin]; jl[i] = jl[j]; jl[j] = i;
23262ed133dbSHong Zhang         }
23272ed133dbSHong Zhang         i = nexti;
23282ed133dbSHong Zhang       }
23292ed133dbSHong Zhang 
23300697b6caSHong Zhang       /* shift the diagonals when zero pivot is detected */
23310697b6caSHong Zhang       /* compute rs=sum of abs(off-diagonal) */
23320697b6caSHong Zhang       rs   = 0.0;
23333c9e8598SHong Zhang       jmin = bi[k]+1;
23343c9e8598SHong Zhang       nz   = bi[k+1] - jmin;
23353c9e8598SHong Zhang       bcol = bj + jmin;
233604fbf559SBarry Smith       for (j=0; j<nz; j++) {
233704fbf559SBarry Smith         rs += PetscAbsScalar(rtmp[bcol[j]]);
23383c9e8598SHong Zhang       }
2339540703acSHong Zhang 
2340540703acSHong Zhang       sctx.rs = rs;
2341540703acSHong Zhang       sctx.pv = dk;
2342d0660788SBarry Smith       ierr = MatPivotCheck(A,info,&sctx,k);CHKERRQ(ierr);
234307b50cabSHong Zhang       if (sctx.newshift) break;
23440e95ead3SHong Zhang       dk = sctx.pv;
23453c9e8598SHong Zhang 
23463c9e8598SHong Zhang       /* copy data into U(k,:) */
234739e3d630SHong Zhang       ba[bi[k]] = 1.0/dk; /* U(k,k) */
234839e3d630SHong Zhang       jmin = bi[k]+1; jmax = bi[k+1];
234939e3d630SHong Zhang       if (jmin < jmax) {
235039e3d630SHong Zhang         for (j=jmin; j<jmax; j++){
2351c6ae9bfcSHong Zhang           col = bj[j]; ba[j] = rtmp[col];
23523c9e8598SHong Zhang         }
235339e3d630SHong Zhang         /* add the k-th row into il and jl */
23543c9e8598SHong Zhang         il[k] = jmin;
23553c9e8598SHong Zhang         i = bj[jmin]; jl[k] = jl[i]; jl[i] = k;
23563c9e8598SHong Zhang       }
23573c9e8598SHong Zhang     }
235807b50cabSHong Zhang   } while (sctx.newshift);
23590e95ead3SHong Zhang 
23600e83c824SBarry Smith   ierr = PetscFree3(rtmp,il,jl);CHKERRQ(ierr);
236139e3d630SHong Zhang   ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr);
23629bfd6278SHong Zhang   ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr);
2363db4efbfdSBarry Smith 
2364db4efbfdSBarry Smith   ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr);
2365db4efbfdSBarry Smith   if (perm_identity){
23660a3c351aSHong Zhang     B->ops->solve           = MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace;
23670a3c351aSHong Zhang     B->ops->solvetranspose  = MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace;
23680a3c351aSHong Zhang     B->ops->forwardsolve    = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace;
23690a3c351aSHong Zhang     B->ops->backwardsolve   = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace;
2370db4efbfdSBarry Smith   } else {
23710a3c351aSHong Zhang     B->ops->solve           = MatSolve_SeqSBAIJ_1_inplace;
23720a3c351aSHong Zhang     B->ops->solvetranspose  = MatSolve_SeqSBAIJ_1_inplace;
23730a3c351aSHong Zhang     B->ops->forwardsolve    = MatForwardSolve_SeqSBAIJ_1_inplace;
23740a3c351aSHong Zhang     B->ops->backwardsolve   = MatBackwardSolve_SeqSBAIJ_1_inplace;
2375db4efbfdSBarry Smith   }
2376db4efbfdSBarry Smith 
23773c9e8598SHong Zhang   C->assembled    = PETSC_TRUE;
23783c9e8598SHong Zhang   C->preallocated = PETSC_TRUE;
2379d0f46423SBarry Smith   ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr);
2380540703acSHong Zhang   if (sctx.nshift){
2381f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
23821e2582c4SBarry Smith       ierr = PetscInfo2(A,"number of shiftnz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
2383f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
23841e2582c4SBarry Smith       ierr = PetscInfo2(A,"number of shiftpd tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
23850697b6caSHong Zhang     }
23863c9e8598SHong Zhang   }
23873c9e8598SHong Zhang   PetscFunctionReturn(0);
23883c9e8598SHong Zhang }
2389a6175056SHong Zhang 
23908a76ed4fSHong Zhang /*
23918a76ed4fSHong Zhang    icc() under revised new data structure.
23928a76ed4fSHong Zhang    Factored arrays bj and ba are stored as
23938a76ed4fSHong Zhang      U(0,:),...,U(i,:),U(n-1,:)
23948a76ed4fSHong Zhang 
23958a76ed4fSHong Zhang    ui=fact->i is an array of size n+1, in which
23968a76ed4fSHong Zhang    ui+
23978a76ed4fSHong Zhang      ui[i]:  points to 1st entry of U(i,:),i=0,...,n-1
23988a76ed4fSHong Zhang      ui[n]:  points to U(n-1,n-1)+1
23998a76ed4fSHong Zhang 
24008a76ed4fSHong Zhang   udiag=fact->diag is an array of size n,in which
24018a76ed4fSHong Zhang      udiag[i]: points to diagonal of U(i,:), i=0,...,n-1
24028a76ed4fSHong Zhang 
24038a76ed4fSHong Zhang    U(i,:) contains udiag[i] as its last entry, i.e.,
24048a76ed4fSHong Zhang     U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i])
24058a76ed4fSHong Zhang */
24068a76ed4fSHong Zhang 
2407a6175056SHong Zhang #undef __FUNCT__
240835233d99SShri Abhyankar #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ"
240935233d99SShri Abhyankar PetscErrorCode MatICCFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
2410a6175056SHong Zhang {
24110968510aSHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
2412ed59401aSHong Zhang   Mat_SeqSBAIJ       *b;
2413dfbe8321SBarry Smith   PetscErrorCode     ierr;
2414ace3abfcSBarry Smith   PetscBool          perm_identity,missing;
24155f44c854SHong Zhang   PetscInt           reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag;
24165d0c19d7SBarry Smith   const PetscInt     *rip,*riip;
2417ed59401aSHong Zhang   PetscInt           jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow;
241858ebbce7SBarry Smith   PetscInt           nlnk,*lnk,*lnk_lvl=PETSC_NULL,d;
24195a8e39fbSHong Zhang   PetscInt           ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr;
2420ed59401aSHong Zhang   PetscReal          fill=info->fill,levels=info->levels;
2421a1a86e44SBarry Smith   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
2422a1a86e44SBarry Smith   PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL;
24237a48dd6fSHong Zhang   PetscBT            lnkbt;
2424b635d36bSHong Zhang   IS                 iperm;
2425a6175056SHong Zhang 
2426a6175056SHong Zhang   PetscFunctionBegin;
2427e32f2f54SBarry 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);
242858ebbce7SBarry Smith   ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr);
2429e32f2f54SBarry Smith   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d);
2430653a6975SHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
2431b635d36bSHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
24327a48dd6fSHong Zhang 
243339e3d630SHong Zhang   ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr);
24345f44c854SHong Zhang   ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr);
243539e3d630SHong Zhang   ui[0] = 0;
243639e3d630SHong Zhang 
24378a76ed4fSHong Zhang   /* ICC(0) without matrix ordering: simply rearrange column indices */
2438622e2028SHong Zhang   if (!levels && perm_identity) {
24395f44c854SHong Zhang     for (i=0; i<am; i++) {
2440c5546cabSHong Zhang       ncols    = ai[i+1] - a->diag[i];
2441c5546cabSHong Zhang       ui[i+1]  = ui[i] + ncols;
2442c5546cabSHong Zhang       udiag[i] = ui[i+1] - 1; /* points to the last entry of U(i,:) */
2443dc1e2a3fSHong Zhang     }
2444dc1e2a3fSHong Zhang     ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
2445dc1e2a3fSHong Zhang     cols = uj;
2446dc1e2a3fSHong Zhang     for (i=0; i<am; i++) {
24475f44c854SHong Zhang       aj    = a->j + a->diag[i] + 1; /* 1st entry of U(i,:) without diagonal */
2448dc1e2a3fSHong Zhang       ncols = ai[i+1] - a->diag[i] -1;
24495f44c854SHong Zhang       for (j=0; j<ncols; j++) *cols++ = aj[j];
24505f44c854SHong Zhang       *cols++ = i; /* diagoanl is located as the last entry of U(i,:) */
24515f44c854SHong Zhang     }
24525f44c854SHong Zhang   } else { /* case: levels>0 || (levels=0 && !perm_identity) */
24535f44c854SHong Zhang     ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
24545f44c854SHong Zhang     ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
24555f44c854SHong Zhang 
24565f44c854SHong Zhang     /* initialization */
24575f44c854SHong Zhang     ierr  = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr);
24585f44c854SHong Zhang 
24595f44c854SHong Zhang     /* jl: linked list for storing indices of the pivot rows
24605f44c854SHong Zhang        il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
24610e83c824SBarry Smith     ierr = PetscMalloc4(am,PetscInt*,&uj_ptr,am,PetscInt*,&uj_lvl_ptr,am,PetscInt,&jl,am,PetscInt,&il);CHKERRQ(ierr);
24625f44c854SHong Zhang     for (i=0; i<am; i++){
24635f44c854SHong Zhang       jl[i] = am; il[i] = 0;
24645f44c854SHong Zhang     }
24655f44c854SHong Zhang 
24665f44c854SHong Zhang     /* create and initialize a linked list for storing column indices of the active row k */
24675f44c854SHong Zhang     nlnk = am + 1;
24685f44c854SHong Zhang     ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
24695f44c854SHong Zhang 
247095b5ac22SHong Zhang     /* initial FreeSpace size is fill*(ai[am]+am)/2 */
247195b5ac22SHong Zhang     ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+am)/2),&free_space);CHKERRQ(ierr);
24725f44c854SHong Zhang     current_space = free_space;
247395b5ac22SHong Zhang     ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+am)/2),&free_space_lvl);CHKERRQ(ierr);
24745f44c854SHong Zhang     current_space_lvl = free_space_lvl;
24755f44c854SHong Zhang 
24765f44c854SHong Zhang     for (k=0; k<am; k++){  /* for each active row k */
24775f44c854SHong Zhang       /* initialize lnk by the column indices of row rip[k] of A */
24785f44c854SHong Zhang       nzk   = 0;
24795f44c854SHong Zhang       ncols = ai[rip[k]+1] - ai[rip[k]];
2480e32f2f54SBarry 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);
24815f44c854SHong Zhang       ncols_upper = 0;
24825f44c854SHong Zhang       for (j=0; j<ncols; j++){
24835f44c854SHong Zhang         i = *(aj + ai[rip[k]] + j); /* unpermuted column index */
24845f44c854SHong Zhang         if (riip[i] >= k){ /* only take upper triangular entry */
24855f44c854SHong Zhang           ajtmp[ncols_upper] = i;
24865f44c854SHong Zhang           ncols_upper++;
24875f44c854SHong Zhang         }
24885f44c854SHong Zhang       }
24895f44c854SHong Zhang       ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
24905f44c854SHong Zhang       nzk += nlnk;
24915f44c854SHong Zhang 
24925f44c854SHong Zhang       /* update lnk by computing fill-in for each pivot row to be merged in */
24935f44c854SHong Zhang       prow = jl[k]; /* 1st pivot row */
24945f44c854SHong Zhang 
24955f44c854SHong Zhang       while (prow < k){
24965f44c854SHong Zhang         nextprow = jl[prow];
24975f44c854SHong Zhang 
24985f44c854SHong Zhang         /* merge prow into k-th row */
24995f44c854SHong Zhang         jmin = il[prow] + 1;  /* index of the 2nd nzero entry in U(prow,k:am-1) */
25005f44c854SHong Zhang         jmax = ui[prow+1];
25015f44c854SHong Zhang         ncols = jmax-jmin;
25025f44c854SHong Zhang         i     = jmin - ui[prow];
25035f44c854SHong Zhang         cols  = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */
25045f44c854SHong Zhang         uj    = uj_lvl_ptr[prow] + i; /* levels of cols */
25055f44c854SHong Zhang         j     = *(uj - 1);
25065f44c854SHong Zhang         ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr);
25075f44c854SHong Zhang         nzk += nlnk;
25085f44c854SHong Zhang 
25095f44c854SHong Zhang         /* update il and jl for prow */
25105f44c854SHong Zhang         if (jmin < jmax){
25115f44c854SHong Zhang           il[prow] = jmin;
25125f44c854SHong Zhang           j = *cols; jl[prow] = jl[j]; jl[j] = prow;
25135f44c854SHong Zhang         }
25145f44c854SHong Zhang         prow = nextprow;
25155f44c854SHong Zhang       }
25165f44c854SHong Zhang 
25175f44c854SHong Zhang       /* if free space is not available, make more free space */
25185f44c854SHong Zhang       if (current_space->local_remaining<nzk) {
25195f44c854SHong Zhang         i  = am - k + 1; /* num of unfactored rows */
2520c6ae9bfcSHong Zhang         i *= PetscMin(nzk, i-1); /* i*nzk, i*(i-1): estimated and max additional space needed */
25215f44c854SHong Zhang         ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
25225f44c854SHong Zhang         ierr = PetscFreeSpaceGet(i,&current_space_lvl);CHKERRQ(ierr);
25235f44c854SHong Zhang         reallocs++;
25245f44c854SHong Zhang       }
25255f44c854SHong Zhang 
25265f44c854SHong Zhang       /* copy data into free_space and free_space_lvl, then initialize lnk */
2527e32f2f54SBarry Smith       if (nzk == 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k);
25285f44c854SHong Zhang       ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
25295f44c854SHong Zhang 
25305f44c854SHong Zhang       /* add the k-th row into il and jl */
25315f44c854SHong Zhang       if (nzk > 1){
25325f44c854SHong Zhang         i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
25335f44c854SHong Zhang         jl[k] = jl[i]; jl[i] = k;
25345f44c854SHong Zhang         il[k] = ui[k] + 1;
25355f44c854SHong Zhang       }
25365f44c854SHong Zhang       uj_ptr[k]     = current_space->array;
25375f44c854SHong Zhang       uj_lvl_ptr[k] = current_space_lvl->array;
25385f44c854SHong Zhang 
25395f44c854SHong Zhang       current_space->array           += nzk;
25405f44c854SHong Zhang       current_space->local_used      += nzk;
25415f44c854SHong Zhang       current_space->local_remaining -= nzk;
25425f44c854SHong Zhang 
25435f44c854SHong Zhang       current_space_lvl->array           += nzk;
25445f44c854SHong Zhang       current_space_lvl->local_used      += nzk;
25455f44c854SHong Zhang       current_space_lvl->local_remaining -= nzk;
25465f44c854SHong Zhang 
25475f44c854SHong Zhang       ui[k+1] = ui[k] + nzk;
25485f44c854SHong Zhang     }
25495f44c854SHong Zhang 
25505f44c854SHong Zhang     ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
25515f44c854SHong Zhang     ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
25520e83c824SBarry Smith     ierr = PetscFree4(uj_ptr,uj_lvl_ptr,jl,il);CHKERRQ(ierr);
25535f44c854SHong Zhang     ierr = PetscFree(ajtmp);CHKERRQ(ierr);
25545f44c854SHong Zhang 
25559263d837SHong Zhang     /* copy free_space into uj and free free_space; set ui, uj, udiag in new datastructure; */
25565f44c854SHong Zhang     ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
255735233d99SShri Abhyankar     ierr = PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag);CHKERRQ(ierr); /* store matrix factor  */
25585f44c854SHong Zhang     ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
25595f44c854SHong Zhang     ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
25605f44c854SHong Zhang 
25615f44c854SHong Zhang   } /* end of case: levels>0 || (levels=0 && !perm_identity) */
25625f44c854SHong Zhang 
25635f44c854SHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
25645f44c854SHong Zhang   b    = (Mat_SeqSBAIJ*)(fact)->data;
25655f44c854SHong Zhang   b->singlemalloc = PETSC_FALSE;
25665f44c854SHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr);
25675f44c854SHong Zhang   b->j    = uj;
25685f44c854SHong Zhang   b->i    = ui;
25695f44c854SHong Zhang   b->diag = udiag;
25707f53bb6cSHong Zhang   b->free_diag = PETSC_TRUE;
25715f44c854SHong Zhang   b->ilen = 0;
25725f44c854SHong Zhang   b->imax = 0;
25735f44c854SHong Zhang   b->row  = perm;
25745f44c854SHong Zhang   b->col  = perm;
25755f44c854SHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
25765f44c854SHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
25775f44c854SHong Zhang   b->icol = iperm;
25785f44c854SHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
25795f44c854SHong Zhang   ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
2580d595f711SHong Zhang   ierr = PetscLogObjectMemory(fact,ui[am]*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
25815f44c854SHong Zhang   b->maxnz   = b->nz = ui[am];
25825f44c854SHong Zhang   b->free_a  = PETSC_TRUE;
25835f44c854SHong Zhang   b->free_ij = PETSC_TRUE;
25845f44c854SHong Zhang 
2585f284f12aSHong Zhang   fact->info.factor_mallocs   = reallocs;
2586f284f12aSHong Zhang   fact->info.fill_ratio_given = fill;
25875f44c854SHong Zhang   if (ai[am] != 0) {
25886a69fef8SHong Zhang     /* nonzeros in lower triangular part of A (including diagonals) = (ai[am]+am)/2 */
258995b5ac22SHong Zhang     fact->info.fill_ratio_needed = ((PetscReal)2*ui[am])/(ai[am]+am);
25905f44c854SHong Zhang   } else {
2591f284f12aSHong Zhang     fact->info.fill_ratio_needed = 0.0;
25925f44c854SHong Zhang   }
25939263d837SHong Zhang #if defined(PETSC_USE_INFO)
25949263d837SHong Zhang     if (ai[am] != 0) {
25959263d837SHong Zhang       PetscReal af = fact->info.fill_ratio_needed;
25969263d837SHong Zhang       ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr);
25979263d837SHong Zhang       ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
25989263d837SHong Zhang       ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr);
25999263d837SHong Zhang     } else {
26009263d837SHong Zhang       ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr);
26019263d837SHong Zhang     }
26029263d837SHong Zhang #endif
260335233d99SShri Abhyankar   fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ;
26045f44c854SHong Zhang   PetscFunctionReturn(0);
26055f44c854SHong Zhang }
26065f44c854SHong Zhang 
26075f44c854SHong Zhang #undef __FUNCT__
2608ad04f41aSHong Zhang #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ_inplace"
2609ad04f41aSHong Zhang PetscErrorCode MatICCFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
26105f44c854SHong Zhang {
26115f44c854SHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
26125f44c854SHong Zhang   Mat_SeqSBAIJ       *b;
26135f44c854SHong Zhang   PetscErrorCode     ierr;
2614ace3abfcSBarry Smith   PetscBool          perm_identity,missing;
26155f44c854SHong Zhang   PetscInt           reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag;
26165f44c854SHong Zhang   const PetscInt     *rip,*riip;
26175f44c854SHong Zhang   PetscInt           jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow;
26185f44c854SHong Zhang   PetscInt           nlnk,*lnk,*lnk_lvl=PETSC_NULL,d;
26195f44c854SHong Zhang   PetscInt           ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr;
26205f44c854SHong Zhang   PetscReal          fill=info->fill,levels=info->levels;
26215f44c854SHong Zhang   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
26225f44c854SHong Zhang   PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL;
26235f44c854SHong Zhang   PetscBT            lnkbt;
26245f44c854SHong Zhang   IS                 iperm;
26255f44c854SHong Zhang 
26265f44c854SHong Zhang   PetscFunctionBegin;
2627e32f2f54SBarry 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);
26285f44c854SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr);
2629e32f2f54SBarry Smith   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d);
26305f44c854SHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
26315f44c854SHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
26325f44c854SHong Zhang 
26335f44c854SHong Zhang   ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr);
26345f44c854SHong Zhang   ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr);
26355f44c854SHong Zhang   ui[0] = 0;
26365f44c854SHong Zhang 
26375f44c854SHong Zhang   /* ICC(0) without matrix ordering: simply copies fill pattern */
26385f44c854SHong Zhang   if (!levels && perm_identity) {
26395f44c854SHong Zhang 
26405f44c854SHong Zhang     for (i=0; i<am; i++) {
26415f44c854SHong Zhang       ui[i+1]  = ui[i] + ai[i+1] - a->diag[i];
26425f44c854SHong Zhang       udiag[i] = ui[i];
2643ed59401aSHong Zhang     }
264439e3d630SHong Zhang     ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
264539e3d630SHong Zhang     cols = uj;
2646ed59401aSHong Zhang     for (i=0; i<am; i++) {
2647ed59401aSHong Zhang       aj    = a->j + a->diag[i];
264839e3d630SHong Zhang       ncols = ui[i+1] - ui[i];
264939e3d630SHong Zhang       for (j=0; j<ncols; j++) *cols++ = *aj++;
2650ed59401aSHong Zhang     }
265139e3d630SHong Zhang   } else { /* case: levels>0 || (levels=0 && !perm_identity) */
2652b635d36bSHong Zhang     ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
2653b635d36bSHong Zhang     ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
2654b635d36bSHong Zhang 
26557a48dd6fSHong Zhang     /* initialization */
26565a8e39fbSHong Zhang     ierr  = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr);
26577a48dd6fSHong Zhang 
26587a48dd6fSHong Zhang     /* jl: linked list for storing indices of the pivot rows
26597a48dd6fSHong Zhang        il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
26600e83c824SBarry Smith     ierr = PetscMalloc4(am,PetscInt*,&uj_ptr,am,PetscInt*,&uj_lvl_ptr,am,PetscInt,&jl,am,PetscInt,&il);CHKERRQ(ierr);
26617a48dd6fSHong Zhang     for (i=0; i<am; i++){
26627a48dd6fSHong Zhang       jl[i] = am; il[i] = 0;
26637a48dd6fSHong Zhang     }
26647a48dd6fSHong Zhang 
26657a48dd6fSHong Zhang     /* create and initialize a linked list for storing column indices of the active row k */
26667a48dd6fSHong Zhang     nlnk = am + 1;
26672ed133dbSHong Zhang     ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
26687a48dd6fSHong Zhang 
26697a48dd6fSHong Zhang     /* initial FreeSpace size is fill*(ai[am]+1) */
2670a1a86e44SBarry Smith     ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr);
26717a48dd6fSHong Zhang     current_space = free_space;
2672a1a86e44SBarry Smith     ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space_lvl);CHKERRQ(ierr);
26737a48dd6fSHong Zhang     current_space_lvl = free_space_lvl;
26747a48dd6fSHong Zhang 
26757a48dd6fSHong Zhang     for (k=0; k<am; k++){  /* for each active row k */
26767a48dd6fSHong Zhang       /* initialize lnk by the column indices of row rip[k] of A */
26777a48dd6fSHong Zhang       nzk   = 0;
2678622e2028SHong Zhang       ncols = ai[rip[k]+1] - ai[rip[k]];
2679e32f2f54SBarry 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);
2680622e2028SHong Zhang       ncols_upper = 0;
2681622e2028SHong Zhang       for (j=0; j<ncols; j++){
2682b635d36bSHong Zhang         i = *(aj + ai[rip[k]] + j); /* unpermuted column index */
2683b635d36bSHong Zhang         if (riip[i] >= k){ /* only take upper triangular entry */
26845a8e39fbSHong Zhang           ajtmp[ncols_upper] = i;
2685622e2028SHong Zhang           ncols_upper++;
2686622e2028SHong Zhang         }
2687622e2028SHong Zhang       }
2688b635d36bSHong Zhang       ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
26897a48dd6fSHong Zhang       nzk += nlnk;
26907a48dd6fSHong Zhang 
26917a48dd6fSHong Zhang       /* update lnk by computing fill-in for each pivot row to be merged in */
26927a48dd6fSHong Zhang       prow = jl[k]; /* 1st pivot row */
26937a48dd6fSHong Zhang 
26947a48dd6fSHong Zhang       while (prow < k){
26957a48dd6fSHong Zhang         nextprow = jl[prow];
26967a48dd6fSHong Zhang 
26977a48dd6fSHong Zhang         /* merge prow into k-th row */
26987a48dd6fSHong Zhang         jmin = il[prow] + 1;  /* index of the 2nd nzero entry in U(prow,k:am-1) */
26997a48dd6fSHong Zhang         jmax = ui[prow+1];
27007a48dd6fSHong Zhang         ncols = jmax-jmin;
2701ed59401aSHong Zhang         i     = jmin - ui[prow];
2702ed59401aSHong Zhang         cols  = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */
27035a8e39fbSHong Zhang         uj    = uj_lvl_ptr[prow] + i; /* levels of cols */
27045a8e39fbSHong Zhang         j     = *(uj - 1);
27055a8e39fbSHong Zhang         ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr);
27067a48dd6fSHong Zhang         nzk += nlnk;
27077a48dd6fSHong Zhang 
27087a48dd6fSHong Zhang         /* update il and jl for prow */
27097a48dd6fSHong Zhang         if (jmin < jmax){
27107a48dd6fSHong Zhang           il[prow] = jmin;
27117a48dd6fSHong Zhang           j = *cols; jl[prow] = jl[j]; jl[j] = prow;
27127a48dd6fSHong Zhang         }
27137a48dd6fSHong Zhang         prow = nextprow;
27147a48dd6fSHong Zhang       }
27157a48dd6fSHong Zhang 
27167a48dd6fSHong Zhang       /* if free space is not available, make more free space */
27177a48dd6fSHong Zhang       if (current_space->local_remaining<nzk) {
27187a48dd6fSHong Zhang         i = am - k + 1; /* num of unfactored rows */
2719fd3f55cbSMatthew Knepley         i *= PetscMin(nzk, (i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */
2720a1a86e44SBarry Smith         ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
2721a1a86e44SBarry Smith         ierr = PetscFreeSpaceGet(i,&current_space_lvl);CHKERRQ(ierr);
27227a48dd6fSHong Zhang         reallocs++;
27237a48dd6fSHong Zhang       }
27247a48dd6fSHong Zhang 
27257a48dd6fSHong Zhang       /* copy data into free_space and free_space_lvl, then initialize lnk */
272665e19b50SBarry Smith       if (!nzk) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k);
27272ed133dbSHong Zhang       ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
27287a48dd6fSHong Zhang 
27297a48dd6fSHong Zhang       /* add the k-th row into il and jl */
273039e3d630SHong Zhang       if (nzk > 1){
27317a48dd6fSHong Zhang         i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
27327a48dd6fSHong Zhang         jl[k] = jl[i]; jl[i] = k;
27337a48dd6fSHong Zhang         il[k] = ui[k] + 1;
27347a48dd6fSHong Zhang       }
27357a48dd6fSHong Zhang       uj_ptr[k]     = current_space->array;
27367a48dd6fSHong Zhang       uj_lvl_ptr[k] = current_space_lvl->array;
27377a48dd6fSHong Zhang 
27387a48dd6fSHong Zhang       current_space->array           += nzk;
27397a48dd6fSHong Zhang       current_space->local_used      += nzk;
27407a48dd6fSHong Zhang       current_space->local_remaining -= nzk;
27417a48dd6fSHong Zhang 
27427a48dd6fSHong Zhang       current_space_lvl->array           += nzk;
27437a48dd6fSHong Zhang       current_space_lvl->local_used      += nzk;
27447a48dd6fSHong Zhang       current_space_lvl->local_remaining -= nzk;
27457a48dd6fSHong Zhang 
27467a48dd6fSHong Zhang       ui[k+1] = ui[k] + nzk;
27477a48dd6fSHong Zhang     }
27487a48dd6fSHong Zhang 
27496cf91177SBarry Smith #if defined(PETSC_USE_INFO)
27507a48dd6fSHong Zhang     if (ai[am] != 0) {
275139e3d630SHong Zhang       PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]);
2752ae15b995SBarry Smith       ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr);
2753ae15b995SBarry Smith       ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
2754ae15b995SBarry Smith       ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr);
27557a48dd6fSHong Zhang     } else {
2756ae15b995SBarry Smith       ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr);
27577a48dd6fSHong Zhang     }
275863ba0a88SBarry Smith #endif
27597a48dd6fSHong Zhang 
27607a48dd6fSHong Zhang     ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
2761b635d36bSHong Zhang     ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
27620e83c824SBarry Smith     ierr = PetscFree4(uj_ptr,uj_lvl_ptr,jl,il);CHKERRQ(ierr);
27635a8e39fbSHong Zhang     ierr = PetscFree(ajtmp);CHKERRQ(ierr);
27647a48dd6fSHong Zhang 
27657a48dd6fSHong Zhang     /* destroy list of free space and other temporary array(s) */
27667a48dd6fSHong Zhang     ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
2767a1a86e44SBarry Smith     ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr);
27682ed133dbSHong Zhang     ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
2769a1a86e44SBarry Smith     ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
27707a48dd6fSHong Zhang 
277139e3d630SHong Zhang   } /* end of case: levels>0 || (levels=0 && !perm_identity) */
277239e3d630SHong Zhang 
27737a48dd6fSHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
27747a48dd6fSHong Zhang 
2775f284f12aSHong Zhang   b    = (Mat_SeqSBAIJ*)fact->data;
27767a48dd6fSHong Zhang   b->singlemalloc = PETSC_FALSE;
27777a48dd6fSHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr);
27787a48dd6fSHong Zhang   b->j    = uj;
27797a48dd6fSHong Zhang   b->i    = ui;
27805f44c854SHong Zhang   b->diag = udiag;
27817f53bb6cSHong Zhang   b->free_diag = PETSC_TRUE;
27827a48dd6fSHong Zhang   b->ilen = 0;
27837a48dd6fSHong Zhang   b->imax = 0;
27847a48dd6fSHong Zhang   b->row  = perm;
2785b635d36bSHong Zhang   b->col  = perm;
2786b635d36bSHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
2787b635d36bSHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
2788b635d36bSHong Zhang   b->icol = iperm;
27897a48dd6fSHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
27907a48dd6fSHong Zhang   ierr    = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
2791f284f12aSHong Zhang   ierr = PetscLogObjectMemory(fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
27927a48dd6fSHong Zhang   b->maxnz   = b->nz = ui[am];
2793e6b907acSBarry Smith   b->free_a  = PETSC_TRUE;
2794e6b907acSBarry Smith   b->free_ij = PETSC_TRUE;
27957a48dd6fSHong Zhang 
2796f284f12aSHong Zhang   fact->info.factor_mallocs    = reallocs;
2797f284f12aSHong Zhang   fact->info.fill_ratio_given  = fill;
27987a48dd6fSHong Zhang   if (ai[am] != 0) {
2799f284f12aSHong Zhang     fact->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]);
28007a48dd6fSHong Zhang   } else {
2801f284f12aSHong Zhang     fact->info.fill_ratio_needed = 0.0;
28027a48dd6fSHong Zhang   }
28030a3c351aSHong Zhang   fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_inplace;
2804a6175056SHong Zhang   PetscFunctionReturn(0);
2805a6175056SHong Zhang }
2806d5d45c9bSBarry Smith 
2807177d4faaSHong Zhang #undef __FUNCT__
2808177d4faaSHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_SeqAIJ"
280935233d99SShri Abhyankar PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
28100be760fbSHong Zhang {
28110be760fbSHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
28120be760fbSHong Zhang   Mat_SeqSBAIJ       *b;
28130be760fbSHong Zhang   PetscErrorCode     ierr;
2814ace3abfcSBarry Smith   PetscBool          perm_identity;
28150be760fbSHong Zhang   PetscReal          fill = info->fill;
28160be760fbSHong Zhang   const PetscInt     *rip,*riip;
28170be760fbSHong Zhang   PetscInt           i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow;
28180be760fbSHong Zhang   PetscInt           *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow;
28190be760fbSHong Zhang   PetscInt           nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr,*udiag;
28200be760fbSHong Zhang   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
28210be760fbSHong Zhang   PetscBT            lnkbt;
28220be760fbSHong Zhang   IS                 iperm;
28230be760fbSHong Zhang 
28240be760fbSHong Zhang   PetscFunctionBegin;
2825e32f2f54SBarry 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);
28260be760fbSHong Zhang   /* check whether perm is the identity mapping */
28270be760fbSHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
28280be760fbSHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
28290be760fbSHong Zhang   ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
28300be760fbSHong Zhang   ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
28310be760fbSHong Zhang 
28320be760fbSHong Zhang   /* initialization */
28330be760fbSHong Zhang   ierr  = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr);
28340be760fbSHong Zhang   ierr  = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr);
28350be760fbSHong Zhang   ui[0] = 0;
28360be760fbSHong Zhang 
28370be760fbSHong Zhang   /* jl: linked list for storing indices of the pivot rows
28380be760fbSHong Zhang      il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
28390be760fbSHong Zhang   ierr = PetscMalloc4(am,PetscInt*,&ui_ptr,am,PetscInt,&jl,am,PetscInt,&il,am,PetscInt,&cols);CHKERRQ(ierr);
28400be760fbSHong Zhang   for (i=0; i<am; i++){
28410be760fbSHong Zhang     jl[i] = am; il[i] = 0;
28420be760fbSHong Zhang   }
28430be760fbSHong Zhang 
28440be760fbSHong Zhang   /* create and initialize a linked list for storing column indices of the active row k */
28450be760fbSHong Zhang   nlnk = am + 1;
28460be760fbSHong Zhang   ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
28470be760fbSHong Zhang 
284895b5ac22SHong Zhang   /* initial FreeSpace size is fill*(ai[am]+am)/2 */
284995b5ac22SHong Zhang   ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+am)/2),&free_space);CHKERRQ(ierr);
28500be760fbSHong Zhang   current_space = free_space;
28510be760fbSHong Zhang 
28520be760fbSHong Zhang   for (k=0; k<am; k++){  /* for each active row k */
28530be760fbSHong Zhang     /* initialize lnk by the column indices of row rip[k] of A */
28540be760fbSHong Zhang     nzk   = 0;
28550be760fbSHong Zhang     ncols = ai[rip[k]+1] - ai[rip[k]];
2856e32f2f54SBarry 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);
28570be760fbSHong Zhang     ncols_upper = 0;
28580be760fbSHong Zhang     for (j=0; j<ncols; j++){
28590be760fbSHong Zhang       i = riip[*(aj + ai[rip[k]] + j)];
28600be760fbSHong Zhang       if (i >= k){ /* only take upper triangular entry */
28610be760fbSHong Zhang         cols[ncols_upper] = i;
28620be760fbSHong Zhang         ncols_upper++;
28630be760fbSHong Zhang       }
28640be760fbSHong Zhang     }
28650be760fbSHong Zhang     ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
28660be760fbSHong Zhang     nzk += nlnk;
28670be760fbSHong Zhang 
28680be760fbSHong Zhang     /* update lnk by computing fill-in for each pivot row to be merged in */
28690be760fbSHong Zhang     prow = jl[k]; /* 1st pivot row */
28700be760fbSHong Zhang 
28710be760fbSHong Zhang     while (prow < k){
28720be760fbSHong Zhang       nextprow = jl[prow];
28730be760fbSHong Zhang       /* merge prow into k-th row */
28740be760fbSHong Zhang       jmin = il[prow] + 1;  /* index of the 2nd nzero entry in U(prow,k:am-1) */
28750be760fbSHong Zhang       jmax = ui[prow+1];
28760be760fbSHong Zhang       ncols = jmax-jmin;
28770be760fbSHong Zhang       uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */
28780be760fbSHong Zhang       ierr = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
28790be760fbSHong Zhang       nzk += nlnk;
28800be760fbSHong Zhang 
28810be760fbSHong Zhang       /* update il and jl for prow */
28820be760fbSHong Zhang       if (jmin < jmax){
28830be760fbSHong Zhang         il[prow] = jmin;
28840be760fbSHong Zhang         j = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow;
28850be760fbSHong Zhang       }
28860be760fbSHong Zhang       prow = nextprow;
28870be760fbSHong Zhang     }
28880be760fbSHong Zhang 
28890be760fbSHong Zhang     /* if free space is not available, make more free space */
28900be760fbSHong Zhang     if (current_space->local_remaining<nzk) {
28910be760fbSHong Zhang       i  = am - k + 1; /* num of unfactored rows */
2892c6ae9bfcSHong Zhang       i *= PetscMin(nzk,i-1); /* i*nzk, i*(i-1): estimated and max additional space needed */
28930be760fbSHong Zhang       ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
28940be760fbSHong Zhang       reallocs++;
28950be760fbSHong Zhang     }
28960be760fbSHong Zhang 
28970be760fbSHong Zhang     /* copy data into free space, then initialize lnk */
28980be760fbSHong Zhang     ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
28990be760fbSHong Zhang 
29000be760fbSHong Zhang     /* add the k-th row into il and jl */
29017b056e98SHong Zhang     if (nzk > 1){
29020be760fbSHong Zhang       i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
29030be760fbSHong Zhang       jl[k] = jl[i]; jl[i] = k;
29040be760fbSHong Zhang       il[k] = ui[k] + 1;
29050be760fbSHong Zhang     }
29060be760fbSHong Zhang     ui_ptr[k] = current_space->array;
29070be760fbSHong Zhang     current_space->array           += nzk;
29080be760fbSHong Zhang     current_space->local_used      += nzk;
29090be760fbSHong Zhang     current_space->local_remaining -= nzk;
29100be760fbSHong Zhang 
29110be760fbSHong Zhang     ui[k+1] = ui[k] + nzk;
29120be760fbSHong Zhang   }
29130be760fbSHong Zhang 
29140be760fbSHong Zhang   ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
29150be760fbSHong Zhang   ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
29160be760fbSHong Zhang   ierr = PetscFree4(ui_ptr,jl,il,cols);CHKERRQ(ierr);
29170be760fbSHong Zhang 
29189263d837SHong Zhang   /* copy free_space into uj and free free_space; set ui, uj, udiag in new datastructure; */
29190be760fbSHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
292035233d99SShri Abhyankar   ierr = PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag);CHKERRQ(ierr); /* store matrix factor */
29210be760fbSHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
29220be760fbSHong Zhang 
29230be760fbSHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
29240be760fbSHong Zhang 
2925f284f12aSHong Zhang   b = (Mat_SeqSBAIJ*)fact->data;
29260be760fbSHong Zhang   b->singlemalloc = PETSC_FALSE;
29270be760fbSHong Zhang   b->free_a       = PETSC_TRUE;
29280be760fbSHong Zhang   b->free_ij      = PETSC_TRUE;
29290be760fbSHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr);
29300be760fbSHong Zhang   b->j    = uj;
29310be760fbSHong Zhang   b->i    = ui;
29320be760fbSHong Zhang   b->diag = udiag;
29330be760fbSHong Zhang   b->free_diag = PETSC_TRUE;
29340be760fbSHong Zhang   b->ilen = 0;
29350be760fbSHong Zhang   b->imax = 0;
29360be760fbSHong Zhang   b->row  = perm;
29370be760fbSHong Zhang   b->col  = perm;
29380be760fbSHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
29390be760fbSHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
29400be760fbSHong Zhang   b->icol = iperm;
29410be760fbSHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
29420be760fbSHong Zhang   ierr    = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
29430860e08cSHong Zhang   ierr    = PetscLogObjectMemory(fact,ui[am]*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
29440be760fbSHong Zhang   b->maxnz = b->nz = ui[am];
29450be760fbSHong Zhang 
2946f284f12aSHong Zhang   fact->info.factor_mallocs    = reallocs;
2947f284f12aSHong Zhang   fact->info.fill_ratio_given  = fill;
29480be760fbSHong Zhang   if (ai[am] != 0) {
29496a69fef8SHong Zhang     /* nonzeros in lower triangular part of A (including diagonals) = (ai[am]+am)/2 */
295095b5ac22SHong Zhang     fact->info.fill_ratio_needed = ((PetscReal)2*ui[am])/(ai[am]+am);
29510be760fbSHong Zhang   } else {
2952f284f12aSHong Zhang     fact->info.fill_ratio_needed = 0.0;
29530be760fbSHong Zhang   }
29549263d837SHong Zhang #if defined(PETSC_USE_INFO)
29559263d837SHong Zhang   if (ai[am] != 0) {
29569263d837SHong Zhang     PetscReal af = fact->info.fill_ratio_needed;
29579263d837SHong Zhang     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr);
29589263d837SHong Zhang     ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
29599263d837SHong Zhang     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr);
29609263d837SHong Zhang   } else {
29619263d837SHong Zhang      ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr);
29629263d837SHong Zhang   }
29639263d837SHong Zhang #endif
296435233d99SShri Abhyankar   fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ;
29650be760fbSHong Zhang   PetscFunctionReturn(0);
29660be760fbSHong Zhang }
29670be760fbSHong Zhang 
2968f76d2b81SHong Zhang #undef __FUNCT__
2969ad04f41aSHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_SeqAIJ_inplace"
2970ad04f41aSHong Zhang PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
2971f76d2b81SHong Zhang {
2972f76d2b81SHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
297310c27e3dSHong Zhang   Mat_SeqSBAIJ       *b;
2974dfbe8321SBarry Smith   PetscErrorCode     ierr;
2975ace3abfcSBarry Smith   PetscBool          perm_identity;
297610c27e3dSHong Zhang   PetscReal          fill = info->fill;
29775d0c19d7SBarry Smith   const PetscInt     *rip,*riip;
29785d0c19d7SBarry Smith   PetscInt           i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow;
297910c27e3dSHong Zhang   PetscInt           *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow;
29802ed133dbSHong Zhang   PetscInt           nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr;
2981a1a86e44SBarry Smith   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
298210c27e3dSHong Zhang   PetscBT            lnkbt;
29832ed133dbSHong Zhang   IS                 iperm;
2984f76d2b81SHong Zhang 
2985f76d2b81SHong Zhang   PetscFunctionBegin;
2986e32f2f54SBarry 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);
29872ed133dbSHong Zhang   /* check whether perm is the identity mapping */
2988f76d2b81SHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
29892ed133dbSHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
29902ed133dbSHong Zhang   ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
29919bfd6278SHong Zhang   ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
299210c27e3dSHong Zhang 
299310c27e3dSHong Zhang   /* initialization */
29941393bc97SHong Zhang   ierr  = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr);
299510c27e3dSHong Zhang   ui[0] = 0;
299610c27e3dSHong Zhang 
299710c27e3dSHong Zhang   /* jl: linked list for storing indices of the pivot rows
29981393bc97SHong Zhang      il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
29990e83c824SBarry Smith   ierr = PetscMalloc4(am,PetscInt*,&ui_ptr,am,PetscInt,&jl,am,PetscInt,&il,am,PetscInt,&cols);CHKERRQ(ierr);
30001393bc97SHong Zhang   for (i=0; i<am; i++){
30011393bc97SHong Zhang     jl[i] = am; il[i] = 0;
3002f76d2b81SHong Zhang   }
300310c27e3dSHong Zhang 
300410c27e3dSHong Zhang   /* create and initialize a linked list for storing column indices of the active row k */
30051393bc97SHong Zhang   nlnk = am + 1;
30061393bc97SHong Zhang   ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
300710c27e3dSHong Zhang 
30081393bc97SHong Zhang   /* initial FreeSpace size is fill*(ai[am]+1) */
3009a1a86e44SBarry Smith   ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr);
301010c27e3dSHong Zhang   current_space = free_space;
301110c27e3dSHong Zhang 
30121393bc97SHong Zhang   for (k=0; k<am; k++){  /* for each active row k */
301310c27e3dSHong Zhang     /* initialize lnk by the column indices of row rip[k] of A */
301410c27e3dSHong Zhang     nzk   = 0;
30152ed133dbSHong Zhang     ncols = ai[rip[k]+1] - ai[rip[k]];
3016e32f2f54SBarry 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);
30172ed133dbSHong Zhang     ncols_upper = 0;
30182ed133dbSHong Zhang     for (j=0; j<ncols; j++){
30199bfd6278SHong Zhang       i = riip[*(aj + ai[rip[k]] + j)];
30202ed133dbSHong Zhang       if (i >= k){ /* only take upper triangular entry */
30212ed133dbSHong Zhang         cols[ncols_upper] = i;
30222ed133dbSHong Zhang         ncols_upper++;
30232ed133dbSHong Zhang       }
30242ed133dbSHong Zhang     }
30251393bc97SHong Zhang     ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
302610c27e3dSHong Zhang     nzk += nlnk;
302710c27e3dSHong Zhang 
302810c27e3dSHong Zhang     /* update lnk by computing fill-in for each pivot row to be merged in */
302910c27e3dSHong Zhang     prow = jl[k]; /* 1st pivot row */
303010c27e3dSHong Zhang 
303110c27e3dSHong Zhang     while (prow < k){
303210c27e3dSHong Zhang       nextprow = jl[prow];
303310c27e3dSHong Zhang       /* merge prow into k-th row */
30341393bc97SHong Zhang       jmin = il[prow] + 1;  /* index of the 2nd nzero entry in U(prow,k:am-1) */
303510c27e3dSHong Zhang       jmax = ui[prow+1];
303610c27e3dSHong Zhang       ncols = jmax-jmin;
30371393bc97SHong Zhang       uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */
30385a8e39fbSHong Zhang       ierr = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
303910c27e3dSHong Zhang       nzk += nlnk;
304010c27e3dSHong Zhang 
304110c27e3dSHong Zhang       /* update il and jl for prow */
304210c27e3dSHong Zhang       if (jmin < jmax){
304310c27e3dSHong Zhang         il[prow] = jmin;
30442ed133dbSHong Zhang         j = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow;
304510c27e3dSHong Zhang       }
304610c27e3dSHong Zhang       prow = nextprow;
304710c27e3dSHong Zhang     }
304810c27e3dSHong Zhang 
304910c27e3dSHong Zhang     /* if free space is not available, make more free space */
305010c27e3dSHong Zhang     if (current_space->local_remaining<nzk) {
30511393bc97SHong Zhang       i = am - k + 1; /* num of unfactored rows */
305210c27e3dSHong Zhang       i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */
3053a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
305410c27e3dSHong Zhang       reallocs++;
305510c27e3dSHong Zhang     }
305610c27e3dSHong Zhang 
305710c27e3dSHong Zhang     /* copy data into free space, then initialize lnk */
30581393bc97SHong Zhang     ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
305910c27e3dSHong Zhang 
306010c27e3dSHong Zhang     /* add the k-th row into il and jl */
306110c27e3dSHong Zhang     if (nzk-1 > 0){
30621393bc97SHong Zhang       i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
306310c27e3dSHong Zhang       jl[k] = jl[i]; jl[i] = k;
306410c27e3dSHong Zhang       il[k] = ui[k] + 1;
306510c27e3dSHong Zhang     }
30662ed133dbSHong Zhang     ui_ptr[k] = current_space->array;
306710c27e3dSHong Zhang     current_space->array           += nzk;
306810c27e3dSHong Zhang     current_space->local_used      += nzk;
306910c27e3dSHong Zhang     current_space->local_remaining -= nzk;
307010c27e3dSHong Zhang 
307110c27e3dSHong Zhang     ui[k+1] = ui[k] + nzk;
307210c27e3dSHong Zhang   }
307310c27e3dSHong Zhang 
30746cf91177SBarry Smith #if defined(PETSC_USE_INFO)
30751393bc97SHong Zhang   if (ai[am] != 0) {
30761393bc97SHong Zhang     PetscReal af = (PetscReal)(ui[am])/((PetscReal)ai[am]);
3077ae15b995SBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr);
3078ae15b995SBarry Smith     ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
3079ae15b995SBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr);
308010c27e3dSHong Zhang   } else {
3081ae15b995SBarry Smith      ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr);
308210c27e3dSHong Zhang   }
308363ba0a88SBarry Smith #endif
308410c27e3dSHong Zhang 
308510c27e3dSHong Zhang   ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
30869bfd6278SHong Zhang   ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
30870e83c824SBarry Smith   ierr = PetscFree4(ui_ptr,jl,il,cols);CHKERRQ(ierr);
308810c27e3dSHong Zhang 
308910c27e3dSHong Zhang   /* destroy list of free space and other temporary array(s) */
30901393bc97SHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
3091a1a86e44SBarry Smith   ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr);
309210c27e3dSHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
309310c27e3dSHong Zhang 
309410c27e3dSHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
309510c27e3dSHong Zhang 
3096f284f12aSHong Zhang   b = (Mat_SeqSBAIJ*)fact->data;
309710c27e3dSHong Zhang   b->singlemalloc = PETSC_FALSE;
3098e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
3099e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
31001393bc97SHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr);
310110c27e3dSHong Zhang   b->j    = uj;
310210c27e3dSHong Zhang   b->i    = ui;
310310c27e3dSHong Zhang   b->diag = 0;
310410c27e3dSHong Zhang   b->ilen = 0;
310510c27e3dSHong Zhang   b->imax = 0;
310610c27e3dSHong Zhang   b->row  = perm;
31079bfd6278SHong Zhang   b->col  = perm;
31089bfd6278SHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
31099bfd6278SHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
31109bfd6278SHong Zhang   b->icol = iperm;
311110c27e3dSHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
31121393bc97SHong Zhang   ierr    = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
3113719d5645SBarry Smith   ierr    = PetscLogObjectMemory(fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
31141393bc97SHong Zhang   b->maxnz = b->nz = ui[am];
311510c27e3dSHong Zhang 
3116f284f12aSHong Zhang   fact->info.factor_mallocs    = reallocs;
3117f284f12aSHong Zhang   fact->info.fill_ratio_given  = fill;
31181393bc97SHong Zhang   if (ai[am] != 0) {
3119f284f12aSHong Zhang     fact->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]);
312010c27e3dSHong Zhang   } else {
3121f284f12aSHong Zhang     fact->info.fill_ratio_needed = 0.0;
312210c27e3dSHong Zhang   }
31230a3c351aSHong Zhang   fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_inplace;
3124f76d2b81SHong Zhang   PetscFunctionReturn(0);
3125f76d2b81SHong Zhang }
3126599ef60dSHong Zhang 
31271d098868SHong Zhang #undef __FUNCT__
312835233d99SShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering"
312935233d99SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering(Mat A,Vec bb,Vec xx)
3130813a1f2bSShri Abhyankar {
3131813a1f2bSShri Abhyankar   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
3132813a1f2bSShri Abhyankar   PetscErrorCode    ierr;
3133813a1f2bSShri Abhyankar   PetscInt          n = A->rmap->n;
3134813a1f2bSShri Abhyankar   const PetscInt    *ai = a->i,*aj = a->j,*adiag = a->diag,*vi;
3135813a1f2bSShri Abhyankar   PetscScalar       *x,sum;
3136813a1f2bSShri Abhyankar   const PetscScalar *b;
3137813a1f2bSShri Abhyankar   const MatScalar   *aa = a->a,*v;
3138813a1f2bSShri Abhyankar   PetscInt          i,nz;
3139813a1f2bSShri Abhyankar 
3140813a1f2bSShri Abhyankar   PetscFunctionBegin;
3141813a1f2bSShri Abhyankar   if (!n) PetscFunctionReturn(0);
3142813a1f2bSShri Abhyankar 
31433649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
3144813a1f2bSShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
3145813a1f2bSShri Abhyankar 
3146813a1f2bSShri Abhyankar   /* forward solve the lower triangular */
3147813a1f2bSShri Abhyankar   x[0] = b[0];
3148813a1f2bSShri Abhyankar   v    = aa;
3149813a1f2bSShri Abhyankar   vi   = aj;
3150813a1f2bSShri Abhyankar   for (i=1; i<n; i++) {
3151813a1f2bSShri Abhyankar     nz  = ai[i+1] - ai[i];
3152813a1f2bSShri Abhyankar     sum = b[i];
3153813a1f2bSShri Abhyankar     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
3154813a1f2bSShri Abhyankar     v  += nz;
3155813a1f2bSShri Abhyankar     vi += nz;
3156813a1f2bSShri Abhyankar     x[i] = sum;
3157813a1f2bSShri Abhyankar   }
3158813a1f2bSShri Abhyankar 
3159813a1f2bSShri Abhyankar   /* backward solve the upper triangular */
316062fbd6afSShri Abhyankar   for (i=n-1; i>=0; i--){
31613c0119dfSShri Abhyankar     v   = aa + adiag[i+1] + 1;
31623c0119dfSShri Abhyankar     vi  = aj + adiag[i+1] + 1;
3163813a1f2bSShri Abhyankar     nz = adiag[i] - adiag[i+1]-1;
3164813a1f2bSShri Abhyankar     sum = x[i];
3165813a1f2bSShri Abhyankar     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
31663c0119dfSShri Abhyankar     x[i] = sum*v[nz]; /* x[i]=aa[adiag[i]]*sum; v++; */
3167813a1f2bSShri Abhyankar   }
3168813a1f2bSShri Abhyankar 
3169813a1f2bSShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
31703649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
3171813a1f2bSShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
3172813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
3173813a1f2bSShri Abhyankar }
3174813a1f2bSShri Abhyankar 
31751d098868SHong Zhang #undef __FUNCT__
317635233d99SShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ"
317735233d99SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ(Mat A,Vec bb,Vec xx)
3178f268cba8SShri Abhyankar {
3179f268cba8SShri Abhyankar   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
3180f268cba8SShri Abhyankar   IS                iscol = a->col,isrow = a->row;
3181f268cba8SShri Abhyankar   PetscErrorCode    ierr;
3182f268cba8SShri Abhyankar   PetscInt          i,n=A->rmap->n,*vi,*ai=a->i,*aj=a->j,*adiag = a->diag,nz;
3183f268cba8SShri Abhyankar   const PetscInt    *rout,*cout,*r,*c;
3184f268cba8SShri Abhyankar   PetscScalar       *x,*tmp,sum;
3185f268cba8SShri Abhyankar   const PetscScalar *b;
3186f268cba8SShri Abhyankar   const MatScalar   *aa = a->a,*v;
3187f268cba8SShri Abhyankar 
3188f268cba8SShri Abhyankar   PetscFunctionBegin;
3189f268cba8SShri Abhyankar   if (!n) PetscFunctionReturn(0);
3190f268cba8SShri Abhyankar 
31913649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
3192f268cba8SShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
3193f268cba8SShri Abhyankar   tmp  = a->solve_work;
3194f268cba8SShri Abhyankar 
3195f268cba8SShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
3196f268cba8SShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
3197f268cba8SShri Abhyankar 
3198f268cba8SShri Abhyankar   /* forward solve the lower triangular */
3199f268cba8SShri Abhyankar   tmp[0] = b[r[0]];
3200f268cba8SShri Abhyankar   v      = aa;
3201f268cba8SShri Abhyankar   vi     = aj;
3202f268cba8SShri Abhyankar   for (i=1; i<n; i++) {
3203f268cba8SShri Abhyankar     nz  = ai[i+1] - ai[i];
3204f268cba8SShri Abhyankar     sum = b[r[i]];
3205f268cba8SShri Abhyankar     PetscSparseDenseMinusDot(sum,tmp,v,vi,nz);
3206f268cba8SShri Abhyankar     tmp[i] = sum;
3207f268cba8SShri Abhyankar     v += nz; vi += nz;
3208f268cba8SShri Abhyankar   }
3209f268cba8SShri Abhyankar 
3210f268cba8SShri Abhyankar   /* backward solve the upper triangular */
3211f268cba8SShri Abhyankar   for (i=n-1; i>=0; i--){
32123c0119dfSShri Abhyankar     v   = aa + adiag[i+1]+1;
32133c0119dfSShri Abhyankar     vi  = aj + adiag[i+1]+1;
3214f268cba8SShri Abhyankar     nz  = adiag[i]-adiag[i+1]-1;
3215f268cba8SShri Abhyankar     sum = tmp[i];
3216f268cba8SShri Abhyankar     PetscSparseDenseMinusDot(sum,tmp,v,vi,nz);
3217f268cba8SShri Abhyankar     x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */
3218f268cba8SShri Abhyankar   }
3219f268cba8SShri Abhyankar 
3220f268cba8SShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
3221f268cba8SShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
32223649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
3223f268cba8SShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
3224f268cba8SShri Abhyankar   ierr = PetscLogFlops(2*a->nz - A->cmap->n);CHKERRQ(ierr);
3225f268cba8SShri Abhyankar   PetscFunctionReturn(0);
3226f268cba8SShri Abhyankar }
3227f268cba8SShri Abhyankar 
3228f268cba8SShri Abhyankar #undef __FUNCT__
32291d098868SHong Zhang #define __FUNCT__ "MatILUDTFactor_SeqAIJ"
3230fe97e370SBarry Smith /*
3231fe97e370SBarry 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
3232fe97e370SBarry Smith */
32331d098868SHong Zhang PetscErrorCode MatILUDTFactor_SeqAIJ(Mat A,IS isrow,IS iscol,const MatFactorInfo *info,Mat *fact)
32341d098868SHong Zhang {
32351d098868SHong Zhang   Mat                B = *fact;
3236599ef60dSHong Zhang   Mat_SeqAIJ         *a=(Mat_SeqAIJ*)A->data,*b;
3237599ef60dSHong Zhang   IS                 isicol;
3238599ef60dSHong Zhang   PetscErrorCode     ierr;
32391d098868SHong Zhang   const PetscInt     *r,*ic;
32401d098868SHong Zhang   PetscInt           i,n=A->rmap->n,*ai=a->i,*aj=a->j,*ajtmp,*adiag;
3241dc3a2fd3SHong Zhang   PetscInt           *bi,*bj,*bdiag,*bdiag_rev;
3242f61a948aSLisandro Dalcin   PetscInt           row,nzi,nzi_bl,nzi_bu,*im,nzi_al,nzi_au;
32431d098868SHong Zhang   PetscInt           nlnk,*lnk;
32441d098868SHong Zhang   PetscBT            lnkbt;
3245ace3abfcSBarry Smith   PetscBool          row_identity,icol_identity;
32468aee2decSHong Zhang   MatScalar          *aatmp,*pv,*batmp,*ba,*rtmp,*pc,multiplier,*vtmp,diag_tmp;
32471d098868SHong Zhang   const PetscInt     *ics;
32481d098868SHong Zhang   PetscInt           j,nz,*pj,*bjtmp,k,ncut,*jtmp;
3249182b8fbaSHong Zhang   PetscReal          dt=info->dt,dtcol=info->dtcol,shift=info->shiftamount;
3250f61a948aSLisandro Dalcin   PetscInt           dtcount=(PetscInt)info->dtcount,nnz_max;
3251ace3abfcSBarry Smith   PetscBool          missing;
3252599ef60dSHong Zhang 
3253599ef60dSHong Zhang   PetscFunctionBegin;
3254f61a948aSLisandro Dalcin 
3255f61a948aSLisandro Dalcin   if (dt      == PETSC_DEFAULT) dt      = 0.005;
3256f61a948aSLisandro Dalcin   if (dtcol   == PETSC_DEFAULT) dtcol   = 0.01; /* XXX unused! */
3257f61a948aSLisandro Dalcin   if (dtcount == PETSC_DEFAULT) dtcount = (PetscInt)(1.5*a->rmax);
3258f61a948aSLisandro Dalcin 
32591d098868SHong Zhang   /* ------- symbolic factorization, can be reused ---------*/
32601d098868SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr);
3261e32f2f54SBarry Smith   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i);
32621d098868SHong Zhang   adiag=a->diag;
3263599ef60dSHong Zhang 
3264599ef60dSHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
3265599ef60dSHong Zhang 
3266599ef60dSHong Zhang   /* bdiag is location of diagonal in factor */
32671bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);     /* becomes b->diag */
32681bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag_rev);CHKERRQ(ierr); /* temporary */
3269599ef60dSHong Zhang 
32701d098868SHong Zhang   /* allocate row pointers bi */
32718fc3a347SHong Zhang   ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
32721d098868SHong Zhang 
32731d098868SHong Zhang   /* allocate bj and ba; max num of nonzero entries is (ai[n]+2*n*dtcount+2) */
3274393d3a68SHong Zhang   if (dtcount > n-1) dtcount = n-1; /* diagonal is excluded */
32751d098868SHong Zhang   nnz_max  = ai[n]+2*n*dtcount+2;
32768fc3a347SHong Zhang 
32778fc3a347SHong Zhang   ierr = PetscMalloc((nnz_max+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
32788fc3a347SHong Zhang   ierr = PetscMalloc((nnz_max+1)*sizeof(MatScalar),&ba);CHKERRQ(ierr);
3279599ef60dSHong Zhang 
3280599ef60dSHong Zhang   /* put together the new matrix */
3281599ef60dSHong Zhang   ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
3282599ef60dSHong Zhang   ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr);
3283f284f12aSHong Zhang   b    = (Mat_SeqAIJ*)B->data;
3284599ef60dSHong Zhang   b->free_a       = PETSC_TRUE;
3285599ef60dSHong Zhang   b->free_ij      = PETSC_TRUE;
3286599ef60dSHong Zhang   b->singlemalloc = PETSC_FALSE;
3287599ef60dSHong Zhang   b->a          = ba;
3288599ef60dSHong Zhang   b->j          = bj;
3289599ef60dSHong Zhang   b->i          = bi;
3290599ef60dSHong Zhang   b->diag       = bdiag;
3291599ef60dSHong Zhang   b->ilen       = 0;
3292599ef60dSHong Zhang   b->imax       = 0;
3293599ef60dSHong Zhang   b->row        = isrow;
3294599ef60dSHong Zhang   b->col        = iscol;
3295599ef60dSHong Zhang   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
3296599ef60dSHong Zhang   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
3297599ef60dSHong Zhang   b->icol       = isicol;
3298599ef60dSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
3299599ef60dSHong Zhang 
33001d098868SHong Zhang   ierr = PetscLogObjectMemory(B,nnz_max*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
33011d098868SHong Zhang   b->maxnz = nnz_max;
3302599ef60dSHong Zhang 
3303d5f3da31SBarry Smith   B->factortype            = MAT_FACTOR_ILUDT;
3304f284f12aSHong Zhang   B->info.factor_mallocs   = 0;
3305f284f12aSHong Zhang   B->info.fill_ratio_given = ((PetscReal)nnz_max)/((PetscReal)ai[n]);
33061d098868SHong Zhang   CHKMEMQ;
33071d098868SHong Zhang   /* ------- end of symbolic factorization ---------*/
3308599ef60dSHong Zhang 
3309599ef60dSHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
3310599ef60dSHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
3311599ef60dSHong Zhang   ics  = ic;
3312599ef60dSHong Zhang 
3313599ef60dSHong Zhang   /* linked list for storing column indices of the active row */
3314599ef60dSHong Zhang   nlnk = n + 1;
3315599ef60dSHong Zhang   ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
33161d098868SHong Zhang 
33171d098868SHong Zhang   /* im: used by PetscLLAddSortedLU(); jtmp: working array for column indices of active row */
33180e83c824SBarry Smith   ierr = PetscMalloc2(n,PetscInt,&im,n,PetscInt,&jtmp);CHKERRQ(ierr);
33191d098868SHong Zhang   /* rtmp, vtmp: working arrays for sparse and contiguous row entries of active row */
33200e83c824SBarry Smith   ierr = PetscMalloc2(n,MatScalar,&rtmp,n,MatScalar,&vtmp);CHKERRQ(ierr);
33210e83c824SBarry Smith   ierr = PetscMemzero(rtmp,n*sizeof(MatScalar));CHKERRQ(ierr);
3322599ef60dSHong Zhang 
3323599ef60dSHong Zhang   bi[0]    = 0;
33248fc3a347SHong Zhang   bdiag[0] = nnz_max-1; /* location of diag[0] in factor B */
3325dc3a2fd3SHong Zhang   bdiag_rev[n] = bdiag[0];
33268fc3a347SHong Zhang   bi[2*n+1] = bdiag[0]+1; /* endof bj and ba array */
3327599ef60dSHong Zhang   for (i=0; i<n; i++) {
3328599ef60dSHong Zhang     /* copy initial fill into linked list */
3329599ef60dSHong Zhang     nzi = 0; /* nonzeros for active row i */
33308369b28dSHong Zhang     nzi = ai[r[i]+1] - ai[r[i]];
3331e32f2f54SBarry 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);
33328369b28dSHong Zhang     nzi_al = adiag[r[i]] - ai[r[i]];
33338369b28dSHong Zhang     nzi_au = ai[r[i]+1] - adiag[r[i]] -1;
3334599ef60dSHong Zhang     ajtmp = aj + ai[r[i]];
33358369b28dSHong Zhang     ierr = PetscLLAddPerm(nzi,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
3336599ef60dSHong Zhang 
3337599ef60dSHong Zhang     /* load in initial (unfactored row) */
33381d098868SHong Zhang     aatmp = a->a + ai[r[i]];
33398369b28dSHong Zhang     for (j=0; j<nzi; j++) {
33401d098868SHong Zhang       rtmp[ics[*ajtmp++]] = *aatmp++;
3341599ef60dSHong Zhang     }
3342599ef60dSHong Zhang 
3343599ef60dSHong Zhang     /* add pivot rows into linked list */
3344599ef60dSHong Zhang     row = lnk[n];
3345599ef60dSHong Zhang     while (row < i ) {
33461d098868SHong Zhang       nzi_bl = bi[row+1] - bi[row] + 1;
33471d098868SHong Zhang       bjtmp = bj + bdiag[row+1]+1; /* points to 1st column next to the diagonal in U */
33481d098868SHong Zhang       ierr  = PetscLLAddSortedLU(bjtmp,row,nlnk,lnk,lnkbt,i,nzi_bl,im);CHKERRQ(ierr);
3349599ef60dSHong Zhang       nzi  += nlnk;
3350599ef60dSHong Zhang       row   = lnk[row];
3351599ef60dSHong Zhang     }
3352599ef60dSHong Zhang 
33538369b28dSHong Zhang     /* copy data from lnk into jtmp, then initialize lnk */
33548369b28dSHong Zhang     ierr = PetscLLClean(n,n,nzi,lnk,jtmp,lnkbt);CHKERRQ(ierr);
3355599ef60dSHong Zhang 
3356599ef60dSHong Zhang     /* numerical factorization */
33578369b28dSHong Zhang     bjtmp = jtmp;
3358599ef60dSHong Zhang     row   = *bjtmp++; /* 1st pivot row */
3359599ef60dSHong Zhang     while  ( row < i ) {
3360599ef60dSHong Zhang       pc         = rtmp + row;
33613c2a7987SHong Zhang       pv         = ba + bdiag[row]; /* 1./(diag of the pivot row) */
33623c2a7987SHong Zhang       multiplier = (*pc) * (*pv);
3363599ef60dSHong Zhang       *pc        = multiplier;
33643c2a7987SHong Zhang       if (PetscAbsScalar(*pc) > dt){ /* apply tolerance dropping rule */
33651d098868SHong Zhang         pj         = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */
33661d098868SHong Zhang         pv         = ba + bdiag[row+1] + 1;
33671d098868SHong Zhang         /* if (multiplier < -1.0 or multiplier >1.0) printf("row/prow %d, %d, multiplier %g\n",i,row,multiplier); */
33681d098868SHong Zhang         nz         = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */
33691d098868SHong Zhang         for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++);
3370dc0b31edSSatish Balay         ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr);
3371599ef60dSHong Zhang       }
3372599ef60dSHong Zhang       row = *bjtmp++;
3373599ef60dSHong Zhang     }
3374599ef60dSHong Zhang 
33758369b28dSHong Zhang     /* copy sparse rtmp into contiguous vtmp; separate L and U part */
33768aee2decSHong Zhang     diag_tmp = rtmp[i];  /* save diagonal value - may not needed?? */
33778369b28dSHong Zhang     nzi_bl = 0; j = 0;
33788369b28dSHong Zhang     while (jtmp[j] < i){ /* Note: jtmp is sorted */
33798aee2decSHong Zhang       vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0;
33808369b28dSHong Zhang       nzi_bl++; j++;
33818369b28dSHong Zhang     }
33828369b28dSHong Zhang     nzi_bu = nzi - nzi_bl -1;
33838369b28dSHong Zhang     while (j < nzi){
33848aee2decSHong Zhang       vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0;
33858369b28dSHong Zhang       j++;
33868369b28dSHong Zhang     }
33878369b28dSHong Zhang 
3388599ef60dSHong Zhang     bjtmp = bj + bi[i];
3389599ef60dSHong Zhang     batmp = ba + bi[i];
33908369b28dSHong Zhang     /* apply level dropping rule to L part */
33918369b28dSHong Zhang     ncut = nzi_al + dtcount;
33928369b28dSHong Zhang     if (ncut < nzi_bl){
33938369b28dSHong Zhang       ierr = PetscSortSplit(ncut,nzi_bl,vtmp,jtmp);CHKERRQ(ierr);
33948369b28dSHong Zhang       ierr = PetscSortIntWithScalarArray(ncut,jtmp,vtmp);CHKERRQ(ierr);
3395599ef60dSHong Zhang     } else {
33968369b28dSHong Zhang       ncut = nzi_bl;
33978369b28dSHong Zhang     }
33988369b28dSHong Zhang     for (j=0; j<ncut; j++){
33998369b28dSHong Zhang       bjtmp[j] = jtmp[j];
34008369b28dSHong Zhang       batmp[j] = vtmp[j];
34016da515a1SHong Zhang       /* printf(" (%d,%g),",bjtmp[j],batmp[j]); */
34028369b28dSHong Zhang     }
34031d098868SHong Zhang     bi[i+1] = bi[i] + ncut;
34048369b28dSHong Zhang     nzi = ncut + 1;
34058369b28dSHong Zhang 
34068369b28dSHong Zhang     /* apply level dropping rule to U part */
34078369b28dSHong Zhang     ncut = nzi_au + dtcount;
34088369b28dSHong Zhang     if (ncut < nzi_bu){
34098369b28dSHong Zhang       ierr = PetscSortSplit(ncut,nzi_bu,vtmp+nzi_bl+1,jtmp+nzi_bl+1);CHKERRQ(ierr);
34108369b28dSHong Zhang       ierr = PetscSortIntWithScalarArray(ncut,jtmp+nzi_bl+1,vtmp+nzi_bl+1);CHKERRQ(ierr);
34118369b28dSHong Zhang     } else {
34128369b28dSHong Zhang       ncut = nzi_bu;
34138369b28dSHong Zhang     }
34148369b28dSHong Zhang     nzi += ncut;
34151d098868SHong Zhang 
34161d098868SHong Zhang     /* mark bdiagonal */
34171d098868SHong Zhang     bdiag[i+1]       = bdiag[i] - (ncut + 1);
3418dc3a2fd3SHong Zhang     bdiag_rev[n-i-1] = bdiag[i+1];
34198fc3a347SHong Zhang     bi[2*n - i]      = bi[2*n - i +1] - (ncut + 1);
34201d098868SHong Zhang     bjtmp = bj + bdiag[i];
34211d098868SHong Zhang     batmp = ba + bdiag[i];
34221d098868SHong Zhang     *bjtmp = i;
34238aee2decSHong Zhang     *batmp = diag_tmp; /* rtmp[i]; */
3424c9c72f8fSHong Zhang     if (*batmp == 0.0) {
3425c9c72f8fSHong Zhang       *batmp = dt+shift;
3426bd1bc851SBarry Smith       /* printf(" row %d add shift %g\n",i,shift); */
3427c9c72f8fSHong Zhang     }
3428b78a477dSHong Zhang     *batmp = 1.0/(*batmp); /* invert diagonal entries for simplier triangular solves */
34296da515a1SHong Zhang     /* printf(" (%d,%g),",*bjtmp,*batmp); */
34301d098868SHong Zhang 
34311d098868SHong Zhang     bjtmp = bj + bdiag[i+1]+1;
34321d098868SHong Zhang     batmp = ba + bdiag[i+1]+1;
34338369b28dSHong Zhang     for (k=0; k<ncut; k++){
34341d098868SHong Zhang       bjtmp[k] = jtmp[nzi_bl+1+k];
34351d098868SHong Zhang       batmp[k] = vtmp[nzi_bl+1+k];
34366da515a1SHong Zhang       /* printf(" (%d,%g),",bjtmp[k],batmp[k]); */
3437599ef60dSHong Zhang     }
34386da515a1SHong Zhang     /* printf("\n"); */
3439599ef60dSHong Zhang 
3440599ef60dSHong Zhang     im[i]   = nzi; /* used by PetscLLAddSortedLU() */
3441599ef60dSHong Zhang     /*
34421d098868SHong Zhang     printf("row %d: bi %d, bdiag %d\n",i,bi[i],bdiag[i]);
34438369b28dSHong Zhang     printf(" ----------------------------\n");
3444599ef60dSHong Zhang     */
34458369b28dSHong Zhang   } /* for (i=0; i<n; i++) */
34461d098868SHong Zhang   /* printf("end of L %d, beginning of U %d\n",bi[n],bdiag[n]); */
3447e32f2f54SBarry 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]);
3448599ef60dSHong Zhang 
3449599ef60dSHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
3450599ef60dSHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
3451599ef60dSHong Zhang 
3452599ef60dSHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
34530e83c824SBarry Smith   ierr = PetscFree2(im,jtmp);CHKERRQ(ierr);
34540e83c824SBarry Smith   ierr = PetscFree2(rtmp,vtmp);CHKERRQ(ierr);
34551bfa06eaSJed Brown   ierr = PetscFree(bdiag_rev);CHKERRQ(ierr);
3456599ef60dSHong Zhang 
3457599ef60dSHong Zhang   ierr = PetscLogFlops(B->cmap->n);CHKERRQ(ierr);
34581d098868SHong Zhang   b->maxnz = b->nz = bi[n] + bdiag[0] - bdiag[n];
3459599ef60dSHong Zhang 
3460599ef60dSHong Zhang   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
3461599ef60dSHong Zhang   ierr = ISIdentity(isicol,&icol_identity);CHKERRQ(ierr);
3462599ef60dSHong Zhang   if (row_identity && icol_identity) {
346335233d99SShri Abhyankar     B->ops->solve = MatSolve_SeqAIJ_NaturalOrdering;
3464599ef60dSHong Zhang   } else {
346535233d99SShri Abhyankar     B->ops->solve = MatSolve_SeqAIJ;
3466599ef60dSHong Zhang   }
3467599ef60dSHong Zhang 
34681d098868SHong Zhang   B->ops->solveadd          = 0;
34691d098868SHong Zhang   B->ops->solvetranspose    = 0;
34701d098868SHong Zhang   B->ops->solvetransposeadd = 0;
34711d098868SHong Zhang   B->ops->matsolve          = 0;
3472599ef60dSHong Zhang   B->assembled              = PETSC_TRUE;
3473599ef60dSHong Zhang   B->preallocated           = PETSC_TRUE;
3474599ef60dSHong Zhang   PetscFunctionReturn(0);
3475599ef60dSHong Zhang }
3476599ef60dSHong Zhang 
34773c2a7987SHong Zhang /* a wraper of MatILUDTFactor_SeqAIJ() */
34783c2a7987SHong Zhang #undef __FUNCT__
34793c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorSymbolic_SeqAIJ"
3480fe97e370SBarry Smith /*
3481fe97e370SBarry 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
3482fe97e370SBarry Smith */
3483fe97e370SBarry Smith 
34847087cfbeSBarry Smith PetscErrorCode  MatILUDTFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS row,IS col,const MatFactorInfo *info)
34853c2a7987SHong Zhang {
34863c2a7987SHong Zhang   PetscErrorCode     ierr;
34873c2a7987SHong Zhang 
34883c2a7987SHong Zhang   PetscFunctionBegin;
34893c2a7987SHong Zhang   ierr = MatILUDTFactor_SeqAIJ(A,row,col,info,&fact);CHKERRQ(ierr);
34903c2a7987SHong Zhang   PetscFunctionReturn(0);
34913c2a7987SHong Zhang }
34923c2a7987SHong Zhang 
34933c2a7987SHong Zhang /*
34943c2a7987SHong Zhang    same as MatLUFactorNumeric_SeqAIJ(), except using contiguous array matrix factors
34953c2a7987SHong Zhang    - intend to replace existing MatLUFactorNumeric_SeqAIJ()
34963c2a7987SHong Zhang */
34973c2a7987SHong Zhang #undef __FUNCT__
34983c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorNumeric_SeqAIJ"
3499fe97e370SBarry Smith /*
3500fe97e370SBarry 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
3501fe97e370SBarry Smith */
3502fe97e370SBarry Smith 
35037087cfbeSBarry Smith PetscErrorCode  MatILUDTFactorNumeric_SeqAIJ(Mat fact,Mat A,const MatFactorInfo *info)
35043c2a7987SHong Zhang {
35053c2a7987SHong Zhang   Mat            C=fact;
35063c2a7987SHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data;
35073c2a7987SHong Zhang   IS             isrow = b->row,isicol = b->icol;
35083c2a7987SHong Zhang   PetscErrorCode ierr;
35093c2a7987SHong Zhang   const PetscInt *r,*ic,*ics;
3510b78a477dSHong Zhang   PetscInt       i,j,k,n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j;
3511b78a477dSHong Zhang   PetscInt       *ajtmp,*bjtmp,nz,nzl,nzu,row,*bdiag = b->diag,*pj;
35123c2a7987SHong Zhang   MatScalar      *rtmp,*pc,multiplier,*v,*pv,*aa=a->a;
3513182b8fbaSHong Zhang   PetscReal      dt=info->dt,shift=info->shiftamount;
3514ace3abfcSBarry Smith   PetscBool      row_identity, col_identity;
35153c2a7987SHong Zhang 
35163c2a7987SHong Zhang   PetscFunctionBegin;
35173c2a7987SHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
35183c2a7987SHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
3519b78a477dSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr);
35203c2a7987SHong Zhang   ics  = ic;
35213c2a7987SHong Zhang 
35223c2a7987SHong Zhang   for (i=0; i<n; i++){
3523b78a477dSHong Zhang     /* initialize rtmp array */
3524b78a477dSHong Zhang     nzl   = bi[i+1] - bi[i];       /* num of nozeros in L(i,:) */
35253c2a7987SHong Zhang     bjtmp = bj + bi[i];
3526b78a477dSHong Zhang     for  (j=0; j<nzl; j++) rtmp[*bjtmp++] = 0.0;
3527b78a477dSHong Zhang     rtmp[i] = 0.0;
3528b78a477dSHong Zhang     nzu   = bdiag[i] - bdiag[i+1]; /* num of nozeros in U(i,:) */
3529b78a477dSHong Zhang     bjtmp = bj + bdiag[i+1] + 1;
3530b78a477dSHong Zhang     for  (j=0; j<nzu; j++) rtmp[*bjtmp++] = 0.0;
35313c2a7987SHong Zhang 
3532b78a477dSHong Zhang     /* load in initial unfactored row of A */
35336da515a1SHong Zhang     /* printf("row %d\n",i); */
35343c2a7987SHong Zhang     nz    = ai[r[i]+1] - ai[r[i]];
35353c2a7987SHong Zhang     ajtmp = aj + ai[r[i]];
35363c2a7987SHong Zhang     v     = aa + ai[r[i]];
35373c2a7987SHong Zhang     for (j=0; j<nz; j++) {
3538b78a477dSHong Zhang       rtmp[ics[*ajtmp++]] = v[j];
35396da515a1SHong Zhang       /* printf(" (%d,%g),",ics[ajtmp[j]],rtmp[ics[ajtmp[j]]]); */
35403c2a7987SHong Zhang     }
35416da515a1SHong Zhang     /* printf("\n"); */
35423c2a7987SHong Zhang 
3543b78a477dSHong Zhang     /* numerical factorization */
3544b78a477dSHong Zhang     bjtmp = bj + bi[i]; /* point to 1st entry of L(i,:) */
3545b78a477dSHong Zhang     nzl   = bi[i+1] - bi[i]; /* num of entries in L(i,:) */
3546b78a477dSHong Zhang     k = 0;
3547b78a477dSHong Zhang     while (k < nzl){
35483c2a7987SHong Zhang       row   = *bjtmp++;
35496da515a1SHong Zhang       /* printf("  prow %d\n",row); */
35503c2a7987SHong Zhang       pc         = rtmp + row;
3551b78a477dSHong Zhang       pv         = b->a + bdiag[row]; /* 1./(diag of the pivot row) */
3552b78a477dSHong Zhang       multiplier = (*pc) * (*pv);
35533c2a7987SHong Zhang       *pc        = multiplier;
3554b78a477dSHong Zhang       if (PetscAbsScalar(multiplier) > dt){
3555b78a477dSHong Zhang         pj         = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */
3556b78a477dSHong Zhang         pv         = b->a + bdiag[row+1] + 1;
3557b78a477dSHong Zhang         nz         = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */
3558b78a477dSHong Zhang         for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++);
35596da515a1SHong Zhang         /* ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); */
35603c2a7987SHong Zhang       }
3561b78a477dSHong Zhang       k++;
35623c2a7987SHong Zhang     }
35633c2a7987SHong Zhang 
3564b78a477dSHong Zhang     /* finished row so stick it into b->a */
3565b78a477dSHong Zhang     /* L-part */
3566b78a477dSHong Zhang     pv = b->a + bi[i] ;
3567b78a477dSHong Zhang     pj = bj + bi[i] ;
3568b78a477dSHong Zhang     nzl = bi[i+1] - bi[i];
3569b78a477dSHong Zhang     for (j=0; j<nzl; j++) {
3570b78a477dSHong Zhang       pv[j] = rtmp[pj[j]];
35716da515a1SHong Zhang       /* printf(" (%d,%g),",pj[j],pv[j]); */
35723c2a7987SHong Zhang     }
3573b78a477dSHong Zhang 
3574b78a477dSHong Zhang     /* diagonal: invert diagonal entries for simplier triangular solves */
3575b78a477dSHong Zhang     if (rtmp[i] == 0.0) rtmp[i] = dt+shift;
3576b78a477dSHong Zhang     b->a[bdiag[i]] = 1.0/rtmp[i];
35776da515a1SHong Zhang     /* printf(" (%d,%g),",i,b->a[bdiag[i]]); */
3578b78a477dSHong Zhang 
3579b78a477dSHong Zhang     /* U-part */
3580b78a477dSHong Zhang     pv = b->a + bdiag[i+1] + 1;
3581b78a477dSHong Zhang     pj = bj + bdiag[i+1] + 1;
3582b78a477dSHong Zhang     nzu = bdiag[i] - bdiag[i+1] - 1;
3583b78a477dSHong Zhang     for (j=0; j<nzu; j++) {
3584b78a477dSHong Zhang       pv[j] = rtmp[pj[j]];
35856da515a1SHong Zhang       /* printf(" (%d,%g),",pj[j],pv[j]); */
3586b78a477dSHong Zhang     }
35876da515a1SHong Zhang     /* printf("\n"); */
3588b78a477dSHong Zhang   }
3589b78a477dSHong Zhang 
35903c2a7987SHong Zhang   ierr = PetscFree(rtmp);CHKERRQ(ierr);
35913c2a7987SHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
35923c2a7987SHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
3593b78a477dSHong Zhang 
35943c2a7987SHong Zhang   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
35953c2a7987SHong Zhang   ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr);
35963c2a7987SHong Zhang   if (row_identity && col_identity) {
359735233d99SShri Abhyankar     C->ops->solve   = MatSolve_SeqAIJ_NaturalOrdering;
35983c2a7987SHong Zhang   } else {
359935233d99SShri Abhyankar     C->ops->solve   = MatSolve_SeqAIJ;
36003c2a7987SHong Zhang   }
3601b78a477dSHong Zhang   C->ops->solveadd           = 0;
3602b78a477dSHong Zhang   C->ops->solvetranspose     = 0;
3603b78a477dSHong Zhang   C->ops->solvetransposeadd  = 0;
3604b78a477dSHong Zhang   C->ops->matsolve           = 0;
36053c2a7987SHong Zhang   C->assembled    = PETSC_TRUE;
36063c2a7987SHong Zhang   C->preallocated = PETSC_TRUE;
36073c2a7987SHong Zhang   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
36083c2a7987SHong Zhang   PetscFunctionReturn(0);
36093c2a7987SHong Zhang }
3610