xref: /petsc/src/mat/impls/aij/seq/aijfact.c (revision d4ad06f727c2ec2ca105edb394e1b331e11fb9dd)
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;
149289bc588SBarry Smith 
1503a40ed3dSBarry Smith   PetscFunctionBegin;
151ce3d78c0SShri Abhyankar   ierr = PetscOptionsGetTruth(PETSC_NULL,"-lu_new",&newdatastruct,PETSC_NULL);CHKERRQ(ierr);
152ce3d78c0SShri Abhyankar   if(newdatastruct){
153ce3d78c0SShri Abhyankar     ierr = MatLUFactorSymbolic_SeqAIJ_newdatastruct(B,A,isrow,iscol,info);CHKERRQ(ierr);
154ce3d78c0SShri Abhyankar     PetscFunctionReturn(0);
155ce3d78c0SShri Abhyankar   }
156ce3d78c0SShri Abhyankar 
157d0f46423SBarry Smith   if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_ERR_ARG_WRONG,"matrix must be square");
1584c49b128SBarry Smith   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
159ac121b3dSBarry Smith   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
160ac121b3dSBarry Smith   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
161289bc588SBarry Smith 
162289bc588SBarry Smith   /* get new row pointers */
1631393bc97SHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
1641393bc97SHong Zhang   bi[0] = 0;
1651393bc97SHong Zhang 
1661393bc97SHong Zhang   /* bdiag is location of diagonal in factor */
1671393bc97SHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
1681393bc97SHong Zhang   bdiag[0] = 0;
1691393bc97SHong Zhang 
1701393bc97SHong Zhang   /* linked list for storing column indices of the active row */
1711393bc97SHong Zhang   nlnk = n + 1;
1721393bc97SHong Zhang   ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
1731393bc97SHong Zhang 
17435baf27eSSatish Balay   ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr);
1751393bc97SHong Zhang 
1761393bc97SHong Zhang   /* initial FreeSpace size is f*(ai[n]+1) */
177d3d32019SBarry Smith   f = info->fill;
178a1a86e44SBarry Smith   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr);
1791393bc97SHong Zhang   current_space = free_space;
180289bc588SBarry Smith 
181289bc588SBarry Smith   for (i=0; i<n; i++) {
1821393bc97SHong Zhang     /* copy previous fill into linked list */
183289bc588SBarry Smith     nzi = 0;
1841393bc97SHong Zhang     nnz = ai[r[i]+1] - ai[r[i]];
1853b4a8b6dSBarry 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);
1861393bc97SHong Zhang     ajtmp = aj + ai[r[i]];
187afcc9904SHong Zhang     ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
1881393bc97SHong Zhang     nzi += nlnk;
1891393bc97SHong Zhang 
1901393bc97SHong Zhang     /* add pivot rows into linked list */
1911393bc97SHong Zhang     row = lnk[n];
1921393bc97SHong Zhang     while (row < i) {
193d1170560SHong Zhang       nzbd    = bdiag[row] - bi[row] + 1; /* num of entries in the row with column index <= row */
194d1170560SHong Zhang       ajtmp   = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */
195d1170560SHong Zhang       ierr = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr);
1961393bc97SHong Zhang       nzi += nlnk;
1971393bc97SHong Zhang       row  = lnk[row];
198289bc588SBarry Smith     }
1991393bc97SHong Zhang     bi[i+1] = bi[i] + nzi;
2001393bc97SHong Zhang     im[i]   = nzi;
2011393bc97SHong Zhang 
2021393bc97SHong Zhang     /* mark bdiag */
2031393bc97SHong Zhang     nzbd = 0;
2041393bc97SHong Zhang     nnz  = nzi;
2051393bc97SHong Zhang     k    = lnk[n];
2061393bc97SHong Zhang     while (nnz-- && k < i){
2071393bc97SHong Zhang       nzbd++;
2081393bc97SHong Zhang       k = lnk[k];
2091393bc97SHong Zhang     }
2101393bc97SHong Zhang     bdiag[i] = bi[i] + nzbd;
2111393bc97SHong Zhang 
2121393bc97SHong Zhang     /* if free space is not available, make more free space */
2131393bc97SHong Zhang     if (current_space->local_remaining<nzi) {
2141393bc97SHong Zhang       nnz = (n - i)*nzi; /* estimated and max additional space needed */
215a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
2161393bc97SHong Zhang       reallocs++;
2171393bc97SHong Zhang     }
2181393bc97SHong Zhang 
2191393bc97SHong Zhang     /* copy data into free space, then initialize lnk */
2201393bc97SHong Zhang     ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
2211393bc97SHong Zhang     bi_ptr[i] = current_space->array;
2221393bc97SHong Zhang     current_space->array           += nzi;
2231393bc97SHong Zhang     current_space->local_used      += nzi;
2241393bc97SHong Zhang     current_space->local_remaining -= nzi;
225289bc588SBarry Smith   }
2266cf91177SBarry Smith #if defined(PETSC_USE_INFO)
227464e8d44SSatish Balay   if (ai[n] != 0) {
2281393bc97SHong Zhang     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
229ae15b995SBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr);
230ae15b995SBarry Smith     ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
231ae15b995SBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr);
232ae15b995SBarry Smith     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
23305bf559cSSatish Balay   } else {
234ae15b995SBarry Smith     ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr);
23505bf559cSSatish Balay   }
23663ba0a88SBarry Smith #endif
2372fb3ed76SBarry Smith 
238898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
239898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
2401987afe7SBarry Smith 
2411393bc97SHong Zhang   /* destroy list of free space and other temporary array(s) */
2421393bc97SHong Zhang   ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
243a1a86e44SBarry Smith   ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr);
2441393bc97SHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
24535baf27eSSatish Balay   ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr);
246289bc588SBarry Smith 
247289bc588SBarry Smith   /* put together the new matrix */
248719d5645SBarry Smith   ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
249719d5645SBarry Smith   ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr);
250719d5645SBarry Smith   b    = (Mat_SeqAIJ*)(B)->data;
251e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
252e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
2537c922b88SBarry Smith   b->singlemalloc = PETSC_FALSE;
2541393bc97SHong Zhang   ierr          = PetscMalloc((bi[n]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);
2551393bc97SHong Zhang   b->j          = bj;
2561393bc97SHong Zhang   b->i          = bi;
2571393bc97SHong Zhang   b->diag       = bdiag;
258416022c9SBarry Smith   b->ilen       = 0;
259416022c9SBarry Smith   b->imax       = 0;
260416022c9SBarry Smith   b->row        = isrow;
261416022c9SBarry Smith   b->col        = iscol;
262c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
263c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
26482bf6240SBarry Smith   b->icol       = isicol;
26587828ca2SBarry Smith   ierr          = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
2661393bc97SHong Zhang 
2671393bc97SHong Zhang   /* In b structure:  Free imax, ilen, old a, old j.  Allocate solve_work, new a, new j */
268719d5645SBarry Smith   ierr = PetscLogObjectMemory(B,(bi[n]-n)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
2691393bc97SHong Zhang   b->maxnz = b->nz = bi[n] ;
2707fd98bd6SLois Curfman McInnes 
271719d5645SBarry Smith   (B)->factor                = MAT_FACTOR_LU;
272719d5645SBarry Smith   (B)->info.factor_mallocs   = reallocs;
273719d5645SBarry Smith   (B)->info.fill_ratio_given = f;
274703deb49SSatish Balay 
2759f7d0b68SHong Zhang   if (ai[n]) {
276719d5645SBarry Smith     (B)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]);
277eea2913aSSatish Balay   } else {
278719d5645SBarry Smith     (B)->info.fill_ratio_needed = 0.0;
279eea2913aSSatish Balay   }
280719d5645SBarry Smith   (B)->ops->lufactornumeric  = MatLUFactorNumeric_SeqAIJ;
281719d5645SBarry Smith   (B)->ops->solve            = MatSolve_SeqAIJ;
282719d5645SBarry Smith   (B)->ops->solvetranspose   = MatSolveTranspose_SeqAIJ;
283db4efbfdSBarry Smith   /* switch to inodes if appropriate */
284719d5645SBarry Smith   ierr = MatLUFactorSymbolic_Inode(B,A,isrow,iscol,info);CHKERRQ(ierr);
2853a40ed3dSBarry Smith   PetscFunctionReturn(0);
286289bc588SBarry Smith }
2871393bc97SHong Zhang 
288ce3d78c0SShri Abhyankar #undef __FUNCT__
289ce3d78c0SShri Abhyankar #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ_newdatastruct"
290ce3d78c0SShri Abhyankar PetscErrorCode MatLUFactorSymbolic_SeqAIJ_newdatastruct(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
291ce3d78c0SShri Abhyankar {
292ce3d78c0SShri Abhyankar   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
293ce3d78c0SShri Abhyankar   IS                 isicol;
294ce3d78c0SShri Abhyankar   PetscErrorCode     ierr;
295ce3d78c0SShri Abhyankar   const PetscInt     *r,*ic;
296ce3d78c0SShri Abhyankar   PetscInt           i,n=A->rmap->n,*ai=a->i,*aj=a->j;
297ce3d78c0SShri Abhyankar   PetscInt           *bi,*bj,*ajtmp;
298ce3d78c0SShri Abhyankar   PetscInt           *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im;
299ce3d78c0SShri Abhyankar   PetscReal          f;
300ce3d78c0SShri Abhyankar   PetscInt           nlnk,*lnk,k,**bi_ptr;
301ce3d78c0SShri Abhyankar   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
302ce3d78c0SShri Abhyankar   PetscBT            lnkbt;
303ce3d78c0SShri Abhyankar 
304ce3d78c0SShri Abhyankar   PetscFunctionBegin;
305ce3d78c0SShri Abhyankar   if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_ERR_ARG_WRONG,"matrix must be square");
306ce3d78c0SShri Abhyankar   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
307ce3d78c0SShri Abhyankar   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
308ce3d78c0SShri Abhyankar   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
309ce3d78c0SShri Abhyankar 
310ce3d78c0SShri Abhyankar   /* get new row pointers */
311ce3d78c0SShri Abhyankar   ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
312ce3d78c0SShri Abhyankar   bi[0] = 0;
313ce3d78c0SShri Abhyankar 
314ce3d78c0SShri Abhyankar   /* bdiag is location of diagonal in factor */
315ce3d78c0SShri Abhyankar   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
316ce3d78c0SShri Abhyankar   bdiag[0] = 0;
317ce3d78c0SShri Abhyankar 
318ce3d78c0SShri Abhyankar   /* linked list for storing column indices of the active row */
319ce3d78c0SShri Abhyankar   nlnk = n + 1;
320ce3d78c0SShri Abhyankar   ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
321ce3d78c0SShri Abhyankar 
322ce3d78c0SShri Abhyankar   ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr);
323ce3d78c0SShri Abhyankar 
324ce3d78c0SShri Abhyankar   /* initial FreeSpace size is f*(ai[n]+1) */
325ce3d78c0SShri Abhyankar   f = info->fill;
326ce3d78c0SShri Abhyankar   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr);
327ce3d78c0SShri Abhyankar   current_space = free_space;
328ce3d78c0SShri Abhyankar 
329ce3d78c0SShri Abhyankar   for (i=0; i<n; i++) {
330ce3d78c0SShri Abhyankar     /* copy previous fill into linked list */
331ce3d78c0SShri Abhyankar     nzi = 0;
332ce3d78c0SShri Abhyankar     nnz = ai[r[i]+1] - ai[r[i]];
333ce3d78c0SShri 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);
334ce3d78c0SShri Abhyankar     ajtmp = aj + ai[r[i]];
335ce3d78c0SShri Abhyankar     ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
336ce3d78c0SShri Abhyankar     nzi += nlnk;
337ce3d78c0SShri Abhyankar 
338ce3d78c0SShri Abhyankar     /* add pivot rows into linked list */
339ce3d78c0SShri Abhyankar     row = lnk[n];
340ce3d78c0SShri Abhyankar     while (row < i){
34170319c84SHong Zhang       nzbd  = bdiag[row] + 1; /* num of entries in the row with column index <= row */
342ce3d78c0SShri Abhyankar       ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */
343ce3d78c0SShri Abhyankar       ierr  = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr);
344ce3d78c0SShri Abhyankar       nzi  += nlnk;
345ce3d78c0SShri Abhyankar       row   = lnk[row];
346ce3d78c0SShri Abhyankar     }
347ce3d78c0SShri Abhyankar     bi[i+1] = bi[i] + nzi;
348ce3d78c0SShri Abhyankar     im[i]   = nzi;
349ce3d78c0SShri Abhyankar 
350ce3d78c0SShri Abhyankar     /* mark bdiag */
351ce3d78c0SShri Abhyankar     nzbd = 0;
352ce3d78c0SShri Abhyankar     nnz  = nzi;
353ce3d78c0SShri Abhyankar     k    = lnk[n];
354ce3d78c0SShri Abhyankar     while (nnz-- && k < i){
355ce3d78c0SShri Abhyankar       nzbd++;
356ce3d78c0SShri Abhyankar       k = lnk[k];
357ce3d78c0SShri Abhyankar     }
358783ef271SHong Zhang     bdiag[i] = nzbd; /* note: bdiag[i] = nnzL as input for PetscFreeSpaceContiguous_LU() */
359ce3d78c0SShri Abhyankar 
360ce3d78c0SShri Abhyankar     /* if free space is not available, make more free space */
361ce3d78c0SShri Abhyankar     if (current_space->local_remaining<nzi) {
362ce3d78c0SShri Abhyankar       nnz = 2*(n - i)*nzi; /* estimated and max additional space needed */
363ce3d78c0SShri Abhyankar       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
364ce3d78c0SShri Abhyankar       reallocs++;
365ce3d78c0SShri Abhyankar     }
366ce3d78c0SShri Abhyankar 
367ce3d78c0SShri Abhyankar     /* copy data into free space, then initialize lnk */
368ce3d78c0SShri Abhyankar     ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
369ce3d78c0SShri Abhyankar     bi_ptr[i] = current_space->array;
370ce3d78c0SShri Abhyankar     current_space->array           += nzi;
371ce3d78c0SShri Abhyankar     current_space->local_used      += nzi;
372ce3d78c0SShri Abhyankar     current_space->local_remaining -= nzi;
373ce3d78c0SShri Abhyankar   }
374ce3d78c0SShri Abhyankar #if defined(PETSC_USE_INFO)
375ce3d78c0SShri Abhyankar   if (ai[n] != 0) {
376ce3d78c0SShri Abhyankar     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
377ce3d78c0SShri Abhyankar     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr);
378ce3d78c0SShri Abhyankar     ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
379ce3d78c0SShri Abhyankar     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr);
380ce3d78c0SShri Abhyankar     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
381ce3d78c0SShri Abhyankar   } else {
382ce3d78c0SShri Abhyankar     ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr);
383ce3d78c0SShri Abhyankar   }
384ce3d78c0SShri Abhyankar #endif
385ce3d78c0SShri Abhyankar 
386ce3d78c0SShri Abhyankar   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
387ce3d78c0SShri Abhyankar   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
388ce3d78c0SShri Abhyankar 
389ce3d78c0SShri Abhyankar   /* destroy list of free space and other temporary array(s) */
390ce3d78c0SShri Abhyankar   ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
391783ef271SHong Zhang   ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr);
392ce3d78c0SShri Abhyankar   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
393ce3d78c0SShri Abhyankar   ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr);
394ce3d78c0SShri Abhyankar 
395ce3d78c0SShri Abhyankar   /* put together the new matrix */
396ce3d78c0SShri Abhyankar   ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
397ce3d78c0SShri Abhyankar   ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr);
398ce3d78c0SShri Abhyankar   b    = (Mat_SeqAIJ*)(B)->data;
399ce3d78c0SShri Abhyankar   b->free_a       = PETSC_TRUE;
400ce3d78c0SShri Abhyankar   b->free_ij      = PETSC_TRUE;
401ce3d78c0SShri Abhyankar   b->singlemalloc = PETSC_FALSE;
402ce3d78c0SShri Abhyankar   ierr          = PetscMalloc((bi[2*n+1])*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);
403ce3d78c0SShri Abhyankar   b->j          = bj;
404ce3d78c0SShri Abhyankar   b->i          = bi;
405ce3d78c0SShri Abhyankar   b->diag       = bdiag;
406ce3d78c0SShri Abhyankar   b->ilen       = 0;
407ce3d78c0SShri Abhyankar   b->imax       = 0;
408ce3d78c0SShri Abhyankar   b->row        = isrow;
409ce3d78c0SShri Abhyankar   b->col        = iscol;
410ce3d78c0SShri Abhyankar   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
411ce3d78c0SShri Abhyankar   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
412ce3d78c0SShri Abhyankar   b->icol       = isicol;
413ce3d78c0SShri Abhyankar   ierr          = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
414ce3d78c0SShri Abhyankar 
415ce3d78c0SShri Abhyankar   /* In b structure:  Free imax, ilen, old a, old j.  Allocate solve_work, new a, new j */
416ce3d78c0SShri Abhyankar   ierr = PetscLogObjectMemory(B,bi[2*n+1]*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
417ce3d78c0SShri Abhyankar   b->maxnz = b->nz = bi[2*n+1] ;
418ce3d78c0SShri Abhyankar 
419ce3d78c0SShri Abhyankar   (B)->factor                = MAT_FACTOR_LU;
420ce3d78c0SShri Abhyankar   (B)->info.factor_mallocs   = reallocs;
421ce3d78c0SShri Abhyankar   (B)->info.fill_ratio_given = f;
422ce3d78c0SShri Abhyankar 
423ce3d78c0SShri Abhyankar   if (ai[n]) {
424894a2336SHong Zhang     (B)->info.fill_ratio_needed = ((PetscReal)bi[2*n+1])/((PetscReal)ai[n]);
425ce3d78c0SShri Abhyankar   } else {
426ce3d78c0SShri Abhyankar     (B)->info.fill_ratio_needed = 0.0;
427ce3d78c0SShri Abhyankar   }
428ce3d78c0SShri Abhyankar   (B)->ops->lufactornumeric  = MatLUFactorNumeric_SeqAIJ_newdatastruct;
429ce3d78c0SShri Abhyankar   PetscFunctionReturn(0);
430ce3d78c0SShri Abhyankar }
431ce3d78c0SShri Abhyankar 
4325b5bc046SBarry Smith /*
4335b5bc046SBarry Smith     Trouble in factorization, should we dump the original matrix?
4345b5bc046SBarry Smith */
4355b5bc046SBarry Smith #undef __FUNCT__
4365b5bc046SBarry Smith #define __FUNCT__ "MatFactorDumpMatrix"
4375b5bc046SBarry Smith PetscErrorCode MatFactorDumpMatrix(Mat A)
4385b5bc046SBarry Smith {
4395b5bc046SBarry Smith   PetscErrorCode ierr;
44090d69ab7SBarry Smith   PetscTruth     flg = PETSC_FALSE;
4415b5bc046SBarry Smith 
4425b5bc046SBarry Smith   PetscFunctionBegin;
44390d69ab7SBarry Smith   ierr = PetscOptionsGetTruth(PETSC_NULL,"-mat_factor_dump_on_error",&flg,PETSC_NULL);CHKERRQ(ierr);
4445b5bc046SBarry Smith   if (flg) {
4455b5bc046SBarry Smith     PetscViewer viewer;
4465b5bc046SBarry Smith     char        filename[PETSC_MAX_PATH_LEN];
4475b5bc046SBarry Smith 
4485b5bc046SBarry Smith     ierr = PetscSNPrintf(filename,PETSC_MAX_PATH_LEN,"matrix_factor_error.%d",PetscGlobalRank);CHKERRQ(ierr);
4497adad957SLisandro Dalcin     ierr = PetscViewerBinaryOpen(((PetscObject)A)->comm,filename,FILE_MODE_WRITE,&viewer);CHKERRQ(ierr);
4505b5bc046SBarry Smith     ierr = MatView(A,viewer);CHKERRQ(ierr);
4515b5bc046SBarry Smith     ierr = PetscViewerDestroy(viewer);CHKERRQ(ierr);
4525b5bc046SBarry Smith   }
4535b5bc046SBarry Smith   PetscFunctionReturn(0);
4545b5bc046SBarry Smith }
4555b5bc046SBarry Smith 
456db4efbfdSBarry Smith extern PetscErrorCode MatSolve_Inode(Mat,Vec,Vec);
457db4efbfdSBarry Smith 
4580a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
459894a2336SHong Zhang extern PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering_newdatastruct(Mat,Vec,Vec);
460894a2336SHong Zhang extern PetscErrorCode MatSolve_SeqAIJ_newdatastruct(Mat,Vec,Vec);
461c9c72f8fSHong Zhang 
462c9c72f8fSHong Zhang #undef __FUNCT__
463c9c72f8fSHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_newdatastruct"
464c9c72f8fSHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ_newdatastruct(Mat B,Mat A,const MatFactorInfo *info)
465c9c72f8fSHong Zhang {
466c9c72f8fSHong Zhang   Mat            C=B;
467c9c72f8fSHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data;
468c9c72f8fSHong Zhang   IS             isrow = b->row,isicol = b->icol;
469c9c72f8fSHong Zhang   PetscErrorCode ierr;
470c9c72f8fSHong Zhang   const PetscInt *r,*ic,*ics;
471c9c72f8fSHong Zhang   PetscInt       i,j,k,n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j;
472c9c72f8fSHong Zhang   PetscInt       *ajtmp,*bjtmp,nz,nzL,row,*bdiag=b->diag,*pj;
473c9c72f8fSHong Zhang   MatScalar      *rtmp,*pc,multiplier,*v,*pv,*aa=a->a;
474c9c72f8fSHong Zhang   PetscTruth     row_identity,col_identity;
475c9c72f8fSHong Zhang 
476*d4ad06f7SHong Zhang   LUShift_Ctx    sctx;
477*d4ad06f7SHong Zhang   PetscInt       *ddiag,newshift;
478*d4ad06f7SHong Zhang   PetscReal      rs;
479*d4ad06f7SHong Zhang   MatScalar      d;
480*d4ad06f7SHong Zhang 
481c9c72f8fSHong Zhang   PetscFunctionBegin;
482*d4ad06f7SHong Zhang   /* ZeropivotSetUp(): initialize shift context sctx */
483*d4ad06f7SHong Zhang   sctx.nshift         = 0;
484*d4ad06f7SHong Zhang   sctx.nshift_max     = 0;
485*d4ad06f7SHong Zhang   sctx.shift_top      = 0.0;
486*d4ad06f7SHong Zhang   sctx.shift_lo       = 0.0;
487*d4ad06f7SHong Zhang   sctx.shift_hi       = 0.0;
488*d4ad06f7SHong Zhang   sctx.shift_fraction = 0.0;
489*d4ad06f7SHong Zhang   sctx.shift_amount   = 0.0;
490*d4ad06f7SHong Zhang 
491*d4ad06f7SHong Zhang   /* if both shift schemes are chosen by user, only use info->shiftpd */
492*d4ad06f7SHong Zhang   if (info->shiftpd) { /* set sctx.shift_top=max{rs} */
493*d4ad06f7SHong Zhang     ddiag          = a->diag;
494*d4ad06f7SHong Zhang     sctx.shift_top = info->zeropivot;
495*d4ad06f7SHong Zhang     for (i=0; i<n; i++) {
496*d4ad06f7SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
497*d4ad06f7SHong Zhang       d  = (aa)[ddiag[i]];
498*d4ad06f7SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
499*d4ad06f7SHong Zhang       v  = aa+ai[i];
500*d4ad06f7SHong Zhang       nz = ai[i+1] - ai[i];
501*d4ad06f7SHong Zhang       for (j=0; j<nz; j++)
502*d4ad06f7SHong Zhang 	rs += PetscAbsScalar(v[j]);
503*d4ad06f7SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
504*d4ad06f7SHong Zhang     }
505*d4ad06f7SHong Zhang     sctx.shift_top   *= 1.1;
506*d4ad06f7SHong Zhang     sctx.nshift_max   = 5;
507*d4ad06f7SHong Zhang     sctx.shift_lo     = 0.;
508*d4ad06f7SHong Zhang     sctx.shift_hi     = 1.;
509*d4ad06f7SHong Zhang   }
510*d4ad06f7SHong Zhang 
511c9c72f8fSHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
512c9c72f8fSHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
513c9c72f8fSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr);
514c9c72f8fSHong Zhang   ics  = ic;
515c9c72f8fSHong Zhang 
516*d4ad06f7SHong Zhang   do {
517*d4ad06f7SHong Zhang     sctx.lushift = PETSC_FALSE;
518c9c72f8fSHong Zhang     for (i=0; i<n; i++){
519c9c72f8fSHong Zhang       /* zero rtmp */
520c9c72f8fSHong Zhang       /* L part */
521c9c72f8fSHong Zhang       nz    = bi[i+1] - bi[i];
522c9c72f8fSHong Zhang       bjtmp = bj + bi[i];
523c9c72f8fSHong Zhang       for  (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
524c9c72f8fSHong Zhang 
525c9c72f8fSHong Zhang       /* U part */
5268fc3a347SHong Zhang       nz = bi[2*n-i+1] - bi[2*n-i];
5278fc3a347SHong Zhang       bjtmp = bj + bi[2*n-i];
528c9c72f8fSHong Zhang       for  (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
529c9c72f8fSHong Zhang 
530c9c72f8fSHong Zhang       /* load in initial (unfactored row) */
531c9c72f8fSHong Zhang       nz    = ai[r[i]+1] - ai[r[i]];
532c9c72f8fSHong Zhang       ajtmp = aj + ai[r[i]];
533c9c72f8fSHong Zhang       v     = aa + ai[r[i]];
534c9c72f8fSHong Zhang       for (j=0; j<nz; j++) {
535c9c72f8fSHong Zhang         rtmp[ics[ajtmp[j]]] = v[j];
536c9c72f8fSHong Zhang       }
537*d4ad06f7SHong Zhang       /* ZeropivotApply() */
538*d4ad06f7SHong Zhang       rtmp[ics[r[i]]] += sctx.shift_amount;  /* shift the diagonal of the matrix */
539c9c72f8fSHong Zhang 
540c9c72f8fSHong Zhang       /* elimination */
541c9c72f8fSHong Zhang       bjtmp = bj + bi[i];
542c9c72f8fSHong Zhang       row   = *bjtmp++;
543c9c72f8fSHong Zhang       nzL   = bi[i+1] - bi[i];
544c9c72f8fSHong Zhang       k   = 0;
545c9c72f8fSHong Zhang       while  (k < nzL) {
546c9c72f8fSHong Zhang         pc = rtmp + row;
547c9c72f8fSHong Zhang         if (*pc != 0.0) {
548c9c72f8fSHong Zhang           pv         = b->a + bdiag[row];
549c9c72f8fSHong Zhang           multiplier = *pc * (*pv);
550c9c72f8fSHong Zhang           *pc        = multiplier;
5518fc3a347SHong Zhang           pj         = b->j + bi[2*n-row]; /* begining of U(row,:) */
5528fc3a347SHong Zhang           pv         = b->a + bi[2*n-row];
5538fc3a347SHong Zhang           nz         = bi[2*n-row+1] - bi[2*n-row] - 1; /* num of entries in U(row,:), excluding diag */
554c9c72f8fSHong Zhang           for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j];
555c9c72f8fSHong Zhang           ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr);
556c9c72f8fSHong Zhang         }
557c9c72f8fSHong Zhang         row = *bjtmp++; k++;
558c9c72f8fSHong Zhang       }
559c9c72f8fSHong Zhang 
560c9c72f8fSHong Zhang       /* finished row so stick it into b->a */
561*d4ad06f7SHong Zhang       rs = 0.0;
562c9c72f8fSHong Zhang       /* L part */
563c9c72f8fSHong Zhang       pv   = b->a + bi[i] ;
564c9c72f8fSHong Zhang       pj   = b->j + bi[i] ;
565c9c72f8fSHong Zhang       nz   = bi[i+1] - bi[i];
566c9c72f8fSHong Zhang       for (j=0; j<nz; j++) {
567*d4ad06f7SHong Zhang         pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]);
568c9c72f8fSHong Zhang       }
569c9c72f8fSHong Zhang 
570c9c72f8fSHong Zhang       /* U part */
5718fc3a347SHong Zhang       pv = b->a + bi[2*n-i];
5728fc3a347SHong Zhang       pj = b->j + bi[2*n-i];
5738fc3a347SHong Zhang       nz = bi[2*n-i+1] - bi[2*n-i] - 1;
574*d4ad06f7SHong Zhang       for (j=0; j<nz; j++) {
575*d4ad06f7SHong Zhang         pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]);
576c9c72f8fSHong Zhang       }
577*d4ad06f7SHong Zhang 
578*d4ad06f7SHong Zhang       /* ZeropivotCheck() */
579*d4ad06f7SHong Zhang       sctx.rs  = rs;
580*d4ad06f7SHong Zhang       sctx.pv  = rtmp[i];
581*d4ad06f7SHong Zhang       ierr = MatLUCheckShift_inline(info,sctx,i,newshift);CHKERRQ(ierr);
582*d4ad06f7SHong Zhang       if (newshift == 1) break;
583*d4ad06f7SHong Zhang 
584*d4ad06f7SHong Zhang       /* Mark diagonal and invert diagonal for simplier triangular solves */
585*d4ad06f7SHong Zhang       pv  = b->a + bdiag[i];
586*d4ad06f7SHong Zhang       *pv = 1.0/rtmp[i];
587*d4ad06f7SHong Zhang 
588*d4ad06f7SHong Zhang     } /* endof for (i=0; i<n; i++){ */
589*d4ad06f7SHong Zhang 
590*d4ad06f7SHong Zhang     /* ZeropivotRefine() */
591*d4ad06f7SHong Zhang     if (info->shiftpd && !sctx.lushift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max){
592*d4ad06f7SHong Zhang       /*
593*d4ad06f7SHong Zhang        * if no shift in this attempt & shifting & started shifting & can refine,
594*d4ad06f7SHong Zhang        * then try lower shift
595*d4ad06f7SHong Zhang        */
596*d4ad06f7SHong Zhang       sctx.shift_hi       = sctx.shift_fraction;
597*d4ad06f7SHong Zhang       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
598*d4ad06f7SHong Zhang       sctx.shift_amount   = sctx.shift_fraction * sctx.shift_top;
599*d4ad06f7SHong Zhang       sctx.lushift        = PETSC_TRUE;
600*d4ad06f7SHong Zhang       sctx.nshift++;
601*d4ad06f7SHong Zhang     }
602*d4ad06f7SHong Zhang   } while (sctx.lushift);
603*d4ad06f7SHong Zhang 
604c9c72f8fSHong Zhang   ierr = PetscFree(rtmp);CHKERRQ(ierr);
605c9c72f8fSHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
606c9c72f8fSHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
607c9c72f8fSHong Zhang 
608c9c72f8fSHong Zhang   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
609c9c72f8fSHong Zhang   ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr);
610c9c72f8fSHong Zhang   if (row_identity && col_identity) {
611894a2336SHong Zhang     C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_newdatastruct;
612c9c72f8fSHong Zhang   } else {
613894a2336SHong Zhang     C->ops->solve = MatSolve_SeqAIJ_newdatastruct;
614c9c72f8fSHong Zhang   }
615c9c72f8fSHong Zhang 
616c9c72f8fSHong Zhang   C->ops->solveadd           = 0;
617c9c72f8fSHong Zhang   C->ops->solvetranspose     = 0;
618c9c72f8fSHong Zhang   C->ops->solvetransposeadd  = 0;
619c9c72f8fSHong Zhang   C->ops->matsolve           = 0;
620c9c72f8fSHong Zhang   C->assembled    = PETSC_TRUE;
621c9c72f8fSHong Zhang   C->preallocated = PETSC_TRUE;
622c9c72f8fSHong Zhang   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
623*d4ad06f7SHong Zhang 
624*d4ad06f7SHong Zhang   /* ZeropivotView() */
625*d4ad06f7SHong Zhang   if (sctx.nshift){
626*d4ad06f7SHong Zhang     if (info->shiftpd) {
627*d4ad06f7SHong 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);
628*d4ad06f7SHong Zhang     } else if (info->shiftnz) {
629*d4ad06f7SHong Zhang       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
630*d4ad06f7SHong Zhang     }
631*d4ad06f7SHong Zhang   }
632c9c72f8fSHong Zhang   PetscFunctionReturn(0);
633c9c72f8fSHong Zhang }
634c9c72f8fSHong Zhang 
6354a2ae208SSatish Balay #undef __FUNCT__
6364a2ae208SSatish Balay #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ"
6370481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info)
638289bc588SBarry Smith {
639719d5645SBarry Smith   Mat             C=B;
640416022c9SBarry Smith   Mat_SeqAIJ      *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data;
64182bf6240SBarry Smith   IS              isrow = b->row,isicol = b->icol;
6426849ba73SBarry Smith   PetscErrorCode  ierr;
6435d0c19d7SBarry Smith   const PetscInt   *r,*ic,*ics;
644d278edc7SBarry Smith   PetscInt        nz,row,i,j,n=A->rmap->n,diag;
645d278edc7SBarry Smith   const PetscInt  *ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j;
646d278edc7SBarry Smith   const PetscInt  *ajtmp,*bjtmp,*diag_offset = b->diag,*pj;
647d278edc7SBarry Smith   MatScalar       *pv,*rtmp,*pc,multiplier,d;
648d278edc7SBarry Smith   const MatScalar *v,*aa=a->a;
64933f9893dSHong Zhang   PetscReal       rs=0.0;
650b3bf805bSHong Zhang   LUShift_Ctx     sctx;
65142898933SHong Zhang   PetscInt        newshift,*ddiag;
652289bc588SBarry Smith 
6533a40ed3dSBarry Smith   PetscFunctionBegin;
65478b31e54SBarry Smith   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
65578b31e54SBarry Smith   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
656b78a477dSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr);
6579f7d0b68SHong Zhang   ics  = ic;
658289bc588SBarry Smith 
659*d4ad06f7SHong Zhang   /* initialize shift context sctx */
660*d4ad06f7SHong Zhang   sctx.nshift         = 0;
661ada7143aSSatish Balay   sctx.nshift_max     = 0;
662*d4ad06f7SHong Zhang   sctx.shift_top      = 0.0;
663*d4ad06f7SHong Zhang   sctx.shift_lo       = 0.0;
664*d4ad06f7SHong Zhang   sctx.shift_hi       = 0.0;
665*d4ad06f7SHong Zhang   sctx.shift_fraction = 0.0;
666*d4ad06f7SHong Zhang   sctx.shift_amount   = 0.0;
667ada7143aSSatish Balay 
668118f5789SBarry Smith   /* if both shift schemes are chosen by user, only use info->shiftpd */
6691a890ab1SHong Zhang   if (info->shiftpd) { /* set sctx.shift_top=max{rs} */
67042898933SHong Zhang     ddiag          = a->diag;
6719f7d0b68SHong Zhang     sctx.shift_top = info->zeropivot;
6726cc28720Svictorle     for (i=0; i<n; i++) {
6739f95998fSHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
6749f7d0b68SHong Zhang       d  = (aa)[ddiag[i]];
6751a890ab1SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
6769f7d0b68SHong Zhang       v  = aa+ai[i];
6779f7d0b68SHong Zhang       nz = ai[i+1] - ai[i];
678e24cfe64SHong Zhang       for (j=0; j<nz; j++)
6791a890ab1SHong Zhang 	rs += PetscAbsScalar(v[j]);
6801a890ab1SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
6816cc28720Svictorle     }
682118f5789SBarry Smith     sctx.shift_top   *= 1.1;
683d2276718SHong Zhang     sctx.nshift_max   = 5;
684d2276718SHong Zhang     sctx.shift_lo     = 0.;
685d2276718SHong Zhang     sctx.shift_hi     = 1.;
686a0a356efSHong Zhang   }
687a0a356efSHong Zhang 
688085256b3SBarry Smith   do {
689d2276718SHong Zhang     sctx.lushift = PETSC_FALSE;
690289bc588SBarry Smith     for (i=0; i<n; i++){
691b3bf805bSHong Zhang       nz    = bi[i+1] - bi[i];
692b3bf805bSHong Zhang       bjtmp = bj + bi[i];
6939f7d0b68SHong Zhang       for  (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
694289bc588SBarry Smith 
695289bc588SBarry Smith       /* load in initial (unfactored row) */
6969f7d0b68SHong Zhang       nz    = ai[r[i]+1] - ai[r[i]];
6979f7d0b68SHong Zhang       ajtmp = aj + ai[r[i]];
6989f7d0b68SHong Zhang       v     = aa + ai[r[i]];
699085256b3SBarry Smith       for (j=0; j<nz; j++) {
700b3bf805bSHong Zhang         rtmp[ics[ajtmp[j]]] = v[j];
701085256b3SBarry Smith       }
702d2276718SHong Zhang       rtmp[ics[r[i]]] += sctx.shift_amount; /* shift the diagonal of the matrix */
703bd1bc851SBarry Smith       /* if (sctx.shift_amount > 0.0) printf("row %d, shift %g\n",i,sctx.shift_amount); */
704289bc588SBarry Smith 
705b3bf805bSHong Zhang       row = *bjtmp++;
706f2582111SSatish Balay       while  (row < i) {
7078c37ef55SBarry Smith         pc = rtmp + row;
7088c37ef55SBarry Smith         if (*pc != 0.0) {
709010a20caSHong Zhang           pv         = b->a + diag_offset[row];
710010a20caSHong Zhang           pj         = b->j + diag_offset[row] + 1;
71135aab85fSBarry Smith           multiplier = *pc / *pv++;
7128c37ef55SBarry Smith           *pc        = multiplier;
713b3bf805bSHong Zhang           nz         = bi[row+1] - diag_offset[row] - 1;
7149f7d0b68SHong Zhang           for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j];
715dc0b31edSSatish Balay           ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr);
716289bc588SBarry Smith         }
717b3bf805bSHong Zhang         row = *bjtmp++;
718289bc588SBarry Smith       }
719416022c9SBarry Smith       /* finished row so stick it into b->a */
720b3bf805bSHong Zhang       pv   = b->a + bi[i] ;
721b3bf805bSHong Zhang       pj   = b->j + bi[i] ;
722b3bf805bSHong Zhang       nz   = bi[i+1] - bi[i];
723b3bf805bSHong Zhang       diag = diag_offset[i] - bi[i];
724e57ff13aSBarry Smith       rs   = 0.0;
725d3d32019SBarry Smith       for (j=0; j<nz; j++) {
7269f7d0b68SHong Zhang         pv[j] = rtmp[pj[j]];
7279f7d0b68SHong Zhang         rs   += PetscAbsScalar(pv[j]);
728d3d32019SBarry Smith       }
729e57ff13aSBarry Smith       rs   -= PetscAbsScalar(pv[diag]);
730d2276718SHong Zhang 
731b3bf805bSHong Zhang       /* 9/13/02 Victor Eijkhout suggested scaling zeropivot by rs for matrices with funny scalings */
732d2276718SHong Zhang       sctx.rs  = rs;
733d2276718SHong Zhang       sctx.pv  = pv[diag];
734c6b1f410SHong Zhang       ierr = MatLUCheckShift_inline(info,sctx,i,newshift);CHKERRQ(ierr);
7355b5bc046SBarry Smith       if (newshift == 1) break;
73635aab85fSBarry Smith     }
737d2276718SHong Zhang 
7380481f469SBarry Smith     if (info->shiftpd && !sctx.lushift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) {
7396cc28720Svictorle       /*
7406c037d1bSvictorle        * if no shift in this attempt & shifting & started shifting & can refine,
7416cc28720Svictorle        * then try lower shift
7426cc28720Svictorle        */
7430481f469SBarry Smith       sctx.shift_hi       = sctx.shift_fraction;
7440481f469SBarry Smith       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
7450481f469SBarry Smith       sctx.shift_amount   = sctx.shift_fraction * sctx.shift_top;
746d2276718SHong Zhang       sctx.lushift        = PETSC_TRUE;
747d2276718SHong Zhang       sctx.nshift++;
7486cc28720Svictorle     }
749d2276718SHong Zhang   } while (sctx.lushift);
7500f11f9aeSSatish Balay 
75135aab85fSBarry Smith   /* invert diagonal entries for simplier triangular solves */
75235aab85fSBarry Smith   for (i=0; i<n; i++) {
753010a20caSHong Zhang     b->a[diag_offset[i]] = 1.0/b->a[diag_offset[i]];
75435aab85fSBarry Smith   }
755606d414cSSatish Balay   ierr = PetscFree(rtmp);CHKERRQ(ierr);
75678b31e54SBarry Smith   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
75778b31e54SBarry Smith   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
758db4efbfdSBarry Smith   if (b->inode.use) {
759db4efbfdSBarry Smith     C->ops->solve   = MatSolve_Inode;
760db4efbfdSBarry Smith   } else {
7618d8c7c43SBarry Smith     PetscTruth row_identity, col_identity;
7628d8c7c43SBarry Smith     ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
7638d8c7c43SBarry Smith     ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr);
7648d8c7c43SBarry Smith     if (row_identity && col_identity) {
7658d8c7c43SBarry Smith       C->ops->solve   = MatSolve_SeqAIJ_NaturalOrdering;
7668d8c7c43SBarry Smith     } else {
767db4efbfdSBarry Smith       C->ops->solve   = MatSolve_SeqAIJ;
768db4efbfdSBarry Smith     }
7698d8c7c43SBarry Smith   }
770db4efbfdSBarry Smith   C->ops->solveadd           = MatSolveAdd_SeqAIJ;
771db4efbfdSBarry Smith   C->ops->solvetranspose     = MatSolveTranspose_SeqAIJ;
772db4efbfdSBarry Smith   C->ops->solvetransposeadd  = MatSolveTransposeAdd_SeqAIJ;
773de26497bSBarry Smith   C->ops->matsolve           = MatMatSolve_SeqAIJ;
774c456f294SBarry Smith   C->assembled    = PETSC_TRUE;
7755c9eb25fSBarry Smith   C->preallocated = PETSC_TRUE;
776d0f46423SBarry Smith   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
777d2276718SHong Zhang   if (sctx.nshift){
778e738e422SBarry Smith      if (info->shiftpd) {
7790481f469SBarry 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);
780e738e422SBarry Smith     } else if (info->shiftnz) {
781e738e422SBarry Smith       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
7826cc28720Svictorle     }
7830697b6caSHong Zhang   }
7843a40ed3dSBarry Smith   PetscFunctionReturn(0);
785289bc588SBarry Smith }
7860889b5dcSvictorle 
787137fb511SHong Zhang /*
788137fb511SHong Zhang    This routine implements inplace ILU(0) with row or/and column permutations.
789137fb511SHong Zhang    Input:
790137fb511SHong Zhang      A - original matrix
791137fb511SHong Zhang    Output;
792137fb511SHong Zhang      A - a->i (rowptr) is same as original rowptr, but factored i-the row is stored in rowperm[i]
793137fb511SHong Zhang          a->j (col index) is permuted by the inverse of colperm, then sorted
794137fb511SHong Zhang          a->a reordered accordingly with a->j
795137fb511SHong Zhang          a->diag (ptr to diagonal elements) is updated.
796137fb511SHong Zhang */
797137fb511SHong Zhang #undef __FUNCT__
798137fb511SHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_InplaceWithPerm"
7990481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_SeqAIJ_InplaceWithPerm(Mat B,Mat A,const MatFactorInfo *info)
800137fb511SHong Zhang {
801b051339dSHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
802b051339dSHong Zhang   IS             isrow = a->row,isicol = a->icol;
803137fb511SHong Zhang   PetscErrorCode ierr;
8045d0c19d7SBarry Smith   const PetscInt *r,*ic,*ics;
8055d0c19d7SBarry Smith   PetscInt       i,j,n=A->rmap->n,*ai=a->i,*aj=a->j;
8065d0c19d7SBarry Smith   PetscInt       *ajtmp,nz,row;
807b051339dSHong Zhang   PetscInt       *diag = a->diag,nbdiag,*pj;
808a77337e4SBarry Smith   PetscScalar    *rtmp,*pc,multiplier,d;
809a77337e4SBarry Smith   MatScalar      *v,*pv;
810137fb511SHong Zhang   PetscReal      rs;
811137fb511SHong Zhang   LUShift_Ctx    sctx;
812b051339dSHong Zhang   PetscInt       newshift;
813137fb511SHong Zhang 
814137fb511SHong Zhang   PetscFunctionBegin;
815719d5645SBarry Smith   if (A != B) SETERRQ(PETSC_ERR_ARG_INCOMP,"input and output matrix must have same address");
816137fb511SHong Zhang   ierr  = ISGetIndices(isrow,&r);CHKERRQ(ierr);
817137fb511SHong Zhang   ierr  = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
818137fb511SHong Zhang   ierr  = PetscMalloc((n+1)*sizeof(PetscScalar),&rtmp);CHKERRQ(ierr);
819137fb511SHong Zhang   ierr  = PetscMemzero(rtmp,(n+1)*sizeof(PetscScalar));CHKERRQ(ierr);
820b051339dSHong Zhang   ics = ic;
821137fb511SHong Zhang 
822137fb511SHong Zhang   sctx.shift_top      = 0;
823137fb511SHong Zhang   sctx.nshift_max     = 0;
824137fb511SHong Zhang   sctx.shift_lo       = 0;
825137fb511SHong Zhang   sctx.shift_hi       = 0;
8260481f469SBarry Smith   sctx.shift_fraction = 0;
827137fb511SHong Zhang 
828137fb511SHong Zhang   /* if both shift schemes are chosen by user, only use info->shiftpd */
829137fb511SHong Zhang   if (info->shiftpd) { /* set sctx.shift_top=max{rs} */
830137fb511SHong Zhang     sctx.shift_top = 0;
831137fb511SHong Zhang     for (i=0; i<n; i++) {
832137fb511SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
833b051339dSHong Zhang       d  = (a->a)[diag[i]];
834137fb511SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
835b051339dSHong Zhang       v  = a->a+ai[i];
836b051339dSHong Zhang       nz = ai[i+1] - ai[i];
837137fb511SHong Zhang       for (j=0; j<nz; j++)
838137fb511SHong Zhang 	rs += PetscAbsScalar(v[j]);
839137fb511SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
840137fb511SHong Zhang     }
841137fb511SHong Zhang     if (sctx.shift_top < info->zeropivot) sctx.shift_top = info->zeropivot;
842137fb511SHong Zhang     sctx.shift_top    *= 1.1;
843137fb511SHong Zhang     sctx.nshift_max   = 5;
844137fb511SHong Zhang     sctx.shift_lo     = 0.;
845137fb511SHong Zhang     sctx.shift_hi     = 1.;
846137fb511SHong Zhang   }
847137fb511SHong Zhang 
848137fb511SHong Zhang   sctx.shift_amount = 0;
849137fb511SHong Zhang   sctx.nshift       = 0;
850137fb511SHong Zhang   do {
851137fb511SHong Zhang     sctx.lushift = PETSC_FALSE;
852137fb511SHong Zhang     for (i=0; i<n; i++){
853b051339dSHong Zhang       /* load in initial unfactored row */
854b051339dSHong Zhang       nz    = ai[r[i]+1] - ai[r[i]];
855b051339dSHong Zhang       ajtmp = aj + ai[r[i]];
856b051339dSHong Zhang       v     = a->a + ai[r[i]];
857137fb511SHong Zhang       /* sort permuted ajtmp and values v accordingly */
858b051339dSHong Zhang       for (j=0; j<nz; j++) ajtmp[j] = ics[ajtmp[j]];
859137fb511SHong Zhang       ierr = PetscSortIntWithScalarArray(nz,ajtmp,v);CHKERRQ(ierr);
860137fb511SHong Zhang 
861b051339dSHong Zhang       diag[r[i]] = ai[r[i]];
862137fb511SHong Zhang       for (j=0; j<nz; j++) {
863137fb511SHong Zhang         rtmp[ajtmp[j]] = v[j];
864b051339dSHong Zhang         if (ajtmp[j] < i) diag[r[i]]++; /* update a->diag */
865137fb511SHong Zhang       }
866137fb511SHong Zhang       rtmp[r[i]] += sctx.shift_amount; /* shift the diagonal of the matrix */
867137fb511SHong Zhang 
868b051339dSHong Zhang       row = *ajtmp++;
869137fb511SHong Zhang       while  (row < i) {
870137fb511SHong Zhang         pc = rtmp + row;
871137fb511SHong Zhang         if (*pc != 0.0) {
872b051339dSHong Zhang           pv         = a->a + diag[r[row]];
873b051339dSHong Zhang           pj         = aj + diag[r[row]] + 1;
874137fb511SHong Zhang 
875137fb511SHong Zhang           multiplier = *pc / *pv++;
876137fb511SHong Zhang           *pc        = multiplier;
877b051339dSHong Zhang           nz         = ai[r[row]+1] - diag[r[row]] - 1;
878b051339dSHong Zhang           for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j];
879dc0b31edSSatish Balay           ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr);
880137fb511SHong Zhang         }
881b051339dSHong Zhang         row = *ajtmp++;
882137fb511SHong Zhang       }
883b051339dSHong Zhang       /* finished row so overwrite it onto a->a */
884b051339dSHong Zhang       pv   = a->a + ai[r[i]] ;
885b051339dSHong Zhang       pj   = aj + ai[r[i]] ;
886b051339dSHong Zhang       nz   = ai[r[i]+1] - ai[r[i]];
887b051339dSHong Zhang       nbdiag = diag[r[i]] - ai[r[i]]; /* num of entries before the diagonal */
888137fb511SHong Zhang 
889137fb511SHong Zhang       rs   = 0.0;
890137fb511SHong Zhang       for (j=0; j<nz; j++) {
891b051339dSHong Zhang         pv[j] = rtmp[pj[j]];
892b051339dSHong Zhang         if (j != nbdiag) rs += PetscAbsScalar(pv[j]);
893137fb511SHong Zhang       }
894137fb511SHong Zhang 
895137fb511SHong Zhang       /* 9/13/02 Victor Eijkhout suggested scaling zeropivot by rs for matrices with funny scalings */
896137fb511SHong Zhang       sctx.rs  = rs;
897b051339dSHong Zhang       sctx.pv  = pv[nbdiag];
898c6b1f410SHong Zhang       ierr = MatLUCheckShift_inline(info,sctx,i,newshift);CHKERRQ(ierr);
899137fb511SHong Zhang       if (newshift == 1) break;
900137fb511SHong Zhang     }
901137fb511SHong Zhang 
9020481f469SBarry Smith     if (info->shiftpd && !sctx.lushift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) {
903137fb511SHong Zhang       /*
904137fb511SHong Zhang        * if no shift in this attempt & shifting & started shifting & can refine,
905137fb511SHong Zhang        * then try lower shift
906137fb511SHong Zhang        */
9070481f469SBarry Smith       sctx.shift_hi        = sctx.shift_fraction;
9080481f469SBarry Smith       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
9090481f469SBarry Smith       sctx.shift_amount    = sctx.shift_fraction * sctx.shift_top;
910137fb511SHong Zhang       sctx.lushift         = PETSC_TRUE;
911137fb511SHong Zhang       sctx.nshift++;
912137fb511SHong Zhang     }
913137fb511SHong Zhang   } while (sctx.lushift);
914137fb511SHong Zhang 
915137fb511SHong Zhang   /* invert diagonal entries for simplier triangular solves */
916137fb511SHong Zhang   for (i=0; i<n; i++) {
917b051339dSHong Zhang     a->a[diag[r[i]]] = 1.0/a->a[diag[r[i]]];
918137fb511SHong Zhang   }
919137fb511SHong Zhang 
920137fb511SHong Zhang   ierr = PetscFree(rtmp);CHKERRQ(ierr);
921137fb511SHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
922137fb511SHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
923b051339dSHong Zhang   A->ops->solve             = MatSolve_SeqAIJ_InplaceWithPerm;
924db4efbfdSBarry Smith   A->ops->solveadd          = MatSolveAdd_SeqAIJ;
925db4efbfdSBarry Smith   A->ops->solvetranspose    = MatSolveTranspose_SeqAIJ;
926db4efbfdSBarry Smith   A->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ;
927b051339dSHong Zhang   A->assembled = PETSC_TRUE;
9285c9eb25fSBarry Smith   A->preallocated = PETSC_TRUE;
929d0f46423SBarry Smith   ierr = PetscLogFlops(A->cmap->n);CHKERRQ(ierr);
930137fb511SHong Zhang   if (sctx.nshift){
931e738e422SBarry Smith     if (info->shiftpd) {
9320481f469SBarry 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);
933e738e422SBarry Smith     } else if (info->shiftnz) {
934e738e422SBarry Smith       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
935137fb511SHong Zhang     }
936137fb511SHong Zhang   }
937137fb511SHong Zhang   PetscFunctionReturn(0);
938137fb511SHong Zhang }
939137fb511SHong Zhang 
9400a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
9414a2ae208SSatish Balay #undef __FUNCT__
9424a2ae208SSatish Balay #define __FUNCT__ "MatLUFactor_SeqAIJ"
9430481f469SBarry Smith PetscErrorCode MatLUFactor_SeqAIJ(Mat A,IS row,IS col,const MatFactorInfo *info)
944da3a660dSBarry Smith {
945dfbe8321SBarry Smith   PetscErrorCode ierr;
946416022c9SBarry Smith   Mat            C;
947416022c9SBarry Smith 
9483a40ed3dSBarry Smith   PetscFunctionBegin;
94943244d56SBarry Smith   ierr = MatGetFactor(A,MAT_SOLVER_PETSC,MAT_FACTOR_LU,&C);CHKERRQ(ierr);
950719d5645SBarry Smith   ierr = MatLUFactorSymbolic(C,A,row,col,info);CHKERRQ(ierr);
951719d5645SBarry Smith   ierr = MatLUFactorNumeric(C,A,info);CHKERRQ(ierr);
952db4efbfdSBarry Smith   A->ops->solve            = C->ops->solve;
953db4efbfdSBarry Smith   A->ops->solvetranspose   = C->ops->solvetranspose;
954273d9f13SBarry Smith   ierr = MatHeaderCopy(A,C);CHKERRQ(ierr);
95552e6d16bSBarry Smith   ierr = PetscLogObjectParent(A,((Mat_SeqAIJ*)(A->data))->icol);CHKERRQ(ierr);
9563a40ed3dSBarry Smith   PetscFunctionReturn(0);
957da3a660dSBarry Smith }
9580a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
9591d098868SHong Zhang 
960889a8dedSBarry Smith 
9614a2ae208SSatish Balay #undef __FUNCT__
9624a2ae208SSatish Balay #define __FUNCT__ "MatSolve_SeqAIJ"
963dfbe8321SBarry Smith PetscErrorCode MatSolve_SeqAIJ(Mat A,Vec bb,Vec xx)
9648c37ef55SBarry Smith {
965416022c9SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
966416022c9SBarry Smith   IS                iscol = a->col,isrow = a->row;
9676849ba73SBarry Smith   PetscErrorCode    ierr;
9685d0c19d7SBarry Smith   PetscInt          i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
9695d0c19d7SBarry Smith   PetscInt          nz;
9705d0c19d7SBarry Smith   const PetscInt    *rout,*cout,*r,*c;
971dd6ea824SBarry Smith   PetscScalar       *x,*tmp,*tmps,sum;
972d9fead3dSBarry Smith   const PetscScalar *b;
973dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
9748c37ef55SBarry Smith 
9753a40ed3dSBarry Smith   PetscFunctionBegin;
9763a40ed3dSBarry Smith   if (!n) PetscFunctionReturn(0);
97791bf9adeSBarry Smith 
978d9fead3dSBarry Smith   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
9791ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
980416022c9SBarry Smith   tmp  = a->solve_work;
9818c37ef55SBarry Smith 
9823b2fbd54SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
9833b2fbd54SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
9848c37ef55SBarry Smith 
9858c37ef55SBarry Smith   /* forward solve the lower triangular */
9868c37ef55SBarry Smith   tmp[0] = b[*r++];
987010a20caSHong Zhang   tmps   = tmp;
9888c37ef55SBarry Smith   for (i=1; i<n; i++) {
989010a20caSHong Zhang     v   = aa + ai[i] ;
990010a20caSHong Zhang     vi  = aj + ai[i] ;
99153e7425aSBarry Smith     nz  = a->diag[i] - ai[i];
99253e7425aSBarry Smith     sum = b[*r++];
993003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
9948c37ef55SBarry Smith     tmp[i] = sum;
9958c37ef55SBarry Smith   }
9968c37ef55SBarry Smith 
9978c37ef55SBarry Smith   /* backward solve the upper triangular */
9988c37ef55SBarry Smith   for (i=n-1; i>=0; i--){
999010a20caSHong Zhang     v   = aa + a->diag[i] + 1;
1000010a20caSHong Zhang     vi  = aj + a->diag[i] + 1;
1001416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
10028c37ef55SBarry Smith     sum = tmp[i];
1003003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
1004010a20caSHong Zhang     x[*c--] = tmp[i] = sum*aa[a->diag[i]];
10058c37ef55SBarry Smith   }
10068c37ef55SBarry Smith 
10073b2fbd54SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
10083b2fbd54SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
1009d9fead3dSBarry Smith   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
10101ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1011dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
10123a40ed3dSBarry Smith   PetscFunctionReturn(0);
10138c37ef55SBarry Smith }
1014026e39d0SSatish Balay 
10152bebee5dSHong Zhang #undef __FUNCT__
10162bebee5dSHong Zhang #define __FUNCT__ "MatMatSolve_SeqAIJ"
10172bebee5dSHong Zhang PetscErrorCode MatMatSolve_SeqAIJ(Mat A,Mat B,Mat X)
10182bebee5dSHong Zhang {
10192bebee5dSHong Zhang   Mat_SeqAIJ      *a = (Mat_SeqAIJ*)A->data;
10202bebee5dSHong Zhang   IS              iscol = a->col,isrow = a->row;
10212bebee5dSHong Zhang   PetscErrorCode  ierr;
10225d0c19d7SBarry Smith   PetscInt        i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
10235d0c19d7SBarry Smith   PetscInt        nz,neq;
10245d0c19d7SBarry Smith   const PetscInt  *rout,*cout,*r,*c;
1025dd6ea824SBarry Smith   PetscScalar     *x,*b,*tmp,*tmps,sum;
1026dd6ea824SBarry Smith   const MatScalar *aa = a->a,*v;
1027db4efbfdSBarry Smith   PetscTruth      bisdense,xisdense;
10282bebee5dSHong Zhang 
10292bebee5dSHong Zhang   PetscFunctionBegin;
10302bebee5dSHong Zhang   if (!n) PetscFunctionReturn(0);
10312bebee5dSHong Zhang 
1032db4efbfdSBarry Smith   ierr = PetscTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr);
1033db4efbfdSBarry Smith   if (!bisdense) SETERRQ(PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix");
1034db4efbfdSBarry Smith   ierr = PetscTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr);
1035db4efbfdSBarry Smith   if (!xisdense) SETERRQ(PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix");
1036db4efbfdSBarry Smith 
10372bebee5dSHong Zhang   ierr = MatGetArray(B,&b);CHKERRQ(ierr);
10382bebee5dSHong Zhang   ierr = MatGetArray(X,&x);CHKERRQ(ierr);
10392bebee5dSHong Zhang 
10402bebee5dSHong Zhang   tmp  = a->solve_work;
10412bebee5dSHong Zhang   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
10422bebee5dSHong Zhang   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
10432bebee5dSHong Zhang 
1044a36b22ccSSatish Balay   for (neq=0; neq<B->cmap->n; neq++){
10452bebee5dSHong Zhang     /* forward solve the lower triangular */
10462bebee5dSHong Zhang     tmp[0] = b[r[0]];
10472bebee5dSHong Zhang     tmps   = tmp;
10482bebee5dSHong Zhang     for (i=1; i<n; i++) {
10492bebee5dSHong Zhang       v   = aa + ai[i] ;
10502bebee5dSHong Zhang       vi  = aj + ai[i] ;
10512bebee5dSHong Zhang       nz  = a->diag[i] - ai[i];
10522bebee5dSHong Zhang       sum = b[r[i]];
1053003131ecSBarry Smith       PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
10542bebee5dSHong Zhang       tmp[i] = sum;
10552bebee5dSHong Zhang     }
10562bebee5dSHong Zhang     /* backward solve the upper triangular */
10572bebee5dSHong Zhang     for (i=n-1; i>=0; i--){
10582bebee5dSHong Zhang       v   = aa + a->diag[i] + 1;
10592bebee5dSHong Zhang       vi  = aj + a->diag[i] + 1;
10602bebee5dSHong Zhang       nz  = ai[i+1] - a->diag[i] - 1;
10612bebee5dSHong Zhang       sum = tmp[i];
1062003131ecSBarry Smith       PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
10632bebee5dSHong Zhang       x[c[i]] = tmp[i] = sum*aa[a->diag[i]];
10642bebee5dSHong Zhang     }
10652bebee5dSHong Zhang 
10662bebee5dSHong Zhang     b += n;
10672bebee5dSHong Zhang     x += n;
10682bebee5dSHong Zhang   }
10692bebee5dSHong Zhang   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
10702bebee5dSHong Zhang   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
10712bebee5dSHong Zhang   ierr = MatRestoreArray(B,&b);CHKERRQ(ierr);
10722bebee5dSHong Zhang   ierr = MatRestoreArray(X,&x);CHKERRQ(ierr);
1073dc0b31edSSatish Balay   ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr);
10742bebee5dSHong Zhang   PetscFunctionReturn(0);
10752bebee5dSHong Zhang }
10762bebee5dSHong Zhang 
1077137fb511SHong Zhang #undef __FUNCT__
1078137fb511SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_InplaceWithPerm"
1079137fb511SHong Zhang PetscErrorCode MatSolve_SeqAIJ_InplaceWithPerm(Mat A,Vec bb,Vec xx)
1080137fb511SHong Zhang {
1081137fb511SHong Zhang   Mat_SeqAIJ      *a = (Mat_SeqAIJ*)A->data;
1082137fb511SHong Zhang   IS              iscol = a->col,isrow = a->row;
1083137fb511SHong Zhang   PetscErrorCode  ierr;
10845d0c19d7SBarry Smith   const PetscInt  *r,*c,*rout,*cout;
10855d0c19d7SBarry Smith   PetscInt        i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
10865d0c19d7SBarry Smith   PetscInt        nz,row;
1087dd6ea824SBarry Smith   PetscScalar     *x,*b,*tmp,*tmps,sum;
1088dd6ea824SBarry Smith   const MatScalar *aa = a->a,*v;
1089137fb511SHong Zhang 
1090137fb511SHong Zhang   PetscFunctionBegin;
1091137fb511SHong Zhang   if (!n) PetscFunctionReturn(0);
1092137fb511SHong Zhang 
1093137fb511SHong Zhang   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
1094137fb511SHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1095137fb511SHong Zhang   tmp  = a->solve_work;
1096137fb511SHong Zhang 
1097137fb511SHong Zhang   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
1098137fb511SHong Zhang   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
1099137fb511SHong Zhang 
1100137fb511SHong Zhang   /* forward solve the lower triangular */
1101137fb511SHong Zhang   tmp[0] = b[*r++];
1102137fb511SHong Zhang   tmps   = tmp;
1103137fb511SHong Zhang   for (row=1; row<n; row++) {
1104137fb511SHong Zhang     i   = rout[row]; /* permuted row */
1105137fb511SHong Zhang     v   = aa + ai[i] ;
1106137fb511SHong Zhang     vi  = aj + ai[i] ;
1107137fb511SHong Zhang     nz  = a->diag[i] - ai[i];
1108137fb511SHong Zhang     sum = b[*r++];
1109003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
1110137fb511SHong Zhang     tmp[row] = sum;
1111137fb511SHong Zhang   }
1112137fb511SHong Zhang 
1113137fb511SHong Zhang   /* backward solve the upper triangular */
1114137fb511SHong Zhang   for (row=n-1; row>=0; row--){
1115137fb511SHong Zhang     i   = rout[row]; /* permuted row */
1116137fb511SHong Zhang     v   = aa + a->diag[i] + 1;
1117137fb511SHong Zhang     vi  = aj + a->diag[i] + 1;
1118137fb511SHong Zhang     nz  = ai[i+1] - a->diag[i] - 1;
1119137fb511SHong Zhang     sum = tmp[row];
1120003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
1121137fb511SHong Zhang     x[*c--] = tmp[row] = sum*aa[a->diag[i]];
1122137fb511SHong Zhang   }
1123137fb511SHong Zhang 
1124137fb511SHong Zhang   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
1125137fb511SHong Zhang   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
1126137fb511SHong Zhang   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
1127137fb511SHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1128dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
1129137fb511SHong Zhang   PetscFunctionReturn(0);
1130137fb511SHong Zhang }
1131137fb511SHong Zhang 
1132930ae53cSBarry Smith /* ----------------------------------------------------------- */
1133a4005a5dSBarry Smith #include "../src/mat/impls/aij/seq/ftn-kernels/fsolve.h"
11344a2ae208SSatish Balay #undef __FUNCT__
11354a2ae208SSatish Balay #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering"
1136dfbe8321SBarry Smith PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering(Mat A,Vec bb,Vec xx)
1137930ae53cSBarry Smith {
1138930ae53cSBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
11396849ba73SBarry Smith   PetscErrorCode    ierr;
1140d0f46423SBarry Smith   PetscInt          n = A->rmap->n;
1141b377110cSBarry Smith   const PetscInt    *ai = a->i,*aj = a->j,*adiag = a->diag;
1142356650c2SBarry Smith   PetscScalar       *x;
1143356650c2SBarry Smith   const PetscScalar *b;
1144dd6ea824SBarry Smith   const MatScalar   *aa = a->a;
1145aa482453SBarry Smith #if !defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ)
1146356650c2SBarry Smith   PetscInt          adiag_i,i,nz,ai_i;
1147b377110cSBarry Smith   const PetscInt    *vi;
1148dd6ea824SBarry Smith   const MatScalar   *v;
1149dd6ea824SBarry Smith   PetscScalar       sum;
1150d85afc42SSatish Balay #endif
1151930ae53cSBarry Smith 
11523a40ed3dSBarry Smith   PetscFunctionBegin;
11533a40ed3dSBarry Smith   if (!n) PetscFunctionReturn(0);
1154930ae53cSBarry Smith 
1155356650c2SBarry Smith   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
11561ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1157930ae53cSBarry Smith 
1158aa482453SBarry Smith #if defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ)
11591c4feecaSSatish Balay   fortransolveaij_(&n,x,ai,aj,adiag,aa,b);
11601c4feecaSSatish Balay #else
1161930ae53cSBarry Smith   /* forward solve the lower triangular */
1162930ae53cSBarry Smith   x[0] = b[0];
1163930ae53cSBarry Smith   for (i=1; i<n; i++) {
1164930ae53cSBarry Smith     ai_i = ai[i];
1165930ae53cSBarry Smith     v    = aa + ai_i;
1166930ae53cSBarry Smith     vi   = aj + ai_i;
1167930ae53cSBarry Smith     nz   = adiag[i] - ai_i;
1168930ae53cSBarry Smith     sum  = b[i];
1169003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
1170930ae53cSBarry Smith     x[i] = sum;
1171930ae53cSBarry Smith   }
1172930ae53cSBarry Smith 
1173930ae53cSBarry Smith   /* backward solve the upper triangular */
1174930ae53cSBarry Smith   for (i=n-1; i>=0; i--){
1175930ae53cSBarry Smith     adiag_i = adiag[i];
1176d708749eSBarry Smith     v       = aa + adiag_i + 1;
1177d708749eSBarry Smith     vi      = aj + adiag_i + 1;
1178930ae53cSBarry Smith     nz      = ai[i+1] - adiag_i - 1;
1179930ae53cSBarry Smith     sum     = x[i];
1180003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
1181930ae53cSBarry Smith     x[i]    = sum*aa[adiag_i];
1182930ae53cSBarry Smith   }
11831c4feecaSSatish Balay #endif
1184dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
1185356650c2SBarry Smith   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
11861ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
11873a40ed3dSBarry Smith   PetscFunctionReturn(0);
1188930ae53cSBarry Smith }
1189930ae53cSBarry Smith 
11904a2ae208SSatish Balay #undef __FUNCT__
11914a2ae208SSatish Balay #define __FUNCT__ "MatSolveAdd_SeqAIJ"
1192dfbe8321SBarry Smith PetscErrorCode MatSolveAdd_SeqAIJ(Mat A,Vec bb,Vec yy,Vec xx)
1193da3a660dSBarry Smith {
1194416022c9SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
1195416022c9SBarry Smith   IS                iscol = a->col,isrow = a->row;
11966849ba73SBarry Smith   PetscErrorCode    ierr;
119704fbf559SBarry Smith   PetscInt          i, n = A->rmap->n,j;
11985d0c19d7SBarry Smith   PetscInt          nz;
119904fbf559SBarry Smith   const PetscInt    *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j;
120004fbf559SBarry Smith   PetscScalar       *x,*tmp,sum;
120104fbf559SBarry Smith   const PetscScalar *b;
1202dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
1203da3a660dSBarry Smith 
12043a40ed3dSBarry Smith   PetscFunctionBegin;
120578b31e54SBarry Smith   if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);}
1206da3a660dSBarry Smith 
120704fbf559SBarry Smith   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
12081ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1209416022c9SBarry Smith   tmp  = a->solve_work;
1210da3a660dSBarry Smith 
12113b2fbd54SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
12123b2fbd54SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
1213da3a660dSBarry Smith 
1214da3a660dSBarry Smith   /* forward solve the lower triangular */
1215da3a660dSBarry Smith   tmp[0] = b[*r++];
1216da3a660dSBarry Smith   for (i=1; i<n; i++) {
1217010a20caSHong Zhang     v   = aa + ai[i] ;
1218010a20caSHong Zhang     vi  = aj + ai[i] ;
1219416022c9SBarry Smith     nz  = a->diag[i] - ai[i];
1220da3a660dSBarry Smith     sum = b[*r++];
122104fbf559SBarry Smith     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
1222da3a660dSBarry Smith     tmp[i] = sum;
1223da3a660dSBarry Smith   }
1224da3a660dSBarry Smith 
1225da3a660dSBarry Smith   /* backward solve the upper triangular */
1226da3a660dSBarry Smith   for (i=n-1; i>=0; i--){
1227010a20caSHong Zhang     v   = aa + a->diag[i] + 1;
1228010a20caSHong Zhang     vi  = aj + a->diag[i] + 1;
1229416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
1230da3a660dSBarry Smith     sum = tmp[i];
123104fbf559SBarry Smith     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
1232010a20caSHong Zhang     tmp[i] = sum*aa[a->diag[i]];
1233da3a660dSBarry Smith     x[*c--] += tmp[i];
1234da3a660dSBarry Smith   }
1235da3a660dSBarry Smith 
12363b2fbd54SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
12373b2fbd54SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
123804fbf559SBarry Smith   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
12391ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1240dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
1241898183ecSLois Curfman McInnes 
12423a40ed3dSBarry Smith   PetscFunctionReturn(0);
1243da3a660dSBarry Smith }
124404fbf559SBarry Smith 
12454a2ae208SSatish Balay #undef __FUNCT__
12464a2ae208SSatish Balay #define __FUNCT__ "MatSolveTranspose_SeqAIJ"
1247dfbe8321SBarry Smith PetscErrorCode MatSolveTranspose_SeqAIJ(Mat A,Vec bb,Vec xx)
1248da3a660dSBarry Smith {
1249416022c9SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
12502235e667SBarry Smith   IS                iscol = a->col,isrow = a->row;
12516849ba73SBarry Smith   PetscErrorCode    ierr;
125204fbf559SBarry Smith   const PetscInt    *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi;
125304fbf559SBarry Smith   PetscInt          i,n = A->rmap->n,j;
125404fbf559SBarry Smith   PetscInt          nz;
125504fbf559SBarry Smith   PetscScalar       *x,*tmp,s1;
1256dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
125704fbf559SBarry Smith   const PetscScalar *b;
1258da3a660dSBarry Smith 
12593a40ed3dSBarry Smith   PetscFunctionBegin;
126004fbf559SBarry Smith   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
12611ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1262416022c9SBarry Smith   tmp  = a->solve_work;
1263da3a660dSBarry Smith 
12642235e667SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
12652235e667SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
1266da3a660dSBarry Smith 
1267da3a660dSBarry Smith   /* copy the b into temp work space according to permutation */
12682235e667SBarry Smith   for (i=0; i<n; i++) tmp[i] = b[c[i]];
1269da3a660dSBarry Smith 
1270da3a660dSBarry Smith   /* forward solve the U^T */
1271da3a660dSBarry Smith   for (i=0; i<n; i++) {
1272010a20caSHong Zhang     v   = aa + diag[i] ;
1273010a20caSHong Zhang     vi  = aj + diag[i] + 1;
1274f1af5d2fSBarry Smith     nz  = ai[i+1] - diag[i] - 1;
1275c38d4ed2SBarry Smith     s1  = tmp[i];
1276ef66eb69SBarry Smith     s1 *= (*v++);  /* multiply by inverse of diagonal entry */
127704fbf559SBarry Smith     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1278c38d4ed2SBarry Smith     tmp[i] = s1;
1279da3a660dSBarry Smith   }
1280da3a660dSBarry Smith 
1281da3a660dSBarry Smith   /* backward solve the L^T */
1282da3a660dSBarry Smith   for (i=n-1; i>=0; i--){
1283010a20caSHong Zhang     v   = aa + diag[i] - 1 ;
1284010a20caSHong Zhang     vi  = aj + diag[i] - 1 ;
1285f1af5d2fSBarry Smith     nz  = diag[i] - ai[i];
1286f1af5d2fSBarry Smith     s1  = tmp[i];
128704fbf559SBarry Smith     for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j];
1288da3a660dSBarry Smith   }
1289da3a660dSBarry Smith 
1290da3a660dSBarry Smith   /* copy tmp into x according to permutation */
1291591294e4SBarry Smith   for (i=0; i<n; i++) x[r[i]] = tmp[i];
1292da3a660dSBarry Smith 
12932235e667SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
12942235e667SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
129504fbf559SBarry Smith   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
12961ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1297da3a660dSBarry Smith 
1298dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr);
12993a40ed3dSBarry Smith   PetscFunctionReturn(0);
1300da3a660dSBarry Smith }
1301da3a660dSBarry Smith 
13024a2ae208SSatish Balay #undef __FUNCT__
13034a2ae208SSatish Balay #define __FUNCT__ "MatSolveTransposeAdd_SeqAIJ"
1304dfbe8321SBarry Smith PetscErrorCode MatSolveTransposeAdd_SeqAIJ(Mat A,Vec bb,Vec zz,Vec xx)
1305da3a660dSBarry Smith {
1306416022c9SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
13072235e667SBarry Smith   IS                iscol = a->col,isrow = a->row;
13086849ba73SBarry Smith   PetscErrorCode    ierr;
130904fbf559SBarry Smith   const PetscInt    *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi;
131004fbf559SBarry Smith   PetscInt          i,n = A->rmap->n,j;
131104fbf559SBarry Smith   PetscInt          nz;
131204fbf559SBarry Smith   PetscScalar       *x,*tmp,s1;
1313dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
131404fbf559SBarry Smith   const PetscScalar *b;
13156abc6512SBarry Smith 
13163a40ed3dSBarry Smith   PetscFunctionBegin;
131723bc6035SMatthew Knepley   if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);}
131804fbf559SBarry Smith   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
13191ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1320416022c9SBarry Smith   tmp  = a->solve_work;
13216abc6512SBarry Smith 
13222235e667SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
13232235e667SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
13246abc6512SBarry Smith 
13256abc6512SBarry Smith   /* copy the b into temp work space according to permutation */
13262235e667SBarry Smith   for (i=0; i<n; i++) tmp[i] = b[c[i]];
13276abc6512SBarry Smith 
13286abc6512SBarry Smith   /* forward solve the U^T */
13296abc6512SBarry Smith   for (i=0; i<n; i++) {
1330010a20caSHong Zhang     v   = aa + diag[i] ;
1331010a20caSHong Zhang     vi  = aj + diag[i] + 1;
1332f1af5d2fSBarry Smith     nz  = ai[i+1] - diag[i] - 1;
133304fbf559SBarry Smith     s1  = tmp[i];
133404fbf559SBarry Smith     s1 *= (*v++);  /* multiply by inverse of diagonal entry */
133504fbf559SBarry Smith     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
133604fbf559SBarry Smith     tmp[i] = s1;
13376abc6512SBarry Smith   }
13386abc6512SBarry Smith 
13396abc6512SBarry Smith   /* backward solve the L^T */
13406abc6512SBarry Smith   for (i=n-1; i>=0; i--){
1341010a20caSHong Zhang     v   = aa + diag[i] - 1 ;
1342010a20caSHong Zhang     vi  = aj + diag[i] - 1 ;
1343f1af5d2fSBarry Smith     nz  = diag[i] - ai[i];
134404fbf559SBarry Smith     s1  = tmp[i];
134504fbf559SBarry Smith     for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j];
13466abc6512SBarry Smith   }
13476abc6512SBarry Smith 
13486abc6512SBarry Smith   /* copy tmp into x according to permutation */
13496abc6512SBarry Smith   for (i=0; i<n; i++) x[r[i]] += tmp[i];
13506abc6512SBarry Smith 
13512235e667SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
13522235e667SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
135304fbf559SBarry Smith   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
13541ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
13556abc6512SBarry Smith 
135604fbf559SBarry Smith   ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr);
13573a40ed3dSBarry Smith   PetscFunctionReturn(0);
1358da3a660dSBarry Smith }
135904fbf559SBarry Smith 
13609e25ed09SBarry Smith /* ----------------------------------------------------------------*/
13613c5fc038SBarry Smith EXTERN PetscErrorCode Mat_CheckInode(Mat,PetscTruth);
1362f77e22a1SHong Zhang EXTERN PetscErrorCode MatDuplicateNoCreate_SeqAIJ(Mat,Mat,MatDuplicateOption,PetscTruth);
136308480c60SBarry Smith 
13648fc3a347SHong Zhang /*
13658fc3a347SHong Zhang    ilu(0) with natural ordering under new data structure.
13668fc3a347SHong Zhang    Factored arrays bj and ba are stored as
13678fc3a347SHong Zhang      L(0,:), L(1,:), ...,L(n-1,:),  U(n-1,:),...,U(i,:),U(i-1,:),...,U(0,:)
13688fc3a347SHong Zhang 
13698fc3a347SHong Zhang    bi=fact->i is an array of size 2n+2, in which
13708fc3a347SHong Zhang    bi+
13718fc3a347SHong Zhang      bi[i]      ->  1st entry of L(i,:),i=0,...,i-1
1372f77e22a1SHong Zhang      bi[n]      ->  points to L(n-1,:)+1
13738fc3a347SHong Zhang      bi[n+1]    ->  1st entry of U(n-1,:)
13748fc3a347SHong Zhang      bi[2n-i]   ->  1st entry of U(i,:)
13758fc3a347SHong Zhang      bi[2n-i+1] ->  end of U(i,:)+1, the 1st entry of U(i-1,:)
1376f77e22a1SHong Zhang      bi[2n]     ->  1st entry of U(0,:)
1377f77e22a1SHong Zhang      bi[2n+1]   ->  points to U(0,:)+1
13788fc3a347SHong Zhang 
13798fc3a347SHong Zhang    U(i,:) contains diag[i] as its last entry, i.e.,
13808fc3a347SHong Zhang     U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i])
13818fc3a347SHong Zhang */
1382c9c72f8fSHong Zhang #undef __FUNCT__
13838fc3a347SHong Zhang #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_ilu0_newdatastruct"
13848fc3a347SHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ_ilu0_newdatastruct(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
1385c9c72f8fSHong Zhang {
1386c9c72f8fSHong Zhang 
1387c9c72f8fSHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
1388c9c72f8fSHong Zhang   PetscErrorCode     ierr;
1389c9c72f8fSHong Zhang   PetscInt           n=A->rmap->n,*ai=a->i,*aj,*adiag=a->diag;
13908fc3a347SHong Zhang   PetscInt           i,j,nz,*bi,*bj,*bdiag;
1391c9c72f8fSHong Zhang 
1392c9c72f8fSHong Zhang   PetscFunctionBegin;
1393f77e22a1SHong Zhang   ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_FALSE);CHKERRQ(ierr);
1394c9c72f8fSHong Zhang   b    = (Mat_SeqAIJ*)(fact)->data;
1395c9c72f8fSHong Zhang 
1396f77e22a1SHong Zhang   /* allocate matrix arrays for new data structure */
1397f77e22a1SHong Zhang   ierr = PetscMalloc3(ai[n]+1,PetscScalar,&b->a,ai[n]+1,PetscInt,&b->j,2*n+2,PetscInt,&b->i);CHKERRQ(ierr);
1398f77e22a1SHong Zhang   ierr = PetscLogObjectMemory(fact,ai[n]*(sizeof(PetscScalar)+sizeof(PetscInt))+(2*n+2)*sizeof(PetscInt));CHKERRQ(ierr);
1399f77e22a1SHong Zhang   b->singlemalloc = PETSC_TRUE;
1400f77e22a1SHong Zhang   if (!b->diag){
14018fc3a347SHong Zhang     ierr = PetscMalloc((n+1)*sizeof(PetscInt),&b->diag);CHKERRQ(ierr);
1402f77e22a1SHong Zhang   }
1403c9c72f8fSHong Zhang   bdiag = b->diag;
1404c9c72f8fSHong Zhang 
1405c9c72f8fSHong Zhang   if (n > 0) {
140616a2bf60SHong Zhang     ierr = PetscMemzero(b->a,(ai[n])*sizeof(MatScalar));CHKERRQ(ierr);
1407c9c72f8fSHong Zhang   }
1408c9c72f8fSHong Zhang 
1409c9c72f8fSHong Zhang   /* set bi and bj with new data structure */
1410c9c72f8fSHong Zhang   bi = b->i;
1411c9c72f8fSHong Zhang   bj = b->j;
1412c9c72f8fSHong Zhang 
1413c9c72f8fSHong Zhang   /* L part */
1414c9c72f8fSHong Zhang   bi[0] = 0;
1415c9c72f8fSHong Zhang   for (i=0; i<n; i++){
1416c9c72f8fSHong Zhang     nz = adiag[i] - ai[i];
1417c9c72f8fSHong Zhang     bi[i+1] = bi[i] + nz;
1418c9c72f8fSHong Zhang     aj = a->j + ai[i];
1419c9c72f8fSHong Zhang     for (j=0; j<nz; j++){
1420c9c72f8fSHong Zhang       *bj = aj[j]; bj++;
1421c9c72f8fSHong Zhang     }
1422c9c72f8fSHong Zhang   }
1423c9c72f8fSHong Zhang 
1424c9c72f8fSHong Zhang   /* U part */
14258fc3a347SHong Zhang   bi[n+1] = bi[n];
1426c9c72f8fSHong Zhang   for (i=n-1; i>=0; i--){
1427c9c72f8fSHong Zhang     nz = ai[i+1] - adiag[i] - 1;
14288fc3a347SHong Zhang     bi[2*n-i+1] = bi[2*n-i] + nz + 1;
1429c9c72f8fSHong Zhang     aj = a->j + adiag[i] + 1;
1430c9c72f8fSHong Zhang     for (j=0; j<nz; j++){
1431c9c72f8fSHong Zhang       *bj = aj[j]; bj++;
1432c9c72f8fSHong Zhang     }
1433c9c72f8fSHong Zhang     /* diag[i] */
1434c9c72f8fSHong Zhang     *bj = i; bj++;
14358fc3a347SHong Zhang     bdiag[i] = bi[2*n-i+1]-1;
1436c9c72f8fSHong Zhang   }
1437c9c72f8fSHong Zhang   PetscFunctionReturn(0);
1438c9c72f8fSHong Zhang }
1439c9c72f8fSHong Zhang 
14404a2ae208SSatish Balay #undef __FUNCT__
14416516239fSHong Zhang #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_newdatastruct"
14426516239fSHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ_newdatastruct(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
14439e25ed09SBarry Smith {
1444416022c9SBarry Smith   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
14459e25ed09SBarry Smith   IS                 isicol;
14466849ba73SBarry Smith   PetscErrorCode     ierr;
14475d0c19d7SBarry Smith   const PetscInt     *r,*ic;
14485d0c19d7SBarry Smith   PetscInt           n=A->rmap->n,*ai=a->i,*aj=a->j,d;
14495a8e39fbSHong Zhang   PetscInt           *bi,*cols,nnz,*cols_lvl;
1450052f0c41SBarry Smith   PetscInt           *bdiag,prow,fm,nzbd,reallocs=0,dcount=0;
14515a8e39fbSHong Zhang   PetscInt           i,levels,diagonal_fill;
145277c4ece6SBarry Smith   PetscTruth         col_identity,row_identity;
1453329f5518SBarry Smith   PetscReal          f;
14545a8e39fbSHong Zhang   PetscInt           nlnk,*lnk,*lnk_lvl=PETSC_NULL;
14555a8e39fbSHong Zhang   PetscBT            lnkbt;
14565a8e39fbSHong Zhang   PetscInt           nzi,*bj,**bj_ptr,**bjlvl_ptr;
1457a1a86e44SBarry Smith   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
1458a1a86e44SBarry Smith   PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL;
145909f38230SBarry Smith   PetscTruth         missing;
14609e25ed09SBarry Smith 
14613a40ed3dSBarry Smith   PetscFunctionBegin;
1462d0f46423SBarry 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);
14636cf997b0SBarry Smith   f             = info->fill;
146438baddfdSBarry Smith   levels        = (PetscInt)info->levels;
146538baddfdSBarry Smith   diagonal_fill = (PetscInt)info->diagonal_fill;
14664c49b128SBarry Smith   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
146782bf6240SBarry Smith 
1468be0abb6dSBarry Smith   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
1469be0abb6dSBarry Smith   ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr);
1470c9c72f8fSHong Zhang 
14716516239fSHong Zhang   if (!levels && row_identity && col_identity) {
14726516239fSHong Zhang     /* special case: ilu(0) with natural ordering */
14738fc3a347SHong Zhang     ierr = MatILUFactorSymbolic_SeqAIJ_ilu0_newdatastruct(fact,A,isrow,iscol,info);CHKERRQ(ierr);
1474c9c72f8fSHong Zhang     (fact)->ops->lufactornumeric =  MatLUFactorNumeric_SeqAIJ_newdatastruct;
14756516239fSHong Zhang 
14766516239fSHong Zhang     fact->factor = MAT_FACTOR_ILU;
14776516239fSHong Zhang     (fact)->info.factor_mallocs    = 0;
14786516239fSHong Zhang     (fact)->info.fill_ratio_given  = info->fill;
14796516239fSHong Zhang     (fact)->info.fill_ratio_needed = 1.0;
14806516239fSHong Zhang     b               = (Mat_SeqAIJ*)(fact)->data;
14816516239fSHong Zhang     ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr);
14826516239fSHong Zhang     if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d);
14836516239fSHong Zhang     b->row              = isrow;
14846516239fSHong Zhang     b->col              = iscol;
14856516239fSHong Zhang     b->icol             = isicol;
14866516239fSHong Zhang     ierr                = PetscMalloc(((fact)->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
14876516239fSHong Zhang     ierr                = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
14886516239fSHong Zhang     ierr                = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
14896516239fSHong Zhang     /* ierr = MatILUFactorSymbolic_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr); */
14906516239fSHong Zhang     PetscFunctionReturn(0);
14916516239fSHong Zhang   }
14926516239fSHong Zhang 
14936516239fSHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
14946516239fSHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
14956516239fSHong Zhang 
14966516239fSHong Zhang   /* get new row pointers */
14976516239fSHong Zhang   ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
14986516239fSHong Zhang   bi[0] = 0;
14996516239fSHong Zhang   /* bdiag is location of diagonal in factor */
15006516239fSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
15016516239fSHong Zhang   bdiag[0]  = 0;
15026516239fSHong Zhang 
15036516239fSHong Zhang   ierr = PetscMalloc((2*n+1)*sizeof(PetscInt**),&bj_ptr);CHKERRQ(ierr);
15046516239fSHong Zhang   bjlvl_ptr = (PetscInt**)(bj_ptr + n);
15056516239fSHong Zhang 
15066516239fSHong Zhang   /* create a linked list for storing column indices of the active row */
15076516239fSHong Zhang   nlnk = n + 1;
15086516239fSHong Zhang   ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
15096516239fSHong Zhang 
15106516239fSHong Zhang   /* initial FreeSpace size is f*(ai[n]+1) */
15116516239fSHong Zhang   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr);
15126516239fSHong Zhang   current_space = free_space;
15136516239fSHong Zhang   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr);
15146516239fSHong Zhang   current_space_lvl = free_space_lvl;
15156516239fSHong Zhang 
15166516239fSHong Zhang   for (i=0; i<n; i++) {
15176516239fSHong Zhang     nzi = 0;
15186516239fSHong Zhang     /* copy current row into linked list */
15196516239fSHong Zhang     nnz  = ai[r[i]+1] - ai[r[i]];
15206516239fSHong 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);
15216516239fSHong Zhang     cols = aj + ai[r[i]];
15226516239fSHong Zhang     lnk[i] = -1; /* marker to indicate if diagonal exists */
15236516239fSHong Zhang     ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
15246516239fSHong Zhang     nzi += nlnk;
15256516239fSHong Zhang 
15266516239fSHong Zhang     /* make sure diagonal entry is included */
15276516239fSHong Zhang     if (diagonal_fill && lnk[i] == -1) {
15286516239fSHong Zhang       fm = n;
15296516239fSHong Zhang       while (lnk[fm] < i) fm = lnk[fm];
15306516239fSHong Zhang       lnk[i]     = lnk[fm]; /* insert diagonal into linked list */
15316516239fSHong Zhang       lnk[fm]    = i;
15326516239fSHong Zhang       lnk_lvl[i] = 0;
15336516239fSHong Zhang       nzi++; dcount++;
15346516239fSHong Zhang     }
15356516239fSHong Zhang 
15366516239fSHong Zhang     /* add pivot rows into the active row */
15376516239fSHong Zhang     nzbd = 0;
15386516239fSHong Zhang     prow = lnk[n];
15396516239fSHong Zhang     while (prow < i) {
15406516239fSHong Zhang       nnz      = bdiag[prow];
15416516239fSHong Zhang       cols     = bj_ptr[prow] + nnz + 1;
15426516239fSHong Zhang       cols_lvl = bjlvl_ptr[prow] + nnz + 1;
15436516239fSHong Zhang       nnz      = bi[prow+1] - bi[prow] - nnz - 1;
15446516239fSHong Zhang       ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr);
15456516239fSHong Zhang       nzi += nlnk;
15466516239fSHong Zhang       prow = lnk[prow];
15476516239fSHong Zhang       nzbd++;
15486516239fSHong Zhang     }
15496516239fSHong Zhang     bdiag[i] = nzbd;
15506516239fSHong Zhang     bi[i+1]  = bi[i] + nzi;
15516516239fSHong Zhang 
15526516239fSHong Zhang     /* if free space is not available, make more free space */
15536516239fSHong Zhang     if (current_space->local_remaining<nzi) {
15546516239fSHong Zhang       nnz = 2*nzi*(n - i); /* estimated and max additional space needed */
15556516239fSHong Zhang       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
15566516239fSHong Zhang       ierr = PetscFreeSpaceGet(nnz,&current_space_lvl);CHKERRQ(ierr);
15576516239fSHong Zhang       reallocs++;
15586516239fSHong Zhang     }
15596516239fSHong Zhang 
15606516239fSHong Zhang     /* copy data into free_space and free_space_lvl, then initialize lnk */
15616516239fSHong Zhang     ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
15626516239fSHong Zhang     bj_ptr[i]    = current_space->array;
15636516239fSHong Zhang     bjlvl_ptr[i] = current_space_lvl->array;
15646516239fSHong Zhang 
15656516239fSHong Zhang     /* make sure the active row i has diagonal entry */
15666516239fSHong Zhang     if (*(bj_ptr[i]+bdiag[i]) != i) {
15676516239fSHong Zhang       SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,"Row %D has missing diagonal in factored matrix\n\
15686516239fSHong Zhang     try running with -pc_factor_nonzeros_along_diagonal or -pc_factor_diagonal_fill",i);
15696516239fSHong Zhang     }
15706516239fSHong Zhang 
15716516239fSHong Zhang     current_space->array           += nzi;
15726516239fSHong Zhang     current_space->local_used      += nzi;
15736516239fSHong Zhang     current_space->local_remaining -= nzi;
15746516239fSHong Zhang     current_space_lvl->array           += nzi;
15756516239fSHong Zhang     current_space_lvl->local_used      += nzi;
15766516239fSHong Zhang     current_space_lvl->local_remaining -= nzi;
15776516239fSHong Zhang   }
15786516239fSHong Zhang 
15796516239fSHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
15806516239fSHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
15816516239fSHong Zhang 
15826516239fSHong Zhang   /* destroy list of free space and other temporary arrays */
15836516239fSHong Zhang   ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
15846516239fSHong Zhang 
15856516239fSHong Zhang   /* copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */
1586783ef271SHong Zhang   ierr = PetscFreeSpaceContiguous_LU(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr);
15876516239fSHong Zhang 
15886516239fSHong Zhang   ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
15896516239fSHong Zhang   ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
15906516239fSHong Zhang   ierr = PetscFree(bj_ptr);CHKERRQ(ierr);
15916516239fSHong Zhang 
15926516239fSHong Zhang #if defined(PETSC_USE_INFO)
15936516239fSHong Zhang   {
15946516239fSHong Zhang     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
15956516239fSHong Zhang     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr);
15966516239fSHong Zhang     ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
15976516239fSHong Zhang     ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr);
15986516239fSHong Zhang     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
15996516239fSHong Zhang     if (diagonal_fill) {
16006516239fSHong Zhang       ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr);
16016516239fSHong Zhang     }
16026516239fSHong Zhang   }
16036516239fSHong Zhang #endif
16046516239fSHong Zhang 
16056516239fSHong Zhang   /* put together the new matrix */
16066516239fSHong Zhang   ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
16076516239fSHong Zhang   ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr);
16086516239fSHong Zhang   b = (Mat_SeqAIJ*)(fact)->data;
16096516239fSHong Zhang   b->free_a       = PETSC_TRUE;
16106516239fSHong Zhang   b->free_ij      = PETSC_TRUE;
16116516239fSHong Zhang   b->singlemalloc = PETSC_FALSE;
16126516239fSHong Zhang   ierr = PetscMalloc( (bi[2*n+1] )*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);
16136516239fSHong Zhang   b->j          = bj;
16146516239fSHong Zhang   b->i          = bi;
16156516239fSHong Zhang   b->diag       = bdiag;
16166516239fSHong Zhang   b->ilen       = 0;
16176516239fSHong Zhang   b->imax       = 0;
16186516239fSHong Zhang   b->row        = isrow;
16196516239fSHong Zhang   b->col        = iscol;
16206516239fSHong Zhang   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
16216516239fSHong Zhang   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
16226516239fSHong Zhang   b->icol       = isicol;
16236516239fSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
16246516239fSHong Zhang   /* In b structure:  Free imax, ilen, old a, old j.
16256516239fSHong Zhang      Allocate bdiag, solve_work, new a, new j */
16266516239fSHong Zhang   ierr = PetscLogObjectMemory(fact,bi[2*n+1] * (sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
16276516239fSHong Zhang   b->maxnz = b->nz = bi[2*n+1] ;
16286516239fSHong Zhang   (fact)->info.factor_mallocs    = reallocs;
16296516239fSHong Zhang   (fact)->info.fill_ratio_given  = f;
16306516239fSHong Zhang   (fact)->info.fill_ratio_needed = ((PetscReal)bi[2*n+1])/((PetscReal)ai[n]);
16316516239fSHong Zhang   (fact)->ops->lufactornumeric   = MatLUFactorNumeric_SeqAIJ_newdatastruct;
16326516239fSHong Zhang   /* ierr = MatILUFactorSymbolic_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr); */
16336516239fSHong Zhang   PetscFunctionReturn(0);
16346516239fSHong Zhang }
16356516239fSHong Zhang 
16366516239fSHong Zhang #undef __FUNCT__
16376516239fSHong Zhang #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ"
16386516239fSHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
16396516239fSHong Zhang {
16406516239fSHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
16416516239fSHong Zhang   IS                 isicol;
16426516239fSHong Zhang   PetscErrorCode     ierr;
16436516239fSHong Zhang   const PetscInt     *r,*ic;
16446516239fSHong Zhang   PetscInt           n=A->rmap->n,*ai=a->i,*aj=a->j,d;
16456516239fSHong Zhang   PetscInt           *bi,*cols,nnz,*cols_lvl;
16466516239fSHong Zhang   PetscInt           *bdiag,prow,fm,nzbd,reallocs=0,dcount=0;
16476516239fSHong Zhang   PetscInt           i,levels,diagonal_fill;
16486516239fSHong Zhang   PetscTruth         col_identity,row_identity;
16496516239fSHong Zhang   PetscReal          f;
16506516239fSHong Zhang   PetscInt           nlnk,*lnk,*lnk_lvl=PETSC_NULL;
16516516239fSHong Zhang   PetscBT            lnkbt;
16526516239fSHong Zhang   PetscInt           nzi,*bj,**bj_ptr,**bjlvl_ptr;
16536516239fSHong Zhang   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
16546516239fSHong Zhang   PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL;
16556516239fSHong Zhang   PetscTruth         missing;
16566516239fSHong Zhang   PetscTruth         newdatastruct=PETSC_FALSE;
16576516239fSHong Zhang 
16586516239fSHong Zhang   PetscFunctionBegin;
16596516239fSHong Zhang   ierr = PetscOptionsGetTruth(PETSC_NULL,"-ilu_new",&newdatastruct,PETSC_NULL);CHKERRQ(ierr);
16606516239fSHong Zhang   if (newdatastruct){
16616516239fSHong Zhang     ierr = MatILUFactorSymbolic_SeqAIJ_newdatastruct(fact,A,isrow,iscol,info);CHKERRQ(ierr);
16626516239fSHong Zhang     PetscFunctionReturn(0);
16636516239fSHong Zhang   }
16646516239fSHong Zhang 
16656516239fSHong 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);
16666516239fSHong Zhang   f             = info->fill;
16676516239fSHong Zhang   levels        = (PetscInt)info->levels;
16686516239fSHong Zhang   diagonal_fill = (PetscInt)info->diagonal_fill;
16696516239fSHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
16706516239fSHong Zhang 
16716516239fSHong Zhang   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
16726516239fSHong Zhang   ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr);
16736516239fSHong Zhang   if (!levels && row_identity && col_identity) { /* special case: ilu(0) with natural ordering */
1674f77e22a1SHong Zhang     ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_TRUE);CHKERRQ(ierr);
1675c9c72f8fSHong Zhang     (fact)->ops->lufactornumeric =  MatLUFactorNumeric_SeqAIJ;
1676c9c72f8fSHong Zhang 
1677719d5645SBarry Smith     fact->factor = MAT_FACTOR_ILU;
1678719d5645SBarry Smith     (fact)->info.factor_mallocs    = 0;
1679719d5645SBarry Smith     (fact)->info.fill_ratio_given  = info->fill;
1680719d5645SBarry Smith     (fact)->info.fill_ratio_needed = 1.0;
1681719d5645SBarry Smith     b               = (Mat_SeqAIJ*)(fact)->data;
16828ef3462fSBarry Smith     ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr);
168309f38230SBarry Smith     if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d);
168408480c60SBarry Smith     b->row              = isrow;
168508480c60SBarry Smith     b->col              = iscol;
168682bf6240SBarry Smith     b->icol             = isicol;
1687719d5645SBarry Smith     ierr                = PetscMalloc(((fact)->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
1688c38d4ed2SBarry Smith     ierr                = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
1689c38d4ed2SBarry Smith     ierr                = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
1690719d5645SBarry Smith     ierr = MatILUFactorSymbolic_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr);
16913a40ed3dSBarry Smith     PetscFunctionReturn(0);
169208480c60SBarry Smith   }
169308480c60SBarry Smith 
1694898183ecSLois Curfman McInnes   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
1695898183ecSLois Curfman McInnes   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
16969e25ed09SBarry Smith 
16979e25ed09SBarry Smith   /* get new row pointers */
16985a8e39fbSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
16995a8e39fbSHong Zhang   bi[0] = 0;
17005a8e39fbSHong Zhang   /* bdiag is location of diagonal in factor */
17015a8e39fbSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
17025a8e39fbSHong Zhang   bdiag[0]  = 0;
17036cf997b0SBarry Smith 
17045a8e39fbSHong Zhang   ierr = PetscMalloc((2*n+1)*sizeof(PetscInt**),&bj_ptr);CHKERRQ(ierr);
17055a8e39fbSHong Zhang   bjlvl_ptr = (PetscInt**)(bj_ptr + n);
17065a8e39fbSHong Zhang 
17075a8e39fbSHong Zhang   /* create a linked list for storing column indices of the active row */
17085a8e39fbSHong Zhang   nlnk = n + 1;
17095a8e39fbSHong Zhang   ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
17105a8e39fbSHong Zhang 
17115a8e39fbSHong Zhang   /* initial FreeSpace size is f*(ai[n]+1) */
1712a1a86e44SBarry Smith   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr);
17135a8e39fbSHong Zhang   current_space = free_space;
1714a1a86e44SBarry Smith   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr);
17155a8e39fbSHong Zhang   current_space_lvl = free_space_lvl;
17165a8e39fbSHong Zhang 
17175a8e39fbSHong Zhang   for (i=0; i<n; i++) {
17185a8e39fbSHong Zhang     nzi = 0;
17196cf997b0SBarry Smith     /* copy current row into linked list */
17205a8e39fbSHong Zhang     nnz  = ai[r[i]+1] - ai[r[i]];
17213b4a8b6dSBarry 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);
17225a8e39fbSHong Zhang     cols = aj + ai[r[i]];
17235a8e39fbSHong Zhang     lnk[i] = -1; /* marker to indicate if diagonal exists */
17245a8e39fbSHong Zhang     ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
17255a8e39fbSHong Zhang     nzi += nlnk;
17266cf997b0SBarry Smith 
17276cf997b0SBarry Smith     /* make sure diagonal entry is included */
17285a8e39fbSHong Zhang     if (diagonal_fill && lnk[i] == -1) {
17296cf997b0SBarry Smith       fm = n;
17305a8e39fbSHong Zhang       while (lnk[fm] < i) fm = lnk[fm];
17315a8e39fbSHong Zhang       lnk[i]     = lnk[fm]; /* insert diagonal into linked list */
17325a8e39fbSHong Zhang       lnk[fm]    = i;
17335a8e39fbSHong Zhang       lnk_lvl[i] = 0;
17345a8e39fbSHong Zhang       nzi++; dcount++;
17356cf997b0SBarry Smith     }
17366cf997b0SBarry Smith 
17375a8e39fbSHong Zhang     /* add pivot rows into the active row */
17385a8e39fbSHong Zhang     nzbd = 0;
17395a8e39fbSHong Zhang     prow = lnk[n];
17405a8e39fbSHong Zhang     while (prow < i) {
17415a8e39fbSHong Zhang       nnz      = bdiag[prow];
17425a8e39fbSHong Zhang       cols     = bj_ptr[prow] + nnz + 1;
17435a8e39fbSHong Zhang       cols_lvl = bjlvl_ptr[prow] + nnz + 1;
17445a8e39fbSHong Zhang       nnz      = bi[prow+1] - bi[prow] - nnz - 1;
17455a8e39fbSHong Zhang       ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr);
17465a8e39fbSHong Zhang       nzi += nlnk;
17475a8e39fbSHong Zhang       prow = lnk[prow];
17485a8e39fbSHong Zhang       nzbd++;
174956beaf04SBarry Smith     }
17505a8e39fbSHong Zhang     bdiag[i] = nzbd;
17515a8e39fbSHong Zhang     bi[i+1]  = bi[i] + nzi;
1752ecf371e4SBarry Smith 
17535a8e39fbSHong Zhang     /* if free space is not available, make more free space */
17545a8e39fbSHong Zhang     if (current_space->local_remaining<nzi) {
17555a8e39fbSHong Zhang       nnz = nzi*(n - i); /* estimated and max additional space needed */
1756a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
1757a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(nnz,&current_space_lvl);CHKERRQ(ierr);
17585a8e39fbSHong Zhang       reallocs++;
17595a8e39fbSHong Zhang     }
1760ecf371e4SBarry Smith 
17615a8e39fbSHong Zhang     /* copy data into free_space and free_space_lvl, then initialize lnk */
17625a8e39fbSHong Zhang     ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
17635a8e39fbSHong Zhang     bj_ptr[i]    = current_space->array;
17645a8e39fbSHong Zhang     bjlvl_ptr[i] = current_space_lvl->array;
17655a8e39fbSHong Zhang 
17665a8e39fbSHong Zhang     /* make sure the active row i has diagonal entry */
17675a8e39fbSHong Zhang     if (*(bj_ptr[i]+bdiag[i]) != i) {
176877431f27SBarry Smith       SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,"Row %D has missing diagonal in factored matrix\n\
1769d7ee0231SBarry Smith     try running with -pc_factor_nonzeros_along_diagonal or -pc_factor_diagonal_fill",i);
17706cf997b0SBarry Smith     }
17715a8e39fbSHong Zhang 
17725a8e39fbSHong Zhang     current_space->array           += nzi;
17735a8e39fbSHong Zhang     current_space->local_used      += nzi;
17745a8e39fbSHong Zhang     current_space->local_remaining -= nzi;
17755a8e39fbSHong Zhang     current_space_lvl->array           += nzi;
17765a8e39fbSHong Zhang     current_space_lvl->local_used      += nzi;
17775a8e39fbSHong Zhang     current_space_lvl->local_remaining -= nzi;
17789e25ed09SBarry Smith   }
17795a8e39fbSHong Zhang 
1780898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
1781898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
17825a8e39fbSHong Zhang 
17835a8e39fbSHong Zhang   /* destroy list of free space and other temporary arrays */
17845a8e39fbSHong Zhang   ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
17856516239fSHong Zhang   ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); /* copy free_space -> bj */
17865a8e39fbSHong Zhang   ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
1787a1a86e44SBarry Smith   ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
17885a8e39fbSHong Zhang   ierr = PetscFree(bj_ptr);CHKERRQ(ierr);
17899e25ed09SBarry Smith 
17906cf91177SBarry Smith #if defined(PETSC_USE_INFO)
1791f64065bbSBarry Smith   {
17925a8e39fbSHong Zhang     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
1793ae15b995SBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr);
1794ae15b995SBarry Smith     ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
1795ae15b995SBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr);
1796ae15b995SBarry Smith     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
1797335d9088SBarry Smith     if (diagonal_fill) {
1798ae15b995SBarry Smith       ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr);
1799335d9088SBarry Smith     }
1800f64065bbSBarry Smith   }
180163ba0a88SBarry Smith #endif
1802bbb6d6a8SBarry Smith 
18039e25ed09SBarry Smith   /* put together the new matrix */
180407dbf95fSBarry Smith   ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
1805719d5645SBarry Smith   ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr);
1806719d5645SBarry Smith   b = (Mat_SeqAIJ*)(fact)->data;
1807e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
1808e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
18097c922b88SBarry Smith   b->singlemalloc = PETSC_FALSE;
1810052f0c41SBarry Smith   ierr = PetscMalloc( (bi[n] )*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);
18115a8e39fbSHong Zhang   b->j          = bj;
18125a8e39fbSHong Zhang   b->i          = bi;
18135a8e39fbSHong Zhang   for (i=0; i<n; i++) bdiag[i] += bi[i];
18145a8e39fbSHong Zhang   b->diag       = bdiag;
1815416022c9SBarry Smith   b->ilen       = 0;
1816416022c9SBarry Smith   b->imax       = 0;
1817416022c9SBarry Smith   b->row        = isrow;
1818416022c9SBarry Smith   b->col        = iscol;
1819c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
1820c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
182182bf6240SBarry Smith   b->icol       = isicol;
182287828ca2SBarry Smith   ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
1823416022c9SBarry Smith   /* In b structure:  Free imax, ilen, old a, old j.
18245a8e39fbSHong Zhang      Allocate bdiag, solve_work, new a, new j */
1825719d5645SBarry Smith   ierr = PetscLogObjectMemory(fact,(bi[n]-n) * (sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
18265a8e39fbSHong Zhang   b->maxnz             = b->nz = bi[n] ;
1827719d5645SBarry Smith   (fact)->info.factor_mallocs    = reallocs;
1828719d5645SBarry Smith   (fact)->info.fill_ratio_given  = f;
1829719d5645SBarry Smith   (fact)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]);
1830719d5645SBarry Smith   (fact)->ops->lufactornumeric =  MatLUFactorNumeric_SeqAIJ;
1831719d5645SBarry Smith   ierr = MatILUFactorSymbolic_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr);
18323a40ed3dSBarry Smith   PetscFunctionReturn(0);
18339e25ed09SBarry Smith }
1834d5d45c9bSBarry Smith 
1835a6175056SHong Zhang #undef __FUNCT__
18365f44c854SHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ_newdatastruct"
18375f44c854SHong Zhang PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ_newdatastruct(Mat B,Mat A,const MatFactorInfo *info)
18385f44c854SHong Zhang {
18395f44c854SHong Zhang   Mat            C = B;
18405f44c854SHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
18415f44c854SHong Zhang   Mat_SeqSBAIJ   *b=(Mat_SeqSBAIJ*)C->data;
18425f44c854SHong Zhang   IS             ip=b->row,iip = b->icol;
18435f44c854SHong Zhang   PetscErrorCode ierr;
18445f44c854SHong Zhang   const PetscInt *rip,*riip;
18455f44c854SHong Zhang   PetscInt       i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bdiag=b->diag,*bjtmp;
18465f44c854SHong Zhang   PetscInt       *ai=a->i,*aj=a->j;
18475f44c854SHong Zhang   PetscInt       k,jmin,jmax,*c2r,*il,col,nexti,ili,nz;
18485f44c854SHong Zhang   MatScalar      *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi;
18495f44c854SHong Zhang   PetscReal      zeropivot,shiftnz;
18505f44c854SHong Zhang   PetscReal      shiftpd;
18515f44c854SHong Zhang   PetscTruth     perm_identity;
18525f44c854SHong Zhang 
18535f44c854SHong Zhang   PetscFunctionBegin;
18545f44c854SHong Zhang 
18555f44c854SHong Zhang   shiftnz   = info->shiftnz;
18565f44c854SHong Zhang   shiftpd   = info->shiftpd;
18575f44c854SHong Zhang   zeropivot = info->zeropivot;
18585f44c854SHong Zhang 
18595f44c854SHong Zhang   ierr  = ISGetIndices(ip,&rip);CHKERRQ(ierr);
18605f44c854SHong Zhang   ierr  = ISGetIndices(iip,&riip);CHKERRQ(ierr);
18615f44c854SHong Zhang 
18625f44c854SHong Zhang   /* allocate working arrays
18635f44c854SHong Zhang      c2r: linked list, keep track of pivot rows for a given column. c2r[col]: head of the list for a given col
18645f44c854SHong 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
18655f44c854SHong Zhang   */
18665f44c854SHong Zhang   nz   = (2*mbs+1)*sizeof(PetscInt)+mbs*sizeof(MatScalar);
18675f44c854SHong Zhang   ierr = PetscMalloc(nz,&il);CHKERRQ(ierr);
18685f44c854SHong Zhang   c2r  = il + mbs;
18695f44c854SHong Zhang   rtmp = (MatScalar*)(c2r + mbs);
18705f44c854SHong Zhang 
18715f44c854SHong Zhang   for (i=0; i<mbs; i++) {
18725f44c854SHong Zhang     c2r[i] = mbs;
18735f44c854SHong Zhang   }
18745f44c854SHong Zhang   il[0] = 0;
18755f44c854SHong Zhang 
18765f44c854SHong Zhang   for (k = 0; k<mbs; k++){
18775f44c854SHong Zhang     /* zero rtmp */
18785f44c854SHong Zhang     nz = bi[k+1] - bi[k];
18795f44c854SHong Zhang     bjtmp = bj + bi[k];
18805f44c854SHong Zhang     for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
18815f44c854SHong Zhang 
18825f44c854SHong Zhang     /* load in initial unfactored row */
18835f44c854SHong Zhang     bval = ba + bi[k];
18845f44c854SHong Zhang     jmin = ai[rip[k]]; jmax = ai[rip[k]+1];
18855f44c854SHong Zhang     for (j = jmin; j < jmax; j++){
18865f44c854SHong Zhang       col = riip[aj[j]];
18875f44c854SHong Zhang       if (col >= k){ /* only take upper triangular entry */
18885f44c854SHong Zhang         rtmp[col] = aa[j];
18895f44c854SHong Zhang         *bval++   = 0.0; /* for in-place factorization */
18905f44c854SHong Zhang       }
18915f44c854SHong Zhang     }
18925f44c854SHong Zhang     /* shift the diagonal of the matrix */
18935f44c854SHong Zhang 
18945f44c854SHong Zhang     /* modify k-th row by adding in those rows i with U(i,k)!=0 */
18955f44c854SHong Zhang     dk = rtmp[k];
18965f44c854SHong Zhang     i  = c2r[k]; /* first row to be added to k_th row  */
18975f44c854SHong Zhang 
18985f44c854SHong Zhang     while (i < k){
18995f44c854SHong Zhang       nexti = c2r[i]; /* next row to be added to k_th row */
19005f44c854SHong Zhang 
19015f44c854SHong Zhang       /* compute multiplier, update diag(k) and U(i,k) */
19025f44c854SHong Zhang       ili   = il[i];  /* index of first nonzero element in U(i,k:bms-1) */
19035f44c854SHong Zhang       uikdi = - ba[ili]*ba[bdiag[i]];  /* diagonal(k) */
19045f44c854SHong Zhang       dk   += uikdi*ba[ili]; /* update diag[k] */
19055f44c854SHong Zhang       ba[ili] = uikdi; /* -U(i,k) */
19065f44c854SHong Zhang 
19075f44c854SHong Zhang       /* add multiple of row i to k-th row */
19085f44c854SHong Zhang       jmin = ili + 1; jmax = bi[i+1];
19095f44c854SHong Zhang       if (jmin < jmax){
19105f44c854SHong Zhang         for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j];
19115f44c854SHong Zhang         /* update il and c2r for row i */
19125f44c854SHong Zhang         il[i] = jmin;
19135f44c854SHong Zhang         j = bj[jmin]; c2r[i] = c2r[j]; c2r[j] = i;
19145f44c854SHong Zhang       }
19155f44c854SHong Zhang       i = nexti;
19165f44c854SHong Zhang     }
19175f44c854SHong Zhang 
19185f44c854SHong Zhang     /* copy data into U(k,:) */
19195f44c854SHong Zhang     ba[bdiag[k]] = 1.0/dk; /* U(k,k) */
19205f44c854SHong Zhang     jmin = bi[k]; jmax = bi[k+1]-1;
19215f44c854SHong Zhang     if (jmin < jmax) {
19225f44c854SHong Zhang       for (j=jmin; j<jmax; j++){
19235f44c854SHong Zhang         col = bj[j]; ba[j] = rtmp[col];
19245f44c854SHong Zhang       }
19255f44c854SHong Zhang       /* add the k-th row into il and c2r */
19265f44c854SHong Zhang       il[k] = jmin;
19275f44c854SHong Zhang       i = bj[jmin]; c2r[k] = c2r[i]; c2r[i] = k;
19285f44c854SHong Zhang     }
19295f44c854SHong Zhang   }
19305f44c854SHong Zhang 
19315f44c854SHong Zhang   ierr = PetscFree(il);CHKERRQ(ierr);
19325f44c854SHong Zhang   ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr);
19335f44c854SHong Zhang   ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr);
19345f44c854SHong Zhang 
19355f44c854SHong Zhang   ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr);
19365f44c854SHong Zhang   if (perm_identity){
19375f44c854SHong Zhang     (B)->ops->solve           = MatSolve_SeqSBAIJ_1_NaturalOrdering_newdatastruct;
19385f44c854SHong Zhang     (B)->ops->solvetranspose  = 0;
19395f44c854SHong Zhang     (B)->ops->forwardsolve    = 0;
19405f44c854SHong Zhang     (B)->ops->backwardsolve   = 0;
19415f44c854SHong Zhang   } else {
1942783ef271SHong Zhang     (B)->ops->solve           = MatSolve_SeqSBAIJ_1_newdatastruct;
19435f44c854SHong Zhang     (B)->ops->solvetranspose  = 0;
19445f44c854SHong Zhang     (B)->ops->forwardsolve    = 0;
19455f44c854SHong Zhang     (B)->ops->backwardsolve   = 0;
19465f44c854SHong Zhang   }
19475f44c854SHong Zhang 
19485f44c854SHong Zhang   C->assembled    = PETSC_TRUE;
19495f44c854SHong Zhang   C->preallocated = PETSC_TRUE;
19505f44c854SHong Zhang   ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr);
19515f44c854SHong Zhang   PetscFunctionReturn(0);
19525f44c854SHong Zhang }
19535f44c854SHong Zhang 
19545f44c854SHong Zhang #undef __FUNCT__
1955a6175056SHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ"
19560481f469SBarry Smith PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info)
1957a6175056SHong Zhang {
1958719d5645SBarry Smith   Mat            C = B;
19590968510aSHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
19602ed133dbSHong Zhang   Mat_SeqSBAIJ   *b=(Mat_SeqSBAIJ*)C->data;
19619bfd6278SHong Zhang   IS             ip=b->row,iip = b->icol;
19622ed133dbSHong Zhang   PetscErrorCode ierr;
19635d0c19d7SBarry Smith   const PetscInt *rip,*riip;
19645d0c19d7SBarry Smith   PetscInt       i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bcol;
19652ed133dbSHong Zhang   PetscInt       *ai=a->i,*aj=a->j;
19662ed133dbSHong Zhang   PetscInt       k,jmin,jmax,*jl,*il,col,nexti,ili,nz;
1967622e2028SHong Zhang   MatScalar      *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi;
1968540703acSHong Zhang   PetscReal      zeropivot,rs,shiftnz;
1969fbf22428SSatish Balay   PetscReal      shiftpd;
1970540703acSHong Zhang   ChShift_Ctx    sctx;
1971540703acSHong Zhang   PetscInt       newshift;
1972db4efbfdSBarry Smith   PetscTruth     perm_identity;
1973d5d45c9bSBarry Smith 
1974a6175056SHong Zhang   PetscFunctionBegin;
1975540703acSHong Zhang   shiftnz   = info->shiftnz;
1976540703acSHong Zhang   shiftpd   = info->shiftpd;
1977ee45ca4aSHong Zhang   zeropivot = info->zeropivot;
1978ee45ca4aSHong Zhang 
19792ed133dbSHong Zhang   ierr  = ISGetIndices(ip,&rip);CHKERRQ(ierr);
19809bfd6278SHong Zhang   ierr  = ISGetIndices(iip,&riip);CHKERRQ(ierr);
19812ed133dbSHong Zhang 
19822ed133dbSHong Zhang   /* initialization */
19832ed133dbSHong Zhang   nz   = (2*mbs+1)*sizeof(PetscInt)+mbs*sizeof(MatScalar);
19842ed133dbSHong Zhang   ierr = PetscMalloc(nz,&il);CHKERRQ(ierr);
19852ed133dbSHong Zhang   jl   = il + mbs;
19862ed133dbSHong Zhang   rtmp = (MatScalar*)(jl + mbs);
19872ed133dbSHong Zhang 
1988540703acSHong Zhang   sctx.shift_amount = 0;
1989540703acSHong Zhang   sctx.nshift       = 0;
19902ed133dbSHong Zhang   do {
1991540703acSHong Zhang     sctx.chshift = PETSC_FALSE;
19922ed133dbSHong Zhang     for (i=0; i<mbs; i++) {
19932ed133dbSHong Zhang       rtmp[i] = 0.0; jl[i] = mbs; il[0] = 0;
1994a6175056SHong Zhang     }
19952ed133dbSHong Zhang 
19962ed133dbSHong Zhang     for (k = 0; k<mbs; k++){
1997c05c3958SHong Zhang       bval = ba + bi[k];
19982ed133dbSHong Zhang       /* initialize k-th row by the perm[k]-th row of A */
19992ed133dbSHong Zhang       jmin = ai[rip[k]]; jmax = ai[rip[k]+1];
20002ed133dbSHong Zhang       for (j = jmin; j < jmax; j++){
20019bfd6278SHong Zhang         col = riip[aj[j]];
20022ed133dbSHong Zhang         if (col >= k){ /* only take upper triangular entry */
20032ed133dbSHong Zhang           rtmp[col] = aa[j];
2004c05c3958SHong Zhang           *bval++  = 0.0; /* for in-place factorization */
20052ed133dbSHong Zhang         }
20062ed133dbSHong Zhang       }
200739e3d630SHong Zhang       /* shift the diagonal of the matrix */
2008540703acSHong Zhang       if (sctx.nshift) rtmp[k] += sctx.shift_amount;
20092ed133dbSHong Zhang 
20102ed133dbSHong Zhang       /* modify k-th row by adding in those rows i with U(i,k)!=0 */
20112ed133dbSHong Zhang       dk = rtmp[k];
20122ed133dbSHong Zhang       i = jl[k]; /* first row to be added to k_th row  */
20132ed133dbSHong Zhang 
20142ed133dbSHong Zhang       while (i < k){
20152ed133dbSHong Zhang         nexti = jl[i]; /* next row to be added to k_th row */
20162ed133dbSHong Zhang 
20172ed133dbSHong Zhang         /* compute multiplier, update diag(k) and U(i,k) */
20182ed133dbSHong Zhang         ili = il[i];  /* index of first nonzero element in U(i,k:bms-1) */
20192ed133dbSHong Zhang         uikdi = - ba[ili]*ba[bi[i]];  /* diagonal(k) */
20202ed133dbSHong Zhang         dk += uikdi*ba[ili];
20212ed133dbSHong Zhang         ba[ili] = uikdi; /* -U(i,k) */
20222ed133dbSHong Zhang 
20232ed133dbSHong Zhang         /* add multiple of row i to k-th row */
20242ed133dbSHong Zhang         jmin = ili + 1; jmax = bi[i+1];
20252ed133dbSHong Zhang         if (jmin < jmax){
20262ed133dbSHong Zhang           for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j];
20272ed133dbSHong Zhang           /* update il and jl for row i */
20282ed133dbSHong Zhang           il[i] = jmin;
20292ed133dbSHong Zhang           j = bj[jmin]; jl[i] = jl[j]; jl[j] = i;
20302ed133dbSHong Zhang         }
20312ed133dbSHong Zhang         i = nexti;
20322ed133dbSHong Zhang       }
20332ed133dbSHong Zhang 
20340697b6caSHong Zhang       /* shift the diagonals when zero pivot is detected */
20350697b6caSHong Zhang       /* compute rs=sum of abs(off-diagonal) */
20360697b6caSHong Zhang       rs   = 0.0;
20373c9e8598SHong Zhang       jmin = bi[k]+1;
20383c9e8598SHong Zhang       nz   = bi[k+1] - jmin;
20393c9e8598SHong Zhang       bcol = bj + jmin;
204004fbf559SBarry Smith       for (j=0; j<nz; j++) {
204104fbf559SBarry Smith         rs += PetscAbsScalar(rtmp[bcol[j]]);
20423c9e8598SHong Zhang       }
2043540703acSHong Zhang 
2044540703acSHong Zhang       sctx.rs = rs;
2045540703acSHong Zhang       sctx.pv = dk;
20465b5bc046SBarry Smith       ierr = MatCholeskyCheckShift_inline(info,sctx,k,newshift);CHKERRQ(ierr);
2047117f1891Stmunson 
2048117f1891Stmunson       if (newshift == 1) {
2049117f1891Stmunson         if (!sctx.shift_amount) {
2050117f1891Stmunson           sctx.shift_amount = 1e-5;
2051117f1891Stmunson         }
2052117f1891Stmunson         break;
2053117f1891Stmunson       }
20543c9e8598SHong Zhang 
20553c9e8598SHong Zhang       /* copy data into U(k,:) */
205639e3d630SHong Zhang       ba[bi[k]] = 1.0/dk; /* U(k,k) */
205739e3d630SHong Zhang       jmin = bi[k]+1; jmax = bi[k+1];
205839e3d630SHong Zhang       if (jmin < jmax) {
205939e3d630SHong Zhang         for (j=jmin; j<jmax; j++){
206039e3d630SHong Zhang           col = bj[j]; ba[j] = rtmp[col]; rtmp[col] = 0.0;
20613c9e8598SHong Zhang         }
206239e3d630SHong Zhang         /* add the k-th row into il and jl */
20633c9e8598SHong Zhang         il[k] = jmin;
20643c9e8598SHong Zhang         i = bj[jmin]; jl[k] = jl[i]; jl[i] = k;
20653c9e8598SHong Zhang       }
20663c9e8598SHong Zhang     }
2067540703acSHong Zhang   } while (sctx.chshift);
20683c9e8598SHong Zhang   ierr = PetscFree(il);CHKERRQ(ierr);
20693c9e8598SHong Zhang 
207039e3d630SHong Zhang   ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr);
20719bfd6278SHong Zhang   ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr);
2072db4efbfdSBarry Smith 
2073db4efbfdSBarry Smith   ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr);
2074db4efbfdSBarry Smith   if (perm_identity){
2075719d5645SBarry Smith     (B)->ops->solve           = MatSolve_SeqSBAIJ_1_NaturalOrdering;
2076719d5645SBarry Smith     (B)->ops->solvetranspose  = MatSolve_SeqSBAIJ_1_NaturalOrdering;
2077719d5645SBarry Smith     (B)->ops->forwardsolve    = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering;
2078719d5645SBarry Smith     (B)->ops->backwardsolve   = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering;
2079db4efbfdSBarry Smith   } else {
2080719d5645SBarry Smith     (B)->ops->solve           = MatSolve_SeqSBAIJ_1;
2081719d5645SBarry Smith     (B)->ops->solvetranspose  = MatSolve_SeqSBAIJ_1;
2082719d5645SBarry Smith     (B)->ops->forwardsolve    = MatForwardSolve_SeqSBAIJ_1;
2083719d5645SBarry Smith     (B)->ops->backwardsolve   = MatBackwardSolve_SeqSBAIJ_1;
2084db4efbfdSBarry Smith   }
2085db4efbfdSBarry Smith 
20863c9e8598SHong Zhang   C->assembled    = PETSC_TRUE;
20873c9e8598SHong Zhang   C->preallocated = PETSC_TRUE;
2088d0f46423SBarry Smith   ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr);
2089540703acSHong Zhang   if (sctx.nshift){
2090540703acSHong Zhang     if (shiftnz) {
20911e2582c4SBarry Smith       ierr = PetscInfo2(A,"number of shiftnz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
2092540703acSHong Zhang     } else if (shiftpd) {
20931e2582c4SBarry Smith       ierr = PetscInfo2(A,"number of shiftpd tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
20940697b6caSHong Zhang     }
20953c9e8598SHong Zhang   }
20963c9e8598SHong Zhang   PetscFunctionReturn(0);
20973c9e8598SHong Zhang }
2098a6175056SHong Zhang 
2099a6175056SHong Zhang #undef __FUNCT__
21005f44c854SHong Zhang #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ_newdatastruct"
21015f44c854SHong Zhang PetscErrorCode MatICCFactorSymbolic_SeqAIJ_newdatastruct(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
2102a6175056SHong Zhang {
21030968510aSHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
2104ed59401aSHong Zhang   Mat_SeqSBAIJ       *b;
2105dfbe8321SBarry Smith   PetscErrorCode     ierr;
210658ebbce7SBarry Smith   PetscTruth         perm_identity,missing;
21075f44c854SHong Zhang   PetscInt           reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag;
21085d0c19d7SBarry Smith   const PetscInt     *rip,*riip;
2109ed59401aSHong Zhang   PetscInt           jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow;
211058ebbce7SBarry Smith   PetscInt           nlnk,*lnk,*lnk_lvl=PETSC_NULL,d;
21115a8e39fbSHong Zhang   PetscInt           ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr;
2112ed59401aSHong Zhang   PetscReal          fill=info->fill,levels=info->levels;
2113a1a86e44SBarry Smith   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
2114a1a86e44SBarry Smith   PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL;
21157a48dd6fSHong Zhang   PetscBT            lnkbt;
2116b635d36bSHong Zhang   IS                 iperm;
2117a6175056SHong Zhang 
2118a6175056SHong Zhang   PetscFunctionBegin;
2119d0f46423SBarry 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);
212058ebbce7SBarry Smith   ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr);
212158ebbce7SBarry Smith   if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d);
2122653a6975SHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
2123b635d36bSHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
21247a48dd6fSHong Zhang 
212539e3d630SHong Zhang   ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr);
21265f44c854SHong Zhang   ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr);
212739e3d630SHong Zhang   ui[0] = 0;
212839e3d630SHong Zhang 
2129b635d36bSHong Zhang   /* ICC(0) without matrix ordering: simply copies fill pattern */
2130622e2028SHong Zhang   if (!levels && perm_identity) {
213158ebbce7SBarry Smith 
2132ed59401aSHong Zhang     for (i=0; i<am; i++) {
213339e3d630SHong Zhang       ui[i+1]  = ui[i] + ai[i+1] - a->diag[i];
21345f44c854SHong Zhang       udiag[i] = ui[i+1] - 1; /* points to the last entry of U(i,:) */
21355f44c854SHong Zhang     }
21365f44c854SHong Zhang     ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
21375f44c854SHong Zhang     cols = uj;
21385f44c854SHong Zhang     for (i=0; i<am; i++) {
21395f44c854SHong Zhang       aj    = a->j + a->diag[i] + 1; /* 1st entry of U(i,:) without diagonal */
21405f44c854SHong Zhang       ncols = ui[i+1] - ui[i] - 1;
21415f44c854SHong Zhang       for (j=0; j<ncols; j++) *cols++ = aj[j];
21425f44c854SHong Zhang       *cols++ = i; /* diagoanl is located as the last entry of U(i,:) */
21435f44c854SHong Zhang     }
21445f44c854SHong Zhang   } else { /* case: levels>0 || (levels=0 && !perm_identity) */
2145783ef271SHong Zhang 
21465f44c854SHong Zhang     ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
21475f44c854SHong Zhang     ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
21485f44c854SHong Zhang 
21495f44c854SHong Zhang     /* initialization */
21505f44c854SHong Zhang     ierr  = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr);
21515f44c854SHong Zhang 
21525f44c854SHong Zhang     /* jl: linked list for storing indices of the pivot rows
21535f44c854SHong Zhang        il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
21545f44c854SHong Zhang     ierr = PetscMalloc((2*am+1)*sizeof(PetscInt)+2*am*sizeof(PetscInt**),&jl);CHKERRQ(ierr);
21555f44c854SHong Zhang     il         = jl + am;
21565f44c854SHong Zhang     uj_ptr     = (PetscInt**)(il + am);
21575f44c854SHong Zhang     uj_lvl_ptr = (PetscInt**)(uj_ptr + am);
21585f44c854SHong Zhang     for (i=0; i<am; i++){
21595f44c854SHong Zhang       jl[i] = am; il[i] = 0;
21605f44c854SHong Zhang     }
21615f44c854SHong Zhang 
21625f44c854SHong Zhang     /* create and initialize a linked list for storing column indices of the active row k */
21635f44c854SHong Zhang     nlnk = am + 1;
21645f44c854SHong Zhang     ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
21655f44c854SHong Zhang 
21665f44c854SHong Zhang     /* initial FreeSpace size is fill*(ai[am]+1) */
21675f44c854SHong Zhang     ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr);
21685f44c854SHong Zhang     current_space = free_space;
21695f44c854SHong Zhang     ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space_lvl);CHKERRQ(ierr);
21705f44c854SHong Zhang     current_space_lvl = free_space_lvl;
21715f44c854SHong Zhang 
21725f44c854SHong Zhang     for (k=0; k<am; k++){  /* for each active row k */
21735f44c854SHong Zhang       /* initialize lnk by the column indices of row rip[k] of A */
21745f44c854SHong Zhang       nzk   = 0;
21755f44c854SHong Zhang       ncols = ai[rip[k]+1] - ai[rip[k]];
21765f44c854SHong 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);
21775f44c854SHong Zhang       ncols_upper = 0;
21785f44c854SHong Zhang       for (j=0; j<ncols; j++){
21795f44c854SHong Zhang         i = *(aj + ai[rip[k]] + j); /* unpermuted column index */
21805f44c854SHong Zhang         if (riip[i] >= k){ /* only take upper triangular entry */
21815f44c854SHong Zhang           ajtmp[ncols_upper] = i;
21825f44c854SHong Zhang           ncols_upper++;
21835f44c854SHong Zhang         }
21845f44c854SHong Zhang       }
21855f44c854SHong Zhang       ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
21865f44c854SHong Zhang       nzk += nlnk;
21875f44c854SHong Zhang 
21885f44c854SHong Zhang       /* update lnk by computing fill-in for each pivot row to be merged in */
21895f44c854SHong Zhang       prow = jl[k]; /* 1st pivot row */
21905f44c854SHong Zhang 
21915f44c854SHong Zhang       while (prow < k){
21925f44c854SHong Zhang         nextprow = jl[prow];
21935f44c854SHong Zhang 
21945f44c854SHong Zhang         /* merge prow into k-th row */
21955f44c854SHong Zhang         jmin = il[prow] + 1;  /* index of the 2nd nzero entry in U(prow,k:am-1) */
21965f44c854SHong Zhang         jmax = ui[prow+1];
21975f44c854SHong Zhang         ncols = jmax-jmin;
21985f44c854SHong Zhang         i     = jmin - ui[prow];
21995f44c854SHong Zhang         cols  = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */
22005f44c854SHong Zhang         uj    = uj_lvl_ptr[prow] + i; /* levels of cols */
22015f44c854SHong Zhang         j     = *(uj - 1);
22025f44c854SHong Zhang         ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr);
22035f44c854SHong Zhang         nzk += nlnk;
22045f44c854SHong Zhang 
22055f44c854SHong Zhang         /* update il and jl for prow */
22065f44c854SHong Zhang         if (jmin < jmax){
22075f44c854SHong Zhang           il[prow] = jmin;
22085f44c854SHong Zhang           j = *cols; jl[prow] = jl[j]; jl[j] = prow;
22095f44c854SHong Zhang         }
22105f44c854SHong Zhang         prow = nextprow;
22115f44c854SHong Zhang       }
22125f44c854SHong Zhang 
22135f44c854SHong Zhang       /* if free space is not available, make more free space */
22145f44c854SHong Zhang       if (current_space->local_remaining<nzk) {
22155f44c854SHong Zhang         i = am - k + 1; /* num of unfactored rows */
22165f44c854SHong Zhang         i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */
22175f44c854SHong Zhang         ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
22185f44c854SHong Zhang         ierr = PetscFreeSpaceGet(i,&current_space_lvl);CHKERRQ(ierr);
22195f44c854SHong Zhang         reallocs++;
22205f44c854SHong Zhang       }
22215f44c854SHong Zhang 
22225f44c854SHong Zhang       /* copy data into free_space and free_space_lvl, then initialize lnk */
22235f44c854SHong Zhang       if (nzk == 0) SETERRQ1(PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k);
22245f44c854SHong Zhang       ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
22255f44c854SHong Zhang 
22265f44c854SHong Zhang       /* add the k-th row into il and jl */
22275f44c854SHong Zhang       if (nzk > 1){
22285f44c854SHong Zhang         i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
22295f44c854SHong Zhang         jl[k] = jl[i]; jl[i] = k;
22305f44c854SHong Zhang         il[k] = ui[k] + 1;
22315f44c854SHong Zhang       }
22325f44c854SHong Zhang       uj_ptr[k]     = current_space->array;
22335f44c854SHong Zhang       uj_lvl_ptr[k] = current_space_lvl->array;
22345f44c854SHong Zhang 
22355f44c854SHong Zhang       current_space->array           += nzk;
22365f44c854SHong Zhang       current_space->local_used      += nzk;
22375f44c854SHong Zhang       current_space->local_remaining -= nzk;
22385f44c854SHong Zhang 
22395f44c854SHong Zhang       current_space_lvl->array           += nzk;
22405f44c854SHong Zhang       current_space_lvl->local_used      += nzk;
22415f44c854SHong Zhang       current_space_lvl->local_remaining -= nzk;
22425f44c854SHong Zhang 
22435f44c854SHong Zhang       ui[k+1] = ui[k] + nzk;
22445f44c854SHong Zhang     }
22455f44c854SHong Zhang 
22465f44c854SHong Zhang #if defined(PETSC_USE_INFO)
22475f44c854SHong Zhang     if (ai[am] != 0) {
22485f44c854SHong Zhang       PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]);
22495f44c854SHong Zhang       ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr);
22505f44c854SHong Zhang       ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
22515f44c854SHong Zhang       ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr);
22525f44c854SHong Zhang     } else {
22535f44c854SHong Zhang       ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr);
22545f44c854SHong Zhang     }
22555f44c854SHong Zhang #endif
22565f44c854SHong Zhang 
22575f44c854SHong Zhang     ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
22585f44c854SHong Zhang     ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
22595f44c854SHong Zhang     ierr = PetscFree(jl);CHKERRQ(ierr);
22605f44c854SHong Zhang     ierr = PetscFree(ajtmp);CHKERRQ(ierr);
22615f44c854SHong Zhang 
22625f44c854SHong Zhang     /* destroy list of free space and other temporary array(s) */
22635f44c854SHong Zhang     ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
2264783ef271SHong Zhang     ierr = PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag);CHKERRQ(ierr);
22655f44c854SHong Zhang     ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
22665f44c854SHong Zhang     ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
22675f44c854SHong Zhang 
22685f44c854SHong Zhang   } /* end of case: levels>0 || (levels=0 && !perm_identity) */
22695f44c854SHong Zhang 
22705f44c854SHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
22715f44c854SHong Zhang 
22725f44c854SHong Zhang   b    = (Mat_SeqSBAIJ*)(fact)->data;
22735f44c854SHong Zhang   b->singlemalloc = PETSC_FALSE;
22745f44c854SHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr);
22755f44c854SHong Zhang   b->j    = uj;
22765f44c854SHong Zhang   b->i    = ui;
22775f44c854SHong Zhang   b->diag = udiag;
22787f53bb6cSHong Zhang   b->free_diag = PETSC_TRUE;
22795f44c854SHong Zhang   b->ilen = 0;
22805f44c854SHong Zhang   b->imax = 0;
22815f44c854SHong Zhang   b->row  = perm;
22825f44c854SHong Zhang   b->col  = perm;
22835f44c854SHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
22845f44c854SHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
22855f44c854SHong Zhang   b->icol = iperm;
22865f44c854SHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
22875f44c854SHong Zhang   ierr    = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
22885f44c854SHong Zhang   ierr = PetscLogObjectMemory((fact),(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
22895f44c854SHong Zhang   b->maxnz   = b->nz = ui[am];
22905f44c854SHong Zhang   b->free_a  = PETSC_TRUE;
22915f44c854SHong Zhang   b->free_ij = PETSC_TRUE;
22925f44c854SHong Zhang 
22935f44c854SHong Zhang   (fact)->info.factor_mallocs    = reallocs;
22945f44c854SHong Zhang   (fact)->info.fill_ratio_given  = fill;
22955f44c854SHong Zhang   if (ai[am] != 0) {
22965f44c854SHong Zhang     (fact)->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]);
22975f44c854SHong Zhang   } else {
22985f44c854SHong Zhang     (fact)->info.fill_ratio_needed = 0.0;
22995f44c854SHong Zhang   }
23005f44c854SHong Zhang   (fact)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_newdatastruct;
23015f44c854SHong Zhang   PetscFunctionReturn(0);
23025f44c854SHong Zhang }
23035f44c854SHong Zhang 
23045f44c854SHong Zhang #undef __FUNCT__
23055f44c854SHong Zhang #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ"
23065f44c854SHong Zhang PetscErrorCode MatICCFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
23075f44c854SHong Zhang {
23085f44c854SHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
23095f44c854SHong Zhang   Mat_SeqSBAIJ       *b;
23105f44c854SHong Zhang   PetscErrorCode     ierr;
23115f44c854SHong Zhang   PetscTruth         perm_identity,missing;
23125f44c854SHong Zhang   PetscInt           reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag;
23135f44c854SHong Zhang   const PetscInt     *rip,*riip;
23145f44c854SHong Zhang   PetscInt           jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow;
23155f44c854SHong Zhang   PetscInt           nlnk,*lnk,*lnk_lvl=PETSC_NULL,d;
23165f44c854SHong Zhang   PetscInt           ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr;
23175f44c854SHong Zhang   PetscReal          fill=info->fill,levels=info->levels;
23185f44c854SHong Zhang   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
23195f44c854SHong Zhang   PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL;
23205f44c854SHong Zhang   PetscBT            lnkbt;
23215f44c854SHong Zhang   IS                 iperm;
23225f44c854SHong Zhang   PetscTruth         newdatastruct=PETSC_FALSE;
23235f44c854SHong Zhang 
23245f44c854SHong Zhang   PetscFunctionBegin;
23255f44c854SHong Zhang   ierr = PetscOptionsGetTruth(PETSC_NULL,"-icc_new",&newdatastruct,PETSC_NULL);CHKERRQ(ierr);
23265f44c854SHong Zhang   if(newdatastruct){
23275f44c854SHong Zhang     ierr = MatICCFactorSymbolic_SeqAIJ_newdatastruct(fact,A,perm,info);CHKERRQ(ierr);
23285f44c854SHong Zhang     PetscFunctionReturn(0);
23295f44c854SHong Zhang   }
23305f44c854SHong Zhang 
23315f44c854SHong 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);
23325f44c854SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr);
23335f44c854SHong Zhang   if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d);
23345f44c854SHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
23355f44c854SHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
23365f44c854SHong Zhang 
23375f44c854SHong Zhang   ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr);
23385f44c854SHong Zhang   ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr);
23395f44c854SHong Zhang   ui[0] = 0;
23405f44c854SHong Zhang 
23415f44c854SHong Zhang   /* ICC(0) without matrix ordering: simply copies fill pattern */
23425f44c854SHong Zhang   if (!levels && perm_identity) {
23435f44c854SHong Zhang 
23445f44c854SHong Zhang     for (i=0; i<am; i++) {
23455f44c854SHong Zhang       ui[i+1]  = ui[i] + ai[i+1] - a->diag[i];
23465f44c854SHong Zhang       udiag[i] = ui[i];
2347ed59401aSHong Zhang     }
234839e3d630SHong Zhang     ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
234939e3d630SHong Zhang     cols = uj;
2350ed59401aSHong Zhang     for (i=0; i<am; i++) {
2351ed59401aSHong Zhang       aj    = a->j + a->diag[i];
235239e3d630SHong Zhang       ncols = ui[i+1] - ui[i];
235339e3d630SHong Zhang       for (j=0; j<ncols; j++) *cols++ = *aj++;
2354ed59401aSHong Zhang     }
235539e3d630SHong Zhang   } else { /* case: levels>0 || (levels=0 && !perm_identity) */
2356b635d36bSHong Zhang     ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
2357b635d36bSHong Zhang     ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
2358b635d36bSHong Zhang 
23597a48dd6fSHong Zhang     /* initialization */
23605a8e39fbSHong Zhang     ierr  = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr);
23617a48dd6fSHong Zhang 
23627a48dd6fSHong Zhang     /* jl: linked list for storing indices of the pivot rows
23637a48dd6fSHong Zhang        il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
23641393bc97SHong Zhang     ierr = PetscMalloc((2*am+1)*sizeof(PetscInt)+2*am*sizeof(PetscInt**),&jl);CHKERRQ(ierr);
23657a48dd6fSHong Zhang     il         = jl + am;
23667a48dd6fSHong Zhang     uj_ptr     = (PetscInt**)(il + am);
23677a48dd6fSHong Zhang     uj_lvl_ptr = (PetscInt**)(uj_ptr + am);
23687a48dd6fSHong Zhang     for (i=0; i<am; i++){
23697a48dd6fSHong Zhang       jl[i] = am; il[i] = 0;
23707a48dd6fSHong Zhang     }
23717a48dd6fSHong Zhang 
23727a48dd6fSHong Zhang     /* create and initialize a linked list for storing column indices of the active row k */
23737a48dd6fSHong Zhang     nlnk = am + 1;
23742ed133dbSHong Zhang     ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
23757a48dd6fSHong Zhang 
23767a48dd6fSHong Zhang     /* initial FreeSpace size is fill*(ai[am]+1) */
2377a1a86e44SBarry Smith     ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr);
23787a48dd6fSHong Zhang     current_space = free_space;
2379a1a86e44SBarry Smith     ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space_lvl);CHKERRQ(ierr);
23807a48dd6fSHong Zhang     current_space_lvl = free_space_lvl;
23817a48dd6fSHong Zhang 
23827a48dd6fSHong Zhang     for (k=0; k<am; k++){  /* for each active row k */
23837a48dd6fSHong Zhang       /* initialize lnk by the column indices of row rip[k] of A */
23847a48dd6fSHong Zhang       nzk   = 0;
2385622e2028SHong Zhang       ncols = ai[rip[k]+1] - ai[rip[k]];
23863b4a8b6dSBarry 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);
2387622e2028SHong Zhang       ncols_upper = 0;
2388622e2028SHong Zhang       for (j=0; j<ncols; j++){
2389b635d36bSHong Zhang         i = *(aj + ai[rip[k]] + j); /* unpermuted column index */
2390b635d36bSHong Zhang         if (riip[i] >= k){ /* only take upper triangular entry */
23915a8e39fbSHong Zhang           ajtmp[ncols_upper] = i;
2392622e2028SHong Zhang           ncols_upper++;
2393622e2028SHong Zhang         }
2394622e2028SHong Zhang       }
2395b635d36bSHong Zhang       ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
23967a48dd6fSHong Zhang       nzk += nlnk;
23977a48dd6fSHong Zhang 
23987a48dd6fSHong Zhang       /* update lnk by computing fill-in for each pivot row to be merged in */
23997a48dd6fSHong Zhang       prow = jl[k]; /* 1st pivot row */
24007a48dd6fSHong Zhang 
24017a48dd6fSHong Zhang       while (prow < k){
24027a48dd6fSHong Zhang         nextprow = jl[prow];
24037a48dd6fSHong Zhang 
24047a48dd6fSHong Zhang         /* merge prow into k-th row */
24057a48dd6fSHong Zhang         jmin = il[prow] + 1;  /* index of the 2nd nzero entry in U(prow,k:am-1) */
24067a48dd6fSHong Zhang         jmax = ui[prow+1];
24077a48dd6fSHong Zhang         ncols = jmax-jmin;
2408ed59401aSHong Zhang         i     = jmin - ui[prow];
2409ed59401aSHong Zhang         cols  = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */
24105a8e39fbSHong Zhang         uj    = uj_lvl_ptr[prow] + i; /* levels of cols */
24115a8e39fbSHong Zhang         j     = *(uj - 1);
24125a8e39fbSHong Zhang         ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr);
24137a48dd6fSHong Zhang         nzk += nlnk;
24147a48dd6fSHong Zhang 
24157a48dd6fSHong Zhang         /* update il and jl for prow */
24167a48dd6fSHong Zhang         if (jmin < jmax){
24177a48dd6fSHong Zhang           il[prow] = jmin;
24187a48dd6fSHong Zhang           j = *cols; jl[prow] = jl[j]; jl[j] = prow;
24197a48dd6fSHong Zhang         }
24207a48dd6fSHong Zhang         prow = nextprow;
24217a48dd6fSHong Zhang       }
24227a48dd6fSHong Zhang 
24237a48dd6fSHong Zhang       /* if free space is not available, make more free space */
24247a48dd6fSHong Zhang       if (current_space->local_remaining<nzk) {
24257a48dd6fSHong Zhang         i = am - k + 1; /* num of unfactored rows */
24267a48dd6fSHong Zhang         i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */
2427a1a86e44SBarry Smith         ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
2428a1a86e44SBarry Smith         ierr = PetscFreeSpaceGet(i,&current_space_lvl);CHKERRQ(ierr);
24297a48dd6fSHong Zhang         reallocs++;
24307a48dd6fSHong Zhang       }
24317a48dd6fSHong Zhang 
24327a48dd6fSHong Zhang       /* copy data into free_space and free_space_lvl, then initialize lnk */
2433b635d36bSHong Zhang       if (nzk == 0) SETERRQ1(PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k);
24342ed133dbSHong Zhang       ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
24357a48dd6fSHong Zhang 
24367a48dd6fSHong Zhang       /* add the k-th row into il and jl */
243739e3d630SHong Zhang       if (nzk > 1){
24387a48dd6fSHong Zhang         i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
24397a48dd6fSHong Zhang         jl[k] = jl[i]; jl[i] = k;
24407a48dd6fSHong Zhang         il[k] = ui[k] + 1;
24417a48dd6fSHong Zhang       }
24427a48dd6fSHong Zhang       uj_ptr[k]     = current_space->array;
24437a48dd6fSHong Zhang       uj_lvl_ptr[k] = current_space_lvl->array;
24447a48dd6fSHong Zhang 
24457a48dd6fSHong Zhang       current_space->array           += nzk;
24467a48dd6fSHong Zhang       current_space->local_used      += nzk;
24477a48dd6fSHong Zhang       current_space->local_remaining -= nzk;
24487a48dd6fSHong Zhang 
24497a48dd6fSHong Zhang       current_space_lvl->array           += nzk;
24507a48dd6fSHong Zhang       current_space_lvl->local_used      += nzk;
24517a48dd6fSHong Zhang       current_space_lvl->local_remaining -= nzk;
24527a48dd6fSHong Zhang 
24537a48dd6fSHong Zhang       ui[k+1] = ui[k] + nzk;
24547a48dd6fSHong Zhang     }
24557a48dd6fSHong Zhang 
24566cf91177SBarry Smith #if defined(PETSC_USE_INFO)
24577a48dd6fSHong Zhang     if (ai[am] != 0) {
245839e3d630SHong Zhang       PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]);
2459ae15b995SBarry Smith       ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr);
2460ae15b995SBarry Smith       ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
2461ae15b995SBarry Smith       ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr);
24627a48dd6fSHong Zhang     } else {
2463ae15b995SBarry Smith       ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr);
24647a48dd6fSHong Zhang     }
246563ba0a88SBarry Smith #endif
24667a48dd6fSHong Zhang 
24677a48dd6fSHong Zhang     ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
2468b635d36bSHong Zhang     ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
24697a48dd6fSHong Zhang     ierr = PetscFree(jl);CHKERRQ(ierr);
24705a8e39fbSHong Zhang     ierr = PetscFree(ajtmp);CHKERRQ(ierr);
24717a48dd6fSHong Zhang 
24727a48dd6fSHong Zhang     /* destroy list of free space and other temporary array(s) */
24737a48dd6fSHong Zhang     ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
2474a1a86e44SBarry Smith     ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr);
24752ed133dbSHong Zhang     ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
2476a1a86e44SBarry Smith     ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
24777a48dd6fSHong Zhang 
247839e3d630SHong Zhang   } /* end of case: levels>0 || (levels=0 && !perm_identity) */
247939e3d630SHong Zhang 
24807a48dd6fSHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
24817a48dd6fSHong Zhang 
2482719d5645SBarry Smith   b    = (Mat_SeqSBAIJ*)(fact)->data;
24837a48dd6fSHong Zhang   b->singlemalloc = PETSC_FALSE;
24847a48dd6fSHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr);
24857a48dd6fSHong Zhang   b->j    = uj;
24867a48dd6fSHong Zhang   b->i    = ui;
24875f44c854SHong Zhang   b->diag = udiag;
24887f53bb6cSHong Zhang   b->free_diag = PETSC_TRUE;
24897a48dd6fSHong Zhang   b->ilen = 0;
24907a48dd6fSHong Zhang   b->imax = 0;
24917a48dd6fSHong Zhang   b->row  = perm;
2492b635d36bSHong Zhang   b->col  = perm;
2493b635d36bSHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
2494b635d36bSHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
2495b635d36bSHong Zhang   b->icol = iperm;
24967a48dd6fSHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
24977a48dd6fSHong Zhang   ierr    = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
2498719d5645SBarry Smith   ierr = PetscLogObjectMemory((fact),(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
24997a48dd6fSHong Zhang   b->maxnz   = b->nz = ui[am];
2500e6b907acSBarry Smith   b->free_a  = PETSC_TRUE;
2501e6b907acSBarry Smith   b->free_ij = PETSC_TRUE;
25027a48dd6fSHong Zhang 
2503719d5645SBarry Smith   (fact)->info.factor_mallocs    = reallocs;
2504719d5645SBarry Smith   (fact)->info.fill_ratio_given  = fill;
25057a48dd6fSHong Zhang   if (ai[am] != 0) {
2506719d5645SBarry Smith     (fact)->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]);
25077a48dd6fSHong Zhang   } else {
2508719d5645SBarry Smith     (fact)->info.fill_ratio_needed = 0.0;
25097a48dd6fSHong Zhang   }
2510719d5645SBarry Smith   (fact)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ;
2511a6175056SHong Zhang   PetscFunctionReturn(0);
2512a6175056SHong Zhang }
2513d5d45c9bSBarry Smith 
2514f76d2b81SHong Zhang #undef __FUNCT__
2515f76d2b81SHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_SeqAIJ"
25160481f469SBarry Smith PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
2517f76d2b81SHong Zhang {
2518f76d2b81SHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
251910c27e3dSHong Zhang   Mat_SeqSBAIJ       *b;
2520dfbe8321SBarry Smith   PetscErrorCode     ierr;
2521f76d2b81SHong Zhang   PetscTruth         perm_identity;
252210c27e3dSHong Zhang   PetscReal          fill = info->fill;
25235d0c19d7SBarry Smith   const PetscInt     *rip,*riip;
25245d0c19d7SBarry Smith   PetscInt           i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow;
252510c27e3dSHong Zhang   PetscInt           *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow;
25262ed133dbSHong Zhang   PetscInt           nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr;
2527a1a86e44SBarry Smith   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
252810c27e3dSHong Zhang   PetscBT            lnkbt;
25292ed133dbSHong Zhang   IS                 iperm;
2530f76d2b81SHong Zhang 
2531f76d2b81SHong Zhang   PetscFunctionBegin;
2532d0f46423SBarry 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);
25332ed133dbSHong Zhang   /* check whether perm is the identity mapping */
2534f76d2b81SHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
25352ed133dbSHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
25362ed133dbSHong Zhang   ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
25379bfd6278SHong Zhang   ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
253810c27e3dSHong Zhang 
253910c27e3dSHong Zhang   /* initialization */
25401393bc97SHong Zhang   ierr  = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr);
254110c27e3dSHong Zhang   ui[0] = 0;
254210c27e3dSHong Zhang 
254310c27e3dSHong Zhang   /* jl: linked list for storing indices of the pivot rows
25441393bc97SHong Zhang      il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
25451393bc97SHong Zhang   ierr = PetscMalloc((3*am+1)*sizeof(PetscInt)+am*sizeof(PetscInt**),&jl);CHKERRQ(ierr);
25461393bc97SHong Zhang   il     = jl + am;
25471393bc97SHong Zhang   cols   = il + am;
25481393bc97SHong Zhang   ui_ptr = (PetscInt**)(cols + am);
25491393bc97SHong Zhang   for (i=0; i<am; i++){
25501393bc97SHong Zhang     jl[i] = am; il[i] = 0;
2551f76d2b81SHong Zhang   }
255210c27e3dSHong Zhang 
255310c27e3dSHong Zhang   /* create and initialize a linked list for storing column indices of the active row k */
25541393bc97SHong Zhang   nlnk = am + 1;
25551393bc97SHong Zhang   ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
255610c27e3dSHong Zhang 
25571393bc97SHong Zhang   /* initial FreeSpace size is fill*(ai[am]+1) */
2558a1a86e44SBarry Smith   ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr);
255910c27e3dSHong Zhang   current_space = free_space;
256010c27e3dSHong Zhang 
25611393bc97SHong Zhang   for (k=0; k<am; k++){  /* for each active row k */
256210c27e3dSHong Zhang     /* initialize lnk by the column indices of row rip[k] of A */
256310c27e3dSHong Zhang     nzk   = 0;
25642ed133dbSHong Zhang     ncols = ai[rip[k]+1] - ai[rip[k]];
25653b4a8b6dSBarry 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);
25662ed133dbSHong Zhang     ncols_upper = 0;
25672ed133dbSHong Zhang     for (j=0; j<ncols; j++){
25689bfd6278SHong Zhang       i = riip[*(aj + ai[rip[k]] + j)];
25692ed133dbSHong Zhang       if (i >= k){ /* only take upper triangular entry */
25702ed133dbSHong Zhang         cols[ncols_upper] = i;
25712ed133dbSHong Zhang         ncols_upper++;
25722ed133dbSHong Zhang       }
25732ed133dbSHong Zhang     }
25741393bc97SHong Zhang     ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
257510c27e3dSHong Zhang     nzk += nlnk;
257610c27e3dSHong Zhang 
257710c27e3dSHong Zhang     /* update lnk by computing fill-in for each pivot row to be merged in */
257810c27e3dSHong Zhang     prow = jl[k]; /* 1st pivot row */
257910c27e3dSHong Zhang 
258010c27e3dSHong Zhang     while (prow < k){
258110c27e3dSHong Zhang       nextprow = jl[prow];
258210c27e3dSHong Zhang       /* merge prow into k-th row */
25831393bc97SHong Zhang       jmin = il[prow] + 1;  /* index of the 2nd nzero entry in U(prow,k:am-1) */
258410c27e3dSHong Zhang       jmax = ui[prow+1];
258510c27e3dSHong Zhang       ncols = jmax-jmin;
25861393bc97SHong Zhang       uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */
25875a8e39fbSHong Zhang       ierr = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
258810c27e3dSHong Zhang       nzk += nlnk;
258910c27e3dSHong Zhang 
259010c27e3dSHong Zhang       /* update il and jl for prow */
259110c27e3dSHong Zhang       if (jmin < jmax){
259210c27e3dSHong Zhang         il[prow] = jmin;
25932ed133dbSHong Zhang         j = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow;
259410c27e3dSHong Zhang       }
259510c27e3dSHong Zhang       prow = nextprow;
259610c27e3dSHong Zhang     }
259710c27e3dSHong Zhang 
259810c27e3dSHong Zhang     /* if free space is not available, make more free space */
259910c27e3dSHong Zhang     if (current_space->local_remaining<nzk) {
26001393bc97SHong Zhang       i = am - k + 1; /* num of unfactored rows */
260110c27e3dSHong Zhang       i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */
2602a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
260310c27e3dSHong Zhang       reallocs++;
260410c27e3dSHong Zhang     }
260510c27e3dSHong Zhang 
260610c27e3dSHong Zhang     /* copy data into free space, then initialize lnk */
26071393bc97SHong Zhang     ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
260810c27e3dSHong Zhang 
260910c27e3dSHong Zhang     /* add the k-th row into il and jl */
261010c27e3dSHong Zhang     if (nzk-1 > 0){
26111393bc97SHong Zhang       i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
261210c27e3dSHong Zhang       jl[k] = jl[i]; jl[i] = k;
261310c27e3dSHong Zhang       il[k] = ui[k] + 1;
261410c27e3dSHong Zhang     }
26152ed133dbSHong Zhang     ui_ptr[k] = current_space->array;
261610c27e3dSHong Zhang     current_space->array           += nzk;
261710c27e3dSHong Zhang     current_space->local_used      += nzk;
261810c27e3dSHong Zhang     current_space->local_remaining -= nzk;
261910c27e3dSHong Zhang 
262010c27e3dSHong Zhang     ui[k+1] = ui[k] + nzk;
262110c27e3dSHong Zhang   }
262210c27e3dSHong Zhang 
26236cf91177SBarry Smith #if defined(PETSC_USE_INFO)
26241393bc97SHong Zhang   if (ai[am] != 0) {
26251393bc97SHong Zhang     PetscReal af = (PetscReal)(ui[am])/((PetscReal)ai[am]);
2626ae15b995SBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr);
2627ae15b995SBarry Smith     ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
2628ae15b995SBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr);
262910c27e3dSHong Zhang   } else {
2630ae15b995SBarry Smith      ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr);
263110c27e3dSHong Zhang   }
263263ba0a88SBarry Smith #endif
263310c27e3dSHong Zhang 
263410c27e3dSHong Zhang   ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
26359bfd6278SHong Zhang   ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
263610c27e3dSHong Zhang   ierr = PetscFree(jl);CHKERRQ(ierr);
263710c27e3dSHong Zhang 
263810c27e3dSHong Zhang   /* destroy list of free space and other temporary array(s) */
26391393bc97SHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
2640a1a86e44SBarry Smith   ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr);
264110c27e3dSHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
264210c27e3dSHong Zhang 
264310c27e3dSHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
264410c27e3dSHong Zhang 
2645719d5645SBarry Smith   b = (Mat_SeqSBAIJ*)(fact)->data;
264610c27e3dSHong Zhang   b->singlemalloc = PETSC_FALSE;
2647e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
2648e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
26491393bc97SHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr);
265010c27e3dSHong Zhang   b->j    = uj;
265110c27e3dSHong Zhang   b->i    = ui;
265210c27e3dSHong Zhang   b->diag = 0;
265310c27e3dSHong Zhang   b->ilen = 0;
265410c27e3dSHong Zhang   b->imax = 0;
265510c27e3dSHong Zhang   b->row  = perm;
26569bfd6278SHong Zhang   b->col  = perm;
26579bfd6278SHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
26589bfd6278SHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
26599bfd6278SHong Zhang   b->icol = iperm;
266010c27e3dSHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
26611393bc97SHong Zhang   ierr    = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
2662719d5645SBarry Smith   ierr    = PetscLogObjectMemory(fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
26631393bc97SHong Zhang   b->maxnz = b->nz = ui[am];
266410c27e3dSHong Zhang 
2665719d5645SBarry Smith   (fact)->info.factor_mallocs    = reallocs;
2666719d5645SBarry Smith   (fact)->info.fill_ratio_given  = fill;
26671393bc97SHong Zhang   if (ai[am] != 0) {
2668719d5645SBarry Smith     (fact)->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]);
266910c27e3dSHong Zhang   } else {
2670719d5645SBarry Smith     (fact)->info.fill_ratio_needed = 0.0;
267110c27e3dSHong Zhang   }
2672719d5645SBarry Smith   (fact)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ;
2673f76d2b81SHong Zhang   PetscFunctionReturn(0);
2674f76d2b81SHong Zhang }
2675599ef60dSHong Zhang 
2676599ef60dSHong Zhang #undef __FUNCT__
2677894a2336SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering_newdatastruct"
2678894a2336SHong Zhang PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering_newdatastruct(Mat A,Vec bb,Vec xx)
2679599ef60dSHong Zhang {
26801d098868SHong Zhang   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
26811d098868SHong Zhang   PetscErrorCode    ierr;
26821d098868SHong Zhang   PetscInt          n = A->rmap->n;
26838fc3a347SHong Zhang   const PetscInt    *ai = a->i,*aj = a->j,*vi;
26841d098868SHong Zhang   PetscScalar       *x,sum;
26851d098868SHong Zhang   const PetscScalar *b;
26861d098868SHong Zhang   const MatScalar   *aa = a->a,*v;
26878fc3a347SHong Zhang   PetscInt          i,nz;
26881d098868SHong Zhang 
26891d098868SHong Zhang   PetscFunctionBegin;
26901d098868SHong Zhang   if (!n) PetscFunctionReturn(0);
26911d098868SHong Zhang 
26921d098868SHong Zhang   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
26931d098868SHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
26941d098868SHong Zhang 
26951d098868SHong Zhang   /* forward solve the lower triangular */
26961d098868SHong Zhang   x[0] = b[0];
26971d098868SHong Zhang   v    = aa;
26981d098868SHong Zhang   vi   = aj;
26991d098868SHong Zhang   for (i=1; i<n; i++) {
27001d098868SHong Zhang     nz  = ai[i+1] - ai[i];
27011d098868SHong Zhang     sum = b[i];
2702003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
2703e0e43300SBarry Smith     v  += nz;
2704e0e43300SBarry Smith     vi += nz;
27051d098868SHong Zhang     x[i] = sum;
27061d098868SHong Zhang   }
27071d098868SHong Zhang 
27081d098868SHong Zhang   /* backward solve the upper triangular */
27098fc3a347SHong Zhang   v   = aa + ai[n+1];
27108fc3a347SHong Zhang   vi  = aj + ai[n+1];
27111d098868SHong Zhang   for (i=n-1; i>=0; i--){
27128fc3a347SHong Zhang     nz = ai[2*n-i +1] - ai[2*n-i]-1;
27131d098868SHong Zhang     sum = x[i];
2714003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
2715e0e43300SBarry Smith     v   += nz;
27168fc3a347SHong Zhang     vi  += nz; vi++;
2717fe31b70dSHong Zhang     x[i] = *v++ *sum; /* x[i]=aa[adiag[i]]*sum; v++; */
27181d098868SHong Zhang   }
27191d098868SHong Zhang 
27201d098868SHong Zhang   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
27211d098868SHong Zhang   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
27221d098868SHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
27231d098868SHong Zhang   PetscFunctionReturn(0);
27241d098868SHong Zhang }
27251d098868SHong Zhang 
27261d098868SHong Zhang #undef __FUNCT__
2727894a2336SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_newdatastruct"
2728894a2336SHong Zhang PetscErrorCode MatSolve_SeqAIJ_newdatastruct(Mat A,Vec bb,Vec xx)
27291d098868SHong Zhang {
27301d098868SHong Zhang   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
27311d098868SHong Zhang   IS                iscol = a->col,isrow = a->row;
27321d098868SHong Zhang   PetscErrorCode    ierr;
2733894a2336SHong Zhang   PetscInt          i,n=A->rmap->n,*vi,*ai=a->i,*aj=a->j,nz,k;
27341d098868SHong Zhang   const PetscInt    *rout,*cout,*r,*c;
2735894a2336SHong Zhang   PetscScalar       *x,*tmp,*tmps,sum;
27361d098868SHong Zhang   const PetscScalar *b;
27371d098868SHong Zhang   const MatScalar   *aa = a->a,*v;
27381d098868SHong Zhang 
27391d098868SHong Zhang   PetscFunctionBegin;
27401d098868SHong Zhang   if (!n) PetscFunctionReturn(0);
27411d098868SHong Zhang 
27421d098868SHong Zhang   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
27431d098868SHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
27441d098868SHong Zhang   tmp  = a->solve_work;
27451d098868SHong Zhang 
27461d098868SHong Zhang   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
2747894a2336SHong Zhang   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
27481d098868SHong Zhang 
27491d098868SHong Zhang   /* forward solve the lower triangular */
27501d098868SHong Zhang   tmp[0] = b[*r++];
27511d098868SHong Zhang   tmps   = tmp;
27521d098868SHong Zhang   v      = aa;
27531d098868SHong Zhang   vi     = aj;
27541d098868SHong Zhang   for (i=1; i<n; i++) {
27551d098868SHong Zhang     nz  = ai[i+1] - ai[i];
2756894a2336SHong Zhang     sum = b[*r++];
2757894a2336SHong Zhang     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
2758894a2336SHong Zhang     tmp[i] = sum;
27591d098868SHong Zhang     v += nz; vi += nz;
27601d098868SHong Zhang   }
27611d098868SHong Zhang 
27621d098868SHong Zhang   /* backward solve the upper triangular */
2763894a2336SHong Zhang   k  = n+1;
2764894a2336SHong Zhang   v  = aa + ai[k]; /* 1st entry of U(n-1,:) */
2765894a2336SHong Zhang   vi = aj + ai[k];
27661d098868SHong Zhang   for (i=n-1; i>=0; i--){
2767894a2336SHong Zhang     k  = 2*n-i;
2768894a2336SHong Zhang     nz = ai[k +1] - ai[k] - 1;
2769894a2336SHong Zhang     sum = tmp[i];
2770894a2336SHong Zhang     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
2771894a2336SHong Zhang     x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */
27721d098868SHong Zhang     v += nz+1; vi += nz+1;
27731d098868SHong Zhang   }
27741d098868SHong Zhang 
27751d098868SHong Zhang   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
27761d098868SHong Zhang   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
27771d098868SHong Zhang   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
27781d098868SHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
27791d098868SHong Zhang   ierr = PetscLogFlops(2*a->nz - A->cmap->n);CHKERRQ(ierr);
27801d098868SHong Zhang   PetscFunctionReturn(0);
27811d098868SHong Zhang }
27821d098868SHong Zhang 
27831d098868SHong Zhang #undef __FUNCT__
27841d098868SHong Zhang #define __FUNCT__ "MatILUDTFactor_SeqAIJ"
27851d098868SHong Zhang PetscErrorCode MatILUDTFactor_SeqAIJ(Mat A,IS isrow,IS iscol,const MatFactorInfo *info,Mat *fact)
27861d098868SHong Zhang {
27871d098868SHong Zhang   Mat                B = *fact;
2788599ef60dSHong Zhang   Mat_SeqAIJ         *a=(Mat_SeqAIJ*)A->data,*b;
2789599ef60dSHong Zhang   IS                 isicol;
2790599ef60dSHong Zhang   PetscErrorCode     ierr;
27911d098868SHong Zhang   const PetscInt     *r,*ic;
27921d098868SHong Zhang   PetscInt           i,n=A->rmap->n,*ai=a->i,*aj=a->j,*ajtmp,*adiag;
2793dc3a2fd3SHong Zhang   PetscInt           *bi,*bj,*bdiag,*bdiag_rev;
2794f61a948aSLisandro Dalcin   PetscInt           row,nzi,nzi_bl,nzi_bu,*im,nzi_al,nzi_au;
27951d098868SHong Zhang   PetscInt           nlnk,*lnk;
27961d098868SHong Zhang   PetscBT            lnkbt;
27971d098868SHong Zhang   PetscTruth         row_identity,icol_identity,both_identity;
27988aee2decSHong Zhang   MatScalar          *aatmp,*pv,*batmp,*ba,*rtmp,*pc,multiplier,*vtmp,diag_tmp;
27991d098868SHong Zhang   const PetscInt     *ics;
28001d098868SHong Zhang   PetscInt           j,nz,*pj,*bjtmp,k,ncut,*jtmp;
2801f61a948aSLisandro Dalcin   PetscReal          dt=info->dt,dtcol=info->dtcol,shift=info->shiftinblocks;
2802f61a948aSLisandro Dalcin   PetscInt           dtcount=(PetscInt)info->dtcount,nnz_max;
2803599ef60dSHong Zhang   PetscTruth         missing;
2804599ef60dSHong Zhang 
2805599ef60dSHong Zhang   PetscFunctionBegin;
2806f61a948aSLisandro Dalcin 
2807f61a948aSLisandro Dalcin   if (dt      == PETSC_DEFAULT) dt      = 0.005;
2808f61a948aSLisandro Dalcin   if (dtcol   == PETSC_DEFAULT) dtcol   = 0.01; /* XXX unused! */
2809f61a948aSLisandro Dalcin   if (dtcount == PETSC_DEFAULT) dtcount = (PetscInt)(1.5*a->rmax);
2810f61a948aSLisandro Dalcin 
28111d098868SHong Zhang   /* ------- symbolic factorization, can be reused ---------*/
28121d098868SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr);
28131d098868SHong Zhang   if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i);
28141d098868SHong Zhang   adiag=a->diag;
2815599ef60dSHong Zhang 
2816599ef60dSHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
2817599ef60dSHong Zhang 
2818599ef60dSHong Zhang   /* bdiag is location of diagonal in factor */
2819a6226a1aSHong Zhang   ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
2820a6226a1aSHong Zhang   bdiag_rev = bdiag + n+1;
2821599ef60dSHong Zhang 
28221d098868SHong Zhang   /* allocate row pointers bi */
28238fc3a347SHong Zhang   ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
28241d098868SHong Zhang 
28251d098868SHong Zhang   /* allocate bj and ba; max num of nonzero entries is (ai[n]+2*n*dtcount+2) */
2826393d3a68SHong Zhang   if (dtcount > n-1) dtcount = n-1; /* diagonal is excluded */
28271d098868SHong Zhang   nnz_max  = ai[n]+2*n*dtcount+2;
28288fc3a347SHong Zhang 
28298fc3a347SHong Zhang   ierr = PetscMalloc((nnz_max+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
28308fc3a347SHong Zhang   ierr = PetscMalloc((nnz_max+1)*sizeof(MatScalar),&ba);CHKERRQ(ierr);
2831599ef60dSHong Zhang 
2832599ef60dSHong Zhang   /* put together the new matrix */
2833599ef60dSHong Zhang   ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
2834599ef60dSHong Zhang   ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr);
2835599ef60dSHong Zhang   b    = (Mat_SeqAIJ*)(B)->data;
2836599ef60dSHong Zhang   b->free_a       = PETSC_TRUE;
2837599ef60dSHong Zhang   b->free_ij      = PETSC_TRUE;
2838599ef60dSHong Zhang   b->singlemalloc = PETSC_FALSE;
2839599ef60dSHong Zhang   b->a          = ba;
2840599ef60dSHong Zhang   b->j          = bj;
2841599ef60dSHong Zhang   b->i          = bi;
2842599ef60dSHong Zhang   b->diag       = bdiag;
2843599ef60dSHong Zhang   b->ilen       = 0;
2844599ef60dSHong Zhang   b->imax       = 0;
2845599ef60dSHong Zhang   b->row        = isrow;
2846599ef60dSHong Zhang   b->col        = iscol;
2847599ef60dSHong Zhang   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
2848599ef60dSHong Zhang   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
2849599ef60dSHong Zhang   b->icol       = isicol;
2850599ef60dSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
2851599ef60dSHong Zhang 
28521d098868SHong Zhang   ierr = PetscLogObjectMemory(B,nnz_max*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
28531d098868SHong Zhang   b->maxnz = nnz_max;
2854599ef60dSHong Zhang 
2855599ef60dSHong Zhang   (B)->factor                = MAT_FACTOR_ILUDT;
2856599ef60dSHong Zhang   (B)->info.factor_mallocs   = 0;
28571d098868SHong Zhang   (B)->info.fill_ratio_given = ((PetscReal)nnz_max)/((PetscReal)ai[n]);
28581d098868SHong Zhang   CHKMEMQ;
28591d098868SHong Zhang   /* ------- end of symbolic factorization ---------*/
2860599ef60dSHong Zhang 
2861599ef60dSHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
2862599ef60dSHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
2863599ef60dSHong Zhang   ics  = ic;
2864599ef60dSHong Zhang 
2865599ef60dSHong Zhang   /* linked list for storing column indices of the active row */
2866599ef60dSHong Zhang   nlnk = n + 1;
2867599ef60dSHong Zhang   ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
28681d098868SHong Zhang 
28691d098868SHong Zhang   /* im: used by PetscLLAddSortedLU(); jtmp: working array for column indices of active row */
28701d098868SHong Zhang   ierr = PetscMalloc((2*n+1)*sizeof(PetscInt),&im);CHKERRQ(ierr);
28711d098868SHong Zhang   jtmp = im + n;
28721d098868SHong Zhang   /* rtmp, vtmp: working arrays for sparse and contiguous row entries of active row */
28738369b28dSHong Zhang   ierr = PetscMalloc((2*n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr);
28748aee2decSHong Zhang   ierr = PetscMemzero(rtmp,(n+1)*sizeof(PetscScalar));CHKERRQ(ierr);
28758369b28dSHong Zhang   vtmp = rtmp + n;
2876599ef60dSHong Zhang 
2877599ef60dSHong Zhang   bi[0]    = 0;
28788fc3a347SHong Zhang   bdiag[0] = nnz_max-1; /* location of diag[0] in factor B */
2879dc3a2fd3SHong Zhang   bdiag_rev[n] = bdiag[0];
28808fc3a347SHong Zhang   bi[2*n+1] = bdiag[0]+1; /* endof bj and ba array */
2881599ef60dSHong Zhang   for (i=0; i<n; i++) {
2882599ef60dSHong Zhang     /* copy initial fill into linked list */
2883599ef60dSHong Zhang     nzi = 0; /* nonzeros for active row i */
28848369b28dSHong Zhang     nzi = ai[r[i]+1] - ai[r[i]];
28858369b28dSHong 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);
28868369b28dSHong Zhang     nzi_al = adiag[r[i]] - ai[r[i]];
28878369b28dSHong Zhang     nzi_au = ai[r[i]+1] - adiag[r[i]] -1;
2888599ef60dSHong Zhang     ajtmp = aj + ai[r[i]];
28898369b28dSHong Zhang     ierr = PetscLLAddPerm(nzi,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
2890599ef60dSHong Zhang 
2891599ef60dSHong Zhang     /* load in initial (unfactored row) */
28921d098868SHong Zhang     aatmp = a->a + ai[r[i]];
28938369b28dSHong Zhang     for (j=0; j<nzi; j++) {
28941d098868SHong Zhang       rtmp[ics[*ajtmp++]] = *aatmp++;
2895599ef60dSHong Zhang     }
2896599ef60dSHong Zhang 
2897599ef60dSHong Zhang     /* add pivot rows into linked list */
2898599ef60dSHong Zhang     row = lnk[n];
2899599ef60dSHong Zhang     while (row < i ) {
29001d098868SHong Zhang       nzi_bl = bi[row+1] - bi[row] + 1;
29011d098868SHong Zhang       bjtmp = bj + bdiag[row+1]+1; /* points to 1st column next to the diagonal in U */
29021d098868SHong Zhang       ierr  = PetscLLAddSortedLU(bjtmp,row,nlnk,lnk,lnkbt,i,nzi_bl,im);CHKERRQ(ierr);
2903599ef60dSHong Zhang       nzi  += nlnk;
2904599ef60dSHong Zhang       row   = lnk[row];
2905599ef60dSHong Zhang     }
2906599ef60dSHong Zhang 
29078369b28dSHong Zhang     /* copy data from lnk into jtmp, then initialize lnk */
29088369b28dSHong Zhang     ierr = PetscLLClean(n,n,nzi,lnk,jtmp,lnkbt);CHKERRQ(ierr);
2909599ef60dSHong Zhang 
2910599ef60dSHong Zhang     /* numerical factorization */
29118369b28dSHong Zhang     bjtmp = jtmp;
2912599ef60dSHong Zhang     row   = *bjtmp++; /* 1st pivot row */
2913599ef60dSHong Zhang     while  ( row < i ) {
2914599ef60dSHong Zhang       pc         = rtmp + row;
29153c2a7987SHong Zhang       pv         = ba + bdiag[row]; /* 1./(diag of the pivot row) */
29163c2a7987SHong Zhang       multiplier = (*pc) * (*pv);
2917599ef60dSHong Zhang       *pc        = multiplier;
29183c2a7987SHong Zhang       if (PetscAbsScalar(*pc) > dt){ /* apply tolerance dropping rule */
29191d098868SHong Zhang         pj         = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */
29201d098868SHong Zhang         pv         = ba + bdiag[row+1] + 1;
29211d098868SHong Zhang         /* if (multiplier < -1.0 or multiplier >1.0) printf("row/prow %d, %d, multiplier %g\n",i,row,multiplier); */
29221d098868SHong Zhang         nz         = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */
29231d098868SHong Zhang         for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++);
2924dc0b31edSSatish Balay         ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr);
2925599ef60dSHong Zhang       }
2926599ef60dSHong Zhang       row = *bjtmp++;
2927599ef60dSHong Zhang     }
2928599ef60dSHong Zhang 
29298369b28dSHong Zhang     /* copy sparse rtmp into contiguous vtmp; separate L and U part */
29308aee2decSHong Zhang     diag_tmp = rtmp[i];  /* save diagonal value - may not needed?? */
29318369b28dSHong Zhang     nzi_bl = 0; j = 0;
29328369b28dSHong Zhang     while (jtmp[j] < i){ /* Note: jtmp is sorted */
29338aee2decSHong Zhang       vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0;
29348369b28dSHong Zhang       nzi_bl++; j++;
29358369b28dSHong Zhang     }
29368369b28dSHong Zhang     nzi_bu = nzi - nzi_bl -1;
29378369b28dSHong Zhang     while (j < nzi){
29388aee2decSHong Zhang       vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0;
29398369b28dSHong Zhang       j++;
29408369b28dSHong Zhang     }
29418369b28dSHong Zhang 
2942599ef60dSHong Zhang     bjtmp = bj + bi[i];
2943599ef60dSHong Zhang     batmp = ba + bi[i];
29448369b28dSHong Zhang     /* apply level dropping rule to L part */
29458369b28dSHong Zhang     ncut = nzi_al + dtcount;
29468369b28dSHong Zhang     if (ncut < nzi_bl){
29478369b28dSHong Zhang       ierr = PetscSortSplit(ncut,nzi_bl,vtmp,jtmp);CHKERRQ(ierr);
29488369b28dSHong Zhang       ierr = PetscSortIntWithScalarArray(ncut,jtmp,vtmp);CHKERRQ(ierr);
2949599ef60dSHong Zhang     } else {
29508369b28dSHong Zhang       ncut = nzi_bl;
29518369b28dSHong Zhang     }
29528369b28dSHong Zhang     for (j=0; j<ncut; j++){
29538369b28dSHong Zhang       bjtmp[j] = jtmp[j];
29548369b28dSHong Zhang       batmp[j] = vtmp[j];
29556da515a1SHong Zhang       /* printf(" (%d,%g),",bjtmp[j],batmp[j]); */
29568369b28dSHong Zhang     }
29571d098868SHong Zhang     bi[i+1] = bi[i] + ncut;
29588369b28dSHong Zhang     nzi = ncut + 1;
29598369b28dSHong Zhang 
29608369b28dSHong Zhang     /* apply level dropping rule to U part */
29618369b28dSHong Zhang     ncut = nzi_au + dtcount;
29628369b28dSHong Zhang     if (ncut < nzi_bu){
29638369b28dSHong Zhang       ierr = PetscSortSplit(ncut,nzi_bu,vtmp+nzi_bl+1,jtmp+nzi_bl+1);CHKERRQ(ierr);
29648369b28dSHong Zhang       ierr = PetscSortIntWithScalarArray(ncut,jtmp+nzi_bl+1,vtmp+nzi_bl+1);CHKERRQ(ierr);
29658369b28dSHong Zhang     } else {
29668369b28dSHong Zhang       ncut = nzi_bu;
29678369b28dSHong Zhang     }
29688369b28dSHong Zhang     nzi += ncut;
29691d098868SHong Zhang 
29701d098868SHong Zhang     /* mark bdiagonal */
29711d098868SHong Zhang     bdiag[i+1]       = bdiag[i] - (ncut + 1);
2972dc3a2fd3SHong Zhang     bdiag_rev[n-i-1] = bdiag[i+1];
29738fc3a347SHong Zhang     bi[2*n - i]      = bi[2*n - i +1] - (ncut + 1);
29741d098868SHong Zhang     bjtmp = bj + bdiag[i];
29751d098868SHong Zhang     batmp = ba + bdiag[i];
29761d098868SHong Zhang     *bjtmp = i;
29778aee2decSHong Zhang     *batmp = diag_tmp; /* rtmp[i]; */
2978c9c72f8fSHong Zhang     if (*batmp == 0.0) {
2979c9c72f8fSHong Zhang       *batmp = dt+shift;
2980bd1bc851SBarry Smith       /* printf(" row %d add shift %g\n",i,shift); */
2981c9c72f8fSHong Zhang     }
2982b78a477dSHong Zhang     *batmp = 1.0/(*batmp); /* invert diagonal entries for simplier triangular solves */
29836da515a1SHong Zhang     /* printf(" (%d,%g),",*bjtmp,*batmp); */
29841d098868SHong Zhang 
29851d098868SHong Zhang     bjtmp = bj + bdiag[i+1]+1;
29861d098868SHong Zhang     batmp = ba + bdiag[i+1]+1;
29878369b28dSHong Zhang     for (k=0; k<ncut; k++){
29881d098868SHong Zhang       bjtmp[k] = jtmp[nzi_bl+1+k];
29891d098868SHong Zhang       batmp[k] = vtmp[nzi_bl+1+k];
29906da515a1SHong Zhang       /* printf(" (%d,%g),",bjtmp[k],batmp[k]); */
2991599ef60dSHong Zhang     }
29926da515a1SHong Zhang     /* printf("\n"); */
2993599ef60dSHong Zhang 
2994599ef60dSHong Zhang     im[i]   = nzi; /* used by PetscLLAddSortedLU() */
2995599ef60dSHong Zhang     /*
29961d098868SHong Zhang     printf("row %d: bi %d, bdiag %d\n",i,bi[i],bdiag[i]);
29978369b28dSHong Zhang     printf(" ----------------------------\n");
2998599ef60dSHong Zhang     */
29998369b28dSHong Zhang   } /* for (i=0; i<n; i++) */
30001d098868SHong Zhang   /* printf("end of L %d, beginning of U %d\n",bi[n],bdiag[n]); */
30011d098868SHong 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]);
3002599ef60dSHong Zhang 
3003599ef60dSHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
3004599ef60dSHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
3005599ef60dSHong Zhang 
3006599ef60dSHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
3007599ef60dSHong Zhang   ierr = PetscFree(im);CHKERRQ(ierr);
3008599ef60dSHong Zhang   ierr = PetscFree(rtmp);CHKERRQ(ierr);
3009599ef60dSHong Zhang 
3010599ef60dSHong Zhang   ierr = PetscLogFlops(B->cmap->n);CHKERRQ(ierr);
30111d098868SHong Zhang   b->maxnz = b->nz = bi[n] + bdiag[0] - bdiag[n];
3012599ef60dSHong Zhang 
3013599ef60dSHong Zhang   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
3014599ef60dSHong Zhang   ierr = ISIdentity(isicol,&icol_identity);CHKERRQ(ierr);
3015599ef60dSHong Zhang   both_identity = (PetscTruth) (row_identity && icol_identity);
3016599ef60dSHong Zhang   if (row_identity && icol_identity) {
3017894a2336SHong Zhang     B->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_newdatastruct;
3018599ef60dSHong Zhang   } else {
3019894a2336SHong Zhang     B->ops->solve = MatSolve_SeqAIJ_newdatastruct;
3020599ef60dSHong Zhang   }
3021599ef60dSHong Zhang 
3022b78a477dSHong Zhang   B->ops->lufactorsymbolic  = MatILUDTFactorSymbolic_SeqAIJ;
3023b78a477dSHong Zhang   B->ops->lufactornumeric   = MatILUDTFactorNumeric_SeqAIJ;
30241d098868SHong Zhang   B->ops->solveadd          = 0;
30251d098868SHong Zhang   B->ops->solvetranspose    = 0;
30261d098868SHong Zhang   B->ops->solvetransposeadd = 0;
30271d098868SHong Zhang   B->ops->matsolve          = 0;
3028599ef60dSHong Zhang   B->assembled              = PETSC_TRUE;
3029599ef60dSHong Zhang   B->preallocated           = PETSC_TRUE;
3030599ef60dSHong Zhang   PetscFunctionReturn(0);
3031599ef60dSHong Zhang }
3032599ef60dSHong Zhang 
30333c2a7987SHong Zhang /* a wraper of MatILUDTFactor_SeqAIJ() */
30343c2a7987SHong Zhang #undef __FUNCT__
30353c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorSymbolic_SeqAIJ"
30363c2a7987SHong Zhang PetscErrorCode PETSCMAT_DLLEXPORT MatILUDTFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS row,IS col,const MatFactorInfo *info)
30373c2a7987SHong Zhang {
30383c2a7987SHong Zhang   PetscErrorCode     ierr;
30393c2a7987SHong Zhang 
30403c2a7987SHong Zhang   PetscFunctionBegin;
30413c2a7987SHong Zhang   ierr = MatILUDTFactor_SeqAIJ(A,row,col,info,&fact);CHKERRQ(ierr);
30423c2a7987SHong Zhang 
30433c2a7987SHong Zhang   fact->ops->lufactornumeric = MatILUDTFactorNumeric_SeqAIJ;
30443c2a7987SHong Zhang   PetscFunctionReturn(0);
30453c2a7987SHong Zhang }
30463c2a7987SHong Zhang 
30473c2a7987SHong Zhang /*
30483c2a7987SHong Zhang    same as MatLUFactorNumeric_SeqAIJ(), except using contiguous array matrix factors
30493c2a7987SHong Zhang    - intend to replace existing MatLUFactorNumeric_SeqAIJ()
30503c2a7987SHong Zhang */
30513c2a7987SHong Zhang #undef __FUNCT__
30523c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorNumeric_SeqAIJ"
30533c2a7987SHong Zhang PetscErrorCode PETSCMAT_DLLEXPORT MatILUDTFactorNumeric_SeqAIJ(Mat fact,Mat A,const MatFactorInfo *info)
30543c2a7987SHong Zhang {
30553c2a7987SHong Zhang   Mat            C=fact;
30563c2a7987SHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data;
30573c2a7987SHong Zhang   IS             isrow = b->row,isicol = b->icol;
30583c2a7987SHong Zhang   PetscErrorCode ierr;
30593c2a7987SHong Zhang   const PetscInt *r,*ic,*ics;
3060b78a477dSHong Zhang   PetscInt       i,j,k,n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j;
3061b78a477dSHong Zhang   PetscInt       *ajtmp,*bjtmp,nz,nzl,nzu,row,*bdiag = b->diag,*pj;
30623c2a7987SHong Zhang   MatScalar      *rtmp,*pc,multiplier,*v,*pv,*aa=a->a;
3063b78a477dSHong Zhang   PetscReal      dt=info->dt,shift=info->shiftinblocks;
3064b78a477dSHong Zhang   PetscTruth     row_identity, col_identity;
30653c2a7987SHong Zhang 
30663c2a7987SHong Zhang   PetscFunctionBegin;
30673c2a7987SHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
30683c2a7987SHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
3069b78a477dSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr);
30703c2a7987SHong Zhang   ics  = ic;
30713c2a7987SHong Zhang 
30723c2a7987SHong Zhang   for (i=0; i<n; i++){
3073b78a477dSHong Zhang     /* initialize rtmp array */
3074b78a477dSHong Zhang     nzl   = bi[i+1] - bi[i];       /* num of nozeros in L(i,:) */
30753c2a7987SHong Zhang     bjtmp = bj + bi[i];
3076b78a477dSHong Zhang     for  (j=0; j<nzl; j++) rtmp[*bjtmp++] = 0.0;
3077b78a477dSHong Zhang     rtmp[i] = 0.0;
3078b78a477dSHong Zhang     nzu   = bdiag[i] - bdiag[i+1]; /* num of nozeros in U(i,:) */
3079b78a477dSHong Zhang     bjtmp = bj + bdiag[i+1] + 1;
3080b78a477dSHong Zhang     for  (j=0; j<nzu; j++) rtmp[*bjtmp++] = 0.0;
30813c2a7987SHong Zhang 
3082b78a477dSHong Zhang     /* load in initial unfactored row of A */
30836da515a1SHong Zhang     /* printf("row %d\n",i); */
30843c2a7987SHong Zhang     nz    = ai[r[i]+1] - ai[r[i]];
30853c2a7987SHong Zhang     ajtmp = aj + ai[r[i]];
30863c2a7987SHong Zhang     v     = aa + ai[r[i]];
30873c2a7987SHong Zhang     for (j=0; j<nz; j++) {
3088b78a477dSHong Zhang       rtmp[ics[*ajtmp++]] = v[j];
30896da515a1SHong Zhang       /* printf(" (%d,%g),",ics[ajtmp[j]],rtmp[ics[ajtmp[j]]]); */
30903c2a7987SHong Zhang     }
30916da515a1SHong Zhang     /* printf("\n"); */
30923c2a7987SHong Zhang 
3093b78a477dSHong Zhang     /* numerical factorization */
3094b78a477dSHong Zhang     bjtmp = bj + bi[i]; /* point to 1st entry of L(i,:) */
3095b78a477dSHong Zhang     nzl   = bi[i+1] - bi[i]; /* num of entries in L(i,:) */
3096b78a477dSHong Zhang     k = 0;
3097b78a477dSHong Zhang     while (k < nzl){
30983c2a7987SHong Zhang       row   = *bjtmp++;
30996da515a1SHong Zhang       /* printf("  prow %d\n",row); */
31003c2a7987SHong Zhang       pc         = rtmp + row;
3101b78a477dSHong Zhang       pv         = b->a + bdiag[row]; /* 1./(diag of the pivot row) */
3102b78a477dSHong Zhang       multiplier = (*pc) * (*pv);
31033c2a7987SHong Zhang       *pc        = multiplier;
3104b78a477dSHong Zhang       if (PetscAbsScalar(multiplier) > dt){
3105b78a477dSHong Zhang         pj         = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */
3106b78a477dSHong Zhang         pv         = b->a + bdiag[row+1] + 1;
3107b78a477dSHong Zhang         nz         = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */
3108b78a477dSHong Zhang         for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++);
31096da515a1SHong Zhang         /* ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); */
31103c2a7987SHong Zhang       }
3111b78a477dSHong Zhang       k++;
31123c2a7987SHong Zhang     }
31133c2a7987SHong Zhang 
3114b78a477dSHong Zhang     /* finished row so stick it into b->a */
3115b78a477dSHong Zhang     /* L-part */
3116b78a477dSHong Zhang     pv = b->a + bi[i] ;
3117b78a477dSHong Zhang     pj = bj + bi[i] ;
3118b78a477dSHong Zhang     nzl = bi[i+1] - bi[i];
3119b78a477dSHong Zhang     for (j=0; j<nzl; j++) {
3120b78a477dSHong Zhang       pv[j] = rtmp[pj[j]];
31216da515a1SHong Zhang       /* printf(" (%d,%g),",pj[j],pv[j]); */
31223c2a7987SHong Zhang     }
3123b78a477dSHong Zhang 
3124b78a477dSHong Zhang     /* diagonal: invert diagonal entries for simplier triangular solves */
3125b78a477dSHong Zhang     if (rtmp[i] == 0.0) rtmp[i] = dt+shift;
3126b78a477dSHong Zhang     b->a[bdiag[i]] = 1.0/rtmp[i];
31276da515a1SHong Zhang     /* printf(" (%d,%g),",i,b->a[bdiag[i]]); */
3128b78a477dSHong Zhang 
3129b78a477dSHong Zhang     /* U-part */
3130b78a477dSHong Zhang     pv = b->a + bdiag[i+1] + 1;
3131b78a477dSHong Zhang     pj = bj + bdiag[i+1] + 1;
3132b78a477dSHong Zhang     nzu = bdiag[i] - bdiag[i+1] - 1;
3133b78a477dSHong Zhang     for (j=0; j<nzu; j++) {
3134b78a477dSHong Zhang       pv[j] = rtmp[pj[j]];
31356da515a1SHong Zhang       /* printf(" (%d,%g),",pj[j],pv[j]); */
3136b78a477dSHong Zhang     }
31376da515a1SHong Zhang     /* printf("\n"); */
3138b78a477dSHong Zhang   }
3139b78a477dSHong Zhang 
31403c2a7987SHong Zhang   ierr = PetscFree(rtmp);CHKERRQ(ierr);
31413c2a7987SHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
31423c2a7987SHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
3143b78a477dSHong Zhang 
31443c2a7987SHong Zhang   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
31453c2a7987SHong Zhang   ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr);
31463c2a7987SHong Zhang   if (row_identity && col_identity) {
3147894a2336SHong Zhang     C->ops->solve   = MatSolve_SeqAIJ_NaturalOrdering_newdatastruct;
31483c2a7987SHong Zhang   } else {
3149894a2336SHong Zhang     C->ops->solve   = MatSolve_SeqAIJ_newdatastruct;
31503c2a7987SHong Zhang   }
3151b78a477dSHong Zhang   C->ops->solveadd           = 0;
3152b78a477dSHong Zhang   C->ops->solvetranspose     = 0;
3153b78a477dSHong Zhang   C->ops->solvetransposeadd  = 0;
3154b78a477dSHong Zhang   C->ops->matsolve           = 0;
31553c2a7987SHong Zhang   C->assembled    = PETSC_TRUE;
31563c2a7987SHong Zhang   C->preallocated = PETSC_TRUE;
31573c2a7987SHong Zhang   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
31583c2a7987SHong Zhang   PetscFunctionReturn(0);
31593c2a7987SHong Zhang }
3160