xref: /petsc/src/mat/impls/aij/seq/aijfact.c (revision dbb450ca0a1bd4568a196678f20f5151425d3a04)
1cb512458SBarry Smith #ifndef lint
2*dbb450caSBarry Smith static char vcid[] = "$Id: aijfact.c,v 1.37 1995/09/12 03:25:17 bsmith Exp bsmith $";
3cb512458SBarry Smith #endif
4289bc588SBarry Smith 
5289bc588SBarry Smith 
6289bc588SBarry Smith #include "aij.h"
78c37ef55SBarry Smith #include "inline/spops.h"
8289bc588SBarry Smith /*
9289bc588SBarry Smith     Factorization code for AIJ format.
10289bc588SBarry Smith */
11289bc588SBarry Smith 
12ec8511deSBarry Smith int MatLUFactorSymbolic_SeqAIJ(Mat mat,IS isrow,IS iscol,double f,Mat *fact)
13289bc588SBarry Smith {
14ec8511deSBarry Smith   Mat_SeqAIJ *aij = (Mat_SeqAIJ *) mat->data, *aijnew;
15289bc588SBarry Smith   IS         isicol;
166abc6512SBarry Smith   int        *r,*ic, ierr, i, n = aij->m, *ai = aij->i, *aj = aij->j;
176abc6512SBarry Smith   int        *ainew,*ajnew, jmax,*fill, *ajtmp, nz;
182fb3ed76SBarry Smith   int        *idnew, idx, row,m,fm, nnz, nzi,len, realloc = 0,nzbd,*im;
19*dbb450caSBarry Smith   int        shift = aij->indexshift;
20289bc588SBarry Smith 
21d35516d3SLois Curfman McInnes   if (n != aij->n)
22ec8511deSBarry Smith     SETERRQ(1,"MatLUFactorSymbolic_SeqAIJ:Matrix must be square");
23d35516d3SLois Curfman McInnes   if (!isrow)
24ec8511deSBarry Smith     SETERRQ(1,"MatLUFactorSymbolic_SeqAIJ:Matrix must have row permutation");
25d35516d3SLois Curfman McInnes   if (!iscol)
26ec8511deSBarry Smith     SETERRQ(1,"MatLUFactorSymbolic_SeqAIJ:Matrix must have column permutation");
27289bc588SBarry Smith 
2878b31e54SBarry Smith   ierr = ISInvertPermutation(iscol,&isicol); CHKERRQ(ierr);
29289bc588SBarry Smith   ISGetIndices(isrow,&r); ISGetIndices(isicol,&ic);
30289bc588SBarry Smith 
31289bc588SBarry Smith   /* get new row pointers */
3278b31e54SBarry Smith   ainew = (int *) PETSCMALLOC( (n+1)*sizeof(int) ); CHKPTRQ(ainew);
33*dbb450caSBarry Smith   ainew[0] = -shift;
34289bc588SBarry Smith   /* don't know how many column pointers are needed so estimate */
35*dbb450caSBarry Smith   jmax = (int) (f*ai[n]+(!shift));
3678b31e54SBarry Smith   ajnew = (int *) PETSCMALLOC( (jmax)*sizeof(int) ); CHKPTRQ(ajnew);
37289bc588SBarry Smith   /* fill is a linked list of nonzeros in active row */
382fb3ed76SBarry Smith   fill = (int *) PETSCMALLOC( (2*n+1)*sizeof(int)); CHKPTRQ(fill);
392fb3ed76SBarry Smith   im = fill + n + 1;
40289bc588SBarry Smith   /* idnew is location of diagonal in factor */
4178b31e54SBarry Smith   idnew = (int *) PETSCMALLOC( (n+1)*sizeof(int)); CHKPTRQ(idnew);
42*dbb450caSBarry Smith   idnew[0] = -shift;
43289bc588SBarry Smith 
44289bc588SBarry Smith   for ( i=0; i<n; i++ ) {
45289bc588SBarry Smith     /* first copy previous fill into linked list */
46289bc588SBarry Smith     nnz = nz    = ai[r[i]+1] - ai[r[i]];
47*dbb450caSBarry Smith     ajtmp = aj + ai[r[i]] + shift;
48289bc588SBarry Smith     fill[n] = n;
49289bc588SBarry Smith     while (nz--) {
50289bc588SBarry Smith       fm = n;
51*dbb450caSBarry Smith       idx = ic[*ajtmp++ + shift];
52289bc588SBarry Smith       do {
53289bc588SBarry Smith         m = fm;
54289bc588SBarry Smith         fm = fill[m];
55289bc588SBarry Smith       } while (fm < idx);
56289bc588SBarry Smith       fill[m] = idx;
57289bc588SBarry Smith       fill[idx] = fm;
58289bc588SBarry Smith     }
59289bc588SBarry Smith     row = fill[n];
60289bc588SBarry Smith     while ( row < i ) {
61*dbb450caSBarry Smith       ajtmp = ajnew + idnew[row] + (!shift);
622fb3ed76SBarry Smith       nzbd = 1 + idnew[row] - ainew[row];
632fb3ed76SBarry Smith       nz = im[row] - nzbd;
64289bc588SBarry Smith       fm = row;
652fb3ed76SBarry Smith       while (nz-- > 0) {
662fb3ed76SBarry Smith         /* fm = n;  */
67*dbb450caSBarry Smith         idx = *ajtmp++ + shift;
682fb3ed76SBarry Smith         nzbd++;
692fb3ed76SBarry Smith         if (idx == i) im[row] = nzbd;
70289bc588SBarry Smith         do {
71289bc588SBarry Smith           m = fm;
72289bc588SBarry Smith           fm = fill[m];
73289bc588SBarry Smith         } while (fm < idx);
74289bc588SBarry Smith         if (fm != idx) {
75289bc588SBarry Smith           fill[m] = idx;
76289bc588SBarry Smith           fill[idx] = fm;
77289bc588SBarry Smith           fm = idx;
78289bc588SBarry Smith           nnz++;
79289bc588SBarry Smith         }
80d5f53d93SBarry Smith /*  printf("i %d row %d nz %d idx %d fm %d\n",i,row,nz,idx,fm);  */
81289bc588SBarry Smith       }
82289bc588SBarry Smith       row = fill[row];
83289bc588SBarry Smith     }
84289bc588SBarry Smith     /* copy new filled row into permanent storage */
85289bc588SBarry Smith     ainew[i+1] = ainew[i] + nnz;
86289bc588SBarry Smith     if (ainew[i+1] > jmax+1) {
87289bc588SBarry Smith       /* allocate a longer ajnew */
882fb3ed76SBarry Smith       int maxadd;
89*dbb450caSBarry Smith       maxadd = (int) ((f*(ai[n]+(!shift))*(n-i+5))/n);
90bbb6d6a8SBarry Smith       if (maxadd < nnz) maxadd = (n-i)*(nnz+1);
912fb3ed76SBarry Smith       jmax += maxadd;
9278b31e54SBarry Smith       ajtmp = (int *) PETSCMALLOC( jmax*sizeof(int) );CHKPTRQ(ajtmp);
93*dbb450caSBarry Smith       PETSCMEMCPY(ajtmp,ajnew,(ainew[i]+shift)*sizeof(int));
9478b31e54SBarry Smith       PETSCFREE(ajnew);
95289bc588SBarry Smith       ajnew = ajtmp;
962fb3ed76SBarry Smith       realloc++; /* count how many times we realloc */
97289bc588SBarry Smith     }
98*dbb450caSBarry Smith     ajtmp = ajnew + ainew[i] + shift;
99289bc588SBarry Smith     fm = fill[n];
100289bc588SBarry Smith     nzi = 0;
1012fb3ed76SBarry Smith     im[i] = nnz;
102289bc588SBarry Smith     while (nnz--) {
103289bc588SBarry Smith       if (fm < i) nzi++;
104*dbb450caSBarry Smith       *ajtmp++ = fm - shift;
105289bc588SBarry Smith       fm = fill[fm];
106289bc588SBarry Smith     }
107289bc588SBarry Smith     idnew[i] = ainew[i] + nzi;
108289bc588SBarry Smith   }
109289bc588SBarry Smith 
1102fb3ed76SBarry Smith   PLogInfo((PetscObject)mat,
111ec8511deSBarry Smith     "Info:MatLUFactorSymbolic_SeqAIJ:Reallocs %d Fill ratio:given %g needed %g\n",
112bbb6d6a8SBarry Smith                              realloc,f,((double)ainew[n])/((double)ai[i]));
1132fb3ed76SBarry Smith 
114898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r); CHKERRQ(ierr);
115898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isicol,&ic); CHKERRQ(ierr);
116898183ecSLois Curfman McInnes   ierr = ISDestroy(isicol); CHKERRQ(ierr);
117898183ecSLois Curfman McInnes   PETSCFREE(fill);
118289bc588SBarry Smith 
119289bc588SBarry Smith   /* put together the new matrix */
120fafbff53SBarry Smith   ierr = MatCreateSeqAIJ(mat->comm,n, n, 0, 0, fact); CHKERRQ(ierr);
121ec8511deSBarry Smith   aijnew = (Mat_SeqAIJ *) (*fact)->data;
12278b31e54SBarry Smith   PETSCFREE(aijnew->imax);
123289bc588SBarry Smith   aijnew->singlemalloc = 0;
124*dbb450caSBarry Smith   len = (ainew[n] + shift)*sizeof(Scalar);
125e8d4e0b9SBarry Smith   /* the next line frees the default space generated by the Create() */
12678b31e54SBarry Smith   PETSCFREE(aijnew->a); PETSCFREE(aijnew->ilen);
12778b31e54SBarry Smith   aijnew->a          = (Scalar *) PETSCMALLOC( len ); CHKPTRQ(aijnew->a);
128289bc588SBarry Smith   aijnew->j          = ajnew;
129289bc588SBarry Smith   aijnew->i          = ainew;
1308c37ef55SBarry Smith   aijnew->diag       = idnew;
131e8d4e0b9SBarry Smith   aijnew->ilen       = 0;
13220563c6bSBarry Smith   aijnew->imax       = 0;
13386d7a8ceSBarry Smith   aijnew->row        = isrow;
13486d7a8ceSBarry Smith   aijnew->col        = iscol;
13578b31e54SBarry Smith   aijnew->solve_work = (Scalar *) PETSCMALLOC( n*sizeof(Scalar));
13678b31e54SBarry Smith   CHKPTRQ(aijnew->solve_work);
1377fd98bd6SLois Curfman McInnes   /* In aijnew structure:  Free imax, ilen, old a, old j.
1387fd98bd6SLois Curfman McInnes      Allocate idnew, solve_work, new a, new j */
139*dbb450caSBarry Smith   PLogObjectMemory(*fact,(ainew[n]+shift-n)*(sizeof(int)+sizeof(Scalar)));
140*dbb450caSBarry Smith   aijnew->maxnz = aijnew->nz = ainew[n] + shift;
1417fd98bd6SLois Curfman McInnes 
1425e42470aSBarry Smith   /* Cannot do this here because child is destroyed before parent created
1435e42470aSBarry Smith      PLogObjectParent(*fact,isicol); */
144289bc588SBarry Smith   return 0;
145289bc588SBarry Smith }
1460a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
147ec8511deSBarry Smith int MatLUFactorNumeric_SeqAIJ(Mat mat,Mat *infact)
148289bc588SBarry Smith {
14920563c6bSBarry Smith   Mat         fact = *infact;
150ec8511deSBarry Smith   Mat_SeqAIJ *aij = (Mat_SeqAIJ *) mat->data, *aijnew = (Mat_SeqAIJ *)fact->data;
15186d7a8ceSBarry Smith   IS      iscol = aijnew->col, isrow = aijnew->row, isicol;
152289bc588SBarry Smith   int     *r,*ic, ierr, i, j, n = aij->m, *ai = aijnew->i, *aj = aijnew->j;
1536abc6512SBarry Smith   int     *ajtmpold, *ajtmp, nz, row,*pj;
1546abc6512SBarry Smith   Scalar  *rtmp,*v, *pv, *pc, multiplier;
155*dbb450caSBarry Smith   int     shift = aij->indexshift;
156289bc588SBarry Smith 
15778b31e54SBarry Smith   ierr = ISInvertPermutation(iscol,&isicol); CHKERRQ(ierr);
1585e42470aSBarry Smith   PLogObjectParent(*infact,isicol);
15978b31e54SBarry Smith   ierr = ISGetIndices(isrow,&r); CHKERRQ(ierr);
16078b31e54SBarry Smith   ierr = ISGetIndices(isicol,&ic); CHKERRQ(ierr);
16178b31e54SBarry Smith   rtmp = (Scalar *) PETSCMALLOC( (n+1)*sizeof(Scalar) ); CHKPTRQ(rtmp);
162289bc588SBarry Smith 
163289bc588SBarry Smith   for ( i=0; i<n; i++ ) {
164289bc588SBarry Smith     nz = ai[i+1] - ai[i];
165*dbb450caSBarry Smith     ajtmp = aj + ai[i] + shift;
166*dbb450caSBarry Smith     for  ( j=0; j<nz; j++ ) rtmp[ajtmp[j]+shift] = 0.0;
167289bc588SBarry Smith 
168289bc588SBarry Smith     /* load in initial (unfactored row) */
1698c37ef55SBarry Smith     nz = aij->i[r[i]+1] - aij->i[r[i]];
170*dbb450caSBarry Smith     ajtmpold = aij->j + aij->i[r[i]] + shift;
171*dbb450caSBarry Smith     v  = aij->a + aij->i[r[i]] + shift;
172*dbb450caSBarry Smith     for ( j=0; j<nz; j++ ) rtmp[ic[ajtmpold[j]+shift]] =  v[j];
173289bc588SBarry Smith 
174*dbb450caSBarry Smith     row = *ajtmp++ + shift;
175289bc588SBarry Smith     while (row < i) {
1768c37ef55SBarry Smith       pc = rtmp + row;
1778c37ef55SBarry Smith       if (*pc != 0.0) {
1788c37ef55SBarry Smith         nz = aijnew->diag[row] - ai[row];
179*dbb450caSBarry Smith         pv = aijnew->a + aijnew->diag[row] + shift;
180*dbb450caSBarry Smith         pj = aijnew->j + aijnew->diag[row] + (!shift);
1818c37ef55SBarry Smith         multiplier = *pc * *pv++;
1828c37ef55SBarry Smith         *pc = multiplier;
1838c37ef55SBarry Smith         nz = ai[row+1] - ai[row] - 1 - nz;
184eef2e97bSBarry Smith         PLogFlops(2*nz);
185*dbb450caSBarry Smith         while (nz-->0) rtmp[*pj++ + shift] -= multiplier* *pv++;
186289bc588SBarry Smith       }
187*dbb450caSBarry Smith       row = *ajtmp++ + shift;
188289bc588SBarry Smith     }
1898c37ef55SBarry Smith     /* finished row so stick it into aijnew->a */
190*dbb450caSBarry Smith     pv = aijnew->a + ai[i] + shift;
191*dbb450caSBarry Smith     pj = aijnew->j + ai[i] + shift;
1928c37ef55SBarry Smith     nz = ai[i+1] - ai[i];
193ec8511deSBarry Smith     if (rtmp[i] == 0.0) {SETERRQ(1,"MatLUFactorNumeric_SeqAIJ:Zero pivot");}
1948c37ef55SBarry Smith     rtmp[i] = 1.0/rtmp[i];
195*dbb450caSBarry Smith     for ( j=0; j<nz; j++ ) {pv[j] = rtmp[pj[j]+shift];}
1968c37ef55SBarry Smith   }
19778b31e54SBarry Smith   PETSCFREE(rtmp);
19878b31e54SBarry Smith   ierr = ISRestoreIndices(isicol,&ic); CHKERRQ(ierr);
19978b31e54SBarry Smith   ierr = ISRestoreIndices(isrow,&r); CHKERRQ(ierr);
20078b31e54SBarry Smith   ierr = ISDestroy(isicol); CHKERRQ(ierr);
2018c37ef55SBarry Smith   fact->factor      = FACTOR_LU;
2029e25ed09SBarry Smith   aijnew->assembled = 1;
203eef2e97bSBarry Smith   PLogFlops(aijnew->n);
204289bc588SBarry Smith   return 0;
205289bc588SBarry Smith }
2060a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
207ec8511deSBarry Smith int MatLUFactor_SeqAIJ(Mat matin,IS row,IS col,double f)
208da3a660dSBarry Smith {
209ec8511deSBarry Smith   Mat_SeqAIJ *mat = (Mat_SeqAIJ *) matin->data;
2106abc6512SBarry Smith   int     ierr;
211da3a660dSBarry Smith   Mat     fact;
212ec8511deSBarry Smith   ierr = MatLUFactorSymbolic_SeqAIJ(matin,row,col,f,&fact); CHKERRQ(ierr);
213ec8511deSBarry Smith   ierr = MatLUFactorNumeric_SeqAIJ(matin,&fact); CHKERRQ(ierr);
214da3a660dSBarry Smith 
215da3a660dSBarry Smith   /* free all the data structures from mat */
21678b31e54SBarry Smith   PETSCFREE(mat->a);
21778b31e54SBarry Smith   if (!mat->singlemalloc) {PETSCFREE(mat->i); PETSCFREE(mat->j);}
21878b31e54SBarry Smith   if (mat->diag) PETSCFREE(mat->diag);
21978b31e54SBarry Smith   if (mat->ilen) PETSCFREE(mat->ilen);
22078b31e54SBarry Smith   if (mat->imax) PETSCFREE(mat->imax);
22194473badSLois Curfman McInnes   if (mat->solve_work) PETSCFREE(mat->solve_work);
22278b31e54SBarry Smith   PETSCFREE(mat);
223da3a660dSBarry Smith 
22478b31e54SBarry Smith   PETSCMEMCPY(matin,fact,sizeof(struct _Mat));
22536d154e2SLois Curfman McInnes   PETSCHEADERDESTROY(fact);
226da3a660dSBarry Smith   return 0;
227da3a660dSBarry Smith }
2280a6dbcc7SLois Curfman McInnes /* ----------------------------------------------------------- */
229ec8511deSBarry Smith int MatSolve_SeqAIJ(Mat mat,Vec bb, Vec xx)
2308c37ef55SBarry Smith {
231ec8511deSBarry Smith   Mat_SeqAIJ *aij = (Mat_SeqAIJ *) mat->data;
23286d7a8ceSBarry Smith   IS      iscol = aij->col, isrow = aij->row;
2336abc6512SBarry Smith   int     *r,*c, ierr, i,  n = aij->m, *vi, *ai = aij->i, *aj = aij->j;
2348c37ef55SBarry Smith   int     nz;
2358c37ef55SBarry Smith   Scalar  *x,*b,*tmp, *aa = aij->a, sum, *v;
236*dbb450caSBarry Smith   int     shift = aij->indexshift;
2378c37ef55SBarry Smith 
238d35516d3SLois Curfman McInnes   if (mat->factor != FACTOR_LU)
239ec8511deSBarry Smith     SETERRQ(1,"MatSolve_SeqAIJ:Cannot solve with unfactored matrix");
2409e25ed09SBarry Smith 
24178b31e54SBarry Smith   ierr = VecGetArray(bb,&b); CHKERRQ(ierr);
24278b31e54SBarry Smith   ierr = VecGetArray(xx,&x); CHKERRQ(ierr);
243eef2e97bSBarry Smith   tmp = aij->solve_work;
2448c37ef55SBarry Smith 
24578b31e54SBarry Smith   ierr = ISGetIndices(isrow,&r);CHKERRQ(ierr);
24678b31e54SBarry Smith   ierr = ISGetIndices(iscol,&c);CHKERRQ(ierr); c = c + (n-1);
2478c37ef55SBarry Smith 
2488c37ef55SBarry Smith   /* forward solve the lower triangular */
2498c37ef55SBarry Smith   tmp[0] = b[*r++];
2508c37ef55SBarry Smith   for ( i=1; i<n; i++ ) {
251*dbb450caSBarry Smith     v   = aa + ai[i] + shift;
252*dbb450caSBarry Smith     vi  = aj + ai[i] + shift;
2538c37ef55SBarry Smith     nz  = aij->diag[i] - ai[i];
2548c37ef55SBarry Smith     sum = b[*r++];
255*dbb450caSBarry Smith     while (nz--) sum -= *v++ * tmp[*vi++ + shift];
2568c37ef55SBarry Smith     tmp[i] = sum;
2578c37ef55SBarry Smith   }
2588c37ef55SBarry Smith 
2598c37ef55SBarry Smith   /* backward solve the upper triangular */
2608c37ef55SBarry Smith   for ( i=n-1; i>=0; i-- ){
261*dbb450caSBarry Smith     v   = aa + aij->diag[i] + (!shift);
262*dbb450caSBarry Smith     vi  = aj + aij->diag[i] + (!shift);
2638c37ef55SBarry Smith     nz  = ai[i+1] - aij->diag[i] - 1;
2648c37ef55SBarry Smith     sum = tmp[i];
265*dbb450caSBarry Smith     while (nz--) sum -= *v++ * tmp[*vi++ + shift];
266*dbb450caSBarry Smith     x[*c--] = tmp[i] = sum*aa[aij->diag[i]+shift];
2678c37ef55SBarry Smith   }
2688c37ef55SBarry Smith 
269898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r); CHKERRQ(ierr);
270898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(iscol,&c); CHKERRQ(ierr);
271eef2e97bSBarry Smith   PLogFlops(2*aij->nz - aij->n);
2728c37ef55SBarry Smith   return 0;
2738c37ef55SBarry Smith }
274ec8511deSBarry Smith int MatSolveAdd_SeqAIJ(Mat mat,Vec bb, Vec yy, Vec xx)
275da3a660dSBarry Smith {
276ec8511deSBarry Smith   Mat_SeqAIJ *aij = (Mat_SeqAIJ *) mat->data;
27786d7a8ceSBarry Smith   IS      iscol = aij->col, isrow = aij->row;
2786abc6512SBarry Smith   int     *r,*c, ierr, i,  n = aij->m, *vi, *ai = aij->i, *aj = aij->j;
279da3a660dSBarry Smith   int     nz;
280da3a660dSBarry Smith   Scalar  *x,*b,*tmp, *aa = aij->a, sum, *v;
281*dbb450caSBarry Smith    int shift = aij->indexshift;
282da3a660dSBarry Smith 
283d35516d3SLois Curfman McInnes   if (mat->factor != FACTOR_LU)
284ec8511deSBarry Smith     SETERRQ(1,"MatSolveAdd_SeqAIJ: Cannot solve with unfactored matrix");
28578b31e54SBarry Smith   if (yy != xx) {ierr = VecCopy(yy,xx); CHKERRQ(ierr);}
286da3a660dSBarry Smith 
28778b31e54SBarry Smith   ierr = VecGetArray(bb,&b); CHKERRQ(ierr);
28878b31e54SBarry Smith   ierr = VecGetArray(xx,&x); CHKERRQ(ierr);
289eef2e97bSBarry Smith   tmp = aij->solve_work;
290da3a660dSBarry Smith 
29178b31e54SBarry Smith   ierr = ISGetIndices(isrow,&r); CHKERRQ(ierr);
29278b31e54SBarry Smith   ierr = ISGetIndices(iscol,&c); CHKERRQ(ierr); c = c + (n-1);
293da3a660dSBarry Smith 
294da3a660dSBarry Smith   /* forward solve the lower triangular */
295da3a660dSBarry Smith   tmp[0] = b[*r++];
296da3a660dSBarry Smith   for ( i=1; i<n; i++ ) {
297*dbb450caSBarry Smith     v   = aa + ai[i] + shift;
298*dbb450caSBarry Smith     vi  = aj + ai[i] + shift;
299da3a660dSBarry Smith     nz  = aij->diag[i] - ai[i];
300da3a660dSBarry Smith     sum = b[*r++];
301*dbb450caSBarry Smith     while (nz--) sum -= *v++ * tmp[*vi++ + shift];
302da3a660dSBarry Smith     tmp[i] = sum;
303da3a660dSBarry Smith   }
304da3a660dSBarry Smith 
305da3a660dSBarry Smith   /* backward solve the upper triangular */
306da3a660dSBarry Smith   for ( i=n-1; i>=0; i-- ){
307*dbb450caSBarry Smith     v   = aa + aij->diag[i] + (!shift);
308*dbb450caSBarry Smith     vi  = aj + aij->diag[i] + (!shift);
309da3a660dSBarry Smith     nz  = ai[i+1] - aij->diag[i] - 1;
310da3a660dSBarry Smith     sum = tmp[i];
311*dbb450caSBarry Smith     while (nz--) sum -= *v++ * tmp[*vi++ + shift];
312*dbb450caSBarry Smith     tmp[i] = sum*aa[aij->diag[i]+shift];
313da3a660dSBarry Smith     x[*c--] += tmp[i];
314da3a660dSBarry Smith   }
315da3a660dSBarry Smith 
316898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r); CHKERRQ(ierr);
317898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(iscol,&c); CHKERRQ(ierr);
318eef2e97bSBarry Smith   PLogFlops(2*aij->nz);
319898183ecSLois Curfman McInnes 
320da3a660dSBarry Smith   return 0;
321da3a660dSBarry Smith }
322da3a660dSBarry Smith /* -------------------------------------------------------------------*/
323ec8511deSBarry Smith int MatSolveTrans_SeqAIJ(Mat mat,Vec bb, Vec xx)
324da3a660dSBarry Smith {
325ec8511deSBarry Smith   Mat_SeqAIJ *aij = (Mat_SeqAIJ *) mat->data;
32686d7a8ceSBarry Smith   IS      iscol = aij->col, isrow = aij->row, invisrow,inviscol;
3276abc6512SBarry Smith   int     *r,*c, ierr, i, n = aij->m, *vi, *ai = aij->i, *aj = aij->j;
328da3a660dSBarry Smith   int     nz;
3296abc6512SBarry Smith   Scalar  *x,*b,*tmp, *aa = aij->a, *v;
330*dbb450caSBarry Smith    int shift = aij->indexshift;
331da3a660dSBarry Smith 
332d35516d3SLois Curfman McInnes   if (mat->factor != FACTOR_LU)
333ec8511deSBarry Smith     SETERRQ(1,"MatSolveTrans_SeqAIJ:Cannot solve with unfactored matrix");
33478b31e54SBarry Smith   ierr = VecGetArray(bb,&b); CHKERRQ(ierr);
33578b31e54SBarry Smith   ierr = VecGetArray(xx,&x); CHKERRQ(ierr);
336eef2e97bSBarry Smith   tmp = aij->solve_work;
337da3a660dSBarry Smith 
338da3a660dSBarry Smith   /* invert the permutations */
33978b31e54SBarry Smith   ierr = ISInvertPermutation(isrow,&invisrow); CHKERRQ(ierr);
34078b31e54SBarry Smith   ierr = ISInvertPermutation(iscol,&inviscol); CHKERRQ(ierr);
341da3a660dSBarry Smith 
34278b31e54SBarry Smith   ierr = ISGetIndices(invisrow,&r); CHKERRQ(ierr);
34378b31e54SBarry Smith   ierr = ISGetIndices(inviscol,&c); CHKERRQ(ierr);
344da3a660dSBarry Smith 
345da3a660dSBarry Smith   /* copy the b into temp work space according to permutation */
346da3a660dSBarry Smith   for ( i=0; i<n; i++ ) tmp[c[i]] = b[i];
347da3a660dSBarry Smith 
348da3a660dSBarry Smith   /* forward solve the U^T */
349da3a660dSBarry Smith   for ( i=0; i<n; i++ ) {
350*dbb450caSBarry Smith     v   = aa + aij->diag[i] + shift;
351*dbb450caSBarry Smith     vi  = aj + aij->diag[i] + (!shift);
352da3a660dSBarry Smith     nz  = ai[i+1] - aij->diag[i] - 1;
353da3a660dSBarry Smith     tmp[i] *= *v++;
354da3a660dSBarry Smith     while (nz--) {
355*dbb450caSBarry Smith       tmp[*vi++ + shift] -= (*v++)*tmp[i];
356da3a660dSBarry Smith     }
357da3a660dSBarry Smith   }
358da3a660dSBarry Smith 
359da3a660dSBarry Smith   /* backward solve the L^T */
360da3a660dSBarry Smith   for ( i=n-1; i>=0; i-- ){
361*dbb450caSBarry Smith     v   = aa + aij->diag[i] - 1 + shift;
362*dbb450caSBarry Smith     vi  = aj + aij->diag[i] - 1 + shift;
363da3a660dSBarry Smith     nz  = aij->diag[i] - ai[i];
364da3a660dSBarry Smith     while (nz--) {
365*dbb450caSBarry Smith       tmp[*vi-- + shift] -= (*v--)*tmp[i];
366da3a660dSBarry Smith     }
367da3a660dSBarry Smith   }
368da3a660dSBarry Smith 
369da3a660dSBarry Smith   /* copy tmp into x according to permutation */
370da3a660dSBarry Smith   for ( i=0; i<n; i++ ) x[r[i]] = tmp[i];
371da3a660dSBarry Smith 
372898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(invisrow,&r); CHKERRQ(ierr);
373898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(inviscol,&c); CHKERRQ(ierr);
374898183ecSLois Curfman McInnes   ierr = ISDestroy(invisrow); CHKERRQ(ierr);
375898183ecSLois Curfman McInnes   ierr = ISDestroy(inviscol); CHKERRQ(ierr);
376da3a660dSBarry Smith 
377eef2e97bSBarry Smith   PLogFlops(2*aij->nz-aij->n);
378da3a660dSBarry Smith   return 0;
379da3a660dSBarry Smith }
380da3a660dSBarry Smith 
381ec8511deSBarry Smith int MatSolveTransAdd_SeqAIJ(Mat mat,Vec bb, Vec zz,Vec xx)
382da3a660dSBarry Smith {
383ec8511deSBarry Smith   Mat_SeqAIJ *aij = (Mat_SeqAIJ *) mat->data;
38486d7a8ceSBarry Smith   IS      iscol = aij->col, isrow = aij->row, invisrow,inviscol;
3856abc6512SBarry Smith   int     *r,*c, ierr, i, n = aij->m, *vi, *ai = aij->i, *aj = aij->j;
3866abc6512SBarry Smith   int     nz;
3876abc6512SBarry Smith   Scalar  *x,*b,*tmp, *aa = aij->a, *v;
388*dbb450caSBarry Smith    int shift = aij->indexshift;
3896abc6512SBarry Smith 
390d35516d3SLois Curfman McInnes   if (mat->factor != FACTOR_LU)
391ec8511deSBarry Smith     SETERRQ(1,"MatSolveTransAdd_SeqAIJ:Cannot solve with unfactored matrix");
3926abc6512SBarry Smith   if (zz != xx) VecCopy(zz,xx);
3936abc6512SBarry Smith 
39478b31e54SBarry Smith   ierr = VecGetArray(bb,&b); CHKERRQ(ierr);
39578b31e54SBarry Smith   ierr = VecGetArray(xx,&x); CHKERRQ(ierr);
396eef2e97bSBarry Smith   tmp = aij->solve_work;
3976abc6512SBarry Smith 
3986abc6512SBarry Smith   /* invert the permutations */
39978b31e54SBarry Smith   ierr = ISInvertPermutation(isrow,&invisrow); CHKERRQ(ierr);
40078b31e54SBarry Smith   ierr = ISInvertPermutation(iscol,&inviscol); CHKERRQ(ierr);
40178b31e54SBarry Smith   ierr = ISGetIndices(invisrow,&r); CHKERRQ(ierr);
40278b31e54SBarry Smith   ierr = ISGetIndices(inviscol,&c); CHKERRQ(ierr);
4036abc6512SBarry Smith 
4046abc6512SBarry Smith   /* copy the b into temp work space according to permutation */
4056abc6512SBarry Smith   for ( i=0; i<n; i++ ) tmp[c[i]] = b[i];
4066abc6512SBarry Smith 
4076abc6512SBarry Smith   /* forward solve the U^T */
4086abc6512SBarry Smith   for ( i=0; i<n; i++ ) {
409*dbb450caSBarry Smith     v   = aa + aij->diag[i] + shift;
410*dbb450caSBarry Smith     vi  = aj + aij->diag[i] + (!shift);
4116abc6512SBarry Smith     nz  = ai[i+1] - aij->diag[i] - 1;
4126abc6512SBarry Smith     tmp[i] *= *v++;
4136abc6512SBarry Smith     while (nz--) {
414*dbb450caSBarry Smith       tmp[*vi++ + shift] -= (*v++)*tmp[i];
4156abc6512SBarry Smith     }
4166abc6512SBarry Smith   }
4176abc6512SBarry Smith 
4186abc6512SBarry Smith   /* backward solve the L^T */
4196abc6512SBarry Smith   for ( i=n-1; i>=0; i-- ){
420*dbb450caSBarry Smith     v   = aa + aij->diag[i] - 1 + shift;
421*dbb450caSBarry Smith     vi  = aj + aij->diag[i] - 1 + shift;
4226abc6512SBarry Smith     nz  = aij->diag[i] - ai[i];
4236abc6512SBarry Smith     while (nz--) {
424*dbb450caSBarry Smith       tmp[*vi-- + shift] -= (*v--)*tmp[i];
4256abc6512SBarry Smith     }
4266abc6512SBarry Smith   }
4276abc6512SBarry Smith 
4286abc6512SBarry Smith   /* copy tmp into x according to permutation */
4296abc6512SBarry Smith   for ( i=0; i<n; i++ ) x[r[i]] += tmp[i];
4306abc6512SBarry Smith 
431898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(invisrow,&r); CHKERRQ(ierr);
432898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(inviscol,&c); CHKERRQ(ierr);
433898183ecSLois Curfman McInnes   ierr = ISDestroy(invisrow); CHKERRQ(ierr);
434898183ecSLois Curfman McInnes   ierr = ISDestroy(inviscol); CHKERRQ(ierr);
4356abc6512SBarry Smith 
436eef2e97bSBarry Smith   PLogFlops(2*aij->nz);
4376abc6512SBarry Smith   return 0;
438da3a660dSBarry Smith }
4399e25ed09SBarry Smith /* ----------------------------------------------------------------*/
440ec8511deSBarry Smith int MatILUFactorSymbolic_SeqAIJ(Mat mat,IS isrow,IS iscol,double f,
44149d8b64dSBarry Smith                              int levels,Mat *fact)
4429e25ed09SBarry Smith {
443ec8511deSBarry Smith   Mat_SeqAIJ *aij = (Mat_SeqAIJ *) mat->data, *aijnew;
4449e25ed09SBarry Smith   IS      isicol;
44556beaf04SBarry Smith   int     *r,*ic, ierr, prow, n = aij->m, *ai = aij->i, *aj = aij->j;
44656beaf04SBarry Smith   int     *ainew,*ajnew, jmax,*fill, *xi, nz, *im,*ajfill,*flev;
44756beaf04SBarry Smith   int     *dloc, idx, row,m,fm, nzf, nzi,len,  realloc = 0;
44856beaf04SBarry Smith   int     incrlev,nnz,i;
449*dbb450caSBarry Smith   int     shift = aij->indexshift;
4509e25ed09SBarry Smith 
451d35516d3SLois Curfman McInnes   if (n != aij->n)
452ec8511deSBarry Smith     SETERRQ(1,"MatILUFactorSymbolic_SeqAIJ:Matrix must be square");
453d35516d3SLois Curfman McInnes   if (!isrow)
454ec8511deSBarry Smith     SETERRQ(1,"MatILUFactorSymbolic_SeqAIJ:Matrix must have row permutation");
455d35516d3SLois Curfman McInnes   if (!iscol) SETERRQ(1,
456ec8511deSBarry Smith     "MatILUFactorSymbolic_SeqAIJ:Matrix must have column permutation");
4579e25ed09SBarry Smith 
45878b31e54SBarry Smith   ierr = ISInvertPermutation(iscol,&isicol); CHKERRQ(ierr);
459898183ecSLois Curfman McInnes   ierr = ISGetIndices(isrow,&r); CHKERRQ(ierr);
460898183ecSLois Curfman McInnes   ierr = ISGetIndices(isicol,&ic); CHKERRQ(ierr);
4619e25ed09SBarry Smith 
4629e25ed09SBarry Smith   /* get new row pointers */
46378b31e54SBarry Smith   ainew = (int *) PETSCMALLOC( (n+1)*sizeof(int) ); CHKPTRQ(ainew);
464*dbb450caSBarry Smith   ainew[0] = -shift;
4659e25ed09SBarry Smith   /* don't know how many column pointers are needed so estimate */
466*dbb450caSBarry Smith   jmax = (int) (f*(ai[n]+!shift));
46778b31e54SBarry Smith   ajnew = (int *) PETSCMALLOC( (jmax)*sizeof(int) ); CHKPTRQ(ajnew);
4689e25ed09SBarry Smith   /* ajfill is level of fill for each fill entry */
46978b31e54SBarry Smith   ajfill = (int *) PETSCMALLOC( (jmax)*sizeof(int) ); CHKPTRQ(ajfill);
4709e25ed09SBarry Smith   /* fill is a linked list of nonzeros in active row */
47156beaf04SBarry Smith   fill = (int *) PETSCMALLOC( (n+1)*sizeof(int)); CHKPTRQ(fill);
47256beaf04SBarry Smith   /* im is level for each filled value */
47356beaf04SBarry Smith   im = (int *) PETSCMALLOC( (n+1)*sizeof(int)); CHKPTRQ(im);
47456beaf04SBarry Smith   /* dloc is location of diagonal in factor */
47556beaf04SBarry Smith   dloc = (int *) PETSCMALLOC( (n+1)*sizeof(int)); CHKPTRQ(dloc);
47656beaf04SBarry Smith   dloc[0]  = 0;
4779e25ed09SBarry Smith 
47856beaf04SBarry Smith   for ( prow=0; prow<n; prow++ ) {
4799e25ed09SBarry Smith     /* first copy previous fill into linked list */
48056beaf04SBarry Smith     nzf = nz  = ai[r[prow]+1] - ai[r[prow]];
481*dbb450caSBarry Smith     xi  = aj + ai[r[prow]] + shift;
4829e25ed09SBarry Smith     fill[n] = n;
4839e25ed09SBarry Smith     while (nz--) {
4849e25ed09SBarry Smith       fm  = n;
485*dbb450caSBarry Smith       idx = ic[*xi++ + shift];
4869e25ed09SBarry Smith       do {
4879e25ed09SBarry Smith         m  = fm;
4889e25ed09SBarry Smith         fm = fill[m];
4899e25ed09SBarry Smith       } while (fm < idx);
4909e25ed09SBarry Smith       fill[m]   = idx;
4919e25ed09SBarry Smith       fill[idx] = fm;
49256beaf04SBarry Smith       im[idx]   = 0;
4939e25ed09SBarry Smith     }
49456beaf04SBarry Smith     nzi = 0;
4959e25ed09SBarry Smith     row = fill[n];
49656beaf04SBarry Smith     while ( row < prow ) {
49756beaf04SBarry Smith       incrlev = im[row] + 1;
49856beaf04SBarry Smith       nz      = dloc[row];
499*dbb450caSBarry Smith       xi      = ajnew  + ainew[row] + shift + nz;
500*dbb450caSBarry Smith       flev    = ajfill + ainew[row] + shift + nz + 1;
501*dbb450caSBarry Smith       nnz     = ainew[row+1] - ainew[row] - nz + shift;
502*dbb450caSBarry Smith       if (*xi++ + shift != row) {
503ec8511deSBarry Smith         SETERRQ(1,"MatILUFactorSymbolic_SeqAIJ:zero pivot");
50456beaf04SBarry Smith       }
50556beaf04SBarry Smith       fm      = row;
50656beaf04SBarry Smith       while (nnz-- > 0) {
507*dbb450caSBarry Smith         idx = *xi++ + shift;
50856beaf04SBarry Smith         if (*flev + incrlev > levels) {
50956beaf04SBarry Smith           flev++;
51056beaf04SBarry Smith           continue;
51156beaf04SBarry Smith         }
51256beaf04SBarry Smith         do {
5139e25ed09SBarry Smith           m  = fm;
5149e25ed09SBarry Smith           fm = fill[m];
51556beaf04SBarry Smith         } while (fm < idx);
5169e25ed09SBarry Smith         if (fm != idx) {
51756beaf04SBarry Smith           im[idx]  = *flev + incrlev;
5189e25ed09SBarry Smith           fill[m] = idx;
5199e25ed09SBarry Smith           fill[idx] = fm;
5209e25ed09SBarry Smith           fm = idx;
52156beaf04SBarry Smith           nzf++;
5229e25ed09SBarry Smith         }
52356beaf04SBarry Smith         else {
52456beaf04SBarry Smith           if (im[idx] > *flev + incrlev) im[idx] = *flev+incrlev;
5259e25ed09SBarry Smith         }
52656beaf04SBarry Smith         flev++;
5279e25ed09SBarry Smith       }
5289e25ed09SBarry Smith       row = fill[row];
52956beaf04SBarry Smith       nzi++;
5309e25ed09SBarry Smith     }
5319e25ed09SBarry Smith     /* copy new filled row into permanent storage */
53256beaf04SBarry Smith     ainew[prow+1] = ainew[prow] + nzf;
53356beaf04SBarry Smith     if (ainew[prow+1] > jmax+1) {
5349e25ed09SBarry Smith       /* allocate a longer ajnew */
535bbb6d6a8SBarry Smith       int maxadd;
536*dbb450caSBarry Smith       maxadd = (int) ((f*(ai[n]+!shift)*(n-prow+5))/n);
53756beaf04SBarry Smith       if (maxadd < nzf) maxadd = (n-prow)*(nzf+1);
538bbb6d6a8SBarry Smith       jmax += maxadd;
53956beaf04SBarry Smith       xi = (int *) PETSCMALLOC( jmax*sizeof(int) );CHKPTRQ(xi);
540*dbb450caSBarry Smith       PETSCMEMCPY(xi,ajnew,(ainew[prow]+shift)*sizeof(int));
54178b31e54SBarry Smith       PETSCFREE(ajnew);
54256beaf04SBarry Smith       ajnew = xi;
5439e25ed09SBarry Smith       /* allocate a longer ajfill */
54456beaf04SBarry Smith       xi = (int *) PETSCMALLOC( jmax*sizeof(int) );CHKPTRQ(xi);
545*dbb450caSBarry Smith       PETSCMEMCPY(xi,ajfill,(ainew[prow]+shift)*sizeof(int));
54678b31e54SBarry Smith       PETSCFREE(ajfill);
54756beaf04SBarry Smith       ajfill = xi;
548bbb6d6a8SBarry Smith       realloc++;
5499e25ed09SBarry Smith     }
550*dbb450caSBarry Smith     xi          = ajnew + ainew[prow] + shift;
551*dbb450caSBarry Smith     flev        = ajfill + ainew[prow] + shift;
55256beaf04SBarry Smith     dloc[prow]  = nzi;
5539e25ed09SBarry Smith     fm          = fill[n];
55456beaf04SBarry Smith     while (nzf--) {
555*dbb450caSBarry Smith       *xi++   = fm - shift;
55656beaf04SBarry Smith       *flev++ = im[fm];
5579e25ed09SBarry Smith       fm      = fill[fm];
5589e25ed09SBarry Smith     }
5599e25ed09SBarry Smith   }
56078b31e54SBarry Smith   PETSCFREE(ajfill);
561898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&r); CHKERRQ(ierr);
562898183ecSLois Curfman McInnes   ierr = ISRestoreIndices(isicol,&ic); CHKERRQ(ierr);
563898183ecSLois Curfman McInnes   ierr = ISDestroy(isicol); CHKERRQ(ierr);
564898183ecSLois Curfman McInnes   PETSCFREE(fill); PETSCFREE(im);
5659e25ed09SBarry Smith 
566bbb6d6a8SBarry Smith   PLogInfo((PetscObject)mat,
567ec8511deSBarry Smith     "Info:MatILUFactorSymbolic_SeqAIJ:Realloc %d Fill ratio:given %g needed %g\n",
56856beaf04SBarry Smith                              realloc,f,((double)ainew[n])/((double)ai[prow]));
569bbb6d6a8SBarry Smith 
5709e25ed09SBarry Smith   /* put together the new matrix */
571fafbff53SBarry Smith   ierr = MatCreateSeqAIJ(mat->comm,n, n, 0, 0, fact); CHKERRQ(ierr);
572ec8511deSBarry Smith   aijnew = (Mat_SeqAIJ *) (*fact)->data;
57378b31e54SBarry Smith   PETSCFREE(aijnew->imax);
5749e25ed09SBarry Smith   aijnew->singlemalloc = 0;
575*dbb450caSBarry Smith   len = (ainew[n] + shift)*sizeof(Scalar);
5769e25ed09SBarry Smith   /* the next line frees the default space generated by the Create() */
57778b31e54SBarry Smith   PETSCFREE(aijnew->a); PETSCFREE(aijnew->ilen);
57878b31e54SBarry Smith   aijnew->a         = (Scalar *) PETSCMALLOC( len ); CHKPTRQ(aijnew->a);
5799e25ed09SBarry Smith   aijnew->j         = ajnew;
5809e25ed09SBarry Smith   aijnew->i         = ainew;
58156beaf04SBarry Smith   for ( i=0; i<n; i++ ) dloc[i] += ainew[i];
58256beaf04SBarry Smith   aijnew->diag      = dloc;
5839e25ed09SBarry Smith   aijnew->ilen      = 0;
5849e25ed09SBarry Smith   aijnew->imax      = 0;
58586d7a8ceSBarry Smith   aijnew->row       = isrow;
58686d7a8ceSBarry Smith   aijnew->col       = iscol;
5875392566eSBarry Smith   aijnew->solve_work = (Scalar *) PETSCMALLOC( (n+1)*sizeof(Scalar));
58878b31e54SBarry Smith   CHKPTRQ(aijnew->solve_work);
5897fd98bd6SLois Curfman McInnes   /* In aijnew structure:  Free imax, ilen, old a, old j.
59056beaf04SBarry Smith      Allocate dloc, solve_work, new a, new j */
591*dbb450caSBarry Smith   PLogObjectMemory(*fact,(ainew[n]+shift-n) * (sizeof(int)+sizeof(Scalar)));
592*dbb450caSBarry Smith   aijnew->maxnz = aijnew->nz = ainew[n] + shift;
5939e25ed09SBarry Smith   (*fact)->factor   = FACTOR_LU;
5949e25ed09SBarry Smith   return 0;
5959e25ed09SBarry Smith }
596