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