xref: /petsc/src/mat/impls/aij/seq/aijfact.c (revision ae068f58437fe7572db20b1072a76b0da1e381df)
1cb512458SBarry Smith #ifndef lint
2*ae068f58SLois Curfman McInnes static char vcid[] = "$Id: aijfact.c,v 1.65 1996/08/08 14:42:46 bsmith Exp curfman $";
3cb512458SBarry Smith #endif
4289bc588SBarry Smith 
570f55243SBarry Smith #include "src/mat/impls/aij/seq/aij.h"
6289bc588SBarry Smith /*
7289bc588SBarry Smith     Factorization code for AIJ format.
8289bc588SBarry Smith */
9289bc588SBarry Smith 
10416022c9SBarry Smith int MatLUFactorSymbolic_SeqAIJ(Mat A,IS isrow,IS iscol,double f,Mat *B)
11289bc588SBarry Smith {
12416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data, *b;
13289bc588SBarry Smith   IS         isicol;
14416022c9SBarry Smith   int        *r,*ic, ierr, i, n = a->m, *ai = a->i, *aj = a->j;
15416022c9SBarry Smith   int        *ainew,*ajnew, jmax,*fill, *ajtmp, nz,shift = a->indexshift;
162fb3ed76SBarry Smith   int        *idnew, idx, row,m,fm, nnz, nzi,len, realloc = 0,nzbd,*im;
17289bc588SBarry Smith 
18d3cbd50cSLois Curfman McInnes   PetscValidHeaderSpecific(isrow,IS_COOKIE);
19d3cbd50cSLois Curfman McInnes   PetscValidHeaderSpecific(iscol,IS_COOKIE);
2078b31e54SBarry Smith   ierr = ISInvertPermutation(iscol,&isicol); CHKERRQ(ierr);
21289bc588SBarry Smith   ISGetIndices(isrow,&r); ISGetIndices(isicol,&ic);
22289bc588SBarry Smith 
23289bc588SBarry Smith   /* get new row pointers */
240452661fSBarry Smith   ainew = (int *) PetscMalloc( (n+1)*sizeof(int) ); CHKPTRQ(ainew);
25dbb450caSBarry Smith   ainew[0] = -shift;
26289bc588SBarry Smith   /* don't know how many column pointers are needed so estimate */
27dbb450caSBarry Smith   jmax = (int) (f*ai[n]+(!shift));
280452661fSBarry Smith   ajnew = (int *) PetscMalloc( (jmax)*sizeof(int) ); CHKPTRQ(ajnew);
29289bc588SBarry Smith   /* fill is a linked list of nonzeros in active row */
300452661fSBarry Smith   fill = (int *) PetscMalloc( (2*n+1)*sizeof(int)); CHKPTRQ(fill);
312fb3ed76SBarry Smith   im = fill + n + 1;
32289bc588SBarry Smith   /* idnew is location of diagonal in factor */
330452661fSBarry Smith   idnew = (int *) PetscMalloc( (n+1)*sizeof(int)); CHKPTRQ(idnew);
34dbb450caSBarry Smith   idnew[0] = -shift;
35289bc588SBarry Smith 
36289bc588SBarry Smith   for ( i=0; i<n; i++ ) {
37289bc588SBarry Smith     /* first copy previous fill into linked list */
38289bc588SBarry Smith     nnz     = nz    = ai[r[i]+1] - ai[r[i]];
39dbb450caSBarry Smith     ajtmp   = aj + ai[r[i]] + shift;
40289bc588SBarry Smith     fill[n] = n;
41289bc588SBarry Smith     while (nz--) {
42289bc588SBarry Smith       fm  = n;
43dbb450caSBarry Smith       idx = ic[*ajtmp++ + shift];
44289bc588SBarry Smith       do {
45289bc588SBarry Smith         m  = fm;
46289bc588SBarry Smith         fm = fill[m];
47289bc588SBarry Smith       } while (fm < idx);
48289bc588SBarry Smith       fill[m]   = idx;
49289bc588SBarry Smith       fill[idx] = fm;
50289bc588SBarry Smith     }
51289bc588SBarry Smith     row = fill[n];
52289bc588SBarry Smith     while ( row < i ) {
53dbb450caSBarry Smith       ajtmp = ajnew + idnew[row] + (!shift);
542fb3ed76SBarry Smith       nzbd  = 1 + idnew[row] - ainew[row];
552fb3ed76SBarry Smith       nz    = im[row] - nzbd;
56289bc588SBarry Smith       fm    = row;
572fb3ed76SBarry Smith       while (nz-- > 0) {
58dbb450caSBarry Smith         idx = *ajtmp++ + shift;
592fb3ed76SBarry Smith         nzbd++;
602fb3ed76SBarry Smith         if (idx == i) im[row] = nzbd;
61289bc588SBarry Smith         do {
62289bc588SBarry Smith           m  = fm;
63289bc588SBarry Smith           fm = fill[m];
64289bc588SBarry Smith         } while (fm < idx);
65289bc588SBarry Smith         if (fm != idx) {
66289bc588SBarry Smith           fill[m]   = idx;
67289bc588SBarry Smith           fill[idx] = fm;
68289bc588SBarry Smith           fm        = idx;
69289bc588SBarry Smith           nnz++;
70289bc588SBarry Smith         }
71289bc588SBarry Smith       }
72289bc588SBarry Smith       row = fill[row];
73289bc588SBarry Smith     }
74289bc588SBarry Smith     /* copy new filled row into permanent storage */
75289bc588SBarry Smith     ainew[i+1] = ainew[i] + nnz;
76496e697eSBarry Smith     if (ainew[i+1] > jmax) {
77289bc588SBarry Smith       /* allocate a longer ajnew */
782fb3ed76SBarry Smith       int maxadd;
79dbb450caSBarry Smith       maxadd = (int) ((f*(ai[n]+(!shift))*(n-i+5))/n);
80bbb6d6a8SBarry Smith       if (maxadd < nnz) maxadd = (n-i)*(nnz+1);
812fb3ed76SBarry Smith       jmax += maxadd;
820452661fSBarry Smith       ajtmp = (int *) PetscMalloc( jmax*sizeof(int) );CHKPTRQ(ajtmp);
83416022c9SBarry Smith       PetscMemcpy(ajtmp,ajnew,(ainew[i]+shift)*sizeof(int));
840452661fSBarry Smith       PetscFree(ajnew);
85289bc588SBarry Smith       ajnew = ajtmp;
862fb3ed76SBarry Smith       realloc++; /* count how many times we realloc */
87289bc588SBarry Smith     }
88dbb450caSBarry Smith     ajtmp = ajnew + ainew[i] + shift;
89289bc588SBarry Smith     fm    = fill[n];
90289bc588SBarry Smith     nzi   = 0;
912fb3ed76SBarry Smith     im[i] = nnz;
92289bc588SBarry Smith     while (nnz--) {
93289bc588SBarry Smith       if (fm < i) nzi++;
94dbb450caSBarry Smith       *ajtmp++ = fm - shift;
95289bc588SBarry Smith       fm       = fill[fm];
96289bc588SBarry Smith     }
97289bc588SBarry Smith     idnew[i] = ainew[i] + nzi;
98289bc588SBarry Smith   }
99289bc588SBarry Smith 
10094a424c1SBarry Smith   PLogInfo(A,
101ec8511deSBarry Smith     "Info:MatLUFactorSymbolic_SeqAIJ:Reallocs %d Fill ratio:given %g needed %g\n",
102bbb6d6a8SBarry Smith                              realloc,f,((double)ainew[n])/((double)ai[i]));
1032fb3ed76SBarry Smith 
104898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r); CHKERRQ(ierr);
105898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isicol,&ic); CHKERRQ(ierr);
1061987afe7SBarry Smith 
1070452661fSBarry Smith   PetscFree(fill);
108289bc588SBarry Smith 
109289bc588SBarry Smith   /* put together the new matrix */
110b4fd4287SBarry Smith   ierr = MatCreateSeqAIJ(A->comm,n,n,0,PETSC_NULL,B); CHKERRQ(ierr);
1111987afe7SBarry Smith   PLogObjectParent(*B,isicol);
1121987afe7SBarry Smith   ierr = ISDestroy(isicol); CHKERRQ(ierr);
113416022c9SBarry Smith   b = (Mat_SeqAIJ *) (*B)->data;
1140452661fSBarry Smith   PetscFree(b->imax);
115416022c9SBarry Smith   b->singlemalloc = 0;
116dbb450caSBarry Smith   len             = (ainew[n] + shift)*sizeof(Scalar);
117e8d4e0b9SBarry Smith   /* the next line frees the default space generated by the Create() */
1180452661fSBarry Smith   PetscFree(b->a); PetscFree(b->ilen);
1190452661fSBarry Smith   b->a          = (Scalar *) PetscMalloc( len ); CHKPTRQ(b->a);
120416022c9SBarry Smith   b->j          = ajnew;
121416022c9SBarry Smith   b->i          = ainew;
122416022c9SBarry Smith   b->diag       = idnew;
123416022c9SBarry Smith   b->ilen       = 0;
124416022c9SBarry Smith   b->imax       = 0;
125416022c9SBarry Smith   b->row        = isrow;
126416022c9SBarry Smith   b->col        = iscol;
1270452661fSBarry Smith   b->solve_work = (Scalar *) PetscMalloc( n*sizeof(Scalar));
128416022c9SBarry Smith   CHKPTRQ(b->solve_work);
129416022c9SBarry Smith   /* In b structure:  Free imax, ilen, old a, old j.
1307fd98bd6SLois Curfman McInnes      Allocate idnew, solve_work, new a, new j */
131416022c9SBarry Smith   PLogObjectMemory(*B,(ainew[n]+shift-n)*(sizeof(int)+sizeof(Scalar)));
132416022c9SBarry Smith   b->maxnz = b->nz = ainew[n] + shift;
1337fd98bd6SLois Curfman McInnes 
134*ae068f58SLois Curfman McInnes   (*B)->info.factor_mallocs    = realloc;
135*ae068f58SLois Curfman McInnes   (*B)->info.fill_ratio_given  = f;
136*ae068f58SLois Curfman McInnes   (*B)->info.fill_ratio_needed = ((double)ainew[n])/((double)ai[i]);
137*ae068f58SLois Curfman McInnes 
138289bc588SBarry Smith   return 0;
139289bc588SBarry Smith }
1400a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
14141c01911SSatish Balay int Mat_AIJ_CheckInode(Mat);
14241c01911SSatish Balay 
143416022c9SBarry Smith int MatLUFactorNumeric_SeqAIJ(Mat A,Mat *B)
144289bc588SBarry Smith {
145416022c9SBarry Smith   Mat        C = *B;
146416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data, *b = (Mat_SeqAIJ *)C->data;
147416022c9SBarry Smith   IS         iscol = b->col, isrow = b->row, isicol;
148416022c9SBarry Smith   int        *r,*ic, ierr, i, j, n = a->m, *ai = b->i, *aj = b->j;
1491987afe7SBarry Smith   int        *ajtmpold, *ajtmp, nz, row, *ics, shift = a->indexshift;
15035aab85fSBarry Smith   int        *diag_offset = b->diag,diag,k;
15135aab85fSBarry Smith   int        preserve_row_sums = (int) a->ilu_preserve_row_sums;
1528ecf7165SBarry Smith   Scalar     *rtmp,*v, *pc, multiplier,sum,inner_sum,*rowsums = 0;
15335aab85fSBarry Smith   double     ssum;
1541987afe7SBarry Smith   /* These declarations are for optimizations.  They reduce the number of
1551987afe7SBarry Smith      memory references that are made by locally storing information; the
1561987afe7SBarry Smith      word "register" used here with pointers can be viewed as "private" or
1571987afe7SBarry Smith      "known only to me"
1581987afe7SBarry Smith    */
15935aab85fSBarry Smith   register Scalar *pv, *rtmps,*u_values;
1601987afe7SBarry Smith   register int    *pj;
161289bc588SBarry Smith 
16278b31e54SBarry Smith   ierr  = ISInvertPermutation(iscol,&isicol); CHKERRQ(ierr);
163416022c9SBarry Smith   PLogObjectParent(*B,isicol);
16478b31e54SBarry Smith   ierr  = ISGetIndices(isrow,&r); CHKERRQ(ierr);
16578b31e54SBarry Smith   ierr  = ISGetIndices(isicol,&ic); CHKERRQ(ierr);
1660452661fSBarry Smith   rtmp  = (Scalar *) PetscMalloc( (n+1)*sizeof(Scalar) ); CHKPTRQ(rtmp);
16713ae1565SBarry Smith   PetscMemzero(rtmp,(n+1)*sizeof(Scalar));
1680cf60383SBarry Smith   rtmps = rtmp + shift; ics = ic + shift;
169289bc588SBarry Smith 
17035aab85fSBarry Smith   /* precalcuate row sums */
17135aab85fSBarry Smith   if (preserve_row_sums) {
17235aab85fSBarry Smith     rowsums = (Scalar *) PetscMalloc( n*sizeof(Scalar) ); CHKPTRQ(rowsums);
17335aab85fSBarry Smith     for ( i=0; i<n; i++ ) {
17435aab85fSBarry Smith       nz  = a->i[r[i]+1] - a->i[r[i]];
17535aab85fSBarry Smith       v   = a->a + a->i[r[i]] + shift;
17635aab85fSBarry Smith       sum = 0.0;
17735aab85fSBarry Smith       for ( j=0; j<nz; j++ ) sum += v[j];
17835aab85fSBarry Smith       rowsums[i] = sum;
17935aab85fSBarry Smith     }
18035aab85fSBarry Smith   }
18135aab85fSBarry Smith 
182289bc588SBarry Smith   for ( i=0; i<n; i++ ) {
183289bc588SBarry Smith     nz    = ai[i+1] - ai[i];
184dbb450caSBarry Smith     ajtmp = aj + ai[i] + shift;
18548e94559SBarry Smith     for  ( j=0; j<nz; j++ ) rtmps[ajtmp[j]] = 0.0;
186289bc588SBarry Smith 
187289bc588SBarry Smith     /* load in initial (unfactored row) */
188416022c9SBarry Smith     nz       = a->i[r[i]+1] - a->i[r[i]];
189416022c9SBarry Smith     ajtmpold = a->j + a->i[r[i]] + shift;
190416022c9SBarry Smith     v        = a->a + a->i[r[i]] + shift;
1910cf60383SBarry Smith     for ( j=0; j<nz; j++ ) rtmp[ics[ajtmpold[j]]] =  v[j];
192289bc588SBarry Smith 
193dbb450caSBarry Smith     row = *ajtmp++ + shift;
194289bc588SBarry Smith     while (row < i) {
1958c37ef55SBarry Smith       pc = rtmp + row;
1968c37ef55SBarry Smith       if (*pc != 0.0) {
1971987afe7SBarry Smith         pv         = b->a + diag_offset[row] + shift;
1981987afe7SBarry Smith         pj         = b->j + diag_offset[row] + (!shift);
19935aab85fSBarry Smith         multiplier = *pc / *pv++;
2008c37ef55SBarry Smith         *pc        = multiplier;
2011987afe7SBarry Smith         nz         = ai[row+1] - diag_offset[row] - 1;
20235aab85fSBarry Smith         for (j=0; j<nz; j++) rtmps[pj[j]] -= multiplier * pv[j];
203eef2e97bSBarry Smith         PLogFlops(2*nz);
204289bc588SBarry Smith       }
205dbb450caSBarry Smith       row = *ajtmp++ + shift;
206289bc588SBarry Smith     }
207416022c9SBarry Smith     /* finished row so stick it into b->a */
208416022c9SBarry Smith     pv = b->a + ai[i] + shift;
209416022c9SBarry Smith     pj = b->j + ai[i] + shift;
2108c37ef55SBarry Smith     nz = ai[i+1] - ai[i];
211416022c9SBarry Smith     for ( j=0; j<nz; j++ ) {pv[j] = rtmps[pj[j]];}
21235aab85fSBarry Smith     diag = diag_offset[i] - ai[i];
21335aab85fSBarry Smith     /*
21435aab85fSBarry Smith           Possibly adjust diagonal entry on current row to force
21535aab85fSBarry Smith         LU matrix to have same row sum as initial matrix.
21635aab85fSBarry Smith     */
21735aab85fSBarry Smith     if (preserve_row_sums) {
21835aab85fSBarry Smith       pj  = b->j + ai[i] + shift;
21935aab85fSBarry Smith       sum = rowsums[i];
22035aab85fSBarry Smith       for ( j=0; j<diag; j++ ) {
22135aab85fSBarry Smith         u_values  = b->a + diag_offset[pj[j]] + shift;
22235aab85fSBarry Smith         nz        = ai[pj[j]+1] - diag_offset[pj[j]];
22335aab85fSBarry Smith         inner_sum = 0.0;
22435aab85fSBarry Smith         for ( k=0; k<nz; k++ ) {
22535aab85fSBarry Smith           inner_sum += u_values[k];
22635aab85fSBarry Smith         }
22735aab85fSBarry Smith         sum -= pv[j]*inner_sum;
22835aab85fSBarry Smith 
22935aab85fSBarry Smith       }
23035aab85fSBarry Smith       nz       = ai[i+1] - diag_offset[i] - 1;
23135aab85fSBarry Smith       u_values = b->a + diag_offset[i] + 1 + shift;
23235aab85fSBarry Smith       for ( k=0; k<nz; k++ ) {
23335aab85fSBarry Smith         sum -= u_values[k];
23435aab85fSBarry Smith       }
23535aab85fSBarry Smith       ssum = PetscAbsScalar(sum/pv[diag]);
23635aab85fSBarry Smith       if (ssum < 1000. && ssum > .001) pv[diag] = sum;
23735aab85fSBarry Smith     }
23835aab85fSBarry Smith     /* check pivot entry for current row */
23935aab85fSBarry Smith     if (pv[diag] == 0.0) {
24035aab85fSBarry Smith       SETERRQ(1,"MatLUFactorNumeric_SeqAIJ:Zero pivot");
24135aab85fSBarry Smith     }
2428c37ef55SBarry Smith   }
2430f11f9aeSSatish Balay 
24435aab85fSBarry Smith   /* invert diagonal entries for simplier triangular solves */
24535aab85fSBarry Smith   for ( i=0; i<n; i++ ) {
24635aab85fSBarry Smith     b->a[diag_offset[i]+shift] = 1.0/b->a[diag_offset[i]+shift];
24735aab85fSBarry Smith   }
24835aab85fSBarry Smith 
24935aab85fSBarry Smith   if (preserve_row_sums) PetscFree(rowsums);
2500452661fSBarry Smith   PetscFree(rtmp);
25178b31e54SBarry Smith   ierr = ISRestoreIndices(isicol,&ic); CHKERRQ(ierr);
25278b31e54SBarry Smith   ierr = ISRestoreIndices(isrow,&r); CHKERRQ(ierr);
25378b31e54SBarry Smith   ierr = ISDestroy(isicol); CHKERRQ(ierr);
254416022c9SBarry Smith   C->factor = FACTOR_LU;
25541c01911SSatish Balay   ierr = Mat_AIJ_CheckInode(C); CHKERRQ(ierr);
256c456f294SBarry Smith   C->assembled = PETSC_TRUE;
257416022c9SBarry Smith   PLogFlops(b->n);
258289bc588SBarry Smith   return 0;
259289bc588SBarry Smith }
2600a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
261416022c9SBarry Smith int MatLUFactor_SeqAIJ(Mat A,IS row,IS col,double f)
262da3a660dSBarry Smith {
263416022c9SBarry Smith   Mat_SeqAIJ *mat = (Mat_SeqAIJ *) A->data;
2646abc6512SBarry Smith   int        ierr;
265416022c9SBarry Smith   Mat        C;
266416022c9SBarry Smith 
267d3cbd50cSLois Curfman McInnes   PetscValidHeaderSpecific(row,IS_COOKIE);
268d3cbd50cSLois Curfman McInnes   PetscValidHeaderSpecific(col,IS_COOKIE);
269416022c9SBarry Smith   ierr = MatLUFactorSymbolic_SeqAIJ(A,row,col,f,&C); CHKERRQ(ierr);
270416022c9SBarry Smith   ierr = MatLUFactorNumeric_SeqAIJ(A,&C); CHKERRQ(ierr);
271da3a660dSBarry Smith 
272da3a660dSBarry Smith   /* free all the data structures from mat */
2730452661fSBarry Smith   PetscFree(mat->a);
2740452661fSBarry Smith   if (!mat->singlemalloc) {PetscFree(mat->i); PetscFree(mat->j);}
2750452661fSBarry Smith   if (mat->diag) PetscFree(mat->diag);
2760452661fSBarry Smith   if (mat->ilen) PetscFree(mat->ilen);
2770452661fSBarry Smith   if (mat->imax) PetscFree(mat->imax);
2780452661fSBarry Smith   if (mat->solve_work) PetscFree(mat->solve_work);
279a7a49059SBarry Smith   if (mat->inode.size) PetscFree(mat->inode.size);
2800452661fSBarry Smith   PetscFree(mat);
281da3a660dSBarry Smith 
282416022c9SBarry Smith   PetscMemcpy(A,C,sizeof(struct _Mat));
2830452661fSBarry Smith   PetscHeaderDestroy(C);
284da3a660dSBarry Smith   return 0;
285da3a660dSBarry Smith }
2860a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
287416022c9SBarry Smith int MatSolve_SeqAIJ(Mat A,Vec bb, Vec xx)
2888c37ef55SBarry Smith {
289416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
290416022c9SBarry Smith   IS         iscol = a->col, isrow = a->row;
291416022c9SBarry Smith   int        *r,*c, ierr, i,  n = a->m, *vi, *ai = a->i, *aj = a->j;
292416022c9SBarry Smith   int        nz,shift = a->indexshift;
293416022c9SBarry Smith   Scalar     *x,*b,*tmp, *tmps, *aa = a->a, sum, *v;
2948c37ef55SBarry Smith 
295416022c9SBarry Smith   if (A->factor != FACTOR_LU) SETERRQ(1,"MatSolve_SeqAIJ:Not for unfactored matrix");
2969e25ed09SBarry Smith 
29778b31e54SBarry Smith   ierr = VecGetArray(bb,&b); CHKERRQ(ierr);
29878b31e54SBarry Smith   ierr = VecGetArray(xx,&x); CHKERRQ(ierr);
299416022c9SBarry Smith   tmp  = a->solve_work;
3008c37ef55SBarry Smith 
30178b31e54SBarry Smith   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
30278b31e54SBarry Smith   ierr = ISGetIndices(iscol,&c);CHKERRQ(ierr); c = c + (n-1);
3038c37ef55SBarry Smith 
3048c37ef55SBarry Smith   /* forward solve the lower triangular */
3058c37ef55SBarry Smith   tmp[0] = b[*r++];
3060cf60383SBarry Smith   tmps   = tmp + shift;
3078c37ef55SBarry Smith   for ( i=1; i<n; i++ ) {
308dbb450caSBarry Smith     v   = aa + ai[i] + shift;
309dbb450caSBarry Smith     vi  = aj + ai[i] + shift;
310416022c9SBarry Smith     nz  = a->diag[i] - ai[i];
3118c37ef55SBarry Smith     sum = b[*r++];
3120cf60383SBarry Smith     while (nz--) sum -= *v++ * tmps[*vi++];
3138c37ef55SBarry Smith     tmp[i] = sum;
3148c37ef55SBarry Smith   }
3158c37ef55SBarry Smith 
3168c37ef55SBarry Smith   /* backward solve the upper triangular */
3178c37ef55SBarry Smith   for ( i=n-1; i>=0; i-- ){
318416022c9SBarry Smith     v   = aa + a->diag[i] + (!shift);
319416022c9SBarry Smith     vi  = aj + a->diag[i] + (!shift);
320416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
3218c37ef55SBarry Smith     sum = tmp[i];
3220cf60383SBarry Smith     while (nz--) sum -= *v++ * tmps[*vi++];
323416022c9SBarry Smith     x[*c--] = tmp[i] = sum*aa[a->diag[i]+shift];
3248c37ef55SBarry Smith   }
3258c37ef55SBarry Smith 
326898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r); CHKERRQ(ierr);
327898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(iscol,&c); CHKERRQ(ierr);
328416022c9SBarry Smith   PLogFlops(2*a->nz - a->n);
3298c37ef55SBarry Smith   return 0;
3308c37ef55SBarry Smith }
331416022c9SBarry Smith int MatSolveAdd_SeqAIJ(Mat A,Vec bb, Vec yy, Vec xx)
332da3a660dSBarry Smith {
333416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
334416022c9SBarry Smith   IS         iscol = a->col, isrow = a->row;
335416022c9SBarry Smith   int        *r,*c, ierr, i,  n = a->m, *vi, *ai = a->i, *aj = a->j;
336416022c9SBarry Smith   int        nz, shift = a->indexshift;
337416022c9SBarry Smith   Scalar     *x,*b,*tmp, *aa = a->a, sum, *v;
338da3a660dSBarry Smith 
339416022c9SBarry Smith   if (A->factor != FACTOR_LU) SETERRQ(1,"MatSolveAdd_SeqAIJ:Not for unfactored matrix");
34078b31e54SBarry Smith   if (yy != xx) {ierr = VecCopy(yy,xx); CHKERRQ(ierr);}
341da3a660dSBarry Smith 
34278b31e54SBarry Smith   ierr = VecGetArray(bb,&b); CHKERRQ(ierr);
34378b31e54SBarry Smith   ierr = VecGetArray(xx,&x); CHKERRQ(ierr);
344416022c9SBarry Smith   tmp  = a->solve_work;
345da3a660dSBarry Smith 
34678b31e54SBarry Smith   ierr = ISGetIndices(isrow,&r); CHKERRQ(ierr);
34778b31e54SBarry Smith   ierr = ISGetIndices(iscol,&c); CHKERRQ(ierr); c = c + (n-1);
348da3a660dSBarry Smith 
349da3a660dSBarry Smith   /* forward solve the lower triangular */
350da3a660dSBarry Smith   tmp[0] = b[*r++];
351da3a660dSBarry Smith   for ( i=1; i<n; i++ ) {
352dbb450caSBarry Smith     v   = aa + ai[i] + shift;
353dbb450caSBarry Smith     vi  = aj + ai[i] + shift;
354416022c9SBarry Smith     nz  = a->diag[i] - ai[i];
355da3a660dSBarry Smith     sum = b[*r++];
356dbb450caSBarry Smith     while (nz--) sum -= *v++ * tmp[*vi++ + shift];
357da3a660dSBarry Smith     tmp[i] = sum;
358da3a660dSBarry Smith   }
359da3a660dSBarry Smith 
360da3a660dSBarry Smith   /* backward solve the upper triangular */
361da3a660dSBarry Smith   for ( i=n-1; i>=0; i-- ){
362416022c9SBarry Smith     v   = aa + a->diag[i] + (!shift);
363416022c9SBarry Smith     vi  = aj + a->diag[i] + (!shift);
364416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
365da3a660dSBarry Smith     sum = tmp[i];
366dbb450caSBarry Smith     while (nz--) sum -= *v++ * tmp[*vi++ + shift];
367416022c9SBarry Smith     tmp[i] = sum*aa[a->diag[i]+shift];
368da3a660dSBarry Smith     x[*c--] += tmp[i];
369da3a660dSBarry Smith   }
370da3a660dSBarry Smith 
371898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r); CHKERRQ(ierr);
372898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(iscol,&c); CHKERRQ(ierr);
373416022c9SBarry Smith   PLogFlops(2*a->nz);
374898183ecSLois Curfman McInnes 
375da3a660dSBarry Smith   return 0;
376da3a660dSBarry Smith }
377da3a660dSBarry Smith /* -------------------------------------------------------------------*/
378416022c9SBarry Smith int MatSolveTrans_SeqAIJ(Mat A,Vec bb, Vec xx)
379da3a660dSBarry Smith {
380416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
381416022c9SBarry Smith   IS         iscol = a->col, isrow = a->row, invisrow,inviscol;
382416022c9SBarry Smith   int        *r,*c, ierr, i, n = a->m, *vi, *ai = a->i, *aj = a->j;
383416022c9SBarry Smith   int        nz,shift = a->indexshift;
384416022c9SBarry Smith   Scalar     *x,*b,*tmp, *aa = a->a, *v;
385da3a660dSBarry Smith 
386416022c9SBarry Smith   if (A->factor != FACTOR_LU)  SETERRQ(1,"MatSolveTrans_SeqAIJ:Not unfactored matrix");
38778b31e54SBarry Smith   ierr = VecGetArray(bb,&b); CHKERRQ(ierr);
38878b31e54SBarry Smith   ierr = VecGetArray(xx,&x); CHKERRQ(ierr);
389416022c9SBarry Smith   tmp  = a->solve_work;
390da3a660dSBarry Smith 
391da3a660dSBarry Smith   /* invert the permutations */
39278b31e54SBarry Smith   ierr = ISInvertPermutation(isrow,&invisrow); CHKERRQ(ierr);
39378b31e54SBarry Smith   ierr = ISInvertPermutation(iscol,&inviscol); CHKERRQ(ierr);
394da3a660dSBarry Smith 
39578b31e54SBarry Smith   ierr = ISGetIndices(invisrow,&r); CHKERRQ(ierr);
39678b31e54SBarry Smith   ierr = ISGetIndices(inviscol,&c); CHKERRQ(ierr);
397da3a660dSBarry Smith 
398da3a660dSBarry Smith   /* copy the b into temp work space according to permutation */
399da3a660dSBarry Smith   for ( i=0; i<n; i++ ) tmp[c[i]] = b[i];
400da3a660dSBarry Smith 
401da3a660dSBarry Smith   /* forward solve the U^T */
402da3a660dSBarry Smith   for ( i=0; i<n; i++ ) {
403416022c9SBarry Smith     v   = aa + a->diag[i] + shift;
404416022c9SBarry Smith     vi  = aj + a->diag[i] + (!shift);
405416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
406da3a660dSBarry Smith     tmp[i] *= *v++;
407da3a660dSBarry Smith     while (nz--) {
408dbb450caSBarry Smith       tmp[*vi++ + shift] -= (*v++)*tmp[i];
409da3a660dSBarry Smith     }
410da3a660dSBarry Smith   }
411da3a660dSBarry Smith 
412da3a660dSBarry Smith   /* backward solve the L^T */
413da3a660dSBarry Smith   for ( i=n-1; i>=0; i-- ){
414416022c9SBarry Smith     v   = aa + a->diag[i] - 1 + shift;
415416022c9SBarry Smith     vi  = aj + a->diag[i] - 1 + shift;
416416022c9SBarry Smith     nz  = a->diag[i] - ai[i];
417da3a660dSBarry Smith     while (nz--) {
418dbb450caSBarry Smith       tmp[*vi-- + shift] -= (*v--)*tmp[i];
419da3a660dSBarry Smith     }
420da3a660dSBarry Smith   }
421da3a660dSBarry Smith 
422da3a660dSBarry Smith   /* copy tmp into x according to permutation */
423da3a660dSBarry Smith   for ( i=0; i<n; i++ ) x[r[i]] = tmp[i];
424da3a660dSBarry Smith 
425898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(invisrow,&r); CHKERRQ(ierr);
426898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(inviscol,&c); CHKERRQ(ierr);
427898183ecSLois Curfman McInnes   ierr = ISDestroy(invisrow); CHKERRQ(ierr);
428898183ecSLois Curfman McInnes   ierr = ISDestroy(inviscol); CHKERRQ(ierr);
429da3a660dSBarry Smith 
430416022c9SBarry Smith   PLogFlops(2*a->nz-a->n);
431da3a660dSBarry Smith   return 0;
432da3a660dSBarry Smith }
433da3a660dSBarry Smith 
434416022c9SBarry Smith int MatSolveTransAdd_SeqAIJ(Mat A,Vec bb, Vec zz,Vec xx)
435da3a660dSBarry Smith {
436416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data;
437416022c9SBarry Smith   IS         iscol = a->col, isrow = a->row, invisrow,inviscol;
438416022c9SBarry Smith   int        *r,*c, ierr, i, n = a->m, *vi, *ai = a->i, *aj = a->j;
439416022c9SBarry Smith   int        nz,shift = a->indexshift;
440416022c9SBarry Smith   Scalar     *x,*b,*tmp, *aa = a->a, *v;
4416abc6512SBarry Smith 
442416022c9SBarry Smith   if (A->factor != FACTOR_LU)SETERRQ(1,"MatSolveTransAdd_SeqAIJ:Not unfactored matrix");
4436abc6512SBarry Smith   if (zz != xx) VecCopy(zz,xx);
4446abc6512SBarry Smith 
44578b31e54SBarry Smith   ierr = VecGetArray(bb,&b); CHKERRQ(ierr);
44678b31e54SBarry Smith   ierr = VecGetArray(xx,&x); CHKERRQ(ierr);
447416022c9SBarry Smith   tmp = a->solve_work;
4486abc6512SBarry Smith 
4496abc6512SBarry Smith   /* invert the permutations */
45078b31e54SBarry Smith   ierr = ISInvertPermutation(isrow,&invisrow); CHKERRQ(ierr);
45178b31e54SBarry Smith   ierr = ISInvertPermutation(iscol,&inviscol); CHKERRQ(ierr);
45278b31e54SBarry Smith   ierr = ISGetIndices(invisrow,&r); CHKERRQ(ierr);
45378b31e54SBarry Smith   ierr = ISGetIndices(inviscol,&c); CHKERRQ(ierr);
4546abc6512SBarry Smith 
4556abc6512SBarry Smith   /* copy the b into temp work space according to permutation */
4566abc6512SBarry Smith   for ( i=0; i<n; i++ ) tmp[c[i]] = b[i];
4576abc6512SBarry Smith 
4586abc6512SBarry Smith   /* forward solve the U^T */
4596abc6512SBarry Smith   for ( i=0; i<n; i++ ) {
460416022c9SBarry Smith     v   = aa + a->diag[i] + shift;
461416022c9SBarry Smith     vi  = aj + a->diag[i] + (!shift);
462416022c9SBarry Smith     nz  = ai[i+1] - a->diag[i] - 1;
4636abc6512SBarry Smith     tmp[i] *= *v++;
4646abc6512SBarry Smith     while (nz--) {
465dbb450caSBarry Smith       tmp[*vi++ + shift] -= (*v++)*tmp[i];
4666abc6512SBarry Smith     }
4676abc6512SBarry Smith   }
4686abc6512SBarry Smith 
4696abc6512SBarry Smith   /* backward solve the L^T */
4706abc6512SBarry Smith   for ( i=n-1; i>=0; i-- ){
471416022c9SBarry Smith     v   = aa + a->diag[i] - 1 + shift;
472416022c9SBarry Smith     vi  = aj + a->diag[i] - 1 + shift;
473416022c9SBarry Smith     nz  = a->diag[i] - ai[i];
4746abc6512SBarry Smith     while (nz--) {
475dbb450caSBarry Smith       tmp[*vi-- + shift] -= (*v--)*tmp[i];
4766abc6512SBarry Smith     }
4776abc6512SBarry Smith   }
4786abc6512SBarry Smith 
4796abc6512SBarry Smith   /* copy tmp into x according to permutation */
4806abc6512SBarry Smith   for ( i=0; i<n; i++ ) x[r[i]] += tmp[i];
4816abc6512SBarry Smith 
482898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(invisrow,&r); CHKERRQ(ierr);
483898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(inviscol,&c); CHKERRQ(ierr);
484898183ecSLois Curfman McInnes   ierr = ISDestroy(invisrow); CHKERRQ(ierr);
485898183ecSLois Curfman McInnes   ierr = ISDestroy(inviscol); CHKERRQ(ierr);
4866abc6512SBarry Smith 
487416022c9SBarry Smith   PLogFlops(2*a->nz);
4886abc6512SBarry Smith   return 0;
489da3a660dSBarry Smith }
4909e25ed09SBarry Smith /* ----------------------------------------------------------------*/
49108480c60SBarry Smith 
49208480c60SBarry Smith int MatILUFactorSymbolic_SeqAIJ(Mat A,IS isrow,IS iscol,double f,int levels,Mat *fact)
4939e25ed09SBarry Smith {
494416022c9SBarry Smith   Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data, *b;
4959e25ed09SBarry Smith   IS         isicol;
496416022c9SBarry Smith   int        *r,*ic, ierr, prow, n = a->m, *ai = a->i, *aj = a->j;
49756beaf04SBarry Smith   int        *ainew,*ajnew, jmax,*fill, *xi, nz, *im,*ajfill,*flev;
49856beaf04SBarry Smith   int        *dloc, idx, row,m,fm, nzf, nzi,len,  realloc = 0;
499416022c9SBarry Smith   int        incrlev,nnz,i,shift = a->indexshift;
50077c4ece6SBarry Smith   PetscTruth col_identity, row_identity;
5019e25ed09SBarry Smith 
50208480c60SBarry Smith   /* special case that simply copies fill pattern */
50377c4ece6SBarry Smith   ISIdentity(isrow,&row_identity); ISIdentity(iscol,&col_identity);
50477c4ece6SBarry Smith   if (levels == 0 && row_identity && col_identity) {
5053d1612f7SBarry Smith     ierr = MatConvertSameType_SeqAIJ(A,fact,DO_NOT_COPY_VALUES); CHKERRQ(ierr);
50608480c60SBarry Smith     (*fact)->factor = FACTOR_LU;
50708480c60SBarry Smith     b               = (Mat_SeqAIJ *) (*fact)->data;
50808480c60SBarry Smith     if (!b->diag) {
50908480c60SBarry Smith       ierr = MatMarkDiag_SeqAIJ(*fact); CHKERRQ(ierr);
51008480c60SBarry Smith     }
51108480c60SBarry Smith     b->row          = isrow;
51208480c60SBarry Smith     b->col          = iscol;
5130452661fSBarry Smith     b->solve_work = (Scalar *) PetscMalloc((b->m+1)*sizeof(Scalar));CHKPTRQ(b->solve_work);
51408480c60SBarry Smith     return 0;
51508480c60SBarry Smith   }
51608480c60SBarry Smith 
51778b31e54SBarry Smith   ierr = ISInvertPermutation(iscol,&isicol); CHKERRQ(ierr);
518898183ecSLois Curfman McInnes   ierr = ISGetIndices(isrow,&r); CHKERRQ(ierr);
519898183ecSLois Curfman McInnes   ierr = ISGetIndices(isicol,&ic); CHKERRQ(ierr);
5209e25ed09SBarry Smith 
5219e25ed09SBarry Smith   /* get new row pointers */
5220452661fSBarry Smith   ainew = (int *) PetscMalloc( (n+1)*sizeof(int) ); CHKPTRQ(ainew);
523dbb450caSBarry Smith   ainew[0] = -shift;
5249e25ed09SBarry Smith   /* don't know how many column pointers are needed so estimate */
525dbb450caSBarry Smith   jmax = (int) (f*(ai[n]+!shift));
5260452661fSBarry Smith   ajnew = (int *) PetscMalloc( (jmax)*sizeof(int) ); CHKPTRQ(ajnew);
5279e25ed09SBarry Smith   /* ajfill is level of fill for each fill entry */
5280452661fSBarry Smith   ajfill = (int *) PetscMalloc( (jmax)*sizeof(int) ); CHKPTRQ(ajfill);
5299e25ed09SBarry Smith   /* fill is a linked list of nonzeros in active row */
5300452661fSBarry Smith   fill = (int *) PetscMalloc( (n+1)*sizeof(int)); CHKPTRQ(fill);
53156beaf04SBarry Smith   /* im is level for each filled value */
5320452661fSBarry Smith   im = (int *) PetscMalloc( (n+1)*sizeof(int)); CHKPTRQ(im);
53356beaf04SBarry Smith   /* dloc is location of diagonal in factor */
5340452661fSBarry Smith   dloc = (int *) PetscMalloc( (n+1)*sizeof(int)); CHKPTRQ(dloc);
53556beaf04SBarry Smith   dloc[0]  = 0;
53656beaf04SBarry Smith   for ( prow=0; prow<n; prow++ ) {
5379e25ed09SBarry Smith     /* first copy previous fill into linked list */
53856beaf04SBarry Smith     nzf     = nz  = ai[r[prow]+1] - ai[r[prow]];
539dbb450caSBarry Smith     xi      = aj + ai[r[prow]] + shift;
5409e25ed09SBarry Smith     fill[n] = n;
5419e25ed09SBarry Smith     while (nz--) {
5429e25ed09SBarry Smith       fm  = n;
543dbb450caSBarry Smith       idx = ic[*xi++ + shift];
5449e25ed09SBarry Smith       do {
5459e25ed09SBarry Smith         m  = fm;
5469e25ed09SBarry Smith         fm = fill[m];
5479e25ed09SBarry Smith       } while (fm < idx);
5489e25ed09SBarry Smith       fill[m]   = idx;
5499e25ed09SBarry Smith       fill[idx] = fm;
55056beaf04SBarry Smith       im[idx]   = 0;
5519e25ed09SBarry Smith     }
55256beaf04SBarry Smith     nzi = 0;
5539e25ed09SBarry Smith     row = fill[n];
55456beaf04SBarry Smith     while ( row < prow ) {
55556beaf04SBarry Smith       incrlev = im[row] + 1;
55656beaf04SBarry Smith       nz      = dloc[row];
557dbb450caSBarry Smith       xi      = ajnew  + ainew[row] + shift + nz;
558dbb450caSBarry Smith       flev    = ajfill + ainew[row] + shift + nz + 1;
559416022c9SBarry Smith       nnz     = ainew[row+1] - ainew[row] - nz - 1;
560dbb450caSBarry Smith       if (*xi++ + shift != row) {
561ec8511deSBarry Smith         SETERRQ(1,"MatILUFactorSymbolic_SeqAIJ:zero pivot");
56256beaf04SBarry Smith       }
56356beaf04SBarry Smith       fm      = row;
56456beaf04SBarry Smith       while (nnz-- > 0) {
565dbb450caSBarry Smith         idx = *xi++ + shift;
56656beaf04SBarry Smith         if (*flev + incrlev > levels) {
56756beaf04SBarry Smith           flev++;
56856beaf04SBarry Smith           continue;
56956beaf04SBarry Smith         }
57056beaf04SBarry Smith         do {
5719e25ed09SBarry Smith           m  = fm;
5729e25ed09SBarry Smith           fm = fill[m];
57356beaf04SBarry Smith         } while (fm < idx);
5749e25ed09SBarry Smith         if (fm != idx) {
57556beaf04SBarry Smith           im[idx]   = *flev + incrlev;
5769e25ed09SBarry Smith           fill[m]   = idx;
5779e25ed09SBarry Smith           fill[idx] = fm;
5789e25ed09SBarry Smith           fm        = idx;
57956beaf04SBarry Smith           nzf++;
5809e25ed09SBarry Smith         }
58156beaf04SBarry Smith         else {
58256beaf04SBarry Smith           if (im[idx] > *flev + incrlev) im[idx] = *flev+incrlev;
5839e25ed09SBarry Smith         }
58456beaf04SBarry Smith         flev++;
5859e25ed09SBarry Smith       }
5869e25ed09SBarry Smith       row = fill[row];
58756beaf04SBarry Smith       nzi++;
5889e25ed09SBarry Smith     }
5899e25ed09SBarry Smith     /* copy new filled row into permanent storage */
59056beaf04SBarry Smith     ainew[prow+1] = ainew[prow] + nzf;
591d7e8b826SBarry Smith     if (ainew[prow+1] > jmax-shift) {
5929e25ed09SBarry Smith       /* allocate a longer ajnew */
593bbb6d6a8SBarry Smith       int maxadd;
594dbb450caSBarry Smith       maxadd = (int) ((f*(ai[n]+!shift)*(n-prow+5))/n);
59556beaf04SBarry Smith       if (maxadd < nzf) maxadd = (n-prow)*(nzf+1);
596bbb6d6a8SBarry Smith       jmax += maxadd;
5970452661fSBarry Smith       xi = (int *) PetscMalloc( jmax*sizeof(int) );CHKPTRQ(xi);
598416022c9SBarry Smith       PetscMemcpy(xi,ajnew,(ainew[prow]+shift)*sizeof(int));
5990452661fSBarry Smith       PetscFree(ajnew);
60056beaf04SBarry Smith       ajnew = xi;
6019e25ed09SBarry Smith       /* allocate a longer ajfill */
6020452661fSBarry Smith       xi = (int *) PetscMalloc( jmax*sizeof(int) );CHKPTRQ(xi);
603416022c9SBarry Smith       PetscMemcpy(xi,ajfill,(ainew[prow]+shift)*sizeof(int));
6040452661fSBarry Smith       PetscFree(ajfill);
60556beaf04SBarry Smith       ajfill = xi;
606bbb6d6a8SBarry Smith       realloc++;
6079e25ed09SBarry Smith     }
608dbb450caSBarry Smith     xi          = ajnew + ainew[prow] + shift;
609dbb450caSBarry Smith     flev        = ajfill + ainew[prow] + shift;
61056beaf04SBarry Smith     dloc[prow]  = nzi;
6119e25ed09SBarry Smith     fm          = fill[n];
61256beaf04SBarry Smith     while (nzf--) {
613dbb450caSBarry Smith       *xi++   = fm - shift;
61456beaf04SBarry Smith       *flev++ = im[fm];
6159e25ed09SBarry Smith       fm      = fill[fm];
6169e25ed09SBarry Smith     }
6179e25ed09SBarry Smith   }
6180452661fSBarry Smith   PetscFree(ajfill);
619898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r); CHKERRQ(ierr);
620898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isicol,&ic); CHKERRQ(ierr);
621898183ecSLois Curfman McInnes   ierr = ISDestroy(isicol); CHKERRQ(ierr);
6220452661fSBarry Smith   PetscFree(fill); PetscFree(im);
6239e25ed09SBarry Smith 
62494a424c1SBarry Smith   PLogInfo(A,
625ec8511deSBarry Smith     "Info:MatILUFactorSymbolic_SeqAIJ:Realloc %d Fill ratio:given %g needed %g\n",
62656beaf04SBarry Smith                              realloc,f,((double)ainew[n])/((double)ai[prow]));
627bbb6d6a8SBarry Smith 
6289e25ed09SBarry Smith   /* put together the new matrix */
629b4fd4287SBarry Smith   ierr = MatCreateSeqAIJ(A->comm,n,n,0,PETSC_NULL,fact); CHKERRQ(ierr);
630416022c9SBarry Smith   b = (Mat_SeqAIJ *) (*fact)->data;
6310452661fSBarry Smith   PetscFree(b->imax);
632416022c9SBarry Smith   b->singlemalloc = 0;
633dbb450caSBarry Smith   len = (ainew[n] + shift)*sizeof(Scalar);
6349e25ed09SBarry Smith   /* the next line frees the default space generated by the Create() */
6350452661fSBarry Smith   PetscFree(b->a); PetscFree(b->ilen);
6360452661fSBarry Smith   b->a          = (Scalar *) PetscMalloc( len ); CHKPTRQ(b->a);
637416022c9SBarry Smith   b->j          = ajnew;
638416022c9SBarry Smith   b->i          = ainew;
63956beaf04SBarry Smith   for ( i=0; i<n; i++ ) dloc[i] += ainew[i];
640416022c9SBarry Smith   b->diag       = dloc;
641416022c9SBarry Smith   b->ilen       = 0;
642416022c9SBarry Smith   b->imax       = 0;
643416022c9SBarry Smith   b->row        = isrow;
644416022c9SBarry Smith   b->col        = iscol;
6450452661fSBarry Smith   b->solve_work = (Scalar *) PetscMalloc( (n+1)*sizeof(Scalar));
646416022c9SBarry Smith   CHKPTRQ(b->solve_work);
647416022c9SBarry Smith   /* In b structure:  Free imax, ilen, old a, old j.
64856beaf04SBarry Smith      Allocate dloc, solve_work, new a, new j */
649dbb450caSBarry Smith   PLogObjectMemory(*fact,(ainew[n]+shift-n) * (sizeof(int)+sizeof(Scalar)));
650416022c9SBarry Smith   b->maxnz          = b->nz = ainew[n] + shift;
6519e25ed09SBarry Smith   (*fact)->factor   = FACTOR_LU;
652*ae068f58SLois Curfman McInnes 
653*ae068f58SLois Curfman McInnes   (*fact)->info.factor_mallocs    = realloc;
654*ae068f58SLois Curfman McInnes   (*fact)->info.fill_ratio_given  = f;
655*ae068f58SLois Curfman McInnes   (*fact)->info.fill_ratio_needed = ((double)ainew[n])/((double)ai[prow]);
656*ae068f58SLois Curfman McInnes 
6579e25ed09SBarry Smith   return 0;
6589e25ed09SBarry Smith }
659d5d45c9bSBarry Smith 
660d5d45c9bSBarry Smith 
661d5d45c9bSBarry Smith 
662d5d45c9bSBarry Smith 
663