xref: /petsc/src/mat/impls/aij/seq/aijfact.c (revision 891bceaeceeef9e699f16c617e24c2bf64eca47e)
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;
115*891bceaeSHong Zhang #if defined(PETSC_USE_COMPLEX)
116*891bceaeSHong Zhang   if (A->hermitian && (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC))SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Hermitian Factor is not supported");
117*891bceaeSHong Zhang #endif
118b24902e0SBarry Smith   ierr = MatCreate(((PetscObject)A)->comm,B);CHKERRQ(ierr);
119b24902e0SBarry Smith   ierr = MatSetSizes(*B,n,n,n,n);CHKERRQ(ierr);
120599ef60dSHong Zhang   if (ftype == MAT_FACTOR_LU || ftype == MAT_FACTOR_ILU || ftype == MAT_FACTOR_ILUDT){
121b24902e0SBarry Smith     ierr = MatSetType(*B,MATSEQAIJ);CHKERRQ(ierr);
12235233d99SShri Abhyankar     (*B)->ops->ilufactorsymbolic = MatILUFactorSymbolic_SeqAIJ;
12335233d99SShri Abhyankar     (*B)->ops->lufactorsymbolic  = MatLUFactorSymbolic_SeqAIJ;
124e0527adcSJed Brown     ierr = MatSetBlockSizes(*B,A->rmap->bs,A->cmap->bs);CHKERRQ(ierr);
125b24902e0SBarry Smith   } else if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) {
126b24902e0SBarry Smith     ierr = MatSetType(*B,MATSEQSBAIJ);CHKERRQ(ierr);
127b24902e0SBarry Smith     ierr = MatSeqSBAIJSetPreallocation(*B,1,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
12835233d99SShri Abhyankar     (*B)->ops->iccfactorsymbolic      = MatICCFactorSymbolic_SeqAIJ;
12935233d99SShri Abhyankar     (*B)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqAIJ;
130e32f2f54SBarry Smith   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Factor type not supported");
131d5f3da31SBarry Smith   (*B)->factortype = ftype;
132b24902e0SBarry Smith   PetscFunctionReturn(0);
133b24902e0SBarry Smith }
134e631078cSBarry Smith EXTERN_C_END
135b24902e0SBarry Smith 
136b24902e0SBarry Smith #undef __FUNCT__
137ad04f41aSHong Zhang #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ_inplace"
138ad04f41aSHong Zhang PetscErrorCode MatLUFactorSymbolic_SeqAIJ_inplace(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
139289bc588SBarry Smith {
140416022c9SBarry Smith   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
141289bc588SBarry Smith   IS                 isicol;
1426849ba73SBarry Smith   PetscErrorCode     ierr;
1435d0c19d7SBarry Smith   const PetscInt     *r,*ic;
1445d0c19d7SBarry Smith   PetscInt           i,n=A->rmap->n,*ai=a->i,*aj=a->j;
1451393bc97SHong Zhang   PetscInt           *bi,*bj,*ajtmp;
1461393bc97SHong Zhang   PetscInt           *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im;
1479e046878SBarry Smith   PetscReal          f;
1484875ba38SHong Zhang   PetscInt           nlnk,*lnk,k,**bi_ptr;
149a1a86e44SBarry Smith   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
1501393bc97SHong Zhang   PetscBT            lnkbt;
151289bc588SBarry Smith 
1523a40ed3dSBarry Smith   PetscFunctionBegin;
153e32f2f54SBarry Smith   if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"matrix must be square");
1544c49b128SBarry Smith   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
155ac121b3dSBarry Smith   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
156ac121b3dSBarry Smith   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
157289bc588SBarry Smith 
158289bc588SBarry Smith   /* get new row pointers */
1591393bc97SHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
1601393bc97SHong Zhang   bi[0] = 0;
1611393bc97SHong Zhang 
1621393bc97SHong Zhang   /* bdiag is location of diagonal in factor */
1631393bc97SHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
1641393bc97SHong Zhang   bdiag[0] = 0;
1651393bc97SHong Zhang 
1661393bc97SHong Zhang   /* linked list for storing column indices of the active row */
1671393bc97SHong Zhang   nlnk = n + 1;
1681393bc97SHong Zhang   ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
1691393bc97SHong Zhang 
17035baf27eSSatish Balay   ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr);
1711393bc97SHong Zhang 
1721393bc97SHong Zhang   /* initial FreeSpace size is f*(ai[n]+1) */
173d3d32019SBarry Smith   f = info->fill;
174a1a86e44SBarry Smith   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr);
1751393bc97SHong Zhang   current_space = free_space;
176289bc588SBarry Smith 
177289bc588SBarry Smith   for (i=0; i<n; i++) {
1781393bc97SHong Zhang     /* copy previous fill into linked list */
179289bc588SBarry Smith     nzi = 0;
1801393bc97SHong Zhang     nnz = ai[r[i]+1] - ai[r[i]];
181e32f2f54SBarry 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);
1821393bc97SHong Zhang     ajtmp = aj + ai[r[i]];
183afcc9904SHong Zhang     ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
1841393bc97SHong Zhang     nzi += nlnk;
1851393bc97SHong Zhang 
1861393bc97SHong Zhang     /* add pivot rows into linked list */
1871393bc97SHong Zhang     row = lnk[n];
1881393bc97SHong Zhang     while (row < i) {
189d1170560SHong Zhang       nzbd    = bdiag[row] - bi[row] + 1; /* num of entries in the row with column index <= row */
190d1170560SHong Zhang       ajtmp   = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */
191d1170560SHong Zhang       ierr = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr);
1921393bc97SHong Zhang       nzi += nlnk;
1931393bc97SHong Zhang       row  = lnk[row];
194289bc588SBarry Smith     }
1951393bc97SHong Zhang     bi[i+1] = bi[i] + nzi;
1961393bc97SHong Zhang     im[i]   = nzi;
1971393bc97SHong Zhang 
1981393bc97SHong Zhang     /* mark bdiag */
1991393bc97SHong Zhang     nzbd = 0;
2001393bc97SHong Zhang     nnz  = nzi;
2011393bc97SHong Zhang     k    = lnk[n];
2021393bc97SHong Zhang     while (nnz-- && k < i){
2031393bc97SHong Zhang       nzbd++;
2041393bc97SHong Zhang       k = lnk[k];
2051393bc97SHong Zhang     }
2061393bc97SHong Zhang     bdiag[i] = bi[i] + nzbd;
2071393bc97SHong Zhang 
2081393bc97SHong Zhang     /* if free space is not available, make more free space */
2091393bc97SHong Zhang     if (current_space->local_remaining<nzi) {
2101393bc97SHong Zhang       nnz = (n - i)*nzi; /* estimated and max additional space needed */
211a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
2121393bc97SHong Zhang       reallocs++;
2131393bc97SHong Zhang     }
2141393bc97SHong Zhang 
2151393bc97SHong Zhang     /* copy data into free space, then initialize lnk */
2161393bc97SHong Zhang     ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
2171393bc97SHong Zhang     bi_ptr[i] = current_space->array;
2181393bc97SHong Zhang     current_space->array           += nzi;
2191393bc97SHong Zhang     current_space->local_used      += nzi;
2201393bc97SHong Zhang     current_space->local_remaining -= nzi;
221289bc588SBarry Smith   }
2226cf91177SBarry Smith #if defined(PETSC_USE_INFO)
223464e8d44SSatish Balay   if (ai[n] != 0) {
2241393bc97SHong Zhang     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
225ae15b995SBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr);
226ae15b995SBarry Smith     ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
227ae15b995SBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr);
228ae15b995SBarry Smith     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
22905bf559cSSatish Balay   } else {
230ae15b995SBarry Smith     ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr);
23105bf559cSSatish Balay   }
23263ba0a88SBarry Smith #endif
2332fb3ed76SBarry Smith 
234898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
235898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
2361987afe7SBarry Smith 
2371393bc97SHong Zhang   /* destroy list of free space and other temporary array(s) */
2381393bc97SHong Zhang   ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
239a1a86e44SBarry Smith   ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr);
2401393bc97SHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
24135baf27eSSatish Balay   ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr);
242289bc588SBarry Smith 
243289bc588SBarry Smith   /* put together the new matrix */
244719d5645SBarry Smith   ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
245719d5645SBarry Smith   ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr);
246719d5645SBarry Smith   b    = (Mat_SeqAIJ*)(B)->data;
247e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
248e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
2497c922b88SBarry Smith   b->singlemalloc = PETSC_FALSE;
2501393bc97SHong Zhang   ierr          = PetscMalloc((bi[n]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);
2511393bc97SHong Zhang   b->j          = bj;
2521393bc97SHong Zhang   b->i          = bi;
2531393bc97SHong Zhang   b->diag       = bdiag;
254416022c9SBarry Smith   b->ilen       = 0;
255416022c9SBarry Smith   b->imax       = 0;
256416022c9SBarry Smith   b->row        = isrow;
257416022c9SBarry Smith   b->col        = iscol;
258c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
259c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
26082bf6240SBarry Smith   b->icol       = isicol;
26187828ca2SBarry Smith   ierr          = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
2621393bc97SHong Zhang 
2631393bc97SHong Zhang   /* In b structure:  Free imax, ilen, old a, old j.  Allocate solve_work, new a, new j */
264719d5645SBarry Smith   ierr = PetscLogObjectMemory(B,(bi[n]-n)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
2651393bc97SHong Zhang   b->maxnz = b->nz = bi[n] ;
2667fd98bd6SLois Curfman McInnes 
267d5f3da31SBarry Smith   (B)->factortype            = MAT_FACTOR_LU;
268719d5645SBarry Smith   (B)->info.factor_mallocs   = reallocs;
269719d5645SBarry Smith   (B)->info.fill_ratio_given = f;
270703deb49SSatish Balay 
2719f7d0b68SHong Zhang   if (ai[n]) {
272719d5645SBarry Smith     (B)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]);
273eea2913aSSatish Balay   } else {
274719d5645SBarry Smith     (B)->info.fill_ratio_needed = 0.0;
275eea2913aSSatish Balay   }
276ad04f41aSHong Zhang   (B)->ops->lufactornumeric  = MatLUFactorNumeric_SeqAIJ_inplace;
277f2344125SBarry Smith   if (a->inode.size) {
278f2344125SBarry Smith     (B)->ops->lufactornumeric  = MatLUFactorNumeric_SeqAIJ_Inode_inplace;
279f2344125SBarry Smith   }
2803a40ed3dSBarry Smith   PetscFunctionReturn(0);
281289bc588SBarry Smith }
2821393bc97SHong Zhang 
283ce3d78c0SShri Abhyankar #undef __FUNCT__
28435233d99SShri Abhyankar #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ"
28535233d99SShri Abhyankar PetscErrorCode MatLUFactorSymbolic_SeqAIJ(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
286ce3d78c0SShri Abhyankar {
287ce3d78c0SShri Abhyankar   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
288ce3d78c0SShri Abhyankar   IS                 isicol;
289ce3d78c0SShri Abhyankar   PetscErrorCode     ierr;
290ce3d78c0SShri Abhyankar   const PetscInt     *r,*ic;
291ce3d78c0SShri Abhyankar   PetscInt           i,n=A->rmap->n,*ai=a->i,*aj=a->j;
292ce3d78c0SShri Abhyankar   PetscInt           *bi,*bj,*ajtmp;
293ce3d78c0SShri Abhyankar   PetscInt           *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im;
294ce3d78c0SShri Abhyankar   PetscReal          f;
295ce3d78c0SShri Abhyankar   PetscInt           nlnk,*lnk,k,**bi_ptr;
296ce3d78c0SShri Abhyankar   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
297ce3d78c0SShri Abhyankar   PetscBT            lnkbt;
298ce3d78c0SShri Abhyankar 
299ce3d78c0SShri Abhyankar   PetscFunctionBegin;
3003898ab1dSSatish Balay   /* Uncomment the oldatastruct part only while testing new data structure for MatSolve() */
301a4be958cSHong Zhang   /*
3021a26ff17SHong Zhang   PetscBool          olddatastruct=PETSC_FALSE;
303acfcf0e5SJed Brown   ierr = PetscOptionsGetBool(PETSC_NULL,"-lu_old",&olddatastruct,PETSC_NULL);CHKERRQ(ierr);
304628f99d7SShri Abhyankar   if(olddatastruct){
305628f99d7SShri Abhyankar     ierr = MatLUFactorSymbolic_SeqAIJ_inplace(B,A,isrow,iscol,info);CHKERRQ(ierr);
306628f99d7SShri Abhyankar     PetscFunctionReturn(0);
307628f99d7SShri Abhyankar   }
308a4be958cSHong Zhang   */
309e32f2f54SBarry Smith   if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"matrix must be square");
310ce3d78c0SShri Abhyankar   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
311ce3d78c0SShri Abhyankar   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
312ce3d78c0SShri Abhyankar   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
313ce3d78c0SShri Abhyankar 
3141bfa06eaSJed Brown   /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */
3151bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
3161bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
317a6df321fSShri Abhyankar   bi[0] = bdiag[0] = 0;
3189b48462bSShri Abhyankar 
3199b48462bSShri Abhyankar   /* linked list for storing column indices of the active row */
3209b48462bSShri Abhyankar   nlnk = n + 1;
3219b48462bSShri Abhyankar   ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
3229b48462bSShri Abhyankar 
3239b48462bSShri Abhyankar   ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr);
3249b48462bSShri Abhyankar 
3259b48462bSShri Abhyankar   /* initial FreeSpace size is f*(ai[n]+1) */
3269b48462bSShri Abhyankar   f = info->fill;
3279b48462bSShri Abhyankar   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr);
3289b48462bSShri Abhyankar   current_space = free_space;
3299b48462bSShri Abhyankar 
3309b48462bSShri Abhyankar   for (i=0; i<n; i++) {
3319b48462bSShri Abhyankar     /* copy previous fill into linked list */
3329b48462bSShri Abhyankar     nzi = 0;
3339b48462bSShri Abhyankar     nnz = ai[r[i]+1] - ai[r[i]];
334e32f2f54SBarry 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);
3359b48462bSShri Abhyankar     ajtmp = aj + ai[r[i]];
3369b48462bSShri Abhyankar     ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
3379b48462bSShri Abhyankar     nzi += nlnk;
3389b48462bSShri Abhyankar 
3399b48462bSShri Abhyankar     /* add pivot rows into linked list */
3409b48462bSShri Abhyankar     row = lnk[n];
3419b48462bSShri Abhyankar     while (row < i){
3429b48462bSShri Abhyankar       nzbd  = bdiag[row] + 1; /* num of entries in the row with column index <= row */
3439b48462bSShri Abhyankar       ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */
3449b48462bSShri Abhyankar       ierr  = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr);
3459b48462bSShri Abhyankar       nzi  += nlnk;
3469b48462bSShri Abhyankar       row   = lnk[row];
3479b48462bSShri Abhyankar     }
3489b48462bSShri Abhyankar     bi[i+1] = bi[i] + nzi;
3499b48462bSShri Abhyankar     im[i]   = nzi;
3509b48462bSShri Abhyankar 
3519b48462bSShri Abhyankar     /* mark bdiag */
3529b48462bSShri Abhyankar     nzbd = 0;
3539b48462bSShri Abhyankar     nnz  = nzi;
3549b48462bSShri Abhyankar     k    = lnk[n];
3559b48462bSShri Abhyankar     while (nnz-- && k < i){
3569b48462bSShri Abhyankar       nzbd++;
3579b48462bSShri Abhyankar       k = lnk[k];
3589b48462bSShri Abhyankar     }
3599b48462bSShri Abhyankar     bdiag[i] = nzbd; /* note: bdiag[i] = nnzL as input for PetscFreeSpaceContiguous_LU() */
3609b48462bSShri Abhyankar 
3619b48462bSShri Abhyankar     /* if free space is not available, make more free space */
3629b48462bSShri Abhyankar     if (current_space->local_remaining<nzi) {
3639b48462bSShri Abhyankar       nnz = 2*(n - i)*nzi; /* estimated and max additional space needed */
3649b48462bSShri Abhyankar       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
3659b48462bSShri Abhyankar       reallocs++;
3669b48462bSShri Abhyankar     }
3679b48462bSShri Abhyankar 
3689b48462bSShri Abhyankar     /* copy data into free space, then initialize lnk */
3699b48462bSShri Abhyankar     ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
3709b48462bSShri Abhyankar     bi_ptr[i] = current_space->array;
3719b48462bSShri Abhyankar     current_space->array           += nzi;
3729b48462bSShri Abhyankar     current_space->local_used      += nzi;
3739b48462bSShri Abhyankar     current_space->local_remaining -= nzi;
3749b48462bSShri Abhyankar   }
3759b48462bSShri Abhyankar 
3769b48462bSShri Abhyankar   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
3779b48462bSShri Abhyankar   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
3789b48462bSShri Abhyankar 
3799263d837SHong Zhang   /*   copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */
3809b48462bSShri Abhyankar   ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
3812ce24eb6SHong Zhang   ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr);
3829b48462bSShri Abhyankar   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
3839b48462bSShri Abhyankar   ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr);
3849b48462bSShri Abhyankar 
3859b48462bSShri Abhyankar   /* put together the new matrix */
3869b48462bSShri Abhyankar   ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
3879b48462bSShri Abhyankar   ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr);
3889b48462bSShri Abhyankar   b    = (Mat_SeqAIJ*)(B)->data;
3899b48462bSShri Abhyankar   b->free_a       = PETSC_TRUE;
3909b48462bSShri Abhyankar   b->free_ij      = PETSC_TRUE;
3919b48462bSShri Abhyankar   b->singlemalloc = PETSC_FALSE;
3929b48462bSShri Abhyankar   ierr = PetscMalloc((bdiag[0]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);
3939b48462bSShri Abhyankar   b->j          = bj;
3949b48462bSShri Abhyankar   b->i          = bi;
3959b48462bSShri Abhyankar   b->diag       = bdiag;
3969b48462bSShri Abhyankar   b->ilen       = 0;
3979b48462bSShri Abhyankar   b->imax       = 0;
3989b48462bSShri Abhyankar   b->row        = isrow;
3999b48462bSShri Abhyankar   b->col        = iscol;
4009b48462bSShri Abhyankar   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
4019b48462bSShri Abhyankar   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
4029b48462bSShri Abhyankar   b->icol       = isicol;
4039b48462bSShri Abhyankar   ierr          = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
4049b48462bSShri Abhyankar 
4059b48462bSShri Abhyankar   /* In b structure:  Free imax, ilen, old a, old j.  Allocate solve_work, new a, new j */
4069b48462bSShri Abhyankar   ierr = PetscLogObjectMemory(B,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
4079b48462bSShri Abhyankar   b->maxnz = b->nz = bdiag[0]+1;
408d5f3da31SBarry Smith   B->factortype            = MAT_FACTOR_LU;
409f284f12aSHong Zhang   B->info.factor_mallocs   = reallocs;
410f284f12aSHong Zhang   B->info.fill_ratio_given = f;
4119b48462bSShri Abhyankar 
4129b48462bSShri Abhyankar   if (ai[n]) {
413f284f12aSHong Zhang     B->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]);
4149b48462bSShri Abhyankar   } else {
415f284f12aSHong Zhang     B->info.fill_ratio_needed = 0.0;
4169b48462bSShri Abhyankar   }
4179263d837SHong Zhang #if defined(PETSC_USE_INFO)
4189263d837SHong Zhang   if (ai[n] != 0) {
4199263d837SHong Zhang     PetscReal af = B->info.fill_ratio_needed;
4209263d837SHong Zhang     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr);
4219263d837SHong Zhang     ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
4229263d837SHong Zhang     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr);
4239263d837SHong Zhang     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
4249263d837SHong Zhang   } else {
4259263d837SHong Zhang     ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr);
4269263d837SHong Zhang   }
4279263d837SHong Zhang #endif
42835233d99SShri Abhyankar   B->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ;
429f2344125SBarry Smith   if (a->inode.size) {
430d3ac4fa3SBarry Smith     B->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode;
431d3ac4fa3SBarry Smith   }
4329b48462bSShri Abhyankar   PetscFunctionReturn(0);
4339b48462bSShri Abhyankar }
4349b48462bSShri Abhyankar 
4355b5bc046SBarry Smith /*
4365b5bc046SBarry Smith     Trouble in factorization, should we dump the original matrix?
4375b5bc046SBarry Smith */
4385b5bc046SBarry Smith #undef __FUNCT__
4395b5bc046SBarry Smith #define __FUNCT__ "MatFactorDumpMatrix"
4405b5bc046SBarry Smith PetscErrorCode MatFactorDumpMatrix(Mat A)
4415b5bc046SBarry Smith {
4425b5bc046SBarry Smith   PetscErrorCode ierr;
443ace3abfcSBarry Smith   PetscBool      flg = PETSC_FALSE;
4445b5bc046SBarry Smith 
4455b5bc046SBarry Smith   PetscFunctionBegin;
446acfcf0e5SJed Brown   ierr = PetscOptionsGetBool(PETSC_NULL,"-mat_factor_dump_on_error",&flg,PETSC_NULL);CHKERRQ(ierr);
4475b5bc046SBarry Smith   if (flg) {
4485b5bc046SBarry Smith     PetscViewer viewer;
4495b5bc046SBarry Smith     char        filename[PETSC_MAX_PATH_LEN];
4505b5bc046SBarry Smith 
4515b5bc046SBarry Smith     ierr = PetscSNPrintf(filename,PETSC_MAX_PATH_LEN,"matrix_factor_error.%d",PetscGlobalRank);CHKERRQ(ierr);
4527adad957SLisandro Dalcin     ierr = PetscViewerBinaryOpen(((PetscObject)A)->comm,filename,FILE_MODE_WRITE,&viewer);CHKERRQ(ierr);
4535b5bc046SBarry Smith     ierr = MatView(A,viewer);CHKERRQ(ierr);
4546bf464f9SBarry Smith     ierr = PetscViewerDestroy(&viewer);CHKERRQ(ierr);
4555b5bc046SBarry Smith   }
4565b5bc046SBarry Smith   PetscFunctionReturn(0);
4575b5bc046SBarry Smith }
4585b5bc046SBarry Smith 
459c9c72f8fSHong Zhang #undef __FUNCT__
46035233d99SShri Abhyankar #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ"
46135233d99SShri Abhyankar PetscErrorCode MatLUFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info)
462c9c72f8fSHong Zhang {
463c9c72f8fSHong Zhang   Mat              C=B;
464c9c72f8fSHong Zhang   Mat_SeqAIJ       *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data;
465c9c72f8fSHong Zhang   IS               isrow = b->row,isicol = b->icol;
466c9c72f8fSHong Zhang   PetscErrorCode   ierr;
467c9c72f8fSHong Zhang   const PetscInt   *r,*ic,*ics;
468bbd65245SShri Abhyankar   const PetscInt   n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j,*bdiag=b->diag;
469bbd65245SShri Abhyankar   PetscInt         i,j,k,nz,nzL,row,*pj;
470bbd65245SShri Abhyankar   const PetscInt   *ajtmp,*bjtmp;
471bbd65245SShri Abhyankar   MatScalar        *rtmp,*pc,multiplier,*pv;
472bbd65245SShri Abhyankar   const  MatScalar *aa=a->a,*v;
473ace3abfcSBarry Smith   PetscBool        row_identity,col_identity;
474d90ac83dSHong Zhang   FactorShiftCtx   sctx;
4754f81c4b7SBarry Smith   const PetscInt   *ddiag;
476d4ad06f7SHong Zhang   PetscReal        rs;
477d4ad06f7SHong Zhang   MatScalar        d;
478d4ad06f7SHong Zhang 
479c9c72f8fSHong Zhang   PetscFunctionBegin;
480d6e5152cSHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
481d90ac83dSHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
482d4ad06f7SHong Zhang 
483f4db908eSBarry Smith   if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
484d4ad06f7SHong Zhang     ddiag          = a->diag;
485d4ad06f7SHong Zhang     sctx.shift_top = info->zeropivot;
486d4ad06f7SHong Zhang     for (i=0; i<n; i++) {
487d4ad06f7SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
488d4ad06f7SHong Zhang       d  = (aa)[ddiag[i]];
489d4ad06f7SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
490d4ad06f7SHong Zhang       v  = aa+ai[i];
491d4ad06f7SHong Zhang       nz = ai[i+1] - ai[i];
492d4ad06f7SHong Zhang       for (j=0; j<nz; j++)
493d4ad06f7SHong Zhang 	rs += PetscAbsScalar(v[j]);
494d4ad06f7SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
495d4ad06f7SHong Zhang     }
496d4ad06f7SHong Zhang     sctx.shift_top   *= 1.1;
497d4ad06f7SHong Zhang     sctx.nshift_max   = 5;
498d4ad06f7SHong Zhang     sctx.shift_lo     = 0.;
499d4ad06f7SHong Zhang     sctx.shift_hi     = 1.;
500d4ad06f7SHong Zhang   }
501d4ad06f7SHong Zhang 
502c9c72f8fSHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
503c9c72f8fSHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
504c9c72f8fSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr);
505c9c72f8fSHong Zhang   ics  = ic;
506c9c72f8fSHong Zhang 
507d4ad06f7SHong Zhang   do {
50807b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
509c9c72f8fSHong Zhang     for (i=0; i<n; i++){
510c9c72f8fSHong Zhang       /* zero rtmp */
511c9c72f8fSHong Zhang       /* L part */
512c9c72f8fSHong Zhang       nz    = bi[i+1] - bi[i];
513c9c72f8fSHong Zhang       bjtmp = bj + bi[i];
514c9c72f8fSHong Zhang       for  (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
515c9c72f8fSHong Zhang 
516c9c72f8fSHong Zhang       /* U part */
51762fbd6afSShri Abhyankar       nz = bdiag[i]-bdiag[i+1];
51862fbd6afSShri Abhyankar       bjtmp = bj + bdiag[i+1]+1;
519813a1f2bSShri Abhyankar       for  (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
520813a1f2bSShri Abhyankar 
521813a1f2bSShri Abhyankar       /* load in initial (unfactored row) */
522813a1f2bSShri Abhyankar       nz    = ai[r[i]+1] - ai[r[i]];
523813a1f2bSShri Abhyankar       ajtmp = aj + ai[r[i]];
524813a1f2bSShri Abhyankar       v     = aa + ai[r[i]];
525813a1f2bSShri Abhyankar       for (j=0; j<nz; j++) {
526813a1f2bSShri Abhyankar         rtmp[ics[ajtmp[j]]] = v[j];
527813a1f2bSShri Abhyankar       }
528813a1f2bSShri Abhyankar       /* ZeropivotApply() */
52998a93161SHong Zhang       rtmp[i] += sctx.shift_amount;  /* shift the diagonal of the matrix */
530813a1f2bSShri Abhyankar 
531813a1f2bSShri Abhyankar       /* elimination */
532813a1f2bSShri Abhyankar       bjtmp = bj + bi[i];
533813a1f2bSShri Abhyankar       row   = *bjtmp++;
534813a1f2bSShri Abhyankar       nzL   = bi[i+1] - bi[i];
535f268cba8SShri Abhyankar       for(k=0; k < nzL;k++) {
536813a1f2bSShri Abhyankar         pc = rtmp + row;
537813a1f2bSShri Abhyankar         if (*pc != 0.0) {
538813a1f2bSShri Abhyankar           pv         = b->a + bdiag[row];
539813a1f2bSShri Abhyankar           multiplier = *pc * (*pv);
540813a1f2bSShri Abhyankar           *pc        = multiplier;
54162fbd6afSShri Abhyankar           pj = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */
54262fbd6afSShri Abhyankar 	  pv = b->a + bdiag[row+1]+1;
54362fbd6afSShri Abhyankar 	  nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */
544813a1f2bSShri Abhyankar           for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j];
54528b1a77fSLisandro Dalcin           ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr);
546813a1f2bSShri Abhyankar         }
547f268cba8SShri Abhyankar         row = *bjtmp++;
548813a1f2bSShri Abhyankar       }
549813a1f2bSShri Abhyankar 
550813a1f2bSShri Abhyankar       /* finished row so stick it into b->a */
551813a1f2bSShri Abhyankar       rs = 0.0;
552813a1f2bSShri Abhyankar       /* L part */
553813a1f2bSShri Abhyankar       pv   = b->a + bi[i] ;
554813a1f2bSShri Abhyankar       pj   = b->j + bi[i] ;
555813a1f2bSShri Abhyankar       nz   = bi[i+1] - bi[i];
556813a1f2bSShri Abhyankar       for (j=0; j<nz; j++) {
557813a1f2bSShri Abhyankar         pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]);
558813a1f2bSShri Abhyankar       }
559813a1f2bSShri Abhyankar 
560813a1f2bSShri Abhyankar       /* U part */
56162fbd6afSShri Abhyankar       pv = b->a + bdiag[i+1]+1;
56262fbd6afSShri Abhyankar       pj = b->j + bdiag[i+1]+1;
56362fbd6afSShri Abhyankar       nz = bdiag[i] - bdiag[i+1]-1;
564813a1f2bSShri Abhyankar       for (j=0; j<nz; j++) {
565813a1f2bSShri Abhyankar         pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]);
566813a1f2bSShri Abhyankar       }
567813a1f2bSShri Abhyankar 
568813a1f2bSShri Abhyankar       sctx.rs  = rs;
569813a1f2bSShri Abhyankar       sctx.pv  = rtmp[i];
570d0660788SBarry Smith       ierr = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr);
57107b50cabSHong Zhang       if(sctx.newshift) break; /* break for-loop */
57207b50cabSHong Zhang       rtmp[i] = sctx.pv; /* sctx.pv might be updated in the case of MAT_SHIFT_INBLOCKS */
573813a1f2bSShri Abhyankar 
574813a1f2bSShri Abhyankar       /* Mark diagonal and invert diagonal for simplier triangular solves */
575813a1f2bSShri Abhyankar       pv  = b->a + bdiag[i];
576813a1f2bSShri Abhyankar       *pv = 1.0/rtmp[i];
577813a1f2bSShri Abhyankar 
578813a1f2bSShri Abhyankar     } /* endof for (i=0; i<n; i++){ */
579813a1f2bSShri Abhyankar 
5808ff23777SHong Zhang     /* MatPivotRefine() */
58107b50cabSHong Zhang     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max){
582813a1f2bSShri Abhyankar       /*
583813a1f2bSShri Abhyankar        * if no shift in this attempt & shifting & started shifting & can refine,
584813a1f2bSShri Abhyankar        * then try lower shift
585813a1f2bSShri Abhyankar        */
586813a1f2bSShri Abhyankar       sctx.shift_hi       = sctx.shift_fraction;
587813a1f2bSShri Abhyankar       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
588813a1f2bSShri Abhyankar       sctx.shift_amount   = sctx.shift_fraction * sctx.shift_top;
58907b50cabSHong Zhang       sctx.newshift       = PETSC_TRUE;
590813a1f2bSShri Abhyankar       sctx.nshift++;
591813a1f2bSShri Abhyankar     }
59207b50cabSHong Zhang   } while (sctx.newshift);
593813a1f2bSShri Abhyankar 
594813a1f2bSShri Abhyankar   ierr = PetscFree(rtmp);CHKERRQ(ierr);
595813a1f2bSShri Abhyankar   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
596813a1f2bSShri Abhyankar   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
597d3ac4fa3SBarry Smith 
598813a1f2bSShri Abhyankar   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
599813a1f2bSShri Abhyankar   ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr);
600813a1f2bSShri Abhyankar   if (row_identity && col_identity) {
60135233d99SShri Abhyankar     C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering;
602813a1f2bSShri Abhyankar   } else {
60335233d99SShri Abhyankar     C->ops->solve = MatSolve_SeqAIJ;
604813a1f2bSShri Abhyankar   }
60535233d99SShri Abhyankar   C->ops->solveadd           = MatSolveAdd_SeqAIJ;
60635233d99SShri Abhyankar   C->ops->solvetranspose     = MatSolveTranspose_SeqAIJ;
60735233d99SShri Abhyankar   C->ops->solvetransposeadd  = MatSolveTransposeAdd_SeqAIJ;
60835233d99SShri Abhyankar   C->ops->matsolve           = MatMatSolve_SeqAIJ;
609813a1f2bSShri Abhyankar   C->assembled    = PETSC_TRUE;
610813a1f2bSShri Abhyankar   C->preallocated = PETSC_TRUE;
611813a1f2bSShri Abhyankar   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
612813a1f2bSShri Abhyankar 
61314d2772eSHong Zhang   /* MatShiftView(A,info,&sctx) */
614813a1f2bSShri Abhyankar   if (sctx.nshift){
615f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
616813a1f2bSShri 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);
617f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
618813a1f2bSShri Abhyankar       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
619f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_INBLOCKS){
620915743fcSHong Zhang       ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %G\n",sctx.nshift,info->shiftamount);CHKERRQ(ierr);
621813a1f2bSShri Abhyankar     }
622813a1f2bSShri Abhyankar   }
623d3ac4fa3SBarry Smith   ierr = Mat_CheckInode_FactorLU(C,PETSC_FALSE);CHKERRQ(ierr);
624813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
625813a1f2bSShri Abhyankar }
626813a1f2bSShri Abhyankar 
6274a2ae208SSatish Balay #undef __FUNCT__
628ad04f41aSHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_inplace"
629ad04f41aSHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ_inplace(Mat B,Mat A,const MatFactorInfo *info)
630289bc588SBarry Smith {
631719d5645SBarry Smith   Mat             C=B;
632416022c9SBarry Smith   Mat_SeqAIJ      *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data;
63382bf6240SBarry Smith   IS              isrow = b->row,isicol = b->icol;
6346849ba73SBarry Smith   PetscErrorCode  ierr;
6355d0c19d7SBarry Smith   const PetscInt   *r,*ic,*ics;
636d278edc7SBarry Smith   PetscInt        nz,row,i,j,n=A->rmap->n,diag;
637d278edc7SBarry Smith   const PetscInt  *ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j;
638d278edc7SBarry Smith   const PetscInt  *ajtmp,*bjtmp,*diag_offset = b->diag,*pj;
639d278edc7SBarry Smith   MatScalar       *pv,*rtmp,*pc,multiplier,d;
640d278edc7SBarry Smith   const MatScalar *v,*aa=a->a;
64133f9893dSHong Zhang   PetscReal       rs=0.0;
642d90ac83dSHong Zhang   FactorShiftCtx  sctx;
6434f81c4b7SBarry Smith   const PetscInt  *ddiag;
644ace3abfcSBarry Smith   PetscBool       row_identity, col_identity;
645289bc588SBarry Smith 
6463a40ed3dSBarry Smith   PetscFunctionBegin;
647504a3fadSHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
648504a3fadSHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
649ada7143aSSatish Balay 
650f4db908eSBarry Smith   if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
65142898933SHong Zhang     ddiag          = a->diag;
6529f7d0b68SHong Zhang     sctx.shift_top = info->zeropivot;
6536cc28720Svictorle     for (i=0; i<n; i++) {
6549f95998fSHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
6559f7d0b68SHong Zhang       d  = (aa)[ddiag[i]];
6561a890ab1SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
6579f7d0b68SHong Zhang       v  = aa+ai[i];
6589f7d0b68SHong Zhang       nz = ai[i+1] - ai[i];
659e24cfe64SHong Zhang       for (j=0; j<nz; j++)
6601a890ab1SHong Zhang 	rs += PetscAbsScalar(v[j]);
6611a890ab1SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
6626cc28720Svictorle     }
663118f5789SBarry Smith     sctx.shift_top   *= 1.1;
664d2276718SHong Zhang     sctx.nshift_max   = 5;
665d2276718SHong Zhang     sctx.shift_lo     = 0.;
666d2276718SHong Zhang     sctx.shift_hi     = 1.;
667a0a356efSHong Zhang   }
668a0a356efSHong Zhang 
669504a3fadSHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
670504a3fadSHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
671504a3fadSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr);
672504a3fadSHong Zhang   ics  = ic;
673504a3fadSHong Zhang 
674085256b3SBarry Smith   do {
67507b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
676289bc588SBarry Smith     for (i=0; i<n; i++){
677b3bf805bSHong Zhang       nz    = bi[i+1] - bi[i];
678b3bf805bSHong Zhang       bjtmp = bj + bi[i];
6799f7d0b68SHong Zhang       for  (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
680289bc588SBarry Smith 
681289bc588SBarry Smith       /* load in initial (unfactored row) */
6829f7d0b68SHong Zhang       nz    = ai[r[i]+1] - ai[r[i]];
6839f7d0b68SHong Zhang       ajtmp = aj + ai[r[i]];
6849f7d0b68SHong Zhang       v     = aa + ai[r[i]];
685085256b3SBarry Smith       for (j=0; j<nz; j++) {
686b3bf805bSHong Zhang         rtmp[ics[ajtmp[j]]] = v[j];
687085256b3SBarry Smith       }
688d2276718SHong Zhang       rtmp[ics[r[i]]] += sctx.shift_amount; /* shift the diagonal of the matrix */
689289bc588SBarry Smith 
690b3bf805bSHong Zhang       row = *bjtmp++;
691f2582111SSatish Balay       while  (row < i) {
6928c37ef55SBarry Smith         pc = rtmp + row;
6938c37ef55SBarry Smith         if (*pc != 0.0) {
694010a20caSHong Zhang           pv         = b->a + diag_offset[row];
695010a20caSHong Zhang           pj         = b->j + diag_offset[row] + 1;
69635aab85fSBarry Smith           multiplier = *pc / *pv++;
6978c37ef55SBarry Smith           *pc        = multiplier;
698b3bf805bSHong Zhang           nz         = bi[row+1] - diag_offset[row] - 1;
6999f7d0b68SHong Zhang           for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j];
70028b1a77fSLisandro Dalcin           ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr);
701289bc588SBarry Smith         }
702b3bf805bSHong Zhang         row = *bjtmp++;
703289bc588SBarry Smith       }
704416022c9SBarry Smith       /* finished row so stick it into b->a */
705b3bf805bSHong Zhang       pv   = b->a + bi[i] ;
706b3bf805bSHong Zhang       pj   = b->j + bi[i] ;
707b3bf805bSHong Zhang       nz   = bi[i+1] - bi[i];
708b3bf805bSHong Zhang       diag = diag_offset[i] - bi[i];
709e57ff13aSBarry Smith       rs   = 0.0;
710d3d32019SBarry Smith       for (j=0; j<nz; j++) {
7119f7d0b68SHong Zhang         pv[j] = rtmp[pj[j]];
7129f7d0b68SHong Zhang         rs   += PetscAbsScalar(pv[j]);
713d3d32019SBarry Smith       }
714e57ff13aSBarry Smith       rs   -= PetscAbsScalar(pv[diag]);
715d2276718SHong Zhang 
716d2276718SHong Zhang       sctx.rs  = rs;
717d2276718SHong Zhang       sctx.pv  = pv[diag];
718d0660788SBarry Smith       ierr = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr);
71907b50cabSHong Zhang       if (sctx.newshift) break;
720504a3fadSHong Zhang       pv[diag] = sctx.pv;
72135aab85fSBarry Smith     }
722d2276718SHong Zhang 
72307b50cabSHong Zhang     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) {
7246cc28720Svictorle       /*
7256c037d1bSvictorle        * if no shift in this attempt & shifting & started shifting & can refine,
7266cc28720Svictorle        * then try lower shift
7276cc28720Svictorle        */
7280481f469SBarry Smith       sctx.shift_hi       = sctx.shift_fraction;
7290481f469SBarry Smith       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
7300481f469SBarry Smith       sctx.shift_amount   = sctx.shift_fraction * sctx.shift_top;
73107b50cabSHong Zhang       sctx.newshift       = PETSC_TRUE;
732d2276718SHong Zhang       sctx.nshift++;
7336cc28720Svictorle     }
73407b50cabSHong Zhang   } while (sctx.newshift);
7350f11f9aeSSatish Balay 
73635aab85fSBarry Smith   /* invert diagonal entries for simplier triangular solves */
73735aab85fSBarry Smith   for (i=0; i<n; i++) {
738010a20caSHong Zhang     b->a[diag_offset[i]] = 1.0/b->a[diag_offset[i]];
73935aab85fSBarry Smith   }
740606d414cSSatish Balay   ierr = PetscFree(rtmp);CHKERRQ(ierr);
74178b31e54SBarry Smith   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
74278b31e54SBarry Smith   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
743d3ac4fa3SBarry Smith 
7448d8c7c43SBarry Smith   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
7458d8c7c43SBarry Smith   ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr);
7468d8c7c43SBarry Smith   if (row_identity && col_identity) {
747ad04f41aSHong Zhang     C->ops->solve   = MatSolve_SeqAIJ_NaturalOrdering_inplace;
7488d8c7c43SBarry Smith   } else {
749ad04f41aSHong Zhang     C->ops->solve   = MatSolve_SeqAIJ_inplace;
750db4efbfdSBarry Smith   }
751ad04f41aSHong Zhang   C->ops->solveadd           = MatSolveAdd_SeqAIJ_inplace;
752ad04f41aSHong Zhang   C->ops->solvetranspose     = MatSolveTranspose_SeqAIJ_inplace;
753ad04f41aSHong Zhang   C->ops->solvetransposeadd  = MatSolveTransposeAdd_SeqAIJ_inplace;
754ad04f41aSHong Zhang   C->ops->matsolve           = MatMatSolve_SeqAIJ_inplace;
755c456f294SBarry Smith   C->assembled    = PETSC_TRUE;
7565c9eb25fSBarry Smith   C->preallocated = PETSC_TRUE;
757d0f46423SBarry Smith   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
758d2276718SHong Zhang   if (sctx.nshift){
759f4db908eSBarry Smith      if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
7600481f469SBarry 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);
761f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
762e738e422SBarry Smith       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
7636cc28720Svictorle     }
7640697b6caSHong Zhang   }
765d3ac4fa3SBarry Smith   (C)->ops->solve            = MatSolve_SeqAIJ_inplace;
766d3ac4fa3SBarry Smith   (C)->ops->solvetranspose   = MatSolveTranspose_SeqAIJ_inplace;
767d3ac4fa3SBarry Smith   ierr = Mat_CheckInode(C,PETSC_FALSE);CHKERRQ(ierr);
7683a40ed3dSBarry Smith   PetscFunctionReturn(0);
769289bc588SBarry Smith }
7700889b5dcSvictorle 
771137fb511SHong Zhang /*
772137fb511SHong Zhang    This routine implements inplace ILU(0) with row or/and column permutations.
773137fb511SHong Zhang    Input:
774137fb511SHong Zhang      A - original matrix
775137fb511SHong Zhang    Output;
776137fb511SHong Zhang      A - a->i (rowptr) is same as original rowptr, but factored i-the row is stored in rowperm[i]
777137fb511SHong Zhang          a->j (col index) is permuted by the inverse of colperm, then sorted
778137fb511SHong Zhang          a->a reordered accordingly with a->j
779137fb511SHong Zhang          a->diag (ptr to diagonal elements) is updated.
780137fb511SHong Zhang */
781137fb511SHong Zhang #undef __FUNCT__
782137fb511SHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_InplaceWithPerm"
7830481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_SeqAIJ_InplaceWithPerm(Mat B,Mat A,const MatFactorInfo *info)
784137fb511SHong Zhang {
785b051339dSHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
786b051339dSHong Zhang   IS             isrow = a->row,isicol = a->icol;
787137fb511SHong Zhang   PetscErrorCode ierr;
7885d0c19d7SBarry Smith   const PetscInt *r,*ic,*ics;
7895d0c19d7SBarry Smith   PetscInt       i,j,n=A->rmap->n,*ai=a->i,*aj=a->j;
7905d0c19d7SBarry Smith   PetscInt       *ajtmp,nz,row;
791b051339dSHong Zhang   PetscInt       *diag = a->diag,nbdiag,*pj;
792a77337e4SBarry Smith   PetscScalar    *rtmp,*pc,multiplier,d;
793504a3fadSHong Zhang   MatScalar      *pv,*v;
794137fb511SHong Zhang   PetscReal      rs;
795d90ac83dSHong Zhang   FactorShiftCtx sctx;
796504a3fadSHong Zhang   const  MatScalar *aa=a->a,*vtmp;
797137fb511SHong Zhang 
798137fb511SHong Zhang   PetscFunctionBegin;
799e32f2f54SBarry Smith   if (A != B) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"input and output matrix must have same address");
800504a3fadSHong Zhang 
801504a3fadSHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
802504a3fadSHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
803504a3fadSHong Zhang 
804504a3fadSHong Zhang   if (info->shifttype == (PetscReal) MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
805504a3fadSHong Zhang     const PetscInt   *ddiag = a->diag;
806504a3fadSHong Zhang     sctx.shift_top = info->zeropivot;
807504a3fadSHong Zhang     for (i=0; i<n; i++) {
808504a3fadSHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
809504a3fadSHong Zhang       d  = (aa)[ddiag[i]];
810504a3fadSHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
811504a3fadSHong Zhang       vtmp  = aa+ai[i];
812504a3fadSHong Zhang       nz = ai[i+1] - ai[i];
813504a3fadSHong Zhang       for (j=0; j<nz; j++)
814504a3fadSHong Zhang 	rs += PetscAbsScalar(vtmp[j]);
815504a3fadSHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
816504a3fadSHong Zhang     }
817504a3fadSHong Zhang     sctx.shift_top   *= 1.1;
818504a3fadSHong Zhang     sctx.nshift_max   = 5;
819504a3fadSHong Zhang     sctx.shift_lo     = 0.;
820504a3fadSHong Zhang     sctx.shift_hi     = 1.;
821504a3fadSHong Zhang   }
822504a3fadSHong Zhang 
823137fb511SHong Zhang   ierr  = ISGetIndices(isrow,&r);CHKERRQ(ierr);
824137fb511SHong Zhang   ierr  = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
825137fb511SHong Zhang   ierr  = PetscMalloc((n+1)*sizeof(PetscScalar),&rtmp);CHKERRQ(ierr);
826137fb511SHong Zhang   ierr  = PetscMemzero(rtmp,(n+1)*sizeof(PetscScalar));CHKERRQ(ierr);
827b051339dSHong Zhang   ics = ic;
828137fb511SHong Zhang 
829504a3fadSHong Zhang #if defined(MV)
83075567043SBarry Smith   sctx.shift_top      = 0.;
831e60cf9a0SBarry Smith   sctx.nshift_max     = 0;
83275567043SBarry Smith   sctx.shift_lo       = 0.;
83375567043SBarry Smith   sctx.shift_hi       = 0.;
83475567043SBarry Smith   sctx.shift_fraction = 0.;
835137fb511SHong Zhang 
836f4db908eSBarry Smith   if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
83775567043SBarry Smith     sctx.shift_top = 0.;
838137fb511SHong Zhang     for (i=0; i<n; i++) {
839137fb511SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
840b051339dSHong Zhang       d  = (a->a)[diag[i]];
841137fb511SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
842b051339dSHong Zhang       v  = a->a+ai[i];
843b051339dSHong Zhang       nz = ai[i+1] - ai[i];
844137fb511SHong Zhang       for (j=0; j<nz; j++)
845137fb511SHong Zhang 	rs += PetscAbsScalar(v[j]);
846137fb511SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
847137fb511SHong Zhang     }
848137fb511SHong Zhang     if (sctx.shift_top < info->zeropivot) sctx.shift_top = info->zeropivot;
849137fb511SHong Zhang     sctx.shift_top    *= 1.1;
850137fb511SHong Zhang     sctx.nshift_max   = 5;
851137fb511SHong Zhang     sctx.shift_lo     = 0.;
852137fb511SHong Zhang     sctx.shift_hi     = 1.;
853137fb511SHong Zhang   }
854137fb511SHong Zhang 
85575567043SBarry Smith   sctx.shift_amount = 0.;
856137fb511SHong Zhang   sctx.nshift       = 0;
857504a3fadSHong Zhang #endif
858504a3fadSHong Zhang 
859137fb511SHong Zhang   do {
86007b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
861137fb511SHong Zhang     for (i=0; i<n; i++){
862b051339dSHong Zhang       /* load in initial unfactored row */
863b051339dSHong Zhang       nz    = ai[r[i]+1] - ai[r[i]];
864b051339dSHong Zhang       ajtmp = aj + ai[r[i]];
865b051339dSHong Zhang       v     = a->a + ai[r[i]];
866137fb511SHong Zhang       /* sort permuted ajtmp and values v accordingly */
867b051339dSHong Zhang       for (j=0; j<nz; j++) ajtmp[j] = ics[ajtmp[j]];
868137fb511SHong Zhang       ierr = PetscSortIntWithScalarArray(nz,ajtmp,v);CHKERRQ(ierr);
869137fb511SHong Zhang 
870b051339dSHong Zhang       diag[r[i]] = ai[r[i]];
871137fb511SHong Zhang       for (j=0; j<nz; j++) {
872137fb511SHong Zhang         rtmp[ajtmp[j]] = v[j];
873b051339dSHong Zhang         if (ajtmp[j] < i) diag[r[i]]++; /* update a->diag */
874137fb511SHong Zhang       }
875137fb511SHong Zhang       rtmp[r[i]] += sctx.shift_amount; /* shift the diagonal of the matrix */
876137fb511SHong Zhang 
877b051339dSHong Zhang       row = *ajtmp++;
878137fb511SHong Zhang       while  (row < i) {
879137fb511SHong Zhang         pc = rtmp + row;
880137fb511SHong Zhang         if (*pc != 0.0) {
881b051339dSHong Zhang           pv         = a->a + diag[r[row]];
882b051339dSHong Zhang           pj         = aj + diag[r[row]] + 1;
883137fb511SHong Zhang 
884137fb511SHong Zhang           multiplier = *pc / *pv++;
885137fb511SHong Zhang           *pc        = multiplier;
886b051339dSHong Zhang           nz         = ai[r[row]+1] - diag[r[row]] - 1;
887b051339dSHong Zhang           for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j];
88828b1a77fSLisandro Dalcin           ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr);
889137fb511SHong Zhang         }
890b051339dSHong Zhang         row = *ajtmp++;
891137fb511SHong Zhang       }
892b051339dSHong Zhang       /* finished row so overwrite it onto a->a */
893b051339dSHong Zhang       pv   = a->a + ai[r[i]] ;
894b051339dSHong Zhang       pj   = aj + ai[r[i]] ;
895b051339dSHong Zhang       nz   = ai[r[i]+1] - ai[r[i]];
896b051339dSHong Zhang       nbdiag = diag[r[i]] - ai[r[i]]; /* num of entries before the diagonal */
897137fb511SHong Zhang 
898137fb511SHong Zhang       rs   = 0.0;
899137fb511SHong Zhang       for (j=0; j<nz; j++) {
900b051339dSHong Zhang         pv[j] = rtmp[pj[j]];
901b051339dSHong Zhang         if (j != nbdiag) rs += PetscAbsScalar(pv[j]);
902137fb511SHong Zhang       }
903137fb511SHong Zhang 
904137fb511SHong Zhang       sctx.rs  = rs;
905b051339dSHong Zhang       sctx.pv  = pv[nbdiag];
906d0660788SBarry Smith       ierr = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr);
90707b50cabSHong Zhang       if (sctx.newshift) break;
908504a3fadSHong Zhang       pv[nbdiag] = sctx.pv;
909137fb511SHong Zhang     }
910137fb511SHong Zhang 
91107b50cabSHong Zhang     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE && !sctx.newshift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) {
912137fb511SHong Zhang       /*
913137fb511SHong Zhang        * if no shift in this attempt & shifting & started shifting & can refine,
914137fb511SHong Zhang        * then try lower shift
915137fb511SHong Zhang        */
9160481f469SBarry Smith       sctx.shift_hi        = sctx.shift_fraction;
9170481f469SBarry Smith       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
9180481f469SBarry Smith       sctx.shift_amount    = sctx.shift_fraction * sctx.shift_top;
91907b50cabSHong Zhang       sctx.newshift         = PETSC_TRUE;
920137fb511SHong Zhang       sctx.nshift++;
921137fb511SHong Zhang     }
92207b50cabSHong Zhang   } while (sctx.newshift);
923137fb511SHong Zhang 
924137fb511SHong Zhang   /* invert diagonal entries for simplier triangular solves */
925137fb511SHong Zhang   for (i=0; i<n; i++) {
926b051339dSHong Zhang     a->a[diag[r[i]]] = 1.0/a->a[diag[r[i]]];
927137fb511SHong Zhang   }
928137fb511SHong Zhang 
929137fb511SHong Zhang   ierr = PetscFree(rtmp);CHKERRQ(ierr);
930137fb511SHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
931137fb511SHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
932b051339dSHong Zhang   A->ops->solve             = MatSolve_SeqAIJ_InplaceWithPerm;
933ad04f41aSHong Zhang   A->ops->solveadd          = MatSolveAdd_SeqAIJ_inplace;
934ad04f41aSHong Zhang   A->ops->solvetranspose    = MatSolveTranspose_SeqAIJ_inplace;
935ad04f41aSHong Zhang   A->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ_inplace;
936b051339dSHong Zhang   A->assembled = PETSC_TRUE;
9375c9eb25fSBarry Smith   A->preallocated = PETSC_TRUE;
938d0f46423SBarry Smith   ierr = PetscLogFlops(A->cmap->n);CHKERRQ(ierr);
939137fb511SHong Zhang   if (sctx.nshift){
940f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
9410481f469SBarry 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);
942f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
943e738e422SBarry Smith       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
944137fb511SHong Zhang     }
945137fb511SHong Zhang   }
946137fb511SHong Zhang   PetscFunctionReturn(0);
947137fb511SHong Zhang }
948137fb511SHong Zhang 
9490a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
9504a2ae208SSatish Balay #undef __FUNCT__
9514a2ae208SSatish Balay #define __FUNCT__ "MatLUFactor_SeqAIJ"
9520481f469SBarry Smith PetscErrorCode MatLUFactor_SeqAIJ(Mat A,IS row,IS col,const MatFactorInfo *info)
953da3a660dSBarry Smith {
954dfbe8321SBarry Smith   PetscErrorCode ierr;
955416022c9SBarry Smith   Mat            C;
956416022c9SBarry Smith 
9573a40ed3dSBarry Smith   PetscFunctionBegin;
9582692d6eeSBarry Smith   ierr = MatGetFactor(A,MATSOLVERPETSC,MAT_FACTOR_LU,&C);CHKERRQ(ierr);
959719d5645SBarry Smith   ierr = MatLUFactorSymbolic(C,A,row,col,info);CHKERRQ(ierr);
960719d5645SBarry Smith   ierr = MatLUFactorNumeric(C,A,info);CHKERRQ(ierr);
961db4efbfdSBarry Smith   A->ops->solve            = C->ops->solve;
962db4efbfdSBarry Smith   A->ops->solvetranspose   = C->ops->solvetranspose;
963eb6b5d47SBarry Smith   ierr = MatHeaderMerge(A,C);CHKERRQ(ierr);
96452e6d16bSBarry Smith   ierr = PetscLogObjectParent(A,((Mat_SeqAIJ*)(A->data))->icol);CHKERRQ(ierr);
9653a40ed3dSBarry Smith   PetscFunctionReturn(0);
966da3a660dSBarry Smith }
9670a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
9681d098868SHong Zhang 
969889a8dedSBarry Smith 
9704a2ae208SSatish Balay #undef __FUNCT__
971ad04f41aSHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_inplace"
972ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_inplace(Mat A,Vec bb,Vec xx)
9738c37ef55SBarry Smith {
974416022c9SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
975416022c9SBarry Smith   IS                iscol = a->col,isrow = a->row;
9766849ba73SBarry Smith   PetscErrorCode    ierr;
9775d0c19d7SBarry Smith   PetscInt          i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
9785d0c19d7SBarry Smith   PetscInt          nz;
9795d0c19d7SBarry Smith   const PetscInt    *rout,*cout,*r,*c;
980dd6ea824SBarry Smith   PetscScalar       *x,*tmp,*tmps,sum;
981d9fead3dSBarry Smith   const PetscScalar *b;
982dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
9838c37ef55SBarry Smith 
9843a40ed3dSBarry Smith   PetscFunctionBegin;
9853a40ed3dSBarry Smith   if (!n) PetscFunctionReturn(0);
98691bf9adeSBarry Smith 
9873649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
9881ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
989416022c9SBarry Smith   tmp  = a->solve_work;
9908c37ef55SBarry Smith 
9913b2fbd54SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
9923b2fbd54SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
9938c37ef55SBarry Smith 
9948c37ef55SBarry Smith   /* forward solve the lower triangular */
9958c37ef55SBarry Smith   tmp[0] = b[*r++];
996010a20caSHong Zhang   tmps   = tmp;
9978c37ef55SBarry Smith   for (i=1; i<n; i++) {
998010a20caSHong Zhang     v   = aa + ai[i] ;
999010a20caSHong Zhang     vi  = aj + ai[i] ;
100053e7425aSBarry Smith     nz  = a->diag[i] - ai[i];
100153e7425aSBarry Smith     sum = b[*r++];
1002003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
10038c37ef55SBarry Smith     tmp[i] = sum;
10048c37ef55SBarry Smith   }
10058c37ef55SBarry Smith 
10068c37ef55SBarry Smith   /* backward solve the upper triangular */
10078c37ef55SBarry Smith   for (i=n-1; i>=0; i--){
1008010a20caSHong Zhang     v   = aa + a->diag[i] + 1;
1009010a20caSHong Zhang     vi  = aj + a->diag[i] + 1;
1010416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
10118c37ef55SBarry Smith     sum = tmp[i];
1012003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
1013010a20caSHong Zhang     x[*c--] = tmp[i] = sum*aa[a->diag[i]];
10148c37ef55SBarry Smith   }
10158c37ef55SBarry Smith 
10163b2fbd54SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
10173b2fbd54SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
10183649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
10191ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1020dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
10213a40ed3dSBarry Smith   PetscFunctionReturn(0);
10228c37ef55SBarry Smith }
1023026e39d0SSatish Balay 
10242bebee5dSHong Zhang #undef __FUNCT__
1025ad04f41aSHong Zhang #define __FUNCT__ "MatMatSolve_SeqAIJ_inplace"
1026ad04f41aSHong Zhang PetscErrorCode MatMatSolve_SeqAIJ_inplace(Mat A,Mat B,Mat X)
10272bebee5dSHong Zhang {
10282bebee5dSHong Zhang   Mat_SeqAIJ      *a = (Mat_SeqAIJ*)A->data;
10292bebee5dSHong Zhang   IS              iscol = a->col,isrow = a->row;
10302bebee5dSHong Zhang   PetscErrorCode  ierr;
10315d0c19d7SBarry Smith   PetscInt        i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
10325d0c19d7SBarry Smith   PetscInt        nz,neq;
10335d0c19d7SBarry Smith   const PetscInt  *rout,*cout,*r,*c;
1034dd6ea824SBarry Smith   PetscScalar     *x,*b,*tmp,*tmps,sum;
1035dd6ea824SBarry Smith   const MatScalar *aa = a->a,*v;
1036ace3abfcSBarry Smith   PetscBool       bisdense,xisdense;
10372bebee5dSHong Zhang 
10382bebee5dSHong Zhang   PetscFunctionBegin;
10392bebee5dSHong Zhang   if (!n) PetscFunctionReturn(0);
10402bebee5dSHong Zhang 
1041251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr);
1042e32f2f54SBarry Smith   if (!bisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix");
1043251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr);
1044e32f2f54SBarry Smith   if (!xisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix");
1045db4efbfdSBarry Smith 
10462bebee5dSHong Zhang   ierr = MatGetArray(B,&b);CHKERRQ(ierr);
10472bebee5dSHong Zhang   ierr = MatGetArray(X,&x);CHKERRQ(ierr);
10482bebee5dSHong Zhang 
10492bebee5dSHong Zhang   tmp  = a->solve_work;
10502bebee5dSHong Zhang   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
10512bebee5dSHong Zhang   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
10522bebee5dSHong Zhang 
1053a36b22ccSSatish Balay   for (neq=0; neq<B->cmap->n; neq++){
10542bebee5dSHong Zhang     /* forward solve the lower triangular */
10552bebee5dSHong Zhang     tmp[0] = b[r[0]];
10562bebee5dSHong Zhang     tmps   = tmp;
10572bebee5dSHong Zhang     for (i=1; i<n; i++) {
10582bebee5dSHong Zhang       v   = aa + ai[i] ;
10592bebee5dSHong Zhang       vi  = aj + ai[i] ;
10602bebee5dSHong Zhang       nz  = a->diag[i] - ai[i];
10612bebee5dSHong Zhang       sum = b[r[i]];
1062003131ecSBarry Smith       PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
10632bebee5dSHong Zhang       tmp[i] = sum;
10642bebee5dSHong Zhang     }
10652bebee5dSHong Zhang     /* backward solve the upper triangular */
10662bebee5dSHong Zhang     for (i=n-1; i>=0; i--){
10672bebee5dSHong Zhang       v   = aa + a->diag[i] + 1;
10682bebee5dSHong Zhang       vi  = aj + a->diag[i] + 1;
10692bebee5dSHong Zhang       nz  = ai[i+1] - a->diag[i] - 1;
10702bebee5dSHong Zhang       sum = tmp[i];
1071003131ecSBarry Smith       PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
10722bebee5dSHong Zhang       x[c[i]] = tmp[i] = sum*aa[a->diag[i]];
10732bebee5dSHong Zhang     }
10742bebee5dSHong Zhang 
10752bebee5dSHong Zhang     b += n;
10762bebee5dSHong Zhang     x += n;
10772bebee5dSHong Zhang   }
10782bebee5dSHong Zhang   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
10792bebee5dSHong Zhang   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
10802bebee5dSHong Zhang   ierr = MatRestoreArray(B,&b);CHKERRQ(ierr);
10812bebee5dSHong Zhang   ierr = MatRestoreArray(X,&x);CHKERRQ(ierr);
1082dc0b31edSSatish Balay   ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr);
10832bebee5dSHong Zhang   PetscFunctionReturn(0);
10842bebee5dSHong Zhang }
10852bebee5dSHong Zhang 
1086137fb511SHong Zhang #undef __FUNCT__
108735233d99SShri Abhyankar #define __FUNCT__ "MatMatSolve_SeqAIJ"
108835233d99SShri Abhyankar PetscErrorCode MatMatSolve_SeqAIJ(Mat A,Mat B,Mat X)
10899bd0847aSShri Abhyankar {
10909bd0847aSShri Abhyankar   Mat_SeqAIJ      *a = (Mat_SeqAIJ*)A->data;
10919bd0847aSShri Abhyankar   IS              iscol = a->col,isrow = a->row;
10929bd0847aSShri Abhyankar   PetscErrorCode  ierr;
10939bd0847aSShri Abhyankar   PetscInt        i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j,*adiag = a->diag;
10949bd0847aSShri Abhyankar   PetscInt        nz,neq;
10959bd0847aSShri Abhyankar   const PetscInt  *rout,*cout,*r,*c;
10969bd0847aSShri Abhyankar   PetscScalar     *x,*b,*tmp,sum;
10979bd0847aSShri Abhyankar   const MatScalar *aa = a->a,*v;
1098ace3abfcSBarry Smith   PetscBool       bisdense,xisdense;
10999bd0847aSShri Abhyankar 
11009bd0847aSShri Abhyankar   PetscFunctionBegin;
11019bd0847aSShri Abhyankar   if (!n) PetscFunctionReturn(0);
11029bd0847aSShri Abhyankar 
1103251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr);
1104e32f2f54SBarry Smith   if (!bisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix");
1105251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr);
1106e32f2f54SBarry Smith   if (!xisdense) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix");
11079bd0847aSShri Abhyankar 
11089bd0847aSShri Abhyankar   ierr = MatGetArray(B,&b);CHKERRQ(ierr);
11099bd0847aSShri Abhyankar   ierr = MatGetArray(X,&x);CHKERRQ(ierr);
11109bd0847aSShri Abhyankar 
11119bd0847aSShri Abhyankar   tmp  = a->solve_work;
11129bd0847aSShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
11139bd0847aSShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
11149bd0847aSShri Abhyankar 
11159bd0847aSShri Abhyankar   for (neq=0; neq<B->cmap->n; neq++){
11169bd0847aSShri Abhyankar     /* forward solve the lower triangular */
11179bd0847aSShri Abhyankar     tmp[0] = b[r[0]];
11189bd0847aSShri Abhyankar     v      = aa;
11199bd0847aSShri Abhyankar     vi     = aj;
11209bd0847aSShri Abhyankar     for (i=1; i<n; i++) {
11219bd0847aSShri Abhyankar       nz  = ai[i+1] - ai[i];
11229bd0847aSShri Abhyankar       sum = b[r[i]];
11239bd0847aSShri Abhyankar       PetscSparseDenseMinusDot(sum,tmp,v,vi,nz);
11249bd0847aSShri Abhyankar       tmp[i] = sum;
11259bd0847aSShri Abhyankar       v += nz; vi += nz;
11269bd0847aSShri Abhyankar     }
11279bd0847aSShri Abhyankar 
11289bd0847aSShri Abhyankar     /* backward solve the upper triangular */
11299bd0847aSShri Abhyankar     for (i=n-1; i>=0; i--){
11309bd0847aSShri Abhyankar       v   = aa + adiag[i+1]+1;
11319bd0847aSShri Abhyankar       vi  = aj + adiag[i+1]+1;
11329bd0847aSShri Abhyankar       nz  = adiag[i]-adiag[i+1]-1;
11339bd0847aSShri Abhyankar       sum = tmp[i];
11349bd0847aSShri Abhyankar       PetscSparseDenseMinusDot(sum,tmp,v,vi,nz);
11359bd0847aSShri Abhyankar       x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */
11369bd0847aSShri Abhyankar     }
11379bd0847aSShri Abhyankar 
11389bd0847aSShri Abhyankar     b += n;
11399bd0847aSShri Abhyankar     x += n;
11409bd0847aSShri Abhyankar   }
11419bd0847aSShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
11429bd0847aSShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
11439bd0847aSShri Abhyankar   ierr = MatRestoreArray(B,&b);CHKERRQ(ierr);
11449bd0847aSShri Abhyankar   ierr = MatRestoreArray(X,&x);CHKERRQ(ierr);
11459bd0847aSShri Abhyankar   ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr);
11469bd0847aSShri Abhyankar   PetscFunctionReturn(0);
11479bd0847aSShri Abhyankar }
11489bd0847aSShri Abhyankar 
11499bd0847aSShri Abhyankar #undef __FUNCT__
1150137fb511SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_InplaceWithPerm"
1151137fb511SHong Zhang PetscErrorCode MatSolve_SeqAIJ_InplaceWithPerm(Mat A,Vec bb,Vec xx)
1152137fb511SHong Zhang {
1153137fb511SHong Zhang   Mat_SeqAIJ      *a = (Mat_SeqAIJ*)A->data;
1154137fb511SHong Zhang   IS              iscol = a->col,isrow = a->row;
1155137fb511SHong Zhang   PetscErrorCode  ierr;
11565d0c19d7SBarry Smith   const PetscInt  *r,*c,*rout,*cout;
11575d0c19d7SBarry Smith   PetscInt        i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
11585d0c19d7SBarry Smith   PetscInt        nz,row;
1159dd6ea824SBarry Smith   PetscScalar     *x,*b,*tmp,*tmps,sum;
1160dd6ea824SBarry Smith   const MatScalar *aa = a->a,*v;
1161137fb511SHong Zhang 
1162137fb511SHong Zhang   PetscFunctionBegin;
1163137fb511SHong Zhang   if (!n) PetscFunctionReturn(0);
1164137fb511SHong Zhang 
1165137fb511SHong Zhang   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
1166137fb511SHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1167137fb511SHong Zhang   tmp  = a->solve_work;
1168137fb511SHong Zhang 
1169137fb511SHong Zhang   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
1170137fb511SHong Zhang   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
1171137fb511SHong Zhang 
1172137fb511SHong Zhang   /* forward solve the lower triangular */
1173137fb511SHong Zhang   tmp[0] = b[*r++];
1174137fb511SHong Zhang   tmps   = tmp;
1175137fb511SHong Zhang   for (row=1; row<n; row++) {
1176137fb511SHong Zhang     i   = rout[row]; /* permuted row */
1177137fb511SHong Zhang     v   = aa + ai[i] ;
1178137fb511SHong Zhang     vi  = aj + ai[i] ;
1179137fb511SHong Zhang     nz  = a->diag[i] - ai[i];
1180137fb511SHong Zhang     sum = b[*r++];
1181003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
1182137fb511SHong Zhang     tmp[row] = sum;
1183137fb511SHong Zhang   }
1184137fb511SHong Zhang 
1185137fb511SHong Zhang   /* backward solve the upper triangular */
1186137fb511SHong Zhang   for (row=n-1; row>=0; row--){
1187137fb511SHong Zhang     i   = rout[row]; /* permuted row */
1188137fb511SHong Zhang     v   = aa + a->diag[i] + 1;
1189137fb511SHong Zhang     vi  = aj + a->diag[i] + 1;
1190137fb511SHong Zhang     nz  = ai[i+1] - a->diag[i] - 1;
1191137fb511SHong Zhang     sum = tmp[row];
1192003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
1193137fb511SHong Zhang     x[*c--] = tmp[row] = sum*aa[a->diag[i]];
1194137fb511SHong Zhang   }
1195137fb511SHong Zhang 
1196137fb511SHong Zhang   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
1197137fb511SHong Zhang   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
1198137fb511SHong Zhang   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
1199137fb511SHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1200dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
1201137fb511SHong Zhang   PetscFunctionReturn(0);
1202137fb511SHong Zhang }
1203137fb511SHong Zhang 
1204930ae53cSBarry Smith /* ----------------------------------------------------------- */
1205c6db04a5SJed Brown #include <../src/mat/impls/aij/seq/ftn-kernels/fsolve.h>
12064a2ae208SSatish Balay #undef __FUNCT__
1207ad04f41aSHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering_inplace"
1208ad04f41aSHong Zhang PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering_inplace(Mat A,Vec bb,Vec xx)
1209930ae53cSBarry Smith {
1210930ae53cSBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
12116849ba73SBarry Smith   PetscErrorCode    ierr;
1212d0f46423SBarry Smith   PetscInt          n = A->rmap->n;
1213b377110cSBarry Smith   const PetscInt    *ai = a->i,*aj = a->j,*adiag = a->diag;
1214356650c2SBarry Smith   PetscScalar       *x;
1215356650c2SBarry Smith   const PetscScalar *b;
1216dd6ea824SBarry Smith   const MatScalar   *aa = a->a;
1217aa482453SBarry Smith #if !defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ)
1218356650c2SBarry Smith   PetscInt          adiag_i,i,nz,ai_i;
1219b377110cSBarry Smith   const PetscInt    *vi;
1220dd6ea824SBarry Smith   const MatScalar   *v;
1221dd6ea824SBarry Smith   PetscScalar       sum;
1222d85afc42SSatish Balay #endif
1223930ae53cSBarry Smith 
12243a40ed3dSBarry Smith   PetscFunctionBegin;
12253a40ed3dSBarry Smith   if (!n) PetscFunctionReturn(0);
1226930ae53cSBarry Smith 
12273649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
12281ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1229930ae53cSBarry Smith 
1230aa482453SBarry Smith #if defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ)
12311c4feecaSSatish Balay   fortransolveaij_(&n,x,ai,aj,adiag,aa,b);
12321c4feecaSSatish Balay #else
1233930ae53cSBarry Smith   /* forward solve the lower triangular */
1234930ae53cSBarry Smith   x[0] = b[0];
1235930ae53cSBarry Smith   for (i=1; i<n; i++) {
1236930ae53cSBarry Smith     ai_i = ai[i];
1237930ae53cSBarry Smith     v    = aa + ai_i;
1238930ae53cSBarry Smith     vi   = aj + ai_i;
1239930ae53cSBarry Smith     nz   = adiag[i] - ai_i;
1240930ae53cSBarry Smith     sum  = b[i];
1241003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
1242930ae53cSBarry Smith     x[i] = sum;
1243930ae53cSBarry Smith   }
1244930ae53cSBarry Smith 
1245930ae53cSBarry Smith   /* backward solve the upper triangular */
1246930ae53cSBarry Smith   for (i=n-1; i>=0; i--){
1247930ae53cSBarry Smith     adiag_i = adiag[i];
1248d708749eSBarry Smith     v       = aa + adiag_i + 1;
1249d708749eSBarry Smith     vi      = aj + adiag_i + 1;
1250930ae53cSBarry Smith     nz      = ai[i+1] - adiag_i - 1;
1251930ae53cSBarry Smith     sum     = x[i];
1252003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
1253930ae53cSBarry Smith     x[i]    = sum*aa[adiag_i];
1254930ae53cSBarry Smith   }
12551c4feecaSSatish Balay #endif
1256dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
12573649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
12581ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
12593a40ed3dSBarry Smith   PetscFunctionReturn(0);
1260930ae53cSBarry Smith }
1261930ae53cSBarry Smith 
12624a2ae208SSatish Balay #undef __FUNCT__
1263ad04f41aSHong Zhang #define __FUNCT__ "MatSolveAdd_SeqAIJ_inplace"
1264ad04f41aSHong Zhang PetscErrorCode MatSolveAdd_SeqAIJ_inplace(Mat A,Vec bb,Vec yy,Vec xx)
1265da3a660dSBarry Smith {
1266416022c9SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
1267416022c9SBarry Smith   IS                iscol = a->col,isrow = a->row;
12686849ba73SBarry Smith   PetscErrorCode    ierr;
126904fbf559SBarry Smith   PetscInt          i, n = A->rmap->n,j;
12705d0c19d7SBarry Smith   PetscInt          nz;
127104fbf559SBarry Smith   const PetscInt    *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j;
127204fbf559SBarry Smith   PetscScalar       *x,*tmp,sum;
127304fbf559SBarry Smith   const PetscScalar *b;
1274dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
1275da3a660dSBarry Smith 
12763a40ed3dSBarry Smith   PetscFunctionBegin;
127778b31e54SBarry Smith   if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);}
1278da3a660dSBarry Smith 
12793649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
12801ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1281416022c9SBarry Smith   tmp  = a->solve_work;
1282da3a660dSBarry Smith 
12833b2fbd54SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
12843b2fbd54SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
1285da3a660dSBarry Smith 
1286da3a660dSBarry Smith   /* forward solve the lower triangular */
1287da3a660dSBarry Smith   tmp[0] = b[*r++];
1288da3a660dSBarry Smith   for (i=1; i<n; i++) {
1289010a20caSHong Zhang     v   = aa + ai[i] ;
1290010a20caSHong Zhang     vi  = aj + ai[i] ;
1291416022c9SBarry Smith     nz  = a->diag[i] - ai[i];
1292da3a660dSBarry Smith     sum = b[*r++];
129304fbf559SBarry Smith     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
1294da3a660dSBarry Smith     tmp[i] = sum;
1295da3a660dSBarry Smith   }
1296da3a660dSBarry Smith 
1297da3a660dSBarry Smith   /* backward solve the upper triangular */
1298da3a660dSBarry Smith   for (i=n-1; i>=0; i--){
1299010a20caSHong Zhang     v   = aa + a->diag[i] + 1;
1300010a20caSHong Zhang     vi  = aj + a->diag[i] + 1;
1301416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
1302da3a660dSBarry Smith     sum = tmp[i];
130304fbf559SBarry Smith     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
1304010a20caSHong Zhang     tmp[i] = sum*aa[a->diag[i]];
1305da3a660dSBarry Smith     x[*c--] += tmp[i];
1306da3a660dSBarry Smith   }
1307da3a660dSBarry Smith 
13083b2fbd54SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
13093b2fbd54SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
13103649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
13111ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1312dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
1313898183ecSLois Curfman McInnes 
13143a40ed3dSBarry Smith   PetscFunctionReturn(0);
1315da3a660dSBarry Smith }
131604fbf559SBarry Smith 
13174a2ae208SSatish Balay #undef __FUNCT__
131835233d99SShri Abhyankar #define __FUNCT__ "MatSolveAdd_SeqAIJ"
131935233d99SShri Abhyankar PetscErrorCode MatSolveAdd_SeqAIJ(Mat A,Vec bb,Vec yy,Vec xx)
13203c0119dfSShri Abhyankar {
13213c0119dfSShri Abhyankar   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
13223c0119dfSShri Abhyankar   IS                iscol = a->col,isrow = a->row;
13233c0119dfSShri Abhyankar   PetscErrorCode    ierr;
13243c0119dfSShri Abhyankar   PetscInt          i, n = A->rmap->n,j;
13253c0119dfSShri Abhyankar   PetscInt          nz;
13263c0119dfSShri Abhyankar   const PetscInt    *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j,*adiag = a->diag;
13273c0119dfSShri Abhyankar   PetscScalar       *x,*tmp,sum;
13283c0119dfSShri Abhyankar   const PetscScalar *b;
13293c0119dfSShri Abhyankar   const MatScalar   *aa = a->a,*v;
13303c0119dfSShri Abhyankar 
13313c0119dfSShri Abhyankar   PetscFunctionBegin;
13323c0119dfSShri Abhyankar   if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);}
13333c0119dfSShri Abhyankar 
13343649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
13353c0119dfSShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
13363c0119dfSShri Abhyankar   tmp  = a->solve_work;
13373c0119dfSShri Abhyankar 
13383c0119dfSShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
13393c0119dfSShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
13403c0119dfSShri Abhyankar 
13413c0119dfSShri Abhyankar   /* forward solve the lower triangular */
13423c0119dfSShri Abhyankar   tmp[0] = b[r[0]];
13433c0119dfSShri Abhyankar   v      = aa;
13443c0119dfSShri Abhyankar   vi     = aj;
13453c0119dfSShri Abhyankar   for (i=1; i<n; i++) {
13463c0119dfSShri Abhyankar     nz  = ai[i+1] - ai[i];
13473c0119dfSShri Abhyankar     sum = b[r[i]];
13483c0119dfSShri Abhyankar     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
13493c0119dfSShri Abhyankar     tmp[i] = sum;
13503c0119dfSShri Abhyankar     v += nz; vi += nz;
13513c0119dfSShri Abhyankar   }
13523c0119dfSShri Abhyankar 
13533c0119dfSShri Abhyankar   /* backward solve the upper triangular */
13543c0119dfSShri Abhyankar   v  = aa + adiag[n-1];
13553c0119dfSShri Abhyankar   vi = aj + adiag[n-1];
13563c0119dfSShri Abhyankar   for (i=n-1; i>=0; i--){
13573c0119dfSShri Abhyankar     nz  = adiag[i] - adiag[i+1] - 1;
13583c0119dfSShri Abhyankar     sum = tmp[i];
13593c0119dfSShri Abhyankar     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
13603c0119dfSShri Abhyankar     tmp[i] = sum*v[nz];
13613c0119dfSShri Abhyankar     x[c[i]] += tmp[i];
13623c0119dfSShri Abhyankar     v += nz+1; vi += nz+1;
13633c0119dfSShri Abhyankar   }
13643c0119dfSShri Abhyankar 
13653c0119dfSShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
13663c0119dfSShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
13673649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
13683c0119dfSShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
13693c0119dfSShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
13703c0119dfSShri Abhyankar 
13713c0119dfSShri Abhyankar   PetscFunctionReturn(0);
13723c0119dfSShri Abhyankar }
13733c0119dfSShri Abhyankar 
13743c0119dfSShri Abhyankar #undef __FUNCT__
1375ad04f41aSHong Zhang #define __FUNCT__ "MatSolveTranspose_SeqAIJ_inplace"
1376ad04f41aSHong Zhang PetscErrorCode MatSolveTranspose_SeqAIJ_inplace(Mat A,Vec bb,Vec xx)
1377da3a660dSBarry Smith {
1378416022c9SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
13792235e667SBarry Smith   IS                iscol = a->col,isrow = a->row;
13806849ba73SBarry Smith   PetscErrorCode    ierr;
138104fbf559SBarry Smith   const PetscInt    *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi;
138204fbf559SBarry Smith   PetscInt          i,n = A->rmap->n,j;
138304fbf559SBarry Smith   PetscInt          nz;
138404fbf559SBarry Smith   PetscScalar       *x,*tmp,s1;
1385dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
138604fbf559SBarry Smith   const PetscScalar *b;
1387da3a660dSBarry Smith 
13883a40ed3dSBarry Smith   PetscFunctionBegin;
13893649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
13901ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1391416022c9SBarry Smith   tmp  = a->solve_work;
1392da3a660dSBarry Smith 
13932235e667SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
13942235e667SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
1395da3a660dSBarry Smith 
1396da3a660dSBarry Smith   /* copy the b into temp work space according to permutation */
13972235e667SBarry Smith   for (i=0; i<n; i++) tmp[i] = b[c[i]];
1398da3a660dSBarry Smith 
1399da3a660dSBarry Smith   /* forward solve the U^T */
1400da3a660dSBarry Smith   for (i=0; i<n; i++) {
1401010a20caSHong Zhang     v   = aa + diag[i] ;
1402010a20caSHong Zhang     vi  = aj + diag[i] + 1;
1403f1af5d2fSBarry Smith     nz  = ai[i+1] - diag[i] - 1;
1404c38d4ed2SBarry Smith     s1  = tmp[i];
1405ef66eb69SBarry Smith     s1 *= (*v++);  /* multiply by inverse of diagonal entry */
140604fbf559SBarry Smith     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1407c38d4ed2SBarry Smith     tmp[i] = s1;
1408da3a660dSBarry Smith   }
1409da3a660dSBarry Smith 
1410da3a660dSBarry Smith   /* backward solve the L^T */
1411da3a660dSBarry Smith   for (i=n-1; i>=0; i--){
1412010a20caSHong Zhang     v   = aa + diag[i] - 1 ;
1413010a20caSHong Zhang     vi  = aj + diag[i] - 1 ;
1414f1af5d2fSBarry Smith     nz  = diag[i] - ai[i];
1415f1af5d2fSBarry Smith     s1  = tmp[i];
141604fbf559SBarry Smith     for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j];
1417da3a660dSBarry Smith   }
1418da3a660dSBarry Smith 
1419da3a660dSBarry Smith   /* copy tmp into x according to permutation */
1420591294e4SBarry Smith   for (i=0; i<n; i++) x[r[i]] = tmp[i];
1421da3a660dSBarry Smith 
14222235e667SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
14232235e667SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
14243649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
14251ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1426da3a660dSBarry Smith 
1427dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr);
14283a40ed3dSBarry Smith   PetscFunctionReturn(0);
1429da3a660dSBarry Smith }
1430da3a660dSBarry Smith 
1431d1fa9404SShri Abhyankar #undef __FUNCT__
143235233d99SShri Abhyankar #define __FUNCT__ "MatSolveTranspose_SeqAIJ"
143335233d99SShri Abhyankar PetscErrorCode MatSolveTranspose_SeqAIJ(Mat A,Vec bb,Vec xx)
1434d1fa9404SShri Abhyankar {
1435d1fa9404SShri Abhyankar   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
1436d1fa9404SShri Abhyankar   IS                iscol = a->col,isrow = a->row;
1437d1fa9404SShri Abhyankar   PetscErrorCode    ierr;
1438d1fa9404SShri Abhyankar   const PetscInt    *rout,*cout,*r,*c,*adiag = a->diag,*ai = a->i,*aj = a->j,*vi;
1439d1fa9404SShri Abhyankar   PetscInt          i,n = A->rmap->n,j;
1440d1fa9404SShri Abhyankar   PetscInt          nz;
1441d1fa9404SShri Abhyankar   PetscScalar       *x,*tmp,s1;
1442d1fa9404SShri Abhyankar   const MatScalar   *aa = a->a,*v;
1443d1fa9404SShri Abhyankar   const PetscScalar *b;
1444d1fa9404SShri Abhyankar 
1445d1fa9404SShri Abhyankar   PetscFunctionBegin;
14463649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
1447d1fa9404SShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1448d1fa9404SShri Abhyankar   tmp  = a->solve_work;
1449d1fa9404SShri Abhyankar 
1450d1fa9404SShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
1451d1fa9404SShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
1452d1fa9404SShri Abhyankar 
1453d1fa9404SShri Abhyankar   /* copy the b into temp work space according to permutation */
1454d1fa9404SShri Abhyankar   for (i=0; i<n; i++) tmp[i] = b[c[i]];
1455d1fa9404SShri Abhyankar 
1456d1fa9404SShri Abhyankar   /* forward solve the U^T */
1457d1fa9404SShri Abhyankar   for (i=0; i<n; i++) {
1458d1fa9404SShri Abhyankar     v   = aa + adiag[i+1] + 1;
1459d1fa9404SShri Abhyankar     vi  = aj + adiag[i+1] + 1;
1460d1fa9404SShri Abhyankar     nz  = adiag[i] - adiag[i+1] - 1;
1461d1fa9404SShri Abhyankar     s1  = tmp[i];
1462d1fa9404SShri Abhyankar     s1 *= v[nz];  /* multiply by inverse of diagonal entry */
1463d1fa9404SShri Abhyankar     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1464d1fa9404SShri Abhyankar     tmp[i] = s1;
1465d1fa9404SShri Abhyankar   }
1466d1fa9404SShri Abhyankar 
1467d1fa9404SShri Abhyankar   /* backward solve the L^T */
1468d1fa9404SShri Abhyankar   for (i=n-1; i>=0; i--){
1469d1fa9404SShri Abhyankar     v   = aa + ai[i];
1470d1fa9404SShri Abhyankar     vi  = aj + ai[i];
1471d1fa9404SShri Abhyankar     nz  = ai[i+1] - ai[i];
1472d1fa9404SShri Abhyankar     s1  = tmp[i];
1473d1fa9404SShri Abhyankar     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1474d1fa9404SShri Abhyankar   }
1475d1fa9404SShri Abhyankar 
1476d1fa9404SShri Abhyankar   /* copy tmp into x according to permutation */
1477d1fa9404SShri Abhyankar   for (i=0; i<n; i++) x[r[i]] = tmp[i];
1478d1fa9404SShri Abhyankar 
1479d1fa9404SShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
1480d1fa9404SShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
14813649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
1482d1fa9404SShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1483d1fa9404SShri Abhyankar 
1484d1fa9404SShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr);
1485d1fa9404SShri Abhyankar   PetscFunctionReturn(0);
1486d1fa9404SShri Abhyankar }
1487d1fa9404SShri Abhyankar 
14884a2ae208SSatish Balay #undef __FUNCT__
1489ad04f41aSHong Zhang #define __FUNCT__ "MatSolveTransposeAdd_SeqAIJ_inplace"
1490ad04f41aSHong Zhang PetscErrorCode MatSolveTransposeAdd_SeqAIJ_inplace(Mat A,Vec bb,Vec zz,Vec xx)
1491da3a660dSBarry Smith {
1492416022c9SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
14932235e667SBarry Smith   IS                iscol = a->col,isrow = a->row;
14946849ba73SBarry Smith   PetscErrorCode    ierr;
149504fbf559SBarry Smith   const PetscInt    *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi;
149604fbf559SBarry Smith   PetscInt          i,n = A->rmap->n,j;
149704fbf559SBarry Smith   PetscInt          nz;
149804fbf559SBarry Smith   PetscScalar       *x,*tmp,s1;
1499dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
150004fbf559SBarry Smith   const PetscScalar *b;
15016abc6512SBarry Smith 
15023a40ed3dSBarry Smith   PetscFunctionBegin;
150323bc6035SMatthew Knepley   if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);}
15043649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
15051ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1506416022c9SBarry Smith   tmp  = a->solve_work;
15076abc6512SBarry Smith 
15082235e667SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
15092235e667SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
15106abc6512SBarry Smith 
15116abc6512SBarry Smith   /* copy the b into temp work space according to permutation */
15122235e667SBarry Smith   for (i=0; i<n; i++) tmp[i] = b[c[i]];
15136abc6512SBarry Smith 
15146abc6512SBarry Smith   /* forward solve the U^T */
15156abc6512SBarry Smith   for (i=0; i<n; i++) {
1516010a20caSHong Zhang     v   = aa + diag[i] ;
1517010a20caSHong Zhang     vi  = aj + diag[i] + 1;
1518f1af5d2fSBarry Smith     nz  = ai[i+1] - diag[i] - 1;
151904fbf559SBarry Smith     s1  = tmp[i];
152004fbf559SBarry Smith     s1 *= (*v++);  /* multiply by inverse of diagonal entry */
152104fbf559SBarry Smith     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
152204fbf559SBarry Smith     tmp[i] = s1;
15236abc6512SBarry Smith   }
15246abc6512SBarry Smith 
15256abc6512SBarry Smith   /* backward solve the L^T */
15266abc6512SBarry Smith   for (i=n-1; i>=0; i--){
1527010a20caSHong Zhang     v   = aa + diag[i] - 1 ;
1528010a20caSHong Zhang     vi  = aj + diag[i] - 1 ;
1529f1af5d2fSBarry Smith     nz  = diag[i] - ai[i];
153004fbf559SBarry Smith     s1  = tmp[i];
153104fbf559SBarry Smith     for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j];
15326abc6512SBarry Smith   }
15336abc6512SBarry Smith 
15346abc6512SBarry Smith   /* copy tmp into x according to permutation */
15356abc6512SBarry Smith   for (i=0; i<n; i++) x[r[i]] += tmp[i];
15366abc6512SBarry Smith 
15372235e667SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
15382235e667SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
15393649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
15401ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
15416abc6512SBarry Smith 
154204fbf559SBarry Smith   ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr);
15433a40ed3dSBarry Smith   PetscFunctionReturn(0);
1544da3a660dSBarry Smith }
154504fbf559SBarry Smith 
1546533e6011SShri Abhyankar #undef __FUNCT__
154735233d99SShri Abhyankar #define __FUNCT__ "MatSolveTransposeAdd_SeqAIJ"
154835233d99SShri Abhyankar PetscErrorCode MatSolveTransposeAdd_SeqAIJ(Mat A,Vec bb,Vec zz,Vec xx)
1549533e6011SShri Abhyankar {
1550533e6011SShri Abhyankar   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
1551533e6011SShri Abhyankar   IS                iscol = a->col,isrow = a->row;
1552533e6011SShri Abhyankar   PetscErrorCode    ierr;
1553533e6011SShri Abhyankar   const PetscInt    *rout,*cout,*r,*c,*adiag = a->diag,*ai = a->i,*aj = a->j,*vi;
1554533e6011SShri Abhyankar   PetscInt          i,n = A->rmap->n,j;
1555533e6011SShri Abhyankar   PetscInt          nz;
1556533e6011SShri Abhyankar   PetscScalar       *x,*tmp,s1;
1557533e6011SShri Abhyankar   const MatScalar   *aa = a->a,*v;
1558533e6011SShri Abhyankar   const PetscScalar *b;
1559533e6011SShri Abhyankar 
1560533e6011SShri Abhyankar   PetscFunctionBegin;
1561533e6011SShri Abhyankar   if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);}
15623649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
1563533e6011SShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1564533e6011SShri Abhyankar   tmp  = a->solve_work;
1565533e6011SShri Abhyankar 
1566533e6011SShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
1567533e6011SShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
1568533e6011SShri Abhyankar 
1569533e6011SShri Abhyankar   /* copy the b into temp work space according to permutation */
1570533e6011SShri Abhyankar   for (i=0; i<n; i++) tmp[i] = b[c[i]];
1571533e6011SShri Abhyankar 
1572533e6011SShri Abhyankar   /* forward solve the U^T */
1573533e6011SShri Abhyankar   for (i=0; i<n; i++) {
1574533e6011SShri Abhyankar     v   = aa + adiag[i+1] + 1;
1575533e6011SShri Abhyankar     vi  = aj + adiag[i+1] + 1;
1576533e6011SShri Abhyankar     nz  = adiag[i] - adiag[i+1] - 1;
1577533e6011SShri Abhyankar     s1  = tmp[i];
1578533e6011SShri Abhyankar     s1 *= v[nz];  /* multiply by inverse of diagonal entry */
1579533e6011SShri Abhyankar     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1580533e6011SShri Abhyankar     tmp[i] = s1;
1581533e6011SShri Abhyankar   }
1582533e6011SShri Abhyankar 
1583533e6011SShri Abhyankar 
1584533e6011SShri Abhyankar   /* backward solve the L^T */
1585533e6011SShri Abhyankar   for (i=n-1; i>=0; i--){
1586533e6011SShri Abhyankar     v   = aa + ai[i] ;
1587533e6011SShri Abhyankar     vi  = aj + ai[i];
1588533e6011SShri Abhyankar     nz  = ai[i+1] - ai[i];
1589533e6011SShri Abhyankar     s1  = tmp[i];
1590c5e3b2a3SShri Abhyankar     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1591533e6011SShri Abhyankar   }
1592533e6011SShri Abhyankar 
1593533e6011SShri Abhyankar   /* copy tmp into x according to permutation */
1594533e6011SShri Abhyankar   for (i=0; i<n; i++) x[r[i]] += tmp[i];
1595533e6011SShri Abhyankar 
1596533e6011SShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
1597533e6011SShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
15983649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
1599533e6011SShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1600533e6011SShri Abhyankar 
1601533e6011SShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr);
1602533e6011SShri Abhyankar   PetscFunctionReturn(0);
1603533e6011SShri Abhyankar }
1604533e6011SShri Abhyankar 
16059e25ed09SBarry Smith /* ----------------------------------------------------------------*/
1606b47f5a65SHong Zhang 
160709573ac7SBarry Smith extern PetscErrorCode MatDuplicateNoCreate_SeqAIJ(Mat,Mat,MatDuplicateOption,PetscBool );
160808480c60SBarry Smith 
16098fc3a347SHong Zhang /*
16108a76ed4fSHong Zhang    ilu() under revised new data structure.
1611813a1f2bSShri Abhyankar    Factored arrays bj and ba are stored as
1612813a1f2bSShri Abhyankar      L(0,:), L(1,:), ...,L(n-1,:),  U(n-1,:),...,U(i,:),U(i-1,:),...,U(0,:)
1613813a1f2bSShri Abhyankar 
1614813a1f2bSShri Abhyankar    bi=fact->i is an array of size n+1, in which
1615813a1f2bSShri Abhyankar    bi+
1616baabb860SHong Zhang      bi[i]:  points to 1st entry of L(i,:),i=0,...,n-1
1617baabb860SHong Zhang      bi[n]:  points to L(n-1,n-1)+1
1618baabb860SHong Zhang 
1619813a1f2bSShri Abhyankar   bdiag=fact->diag is an array of size n+1,in which
1620baabb860SHong Zhang      bdiag[i]: points to diagonal of U(i,:), i=0,...,n-1
1621a24f213cSHong Zhang      bdiag[n]: points to entry of U(n-1,0)-1
1622813a1f2bSShri Abhyankar 
1623813a1f2bSShri Abhyankar    U(i,:) contains bdiag[i] as its last entry, i.e.,
1624813a1f2bSShri Abhyankar     U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i])
1625813a1f2bSShri Abhyankar */
1626813a1f2bSShri Abhyankar #undef __FUNCT__
162735233d99SShri Abhyankar #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_ilu0"
162835233d99SShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ_ilu0(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
1629813a1f2bSShri Abhyankar {
1630813a1f2bSShri Abhyankar 
1631813a1f2bSShri Abhyankar   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
1632813a1f2bSShri Abhyankar   PetscErrorCode     ierr;
1633bbd65245SShri Abhyankar   const PetscInt     n=A->rmap->n,*ai=a->i,*aj,*adiag=a->diag;
1634bbd65245SShri Abhyankar   PetscInt           i,j,k=0,nz,*bi,*bj,*bdiag;
1635ace3abfcSBarry Smith   PetscBool          missing;
16361df811f5SHong Zhang   IS                 isicol;
1637813a1f2bSShri Abhyankar 
1638813a1f2bSShri Abhyankar   PetscFunctionBegin;
1639e32f2f54SBarry 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);
16401df811f5SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr);
1641e32f2f54SBarry Smith   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i);
16421df811f5SHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
16431df811f5SHong Zhang 
1644813a1f2bSShri Abhyankar   ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_FALSE);CHKERRQ(ierr);
1645813a1f2bSShri Abhyankar   b    = (Mat_SeqAIJ*)(fact)->data;
1646813a1f2bSShri Abhyankar 
1647813a1f2bSShri Abhyankar   /* allocate matrix arrays for new data structure */
1648813a1f2bSShri Abhyankar   ierr = PetscMalloc3(ai[n]+1,PetscScalar,&b->a,ai[n]+1,PetscInt,&b->j,n+1,PetscInt,&b->i);CHKERRQ(ierr);
1649813a1f2bSShri Abhyankar   ierr = PetscLogObjectMemory(fact,ai[n]*(sizeof(PetscScalar)+sizeof(PetscInt))+(n+1)*sizeof(PetscInt));CHKERRQ(ierr);
1650813a1f2bSShri Abhyankar   b->singlemalloc = PETSC_TRUE;
1651813a1f2bSShri Abhyankar   if (!b->diag){
1652813a1f2bSShri Abhyankar     ierr = PetscMalloc((n+1)*sizeof(PetscInt),&b->diag);CHKERRQ(ierr);
1653813a1f2bSShri Abhyankar     ierr = PetscLogObjectMemory(fact,(n+1)*sizeof(PetscInt));CHKERRQ(ierr);
1654813a1f2bSShri Abhyankar   }
1655813a1f2bSShri Abhyankar   bdiag = b->diag;
1656813a1f2bSShri Abhyankar 
1657813a1f2bSShri Abhyankar   if (n > 0) {
1658813a1f2bSShri Abhyankar     ierr = PetscMemzero(b->a,(ai[n])*sizeof(MatScalar));CHKERRQ(ierr);
1659813a1f2bSShri Abhyankar   }
1660813a1f2bSShri Abhyankar 
1661813a1f2bSShri Abhyankar   /* set bi and bj with new data structure */
1662813a1f2bSShri Abhyankar   bi = b->i;
1663813a1f2bSShri Abhyankar   bj = b->j;
1664813a1f2bSShri Abhyankar 
1665813a1f2bSShri Abhyankar   /* L part */
1666e60cf9a0SBarry Smith   bi[0] = 0;
1667813a1f2bSShri Abhyankar   for (i=0; i<n; i++){
1668813a1f2bSShri Abhyankar     nz = adiag[i] - ai[i];
1669813a1f2bSShri Abhyankar     bi[i+1] = bi[i] + nz;
1670813a1f2bSShri Abhyankar     aj = a->j + ai[i];
1671813a1f2bSShri Abhyankar     for (j=0; j<nz; j++){
1672bbd65245SShri Abhyankar       /*   *bj = aj[j]; bj++; */
1673bbd65245SShri Abhyankar       bj[k++] = aj[j];
1674813a1f2bSShri Abhyankar     }
1675813a1f2bSShri Abhyankar   }
1676813a1f2bSShri Abhyankar 
1677813a1f2bSShri Abhyankar   /* U part */
167862fbd6afSShri Abhyankar   bdiag[n] = bi[n]-1;
1679813a1f2bSShri Abhyankar   for (i=n-1; i>=0; i--){
1680813a1f2bSShri Abhyankar     nz = ai[i+1] - adiag[i] - 1;
1681813a1f2bSShri Abhyankar     aj = a->j + adiag[i] + 1;
1682813a1f2bSShri Abhyankar     for (j=0; j<nz; j++){
1683bbd65245SShri Abhyankar       /*      *bj = aj[j]; bj++; */
1684bbd65245SShri Abhyankar       bj[k++] = aj[j];
1685813a1f2bSShri Abhyankar     }
1686813a1f2bSShri Abhyankar     /* diag[i] */
1687bbd65245SShri Abhyankar     /*    *bj = i; bj++; */
1688bbd65245SShri Abhyankar     bj[k++] = i;
1689a24f213cSHong Zhang     bdiag[i] = bdiag[i+1] + nz + 1;
1690813a1f2bSShri Abhyankar   }
16911df811f5SHong Zhang 
1692d5f3da31SBarry Smith   fact->factortype             = MAT_FACTOR_ILU;
16931df811f5SHong Zhang   fact->info.factor_mallocs    = 0;
16941df811f5SHong Zhang   fact->info.fill_ratio_given  = info->fill;
16951df811f5SHong Zhang   fact->info.fill_ratio_needed = 1.0;
169635233d99SShri Abhyankar   fact->ops->lufactornumeric   = MatLUFactorNumeric_SeqAIJ;
16971df811f5SHong Zhang 
16981df811f5SHong Zhang   b       = (Mat_SeqAIJ*)(fact)->data;
16991df811f5SHong Zhang   b->row  = isrow;
17001df811f5SHong Zhang   b->col  = iscol;
17011df811f5SHong Zhang   b->icol = isicol;
17021df811f5SHong Zhang   ierr    = PetscMalloc((fact->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
17031df811f5SHong Zhang   ierr    = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
17041df811f5SHong Zhang   ierr    = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
1705813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
1706813a1f2bSShri Abhyankar }
1707813a1f2bSShri Abhyankar 
1708813a1f2bSShri Abhyankar #undef __FUNCT__
170935233d99SShri Abhyankar #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ"
171035233d99SShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
1711813a1f2bSShri Abhyankar {
1712813a1f2bSShri Abhyankar   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
1713813a1f2bSShri Abhyankar   IS                 isicol;
1714813a1f2bSShri Abhyankar   PetscErrorCode     ierr;
1715813a1f2bSShri Abhyankar   const PetscInt     *r,*ic;
17161df811f5SHong Zhang   PetscInt           n=A->rmap->n,*ai=a->i,*aj=a->j;
1717813a1f2bSShri Abhyankar   PetscInt           *bi,*cols,nnz,*cols_lvl;
1718813a1f2bSShri Abhyankar   PetscInt           *bdiag,prow,fm,nzbd,reallocs=0,dcount=0;
1719813a1f2bSShri Abhyankar   PetscInt           i,levels,diagonal_fill;
1720ace3abfcSBarry Smith   PetscBool          col_identity,row_identity;
1721813a1f2bSShri Abhyankar   PetscReal          f;
1722813a1f2bSShri Abhyankar   PetscInt           nlnk,*lnk,*lnk_lvl=PETSC_NULL;
1723813a1f2bSShri Abhyankar   PetscBT            lnkbt;
1724813a1f2bSShri Abhyankar   PetscInt           nzi,*bj,**bj_ptr,**bjlvl_ptr;
1725813a1f2bSShri Abhyankar   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
1726813a1f2bSShri Abhyankar   PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL;
1727813a1f2bSShri Abhyankar 
1728813a1f2bSShri Abhyankar   PetscFunctionBegin;
17293898ab1dSSatish Balay   /* Uncomment the old data struct part only while testing new data structure for MatSolve() */
1730124a3351SHong Zhang   /*
1731ace3abfcSBarry Smith   PetscBool          olddatastruct=PETSC_FALSE;
1732acfcf0e5SJed Brown   ierr = PetscOptionsGetBool(PETSC_NULL,"-ilu_old",&olddatastruct,PETSC_NULL);CHKERRQ(ierr);
173335233d99SShri Abhyankar   if(olddatastruct){
1734ad04f41aSHong Zhang     ierr = MatILUFactorSymbolic_SeqAIJ_inplace(fact,A,isrow,iscol,info);CHKERRQ(ierr);
1735ad04f41aSHong Zhang     PetscFunctionReturn(0);
1736ad04f41aSHong Zhang   }
1737124a3351SHong Zhang   */
1738ad04f41aSHong Zhang 
1739813a1f2bSShri Abhyankar   levels = (PetscInt)info->levels;
1740813a1f2bSShri Abhyankar   ierr   = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
1741813a1f2bSShri Abhyankar   ierr   = ISIdentity(iscol,&col_identity);CHKERRQ(ierr);
1742813a1f2bSShri Abhyankar 
1743813a1f2bSShri Abhyankar   if (!levels && row_identity && col_identity) {
1744813a1f2bSShri Abhyankar     /* special case: ilu(0) with natural ordering */
174535233d99SShri Abhyankar     ierr = MatILUFactorSymbolic_SeqAIJ_ilu0(fact,A,isrow,iscol,info);CHKERRQ(ierr);
1746124a3351SHong Zhang     if (a->inode.size) {
1747124a3351SHong Zhang       fact->ops->lufactornumeric  = MatLUFactorNumeric_SeqAIJ_Inode;
1748124a3351SHong Zhang     }
1749813a1f2bSShri Abhyankar     PetscFunctionReturn(0);
1750813a1f2bSShri Abhyankar   }
1751813a1f2bSShri Abhyankar 
1752e32f2f54SBarry 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);
17531df811f5SHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
1754813a1f2bSShri Abhyankar   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
1755813a1f2bSShri Abhyankar   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
1756813a1f2bSShri Abhyankar 
17571bfa06eaSJed Brown   /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */
17581bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
17591bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
1760e60cf9a0SBarry Smith   bi[0] = bdiag[0] = 0;
1761813a1f2bSShri Abhyankar 
17620e83c824SBarry Smith   ierr = PetscMalloc2(n,PetscInt*,&bj_ptr,n,PetscInt*,&bjlvl_ptr);CHKERRQ(ierr);
1763813a1f2bSShri Abhyankar 
1764813a1f2bSShri Abhyankar   /* create a linked list for storing column indices of the active row */
1765813a1f2bSShri Abhyankar   nlnk = n + 1;
1766813a1f2bSShri Abhyankar   ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
1767813a1f2bSShri Abhyankar 
1768813a1f2bSShri Abhyankar   /* initial FreeSpace size is f*(ai[n]+1) */
17691df811f5SHong Zhang   f             = info->fill;
17701df811f5SHong Zhang   diagonal_fill = (PetscInt)info->diagonal_fill;
1771813a1f2bSShri Abhyankar   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr);
1772813a1f2bSShri Abhyankar   current_space = free_space;
1773813a1f2bSShri Abhyankar   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr);
1774813a1f2bSShri Abhyankar   current_space_lvl = free_space_lvl;
1775813a1f2bSShri Abhyankar 
1776813a1f2bSShri Abhyankar   for (i=0; i<n; i++) {
1777813a1f2bSShri Abhyankar     nzi = 0;
1778813a1f2bSShri Abhyankar     /* copy current row into linked list */
1779813a1f2bSShri Abhyankar     nnz  = ai[r[i]+1] - ai[r[i]];
1780e32f2f54SBarry 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);
1781813a1f2bSShri Abhyankar     cols = aj + ai[r[i]];
1782813a1f2bSShri Abhyankar     lnk[i] = -1; /* marker to indicate if diagonal exists */
1783813a1f2bSShri Abhyankar     ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
1784813a1f2bSShri Abhyankar     nzi += nlnk;
1785813a1f2bSShri Abhyankar 
1786813a1f2bSShri Abhyankar     /* make sure diagonal entry is included */
1787813a1f2bSShri Abhyankar     if (diagonal_fill && lnk[i] == -1) {
1788813a1f2bSShri Abhyankar       fm = n;
1789813a1f2bSShri Abhyankar       while (lnk[fm] < i) fm = lnk[fm];
1790813a1f2bSShri Abhyankar       lnk[i]     = lnk[fm]; /* insert diagonal into linked list */
1791813a1f2bSShri Abhyankar       lnk[fm]    = i;
1792813a1f2bSShri Abhyankar       lnk_lvl[i] = 0;
1793813a1f2bSShri Abhyankar       nzi++; dcount++;
1794813a1f2bSShri Abhyankar     }
1795813a1f2bSShri Abhyankar 
1796813a1f2bSShri Abhyankar     /* add pivot rows into the active row */
1797813a1f2bSShri Abhyankar     nzbd = 0;
1798813a1f2bSShri Abhyankar     prow = lnk[n];
1799813a1f2bSShri Abhyankar     while (prow < i) {
1800813a1f2bSShri Abhyankar       nnz      = bdiag[prow];
1801813a1f2bSShri Abhyankar       cols     = bj_ptr[prow] + nnz + 1;
1802813a1f2bSShri Abhyankar       cols_lvl = bjlvl_ptr[prow] + nnz + 1;
1803813a1f2bSShri Abhyankar       nnz      = bi[prow+1] - bi[prow] - nnz - 1;
1804813a1f2bSShri Abhyankar       ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr);
1805813a1f2bSShri Abhyankar       nzi += nlnk;
1806813a1f2bSShri Abhyankar       prow = lnk[prow];
1807813a1f2bSShri Abhyankar       nzbd++;
1808813a1f2bSShri Abhyankar     }
1809813a1f2bSShri Abhyankar     bdiag[i] = nzbd;
1810813a1f2bSShri Abhyankar     bi[i+1]  = bi[i] + nzi;
1811813a1f2bSShri Abhyankar 
1812813a1f2bSShri Abhyankar     /* if free space is not available, make more free space */
1813813a1f2bSShri Abhyankar     if (current_space->local_remaining<nzi) {
1814813a1f2bSShri Abhyankar       nnz = 2*nzi*(n - i); /* estimated and max additional space needed */
1815813a1f2bSShri Abhyankar       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
1816813a1f2bSShri Abhyankar       ierr = PetscFreeSpaceGet(nnz,&current_space_lvl);CHKERRQ(ierr);
1817813a1f2bSShri Abhyankar       reallocs++;
1818813a1f2bSShri Abhyankar     }
1819813a1f2bSShri Abhyankar 
1820813a1f2bSShri Abhyankar     /* copy data into free_space and free_space_lvl, then initialize lnk */
1821813a1f2bSShri Abhyankar     ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
1822813a1f2bSShri Abhyankar     bj_ptr[i]    = current_space->array;
1823813a1f2bSShri Abhyankar     bjlvl_ptr[i] = current_space_lvl->array;
1824813a1f2bSShri Abhyankar 
1825813a1f2bSShri Abhyankar     /* make sure the active row i has diagonal entry */
182665e19b50SBarry 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);
1827813a1f2bSShri Abhyankar 
1828813a1f2bSShri Abhyankar     current_space->array           += nzi;
1829813a1f2bSShri Abhyankar     current_space->local_used      += nzi;
1830813a1f2bSShri Abhyankar     current_space->local_remaining -= nzi;
1831813a1f2bSShri Abhyankar     current_space_lvl->array           += nzi;
1832813a1f2bSShri Abhyankar     current_space_lvl->local_used      += nzi;
1833813a1f2bSShri Abhyankar     current_space_lvl->local_remaining -= nzi;
1834813a1f2bSShri Abhyankar   }
1835813a1f2bSShri Abhyankar 
1836813a1f2bSShri Abhyankar   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
1837813a1f2bSShri Abhyankar   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
1838813a1f2bSShri Abhyankar 
1839813a1f2bSShri Abhyankar   /* copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */
18409263d837SHong Zhang   ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
18412ce24eb6SHong Zhang   ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr);
1842813a1f2bSShri Abhyankar 
1843813a1f2bSShri Abhyankar   ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
1844813a1f2bSShri Abhyankar   ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
18450e83c824SBarry Smith   ierr = PetscFree2(bj_ptr,bjlvl_ptr);CHKERRQ(ierr);
1846813a1f2bSShri Abhyankar 
1847813a1f2bSShri Abhyankar #if defined(PETSC_USE_INFO)
1848813a1f2bSShri Abhyankar   {
1849aef85c9fSShri Abhyankar     PetscReal af = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]);
1850813a1f2bSShri Abhyankar     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr);
1851813a1f2bSShri Abhyankar     ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
1852813a1f2bSShri Abhyankar     ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr);
1853813a1f2bSShri Abhyankar     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
1854813a1f2bSShri Abhyankar     if (diagonal_fill) {
1855813a1f2bSShri Abhyankar       ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr);
1856813a1f2bSShri Abhyankar     }
1857813a1f2bSShri Abhyankar   }
1858813a1f2bSShri Abhyankar #endif
1859813a1f2bSShri Abhyankar 
1860813a1f2bSShri Abhyankar   /* put together the new matrix */
1861813a1f2bSShri Abhyankar   ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
1862813a1f2bSShri Abhyankar   ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr);
1863813a1f2bSShri Abhyankar   b = (Mat_SeqAIJ*)(fact)->data;
1864813a1f2bSShri Abhyankar   b->free_a       = PETSC_TRUE;
1865813a1f2bSShri Abhyankar   b->free_ij      = PETSC_TRUE;
1866813a1f2bSShri Abhyankar   b->singlemalloc = PETSC_FALSE;
1867f268cba8SShri Abhyankar   ierr = PetscMalloc((bdiag[0]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);
1868813a1f2bSShri Abhyankar   b->j          = bj;
1869813a1f2bSShri Abhyankar   b->i          = bi;
1870813a1f2bSShri Abhyankar   b->diag       = bdiag;
1871813a1f2bSShri Abhyankar   b->ilen       = 0;
1872813a1f2bSShri Abhyankar   b->imax       = 0;
1873813a1f2bSShri Abhyankar   b->row        = isrow;
1874813a1f2bSShri Abhyankar   b->col        = iscol;
1875813a1f2bSShri Abhyankar   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
1876813a1f2bSShri Abhyankar   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
1877813a1f2bSShri Abhyankar   b->icol       = isicol;
1878813a1f2bSShri Abhyankar   ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
1879813a1f2bSShri Abhyankar   /* In b structure:  Free imax, ilen, old a, old j.
1880813a1f2bSShri Abhyankar      Allocate bdiag, solve_work, new a, new j */
1881f268cba8SShri Abhyankar   ierr = PetscLogObjectMemory(fact,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
1882f268cba8SShri Abhyankar   b->maxnz = b->nz = bdiag[0]+1;
1883813a1f2bSShri Abhyankar   (fact)->info.factor_mallocs    = reallocs;
1884813a1f2bSShri Abhyankar   (fact)->info.fill_ratio_given  = f;
1885f268cba8SShri Abhyankar   (fact)->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]);
188635233d99SShri Abhyankar   (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ;
1887f2344125SBarry Smith   if (a->inode.size) {
1888f2344125SBarry Smith     (fact)->ops->lufactornumeric  = MatLUFactorNumeric_SeqAIJ_Inode;
1889f2344125SBarry Smith   }
1890813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
1891813a1f2bSShri Abhyankar }
1892813a1f2bSShri Abhyankar 
18936516239fSHong Zhang #undef __FUNCT__
1894ad04f41aSHong Zhang #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_inplace"
1895ad04f41aSHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
18966516239fSHong Zhang {
18976516239fSHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
18986516239fSHong Zhang   IS                 isicol;
18996516239fSHong Zhang   PetscErrorCode     ierr;
19006516239fSHong Zhang   const PetscInt     *r,*ic;
19016516239fSHong Zhang   PetscInt           n=A->rmap->n,*ai=a->i,*aj=a->j,d;
19026516239fSHong Zhang   PetscInt           *bi,*cols,nnz,*cols_lvl;
19036516239fSHong Zhang   PetscInt           *bdiag,prow,fm,nzbd,reallocs=0,dcount=0;
19046516239fSHong Zhang   PetscInt           i,levels,diagonal_fill;
1905ace3abfcSBarry Smith   PetscBool          col_identity,row_identity;
19066516239fSHong Zhang   PetscReal          f;
19076516239fSHong Zhang   PetscInt           nlnk,*lnk,*lnk_lvl=PETSC_NULL;
19086516239fSHong Zhang   PetscBT            lnkbt;
19096516239fSHong Zhang   PetscInt           nzi,*bj,**bj_ptr,**bjlvl_ptr;
19106516239fSHong Zhang   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
19116516239fSHong Zhang   PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL;
1912ace3abfcSBarry Smith   PetscBool          missing;
19136516239fSHong Zhang 
19146516239fSHong Zhang   PetscFunctionBegin;
1915e32f2f54SBarry Smith   if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n);
19166516239fSHong Zhang   f             = info->fill;
19176516239fSHong Zhang   levels        = (PetscInt)info->levels;
19186516239fSHong Zhang   diagonal_fill = (PetscInt)info->diagonal_fill;
19196516239fSHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
19206516239fSHong Zhang 
19216516239fSHong Zhang   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
19226516239fSHong Zhang   ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr);
19236516239fSHong Zhang   if (!levels && row_identity && col_identity) { /* special case: ilu(0) with natural ordering */
1924f77e22a1SHong Zhang     ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_TRUE);CHKERRQ(ierr);
1925ad04f41aSHong Zhang     (fact)->ops->lufactornumeric =  MatLUFactorNumeric_SeqAIJ_inplace;
1926f2344125SBarry Smith     if (a->inode.size) {
1927f2344125SBarry Smith       (fact)->ops->lufactornumeric  = MatLUFactorNumeric_SeqAIJ_Inode_inplace;
1928f2344125SBarry Smith     }
1929d5f3da31SBarry Smith     fact->factortype = MAT_FACTOR_ILU;
1930719d5645SBarry Smith     (fact)->info.factor_mallocs    = 0;
1931719d5645SBarry Smith     (fact)->info.fill_ratio_given  = info->fill;
1932719d5645SBarry Smith     (fact)->info.fill_ratio_needed = 1.0;
1933719d5645SBarry Smith     b               = (Mat_SeqAIJ*)(fact)->data;
19348ef3462fSBarry Smith     ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr);
1935e32f2f54SBarry Smith     if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d);
193608480c60SBarry Smith     b->row              = isrow;
193708480c60SBarry Smith     b->col              = iscol;
193882bf6240SBarry Smith     b->icol             = isicol;
1939719d5645SBarry Smith     ierr                = PetscMalloc(((fact)->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
1940c38d4ed2SBarry Smith     ierr                = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
1941c38d4ed2SBarry Smith     ierr                = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
19423a40ed3dSBarry Smith     PetscFunctionReturn(0);
194308480c60SBarry Smith   }
194408480c60SBarry Smith 
1945898183ecSLois Curfman McInnes   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
1946898183ecSLois Curfman McInnes   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
19479e25ed09SBarry Smith 
19481bfa06eaSJed Brown   /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */
19491bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
19501bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
1951a6df321fSShri Abhyankar   bi[0] = bdiag[0] = 0;
19526cf997b0SBarry Smith 
19530e83c824SBarry Smith   ierr = PetscMalloc2(n,PetscInt*,&bj_ptr,n,PetscInt*,&bjlvl_ptr);CHKERRQ(ierr);
19545a8e39fbSHong Zhang 
19555a8e39fbSHong Zhang   /* create a linked list for storing column indices of the active row */
19565a8e39fbSHong Zhang   nlnk = n + 1;
19575a8e39fbSHong Zhang   ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
19585a8e39fbSHong Zhang 
19595a8e39fbSHong Zhang   /* initial FreeSpace size is f*(ai[n]+1) */
1960a1a86e44SBarry Smith   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr);
19615a8e39fbSHong Zhang   current_space = free_space;
1962a1a86e44SBarry Smith   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr);
19635a8e39fbSHong Zhang   current_space_lvl = free_space_lvl;
19645a8e39fbSHong Zhang 
19655a8e39fbSHong Zhang   for (i=0; i<n; i++) {
19665a8e39fbSHong Zhang     nzi = 0;
19676cf997b0SBarry Smith     /* copy current row into linked list */
19685a8e39fbSHong Zhang     nnz  = ai[r[i]+1] - ai[r[i]];
1969e32f2f54SBarry Smith     if (!nnz) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i);
19705a8e39fbSHong Zhang     cols = aj + ai[r[i]];
19715a8e39fbSHong Zhang     lnk[i] = -1; /* marker to indicate if diagonal exists */
19725a8e39fbSHong Zhang     ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
19735a8e39fbSHong Zhang     nzi += nlnk;
19746cf997b0SBarry Smith 
19756cf997b0SBarry Smith     /* make sure diagonal entry is included */
19765a8e39fbSHong Zhang     if (diagonal_fill && lnk[i] == -1) {
19776cf997b0SBarry Smith       fm = n;
19785a8e39fbSHong Zhang       while (lnk[fm] < i) fm = lnk[fm];
19795a8e39fbSHong Zhang       lnk[i]     = lnk[fm]; /* insert diagonal into linked list */
19805a8e39fbSHong Zhang       lnk[fm]    = i;
19815a8e39fbSHong Zhang       lnk_lvl[i] = 0;
19825a8e39fbSHong Zhang       nzi++; dcount++;
19836cf997b0SBarry Smith     }
19846cf997b0SBarry Smith 
19855a8e39fbSHong Zhang     /* add pivot rows into the active row */
19865a8e39fbSHong Zhang     nzbd = 0;
19875a8e39fbSHong Zhang     prow = lnk[n];
19885a8e39fbSHong Zhang     while (prow < i) {
19895a8e39fbSHong Zhang       nnz      = bdiag[prow];
19905a8e39fbSHong Zhang       cols     = bj_ptr[prow] + nnz + 1;
19915a8e39fbSHong Zhang       cols_lvl = bjlvl_ptr[prow] + nnz + 1;
19925a8e39fbSHong Zhang       nnz      = bi[prow+1] - bi[prow] - nnz - 1;
19935a8e39fbSHong Zhang       ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr);
19945a8e39fbSHong Zhang       nzi += nlnk;
19955a8e39fbSHong Zhang       prow = lnk[prow];
19965a8e39fbSHong Zhang       nzbd++;
199756beaf04SBarry Smith     }
19985a8e39fbSHong Zhang     bdiag[i] = nzbd;
19995a8e39fbSHong Zhang     bi[i+1]  = bi[i] + nzi;
2000ecf371e4SBarry Smith 
20015a8e39fbSHong Zhang     /* if free space is not available, make more free space */
20025a8e39fbSHong Zhang     if (current_space->local_remaining<nzi) {
20035a8e39fbSHong Zhang       nnz = nzi*(n - i); /* estimated and max additional space needed */
2004a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
2005a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(nnz,&current_space_lvl);CHKERRQ(ierr);
20065a8e39fbSHong Zhang       reallocs++;
20075a8e39fbSHong Zhang     }
2008ecf371e4SBarry Smith 
20095a8e39fbSHong Zhang     /* copy data into free_space and free_space_lvl, then initialize lnk */
20105a8e39fbSHong Zhang     ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
20115a8e39fbSHong Zhang     bj_ptr[i]    = current_space->array;
20125a8e39fbSHong Zhang     bjlvl_ptr[i] = current_space_lvl->array;
20135a8e39fbSHong Zhang 
20145a8e39fbSHong Zhang     /* make sure the active row i has diagonal entry */
201565e19b50SBarry Smith     if (*(bj_ptr[i]+bdiag[i]) != i) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Row %D has missing diagonal in factored matrix\ntry running with -pc_factor_nonzeros_along_diagonal or -pc_factor_diagonal_fill",i);
20165a8e39fbSHong Zhang 
20175a8e39fbSHong Zhang     current_space->array           += nzi;
20185a8e39fbSHong Zhang     current_space->local_used      += nzi;
20195a8e39fbSHong Zhang     current_space->local_remaining -= nzi;
20205a8e39fbSHong Zhang     current_space_lvl->array           += nzi;
20215a8e39fbSHong Zhang     current_space_lvl->local_used      += nzi;
20225a8e39fbSHong Zhang     current_space_lvl->local_remaining -= nzi;
20239e25ed09SBarry Smith   }
20245a8e39fbSHong Zhang 
2025898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
2026898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
20275a8e39fbSHong Zhang 
20285a8e39fbSHong Zhang   /* destroy list of free space and other temporary arrays */
20295a8e39fbSHong Zhang   ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
20306516239fSHong Zhang   ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); /* copy free_space -> bj */
20315a8e39fbSHong Zhang   ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
2032a1a86e44SBarry Smith   ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
20330e83c824SBarry Smith   ierr = PetscFree2(bj_ptr,bjlvl_ptr);CHKERRQ(ierr);
20349e25ed09SBarry Smith 
20356cf91177SBarry Smith #if defined(PETSC_USE_INFO)
2036f64065bbSBarry Smith   {
20375a8e39fbSHong Zhang     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
2038ae15b995SBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr);
2039ae15b995SBarry Smith     ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
2040ae15b995SBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr);
2041ae15b995SBarry Smith     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
2042335d9088SBarry Smith     if (diagonal_fill) {
2043ae15b995SBarry Smith       ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr);
2044335d9088SBarry Smith     }
2045f64065bbSBarry Smith   }
204663ba0a88SBarry Smith #endif
2047bbb6d6a8SBarry Smith 
20489e25ed09SBarry Smith   /* put together the new matrix */
204907dbf95fSBarry Smith   ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
2050719d5645SBarry Smith   ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr);
2051719d5645SBarry Smith   b = (Mat_SeqAIJ*)(fact)->data;
2052e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
2053e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
20547c922b88SBarry Smith   b->singlemalloc = PETSC_FALSE;
20550e83c824SBarry Smith   ierr = PetscMalloc(bi[n]*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);
20565a8e39fbSHong Zhang   b->j          = bj;
20575a8e39fbSHong Zhang   b->i          = bi;
20585a8e39fbSHong Zhang   for (i=0; i<n; i++) bdiag[i] += bi[i];
20595a8e39fbSHong Zhang   b->diag       = bdiag;
2060416022c9SBarry Smith   b->ilen       = 0;
2061416022c9SBarry Smith   b->imax       = 0;
2062416022c9SBarry Smith   b->row        = isrow;
2063416022c9SBarry Smith   b->col        = iscol;
2064c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
2065c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
206682bf6240SBarry Smith   b->icol       = isicol;
206787828ca2SBarry Smith   ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
2068416022c9SBarry Smith   /* In b structure:  Free imax, ilen, old a, old j.
20695a8e39fbSHong Zhang      Allocate bdiag, solve_work, new a, new j */
2070719d5645SBarry Smith   ierr = PetscLogObjectMemory(fact,(bi[n]-n) * (sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
20715a8e39fbSHong Zhang   b->maxnz             = b->nz = bi[n] ;
2072719d5645SBarry Smith   (fact)->info.factor_mallocs    = reallocs;
2073719d5645SBarry Smith   (fact)->info.fill_ratio_given  = f;
2074719d5645SBarry Smith   (fact)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]);
2075ad04f41aSHong Zhang   (fact)->ops->lufactornumeric =  MatLUFactorNumeric_SeqAIJ_inplace;
2076f2344125SBarry Smith   if (a->inode.size) {
2077f2344125SBarry Smith     (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_Inode_inplace;
2078d3ac4fa3SBarry Smith   }
20793a40ed3dSBarry Smith   PetscFunctionReturn(0);
20809e25ed09SBarry Smith }
2081d5d45c9bSBarry Smith 
2082a6175056SHong Zhang #undef __FUNCT__
208335233d99SShri Abhyankar #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ"
208435233d99SShri Abhyankar PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info)
20855f44c854SHong Zhang {
20865f44c854SHong Zhang   Mat            C = B;
20875f44c854SHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
20885f44c854SHong Zhang   Mat_SeqSBAIJ   *b=(Mat_SeqSBAIJ*)C->data;
20895f44c854SHong Zhang   IS             ip=b->row,iip = b->icol;
20905f44c854SHong Zhang   PetscErrorCode ierr;
20915f44c854SHong Zhang   const PetscInt *rip,*riip;
20925f44c854SHong Zhang   PetscInt       i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bdiag=b->diag,*bjtmp;
20935f44c854SHong Zhang   PetscInt       *ai=a->i,*aj=a->j;
20945f44c854SHong Zhang   PetscInt       k,jmin,jmax,*c2r,*il,col,nexti,ili,nz;
20955f44c854SHong Zhang   MatScalar      *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi;
2096ace3abfcSBarry Smith   PetscBool      perm_identity;
2097d90ac83dSHong Zhang   FactorShiftCtx sctx;
209898a93161SHong Zhang   PetscReal      rs;
209998a93161SHong Zhang   MatScalar      d,*v;
210098a93161SHong Zhang 
21015f44c854SHong Zhang   PetscFunctionBegin;
210298a93161SHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
2103d90ac83dSHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
210498a93161SHong Zhang 
2105f4db908eSBarry Smith   if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
210698a93161SHong Zhang     sctx.shift_top = info->zeropivot;
210798a93161SHong Zhang     for (i=0; i<mbs; i++) {
210898a93161SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
210998a93161SHong Zhang       d  = (aa)[a->diag[i]];
211098a93161SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
211198a93161SHong Zhang       v  = aa+ai[i];
211298a93161SHong Zhang       nz = ai[i+1] - ai[i];
211398a93161SHong Zhang       for (j=0; j<nz; j++)
211498a93161SHong Zhang 	rs += PetscAbsScalar(v[j]);
211598a93161SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
211698a93161SHong Zhang     }
211798a93161SHong Zhang     sctx.shift_top   *= 1.1;
211898a93161SHong Zhang     sctx.nshift_max   = 5;
211998a93161SHong Zhang     sctx.shift_lo     = 0.;
212098a93161SHong Zhang     sctx.shift_hi     = 1.;
212198a93161SHong Zhang   }
21225f44c854SHong Zhang 
21235f44c854SHong Zhang   ierr  = ISGetIndices(ip,&rip);CHKERRQ(ierr);
21245f44c854SHong Zhang   ierr  = ISGetIndices(iip,&riip);CHKERRQ(ierr);
21255f44c854SHong Zhang 
21265f44c854SHong Zhang   /* allocate working arrays
21275f44c854SHong Zhang      c2r: linked list, keep track of pivot rows for a given column. c2r[col]: head of the list for a given col
21285f44c854SHong 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
21295f44c854SHong Zhang   */
21300e83c824SBarry Smith   ierr = PetscMalloc3(mbs,MatScalar,&rtmp,mbs,PetscInt,&il,mbs,PetscInt,&c2r);CHKERRQ(ierr);
21315f44c854SHong Zhang 
213298a93161SHong Zhang   do {
213307b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
213498a93161SHong Zhang 
2135c5546cabSHong Zhang     for (i=0; i<mbs; i++) c2r[i] = mbs;
21365f44c854SHong Zhang     il[0] = 0;
21375f44c854SHong Zhang 
21385f44c854SHong Zhang     for (k = 0; k<mbs; k++){
21395f44c854SHong Zhang       /* zero rtmp */
21405f44c854SHong Zhang       nz = bi[k+1] - bi[k];
21415f44c854SHong Zhang       bjtmp = bj + bi[k];
21425f44c854SHong Zhang       for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
21435f44c854SHong Zhang 
21445f44c854SHong Zhang       /* load in initial unfactored row */
21455f44c854SHong Zhang       bval = ba + bi[k];
21465f44c854SHong Zhang       jmin = ai[rip[k]]; jmax = ai[rip[k]+1];
21475f44c854SHong Zhang       for (j = jmin; j < jmax; j++){
21485f44c854SHong Zhang         col = riip[aj[j]];
21495f44c854SHong Zhang         if (col >= k){ /* only take upper triangular entry */
21505f44c854SHong Zhang           rtmp[col] = aa[j];
21515f44c854SHong Zhang           *bval++   = 0.0; /* for in-place factorization */
21525f44c854SHong Zhang         }
21535f44c854SHong Zhang       }
215498a93161SHong Zhang       /* shift the diagonal of the matrix: ZeropivotApply() */
215598a93161SHong Zhang       rtmp[k] += sctx.shift_amount;  /* shift the diagonal of the matrix */
21565f44c854SHong Zhang 
21575f44c854SHong Zhang       /* modify k-th row by adding in those rows i with U(i,k)!=0 */
21585f44c854SHong Zhang       dk = rtmp[k];
21595f44c854SHong Zhang       i  = c2r[k]; /* first row to be added to k_th row  */
21605f44c854SHong Zhang 
21615f44c854SHong Zhang       while (i < k){
21625f44c854SHong Zhang         nexti = c2r[i]; /* next row to be added to k_th row */
21635f44c854SHong Zhang 
21645f44c854SHong Zhang         /* compute multiplier, update diag(k) and U(i,k) */
21655f44c854SHong Zhang         ili   = il[i];  /* index of first nonzero element in U(i,k:bms-1) */
21665f44c854SHong Zhang         uikdi = - ba[ili]*ba[bdiag[i]];  /* diagonal(k) */
21675f44c854SHong Zhang         dk   += uikdi*ba[ili]; /* update diag[k] */
21685f44c854SHong Zhang         ba[ili] = uikdi; /* -U(i,k) */
21695f44c854SHong Zhang 
21705f44c854SHong Zhang         /* add multiple of row i to k-th row */
21715f44c854SHong Zhang         jmin = ili + 1; jmax = bi[i+1];
21725f44c854SHong Zhang         if (jmin < jmax){
21735f44c854SHong Zhang           for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j];
21745f44c854SHong Zhang           /* update il and c2r for row i */
21755f44c854SHong Zhang           il[i] = jmin;
21765f44c854SHong Zhang           j = bj[jmin]; c2r[i] = c2r[j]; c2r[j] = i;
21775f44c854SHong Zhang         }
21785f44c854SHong Zhang         i = nexti;
21795f44c854SHong Zhang       }
21805f44c854SHong Zhang 
21815f44c854SHong Zhang       /* copy data into U(k,:) */
218298a93161SHong Zhang       rs   = 0.0;
21835f44c854SHong Zhang       jmin = bi[k]; jmax = bi[k+1]-1;
21845f44c854SHong Zhang       if (jmin < jmax) {
21855f44c854SHong Zhang         for (j=jmin; j<jmax; j++){
218698a93161SHong Zhang           col = bj[j]; ba[j] = rtmp[col]; rs += PetscAbsScalar(ba[j]);
21875f44c854SHong Zhang         }
21885f44c854SHong Zhang         /* add the k-th row into il and c2r */
21895f44c854SHong Zhang         il[k] = jmin;
21905f44c854SHong Zhang         i = bj[jmin]; c2r[k] = c2r[i]; c2r[i] = k;
21915f44c854SHong Zhang       }
219298a93161SHong Zhang 
219398a93161SHong Zhang       /* MatPivotCheck() */
219498a93161SHong Zhang       sctx.rs  = rs;
219598a93161SHong Zhang       sctx.pv  = dk;
2196d0660788SBarry Smith       ierr = MatPivotCheck(A,info,&sctx,i);CHKERRQ(ierr);
219707b50cabSHong Zhang       if(sctx.newshift) break;
219898a93161SHong Zhang       dk = sctx.pv;
219998a93161SHong Zhang 
220098a93161SHong Zhang       ba[bdiag[k]] = 1.0/dk; /* U(k,k) */
220198a93161SHong Zhang     }
220207b50cabSHong Zhang   } while (sctx.newshift);
22035f44c854SHong Zhang 
22040e83c824SBarry Smith   ierr = PetscFree3(rtmp,il,c2r);CHKERRQ(ierr);
22055f44c854SHong Zhang   ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr);
22065f44c854SHong Zhang   ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr);
22075f44c854SHong Zhang 
22085f44c854SHong Zhang   ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr);
22095f44c854SHong Zhang   if (perm_identity){
221035233d99SShri Abhyankar     B->ops->solve           = MatSolve_SeqSBAIJ_1_NaturalOrdering;
221135233d99SShri Abhyankar     B->ops->solvetranspose  = MatSolve_SeqSBAIJ_1_NaturalOrdering;
22122169352eSHong Zhang     B->ops->forwardsolve    = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering;
22132169352eSHong Zhang     B->ops->backwardsolve   = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering;
22145f44c854SHong Zhang   } else {
221535233d99SShri Abhyankar     B->ops->solve           = MatSolve_SeqSBAIJ_1;
221635233d99SShri Abhyankar     B->ops->solvetranspose  = MatSolve_SeqSBAIJ_1;
221735233d99SShri Abhyankar     B->ops->forwardsolve    = MatForwardSolve_SeqSBAIJ_1;
221835233d99SShri Abhyankar     B->ops->backwardsolve   = MatBackwardSolve_SeqSBAIJ_1;
22195f44c854SHong Zhang   }
22205f44c854SHong Zhang 
22215f44c854SHong Zhang   C->assembled    = PETSC_TRUE;
22225f44c854SHong Zhang   C->preallocated = PETSC_TRUE;
22235f44c854SHong Zhang   ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr);
222498a93161SHong Zhang 
222598a93161SHong Zhang   /* MatPivotView() */
222698a93161SHong Zhang   if (sctx.nshift){
2227f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
222898a93161SHong 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);
2229f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
223098a93161SHong Zhang       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
2231f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_INBLOCKS){
2232b2a402fcSHong Zhang       ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %G\n",sctx.nshift,info->shiftamount);CHKERRQ(ierr);
223398a93161SHong Zhang     }
223498a93161SHong Zhang   }
22355f44c854SHong Zhang   PetscFunctionReturn(0);
22365f44c854SHong Zhang }
22375f44c854SHong Zhang 
22385f44c854SHong Zhang #undef __FUNCT__
22390a3c351aSHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ_inplace"
22400a3c351aSHong Zhang PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ_inplace(Mat B,Mat A,const MatFactorInfo *info)
2241a6175056SHong Zhang {
2242719d5645SBarry Smith   Mat            C = B;
22430968510aSHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
22442ed133dbSHong Zhang   Mat_SeqSBAIJ   *b=(Mat_SeqSBAIJ*)C->data;
22459bfd6278SHong Zhang   IS             ip=b->row,iip = b->icol;
22462ed133dbSHong Zhang   PetscErrorCode ierr;
22475d0c19d7SBarry Smith   const PetscInt *rip,*riip;
2248c5546cabSHong Zhang   PetscInt       i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bcol,*bjtmp;
22492ed133dbSHong Zhang   PetscInt       *ai=a->i,*aj=a->j;
22502ed133dbSHong Zhang   PetscInt       k,jmin,jmax,*jl,*il,col,nexti,ili,nz;
2251622e2028SHong Zhang   MatScalar      *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi;
2252ace3abfcSBarry Smith   PetscBool      perm_identity;
22530e95ead3SHong Zhang   FactorShiftCtx sctx;
22540e95ead3SHong Zhang   PetscReal      rs;
22550e95ead3SHong Zhang   MatScalar      d,*v;
2256d5d45c9bSBarry Smith 
2257a6175056SHong Zhang   PetscFunctionBegin;
22580e95ead3SHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
22590e95ead3SHong Zhang   ierr = PetscMemzero(&sctx,sizeof(FactorShiftCtx));CHKERRQ(ierr);
22600e95ead3SHong Zhang 
22610e95ead3SHong Zhang   if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) { /* set sctx.shift_top=max{rs} */
22620e95ead3SHong Zhang     sctx.shift_top = info->zeropivot;
22630e95ead3SHong Zhang     for (i=0; i<mbs; i++) {
22640e95ead3SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
22650e95ead3SHong Zhang       d  = (aa)[a->diag[i]];
22660e95ead3SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
22670e95ead3SHong Zhang       v  = aa+ai[i];
22680e95ead3SHong Zhang       nz = ai[i+1] - ai[i];
22690e95ead3SHong Zhang       for (j=0; j<nz; j++)
22700e95ead3SHong Zhang 	rs += PetscAbsScalar(v[j]);
22710e95ead3SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
22720e95ead3SHong Zhang     }
22730e95ead3SHong Zhang     sctx.shift_top   *= 1.1;
22740e95ead3SHong Zhang     sctx.nshift_max   = 5;
22750e95ead3SHong Zhang     sctx.shift_lo     = 0.;
22760e95ead3SHong Zhang     sctx.shift_hi     = 1.;
22770e95ead3SHong Zhang   }
2278ee45ca4aSHong Zhang 
22792ed133dbSHong Zhang   ierr  = ISGetIndices(ip,&rip);CHKERRQ(ierr);
22809bfd6278SHong Zhang   ierr  = ISGetIndices(iip,&riip);CHKERRQ(ierr);
22812ed133dbSHong Zhang 
22822ed133dbSHong Zhang   /* initialization */
22830e83c824SBarry Smith   ierr = PetscMalloc3(mbs,MatScalar,&rtmp,mbs,PetscInt,&il,mbs,PetscInt,&jl);CHKERRQ(ierr);
22840e95ead3SHong Zhang 
22852ed133dbSHong Zhang   do {
228607b50cabSHong Zhang     sctx.newshift = PETSC_FALSE;
22870e95ead3SHong Zhang 
2288c5546cabSHong Zhang     for (i=0; i<mbs; i++) jl[i] = mbs;
2289c5546cabSHong Zhang     il[0] = 0;
22902ed133dbSHong Zhang 
22912ed133dbSHong Zhang     for (k = 0; k<mbs; k++){
2292c5546cabSHong Zhang       /* zero rtmp */
2293c5546cabSHong Zhang       nz = bi[k+1] - bi[k];
2294c5546cabSHong Zhang       bjtmp = bj + bi[k];
2295c5546cabSHong Zhang       for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
2296c5546cabSHong Zhang 
2297c05c3958SHong Zhang       bval = ba + bi[k];
22982ed133dbSHong Zhang       /* initialize k-th row by the perm[k]-th row of A */
22992ed133dbSHong Zhang       jmin = ai[rip[k]]; jmax = ai[rip[k]+1];
23002ed133dbSHong Zhang       for (j = jmin; j < jmax; j++){
23019bfd6278SHong Zhang         col = riip[aj[j]];
23022ed133dbSHong Zhang         if (col >= k){ /* only take upper triangular entry */
23032ed133dbSHong Zhang           rtmp[col] = aa[j];
2304c05c3958SHong Zhang           *bval++  = 0.0; /* for in-place factorization */
23052ed133dbSHong Zhang         }
23062ed133dbSHong Zhang       }
230739e3d630SHong Zhang       /* shift the diagonal of the matrix */
2308540703acSHong Zhang       if (sctx.nshift) rtmp[k] += sctx.shift_amount;
23092ed133dbSHong Zhang 
23102ed133dbSHong Zhang       /* modify k-th row by adding in those rows i with U(i,k)!=0 */
23112ed133dbSHong Zhang       dk = rtmp[k];
23122ed133dbSHong Zhang       i = jl[k]; /* first row to be added to k_th row  */
23132ed133dbSHong Zhang 
23142ed133dbSHong Zhang       while (i < k){
23152ed133dbSHong Zhang         nexti = jl[i]; /* next row to be added to k_th row */
23162ed133dbSHong Zhang 
23172ed133dbSHong Zhang         /* compute multiplier, update diag(k) and U(i,k) */
23182ed133dbSHong Zhang         ili = il[i];  /* index of first nonzero element in U(i,k:bms-1) */
23192ed133dbSHong Zhang         uikdi = - ba[ili]*ba[bi[i]];  /* diagonal(k) */
23202ed133dbSHong Zhang         dk += uikdi*ba[ili];
23212ed133dbSHong Zhang         ba[ili] = uikdi; /* -U(i,k) */
23222ed133dbSHong Zhang 
23232ed133dbSHong Zhang         /* add multiple of row i to k-th row */
23242ed133dbSHong Zhang         jmin = ili + 1; jmax = bi[i+1];
23252ed133dbSHong Zhang         if (jmin < jmax){
23262ed133dbSHong Zhang           for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j];
23272ed133dbSHong Zhang           /* update il and jl for row i */
23282ed133dbSHong Zhang           il[i] = jmin;
23292ed133dbSHong Zhang           j = bj[jmin]; jl[i] = jl[j]; jl[j] = i;
23302ed133dbSHong Zhang         }
23312ed133dbSHong Zhang         i = nexti;
23322ed133dbSHong Zhang       }
23332ed133dbSHong Zhang 
23340697b6caSHong Zhang       /* shift the diagonals when zero pivot is detected */
23350697b6caSHong Zhang       /* compute rs=sum of abs(off-diagonal) */
23360697b6caSHong Zhang       rs   = 0.0;
23373c9e8598SHong Zhang       jmin = bi[k]+1;
23383c9e8598SHong Zhang       nz   = bi[k+1] - jmin;
23393c9e8598SHong Zhang       bcol = bj + jmin;
234004fbf559SBarry Smith       for (j=0; j<nz; j++) {
234104fbf559SBarry Smith         rs += PetscAbsScalar(rtmp[bcol[j]]);
23423c9e8598SHong Zhang       }
2343540703acSHong Zhang 
2344540703acSHong Zhang       sctx.rs = rs;
2345540703acSHong Zhang       sctx.pv = dk;
2346d0660788SBarry Smith       ierr = MatPivotCheck(A,info,&sctx,k);CHKERRQ(ierr);
234707b50cabSHong Zhang       if (sctx.newshift) break;
23480e95ead3SHong Zhang       dk = sctx.pv;
23493c9e8598SHong Zhang 
23503c9e8598SHong Zhang       /* copy data into U(k,:) */
235139e3d630SHong Zhang       ba[bi[k]] = 1.0/dk; /* U(k,k) */
235239e3d630SHong Zhang       jmin = bi[k]+1; jmax = bi[k+1];
235339e3d630SHong Zhang       if (jmin < jmax) {
235439e3d630SHong Zhang         for (j=jmin; j<jmax; j++){
2355c6ae9bfcSHong Zhang           col = bj[j]; ba[j] = rtmp[col];
23563c9e8598SHong Zhang         }
235739e3d630SHong Zhang         /* add the k-th row into il and jl */
23583c9e8598SHong Zhang         il[k] = jmin;
23593c9e8598SHong Zhang         i = bj[jmin]; jl[k] = jl[i]; jl[i] = k;
23603c9e8598SHong Zhang       }
23613c9e8598SHong Zhang     }
236207b50cabSHong Zhang   } while (sctx.newshift);
23630e95ead3SHong Zhang 
23640e83c824SBarry Smith   ierr = PetscFree3(rtmp,il,jl);CHKERRQ(ierr);
236539e3d630SHong Zhang   ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr);
23669bfd6278SHong Zhang   ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr);
2367db4efbfdSBarry Smith 
2368db4efbfdSBarry Smith   ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr);
2369db4efbfdSBarry Smith   if (perm_identity){
23700a3c351aSHong Zhang     B->ops->solve           = MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace;
23710a3c351aSHong Zhang     B->ops->solvetranspose  = MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace;
23720a3c351aSHong Zhang     B->ops->forwardsolve    = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace;
23730a3c351aSHong Zhang     B->ops->backwardsolve   = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace;
2374db4efbfdSBarry Smith   } else {
23750a3c351aSHong Zhang     B->ops->solve           = MatSolve_SeqSBAIJ_1_inplace;
23760a3c351aSHong Zhang     B->ops->solvetranspose  = MatSolve_SeqSBAIJ_1_inplace;
23770a3c351aSHong Zhang     B->ops->forwardsolve    = MatForwardSolve_SeqSBAIJ_1_inplace;
23780a3c351aSHong Zhang     B->ops->backwardsolve   = MatBackwardSolve_SeqSBAIJ_1_inplace;
2379db4efbfdSBarry Smith   }
2380db4efbfdSBarry Smith 
23813c9e8598SHong Zhang   C->assembled    = PETSC_TRUE;
23823c9e8598SHong Zhang   C->preallocated = PETSC_TRUE;
2383d0f46423SBarry Smith   ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr);
2384540703acSHong Zhang   if (sctx.nshift){
2385f4db908eSBarry Smith     if (info->shifttype == (PetscReal)MAT_SHIFT_NONZERO) {
23861e2582c4SBarry Smith       ierr = PetscInfo2(A,"number of shiftnz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
2387f4db908eSBarry Smith     } else if (info->shifttype == (PetscReal)MAT_SHIFT_POSITIVE_DEFINITE) {
23881e2582c4SBarry Smith       ierr = PetscInfo2(A,"number of shiftpd tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
23890697b6caSHong Zhang     }
23903c9e8598SHong Zhang   }
23913c9e8598SHong Zhang   PetscFunctionReturn(0);
23923c9e8598SHong Zhang }
2393a6175056SHong Zhang 
23948a76ed4fSHong Zhang /*
23958a76ed4fSHong Zhang    icc() under revised new data structure.
23968a76ed4fSHong Zhang    Factored arrays bj and ba are stored as
23978a76ed4fSHong Zhang      U(0,:),...,U(i,:),U(n-1,:)
23988a76ed4fSHong Zhang 
23998a76ed4fSHong Zhang    ui=fact->i is an array of size n+1, in which
24008a76ed4fSHong Zhang    ui+
24018a76ed4fSHong Zhang      ui[i]:  points to 1st entry of U(i,:),i=0,...,n-1
24028a76ed4fSHong Zhang      ui[n]:  points to U(n-1,n-1)+1
24038a76ed4fSHong Zhang 
24048a76ed4fSHong Zhang   udiag=fact->diag is an array of size n,in which
24058a76ed4fSHong Zhang      udiag[i]: points to diagonal of U(i,:), i=0,...,n-1
24068a76ed4fSHong Zhang 
24078a76ed4fSHong Zhang    U(i,:) contains udiag[i] as its last entry, i.e.,
24088a76ed4fSHong Zhang     U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i])
24098a76ed4fSHong Zhang */
24108a76ed4fSHong Zhang 
2411a6175056SHong Zhang #undef __FUNCT__
241235233d99SShri Abhyankar #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ"
241335233d99SShri Abhyankar PetscErrorCode MatICCFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
2414a6175056SHong Zhang {
24150968510aSHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
2416ed59401aSHong Zhang   Mat_SeqSBAIJ       *b;
2417dfbe8321SBarry Smith   PetscErrorCode     ierr;
2418ace3abfcSBarry Smith   PetscBool          perm_identity,missing;
24195f44c854SHong Zhang   PetscInt           reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag;
24205d0c19d7SBarry Smith   const PetscInt     *rip,*riip;
2421ed59401aSHong Zhang   PetscInt           jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow;
242258ebbce7SBarry Smith   PetscInt           nlnk,*lnk,*lnk_lvl=PETSC_NULL,d;
24235a8e39fbSHong Zhang   PetscInt           ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr;
2424ed59401aSHong Zhang   PetscReal          fill=info->fill,levels=info->levels;
2425a1a86e44SBarry Smith   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
2426a1a86e44SBarry Smith   PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL;
24277a48dd6fSHong Zhang   PetscBT            lnkbt;
2428b635d36bSHong Zhang   IS                 iperm;
2429a6175056SHong Zhang 
2430a6175056SHong Zhang   PetscFunctionBegin;
2431e32f2f54SBarry 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);
243258ebbce7SBarry Smith   ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr);
2433e32f2f54SBarry Smith   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d);
2434653a6975SHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
2435b635d36bSHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
24367a48dd6fSHong Zhang 
243739e3d630SHong Zhang   ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr);
24385f44c854SHong Zhang   ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr);
243939e3d630SHong Zhang   ui[0] = 0;
244039e3d630SHong Zhang 
24418a76ed4fSHong Zhang   /* ICC(0) without matrix ordering: simply rearrange column indices */
2442622e2028SHong Zhang   if (!levels && perm_identity) {
24435f44c854SHong Zhang     for (i=0; i<am; i++) {
2444c5546cabSHong Zhang       ncols    = ai[i+1] - a->diag[i];
2445c5546cabSHong Zhang       ui[i+1]  = ui[i] + ncols;
2446c5546cabSHong Zhang       udiag[i] = ui[i+1] - 1; /* points to the last entry of U(i,:) */
2447dc1e2a3fSHong Zhang     }
2448dc1e2a3fSHong Zhang     ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
2449dc1e2a3fSHong Zhang     cols = uj;
2450dc1e2a3fSHong Zhang     for (i=0; i<am; i++) {
24515f44c854SHong Zhang       aj    = a->j + a->diag[i] + 1; /* 1st entry of U(i,:) without diagonal */
2452dc1e2a3fSHong Zhang       ncols = ai[i+1] - a->diag[i] -1;
24535f44c854SHong Zhang       for (j=0; j<ncols; j++) *cols++ = aj[j];
24545f44c854SHong Zhang       *cols++ = i; /* diagoanl is located as the last entry of U(i,:) */
24555f44c854SHong Zhang     }
24565f44c854SHong Zhang   } else { /* case: levels>0 || (levels=0 && !perm_identity) */
24575f44c854SHong Zhang     ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
24585f44c854SHong Zhang     ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
24595f44c854SHong Zhang 
24605f44c854SHong Zhang     /* initialization */
24615f44c854SHong Zhang     ierr  = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr);
24625f44c854SHong Zhang 
24635f44c854SHong Zhang     /* jl: linked list for storing indices of the pivot rows
24645f44c854SHong Zhang        il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
24650e83c824SBarry Smith     ierr = PetscMalloc4(am,PetscInt*,&uj_ptr,am,PetscInt*,&uj_lvl_ptr,am,PetscInt,&jl,am,PetscInt,&il);CHKERRQ(ierr);
24665f44c854SHong Zhang     for (i=0; i<am; i++){
24675f44c854SHong Zhang       jl[i] = am; il[i] = 0;
24685f44c854SHong Zhang     }
24695f44c854SHong Zhang 
24705f44c854SHong Zhang     /* create and initialize a linked list for storing column indices of the active row k */
24715f44c854SHong Zhang     nlnk = am + 1;
24725f44c854SHong Zhang     ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
24735f44c854SHong Zhang 
247495b5ac22SHong Zhang     /* initial FreeSpace size is fill*(ai[am]+am)/2 */
247595b5ac22SHong Zhang     ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+am)/2),&free_space);CHKERRQ(ierr);
24765f44c854SHong Zhang     current_space = free_space;
247795b5ac22SHong Zhang     ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+am)/2),&free_space_lvl);CHKERRQ(ierr);
24785f44c854SHong Zhang     current_space_lvl = free_space_lvl;
24795f44c854SHong Zhang 
24805f44c854SHong Zhang     for (k=0; k<am; k++){  /* for each active row k */
24815f44c854SHong Zhang       /* initialize lnk by the column indices of row rip[k] of A */
24825f44c854SHong Zhang       nzk   = 0;
24835f44c854SHong Zhang       ncols = ai[rip[k]+1] - ai[rip[k]];
2484e32f2f54SBarry 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);
24855f44c854SHong Zhang       ncols_upper = 0;
24865f44c854SHong Zhang       for (j=0; j<ncols; j++){
24875f44c854SHong Zhang         i = *(aj + ai[rip[k]] + j); /* unpermuted column index */
24885f44c854SHong Zhang         if (riip[i] >= k){ /* only take upper triangular entry */
24895f44c854SHong Zhang           ajtmp[ncols_upper] = i;
24905f44c854SHong Zhang           ncols_upper++;
24915f44c854SHong Zhang         }
24925f44c854SHong Zhang       }
24935f44c854SHong Zhang       ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
24945f44c854SHong Zhang       nzk += nlnk;
24955f44c854SHong Zhang 
24965f44c854SHong Zhang       /* update lnk by computing fill-in for each pivot row to be merged in */
24975f44c854SHong Zhang       prow = jl[k]; /* 1st pivot row */
24985f44c854SHong Zhang 
24995f44c854SHong Zhang       while (prow < k){
25005f44c854SHong Zhang         nextprow = jl[prow];
25015f44c854SHong Zhang 
25025f44c854SHong Zhang         /* merge prow into k-th row */
25035f44c854SHong Zhang         jmin = il[prow] + 1;  /* index of the 2nd nzero entry in U(prow,k:am-1) */
25045f44c854SHong Zhang         jmax = ui[prow+1];
25055f44c854SHong Zhang         ncols = jmax-jmin;
25065f44c854SHong Zhang         i     = jmin - ui[prow];
25075f44c854SHong Zhang         cols  = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */
25085f44c854SHong Zhang         uj    = uj_lvl_ptr[prow] + i; /* levels of cols */
25095f44c854SHong Zhang         j     = *(uj - 1);
25105f44c854SHong Zhang         ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr);
25115f44c854SHong Zhang         nzk += nlnk;
25125f44c854SHong Zhang 
25135f44c854SHong Zhang         /* update il and jl for prow */
25145f44c854SHong Zhang         if (jmin < jmax){
25155f44c854SHong Zhang           il[prow] = jmin;
25165f44c854SHong Zhang           j = *cols; jl[prow] = jl[j]; jl[j] = prow;
25175f44c854SHong Zhang         }
25185f44c854SHong Zhang         prow = nextprow;
25195f44c854SHong Zhang       }
25205f44c854SHong Zhang 
25215f44c854SHong Zhang       /* if free space is not available, make more free space */
25225f44c854SHong Zhang       if (current_space->local_remaining<nzk) {
25235f44c854SHong Zhang         i  = am - k + 1; /* num of unfactored rows */
2524c6ae9bfcSHong Zhang         i *= PetscMin(nzk, i-1); /* i*nzk, i*(i-1): estimated and max additional space needed */
25255f44c854SHong Zhang         ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
25265f44c854SHong Zhang         ierr = PetscFreeSpaceGet(i,&current_space_lvl);CHKERRQ(ierr);
25275f44c854SHong Zhang         reallocs++;
25285f44c854SHong Zhang       }
25295f44c854SHong Zhang 
25305f44c854SHong Zhang       /* copy data into free_space and free_space_lvl, then initialize lnk */
2531e32f2f54SBarry Smith       if (nzk == 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k);
25325f44c854SHong Zhang       ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
25335f44c854SHong Zhang 
25345f44c854SHong Zhang       /* add the k-th row into il and jl */
25355f44c854SHong Zhang       if (nzk > 1){
25365f44c854SHong Zhang         i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
25375f44c854SHong Zhang         jl[k] = jl[i]; jl[i] = k;
25385f44c854SHong Zhang         il[k] = ui[k] + 1;
25395f44c854SHong Zhang       }
25405f44c854SHong Zhang       uj_ptr[k]     = current_space->array;
25415f44c854SHong Zhang       uj_lvl_ptr[k] = current_space_lvl->array;
25425f44c854SHong Zhang 
25435f44c854SHong Zhang       current_space->array           += nzk;
25445f44c854SHong Zhang       current_space->local_used      += nzk;
25455f44c854SHong Zhang       current_space->local_remaining -= nzk;
25465f44c854SHong Zhang 
25475f44c854SHong Zhang       current_space_lvl->array           += nzk;
25485f44c854SHong Zhang       current_space_lvl->local_used      += nzk;
25495f44c854SHong Zhang       current_space_lvl->local_remaining -= nzk;
25505f44c854SHong Zhang 
25515f44c854SHong Zhang       ui[k+1] = ui[k] + nzk;
25525f44c854SHong Zhang     }
25535f44c854SHong Zhang 
25545f44c854SHong Zhang     ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
25555f44c854SHong Zhang     ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
25560e83c824SBarry Smith     ierr = PetscFree4(uj_ptr,uj_lvl_ptr,jl,il);CHKERRQ(ierr);
25575f44c854SHong Zhang     ierr = PetscFree(ajtmp);CHKERRQ(ierr);
25585f44c854SHong Zhang 
25599263d837SHong Zhang     /* copy free_space into uj and free free_space; set ui, uj, udiag in new datastructure; */
25605f44c854SHong Zhang     ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
256135233d99SShri Abhyankar     ierr = PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag);CHKERRQ(ierr); /* store matrix factor  */
25625f44c854SHong Zhang     ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
25635f44c854SHong Zhang     ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
25645f44c854SHong Zhang 
25655f44c854SHong Zhang   } /* end of case: levels>0 || (levels=0 && !perm_identity) */
25665f44c854SHong Zhang 
25675f44c854SHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
25685f44c854SHong Zhang   b    = (Mat_SeqSBAIJ*)(fact)->data;
25695f44c854SHong Zhang   b->singlemalloc = PETSC_FALSE;
25705f44c854SHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr);
25715f44c854SHong Zhang   b->j    = uj;
25725f44c854SHong Zhang   b->i    = ui;
25735f44c854SHong Zhang   b->diag = udiag;
25747f53bb6cSHong Zhang   b->free_diag = PETSC_TRUE;
25755f44c854SHong Zhang   b->ilen = 0;
25765f44c854SHong Zhang   b->imax = 0;
25775f44c854SHong Zhang   b->row  = perm;
25785f44c854SHong Zhang   b->col  = perm;
25795f44c854SHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
25805f44c854SHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
25815f44c854SHong Zhang   b->icol = iperm;
25825f44c854SHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
25835f44c854SHong Zhang   ierr = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
2584d595f711SHong Zhang   ierr = PetscLogObjectMemory(fact,ui[am]*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
25855f44c854SHong Zhang   b->maxnz   = b->nz = ui[am];
25865f44c854SHong Zhang   b->free_a  = PETSC_TRUE;
25875f44c854SHong Zhang   b->free_ij = PETSC_TRUE;
25885f44c854SHong Zhang 
2589f284f12aSHong Zhang   fact->info.factor_mallocs   = reallocs;
2590f284f12aSHong Zhang   fact->info.fill_ratio_given = fill;
25915f44c854SHong Zhang   if (ai[am] != 0) {
25926a69fef8SHong Zhang     /* nonzeros in lower triangular part of A (including diagonals) = (ai[am]+am)/2 */
259395b5ac22SHong Zhang     fact->info.fill_ratio_needed = ((PetscReal)2*ui[am])/(ai[am]+am);
25945f44c854SHong Zhang   } else {
2595f284f12aSHong Zhang     fact->info.fill_ratio_needed = 0.0;
25965f44c854SHong Zhang   }
25979263d837SHong Zhang #if defined(PETSC_USE_INFO)
25989263d837SHong Zhang     if (ai[am] != 0) {
25999263d837SHong Zhang       PetscReal af = fact->info.fill_ratio_needed;
26009263d837SHong Zhang       ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr);
26019263d837SHong Zhang       ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
26029263d837SHong Zhang       ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr);
26039263d837SHong Zhang     } else {
26049263d837SHong Zhang       ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr);
26059263d837SHong Zhang     }
26069263d837SHong Zhang #endif
260735233d99SShri Abhyankar   fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ;
26085f44c854SHong Zhang   PetscFunctionReturn(0);
26095f44c854SHong Zhang }
26105f44c854SHong Zhang 
26115f44c854SHong Zhang #undef __FUNCT__
2612ad04f41aSHong Zhang #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ_inplace"
2613ad04f41aSHong Zhang PetscErrorCode MatICCFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
26145f44c854SHong Zhang {
26155f44c854SHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
26165f44c854SHong Zhang   Mat_SeqSBAIJ       *b;
26175f44c854SHong Zhang   PetscErrorCode     ierr;
2618ace3abfcSBarry Smith   PetscBool          perm_identity,missing;
26195f44c854SHong Zhang   PetscInt           reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag;
26205f44c854SHong Zhang   const PetscInt     *rip,*riip;
26215f44c854SHong Zhang   PetscInt           jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow;
26225f44c854SHong Zhang   PetscInt           nlnk,*lnk,*lnk_lvl=PETSC_NULL,d;
26235f44c854SHong Zhang   PetscInt           ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr;
26245f44c854SHong Zhang   PetscReal          fill=info->fill,levels=info->levels;
26255f44c854SHong Zhang   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
26265f44c854SHong Zhang   PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL;
26275f44c854SHong Zhang   PetscBT            lnkbt;
26285f44c854SHong Zhang   IS                 iperm;
26295f44c854SHong Zhang 
26305f44c854SHong Zhang   PetscFunctionBegin;
2631e32f2f54SBarry 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);
26325f44c854SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr);
2633e32f2f54SBarry Smith   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d);
26345f44c854SHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
26355f44c854SHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
26365f44c854SHong Zhang 
26375f44c854SHong Zhang   ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr);
26385f44c854SHong Zhang   ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr);
26395f44c854SHong Zhang   ui[0] = 0;
26405f44c854SHong Zhang 
26415f44c854SHong Zhang   /* ICC(0) without matrix ordering: simply copies fill pattern */
26425f44c854SHong Zhang   if (!levels && perm_identity) {
26435f44c854SHong Zhang 
26445f44c854SHong Zhang     for (i=0; i<am; i++) {
26455f44c854SHong Zhang       ui[i+1]  = ui[i] + ai[i+1] - a->diag[i];
26465f44c854SHong Zhang       udiag[i] = ui[i];
2647ed59401aSHong Zhang     }
264839e3d630SHong Zhang     ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
264939e3d630SHong Zhang     cols = uj;
2650ed59401aSHong Zhang     for (i=0; i<am; i++) {
2651ed59401aSHong Zhang       aj    = a->j + a->diag[i];
265239e3d630SHong Zhang       ncols = ui[i+1] - ui[i];
265339e3d630SHong Zhang       for (j=0; j<ncols; j++) *cols++ = *aj++;
2654ed59401aSHong Zhang     }
265539e3d630SHong Zhang   } else { /* case: levels>0 || (levels=0 && !perm_identity) */
2656b635d36bSHong Zhang     ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
2657b635d36bSHong Zhang     ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
2658b635d36bSHong Zhang 
26597a48dd6fSHong Zhang     /* initialization */
26605a8e39fbSHong Zhang     ierr  = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr);
26617a48dd6fSHong Zhang 
26627a48dd6fSHong Zhang     /* jl: linked list for storing indices of the pivot rows
26637a48dd6fSHong Zhang        il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
26640e83c824SBarry Smith     ierr = PetscMalloc4(am,PetscInt*,&uj_ptr,am,PetscInt*,&uj_lvl_ptr,am,PetscInt,&jl,am,PetscInt,&il);CHKERRQ(ierr);
26657a48dd6fSHong Zhang     for (i=0; i<am; i++){
26667a48dd6fSHong Zhang       jl[i] = am; il[i] = 0;
26677a48dd6fSHong Zhang     }
26687a48dd6fSHong Zhang 
26697a48dd6fSHong Zhang     /* create and initialize a linked list for storing column indices of the active row k */
26707a48dd6fSHong Zhang     nlnk = am + 1;
26712ed133dbSHong Zhang     ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
26727a48dd6fSHong Zhang 
26737a48dd6fSHong Zhang     /* initial FreeSpace size is fill*(ai[am]+1) */
2674a1a86e44SBarry Smith     ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr);
26757a48dd6fSHong Zhang     current_space = free_space;
2676a1a86e44SBarry Smith     ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space_lvl);CHKERRQ(ierr);
26777a48dd6fSHong Zhang     current_space_lvl = free_space_lvl;
26787a48dd6fSHong Zhang 
26797a48dd6fSHong Zhang     for (k=0; k<am; k++){  /* for each active row k */
26807a48dd6fSHong Zhang       /* initialize lnk by the column indices of row rip[k] of A */
26817a48dd6fSHong Zhang       nzk   = 0;
2682622e2028SHong Zhang       ncols = ai[rip[k]+1] - ai[rip[k]];
2683e32f2f54SBarry 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);
2684622e2028SHong Zhang       ncols_upper = 0;
2685622e2028SHong Zhang       for (j=0; j<ncols; j++){
2686b635d36bSHong Zhang         i = *(aj + ai[rip[k]] + j); /* unpermuted column index */
2687b635d36bSHong Zhang         if (riip[i] >= k){ /* only take upper triangular entry */
26885a8e39fbSHong Zhang           ajtmp[ncols_upper] = i;
2689622e2028SHong Zhang           ncols_upper++;
2690622e2028SHong Zhang         }
2691622e2028SHong Zhang       }
2692b635d36bSHong Zhang       ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
26937a48dd6fSHong Zhang       nzk += nlnk;
26947a48dd6fSHong Zhang 
26957a48dd6fSHong Zhang       /* update lnk by computing fill-in for each pivot row to be merged in */
26967a48dd6fSHong Zhang       prow = jl[k]; /* 1st pivot row */
26977a48dd6fSHong Zhang 
26987a48dd6fSHong Zhang       while (prow < k){
26997a48dd6fSHong Zhang         nextprow = jl[prow];
27007a48dd6fSHong Zhang 
27017a48dd6fSHong Zhang         /* merge prow into k-th row */
27027a48dd6fSHong Zhang         jmin = il[prow] + 1;  /* index of the 2nd nzero entry in U(prow,k:am-1) */
27037a48dd6fSHong Zhang         jmax = ui[prow+1];
27047a48dd6fSHong Zhang         ncols = jmax-jmin;
2705ed59401aSHong Zhang         i     = jmin - ui[prow];
2706ed59401aSHong Zhang         cols  = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */
27075a8e39fbSHong Zhang         uj    = uj_lvl_ptr[prow] + i; /* levels of cols */
27085a8e39fbSHong Zhang         j     = *(uj - 1);
27095a8e39fbSHong Zhang         ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr);
27107a48dd6fSHong Zhang         nzk += nlnk;
27117a48dd6fSHong Zhang 
27127a48dd6fSHong Zhang         /* update il and jl for prow */
27137a48dd6fSHong Zhang         if (jmin < jmax){
27147a48dd6fSHong Zhang           il[prow] = jmin;
27157a48dd6fSHong Zhang           j = *cols; jl[prow] = jl[j]; jl[j] = prow;
27167a48dd6fSHong Zhang         }
27177a48dd6fSHong Zhang         prow = nextprow;
27187a48dd6fSHong Zhang       }
27197a48dd6fSHong Zhang 
27207a48dd6fSHong Zhang       /* if free space is not available, make more free space */
27217a48dd6fSHong Zhang       if (current_space->local_remaining<nzk) {
27227a48dd6fSHong Zhang         i = am - k + 1; /* num of unfactored rows */
2723fd3f55cbSMatthew Knepley         i *= PetscMin(nzk, (i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */
2724a1a86e44SBarry Smith         ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
2725a1a86e44SBarry Smith         ierr = PetscFreeSpaceGet(i,&current_space_lvl);CHKERRQ(ierr);
27267a48dd6fSHong Zhang         reallocs++;
27277a48dd6fSHong Zhang       }
27287a48dd6fSHong Zhang 
27297a48dd6fSHong Zhang       /* copy data into free_space and free_space_lvl, then initialize lnk */
273065e19b50SBarry Smith       if (!nzk) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k);
27312ed133dbSHong Zhang       ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
27327a48dd6fSHong Zhang 
27337a48dd6fSHong Zhang       /* add the k-th row into il and jl */
273439e3d630SHong Zhang       if (nzk > 1){
27357a48dd6fSHong Zhang         i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
27367a48dd6fSHong Zhang         jl[k] = jl[i]; jl[i] = k;
27377a48dd6fSHong Zhang         il[k] = ui[k] + 1;
27387a48dd6fSHong Zhang       }
27397a48dd6fSHong Zhang       uj_ptr[k]     = current_space->array;
27407a48dd6fSHong Zhang       uj_lvl_ptr[k] = current_space_lvl->array;
27417a48dd6fSHong Zhang 
27427a48dd6fSHong Zhang       current_space->array           += nzk;
27437a48dd6fSHong Zhang       current_space->local_used      += nzk;
27447a48dd6fSHong Zhang       current_space->local_remaining -= nzk;
27457a48dd6fSHong Zhang 
27467a48dd6fSHong Zhang       current_space_lvl->array           += nzk;
27477a48dd6fSHong Zhang       current_space_lvl->local_used      += nzk;
27487a48dd6fSHong Zhang       current_space_lvl->local_remaining -= nzk;
27497a48dd6fSHong Zhang 
27507a48dd6fSHong Zhang       ui[k+1] = ui[k] + nzk;
27517a48dd6fSHong Zhang     }
27527a48dd6fSHong Zhang 
27536cf91177SBarry Smith #if defined(PETSC_USE_INFO)
27547a48dd6fSHong Zhang     if (ai[am] != 0) {
275539e3d630SHong Zhang       PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]);
2756ae15b995SBarry Smith       ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr);
2757ae15b995SBarry Smith       ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
2758ae15b995SBarry Smith       ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr);
27597a48dd6fSHong Zhang     } else {
2760ae15b995SBarry Smith       ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr);
27617a48dd6fSHong Zhang     }
276263ba0a88SBarry Smith #endif
27637a48dd6fSHong Zhang 
27647a48dd6fSHong Zhang     ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
2765b635d36bSHong Zhang     ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
27660e83c824SBarry Smith     ierr = PetscFree4(uj_ptr,uj_lvl_ptr,jl,il);CHKERRQ(ierr);
27675a8e39fbSHong Zhang     ierr = PetscFree(ajtmp);CHKERRQ(ierr);
27687a48dd6fSHong Zhang 
27697a48dd6fSHong Zhang     /* destroy list of free space and other temporary array(s) */
27707a48dd6fSHong Zhang     ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
2771a1a86e44SBarry Smith     ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr);
27722ed133dbSHong Zhang     ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
2773a1a86e44SBarry Smith     ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
27747a48dd6fSHong Zhang 
277539e3d630SHong Zhang   } /* end of case: levels>0 || (levels=0 && !perm_identity) */
277639e3d630SHong Zhang 
27777a48dd6fSHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
27787a48dd6fSHong Zhang 
2779f284f12aSHong Zhang   b    = (Mat_SeqSBAIJ*)fact->data;
27807a48dd6fSHong Zhang   b->singlemalloc = PETSC_FALSE;
27817a48dd6fSHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr);
27827a48dd6fSHong Zhang   b->j    = uj;
27837a48dd6fSHong Zhang   b->i    = ui;
27845f44c854SHong Zhang   b->diag = udiag;
27857f53bb6cSHong Zhang   b->free_diag = PETSC_TRUE;
27867a48dd6fSHong Zhang   b->ilen = 0;
27877a48dd6fSHong Zhang   b->imax = 0;
27887a48dd6fSHong Zhang   b->row  = perm;
2789b635d36bSHong Zhang   b->col  = perm;
2790b635d36bSHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
2791b635d36bSHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
2792b635d36bSHong Zhang   b->icol = iperm;
27937a48dd6fSHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
27947a48dd6fSHong Zhang   ierr    = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
2795f284f12aSHong Zhang   ierr = PetscLogObjectMemory(fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
27967a48dd6fSHong Zhang   b->maxnz   = b->nz = ui[am];
2797e6b907acSBarry Smith   b->free_a  = PETSC_TRUE;
2798e6b907acSBarry Smith   b->free_ij = PETSC_TRUE;
27997a48dd6fSHong Zhang 
2800f284f12aSHong Zhang   fact->info.factor_mallocs    = reallocs;
2801f284f12aSHong Zhang   fact->info.fill_ratio_given  = fill;
28027a48dd6fSHong Zhang   if (ai[am] != 0) {
2803f284f12aSHong Zhang     fact->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]);
28047a48dd6fSHong Zhang   } else {
2805f284f12aSHong Zhang     fact->info.fill_ratio_needed = 0.0;
28067a48dd6fSHong Zhang   }
28070a3c351aSHong Zhang   fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_inplace;
2808a6175056SHong Zhang   PetscFunctionReturn(0);
2809a6175056SHong Zhang }
2810d5d45c9bSBarry Smith 
2811177d4faaSHong Zhang #undef __FUNCT__
2812177d4faaSHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_SeqAIJ"
281335233d99SShri Abhyankar PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
28140be760fbSHong Zhang {
28150be760fbSHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
28160be760fbSHong Zhang   Mat_SeqSBAIJ       *b;
28170be760fbSHong Zhang   PetscErrorCode     ierr;
2818ace3abfcSBarry Smith   PetscBool          perm_identity;
28190be760fbSHong Zhang   PetscReal          fill = info->fill;
28200be760fbSHong Zhang   const PetscInt     *rip,*riip;
28210be760fbSHong Zhang   PetscInt           i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow;
28220be760fbSHong Zhang   PetscInt           *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow;
28230be760fbSHong Zhang   PetscInt           nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr,*udiag;
28240be760fbSHong Zhang   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
28250be760fbSHong Zhang   PetscBT            lnkbt;
28260be760fbSHong Zhang   IS                 iperm;
28270be760fbSHong Zhang 
28280be760fbSHong Zhang   PetscFunctionBegin;
2829e32f2f54SBarry 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);
28300be760fbSHong Zhang   /* check whether perm is the identity mapping */
28310be760fbSHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
28320be760fbSHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
28330be760fbSHong Zhang   ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
28340be760fbSHong Zhang   ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
28350be760fbSHong Zhang 
28360be760fbSHong Zhang   /* initialization */
28370be760fbSHong Zhang   ierr  = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr);
28380be760fbSHong Zhang   ierr  = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr);
28390be760fbSHong Zhang   ui[0] = 0;
28400be760fbSHong Zhang 
28410be760fbSHong Zhang   /* jl: linked list for storing indices of the pivot rows
28420be760fbSHong Zhang      il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
28430be760fbSHong Zhang   ierr = PetscMalloc4(am,PetscInt*,&ui_ptr,am,PetscInt,&jl,am,PetscInt,&il,am,PetscInt,&cols);CHKERRQ(ierr);
28440be760fbSHong Zhang   for (i=0; i<am; i++){
28450be760fbSHong Zhang     jl[i] = am; il[i] = 0;
28460be760fbSHong Zhang   }
28470be760fbSHong Zhang 
28480be760fbSHong Zhang   /* create and initialize a linked list for storing column indices of the active row k */
28490be760fbSHong Zhang   nlnk = am + 1;
28500be760fbSHong Zhang   ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
28510be760fbSHong Zhang 
285295b5ac22SHong Zhang   /* initial FreeSpace size is fill*(ai[am]+am)/2 */
285395b5ac22SHong Zhang   ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+am)/2),&free_space);CHKERRQ(ierr);
28540be760fbSHong Zhang   current_space = free_space;
28550be760fbSHong Zhang 
28560be760fbSHong Zhang   for (k=0; k<am; k++){  /* for each active row k */
28570be760fbSHong Zhang     /* initialize lnk by the column indices of row rip[k] of A */
28580be760fbSHong Zhang     nzk   = 0;
28590be760fbSHong Zhang     ncols = ai[rip[k]+1] - ai[rip[k]];
2860e32f2f54SBarry 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);
28610be760fbSHong Zhang     ncols_upper = 0;
28620be760fbSHong Zhang     for (j=0; j<ncols; j++){
28630be760fbSHong Zhang       i = riip[*(aj + ai[rip[k]] + j)];
28640be760fbSHong Zhang       if (i >= k){ /* only take upper triangular entry */
28650be760fbSHong Zhang         cols[ncols_upper] = i;
28660be760fbSHong Zhang         ncols_upper++;
28670be760fbSHong Zhang       }
28680be760fbSHong Zhang     }
28690be760fbSHong Zhang     ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
28700be760fbSHong Zhang     nzk += nlnk;
28710be760fbSHong Zhang 
28720be760fbSHong Zhang     /* update lnk by computing fill-in for each pivot row to be merged in */
28730be760fbSHong Zhang     prow = jl[k]; /* 1st pivot row */
28740be760fbSHong Zhang 
28750be760fbSHong Zhang     while (prow < k){
28760be760fbSHong Zhang       nextprow = jl[prow];
28770be760fbSHong Zhang       /* merge prow into k-th row */
28780be760fbSHong Zhang       jmin = il[prow] + 1;  /* index of the 2nd nzero entry in U(prow,k:am-1) */
28790be760fbSHong Zhang       jmax = ui[prow+1];
28800be760fbSHong Zhang       ncols = jmax-jmin;
28810be760fbSHong Zhang       uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */
28820be760fbSHong Zhang       ierr = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
28830be760fbSHong Zhang       nzk += nlnk;
28840be760fbSHong Zhang 
28850be760fbSHong Zhang       /* update il and jl for prow */
28860be760fbSHong Zhang       if (jmin < jmax){
28870be760fbSHong Zhang         il[prow] = jmin;
28880be760fbSHong Zhang         j = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow;
28890be760fbSHong Zhang       }
28900be760fbSHong Zhang       prow = nextprow;
28910be760fbSHong Zhang     }
28920be760fbSHong Zhang 
28930be760fbSHong Zhang     /* if free space is not available, make more free space */
28940be760fbSHong Zhang     if (current_space->local_remaining<nzk) {
28950be760fbSHong Zhang       i  = am - k + 1; /* num of unfactored rows */
2896c6ae9bfcSHong Zhang       i *= PetscMin(nzk,i-1); /* i*nzk, i*(i-1): estimated and max additional space needed */
28970be760fbSHong Zhang       ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
28980be760fbSHong Zhang       reallocs++;
28990be760fbSHong Zhang     }
29000be760fbSHong Zhang 
29010be760fbSHong Zhang     /* copy data into free space, then initialize lnk */
29020be760fbSHong Zhang     ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
29030be760fbSHong Zhang 
29040be760fbSHong Zhang     /* add the k-th row into il and jl */
29057b056e98SHong Zhang     if (nzk > 1){
29060be760fbSHong Zhang       i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
29070be760fbSHong Zhang       jl[k] = jl[i]; jl[i] = k;
29080be760fbSHong Zhang       il[k] = ui[k] + 1;
29090be760fbSHong Zhang     }
29100be760fbSHong Zhang     ui_ptr[k] = current_space->array;
29110be760fbSHong Zhang     current_space->array           += nzk;
29120be760fbSHong Zhang     current_space->local_used      += nzk;
29130be760fbSHong Zhang     current_space->local_remaining -= nzk;
29140be760fbSHong Zhang 
29150be760fbSHong Zhang     ui[k+1] = ui[k] + nzk;
29160be760fbSHong Zhang   }
29170be760fbSHong Zhang 
29180be760fbSHong Zhang   ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
29190be760fbSHong Zhang   ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
29200be760fbSHong Zhang   ierr = PetscFree4(ui_ptr,jl,il,cols);CHKERRQ(ierr);
29210be760fbSHong Zhang 
29229263d837SHong Zhang   /* copy free_space into uj and free free_space; set ui, uj, udiag in new datastructure; */
29230be760fbSHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
292435233d99SShri Abhyankar   ierr = PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag);CHKERRQ(ierr); /* store matrix factor */
29250be760fbSHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
29260be760fbSHong Zhang 
29270be760fbSHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
29280be760fbSHong Zhang 
2929f284f12aSHong Zhang   b = (Mat_SeqSBAIJ*)fact->data;
29300be760fbSHong Zhang   b->singlemalloc = PETSC_FALSE;
29310be760fbSHong Zhang   b->free_a       = PETSC_TRUE;
29320be760fbSHong Zhang   b->free_ij      = PETSC_TRUE;
29330be760fbSHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr);
29340be760fbSHong Zhang   b->j    = uj;
29350be760fbSHong Zhang   b->i    = ui;
29360be760fbSHong Zhang   b->diag = udiag;
29370be760fbSHong Zhang   b->free_diag = PETSC_TRUE;
29380be760fbSHong Zhang   b->ilen = 0;
29390be760fbSHong Zhang   b->imax = 0;
29400be760fbSHong Zhang   b->row  = perm;
29410be760fbSHong Zhang   b->col  = perm;
29420be760fbSHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
29430be760fbSHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
29440be760fbSHong Zhang   b->icol = iperm;
29450be760fbSHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
29460be760fbSHong Zhang   ierr    = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
29470860e08cSHong Zhang   ierr    = PetscLogObjectMemory(fact,ui[am]*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
29480be760fbSHong Zhang   b->maxnz = b->nz = ui[am];
29490be760fbSHong Zhang 
2950f284f12aSHong Zhang   fact->info.factor_mallocs    = reallocs;
2951f284f12aSHong Zhang   fact->info.fill_ratio_given  = fill;
29520be760fbSHong Zhang   if (ai[am] != 0) {
29536a69fef8SHong Zhang     /* nonzeros in lower triangular part of A (including diagonals) = (ai[am]+am)/2 */
295495b5ac22SHong Zhang     fact->info.fill_ratio_needed = ((PetscReal)2*ui[am])/(ai[am]+am);
29550be760fbSHong Zhang   } else {
2956f284f12aSHong Zhang     fact->info.fill_ratio_needed = 0.0;
29570be760fbSHong Zhang   }
29589263d837SHong Zhang #if defined(PETSC_USE_INFO)
29599263d837SHong Zhang   if (ai[am] != 0) {
29609263d837SHong Zhang     PetscReal af = fact->info.fill_ratio_needed;
29619263d837SHong Zhang     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr);
29629263d837SHong Zhang     ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
29639263d837SHong Zhang     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr);
29649263d837SHong Zhang   } else {
29659263d837SHong Zhang      ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr);
29669263d837SHong Zhang   }
29679263d837SHong Zhang #endif
296835233d99SShri Abhyankar   fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ;
29690be760fbSHong Zhang   PetscFunctionReturn(0);
29700be760fbSHong Zhang }
29710be760fbSHong Zhang 
2972f76d2b81SHong Zhang #undef __FUNCT__
2973ad04f41aSHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_SeqAIJ_inplace"
2974ad04f41aSHong Zhang PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ_inplace(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
2975f76d2b81SHong Zhang {
2976f76d2b81SHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
297710c27e3dSHong Zhang   Mat_SeqSBAIJ       *b;
2978dfbe8321SBarry Smith   PetscErrorCode     ierr;
2979ace3abfcSBarry Smith   PetscBool          perm_identity;
298010c27e3dSHong Zhang   PetscReal          fill = info->fill;
29815d0c19d7SBarry Smith   const PetscInt     *rip,*riip;
29825d0c19d7SBarry Smith   PetscInt           i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow;
298310c27e3dSHong Zhang   PetscInt           *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow;
29842ed133dbSHong Zhang   PetscInt           nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr;
2985a1a86e44SBarry Smith   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
298610c27e3dSHong Zhang   PetscBT            lnkbt;
29872ed133dbSHong Zhang   IS                 iperm;
2988f76d2b81SHong Zhang 
2989f76d2b81SHong Zhang   PetscFunctionBegin;
2990e32f2f54SBarry 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);
29912ed133dbSHong Zhang   /* check whether perm is the identity mapping */
2992f76d2b81SHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
29932ed133dbSHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
29942ed133dbSHong Zhang   ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
29959bfd6278SHong Zhang   ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
299610c27e3dSHong Zhang 
299710c27e3dSHong Zhang   /* initialization */
29981393bc97SHong Zhang   ierr  = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr);
299910c27e3dSHong Zhang   ui[0] = 0;
300010c27e3dSHong Zhang 
300110c27e3dSHong Zhang   /* jl: linked list for storing indices of the pivot rows
30021393bc97SHong Zhang      il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
30030e83c824SBarry Smith   ierr = PetscMalloc4(am,PetscInt*,&ui_ptr,am,PetscInt,&jl,am,PetscInt,&il,am,PetscInt,&cols);CHKERRQ(ierr);
30041393bc97SHong Zhang   for (i=0; i<am; i++){
30051393bc97SHong Zhang     jl[i] = am; il[i] = 0;
3006f76d2b81SHong Zhang   }
300710c27e3dSHong Zhang 
300810c27e3dSHong Zhang   /* create and initialize a linked list for storing column indices of the active row k */
30091393bc97SHong Zhang   nlnk = am + 1;
30101393bc97SHong Zhang   ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
301110c27e3dSHong Zhang 
30121393bc97SHong Zhang   /* initial FreeSpace size is fill*(ai[am]+1) */
3013a1a86e44SBarry Smith   ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr);
301410c27e3dSHong Zhang   current_space = free_space;
301510c27e3dSHong Zhang 
30161393bc97SHong Zhang   for (k=0; k<am; k++){  /* for each active row k */
301710c27e3dSHong Zhang     /* initialize lnk by the column indices of row rip[k] of A */
301810c27e3dSHong Zhang     nzk   = 0;
30192ed133dbSHong Zhang     ncols = ai[rip[k]+1] - ai[rip[k]];
3020e32f2f54SBarry 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);
30212ed133dbSHong Zhang     ncols_upper = 0;
30222ed133dbSHong Zhang     for (j=0; j<ncols; j++){
30239bfd6278SHong Zhang       i = riip[*(aj + ai[rip[k]] + j)];
30242ed133dbSHong Zhang       if (i >= k){ /* only take upper triangular entry */
30252ed133dbSHong Zhang         cols[ncols_upper] = i;
30262ed133dbSHong Zhang         ncols_upper++;
30272ed133dbSHong Zhang       }
30282ed133dbSHong Zhang     }
30291393bc97SHong Zhang     ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
303010c27e3dSHong Zhang     nzk += nlnk;
303110c27e3dSHong Zhang 
303210c27e3dSHong Zhang     /* update lnk by computing fill-in for each pivot row to be merged in */
303310c27e3dSHong Zhang     prow = jl[k]; /* 1st pivot row */
303410c27e3dSHong Zhang 
303510c27e3dSHong Zhang     while (prow < k){
303610c27e3dSHong Zhang       nextprow = jl[prow];
303710c27e3dSHong Zhang       /* merge prow into k-th row */
30381393bc97SHong Zhang       jmin = il[prow] + 1;  /* index of the 2nd nzero entry in U(prow,k:am-1) */
303910c27e3dSHong Zhang       jmax = ui[prow+1];
304010c27e3dSHong Zhang       ncols = jmax-jmin;
30411393bc97SHong Zhang       uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */
30425a8e39fbSHong Zhang       ierr = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
304310c27e3dSHong Zhang       nzk += nlnk;
304410c27e3dSHong Zhang 
304510c27e3dSHong Zhang       /* update il and jl for prow */
304610c27e3dSHong Zhang       if (jmin < jmax){
304710c27e3dSHong Zhang         il[prow] = jmin;
30482ed133dbSHong Zhang         j = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow;
304910c27e3dSHong Zhang       }
305010c27e3dSHong Zhang       prow = nextprow;
305110c27e3dSHong Zhang     }
305210c27e3dSHong Zhang 
305310c27e3dSHong Zhang     /* if free space is not available, make more free space */
305410c27e3dSHong Zhang     if (current_space->local_remaining<nzk) {
30551393bc97SHong Zhang       i = am - k + 1; /* num of unfactored rows */
305610c27e3dSHong Zhang       i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */
3057a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
305810c27e3dSHong Zhang       reallocs++;
305910c27e3dSHong Zhang     }
306010c27e3dSHong Zhang 
306110c27e3dSHong Zhang     /* copy data into free space, then initialize lnk */
30621393bc97SHong Zhang     ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
306310c27e3dSHong Zhang 
306410c27e3dSHong Zhang     /* add the k-th row into il and jl */
306510c27e3dSHong Zhang     if (nzk-1 > 0){
30661393bc97SHong Zhang       i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
306710c27e3dSHong Zhang       jl[k] = jl[i]; jl[i] = k;
306810c27e3dSHong Zhang       il[k] = ui[k] + 1;
306910c27e3dSHong Zhang     }
30702ed133dbSHong Zhang     ui_ptr[k] = current_space->array;
307110c27e3dSHong Zhang     current_space->array           += nzk;
307210c27e3dSHong Zhang     current_space->local_used      += nzk;
307310c27e3dSHong Zhang     current_space->local_remaining -= nzk;
307410c27e3dSHong Zhang 
307510c27e3dSHong Zhang     ui[k+1] = ui[k] + nzk;
307610c27e3dSHong Zhang   }
307710c27e3dSHong Zhang 
30786cf91177SBarry Smith #if defined(PETSC_USE_INFO)
30791393bc97SHong Zhang   if (ai[am] != 0) {
30801393bc97SHong Zhang     PetscReal af = (PetscReal)(ui[am])/((PetscReal)ai[am]);
3081ae15b995SBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr);
3082ae15b995SBarry Smith     ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
3083ae15b995SBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr);
308410c27e3dSHong Zhang   } else {
3085ae15b995SBarry Smith      ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr);
308610c27e3dSHong Zhang   }
308763ba0a88SBarry Smith #endif
308810c27e3dSHong Zhang 
308910c27e3dSHong Zhang   ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
30909bfd6278SHong Zhang   ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
30910e83c824SBarry Smith   ierr = PetscFree4(ui_ptr,jl,il,cols);CHKERRQ(ierr);
309210c27e3dSHong Zhang 
309310c27e3dSHong Zhang   /* destroy list of free space and other temporary array(s) */
30941393bc97SHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
3095a1a86e44SBarry Smith   ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr);
309610c27e3dSHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
309710c27e3dSHong Zhang 
309810c27e3dSHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
309910c27e3dSHong Zhang 
3100f284f12aSHong Zhang   b = (Mat_SeqSBAIJ*)fact->data;
310110c27e3dSHong Zhang   b->singlemalloc = PETSC_FALSE;
3102e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
3103e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
31041393bc97SHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr);
310510c27e3dSHong Zhang   b->j    = uj;
310610c27e3dSHong Zhang   b->i    = ui;
310710c27e3dSHong Zhang   b->diag = 0;
310810c27e3dSHong Zhang   b->ilen = 0;
310910c27e3dSHong Zhang   b->imax = 0;
311010c27e3dSHong Zhang   b->row  = perm;
31119bfd6278SHong Zhang   b->col  = perm;
31129bfd6278SHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
31139bfd6278SHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
31149bfd6278SHong Zhang   b->icol = iperm;
311510c27e3dSHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
31161393bc97SHong Zhang   ierr    = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
3117719d5645SBarry Smith   ierr    = PetscLogObjectMemory(fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
31181393bc97SHong Zhang   b->maxnz = b->nz = ui[am];
311910c27e3dSHong Zhang 
3120f284f12aSHong Zhang   fact->info.factor_mallocs    = reallocs;
3121f284f12aSHong Zhang   fact->info.fill_ratio_given  = fill;
31221393bc97SHong Zhang   if (ai[am] != 0) {
3123f284f12aSHong Zhang     fact->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]);
312410c27e3dSHong Zhang   } else {
3125f284f12aSHong Zhang     fact->info.fill_ratio_needed = 0.0;
312610c27e3dSHong Zhang   }
31270a3c351aSHong Zhang   fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_inplace;
3128f76d2b81SHong Zhang   PetscFunctionReturn(0);
3129f76d2b81SHong Zhang }
3130599ef60dSHong Zhang 
31311d098868SHong Zhang #undef __FUNCT__
313235233d99SShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering"
313335233d99SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering(Mat A,Vec bb,Vec xx)
3134813a1f2bSShri Abhyankar {
3135813a1f2bSShri Abhyankar   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
3136813a1f2bSShri Abhyankar   PetscErrorCode    ierr;
3137813a1f2bSShri Abhyankar   PetscInt          n = A->rmap->n;
3138813a1f2bSShri Abhyankar   const PetscInt    *ai = a->i,*aj = a->j,*adiag = a->diag,*vi;
3139813a1f2bSShri Abhyankar   PetscScalar       *x,sum;
3140813a1f2bSShri Abhyankar   const PetscScalar *b;
3141813a1f2bSShri Abhyankar   const MatScalar   *aa = a->a,*v;
3142813a1f2bSShri Abhyankar   PetscInt          i,nz;
3143813a1f2bSShri Abhyankar 
3144813a1f2bSShri Abhyankar   PetscFunctionBegin;
3145813a1f2bSShri Abhyankar   if (!n) PetscFunctionReturn(0);
3146813a1f2bSShri Abhyankar 
31473649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
3148813a1f2bSShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
3149813a1f2bSShri Abhyankar 
3150813a1f2bSShri Abhyankar   /* forward solve the lower triangular */
3151813a1f2bSShri Abhyankar   x[0] = b[0];
3152813a1f2bSShri Abhyankar   v    = aa;
3153813a1f2bSShri Abhyankar   vi   = aj;
3154813a1f2bSShri Abhyankar   for (i=1; i<n; i++) {
3155813a1f2bSShri Abhyankar     nz  = ai[i+1] - ai[i];
3156813a1f2bSShri Abhyankar     sum = b[i];
3157813a1f2bSShri Abhyankar     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
3158813a1f2bSShri Abhyankar     v  += nz;
3159813a1f2bSShri Abhyankar     vi += nz;
3160813a1f2bSShri Abhyankar     x[i] = sum;
3161813a1f2bSShri Abhyankar   }
3162813a1f2bSShri Abhyankar 
3163813a1f2bSShri Abhyankar   /* backward solve the upper triangular */
316462fbd6afSShri Abhyankar   for (i=n-1; i>=0; i--){
31653c0119dfSShri Abhyankar     v   = aa + adiag[i+1] + 1;
31663c0119dfSShri Abhyankar     vi  = aj + adiag[i+1] + 1;
3167813a1f2bSShri Abhyankar     nz = adiag[i] - adiag[i+1]-1;
3168813a1f2bSShri Abhyankar     sum = x[i];
3169813a1f2bSShri Abhyankar     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
31703c0119dfSShri Abhyankar     x[i] = sum*v[nz]; /* x[i]=aa[adiag[i]]*sum; v++; */
3171813a1f2bSShri Abhyankar   }
3172813a1f2bSShri Abhyankar 
3173813a1f2bSShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
31743649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
3175813a1f2bSShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
3176813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
3177813a1f2bSShri Abhyankar }
3178813a1f2bSShri Abhyankar 
31791d098868SHong Zhang #undef __FUNCT__
318035233d99SShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ"
318135233d99SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ(Mat A,Vec bb,Vec xx)
3182f268cba8SShri Abhyankar {
3183f268cba8SShri Abhyankar   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
3184f268cba8SShri Abhyankar   IS                iscol = a->col,isrow = a->row;
3185f268cba8SShri Abhyankar   PetscErrorCode    ierr;
3186f268cba8SShri Abhyankar   PetscInt          i,n=A->rmap->n,*vi,*ai=a->i,*aj=a->j,*adiag = a->diag,nz;
3187f268cba8SShri Abhyankar   const PetscInt    *rout,*cout,*r,*c;
3188f268cba8SShri Abhyankar   PetscScalar       *x,*tmp,sum;
3189f268cba8SShri Abhyankar   const PetscScalar *b;
3190f268cba8SShri Abhyankar   const MatScalar   *aa = a->a,*v;
3191f268cba8SShri Abhyankar 
3192f268cba8SShri Abhyankar   PetscFunctionBegin;
3193f268cba8SShri Abhyankar   if (!n) PetscFunctionReturn(0);
3194f268cba8SShri Abhyankar 
31953649974fSBarry Smith   ierr = VecGetArrayRead(bb,&b);CHKERRQ(ierr);
3196f268cba8SShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
3197f268cba8SShri Abhyankar   tmp  = a->solve_work;
3198f268cba8SShri Abhyankar 
3199f268cba8SShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
3200f268cba8SShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
3201f268cba8SShri Abhyankar 
3202f268cba8SShri Abhyankar   /* forward solve the lower triangular */
3203f268cba8SShri Abhyankar   tmp[0] = b[r[0]];
3204f268cba8SShri Abhyankar   v      = aa;
3205f268cba8SShri Abhyankar   vi     = aj;
3206f268cba8SShri Abhyankar   for (i=1; i<n; i++) {
3207f268cba8SShri Abhyankar     nz  = ai[i+1] - ai[i];
3208f268cba8SShri Abhyankar     sum = b[r[i]];
3209f268cba8SShri Abhyankar     PetscSparseDenseMinusDot(sum,tmp,v,vi,nz);
3210f268cba8SShri Abhyankar     tmp[i] = sum;
3211f268cba8SShri Abhyankar     v += nz; vi += nz;
3212f268cba8SShri Abhyankar   }
3213f268cba8SShri Abhyankar 
3214f268cba8SShri Abhyankar   /* backward solve the upper triangular */
3215f268cba8SShri Abhyankar   for (i=n-1; i>=0; i--){
32163c0119dfSShri Abhyankar     v   = aa + adiag[i+1]+1;
32173c0119dfSShri Abhyankar     vi  = aj + adiag[i+1]+1;
3218f268cba8SShri Abhyankar     nz  = adiag[i]-adiag[i+1]-1;
3219f268cba8SShri Abhyankar     sum = tmp[i];
3220f268cba8SShri Abhyankar     PetscSparseDenseMinusDot(sum,tmp,v,vi,nz);
3221f268cba8SShri Abhyankar     x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */
3222f268cba8SShri Abhyankar   }
3223f268cba8SShri Abhyankar 
3224f268cba8SShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
3225f268cba8SShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
32263649974fSBarry Smith   ierr = VecRestoreArrayRead(bb,&b);CHKERRQ(ierr);
3227f268cba8SShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
3228f268cba8SShri Abhyankar   ierr = PetscLogFlops(2*a->nz - A->cmap->n);CHKERRQ(ierr);
3229f268cba8SShri Abhyankar   PetscFunctionReturn(0);
3230f268cba8SShri Abhyankar }
3231f268cba8SShri Abhyankar 
3232f268cba8SShri Abhyankar #undef __FUNCT__
32331d098868SHong Zhang #define __FUNCT__ "MatILUDTFactor_SeqAIJ"
3234fe97e370SBarry Smith /*
3235fe97e370SBarry 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
3236fe97e370SBarry Smith */
32371d098868SHong Zhang PetscErrorCode MatILUDTFactor_SeqAIJ(Mat A,IS isrow,IS iscol,const MatFactorInfo *info,Mat *fact)
32381d098868SHong Zhang {
32391d098868SHong Zhang   Mat                B = *fact;
3240599ef60dSHong Zhang   Mat_SeqAIJ         *a=(Mat_SeqAIJ*)A->data,*b;
3241599ef60dSHong Zhang   IS                 isicol;
3242599ef60dSHong Zhang   PetscErrorCode     ierr;
32431d098868SHong Zhang   const PetscInt     *r,*ic;
32441d098868SHong Zhang   PetscInt           i,n=A->rmap->n,*ai=a->i,*aj=a->j,*ajtmp,*adiag;
3245dc3a2fd3SHong Zhang   PetscInt           *bi,*bj,*bdiag,*bdiag_rev;
3246f61a948aSLisandro Dalcin   PetscInt           row,nzi,nzi_bl,nzi_bu,*im,nzi_al,nzi_au;
32471d098868SHong Zhang   PetscInt           nlnk,*lnk;
32481d098868SHong Zhang   PetscBT            lnkbt;
3249ace3abfcSBarry Smith   PetscBool          row_identity,icol_identity;
32508aee2decSHong Zhang   MatScalar          *aatmp,*pv,*batmp,*ba,*rtmp,*pc,multiplier,*vtmp,diag_tmp;
32511d098868SHong Zhang   const PetscInt     *ics;
32521d098868SHong Zhang   PetscInt           j,nz,*pj,*bjtmp,k,ncut,*jtmp;
3253182b8fbaSHong Zhang   PetscReal          dt=info->dt,dtcol=info->dtcol,shift=info->shiftamount;
3254f61a948aSLisandro Dalcin   PetscInt           dtcount=(PetscInt)info->dtcount,nnz_max;
3255ace3abfcSBarry Smith   PetscBool          missing;
3256599ef60dSHong Zhang 
3257599ef60dSHong Zhang   PetscFunctionBegin;
3258f61a948aSLisandro Dalcin 
3259f61a948aSLisandro Dalcin   if (dt      == PETSC_DEFAULT) dt      = 0.005;
3260f61a948aSLisandro Dalcin   if (dtcol   == PETSC_DEFAULT) dtcol   = 0.01; /* XXX unused! */
3261f61a948aSLisandro Dalcin   if (dtcount == PETSC_DEFAULT) dtcount = (PetscInt)(1.5*a->rmax);
3262f61a948aSLisandro Dalcin 
32631d098868SHong Zhang   /* ------- symbolic factorization, can be reused ---------*/
32641d098868SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr);
3265e32f2f54SBarry Smith   if (missing) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i);
32661d098868SHong Zhang   adiag=a->diag;
3267599ef60dSHong Zhang 
3268599ef60dSHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
3269599ef60dSHong Zhang 
3270599ef60dSHong Zhang   /* bdiag is location of diagonal in factor */
32711bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);     /* becomes b->diag */
32721bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag_rev);CHKERRQ(ierr); /* temporary */
3273599ef60dSHong Zhang 
32741d098868SHong Zhang   /* allocate row pointers bi */
32758fc3a347SHong Zhang   ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
32761d098868SHong Zhang 
32771d098868SHong Zhang   /* allocate bj and ba; max num of nonzero entries is (ai[n]+2*n*dtcount+2) */
3278393d3a68SHong Zhang   if (dtcount > n-1) dtcount = n-1; /* diagonal is excluded */
32791d098868SHong Zhang   nnz_max  = ai[n]+2*n*dtcount+2;
32808fc3a347SHong Zhang 
32818fc3a347SHong Zhang   ierr = PetscMalloc((nnz_max+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
32828fc3a347SHong Zhang   ierr = PetscMalloc((nnz_max+1)*sizeof(MatScalar),&ba);CHKERRQ(ierr);
3283599ef60dSHong Zhang 
3284599ef60dSHong Zhang   /* put together the new matrix */
3285599ef60dSHong Zhang   ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
3286599ef60dSHong Zhang   ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr);
3287f284f12aSHong Zhang   b    = (Mat_SeqAIJ*)B->data;
3288599ef60dSHong Zhang   b->free_a       = PETSC_TRUE;
3289599ef60dSHong Zhang   b->free_ij      = PETSC_TRUE;
3290599ef60dSHong Zhang   b->singlemalloc = PETSC_FALSE;
3291599ef60dSHong Zhang   b->a          = ba;
3292599ef60dSHong Zhang   b->j          = bj;
3293599ef60dSHong Zhang   b->i          = bi;
3294599ef60dSHong Zhang   b->diag       = bdiag;
3295599ef60dSHong Zhang   b->ilen       = 0;
3296599ef60dSHong Zhang   b->imax       = 0;
3297599ef60dSHong Zhang   b->row        = isrow;
3298599ef60dSHong Zhang   b->col        = iscol;
3299599ef60dSHong Zhang   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
3300599ef60dSHong Zhang   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
3301599ef60dSHong Zhang   b->icol       = isicol;
3302599ef60dSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
3303599ef60dSHong Zhang 
33041d098868SHong Zhang   ierr = PetscLogObjectMemory(B,nnz_max*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
33051d098868SHong Zhang   b->maxnz = nnz_max;
3306599ef60dSHong Zhang 
3307d5f3da31SBarry Smith   B->factortype            = MAT_FACTOR_ILUDT;
3308f284f12aSHong Zhang   B->info.factor_mallocs   = 0;
3309f284f12aSHong Zhang   B->info.fill_ratio_given = ((PetscReal)nnz_max)/((PetscReal)ai[n]);
33101d098868SHong Zhang   CHKMEMQ;
33111d098868SHong Zhang   /* ------- end of symbolic factorization ---------*/
3312599ef60dSHong Zhang 
3313599ef60dSHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
3314599ef60dSHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
3315599ef60dSHong Zhang   ics  = ic;
3316599ef60dSHong Zhang 
3317599ef60dSHong Zhang   /* linked list for storing column indices of the active row */
3318599ef60dSHong Zhang   nlnk = n + 1;
3319599ef60dSHong Zhang   ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
33201d098868SHong Zhang 
33211d098868SHong Zhang   /* im: used by PetscLLAddSortedLU(); jtmp: working array for column indices of active row */
33220e83c824SBarry Smith   ierr = PetscMalloc2(n,PetscInt,&im,n,PetscInt,&jtmp);CHKERRQ(ierr);
33231d098868SHong Zhang   /* rtmp, vtmp: working arrays for sparse and contiguous row entries of active row */
33240e83c824SBarry Smith   ierr = PetscMalloc2(n,MatScalar,&rtmp,n,MatScalar,&vtmp);CHKERRQ(ierr);
33250e83c824SBarry Smith   ierr = PetscMemzero(rtmp,n*sizeof(MatScalar));CHKERRQ(ierr);
3326599ef60dSHong Zhang 
3327599ef60dSHong Zhang   bi[0]    = 0;
33288fc3a347SHong Zhang   bdiag[0] = nnz_max-1; /* location of diag[0] in factor B */
3329dc3a2fd3SHong Zhang   bdiag_rev[n] = bdiag[0];
33308fc3a347SHong Zhang   bi[2*n+1] = bdiag[0]+1; /* endof bj and ba array */
3331599ef60dSHong Zhang   for (i=0; i<n; i++) {
3332599ef60dSHong Zhang     /* copy initial fill into linked list */
3333599ef60dSHong Zhang     nzi = 0; /* nonzeros for active row i */
33348369b28dSHong Zhang     nzi = ai[r[i]+1] - ai[r[i]];
3335e32f2f54SBarry 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);
33368369b28dSHong Zhang     nzi_al = adiag[r[i]] - ai[r[i]];
33378369b28dSHong Zhang     nzi_au = ai[r[i]+1] - adiag[r[i]] -1;
3338599ef60dSHong Zhang     ajtmp = aj + ai[r[i]];
33398369b28dSHong Zhang     ierr = PetscLLAddPerm(nzi,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
3340599ef60dSHong Zhang 
3341599ef60dSHong Zhang     /* load in initial (unfactored row) */
33421d098868SHong Zhang     aatmp = a->a + ai[r[i]];
33438369b28dSHong Zhang     for (j=0; j<nzi; j++) {
33441d098868SHong Zhang       rtmp[ics[*ajtmp++]] = *aatmp++;
3345599ef60dSHong Zhang     }
3346599ef60dSHong Zhang 
3347599ef60dSHong Zhang     /* add pivot rows into linked list */
3348599ef60dSHong Zhang     row = lnk[n];
3349599ef60dSHong Zhang     while (row < i ) {
33501d098868SHong Zhang       nzi_bl = bi[row+1] - bi[row] + 1;
33511d098868SHong Zhang       bjtmp = bj + bdiag[row+1]+1; /* points to 1st column next to the diagonal in U */
33521d098868SHong Zhang       ierr  = PetscLLAddSortedLU(bjtmp,row,nlnk,lnk,lnkbt,i,nzi_bl,im);CHKERRQ(ierr);
3353599ef60dSHong Zhang       nzi  += nlnk;
3354599ef60dSHong Zhang       row   = lnk[row];
3355599ef60dSHong Zhang     }
3356599ef60dSHong Zhang 
33578369b28dSHong Zhang     /* copy data from lnk into jtmp, then initialize lnk */
33588369b28dSHong Zhang     ierr = PetscLLClean(n,n,nzi,lnk,jtmp,lnkbt);CHKERRQ(ierr);
3359599ef60dSHong Zhang 
3360599ef60dSHong Zhang     /* numerical factorization */
33618369b28dSHong Zhang     bjtmp = jtmp;
3362599ef60dSHong Zhang     row   = *bjtmp++; /* 1st pivot row */
3363599ef60dSHong Zhang     while  ( row < i ) {
3364599ef60dSHong Zhang       pc         = rtmp + row;
33653c2a7987SHong Zhang       pv         = ba + bdiag[row]; /* 1./(diag of the pivot row) */
33663c2a7987SHong Zhang       multiplier = (*pc) * (*pv);
3367599ef60dSHong Zhang       *pc        = multiplier;
33683c2a7987SHong Zhang       if (PetscAbsScalar(*pc) > dt){ /* apply tolerance dropping rule */
33691d098868SHong Zhang         pj         = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */
33701d098868SHong Zhang         pv         = ba + bdiag[row+1] + 1;
33711d098868SHong Zhang         /* if (multiplier < -1.0 or multiplier >1.0) printf("row/prow %d, %d, multiplier %g\n",i,row,multiplier); */
33721d098868SHong Zhang         nz         = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */
33731d098868SHong Zhang         for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++);
337428b1a77fSLisandro Dalcin         ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr);
3375599ef60dSHong Zhang       }
3376599ef60dSHong Zhang       row = *bjtmp++;
3377599ef60dSHong Zhang     }
3378599ef60dSHong Zhang 
33798369b28dSHong Zhang     /* copy sparse rtmp into contiguous vtmp; separate L and U part */
33808aee2decSHong Zhang     diag_tmp = rtmp[i];  /* save diagonal value - may not needed?? */
33818369b28dSHong Zhang     nzi_bl = 0; j = 0;
33828369b28dSHong Zhang     while (jtmp[j] < i){ /* Note: jtmp is sorted */
33838aee2decSHong Zhang       vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0;
33848369b28dSHong Zhang       nzi_bl++; j++;
33858369b28dSHong Zhang     }
33868369b28dSHong Zhang     nzi_bu = nzi - nzi_bl -1;
33878369b28dSHong Zhang     while (j < nzi){
33888aee2decSHong Zhang       vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0;
33898369b28dSHong Zhang       j++;
33908369b28dSHong Zhang     }
33918369b28dSHong Zhang 
3392599ef60dSHong Zhang     bjtmp = bj + bi[i];
3393599ef60dSHong Zhang     batmp = ba + bi[i];
33948369b28dSHong Zhang     /* apply level dropping rule to L part */
33958369b28dSHong Zhang     ncut = nzi_al + dtcount;
33968369b28dSHong Zhang     if (ncut < nzi_bl){
33978369b28dSHong Zhang       ierr = PetscSortSplit(ncut,nzi_bl,vtmp,jtmp);CHKERRQ(ierr);
33988369b28dSHong Zhang       ierr = PetscSortIntWithScalarArray(ncut,jtmp,vtmp);CHKERRQ(ierr);
3399599ef60dSHong Zhang     } else {
34008369b28dSHong Zhang       ncut = nzi_bl;
34018369b28dSHong Zhang     }
34028369b28dSHong Zhang     for (j=0; j<ncut; j++){
34038369b28dSHong Zhang       bjtmp[j] = jtmp[j];
34048369b28dSHong Zhang       batmp[j] = vtmp[j];
34056da515a1SHong Zhang       /* printf(" (%d,%g),",bjtmp[j],batmp[j]); */
34068369b28dSHong Zhang     }
34071d098868SHong Zhang     bi[i+1] = bi[i] + ncut;
34088369b28dSHong Zhang     nzi = ncut + 1;
34098369b28dSHong Zhang 
34108369b28dSHong Zhang     /* apply level dropping rule to U part */
34118369b28dSHong Zhang     ncut = nzi_au + dtcount;
34128369b28dSHong Zhang     if (ncut < nzi_bu){
34138369b28dSHong Zhang       ierr = PetscSortSplit(ncut,nzi_bu,vtmp+nzi_bl+1,jtmp+nzi_bl+1);CHKERRQ(ierr);
34148369b28dSHong Zhang       ierr = PetscSortIntWithScalarArray(ncut,jtmp+nzi_bl+1,vtmp+nzi_bl+1);CHKERRQ(ierr);
34158369b28dSHong Zhang     } else {
34168369b28dSHong Zhang       ncut = nzi_bu;
34178369b28dSHong Zhang     }
34188369b28dSHong Zhang     nzi += ncut;
34191d098868SHong Zhang 
34201d098868SHong Zhang     /* mark bdiagonal */
34211d098868SHong Zhang     bdiag[i+1]       = bdiag[i] - (ncut + 1);
3422dc3a2fd3SHong Zhang     bdiag_rev[n-i-1] = bdiag[i+1];
34238fc3a347SHong Zhang     bi[2*n - i]      = bi[2*n - i +1] - (ncut + 1);
34241d098868SHong Zhang     bjtmp = bj + bdiag[i];
34251d098868SHong Zhang     batmp = ba + bdiag[i];
34261d098868SHong Zhang     *bjtmp = i;
34278aee2decSHong Zhang     *batmp = diag_tmp; /* rtmp[i]; */
3428c9c72f8fSHong Zhang     if (*batmp == 0.0) {
3429c9c72f8fSHong Zhang       *batmp = dt+shift;
3430bd1bc851SBarry Smith       /* printf(" row %d add shift %g\n",i,shift); */
3431c9c72f8fSHong Zhang     }
3432b78a477dSHong Zhang     *batmp = 1.0/(*batmp); /* invert diagonal entries for simplier triangular solves */
34336da515a1SHong Zhang     /* printf(" (%d,%g),",*bjtmp,*batmp); */
34341d098868SHong Zhang 
34351d098868SHong Zhang     bjtmp = bj + bdiag[i+1]+1;
34361d098868SHong Zhang     batmp = ba + bdiag[i+1]+1;
34378369b28dSHong Zhang     for (k=0; k<ncut; k++){
34381d098868SHong Zhang       bjtmp[k] = jtmp[nzi_bl+1+k];
34391d098868SHong Zhang       batmp[k] = vtmp[nzi_bl+1+k];
34406da515a1SHong Zhang       /* printf(" (%d,%g),",bjtmp[k],batmp[k]); */
3441599ef60dSHong Zhang     }
34426da515a1SHong Zhang     /* printf("\n"); */
3443599ef60dSHong Zhang 
3444599ef60dSHong Zhang     im[i]   = nzi; /* used by PetscLLAddSortedLU() */
3445599ef60dSHong Zhang     /*
34461d098868SHong Zhang     printf("row %d: bi %d, bdiag %d\n",i,bi[i],bdiag[i]);
34478369b28dSHong Zhang     printf(" ----------------------------\n");
3448599ef60dSHong Zhang     */
34498369b28dSHong Zhang   } /* for (i=0; i<n; i++) */
34501d098868SHong Zhang   /* printf("end of L %d, beginning of U %d\n",bi[n],bdiag[n]); */
3451e32f2f54SBarry 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]);
3452599ef60dSHong Zhang 
3453599ef60dSHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
3454599ef60dSHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
3455599ef60dSHong Zhang 
3456599ef60dSHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
34570e83c824SBarry Smith   ierr = PetscFree2(im,jtmp);CHKERRQ(ierr);
34580e83c824SBarry Smith   ierr = PetscFree2(rtmp,vtmp);CHKERRQ(ierr);
34591bfa06eaSJed Brown   ierr = PetscFree(bdiag_rev);CHKERRQ(ierr);
3460599ef60dSHong Zhang 
3461599ef60dSHong Zhang   ierr = PetscLogFlops(B->cmap->n);CHKERRQ(ierr);
34621d098868SHong Zhang   b->maxnz = b->nz = bi[n] + bdiag[0] - bdiag[n];
3463599ef60dSHong Zhang 
3464599ef60dSHong Zhang   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
3465599ef60dSHong Zhang   ierr = ISIdentity(isicol,&icol_identity);CHKERRQ(ierr);
3466599ef60dSHong Zhang   if (row_identity && icol_identity) {
346735233d99SShri Abhyankar     B->ops->solve = MatSolve_SeqAIJ_NaturalOrdering;
3468599ef60dSHong Zhang   } else {
346935233d99SShri Abhyankar     B->ops->solve = MatSolve_SeqAIJ;
3470599ef60dSHong Zhang   }
3471599ef60dSHong Zhang 
34721d098868SHong Zhang   B->ops->solveadd          = 0;
34731d098868SHong Zhang   B->ops->solvetranspose    = 0;
34741d098868SHong Zhang   B->ops->solvetransposeadd = 0;
34751d098868SHong Zhang   B->ops->matsolve          = 0;
3476599ef60dSHong Zhang   B->assembled              = PETSC_TRUE;
3477599ef60dSHong Zhang   B->preallocated           = PETSC_TRUE;
3478599ef60dSHong Zhang   PetscFunctionReturn(0);
3479599ef60dSHong Zhang }
3480599ef60dSHong Zhang 
34813c2a7987SHong Zhang /* a wraper of MatILUDTFactor_SeqAIJ() */
34823c2a7987SHong Zhang #undef __FUNCT__
34833c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorSymbolic_SeqAIJ"
3484fe97e370SBarry Smith /*
3485fe97e370SBarry 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
3486fe97e370SBarry Smith */
3487fe97e370SBarry Smith 
34887087cfbeSBarry Smith PetscErrorCode  MatILUDTFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS row,IS col,const MatFactorInfo *info)
34893c2a7987SHong Zhang {
34903c2a7987SHong Zhang   PetscErrorCode     ierr;
34913c2a7987SHong Zhang 
34923c2a7987SHong Zhang   PetscFunctionBegin;
34933c2a7987SHong Zhang   ierr = MatILUDTFactor_SeqAIJ(A,row,col,info,&fact);CHKERRQ(ierr);
34943c2a7987SHong Zhang   PetscFunctionReturn(0);
34953c2a7987SHong Zhang }
34963c2a7987SHong Zhang 
34973c2a7987SHong Zhang /*
34983c2a7987SHong Zhang    same as MatLUFactorNumeric_SeqAIJ(), except using contiguous array matrix factors
34993c2a7987SHong Zhang    - intend to replace existing MatLUFactorNumeric_SeqAIJ()
35003c2a7987SHong Zhang */
35013c2a7987SHong Zhang #undef __FUNCT__
35023c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorNumeric_SeqAIJ"
3503fe97e370SBarry Smith /*
3504fe97e370SBarry 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
3505fe97e370SBarry Smith */
3506fe97e370SBarry Smith 
35077087cfbeSBarry Smith PetscErrorCode  MatILUDTFactorNumeric_SeqAIJ(Mat fact,Mat A,const MatFactorInfo *info)
35083c2a7987SHong Zhang {
35093c2a7987SHong Zhang   Mat            C=fact;
35103c2a7987SHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data;
35113c2a7987SHong Zhang   IS             isrow = b->row,isicol = b->icol;
35123c2a7987SHong Zhang   PetscErrorCode ierr;
35133c2a7987SHong Zhang   const PetscInt *r,*ic,*ics;
3514b78a477dSHong Zhang   PetscInt       i,j,k,n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j;
3515b78a477dSHong Zhang   PetscInt       *ajtmp,*bjtmp,nz,nzl,nzu,row,*bdiag = b->diag,*pj;
35163c2a7987SHong Zhang   MatScalar      *rtmp,*pc,multiplier,*v,*pv,*aa=a->a;
3517182b8fbaSHong Zhang   PetscReal      dt=info->dt,shift=info->shiftamount;
3518ace3abfcSBarry Smith   PetscBool      row_identity, col_identity;
35193c2a7987SHong Zhang 
35203c2a7987SHong Zhang   PetscFunctionBegin;
35213c2a7987SHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
35223c2a7987SHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
3523b78a477dSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr);
35243c2a7987SHong Zhang   ics  = ic;
35253c2a7987SHong Zhang 
35263c2a7987SHong Zhang   for (i=0; i<n; i++){
3527b78a477dSHong Zhang     /* initialize rtmp array */
3528b78a477dSHong Zhang     nzl   = bi[i+1] - bi[i];       /* num of nozeros in L(i,:) */
35293c2a7987SHong Zhang     bjtmp = bj + bi[i];
3530b78a477dSHong Zhang     for  (j=0; j<nzl; j++) rtmp[*bjtmp++] = 0.0;
3531b78a477dSHong Zhang     rtmp[i] = 0.0;
3532b78a477dSHong Zhang     nzu   = bdiag[i] - bdiag[i+1]; /* num of nozeros in U(i,:) */
3533b78a477dSHong Zhang     bjtmp = bj + bdiag[i+1] + 1;
3534b78a477dSHong Zhang     for  (j=0; j<nzu; j++) rtmp[*bjtmp++] = 0.0;
35353c2a7987SHong Zhang 
3536b78a477dSHong Zhang     /* load in initial unfactored row of A */
35376da515a1SHong Zhang     /* printf("row %d\n",i); */
35383c2a7987SHong Zhang     nz    = ai[r[i]+1] - ai[r[i]];
35393c2a7987SHong Zhang     ajtmp = aj + ai[r[i]];
35403c2a7987SHong Zhang     v     = aa + ai[r[i]];
35413c2a7987SHong Zhang     for (j=0; j<nz; j++) {
3542b78a477dSHong Zhang       rtmp[ics[*ajtmp++]] = v[j];
35436da515a1SHong Zhang       /* printf(" (%d,%g),",ics[ajtmp[j]],rtmp[ics[ajtmp[j]]]); */
35443c2a7987SHong Zhang     }
35456da515a1SHong Zhang     /* printf("\n"); */
35463c2a7987SHong Zhang 
3547b78a477dSHong Zhang     /* numerical factorization */
3548b78a477dSHong Zhang     bjtmp = bj + bi[i]; /* point to 1st entry of L(i,:) */
3549b78a477dSHong Zhang     nzl   = bi[i+1] - bi[i]; /* num of entries in L(i,:) */
3550b78a477dSHong Zhang     k = 0;
3551b78a477dSHong Zhang     while (k < nzl){
35523c2a7987SHong Zhang       row   = *bjtmp++;
35536da515a1SHong Zhang       /* printf("  prow %d\n",row); */
35543c2a7987SHong Zhang       pc         = rtmp + row;
3555b78a477dSHong Zhang       pv         = b->a + bdiag[row]; /* 1./(diag of the pivot row) */
3556b78a477dSHong Zhang       multiplier = (*pc) * (*pv);
35573c2a7987SHong Zhang       *pc        = multiplier;
3558b78a477dSHong Zhang       if (PetscAbsScalar(multiplier) > dt){
3559b78a477dSHong Zhang         pj         = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */
3560b78a477dSHong Zhang         pv         = b->a + bdiag[row+1] + 1;
3561b78a477dSHong Zhang         nz         = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */
3562b78a477dSHong Zhang         for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++);
356328b1a77fSLisandro Dalcin         ierr = PetscLogFlops(1+2*nz);CHKERRQ(ierr);
35643c2a7987SHong Zhang       }
3565b78a477dSHong Zhang       k++;
35663c2a7987SHong Zhang     }
35673c2a7987SHong Zhang 
3568b78a477dSHong Zhang     /* finished row so stick it into b->a */
3569b78a477dSHong Zhang     /* L-part */
3570b78a477dSHong Zhang     pv = b->a + bi[i] ;
3571b78a477dSHong Zhang     pj = bj + bi[i] ;
3572b78a477dSHong Zhang     nzl = bi[i+1] - bi[i];
3573b78a477dSHong Zhang     for (j=0; j<nzl; j++) {
3574b78a477dSHong Zhang       pv[j] = rtmp[pj[j]];
35756da515a1SHong Zhang       /* printf(" (%d,%g),",pj[j],pv[j]); */
35763c2a7987SHong Zhang     }
3577b78a477dSHong Zhang 
3578b78a477dSHong Zhang     /* diagonal: invert diagonal entries for simplier triangular solves */
3579b78a477dSHong Zhang     if (rtmp[i] == 0.0) rtmp[i] = dt+shift;
3580b78a477dSHong Zhang     b->a[bdiag[i]] = 1.0/rtmp[i];
35816da515a1SHong Zhang     /* printf(" (%d,%g),",i,b->a[bdiag[i]]); */
3582b78a477dSHong Zhang 
3583b78a477dSHong Zhang     /* U-part */
3584b78a477dSHong Zhang     pv = b->a + bdiag[i+1] + 1;
3585b78a477dSHong Zhang     pj = bj + bdiag[i+1] + 1;
3586b78a477dSHong Zhang     nzu = bdiag[i] - bdiag[i+1] - 1;
3587b78a477dSHong Zhang     for (j=0; j<nzu; j++) {
3588b78a477dSHong Zhang       pv[j] = rtmp[pj[j]];
35896da515a1SHong Zhang       /* printf(" (%d,%g),",pj[j],pv[j]); */
3590b78a477dSHong Zhang     }
35916da515a1SHong Zhang     /* printf("\n"); */
3592b78a477dSHong Zhang   }
3593b78a477dSHong Zhang 
35943c2a7987SHong Zhang   ierr = PetscFree(rtmp);CHKERRQ(ierr);
35953c2a7987SHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
35963c2a7987SHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
3597b78a477dSHong Zhang 
35983c2a7987SHong Zhang   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
35993c2a7987SHong Zhang   ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr);
36003c2a7987SHong Zhang   if (row_identity && col_identity) {
360135233d99SShri Abhyankar     C->ops->solve   = MatSolve_SeqAIJ_NaturalOrdering;
36023c2a7987SHong Zhang   } else {
360335233d99SShri Abhyankar     C->ops->solve   = MatSolve_SeqAIJ;
36043c2a7987SHong Zhang   }
3605b78a477dSHong Zhang   C->ops->solveadd           = 0;
3606b78a477dSHong Zhang   C->ops->solvetranspose     = 0;
3607b78a477dSHong Zhang   C->ops->solvetransposeadd  = 0;
3608b78a477dSHong Zhang   C->ops->matsolve           = 0;
36093c2a7987SHong Zhang   C->assembled    = PETSC_TRUE;
36103c2a7987SHong Zhang   C->preallocated = PETSC_TRUE;
36113c2a7987SHong Zhang   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
36123c2a7987SHong Zhang   PetscFunctionReturn(0);
36133c2a7987SHong Zhang }
3614