16d84be18SBarry Smith 26945ee14SBarry Smith 317ab2063SBarry Smith #ifndef lint 4*dd097bc3SLois Curfman McInnes static char vcid[] = "$Id: aij.c,v 1.207 1997/02/11 19:46:31 curfman Exp curfman $"; 517ab2063SBarry Smith #endif 617ab2063SBarry Smith 7d5d45c9bSBarry Smith /* 85a838052SSatish Balay B Defines the basic matrix operations for the AIJ (compressed row) 9d5d45c9bSBarry Smith matrix storage format. 10d5d45c9bSBarry Smith */ 1170f55243SBarry Smith #include "src/mat/impls/aij/seq/aij.h" 12f5eb4b81SSatish Balay #include "src/vec/vecimpl.h" 13f5eb4b81SSatish Balay #include "src/inline/spops.h" 14e4d965acSSatish Balay #include "petsc.h" 15f5eb4b81SSatish Balay #include "src/inline/bitarray.h" 1617ab2063SBarry Smith 17a2ce50c7SBarry Smith /* 18a2ce50c7SBarry Smith Basic AIJ format ILU based on drop tolerance 19a2ce50c7SBarry Smith */ 205615d1e5SSatish Balay #undef __FUNC__ 215615d1e5SSatish Balay #define __FUNC__ "MatILUDTFactor_SeqAIJ" 22a2ce50c7SBarry Smith int MatILUDTFactor_SeqAIJ(Mat A,double dt,int maxnz,IS row,IS col,Mat *fact) 23a2ce50c7SBarry Smith { 24a2ce50c7SBarry Smith /* Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; */ 25a2ce50c7SBarry Smith int ierr = 1; 26a2ce50c7SBarry Smith 27e3372554SBarry Smith SETERRQ(ierr,0,"Not implemented"); 28a2ce50c7SBarry Smith } 29a2ce50c7SBarry Smith 30bcd2baecSBarry Smith extern int MatToSymmetricIJ_SeqAIJ(int,int*,int*,int,int,int**,int**); 3117ab2063SBarry Smith 325615d1e5SSatish Balay #undef __FUNC__ 335615d1e5SSatish Balay #define __FUNC__ "MatGetRowIJ_SeqAIJ" 3431625ec5SSatish Balay static int MatGetRowIJ_SeqAIJ(Mat A,int oshift,PetscTruth symmetric,int *m,int **ia,int **ja, 356945ee14SBarry Smith PetscTruth *done) 3617ab2063SBarry Smith { 37416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 386945ee14SBarry Smith int ierr,i,ishift; 3917ab2063SBarry Smith 4031625ec5SSatish Balay *m = A->m; 416945ee14SBarry Smith if (!ia) return 0; 426945ee14SBarry Smith ishift = a->indexshift; 436945ee14SBarry Smith if (symmetric) { 4431625ec5SSatish Balay ierr = MatToSymmetricIJ_SeqAIJ(a->m,a->i,a->j,ishift,oshift,ia,ja); CHKERRQ(ierr); 456945ee14SBarry Smith } else if (oshift == 0 && ishift == -1) { 4631625ec5SSatish Balay int nz = a->i[a->m]; 473b2fbd54SBarry Smith /* malloc space and subtract 1 from i and j indices */ 4831625ec5SSatish Balay *ia = (int *) PetscMalloc( (a->m+1)*sizeof(int) ); CHKPTRQ(*ia); 493b2fbd54SBarry Smith *ja = (int *) PetscMalloc( (nz+1)*sizeof(int) ); CHKPTRQ(*ja); 503b2fbd54SBarry Smith for ( i=0; i<nz; i++ ) (*ja)[i] = a->j[i] - 1; 5131625ec5SSatish Balay for ( i=0; i<a->m+1; i++ ) (*ia)[i] = a->i[i] - 1; 526945ee14SBarry Smith } else if (oshift == 1 && ishift == 0) { 5331625ec5SSatish Balay int nz = a->i[a->m] + 1; 543b2fbd54SBarry Smith /* malloc space and add 1 to i and j indices */ 5531625ec5SSatish Balay *ia = (int *) PetscMalloc( (a->m+1)*sizeof(int) ); CHKPTRQ(*ia); 563b2fbd54SBarry Smith *ja = (int *) PetscMalloc( (nz+1)*sizeof(int) ); CHKPTRQ(*ja); 573b2fbd54SBarry Smith for ( i=0; i<nz; i++ ) (*ja)[i] = a->j[i] + 1; 5831625ec5SSatish Balay for ( i=0; i<a->m+1; i++ ) (*ia)[i] = a->i[i] + 1; 596945ee14SBarry Smith } else { 606945ee14SBarry Smith *ia = a->i; *ja = a->j; 61a2ce50c7SBarry Smith } 62a2ce50c7SBarry Smith 63a2744918SBarry Smith return 0; 64a2744918SBarry Smith } 65a2744918SBarry Smith 665615d1e5SSatish Balay #undef __FUNC__ 675615d1e5SSatish Balay #define __FUNC__ "MatRestoreRowIJ_SeqAIJ" 683b2fbd54SBarry Smith static int MatRestoreRowIJ_SeqAIJ(Mat A,int oshift,PetscTruth symmetric,int *n,int **ia,int **ja, 696945ee14SBarry Smith PetscTruth *done) 706945ee14SBarry Smith { 716945ee14SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 723b2fbd54SBarry Smith int ishift = a->indexshift; 736945ee14SBarry Smith 746945ee14SBarry Smith if (!ia) return 0; 753b2fbd54SBarry Smith if (symmetric || (oshift == 0 && ishift == -1) || (oshift == 1 && ishift == 0)) { 766945ee14SBarry Smith PetscFree(*ia); 776945ee14SBarry Smith PetscFree(*ja); 78bcd2baecSBarry Smith } 7917ab2063SBarry Smith return 0; 8017ab2063SBarry Smith } 8117ab2063SBarry Smith 825615d1e5SSatish Balay #undef __FUNC__ 835615d1e5SSatish Balay #define __FUNC__ "MatGetColumnIJ_SeqAIJ" 8443a90d84SBarry Smith int MatGetColumnIJ_SeqAIJ(Mat A,int oshift,PetscTruth symmetric,int *nn,int **ia,int **ja, 853b2fbd54SBarry Smith PetscTruth *done) 863b2fbd54SBarry Smith { 873b2fbd54SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 88a93ec695SBarry Smith int ierr,i,ishift = a->indexshift,*collengths,*cia,*cja,n = A->n,m = A->m; 89a93ec695SBarry Smith int nz = a->i[m]+ishift,row,*jj,mr,col; 903b2fbd54SBarry Smith 913b2fbd54SBarry Smith *nn = A->n; 923b2fbd54SBarry Smith if (!ia) return 0; 933b2fbd54SBarry Smith if (symmetric) { 94179192dfSSatish Balay ierr = MatToSymmetricIJ_SeqAIJ(a->m,a->i,a->j,ishift,oshift,ia,ja); CHKERRQ(ierr); 953b2fbd54SBarry Smith } else { 9661d2ded1SBarry Smith collengths = (int *) PetscMalloc( (n+1)*sizeof(int) ); CHKPTRQ(collengths); 973b2fbd54SBarry Smith PetscMemzero(collengths,n*sizeof(int)); 983b2fbd54SBarry Smith cia = (int *) PetscMalloc( (n+1)*sizeof(int) ); CHKPTRQ(cia); 99a93ec695SBarry Smith cja = (int *) PetscMalloc( (nz+1)*sizeof(int) ); CHKPTRQ(cja); 1003b2fbd54SBarry Smith jj = a->j; 1013b2fbd54SBarry Smith for ( i=0; i<nz; i++ ) { 1023b2fbd54SBarry Smith collengths[jj[i] + ishift]++; 1033b2fbd54SBarry Smith } 1043b2fbd54SBarry Smith cia[0] = oshift; 1053b2fbd54SBarry Smith for ( i=0; i<n; i++) { 1063b2fbd54SBarry Smith cia[i+1] = cia[i] + collengths[i]; 1073b2fbd54SBarry Smith } 1083b2fbd54SBarry Smith PetscMemzero(collengths,n*sizeof(int)); 1093b2fbd54SBarry Smith jj = a->j; 110a93ec695SBarry Smith for ( row=0; row<m; row++ ) { 111a93ec695SBarry Smith mr = a->i[row+1] - a->i[row]; 112a93ec695SBarry Smith for ( i=0; i<mr; i++ ) { 1133b2fbd54SBarry Smith col = *jj++ + ishift; 1143b2fbd54SBarry Smith cja[cia[col] + collengths[col]++ - oshift] = row + oshift; 1153b2fbd54SBarry Smith } 1163b2fbd54SBarry Smith } 1173b2fbd54SBarry Smith PetscFree(collengths); 1183b2fbd54SBarry Smith *ia = cia; *ja = cja; 1193b2fbd54SBarry Smith } 1203b2fbd54SBarry Smith 1213b2fbd54SBarry Smith return 0; 1223b2fbd54SBarry Smith } 1233b2fbd54SBarry Smith 1245615d1e5SSatish Balay #undef __FUNC__ 1255615d1e5SSatish Balay #define __FUNC__ "MatRestoreColumnIJ_SeqAIJ" 12643a90d84SBarry Smith int MatRestoreColumnIJ_SeqAIJ(Mat A,int oshift,PetscTruth symmetric,int *n,int **ia, 1273b2fbd54SBarry Smith int **ja,PetscTruth *done) 1283b2fbd54SBarry Smith { 1293b2fbd54SBarry Smith if (!ia) return 0; 1303b2fbd54SBarry Smith 1313b2fbd54SBarry Smith PetscFree(*ia); 1323b2fbd54SBarry Smith PetscFree(*ja); 1333b2fbd54SBarry Smith 1343b2fbd54SBarry Smith return 0; 1353b2fbd54SBarry Smith } 1363b2fbd54SBarry Smith 137227d817aSBarry Smith #define CHUNKSIZE 15 13817ab2063SBarry Smith 1395615d1e5SSatish Balay #undef __FUNC__ 1405615d1e5SSatish Balay #define __FUNC__ "MatSetValues_SeqAIJ" 14161d2ded1SBarry Smith int MatSetValues_SeqAIJ(Mat A,int m,int *im,int n,int *in,Scalar *v,InsertMode is) 14217ab2063SBarry Smith { 143416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 144416022c9SBarry Smith int *rp,k,low,high,t,ii,row,nrow,i,col,l,rmax, N, sorted = a->sorted; 1454b0e389bSBarry Smith int *imax = a->imax, *ai = a->i, *ailen = a->ilen,roworiented = a->roworiented; 146d5d45c9bSBarry Smith int *aj = a->j, nonew = a->nonew,shift = a->indexshift; 147416022c9SBarry Smith Scalar *ap,value, *aa = a->a; 14817ab2063SBarry Smith 14917ab2063SBarry Smith for ( k=0; k<m; k++ ) { /* loop over added rows */ 150416022c9SBarry Smith row = im[k]; 1513b2fbd54SBarry Smith #if defined(PETSC_BOPT_g) 152e3372554SBarry Smith if (row < 0) SETERRQ(1,0,"Negative row"); 153e3372554SBarry Smith if (row >= a->m) SETERRQ(1,0,"Row too large"); 1543b2fbd54SBarry Smith #endif 15517ab2063SBarry Smith rp = aj + ai[row] + shift; ap = aa + ai[row] + shift; 15617ab2063SBarry Smith rmax = imax[row]; nrow = ailen[row]; 157416022c9SBarry Smith low = 0; 15817ab2063SBarry Smith for ( l=0; l<n; l++ ) { /* loop over added columns */ 1593b2fbd54SBarry Smith #if defined(PETSC_BOPT_g) 160e3372554SBarry Smith if (in[l] < 0) SETERRQ(1,0,"Negative column"); 161e3372554SBarry Smith if (in[l] >= a->n) SETERRQ(1,0,"Column too large"); 1623b2fbd54SBarry Smith #endif 1634b0e389bSBarry Smith col = in[l] - shift; 1644b0e389bSBarry Smith if (roworiented) { 1654b0e389bSBarry Smith value = *v++; 1664b0e389bSBarry Smith } 1674b0e389bSBarry Smith else { 1684b0e389bSBarry Smith value = v[k + l*m]; 1694b0e389bSBarry Smith } 170416022c9SBarry Smith if (!sorted) low = 0; high = nrow; 171416022c9SBarry Smith while (high-low > 5) { 172416022c9SBarry Smith t = (low+high)/2; 173416022c9SBarry Smith if (rp[t] > col) high = t; 174416022c9SBarry Smith else low = t; 17517ab2063SBarry Smith } 176416022c9SBarry Smith for ( i=low; i<high; i++ ) { 17717ab2063SBarry Smith if (rp[i] > col) break; 17817ab2063SBarry Smith if (rp[i] == col) { 179416022c9SBarry Smith if (is == ADD_VALUES) ap[i] += value; 18017ab2063SBarry Smith else ap[i] = value; 18117ab2063SBarry Smith goto noinsert; 18217ab2063SBarry Smith } 18317ab2063SBarry Smith } 18417ab2063SBarry Smith if (nonew) goto noinsert; 18517ab2063SBarry Smith if (nrow >= rmax) { 18617ab2063SBarry Smith /* there is no extra room in row, therefore enlarge */ 187416022c9SBarry Smith int new_nz = ai[a->m] + CHUNKSIZE,len,*new_i,*new_j; 18817ab2063SBarry Smith Scalar *new_a; 18917ab2063SBarry Smith 19017ab2063SBarry Smith /* malloc new storage space */ 191416022c9SBarry Smith len = new_nz*(sizeof(int)+sizeof(Scalar))+(a->m+1)*sizeof(int); 1920452661fSBarry Smith new_a = (Scalar *) PetscMalloc( len ); CHKPTRQ(new_a); 19317ab2063SBarry Smith new_j = (int *) (new_a + new_nz); 19417ab2063SBarry Smith new_i = new_j + new_nz; 19517ab2063SBarry Smith 19617ab2063SBarry Smith /* copy over old data into new slots */ 19717ab2063SBarry Smith for ( ii=0; ii<row+1; ii++ ) {new_i[ii] = ai[ii];} 198416022c9SBarry Smith for ( ii=row+1; ii<a->m+1; ii++ ) {new_i[ii] = ai[ii]+CHUNKSIZE;} 199416022c9SBarry Smith PetscMemcpy(new_j,aj,(ai[row]+nrow+shift)*sizeof(int)); 200416022c9SBarry Smith len = (new_nz - CHUNKSIZE - ai[row] - nrow - shift); 201416022c9SBarry Smith PetscMemcpy(new_j+ai[row]+shift+nrow+CHUNKSIZE,aj+ai[row]+shift+nrow, 20217ab2063SBarry Smith len*sizeof(int)); 203416022c9SBarry Smith PetscMemcpy(new_a,aa,(ai[row]+nrow+shift)*sizeof(Scalar)); 204416022c9SBarry Smith PetscMemcpy(new_a+ai[row]+shift+nrow+CHUNKSIZE,aa+ai[row]+shift+nrow, 20517ab2063SBarry Smith len*sizeof(Scalar)); 20617ab2063SBarry Smith /* free up old matrix storage */ 2070452661fSBarry Smith PetscFree(a->a); 2080452661fSBarry Smith if (!a->singlemalloc) {PetscFree(a->i);PetscFree(a->j);} 209416022c9SBarry Smith aa = a->a = new_a; ai = a->i = new_i; aj = a->j = new_j; 210416022c9SBarry Smith a->singlemalloc = 1; 21117ab2063SBarry Smith 21217ab2063SBarry Smith rp = aj + ai[row] + shift; ap = aa + ai[row] + shift; 213416022c9SBarry Smith rmax = imax[row] = imax[row] + CHUNKSIZE; 214416022c9SBarry Smith PLogObjectMemory(A,CHUNKSIZE*(sizeof(int) + sizeof(Scalar))); 215416022c9SBarry Smith a->maxnz += CHUNKSIZE; 216b810aeb4SBarry Smith a->reallocs++; 21717ab2063SBarry Smith } 218416022c9SBarry Smith N = nrow++ - 1; a->nz++; 219416022c9SBarry Smith /* shift up all the later entries in this row */ 220416022c9SBarry Smith for ( ii=N; ii>=i; ii-- ) { 22117ab2063SBarry Smith rp[ii+1] = rp[ii]; 22217ab2063SBarry Smith ap[ii+1] = ap[ii]; 22317ab2063SBarry Smith } 22417ab2063SBarry Smith rp[i] = col; 22517ab2063SBarry Smith ap[i] = value; 22617ab2063SBarry Smith noinsert:; 227416022c9SBarry Smith low = i + 1; 22817ab2063SBarry Smith } 22917ab2063SBarry Smith ailen[row] = nrow; 23017ab2063SBarry Smith } 23117ab2063SBarry Smith return 0; 23217ab2063SBarry Smith } 23317ab2063SBarry Smith 2345615d1e5SSatish Balay #undef __FUNC__ 2355615d1e5SSatish Balay #define __FUNC__ "MatGetValues_SeqAIJ" 2367eb43aa7SLois Curfman McInnes static int MatGetValues_SeqAIJ(Mat A,int m,int *im,int n,int *in,Scalar *v) 2377eb43aa7SLois Curfman McInnes { 2387eb43aa7SLois Curfman McInnes Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 239b49de8d1SLois Curfman McInnes int *rp, k, low, high, t, row, nrow, i, col, l, *aj = a->j; 2407eb43aa7SLois Curfman McInnes int *ai = a->i, *ailen = a->ilen, shift = a->indexshift; 2417eb43aa7SLois Curfman McInnes Scalar *ap, *aa = a->a, zero = 0.0; 2427eb43aa7SLois Curfman McInnes 2437eb43aa7SLois Curfman McInnes for ( k=0; k<m; k++ ) { /* loop over rows */ 2447eb43aa7SLois Curfman McInnes row = im[k]; 245e3372554SBarry Smith if (row < 0) SETERRQ(1,0,"Negative row"); 246e3372554SBarry Smith if (row >= a->m) SETERRQ(1,0,"Row too large"); 2477eb43aa7SLois Curfman McInnes rp = aj + ai[row] + shift; ap = aa + ai[row] + shift; 2487eb43aa7SLois Curfman McInnes nrow = ailen[row]; 2497eb43aa7SLois Curfman McInnes for ( l=0; l<n; l++ ) { /* loop over columns */ 250e3372554SBarry Smith if (in[l] < 0) SETERRQ(1,0,"Negative column"); 251e3372554SBarry Smith if (in[l] >= a->n) SETERRQ(1,0,"Column too large"); 2527eb43aa7SLois Curfman McInnes col = in[l] - shift; 2537eb43aa7SLois Curfman McInnes high = nrow; low = 0; /* assume unsorted */ 2547eb43aa7SLois Curfman McInnes while (high-low > 5) { 2557eb43aa7SLois Curfman McInnes t = (low+high)/2; 2567eb43aa7SLois Curfman McInnes if (rp[t] > col) high = t; 2577eb43aa7SLois Curfman McInnes else low = t; 2587eb43aa7SLois Curfman McInnes } 2597eb43aa7SLois Curfman McInnes for ( i=low; i<high; i++ ) { 2607eb43aa7SLois Curfman McInnes if (rp[i] > col) break; 2617eb43aa7SLois Curfman McInnes if (rp[i] == col) { 262b49de8d1SLois Curfman McInnes *v++ = ap[i]; 2637eb43aa7SLois Curfman McInnes goto finished; 2647eb43aa7SLois Curfman McInnes } 2657eb43aa7SLois Curfman McInnes } 266b49de8d1SLois Curfman McInnes *v++ = zero; 2677eb43aa7SLois Curfman McInnes finished:; 2687eb43aa7SLois Curfman McInnes } 2697eb43aa7SLois Curfman McInnes } 2707eb43aa7SLois Curfman McInnes return 0; 2717eb43aa7SLois Curfman McInnes } 2727eb43aa7SLois Curfman McInnes 27317ab2063SBarry Smith #include "draw.h" 27417ab2063SBarry Smith #include "pinclude/pviewer.h" 27577c4ece6SBarry Smith #include "sys.h" 27617ab2063SBarry Smith 2775615d1e5SSatish Balay #undef __FUNC__ 2785615d1e5SSatish Balay #define __FUNC__ "MatView_SeqAIJ_Binary" 279416022c9SBarry Smith static int MatView_SeqAIJ_Binary(Mat A,Viewer viewer) 28017ab2063SBarry Smith { 281416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 282416022c9SBarry Smith int i, fd, *col_lens, ierr; 28317ab2063SBarry Smith 28490ace30eSBarry Smith ierr = ViewerBinaryGetDescriptor(viewer,&fd); CHKERRQ(ierr); 2850452661fSBarry Smith col_lens = (int *) PetscMalloc( (4+a->m)*sizeof(int) ); CHKPTRQ(col_lens); 286416022c9SBarry Smith col_lens[0] = MAT_COOKIE; 287416022c9SBarry Smith col_lens[1] = a->m; 288416022c9SBarry Smith col_lens[2] = a->n; 289416022c9SBarry Smith col_lens[3] = a->nz; 290416022c9SBarry Smith 291416022c9SBarry Smith /* store lengths of each row and write (including header) to file */ 292416022c9SBarry Smith for ( i=0; i<a->m; i++ ) { 293416022c9SBarry Smith col_lens[4+i] = a->i[i+1] - a->i[i]; 29417ab2063SBarry Smith } 29577c4ece6SBarry Smith ierr = PetscBinaryWrite(fd,col_lens,4+a->m,BINARY_INT,1); CHKERRQ(ierr); 2960452661fSBarry Smith PetscFree(col_lens); 297416022c9SBarry Smith 298416022c9SBarry Smith /* store column indices (zero start index) */ 299416022c9SBarry Smith if (a->indexshift) { 300416022c9SBarry Smith for ( i=0; i<a->nz; i++ ) a->j[i]--; 30117ab2063SBarry Smith } 30277c4ece6SBarry Smith ierr = PetscBinaryWrite(fd,a->j,a->nz,BINARY_INT,0); CHKERRQ(ierr); 303416022c9SBarry Smith if (a->indexshift) { 304416022c9SBarry Smith for ( i=0; i<a->nz; i++ ) a->j[i]++; 30517ab2063SBarry Smith } 306416022c9SBarry Smith 307416022c9SBarry Smith /* store nonzero values */ 30877c4ece6SBarry Smith ierr = PetscBinaryWrite(fd,a->a,a->nz,BINARY_SCALAR,0); CHKERRQ(ierr); 30917ab2063SBarry Smith return 0; 31017ab2063SBarry Smith } 311416022c9SBarry Smith 3125615d1e5SSatish Balay #undef __FUNC__ 3135615d1e5SSatish Balay #define __FUNC__ "MatView_SeqAIJ_ASCII" 314416022c9SBarry Smith static int MatView_SeqAIJ_ASCII(Mat A,Viewer viewer) 315416022c9SBarry Smith { 316416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 317496e697eSBarry Smith int ierr, i,j, m = a->m, shift = a->indexshift, format, flg1,flg2; 31817ab2063SBarry Smith FILE *fd; 31917ab2063SBarry Smith char *outputname; 32017ab2063SBarry Smith 32190ace30eSBarry Smith ierr = ViewerASCIIGetPointer(viewer,&fd); CHKERRQ(ierr); 322416022c9SBarry Smith ierr = ViewerFileGetOutputname_Private(viewer,&outputname); CHKERRQ(ierr); 32390ace30eSBarry Smith ierr = ViewerGetFormat(viewer,&format); 324a93ec695SBarry Smith if (format == VIEWER_FORMAT_ASCII_INFO) { 32595e01e2fSLois Curfman McInnes return 0; 32695e01e2fSLois Curfman McInnes } 327a93ec695SBarry Smith else if (format == VIEWER_FORMAT_ASCII_INFO_LONG) { 328496e697eSBarry Smith ierr = OptionsHasName(PETSC_NULL,"-mat_aij_no_inode",&flg1); CHKERRQ(ierr); 329496e697eSBarry Smith ierr = OptionsHasName(PETSC_NULL,"-mat_no_unroll",&flg2); CHKERRQ(ierr); 330496e697eSBarry Smith if (flg1 || flg2) fprintf(fd," not using I-node routines\n"); 33195e01e2fSLois Curfman McInnes else fprintf(fd," using I-node routines: found %d nodes, limit used is %d\n", 33295e01e2fSLois Curfman McInnes a->inode.node_count,a->inode.limit); 33317ab2063SBarry Smith } 334a93ec695SBarry Smith else if (format == VIEWER_FORMAT_ASCII_MATLAB) { 335416022c9SBarry Smith fprintf(fd,"%% Size = %d %d \n",m,a->n); 3364e220ebcSLois Curfman McInnes fprintf(fd,"%% Nonzeros = %d \n",a->nz); 3374e220ebcSLois Curfman McInnes fprintf(fd,"zzz = zeros(%d,3);\n",a->nz); 33817ab2063SBarry Smith fprintf(fd,"zzz = [\n"); 33917ab2063SBarry Smith 34017ab2063SBarry Smith for (i=0; i<m; i++) { 341416022c9SBarry Smith for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 34217ab2063SBarry Smith #if defined(PETSC_COMPLEX) 3436945ee14SBarry Smith fprintf(fd,"%d %d %18.16e + %18.16e i \n",i+1,a->j[j]+!shift,real(a->a[j]), 344416022c9SBarry Smith imag(a->a[j])); 34517ab2063SBarry Smith #else 3467a743949SBarry Smith fprintf(fd,"%d %d %18.16e\n", i+1, a->j[j]+!shift, a->a[j]); 34717ab2063SBarry Smith #endif 34817ab2063SBarry Smith } 34917ab2063SBarry Smith } 35017ab2063SBarry Smith fprintf(fd,"];\n %s = spconvert(zzz);\n",outputname); 35117ab2063SBarry Smith } 352a93ec695SBarry Smith else if (format == VIEWER_FORMAT_ASCII_COMMON) { 35344cd7ae7SLois Curfman McInnes for ( i=0; i<m; i++ ) { 35444cd7ae7SLois Curfman McInnes fprintf(fd,"row %d:",i); 35544cd7ae7SLois Curfman McInnes for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 35644cd7ae7SLois Curfman McInnes #if defined(PETSC_COMPLEX) 35744cd7ae7SLois Curfman McInnes if (imag(a->a[j]) != 0.0 && real(a->a[j]) != 0.0) 35844cd7ae7SLois Curfman McInnes fprintf(fd," %d %g + %g i",a->j[j]+shift,real(a->a[j]),imag(a->a[j])); 35944cd7ae7SLois Curfman McInnes else if (real(a->a[j]) != 0.0) 36044cd7ae7SLois Curfman McInnes fprintf(fd," %d %g ",a->j[j]+shift,real(a->a[j])); 36144cd7ae7SLois Curfman McInnes #else 36244cd7ae7SLois Curfman McInnes if (a->a[j] != 0.0) fprintf(fd," %d %g ",a->j[j]+shift,a->a[j]); 36344cd7ae7SLois Curfman McInnes #endif 36444cd7ae7SLois Curfman McInnes } 36544cd7ae7SLois Curfman McInnes fprintf(fd,"\n"); 36644cd7ae7SLois Curfman McInnes } 36744cd7ae7SLois Curfman McInnes } 36817ab2063SBarry Smith else { 36917ab2063SBarry Smith for ( i=0; i<m; i++ ) { 37017ab2063SBarry Smith fprintf(fd,"row %d:",i); 371416022c9SBarry Smith for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 37217ab2063SBarry Smith #if defined(PETSC_COMPLEX) 373416022c9SBarry Smith if (imag(a->a[j]) != 0.0) { 374416022c9SBarry Smith fprintf(fd," %d %g + %g i",a->j[j]+shift,real(a->a[j]),imag(a->a[j])); 37517ab2063SBarry Smith } 37617ab2063SBarry Smith else { 377416022c9SBarry Smith fprintf(fd," %d %g ",a->j[j]+shift,real(a->a[j])); 37817ab2063SBarry Smith } 37917ab2063SBarry Smith #else 380416022c9SBarry Smith fprintf(fd," %d %g ",a->j[j]+shift,a->a[j]); 38117ab2063SBarry Smith #endif 38217ab2063SBarry Smith } 38317ab2063SBarry Smith fprintf(fd,"\n"); 38417ab2063SBarry Smith } 38517ab2063SBarry Smith } 38617ab2063SBarry Smith fflush(fd); 387416022c9SBarry Smith return 0; 388416022c9SBarry Smith } 389416022c9SBarry Smith 3905615d1e5SSatish Balay #undef __FUNC__ 3915615d1e5SSatish Balay #define __FUNC__ "MatView_SeqAIJ_Draw" 392416022c9SBarry Smith static int MatView_SeqAIJ_Draw(Mat A,Viewer viewer) 393416022c9SBarry Smith { 394416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 395cddf8d76SBarry Smith int ierr, i,j, m = a->m, shift = a->indexshift,pause,color; 3960513a670SBarry Smith int format; 39794a9d846SBarry Smith double xl,yl,xr,yr,w,h,xc,yc,scale = 1.0,x_l,x_r,y_l,y_r,maxv = 0.0; 398bcd2baecSBarry Smith Draw draw; 399cddf8d76SBarry Smith DrawButton button; 40019bcc07fSBarry Smith PetscTruth isnull; 401cddf8d76SBarry Smith 4020513a670SBarry Smith ierr = ViewerDrawGetDraw(viewer,&draw); CHKERRQ(ierr); 4030513a670SBarry Smith ierr = DrawClear(draw); CHKERRQ(ierr); 4040513a670SBarry Smith ierr = ViewerGetFormat(viewer,&format); CHKERRQ(ierr); 40519bcc07fSBarry Smith ierr = DrawIsNull(draw,&isnull); CHKERRQ(ierr); if (isnull) return 0; 40619bcc07fSBarry Smith 407416022c9SBarry Smith xr = a->n; yr = a->m; h = yr/10.0; w = xr/10.0; 408416022c9SBarry Smith xr += w; yr += h; xl = -w; yl = -h; 409416022c9SBarry Smith ierr = DrawSetCoordinates(draw,xl,yl,xr,yr); CHKERRQ(ierr); 410416022c9SBarry Smith /* loop over matrix elements drawing boxes */ 4110513a670SBarry Smith 4120513a670SBarry Smith if (format != VIEWER_FORMAT_DRAW_CONTOUR) { 4130513a670SBarry Smith /* Blue for negative, Cyan for zero and Red for positive */ 414cddf8d76SBarry Smith color = DRAW_BLUE; 415416022c9SBarry Smith for ( i=0; i<m; i++ ) { 416cddf8d76SBarry Smith y_l = m - i - 1.0; y_r = y_l + 1.0; 417416022c9SBarry Smith for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 418cddf8d76SBarry Smith x_l = a->j[j] + shift; x_r = x_l + 1.0; 419cddf8d76SBarry Smith #if defined(PETSC_COMPLEX) 420cddf8d76SBarry Smith if (real(a->a[j]) >= 0.) continue; 421cddf8d76SBarry Smith #else 422cddf8d76SBarry Smith if (a->a[j] >= 0.) continue; 423cddf8d76SBarry Smith #endif 424cddf8d76SBarry Smith DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color); 425cddf8d76SBarry Smith } 426cddf8d76SBarry Smith } 427cddf8d76SBarry Smith color = DRAW_CYAN; 428cddf8d76SBarry Smith for ( i=0; i<m; i++ ) { 429cddf8d76SBarry Smith y_l = m - i - 1.0; y_r = y_l + 1.0; 430cddf8d76SBarry Smith for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 431cddf8d76SBarry Smith x_l = a->j[j] + shift; x_r = x_l + 1.0; 432cddf8d76SBarry Smith if (a->a[j] != 0.) continue; 433cddf8d76SBarry Smith DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color); 434cddf8d76SBarry Smith } 435cddf8d76SBarry Smith } 436cddf8d76SBarry Smith color = DRAW_RED; 437cddf8d76SBarry Smith for ( i=0; i<m; i++ ) { 438cddf8d76SBarry Smith y_l = m - i - 1.0; y_r = y_l + 1.0; 439cddf8d76SBarry Smith for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 440cddf8d76SBarry Smith x_l = a->j[j] + shift; x_r = x_l + 1.0; 441cddf8d76SBarry Smith #if defined(PETSC_COMPLEX) 442cddf8d76SBarry Smith if (real(a->a[j]) <= 0.) continue; 443cddf8d76SBarry Smith #else 444cddf8d76SBarry Smith if (a->a[j] <= 0.) continue; 445cddf8d76SBarry Smith #endif 446cddf8d76SBarry Smith DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color); 447416022c9SBarry Smith } 448416022c9SBarry Smith } 4490513a670SBarry Smith } else { 4500513a670SBarry Smith /* use contour shading to indicate magnitude of values */ 4510513a670SBarry Smith /* first determine max of all nonzero values */ 4520513a670SBarry Smith int nz = a->nz,count; 4530513a670SBarry Smith Draw popup; 4540513a670SBarry Smith 4550513a670SBarry Smith for ( i=0; i<nz; i++ ) { 4560513a670SBarry Smith if (PetscAbsScalar(a->a[i]) > maxv) maxv = PetscAbsScalar(a->a[i]); 4570513a670SBarry Smith } 4580513a670SBarry Smith ierr = DrawCreatePopUp(draw,&popup); CHKERRQ(ierr); 4590513a670SBarry Smith ierr = DrawScalePopup(popup,0.0,maxv); CHKERRQ(ierr); 4600513a670SBarry Smith count = 0; 4610513a670SBarry Smith for ( i=0; i<m; i++ ) { 4620513a670SBarry Smith y_l = m - i - 1.0; y_r = y_l + 1.0; 4630513a670SBarry Smith for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 4640513a670SBarry Smith x_l = a->j[j] + shift; x_r = x_l + 1.0; 4650513a670SBarry Smith color = 32 + (int) ((200.0 - 32.0)*PetscAbsScalar(a->a[count])/maxv); 4660513a670SBarry Smith DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color); 4670513a670SBarry Smith count++; 4680513a670SBarry Smith } 4690513a670SBarry Smith } 4700513a670SBarry Smith } 471416022c9SBarry Smith DrawFlush(draw); 472cddf8d76SBarry Smith DrawGetPause(draw,&pause); 473cddf8d76SBarry Smith if (pause >= 0) { PetscSleep(pause); return 0;} 474cddf8d76SBarry Smith 475cddf8d76SBarry Smith /* allow the matrix to zoom or shrink */ 4766945ee14SBarry Smith ierr = DrawCheckResizedWindow(draw); 477cddf8d76SBarry Smith ierr = DrawGetMouseButton(draw,&button,&xc,&yc,0,0); 478cddf8d76SBarry Smith while (button != BUTTON_RIGHT) { 479cddf8d76SBarry Smith DrawClear(draw); 480cddf8d76SBarry Smith if (button == BUTTON_LEFT) scale = .5; 481cddf8d76SBarry Smith else if (button == BUTTON_CENTER) scale = 2.; 482cddf8d76SBarry Smith xl = scale*(xl + w - xc) + xc - w*scale; 483cddf8d76SBarry Smith xr = scale*(xr - w - xc) + xc + w*scale; 484cddf8d76SBarry Smith yl = scale*(yl + h - yc) + yc - h*scale; 485cddf8d76SBarry Smith yr = scale*(yr - h - yc) + yc + h*scale; 486cddf8d76SBarry Smith w *= scale; h *= scale; 487cddf8d76SBarry Smith ierr = DrawSetCoordinates(draw,xl,yl,xr,yr); CHKERRQ(ierr); 4880513a670SBarry Smith if (format != VIEWER_FORMAT_DRAW_CONTOUR) { 4890513a670SBarry Smith /* Blue for negative, Cyan for zero and Red for positive */ 490cddf8d76SBarry Smith color = DRAW_BLUE; 491cddf8d76SBarry Smith for ( i=0; i<m; i++ ) { 492cddf8d76SBarry Smith y_l = m - i - 1.0; y_r = y_l + 1.0; 493cddf8d76SBarry Smith for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 494cddf8d76SBarry Smith x_l = a->j[j] + shift; x_r = x_l + 1.0; 495cddf8d76SBarry Smith #if defined(PETSC_COMPLEX) 496cddf8d76SBarry Smith if (real(a->a[j]) >= 0.) continue; 497cddf8d76SBarry Smith #else 498cddf8d76SBarry Smith if (a->a[j] >= 0.) continue; 499cddf8d76SBarry Smith #endif 500cddf8d76SBarry Smith DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color); 501cddf8d76SBarry Smith } 502cddf8d76SBarry Smith } 503cddf8d76SBarry Smith color = DRAW_CYAN; 504cddf8d76SBarry Smith for ( i=0; i<m; i++ ) { 505cddf8d76SBarry Smith y_l = m - i - 1.0; y_r = y_l + 1.0; 506cddf8d76SBarry Smith for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 507cddf8d76SBarry Smith x_l = a->j[j] + shift; x_r = x_l + 1.0; 508cddf8d76SBarry Smith if (a->a[j] != 0.) continue; 509cddf8d76SBarry Smith DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color); 510cddf8d76SBarry Smith } 511cddf8d76SBarry Smith } 512cddf8d76SBarry Smith color = DRAW_RED; 513cddf8d76SBarry Smith for ( i=0; i<m; i++ ) { 514cddf8d76SBarry Smith y_l = m - i - 1.0; y_r = y_l + 1.0; 515cddf8d76SBarry Smith for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 516cddf8d76SBarry Smith x_l = a->j[j] + shift; x_r = x_l + 1.0; 517cddf8d76SBarry Smith #if defined(PETSC_COMPLEX) 518cddf8d76SBarry Smith if (real(a->a[j]) <= 0.) continue; 519cddf8d76SBarry Smith #else 520cddf8d76SBarry Smith if (a->a[j] <= 0.) continue; 521cddf8d76SBarry Smith #endif 522cddf8d76SBarry Smith DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color); 523cddf8d76SBarry Smith } 524cddf8d76SBarry Smith } 5250513a670SBarry Smith } else { 5260513a670SBarry Smith /* use contour shading to indicate magnitude of values */ 5270513a670SBarry Smith int count = 0; 5280513a670SBarry Smith for ( i=0; i<m; i++ ) { 5290513a670SBarry Smith y_l = m - i - 1.0; y_r = y_l + 1.0; 5300513a670SBarry Smith for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 5310513a670SBarry Smith x_l = a->j[j] + shift; x_r = x_l + 1.0; 5320513a670SBarry Smith color = 32 + (int) ((200.0 - 32.0)*PetscAbsScalar(a->a[count])/maxv); 5330513a670SBarry Smith DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color); 5340513a670SBarry Smith count++; 5350513a670SBarry Smith } 5360513a670SBarry Smith } 5370513a670SBarry Smith } 5380513a670SBarry Smith 5396945ee14SBarry Smith ierr = DrawCheckResizedWindow(draw); 540cddf8d76SBarry Smith ierr = DrawGetMouseButton(draw,&button,&xc,&yc,0,0); 541cddf8d76SBarry Smith } 542416022c9SBarry Smith return 0; 543416022c9SBarry Smith } 544416022c9SBarry Smith 5455615d1e5SSatish Balay #undef __FUNC__ 5465615d1e5SSatish Balay #define __FUNC__ "MatView_SeqAIJ" 547416022c9SBarry Smith static int MatView_SeqAIJ(PetscObject obj,Viewer viewer) 548416022c9SBarry Smith { 549416022c9SBarry Smith Mat A = (Mat) obj; 550416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*) A->data; 551bcd2baecSBarry Smith ViewerType vtype; 552bcd2baecSBarry Smith int ierr; 553416022c9SBarry Smith 554bcd2baecSBarry Smith ierr = ViewerGetType(viewer,&vtype); CHKERRQ(ierr); 555bcd2baecSBarry Smith if (vtype == MATLAB_VIEWER) { 556416022c9SBarry Smith return ViewerMatlabPutSparse_Private(viewer,a->m,a->n,a->nz,a->a,a->i,a->j); 557416022c9SBarry Smith } 558bcd2baecSBarry Smith else if (vtype == ASCII_FILE_VIEWER || vtype == ASCII_FILES_VIEWER){ 559416022c9SBarry Smith return MatView_SeqAIJ_ASCII(A,viewer); 560416022c9SBarry Smith } 561bcd2baecSBarry Smith else if (vtype == BINARY_FILE_VIEWER) { 562416022c9SBarry Smith return MatView_SeqAIJ_Binary(A,viewer); 563416022c9SBarry Smith } 564bcd2baecSBarry Smith else if (vtype == DRAW_VIEWER) { 565bcd2baecSBarry Smith return MatView_SeqAIJ_Draw(A,viewer); 56617ab2063SBarry Smith } 56717ab2063SBarry Smith return 0; 56817ab2063SBarry Smith } 56919bcc07fSBarry Smith 570c456f294SBarry Smith extern int Mat_AIJ_CheckInode(Mat); 5715615d1e5SSatish Balay #undef __FUNC__ 5725615d1e5SSatish Balay #define __FUNC__ "MatAssemblyEnd_SeqAIJ" 573416022c9SBarry Smith static int MatAssemblyEnd_SeqAIJ(Mat A,MatAssemblyType mode) 57417ab2063SBarry Smith { 575416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 57641c01911SSatish Balay int fshift = 0,i,j,*ai = a->i, *aj = a->j, *imax = a->imax,ierr; 57794a9d846SBarry Smith int m = a->m, *ip, N, *ailen = a->ilen,shift = a->indexshift,rmax; 578416022c9SBarry Smith Scalar *aa = a->a, *ap; 57917ab2063SBarry Smith 5806d4a8577SBarry Smith if (mode == MAT_FLUSH_ASSEMBLY) return 0; 58117ab2063SBarry Smith 58294a9d846SBarry Smith rmax = ailen[0]; /* determine row with most nonzeros */ 58317ab2063SBarry Smith for ( i=1; i<m; i++ ) { 584416022c9SBarry Smith /* move each row back by the amount of empty slots (fshift) before it*/ 58517ab2063SBarry Smith fshift += imax[i-1] - ailen[i-1]; 58694a9d846SBarry Smith rmax = PetscMax(rmax,ailen[i]); 58717ab2063SBarry Smith if (fshift) { 588416022c9SBarry Smith ip = aj + ai[i] + shift; ap = aa + ai[i] + shift; 58917ab2063SBarry Smith N = ailen[i]; 59017ab2063SBarry Smith for ( j=0; j<N; j++ ) { 59117ab2063SBarry Smith ip[j-fshift] = ip[j]; 59217ab2063SBarry Smith ap[j-fshift] = ap[j]; 59317ab2063SBarry Smith } 59417ab2063SBarry Smith } 59517ab2063SBarry Smith ai[i] = ai[i-1] + ailen[i-1]; 59617ab2063SBarry Smith } 59717ab2063SBarry Smith if (m) { 59817ab2063SBarry Smith fshift += imax[m-1] - ailen[m-1]; 59917ab2063SBarry Smith ai[m] = ai[m-1] + ailen[m-1]; 60017ab2063SBarry Smith } 60117ab2063SBarry Smith /* reset ilen and imax for each row */ 60217ab2063SBarry Smith for ( i=0; i<m; i++ ) { 60317ab2063SBarry Smith ailen[i] = imax[i] = ai[i+1] - ai[i]; 60417ab2063SBarry Smith } 605416022c9SBarry Smith a->nz = ai[m] + shift; 60617ab2063SBarry Smith 60717ab2063SBarry Smith /* diagonals may have moved, so kill the diagonal pointers */ 608416022c9SBarry Smith if (fshift && a->diag) { 6090452661fSBarry Smith PetscFree(a->diag); 610416022c9SBarry Smith PLogObjectMemory(A,-(m+1)*sizeof(int)); 611416022c9SBarry Smith a->diag = 0; 61217ab2063SBarry Smith } 6134e220ebcSLois Curfman McInnes PLogInfo(A,"MatAssemblyEnd_SeqAIJ:Matrix size: %d X %d; storage space: %d unneeded, %d used\n", 6144e220ebcSLois Curfman McInnes m,a->n,fshift,a->nz); 6154e220ebcSLois Curfman McInnes PLogInfo(A,"MatAssemblyEnd_SeqAIJ:Number of mallocs during MatSetValues is %d\n", 616b810aeb4SBarry Smith a->reallocs); 61794a9d846SBarry Smith PLogInfo(A,"MatAssemblyEnd_SeqAIJ:Most nonzeros in any row is %d\n",rmax); 618dd5f02e7SSatish Balay a->reallocs = 0; 6194e220ebcSLois Curfman McInnes A->info.nz_unneeded = (double)fshift; 6204e220ebcSLois Curfman McInnes 62176dd722bSSatish Balay /* check out for identical nodes. If found, use inode functions */ 62241c01911SSatish Balay ierr = Mat_AIJ_CheckInode(A); CHKERRQ(ierr); 62317ab2063SBarry Smith return 0; 62417ab2063SBarry Smith } 62517ab2063SBarry Smith 6265615d1e5SSatish Balay #undef __FUNC__ 6275615d1e5SSatish Balay #define __FUNC__ "MatZeroEntries_SeqAIJ" 628416022c9SBarry Smith static int MatZeroEntries_SeqAIJ(Mat A) 62917ab2063SBarry Smith { 630416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 631cddf8d76SBarry Smith PetscMemzero(a->a,(a->i[a->m]+a->indexshift)*sizeof(Scalar)); 63217ab2063SBarry Smith return 0; 63317ab2063SBarry Smith } 634416022c9SBarry Smith 6355615d1e5SSatish Balay #undef __FUNC__ 6365615d1e5SSatish Balay #define __FUNC__ "MatDestroy_SeqAIJ" 63717ab2063SBarry Smith int MatDestroy_SeqAIJ(PetscObject obj) 63817ab2063SBarry Smith { 639416022c9SBarry Smith Mat A = (Mat) obj; 640416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 64190f02eecSBarry Smith int ierr; 642d5d45c9bSBarry Smith 64317ab2063SBarry Smith #if defined(PETSC_LOG) 644416022c9SBarry Smith PLogObjectState(obj,"Rows=%d, Cols=%d, NZ=%d",a->m,a->n,a->nz); 64517ab2063SBarry Smith #endif 6460452661fSBarry Smith PetscFree(a->a); 6470452661fSBarry Smith if (!a->singlemalloc) { PetscFree(a->i); PetscFree(a->j);} 6480452661fSBarry Smith if (a->diag) PetscFree(a->diag); 6490452661fSBarry Smith if (a->ilen) PetscFree(a->ilen); 6500452661fSBarry Smith if (a->imax) PetscFree(a->imax); 6510452661fSBarry Smith if (a->solve_work) PetscFree(a->solve_work); 65276dd722bSSatish Balay if (a->inode.size) PetscFree(a->inode.size); 6530452661fSBarry Smith PetscFree(a); 65490f02eecSBarry Smith if (A->mapping) { 65590f02eecSBarry Smith ierr = ISLocalToGlobalMappingDestroy(A->mapping); CHKERRQ(ierr); 65690f02eecSBarry Smith } 657f2655603SLois Curfman McInnes PLogObjectDestroy(A); 658f2655603SLois Curfman McInnes PetscHeaderDestroy(A); 65917ab2063SBarry Smith return 0; 66017ab2063SBarry Smith } 66117ab2063SBarry Smith 6625615d1e5SSatish Balay #undef __FUNC__ 6635615d1e5SSatish Balay #define __FUNC__ "MatCompress_SeqAIJ" 664416022c9SBarry Smith static int MatCompress_SeqAIJ(Mat A) 66517ab2063SBarry Smith { 66617ab2063SBarry Smith return 0; 66717ab2063SBarry Smith } 66817ab2063SBarry Smith 6695615d1e5SSatish Balay #undef __FUNC__ 6705615d1e5SSatish Balay #define __FUNC__ "MatSetOption_SeqAIJ" 671416022c9SBarry Smith static int MatSetOption_SeqAIJ(Mat A,MatOption op) 67217ab2063SBarry Smith { 673416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 6746d4a8577SBarry Smith if (op == MAT_ROW_ORIENTED) a->roworiented = 1; 6756d4a8577SBarry Smith else if (op == MAT_COLUMN_ORIENTED) a->roworiented = 0; 6766d4a8577SBarry Smith else if (op == MAT_COLUMNS_SORTED) a->sorted = 1; 677219d9a1aSLois Curfman McInnes else if (op == MAT_COLUMNS_UNSORTED) a->sorted = 0; 6786d4a8577SBarry Smith else if (op == MAT_NO_NEW_NONZERO_LOCATIONS) a->nonew = 1; 6796d4a8577SBarry Smith else if (op == MAT_YES_NEW_NONZERO_LOCATIONS) a->nonew = 0; 6806d4a8577SBarry Smith else if (op == MAT_ROWS_SORTED || 681219d9a1aSLois Curfman McInnes op == MAT_ROWS_UNSORTED || 6826d4a8577SBarry Smith op == MAT_SYMMETRIC || 6836d4a8577SBarry Smith op == MAT_STRUCTURALLY_SYMMETRIC || 68490f02eecSBarry Smith op == MAT_YES_NEW_DIAGONALS || 68590f02eecSBarry Smith op == MAT_IGNORE_OFF_PROCESSOR_ENTRIES) 68694a424c1SBarry Smith PLogInfo(A,"Info:MatSetOption_SeqAIJ:Option ignored\n"); 6876d4a8577SBarry Smith else if (op == MAT_NO_NEW_DIAGONALS) 688e3372554SBarry Smith {SETERRQ(PETSC_ERR_SUP,0,"MAT_NO_NEW_DIAGONALS");} 6896d4a8577SBarry Smith else if (op == MAT_INODE_LIMIT_1) a->inode.limit = 1; 6906d4a8577SBarry Smith else if (op == MAT_INODE_LIMIT_2) a->inode.limit = 2; 6916d4a8577SBarry Smith else if (op == MAT_INODE_LIMIT_3) a->inode.limit = 3; 6926d4a8577SBarry Smith else if (op == MAT_INODE_LIMIT_4) a->inode.limit = 4; 6936d4a8577SBarry Smith else if (op == MAT_INODE_LIMIT_5) a->inode.limit = 5; 694e2f28af5SBarry Smith else 695e3372554SBarry Smith {SETERRQ(PETSC_ERR_SUP,0,"unknown option");} 69617ab2063SBarry Smith return 0; 69717ab2063SBarry Smith } 69817ab2063SBarry Smith 6995615d1e5SSatish Balay #undef __FUNC__ 7005615d1e5SSatish Balay #define __FUNC__ "MatGetDiagonal_SeqAIJ" 701416022c9SBarry Smith static int MatGetDiagonal_SeqAIJ(Mat A,Vec v) 70217ab2063SBarry Smith { 703416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 704416022c9SBarry Smith int i,j, n,shift = a->indexshift; 70517ab2063SBarry Smith Scalar *x, zero = 0.0; 70617ab2063SBarry Smith 70717ab2063SBarry Smith VecSet(&zero,v); 70890f02eecSBarry Smith VecGetArray_Fast(v,x); VecGetLocalSize(v,&n); 709e3372554SBarry Smith if (n != a->m) SETERRQ(1,0,"Nonconforming matrix and vector"); 710416022c9SBarry Smith for ( i=0; i<a->m; i++ ) { 711416022c9SBarry Smith for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 712416022c9SBarry Smith if (a->j[j]+shift == i) { 713416022c9SBarry Smith x[i] = a->a[j]; 71417ab2063SBarry Smith break; 71517ab2063SBarry Smith } 71617ab2063SBarry Smith } 71717ab2063SBarry Smith } 71817ab2063SBarry Smith return 0; 71917ab2063SBarry Smith } 72017ab2063SBarry Smith 72117ab2063SBarry Smith /* -------------------------------------------------------*/ 72217ab2063SBarry Smith /* Should check that shapes of vectors and matrices match */ 72317ab2063SBarry Smith /* -------------------------------------------------------*/ 7245615d1e5SSatish Balay #undef __FUNC__ 7255615d1e5SSatish Balay #define __FUNC__ "MatMultTrans_SeqAIJ" 72644cd7ae7SLois Curfman McInnes int MatMultTrans_SeqAIJ(Mat A,Vec xx,Vec yy) 72717ab2063SBarry Smith { 728416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 72917ab2063SBarry Smith Scalar *x, *y, *v, alpha; 730416022c9SBarry Smith int m = a->m, n, i, *idx, shift = a->indexshift; 73117ab2063SBarry Smith 73290f02eecSBarry Smith VecGetArray_Fast(xx,x); VecGetArray_Fast(yy,y); 733cddf8d76SBarry Smith PetscMemzero(y,a->n*sizeof(Scalar)); 73417ab2063SBarry Smith y = y + shift; /* shift for Fortran start by 1 indexing */ 73517ab2063SBarry Smith for ( i=0; i<m; i++ ) { 736416022c9SBarry Smith idx = a->j + a->i[i] + shift; 737416022c9SBarry Smith v = a->a + a->i[i] + shift; 738416022c9SBarry Smith n = a->i[i+1] - a->i[i]; 73917ab2063SBarry Smith alpha = x[i]; 74017ab2063SBarry Smith while (n-->0) {y[*idx++] += alpha * *v++;} 74117ab2063SBarry Smith } 742416022c9SBarry Smith PLogFlops(2*a->nz - a->n); 74317ab2063SBarry Smith return 0; 74417ab2063SBarry Smith } 745d5d45c9bSBarry Smith 7465615d1e5SSatish Balay #undef __FUNC__ 7475615d1e5SSatish Balay #define __FUNC__ "MatMultTransAdd_SeqAIJ" 74844cd7ae7SLois Curfman McInnes int MatMultTransAdd_SeqAIJ(Mat A,Vec xx,Vec zz,Vec yy) 74917ab2063SBarry Smith { 750416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 75117ab2063SBarry Smith Scalar *x, *y, *v, alpha; 752416022c9SBarry Smith int m = a->m, n, i, *idx,shift = a->indexshift; 75317ab2063SBarry Smith 75490f02eecSBarry Smith VecGetArray_Fast(xx,x); VecGetArray_Fast(yy,y); 75517ab2063SBarry Smith if (zz != yy) VecCopy(zz,yy); 75617ab2063SBarry Smith y = y + shift; /* shift for Fortran start by 1 indexing */ 75717ab2063SBarry Smith for ( i=0; i<m; i++ ) { 758416022c9SBarry Smith idx = a->j + a->i[i] + shift; 759416022c9SBarry Smith v = a->a + a->i[i] + shift; 760416022c9SBarry Smith n = a->i[i+1] - a->i[i]; 76117ab2063SBarry Smith alpha = x[i]; 76217ab2063SBarry Smith while (n-->0) {y[*idx++] += alpha * *v++;} 76317ab2063SBarry Smith } 76490f02eecSBarry Smith PLogFlops(2*a->nz); 76517ab2063SBarry Smith return 0; 76617ab2063SBarry Smith } 76717ab2063SBarry Smith 7685615d1e5SSatish Balay #undef __FUNC__ 7695615d1e5SSatish Balay #define __FUNC__ "MatMult_SeqAIJ" 77044cd7ae7SLois Curfman McInnes int MatMult_SeqAIJ(Mat A,Vec xx,Vec yy) 77117ab2063SBarry Smith { 772416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 77317ab2063SBarry Smith Scalar *x, *y, *v, sum; 774416022c9SBarry Smith int m = a->m, n, i, *idx, shift = a->indexshift,*ii; 77517ab2063SBarry Smith 77690f02eecSBarry Smith VecGetArray_Fast(xx,x); VecGetArray_Fast(yy,y); 77717ab2063SBarry Smith x = x + shift; /* shift for Fortran start by 1 indexing */ 778416022c9SBarry Smith idx = a->j; 779416022c9SBarry Smith v = a->a; 780416022c9SBarry Smith ii = a->i; 78117ab2063SBarry Smith for ( i=0; i<m; i++ ) { 782416022c9SBarry Smith n = ii[1] - ii[0]; ii++; 78317ab2063SBarry Smith sum = 0.0; 78417ab2063SBarry Smith /* SPARSEDENSEDOT(sum,x,v,idx,n); */ 78517ab2063SBarry Smith /* for ( j=n-1; j>-1; j--) sum += v[j]*x[idx[j]]; */ 786416022c9SBarry Smith while (n--) sum += *v++ * x[*idx++]; 78717ab2063SBarry Smith y[i] = sum; 78817ab2063SBarry Smith } 789416022c9SBarry Smith PLogFlops(2*a->nz - m); 79017ab2063SBarry Smith return 0; 79117ab2063SBarry Smith } 79217ab2063SBarry Smith 7935615d1e5SSatish Balay #undef __FUNC__ 7945615d1e5SSatish Balay #define __FUNC__ "MatMultAdd_SeqAIJ" 79544cd7ae7SLois Curfman McInnes int MatMultAdd_SeqAIJ(Mat A,Vec xx,Vec yy,Vec zz) 79617ab2063SBarry Smith { 797416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 79817ab2063SBarry Smith Scalar *x, *y, *z, *v, sum; 799cddf8d76SBarry Smith int m = a->m, n, i, *idx, shift = a->indexshift,*ii; 80017ab2063SBarry Smith 80190f02eecSBarry Smith VecGetArray_Fast(xx,x); VecGetArray_Fast(yy,y); VecGetArray_Fast(zz,z); 80217ab2063SBarry Smith x = x + shift; /* shift for Fortran start by 1 indexing */ 803cddf8d76SBarry Smith idx = a->j; 804cddf8d76SBarry Smith v = a->a; 805cddf8d76SBarry Smith ii = a->i; 80617ab2063SBarry Smith for ( i=0; i<m; i++ ) { 807cddf8d76SBarry Smith n = ii[1] - ii[0]; ii++; 80817ab2063SBarry Smith sum = y[i]; 809cddf8d76SBarry Smith /* SPARSEDENSEDOT(sum,x,v,idx,n); */ 810cddf8d76SBarry Smith while (n--) sum += *v++ * x[*idx++]; 81117ab2063SBarry Smith z[i] = sum; 81217ab2063SBarry Smith } 813416022c9SBarry Smith PLogFlops(2*a->nz); 81417ab2063SBarry Smith return 0; 81517ab2063SBarry Smith } 81617ab2063SBarry Smith 81717ab2063SBarry Smith /* 81817ab2063SBarry Smith Adds diagonal pointers to sparse matrix structure. 81917ab2063SBarry Smith */ 82017ab2063SBarry Smith 8215615d1e5SSatish Balay #undef __FUNC__ 8225615d1e5SSatish Balay #define __FUNC__ "MatMarkDiag_SeqAIJ" 823416022c9SBarry Smith int MatMarkDiag_SeqAIJ(Mat A) 82417ab2063SBarry Smith { 825416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 826416022c9SBarry Smith int i,j, *diag, m = a->m,shift = a->indexshift; 82717ab2063SBarry Smith 8280452661fSBarry Smith diag = (int *) PetscMalloc( (m+1)*sizeof(int)); CHKPTRQ(diag); 829416022c9SBarry Smith PLogObjectMemory(A,(m+1)*sizeof(int)); 830416022c9SBarry Smith for ( i=0; i<a->m; i++ ) { 831416022c9SBarry Smith for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 832416022c9SBarry Smith if (a->j[j]+shift == i) { 83317ab2063SBarry Smith diag[i] = j - shift; 83417ab2063SBarry Smith break; 83517ab2063SBarry Smith } 83617ab2063SBarry Smith } 83717ab2063SBarry Smith } 838416022c9SBarry Smith a->diag = diag; 83917ab2063SBarry Smith return 0; 84017ab2063SBarry Smith } 84117ab2063SBarry Smith 8425615d1e5SSatish Balay #undef __FUNC__ 8435615d1e5SSatish Balay #define __FUNC__ "MatRelax_SeqAIJ" 84444cd7ae7SLois Curfman McInnes int MatRelax_SeqAIJ(Mat A,Vec bb,double omega,MatSORType flag, 84517ab2063SBarry Smith double fshift,int its,Vec xx) 84617ab2063SBarry Smith { 847416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 848416022c9SBarry Smith Scalar *x, *b, *bs, d, *xs, sum, *v = a->a,*t,scale,*ts, *xb; 849d5d45c9bSBarry Smith int ierr, *idx, *diag,n = a->n, m = a->m, i, shift = a->indexshift; 85017ab2063SBarry Smith 85190f02eecSBarry Smith VecGetArray_Fast(xx,x); VecGetArray_Fast(bb,b); 852416022c9SBarry Smith if (!a->diag) {if ((ierr = MatMarkDiag_SeqAIJ(A))) return ierr;} 853416022c9SBarry Smith diag = a->diag; 854416022c9SBarry Smith xs = x + shift; /* shifted by one for index start of a or a->j*/ 85517ab2063SBarry Smith if (flag == SOR_APPLY_UPPER) { 85617ab2063SBarry Smith /* apply ( U + D/omega) to the vector */ 85717ab2063SBarry Smith bs = b + shift; 85817ab2063SBarry Smith for ( i=0; i<m; i++ ) { 859416022c9SBarry Smith d = fshift + a->a[diag[i] + shift]; 860416022c9SBarry Smith n = a->i[i+1] - diag[i] - 1; 861416022c9SBarry Smith idx = a->j + diag[i] + (!shift); 862416022c9SBarry Smith v = a->a + diag[i] + (!shift); 86317ab2063SBarry Smith sum = b[i]*d/omega; 86417ab2063SBarry Smith SPARSEDENSEDOT(sum,bs,v,idx,n); 86517ab2063SBarry Smith x[i] = sum; 86617ab2063SBarry Smith } 86717ab2063SBarry Smith return 0; 86817ab2063SBarry Smith } 86917ab2063SBarry Smith if (flag == SOR_APPLY_LOWER) { 870e3372554SBarry Smith SETERRQ(1,0,"SOR_APPLY_LOWER is not done"); 87117ab2063SBarry Smith } 872416022c9SBarry Smith else if (flag & SOR_EISENSTAT) { 87317ab2063SBarry Smith /* Let A = L + U + D; where L is lower trianglar, 87417ab2063SBarry Smith U is upper triangular, E is diagonal; This routine applies 87517ab2063SBarry Smith 87617ab2063SBarry Smith (L + E)^{-1} A (U + E)^{-1} 87717ab2063SBarry Smith 87817ab2063SBarry Smith to a vector efficiently using Eisenstat's trick. This is for 87917ab2063SBarry Smith the case of SSOR preconditioner, so E is D/omega where omega 88017ab2063SBarry Smith is the relaxation factor. 88117ab2063SBarry Smith */ 8820452661fSBarry Smith t = (Scalar *) PetscMalloc( m*sizeof(Scalar) ); CHKPTRQ(t); 88317ab2063SBarry Smith scale = (2.0/omega) - 1.0; 88417ab2063SBarry Smith 88517ab2063SBarry Smith /* x = (E + U)^{-1} b */ 88617ab2063SBarry Smith for ( i=m-1; i>=0; i-- ) { 887416022c9SBarry Smith d = fshift + a->a[diag[i] + shift]; 888416022c9SBarry Smith n = a->i[i+1] - diag[i] - 1; 889416022c9SBarry Smith idx = a->j + diag[i] + (!shift); 890416022c9SBarry Smith v = a->a + diag[i] + (!shift); 89117ab2063SBarry Smith sum = b[i]; 89217ab2063SBarry Smith SPARSEDENSEMDOT(sum,xs,v,idx,n); 89317ab2063SBarry Smith x[i] = omega*(sum/d); 89417ab2063SBarry Smith } 89517ab2063SBarry Smith 89617ab2063SBarry Smith /* t = b - (2*E - D)x */ 897416022c9SBarry Smith v = a->a; 89817ab2063SBarry Smith for ( i=0; i<m; i++ ) { t[i] = b[i] - scale*(v[*diag++ + shift])*x[i]; } 89917ab2063SBarry Smith 90017ab2063SBarry Smith /* t = (E + L)^{-1}t */ 901416022c9SBarry Smith ts = t + shift; /* shifted by one for index start of a or a->j*/ 902416022c9SBarry Smith diag = a->diag; 90317ab2063SBarry Smith for ( i=0; i<m; i++ ) { 904416022c9SBarry Smith d = fshift + a->a[diag[i]+shift]; 905416022c9SBarry Smith n = diag[i] - a->i[i]; 906416022c9SBarry Smith idx = a->j + a->i[i] + shift; 907416022c9SBarry Smith v = a->a + a->i[i] + shift; 90817ab2063SBarry Smith sum = t[i]; 90917ab2063SBarry Smith SPARSEDENSEMDOT(sum,ts,v,idx,n); 91017ab2063SBarry Smith t[i] = omega*(sum/d); 91117ab2063SBarry Smith } 91217ab2063SBarry Smith 91317ab2063SBarry Smith /* x = x + t */ 91417ab2063SBarry Smith for ( i=0; i<m; i++ ) { x[i] += t[i]; } 9150452661fSBarry Smith PetscFree(t); 91617ab2063SBarry Smith return 0; 91717ab2063SBarry Smith } 91817ab2063SBarry Smith if (flag & SOR_ZERO_INITIAL_GUESS) { 91917ab2063SBarry Smith if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP){ 92017ab2063SBarry Smith for ( i=0; i<m; i++ ) { 921416022c9SBarry Smith d = fshift + a->a[diag[i]+shift]; 922416022c9SBarry Smith n = diag[i] - a->i[i]; 923416022c9SBarry Smith idx = a->j + a->i[i] + shift; 924416022c9SBarry Smith v = a->a + a->i[i] + shift; 92517ab2063SBarry Smith sum = b[i]; 92617ab2063SBarry Smith SPARSEDENSEMDOT(sum,xs,v,idx,n); 92717ab2063SBarry Smith x[i] = omega*(sum/d); 92817ab2063SBarry Smith } 92917ab2063SBarry Smith xb = x; 93017ab2063SBarry Smith } 93117ab2063SBarry Smith else xb = b; 93217ab2063SBarry Smith if ((flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP) && 93317ab2063SBarry Smith (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP)) { 93417ab2063SBarry Smith for ( i=0; i<m; i++ ) { 935416022c9SBarry Smith x[i] *= a->a[diag[i]+shift]; 93617ab2063SBarry Smith } 93717ab2063SBarry Smith } 93817ab2063SBarry Smith if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP){ 93917ab2063SBarry Smith for ( i=m-1; i>=0; i-- ) { 940416022c9SBarry Smith d = fshift + a->a[diag[i] + shift]; 941416022c9SBarry Smith n = a->i[i+1] - diag[i] - 1; 942416022c9SBarry Smith idx = a->j + diag[i] + (!shift); 943416022c9SBarry Smith v = a->a + diag[i] + (!shift); 94417ab2063SBarry Smith sum = xb[i]; 94517ab2063SBarry Smith SPARSEDENSEMDOT(sum,xs,v,idx,n); 94617ab2063SBarry Smith x[i] = omega*(sum/d); 94717ab2063SBarry Smith } 94817ab2063SBarry Smith } 94917ab2063SBarry Smith its--; 95017ab2063SBarry Smith } 95117ab2063SBarry Smith while (its--) { 95217ab2063SBarry Smith if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP){ 95317ab2063SBarry Smith for ( i=0; i<m; i++ ) { 954416022c9SBarry Smith d = fshift + a->a[diag[i]+shift]; 955416022c9SBarry Smith n = a->i[i+1] - a->i[i]; 956416022c9SBarry Smith idx = a->j + a->i[i] + shift; 957416022c9SBarry Smith v = a->a + a->i[i] + shift; 95817ab2063SBarry Smith sum = b[i]; 95917ab2063SBarry Smith SPARSEDENSEMDOT(sum,xs,v,idx,n); 9607e33a6baSBarry Smith x[i] = (1. - omega)*x[i] + omega*(sum + a->a[diag[i]+shift]*x[i])/d; 96117ab2063SBarry Smith } 96217ab2063SBarry Smith } 96317ab2063SBarry Smith if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP){ 96417ab2063SBarry Smith for ( i=m-1; i>=0; i-- ) { 965416022c9SBarry Smith d = fshift + a->a[diag[i] + shift]; 966416022c9SBarry Smith n = a->i[i+1] - a->i[i]; 967416022c9SBarry Smith idx = a->j + a->i[i] + shift; 968416022c9SBarry Smith v = a->a + a->i[i] + shift; 96917ab2063SBarry Smith sum = b[i]; 97017ab2063SBarry Smith SPARSEDENSEMDOT(sum,xs,v,idx,n); 9717e33a6baSBarry Smith x[i] = (1. - omega)*x[i] + omega*(sum + a->a[diag[i]+shift]*x[i])/d; 97217ab2063SBarry Smith } 97317ab2063SBarry Smith } 97417ab2063SBarry Smith } 97517ab2063SBarry Smith return 0; 97617ab2063SBarry Smith } 97717ab2063SBarry Smith 9785615d1e5SSatish Balay #undef __FUNC__ 9795615d1e5SSatish Balay #define __FUNC__ "MatGetInfo_SeqAIJ" 9804e220ebcSLois Curfman McInnes static int MatGetInfo_SeqAIJ(Mat A,MatInfoType flag,MatInfo *info) 98117ab2063SBarry Smith { 982416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 9834e220ebcSLois Curfman McInnes 9844e220ebcSLois Curfman McInnes info->rows_global = (double)a->m; 9854e220ebcSLois Curfman McInnes info->columns_global = (double)a->n; 9864e220ebcSLois Curfman McInnes info->rows_local = (double)a->m; 9874e220ebcSLois Curfman McInnes info->columns_local = (double)a->n; 9884e220ebcSLois Curfman McInnes info->block_size = 1.0; 9894e220ebcSLois Curfman McInnes info->nz_allocated = (double)a->maxnz; 9904e220ebcSLois Curfman McInnes info->nz_used = (double)a->nz; 9914e220ebcSLois Curfman McInnes info->nz_unneeded = (double)(a->maxnz - a->nz); 9924e220ebcSLois Curfman McInnes /* if (info->nz_unneeded != A->info.nz_unneeded) 9934e220ebcSLois Curfman McInnes printf("space descrepancy: maxnz-nz = %d, nz_unneeded = %d\n",(int)info->nz_unneeded,(int)A->info.nz_unneeded); */ 9944e220ebcSLois Curfman McInnes info->assemblies = (double)A->num_ass; 9954e220ebcSLois Curfman McInnes info->mallocs = (double)a->reallocs; 9964e220ebcSLois Curfman McInnes info->memory = A->mem; 9974e220ebcSLois Curfman McInnes if (A->factor) { 9984e220ebcSLois Curfman McInnes info->fill_ratio_given = A->info.fill_ratio_given; 9994e220ebcSLois Curfman McInnes info->fill_ratio_needed = A->info.fill_ratio_needed; 10004e220ebcSLois Curfman McInnes info->factor_mallocs = A->info.factor_mallocs; 10014e220ebcSLois Curfman McInnes } else { 10024e220ebcSLois Curfman McInnes info->fill_ratio_given = 0; 10034e220ebcSLois Curfman McInnes info->fill_ratio_needed = 0; 10044e220ebcSLois Curfman McInnes info->factor_mallocs = 0; 10054e220ebcSLois Curfman McInnes } 100617ab2063SBarry Smith return 0; 100717ab2063SBarry Smith } 100817ab2063SBarry Smith 100917ab2063SBarry Smith extern int MatLUFactorSymbolic_SeqAIJ(Mat,IS,IS,double,Mat*); 101017ab2063SBarry Smith extern int MatLUFactorNumeric_SeqAIJ(Mat,Mat*); 101117ab2063SBarry Smith extern int MatLUFactor_SeqAIJ(Mat,IS,IS,double); 101217ab2063SBarry Smith extern int MatSolve_SeqAIJ(Mat,Vec,Vec); 101317ab2063SBarry Smith extern int MatSolveAdd_SeqAIJ(Mat,Vec,Vec,Vec); 101417ab2063SBarry Smith extern int MatSolveTrans_SeqAIJ(Mat,Vec,Vec); 101517ab2063SBarry Smith extern int MatSolveTransAdd_SeqAIJ(Mat,Vec,Vec,Vec); 101617ab2063SBarry Smith 10175615d1e5SSatish Balay #undef __FUNC__ 10185615d1e5SSatish Balay #define __FUNC__ "MatZeroRows_SeqAIJ" 101917ab2063SBarry Smith static int MatZeroRows_SeqAIJ(Mat A,IS is,Scalar *diag) 102017ab2063SBarry Smith { 1021416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 1022416022c9SBarry Smith int i,ierr,N, *rows,m = a->m - 1,shift = a->indexshift; 102317ab2063SBarry Smith 102477c4ece6SBarry Smith ierr = ISGetSize(is,&N); CHKERRQ(ierr); 102517ab2063SBarry Smith ierr = ISGetIndices(is,&rows); CHKERRQ(ierr); 102617ab2063SBarry Smith if (diag) { 102717ab2063SBarry Smith for ( i=0; i<N; i++ ) { 1028e3372554SBarry Smith if (rows[i] < 0 || rows[i] > m) SETERRQ(1,0,"row out of range"); 1029416022c9SBarry Smith if (a->ilen[rows[i]] > 0) { /* in case row was completely empty */ 1030416022c9SBarry Smith a->ilen[rows[i]] = 1; 1031416022c9SBarry Smith a->a[a->i[rows[i]]+shift] = *diag; 1032416022c9SBarry Smith a->j[a->i[rows[i]]+shift] = rows[i]+shift; 103317ab2063SBarry Smith } 103417ab2063SBarry Smith else { 103517ab2063SBarry Smith ierr = MatSetValues_SeqAIJ(A,1,&rows[i],1,&rows[i],diag,INSERT_VALUES); 103617ab2063SBarry Smith CHKERRQ(ierr); 103717ab2063SBarry Smith } 103817ab2063SBarry Smith } 103917ab2063SBarry Smith } 104017ab2063SBarry Smith else { 104117ab2063SBarry Smith for ( i=0; i<N; i++ ) { 1042e3372554SBarry Smith if (rows[i] < 0 || rows[i] > m) SETERRQ(1,0,"row out of range"); 1043416022c9SBarry Smith a->ilen[rows[i]] = 0; 104417ab2063SBarry Smith } 104517ab2063SBarry Smith } 104617ab2063SBarry Smith ISRestoreIndices(is,&rows); 104743a90d84SBarry Smith ierr = MatAssemblyEnd_SeqAIJ(A,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr); 104817ab2063SBarry Smith return 0; 104917ab2063SBarry Smith } 105017ab2063SBarry Smith 10515615d1e5SSatish Balay #undef __FUNC__ 10525615d1e5SSatish Balay #define __FUNC__ "MatGetSize_SeqAIJ" 1053416022c9SBarry Smith static int MatGetSize_SeqAIJ(Mat A,int *m,int *n) 105417ab2063SBarry Smith { 1055416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 1056416022c9SBarry Smith *m = a->m; *n = a->n; 105717ab2063SBarry Smith return 0; 105817ab2063SBarry Smith } 105917ab2063SBarry Smith 10605615d1e5SSatish Balay #undef __FUNC__ 10615615d1e5SSatish Balay #define __FUNC__ "MatGetOwnershipRange_SeqAIJ" 1062416022c9SBarry Smith static int MatGetOwnershipRange_SeqAIJ(Mat A,int *m,int *n) 106317ab2063SBarry Smith { 1064416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 1065416022c9SBarry Smith *m = 0; *n = a->m; 106617ab2063SBarry Smith return 0; 106717ab2063SBarry Smith } 1068026e39d0SSatish Balay 10695615d1e5SSatish Balay #undef __FUNC__ 10705615d1e5SSatish Balay #define __FUNC__ "MatGetRow_SeqAIJ" 10714e093b46SBarry Smith int MatGetRow_SeqAIJ(Mat A,int row,int *nz,int **idx,Scalar **v) 107217ab2063SBarry Smith { 1073416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 1074c456f294SBarry Smith int *itmp,i,shift = a->indexshift; 107517ab2063SBarry Smith 1076e3372554SBarry Smith if (row < 0 || row >= a->m) SETERRQ(1,0,"Row out of range"); 107717ab2063SBarry Smith 1078416022c9SBarry Smith *nz = a->i[row+1] - a->i[row]; 1079416022c9SBarry Smith if (v) *v = a->a + a->i[row] + shift; 108017ab2063SBarry Smith if (idx) { 1081416022c9SBarry Smith itmp = a->j + a->i[row] + shift; 10824e093b46SBarry Smith if (*nz && shift) { 10830452661fSBarry Smith *idx = (int *) PetscMalloc( (*nz)*sizeof(int) ); CHKPTRQ(*idx); 108417ab2063SBarry Smith for ( i=0; i<(*nz); i++ ) {(*idx)[i] = itmp[i] + shift;} 10854e093b46SBarry Smith } else if (*nz) { 10864e093b46SBarry Smith *idx = itmp; 108717ab2063SBarry Smith } 108817ab2063SBarry Smith else *idx = 0; 108917ab2063SBarry Smith } 109017ab2063SBarry Smith return 0; 109117ab2063SBarry Smith } 109217ab2063SBarry Smith 10935615d1e5SSatish Balay #undef __FUNC__ 10945615d1e5SSatish Balay #define __FUNC__ "MatRestoreRow_SeqAIJ" 10954e093b46SBarry Smith int MatRestoreRow_SeqAIJ(Mat A,int row,int *nz,int **idx,Scalar **v) 109617ab2063SBarry Smith { 10974e093b46SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 10984e093b46SBarry Smith if (idx) {if (*idx && a->indexshift) PetscFree(*idx);} 109917ab2063SBarry Smith return 0; 110017ab2063SBarry Smith } 110117ab2063SBarry Smith 11025615d1e5SSatish Balay #undef __FUNC__ 11035615d1e5SSatish Balay #define __FUNC__ "MatNorm_SeqAIJ" 1104cddf8d76SBarry Smith static int MatNorm_SeqAIJ(Mat A,NormType type,double *norm) 110517ab2063SBarry Smith { 1106416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 1107416022c9SBarry Smith Scalar *v = a->a; 110817ab2063SBarry Smith double sum = 0.0; 1109416022c9SBarry Smith int i, j,shift = a->indexshift; 111017ab2063SBarry Smith 111117ab2063SBarry Smith if (type == NORM_FROBENIUS) { 1112416022c9SBarry Smith for (i=0; i<a->nz; i++ ) { 111317ab2063SBarry Smith #if defined(PETSC_COMPLEX) 111417ab2063SBarry Smith sum += real(conj(*v)*(*v)); v++; 111517ab2063SBarry Smith #else 111617ab2063SBarry Smith sum += (*v)*(*v); v++; 111717ab2063SBarry Smith #endif 111817ab2063SBarry Smith } 111917ab2063SBarry Smith *norm = sqrt(sum); 112017ab2063SBarry Smith } 112117ab2063SBarry Smith else if (type == NORM_1) { 112217ab2063SBarry Smith double *tmp; 1123416022c9SBarry Smith int *jj = a->j; 11240452661fSBarry Smith tmp = (double *) PetscMalloc( a->n*sizeof(double) ); CHKPTRQ(tmp); 1125cddf8d76SBarry Smith PetscMemzero(tmp,a->n*sizeof(double)); 112617ab2063SBarry Smith *norm = 0.0; 1127416022c9SBarry Smith for ( j=0; j<a->nz; j++ ) { 1128a2744918SBarry Smith tmp[*jj++ + shift] += PetscAbsScalar(*v); v++; 112917ab2063SBarry Smith } 1130416022c9SBarry Smith for ( j=0; j<a->n; j++ ) { 113117ab2063SBarry Smith if (tmp[j] > *norm) *norm = tmp[j]; 113217ab2063SBarry Smith } 11330452661fSBarry Smith PetscFree(tmp); 113417ab2063SBarry Smith } 113517ab2063SBarry Smith else if (type == NORM_INFINITY) { 113617ab2063SBarry Smith *norm = 0.0; 1137416022c9SBarry Smith for ( j=0; j<a->m; j++ ) { 1138416022c9SBarry Smith v = a->a + a->i[j] + shift; 113917ab2063SBarry Smith sum = 0.0; 1140416022c9SBarry Smith for ( i=0; i<a->i[j+1]-a->i[j]; i++ ) { 1141cddf8d76SBarry Smith sum += PetscAbsScalar(*v); v++; 114217ab2063SBarry Smith } 114317ab2063SBarry Smith if (sum > *norm) *norm = sum; 114417ab2063SBarry Smith } 114517ab2063SBarry Smith } 114617ab2063SBarry Smith else { 1147e3372554SBarry Smith SETERRQ(1,0,"No support for two norm yet"); 114817ab2063SBarry Smith } 114917ab2063SBarry Smith return 0; 115017ab2063SBarry Smith } 115117ab2063SBarry Smith 11525615d1e5SSatish Balay #undef __FUNC__ 11535615d1e5SSatish Balay #define __FUNC__ "MatTranspose_SeqAIJ" 1154416022c9SBarry Smith static int MatTranspose_SeqAIJ(Mat A,Mat *B) 115517ab2063SBarry Smith { 1156416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 1157416022c9SBarry Smith Mat C; 1158416022c9SBarry Smith int i, ierr, *aj = a->j, *ai = a->i, m = a->m, len, *col; 1159416022c9SBarry Smith int shift = a->indexshift; 1160d5d45c9bSBarry Smith Scalar *array = a->a; 116117ab2063SBarry Smith 11623638b69dSLois Curfman McInnes if (B == PETSC_NULL && m != a->n) 1163e3372554SBarry Smith SETERRQ(1,0,"Square matrix only for in-place"); 11640452661fSBarry Smith col = (int *) PetscMalloc((1+a->n)*sizeof(int)); CHKPTRQ(col); 1165cddf8d76SBarry Smith PetscMemzero(col,(1+a->n)*sizeof(int)); 116617ab2063SBarry Smith if (shift) { 116717ab2063SBarry Smith for ( i=0; i<ai[m]-1; i++ ) aj[i] -= 1; 116817ab2063SBarry Smith } 116917ab2063SBarry Smith for ( i=0; i<ai[m]+shift; i++ ) col[aj[i]] += 1; 1170416022c9SBarry Smith ierr = MatCreateSeqAIJ(A->comm,a->n,m,0,col,&C); CHKERRQ(ierr); 11710452661fSBarry Smith PetscFree(col); 117217ab2063SBarry Smith for ( i=0; i<m; i++ ) { 117317ab2063SBarry Smith len = ai[i+1]-ai[i]; 1174416022c9SBarry Smith ierr = MatSetValues(C,len,aj,1,&i,array,INSERT_VALUES); CHKERRQ(ierr); 117517ab2063SBarry Smith array += len; aj += len; 117617ab2063SBarry Smith } 117717ab2063SBarry Smith if (shift) { 117817ab2063SBarry Smith for ( i=0; i<ai[m]-1; i++ ) aj[i] += 1; 117917ab2063SBarry Smith } 118017ab2063SBarry Smith 11816d4a8577SBarry Smith ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr); 11826d4a8577SBarry Smith ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr); 118317ab2063SBarry Smith 11843638b69dSLois Curfman McInnes if (B != PETSC_NULL) { 1185416022c9SBarry Smith *B = C; 118617ab2063SBarry Smith } else { 1187416022c9SBarry Smith /* This isn't really an in-place transpose */ 11880452661fSBarry Smith PetscFree(a->a); 11890452661fSBarry Smith if (!a->singlemalloc) {PetscFree(a->i); PetscFree(a->j);} 11900452661fSBarry Smith if (a->diag) PetscFree(a->diag); 11910452661fSBarry Smith if (a->ilen) PetscFree(a->ilen); 11920452661fSBarry Smith if (a->imax) PetscFree(a->imax); 11930452661fSBarry Smith if (a->solve_work) PetscFree(a->solve_work); 11941073c447SSatish Balay if (a->inode.size) PetscFree(a->inode.size); 11950452661fSBarry Smith PetscFree(a); 1196416022c9SBarry Smith PetscMemcpy(A,C,sizeof(struct _Mat)); 11970452661fSBarry Smith PetscHeaderDestroy(C); 119817ab2063SBarry Smith } 119917ab2063SBarry Smith return 0; 120017ab2063SBarry Smith } 120117ab2063SBarry Smith 12025615d1e5SSatish Balay #undef __FUNC__ 12035615d1e5SSatish Balay #define __FUNC__ "MatDiagonalScale_SeqAIJ" 1204f0b747eeSBarry Smith static int MatDiagonalScale_SeqAIJ(Mat A,Vec ll,Vec rr) 120517ab2063SBarry Smith { 1206416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 120717ab2063SBarry Smith Scalar *l,*r,x,*v; 1208d5d45c9bSBarry Smith int i,j,m = a->m, n = a->n, M, nz = a->nz, *jj,shift = a->indexshift; 120917ab2063SBarry Smith 121017ab2063SBarry Smith if (ll) { 12113ea7c6a1SSatish Balay /* The local size is used so that VecMPI can be passed to this routine 12123ea7c6a1SSatish Balay by MatDiagonalScale_MPIAIJ */ 12139b1297e1SSatish Balay VecGetLocalSize_Fast(ll,m); 1214e3372554SBarry Smith if (m != a->m) SETERRQ(1,0,"Left scaling vector wrong length"); 121590f02eecSBarry Smith VecGetArray_Fast(ll,l); 1216416022c9SBarry Smith v = a->a; 121717ab2063SBarry Smith for ( i=0; i<m; i++ ) { 121817ab2063SBarry Smith x = l[i]; 1219416022c9SBarry Smith M = a->i[i+1] - a->i[i]; 122017ab2063SBarry Smith for ( j=0; j<M; j++ ) { (*v++) *= x;} 122117ab2063SBarry Smith } 122244cd7ae7SLois Curfman McInnes PLogFlops(nz); 122317ab2063SBarry Smith } 122417ab2063SBarry Smith if (rr) { 12259b1297e1SSatish Balay VecGetLocalSize_Fast(rr,n); 1226e3372554SBarry Smith if (n != a->n) SETERRQ(1,0,"Right scaling vector wrong length"); 122790f02eecSBarry Smith VecGetArray_Fast(rr,r); 1228416022c9SBarry Smith v = a->a; jj = a->j; 122917ab2063SBarry Smith for ( i=0; i<nz; i++ ) { 123017ab2063SBarry Smith (*v++) *= r[*jj++ + shift]; 123117ab2063SBarry Smith } 123244cd7ae7SLois Curfman McInnes PLogFlops(nz); 123317ab2063SBarry Smith } 123417ab2063SBarry Smith return 0; 123517ab2063SBarry Smith } 123617ab2063SBarry Smith 12375615d1e5SSatish Balay #undef __FUNC__ 12385615d1e5SSatish Balay #define __FUNC__ "MatGetSubMatrix_SeqAIJ" 1239cddf8d76SBarry Smith static int MatGetSubMatrix_SeqAIJ(Mat A,IS isrow,IS iscol,MatGetSubMatrixCall scall,Mat *B) 124017ab2063SBarry Smith { 1241db02288aSLois Curfman McInnes Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data,*c; 124202834360SBarry Smith int nznew, *smap, i, k, kstart, kend, ierr, oldcols = a->n,*lens; 124399141d43SSatish Balay int row,mat_i,*mat_j,tcol,first,step,*mat_ilen; 1244a2744918SBarry Smith register int sum,lensi; 124599141d43SSatish Balay int *irow, *icol, nrows, ncols, shift = a->indexshift,*ssmap; 124699141d43SSatish Balay int *starts,*j_new,*i_new,*aj = a->j, *ai = a->i,ii,*ailen = a->ilen; 124799141d43SSatish Balay Scalar *a_new,*mat_a; 1248416022c9SBarry Smith Mat C; 124917ab2063SBarry Smith 1250b48a1e75SSatish Balay ierr = ISSorted(isrow,(PetscTruth*)&i); 1251e3372554SBarry Smith if (!i) SETERRQ(1,0,"ISrow is not sorted"); 125299141d43SSatish Balay ierr = ISSorted(iscol,(PetscTruth*)&i); 1253e3372554SBarry Smith if (!i) SETERRQ(1,0,"IScol is not sorted"); 125499141d43SSatish Balay 125517ab2063SBarry Smith ierr = ISGetIndices(isrow,&irow); CHKERRQ(ierr); 125617ab2063SBarry Smith ierr = ISGetSize(isrow,&nrows); CHKERRQ(ierr); 125717ab2063SBarry Smith ierr = ISGetSize(iscol,&ncols); CHKERRQ(ierr); 125817ab2063SBarry Smith 12597264ac53SSatish Balay if (ISStrideGetInfo(iscol,&first,&step) && step == 1) { /* no need to sort */ 126002834360SBarry Smith /* special case of contiguous rows */ 126157faeb66SBarry Smith lens = (int *) PetscMalloc((ncols+nrows+1)*sizeof(int)); CHKPTRQ(lens); 126202834360SBarry Smith starts = lens + ncols; 126302834360SBarry Smith /* loop over new rows determining lens and starting points */ 126402834360SBarry Smith for (i=0; i<nrows; i++) { 1265a2744918SBarry Smith kstart = ai[irow[i]]+shift; 1266a2744918SBarry Smith kend = kstart + ailen[irow[i]]; 126702834360SBarry Smith for ( k=kstart; k<kend; k++ ) { 1268d8ced48eSBarry Smith if (aj[k]+shift >= first) { 126902834360SBarry Smith starts[i] = k; 127002834360SBarry Smith break; 127102834360SBarry Smith } 127202834360SBarry Smith } 1273a2744918SBarry Smith sum = 0; 127402834360SBarry Smith while (k < kend) { 1275d8ced48eSBarry Smith if (aj[k++]+shift >= first+ncols) break; 1276a2744918SBarry Smith sum++; 127702834360SBarry Smith } 1278a2744918SBarry Smith lens[i] = sum; 127902834360SBarry Smith } 128002834360SBarry Smith /* create submatrix */ 1281cddf8d76SBarry Smith if (scall == MAT_REUSE_MATRIX) { 128208480c60SBarry Smith int n_cols,n_rows; 128308480c60SBarry Smith ierr = MatGetSize(*B,&n_rows,&n_cols); CHKERRQ(ierr); 1284e3372554SBarry Smith if (n_rows != nrows || n_cols != ncols) SETERRQ(1,0,""); 1285d8ced48eSBarry Smith ierr = MatZeroEntries(*B); CHKERRQ(ierr); 128608480c60SBarry Smith C = *B; 128708480c60SBarry Smith } 128808480c60SBarry Smith else { 128902834360SBarry Smith ierr = MatCreateSeqAIJ(A->comm,nrows,ncols,0,lens,&C);CHKERRQ(ierr); 129008480c60SBarry Smith } 1291db02288aSLois Curfman McInnes c = (Mat_SeqAIJ*) C->data; 1292db02288aSLois Curfman McInnes 129302834360SBarry Smith /* loop over rows inserting into submatrix */ 1294db02288aSLois Curfman McInnes a_new = c->a; 1295db02288aSLois Curfman McInnes j_new = c->j; 1296db02288aSLois Curfman McInnes i_new = c->i; 1297db02288aSLois Curfman McInnes i_new[0] = -shift; 129802834360SBarry Smith for (i=0; i<nrows; i++) { 1299a2744918SBarry Smith ii = starts[i]; 1300a2744918SBarry Smith lensi = lens[i]; 1301a2744918SBarry Smith for ( k=0; k<lensi; k++ ) { 1302a2744918SBarry Smith *j_new++ = aj[ii+k] - first; 130302834360SBarry Smith } 1304a2744918SBarry Smith PetscMemcpy(a_new,a->a + starts[i],lensi*sizeof(Scalar)); 1305a2744918SBarry Smith a_new += lensi; 1306a2744918SBarry Smith i_new[i+1] = i_new[i] + lensi; 1307a2744918SBarry Smith c->ilen[i] = lensi; 130802834360SBarry Smith } 13090452661fSBarry Smith PetscFree(lens); 131002834360SBarry Smith } 131102834360SBarry Smith else { 131202834360SBarry Smith ierr = ISGetIndices(iscol,&icol); CHKERRQ(ierr); 13130452661fSBarry Smith smap = (int *) PetscMalloc((1+oldcols)*sizeof(int)); CHKPTRQ(smap); 131402834360SBarry Smith ssmap = smap + shift; 131599141d43SSatish Balay lens = (int *) PetscMalloc((1+nrows)*sizeof(int)); CHKPTRQ(lens); 1316cddf8d76SBarry Smith PetscMemzero(smap,oldcols*sizeof(int)); 131717ab2063SBarry Smith for ( i=0; i<ncols; i++ ) smap[icol[i]] = i+1; 131802834360SBarry Smith /* determine lens of each row */ 131902834360SBarry Smith for (i=0; i<nrows; i++) { 1320d8ced48eSBarry Smith kstart = ai[irow[i]]+shift; 132102834360SBarry Smith kend = kstart + a->ilen[irow[i]]; 132202834360SBarry Smith lens[i] = 0; 132302834360SBarry Smith for ( k=kstart; k<kend; k++ ) { 1324d8ced48eSBarry Smith if (ssmap[aj[k]]) { 132502834360SBarry Smith lens[i]++; 132602834360SBarry Smith } 132702834360SBarry Smith } 132802834360SBarry Smith } 132917ab2063SBarry Smith /* Create and fill new matrix */ 1330a2744918SBarry Smith if (scall == MAT_REUSE_MATRIX) { 133199141d43SSatish Balay c = (Mat_SeqAIJ *)((*B)->data); 133299141d43SSatish Balay 1333e3372554SBarry Smith if (c->m != nrows || c->n != ncols) SETERRQ(1,0,""); 133499141d43SSatish Balay if (PetscMemcmp(c->ilen,lens, c->m *sizeof(int))) { 1335e3372554SBarry Smith SETERRQ(1,0,"Cannot reuse matrix. wrong no of nonzeros"); 133699141d43SSatish Balay } 133799141d43SSatish Balay PetscMemzero(c->ilen,c->m*sizeof(int)); 133808480c60SBarry Smith C = *B; 133908480c60SBarry Smith } 134008480c60SBarry Smith else { 134102834360SBarry Smith ierr = MatCreateSeqAIJ(A->comm,nrows,ncols,0,lens,&C);CHKERRQ(ierr); 134208480c60SBarry Smith } 134399141d43SSatish Balay c = (Mat_SeqAIJ *)(C->data); 134417ab2063SBarry Smith for (i=0; i<nrows; i++) { 134599141d43SSatish Balay row = irow[i]; 134617ab2063SBarry Smith nznew = 0; 134799141d43SSatish Balay kstart = ai[row]+shift; 134899141d43SSatish Balay kend = kstart + a->ilen[row]; 134999141d43SSatish Balay mat_i = c->i[i]+shift; 135099141d43SSatish Balay mat_j = c->j + mat_i; 135199141d43SSatish Balay mat_a = c->a + mat_i; 135299141d43SSatish Balay mat_ilen = c->ilen + i; 135317ab2063SBarry Smith for ( k=kstart; k<kend; k++ ) { 135499141d43SSatish Balay if ((tcol=ssmap[a->j[k]])) { 135599141d43SSatish Balay *mat_j++ = tcol - (!shift); 135699141d43SSatish Balay *mat_a++ = a->a[k]; 135799141d43SSatish Balay (*mat_ilen)++; 135899141d43SSatish Balay 135917ab2063SBarry Smith } 136017ab2063SBarry Smith } 136117ab2063SBarry Smith } 136202834360SBarry Smith /* Free work space */ 136302834360SBarry Smith ierr = ISRestoreIndices(iscol,&icol); CHKERRQ(ierr); 136499141d43SSatish Balay PetscFree(smap); PetscFree(lens); 136502834360SBarry Smith } 13666d4a8577SBarry Smith ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr); 13676d4a8577SBarry Smith ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr); 136817ab2063SBarry Smith 136917ab2063SBarry Smith ierr = ISRestoreIndices(isrow,&irow); CHKERRQ(ierr); 1370416022c9SBarry Smith *B = C; 137117ab2063SBarry Smith return 0; 137217ab2063SBarry Smith } 137317ab2063SBarry Smith 1374a871dcd8SBarry Smith /* 137563b91edcSBarry Smith note: This can only work for identity for row and col. It would 137663b91edcSBarry Smith be good to check this and otherwise generate an error. 1377a871dcd8SBarry Smith */ 13785615d1e5SSatish Balay #undef __FUNC__ 13795615d1e5SSatish Balay #define __FUNC__ "MatILUFactor_SeqAIJ" 138063b91edcSBarry Smith static int MatILUFactor_SeqAIJ(Mat inA,IS row,IS col,double efill,int fill) 1381a871dcd8SBarry Smith { 138263b91edcSBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) inA->data; 138308480c60SBarry Smith int ierr; 138463b91edcSBarry Smith Mat outA; 138563b91edcSBarry Smith 1386e3372554SBarry Smith if (fill != 0) SETERRQ(1,0,"Only fill=0 supported"); 1387a871dcd8SBarry Smith 138863b91edcSBarry Smith outA = inA; 138963b91edcSBarry Smith inA->factor = FACTOR_LU; 139063b91edcSBarry Smith a->row = row; 139163b91edcSBarry Smith a->col = col; 139263b91edcSBarry Smith 139394a9d846SBarry Smith if (!a->solve_work) { /* this matrix may have been factored before */ 13940452661fSBarry Smith a->solve_work = (Scalar *) PetscMalloc( (a->m+1)*sizeof(Scalar)); CHKPTRQ(a->solve_work); 139594a9d846SBarry Smith } 139663b91edcSBarry Smith 139708480c60SBarry Smith if (!a->diag) { 139808480c60SBarry Smith ierr = MatMarkDiag_SeqAIJ(inA); CHKERRQ(ierr); 139963b91edcSBarry Smith } 140063b91edcSBarry Smith ierr = MatLUFactorNumeric_SeqAIJ(inA,&outA); CHKERRQ(ierr); 1401a871dcd8SBarry Smith return 0; 1402a871dcd8SBarry Smith } 1403a871dcd8SBarry Smith 1404f0b747eeSBarry Smith #include "pinclude/plapack.h" 14055615d1e5SSatish Balay #undef __FUNC__ 14065615d1e5SSatish Balay #define __FUNC__ "MatScale_SeqAIJ" 1407f0b747eeSBarry Smith static int MatScale_SeqAIJ(Scalar *alpha,Mat inA) 1408f0b747eeSBarry Smith { 1409f0b747eeSBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) inA->data; 1410f0b747eeSBarry Smith int one = 1; 1411f0b747eeSBarry Smith BLscal_( &a->nz, alpha, a->a, &one ); 1412f0b747eeSBarry Smith PLogFlops(a->nz); 1413f0b747eeSBarry Smith return 0; 1414f0b747eeSBarry Smith } 1415f0b747eeSBarry Smith 14165615d1e5SSatish Balay #undef __FUNC__ 14175615d1e5SSatish Balay #define __FUNC__ "MatGetSubMatrices_SeqAIJ" 1418cddf8d76SBarry Smith static int MatGetSubMatrices_SeqAIJ(Mat A,int n, IS *irow,IS *icol,MatGetSubMatrixCall scall, 1419cddf8d76SBarry Smith Mat **B) 1420cddf8d76SBarry Smith { 1421cddf8d76SBarry Smith int ierr,i; 1422cddf8d76SBarry Smith 1423cddf8d76SBarry Smith if (scall == MAT_INITIAL_MATRIX) { 14240452661fSBarry Smith *B = (Mat *) PetscMalloc( (n+1)*sizeof(Mat) ); CHKPTRQ(*B); 1425cddf8d76SBarry Smith } 1426cddf8d76SBarry Smith 1427cddf8d76SBarry Smith for ( i=0; i<n; i++ ) { 1428905e6a2fSBarry Smith ierr = MatGetSubMatrix_SeqAIJ(A,irow[i],icol[i],scall,&(*B)[i]);CHKERRQ(ierr); 1429cddf8d76SBarry Smith } 1430cddf8d76SBarry Smith return 0; 1431cddf8d76SBarry Smith } 1432cddf8d76SBarry Smith 14335615d1e5SSatish Balay #undef __FUNC__ 14345615d1e5SSatish Balay #define __FUNC__ "MatGetBlockSize_SeqAIJ" 14355a838052SSatish Balay static int MatGetBlockSize_SeqAIJ(Mat A, int *bs) 14365a838052SSatish Balay { 14375a838052SSatish Balay *bs = 1; 14385a838052SSatish Balay return 0; 14395a838052SSatish Balay } 14405a838052SSatish Balay 14415615d1e5SSatish Balay #undef __FUNC__ 14425615d1e5SSatish Balay #define __FUNC__ "MatIncreaseOverlap_SeqAIJ" 1443e4d965acSSatish Balay static int MatIncreaseOverlap_SeqAIJ(Mat A, int is_max, IS *is, int ov) 14444dcbc457SBarry Smith { 1445e4d965acSSatish Balay Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 144606763907SSatish Balay int shift, row, i,j,k,l,m,n, *idx,ierr, *nidx, isz, val; 14478a047759SSatish Balay int start, end, *ai, *aj; 144806763907SSatish Balay char *table; 14498a047759SSatish Balay shift = a->indexshift; 1450e4d965acSSatish Balay m = a->m; 1451e4d965acSSatish Balay ai = a->i; 14528a047759SSatish Balay aj = a->j+shift; 14538a047759SSatish Balay 1454e3372554SBarry Smith if (ov < 0) SETERRQ(1,0,"illegal overlap value used"); 145506763907SSatish Balay 145606763907SSatish Balay table = (char *) PetscMalloc((m/BITSPERBYTE +1)*sizeof(char)); CHKPTRQ(table); 145706763907SSatish Balay nidx = (int *) PetscMalloc((m+1)*sizeof(int)); CHKPTRQ(nidx); 145806763907SSatish Balay 1459e4d965acSSatish Balay for ( i=0; i<is_max; i++ ) { 1460b97fc60eSLois Curfman McInnes /* Initialize the two local arrays */ 1461e4d965acSSatish Balay isz = 0; 146206763907SSatish Balay PetscMemzero(table,(m/BITSPERBYTE +1)*sizeof(char)); 1463e4d965acSSatish Balay 1464e4d965acSSatish Balay /* Extract the indices, assume there can be duplicate entries */ 14654dcbc457SBarry Smith ierr = ISGetIndices(is[i],&idx); CHKERRQ(ierr); 146677c4ece6SBarry Smith ierr = ISGetSize(is[i],&n); CHKERRQ(ierr); 1467e4d965acSSatish Balay 1468*dd097bc3SLois Curfman McInnes /* Enter these into the temp arrays. I.e., mark table[row], enter row into new index */ 1469e4d965acSSatish Balay for ( j=0; j<n ; ++j){ 147006763907SSatish Balay if(!BT_LOOKUP(table, idx[j])) { nidx[isz++] = idx[j];} 14714dcbc457SBarry Smith } 147206763907SSatish Balay ierr = ISRestoreIndices(is[i],&idx); CHKERRQ(ierr); 147306763907SSatish Balay ierr = ISDestroy(is[i]); CHKERRQ(ierr); 1474e4d965acSSatish Balay 147504a348a9SBarry Smith k = 0; 147604a348a9SBarry Smith for ( j=0; j<ov; j++){ /* for each overlap */ 147704a348a9SBarry Smith n = isz; 147806763907SSatish Balay for ( ; k<n ; k++){ /* do only those rows in nidx[k], which are not done yet */ 1479e4d965acSSatish Balay row = nidx[k]; 1480e4d965acSSatish Balay start = ai[row]; 1481e4d965acSSatish Balay end = ai[row+1]; 148204a348a9SBarry Smith for ( l = start; l<end ; l++){ 14838a047759SSatish Balay val = aj[l] + shift; 148406763907SSatish Balay if (!BT_LOOKUP(table,val)) {nidx[isz++] = val;} 1485e4d965acSSatish Balay } 1486e4d965acSSatish Balay } 1487e4d965acSSatish Balay } 1488537820f0SBarry Smith ierr = ISCreateGeneral(MPI_COMM_SELF, isz, nidx, (is+i)); CHKERRQ(ierr); 1489e4d965acSSatish Balay } 149004a348a9SBarry Smith PetscFree(table); 149106763907SSatish Balay PetscFree(nidx); 1492e4d965acSSatish Balay return 0; 14934dcbc457SBarry Smith } 149417ab2063SBarry Smith 14950513a670SBarry Smith /* -------------------------------------------------------------- */ 14960513a670SBarry Smith #undef __FUNC__ 14970513a670SBarry Smith #define __FUNC__ "MatPermute_SeqAIJ" 14980513a670SBarry Smith int MatPermute_SeqAIJ(Mat A, IS rowp, IS colp, Mat *B) 14990513a670SBarry Smith { 15000513a670SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 15010513a670SBarry Smith Scalar *vwork; 15020513a670SBarry Smith int i, ierr, nz, m = a->m, n = a->n, *cwork; 15030513a670SBarry Smith int *row,*col,*cnew,j,*lens; 15040513a670SBarry Smith 15050513a670SBarry Smith ierr = ISGetIndices(rowp,&row); CHKERRQ(ierr); 15060513a670SBarry Smith ierr = ISGetIndices(colp,&col); CHKERRQ(ierr); 15070513a670SBarry Smith 15080513a670SBarry Smith /* determine lengths of permuted rows */ 15090513a670SBarry Smith lens = (int *) PetscMalloc( (m+1)*sizeof(int) ); CHKPTRQ(lens); 15100513a670SBarry Smith for (i=0; i<m; i++ ) { 15110513a670SBarry Smith lens[row[i]] = a->i[i+1] - a->i[i]; 15120513a670SBarry Smith } 15130513a670SBarry Smith ierr = MatCreateSeqAIJ(A->comm,m,n,0,lens,B);CHKERRQ(ierr); 15140513a670SBarry Smith PetscFree(lens); 15150513a670SBarry Smith 15160513a670SBarry Smith cnew = (int *) PetscMalloc( n*sizeof(int) ); CHKPTRQ(cnew); 15170513a670SBarry Smith for (i=0; i<m; i++) { 15180513a670SBarry Smith ierr = MatGetRow(A,i,&nz,&cwork,&vwork); CHKERRQ(ierr); 15190513a670SBarry Smith for (j=0; j<nz; j++ ) { cnew[j] = col[cwork[j]];} 15200513a670SBarry Smith ierr = MatSetValues(*B,1,&row[i],nz,cnew,vwork,INSERT_VALUES); CHKERRQ(ierr); 15210513a670SBarry Smith ierr = MatRestoreRow(A,i,&nz,&cwork,&vwork); CHKERRQ(ierr); 15220513a670SBarry Smith } 15230513a670SBarry Smith PetscFree(cnew); 15240513a670SBarry Smith ierr = MatAssemblyBegin(*B,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr); 15250513a670SBarry Smith ierr = MatAssemblyEnd(*B,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr); 15260513a670SBarry Smith ierr = ISRestoreIndices(rowp,&row); CHKERRQ(ierr); 15270513a670SBarry Smith ierr = ISRestoreIndices(colp,&col); CHKERRQ(ierr); 15280513a670SBarry Smith return 0; 15290513a670SBarry Smith } 15300513a670SBarry Smith 15315615d1e5SSatish Balay #undef __FUNC__ 15325615d1e5SSatish Balay #define __FUNC__ "MatPrintHelp_SeqAIJ" 1533682d7d0cSBarry Smith int MatPrintHelp_SeqAIJ(Mat A) 1534682d7d0cSBarry Smith { 1535682d7d0cSBarry Smith static int called = 0; 1536682d7d0cSBarry Smith MPI_Comm comm = A->comm; 1537682d7d0cSBarry Smith 1538682d7d0cSBarry Smith if (called) return 0; else called = 1; 153977c4ece6SBarry Smith PetscPrintf(comm," Options for MATSEQAIJ and MATMPIAIJ matrix formats (the defaults):\n"); 15400f665d81SLois Curfman McInnes PetscPrintf(comm," -mat_lu_pivotthreshold <threshold>: Set pivoting threshold\n"); 15410f665d81SLois Curfman McInnes PetscPrintf(comm," -mat_aij_oneindex: internal indices begin at 1 instead of the default 0.\n"); 15420f665d81SLois Curfman McInnes PetscPrintf(comm," -mat_aij_no_inode: Do not use inodes\n"); 15430f665d81SLois Curfman McInnes PetscPrintf(comm," -mat_aij_inode_limit <limit>: Set inode limit (max limit=5)\n"); 1544682d7d0cSBarry Smith #if defined(HAVE_ESSL) 15450f665d81SLois Curfman McInnes PetscPrintf(comm," -mat_aij_essl: Use IBM sparse LU factorization and solve.\n"); 1546682d7d0cSBarry Smith #endif 1547682d7d0cSBarry Smith return 0; 1548682d7d0cSBarry Smith } 1549205e38a3SBarry Smith static int MatEqual_SeqAIJ(Mat A,Mat B, PetscTruth* flg); 1550a93ec695SBarry Smith extern int MatFDColoringCreate_SeqAIJ(Mat,ISColoring,MatFDColoring); 1551a93ec695SBarry Smith extern int MatColoringPatch_SeqAIJ(Mat,int,int *,ISColoring *); 1552a93ec695SBarry Smith 1553682d7d0cSBarry Smith /* -------------------------------------------------------------------*/ 155417ab2063SBarry Smith static struct _MatOps MatOps = {MatSetValues_SeqAIJ, 155517ab2063SBarry Smith MatGetRow_SeqAIJ,MatRestoreRow_SeqAIJ, 1556416022c9SBarry Smith MatMult_SeqAIJ,MatMultAdd_SeqAIJ, 1557416022c9SBarry Smith MatMultTrans_SeqAIJ,MatMultTransAdd_SeqAIJ, 155817ab2063SBarry Smith MatSolve_SeqAIJ,MatSolveAdd_SeqAIJ, 155917ab2063SBarry Smith MatSolveTrans_SeqAIJ,MatSolveTransAdd_SeqAIJ, 156017ab2063SBarry Smith MatLUFactor_SeqAIJ,0, 156117ab2063SBarry Smith MatRelax_SeqAIJ, 156217ab2063SBarry Smith MatTranspose_SeqAIJ, 15637264ac53SSatish Balay MatGetInfo_SeqAIJ,MatEqual_SeqAIJ, 1564f0b747eeSBarry Smith MatGetDiagonal_SeqAIJ,MatDiagonalScale_SeqAIJ,MatNorm_SeqAIJ, 156517ab2063SBarry Smith 0,MatAssemblyEnd_SeqAIJ, 156617ab2063SBarry Smith MatCompress_SeqAIJ, 156717ab2063SBarry Smith MatSetOption_SeqAIJ,MatZeroEntries_SeqAIJ,MatZeroRows_SeqAIJ, 156817ab2063SBarry Smith MatLUFactorSymbolic_SeqAIJ,MatLUFactorNumeric_SeqAIJ,0,0, 156917ab2063SBarry Smith MatGetSize_SeqAIJ,MatGetSize_SeqAIJ,MatGetOwnershipRange_SeqAIJ, 157017ab2063SBarry Smith MatILUFactorSymbolic_SeqAIJ,0, 157194a9d846SBarry Smith 0,0, 15723d1612f7SBarry Smith MatConvertSameType_SeqAIJ,0,0, 1573cddf8d76SBarry Smith MatILUFactor_SeqAIJ,0,0, 15747eb43aa7SLois Curfman McInnes MatGetSubMatrices_SeqAIJ,MatIncreaseOverlap_SeqAIJ, 1575682d7d0cSBarry Smith MatGetValues_SeqAIJ,0, 1576f0b747eeSBarry Smith MatPrintHelp_SeqAIJ, 15775a838052SSatish Balay MatScale_SeqAIJ,0,0, 15786945ee14SBarry Smith MatILUDTFactor_SeqAIJ, 15796945ee14SBarry Smith MatGetBlockSize_SeqAIJ, 15803b2fbd54SBarry Smith MatGetRowIJ_SeqAIJ, 15813b2fbd54SBarry Smith MatRestoreRowIJ_SeqAIJ, 15823b2fbd54SBarry Smith MatGetColumnIJ_SeqAIJ, 1583a93ec695SBarry Smith MatRestoreColumnIJ_SeqAIJ, 1584a93ec695SBarry Smith MatFDColoringCreate_SeqAIJ, 15850513a670SBarry Smith MatColoringPatch_SeqAIJ, 15860513a670SBarry Smith 0, 15870513a670SBarry Smith MatPermute_SeqAIJ}; 158817ab2063SBarry Smith 158917ab2063SBarry Smith extern int MatUseSuperLU_SeqAIJ(Mat); 159017ab2063SBarry Smith extern int MatUseEssl_SeqAIJ(Mat); 159117ab2063SBarry Smith extern int MatUseDXML_SeqAIJ(Mat); 159217ab2063SBarry Smith 15935615d1e5SSatish Balay #undef __FUNC__ 15945615d1e5SSatish Balay #define __FUNC__ "MatCreateSeqAIJ" 159517ab2063SBarry Smith /*@C 1596682d7d0cSBarry Smith MatCreateSeqAIJ - Creates a sparse matrix in AIJ (compressed row) format 15970d15e28bSLois Curfman McInnes (the default parallel PETSc format). For good matrix assembly performance 15986e62573dSLois Curfman McInnes the user should preallocate the matrix storage by setting the parameter nz 15992bd5e0b2SLois Curfman McInnes (or the array nzz). By setting these parameters accurately, performance 16002bd5e0b2SLois Curfman McInnes during matrix assembly can be increased by more than a factor of 50. 160117ab2063SBarry Smith 160217ab2063SBarry Smith Input Parameters: 160317ab2063SBarry Smith . comm - MPI communicator, set to MPI_COMM_SELF 160417ab2063SBarry Smith . m - number of rows 160517ab2063SBarry Smith . n - number of columns 160617ab2063SBarry Smith . nz - number of nonzeros per row (same for all rows) 16072bd5e0b2SLois Curfman McInnes . nzz - array containing the number of nonzeros in the various rows 16082bd5e0b2SLois Curfman McInnes (possibly different for each row) or PETSC_NULL 160917ab2063SBarry Smith 161017ab2063SBarry Smith Output Parameter: 1611416022c9SBarry Smith . A - the matrix 161217ab2063SBarry Smith 161317ab2063SBarry Smith Notes: 161417ab2063SBarry Smith The AIJ format (also called the Yale sparse matrix format or 161517ab2063SBarry Smith compressed row storage), is fully compatible with standard Fortran 77 16160002213bSLois Curfman McInnes storage. That is, the stored row and column indices can begin at 161744cd7ae7SLois Curfman McInnes either one (as in Fortran) or zero. See the users' manual for details. 161817ab2063SBarry Smith 161917ab2063SBarry Smith Specify the preallocated storage with either nz or nnz (not both). 1620a40aa06bSLois Curfman McInnes Set nz=PETSC_DEFAULT and nnz=PETSC_NULL for PETSc to control dynamic memory 16213d323bbdSBarry Smith allocation. For large problems you MUST preallocate memory or you 16226da5968aSLois Curfman McInnes will get TERRIBLE performance, see the users' manual chapter on matrices. 162317ab2063SBarry Smith 1624682d7d0cSBarry Smith By default, this format uses inodes (identical nodes) when possible, to 1625682d7d0cSBarry Smith improve numerical efficiency of Matrix vector products and solves. We 1626682d7d0cSBarry Smith search for consecutive rows with the same nonzero structure, thereby 16276c7ebb05SLois Curfman McInnes reusing matrix information to achieve increased efficiency. 16286c7ebb05SLois Curfman McInnes 16296c7ebb05SLois Curfman McInnes Options Database Keys: 16306c7ebb05SLois Curfman McInnes $ -mat_aij_no_inode - Do not use inodes 16316e62573dSLois Curfman McInnes $ -mat_aij_inode_limit <limit> - Set inode limit. 16326e62573dSLois Curfman McInnes $ (max limit=5) 16336e62573dSLois Curfman McInnes $ -mat_aij_oneindex - Internally use indexing starting at 1 16346e62573dSLois Curfman McInnes $ rather than 0. Note: When calling MatSetValues(), 16356e62573dSLois Curfman McInnes $ the user still MUST index entries starting at 0! 163617ab2063SBarry Smith 163717ab2063SBarry Smith .seealso: MatCreate(), MatCreateMPIAIJ(), MatSetValues() 163817ab2063SBarry Smith @*/ 1639416022c9SBarry Smith int MatCreateSeqAIJ(MPI_Comm comm,int m,int n,int nz,int *nnz, Mat *A) 164017ab2063SBarry Smith { 1641416022c9SBarry Smith Mat B; 1642416022c9SBarry Smith Mat_SeqAIJ *b; 16436945ee14SBarry Smith int i, len, ierr, flg,size; 16446945ee14SBarry Smith 16456945ee14SBarry Smith MPI_Comm_size(comm,&size); 1646e3372554SBarry Smith if (size > 1) SETERRQ(1,0,"Comm must be of size 1"); 1647d5d45c9bSBarry Smith 1648416022c9SBarry Smith *A = 0; 16490452661fSBarry Smith PetscHeaderCreate(B,_Mat,MAT_COOKIE,MATSEQAIJ,comm); 1650416022c9SBarry Smith PLogObjectCreate(B); 16510452661fSBarry Smith B->data = (void *) (b = PetscNew(Mat_SeqAIJ)); CHKPTRQ(b); 165244cd7ae7SLois Curfman McInnes PetscMemzero(b,sizeof(Mat_SeqAIJ)); 1653416022c9SBarry Smith PetscMemcpy(&B->ops,&MatOps,sizeof(struct _MatOps)); 1654416022c9SBarry Smith B->destroy = MatDestroy_SeqAIJ; 1655416022c9SBarry Smith B->view = MatView_SeqAIJ; 1656416022c9SBarry Smith B->factor = 0; 1657416022c9SBarry Smith B->lupivotthreshold = 1.0; 165890f02eecSBarry Smith B->mapping = 0; 16597a743949SBarry Smith ierr = OptionsGetDouble(PETSC_NULL,"-mat_lu_pivotthreshold",&B->lupivotthreshold, 166069957df2SSatish Balay &flg); CHKERRQ(ierr); 16617a743949SBarry Smith b->ilu_preserve_row_sums = PETSC_FALSE; 16627a743949SBarry Smith ierr = OptionsHasName(PETSC_NULL,"-pc_ilu_preserve_row_sums", 16637a743949SBarry Smith (int*) &b->ilu_preserve_row_sums); CHKERRQ(ierr); 1664416022c9SBarry Smith b->row = 0; 1665416022c9SBarry Smith b->col = 0; 1666416022c9SBarry Smith b->indexshift = 0; 1667b810aeb4SBarry Smith b->reallocs = 0; 166869957df2SSatish Balay ierr = OptionsHasName(PETSC_NULL,"-mat_aij_oneindex", &flg); CHKERRQ(ierr); 166969957df2SSatish Balay if (flg) b->indexshift = -1; 167017ab2063SBarry Smith 167144cd7ae7SLois Curfman McInnes b->m = m; B->m = m; B->M = m; 167244cd7ae7SLois Curfman McInnes b->n = n; B->n = n; B->N = n; 16730452661fSBarry Smith b->imax = (int *) PetscMalloc( (m+1)*sizeof(int) ); CHKPTRQ(b->imax); 1674b4fd4287SBarry Smith if (nnz == PETSC_NULL) { 16757b8455f0SLois Curfman McInnes if (nz == PETSC_DEFAULT) nz = 10; 16767b8455f0SLois Curfman McInnes else if (nz <= 0) nz = 1; 1677416022c9SBarry Smith for ( i=0; i<m; i++ ) b->imax[i] = nz; 167817ab2063SBarry Smith nz = nz*m; 167917ab2063SBarry Smith } 168017ab2063SBarry Smith else { 168117ab2063SBarry Smith nz = 0; 1682416022c9SBarry Smith for ( i=0; i<m; i++ ) {b->imax[i] = nnz[i]; nz += nnz[i];} 168317ab2063SBarry Smith } 168417ab2063SBarry Smith 168517ab2063SBarry Smith /* allocate the matrix space */ 1686416022c9SBarry Smith len = nz*(sizeof(int) + sizeof(Scalar)) + (b->m+1)*sizeof(int); 16870452661fSBarry Smith b->a = (Scalar *) PetscMalloc( len ); CHKPTRQ(b->a); 1688416022c9SBarry Smith b->j = (int *) (b->a + nz); 1689cddf8d76SBarry Smith PetscMemzero(b->j,nz*sizeof(int)); 1690416022c9SBarry Smith b->i = b->j + nz; 1691416022c9SBarry Smith b->singlemalloc = 1; 169217ab2063SBarry Smith 1693416022c9SBarry Smith b->i[0] = -b->indexshift; 169417ab2063SBarry Smith for (i=1; i<m+1; i++) { 1695416022c9SBarry Smith b->i[i] = b->i[i-1] + b->imax[i-1]; 169617ab2063SBarry Smith } 169717ab2063SBarry Smith 1698416022c9SBarry Smith /* b->ilen will count nonzeros in each row so far. */ 16990452661fSBarry Smith b->ilen = (int *) PetscMalloc((m+1)*sizeof(int)); 1700416022c9SBarry Smith PLogObjectMemory(B,len+2*(m+1)*sizeof(int)+sizeof(struct _Mat)+sizeof(Mat_SeqAIJ)); 1701416022c9SBarry Smith for ( i=0; i<b->m; i++ ) { b->ilen[i] = 0;} 170217ab2063SBarry Smith 1703416022c9SBarry Smith b->nz = 0; 1704416022c9SBarry Smith b->maxnz = nz; 1705416022c9SBarry Smith b->sorted = 0; 1706416022c9SBarry Smith b->roworiented = 1; 1707416022c9SBarry Smith b->nonew = 0; 1708416022c9SBarry Smith b->diag = 0; 1709416022c9SBarry Smith b->solve_work = 0; 171071bd300dSLois Curfman McInnes b->spptr = 0; 1711754ec7b1SSatish Balay b->inode.node_count = 0; 1712754ec7b1SSatish Balay b->inode.size = 0; 17136c7ebb05SLois Curfman McInnes b->inode.limit = 5; 17146c7ebb05SLois Curfman McInnes b->inode.max_limit = 5; 17154e220ebcSLois Curfman McInnes B->info.nz_unneeded = (double)b->maxnz; 171617ab2063SBarry Smith 1717416022c9SBarry Smith *A = B; 17184e220ebcSLois Curfman McInnes 17194b14c69eSBarry Smith /* SuperLU is not currently supported through PETSc */ 17204b14c69eSBarry Smith #if defined(HAVE_SUPERLU) 172169957df2SSatish Balay ierr = OptionsHasName(PETSC_NULL,"-mat_aij_superlu", &flg); CHKERRQ(ierr); 172269957df2SSatish Balay if (flg) { ierr = MatUseSuperLU_SeqAIJ(B); CHKERRQ(ierr); } 17234b14c69eSBarry Smith #endif 172469957df2SSatish Balay ierr = OptionsHasName(PETSC_NULL,"-mat_aij_essl", &flg); CHKERRQ(ierr); 172569957df2SSatish Balay if (flg) { ierr = MatUseEssl_SeqAIJ(B); CHKERRQ(ierr); } 172669957df2SSatish Balay ierr = OptionsHasName(PETSC_NULL,"-mat_aij_dxml", &flg); CHKERRQ(ierr); 172769957df2SSatish Balay if (flg) { 1728e3372554SBarry Smith if (!b->indexshift) SETERRQ(1,0,"need -mat_aij_oneindex with -mat_aij_dxml"); 1729416022c9SBarry Smith ierr = MatUseDXML_SeqAIJ(B); CHKERRQ(ierr); 173017ab2063SBarry Smith } 173169957df2SSatish Balay ierr = OptionsHasName(PETSC_NULL,"-help", &flg); CHKERRQ(ierr); 173269957df2SSatish Balay if (flg) {ierr = MatPrintHelp(B); CHKERRQ(ierr); } 173317ab2063SBarry Smith return 0; 173417ab2063SBarry Smith } 173517ab2063SBarry Smith 17365615d1e5SSatish Balay #undef __FUNC__ 17375615d1e5SSatish Balay #define __FUNC__ "MatConvertSameType_SeqAIJ" 17383d1612f7SBarry Smith int MatConvertSameType_SeqAIJ(Mat A,Mat *B,int cpvalues) 173917ab2063SBarry Smith { 1740416022c9SBarry Smith Mat C; 1741416022c9SBarry Smith Mat_SeqAIJ *c,*a = (Mat_SeqAIJ *) A->data; 174208480c60SBarry Smith int i,len, m = a->m,shift = a->indexshift; 174317ab2063SBarry Smith 17444043dd9cSLois Curfman McInnes *B = 0; 17450452661fSBarry Smith PetscHeaderCreate(C,_Mat,MAT_COOKIE,MATSEQAIJ,A->comm); 1746416022c9SBarry Smith PLogObjectCreate(C); 17470452661fSBarry Smith C->data = (void *) (c = PetscNew(Mat_SeqAIJ)); CHKPTRQ(c); 174841c01911SSatish Balay PetscMemcpy(&C->ops,&A->ops,sizeof(struct _MatOps)); 1749416022c9SBarry Smith C->destroy = MatDestroy_SeqAIJ; 1750416022c9SBarry Smith C->view = MatView_SeqAIJ; 1751416022c9SBarry Smith C->factor = A->factor; 1752416022c9SBarry Smith c->row = 0; 1753416022c9SBarry Smith c->col = 0; 1754416022c9SBarry Smith c->indexshift = shift; 1755c456f294SBarry Smith C->assembled = PETSC_TRUE; 175617ab2063SBarry Smith 175744cd7ae7SLois Curfman McInnes c->m = C->m = a->m; 175844cd7ae7SLois Curfman McInnes c->n = C->n = a->n; 175944cd7ae7SLois Curfman McInnes C->M = a->m; 176044cd7ae7SLois Curfman McInnes C->N = a->n; 176117ab2063SBarry Smith 17620452661fSBarry Smith c->imax = (int *) PetscMalloc((m+1)*sizeof(int)); CHKPTRQ(c->imax); 17630452661fSBarry Smith c->ilen = (int *) PetscMalloc((m+1)*sizeof(int)); CHKPTRQ(c->ilen); 176417ab2063SBarry Smith for ( i=0; i<m; i++ ) { 1765416022c9SBarry Smith c->imax[i] = a->imax[i]; 1766416022c9SBarry Smith c->ilen[i] = a->ilen[i]; 176717ab2063SBarry Smith } 176817ab2063SBarry Smith 176917ab2063SBarry Smith /* allocate the matrix space */ 1770416022c9SBarry Smith c->singlemalloc = 1; 1771416022c9SBarry Smith len = (m+1)*sizeof(int)+(a->i[m])*(sizeof(Scalar)+sizeof(int)); 17720452661fSBarry Smith c->a = (Scalar *) PetscMalloc( len ); CHKPTRQ(c->a); 1773416022c9SBarry Smith c->j = (int *) (c->a + a->i[m] + shift); 1774416022c9SBarry Smith c->i = c->j + a->i[m] + shift; 1775416022c9SBarry Smith PetscMemcpy(c->i,a->i,(m+1)*sizeof(int)); 177617ab2063SBarry Smith if (m > 0) { 1777416022c9SBarry Smith PetscMemcpy(c->j,a->j,(a->i[m]+shift)*sizeof(int)); 177808480c60SBarry Smith if (cpvalues == COPY_VALUES) { 1779416022c9SBarry Smith PetscMemcpy(c->a,a->a,(a->i[m]+shift)*sizeof(Scalar)); 178017ab2063SBarry Smith } 178108480c60SBarry Smith } 178217ab2063SBarry Smith 1783416022c9SBarry Smith PLogObjectMemory(C,len+2*(m+1)*sizeof(int)+sizeof(struct _Mat)+sizeof(Mat_SeqAIJ)); 1784416022c9SBarry Smith c->sorted = a->sorted; 1785416022c9SBarry Smith c->roworiented = a->roworiented; 1786416022c9SBarry Smith c->nonew = a->nonew; 17877a743949SBarry Smith c->ilu_preserve_row_sums = a->ilu_preserve_row_sums; 178817ab2063SBarry Smith 1789416022c9SBarry Smith if (a->diag) { 17900452661fSBarry Smith c->diag = (int *) PetscMalloc( (m+1)*sizeof(int) ); CHKPTRQ(c->diag); 1791416022c9SBarry Smith PLogObjectMemory(C,(m+1)*sizeof(int)); 179217ab2063SBarry Smith for ( i=0; i<m; i++ ) { 1793416022c9SBarry Smith c->diag[i] = a->diag[i]; 179417ab2063SBarry Smith } 179517ab2063SBarry Smith } 1796416022c9SBarry Smith else c->diag = 0; 17976c7ebb05SLois Curfman McInnes c->inode.limit = a->inode.limit; 17986c7ebb05SLois Curfman McInnes c->inode.max_limit = a->inode.max_limit; 1799754ec7b1SSatish Balay if (a->inode.size){ 1800daed632aSSatish Balay c->inode.size = (int *) PetscMalloc( (m+1)*sizeof(int) ); CHKPTRQ(c->inode.size); 1801754ec7b1SSatish Balay c->inode.node_count = a->inode.node_count; 1802daed632aSSatish Balay PetscMemcpy( c->inode.size, a->inode.size, (m+1)*sizeof(int)); 1803754ec7b1SSatish Balay } else { 1804754ec7b1SSatish Balay c->inode.size = 0; 1805754ec7b1SSatish Balay c->inode.node_count = 0; 1806754ec7b1SSatish Balay } 1807416022c9SBarry Smith c->nz = a->nz; 1808416022c9SBarry Smith c->maxnz = a->maxnz; 1809416022c9SBarry Smith c->solve_work = 0; 181076dd722bSSatish Balay c->spptr = 0; /* Dangerous -I'm throwing away a->spptr */ 1811754ec7b1SSatish Balay 1812416022c9SBarry Smith *B = C; 181317ab2063SBarry Smith return 0; 181417ab2063SBarry Smith } 181517ab2063SBarry Smith 18165615d1e5SSatish Balay #undef __FUNC__ 18175615d1e5SSatish Balay #define __FUNC__ "MatLoad_SeqAIJ" 181819bcc07fSBarry Smith int MatLoad_SeqAIJ(Viewer viewer,MatType type,Mat *A) 181917ab2063SBarry Smith { 1820416022c9SBarry Smith Mat_SeqAIJ *a; 1821416022c9SBarry Smith Mat B; 182217699dbbSLois Curfman McInnes int i, nz, ierr, fd, header[4],size,*rowlengths = 0,M,N,shift; 1823bcd2baecSBarry Smith MPI_Comm comm; 182417ab2063SBarry Smith 182519bcc07fSBarry Smith PetscObjectGetComm((PetscObject) viewer,&comm); 182617699dbbSLois Curfman McInnes MPI_Comm_size(comm,&size); 1827e3372554SBarry Smith if (size > 1) SETERRQ(1,0,"view must have one processor"); 182890ace30eSBarry Smith ierr = ViewerBinaryGetDescriptor(viewer,&fd); CHKERRQ(ierr); 182977c4ece6SBarry Smith ierr = PetscBinaryRead(fd,header,4,BINARY_INT); CHKERRQ(ierr); 1830e3372554SBarry Smith if (header[0] != MAT_COOKIE) SETERRQ(1,0,"not matrix object in file"); 183117ab2063SBarry Smith M = header[1]; N = header[2]; nz = header[3]; 183217ab2063SBarry Smith 183317ab2063SBarry Smith /* read in row lengths */ 18340452661fSBarry Smith rowlengths = (int*) PetscMalloc( M*sizeof(int) ); CHKPTRQ(rowlengths); 183577c4ece6SBarry Smith ierr = PetscBinaryRead(fd,rowlengths,M,BINARY_INT); CHKERRQ(ierr); 183617ab2063SBarry Smith 183717ab2063SBarry Smith /* create our matrix */ 1838416022c9SBarry Smith ierr = MatCreateSeqAIJ(comm,M,N,0,rowlengths,A); CHKERRQ(ierr); 1839416022c9SBarry Smith B = *A; 1840416022c9SBarry Smith a = (Mat_SeqAIJ *) B->data; 1841416022c9SBarry Smith shift = a->indexshift; 184217ab2063SBarry Smith 184317ab2063SBarry Smith /* read in column indices and adjust for Fortran indexing*/ 184477c4ece6SBarry Smith ierr = PetscBinaryRead(fd,a->j,nz,BINARY_INT); CHKERRQ(ierr); 184517ab2063SBarry Smith if (shift) { 184617ab2063SBarry Smith for ( i=0; i<nz; i++ ) { 1847416022c9SBarry Smith a->j[i] += 1; 184817ab2063SBarry Smith } 184917ab2063SBarry Smith } 185017ab2063SBarry Smith 185117ab2063SBarry Smith /* read in nonzero values */ 185277c4ece6SBarry Smith ierr = PetscBinaryRead(fd,a->a,nz,BINARY_SCALAR); CHKERRQ(ierr); 185317ab2063SBarry Smith 185417ab2063SBarry Smith /* set matrix "i" values */ 1855416022c9SBarry Smith a->i[0] = -shift; 185617ab2063SBarry Smith for ( i=1; i<= M; i++ ) { 1857416022c9SBarry Smith a->i[i] = a->i[i-1] + rowlengths[i-1]; 1858416022c9SBarry Smith a->ilen[i-1] = rowlengths[i-1]; 185917ab2063SBarry Smith } 18600452661fSBarry Smith PetscFree(rowlengths); 186117ab2063SBarry Smith 18626d4a8577SBarry Smith ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr); 18636d4a8577SBarry Smith ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr); 186417ab2063SBarry Smith return 0; 186517ab2063SBarry Smith } 186617ab2063SBarry Smith 18675615d1e5SSatish Balay #undef __FUNC__ 18685615d1e5SSatish Balay #define __FUNC__ "MatEqual_SeqAIJ" 1869686e14cbSSatish Balay static int MatEqual_SeqAIJ(Mat A,Mat B, PetscTruth* flg) 18707264ac53SSatish Balay { 18717264ac53SSatish Balay Mat_SeqAIJ *a = (Mat_SeqAIJ *)A->data, *b = (Mat_SeqAIJ *)B->data; 18727264ac53SSatish Balay 1873e3372554SBarry Smith if (B->type !=MATSEQAIJ)SETERRQ(1,0,"Matrices must be same type"); 18747264ac53SSatish Balay 18757264ac53SSatish Balay /* If the matrix dimensions are not equal, or no of nonzeros or shift */ 18767264ac53SSatish Balay if ((a->m != b->m ) || (a->n !=b->n) ||( a->nz != b->nz)|| 1877bcd2baecSBarry Smith (a->indexshift != b->indexshift)) { 187877c4ece6SBarry Smith *flg = PETSC_FALSE; return 0; 1879bcd2baecSBarry Smith } 18807264ac53SSatish Balay 18817264ac53SSatish Balay /* if the a->i are the same */ 1882bcd2baecSBarry Smith if (PetscMemcmp(a->i,b->i, (a->n+1)*sizeof(int))) { 188377c4ece6SBarry Smith *flg = PETSC_FALSE; return 0; 18847264ac53SSatish Balay } 18857264ac53SSatish Balay 18867264ac53SSatish Balay /* if a->j are the same */ 1887bcd2baecSBarry Smith if (PetscMemcmp(a->j, b->j, (a->nz)*sizeof(int))) { 188877c4ece6SBarry Smith *flg = PETSC_FALSE; return 0; 1889bcd2baecSBarry Smith } 1890bcd2baecSBarry Smith 1891bcd2baecSBarry Smith /* if a->a are the same */ 189219bcc07fSBarry Smith if (PetscMemcmp(a->a, b->a, (a->nz)*sizeof(Scalar))) { 189377c4ece6SBarry Smith *flg = PETSC_FALSE; return 0; 18947264ac53SSatish Balay } 189577c4ece6SBarry Smith *flg = PETSC_TRUE; 18967264ac53SSatish Balay return 0; 18977264ac53SSatish Balay 18987264ac53SSatish Balay } 1899