xref: /petsc/src/mat/impls/aij/seq/aijfact.c (revision 8a76ed4f3fc051f86a2481820ea0cb0a957103d2)
1be1d678aSKris Buschelman #define PETSCMAT_DLL
2289bc588SBarry Smith 
3b377110cSBarry Smith 
47c4f633dSBarry Smith #include "../src/mat/impls/aij/seq/aij.h"
5783ef271SHong Zhang #include "../src/mat/impls/sbaij/seq/sbaij.h"
61393bc97SHong Zhang #include "petscbt.h"
77c4f633dSBarry Smith #include "../src/mat/utils/freespace.h"
83b2fbd54SBarry Smith 
9f6a26d55SMatthew Knepley EXTERN_C_BEGIN
104a2ae208SSatish Balay #undef __FUNCT__
114a2ae208SSatish Balay #define __FUNCT__ "MatOrdering_Flow_SeqAIJ"
128fa24674SBarry Smith /*
138fa24674SBarry Smith       Computes an ordering to get most of the large numerical values in the lower triangular part of the matrix
148fa24674SBarry Smith */
15dfbe8321SBarry Smith PetscErrorCode MatOrdering_Flow_SeqAIJ(Mat mat,const MatOrderingType type,IS *irow,IS *icol)
163b2fbd54SBarry Smith {
178fa24674SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)mat->data;
188fa24674SBarry Smith   PetscErrorCode    ierr;
19889a8dedSBarry Smith   PetscInt          i,j,jj,k, kk,n = mat->rmap->n, current = 0, newcurrent = 0,*order;
208fa24674SBarry Smith   const PetscInt    *ai = a->i, *aj = a->j;
218fa24674SBarry Smith   const PetscScalar *aa = a->a;
228fa24674SBarry Smith   PetscTruth        *done;
23889a8dedSBarry Smith   PetscReal         best,past = 0,future;
243a40ed3dSBarry Smith 
258fa24674SBarry Smith   PetscFunctionBegin;
268fa24674SBarry Smith   /* pick initial row */
278fa24674SBarry Smith   best = -1;
288fa24674SBarry Smith   for (i=0; i<n; i++) {
298fa24674SBarry Smith     future = 0;
308fa24674SBarry Smith     for (j=ai[i]; j<ai[i+1]; j++) {
318fa24674SBarry Smith       if (aj[j] != i) future  += PetscAbsScalar(aa[j]); else past = PetscAbsScalar(aa[j]);
328fa24674SBarry Smith     }
338fa24674SBarry Smith     if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */
348fa24674SBarry Smith     if (past/future > best) {
358fa24674SBarry Smith       best = past/future;
368fa24674SBarry Smith       current = i;
378fa24674SBarry Smith     }
388fa24674SBarry Smith   }
398fa24674SBarry Smith 
408fa24674SBarry Smith   ierr = PetscMalloc(n*sizeof(PetscTruth),&done);CHKERRQ(ierr);
418fa24674SBarry Smith   ierr = PetscMalloc(n*sizeof(PetscInt),&order);CHKERRQ(ierr);
428fa24674SBarry Smith   ierr = PetscMemzero(done,n*sizeof(PetscTruth));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;
688fa24674SBarry Smith         future = 0;
698fa24674SBarry Smith         past   = 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     }
828fa24674SBarry Smith     if (current == newcurrent) SETERRQ(PETSC_ERR_PLIB,"newcurrent cannot be current");
838fa24674SBarry Smith     current = newcurrent;
848fa24674SBarry Smith     order[i+1] = current;
858fa24674SBarry Smith   }
868fa24674SBarry Smith   ierr = ISCreateGeneral(PETSC_COMM_SELF,n,order,irow);CHKERRQ(ierr);
878fa24674SBarry Smith   *icol = *irow;
888fa24674SBarry Smith   ierr = PetscObjectReference((PetscObject)*irow);CHKERRQ(ierr);
898fa24674SBarry Smith   ierr = PetscFree(done);CHKERRQ(ierr);
908fa24674SBarry Smith   ierr = PetscFree(order);CHKERRQ(ierr);
91d64ed03dSBarry Smith   PetscFunctionReturn(0);
923b2fbd54SBarry Smith }
93f6a26d55SMatthew Knepley EXTERN_C_END
943b2fbd54SBarry Smith 
95e631078cSBarry Smith EXTERN_C_BEGIN
964a2ae208SSatish Balay #undef __FUNCT__
97db4efbfdSBarry Smith #define __FUNCT__ "MatGetFactorAvailable_seqaij_petsc"
98db4efbfdSBarry Smith PetscErrorCode MatGetFactorAvailable_seqaij_petsc(Mat A,MatFactorType ftype,PetscTruth *flg)
99db4efbfdSBarry Smith {
100db4efbfdSBarry Smith   PetscFunctionBegin;
101db4efbfdSBarry Smith   *flg = PETSC_TRUE;
102db4efbfdSBarry Smith   PetscFunctionReturn(0);
103db4efbfdSBarry Smith }
104db4efbfdSBarry Smith EXTERN_C_END
105db4efbfdSBarry Smith 
106db4efbfdSBarry Smith EXTERN_C_BEGIN
107db4efbfdSBarry Smith #undef __FUNCT__
108b24902e0SBarry Smith #define __FUNCT__ "MatGetFactor_seqaij_petsc"
109b24902e0SBarry Smith PetscErrorCode MatGetFactor_seqaij_petsc(Mat A,MatFactorType ftype,Mat *B)
110b24902e0SBarry Smith {
111d0f46423SBarry Smith   PetscInt           n = A->rmap->n;
112b24902e0SBarry Smith   PetscErrorCode     ierr;
113b24902e0SBarry Smith 
114b24902e0SBarry Smith   PetscFunctionBegin;
115b24902e0SBarry Smith   ierr = MatCreate(((PetscObject)A)->comm,B);CHKERRQ(ierr);
116b24902e0SBarry Smith   ierr = MatSetSizes(*B,n,n,n,n);CHKERRQ(ierr);
117599ef60dSHong Zhang   if (ftype == MAT_FACTOR_LU || ftype == MAT_FACTOR_ILU || ftype == MAT_FACTOR_ILUDT){
118b24902e0SBarry Smith     ierr = MatSetType(*B,MATSEQAIJ);CHKERRQ(ierr);
119db4efbfdSBarry Smith     (*B)->ops->ilufactorsymbolic = MatILUFactorSymbolic_SeqAIJ;
12035bd34faSBarry Smith     (*B)->ops->lufactorsymbolic  = MatLUFactorSymbolic_SeqAIJ;
1211da05e5aSHong Zhang     (*B)->ops->iludtfactor       = MatILUDTFactor_SeqAIJ;
122b24902e0SBarry Smith   } else if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) {
123b24902e0SBarry Smith     ierr = MatSetType(*B,MATSEQSBAIJ);CHKERRQ(ierr);
124b24902e0SBarry Smith     ierr = MatSeqSBAIJSetPreallocation(*B,1,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
1255c9eb25fSBarry Smith     (*B)->ops->iccfactorsymbolic      = MatICCFactorSymbolic_SeqAIJ;
1265c9eb25fSBarry Smith     (*B)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqAIJ;
127b24902e0SBarry Smith   } else SETERRQ(PETSC_ERR_SUP,"Factor type not supported");
128db4efbfdSBarry Smith   (*B)->factor = ftype;
129b24902e0SBarry Smith   PetscFunctionReturn(0);
130b24902e0SBarry Smith }
131e631078cSBarry Smith EXTERN_C_END
132b24902e0SBarry Smith 
133b24902e0SBarry Smith #undef __FUNCT__
134b9617806SBarry Smith #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ"
1350481f469SBarry Smith PetscErrorCode MatLUFactorSymbolic_SeqAIJ(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
136289bc588SBarry Smith {
137416022c9SBarry Smith   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
138289bc588SBarry Smith   IS                 isicol;
1396849ba73SBarry Smith   PetscErrorCode     ierr;
1405d0c19d7SBarry Smith   const PetscInt     *r,*ic;
1415d0c19d7SBarry Smith   PetscInt           i,n=A->rmap->n,*ai=a->i,*aj=a->j;
1421393bc97SHong Zhang   PetscInt           *bi,*bj,*ajtmp;
1431393bc97SHong Zhang   PetscInt           *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im;
1449e046878SBarry Smith   PetscReal          f;
1454875ba38SHong Zhang   PetscInt           nlnk,*lnk,k,**bi_ptr;
146a1a86e44SBarry Smith   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
1471393bc97SHong Zhang   PetscBT            lnkbt;
148ce3d78c0SShri Abhyankar   PetscTruth         newdatastruct=PETSC_FALSE;
1499b48462bSShri Abhyankar   PetscTruth         newdatastruct_v2 = PETSC_FALSE;
150289bc588SBarry Smith 
1513a40ed3dSBarry Smith   PetscFunctionBegin;
152ce3d78c0SShri Abhyankar   ierr = PetscOptionsGetTruth(PETSC_NULL,"-lu_new",&newdatastruct,PETSC_NULL);CHKERRQ(ierr);
153ce3d78c0SShri Abhyankar   if(newdatastruct){
154ce3d78c0SShri Abhyankar     ierr = MatLUFactorSymbolic_SeqAIJ_newdatastruct(B,A,isrow,iscol,info);CHKERRQ(ierr);
155ce3d78c0SShri Abhyankar     PetscFunctionReturn(0);
156ce3d78c0SShri Abhyankar   }
1579b48462bSShri Abhyankar   ierr = PetscOptionsGetTruth(PETSC_NULL,"-lu_new_v2",&newdatastruct_v2,PETSC_NULL);CHKERRQ(ierr);
1589b48462bSShri Abhyankar   if(newdatastruct_v2){
1599b48462bSShri Abhyankar     ierr = MatLUFactorSymbolic_SeqAIJ_newdatastruct_v2(B,A,isrow,iscol,info);CHKERRQ(ierr);
1609b48462bSShri Abhyankar     PetscFunctionReturn(0);
1619b48462bSShri Abhyankar   }
1629b48462bSShri Abhyankar 
163ce3d78c0SShri Abhyankar 
164d0f46423SBarry Smith   if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_ERR_ARG_WRONG,"matrix must be square");
1654c49b128SBarry Smith   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
166ac121b3dSBarry Smith   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
167ac121b3dSBarry Smith   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
168289bc588SBarry Smith 
169289bc588SBarry Smith   /* get new row pointers */
1701393bc97SHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
1711393bc97SHong Zhang   bi[0] = 0;
1721393bc97SHong Zhang 
1731393bc97SHong Zhang   /* bdiag is location of diagonal in factor */
1741393bc97SHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
1751393bc97SHong Zhang   bdiag[0] = 0;
1761393bc97SHong Zhang 
1771393bc97SHong Zhang   /* linked list for storing column indices of the active row */
1781393bc97SHong Zhang   nlnk = n + 1;
1791393bc97SHong Zhang   ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
1801393bc97SHong Zhang 
18135baf27eSSatish Balay   ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr);
1821393bc97SHong Zhang 
1831393bc97SHong Zhang   /* initial FreeSpace size is f*(ai[n]+1) */
184d3d32019SBarry Smith   f = info->fill;
185a1a86e44SBarry Smith   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr);
1861393bc97SHong Zhang   current_space = free_space;
187289bc588SBarry Smith 
188289bc588SBarry Smith   for (i=0; i<n; i++) {
1891393bc97SHong Zhang     /* copy previous fill into linked list */
190289bc588SBarry Smith     nzi = 0;
1911393bc97SHong Zhang     nnz = ai[r[i]+1] - ai[r[i]];
1923b4a8b6dSBarry Smith     if (!nnz) SETERRQ2(PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i);
1931393bc97SHong Zhang     ajtmp = aj + ai[r[i]];
194afcc9904SHong Zhang     ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
1951393bc97SHong Zhang     nzi += nlnk;
1961393bc97SHong Zhang 
1971393bc97SHong Zhang     /* add pivot rows into linked list */
1981393bc97SHong Zhang     row = lnk[n];
1991393bc97SHong Zhang     while (row < i) {
200d1170560SHong Zhang       nzbd    = bdiag[row] - bi[row] + 1; /* num of entries in the row with column index <= row */
201d1170560SHong Zhang       ajtmp   = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */
202d1170560SHong Zhang       ierr = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr);
2031393bc97SHong Zhang       nzi += nlnk;
2041393bc97SHong Zhang       row  = lnk[row];
205289bc588SBarry Smith     }
2061393bc97SHong Zhang     bi[i+1] = bi[i] + nzi;
2071393bc97SHong Zhang     im[i]   = nzi;
2081393bc97SHong Zhang 
2091393bc97SHong Zhang     /* mark bdiag */
2101393bc97SHong Zhang     nzbd = 0;
2111393bc97SHong Zhang     nnz  = nzi;
2121393bc97SHong Zhang     k    = lnk[n];
2131393bc97SHong Zhang     while (nnz-- && k < i){
2141393bc97SHong Zhang       nzbd++;
2151393bc97SHong Zhang       k = lnk[k];
2161393bc97SHong Zhang     }
2171393bc97SHong Zhang     bdiag[i] = bi[i] + nzbd;
2181393bc97SHong Zhang 
2191393bc97SHong Zhang     /* if free space is not available, make more free space */
2201393bc97SHong Zhang     if (current_space->local_remaining<nzi) {
2211393bc97SHong Zhang       nnz = (n - i)*nzi; /* estimated and max additional space needed */
222a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
2231393bc97SHong Zhang       reallocs++;
2241393bc97SHong Zhang     }
2251393bc97SHong Zhang 
2261393bc97SHong Zhang     /* copy data into free space, then initialize lnk */
2271393bc97SHong Zhang     ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
2281393bc97SHong Zhang     bi_ptr[i] = current_space->array;
2291393bc97SHong Zhang     current_space->array           += nzi;
2301393bc97SHong Zhang     current_space->local_used      += nzi;
2311393bc97SHong Zhang     current_space->local_remaining -= nzi;
232289bc588SBarry Smith   }
2336cf91177SBarry Smith #if defined(PETSC_USE_INFO)
234464e8d44SSatish Balay   if (ai[n] != 0) {
2351393bc97SHong Zhang     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
236ae15b995SBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr);
237ae15b995SBarry Smith     ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
238ae15b995SBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr);
239ae15b995SBarry Smith     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
24005bf559cSSatish Balay   } else {
241ae15b995SBarry Smith     ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr);
24205bf559cSSatish Balay   }
24363ba0a88SBarry Smith #endif
2442fb3ed76SBarry Smith 
245898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
246898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
2471987afe7SBarry Smith 
2481393bc97SHong Zhang   /* destroy list of free space and other temporary array(s) */
2491393bc97SHong Zhang   ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
250a1a86e44SBarry Smith   ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr);
2511393bc97SHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
25235baf27eSSatish Balay   ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr);
253289bc588SBarry Smith 
254289bc588SBarry Smith   /* put together the new matrix */
255719d5645SBarry Smith   ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
256719d5645SBarry Smith   ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr);
257719d5645SBarry Smith   b    = (Mat_SeqAIJ*)(B)->data;
258e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
259e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
2607c922b88SBarry Smith   b->singlemalloc = PETSC_FALSE;
2611393bc97SHong Zhang   ierr          = PetscMalloc((bi[n]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);
2621393bc97SHong Zhang   b->j          = bj;
2631393bc97SHong Zhang   b->i          = bi;
2641393bc97SHong Zhang   b->diag       = bdiag;
265416022c9SBarry Smith   b->ilen       = 0;
266416022c9SBarry Smith   b->imax       = 0;
267416022c9SBarry Smith   b->row        = isrow;
268416022c9SBarry Smith   b->col        = iscol;
269c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
270c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
27182bf6240SBarry Smith   b->icol       = isicol;
27287828ca2SBarry Smith   ierr          = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
2731393bc97SHong Zhang 
2741393bc97SHong Zhang   /* In b structure:  Free imax, ilen, old a, old j.  Allocate solve_work, new a, new j */
275719d5645SBarry Smith   ierr = PetscLogObjectMemory(B,(bi[n]-n)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
2761393bc97SHong Zhang   b->maxnz = b->nz = bi[n] ;
2777fd98bd6SLois Curfman McInnes 
278719d5645SBarry Smith   (B)->factor                = MAT_FACTOR_LU;
279719d5645SBarry Smith   (B)->info.factor_mallocs   = reallocs;
280719d5645SBarry Smith   (B)->info.fill_ratio_given = f;
281703deb49SSatish Balay 
2829f7d0b68SHong Zhang   if (ai[n]) {
283719d5645SBarry Smith     (B)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]);
284eea2913aSSatish Balay   } else {
285719d5645SBarry Smith     (B)->info.fill_ratio_needed = 0.0;
286eea2913aSSatish Balay   }
287719d5645SBarry Smith   (B)->ops->lufactornumeric  = MatLUFactorNumeric_SeqAIJ;
288719d5645SBarry Smith   (B)->ops->solve            = MatSolve_SeqAIJ;
289719d5645SBarry Smith   (B)->ops->solvetranspose   = MatSolveTranspose_SeqAIJ;
290db4efbfdSBarry Smith   /* switch to inodes if appropriate */
2914108e4d5SBarry Smith   ierr = MatLUFactorSymbolic_SeqAIJ_Inode(B,A,isrow,iscol,info);CHKERRQ(ierr);
2923a40ed3dSBarry Smith   PetscFunctionReturn(0);
293289bc588SBarry Smith }
2941393bc97SHong Zhang 
295ce3d78c0SShri Abhyankar #undef __FUNCT__
296ce3d78c0SShri Abhyankar #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ_newdatastruct"
297ce3d78c0SShri Abhyankar PetscErrorCode MatLUFactorSymbolic_SeqAIJ_newdatastruct(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
298ce3d78c0SShri Abhyankar {
299ce3d78c0SShri Abhyankar   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
300ce3d78c0SShri Abhyankar   IS                 isicol;
301ce3d78c0SShri Abhyankar   PetscErrorCode     ierr;
302ce3d78c0SShri Abhyankar   const PetscInt     *r,*ic;
303ce3d78c0SShri Abhyankar   PetscInt           i,n=A->rmap->n,*ai=a->i,*aj=a->j;
304ce3d78c0SShri Abhyankar   PetscInt           *bi,*bj,*ajtmp;
305ce3d78c0SShri Abhyankar   PetscInt           *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im;
306ce3d78c0SShri Abhyankar   PetscReal          f;
307ce3d78c0SShri Abhyankar   PetscInt           nlnk,*lnk,k,**bi_ptr;
308ce3d78c0SShri Abhyankar   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
309ce3d78c0SShri Abhyankar   PetscBT            lnkbt;
310ce3d78c0SShri Abhyankar 
311ce3d78c0SShri Abhyankar   PetscFunctionBegin;
312ce3d78c0SShri Abhyankar   if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_ERR_ARG_WRONG,"matrix must be square");
313ce3d78c0SShri Abhyankar   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
314ce3d78c0SShri Abhyankar   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
315ce3d78c0SShri Abhyankar   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
316ce3d78c0SShri Abhyankar 
317ce3d78c0SShri Abhyankar   /* get new row pointers */
318ce3d78c0SShri Abhyankar   ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
319ce3d78c0SShri Abhyankar   bi[0] = 0;
320ce3d78c0SShri Abhyankar 
321ce3d78c0SShri Abhyankar   /* bdiag is location of diagonal in factor */
322ce3d78c0SShri Abhyankar   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
323ce3d78c0SShri Abhyankar   bdiag[0] = 0;
324ce3d78c0SShri Abhyankar 
325ce3d78c0SShri Abhyankar   /* linked list for storing column indices of the active row */
326ce3d78c0SShri Abhyankar   nlnk = n + 1;
327ce3d78c0SShri Abhyankar   ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
328ce3d78c0SShri Abhyankar 
329ce3d78c0SShri Abhyankar   ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr);
330ce3d78c0SShri Abhyankar 
331ce3d78c0SShri Abhyankar   /* initial FreeSpace size is f*(ai[n]+1) */
332ce3d78c0SShri Abhyankar   f = info->fill;
333ce3d78c0SShri Abhyankar   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr);
334ce3d78c0SShri Abhyankar   current_space = free_space;
335ce3d78c0SShri Abhyankar 
336ce3d78c0SShri Abhyankar   for (i=0; i<n; i++) {
337ce3d78c0SShri Abhyankar     /* copy previous fill into linked list */
338ce3d78c0SShri Abhyankar     nzi = 0;
339ce3d78c0SShri Abhyankar     nnz = ai[r[i]+1] - ai[r[i]];
340ce3d78c0SShri Abhyankar     if (!nnz) SETERRQ2(PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i);
341ce3d78c0SShri Abhyankar     ajtmp = aj + ai[r[i]];
342ce3d78c0SShri Abhyankar     ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
343ce3d78c0SShri Abhyankar     nzi += nlnk;
344ce3d78c0SShri Abhyankar 
345ce3d78c0SShri Abhyankar     /* add pivot rows into linked list */
346ce3d78c0SShri Abhyankar     row = lnk[n];
347ce3d78c0SShri Abhyankar     while (row < i){
34870319c84SHong Zhang       nzbd  = bdiag[row] + 1; /* num of entries in the row with column index <= row */
349ce3d78c0SShri Abhyankar       ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */
350ce3d78c0SShri Abhyankar       ierr  = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr);
351ce3d78c0SShri Abhyankar       nzi  += nlnk;
352ce3d78c0SShri Abhyankar       row   = lnk[row];
353ce3d78c0SShri Abhyankar     }
354ce3d78c0SShri Abhyankar     bi[i+1] = bi[i] + nzi;
355ce3d78c0SShri Abhyankar     im[i]   = nzi;
356ce3d78c0SShri Abhyankar 
357ce3d78c0SShri Abhyankar     /* mark bdiag */
358ce3d78c0SShri Abhyankar     nzbd = 0;
359ce3d78c0SShri Abhyankar     nnz  = nzi;
360ce3d78c0SShri Abhyankar     k    = lnk[n];
361ce3d78c0SShri Abhyankar     while (nnz-- && k < i){
362ce3d78c0SShri Abhyankar       nzbd++;
363ce3d78c0SShri Abhyankar       k = lnk[k];
364ce3d78c0SShri Abhyankar     }
365783ef271SHong Zhang     bdiag[i] = nzbd; /* note: bdiag[i] = nnzL as input for PetscFreeSpaceContiguous_LU() */
366ce3d78c0SShri Abhyankar 
367ce3d78c0SShri Abhyankar     /* if free space is not available, make more free space */
368ce3d78c0SShri Abhyankar     if (current_space->local_remaining<nzi) {
369ce3d78c0SShri Abhyankar       nnz = 2*(n - i)*nzi; /* estimated and max additional space needed */
370ce3d78c0SShri Abhyankar       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
371ce3d78c0SShri Abhyankar       reallocs++;
372ce3d78c0SShri Abhyankar     }
373ce3d78c0SShri Abhyankar 
374ce3d78c0SShri Abhyankar     /* copy data into free space, then initialize lnk */
375ce3d78c0SShri Abhyankar     ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
376ce3d78c0SShri Abhyankar     bi_ptr[i] = current_space->array;
377ce3d78c0SShri Abhyankar     current_space->array           += nzi;
378ce3d78c0SShri Abhyankar     current_space->local_used      += nzi;
379ce3d78c0SShri Abhyankar     current_space->local_remaining -= nzi;
380ce3d78c0SShri Abhyankar   }
381ce3d78c0SShri Abhyankar #if defined(PETSC_USE_INFO)
382ce3d78c0SShri Abhyankar   if (ai[n] != 0) {
383ce3d78c0SShri Abhyankar     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
384ce3d78c0SShri Abhyankar     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr);
385ce3d78c0SShri Abhyankar     ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
386ce3d78c0SShri Abhyankar     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr);
387ce3d78c0SShri Abhyankar     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
388ce3d78c0SShri Abhyankar   } else {
389ce3d78c0SShri Abhyankar     ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr);
390ce3d78c0SShri Abhyankar   }
391ce3d78c0SShri Abhyankar #endif
392ce3d78c0SShri Abhyankar 
393ce3d78c0SShri Abhyankar   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
394ce3d78c0SShri Abhyankar   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
395ce3d78c0SShri Abhyankar 
396ce3d78c0SShri Abhyankar   /* destroy list of free space and other temporary array(s) */
397ce3d78c0SShri Abhyankar   ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
398783ef271SHong Zhang   ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr);
399ce3d78c0SShri Abhyankar   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
400ce3d78c0SShri Abhyankar   ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr);
401ce3d78c0SShri Abhyankar 
402ce3d78c0SShri Abhyankar   /* put together the new matrix */
403ce3d78c0SShri Abhyankar   ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
404ce3d78c0SShri Abhyankar   ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr);
405ce3d78c0SShri Abhyankar   b    = (Mat_SeqAIJ*)(B)->data;
406ce3d78c0SShri Abhyankar   b->free_a       = PETSC_TRUE;
407ce3d78c0SShri Abhyankar   b->free_ij      = PETSC_TRUE;
408ce3d78c0SShri Abhyankar   b->singlemalloc = PETSC_FALSE;
409ce3d78c0SShri Abhyankar   ierr          = PetscMalloc((bi[2*n+1])*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);
410ce3d78c0SShri Abhyankar   b->j          = bj;
411ce3d78c0SShri Abhyankar   b->i          = bi;
412ce3d78c0SShri Abhyankar   b->diag       = bdiag;
413ce3d78c0SShri Abhyankar   b->ilen       = 0;
414ce3d78c0SShri Abhyankar   b->imax       = 0;
415ce3d78c0SShri Abhyankar   b->row        = isrow;
416ce3d78c0SShri Abhyankar   b->col        = iscol;
417ce3d78c0SShri Abhyankar   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
418ce3d78c0SShri Abhyankar   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
419ce3d78c0SShri Abhyankar   b->icol       = isicol;
420ce3d78c0SShri Abhyankar   ierr          = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
421ce3d78c0SShri Abhyankar 
422ce3d78c0SShri Abhyankar   /* In b structure:  Free imax, ilen, old a, old j.  Allocate solve_work, new a, new j */
423ce3d78c0SShri Abhyankar   ierr = PetscLogObjectMemory(B,bi[2*n+1]*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
424ce3d78c0SShri Abhyankar   b->maxnz = b->nz = bi[2*n+1] ;
425ce3d78c0SShri Abhyankar 
426ce3d78c0SShri Abhyankar   (B)->factor                = MAT_FACTOR_LU;
427ce3d78c0SShri Abhyankar   (B)->info.factor_mallocs   = reallocs;
428ce3d78c0SShri Abhyankar   (B)->info.fill_ratio_given = f;
429ce3d78c0SShri Abhyankar 
430ce3d78c0SShri Abhyankar   if (ai[n]) {
431894a2336SHong Zhang     (B)->info.fill_ratio_needed = ((PetscReal)bi[2*n+1])/((PetscReal)ai[n]);
432ce3d78c0SShri Abhyankar   } else {
433ce3d78c0SShri Abhyankar     (B)->info.fill_ratio_needed = 0.0;
434ce3d78c0SShri Abhyankar   }
435ce3d78c0SShri Abhyankar   (B)->ops->lufactornumeric  = MatLUFactorNumeric_SeqAIJ_newdatastruct;
436ce3d78c0SShri Abhyankar   PetscFunctionReturn(0);
437ce3d78c0SShri Abhyankar }
438ce3d78c0SShri Abhyankar 
4399b48462bSShri Abhyankar #undef __FUNCT__
4409b48462bSShri Abhyankar #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ_newdatastruct_v2"
4419b48462bSShri Abhyankar PetscErrorCode MatLUFactorSymbolic_SeqAIJ_newdatastruct_v2(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
4429b48462bSShri Abhyankar {
4439b48462bSShri Abhyankar   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
4449b48462bSShri Abhyankar   IS                 isicol;
4459b48462bSShri Abhyankar   PetscErrorCode     ierr;
4469b48462bSShri Abhyankar   const PetscInt     *r,*ic;
4479b48462bSShri Abhyankar   PetscInt           i,n=A->rmap->n,*ai=a->i,*aj=a->j;
4489b48462bSShri Abhyankar   PetscInt           *bi,*bj,*ajtmp;
4499b48462bSShri Abhyankar   PetscInt           *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im;
4509b48462bSShri Abhyankar   PetscReal          f;
4519b48462bSShri Abhyankar   PetscInt           nlnk,*lnk,k,**bi_ptr;
4529b48462bSShri Abhyankar   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
4539b48462bSShri Abhyankar   PetscBT            lnkbt;
4549b48462bSShri Abhyankar 
4559b48462bSShri Abhyankar   PetscFunctionBegin;
4569b48462bSShri Abhyankar   if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_ERR_ARG_WRONG,"matrix must be square");
4579b48462bSShri Abhyankar   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
4589b48462bSShri Abhyankar   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
4599b48462bSShri Abhyankar   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
4609b48462bSShri Abhyankar 
4619b48462bSShri Abhyankar   /* get new row pointers */
4629b48462bSShri Abhyankar   /* ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bi);CHKERRQ(ierr); */
4639b48462bSShri Abhyankar   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
4649b48462bSShri Abhyankar   bi[0] = 0;
4659b48462bSShri Abhyankar 
4669b48462bSShri Abhyankar   /* bdiag is location of diagonal in factor */
4679b48462bSShri Abhyankar   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
4689b48462bSShri Abhyankar   bdiag[0] = 0;
4699b48462bSShri Abhyankar 
4709b48462bSShri Abhyankar   /* linked list for storing column indices of the active row */
4719b48462bSShri Abhyankar   nlnk = n + 1;
4729b48462bSShri Abhyankar   ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
4739b48462bSShri Abhyankar 
4749b48462bSShri Abhyankar   ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr);
4759b48462bSShri Abhyankar 
4769b48462bSShri Abhyankar   /* initial FreeSpace size is f*(ai[n]+1) */
4779b48462bSShri Abhyankar   f = info->fill;
4789b48462bSShri Abhyankar   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr);
4799b48462bSShri Abhyankar   current_space = free_space;
4809b48462bSShri Abhyankar 
4819b48462bSShri Abhyankar   for (i=0; i<n; i++) {
4829b48462bSShri Abhyankar     /* copy previous fill into linked list */
4839b48462bSShri Abhyankar     nzi = 0;
4849b48462bSShri Abhyankar     nnz = ai[r[i]+1] - ai[r[i]];
4859b48462bSShri Abhyankar     if (!nnz) SETERRQ2(PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i);
4869b48462bSShri Abhyankar     ajtmp = aj + ai[r[i]];
4879b48462bSShri Abhyankar     ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
4889b48462bSShri Abhyankar     nzi += nlnk;
4899b48462bSShri Abhyankar 
4909b48462bSShri Abhyankar     /* add pivot rows into linked list */
4919b48462bSShri Abhyankar     row = lnk[n];
4929b48462bSShri Abhyankar     while (row < i){
4939b48462bSShri Abhyankar       nzbd  = bdiag[row] + 1; /* num of entries in the row with column index <= row */
4949b48462bSShri Abhyankar       ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */
4959b48462bSShri Abhyankar       ierr  = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr);
4969b48462bSShri Abhyankar       nzi  += nlnk;
4979b48462bSShri Abhyankar       row   = lnk[row];
4989b48462bSShri Abhyankar     }
4999b48462bSShri Abhyankar     bi[i+1] = bi[i] + nzi;
5009b48462bSShri Abhyankar     im[i]   = nzi;
5019b48462bSShri Abhyankar 
5029b48462bSShri Abhyankar     /* mark bdiag */
5039b48462bSShri Abhyankar     nzbd = 0;
5049b48462bSShri Abhyankar     nnz  = nzi;
5059b48462bSShri Abhyankar     k    = lnk[n];
5069b48462bSShri Abhyankar     while (nnz-- && k < i){
5079b48462bSShri Abhyankar       nzbd++;
5089b48462bSShri Abhyankar       k = lnk[k];
5099b48462bSShri Abhyankar     }
5109b48462bSShri Abhyankar     bdiag[i] = nzbd; /* note: bdiag[i] = nnzL as input for PetscFreeSpaceContiguous_LU() */
5119b48462bSShri Abhyankar 
5129b48462bSShri Abhyankar     /* if free space is not available, make more free space */
5139b48462bSShri Abhyankar     if (current_space->local_remaining<nzi) {
5149b48462bSShri Abhyankar       nnz = 2*(n - i)*nzi; /* estimated and max additional space needed */
5159b48462bSShri Abhyankar       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
5169b48462bSShri Abhyankar       reallocs++;
5179b48462bSShri Abhyankar     }
5189b48462bSShri Abhyankar 
5199b48462bSShri Abhyankar     /* copy data into free space, then initialize lnk */
5209b48462bSShri Abhyankar     ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
5219b48462bSShri Abhyankar     bi_ptr[i] = current_space->array;
5229b48462bSShri Abhyankar     current_space->array           += nzi;
5239b48462bSShri Abhyankar     current_space->local_used      += nzi;
5249b48462bSShri Abhyankar     current_space->local_remaining -= nzi;
5259b48462bSShri Abhyankar   }
5269b48462bSShri Abhyankar #if defined(PETSC_USE_INFO)
5279b48462bSShri Abhyankar   if (ai[n] != 0) {
5289b48462bSShri Abhyankar     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
5299b48462bSShri Abhyankar     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr);
5309b48462bSShri Abhyankar     ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
5319b48462bSShri Abhyankar     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr);
5329b48462bSShri Abhyankar     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
5339b48462bSShri Abhyankar   } else {
5349b48462bSShri Abhyankar     ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr);
5359b48462bSShri Abhyankar   }
5369b48462bSShri Abhyankar #endif
5379b48462bSShri Abhyankar 
5389b48462bSShri Abhyankar   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
5399b48462bSShri Abhyankar   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
5409b48462bSShri Abhyankar 
5419b48462bSShri Abhyankar   /* destroy list of free space and other temporary array(s) */
5429b48462bSShri Abhyankar   ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
5439b48462bSShri Abhyankar   /* ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr); */
5449b48462bSShri Abhyankar   ierr = PetscFreeSpaceContiguous_LU_v2(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr);
5459b48462bSShri Abhyankar   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
5469b48462bSShri Abhyankar   ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr);
5479b48462bSShri Abhyankar 
5489b48462bSShri Abhyankar   /* put together the new matrix */
5499b48462bSShri Abhyankar   ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
5509b48462bSShri Abhyankar   ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr);
5519b48462bSShri Abhyankar   b    = (Mat_SeqAIJ*)(B)->data;
5529b48462bSShri Abhyankar   b->free_a       = PETSC_TRUE;
5539b48462bSShri Abhyankar   b->free_ij      = PETSC_TRUE;
5549b48462bSShri Abhyankar   b->singlemalloc = PETSC_FALSE;
5559b48462bSShri Abhyankar   /*ierr          = PetscMalloc((bi[2*n+1])*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);*/
5569b48462bSShri Abhyankar   ierr = PetscMalloc((bdiag[0]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);
5579b48462bSShri Abhyankar   b->j          = bj;
5589b48462bSShri Abhyankar   b->i          = bi;
5599b48462bSShri Abhyankar   b->diag       = bdiag;
5609b48462bSShri Abhyankar   b->ilen       = 0;
5619b48462bSShri Abhyankar   b->imax       = 0;
5629b48462bSShri Abhyankar   b->row        = isrow;
5639b48462bSShri Abhyankar   b->col        = iscol;
5649b48462bSShri Abhyankar   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
5659b48462bSShri Abhyankar   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
5669b48462bSShri Abhyankar   b->icol       = isicol;
5679b48462bSShri Abhyankar   ierr          = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
5689b48462bSShri Abhyankar 
5699b48462bSShri Abhyankar   /* In b structure:  Free imax, ilen, old a, old j.  Allocate solve_work, new a, new j */
5709b48462bSShri Abhyankar   /* ierr = PetscLogObjectMemory(B,bi[2*n+1]*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
5719b48462bSShri Abhyankar      b->maxnz = b->nz = bi[2*n+1] ; */
5729b48462bSShri Abhyankar   ierr = PetscLogObjectMemory(B,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
5739b48462bSShri Abhyankar   b->maxnz = b->nz = bdiag[0]+1;
5749b48462bSShri Abhyankar   (B)->factor                = MAT_FACTOR_LU;
5759b48462bSShri Abhyankar   (B)->info.factor_mallocs   = reallocs;
5769b48462bSShri Abhyankar   (B)->info.fill_ratio_given = f;
5779b48462bSShri Abhyankar 
5789b48462bSShri Abhyankar   if (ai[n]) {
5799b48462bSShri Abhyankar     /*  (B)->info.fill_ratio_needed = ((PetscReal)bi[2*n+1])/((PetscReal)ai[n]); */
5809b48462bSShri Abhyankar     (B)->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]);
5819b48462bSShri Abhyankar   } else {
5829b48462bSShri Abhyankar     (B)->info.fill_ratio_needed = 0.0;
5839b48462bSShri Abhyankar   }
5849b48462bSShri Abhyankar   /* (B)->ops->lufactornumeric  = MatLUFactorNumeric_SeqAIJ_newdatastruct;*/
5859b48462bSShri Abhyankar   (B)->ops->lufactornumeric  = MatLUFactorNumeric_SeqAIJ_newdatastruct_v2;
5869b48462bSShri Abhyankar   PetscFunctionReturn(0);
5879b48462bSShri Abhyankar }
5889b48462bSShri Abhyankar 
5895b5bc046SBarry Smith /*
5905b5bc046SBarry Smith     Trouble in factorization, should we dump the original matrix?
5915b5bc046SBarry Smith */
5925b5bc046SBarry Smith #undef __FUNCT__
5935b5bc046SBarry Smith #define __FUNCT__ "MatFactorDumpMatrix"
5945b5bc046SBarry Smith PetscErrorCode MatFactorDumpMatrix(Mat A)
5955b5bc046SBarry Smith {
5965b5bc046SBarry Smith   PetscErrorCode ierr;
59790d69ab7SBarry Smith   PetscTruth     flg = PETSC_FALSE;
5985b5bc046SBarry Smith 
5995b5bc046SBarry Smith   PetscFunctionBegin;
60090d69ab7SBarry Smith   ierr = PetscOptionsGetTruth(PETSC_NULL,"-mat_factor_dump_on_error",&flg,PETSC_NULL);CHKERRQ(ierr);
6015b5bc046SBarry Smith   if (flg) {
6025b5bc046SBarry Smith     PetscViewer viewer;
6035b5bc046SBarry Smith     char        filename[PETSC_MAX_PATH_LEN];
6045b5bc046SBarry Smith 
6055b5bc046SBarry Smith     ierr = PetscSNPrintf(filename,PETSC_MAX_PATH_LEN,"matrix_factor_error.%d",PetscGlobalRank);CHKERRQ(ierr);
6067adad957SLisandro Dalcin     ierr = PetscViewerBinaryOpen(((PetscObject)A)->comm,filename,FILE_MODE_WRITE,&viewer);CHKERRQ(ierr);
6075b5bc046SBarry Smith     ierr = MatView(A,viewer);CHKERRQ(ierr);
6085b5bc046SBarry Smith     ierr = PetscViewerDestroy(viewer);CHKERRQ(ierr);
6095b5bc046SBarry Smith   }
6105b5bc046SBarry Smith   PetscFunctionReturn(0);
6115b5bc046SBarry Smith }
6125b5bc046SBarry Smith 
6134108e4d5SBarry Smith extern PetscErrorCode MatSolve_SeqAIJ_Inode(Mat,Vec,Vec);
614db4efbfdSBarry Smith 
6150a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
616894a2336SHong Zhang extern PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering_newdatastruct(Mat,Vec,Vec);
617894a2336SHong Zhang extern PetscErrorCode MatSolve_SeqAIJ_newdatastruct(Mat,Vec,Vec);
618c9c72f8fSHong Zhang 
619c9c72f8fSHong Zhang #undef __FUNCT__
620c9c72f8fSHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_newdatastruct"
621c9c72f8fSHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ_newdatastruct(Mat B,Mat A,const MatFactorInfo *info)
622c9c72f8fSHong Zhang {
623c9c72f8fSHong Zhang   Mat            C=B;
624c9c72f8fSHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data;
625c9c72f8fSHong Zhang   IS             isrow = b->row,isicol = b->icol;
626c9c72f8fSHong Zhang   PetscErrorCode ierr;
627c9c72f8fSHong Zhang   const PetscInt *r,*ic,*ics;
628c9c72f8fSHong Zhang   PetscInt       i,j,k,n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j;
629c9c72f8fSHong Zhang   PetscInt       *ajtmp,*bjtmp,nz,nzL,row,*bdiag=b->diag,*pj;
630c9c72f8fSHong Zhang   MatScalar      *rtmp,*pc,multiplier,*v,*pv,*aa=a->a;
631c9c72f8fSHong Zhang   PetscTruth     row_identity,col_identity;
632c9c72f8fSHong Zhang 
633d4ad06f7SHong Zhang   LUShift_Ctx    sctx;
634d4ad06f7SHong Zhang   PetscInt       *ddiag,newshift;
635d4ad06f7SHong Zhang   PetscReal      rs;
636d4ad06f7SHong Zhang   MatScalar      d;
637d4ad06f7SHong Zhang 
638c9c72f8fSHong Zhang   PetscFunctionBegin;
639d6e5152cSHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
640d6e5152cSHong Zhang   ierr = PetscMemzero(&sctx,sizeof(LUShift_Ctx));CHKERRQ(ierr);
641d4ad06f7SHong Zhang 
642d4ad06f7SHong Zhang   /* if both shift schemes are chosen by user, only use info->shiftpd */
643d4ad06f7SHong Zhang   if (info->shiftpd) { /* set sctx.shift_top=max{rs} */
644d4ad06f7SHong Zhang     ddiag          = a->diag;
645d4ad06f7SHong Zhang     sctx.shift_top = info->zeropivot;
646d4ad06f7SHong Zhang     for (i=0; i<n; i++) {
647d4ad06f7SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
648d4ad06f7SHong Zhang       d  = (aa)[ddiag[i]];
649d4ad06f7SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
650d4ad06f7SHong Zhang       v  = aa+ai[i];
651d4ad06f7SHong Zhang       nz = ai[i+1] - ai[i];
652d4ad06f7SHong Zhang       for (j=0; j<nz; j++)
653d4ad06f7SHong Zhang 	rs += PetscAbsScalar(v[j]);
654d4ad06f7SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
655d4ad06f7SHong Zhang     }
656d4ad06f7SHong Zhang     sctx.shift_top   *= 1.1;
657d4ad06f7SHong Zhang     sctx.nshift_max   = 5;
658d4ad06f7SHong Zhang     sctx.shift_lo     = 0.;
659d4ad06f7SHong Zhang     sctx.shift_hi     = 1.;
660d4ad06f7SHong Zhang   }
661d4ad06f7SHong Zhang 
662c9c72f8fSHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
663c9c72f8fSHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
664c9c72f8fSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr);
665c9c72f8fSHong Zhang   ics  = ic;
666c9c72f8fSHong Zhang 
667d4ad06f7SHong Zhang   do {
668d4ad06f7SHong Zhang     sctx.lushift = PETSC_FALSE;
669c9c72f8fSHong Zhang     for (i=0; i<n; i++){
670c9c72f8fSHong Zhang       /* zero rtmp */
671c9c72f8fSHong Zhang       /* L part */
672c9c72f8fSHong Zhang       nz    = bi[i+1] - bi[i];
673c9c72f8fSHong Zhang       bjtmp = bj + bi[i];
674c9c72f8fSHong Zhang       for  (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
675c9c72f8fSHong Zhang 
676c9c72f8fSHong Zhang       /* U part */
6778fc3a347SHong Zhang       nz = bi[2*n-i+1] - bi[2*n-i];
6788fc3a347SHong Zhang       bjtmp = bj + bi[2*n-i];
679c9c72f8fSHong Zhang       for  (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
680c9c72f8fSHong Zhang 
681c9c72f8fSHong Zhang       /* load in initial (unfactored row) */
682c9c72f8fSHong Zhang       nz    = ai[r[i]+1] - ai[r[i]];
683c9c72f8fSHong Zhang       ajtmp = aj + ai[r[i]];
684c9c72f8fSHong Zhang       v     = aa + ai[r[i]];
685c9c72f8fSHong Zhang       for (j=0; j<nz; j++) {
686c9c72f8fSHong Zhang         rtmp[ics[ajtmp[j]]] = v[j];
687c9c72f8fSHong Zhang       }
688d4ad06f7SHong Zhang       /* ZeropivotApply() */
689d4ad06f7SHong Zhang       rtmp[ics[r[i]]] += sctx.shift_amount;  /* shift the diagonal of the matrix */
690c9c72f8fSHong Zhang 
691c9c72f8fSHong Zhang       /* elimination */
692c9c72f8fSHong Zhang       bjtmp = bj + bi[i];
693c9c72f8fSHong Zhang       row   = *bjtmp++;
694c9c72f8fSHong Zhang       nzL   = bi[i+1] - bi[i];
695c9c72f8fSHong Zhang       k   = 0;
696c9c72f8fSHong Zhang       while  (k < nzL) {
697c9c72f8fSHong Zhang         pc = rtmp + row;
698c9c72f8fSHong Zhang         if (*pc != 0.0) {
699c9c72f8fSHong Zhang           pv         = b->a + bdiag[row];
700c9c72f8fSHong Zhang           multiplier = *pc * (*pv);
701c9c72f8fSHong Zhang           *pc        = multiplier;
7028fc3a347SHong Zhang           pj         = b->j + bi[2*n-row]; /* begining of U(row,:) */
7038fc3a347SHong Zhang           pv         = b->a + bi[2*n-row];
7048fc3a347SHong Zhang           nz         = bi[2*n-row+1] - bi[2*n-row] - 1; /* num of entries in U(row,:), excluding diag */
705c9c72f8fSHong Zhang           for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j];
706c9c72f8fSHong Zhang           ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr);
707c9c72f8fSHong Zhang         }
708c9c72f8fSHong Zhang         row = *bjtmp++; k++;
709c9c72f8fSHong Zhang       }
710c9c72f8fSHong Zhang 
711c9c72f8fSHong Zhang       /* finished row so stick it into b->a */
712d4ad06f7SHong Zhang       rs = 0.0;
713c9c72f8fSHong Zhang       /* L part */
714c9c72f8fSHong Zhang       pv   = b->a + bi[i] ;
715c9c72f8fSHong Zhang       pj   = b->j + bi[i] ;
716c9c72f8fSHong Zhang       nz   = bi[i+1] - bi[i];
717c9c72f8fSHong Zhang       for (j=0; j<nz; j++) {
718d4ad06f7SHong Zhang         pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]);
719c9c72f8fSHong Zhang       }
720c9c72f8fSHong Zhang 
721c9c72f8fSHong Zhang       /* U part */
7228fc3a347SHong Zhang       pv = b->a + bi[2*n-i];
7238fc3a347SHong Zhang       pj = b->j + bi[2*n-i];
7248fc3a347SHong Zhang       nz = bi[2*n-i+1] - bi[2*n-i] - 1;
725d4ad06f7SHong Zhang       for (j=0; j<nz; j++) {
726d4ad06f7SHong Zhang         pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]);
727c9c72f8fSHong Zhang       }
728d4ad06f7SHong Zhang 
729d6e5152cSHong Zhang       /* MatPivotCheck() */
730d4ad06f7SHong Zhang       sctx.rs  = rs;
731d4ad06f7SHong Zhang       sctx.pv  = rtmp[i];
732d6e5152cSHong Zhang       /* ierr = MatLUCheckShift_inline(info,sctx,i,newshift);CHKERRQ(ierr); */
733d6e5152cSHong Zhang       if (info->shiftnz){
734d6e5152cSHong Zhang         ierr = MatPivotCheck_nz(info,sctx,i,newshift);CHKERRQ(ierr);
735d6e5152cSHong Zhang       } else if (info->shiftpd){
736d6e5152cSHong Zhang         ierr = MatPivotCheck_pd(info,sctx,i,newshift);CHKERRQ(ierr);
737d6e5152cSHong Zhang       } else if (info->shiftinblocks){
738d6e5152cSHong Zhang         ierr = MatPivotCheck_inblocks(info,sctx,i,newshift);CHKERRQ(ierr);
739d6e5152cSHong Zhang       } else {
740d6e5152cSHong Zhang         ierr = MatPivotCheck_none(info,sctx,i,newshift);CHKERRQ(ierr);
741d6e5152cSHong Zhang       }
742d6e5152cSHong Zhang       rtmp[i] = sctx.pv;
743d4ad06f7SHong Zhang       if (newshift == 1) break;
744d4ad06f7SHong Zhang 
745d4ad06f7SHong Zhang       /* Mark diagonal and invert diagonal for simplier triangular solves */
746d4ad06f7SHong Zhang       pv  = b->a + bdiag[i];
747d4ad06f7SHong Zhang       *pv = 1.0/rtmp[i];
748d4ad06f7SHong Zhang 
749d4ad06f7SHong Zhang     } /* endof for (i=0; i<n; i++){ */
750d4ad06f7SHong Zhang 
751d6e5152cSHong Zhang     /* MatPivotRefine() */
752d4ad06f7SHong Zhang     if (info->shiftpd && !sctx.lushift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max){
753d4ad06f7SHong Zhang       /*
754d4ad06f7SHong Zhang        * if no shift in this attempt & shifting & started shifting & can refine,
755d4ad06f7SHong Zhang        * then try lower shift
756d4ad06f7SHong Zhang        */
757d4ad06f7SHong Zhang       sctx.shift_hi       = sctx.shift_fraction;
758d4ad06f7SHong Zhang       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
759d4ad06f7SHong Zhang       sctx.shift_amount   = sctx.shift_fraction * sctx.shift_top;
760d4ad06f7SHong Zhang       sctx.lushift        = PETSC_TRUE;
761d4ad06f7SHong Zhang       sctx.nshift++;
762d4ad06f7SHong Zhang     }
763d4ad06f7SHong Zhang   } while (sctx.lushift);
764d4ad06f7SHong Zhang 
765c9c72f8fSHong Zhang   ierr = PetscFree(rtmp);CHKERRQ(ierr);
766c9c72f8fSHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
767c9c72f8fSHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
768c9c72f8fSHong Zhang 
769c9c72f8fSHong Zhang   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
770c9c72f8fSHong Zhang   ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr);
771c9c72f8fSHong Zhang   if (row_identity && col_identity) {
772894a2336SHong Zhang     C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_newdatastruct;
773c9c72f8fSHong Zhang   } else {
774894a2336SHong Zhang     C->ops->solve = MatSolve_SeqAIJ_newdatastruct;
775c9c72f8fSHong Zhang   }
776c9c72f8fSHong Zhang 
777c9c72f8fSHong Zhang   C->ops->solveadd           = 0;
778c9c72f8fSHong Zhang   C->ops->solvetranspose     = 0;
779c9c72f8fSHong Zhang   C->ops->solvetransposeadd  = 0;
780c9c72f8fSHong Zhang   C->ops->matsolve           = 0;
781c9c72f8fSHong Zhang   C->assembled    = PETSC_TRUE;
782c9c72f8fSHong Zhang   C->preallocated = PETSC_TRUE;
783c9c72f8fSHong Zhang   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
784d4ad06f7SHong Zhang 
785d6e5152cSHong Zhang   /* MatPivotView() */
786d4ad06f7SHong Zhang   if (sctx.nshift){
787d4ad06f7SHong Zhang     if (info->shiftpd) {
788d4ad06f7SHong 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);
789d4ad06f7SHong Zhang     } else if (info->shiftnz) {
790d4ad06f7SHong Zhang       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
791d4ad06f7SHong Zhang     }
792d4ad06f7SHong Zhang   }
793c9c72f8fSHong Zhang   PetscFunctionReturn(0);
794c9c72f8fSHong Zhang }
795c9c72f8fSHong Zhang 
796813a1f2bSShri Abhyankar /* ----------------------------------------------------------- */
797813a1f2bSShri Abhyankar extern PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering_newdatastruct_v2(Mat,Vec,Vec);
798f268cba8SShri Abhyankar extern PetscErrorCode MatSolve_SeqAIJ_newdatastruct_v2(Mat,Vec,Vec);
799813a1f2bSShri Abhyankar 
800813a1f2bSShri Abhyankar #undef __FUNCT__
801813a1f2bSShri Abhyankar #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_newdatastruct_v2"
802813a1f2bSShri Abhyankar PetscErrorCode MatLUFactorNumeric_SeqAIJ_newdatastruct_v2(Mat B,Mat A,const MatFactorInfo *info)
803813a1f2bSShri Abhyankar {
804813a1f2bSShri Abhyankar   Mat            C=B;
805813a1f2bSShri Abhyankar   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data;
806813a1f2bSShri Abhyankar   IS             isrow = b->row,isicol = b->icol;
807813a1f2bSShri Abhyankar   PetscErrorCode ierr;
808813a1f2bSShri Abhyankar   const PetscInt *r,*ic,*ics;
809813a1f2bSShri Abhyankar   PetscInt       i,j,k,n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j;
810813a1f2bSShri Abhyankar   PetscInt       *ajtmp,*bjtmp,nz,nzL,row,*bdiag=b->diag,*pj;
811813a1f2bSShri Abhyankar   MatScalar      *rtmp,*pc,multiplier,*v,*pv,*aa=a->a;
812813a1f2bSShri Abhyankar   PetscTruth     row_identity,col_identity;
813813a1f2bSShri Abhyankar 
814813a1f2bSShri Abhyankar   LUShift_Ctx    sctx;
815813a1f2bSShri Abhyankar   PetscInt       *ddiag,newshift;
816813a1f2bSShri Abhyankar   PetscReal      rs;
817813a1f2bSShri Abhyankar   MatScalar      d;
818813a1f2bSShri Abhyankar 
819813a1f2bSShri Abhyankar   PetscFunctionBegin;
8208ff23777SHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
8218ff23777SHong Zhang   ierr = PetscMemzero(&sctx,sizeof(LUShift_Ctx));CHKERRQ(ierr);
822813a1f2bSShri Abhyankar 
823813a1f2bSShri Abhyankar   /* if both shift schemes are chosen by user, only use info->shiftpd */
824813a1f2bSShri Abhyankar   if (info->shiftpd) { /* set sctx.shift_top=max{rs} */
825813a1f2bSShri Abhyankar     ddiag          = a->diag;
826813a1f2bSShri Abhyankar     sctx.shift_top = info->zeropivot;
827813a1f2bSShri Abhyankar     for (i=0; i<n; i++) {
828813a1f2bSShri Abhyankar       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
829813a1f2bSShri Abhyankar       d  = (aa)[ddiag[i]];
830813a1f2bSShri Abhyankar       rs = -PetscAbsScalar(d) - PetscRealPart(d);
831813a1f2bSShri Abhyankar       v  = aa+ai[i];
832813a1f2bSShri Abhyankar       nz = ai[i+1] - ai[i];
833813a1f2bSShri Abhyankar       for (j=0; j<nz; j++)
834813a1f2bSShri Abhyankar 	rs += PetscAbsScalar(v[j]);
835813a1f2bSShri Abhyankar       if (rs>sctx.shift_top) sctx.shift_top = rs;
836813a1f2bSShri Abhyankar     }
837813a1f2bSShri Abhyankar     sctx.shift_top   *= 1.1;
838813a1f2bSShri Abhyankar     sctx.nshift_max   = 5;
839813a1f2bSShri Abhyankar     sctx.shift_lo     = 0.;
840813a1f2bSShri Abhyankar     sctx.shift_hi     = 1.;
841813a1f2bSShri Abhyankar   }
842813a1f2bSShri Abhyankar 
843813a1f2bSShri Abhyankar   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
844813a1f2bSShri Abhyankar   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
845813a1f2bSShri Abhyankar   ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr);
846813a1f2bSShri Abhyankar   ics  = ic;
847813a1f2bSShri Abhyankar 
848813a1f2bSShri Abhyankar   do {
849813a1f2bSShri Abhyankar     sctx.lushift = PETSC_FALSE;
850813a1f2bSShri Abhyankar     for (i=0; i<n; i++){
851813a1f2bSShri Abhyankar       /* zero rtmp */
852813a1f2bSShri Abhyankar       /* L part */
853813a1f2bSShri Abhyankar       nz    = bi[i+1] - bi[i];
854813a1f2bSShri Abhyankar       bjtmp = bj + bi[i];
855813a1f2bSShri Abhyankar       for  (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
856813a1f2bSShri Abhyankar 
857813a1f2bSShri Abhyankar       /* U part */
85862fbd6afSShri Abhyankar       /*   nz = bi[2*n-i+1] - bi[2*n-i];
85962fbd6afSShri Abhyankar 	   bjtmp = bj + bi[2*n-i];  */
86062fbd6afSShri Abhyankar       nz = bdiag[i]-bdiag[i+1];
86162fbd6afSShri Abhyankar       bjtmp = bj + bdiag[i+1]+1;
862813a1f2bSShri Abhyankar       for  (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
863813a1f2bSShri Abhyankar 
864813a1f2bSShri Abhyankar       /* load in initial (unfactored row) */
865813a1f2bSShri Abhyankar       nz    = ai[r[i]+1] - ai[r[i]];
866813a1f2bSShri Abhyankar       ajtmp = aj + ai[r[i]];
867813a1f2bSShri Abhyankar       v     = aa + ai[r[i]];
868813a1f2bSShri Abhyankar       for (j=0; j<nz; j++) {
869813a1f2bSShri Abhyankar         rtmp[ics[ajtmp[j]]] = v[j];
870813a1f2bSShri Abhyankar       }
871813a1f2bSShri Abhyankar       /* ZeropivotApply() */
872813a1f2bSShri Abhyankar       rtmp[ics[r[i]]] += sctx.shift_amount;  /* shift the diagonal of the matrix */
873813a1f2bSShri Abhyankar 
874813a1f2bSShri Abhyankar       /* elimination */
875813a1f2bSShri Abhyankar       bjtmp = bj + bi[i];
876813a1f2bSShri Abhyankar       row   = *bjtmp++;
877813a1f2bSShri Abhyankar       nzL   = bi[i+1] - bi[i];
878f268cba8SShri Abhyankar       for(k=0; k < nzL;k++) {
879813a1f2bSShri Abhyankar         pc = rtmp + row;
880813a1f2bSShri Abhyankar         if (*pc != 0.0) {
881813a1f2bSShri Abhyankar           pv         = b->a + bdiag[row];
882813a1f2bSShri Abhyankar           multiplier = *pc * (*pv);
883813a1f2bSShri Abhyankar           *pc        = multiplier;
88462fbd6afSShri Abhyankar 	  /*   pj         = b->j + bi[2*n-row]; */ /* begining of U(row,:) */
88562fbd6afSShri Abhyankar 	  /* pv         = b->a + bi[2*n-row];
88662fbd6afSShri Abhyankar 	     nz         = bi[2*n-row+1] - bi[2*n-row] - 1; */ /* num of entries in U(row,:), excluding diag */
88762fbd6afSShri Abhyankar           pj = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */
88862fbd6afSShri Abhyankar 	  pv = b->a + bdiag[row+1]+1;
88962fbd6afSShri Abhyankar 	  nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */
890813a1f2bSShri Abhyankar           for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j];
891813a1f2bSShri Abhyankar           ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr);
892813a1f2bSShri Abhyankar         }
893f268cba8SShri Abhyankar         row = *bjtmp++;
894813a1f2bSShri Abhyankar       }
895813a1f2bSShri Abhyankar 
896813a1f2bSShri Abhyankar       /* finished row so stick it into b->a */
897813a1f2bSShri Abhyankar       rs = 0.0;
898813a1f2bSShri Abhyankar       /* L part */
899813a1f2bSShri Abhyankar       pv   = b->a + bi[i] ;
900813a1f2bSShri Abhyankar       pj   = b->j + bi[i] ;
901813a1f2bSShri Abhyankar       nz   = bi[i+1] - bi[i];
902813a1f2bSShri Abhyankar       for (j=0; j<nz; j++) {
903813a1f2bSShri Abhyankar         pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]);
904813a1f2bSShri Abhyankar       }
905813a1f2bSShri Abhyankar 
906813a1f2bSShri Abhyankar       /* U part */
90762fbd6afSShri Abhyankar       /*  pv = b->a + bi[2*n-i];
908813a1f2bSShri Abhyankar       pj = b->j + bi[2*n-i];
909813a1f2bSShri Abhyankar       nz = bi[2*n-i+1] - bi[2*n-i] - 1;
91062fbd6afSShri Abhyankar       */
91162fbd6afSShri Abhyankar       pv = b->a + bdiag[i+1]+1;
91262fbd6afSShri Abhyankar       pj = b->j + bdiag[i+1]+1;
91362fbd6afSShri Abhyankar       nz = bdiag[i] - bdiag[i+1]-1;
914813a1f2bSShri Abhyankar       for (j=0; j<nz; j++) {
915813a1f2bSShri Abhyankar         pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]);
916813a1f2bSShri Abhyankar       }
917813a1f2bSShri Abhyankar 
9188ff23777SHong Zhang       /* MatPivotCheck() */
919813a1f2bSShri Abhyankar       sctx.rs  = rs;
920813a1f2bSShri Abhyankar       sctx.pv  = rtmp[i];
9218ff23777SHong Zhang       /* ierr = MatLUCheckShift_inline(info,sctx,i,newshift);CHKERRQ(ierr); */
9228ff23777SHong Zhang       if (info->shiftnz){
9238ff23777SHong Zhang         ierr = MatPivotCheck_nz(info,sctx,i,newshift);CHKERRQ(ierr);
9248ff23777SHong Zhang       } else if (info->shiftpd){
9258ff23777SHong Zhang         ierr = MatPivotCheck_pd(info,sctx,i,newshift);CHKERRQ(ierr);
9268ff23777SHong Zhang       } else if (info->shiftinblocks){
9278ff23777SHong Zhang         ierr = MatPivotCheck_inblocks(info,sctx,i,newshift);CHKERRQ(ierr);
9288ff23777SHong Zhang       } else {
9298ff23777SHong Zhang         ierr = MatPivotCheck_none(info,sctx,i,newshift);CHKERRQ(ierr);
9308ff23777SHong Zhang       }
9318ff23777SHong Zhang       rtmp[i] = sctx.pv;
932813a1f2bSShri Abhyankar       if (newshift == 1) break;
933813a1f2bSShri Abhyankar 
934813a1f2bSShri Abhyankar       /* Mark diagonal and invert diagonal for simplier triangular solves */
935813a1f2bSShri Abhyankar       pv  = b->a + bdiag[i];
936813a1f2bSShri Abhyankar       *pv = 1.0/rtmp[i];
937813a1f2bSShri Abhyankar 
938813a1f2bSShri Abhyankar     } /* endof for (i=0; i<n; i++){ */
939813a1f2bSShri Abhyankar 
9408ff23777SHong Zhang     /* MatPivotRefine() */
941813a1f2bSShri Abhyankar     if (info->shiftpd && !sctx.lushift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max){
942813a1f2bSShri Abhyankar       /*
943813a1f2bSShri Abhyankar        * if no shift in this attempt & shifting & started shifting & can refine,
944813a1f2bSShri Abhyankar        * then try lower shift
945813a1f2bSShri Abhyankar        */
946813a1f2bSShri Abhyankar       sctx.shift_hi       = sctx.shift_fraction;
947813a1f2bSShri Abhyankar       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
948813a1f2bSShri Abhyankar       sctx.shift_amount   = sctx.shift_fraction * sctx.shift_top;
949813a1f2bSShri Abhyankar       sctx.lushift        = PETSC_TRUE;
950813a1f2bSShri Abhyankar       sctx.nshift++;
951813a1f2bSShri Abhyankar     }
952813a1f2bSShri Abhyankar   } while (sctx.lushift);
953813a1f2bSShri Abhyankar 
954813a1f2bSShri Abhyankar   ierr = PetscFree(rtmp);CHKERRQ(ierr);
955813a1f2bSShri Abhyankar   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
956813a1f2bSShri Abhyankar   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
957813a1f2bSShri Abhyankar 
958813a1f2bSShri Abhyankar   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
959813a1f2bSShri Abhyankar   ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr);
960813a1f2bSShri Abhyankar   if (row_identity && col_identity) {
96162fbd6afSShri Abhyankar     C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_newdatastruct_v2;
962813a1f2bSShri Abhyankar   } else {
963f268cba8SShri Abhyankar     C->ops->solve = MatSolve_SeqAIJ_newdatastruct_v2;
964813a1f2bSShri Abhyankar   }
965813a1f2bSShri Abhyankar 
966813a1f2bSShri Abhyankar   C->ops->solveadd           = 0;
967813a1f2bSShri Abhyankar   C->ops->solvetranspose     = 0;
968813a1f2bSShri Abhyankar   C->ops->solvetransposeadd  = 0;
969813a1f2bSShri Abhyankar   C->ops->matsolve           = 0;
970813a1f2bSShri Abhyankar   C->assembled    = PETSC_TRUE;
971813a1f2bSShri Abhyankar   C->preallocated = PETSC_TRUE;
972813a1f2bSShri Abhyankar   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
973813a1f2bSShri Abhyankar 
9748ff23777SHong Zhang   /* MatPivotView() */
975813a1f2bSShri Abhyankar   if (sctx.nshift){
976813a1f2bSShri Abhyankar     if (info->shiftpd) {
977813a1f2bSShri 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);
978813a1f2bSShri Abhyankar     } else if (info->shiftnz) {
979813a1f2bSShri Abhyankar       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
9808ff23777SHong Zhang     } else if (info->shiftinblocks){
9818ff23777SHong Zhang       ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %G\n",sctx.nshift,info->shiftinblocks);CHKERRQ(ierr);
982813a1f2bSShri Abhyankar     }
983813a1f2bSShri Abhyankar   }
984813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
985813a1f2bSShri Abhyankar }
986813a1f2bSShri Abhyankar 
9874a2ae208SSatish Balay #undef __FUNCT__
9884a2ae208SSatish Balay #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ"
9890481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info)
990289bc588SBarry Smith {
991719d5645SBarry Smith   Mat             C=B;
992416022c9SBarry Smith   Mat_SeqAIJ      *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data;
99382bf6240SBarry Smith   IS              isrow = b->row,isicol = b->icol;
9946849ba73SBarry Smith   PetscErrorCode  ierr;
9955d0c19d7SBarry Smith   const PetscInt   *r,*ic,*ics;
996d278edc7SBarry Smith   PetscInt        nz,row,i,j,n=A->rmap->n,diag;
997d278edc7SBarry Smith   const PetscInt  *ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j;
998d278edc7SBarry Smith   const PetscInt  *ajtmp,*bjtmp,*diag_offset = b->diag,*pj;
999d278edc7SBarry Smith   MatScalar       *pv,*rtmp,*pc,multiplier,d;
1000d278edc7SBarry Smith   const MatScalar *v,*aa=a->a;
100133f9893dSHong Zhang   PetscReal       rs=0.0;
1002b3bf805bSHong Zhang   LUShift_Ctx     sctx;
100342898933SHong Zhang   PetscInt        newshift,*ddiag;
1004289bc588SBarry Smith 
10053a40ed3dSBarry Smith   PetscFunctionBegin;
100678b31e54SBarry Smith   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
100778b31e54SBarry Smith   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
1008b78a477dSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr);
10099f7d0b68SHong Zhang   ics  = ic;
1010289bc588SBarry Smith 
1011d4ad06f7SHong Zhang   /* initialize shift context sctx */
1012d4ad06f7SHong Zhang   sctx.nshift         = 0;
1013ada7143aSSatish Balay   sctx.nshift_max     = 0;
1014d4ad06f7SHong Zhang   sctx.shift_top      = 0.0;
1015d4ad06f7SHong Zhang   sctx.shift_lo       = 0.0;
1016d4ad06f7SHong Zhang   sctx.shift_hi       = 0.0;
1017d4ad06f7SHong Zhang   sctx.shift_fraction = 0.0;
1018d4ad06f7SHong Zhang   sctx.shift_amount   = 0.0;
1019ada7143aSSatish Balay 
1020118f5789SBarry Smith   /* if both shift schemes are chosen by user, only use info->shiftpd */
10211a890ab1SHong Zhang   if (info->shiftpd) { /* set sctx.shift_top=max{rs} */
102242898933SHong Zhang     ddiag          = a->diag;
10239f7d0b68SHong Zhang     sctx.shift_top = info->zeropivot;
10246cc28720Svictorle     for (i=0; i<n; i++) {
10259f95998fSHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
10269f7d0b68SHong Zhang       d  = (aa)[ddiag[i]];
10271a890ab1SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
10289f7d0b68SHong Zhang       v  = aa+ai[i];
10299f7d0b68SHong Zhang       nz = ai[i+1] - ai[i];
1030e24cfe64SHong Zhang       for (j=0; j<nz; j++)
10311a890ab1SHong Zhang 	rs += PetscAbsScalar(v[j]);
10321a890ab1SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
10336cc28720Svictorle     }
1034118f5789SBarry Smith     sctx.shift_top   *= 1.1;
1035d2276718SHong Zhang     sctx.nshift_max   = 5;
1036d2276718SHong Zhang     sctx.shift_lo     = 0.;
1037d2276718SHong Zhang     sctx.shift_hi     = 1.;
1038a0a356efSHong Zhang   }
1039a0a356efSHong Zhang 
1040085256b3SBarry Smith   do {
1041d2276718SHong Zhang     sctx.lushift = PETSC_FALSE;
1042289bc588SBarry Smith     for (i=0; i<n; i++){
1043b3bf805bSHong Zhang       nz    = bi[i+1] - bi[i];
1044b3bf805bSHong Zhang       bjtmp = bj + bi[i];
10459f7d0b68SHong Zhang       for  (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
1046289bc588SBarry Smith 
1047289bc588SBarry Smith       /* load in initial (unfactored row) */
10489f7d0b68SHong Zhang       nz    = ai[r[i]+1] - ai[r[i]];
10499f7d0b68SHong Zhang       ajtmp = aj + ai[r[i]];
10509f7d0b68SHong Zhang       v     = aa + ai[r[i]];
1051085256b3SBarry Smith       for (j=0; j<nz; j++) {
1052b3bf805bSHong Zhang         rtmp[ics[ajtmp[j]]] = v[j];
1053085256b3SBarry Smith       }
1054d2276718SHong Zhang       rtmp[ics[r[i]]] += sctx.shift_amount; /* shift the diagonal of the matrix */
1055bd1bc851SBarry Smith       /* if (sctx.shift_amount > 0.0) printf("row %d, shift %g\n",i,sctx.shift_amount); */
1056289bc588SBarry Smith 
1057b3bf805bSHong Zhang       row = *bjtmp++;
1058f2582111SSatish Balay       while  (row < i) {
10598c37ef55SBarry Smith         pc = rtmp + row;
10608c37ef55SBarry Smith         if (*pc != 0.0) {
1061010a20caSHong Zhang           pv         = b->a + diag_offset[row];
1062010a20caSHong Zhang           pj         = b->j + diag_offset[row] + 1;
106335aab85fSBarry Smith           multiplier = *pc / *pv++;
10648c37ef55SBarry Smith           *pc        = multiplier;
1065b3bf805bSHong Zhang           nz         = bi[row+1] - diag_offset[row] - 1;
10669f7d0b68SHong Zhang           for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j];
1067dc0b31edSSatish Balay           ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr);
1068289bc588SBarry Smith         }
1069b3bf805bSHong Zhang         row = *bjtmp++;
1070289bc588SBarry Smith       }
1071416022c9SBarry Smith       /* finished row so stick it into b->a */
1072b3bf805bSHong Zhang       pv   = b->a + bi[i] ;
1073b3bf805bSHong Zhang       pj   = b->j + bi[i] ;
1074b3bf805bSHong Zhang       nz   = bi[i+1] - bi[i];
1075b3bf805bSHong Zhang       diag = diag_offset[i] - bi[i];
1076e57ff13aSBarry Smith       rs   = 0.0;
1077d3d32019SBarry Smith       for (j=0; j<nz; j++) {
10789f7d0b68SHong Zhang         pv[j] = rtmp[pj[j]];
10799f7d0b68SHong Zhang         rs   += PetscAbsScalar(pv[j]);
1080d3d32019SBarry Smith       }
1081e57ff13aSBarry Smith       rs   -= PetscAbsScalar(pv[diag]);
1082d2276718SHong Zhang 
1083b3bf805bSHong Zhang       /* 9/13/02 Victor Eijkhout suggested scaling zeropivot by rs for matrices with funny scalings */
1084d2276718SHong Zhang       sctx.rs  = rs;
1085d2276718SHong Zhang       sctx.pv  = pv[diag];
1086c6b1f410SHong Zhang       ierr = MatLUCheckShift_inline(info,sctx,i,newshift);CHKERRQ(ierr);
10875b5bc046SBarry Smith       if (newshift == 1) break;
108835aab85fSBarry Smith     }
1089d2276718SHong Zhang 
10900481f469SBarry Smith     if (info->shiftpd && !sctx.lushift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) {
10916cc28720Svictorle       /*
10926c037d1bSvictorle        * if no shift in this attempt & shifting & started shifting & can refine,
10936cc28720Svictorle        * then try lower shift
10946cc28720Svictorle        */
10950481f469SBarry Smith       sctx.shift_hi       = sctx.shift_fraction;
10960481f469SBarry Smith       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
10970481f469SBarry Smith       sctx.shift_amount   = sctx.shift_fraction * sctx.shift_top;
1098d2276718SHong Zhang       sctx.lushift        = PETSC_TRUE;
1099d2276718SHong Zhang       sctx.nshift++;
11006cc28720Svictorle     }
1101d2276718SHong Zhang   } while (sctx.lushift);
11020f11f9aeSSatish Balay 
110335aab85fSBarry Smith   /* invert diagonal entries for simplier triangular solves */
110435aab85fSBarry Smith   for (i=0; i<n; i++) {
1105010a20caSHong Zhang     b->a[diag_offset[i]] = 1.0/b->a[diag_offset[i]];
110635aab85fSBarry Smith   }
1107606d414cSSatish Balay   ierr = PetscFree(rtmp);CHKERRQ(ierr);
110878b31e54SBarry Smith   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
110978b31e54SBarry Smith   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
1110db4efbfdSBarry Smith   if (b->inode.use) {
11114108e4d5SBarry Smith     C->ops->solve   = MatSolve_SeqAIJ_Inode;
1112db4efbfdSBarry Smith   } else {
11138d8c7c43SBarry Smith     PetscTruth row_identity, col_identity;
11148d8c7c43SBarry Smith     ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
11158d8c7c43SBarry Smith     ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr);
11168d8c7c43SBarry Smith     if (row_identity && col_identity) {
11178d8c7c43SBarry Smith       C->ops->solve   = MatSolve_SeqAIJ_NaturalOrdering;
11188d8c7c43SBarry Smith     } else {
1119db4efbfdSBarry Smith       C->ops->solve   = MatSolve_SeqAIJ;
1120db4efbfdSBarry Smith     }
11218d8c7c43SBarry Smith   }
1122db4efbfdSBarry Smith   C->ops->solveadd           = MatSolveAdd_SeqAIJ;
1123db4efbfdSBarry Smith   C->ops->solvetranspose     = MatSolveTranspose_SeqAIJ;
1124db4efbfdSBarry Smith   C->ops->solvetransposeadd  = MatSolveTransposeAdd_SeqAIJ;
1125de26497bSBarry Smith   C->ops->matsolve           = MatMatSolve_SeqAIJ;
1126c456f294SBarry Smith   C->assembled    = PETSC_TRUE;
11275c9eb25fSBarry Smith   C->preallocated = PETSC_TRUE;
1128d0f46423SBarry Smith   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
1129d2276718SHong Zhang   if (sctx.nshift){
1130e738e422SBarry Smith      if (info->shiftpd) {
11310481f469SBarry 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);
1132e738e422SBarry Smith     } else if (info->shiftnz) {
1133e738e422SBarry Smith       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
11346cc28720Svictorle     }
11350697b6caSHong Zhang   }
11363a40ed3dSBarry Smith   PetscFunctionReturn(0);
1137289bc588SBarry Smith }
11380889b5dcSvictorle 
1139137fb511SHong Zhang /*
1140137fb511SHong Zhang    This routine implements inplace ILU(0) with row or/and column permutations.
1141137fb511SHong Zhang    Input:
1142137fb511SHong Zhang      A - original matrix
1143137fb511SHong Zhang    Output;
1144137fb511SHong Zhang      A - a->i (rowptr) is same as original rowptr, but factored i-the row is stored in rowperm[i]
1145137fb511SHong Zhang          a->j (col index) is permuted by the inverse of colperm, then sorted
1146137fb511SHong Zhang          a->a reordered accordingly with a->j
1147137fb511SHong Zhang          a->diag (ptr to diagonal elements) is updated.
1148137fb511SHong Zhang */
1149137fb511SHong Zhang #undef __FUNCT__
1150137fb511SHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_InplaceWithPerm"
11510481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_SeqAIJ_InplaceWithPerm(Mat B,Mat A,const MatFactorInfo *info)
1152137fb511SHong Zhang {
1153b051339dSHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
1154b051339dSHong Zhang   IS             isrow = a->row,isicol = a->icol;
1155137fb511SHong Zhang   PetscErrorCode ierr;
11565d0c19d7SBarry Smith   const PetscInt *r,*ic,*ics;
11575d0c19d7SBarry Smith   PetscInt       i,j,n=A->rmap->n,*ai=a->i,*aj=a->j;
11585d0c19d7SBarry Smith   PetscInt       *ajtmp,nz,row;
1159b051339dSHong Zhang   PetscInt       *diag = a->diag,nbdiag,*pj;
1160a77337e4SBarry Smith   PetscScalar    *rtmp,*pc,multiplier,d;
1161a77337e4SBarry Smith   MatScalar      *v,*pv;
1162137fb511SHong Zhang   PetscReal      rs;
1163137fb511SHong Zhang   LUShift_Ctx    sctx;
1164b051339dSHong Zhang   PetscInt       newshift;
1165137fb511SHong Zhang 
1166137fb511SHong Zhang   PetscFunctionBegin;
1167719d5645SBarry Smith   if (A != B) SETERRQ(PETSC_ERR_ARG_INCOMP,"input and output matrix must have same address");
1168137fb511SHong Zhang   ierr  = ISGetIndices(isrow,&r);CHKERRQ(ierr);
1169137fb511SHong Zhang   ierr  = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
1170137fb511SHong Zhang   ierr  = PetscMalloc((n+1)*sizeof(PetscScalar),&rtmp);CHKERRQ(ierr);
1171137fb511SHong Zhang   ierr  = PetscMemzero(rtmp,(n+1)*sizeof(PetscScalar));CHKERRQ(ierr);
1172b051339dSHong Zhang   ics = ic;
1173137fb511SHong Zhang 
1174137fb511SHong Zhang   sctx.shift_top      = 0;
1175137fb511SHong Zhang   sctx.nshift_max     = 0;
1176137fb511SHong Zhang   sctx.shift_lo       = 0;
1177137fb511SHong Zhang   sctx.shift_hi       = 0;
11780481f469SBarry Smith   sctx.shift_fraction = 0;
1179137fb511SHong Zhang 
1180137fb511SHong Zhang   /* if both shift schemes are chosen by user, only use info->shiftpd */
1181137fb511SHong Zhang   if (info->shiftpd) { /* set sctx.shift_top=max{rs} */
1182137fb511SHong Zhang     sctx.shift_top = 0;
1183137fb511SHong Zhang     for (i=0; i<n; i++) {
1184137fb511SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
1185b051339dSHong Zhang       d  = (a->a)[diag[i]];
1186137fb511SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
1187b051339dSHong Zhang       v  = a->a+ai[i];
1188b051339dSHong Zhang       nz = ai[i+1] - ai[i];
1189137fb511SHong Zhang       for (j=0; j<nz; j++)
1190137fb511SHong Zhang 	rs += PetscAbsScalar(v[j]);
1191137fb511SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
1192137fb511SHong Zhang     }
1193137fb511SHong Zhang     if (sctx.shift_top < info->zeropivot) sctx.shift_top = info->zeropivot;
1194137fb511SHong Zhang     sctx.shift_top    *= 1.1;
1195137fb511SHong Zhang     sctx.nshift_max   = 5;
1196137fb511SHong Zhang     sctx.shift_lo     = 0.;
1197137fb511SHong Zhang     sctx.shift_hi     = 1.;
1198137fb511SHong Zhang   }
1199137fb511SHong Zhang 
1200137fb511SHong Zhang   sctx.shift_amount = 0;
1201137fb511SHong Zhang   sctx.nshift       = 0;
1202137fb511SHong Zhang   do {
1203137fb511SHong Zhang     sctx.lushift = PETSC_FALSE;
1204137fb511SHong Zhang     for (i=0; i<n; i++){
1205b051339dSHong Zhang       /* load in initial unfactored row */
1206b051339dSHong Zhang       nz    = ai[r[i]+1] - ai[r[i]];
1207b051339dSHong Zhang       ajtmp = aj + ai[r[i]];
1208b051339dSHong Zhang       v     = a->a + ai[r[i]];
1209137fb511SHong Zhang       /* sort permuted ajtmp and values v accordingly */
1210b051339dSHong Zhang       for (j=0; j<nz; j++) ajtmp[j] = ics[ajtmp[j]];
1211137fb511SHong Zhang       ierr = PetscSortIntWithScalarArray(nz,ajtmp,v);CHKERRQ(ierr);
1212137fb511SHong Zhang 
1213b051339dSHong Zhang       diag[r[i]] = ai[r[i]];
1214137fb511SHong Zhang       for (j=0; j<nz; j++) {
1215137fb511SHong Zhang         rtmp[ajtmp[j]] = v[j];
1216b051339dSHong Zhang         if (ajtmp[j] < i) diag[r[i]]++; /* update a->diag */
1217137fb511SHong Zhang       }
1218137fb511SHong Zhang       rtmp[r[i]] += sctx.shift_amount; /* shift the diagonal of the matrix */
1219137fb511SHong Zhang 
1220b051339dSHong Zhang       row = *ajtmp++;
1221137fb511SHong Zhang       while  (row < i) {
1222137fb511SHong Zhang         pc = rtmp + row;
1223137fb511SHong Zhang         if (*pc != 0.0) {
1224b051339dSHong Zhang           pv         = a->a + diag[r[row]];
1225b051339dSHong Zhang           pj         = aj + diag[r[row]] + 1;
1226137fb511SHong Zhang 
1227137fb511SHong Zhang           multiplier = *pc / *pv++;
1228137fb511SHong Zhang           *pc        = multiplier;
1229b051339dSHong Zhang           nz         = ai[r[row]+1] - diag[r[row]] - 1;
1230b051339dSHong Zhang           for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j];
1231dc0b31edSSatish Balay           ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr);
1232137fb511SHong Zhang         }
1233b051339dSHong Zhang         row = *ajtmp++;
1234137fb511SHong Zhang       }
1235b051339dSHong Zhang       /* finished row so overwrite it onto a->a */
1236b051339dSHong Zhang       pv   = a->a + ai[r[i]] ;
1237b051339dSHong Zhang       pj   = aj + ai[r[i]] ;
1238b051339dSHong Zhang       nz   = ai[r[i]+1] - ai[r[i]];
1239b051339dSHong Zhang       nbdiag = diag[r[i]] - ai[r[i]]; /* num of entries before the diagonal */
1240137fb511SHong Zhang 
1241137fb511SHong Zhang       rs   = 0.0;
1242137fb511SHong Zhang       for (j=0; j<nz; j++) {
1243b051339dSHong Zhang         pv[j] = rtmp[pj[j]];
1244b051339dSHong Zhang         if (j != nbdiag) rs += PetscAbsScalar(pv[j]);
1245137fb511SHong Zhang       }
1246137fb511SHong Zhang 
1247137fb511SHong Zhang       /* 9/13/02 Victor Eijkhout suggested scaling zeropivot by rs for matrices with funny scalings */
1248137fb511SHong Zhang       sctx.rs  = rs;
1249b051339dSHong Zhang       sctx.pv  = pv[nbdiag];
1250c6b1f410SHong Zhang       ierr = MatLUCheckShift_inline(info,sctx,i,newshift);CHKERRQ(ierr);
1251137fb511SHong Zhang       if (newshift == 1) break;
1252137fb511SHong Zhang     }
1253137fb511SHong Zhang 
12540481f469SBarry Smith     if (info->shiftpd && !sctx.lushift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) {
1255137fb511SHong Zhang       /*
1256137fb511SHong Zhang        * if no shift in this attempt & shifting & started shifting & can refine,
1257137fb511SHong Zhang        * then try lower shift
1258137fb511SHong Zhang        */
12590481f469SBarry Smith       sctx.shift_hi        = sctx.shift_fraction;
12600481f469SBarry Smith       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
12610481f469SBarry Smith       sctx.shift_amount    = sctx.shift_fraction * sctx.shift_top;
1262137fb511SHong Zhang       sctx.lushift         = PETSC_TRUE;
1263137fb511SHong Zhang       sctx.nshift++;
1264137fb511SHong Zhang     }
1265137fb511SHong Zhang   } while (sctx.lushift);
1266137fb511SHong Zhang 
1267137fb511SHong Zhang   /* invert diagonal entries for simplier triangular solves */
1268137fb511SHong Zhang   for (i=0; i<n; i++) {
1269b051339dSHong Zhang     a->a[diag[r[i]]] = 1.0/a->a[diag[r[i]]];
1270137fb511SHong Zhang   }
1271137fb511SHong Zhang 
1272137fb511SHong Zhang   ierr = PetscFree(rtmp);CHKERRQ(ierr);
1273137fb511SHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
1274137fb511SHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
1275b051339dSHong Zhang   A->ops->solve             = MatSolve_SeqAIJ_InplaceWithPerm;
1276db4efbfdSBarry Smith   A->ops->solveadd          = MatSolveAdd_SeqAIJ;
1277db4efbfdSBarry Smith   A->ops->solvetranspose    = MatSolveTranspose_SeqAIJ;
1278db4efbfdSBarry Smith   A->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ;
1279b051339dSHong Zhang   A->assembled = PETSC_TRUE;
12805c9eb25fSBarry Smith   A->preallocated = PETSC_TRUE;
1281d0f46423SBarry Smith   ierr = PetscLogFlops(A->cmap->n);CHKERRQ(ierr);
1282137fb511SHong Zhang   if (sctx.nshift){
1283e738e422SBarry Smith     if (info->shiftpd) {
12840481f469SBarry 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);
1285e738e422SBarry Smith     } else if (info->shiftnz) {
1286e738e422SBarry Smith       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
1287137fb511SHong Zhang     }
1288137fb511SHong Zhang   }
1289137fb511SHong Zhang   PetscFunctionReturn(0);
1290137fb511SHong Zhang }
1291137fb511SHong Zhang 
12920a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
12934a2ae208SSatish Balay #undef __FUNCT__
12944a2ae208SSatish Balay #define __FUNCT__ "MatLUFactor_SeqAIJ"
12950481f469SBarry Smith PetscErrorCode MatLUFactor_SeqAIJ(Mat A,IS row,IS col,const MatFactorInfo *info)
1296da3a660dSBarry Smith {
1297dfbe8321SBarry Smith   PetscErrorCode ierr;
1298416022c9SBarry Smith   Mat            C;
1299416022c9SBarry Smith 
13003a40ed3dSBarry Smith   PetscFunctionBegin;
130143244d56SBarry Smith   ierr = MatGetFactor(A,MAT_SOLVER_PETSC,MAT_FACTOR_LU,&C);CHKERRQ(ierr);
1302719d5645SBarry Smith   ierr = MatLUFactorSymbolic(C,A,row,col,info);CHKERRQ(ierr);
1303719d5645SBarry Smith   ierr = MatLUFactorNumeric(C,A,info);CHKERRQ(ierr);
1304db4efbfdSBarry Smith   A->ops->solve            = C->ops->solve;
1305db4efbfdSBarry Smith   A->ops->solvetranspose   = C->ops->solvetranspose;
1306273d9f13SBarry Smith   ierr = MatHeaderCopy(A,C);CHKERRQ(ierr);
130752e6d16bSBarry Smith   ierr = PetscLogObjectParent(A,((Mat_SeqAIJ*)(A->data))->icol);CHKERRQ(ierr);
13083a40ed3dSBarry Smith   PetscFunctionReturn(0);
1309da3a660dSBarry Smith }
13100a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
13111d098868SHong Zhang 
1312889a8dedSBarry Smith 
13134a2ae208SSatish Balay #undef __FUNCT__
13144a2ae208SSatish Balay #define __FUNCT__ "MatSolve_SeqAIJ"
1315dfbe8321SBarry Smith PetscErrorCode MatSolve_SeqAIJ(Mat A,Vec bb,Vec xx)
13168c37ef55SBarry Smith {
1317416022c9SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
1318416022c9SBarry Smith   IS                iscol = a->col,isrow = a->row;
13196849ba73SBarry Smith   PetscErrorCode    ierr;
13205d0c19d7SBarry Smith   PetscInt          i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
13215d0c19d7SBarry Smith   PetscInt          nz;
13225d0c19d7SBarry Smith   const PetscInt    *rout,*cout,*r,*c;
1323dd6ea824SBarry Smith   PetscScalar       *x,*tmp,*tmps,sum;
1324d9fead3dSBarry Smith   const PetscScalar *b;
1325dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
13268c37ef55SBarry Smith 
13273a40ed3dSBarry Smith   PetscFunctionBegin;
13283a40ed3dSBarry Smith   if (!n) PetscFunctionReturn(0);
132991bf9adeSBarry Smith 
1330d9fead3dSBarry Smith   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
13311ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1332416022c9SBarry Smith   tmp  = a->solve_work;
13338c37ef55SBarry Smith 
13343b2fbd54SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
13353b2fbd54SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
13368c37ef55SBarry Smith 
13378c37ef55SBarry Smith   /* forward solve the lower triangular */
13388c37ef55SBarry Smith   tmp[0] = b[*r++];
1339010a20caSHong Zhang   tmps   = tmp;
13408c37ef55SBarry Smith   for (i=1; i<n; i++) {
1341010a20caSHong Zhang     v   = aa + ai[i] ;
1342010a20caSHong Zhang     vi  = aj + ai[i] ;
134353e7425aSBarry Smith     nz  = a->diag[i] - ai[i];
134453e7425aSBarry Smith     sum = b[*r++];
1345003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
13468c37ef55SBarry Smith     tmp[i] = sum;
13478c37ef55SBarry Smith   }
13488c37ef55SBarry Smith 
13498c37ef55SBarry Smith   /* backward solve the upper triangular */
13508c37ef55SBarry Smith   for (i=n-1; i>=0; i--){
1351010a20caSHong Zhang     v   = aa + a->diag[i] + 1;
1352010a20caSHong Zhang     vi  = aj + a->diag[i] + 1;
1353416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
13548c37ef55SBarry Smith     sum = tmp[i];
1355003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
1356010a20caSHong Zhang     x[*c--] = tmp[i] = sum*aa[a->diag[i]];
13578c37ef55SBarry Smith   }
13588c37ef55SBarry Smith 
13593b2fbd54SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
13603b2fbd54SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
1361d9fead3dSBarry Smith   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
13621ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1363dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
13643a40ed3dSBarry Smith   PetscFunctionReturn(0);
13658c37ef55SBarry Smith }
1366026e39d0SSatish Balay 
13672bebee5dSHong Zhang #undef __FUNCT__
13682bebee5dSHong Zhang #define __FUNCT__ "MatMatSolve_SeqAIJ"
13692bebee5dSHong Zhang PetscErrorCode MatMatSolve_SeqAIJ(Mat A,Mat B,Mat X)
13702bebee5dSHong Zhang {
13712bebee5dSHong Zhang   Mat_SeqAIJ      *a = (Mat_SeqAIJ*)A->data;
13722bebee5dSHong Zhang   IS              iscol = a->col,isrow = a->row;
13732bebee5dSHong Zhang   PetscErrorCode  ierr;
13745d0c19d7SBarry Smith   PetscInt        i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
13755d0c19d7SBarry Smith   PetscInt        nz,neq;
13765d0c19d7SBarry Smith   const PetscInt  *rout,*cout,*r,*c;
1377dd6ea824SBarry Smith   PetscScalar     *x,*b,*tmp,*tmps,sum;
1378dd6ea824SBarry Smith   const MatScalar *aa = a->a,*v;
1379db4efbfdSBarry Smith   PetscTruth      bisdense,xisdense;
13802bebee5dSHong Zhang 
13812bebee5dSHong Zhang   PetscFunctionBegin;
13822bebee5dSHong Zhang   if (!n) PetscFunctionReturn(0);
13832bebee5dSHong Zhang 
1384db4efbfdSBarry Smith   ierr = PetscTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr);
1385db4efbfdSBarry Smith   if (!bisdense) SETERRQ(PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix");
1386db4efbfdSBarry Smith   ierr = PetscTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr);
1387db4efbfdSBarry Smith   if (!xisdense) SETERRQ(PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix");
1388db4efbfdSBarry Smith 
13892bebee5dSHong Zhang   ierr = MatGetArray(B,&b);CHKERRQ(ierr);
13902bebee5dSHong Zhang   ierr = MatGetArray(X,&x);CHKERRQ(ierr);
13912bebee5dSHong Zhang 
13922bebee5dSHong Zhang   tmp  = a->solve_work;
13932bebee5dSHong Zhang   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
13942bebee5dSHong Zhang   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
13952bebee5dSHong Zhang 
1396a36b22ccSSatish Balay   for (neq=0; neq<B->cmap->n; neq++){
13972bebee5dSHong Zhang     /* forward solve the lower triangular */
13982bebee5dSHong Zhang     tmp[0] = b[r[0]];
13992bebee5dSHong Zhang     tmps   = tmp;
14002bebee5dSHong Zhang     for (i=1; i<n; i++) {
14012bebee5dSHong Zhang       v   = aa + ai[i] ;
14022bebee5dSHong Zhang       vi  = aj + ai[i] ;
14032bebee5dSHong Zhang       nz  = a->diag[i] - ai[i];
14042bebee5dSHong Zhang       sum = b[r[i]];
1405003131ecSBarry Smith       PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
14062bebee5dSHong Zhang       tmp[i] = sum;
14072bebee5dSHong Zhang     }
14082bebee5dSHong Zhang     /* backward solve the upper triangular */
14092bebee5dSHong Zhang     for (i=n-1; i>=0; i--){
14102bebee5dSHong Zhang       v   = aa + a->diag[i] + 1;
14112bebee5dSHong Zhang       vi  = aj + a->diag[i] + 1;
14122bebee5dSHong Zhang       nz  = ai[i+1] - a->diag[i] - 1;
14132bebee5dSHong Zhang       sum = tmp[i];
1414003131ecSBarry Smith       PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
14152bebee5dSHong Zhang       x[c[i]] = tmp[i] = sum*aa[a->diag[i]];
14162bebee5dSHong Zhang     }
14172bebee5dSHong Zhang 
14182bebee5dSHong Zhang     b += n;
14192bebee5dSHong Zhang     x += n;
14202bebee5dSHong Zhang   }
14212bebee5dSHong Zhang   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
14222bebee5dSHong Zhang   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
14232bebee5dSHong Zhang   ierr = MatRestoreArray(B,&b);CHKERRQ(ierr);
14242bebee5dSHong Zhang   ierr = MatRestoreArray(X,&x);CHKERRQ(ierr);
1425dc0b31edSSatish Balay   ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr);
14262bebee5dSHong Zhang   PetscFunctionReturn(0);
14272bebee5dSHong Zhang }
14282bebee5dSHong Zhang 
1429137fb511SHong Zhang #undef __FUNCT__
1430137fb511SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_InplaceWithPerm"
1431137fb511SHong Zhang PetscErrorCode MatSolve_SeqAIJ_InplaceWithPerm(Mat A,Vec bb,Vec xx)
1432137fb511SHong Zhang {
1433137fb511SHong Zhang   Mat_SeqAIJ      *a = (Mat_SeqAIJ*)A->data;
1434137fb511SHong Zhang   IS              iscol = a->col,isrow = a->row;
1435137fb511SHong Zhang   PetscErrorCode  ierr;
14365d0c19d7SBarry Smith   const PetscInt  *r,*c,*rout,*cout;
14375d0c19d7SBarry Smith   PetscInt        i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
14385d0c19d7SBarry Smith   PetscInt        nz,row;
1439dd6ea824SBarry Smith   PetscScalar     *x,*b,*tmp,*tmps,sum;
1440dd6ea824SBarry Smith   const MatScalar *aa = a->a,*v;
1441137fb511SHong Zhang 
1442137fb511SHong Zhang   PetscFunctionBegin;
1443137fb511SHong Zhang   if (!n) PetscFunctionReturn(0);
1444137fb511SHong Zhang 
1445137fb511SHong Zhang   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
1446137fb511SHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1447137fb511SHong Zhang   tmp  = a->solve_work;
1448137fb511SHong Zhang 
1449137fb511SHong Zhang   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
1450137fb511SHong Zhang   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
1451137fb511SHong Zhang 
1452137fb511SHong Zhang   /* forward solve the lower triangular */
1453137fb511SHong Zhang   tmp[0] = b[*r++];
1454137fb511SHong Zhang   tmps   = tmp;
1455137fb511SHong Zhang   for (row=1; row<n; row++) {
1456137fb511SHong Zhang     i   = rout[row]; /* permuted row */
1457137fb511SHong Zhang     v   = aa + ai[i] ;
1458137fb511SHong Zhang     vi  = aj + ai[i] ;
1459137fb511SHong Zhang     nz  = a->diag[i] - ai[i];
1460137fb511SHong Zhang     sum = b[*r++];
1461003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
1462137fb511SHong Zhang     tmp[row] = sum;
1463137fb511SHong Zhang   }
1464137fb511SHong Zhang 
1465137fb511SHong Zhang   /* backward solve the upper triangular */
1466137fb511SHong Zhang   for (row=n-1; row>=0; row--){
1467137fb511SHong Zhang     i   = rout[row]; /* permuted row */
1468137fb511SHong Zhang     v   = aa + a->diag[i] + 1;
1469137fb511SHong Zhang     vi  = aj + a->diag[i] + 1;
1470137fb511SHong Zhang     nz  = ai[i+1] - a->diag[i] - 1;
1471137fb511SHong Zhang     sum = tmp[row];
1472003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
1473137fb511SHong Zhang     x[*c--] = tmp[row] = sum*aa[a->diag[i]];
1474137fb511SHong Zhang   }
1475137fb511SHong Zhang 
1476137fb511SHong Zhang   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
1477137fb511SHong Zhang   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
1478137fb511SHong Zhang   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
1479137fb511SHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1480dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
1481137fb511SHong Zhang   PetscFunctionReturn(0);
1482137fb511SHong Zhang }
1483137fb511SHong Zhang 
1484930ae53cSBarry Smith /* ----------------------------------------------------------- */
1485a4005a5dSBarry Smith #include "../src/mat/impls/aij/seq/ftn-kernels/fsolve.h"
14864a2ae208SSatish Balay #undef __FUNCT__
14874a2ae208SSatish Balay #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering"
1488dfbe8321SBarry Smith PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering(Mat A,Vec bb,Vec xx)
1489930ae53cSBarry Smith {
1490930ae53cSBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
14916849ba73SBarry Smith   PetscErrorCode    ierr;
1492d0f46423SBarry Smith   PetscInt          n = A->rmap->n;
1493b377110cSBarry Smith   const PetscInt    *ai = a->i,*aj = a->j,*adiag = a->diag;
1494356650c2SBarry Smith   PetscScalar       *x;
1495356650c2SBarry Smith   const PetscScalar *b;
1496dd6ea824SBarry Smith   const MatScalar   *aa = a->a;
1497aa482453SBarry Smith #if !defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ)
1498356650c2SBarry Smith   PetscInt          adiag_i,i,nz,ai_i;
1499b377110cSBarry Smith   const PetscInt    *vi;
1500dd6ea824SBarry Smith   const MatScalar   *v;
1501dd6ea824SBarry Smith   PetscScalar       sum;
1502d85afc42SSatish Balay #endif
1503930ae53cSBarry Smith 
15043a40ed3dSBarry Smith   PetscFunctionBegin;
15053a40ed3dSBarry Smith   if (!n) PetscFunctionReturn(0);
1506930ae53cSBarry Smith 
1507356650c2SBarry Smith   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
15081ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1509930ae53cSBarry Smith 
1510aa482453SBarry Smith #if defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ)
15111c4feecaSSatish Balay   fortransolveaij_(&n,x,ai,aj,adiag,aa,b);
15121c4feecaSSatish Balay #else
1513930ae53cSBarry Smith   /* forward solve the lower triangular */
1514930ae53cSBarry Smith   x[0] = b[0];
1515930ae53cSBarry Smith   for (i=1; i<n; i++) {
1516930ae53cSBarry Smith     ai_i = ai[i];
1517930ae53cSBarry Smith     v    = aa + ai_i;
1518930ae53cSBarry Smith     vi   = aj + ai_i;
1519930ae53cSBarry Smith     nz   = adiag[i] - ai_i;
1520930ae53cSBarry Smith     sum  = b[i];
1521003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
1522930ae53cSBarry Smith     x[i] = sum;
1523930ae53cSBarry Smith   }
1524930ae53cSBarry Smith 
1525930ae53cSBarry Smith   /* backward solve the upper triangular */
1526930ae53cSBarry Smith   for (i=n-1; i>=0; i--){
1527930ae53cSBarry Smith     adiag_i = adiag[i];
1528d708749eSBarry Smith     v       = aa + adiag_i + 1;
1529d708749eSBarry Smith     vi      = aj + adiag_i + 1;
1530930ae53cSBarry Smith     nz      = ai[i+1] - adiag_i - 1;
1531930ae53cSBarry Smith     sum     = x[i];
1532003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
1533930ae53cSBarry Smith     x[i]    = sum*aa[adiag_i];
1534930ae53cSBarry Smith   }
15351c4feecaSSatish Balay #endif
1536dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
1537356650c2SBarry Smith   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
15381ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
15393a40ed3dSBarry Smith   PetscFunctionReturn(0);
1540930ae53cSBarry Smith }
1541930ae53cSBarry Smith 
15424a2ae208SSatish Balay #undef __FUNCT__
15434a2ae208SSatish Balay #define __FUNCT__ "MatSolveAdd_SeqAIJ"
1544dfbe8321SBarry Smith PetscErrorCode MatSolveAdd_SeqAIJ(Mat A,Vec bb,Vec yy,Vec xx)
1545da3a660dSBarry Smith {
1546416022c9SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
1547416022c9SBarry Smith   IS                iscol = a->col,isrow = a->row;
15486849ba73SBarry Smith   PetscErrorCode    ierr;
154904fbf559SBarry Smith   PetscInt          i, n = A->rmap->n,j;
15505d0c19d7SBarry Smith   PetscInt          nz;
155104fbf559SBarry Smith   const PetscInt    *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j;
155204fbf559SBarry Smith   PetscScalar       *x,*tmp,sum;
155304fbf559SBarry Smith   const PetscScalar *b;
1554dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
1555da3a660dSBarry Smith 
15563a40ed3dSBarry Smith   PetscFunctionBegin;
155778b31e54SBarry Smith   if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);}
1558da3a660dSBarry Smith 
155904fbf559SBarry Smith   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
15601ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1561416022c9SBarry Smith   tmp  = a->solve_work;
1562da3a660dSBarry Smith 
15633b2fbd54SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
15643b2fbd54SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
1565da3a660dSBarry Smith 
1566da3a660dSBarry Smith   /* forward solve the lower triangular */
1567da3a660dSBarry Smith   tmp[0] = b[*r++];
1568da3a660dSBarry Smith   for (i=1; i<n; i++) {
1569010a20caSHong Zhang     v   = aa + ai[i] ;
1570010a20caSHong Zhang     vi  = aj + ai[i] ;
1571416022c9SBarry Smith     nz  = a->diag[i] - ai[i];
1572da3a660dSBarry Smith     sum = b[*r++];
157304fbf559SBarry Smith     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
1574da3a660dSBarry Smith     tmp[i] = sum;
1575da3a660dSBarry Smith   }
1576da3a660dSBarry Smith 
1577da3a660dSBarry Smith   /* backward solve the upper triangular */
1578da3a660dSBarry Smith   for (i=n-1; i>=0; i--){
1579010a20caSHong Zhang     v   = aa + a->diag[i] + 1;
1580010a20caSHong Zhang     vi  = aj + a->diag[i] + 1;
1581416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
1582da3a660dSBarry Smith     sum = tmp[i];
158304fbf559SBarry Smith     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
1584010a20caSHong Zhang     tmp[i] = sum*aa[a->diag[i]];
1585da3a660dSBarry Smith     x[*c--] += tmp[i];
1586da3a660dSBarry Smith   }
1587da3a660dSBarry Smith 
15883b2fbd54SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
15893b2fbd54SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
159004fbf559SBarry Smith   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
15911ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1592dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
1593898183ecSLois Curfman McInnes 
15943a40ed3dSBarry Smith   PetscFunctionReturn(0);
1595da3a660dSBarry Smith }
159604fbf559SBarry Smith 
15974a2ae208SSatish Balay #undef __FUNCT__
15984a2ae208SSatish Balay #define __FUNCT__ "MatSolveTranspose_SeqAIJ"
1599dfbe8321SBarry Smith PetscErrorCode MatSolveTranspose_SeqAIJ(Mat A,Vec bb,Vec xx)
1600da3a660dSBarry Smith {
1601416022c9SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
16022235e667SBarry Smith   IS                iscol = a->col,isrow = a->row;
16036849ba73SBarry Smith   PetscErrorCode    ierr;
160404fbf559SBarry Smith   const PetscInt    *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi;
160504fbf559SBarry Smith   PetscInt          i,n = A->rmap->n,j;
160604fbf559SBarry Smith   PetscInt          nz;
160704fbf559SBarry Smith   PetscScalar       *x,*tmp,s1;
1608dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
160904fbf559SBarry Smith   const PetscScalar *b;
1610da3a660dSBarry Smith 
16113a40ed3dSBarry Smith   PetscFunctionBegin;
161204fbf559SBarry Smith   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
16131ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1614416022c9SBarry Smith   tmp  = a->solve_work;
1615da3a660dSBarry Smith 
16162235e667SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
16172235e667SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
1618da3a660dSBarry Smith 
1619da3a660dSBarry Smith   /* copy the b into temp work space according to permutation */
16202235e667SBarry Smith   for (i=0; i<n; i++) tmp[i] = b[c[i]];
1621da3a660dSBarry Smith 
1622da3a660dSBarry Smith   /* forward solve the U^T */
1623da3a660dSBarry Smith   for (i=0; i<n; i++) {
1624010a20caSHong Zhang     v   = aa + diag[i] ;
1625010a20caSHong Zhang     vi  = aj + diag[i] + 1;
1626f1af5d2fSBarry Smith     nz  = ai[i+1] - diag[i] - 1;
1627c38d4ed2SBarry Smith     s1  = tmp[i];
1628ef66eb69SBarry Smith     s1 *= (*v++);  /* multiply by inverse of diagonal entry */
162904fbf559SBarry Smith     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1630c38d4ed2SBarry Smith     tmp[i] = s1;
1631da3a660dSBarry Smith   }
1632da3a660dSBarry Smith 
1633da3a660dSBarry Smith   /* backward solve the L^T */
1634da3a660dSBarry Smith   for (i=n-1; i>=0; i--){
1635010a20caSHong Zhang     v   = aa + diag[i] - 1 ;
1636010a20caSHong Zhang     vi  = aj + diag[i] - 1 ;
1637f1af5d2fSBarry Smith     nz  = diag[i] - ai[i];
1638f1af5d2fSBarry Smith     s1  = tmp[i];
163904fbf559SBarry Smith     for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j];
1640da3a660dSBarry Smith   }
1641da3a660dSBarry Smith 
1642da3a660dSBarry Smith   /* copy tmp into x according to permutation */
1643591294e4SBarry Smith   for (i=0; i<n; i++) x[r[i]] = tmp[i];
1644da3a660dSBarry Smith 
16452235e667SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
16462235e667SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
164704fbf559SBarry Smith   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
16481ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1649da3a660dSBarry Smith 
1650dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr);
16513a40ed3dSBarry Smith   PetscFunctionReturn(0);
1652da3a660dSBarry Smith }
1653da3a660dSBarry Smith 
16544a2ae208SSatish Balay #undef __FUNCT__
16554a2ae208SSatish Balay #define __FUNCT__ "MatSolveTransposeAdd_SeqAIJ"
1656dfbe8321SBarry Smith PetscErrorCode MatSolveTransposeAdd_SeqAIJ(Mat A,Vec bb,Vec zz,Vec xx)
1657da3a660dSBarry Smith {
1658416022c9SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
16592235e667SBarry Smith   IS                iscol = a->col,isrow = a->row;
16606849ba73SBarry Smith   PetscErrorCode    ierr;
166104fbf559SBarry Smith   const PetscInt    *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi;
166204fbf559SBarry Smith   PetscInt          i,n = A->rmap->n,j;
166304fbf559SBarry Smith   PetscInt          nz;
166404fbf559SBarry Smith   PetscScalar       *x,*tmp,s1;
1665dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
166604fbf559SBarry Smith   const PetscScalar *b;
16676abc6512SBarry Smith 
16683a40ed3dSBarry Smith   PetscFunctionBegin;
166923bc6035SMatthew Knepley   if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);}
167004fbf559SBarry Smith   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
16711ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1672416022c9SBarry Smith   tmp  = a->solve_work;
16736abc6512SBarry Smith 
16742235e667SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
16752235e667SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
16766abc6512SBarry Smith 
16776abc6512SBarry Smith   /* copy the b into temp work space according to permutation */
16782235e667SBarry Smith   for (i=0; i<n; i++) tmp[i] = b[c[i]];
16796abc6512SBarry Smith 
16806abc6512SBarry Smith   /* forward solve the U^T */
16816abc6512SBarry Smith   for (i=0; i<n; i++) {
1682010a20caSHong Zhang     v   = aa + diag[i] ;
1683010a20caSHong Zhang     vi  = aj + diag[i] + 1;
1684f1af5d2fSBarry Smith     nz  = ai[i+1] - diag[i] - 1;
168504fbf559SBarry Smith     s1  = tmp[i];
168604fbf559SBarry Smith     s1 *= (*v++);  /* multiply by inverse of diagonal entry */
168704fbf559SBarry Smith     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
168804fbf559SBarry Smith     tmp[i] = s1;
16896abc6512SBarry Smith   }
16906abc6512SBarry Smith 
16916abc6512SBarry Smith   /* backward solve the L^T */
16926abc6512SBarry Smith   for (i=n-1; i>=0; i--){
1693010a20caSHong Zhang     v   = aa + diag[i] - 1 ;
1694010a20caSHong Zhang     vi  = aj + diag[i] - 1 ;
1695f1af5d2fSBarry Smith     nz  = diag[i] - ai[i];
169604fbf559SBarry Smith     s1  = tmp[i];
169704fbf559SBarry Smith     for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j];
16986abc6512SBarry Smith   }
16996abc6512SBarry Smith 
17006abc6512SBarry Smith   /* copy tmp into x according to permutation */
17016abc6512SBarry Smith   for (i=0; i<n; i++) x[r[i]] += tmp[i];
17026abc6512SBarry Smith 
17032235e667SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
17042235e667SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
170504fbf559SBarry Smith   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
17061ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
17076abc6512SBarry Smith 
170804fbf559SBarry Smith   ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr);
17093a40ed3dSBarry Smith   PetscFunctionReturn(0);
1710da3a660dSBarry Smith }
171104fbf559SBarry Smith 
17129e25ed09SBarry Smith /* ----------------------------------------------------------------*/
17133c5fc038SBarry Smith EXTERN PetscErrorCode Mat_CheckInode(Mat,PetscTruth);
1714f77e22a1SHong Zhang EXTERN PetscErrorCode MatDuplicateNoCreate_SeqAIJ(Mat,Mat,MatDuplicateOption,PetscTruth);
171508480c60SBarry Smith 
17168fc3a347SHong Zhang /*
17178fc3a347SHong Zhang    ilu(0) with natural ordering under new data structure.
17188fc3a347SHong Zhang    Factored arrays bj and ba are stored as
17198fc3a347SHong Zhang      L(0,:), L(1,:), ...,L(n-1,:),  U(n-1,:),...,U(i,:),U(i-1,:),...,U(0,:)
17208fc3a347SHong Zhang 
17218fc3a347SHong Zhang    bi=fact->i is an array of size 2n+2, in which
17228fc3a347SHong Zhang    bi+
17238fc3a347SHong Zhang      bi[i]      ->  1st entry of L(i,:),i=0,...,i-1
1724f77e22a1SHong Zhang      bi[n]      ->  points to L(n-1,:)+1
17258fc3a347SHong Zhang      bi[n+1]    ->  1st entry of U(n-1,:)
17268fc3a347SHong Zhang      bi[2n-i]   ->  1st entry of U(i,:)
17278fc3a347SHong Zhang      bi[2n-i+1] ->  end of U(i,:)+1, the 1st entry of U(i-1,:)
1728f77e22a1SHong Zhang      bi[2n]     ->  1st entry of U(0,:)
1729f77e22a1SHong Zhang      bi[2n+1]   ->  points to U(0,:)+1
17308fc3a347SHong Zhang 
17318fc3a347SHong Zhang    U(i,:) contains diag[i] as its last entry, i.e.,
17328fc3a347SHong Zhang     U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i])
17338fc3a347SHong Zhang */
1734c9c72f8fSHong Zhang #undef __FUNCT__
17358fc3a347SHong Zhang #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_ilu0_newdatastruct"
17368fc3a347SHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ_ilu0_newdatastruct(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
1737c9c72f8fSHong Zhang {
1738c9c72f8fSHong Zhang 
1739c9c72f8fSHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
1740c9c72f8fSHong Zhang   PetscErrorCode     ierr;
1741c9c72f8fSHong Zhang   PetscInt           n=A->rmap->n,*ai=a->i,*aj,*adiag=a->diag;
17428fc3a347SHong Zhang   PetscInt           i,j,nz,*bi,*bj,*bdiag;
1743c9c72f8fSHong Zhang 
1744c9c72f8fSHong Zhang   PetscFunctionBegin;
1745f77e22a1SHong Zhang   ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_FALSE);CHKERRQ(ierr);
1746c9c72f8fSHong Zhang   b    = (Mat_SeqAIJ*)(fact)->data;
1747c9c72f8fSHong Zhang 
1748f77e22a1SHong Zhang   /* allocate matrix arrays for new data structure */
1749f77e22a1SHong Zhang   ierr = PetscMalloc3(ai[n]+1,PetscScalar,&b->a,ai[n]+1,PetscInt,&b->j,2*n+2,PetscInt,&b->i);CHKERRQ(ierr);
1750f77e22a1SHong Zhang   ierr = PetscLogObjectMemory(fact,ai[n]*(sizeof(PetscScalar)+sizeof(PetscInt))+(2*n+2)*sizeof(PetscInt));CHKERRQ(ierr);
1751f77e22a1SHong Zhang   b->singlemalloc = PETSC_TRUE;
1752f77e22a1SHong Zhang   if (!b->diag){
17538fc3a347SHong Zhang     ierr = PetscMalloc((n+1)*sizeof(PetscInt),&b->diag);CHKERRQ(ierr);
1754f77e22a1SHong Zhang   }
1755c9c72f8fSHong Zhang   bdiag = b->diag;
1756c9c72f8fSHong Zhang 
1757c9c72f8fSHong Zhang   if (n > 0) {
175816a2bf60SHong Zhang     ierr = PetscMemzero(b->a,(ai[n])*sizeof(MatScalar));CHKERRQ(ierr);
1759c9c72f8fSHong Zhang   }
1760c9c72f8fSHong Zhang 
1761c9c72f8fSHong Zhang   /* set bi and bj with new data structure */
1762c9c72f8fSHong Zhang   bi = b->i;
1763c9c72f8fSHong Zhang   bj = b->j;
1764c9c72f8fSHong Zhang 
1765c9c72f8fSHong Zhang   /* L part */
1766c9c72f8fSHong Zhang   bi[0] = 0;
1767c9c72f8fSHong Zhang   for (i=0; i<n; i++){
1768c9c72f8fSHong Zhang     nz = adiag[i] - ai[i];
1769c9c72f8fSHong Zhang     bi[i+1] = bi[i] + nz;
1770c9c72f8fSHong Zhang     aj = a->j + ai[i];
1771c9c72f8fSHong Zhang     for (j=0; j<nz; j++){
1772c9c72f8fSHong Zhang       *bj = aj[j]; bj++;
1773c9c72f8fSHong Zhang     }
1774c9c72f8fSHong Zhang   }
1775c9c72f8fSHong Zhang 
1776c9c72f8fSHong Zhang   /* U part */
17778fc3a347SHong Zhang   bi[n+1] = bi[n];
1778c9c72f8fSHong Zhang   for (i=n-1; i>=0; i--){
1779c9c72f8fSHong Zhang     nz = ai[i+1] - adiag[i] - 1;
17808fc3a347SHong Zhang     bi[2*n-i+1] = bi[2*n-i] + nz + 1;
1781c9c72f8fSHong Zhang     aj = a->j + adiag[i] + 1;
1782c9c72f8fSHong Zhang     for (j=0; j<nz; j++){
1783c9c72f8fSHong Zhang       *bj = aj[j]; bj++;
1784c9c72f8fSHong Zhang     }
1785c9c72f8fSHong Zhang     /* diag[i] */
1786c9c72f8fSHong Zhang     *bj = i; bj++;
17878fc3a347SHong Zhang     bdiag[i] = bi[2*n-i+1]-1;
1788c9c72f8fSHong Zhang   }
1789c9c72f8fSHong Zhang   PetscFunctionReturn(0);
1790c9c72f8fSHong Zhang }
1791c9c72f8fSHong Zhang 
17924a2ae208SSatish Balay #undef __FUNCT__
17936516239fSHong Zhang #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_newdatastruct"
17946516239fSHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ_newdatastruct(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
17959e25ed09SBarry Smith {
1796416022c9SBarry Smith   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
17979e25ed09SBarry Smith   IS                 isicol;
17986849ba73SBarry Smith   PetscErrorCode     ierr;
17995d0c19d7SBarry Smith   const PetscInt     *r,*ic;
18005d0c19d7SBarry Smith   PetscInt           n=A->rmap->n,*ai=a->i,*aj=a->j,d;
18015a8e39fbSHong Zhang   PetscInt           *bi,*cols,nnz,*cols_lvl;
1802052f0c41SBarry Smith   PetscInt           *bdiag,prow,fm,nzbd,reallocs=0,dcount=0;
18035a8e39fbSHong Zhang   PetscInt           i,levels,diagonal_fill;
180477c4ece6SBarry Smith   PetscTruth         col_identity,row_identity;
1805329f5518SBarry Smith   PetscReal          f;
18065a8e39fbSHong Zhang   PetscInt           nlnk,*lnk,*lnk_lvl=PETSC_NULL;
18075a8e39fbSHong Zhang   PetscBT            lnkbt;
18085a8e39fbSHong Zhang   PetscInt           nzi,*bj,**bj_ptr,**bjlvl_ptr;
1809a1a86e44SBarry Smith   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
1810a1a86e44SBarry Smith   PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL;
181109f38230SBarry Smith   PetscTruth         missing;
18129e25ed09SBarry Smith 
18133a40ed3dSBarry Smith   PetscFunctionBegin;
1814d0f46423SBarry Smith   if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n);
18156cf997b0SBarry Smith   f             = info->fill;
181638baddfdSBarry Smith   levels        = (PetscInt)info->levels;
181738baddfdSBarry Smith   diagonal_fill = (PetscInt)info->diagonal_fill;
18184c49b128SBarry Smith   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
181982bf6240SBarry Smith 
1820be0abb6dSBarry Smith   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
1821be0abb6dSBarry Smith   ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr);
1822c9c72f8fSHong Zhang 
18236516239fSHong Zhang   if (!levels && row_identity && col_identity) {
18246516239fSHong Zhang     /* special case: ilu(0) with natural ordering */
18258fc3a347SHong Zhang     ierr = MatILUFactorSymbolic_SeqAIJ_ilu0_newdatastruct(fact,A,isrow,iscol,info);CHKERRQ(ierr);
1826c9c72f8fSHong Zhang     (fact)->ops->lufactornumeric =  MatLUFactorNumeric_SeqAIJ_newdatastruct;
18276516239fSHong Zhang 
18286516239fSHong Zhang     fact->factor = MAT_FACTOR_ILU;
18296516239fSHong Zhang     (fact)->info.factor_mallocs    = 0;
18306516239fSHong Zhang     (fact)->info.fill_ratio_given  = info->fill;
18316516239fSHong Zhang     (fact)->info.fill_ratio_needed = 1.0;
18326516239fSHong Zhang     b               = (Mat_SeqAIJ*)(fact)->data;
18336516239fSHong Zhang     ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr);
18346516239fSHong Zhang     if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d);
18356516239fSHong Zhang     b->row              = isrow;
18366516239fSHong Zhang     b->col              = iscol;
18376516239fSHong Zhang     b->icol             = isicol;
18386516239fSHong Zhang     ierr                = PetscMalloc(((fact)->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
18396516239fSHong Zhang     ierr                = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
18406516239fSHong Zhang     ierr                = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
18414108e4d5SBarry Smith     /* ierr = MatILUFactorSymbolic_SeqAIJ_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr); */
18426516239fSHong Zhang     PetscFunctionReturn(0);
18436516239fSHong Zhang   }
18446516239fSHong Zhang 
18456516239fSHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
18466516239fSHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
18476516239fSHong Zhang 
18486516239fSHong Zhang   /* get new row pointers */
18496516239fSHong Zhang   ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
18506516239fSHong Zhang   bi[0] = 0;
18516516239fSHong Zhang   /* bdiag is location of diagonal in factor */
18526516239fSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
18536516239fSHong Zhang   bdiag[0]  = 0;
18546516239fSHong Zhang 
18556516239fSHong Zhang   ierr = PetscMalloc((2*n+1)*sizeof(PetscInt**),&bj_ptr);CHKERRQ(ierr);
18566516239fSHong Zhang   bjlvl_ptr = (PetscInt**)(bj_ptr + n);
18576516239fSHong Zhang 
18586516239fSHong Zhang   /* create a linked list for storing column indices of the active row */
18596516239fSHong Zhang   nlnk = n + 1;
18606516239fSHong Zhang   ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
18616516239fSHong Zhang 
18626516239fSHong Zhang   /* initial FreeSpace size is f*(ai[n]+1) */
18636516239fSHong Zhang   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr);
18646516239fSHong Zhang   current_space = free_space;
18656516239fSHong Zhang   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr);
18666516239fSHong Zhang   current_space_lvl = free_space_lvl;
18676516239fSHong Zhang 
18686516239fSHong Zhang   for (i=0; i<n; i++) {
18696516239fSHong Zhang     nzi = 0;
18706516239fSHong Zhang     /* copy current row into linked list */
18716516239fSHong Zhang     nnz  = ai[r[i]+1] - ai[r[i]];
18726516239fSHong Zhang     if (!nnz) SETERRQ2(PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i);
18736516239fSHong Zhang     cols = aj + ai[r[i]];
18746516239fSHong Zhang     lnk[i] = -1; /* marker to indicate if diagonal exists */
18756516239fSHong Zhang     ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
18766516239fSHong Zhang     nzi += nlnk;
18776516239fSHong Zhang 
18786516239fSHong Zhang     /* make sure diagonal entry is included */
18796516239fSHong Zhang     if (diagonal_fill && lnk[i] == -1) {
18806516239fSHong Zhang       fm = n;
18816516239fSHong Zhang       while (lnk[fm] < i) fm = lnk[fm];
18826516239fSHong Zhang       lnk[i]     = lnk[fm]; /* insert diagonal into linked list */
18836516239fSHong Zhang       lnk[fm]    = i;
18846516239fSHong Zhang       lnk_lvl[i] = 0;
18856516239fSHong Zhang       nzi++; dcount++;
18866516239fSHong Zhang     }
18876516239fSHong Zhang 
18886516239fSHong Zhang     /* add pivot rows into the active row */
18896516239fSHong Zhang     nzbd = 0;
18906516239fSHong Zhang     prow = lnk[n];
18916516239fSHong Zhang     while (prow < i) {
18926516239fSHong Zhang       nnz      = bdiag[prow];
18936516239fSHong Zhang       cols     = bj_ptr[prow] + nnz + 1;
18946516239fSHong Zhang       cols_lvl = bjlvl_ptr[prow] + nnz + 1;
18956516239fSHong Zhang       nnz      = bi[prow+1] - bi[prow] - nnz - 1;
18966516239fSHong Zhang       ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr);
18976516239fSHong Zhang       nzi += nlnk;
18986516239fSHong Zhang       prow = lnk[prow];
18996516239fSHong Zhang       nzbd++;
19006516239fSHong Zhang     }
19016516239fSHong Zhang     bdiag[i] = nzbd;
19026516239fSHong Zhang     bi[i+1]  = bi[i] + nzi;
19036516239fSHong Zhang 
19046516239fSHong Zhang     /* if free space is not available, make more free space */
19056516239fSHong Zhang     if (current_space->local_remaining<nzi) {
19066516239fSHong Zhang       nnz = 2*nzi*(n - i); /* estimated and max additional space needed */
19076516239fSHong Zhang       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
19086516239fSHong Zhang       ierr = PetscFreeSpaceGet(nnz,&current_space_lvl);CHKERRQ(ierr);
19096516239fSHong Zhang       reallocs++;
19106516239fSHong Zhang     }
19116516239fSHong Zhang 
19126516239fSHong Zhang     /* copy data into free_space and free_space_lvl, then initialize lnk */
19136516239fSHong Zhang     ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
19146516239fSHong Zhang     bj_ptr[i]    = current_space->array;
19156516239fSHong Zhang     bjlvl_ptr[i] = current_space_lvl->array;
19166516239fSHong Zhang 
19176516239fSHong Zhang     /* make sure the active row i has diagonal entry */
19186516239fSHong Zhang     if (*(bj_ptr[i]+bdiag[i]) != i) {
19196516239fSHong Zhang       SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,"Row %D has missing diagonal in factored matrix\n\
19206516239fSHong Zhang     try running with -pc_factor_nonzeros_along_diagonal or -pc_factor_diagonal_fill",i);
19216516239fSHong Zhang     }
19226516239fSHong Zhang 
19236516239fSHong Zhang     current_space->array           += nzi;
19246516239fSHong Zhang     current_space->local_used      += nzi;
19256516239fSHong Zhang     current_space->local_remaining -= nzi;
19266516239fSHong Zhang     current_space_lvl->array           += nzi;
19276516239fSHong Zhang     current_space_lvl->local_used      += nzi;
19286516239fSHong Zhang     current_space_lvl->local_remaining -= nzi;
19296516239fSHong Zhang   }
19306516239fSHong Zhang 
19316516239fSHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
19326516239fSHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
19336516239fSHong Zhang 
19346516239fSHong Zhang   /* destroy list of free space and other temporary arrays */
19356516239fSHong Zhang   ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
19366516239fSHong Zhang 
19376516239fSHong Zhang   /* copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */
1938783ef271SHong Zhang   ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr);
19396516239fSHong Zhang 
19406516239fSHong Zhang   ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
19416516239fSHong Zhang   ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
19426516239fSHong Zhang   ierr = PetscFree(bj_ptr);CHKERRQ(ierr);
19436516239fSHong Zhang 
19446516239fSHong Zhang #if defined(PETSC_USE_INFO)
19456516239fSHong Zhang   {
19466516239fSHong Zhang     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
19476516239fSHong Zhang     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr);
19486516239fSHong Zhang     ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
19496516239fSHong Zhang     ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr);
19506516239fSHong Zhang     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
19516516239fSHong Zhang     if (diagonal_fill) {
19526516239fSHong Zhang       ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr);
19536516239fSHong Zhang     }
19546516239fSHong Zhang   }
19556516239fSHong Zhang #endif
19566516239fSHong Zhang 
19576516239fSHong Zhang   /* put together the new matrix */
19586516239fSHong Zhang   ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
19596516239fSHong Zhang   ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr);
19606516239fSHong Zhang   b = (Mat_SeqAIJ*)(fact)->data;
19616516239fSHong Zhang   b->free_a       = PETSC_TRUE;
19626516239fSHong Zhang   b->free_ij      = PETSC_TRUE;
19636516239fSHong Zhang   b->singlemalloc = PETSC_FALSE;
19646516239fSHong Zhang   ierr = PetscMalloc( (bi[2*n+1] )*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);
19656516239fSHong Zhang   b->j          = bj;
19666516239fSHong Zhang   b->i          = bi;
19676516239fSHong Zhang   b->diag       = bdiag;
19686516239fSHong Zhang   b->ilen       = 0;
19696516239fSHong Zhang   b->imax       = 0;
19706516239fSHong Zhang   b->row        = isrow;
19716516239fSHong Zhang   b->col        = iscol;
19726516239fSHong Zhang   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
19736516239fSHong Zhang   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
19746516239fSHong Zhang   b->icol       = isicol;
19756516239fSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
19766516239fSHong Zhang   /* In b structure:  Free imax, ilen, old a, old j.
19776516239fSHong Zhang      Allocate bdiag, solve_work, new a, new j */
19786516239fSHong Zhang   ierr = PetscLogObjectMemory(fact,bi[2*n+1] * (sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
19796516239fSHong Zhang   b->maxnz = b->nz = bi[2*n+1] ;
19806516239fSHong Zhang   (fact)->info.factor_mallocs    = reallocs;
19816516239fSHong Zhang   (fact)->info.fill_ratio_given  = f;
19826516239fSHong Zhang   (fact)->info.fill_ratio_needed = ((PetscReal)bi[2*n+1])/((PetscReal)ai[n]);
19836516239fSHong Zhang   (fact)->ops->lufactornumeric   = MatLUFactorNumeric_SeqAIJ_newdatastruct;
19844108e4d5SBarry Smith   /* ierr = MatILUFactorSymbolic_SeqAIJ_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr); */
19856516239fSHong Zhang   PetscFunctionReturn(0);
19866516239fSHong Zhang }
19876516239fSHong Zhang 
1988813a1f2bSShri Abhyankar 
1989813a1f2bSShri Abhyankar /*
1990*8a76ed4fSHong Zhang    ilu() under revised new data structure.
1991813a1f2bSShri Abhyankar    Factored arrays bj and ba are stored as
1992813a1f2bSShri Abhyankar      L(0,:), L(1,:), ...,L(n-1,:),  U(n-1,:),...,U(i,:),U(i-1,:),...,U(0,:)
1993813a1f2bSShri Abhyankar 
1994813a1f2bSShri Abhyankar    bi=fact->i is an array of size n+1, in which
1995813a1f2bSShri Abhyankar    bi+
1996baabb860SHong Zhang      bi[i]:  points to 1st entry of L(i,:),i=0,...,n-1
1997baabb860SHong Zhang      bi[n]:  points to L(n-1,n-1)+1
1998baabb860SHong Zhang 
1999813a1f2bSShri Abhyankar   bdiag=fact->diag is an array of size n+1,in which
2000baabb860SHong Zhang      bdiag[i]: points to diagonal of U(i,:), i=0,...,n-1
2001baabb860SHong Zhang      bdiag[n]: points to diagonal of U(n-1,0)-1
2002813a1f2bSShri Abhyankar 
2003813a1f2bSShri Abhyankar    U(i,:) contains bdiag[i] as its last entry, i.e.,
2004813a1f2bSShri Abhyankar     U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i])
2005813a1f2bSShri Abhyankar */
2006813a1f2bSShri Abhyankar #undef __FUNCT__
2007813a1f2bSShri Abhyankar #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_ilu0_newdatastruct_v2"
2008813a1f2bSShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ_ilu0_newdatastruct_v2(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
2009813a1f2bSShri Abhyankar {
2010813a1f2bSShri Abhyankar 
2011813a1f2bSShri Abhyankar   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
2012813a1f2bSShri Abhyankar   PetscErrorCode     ierr;
2013813a1f2bSShri Abhyankar   PetscInt           n=A->rmap->n,*ai=a->i,*aj,*adiag=a->diag;
2014813a1f2bSShri Abhyankar   PetscInt           i,j,nz,*bi,*bj,*bdiag,bi_temp;
2015813a1f2bSShri Abhyankar 
2016813a1f2bSShri Abhyankar   PetscFunctionBegin;
2017813a1f2bSShri Abhyankar   ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_FALSE);CHKERRQ(ierr);
2018813a1f2bSShri Abhyankar   b    = (Mat_SeqAIJ*)(fact)->data;
2019813a1f2bSShri Abhyankar 
2020813a1f2bSShri Abhyankar   /* allocate matrix arrays for new data structure */
2021813a1f2bSShri Abhyankar   ierr = PetscMalloc3(ai[n]+1,PetscScalar,&b->a,ai[n]+1,PetscInt,&b->j,n+1,PetscInt,&b->i);CHKERRQ(ierr);
2022813a1f2bSShri Abhyankar   ierr = PetscLogObjectMemory(fact,ai[n]*(sizeof(PetscScalar)+sizeof(PetscInt))+(n+1)*sizeof(PetscInt));CHKERRQ(ierr);
2023813a1f2bSShri Abhyankar   b->singlemalloc = PETSC_TRUE;
2024813a1f2bSShri Abhyankar   if (!b->diag){
2025813a1f2bSShri Abhyankar     ierr = PetscMalloc((n+1)*sizeof(PetscInt),&b->diag);CHKERRQ(ierr);
2026813a1f2bSShri Abhyankar     ierr = PetscLogObjectMemory(fact,(n+1)*sizeof(PetscInt));CHKERRQ(ierr);
2027813a1f2bSShri Abhyankar   }
2028813a1f2bSShri Abhyankar   bdiag = b->diag;
2029813a1f2bSShri Abhyankar 
2030813a1f2bSShri Abhyankar   if (n > 0) {
2031813a1f2bSShri Abhyankar     ierr = PetscMemzero(b->a,(ai[n])*sizeof(MatScalar));CHKERRQ(ierr);
2032813a1f2bSShri Abhyankar   }
2033813a1f2bSShri Abhyankar 
2034813a1f2bSShri Abhyankar   /* set bi and bj with new data structure */
2035813a1f2bSShri Abhyankar   bi = b->i;
2036813a1f2bSShri Abhyankar   bj = b->j;
2037813a1f2bSShri Abhyankar 
2038813a1f2bSShri Abhyankar   /* L part */
2039813a1f2bSShri Abhyankar   bi[0] = 0;
2040813a1f2bSShri Abhyankar   for (i=0; i<n; i++){
2041813a1f2bSShri Abhyankar     nz = adiag[i] - ai[i];
2042813a1f2bSShri Abhyankar     bi[i+1] = bi[i] + nz;
2043813a1f2bSShri Abhyankar     aj = a->j + ai[i];
2044813a1f2bSShri Abhyankar     for (j=0; j<nz; j++){
2045813a1f2bSShri Abhyankar       *bj = aj[j]; bj++;
2046813a1f2bSShri Abhyankar     }
2047813a1f2bSShri Abhyankar   }
2048813a1f2bSShri Abhyankar 
2049813a1f2bSShri Abhyankar   /* U part */
2050813a1f2bSShri Abhyankar   /* bi[n+1] = bi[n]; */
2051813a1f2bSShri Abhyankar   bi_temp = bi[n];
205262fbd6afSShri Abhyankar   bdiag[n] = bi[n]-1;
2053813a1f2bSShri Abhyankar   for (i=n-1; i>=0; i--){
2054813a1f2bSShri Abhyankar     nz = ai[i+1] - adiag[i] - 1;
2055813a1f2bSShri Abhyankar     /*  bi[2*n-i+1] = bi[2*n-i] + nz + 1; */
2056813a1f2bSShri Abhyankar     bi_temp = bi_temp + nz + 1;
2057813a1f2bSShri Abhyankar     aj = a->j + adiag[i] + 1;
2058813a1f2bSShri Abhyankar     for (j=0; j<nz; j++){
2059813a1f2bSShri Abhyankar       *bj = aj[j]; bj++;
2060813a1f2bSShri Abhyankar     }
2061813a1f2bSShri Abhyankar     /* diag[i] */
2062813a1f2bSShri Abhyankar     *bj = i; bj++;
2063813a1f2bSShri Abhyankar     /*   bdiag[i] = bi[2*n-i+1]-1; */
2064813a1f2bSShri Abhyankar     bdiag[i] = bi_temp - 1;
2065813a1f2bSShri Abhyankar   }
2066813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
2067813a1f2bSShri Abhyankar }
2068813a1f2bSShri Abhyankar 
2069813a1f2bSShri Abhyankar #undef __FUNCT__
2070813a1f2bSShri Abhyankar #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_newdatastruct_v2"
2071813a1f2bSShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ_newdatastruct_v2(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
2072813a1f2bSShri Abhyankar {
2073813a1f2bSShri Abhyankar   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
2074813a1f2bSShri Abhyankar   IS                 isicol;
2075813a1f2bSShri Abhyankar   PetscErrorCode     ierr;
2076813a1f2bSShri Abhyankar   const PetscInt     *r,*ic;
2077813a1f2bSShri Abhyankar   PetscInt           n=A->rmap->n,*ai=a->i,*aj=a->j,d;
2078813a1f2bSShri Abhyankar   PetscInt           *bi,*cols,nnz,*cols_lvl;
2079813a1f2bSShri Abhyankar   PetscInt           *bdiag,prow,fm,nzbd,reallocs=0,dcount=0;
2080813a1f2bSShri Abhyankar   PetscInt           i,levels,diagonal_fill;
2081813a1f2bSShri Abhyankar   PetscTruth         col_identity,row_identity;
2082813a1f2bSShri Abhyankar   PetscReal          f;
2083813a1f2bSShri Abhyankar   PetscInt           nlnk,*lnk,*lnk_lvl=PETSC_NULL;
2084813a1f2bSShri Abhyankar   PetscBT            lnkbt;
2085813a1f2bSShri Abhyankar   PetscInt           nzi,*bj,**bj_ptr,**bjlvl_ptr;
2086813a1f2bSShri Abhyankar   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
2087813a1f2bSShri Abhyankar   PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL;
2088813a1f2bSShri Abhyankar   PetscTruth         missing;
2089813a1f2bSShri Abhyankar 
2090813a1f2bSShri Abhyankar   PetscFunctionBegin;
2091813a1f2bSShri Abhyankar   if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n);
2092813a1f2bSShri Abhyankar   f             = info->fill;
2093813a1f2bSShri Abhyankar   levels        = (PetscInt)info->levels;
2094813a1f2bSShri Abhyankar   diagonal_fill = (PetscInt)info->diagonal_fill;
2095813a1f2bSShri Abhyankar   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
2096813a1f2bSShri Abhyankar 
2097813a1f2bSShri Abhyankar   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
2098813a1f2bSShri Abhyankar   ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr);
2099813a1f2bSShri Abhyankar 
2100813a1f2bSShri Abhyankar   if (!levels && row_identity && col_identity) {
2101813a1f2bSShri Abhyankar     /* special case: ilu(0) with natural ordering */
2102813a1f2bSShri Abhyankar     ierr = MatILUFactorSymbolic_SeqAIJ_ilu0_newdatastruct_v2(fact,A,isrow,iscol,info);CHKERRQ(ierr);
2103813a1f2bSShri Abhyankar     (fact)->ops->lufactornumeric =  MatLUFactorNumeric_SeqAIJ_newdatastruct_v2;
2104813a1f2bSShri Abhyankar 
2105813a1f2bSShri Abhyankar     fact->factor = MAT_FACTOR_ILU;
2106813a1f2bSShri Abhyankar     (fact)->info.factor_mallocs    = 0;
2107813a1f2bSShri Abhyankar     (fact)->info.fill_ratio_given  = info->fill;
2108813a1f2bSShri Abhyankar     (fact)->info.fill_ratio_needed = 1.0;
2109813a1f2bSShri Abhyankar     b               = (Mat_SeqAIJ*)(fact)->data;
2110813a1f2bSShri Abhyankar     ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr);
2111813a1f2bSShri Abhyankar     if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d);
2112813a1f2bSShri Abhyankar     b->row              = isrow;
2113813a1f2bSShri Abhyankar     b->col              = iscol;
2114813a1f2bSShri Abhyankar     b->icol             = isicol;
2115813a1f2bSShri Abhyankar     ierr                = PetscMalloc(((fact)->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
2116813a1f2bSShri Abhyankar     ierr                = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
2117813a1f2bSShri Abhyankar     ierr                = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
21184108e4d5SBarry Smith     /* ierr = MatILUFactorSymbolic_SeqAIJ_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr); */
2119813a1f2bSShri Abhyankar     PetscFunctionReturn(0);
2120813a1f2bSShri Abhyankar   }
2121813a1f2bSShri Abhyankar 
2122813a1f2bSShri Abhyankar   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
2123813a1f2bSShri Abhyankar   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
2124813a1f2bSShri Abhyankar 
2125813a1f2bSShri Abhyankar   /* get new row pointers */
2126f268cba8SShri Abhyankar   /*  ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bi);CHKERRQ(ierr); */
2127f268cba8SShri Abhyankar   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
2128813a1f2bSShri Abhyankar   bi[0] = 0;
2129813a1f2bSShri Abhyankar   /* bdiag is location of diagonal in factor */
2130813a1f2bSShri Abhyankar   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
2131813a1f2bSShri Abhyankar   bdiag[0]  = 0;
2132813a1f2bSShri Abhyankar 
2133813a1f2bSShri Abhyankar   ierr = PetscMalloc((2*n+1)*sizeof(PetscInt**),&bj_ptr);CHKERRQ(ierr);
2134813a1f2bSShri Abhyankar   bjlvl_ptr = (PetscInt**)(bj_ptr + n);
2135813a1f2bSShri Abhyankar 
2136813a1f2bSShri Abhyankar   /* create a linked list for storing column indices of the active row */
2137813a1f2bSShri Abhyankar   nlnk = n + 1;
2138813a1f2bSShri Abhyankar   ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
2139813a1f2bSShri Abhyankar 
2140813a1f2bSShri Abhyankar   /* initial FreeSpace size is f*(ai[n]+1) */
2141813a1f2bSShri Abhyankar   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr);
2142813a1f2bSShri Abhyankar   current_space = free_space;
2143813a1f2bSShri Abhyankar   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr);
2144813a1f2bSShri Abhyankar   current_space_lvl = free_space_lvl;
2145813a1f2bSShri Abhyankar 
2146813a1f2bSShri Abhyankar   for (i=0; i<n; i++) {
2147813a1f2bSShri Abhyankar     nzi = 0;
2148813a1f2bSShri Abhyankar     /* copy current row into linked list */
2149813a1f2bSShri Abhyankar     nnz  = ai[r[i]+1] - ai[r[i]];
2150813a1f2bSShri Abhyankar     if (!nnz) SETERRQ2(PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i);
2151813a1f2bSShri Abhyankar     cols = aj + ai[r[i]];
2152813a1f2bSShri Abhyankar     lnk[i] = -1; /* marker to indicate if diagonal exists */
2153813a1f2bSShri Abhyankar     ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
2154813a1f2bSShri Abhyankar     nzi += nlnk;
2155813a1f2bSShri Abhyankar 
2156813a1f2bSShri Abhyankar     /* make sure diagonal entry is included */
2157813a1f2bSShri Abhyankar     if (diagonal_fill && lnk[i] == -1) {
2158813a1f2bSShri Abhyankar       fm = n;
2159813a1f2bSShri Abhyankar       while (lnk[fm] < i) fm = lnk[fm];
2160813a1f2bSShri Abhyankar       lnk[i]     = lnk[fm]; /* insert diagonal into linked list */
2161813a1f2bSShri Abhyankar       lnk[fm]    = i;
2162813a1f2bSShri Abhyankar       lnk_lvl[i] = 0;
2163813a1f2bSShri Abhyankar       nzi++; dcount++;
2164813a1f2bSShri Abhyankar     }
2165813a1f2bSShri Abhyankar 
2166813a1f2bSShri Abhyankar     /* add pivot rows into the active row */
2167813a1f2bSShri Abhyankar     nzbd = 0;
2168813a1f2bSShri Abhyankar     prow = lnk[n];
2169813a1f2bSShri Abhyankar     while (prow < i) {
2170813a1f2bSShri Abhyankar       nnz      = bdiag[prow];
2171813a1f2bSShri Abhyankar       cols     = bj_ptr[prow] + nnz + 1;
2172813a1f2bSShri Abhyankar       cols_lvl = bjlvl_ptr[prow] + nnz + 1;
2173813a1f2bSShri Abhyankar       nnz      = bi[prow+1] - bi[prow] - nnz - 1;
2174813a1f2bSShri Abhyankar       ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr);
2175813a1f2bSShri Abhyankar       nzi += nlnk;
2176813a1f2bSShri Abhyankar       prow = lnk[prow];
2177813a1f2bSShri Abhyankar       nzbd++;
2178813a1f2bSShri Abhyankar     }
2179813a1f2bSShri Abhyankar     bdiag[i] = nzbd;
2180813a1f2bSShri Abhyankar     bi[i+1]  = bi[i] + nzi;
2181813a1f2bSShri Abhyankar 
2182813a1f2bSShri Abhyankar     /* if free space is not available, make more free space */
2183813a1f2bSShri Abhyankar     if (current_space->local_remaining<nzi) {
2184813a1f2bSShri Abhyankar       nnz = 2*nzi*(n - i); /* estimated and max additional space needed */
2185813a1f2bSShri Abhyankar       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
2186813a1f2bSShri Abhyankar       ierr = PetscFreeSpaceGet(nnz,&current_space_lvl);CHKERRQ(ierr);
2187813a1f2bSShri Abhyankar       reallocs++;
2188813a1f2bSShri Abhyankar     }
2189813a1f2bSShri Abhyankar 
2190813a1f2bSShri Abhyankar     /* copy data into free_space and free_space_lvl, then initialize lnk */
2191813a1f2bSShri Abhyankar     ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
2192813a1f2bSShri Abhyankar     bj_ptr[i]    = current_space->array;
2193813a1f2bSShri Abhyankar     bjlvl_ptr[i] = current_space_lvl->array;
2194813a1f2bSShri Abhyankar 
2195813a1f2bSShri Abhyankar     /* make sure the active row i has diagonal entry */
2196813a1f2bSShri Abhyankar     if (*(bj_ptr[i]+bdiag[i]) != i) {
2197813a1f2bSShri Abhyankar       SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,"Row %D has missing diagonal in factored matrix\n\
2198813a1f2bSShri Abhyankar     try running with -pc_factor_nonzeros_along_diagonal or -pc_factor_diagonal_fill",i);
2199813a1f2bSShri Abhyankar     }
2200813a1f2bSShri Abhyankar 
2201813a1f2bSShri Abhyankar     current_space->array           += nzi;
2202813a1f2bSShri Abhyankar     current_space->local_used      += nzi;
2203813a1f2bSShri Abhyankar     current_space->local_remaining -= nzi;
2204813a1f2bSShri Abhyankar     current_space_lvl->array           += nzi;
2205813a1f2bSShri Abhyankar     current_space_lvl->local_used      += nzi;
2206813a1f2bSShri Abhyankar     current_space_lvl->local_remaining -= nzi;
2207813a1f2bSShri Abhyankar   }
2208813a1f2bSShri Abhyankar 
2209813a1f2bSShri Abhyankar   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
2210813a1f2bSShri Abhyankar   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
2211813a1f2bSShri Abhyankar 
2212813a1f2bSShri Abhyankar   /* destroy list of free space and other temporary arrays */
2213813a1f2bSShri Abhyankar   ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
2214813a1f2bSShri Abhyankar 
2215813a1f2bSShri Abhyankar   /* copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */
2216f268cba8SShri Abhyankar   /* ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr); */
2217f268cba8SShri Abhyankar   ierr = PetscFreeSpaceContiguous_LU_v2(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr);
2218813a1f2bSShri Abhyankar 
2219813a1f2bSShri Abhyankar   ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
2220813a1f2bSShri Abhyankar   ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
2221813a1f2bSShri Abhyankar   ierr = PetscFree(bj_ptr);CHKERRQ(ierr);
2222813a1f2bSShri Abhyankar 
2223813a1f2bSShri Abhyankar #if defined(PETSC_USE_INFO)
2224813a1f2bSShri Abhyankar   {
2225813a1f2bSShri Abhyankar     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
2226813a1f2bSShri Abhyankar     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr);
2227813a1f2bSShri Abhyankar     ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
2228813a1f2bSShri Abhyankar     ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr);
2229813a1f2bSShri Abhyankar     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
2230813a1f2bSShri Abhyankar     if (diagonal_fill) {
2231813a1f2bSShri Abhyankar       ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr);
2232813a1f2bSShri Abhyankar     }
2233813a1f2bSShri Abhyankar   }
2234813a1f2bSShri Abhyankar #endif
2235813a1f2bSShri Abhyankar 
2236813a1f2bSShri Abhyankar   /* put together the new matrix */
2237813a1f2bSShri Abhyankar   ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
2238813a1f2bSShri Abhyankar   ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr);
2239813a1f2bSShri Abhyankar   b = (Mat_SeqAIJ*)(fact)->data;
2240813a1f2bSShri Abhyankar   b->free_a       = PETSC_TRUE;
2241813a1f2bSShri Abhyankar   b->free_ij      = PETSC_TRUE;
2242813a1f2bSShri Abhyankar   b->singlemalloc = PETSC_FALSE;
2243f268cba8SShri Abhyankar   /*  ierr = PetscMalloc( (bi[2*n+1] )*sizeof(PetscScalar),&b->a);CHKERRQ(ierr); */
2244f268cba8SShri Abhyankar   ierr = PetscMalloc((bdiag[0]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);
2245813a1f2bSShri Abhyankar   b->j          = bj;
2246813a1f2bSShri Abhyankar   b->i          = bi;
2247813a1f2bSShri Abhyankar   b->diag       = bdiag;
2248813a1f2bSShri Abhyankar   b->ilen       = 0;
2249813a1f2bSShri Abhyankar   b->imax       = 0;
2250813a1f2bSShri Abhyankar   b->row        = isrow;
2251813a1f2bSShri Abhyankar   b->col        = iscol;
2252813a1f2bSShri Abhyankar   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
2253813a1f2bSShri Abhyankar   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
2254813a1f2bSShri Abhyankar   b->icol       = isicol;
2255813a1f2bSShri Abhyankar   ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
2256813a1f2bSShri Abhyankar   /* In b structure:  Free imax, ilen, old a, old j.
2257813a1f2bSShri Abhyankar      Allocate bdiag, solve_work, new a, new j */
2258f268cba8SShri Abhyankar 
2259f268cba8SShri Abhyankar   /*  ierr = PetscLogObjectMemory(fact,bi[2*n+1] * (sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
2260f268cba8SShri Abhyankar       b->maxnz = b->nz = bi[2*n+1] ; */
2261f268cba8SShri Abhyankar   ierr = PetscLogObjectMemory(fact,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
2262f268cba8SShri Abhyankar   b->maxnz = b->nz = bdiag[0]+1;
2263813a1f2bSShri Abhyankar   (fact)->info.factor_mallocs    = reallocs;
2264813a1f2bSShri Abhyankar   (fact)->info.fill_ratio_given  = f;
2265f268cba8SShri Abhyankar   /*  (fact)->info.fill_ratio_needed = ((PetscReal)bi[2*n+1])/((PetscReal)ai[n]); */
2266f268cba8SShri Abhyankar   (fact)->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]);
2267f268cba8SShri Abhyankar   /*  (fact)->ops->lufactornumeric   = MatLUFactorNumeric_SeqAIJ_newdatastruct; */
2268f268cba8SShri Abhyankar   (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_newdatastruct_v2;
22694108e4d5SBarry Smith   /* ierr = MatILUFactorSymbolic_SeqAIJ_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr); */
2270813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
2271813a1f2bSShri Abhyankar }
2272813a1f2bSShri Abhyankar 
22736516239fSHong Zhang #undef __FUNCT__
22746516239fSHong Zhang #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ"
22756516239fSHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
22766516239fSHong Zhang {
22776516239fSHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
22786516239fSHong Zhang   IS                 isicol;
22796516239fSHong Zhang   PetscErrorCode     ierr;
22806516239fSHong Zhang   const PetscInt     *r,*ic;
22816516239fSHong Zhang   PetscInt           n=A->rmap->n,*ai=a->i,*aj=a->j,d;
22826516239fSHong Zhang   PetscInt           *bi,*cols,nnz,*cols_lvl;
22836516239fSHong Zhang   PetscInt           *bdiag,prow,fm,nzbd,reallocs=0,dcount=0;
22846516239fSHong Zhang   PetscInt           i,levels,diagonal_fill;
22856516239fSHong Zhang   PetscTruth         col_identity,row_identity;
22866516239fSHong Zhang   PetscReal          f;
22876516239fSHong Zhang   PetscInt           nlnk,*lnk,*lnk_lvl=PETSC_NULL;
22886516239fSHong Zhang   PetscBT            lnkbt;
22896516239fSHong Zhang   PetscInt           nzi,*bj,**bj_ptr,**bjlvl_ptr;
22906516239fSHong Zhang   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
22916516239fSHong Zhang   PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL;
22926516239fSHong Zhang   PetscTruth         missing;
2293813a1f2bSShri Abhyankar   PetscTruth         newdatastruct=PETSC_FALSE,newdatastruct_v2=PETSC_FALSE;
22946516239fSHong Zhang 
22956516239fSHong Zhang   PetscFunctionBegin;
22966516239fSHong Zhang   ierr = PetscOptionsGetTruth(PETSC_NULL,"-ilu_new",&newdatastruct,PETSC_NULL);CHKERRQ(ierr);
22976516239fSHong Zhang   if (newdatastruct){
22986516239fSHong Zhang     ierr = MatILUFactorSymbolic_SeqAIJ_newdatastruct(fact,A,isrow,iscol,info);CHKERRQ(ierr);
22996516239fSHong Zhang     PetscFunctionReturn(0);
23006516239fSHong Zhang   }
2301813a1f2bSShri Abhyankar   ierr = PetscOptionsGetTruth(PETSC_NULL,"-ilu_new_v2",&newdatastruct_v2,PETSC_NULL);CHKERRQ(ierr);
2302813a1f2bSShri Abhyankar   if(newdatastruct_v2){
2303813a1f2bSShri Abhyankar     ierr = MatILUFactorSymbolic_SeqAIJ_newdatastruct_v2(fact,A,isrow,iscol,info);CHKERRQ(ierr);
2304813a1f2bSShri Abhyankar     PetscFunctionReturn(0);
2305813a1f2bSShri Abhyankar   }
23066516239fSHong Zhang 
23076516239fSHong Zhang   if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n);
23086516239fSHong Zhang   f             = info->fill;
23096516239fSHong Zhang   levels        = (PetscInt)info->levels;
23106516239fSHong Zhang   diagonal_fill = (PetscInt)info->diagonal_fill;
23116516239fSHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
23126516239fSHong Zhang 
23136516239fSHong Zhang   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
23146516239fSHong Zhang   ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr);
23156516239fSHong Zhang   if (!levels && row_identity && col_identity) { /* special case: ilu(0) with natural ordering */
2316f77e22a1SHong Zhang     ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_TRUE);CHKERRQ(ierr);
2317c9c72f8fSHong Zhang     (fact)->ops->lufactornumeric =  MatLUFactorNumeric_SeqAIJ;
2318c9c72f8fSHong Zhang 
2319719d5645SBarry Smith     fact->factor = MAT_FACTOR_ILU;
2320719d5645SBarry Smith     (fact)->info.factor_mallocs    = 0;
2321719d5645SBarry Smith     (fact)->info.fill_ratio_given  = info->fill;
2322719d5645SBarry Smith     (fact)->info.fill_ratio_needed = 1.0;
2323719d5645SBarry Smith     b               = (Mat_SeqAIJ*)(fact)->data;
23248ef3462fSBarry Smith     ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr);
232509f38230SBarry Smith     if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d);
232608480c60SBarry Smith     b->row              = isrow;
232708480c60SBarry Smith     b->col              = iscol;
232882bf6240SBarry Smith     b->icol             = isicol;
2329719d5645SBarry Smith     ierr                = PetscMalloc(((fact)->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
2330c38d4ed2SBarry Smith     ierr                = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
2331c38d4ed2SBarry Smith     ierr                = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
23324108e4d5SBarry Smith     ierr = MatILUFactorSymbolic_SeqAIJ_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr);
23333a40ed3dSBarry Smith     PetscFunctionReturn(0);
233408480c60SBarry Smith   }
233508480c60SBarry Smith 
2336898183ecSLois Curfman McInnes   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
2337898183ecSLois Curfman McInnes   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
23389e25ed09SBarry Smith 
23399e25ed09SBarry Smith   /* get new row pointers */
23405a8e39fbSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
23415a8e39fbSHong Zhang   bi[0] = 0;
23425a8e39fbSHong Zhang   /* bdiag is location of diagonal in factor */
23435a8e39fbSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
23445a8e39fbSHong Zhang   bdiag[0]  = 0;
23456cf997b0SBarry Smith 
23465a8e39fbSHong Zhang   ierr = PetscMalloc((2*n+1)*sizeof(PetscInt**),&bj_ptr);CHKERRQ(ierr);
23475a8e39fbSHong Zhang   bjlvl_ptr = (PetscInt**)(bj_ptr + n);
23485a8e39fbSHong Zhang 
23495a8e39fbSHong Zhang   /* create a linked list for storing column indices of the active row */
23505a8e39fbSHong Zhang   nlnk = n + 1;
23515a8e39fbSHong Zhang   ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
23525a8e39fbSHong Zhang 
23535a8e39fbSHong Zhang   /* initial FreeSpace size is f*(ai[n]+1) */
2354a1a86e44SBarry Smith   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr);
23555a8e39fbSHong Zhang   current_space = free_space;
2356a1a86e44SBarry Smith   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr);
23575a8e39fbSHong Zhang   current_space_lvl = free_space_lvl;
23585a8e39fbSHong Zhang 
23595a8e39fbSHong Zhang   for (i=0; i<n; i++) {
23605a8e39fbSHong Zhang     nzi = 0;
23616cf997b0SBarry Smith     /* copy current row into linked list */
23625a8e39fbSHong Zhang     nnz  = ai[r[i]+1] - ai[r[i]];
23633b4a8b6dSBarry Smith     if (!nnz) SETERRQ2(PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i);
23645a8e39fbSHong Zhang     cols = aj + ai[r[i]];
23655a8e39fbSHong Zhang     lnk[i] = -1; /* marker to indicate if diagonal exists */
23665a8e39fbSHong Zhang     ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
23675a8e39fbSHong Zhang     nzi += nlnk;
23686cf997b0SBarry Smith 
23696cf997b0SBarry Smith     /* make sure diagonal entry is included */
23705a8e39fbSHong Zhang     if (diagonal_fill && lnk[i] == -1) {
23716cf997b0SBarry Smith       fm = n;
23725a8e39fbSHong Zhang       while (lnk[fm] < i) fm = lnk[fm];
23735a8e39fbSHong Zhang       lnk[i]     = lnk[fm]; /* insert diagonal into linked list */
23745a8e39fbSHong Zhang       lnk[fm]    = i;
23755a8e39fbSHong Zhang       lnk_lvl[i] = 0;
23765a8e39fbSHong Zhang       nzi++; dcount++;
23776cf997b0SBarry Smith     }
23786cf997b0SBarry Smith 
23795a8e39fbSHong Zhang     /* add pivot rows into the active row */
23805a8e39fbSHong Zhang     nzbd = 0;
23815a8e39fbSHong Zhang     prow = lnk[n];
23825a8e39fbSHong Zhang     while (prow < i) {
23835a8e39fbSHong Zhang       nnz      = bdiag[prow];
23845a8e39fbSHong Zhang       cols     = bj_ptr[prow] + nnz + 1;
23855a8e39fbSHong Zhang       cols_lvl = bjlvl_ptr[prow] + nnz + 1;
23865a8e39fbSHong Zhang       nnz      = bi[prow+1] - bi[prow] - nnz - 1;
23875a8e39fbSHong Zhang       ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr);
23885a8e39fbSHong Zhang       nzi += nlnk;
23895a8e39fbSHong Zhang       prow = lnk[prow];
23905a8e39fbSHong Zhang       nzbd++;
239156beaf04SBarry Smith     }
23925a8e39fbSHong Zhang     bdiag[i] = nzbd;
23935a8e39fbSHong Zhang     bi[i+1]  = bi[i] + nzi;
2394ecf371e4SBarry Smith 
23955a8e39fbSHong Zhang     /* if free space is not available, make more free space */
23965a8e39fbSHong Zhang     if (current_space->local_remaining<nzi) {
23975a8e39fbSHong Zhang       nnz = nzi*(n - i); /* estimated and max additional space needed */
2398a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
2399a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(nnz,&current_space_lvl);CHKERRQ(ierr);
24005a8e39fbSHong Zhang       reallocs++;
24015a8e39fbSHong Zhang     }
2402ecf371e4SBarry Smith 
24035a8e39fbSHong Zhang     /* copy data into free_space and free_space_lvl, then initialize lnk */
24045a8e39fbSHong Zhang     ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
24055a8e39fbSHong Zhang     bj_ptr[i]    = current_space->array;
24065a8e39fbSHong Zhang     bjlvl_ptr[i] = current_space_lvl->array;
24075a8e39fbSHong Zhang 
24085a8e39fbSHong Zhang     /* make sure the active row i has diagonal entry */
24095a8e39fbSHong Zhang     if (*(bj_ptr[i]+bdiag[i]) != i) {
241077431f27SBarry Smith       SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,"Row %D has missing diagonal in factored matrix\n\
2411d7ee0231SBarry Smith     try running with -pc_factor_nonzeros_along_diagonal or -pc_factor_diagonal_fill",i);
24126cf997b0SBarry Smith     }
24135a8e39fbSHong Zhang 
24145a8e39fbSHong Zhang     current_space->array           += nzi;
24155a8e39fbSHong Zhang     current_space->local_used      += nzi;
24165a8e39fbSHong Zhang     current_space->local_remaining -= nzi;
24175a8e39fbSHong Zhang     current_space_lvl->array           += nzi;
24185a8e39fbSHong Zhang     current_space_lvl->local_used      += nzi;
24195a8e39fbSHong Zhang     current_space_lvl->local_remaining -= nzi;
24209e25ed09SBarry Smith   }
24215a8e39fbSHong Zhang 
2422898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
2423898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
24245a8e39fbSHong Zhang 
24255a8e39fbSHong Zhang   /* destroy list of free space and other temporary arrays */
24265a8e39fbSHong Zhang   ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
24276516239fSHong Zhang   ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); /* copy free_space -> bj */
24285a8e39fbSHong Zhang   ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
2429a1a86e44SBarry Smith   ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
24305a8e39fbSHong Zhang   ierr = PetscFree(bj_ptr);CHKERRQ(ierr);
24319e25ed09SBarry Smith 
24326cf91177SBarry Smith #if defined(PETSC_USE_INFO)
2433f64065bbSBarry Smith   {
24345a8e39fbSHong Zhang     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
2435ae15b995SBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr);
2436ae15b995SBarry Smith     ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
2437ae15b995SBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr);
2438ae15b995SBarry Smith     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
2439335d9088SBarry Smith     if (diagonal_fill) {
2440ae15b995SBarry Smith       ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr);
2441335d9088SBarry Smith     }
2442f64065bbSBarry Smith   }
244363ba0a88SBarry Smith #endif
2444bbb6d6a8SBarry Smith 
24459e25ed09SBarry Smith   /* put together the new matrix */
244607dbf95fSBarry Smith   ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
2447719d5645SBarry Smith   ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr);
2448719d5645SBarry Smith   b = (Mat_SeqAIJ*)(fact)->data;
2449e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
2450e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
24517c922b88SBarry Smith   b->singlemalloc = PETSC_FALSE;
2452052f0c41SBarry Smith   ierr = PetscMalloc( (bi[n] )*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);
24535a8e39fbSHong Zhang   b->j          = bj;
24545a8e39fbSHong Zhang   b->i          = bi;
24555a8e39fbSHong Zhang   for (i=0; i<n; i++) bdiag[i] += bi[i];
24565a8e39fbSHong Zhang   b->diag       = bdiag;
2457416022c9SBarry Smith   b->ilen       = 0;
2458416022c9SBarry Smith   b->imax       = 0;
2459416022c9SBarry Smith   b->row        = isrow;
2460416022c9SBarry Smith   b->col        = iscol;
2461c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
2462c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
246382bf6240SBarry Smith   b->icol       = isicol;
246487828ca2SBarry Smith   ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
2465416022c9SBarry Smith   /* In b structure:  Free imax, ilen, old a, old j.
24665a8e39fbSHong Zhang      Allocate bdiag, solve_work, new a, new j */
2467719d5645SBarry Smith   ierr = PetscLogObjectMemory(fact,(bi[n]-n) * (sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
24685a8e39fbSHong Zhang   b->maxnz             = b->nz = bi[n] ;
2469719d5645SBarry Smith   (fact)->info.factor_mallocs    = reallocs;
2470719d5645SBarry Smith   (fact)->info.fill_ratio_given  = f;
2471719d5645SBarry Smith   (fact)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]);
2472719d5645SBarry Smith   (fact)->ops->lufactornumeric =  MatLUFactorNumeric_SeqAIJ;
24734108e4d5SBarry Smith   ierr = MatILUFactorSymbolic_SeqAIJ_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr);
24743a40ed3dSBarry Smith   PetscFunctionReturn(0);
24759e25ed09SBarry Smith }
2476d5d45c9bSBarry Smith 
2477a6175056SHong Zhang #undef __FUNCT__
24785f44c854SHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ_newdatastruct"
24795f44c854SHong Zhang PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ_newdatastruct(Mat B,Mat A,const MatFactorInfo *info)
24805f44c854SHong Zhang {
24815f44c854SHong Zhang   Mat            C = B;
24825f44c854SHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
24835f44c854SHong Zhang   Mat_SeqSBAIJ   *b=(Mat_SeqSBAIJ*)C->data;
24845f44c854SHong Zhang   IS             ip=b->row,iip = b->icol;
24855f44c854SHong Zhang   PetscErrorCode ierr;
24865f44c854SHong Zhang   const PetscInt *rip,*riip;
24875f44c854SHong Zhang   PetscInt       i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bdiag=b->diag,*bjtmp;
24885f44c854SHong Zhang   PetscInt       *ai=a->i,*aj=a->j;
24895f44c854SHong Zhang   PetscInt       k,jmin,jmax,*c2r,*il,col,nexti,ili,nz;
24905f44c854SHong Zhang   MatScalar      *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi;
24915f44c854SHong Zhang   PetscReal      zeropivot,shiftnz;
24925f44c854SHong Zhang   PetscReal      shiftpd;
24935f44c854SHong Zhang   PetscTruth     perm_identity;
24945f44c854SHong Zhang 
24955f44c854SHong Zhang   PetscFunctionBegin;
24965f44c854SHong Zhang 
24975f44c854SHong Zhang   shiftnz   = info->shiftnz;
24985f44c854SHong Zhang   shiftpd   = info->shiftpd;
24995f44c854SHong Zhang   zeropivot = info->zeropivot;
25005f44c854SHong Zhang 
25015f44c854SHong Zhang   ierr  = ISGetIndices(ip,&rip);CHKERRQ(ierr);
25025f44c854SHong Zhang   ierr  = ISGetIndices(iip,&riip);CHKERRQ(ierr);
25035f44c854SHong Zhang 
25045f44c854SHong Zhang   /* allocate working arrays
25055f44c854SHong Zhang      c2r: linked list, keep track of pivot rows for a given column. c2r[col]: head of the list for a given col
25065f44c854SHong 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
25075f44c854SHong Zhang   */
25085f44c854SHong Zhang   nz   = (2*mbs+1)*sizeof(PetscInt)+mbs*sizeof(MatScalar);
25095f44c854SHong Zhang   ierr = PetscMalloc(nz,&il);CHKERRQ(ierr);
25105f44c854SHong Zhang   c2r  = il + mbs;
25115f44c854SHong Zhang   rtmp = (MatScalar*)(c2r + mbs);
25125f44c854SHong Zhang 
25135f44c854SHong Zhang   for (i=0; i<mbs; i++) {
25145f44c854SHong Zhang     c2r[i] = mbs;
25155f44c854SHong Zhang   }
25165f44c854SHong Zhang   il[0] = 0;
25175f44c854SHong Zhang 
25185f44c854SHong Zhang   for (k = 0; k<mbs; k++){
25195f44c854SHong Zhang     /* zero rtmp */
25205f44c854SHong Zhang     nz = bi[k+1] - bi[k];
25215f44c854SHong Zhang     bjtmp = bj + bi[k];
25225f44c854SHong Zhang     for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
25235f44c854SHong Zhang 
25245f44c854SHong Zhang     /* load in initial unfactored row */
25255f44c854SHong Zhang     bval = ba + bi[k];
25265f44c854SHong Zhang     jmin = ai[rip[k]]; jmax = ai[rip[k]+1];
25275f44c854SHong Zhang     for (j = jmin; j < jmax; j++){
25285f44c854SHong Zhang       col = riip[aj[j]];
25295f44c854SHong Zhang       if (col >= k){ /* only take upper triangular entry */
25305f44c854SHong Zhang         rtmp[col] = aa[j];
25315f44c854SHong Zhang         *bval++   = 0.0; /* for in-place factorization */
25325f44c854SHong Zhang       }
25335f44c854SHong Zhang     }
25345f44c854SHong Zhang     /* shift the diagonal of the matrix */
25355f44c854SHong Zhang 
25365f44c854SHong Zhang     /* modify k-th row by adding in those rows i with U(i,k)!=0 */
25375f44c854SHong Zhang     dk = rtmp[k];
25385f44c854SHong Zhang     i  = c2r[k]; /* first row to be added to k_th row  */
25395f44c854SHong Zhang 
25405f44c854SHong Zhang     while (i < k){
25415f44c854SHong Zhang       nexti = c2r[i]; /* next row to be added to k_th row */
25425f44c854SHong Zhang 
25435f44c854SHong Zhang       /* compute multiplier, update diag(k) and U(i,k) */
25445f44c854SHong Zhang       ili   = il[i];  /* index of first nonzero element in U(i,k:bms-1) */
25455f44c854SHong Zhang       uikdi = - ba[ili]*ba[bdiag[i]];  /* diagonal(k) */
25465f44c854SHong Zhang       dk   += uikdi*ba[ili]; /* update diag[k] */
25475f44c854SHong Zhang       ba[ili] = uikdi; /* -U(i,k) */
25485f44c854SHong Zhang 
25495f44c854SHong Zhang       /* add multiple of row i to k-th row */
25505f44c854SHong Zhang       jmin = ili + 1; jmax = bi[i+1];
25515f44c854SHong Zhang       if (jmin < jmax){
25525f44c854SHong Zhang         for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j];
25535f44c854SHong Zhang         /* update il and c2r for row i */
25545f44c854SHong Zhang         il[i] = jmin;
25555f44c854SHong Zhang         j = bj[jmin]; c2r[i] = c2r[j]; c2r[j] = i;
25565f44c854SHong Zhang       }
25575f44c854SHong Zhang       i = nexti;
25585f44c854SHong Zhang     }
25595f44c854SHong Zhang 
25605f44c854SHong Zhang     /* copy data into U(k,:) */
25615f44c854SHong Zhang     ba[bdiag[k]] = 1.0/dk; /* U(k,k) */
25625f44c854SHong Zhang     jmin = bi[k]; jmax = bi[k+1]-1;
25635f44c854SHong Zhang     if (jmin < jmax) {
25645f44c854SHong Zhang       for (j=jmin; j<jmax; j++){
25655f44c854SHong Zhang         col = bj[j]; ba[j] = rtmp[col];
25665f44c854SHong Zhang       }
25675f44c854SHong Zhang       /* add the k-th row into il and c2r */
25685f44c854SHong Zhang       il[k] = jmin;
25695f44c854SHong Zhang       i = bj[jmin]; c2r[k] = c2r[i]; c2r[i] = k;
25705f44c854SHong Zhang     }
25715f44c854SHong Zhang   }
25725f44c854SHong Zhang 
25735f44c854SHong Zhang   ierr = PetscFree(il);CHKERRQ(ierr);
25745f44c854SHong Zhang   ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr);
25755f44c854SHong Zhang   ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr);
25765f44c854SHong Zhang 
25775f44c854SHong Zhang   ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr);
25785f44c854SHong Zhang   if (perm_identity){
25795f44c854SHong Zhang     (B)->ops->solve           = MatSolve_SeqSBAIJ_1_NaturalOrdering_newdatastruct;
25805f44c854SHong Zhang     (B)->ops->solvetranspose  = 0;
25815f44c854SHong Zhang     (B)->ops->forwardsolve    = 0;
25825f44c854SHong Zhang     (B)->ops->backwardsolve   = 0;
25835f44c854SHong Zhang   } else {
2584783ef271SHong Zhang     (B)->ops->solve           = MatSolve_SeqSBAIJ_1_newdatastruct;
25855f44c854SHong Zhang     (B)->ops->solvetranspose  = 0;
25865f44c854SHong Zhang     (B)->ops->forwardsolve    = 0;
25875f44c854SHong Zhang     (B)->ops->backwardsolve   = 0;
25885f44c854SHong Zhang   }
25895f44c854SHong Zhang 
25905f44c854SHong Zhang   C->assembled    = PETSC_TRUE;
25915f44c854SHong Zhang   C->preallocated = PETSC_TRUE;
25925f44c854SHong Zhang   ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr);
25935f44c854SHong Zhang   PetscFunctionReturn(0);
25945f44c854SHong Zhang }
25955f44c854SHong Zhang 
25965f44c854SHong Zhang #undef __FUNCT__
2597a6175056SHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ"
25980481f469SBarry Smith PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info)
2599a6175056SHong Zhang {
2600719d5645SBarry Smith   Mat            C = B;
26010968510aSHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
26022ed133dbSHong Zhang   Mat_SeqSBAIJ   *b=(Mat_SeqSBAIJ*)C->data;
26039bfd6278SHong Zhang   IS             ip=b->row,iip = b->icol;
26042ed133dbSHong Zhang   PetscErrorCode ierr;
26055d0c19d7SBarry Smith   const PetscInt *rip,*riip;
26065d0c19d7SBarry Smith   PetscInt       i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bcol;
26072ed133dbSHong Zhang   PetscInt       *ai=a->i,*aj=a->j;
26082ed133dbSHong Zhang   PetscInt       k,jmin,jmax,*jl,*il,col,nexti,ili,nz;
2609622e2028SHong Zhang   MatScalar      *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi;
2610540703acSHong Zhang   PetscReal      zeropivot,rs,shiftnz;
2611fbf22428SSatish Balay   PetscReal      shiftpd;
2612540703acSHong Zhang   ChShift_Ctx    sctx;
2613540703acSHong Zhang   PetscInt       newshift;
2614db4efbfdSBarry Smith   PetscTruth     perm_identity;
2615d5d45c9bSBarry Smith 
2616a6175056SHong Zhang   PetscFunctionBegin;
2617540703acSHong Zhang   shiftnz   = info->shiftnz;
2618540703acSHong Zhang   shiftpd   = info->shiftpd;
2619ee45ca4aSHong Zhang   zeropivot = info->zeropivot;
2620ee45ca4aSHong Zhang 
26212ed133dbSHong Zhang   ierr  = ISGetIndices(ip,&rip);CHKERRQ(ierr);
26229bfd6278SHong Zhang   ierr  = ISGetIndices(iip,&riip);CHKERRQ(ierr);
26232ed133dbSHong Zhang 
26242ed133dbSHong Zhang   /* initialization */
26252ed133dbSHong Zhang   nz   = (2*mbs+1)*sizeof(PetscInt)+mbs*sizeof(MatScalar);
26262ed133dbSHong Zhang   ierr = PetscMalloc(nz,&il);CHKERRQ(ierr);
26272ed133dbSHong Zhang   jl   = il + mbs;
26282ed133dbSHong Zhang   rtmp = (MatScalar*)(jl + mbs);
26292ed133dbSHong Zhang 
2630540703acSHong Zhang   sctx.shift_amount = 0;
2631540703acSHong Zhang   sctx.nshift       = 0;
26322ed133dbSHong Zhang   do {
2633540703acSHong Zhang     sctx.chshift = PETSC_FALSE;
26342ed133dbSHong Zhang     for (i=0; i<mbs; i++) {
26352ed133dbSHong Zhang       rtmp[i] = 0.0; jl[i] = mbs; il[0] = 0;
2636a6175056SHong Zhang     }
26372ed133dbSHong Zhang 
26382ed133dbSHong Zhang     for (k = 0; k<mbs; k++){
2639c05c3958SHong Zhang       bval = ba + bi[k];
26402ed133dbSHong Zhang       /* initialize k-th row by the perm[k]-th row of A */
26412ed133dbSHong Zhang       jmin = ai[rip[k]]; jmax = ai[rip[k]+1];
26422ed133dbSHong Zhang       for (j = jmin; j < jmax; j++){
26439bfd6278SHong Zhang         col = riip[aj[j]];
26442ed133dbSHong Zhang         if (col >= k){ /* only take upper triangular entry */
26452ed133dbSHong Zhang           rtmp[col] = aa[j];
2646c05c3958SHong Zhang           *bval++  = 0.0; /* for in-place factorization */
26472ed133dbSHong Zhang         }
26482ed133dbSHong Zhang       }
264939e3d630SHong Zhang       /* shift the diagonal of the matrix */
2650540703acSHong Zhang       if (sctx.nshift) rtmp[k] += sctx.shift_amount;
26512ed133dbSHong Zhang 
26522ed133dbSHong Zhang       /* modify k-th row by adding in those rows i with U(i,k)!=0 */
26532ed133dbSHong Zhang       dk = rtmp[k];
26542ed133dbSHong Zhang       i = jl[k]; /* first row to be added to k_th row  */
26552ed133dbSHong Zhang 
26562ed133dbSHong Zhang       while (i < k){
26572ed133dbSHong Zhang         nexti = jl[i]; /* next row to be added to k_th row */
26582ed133dbSHong Zhang 
26592ed133dbSHong Zhang         /* compute multiplier, update diag(k) and U(i,k) */
26602ed133dbSHong Zhang         ili = il[i];  /* index of first nonzero element in U(i,k:bms-1) */
26612ed133dbSHong Zhang         uikdi = - ba[ili]*ba[bi[i]];  /* diagonal(k) */
26622ed133dbSHong Zhang         dk += uikdi*ba[ili];
26632ed133dbSHong Zhang         ba[ili] = uikdi; /* -U(i,k) */
26642ed133dbSHong Zhang 
26652ed133dbSHong Zhang         /* add multiple of row i to k-th row */
26662ed133dbSHong Zhang         jmin = ili + 1; jmax = bi[i+1];
26672ed133dbSHong Zhang         if (jmin < jmax){
26682ed133dbSHong Zhang           for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j];
26692ed133dbSHong Zhang           /* update il and jl for row i */
26702ed133dbSHong Zhang           il[i] = jmin;
26712ed133dbSHong Zhang           j = bj[jmin]; jl[i] = jl[j]; jl[j] = i;
26722ed133dbSHong Zhang         }
26732ed133dbSHong Zhang         i = nexti;
26742ed133dbSHong Zhang       }
26752ed133dbSHong Zhang 
26760697b6caSHong Zhang       /* shift the diagonals when zero pivot is detected */
26770697b6caSHong Zhang       /* compute rs=sum of abs(off-diagonal) */
26780697b6caSHong Zhang       rs   = 0.0;
26793c9e8598SHong Zhang       jmin = bi[k]+1;
26803c9e8598SHong Zhang       nz   = bi[k+1] - jmin;
26813c9e8598SHong Zhang       bcol = bj + jmin;
268204fbf559SBarry Smith       for (j=0; j<nz; j++) {
268304fbf559SBarry Smith         rs += PetscAbsScalar(rtmp[bcol[j]]);
26843c9e8598SHong Zhang       }
2685540703acSHong Zhang 
2686540703acSHong Zhang       sctx.rs = rs;
2687540703acSHong Zhang       sctx.pv = dk;
26885b5bc046SBarry Smith       ierr = MatCholeskyCheckShift_inline(info,sctx,k,newshift);CHKERRQ(ierr);
2689117f1891Stmunson 
2690117f1891Stmunson       if (newshift == 1) {
2691117f1891Stmunson         if (!sctx.shift_amount) {
2692117f1891Stmunson           sctx.shift_amount = 1e-5;
2693117f1891Stmunson         }
2694117f1891Stmunson         break;
2695117f1891Stmunson       }
26963c9e8598SHong Zhang 
26973c9e8598SHong Zhang       /* copy data into U(k,:) */
269839e3d630SHong Zhang       ba[bi[k]] = 1.0/dk; /* U(k,k) */
269939e3d630SHong Zhang       jmin = bi[k]+1; jmax = bi[k+1];
270039e3d630SHong Zhang       if (jmin < jmax) {
270139e3d630SHong Zhang         for (j=jmin; j<jmax; j++){
270239e3d630SHong Zhang           col = bj[j]; ba[j] = rtmp[col]; rtmp[col] = 0.0;
27033c9e8598SHong Zhang         }
270439e3d630SHong Zhang         /* add the k-th row into il and jl */
27053c9e8598SHong Zhang         il[k] = jmin;
27063c9e8598SHong Zhang         i = bj[jmin]; jl[k] = jl[i]; jl[i] = k;
27073c9e8598SHong Zhang       }
27083c9e8598SHong Zhang     }
2709540703acSHong Zhang   } while (sctx.chshift);
27103c9e8598SHong Zhang   ierr = PetscFree(il);CHKERRQ(ierr);
27113c9e8598SHong Zhang 
271239e3d630SHong Zhang   ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr);
27139bfd6278SHong Zhang   ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr);
2714db4efbfdSBarry Smith 
2715db4efbfdSBarry Smith   ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr);
2716db4efbfdSBarry Smith   if (perm_identity){
2717719d5645SBarry Smith     (B)->ops->solve           = MatSolve_SeqSBAIJ_1_NaturalOrdering;
2718719d5645SBarry Smith     (B)->ops->solvetranspose  = MatSolve_SeqSBAIJ_1_NaturalOrdering;
2719719d5645SBarry Smith     (B)->ops->forwardsolve    = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering;
2720719d5645SBarry Smith     (B)->ops->backwardsolve   = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering;
2721db4efbfdSBarry Smith   } else {
2722719d5645SBarry Smith     (B)->ops->solve           = MatSolve_SeqSBAIJ_1;
2723719d5645SBarry Smith     (B)->ops->solvetranspose  = MatSolve_SeqSBAIJ_1;
2724719d5645SBarry Smith     (B)->ops->forwardsolve    = MatForwardSolve_SeqSBAIJ_1;
2725719d5645SBarry Smith     (B)->ops->backwardsolve   = MatBackwardSolve_SeqSBAIJ_1;
2726db4efbfdSBarry Smith   }
2727db4efbfdSBarry Smith 
27283c9e8598SHong Zhang   C->assembled    = PETSC_TRUE;
27293c9e8598SHong Zhang   C->preallocated = PETSC_TRUE;
2730d0f46423SBarry Smith   ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr);
2731540703acSHong Zhang   if (sctx.nshift){
2732540703acSHong Zhang     if (shiftnz) {
27331e2582c4SBarry Smith       ierr = PetscInfo2(A,"number of shiftnz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
2734540703acSHong Zhang     } else if (shiftpd) {
27351e2582c4SBarry Smith       ierr = PetscInfo2(A,"number of shiftpd tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
27360697b6caSHong Zhang     }
27373c9e8598SHong Zhang   }
27383c9e8598SHong Zhang   PetscFunctionReturn(0);
27393c9e8598SHong Zhang }
2740a6175056SHong Zhang 
2741*8a76ed4fSHong Zhang /*
2742*8a76ed4fSHong Zhang    icc() under revised new data structure.
2743*8a76ed4fSHong Zhang    Factored arrays bj and ba are stored as
2744*8a76ed4fSHong Zhang      U(0,:),...,U(i,:),U(n-1,:)
2745*8a76ed4fSHong Zhang 
2746*8a76ed4fSHong Zhang    ui=fact->i is an array of size n+1, in which
2747*8a76ed4fSHong Zhang    ui+
2748*8a76ed4fSHong Zhang      ui[i]:  points to 1st entry of U(i,:),i=0,...,n-1
2749*8a76ed4fSHong Zhang      ui[n]:  points to U(n-1,n-1)+1
2750*8a76ed4fSHong Zhang 
2751*8a76ed4fSHong Zhang   udiag=fact->diag is an array of size n,in which
2752*8a76ed4fSHong Zhang      udiag[i]: points to diagonal of U(i,:), i=0,...,n-1
2753*8a76ed4fSHong Zhang 
2754*8a76ed4fSHong Zhang    U(i,:) contains udiag[i] as its last entry, i.e.,
2755*8a76ed4fSHong Zhang     U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i])
2756*8a76ed4fSHong Zhang */
2757*8a76ed4fSHong Zhang 
2758a6175056SHong Zhang #undef __FUNCT__
27595f44c854SHong Zhang #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ_newdatastruct"
27605f44c854SHong Zhang PetscErrorCode MatICCFactorSymbolic_SeqAIJ_newdatastruct(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
2761a6175056SHong Zhang {
27620968510aSHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
2763ed59401aSHong Zhang   Mat_SeqSBAIJ       *b;
2764dfbe8321SBarry Smith   PetscErrorCode     ierr;
276558ebbce7SBarry Smith   PetscTruth         perm_identity,missing;
27665f44c854SHong Zhang   PetscInt           reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag;
27675d0c19d7SBarry Smith   const PetscInt     *rip,*riip;
2768ed59401aSHong Zhang   PetscInt           jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow;
276958ebbce7SBarry Smith   PetscInt           nlnk,*lnk,*lnk_lvl=PETSC_NULL,d;
27705a8e39fbSHong Zhang   PetscInt           ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr;
2771ed59401aSHong Zhang   PetscReal          fill=info->fill,levels=info->levels;
2772a1a86e44SBarry Smith   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
2773a1a86e44SBarry Smith   PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL;
27747a48dd6fSHong Zhang   PetscBT            lnkbt;
2775b635d36bSHong Zhang   IS                 iperm;
2776a6175056SHong Zhang 
2777a6175056SHong Zhang   PetscFunctionBegin;
2778d0f46423SBarry Smith   if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n);
277958ebbce7SBarry Smith   ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr);
278058ebbce7SBarry Smith   if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d);
2781653a6975SHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
2782b635d36bSHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
27837a48dd6fSHong Zhang 
278439e3d630SHong Zhang   ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr);
27855f44c854SHong Zhang   ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr);
278639e3d630SHong Zhang   ui[0] = 0;
278739e3d630SHong Zhang 
2788*8a76ed4fSHong Zhang   /* ICC(0) without matrix ordering: simply rearrange column indices */
2789622e2028SHong Zhang   if (!levels && perm_identity) {
279058ebbce7SBarry Smith 
2791ed59401aSHong Zhang     for (i=0; i<am; i++) {
279239e3d630SHong Zhang       ui[i+1]  = ui[i] + ai[i+1] - a->diag[i];
27935f44c854SHong Zhang       udiag[i] = ui[i+1] - 1; /* points to the last entry of U(i,:) */
27945f44c854SHong Zhang     }
27955f44c854SHong Zhang     ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
27965f44c854SHong Zhang     cols = uj;
27975f44c854SHong Zhang     for (i=0; i<am; i++) {
27985f44c854SHong Zhang       aj    = a->j + a->diag[i] + 1; /* 1st entry of U(i,:) without diagonal */
27995f44c854SHong Zhang       ncols = ui[i+1] - ui[i] - 1;
28005f44c854SHong Zhang       for (j=0; j<ncols; j++) *cols++ = aj[j];
28015f44c854SHong Zhang       *cols++ = i; /* diagoanl is located as the last entry of U(i,:) */
28025f44c854SHong Zhang     }
28035f44c854SHong Zhang   } else { /* case: levels>0 || (levels=0 && !perm_identity) */
2804*8a76ed4fSHong Zhang     if (levels) SETERRQ(PETSC_ERR_ARG_WRONG,"not done yet");
28055f44c854SHong Zhang     ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
28065f44c854SHong Zhang     ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
28075f44c854SHong Zhang 
28085f44c854SHong Zhang     /* initialization */
28095f44c854SHong Zhang     ierr  = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr);
28105f44c854SHong Zhang 
28115f44c854SHong Zhang     /* jl: linked list for storing indices of the pivot rows
28125f44c854SHong Zhang        il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
28135f44c854SHong Zhang     ierr = PetscMalloc((2*am+1)*sizeof(PetscInt)+2*am*sizeof(PetscInt**),&jl);CHKERRQ(ierr);
28145f44c854SHong Zhang     il         = jl + am;
28155f44c854SHong Zhang     uj_ptr     = (PetscInt**)(il + am);
28165f44c854SHong Zhang     uj_lvl_ptr = (PetscInt**)(uj_ptr + am);
28175f44c854SHong Zhang     for (i=0; i<am; i++){
28185f44c854SHong Zhang       jl[i] = am; il[i] = 0;
28195f44c854SHong Zhang     }
28205f44c854SHong Zhang 
28215f44c854SHong Zhang     /* create and initialize a linked list for storing column indices of the active row k */
28225f44c854SHong Zhang     nlnk = am + 1;
28235f44c854SHong Zhang     ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
28245f44c854SHong Zhang 
28255f44c854SHong Zhang     /* initial FreeSpace size is fill*(ai[am]+1) */
28265f44c854SHong Zhang     ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr);
28275f44c854SHong Zhang     current_space = free_space;
28285f44c854SHong Zhang     ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space_lvl);CHKERRQ(ierr);
28295f44c854SHong Zhang     current_space_lvl = free_space_lvl;
28305f44c854SHong Zhang 
28315f44c854SHong Zhang     for (k=0; k<am; k++){  /* for each active row k */
28325f44c854SHong Zhang       /* initialize lnk by the column indices of row rip[k] of A */
28335f44c854SHong Zhang       nzk   = 0;
28345f44c854SHong Zhang       ncols = ai[rip[k]+1] - ai[rip[k]];
28355f44c854SHong Zhang       if (!ncols) SETERRQ2(PETSC_ERR_MAT_CH_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",rip[k],k);
28365f44c854SHong Zhang       ncols_upper = 0;
28375f44c854SHong Zhang       for (j=0; j<ncols; j++){
28385f44c854SHong Zhang         i = *(aj + ai[rip[k]] + j); /* unpermuted column index */
28395f44c854SHong Zhang         if (riip[i] >= k){ /* only take upper triangular entry */
28405f44c854SHong Zhang           ajtmp[ncols_upper] = i;
28415f44c854SHong Zhang           ncols_upper++;
28425f44c854SHong Zhang         }
28435f44c854SHong Zhang       }
28445f44c854SHong Zhang       ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
28455f44c854SHong Zhang       nzk += nlnk;
28465f44c854SHong Zhang 
28475f44c854SHong Zhang       /* update lnk by computing fill-in for each pivot row to be merged in */
28485f44c854SHong Zhang       prow = jl[k]; /* 1st pivot row */
28495f44c854SHong Zhang 
28505f44c854SHong Zhang       while (prow < k){
28515f44c854SHong Zhang         nextprow = jl[prow];
28525f44c854SHong Zhang 
28535f44c854SHong Zhang         /* merge prow into k-th row */
28545f44c854SHong Zhang         jmin = il[prow] + 1;  /* index of the 2nd nzero entry in U(prow,k:am-1) */
28555f44c854SHong Zhang         jmax = ui[prow+1];
28565f44c854SHong Zhang         ncols = jmax-jmin;
28575f44c854SHong Zhang         i     = jmin - ui[prow];
28585f44c854SHong Zhang         cols  = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */
28595f44c854SHong Zhang         uj    = uj_lvl_ptr[prow] + i; /* levels of cols */
28605f44c854SHong Zhang         j     = *(uj - 1);
28615f44c854SHong Zhang         ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr);
28625f44c854SHong Zhang         nzk += nlnk;
28635f44c854SHong Zhang 
28645f44c854SHong Zhang         /* update il and jl for prow */
28655f44c854SHong Zhang         if (jmin < jmax){
28665f44c854SHong Zhang           il[prow] = jmin;
28675f44c854SHong Zhang           j = *cols; jl[prow] = jl[j]; jl[j] = prow;
28685f44c854SHong Zhang         }
28695f44c854SHong Zhang         prow = nextprow;
28705f44c854SHong Zhang       }
28715f44c854SHong Zhang 
28725f44c854SHong Zhang       /* if free space is not available, make more free space */
28735f44c854SHong Zhang       if (current_space->local_remaining<nzk) {
28745f44c854SHong Zhang         i = am - k + 1; /* num of unfactored rows */
28755f44c854SHong Zhang         i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */
28765f44c854SHong Zhang         ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
28775f44c854SHong Zhang         ierr = PetscFreeSpaceGet(i,&current_space_lvl);CHKERRQ(ierr);
28785f44c854SHong Zhang         reallocs++;
28795f44c854SHong Zhang       }
28805f44c854SHong Zhang 
28815f44c854SHong Zhang       /* copy data into free_space and free_space_lvl, then initialize lnk */
28825f44c854SHong Zhang       if (nzk == 0) SETERRQ1(PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k);
28835f44c854SHong Zhang       ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
28845f44c854SHong Zhang 
28855f44c854SHong Zhang       /* add the k-th row into il and jl */
28865f44c854SHong Zhang       if (nzk > 1){
28875f44c854SHong Zhang         i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
28885f44c854SHong Zhang         jl[k] = jl[i]; jl[i] = k;
28895f44c854SHong Zhang         il[k] = ui[k] + 1;
28905f44c854SHong Zhang       }
28915f44c854SHong Zhang       uj_ptr[k]     = current_space->array;
28925f44c854SHong Zhang       uj_lvl_ptr[k] = current_space_lvl->array;
28935f44c854SHong Zhang 
28945f44c854SHong Zhang       current_space->array           += nzk;
28955f44c854SHong Zhang       current_space->local_used      += nzk;
28965f44c854SHong Zhang       current_space->local_remaining -= nzk;
28975f44c854SHong Zhang 
28985f44c854SHong Zhang       current_space_lvl->array           += nzk;
28995f44c854SHong Zhang       current_space_lvl->local_used      += nzk;
29005f44c854SHong Zhang       current_space_lvl->local_remaining -= nzk;
29015f44c854SHong Zhang 
29025f44c854SHong Zhang       ui[k+1] = ui[k] + nzk;
29035f44c854SHong Zhang     }
29045f44c854SHong Zhang 
29055f44c854SHong Zhang #if defined(PETSC_USE_INFO)
29065f44c854SHong Zhang     if (ai[am] != 0) {
29075f44c854SHong Zhang       PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]);
29085f44c854SHong Zhang       ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr);
29095f44c854SHong Zhang       ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
29105f44c854SHong Zhang       ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr);
29115f44c854SHong Zhang     } else {
29125f44c854SHong Zhang       ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr);
29135f44c854SHong Zhang     }
29145f44c854SHong Zhang #endif
29155f44c854SHong Zhang 
29165f44c854SHong Zhang     ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
29175f44c854SHong Zhang     ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
29185f44c854SHong Zhang     ierr = PetscFree(jl);CHKERRQ(ierr);
29195f44c854SHong Zhang     ierr = PetscFree(ajtmp);CHKERRQ(ierr);
29205f44c854SHong Zhang 
29215f44c854SHong Zhang     /* destroy list of free space and other temporary array(s) */
29225f44c854SHong Zhang     ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
2923783ef271SHong Zhang     ierr = PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag);CHKERRQ(ierr);
29245f44c854SHong Zhang     ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
29255f44c854SHong Zhang     ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
29265f44c854SHong Zhang 
29275f44c854SHong Zhang   } /* end of case: levels>0 || (levels=0 && !perm_identity) */
29285f44c854SHong Zhang 
29295f44c854SHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
29305f44c854SHong Zhang 
29315f44c854SHong Zhang   b    = (Mat_SeqSBAIJ*)(fact)->data;
29325f44c854SHong Zhang   b->singlemalloc = PETSC_FALSE;
29335f44c854SHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr);
29345f44c854SHong Zhang   b->j    = uj;
29355f44c854SHong Zhang   b->i    = ui;
29365f44c854SHong Zhang   b->diag = udiag;
29377f53bb6cSHong Zhang   b->free_diag = PETSC_TRUE;
29385f44c854SHong Zhang   b->ilen = 0;
29395f44c854SHong Zhang   b->imax = 0;
29405f44c854SHong Zhang   b->row  = perm;
29415f44c854SHong Zhang   b->col  = perm;
29425f44c854SHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
29435f44c854SHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
29445f44c854SHong Zhang   b->icol = iperm;
29455f44c854SHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
29465f44c854SHong Zhang   ierr    = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
29475f44c854SHong Zhang   ierr = PetscLogObjectMemory((fact),(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
29485f44c854SHong Zhang   b->maxnz   = b->nz = ui[am];
29495f44c854SHong Zhang   b->free_a  = PETSC_TRUE;
29505f44c854SHong Zhang   b->free_ij = PETSC_TRUE;
29515f44c854SHong Zhang 
29525f44c854SHong Zhang   (fact)->info.factor_mallocs    = reallocs;
29535f44c854SHong Zhang   (fact)->info.fill_ratio_given  = fill;
29545f44c854SHong Zhang   if (ai[am] != 0) {
29555f44c854SHong Zhang     (fact)->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]);
29565f44c854SHong Zhang   } else {
29575f44c854SHong Zhang     (fact)->info.fill_ratio_needed = 0.0;
29585f44c854SHong Zhang   }
29595f44c854SHong Zhang   (fact)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_newdatastruct;
29605f44c854SHong Zhang   PetscFunctionReturn(0);
29615f44c854SHong Zhang }
29625f44c854SHong Zhang 
29635f44c854SHong Zhang #undef __FUNCT__
29645f44c854SHong Zhang #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ"
29655f44c854SHong Zhang PetscErrorCode MatICCFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
29665f44c854SHong Zhang {
29675f44c854SHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
29685f44c854SHong Zhang   Mat_SeqSBAIJ       *b;
29695f44c854SHong Zhang   PetscErrorCode     ierr;
29705f44c854SHong Zhang   PetscTruth         perm_identity,missing;
29715f44c854SHong Zhang   PetscInt           reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag;
29725f44c854SHong Zhang   const PetscInt     *rip,*riip;
29735f44c854SHong Zhang   PetscInt           jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow;
29745f44c854SHong Zhang   PetscInt           nlnk,*lnk,*lnk_lvl=PETSC_NULL,d;
29755f44c854SHong Zhang   PetscInt           ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr;
29765f44c854SHong Zhang   PetscReal          fill=info->fill,levels=info->levels;
29775f44c854SHong Zhang   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
29785f44c854SHong Zhang   PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL;
29795f44c854SHong Zhang   PetscBT            lnkbt;
29805f44c854SHong Zhang   IS                 iperm;
29815f44c854SHong Zhang   PetscTruth         newdatastruct=PETSC_FALSE;
29825f44c854SHong Zhang 
29835f44c854SHong Zhang   PetscFunctionBegin;
29845f44c854SHong Zhang   ierr = PetscOptionsGetTruth(PETSC_NULL,"-icc_new",&newdatastruct,PETSC_NULL);CHKERRQ(ierr);
29855f44c854SHong Zhang   if(newdatastruct){
29865f44c854SHong Zhang     ierr = MatICCFactorSymbolic_SeqAIJ_newdatastruct(fact,A,perm,info);CHKERRQ(ierr);
29875f44c854SHong Zhang     PetscFunctionReturn(0);
29885f44c854SHong Zhang   }
29895f44c854SHong Zhang 
29905f44c854SHong Zhang   if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n);
29915f44c854SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr);
29925f44c854SHong Zhang   if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d);
29935f44c854SHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
29945f44c854SHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
29955f44c854SHong Zhang 
29965f44c854SHong Zhang   ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr);
29975f44c854SHong Zhang   ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr);
29985f44c854SHong Zhang   ui[0] = 0;
29995f44c854SHong Zhang 
30005f44c854SHong Zhang   /* ICC(0) without matrix ordering: simply copies fill pattern */
30015f44c854SHong Zhang   if (!levels && perm_identity) {
30025f44c854SHong Zhang 
30035f44c854SHong Zhang     for (i=0; i<am; i++) {
30045f44c854SHong Zhang       ui[i+1]  = ui[i] + ai[i+1] - a->diag[i];
30055f44c854SHong Zhang       udiag[i] = ui[i];
3006ed59401aSHong Zhang     }
300739e3d630SHong Zhang     ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
300839e3d630SHong Zhang     cols = uj;
3009ed59401aSHong Zhang     for (i=0; i<am; i++) {
3010ed59401aSHong Zhang       aj    = a->j + a->diag[i];
301139e3d630SHong Zhang       ncols = ui[i+1] - ui[i];
301239e3d630SHong Zhang       for (j=0; j<ncols; j++) *cols++ = *aj++;
3013ed59401aSHong Zhang     }
301439e3d630SHong Zhang   } else { /* case: levels>0 || (levels=0 && !perm_identity) */
3015b635d36bSHong Zhang     ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
3016b635d36bSHong Zhang     ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
3017b635d36bSHong Zhang 
30187a48dd6fSHong Zhang     /* initialization */
30195a8e39fbSHong Zhang     ierr  = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr);
30207a48dd6fSHong Zhang 
30217a48dd6fSHong Zhang     /* jl: linked list for storing indices of the pivot rows
30227a48dd6fSHong Zhang        il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
30231393bc97SHong Zhang     ierr = PetscMalloc((2*am+1)*sizeof(PetscInt)+2*am*sizeof(PetscInt**),&jl);CHKERRQ(ierr);
30247a48dd6fSHong Zhang     il         = jl + am;
30257a48dd6fSHong Zhang     uj_ptr     = (PetscInt**)(il + am);
30267a48dd6fSHong Zhang     uj_lvl_ptr = (PetscInt**)(uj_ptr + am);
30277a48dd6fSHong Zhang     for (i=0; i<am; i++){
30287a48dd6fSHong Zhang       jl[i] = am; il[i] = 0;
30297a48dd6fSHong Zhang     }
30307a48dd6fSHong Zhang 
30317a48dd6fSHong Zhang     /* create and initialize a linked list for storing column indices of the active row k */
30327a48dd6fSHong Zhang     nlnk = am + 1;
30332ed133dbSHong Zhang     ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
30347a48dd6fSHong Zhang 
30357a48dd6fSHong Zhang     /* initial FreeSpace size is fill*(ai[am]+1) */
3036a1a86e44SBarry Smith     ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr);
30377a48dd6fSHong Zhang     current_space = free_space;
3038a1a86e44SBarry Smith     ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space_lvl);CHKERRQ(ierr);
30397a48dd6fSHong Zhang     current_space_lvl = free_space_lvl;
30407a48dd6fSHong Zhang 
30417a48dd6fSHong Zhang     for (k=0; k<am; k++){  /* for each active row k */
30427a48dd6fSHong Zhang       /* initialize lnk by the column indices of row rip[k] of A */
30437a48dd6fSHong Zhang       nzk   = 0;
3044622e2028SHong Zhang       ncols = ai[rip[k]+1] - ai[rip[k]];
30453b4a8b6dSBarry Smith       if (!ncols) SETERRQ2(PETSC_ERR_MAT_CH_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",rip[k],k);
3046622e2028SHong Zhang       ncols_upper = 0;
3047622e2028SHong Zhang       for (j=0; j<ncols; j++){
3048b635d36bSHong Zhang         i = *(aj + ai[rip[k]] + j); /* unpermuted column index */
3049b635d36bSHong Zhang         if (riip[i] >= k){ /* only take upper triangular entry */
30505a8e39fbSHong Zhang           ajtmp[ncols_upper] = i;
3051622e2028SHong Zhang           ncols_upper++;
3052622e2028SHong Zhang         }
3053622e2028SHong Zhang       }
3054b635d36bSHong Zhang       ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
30557a48dd6fSHong Zhang       nzk += nlnk;
30567a48dd6fSHong Zhang 
30577a48dd6fSHong Zhang       /* update lnk by computing fill-in for each pivot row to be merged in */
30587a48dd6fSHong Zhang       prow = jl[k]; /* 1st pivot row */
30597a48dd6fSHong Zhang 
30607a48dd6fSHong Zhang       while (prow < k){
30617a48dd6fSHong Zhang         nextprow = jl[prow];
30627a48dd6fSHong Zhang 
30637a48dd6fSHong Zhang         /* merge prow into k-th row */
30647a48dd6fSHong Zhang         jmin = il[prow] + 1;  /* index of the 2nd nzero entry in U(prow,k:am-1) */
30657a48dd6fSHong Zhang         jmax = ui[prow+1];
30667a48dd6fSHong Zhang         ncols = jmax-jmin;
3067ed59401aSHong Zhang         i     = jmin - ui[prow];
3068ed59401aSHong Zhang         cols  = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */
30695a8e39fbSHong Zhang         uj    = uj_lvl_ptr[prow] + i; /* levels of cols */
30705a8e39fbSHong Zhang         j     = *(uj - 1);
30715a8e39fbSHong Zhang         ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr);
30727a48dd6fSHong Zhang         nzk += nlnk;
30737a48dd6fSHong Zhang 
30747a48dd6fSHong Zhang         /* update il and jl for prow */
30757a48dd6fSHong Zhang         if (jmin < jmax){
30767a48dd6fSHong Zhang           il[prow] = jmin;
30777a48dd6fSHong Zhang           j = *cols; jl[prow] = jl[j]; jl[j] = prow;
30787a48dd6fSHong Zhang         }
30797a48dd6fSHong Zhang         prow = nextprow;
30807a48dd6fSHong Zhang       }
30817a48dd6fSHong Zhang 
30827a48dd6fSHong Zhang       /* if free space is not available, make more free space */
30837a48dd6fSHong Zhang       if (current_space->local_remaining<nzk) {
30847a48dd6fSHong Zhang         i = am - k + 1; /* num of unfactored rows */
30857a48dd6fSHong Zhang         i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */
3086a1a86e44SBarry Smith         ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
3087a1a86e44SBarry Smith         ierr = PetscFreeSpaceGet(i,&current_space_lvl);CHKERRQ(ierr);
30887a48dd6fSHong Zhang         reallocs++;
30897a48dd6fSHong Zhang       }
30907a48dd6fSHong Zhang 
30917a48dd6fSHong Zhang       /* copy data into free_space and free_space_lvl, then initialize lnk */
3092b635d36bSHong Zhang       if (nzk == 0) SETERRQ1(PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k);
30932ed133dbSHong Zhang       ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
30947a48dd6fSHong Zhang 
30957a48dd6fSHong Zhang       /* add the k-th row into il and jl */
309639e3d630SHong Zhang       if (nzk > 1){
30977a48dd6fSHong Zhang         i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
30987a48dd6fSHong Zhang         jl[k] = jl[i]; jl[i] = k;
30997a48dd6fSHong Zhang         il[k] = ui[k] + 1;
31007a48dd6fSHong Zhang       }
31017a48dd6fSHong Zhang       uj_ptr[k]     = current_space->array;
31027a48dd6fSHong Zhang       uj_lvl_ptr[k] = current_space_lvl->array;
31037a48dd6fSHong Zhang 
31047a48dd6fSHong Zhang       current_space->array           += nzk;
31057a48dd6fSHong Zhang       current_space->local_used      += nzk;
31067a48dd6fSHong Zhang       current_space->local_remaining -= nzk;
31077a48dd6fSHong Zhang 
31087a48dd6fSHong Zhang       current_space_lvl->array           += nzk;
31097a48dd6fSHong Zhang       current_space_lvl->local_used      += nzk;
31107a48dd6fSHong Zhang       current_space_lvl->local_remaining -= nzk;
31117a48dd6fSHong Zhang 
31127a48dd6fSHong Zhang       ui[k+1] = ui[k] + nzk;
31137a48dd6fSHong Zhang     }
31147a48dd6fSHong Zhang 
31156cf91177SBarry Smith #if defined(PETSC_USE_INFO)
31167a48dd6fSHong Zhang     if (ai[am] != 0) {
311739e3d630SHong Zhang       PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]);
3118ae15b995SBarry Smith       ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr);
3119ae15b995SBarry Smith       ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
3120ae15b995SBarry Smith       ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr);
31217a48dd6fSHong Zhang     } else {
3122ae15b995SBarry Smith       ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr);
31237a48dd6fSHong Zhang     }
312463ba0a88SBarry Smith #endif
31257a48dd6fSHong Zhang 
31267a48dd6fSHong Zhang     ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
3127b635d36bSHong Zhang     ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
31287a48dd6fSHong Zhang     ierr = PetscFree(jl);CHKERRQ(ierr);
31295a8e39fbSHong Zhang     ierr = PetscFree(ajtmp);CHKERRQ(ierr);
31307a48dd6fSHong Zhang 
31317a48dd6fSHong Zhang     /* destroy list of free space and other temporary array(s) */
31327a48dd6fSHong Zhang     ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
3133a1a86e44SBarry Smith     ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr);
31342ed133dbSHong Zhang     ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
3135a1a86e44SBarry Smith     ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
31367a48dd6fSHong Zhang 
313739e3d630SHong Zhang   } /* end of case: levels>0 || (levels=0 && !perm_identity) */
313839e3d630SHong Zhang 
31397a48dd6fSHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
31407a48dd6fSHong Zhang 
3141719d5645SBarry Smith   b    = (Mat_SeqSBAIJ*)(fact)->data;
31427a48dd6fSHong Zhang   b->singlemalloc = PETSC_FALSE;
31437a48dd6fSHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr);
31447a48dd6fSHong Zhang   b->j    = uj;
31457a48dd6fSHong Zhang   b->i    = ui;
31465f44c854SHong Zhang   b->diag = udiag;
31477f53bb6cSHong Zhang   b->free_diag = PETSC_TRUE;
31487a48dd6fSHong Zhang   b->ilen = 0;
31497a48dd6fSHong Zhang   b->imax = 0;
31507a48dd6fSHong Zhang   b->row  = perm;
3151b635d36bSHong Zhang   b->col  = perm;
3152b635d36bSHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
3153b635d36bSHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
3154b635d36bSHong Zhang   b->icol = iperm;
31557a48dd6fSHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
31567a48dd6fSHong Zhang   ierr    = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
3157719d5645SBarry Smith   ierr = PetscLogObjectMemory((fact),(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
31587a48dd6fSHong Zhang   b->maxnz   = b->nz = ui[am];
3159e6b907acSBarry Smith   b->free_a  = PETSC_TRUE;
3160e6b907acSBarry Smith   b->free_ij = PETSC_TRUE;
31617a48dd6fSHong Zhang 
3162719d5645SBarry Smith   (fact)->info.factor_mallocs    = reallocs;
3163719d5645SBarry Smith   (fact)->info.fill_ratio_given  = fill;
31647a48dd6fSHong Zhang   if (ai[am] != 0) {
3165719d5645SBarry Smith     (fact)->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]);
31667a48dd6fSHong Zhang   } else {
3167719d5645SBarry Smith     (fact)->info.fill_ratio_needed = 0.0;
31687a48dd6fSHong Zhang   }
3169719d5645SBarry Smith   (fact)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ;
3170a6175056SHong Zhang   PetscFunctionReturn(0);
3171a6175056SHong Zhang }
3172d5d45c9bSBarry Smith 
3173f76d2b81SHong Zhang #undef __FUNCT__
3174f76d2b81SHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_SeqAIJ"
31750481f469SBarry Smith PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
3176f76d2b81SHong Zhang {
3177f76d2b81SHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
317810c27e3dSHong Zhang   Mat_SeqSBAIJ       *b;
3179dfbe8321SBarry Smith   PetscErrorCode     ierr;
3180f76d2b81SHong Zhang   PetscTruth         perm_identity;
318110c27e3dSHong Zhang   PetscReal          fill = info->fill;
31825d0c19d7SBarry Smith   const PetscInt     *rip,*riip;
31835d0c19d7SBarry Smith   PetscInt           i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow;
318410c27e3dSHong Zhang   PetscInt           *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow;
31852ed133dbSHong Zhang   PetscInt           nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr;
3186a1a86e44SBarry Smith   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
318710c27e3dSHong Zhang   PetscBT            lnkbt;
31882ed133dbSHong Zhang   IS                 iperm;
3189f76d2b81SHong Zhang 
3190f76d2b81SHong Zhang   PetscFunctionBegin;
3191d0f46423SBarry Smith   if (A->rmap->n != A->cmap->n) SETERRQ2(PETSC_ERR_ARG_WRONG,"Must be square matrix, rows %D columns %D",A->rmap->n,A->cmap->n);
31922ed133dbSHong Zhang   /* check whether perm is the identity mapping */
3193f76d2b81SHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
31942ed133dbSHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
31952ed133dbSHong Zhang   ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
31969bfd6278SHong Zhang   ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
319710c27e3dSHong Zhang 
319810c27e3dSHong Zhang   /* initialization */
31991393bc97SHong Zhang   ierr  = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr);
320010c27e3dSHong Zhang   ui[0] = 0;
320110c27e3dSHong Zhang 
320210c27e3dSHong Zhang   /* jl: linked list for storing indices of the pivot rows
32031393bc97SHong Zhang      il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
32041393bc97SHong Zhang   ierr = PetscMalloc((3*am+1)*sizeof(PetscInt)+am*sizeof(PetscInt**),&jl);CHKERRQ(ierr);
32051393bc97SHong Zhang   il     = jl + am;
32061393bc97SHong Zhang   cols   = il + am;
32071393bc97SHong Zhang   ui_ptr = (PetscInt**)(cols + am);
32081393bc97SHong Zhang   for (i=0; i<am; i++){
32091393bc97SHong Zhang     jl[i] = am; il[i] = 0;
3210f76d2b81SHong Zhang   }
321110c27e3dSHong Zhang 
321210c27e3dSHong Zhang   /* create and initialize a linked list for storing column indices of the active row k */
32131393bc97SHong Zhang   nlnk = am + 1;
32141393bc97SHong Zhang   ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
321510c27e3dSHong Zhang 
32161393bc97SHong Zhang   /* initial FreeSpace size is fill*(ai[am]+1) */
3217a1a86e44SBarry Smith   ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr);
321810c27e3dSHong Zhang   current_space = free_space;
321910c27e3dSHong Zhang 
32201393bc97SHong Zhang   for (k=0; k<am; k++){  /* for each active row k */
322110c27e3dSHong Zhang     /* initialize lnk by the column indices of row rip[k] of A */
322210c27e3dSHong Zhang     nzk   = 0;
32232ed133dbSHong Zhang     ncols = ai[rip[k]+1] - ai[rip[k]];
32243b4a8b6dSBarry Smith     if (!ncols) SETERRQ2(PETSC_ERR_MAT_CH_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",rip[k],k);
32252ed133dbSHong Zhang     ncols_upper = 0;
32262ed133dbSHong Zhang     for (j=0; j<ncols; j++){
32279bfd6278SHong Zhang       i = riip[*(aj + ai[rip[k]] + j)];
32282ed133dbSHong Zhang       if (i >= k){ /* only take upper triangular entry */
32292ed133dbSHong Zhang         cols[ncols_upper] = i;
32302ed133dbSHong Zhang         ncols_upper++;
32312ed133dbSHong Zhang       }
32322ed133dbSHong Zhang     }
32331393bc97SHong Zhang     ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
323410c27e3dSHong Zhang     nzk += nlnk;
323510c27e3dSHong Zhang 
323610c27e3dSHong Zhang     /* update lnk by computing fill-in for each pivot row to be merged in */
323710c27e3dSHong Zhang     prow = jl[k]; /* 1st pivot row */
323810c27e3dSHong Zhang 
323910c27e3dSHong Zhang     while (prow < k){
324010c27e3dSHong Zhang       nextprow = jl[prow];
324110c27e3dSHong Zhang       /* merge prow into k-th row */
32421393bc97SHong Zhang       jmin = il[prow] + 1;  /* index of the 2nd nzero entry in U(prow,k:am-1) */
324310c27e3dSHong Zhang       jmax = ui[prow+1];
324410c27e3dSHong Zhang       ncols = jmax-jmin;
32451393bc97SHong Zhang       uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */
32465a8e39fbSHong Zhang       ierr = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
324710c27e3dSHong Zhang       nzk += nlnk;
324810c27e3dSHong Zhang 
324910c27e3dSHong Zhang       /* update il and jl for prow */
325010c27e3dSHong Zhang       if (jmin < jmax){
325110c27e3dSHong Zhang         il[prow] = jmin;
32522ed133dbSHong Zhang         j = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow;
325310c27e3dSHong Zhang       }
325410c27e3dSHong Zhang       prow = nextprow;
325510c27e3dSHong Zhang     }
325610c27e3dSHong Zhang 
325710c27e3dSHong Zhang     /* if free space is not available, make more free space */
325810c27e3dSHong Zhang     if (current_space->local_remaining<nzk) {
32591393bc97SHong Zhang       i = am - k + 1; /* num of unfactored rows */
326010c27e3dSHong Zhang       i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */
3261a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
326210c27e3dSHong Zhang       reallocs++;
326310c27e3dSHong Zhang     }
326410c27e3dSHong Zhang 
326510c27e3dSHong Zhang     /* copy data into free space, then initialize lnk */
32661393bc97SHong Zhang     ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
326710c27e3dSHong Zhang 
326810c27e3dSHong Zhang     /* add the k-th row into il and jl */
326910c27e3dSHong Zhang     if (nzk-1 > 0){
32701393bc97SHong Zhang       i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
327110c27e3dSHong Zhang       jl[k] = jl[i]; jl[i] = k;
327210c27e3dSHong Zhang       il[k] = ui[k] + 1;
327310c27e3dSHong Zhang     }
32742ed133dbSHong Zhang     ui_ptr[k] = current_space->array;
327510c27e3dSHong Zhang     current_space->array           += nzk;
327610c27e3dSHong Zhang     current_space->local_used      += nzk;
327710c27e3dSHong Zhang     current_space->local_remaining -= nzk;
327810c27e3dSHong Zhang 
327910c27e3dSHong Zhang     ui[k+1] = ui[k] + nzk;
328010c27e3dSHong Zhang   }
328110c27e3dSHong Zhang 
32826cf91177SBarry Smith #if defined(PETSC_USE_INFO)
32831393bc97SHong Zhang   if (ai[am] != 0) {
32841393bc97SHong Zhang     PetscReal af = (PetscReal)(ui[am])/((PetscReal)ai[am]);
3285ae15b995SBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr);
3286ae15b995SBarry Smith     ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
3287ae15b995SBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr);
328810c27e3dSHong Zhang   } else {
3289ae15b995SBarry Smith      ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr);
329010c27e3dSHong Zhang   }
329163ba0a88SBarry Smith #endif
329210c27e3dSHong Zhang 
329310c27e3dSHong Zhang   ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
32949bfd6278SHong Zhang   ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
329510c27e3dSHong Zhang   ierr = PetscFree(jl);CHKERRQ(ierr);
329610c27e3dSHong Zhang 
329710c27e3dSHong Zhang   /* destroy list of free space and other temporary array(s) */
32981393bc97SHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
3299a1a86e44SBarry Smith   ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr);
330010c27e3dSHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
330110c27e3dSHong Zhang 
330210c27e3dSHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
330310c27e3dSHong Zhang 
3304719d5645SBarry Smith   b = (Mat_SeqSBAIJ*)(fact)->data;
330510c27e3dSHong Zhang   b->singlemalloc = PETSC_FALSE;
3306e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
3307e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
33081393bc97SHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr);
330910c27e3dSHong Zhang   b->j    = uj;
331010c27e3dSHong Zhang   b->i    = ui;
331110c27e3dSHong Zhang   b->diag = 0;
331210c27e3dSHong Zhang   b->ilen = 0;
331310c27e3dSHong Zhang   b->imax = 0;
331410c27e3dSHong Zhang   b->row  = perm;
33159bfd6278SHong Zhang   b->col  = perm;
33169bfd6278SHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
33179bfd6278SHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
33189bfd6278SHong Zhang   b->icol = iperm;
331910c27e3dSHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
33201393bc97SHong Zhang   ierr    = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
3321719d5645SBarry Smith   ierr    = PetscLogObjectMemory(fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
33221393bc97SHong Zhang   b->maxnz = b->nz = ui[am];
332310c27e3dSHong Zhang 
3324719d5645SBarry Smith   (fact)->info.factor_mallocs    = reallocs;
3325719d5645SBarry Smith   (fact)->info.fill_ratio_given  = fill;
33261393bc97SHong Zhang   if (ai[am] != 0) {
3327719d5645SBarry Smith     (fact)->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]);
332810c27e3dSHong Zhang   } else {
3329719d5645SBarry Smith     (fact)->info.fill_ratio_needed = 0.0;
333010c27e3dSHong Zhang   }
3331719d5645SBarry Smith   (fact)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ;
3332f76d2b81SHong Zhang   PetscFunctionReturn(0);
3333f76d2b81SHong Zhang }
3334599ef60dSHong Zhang 
3335599ef60dSHong Zhang #undef __FUNCT__
3336894a2336SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering_newdatastruct"
3337894a2336SHong Zhang PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering_newdatastruct(Mat A,Vec bb,Vec xx)
3338599ef60dSHong Zhang {
33391d098868SHong Zhang   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
33401d098868SHong Zhang   PetscErrorCode    ierr;
33411d098868SHong Zhang   PetscInt          n = A->rmap->n;
33428fc3a347SHong Zhang   const PetscInt    *ai = a->i,*aj = a->j,*vi;
33431d098868SHong Zhang   PetscScalar       *x,sum;
33441d098868SHong Zhang   const PetscScalar *b;
33451d098868SHong Zhang   const MatScalar   *aa = a->a,*v;
33468fc3a347SHong Zhang   PetscInt          i,nz;
33471d098868SHong Zhang 
33481d098868SHong Zhang   PetscFunctionBegin;
33491d098868SHong Zhang   if (!n) PetscFunctionReturn(0);
33501d098868SHong Zhang 
33511d098868SHong Zhang   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
33521d098868SHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
33531d098868SHong Zhang 
33541d098868SHong Zhang   /* forward solve the lower triangular */
33551d098868SHong Zhang   x[0] = b[0];
33561d098868SHong Zhang   v    = aa;
33571d098868SHong Zhang   vi   = aj;
33581d098868SHong Zhang   for (i=1; i<n; i++) {
33591d098868SHong Zhang     nz  = ai[i+1] - ai[i];
33601d098868SHong Zhang     sum = b[i];
3361003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
3362e0e43300SBarry Smith     v  += nz;
3363e0e43300SBarry Smith     vi += nz;
33641d098868SHong Zhang     x[i] = sum;
33651d098868SHong Zhang   }
33661d098868SHong Zhang 
33671d098868SHong Zhang   /* backward solve the upper triangular */
33688fc3a347SHong Zhang   v   = aa + ai[n+1];
33698fc3a347SHong Zhang   vi  = aj + ai[n+1];
33701d098868SHong Zhang   for (i=n-1; i>=0; i--){
33718fc3a347SHong Zhang     nz = ai[2*n-i +1] - ai[2*n-i]-1;
33721d098868SHong Zhang     sum = x[i];
3373003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
3374e0e43300SBarry Smith     v   += nz;
33758fc3a347SHong Zhang     vi  += nz; vi++;
3376fe31b70dSHong Zhang     x[i] = *v++ *sum; /* x[i]=aa[adiag[i]]*sum; v++; */
33771d098868SHong Zhang   }
33781d098868SHong Zhang 
33791d098868SHong Zhang   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
33801d098868SHong Zhang   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
33811d098868SHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
33821d098868SHong Zhang   PetscFunctionReturn(0);
33831d098868SHong Zhang }
33841d098868SHong Zhang 
33851d098868SHong Zhang #undef __FUNCT__
3386813a1f2bSShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering_newdatastruct_v2"
3387813a1f2bSShri Abhyankar PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering_newdatastruct_v2(Mat A,Vec bb,Vec xx)
3388813a1f2bSShri Abhyankar {
3389813a1f2bSShri Abhyankar   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
3390813a1f2bSShri Abhyankar   PetscErrorCode    ierr;
3391813a1f2bSShri Abhyankar   PetscInt          n = A->rmap->n;
3392813a1f2bSShri Abhyankar   const PetscInt    *ai = a->i,*aj = a->j,*adiag = a->diag,*vi;
3393813a1f2bSShri Abhyankar   PetscScalar       *x,sum;
3394813a1f2bSShri Abhyankar   const PetscScalar *b;
3395813a1f2bSShri Abhyankar   const MatScalar   *aa = a->a,*v;
3396813a1f2bSShri Abhyankar   PetscInt          i,nz;
3397813a1f2bSShri Abhyankar 
3398813a1f2bSShri Abhyankar   PetscFunctionBegin;
3399813a1f2bSShri Abhyankar   if (!n) PetscFunctionReturn(0);
3400813a1f2bSShri Abhyankar 
3401813a1f2bSShri Abhyankar   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
3402813a1f2bSShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
3403813a1f2bSShri Abhyankar 
3404813a1f2bSShri Abhyankar   /* forward solve the lower triangular */
3405813a1f2bSShri Abhyankar   x[0] = b[0];
3406813a1f2bSShri Abhyankar   v    = aa;
3407813a1f2bSShri Abhyankar   vi   = aj;
3408813a1f2bSShri Abhyankar   for (i=1; i<n; i++) {
3409813a1f2bSShri Abhyankar     nz  = ai[i+1] - ai[i];
3410813a1f2bSShri Abhyankar     sum = b[i];
3411813a1f2bSShri Abhyankar     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
3412813a1f2bSShri Abhyankar     v  += nz;
3413813a1f2bSShri Abhyankar     vi += nz;
3414813a1f2bSShri Abhyankar     x[i] = sum;
3415813a1f2bSShri Abhyankar   }
3416813a1f2bSShri Abhyankar 
3417813a1f2bSShri Abhyankar   /* backward solve the upper triangular */
3418813a1f2bSShri Abhyankar   /*  v   = aa + ai[n+1];
3419813a1f2bSShri Abhyankar       vi  = aj + ai[n+1]; */
3420813a1f2bSShri Abhyankar   v  = aa + adiag[n-1];
3421813a1f2bSShri Abhyankar   vi = aj + adiag[n-1];
342262fbd6afSShri Abhyankar   for (i=n-1; i>=0; i--){
3423813a1f2bSShri Abhyankar     nz = adiag[i] - adiag[i+1]-1;
3424813a1f2bSShri Abhyankar     sum = x[i];
3425813a1f2bSShri Abhyankar     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
3426813a1f2bSShri Abhyankar     v   += nz;
3427813a1f2bSShri Abhyankar     vi  += nz; vi++;
3428813a1f2bSShri Abhyankar     x[i] = *v++ *sum; /* x[i]=aa[adiag[i]]*sum; v++; */
3429813a1f2bSShri Abhyankar   }
3430813a1f2bSShri Abhyankar 
3431813a1f2bSShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
3432813a1f2bSShri Abhyankar   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
3433813a1f2bSShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
3434813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
3435813a1f2bSShri Abhyankar }
3436813a1f2bSShri Abhyankar 
3437813a1f2bSShri Abhyankar #undef __FUNCT__
3438894a2336SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_newdatastruct"
3439894a2336SHong Zhang PetscErrorCode MatSolve_SeqAIJ_newdatastruct(Mat A,Vec bb,Vec xx)
34401d098868SHong Zhang {
34411d098868SHong Zhang   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
34421d098868SHong Zhang   IS                iscol = a->col,isrow = a->row;
34431d098868SHong Zhang   PetscErrorCode    ierr;
3444894a2336SHong Zhang   PetscInt          i,n=A->rmap->n,*vi,*ai=a->i,*aj=a->j,nz,k;
34451d098868SHong Zhang   const PetscInt    *rout,*cout,*r,*c;
3446894a2336SHong Zhang   PetscScalar       *x,*tmp,*tmps,sum;
34471d098868SHong Zhang   const PetscScalar *b;
34481d098868SHong Zhang   const MatScalar   *aa = a->a,*v;
34491d098868SHong Zhang 
34501d098868SHong Zhang   PetscFunctionBegin;
34511d098868SHong Zhang   if (!n) PetscFunctionReturn(0);
34521d098868SHong Zhang 
34531d098868SHong Zhang   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
34541d098868SHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
34551d098868SHong Zhang   tmp  = a->solve_work;
34561d098868SHong Zhang 
34571d098868SHong Zhang   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
3458894a2336SHong Zhang   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
34591d098868SHong Zhang 
34601d098868SHong Zhang   /* forward solve the lower triangular */
34611d098868SHong Zhang   tmp[0] = b[*r++];
34621d098868SHong Zhang   tmps   = tmp;
34631d098868SHong Zhang   v      = aa;
34641d098868SHong Zhang   vi     = aj;
34651d098868SHong Zhang   for (i=1; i<n; i++) {
34661d098868SHong Zhang     nz  = ai[i+1] - ai[i];
3467894a2336SHong Zhang     sum = b[*r++];
3468894a2336SHong Zhang     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
3469894a2336SHong Zhang     tmp[i] = sum;
34701d098868SHong Zhang     v += nz; vi += nz;
34711d098868SHong Zhang   }
34721d098868SHong Zhang 
34731d098868SHong Zhang   /* backward solve the upper triangular */
3474894a2336SHong Zhang   k  = n+1;
3475894a2336SHong Zhang   v  = aa + ai[k]; /* 1st entry of U(n-1,:) */
3476894a2336SHong Zhang   vi = aj + ai[k];
34771d098868SHong Zhang   for (i=n-1; i>=0; i--){
3478894a2336SHong Zhang     k  = 2*n-i;
3479894a2336SHong Zhang     nz = ai[k +1] - ai[k] - 1;
3480894a2336SHong Zhang     sum = tmp[i];
3481894a2336SHong Zhang     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
3482894a2336SHong Zhang     x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */
34831d098868SHong Zhang     v += nz+1; vi += nz+1;
34841d098868SHong Zhang   }
34851d098868SHong Zhang 
34861d098868SHong Zhang   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
34871d098868SHong Zhang   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
34881d098868SHong Zhang   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
34891d098868SHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
34901d098868SHong Zhang   ierr = PetscLogFlops(2*a->nz - A->cmap->n);CHKERRQ(ierr);
34911d098868SHong Zhang   PetscFunctionReturn(0);
34921d098868SHong Zhang }
34931d098868SHong Zhang 
34941d098868SHong Zhang #undef __FUNCT__
3495f268cba8SShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ_newdatastruct_v2"
3496f268cba8SShri Abhyankar PetscErrorCode MatSolve_SeqAIJ_newdatastruct_v2(Mat A,Vec bb,Vec xx)
3497f268cba8SShri Abhyankar {
3498f268cba8SShri Abhyankar   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
3499f268cba8SShri Abhyankar   IS                iscol = a->col,isrow = a->row;
3500f268cba8SShri Abhyankar   PetscErrorCode    ierr;
3501f268cba8SShri Abhyankar   PetscInt          i,n=A->rmap->n,*vi,*ai=a->i,*aj=a->j,*adiag = a->diag,nz;
3502f268cba8SShri Abhyankar   const PetscInt    *rout,*cout,*r,*c;
3503f268cba8SShri Abhyankar   PetscScalar       *x,*tmp,sum;
3504f268cba8SShri Abhyankar   const PetscScalar *b;
3505f268cba8SShri Abhyankar   const MatScalar   *aa = a->a,*v;
3506f268cba8SShri Abhyankar 
3507f268cba8SShri Abhyankar   PetscFunctionBegin;
3508f268cba8SShri Abhyankar   if (!n) PetscFunctionReturn(0);
3509f268cba8SShri Abhyankar 
3510f268cba8SShri Abhyankar   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
3511f268cba8SShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
3512f268cba8SShri Abhyankar   tmp  = a->solve_work;
3513f268cba8SShri Abhyankar 
3514f268cba8SShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
3515f268cba8SShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
3516f268cba8SShri Abhyankar 
3517f268cba8SShri Abhyankar   /* forward solve the lower triangular */
3518f268cba8SShri Abhyankar   /*  tmp[0] = b[*r++]; */
3519f268cba8SShri Abhyankar   tmp[0] = b[r[0]];
3520f268cba8SShri Abhyankar   v      = aa;
3521f268cba8SShri Abhyankar   vi     = aj;
3522f268cba8SShri Abhyankar   for (i=1; i<n; i++) {
3523f268cba8SShri Abhyankar     nz  = ai[i+1] - ai[i];
3524f268cba8SShri Abhyankar     sum = b[r[i]];
3525f268cba8SShri Abhyankar     PetscSparseDenseMinusDot(sum,tmp,v,vi,nz);
3526f268cba8SShri Abhyankar     tmp[i] = sum;
3527f268cba8SShri Abhyankar     v += nz; vi += nz;
3528f268cba8SShri Abhyankar   }
3529f268cba8SShri Abhyankar 
3530f268cba8SShri Abhyankar   /* backward solve the upper triangular */
3531f268cba8SShri Abhyankar   /*  v  = aa + ai[k]; *//* 1st entry of U(n-1,:) */
3532f268cba8SShri Abhyankar   /* vi = aj + ai[k]; */
3533f268cba8SShri Abhyankar   v  = aa + adiag[n-1];
3534f268cba8SShri Abhyankar   vi = aj + adiag[n-1];
3535f268cba8SShri Abhyankar   for (i=n-1; i>=0; i--){
3536f268cba8SShri Abhyankar     /* nz = ai[k +1] - ai[k] - 1;*/
3537f268cba8SShri Abhyankar     nz = adiag[i]-adiag[i+1]-1;
3538f268cba8SShri Abhyankar     sum = tmp[i];
3539f268cba8SShri Abhyankar     PetscSparseDenseMinusDot(sum,tmp,v,vi,nz);
3540f268cba8SShri Abhyankar     x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */
3541f268cba8SShri Abhyankar     v += nz+1; vi += nz+1;
3542f268cba8SShri Abhyankar   }
3543f268cba8SShri Abhyankar 
3544f268cba8SShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
3545f268cba8SShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
3546f268cba8SShri Abhyankar   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
3547f268cba8SShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
3548f268cba8SShri Abhyankar   ierr = PetscLogFlops(2*a->nz - A->cmap->n);CHKERRQ(ierr);
3549f268cba8SShri Abhyankar   PetscFunctionReturn(0);
3550f268cba8SShri Abhyankar }
3551f268cba8SShri Abhyankar 
3552f268cba8SShri Abhyankar #undef __FUNCT__
35531d098868SHong Zhang #define __FUNCT__ "MatILUDTFactor_SeqAIJ"
35541d098868SHong Zhang PetscErrorCode MatILUDTFactor_SeqAIJ(Mat A,IS isrow,IS iscol,const MatFactorInfo *info,Mat *fact)
35551d098868SHong Zhang {
35561d098868SHong Zhang   Mat                B = *fact;
3557599ef60dSHong Zhang   Mat_SeqAIJ         *a=(Mat_SeqAIJ*)A->data,*b;
3558599ef60dSHong Zhang   IS                 isicol;
3559599ef60dSHong Zhang   PetscErrorCode     ierr;
35601d098868SHong Zhang   const PetscInt     *r,*ic;
35611d098868SHong Zhang   PetscInt           i,n=A->rmap->n,*ai=a->i,*aj=a->j,*ajtmp,*adiag;
3562dc3a2fd3SHong Zhang   PetscInt           *bi,*bj,*bdiag,*bdiag_rev;
3563f61a948aSLisandro Dalcin   PetscInt           row,nzi,nzi_bl,nzi_bu,*im,nzi_al,nzi_au;
35641d098868SHong Zhang   PetscInt           nlnk,*lnk;
35651d098868SHong Zhang   PetscBT            lnkbt;
35661d098868SHong Zhang   PetscTruth         row_identity,icol_identity,both_identity;
35678aee2decSHong Zhang   MatScalar          *aatmp,*pv,*batmp,*ba,*rtmp,*pc,multiplier,*vtmp,diag_tmp;
35681d098868SHong Zhang   const PetscInt     *ics;
35691d098868SHong Zhang   PetscInt           j,nz,*pj,*bjtmp,k,ncut,*jtmp;
3570f61a948aSLisandro Dalcin   PetscReal          dt=info->dt,dtcol=info->dtcol,shift=info->shiftinblocks;
3571f61a948aSLisandro Dalcin   PetscInt           dtcount=(PetscInt)info->dtcount,nnz_max;
3572599ef60dSHong Zhang   PetscTruth         missing;
3573599ef60dSHong Zhang 
3574599ef60dSHong Zhang   PetscFunctionBegin;
3575f61a948aSLisandro Dalcin 
3576f61a948aSLisandro Dalcin   if (dt      == PETSC_DEFAULT) dt      = 0.005;
3577f61a948aSLisandro Dalcin   if (dtcol   == PETSC_DEFAULT) dtcol   = 0.01; /* XXX unused! */
3578f61a948aSLisandro Dalcin   if (dtcount == PETSC_DEFAULT) dtcount = (PetscInt)(1.5*a->rmax);
3579f61a948aSLisandro Dalcin 
35801d098868SHong Zhang   /* ------- symbolic factorization, can be reused ---------*/
35811d098868SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr);
35821d098868SHong Zhang   if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i);
35831d098868SHong Zhang   adiag=a->diag;
3584599ef60dSHong Zhang 
3585599ef60dSHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
3586599ef60dSHong Zhang 
3587599ef60dSHong Zhang   /* bdiag is location of diagonal in factor */
3588a6226a1aSHong Zhang   ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
3589a6226a1aSHong Zhang   bdiag_rev = bdiag + n+1;
3590599ef60dSHong Zhang 
35911d098868SHong Zhang   /* allocate row pointers bi */
35928fc3a347SHong Zhang   ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
35931d098868SHong Zhang 
35941d098868SHong Zhang   /* allocate bj and ba; max num of nonzero entries is (ai[n]+2*n*dtcount+2) */
3595393d3a68SHong Zhang   if (dtcount > n-1) dtcount = n-1; /* diagonal is excluded */
35961d098868SHong Zhang   nnz_max  = ai[n]+2*n*dtcount+2;
35978fc3a347SHong Zhang 
35988fc3a347SHong Zhang   ierr = PetscMalloc((nnz_max+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
35998fc3a347SHong Zhang   ierr = PetscMalloc((nnz_max+1)*sizeof(MatScalar),&ba);CHKERRQ(ierr);
3600599ef60dSHong Zhang 
3601599ef60dSHong Zhang   /* put together the new matrix */
3602599ef60dSHong Zhang   ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
3603599ef60dSHong Zhang   ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr);
3604599ef60dSHong Zhang   b    = (Mat_SeqAIJ*)(B)->data;
3605599ef60dSHong Zhang   b->free_a       = PETSC_TRUE;
3606599ef60dSHong Zhang   b->free_ij      = PETSC_TRUE;
3607599ef60dSHong Zhang   b->singlemalloc = PETSC_FALSE;
3608599ef60dSHong Zhang   b->a          = ba;
3609599ef60dSHong Zhang   b->j          = bj;
3610599ef60dSHong Zhang   b->i          = bi;
3611599ef60dSHong Zhang   b->diag       = bdiag;
3612599ef60dSHong Zhang   b->ilen       = 0;
3613599ef60dSHong Zhang   b->imax       = 0;
3614599ef60dSHong Zhang   b->row        = isrow;
3615599ef60dSHong Zhang   b->col        = iscol;
3616599ef60dSHong Zhang   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
3617599ef60dSHong Zhang   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
3618599ef60dSHong Zhang   b->icol       = isicol;
3619599ef60dSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
3620599ef60dSHong Zhang 
36211d098868SHong Zhang   ierr = PetscLogObjectMemory(B,nnz_max*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
36221d098868SHong Zhang   b->maxnz = nnz_max;
3623599ef60dSHong Zhang 
3624599ef60dSHong Zhang   (B)->factor                = MAT_FACTOR_ILUDT;
3625599ef60dSHong Zhang   (B)->info.factor_mallocs   = 0;
36261d098868SHong Zhang   (B)->info.fill_ratio_given = ((PetscReal)nnz_max)/((PetscReal)ai[n]);
36271d098868SHong Zhang   CHKMEMQ;
36281d098868SHong Zhang   /* ------- end of symbolic factorization ---------*/
3629599ef60dSHong Zhang 
3630599ef60dSHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
3631599ef60dSHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
3632599ef60dSHong Zhang   ics  = ic;
3633599ef60dSHong Zhang 
3634599ef60dSHong Zhang   /* linked list for storing column indices of the active row */
3635599ef60dSHong Zhang   nlnk = n + 1;
3636599ef60dSHong Zhang   ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
36371d098868SHong Zhang 
36381d098868SHong Zhang   /* im: used by PetscLLAddSortedLU(); jtmp: working array for column indices of active row */
36391d098868SHong Zhang   ierr = PetscMalloc((2*n+1)*sizeof(PetscInt),&im);CHKERRQ(ierr);
36401d098868SHong Zhang   jtmp = im + n;
36411d098868SHong Zhang   /* rtmp, vtmp: working arrays for sparse and contiguous row entries of active row */
36428369b28dSHong Zhang   ierr = PetscMalloc((2*n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr);
36438aee2decSHong Zhang   ierr = PetscMemzero(rtmp,(n+1)*sizeof(PetscScalar));CHKERRQ(ierr);
36448369b28dSHong Zhang   vtmp = rtmp + n;
3645599ef60dSHong Zhang 
3646599ef60dSHong Zhang   bi[0]    = 0;
36478fc3a347SHong Zhang   bdiag[0] = nnz_max-1; /* location of diag[0] in factor B */
3648dc3a2fd3SHong Zhang   bdiag_rev[n] = bdiag[0];
36498fc3a347SHong Zhang   bi[2*n+1] = bdiag[0]+1; /* endof bj and ba array */
3650599ef60dSHong Zhang   for (i=0; i<n; i++) {
3651599ef60dSHong Zhang     /* copy initial fill into linked list */
3652599ef60dSHong Zhang     nzi = 0; /* nonzeros for active row i */
36538369b28dSHong Zhang     nzi = ai[r[i]+1] - ai[r[i]];
36548369b28dSHong Zhang     if (!nzi) SETERRQ2(PETSC_ERR_MAT_LU_ZRPVT,"Empty row in matrix: row in original ordering %D in permuted ordering %D",r[i],i);
36558369b28dSHong Zhang     nzi_al = adiag[r[i]] - ai[r[i]];
36568369b28dSHong Zhang     nzi_au = ai[r[i]+1] - adiag[r[i]] -1;
3657599ef60dSHong Zhang     ajtmp = aj + ai[r[i]];
36588369b28dSHong Zhang     ierr = PetscLLAddPerm(nzi,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
3659599ef60dSHong Zhang 
3660599ef60dSHong Zhang     /* load in initial (unfactored row) */
36611d098868SHong Zhang     aatmp = a->a + ai[r[i]];
36628369b28dSHong Zhang     for (j=0; j<nzi; j++) {
36631d098868SHong Zhang       rtmp[ics[*ajtmp++]] = *aatmp++;
3664599ef60dSHong Zhang     }
3665599ef60dSHong Zhang 
3666599ef60dSHong Zhang     /* add pivot rows into linked list */
3667599ef60dSHong Zhang     row = lnk[n];
3668599ef60dSHong Zhang     while (row < i ) {
36691d098868SHong Zhang       nzi_bl = bi[row+1] - bi[row] + 1;
36701d098868SHong Zhang       bjtmp = bj + bdiag[row+1]+1; /* points to 1st column next to the diagonal in U */
36711d098868SHong Zhang       ierr  = PetscLLAddSortedLU(bjtmp,row,nlnk,lnk,lnkbt,i,nzi_bl,im);CHKERRQ(ierr);
3672599ef60dSHong Zhang       nzi  += nlnk;
3673599ef60dSHong Zhang       row   = lnk[row];
3674599ef60dSHong Zhang     }
3675599ef60dSHong Zhang 
36768369b28dSHong Zhang     /* copy data from lnk into jtmp, then initialize lnk */
36778369b28dSHong Zhang     ierr = PetscLLClean(n,n,nzi,lnk,jtmp,lnkbt);CHKERRQ(ierr);
3678599ef60dSHong Zhang 
3679599ef60dSHong Zhang     /* numerical factorization */
36808369b28dSHong Zhang     bjtmp = jtmp;
3681599ef60dSHong Zhang     row   = *bjtmp++; /* 1st pivot row */
3682599ef60dSHong Zhang     while  ( row < i ) {
3683599ef60dSHong Zhang       pc         = rtmp + row;
36843c2a7987SHong Zhang       pv         = ba + bdiag[row]; /* 1./(diag of the pivot row) */
36853c2a7987SHong Zhang       multiplier = (*pc) * (*pv);
3686599ef60dSHong Zhang       *pc        = multiplier;
36873c2a7987SHong Zhang       if (PetscAbsScalar(*pc) > dt){ /* apply tolerance dropping rule */
36881d098868SHong Zhang         pj         = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */
36891d098868SHong Zhang         pv         = ba + bdiag[row+1] + 1;
36901d098868SHong Zhang         /* if (multiplier < -1.0 or multiplier >1.0) printf("row/prow %d, %d, multiplier %g\n",i,row,multiplier); */
36911d098868SHong Zhang         nz         = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */
36921d098868SHong Zhang         for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++);
3693dc0b31edSSatish Balay         ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr);
3694599ef60dSHong Zhang       }
3695599ef60dSHong Zhang       row = *bjtmp++;
3696599ef60dSHong Zhang     }
3697599ef60dSHong Zhang 
36988369b28dSHong Zhang     /* copy sparse rtmp into contiguous vtmp; separate L and U part */
36998aee2decSHong Zhang     diag_tmp = rtmp[i];  /* save diagonal value - may not needed?? */
37008369b28dSHong Zhang     nzi_bl = 0; j = 0;
37018369b28dSHong Zhang     while (jtmp[j] < i){ /* Note: jtmp is sorted */
37028aee2decSHong Zhang       vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0;
37038369b28dSHong Zhang       nzi_bl++; j++;
37048369b28dSHong Zhang     }
37058369b28dSHong Zhang     nzi_bu = nzi - nzi_bl -1;
37068369b28dSHong Zhang     while (j < nzi){
37078aee2decSHong Zhang       vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0;
37088369b28dSHong Zhang       j++;
37098369b28dSHong Zhang     }
37108369b28dSHong Zhang 
3711599ef60dSHong Zhang     bjtmp = bj + bi[i];
3712599ef60dSHong Zhang     batmp = ba + bi[i];
37138369b28dSHong Zhang     /* apply level dropping rule to L part */
37148369b28dSHong Zhang     ncut = nzi_al + dtcount;
37158369b28dSHong Zhang     if (ncut < nzi_bl){
37168369b28dSHong Zhang       ierr = PetscSortSplit(ncut,nzi_bl,vtmp,jtmp);CHKERRQ(ierr);
37178369b28dSHong Zhang       ierr = PetscSortIntWithScalarArray(ncut,jtmp,vtmp);CHKERRQ(ierr);
3718599ef60dSHong Zhang     } else {
37198369b28dSHong Zhang       ncut = nzi_bl;
37208369b28dSHong Zhang     }
37218369b28dSHong Zhang     for (j=0; j<ncut; j++){
37228369b28dSHong Zhang       bjtmp[j] = jtmp[j];
37238369b28dSHong Zhang       batmp[j] = vtmp[j];
37246da515a1SHong Zhang       /* printf(" (%d,%g),",bjtmp[j],batmp[j]); */
37258369b28dSHong Zhang     }
37261d098868SHong Zhang     bi[i+1] = bi[i] + ncut;
37278369b28dSHong Zhang     nzi = ncut + 1;
37288369b28dSHong Zhang 
37298369b28dSHong Zhang     /* apply level dropping rule to U part */
37308369b28dSHong Zhang     ncut = nzi_au + dtcount;
37318369b28dSHong Zhang     if (ncut < nzi_bu){
37328369b28dSHong Zhang       ierr = PetscSortSplit(ncut,nzi_bu,vtmp+nzi_bl+1,jtmp+nzi_bl+1);CHKERRQ(ierr);
37338369b28dSHong Zhang       ierr = PetscSortIntWithScalarArray(ncut,jtmp+nzi_bl+1,vtmp+nzi_bl+1);CHKERRQ(ierr);
37348369b28dSHong Zhang     } else {
37358369b28dSHong Zhang       ncut = nzi_bu;
37368369b28dSHong Zhang     }
37378369b28dSHong Zhang     nzi += ncut;
37381d098868SHong Zhang 
37391d098868SHong Zhang     /* mark bdiagonal */
37401d098868SHong Zhang     bdiag[i+1]       = bdiag[i] - (ncut + 1);
3741dc3a2fd3SHong Zhang     bdiag_rev[n-i-1] = bdiag[i+1];
37428fc3a347SHong Zhang     bi[2*n - i]      = bi[2*n - i +1] - (ncut + 1);
37431d098868SHong Zhang     bjtmp = bj + bdiag[i];
37441d098868SHong Zhang     batmp = ba + bdiag[i];
37451d098868SHong Zhang     *bjtmp = i;
37468aee2decSHong Zhang     *batmp = diag_tmp; /* rtmp[i]; */
3747c9c72f8fSHong Zhang     if (*batmp == 0.0) {
3748c9c72f8fSHong Zhang       *batmp = dt+shift;
3749bd1bc851SBarry Smith       /* printf(" row %d add shift %g\n",i,shift); */
3750c9c72f8fSHong Zhang     }
3751b78a477dSHong Zhang     *batmp = 1.0/(*batmp); /* invert diagonal entries for simplier triangular solves */
37526da515a1SHong Zhang     /* printf(" (%d,%g),",*bjtmp,*batmp); */
37531d098868SHong Zhang 
37541d098868SHong Zhang     bjtmp = bj + bdiag[i+1]+1;
37551d098868SHong Zhang     batmp = ba + bdiag[i+1]+1;
37568369b28dSHong Zhang     for (k=0; k<ncut; k++){
37571d098868SHong Zhang       bjtmp[k] = jtmp[nzi_bl+1+k];
37581d098868SHong Zhang       batmp[k] = vtmp[nzi_bl+1+k];
37596da515a1SHong Zhang       /* printf(" (%d,%g),",bjtmp[k],batmp[k]); */
3760599ef60dSHong Zhang     }
37616da515a1SHong Zhang     /* printf("\n"); */
3762599ef60dSHong Zhang 
3763599ef60dSHong Zhang     im[i]   = nzi; /* used by PetscLLAddSortedLU() */
3764599ef60dSHong Zhang     /*
37651d098868SHong Zhang     printf("row %d: bi %d, bdiag %d\n",i,bi[i],bdiag[i]);
37668369b28dSHong Zhang     printf(" ----------------------------\n");
3767599ef60dSHong Zhang     */
37688369b28dSHong Zhang   } /* for (i=0; i<n; i++) */
37691d098868SHong Zhang   /* printf("end of L %d, beginning of U %d\n",bi[n],bdiag[n]); */
37701d098868SHong Zhang   if (bi[n] >= bdiag[n]) SETERRQ2(PETSC_ERR_ARG_SIZ,"end of L array %d cannot >= the beginning of U array %d",bi[n],bdiag[n]);
3771599ef60dSHong Zhang 
3772599ef60dSHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
3773599ef60dSHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
3774599ef60dSHong Zhang 
3775599ef60dSHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
3776599ef60dSHong Zhang   ierr = PetscFree(im);CHKERRQ(ierr);
3777599ef60dSHong Zhang   ierr = PetscFree(rtmp);CHKERRQ(ierr);
3778599ef60dSHong Zhang 
3779599ef60dSHong Zhang   ierr = PetscLogFlops(B->cmap->n);CHKERRQ(ierr);
37801d098868SHong Zhang   b->maxnz = b->nz = bi[n] + bdiag[0] - bdiag[n];
3781599ef60dSHong Zhang 
3782599ef60dSHong Zhang   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
3783599ef60dSHong Zhang   ierr = ISIdentity(isicol,&icol_identity);CHKERRQ(ierr);
3784599ef60dSHong Zhang   both_identity = (PetscTruth) (row_identity && icol_identity);
3785599ef60dSHong Zhang   if (row_identity && icol_identity) {
3786894a2336SHong Zhang     B->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_newdatastruct;
3787599ef60dSHong Zhang   } else {
3788894a2336SHong Zhang     B->ops->solve = MatSolve_SeqAIJ_newdatastruct;
3789599ef60dSHong Zhang   }
3790599ef60dSHong Zhang 
3791b78a477dSHong Zhang   B->ops->lufactorsymbolic  = MatILUDTFactorSymbolic_SeqAIJ;
3792b78a477dSHong Zhang   B->ops->lufactornumeric   = MatILUDTFactorNumeric_SeqAIJ;
37931d098868SHong Zhang   B->ops->solveadd          = 0;
37941d098868SHong Zhang   B->ops->solvetranspose    = 0;
37951d098868SHong Zhang   B->ops->solvetransposeadd = 0;
37961d098868SHong Zhang   B->ops->matsolve          = 0;
3797599ef60dSHong Zhang   B->assembled              = PETSC_TRUE;
3798599ef60dSHong Zhang   B->preallocated           = PETSC_TRUE;
3799599ef60dSHong Zhang   PetscFunctionReturn(0);
3800599ef60dSHong Zhang }
3801599ef60dSHong Zhang 
38023c2a7987SHong Zhang /* a wraper of MatILUDTFactor_SeqAIJ() */
38033c2a7987SHong Zhang #undef __FUNCT__
38043c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorSymbolic_SeqAIJ"
38053c2a7987SHong Zhang PetscErrorCode PETSCMAT_DLLEXPORT MatILUDTFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS row,IS col,const MatFactorInfo *info)
38063c2a7987SHong Zhang {
38073c2a7987SHong Zhang   PetscErrorCode     ierr;
38083c2a7987SHong Zhang 
38093c2a7987SHong Zhang   PetscFunctionBegin;
38103c2a7987SHong Zhang   ierr = MatILUDTFactor_SeqAIJ(A,row,col,info,&fact);CHKERRQ(ierr);
38113c2a7987SHong Zhang 
38123c2a7987SHong Zhang   fact->ops->lufactornumeric = MatILUDTFactorNumeric_SeqAIJ;
38133c2a7987SHong Zhang   PetscFunctionReturn(0);
38143c2a7987SHong Zhang }
38153c2a7987SHong Zhang 
38163c2a7987SHong Zhang /*
38173c2a7987SHong Zhang    same as MatLUFactorNumeric_SeqAIJ(), except using contiguous array matrix factors
38183c2a7987SHong Zhang    - intend to replace existing MatLUFactorNumeric_SeqAIJ()
38193c2a7987SHong Zhang */
38203c2a7987SHong Zhang #undef __FUNCT__
38213c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorNumeric_SeqAIJ"
38223c2a7987SHong Zhang PetscErrorCode PETSCMAT_DLLEXPORT MatILUDTFactorNumeric_SeqAIJ(Mat fact,Mat A,const MatFactorInfo *info)
38233c2a7987SHong Zhang {
38243c2a7987SHong Zhang   Mat            C=fact;
38253c2a7987SHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data;
38263c2a7987SHong Zhang   IS             isrow = b->row,isicol = b->icol;
38273c2a7987SHong Zhang   PetscErrorCode ierr;
38283c2a7987SHong Zhang   const PetscInt *r,*ic,*ics;
3829b78a477dSHong Zhang   PetscInt       i,j,k,n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j;
3830b78a477dSHong Zhang   PetscInt       *ajtmp,*bjtmp,nz,nzl,nzu,row,*bdiag = b->diag,*pj;
38313c2a7987SHong Zhang   MatScalar      *rtmp,*pc,multiplier,*v,*pv,*aa=a->a;
3832b78a477dSHong Zhang   PetscReal      dt=info->dt,shift=info->shiftinblocks;
3833b78a477dSHong Zhang   PetscTruth     row_identity, col_identity;
38343c2a7987SHong Zhang 
38353c2a7987SHong Zhang   PetscFunctionBegin;
38363c2a7987SHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
38373c2a7987SHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
3838b78a477dSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr);
38393c2a7987SHong Zhang   ics  = ic;
38403c2a7987SHong Zhang 
38413c2a7987SHong Zhang   for (i=0; i<n; i++){
3842b78a477dSHong Zhang     /* initialize rtmp array */
3843b78a477dSHong Zhang     nzl   = bi[i+1] - bi[i];       /* num of nozeros in L(i,:) */
38443c2a7987SHong Zhang     bjtmp = bj + bi[i];
3845b78a477dSHong Zhang     for  (j=0; j<nzl; j++) rtmp[*bjtmp++] = 0.0;
3846b78a477dSHong Zhang     rtmp[i] = 0.0;
3847b78a477dSHong Zhang     nzu   = bdiag[i] - bdiag[i+1]; /* num of nozeros in U(i,:) */
3848b78a477dSHong Zhang     bjtmp = bj + bdiag[i+1] + 1;
3849b78a477dSHong Zhang     for  (j=0; j<nzu; j++) rtmp[*bjtmp++] = 0.0;
38503c2a7987SHong Zhang 
3851b78a477dSHong Zhang     /* load in initial unfactored row of A */
38526da515a1SHong Zhang     /* printf("row %d\n",i); */
38533c2a7987SHong Zhang     nz    = ai[r[i]+1] - ai[r[i]];
38543c2a7987SHong Zhang     ajtmp = aj + ai[r[i]];
38553c2a7987SHong Zhang     v     = aa + ai[r[i]];
38563c2a7987SHong Zhang     for (j=0; j<nz; j++) {
3857b78a477dSHong Zhang       rtmp[ics[*ajtmp++]] = v[j];
38586da515a1SHong Zhang       /* printf(" (%d,%g),",ics[ajtmp[j]],rtmp[ics[ajtmp[j]]]); */
38593c2a7987SHong Zhang     }
38606da515a1SHong Zhang     /* printf("\n"); */
38613c2a7987SHong Zhang 
3862b78a477dSHong Zhang     /* numerical factorization */
3863b78a477dSHong Zhang     bjtmp = bj + bi[i]; /* point to 1st entry of L(i,:) */
3864b78a477dSHong Zhang     nzl   = bi[i+1] - bi[i]; /* num of entries in L(i,:) */
3865b78a477dSHong Zhang     k = 0;
3866b78a477dSHong Zhang     while (k < nzl){
38673c2a7987SHong Zhang       row   = *bjtmp++;
38686da515a1SHong Zhang       /* printf("  prow %d\n",row); */
38693c2a7987SHong Zhang       pc         = rtmp + row;
3870b78a477dSHong Zhang       pv         = b->a + bdiag[row]; /* 1./(diag of the pivot row) */
3871b78a477dSHong Zhang       multiplier = (*pc) * (*pv);
38723c2a7987SHong Zhang       *pc        = multiplier;
3873b78a477dSHong Zhang       if (PetscAbsScalar(multiplier) > dt){
3874b78a477dSHong Zhang         pj         = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */
3875b78a477dSHong Zhang         pv         = b->a + bdiag[row+1] + 1;
3876b78a477dSHong Zhang         nz         = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */
3877b78a477dSHong Zhang         for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++);
38786da515a1SHong Zhang         /* ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); */
38793c2a7987SHong Zhang       }
3880b78a477dSHong Zhang       k++;
38813c2a7987SHong Zhang     }
38823c2a7987SHong Zhang 
3883b78a477dSHong Zhang     /* finished row so stick it into b->a */
3884b78a477dSHong Zhang     /* L-part */
3885b78a477dSHong Zhang     pv = b->a + bi[i] ;
3886b78a477dSHong Zhang     pj = bj + bi[i] ;
3887b78a477dSHong Zhang     nzl = bi[i+1] - bi[i];
3888b78a477dSHong Zhang     for (j=0; j<nzl; j++) {
3889b78a477dSHong Zhang       pv[j] = rtmp[pj[j]];
38906da515a1SHong Zhang       /* printf(" (%d,%g),",pj[j],pv[j]); */
38913c2a7987SHong Zhang     }
3892b78a477dSHong Zhang 
3893b78a477dSHong Zhang     /* diagonal: invert diagonal entries for simplier triangular solves */
3894b78a477dSHong Zhang     if (rtmp[i] == 0.0) rtmp[i] = dt+shift;
3895b78a477dSHong Zhang     b->a[bdiag[i]] = 1.0/rtmp[i];
38966da515a1SHong Zhang     /* printf(" (%d,%g),",i,b->a[bdiag[i]]); */
3897b78a477dSHong Zhang 
3898b78a477dSHong Zhang     /* U-part */
3899b78a477dSHong Zhang     pv = b->a + bdiag[i+1] + 1;
3900b78a477dSHong Zhang     pj = bj + bdiag[i+1] + 1;
3901b78a477dSHong Zhang     nzu = bdiag[i] - bdiag[i+1] - 1;
3902b78a477dSHong Zhang     for (j=0; j<nzu; j++) {
3903b78a477dSHong Zhang       pv[j] = rtmp[pj[j]];
39046da515a1SHong Zhang       /* printf(" (%d,%g),",pj[j],pv[j]); */
3905b78a477dSHong Zhang     }
39066da515a1SHong Zhang     /* printf("\n"); */
3907b78a477dSHong Zhang   }
3908b78a477dSHong Zhang 
39093c2a7987SHong Zhang   ierr = PetscFree(rtmp);CHKERRQ(ierr);
39103c2a7987SHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
39113c2a7987SHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
3912b78a477dSHong Zhang 
39133c2a7987SHong Zhang   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
39143c2a7987SHong Zhang   ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr);
39153c2a7987SHong Zhang   if (row_identity && col_identity) {
3916894a2336SHong Zhang     C->ops->solve   = MatSolve_SeqAIJ_NaturalOrdering_newdatastruct;
39173c2a7987SHong Zhang   } else {
3918894a2336SHong Zhang     C->ops->solve   = MatSolve_SeqAIJ_newdatastruct;
39193c2a7987SHong Zhang   }
3920b78a477dSHong Zhang   C->ops->solveadd           = 0;
3921b78a477dSHong Zhang   C->ops->solvetranspose     = 0;
3922b78a477dSHong Zhang   C->ops->solvetransposeadd  = 0;
3923b78a477dSHong Zhang   C->ops->matsolve           = 0;
39243c2a7987SHong Zhang   C->assembled    = PETSC_TRUE;
39253c2a7987SHong Zhang   C->preallocated = PETSC_TRUE;
39263c2a7987SHong Zhang   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
39273c2a7987SHong Zhang   PetscFunctionReturn(0);
39283c2a7987SHong Zhang }
3929