xref: /petsc/src/mat/impls/aij/seq/aijfact.c (revision e60cf9a048d47efee666bcb586e0da2d34090e24)
1be1d678aSKris Buschelman #define PETSCMAT_DLL
2289bc588SBarry Smith 
3b377110cSBarry Smith 
47c4f633dSBarry Smith #include "../src/mat/impls/aij/seq/aij.h"
5783ef271SHong Zhang #include "../src/mat/impls/sbaij/seq/sbaij.h"
61393bc97SHong Zhang #include "petscbt.h"
77c4f633dSBarry Smith #include "../src/mat/utils/freespace.h"
83b2fbd54SBarry Smith 
9f6a26d55SMatthew Knepley EXTERN_C_BEGIN
104a2ae208SSatish Balay #undef __FUNCT__
114a2ae208SSatish Balay #define __FUNCT__ "MatOrdering_Flow_SeqAIJ"
128fa24674SBarry Smith /*
138fa24674SBarry Smith       Computes an ordering to get most of the large numerical values in the lower triangular part of the matrix
148fa24674SBarry Smith */
15dfbe8321SBarry Smith PetscErrorCode MatOrdering_Flow_SeqAIJ(Mat mat,const MatOrderingType type,IS *irow,IS *icol)
163b2fbd54SBarry Smith {
178fa24674SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)mat->data;
188fa24674SBarry Smith   PetscErrorCode    ierr;
19889a8dedSBarry Smith   PetscInt          i,j,jj,k, kk,n = mat->rmap->n, current = 0, newcurrent = 0,*order;
208fa24674SBarry Smith   const PetscInt    *ai = a->i, *aj = a->j;
218fa24674SBarry Smith   const PetscScalar *aa = a->a;
228fa24674SBarry Smith   PetscTruth        *done;
23889a8dedSBarry Smith   PetscReal         best,past = 0,future;
243a40ed3dSBarry Smith 
258fa24674SBarry Smith   PetscFunctionBegin;
268fa24674SBarry Smith   /* pick initial row */
278fa24674SBarry Smith   best = -1;
288fa24674SBarry Smith   for (i=0; i<n; i++) {
2975567043SBarry Smith     future = 0.0;
308fa24674SBarry Smith     for (j=ai[i]; j<ai[i+1]; j++) {
318fa24674SBarry Smith       if (aj[j] != i) future  += PetscAbsScalar(aa[j]); else past = PetscAbsScalar(aa[j]);
328fa24674SBarry Smith     }
338fa24674SBarry Smith     if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */
348fa24674SBarry Smith     if (past/future > best) {
358fa24674SBarry Smith       best = past/future;
368fa24674SBarry Smith       current = i;
378fa24674SBarry Smith     }
388fa24674SBarry Smith   }
398fa24674SBarry Smith 
408fa24674SBarry Smith   ierr = PetscMalloc(n*sizeof(PetscTruth),&done);CHKERRQ(ierr);
418fa24674SBarry Smith   ierr = PetscMemzero(done,n*sizeof(PetscTruth));CHKERRQ(ierr);
420e83c824SBarry Smith   ierr = PetscMalloc(n*sizeof(PetscInt),&order);CHKERRQ(ierr);
438fa24674SBarry Smith   order[0] = current;
448fa24674SBarry Smith   for (i=0; i<n-1; i++) {
458fa24674SBarry Smith     done[current] = PETSC_TRUE;
468fa24674SBarry Smith     best          = -1;
478fa24674SBarry Smith     /* loop over all neighbors of current pivot */
488fa24674SBarry Smith     for (j=ai[current]; j<ai[current+1]; j++) {
498fa24674SBarry Smith       jj = aj[j];
508fa24674SBarry Smith       if (done[jj]) continue;
518fa24674SBarry Smith       /* loop over columns of potential next row computing weights for below and above diagonal */
528fa24674SBarry Smith       past = future = 0.0;
538fa24674SBarry Smith       for (k=ai[jj]; k<ai[jj+1]; k++) {
548fa24674SBarry Smith         kk = aj[k];
558fa24674SBarry Smith         if (done[kk]) past += PetscAbsScalar(aa[k]);
568fa24674SBarry Smith         else if (kk != jj) future  += PetscAbsScalar(aa[k]);
578fa24674SBarry Smith       }
588fa24674SBarry Smith       if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */
598fa24674SBarry Smith       if (past/future > best) {
608fa24674SBarry Smith         best = past/future;
618fa24674SBarry Smith         newcurrent = jj;
628fa24674SBarry Smith       }
638fa24674SBarry Smith     }
648fa24674SBarry Smith     if (best == -1) { /* no neighbors to select from so select best of all that remain */
658fa24674SBarry Smith       best = -1;
668fa24674SBarry Smith       for (k=0; k<n; k++) {
678fa24674SBarry Smith         if (done[k]) continue;
6875567043SBarry Smith         future = 0.0;
6975567043SBarry Smith         past   = 0.0;
708fa24674SBarry Smith         for (j=ai[k]; j<ai[k+1]; j++) {
718fa24674SBarry Smith           kk = aj[j];
728fa24674SBarry Smith           if (done[kk]) past += PetscAbsScalar(aa[j]);
738fa24674SBarry Smith           else if (kk != k) future  += PetscAbsScalar(aa[j]);
748fa24674SBarry Smith         }
758fa24674SBarry Smith         if (!future) future = 1.e-10; /* if there is zero in the upper diagonal part want to rank this row high */
768fa24674SBarry Smith         if (past/future > best) {
778fa24674SBarry Smith           best = past/future;
788fa24674SBarry Smith           newcurrent = k;
798fa24674SBarry Smith         }
808fa24674SBarry Smith       }
818fa24674SBarry Smith     }
828fa24674SBarry Smith     if (current == newcurrent) SETERRQ(PETSC_ERR_PLIB,"newcurrent cannot be current");
838fa24674SBarry Smith     current = newcurrent;
848fa24674SBarry Smith     order[i+1] = current;
858fa24674SBarry Smith   }
868fa24674SBarry Smith   ierr = ISCreateGeneral(PETSC_COMM_SELF,n,order,irow);CHKERRQ(ierr);
878fa24674SBarry Smith   *icol = *irow;
888fa24674SBarry Smith   ierr = PetscObjectReference((PetscObject)*irow);CHKERRQ(ierr);
898fa24674SBarry Smith   ierr = PetscFree(done);CHKERRQ(ierr);
908fa24674SBarry Smith   ierr = PetscFree(order);CHKERRQ(ierr);
91d64ed03dSBarry Smith   PetscFunctionReturn(0);
923b2fbd54SBarry Smith }
93f6a26d55SMatthew Knepley EXTERN_C_END
943b2fbd54SBarry Smith 
95e631078cSBarry Smith EXTERN_C_BEGIN
964a2ae208SSatish Balay #undef __FUNCT__
97db4efbfdSBarry Smith #define __FUNCT__ "MatGetFactorAvailable_seqaij_petsc"
98db4efbfdSBarry Smith PetscErrorCode MatGetFactorAvailable_seqaij_petsc(Mat A,MatFactorType ftype,PetscTruth *flg)
99db4efbfdSBarry Smith {
100db4efbfdSBarry Smith   PetscFunctionBegin;
101db4efbfdSBarry Smith   *flg = PETSC_TRUE;
102db4efbfdSBarry Smith   PetscFunctionReturn(0);
103db4efbfdSBarry Smith }
104db4efbfdSBarry Smith EXTERN_C_END
105db4efbfdSBarry Smith 
106db4efbfdSBarry Smith EXTERN_C_BEGIN
107db4efbfdSBarry Smith #undef __FUNCT__
108b24902e0SBarry Smith #define __FUNCT__ "MatGetFactor_seqaij_petsc"
109b24902e0SBarry Smith PetscErrorCode MatGetFactor_seqaij_petsc(Mat A,MatFactorType ftype,Mat *B)
110b24902e0SBarry Smith {
111d0f46423SBarry Smith   PetscInt           n = A->rmap->n;
112b24902e0SBarry Smith   PetscErrorCode     ierr;
113b24902e0SBarry Smith 
114b24902e0SBarry Smith   PetscFunctionBegin;
115b24902e0SBarry Smith   ierr = MatCreate(((PetscObject)A)->comm,B);CHKERRQ(ierr);
116b24902e0SBarry Smith   ierr = MatSetSizes(*B,n,n,n,n);CHKERRQ(ierr);
117599ef60dSHong Zhang   if (ftype == MAT_FACTOR_LU || ftype == MAT_FACTOR_ILU || ftype == MAT_FACTOR_ILUDT){
118b24902e0SBarry Smith     ierr = MatSetType(*B,MATSEQAIJ);CHKERRQ(ierr);
119db4efbfdSBarry Smith     (*B)->ops->ilufactorsymbolic = MatILUFactorSymbolic_SeqAIJ;
12035bd34faSBarry Smith     (*B)->ops->lufactorsymbolic  = MatLUFactorSymbolic_SeqAIJ;
1211da05e5aSHong Zhang     (*B)->ops->iludtfactor       = MatILUDTFactor_SeqAIJ;
122b24902e0SBarry Smith   } else if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) {
123b24902e0SBarry Smith     ierr = MatSetType(*B,MATSEQSBAIJ);CHKERRQ(ierr);
124b24902e0SBarry Smith     ierr = MatSeqSBAIJSetPreallocation(*B,1,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
1255c9eb25fSBarry Smith     (*B)->ops->iccfactorsymbolic      = MatICCFactorSymbolic_SeqAIJ;
1265c9eb25fSBarry Smith     (*B)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqAIJ;
127b24902e0SBarry Smith   } else SETERRQ(PETSC_ERR_SUP,"Factor type not supported");
128db4efbfdSBarry Smith   (*B)->factor = ftype;
129b24902e0SBarry Smith   PetscFunctionReturn(0);
130b24902e0SBarry Smith }
131e631078cSBarry Smith EXTERN_C_END
132b24902e0SBarry Smith 
133b24902e0SBarry Smith #undef __FUNCT__
134b9617806SBarry Smith #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ"
1350481f469SBarry Smith PetscErrorCode MatLUFactorSymbolic_SeqAIJ(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
136289bc588SBarry Smith {
137416022c9SBarry Smith   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
138289bc588SBarry Smith   IS                 isicol;
1396849ba73SBarry Smith   PetscErrorCode     ierr;
1405d0c19d7SBarry Smith   const PetscInt     *r,*ic;
1415d0c19d7SBarry Smith   PetscInt           i,n=A->rmap->n,*ai=a->i,*aj=a->j;
1421393bc97SHong Zhang   PetscInt           *bi,*bj,*ajtmp;
1431393bc97SHong Zhang   PetscInt           *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im;
1449e046878SBarry Smith   PetscReal          f;
1454875ba38SHong Zhang   PetscInt           nlnk,*lnk,k,**bi_ptr;
146a1a86e44SBarry Smith   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
1471393bc97SHong Zhang   PetscBT            lnkbt;
148ce3d78c0SShri Abhyankar   PetscTruth         newdatastruct= PETSC_FALSE;
149289bc588SBarry Smith 
1503a40ed3dSBarry Smith   PetscFunctionBegin;
151ce3d78c0SShri Abhyankar   ierr = PetscOptionsGetTruth(PETSC_NULL,"-lu_new",&newdatastruct,PETSC_NULL);CHKERRQ(ierr);
152ce3d78c0SShri Abhyankar   if(newdatastruct){
153ce3d78c0SShri Abhyankar     ierr = MatLUFactorSymbolic_SeqAIJ_newdatastruct(B,A,isrow,iscol,info);CHKERRQ(ierr);
154ce3d78c0SShri Abhyankar     PetscFunctionReturn(0);
155ce3d78c0SShri Abhyankar   }
1569b48462bSShri Abhyankar 
157ce3d78c0SShri Abhyankar 
158d0f46423SBarry Smith   if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_ERR_ARG_WRONG,"matrix must be square");
1594c49b128SBarry Smith   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
160ac121b3dSBarry Smith   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
161ac121b3dSBarry Smith   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
162289bc588SBarry Smith 
163289bc588SBarry Smith   /* get new row pointers */
1641393bc97SHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
1651393bc97SHong Zhang   bi[0] = 0;
1661393bc97SHong Zhang 
1671393bc97SHong Zhang   /* bdiag is location of diagonal in factor */
1681393bc97SHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
1691393bc97SHong Zhang   bdiag[0] = 0;
1701393bc97SHong Zhang 
1711393bc97SHong Zhang   /* linked list for storing column indices of the active row */
1721393bc97SHong Zhang   nlnk = n + 1;
1731393bc97SHong Zhang   ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
1741393bc97SHong Zhang 
17535baf27eSSatish Balay   ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr);
1761393bc97SHong Zhang 
1771393bc97SHong Zhang   /* initial FreeSpace size is f*(ai[n]+1) */
178d3d32019SBarry Smith   f = info->fill;
179a1a86e44SBarry Smith   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr);
1801393bc97SHong Zhang   current_space = free_space;
181289bc588SBarry Smith 
182289bc588SBarry Smith   for (i=0; i<n; i++) {
1831393bc97SHong Zhang     /* copy previous fill into linked list */
184289bc588SBarry Smith     nzi = 0;
1851393bc97SHong Zhang     nnz = ai[r[i]+1] - ai[r[i]];
1863b4a8b6dSBarry 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);
1871393bc97SHong Zhang     ajtmp = aj + ai[r[i]];
188afcc9904SHong Zhang     ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
1891393bc97SHong Zhang     nzi += nlnk;
1901393bc97SHong Zhang 
1911393bc97SHong Zhang     /* add pivot rows into linked list */
1921393bc97SHong Zhang     row = lnk[n];
1931393bc97SHong Zhang     while (row < i) {
194d1170560SHong Zhang       nzbd    = bdiag[row] - bi[row] + 1; /* num of entries in the row with column index <= row */
195d1170560SHong Zhang       ajtmp   = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */
196d1170560SHong Zhang       ierr = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr);
1971393bc97SHong Zhang       nzi += nlnk;
1981393bc97SHong Zhang       row  = lnk[row];
199289bc588SBarry Smith     }
2001393bc97SHong Zhang     bi[i+1] = bi[i] + nzi;
2011393bc97SHong Zhang     im[i]   = nzi;
2021393bc97SHong Zhang 
2031393bc97SHong Zhang     /* mark bdiag */
2041393bc97SHong Zhang     nzbd = 0;
2051393bc97SHong Zhang     nnz  = nzi;
2061393bc97SHong Zhang     k    = lnk[n];
2071393bc97SHong Zhang     while (nnz-- && k < i){
2081393bc97SHong Zhang       nzbd++;
2091393bc97SHong Zhang       k = lnk[k];
2101393bc97SHong Zhang     }
2111393bc97SHong Zhang     bdiag[i] = bi[i] + nzbd;
2121393bc97SHong Zhang 
2131393bc97SHong Zhang     /* if free space is not available, make more free space */
2141393bc97SHong Zhang     if (current_space->local_remaining<nzi) {
2151393bc97SHong Zhang       nnz = (n - i)*nzi; /* estimated and max additional space needed */
216a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
2171393bc97SHong Zhang       reallocs++;
2181393bc97SHong Zhang     }
2191393bc97SHong Zhang 
2201393bc97SHong Zhang     /* copy data into free space, then initialize lnk */
2211393bc97SHong Zhang     ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
2221393bc97SHong Zhang     bi_ptr[i] = current_space->array;
2231393bc97SHong Zhang     current_space->array           += nzi;
2241393bc97SHong Zhang     current_space->local_used      += nzi;
2251393bc97SHong Zhang     current_space->local_remaining -= nzi;
226289bc588SBarry Smith   }
2276cf91177SBarry Smith #if defined(PETSC_USE_INFO)
228464e8d44SSatish Balay   if (ai[n] != 0) {
2291393bc97SHong Zhang     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
230ae15b995SBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr);
231ae15b995SBarry Smith     ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
232ae15b995SBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr);
233ae15b995SBarry Smith     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
23405bf559cSSatish Balay   } else {
235ae15b995SBarry Smith     ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr);
23605bf559cSSatish Balay   }
23763ba0a88SBarry Smith #endif
2382fb3ed76SBarry Smith 
239898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
240898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
2411987afe7SBarry Smith 
2421393bc97SHong Zhang   /* destroy list of free space and other temporary array(s) */
2431393bc97SHong Zhang   ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
244a1a86e44SBarry Smith   ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr);
2451393bc97SHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
24635baf27eSSatish Balay   ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr);
247289bc588SBarry Smith 
248289bc588SBarry Smith   /* put together the new matrix */
249719d5645SBarry Smith   ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
250719d5645SBarry Smith   ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr);
251719d5645SBarry Smith   b    = (Mat_SeqAIJ*)(B)->data;
252e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
253e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
2547c922b88SBarry Smith   b->singlemalloc = PETSC_FALSE;
2551393bc97SHong Zhang   ierr          = PetscMalloc((bi[n]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);
2561393bc97SHong Zhang   b->j          = bj;
2571393bc97SHong Zhang   b->i          = bi;
2581393bc97SHong Zhang   b->diag       = bdiag;
259416022c9SBarry Smith   b->ilen       = 0;
260416022c9SBarry Smith   b->imax       = 0;
261416022c9SBarry Smith   b->row        = isrow;
262416022c9SBarry Smith   b->col        = iscol;
263c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
264c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
26582bf6240SBarry Smith   b->icol       = isicol;
26687828ca2SBarry Smith   ierr          = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
2671393bc97SHong Zhang 
2681393bc97SHong Zhang   /* In b structure:  Free imax, ilen, old a, old j.  Allocate solve_work, new a, new j */
269719d5645SBarry Smith   ierr = PetscLogObjectMemory(B,(bi[n]-n)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
2701393bc97SHong Zhang   b->maxnz = b->nz = bi[n] ;
2717fd98bd6SLois Curfman McInnes 
272719d5645SBarry Smith   (B)->factor                = MAT_FACTOR_LU;
273719d5645SBarry Smith   (B)->info.factor_mallocs   = reallocs;
274719d5645SBarry Smith   (B)->info.fill_ratio_given = f;
275703deb49SSatish Balay 
2769f7d0b68SHong Zhang   if (ai[n]) {
277719d5645SBarry Smith     (B)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]);
278eea2913aSSatish Balay   } else {
279719d5645SBarry Smith     (B)->info.fill_ratio_needed = 0.0;
280eea2913aSSatish Balay   }
281719d5645SBarry Smith   (B)->ops->lufactornumeric  = MatLUFactorNumeric_SeqAIJ;
282719d5645SBarry Smith   (B)->ops->solve            = MatSolve_SeqAIJ;
283719d5645SBarry Smith   (B)->ops->solvetranspose   = MatSolveTranspose_SeqAIJ;
284db4efbfdSBarry Smith   /* switch to inodes if appropriate */
2854108e4d5SBarry Smith   ierr = MatLUFactorSymbolic_SeqAIJ_Inode(B,A,isrow,iscol,info);CHKERRQ(ierr);
2863a40ed3dSBarry Smith   PetscFunctionReturn(0);
287289bc588SBarry Smith }
2881393bc97SHong Zhang 
289ce3d78c0SShri Abhyankar #undef __FUNCT__
290ce3d78c0SShri Abhyankar #define __FUNCT__ "MatLUFactorSymbolic_SeqAIJ_newdatastruct"
291ce3d78c0SShri Abhyankar PetscErrorCode MatLUFactorSymbolic_SeqAIJ_newdatastruct(Mat B,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
292ce3d78c0SShri Abhyankar {
293ce3d78c0SShri Abhyankar   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
294ce3d78c0SShri Abhyankar   IS                 isicol;
295ce3d78c0SShri Abhyankar   PetscErrorCode     ierr;
296ce3d78c0SShri Abhyankar   const PetscInt     *r,*ic;
297ce3d78c0SShri Abhyankar   PetscInt           i,n=A->rmap->n,*ai=a->i,*aj=a->j;
298ce3d78c0SShri Abhyankar   PetscInt           *bi,*bj,*ajtmp;
299ce3d78c0SShri Abhyankar   PetscInt           *bdiag,row,nnz,nzi,reallocs=0,nzbd,*im;
300ce3d78c0SShri Abhyankar   PetscReal          f;
301ce3d78c0SShri Abhyankar   PetscInt           nlnk,*lnk,k,**bi_ptr;
302ce3d78c0SShri Abhyankar   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
303ce3d78c0SShri Abhyankar   PetscBT            lnkbt;
304ce3d78c0SShri Abhyankar 
305ce3d78c0SShri Abhyankar   PetscFunctionBegin;
306ce3d78c0SShri Abhyankar   if (A->rmap->N != A->cmap->N) SETERRQ(PETSC_ERR_ARG_WRONG,"matrix must be square");
307ce3d78c0SShri Abhyankar   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
308ce3d78c0SShri Abhyankar   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
309ce3d78c0SShri Abhyankar   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
310ce3d78c0SShri Abhyankar 
3111bfa06eaSJed Brown   /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */
3121bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
3131bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
314a6df321fSShri Abhyankar   bi[0] = bdiag[0] = 0;
3159b48462bSShri Abhyankar 
3169b48462bSShri Abhyankar   /* linked list for storing column indices of the active row */
3179b48462bSShri Abhyankar   nlnk = n + 1;
3189b48462bSShri Abhyankar   ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
3199b48462bSShri Abhyankar 
3209b48462bSShri Abhyankar   ierr = PetscMalloc2(n+1,PetscInt**,&bi_ptr,n+1,PetscInt,&im);CHKERRQ(ierr);
3219b48462bSShri Abhyankar 
3229b48462bSShri Abhyankar   /* initial FreeSpace size is f*(ai[n]+1) */
3239b48462bSShri Abhyankar   f = info->fill;
3249b48462bSShri Abhyankar   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr);
3259b48462bSShri Abhyankar   current_space = free_space;
3269b48462bSShri Abhyankar 
3279b48462bSShri Abhyankar   for (i=0; i<n; i++) {
3289b48462bSShri Abhyankar     /* copy previous fill into linked list */
3299b48462bSShri Abhyankar     nzi = 0;
3309b48462bSShri Abhyankar     nnz = ai[r[i]+1] - ai[r[i]];
3319b48462bSShri 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);
3329b48462bSShri Abhyankar     ajtmp = aj + ai[r[i]];
3339b48462bSShri Abhyankar     ierr = PetscLLAddPerm(nnz,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
3349b48462bSShri Abhyankar     nzi += nlnk;
3359b48462bSShri Abhyankar 
3369b48462bSShri Abhyankar     /* add pivot rows into linked list */
3379b48462bSShri Abhyankar     row = lnk[n];
3389b48462bSShri Abhyankar     while (row < i){
3399b48462bSShri Abhyankar       nzbd  = bdiag[row] + 1; /* num of entries in the row with column index <= row */
3409b48462bSShri Abhyankar       ajtmp = bi_ptr[row] + nzbd; /* points to the entry next to the diagonal */
3419b48462bSShri Abhyankar       ierr  = PetscLLAddSortedLU(ajtmp,row,nlnk,lnk,lnkbt,i,nzbd,im);CHKERRQ(ierr);
3429b48462bSShri Abhyankar       nzi  += nlnk;
3439b48462bSShri Abhyankar       row   = lnk[row];
3449b48462bSShri Abhyankar     }
3459b48462bSShri Abhyankar     bi[i+1] = bi[i] + nzi;
3469b48462bSShri Abhyankar     im[i]   = nzi;
3479b48462bSShri Abhyankar 
3489b48462bSShri Abhyankar     /* mark bdiag */
3499b48462bSShri Abhyankar     nzbd = 0;
3509b48462bSShri Abhyankar     nnz  = nzi;
3519b48462bSShri Abhyankar     k    = lnk[n];
3529b48462bSShri Abhyankar     while (nnz-- && k < i){
3539b48462bSShri Abhyankar       nzbd++;
3549b48462bSShri Abhyankar       k = lnk[k];
3559b48462bSShri Abhyankar     }
3569b48462bSShri Abhyankar     bdiag[i] = nzbd; /* note: bdiag[i] = nnzL as input for PetscFreeSpaceContiguous_LU() */
3579b48462bSShri Abhyankar 
3589b48462bSShri Abhyankar     /* if free space is not available, make more free space */
3599b48462bSShri Abhyankar     if (current_space->local_remaining<nzi) {
3609b48462bSShri Abhyankar       nnz = 2*(n - i)*nzi; /* estimated and max additional space needed */
3619b48462bSShri Abhyankar       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
3629b48462bSShri Abhyankar       reallocs++;
3639b48462bSShri Abhyankar     }
3649b48462bSShri Abhyankar 
3659b48462bSShri Abhyankar     /* copy data into free space, then initialize lnk */
3669b48462bSShri Abhyankar     ierr = PetscLLClean(n,n,nzi,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
3679b48462bSShri Abhyankar     bi_ptr[i] = current_space->array;
3689b48462bSShri Abhyankar     current_space->array           += nzi;
3699b48462bSShri Abhyankar     current_space->local_used      += nzi;
3709b48462bSShri Abhyankar     current_space->local_remaining -= nzi;
3719b48462bSShri Abhyankar   }
3729b48462bSShri Abhyankar #if defined(PETSC_USE_INFO)
3739b48462bSShri Abhyankar   if (ai[n] != 0) {
3749b48462bSShri Abhyankar     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
3759b48462bSShri Abhyankar     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr);
3769b48462bSShri Abhyankar     ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
3779b48462bSShri Abhyankar     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G);\n",af);CHKERRQ(ierr);
3789b48462bSShri Abhyankar     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
3799b48462bSShri Abhyankar   } else {
3809b48462bSShri Abhyankar     ierr = PetscInfo(A,"Empty matrix\n");CHKERRQ(ierr);
3819b48462bSShri Abhyankar   }
3829b48462bSShri Abhyankar #endif
3839b48462bSShri Abhyankar 
3849b48462bSShri Abhyankar   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
3859b48462bSShri Abhyankar   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
3869b48462bSShri Abhyankar 
3879b48462bSShri Abhyankar   /* destroy list of free space and other temporary array(s) */
3889b48462bSShri Abhyankar   ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
3899b48462bSShri Abhyankar   ierr = PetscFreeSpaceContiguous_LU_v2(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr);
3909b48462bSShri Abhyankar   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
3919b48462bSShri Abhyankar   ierr = PetscFree2(bi_ptr,im);CHKERRQ(ierr);
3929b48462bSShri Abhyankar 
3939b48462bSShri Abhyankar   /* put together the new matrix */
3949b48462bSShri Abhyankar   ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
3959b48462bSShri Abhyankar   ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr);
3969b48462bSShri Abhyankar   b    = (Mat_SeqAIJ*)(B)->data;
3979b48462bSShri Abhyankar   b->free_a       = PETSC_TRUE;
3989b48462bSShri Abhyankar   b->free_ij      = PETSC_TRUE;
3999b48462bSShri Abhyankar   b->singlemalloc = PETSC_FALSE;
4009b48462bSShri Abhyankar   ierr = PetscMalloc((bdiag[0]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);
4019b48462bSShri Abhyankar   b->j          = bj;
4029b48462bSShri Abhyankar   b->i          = bi;
4039b48462bSShri Abhyankar   b->diag       = bdiag;
4049b48462bSShri Abhyankar   b->ilen       = 0;
4059b48462bSShri Abhyankar   b->imax       = 0;
4069b48462bSShri Abhyankar   b->row        = isrow;
4079b48462bSShri Abhyankar   b->col        = iscol;
4089b48462bSShri Abhyankar   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
4099b48462bSShri Abhyankar   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
4109b48462bSShri Abhyankar   b->icol       = isicol;
4119b48462bSShri Abhyankar   ierr          = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
4129b48462bSShri Abhyankar 
4139b48462bSShri Abhyankar   /* In b structure:  Free imax, ilen, old a, old j.  Allocate solve_work, new a, new j */
4149b48462bSShri Abhyankar   ierr = PetscLogObjectMemory(B,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
4159b48462bSShri Abhyankar   b->maxnz = b->nz = bdiag[0]+1;
4169b48462bSShri Abhyankar   (B)->factor                = MAT_FACTOR_LU;
4179b48462bSShri Abhyankar   (B)->info.factor_mallocs   = reallocs;
4189b48462bSShri Abhyankar   (B)->info.fill_ratio_given = f;
4199b48462bSShri Abhyankar 
4209b48462bSShri Abhyankar   if (ai[n]) {
4219b48462bSShri Abhyankar     (B)->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]);
4229b48462bSShri Abhyankar   } else {
4239b48462bSShri Abhyankar     (B)->info.fill_ratio_needed = 0.0;
4249b48462bSShri Abhyankar   }
425b4f5729fSShri Abhyankar   (B)->ops->lufactornumeric  = MatLUFactorNumeric_SeqAIJ_newdatastruct;
4269b48462bSShri Abhyankar   PetscFunctionReturn(0);
4279b48462bSShri Abhyankar }
4289b48462bSShri Abhyankar 
4295b5bc046SBarry Smith /*
4305b5bc046SBarry Smith     Trouble in factorization, should we dump the original matrix?
4315b5bc046SBarry Smith */
4325b5bc046SBarry Smith #undef __FUNCT__
4335b5bc046SBarry Smith #define __FUNCT__ "MatFactorDumpMatrix"
4345b5bc046SBarry Smith PetscErrorCode MatFactorDumpMatrix(Mat A)
4355b5bc046SBarry Smith {
4365b5bc046SBarry Smith   PetscErrorCode ierr;
43790d69ab7SBarry Smith   PetscTruth     flg = PETSC_FALSE;
4385b5bc046SBarry Smith 
4395b5bc046SBarry Smith   PetscFunctionBegin;
44090d69ab7SBarry Smith   ierr = PetscOptionsGetTruth(PETSC_NULL,"-mat_factor_dump_on_error",&flg,PETSC_NULL);CHKERRQ(ierr);
4415b5bc046SBarry Smith   if (flg) {
4425b5bc046SBarry Smith     PetscViewer viewer;
4435b5bc046SBarry Smith     char        filename[PETSC_MAX_PATH_LEN];
4445b5bc046SBarry Smith 
4455b5bc046SBarry Smith     ierr = PetscSNPrintf(filename,PETSC_MAX_PATH_LEN,"matrix_factor_error.%d",PetscGlobalRank);CHKERRQ(ierr);
4467adad957SLisandro Dalcin     ierr = PetscViewerBinaryOpen(((PetscObject)A)->comm,filename,FILE_MODE_WRITE,&viewer);CHKERRQ(ierr);
4475b5bc046SBarry Smith     ierr = MatView(A,viewer);CHKERRQ(ierr);
4485b5bc046SBarry Smith     ierr = PetscViewerDestroy(viewer);CHKERRQ(ierr);
4495b5bc046SBarry Smith   }
4505b5bc046SBarry Smith   PetscFunctionReturn(0);
4515b5bc046SBarry Smith }
4525b5bc046SBarry Smith 
4534108e4d5SBarry Smith extern PetscErrorCode MatSolve_SeqAIJ_Inode(Mat,Vec,Vec);
454db4efbfdSBarry Smith 
455c9c72f8fSHong Zhang #undef __FUNCT__
456c9c72f8fSHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_newdatastruct"
457c9c72f8fSHong Zhang PetscErrorCode MatLUFactorNumeric_SeqAIJ_newdatastruct(Mat B,Mat A,const MatFactorInfo *info)
458c9c72f8fSHong Zhang {
459c9c72f8fSHong Zhang   Mat            C=B;
460c9c72f8fSHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data;
461c9c72f8fSHong Zhang   IS             isrow = b->row,isicol = b->icol;
462c9c72f8fSHong Zhang   PetscErrorCode ierr;
463c9c72f8fSHong Zhang   const PetscInt *r,*ic,*ics;
464c9c72f8fSHong Zhang   PetscInt       i,j,k,n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j;
465c9c72f8fSHong Zhang   PetscInt       *ajtmp,*bjtmp,nz,nzL,row,*bdiag=b->diag,*pj;
466c9c72f8fSHong Zhang   MatScalar      *rtmp,*pc,multiplier,*v,*pv,*aa=a->a;
467c9c72f8fSHong Zhang   PetscTruth     row_identity,col_identity;
468c9c72f8fSHong Zhang 
469d4ad06f7SHong Zhang   LUShift_Ctx    sctx;
470d4ad06f7SHong Zhang   PetscInt       *ddiag,newshift;
471d4ad06f7SHong Zhang   PetscReal      rs;
472d4ad06f7SHong Zhang   MatScalar      d;
473d4ad06f7SHong Zhang 
474c9c72f8fSHong Zhang   PetscFunctionBegin;
475d6e5152cSHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
476d6e5152cSHong Zhang   ierr = PetscMemzero(&sctx,sizeof(LUShift_Ctx));CHKERRQ(ierr);
477d4ad06f7SHong Zhang 
478d4ad06f7SHong Zhang   /* if both shift schemes are chosen by user, only use info->shiftpd */
479d4ad06f7SHong Zhang   if (info->shiftpd) { /* set sctx.shift_top=max{rs} */
480d4ad06f7SHong Zhang     ddiag          = a->diag;
481d4ad06f7SHong Zhang     sctx.shift_top = info->zeropivot;
482d4ad06f7SHong Zhang     for (i=0; i<n; i++) {
483d4ad06f7SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
484d4ad06f7SHong Zhang       d  = (aa)[ddiag[i]];
485d4ad06f7SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
486d4ad06f7SHong Zhang       v  = aa+ai[i];
487d4ad06f7SHong Zhang       nz = ai[i+1] - ai[i];
488d4ad06f7SHong Zhang       for (j=0; j<nz; j++)
489d4ad06f7SHong Zhang 	rs += PetscAbsScalar(v[j]);
490d4ad06f7SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
491d4ad06f7SHong Zhang     }
492d4ad06f7SHong Zhang     sctx.shift_top   *= 1.1;
493d4ad06f7SHong Zhang     sctx.nshift_max   = 5;
494d4ad06f7SHong Zhang     sctx.shift_lo     = 0.;
495d4ad06f7SHong Zhang     sctx.shift_hi     = 1.;
496d4ad06f7SHong Zhang   }
497d4ad06f7SHong Zhang 
498c9c72f8fSHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
499c9c72f8fSHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
500c9c72f8fSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr);
501c9c72f8fSHong Zhang   ics  = ic;
502c9c72f8fSHong Zhang 
503d4ad06f7SHong Zhang   do {
504d4ad06f7SHong Zhang     sctx.lushift = PETSC_FALSE;
505c9c72f8fSHong Zhang     for (i=0; i<n; i++){
506c9c72f8fSHong Zhang       /* zero rtmp */
507c9c72f8fSHong Zhang       /* L part */
508c9c72f8fSHong Zhang       nz    = bi[i+1] - bi[i];
509c9c72f8fSHong Zhang       bjtmp = bj + bi[i];
510c9c72f8fSHong Zhang       for  (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
511c9c72f8fSHong Zhang 
512c9c72f8fSHong Zhang       /* U part */
51362fbd6afSShri Abhyankar       nz = bdiag[i]-bdiag[i+1];
51462fbd6afSShri Abhyankar       bjtmp = bj + bdiag[i+1]+1;
515813a1f2bSShri Abhyankar       for  (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
516813a1f2bSShri Abhyankar 
517813a1f2bSShri Abhyankar       /* load in initial (unfactored row) */
518813a1f2bSShri Abhyankar       nz    = ai[r[i]+1] - ai[r[i]];
519813a1f2bSShri Abhyankar       ajtmp = aj + ai[r[i]];
520813a1f2bSShri Abhyankar       v     = aa + ai[r[i]];
521813a1f2bSShri Abhyankar       for (j=0; j<nz; j++) {
522813a1f2bSShri Abhyankar         rtmp[ics[ajtmp[j]]] = v[j];
523813a1f2bSShri Abhyankar       }
524813a1f2bSShri Abhyankar       /* ZeropivotApply() */
52598a93161SHong Zhang       rtmp[i] += sctx.shift_amount;  /* shift the diagonal of the matrix */
526813a1f2bSShri Abhyankar 
527813a1f2bSShri Abhyankar       /* elimination */
528813a1f2bSShri Abhyankar       bjtmp = bj + bi[i];
529813a1f2bSShri Abhyankar       row   = *bjtmp++;
530813a1f2bSShri Abhyankar       nzL   = bi[i+1] - bi[i];
531f268cba8SShri Abhyankar       for(k=0; k < nzL;k++) {
532813a1f2bSShri Abhyankar         pc = rtmp + row;
533813a1f2bSShri Abhyankar         if (*pc != 0.0) {
534813a1f2bSShri Abhyankar           pv         = b->a + bdiag[row];
535813a1f2bSShri Abhyankar           multiplier = *pc * (*pv);
536813a1f2bSShri Abhyankar           *pc        = multiplier;
53762fbd6afSShri Abhyankar           pj = b->j + bdiag[row+1]+1; /* beginning of U(row,:) */
53862fbd6afSShri Abhyankar 	  pv = b->a + bdiag[row+1]+1;
53962fbd6afSShri Abhyankar 	  nz = bdiag[row]-bdiag[row+1]-1; /* num of entries in U(row,:) excluding diag */
540813a1f2bSShri Abhyankar           for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j];
541813a1f2bSShri Abhyankar           ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr);
542813a1f2bSShri Abhyankar         }
543f268cba8SShri Abhyankar         row = *bjtmp++;
544813a1f2bSShri Abhyankar       }
545813a1f2bSShri Abhyankar 
546813a1f2bSShri Abhyankar       /* finished row so stick it into b->a */
547813a1f2bSShri Abhyankar       rs = 0.0;
548813a1f2bSShri Abhyankar       /* L part */
549813a1f2bSShri Abhyankar       pv   = b->a + bi[i] ;
550813a1f2bSShri Abhyankar       pj   = b->j + bi[i] ;
551813a1f2bSShri Abhyankar       nz   = bi[i+1] - bi[i];
552813a1f2bSShri Abhyankar       for (j=0; j<nz; j++) {
553813a1f2bSShri Abhyankar         pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]);
554813a1f2bSShri Abhyankar       }
555813a1f2bSShri Abhyankar 
556813a1f2bSShri Abhyankar       /* U part */
55762fbd6afSShri Abhyankar       pv = b->a + bdiag[i+1]+1;
55862fbd6afSShri Abhyankar       pj = b->j + bdiag[i+1]+1;
55962fbd6afSShri Abhyankar       nz = bdiag[i] - bdiag[i+1]-1;
560813a1f2bSShri Abhyankar       for (j=0; j<nz; j++) {
561813a1f2bSShri Abhyankar         pv[j] = rtmp[pj[j]]; rs += PetscAbsScalar(pv[j]);
562813a1f2bSShri Abhyankar       }
563813a1f2bSShri Abhyankar 
5648ff23777SHong Zhang       /* MatPivotCheck() */
565813a1f2bSShri Abhyankar       sctx.rs  = rs;
566813a1f2bSShri Abhyankar       sctx.pv  = rtmp[i];
5678ff23777SHong Zhang       /* ierr = MatLUCheckShift_inline(info,sctx,i,newshift);CHKERRQ(ierr); */
5688ff23777SHong Zhang       if (info->shiftnz){
5698ff23777SHong Zhang         ierr = MatPivotCheck_nz(info,sctx,i,newshift);CHKERRQ(ierr);
5708ff23777SHong Zhang       } else if (info->shiftpd){
5718ff23777SHong Zhang         ierr = MatPivotCheck_pd(info,sctx,i,newshift);CHKERRQ(ierr);
5728ff23777SHong Zhang       } else if (info->shiftinblocks){
5738ff23777SHong Zhang         ierr = MatPivotCheck_inblocks(info,sctx,i,newshift);CHKERRQ(ierr);
5748ff23777SHong Zhang       } else {
5758ff23777SHong Zhang         ierr = MatPivotCheck_none(info,sctx,i,newshift);CHKERRQ(ierr);
5768ff23777SHong Zhang       }
5778ff23777SHong Zhang       rtmp[i] = sctx.pv;
578813a1f2bSShri Abhyankar       if (newshift == 1) break;
579813a1f2bSShri Abhyankar 
580813a1f2bSShri Abhyankar       /* Mark diagonal and invert diagonal for simplier triangular solves */
581813a1f2bSShri Abhyankar       pv  = b->a + bdiag[i];
582813a1f2bSShri Abhyankar       *pv = 1.0/rtmp[i];
583813a1f2bSShri Abhyankar 
584813a1f2bSShri Abhyankar     } /* endof for (i=0; i<n; i++){ */
585813a1f2bSShri Abhyankar 
5868ff23777SHong Zhang     /* MatPivotRefine() */
587813a1f2bSShri Abhyankar     if (info->shiftpd && !sctx.lushift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max){
588813a1f2bSShri Abhyankar       /*
589813a1f2bSShri Abhyankar        * if no shift in this attempt & shifting & started shifting & can refine,
590813a1f2bSShri Abhyankar        * then try lower shift
591813a1f2bSShri Abhyankar        */
592813a1f2bSShri Abhyankar       sctx.shift_hi       = sctx.shift_fraction;
593813a1f2bSShri Abhyankar       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
594813a1f2bSShri Abhyankar       sctx.shift_amount   = sctx.shift_fraction * sctx.shift_top;
595813a1f2bSShri Abhyankar       sctx.lushift        = PETSC_TRUE;
596813a1f2bSShri Abhyankar       sctx.nshift++;
597813a1f2bSShri Abhyankar     }
598813a1f2bSShri Abhyankar   } while (sctx.lushift);
599813a1f2bSShri Abhyankar 
600813a1f2bSShri Abhyankar   ierr = PetscFree(rtmp);CHKERRQ(ierr);
601813a1f2bSShri Abhyankar   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
602813a1f2bSShri Abhyankar   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
603813a1f2bSShri Abhyankar 
604813a1f2bSShri Abhyankar   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
605813a1f2bSShri Abhyankar   ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr);
606813a1f2bSShri Abhyankar   if (row_identity && col_identity) {
6074395751fSShri Abhyankar     C->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_newdatastruct;
608813a1f2bSShri Abhyankar   } else {
6094395751fSShri Abhyankar     C->ops->solve = MatSolve_SeqAIJ_newdatastruct;
610813a1f2bSShri Abhyankar   }
611813a1f2bSShri Abhyankar 
6123c0119dfSShri Abhyankar   C->ops->solveadd           = MatSolveAdd_SeqAIJ_newdatastruct;
6134395751fSShri Abhyankar   C->ops->solvetranspose     = MatSolveTranspose_SeqAIJ_newdatastruct;
614533e6011SShri Abhyankar   C->ops->solvetransposeadd  = MatSolveTransposeAdd_SeqAIJ_newdatastruct;
615813a1f2bSShri Abhyankar   C->ops->matsolve           = 0;
616813a1f2bSShri Abhyankar   C->assembled    = PETSC_TRUE;
617813a1f2bSShri Abhyankar   C->preallocated = PETSC_TRUE;
618813a1f2bSShri Abhyankar   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
619813a1f2bSShri Abhyankar 
6208ff23777SHong Zhang   /* MatPivotView() */
621813a1f2bSShri Abhyankar   if (sctx.nshift){
622813a1f2bSShri Abhyankar     if (info->shiftpd) {
623813a1f2bSShri Abhyankar       ierr = PetscInfo4(A,"number of shift_pd tries %D, shift_amount %G, diagonal shifted up by %e fraction top_value %e\n",sctx.nshift,sctx.shift_amount,sctx.shift_fraction,sctx.shift_top);CHKERRQ(ierr);
624813a1f2bSShri Abhyankar     } else if (info->shiftnz) {
625813a1f2bSShri Abhyankar       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
6268ff23777SHong Zhang     } else if (info->shiftinblocks){
6278ff23777SHong Zhang       ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %G\n",sctx.nshift,info->shiftinblocks);CHKERRQ(ierr);
628813a1f2bSShri Abhyankar     }
629813a1f2bSShri Abhyankar   }
630813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
631813a1f2bSShri Abhyankar }
632813a1f2bSShri Abhyankar 
6334a2ae208SSatish Balay #undef __FUNCT__
6344a2ae208SSatish Balay #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ"
6350481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info)
636289bc588SBarry Smith {
637719d5645SBarry Smith   Mat             C=B;
638416022c9SBarry Smith   Mat_SeqAIJ      *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data;
63982bf6240SBarry Smith   IS              isrow = b->row,isicol = b->icol;
6406849ba73SBarry Smith   PetscErrorCode  ierr;
6415d0c19d7SBarry Smith   const PetscInt   *r,*ic,*ics;
642d278edc7SBarry Smith   PetscInt        nz,row,i,j,n=A->rmap->n,diag;
643d278edc7SBarry Smith   const PetscInt  *ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j;
644d278edc7SBarry Smith   const PetscInt  *ajtmp,*bjtmp,*diag_offset = b->diag,*pj;
645d278edc7SBarry Smith   MatScalar       *pv,*rtmp,*pc,multiplier,d;
646d278edc7SBarry Smith   const MatScalar *v,*aa=a->a;
64733f9893dSHong Zhang   PetscReal       rs=0.0;
648b3bf805bSHong Zhang   LUShift_Ctx     sctx;
64942898933SHong Zhang   PetscInt        newshift,*ddiag;
650289bc588SBarry Smith 
6513a40ed3dSBarry Smith   PetscFunctionBegin;
65278b31e54SBarry Smith   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
65378b31e54SBarry Smith   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
654b78a477dSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr);
6559f7d0b68SHong Zhang   ics  = ic;
656289bc588SBarry Smith 
657d4ad06f7SHong Zhang   /* initialize shift context sctx */
658d4ad06f7SHong Zhang   sctx.nshift         = 0;
659ada7143aSSatish Balay   sctx.nshift_max     = 0;
660d4ad06f7SHong Zhang   sctx.shift_top      = 0.0;
661d4ad06f7SHong Zhang   sctx.shift_lo       = 0.0;
662d4ad06f7SHong Zhang   sctx.shift_hi       = 0.0;
663d4ad06f7SHong Zhang   sctx.shift_fraction = 0.0;
664d4ad06f7SHong Zhang   sctx.shift_amount   = 0.0;
665ada7143aSSatish Balay 
666118f5789SBarry Smith   /* if both shift schemes are chosen by user, only use info->shiftpd */
6671a890ab1SHong Zhang   if (info->shiftpd) { /* set sctx.shift_top=max{rs} */
66842898933SHong Zhang     ddiag          = a->diag;
6699f7d0b68SHong Zhang     sctx.shift_top = info->zeropivot;
6706cc28720Svictorle     for (i=0; i<n; i++) {
6719f95998fSHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
6729f7d0b68SHong Zhang       d  = (aa)[ddiag[i]];
6731a890ab1SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
6749f7d0b68SHong Zhang       v  = aa+ai[i];
6759f7d0b68SHong Zhang       nz = ai[i+1] - ai[i];
676e24cfe64SHong Zhang       for (j=0; j<nz; j++)
6771a890ab1SHong Zhang 	rs += PetscAbsScalar(v[j]);
6781a890ab1SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
6796cc28720Svictorle     }
680118f5789SBarry Smith     sctx.shift_top   *= 1.1;
681d2276718SHong Zhang     sctx.nshift_max   = 5;
682d2276718SHong Zhang     sctx.shift_lo     = 0.;
683d2276718SHong Zhang     sctx.shift_hi     = 1.;
684a0a356efSHong Zhang   }
685a0a356efSHong Zhang 
686085256b3SBarry Smith   do {
687d2276718SHong Zhang     sctx.lushift = PETSC_FALSE;
688289bc588SBarry Smith     for (i=0; i<n; i++){
689b3bf805bSHong Zhang       nz    = bi[i+1] - bi[i];
690b3bf805bSHong Zhang       bjtmp = bj + bi[i];
6919f7d0b68SHong Zhang       for  (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
692289bc588SBarry Smith 
693289bc588SBarry Smith       /* load in initial (unfactored row) */
6949f7d0b68SHong Zhang       nz    = ai[r[i]+1] - ai[r[i]];
6959f7d0b68SHong Zhang       ajtmp = aj + ai[r[i]];
6969f7d0b68SHong Zhang       v     = aa + ai[r[i]];
697085256b3SBarry Smith       for (j=0; j<nz; j++) {
698b3bf805bSHong Zhang         rtmp[ics[ajtmp[j]]] = v[j];
699085256b3SBarry Smith       }
700d2276718SHong Zhang       rtmp[ics[r[i]]] += sctx.shift_amount; /* shift the diagonal of the matrix */
701bd1bc851SBarry Smith       /* if (sctx.shift_amount > 0.0) printf("row %d, shift %g\n",i,sctx.shift_amount); */
702289bc588SBarry Smith 
703b3bf805bSHong Zhang       row = *bjtmp++;
704f2582111SSatish Balay       while  (row < i) {
7058c37ef55SBarry Smith         pc = rtmp + row;
7068c37ef55SBarry Smith         if (*pc != 0.0) {
707010a20caSHong Zhang           pv         = b->a + diag_offset[row];
708010a20caSHong Zhang           pj         = b->j + diag_offset[row] + 1;
70935aab85fSBarry Smith           multiplier = *pc / *pv++;
7108c37ef55SBarry Smith           *pc        = multiplier;
711b3bf805bSHong Zhang           nz         = bi[row+1] - diag_offset[row] - 1;
7129f7d0b68SHong Zhang           for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j];
713dc0b31edSSatish Balay           ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr);
714289bc588SBarry Smith         }
715b3bf805bSHong Zhang         row = *bjtmp++;
716289bc588SBarry Smith       }
717416022c9SBarry Smith       /* finished row so stick it into b->a */
718b3bf805bSHong Zhang       pv   = b->a + bi[i] ;
719b3bf805bSHong Zhang       pj   = b->j + bi[i] ;
720b3bf805bSHong Zhang       nz   = bi[i+1] - bi[i];
721b3bf805bSHong Zhang       diag = diag_offset[i] - bi[i];
722e57ff13aSBarry Smith       rs   = 0.0;
723d3d32019SBarry Smith       for (j=0; j<nz; j++) {
7249f7d0b68SHong Zhang         pv[j] = rtmp[pj[j]];
7259f7d0b68SHong Zhang         rs   += PetscAbsScalar(pv[j]);
726d3d32019SBarry Smith       }
727e57ff13aSBarry Smith       rs   -= PetscAbsScalar(pv[diag]);
728d2276718SHong Zhang 
729b3bf805bSHong Zhang       /* 9/13/02 Victor Eijkhout suggested scaling zeropivot by rs for matrices with funny scalings */
730d2276718SHong Zhang       sctx.rs  = rs;
731d2276718SHong Zhang       sctx.pv  = pv[diag];
732c6b1f410SHong Zhang       ierr = MatLUCheckShift_inline(info,sctx,i,newshift);CHKERRQ(ierr);
7335b5bc046SBarry Smith       if (newshift == 1) break;
73435aab85fSBarry Smith     }
735d2276718SHong Zhang 
7360481f469SBarry Smith     if (info->shiftpd && !sctx.lushift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) {
7376cc28720Svictorle       /*
7386c037d1bSvictorle        * if no shift in this attempt & shifting & started shifting & can refine,
7396cc28720Svictorle        * then try lower shift
7406cc28720Svictorle        */
7410481f469SBarry Smith       sctx.shift_hi       = sctx.shift_fraction;
7420481f469SBarry Smith       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
7430481f469SBarry Smith       sctx.shift_amount   = sctx.shift_fraction * sctx.shift_top;
744d2276718SHong Zhang       sctx.lushift        = PETSC_TRUE;
745d2276718SHong Zhang       sctx.nshift++;
7466cc28720Svictorle     }
747d2276718SHong Zhang   } while (sctx.lushift);
7480f11f9aeSSatish Balay 
74935aab85fSBarry Smith   /* invert diagonal entries for simplier triangular solves */
75035aab85fSBarry Smith   for (i=0; i<n; i++) {
751010a20caSHong Zhang     b->a[diag_offset[i]] = 1.0/b->a[diag_offset[i]];
75235aab85fSBarry Smith   }
753606d414cSSatish Balay   ierr = PetscFree(rtmp);CHKERRQ(ierr);
75478b31e54SBarry Smith   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
75578b31e54SBarry Smith   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
756db4efbfdSBarry Smith   if (b->inode.use) {
7574108e4d5SBarry Smith     C->ops->solve   = MatSolve_SeqAIJ_Inode;
758db4efbfdSBarry Smith   } else {
7598d8c7c43SBarry Smith     PetscTruth row_identity, col_identity;
7608d8c7c43SBarry Smith     ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
7618d8c7c43SBarry Smith     ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr);
7628d8c7c43SBarry Smith     if (row_identity && col_identity) {
7638d8c7c43SBarry Smith       C->ops->solve   = MatSolve_SeqAIJ_NaturalOrdering;
7648d8c7c43SBarry Smith     } else {
765db4efbfdSBarry Smith       C->ops->solve   = MatSolve_SeqAIJ;
766db4efbfdSBarry Smith     }
7678d8c7c43SBarry Smith   }
768db4efbfdSBarry Smith   C->ops->solveadd           = MatSolveAdd_SeqAIJ;
769db4efbfdSBarry Smith   C->ops->solvetranspose     = MatSolveTranspose_SeqAIJ;
770db4efbfdSBarry Smith   C->ops->solvetransposeadd  = MatSolveTransposeAdd_SeqAIJ;
771de26497bSBarry Smith   C->ops->matsolve           = MatMatSolve_SeqAIJ;
772c456f294SBarry Smith   C->assembled    = PETSC_TRUE;
7735c9eb25fSBarry Smith   C->preallocated = PETSC_TRUE;
774d0f46423SBarry Smith   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
775d2276718SHong Zhang   if (sctx.nshift){
776e738e422SBarry Smith      if (info->shiftpd) {
7770481f469SBarry 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);
778e738e422SBarry Smith     } else if (info->shiftnz) {
779e738e422SBarry Smith       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
7806cc28720Svictorle     }
7810697b6caSHong Zhang   }
7823a40ed3dSBarry Smith   PetscFunctionReturn(0);
783289bc588SBarry Smith }
7840889b5dcSvictorle 
785137fb511SHong Zhang /*
786137fb511SHong Zhang    This routine implements inplace ILU(0) with row or/and column permutations.
787137fb511SHong Zhang    Input:
788137fb511SHong Zhang      A - original matrix
789137fb511SHong Zhang    Output;
790137fb511SHong Zhang      A - a->i (rowptr) is same as original rowptr, but factored i-the row is stored in rowperm[i]
791137fb511SHong Zhang          a->j (col index) is permuted by the inverse of colperm, then sorted
792137fb511SHong Zhang          a->a reordered accordingly with a->j
793137fb511SHong Zhang          a->diag (ptr to diagonal elements) is updated.
794137fb511SHong Zhang */
795137fb511SHong Zhang #undef __FUNCT__
796137fb511SHong Zhang #define __FUNCT__ "MatLUFactorNumeric_SeqAIJ_InplaceWithPerm"
7970481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_SeqAIJ_InplaceWithPerm(Mat B,Mat A,const MatFactorInfo *info)
798137fb511SHong Zhang {
799b051339dSHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
800b051339dSHong Zhang   IS             isrow = a->row,isicol = a->icol;
801137fb511SHong Zhang   PetscErrorCode ierr;
8025d0c19d7SBarry Smith   const PetscInt *r,*ic,*ics;
8035d0c19d7SBarry Smith   PetscInt       i,j,n=A->rmap->n,*ai=a->i,*aj=a->j;
8045d0c19d7SBarry Smith   PetscInt       *ajtmp,nz,row;
805b051339dSHong Zhang   PetscInt       *diag = a->diag,nbdiag,*pj;
806a77337e4SBarry Smith   PetscScalar    *rtmp,*pc,multiplier,d;
807a77337e4SBarry Smith   MatScalar      *v,*pv;
808137fb511SHong Zhang   PetscReal      rs;
809137fb511SHong Zhang   LUShift_Ctx    sctx;
810b051339dSHong Zhang   PetscInt       newshift;
811137fb511SHong Zhang 
812137fb511SHong Zhang   PetscFunctionBegin;
813719d5645SBarry Smith   if (A != B) SETERRQ(PETSC_ERR_ARG_INCOMP,"input and output matrix must have same address");
814137fb511SHong Zhang   ierr  = ISGetIndices(isrow,&r);CHKERRQ(ierr);
815137fb511SHong Zhang   ierr  = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
816137fb511SHong Zhang   ierr  = PetscMalloc((n+1)*sizeof(PetscScalar),&rtmp);CHKERRQ(ierr);
817137fb511SHong Zhang   ierr  = PetscMemzero(rtmp,(n+1)*sizeof(PetscScalar));CHKERRQ(ierr);
818b051339dSHong Zhang   ics = ic;
819137fb511SHong Zhang 
82075567043SBarry Smith   sctx.shift_top      = 0.;
821*e60cf9a0SBarry Smith   sctx.nshift_max     = 0;
82275567043SBarry Smith   sctx.shift_lo       = 0.;
82375567043SBarry Smith   sctx.shift_hi       = 0.;
82475567043SBarry Smith   sctx.shift_fraction = 0.;
825137fb511SHong Zhang 
826137fb511SHong Zhang   /* if both shift schemes are chosen by user, only use info->shiftpd */
827137fb511SHong Zhang   if (info->shiftpd) { /* set sctx.shift_top=max{rs} */
82875567043SBarry Smith     sctx.shift_top = 0.;
829137fb511SHong Zhang     for (i=0; i<n; i++) {
830137fb511SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
831b051339dSHong Zhang       d  = (a->a)[diag[i]];
832137fb511SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
833b051339dSHong Zhang       v  = a->a+ai[i];
834b051339dSHong Zhang       nz = ai[i+1] - ai[i];
835137fb511SHong Zhang       for (j=0; j<nz; j++)
836137fb511SHong Zhang 	rs += PetscAbsScalar(v[j]);
837137fb511SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
838137fb511SHong Zhang     }
839137fb511SHong Zhang     if (sctx.shift_top < info->zeropivot) sctx.shift_top = info->zeropivot;
840137fb511SHong Zhang     sctx.shift_top    *= 1.1;
841137fb511SHong Zhang     sctx.nshift_max   = 5;
842137fb511SHong Zhang     sctx.shift_lo     = 0.;
843137fb511SHong Zhang     sctx.shift_hi     = 1.;
844137fb511SHong Zhang   }
845137fb511SHong Zhang 
84675567043SBarry Smith   sctx.shift_amount = 0.;
847137fb511SHong Zhang   sctx.nshift       = 0;
848137fb511SHong Zhang   do {
849137fb511SHong Zhang     sctx.lushift = PETSC_FALSE;
850137fb511SHong Zhang     for (i=0; i<n; i++){
851b051339dSHong Zhang       /* load in initial unfactored row */
852b051339dSHong Zhang       nz    = ai[r[i]+1] - ai[r[i]];
853b051339dSHong Zhang       ajtmp = aj + ai[r[i]];
854b051339dSHong Zhang       v     = a->a + ai[r[i]];
855137fb511SHong Zhang       /* sort permuted ajtmp and values v accordingly */
856b051339dSHong Zhang       for (j=0; j<nz; j++) ajtmp[j] = ics[ajtmp[j]];
857137fb511SHong Zhang       ierr = PetscSortIntWithScalarArray(nz,ajtmp,v);CHKERRQ(ierr);
858137fb511SHong Zhang 
859b051339dSHong Zhang       diag[r[i]] = ai[r[i]];
860137fb511SHong Zhang       for (j=0; j<nz; j++) {
861137fb511SHong Zhang         rtmp[ajtmp[j]] = v[j];
862b051339dSHong Zhang         if (ajtmp[j] < i) diag[r[i]]++; /* update a->diag */
863137fb511SHong Zhang       }
864137fb511SHong Zhang       rtmp[r[i]] += sctx.shift_amount; /* shift the diagonal of the matrix */
865137fb511SHong Zhang 
866b051339dSHong Zhang       row = *ajtmp++;
867137fb511SHong Zhang       while  (row < i) {
868137fb511SHong Zhang         pc = rtmp + row;
869137fb511SHong Zhang         if (*pc != 0.0) {
870b051339dSHong Zhang           pv         = a->a + diag[r[row]];
871b051339dSHong Zhang           pj         = aj + diag[r[row]] + 1;
872137fb511SHong Zhang 
873137fb511SHong Zhang           multiplier = *pc / *pv++;
874137fb511SHong Zhang           *pc        = multiplier;
875b051339dSHong Zhang           nz         = ai[r[row]+1] - diag[r[row]] - 1;
876b051339dSHong Zhang           for (j=0; j<nz; j++) rtmp[pj[j]] -= multiplier * pv[j];
877dc0b31edSSatish Balay           ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr);
878137fb511SHong Zhang         }
879b051339dSHong Zhang         row = *ajtmp++;
880137fb511SHong Zhang       }
881b051339dSHong Zhang       /* finished row so overwrite it onto a->a */
882b051339dSHong Zhang       pv   = a->a + ai[r[i]] ;
883b051339dSHong Zhang       pj   = aj + ai[r[i]] ;
884b051339dSHong Zhang       nz   = ai[r[i]+1] - ai[r[i]];
885b051339dSHong Zhang       nbdiag = diag[r[i]] - ai[r[i]]; /* num of entries before the diagonal */
886137fb511SHong Zhang 
887137fb511SHong Zhang       rs   = 0.0;
888137fb511SHong Zhang       for (j=0; j<nz; j++) {
889b051339dSHong Zhang         pv[j] = rtmp[pj[j]];
890b051339dSHong Zhang         if (j != nbdiag) rs += PetscAbsScalar(pv[j]);
891137fb511SHong Zhang       }
892137fb511SHong Zhang 
893137fb511SHong Zhang       /* 9/13/02 Victor Eijkhout suggested scaling zeropivot by rs for matrices with funny scalings */
894137fb511SHong Zhang       sctx.rs  = rs;
895b051339dSHong Zhang       sctx.pv  = pv[nbdiag];
896c6b1f410SHong Zhang       ierr = MatLUCheckShift_inline(info,sctx,i,newshift);CHKERRQ(ierr);
897137fb511SHong Zhang       if (newshift == 1) break;
898137fb511SHong Zhang     }
899137fb511SHong Zhang 
9000481f469SBarry Smith     if (info->shiftpd && !sctx.lushift && sctx.shift_fraction>0 && sctx.nshift<sctx.nshift_max) {
901137fb511SHong Zhang       /*
902137fb511SHong Zhang        * if no shift in this attempt & shifting & started shifting & can refine,
903137fb511SHong Zhang        * then try lower shift
904137fb511SHong Zhang        */
9050481f469SBarry Smith       sctx.shift_hi        = sctx.shift_fraction;
9060481f469SBarry Smith       sctx.shift_fraction = (sctx.shift_hi+sctx.shift_lo)/2.;
9070481f469SBarry Smith       sctx.shift_amount    = sctx.shift_fraction * sctx.shift_top;
908137fb511SHong Zhang       sctx.lushift         = PETSC_TRUE;
909137fb511SHong Zhang       sctx.nshift++;
910137fb511SHong Zhang     }
911137fb511SHong Zhang   } while (sctx.lushift);
912137fb511SHong Zhang 
913137fb511SHong Zhang   /* invert diagonal entries for simplier triangular solves */
914137fb511SHong Zhang   for (i=0; i<n; i++) {
915b051339dSHong Zhang     a->a[diag[r[i]]] = 1.0/a->a[diag[r[i]]];
916137fb511SHong Zhang   }
917137fb511SHong Zhang 
918137fb511SHong Zhang   ierr = PetscFree(rtmp);CHKERRQ(ierr);
919137fb511SHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
920137fb511SHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
921b051339dSHong Zhang   A->ops->solve             = MatSolve_SeqAIJ_InplaceWithPerm;
922db4efbfdSBarry Smith   A->ops->solveadd          = MatSolveAdd_SeqAIJ;
923db4efbfdSBarry Smith   A->ops->solvetranspose    = MatSolveTranspose_SeqAIJ;
924db4efbfdSBarry Smith   A->ops->solvetransposeadd = MatSolveTransposeAdd_SeqAIJ;
925b051339dSHong Zhang   A->assembled = PETSC_TRUE;
9265c9eb25fSBarry Smith   A->preallocated = PETSC_TRUE;
927d0f46423SBarry Smith   ierr = PetscLogFlops(A->cmap->n);CHKERRQ(ierr);
928137fb511SHong Zhang   if (sctx.nshift){
929e738e422SBarry Smith     if (info->shiftpd) {
9300481f469SBarry 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);
931e738e422SBarry Smith     } else if (info->shiftnz) {
932e738e422SBarry Smith       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
933137fb511SHong Zhang     }
934137fb511SHong Zhang   }
935137fb511SHong Zhang   PetscFunctionReturn(0);
936137fb511SHong Zhang }
937137fb511SHong Zhang 
9380a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
9394a2ae208SSatish Balay #undef __FUNCT__
9404a2ae208SSatish Balay #define __FUNCT__ "MatLUFactor_SeqAIJ"
9410481f469SBarry Smith PetscErrorCode MatLUFactor_SeqAIJ(Mat A,IS row,IS col,const MatFactorInfo *info)
942da3a660dSBarry Smith {
943dfbe8321SBarry Smith   PetscErrorCode ierr;
944416022c9SBarry Smith   Mat            C;
945416022c9SBarry Smith 
9463a40ed3dSBarry Smith   PetscFunctionBegin;
94743244d56SBarry Smith   ierr = MatGetFactor(A,MAT_SOLVER_PETSC,MAT_FACTOR_LU,&C);CHKERRQ(ierr);
948719d5645SBarry Smith   ierr = MatLUFactorSymbolic(C,A,row,col,info);CHKERRQ(ierr);
949719d5645SBarry Smith   ierr = MatLUFactorNumeric(C,A,info);CHKERRQ(ierr);
950db4efbfdSBarry Smith   A->ops->solve            = C->ops->solve;
951db4efbfdSBarry Smith   A->ops->solvetranspose   = C->ops->solvetranspose;
952273d9f13SBarry Smith   ierr = MatHeaderCopy(A,C);CHKERRQ(ierr);
95352e6d16bSBarry Smith   ierr = PetscLogObjectParent(A,((Mat_SeqAIJ*)(A->data))->icol);CHKERRQ(ierr);
9543a40ed3dSBarry Smith   PetscFunctionReturn(0);
955da3a660dSBarry Smith }
9560a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
9571d098868SHong Zhang 
958889a8dedSBarry Smith 
9594a2ae208SSatish Balay #undef __FUNCT__
9604a2ae208SSatish Balay #define __FUNCT__ "MatSolve_SeqAIJ"
961dfbe8321SBarry Smith PetscErrorCode MatSolve_SeqAIJ(Mat A,Vec bb,Vec xx)
9628c37ef55SBarry Smith {
963416022c9SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
964416022c9SBarry Smith   IS                iscol = a->col,isrow = a->row;
9656849ba73SBarry Smith   PetscErrorCode    ierr;
9665d0c19d7SBarry Smith   PetscInt          i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
9675d0c19d7SBarry Smith   PetscInt          nz;
9685d0c19d7SBarry Smith   const PetscInt    *rout,*cout,*r,*c;
969dd6ea824SBarry Smith   PetscScalar       *x,*tmp,*tmps,sum;
970d9fead3dSBarry Smith   const PetscScalar *b;
971dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
9728c37ef55SBarry Smith 
9733a40ed3dSBarry Smith   PetscFunctionBegin;
9743a40ed3dSBarry Smith   if (!n) PetscFunctionReturn(0);
97591bf9adeSBarry Smith 
976d9fead3dSBarry Smith   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
9771ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
978416022c9SBarry Smith   tmp  = a->solve_work;
9798c37ef55SBarry Smith 
9803b2fbd54SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
9813b2fbd54SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
9828c37ef55SBarry Smith 
9838c37ef55SBarry Smith   /* forward solve the lower triangular */
9848c37ef55SBarry Smith   tmp[0] = b[*r++];
985010a20caSHong Zhang   tmps   = tmp;
9868c37ef55SBarry Smith   for (i=1; i<n; i++) {
987010a20caSHong Zhang     v   = aa + ai[i] ;
988010a20caSHong Zhang     vi  = aj + ai[i] ;
98953e7425aSBarry Smith     nz  = a->diag[i] - ai[i];
99053e7425aSBarry Smith     sum = b[*r++];
991003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
9928c37ef55SBarry Smith     tmp[i] = sum;
9938c37ef55SBarry Smith   }
9948c37ef55SBarry Smith 
9958c37ef55SBarry Smith   /* backward solve the upper triangular */
9968c37ef55SBarry Smith   for (i=n-1; i>=0; i--){
997010a20caSHong Zhang     v   = aa + a->diag[i] + 1;
998010a20caSHong Zhang     vi  = aj + a->diag[i] + 1;
999416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
10008c37ef55SBarry Smith     sum = tmp[i];
1001003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
1002010a20caSHong Zhang     x[*c--] = tmp[i] = sum*aa[a->diag[i]];
10038c37ef55SBarry Smith   }
10048c37ef55SBarry Smith 
10053b2fbd54SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
10063b2fbd54SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
1007d9fead3dSBarry Smith   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
10081ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1009dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
10103a40ed3dSBarry Smith   PetscFunctionReturn(0);
10118c37ef55SBarry Smith }
1012026e39d0SSatish Balay 
10132bebee5dSHong Zhang #undef __FUNCT__
10142bebee5dSHong Zhang #define __FUNCT__ "MatMatSolve_SeqAIJ"
10152bebee5dSHong Zhang PetscErrorCode MatMatSolve_SeqAIJ(Mat A,Mat B,Mat X)
10162bebee5dSHong Zhang {
10172bebee5dSHong Zhang   Mat_SeqAIJ      *a = (Mat_SeqAIJ*)A->data;
10182bebee5dSHong Zhang   IS              iscol = a->col,isrow = a->row;
10192bebee5dSHong Zhang   PetscErrorCode  ierr;
10205d0c19d7SBarry Smith   PetscInt        i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
10215d0c19d7SBarry Smith   PetscInt        nz,neq;
10225d0c19d7SBarry Smith   const PetscInt  *rout,*cout,*r,*c;
1023dd6ea824SBarry Smith   PetscScalar     *x,*b,*tmp,*tmps,sum;
1024dd6ea824SBarry Smith   const MatScalar *aa = a->a,*v;
1025db4efbfdSBarry Smith   PetscTruth      bisdense,xisdense;
10262bebee5dSHong Zhang 
10272bebee5dSHong Zhang   PetscFunctionBegin;
10282bebee5dSHong Zhang   if (!n) PetscFunctionReturn(0);
10292bebee5dSHong Zhang 
1030db4efbfdSBarry Smith   ierr = PetscTypeCompare((PetscObject)B,MATSEQDENSE,&bisdense);CHKERRQ(ierr);
1031db4efbfdSBarry Smith   if (!bisdense) SETERRQ(PETSC_ERR_ARG_INCOMP,"B matrix must be a SeqDense matrix");
1032db4efbfdSBarry Smith   ierr = PetscTypeCompare((PetscObject)X,MATSEQDENSE,&xisdense);CHKERRQ(ierr);
1033db4efbfdSBarry Smith   if (!xisdense) SETERRQ(PETSC_ERR_ARG_INCOMP,"X matrix must be a SeqDense matrix");
1034db4efbfdSBarry Smith 
10352bebee5dSHong Zhang   ierr = MatGetArray(B,&b);CHKERRQ(ierr);
10362bebee5dSHong Zhang   ierr = MatGetArray(X,&x);CHKERRQ(ierr);
10372bebee5dSHong Zhang 
10382bebee5dSHong Zhang   tmp  = a->solve_work;
10392bebee5dSHong Zhang   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
10402bebee5dSHong Zhang   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
10412bebee5dSHong Zhang 
1042a36b22ccSSatish Balay   for (neq=0; neq<B->cmap->n; neq++){
10432bebee5dSHong Zhang     /* forward solve the lower triangular */
10442bebee5dSHong Zhang     tmp[0] = b[r[0]];
10452bebee5dSHong Zhang     tmps   = tmp;
10462bebee5dSHong Zhang     for (i=1; i<n; i++) {
10472bebee5dSHong Zhang       v   = aa + ai[i] ;
10482bebee5dSHong Zhang       vi  = aj + ai[i] ;
10492bebee5dSHong Zhang       nz  = a->diag[i] - ai[i];
10502bebee5dSHong Zhang       sum = b[r[i]];
1051003131ecSBarry Smith       PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
10522bebee5dSHong Zhang       tmp[i] = sum;
10532bebee5dSHong Zhang     }
10542bebee5dSHong Zhang     /* backward solve the upper triangular */
10552bebee5dSHong Zhang     for (i=n-1; i>=0; i--){
10562bebee5dSHong Zhang       v   = aa + a->diag[i] + 1;
10572bebee5dSHong Zhang       vi  = aj + a->diag[i] + 1;
10582bebee5dSHong Zhang       nz  = ai[i+1] - a->diag[i] - 1;
10592bebee5dSHong Zhang       sum = tmp[i];
1060003131ecSBarry Smith       PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
10612bebee5dSHong Zhang       x[c[i]] = tmp[i] = sum*aa[a->diag[i]];
10622bebee5dSHong Zhang     }
10632bebee5dSHong Zhang 
10642bebee5dSHong Zhang     b += n;
10652bebee5dSHong Zhang     x += n;
10662bebee5dSHong Zhang   }
10672bebee5dSHong Zhang   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
10682bebee5dSHong Zhang   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
10692bebee5dSHong Zhang   ierr = MatRestoreArray(B,&b);CHKERRQ(ierr);
10702bebee5dSHong Zhang   ierr = MatRestoreArray(X,&x);CHKERRQ(ierr);
1071dc0b31edSSatish Balay   ierr = PetscLogFlops(B->cmap->n*(2.0*a->nz - n));CHKERRQ(ierr);
10722bebee5dSHong Zhang   PetscFunctionReturn(0);
10732bebee5dSHong Zhang }
10742bebee5dSHong Zhang 
1075137fb511SHong Zhang #undef __FUNCT__
1076137fb511SHong Zhang #define __FUNCT__ "MatSolve_SeqAIJ_InplaceWithPerm"
1077137fb511SHong Zhang PetscErrorCode MatSolve_SeqAIJ_InplaceWithPerm(Mat A,Vec bb,Vec xx)
1078137fb511SHong Zhang {
1079137fb511SHong Zhang   Mat_SeqAIJ      *a = (Mat_SeqAIJ*)A->data;
1080137fb511SHong Zhang   IS              iscol = a->col,isrow = a->row;
1081137fb511SHong Zhang   PetscErrorCode  ierr;
10825d0c19d7SBarry Smith   const PetscInt  *r,*c,*rout,*cout;
10835d0c19d7SBarry Smith   PetscInt        i, n = A->rmap->n,*vi,*ai = a->i,*aj = a->j;
10845d0c19d7SBarry Smith   PetscInt        nz,row;
1085dd6ea824SBarry Smith   PetscScalar     *x,*b,*tmp,*tmps,sum;
1086dd6ea824SBarry Smith   const MatScalar *aa = a->a,*v;
1087137fb511SHong Zhang 
1088137fb511SHong Zhang   PetscFunctionBegin;
1089137fb511SHong Zhang   if (!n) PetscFunctionReturn(0);
1090137fb511SHong Zhang 
1091137fb511SHong Zhang   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
1092137fb511SHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1093137fb511SHong Zhang   tmp  = a->solve_work;
1094137fb511SHong Zhang 
1095137fb511SHong Zhang   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
1096137fb511SHong Zhang   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
1097137fb511SHong Zhang 
1098137fb511SHong Zhang   /* forward solve the lower triangular */
1099137fb511SHong Zhang   tmp[0] = b[*r++];
1100137fb511SHong Zhang   tmps   = tmp;
1101137fb511SHong Zhang   for (row=1; row<n; row++) {
1102137fb511SHong Zhang     i   = rout[row]; /* permuted row */
1103137fb511SHong Zhang     v   = aa + ai[i] ;
1104137fb511SHong Zhang     vi  = aj + ai[i] ;
1105137fb511SHong Zhang     nz  = a->diag[i] - ai[i];
1106137fb511SHong Zhang     sum = b[*r++];
1107003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
1108137fb511SHong Zhang     tmp[row] = sum;
1109137fb511SHong Zhang   }
1110137fb511SHong Zhang 
1111137fb511SHong Zhang   /* backward solve the upper triangular */
1112137fb511SHong Zhang   for (row=n-1; row>=0; row--){
1113137fb511SHong Zhang     i   = rout[row]; /* permuted row */
1114137fb511SHong Zhang     v   = aa + a->diag[i] + 1;
1115137fb511SHong Zhang     vi  = aj + a->diag[i] + 1;
1116137fb511SHong Zhang     nz  = ai[i+1] - a->diag[i] - 1;
1117137fb511SHong Zhang     sum = tmp[row];
1118003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,tmps,v,vi,nz);
1119137fb511SHong Zhang     x[*c--] = tmp[row] = sum*aa[a->diag[i]];
1120137fb511SHong Zhang   }
1121137fb511SHong Zhang 
1122137fb511SHong Zhang   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
1123137fb511SHong Zhang   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
1124137fb511SHong Zhang   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
1125137fb511SHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1126dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
1127137fb511SHong Zhang   PetscFunctionReturn(0);
1128137fb511SHong Zhang }
1129137fb511SHong Zhang 
1130930ae53cSBarry Smith /* ----------------------------------------------------------- */
1131a4005a5dSBarry Smith #include "../src/mat/impls/aij/seq/ftn-kernels/fsolve.h"
11324a2ae208SSatish Balay #undef __FUNCT__
11334a2ae208SSatish Balay #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering"
1134dfbe8321SBarry Smith PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering(Mat A,Vec bb,Vec xx)
1135930ae53cSBarry Smith {
1136930ae53cSBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
11376849ba73SBarry Smith   PetscErrorCode    ierr;
1138d0f46423SBarry Smith   PetscInt          n = A->rmap->n;
1139b377110cSBarry Smith   const PetscInt    *ai = a->i,*aj = a->j,*adiag = a->diag;
1140356650c2SBarry Smith   PetscScalar       *x;
1141356650c2SBarry Smith   const PetscScalar *b;
1142dd6ea824SBarry Smith   const MatScalar   *aa = a->a;
1143aa482453SBarry Smith #if !defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ)
1144356650c2SBarry Smith   PetscInt          adiag_i,i,nz,ai_i;
1145b377110cSBarry Smith   const PetscInt    *vi;
1146dd6ea824SBarry Smith   const MatScalar   *v;
1147dd6ea824SBarry Smith   PetscScalar       sum;
1148d85afc42SSatish Balay #endif
1149930ae53cSBarry Smith 
11503a40ed3dSBarry Smith   PetscFunctionBegin;
11513a40ed3dSBarry Smith   if (!n) PetscFunctionReturn(0);
1152930ae53cSBarry Smith 
1153356650c2SBarry Smith   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
11541ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1155930ae53cSBarry Smith 
1156aa482453SBarry Smith #if defined(PETSC_USE_FORTRAN_KERNEL_SOLVEAIJ)
11571c4feecaSSatish Balay   fortransolveaij_(&n,x,ai,aj,adiag,aa,b);
11581c4feecaSSatish Balay #else
1159930ae53cSBarry Smith   /* forward solve the lower triangular */
1160930ae53cSBarry Smith   x[0] = b[0];
1161930ae53cSBarry Smith   for (i=1; i<n; i++) {
1162930ae53cSBarry Smith     ai_i = ai[i];
1163930ae53cSBarry Smith     v    = aa + ai_i;
1164930ae53cSBarry Smith     vi   = aj + ai_i;
1165930ae53cSBarry Smith     nz   = adiag[i] - ai_i;
1166930ae53cSBarry Smith     sum  = b[i];
1167003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
1168930ae53cSBarry Smith     x[i] = sum;
1169930ae53cSBarry Smith   }
1170930ae53cSBarry Smith 
1171930ae53cSBarry Smith   /* backward solve the upper triangular */
1172930ae53cSBarry Smith   for (i=n-1; i>=0; i--){
1173930ae53cSBarry Smith     adiag_i = adiag[i];
1174d708749eSBarry Smith     v       = aa + adiag_i + 1;
1175d708749eSBarry Smith     vi      = aj + adiag_i + 1;
1176930ae53cSBarry Smith     nz      = ai[i+1] - adiag_i - 1;
1177930ae53cSBarry Smith     sum     = x[i];
1178003131ecSBarry Smith     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
1179930ae53cSBarry Smith     x[i]    = sum*aa[adiag_i];
1180930ae53cSBarry Smith   }
11811c4feecaSSatish Balay #endif
1182dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
1183356650c2SBarry Smith   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
11841ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
11853a40ed3dSBarry Smith   PetscFunctionReturn(0);
1186930ae53cSBarry Smith }
1187930ae53cSBarry Smith 
11884a2ae208SSatish Balay #undef __FUNCT__
11894a2ae208SSatish Balay #define __FUNCT__ "MatSolveAdd_SeqAIJ"
1190dfbe8321SBarry Smith PetscErrorCode MatSolveAdd_SeqAIJ(Mat A,Vec bb,Vec yy,Vec xx)
1191da3a660dSBarry Smith {
1192416022c9SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
1193416022c9SBarry Smith   IS                iscol = a->col,isrow = a->row;
11946849ba73SBarry Smith   PetscErrorCode    ierr;
119504fbf559SBarry Smith   PetscInt          i, n = A->rmap->n,j;
11965d0c19d7SBarry Smith   PetscInt          nz;
119704fbf559SBarry Smith   const PetscInt    *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j;
119804fbf559SBarry Smith   PetscScalar       *x,*tmp,sum;
119904fbf559SBarry Smith   const PetscScalar *b;
1200dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
1201da3a660dSBarry Smith 
12023a40ed3dSBarry Smith   PetscFunctionBegin;
120378b31e54SBarry Smith   if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);}
1204da3a660dSBarry Smith 
120504fbf559SBarry Smith   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
12061ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1207416022c9SBarry Smith   tmp  = a->solve_work;
1208da3a660dSBarry Smith 
12093b2fbd54SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
12103b2fbd54SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout + (n-1);
1211da3a660dSBarry Smith 
1212da3a660dSBarry Smith   /* forward solve the lower triangular */
1213da3a660dSBarry Smith   tmp[0] = b[*r++];
1214da3a660dSBarry Smith   for (i=1; i<n; i++) {
1215010a20caSHong Zhang     v   = aa + ai[i] ;
1216010a20caSHong Zhang     vi  = aj + ai[i] ;
1217416022c9SBarry Smith     nz  = a->diag[i] - ai[i];
1218da3a660dSBarry Smith     sum = b[*r++];
121904fbf559SBarry Smith     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
1220da3a660dSBarry Smith     tmp[i] = sum;
1221da3a660dSBarry Smith   }
1222da3a660dSBarry Smith 
1223da3a660dSBarry Smith   /* backward solve the upper triangular */
1224da3a660dSBarry Smith   for (i=n-1; i>=0; i--){
1225010a20caSHong Zhang     v   = aa + a->diag[i] + 1;
1226010a20caSHong Zhang     vi  = aj + a->diag[i] + 1;
1227416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
1228da3a660dSBarry Smith     sum = tmp[i];
122904fbf559SBarry Smith     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
1230010a20caSHong Zhang     tmp[i] = sum*aa[a->diag[i]];
1231da3a660dSBarry Smith     x[*c--] += tmp[i];
1232da3a660dSBarry Smith   }
1233da3a660dSBarry Smith 
12343b2fbd54SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
12353b2fbd54SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
123604fbf559SBarry Smith   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
12371ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1238dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
1239898183ecSLois Curfman McInnes 
12403a40ed3dSBarry Smith   PetscFunctionReturn(0);
1241da3a660dSBarry Smith }
124204fbf559SBarry Smith 
12434a2ae208SSatish Balay #undef __FUNCT__
12443c0119dfSShri Abhyankar #define __FUNCT__ "MatSolveAdd_SeqAIJ_newdatastruct"
12453c0119dfSShri Abhyankar PetscErrorCode MatSolveAdd_SeqAIJ_newdatastruct(Mat A,Vec bb,Vec yy,Vec xx)
12463c0119dfSShri Abhyankar {
12473c0119dfSShri Abhyankar   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
12483c0119dfSShri Abhyankar   IS                iscol = a->col,isrow = a->row;
12493c0119dfSShri Abhyankar   PetscErrorCode    ierr;
12503c0119dfSShri Abhyankar   PetscInt          i, n = A->rmap->n,j;
12513c0119dfSShri Abhyankar   PetscInt          nz;
12523c0119dfSShri Abhyankar   const PetscInt    *rout,*cout,*r,*c,*vi,*ai = a->i,*aj = a->j,*adiag = a->diag;
12533c0119dfSShri Abhyankar   PetscScalar       *x,*tmp,sum;
12543c0119dfSShri Abhyankar   const PetscScalar *b;
12553c0119dfSShri Abhyankar   const MatScalar   *aa = a->a,*v;
12563c0119dfSShri Abhyankar 
12573c0119dfSShri Abhyankar   PetscFunctionBegin;
12583c0119dfSShri Abhyankar   if (yy != xx) {ierr = VecCopy(yy,xx);CHKERRQ(ierr);}
12593c0119dfSShri Abhyankar 
12603c0119dfSShri Abhyankar   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
12613c0119dfSShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
12623c0119dfSShri Abhyankar   tmp  = a->solve_work;
12633c0119dfSShri Abhyankar 
12643c0119dfSShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
12653c0119dfSShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
12663c0119dfSShri Abhyankar 
12673c0119dfSShri Abhyankar   /* forward solve the lower triangular */
12683c0119dfSShri Abhyankar   tmp[0] = b[r[0]];
12693c0119dfSShri Abhyankar   v      = aa;
12703c0119dfSShri Abhyankar   vi     = aj;
12713c0119dfSShri Abhyankar   for (i=1; i<n; i++) {
12723c0119dfSShri Abhyankar     nz  = ai[i+1] - ai[i];
12733c0119dfSShri Abhyankar     sum = b[r[i]];
12743c0119dfSShri Abhyankar     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
12753c0119dfSShri Abhyankar     tmp[i] = sum;
12763c0119dfSShri Abhyankar     v += nz; vi += nz;
12773c0119dfSShri Abhyankar   }
12783c0119dfSShri Abhyankar 
12793c0119dfSShri Abhyankar   /* backward solve the upper triangular */
12803c0119dfSShri Abhyankar   v  = aa + adiag[n-1];
12813c0119dfSShri Abhyankar   vi = aj + adiag[n-1];
12823c0119dfSShri Abhyankar   for (i=n-1; i>=0; i--){
12833c0119dfSShri Abhyankar     nz  = adiag[i] - adiag[i+1] - 1;
12843c0119dfSShri Abhyankar     sum = tmp[i];
12853c0119dfSShri Abhyankar     for (j=0; j<nz; j++) sum -= v[j]*tmp[vi[j]];
12863c0119dfSShri Abhyankar     tmp[i] = sum*v[nz];
12873c0119dfSShri Abhyankar     x[c[i]] += tmp[i];
12883c0119dfSShri Abhyankar     v += nz+1; vi += nz+1;
12893c0119dfSShri Abhyankar   }
12903c0119dfSShri Abhyankar 
12913c0119dfSShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
12923c0119dfSShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
12933c0119dfSShri Abhyankar   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
12943c0119dfSShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
12953c0119dfSShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz);CHKERRQ(ierr);
12963c0119dfSShri Abhyankar 
12973c0119dfSShri Abhyankar   PetscFunctionReturn(0);
12983c0119dfSShri Abhyankar }
12993c0119dfSShri Abhyankar 
13003c0119dfSShri Abhyankar #undef __FUNCT__
13014a2ae208SSatish Balay #define __FUNCT__ "MatSolveTranspose_SeqAIJ"
1302dfbe8321SBarry Smith PetscErrorCode MatSolveTranspose_SeqAIJ(Mat A,Vec bb,Vec xx)
1303da3a660dSBarry Smith {
1304416022c9SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
13052235e667SBarry Smith   IS                iscol = a->col,isrow = a->row;
13066849ba73SBarry Smith   PetscErrorCode    ierr;
130704fbf559SBarry Smith   const PetscInt    *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi;
130804fbf559SBarry Smith   PetscInt          i,n = A->rmap->n,j;
130904fbf559SBarry Smith   PetscInt          nz;
131004fbf559SBarry Smith   PetscScalar       *x,*tmp,s1;
1311dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
131204fbf559SBarry Smith   const PetscScalar *b;
1313da3a660dSBarry Smith 
13143a40ed3dSBarry Smith   PetscFunctionBegin;
131504fbf559SBarry Smith   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
13161ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1317416022c9SBarry Smith   tmp  = a->solve_work;
1318da3a660dSBarry Smith 
13192235e667SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
13202235e667SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
1321da3a660dSBarry Smith 
1322da3a660dSBarry Smith   /* copy the b into temp work space according to permutation */
13232235e667SBarry Smith   for (i=0; i<n; i++) tmp[i] = b[c[i]];
1324da3a660dSBarry Smith 
1325da3a660dSBarry Smith   /* forward solve the U^T */
1326da3a660dSBarry Smith   for (i=0; i<n; i++) {
1327010a20caSHong Zhang     v   = aa + diag[i] ;
1328010a20caSHong Zhang     vi  = aj + diag[i] + 1;
1329f1af5d2fSBarry Smith     nz  = ai[i+1] - diag[i] - 1;
1330c38d4ed2SBarry Smith     s1  = tmp[i];
1331ef66eb69SBarry Smith     s1 *= (*v++);  /* multiply by inverse of diagonal entry */
133204fbf559SBarry Smith     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1333c38d4ed2SBarry Smith     tmp[i] = s1;
1334da3a660dSBarry Smith   }
1335da3a660dSBarry Smith 
1336da3a660dSBarry Smith   /* backward solve the L^T */
1337da3a660dSBarry Smith   for (i=n-1; i>=0; i--){
1338010a20caSHong Zhang     v   = aa + diag[i] - 1 ;
1339010a20caSHong Zhang     vi  = aj + diag[i] - 1 ;
1340f1af5d2fSBarry Smith     nz  = diag[i] - ai[i];
1341f1af5d2fSBarry Smith     s1  = tmp[i];
134204fbf559SBarry Smith     for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j];
1343da3a660dSBarry Smith   }
1344da3a660dSBarry Smith 
1345da3a660dSBarry Smith   /* copy tmp into x according to permutation */
1346591294e4SBarry Smith   for (i=0; i<n; i++) x[r[i]] = tmp[i];
1347da3a660dSBarry Smith 
13482235e667SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
13492235e667SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
135004fbf559SBarry Smith   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
13511ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1352da3a660dSBarry Smith 
1353dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr);
13543a40ed3dSBarry Smith   PetscFunctionReturn(0);
1355da3a660dSBarry Smith }
1356da3a660dSBarry Smith 
1357d1fa9404SShri Abhyankar #undef __FUNCT__
13584395751fSShri Abhyankar #define __FUNCT__ "MatSolveTranspose_SeqAIJ_newdatastruct"
13594395751fSShri Abhyankar PetscErrorCode MatSolveTranspose_SeqAIJ_newdatastruct(Mat A,Vec bb,Vec xx)
1360d1fa9404SShri Abhyankar {
1361d1fa9404SShri Abhyankar   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
1362d1fa9404SShri Abhyankar   IS                iscol = a->col,isrow = a->row;
1363d1fa9404SShri Abhyankar   PetscErrorCode    ierr;
1364d1fa9404SShri Abhyankar   const PetscInt    *rout,*cout,*r,*c,*adiag = a->diag,*ai = a->i,*aj = a->j,*vi;
1365d1fa9404SShri Abhyankar   PetscInt          i,n = A->rmap->n,j;
1366d1fa9404SShri Abhyankar   PetscInt          nz;
1367d1fa9404SShri Abhyankar   PetscScalar       *x,*tmp,s1;
1368d1fa9404SShri Abhyankar   const MatScalar   *aa = a->a,*v;
1369d1fa9404SShri Abhyankar   const PetscScalar *b;
1370d1fa9404SShri Abhyankar 
1371d1fa9404SShri Abhyankar   PetscFunctionBegin;
1372d1fa9404SShri Abhyankar   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
1373d1fa9404SShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1374d1fa9404SShri Abhyankar   tmp  = a->solve_work;
1375d1fa9404SShri Abhyankar 
1376d1fa9404SShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
1377d1fa9404SShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
1378d1fa9404SShri Abhyankar 
1379d1fa9404SShri Abhyankar   /* copy the b into temp work space according to permutation */
1380d1fa9404SShri Abhyankar   for (i=0; i<n; i++) tmp[i] = b[c[i]];
1381d1fa9404SShri Abhyankar 
1382d1fa9404SShri Abhyankar   /* forward solve the U^T */
1383d1fa9404SShri Abhyankar   for (i=0; i<n; i++) {
1384d1fa9404SShri Abhyankar     v   = aa + adiag[i+1] + 1;
1385d1fa9404SShri Abhyankar     vi  = aj + adiag[i+1] + 1;
1386d1fa9404SShri Abhyankar     nz  = adiag[i] - adiag[i+1] - 1;
1387d1fa9404SShri Abhyankar     s1  = tmp[i];
1388d1fa9404SShri Abhyankar     s1 *= v[nz];  /* multiply by inverse of diagonal entry */
1389d1fa9404SShri Abhyankar     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1390d1fa9404SShri Abhyankar     tmp[i] = s1;
1391d1fa9404SShri Abhyankar   }
1392d1fa9404SShri Abhyankar 
1393d1fa9404SShri Abhyankar   /* backward solve the L^T */
1394d1fa9404SShri Abhyankar   for (i=n-1; i>=0; i--){
1395d1fa9404SShri Abhyankar     v   = aa + ai[i];
1396d1fa9404SShri Abhyankar     vi  = aj + ai[i];
1397d1fa9404SShri Abhyankar     nz  = ai[i+1] - ai[i];
1398d1fa9404SShri Abhyankar     s1  = tmp[i];
1399d1fa9404SShri Abhyankar     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1400d1fa9404SShri Abhyankar   }
1401d1fa9404SShri Abhyankar 
1402d1fa9404SShri Abhyankar   /* copy tmp into x according to permutation */
1403d1fa9404SShri Abhyankar   for (i=0; i<n; i++) x[r[i]] = tmp[i];
1404d1fa9404SShri Abhyankar 
1405d1fa9404SShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
1406d1fa9404SShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
1407d1fa9404SShri Abhyankar   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
1408d1fa9404SShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1409d1fa9404SShri Abhyankar 
1410d1fa9404SShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr);
1411d1fa9404SShri Abhyankar   PetscFunctionReturn(0);
1412d1fa9404SShri Abhyankar }
1413d1fa9404SShri Abhyankar 
14144a2ae208SSatish Balay #undef __FUNCT__
14154a2ae208SSatish Balay #define __FUNCT__ "MatSolveTransposeAdd_SeqAIJ"
1416dfbe8321SBarry Smith PetscErrorCode MatSolveTransposeAdd_SeqAIJ(Mat A,Vec bb,Vec zz,Vec xx)
1417da3a660dSBarry Smith {
1418416022c9SBarry Smith   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
14192235e667SBarry Smith   IS                iscol = a->col,isrow = a->row;
14206849ba73SBarry Smith   PetscErrorCode    ierr;
142104fbf559SBarry Smith   const PetscInt    *rout,*cout,*r,*c,*diag = a->diag,*ai = a->i,*aj = a->j,*vi;
142204fbf559SBarry Smith   PetscInt          i,n = A->rmap->n,j;
142304fbf559SBarry Smith   PetscInt          nz;
142404fbf559SBarry Smith   PetscScalar       *x,*tmp,s1;
1425dd6ea824SBarry Smith   const MatScalar   *aa = a->a,*v;
142604fbf559SBarry Smith   const PetscScalar *b;
14276abc6512SBarry Smith 
14283a40ed3dSBarry Smith   PetscFunctionBegin;
142923bc6035SMatthew Knepley   if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);}
143004fbf559SBarry Smith   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
14311ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1432416022c9SBarry Smith   tmp  = a->solve_work;
14336abc6512SBarry Smith 
14342235e667SBarry Smith   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
14352235e667SBarry Smith   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
14366abc6512SBarry Smith 
14376abc6512SBarry Smith   /* copy the b into temp work space according to permutation */
14382235e667SBarry Smith   for (i=0; i<n; i++) tmp[i] = b[c[i]];
14396abc6512SBarry Smith 
14406abc6512SBarry Smith   /* forward solve the U^T */
14416abc6512SBarry Smith   for (i=0; i<n; i++) {
1442010a20caSHong Zhang     v   = aa + diag[i] ;
1443010a20caSHong Zhang     vi  = aj + diag[i] + 1;
1444f1af5d2fSBarry Smith     nz  = ai[i+1] - diag[i] - 1;
144504fbf559SBarry Smith     s1  = tmp[i];
144604fbf559SBarry Smith     s1 *= (*v++);  /* multiply by inverse of diagonal entry */
144704fbf559SBarry Smith     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
144804fbf559SBarry Smith     tmp[i] = s1;
14496abc6512SBarry Smith   }
14506abc6512SBarry Smith 
14516abc6512SBarry Smith   /* backward solve the L^T */
14526abc6512SBarry Smith   for (i=n-1; i>=0; i--){
1453010a20caSHong Zhang     v   = aa + diag[i] - 1 ;
1454010a20caSHong Zhang     vi  = aj + diag[i] - 1 ;
1455f1af5d2fSBarry Smith     nz  = diag[i] - ai[i];
145604fbf559SBarry Smith     s1  = tmp[i];
145704fbf559SBarry Smith     for (j=0; j>-nz; j--) tmp[vi[j]] -= s1*v[j];
14586abc6512SBarry Smith   }
14596abc6512SBarry Smith 
14606abc6512SBarry Smith   /* copy tmp into x according to permutation */
14616abc6512SBarry Smith   for (i=0; i<n; i++) x[r[i]] += tmp[i];
14626abc6512SBarry Smith 
14632235e667SBarry Smith   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
14642235e667SBarry Smith   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
146504fbf559SBarry Smith   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
14661ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
14676abc6512SBarry Smith 
146804fbf559SBarry Smith   ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr);
14693a40ed3dSBarry Smith   PetscFunctionReturn(0);
1470da3a660dSBarry Smith }
147104fbf559SBarry Smith 
1472533e6011SShri Abhyankar #undef __FUNCT__
1473533e6011SShri Abhyankar #define __FUNCT__ "MatSolveTransposeAdd_SeqAIJ_newdatastruct"
1474533e6011SShri Abhyankar PetscErrorCode MatSolveTransposeAdd_SeqAIJ_newdatastruct(Mat A,Vec bb,Vec zz,Vec xx)
1475533e6011SShri Abhyankar {
1476533e6011SShri Abhyankar   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
1477533e6011SShri Abhyankar   IS                iscol = a->col,isrow = a->row;
1478533e6011SShri Abhyankar   PetscErrorCode    ierr;
1479533e6011SShri Abhyankar   const PetscInt    *rout,*cout,*r,*c,*adiag = a->diag,*ai = a->i,*aj = a->j,*vi;
1480533e6011SShri Abhyankar   PetscInt          i,n = A->rmap->n,j;
1481533e6011SShri Abhyankar   PetscInt          nz;
1482533e6011SShri Abhyankar   PetscScalar       *x,*tmp,s1;
1483533e6011SShri Abhyankar   const MatScalar   *aa = a->a,*v;
1484533e6011SShri Abhyankar   const PetscScalar *b;
1485533e6011SShri Abhyankar 
1486533e6011SShri Abhyankar   PetscFunctionBegin;
1487533e6011SShri Abhyankar   if (zz != xx) {ierr = VecCopy(zz,xx);CHKERRQ(ierr);}
1488533e6011SShri Abhyankar   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
1489533e6011SShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1490533e6011SShri Abhyankar   tmp  = a->solve_work;
1491533e6011SShri Abhyankar 
1492533e6011SShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
1493533e6011SShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
1494533e6011SShri Abhyankar 
1495533e6011SShri Abhyankar   /* copy the b into temp work space according to permutation */
1496533e6011SShri Abhyankar   for (i=0; i<n; i++) tmp[i] = b[c[i]];
1497533e6011SShri Abhyankar 
1498533e6011SShri Abhyankar   /* forward solve the U^T */
1499533e6011SShri Abhyankar   for (i=0; i<n; i++) {
1500533e6011SShri Abhyankar     v   = aa + adiag[i+1] + 1;
1501533e6011SShri Abhyankar     vi  = aj + adiag[i+1] + 1;
1502533e6011SShri Abhyankar     nz  = adiag[i] - adiag[i+1] - 1;
1503533e6011SShri Abhyankar     s1  = tmp[i];
1504533e6011SShri Abhyankar     s1 *= v[nz];  /* multiply by inverse of diagonal entry */
1505533e6011SShri Abhyankar     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1506533e6011SShri Abhyankar     tmp[i] = s1;
1507533e6011SShri Abhyankar   }
1508533e6011SShri Abhyankar 
1509533e6011SShri Abhyankar 
1510533e6011SShri Abhyankar   /* backward solve the L^T */
1511533e6011SShri Abhyankar   for (i=n-1; i>=0; i--){
1512533e6011SShri Abhyankar     v   = aa + ai[i] ;
1513533e6011SShri Abhyankar     vi  = aj + ai[i];
1514533e6011SShri Abhyankar     nz  = ai[i+1] - ai[i];
1515533e6011SShri Abhyankar     s1  = tmp[i];
1516c5e3b2a3SShri Abhyankar     for (j=0; j<nz; j++) tmp[vi[j]] -= s1*v[j];
1517533e6011SShri Abhyankar   }
1518533e6011SShri Abhyankar 
1519533e6011SShri Abhyankar   /* copy tmp into x according to permutation */
1520533e6011SShri Abhyankar   for (i=0; i<n; i++) x[r[i]] += tmp[i];
1521533e6011SShri Abhyankar 
1522533e6011SShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
1523533e6011SShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
1524533e6011SShri Abhyankar   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
1525533e6011SShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1526533e6011SShri Abhyankar 
1527533e6011SShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz-A->cmap->n);CHKERRQ(ierr);
1528533e6011SShri Abhyankar   PetscFunctionReturn(0);
1529533e6011SShri Abhyankar }
1530533e6011SShri Abhyankar 
15319e25ed09SBarry Smith /* ----------------------------------------------------------------*/
15323c5fc038SBarry Smith EXTERN PetscErrorCode Mat_CheckInode(Mat,PetscTruth);
1533f77e22a1SHong Zhang EXTERN PetscErrorCode MatDuplicateNoCreate_SeqAIJ(Mat,Mat,MatDuplicateOption,PetscTruth);
153408480c60SBarry Smith 
15358fc3a347SHong Zhang /*
15368a76ed4fSHong Zhang    ilu() under revised new data structure.
1537813a1f2bSShri Abhyankar    Factored arrays bj and ba are stored as
1538813a1f2bSShri Abhyankar      L(0,:), L(1,:), ...,L(n-1,:),  U(n-1,:),...,U(i,:),U(i-1,:),...,U(0,:)
1539813a1f2bSShri Abhyankar 
1540813a1f2bSShri Abhyankar    bi=fact->i is an array of size n+1, in which
1541813a1f2bSShri Abhyankar    bi+
1542baabb860SHong Zhang      bi[i]:  points to 1st entry of L(i,:),i=0,...,n-1
1543baabb860SHong Zhang      bi[n]:  points to L(n-1,n-1)+1
1544baabb860SHong Zhang 
1545813a1f2bSShri Abhyankar   bdiag=fact->diag is an array of size n+1,in which
1546baabb860SHong Zhang      bdiag[i]: points to diagonal of U(i,:), i=0,...,n-1
1547a24f213cSHong Zhang      bdiag[n]: points to entry of U(n-1,0)-1
1548813a1f2bSShri Abhyankar 
1549813a1f2bSShri Abhyankar    U(i,:) contains bdiag[i] as its last entry, i.e.,
1550813a1f2bSShri Abhyankar     U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i])
1551813a1f2bSShri Abhyankar */
1552813a1f2bSShri Abhyankar #undef __FUNCT__
1553b4f5729fSShri Abhyankar #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_ilu0_newdatastruct"
1554b4f5729fSShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ_ilu0_newdatastruct(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
1555813a1f2bSShri Abhyankar {
1556813a1f2bSShri Abhyankar 
1557813a1f2bSShri Abhyankar   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
1558813a1f2bSShri Abhyankar   PetscErrorCode     ierr;
1559813a1f2bSShri Abhyankar   PetscInt           n=A->rmap->n,*ai=a->i,*aj,*adiag=a->diag;
1560a24f213cSHong Zhang   PetscInt           i,j,nz,*bi,*bj,*bdiag;
15611df811f5SHong Zhang   PetscTruth         missing;
15621df811f5SHong Zhang   IS                 isicol;
1563813a1f2bSShri Abhyankar 
1564813a1f2bSShri Abhyankar   PetscFunctionBegin;
15651df811f5SHong 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);
15661df811f5SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr);
15671df811f5SHong Zhang   if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i);
15681df811f5SHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
15691df811f5SHong Zhang 
1570813a1f2bSShri Abhyankar   ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_FALSE);CHKERRQ(ierr);
1571813a1f2bSShri Abhyankar   b    = (Mat_SeqAIJ*)(fact)->data;
1572813a1f2bSShri Abhyankar 
1573813a1f2bSShri Abhyankar   /* allocate matrix arrays for new data structure */
1574813a1f2bSShri Abhyankar   ierr = PetscMalloc3(ai[n]+1,PetscScalar,&b->a,ai[n]+1,PetscInt,&b->j,n+1,PetscInt,&b->i);CHKERRQ(ierr);
1575813a1f2bSShri Abhyankar   ierr = PetscLogObjectMemory(fact,ai[n]*(sizeof(PetscScalar)+sizeof(PetscInt))+(n+1)*sizeof(PetscInt));CHKERRQ(ierr);
1576813a1f2bSShri Abhyankar   b->singlemalloc = PETSC_TRUE;
1577813a1f2bSShri Abhyankar   if (!b->diag){
1578813a1f2bSShri Abhyankar     ierr = PetscMalloc((n+1)*sizeof(PetscInt),&b->diag);CHKERRQ(ierr);
1579813a1f2bSShri Abhyankar     ierr = PetscLogObjectMemory(fact,(n+1)*sizeof(PetscInt));CHKERRQ(ierr);
1580813a1f2bSShri Abhyankar   }
1581813a1f2bSShri Abhyankar   bdiag = b->diag;
1582813a1f2bSShri Abhyankar 
1583813a1f2bSShri Abhyankar   if (n > 0) {
1584813a1f2bSShri Abhyankar     ierr = PetscMemzero(b->a,(ai[n])*sizeof(MatScalar));CHKERRQ(ierr);
1585813a1f2bSShri Abhyankar   }
1586813a1f2bSShri Abhyankar 
1587813a1f2bSShri Abhyankar   /* set bi and bj with new data structure */
1588813a1f2bSShri Abhyankar   bi = b->i;
1589813a1f2bSShri Abhyankar   bj = b->j;
1590813a1f2bSShri Abhyankar 
1591813a1f2bSShri Abhyankar   /* L part */
1592*e60cf9a0SBarry Smith   bi[0] = 0;
1593813a1f2bSShri Abhyankar   for (i=0; i<n; i++){
1594813a1f2bSShri Abhyankar     nz = adiag[i] - ai[i];
1595813a1f2bSShri Abhyankar     bi[i+1] = bi[i] + nz;
1596813a1f2bSShri Abhyankar     aj = a->j + ai[i];
1597813a1f2bSShri Abhyankar     for (j=0; j<nz; j++){
1598813a1f2bSShri Abhyankar       *bj = aj[j]; bj++;
1599813a1f2bSShri Abhyankar     }
1600813a1f2bSShri Abhyankar   }
1601813a1f2bSShri Abhyankar 
1602813a1f2bSShri Abhyankar   /* U part */
160362fbd6afSShri Abhyankar   bdiag[n] = bi[n]-1;
1604813a1f2bSShri Abhyankar   for (i=n-1; i>=0; i--){
1605813a1f2bSShri Abhyankar     nz = ai[i+1] - adiag[i] - 1;
1606813a1f2bSShri Abhyankar     aj = a->j + adiag[i] + 1;
1607813a1f2bSShri Abhyankar     for (j=0; j<nz; j++){
1608813a1f2bSShri Abhyankar       *bj = aj[j]; bj++;
1609813a1f2bSShri Abhyankar     }
1610813a1f2bSShri Abhyankar     /* diag[i] */
1611813a1f2bSShri Abhyankar     *bj = i; bj++;
1612a24f213cSHong Zhang     bdiag[i] = bdiag[i+1] + nz + 1;
1613813a1f2bSShri Abhyankar   }
16141df811f5SHong Zhang 
16151df811f5SHong Zhang   fact->factor                 = MAT_FACTOR_ILU;
16161df811f5SHong Zhang   fact->info.factor_mallocs    = 0;
16171df811f5SHong Zhang   fact->info.fill_ratio_given  = info->fill;
16181df811f5SHong Zhang   fact->info.fill_ratio_needed = 1.0;
16191df811f5SHong Zhang   fact->ops->lufactornumeric   = MatLUFactorNumeric_SeqAIJ_newdatastruct;
16201df811f5SHong Zhang 
16211df811f5SHong Zhang   b       = (Mat_SeqAIJ*)(fact)->data;
16221df811f5SHong Zhang   b->row  = isrow;
16231df811f5SHong Zhang   b->col  = iscol;
16241df811f5SHong Zhang   b->icol = isicol;
16251df811f5SHong Zhang   ierr    = PetscMalloc((fact->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
16261df811f5SHong Zhang   ierr    = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
16271df811f5SHong Zhang   ierr    = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
1628813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
1629813a1f2bSShri Abhyankar }
1630813a1f2bSShri Abhyankar 
1631813a1f2bSShri Abhyankar #undef __FUNCT__
1632b4f5729fSShri Abhyankar #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ_newdatastruct"
1633b4f5729fSShri Abhyankar PetscErrorCode MatILUFactorSymbolic_SeqAIJ_newdatastruct(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
1634813a1f2bSShri Abhyankar {
1635813a1f2bSShri Abhyankar   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
1636813a1f2bSShri Abhyankar   IS                 isicol;
1637813a1f2bSShri Abhyankar   PetscErrorCode     ierr;
1638813a1f2bSShri Abhyankar   const PetscInt     *r,*ic;
16391df811f5SHong Zhang   PetscInt           n=A->rmap->n,*ai=a->i,*aj=a->j;
1640813a1f2bSShri Abhyankar   PetscInt           *bi,*cols,nnz,*cols_lvl;
1641813a1f2bSShri Abhyankar   PetscInt           *bdiag,prow,fm,nzbd,reallocs=0,dcount=0;
1642813a1f2bSShri Abhyankar   PetscInt           i,levels,diagonal_fill;
1643813a1f2bSShri Abhyankar   PetscTruth         col_identity,row_identity;
1644813a1f2bSShri Abhyankar   PetscReal          f;
1645813a1f2bSShri Abhyankar   PetscInt           nlnk,*lnk,*lnk_lvl=PETSC_NULL;
1646813a1f2bSShri Abhyankar   PetscBT            lnkbt;
1647813a1f2bSShri Abhyankar   PetscInt           nzi,*bj,**bj_ptr,**bjlvl_ptr;
1648813a1f2bSShri Abhyankar   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
1649813a1f2bSShri Abhyankar   PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL;
1650813a1f2bSShri Abhyankar 
1651813a1f2bSShri Abhyankar   PetscFunctionBegin;
16521df811f5SHong Zhang   /* printf("MatILUFactorSymbolic_SeqAIJ_newdatastruct ...\n"); */
1653813a1f2bSShri Abhyankar   levels = (PetscInt)info->levels;
1654813a1f2bSShri Abhyankar   ierr   = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
1655813a1f2bSShri Abhyankar   ierr   = ISIdentity(iscol,&col_identity);CHKERRQ(ierr);
1656813a1f2bSShri Abhyankar 
1657813a1f2bSShri Abhyankar   if (!levels && row_identity && col_identity) {
1658813a1f2bSShri Abhyankar     /* special case: ilu(0) with natural ordering */
1659b4f5729fSShri Abhyankar     ierr = MatILUFactorSymbolic_SeqAIJ_ilu0_newdatastruct(fact,A,isrow,iscol,info);CHKERRQ(ierr);
1660813a1f2bSShri Abhyankar     PetscFunctionReturn(0);
1661813a1f2bSShri Abhyankar   }
1662813a1f2bSShri Abhyankar 
16631df811f5SHong 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);
16641df811f5SHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
1665813a1f2bSShri Abhyankar   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
1666813a1f2bSShri Abhyankar   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
1667813a1f2bSShri Abhyankar 
16681bfa06eaSJed Brown   /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */
16691bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
16701bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
1671*e60cf9a0SBarry Smith   bi[0] = bdiag[0] = 0;
1672813a1f2bSShri Abhyankar 
16730e83c824SBarry Smith   ierr = PetscMalloc2(n,PetscInt*,&bj_ptr,n,PetscInt*,&bjlvl_ptr);CHKERRQ(ierr);
1674813a1f2bSShri Abhyankar 
1675813a1f2bSShri Abhyankar   /* create a linked list for storing column indices of the active row */
1676813a1f2bSShri Abhyankar   nlnk = n + 1;
1677813a1f2bSShri Abhyankar   ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
1678813a1f2bSShri Abhyankar 
1679813a1f2bSShri Abhyankar   /* initial FreeSpace size is f*(ai[n]+1) */
16801df811f5SHong Zhang   f             = info->fill;
16811df811f5SHong Zhang   diagonal_fill = (PetscInt)info->diagonal_fill;
1682813a1f2bSShri Abhyankar   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr);
1683813a1f2bSShri Abhyankar   current_space = free_space;
1684813a1f2bSShri Abhyankar   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr);
1685813a1f2bSShri Abhyankar   current_space_lvl = free_space_lvl;
1686813a1f2bSShri Abhyankar 
1687813a1f2bSShri Abhyankar   for (i=0; i<n; i++) {
1688813a1f2bSShri Abhyankar     nzi = 0;
1689813a1f2bSShri Abhyankar     /* copy current row into linked list */
1690813a1f2bSShri Abhyankar     nnz  = ai[r[i]+1] - ai[r[i]];
1691813a1f2bSShri 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);
1692813a1f2bSShri Abhyankar     cols = aj + ai[r[i]];
1693813a1f2bSShri Abhyankar     lnk[i] = -1; /* marker to indicate if diagonal exists */
1694813a1f2bSShri Abhyankar     ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
1695813a1f2bSShri Abhyankar     nzi += nlnk;
1696813a1f2bSShri Abhyankar 
1697813a1f2bSShri Abhyankar     /* make sure diagonal entry is included */
1698813a1f2bSShri Abhyankar     if (diagonal_fill && lnk[i] == -1) {
1699813a1f2bSShri Abhyankar       fm = n;
1700813a1f2bSShri Abhyankar       while (lnk[fm] < i) fm = lnk[fm];
1701813a1f2bSShri Abhyankar       lnk[i]     = lnk[fm]; /* insert diagonal into linked list */
1702813a1f2bSShri Abhyankar       lnk[fm]    = i;
1703813a1f2bSShri Abhyankar       lnk_lvl[i] = 0;
1704813a1f2bSShri Abhyankar       nzi++; dcount++;
1705813a1f2bSShri Abhyankar     }
1706813a1f2bSShri Abhyankar 
1707813a1f2bSShri Abhyankar     /* add pivot rows into the active row */
1708813a1f2bSShri Abhyankar     nzbd = 0;
1709813a1f2bSShri Abhyankar     prow = lnk[n];
1710813a1f2bSShri Abhyankar     while (prow < i) {
1711813a1f2bSShri Abhyankar       nnz      = bdiag[prow];
1712813a1f2bSShri Abhyankar       cols     = bj_ptr[prow] + nnz + 1;
1713813a1f2bSShri Abhyankar       cols_lvl = bjlvl_ptr[prow] + nnz + 1;
1714813a1f2bSShri Abhyankar       nnz      = bi[prow+1] - bi[prow] - nnz - 1;
1715813a1f2bSShri Abhyankar       ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr);
1716813a1f2bSShri Abhyankar       nzi += nlnk;
1717813a1f2bSShri Abhyankar       prow = lnk[prow];
1718813a1f2bSShri Abhyankar       nzbd++;
1719813a1f2bSShri Abhyankar     }
1720813a1f2bSShri Abhyankar     bdiag[i] = nzbd;
1721813a1f2bSShri Abhyankar     bi[i+1]  = bi[i] + nzi;
1722813a1f2bSShri Abhyankar 
1723813a1f2bSShri Abhyankar     /* if free space is not available, make more free space */
1724813a1f2bSShri Abhyankar     if (current_space->local_remaining<nzi) {
1725813a1f2bSShri Abhyankar       nnz = 2*nzi*(n - i); /* estimated and max additional space needed */
1726813a1f2bSShri Abhyankar       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
1727813a1f2bSShri Abhyankar       ierr = PetscFreeSpaceGet(nnz,&current_space_lvl);CHKERRQ(ierr);
1728813a1f2bSShri Abhyankar       reallocs++;
1729813a1f2bSShri Abhyankar     }
1730813a1f2bSShri Abhyankar 
1731813a1f2bSShri Abhyankar     /* copy data into free_space and free_space_lvl, then initialize lnk */
1732813a1f2bSShri Abhyankar     ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
1733813a1f2bSShri Abhyankar     bj_ptr[i]    = current_space->array;
1734813a1f2bSShri Abhyankar     bjlvl_ptr[i] = current_space_lvl->array;
1735813a1f2bSShri Abhyankar 
1736813a1f2bSShri Abhyankar     /* make sure the active row i has diagonal entry */
1737813a1f2bSShri Abhyankar     if (*(bj_ptr[i]+bdiag[i]) != i) {
1738813a1f2bSShri Abhyankar       SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,"Row %D has missing diagonal in factored matrix\n\
1739813a1f2bSShri Abhyankar     try running with -pc_factor_nonzeros_along_diagonal or -pc_factor_diagonal_fill",i);
1740813a1f2bSShri Abhyankar     }
1741813a1f2bSShri Abhyankar 
1742813a1f2bSShri Abhyankar     current_space->array           += nzi;
1743813a1f2bSShri Abhyankar     current_space->local_used      += nzi;
1744813a1f2bSShri Abhyankar     current_space->local_remaining -= nzi;
1745813a1f2bSShri Abhyankar     current_space_lvl->array           += nzi;
1746813a1f2bSShri Abhyankar     current_space_lvl->local_used      += nzi;
1747813a1f2bSShri Abhyankar     current_space_lvl->local_remaining -= nzi;
1748813a1f2bSShri Abhyankar   }
1749813a1f2bSShri Abhyankar 
1750813a1f2bSShri Abhyankar   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
1751813a1f2bSShri Abhyankar   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
1752813a1f2bSShri Abhyankar 
1753813a1f2bSShri Abhyankar   /* destroy list of free space and other temporary arrays */
1754813a1f2bSShri Abhyankar   ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
1755813a1f2bSShri Abhyankar 
1756813a1f2bSShri Abhyankar   /* copy free_space into bj and free free_space; set bi, bj, bdiag in new datastructure; */
1757f268cba8SShri Abhyankar   ierr = PetscFreeSpaceContiguous_LU_v2(&free_space,bj,n,bi,bdiag);CHKERRQ(ierr);
1758813a1f2bSShri Abhyankar 
1759813a1f2bSShri Abhyankar   ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
1760813a1f2bSShri Abhyankar   ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
17610e83c824SBarry Smith   ierr = PetscFree2(bj_ptr,bjlvl_ptr);CHKERRQ(ierr);
1762813a1f2bSShri Abhyankar 
1763813a1f2bSShri Abhyankar #if defined(PETSC_USE_INFO)
1764813a1f2bSShri Abhyankar   {
1765813a1f2bSShri Abhyankar     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
1766813a1f2bSShri Abhyankar     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr);
1767813a1f2bSShri Abhyankar     ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
1768813a1f2bSShri Abhyankar     ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr);
1769813a1f2bSShri Abhyankar     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
1770813a1f2bSShri Abhyankar     if (diagonal_fill) {
1771813a1f2bSShri Abhyankar       ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr);
1772813a1f2bSShri Abhyankar     }
1773813a1f2bSShri Abhyankar   }
1774813a1f2bSShri Abhyankar #endif
1775813a1f2bSShri Abhyankar 
1776813a1f2bSShri Abhyankar   /* put together the new matrix */
1777813a1f2bSShri Abhyankar   ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
1778813a1f2bSShri Abhyankar   ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr);
1779813a1f2bSShri Abhyankar   b = (Mat_SeqAIJ*)(fact)->data;
1780813a1f2bSShri Abhyankar   b->free_a       = PETSC_TRUE;
1781813a1f2bSShri Abhyankar   b->free_ij      = PETSC_TRUE;
1782813a1f2bSShri Abhyankar   b->singlemalloc = PETSC_FALSE;
1783f268cba8SShri Abhyankar   ierr = PetscMalloc((bdiag[0]+1)*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);
1784813a1f2bSShri Abhyankar   b->j          = bj;
1785813a1f2bSShri Abhyankar   b->i          = bi;
1786813a1f2bSShri Abhyankar   b->diag       = bdiag;
1787813a1f2bSShri Abhyankar   b->ilen       = 0;
1788813a1f2bSShri Abhyankar   b->imax       = 0;
1789813a1f2bSShri Abhyankar   b->row        = isrow;
1790813a1f2bSShri Abhyankar   b->col        = iscol;
1791813a1f2bSShri Abhyankar   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
1792813a1f2bSShri Abhyankar   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
1793813a1f2bSShri Abhyankar   b->icol       = isicol;
1794813a1f2bSShri Abhyankar   ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
1795813a1f2bSShri Abhyankar   /* In b structure:  Free imax, ilen, old a, old j.
1796813a1f2bSShri Abhyankar      Allocate bdiag, solve_work, new a, new j */
1797f268cba8SShri Abhyankar   ierr = PetscLogObjectMemory(fact,(bdiag[0]+1)*(sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
1798f268cba8SShri Abhyankar   b->maxnz = b->nz = bdiag[0]+1;
1799813a1f2bSShri Abhyankar   (fact)->info.factor_mallocs    = reallocs;
1800813a1f2bSShri Abhyankar   (fact)->info.fill_ratio_given  = f;
1801f268cba8SShri Abhyankar   (fact)->info.fill_ratio_needed = ((PetscReal)(bdiag[0]+1))/((PetscReal)ai[n]);
1802b4f5729fSShri Abhyankar   (fact)->ops->lufactornumeric = MatLUFactorNumeric_SeqAIJ_newdatastruct;
18034108e4d5SBarry Smith   /* ierr = MatILUFactorSymbolic_SeqAIJ_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr); */
1804813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
1805813a1f2bSShri Abhyankar }
1806813a1f2bSShri Abhyankar 
18076516239fSHong Zhang #undef __FUNCT__
18086516239fSHong Zhang #define __FUNCT__ "MatILUFactorSymbolic_SeqAIJ"
18096516239fSHong Zhang PetscErrorCode MatILUFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS isrow,IS iscol,const MatFactorInfo *info)
18106516239fSHong Zhang {
18116516239fSHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data,*b;
18126516239fSHong Zhang   IS                 isicol;
18136516239fSHong Zhang   PetscErrorCode     ierr;
18146516239fSHong Zhang   const PetscInt     *r,*ic;
18156516239fSHong Zhang   PetscInt           n=A->rmap->n,*ai=a->i,*aj=a->j,d;
18166516239fSHong Zhang   PetscInt           *bi,*cols,nnz,*cols_lvl;
18176516239fSHong Zhang   PetscInt           *bdiag,prow,fm,nzbd,reallocs=0,dcount=0;
18186516239fSHong Zhang   PetscInt           i,levels,diagonal_fill;
18196516239fSHong Zhang   PetscTruth         col_identity,row_identity;
18206516239fSHong Zhang   PetscReal          f;
18216516239fSHong Zhang   PetscInt           nlnk,*lnk,*lnk_lvl=PETSC_NULL;
18226516239fSHong Zhang   PetscBT            lnkbt;
18236516239fSHong Zhang   PetscInt           nzi,*bj,**bj_ptr,**bjlvl_ptr;
18246516239fSHong Zhang   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
18256516239fSHong Zhang   PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL;
18266516239fSHong Zhang   PetscTruth         missing;
1827b4f5729fSShri Abhyankar   PetscTruth         newdatastruct=PETSC_FALSE;
18286516239fSHong Zhang 
18296516239fSHong Zhang   PetscFunctionBegin;
18306516239fSHong Zhang   ierr = PetscOptionsGetTruth(PETSC_NULL,"-ilu_new",&newdatastruct,PETSC_NULL);CHKERRQ(ierr);
18316516239fSHong Zhang   if(newdatastruct){
18326516239fSHong Zhang     ierr = MatILUFactorSymbolic_SeqAIJ_newdatastruct(fact,A,isrow,iscol,info);CHKERRQ(ierr);
18336516239fSHong Zhang     PetscFunctionReturn(0);
18346516239fSHong Zhang   }
18356516239fSHong Zhang 
18366516239fSHong 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);
18376516239fSHong Zhang   f             = info->fill;
18386516239fSHong Zhang   levels        = (PetscInt)info->levels;
18396516239fSHong Zhang   diagonal_fill = (PetscInt)info->diagonal_fill;
18406516239fSHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
18416516239fSHong Zhang 
18426516239fSHong Zhang   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
18436516239fSHong Zhang   ierr = ISIdentity(iscol,&col_identity);CHKERRQ(ierr);
18446516239fSHong Zhang   if (!levels && row_identity && col_identity) { /* special case: ilu(0) with natural ordering */
1845f77e22a1SHong Zhang     ierr = MatDuplicateNoCreate_SeqAIJ(fact,A,MAT_DO_NOT_COPY_VALUES,PETSC_TRUE);CHKERRQ(ierr);
1846c9c72f8fSHong Zhang     (fact)->ops->lufactornumeric =  MatLUFactorNumeric_SeqAIJ;
1847c9c72f8fSHong Zhang 
1848719d5645SBarry Smith     fact->factor = MAT_FACTOR_ILU;
1849719d5645SBarry Smith     (fact)->info.factor_mallocs    = 0;
1850719d5645SBarry Smith     (fact)->info.fill_ratio_given  = info->fill;
1851719d5645SBarry Smith     (fact)->info.fill_ratio_needed = 1.0;
1852719d5645SBarry Smith     b               = (Mat_SeqAIJ*)(fact)->data;
18538ef3462fSBarry Smith     ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr);
185409f38230SBarry Smith     if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d);
185508480c60SBarry Smith     b->row              = isrow;
185608480c60SBarry Smith     b->col              = iscol;
185782bf6240SBarry Smith     b->icol             = isicol;
1858719d5645SBarry Smith     ierr                = PetscMalloc(((fact)->rmap->n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
1859c38d4ed2SBarry Smith     ierr                = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
1860c38d4ed2SBarry Smith     ierr                = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
18614108e4d5SBarry Smith     ierr = MatILUFactorSymbolic_SeqAIJ_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr);
18623a40ed3dSBarry Smith     PetscFunctionReturn(0);
186308480c60SBarry Smith   }
186408480c60SBarry Smith 
1865898183ecSLois Curfman McInnes   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
1866898183ecSLois Curfman McInnes   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
18679e25ed09SBarry Smith 
18681bfa06eaSJed Brown   /* get new row and diagonal pointers, must be allocated separately because they will be given to the Mat_SeqAIJ and freed separately */
18691bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
18701bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);
1871a6df321fSShri Abhyankar   bi[0] = bdiag[0] = 0;
18726cf997b0SBarry Smith 
18730e83c824SBarry Smith   ierr = PetscMalloc2(n,PetscInt*,&bj_ptr,n,PetscInt*,&bjlvl_ptr);CHKERRQ(ierr);
18745a8e39fbSHong Zhang 
18755a8e39fbSHong Zhang   /* create a linked list for storing column indices of the active row */
18765a8e39fbSHong Zhang   nlnk = n + 1;
18775a8e39fbSHong Zhang   ierr = PetscIncompleteLLCreate(n,n,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
18785a8e39fbSHong Zhang 
18795a8e39fbSHong Zhang   /* initial FreeSpace size is f*(ai[n]+1) */
1880a1a86e44SBarry Smith   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space);CHKERRQ(ierr);
18815a8e39fbSHong Zhang   current_space = free_space;
1882a1a86e44SBarry Smith   ierr = PetscFreeSpaceGet((PetscInt)(f*(ai[n]+1)),&free_space_lvl);CHKERRQ(ierr);
18835a8e39fbSHong Zhang   current_space_lvl = free_space_lvl;
18845a8e39fbSHong Zhang 
18855a8e39fbSHong Zhang   for (i=0; i<n; i++) {
18865a8e39fbSHong Zhang     nzi = 0;
18876cf997b0SBarry Smith     /* copy current row into linked list */
18885a8e39fbSHong Zhang     nnz  = ai[r[i]+1] - ai[r[i]];
18893b4a8b6dSBarry 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);
18905a8e39fbSHong Zhang     cols = aj + ai[r[i]];
18915a8e39fbSHong Zhang     lnk[i] = -1; /* marker to indicate if diagonal exists */
18925a8e39fbSHong Zhang     ierr = PetscIncompleteLLInit(nnz,cols,n,ic,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
18935a8e39fbSHong Zhang     nzi += nlnk;
18946cf997b0SBarry Smith 
18956cf997b0SBarry Smith     /* make sure diagonal entry is included */
18965a8e39fbSHong Zhang     if (diagonal_fill && lnk[i] == -1) {
18976cf997b0SBarry Smith       fm = n;
18985a8e39fbSHong Zhang       while (lnk[fm] < i) fm = lnk[fm];
18995a8e39fbSHong Zhang       lnk[i]     = lnk[fm]; /* insert diagonal into linked list */
19005a8e39fbSHong Zhang       lnk[fm]    = i;
19015a8e39fbSHong Zhang       lnk_lvl[i] = 0;
19025a8e39fbSHong Zhang       nzi++; dcount++;
19036cf997b0SBarry Smith     }
19046cf997b0SBarry Smith 
19055a8e39fbSHong Zhang     /* add pivot rows into the active row */
19065a8e39fbSHong Zhang     nzbd = 0;
19075a8e39fbSHong Zhang     prow = lnk[n];
19085a8e39fbSHong Zhang     while (prow < i) {
19095a8e39fbSHong Zhang       nnz      = bdiag[prow];
19105a8e39fbSHong Zhang       cols     = bj_ptr[prow] + nnz + 1;
19115a8e39fbSHong Zhang       cols_lvl = bjlvl_ptr[prow] + nnz + 1;
19125a8e39fbSHong Zhang       nnz      = bi[prow+1] - bi[prow] - nnz - 1;
19135a8e39fbSHong Zhang       ierr = PetscILULLAddSorted(nnz,cols,levels,cols_lvl,prow,nlnk,lnk,lnk_lvl,lnkbt,prow);CHKERRQ(ierr);
19145a8e39fbSHong Zhang       nzi += nlnk;
19155a8e39fbSHong Zhang       prow = lnk[prow];
19165a8e39fbSHong Zhang       nzbd++;
191756beaf04SBarry Smith     }
19185a8e39fbSHong Zhang     bdiag[i] = nzbd;
19195a8e39fbSHong Zhang     bi[i+1]  = bi[i] + nzi;
1920ecf371e4SBarry Smith 
19215a8e39fbSHong Zhang     /* if free space is not available, make more free space */
19225a8e39fbSHong Zhang     if (current_space->local_remaining<nzi) {
19235a8e39fbSHong Zhang       nnz = nzi*(n - i); /* estimated and max additional space needed */
1924a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(nnz,&current_space);CHKERRQ(ierr);
1925a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(nnz,&current_space_lvl);CHKERRQ(ierr);
19265a8e39fbSHong Zhang       reallocs++;
19275a8e39fbSHong Zhang     }
1928ecf371e4SBarry Smith 
19295a8e39fbSHong Zhang     /* copy data into free_space and free_space_lvl, then initialize lnk */
19305a8e39fbSHong Zhang     ierr = PetscIncompleteLLClean(n,n,nzi,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
19315a8e39fbSHong Zhang     bj_ptr[i]    = current_space->array;
19325a8e39fbSHong Zhang     bjlvl_ptr[i] = current_space_lvl->array;
19335a8e39fbSHong Zhang 
19345a8e39fbSHong Zhang     /* make sure the active row i has diagonal entry */
19355a8e39fbSHong Zhang     if (*(bj_ptr[i]+bdiag[i]) != i) {
193677431f27SBarry Smith       SETERRQ1(PETSC_ERR_MAT_LU_ZRPVT,"Row %D has missing diagonal in factored matrix\n\
1937d7ee0231SBarry Smith     try running with -pc_factor_nonzeros_along_diagonal or -pc_factor_diagonal_fill",i);
19386cf997b0SBarry Smith     }
19395a8e39fbSHong Zhang 
19405a8e39fbSHong Zhang     current_space->array           += nzi;
19415a8e39fbSHong Zhang     current_space->local_used      += nzi;
19425a8e39fbSHong Zhang     current_space->local_remaining -= nzi;
19435a8e39fbSHong Zhang     current_space_lvl->array           += nzi;
19445a8e39fbSHong Zhang     current_space_lvl->local_used      += nzi;
19455a8e39fbSHong Zhang     current_space_lvl->local_remaining -= nzi;
19469e25ed09SBarry Smith   }
19475a8e39fbSHong Zhang 
1948898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
1949898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
19505a8e39fbSHong Zhang 
19515a8e39fbSHong Zhang   /* destroy list of free space and other temporary arrays */
19525a8e39fbSHong Zhang   ierr = PetscMalloc((bi[n]+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
19536516239fSHong Zhang   ierr = PetscFreeSpaceContiguous(&free_space,bj);CHKERRQ(ierr); /* copy free_space -> bj */
19545a8e39fbSHong Zhang   ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
1955a1a86e44SBarry Smith   ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
19560e83c824SBarry Smith   ierr = PetscFree2(bj_ptr,bjlvl_ptr);CHKERRQ(ierr);
19579e25ed09SBarry Smith 
19586cf91177SBarry Smith #if defined(PETSC_USE_INFO)
1959f64065bbSBarry Smith   {
19605a8e39fbSHong Zhang     PetscReal af = ((PetscReal)bi[n])/((PetscReal)ai[n]);
1961ae15b995SBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,f,af);CHKERRQ(ierr);
1962ae15b995SBarry Smith     ierr = PetscInfo1(A,"Run with -[sub_]pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
1963ae15b995SBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill([sub]pc,%G);\n",af);CHKERRQ(ierr);
1964ae15b995SBarry Smith     ierr = PetscInfo(A,"for best performance.\n");CHKERRQ(ierr);
1965335d9088SBarry Smith     if (diagonal_fill) {
1966ae15b995SBarry Smith       ierr = PetscInfo1(A,"Detected and replaced %D missing diagonals",dcount);CHKERRQ(ierr);
1967335d9088SBarry Smith     }
1968f64065bbSBarry Smith   }
196963ba0a88SBarry Smith #endif
1970bbb6d6a8SBarry Smith 
19719e25ed09SBarry Smith   /* put together the new matrix */
197207dbf95fSBarry Smith   ierr = MatSeqAIJSetPreallocation_SeqAIJ(fact,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
1973719d5645SBarry Smith   ierr = PetscLogObjectParent(fact,isicol);CHKERRQ(ierr);
1974719d5645SBarry Smith   b = (Mat_SeqAIJ*)(fact)->data;
1975e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
1976e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
19777c922b88SBarry Smith   b->singlemalloc = PETSC_FALSE;
19780e83c824SBarry Smith   ierr = PetscMalloc(bi[n]*sizeof(PetscScalar),&b->a);CHKERRQ(ierr);
19795a8e39fbSHong Zhang   b->j          = bj;
19805a8e39fbSHong Zhang   b->i          = bi;
19815a8e39fbSHong Zhang   for (i=0; i<n; i++) bdiag[i] += bi[i];
19825a8e39fbSHong Zhang   b->diag       = bdiag;
1983416022c9SBarry Smith   b->ilen       = 0;
1984416022c9SBarry Smith   b->imax       = 0;
1985416022c9SBarry Smith   b->row        = isrow;
1986416022c9SBarry Smith   b->col        = iscol;
1987c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
1988c38d4ed2SBarry Smith   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
198982bf6240SBarry Smith   b->icol       = isicol;
199087828ca2SBarry Smith   ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
1991416022c9SBarry Smith   /* In b structure:  Free imax, ilen, old a, old j.
19925a8e39fbSHong Zhang      Allocate bdiag, solve_work, new a, new j */
1993719d5645SBarry Smith   ierr = PetscLogObjectMemory(fact,(bi[n]-n) * (sizeof(PetscInt)+sizeof(PetscScalar)));CHKERRQ(ierr);
19945a8e39fbSHong Zhang   b->maxnz             = b->nz = bi[n] ;
1995719d5645SBarry Smith   (fact)->info.factor_mallocs    = reallocs;
1996719d5645SBarry Smith   (fact)->info.fill_ratio_given  = f;
1997719d5645SBarry Smith   (fact)->info.fill_ratio_needed = ((PetscReal)bi[n])/((PetscReal)ai[n]);
1998719d5645SBarry Smith   (fact)->ops->lufactornumeric =  MatLUFactorNumeric_SeqAIJ;
19994108e4d5SBarry Smith   ierr = MatILUFactorSymbolic_SeqAIJ_Inode(fact,A,isrow,iscol,info);CHKERRQ(ierr);
20003a40ed3dSBarry Smith   PetscFunctionReturn(0);
20019e25ed09SBarry Smith }
2002d5d45c9bSBarry Smith 
2003a6175056SHong Zhang #undef __FUNCT__
20045f44c854SHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ_newdatastruct"
20055f44c854SHong Zhang PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ_newdatastruct(Mat B,Mat A,const MatFactorInfo *info)
20065f44c854SHong Zhang {
20075f44c854SHong Zhang   Mat            C = B;
20085f44c854SHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
20095f44c854SHong Zhang   Mat_SeqSBAIJ   *b=(Mat_SeqSBAIJ*)C->data;
20105f44c854SHong Zhang   IS             ip=b->row,iip = b->icol;
20115f44c854SHong Zhang   PetscErrorCode ierr;
20125f44c854SHong Zhang   const PetscInt *rip,*riip;
20135f44c854SHong Zhang   PetscInt       i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bdiag=b->diag,*bjtmp;
20145f44c854SHong Zhang   PetscInt       *ai=a->i,*aj=a->j;
20155f44c854SHong Zhang   PetscInt       k,jmin,jmax,*c2r,*il,col,nexti,ili,nz;
20165f44c854SHong Zhang   MatScalar      *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi;
20175f44c854SHong Zhang   PetscTruth     perm_identity;
20185f44c854SHong Zhang 
201998a93161SHong Zhang   LUShift_Ctx    sctx;
202098a93161SHong Zhang   PetscInt       newshift;
202198a93161SHong Zhang   PetscReal      rs;
202298a93161SHong Zhang   MatScalar      d,*v;
202398a93161SHong Zhang 
20245f44c854SHong Zhang   PetscFunctionBegin;
202598a93161SHong Zhang   /* MatPivotSetUp(): initialize shift context sctx */
202698a93161SHong Zhang   ierr = PetscMemzero(&sctx,sizeof(LUShift_Ctx));CHKERRQ(ierr);
202798a93161SHong Zhang 
202898a93161SHong Zhang   /* if both shift schemes are chosen by user, only use info->shiftpd */
202998a93161SHong Zhang   if (info->shiftpd) { /* set sctx.shift_top=max{rs} */
203098a93161SHong Zhang     sctx.shift_top = info->zeropivot;
203198a93161SHong Zhang     for (i=0; i<mbs; i++) {
203298a93161SHong Zhang       /* calculate sum(|aij|)-RealPart(aii), amt of shift needed for this row */
203398a93161SHong Zhang       d  = (aa)[a->diag[i]];
203498a93161SHong Zhang       rs = -PetscAbsScalar(d) - PetscRealPart(d);
203598a93161SHong Zhang       v  = aa+ai[i];
203698a93161SHong Zhang       nz = ai[i+1] - ai[i];
203798a93161SHong Zhang       for (j=0; j<nz; j++)
203898a93161SHong Zhang 	rs += PetscAbsScalar(v[j]);
203998a93161SHong Zhang       if (rs>sctx.shift_top) sctx.shift_top = rs;
204098a93161SHong Zhang     }
204198a93161SHong Zhang     sctx.shift_top   *= 1.1;
204298a93161SHong Zhang     sctx.nshift_max   = 5;
204398a93161SHong Zhang     sctx.shift_lo     = 0.;
204498a93161SHong Zhang     sctx.shift_hi     = 1.;
204598a93161SHong Zhang   }
20465f44c854SHong Zhang 
20475f44c854SHong Zhang   ierr  = ISGetIndices(ip,&rip);CHKERRQ(ierr);
20485f44c854SHong Zhang   ierr  = ISGetIndices(iip,&riip);CHKERRQ(ierr);
20495f44c854SHong Zhang 
20505f44c854SHong Zhang   /* allocate working arrays
20515f44c854SHong Zhang      c2r: linked list, keep track of pivot rows for a given column. c2r[col]: head of the list for a given col
20525f44c854SHong 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
20535f44c854SHong Zhang   */
20540e83c824SBarry Smith   ierr = PetscMalloc3(mbs,MatScalar,&rtmp,mbs,PetscInt,&il,mbs,PetscInt,&c2r);CHKERRQ(ierr);
20555f44c854SHong Zhang 
205698a93161SHong Zhang   do {
205798a93161SHong Zhang     sctx.lushift = PETSC_FALSE;
205898a93161SHong Zhang 
2059c5546cabSHong Zhang     for (i=0; i<mbs; i++) c2r[i] = mbs;
20605f44c854SHong Zhang     il[0] = 0;
20615f44c854SHong Zhang 
20625f44c854SHong Zhang     for (k = 0; k<mbs; k++){
20635f44c854SHong Zhang       /* zero rtmp */
20645f44c854SHong Zhang       nz = bi[k+1] - bi[k];
20655f44c854SHong Zhang       bjtmp = bj + bi[k];
20665f44c854SHong Zhang       for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
20675f44c854SHong Zhang 
20685f44c854SHong Zhang       /* load in initial unfactored row */
20695f44c854SHong Zhang       bval = ba + bi[k];
20705f44c854SHong Zhang       jmin = ai[rip[k]]; jmax = ai[rip[k]+1];
20715f44c854SHong Zhang       for (j = jmin; j < jmax; j++){
20725f44c854SHong Zhang         col = riip[aj[j]];
20735f44c854SHong Zhang         if (col >= k){ /* only take upper triangular entry */
20745f44c854SHong Zhang           rtmp[col] = aa[j];
20755f44c854SHong Zhang           *bval++   = 0.0; /* for in-place factorization */
20765f44c854SHong Zhang         }
20775f44c854SHong Zhang       }
207898a93161SHong Zhang       /* shift the diagonal of the matrix: ZeropivotApply() */
207998a93161SHong Zhang       rtmp[k] += sctx.shift_amount;  /* shift the diagonal of the matrix */
20805f44c854SHong Zhang 
20815f44c854SHong Zhang       /* modify k-th row by adding in those rows i with U(i,k)!=0 */
20825f44c854SHong Zhang       dk = rtmp[k];
20835f44c854SHong Zhang       i  = c2r[k]; /* first row to be added to k_th row  */
20845f44c854SHong Zhang 
20855f44c854SHong Zhang       while (i < k){
20865f44c854SHong Zhang         nexti = c2r[i]; /* next row to be added to k_th row */
20875f44c854SHong Zhang 
20885f44c854SHong Zhang         /* compute multiplier, update diag(k) and U(i,k) */
20895f44c854SHong Zhang         ili   = il[i];  /* index of first nonzero element in U(i,k:bms-1) */
20905f44c854SHong Zhang         uikdi = - ba[ili]*ba[bdiag[i]];  /* diagonal(k) */
20915f44c854SHong Zhang         dk   += uikdi*ba[ili]; /* update diag[k] */
20925f44c854SHong Zhang         ba[ili] = uikdi; /* -U(i,k) */
20935f44c854SHong Zhang 
20945f44c854SHong Zhang         /* add multiple of row i to k-th row */
20955f44c854SHong Zhang         jmin = ili + 1; jmax = bi[i+1];
20965f44c854SHong Zhang         if (jmin < jmax){
20975f44c854SHong Zhang           for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j];
20985f44c854SHong Zhang           /* update il and c2r for row i */
20995f44c854SHong Zhang           il[i] = jmin;
21005f44c854SHong Zhang           j = bj[jmin]; c2r[i] = c2r[j]; c2r[j] = i;
21015f44c854SHong Zhang         }
21025f44c854SHong Zhang         i = nexti;
21035f44c854SHong Zhang       }
21045f44c854SHong Zhang 
21055f44c854SHong Zhang       /* copy data into U(k,:) */
210698a93161SHong Zhang       rs   = 0.0;
21075f44c854SHong Zhang       jmin = bi[k]; jmax = bi[k+1]-1;
21085f44c854SHong Zhang       if (jmin < jmax) {
21095f44c854SHong Zhang         for (j=jmin; j<jmax; j++){
211098a93161SHong Zhang           col = bj[j]; ba[j] = rtmp[col]; rs += PetscAbsScalar(ba[j]);
21115f44c854SHong Zhang         }
21125f44c854SHong Zhang         /* add the k-th row into il and c2r */
21135f44c854SHong Zhang         il[k] = jmin;
21145f44c854SHong Zhang         i = bj[jmin]; c2r[k] = c2r[i]; c2r[i] = k;
21155f44c854SHong Zhang       }
211698a93161SHong Zhang 
211798a93161SHong Zhang       /* MatPivotCheck() */
211898a93161SHong Zhang       sctx.rs  = rs;
211998a93161SHong Zhang       sctx.pv  = dk;
212098a93161SHong Zhang       if (info->shiftnz){
212198a93161SHong Zhang         ierr = MatPivotCheck_nz(info,sctx,k,newshift);CHKERRQ(ierr);
212298a93161SHong Zhang       } else if (info->shiftpd){
212398a93161SHong Zhang         ierr = MatPivotCheck_pd(info,sctx,k,newshift);CHKERRQ(ierr);
212498a93161SHong Zhang       } else if (info->shiftinblocks){
212598a93161SHong Zhang         ierr = MatPivotCheck_inblocks(info,sctx,k,newshift);CHKERRQ(ierr);
212698a93161SHong Zhang       } else {
212798a93161SHong Zhang         ierr = MatPivotCheck_none(info,sctx,k,newshift);CHKERRQ(ierr);
21285f44c854SHong Zhang       }
212998a93161SHong Zhang       dk = sctx.pv;
213098a93161SHong Zhang       if (newshift == 1) break;
213198a93161SHong Zhang 
213298a93161SHong Zhang       ba[bdiag[k]] = 1.0/dk; /* U(k,k) */
213398a93161SHong Zhang     }
213498a93161SHong Zhang   } while (sctx.lushift);
21355f44c854SHong Zhang 
21360e83c824SBarry Smith   ierr = PetscFree3(rtmp,il,c2r);CHKERRQ(ierr);
21375f44c854SHong Zhang   ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr);
21385f44c854SHong Zhang   ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr);
21395f44c854SHong Zhang 
21405f44c854SHong Zhang   ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr);
21415f44c854SHong Zhang   if (perm_identity){
21425f44c854SHong Zhang     (B)->ops->solve           = MatSolve_SeqSBAIJ_1_NaturalOrdering_newdatastruct;
21435f44c854SHong Zhang     (B)->ops->solvetranspose  = 0;
21445f44c854SHong Zhang     (B)->ops->forwardsolve    = 0;
21455f44c854SHong Zhang     (B)->ops->backwardsolve   = 0;
21465f44c854SHong Zhang   } else {
2147783ef271SHong Zhang     (B)->ops->solve           = MatSolve_SeqSBAIJ_1_newdatastruct;
21485f44c854SHong Zhang     (B)->ops->solvetranspose  = 0;
21495f44c854SHong Zhang     (B)->ops->forwardsolve    = 0;
21505f44c854SHong Zhang     (B)->ops->backwardsolve   = 0;
21515f44c854SHong Zhang   }
21525f44c854SHong Zhang 
21535f44c854SHong Zhang   C->assembled    = PETSC_TRUE;
21545f44c854SHong Zhang   C->preallocated = PETSC_TRUE;
21555f44c854SHong Zhang   ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr);
215698a93161SHong Zhang 
215798a93161SHong Zhang   /* MatPivotView() */
215898a93161SHong Zhang   if (sctx.nshift){
215998a93161SHong Zhang     if (info->shiftpd) {
216098a93161SHong Zhang       ierr = PetscInfo4(A,"number of shift_pd tries %D, shift_amount %G, diagonal shifted up by %e fraction top_value %e\n",sctx.nshift,sctx.shift_amount,sctx.shift_fraction,sctx.shift_top);CHKERRQ(ierr);
216198a93161SHong Zhang     } else if (info->shiftnz) {
216298a93161SHong Zhang       ierr = PetscInfo2(A,"number of shift_nz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
216398a93161SHong Zhang     } else if (info->shiftinblocks){
216498a93161SHong Zhang       ierr = PetscInfo2(A,"number of shift_inblocks applied %D, each shift_amount %G\n",sctx.nshift,info->shiftinblocks);CHKERRQ(ierr);
216598a93161SHong Zhang     }
216698a93161SHong Zhang   }
21675f44c854SHong Zhang   PetscFunctionReturn(0);
21685f44c854SHong Zhang }
21695f44c854SHong Zhang 
21705f44c854SHong Zhang #undef __FUNCT__
2171a6175056SHong Zhang #define __FUNCT__ "MatCholeskyFactorNumeric_SeqAIJ"
21720481f469SBarry Smith PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ(Mat B,Mat A,const MatFactorInfo *info)
2173a6175056SHong Zhang {
2174719d5645SBarry Smith   Mat            C = B;
21750968510aSHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data;
21762ed133dbSHong Zhang   Mat_SeqSBAIJ   *b=(Mat_SeqSBAIJ*)C->data;
21779bfd6278SHong Zhang   IS             ip=b->row,iip = b->icol;
21782ed133dbSHong Zhang   PetscErrorCode ierr;
21795d0c19d7SBarry Smith   const PetscInt *rip,*riip;
2180c5546cabSHong Zhang   PetscInt       i,j,mbs=A->rmap->n,*bi=b->i,*bj=b->j,*bcol,*bjtmp;
21812ed133dbSHong Zhang   PetscInt       *ai=a->i,*aj=a->j;
21822ed133dbSHong Zhang   PetscInt       k,jmin,jmax,*jl,*il,col,nexti,ili,nz;
2183622e2028SHong Zhang   MatScalar      *rtmp,*ba=b->a,*bval,*aa=a->a,dk,uikdi;
2184540703acSHong Zhang   PetscReal      zeropivot,rs,shiftnz;
2185fbf22428SSatish Balay   PetscReal      shiftpd;
2186540703acSHong Zhang   ChShift_Ctx    sctx;
2187540703acSHong Zhang   PetscInt       newshift;
2188db4efbfdSBarry Smith   PetscTruth     perm_identity;
2189d5d45c9bSBarry Smith 
2190a6175056SHong Zhang   PetscFunctionBegin;
2191540703acSHong Zhang   shiftnz   = info->shiftnz;
2192540703acSHong Zhang   shiftpd   = info->shiftpd;
2193ee45ca4aSHong Zhang   zeropivot = info->zeropivot;
2194ee45ca4aSHong Zhang 
21952ed133dbSHong Zhang   ierr  = ISGetIndices(ip,&rip);CHKERRQ(ierr);
21969bfd6278SHong Zhang   ierr  = ISGetIndices(iip,&riip);CHKERRQ(ierr);
21972ed133dbSHong Zhang 
21982ed133dbSHong Zhang   /* initialization */
21990e83c824SBarry Smith   ierr = PetscMalloc3(mbs,MatScalar,&rtmp,mbs,PetscInt,&il,mbs,PetscInt,&jl);CHKERRQ(ierr);
2200540703acSHong Zhang   sctx.shift_amount = 0;
2201540703acSHong Zhang   sctx.nshift       = 0;
22022ed133dbSHong Zhang   do {
2203540703acSHong Zhang     sctx.chshift = PETSC_FALSE;
2204c5546cabSHong Zhang     for (i=0; i<mbs; i++) jl[i] = mbs;
2205c5546cabSHong Zhang     il[0] = 0;
22062ed133dbSHong Zhang 
22072ed133dbSHong Zhang     for (k = 0; k<mbs; k++){
2208c5546cabSHong Zhang       /* zero rtmp */
2209c5546cabSHong Zhang       nz = bi[k+1] - bi[k];
2210c5546cabSHong Zhang       bjtmp = bj + bi[k];
2211c5546cabSHong Zhang       for (j=0; j<nz; j++) rtmp[bjtmp[j]] = 0.0;
2212c5546cabSHong Zhang 
2213c05c3958SHong Zhang       bval = ba + bi[k];
22142ed133dbSHong Zhang       /* initialize k-th row by the perm[k]-th row of A */
22152ed133dbSHong Zhang       jmin = ai[rip[k]]; jmax = ai[rip[k]+1];
22162ed133dbSHong Zhang       for (j = jmin; j < jmax; j++){
22179bfd6278SHong Zhang         col = riip[aj[j]];
22182ed133dbSHong Zhang         if (col >= k){ /* only take upper triangular entry */
22192ed133dbSHong Zhang           rtmp[col] = aa[j];
2220c05c3958SHong Zhang           *bval++  = 0.0; /* for in-place factorization */
22212ed133dbSHong Zhang         }
22222ed133dbSHong Zhang       }
222339e3d630SHong Zhang       /* shift the diagonal of the matrix */
2224540703acSHong Zhang       if (sctx.nshift) rtmp[k] += sctx.shift_amount;
22252ed133dbSHong Zhang 
22262ed133dbSHong Zhang       /* modify k-th row by adding in those rows i with U(i,k)!=0 */
22272ed133dbSHong Zhang       dk = rtmp[k];
22282ed133dbSHong Zhang       i = jl[k]; /* first row to be added to k_th row  */
22292ed133dbSHong Zhang 
22302ed133dbSHong Zhang       while (i < k){
22312ed133dbSHong Zhang         nexti = jl[i]; /* next row to be added to k_th row */
22322ed133dbSHong Zhang 
22332ed133dbSHong Zhang         /* compute multiplier, update diag(k) and U(i,k) */
22342ed133dbSHong Zhang         ili = il[i];  /* index of first nonzero element in U(i,k:bms-1) */
22352ed133dbSHong Zhang         uikdi = - ba[ili]*ba[bi[i]];  /* diagonal(k) */
22362ed133dbSHong Zhang         dk += uikdi*ba[ili];
22372ed133dbSHong Zhang         ba[ili] = uikdi; /* -U(i,k) */
22382ed133dbSHong Zhang 
22392ed133dbSHong Zhang         /* add multiple of row i to k-th row */
22402ed133dbSHong Zhang         jmin = ili + 1; jmax = bi[i+1];
22412ed133dbSHong Zhang         if (jmin < jmax){
22422ed133dbSHong Zhang           for (j=jmin; j<jmax; j++) rtmp[bj[j]] += uikdi*ba[j];
22432ed133dbSHong Zhang           /* update il and jl for row i */
22442ed133dbSHong Zhang           il[i] = jmin;
22452ed133dbSHong Zhang           j = bj[jmin]; jl[i] = jl[j]; jl[j] = i;
22462ed133dbSHong Zhang         }
22472ed133dbSHong Zhang         i = nexti;
22482ed133dbSHong Zhang       }
22492ed133dbSHong Zhang 
22500697b6caSHong Zhang       /* shift the diagonals when zero pivot is detected */
22510697b6caSHong Zhang       /* compute rs=sum of abs(off-diagonal) */
22520697b6caSHong Zhang       rs   = 0.0;
22533c9e8598SHong Zhang       jmin = bi[k]+1;
22543c9e8598SHong Zhang       nz   = bi[k+1] - jmin;
22553c9e8598SHong Zhang       bcol = bj + jmin;
225604fbf559SBarry Smith       for (j=0; j<nz; j++) {
225704fbf559SBarry Smith         rs += PetscAbsScalar(rtmp[bcol[j]]);
22583c9e8598SHong Zhang       }
2259540703acSHong Zhang 
2260540703acSHong Zhang       sctx.rs = rs;
2261540703acSHong Zhang       sctx.pv = dk;
22625b5bc046SBarry Smith       ierr = MatCholeskyCheckShift_inline(info,sctx,k,newshift);CHKERRQ(ierr);
2263117f1891Stmunson 
2264117f1891Stmunson       if (newshift == 1) {
2265117f1891Stmunson         if (!sctx.shift_amount) {
2266117f1891Stmunson           sctx.shift_amount = 1e-5;
2267117f1891Stmunson         }
2268117f1891Stmunson         break;
2269117f1891Stmunson       }
22703c9e8598SHong Zhang 
22713c9e8598SHong Zhang       /* copy data into U(k,:) */
227239e3d630SHong Zhang       ba[bi[k]] = 1.0/dk; /* U(k,k) */
227339e3d630SHong Zhang       jmin = bi[k]+1; jmax = bi[k+1];
227439e3d630SHong Zhang       if (jmin < jmax) {
227539e3d630SHong Zhang         for (j=jmin; j<jmax; j++){
2276c6ae9bfcSHong Zhang           col = bj[j]; ba[j] = rtmp[col];
22773c9e8598SHong Zhang         }
227839e3d630SHong Zhang         /* add the k-th row into il and jl */
22793c9e8598SHong Zhang         il[k] = jmin;
22803c9e8598SHong Zhang         i = bj[jmin]; jl[k] = jl[i]; jl[i] = k;
22813c9e8598SHong Zhang       }
22823c9e8598SHong Zhang     }
2283540703acSHong Zhang   } while (sctx.chshift);
22840e83c824SBarry Smith   ierr = PetscFree3(rtmp,il,jl);CHKERRQ(ierr);
228539e3d630SHong Zhang   ierr = ISRestoreIndices(ip,&rip);CHKERRQ(ierr);
22869bfd6278SHong Zhang   ierr = ISRestoreIndices(iip,&riip);CHKERRQ(ierr);
2287db4efbfdSBarry Smith 
2288db4efbfdSBarry Smith   ierr = ISIdentity(ip,&perm_identity);CHKERRQ(ierr);
2289db4efbfdSBarry Smith   if (perm_identity){
2290719d5645SBarry Smith     (B)->ops->solve           = MatSolve_SeqSBAIJ_1_NaturalOrdering;
2291719d5645SBarry Smith     (B)->ops->solvetranspose  = MatSolve_SeqSBAIJ_1_NaturalOrdering;
2292719d5645SBarry Smith     (B)->ops->forwardsolve    = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering;
2293719d5645SBarry Smith     (B)->ops->backwardsolve   = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering;
2294db4efbfdSBarry Smith   } else {
2295719d5645SBarry Smith     (B)->ops->solve           = MatSolve_SeqSBAIJ_1;
2296719d5645SBarry Smith     (B)->ops->solvetranspose  = MatSolve_SeqSBAIJ_1;
2297719d5645SBarry Smith     (B)->ops->forwardsolve    = MatForwardSolve_SeqSBAIJ_1;
2298719d5645SBarry Smith     (B)->ops->backwardsolve   = MatBackwardSolve_SeqSBAIJ_1;
2299db4efbfdSBarry Smith   }
2300db4efbfdSBarry Smith 
23013c9e8598SHong Zhang   C->assembled    = PETSC_TRUE;
23023c9e8598SHong Zhang   C->preallocated = PETSC_TRUE;
2303d0f46423SBarry Smith   ierr = PetscLogFlops(C->rmap->n);CHKERRQ(ierr);
2304540703acSHong Zhang   if (sctx.nshift){
2305540703acSHong Zhang     if (shiftnz) {
23061e2582c4SBarry Smith       ierr = PetscInfo2(A,"number of shiftnz tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
2307540703acSHong Zhang     } else if (shiftpd) {
23081e2582c4SBarry Smith       ierr = PetscInfo2(A,"number of shiftpd tries %D, shift_amount %G\n",sctx.nshift,sctx.shift_amount);CHKERRQ(ierr);
23090697b6caSHong Zhang     }
23103c9e8598SHong Zhang   }
23113c9e8598SHong Zhang   PetscFunctionReturn(0);
23123c9e8598SHong Zhang }
2313a6175056SHong Zhang 
23148a76ed4fSHong Zhang /*
23158a76ed4fSHong Zhang    icc() under revised new data structure.
23168a76ed4fSHong Zhang    Factored arrays bj and ba are stored as
23178a76ed4fSHong Zhang      U(0,:),...,U(i,:),U(n-1,:)
23188a76ed4fSHong Zhang 
23198a76ed4fSHong Zhang    ui=fact->i is an array of size n+1, in which
23208a76ed4fSHong Zhang    ui+
23218a76ed4fSHong Zhang      ui[i]:  points to 1st entry of U(i,:),i=0,...,n-1
23228a76ed4fSHong Zhang      ui[n]:  points to U(n-1,n-1)+1
23238a76ed4fSHong Zhang 
23248a76ed4fSHong Zhang   udiag=fact->diag is an array of size n,in which
23258a76ed4fSHong Zhang      udiag[i]: points to diagonal of U(i,:), i=0,...,n-1
23268a76ed4fSHong Zhang 
23278a76ed4fSHong Zhang    U(i,:) contains udiag[i] as its last entry, i.e.,
23288a76ed4fSHong Zhang     U(i,:) = (u[i,i+1],...,u[i,n-1],diag[i])
23298a76ed4fSHong Zhang */
23308a76ed4fSHong Zhang 
2331a6175056SHong Zhang #undef __FUNCT__
23325f44c854SHong Zhang #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ_newdatastruct"
23335f44c854SHong Zhang PetscErrorCode MatICCFactorSymbolic_SeqAIJ_newdatastruct(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
2334a6175056SHong Zhang {
23350968510aSHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
2336ed59401aSHong Zhang   Mat_SeqSBAIJ       *b;
2337dfbe8321SBarry Smith   PetscErrorCode     ierr;
233858ebbce7SBarry Smith   PetscTruth         perm_identity,missing;
23395f44c854SHong Zhang   PetscInt           reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag;
23405d0c19d7SBarry Smith   const PetscInt     *rip,*riip;
2341ed59401aSHong Zhang   PetscInt           jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow;
234258ebbce7SBarry Smith   PetscInt           nlnk,*lnk,*lnk_lvl=PETSC_NULL,d;
23435a8e39fbSHong Zhang   PetscInt           ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr;
2344ed59401aSHong Zhang   PetscReal          fill=info->fill,levels=info->levels;
2345a1a86e44SBarry Smith   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
2346a1a86e44SBarry Smith   PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL;
23477a48dd6fSHong Zhang   PetscBT            lnkbt;
2348b635d36bSHong Zhang   IS                 iperm;
2349a6175056SHong Zhang 
2350a6175056SHong Zhang   PetscFunctionBegin;
2351d0f46423SBarry 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);
235258ebbce7SBarry Smith   ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr);
235358ebbce7SBarry Smith   if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d);
2354653a6975SHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
2355b635d36bSHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
23567a48dd6fSHong Zhang 
235739e3d630SHong Zhang   ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr);
23585f44c854SHong Zhang   ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr);
235939e3d630SHong Zhang   ui[0] = 0;
236039e3d630SHong Zhang 
23618a76ed4fSHong Zhang   /* ICC(0) without matrix ordering: simply rearrange column indices */
2362622e2028SHong Zhang   if (!levels && perm_identity) {
23635f44c854SHong Zhang     ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
23645f44c854SHong Zhang     cols = uj;
23655f44c854SHong Zhang     for (i=0; i<am; i++) {
2366c5546cabSHong Zhang       ncols    = ai[i+1] - a->diag[i];
2367c5546cabSHong Zhang       ui[i+1]  = ui[i] + ncols;
2368c5546cabSHong Zhang       udiag[i] = ui[i+1] - 1; /* points to the last entry of U(i,:) */
2369c5546cabSHong Zhang 
23705f44c854SHong Zhang       aj   = a->j + a->diag[i] + 1; /* 1st entry of U(i,:) without diagonal */
2371c5546cabSHong Zhang       ncols--; /* exclude diagonal */
23725f44c854SHong Zhang       for (j=0; j<ncols; j++) *cols++ = aj[j];
23735f44c854SHong Zhang       *cols++ = i; /* diagoanl is located as the last entry of U(i,:) */
23745f44c854SHong Zhang     }
23755f44c854SHong Zhang   } else { /* case: levels>0 || (levels=0 && !perm_identity) */
23765f44c854SHong Zhang     ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
23775f44c854SHong Zhang     ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
23785f44c854SHong Zhang 
23795f44c854SHong Zhang     /* initialization */
23805f44c854SHong Zhang     ierr  = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr);
23815f44c854SHong Zhang 
23825f44c854SHong Zhang     /* jl: linked list for storing indices of the pivot rows
23835f44c854SHong Zhang        il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
23840e83c824SBarry Smith     ierr = PetscMalloc4(am,PetscInt*,&uj_ptr,am,PetscInt*,&uj_lvl_ptr,am,PetscInt,&jl,am,PetscInt,&il);CHKERRQ(ierr);
23855f44c854SHong Zhang     for (i=0; i<am; i++){
23865f44c854SHong Zhang       jl[i] = am; il[i] = 0;
23875f44c854SHong Zhang     }
23885f44c854SHong Zhang 
23895f44c854SHong Zhang     /* create and initialize a linked list for storing column indices of the active row k */
23905f44c854SHong Zhang     nlnk = am + 1;
23915f44c854SHong Zhang     ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
23925f44c854SHong Zhang 
23935f44c854SHong Zhang     /* initial FreeSpace size is fill*(ai[am]+1) */
23945f44c854SHong Zhang     ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr);
23955f44c854SHong Zhang     current_space = free_space;
23965f44c854SHong Zhang     ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space_lvl);CHKERRQ(ierr);
23975f44c854SHong Zhang     current_space_lvl = free_space_lvl;
23985f44c854SHong Zhang 
23995f44c854SHong Zhang     for (k=0; k<am; k++){  /* for each active row k */
24005f44c854SHong Zhang       /* initialize lnk by the column indices of row rip[k] of A */
24015f44c854SHong Zhang       nzk   = 0;
24025f44c854SHong Zhang       ncols = ai[rip[k]+1] - ai[rip[k]];
24035f44c854SHong 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);
24045f44c854SHong Zhang       ncols_upper = 0;
24055f44c854SHong Zhang       for (j=0; j<ncols; j++){
24065f44c854SHong Zhang         i = *(aj + ai[rip[k]] + j); /* unpermuted column index */
24075f44c854SHong Zhang         if (riip[i] >= k){ /* only take upper triangular entry */
24085f44c854SHong Zhang           ajtmp[ncols_upper] = i;
24095f44c854SHong Zhang           ncols_upper++;
24105f44c854SHong Zhang         }
24115f44c854SHong Zhang       }
24125f44c854SHong Zhang       ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
24135f44c854SHong Zhang       nzk += nlnk;
24145f44c854SHong Zhang 
24155f44c854SHong Zhang       /* update lnk by computing fill-in for each pivot row to be merged in */
24165f44c854SHong Zhang       prow = jl[k]; /* 1st pivot row */
24175f44c854SHong Zhang 
24185f44c854SHong Zhang       while (prow < k){
24195f44c854SHong Zhang         nextprow = jl[prow];
24205f44c854SHong Zhang 
24215f44c854SHong Zhang         /* merge prow into k-th row */
24225f44c854SHong Zhang         jmin = il[prow] + 1;  /* index of the 2nd nzero entry in U(prow,k:am-1) */
24235f44c854SHong Zhang         jmax = ui[prow+1];
24245f44c854SHong Zhang         ncols = jmax-jmin;
24255f44c854SHong Zhang         i     = jmin - ui[prow];
24265f44c854SHong Zhang         cols  = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */
24275f44c854SHong Zhang         uj    = uj_lvl_ptr[prow] + i; /* levels of cols */
24285f44c854SHong Zhang         j     = *(uj - 1);
24295f44c854SHong Zhang         ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr);
24305f44c854SHong Zhang         nzk += nlnk;
24315f44c854SHong Zhang 
24325f44c854SHong Zhang         /* update il and jl for prow */
24335f44c854SHong Zhang         if (jmin < jmax){
24345f44c854SHong Zhang           il[prow] = jmin;
24355f44c854SHong Zhang           j = *cols; jl[prow] = jl[j]; jl[j] = prow;
24365f44c854SHong Zhang         }
24375f44c854SHong Zhang         prow = nextprow;
24385f44c854SHong Zhang       }
24395f44c854SHong Zhang 
24405f44c854SHong Zhang       /* if free space is not available, make more free space */
24415f44c854SHong Zhang       if (current_space->local_remaining<nzk) {
24425f44c854SHong Zhang         i  = am - k + 1; /* num of unfactored rows */
2443c6ae9bfcSHong Zhang         i *= PetscMin(nzk, i-1); /* i*nzk, i*(i-1): estimated and max additional space needed */
24445f44c854SHong Zhang         ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
24455f44c854SHong Zhang         ierr = PetscFreeSpaceGet(i,&current_space_lvl);CHKERRQ(ierr);
24465f44c854SHong Zhang         reallocs++;
24475f44c854SHong Zhang       }
24485f44c854SHong Zhang 
24495f44c854SHong Zhang       /* copy data into free_space and free_space_lvl, then initialize lnk */
24505f44c854SHong Zhang       if (nzk == 0) SETERRQ1(PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k);
24515f44c854SHong Zhang       ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
24525f44c854SHong Zhang 
24535f44c854SHong Zhang       /* add the k-th row into il and jl */
24545f44c854SHong Zhang       if (nzk > 1){
24555f44c854SHong Zhang         i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
24565f44c854SHong Zhang         jl[k] = jl[i]; jl[i] = k;
24575f44c854SHong Zhang         il[k] = ui[k] + 1;
24585f44c854SHong Zhang       }
24595f44c854SHong Zhang       uj_ptr[k]     = current_space->array;
24605f44c854SHong Zhang       uj_lvl_ptr[k] = current_space_lvl->array;
24615f44c854SHong Zhang 
24625f44c854SHong Zhang       current_space->array           += nzk;
24635f44c854SHong Zhang       current_space->local_used      += nzk;
24645f44c854SHong Zhang       current_space->local_remaining -= nzk;
24655f44c854SHong Zhang 
24665f44c854SHong Zhang       current_space_lvl->array           += nzk;
24675f44c854SHong Zhang       current_space_lvl->local_used      += nzk;
24685f44c854SHong Zhang       current_space_lvl->local_remaining -= nzk;
24695f44c854SHong Zhang 
24705f44c854SHong Zhang       ui[k+1] = ui[k] + nzk;
24715f44c854SHong Zhang     }
24725f44c854SHong Zhang 
24735f44c854SHong Zhang #if defined(PETSC_USE_INFO)
24745f44c854SHong Zhang     if (ai[am] != 0) {
24755f44c854SHong Zhang       PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]);
24765f44c854SHong Zhang       ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr);
24775f44c854SHong Zhang       ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
24785f44c854SHong Zhang       ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr);
24795f44c854SHong Zhang     } else {
24805f44c854SHong Zhang       ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr);
24815f44c854SHong Zhang     }
24825f44c854SHong Zhang #endif
24835f44c854SHong Zhang 
24845f44c854SHong Zhang     ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
24855f44c854SHong Zhang     ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
24860e83c824SBarry Smith     ierr = PetscFree4(uj_ptr,uj_lvl_ptr,jl,il);CHKERRQ(ierr);
24875f44c854SHong Zhang     ierr = PetscFree(ajtmp);CHKERRQ(ierr);
24885f44c854SHong Zhang 
24895f44c854SHong Zhang     /* destroy list of free space and other temporary array(s) */
24905f44c854SHong Zhang     ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
2491c5546cabSHong Zhang     ierr = PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag);CHKERRQ(ierr); /* store matrix factor in newdatastruct */
24925f44c854SHong Zhang     ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
24935f44c854SHong Zhang     ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
24945f44c854SHong Zhang 
24955f44c854SHong Zhang   } /* end of case: levels>0 || (levels=0 && !perm_identity) */
24965f44c854SHong Zhang 
24975f44c854SHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
24985f44c854SHong Zhang 
24995f44c854SHong Zhang   b    = (Mat_SeqSBAIJ*)(fact)->data;
25005f44c854SHong Zhang   b->singlemalloc = PETSC_FALSE;
25015f44c854SHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr);
25025f44c854SHong Zhang   b->j    = uj;
25035f44c854SHong Zhang   b->i    = ui;
25045f44c854SHong Zhang   b->diag = udiag;
25057f53bb6cSHong Zhang   b->free_diag = PETSC_TRUE;
25065f44c854SHong Zhang   b->ilen = 0;
25075f44c854SHong Zhang   b->imax = 0;
25085f44c854SHong Zhang   b->row  = perm;
25095f44c854SHong Zhang   b->col  = perm;
25105f44c854SHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
25115f44c854SHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
25125f44c854SHong Zhang   b->icol = iperm;
25135f44c854SHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
25145f44c854SHong Zhang   ierr    = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
25155f44c854SHong Zhang   ierr = PetscLogObjectMemory((fact),(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
25165f44c854SHong Zhang   b->maxnz   = b->nz = ui[am];
25175f44c854SHong Zhang   b->free_a  = PETSC_TRUE;
25185f44c854SHong Zhang   b->free_ij = PETSC_TRUE;
25195f44c854SHong Zhang 
25205f44c854SHong Zhang   (fact)->info.factor_mallocs    = reallocs;
25215f44c854SHong Zhang   (fact)->info.fill_ratio_given  = fill;
25225f44c854SHong Zhang   if (ai[am] != 0) {
25235f44c854SHong Zhang     (fact)->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]);
25245f44c854SHong Zhang   } else {
25255f44c854SHong Zhang     (fact)->info.fill_ratio_needed = 0.0;
25265f44c854SHong Zhang   }
25275f44c854SHong Zhang   (fact)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_newdatastruct;
25285f44c854SHong Zhang   PetscFunctionReturn(0);
25295f44c854SHong Zhang }
25305f44c854SHong Zhang 
25315f44c854SHong Zhang #undef __FUNCT__
25325f44c854SHong Zhang #define __FUNCT__ "MatICCFactorSymbolic_SeqAIJ"
25335f44c854SHong Zhang PetscErrorCode MatICCFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
25345f44c854SHong Zhang {
25355f44c854SHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
25365f44c854SHong Zhang   Mat_SeqSBAIJ       *b;
25375f44c854SHong Zhang   PetscErrorCode     ierr;
25385f44c854SHong Zhang   PetscTruth         perm_identity,missing;
25395f44c854SHong Zhang   PetscInt           reallocs=0,i,*ai=a->i,*aj=a->j,am=A->rmap->n,*ui,*udiag;
25405f44c854SHong Zhang   const PetscInt     *rip,*riip;
25415f44c854SHong Zhang   PetscInt           jmin,jmax,nzk,k,j,*jl,prow,*il,nextprow;
25425f44c854SHong Zhang   PetscInt           nlnk,*lnk,*lnk_lvl=PETSC_NULL,d;
25435f44c854SHong Zhang   PetscInt           ncols,ncols_upper,*cols,*ajtmp,*uj,**uj_ptr,**uj_lvl_ptr;
25445f44c854SHong Zhang   PetscReal          fill=info->fill,levels=info->levels;
25455f44c854SHong Zhang   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
25465f44c854SHong Zhang   PetscFreeSpaceList free_space_lvl=PETSC_NULL,current_space_lvl=PETSC_NULL;
25475f44c854SHong Zhang   PetscBT            lnkbt;
25485f44c854SHong Zhang   IS                 iperm;
25495f44c854SHong Zhang   PetscTruth         newdatastruct=PETSC_FALSE;
25505f44c854SHong Zhang 
25515f44c854SHong Zhang   PetscFunctionBegin;
25525f44c854SHong Zhang   ierr = PetscOptionsGetTruth(PETSC_NULL,"-icc_new",&newdatastruct,PETSC_NULL);CHKERRQ(ierr);
25535f44c854SHong Zhang   if(newdatastruct){
25545f44c854SHong Zhang     ierr = MatICCFactorSymbolic_SeqAIJ_newdatastruct(fact,A,perm,info);CHKERRQ(ierr);
25555f44c854SHong Zhang     PetscFunctionReturn(0);
25565f44c854SHong Zhang   }
25575f44c854SHong Zhang 
25585f44c854SHong 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);
25595f44c854SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&d);CHKERRQ(ierr);
25605f44c854SHong Zhang   if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",d);
25615f44c854SHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
25625f44c854SHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
25635f44c854SHong Zhang 
25645f44c854SHong Zhang   ierr = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr);
25655f44c854SHong Zhang   ierr = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr);
25665f44c854SHong Zhang   ui[0] = 0;
25675f44c854SHong Zhang 
25685f44c854SHong Zhang   /* ICC(0) without matrix ordering: simply copies fill pattern */
25695f44c854SHong Zhang   if (!levels && perm_identity) {
25705f44c854SHong Zhang 
25715f44c854SHong Zhang     for (i=0; i<am; i++) {
25725f44c854SHong Zhang       ui[i+1]  = ui[i] + ai[i+1] - a->diag[i];
25735f44c854SHong Zhang       udiag[i] = ui[i];
2574ed59401aSHong Zhang     }
257539e3d630SHong Zhang     ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
257639e3d630SHong Zhang     cols = uj;
2577ed59401aSHong Zhang     for (i=0; i<am; i++) {
2578ed59401aSHong Zhang       aj    = a->j + a->diag[i];
257939e3d630SHong Zhang       ncols = ui[i+1] - ui[i];
258039e3d630SHong Zhang       for (j=0; j<ncols; j++) *cols++ = *aj++;
2581ed59401aSHong Zhang     }
258239e3d630SHong Zhang   } else { /* case: levels>0 || (levels=0 && !perm_identity) */
2583b635d36bSHong Zhang     ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
2584b635d36bSHong Zhang     ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
2585b635d36bSHong Zhang 
25867a48dd6fSHong Zhang     /* initialization */
25875a8e39fbSHong Zhang     ierr  = PetscMalloc((am+1)*sizeof(PetscInt),&ajtmp);CHKERRQ(ierr);
25887a48dd6fSHong Zhang 
25897a48dd6fSHong Zhang     /* jl: linked list for storing indices of the pivot rows
25907a48dd6fSHong Zhang        il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
25910e83c824SBarry Smith     ierr = PetscMalloc4(am,PetscInt*,&uj_ptr,am,PetscInt*,&uj_lvl_ptr,am,PetscInt,&jl,am,PetscInt,&il);CHKERRQ(ierr);
25927a48dd6fSHong Zhang     for (i=0; i<am; i++){
25937a48dd6fSHong Zhang       jl[i] = am; il[i] = 0;
25947a48dd6fSHong Zhang     }
25957a48dd6fSHong Zhang 
25967a48dd6fSHong Zhang     /* create and initialize a linked list for storing column indices of the active row k */
25977a48dd6fSHong Zhang     nlnk = am + 1;
25982ed133dbSHong Zhang     ierr = PetscIncompleteLLCreate(am,am,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
25997a48dd6fSHong Zhang 
26007a48dd6fSHong Zhang     /* initial FreeSpace size is fill*(ai[am]+1) */
2601a1a86e44SBarry Smith     ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr);
26027a48dd6fSHong Zhang     current_space = free_space;
2603a1a86e44SBarry Smith     ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space_lvl);CHKERRQ(ierr);
26047a48dd6fSHong Zhang     current_space_lvl = free_space_lvl;
26057a48dd6fSHong Zhang 
26067a48dd6fSHong Zhang     for (k=0; k<am; k++){  /* for each active row k */
26077a48dd6fSHong Zhang       /* initialize lnk by the column indices of row rip[k] of A */
26087a48dd6fSHong Zhang       nzk   = 0;
2609622e2028SHong Zhang       ncols = ai[rip[k]+1] - ai[rip[k]];
26103b4a8b6dSBarry 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);
2611622e2028SHong Zhang       ncols_upper = 0;
2612622e2028SHong Zhang       for (j=0; j<ncols; j++){
2613b635d36bSHong Zhang         i = *(aj + ai[rip[k]] + j); /* unpermuted column index */
2614b635d36bSHong Zhang         if (riip[i] >= k){ /* only take upper triangular entry */
26155a8e39fbSHong Zhang           ajtmp[ncols_upper] = i;
2616622e2028SHong Zhang           ncols_upper++;
2617622e2028SHong Zhang         }
2618622e2028SHong Zhang       }
2619b635d36bSHong Zhang       ierr = PetscIncompleteLLInit(ncols_upper,ajtmp,am,riip,nlnk,lnk,lnk_lvl,lnkbt);CHKERRQ(ierr);
26207a48dd6fSHong Zhang       nzk += nlnk;
26217a48dd6fSHong Zhang 
26227a48dd6fSHong Zhang       /* update lnk by computing fill-in for each pivot row to be merged in */
26237a48dd6fSHong Zhang       prow = jl[k]; /* 1st pivot row */
26247a48dd6fSHong Zhang 
26257a48dd6fSHong Zhang       while (prow < k){
26267a48dd6fSHong Zhang         nextprow = jl[prow];
26277a48dd6fSHong Zhang 
26287a48dd6fSHong Zhang         /* merge prow into k-th row */
26297a48dd6fSHong Zhang         jmin = il[prow] + 1;  /* index of the 2nd nzero entry in U(prow,k:am-1) */
26307a48dd6fSHong Zhang         jmax = ui[prow+1];
26317a48dd6fSHong Zhang         ncols = jmax-jmin;
2632ed59401aSHong Zhang         i     = jmin - ui[prow];
2633ed59401aSHong Zhang         cols  = uj_ptr[prow] + i; /* points to the 2nd nzero entry in U(prow,k:am-1) */
26345a8e39fbSHong Zhang         uj    = uj_lvl_ptr[prow] + i; /* levels of cols */
26355a8e39fbSHong Zhang         j     = *(uj - 1);
26365a8e39fbSHong Zhang         ierr = PetscICCLLAddSorted(ncols,cols,levels,uj,am,nlnk,lnk,lnk_lvl,lnkbt,j);CHKERRQ(ierr);
26377a48dd6fSHong Zhang         nzk += nlnk;
26387a48dd6fSHong Zhang 
26397a48dd6fSHong Zhang         /* update il and jl for prow */
26407a48dd6fSHong Zhang         if (jmin < jmax){
26417a48dd6fSHong Zhang           il[prow] = jmin;
26427a48dd6fSHong Zhang           j = *cols; jl[prow] = jl[j]; jl[j] = prow;
26437a48dd6fSHong Zhang         }
26447a48dd6fSHong Zhang         prow = nextprow;
26457a48dd6fSHong Zhang       }
26467a48dd6fSHong Zhang 
26477a48dd6fSHong Zhang       /* if free space is not available, make more free space */
26487a48dd6fSHong Zhang       if (current_space->local_remaining<nzk) {
26497a48dd6fSHong Zhang         i = am - k + 1; /* num of unfactored rows */
2650fd3f55cbSMatthew Knepley         i *= PetscMin(nzk, (i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */
2651a1a86e44SBarry Smith         ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
2652a1a86e44SBarry Smith         ierr = PetscFreeSpaceGet(i,&current_space_lvl);CHKERRQ(ierr);
26537a48dd6fSHong Zhang         reallocs++;
26547a48dd6fSHong Zhang       }
26557a48dd6fSHong Zhang 
26567a48dd6fSHong Zhang       /* copy data into free_space and free_space_lvl, then initialize lnk */
2657b635d36bSHong Zhang       if (nzk == 0) SETERRQ1(PETSC_ERR_ARG_WRONG,"Empty row %D in ICC matrix factor",k);
26582ed133dbSHong Zhang       ierr = PetscIncompleteLLClean(am,am,nzk,lnk,lnk_lvl,current_space->array,current_space_lvl->array,lnkbt);CHKERRQ(ierr);
26597a48dd6fSHong Zhang 
26607a48dd6fSHong Zhang       /* add the k-th row into il and jl */
266139e3d630SHong Zhang       if (nzk > 1){
26627a48dd6fSHong Zhang         i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
26637a48dd6fSHong Zhang         jl[k] = jl[i]; jl[i] = k;
26647a48dd6fSHong Zhang         il[k] = ui[k] + 1;
26657a48dd6fSHong Zhang       }
26667a48dd6fSHong Zhang       uj_ptr[k]     = current_space->array;
26677a48dd6fSHong Zhang       uj_lvl_ptr[k] = current_space_lvl->array;
26687a48dd6fSHong Zhang 
26697a48dd6fSHong Zhang       current_space->array           += nzk;
26707a48dd6fSHong Zhang       current_space->local_used      += nzk;
26717a48dd6fSHong Zhang       current_space->local_remaining -= nzk;
26727a48dd6fSHong Zhang 
26737a48dd6fSHong Zhang       current_space_lvl->array           += nzk;
26747a48dd6fSHong Zhang       current_space_lvl->local_used      += nzk;
26757a48dd6fSHong Zhang       current_space_lvl->local_remaining -= nzk;
26767a48dd6fSHong Zhang 
26777a48dd6fSHong Zhang       ui[k+1] = ui[k] + nzk;
26787a48dd6fSHong Zhang     }
26797a48dd6fSHong Zhang 
26806cf91177SBarry Smith #if defined(PETSC_USE_INFO)
26817a48dd6fSHong Zhang     if (ai[am] != 0) {
268239e3d630SHong Zhang       PetscReal af = (PetscReal)ui[am]/((PetscReal)ai[am]);
2683ae15b995SBarry Smith       ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr);
2684ae15b995SBarry Smith       ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
2685ae15b995SBarry Smith       ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr);
26867a48dd6fSHong Zhang     } else {
2687ae15b995SBarry Smith       ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr);
26887a48dd6fSHong Zhang     }
268963ba0a88SBarry Smith #endif
26907a48dd6fSHong Zhang 
26917a48dd6fSHong Zhang     ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
2692b635d36bSHong Zhang     ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
26930e83c824SBarry Smith     ierr = PetscFree4(uj_ptr,uj_lvl_ptr,jl,il);CHKERRQ(ierr);
26945a8e39fbSHong Zhang     ierr = PetscFree(ajtmp);CHKERRQ(ierr);
26957a48dd6fSHong Zhang 
26967a48dd6fSHong Zhang     /* destroy list of free space and other temporary array(s) */
26977a48dd6fSHong Zhang     ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
2698a1a86e44SBarry Smith     ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr);
26992ed133dbSHong Zhang     ierr = PetscIncompleteLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
2700a1a86e44SBarry Smith     ierr = PetscFreeSpaceDestroy(free_space_lvl);CHKERRQ(ierr);
27017a48dd6fSHong Zhang 
270239e3d630SHong Zhang   } /* end of case: levels>0 || (levels=0 && !perm_identity) */
270339e3d630SHong Zhang 
27047a48dd6fSHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
27057a48dd6fSHong Zhang 
2706719d5645SBarry Smith   b    = (Mat_SeqSBAIJ*)(fact)->data;
27077a48dd6fSHong Zhang   b->singlemalloc = PETSC_FALSE;
27087a48dd6fSHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr);
27097a48dd6fSHong Zhang   b->j    = uj;
27107a48dd6fSHong Zhang   b->i    = ui;
27115f44c854SHong Zhang   b->diag = udiag;
27127f53bb6cSHong Zhang   b->free_diag = PETSC_TRUE;
27137a48dd6fSHong Zhang   b->ilen = 0;
27147a48dd6fSHong Zhang   b->imax = 0;
27157a48dd6fSHong Zhang   b->row  = perm;
2716b635d36bSHong Zhang   b->col  = perm;
2717b635d36bSHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
2718b635d36bSHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
2719b635d36bSHong Zhang   b->icol = iperm;
27207a48dd6fSHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
27217a48dd6fSHong Zhang   ierr    = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
2722719d5645SBarry Smith   ierr = PetscLogObjectMemory((fact),(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
27237a48dd6fSHong Zhang   b->maxnz   = b->nz = ui[am];
2724e6b907acSBarry Smith   b->free_a  = PETSC_TRUE;
2725e6b907acSBarry Smith   b->free_ij = PETSC_TRUE;
27267a48dd6fSHong Zhang 
2727719d5645SBarry Smith   (fact)->info.factor_mallocs    = reallocs;
2728719d5645SBarry Smith   (fact)->info.fill_ratio_given  = fill;
27297a48dd6fSHong Zhang   if (ai[am] != 0) {
2730719d5645SBarry Smith     (fact)->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]);
27317a48dd6fSHong Zhang   } else {
2732719d5645SBarry Smith     (fact)->info.fill_ratio_needed = 0.0;
27337a48dd6fSHong Zhang   }
2734719d5645SBarry Smith   (fact)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ;
2735a6175056SHong Zhang   PetscFunctionReturn(0);
2736a6175056SHong Zhang }
2737d5d45c9bSBarry Smith 
27380be760fbSHong Zhang PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ_newdatastruct(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
27390be760fbSHong Zhang {
27400be760fbSHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
27410be760fbSHong Zhang   Mat_SeqSBAIJ       *b;
27420be760fbSHong Zhang   PetscErrorCode     ierr;
27430be760fbSHong Zhang   PetscTruth         perm_identity;
27440be760fbSHong Zhang   PetscReal          fill = info->fill;
27450be760fbSHong Zhang   const PetscInt     *rip,*riip;
27460be760fbSHong Zhang   PetscInt           i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow;
27470be760fbSHong Zhang   PetscInt           *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow;
27480be760fbSHong Zhang   PetscInt           nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr,*udiag;
27490be760fbSHong Zhang   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
27500be760fbSHong Zhang   PetscBT            lnkbt;
27510be760fbSHong Zhang   IS                 iperm;
27520be760fbSHong Zhang 
27530be760fbSHong Zhang   PetscFunctionBegin;
27540be760fbSHong 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);
27550be760fbSHong Zhang   /* check whether perm is the identity mapping */
27560be760fbSHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
27570be760fbSHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
27580be760fbSHong Zhang   ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
27590be760fbSHong Zhang   ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
27600be760fbSHong Zhang 
27610be760fbSHong Zhang   /* initialization */
27620be760fbSHong Zhang   ierr  = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr);
27630be760fbSHong Zhang   ierr  = PetscMalloc((am+1)*sizeof(PetscInt),&udiag);CHKERRQ(ierr);
27640be760fbSHong Zhang   ui[0] = 0;
27650be760fbSHong Zhang 
27660be760fbSHong Zhang   /* jl: linked list for storing indices of the pivot rows
27670be760fbSHong Zhang      il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
27680be760fbSHong Zhang   ierr = PetscMalloc4(am,PetscInt*,&ui_ptr,am,PetscInt,&jl,am,PetscInt,&il,am,PetscInt,&cols);CHKERRQ(ierr);
27690be760fbSHong Zhang   for (i=0; i<am; i++){
27700be760fbSHong Zhang     jl[i] = am; il[i] = 0;
27710be760fbSHong Zhang   }
27720be760fbSHong Zhang 
27730be760fbSHong Zhang   /* create and initialize a linked list for storing column indices of the active row k */
27740be760fbSHong Zhang   nlnk = am + 1;
27750be760fbSHong Zhang   ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
27760be760fbSHong Zhang 
27770be760fbSHong Zhang   /* initial FreeSpace size is fill*(ai[am]+1) */
27780be760fbSHong Zhang   ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr);
27790be760fbSHong Zhang   current_space = free_space;
27800be760fbSHong Zhang 
27810be760fbSHong Zhang   for (k=0; k<am; k++){  /* for each active row k */
27820be760fbSHong Zhang     /* initialize lnk by the column indices of row rip[k] of A */
27830be760fbSHong Zhang     nzk   = 0;
27840be760fbSHong Zhang     ncols = ai[rip[k]+1] - ai[rip[k]];
27850be760fbSHong 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);
27860be760fbSHong Zhang     ncols_upper = 0;
27870be760fbSHong Zhang     for (j=0; j<ncols; j++){
27880be760fbSHong Zhang       i = riip[*(aj + ai[rip[k]] + j)];
27890be760fbSHong Zhang       if (i >= k){ /* only take upper triangular entry */
27900be760fbSHong Zhang         cols[ncols_upper] = i;
27910be760fbSHong Zhang         ncols_upper++;
27920be760fbSHong Zhang       }
27930be760fbSHong Zhang     }
27940be760fbSHong Zhang     ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
27950be760fbSHong Zhang     nzk += nlnk;
27960be760fbSHong Zhang 
27970be760fbSHong Zhang     /* update lnk by computing fill-in for each pivot row to be merged in */
27980be760fbSHong Zhang     prow = jl[k]; /* 1st pivot row */
27990be760fbSHong Zhang 
28000be760fbSHong Zhang     while (prow < k){
28010be760fbSHong Zhang       nextprow = jl[prow];
28020be760fbSHong Zhang       /* merge prow into k-th row */
28030be760fbSHong Zhang       jmin = il[prow] + 1;  /* index of the 2nd nzero entry in U(prow,k:am-1) */
28040be760fbSHong Zhang       jmax = ui[prow+1];
28050be760fbSHong Zhang       ncols = jmax-jmin;
28060be760fbSHong Zhang       uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */
28070be760fbSHong Zhang       ierr = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
28080be760fbSHong Zhang       nzk += nlnk;
28090be760fbSHong Zhang 
28100be760fbSHong Zhang       /* update il and jl for prow */
28110be760fbSHong Zhang       if (jmin < jmax){
28120be760fbSHong Zhang         il[prow] = jmin;
28130be760fbSHong Zhang         j = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow;
28140be760fbSHong Zhang       }
28150be760fbSHong Zhang       prow = nextprow;
28160be760fbSHong Zhang     }
28170be760fbSHong Zhang 
28180be760fbSHong Zhang     /* if free space is not available, make more free space */
28190be760fbSHong Zhang     if (current_space->local_remaining<nzk) {
28200be760fbSHong Zhang       i  = am - k + 1; /* num of unfactored rows */
2821c6ae9bfcSHong Zhang       i *= PetscMin(nzk,i-1); /* i*nzk, i*(i-1): estimated and max additional space needed */
28220be760fbSHong Zhang       ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
28230be760fbSHong Zhang       reallocs++;
28240be760fbSHong Zhang     }
28250be760fbSHong Zhang 
28260be760fbSHong Zhang     /* copy data into free space, then initialize lnk */
28270be760fbSHong Zhang     ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
28280be760fbSHong Zhang 
28290be760fbSHong Zhang     /* add the k-th row into il and jl */
28300be760fbSHong Zhang     if (nzk-1 > 0){
28310be760fbSHong Zhang       i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
28320be760fbSHong Zhang       jl[k] = jl[i]; jl[i] = k;
28330be760fbSHong Zhang       il[k] = ui[k] + 1;
28340be760fbSHong Zhang     }
28350be760fbSHong Zhang     ui_ptr[k] = current_space->array;
28360be760fbSHong Zhang     current_space->array           += nzk;
28370be760fbSHong Zhang     current_space->local_used      += nzk;
28380be760fbSHong Zhang     current_space->local_remaining -= nzk;
28390be760fbSHong Zhang 
28400be760fbSHong Zhang     ui[k+1] = ui[k] + nzk;
28410be760fbSHong Zhang   }
28420be760fbSHong Zhang 
28430be760fbSHong Zhang #if defined(PETSC_USE_INFO)
28440be760fbSHong Zhang   if (ai[am] != 0) {
28450be760fbSHong Zhang     PetscReal af = (PetscReal)(ui[am])/((PetscReal)ai[am]);
28460be760fbSHong Zhang     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr);
28470be760fbSHong Zhang     ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
28480be760fbSHong Zhang     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr);
28490be760fbSHong Zhang   } else {
28500be760fbSHong Zhang      ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr);
28510be760fbSHong Zhang   }
28520be760fbSHong Zhang #endif
28530be760fbSHong Zhang 
28540be760fbSHong Zhang   ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
28550be760fbSHong Zhang   ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
28560be760fbSHong Zhang   ierr = PetscFree4(ui_ptr,jl,il,cols);CHKERRQ(ierr);
28570be760fbSHong Zhang 
28580be760fbSHong Zhang   /* destroy list of free space and other temporary array(s) */
28590be760fbSHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
28600be760fbSHong Zhang   ierr = PetscFreeSpaceContiguous_Cholesky(&free_space,uj,am,ui,udiag);CHKERRQ(ierr); /* store matrix factor in newdatastruct */
28610be760fbSHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
28620be760fbSHong Zhang 
28630be760fbSHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
28640be760fbSHong Zhang 
28650be760fbSHong Zhang   b = (Mat_SeqSBAIJ*)(fact)->data;
28660be760fbSHong Zhang   b->singlemalloc = PETSC_FALSE;
28670be760fbSHong Zhang   b->free_a       = PETSC_TRUE;
28680be760fbSHong Zhang   b->free_ij      = PETSC_TRUE;
28690be760fbSHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr);
28700be760fbSHong Zhang   b->j    = uj;
28710be760fbSHong Zhang   b->i    = ui;
28720be760fbSHong Zhang   b->diag = udiag;
28730be760fbSHong Zhang   b->free_diag = PETSC_TRUE;
28740be760fbSHong Zhang   b->ilen = 0;
28750be760fbSHong Zhang   b->imax = 0;
28760be760fbSHong Zhang   b->row  = perm;
28770be760fbSHong Zhang   b->col  = perm;
28780be760fbSHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
28790be760fbSHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
28800be760fbSHong Zhang   b->icol = iperm;
28810be760fbSHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
28820be760fbSHong Zhang   ierr    = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
28830be760fbSHong Zhang   ierr    = PetscLogObjectMemory(fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
28840be760fbSHong Zhang   b->maxnz = b->nz = ui[am];
28850be760fbSHong Zhang 
28860be760fbSHong Zhang   (fact)->info.factor_mallocs    = reallocs;
28870be760fbSHong Zhang   (fact)->info.fill_ratio_given  = fill;
28880be760fbSHong Zhang   if (ai[am] != 0) {
28890be760fbSHong Zhang     (fact)->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]);
28900be760fbSHong Zhang   } else {
28910be760fbSHong Zhang     (fact)->info.fill_ratio_needed = 0.0;
28920be760fbSHong Zhang   }
28930be760fbSHong Zhang   (fact)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ_newdatastruct;
28940be760fbSHong Zhang   PetscFunctionReturn(0);
28950be760fbSHong Zhang }
28960be760fbSHong Zhang 
2897f76d2b81SHong Zhang #undef __FUNCT__
2898f76d2b81SHong Zhang #define __FUNCT__ "MatCholeskyFactorSymbolic_SeqAIJ"
28990481f469SBarry Smith PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS perm,const MatFactorInfo *info)
2900f76d2b81SHong Zhang {
2901f76d2b81SHong Zhang   Mat_SeqAIJ         *a = (Mat_SeqAIJ*)A->data;
290210c27e3dSHong Zhang   Mat_SeqSBAIJ       *b;
2903dfbe8321SBarry Smith   PetscErrorCode     ierr;
2904f76d2b81SHong Zhang   PetscTruth         perm_identity;
290510c27e3dSHong Zhang   PetscReal          fill = info->fill;
29065d0c19d7SBarry Smith   const PetscInt     *rip,*riip;
29075d0c19d7SBarry Smith   PetscInt           i,am=A->rmap->n,*ai=a->i,*aj=a->j,reallocs=0,prow;
290810c27e3dSHong Zhang   PetscInt           *jl,jmin,jmax,nzk,*ui,k,j,*il,nextprow;
29092ed133dbSHong Zhang   PetscInt           nlnk,*lnk,ncols,ncols_upper,*cols,*uj,**ui_ptr,*uj_ptr;
2910a1a86e44SBarry Smith   PetscFreeSpaceList free_space=PETSC_NULL,current_space=PETSC_NULL;
291110c27e3dSHong Zhang   PetscBT            lnkbt;
29122ed133dbSHong Zhang   IS                 iperm;
29130be760fbSHong Zhang   PetscTruth         newdatastruct=PETSC_FALSE;
2914f76d2b81SHong Zhang 
2915f76d2b81SHong Zhang   PetscFunctionBegin;
29160be760fbSHong Zhang   ierr = PetscOptionsGetTruth(PETSC_NULL,"-cholesky_new",&newdatastruct,PETSC_NULL);CHKERRQ(ierr);
29170be760fbSHong Zhang   if(newdatastruct){
29180be760fbSHong Zhang     ierr = MatCholeskyFactorSymbolic_SeqAIJ_newdatastruct(fact,A,perm,info);CHKERRQ(ierr);
29190be760fbSHong Zhang     PetscFunctionReturn(0);
29200be760fbSHong Zhang   }
29210be760fbSHong Zhang 
2922d0f46423SBarry 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);
29232ed133dbSHong Zhang   /* check whether perm is the identity mapping */
2924f76d2b81SHong Zhang   ierr = ISIdentity(perm,&perm_identity);CHKERRQ(ierr);
29252ed133dbSHong Zhang   ierr = ISInvertPermutation(perm,PETSC_DECIDE,&iperm);CHKERRQ(ierr);
29262ed133dbSHong Zhang   ierr = ISGetIndices(iperm,&riip);CHKERRQ(ierr);
29279bfd6278SHong Zhang   ierr = ISGetIndices(perm,&rip);CHKERRQ(ierr);
292810c27e3dSHong Zhang 
292910c27e3dSHong Zhang   /* initialization */
29301393bc97SHong Zhang   ierr  = PetscMalloc((am+1)*sizeof(PetscInt),&ui);CHKERRQ(ierr);
293110c27e3dSHong Zhang   ui[0] = 0;
293210c27e3dSHong Zhang 
293310c27e3dSHong Zhang   /* jl: linked list for storing indices of the pivot rows
29341393bc97SHong Zhang      il: il[i] points to the 1st nonzero entry of U(i,k:am-1) */
29350e83c824SBarry Smith   ierr = PetscMalloc4(am,PetscInt*,&ui_ptr,am,PetscInt,&jl,am,PetscInt,&il,am,PetscInt,&cols);CHKERRQ(ierr);
29361393bc97SHong Zhang   for (i=0; i<am; i++){
29371393bc97SHong Zhang     jl[i] = am; il[i] = 0;
2938f76d2b81SHong Zhang   }
293910c27e3dSHong Zhang 
294010c27e3dSHong Zhang   /* create and initialize a linked list for storing column indices of the active row k */
29411393bc97SHong Zhang   nlnk = am + 1;
29421393bc97SHong Zhang   ierr = PetscLLCreate(am,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
294310c27e3dSHong Zhang 
29441393bc97SHong Zhang   /* initial FreeSpace size is fill*(ai[am]+1) */
2945a1a86e44SBarry Smith   ierr = PetscFreeSpaceGet((PetscInt)(fill*(ai[am]+1)),&free_space);CHKERRQ(ierr);
294610c27e3dSHong Zhang   current_space = free_space;
294710c27e3dSHong Zhang 
29481393bc97SHong Zhang   for (k=0; k<am; k++){  /* for each active row k */
294910c27e3dSHong Zhang     /* initialize lnk by the column indices of row rip[k] of A */
295010c27e3dSHong Zhang     nzk   = 0;
29512ed133dbSHong Zhang     ncols = ai[rip[k]+1] - ai[rip[k]];
29523b4a8b6dSBarry 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);
29532ed133dbSHong Zhang     ncols_upper = 0;
29542ed133dbSHong Zhang     for (j=0; j<ncols; j++){
29559bfd6278SHong Zhang       i = riip[*(aj + ai[rip[k]] + j)];
29562ed133dbSHong Zhang       if (i >= k){ /* only take upper triangular entry */
29572ed133dbSHong Zhang         cols[ncols_upper] = i;
29582ed133dbSHong Zhang         ncols_upper++;
29592ed133dbSHong Zhang       }
29602ed133dbSHong Zhang     }
29611393bc97SHong Zhang     ierr = PetscLLAdd(ncols_upper,cols,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
296210c27e3dSHong Zhang     nzk += nlnk;
296310c27e3dSHong Zhang 
296410c27e3dSHong Zhang     /* update lnk by computing fill-in for each pivot row to be merged in */
296510c27e3dSHong Zhang     prow = jl[k]; /* 1st pivot row */
296610c27e3dSHong Zhang 
296710c27e3dSHong Zhang     while (prow < k){
296810c27e3dSHong Zhang       nextprow = jl[prow];
296910c27e3dSHong Zhang       /* merge prow into k-th row */
29701393bc97SHong Zhang       jmin = il[prow] + 1;  /* index of the 2nd nzero entry in U(prow,k:am-1) */
297110c27e3dSHong Zhang       jmax = ui[prow+1];
297210c27e3dSHong Zhang       ncols = jmax-jmin;
29731393bc97SHong Zhang       uj_ptr = ui_ptr[prow] + jmin - ui[prow]; /* points to the 2nd nzero entry in U(prow,k:am-1) */
29745a8e39fbSHong Zhang       ierr = PetscLLAddSorted(ncols,uj_ptr,am,nlnk,lnk,lnkbt);CHKERRQ(ierr);
297510c27e3dSHong Zhang       nzk += nlnk;
297610c27e3dSHong Zhang 
297710c27e3dSHong Zhang       /* update il and jl for prow */
297810c27e3dSHong Zhang       if (jmin < jmax){
297910c27e3dSHong Zhang         il[prow] = jmin;
29802ed133dbSHong Zhang         j = *uj_ptr; jl[prow] = jl[j]; jl[j] = prow;
298110c27e3dSHong Zhang       }
298210c27e3dSHong Zhang       prow = nextprow;
298310c27e3dSHong Zhang     }
298410c27e3dSHong Zhang 
298510c27e3dSHong Zhang     /* if free space is not available, make more free space */
298610c27e3dSHong Zhang     if (current_space->local_remaining<nzk) {
29871393bc97SHong Zhang       i = am - k + 1; /* num of unfactored rows */
298810c27e3dSHong Zhang       i = PetscMin(i*nzk, i*(i-1)); /* i*nzk, i*(i-1): estimated and max additional space needed */
2989a1a86e44SBarry Smith       ierr = PetscFreeSpaceGet(i,&current_space);CHKERRQ(ierr);
299010c27e3dSHong Zhang       reallocs++;
299110c27e3dSHong Zhang     }
299210c27e3dSHong Zhang 
299310c27e3dSHong Zhang     /* copy data into free space, then initialize lnk */
29941393bc97SHong Zhang     ierr = PetscLLClean(am,am,nzk,lnk,current_space->array,lnkbt);CHKERRQ(ierr);
299510c27e3dSHong Zhang 
299610c27e3dSHong Zhang     /* add the k-th row into il and jl */
299710c27e3dSHong Zhang     if (nzk-1 > 0){
29981393bc97SHong Zhang       i = current_space->array[1]; /* col value of the first nonzero element in U(k, k+1:am-1) */
299910c27e3dSHong Zhang       jl[k] = jl[i]; jl[i] = k;
300010c27e3dSHong Zhang       il[k] = ui[k] + 1;
300110c27e3dSHong Zhang     }
30022ed133dbSHong Zhang     ui_ptr[k] = current_space->array;
300310c27e3dSHong Zhang     current_space->array           += nzk;
300410c27e3dSHong Zhang     current_space->local_used      += nzk;
300510c27e3dSHong Zhang     current_space->local_remaining -= nzk;
300610c27e3dSHong Zhang 
300710c27e3dSHong Zhang     ui[k+1] = ui[k] + nzk;
300810c27e3dSHong Zhang   }
300910c27e3dSHong Zhang 
30106cf91177SBarry Smith #if defined(PETSC_USE_INFO)
30111393bc97SHong Zhang   if (ai[am] != 0) {
30121393bc97SHong Zhang     PetscReal af = (PetscReal)(ui[am])/((PetscReal)ai[am]);
3013ae15b995SBarry Smith     ierr = PetscInfo3(A,"Reallocs %D Fill ratio:given %G needed %G\n",reallocs,fill,af);CHKERRQ(ierr);
3014ae15b995SBarry Smith     ierr = PetscInfo1(A,"Run with -pc_factor_fill %G or use \n",af);CHKERRQ(ierr);
3015ae15b995SBarry Smith     ierr = PetscInfo1(A,"PCFactorSetFill(pc,%G) for best performance.\n",af);CHKERRQ(ierr);
301610c27e3dSHong Zhang   } else {
3017ae15b995SBarry Smith      ierr = PetscInfo(A,"Empty matrix.\n");CHKERRQ(ierr);
301810c27e3dSHong Zhang   }
301963ba0a88SBarry Smith #endif
302010c27e3dSHong Zhang 
302110c27e3dSHong Zhang   ierr = ISRestoreIndices(perm,&rip);CHKERRQ(ierr);
30229bfd6278SHong Zhang   ierr = ISRestoreIndices(iperm,&riip);CHKERRQ(ierr);
30230e83c824SBarry Smith   ierr = PetscFree4(ui_ptr,jl,il,cols);CHKERRQ(ierr);
302410c27e3dSHong Zhang 
302510c27e3dSHong Zhang   /* destroy list of free space and other temporary array(s) */
30261393bc97SHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(PetscInt),&uj);CHKERRQ(ierr);
3027a1a86e44SBarry Smith   ierr = PetscFreeSpaceContiguous(&free_space,uj);CHKERRQ(ierr);
302810c27e3dSHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
302910c27e3dSHong Zhang 
303010c27e3dSHong Zhang   /* put together the new matrix in MATSEQSBAIJ format */
303110c27e3dSHong Zhang 
3032719d5645SBarry Smith   b = (Mat_SeqSBAIJ*)(fact)->data;
303310c27e3dSHong Zhang   b->singlemalloc = PETSC_FALSE;
3034e6b907acSBarry Smith   b->free_a       = PETSC_TRUE;
3035e6b907acSBarry Smith   b->free_ij      = PETSC_TRUE;
30361393bc97SHong Zhang   ierr = PetscMalloc((ui[am]+1)*sizeof(MatScalar),&b->a);CHKERRQ(ierr);
303710c27e3dSHong Zhang   b->j    = uj;
303810c27e3dSHong Zhang   b->i    = ui;
303910c27e3dSHong Zhang   b->diag = 0;
304010c27e3dSHong Zhang   b->ilen = 0;
304110c27e3dSHong Zhang   b->imax = 0;
304210c27e3dSHong Zhang   b->row  = perm;
30439bfd6278SHong Zhang   b->col  = perm;
30449bfd6278SHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
30459bfd6278SHong Zhang   ierr    = PetscObjectReference((PetscObject)perm);CHKERRQ(ierr);
30469bfd6278SHong Zhang   b->icol = iperm;
304710c27e3dSHong Zhang   b->pivotinblocks = PETSC_FALSE; /* need to get from MatFactorInfo */
30481393bc97SHong Zhang   ierr    = PetscMalloc((am+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
3049719d5645SBarry Smith   ierr    = PetscLogObjectMemory(fact,(ui[am]-am)*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
30501393bc97SHong Zhang   b->maxnz = b->nz = ui[am];
305110c27e3dSHong Zhang 
3052719d5645SBarry Smith   (fact)->info.factor_mallocs    = reallocs;
3053719d5645SBarry Smith   (fact)->info.fill_ratio_given  = fill;
30541393bc97SHong Zhang   if (ai[am] != 0) {
3055719d5645SBarry Smith     (fact)->info.fill_ratio_needed = ((PetscReal)ui[am])/((PetscReal)ai[am]);
305610c27e3dSHong Zhang   } else {
3057719d5645SBarry Smith     (fact)->info.fill_ratio_needed = 0.0;
305810c27e3dSHong Zhang   }
3059719d5645SBarry Smith   (fact)->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqAIJ;
3060f76d2b81SHong Zhang   PetscFunctionReturn(0);
3061f76d2b81SHong Zhang }
3062599ef60dSHong Zhang 
30631d098868SHong Zhang #undef __FUNCT__
30644395751fSShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ_NaturalOrdering_newdatastruct"
30654395751fSShri Abhyankar PetscErrorCode MatSolve_SeqAIJ_NaturalOrdering_newdatastruct(Mat A,Vec bb,Vec xx)
3066813a1f2bSShri Abhyankar {
3067813a1f2bSShri Abhyankar   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
3068813a1f2bSShri Abhyankar   PetscErrorCode    ierr;
3069813a1f2bSShri Abhyankar   PetscInt          n = A->rmap->n;
3070813a1f2bSShri Abhyankar   const PetscInt    *ai = a->i,*aj = a->j,*adiag = a->diag,*vi;
3071813a1f2bSShri Abhyankar   PetscScalar       *x,sum;
3072813a1f2bSShri Abhyankar   const PetscScalar *b;
3073813a1f2bSShri Abhyankar   const MatScalar   *aa = a->a,*v;
3074813a1f2bSShri Abhyankar   PetscInt          i,nz;
3075813a1f2bSShri Abhyankar 
3076813a1f2bSShri Abhyankar   PetscFunctionBegin;
3077813a1f2bSShri Abhyankar   if (!n) PetscFunctionReturn(0);
3078813a1f2bSShri Abhyankar 
3079813a1f2bSShri Abhyankar   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
3080813a1f2bSShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
3081813a1f2bSShri Abhyankar 
3082813a1f2bSShri Abhyankar   /* forward solve the lower triangular */
3083813a1f2bSShri Abhyankar   x[0] = b[0];
3084813a1f2bSShri Abhyankar   v    = aa;
3085813a1f2bSShri Abhyankar   vi   = aj;
3086813a1f2bSShri Abhyankar   for (i=1; i<n; i++) {
3087813a1f2bSShri Abhyankar     nz  = ai[i+1] - ai[i];
3088813a1f2bSShri Abhyankar     sum = b[i];
3089813a1f2bSShri Abhyankar     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
3090813a1f2bSShri Abhyankar     v  += nz;
3091813a1f2bSShri Abhyankar     vi += nz;
3092813a1f2bSShri Abhyankar     x[i] = sum;
3093813a1f2bSShri Abhyankar   }
3094813a1f2bSShri Abhyankar 
3095813a1f2bSShri Abhyankar   /* backward solve the upper triangular */
309662fbd6afSShri Abhyankar   for (i=n-1; i>=0; i--){
30973c0119dfSShri Abhyankar     v   = aa + adiag[i+1] + 1;
30983c0119dfSShri Abhyankar     vi  = aj + adiag[i+1] + 1;
3099813a1f2bSShri Abhyankar     nz = adiag[i] - adiag[i+1]-1;
3100813a1f2bSShri Abhyankar     sum = x[i];
3101813a1f2bSShri Abhyankar     PetscSparseDenseMinusDot(sum,x,v,vi,nz);
31023c0119dfSShri Abhyankar     x[i] = sum*v[nz]; /* x[i]=aa[adiag[i]]*sum; v++; */
3103813a1f2bSShri Abhyankar   }
3104813a1f2bSShri Abhyankar 
3105813a1f2bSShri Abhyankar   ierr = PetscLogFlops(2.0*a->nz - A->cmap->n);CHKERRQ(ierr);
3106813a1f2bSShri Abhyankar   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
3107813a1f2bSShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
3108813a1f2bSShri Abhyankar   PetscFunctionReturn(0);
3109813a1f2bSShri Abhyankar }
3110813a1f2bSShri Abhyankar 
31111d098868SHong Zhang #undef __FUNCT__
31124395751fSShri Abhyankar #define __FUNCT__ "MatSolve_SeqAIJ_newdatastruct"
31134395751fSShri Abhyankar PetscErrorCode MatSolve_SeqAIJ_newdatastruct(Mat A,Vec bb,Vec xx)
3114f268cba8SShri Abhyankar {
3115f268cba8SShri Abhyankar   Mat_SeqAIJ        *a = (Mat_SeqAIJ*)A->data;
3116f268cba8SShri Abhyankar   IS                iscol = a->col,isrow = a->row;
3117f268cba8SShri Abhyankar   PetscErrorCode    ierr;
3118f268cba8SShri Abhyankar   PetscInt          i,n=A->rmap->n,*vi,*ai=a->i,*aj=a->j,*adiag = a->diag,nz;
3119f268cba8SShri Abhyankar   const PetscInt    *rout,*cout,*r,*c;
3120f268cba8SShri Abhyankar   PetscScalar       *x,*tmp,sum;
3121f268cba8SShri Abhyankar   const PetscScalar *b;
3122f268cba8SShri Abhyankar   const MatScalar   *aa = a->a,*v;
3123f268cba8SShri Abhyankar 
3124f268cba8SShri Abhyankar   PetscFunctionBegin;
3125f268cba8SShri Abhyankar   if (!n) PetscFunctionReturn(0);
3126f268cba8SShri Abhyankar 
3127f268cba8SShri Abhyankar   ierr = VecGetArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
3128f268cba8SShri Abhyankar   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
3129f268cba8SShri Abhyankar   tmp  = a->solve_work;
3130f268cba8SShri Abhyankar 
3131f268cba8SShri Abhyankar   ierr = ISGetIndices(isrow,&rout);CHKERRQ(ierr); r = rout;
3132f268cba8SShri Abhyankar   ierr = ISGetIndices(iscol,&cout);CHKERRQ(ierr); c = cout;
3133f268cba8SShri Abhyankar 
3134f268cba8SShri Abhyankar   /* forward solve the lower triangular */
3135f268cba8SShri Abhyankar   tmp[0] = b[r[0]];
3136f268cba8SShri Abhyankar   v      = aa;
3137f268cba8SShri Abhyankar   vi     = aj;
3138f268cba8SShri Abhyankar   for (i=1; i<n; i++) {
3139f268cba8SShri Abhyankar     nz  = ai[i+1] - ai[i];
3140f268cba8SShri Abhyankar     sum = b[r[i]];
3141f268cba8SShri Abhyankar     PetscSparseDenseMinusDot(sum,tmp,v,vi,nz);
3142f268cba8SShri Abhyankar     tmp[i] = sum;
3143f268cba8SShri Abhyankar     v += nz; vi += nz;
3144f268cba8SShri Abhyankar   }
3145f268cba8SShri Abhyankar 
3146f268cba8SShri Abhyankar   /* backward solve the upper triangular */
3147f268cba8SShri Abhyankar   for (i=n-1; i>=0; i--){
31483c0119dfSShri Abhyankar     v   = aa + adiag[i+1]+1;
31493c0119dfSShri Abhyankar     vi  = aj + adiag[i+1]+1;
3150f268cba8SShri Abhyankar     nz  = adiag[i]-adiag[i+1]-1;
3151f268cba8SShri Abhyankar     sum = tmp[i];
3152f268cba8SShri Abhyankar     PetscSparseDenseMinusDot(sum,tmp,v,vi,nz);
3153f268cba8SShri Abhyankar     x[c[i]] = tmp[i] = sum*v[nz]; /* v[nz] = aa[adiag[i]] */
3154f268cba8SShri Abhyankar   }
3155f268cba8SShri Abhyankar 
3156f268cba8SShri Abhyankar   ierr = ISRestoreIndices(isrow,&rout);CHKERRQ(ierr);
3157f268cba8SShri Abhyankar   ierr = ISRestoreIndices(iscol,&cout);CHKERRQ(ierr);
3158f268cba8SShri Abhyankar   ierr = VecRestoreArray(bb,(PetscScalar**)&b);CHKERRQ(ierr);
3159f268cba8SShri Abhyankar   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
3160f268cba8SShri Abhyankar   ierr = PetscLogFlops(2*a->nz - A->cmap->n);CHKERRQ(ierr);
3161f268cba8SShri Abhyankar   PetscFunctionReturn(0);
3162f268cba8SShri Abhyankar }
3163f268cba8SShri Abhyankar 
3164f268cba8SShri Abhyankar #undef __FUNCT__
31651d098868SHong Zhang #define __FUNCT__ "MatILUDTFactor_SeqAIJ"
31661d098868SHong Zhang PetscErrorCode MatILUDTFactor_SeqAIJ(Mat A,IS isrow,IS iscol,const MatFactorInfo *info,Mat *fact)
31671d098868SHong Zhang {
31681d098868SHong Zhang   Mat                B = *fact;
3169599ef60dSHong Zhang   Mat_SeqAIJ         *a=(Mat_SeqAIJ*)A->data,*b;
3170599ef60dSHong Zhang   IS                 isicol;
3171599ef60dSHong Zhang   PetscErrorCode     ierr;
31721d098868SHong Zhang   const PetscInt     *r,*ic;
31731d098868SHong Zhang   PetscInt           i,n=A->rmap->n,*ai=a->i,*aj=a->j,*ajtmp,*adiag;
3174dc3a2fd3SHong Zhang   PetscInt           *bi,*bj,*bdiag,*bdiag_rev;
3175f61a948aSLisandro Dalcin   PetscInt           row,nzi,nzi_bl,nzi_bu,*im,nzi_al,nzi_au;
31761d098868SHong Zhang   PetscInt           nlnk,*lnk;
31771d098868SHong Zhang   PetscBT            lnkbt;
31781d098868SHong Zhang   PetscTruth         row_identity,icol_identity,both_identity;
31798aee2decSHong Zhang   MatScalar          *aatmp,*pv,*batmp,*ba,*rtmp,*pc,multiplier,*vtmp,diag_tmp;
31801d098868SHong Zhang   const PetscInt     *ics;
31811d098868SHong Zhang   PetscInt           j,nz,*pj,*bjtmp,k,ncut,*jtmp;
3182f61a948aSLisandro Dalcin   PetscReal          dt=info->dt,dtcol=info->dtcol,shift=info->shiftinblocks;
3183f61a948aSLisandro Dalcin   PetscInt           dtcount=(PetscInt)info->dtcount,nnz_max;
3184599ef60dSHong Zhang   PetscTruth         missing;
3185599ef60dSHong Zhang 
3186599ef60dSHong Zhang   PetscFunctionBegin;
3187f61a948aSLisandro Dalcin 
3188f61a948aSLisandro Dalcin   if (dt      == PETSC_DEFAULT) dt      = 0.005;
3189f61a948aSLisandro Dalcin   if (dtcol   == PETSC_DEFAULT) dtcol   = 0.01; /* XXX unused! */
3190f61a948aSLisandro Dalcin   if (dtcount == PETSC_DEFAULT) dtcount = (PetscInt)(1.5*a->rmax);
3191f61a948aSLisandro Dalcin 
31921d098868SHong Zhang   /* ------- symbolic factorization, can be reused ---------*/
31931d098868SHong Zhang   ierr = MatMissingDiagonal(A,&missing,&i);CHKERRQ(ierr);
31941d098868SHong Zhang   if (missing) SETERRQ1(PETSC_ERR_ARG_WRONGSTATE,"Matrix is missing diagonal entry %D",i);
31951d098868SHong Zhang   adiag=a->diag;
3196599ef60dSHong Zhang 
3197599ef60dSHong Zhang   ierr = ISInvertPermutation(iscol,PETSC_DECIDE,&isicol);CHKERRQ(ierr);
3198599ef60dSHong Zhang 
3199599ef60dSHong Zhang   /* bdiag is location of diagonal in factor */
32001bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag);CHKERRQ(ierr);     /* becomes b->diag */
32011bfa06eaSJed Brown   ierr = PetscMalloc((n+1)*sizeof(PetscInt),&bdiag_rev);CHKERRQ(ierr); /* temporary */
3202599ef60dSHong Zhang 
32031d098868SHong Zhang   /* allocate row pointers bi */
32048fc3a347SHong Zhang   ierr = PetscMalloc((2*n+2)*sizeof(PetscInt),&bi);CHKERRQ(ierr);
32051d098868SHong Zhang 
32061d098868SHong Zhang   /* allocate bj and ba; max num of nonzero entries is (ai[n]+2*n*dtcount+2) */
3207393d3a68SHong Zhang   if (dtcount > n-1) dtcount = n-1; /* diagonal is excluded */
32081d098868SHong Zhang   nnz_max  = ai[n]+2*n*dtcount+2;
32098fc3a347SHong Zhang 
32108fc3a347SHong Zhang   ierr = PetscMalloc((nnz_max+1)*sizeof(PetscInt),&bj);CHKERRQ(ierr);
32118fc3a347SHong Zhang   ierr = PetscMalloc((nnz_max+1)*sizeof(MatScalar),&ba);CHKERRQ(ierr);
3212599ef60dSHong Zhang 
3213599ef60dSHong Zhang   /* put together the new matrix */
3214599ef60dSHong Zhang   ierr = MatSeqAIJSetPreallocation_SeqAIJ(B,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
3215599ef60dSHong Zhang   ierr = PetscLogObjectParent(B,isicol);CHKERRQ(ierr);
3216599ef60dSHong Zhang   b    = (Mat_SeqAIJ*)(B)->data;
3217599ef60dSHong Zhang   b->free_a       = PETSC_TRUE;
3218599ef60dSHong Zhang   b->free_ij      = PETSC_TRUE;
3219599ef60dSHong Zhang   b->singlemalloc = PETSC_FALSE;
3220599ef60dSHong Zhang   b->a          = ba;
3221599ef60dSHong Zhang   b->j          = bj;
3222599ef60dSHong Zhang   b->i          = bi;
3223599ef60dSHong Zhang   b->diag       = bdiag;
3224599ef60dSHong Zhang   b->ilen       = 0;
3225599ef60dSHong Zhang   b->imax       = 0;
3226599ef60dSHong Zhang   b->row        = isrow;
3227599ef60dSHong Zhang   b->col        = iscol;
3228599ef60dSHong Zhang   ierr          = PetscObjectReference((PetscObject)isrow);CHKERRQ(ierr);
3229599ef60dSHong Zhang   ierr          = PetscObjectReference((PetscObject)iscol);CHKERRQ(ierr);
3230599ef60dSHong Zhang   b->icol       = isicol;
3231599ef60dSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(PetscScalar),&b->solve_work);CHKERRQ(ierr);
3232599ef60dSHong Zhang 
32331d098868SHong Zhang   ierr = PetscLogObjectMemory(B,nnz_max*(sizeof(PetscInt)+sizeof(MatScalar)));CHKERRQ(ierr);
32341d098868SHong Zhang   b->maxnz = nnz_max;
3235599ef60dSHong Zhang 
3236599ef60dSHong Zhang   (B)->factor                = MAT_FACTOR_ILUDT;
3237599ef60dSHong Zhang   (B)->info.factor_mallocs   = 0;
32381d098868SHong Zhang   (B)->info.fill_ratio_given = ((PetscReal)nnz_max)/((PetscReal)ai[n]);
32391d098868SHong Zhang   CHKMEMQ;
32401d098868SHong Zhang   /* ------- end of symbolic factorization ---------*/
3241599ef60dSHong Zhang 
3242599ef60dSHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
3243599ef60dSHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
3244599ef60dSHong Zhang   ics  = ic;
3245599ef60dSHong Zhang 
3246599ef60dSHong Zhang   /* linked list for storing column indices of the active row */
3247599ef60dSHong Zhang   nlnk = n + 1;
3248599ef60dSHong Zhang   ierr = PetscLLCreate(n,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
32491d098868SHong Zhang 
32501d098868SHong Zhang   /* im: used by PetscLLAddSortedLU(); jtmp: working array for column indices of active row */
32510e83c824SBarry Smith   ierr = PetscMalloc2(n,PetscInt,&im,n,PetscInt,&jtmp);CHKERRQ(ierr);
32521d098868SHong Zhang   /* rtmp, vtmp: working arrays for sparse and contiguous row entries of active row */
32530e83c824SBarry Smith   ierr = PetscMalloc2(n,MatScalar,&rtmp,n,MatScalar,&vtmp);CHKERRQ(ierr);
32540e83c824SBarry Smith   ierr = PetscMemzero(rtmp,n*sizeof(MatScalar));CHKERRQ(ierr);
3255599ef60dSHong Zhang 
3256599ef60dSHong Zhang   bi[0]    = 0;
32578fc3a347SHong Zhang   bdiag[0] = nnz_max-1; /* location of diag[0] in factor B */
3258dc3a2fd3SHong Zhang   bdiag_rev[n] = bdiag[0];
32598fc3a347SHong Zhang   bi[2*n+1] = bdiag[0]+1; /* endof bj and ba array */
3260599ef60dSHong Zhang   for (i=0; i<n; i++) {
3261599ef60dSHong Zhang     /* copy initial fill into linked list */
3262599ef60dSHong Zhang     nzi = 0; /* nonzeros for active row i */
32638369b28dSHong Zhang     nzi = ai[r[i]+1] - ai[r[i]];
32648369b28dSHong 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);
32658369b28dSHong Zhang     nzi_al = adiag[r[i]] - ai[r[i]];
32668369b28dSHong Zhang     nzi_au = ai[r[i]+1] - adiag[r[i]] -1;
3267599ef60dSHong Zhang     ajtmp = aj + ai[r[i]];
32688369b28dSHong Zhang     ierr = PetscLLAddPerm(nzi,ajtmp,ic,n,nlnk,lnk,lnkbt);CHKERRQ(ierr);
3269599ef60dSHong Zhang 
3270599ef60dSHong Zhang     /* load in initial (unfactored row) */
32711d098868SHong Zhang     aatmp = a->a + ai[r[i]];
32728369b28dSHong Zhang     for (j=0; j<nzi; j++) {
32731d098868SHong Zhang       rtmp[ics[*ajtmp++]] = *aatmp++;
3274599ef60dSHong Zhang     }
3275599ef60dSHong Zhang 
3276599ef60dSHong Zhang     /* add pivot rows into linked list */
3277599ef60dSHong Zhang     row = lnk[n];
3278599ef60dSHong Zhang     while (row < i ) {
32791d098868SHong Zhang       nzi_bl = bi[row+1] - bi[row] + 1;
32801d098868SHong Zhang       bjtmp = bj + bdiag[row+1]+1; /* points to 1st column next to the diagonal in U */
32811d098868SHong Zhang       ierr  = PetscLLAddSortedLU(bjtmp,row,nlnk,lnk,lnkbt,i,nzi_bl,im);CHKERRQ(ierr);
3282599ef60dSHong Zhang       nzi  += nlnk;
3283599ef60dSHong Zhang       row   = lnk[row];
3284599ef60dSHong Zhang     }
3285599ef60dSHong Zhang 
32868369b28dSHong Zhang     /* copy data from lnk into jtmp, then initialize lnk */
32878369b28dSHong Zhang     ierr = PetscLLClean(n,n,nzi,lnk,jtmp,lnkbt);CHKERRQ(ierr);
3288599ef60dSHong Zhang 
3289599ef60dSHong Zhang     /* numerical factorization */
32908369b28dSHong Zhang     bjtmp = jtmp;
3291599ef60dSHong Zhang     row   = *bjtmp++; /* 1st pivot row */
3292599ef60dSHong Zhang     while  ( row < i ) {
3293599ef60dSHong Zhang       pc         = rtmp + row;
32943c2a7987SHong Zhang       pv         = ba + bdiag[row]; /* 1./(diag of the pivot row) */
32953c2a7987SHong Zhang       multiplier = (*pc) * (*pv);
3296599ef60dSHong Zhang       *pc        = multiplier;
32973c2a7987SHong Zhang       if (PetscAbsScalar(*pc) > dt){ /* apply tolerance dropping rule */
32981d098868SHong Zhang         pj         = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */
32991d098868SHong Zhang         pv         = ba + bdiag[row+1] + 1;
33001d098868SHong Zhang         /* if (multiplier < -1.0 or multiplier >1.0) printf("row/prow %d, %d, multiplier %g\n",i,row,multiplier); */
33011d098868SHong Zhang         nz         = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */
33021d098868SHong Zhang         for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++);
3303dc0b31edSSatish Balay         ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr);
3304599ef60dSHong Zhang       }
3305599ef60dSHong Zhang       row = *bjtmp++;
3306599ef60dSHong Zhang     }
3307599ef60dSHong Zhang 
33088369b28dSHong Zhang     /* copy sparse rtmp into contiguous vtmp; separate L and U part */
33098aee2decSHong Zhang     diag_tmp = rtmp[i];  /* save diagonal value - may not needed?? */
33108369b28dSHong Zhang     nzi_bl = 0; j = 0;
33118369b28dSHong Zhang     while (jtmp[j] < i){ /* Note: jtmp is sorted */
33128aee2decSHong Zhang       vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0;
33138369b28dSHong Zhang       nzi_bl++; j++;
33148369b28dSHong Zhang     }
33158369b28dSHong Zhang     nzi_bu = nzi - nzi_bl -1;
33168369b28dSHong Zhang     while (j < nzi){
33178aee2decSHong Zhang       vtmp[j] = rtmp[jtmp[j]]; rtmp[jtmp[j]]=0.0;
33188369b28dSHong Zhang       j++;
33198369b28dSHong Zhang     }
33208369b28dSHong Zhang 
3321599ef60dSHong Zhang     bjtmp = bj + bi[i];
3322599ef60dSHong Zhang     batmp = ba + bi[i];
33238369b28dSHong Zhang     /* apply level dropping rule to L part */
33248369b28dSHong Zhang     ncut = nzi_al + dtcount;
33258369b28dSHong Zhang     if (ncut < nzi_bl){
33268369b28dSHong Zhang       ierr = PetscSortSplit(ncut,nzi_bl,vtmp,jtmp);CHKERRQ(ierr);
33278369b28dSHong Zhang       ierr = PetscSortIntWithScalarArray(ncut,jtmp,vtmp);CHKERRQ(ierr);
3328599ef60dSHong Zhang     } else {
33298369b28dSHong Zhang       ncut = nzi_bl;
33308369b28dSHong Zhang     }
33318369b28dSHong Zhang     for (j=0; j<ncut; j++){
33328369b28dSHong Zhang       bjtmp[j] = jtmp[j];
33338369b28dSHong Zhang       batmp[j] = vtmp[j];
33346da515a1SHong Zhang       /* printf(" (%d,%g),",bjtmp[j],batmp[j]); */
33358369b28dSHong Zhang     }
33361d098868SHong Zhang     bi[i+1] = bi[i] + ncut;
33378369b28dSHong Zhang     nzi = ncut + 1;
33388369b28dSHong Zhang 
33398369b28dSHong Zhang     /* apply level dropping rule to U part */
33408369b28dSHong Zhang     ncut = nzi_au + dtcount;
33418369b28dSHong Zhang     if (ncut < nzi_bu){
33428369b28dSHong Zhang       ierr = PetscSortSplit(ncut,nzi_bu,vtmp+nzi_bl+1,jtmp+nzi_bl+1);CHKERRQ(ierr);
33438369b28dSHong Zhang       ierr = PetscSortIntWithScalarArray(ncut,jtmp+nzi_bl+1,vtmp+nzi_bl+1);CHKERRQ(ierr);
33448369b28dSHong Zhang     } else {
33458369b28dSHong Zhang       ncut = nzi_bu;
33468369b28dSHong Zhang     }
33478369b28dSHong Zhang     nzi += ncut;
33481d098868SHong Zhang 
33491d098868SHong Zhang     /* mark bdiagonal */
33501d098868SHong Zhang     bdiag[i+1]       = bdiag[i] - (ncut + 1);
3351dc3a2fd3SHong Zhang     bdiag_rev[n-i-1] = bdiag[i+1];
33528fc3a347SHong Zhang     bi[2*n - i]      = bi[2*n - i +1] - (ncut + 1);
33531d098868SHong Zhang     bjtmp = bj + bdiag[i];
33541d098868SHong Zhang     batmp = ba + bdiag[i];
33551d098868SHong Zhang     *bjtmp = i;
33568aee2decSHong Zhang     *batmp = diag_tmp; /* rtmp[i]; */
3357c9c72f8fSHong Zhang     if (*batmp == 0.0) {
3358c9c72f8fSHong Zhang       *batmp = dt+shift;
3359bd1bc851SBarry Smith       /* printf(" row %d add shift %g\n",i,shift); */
3360c9c72f8fSHong Zhang     }
3361b78a477dSHong Zhang     *batmp = 1.0/(*batmp); /* invert diagonal entries for simplier triangular solves */
33626da515a1SHong Zhang     /* printf(" (%d,%g),",*bjtmp,*batmp); */
33631d098868SHong Zhang 
33641d098868SHong Zhang     bjtmp = bj + bdiag[i+1]+1;
33651d098868SHong Zhang     batmp = ba + bdiag[i+1]+1;
33668369b28dSHong Zhang     for (k=0; k<ncut; k++){
33671d098868SHong Zhang       bjtmp[k] = jtmp[nzi_bl+1+k];
33681d098868SHong Zhang       batmp[k] = vtmp[nzi_bl+1+k];
33696da515a1SHong Zhang       /* printf(" (%d,%g),",bjtmp[k],batmp[k]); */
3370599ef60dSHong Zhang     }
33716da515a1SHong Zhang     /* printf("\n"); */
3372599ef60dSHong Zhang 
3373599ef60dSHong Zhang     im[i]   = nzi; /* used by PetscLLAddSortedLU() */
3374599ef60dSHong Zhang     /*
33751d098868SHong Zhang     printf("row %d: bi %d, bdiag %d\n",i,bi[i],bdiag[i]);
33768369b28dSHong Zhang     printf(" ----------------------------\n");
3377599ef60dSHong Zhang     */
33788369b28dSHong Zhang   } /* for (i=0; i<n; i++) */
33791d098868SHong Zhang   /* printf("end of L %d, beginning of U %d\n",bi[n],bdiag[n]); */
33801d098868SHong 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]);
3381599ef60dSHong Zhang 
3382599ef60dSHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
3383599ef60dSHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
3384599ef60dSHong Zhang 
3385599ef60dSHong Zhang   ierr = PetscLLDestroy(lnk,lnkbt);CHKERRQ(ierr);
33860e83c824SBarry Smith   ierr = PetscFree2(im,jtmp);CHKERRQ(ierr);
33870e83c824SBarry Smith   ierr = PetscFree2(rtmp,vtmp);CHKERRQ(ierr);
33881bfa06eaSJed Brown   ierr = PetscFree(bdiag_rev);CHKERRQ(ierr);
3389599ef60dSHong Zhang 
3390599ef60dSHong Zhang   ierr = PetscLogFlops(B->cmap->n);CHKERRQ(ierr);
33911d098868SHong Zhang   b->maxnz = b->nz = bi[n] + bdiag[0] - bdiag[n];
3392599ef60dSHong Zhang 
3393599ef60dSHong Zhang   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
3394599ef60dSHong Zhang   ierr = ISIdentity(isicol,&icol_identity);CHKERRQ(ierr);
3395599ef60dSHong Zhang   both_identity = (PetscTruth) (row_identity && icol_identity);
3396599ef60dSHong Zhang   if (row_identity && icol_identity) {
3397894a2336SHong Zhang     B->ops->solve = MatSolve_SeqAIJ_NaturalOrdering_newdatastruct;
3398599ef60dSHong Zhang   } else {
3399894a2336SHong Zhang     B->ops->solve = MatSolve_SeqAIJ_newdatastruct;
3400599ef60dSHong Zhang   }
3401599ef60dSHong Zhang 
3402b78a477dSHong Zhang   B->ops->lufactorsymbolic  = MatILUDTFactorSymbolic_SeqAIJ;
3403b78a477dSHong Zhang   B->ops->lufactornumeric   = MatILUDTFactorNumeric_SeqAIJ;
34041d098868SHong Zhang   B->ops->solveadd          = 0;
34051d098868SHong Zhang   B->ops->solvetranspose    = 0;
34061d098868SHong Zhang   B->ops->solvetransposeadd = 0;
34071d098868SHong Zhang   B->ops->matsolve          = 0;
3408599ef60dSHong Zhang   B->assembled              = PETSC_TRUE;
3409599ef60dSHong Zhang   B->preallocated           = PETSC_TRUE;
3410599ef60dSHong Zhang   PetscFunctionReturn(0);
3411599ef60dSHong Zhang }
3412599ef60dSHong Zhang 
34133c2a7987SHong Zhang /* a wraper of MatILUDTFactor_SeqAIJ() */
34143c2a7987SHong Zhang #undef __FUNCT__
34153c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorSymbolic_SeqAIJ"
34163c2a7987SHong Zhang PetscErrorCode PETSCMAT_DLLEXPORT MatILUDTFactorSymbolic_SeqAIJ(Mat fact,Mat A,IS row,IS col,const MatFactorInfo *info)
34173c2a7987SHong Zhang {
34183c2a7987SHong Zhang   PetscErrorCode     ierr;
34193c2a7987SHong Zhang 
34203c2a7987SHong Zhang   PetscFunctionBegin;
34213c2a7987SHong Zhang   ierr = MatILUDTFactor_SeqAIJ(A,row,col,info,&fact);CHKERRQ(ierr);
34223c2a7987SHong Zhang 
34233c2a7987SHong Zhang   fact->ops->lufactornumeric = MatILUDTFactorNumeric_SeqAIJ;
34243c2a7987SHong Zhang   PetscFunctionReturn(0);
34253c2a7987SHong Zhang }
34263c2a7987SHong Zhang 
34273c2a7987SHong Zhang /*
34283c2a7987SHong Zhang    same as MatLUFactorNumeric_SeqAIJ(), except using contiguous array matrix factors
34293c2a7987SHong Zhang    - intend to replace existing MatLUFactorNumeric_SeqAIJ()
34303c2a7987SHong Zhang */
34313c2a7987SHong Zhang #undef __FUNCT__
34323c2a7987SHong Zhang #define __FUNCT__ "MatILUDTFactorNumeric_SeqAIJ"
34333c2a7987SHong Zhang PetscErrorCode PETSCMAT_DLLEXPORT MatILUDTFactorNumeric_SeqAIJ(Mat fact,Mat A,const MatFactorInfo *info)
34343c2a7987SHong Zhang {
34353c2a7987SHong Zhang   Mat            C=fact;
34363c2a7987SHong Zhang   Mat_SeqAIJ     *a=(Mat_SeqAIJ*)A->data,*b=(Mat_SeqAIJ *)C->data;
34373c2a7987SHong Zhang   IS             isrow = b->row,isicol = b->icol;
34383c2a7987SHong Zhang   PetscErrorCode ierr;
34393c2a7987SHong Zhang   const PetscInt *r,*ic,*ics;
3440b78a477dSHong Zhang   PetscInt       i,j,k,n=A->rmap->n,*ai=a->i,*aj=a->j,*bi=b->i,*bj=b->j;
3441b78a477dSHong Zhang   PetscInt       *ajtmp,*bjtmp,nz,nzl,nzu,row,*bdiag = b->diag,*pj;
34423c2a7987SHong Zhang   MatScalar      *rtmp,*pc,multiplier,*v,*pv,*aa=a->a;
3443b78a477dSHong Zhang   PetscReal      dt=info->dt,shift=info->shiftinblocks;
3444b78a477dSHong Zhang   PetscTruth     row_identity, col_identity;
34453c2a7987SHong Zhang 
34463c2a7987SHong Zhang   PetscFunctionBegin;
34473c2a7987SHong Zhang   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
34483c2a7987SHong Zhang   ierr = ISGetIndices(isicol,&ic);CHKERRQ(ierr);
3449b78a477dSHong Zhang   ierr = PetscMalloc((n+1)*sizeof(MatScalar),&rtmp);CHKERRQ(ierr);
34503c2a7987SHong Zhang   ics  = ic;
34513c2a7987SHong Zhang 
34523c2a7987SHong Zhang   for (i=0; i<n; i++){
3453b78a477dSHong Zhang     /* initialize rtmp array */
3454b78a477dSHong Zhang     nzl   = bi[i+1] - bi[i];       /* num of nozeros in L(i,:) */
34553c2a7987SHong Zhang     bjtmp = bj + bi[i];
3456b78a477dSHong Zhang     for  (j=0; j<nzl; j++) rtmp[*bjtmp++] = 0.0;
3457b78a477dSHong Zhang     rtmp[i] = 0.0;
3458b78a477dSHong Zhang     nzu   = bdiag[i] - bdiag[i+1]; /* num of nozeros in U(i,:) */
3459b78a477dSHong Zhang     bjtmp = bj + bdiag[i+1] + 1;
3460b78a477dSHong Zhang     for  (j=0; j<nzu; j++) rtmp[*bjtmp++] = 0.0;
34613c2a7987SHong Zhang 
3462b78a477dSHong Zhang     /* load in initial unfactored row of A */
34636da515a1SHong Zhang     /* printf("row %d\n",i); */
34643c2a7987SHong Zhang     nz    = ai[r[i]+1] - ai[r[i]];
34653c2a7987SHong Zhang     ajtmp = aj + ai[r[i]];
34663c2a7987SHong Zhang     v     = aa + ai[r[i]];
34673c2a7987SHong Zhang     for (j=0; j<nz; j++) {
3468b78a477dSHong Zhang       rtmp[ics[*ajtmp++]] = v[j];
34696da515a1SHong Zhang       /* printf(" (%d,%g),",ics[ajtmp[j]],rtmp[ics[ajtmp[j]]]); */
34703c2a7987SHong Zhang     }
34716da515a1SHong Zhang     /* printf("\n"); */
34723c2a7987SHong Zhang 
3473b78a477dSHong Zhang     /* numerical factorization */
3474b78a477dSHong Zhang     bjtmp = bj + bi[i]; /* point to 1st entry of L(i,:) */
3475b78a477dSHong Zhang     nzl   = bi[i+1] - bi[i]; /* num of entries in L(i,:) */
3476b78a477dSHong Zhang     k = 0;
3477b78a477dSHong Zhang     while (k < nzl){
34783c2a7987SHong Zhang       row   = *bjtmp++;
34796da515a1SHong Zhang       /* printf("  prow %d\n",row); */
34803c2a7987SHong Zhang       pc         = rtmp + row;
3481b78a477dSHong Zhang       pv         = b->a + bdiag[row]; /* 1./(diag of the pivot row) */
3482b78a477dSHong Zhang       multiplier = (*pc) * (*pv);
34833c2a7987SHong Zhang       *pc        = multiplier;
3484b78a477dSHong Zhang       if (PetscAbsScalar(multiplier) > dt){
3485b78a477dSHong Zhang         pj         = bj + bdiag[row+1] + 1; /* point to 1st entry of U(row,:) */
3486b78a477dSHong Zhang         pv         = b->a + bdiag[row+1] + 1;
3487b78a477dSHong Zhang         nz         = bdiag[row] - bdiag[row+1] - 1; /* num of entries in U(row,:), excluding diagonal */
3488b78a477dSHong Zhang         for (j=0; j<nz; j++) rtmp[*pj++] -= multiplier * (*pv++);
34896da515a1SHong Zhang         /* ierr = PetscLogFlops(2.0*nz);CHKERRQ(ierr); */
34903c2a7987SHong Zhang       }
3491b78a477dSHong Zhang       k++;
34923c2a7987SHong Zhang     }
34933c2a7987SHong Zhang 
3494b78a477dSHong Zhang     /* finished row so stick it into b->a */
3495b78a477dSHong Zhang     /* L-part */
3496b78a477dSHong Zhang     pv = b->a + bi[i] ;
3497b78a477dSHong Zhang     pj = bj + bi[i] ;
3498b78a477dSHong Zhang     nzl = bi[i+1] - bi[i];
3499b78a477dSHong Zhang     for (j=0; j<nzl; j++) {
3500b78a477dSHong Zhang       pv[j] = rtmp[pj[j]];
35016da515a1SHong Zhang       /* printf(" (%d,%g),",pj[j],pv[j]); */
35023c2a7987SHong Zhang     }
3503b78a477dSHong Zhang 
3504b78a477dSHong Zhang     /* diagonal: invert diagonal entries for simplier triangular solves */
3505b78a477dSHong Zhang     if (rtmp[i] == 0.0) rtmp[i] = dt+shift;
3506b78a477dSHong Zhang     b->a[bdiag[i]] = 1.0/rtmp[i];
35076da515a1SHong Zhang     /* printf(" (%d,%g),",i,b->a[bdiag[i]]); */
3508b78a477dSHong Zhang 
3509b78a477dSHong Zhang     /* U-part */
3510b78a477dSHong Zhang     pv = b->a + bdiag[i+1] + 1;
3511b78a477dSHong Zhang     pj = bj + bdiag[i+1] + 1;
3512b78a477dSHong Zhang     nzu = bdiag[i] - bdiag[i+1] - 1;
3513b78a477dSHong Zhang     for (j=0; j<nzu; j++) {
3514b78a477dSHong Zhang       pv[j] = rtmp[pj[j]];
35156da515a1SHong Zhang       /* printf(" (%d,%g),",pj[j],pv[j]); */
3516b78a477dSHong Zhang     }
35176da515a1SHong Zhang     /* printf("\n"); */
3518b78a477dSHong Zhang   }
3519b78a477dSHong Zhang 
35203c2a7987SHong Zhang   ierr = PetscFree(rtmp);CHKERRQ(ierr);
35213c2a7987SHong Zhang   ierr = ISRestoreIndices(isicol,&ic);CHKERRQ(ierr);
35223c2a7987SHong Zhang   ierr = ISRestoreIndices(isrow,&r);CHKERRQ(ierr);
3523b78a477dSHong Zhang 
35243c2a7987SHong Zhang   ierr = ISIdentity(isrow,&row_identity);CHKERRQ(ierr);
35253c2a7987SHong Zhang   ierr = ISIdentity(isicol,&col_identity);CHKERRQ(ierr);
35263c2a7987SHong Zhang   if (row_identity && col_identity) {
3527894a2336SHong Zhang     C->ops->solve   = MatSolve_SeqAIJ_NaturalOrdering_newdatastruct;
35283c2a7987SHong Zhang   } else {
3529894a2336SHong Zhang     C->ops->solve   = MatSolve_SeqAIJ_newdatastruct;
35303c2a7987SHong Zhang   }
3531b78a477dSHong Zhang   C->ops->solveadd           = 0;
3532b78a477dSHong Zhang   C->ops->solvetranspose     = 0;
3533b78a477dSHong Zhang   C->ops->solvetransposeadd  = 0;
3534b78a477dSHong Zhang   C->ops->matsolve           = 0;
35353c2a7987SHong Zhang   C->assembled    = PETSC_TRUE;
35363c2a7987SHong Zhang   C->preallocated = PETSC_TRUE;
35373c2a7987SHong Zhang   ierr = PetscLogFlops(C->cmap->n);CHKERRQ(ierr);
35383c2a7987SHong Zhang   PetscFunctionReturn(0);
35393c2a7987SHong Zhang }
3540