xref: /petsc/src/mat/impls/aij/seq/aijfact.c (revision 5f44c854a1643e5474f3e58a6ba8329779ac3923)
1be1d678aSKris Buschelman #define PETSCMAT_DLL
2289bc588SBarry Smith 
3b377110cSBarry Smith 
47c4f633dSBarry Smith #include "../src/mat/impls/aij/seq/aij.h"
51393bc97SHong Zhang #include "petscbt.h"
67c4f633dSBarry Smith #include "../src/mat/utils/freespace.h"
73b2fbd54SBarry Smith 
8f6a26d55SMatthew Knepley EXTERN_C_BEGIN
94a2ae208SSatish Balay #undef __FUNCT__
104a2ae208SSatish Balay #define __FUNCT__ "MatOrdering_Flow_SeqAIJ"
118fa24674SBarry Smith /*
128fa24674SBarry Smith       Computes an ordering to get most of the large numerical values in the lower triangular part of the matrix
138fa24674SBarry Smith */
14dfbe8321SBarry Smith PetscErrorCode MatOrdering_Flow_SeqAIJ(Mat mat,const MatOrderingType type,IS *irow,IS *icol)
153b2fbd54SBarry Smith {
168fa24674SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)mat->data;
178fa24674SBarry Smith   PetscErrorCode    ierr;
18889a8dedSBarry Smith   PetscInt          i,j,jj,k, kk,n = mat->rmap->n, current = 0, newcurrent = 0,*order;
198fa24674SBarry Smith   const PetscInt    *ai = a->i, *aj = a->j;
208fa24674SBarry Smith   const PetscScalar *aa = a->a;
218fa24674SBarry Smith   PetscTruth        *done;
22889a8dedSBarry Smith   PetscReal         best,past = 0,future;
233a40ed3dSBarry Smith 
248fa24674SBarry Smith   PetscFunctionBegin;
258fa24674SBarry Smith   /* pick initial row */
268fa24674SBarry Smith   best = -1;
278fa24674SBarry Smith   for (i=0; i<n; i++) {
288fa24674SBarry Smith     future = 0;
298fa24674SBarry Smith     for (j=ai[i]; j<ai[i+1]; j++) {
308fa24674SBarry Smith       if (aj[j] != i) future  += PetscAbsScalar(aa[j]); else past = PetscAbsScalar(aa[j]);
318fa24674SBarry Smith     }
328fa24674SBarry Smith     if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */
338fa24674SBarry Smith     if (past/future > best) {
348fa24674SBarry Smith       best = past/future;
358fa24674SBarry Smith       current = i;
368fa24674SBarry Smith     }
378fa24674SBarry Smith   }
388fa24674SBarry Smith 
398fa24674SBarry Smith   ierr = PetscMalloc(n*sizeof(PetscTruth),&done);CHKERRQ(ierr);
408fa24674SBarry Smith   ierr = PetscMalloc(n*sizeof(PetscInt),&order);CHKERRQ(ierr);
418fa24674SBarry Smith   ierr = PetscMemzero(done,n*sizeof(PetscTruth));CHKERRQ(ierr);
428fa24674SBarry Smith   order[0] = current;
438fa24674SBarry Smith   for (i=0; i<n-1; i++) {
448fa24674SBarry Smith     done[current] = PETSC_TRUE;
458fa24674SBarry Smith     best          = -1;
468fa24674SBarry Smith     /* loop over all neighbors of current pivot */
478fa24674SBarry Smith     for (j=ai[current]; j<ai[current+1]; j++) {
488fa24674SBarry Smith       jj = aj[j];
498fa24674SBarry Smith       if (done[jj]) continue;
508fa24674SBarry Smith       /* loop over columns of potential next row computing weights for below and above diagonal */
518fa24674SBarry Smith       past = future = 0.0;
528fa24674SBarry Smith       for (k=ai[jj]; k<ai[jj+1]; k++) {
538fa24674SBarry Smith         kk = aj[k];
548fa24674SBarry Smith         if (done[kk]) past += PetscAbsScalar(aa[k]);
558fa24674SBarry Smith         else if (kk != jj) future  += PetscAbsScalar(aa[k]);
568fa24674SBarry Smith       }
578fa24674SBarry Smith       if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */
588fa24674SBarry Smith       if (past/future > best) {
598fa24674SBarry Smith         best = past/future;
608fa24674SBarry Smith         newcurrent = jj;
618fa24674SBarry Smith       }
628fa24674SBarry Smith     }
638fa24674SBarry Smith     if (best == -1) { /* no neighbors to select from so select best of all that remain */
648fa24674SBarry Smith       best = -1;
658fa24674SBarry Smith       for (k=0; k<n; k++) {
668fa24674SBarry Smith         if (done[k]) continue;
678fa24674SBarry Smith         future = 0;
688fa24674SBarry Smith         past   = 0;
698fa24674SBarry Smith         for (j=ai[k]; j<ai[k+1]; j++) {
708fa24674SBarry Smith           kk = aj[j];
718fa24674SBarry Smith           if (done[kk]) past += PetscAbsScalar(aa[j]);
728fa24674SBarry Smith           else if (kk != k) future  += PetscAbsScalar(aa[j]);
738fa24674SBarry Smith         }
748fa24674SBarry Smith         if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */
758fa24674SBarry Smith         if (past/future > best) {
768fa24674SBarry Smith           best = past/future;
778fa24674SBarry Smith           newcurrent = k;
788fa24674SBarry Smith         }
798fa24674SBarry Smith       }
808fa24674SBarry Smith     }
818fa24674SBarry Smith     if (current == newcurrent) SETERRQ(PETSC_ERR_PLIB,"newcurrent cannot be current");
828fa24674SBarry Smith     current = newcurrent;
838fa24674SBarry Smith     order[i+1] = current;
848fa24674SBarry Smith   }
858fa24674SBarry Smith   ierr = ISCreateGeneral(PETSC_COMM_SELF,n,order,irow);CHKERRQ(ierr);
868fa24674SBarry Smith   *icol = *irow;
878fa24674SBarry Smith   ierr = PetscObjectReference((PetscObject)*irow);CHKERRQ(ierr);
888fa24674SBarry Smith   ierr = PetscFree(done);CHKERRQ(ierr);
898fa24674SBarry Smith   ierr = PetscFree(order);CHKERRQ(ierr);
90d64ed03dSBarry Smith   PetscFunctionReturn(0);
913b2fbd54SBarry Smith }
92f6a26d55SMatthew Knepley EXTERN_C_END
933b2fbd54SBarry Smith 
94e631078cSBarry Smith EXTERN_C_BEGIN
954a2ae208SSatish Balay #undef __FUNCT__
96db4efbfdSBarry Smith #define __FUNCT__ "MatGetFactorAvailable_seqaij_petsc"
97db4efbfdSBarry Smith PetscErrorCode MatGetFactorAvailable_seqaij_petsc(Mat A,MatFactorType ftype,PetscTruth *flg)
98db4efbfdSBarry Smith {
99db4efbfdSBarry Smith   PetscFunctionBegin;
100db4efbfdSBarry Smith   *flg = PETSC_TRUE;
101db4efbfdSBarry Smith   PetscFunctionReturn(0);
102db4efbfdSBarry Smith }
103db4efbfdSBarry Smith EXTERN_C_END
104db4efbfdSBarry Smith 
105db4efbfdSBarry Smith EXTERN_C_BEGIN
106db4efbfdSBarry Smith #undef __FUNCT__
107b24902e0SBarry Smith #define __FUNCT__ "MatGetFactor_seqaij_petsc"
108b24902e0SBarry Smith PetscErrorCode MatGetFactor_seqaij_petsc(Mat A,MatFactorType ftype,Mat *B)
109b24902e0SBarry Smith {
110d0f46423SBarry Smith   PetscInt           n = A->rmap->n;
111b24902e0SBarry Smith   PetscErrorCode     ierr;
112b24902e0SBarry Smith 
113b24902e0SBarry Smith   PetscFunctionBegin;
114b24902e0SBarry Smith   ierr = MatCreate(((PetscObject)A)->comm,B);CHKERRQ(ierr);
115b24902e0SBarry Smith   ierr = MatSetSizes(*B,n,n,n,n);CHKERRQ(ierr);
116599ef60dSHong Zhang   if (ftype == MAT_FACTOR_LU || ftype == MAT_FACTOR_ILU || ftype == MAT_FACTOR_ILUDT){
117b24902e0SBarry Smith     ierr = MatSetType(*B,MATSEQAIJ);CHKERRQ(ierr);
118db4efbfdSBarry Smith     (*B)->ops->ilufactorsymbolic = MatILUFactorSymbolic_SeqAIJ;
11935bd34faSBarry Smith     (*B)->ops->lufactorsymbolic  = MatLUFactorSymbolic_SeqAIJ;
1201da05e5aSHong Zhang     (*B)->ops->iludtfactor       = MatILUDTFactor_SeqAIJ;
121b24902e0SBarry Smith   } else if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) {
122b24902e0SBarry Smith     ierr = MatSetType(*B,MATSEQSBAIJ);CHKERRQ(ierr);
123b24902e0SBarry Smith     ierr = MatSeqSBAIJSetPreallocation(*B,1,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
1245c9eb25fSBarry Smith     (*B)->ops->iccfactorsymbolic      = MatICCFactorSymbolic_SeqAIJ;
1255c9eb25fSBarry Smith     (*B)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqAIJ;
126b24902e0SBarry Smith   } else SETERRQ(PETSC_ERR_SUP,"Factor type not supported");
127db4efbfdSBarry Smith   (*B)->factor = ftype;
128b24902e0SBarry Smith   PetscFunctionReturn(0);
129b24902e0SBarry Smith }
130e631078cSBarry Smith EXTERN_C_END
131b24902e0SBarry Smith 
132b24902e0SBarry Smith #undef __FUNCT__
133b9617806SBarry Smith #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ"
1340481f469SBarry Smith PetscErrorCode MatLUFactorSymbolic_SeqAIJ(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
135289bc588SBarry Smith {
136416022c9SBarry Smith   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
137289bc588SBarry Smith   IS                 isicol;
1386849ba73SBarry Smith   PetscErrorCode     ierr;
1395d0c19d7SBarry Smith   const PetscInt     *r,*ic;
1405d0c19d7SBarry Smith   PetscInt           i,n=A->rmap->n,*ai=a->i,*aj=a->j;
1411393bc97SHong Zhang   PetscInt           *bi,*bj,*ajtmp;
1421393bc97SHong Zhang   PetscInt           *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im;
1439e046878SBarry Smith   PetscReal          f;
1444875ba38SHong Zhang   PetscInt           nlnk,*lnk,k,**bi_ptr;
145a1a86e44SBarry Smith   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
1461393bc97SHong Zhang   PetscBT            lnkbt;
147ce3d78c0SShri Abhyankar   PetscTruth         newdatastruct=PETSC_FALSE;
148289bc588SBarry Smith 
1493a40ed3dSBarry Smith   PetscFunctionBegin;
150ce3d78c0SShri Abhyankar   ierr = PetscOptionsGetTruth(PETSC_NULL,"-lu_new",&newdatastruct,PETSC_NULL);CHKERRQ(ierr);
151ce3d78c0SShri Abhyankar   if(newdatastruct){
152ce3d78c0SShri Abhyankar     ierr = MatLUFactorSymbolic_SeqAIJ_newdatastruct(B,A,isrow,iscol,info);CHKERRQ(ierr);
153ce3d78c0SShri Abhyankar     PetscFunctionReturn(0);
154ce3d78c0SShri Abhyankar   }
155ce3d78c0SShri Abhyankar 
156d0f46423SBarry Smith   if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_ERR_ARG_WRONG,"matrix must be square");
1574c49b128SBarry Smith   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
158ac121b3dSBarry Smith   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
159ac121b3dSBarry Smith   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
160289bc588SBarry Smith 
161289bc588SBarry Smith   /* get new row pointers */
1621393bc97SHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
1631393bc97SHong Zhang   bi[0] = 0;
1641393bc97SHong Zhang 
1651393bc97SHong Zhang   /* bdiag is location of diagonal in factor */
1661393bc97SHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
1671393bc97SHong Zhang   bdiag[0] = 0;
1681393bc97SHong Zhang 
1691393bc97SHong Zhang   /* linked list for storing column indices of the active row */
1701393bc97SHong Zhang   nlnk = n + 1;
1711393bc97SHong Zhang   ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
1721393bc97SHong Zhang 
17335baf27eSSatish Balay   ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr);
1741393bc97SHong Zhang 
1751393bc97SHong Zhang   /* initial FreeSpace size is f*(ai[n]+1) */
176d3d32019SBarry Smith   f = info->fill;
177a1a86e44SBarry Smith   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr);
1781393bc97SHong Zhang   current_space = free_space;
179289bc588SBarry Smith 
180289bc588SBarry Smith   for (i=0; i<n; i++) {
1811393bc97SHong Zhang     /* copy previous fill into linked list */
182289bc588SBarry Smith     nzi = 0;
1831393bc97SHong Zhang     nnz = ai[r[i]+1] - ai[r[i]];
1843b4a8b6dSBarry 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);
1851393bc97SHong Zhang     ajtmp = aj + ai[r[i]];
186afcc9904SHong Zhang     ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
1871393bc97SHong Zhang     nzi += nlnk;
1881393bc97SHong Zhang 
1891393bc97SHong Zhang     /* add pivot rows into linked list */
1901393bc97SHong Zhang     row = lnk[n];
1911393bc97SHong Zhang     while (row < i) {
192d1170560SHong Zhang       nzbd    = bdiag[row] - bi[row] + 1; /* num of entries in the row with column index <= row */
193d1170560SHong Zhang       ajtmp   = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */
194d1170560SHong Zhang       ierr = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr);
1951393bc97SHong Zhang       nzi += nlnk;
1961393bc97SHong Zhang       row  = lnk[row];
197289bc588SBarry Smith     }
1981393bc97SHong Zhang     bi[i+1] = bi[i] + nzi;
1991393bc97SHong Zhang     im[i]   = nzi;
2001393bc97SHong Zhang 
2011393bc97SHong Zhang     /* mark bdiag */
2021393bc97SHong Zhang     nzbd = 0;
2031393bc97SHong Zhang     nnz  = nzi;
2041393bc97SHong Zhang     k    = lnk[n];
2051393bc97SHong Zhang     while (nnz-- && k < i){
2061393bc97SHong Zhang       nzbd++;
2071393bc97SHong Zhang       k = lnk[k];
2081393bc97SHong Zhang     }
2091393bc97SHong Zhang     bdiag[i] = bi[i] + nzbd;
2101393bc97SHong Zhang 
2111393bc97SHong Zhang     /* if free space is not available, make more free space */
2121393bc97SHong Zhang     if (current_space->local_remaining<nzi) {
2131393bc97SHong Zhang       nnz = (n - i)*nzi; /* estimated and max additional space needed */
214a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
2151393bc97SHong Zhang       reallocs++;
2161393bc97SHong Zhang     }
2171393bc97SHong Zhang 
2181393bc97SHong Zhang     /* copy data into free space, then initialize lnk */
2191393bc97SHong Zhang     ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
2201393bc97SHong Zhang     bi_ptr[i] = current_space->array;
2211393bc97SHong Zhang     current_space->array           += nzi;
2221393bc97SHong Zhang     current_space->local_used      += nzi;
2231393bc97SHong Zhang     current_space->local_remaining -= nzi;
224289bc588SBarry Smith   }
2256cf91177SBarry Smith #if defined(PETSC_USE_INFO)
226464e8d44SSatish Balay   if (ai[n] != 0) {
2271393bc97SHong Zhang     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
228ae15b995SBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr);
229ae15b995SBarry Smith     ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
230ae15b995SBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr);
231ae15b995SBarry Smith     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
23205bf559cSSatish Balay   } else {
233ae15b995SBarry Smith     ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr);
23405bf559cSSatish Balay   }
23563ba0a88SBarry Smith #endif
2362fb3ed76SBarry Smith 
237898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
238898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
2391987afe7SBarry Smith 
2401393bc97SHong Zhang   /* destroy list of free space and other temporary array(s) */
2411393bc97SHong Zhang   ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
242a1a86e44SBarry Smith   ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr);
2431393bc97SHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
24435baf27eSSatish Balay   ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr);
245289bc588SBarry Smith 
246289bc588SBarry Smith   /* put together the new matrix */
247719d5645SBarry Smith   ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
248719d5645SBarry Smith   ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr);
249719d5645SBarry Smith   b    = (Mat_SeqAIJ*)(B)->data;
250e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
251e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
2527c922b88SBarry Smith   b->singlemalloc = PETSC_FALSE;
2531393bc97SHong Zhang   ierr          = PetscMalloc((bi[n]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);
2541393bc97SHong Zhang   b->j          = bj;
2551393bc97SHong Zhang   b->i          = bi;
2561393bc97SHong Zhang   b->diag       = bdiag;
257416022c9SBarry Smith   b->ilen       = 0;
258416022c9SBarry Smith   b->imax       = 0;
259416022c9SBarry Smith   b->row        = isrow;
260416022c9SBarry Smith   b->col        = iscol;
261c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
262c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
26382bf6240SBarry Smith   b->icol       = isicol;
26487828ca2SBarry Smith   ierr          = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
2651393bc97SHong Zhang 
2661393bc97SHong Zhang   /* In b structure:  Free imax, ilen, old a, old j.  Allocate solve_work, new a, new j */
267719d5645SBarry Smith   ierr = PetscLogObjectMemory(B,(bi[n]-n)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
2681393bc97SHong Zhang   b->maxnz = b->nz = bi[n] ;
2697fd98bd6SLois Curfman McInnes 
270719d5645SBarry Smith   (B)->factor                = MAT_FACTOR_LU;
271719d5645SBarry Smith   (B)->info.factor_mallocs   = reallocs;
272719d5645SBarry Smith   (B)->info.fill_ratio_given = f;
273703deb49SSatish Balay 
2749f7d0b68SHong Zhang   if (ai[n]) {
275719d5645SBarry Smith     (B)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]);
276eea2913aSSatish Balay   } else {
277719d5645SBarry Smith     (B)->info.fill_ratio_needed = 0.0;
278eea2913aSSatish Balay   }
279719d5645SBarry Smith   (B)->ops->lufactornumeric  = MatLUFactorNumeric_SeqAIJ;
280719d5645SBarry Smith   (B)->ops->solve            = MatSolve_SeqAIJ;
281719d5645SBarry Smith   (B)->ops->solvetranspose   = MatSolveTranspose_SeqAIJ;
282db4efbfdSBarry Smith   /* switch to inodes if appropriate */
283719d5645SBarry Smith   ierr = MatLUFactorSymbolic_Inode(B,A,isrow,iscol,info);CHKERRQ(ierr);
2843a40ed3dSBarry Smith   PetscFunctionReturn(0);
285289bc588SBarry Smith }
2861393bc97SHong Zhang 
287ce3d78c0SShri Abhyankar #undef __FUNCT__
288ce3d78c0SShri Abhyankar #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ_newdatastruct"
289ce3d78c0SShri Abhyankar PetscErrorCode MatLUFactorSymbolic_SeqAIJ_newdatastruct(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
290ce3d78c0SShri Abhyankar {
291ce3d78c0SShri Abhyankar   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
292ce3d78c0SShri Abhyankar   IS                 isicol;
293ce3d78c0SShri Abhyankar   PetscErrorCode     ierr;
294ce3d78c0SShri Abhyankar   const PetscInt     *r,*ic;
295ce3d78c0SShri Abhyankar   PetscInt           i,n=A->rmap->n,*ai=a->i,*aj=a->j;
296ce3d78c0SShri Abhyankar   PetscInt           *bi,*bj,*ajtmp;
297ce3d78c0SShri Abhyankar   PetscInt           *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im;
298ce3d78c0SShri Abhyankar   PetscReal          f;
299ce3d78c0SShri Abhyankar   PetscInt           nlnk,*lnk,k,**bi_ptr;
300ce3d78c0SShri Abhyankar   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
301ce3d78c0SShri Abhyankar   PetscBT            lnkbt;
302ce3d78c0SShri Abhyankar 
303ce3d78c0SShri Abhyankar   PetscFunctionBegin;
304ce3d78c0SShri Abhyankar   if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_ERR_ARG_WRONG,"matrix must be square");
305ce3d78c0SShri Abhyankar   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
306ce3d78c0SShri Abhyankar   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
307ce3d78c0SShri Abhyankar   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
308ce3d78c0SShri Abhyankar 
309ce3d78c0SShri Abhyankar   /* get new row pointers */
310ce3d78c0SShri Abhyankar   ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
311ce3d78c0SShri Abhyankar   bi[0] = 0;
312ce3d78c0SShri Abhyankar 
313ce3d78c0SShri Abhyankar   /* bdiag is location of diagonal in factor */
314ce3d78c0SShri Abhyankar   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
315ce3d78c0SShri Abhyankar   bdiag[0] = 0;
316ce3d78c0SShri Abhyankar 
317ce3d78c0SShri Abhyankar   /* linked list for storing column indices of the active row */
318ce3d78c0SShri Abhyankar   nlnk = n + 1;
319ce3d78c0SShri Abhyankar   ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
320ce3d78c0SShri Abhyankar 
321ce3d78c0SShri Abhyankar   ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr);
322ce3d78c0SShri Abhyankar 
323ce3d78c0SShri Abhyankar   /* initial FreeSpace size is f*(ai[n]+1) */
324ce3d78c0SShri Abhyankar   f = info->fill;
325ce3d78c0SShri Abhyankar   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr);
326ce3d78c0SShri Abhyankar   current_space = free_space;
327ce3d78c0SShri Abhyankar 
328ce3d78c0SShri Abhyankar   for (i=0; i<n; i++) {
329ce3d78c0SShri Abhyankar     /* copy previous fill into linked list */
330ce3d78c0SShri Abhyankar     nzi = 0;
331ce3d78c0SShri Abhyankar     nnz = ai[r[i]+1] - ai[r[i]];
332ce3d78c0SShri 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);
333ce3d78c0SShri Abhyankar     ajtmp = aj + ai[r[i]];
334ce3d78c0SShri Abhyankar     ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
335ce3d78c0SShri Abhyankar     nzi += nlnk;
336ce3d78c0SShri Abhyankar 
337ce3d78c0SShri Abhyankar     /* add pivot rows into linked list */
338ce3d78c0SShri Abhyankar     row = lnk[n];
339ce3d78c0SShri Abhyankar     while (row < i){
34070319c84SHong Zhang       nzbd  = bdiag[row] + 1; /* num of entries in the row with column index <= row */
341ce3d78c0SShri Abhyankar       ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */
342ce3d78c0SShri Abhyankar       ierr  = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr);
343ce3d78c0SShri Abhyankar       nzi  += nlnk;
344ce3d78c0SShri Abhyankar       row   = lnk[row];
345ce3d78c0SShri Abhyankar     }
346ce3d78c0SShri Abhyankar     bi[i+1] = bi[i] + nzi;
347ce3d78c0SShri Abhyankar     im[i]   = nzi;
348ce3d78c0SShri Abhyankar 
349ce3d78c0SShri Abhyankar     /* mark bdiag */
350ce3d78c0SShri Abhyankar     nzbd = 0;
351ce3d78c0SShri Abhyankar     nnz  = nzi;
352ce3d78c0SShri Abhyankar     k    = lnk[n];
353ce3d78c0SShri Abhyankar     while (nnz-- && k < i){
354ce3d78c0SShri Abhyankar       nzbd++;
355ce3d78c0SShri Abhyankar       k = lnk[k];
356ce3d78c0SShri Abhyankar     }
35770319c84SHong Zhang     bdiag[i] = nzbd; /* note: bdiag[i] = nnzL as input for PetscFreeSpaceContiguous_newdatastruct() */
358ce3d78c0SShri Abhyankar 
359ce3d78c0SShri Abhyankar     /* if free space is not available, make more free space */
360ce3d78c0SShri Abhyankar     if (current_space->local_remaining<nzi) {
361ce3d78c0SShri Abhyankar       nnz = 2*(n - i)*nzi; /* estimated and max additional space needed */
362ce3d78c0SShri Abhyankar       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
363ce3d78c0SShri Abhyankar       reallocs++;
364ce3d78c0SShri Abhyankar     }
365ce3d78c0SShri Abhyankar 
366ce3d78c0SShri Abhyankar     /* copy data into free space, then initialize lnk */
367ce3d78c0SShri Abhyankar     ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
368ce3d78c0SShri Abhyankar     bi_ptr[i] = current_space->array;
369ce3d78c0SShri Abhyankar     current_space->array           += nzi;
370ce3d78c0SShri Abhyankar     current_space->local_used      += nzi;
371ce3d78c0SShri Abhyankar     current_space->local_remaining -= nzi;
372ce3d78c0SShri Abhyankar   }
373ce3d78c0SShri Abhyankar #if defined(PETSC_USE_INFO)
374ce3d78c0SShri Abhyankar   if (ai[n] != 0) {
375ce3d78c0SShri Abhyankar     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
376ce3d78c0SShri Abhyankar     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr);
377ce3d78c0SShri Abhyankar     ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
378ce3d78c0SShri Abhyankar     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr);
379ce3d78c0SShri Abhyankar     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
380ce3d78c0SShri Abhyankar   } else {
381ce3d78c0SShri Abhyankar     ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr);
382ce3d78c0SShri Abhyankar   }
383ce3d78c0SShri Abhyankar #endif
384ce3d78c0SShri Abhyankar 
385ce3d78c0SShri Abhyankar   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
386ce3d78c0SShri Abhyankar   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
387ce3d78c0SShri Abhyankar 
388ce3d78c0SShri Abhyankar   /* destroy list of free space and other temporary array(s) */
389ce3d78c0SShri Abhyankar   ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
390ce3d78c0SShri Abhyankar   ierr = PetscFreeSpaceContiguous_newdatastruct(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr);
391ce3d78c0SShri Abhyankar   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
392ce3d78c0SShri Abhyankar   ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr);
393ce3d78c0SShri Abhyankar 
394ce3d78c0SShri Abhyankar   /* put together the new matrix */
395ce3d78c0SShri Abhyankar   ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
396ce3d78c0SShri Abhyankar   ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr);
397ce3d78c0SShri Abhyankar   b    = (Mat_SeqAIJ*)(B)->data;
398ce3d78c0SShri Abhyankar   b->free_a       = PETSC_TRUE;
399ce3d78c0SShri Abhyankar   b->free_ij      = PETSC_TRUE;
400ce3d78c0SShri Abhyankar   b->singlemalloc = PETSC_FALSE;
401ce3d78c0SShri Abhyankar   ierr          = PetscMalloc((bi[2*n+1])*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);
402ce3d78c0SShri Abhyankar   b->j          = bj;
403ce3d78c0SShri Abhyankar   b->i          = bi;
404ce3d78c0SShri Abhyankar   b->diag       = bdiag;
405ce3d78c0SShri Abhyankar   b->ilen       = 0;
406ce3d78c0SShri Abhyankar   b->imax       = 0;
407ce3d78c0SShri Abhyankar   b->row        = isrow;
408ce3d78c0SShri Abhyankar   b->col        = iscol;
409ce3d78c0SShri Abhyankar   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
410ce3d78c0SShri Abhyankar   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
411ce3d78c0SShri Abhyankar   b->icol       = isicol;
412ce3d78c0SShri Abhyankar   ierr          = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
413ce3d78c0SShri Abhyankar 
414ce3d78c0SShri Abhyankar   /* In b structure:  Free imax, ilen, old a, old j.  Allocate solve_work, new a, new j */
415ce3d78c0SShri Abhyankar   ierr = PetscLogObjectMemory(B,bi[2*n+1]*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
416ce3d78c0SShri Abhyankar   b->maxnz = b->nz = bi[2*n+1] ;
417ce3d78c0SShri Abhyankar 
418ce3d78c0SShri Abhyankar   (B)->factor                = MAT_FACTOR_LU;
419ce3d78c0SShri Abhyankar   (B)->info.factor_mallocs   = reallocs;
420ce3d78c0SShri Abhyankar   (B)->info.fill_ratio_given = f;
421ce3d78c0SShri Abhyankar 
422ce3d78c0SShri Abhyankar   if (ai[n]) {
423894a2336SHong Zhang     (B)->info.fill_ratio_needed = ((PetscReal)bi[2*n+1])/((PetscReal)ai[n]);
424ce3d78c0SShri Abhyankar   } else {
425ce3d78c0SShri Abhyankar     (B)->info.fill_ratio_needed = 0.0;
426ce3d78c0SShri Abhyankar   }
427ce3d78c0SShri Abhyankar   (B)->ops->lufactornumeric  = MatLUFactorNumeric_SeqAIJ_newdatastruct;
428ce3d78c0SShri Abhyankar   PetscFunctionReturn(0);
429ce3d78c0SShri Abhyankar }
430ce3d78c0SShri Abhyankar 
4315b5bc046SBarry Smith /*
4325b5bc046SBarry Smith     Trouble in factorization, should we dump the original matrix?
4335b5bc046SBarry Smith */
4345b5bc046SBarry Smith #undef __FUNCT__
4355b5bc046SBarry Smith #define __FUNCT__ "MatFactorDumpMatrix"
4365b5bc046SBarry Smith PetscErrorCode MatFactorDumpMatrix(Mat A)
4375b5bc046SBarry Smith {
4385b5bc046SBarry Smith   PetscErrorCode ierr;
43990d69ab7SBarry Smith   PetscTruth     flg = PETSC_FALSE;
4405b5bc046SBarry Smith 
4415b5bc046SBarry Smith   PetscFunctionBegin;
44290d69ab7SBarry Smith   ierr = PetscOptionsGetTruth(PETSC_NULL,"-mat_factor_dump_on_error",&flg,PETSC_NULL);CHKERRQ(ierr);
4435b5bc046SBarry Smith   if (flg) {
4445b5bc046SBarry Smith     PetscViewer viewer;
4455b5bc046SBarry Smith     char        filename[PETSC_MAX_PATH_LEN];
4465b5bc046SBarry Smith 
4475b5bc046SBarry Smith     ierr = PetscSNPrintf(filename,PETSC_MAX_PATH_LEN,"matrix_factor_error.%d",PetscGlobalRank);CHKERRQ(ierr);
4487adad957SLisandro Dalcin     ierr = PetscViewerBinaryOpen(((PetscObject)A)->comm,filename,FILE_MODE_WRITE,&viewer);CHKERRQ(ierr);
4495b5bc046SBarry Smith     ierr = MatView(A,viewer);CHKERRQ(ierr);
4505b5bc046SBarry Smith     ierr = PetscViewerDestroy(viewer);CHKERRQ(ierr);
4515b5bc046SBarry Smith   }
4525b5bc046SBarry Smith   PetscFunctionReturn(0);
4535b5bc046SBarry Smith }
4545b5bc046SBarry Smith 
455db4efbfdSBarry Smith extern PetscErrorCode MatSolve_Inode(Mat,Vec,Vec);
456db4efbfdSBarry Smith 
4570a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
458894a2336SHong Zhang extern PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering_newdatastruct(Mat,Vec,Vec);
459894a2336SHong Zhang extern PetscErrorCode MatSolve_SeqAIJ_newdatastruct(Mat,Vec,Vec);
460c9c72f8fSHong Zhang 
461c9c72f8fSHong Zhang #undef __FUNCT__
462c9c72f8fSHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_newdatastruct"
463c9c72f8fSHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ_newdatastruct(Mat B,Mat A,const MatFactorInfo *info)
464c9c72f8fSHong Zhang {
465c9c72f8fSHong Zhang   Mat            C=B;
466c9c72f8fSHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data;
467c9c72f8fSHong Zhang   IS             isrow = b->row,isicol = b->icol;
468c9c72f8fSHong Zhang   PetscErrorCode ierr;
469c9c72f8fSHong Zhang   const PetscInt *r,*ic,*ics;
470c9c72f8fSHong Zhang   PetscInt       i,j,k,n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j;
471c9c72f8fSHong Zhang   PetscInt       *ajtmp,*bjtmp,nz,nzL,row,*bdiag=b->diag,*pj;
472c9c72f8fSHong Zhang   MatScalar      *rtmp,*pc,multiplier,*v,*pv,*aa=a->a;
473c9c72f8fSHong Zhang   PetscReal      shift=info->shiftinblocks;
474c9c72f8fSHong Zhang   PetscTruth     row_identity,col_identity;
475c9c72f8fSHong Zhang 
476c9c72f8fSHong Zhang   PetscFunctionBegin;
477c9c72f8fSHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
478c9c72f8fSHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
479c9c72f8fSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr);
480c9c72f8fSHong Zhang   ics  = ic;
481c9c72f8fSHong Zhang 
482c9c72f8fSHong Zhang   for (i=0; i<n; i++){
483c9c72f8fSHong Zhang     /* zero rtmp */
484c9c72f8fSHong Zhang     /* L part */
485c9c72f8fSHong Zhang     nz    = bi[i+1] - bi[i];
486c9c72f8fSHong Zhang     bjtmp = bj + bi[i];
487c9c72f8fSHong Zhang     for  (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
488c9c72f8fSHong Zhang 
489c9c72f8fSHong Zhang     /* U part */
4908fc3a347SHong Zhang     nz = bi[2*n-i+1] - bi[2*n-i];
4918fc3a347SHong Zhang     bjtmp = bj + bi[2*n-i];
492c9c72f8fSHong Zhang     for  (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
493c9c72f8fSHong Zhang 
494c9c72f8fSHong Zhang     /* load in initial (unfactored row) */
495c9c72f8fSHong Zhang     nz    = ai[r[i]+1] - ai[r[i]];
496c9c72f8fSHong Zhang     ajtmp = aj + ai[r[i]];
497c9c72f8fSHong Zhang     v     = aa + ai[r[i]];
498c9c72f8fSHong Zhang     for (j=0; j<nz; j++) {
499c9c72f8fSHong Zhang       rtmp[ics[ajtmp[j]]] = v[j];
500c9c72f8fSHong Zhang     }
501c9c72f8fSHong Zhang     if (rtmp[ics[r[i]]] == 0.0){
502c9c72f8fSHong Zhang       rtmp[ics[r[i]]] += shift; /* shift the diagonal of the matrix */
503c9c72f8fSHong Zhang       /* printf("row %d, shift %g\n",i,shift); */
504c9c72f8fSHong Zhang     }
505c9c72f8fSHong Zhang 
506c9c72f8fSHong Zhang     /* elimination */
507c9c72f8fSHong Zhang     bjtmp = bj + bi[i];
508c9c72f8fSHong Zhang     row   = *bjtmp++;
509c9c72f8fSHong Zhang     nzL   = bi[i+1] - bi[i];
510c9c72f8fSHong Zhang     k   = 0;
511c9c72f8fSHong Zhang     while  (k < nzL) {
512c9c72f8fSHong Zhang       pc = rtmp + row;
513c9c72f8fSHong Zhang       if (*pc != 0.0) {
514c9c72f8fSHong Zhang         pv         = b->a + bdiag[row];
515c9c72f8fSHong Zhang         multiplier = *pc * (*pv);
516c9c72f8fSHong Zhang         *pc        = multiplier;
5178fc3a347SHong Zhang         pj         = b->j + bi[2*n-row]; /* begining of U(row,:) */
5188fc3a347SHong Zhang         pv         = b->a + bi[2*n-row];
5198fc3a347SHong Zhang         nz         = bi[2*n-row+1] - bi[2*n-row] - 1; /* num of entries in U(row,:), excluding diag */
520c9c72f8fSHong Zhang         for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j];
521c9c72f8fSHong Zhang         ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr);
522c9c72f8fSHong Zhang       }
523c9c72f8fSHong Zhang       row = *bjtmp++; k++;
524c9c72f8fSHong Zhang     }
525c9c72f8fSHong Zhang 
526c9c72f8fSHong Zhang     /* finished row so stick it into b->a */
527c9c72f8fSHong Zhang     /* L part */
528c9c72f8fSHong Zhang     pv   = b->a + bi[i] ;
529c9c72f8fSHong Zhang     pj   = b->j + bi[i] ;
530c9c72f8fSHong Zhang     nz   = bi[i+1] - bi[i];
531c9c72f8fSHong Zhang     for (j=0; j<nz; j++) {
532c9c72f8fSHong Zhang       pv[j] = rtmp[pj[j]];
533c9c72f8fSHong Zhang     }
534c9c72f8fSHong Zhang 
535c9c72f8fSHong Zhang     /* Mark diagonal and invert diagonal for simplier triangular solves */
536c9c72f8fSHong Zhang     pv  = b->a + bdiag[i];
537c9c72f8fSHong Zhang     pj  = b->j + bdiag[i];
538c9c72f8fSHong Zhang     /* if (*pj != i)SETERRQ2(PETSC_ERR_SUP,"row %d != *pj %d",i,*pj) */
539c9c72f8fSHong Zhang     *pv = 1.0/rtmp[*pj];
540c9c72f8fSHong Zhang 
541c9c72f8fSHong Zhang     /* U part */
5428fc3a347SHong Zhang     pv = b->a + bi[2*n-i];
5438fc3a347SHong Zhang     pj = b->j + bi[2*n-i];
5448fc3a347SHong Zhang     nz = bi[2*n-i+1] - bi[2*n-i] - 1;
5458fc3a347SHong Zhang     for (j=0; j<nz; j++) pv[j] = rtmp[pj[j]];
546c9c72f8fSHong Zhang   }
547c9c72f8fSHong Zhang   ierr = PetscFree(rtmp);CHKERRQ(ierr);
548c9c72f8fSHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
549c9c72f8fSHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
550c9c72f8fSHong Zhang 
551c9c72f8fSHong Zhang   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
552c9c72f8fSHong Zhang   ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr);
553c9c72f8fSHong Zhang   if (row_identity && col_identity) {
554894a2336SHong Zhang     C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_newdatastruct;
555c9c72f8fSHong Zhang   } else {
556894a2336SHong Zhang     C->ops->solve = MatSolve_SeqAIJ_newdatastruct;
557c9c72f8fSHong Zhang   }
558c9c72f8fSHong Zhang 
559c9c72f8fSHong Zhang   C->ops->solveadd           = 0;
560c9c72f8fSHong Zhang   C->ops->solvetranspose     = 0;
561c9c72f8fSHong Zhang   C->ops->solvetransposeadd  = 0;
562c9c72f8fSHong Zhang   C->ops->matsolve           = 0;
563c9c72f8fSHong Zhang   C->assembled    = PETSC_TRUE;
564c9c72f8fSHong Zhang   C->preallocated = PETSC_TRUE;
565c9c72f8fSHong Zhang   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
566c9c72f8fSHong Zhang   PetscFunctionReturn(0);
567c9c72f8fSHong Zhang }
568c9c72f8fSHong Zhang 
5694a2ae208SSatish Balay #undef __FUNCT__
5704a2ae208SSatish Balay #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ"
5710481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info)
572289bc588SBarry Smith {
573719d5645SBarry Smith   Mat             C=B;
574416022c9SBarry Smith   Mat_SeqAIJ      *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data;
57582bf6240SBarry Smith   IS              isrow = b->row,isicol = b->icol;
5766849ba73SBarry Smith   PetscErrorCode  ierr;
5775d0c19d7SBarry Smith   const PetscInt   *r,*ic,*ics;
578d278edc7SBarry Smith   PetscInt        nz,row,i,j,n=A->rmap->n,diag;
579d278edc7SBarry Smith   const PetscInt  *ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j;
580d278edc7SBarry Smith   const PetscInt  *ajtmp,*bjtmp,*diag_offset = b->diag,*pj;
581d278edc7SBarry Smith   MatScalar       *pv,*rtmp,*pc,multiplier,d;
582d278edc7SBarry Smith   const MatScalar *v,*aa=a->a;
58333f9893dSHong Zhang   PetscReal       rs=0.0;
584b3bf805bSHong Zhang   LUShift_Ctx     sctx;
58542898933SHong Zhang   PetscInt        newshift,*ddiag;
586289bc588SBarry Smith 
5873a40ed3dSBarry Smith   PetscFunctionBegin;
58878b31e54SBarry Smith   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
58978b31e54SBarry Smith   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
590b78a477dSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr);
5919f7d0b68SHong Zhang   ics  = ic;
592289bc588SBarry Smith 
593ada7143aSSatish Balay   sctx.shift_top      = 0;
594ada7143aSSatish Balay   sctx.nshift_max     = 0;
595ada7143aSSatish Balay   sctx.shift_lo       = 0;
596ada7143aSSatish Balay   sctx.shift_hi       = 0;
5970481f469SBarry Smith   sctx.shift_fraction = 0;
598ada7143aSSatish Balay 
599118f5789SBarry Smith   /* if both shift schemes are chosen by user, only use info->shiftpd */
6001a890ab1SHong Zhang   if (info->shiftpd) { /* set sctx.shift_top=max{rs} */
60142898933SHong Zhang     ddiag          = a->diag;
6029f7d0b68SHong Zhang     sctx.shift_top = info->zeropivot;
6036cc28720Svictorle     for (i=0; i<n; i++) {
6049f95998fSHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
6059f7d0b68SHong Zhang       d  = (aa)[ddiag[i]];
6061a890ab1SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
6079f7d0b68SHong Zhang       v  = aa+ai[i];
6089f7d0b68SHong Zhang       nz = ai[i+1] - ai[i];
609e24cfe64SHong Zhang       for (j=0; j<nz; j++)
6101a890ab1SHong Zhang 	rs += PetscAbsScalar(v[j]);
6111a890ab1SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
6126cc28720Svictorle     }
613118f5789SBarry Smith     sctx.shift_top   *= 1.1;
614d2276718SHong Zhang     sctx.nshift_max   = 5;
615d2276718SHong Zhang     sctx.shift_lo     = 0.;
616d2276718SHong Zhang     sctx.shift_hi     = 1.;
617a0a356efSHong Zhang   }
618a0a356efSHong Zhang 
6199f7d0b68SHong Zhang   sctx.shift_amount = 0.0;
620a0a356efSHong Zhang   sctx.nshift       = 0;
621085256b3SBarry Smith   do {
622d2276718SHong Zhang     sctx.lushift = PETSC_FALSE;
623289bc588SBarry Smith     for (i=0; i<n; i++){
624b3bf805bSHong Zhang       nz    = bi[i+1] - bi[i];
625b3bf805bSHong Zhang       bjtmp = bj + bi[i];
6269f7d0b68SHong Zhang       for  (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
627289bc588SBarry Smith 
628289bc588SBarry Smith       /* load in initial (unfactored row) */
6299f7d0b68SHong Zhang       nz    = ai[r[i]+1] - ai[r[i]];
6309f7d0b68SHong Zhang       ajtmp = aj + ai[r[i]];
6319f7d0b68SHong Zhang       v     = aa + ai[r[i]];
632085256b3SBarry Smith       for (j=0; j<nz; j++) {
633b3bf805bSHong Zhang         rtmp[ics[ajtmp[j]]] = v[j];
634085256b3SBarry Smith       }
635d2276718SHong Zhang       rtmp[ics[r[i]]] += sctx.shift_amount; /* shift the diagonal of the matrix */
636bd1bc851SBarry Smith       /* if (sctx.shift_amount > 0.0) printf("row %d, shift %g\n",i,sctx.shift_amount); */
637289bc588SBarry Smith 
638b3bf805bSHong Zhang       row = *bjtmp++;
639f2582111SSatish Balay       while  (row < i) {
6408c37ef55SBarry Smith         pc = rtmp + row;
6418c37ef55SBarry Smith         if (*pc != 0.0) {
642010a20caSHong Zhang           pv         = b->a + diag_offset[row];
643010a20caSHong Zhang           pj         = b->j + diag_offset[row] + 1;
64435aab85fSBarry Smith           multiplier = *pc / *pv++;
6458c37ef55SBarry Smith           *pc        = multiplier;
646b3bf805bSHong Zhang           nz         = bi[row+1] - diag_offset[row] - 1;
6479f7d0b68SHong Zhang           for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j];
648dc0b31edSSatish Balay           ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr);
649289bc588SBarry Smith         }
650b3bf805bSHong Zhang         row = *bjtmp++;
651289bc588SBarry Smith       }
652416022c9SBarry Smith       /* finished row so stick it into b->a */
653b3bf805bSHong Zhang       pv   = b->a + bi[i] ;
654b3bf805bSHong Zhang       pj   = b->j + bi[i] ;
655b3bf805bSHong Zhang       nz   = bi[i+1] - bi[i];
656b3bf805bSHong Zhang       diag = diag_offset[i] - bi[i];
657e57ff13aSBarry Smith       rs   = 0.0;
658d3d32019SBarry Smith       for (j=0; j<nz; j++) {
6599f7d0b68SHong Zhang         pv[j] = rtmp[pj[j]];
6609f7d0b68SHong Zhang         rs   += PetscAbsScalar(pv[j]);
661d3d32019SBarry Smith       }
662e57ff13aSBarry Smith       rs   -= PetscAbsScalar(pv[diag]);
663d2276718SHong Zhang 
664b3bf805bSHong Zhang       /* 9/13/02 Victor Eijkhout suggested scaling zeropivot by rs for matrices with funny scalings */
665d2276718SHong Zhang       sctx.rs  = rs;
666d2276718SHong Zhang       sctx.pv  = pv[diag];
667c6b1f410SHong Zhang       ierr = MatLUCheckShift_inline(info,sctx,i,newshift);CHKERRQ(ierr);
6685b5bc046SBarry Smith       if (newshift == 1) break;
66935aab85fSBarry Smith     }
670d2276718SHong Zhang 
6710481f469SBarry Smith     if (info->shiftpd && !sctx.lushift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) {
6726cc28720Svictorle       /*
6736c037d1bSvictorle        * if no shift in this attempt & shifting & started shifting & can refine,
6746cc28720Svictorle        * then try lower shift
6756cc28720Svictorle        */
6760481f469SBarry Smith       sctx.shift_hi       = sctx.shift_fraction;
6770481f469SBarry Smith       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
6780481f469SBarry Smith       sctx.shift_amount   = sctx.shift_fraction * sctx.shift_top;
679d2276718SHong Zhang       sctx.lushift        = PETSC_TRUE;
680d2276718SHong Zhang       sctx.nshift++;
6816cc28720Svictorle     }
682d2276718SHong Zhang   } while (sctx.lushift);
6830f11f9aeSSatish Balay 
68435aab85fSBarry Smith   /* invert diagonal entries for simplier triangular solves */
68535aab85fSBarry Smith   for (i=0; i<n; i++) {
686010a20caSHong Zhang     b->a[diag_offset[i]] = 1.0/b->a[diag_offset[i]];
68735aab85fSBarry Smith   }
688606d414cSSatish Balay   ierr = PetscFree(rtmp);CHKERRQ(ierr);
68978b31e54SBarry Smith   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
69078b31e54SBarry Smith   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
691db4efbfdSBarry Smith   if (b->inode.use) {
692db4efbfdSBarry Smith     C->ops->solve   = MatSolve_Inode;
693db4efbfdSBarry Smith   } else {
6948d8c7c43SBarry Smith     PetscTruth row_identity, col_identity;
6958d8c7c43SBarry Smith     ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
6968d8c7c43SBarry Smith     ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr);
6978d8c7c43SBarry Smith     if (row_identity && col_identity) {
6988d8c7c43SBarry Smith       C->ops->solve   = MatSolve_SeqAIJ_NaturalOrdering;
6998d8c7c43SBarry Smith     } else {
700db4efbfdSBarry Smith       C->ops->solve   = MatSolve_SeqAIJ;
701db4efbfdSBarry Smith     }
7028d8c7c43SBarry Smith   }
703db4efbfdSBarry Smith   C->ops->solveadd           = MatSolveAdd_SeqAIJ;
704db4efbfdSBarry Smith   C->ops->solvetranspose     = MatSolveTranspose_SeqAIJ;
705db4efbfdSBarry Smith   C->ops->solvetransposeadd  = MatSolveTransposeAdd_SeqAIJ;
706de26497bSBarry Smith   C->ops->matsolve           = MatMatSolve_SeqAIJ;
707c456f294SBarry Smith   C->assembled    = PETSC_TRUE;
7085c9eb25fSBarry Smith   C->preallocated = PETSC_TRUE;
709d0f46423SBarry Smith   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
710d2276718SHong Zhang   if (sctx.nshift){
711e738e422SBarry Smith      if (info->shiftpd) {
7120481f469SBarry 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);
713e738e422SBarry Smith     } else if (info->shiftnz) {
714e738e422SBarry Smith       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
7156cc28720Svictorle     }
7160697b6caSHong Zhang   }
7173a40ed3dSBarry Smith   PetscFunctionReturn(0);
718289bc588SBarry Smith }
7190889b5dcSvictorle 
720137fb511SHong Zhang /*
721137fb511SHong Zhang    This routine implements inplace ILU(0) with row or/and column permutations.
722137fb511SHong Zhang    Input:
723137fb511SHong Zhang      A - original matrix
724137fb511SHong Zhang    Output;
725137fb511SHong Zhang      A - a->i (rowptr) is same as original rowptr, but factored i-the row is stored in rowperm[i]
726137fb511SHong Zhang          a->j (col index) is permuted by the inverse of colperm, then sorted
727137fb511SHong Zhang          a->a reordered accordingly with a->j
728137fb511SHong Zhang          a->diag (ptr to diagonal elements) is updated.
729137fb511SHong Zhang */
730137fb511SHong Zhang #undef __FUNCT__
731137fb511SHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_InplaceWithPerm"
7320481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_SeqAIJ_InplaceWithPerm(Mat B,Mat A,const MatFactorInfo *info)
733137fb511SHong Zhang {
734b051339dSHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
735b051339dSHong Zhang   IS             isrow = a->row,isicol = a->icol;
736137fb511SHong Zhang   PetscErrorCode ierr;
7375d0c19d7SBarry Smith   const PetscInt *r,*ic,*ics;
7385d0c19d7SBarry Smith   PetscInt       i,j,n=A->rmap->n,*ai=a->i,*aj=a->j;
7395d0c19d7SBarry Smith   PetscInt       *ajtmp,nz,row;
740b051339dSHong Zhang   PetscInt       *diag = a->diag,nbdiag,*pj;
741a77337e4SBarry Smith   PetscScalar    *rtmp,*pc,multiplier,d;
742a77337e4SBarry Smith   MatScalar      *v,*pv;
743137fb511SHong Zhang   PetscReal      rs;
744137fb511SHong Zhang   LUShift_Ctx    sctx;
745b051339dSHong Zhang   PetscInt       newshift;
746137fb511SHong Zhang 
747137fb511SHong Zhang   PetscFunctionBegin;
748719d5645SBarry Smith   if (A != B) SETERRQ(PETSC_ERR_ARG_INCOMP,"input and output matrix must have same address");
749137fb511SHong Zhang   ierr  = ISGetIndices(isrow,&r);CHKERRQ(ierr);
750137fb511SHong Zhang   ierr  = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
751137fb511SHong Zhang   ierr  = PetscMalloc((n+1)*sizeof(PetscScalar),&rtmp);CHKERRQ(ierr);
752137fb511SHong Zhang   ierr  = PetscMemzero(rtmp,(n+1)*sizeof(PetscScalar));CHKERRQ(ierr);
753b051339dSHong Zhang   ics = ic;
754137fb511SHong Zhang 
755137fb511SHong Zhang   sctx.shift_top      = 0;
756137fb511SHong Zhang   sctx.nshift_max     = 0;
757137fb511SHong Zhang   sctx.shift_lo       = 0;
758137fb511SHong Zhang   sctx.shift_hi       = 0;
7590481f469SBarry Smith   sctx.shift_fraction = 0;
760137fb511SHong Zhang 
761137fb511SHong Zhang   /* if both shift schemes are chosen by user, only use info->shiftpd */
762137fb511SHong Zhang   if (info->shiftpd) { /* set sctx.shift_top=max{rs} */
763137fb511SHong Zhang     sctx.shift_top = 0;
764137fb511SHong Zhang     for (i=0; i<n; i++) {
765137fb511SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
766b051339dSHong Zhang       d  = (a->a)[diag[i]];
767137fb511SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
768b051339dSHong Zhang       v  = a->a+ai[i];
769b051339dSHong Zhang       nz = ai[i+1] - ai[i];
770137fb511SHong Zhang       for (j=0; j<nz; j++)
771137fb511SHong Zhang 	rs += PetscAbsScalar(v[j]);
772137fb511SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
773137fb511SHong Zhang     }
774137fb511SHong Zhang     if (sctx.shift_top < info->zeropivot) sctx.shift_top = info->zeropivot;
775137fb511SHong Zhang     sctx.shift_top    *= 1.1;
776137fb511SHong Zhang     sctx.nshift_max   = 5;
777137fb511SHong Zhang     sctx.shift_lo     = 0.;
778137fb511SHong Zhang     sctx.shift_hi     = 1.;
779137fb511SHong Zhang   }
780137fb511SHong Zhang 
781137fb511SHong Zhang   sctx.shift_amount = 0;
782137fb511SHong Zhang   sctx.nshift       = 0;
783137fb511SHong Zhang   do {
784137fb511SHong Zhang     sctx.lushift = PETSC_FALSE;
785137fb511SHong Zhang     for (i=0; i<n; i++){
786b051339dSHong Zhang       /* load in initial unfactored row */
787b051339dSHong Zhang       nz    = ai[r[i]+1] - ai[r[i]];
788b051339dSHong Zhang       ajtmp = aj + ai[r[i]];
789b051339dSHong Zhang       v     = a->a + ai[r[i]];
790137fb511SHong Zhang       /* sort permuted ajtmp and values v accordingly */
791b051339dSHong Zhang       for (j=0; j<nz; j++) ajtmp[j] = ics[ajtmp[j]];
792137fb511SHong Zhang       ierr = PetscSortIntWithScalarArray(nz,ajtmp,v);CHKERRQ(ierr);
793137fb511SHong Zhang 
794b051339dSHong Zhang       diag[r[i]] = ai[r[i]];
795137fb511SHong Zhang       for (j=0; j<nz; j++) {
796137fb511SHong Zhang         rtmp[ajtmp[j]] = v[j];
797b051339dSHong Zhang         if (ajtmp[j] < i) diag[r[i]]++; /* update a->diag */
798137fb511SHong Zhang       }
799137fb511SHong Zhang       rtmp[r[i]] += sctx.shift_amount; /* shift the diagonal of the matrix */
800137fb511SHong Zhang 
801b051339dSHong Zhang       row = *ajtmp++;
802137fb511SHong Zhang       while  (row < i) {
803137fb511SHong Zhang         pc = rtmp + row;
804137fb511SHong Zhang         if (*pc != 0.0) {
805b051339dSHong Zhang           pv         = a->a + diag[r[row]];
806b051339dSHong Zhang           pj         = aj + diag[r[row]] + 1;
807137fb511SHong Zhang 
808137fb511SHong Zhang           multiplier = *pc / *pv++;
809137fb511SHong Zhang           *pc        = multiplier;
810b051339dSHong Zhang           nz         = ai[r[row]+1] - diag[r[row]] - 1;
811b051339dSHong Zhang           for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j];
812dc0b31edSSatish Balay           ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr);
813137fb511SHong Zhang         }
814b051339dSHong Zhang         row = *ajtmp++;
815137fb511SHong Zhang       }
816b051339dSHong Zhang       /* finished row so overwrite it onto a->a */
817b051339dSHong Zhang       pv   = a->a + ai[r[i]] ;
818b051339dSHong Zhang       pj   = aj + ai[r[i]] ;
819b051339dSHong Zhang       nz   = ai[r[i]+1] - ai[r[i]];
820b051339dSHong Zhang       nbdiag = diag[r[i]] - ai[r[i]]; /* num of entries before the diagonal */
821137fb511SHong Zhang 
822137fb511SHong Zhang       rs   = 0.0;
823137fb511SHong Zhang       for (j=0; j<nz; j++) {
824b051339dSHong Zhang         pv[j] = rtmp[pj[j]];
825b051339dSHong Zhang         if (j != nbdiag) rs += PetscAbsScalar(pv[j]);
826137fb511SHong Zhang       }
827137fb511SHong Zhang 
828137fb511SHong Zhang       /* 9/13/02 Victor Eijkhout suggested scaling zeropivot by rs for matrices with funny scalings */
829137fb511SHong Zhang       sctx.rs  = rs;
830b051339dSHong Zhang       sctx.pv  = pv[nbdiag];
831c6b1f410SHong Zhang       ierr = MatLUCheckShift_inline(info,sctx,i,newshift);CHKERRQ(ierr);
832137fb511SHong Zhang       if (newshift == 1) break;
833137fb511SHong Zhang     }
834137fb511SHong Zhang 
8350481f469SBarry Smith     if (info->shiftpd && !sctx.lushift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) {
836137fb511SHong Zhang       /*
837137fb511SHong Zhang        * if no shift in this attempt & shifting & started shifting & can refine,
838137fb511SHong Zhang        * then try lower shift
839137fb511SHong Zhang        */
8400481f469SBarry Smith       sctx.shift_hi        = sctx.shift_fraction;
8410481f469SBarry Smith       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
8420481f469SBarry Smith       sctx.shift_amount    = sctx.shift_fraction * sctx.shift_top;
843137fb511SHong Zhang       sctx.lushift         = PETSC_TRUE;
844137fb511SHong Zhang       sctx.nshift++;
845137fb511SHong Zhang     }
846137fb511SHong Zhang   } while (sctx.lushift);
847137fb511SHong Zhang 
848137fb511SHong Zhang   /* invert diagonal entries for simplier triangular solves */
849137fb511SHong Zhang   for (i=0; i<n; i++) {
850b051339dSHong Zhang     a->a[diag[r[i]]] = 1.0/a->a[diag[r[i]]];
851137fb511SHong Zhang   }
852137fb511SHong Zhang 
853137fb511SHong Zhang   ierr = PetscFree(rtmp);CHKERRQ(ierr);
854137fb511SHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
855137fb511SHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
856b051339dSHong Zhang   A->ops->solve             = MatSolve_SeqAIJ_InplaceWithPerm;
857db4efbfdSBarry Smith   A->ops->solveadd          = MatSolveAdd_SeqAIJ;
858db4efbfdSBarry Smith   A->ops->solvetranspose    = MatSolveTranspose_SeqAIJ;
859db4efbfdSBarry Smith   A->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ;
860b051339dSHong Zhang   A->assembled = PETSC_TRUE;
8615c9eb25fSBarry Smith   A->preallocated = PETSC_TRUE;
862d0f46423SBarry Smith   ierr = PetscLogFlops(A->cmap->n);CHKERRQ(ierr);
863137fb511SHong Zhang   if (sctx.nshift){
864e738e422SBarry Smith     if (info->shiftpd) {
8650481f469SBarry 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);
866e738e422SBarry Smith     } else if (info->shiftnz) {
867e738e422SBarry Smith       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
868137fb511SHong Zhang     }
869137fb511SHong Zhang   }
870137fb511SHong Zhang   PetscFunctionReturn(0);
871137fb511SHong Zhang }
872137fb511SHong Zhang 
8730a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
8744a2ae208SSatish Balay #undef __FUNCT__
8754a2ae208SSatish Balay #define __FUNCT__ "MatLUFactor_SeqAIJ"
8760481f469SBarry Smith PetscErrorCode MatLUFactor_SeqAIJ(Mat A,IS row,IS col,const MatFactorInfo *info)
877da3a660dSBarry Smith {
878dfbe8321SBarry Smith   PetscErrorCode ierr;
879416022c9SBarry Smith   Mat            C;
880416022c9SBarry Smith 
8813a40ed3dSBarry Smith   PetscFunctionBegin;
88243244d56SBarry Smith   ierr = MatGetFactor(A,MAT_SOLVER_PETSC,MAT_FACTOR_LU,&C);CHKERRQ(ierr);
883719d5645SBarry Smith   ierr = MatLUFactorSymbolic(C,A,row,col,info);CHKERRQ(ierr);
884719d5645SBarry Smith   ierr = MatLUFactorNumeric(C,A,info);CHKERRQ(ierr);
885db4efbfdSBarry Smith   A->ops->solve            = C->ops->solve;
886db4efbfdSBarry Smith   A->ops->solvetranspose   = C->ops->solvetranspose;
887273d9f13SBarry Smith   ierr = MatHeaderCopy(A,C);CHKERRQ(ierr);
88852e6d16bSBarry Smith   ierr = PetscLogObjectParent(A,((Mat_SeqAIJ*)(A->data))->icol);CHKERRQ(ierr);
8893a40ed3dSBarry Smith   PetscFunctionReturn(0);
890da3a660dSBarry Smith }
8910a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
8921d098868SHong Zhang 
893889a8dedSBarry Smith 
8944a2ae208SSatish Balay #undef __FUNCT__
8954a2ae208SSatish Balay #define __FUNCT__ "MatSolve_SeqAIJ"
896dfbe8321SBarry Smith PetscErrorCode MatSolve_SeqAIJ(Mat A,Vec bb,Vec xx)
8978c37ef55SBarry Smith {
898416022c9SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
899416022c9SBarry Smith   IS                iscol = a->col,isrow = a->row;
9006849ba73SBarry Smith   PetscErrorCode    ierr;
9015d0c19d7SBarry Smith   PetscInt          i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
9025d0c19d7SBarry Smith   PetscInt          nz;
9035d0c19d7SBarry Smith   const PetscInt    *rout,*cout,*r,*c;
904dd6ea824SBarry Smith   PetscScalar       *x,*tmp,*tmps,sum;
905d9fead3dSBarry Smith   const PetscScalar *b;
906dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
9078c37ef55SBarry Smith 
9083a40ed3dSBarry Smith   PetscFunctionBegin;
9093a40ed3dSBarry Smith   if (!n) PetscFunctionReturn(0);
91091bf9adeSBarry Smith 
911d9fead3dSBarry Smith   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
9121ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
913416022c9SBarry Smith   tmp  = a->solve_work;
9148c37ef55SBarry Smith 
9153b2fbd54SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
9163b2fbd54SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
9178c37ef55SBarry Smith 
9188c37ef55SBarry Smith   /* forward solve the lower triangular */
9198c37ef55SBarry Smith   tmp[0] = b[*r++];
920010a20caSHong Zhang   tmps   = tmp;
9218c37ef55SBarry Smith   for (i=1; i<n; i++) {
922010a20caSHong Zhang     v   = aa + ai[i] ;
923010a20caSHong Zhang     vi  = aj + ai[i] ;
92453e7425aSBarry Smith     nz  = a->diag[i] - ai[i];
92553e7425aSBarry Smith     sum = b[*r++];
926003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
9278c37ef55SBarry Smith     tmp[i] = sum;
9288c37ef55SBarry Smith   }
9298c37ef55SBarry Smith 
9308c37ef55SBarry Smith   /* backward solve the upper triangular */
9318c37ef55SBarry Smith   for (i=n-1; i>=0; i--){
932010a20caSHong Zhang     v   = aa + a->diag[i] + 1;
933010a20caSHong Zhang     vi  = aj + a->diag[i] + 1;
934416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
9358c37ef55SBarry Smith     sum = tmp[i];
936003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
937010a20caSHong Zhang     x[*c--] = tmp[i] = sum*aa[a->diag[i]];
9388c37ef55SBarry Smith   }
9398c37ef55SBarry Smith 
9403b2fbd54SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
9413b2fbd54SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
942d9fead3dSBarry Smith   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
9431ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
944dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
9453a40ed3dSBarry Smith   PetscFunctionReturn(0);
9468c37ef55SBarry Smith }
947026e39d0SSatish Balay 
9482bebee5dSHong Zhang #undef __FUNCT__
9492bebee5dSHong Zhang #define __FUNCT__ "MatMatSolve_SeqAIJ"
9502bebee5dSHong Zhang PetscErrorCode MatMatSolve_SeqAIJ(Mat A,Mat B,Mat X)
9512bebee5dSHong Zhang {
9522bebee5dSHong Zhang   Mat_SeqAIJ      *a = (Mat_SeqAIJ*)A->data;
9532bebee5dSHong Zhang   IS              iscol = a->col,isrow = a->row;
9542bebee5dSHong Zhang   PetscErrorCode  ierr;
9555d0c19d7SBarry Smith   PetscInt        i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
9565d0c19d7SBarry Smith   PetscInt        nz,neq;
9575d0c19d7SBarry Smith   const PetscInt  *rout,*cout,*r,*c;
958dd6ea824SBarry Smith   PetscScalar     *x,*b,*tmp,*tmps,sum;
959dd6ea824SBarry Smith   const MatScalar *aa = a->a,*v;
960db4efbfdSBarry Smith   PetscTruth      bisdense,xisdense;
9612bebee5dSHong Zhang 
9622bebee5dSHong Zhang   PetscFunctionBegin;
9632bebee5dSHong Zhang   if (!n) PetscFunctionReturn(0);
9642bebee5dSHong Zhang 
965db4efbfdSBarry Smith   ierr = PetscTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr);
966db4efbfdSBarry Smith   if (!bisdense) SETERRQ(PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix");
967db4efbfdSBarry Smith   ierr = PetscTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr);
968db4efbfdSBarry Smith   if (!xisdense) SETERRQ(PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix");
969db4efbfdSBarry Smith 
9702bebee5dSHong Zhang   ierr = MatGetArray(B,&b);CHKERRQ(ierr);
9712bebee5dSHong Zhang   ierr = MatGetArray(X,&x);CHKERRQ(ierr);
9722bebee5dSHong Zhang 
9732bebee5dSHong Zhang   tmp  = a->solve_work;
9742bebee5dSHong Zhang   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
9752bebee5dSHong Zhang   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
9762bebee5dSHong Zhang 
977a36b22ccSSatish Balay   for (neq=0; neq<B->cmap->n; neq++){
9782bebee5dSHong Zhang     /* forward solve the lower triangular */
9792bebee5dSHong Zhang     tmp[0] = b[r[0]];
9802bebee5dSHong Zhang     tmps   = tmp;
9812bebee5dSHong Zhang     for (i=1; i<n; i++) {
9822bebee5dSHong Zhang       v   = aa + ai[i] ;
9832bebee5dSHong Zhang       vi  = aj + ai[i] ;
9842bebee5dSHong Zhang       nz  = a->diag[i] - ai[i];
9852bebee5dSHong Zhang       sum = b[r[i]];
986003131ecSBarry Smith       PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
9872bebee5dSHong Zhang       tmp[i] = sum;
9882bebee5dSHong Zhang     }
9892bebee5dSHong Zhang     /* backward solve the upper triangular */
9902bebee5dSHong Zhang     for (i=n-1; i>=0; i--){
9912bebee5dSHong Zhang       v   = aa + a->diag[i] + 1;
9922bebee5dSHong Zhang       vi  = aj + a->diag[i] + 1;
9932bebee5dSHong Zhang       nz  = ai[i+1] - a->diag[i] - 1;
9942bebee5dSHong Zhang       sum = tmp[i];
995003131ecSBarry Smith       PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
9962bebee5dSHong Zhang       x[c[i]] = tmp[i] = sum*aa[a->diag[i]];
9972bebee5dSHong Zhang     }
9982bebee5dSHong Zhang 
9992bebee5dSHong Zhang     b += n;
10002bebee5dSHong Zhang     x += n;
10012bebee5dSHong Zhang   }
10022bebee5dSHong Zhang   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
10032bebee5dSHong Zhang   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
10042bebee5dSHong Zhang   ierr = MatRestoreArray(B,&b);CHKERRQ(ierr);
10052bebee5dSHong Zhang   ierr = MatRestoreArray(X,&x);CHKERRQ(ierr);
1006dc0b31edSSatish Balay   ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr);
10072bebee5dSHong Zhang   PetscFunctionReturn(0);
10082bebee5dSHong Zhang }
10092bebee5dSHong Zhang 
1010137fb511SHong Zhang #undef __FUNCT__
1011137fb511SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_InplaceWithPerm"
1012137fb511SHong Zhang PetscErrorCode MatSolve_SeqAIJ_InplaceWithPerm(Mat A,Vec bb,Vec xx)
1013137fb511SHong Zhang {
1014137fb511SHong Zhang   Mat_SeqAIJ      *a = (Mat_SeqAIJ*)A->data;
1015137fb511SHong Zhang   IS              iscol = a->col,isrow = a->row;
1016137fb511SHong Zhang   PetscErrorCode  ierr;
10175d0c19d7SBarry Smith   const PetscInt  *r,*c,*rout,*cout;
10185d0c19d7SBarry Smith   PetscInt        i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
10195d0c19d7SBarry Smith   PetscInt        nz,row;
1020dd6ea824SBarry Smith   PetscScalar     *x,*b,*tmp,*tmps,sum;
1021dd6ea824SBarry Smith   const MatScalar *aa = a->a,*v;
1022137fb511SHong Zhang 
1023137fb511SHong Zhang   PetscFunctionBegin;
1024137fb511SHong Zhang   if (!n) PetscFunctionReturn(0);
1025137fb511SHong Zhang 
1026137fb511SHong Zhang   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
1027137fb511SHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1028137fb511SHong Zhang   tmp  = a->solve_work;
1029137fb511SHong Zhang 
1030137fb511SHong Zhang   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
1031137fb511SHong Zhang   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
1032137fb511SHong Zhang 
1033137fb511SHong Zhang   /* forward solve the lower triangular */
1034137fb511SHong Zhang   tmp[0] = b[*r++];
1035137fb511SHong Zhang   tmps   = tmp;
1036137fb511SHong Zhang   for (row=1; row<n; row++) {
1037137fb511SHong Zhang     i   = rout[row]; /* permuted row */
1038137fb511SHong Zhang     v   = aa + ai[i] ;
1039137fb511SHong Zhang     vi  = aj + ai[i] ;
1040137fb511SHong Zhang     nz  = a->diag[i] - ai[i];
1041137fb511SHong Zhang     sum = b[*r++];
1042003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
1043137fb511SHong Zhang     tmp[row] = sum;
1044137fb511SHong Zhang   }
1045137fb511SHong Zhang 
1046137fb511SHong Zhang   /* backward solve the upper triangular */
1047137fb511SHong Zhang   for (row=n-1; row>=0; row--){
1048137fb511SHong Zhang     i   = rout[row]; /* permuted row */
1049137fb511SHong Zhang     v   = aa + a->diag[i] + 1;
1050137fb511SHong Zhang     vi  = aj + a->diag[i] + 1;
1051137fb511SHong Zhang     nz  = ai[i+1] - a->diag[i] - 1;
1052137fb511SHong Zhang     sum = tmp[row];
1053003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
1054137fb511SHong Zhang     x[*c--] = tmp[row] = sum*aa[a->diag[i]];
1055137fb511SHong Zhang   }
1056137fb511SHong Zhang 
1057137fb511SHong Zhang   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
1058137fb511SHong Zhang   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
1059137fb511SHong Zhang   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
1060137fb511SHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1061dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
1062137fb511SHong Zhang   PetscFunctionReturn(0);
1063137fb511SHong Zhang }
1064137fb511SHong Zhang 
1065930ae53cSBarry Smith /* ----------------------------------------------------------- */
1066a4005a5dSBarry Smith #include "../src/mat/impls/aij/seq/ftn-kernels/fsolve.h"
10674a2ae208SSatish Balay #undef __FUNCT__
10684a2ae208SSatish Balay #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering"
1069dfbe8321SBarry Smith PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering(Mat A,Vec bb,Vec xx)
1070930ae53cSBarry Smith {
1071930ae53cSBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
10726849ba73SBarry Smith   PetscErrorCode    ierr;
1073d0f46423SBarry Smith   PetscInt          n = A->rmap->n;
1074b377110cSBarry Smith   const PetscInt    *ai = a->i,*aj = a->j,*adiag = a->diag;
1075356650c2SBarry Smith   PetscScalar       *x;
1076356650c2SBarry Smith   const PetscScalar *b;
1077dd6ea824SBarry Smith   const MatScalar   *aa = a->a;
1078aa482453SBarry Smith #if !defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ)
1079356650c2SBarry Smith   PetscInt          adiag_i,i,nz,ai_i;
1080b377110cSBarry Smith   const PetscInt    *vi;
1081dd6ea824SBarry Smith   const MatScalar   *v;
1082dd6ea824SBarry Smith   PetscScalar       sum;
1083d85afc42SSatish Balay #endif
1084930ae53cSBarry Smith 
10853a40ed3dSBarry Smith   PetscFunctionBegin;
10863a40ed3dSBarry Smith   if (!n) PetscFunctionReturn(0);
1087930ae53cSBarry Smith 
1088356650c2SBarry Smith   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
10891ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1090930ae53cSBarry Smith 
1091aa482453SBarry Smith #if defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ)
10921c4feecaSSatish Balay   fortransolveaij_(&n,x,ai,aj,adiag,aa,b);
10931c4feecaSSatish Balay #else
1094930ae53cSBarry Smith   /* forward solve the lower triangular */
1095930ae53cSBarry Smith   x[0] = b[0];
1096930ae53cSBarry Smith   for (i=1; i<n; i++) {
1097930ae53cSBarry Smith     ai_i = ai[i];
1098930ae53cSBarry Smith     v    = aa + ai_i;
1099930ae53cSBarry Smith     vi   = aj + ai_i;
1100930ae53cSBarry Smith     nz   = adiag[i] - ai_i;
1101930ae53cSBarry Smith     sum  = b[i];
1102003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
1103930ae53cSBarry Smith     x[i] = sum;
1104930ae53cSBarry Smith   }
1105930ae53cSBarry Smith 
1106930ae53cSBarry Smith   /* backward solve the upper triangular */
1107930ae53cSBarry Smith   for (i=n-1; i>=0; i--){
1108930ae53cSBarry Smith     adiag_i = adiag[i];
1109d708749eSBarry Smith     v       = aa + adiag_i + 1;
1110d708749eSBarry Smith     vi      = aj + adiag_i + 1;
1111930ae53cSBarry Smith     nz      = ai[i+1] - adiag_i - 1;
1112930ae53cSBarry Smith     sum     = x[i];
1113003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
1114930ae53cSBarry Smith     x[i]    = sum*aa[adiag_i];
1115930ae53cSBarry Smith   }
11161c4feecaSSatish Balay #endif
1117dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
1118356650c2SBarry Smith   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
11191ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
11203a40ed3dSBarry Smith   PetscFunctionReturn(0);
1121930ae53cSBarry Smith }
1122930ae53cSBarry Smith 
11234a2ae208SSatish Balay #undef __FUNCT__
11244a2ae208SSatish Balay #define __FUNCT__ "MatSolveAdd_SeqAIJ"
1125dfbe8321SBarry Smith PetscErrorCode MatSolveAdd_SeqAIJ(Mat A,Vec bb,Vec yy,Vec xx)
1126da3a660dSBarry Smith {
1127416022c9SBarry Smith   Mat_SeqAIJ      *a = (Mat_SeqAIJ*)A->data;
1128416022c9SBarry Smith   IS              iscol = a->col,isrow = a->row;
11296849ba73SBarry Smith   PetscErrorCode  ierr;
11305d0c19d7SBarry Smith   PetscInt        i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
11315d0c19d7SBarry Smith   PetscInt        nz;
11325d0c19d7SBarry Smith   const PetscInt  *rout,*cout,*r,*c;
1133dd6ea824SBarry Smith   PetscScalar     *x,*b,*tmp,sum;
1134dd6ea824SBarry Smith   const MatScalar *aa = a->a,*v;
1135da3a660dSBarry Smith 
11363a40ed3dSBarry Smith   PetscFunctionBegin;
113778b31e54SBarry Smith   if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);}
1138da3a660dSBarry Smith 
11391ebc52fbSHong Zhang   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
11401ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1141416022c9SBarry Smith   tmp  = a->solve_work;
1142da3a660dSBarry Smith 
11433b2fbd54SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
11443b2fbd54SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
1145da3a660dSBarry Smith 
1146da3a660dSBarry Smith   /* forward solve the lower triangular */
1147da3a660dSBarry Smith   tmp[0] = b[*r++];
1148da3a660dSBarry Smith   for (i=1; i<n; i++) {
1149010a20caSHong Zhang     v   = aa + ai[i] ;
1150010a20caSHong Zhang     vi  = aj + ai[i] ;
1151416022c9SBarry Smith     nz  = a->diag[i] - ai[i];
1152da3a660dSBarry Smith     sum = b[*r++];
1153010a20caSHong Zhang     while (nz--) sum -= *v++ * tmp[*vi++ ];
1154da3a660dSBarry Smith     tmp[i] = sum;
1155da3a660dSBarry Smith   }
1156da3a660dSBarry Smith 
1157da3a660dSBarry Smith   /* backward solve the upper triangular */
1158da3a660dSBarry Smith   for (i=n-1; i>=0; i--){
1159010a20caSHong Zhang     v   = aa + a->diag[i] + 1;
1160010a20caSHong Zhang     vi  = aj + a->diag[i] + 1;
1161416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
1162da3a660dSBarry Smith     sum = tmp[i];
1163010a20caSHong Zhang     while (nz--) sum -= *v++ * tmp[*vi++ ];
1164010a20caSHong Zhang     tmp[i] = sum*aa[a->diag[i]];
1165da3a660dSBarry Smith     x[*c--] += tmp[i];
1166da3a660dSBarry Smith   }
1167da3a660dSBarry Smith 
11683b2fbd54SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
11693b2fbd54SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
11701ebc52fbSHong Zhang   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
11711ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1172dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
1173898183ecSLois Curfman McInnes 
11743a40ed3dSBarry Smith   PetscFunctionReturn(0);
1175da3a660dSBarry Smith }
1176da3a660dSBarry Smith /* -------------------------------------------------------------------*/
11774a2ae208SSatish Balay #undef __FUNCT__
11784a2ae208SSatish Balay #define __FUNCT__ "MatSolveTranspose_SeqAIJ"
1179dfbe8321SBarry Smith PetscErrorCode MatSolveTranspose_SeqAIJ(Mat A,Vec bb,Vec xx)
1180da3a660dSBarry Smith {
1181416022c9SBarry Smith   Mat_SeqAIJ      *a = (Mat_SeqAIJ*)A->data;
11822235e667SBarry Smith   IS              iscol = a->col,isrow = a->row;
11836849ba73SBarry Smith   PetscErrorCode  ierr;
11845d0c19d7SBarry Smith   const PetscInt  *rout,*cout,*r,*c;
11855d0c19d7SBarry Smith   PetscInt        i,n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
11865d0c19d7SBarry Smith   PetscInt        nz,*diag = a->diag;
1187dd6ea824SBarry Smith   PetscScalar     *x,*b,*tmp,s1;
1188dd6ea824SBarry Smith   const MatScalar *aa = a->a,*v;
1189da3a660dSBarry Smith 
11903a40ed3dSBarry Smith   PetscFunctionBegin;
11911ebc52fbSHong Zhang   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
11921ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1193416022c9SBarry Smith   tmp  = a->solve_work;
1194da3a660dSBarry Smith 
11952235e667SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
11962235e667SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
1197da3a660dSBarry Smith 
1198da3a660dSBarry Smith   /* copy the b into temp work space according to permutation */
11992235e667SBarry Smith   for (i=0; i<n; i++) tmp[i] = b[c[i]];
1200da3a660dSBarry Smith 
1201da3a660dSBarry Smith   /* forward solve the U^T */
1202da3a660dSBarry Smith   for (i=0; i<n; i++) {
1203010a20caSHong Zhang     v   = aa + diag[i] ;
1204010a20caSHong Zhang     vi  = aj + diag[i] + 1;
1205f1af5d2fSBarry Smith     nz  = ai[i+1] - diag[i] - 1;
1206c38d4ed2SBarry Smith     s1  = tmp[i];
1207ef66eb69SBarry Smith     s1 *= (*v++);  /* multiply by inverse of diagonal entry */
1208da3a660dSBarry Smith     while (nz--) {
1209010a20caSHong Zhang       tmp[*vi++ ] -= (*v++)*s1;
1210da3a660dSBarry Smith     }
1211c38d4ed2SBarry Smith     tmp[i] = s1;
1212da3a660dSBarry Smith   }
1213da3a660dSBarry Smith 
1214da3a660dSBarry Smith   /* backward solve the L^T */
1215da3a660dSBarry Smith   for (i=n-1; i>=0; i--){
1216010a20caSHong Zhang     v   = aa + diag[i] - 1 ;
1217010a20caSHong Zhang     vi  = aj + diag[i] - 1 ;
1218f1af5d2fSBarry Smith     nz  = diag[i] - ai[i];
1219f1af5d2fSBarry Smith     s1  = tmp[i];
1220da3a660dSBarry Smith     while (nz--) {
1221010a20caSHong Zhang       tmp[*vi-- ] -= (*v--)*s1;
1222da3a660dSBarry Smith     }
1223da3a660dSBarry Smith   }
1224da3a660dSBarry Smith 
1225da3a660dSBarry Smith   /* copy tmp into x according to permutation */
1226da3a660dSBarry Smith   for (i=0; i<n; i++) x[r[i]] = tmp[i];
1227da3a660dSBarry Smith 
12282235e667SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
12292235e667SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
12301ebc52fbSHong Zhang   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
12311ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1232da3a660dSBarry Smith 
1233dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr);
12343a40ed3dSBarry Smith   PetscFunctionReturn(0);
1235da3a660dSBarry Smith }
1236da3a660dSBarry Smith 
12374a2ae208SSatish Balay #undef __FUNCT__
12384a2ae208SSatish Balay #define __FUNCT__ "MatSolveTransposeAdd_SeqAIJ"
1239dfbe8321SBarry Smith PetscErrorCode MatSolveTransposeAdd_SeqAIJ(Mat A,Vec bb,Vec zz,Vec xx)
1240da3a660dSBarry Smith {
1241416022c9SBarry Smith   Mat_SeqAIJ      *a = (Mat_SeqAIJ*)A->data;
12422235e667SBarry Smith   IS              iscol = a->col,isrow = a->row;
12436849ba73SBarry Smith   PetscErrorCode  ierr;
12445d0c19d7SBarry Smith   const PetscInt  *r,*c,*rout,*cout;
12455d0c19d7SBarry Smith   PetscInt        i,n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
12465d0c19d7SBarry Smith   PetscInt        nz,*diag = a->diag;
1247dd6ea824SBarry Smith   PetscScalar     *x,*b,*tmp;
1248dd6ea824SBarry Smith   const MatScalar *aa = a->a,*v;
12496abc6512SBarry Smith 
12503a40ed3dSBarry Smith   PetscFunctionBegin;
125123bc6035SMatthew Knepley   if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);}
12526abc6512SBarry Smith 
12531ebc52fbSHong Zhang   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
12541ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1255416022c9SBarry Smith   tmp = a->solve_work;
12566abc6512SBarry Smith 
12572235e667SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
12582235e667SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
12596abc6512SBarry Smith 
12606abc6512SBarry Smith   /* copy the b into temp work space according to permutation */
12612235e667SBarry Smith   for (i=0; i<n; i++) tmp[i] = b[c[i]];
12626abc6512SBarry Smith 
12636abc6512SBarry Smith   /* forward solve the U^T */
12646abc6512SBarry Smith   for (i=0; i<n; i++) {
1265010a20caSHong Zhang     v   = aa + diag[i] ;
1266010a20caSHong Zhang     vi  = aj + diag[i] + 1;
1267f1af5d2fSBarry Smith     nz  = ai[i+1] - diag[i] - 1;
12686abc6512SBarry Smith     tmp[i] *= *v++;
12696abc6512SBarry Smith     while (nz--) {
1270010a20caSHong Zhang       tmp[*vi++ ] -= (*v++)*tmp[i];
12716abc6512SBarry Smith     }
12726abc6512SBarry Smith   }
12736abc6512SBarry Smith 
12746abc6512SBarry Smith   /* backward solve the L^T */
12756abc6512SBarry Smith   for (i=n-1; i>=0; i--){
1276010a20caSHong Zhang     v   = aa + diag[i] - 1 ;
1277010a20caSHong Zhang     vi  = aj + diag[i] - 1 ;
1278f1af5d2fSBarry Smith     nz  = diag[i] - ai[i];
12796abc6512SBarry Smith     while (nz--) {
1280010a20caSHong Zhang       tmp[*vi-- ] -= (*v--)*tmp[i];
12816abc6512SBarry Smith     }
12826abc6512SBarry Smith   }
12836abc6512SBarry Smith 
12846abc6512SBarry Smith   /* copy tmp into x according to permutation */
12856abc6512SBarry Smith   for (i=0; i<n; i++) x[r[i]] += tmp[i];
12866abc6512SBarry Smith 
12872235e667SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
12882235e667SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
12891ebc52fbSHong Zhang   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
12901ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
12916abc6512SBarry Smith 
1292dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
12933a40ed3dSBarry Smith   PetscFunctionReturn(0);
1294da3a660dSBarry Smith }
12959e25ed09SBarry Smith /* ----------------------------------------------------------------*/
12963c5fc038SBarry Smith EXTERN PetscErrorCode Mat_CheckInode(Mat,PetscTruth);
1297f77e22a1SHong Zhang EXTERN PetscErrorCode MatDuplicateNoCreate_SeqAIJ(Mat,Mat,MatDuplicateOption,PetscTruth);
129808480c60SBarry Smith 
12998fc3a347SHong Zhang /*
13008fc3a347SHong Zhang    ilu(0) with natural ordering under new data structure.
13018fc3a347SHong Zhang    Factored arrays bj and ba are stored as
13028fc3a347SHong Zhang      L(0,:), L(1,:), ...,L(n-1,:),  U(n-1,:),...,U(i,:),U(i-1,:),...,U(0,:)
13038fc3a347SHong Zhang 
13048fc3a347SHong Zhang    bi=fact->i is an array of size 2n+2, in which
13058fc3a347SHong Zhang    bi+
13068fc3a347SHong Zhang      bi[i]      ->  1st entry of L(i,:),i=0,...,i-1
1307f77e22a1SHong Zhang      bi[n]      ->  points to L(n-1,:)+1
13088fc3a347SHong Zhang      bi[n+1]    ->  1st entry of U(n-1,:)
13098fc3a347SHong Zhang      bi[2n-i]   ->  1st entry of U(i,:)
13108fc3a347SHong Zhang      bi[2n-i+1] ->  end of U(i,:)+1, the 1st entry of U(i-1,:)
1311f77e22a1SHong Zhang      bi[2n]     ->  1st entry of U(0,:)
1312f77e22a1SHong Zhang      bi[2n+1]   ->  points to U(0,:)+1
13138fc3a347SHong Zhang 
13148fc3a347SHong Zhang    U(i,:) contains diag[i] as its last entry, i.e.,
13158fc3a347SHong Zhang     U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i])
13168fc3a347SHong Zhang */
1317c9c72f8fSHong Zhang #undef __FUNCT__
13188fc3a347SHong Zhang #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_ilu0_newdatastruct"
13198fc3a347SHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ_ilu0_newdatastruct(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
1320c9c72f8fSHong Zhang {
1321c9c72f8fSHong Zhang 
1322c9c72f8fSHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
1323c9c72f8fSHong Zhang   PetscErrorCode     ierr;
1324c9c72f8fSHong Zhang   PetscInt           n=A->rmap->n,*ai=a->i,*aj,*adiag=a->diag;
13258fc3a347SHong Zhang   PetscInt           i,j,nz,*bi,*bj,*bdiag;
1326c9c72f8fSHong Zhang 
1327c9c72f8fSHong Zhang   PetscFunctionBegin;
1328f77e22a1SHong Zhang   ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_FALSE);CHKERRQ(ierr);
1329c9c72f8fSHong Zhang   b    = (Mat_SeqAIJ*)(fact)->data;
1330c9c72f8fSHong Zhang 
1331f77e22a1SHong Zhang   /* allocate matrix arrays for new data structure */
1332f77e22a1SHong Zhang   ierr = PetscMalloc3(ai[n]+1,PetscScalar,&b->a,ai[n]+1,PetscInt,&b->j,2*n+2,PetscInt,&b->i);CHKERRQ(ierr);
1333f77e22a1SHong Zhang   ierr = PetscLogObjectMemory(fact,ai[n]*(sizeof(PetscScalar)+sizeof(PetscInt))+(2*n+2)*sizeof(PetscInt));CHKERRQ(ierr);
1334f77e22a1SHong Zhang   b->singlemalloc = PETSC_TRUE;
1335f77e22a1SHong Zhang   if (!b->diag){
13368fc3a347SHong Zhang     ierr = PetscMalloc((n+1)*sizeof(PetscInt),&b->diag);CHKERRQ(ierr);
1337f77e22a1SHong Zhang   }
1338c9c72f8fSHong Zhang   bdiag = b->diag;
1339c9c72f8fSHong Zhang 
1340c9c72f8fSHong Zhang   if (n > 0) {
134116a2bf60SHong Zhang     ierr = PetscMemzero(b->a,(ai[n])*sizeof(MatScalar));CHKERRQ(ierr);
1342c9c72f8fSHong Zhang   }
1343c9c72f8fSHong Zhang 
1344c9c72f8fSHong Zhang   /* set bi and bj with new data structure */
1345c9c72f8fSHong Zhang   bi = b->i;
1346c9c72f8fSHong Zhang   bj = b->j;
1347c9c72f8fSHong Zhang 
1348c9c72f8fSHong Zhang   /* L part */
1349c9c72f8fSHong Zhang   bi[0] = 0;
1350c9c72f8fSHong Zhang   for (i=0; i<n; i++){
1351c9c72f8fSHong Zhang     nz = adiag[i] - ai[i];
1352c9c72f8fSHong Zhang     bi[i+1] = bi[i] + nz;
1353c9c72f8fSHong Zhang     aj = a->j + ai[i];
1354c9c72f8fSHong Zhang     for (j=0; j<nz; j++){
1355c9c72f8fSHong Zhang       *bj = aj[j]; bj++;
1356c9c72f8fSHong Zhang     }
1357c9c72f8fSHong Zhang   }
1358c9c72f8fSHong Zhang 
1359c9c72f8fSHong Zhang   /* U part */
13608fc3a347SHong Zhang   bi[n+1] = bi[n];
1361c9c72f8fSHong Zhang   for (i=n-1; i>=0; i--){
1362c9c72f8fSHong Zhang     nz = ai[i+1] - adiag[i] - 1;
13638fc3a347SHong Zhang     bi[2*n-i+1] = bi[2*n-i] + nz + 1;
1364c9c72f8fSHong Zhang     aj = a->j + adiag[i] + 1;
1365c9c72f8fSHong Zhang     for (j=0; j<nz; j++){
1366c9c72f8fSHong Zhang       *bj = aj[j]; bj++;
1367c9c72f8fSHong Zhang     }
1368c9c72f8fSHong Zhang     /* diag[i] */
1369c9c72f8fSHong Zhang     *bj = i; bj++;
13708fc3a347SHong Zhang     bdiag[i] = bi[2*n-i+1]-1;
1371c9c72f8fSHong Zhang   }
1372c9c72f8fSHong Zhang   PetscFunctionReturn(0);
1373c9c72f8fSHong Zhang }
1374c9c72f8fSHong Zhang 
13754a2ae208SSatish Balay #undef __FUNCT__
13766516239fSHong Zhang #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_newdatastruct"
13776516239fSHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ_newdatastruct(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
13789e25ed09SBarry Smith {
1379416022c9SBarry Smith   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
13809e25ed09SBarry Smith   IS                 isicol;
13816849ba73SBarry Smith   PetscErrorCode     ierr;
13825d0c19d7SBarry Smith   const PetscInt     *r,*ic;
13835d0c19d7SBarry Smith   PetscInt           n=A->rmap->n,*ai=a->i,*aj=a->j,d;
13845a8e39fbSHong Zhang   PetscInt           *bi,*cols,nnz,*cols_lvl;
1385052f0c41SBarry Smith   PetscInt           *bdiag,prow,fm,nzbd,reallocs=0,dcount=0;
13865a8e39fbSHong Zhang   PetscInt           i,levels,diagonal_fill;
138777c4ece6SBarry Smith   PetscTruth         col_identity,row_identity;
1388329f5518SBarry Smith   PetscReal          f;
13895a8e39fbSHong Zhang   PetscInt           nlnk,*lnk,*lnk_lvl=PETSC_NULL;
13905a8e39fbSHong Zhang   PetscBT            lnkbt;
13915a8e39fbSHong Zhang   PetscInt           nzi,*bj,**bj_ptr,**bjlvl_ptr;
1392a1a86e44SBarry Smith   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
1393a1a86e44SBarry Smith   PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL;
139409f38230SBarry Smith   PetscTruth         missing;
13959e25ed09SBarry Smith 
13963a40ed3dSBarry Smith   PetscFunctionBegin;
13976516239fSHong Zhang   //printf("MatILUFactorSymbolic_SeqAIJ_newdatastruct ...\n");
1398d0f46423SBarry 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);
13996cf997b0SBarry Smith   f             = info->fill;
140038baddfdSBarry Smith   levels        = (PetscInt)info->levels;
140138baddfdSBarry Smith   diagonal_fill = (PetscInt)info->diagonal_fill;
14024c49b128SBarry Smith   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
140382bf6240SBarry Smith 
1404be0abb6dSBarry Smith   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
1405be0abb6dSBarry Smith   ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr);
1406c9c72f8fSHong Zhang 
14076516239fSHong Zhang   if (!levels && row_identity && col_identity) {
14086516239fSHong Zhang     /* special case: ilu(0) with natural ordering */
14098fc3a347SHong Zhang     ierr = MatILUFactorSymbolic_SeqAIJ_ilu0_newdatastruct(fact,A,isrow,iscol,info);CHKERRQ(ierr);
1410c9c72f8fSHong Zhang     (fact)->ops->lufactornumeric =  MatLUFactorNumeric_SeqAIJ_newdatastruct;
14116516239fSHong Zhang 
14126516239fSHong Zhang     fact->factor = MAT_FACTOR_ILU;
14136516239fSHong Zhang     (fact)->info.factor_mallocs    = 0;
14146516239fSHong Zhang     (fact)->info.fill_ratio_given  = info->fill;
14156516239fSHong Zhang     (fact)->info.fill_ratio_needed = 1.0;
14166516239fSHong Zhang     b               = (Mat_SeqAIJ*)(fact)->data;
14176516239fSHong Zhang     ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr);
14186516239fSHong Zhang     if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d);
14196516239fSHong Zhang     b->row              = isrow;
14206516239fSHong Zhang     b->col              = iscol;
14216516239fSHong Zhang     b->icol             = isicol;
14226516239fSHong Zhang     ierr                = PetscMalloc(((fact)->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
14236516239fSHong Zhang     ierr                = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
14246516239fSHong Zhang     ierr                = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
14256516239fSHong Zhang     /* ierr = MatILUFactorSymbolic_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr); */
14266516239fSHong Zhang     PetscFunctionReturn(0);
14276516239fSHong Zhang   }
14286516239fSHong Zhang 
14296516239fSHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
14306516239fSHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
14316516239fSHong Zhang 
14326516239fSHong Zhang   /* get new row pointers */
14336516239fSHong Zhang   ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
14346516239fSHong Zhang   bi[0] = 0;
14356516239fSHong Zhang   /* bdiag is location of diagonal in factor */
14366516239fSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
14376516239fSHong Zhang   bdiag[0]  = 0;
14386516239fSHong Zhang 
14396516239fSHong Zhang   ierr = PetscMalloc((2*n+1)*sizeof(PetscInt**),&bj_ptr);CHKERRQ(ierr);
14406516239fSHong Zhang   bjlvl_ptr = (PetscInt**)(bj_ptr + n);
14416516239fSHong Zhang 
14426516239fSHong Zhang   /* create a linked list for storing column indices of the active row */
14436516239fSHong Zhang   nlnk = n + 1;
14446516239fSHong Zhang   ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
14456516239fSHong Zhang 
14466516239fSHong Zhang   /* initial FreeSpace size is f*(ai[n]+1) */
14476516239fSHong Zhang   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr);
14486516239fSHong Zhang   current_space = free_space;
14496516239fSHong Zhang   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr);
14506516239fSHong Zhang   current_space_lvl = free_space_lvl;
14516516239fSHong Zhang 
14526516239fSHong Zhang   for (i=0; i<n; i++) {
14536516239fSHong Zhang     nzi = 0;
14546516239fSHong Zhang     /* copy current row into linked list */
14556516239fSHong Zhang     nnz  = ai[r[i]+1] - ai[r[i]];
14566516239fSHong 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);
14576516239fSHong Zhang     cols = aj + ai[r[i]];
14586516239fSHong Zhang     lnk[i] = -1; /* marker to indicate if diagonal exists */
14596516239fSHong Zhang     ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
14606516239fSHong Zhang     nzi += nlnk;
14616516239fSHong Zhang 
14626516239fSHong Zhang     /* make sure diagonal entry is included */
14636516239fSHong Zhang     if (diagonal_fill && lnk[i] == -1) {
14646516239fSHong Zhang       fm = n;
14656516239fSHong Zhang       while (lnk[fm] < i) fm = lnk[fm];
14666516239fSHong Zhang       lnk[i]     = lnk[fm]; /* insert diagonal into linked list */
14676516239fSHong Zhang       lnk[fm]    = i;
14686516239fSHong Zhang       lnk_lvl[i] = 0;
14696516239fSHong Zhang       nzi++; dcount++;
14706516239fSHong Zhang     }
14716516239fSHong Zhang 
14726516239fSHong Zhang     /* add pivot rows into the active row */
14736516239fSHong Zhang     nzbd = 0;
14746516239fSHong Zhang     prow = lnk[n];
14756516239fSHong Zhang     while (prow < i) {
14766516239fSHong Zhang       nnz      = bdiag[prow];
14776516239fSHong Zhang       cols     = bj_ptr[prow] + nnz + 1;
14786516239fSHong Zhang       cols_lvl = bjlvl_ptr[prow] + nnz + 1;
14796516239fSHong Zhang       nnz      = bi[prow+1] - bi[prow] - nnz - 1;
14806516239fSHong Zhang       ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr);
14816516239fSHong Zhang       nzi += nlnk;
14826516239fSHong Zhang       prow = lnk[prow];
14836516239fSHong Zhang       nzbd++;
14846516239fSHong Zhang     }
14856516239fSHong Zhang     bdiag[i] = nzbd;
14866516239fSHong Zhang     bi[i+1]  = bi[i] + nzi;
14876516239fSHong Zhang 
14886516239fSHong Zhang     /* if free space is not available, make more free space */
14896516239fSHong Zhang     if (current_space->local_remaining<nzi) {
14906516239fSHong Zhang       nnz = 2*nzi*(n - i); /* estimated and max additional space needed */
14916516239fSHong Zhang       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
14926516239fSHong Zhang       ierr = PetscFreeSpaceGet(nnz,&current_space_lvl);CHKERRQ(ierr);
14936516239fSHong Zhang       reallocs++;
14946516239fSHong Zhang     }
14956516239fSHong Zhang 
14966516239fSHong Zhang     /* copy data into free_space and free_space_lvl, then initialize lnk */
14976516239fSHong Zhang     ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
14986516239fSHong Zhang     bj_ptr[i]    = current_space->array;
14996516239fSHong Zhang     bjlvl_ptr[i] = current_space_lvl->array;
15006516239fSHong Zhang 
15016516239fSHong Zhang     /* make sure the active row i has diagonal entry */
15026516239fSHong Zhang     if (*(bj_ptr[i]+bdiag[i]) != i) {
15036516239fSHong Zhang       SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,"Row %D has missing diagonal in factored matrix\n\
15046516239fSHong Zhang     try running with -pc_factor_nonzeros_along_diagonal or -pc_factor_diagonal_fill",i);
15056516239fSHong Zhang     }
15066516239fSHong Zhang 
15076516239fSHong Zhang     current_space->array           += nzi;
15086516239fSHong Zhang     current_space->local_used      += nzi;
15096516239fSHong Zhang     current_space->local_remaining -= nzi;
15106516239fSHong Zhang     current_space_lvl->array           += nzi;
15116516239fSHong Zhang     current_space_lvl->local_used      += nzi;
15126516239fSHong Zhang     current_space_lvl->local_remaining -= nzi;
15136516239fSHong Zhang   }
15146516239fSHong Zhang 
15156516239fSHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
15166516239fSHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
15176516239fSHong Zhang 
15186516239fSHong Zhang   /* destroy list of free space and other temporary arrays */
15196516239fSHong Zhang   ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
15206516239fSHong Zhang 
15216516239fSHong Zhang   /* copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */
15226516239fSHong Zhang   ierr = PetscFreeSpaceContiguous_newdatastruct(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr);
15236516239fSHong Zhang 
15246516239fSHong Zhang   ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
15256516239fSHong Zhang   ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
15266516239fSHong Zhang   ierr = PetscFree(bj_ptr);CHKERRQ(ierr);
15276516239fSHong Zhang 
15286516239fSHong Zhang #if defined(PETSC_USE_INFO)
15296516239fSHong Zhang   {
15306516239fSHong Zhang     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
15316516239fSHong Zhang     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr);
15326516239fSHong Zhang     ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
15336516239fSHong Zhang     ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr);
15346516239fSHong Zhang     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
15356516239fSHong Zhang     if (diagonal_fill) {
15366516239fSHong Zhang       ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr);
15376516239fSHong Zhang     }
15386516239fSHong Zhang   }
15396516239fSHong Zhang #endif
15406516239fSHong Zhang 
15416516239fSHong Zhang   /* put together the new matrix */
15426516239fSHong Zhang   ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
15436516239fSHong Zhang   ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr);
15446516239fSHong Zhang   b = (Mat_SeqAIJ*)(fact)->data;
15456516239fSHong Zhang   b->free_a       = PETSC_TRUE;
15466516239fSHong Zhang   b->free_ij      = PETSC_TRUE;
15476516239fSHong Zhang   b->singlemalloc = PETSC_FALSE;
15486516239fSHong Zhang   ierr = PetscMalloc( (bi[2*n+1] )*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);
15496516239fSHong Zhang   b->j          = bj;
15506516239fSHong Zhang   b->i          = bi;
15516516239fSHong Zhang   b->diag       = bdiag;
15526516239fSHong Zhang   b->ilen       = 0;
15536516239fSHong Zhang   b->imax       = 0;
15546516239fSHong Zhang   b->row        = isrow;
15556516239fSHong Zhang   b->col        = iscol;
15566516239fSHong Zhang   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
15576516239fSHong Zhang   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
15586516239fSHong Zhang   b->icol       = isicol;
15596516239fSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
15606516239fSHong Zhang   /* In b structure:  Free imax, ilen, old a, old j.
15616516239fSHong Zhang      Allocate bdiag, solve_work, new a, new j */
15626516239fSHong Zhang   ierr = PetscLogObjectMemory(fact,bi[2*n+1] * (sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
15636516239fSHong Zhang   b->maxnz = b->nz = bi[2*n+1] ;
15646516239fSHong Zhang   (fact)->info.factor_mallocs    = reallocs;
15656516239fSHong Zhang   (fact)->info.fill_ratio_given  = f;
15666516239fSHong Zhang   (fact)->info.fill_ratio_needed = ((PetscReal)bi[2*n+1])/((PetscReal)ai[n]);
15676516239fSHong Zhang   (fact)->ops->lufactornumeric   = MatLUFactorNumeric_SeqAIJ_newdatastruct;
15686516239fSHong Zhang   /* ierr = MatILUFactorSymbolic_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr); */
15696516239fSHong Zhang   PetscFunctionReturn(0);
15706516239fSHong Zhang }
15716516239fSHong Zhang 
15726516239fSHong Zhang #undef __FUNCT__
15736516239fSHong Zhang #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ"
15746516239fSHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
15756516239fSHong Zhang {
15766516239fSHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
15776516239fSHong Zhang   IS                 isicol;
15786516239fSHong Zhang   PetscErrorCode     ierr;
15796516239fSHong Zhang   const PetscInt     *r,*ic;
15806516239fSHong Zhang   PetscInt           n=A->rmap->n,*ai=a->i,*aj=a->j,d;
15816516239fSHong Zhang   PetscInt           *bi,*cols,nnz,*cols_lvl;
15826516239fSHong Zhang   PetscInt           *bdiag,prow,fm,nzbd,reallocs=0,dcount=0;
15836516239fSHong Zhang   PetscInt           i,levels,diagonal_fill;
15846516239fSHong Zhang   PetscTruth         col_identity,row_identity;
15856516239fSHong Zhang   PetscReal          f;
15866516239fSHong Zhang   PetscInt           nlnk,*lnk,*lnk_lvl=PETSC_NULL;
15876516239fSHong Zhang   PetscBT            lnkbt;
15886516239fSHong Zhang   PetscInt           nzi,*bj,**bj_ptr,**bjlvl_ptr;
15896516239fSHong Zhang   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
15906516239fSHong Zhang   PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL;
15916516239fSHong Zhang   PetscTruth         missing;
15926516239fSHong Zhang   PetscTruth         newdatastruct=PETSC_FALSE;
15936516239fSHong Zhang 
15946516239fSHong Zhang   PetscFunctionBegin;
15956516239fSHong Zhang   ierr = PetscOptionsGetTruth(PETSC_NULL,"-ilu_new",&newdatastruct,PETSC_NULL);CHKERRQ(ierr);
15966516239fSHong Zhang   if (newdatastruct){
15976516239fSHong Zhang     ierr = MatILUFactorSymbolic_SeqAIJ_newdatastruct(fact,A,isrow,iscol,info);CHKERRQ(ierr);
15986516239fSHong Zhang     PetscFunctionReturn(0);
15996516239fSHong Zhang   }
16006516239fSHong Zhang 
16016516239fSHong 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);
16026516239fSHong Zhang   f             = info->fill;
16036516239fSHong Zhang   levels        = (PetscInt)info->levels;
16046516239fSHong Zhang   diagonal_fill = (PetscInt)info->diagonal_fill;
16056516239fSHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
16066516239fSHong Zhang 
16076516239fSHong Zhang   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
16086516239fSHong Zhang   ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr);
16096516239fSHong Zhang   if (!levels && row_identity && col_identity) { /* special case: ilu(0) with natural ordering */
1610f77e22a1SHong Zhang     ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_TRUE);CHKERRQ(ierr);
1611c9c72f8fSHong Zhang     (fact)->ops->lufactornumeric =  MatLUFactorNumeric_SeqAIJ;
1612c9c72f8fSHong Zhang 
1613719d5645SBarry Smith     fact->factor = MAT_FACTOR_ILU;
1614719d5645SBarry Smith     (fact)->info.factor_mallocs    = 0;
1615719d5645SBarry Smith     (fact)->info.fill_ratio_given  = info->fill;
1616719d5645SBarry Smith     (fact)->info.fill_ratio_needed = 1.0;
1617719d5645SBarry Smith     b               = (Mat_SeqAIJ*)(fact)->data;
16188ef3462fSBarry Smith     ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr);
161909f38230SBarry Smith     if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d);
162008480c60SBarry Smith     b->row              = isrow;
162108480c60SBarry Smith     b->col              = iscol;
162282bf6240SBarry Smith     b->icol             = isicol;
1623719d5645SBarry Smith     ierr                = PetscMalloc(((fact)->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
1624c38d4ed2SBarry Smith     ierr                = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
1625c38d4ed2SBarry Smith     ierr                = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
1626719d5645SBarry Smith     ierr = MatILUFactorSymbolic_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr);
16273a40ed3dSBarry Smith     PetscFunctionReturn(0);
162808480c60SBarry Smith   }
162908480c60SBarry Smith 
1630898183ecSLois Curfman McInnes   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
1631898183ecSLois Curfman McInnes   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
16329e25ed09SBarry Smith 
16339e25ed09SBarry Smith   /* get new row pointers */
16345a8e39fbSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
16355a8e39fbSHong Zhang   bi[0] = 0;
16365a8e39fbSHong Zhang   /* bdiag is location of diagonal in factor */
16375a8e39fbSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
16385a8e39fbSHong Zhang   bdiag[0]  = 0;
16396cf997b0SBarry Smith 
16405a8e39fbSHong Zhang   ierr = PetscMalloc((2*n+1)*sizeof(PetscInt**),&bj_ptr);CHKERRQ(ierr);
16415a8e39fbSHong Zhang   bjlvl_ptr = (PetscInt**)(bj_ptr + n);
16425a8e39fbSHong Zhang 
16435a8e39fbSHong Zhang   /* create a linked list for storing column indices of the active row */
16445a8e39fbSHong Zhang   nlnk = n + 1;
16455a8e39fbSHong Zhang   ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
16465a8e39fbSHong Zhang 
16475a8e39fbSHong Zhang   /* initial FreeSpace size is f*(ai[n]+1) */
1648a1a86e44SBarry Smith   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr);
16495a8e39fbSHong Zhang   current_space = free_space;
1650a1a86e44SBarry Smith   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr);
16515a8e39fbSHong Zhang   current_space_lvl = free_space_lvl;
16525a8e39fbSHong Zhang 
16535a8e39fbSHong Zhang   for (i=0; i<n; i++) {
16545a8e39fbSHong Zhang     nzi = 0;
16556cf997b0SBarry Smith     /* copy current row into linked list */
16565a8e39fbSHong Zhang     nnz  = ai[r[i]+1] - ai[r[i]];
16573b4a8b6dSBarry 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);
16585a8e39fbSHong Zhang     cols = aj + ai[r[i]];
16595a8e39fbSHong Zhang     lnk[i] = -1; /* marker to indicate if diagonal exists */
16605a8e39fbSHong Zhang     ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
16615a8e39fbSHong Zhang     nzi += nlnk;
16626cf997b0SBarry Smith 
16636cf997b0SBarry Smith     /* make sure diagonal entry is included */
16645a8e39fbSHong Zhang     if (diagonal_fill && lnk[i] == -1) {
16656cf997b0SBarry Smith       fm = n;
16665a8e39fbSHong Zhang       while (lnk[fm] < i) fm = lnk[fm];
16675a8e39fbSHong Zhang       lnk[i]     = lnk[fm]; /* insert diagonal into linked list */
16685a8e39fbSHong Zhang       lnk[fm]    = i;
16695a8e39fbSHong Zhang       lnk_lvl[i] = 0;
16705a8e39fbSHong Zhang       nzi++; dcount++;
16716cf997b0SBarry Smith     }
16726cf997b0SBarry Smith 
16735a8e39fbSHong Zhang     /* add pivot rows into the active row */
16745a8e39fbSHong Zhang     nzbd = 0;
16755a8e39fbSHong Zhang     prow = lnk[n];
16765a8e39fbSHong Zhang     while (prow < i) {
16775a8e39fbSHong Zhang       nnz      = bdiag[prow];
16785a8e39fbSHong Zhang       cols     = bj_ptr[prow] + nnz + 1;
16795a8e39fbSHong Zhang       cols_lvl = bjlvl_ptr[prow] + nnz + 1;
16805a8e39fbSHong Zhang       nnz      = bi[prow+1] - bi[prow] - nnz - 1;
16815a8e39fbSHong Zhang       ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr);
16825a8e39fbSHong Zhang       nzi += nlnk;
16835a8e39fbSHong Zhang       prow = lnk[prow];
16845a8e39fbSHong Zhang       nzbd++;
168556beaf04SBarry Smith     }
16865a8e39fbSHong Zhang     bdiag[i] = nzbd;
16875a8e39fbSHong Zhang     bi[i+1]  = bi[i] + nzi;
1688ecf371e4SBarry Smith 
16895a8e39fbSHong Zhang     /* if free space is not available, make more free space */
16905a8e39fbSHong Zhang     if (current_space->local_remaining<nzi) {
16915a8e39fbSHong Zhang       nnz = nzi*(n - i); /* estimated and max additional space needed */
1692a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
1693a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(nnz,&current_space_lvl);CHKERRQ(ierr);
16945a8e39fbSHong Zhang       reallocs++;
16955a8e39fbSHong Zhang     }
1696ecf371e4SBarry Smith 
16975a8e39fbSHong Zhang     /* copy data into free_space and free_space_lvl, then initialize lnk */
16985a8e39fbSHong Zhang     ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
16995a8e39fbSHong Zhang     bj_ptr[i]    = current_space->array;
17005a8e39fbSHong Zhang     bjlvl_ptr[i] = current_space_lvl->array;
17015a8e39fbSHong Zhang 
17025a8e39fbSHong Zhang     /* make sure the active row i has diagonal entry */
17035a8e39fbSHong Zhang     if (*(bj_ptr[i]+bdiag[i]) != i) {
170477431f27SBarry Smith       SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,"Row %D has missing diagonal in factored matrix\n\
1705d7ee0231SBarry Smith     try running with -pc_factor_nonzeros_along_diagonal or -pc_factor_diagonal_fill",i);
17066cf997b0SBarry Smith     }
17075a8e39fbSHong Zhang 
17085a8e39fbSHong Zhang     current_space->array           += nzi;
17095a8e39fbSHong Zhang     current_space->local_used      += nzi;
17105a8e39fbSHong Zhang     current_space->local_remaining -= nzi;
17115a8e39fbSHong Zhang     current_space_lvl->array           += nzi;
17125a8e39fbSHong Zhang     current_space_lvl->local_used      += nzi;
17135a8e39fbSHong Zhang     current_space_lvl->local_remaining -= nzi;
17149e25ed09SBarry Smith   }
17155a8e39fbSHong Zhang 
1716898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
1717898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
17185a8e39fbSHong Zhang 
17195a8e39fbSHong Zhang   /* destroy list of free space and other temporary arrays */
17205a8e39fbSHong Zhang   ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
17216516239fSHong Zhang   ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); /* copy free_space -> bj */
17225a8e39fbSHong Zhang   ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
1723a1a86e44SBarry Smith   ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
17245a8e39fbSHong Zhang   ierr = PetscFree(bj_ptr);CHKERRQ(ierr);
17259e25ed09SBarry Smith 
17266cf91177SBarry Smith #if defined(PETSC_USE_INFO)
1727f64065bbSBarry Smith   {
17285a8e39fbSHong Zhang     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
1729ae15b995SBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr);
1730ae15b995SBarry Smith     ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
1731ae15b995SBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr);
1732ae15b995SBarry Smith     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
1733335d9088SBarry Smith     if (diagonal_fill) {
1734ae15b995SBarry Smith       ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr);
1735335d9088SBarry Smith     }
1736f64065bbSBarry Smith   }
173763ba0a88SBarry Smith #endif
1738bbb6d6a8SBarry Smith 
17399e25ed09SBarry Smith   /* put together the new matrix */
174007dbf95fSBarry Smith   ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
1741719d5645SBarry Smith   ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr);
1742719d5645SBarry Smith   b = (Mat_SeqAIJ*)(fact)->data;
1743e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
1744e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
17457c922b88SBarry Smith   b->singlemalloc = PETSC_FALSE;
1746052f0c41SBarry Smith   ierr = PetscMalloc( (bi[n] )*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);
17475a8e39fbSHong Zhang   b->j          = bj;
17485a8e39fbSHong Zhang   b->i          = bi;
17495a8e39fbSHong Zhang   for (i=0; i<n; i++) bdiag[i] += bi[i];
17505a8e39fbSHong Zhang   b->diag       = bdiag;
1751416022c9SBarry Smith   b->ilen       = 0;
1752416022c9SBarry Smith   b->imax       = 0;
1753416022c9SBarry Smith   b->row        = isrow;
1754416022c9SBarry Smith   b->col        = iscol;
1755c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
1756c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
175782bf6240SBarry Smith   b->icol       = isicol;
175887828ca2SBarry Smith   ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
1759416022c9SBarry Smith   /* In b structure:  Free imax, ilen, old a, old j.
17605a8e39fbSHong Zhang      Allocate bdiag, solve_work, new a, new j */
1761719d5645SBarry Smith   ierr = PetscLogObjectMemory(fact,(bi[n]-n) * (sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
17625a8e39fbSHong Zhang   b->maxnz             = b->nz = bi[n] ;
1763719d5645SBarry Smith   (fact)->info.factor_mallocs    = reallocs;
1764719d5645SBarry Smith   (fact)->info.fill_ratio_given  = f;
1765719d5645SBarry Smith   (fact)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]);
1766719d5645SBarry Smith   (fact)->ops->lufactornumeric =  MatLUFactorNumeric_SeqAIJ;
1767719d5645SBarry Smith   ierr = MatILUFactorSymbolic_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr);
17683a40ed3dSBarry Smith   PetscFunctionReturn(0);
17699e25ed09SBarry Smith }
1770d5d45c9bSBarry Smith 
17717c4f633dSBarry Smith #include "../src/mat/impls/sbaij/seq/sbaij.h"
1772a6175056SHong Zhang #undef __FUNCT__
1773*5f44c854SHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ_newdatastruct"
1774*5f44c854SHong Zhang PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ_newdatastruct(Mat B,Mat A,const MatFactorInfo *info)
1775*5f44c854SHong Zhang {
1776*5f44c854SHong Zhang   Mat            C = B;
1777*5f44c854SHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
1778*5f44c854SHong Zhang   Mat_SeqSBAIJ   *b=(Mat_SeqSBAIJ*)C->data;
1779*5f44c854SHong Zhang   IS             ip=b->row,iip = b->icol;
1780*5f44c854SHong Zhang   PetscErrorCode ierr;
1781*5f44c854SHong Zhang   const PetscInt *rip,*riip;
1782*5f44c854SHong Zhang   PetscInt       i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bdiag=b->diag,*bjtmp;
1783*5f44c854SHong Zhang   PetscInt       *ai=a->i,*aj=a->j;
1784*5f44c854SHong Zhang   PetscInt       k,jmin,jmax,*c2r,*il,col,nexti,ili,nz;
1785*5f44c854SHong Zhang   MatScalar      *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi;
1786*5f44c854SHong Zhang   PetscReal      zeropivot,shiftnz;
1787*5f44c854SHong Zhang   PetscReal      shiftpd;
1788*5f44c854SHong Zhang   PetscTruth     perm_identity;
1789*5f44c854SHong Zhang 
1790*5f44c854SHong Zhang   PetscFunctionBegin;
1791*5f44c854SHong Zhang 
1792*5f44c854SHong Zhang   shiftnz   = info->shiftnz;
1793*5f44c854SHong Zhang   shiftpd   = info->shiftpd;
1794*5f44c854SHong Zhang   zeropivot = info->zeropivot;
1795*5f44c854SHong Zhang 
1796*5f44c854SHong Zhang   ierr  = ISGetIndices(ip,&rip);CHKERRQ(ierr);
1797*5f44c854SHong Zhang   ierr  = ISGetIndices(iip,&riip);CHKERRQ(ierr);
1798*5f44c854SHong Zhang 
1799*5f44c854SHong Zhang   /* allocate working arrays
1800*5f44c854SHong Zhang      c2r: linked list, keep track of pivot rows for a given column. c2r[col]: head of the list for a given col
1801*5f44c854SHong 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
1802*5f44c854SHong Zhang   */
1803*5f44c854SHong Zhang   nz   = (2*mbs+1)*sizeof(PetscInt)+mbs*sizeof(MatScalar);
1804*5f44c854SHong Zhang   ierr = PetscMalloc(nz,&il);CHKERRQ(ierr);
1805*5f44c854SHong Zhang   c2r  = il + mbs;
1806*5f44c854SHong Zhang   rtmp = (MatScalar*)(c2r + mbs);
1807*5f44c854SHong Zhang 
1808*5f44c854SHong Zhang   for (i=0; i<mbs; i++) {
1809*5f44c854SHong Zhang     c2r[i] = mbs;
1810*5f44c854SHong Zhang   }
1811*5f44c854SHong Zhang   il[0] = 0;
1812*5f44c854SHong Zhang 
1813*5f44c854SHong Zhang   for (k = 0; k<mbs; k++){
1814*5f44c854SHong Zhang     /* zero rtmp */
1815*5f44c854SHong Zhang     nz = bi[k+1] - bi[k];
1816*5f44c854SHong Zhang     bjtmp = bj + bi[k];
1817*5f44c854SHong Zhang     for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
1818*5f44c854SHong Zhang 
1819*5f44c854SHong Zhang     /* load in initial unfactored row */
1820*5f44c854SHong Zhang     bval = ba + bi[k];
1821*5f44c854SHong Zhang     jmin = ai[rip[k]]; jmax = ai[rip[k]+1];
1822*5f44c854SHong Zhang     for (j = jmin; j < jmax; j++){
1823*5f44c854SHong Zhang       col = riip[aj[j]];
1824*5f44c854SHong Zhang       if (col >= k){ /* only take upper triangular entry */
1825*5f44c854SHong Zhang         rtmp[col] = aa[j];
1826*5f44c854SHong Zhang         *bval++   = 0.0; /* for in-place factorization */
1827*5f44c854SHong Zhang       }
1828*5f44c854SHong Zhang     }
1829*5f44c854SHong Zhang     /* shift the diagonal of the matrix */
1830*5f44c854SHong Zhang 
1831*5f44c854SHong Zhang     /* modify k-th row by adding in those rows i with U(i,k)!=0 */
1832*5f44c854SHong Zhang     dk = rtmp[k];
1833*5f44c854SHong Zhang     i  = c2r[k]; /* first row to be added to k_th row  */
1834*5f44c854SHong Zhang 
1835*5f44c854SHong Zhang     while (i < k){
1836*5f44c854SHong Zhang       nexti = c2r[i]; /* next row to be added to k_th row */
1837*5f44c854SHong Zhang 
1838*5f44c854SHong Zhang       /* compute multiplier, update diag(k) and U(i,k) */
1839*5f44c854SHong Zhang       ili   = il[i];  /* index of first nonzero element in U(i,k:bms-1) */
1840*5f44c854SHong Zhang       uikdi = - ba[ili]*ba[bdiag[i]];  /* diagonal(k) */
1841*5f44c854SHong Zhang       dk   += uikdi*ba[ili]; /* update diag[k] */
1842*5f44c854SHong Zhang       ba[ili] = uikdi; /* -U(i,k) */
1843*5f44c854SHong Zhang 
1844*5f44c854SHong Zhang       /* add multiple of row i to k-th row */
1845*5f44c854SHong Zhang       jmin = ili + 1; jmax = bi[i+1];
1846*5f44c854SHong Zhang       if (jmin < jmax){
1847*5f44c854SHong Zhang         for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j];
1848*5f44c854SHong Zhang         /* update il and c2r for row i */
1849*5f44c854SHong Zhang         il[i] = jmin;
1850*5f44c854SHong Zhang         j = bj[jmin]; c2r[i] = c2r[j]; c2r[j] = i;
1851*5f44c854SHong Zhang       }
1852*5f44c854SHong Zhang       i = nexti;
1853*5f44c854SHong Zhang     }
1854*5f44c854SHong Zhang 
1855*5f44c854SHong Zhang     /* copy data into U(k,:) */
1856*5f44c854SHong Zhang     ba[bdiag[k]] = 1.0/dk; /* U(k,k) */
1857*5f44c854SHong Zhang     jmin = bi[k]; jmax = bi[k+1]-1;
1858*5f44c854SHong Zhang     if (jmin < jmax) {
1859*5f44c854SHong Zhang       for (j=jmin; j<jmax; j++){
1860*5f44c854SHong Zhang         col = bj[j]; ba[j] = rtmp[col];
1861*5f44c854SHong Zhang       }
1862*5f44c854SHong Zhang       /* add the k-th row into il and c2r */
1863*5f44c854SHong Zhang       il[k] = jmin;
1864*5f44c854SHong Zhang       i = bj[jmin]; c2r[k] = c2r[i]; c2r[i] = k;
1865*5f44c854SHong Zhang     }
1866*5f44c854SHong Zhang   }
1867*5f44c854SHong Zhang 
1868*5f44c854SHong Zhang   ierr = PetscFree(il);CHKERRQ(ierr);
1869*5f44c854SHong Zhang   ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr);
1870*5f44c854SHong Zhang   ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr);
1871*5f44c854SHong Zhang 
1872*5f44c854SHong Zhang   ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr);
1873*5f44c854SHong Zhang   if (perm_identity){
1874*5f44c854SHong Zhang     (B)->ops->solve           = MatSolve_SeqSBAIJ_1_NaturalOrdering_newdatastruct;
1875*5f44c854SHong Zhang     (B)->ops->solvetranspose  = 0;
1876*5f44c854SHong Zhang     (B)->ops->forwardsolve    = 0;
1877*5f44c854SHong Zhang     (B)->ops->backwardsolve   = 0;
1878*5f44c854SHong Zhang   } else {
1879*5f44c854SHong Zhang     (B)->ops->solve           = 0;
1880*5f44c854SHong Zhang     (B)->ops->solvetranspose  = 0;
1881*5f44c854SHong Zhang     (B)->ops->forwardsolve    = 0;
1882*5f44c854SHong Zhang     (B)->ops->backwardsolve   = 0;
1883*5f44c854SHong Zhang   }
1884*5f44c854SHong Zhang 
1885*5f44c854SHong Zhang   C->assembled    = PETSC_TRUE;
1886*5f44c854SHong Zhang   C->preallocated = PETSC_TRUE;
1887*5f44c854SHong Zhang   ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr);
1888*5f44c854SHong Zhang   PetscFunctionReturn(0);
1889*5f44c854SHong Zhang }
1890*5f44c854SHong Zhang 
1891*5f44c854SHong Zhang #undef __FUNCT__
1892a6175056SHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ"
18930481f469SBarry Smith PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info)
1894a6175056SHong Zhang {
1895719d5645SBarry Smith   Mat            C = B;
18960968510aSHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
18972ed133dbSHong Zhang   Mat_SeqSBAIJ   *b=(Mat_SeqSBAIJ*)C->data;
18989bfd6278SHong Zhang   IS             ip=b->row,iip = b->icol;
18992ed133dbSHong Zhang   PetscErrorCode ierr;
19005d0c19d7SBarry Smith   const PetscInt *rip,*riip;
19015d0c19d7SBarry Smith   PetscInt       i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bcol;
19022ed133dbSHong Zhang   PetscInt       *ai=a->i,*aj=a->j;
19032ed133dbSHong Zhang   PetscInt       k,jmin,jmax,*jl,*il,col,nexti,ili,nz;
1904622e2028SHong Zhang   MatScalar      *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi;
1905540703acSHong Zhang   PetscReal      zeropivot,rs,shiftnz;
1906fbf22428SSatish Balay   PetscReal      shiftpd;
1907540703acSHong Zhang   ChShift_Ctx    sctx;
1908540703acSHong Zhang   PetscInt       newshift;
1909db4efbfdSBarry Smith   PetscTruth     perm_identity;
1910d5d45c9bSBarry Smith 
1911a6175056SHong Zhang   PetscFunctionBegin;
1912540703acSHong Zhang   shiftnz   = info->shiftnz;
1913540703acSHong Zhang   shiftpd   = info->shiftpd;
1914ee45ca4aSHong Zhang   zeropivot = info->zeropivot;
1915ee45ca4aSHong Zhang 
19162ed133dbSHong Zhang   ierr  = ISGetIndices(ip,&rip);CHKERRQ(ierr);
19179bfd6278SHong Zhang   ierr  = ISGetIndices(iip,&riip);CHKERRQ(ierr);
19182ed133dbSHong Zhang 
19192ed133dbSHong Zhang   /* initialization */
19202ed133dbSHong Zhang   nz   = (2*mbs+1)*sizeof(PetscInt)+mbs*sizeof(MatScalar);
19212ed133dbSHong Zhang   ierr = PetscMalloc(nz,&il);CHKERRQ(ierr);
19222ed133dbSHong Zhang   jl   = il + mbs;
19232ed133dbSHong Zhang   rtmp = (MatScalar*)(jl + mbs);
19242ed133dbSHong Zhang 
1925540703acSHong Zhang   sctx.shift_amount = 0;
1926540703acSHong Zhang   sctx.nshift       = 0;
19272ed133dbSHong Zhang   do {
1928540703acSHong Zhang     sctx.chshift = PETSC_FALSE;
19292ed133dbSHong Zhang     for (i=0; i<mbs; i++) {
19302ed133dbSHong Zhang       rtmp[i] = 0.0; jl[i] = mbs; il[0] = 0;
1931a6175056SHong Zhang     }
19322ed133dbSHong Zhang 
19332ed133dbSHong Zhang     for (k = 0; k<mbs; k++){
1934c05c3958SHong Zhang       bval = ba + bi[k];
19352ed133dbSHong Zhang       /* initialize k-th row by the perm[k]-th row of A */
19362ed133dbSHong Zhang       jmin = ai[rip[k]]; jmax = ai[rip[k]+1];
19372ed133dbSHong Zhang       for (j = jmin; j < jmax; j++){
19389bfd6278SHong Zhang         col = riip[aj[j]];
19392ed133dbSHong Zhang         if (col >= k){ /* only take upper triangular entry */
19402ed133dbSHong Zhang           rtmp[col] = aa[j];
1941c05c3958SHong Zhang           *bval++  = 0.0; /* for in-place factorization */
19422ed133dbSHong Zhang         }
19432ed133dbSHong Zhang       }
194439e3d630SHong Zhang       /* shift the diagonal of the matrix */
1945540703acSHong Zhang       if (sctx.nshift) rtmp[k] += sctx.shift_amount;
19462ed133dbSHong Zhang 
19472ed133dbSHong Zhang       /* modify k-th row by adding in those rows i with U(i,k)!=0 */
19482ed133dbSHong Zhang       dk = rtmp[k];
19492ed133dbSHong Zhang       i = jl[k]; /* first row to be added to k_th row  */
19502ed133dbSHong Zhang 
19512ed133dbSHong Zhang       while (i < k){
19522ed133dbSHong Zhang         nexti = jl[i]; /* next row to be added to k_th row */
19532ed133dbSHong Zhang 
19542ed133dbSHong Zhang         /* compute multiplier, update diag(k) and U(i,k) */
19552ed133dbSHong Zhang         ili = il[i];  /* index of first nonzero element in U(i,k:bms-1) */
19562ed133dbSHong Zhang         uikdi = - ba[ili]*ba[bi[i]];  /* diagonal(k) */
19572ed133dbSHong Zhang         dk += uikdi*ba[ili];
19582ed133dbSHong Zhang         ba[ili] = uikdi; /* -U(i,k) */
19592ed133dbSHong Zhang 
19602ed133dbSHong Zhang         /* add multiple of row i to k-th row */
19612ed133dbSHong Zhang         jmin = ili + 1; jmax = bi[i+1];
19622ed133dbSHong Zhang         if (jmin < jmax){
19632ed133dbSHong Zhang           for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j];
19642ed133dbSHong Zhang           /* update il and jl for row i */
19652ed133dbSHong Zhang           il[i] = jmin;
19662ed133dbSHong Zhang           j = bj[jmin]; jl[i] = jl[j]; jl[j] = i;
19672ed133dbSHong Zhang         }
19682ed133dbSHong Zhang         i = nexti;
19692ed133dbSHong Zhang       }
19702ed133dbSHong Zhang 
19710697b6caSHong Zhang       /* shift the diagonals when zero pivot is detected */
19720697b6caSHong Zhang       /* compute rs=sum of abs(off-diagonal) */
19730697b6caSHong Zhang       rs   = 0.0;
19743c9e8598SHong Zhang       jmin = bi[k]+1;
19753c9e8598SHong Zhang       nz   = bi[k+1] - jmin;
19763c9e8598SHong Zhang       bcol = bj + jmin;
19773c9e8598SHong Zhang       while (nz--){
197889f845c8SHong Zhang         rs += PetscAbsScalar(rtmp[*bcol]);
197989f845c8SHong Zhang         bcol++;
19803c9e8598SHong Zhang       }
1981540703acSHong Zhang 
1982540703acSHong Zhang       sctx.rs = rs;
1983540703acSHong Zhang       sctx.pv = dk;
19845b5bc046SBarry Smith       ierr = MatCholeskyCheckShift_inline(info,sctx,k,newshift);CHKERRQ(ierr);
1985117f1891Stmunson 
1986117f1891Stmunson       if (newshift == 1) {
1987117f1891Stmunson         if (!sctx.shift_amount) {
1988117f1891Stmunson           sctx.shift_amount = 1e-5;
1989117f1891Stmunson         }
1990117f1891Stmunson         break;
1991117f1891Stmunson       }
19923c9e8598SHong Zhang 
19933c9e8598SHong Zhang       /* copy data into U(k,:) */
199439e3d630SHong Zhang       ba[bi[k]] = 1.0/dk; /* U(k,k) */
199539e3d630SHong Zhang       jmin = bi[k]+1; jmax = bi[k+1];
199639e3d630SHong Zhang       if (jmin < jmax) {
199739e3d630SHong Zhang         for (j=jmin; j<jmax; j++){
199839e3d630SHong Zhang           col = bj[j]; ba[j] = rtmp[col]; rtmp[col] = 0.0;
19993c9e8598SHong Zhang         }
200039e3d630SHong Zhang         /* add the k-th row into il and jl */
20013c9e8598SHong Zhang         il[k] = jmin;
20023c9e8598SHong Zhang         i = bj[jmin]; jl[k] = jl[i]; jl[i] = k;
20033c9e8598SHong Zhang       }
20043c9e8598SHong Zhang     }
2005540703acSHong Zhang   } while (sctx.chshift);
20063c9e8598SHong Zhang   ierr = PetscFree(il);CHKERRQ(ierr);
20073c9e8598SHong Zhang 
200839e3d630SHong Zhang   ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr);
20099bfd6278SHong Zhang   ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr);
2010db4efbfdSBarry Smith 
2011db4efbfdSBarry Smith   ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr);
2012db4efbfdSBarry Smith   if (perm_identity){
2013719d5645SBarry Smith     (B)->ops->solve           = MatSolve_SeqSBAIJ_1_NaturalOrdering;
2014719d5645SBarry Smith     (B)->ops->solvetranspose  = MatSolve_SeqSBAIJ_1_NaturalOrdering;
2015719d5645SBarry Smith     (B)->ops->forwardsolve    = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering;
2016719d5645SBarry Smith     (B)->ops->backwardsolve   = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering;
2017db4efbfdSBarry Smith   } else {
2018719d5645SBarry Smith     (B)->ops->solve           = MatSolve_SeqSBAIJ_1;
2019719d5645SBarry Smith     (B)->ops->solvetranspose  = MatSolve_SeqSBAIJ_1;
2020719d5645SBarry Smith     (B)->ops->forwardsolve    = MatForwardSolve_SeqSBAIJ_1;
2021719d5645SBarry Smith     (B)->ops->backwardsolve   = MatBackwardSolve_SeqSBAIJ_1;
2022db4efbfdSBarry Smith   }
2023db4efbfdSBarry Smith 
20243c9e8598SHong Zhang   C->assembled    = PETSC_TRUE;
20253c9e8598SHong Zhang   C->preallocated = PETSC_TRUE;
2026d0f46423SBarry Smith   ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr);
2027540703acSHong Zhang   if (sctx.nshift){
2028540703acSHong Zhang     if (shiftnz) {
20291e2582c4SBarry Smith       ierr = PetscInfo2(A,"number of shiftnz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
2030540703acSHong Zhang     } else if (shiftpd) {
20311e2582c4SBarry Smith       ierr = PetscInfo2(A,"number of shiftpd tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
20320697b6caSHong Zhang     }
20333c9e8598SHong Zhang   }
20343c9e8598SHong Zhang   PetscFunctionReturn(0);
20353c9e8598SHong Zhang }
2036a6175056SHong Zhang 
2037a6175056SHong Zhang #undef __FUNCT__
2038*5f44c854SHong Zhang #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ_newdatastruct"
2039*5f44c854SHong Zhang PetscErrorCode MatICCFactorSymbolic_SeqAIJ_newdatastruct(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
2040a6175056SHong Zhang {
20410968510aSHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
2042ed59401aSHong Zhang   Mat_SeqSBAIJ       *b;
2043dfbe8321SBarry Smith   PetscErrorCode     ierr;
204458ebbce7SBarry Smith   PetscTruth         perm_identity,missing;
2045*5f44c854SHong Zhang   PetscInt           reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag;
20465d0c19d7SBarry Smith   const PetscInt     *rip,*riip;
2047ed59401aSHong Zhang   PetscInt           jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow;
204858ebbce7SBarry Smith   PetscInt           nlnk,*lnk,*lnk_lvl=PETSC_NULL,d;
20495a8e39fbSHong Zhang   PetscInt           ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr;
2050ed59401aSHong Zhang   PetscReal          fill=info->fill,levels=info->levels;
2051a1a86e44SBarry Smith   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
2052a1a86e44SBarry Smith   PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL;
20537a48dd6fSHong Zhang   PetscBT            lnkbt;
2054b635d36bSHong Zhang   IS                 iperm;
2055a6175056SHong Zhang 
2056a6175056SHong Zhang   PetscFunctionBegin;
2057d0f46423SBarry 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);
205858ebbce7SBarry Smith   ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr);
205958ebbce7SBarry Smith   if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d);
2060653a6975SHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
2061b635d36bSHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
20627a48dd6fSHong Zhang 
206339e3d630SHong Zhang   ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr);
2064*5f44c854SHong Zhang   ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr);
206539e3d630SHong Zhang   ui[0] = 0;
206639e3d630SHong Zhang 
2067b635d36bSHong Zhang   /* ICC(0) without matrix ordering: simply copies fill pattern */
2068622e2028SHong Zhang   if (!levels && perm_identity) {
206958ebbce7SBarry Smith 
2070ed59401aSHong Zhang     for (i=0; i<am; i++) {
207139e3d630SHong Zhang       ui[i+1]  = ui[i] + ai[i+1] - a->diag[i];
2072*5f44c854SHong Zhang       udiag[i] = ui[i+1] - 1; /* points to the last entry of U(i,:) */
2073*5f44c854SHong Zhang     }
2074*5f44c854SHong Zhang     ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
2075*5f44c854SHong Zhang     cols = uj;
2076*5f44c854SHong Zhang     for (i=0; i<am; i++) {
2077*5f44c854SHong Zhang       aj    = a->j + a->diag[i] + 1; /* 1st entry of U(i,:) without diagonal */
2078*5f44c854SHong Zhang       ncols = ui[i+1] - ui[i] - 1;
2079*5f44c854SHong Zhang       for (j=0; j<ncols; j++) *cols++ = aj[j];
2080*5f44c854SHong Zhang       *cols++ = i; /* diagoanl is located as the last entry of U(i,:) */
2081*5f44c854SHong Zhang     }
2082*5f44c854SHong Zhang   } else { /* case: levels>0 || (levels=0 && !perm_identity) */
2083*5f44c854SHong Zhang     SETERRQ(1,"Not done yet");
2084*5f44c854SHong Zhang     ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
2085*5f44c854SHong Zhang     ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
2086*5f44c854SHong Zhang 
2087*5f44c854SHong Zhang     /* initialization */
2088*5f44c854SHong Zhang     ierr  = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr);
2089*5f44c854SHong Zhang 
2090*5f44c854SHong Zhang     /* jl: linked list for storing indices of the pivot rows
2091*5f44c854SHong Zhang        il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
2092*5f44c854SHong Zhang     ierr = PetscMalloc((2*am+1)*sizeof(PetscInt)+2*am*sizeof(PetscInt**),&jl);CHKERRQ(ierr);
2093*5f44c854SHong Zhang     il         = jl + am;
2094*5f44c854SHong Zhang     uj_ptr     = (PetscInt**)(il + am);
2095*5f44c854SHong Zhang     uj_lvl_ptr = (PetscInt**)(uj_ptr + am);
2096*5f44c854SHong Zhang     for (i=0; i<am; i++){
2097*5f44c854SHong Zhang       jl[i] = am; il[i] = 0;
2098*5f44c854SHong Zhang     }
2099*5f44c854SHong Zhang 
2100*5f44c854SHong Zhang     /* create and initialize a linked list for storing column indices of the active row k */
2101*5f44c854SHong Zhang     nlnk = am + 1;
2102*5f44c854SHong Zhang     ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
2103*5f44c854SHong Zhang 
2104*5f44c854SHong Zhang     /* initial FreeSpace size is fill*(ai[am]+1) */
2105*5f44c854SHong Zhang     ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr);
2106*5f44c854SHong Zhang     current_space = free_space;
2107*5f44c854SHong Zhang     ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space_lvl);CHKERRQ(ierr);
2108*5f44c854SHong Zhang     current_space_lvl = free_space_lvl;
2109*5f44c854SHong Zhang 
2110*5f44c854SHong Zhang     for (k=0; k<am; k++){  /* for each active row k */
2111*5f44c854SHong Zhang       /* initialize lnk by the column indices of row rip[k] of A */
2112*5f44c854SHong Zhang       nzk   = 0;
2113*5f44c854SHong Zhang       ncols = ai[rip[k]+1] - ai[rip[k]];
2114*5f44c854SHong 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);
2115*5f44c854SHong Zhang       ncols_upper = 0;
2116*5f44c854SHong Zhang       for (j=0; j<ncols; j++){
2117*5f44c854SHong Zhang         i = *(aj + ai[rip[k]] + j); /* unpermuted column index */
2118*5f44c854SHong Zhang         if (riip[i] >= k){ /* only take upper triangular entry */
2119*5f44c854SHong Zhang           ajtmp[ncols_upper] = i;
2120*5f44c854SHong Zhang           ncols_upper++;
2121*5f44c854SHong Zhang         }
2122*5f44c854SHong Zhang       }
2123*5f44c854SHong Zhang       ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
2124*5f44c854SHong Zhang       nzk += nlnk;
2125*5f44c854SHong Zhang 
2126*5f44c854SHong Zhang       /* update lnk by computing fill-in for each pivot row to be merged in */
2127*5f44c854SHong Zhang       prow = jl[k]; /* 1st pivot row */
2128*5f44c854SHong Zhang 
2129*5f44c854SHong Zhang       while (prow < k){
2130*5f44c854SHong Zhang         nextprow = jl[prow];
2131*5f44c854SHong Zhang 
2132*5f44c854SHong Zhang         /* merge prow into k-th row */
2133*5f44c854SHong Zhang         jmin = il[prow] + 1;  /* index of the 2nd nzero entry in U(prow,k:am-1) */
2134*5f44c854SHong Zhang         jmax = ui[prow+1];
2135*5f44c854SHong Zhang         ncols = jmax-jmin;
2136*5f44c854SHong Zhang         i     = jmin - ui[prow];
2137*5f44c854SHong Zhang         cols  = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */
2138*5f44c854SHong Zhang         uj    = uj_lvl_ptr[prow] + i; /* levels of cols */
2139*5f44c854SHong Zhang         j     = *(uj - 1);
2140*5f44c854SHong Zhang         ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr);
2141*5f44c854SHong Zhang         nzk += nlnk;
2142*5f44c854SHong Zhang 
2143*5f44c854SHong Zhang         /* update il and jl for prow */
2144*5f44c854SHong Zhang         if (jmin < jmax){
2145*5f44c854SHong Zhang           il[prow] = jmin;
2146*5f44c854SHong Zhang           j = *cols; jl[prow] = jl[j]; jl[j] = prow;
2147*5f44c854SHong Zhang         }
2148*5f44c854SHong Zhang         prow = nextprow;
2149*5f44c854SHong Zhang       }
2150*5f44c854SHong Zhang 
2151*5f44c854SHong Zhang       /* if free space is not available, make more free space */
2152*5f44c854SHong Zhang       if (current_space->local_remaining<nzk) {
2153*5f44c854SHong Zhang         i = am - k + 1; /* num of unfactored rows */
2154*5f44c854SHong Zhang         i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */
2155*5f44c854SHong Zhang         ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
2156*5f44c854SHong Zhang         ierr = PetscFreeSpaceGet(i,&current_space_lvl);CHKERRQ(ierr);
2157*5f44c854SHong Zhang         reallocs++;
2158*5f44c854SHong Zhang       }
2159*5f44c854SHong Zhang 
2160*5f44c854SHong Zhang       /* copy data into free_space and free_space_lvl, then initialize lnk */
2161*5f44c854SHong Zhang       if (nzk == 0) SETERRQ1(PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k);
2162*5f44c854SHong Zhang       ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
2163*5f44c854SHong Zhang 
2164*5f44c854SHong Zhang       /* add the k-th row into il and jl */
2165*5f44c854SHong Zhang       if (nzk > 1){
2166*5f44c854SHong Zhang         i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
2167*5f44c854SHong Zhang         jl[k] = jl[i]; jl[i] = k;
2168*5f44c854SHong Zhang         il[k] = ui[k] + 1;
2169*5f44c854SHong Zhang       }
2170*5f44c854SHong Zhang       uj_ptr[k]     = current_space->array;
2171*5f44c854SHong Zhang       uj_lvl_ptr[k] = current_space_lvl->array;
2172*5f44c854SHong Zhang 
2173*5f44c854SHong Zhang       current_space->array           += nzk;
2174*5f44c854SHong Zhang       current_space->local_used      += nzk;
2175*5f44c854SHong Zhang       current_space->local_remaining -= nzk;
2176*5f44c854SHong Zhang 
2177*5f44c854SHong Zhang       current_space_lvl->array           += nzk;
2178*5f44c854SHong Zhang       current_space_lvl->local_used      += nzk;
2179*5f44c854SHong Zhang       current_space_lvl->local_remaining -= nzk;
2180*5f44c854SHong Zhang 
2181*5f44c854SHong Zhang       ui[k+1] = ui[k] + nzk;
2182*5f44c854SHong Zhang     }
2183*5f44c854SHong Zhang 
2184*5f44c854SHong Zhang #if defined(PETSC_USE_INFO)
2185*5f44c854SHong Zhang     if (ai[am] != 0) {
2186*5f44c854SHong Zhang       PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]);
2187*5f44c854SHong Zhang       ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr);
2188*5f44c854SHong Zhang       ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
2189*5f44c854SHong Zhang       ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr);
2190*5f44c854SHong Zhang     } else {
2191*5f44c854SHong Zhang       ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr);
2192*5f44c854SHong Zhang     }
2193*5f44c854SHong Zhang #endif
2194*5f44c854SHong Zhang 
2195*5f44c854SHong Zhang     ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
2196*5f44c854SHong Zhang     ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
2197*5f44c854SHong Zhang     ierr = PetscFree(jl);CHKERRQ(ierr);
2198*5f44c854SHong Zhang     ierr = PetscFree(ajtmp);CHKERRQ(ierr);
2199*5f44c854SHong Zhang 
2200*5f44c854SHong Zhang     /* destroy list of free space and other temporary array(s) */
2201*5f44c854SHong Zhang     ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
2202*5f44c854SHong Zhang     ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr);
2203*5f44c854SHong Zhang     ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
2204*5f44c854SHong Zhang     ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
2205*5f44c854SHong Zhang 
2206*5f44c854SHong Zhang   } /* end of case: levels>0 || (levels=0 && !perm_identity) */
2207*5f44c854SHong Zhang 
2208*5f44c854SHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
2209*5f44c854SHong Zhang 
2210*5f44c854SHong Zhang   b    = (Mat_SeqSBAIJ*)(fact)->data;
2211*5f44c854SHong Zhang   b->singlemalloc = PETSC_FALSE;
2212*5f44c854SHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr);
2213*5f44c854SHong Zhang   b->j    = uj;
2214*5f44c854SHong Zhang   b->i    = ui;
2215*5f44c854SHong Zhang   b->diag = udiag;
2216*5f44c854SHong Zhang   b->ilen = 0;
2217*5f44c854SHong Zhang   b->imax = 0;
2218*5f44c854SHong Zhang   b->row  = perm;
2219*5f44c854SHong Zhang   b->col  = perm;
2220*5f44c854SHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
2221*5f44c854SHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
2222*5f44c854SHong Zhang   b->icol = iperm;
2223*5f44c854SHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
2224*5f44c854SHong Zhang   ierr    = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
2225*5f44c854SHong Zhang   ierr = PetscLogObjectMemory((fact),(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
2226*5f44c854SHong Zhang   b->maxnz   = b->nz = ui[am];
2227*5f44c854SHong Zhang   b->free_a  = PETSC_TRUE;
2228*5f44c854SHong Zhang   b->free_ij = PETSC_TRUE;
2229*5f44c854SHong Zhang 
2230*5f44c854SHong Zhang   (fact)->info.factor_mallocs    = reallocs;
2231*5f44c854SHong Zhang   (fact)->info.fill_ratio_given  = fill;
2232*5f44c854SHong Zhang   if (ai[am] != 0) {
2233*5f44c854SHong Zhang     (fact)->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]);
2234*5f44c854SHong Zhang   } else {
2235*5f44c854SHong Zhang     (fact)->info.fill_ratio_needed = 0.0;
2236*5f44c854SHong Zhang   }
2237*5f44c854SHong Zhang   (fact)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_newdatastruct;
2238*5f44c854SHong Zhang   PetscFunctionReturn(0);
2239*5f44c854SHong Zhang }
2240*5f44c854SHong Zhang 
2241*5f44c854SHong Zhang #undef __FUNCT__
2242*5f44c854SHong Zhang #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ"
2243*5f44c854SHong Zhang PetscErrorCode MatICCFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
2244*5f44c854SHong Zhang {
2245*5f44c854SHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
2246*5f44c854SHong Zhang   Mat_SeqSBAIJ       *b;
2247*5f44c854SHong Zhang   PetscErrorCode     ierr;
2248*5f44c854SHong Zhang   PetscTruth         perm_identity,missing;
2249*5f44c854SHong Zhang   PetscInt           reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag;
2250*5f44c854SHong Zhang   const PetscInt     *rip,*riip;
2251*5f44c854SHong Zhang   PetscInt           jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow;
2252*5f44c854SHong Zhang   PetscInt           nlnk,*lnk,*lnk_lvl=PETSC_NULL,d;
2253*5f44c854SHong Zhang   PetscInt           ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr;
2254*5f44c854SHong Zhang   PetscReal          fill=info->fill,levels=info->levels;
2255*5f44c854SHong Zhang   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
2256*5f44c854SHong Zhang   PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL;
2257*5f44c854SHong Zhang   PetscBT            lnkbt;
2258*5f44c854SHong Zhang   IS                 iperm;
2259*5f44c854SHong Zhang   PetscTruth         newdatastruct=PETSC_FALSE;
2260*5f44c854SHong Zhang 
2261*5f44c854SHong Zhang   PetscFunctionBegin;
2262*5f44c854SHong Zhang   ierr = PetscOptionsGetTruth(PETSC_NULL,"-icc_new",&newdatastruct,PETSC_NULL);CHKERRQ(ierr);
2263*5f44c854SHong Zhang   if(newdatastruct){
2264*5f44c854SHong Zhang     ierr = MatICCFactorSymbolic_SeqAIJ_newdatastruct(fact,A,perm,info);CHKERRQ(ierr);
2265*5f44c854SHong Zhang     PetscFunctionReturn(0);
2266*5f44c854SHong Zhang   }
2267*5f44c854SHong Zhang 
2268*5f44c854SHong 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);
2269*5f44c854SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr);
2270*5f44c854SHong Zhang   if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d);
2271*5f44c854SHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
2272*5f44c854SHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
2273*5f44c854SHong Zhang 
2274*5f44c854SHong Zhang   ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr);
2275*5f44c854SHong Zhang   ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr);
2276*5f44c854SHong Zhang   ui[0] = 0;
2277*5f44c854SHong Zhang 
2278*5f44c854SHong Zhang   /* ICC(0) without matrix ordering: simply copies fill pattern */
2279*5f44c854SHong Zhang   if (!levels && perm_identity) {
2280*5f44c854SHong Zhang 
2281*5f44c854SHong Zhang     for (i=0; i<am; i++) {
2282*5f44c854SHong Zhang       ui[i+1]  = ui[i] + ai[i+1] - a->diag[i];
2283*5f44c854SHong Zhang       udiag[i] = ui[i];
2284ed59401aSHong Zhang     }
228539e3d630SHong Zhang     ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
228639e3d630SHong Zhang     cols = uj;
2287ed59401aSHong Zhang     for (i=0; i<am; i++) {
2288ed59401aSHong Zhang       aj    = a->j + a->diag[i];
228939e3d630SHong Zhang       ncols = ui[i+1] - ui[i];
229039e3d630SHong Zhang       for (j=0; j<ncols; j++) *cols++ = *aj++;
2291ed59401aSHong Zhang     }
229239e3d630SHong Zhang   } else { /* case: levels>0 || (levels=0 && !perm_identity) */
2293b635d36bSHong Zhang     ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
2294b635d36bSHong Zhang     ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
2295b635d36bSHong Zhang 
22967a48dd6fSHong Zhang     /* initialization */
22975a8e39fbSHong Zhang     ierr  = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr);
22987a48dd6fSHong Zhang 
22997a48dd6fSHong Zhang     /* jl: linked list for storing indices of the pivot rows
23007a48dd6fSHong Zhang        il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
23011393bc97SHong Zhang     ierr = PetscMalloc((2*am+1)*sizeof(PetscInt)+2*am*sizeof(PetscInt**),&jl);CHKERRQ(ierr);
23027a48dd6fSHong Zhang     il         = jl + am;
23037a48dd6fSHong Zhang     uj_ptr     = (PetscInt**)(il + am);
23047a48dd6fSHong Zhang     uj_lvl_ptr = (PetscInt**)(uj_ptr + am);
23057a48dd6fSHong Zhang     for (i=0; i<am; i++){
23067a48dd6fSHong Zhang       jl[i] = am; il[i] = 0;
23077a48dd6fSHong Zhang     }
23087a48dd6fSHong Zhang 
23097a48dd6fSHong Zhang     /* create and initialize a linked list for storing column indices of the active row k */
23107a48dd6fSHong Zhang     nlnk = am + 1;
23112ed133dbSHong Zhang     ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
23127a48dd6fSHong Zhang 
23137a48dd6fSHong Zhang     /* initial FreeSpace size is fill*(ai[am]+1) */
2314a1a86e44SBarry Smith     ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr);
23157a48dd6fSHong Zhang     current_space = free_space;
2316a1a86e44SBarry Smith     ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space_lvl);CHKERRQ(ierr);
23177a48dd6fSHong Zhang     current_space_lvl = free_space_lvl;
23187a48dd6fSHong Zhang 
23197a48dd6fSHong Zhang     for (k=0; k<am; k++){  /* for each active row k */
23207a48dd6fSHong Zhang       /* initialize lnk by the column indices of row rip[k] of A */
23217a48dd6fSHong Zhang       nzk   = 0;
2322622e2028SHong Zhang       ncols = ai[rip[k]+1] - ai[rip[k]];
23233b4a8b6dSBarry 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);
2324622e2028SHong Zhang       ncols_upper = 0;
2325622e2028SHong Zhang       for (j=0; j<ncols; j++){
2326b635d36bSHong Zhang         i = *(aj + ai[rip[k]] + j); /* unpermuted column index */
2327b635d36bSHong Zhang         if (riip[i] >= k){ /* only take upper triangular entry */
23285a8e39fbSHong Zhang           ajtmp[ncols_upper] = i;
2329622e2028SHong Zhang           ncols_upper++;
2330622e2028SHong Zhang         }
2331622e2028SHong Zhang       }
2332b635d36bSHong Zhang       ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
23337a48dd6fSHong Zhang       nzk += nlnk;
23347a48dd6fSHong Zhang 
23357a48dd6fSHong Zhang       /* update lnk by computing fill-in for each pivot row to be merged in */
23367a48dd6fSHong Zhang       prow = jl[k]; /* 1st pivot row */
23377a48dd6fSHong Zhang 
23387a48dd6fSHong Zhang       while (prow < k){
23397a48dd6fSHong Zhang         nextprow = jl[prow];
23407a48dd6fSHong Zhang 
23417a48dd6fSHong Zhang         /* merge prow into k-th row */
23427a48dd6fSHong Zhang         jmin = il[prow] + 1;  /* index of the 2nd nzero entry in U(prow,k:am-1) */
23437a48dd6fSHong Zhang         jmax = ui[prow+1];
23447a48dd6fSHong Zhang         ncols = jmax-jmin;
2345ed59401aSHong Zhang         i     = jmin - ui[prow];
2346ed59401aSHong Zhang         cols  = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */
23475a8e39fbSHong Zhang         uj    = uj_lvl_ptr[prow] + i; /* levels of cols */
23485a8e39fbSHong Zhang         j     = *(uj - 1);
23495a8e39fbSHong Zhang         ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr);
23507a48dd6fSHong Zhang         nzk += nlnk;
23517a48dd6fSHong Zhang 
23527a48dd6fSHong Zhang         /* update il and jl for prow */
23537a48dd6fSHong Zhang         if (jmin < jmax){
23547a48dd6fSHong Zhang           il[prow] = jmin;
23557a48dd6fSHong Zhang           j = *cols; jl[prow] = jl[j]; jl[j] = prow;
23567a48dd6fSHong Zhang         }
23577a48dd6fSHong Zhang         prow = nextprow;
23587a48dd6fSHong Zhang       }
23597a48dd6fSHong Zhang 
23607a48dd6fSHong Zhang       /* if free space is not available, make more free space */
23617a48dd6fSHong Zhang       if (current_space->local_remaining<nzk) {
23627a48dd6fSHong Zhang         i = am - k + 1; /* num of unfactored rows */
23637a48dd6fSHong Zhang         i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */
2364a1a86e44SBarry Smith         ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
2365a1a86e44SBarry Smith         ierr = PetscFreeSpaceGet(i,&current_space_lvl);CHKERRQ(ierr);
23667a48dd6fSHong Zhang         reallocs++;
23677a48dd6fSHong Zhang       }
23687a48dd6fSHong Zhang 
23697a48dd6fSHong Zhang       /* copy data into free_space and free_space_lvl, then initialize lnk */
2370b635d36bSHong Zhang       if (nzk == 0) SETERRQ1(PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k);
23712ed133dbSHong Zhang       ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
23727a48dd6fSHong Zhang 
23737a48dd6fSHong Zhang       /* add the k-th row into il and jl */
237439e3d630SHong Zhang       if (nzk > 1){
23757a48dd6fSHong Zhang         i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
23767a48dd6fSHong Zhang         jl[k] = jl[i]; jl[i] = k;
23777a48dd6fSHong Zhang         il[k] = ui[k] + 1;
23787a48dd6fSHong Zhang       }
23797a48dd6fSHong Zhang       uj_ptr[k]     = current_space->array;
23807a48dd6fSHong Zhang       uj_lvl_ptr[k] = current_space_lvl->array;
23817a48dd6fSHong Zhang 
23827a48dd6fSHong Zhang       current_space->array           += nzk;
23837a48dd6fSHong Zhang       current_space->local_used      += nzk;
23847a48dd6fSHong Zhang       current_space->local_remaining -= nzk;
23857a48dd6fSHong Zhang 
23867a48dd6fSHong Zhang       current_space_lvl->array           += nzk;
23877a48dd6fSHong Zhang       current_space_lvl->local_used      += nzk;
23887a48dd6fSHong Zhang       current_space_lvl->local_remaining -= nzk;
23897a48dd6fSHong Zhang 
23907a48dd6fSHong Zhang       ui[k+1] = ui[k] + nzk;
23917a48dd6fSHong Zhang     }
23927a48dd6fSHong Zhang 
23936cf91177SBarry Smith #if defined(PETSC_USE_INFO)
23947a48dd6fSHong Zhang     if (ai[am] != 0) {
239539e3d630SHong Zhang       PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]);
2396ae15b995SBarry Smith       ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr);
2397ae15b995SBarry Smith       ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
2398ae15b995SBarry Smith       ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr);
23997a48dd6fSHong Zhang     } else {
2400ae15b995SBarry Smith       ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr);
24017a48dd6fSHong Zhang     }
240263ba0a88SBarry Smith #endif
24037a48dd6fSHong Zhang 
24047a48dd6fSHong Zhang     ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
2405b635d36bSHong Zhang     ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
24067a48dd6fSHong Zhang     ierr = PetscFree(jl);CHKERRQ(ierr);
24075a8e39fbSHong Zhang     ierr = PetscFree(ajtmp);CHKERRQ(ierr);
24087a48dd6fSHong Zhang 
24097a48dd6fSHong Zhang     /* destroy list of free space and other temporary array(s) */
24107a48dd6fSHong Zhang     ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
2411a1a86e44SBarry Smith     ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr);
24122ed133dbSHong Zhang     ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
2413a1a86e44SBarry Smith     ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
24147a48dd6fSHong Zhang 
241539e3d630SHong Zhang   } /* end of case: levels>0 || (levels=0 && !perm_identity) */
241639e3d630SHong Zhang 
24177a48dd6fSHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
24187a48dd6fSHong Zhang 
2419719d5645SBarry Smith   b    = (Mat_SeqSBAIJ*)(fact)->data;
24207a48dd6fSHong Zhang   b->singlemalloc = PETSC_FALSE;
24217a48dd6fSHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr);
24227a48dd6fSHong Zhang   b->j    = uj;
24237a48dd6fSHong Zhang   b->i    = ui;
2424*5f44c854SHong Zhang   b->diag = udiag;
24257a48dd6fSHong Zhang   b->ilen = 0;
24267a48dd6fSHong Zhang   b->imax = 0;
24277a48dd6fSHong Zhang   b->row  = perm;
2428b635d36bSHong Zhang   b->col  = perm;
2429b635d36bSHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
2430b635d36bSHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
2431b635d36bSHong Zhang   b->icol = iperm;
24327a48dd6fSHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
24337a48dd6fSHong Zhang   ierr    = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
2434719d5645SBarry Smith   ierr = PetscLogObjectMemory((fact),(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
24357a48dd6fSHong Zhang   b->maxnz   = b->nz = ui[am];
2436e6b907acSBarry Smith   b->free_a  = PETSC_TRUE;
2437e6b907acSBarry Smith   b->free_ij = PETSC_TRUE;
24387a48dd6fSHong Zhang 
2439719d5645SBarry Smith   (fact)->info.factor_mallocs    = reallocs;
2440719d5645SBarry Smith   (fact)->info.fill_ratio_given  = fill;
24417a48dd6fSHong Zhang   if (ai[am] != 0) {
2442719d5645SBarry Smith     (fact)->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]);
24437a48dd6fSHong Zhang   } else {
2444719d5645SBarry Smith     (fact)->info.fill_ratio_needed = 0.0;
24457a48dd6fSHong Zhang   }
2446719d5645SBarry Smith   (fact)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ;
2447a6175056SHong Zhang   PetscFunctionReturn(0);
2448a6175056SHong Zhang }
2449d5d45c9bSBarry Smith 
2450f76d2b81SHong Zhang #undef __FUNCT__
2451f76d2b81SHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_SeqAIJ"
24520481f469SBarry Smith PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
2453f76d2b81SHong Zhang {
2454f76d2b81SHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
245510c27e3dSHong Zhang   Mat_SeqSBAIJ       *b;
2456dfbe8321SBarry Smith   PetscErrorCode     ierr;
2457f76d2b81SHong Zhang   PetscTruth         perm_identity;
245810c27e3dSHong Zhang   PetscReal          fill = info->fill;
24595d0c19d7SBarry Smith   const PetscInt     *rip,*riip;
24605d0c19d7SBarry Smith   PetscInt           i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow;
246110c27e3dSHong Zhang   PetscInt           *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow;
24622ed133dbSHong Zhang   PetscInt           nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr;
2463a1a86e44SBarry Smith   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
246410c27e3dSHong Zhang   PetscBT            lnkbt;
24652ed133dbSHong Zhang   IS                 iperm;
2466f76d2b81SHong Zhang 
2467f76d2b81SHong Zhang   PetscFunctionBegin;
2468d0f46423SBarry 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);
24692ed133dbSHong Zhang   /* check whether perm is the identity mapping */
2470f76d2b81SHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
24712ed133dbSHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
24722ed133dbSHong Zhang   ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
24739bfd6278SHong Zhang   ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
247410c27e3dSHong Zhang 
247510c27e3dSHong Zhang   /* initialization */
24761393bc97SHong Zhang   ierr  = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr);
247710c27e3dSHong Zhang   ui[0] = 0;
247810c27e3dSHong Zhang 
247910c27e3dSHong Zhang   /* jl: linked list for storing indices of the pivot rows
24801393bc97SHong Zhang      il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
24811393bc97SHong Zhang   ierr = PetscMalloc((3*am+1)*sizeof(PetscInt)+am*sizeof(PetscInt**),&jl);CHKERRQ(ierr);
24821393bc97SHong Zhang   il     = jl + am;
24831393bc97SHong Zhang   cols   = il + am;
24841393bc97SHong Zhang   ui_ptr = (PetscInt**)(cols + am);
24851393bc97SHong Zhang   for (i=0; i<am; i++){
24861393bc97SHong Zhang     jl[i] = am; il[i] = 0;
2487f76d2b81SHong Zhang   }
248810c27e3dSHong Zhang 
248910c27e3dSHong Zhang   /* create and initialize a linked list for storing column indices of the active row k */
24901393bc97SHong Zhang   nlnk = am + 1;
24911393bc97SHong Zhang   ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
249210c27e3dSHong Zhang 
24931393bc97SHong Zhang   /* initial FreeSpace size is fill*(ai[am]+1) */
2494a1a86e44SBarry Smith   ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr);
249510c27e3dSHong Zhang   current_space = free_space;
249610c27e3dSHong Zhang 
24971393bc97SHong Zhang   for (k=0; k<am; k++){  /* for each active row k */
249810c27e3dSHong Zhang     /* initialize lnk by the column indices of row rip[k] of A */
249910c27e3dSHong Zhang     nzk   = 0;
25002ed133dbSHong Zhang     ncols = ai[rip[k]+1] - ai[rip[k]];
25013b4a8b6dSBarry 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);
25022ed133dbSHong Zhang     ncols_upper = 0;
25032ed133dbSHong Zhang     for (j=0; j<ncols; j++){
25049bfd6278SHong Zhang       i = riip[*(aj + ai[rip[k]] + j)];
25052ed133dbSHong Zhang       if (i >= k){ /* only take upper triangular entry */
25062ed133dbSHong Zhang         cols[ncols_upper] = i;
25072ed133dbSHong Zhang         ncols_upper++;
25082ed133dbSHong Zhang       }
25092ed133dbSHong Zhang     }
25101393bc97SHong Zhang     ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
251110c27e3dSHong Zhang     nzk += nlnk;
251210c27e3dSHong Zhang 
251310c27e3dSHong Zhang     /* update lnk by computing fill-in for each pivot row to be merged in */
251410c27e3dSHong Zhang     prow = jl[k]; /* 1st pivot row */
251510c27e3dSHong Zhang 
251610c27e3dSHong Zhang     while (prow < k){
251710c27e3dSHong Zhang       nextprow = jl[prow];
251810c27e3dSHong Zhang       /* merge prow into k-th row */
25191393bc97SHong Zhang       jmin = il[prow] + 1;  /* index of the 2nd nzero entry in U(prow,k:am-1) */
252010c27e3dSHong Zhang       jmax = ui[prow+1];
252110c27e3dSHong Zhang       ncols = jmax-jmin;
25221393bc97SHong Zhang       uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */
25235a8e39fbSHong Zhang       ierr = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
252410c27e3dSHong Zhang       nzk += nlnk;
252510c27e3dSHong Zhang 
252610c27e3dSHong Zhang       /* update il and jl for prow */
252710c27e3dSHong Zhang       if (jmin < jmax){
252810c27e3dSHong Zhang         il[prow] = jmin;
25292ed133dbSHong Zhang         j = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow;
253010c27e3dSHong Zhang       }
253110c27e3dSHong Zhang       prow = nextprow;
253210c27e3dSHong Zhang     }
253310c27e3dSHong Zhang 
253410c27e3dSHong Zhang     /* if free space is not available, make more free space */
253510c27e3dSHong Zhang     if (current_space->local_remaining<nzk) {
25361393bc97SHong Zhang       i = am - k + 1; /* num of unfactored rows */
253710c27e3dSHong Zhang       i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */
2538a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
253910c27e3dSHong Zhang       reallocs++;
254010c27e3dSHong Zhang     }
254110c27e3dSHong Zhang 
254210c27e3dSHong Zhang     /* copy data into free space, then initialize lnk */
25431393bc97SHong Zhang     ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
254410c27e3dSHong Zhang 
254510c27e3dSHong Zhang     /* add the k-th row into il and jl */
254610c27e3dSHong Zhang     if (nzk-1 > 0){
25471393bc97SHong Zhang       i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
254810c27e3dSHong Zhang       jl[k] = jl[i]; jl[i] = k;
254910c27e3dSHong Zhang       il[k] = ui[k] + 1;
255010c27e3dSHong Zhang     }
25512ed133dbSHong Zhang     ui_ptr[k] = current_space->array;
255210c27e3dSHong Zhang     current_space->array           += nzk;
255310c27e3dSHong Zhang     current_space->local_used      += nzk;
255410c27e3dSHong Zhang     current_space->local_remaining -= nzk;
255510c27e3dSHong Zhang 
255610c27e3dSHong Zhang     ui[k+1] = ui[k] + nzk;
255710c27e3dSHong Zhang   }
255810c27e3dSHong Zhang 
25596cf91177SBarry Smith #if defined(PETSC_USE_INFO)
25601393bc97SHong Zhang   if (ai[am] != 0) {
25611393bc97SHong Zhang     PetscReal af = (PetscReal)(ui[am])/((PetscReal)ai[am]);
2562ae15b995SBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr);
2563ae15b995SBarry Smith     ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
2564ae15b995SBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr);
256510c27e3dSHong Zhang   } else {
2566ae15b995SBarry Smith      ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr);
256710c27e3dSHong Zhang   }
256863ba0a88SBarry Smith #endif
256910c27e3dSHong Zhang 
257010c27e3dSHong Zhang   ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
25719bfd6278SHong Zhang   ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
257210c27e3dSHong Zhang   ierr = PetscFree(jl);CHKERRQ(ierr);
257310c27e3dSHong Zhang 
257410c27e3dSHong Zhang   /* destroy list of free space and other temporary array(s) */
25751393bc97SHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
2576a1a86e44SBarry Smith   ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr);
257710c27e3dSHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
257810c27e3dSHong Zhang 
257910c27e3dSHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
258010c27e3dSHong Zhang 
2581719d5645SBarry Smith   b = (Mat_SeqSBAIJ*)(fact)->data;
258210c27e3dSHong Zhang   b->singlemalloc = PETSC_FALSE;
2583e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
2584e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
25851393bc97SHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr);
258610c27e3dSHong Zhang   b->j    = uj;
258710c27e3dSHong Zhang   b->i    = ui;
258810c27e3dSHong Zhang   b->diag = 0;
258910c27e3dSHong Zhang   b->ilen = 0;
259010c27e3dSHong Zhang   b->imax = 0;
259110c27e3dSHong Zhang   b->row  = perm;
25929bfd6278SHong Zhang   b->col  = perm;
25939bfd6278SHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
25949bfd6278SHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
25959bfd6278SHong Zhang   b->icol = iperm;
259610c27e3dSHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
25971393bc97SHong Zhang   ierr    = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
2598719d5645SBarry Smith   ierr    = PetscLogObjectMemory(fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
25991393bc97SHong Zhang   b->maxnz = b->nz = ui[am];
260010c27e3dSHong Zhang 
2601719d5645SBarry Smith   (fact)->info.factor_mallocs    = reallocs;
2602719d5645SBarry Smith   (fact)->info.fill_ratio_given  = fill;
26031393bc97SHong Zhang   if (ai[am] != 0) {
2604719d5645SBarry Smith     (fact)->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]);
260510c27e3dSHong Zhang   } else {
2606719d5645SBarry Smith     (fact)->info.fill_ratio_needed = 0.0;
260710c27e3dSHong Zhang   }
2608719d5645SBarry Smith   (fact)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ;
2609f76d2b81SHong Zhang   PetscFunctionReturn(0);
2610f76d2b81SHong Zhang }
2611599ef60dSHong Zhang 
2612599ef60dSHong Zhang #undef __FUNCT__
2613894a2336SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering_newdatastruct"
2614894a2336SHong Zhang PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering_newdatastruct(Mat A,Vec bb,Vec xx)
2615599ef60dSHong Zhang {
26161d098868SHong Zhang   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
26171d098868SHong Zhang   PetscErrorCode    ierr;
26181d098868SHong Zhang   PetscInt          n = A->rmap->n;
26198fc3a347SHong Zhang   const PetscInt    *ai = a->i,*aj = a->j,*vi;
26201d098868SHong Zhang   PetscScalar       *x,sum;
26211d098868SHong Zhang   const PetscScalar *b;
26221d098868SHong Zhang   const MatScalar   *aa = a->a,*v;
26238fc3a347SHong Zhang   PetscInt          i,nz;
26241d098868SHong Zhang 
26251d098868SHong Zhang   PetscFunctionBegin;
26261d098868SHong Zhang   if (!n) PetscFunctionReturn(0);
26271d098868SHong Zhang 
26281d098868SHong Zhang   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
26291d098868SHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
26301d098868SHong Zhang 
26311d098868SHong Zhang   /* forward solve the lower triangular */
26321d098868SHong Zhang   x[0] = b[0];
26331d098868SHong Zhang   v    = aa;
26341d098868SHong Zhang   vi   = aj;
26351d098868SHong Zhang   for (i=1; i<n; i++) {
26361d098868SHong Zhang     nz  = ai[i+1] - ai[i];
26371d098868SHong Zhang     sum = b[i];
2638003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
2639e09e08ccSBarry Smith     /*    while (nz--) sum -= *v++ * x[*vi++];*/
2640e0e43300SBarry Smith     v  += nz;
2641e0e43300SBarry Smith     vi += nz;
26421d098868SHong Zhang     x[i] = sum;
26431d098868SHong Zhang   }
26441d098868SHong Zhang 
26451d098868SHong Zhang   /* backward solve the upper triangular */
26468fc3a347SHong Zhang   v   = aa + ai[n+1];
26478fc3a347SHong Zhang   vi  = aj + ai[n+1];
26481d098868SHong Zhang   for (i=n-1; i>=0; i--){
26498fc3a347SHong Zhang     nz = ai[2*n-i +1] - ai[2*n-i]-1;
26501d098868SHong Zhang     sum = x[i];
2651003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
2652e09e08ccSBarry Smith     /* while (nz--) sum -= *v++ * x[*vi++]; */
2653e0e43300SBarry Smith     v   += nz;
26548fc3a347SHong Zhang     vi  += nz; vi++;
2655fe31b70dSHong Zhang     x[i] = *v++ *sum; /* x[i]=aa[adiag[i]]*sum; v++; */
26561d098868SHong Zhang   }
26571d098868SHong Zhang 
26581d098868SHong Zhang   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
26591d098868SHong Zhang   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
26601d098868SHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
26611d098868SHong Zhang   PetscFunctionReturn(0);
26621d098868SHong Zhang }
26631d098868SHong Zhang 
26641d098868SHong Zhang #undef __FUNCT__
2665894a2336SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_newdatastruct"
2666894a2336SHong Zhang PetscErrorCode MatSolve_SeqAIJ_newdatastruct(Mat A,Vec bb,Vec xx)
26671d098868SHong Zhang {
26681d098868SHong Zhang   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
26691d098868SHong Zhang   IS                iscol = a->col,isrow = a->row;
26701d098868SHong Zhang   PetscErrorCode    ierr;
2671894a2336SHong Zhang   PetscInt          i,n=A->rmap->n,*vi,*ai=a->i,*aj=a->j,nz,k;
26721d098868SHong Zhang   const PetscInt    *rout,*cout,*r,*c;
2673894a2336SHong Zhang   PetscScalar       *x,*tmp,*tmps,sum;
26741d098868SHong Zhang   const PetscScalar *b;
26751d098868SHong Zhang   const MatScalar   *aa = a->a,*v;
26761d098868SHong Zhang 
26771d098868SHong Zhang   PetscFunctionBegin;
26781d098868SHong Zhang   if (!n) PetscFunctionReturn(0);
26791d098868SHong Zhang 
26801d098868SHong Zhang   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
26811d098868SHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
26821d098868SHong Zhang   tmp  = a->solve_work;
26831d098868SHong Zhang 
26841d098868SHong Zhang   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
2685894a2336SHong Zhang   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
26861d098868SHong Zhang 
26871d098868SHong Zhang   /* forward solve the lower triangular */
26881d098868SHong Zhang   tmp[0] = b[*r++];
26891d098868SHong Zhang   tmps   = tmp;
26901d098868SHong Zhang   v      = aa;
26911d098868SHong Zhang   vi     = aj;
26921d098868SHong Zhang   for (i=1; i<n; i++) {
26931d098868SHong Zhang     nz  = ai[i+1] - ai[i];
2694894a2336SHong Zhang     sum = b[*r++];
2695894a2336SHong Zhang     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
2696894a2336SHong Zhang     tmp[i] = sum;
26971d098868SHong Zhang     v += nz; vi += nz;
26981d098868SHong Zhang   }
26991d098868SHong Zhang 
27001d098868SHong Zhang   /* backward solve the upper triangular */
2701894a2336SHong Zhang   k  = n+1;
2702894a2336SHong Zhang   v  = aa + ai[k]; /* 1st entry of U(n-1,:) */
2703894a2336SHong Zhang   vi = aj + ai[k];
27041d098868SHong Zhang   for (i=n-1; i>=0; i--){
2705894a2336SHong Zhang     k  = 2*n-i;
2706894a2336SHong Zhang     nz = ai[k +1] - ai[k] - 1;
2707894a2336SHong Zhang     sum = tmp[i];
2708894a2336SHong Zhang     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
2709894a2336SHong Zhang     x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */
27101d098868SHong Zhang     v += nz+1; vi += nz+1;
27111d098868SHong Zhang   }
27121d098868SHong Zhang 
27131d098868SHong Zhang   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
27141d098868SHong Zhang   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
27151d098868SHong Zhang   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
27161d098868SHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
27171d098868SHong Zhang   ierr = PetscLogFlops(2*a->nz - A->cmap->n);CHKERRQ(ierr);
27181d098868SHong Zhang   PetscFunctionReturn(0);
27191d098868SHong Zhang }
27201d098868SHong Zhang 
27211d098868SHong Zhang #undef __FUNCT__
27221d098868SHong Zhang #define __FUNCT__ "MatILUDTFactor_SeqAIJ"
27231d098868SHong Zhang PetscErrorCode MatILUDTFactor_SeqAIJ(Mat A,IS isrow,IS iscol,const MatFactorInfo *info,Mat *fact)
27241d098868SHong Zhang {
27251d098868SHong Zhang   Mat                B = *fact;
2726599ef60dSHong Zhang   Mat_SeqAIJ         *a=(Mat_SeqAIJ*)A->data,*b;
2727599ef60dSHong Zhang   IS                 isicol;
2728599ef60dSHong Zhang   PetscErrorCode     ierr;
27291d098868SHong Zhang   const PetscInt     *r,*ic;
27301d098868SHong Zhang   PetscInt           i,n=A->rmap->n,*ai=a->i,*aj=a->j,*ajtmp,*adiag;
2731dc3a2fd3SHong Zhang   PetscInt           *bi,*bj,*bdiag,*bdiag_rev;
2732f61a948aSLisandro Dalcin   PetscInt           row,nzi,nzi_bl,nzi_bu,*im,nzi_al,nzi_au;
27331d098868SHong Zhang   PetscInt           nlnk,*lnk;
27341d098868SHong Zhang   PetscBT            lnkbt;
27351d098868SHong Zhang   PetscTruth         row_identity,icol_identity,both_identity;
27368aee2decSHong Zhang   MatScalar          *aatmp,*pv,*batmp,*ba,*rtmp,*pc,multiplier,*vtmp,diag_tmp;
27371d098868SHong Zhang   const PetscInt     *ics;
27381d098868SHong Zhang   PetscInt           j,nz,*pj,*bjtmp,k,ncut,*jtmp;
2739f61a948aSLisandro Dalcin   PetscReal          dt=info->dt,dtcol=info->dtcol,shift=info->shiftinblocks;
2740f61a948aSLisandro Dalcin   PetscInt           dtcount=(PetscInt)info->dtcount,nnz_max;
2741599ef60dSHong Zhang   PetscTruth         missing;
2742599ef60dSHong Zhang 
2743599ef60dSHong Zhang   PetscFunctionBegin;
2744f61a948aSLisandro Dalcin 
2745f61a948aSLisandro Dalcin   if (dt      == PETSC_DEFAULT) dt      = 0.005;
2746f61a948aSLisandro Dalcin   if (dtcol   == PETSC_DEFAULT) dtcol   = 0.01; /* XXX unused! */
2747f61a948aSLisandro Dalcin   if (dtcount == PETSC_DEFAULT) dtcount = (PetscInt)(1.5*a->rmax);
2748f61a948aSLisandro Dalcin 
27491d098868SHong Zhang   /* ------- symbolic factorization, can be reused ---------*/
27501d098868SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr);
27511d098868SHong Zhang   if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i);
27521d098868SHong Zhang   adiag=a->diag;
2753599ef60dSHong Zhang 
2754599ef60dSHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
2755599ef60dSHong Zhang 
2756599ef60dSHong Zhang   /* bdiag is location of diagonal in factor */
2757a6226a1aSHong Zhang   ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
2758a6226a1aSHong Zhang   bdiag_rev = bdiag + n+1;
2759599ef60dSHong Zhang 
27601d098868SHong Zhang   /* allocate row pointers bi */
27618fc3a347SHong Zhang   ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
27621d098868SHong Zhang 
27631d098868SHong Zhang   /* allocate bj and ba; max num of nonzero entries is (ai[n]+2*n*dtcount+2) */
2764393d3a68SHong Zhang   if (dtcount > n-1) dtcount = n-1; /* diagonal is excluded */
27651d098868SHong Zhang   nnz_max  = ai[n]+2*n*dtcount+2;
27668fc3a347SHong Zhang 
27678fc3a347SHong Zhang   ierr = PetscMalloc((nnz_max+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
27688fc3a347SHong Zhang   ierr = PetscMalloc((nnz_max+1)*sizeof(MatScalar),&ba);CHKERRQ(ierr);
2769599ef60dSHong Zhang 
2770599ef60dSHong Zhang   /* put together the new matrix */
2771599ef60dSHong Zhang   ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
2772599ef60dSHong Zhang   ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr);
2773599ef60dSHong Zhang   b    = (Mat_SeqAIJ*)(B)->data;
2774599ef60dSHong Zhang   b->free_a       = PETSC_TRUE;
2775599ef60dSHong Zhang   b->free_ij      = PETSC_TRUE;
2776599ef60dSHong Zhang   b->singlemalloc = PETSC_FALSE;
2777599ef60dSHong Zhang   b->a          = ba;
2778599ef60dSHong Zhang   b->j          = bj;
2779599ef60dSHong Zhang   b->i          = bi;
2780599ef60dSHong Zhang   b->diag       = bdiag;
2781599ef60dSHong Zhang   b->ilen       = 0;
2782599ef60dSHong Zhang   b->imax       = 0;
2783599ef60dSHong Zhang   b->row        = isrow;
2784599ef60dSHong Zhang   b->col        = iscol;
2785599ef60dSHong Zhang   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
2786599ef60dSHong Zhang   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
2787599ef60dSHong Zhang   b->icol       = isicol;
2788599ef60dSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
2789599ef60dSHong Zhang 
27901d098868SHong Zhang   ierr = PetscLogObjectMemory(B,nnz_max*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
27911d098868SHong Zhang   b->maxnz = nnz_max;
2792599ef60dSHong Zhang 
2793599ef60dSHong Zhang   (B)->factor                = MAT_FACTOR_ILUDT;
2794599ef60dSHong Zhang   (B)->info.factor_mallocs   = 0;
27951d098868SHong Zhang   (B)->info.fill_ratio_given = ((PetscReal)nnz_max)/((PetscReal)ai[n]);
27961d098868SHong Zhang   CHKMEMQ;
27971d098868SHong Zhang   /* ------- end of symbolic factorization ---------*/
2798599ef60dSHong Zhang 
2799599ef60dSHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
2800599ef60dSHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
2801599ef60dSHong Zhang   ics  = ic;
2802599ef60dSHong Zhang 
2803599ef60dSHong Zhang   /* linked list for storing column indices of the active row */
2804599ef60dSHong Zhang   nlnk = n + 1;
2805599ef60dSHong Zhang   ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
28061d098868SHong Zhang 
28071d098868SHong Zhang   /* im: used by PetscLLAddSortedLU(); jtmp: working array for column indices of active row */
28081d098868SHong Zhang   ierr = PetscMalloc((2*n+1)*sizeof(PetscInt),&im);CHKERRQ(ierr);
28091d098868SHong Zhang   jtmp = im + n;
28101d098868SHong Zhang   /* rtmp, vtmp: working arrays for sparse and contiguous row entries of active row */
28118369b28dSHong Zhang   ierr = PetscMalloc((2*n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr);
28128aee2decSHong Zhang   ierr = PetscMemzero(rtmp,(n+1)*sizeof(PetscScalar));CHKERRQ(ierr);
28138369b28dSHong Zhang   vtmp = rtmp + n;
2814599ef60dSHong Zhang 
2815599ef60dSHong Zhang   bi[0]    = 0;
28168fc3a347SHong Zhang   bdiag[0] = nnz_max-1; /* location of diag[0] in factor B */
2817dc3a2fd3SHong Zhang   bdiag_rev[n] = bdiag[0];
28188fc3a347SHong Zhang   bi[2*n+1] = bdiag[0]+1; /* endof bj and ba array */
2819599ef60dSHong Zhang   for (i=0; i<n; i++) {
2820599ef60dSHong Zhang     /* copy initial fill into linked list */
2821599ef60dSHong Zhang     nzi = 0; /* nonzeros for active row i */
28228369b28dSHong Zhang     nzi = ai[r[i]+1] - ai[r[i]];
28238369b28dSHong 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);
28248369b28dSHong Zhang     nzi_al = adiag[r[i]] - ai[r[i]];
28258369b28dSHong Zhang     nzi_au = ai[r[i]+1] - adiag[r[i]] -1;
2826599ef60dSHong Zhang     ajtmp = aj + ai[r[i]];
28278369b28dSHong Zhang     ierr = PetscLLAddPerm(nzi,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
2828599ef60dSHong Zhang 
2829599ef60dSHong Zhang     /* load in initial (unfactored row) */
28301d098868SHong Zhang     aatmp = a->a + ai[r[i]];
28318369b28dSHong Zhang     for (j=0; j<nzi; j++) {
28321d098868SHong Zhang       rtmp[ics[*ajtmp++]] = *aatmp++;
2833599ef60dSHong Zhang     }
2834599ef60dSHong Zhang 
2835599ef60dSHong Zhang     /* add pivot rows into linked list */
2836599ef60dSHong Zhang     row = lnk[n];
2837599ef60dSHong Zhang     while (row < i ) {
28381d098868SHong Zhang       nzi_bl = bi[row+1] - bi[row] + 1;
28391d098868SHong Zhang       bjtmp = bj + bdiag[row+1]+1; /* points to 1st column next to the diagonal in U */
28401d098868SHong Zhang       ierr  = PetscLLAddSortedLU(bjtmp,row,nlnk,lnk,lnkbt,i,nzi_bl,im);CHKERRQ(ierr);
2841599ef60dSHong Zhang       nzi  += nlnk;
2842599ef60dSHong Zhang       row   = lnk[row];
2843599ef60dSHong Zhang     }
2844599ef60dSHong Zhang 
28458369b28dSHong Zhang     /* copy data from lnk into jtmp, then initialize lnk */
28468369b28dSHong Zhang     ierr = PetscLLClean(n,n,nzi,lnk,jtmp,lnkbt);CHKERRQ(ierr);
2847599ef60dSHong Zhang 
2848599ef60dSHong Zhang     /* numerical factorization */
28498369b28dSHong Zhang     bjtmp = jtmp;
2850599ef60dSHong Zhang     row   = *bjtmp++; /* 1st pivot row */
2851599ef60dSHong Zhang     while  ( row < i ) {
2852599ef60dSHong Zhang       pc         = rtmp + row;
28533c2a7987SHong Zhang       pv         = ba + bdiag[row]; /* 1./(diag of the pivot row) */
28543c2a7987SHong Zhang       multiplier = (*pc) * (*pv);
2855599ef60dSHong Zhang       *pc        = multiplier;
28563c2a7987SHong Zhang       if (PetscAbsScalar(*pc) > dt){ /* apply tolerance dropping rule */
28571d098868SHong Zhang         pj         = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */
28581d098868SHong Zhang         pv         = ba + bdiag[row+1] + 1;
28591d098868SHong Zhang         /* if (multiplier < -1.0 or multiplier >1.0) printf("row/prow %d, %d, multiplier %g\n",i,row,multiplier); */
28601d098868SHong Zhang         nz         = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */
28611d098868SHong Zhang         for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++);
2862dc0b31edSSatish Balay         ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr);
2863599ef60dSHong Zhang       }
2864599ef60dSHong Zhang       row = *bjtmp++;
2865599ef60dSHong Zhang     }
2866599ef60dSHong Zhang 
28678369b28dSHong Zhang     /* copy sparse rtmp into contiguous vtmp; separate L and U part */
28688aee2decSHong Zhang     diag_tmp = rtmp[i];  /* save diagonal value - may not needed?? */
28698369b28dSHong Zhang     nzi_bl = 0; j = 0;
28708369b28dSHong Zhang     while (jtmp[j] < i){ /* Note: jtmp is sorted */
28718aee2decSHong Zhang       vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0;
28728369b28dSHong Zhang       nzi_bl++; j++;
28738369b28dSHong Zhang     }
28748369b28dSHong Zhang     nzi_bu = nzi - nzi_bl -1;
28758369b28dSHong Zhang     while (j < nzi){
28768aee2decSHong Zhang       vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0;
28778369b28dSHong Zhang       j++;
28788369b28dSHong Zhang     }
28798369b28dSHong Zhang 
2880599ef60dSHong Zhang     bjtmp = bj + bi[i];
2881599ef60dSHong Zhang     batmp = ba + bi[i];
28828369b28dSHong Zhang     /* apply level dropping rule to L part */
28838369b28dSHong Zhang     ncut = nzi_al + dtcount;
28848369b28dSHong Zhang     if (ncut < nzi_bl){
28858369b28dSHong Zhang       ierr = PetscSortSplit(ncut,nzi_bl,vtmp,jtmp);CHKERRQ(ierr);
28868369b28dSHong Zhang       ierr = PetscSortIntWithScalarArray(ncut,jtmp,vtmp);CHKERRQ(ierr);
2887599ef60dSHong Zhang     } else {
28888369b28dSHong Zhang       ncut = nzi_bl;
28898369b28dSHong Zhang     }
28908369b28dSHong Zhang     for (j=0; j<ncut; j++){
28918369b28dSHong Zhang       bjtmp[j] = jtmp[j];
28928369b28dSHong Zhang       batmp[j] = vtmp[j];
28936da515a1SHong Zhang       /* printf(" (%d,%g),",bjtmp[j],batmp[j]); */
28948369b28dSHong Zhang     }
28951d098868SHong Zhang     bi[i+1] = bi[i] + ncut;
28968369b28dSHong Zhang     nzi = ncut + 1;
28978369b28dSHong Zhang 
28988369b28dSHong Zhang     /* apply level dropping rule to U part */
28998369b28dSHong Zhang     ncut = nzi_au + dtcount;
29008369b28dSHong Zhang     if (ncut < nzi_bu){
29018369b28dSHong Zhang       ierr = PetscSortSplit(ncut,nzi_bu,vtmp+nzi_bl+1,jtmp+nzi_bl+1);CHKERRQ(ierr);
29028369b28dSHong Zhang       ierr = PetscSortIntWithScalarArray(ncut,jtmp+nzi_bl+1,vtmp+nzi_bl+1);CHKERRQ(ierr);
29038369b28dSHong Zhang     } else {
29048369b28dSHong Zhang       ncut = nzi_bu;
29058369b28dSHong Zhang     }
29068369b28dSHong Zhang     nzi += ncut;
29071d098868SHong Zhang 
29081d098868SHong Zhang     /* mark bdiagonal */
29091d098868SHong Zhang     bdiag[i+1]       = bdiag[i] - (ncut + 1);
2910dc3a2fd3SHong Zhang     bdiag_rev[n-i-1] = bdiag[i+1];
29118fc3a347SHong Zhang     bi[2*n - i]      = bi[2*n - i +1] - (ncut + 1);
29121d098868SHong Zhang     bjtmp = bj + bdiag[i];
29131d098868SHong Zhang     batmp = ba + bdiag[i];
29141d098868SHong Zhang     *bjtmp = i;
29158aee2decSHong Zhang     *batmp = diag_tmp; /* rtmp[i]; */
2916c9c72f8fSHong Zhang     if (*batmp == 0.0) {
2917c9c72f8fSHong Zhang       *batmp = dt+shift;
2918bd1bc851SBarry Smith       /* printf(" row %d add shift %g\n",i,shift); */
2919c9c72f8fSHong Zhang     }
2920b78a477dSHong Zhang     *batmp = 1.0/(*batmp); /* invert diagonal entries for simplier triangular solves */
29216da515a1SHong Zhang     /* printf(" (%d,%g),",*bjtmp,*batmp); */
29221d098868SHong Zhang 
29231d098868SHong Zhang     bjtmp = bj + bdiag[i+1]+1;
29241d098868SHong Zhang     batmp = ba + bdiag[i+1]+1;
29258369b28dSHong Zhang     for (k=0; k<ncut; k++){
29261d098868SHong Zhang       bjtmp[k] = jtmp[nzi_bl+1+k];
29271d098868SHong Zhang       batmp[k] = vtmp[nzi_bl+1+k];
29286da515a1SHong Zhang       /* printf(" (%d,%g),",bjtmp[k],batmp[k]); */
2929599ef60dSHong Zhang     }
29306da515a1SHong Zhang     /* printf("\n"); */
2931599ef60dSHong Zhang 
2932599ef60dSHong Zhang     im[i]   = nzi; /* used by PetscLLAddSortedLU() */
2933599ef60dSHong Zhang     /*
29341d098868SHong Zhang     printf("row %d: bi %d, bdiag %d\n",i,bi[i],bdiag[i]);
29358369b28dSHong Zhang     printf(" ----------------------------\n");
2936599ef60dSHong Zhang     */
29378369b28dSHong Zhang   } /* for (i=0; i<n; i++) */
29381d098868SHong Zhang   /* printf("end of L %d, beginning of U %d\n",bi[n],bdiag[n]); */
29391d098868SHong 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]);
2940599ef60dSHong Zhang 
2941599ef60dSHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
2942599ef60dSHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
2943599ef60dSHong Zhang 
2944599ef60dSHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
2945599ef60dSHong Zhang   ierr = PetscFree(im);CHKERRQ(ierr);
2946599ef60dSHong Zhang   ierr = PetscFree(rtmp);CHKERRQ(ierr);
2947599ef60dSHong Zhang 
2948599ef60dSHong Zhang   ierr = PetscLogFlops(B->cmap->n);CHKERRQ(ierr);
29491d098868SHong Zhang   b->maxnz = b->nz = bi[n] + bdiag[0] - bdiag[n];
2950599ef60dSHong Zhang 
2951599ef60dSHong Zhang   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
2952599ef60dSHong Zhang   ierr = ISIdentity(isicol,&icol_identity);CHKERRQ(ierr);
2953599ef60dSHong Zhang   both_identity = (PetscTruth) (row_identity && icol_identity);
2954599ef60dSHong Zhang   if (row_identity && icol_identity) {
2955894a2336SHong Zhang     B->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_newdatastruct;
2956599ef60dSHong Zhang   } else {
2957894a2336SHong Zhang     B->ops->solve = MatSolve_SeqAIJ_newdatastruct;
2958599ef60dSHong Zhang   }
2959599ef60dSHong Zhang 
2960b78a477dSHong Zhang   B->ops->lufactorsymbolic  = MatILUDTFactorSymbolic_SeqAIJ;
2961b78a477dSHong Zhang   B->ops->lufactornumeric   = MatILUDTFactorNumeric_SeqAIJ;
29621d098868SHong Zhang   B->ops->solveadd          = 0;
29631d098868SHong Zhang   B->ops->solvetranspose    = 0;
29641d098868SHong Zhang   B->ops->solvetransposeadd = 0;
29651d098868SHong Zhang   B->ops->matsolve          = 0;
2966599ef60dSHong Zhang   B->assembled              = PETSC_TRUE;
2967599ef60dSHong Zhang   B->preallocated           = PETSC_TRUE;
2968599ef60dSHong Zhang   PetscFunctionReturn(0);
2969599ef60dSHong Zhang }
2970599ef60dSHong Zhang 
29713c2a7987SHong Zhang /* a wraper of MatILUDTFactor_SeqAIJ() */
29723c2a7987SHong Zhang #undef __FUNCT__
29733c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorSymbolic_SeqAIJ"
29743c2a7987SHong Zhang PetscErrorCode PETSCMAT_DLLEXPORT MatILUDTFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS row,IS col,const MatFactorInfo *info)
29753c2a7987SHong Zhang {
29763c2a7987SHong Zhang   PetscErrorCode     ierr;
29773c2a7987SHong Zhang 
29783c2a7987SHong Zhang   PetscFunctionBegin;
29793c2a7987SHong Zhang   ierr = MatILUDTFactor_SeqAIJ(A,row,col,info,&fact);CHKERRQ(ierr);
29803c2a7987SHong Zhang 
29813c2a7987SHong Zhang   fact->ops->lufactornumeric = MatILUDTFactorNumeric_SeqAIJ;
29823c2a7987SHong Zhang   PetscFunctionReturn(0);
29833c2a7987SHong Zhang }
29843c2a7987SHong Zhang 
29853c2a7987SHong Zhang /*
29863c2a7987SHong Zhang    same as MatLUFactorNumeric_SeqAIJ(), except using contiguous array matrix factors
29873c2a7987SHong Zhang    - intend to replace existing MatLUFactorNumeric_SeqAIJ()
29883c2a7987SHong Zhang */
29893c2a7987SHong Zhang #undef __FUNCT__
29903c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorNumeric_SeqAIJ"
29913c2a7987SHong Zhang PetscErrorCode PETSCMAT_DLLEXPORT MatILUDTFactorNumeric_SeqAIJ(Mat fact,Mat A,const MatFactorInfo *info)
29923c2a7987SHong Zhang {
29933c2a7987SHong Zhang   Mat            C=fact;
29943c2a7987SHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data;
29953c2a7987SHong Zhang   IS             isrow = b->row,isicol = b->icol;
29963c2a7987SHong Zhang   PetscErrorCode ierr;
29973c2a7987SHong Zhang   const PetscInt *r,*ic,*ics;
2998b78a477dSHong Zhang   PetscInt       i,j,k,n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j;
2999b78a477dSHong Zhang   PetscInt       *ajtmp,*bjtmp,nz,nzl,nzu,row,*bdiag = b->diag,*pj;
30003c2a7987SHong Zhang   MatScalar      *rtmp,*pc,multiplier,*v,*pv,*aa=a->a;
3001b78a477dSHong Zhang   PetscReal      dt=info->dt,shift=info->shiftinblocks;
3002b78a477dSHong Zhang   PetscTruth     row_identity, col_identity;
30033c2a7987SHong Zhang 
30043c2a7987SHong Zhang   PetscFunctionBegin;
30053c2a7987SHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
30063c2a7987SHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
3007b78a477dSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr);
30083c2a7987SHong Zhang   ics  = ic;
30093c2a7987SHong Zhang 
30103c2a7987SHong Zhang   for (i=0; i<n; i++){
3011b78a477dSHong Zhang     /* initialize rtmp array */
3012b78a477dSHong Zhang     nzl   = bi[i+1] - bi[i];       /* num of nozeros in L(i,:) */
30133c2a7987SHong Zhang     bjtmp = bj + bi[i];
3014b78a477dSHong Zhang     for  (j=0; j<nzl; j++) rtmp[*bjtmp++] = 0.0;
3015b78a477dSHong Zhang     rtmp[i] = 0.0;
3016b78a477dSHong Zhang     nzu   = bdiag[i] - bdiag[i+1]; /* num of nozeros in U(i,:) */
3017b78a477dSHong Zhang     bjtmp = bj + bdiag[i+1] + 1;
3018b78a477dSHong Zhang     for  (j=0; j<nzu; j++) rtmp[*bjtmp++] = 0.0;
30193c2a7987SHong Zhang 
3020b78a477dSHong Zhang     /* load in initial unfactored row of A */
30216da515a1SHong Zhang     /* printf("row %d\n",i); */
30223c2a7987SHong Zhang     nz    = ai[r[i]+1] - ai[r[i]];
30233c2a7987SHong Zhang     ajtmp = aj + ai[r[i]];
30243c2a7987SHong Zhang     v     = aa + ai[r[i]];
30253c2a7987SHong Zhang     for (j=0; j<nz; j++) {
3026b78a477dSHong Zhang       rtmp[ics[*ajtmp++]] = v[j];
30276da515a1SHong Zhang       /* printf(" (%d,%g),",ics[ajtmp[j]],rtmp[ics[ajtmp[j]]]); */
30283c2a7987SHong Zhang     }
30296da515a1SHong Zhang     /* printf("\n"); */
30303c2a7987SHong Zhang 
3031b78a477dSHong Zhang     /* numerical factorization */
3032b78a477dSHong Zhang     bjtmp = bj + bi[i]; /* point to 1st entry of L(i,:) */
3033b78a477dSHong Zhang     nzl   = bi[i+1] - bi[i]; /* num of entries in L(i,:) */
3034b78a477dSHong Zhang     k = 0;
3035b78a477dSHong Zhang     while (k < nzl){
30363c2a7987SHong Zhang       row   = *bjtmp++;
30376da515a1SHong Zhang       /* printf("  prow %d\n",row); */
30383c2a7987SHong Zhang       pc         = rtmp + row;
3039b78a477dSHong Zhang       pv         = b->a + bdiag[row]; /* 1./(diag of the pivot row) */
3040b78a477dSHong Zhang       multiplier = (*pc) * (*pv);
30413c2a7987SHong Zhang       *pc        = multiplier;
3042b78a477dSHong Zhang       if (PetscAbsScalar(multiplier) > dt){
3043b78a477dSHong Zhang         pj         = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */
3044b78a477dSHong Zhang         pv         = b->a + bdiag[row+1] + 1;
3045b78a477dSHong Zhang         nz         = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */
3046b78a477dSHong Zhang         for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++);
30476da515a1SHong Zhang         /* ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); */
30483c2a7987SHong Zhang       }
3049b78a477dSHong Zhang       k++;
30503c2a7987SHong Zhang     }
30513c2a7987SHong Zhang 
3052b78a477dSHong Zhang     /* finished row so stick it into b->a */
3053b78a477dSHong Zhang     /* L-part */
3054b78a477dSHong Zhang     pv = b->a + bi[i] ;
3055b78a477dSHong Zhang     pj = bj + bi[i] ;
3056b78a477dSHong Zhang     nzl = bi[i+1] - bi[i];
3057b78a477dSHong Zhang     for (j=0; j<nzl; j++) {
3058b78a477dSHong Zhang       pv[j] = rtmp[pj[j]];
30596da515a1SHong Zhang       /* printf(" (%d,%g),",pj[j],pv[j]); */
30603c2a7987SHong Zhang     }
3061b78a477dSHong Zhang 
3062b78a477dSHong Zhang     /* diagonal: invert diagonal entries for simplier triangular solves */
3063b78a477dSHong Zhang     if (rtmp[i] == 0.0) rtmp[i] = dt+shift;
3064b78a477dSHong Zhang     b->a[bdiag[i]] = 1.0/rtmp[i];
30656da515a1SHong Zhang     /* printf(" (%d,%g),",i,b->a[bdiag[i]]); */
3066b78a477dSHong Zhang 
3067b78a477dSHong Zhang     /* U-part */
3068b78a477dSHong Zhang     pv = b->a + bdiag[i+1] + 1;
3069b78a477dSHong Zhang     pj = bj + bdiag[i+1] + 1;
3070b78a477dSHong Zhang     nzu = bdiag[i] - bdiag[i+1] - 1;
3071b78a477dSHong Zhang     for (j=0; j<nzu; j++) {
3072b78a477dSHong Zhang       pv[j] = rtmp[pj[j]];
30736da515a1SHong Zhang       /* printf(" (%d,%g),",pj[j],pv[j]); */
3074b78a477dSHong Zhang     }
30756da515a1SHong Zhang     /* printf("\n"); */
3076b78a477dSHong Zhang   }
3077b78a477dSHong Zhang 
30783c2a7987SHong Zhang   ierr = PetscFree(rtmp);CHKERRQ(ierr);
30793c2a7987SHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
30803c2a7987SHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
3081b78a477dSHong Zhang 
30823c2a7987SHong Zhang   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
30833c2a7987SHong Zhang   ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr);
30843c2a7987SHong Zhang   if (row_identity && col_identity) {
3085894a2336SHong Zhang     C->ops->solve   = MatSolve_SeqAIJ_NaturalOrdering_newdatastruct;
30863c2a7987SHong Zhang   } else {
3087894a2336SHong Zhang     C->ops->solve   = MatSolve_SeqAIJ_newdatastruct;
30883c2a7987SHong Zhang   }
3089b78a477dSHong Zhang   C->ops->solveadd           = 0;
3090b78a477dSHong Zhang   C->ops->solvetranspose     = 0;
3091b78a477dSHong Zhang   C->ops->solvetransposeadd  = 0;
3092b78a477dSHong Zhang   C->ops->matsolve           = 0;
30933c2a7987SHong Zhang   C->assembled    = PETSC_TRUE;
30943c2a7987SHong Zhang   C->preallocated = PETSC_TRUE;
30953c2a7987SHong Zhang   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
30963c2a7987SHong Zhang   PetscFunctionReturn(0);
30973c2a7987SHong Zhang }
3098