117ab2063SBarry Smith #ifndef lint 2*f09e8eb9SSatish Balay static char vcid[] = "$Id: aij.c,v 1.218 1997/05/03 22:48:31 curfman Exp balay $"; 317ab2063SBarry Smith #endif 417ab2063SBarry Smith 5d5d45c9bSBarry Smith /* 63369ce9aSBarry Smith Defines the basic matrix operations for the AIJ (compressed row) 7d5d45c9bSBarry Smith matrix storage format. 8d5d45c9bSBarry Smith */ 93369ce9aSBarry Smith 103369ce9aSBarry Smith #include "pinclude/pviewer.h" 113369ce9aSBarry Smith #include "sys.h" 1270f55243SBarry Smith #include "src/mat/impls/aij/seq/aij.h" 13f5eb4b81SSatish Balay #include "src/vec/vecimpl.h" 14f5eb4b81SSatish Balay #include "src/inline/spops.h" 15e4d965acSSatish Balay #include "petsc.h" 16f5eb4b81SSatish Balay #include "src/inline/bitarray.h" 1717ab2063SBarry Smith 18a2ce50c7SBarry Smith /* 19a2ce50c7SBarry Smith Basic AIJ format ILU based on drop tolerance 20a2ce50c7SBarry Smith */ 215615d1e5SSatish Balay #undef __FUNC__ 225615d1e5SSatish Balay #define __FUNC__ "MatILUDTFactor_SeqAIJ" 23a2ce50c7SBarry Smith int MatILUDTFactor_SeqAIJ(Mat A,double dt,int maxnz,IS row,IS col,Mat *fact) 24a2ce50c7SBarry Smith { 25a2ce50c7SBarry Smith /* Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; */ 26a2ce50c7SBarry Smith int ierr = 1; 27a2ce50c7SBarry Smith 28e3372554SBarry Smith SETERRQ(ierr,0,"Not implemented"); 29a2ce50c7SBarry Smith } 30a2ce50c7SBarry Smith 31bcd2baecSBarry Smith extern int MatToSymmetricIJ_SeqAIJ(int,int*,int*,int,int,int**,int**); 3217ab2063SBarry Smith 335615d1e5SSatish Balay #undef __FUNC__ 345615d1e5SSatish Balay #define __FUNC__ "MatGetRowIJ_SeqAIJ" 358f6be9afSLois Curfman McInnes int MatGetRowIJ_SeqAIJ(Mat A,int oshift,PetscTruth symmetric,int *m,int **ia,int **ja, 366945ee14SBarry Smith PetscTruth *done) 3717ab2063SBarry Smith { 38416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 396945ee14SBarry Smith int ierr,i,ishift; 4017ab2063SBarry Smith 4131625ec5SSatish Balay *m = A->m; 426945ee14SBarry Smith if (!ia) return 0; 436945ee14SBarry Smith ishift = a->indexshift; 446945ee14SBarry Smith if (symmetric) { 4531625ec5SSatish Balay ierr = MatToSymmetricIJ_SeqAIJ(a->m,a->i,a->j,ishift,oshift,ia,ja); CHKERRQ(ierr); 466945ee14SBarry Smith } else if (oshift == 0 && ishift == -1) { 4731625ec5SSatish Balay int nz = a->i[a->m]; 483b2fbd54SBarry Smith /* malloc space and subtract 1 from i and j indices */ 4931625ec5SSatish Balay *ia = (int *) PetscMalloc( (a->m+1)*sizeof(int) ); CHKPTRQ(*ia); 503b2fbd54SBarry Smith *ja = (int *) PetscMalloc( (nz+1)*sizeof(int) ); CHKPTRQ(*ja); 513b2fbd54SBarry Smith for ( i=0; i<nz; i++ ) (*ja)[i] = a->j[i] - 1; 5231625ec5SSatish Balay for ( i=0; i<a->m+1; i++ ) (*ia)[i] = a->i[i] - 1; 536945ee14SBarry Smith } else if (oshift == 1 && ishift == 0) { 5431625ec5SSatish Balay int nz = a->i[a->m] + 1; 553b2fbd54SBarry Smith /* malloc space and add 1 to i and j indices */ 5631625ec5SSatish Balay *ia = (int *) PetscMalloc( (a->m+1)*sizeof(int) ); CHKPTRQ(*ia); 573b2fbd54SBarry Smith *ja = (int *) PetscMalloc( (nz+1)*sizeof(int) ); CHKPTRQ(*ja); 583b2fbd54SBarry Smith for ( i=0; i<nz; i++ ) (*ja)[i] = a->j[i] + 1; 5931625ec5SSatish Balay for ( i=0; i<a->m+1; i++ ) (*ia)[i] = a->i[i] + 1; 606945ee14SBarry Smith } else { 616945ee14SBarry Smith *ia = a->i; *ja = a->j; 62a2ce50c7SBarry Smith } 63a2ce50c7SBarry Smith 64a2744918SBarry Smith return 0; 65a2744918SBarry Smith } 66a2744918SBarry Smith 675615d1e5SSatish Balay #undef __FUNC__ 685615d1e5SSatish Balay #define __FUNC__ "MatRestoreRowIJ_SeqAIJ" 698f6be9afSLois Curfman McInnes int MatRestoreRowIJ_SeqAIJ(Mat A,int oshift,PetscTruth symmetric,int *n,int **ia,int **ja, 706945ee14SBarry Smith PetscTruth *done) 716945ee14SBarry Smith { 726945ee14SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 733b2fbd54SBarry Smith int ishift = a->indexshift; 746945ee14SBarry Smith 756945ee14SBarry Smith if (!ia) return 0; 763b2fbd54SBarry Smith if (symmetric || (oshift == 0 && ishift == -1) || (oshift == 1 && ishift == 0)) { 776945ee14SBarry Smith PetscFree(*ia); 786945ee14SBarry Smith PetscFree(*ja); 79bcd2baecSBarry Smith } 8017ab2063SBarry Smith return 0; 8117ab2063SBarry Smith } 8217ab2063SBarry Smith 835615d1e5SSatish Balay #undef __FUNC__ 845615d1e5SSatish Balay #define __FUNC__ "MatGetColumnIJ_SeqAIJ" 8543a90d84SBarry Smith int MatGetColumnIJ_SeqAIJ(Mat A,int oshift,PetscTruth symmetric,int *nn,int **ia,int **ja, 863b2fbd54SBarry Smith PetscTruth *done) 873b2fbd54SBarry Smith { 883b2fbd54SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 89a93ec695SBarry Smith int ierr,i,ishift = a->indexshift,*collengths,*cia,*cja,n = A->n,m = A->m; 90a93ec695SBarry Smith int nz = a->i[m]+ishift,row,*jj,mr,col; 913b2fbd54SBarry Smith 923b2fbd54SBarry Smith *nn = A->n; 933b2fbd54SBarry Smith if (!ia) return 0; 943b2fbd54SBarry Smith if (symmetric) { 95179192dfSSatish Balay ierr = MatToSymmetricIJ_SeqAIJ(a->m,a->i,a->j,ishift,oshift,ia,ja); CHKERRQ(ierr); 963b2fbd54SBarry Smith } else { 9761d2ded1SBarry Smith collengths = (int *) PetscMalloc( (n+1)*sizeof(int) ); CHKPTRQ(collengths); 983b2fbd54SBarry Smith PetscMemzero(collengths,n*sizeof(int)); 993b2fbd54SBarry Smith cia = (int *) PetscMalloc( (n+1)*sizeof(int) ); CHKPTRQ(cia); 100a93ec695SBarry Smith cja = (int *) PetscMalloc( (nz+1)*sizeof(int) ); CHKPTRQ(cja); 1013b2fbd54SBarry Smith jj = a->j; 1023b2fbd54SBarry Smith for ( i=0; i<nz; i++ ) { 1033b2fbd54SBarry Smith collengths[jj[i] + ishift]++; 1043b2fbd54SBarry Smith } 1053b2fbd54SBarry Smith cia[0] = oshift; 1063b2fbd54SBarry Smith for ( i=0; i<n; i++) { 1073b2fbd54SBarry Smith cia[i+1] = cia[i] + collengths[i]; 1083b2fbd54SBarry Smith } 1093b2fbd54SBarry Smith PetscMemzero(collengths,n*sizeof(int)); 1103b2fbd54SBarry Smith jj = a->j; 111a93ec695SBarry Smith for ( row=0; row<m; row++ ) { 112a93ec695SBarry Smith mr = a->i[row+1] - a->i[row]; 113a93ec695SBarry Smith for ( i=0; i<mr; i++ ) { 1143b2fbd54SBarry Smith col = *jj++ + ishift; 1153b2fbd54SBarry Smith cja[cia[col] + collengths[col]++ - oshift] = row + oshift; 1163b2fbd54SBarry Smith } 1173b2fbd54SBarry Smith } 1183b2fbd54SBarry Smith PetscFree(collengths); 1193b2fbd54SBarry Smith *ia = cia; *ja = cja; 1203b2fbd54SBarry Smith } 1213b2fbd54SBarry Smith 1223b2fbd54SBarry Smith return 0; 1233b2fbd54SBarry Smith } 1243b2fbd54SBarry Smith 1255615d1e5SSatish Balay #undef __FUNC__ 1265615d1e5SSatish Balay #define __FUNC__ "MatRestoreColumnIJ_SeqAIJ" 12743a90d84SBarry Smith int MatRestoreColumnIJ_SeqAIJ(Mat A,int oshift,PetscTruth symmetric,int *n,int **ia, 1283b2fbd54SBarry Smith int **ja,PetscTruth *done) 1293b2fbd54SBarry Smith { 1303b2fbd54SBarry Smith if (!ia) return 0; 1313b2fbd54SBarry Smith 1323b2fbd54SBarry Smith PetscFree(*ia); 1333b2fbd54SBarry Smith PetscFree(*ja); 1343b2fbd54SBarry Smith 1353b2fbd54SBarry Smith return 0; 1363b2fbd54SBarry Smith } 1373b2fbd54SBarry Smith 138227d817aSBarry Smith #define CHUNKSIZE 15 13917ab2063SBarry Smith 1405615d1e5SSatish Balay #undef __FUNC__ 1415615d1e5SSatish Balay #define __FUNC__ "MatSetValues_SeqAIJ" 14261d2ded1SBarry Smith int MatSetValues_SeqAIJ(Mat A,int m,int *im,int n,int *in,Scalar *v,InsertMode is) 14317ab2063SBarry Smith { 144416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 145416022c9SBarry Smith int *rp,k,low,high,t,ii,row,nrow,i,col,l,rmax, N, sorted = a->sorted; 1464b0e389bSBarry Smith int *imax = a->imax, *ai = a->i, *ailen = a->ilen,roworiented = a->roworiented; 147d5d45c9bSBarry Smith int *aj = a->j, nonew = a->nonew,shift = a->indexshift; 148416022c9SBarry Smith Scalar *ap,value, *aa = a->a; 14917ab2063SBarry Smith 15017ab2063SBarry Smith for ( k=0; k<m; k++ ) { /* loop over added rows */ 151416022c9SBarry Smith row = im[k]; 1523b2fbd54SBarry Smith #if defined(PETSC_BOPT_g) 153e3372554SBarry Smith if (row < 0) SETERRQ(1,0,"Negative row"); 154e3372554SBarry Smith if (row >= a->m) SETERRQ(1,0,"Row too large"); 1553b2fbd54SBarry Smith #endif 15617ab2063SBarry Smith rp = aj + ai[row] + shift; ap = aa + ai[row] + shift; 15717ab2063SBarry Smith rmax = imax[row]; nrow = ailen[row]; 158416022c9SBarry Smith low = 0; 15917ab2063SBarry Smith for ( l=0; l<n; l++ ) { /* loop over added columns */ 1603b2fbd54SBarry Smith #if defined(PETSC_BOPT_g) 161e3372554SBarry Smith if (in[l] < 0) SETERRQ(1,0,"Negative column"); 162e3372554SBarry Smith if (in[l] >= a->n) SETERRQ(1,0,"Column too large"); 1633b2fbd54SBarry Smith #endif 1644b0e389bSBarry Smith col = in[l] - shift; 1654b0e389bSBarry Smith if (roworiented) { 1664b0e389bSBarry Smith value = *v++; 1674b0e389bSBarry Smith } 1684b0e389bSBarry Smith else { 1694b0e389bSBarry Smith value = v[k + l*m]; 1704b0e389bSBarry Smith } 171416022c9SBarry Smith if (!sorted) low = 0; high = nrow; 172416022c9SBarry Smith while (high-low > 5) { 173416022c9SBarry Smith t = (low+high)/2; 174416022c9SBarry Smith if (rp[t] > col) high = t; 175416022c9SBarry Smith else low = t; 17617ab2063SBarry Smith } 177416022c9SBarry Smith for ( i=low; i<high; i++ ) { 17817ab2063SBarry Smith if (rp[i] > col) break; 17917ab2063SBarry Smith if (rp[i] == col) { 180416022c9SBarry Smith if (is == ADD_VALUES) ap[i] += value; 18117ab2063SBarry Smith else ap[i] = value; 18217ab2063SBarry Smith goto noinsert; 18317ab2063SBarry Smith } 18417ab2063SBarry Smith } 185c2653b3dSLois Curfman McInnes if (nonew == 1) goto noinsert; 18611ebbc71SLois Curfman McInnes else if (nonew == -1) SETERRQ(1,0,"Inserting a new nonzero in the matrix"); 18717ab2063SBarry Smith if (nrow >= rmax) { 18817ab2063SBarry Smith /* there is no extra room in row, therefore enlarge */ 189416022c9SBarry Smith int new_nz = ai[a->m] + CHUNKSIZE,len,*new_i,*new_j; 19017ab2063SBarry Smith Scalar *new_a; 19117ab2063SBarry Smith 19211ebbc71SLois Curfman McInnes if (nonew == -2) SETERRQ(1,0,"Inserting a new nonzero in the matrix"); 19396854ed6SLois Curfman McInnes 19417ab2063SBarry Smith /* malloc new storage space */ 195416022c9SBarry Smith len = new_nz*(sizeof(int)+sizeof(Scalar))+(a->m+1)*sizeof(int); 1960452661fSBarry Smith new_a = (Scalar *) PetscMalloc( len ); CHKPTRQ(new_a); 19717ab2063SBarry Smith new_j = (int *) (new_a + new_nz); 19817ab2063SBarry Smith new_i = new_j + new_nz; 19917ab2063SBarry Smith 20017ab2063SBarry Smith /* copy over old data into new slots */ 20117ab2063SBarry Smith for ( ii=0; ii<row+1; ii++ ) {new_i[ii] = ai[ii];} 202416022c9SBarry Smith for ( ii=row+1; ii<a->m+1; ii++ ) {new_i[ii] = ai[ii]+CHUNKSIZE;} 203416022c9SBarry Smith PetscMemcpy(new_j,aj,(ai[row]+nrow+shift)*sizeof(int)); 204416022c9SBarry Smith len = (new_nz - CHUNKSIZE - ai[row] - nrow - shift); 205416022c9SBarry Smith PetscMemcpy(new_j+ai[row]+shift+nrow+CHUNKSIZE,aj+ai[row]+shift+nrow, 20617ab2063SBarry Smith len*sizeof(int)); 207416022c9SBarry Smith PetscMemcpy(new_a,aa,(ai[row]+nrow+shift)*sizeof(Scalar)); 208416022c9SBarry Smith PetscMemcpy(new_a+ai[row]+shift+nrow+CHUNKSIZE,aa+ai[row]+shift+nrow, 20917ab2063SBarry Smith len*sizeof(Scalar)); 21017ab2063SBarry Smith /* free up old matrix storage */ 2110452661fSBarry Smith PetscFree(a->a); 2120452661fSBarry Smith if (!a->singlemalloc) {PetscFree(a->i);PetscFree(a->j);} 213416022c9SBarry Smith aa = a->a = new_a; ai = a->i = new_i; aj = a->j = new_j; 214416022c9SBarry Smith a->singlemalloc = 1; 21517ab2063SBarry Smith 21617ab2063SBarry Smith rp = aj + ai[row] + shift; ap = aa + ai[row] + shift; 217416022c9SBarry Smith rmax = imax[row] = imax[row] + CHUNKSIZE; 218416022c9SBarry Smith PLogObjectMemory(A,CHUNKSIZE*(sizeof(int) + sizeof(Scalar))); 219416022c9SBarry Smith a->maxnz += CHUNKSIZE; 220b810aeb4SBarry Smith a->reallocs++; 22117ab2063SBarry Smith } 222416022c9SBarry Smith N = nrow++ - 1; a->nz++; 223416022c9SBarry Smith /* shift up all the later entries in this row */ 224416022c9SBarry Smith for ( ii=N; ii>=i; ii-- ) { 22517ab2063SBarry Smith rp[ii+1] = rp[ii]; 22617ab2063SBarry Smith ap[ii+1] = ap[ii]; 22717ab2063SBarry Smith } 22817ab2063SBarry Smith rp[i] = col; 22917ab2063SBarry Smith ap[i] = value; 23017ab2063SBarry Smith noinsert:; 231416022c9SBarry Smith low = i + 1; 23217ab2063SBarry Smith } 23317ab2063SBarry Smith ailen[row] = nrow; 23417ab2063SBarry Smith } 23517ab2063SBarry Smith return 0; 23617ab2063SBarry Smith } 23717ab2063SBarry Smith 2385615d1e5SSatish Balay #undef __FUNC__ 2395615d1e5SSatish Balay #define __FUNC__ "MatGetValues_SeqAIJ" 2408f6be9afSLois Curfman McInnes int MatGetValues_SeqAIJ(Mat A,int m,int *im,int n,int *in,Scalar *v) 2417eb43aa7SLois Curfman McInnes { 2427eb43aa7SLois Curfman McInnes Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 243b49de8d1SLois Curfman McInnes int *rp, k, low, high, t, row, nrow, i, col, l, *aj = a->j; 2447eb43aa7SLois Curfman McInnes int *ai = a->i, *ailen = a->ilen, shift = a->indexshift; 2457eb43aa7SLois Curfman McInnes Scalar *ap, *aa = a->a, zero = 0.0; 2467eb43aa7SLois Curfman McInnes 2477eb43aa7SLois Curfman McInnes for ( k=0; k<m; k++ ) { /* loop over rows */ 2487eb43aa7SLois Curfman McInnes row = im[k]; 249e3372554SBarry Smith if (row < 0) SETERRQ(1,0,"Negative row"); 250e3372554SBarry Smith if (row >= a->m) SETERRQ(1,0,"Row too large"); 2517eb43aa7SLois Curfman McInnes rp = aj + ai[row] + shift; ap = aa + ai[row] + shift; 2527eb43aa7SLois Curfman McInnes nrow = ailen[row]; 2537eb43aa7SLois Curfman McInnes for ( l=0; l<n; l++ ) { /* loop over columns */ 254e3372554SBarry Smith if (in[l] < 0) SETERRQ(1,0,"Negative column"); 255e3372554SBarry Smith if (in[l] >= a->n) SETERRQ(1,0,"Column too large"); 2567eb43aa7SLois Curfman McInnes col = in[l] - shift; 2577eb43aa7SLois Curfman McInnes high = nrow; low = 0; /* assume unsorted */ 2587eb43aa7SLois Curfman McInnes while (high-low > 5) { 2597eb43aa7SLois Curfman McInnes t = (low+high)/2; 2607eb43aa7SLois Curfman McInnes if (rp[t] > col) high = t; 2617eb43aa7SLois Curfman McInnes else low = t; 2627eb43aa7SLois Curfman McInnes } 2637eb43aa7SLois Curfman McInnes for ( i=low; i<high; i++ ) { 2647eb43aa7SLois Curfman McInnes if (rp[i] > col) break; 2657eb43aa7SLois Curfman McInnes if (rp[i] == col) { 266b49de8d1SLois Curfman McInnes *v++ = ap[i]; 2677eb43aa7SLois Curfman McInnes goto finished; 2687eb43aa7SLois Curfman McInnes } 2697eb43aa7SLois Curfman McInnes } 270b49de8d1SLois Curfman McInnes *v++ = zero; 2717eb43aa7SLois Curfman McInnes finished:; 2727eb43aa7SLois Curfman McInnes } 2737eb43aa7SLois Curfman McInnes } 2747eb43aa7SLois Curfman McInnes return 0; 2757eb43aa7SLois Curfman McInnes } 2767eb43aa7SLois Curfman McInnes 27717ab2063SBarry Smith 2785615d1e5SSatish Balay #undef __FUNC__ 2795615d1e5SSatish Balay #define __FUNC__ "MatView_SeqAIJ_Binary" 2808f6be9afSLois Curfman McInnes extern int MatView_SeqAIJ_Binary(Mat A,Viewer viewer) 28117ab2063SBarry Smith { 282416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 283416022c9SBarry Smith int i, fd, *col_lens, ierr; 28417ab2063SBarry Smith 28590ace30eSBarry Smith ierr = ViewerBinaryGetDescriptor(viewer,&fd); CHKERRQ(ierr); 2860452661fSBarry Smith col_lens = (int *) PetscMalloc( (4+a->m)*sizeof(int) ); CHKPTRQ(col_lens); 287416022c9SBarry Smith col_lens[0] = MAT_COOKIE; 288416022c9SBarry Smith col_lens[1] = a->m; 289416022c9SBarry Smith col_lens[2] = a->n; 290416022c9SBarry Smith col_lens[3] = a->nz; 291416022c9SBarry Smith 292416022c9SBarry Smith /* store lengths of each row and write (including header) to file */ 293416022c9SBarry Smith for ( i=0; i<a->m; i++ ) { 294416022c9SBarry Smith col_lens[4+i] = a->i[i+1] - a->i[i]; 29517ab2063SBarry Smith } 29677c4ece6SBarry Smith ierr = PetscBinaryWrite(fd,col_lens,4+a->m,BINARY_INT,1); CHKERRQ(ierr); 2970452661fSBarry Smith PetscFree(col_lens); 298416022c9SBarry Smith 299416022c9SBarry Smith /* store column indices (zero start index) */ 300416022c9SBarry Smith if (a->indexshift) { 301416022c9SBarry Smith for ( i=0; i<a->nz; i++ ) a->j[i]--; 30217ab2063SBarry Smith } 30377c4ece6SBarry Smith ierr = PetscBinaryWrite(fd,a->j,a->nz,BINARY_INT,0); CHKERRQ(ierr); 304416022c9SBarry Smith if (a->indexshift) { 305416022c9SBarry Smith for ( i=0; i<a->nz; i++ ) a->j[i]++; 30617ab2063SBarry Smith } 307416022c9SBarry Smith 308416022c9SBarry Smith /* store nonzero values */ 30977c4ece6SBarry Smith ierr = PetscBinaryWrite(fd,a->a,a->nz,BINARY_SCALAR,0); CHKERRQ(ierr); 31017ab2063SBarry Smith return 0; 31117ab2063SBarry Smith } 312416022c9SBarry Smith 3135615d1e5SSatish Balay #undef __FUNC__ 3145615d1e5SSatish Balay #define __FUNC__ "MatView_SeqAIJ_ASCII" 3158f6be9afSLois Curfman McInnes extern int MatView_SeqAIJ_ASCII(Mat A,Viewer viewer) 316416022c9SBarry Smith { 317416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 318496e697eSBarry Smith int ierr, i,j, m = a->m, shift = a->indexshift, format, flg1,flg2; 31917ab2063SBarry Smith FILE *fd; 32017ab2063SBarry Smith char *outputname; 32117ab2063SBarry Smith 32290ace30eSBarry Smith ierr = ViewerASCIIGetPointer(viewer,&fd); CHKERRQ(ierr); 323416022c9SBarry Smith ierr = ViewerFileGetOutputname_Private(viewer,&outputname); CHKERRQ(ierr); 32490ace30eSBarry Smith ierr = ViewerGetFormat(viewer,&format); 325a93ec695SBarry Smith if (format == VIEWER_FORMAT_ASCII_INFO) { 32695e01e2fSLois Curfman McInnes return 0; 32795e01e2fSLois Curfman McInnes } 328a93ec695SBarry Smith else if (format == VIEWER_FORMAT_ASCII_INFO_LONG) { 329496e697eSBarry Smith ierr = OptionsHasName(PETSC_NULL,"-mat_aij_no_inode",&flg1); CHKERRQ(ierr); 330496e697eSBarry Smith ierr = OptionsHasName(PETSC_NULL,"-mat_no_unroll",&flg2); CHKERRQ(ierr); 331496e697eSBarry Smith if (flg1 || flg2) fprintf(fd," not using I-node routines\n"); 33295e01e2fSLois Curfman McInnes else fprintf(fd," using I-node routines: found %d nodes, limit used is %d\n", 33395e01e2fSLois Curfman McInnes a->inode.node_count,a->inode.limit); 33417ab2063SBarry Smith } 335a93ec695SBarry Smith else if (format == VIEWER_FORMAT_ASCII_MATLAB) { 336416022c9SBarry Smith fprintf(fd,"%% Size = %d %d \n",m,a->n); 3374e220ebcSLois Curfman McInnes fprintf(fd,"%% Nonzeros = %d \n",a->nz); 3384e220ebcSLois Curfman McInnes fprintf(fd,"zzz = zeros(%d,3);\n",a->nz); 33917ab2063SBarry Smith fprintf(fd,"zzz = [\n"); 34017ab2063SBarry Smith 34117ab2063SBarry Smith for (i=0; i<m; i++) { 342416022c9SBarry Smith for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 34317ab2063SBarry Smith #if defined(PETSC_COMPLEX) 3446945ee14SBarry Smith fprintf(fd,"%d %d %18.16e + %18.16e i \n",i+1,a->j[j]+!shift,real(a->a[j]), 345416022c9SBarry Smith imag(a->a[j])); 34617ab2063SBarry Smith #else 3477a743949SBarry Smith fprintf(fd,"%d %d %18.16e\n", i+1, a->j[j]+!shift, a->a[j]); 34817ab2063SBarry Smith #endif 34917ab2063SBarry Smith } 35017ab2063SBarry Smith } 35117ab2063SBarry Smith fprintf(fd,"];\n %s = spconvert(zzz);\n",outputname); 35217ab2063SBarry Smith } 353a93ec695SBarry Smith else if (format == VIEWER_FORMAT_ASCII_COMMON) { 35444cd7ae7SLois Curfman McInnes for ( i=0; i<m; i++ ) { 35544cd7ae7SLois Curfman McInnes fprintf(fd,"row %d:",i); 35644cd7ae7SLois Curfman McInnes for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 35744cd7ae7SLois Curfman McInnes #if defined(PETSC_COMPLEX) 35844cd7ae7SLois Curfman McInnes if (imag(a->a[j]) != 0.0 && real(a->a[j]) != 0.0) 35944cd7ae7SLois Curfman McInnes fprintf(fd," %d %g + %g i",a->j[j]+shift,real(a->a[j]),imag(a->a[j])); 36044cd7ae7SLois Curfman McInnes else if (real(a->a[j]) != 0.0) 36144cd7ae7SLois Curfman McInnes fprintf(fd," %d %g ",a->j[j]+shift,real(a->a[j])); 36244cd7ae7SLois Curfman McInnes #else 36344cd7ae7SLois Curfman McInnes if (a->a[j] != 0.0) fprintf(fd," %d %g ",a->j[j]+shift,a->a[j]); 36444cd7ae7SLois Curfman McInnes #endif 36544cd7ae7SLois Curfman McInnes } 36644cd7ae7SLois Curfman McInnes fprintf(fd,"\n"); 36744cd7ae7SLois Curfman McInnes } 36844cd7ae7SLois Curfman McInnes } 36917ab2063SBarry Smith else { 37017ab2063SBarry Smith for ( i=0; i<m; i++ ) { 37117ab2063SBarry Smith fprintf(fd,"row %d:",i); 372416022c9SBarry Smith for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 37317ab2063SBarry Smith #if defined(PETSC_COMPLEX) 374416022c9SBarry Smith if (imag(a->a[j]) != 0.0) { 375416022c9SBarry Smith fprintf(fd," %d %g + %g i",a->j[j]+shift,real(a->a[j]),imag(a->a[j])); 37617ab2063SBarry Smith } 37717ab2063SBarry Smith else { 378416022c9SBarry Smith fprintf(fd," %d %g ",a->j[j]+shift,real(a->a[j])); 37917ab2063SBarry Smith } 38017ab2063SBarry Smith #else 381416022c9SBarry Smith fprintf(fd," %d %g ",a->j[j]+shift,a->a[j]); 38217ab2063SBarry Smith #endif 38317ab2063SBarry Smith } 38417ab2063SBarry Smith fprintf(fd,"\n"); 38517ab2063SBarry Smith } 38617ab2063SBarry Smith } 38717ab2063SBarry Smith fflush(fd); 388416022c9SBarry Smith return 0; 389416022c9SBarry Smith } 390416022c9SBarry Smith 3915615d1e5SSatish Balay #undef __FUNC__ 3925615d1e5SSatish Balay #define __FUNC__ "MatView_SeqAIJ_Draw" 3938f6be9afSLois Curfman McInnes extern int MatView_SeqAIJ_Draw(Mat A,Viewer viewer) 394416022c9SBarry Smith { 395416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 396cddf8d76SBarry Smith int ierr, i,j, m = a->m, shift = a->indexshift,pause,color; 3970513a670SBarry Smith int format; 39894a9d846SBarry Smith double xl,yl,xr,yr,w,h,xc,yc,scale = 1.0,x_l,x_r,y_l,y_r,maxv = 0.0; 399bcd2baecSBarry Smith Draw draw; 400cddf8d76SBarry Smith DrawButton button; 40119bcc07fSBarry Smith PetscTruth isnull; 402cddf8d76SBarry Smith 4030513a670SBarry Smith ierr = ViewerDrawGetDraw(viewer,&draw); CHKERRQ(ierr); 4040513a670SBarry Smith ierr = DrawClear(draw); CHKERRQ(ierr); 4050513a670SBarry Smith ierr = ViewerGetFormat(viewer,&format); CHKERRQ(ierr); 40619bcc07fSBarry Smith ierr = DrawIsNull(draw,&isnull); CHKERRQ(ierr); if (isnull) return 0; 40719bcc07fSBarry Smith 408416022c9SBarry Smith xr = a->n; yr = a->m; h = yr/10.0; w = xr/10.0; 409416022c9SBarry Smith xr += w; yr += h; xl = -w; yl = -h; 410416022c9SBarry Smith ierr = DrawSetCoordinates(draw,xl,yl,xr,yr); CHKERRQ(ierr); 411416022c9SBarry Smith /* loop over matrix elements drawing boxes */ 4120513a670SBarry Smith 4130513a670SBarry Smith if (format != VIEWER_FORMAT_DRAW_CONTOUR) { 4140513a670SBarry Smith /* Blue for negative, Cyan for zero and Red for positive */ 415cddf8d76SBarry Smith color = DRAW_BLUE; 416416022c9SBarry Smith for ( i=0; i<m; i++ ) { 417cddf8d76SBarry Smith y_l = m - i - 1.0; y_r = y_l + 1.0; 418416022c9SBarry Smith for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 419cddf8d76SBarry Smith x_l = a->j[j] + shift; x_r = x_l + 1.0; 420cddf8d76SBarry Smith #if defined(PETSC_COMPLEX) 421cddf8d76SBarry Smith if (real(a->a[j]) >= 0.) continue; 422cddf8d76SBarry Smith #else 423cddf8d76SBarry Smith if (a->a[j] >= 0.) continue; 424cddf8d76SBarry Smith #endif 425cddf8d76SBarry Smith DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color); 426cddf8d76SBarry Smith } 427cddf8d76SBarry Smith } 428cddf8d76SBarry Smith color = DRAW_CYAN; 429cddf8d76SBarry Smith for ( i=0; i<m; i++ ) { 430cddf8d76SBarry Smith y_l = m - i - 1.0; y_r = y_l + 1.0; 431cddf8d76SBarry Smith for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 432cddf8d76SBarry Smith x_l = a->j[j] + shift; x_r = x_l + 1.0; 433cddf8d76SBarry Smith if (a->a[j] != 0.) continue; 434cddf8d76SBarry Smith DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color); 435cddf8d76SBarry Smith } 436cddf8d76SBarry Smith } 437cddf8d76SBarry Smith color = DRAW_RED; 438cddf8d76SBarry Smith for ( i=0; i<m; i++ ) { 439cddf8d76SBarry Smith y_l = m - i - 1.0; y_r = y_l + 1.0; 440cddf8d76SBarry Smith for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 441cddf8d76SBarry Smith x_l = a->j[j] + shift; x_r = x_l + 1.0; 442cddf8d76SBarry Smith #if defined(PETSC_COMPLEX) 443cddf8d76SBarry Smith if (real(a->a[j]) <= 0.) continue; 444cddf8d76SBarry Smith #else 445cddf8d76SBarry Smith if (a->a[j] <= 0.) continue; 446cddf8d76SBarry Smith #endif 447cddf8d76SBarry Smith DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color); 448416022c9SBarry Smith } 449416022c9SBarry Smith } 4500513a670SBarry Smith } else { 4510513a670SBarry Smith /* use contour shading to indicate magnitude of values */ 4520513a670SBarry Smith /* first determine max of all nonzero values */ 4530513a670SBarry Smith int nz = a->nz,count; 4540513a670SBarry Smith Draw popup; 4550513a670SBarry Smith 4560513a670SBarry Smith for ( i=0; i<nz; i++ ) { 4570513a670SBarry Smith if (PetscAbsScalar(a->a[i]) > maxv) maxv = PetscAbsScalar(a->a[i]); 4580513a670SBarry Smith } 4590513a670SBarry Smith ierr = DrawCreatePopUp(draw,&popup); CHKERRQ(ierr); 4600513a670SBarry Smith ierr = DrawScalePopup(popup,0.0,maxv); CHKERRQ(ierr); 4610513a670SBarry Smith count = 0; 4620513a670SBarry Smith for ( i=0; i<m; i++ ) { 4630513a670SBarry Smith y_l = m - i - 1.0; y_r = y_l + 1.0; 4640513a670SBarry Smith for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 4650513a670SBarry Smith x_l = a->j[j] + shift; x_r = x_l + 1.0; 4660513a670SBarry Smith color = 32 + (int) ((200.0 - 32.0)*PetscAbsScalar(a->a[count])/maxv); 4670513a670SBarry Smith DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color); 4680513a670SBarry Smith count++; 4690513a670SBarry Smith } 4700513a670SBarry Smith } 4710513a670SBarry Smith } 472416022c9SBarry Smith DrawFlush(draw); 473cddf8d76SBarry Smith DrawGetPause(draw,&pause); 474cddf8d76SBarry Smith if (pause >= 0) { PetscSleep(pause); return 0;} 475cddf8d76SBarry Smith 476cddf8d76SBarry Smith /* allow the matrix to zoom or shrink */ 4776945ee14SBarry Smith ierr = DrawCheckResizedWindow(draw); 478cddf8d76SBarry Smith ierr = DrawGetMouseButton(draw,&button,&xc,&yc,0,0); 479cddf8d76SBarry Smith while (button != BUTTON_RIGHT) { 480cddf8d76SBarry Smith DrawClear(draw); 481cddf8d76SBarry Smith if (button == BUTTON_LEFT) scale = .5; 482cddf8d76SBarry Smith else if (button == BUTTON_CENTER) scale = 2.; 483cddf8d76SBarry Smith xl = scale*(xl + w - xc) + xc - w*scale; 484cddf8d76SBarry Smith xr = scale*(xr - w - xc) + xc + w*scale; 485cddf8d76SBarry Smith yl = scale*(yl + h - yc) + yc - h*scale; 486cddf8d76SBarry Smith yr = scale*(yr - h - yc) + yc + h*scale; 487cddf8d76SBarry Smith w *= scale; h *= scale; 488cddf8d76SBarry Smith ierr = DrawSetCoordinates(draw,xl,yl,xr,yr); CHKERRQ(ierr); 4890513a670SBarry Smith if (format != VIEWER_FORMAT_DRAW_CONTOUR) { 4900513a670SBarry Smith /* Blue for negative, Cyan for zero and Red for positive */ 491cddf8d76SBarry Smith color = DRAW_BLUE; 492cddf8d76SBarry Smith for ( i=0; i<m; i++ ) { 493cddf8d76SBarry Smith y_l = m - i - 1.0; y_r = y_l + 1.0; 494cddf8d76SBarry Smith for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 495cddf8d76SBarry Smith x_l = a->j[j] + shift; x_r = x_l + 1.0; 496cddf8d76SBarry Smith #if defined(PETSC_COMPLEX) 497cddf8d76SBarry Smith if (real(a->a[j]) >= 0.) continue; 498cddf8d76SBarry Smith #else 499cddf8d76SBarry Smith if (a->a[j] >= 0.) continue; 500cddf8d76SBarry Smith #endif 501cddf8d76SBarry Smith DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color); 502cddf8d76SBarry Smith } 503cddf8d76SBarry Smith } 504cddf8d76SBarry Smith color = DRAW_CYAN; 505cddf8d76SBarry Smith for ( i=0; i<m; i++ ) { 506cddf8d76SBarry Smith y_l = m - i - 1.0; y_r = y_l + 1.0; 507cddf8d76SBarry Smith for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 508cddf8d76SBarry Smith x_l = a->j[j] + shift; x_r = x_l + 1.0; 509cddf8d76SBarry Smith if (a->a[j] != 0.) continue; 510cddf8d76SBarry Smith DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color); 511cddf8d76SBarry Smith } 512cddf8d76SBarry Smith } 513cddf8d76SBarry Smith color = DRAW_RED; 514cddf8d76SBarry Smith for ( i=0; i<m; i++ ) { 515cddf8d76SBarry Smith y_l = m - i - 1.0; y_r = y_l + 1.0; 516cddf8d76SBarry Smith for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 517cddf8d76SBarry Smith x_l = a->j[j] + shift; x_r = x_l + 1.0; 518cddf8d76SBarry Smith #if defined(PETSC_COMPLEX) 519cddf8d76SBarry Smith if (real(a->a[j]) <= 0.) continue; 520cddf8d76SBarry Smith #else 521cddf8d76SBarry Smith if (a->a[j] <= 0.) continue; 522cddf8d76SBarry Smith #endif 523cddf8d76SBarry Smith DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color); 524cddf8d76SBarry Smith } 525cddf8d76SBarry Smith } 5260513a670SBarry Smith } else { 5270513a670SBarry Smith /* use contour shading to indicate magnitude of values */ 5280513a670SBarry Smith int count = 0; 5290513a670SBarry Smith for ( i=0; i<m; i++ ) { 5300513a670SBarry Smith y_l = m - i - 1.0; y_r = y_l + 1.0; 5310513a670SBarry Smith for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 5320513a670SBarry Smith x_l = a->j[j] + shift; x_r = x_l + 1.0; 5330513a670SBarry Smith color = 32 + (int) ((200.0 - 32.0)*PetscAbsScalar(a->a[count])/maxv); 5340513a670SBarry Smith DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color); 5350513a670SBarry Smith count++; 5360513a670SBarry Smith } 5370513a670SBarry Smith } 5380513a670SBarry Smith } 5390513a670SBarry Smith 5406945ee14SBarry Smith ierr = DrawCheckResizedWindow(draw); 541cddf8d76SBarry Smith ierr = DrawGetMouseButton(draw,&button,&xc,&yc,0,0); 542cddf8d76SBarry Smith } 543416022c9SBarry Smith return 0; 544416022c9SBarry Smith } 545416022c9SBarry Smith 5465615d1e5SSatish Balay #undef __FUNC__ 547c22c1629SBarry Smith #define __FUNC__ "MatView_SeqAIJ" /* ADIC Ignore */ 5488f6be9afSLois Curfman McInnes int MatView_SeqAIJ(PetscObject obj,Viewer viewer) 549416022c9SBarry Smith { 550416022c9SBarry Smith Mat A = (Mat) obj; 551416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*) A->data; 552bcd2baecSBarry Smith ViewerType vtype; 553bcd2baecSBarry Smith int ierr; 554416022c9SBarry Smith 555bcd2baecSBarry Smith ierr = ViewerGetType(viewer,&vtype); CHKERRQ(ierr); 556bcd2baecSBarry Smith if (vtype == MATLAB_VIEWER) { 557416022c9SBarry Smith return ViewerMatlabPutSparse_Private(viewer,a->m,a->n,a->nz,a->a,a->i,a->j); 558416022c9SBarry Smith } 559bcd2baecSBarry Smith else if (vtype == ASCII_FILE_VIEWER || vtype == ASCII_FILES_VIEWER){ 560416022c9SBarry Smith return MatView_SeqAIJ_ASCII(A,viewer); 561416022c9SBarry Smith } 562bcd2baecSBarry Smith else if (vtype == BINARY_FILE_VIEWER) { 563416022c9SBarry Smith return MatView_SeqAIJ_Binary(A,viewer); 564416022c9SBarry Smith } 565bcd2baecSBarry Smith else if (vtype == DRAW_VIEWER) { 566bcd2baecSBarry Smith return MatView_SeqAIJ_Draw(A,viewer); 56717ab2063SBarry Smith } 56817ab2063SBarry Smith return 0; 56917ab2063SBarry Smith } 57019bcc07fSBarry Smith 571c456f294SBarry Smith extern int Mat_AIJ_CheckInode(Mat); 5725615d1e5SSatish Balay #undef __FUNC__ 5735615d1e5SSatish Balay #define __FUNC__ "MatAssemblyEnd_SeqAIJ" 5748f6be9afSLois Curfman McInnes int MatAssemblyEnd_SeqAIJ(Mat A,MatAssemblyType mode) 57517ab2063SBarry Smith { 576416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 57741c01911SSatish Balay int fshift = 0,i,j,*ai = a->i, *aj = a->j, *imax = a->imax,ierr; 57894a9d846SBarry Smith int m = a->m, *ip, N, *ailen = a->ilen,shift = a->indexshift,rmax; 579416022c9SBarry Smith Scalar *aa = a->a, *ap; 58017ab2063SBarry Smith 5816d4a8577SBarry Smith if (mode == MAT_FLUSH_ASSEMBLY) return 0; 58217ab2063SBarry Smith 58394a9d846SBarry Smith rmax = ailen[0]; /* determine row with most nonzeros */ 58417ab2063SBarry Smith for ( i=1; i<m; i++ ) { 585416022c9SBarry Smith /* move each row back by the amount of empty slots (fshift) before it*/ 58617ab2063SBarry Smith fshift += imax[i-1] - ailen[i-1]; 58794a9d846SBarry Smith rmax = PetscMax(rmax,ailen[i]); 58817ab2063SBarry Smith if (fshift) { 589416022c9SBarry Smith ip = aj + ai[i] + shift; ap = aa + ai[i] + shift; 59017ab2063SBarry Smith N = ailen[i]; 59117ab2063SBarry Smith for ( j=0; j<N; j++ ) { 59217ab2063SBarry Smith ip[j-fshift] = ip[j]; 59317ab2063SBarry Smith ap[j-fshift] = ap[j]; 59417ab2063SBarry Smith } 59517ab2063SBarry Smith } 59617ab2063SBarry Smith ai[i] = ai[i-1] + ailen[i-1]; 59717ab2063SBarry Smith } 59817ab2063SBarry Smith if (m) { 59917ab2063SBarry Smith fshift += imax[m-1] - ailen[m-1]; 60017ab2063SBarry Smith ai[m] = ai[m-1] + ailen[m-1]; 60117ab2063SBarry Smith } 60217ab2063SBarry Smith /* reset ilen and imax for each row */ 60317ab2063SBarry Smith for ( i=0; i<m; i++ ) { 60417ab2063SBarry Smith ailen[i] = imax[i] = ai[i+1] - ai[i]; 60517ab2063SBarry Smith } 606416022c9SBarry Smith a->nz = ai[m] + shift; 60717ab2063SBarry Smith 60817ab2063SBarry Smith /* diagonals may have moved, so kill the diagonal pointers */ 609416022c9SBarry Smith if (fshift && a->diag) { 6100452661fSBarry Smith PetscFree(a->diag); 611416022c9SBarry Smith PLogObjectMemory(A,-(m+1)*sizeof(int)); 612416022c9SBarry Smith a->diag = 0; 61317ab2063SBarry Smith } 6144e220ebcSLois Curfman McInnes PLogInfo(A,"MatAssemblyEnd_SeqAIJ:Matrix size: %d X %d; storage space: %d unneeded, %d used\n", 6154e220ebcSLois Curfman McInnes m,a->n,fshift,a->nz); 6164e220ebcSLois Curfman McInnes PLogInfo(A,"MatAssemblyEnd_SeqAIJ:Number of mallocs during MatSetValues is %d\n", 617b810aeb4SBarry Smith a->reallocs); 61894a9d846SBarry Smith PLogInfo(A,"MatAssemblyEnd_SeqAIJ:Most nonzeros in any row is %d\n",rmax); 619dd5f02e7SSatish Balay a->reallocs = 0; 6204e220ebcSLois Curfman McInnes A->info.nz_unneeded = (double)fshift; 6214e220ebcSLois Curfman McInnes 62276dd722bSSatish Balay /* check out for identical nodes. If found, use inode functions */ 62341c01911SSatish Balay ierr = Mat_AIJ_CheckInode(A); CHKERRQ(ierr); 62417ab2063SBarry Smith return 0; 62517ab2063SBarry Smith } 62617ab2063SBarry Smith 6275615d1e5SSatish Balay #undef __FUNC__ 6285615d1e5SSatish Balay #define __FUNC__ "MatZeroEntries_SeqAIJ" 6298f6be9afSLois Curfman McInnes int MatZeroEntries_SeqAIJ(Mat A) 63017ab2063SBarry Smith { 631416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 632cddf8d76SBarry Smith PetscMemzero(a->a,(a->i[a->m]+a->indexshift)*sizeof(Scalar)); 63317ab2063SBarry Smith return 0; 63417ab2063SBarry Smith } 635416022c9SBarry Smith 6365615d1e5SSatish Balay #undef __FUNC__ 6375615d1e5SSatish Balay #define __FUNC__ "MatDestroy_SeqAIJ" 63817ab2063SBarry Smith int MatDestroy_SeqAIJ(PetscObject obj) 63917ab2063SBarry Smith { 640416022c9SBarry Smith Mat A = (Mat) obj; 641416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 64290f02eecSBarry Smith int ierr; 643d5d45c9bSBarry Smith 64417ab2063SBarry Smith #if defined(PETSC_LOG) 645416022c9SBarry Smith PLogObjectState(obj,"Rows=%d, Cols=%d, NZ=%d",a->m,a->n,a->nz); 64617ab2063SBarry Smith #endif 6470452661fSBarry Smith PetscFree(a->a); 6480452661fSBarry Smith if (!a->singlemalloc) { PetscFree(a->i); PetscFree(a->j);} 6490452661fSBarry Smith if (a->diag) PetscFree(a->diag); 6500452661fSBarry Smith if (a->ilen) PetscFree(a->ilen); 6510452661fSBarry Smith if (a->imax) PetscFree(a->imax); 6520452661fSBarry Smith if (a->solve_work) PetscFree(a->solve_work); 65376dd722bSSatish Balay if (a->inode.size) PetscFree(a->inode.size); 6540452661fSBarry Smith PetscFree(a); 65590f02eecSBarry Smith if (A->mapping) { 65690f02eecSBarry Smith ierr = ISLocalToGlobalMappingDestroy(A->mapping); CHKERRQ(ierr); 65790f02eecSBarry Smith } 658f2655603SLois Curfman McInnes PLogObjectDestroy(A); 659f2655603SLois Curfman McInnes PetscHeaderDestroy(A); 66017ab2063SBarry Smith return 0; 66117ab2063SBarry Smith } 66217ab2063SBarry Smith 6635615d1e5SSatish Balay #undef __FUNC__ 6645615d1e5SSatish Balay #define __FUNC__ "MatCompress_SeqAIJ" 6658f6be9afSLois Curfman McInnes int MatCompress_SeqAIJ(Mat A) 66617ab2063SBarry Smith { 66717ab2063SBarry Smith return 0; 66817ab2063SBarry Smith } 66917ab2063SBarry Smith 6705615d1e5SSatish Balay #undef __FUNC__ 6715615d1e5SSatish Balay #define __FUNC__ "MatSetOption_SeqAIJ" 6728f6be9afSLois Curfman McInnes int MatSetOption_SeqAIJ(Mat A,MatOption op) 67317ab2063SBarry Smith { 674416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 6756d4a8577SBarry Smith if (op == MAT_ROW_ORIENTED) a->roworiented = 1; 6766d4a8577SBarry Smith else if (op == MAT_COLUMN_ORIENTED) a->roworiented = 0; 6776d4a8577SBarry Smith else if (op == MAT_COLUMNS_SORTED) a->sorted = 1; 678219d9a1aSLois Curfman McInnes else if (op == MAT_COLUMNS_UNSORTED) a->sorted = 0; 6796d4a8577SBarry Smith else if (op == MAT_NO_NEW_NONZERO_LOCATIONS) a->nonew = 1; 680c2653b3dSLois Curfman McInnes else if (op == MAT_NEW_NONZERO_LOCATION_ERROR) a->nonew = -1; 68196854ed6SLois Curfman McInnes else if (op == MAT_NEW_NONZERO_ALLOCATION_ERROR) a->nonew = -2; 6826d4a8577SBarry Smith else if (op == MAT_YES_NEW_NONZERO_LOCATIONS) a->nonew = 0; 6836d4a8577SBarry Smith else if (op == MAT_ROWS_SORTED || 684219d9a1aSLois Curfman McInnes op == MAT_ROWS_UNSORTED || 6856d4a8577SBarry Smith op == MAT_SYMMETRIC || 6866d4a8577SBarry Smith op == MAT_STRUCTURALLY_SYMMETRIC || 68790f02eecSBarry Smith op == MAT_YES_NEW_DIAGONALS || 6882b362799SSatish Balay op == MAT_IGNORE_OFF_PROC_ENTRIES) 68994a424c1SBarry Smith PLogInfo(A,"Info:MatSetOption_SeqAIJ:Option ignored\n"); 6906d4a8577SBarry Smith else if (op == MAT_NO_NEW_DIAGONALS) 691e3372554SBarry Smith {SETERRQ(PETSC_ERR_SUP,0,"MAT_NO_NEW_DIAGONALS");} 6926d4a8577SBarry Smith else if (op == MAT_INODE_LIMIT_1) a->inode.limit = 1; 6936d4a8577SBarry Smith else if (op == MAT_INODE_LIMIT_2) a->inode.limit = 2; 6946d4a8577SBarry Smith else if (op == MAT_INODE_LIMIT_3) a->inode.limit = 3; 6956d4a8577SBarry Smith else if (op == MAT_INODE_LIMIT_4) a->inode.limit = 4; 6966d4a8577SBarry Smith else if (op == MAT_INODE_LIMIT_5) a->inode.limit = 5; 697e2f28af5SBarry Smith else 698e3372554SBarry Smith {SETERRQ(PETSC_ERR_SUP,0,"unknown option");} 69917ab2063SBarry Smith return 0; 70017ab2063SBarry Smith } 70117ab2063SBarry Smith 7025615d1e5SSatish Balay #undef __FUNC__ 7035615d1e5SSatish Balay #define __FUNC__ "MatGetDiagonal_SeqAIJ" 7048f6be9afSLois Curfman McInnes int MatGetDiagonal_SeqAIJ(Mat A,Vec v) 70517ab2063SBarry Smith { 706416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 707416022c9SBarry Smith int i,j, n,shift = a->indexshift; 70817ab2063SBarry Smith Scalar *x, zero = 0.0; 70917ab2063SBarry Smith 71017ab2063SBarry Smith VecSet(&zero,v); 71190f02eecSBarry Smith VecGetArray_Fast(v,x); VecGetLocalSize(v,&n); 712e3372554SBarry Smith if (n != a->m) SETERRQ(1,0,"Nonconforming matrix and vector"); 713416022c9SBarry Smith for ( i=0; i<a->m; i++ ) { 714416022c9SBarry Smith for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 715416022c9SBarry Smith if (a->j[j]+shift == i) { 716416022c9SBarry Smith x[i] = a->a[j]; 71717ab2063SBarry Smith break; 71817ab2063SBarry Smith } 71917ab2063SBarry Smith } 72017ab2063SBarry Smith } 72117ab2063SBarry Smith return 0; 72217ab2063SBarry Smith } 72317ab2063SBarry Smith 72417ab2063SBarry Smith /* -------------------------------------------------------*/ 72517ab2063SBarry Smith /* Should check that shapes of vectors and matrices match */ 72617ab2063SBarry Smith /* -------------------------------------------------------*/ 7275615d1e5SSatish Balay #undef __FUNC__ 7285615d1e5SSatish Balay #define __FUNC__ "MatMultTrans_SeqAIJ" 72944cd7ae7SLois Curfman McInnes int MatMultTrans_SeqAIJ(Mat A,Vec xx,Vec yy) 73017ab2063SBarry Smith { 731416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 73217ab2063SBarry Smith Scalar *x, *y, *v, alpha; 733416022c9SBarry Smith int m = a->m, n, i, *idx, shift = a->indexshift; 73417ab2063SBarry Smith 73590f02eecSBarry Smith VecGetArray_Fast(xx,x); VecGetArray_Fast(yy,y); 736cddf8d76SBarry Smith PetscMemzero(y,a->n*sizeof(Scalar)); 73717ab2063SBarry Smith y = y + shift; /* shift for Fortran start by 1 indexing */ 73817ab2063SBarry Smith for ( i=0; i<m; i++ ) { 739416022c9SBarry Smith idx = a->j + a->i[i] + shift; 740416022c9SBarry Smith v = a->a + a->i[i] + shift; 741416022c9SBarry Smith n = a->i[i+1] - a->i[i]; 74217ab2063SBarry Smith alpha = x[i]; 74317ab2063SBarry Smith while (n-->0) {y[*idx++] += alpha * *v++;} 74417ab2063SBarry Smith } 745416022c9SBarry Smith PLogFlops(2*a->nz - a->n); 74617ab2063SBarry Smith return 0; 74717ab2063SBarry Smith } 748d5d45c9bSBarry Smith 7495615d1e5SSatish Balay #undef __FUNC__ 7505615d1e5SSatish Balay #define __FUNC__ "MatMultTransAdd_SeqAIJ" 75144cd7ae7SLois Curfman McInnes int MatMultTransAdd_SeqAIJ(Mat A,Vec xx,Vec zz,Vec yy) 75217ab2063SBarry Smith { 753416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 75417ab2063SBarry Smith Scalar *x, *y, *v, alpha; 755416022c9SBarry Smith int m = a->m, n, i, *idx,shift = a->indexshift; 75617ab2063SBarry Smith 75790f02eecSBarry Smith VecGetArray_Fast(xx,x); VecGetArray_Fast(yy,y); 75817ab2063SBarry Smith if (zz != yy) VecCopy(zz,yy); 75917ab2063SBarry Smith y = y + shift; /* shift for Fortran start by 1 indexing */ 76017ab2063SBarry Smith for ( i=0; i<m; i++ ) { 761416022c9SBarry Smith idx = a->j + a->i[i] + shift; 762416022c9SBarry Smith v = a->a + a->i[i] + shift; 763416022c9SBarry Smith n = a->i[i+1] - a->i[i]; 76417ab2063SBarry Smith alpha = x[i]; 76517ab2063SBarry Smith while (n-->0) {y[*idx++] += alpha * *v++;} 76617ab2063SBarry Smith } 76790f02eecSBarry Smith PLogFlops(2*a->nz); 76817ab2063SBarry Smith return 0; 76917ab2063SBarry Smith } 77017ab2063SBarry Smith 7715615d1e5SSatish Balay #undef __FUNC__ 7725615d1e5SSatish Balay #define __FUNC__ "MatMult_SeqAIJ" 77344cd7ae7SLois Curfman McInnes int MatMult_SeqAIJ(Mat A,Vec xx,Vec yy) 77417ab2063SBarry Smith { 775416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 77617ab2063SBarry Smith Scalar *x, *y, *v, sum; 7779ea0dfa2SSatish Balay int m = a->m, n, i, *idx, shift = a->indexshift,*ii,jrow,j; 77817ab2063SBarry Smith 77990f02eecSBarry Smith VecGetArray_Fast(xx,x); VecGetArray_Fast(yy,y); 78017ab2063SBarry Smith x = x + shift; /* shift for Fortran start by 1 indexing */ 781416022c9SBarry Smith idx = a->j; 782416022c9SBarry Smith v = a->a; 783416022c9SBarry Smith ii = a->i; 78417ab2063SBarry Smith for ( i=0; i<m; i++ ) { 7859ea0dfa2SSatish Balay jrow = ii[i]; 7869ea0dfa2SSatish Balay n = ii[i+1] - jrow; 78717ab2063SBarry Smith sum = 0.0; 7889ea0dfa2SSatish Balay /* while (n--) sum += *v++ * x[*idx++]; */ 7899ea0dfa2SSatish Balay for ( j=0; j<n; j++) { 7909ea0dfa2SSatish Balay sum += v[jrow]*x[idx[jrow]]; jrow++; 7919ea0dfa2SSatish Balay } 79217ab2063SBarry Smith y[i] = sum; 79317ab2063SBarry Smith } 794416022c9SBarry Smith PLogFlops(2*a->nz - m); 79517ab2063SBarry Smith return 0; 79617ab2063SBarry Smith } 79717ab2063SBarry Smith 7985615d1e5SSatish Balay #undef __FUNC__ 7995615d1e5SSatish Balay #define __FUNC__ "MatMultAdd_SeqAIJ" 80044cd7ae7SLois Curfman McInnes int MatMultAdd_SeqAIJ(Mat A,Vec xx,Vec yy,Vec zz) 80117ab2063SBarry Smith { 802416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 80317ab2063SBarry Smith Scalar *x, *y, *z, *v, sum; 8049ea0dfa2SSatish Balay int m = a->m, n, i, *idx, shift = a->indexshift,*ii,jrow,j; 8059ea0dfa2SSatish Balay 80617ab2063SBarry Smith 80790f02eecSBarry Smith VecGetArray_Fast(xx,x); VecGetArray_Fast(yy,y); VecGetArray_Fast(zz,z); 80817ab2063SBarry Smith x = x + shift; /* shift for Fortran start by 1 indexing */ 809cddf8d76SBarry Smith idx = a->j; 810cddf8d76SBarry Smith v = a->a; 811cddf8d76SBarry Smith ii = a->i; 81217ab2063SBarry Smith for ( i=0; i<m; i++ ) { 8139ea0dfa2SSatish Balay jrow = ii[i]; 8149ea0dfa2SSatish Balay n = ii[i+1] - jrow; 81517ab2063SBarry Smith sum = y[i]; 8169ea0dfa2SSatish Balay /* while (n--) sum += *v++ * x[*idx++]; */ 8179ea0dfa2SSatish Balay for ( j=0; j<n; j++) { 8189ea0dfa2SSatish Balay sum += v[jrow]*x[idx[jrow]]; jrow++; 8199ea0dfa2SSatish Balay } 82017ab2063SBarry Smith z[i] = sum; 82117ab2063SBarry Smith } 822416022c9SBarry Smith PLogFlops(2*a->nz); 82317ab2063SBarry Smith return 0; 82417ab2063SBarry Smith } 82517ab2063SBarry Smith 82617ab2063SBarry Smith /* 82717ab2063SBarry Smith Adds diagonal pointers to sparse matrix structure. 82817ab2063SBarry Smith */ 82917ab2063SBarry Smith 8305615d1e5SSatish Balay #undef __FUNC__ 8315615d1e5SSatish Balay #define __FUNC__ "MatMarkDiag_SeqAIJ" 832416022c9SBarry Smith int MatMarkDiag_SeqAIJ(Mat A) 83317ab2063SBarry Smith { 834416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 835416022c9SBarry Smith int i,j, *diag, m = a->m,shift = a->indexshift; 83617ab2063SBarry Smith 8370452661fSBarry Smith diag = (int *) PetscMalloc( (m+1)*sizeof(int)); CHKPTRQ(diag); 838416022c9SBarry Smith PLogObjectMemory(A,(m+1)*sizeof(int)); 839416022c9SBarry Smith for ( i=0; i<a->m; i++ ) { 840416022c9SBarry Smith for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 841416022c9SBarry Smith if (a->j[j]+shift == i) { 84217ab2063SBarry Smith diag[i] = j - shift; 84317ab2063SBarry Smith break; 84417ab2063SBarry Smith } 84517ab2063SBarry Smith } 84617ab2063SBarry Smith } 847416022c9SBarry Smith a->diag = diag; 84817ab2063SBarry Smith return 0; 84917ab2063SBarry Smith } 85017ab2063SBarry Smith 8515615d1e5SSatish Balay #undef __FUNC__ 8525615d1e5SSatish Balay #define __FUNC__ "MatRelax_SeqAIJ" 85344cd7ae7SLois Curfman McInnes int MatRelax_SeqAIJ(Mat A,Vec bb,double omega,MatSORType flag, 85417ab2063SBarry Smith double fshift,int its,Vec xx) 85517ab2063SBarry Smith { 856416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 857416022c9SBarry Smith Scalar *x, *b, *bs, d, *xs, sum, *v = a->a,*t,scale,*ts, *xb; 858d5d45c9bSBarry Smith int ierr, *idx, *diag,n = a->n, m = a->m, i, shift = a->indexshift; 85917ab2063SBarry Smith 86090f02eecSBarry Smith VecGetArray_Fast(xx,x); VecGetArray_Fast(bb,b); 861416022c9SBarry Smith if (!a->diag) {if ((ierr = MatMarkDiag_SeqAIJ(A))) return ierr;} 862416022c9SBarry Smith diag = a->diag; 863416022c9SBarry Smith xs = x + shift; /* shifted by one for index start of a or a->j*/ 86417ab2063SBarry Smith if (flag == SOR_APPLY_UPPER) { 86517ab2063SBarry Smith /* apply ( U + D/omega) to the vector */ 86617ab2063SBarry Smith bs = b + shift; 86717ab2063SBarry Smith for ( i=0; i<m; i++ ) { 868416022c9SBarry Smith d = fshift + a->a[diag[i] + shift]; 869416022c9SBarry Smith n = a->i[i+1] - diag[i] - 1; 870416022c9SBarry Smith idx = a->j + diag[i] + (!shift); 871416022c9SBarry Smith v = a->a + diag[i] + (!shift); 87217ab2063SBarry Smith sum = b[i]*d/omega; 87317ab2063SBarry Smith SPARSEDENSEDOT(sum,bs,v,idx,n); 87417ab2063SBarry Smith x[i] = sum; 87517ab2063SBarry Smith } 87617ab2063SBarry Smith return 0; 87717ab2063SBarry Smith } 87817ab2063SBarry Smith if (flag == SOR_APPLY_LOWER) { 879e3372554SBarry Smith SETERRQ(1,0,"SOR_APPLY_LOWER is not done"); 88017ab2063SBarry Smith } 881416022c9SBarry Smith else if (flag & SOR_EISENSTAT) { 88217ab2063SBarry Smith /* Let A = L + U + D; where L is lower trianglar, 88317ab2063SBarry Smith U is upper triangular, E is diagonal; This routine applies 88417ab2063SBarry Smith 88517ab2063SBarry Smith (L + E)^{-1} A (U + E)^{-1} 88617ab2063SBarry Smith 88717ab2063SBarry Smith to a vector efficiently using Eisenstat's trick. This is for 88817ab2063SBarry Smith the case of SSOR preconditioner, so E is D/omega where omega 88917ab2063SBarry Smith is the relaxation factor. 89017ab2063SBarry Smith */ 8910452661fSBarry Smith t = (Scalar *) PetscMalloc( m*sizeof(Scalar) ); CHKPTRQ(t); 89217ab2063SBarry Smith scale = (2.0/omega) - 1.0; 89317ab2063SBarry Smith 89417ab2063SBarry Smith /* x = (E + U)^{-1} b */ 89517ab2063SBarry Smith for ( i=m-1; i>=0; i-- ) { 896416022c9SBarry Smith d = fshift + a->a[diag[i] + shift]; 897416022c9SBarry Smith n = a->i[i+1] - diag[i] - 1; 898416022c9SBarry Smith idx = a->j + diag[i] + (!shift); 899416022c9SBarry Smith v = a->a + diag[i] + (!shift); 90017ab2063SBarry Smith sum = b[i]; 90117ab2063SBarry Smith SPARSEDENSEMDOT(sum,xs,v,idx,n); 90217ab2063SBarry Smith x[i] = omega*(sum/d); 90317ab2063SBarry Smith } 90417ab2063SBarry Smith 90517ab2063SBarry Smith /* t = b - (2*E - D)x */ 906416022c9SBarry Smith v = a->a; 90717ab2063SBarry Smith for ( i=0; i<m; i++ ) { t[i] = b[i] - scale*(v[*diag++ + shift])*x[i]; } 90817ab2063SBarry Smith 90917ab2063SBarry Smith /* t = (E + L)^{-1}t */ 910416022c9SBarry Smith ts = t + shift; /* shifted by one for index start of a or a->j*/ 911416022c9SBarry Smith diag = a->diag; 91217ab2063SBarry Smith for ( i=0; i<m; i++ ) { 913416022c9SBarry Smith d = fshift + a->a[diag[i]+shift]; 914416022c9SBarry Smith n = diag[i] - a->i[i]; 915416022c9SBarry Smith idx = a->j + a->i[i] + shift; 916416022c9SBarry Smith v = a->a + a->i[i] + shift; 91717ab2063SBarry Smith sum = t[i]; 91817ab2063SBarry Smith SPARSEDENSEMDOT(sum,ts,v,idx,n); 91917ab2063SBarry Smith t[i] = omega*(sum/d); 92017ab2063SBarry Smith } 92117ab2063SBarry Smith 92217ab2063SBarry Smith /* x = x + t */ 92317ab2063SBarry Smith for ( i=0; i<m; i++ ) { x[i] += t[i]; } 9240452661fSBarry Smith PetscFree(t); 92517ab2063SBarry Smith return 0; 92617ab2063SBarry Smith } 92717ab2063SBarry Smith if (flag & SOR_ZERO_INITIAL_GUESS) { 92817ab2063SBarry Smith if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP){ 92917ab2063SBarry Smith for ( i=0; i<m; i++ ) { 930416022c9SBarry Smith d = fshift + a->a[diag[i]+shift]; 931416022c9SBarry Smith n = diag[i] - a->i[i]; 932416022c9SBarry Smith idx = a->j + a->i[i] + shift; 933416022c9SBarry Smith v = a->a + a->i[i] + shift; 93417ab2063SBarry Smith sum = b[i]; 93517ab2063SBarry Smith SPARSEDENSEMDOT(sum,xs,v,idx,n); 93617ab2063SBarry Smith x[i] = omega*(sum/d); 93717ab2063SBarry Smith } 93817ab2063SBarry Smith xb = x; 93917ab2063SBarry Smith } 94017ab2063SBarry Smith else xb = b; 94117ab2063SBarry Smith if ((flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP) && 94217ab2063SBarry Smith (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP)) { 94317ab2063SBarry Smith for ( i=0; i<m; i++ ) { 944416022c9SBarry Smith x[i] *= a->a[diag[i]+shift]; 94517ab2063SBarry Smith } 94617ab2063SBarry Smith } 94717ab2063SBarry Smith if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP){ 94817ab2063SBarry Smith for ( i=m-1; i>=0; i-- ) { 949416022c9SBarry Smith d = fshift + a->a[diag[i] + shift]; 950416022c9SBarry Smith n = a->i[i+1] - diag[i] - 1; 951416022c9SBarry Smith idx = a->j + diag[i] + (!shift); 952416022c9SBarry Smith v = a->a + diag[i] + (!shift); 95317ab2063SBarry Smith sum = xb[i]; 95417ab2063SBarry Smith SPARSEDENSEMDOT(sum,xs,v,idx,n); 95517ab2063SBarry Smith x[i] = omega*(sum/d); 95617ab2063SBarry Smith } 95717ab2063SBarry Smith } 95817ab2063SBarry Smith its--; 95917ab2063SBarry Smith } 96017ab2063SBarry Smith while (its--) { 96117ab2063SBarry Smith if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP){ 96217ab2063SBarry Smith for ( i=0; i<m; i++ ) { 963416022c9SBarry Smith d = fshift + a->a[diag[i]+shift]; 964416022c9SBarry Smith n = a->i[i+1] - a->i[i]; 965416022c9SBarry Smith idx = a->j + a->i[i] + shift; 966416022c9SBarry Smith v = a->a + a->i[i] + shift; 96717ab2063SBarry Smith sum = b[i]; 96817ab2063SBarry Smith SPARSEDENSEMDOT(sum,xs,v,idx,n); 9697e33a6baSBarry Smith x[i] = (1. - omega)*x[i] + omega*(sum + a->a[diag[i]+shift]*x[i])/d; 97017ab2063SBarry Smith } 97117ab2063SBarry Smith } 97217ab2063SBarry Smith if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP){ 97317ab2063SBarry Smith for ( i=m-1; i>=0; i-- ) { 974416022c9SBarry Smith d = fshift + a->a[diag[i] + shift]; 975416022c9SBarry Smith n = a->i[i+1] - a->i[i]; 976416022c9SBarry Smith idx = a->j + a->i[i] + shift; 977416022c9SBarry Smith v = a->a + a->i[i] + shift; 97817ab2063SBarry Smith sum = b[i]; 97917ab2063SBarry Smith SPARSEDENSEMDOT(sum,xs,v,idx,n); 9807e33a6baSBarry Smith x[i] = (1. - omega)*x[i] + omega*(sum + a->a[diag[i]+shift]*x[i])/d; 98117ab2063SBarry Smith } 98217ab2063SBarry Smith } 98317ab2063SBarry Smith } 98417ab2063SBarry Smith return 0; 98517ab2063SBarry Smith } 98617ab2063SBarry Smith 9875615d1e5SSatish Balay #undef __FUNC__ 9885615d1e5SSatish Balay #define __FUNC__ "MatGetInfo_SeqAIJ" 9898f6be9afSLois Curfman McInnes int MatGetInfo_SeqAIJ(Mat A,MatInfoType flag,MatInfo *info) 99017ab2063SBarry Smith { 991416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 9924e220ebcSLois Curfman McInnes 9934e220ebcSLois Curfman McInnes info->rows_global = (double)a->m; 9944e220ebcSLois Curfman McInnes info->columns_global = (double)a->n; 9954e220ebcSLois Curfman McInnes info->rows_local = (double)a->m; 9964e220ebcSLois Curfman McInnes info->columns_local = (double)a->n; 9974e220ebcSLois Curfman McInnes info->block_size = 1.0; 9984e220ebcSLois Curfman McInnes info->nz_allocated = (double)a->maxnz; 9994e220ebcSLois Curfman McInnes info->nz_used = (double)a->nz; 10004e220ebcSLois Curfman McInnes info->nz_unneeded = (double)(a->maxnz - a->nz); 10014e220ebcSLois Curfman McInnes /* if (info->nz_unneeded != A->info.nz_unneeded) 10024e220ebcSLois Curfman McInnes printf("space descrepancy: maxnz-nz = %d, nz_unneeded = %d\n",(int)info->nz_unneeded,(int)A->info.nz_unneeded); */ 10034e220ebcSLois Curfman McInnes info->assemblies = (double)A->num_ass; 10044e220ebcSLois Curfman McInnes info->mallocs = (double)a->reallocs; 10054e220ebcSLois Curfman McInnes info->memory = A->mem; 10064e220ebcSLois Curfman McInnes if (A->factor) { 10074e220ebcSLois Curfman McInnes info->fill_ratio_given = A->info.fill_ratio_given; 10084e220ebcSLois Curfman McInnes info->fill_ratio_needed = A->info.fill_ratio_needed; 10094e220ebcSLois Curfman McInnes info->factor_mallocs = A->info.factor_mallocs; 10104e220ebcSLois Curfman McInnes } else { 10114e220ebcSLois Curfman McInnes info->fill_ratio_given = 0; 10124e220ebcSLois Curfman McInnes info->fill_ratio_needed = 0; 10134e220ebcSLois Curfman McInnes info->factor_mallocs = 0; 10144e220ebcSLois Curfman McInnes } 101517ab2063SBarry Smith return 0; 101617ab2063SBarry Smith } 101717ab2063SBarry Smith 101817ab2063SBarry Smith extern int MatLUFactorSymbolic_SeqAIJ(Mat,IS,IS,double,Mat*); 101917ab2063SBarry Smith extern int MatLUFactorNumeric_SeqAIJ(Mat,Mat*); 102017ab2063SBarry Smith extern int MatLUFactor_SeqAIJ(Mat,IS,IS,double); 102117ab2063SBarry Smith extern int MatSolve_SeqAIJ(Mat,Vec,Vec); 102217ab2063SBarry Smith extern int MatSolveAdd_SeqAIJ(Mat,Vec,Vec,Vec); 102317ab2063SBarry Smith extern int MatSolveTrans_SeqAIJ(Mat,Vec,Vec); 102417ab2063SBarry Smith extern int MatSolveTransAdd_SeqAIJ(Mat,Vec,Vec,Vec); 102517ab2063SBarry Smith 10265615d1e5SSatish Balay #undef __FUNC__ 10275615d1e5SSatish Balay #define __FUNC__ "MatZeroRows_SeqAIJ" 10288f6be9afSLois Curfman McInnes int MatZeroRows_SeqAIJ(Mat A,IS is,Scalar *diag) 102917ab2063SBarry Smith { 1030416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 1031416022c9SBarry Smith int i,ierr,N, *rows,m = a->m - 1,shift = a->indexshift; 103217ab2063SBarry Smith 103377c4ece6SBarry Smith ierr = ISGetSize(is,&N); CHKERRQ(ierr); 103417ab2063SBarry Smith ierr = ISGetIndices(is,&rows); CHKERRQ(ierr); 103517ab2063SBarry Smith if (diag) { 103617ab2063SBarry Smith for ( i=0; i<N; i++ ) { 1037e3372554SBarry Smith if (rows[i] < 0 || rows[i] > m) SETERRQ(1,0,"row out of range"); 1038416022c9SBarry Smith if (a->ilen[rows[i]] > 0) { /* in case row was completely empty */ 1039416022c9SBarry Smith a->ilen[rows[i]] = 1; 1040416022c9SBarry Smith a->a[a->i[rows[i]]+shift] = *diag; 1041416022c9SBarry Smith a->j[a->i[rows[i]]+shift] = rows[i]+shift; 104217ab2063SBarry Smith } 104317ab2063SBarry Smith else { 104417ab2063SBarry Smith ierr = MatSetValues_SeqAIJ(A,1,&rows[i],1,&rows[i],diag,INSERT_VALUES); 104517ab2063SBarry Smith CHKERRQ(ierr); 104617ab2063SBarry Smith } 104717ab2063SBarry Smith } 104817ab2063SBarry Smith } 104917ab2063SBarry Smith else { 105017ab2063SBarry Smith for ( i=0; i<N; i++ ) { 1051e3372554SBarry Smith if (rows[i] < 0 || rows[i] > m) SETERRQ(1,0,"row out of range"); 1052416022c9SBarry Smith a->ilen[rows[i]] = 0; 105317ab2063SBarry Smith } 105417ab2063SBarry Smith } 105517ab2063SBarry Smith ISRestoreIndices(is,&rows); 105643a90d84SBarry Smith ierr = MatAssemblyEnd_SeqAIJ(A,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr); 105717ab2063SBarry Smith return 0; 105817ab2063SBarry Smith } 105917ab2063SBarry Smith 10605615d1e5SSatish Balay #undef __FUNC__ 10615615d1e5SSatish Balay #define __FUNC__ "MatGetSize_SeqAIJ" 10628f6be9afSLois Curfman McInnes int MatGetSize_SeqAIJ(Mat A,int *m,int *n) 106317ab2063SBarry Smith { 1064416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 1065416022c9SBarry Smith *m = a->m; *n = a->n; 106617ab2063SBarry Smith return 0; 106717ab2063SBarry Smith } 106817ab2063SBarry Smith 10695615d1e5SSatish Balay #undef __FUNC__ 10705615d1e5SSatish Balay #define __FUNC__ "MatGetOwnershipRange_SeqAIJ" 10718f6be9afSLois Curfman McInnes int MatGetOwnershipRange_SeqAIJ(Mat A,int *m,int *n) 107217ab2063SBarry Smith { 1073416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 1074416022c9SBarry Smith *m = 0; *n = a->m; 107517ab2063SBarry Smith return 0; 107617ab2063SBarry Smith } 1077026e39d0SSatish Balay 10785615d1e5SSatish Balay #undef __FUNC__ 10795615d1e5SSatish Balay #define __FUNC__ "MatGetRow_SeqAIJ" 10804e093b46SBarry Smith int MatGetRow_SeqAIJ(Mat A,int row,int *nz,int **idx,Scalar **v) 108117ab2063SBarry Smith { 1082416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 1083c456f294SBarry Smith int *itmp,i,shift = a->indexshift; 108417ab2063SBarry Smith 1085e3372554SBarry Smith if (row < 0 || row >= a->m) SETERRQ(1,0,"Row out of range"); 108617ab2063SBarry Smith 1087416022c9SBarry Smith *nz = a->i[row+1] - a->i[row]; 1088416022c9SBarry Smith if (v) *v = a->a + a->i[row] + shift; 108917ab2063SBarry Smith if (idx) { 1090416022c9SBarry Smith itmp = a->j + a->i[row] + shift; 10914e093b46SBarry Smith if (*nz && shift) { 10920452661fSBarry Smith *idx = (int *) PetscMalloc( (*nz)*sizeof(int) ); CHKPTRQ(*idx); 109317ab2063SBarry Smith for ( i=0; i<(*nz); i++ ) {(*idx)[i] = itmp[i] + shift;} 10944e093b46SBarry Smith } else if (*nz) { 10954e093b46SBarry Smith *idx = itmp; 109617ab2063SBarry Smith } 109717ab2063SBarry Smith else *idx = 0; 109817ab2063SBarry Smith } 109917ab2063SBarry Smith return 0; 110017ab2063SBarry Smith } 110117ab2063SBarry Smith 11025615d1e5SSatish Balay #undef __FUNC__ 11035615d1e5SSatish Balay #define __FUNC__ "MatRestoreRow_SeqAIJ" 11044e093b46SBarry Smith int MatRestoreRow_SeqAIJ(Mat A,int row,int *nz,int **idx,Scalar **v) 110517ab2063SBarry Smith { 11064e093b46SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 11074e093b46SBarry Smith if (idx) {if (*idx && a->indexshift) PetscFree(*idx);} 110817ab2063SBarry Smith return 0; 110917ab2063SBarry Smith } 111017ab2063SBarry Smith 11115615d1e5SSatish Balay #undef __FUNC__ 11125615d1e5SSatish Balay #define __FUNC__ "MatNorm_SeqAIJ" 11138f6be9afSLois Curfman McInnes int MatNorm_SeqAIJ(Mat A,NormType type,double *norm) 111417ab2063SBarry Smith { 1115416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 1116416022c9SBarry Smith Scalar *v = a->a; 111717ab2063SBarry Smith double sum = 0.0; 1118416022c9SBarry Smith int i, j,shift = a->indexshift; 111917ab2063SBarry Smith 112017ab2063SBarry Smith if (type == NORM_FROBENIUS) { 1121416022c9SBarry Smith for (i=0; i<a->nz; i++ ) { 112217ab2063SBarry Smith #if defined(PETSC_COMPLEX) 112317ab2063SBarry Smith sum += real(conj(*v)*(*v)); v++; 112417ab2063SBarry Smith #else 112517ab2063SBarry Smith sum += (*v)*(*v); v++; 112617ab2063SBarry Smith #endif 112717ab2063SBarry Smith } 112817ab2063SBarry Smith *norm = sqrt(sum); 112917ab2063SBarry Smith } 113017ab2063SBarry Smith else if (type == NORM_1) { 113117ab2063SBarry Smith double *tmp; 1132416022c9SBarry Smith int *jj = a->j; 113366963ce1SSatish Balay tmp = (double *) PetscMalloc( (a->n+1)*sizeof(double) ); CHKPTRQ(tmp); 1134cddf8d76SBarry Smith PetscMemzero(tmp,a->n*sizeof(double)); 113517ab2063SBarry Smith *norm = 0.0; 1136416022c9SBarry Smith for ( j=0; j<a->nz; j++ ) { 1137a2744918SBarry Smith tmp[*jj++ + shift] += PetscAbsScalar(*v); v++; 113817ab2063SBarry Smith } 1139416022c9SBarry Smith for ( j=0; j<a->n; j++ ) { 114017ab2063SBarry Smith if (tmp[j] > *norm) *norm = tmp[j]; 114117ab2063SBarry Smith } 11420452661fSBarry Smith PetscFree(tmp); 114317ab2063SBarry Smith } 114417ab2063SBarry Smith else if (type == NORM_INFINITY) { 114517ab2063SBarry Smith *norm = 0.0; 1146416022c9SBarry Smith for ( j=0; j<a->m; j++ ) { 1147416022c9SBarry Smith v = a->a + a->i[j] + shift; 114817ab2063SBarry Smith sum = 0.0; 1149416022c9SBarry Smith for ( i=0; i<a->i[j+1]-a->i[j]; i++ ) { 1150cddf8d76SBarry Smith sum += PetscAbsScalar(*v); v++; 115117ab2063SBarry Smith } 115217ab2063SBarry Smith if (sum > *norm) *norm = sum; 115317ab2063SBarry Smith } 115417ab2063SBarry Smith } 115517ab2063SBarry Smith else { 1156e3372554SBarry Smith SETERRQ(1,0,"No support for two norm yet"); 115717ab2063SBarry Smith } 115817ab2063SBarry Smith return 0; 115917ab2063SBarry Smith } 116017ab2063SBarry Smith 11615615d1e5SSatish Balay #undef __FUNC__ 11625615d1e5SSatish Balay #define __FUNC__ "MatTranspose_SeqAIJ" 11638f6be9afSLois Curfman McInnes int MatTranspose_SeqAIJ(Mat A,Mat *B) 116417ab2063SBarry Smith { 1165416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 1166416022c9SBarry Smith Mat C; 1167416022c9SBarry Smith int i, ierr, *aj = a->j, *ai = a->i, m = a->m, len, *col; 1168416022c9SBarry Smith int shift = a->indexshift; 1169d5d45c9bSBarry Smith Scalar *array = a->a; 117017ab2063SBarry Smith 11713638b69dSLois Curfman McInnes if (B == PETSC_NULL && m != a->n) 1172e3372554SBarry Smith SETERRQ(1,0,"Square matrix only for in-place"); 11730452661fSBarry Smith col = (int *) PetscMalloc((1+a->n)*sizeof(int)); CHKPTRQ(col); 1174cddf8d76SBarry Smith PetscMemzero(col,(1+a->n)*sizeof(int)); 117517ab2063SBarry Smith if (shift) { 117617ab2063SBarry Smith for ( i=0; i<ai[m]-1; i++ ) aj[i] -= 1; 117717ab2063SBarry Smith } 117817ab2063SBarry Smith for ( i=0; i<ai[m]+shift; i++ ) col[aj[i]] += 1; 1179416022c9SBarry Smith ierr = MatCreateSeqAIJ(A->comm,a->n,m,0,col,&C); CHKERRQ(ierr); 11800452661fSBarry Smith PetscFree(col); 118117ab2063SBarry Smith for ( i=0; i<m; i++ ) { 118217ab2063SBarry Smith len = ai[i+1]-ai[i]; 1183416022c9SBarry Smith ierr = MatSetValues(C,len,aj,1,&i,array,INSERT_VALUES); CHKERRQ(ierr); 118417ab2063SBarry Smith array += len; aj += len; 118517ab2063SBarry Smith } 118617ab2063SBarry Smith if (shift) { 118717ab2063SBarry Smith for ( i=0; i<ai[m]-1; i++ ) aj[i] += 1; 118817ab2063SBarry Smith } 118917ab2063SBarry Smith 11906d4a8577SBarry Smith ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr); 11916d4a8577SBarry Smith ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr); 119217ab2063SBarry Smith 11933638b69dSLois Curfman McInnes if (B != PETSC_NULL) { 1194416022c9SBarry Smith *B = C; 119517ab2063SBarry Smith } else { 1196416022c9SBarry Smith /* This isn't really an in-place transpose */ 11970452661fSBarry Smith PetscFree(a->a); 11980452661fSBarry Smith if (!a->singlemalloc) {PetscFree(a->i); PetscFree(a->j);} 11990452661fSBarry Smith if (a->diag) PetscFree(a->diag); 12000452661fSBarry Smith if (a->ilen) PetscFree(a->ilen); 12010452661fSBarry Smith if (a->imax) PetscFree(a->imax); 12020452661fSBarry Smith if (a->solve_work) PetscFree(a->solve_work); 12031073c447SSatish Balay if (a->inode.size) PetscFree(a->inode.size); 12040452661fSBarry Smith PetscFree(a); 1205*f09e8eb9SSatish Balay PetscMemcpy(A,C,sizeof(struct _p_Mat)); 12060452661fSBarry Smith PetscHeaderDestroy(C); 120717ab2063SBarry Smith } 120817ab2063SBarry Smith return 0; 120917ab2063SBarry Smith } 121017ab2063SBarry Smith 12115615d1e5SSatish Balay #undef __FUNC__ 12125615d1e5SSatish Balay #define __FUNC__ "MatDiagonalScale_SeqAIJ" 12138f6be9afSLois Curfman McInnes int MatDiagonalScale_SeqAIJ(Mat A,Vec ll,Vec rr) 121417ab2063SBarry Smith { 1215416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 121617ab2063SBarry Smith Scalar *l,*r,x,*v; 1217d5d45c9bSBarry Smith int i,j,m = a->m, n = a->n, M, nz = a->nz, *jj,shift = a->indexshift; 121817ab2063SBarry Smith 121917ab2063SBarry Smith if (ll) { 12203ea7c6a1SSatish Balay /* The local size is used so that VecMPI can be passed to this routine 12213ea7c6a1SSatish Balay by MatDiagonalScale_MPIAIJ */ 12229b1297e1SSatish Balay VecGetLocalSize_Fast(ll,m); 1223e3372554SBarry Smith if (m != a->m) SETERRQ(1,0,"Left scaling vector wrong length"); 122490f02eecSBarry Smith VecGetArray_Fast(ll,l); 1225416022c9SBarry Smith v = a->a; 122617ab2063SBarry Smith for ( i=0; i<m; i++ ) { 122717ab2063SBarry Smith x = l[i]; 1228416022c9SBarry Smith M = a->i[i+1] - a->i[i]; 122917ab2063SBarry Smith for ( j=0; j<M; j++ ) { (*v++) *= x;} 123017ab2063SBarry Smith } 123144cd7ae7SLois Curfman McInnes PLogFlops(nz); 123217ab2063SBarry Smith } 123317ab2063SBarry Smith if (rr) { 12349b1297e1SSatish Balay VecGetLocalSize_Fast(rr,n); 1235e3372554SBarry Smith if (n != a->n) SETERRQ(1,0,"Right scaling vector wrong length"); 123690f02eecSBarry Smith VecGetArray_Fast(rr,r); 1237416022c9SBarry Smith v = a->a; jj = a->j; 123817ab2063SBarry Smith for ( i=0; i<nz; i++ ) { 123917ab2063SBarry Smith (*v++) *= r[*jj++ + shift]; 124017ab2063SBarry Smith } 124144cd7ae7SLois Curfman McInnes PLogFlops(nz); 124217ab2063SBarry Smith } 124317ab2063SBarry Smith return 0; 124417ab2063SBarry Smith } 124517ab2063SBarry Smith 12465615d1e5SSatish Balay #undef __FUNC__ 12475615d1e5SSatish Balay #define __FUNC__ "MatGetSubMatrix_SeqAIJ" 12488f6be9afSLois Curfman McInnes int MatGetSubMatrix_SeqAIJ(Mat A,IS isrow,IS iscol,MatGetSubMatrixCall scall,Mat *B) 124917ab2063SBarry Smith { 1250db02288aSLois Curfman McInnes Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data,*c; 125102834360SBarry Smith int nznew, *smap, i, k, kstart, kend, ierr, oldcols = a->n,*lens; 125299141d43SSatish Balay int row,mat_i,*mat_j,tcol,first,step,*mat_ilen; 1253a2744918SBarry Smith register int sum,lensi; 125499141d43SSatish Balay int *irow, *icol, nrows, ncols, shift = a->indexshift,*ssmap; 125599141d43SSatish Balay int *starts,*j_new,*i_new,*aj = a->j, *ai = a->i,ii,*ailen = a->ilen; 125699141d43SSatish Balay Scalar *a_new,*mat_a; 1257416022c9SBarry Smith Mat C; 125817ab2063SBarry Smith 1259b48a1e75SSatish Balay ierr = ISSorted(isrow,(PetscTruth*)&i); 1260e3372554SBarry Smith if (!i) SETERRQ(1,0,"ISrow is not sorted"); 126199141d43SSatish Balay ierr = ISSorted(iscol,(PetscTruth*)&i); 1262e3372554SBarry Smith if (!i) SETERRQ(1,0,"IScol is not sorted"); 126399141d43SSatish Balay 126417ab2063SBarry Smith ierr = ISGetIndices(isrow,&irow); CHKERRQ(ierr); 126517ab2063SBarry Smith ierr = ISGetSize(isrow,&nrows); CHKERRQ(ierr); 126617ab2063SBarry Smith ierr = ISGetSize(iscol,&ncols); CHKERRQ(ierr); 126717ab2063SBarry Smith 12687264ac53SSatish Balay if (ISStrideGetInfo(iscol,&first,&step) && step == 1) { /* no need to sort */ 126902834360SBarry Smith /* special case of contiguous rows */ 127057faeb66SBarry Smith lens = (int *) PetscMalloc((ncols+nrows+1)*sizeof(int)); CHKPTRQ(lens); 127102834360SBarry Smith starts = lens + ncols; 127202834360SBarry Smith /* loop over new rows determining lens and starting points */ 127302834360SBarry Smith for (i=0; i<nrows; i++) { 1274a2744918SBarry Smith kstart = ai[irow[i]]+shift; 1275a2744918SBarry Smith kend = kstart + ailen[irow[i]]; 127602834360SBarry Smith for ( k=kstart; k<kend; k++ ) { 1277d8ced48eSBarry Smith if (aj[k]+shift >= first) { 127802834360SBarry Smith starts[i] = k; 127902834360SBarry Smith break; 128002834360SBarry Smith } 128102834360SBarry Smith } 1282a2744918SBarry Smith sum = 0; 128302834360SBarry Smith while (k < kend) { 1284d8ced48eSBarry Smith if (aj[k++]+shift >= first+ncols) break; 1285a2744918SBarry Smith sum++; 128602834360SBarry Smith } 1287a2744918SBarry Smith lens[i] = sum; 128802834360SBarry Smith } 128902834360SBarry Smith /* create submatrix */ 1290cddf8d76SBarry Smith if (scall == MAT_REUSE_MATRIX) { 129108480c60SBarry Smith int n_cols,n_rows; 129208480c60SBarry Smith ierr = MatGetSize(*B,&n_rows,&n_cols); CHKERRQ(ierr); 1293e3372554SBarry Smith if (n_rows != nrows || n_cols != ncols) SETERRQ(1,0,""); 1294d8ced48eSBarry Smith ierr = MatZeroEntries(*B); CHKERRQ(ierr); 129508480c60SBarry Smith C = *B; 129608480c60SBarry Smith } 129708480c60SBarry Smith else { 129802834360SBarry Smith ierr = MatCreateSeqAIJ(A->comm,nrows,ncols,0,lens,&C);CHKERRQ(ierr); 129908480c60SBarry Smith } 1300db02288aSLois Curfman McInnes c = (Mat_SeqAIJ*) C->data; 1301db02288aSLois Curfman McInnes 130202834360SBarry Smith /* loop over rows inserting into submatrix */ 1303db02288aSLois Curfman McInnes a_new = c->a; 1304db02288aSLois Curfman McInnes j_new = c->j; 1305db02288aSLois Curfman McInnes i_new = c->i; 1306db02288aSLois Curfman McInnes i_new[0] = -shift; 130702834360SBarry Smith for (i=0; i<nrows; i++) { 1308a2744918SBarry Smith ii = starts[i]; 1309a2744918SBarry Smith lensi = lens[i]; 1310a2744918SBarry Smith for ( k=0; k<lensi; k++ ) { 1311a2744918SBarry Smith *j_new++ = aj[ii+k] - first; 131202834360SBarry Smith } 1313a2744918SBarry Smith PetscMemcpy(a_new,a->a + starts[i],lensi*sizeof(Scalar)); 1314a2744918SBarry Smith a_new += lensi; 1315a2744918SBarry Smith i_new[i+1] = i_new[i] + lensi; 1316a2744918SBarry Smith c->ilen[i] = lensi; 131702834360SBarry Smith } 13180452661fSBarry Smith PetscFree(lens); 131902834360SBarry Smith } 132002834360SBarry Smith else { 132102834360SBarry Smith ierr = ISGetIndices(iscol,&icol); CHKERRQ(ierr); 13220452661fSBarry Smith smap = (int *) PetscMalloc((1+oldcols)*sizeof(int)); CHKPTRQ(smap); 132302834360SBarry Smith ssmap = smap + shift; 132499141d43SSatish Balay lens = (int *) PetscMalloc((1+nrows)*sizeof(int)); CHKPTRQ(lens); 1325cddf8d76SBarry Smith PetscMemzero(smap,oldcols*sizeof(int)); 132617ab2063SBarry Smith for ( i=0; i<ncols; i++ ) smap[icol[i]] = i+1; 132702834360SBarry Smith /* determine lens of each row */ 132802834360SBarry Smith for (i=0; i<nrows; i++) { 1329d8ced48eSBarry Smith kstart = ai[irow[i]]+shift; 133002834360SBarry Smith kend = kstart + a->ilen[irow[i]]; 133102834360SBarry Smith lens[i] = 0; 133202834360SBarry Smith for ( k=kstart; k<kend; k++ ) { 1333d8ced48eSBarry Smith if (ssmap[aj[k]]) { 133402834360SBarry Smith lens[i]++; 133502834360SBarry Smith } 133602834360SBarry Smith } 133702834360SBarry Smith } 133817ab2063SBarry Smith /* Create and fill new matrix */ 1339a2744918SBarry Smith if (scall == MAT_REUSE_MATRIX) { 134099141d43SSatish Balay c = (Mat_SeqAIJ *)((*B)->data); 134199141d43SSatish Balay 1342e3372554SBarry Smith if (c->m != nrows || c->n != ncols) SETERRQ(1,0,""); 134399141d43SSatish Balay if (PetscMemcmp(c->ilen,lens, c->m *sizeof(int))) { 1344e3372554SBarry Smith SETERRQ(1,0,"Cannot reuse matrix. wrong no of nonzeros"); 134599141d43SSatish Balay } 134699141d43SSatish Balay PetscMemzero(c->ilen,c->m*sizeof(int)); 134708480c60SBarry Smith C = *B; 134808480c60SBarry Smith } 134908480c60SBarry Smith else { 135002834360SBarry Smith ierr = MatCreateSeqAIJ(A->comm,nrows,ncols,0,lens,&C);CHKERRQ(ierr); 135108480c60SBarry Smith } 135299141d43SSatish Balay c = (Mat_SeqAIJ *)(C->data); 135317ab2063SBarry Smith for (i=0; i<nrows; i++) { 135499141d43SSatish Balay row = irow[i]; 135517ab2063SBarry Smith nznew = 0; 135699141d43SSatish Balay kstart = ai[row]+shift; 135799141d43SSatish Balay kend = kstart + a->ilen[row]; 135899141d43SSatish Balay mat_i = c->i[i]+shift; 135999141d43SSatish Balay mat_j = c->j + mat_i; 136099141d43SSatish Balay mat_a = c->a + mat_i; 136199141d43SSatish Balay mat_ilen = c->ilen + i; 136217ab2063SBarry Smith for ( k=kstart; k<kend; k++ ) { 136399141d43SSatish Balay if ((tcol=ssmap[a->j[k]])) { 136499141d43SSatish Balay *mat_j++ = tcol - (!shift); 136599141d43SSatish Balay *mat_a++ = a->a[k]; 136699141d43SSatish Balay (*mat_ilen)++; 136799141d43SSatish Balay 136817ab2063SBarry Smith } 136917ab2063SBarry Smith } 137017ab2063SBarry Smith } 137102834360SBarry Smith /* Free work space */ 137202834360SBarry Smith ierr = ISRestoreIndices(iscol,&icol); CHKERRQ(ierr); 137399141d43SSatish Balay PetscFree(smap); PetscFree(lens); 137402834360SBarry Smith } 13756d4a8577SBarry Smith ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr); 13766d4a8577SBarry Smith ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr); 137717ab2063SBarry Smith 137817ab2063SBarry Smith ierr = ISRestoreIndices(isrow,&irow); CHKERRQ(ierr); 1379416022c9SBarry Smith *B = C; 138017ab2063SBarry Smith return 0; 138117ab2063SBarry Smith } 138217ab2063SBarry Smith 1383a871dcd8SBarry Smith /* 138463b91edcSBarry Smith note: This can only work for identity for row and col. It would 138563b91edcSBarry Smith be good to check this and otherwise generate an error. 1386a871dcd8SBarry Smith */ 13875615d1e5SSatish Balay #undef __FUNC__ 13885615d1e5SSatish Balay #define __FUNC__ "MatILUFactor_SeqAIJ" 13898f6be9afSLois Curfman McInnes int MatILUFactor_SeqAIJ(Mat inA,IS row,IS col,double efill,int fill) 1390a871dcd8SBarry Smith { 139163b91edcSBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) inA->data; 139208480c60SBarry Smith int ierr; 139363b91edcSBarry Smith Mat outA; 139463b91edcSBarry Smith 1395e3372554SBarry Smith if (fill != 0) SETERRQ(1,0,"Only fill=0 supported"); 1396a871dcd8SBarry Smith 139763b91edcSBarry Smith outA = inA; 139863b91edcSBarry Smith inA->factor = FACTOR_LU; 139963b91edcSBarry Smith a->row = row; 140063b91edcSBarry Smith a->col = col; 140163b91edcSBarry Smith 140294a9d846SBarry Smith if (!a->solve_work) { /* this matrix may have been factored before */ 14030452661fSBarry Smith a->solve_work = (Scalar *) PetscMalloc( (a->m+1)*sizeof(Scalar)); CHKPTRQ(a->solve_work); 140494a9d846SBarry Smith } 140563b91edcSBarry Smith 140608480c60SBarry Smith if (!a->diag) { 140708480c60SBarry Smith ierr = MatMarkDiag_SeqAIJ(inA); CHKERRQ(ierr); 140863b91edcSBarry Smith } 140963b91edcSBarry Smith ierr = MatLUFactorNumeric_SeqAIJ(inA,&outA); CHKERRQ(ierr); 1410a871dcd8SBarry Smith return 0; 1411a871dcd8SBarry Smith } 1412a871dcd8SBarry Smith 1413f0b747eeSBarry Smith #include "pinclude/plapack.h" 14145615d1e5SSatish Balay #undef __FUNC__ 14155615d1e5SSatish Balay #define __FUNC__ "MatScale_SeqAIJ" 14168f6be9afSLois Curfman McInnes int MatScale_SeqAIJ(Scalar *alpha,Mat inA) 1417f0b747eeSBarry Smith { 1418f0b747eeSBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) inA->data; 1419f0b747eeSBarry Smith int one = 1; 1420f0b747eeSBarry Smith BLscal_( &a->nz, alpha, a->a, &one ); 1421f0b747eeSBarry Smith PLogFlops(a->nz); 1422f0b747eeSBarry Smith return 0; 1423f0b747eeSBarry Smith } 1424f0b747eeSBarry Smith 14255615d1e5SSatish Balay #undef __FUNC__ 14265615d1e5SSatish Balay #define __FUNC__ "MatGetSubMatrices_SeqAIJ" 14278f6be9afSLois Curfman McInnes int MatGetSubMatrices_SeqAIJ(Mat A,int n, IS *irow,IS *icol,MatGetSubMatrixCall scall, 1428cddf8d76SBarry Smith Mat **B) 1429cddf8d76SBarry Smith { 1430cddf8d76SBarry Smith int ierr,i; 1431cddf8d76SBarry Smith 1432cddf8d76SBarry Smith if (scall == MAT_INITIAL_MATRIX) { 14330452661fSBarry Smith *B = (Mat *) PetscMalloc( (n+1)*sizeof(Mat) ); CHKPTRQ(*B); 1434cddf8d76SBarry Smith } 1435cddf8d76SBarry Smith 1436cddf8d76SBarry Smith for ( i=0; i<n; i++ ) { 1437905e6a2fSBarry Smith ierr = MatGetSubMatrix_SeqAIJ(A,irow[i],icol[i],scall,&(*B)[i]);CHKERRQ(ierr); 1438cddf8d76SBarry Smith } 1439cddf8d76SBarry Smith return 0; 1440cddf8d76SBarry Smith } 1441cddf8d76SBarry Smith 14425615d1e5SSatish Balay #undef __FUNC__ 14435615d1e5SSatish Balay #define __FUNC__ "MatGetBlockSize_SeqAIJ" 14448f6be9afSLois Curfman McInnes int MatGetBlockSize_SeqAIJ(Mat A, int *bs) 14455a838052SSatish Balay { 14465a838052SSatish Balay *bs = 1; 14475a838052SSatish Balay return 0; 14485a838052SSatish Balay } 14495a838052SSatish Balay 14505615d1e5SSatish Balay #undef __FUNC__ 14515615d1e5SSatish Balay #define __FUNC__ "MatIncreaseOverlap_SeqAIJ" 14528f6be9afSLois Curfman McInnes int MatIncreaseOverlap_SeqAIJ(Mat A, int is_max, IS *is, int ov) 14534dcbc457SBarry Smith { 1454e4d965acSSatish Balay Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 145506763907SSatish Balay int shift, row, i,j,k,l,m,n, *idx,ierr, *nidx, isz, val; 14568a047759SSatish Balay int start, end, *ai, *aj; 145706763907SSatish Balay char *table; 14588a047759SSatish Balay shift = a->indexshift; 1459e4d965acSSatish Balay m = a->m; 1460e4d965acSSatish Balay ai = a->i; 14618a047759SSatish Balay aj = a->j+shift; 14628a047759SSatish Balay 1463e3372554SBarry Smith if (ov < 0) SETERRQ(1,0,"illegal overlap value used"); 146406763907SSatish Balay 146506763907SSatish Balay table = (char *) PetscMalloc((m/BITSPERBYTE +1)*sizeof(char)); CHKPTRQ(table); 146606763907SSatish Balay nidx = (int *) PetscMalloc((m+1)*sizeof(int)); CHKPTRQ(nidx); 146706763907SSatish Balay 1468e4d965acSSatish Balay for ( i=0; i<is_max; i++ ) { 1469b97fc60eSLois Curfman McInnes /* Initialize the two local arrays */ 1470e4d965acSSatish Balay isz = 0; 147106763907SSatish Balay PetscMemzero(table,(m/BITSPERBYTE +1)*sizeof(char)); 1472e4d965acSSatish Balay 1473e4d965acSSatish Balay /* Extract the indices, assume there can be duplicate entries */ 14744dcbc457SBarry Smith ierr = ISGetIndices(is[i],&idx); CHKERRQ(ierr); 147577c4ece6SBarry Smith ierr = ISGetSize(is[i],&n); CHKERRQ(ierr); 1476e4d965acSSatish Balay 1477dd097bc3SLois Curfman McInnes /* Enter these into the temp arrays. I.e., mark table[row], enter row into new index */ 1478e4d965acSSatish Balay for ( j=0; j<n ; ++j){ 147906763907SSatish Balay if(!BT_LOOKUP(table, idx[j])) { nidx[isz++] = idx[j];} 14804dcbc457SBarry Smith } 148106763907SSatish Balay ierr = ISRestoreIndices(is[i],&idx); CHKERRQ(ierr); 148206763907SSatish Balay ierr = ISDestroy(is[i]); CHKERRQ(ierr); 1483e4d965acSSatish Balay 148404a348a9SBarry Smith k = 0; 148504a348a9SBarry Smith for ( j=0; j<ov; j++){ /* for each overlap */ 148604a348a9SBarry Smith n = isz; 148706763907SSatish Balay for ( ; k<n ; k++){ /* do only those rows in nidx[k], which are not done yet */ 1488e4d965acSSatish Balay row = nidx[k]; 1489e4d965acSSatish Balay start = ai[row]; 1490e4d965acSSatish Balay end = ai[row+1]; 149104a348a9SBarry Smith for ( l = start; l<end ; l++){ 14928a047759SSatish Balay val = aj[l] + shift; 149306763907SSatish Balay if (!BT_LOOKUP(table,val)) {nidx[isz++] = val;} 1494e4d965acSSatish Balay } 1495e4d965acSSatish Balay } 1496e4d965acSSatish Balay } 1497029af93fSBarry Smith ierr = ISCreateGeneral(PETSC_COMM_SELF, isz, nidx, (is+i)); CHKERRQ(ierr); 1498e4d965acSSatish Balay } 149904a348a9SBarry Smith PetscFree(table); 150006763907SSatish Balay PetscFree(nidx); 1501e4d965acSSatish Balay return 0; 15024dcbc457SBarry Smith } 150317ab2063SBarry Smith 15040513a670SBarry Smith /* -------------------------------------------------------------- */ 15050513a670SBarry Smith #undef __FUNC__ 15060513a670SBarry Smith #define __FUNC__ "MatPermute_SeqAIJ" 15070513a670SBarry Smith int MatPermute_SeqAIJ(Mat A, IS rowp, IS colp, Mat *B) 15080513a670SBarry Smith { 15090513a670SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 15100513a670SBarry Smith Scalar *vwork; 15110513a670SBarry Smith int i, ierr, nz, m = a->m, n = a->n, *cwork; 15120513a670SBarry Smith int *row,*col,*cnew,j,*lens; 15130513a670SBarry Smith 15140513a670SBarry Smith ierr = ISGetIndices(rowp,&row); CHKERRQ(ierr); 15150513a670SBarry Smith ierr = ISGetIndices(colp,&col); CHKERRQ(ierr); 15160513a670SBarry Smith 15170513a670SBarry Smith /* determine lengths of permuted rows */ 15180513a670SBarry Smith lens = (int *) PetscMalloc( (m+1)*sizeof(int) ); CHKPTRQ(lens); 15190513a670SBarry Smith for (i=0; i<m; i++ ) { 15200513a670SBarry Smith lens[row[i]] = a->i[i+1] - a->i[i]; 15210513a670SBarry Smith } 15220513a670SBarry Smith ierr = MatCreateSeqAIJ(A->comm,m,n,0,lens,B);CHKERRQ(ierr); 15230513a670SBarry Smith PetscFree(lens); 15240513a670SBarry Smith 15250513a670SBarry Smith cnew = (int *) PetscMalloc( n*sizeof(int) ); CHKPTRQ(cnew); 15260513a670SBarry Smith for (i=0; i<m; i++) { 15270513a670SBarry Smith ierr = MatGetRow(A,i,&nz,&cwork,&vwork); CHKERRQ(ierr); 15280513a670SBarry Smith for (j=0; j<nz; j++ ) { cnew[j] = col[cwork[j]];} 15290513a670SBarry Smith ierr = MatSetValues(*B,1,&row[i],nz,cnew,vwork,INSERT_VALUES); CHKERRQ(ierr); 15300513a670SBarry Smith ierr = MatRestoreRow(A,i,&nz,&cwork,&vwork); CHKERRQ(ierr); 15310513a670SBarry Smith } 15320513a670SBarry Smith PetscFree(cnew); 15330513a670SBarry Smith ierr = MatAssemblyBegin(*B,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr); 15340513a670SBarry Smith ierr = MatAssemblyEnd(*B,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr); 15350513a670SBarry Smith ierr = ISRestoreIndices(rowp,&row); CHKERRQ(ierr); 15360513a670SBarry Smith ierr = ISRestoreIndices(colp,&col); CHKERRQ(ierr); 15370513a670SBarry Smith return 0; 15380513a670SBarry Smith } 15390513a670SBarry Smith 15405615d1e5SSatish Balay #undef __FUNC__ 15415615d1e5SSatish Balay #define __FUNC__ "MatPrintHelp_SeqAIJ" 1542682d7d0cSBarry Smith int MatPrintHelp_SeqAIJ(Mat A) 1543682d7d0cSBarry Smith { 1544682d7d0cSBarry Smith static int called = 0; 1545682d7d0cSBarry Smith MPI_Comm comm = A->comm; 1546682d7d0cSBarry Smith 1547682d7d0cSBarry Smith if (called) return 0; else called = 1; 154877c4ece6SBarry Smith PetscPrintf(comm," Options for MATSEQAIJ and MATMPIAIJ matrix formats (the defaults):\n"); 15490f665d81SLois Curfman McInnes PetscPrintf(comm," -mat_lu_pivotthreshold <threshold>: Set pivoting threshold\n"); 15500f665d81SLois Curfman McInnes PetscPrintf(comm," -mat_aij_oneindex: internal indices begin at 1 instead of the default 0.\n"); 15510f665d81SLois Curfman McInnes PetscPrintf(comm," -mat_aij_no_inode: Do not use inodes\n"); 15520f665d81SLois Curfman McInnes PetscPrintf(comm," -mat_aij_inode_limit <limit>: Set inode limit (max limit=5)\n"); 1553682d7d0cSBarry Smith #if defined(HAVE_ESSL) 15540f665d81SLois Curfman McInnes PetscPrintf(comm," -mat_aij_essl: Use IBM sparse LU factorization and solve.\n"); 1555682d7d0cSBarry Smith #endif 1556682d7d0cSBarry Smith return 0; 1557682d7d0cSBarry Smith } 15588f6be9afSLois Curfman McInnes extern int MatEqual_SeqAIJ(Mat A,Mat B, PetscTruth* flg); 1559a93ec695SBarry Smith extern int MatFDColoringCreate_SeqAIJ(Mat,ISColoring,MatFDColoring); 1560a93ec695SBarry Smith extern int MatColoringPatch_SeqAIJ(Mat,int,int *,ISColoring *); 1561a93ec695SBarry Smith 1562682d7d0cSBarry Smith /* -------------------------------------------------------------------*/ 156317ab2063SBarry Smith static struct _MatOps MatOps = {MatSetValues_SeqAIJ, 156417ab2063SBarry Smith MatGetRow_SeqAIJ,MatRestoreRow_SeqAIJ, 1565416022c9SBarry Smith MatMult_SeqAIJ,MatMultAdd_SeqAIJ, 1566416022c9SBarry Smith MatMultTrans_SeqAIJ,MatMultTransAdd_SeqAIJ, 156717ab2063SBarry Smith MatSolve_SeqAIJ,MatSolveAdd_SeqAIJ, 156817ab2063SBarry Smith MatSolveTrans_SeqAIJ,MatSolveTransAdd_SeqAIJ, 156917ab2063SBarry Smith MatLUFactor_SeqAIJ,0, 157017ab2063SBarry Smith MatRelax_SeqAIJ, 157117ab2063SBarry Smith MatTranspose_SeqAIJ, 15727264ac53SSatish Balay MatGetInfo_SeqAIJ,MatEqual_SeqAIJ, 1573f0b747eeSBarry Smith MatGetDiagonal_SeqAIJ,MatDiagonalScale_SeqAIJ,MatNorm_SeqAIJ, 157417ab2063SBarry Smith 0,MatAssemblyEnd_SeqAIJ, 157517ab2063SBarry Smith MatCompress_SeqAIJ, 157617ab2063SBarry Smith MatSetOption_SeqAIJ,MatZeroEntries_SeqAIJ,MatZeroRows_SeqAIJ, 157717ab2063SBarry Smith MatLUFactorSymbolic_SeqAIJ,MatLUFactorNumeric_SeqAIJ,0,0, 157817ab2063SBarry Smith MatGetSize_SeqAIJ,MatGetSize_SeqAIJ,MatGetOwnershipRange_SeqAIJ, 157917ab2063SBarry Smith MatILUFactorSymbolic_SeqAIJ,0, 158094a9d846SBarry Smith 0,0, 15813d1612f7SBarry Smith MatConvertSameType_SeqAIJ,0,0, 1582cddf8d76SBarry Smith MatILUFactor_SeqAIJ,0,0, 15837eb43aa7SLois Curfman McInnes MatGetSubMatrices_SeqAIJ,MatIncreaseOverlap_SeqAIJ, 1584682d7d0cSBarry Smith MatGetValues_SeqAIJ,0, 1585f0b747eeSBarry Smith MatPrintHelp_SeqAIJ, 15865a838052SSatish Balay MatScale_SeqAIJ,0,0, 15876945ee14SBarry Smith MatILUDTFactor_SeqAIJ, 15886945ee14SBarry Smith MatGetBlockSize_SeqAIJ, 15893b2fbd54SBarry Smith MatGetRowIJ_SeqAIJ, 15903b2fbd54SBarry Smith MatRestoreRowIJ_SeqAIJ, 15913b2fbd54SBarry Smith MatGetColumnIJ_SeqAIJ, 1592a93ec695SBarry Smith MatRestoreColumnIJ_SeqAIJ, 1593a93ec695SBarry Smith MatFDColoringCreate_SeqAIJ, 15940513a670SBarry Smith MatColoringPatch_SeqAIJ, 15950513a670SBarry Smith 0, 15960513a670SBarry Smith MatPermute_SeqAIJ}; 159717ab2063SBarry Smith 159817ab2063SBarry Smith extern int MatUseSuperLU_SeqAIJ(Mat); 159917ab2063SBarry Smith extern int MatUseEssl_SeqAIJ(Mat); 160017ab2063SBarry Smith extern int MatUseDXML_SeqAIJ(Mat); 160117ab2063SBarry Smith 16025615d1e5SSatish Balay #undef __FUNC__ 16035615d1e5SSatish Balay #define __FUNC__ "MatCreateSeqAIJ" 160417ab2063SBarry Smith /*@C 1605682d7d0cSBarry Smith MatCreateSeqAIJ - Creates a sparse matrix in AIJ (compressed row) format 16060d15e28bSLois Curfman McInnes (the default parallel PETSc format). For good matrix assembly performance 16076e62573dSLois Curfman McInnes the user should preallocate the matrix storage by setting the parameter nz 16082bd5e0b2SLois Curfman McInnes (or the array nzz). By setting these parameters accurately, performance 16092bd5e0b2SLois Curfman McInnes during matrix assembly can be increased by more than a factor of 50. 161017ab2063SBarry Smith 161117ab2063SBarry Smith Input Parameters: 1612029af93fSBarry Smith . comm - MPI communicator, set to PETSC_COMM_SELF 161317ab2063SBarry Smith . m - number of rows 161417ab2063SBarry Smith . n - number of columns 161517ab2063SBarry Smith . nz - number of nonzeros per row (same for all rows) 16162bd5e0b2SLois Curfman McInnes . nzz - array containing the number of nonzeros in the various rows 16172bd5e0b2SLois Curfman McInnes (possibly different for each row) or PETSC_NULL 161817ab2063SBarry Smith 161917ab2063SBarry Smith Output Parameter: 1620416022c9SBarry Smith . A - the matrix 162117ab2063SBarry Smith 162217ab2063SBarry Smith Notes: 162317ab2063SBarry Smith The AIJ format (also called the Yale sparse matrix format or 162417ab2063SBarry Smith compressed row storage), is fully compatible with standard Fortran 77 16250002213bSLois Curfman McInnes storage. That is, the stored row and column indices can begin at 162644cd7ae7SLois Curfman McInnes either one (as in Fortran) or zero. See the users' manual for details. 162717ab2063SBarry Smith 162817ab2063SBarry Smith Specify the preallocated storage with either nz or nnz (not both). 1629a40aa06bSLois Curfman McInnes Set nz=PETSC_DEFAULT and nnz=PETSC_NULL for PETSc to control dynamic memory 16303d323bbdSBarry Smith allocation. For large problems you MUST preallocate memory or you 16316da5968aSLois Curfman McInnes will get TERRIBLE performance, see the users' manual chapter on matrices. 163217ab2063SBarry Smith 1633682d7d0cSBarry Smith By default, this format uses inodes (identical nodes) when possible, to 1634682d7d0cSBarry Smith improve numerical efficiency of Matrix vector products and solves. We 1635682d7d0cSBarry Smith search for consecutive rows with the same nonzero structure, thereby 16366c7ebb05SLois Curfman McInnes reusing matrix information to achieve increased efficiency. 16376c7ebb05SLois Curfman McInnes 16386c7ebb05SLois Curfman McInnes Options Database Keys: 16396c7ebb05SLois Curfman McInnes $ -mat_aij_no_inode - Do not use inodes 16406e62573dSLois Curfman McInnes $ -mat_aij_inode_limit <limit> - Set inode limit. 16416e62573dSLois Curfman McInnes $ (max limit=5) 16426e62573dSLois Curfman McInnes $ -mat_aij_oneindex - Internally use indexing starting at 1 16436e62573dSLois Curfman McInnes $ rather than 0. Note: When calling MatSetValues(), 16446e62573dSLois Curfman McInnes $ the user still MUST index entries starting at 0! 164517ab2063SBarry Smith 164617ab2063SBarry Smith .seealso: MatCreate(), MatCreateMPIAIJ(), MatSetValues() 164717ab2063SBarry Smith @*/ 1648416022c9SBarry Smith int MatCreateSeqAIJ(MPI_Comm comm,int m,int n,int nz,int *nnz, Mat *A) 164917ab2063SBarry Smith { 1650416022c9SBarry Smith Mat B; 1651416022c9SBarry Smith Mat_SeqAIJ *b; 16526945ee14SBarry Smith int i, len, ierr, flg,size; 16536945ee14SBarry Smith 16546945ee14SBarry Smith MPI_Comm_size(comm,&size); 1655e3372554SBarry Smith if (size > 1) SETERRQ(1,0,"Comm must be of size 1"); 1656d5d45c9bSBarry Smith 1657416022c9SBarry Smith *A = 0; 1658*f09e8eb9SSatish Balay PetscHeaderCreate(B,_p_Mat,MAT_COOKIE,MATSEQAIJ,comm); 1659416022c9SBarry Smith PLogObjectCreate(B); 16600452661fSBarry Smith B->data = (void *) (b = PetscNew(Mat_SeqAIJ)); CHKPTRQ(b); 166144cd7ae7SLois Curfman McInnes PetscMemzero(b,sizeof(Mat_SeqAIJ)); 1662416022c9SBarry Smith PetscMemcpy(&B->ops,&MatOps,sizeof(struct _MatOps)); 1663416022c9SBarry Smith B->destroy = MatDestroy_SeqAIJ; 1664416022c9SBarry Smith B->view = MatView_SeqAIJ; 1665416022c9SBarry Smith B->factor = 0; 1666416022c9SBarry Smith B->lupivotthreshold = 1.0; 166790f02eecSBarry Smith B->mapping = 0; 16687a743949SBarry Smith ierr = OptionsGetDouble(PETSC_NULL,"-mat_lu_pivotthreshold",&B->lupivotthreshold, 166969957df2SSatish Balay &flg); CHKERRQ(ierr); 16707a743949SBarry Smith b->ilu_preserve_row_sums = PETSC_FALSE; 16717a743949SBarry Smith ierr = OptionsHasName(PETSC_NULL,"-pc_ilu_preserve_row_sums", 16727a743949SBarry Smith (int*) &b->ilu_preserve_row_sums); CHKERRQ(ierr); 1673416022c9SBarry Smith b->row = 0; 1674416022c9SBarry Smith b->col = 0; 1675416022c9SBarry Smith b->indexshift = 0; 1676b810aeb4SBarry Smith b->reallocs = 0; 167769957df2SSatish Balay ierr = OptionsHasName(PETSC_NULL,"-mat_aij_oneindex", &flg); CHKERRQ(ierr); 167869957df2SSatish Balay if (flg) b->indexshift = -1; 167917ab2063SBarry Smith 168044cd7ae7SLois Curfman McInnes b->m = m; B->m = m; B->M = m; 168144cd7ae7SLois Curfman McInnes b->n = n; B->n = n; B->N = n; 16820452661fSBarry Smith b->imax = (int *) PetscMalloc( (m+1)*sizeof(int) ); CHKPTRQ(b->imax); 1683b4fd4287SBarry Smith if (nnz == PETSC_NULL) { 16847b8455f0SLois Curfman McInnes if (nz == PETSC_DEFAULT) nz = 10; 16857b8455f0SLois Curfman McInnes else if (nz <= 0) nz = 1; 1686416022c9SBarry Smith for ( i=0; i<m; i++ ) b->imax[i] = nz; 168717ab2063SBarry Smith nz = nz*m; 168817ab2063SBarry Smith } 168917ab2063SBarry Smith else { 169017ab2063SBarry Smith nz = 0; 1691416022c9SBarry Smith for ( i=0; i<m; i++ ) {b->imax[i] = nnz[i]; nz += nnz[i];} 169217ab2063SBarry Smith } 169317ab2063SBarry Smith 169417ab2063SBarry Smith /* allocate the matrix space */ 1695416022c9SBarry Smith len = nz*(sizeof(int) + sizeof(Scalar)) + (b->m+1)*sizeof(int); 16960452661fSBarry Smith b->a = (Scalar *) PetscMalloc( len ); CHKPTRQ(b->a); 1697416022c9SBarry Smith b->j = (int *) (b->a + nz); 1698cddf8d76SBarry Smith PetscMemzero(b->j,nz*sizeof(int)); 1699416022c9SBarry Smith b->i = b->j + nz; 1700416022c9SBarry Smith b->singlemalloc = 1; 170117ab2063SBarry Smith 1702416022c9SBarry Smith b->i[0] = -b->indexshift; 170317ab2063SBarry Smith for (i=1; i<m+1; i++) { 1704416022c9SBarry Smith b->i[i] = b->i[i-1] + b->imax[i-1]; 170517ab2063SBarry Smith } 170617ab2063SBarry Smith 1707416022c9SBarry Smith /* b->ilen will count nonzeros in each row so far. */ 17080452661fSBarry Smith b->ilen = (int *) PetscMalloc((m+1)*sizeof(int)); 1709*f09e8eb9SSatish Balay PLogObjectMemory(B,len+2*(m+1)*sizeof(int)+sizeof(struct _p_Mat)+sizeof(Mat_SeqAIJ)); 1710416022c9SBarry Smith for ( i=0; i<b->m; i++ ) { b->ilen[i] = 0;} 171117ab2063SBarry Smith 1712416022c9SBarry Smith b->nz = 0; 1713416022c9SBarry Smith b->maxnz = nz; 1714416022c9SBarry Smith b->sorted = 0; 1715416022c9SBarry Smith b->roworiented = 1; 1716416022c9SBarry Smith b->nonew = 0; 1717416022c9SBarry Smith b->diag = 0; 1718416022c9SBarry Smith b->solve_work = 0; 171971bd300dSLois Curfman McInnes b->spptr = 0; 1720754ec7b1SSatish Balay b->inode.node_count = 0; 1721754ec7b1SSatish Balay b->inode.size = 0; 17226c7ebb05SLois Curfman McInnes b->inode.limit = 5; 17236c7ebb05SLois Curfman McInnes b->inode.max_limit = 5; 17244e220ebcSLois Curfman McInnes B->info.nz_unneeded = (double)b->maxnz; 172517ab2063SBarry Smith 1726416022c9SBarry Smith *A = B; 17274e220ebcSLois Curfman McInnes 17284b14c69eSBarry Smith /* SuperLU is not currently supported through PETSc */ 17294b14c69eSBarry Smith #if defined(HAVE_SUPERLU) 173069957df2SSatish Balay ierr = OptionsHasName(PETSC_NULL,"-mat_aij_superlu", &flg); CHKERRQ(ierr); 173169957df2SSatish Balay if (flg) { ierr = MatUseSuperLU_SeqAIJ(B); CHKERRQ(ierr); } 17324b14c69eSBarry Smith #endif 173369957df2SSatish Balay ierr = OptionsHasName(PETSC_NULL,"-mat_aij_essl", &flg); CHKERRQ(ierr); 173469957df2SSatish Balay if (flg) { ierr = MatUseEssl_SeqAIJ(B); CHKERRQ(ierr); } 173569957df2SSatish Balay ierr = OptionsHasName(PETSC_NULL,"-mat_aij_dxml", &flg); CHKERRQ(ierr); 173669957df2SSatish Balay if (flg) { 1737e3372554SBarry Smith if (!b->indexshift) SETERRQ(1,0,"need -mat_aij_oneindex with -mat_aij_dxml"); 1738416022c9SBarry Smith ierr = MatUseDXML_SeqAIJ(B); CHKERRQ(ierr); 173917ab2063SBarry Smith } 174069957df2SSatish Balay ierr = OptionsHasName(PETSC_NULL,"-help", &flg); CHKERRQ(ierr); 174169957df2SSatish Balay if (flg) {ierr = MatPrintHelp(B); CHKERRQ(ierr); } 174217ab2063SBarry Smith return 0; 174317ab2063SBarry Smith } 174417ab2063SBarry Smith 17455615d1e5SSatish Balay #undef __FUNC__ 17465615d1e5SSatish Balay #define __FUNC__ "MatConvertSameType_SeqAIJ" 17473d1612f7SBarry Smith int MatConvertSameType_SeqAIJ(Mat A,Mat *B,int cpvalues) 174817ab2063SBarry Smith { 1749416022c9SBarry Smith Mat C; 1750416022c9SBarry Smith Mat_SeqAIJ *c,*a = (Mat_SeqAIJ *) A->data; 175108480c60SBarry Smith int i,len, m = a->m,shift = a->indexshift; 175217ab2063SBarry Smith 17534043dd9cSLois Curfman McInnes *B = 0; 1754*f09e8eb9SSatish Balay PetscHeaderCreate(C,_p_Mat,MAT_COOKIE,MATSEQAIJ,A->comm); 1755416022c9SBarry Smith PLogObjectCreate(C); 17560452661fSBarry Smith C->data = (void *) (c = PetscNew(Mat_SeqAIJ)); CHKPTRQ(c); 175741c01911SSatish Balay PetscMemcpy(&C->ops,&A->ops,sizeof(struct _MatOps)); 1758416022c9SBarry Smith C->destroy = MatDestroy_SeqAIJ; 1759416022c9SBarry Smith C->view = MatView_SeqAIJ; 1760416022c9SBarry Smith C->factor = A->factor; 1761416022c9SBarry Smith c->row = 0; 1762416022c9SBarry Smith c->col = 0; 1763416022c9SBarry Smith c->indexshift = shift; 1764c456f294SBarry Smith C->assembled = PETSC_TRUE; 176517ab2063SBarry Smith 176644cd7ae7SLois Curfman McInnes c->m = C->m = a->m; 176744cd7ae7SLois Curfman McInnes c->n = C->n = a->n; 176844cd7ae7SLois Curfman McInnes C->M = a->m; 176944cd7ae7SLois Curfman McInnes C->N = a->n; 177017ab2063SBarry Smith 17710452661fSBarry Smith c->imax = (int *) PetscMalloc((m+1)*sizeof(int)); CHKPTRQ(c->imax); 17720452661fSBarry Smith c->ilen = (int *) PetscMalloc((m+1)*sizeof(int)); CHKPTRQ(c->ilen); 177317ab2063SBarry Smith for ( i=0; i<m; i++ ) { 1774416022c9SBarry Smith c->imax[i] = a->imax[i]; 1775416022c9SBarry Smith c->ilen[i] = a->ilen[i]; 177617ab2063SBarry Smith } 177717ab2063SBarry Smith 177817ab2063SBarry Smith /* allocate the matrix space */ 1779416022c9SBarry Smith c->singlemalloc = 1; 1780416022c9SBarry Smith len = (m+1)*sizeof(int)+(a->i[m])*(sizeof(Scalar)+sizeof(int)); 17810452661fSBarry Smith c->a = (Scalar *) PetscMalloc( len ); CHKPTRQ(c->a); 1782416022c9SBarry Smith c->j = (int *) (c->a + a->i[m] + shift); 1783416022c9SBarry Smith c->i = c->j + a->i[m] + shift; 1784416022c9SBarry Smith PetscMemcpy(c->i,a->i,(m+1)*sizeof(int)); 178517ab2063SBarry Smith if (m > 0) { 1786416022c9SBarry Smith PetscMemcpy(c->j,a->j,(a->i[m]+shift)*sizeof(int)); 178708480c60SBarry Smith if (cpvalues == COPY_VALUES) { 1788416022c9SBarry Smith PetscMemcpy(c->a,a->a,(a->i[m]+shift)*sizeof(Scalar)); 178917ab2063SBarry Smith } 179008480c60SBarry Smith } 179117ab2063SBarry Smith 1792*f09e8eb9SSatish Balay PLogObjectMemory(C,len+2*(m+1)*sizeof(int)+sizeof(struct _p_Mat)+sizeof(Mat_SeqAIJ)); 1793416022c9SBarry Smith c->sorted = a->sorted; 1794416022c9SBarry Smith c->roworiented = a->roworiented; 1795416022c9SBarry Smith c->nonew = a->nonew; 17967a743949SBarry Smith c->ilu_preserve_row_sums = a->ilu_preserve_row_sums; 179717ab2063SBarry Smith 1798416022c9SBarry Smith if (a->diag) { 17990452661fSBarry Smith c->diag = (int *) PetscMalloc( (m+1)*sizeof(int) ); CHKPTRQ(c->diag); 1800416022c9SBarry Smith PLogObjectMemory(C,(m+1)*sizeof(int)); 180117ab2063SBarry Smith for ( i=0; i<m; i++ ) { 1802416022c9SBarry Smith c->diag[i] = a->diag[i]; 180317ab2063SBarry Smith } 180417ab2063SBarry Smith } 1805416022c9SBarry Smith else c->diag = 0; 18066c7ebb05SLois Curfman McInnes c->inode.limit = a->inode.limit; 18076c7ebb05SLois Curfman McInnes c->inode.max_limit = a->inode.max_limit; 1808754ec7b1SSatish Balay if (a->inode.size){ 1809daed632aSSatish Balay c->inode.size = (int *) PetscMalloc( (m+1)*sizeof(int) ); CHKPTRQ(c->inode.size); 1810754ec7b1SSatish Balay c->inode.node_count = a->inode.node_count; 1811daed632aSSatish Balay PetscMemcpy( c->inode.size, a->inode.size, (m+1)*sizeof(int)); 1812754ec7b1SSatish Balay } else { 1813754ec7b1SSatish Balay c->inode.size = 0; 1814754ec7b1SSatish Balay c->inode.node_count = 0; 1815754ec7b1SSatish Balay } 1816416022c9SBarry Smith c->nz = a->nz; 1817416022c9SBarry Smith c->maxnz = a->maxnz; 1818416022c9SBarry Smith c->solve_work = 0; 181976dd722bSSatish Balay c->spptr = 0; /* Dangerous -I'm throwing away a->spptr */ 1820754ec7b1SSatish Balay 1821416022c9SBarry Smith *B = C; 182217ab2063SBarry Smith return 0; 182317ab2063SBarry Smith } 182417ab2063SBarry Smith 18255615d1e5SSatish Balay #undef __FUNC__ 18265615d1e5SSatish Balay #define __FUNC__ "MatLoad_SeqAIJ" 182719bcc07fSBarry Smith int MatLoad_SeqAIJ(Viewer viewer,MatType type,Mat *A) 182817ab2063SBarry Smith { 1829416022c9SBarry Smith Mat_SeqAIJ *a; 1830416022c9SBarry Smith Mat B; 183117699dbbSLois Curfman McInnes int i, nz, ierr, fd, header[4],size,*rowlengths = 0,M,N,shift; 1832bcd2baecSBarry Smith MPI_Comm comm; 183317ab2063SBarry Smith 183419bcc07fSBarry Smith PetscObjectGetComm((PetscObject) viewer,&comm); 183517699dbbSLois Curfman McInnes MPI_Comm_size(comm,&size); 1836e3372554SBarry Smith if (size > 1) SETERRQ(1,0,"view must have one processor"); 183790ace30eSBarry Smith ierr = ViewerBinaryGetDescriptor(viewer,&fd); CHKERRQ(ierr); 183877c4ece6SBarry Smith ierr = PetscBinaryRead(fd,header,4,BINARY_INT); CHKERRQ(ierr); 1839e3372554SBarry Smith if (header[0] != MAT_COOKIE) SETERRQ(1,0,"not matrix object in file"); 184017ab2063SBarry Smith M = header[1]; N = header[2]; nz = header[3]; 184117ab2063SBarry Smith 184217ab2063SBarry Smith /* read in row lengths */ 18430452661fSBarry Smith rowlengths = (int*) PetscMalloc( M*sizeof(int) ); CHKPTRQ(rowlengths); 184477c4ece6SBarry Smith ierr = PetscBinaryRead(fd,rowlengths,M,BINARY_INT); CHKERRQ(ierr); 184517ab2063SBarry Smith 184617ab2063SBarry Smith /* create our matrix */ 1847416022c9SBarry Smith ierr = MatCreateSeqAIJ(comm,M,N,0,rowlengths,A); CHKERRQ(ierr); 1848416022c9SBarry Smith B = *A; 1849416022c9SBarry Smith a = (Mat_SeqAIJ *) B->data; 1850416022c9SBarry Smith shift = a->indexshift; 185117ab2063SBarry Smith 185217ab2063SBarry Smith /* read in column indices and adjust for Fortran indexing*/ 185377c4ece6SBarry Smith ierr = PetscBinaryRead(fd,a->j,nz,BINARY_INT); CHKERRQ(ierr); 185417ab2063SBarry Smith if (shift) { 185517ab2063SBarry Smith for ( i=0; i<nz; i++ ) { 1856416022c9SBarry Smith a->j[i] += 1; 185717ab2063SBarry Smith } 185817ab2063SBarry Smith } 185917ab2063SBarry Smith 186017ab2063SBarry Smith /* read in nonzero values */ 186177c4ece6SBarry Smith ierr = PetscBinaryRead(fd,a->a,nz,BINARY_SCALAR); CHKERRQ(ierr); 186217ab2063SBarry Smith 186317ab2063SBarry Smith /* set matrix "i" values */ 1864416022c9SBarry Smith a->i[0] = -shift; 186517ab2063SBarry Smith for ( i=1; i<= M; i++ ) { 1866416022c9SBarry Smith a->i[i] = a->i[i-1] + rowlengths[i-1]; 1867416022c9SBarry Smith a->ilen[i-1] = rowlengths[i-1]; 186817ab2063SBarry Smith } 18690452661fSBarry Smith PetscFree(rowlengths); 187017ab2063SBarry Smith 18716d4a8577SBarry Smith ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr); 18726d4a8577SBarry Smith ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr); 187317ab2063SBarry Smith return 0; 187417ab2063SBarry Smith } 187517ab2063SBarry Smith 18765615d1e5SSatish Balay #undef __FUNC__ 18775615d1e5SSatish Balay #define __FUNC__ "MatEqual_SeqAIJ" 18788f6be9afSLois Curfman McInnes int MatEqual_SeqAIJ(Mat A,Mat B, PetscTruth* flg) 18797264ac53SSatish Balay { 18807264ac53SSatish Balay Mat_SeqAIJ *a = (Mat_SeqAIJ *)A->data, *b = (Mat_SeqAIJ *)B->data; 18817264ac53SSatish Balay 1882e3372554SBarry Smith if (B->type !=MATSEQAIJ)SETERRQ(1,0,"Matrices must be same type"); 18837264ac53SSatish Balay 18847264ac53SSatish Balay /* If the matrix dimensions are not equal, or no of nonzeros or shift */ 18857264ac53SSatish Balay if ((a->m != b->m ) || (a->n !=b->n) ||( a->nz != b->nz)|| 1886bcd2baecSBarry Smith (a->indexshift != b->indexshift)) { 188777c4ece6SBarry Smith *flg = PETSC_FALSE; return 0; 1888bcd2baecSBarry Smith } 18897264ac53SSatish Balay 18907264ac53SSatish Balay /* if the a->i are the same */ 1891bcd2baecSBarry Smith if (PetscMemcmp(a->i,b->i, (a->n+1)*sizeof(int))) { 189277c4ece6SBarry Smith *flg = PETSC_FALSE; return 0; 18937264ac53SSatish Balay } 18947264ac53SSatish Balay 18957264ac53SSatish Balay /* if a->j are the same */ 1896bcd2baecSBarry Smith if (PetscMemcmp(a->j, b->j, (a->nz)*sizeof(int))) { 189777c4ece6SBarry Smith *flg = PETSC_FALSE; return 0; 1898bcd2baecSBarry Smith } 1899bcd2baecSBarry Smith 1900bcd2baecSBarry Smith /* if a->a are the same */ 190119bcc07fSBarry Smith if (PetscMemcmp(a->a, b->a, (a->nz)*sizeof(Scalar))) { 190277c4ece6SBarry Smith *flg = PETSC_FALSE; return 0; 19037264ac53SSatish Balay } 190477c4ece6SBarry Smith *flg = PETSC_TRUE; 19057264ac53SSatish Balay return 0; 19067264ac53SSatish Balay 19077264ac53SSatish Balay } 1908