1a5eb4965SSatish Balay #ifdef PETSC_RCS_HEADER 2*3a40ed3dSBarry Smith static char vcid[] = "$Id: aij.c,v 1.239 1997/10/06 16:23:56 balay Exp bsmith $"; 317ab2063SBarry Smith #endif 417ab2063SBarry Smith 5d5d45c9bSBarry Smith /* 63369ce9aSBarry Smith Defines the basic matrix operations for the AIJ (compressed row) 7d5d45c9bSBarry Smith matrix storage format. 8d5d45c9bSBarry Smith */ 93369ce9aSBarry Smith 103369ce9aSBarry Smith #include "pinclude/pviewer.h" 113369ce9aSBarry Smith #include "sys.h" 1270f55243SBarry Smith #include "src/mat/impls/aij/seq/aij.h" 13f5eb4b81SSatish Balay #include "src/vec/vecimpl.h" 14f5eb4b81SSatish Balay #include "src/inline/spops.h" 158d195f9aSBarry Smith #include "src/inline/dot.h" 16f5eb4b81SSatish Balay #include "src/inline/bitarray.h" 1717ab2063SBarry Smith 18a2ce50c7SBarry Smith /* 19a2ce50c7SBarry Smith Basic AIJ format ILU based on drop tolerance 20a2ce50c7SBarry Smith */ 215615d1e5SSatish Balay #undef __FUNC__ 225615d1e5SSatish Balay #define __FUNC__ "MatILUDTFactor_SeqAIJ" 23a2ce50c7SBarry Smith int MatILUDTFactor_SeqAIJ(Mat A,double dt,int maxnz,IS row,IS col,Mat *fact) 24a2ce50c7SBarry Smith { 25a2ce50c7SBarry Smith /* Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; */ 26a2ce50c7SBarry Smith int ierr = 1; 27a2ce50c7SBarry Smith 28*3a40ed3dSBarry Smith PetscFunctionBegin; 29e3372554SBarry Smith SETERRQ(ierr,0,"Not implemented"); 30a2ce50c7SBarry Smith } 31a2ce50c7SBarry Smith 32bcd2baecSBarry Smith extern int MatToSymmetricIJ_SeqAIJ(int,int*,int*,int,int,int**,int**); 3317ab2063SBarry Smith 345615d1e5SSatish Balay #undef __FUNC__ 355615d1e5SSatish Balay #define __FUNC__ "MatGetRowIJ_SeqAIJ" 368f6be9afSLois Curfman McInnes int MatGetRowIJ_SeqAIJ(Mat A,int oshift,PetscTruth symmetric,int *m,int **ia,int **ja, 376945ee14SBarry Smith PetscTruth *done) 3817ab2063SBarry Smith { 39416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 406945ee14SBarry Smith int ierr,i,ishift; 4117ab2063SBarry Smith 42*3a40ed3dSBarry Smith PetscFunctionBegin; 4331625ec5SSatish Balay *m = A->m; 44*3a40ed3dSBarry Smith if (!ia) PetscFunctionReturn(0); 456945ee14SBarry Smith ishift = a->indexshift; 466945ee14SBarry Smith if (symmetric) { 4731625ec5SSatish Balay ierr = MatToSymmetricIJ_SeqAIJ(a->m,a->i,a->j,ishift,oshift,ia,ja); CHKERRQ(ierr); 486945ee14SBarry Smith } else if (oshift == 0 && ishift == -1) { 4931625ec5SSatish Balay int nz = a->i[a->m]; 503b2fbd54SBarry Smith /* malloc space and subtract 1 from i and j indices */ 5131625ec5SSatish Balay *ia = (int *) PetscMalloc( (a->m+1)*sizeof(int) ); CHKPTRQ(*ia); 523b2fbd54SBarry Smith *ja = (int *) PetscMalloc( (nz+1)*sizeof(int) ); CHKPTRQ(*ja); 533b2fbd54SBarry Smith for ( i=0; i<nz; i++ ) (*ja)[i] = a->j[i] - 1; 5431625ec5SSatish Balay for ( i=0; i<a->m+1; i++ ) (*ia)[i] = a->i[i] - 1; 556945ee14SBarry Smith } else if (oshift == 1 && ishift == 0) { 5631625ec5SSatish Balay int nz = a->i[a->m] + 1; 573b2fbd54SBarry Smith /* malloc space and add 1 to i and j indices */ 5831625ec5SSatish Balay *ia = (int *) PetscMalloc( (a->m+1)*sizeof(int) ); CHKPTRQ(*ia); 593b2fbd54SBarry Smith *ja = (int *) PetscMalloc( (nz+1)*sizeof(int) ); CHKPTRQ(*ja); 603b2fbd54SBarry Smith for ( i=0; i<nz; i++ ) (*ja)[i] = a->j[i] + 1; 6131625ec5SSatish Balay for ( i=0; i<a->m+1; i++ ) (*ia)[i] = a->i[i] + 1; 626945ee14SBarry Smith } else { 636945ee14SBarry Smith *ia = a->i; *ja = a->j; 64a2ce50c7SBarry Smith } 65a2ce50c7SBarry Smith 66*3a40ed3dSBarry Smith PetscFunctionReturn(0); 67a2744918SBarry Smith } 68a2744918SBarry Smith 695615d1e5SSatish Balay #undef __FUNC__ 705615d1e5SSatish Balay #define __FUNC__ "MatRestoreRowIJ_SeqAIJ" 718f6be9afSLois Curfman McInnes int MatRestoreRowIJ_SeqAIJ(Mat A,int oshift,PetscTruth symmetric,int *n,int **ia,int **ja, 726945ee14SBarry Smith PetscTruth *done) 736945ee14SBarry Smith { 746945ee14SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 753b2fbd54SBarry Smith int ishift = a->indexshift; 766945ee14SBarry Smith 77*3a40ed3dSBarry Smith PetscFunctionBegin; 78*3a40ed3dSBarry Smith if (!ia) PetscFunctionReturn(0); 793b2fbd54SBarry Smith if (symmetric || (oshift == 0 && ishift == -1) || (oshift == 1 && ishift == 0)) { 806945ee14SBarry Smith PetscFree(*ia); 816945ee14SBarry Smith PetscFree(*ja); 82bcd2baecSBarry Smith } 83*3a40ed3dSBarry Smith PetscFunctionReturn(0); 8417ab2063SBarry Smith } 8517ab2063SBarry Smith 865615d1e5SSatish Balay #undef __FUNC__ 875615d1e5SSatish Balay #define __FUNC__ "MatGetColumnIJ_SeqAIJ" 8843a90d84SBarry Smith int MatGetColumnIJ_SeqAIJ(Mat A,int oshift,PetscTruth symmetric,int *nn,int **ia,int **ja, 893b2fbd54SBarry Smith PetscTruth *done) 903b2fbd54SBarry Smith { 913b2fbd54SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 92a93ec695SBarry Smith int ierr,i,ishift = a->indexshift,*collengths,*cia,*cja,n = A->n,m = A->m; 93a93ec695SBarry Smith int nz = a->i[m]+ishift,row,*jj,mr,col; 943b2fbd54SBarry Smith 95*3a40ed3dSBarry Smith PetscFunctionBegin; 963b2fbd54SBarry Smith *nn = A->n; 97*3a40ed3dSBarry Smith if (!ia) PetscFunctionReturn(0); 983b2fbd54SBarry Smith if (symmetric) { 99179192dfSSatish Balay ierr = MatToSymmetricIJ_SeqAIJ(a->m,a->i,a->j,ishift,oshift,ia,ja); CHKERRQ(ierr); 1003b2fbd54SBarry Smith } else { 10161d2ded1SBarry Smith collengths = (int *) PetscMalloc( (n+1)*sizeof(int) ); CHKPTRQ(collengths); 1023b2fbd54SBarry Smith PetscMemzero(collengths,n*sizeof(int)); 1033b2fbd54SBarry Smith cia = (int *) PetscMalloc( (n+1)*sizeof(int) ); CHKPTRQ(cia); 104a93ec695SBarry Smith cja = (int *) PetscMalloc( (nz+1)*sizeof(int) ); CHKPTRQ(cja); 1053b2fbd54SBarry Smith jj = a->j; 1063b2fbd54SBarry Smith for ( i=0; i<nz; i++ ) { 1073b2fbd54SBarry Smith collengths[jj[i] + ishift]++; 1083b2fbd54SBarry Smith } 1093b2fbd54SBarry Smith cia[0] = oshift; 1103b2fbd54SBarry Smith for ( i=0; i<n; i++) { 1113b2fbd54SBarry Smith cia[i+1] = cia[i] + collengths[i]; 1123b2fbd54SBarry Smith } 1133b2fbd54SBarry Smith PetscMemzero(collengths,n*sizeof(int)); 1143b2fbd54SBarry Smith jj = a->j; 115a93ec695SBarry Smith for ( row=0; row<m; row++ ) { 116a93ec695SBarry Smith mr = a->i[row+1] - a->i[row]; 117a93ec695SBarry Smith for ( i=0; i<mr; i++ ) { 1183b2fbd54SBarry Smith col = *jj++ + ishift; 1193b2fbd54SBarry Smith cja[cia[col] + collengths[col]++ - oshift] = row + oshift; 1203b2fbd54SBarry Smith } 1213b2fbd54SBarry Smith } 1223b2fbd54SBarry Smith PetscFree(collengths); 1233b2fbd54SBarry Smith *ia = cia; *ja = cja; 1243b2fbd54SBarry Smith } 1253b2fbd54SBarry Smith 126*3a40ed3dSBarry Smith PetscFunctionReturn(0); 1273b2fbd54SBarry Smith } 1283b2fbd54SBarry Smith 1295615d1e5SSatish Balay #undef __FUNC__ 1305615d1e5SSatish Balay #define __FUNC__ "MatRestoreColumnIJ_SeqAIJ" 13143a90d84SBarry Smith int MatRestoreColumnIJ_SeqAIJ(Mat A,int oshift,PetscTruth symmetric,int *n,int **ia, 1323b2fbd54SBarry Smith int **ja,PetscTruth *done) 1333b2fbd54SBarry Smith { 134*3a40ed3dSBarry Smith PetscFunctionBegin; 135*3a40ed3dSBarry Smith if (!ia) PetscFunctionReturn(0); 1363b2fbd54SBarry Smith 1373b2fbd54SBarry Smith PetscFree(*ia); 1383b2fbd54SBarry Smith PetscFree(*ja); 1393b2fbd54SBarry Smith 140*3a40ed3dSBarry Smith PetscFunctionReturn(0); 1413b2fbd54SBarry Smith } 1423b2fbd54SBarry Smith 143227d817aSBarry Smith #define CHUNKSIZE 15 14417ab2063SBarry Smith 1455615d1e5SSatish Balay #undef __FUNC__ 1465615d1e5SSatish Balay #define __FUNC__ "MatSetValues_SeqAIJ" 14761d2ded1SBarry Smith int MatSetValues_SeqAIJ(Mat A,int m,int *im,int n,int *in,Scalar *v,InsertMode is) 14817ab2063SBarry Smith { 149416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 150416022c9SBarry Smith int *rp,k,low,high,t,ii,row,nrow,i,col,l,rmax, N, sorted = a->sorted; 1514b0e389bSBarry Smith int *imax = a->imax, *ai = a->i, *ailen = a->ilen,roworiented = a->roworiented; 152d5d45c9bSBarry Smith int *aj = a->j, nonew = a->nonew,shift = a->indexshift; 153416022c9SBarry Smith Scalar *ap,value, *aa = a->a; 15417ab2063SBarry Smith 155*3a40ed3dSBarry Smith PetscFunctionBegin; 15617ab2063SBarry Smith for ( k=0; k<m; k++ ) { /* loop over added rows */ 157416022c9SBarry Smith row = im[k]; 158*3a40ed3dSBarry Smith #if defined(USE_PETSC_BOPT_g) 159e3372554SBarry Smith if (row < 0) SETERRQ(1,0,"Negative row"); 160e3372554SBarry Smith if (row >= a->m) SETERRQ(1,0,"Row too large"); 1613b2fbd54SBarry Smith #endif 16217ab2063SBarry Smith rp = aj + ai[row] + shift; ap = aa + ai[row] + shift; 16317ab2063SBarry Smith rmax = imax[row]; nrow = ailen[row]; 164416022c9SBarry Smith low = 0; 16517ab2063SBarry Smith for ( l=0; l<n; l++ ) { /* loop over added columns */ 166*3a40ed3dSBarry Smith #if defined(USE_PETSC_BOPT_g) 167e3372554SBarry Smith if (in[l] < 0) SETERRQ(1,0,"Negative column"); 168e3372554SBarry Smith if (in[l] >= a->n) SETERRQ(1,0,"Column too large"); 1693b2fbd54SBarry Smith #endif 1704b0e389bSBarry Smith col = in[l] - shift; 1714b0e389bSBarry Smith if (roworiented) { 1724b0e389bSBarry Smith value = *v++; 1734b0e389bSBarry Smith } 1744b0e389bSBarry Smith else { 1754b0e389bSBarry Smith value = v[k + l*m]; 1764b0e389bSBarry Smith } 177416022c9SBarry Smith if (!sorted) low = 0; high = nrow; 178416022c9SBarry Smith while (high-low > 5) { 179416022c9SBarry Smith t = (low+high)/2; 180416022c9SBarry Smith if (rp[t] > col) high = t; 181416022c9SBarry Smith else low = t; 18217ab2063SBarry Smith } 183416022c9SBarry Smith for ( i=low; i<high; i++ ) { 18417ab2063SBarry Smith if (rp[i] > col) break; 18517ab2063SBarry Smith if (rp[i] == col) { 186416022c9SBarry Smith if (is == ADD_VALUES) ap[i] += value; 18717ab2063SBarry Smith else ap[i] = value; 18817ab2063SBarry Smith goto noinsert; 18917ab2063SBarry Smith } 19017ab2063SBarry Smith } 191c2653b3dSLois Curfman McInnes if (nonew == 1) goto noinsert; 19211ebbc71SLois Curfman McInnes else if (nonew == -1) SETERRQ(1,0,"Inserting a new nonzero in the matrix"); 19317ab2063SBarry Smith if (nrow >= rmax) { 19417ab2063SBarry Smith /* there is no extra room in row, therefore enlarge */ 195416022c9SBarry Smith int new_nz = ai[a->m] + CHUNKSIZE,len,*new_i,*new_j; 19617ab2063SBarry Smith Scalar *new_a; 19717ab2063SBarry Smith 19811ebbc71SLois Curfman McInnes if (nonew == -2) SETERRQ(1,0,"Inserting a new nonzero in the matrix"); 19996854ed6SLois Curfman McInnes 20017ab2063SBarry Smith /* malloc new storage space */ 201416022c9SBarry Smith len = new_nz*(sizeof(int)+sizeof(Scalar))+(a->m+1)*sizeof(int); 2020452661fSBarry Smith new_a = (Scalar *) PetscMalloc( len ); CHKPTRQ(new_a); 20317ab2063SBarry Smith new_j = (int *) (new_a + new_nz); 20417ab2063SBarry Smith new_i = new_j + new_nz; 20517ab2063SBarry Smith 20617ab2063SBarry Smith /* copy over old data into new slots */ 20717ab2063SBarry Smith for ( ii=0; ii<row+1; ii++ ) {new_i[ii] = ai[ii];} 208416022c9SBarry Smith for ( ii=row+1; ii<a->m+1; ii++ ) {new_i[ii] = ai[ii]+CHUNKSIZE;} 209416022c9SBarry Smith PetscMemcpy(new_j,aj,(ai[row]+nrow+shift)*sizeof(int)); 210416022c9SBarry Smith len = (new_nz - CHUNKSIZE - ai[row] - nrow - shift); 211416022c9SBarry Smith PetscMemcpy(new_j+ai[row]+shift+nrow+CHUNKSIZE,aj+ai[row]+shift+nrow, 21217ab2063SBarry Smith len*sizeof(int)); 213416022c9SBarry Smith PetscMemcpy(new_a,aa,(ai[row]+nrow+shift)*sizeof(Scalar)); 214416022c9SBarry Smith PetscMemcpy(new_a+ai[row]+shift+nrow+CHUNKSIZE,aa+ai[row]+shift+nrow, 21517ab2063SBarry Smith len*sizeof(Scalar)); 21617ab2063SBarry Smith /* free up old matrix storage */ 2170452661fSBarry Smith PetscFree(a->a); 2180452661fSBarry Smith if (!a->singlemalloc) {PetscFree(a->i);PetscFree(a->j);} 219416022c9SBarry Smith aa = a->a = new_a; ai = a->i = new_i; aj = a->j = new_j; 220416022c9SBarry Smith a->singlemalloc = 1; 22117ab2063SBarry Smith 22217ab2063SBarry Smith rp = aj + ai[row] + shift; ap = aa + ai[row] + shift; 223416022c9SBarry Smith rmax = imax[row] = imax[row] + CHUNKSIZE; 224416022c9SBarry Smith PLogObjectMemory(A,CHUNKSIZE*(sizeof(int) + sizeof(Scalar))); 225416022c9SBarry Smith a->maxnz += CHUNKSIZE; 226b810aeb4SBarry Smith a->reallocs++; 22717ab2063SBarry Smith } 228416022c9SBarry Smith N = nrow++ - 1; a->nz++; 229416022c9SBarry Smith /* shift up all the later entries in this row */ 230416022c9SBarry Smith for ( ii=N; ii>=i; ii-- ) { 23117ab2063SBarry Smith rp[ii+1] = rp[ii]; 23217ab2063SBarry Smith ap[ii+1] = ap[ii]; 23317ab2063SBarry Smith } 23417ab2063SBarry Smith rp[i] = col; 23517ab2063SBarry Smith ap[i] = value; 23617ab2063SBarry Smith noinsert:; 237416022c9SBarry Smith low = i + 1; 23817ab2063SBarry Smith } 23917ab2063SBarry Smith ailen[row] = nrow; 24017ab2063SBarry Smith } 241*3a40ed3dSBarry Smith PetscFunctionReturn(0); 24217ab2063SBarry Smith } 24317ab2063SBarry Smith 2445615d1e5SSatish Balay #undef __FUNC__ 2455615d1e5SSatish Balay #define __FUNC__ "MatGetValues_SeqAIJ" 2468f6be9afSLois Curfman McInnes int MatGetValues_SeqAIJ(Mat A,int m,int *im,int n,int *in,Scalar *v) 2477eb43aa7SLois Curfman McInnes { 2487eb43aa7SLois Curfman McInnes Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 249b49de8d1SLois Curfman McInnes int *rp, k, low, high, t, row, nrow, i, col, l, *aj = a->j; 2507eb43aa7SLois Curfman McInnes int *ai = a->i, *ailen = a->ilen, shift = a->indexshift; 2517eb43aa7SLois Curfman McInnes Scalar *ap, *aa = a->a, zero = 0.0; 2527eb43aa7SLois Curfman McInnes 253*3a40ed3dSBarry Smith PetscFunctionBegin; 2547eb43aa7SLois Curfman McInnes for ( k=0; k<m; k++ ) { /* loop over rows */ 2557eb43aa7SLois Curfman McInnes row = im[k]; 256e3372554SBarry Smith if (row < 0) SETERRQ(1,0,"Negative row"); 257e3372554SBarry Smith if (row >= a->m) SETERRQ(1,0,"Row too large"); 2587eb43aa7SLois Curfman McInnes rp = aj + ai[row] + shift; ap = aa + ai[row] + shift; 2597eb43aa7SLois Curfman McInnes nrow = ailen[row]; 2607eb43aa7SLois Curfman McInnes for ( l=0; l<n; l++ ) { /* loop over columns */ 261e3372554SBarry Smith if (in[l] < 0) SETERRQ(1,0,"Negative column"); 262e3372554SBarry Smith if (in[l] >= a->n) SETERRQ(1,0,"Column too large"); 2637eb43aa7SLois Curfman McInnes col = in[l] - shift; 2647eb43aa7SLois Curfman McInnes high = nrow; low = 0; /* assume unsorted */ 2657eb43aa7SLois Curfman McInnes while (high-low > 5) { 2667eb43aa7SLois Curfman McInnes t = (low+high)/2; 2677eb43aa7SLois Curfman McInnes if (rp[t] > col) high = t; 2687eb43aa7SLois Curfman McInnes else low = t; 2697eb43aa7SLois Curfman McInnes } 2707eb43aa7SLois Curfman McInnes for ( i=low; i<high; i++ ) { 2717eb43aa7SLois Curfman McInnes if (rp[i] > col) break; 2727eb43aa7SLois Curfman McInnes if (rp[i] == col) { 273b49de8d1SLois Curfman McInnes *v++ = ap[i]; 2747eb43aa7SLois Curfman McInnes goto finished; 2757eb43aa7SLois Curfman McInnes } 2767eb43aa7SLois Curfman McInnes } 277b49de8d1SLois Curfman McInnes *v++ = zero; 2787eb43aa7SLois Curfman McInnes finished:; 2797eb43aa7SLois Curfman McInnes } 2807eb43aa7SLois Curfman McInnes } 281*3a40ed3dSBarry Smith PetscFunctionReturn(0); 2827eb43aa7SLois Curfman McInnes } 2837eb43aa7SLois Curfman McInnes 28417ab2063SBarry Smith 2855615d1e5SSatish Balay #undef __FUNC__ 2865615d1e5SSatish Balay #define __FUNC__ "MatView_SeqAIJ_Binary" 2878f6be9afSLois Curfman McInnes extern int MatView_SeqAIJ_Binary(Mat A,Viewer viewer) 28817ab2063SBarry Smith { 289416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 290416022c9SBarry Smith int i, fd, *col_lens, ierr; 29117ab2063SBarry Smith 292*3a40ed3dSBarry Smith PetscFunctionBegin; 29390ace30eSBarry Smith ierr = ViewerBinaryGetDescriptor(viewer,&fd); CHKERRQ(ierr); 2940452661fSBarry Smith col_lens = (int *) PetscMalloc( (4+a->m)*sizeof(int) ); CHKPTRQ(col_lens); 295416022c9SBarry Smith col_lens[0] = MAT_COOKIE; 296416022c9SBarry Smith col_lens[1] = a->m; 297416022c9SBarry Smith col_lens[2] = a->n; 298416022c9SBarry Smith col_lens[3] = a->nz; 299416022c9SBarry Smith 300416022c9SBarry Smith /* store lengths of each row and write (including header) to file */ 301416022c9SBarry Smith for ( i=0; i<a->m; i++ ) { 302416022c9SBarry Smith col_lens[4+i] = a->i[i+1] - a->i[i]; 30317ab2063SBarry Smith } 3040752156aSBarry Smith ierr = PetscBinaryWrite(fd,col_lens,4+a->m,PETSC_INT,1); CHKERRQ(ierr); 3050452661fSBarry Smith PetscFree(col_lens); 306416022c9SBarry Smith 307416022c9SBarry Smith /* store column indices (zero start index) */ 308416022c9SBarry Smith if (a->indexshift) { 309416022c9SBarry Smith for ( i=0; i<a->nz; i++ ) a->j[i]--; 31017ab2063SBarry Smith } 3110752156aSBarry Smith ierr = PetscBinaryWrite(fd,a->j,a->nz,PETSC_INT,0); CHKERRQ(ierr); 312416022c9SBarry Smith if (a->indexshift) { 313416022c9SBarry Smith for ( i=0; i<a->nz; i++ ) a->j[i]++; 31417ab2063SBarry Smith } 315416022c9SBarry Smith 316416022c9SBarry Smith /* store nonzero values */ 3170752156aSBarry Smith ierr = PetscBinaryWrite(fd,a->a,a->nz,PETSC_SCALAR,0); CHKERRQ(ierr); 318*3a40ed3dSBarry Smith PetscFunctionReturn(0); 31917ab2063SBarry Smith } 320416022c9SBarry Smith 3215615d1e5SSatish Balay #undef __FUNC__ 3225615d1e5SSatish Balay #define __FUNC__ "MatView_SeqAIJ_ASCII" 3238f6be9afSLois Curfman McInnes extern int MatView_SeqAIJ_ASCII(Mat A,Viewer viewer) 324416022c9SBarry Smith { 325416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 326496e697eSBarry Smith int ierr, i,j, m = a->m, shift = a->indexshift, format, flg1,flg2; 32717ab2063SBarry Smith FILE *fd; 32817ab2063SBarry Smith char *outputname; 32917ab2063SBarry Smith 330*3a40ed3dSBarry Smith PetscFunctionBegin; 33190ace30eSBarry Smith ierr = ViewerASCIIGetPointer(viewer,&fd); CHKERRQ(ierr); 332416022c9SBarry Smith ierr = ViewerFileGetOutputname_Private(viewer,&outputname); CHKERRQ(ierr); 33390ace30eSBarry Smith ierr = ViewerGetFormat(viewer,&format); 334a93ec695SBarry Smith if (format == VIEWER_FORMAT_ASCII_INFO) { 335*3a40ed3dSBarry Smith PetscFunctionReturn(0); 336*3a40ed3dSBarry Smith } else if (format == VIEWER_FORMAT_ASCII_INFO_LONG) { 337496e697eSBarry Smith ierr = OptionsHasName(PETSC_NULL,"-mat_aij_no_inode",&flg1); CHKERRQ(ierr); 338496e697eSBarry Smith ierr = OptionsHasName(PETSC_NULL,"-mat_no_unroll",&flg2); CHKERRQ(ierr); 339496e697eSBarry Smith if (flg1 || flg2) fprintf(fd," not using I-node routines\n"); 34095e01e2fSLois Curfman McInnes else fprintf(fd," using I-node routines: found %d nodes, limit used is %d\n", 34195e01e2fSLois Curfman McInnes a->inode.node_count,a->inode.limit); 342*3a40ed3dSBarry Smith } else if (format == VIEWER_FORMAT_ASCII_MATLAB) { 343d00d2cf4SBarry Smith int nofinalvalue = 0; 344d00d2cf4SBarry Smith if ((a->i[m] == a->i[m-1]) || (a->j[a->nz-1] != a->n-!shift)) { 345d00d2cf4SBarry Smith nofinalvalue = 1; 346d00d2cf4SBarry Smith } 347416022c9SBarry Smith fprintf(fd,"%% Size = %d %d \n",m,a->n); 3484e220ebcSLois Curfman McInnes fprintf(fd,"%% Nonzeros = %d \n",a->nz); 349d00d2cf4SBarry Smith fprintf(fd,"zzz = zeros(%d,3);\n",a->nz+nofinalvalue); 35017ab2063SBarry Smith fprintf(fd,"zzz = [\n"); 35117ab2063SBarry Smith 35217ab2063SBarry Smith for (i=0; i<m; i++) { 353416022c9SBarry Smith for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 354*3a40ed3dSBarry Smith #if defined(USE_PETSC_COMPLEX) 3556945ee14SBarry Smith fprintf(fd,"%d %d %18.16e + %18.16e i \n",i+1,a->j[j]+!shift,real(a->a[j]), 356416022c9SBarry Smith imag(a->a[j])); 35717ab2063SBarry Smith #else 3587a743949SBarry Smith fprintf(fd,"%d %d %18.16e\n", i+1, a->j[j]+!shift, a->a[j]); 35917ab2063SBarry Smith #endif 36017ab2063SBarry Smith } 36117ab2063SBarry Smith } 362d00d2cf4SBarry Smith if (nofinalvalue) { 363d00d2cf4SBarry Smith fprintf(fd,"%d %d %18.16e\n", m, a->n, 0.0); 364d00d2cf4SBarry Smith } 36517ab2063SBarry Smith fprintf(fd,"];\n %s = spconvert(zzz);\n",outputname); 366*3a40ed3dSBarry Smith } else if (format == VIEWER_FORMAT_ASCII_COMMON) { 36744cd7ae7SLois Curfman McInnes for ( i=0; i<m; i++ ) { 36844cd7ae7SLois Curfman McInnes fprintf(fd,"row %d:",i); 36944cd7ae7SLois Curfman McInnes for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 370*3a40ed3dSBarry Smith #if defined(USE_PETSC_COMPLEX) 371766eeae4SLois Curfman McInnes if (imag(a->a[j]) > 0.0 && real(a->a[j]) != 0.0) 37244cd7ae7SLois Curfman McInnes fprintf(fd," %d %g + %g i",a->j[j]+shift,real(a->a[j]),imag(a->a[j])); 373766eeae4SLois Curfman McInnes else if (imag(a->a[j]) < 0.0 && real(a->a[j]) != 0.0) 374766eeae4SLois Curfman McInnes fprintf(fd," %d %g - %g i",a->j[j]+shift,real(a->a[j]),-imag(a->a[j])); 37544cd7ae7SLois Curfman McInnes else if (real(a->a[j]) != 0.0) 37644cd7ae7SLois Curfman McInnes fprintf(fd," %d %g ",a->j[j]+shift,real(a->a[j])); 37744cd7ae7SLois Curfman McInnes #else 37844cd7ae7SLois Curfman McInnes if (a->a[j] != 0.0) fprintf(fd," %d %g ",a->j[j]+shift,a->a[j]); 37944cd7ae7SLois Curfman McInnes #endif 38044cd7ae7SLois Curfman McInnes } 38144cd7ae7SLois Curfman McInnes fprintf(fd,"\n"); 38244cd7ae7SLois Curfman McInnes } 38344cd7ae7SLois Curfman McInnes } 384496be53dSLois Curfman McInnes else if (format == VIEWER_FORMAT_ASCII_SYMMODU) { 385496be53dSLois Curfman McInnes int nzd=0, fshift=1, *sptr; 3862e44a96cSLois Curfman McInnes sptr = (int *) PetscMalloc( (m+1)*sizeof(int) ); CHKPTRQ(sptr); 387496be53dSLois Curfman McInnes for ( i=0; i<m; i++ ) { 388496be53dSLois Curfman McInnes sptr[i] = nzd+1; 389496be53dSLois Curfman McInnes for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 390496be53dSLois Curfman McInnes if (a->j[j] >= i) { 391*3a40ed3dSBarry Smith #if defined(USE_PETSC_COMPLEX) 392496be53dSLois Curfman McInnes if (imag(a->a[j]) != 0.0 || real(a->a[j]) != 0.0) nzd++; 393496be53dSLois Curfman McInnes #else 394496be53dSLois Curfman McInnes if (a->a[j] != 0.0) nzd++; 395496be53dSLois Curfman McInnes #endif 396496be53dSLois Curfman McInnes } 397496be53dSLois Curfman McInnes } 398496be53dSLois Curfman McInnes } 3992e44a96cSLois Curfman McInnes sptr[m] = nzd+1; 400496be53dSLois Curfman McInnes fprintf(fd," %d %d\n\n",m,nzd); 4012e44a96cSLois Curfman McInnes for ( i=0; i<m+1; i+=6 ) { 4022e44a96cSLois Curfman McInnes if (i+4<m) fprintf(fd," %d %d %d %d %d %d\n",sptr[i],sptr[i+1],sptr[i+2],sptr[i+3],sptr[i+4],sptr[i+5]); 4032e44a96cSLois Curfman McInnes else if (i+3<m) fprintf(fd," %d %d %d %d %d\n",sptr[i],sptr[i+1],sptr[i+2],sptr[i+3],sptr[i+4]); 4042e44a96cSLois Curfman McInnes else if (i+2<m) fprintf(fd," %d %d %d %d\n",sptr[i],sptr[i+1],sptr[i+2],sptr[i+3]); 4052e44a96cSLois Curfman McInnes else if (i+1<m) fprintf(fd," %d %d %d\n",sptr[i],sptr[i+1],sptr[i+2]); 4062e44a96cSLois Curfman McInnes else if (i<m) fprintf(fd," %d %d\n",sptr[i],sptr[i+1]); 4077272d637SLois Curfman McInnes else fprintf(fd," %d\n",sptr[i]); 408496be53dSLois Curfman McInnes } 409496be53dSLois Curfman McInnes fprintf(fd,"\n"); 410496be53dSLois Curfman McInnes PetscFree(sptr); 411496be53dSLois Curfman McInnes for ( i=0; i<m; i++ ) { 412496be53dSLois Curfman McInnes for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 413496be53dSLois Curfman McInnes if (a->j[j] >= i) fprintf(fd," %d ",a->j[j]+fshift); 414496be53dSLois Curfman McInnes } 415496be53dSLois Curfman McInnes fprintf(fd,"\n"); 416496be53dSLois Curfman McInnes } 417496be53dSLois Curfman McInnes fprintf(fd,"\n"); 418496be53dSLois Curfman McInnes for ( i=0; i<m; i++ ) { 419496be53dSLois Curfman McInnes for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 420496be53dSLois Curfman McInnes if (a->j[j] >= i) { 421*3a40ed3dSBarry Smith #if defined(USE_PETSC_COMPLEX) 422496be53dSLois Curfman McInnes if (imag(a->a[j]) != 0.0 || real(a->a[j]) != 0.0) 423496be53dSLois Curfman McInnes fprintf(fd," %18.16e %18.16e ",real(a->a[j]),imag(a->a[j])); 424496be53dSLois Curfman McInnes #else 425496be53dSLois Curfman McInnes if (a->a[j] != 0.0) fprintf(fd," %18.16e ",a->a[j]); 426496be53dSLois Curfman McInnes #endif 427496be53dSLois Curfman McInnes } 428496be53dSLois Curfman McInnes } 429496be53dSLois Curfman McInnes fprintf(fd,"\n"); 430496be53dSLois Curfman McInnes } 431*3a40ed3dSBarry Smith } else { 43217ab2063SBarry Smith for ( i=0; i<m; i++ ) { 43317ab2063SBarry Smith fprintf(fd,"row %d:",i); 434416022c9SBarry Smith for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 435*3a40ed3dSBarry Smith #if defined(USE_PETSC_COMPLEX) 436766eeae4SLois Curfman McInnes if (imag(a->a[j]) > 0.0) { 437416022c9SBarry Smith fprintf(fd," %d %g + %g i",a->j[j]+shift,real(a->a[j]),imag(a->a[j])); 438766eeae4SLois Curfman McInnes } else if (imag(a->a[j]) < 0.0) { 439766eeae4SLois Curfman McInnes fprintf(fd," %d %g - %g i",a->j[j]+shift,real(a->a[j]),-imag(a->a[j])); 440*3a40ed3dSBarry Smith } else { 441416022c9SBarry Smith fprintf(fd," %d %g ",a->j[j]+shift,real(a->a[j])); 44217ab2063SBarry Smith } 44317ab2063SBarry Smith #else 444416022c9SBarry Smith fprintf(fd," %d %g ",a->j[j]+shift,a->a[j]); 44517ab2063SBarry Smith #endif 44617ab2063SBarry Smith } 44717ab2063SBarry Smith fprintf(fd,"\n"); 44817ab2063SBarry Smith } 44917ab2063SBarry Smith } 45017ab2063SBarry Smith fflush(fd); 451*3a40ed3dSBarry Smith PetscFunctionReturn(0); 452416022c9SBarry Smith } 453416022c9SBarry Smith 4545615d1e5SSatish Balay #undef __FUNC__ 4555615d1e5SSatish Balay #define __FUNC__ "MatView_SeqAIJ_Draw" 4568f6be9afSLois Curfman McInnes extern int MatView_SeqAIJ_Draw(Mat A,Viewer viewer) 457416022c9SBarry Smith { 458416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 459cddf8d76SBarry Smith int ierr, i,j, m = a->m, shift = a->indexshift,pause,color; 4600513a670SBarry Smith int format; 46194a9d846SBarry Smith double xl,yl,xr,yr,w,h,xc,yc,scale = 1.0,x_l,x_r,y_l,y_r,maxv = 0.0; 462bcd2baecSBarry Smith Draw draw; 463cddf8d76SBarry Smith DrawButton button; 46419bcc07fSBarry Smith PetscTruth isnull; 465cddf8d76SBarry Smith 466*3a40ed3dSBarry Smith PetscFunctionBegin; 4670513a670SBarry Smith ierr = ViewerDrawGetDraw(viewer,&draw); CHKERRQ(ierr); 4680513a670SBarry Smith ierr = DrawClear(draw); CHKERRQ(ierr); 4690513a670SBarry Smith ierr = ViewerGetFormat(viewer,&format); CHKERRQ(ierr); 470*3a40ed3dSBarry Smith ierr = DrawIsNull(draw,&isnull); CHKERRQ(ierr); if (isnull) PetscFunctionReturn(0); 47119bcc07fSBarry Smith 472416022c9SBarry Smith xr = a->n; yr = a->m; h = yr/10.0; w = xr/10.0; 473416022c9SBarry Smith xr += w; yr += h; xl = -w; yl = -h; 474416022c9SBarry Smith ierr = DrawSetCoordinates(draw,xl,yl,xr,yr); CHKERRQ(ierr); 475416022c9SBarry Smith /* loop over matrix elements drawing boxes */ 4760513a670SBarry Smith 4770513a670SBarry Smith if (format != VIEWER_FORMAT_DRAW_CONTOUR) { 4780513a670SBarry Smith /* Blue for negative, Cyan for zero and Red for positive */ 479cddf8d76SBarry Smith color = DRAW_BLUE; 480416022c9SBarry Smith for ( i=0; i<m; i++ ) { 481cddf8d76SBarry Smith y_l = m - i - 1.0; y_r = y_l + 1.0; 482416022c9SBarry Smith for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 483cddf8d76SBarry Smith x_l = a->j[j] + shift; x_r = x_l + 1.0; 484*3a40ed3dSBarry Smith #if defined(USE_PETSC_COMPLEX) 485cddf8d76SBarry Smith if (real(a->a[j]) >= 0.) continue; 486cddf8d76SBarry Smith #else 487cddf8d76SBarry Smith if (a->a[j] >= 0.) continue; 488cddf8d76SBarry Smith #endif 489cddf8d76SBarry Smith DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color); 490cddf8d76SBarry Smith } 491cddf8d76SBarry Smith } 492cddf8d76SBarry Smith color = DRAW_CYAN; 493cddf8d76SBarry Smith for ( i=0; i<m; i++ ) { 494cddf8d76SBarry Smith y_l = m - i - 1.0; y_r = y_l + 1.0; 495cddf8d76SBarry Smith for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 496cddf8d76SBarry Smith x_l = a->j[j] + shift; x_r = x_l + 1.0; 497cddf8d76SBarry Smith if (a->a[j] != 0.) continue; 498cddf8d76SBarry Smith DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color); 499cddf8d76SBarry Smith } 500cddf8d76SBarry Smith } 501cddf8d76SBarry Smith color = DRAW_RED; 502cddf8d76SBarry Smith for ( i=0; i<m; i++ ) { 503cddf8d76SBarry Smith y_l = m - i - 1.0; y_r = y_l + 1.0; 504cddf8d76SBarry Smith for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 505cddf8d76SBarry Smith x_l = a->j[j] + shift; x_r = x_l + 1.0; 506*3a40ed3dSBarry Smith #if defined(USE_PETSC_COMPLEX) 507cddf8d76SBarry Smith if (real(a->a[j]) <= 0.) continue; 508cddf8d76SBarry Smith #else 509cddf8d76SBarry Smith if (a->a[j] <= 0.) continue; 510cddf8d76SBarry Smith #endif 511cddf8d76SBarry Smith DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color); 512416022c9SBarry Smith } 513416022c9SBarry Smith } 5140513a670SBarry Smith } else { 5150513a670SBarry Smith /* use contour shading to indicate magnitude of values */ 5160513a670SBarry Smith /* first determine max of all nonzero values */ 5170513a670SBarry Smith int nz = a->nz,count; 5180513a670SBarry Smith Draw popup; 5190513a670SBarry Smith 5200513a670SBarry Smith for ( i=0; i<nz; i++ ) { 5210513a670SBarry Smith if (PetscAbsScalar(a->a[i]) > maxv) maxv = PetscAbsScalar(a->a[i]); 5220513a670SBarry Smith } 5230513a670SBarry Smith ierr = DrawCreatePopUp(draw,&popup); CHKERRQ(ierr); 5240513a670SBarry Smith ierr = DrawScalePopup(popup,0.0,maxv); CHKERRQ(ierr); 5250513a670SBarry Smith count = 0; 5260513a670SBarry Smith for ( i=0; i<m; i++ ) { 5270513a670SBarry Smith y_l = m - i - 1.0; y_r = y_l + 1.0; 5280513a670SBarry Smith for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 5290513a670SBarry Smith x_l = a->j[j] + shift; x_r = x_l + 1.0; 5300513a670SBarry Smith color = 32 + (int) ((200.0 - 32.0)*PetscAbsScalar(a->a[count])/maxv); 5310513a670SBarry Smith DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color); 5320513a670SBarry Smith count++; 5330513a670SBarry Smith } 5340513a670SBarry Smith } 5350513a670SBarry Smith } 536416022c9SBarry Smith DrawFlush(draw); 537cddf8d76SBarry Smith DrawGetPause(draw,&pause); 538*3a40ed3dSBarry Smith if (pause >= 0) { PetscSleep(pause); PetscFunctionReturn(0);} 539cddf8d76SBarry Smith 540cddf8d76SBarry Smith /* allow the matrix to zoom or shrink */ 5416945ee14SBarry Smith ierr = DrawCheckResizedWindow(draw); 542cddf8d76SBarry Smith ierr = DrawGetMouseButton(draw,&button,&xc,&yc,0,0); 543cddf8d76SBarry Smith while (button != BUTTON_RIGHT) { 544cddf8d76SBarry Smith DrawClear(draw); 545cddf8d76SBarry Smith if (button == BUTTON_LEFT) scale = .5; 546cddf8d76SBarry Smith else if (button == BUTTON_CENTER) scale = 2.; 547cddf8d76SBarry Smith xl = scale*(xl + w - xc) + xc - w*scale; 548cddf8d76SBarry Smith xr = scale*(xr - w - xc) + xc + w*scale; 549cddf8d76SBarry Smith yl = scale*(yl + h - yc) + yc - h*scale; 550cddf8d76SBarry Smith yr = scale*(yr - h - yc) + yc + h*scale; 551cddf8d76SBarry Smith w *= scale; h *= scale; 552cddf8d76SBarry Smith ierr = DrawSetCoordinates(draw,xl,yl,xr,yr); CHKERRQ(ierr); 5530513a670SBarry Smith if (format != VIEWER_FORMAT_DRAW_CONTOUR) { 5540513a670SBarry Smith /* Blue for negative, Cyan for zero and Red for positive */ 555cddf8d76SBarry Smith color = DRAW_BLUE; 556cddf8d76SBarry Smith for ( i=0; i<m; i++ ) { 557cddf8d76SBarry Smith y_l = m - i - 1.0; y_r = y_l + 1.0; 558cddf8d76SBarry Smith for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 559cddf8d76SBarry Smith x_l = a->j[j] + shift; x_r = x_l + 1.0; 560*3a40ed3dSBarry Smith #if defined(USE_PETSC_COMPLEX) 561cddf8d76SBarry Smith if (real(a->a[j]) >= 0.) continue; 562cddf8d76SBarry Smith #else 563cddf8d76SBarry Smith if (a->a[j] >= 0.) continue; 564cddf8d76SBarry Smith #endif 565cddf8d76SBarry Smith DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color); 566cddf8d76SBarry Smith } 567cddf8d76SBarry Smith } 568cddf8d76SBarry Smith color = DRAW_CYAN; 569cddf8d76SBarry Smith for ( i=0; i<m; i++ ) { 570cddf8d76SBarry Smith y_l = m - i - 1.0; y_r = y_l + 1.0; 571cddf8d76SBarry Smith for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 572cddf8d76SBarry Smith x_l = a->j[j] + shift; x_r = x_l + 1.0; 573cddf8d76SBarry Smith if (a->a[j] != 0.) continue; 574cddf8d76SBarry Smith DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color); 575cddf8d76SBarry Smith } 576cddf8d76SBarry Smith } 577cddf8d76SBarry Smith color = DRAW_RED; 578cddf8d76SBarry Smith for ( i=0; i<m; i++ ) { 579cddf8d76SBarry Smith y_l = m - i - 1.0; y_r = y_l + 1.0; 580cddf8d76SBarry Smith for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 581cddf8d76SBarry Smith x_l = a->j[j] + shift; x_r = x_l + 1.0; 582*3a40ed3dSBarry Smith #if defined(USE_PETSC_COMPLEX) 583cddf8d76SBarry Smith if (real(a->a[j]) <= 0.) continue; 584cddf8d76SBarry Smith #else 585cddf8d76SBarry Smith if (a->a[j] <= 0.) continue; 586cddf8d76SBarry Smith #endif 587cddf8d76SBarry Smith DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color); 588cddf8d76SBarry Smith } 589cddf8d76SBarry Smith } 5900513a670SBarry Smith } else { 5910513a670SBarry Smith /* use contour shading to indicate magnitude of values */ 5920513a670SBarry Smith int count = 0; 5930513a670SBarry Smith for ( i=0; i<m; i++ ) { 5940513a670SBarry Smith y_l = m - i - 1.0; y_r = y_l + 1.0; 5950513a670SBarry Smith for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 5960513a670SBarry Smith x_l = a->j[j] + shift; x_r = x_l + 1.0; 5970513a670SBarry Smith color = 32 + (int) ((200.0 - 32.0)*PetscAbsScalar(a->a[count])/maxv); 598*3a40ed3dSBarry Smith DrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color); CHKERRQ(ierr); 5990513a670SBarry Smith count++; 6000513a670SBarry Smith } 6010513a670SBarry Smith } 6020513a670SBarry Smith } 6030513a670SBarry Smith 604*3a40ed3dSBarry Smith ierr = DrawCheckResizedWindow(draw); CHKERRQ(ierr); 605*3a40ed3dSBarry Smith ierr = DrawGetMouseButton(draw,&button,&xc,&yc,0,0); CHKERRQ(ierr); 606cddf8d76SBarry Smith } 607*3a40ed3dSBarry Smith PetscFunctionReturn(0); 608416022c9SBarry Smith } 609416022c9SBarry Smith 6105615d1e5SSatish Balay #undef __FUNC__ 611d4bb536fSBarry Smith #define __FUNC__ "MatView_SeqAIJ" 6128f6be9afSLois Curfman McInnes int MatView_SeqAIJ(PetscObject obj,Viewer viewer) 613416022c9SBarry Smith { 614416022c9SBarry Smith Mat A = (Mat) obj; 615416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ*) A->data; 616bcd2baecSBarry Smith ViewerType vtype; 617bcd2baecSBarry Smith int ierr; 618416022c9SBarry Smith 619*3a40ed3dSBarry Smith PetscFunctionBegin; 620bcd2baecSBarry Smith ierr = ViewerGetType(viewer,&vtype); CHKERRQ(ierr); 621bcd2baecSBarry Smith if (vtype == MATLAB_VIEWER) { 622*3a40ed3dSBarry Smith ierr = ViewerMatlabPutSparse_Private(viewer,a->m,a->n,a->nz,a->a,a->i,a->j); CHKERRQ(ierr); 623416022c9SBarry Smith } 624bcd2baecSBarry Smith else if (vtype == ASCII_FILE_VIEWER || vtype == ASCII_FILES_VIEWER){ 625*3a40ed3dSBarry Smith ierr = MatView_SeqAIJ_ASCII(A,viewer); CHKERRQ(ierr); 626416022c9SBarry Smith } 627bcd2baecSBarry Smith else if (vtype == BINARY_FILE_VIEWER) { 628*3a40ed3dSBarry Smith ierr = MatView_SeqAIJ_Binary(A,viewer); CHKERRQ(ierr); 629416022c9SBarry Smith } 630bcd2baecSBarry Smith else if (vtype == DRAW_VIEWER) { 631*3a40ed3dSBarry Smith ierr = MatView_SeqAIJ_Draw(A,viewer); CHKERRQ(ierr); 63217ab2063SBarry Smith } 633*3a40ed3dSBarry Smith PetscFunctionReturn(0); 63417ab2063SBarry Smith } 63519bcc07fSBarry Smith 636c456f294SBarry Smith extern int Mat_AIJ_CheckInode(Mat); 6375615d1e5SSatish Balay #undef __FUNC__ 6385615d1e5SSatish Balay #define __FUNC__ "MatAssemblyEnd_SeqAIJ" 6398f6be9afSLois Curfman McInnes int MatAssemblyEnd_SeqAIJ(Mat A,MatAssemblyType mode) 64017ab2063SBarry Smith { 641416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 64241c01911SSatish Balay int fshift = 0,i,j,*ai = a->i, *aj = a->j, *imax = a->imax,ierr; 64343ee02c3SBarry Smith int m = a->m, *ip, N, *ailen = a->ilen,shift = a->indexshift,rmax = 0; 644416022c9SBarry Smith Scalar *aa = a->a, *ap; 64517ab2063SBarry Smith 646*3a40ed3dSBarry Smith PetscFunctionBegin; 647*3a40ed3dSBarry Smith if (mode == MAT_FLUSH_ASSEMBLY) PetscFunctionReturn(0); 64817ab2063SBarry Smith 64943ee02c3SBarry Smith if (m) rmax = ailen[0]; /* determine row with most nonzeros */ 65017ab2063SBarry Smith for ( i=1; i<m; i++ ) { 651416022c9SBarry Smith /* move each row back by the amount of empty slots (fshift) before it*/ 65217ab2063SBarry Smith fshift += imax[i-1] - ailen[i-1]; 65394a9d846SBarry Smith rmax = PetscMax(rmax,ailen[i]); 65417ab2063SBarry Smith if (fshift) { 655416022c9SBarry Smith ip = aj + ai[i] + shift; ap = aa + ai[i] + shift; 65617ab2063SBarry Smith N = ailen[i]; 65717ab2063SBarry Smith for ( j=0; j<N; j++ ) { 65817ab2063SBarry Smith ip[j-fshift] = ip[j]; 65917ab2063SBarry Smith ap[j-fshift] = ap[j]; 66017ab2063SBarry Smith } 66117ab2063SBarry Smith } 66217ab2063SBarry Smith ai[i] = ai[i-1] + ailen[i-1]; 66317ab2063SBarry Smith } 66417ab2063SBarry Smith if (m) { 66517ab2063SBarry Smith fshift += imax[m-1] - ailen[m-1]; 66617ab2063SBarry Smith ai[m] = ai[m-1] + ailen[m-1]; 66717ab2063SBarry Smith } 66817ab2063SBarry Smith /* reset ilen and imax for each row */ 66917ab2063SBarry Smith for ( i=0; i<m; i++ ) { 67017ab2063SBarry Smith ailen[i] = imax[i] = ai[i+1] - ai[i]; 67117ab2063SBarry Smith } 672416022c9SBarry Smith a->nz = ai[m] + shift; 67317ab2063SBarry Smith 67417ab2063SBarry Smith /* diagonals may have moved, so kill the diagonal pointers */ 675416022c9SBarry Smith if (fshift && a->diag) { 6760452661fSBarry Smith PetscFree(a->diag); 677416022c9SBarry Smith PLogObjectMemory(A,-(m+1)*sizeof(int)); 678416022c9SBarry Smith a->diag = 0; 67917ab2063SBarry Smith } 6804e220ebcSLois Curfman McInnes PLogInfo(A,"MatAssemblyEnd_SeqAIJ:Matrix size: %d X %d; storage space: %d unneeded, %d used\n", 6814e220ebcSLois Curfman McInnes m,a->n,fshift,a->nz); 6824e220ebcSLois Curfman McInnes PLogInfo(A,"MatAssemblyEnd_SeqAIJ:Number of mallocs during MatSetValues is %d\n", 683b810aeb4SBarry Smith a->reallocs); 68494a9d846SBarry Smith PLogInfo(A,"MatAssemblyEnd_SeqAIJ:Most nonzeros in any row is %d\n",rmax); 685dd5f02e7SSatish Balay a->reallocs = 0; 6864e220ebcSLois Curfman McInnes A->info.nz_unneeded = (double)fshift; 6874e220ebcSLois Curfman McInnes 68876dd722bSSatish Balay /* check out for identical nodes. If found, use inode functions */ 68941c01911SSatish Balay ierr = Mat_AIJ_CheckInode(A); CHKERRQ(ierr); 690*3a40ed3dSBarry Smith PetscFunctionReturn(0); 69117ab2063SBarry Smith } 69217ab2063SBarry Smith 6935615d1e5SSatish Balay #undef __FUNC__ 6945615d1e5SSatish Balay #define __FUNC__ "MatZeroEntries_SeqAIJ" 6958f6be9afSLois Curfman McInnes int MatZeroEntries_SeqAIJ(Mat A) 69617ab2063SBarry Smith { 697416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 698*3a40ed3dSBarry Smith 699*3a40ed3dSBarry Smith PetscFunctionBegin; 700cddf8d76SBarry Smith PetscMemzero(a->a,(a->i[a->m]+a->indexshift)*sizeof(Scalar)); 701*3a40ed3dSBarry Smith PetscFunctionReturn(0); 70217ab2063SBarry Smith } 703416022c9SBarry Smith 7045615d1e5SSatish Balay #undef __FUNC__ 7055615d1e5SSatish Balay #define __FUNC__ "MatDestroy_SeqAIJ" 70617ab2063SBarry Smith int MatDestroy_SeqAIJ(PetscObject obj) 70717ab2063SBarry Smith { 708416022c9SBarry Smith Mat A = (Mat) obj; 709416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 710d5d45c9bSBarry Smith 711*3a40ed3dSBarry Smith PetscFunctionBegin; 712*3a40ed3dSBarry Smith #if defined(USE_PETSC_LOG) 713416022c9SBarry Smith PLogObjectState(obj,"Rows=%d, Cols=%d, NZ=%d",a->m,a->n,a->nz); 71417ab2063SBarry Smith #endif 7150452661fSBarry Smith PetscFree(a->a); 7160452661fSBarry Smith if (!a->singlemalloc) { PetscFree(a->i); PetscFree(a->j);} 7170452661fSBarry Smith if (a->diag) PetscFree(a->diag); 7180452661fSBarry Smith if (a->ilen) PetscFree(a->ilen); 7190452661fSBarry Smith if (a->imax) PetscFree(a->imax); 7200452661fSBarry Smith if (a->solve_work) PetscFree(a->solve_work); 72176dd722bSSatish Balay if (a->inode.size) PetscFree(a->inode.size); 7220452661fSBarry Smith PetscFree(a); 723eed86810SBarry Smith 724f2655603SLois Curfman McInnes PLogObjectDestroy(A); 725f2655603SLois Curfman McInnes PetscHeaderDestroy(A); 726*3a40ed3dSBarry Smith PetscFunctionReturn(0); 72717ab2063SBarry Smith } 72817ab2063SBarry Smith 7295615d1e5SSatish Balay #undef __FUNC__ 7305615d1e5SSatish Balay #define __FUNC__ "MatCompress_SeqAIJ" 7318f6be9afSLois Curfman McInnes int MatCompress_SeqAIJ(Mat A) 73217ab2063SBarry Smith { 733*3a40ed3dSBarry Smith PetscFunctionBegin; 734*3a40ed3dSBarry Smith PetscFunctionReturn(0); 73517ab2063SBarry Smith } 73617ab2063SBarry Smith 7375615d1e5SSatish Balay #undef __FUNC__ 7385615d1e5SSatish Balay #define __FUNC__ "MatSetOption_SeqAIJ" 7398f6be9afSLois Curfman McInnes int MatSetOption_SeqAIJ(Mat A,MatOption op) 74017ab2063SBarry Smith { 741416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 742*3a40ed3dSBarry Smith 743*3a40ed3dSBarry Smith PetscFunctionBegin; 7446d4a8577SBarry Smith if (op == MAT_ROW_ORIENTED) a->roworiented = 1; 7456d4a8577SBarry Smith else if (op == MAT_COLUMN_ORIENTED) a->roworiented = 0; 7466d4a8577SBarry Smith else if (op == MAT_COLUMNS_SORTED) a->sorted = 1; 747219d9a1aSLois Curfman McInnes else if (op == MAT_COLUMNS_UNSORTED) a->sorted = 0; 7486d4a8577SBarry Smith else if (op == MAT_NO_NEW_NONZERO_LOCATIONS) a->nonew = 1; 749c2653b3dSLois Curfman McInnes else if (op == MAT_NEW_NONZERO_LOCATION_ERROR) a->nonew = -1; 75096854ed6SLois Curfman McInnes else if (op == MAT_NEW_NONZERO_ALLOCATION_ERROR) a->nonew = -2; 7516d4a8577SBarry Smith else if (op == MAT_YES_NEW_NONZERO_LOCATIONS) a->nonew = 0; 7526d4a8577SBarry Smith else if (op == MAT_ROWS_SORTED || 753219d9a1aSLois Curfman McInnes op == MAT_ROWS_UNSORTED || 7546d4a8577SBarry Smith op == MAT_SYMMETRIC || 7556d4a8577SBarry Smith op == MAT_STRUCTURALLY_SYMMETRIC || 75690f02eecSBarry Smith op == MAT_YES_NEW_DIAGONALS || 7572b362799SSatish Balay op == MAT_IGNORE_OFF_PROC_ENTRIES) 75894a424c1SBarry Smith PLogInfo(A,"Info:MatSetOption_SeqAIJ:Option ignored\n"); 759*3a40ed3dSBarry Smith else if (op == MAT_NO_NEW_DIAGONALS) { 760*3a40ed3dSBarry Smith SETERRQ(PETSC_ERR_SUP,0,"MAT_NO_NEW_DIAGONALS"); 761*3a40ed3dSBarry Smith } else if (op == MAT_INODE_LIMIT_1) a->inode.limit = 1; 7626d4a8577SBarry Smith else if (op == MAT_INODE_LIMIT_2) a->inode.limit = 2; 7636d4a8577SBarry Smith else if (op == MAT_INODE_LIMIT_3) a->inode.limit = 3; 7646d4a8577SBarry Smith else if (op == MAT_INODE_LIMIT_4) a->inode.limit = 4; 7656d4a8577SBarry Smith else if (op == MAT_INODE_LIMIT_5) a->inode.limit = 5; 766*3a40ed3dSBarry Smith else SETERRQ(PETSC_ERR_SUP,0,"unknown option"); 767*3a40ed3dSBarry Smith PetscFunctionReturn(0); 76817ab2063SBarry Smith } 76917ab2063SBarry Smith 7705615d1e5SSatish Balay #undef __FUNC__ 7715615d1e5SSatish Balay #define __FUNC__ "MatGetDiagonal_SeqAIJ" 7728f6be9afSLois Curfman McInnes int MatGetDiagonal_SeqAIJ(Mat A,Vec v) 77317ab2063SBarry Smith { 774416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 775*3a40ed3dSBarry Smith int i,j, n,shift = a->indexshift,ierr; 77617ab2063SBarry Smith Scalar *x, zero = 0.0; 77717ab2063SBarry Smith 778*3a40ed3dSBarry Smith PetscFunctionBegin; 779*3a40ed3dSBarry Smith ierr = VecSet(&zero,v);CHKERRQ(ierr); 78090f02eecSBarry Smith VecGetArray_Fast(v,x); VecGetLocalSize(v,&n); 781e3372554SBarry Smith if (n != a->m) SETERRQ(1,0,"Nonconforming matrix and vector"); 782416022c9SBarry Smith for ( i=0; i<a->m; i++ ) { 783416022c9SBarry Smith for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 784416022c9SBarry Smith if (a->j[j]+shift == i) { 785416022c9SBarry Smith x[i] = a->a[j]; 78617ab2063SBarry Smith break; 78717ab2063SBarry Smith } 78817ab2063SBarry Smith } 78917ab2063SBarry Smith } 790*3a40ed3dSBarry Smith PetscFunctionReturn(0); 79117ab2063SBarry Smith } 79217ab2063SBarry Smith 79317ab2063SBarry Smith /* -------------------------------------------------------*/ 79417ab2063SBarry Smith /* Should check that shapes of vectors and matrices match */ 79517ab2063SBarry Smith /* -------------------------------------------------------*/ 7965615d1e5SSatish Balay #undef __FUNC__ 7975615d1e5SSatish Balay #define __FUNC__ "MatMultTrans_SeqAIJ" 79844cd7ae7SLois Curfman McInnes int MatMultTrans_SeqAIJ(Mat A,Vec xx,Vec yy) 79917ab2063SBarry Smith { 800416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 80117ab2063SBarry Smith Scalar *x, *y, *v, alpha; 802416022c9SBarry Smith int m = a->m, n, i, *idx, shift = a->indexshift; 80317ab2063SBarry Smith 804*3a40ed3dSBarry Smith PetscFunctionBegin; 80590f02eecSBarry Smith VecGetArray_Fast(xx,x); VecGetArray_Fast(yy,y); 806cddf8d76SBarry Smith PetscMemzero(y,a->n*sizeof(Scalar)); 80717ab2063SBarry Smith y = y + shift; /* shift for Fortran start by 1 indexing */ 80817ab2063SBarry Smith for ( i=0; i<m; i++ ) { 809416022c9SBarry Smith idx = a->j + a->i[i] + shift; 810416022c9SBarry Smith v = a->a + a->i[i] + shift; 811416022c9SBarry Smith n = a->i[i+1] - a->i[i]; 81217ab2063SBarry Smith alpha = x[i]; 81317ab2063SBarry Smith while (n-->0) {y[*idx++] += alpha * *v++;} 81417ab2063SBarry Smith } 815416022c9SBarry Smith PLogFlops(2*a->nz - a->n); 816*3a40ed3dSBarry Smith PetscFunctionReturn(0); 81717ab2063SBarry Smith } 818d5d45c9bSBarry Smith 8195615d1e5SSatish Balay #undef __FUNC__ 8205615d1e5SSatish Balay #define __FUNC__ "MatMultTransAdd_SeqAIJ" 82144cd7ae7SLois Curfman McInnes int MatMultTransAdd_SeqAIJ(Mat A,Vec xx,Vec zz,Vec yy) 82217ab2063SBarry Smith { 823416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 82417ab2063SBarry Smith Scalar *x, *y, *v, alpha; 825416022c9SBarry Smith int m = a->m, n, i, *idx,shift = a->indexshift; 82617ab2063SBarry Smith 827*3a40ed3dSBarry Smith PetscFunctionBegin; 82890f02eecSBarry Smith VecGetArray_Fast(xx,x); VecGetArray_Fast(yy,y); 82917ab2063SBarry Smith if (zz != yy) VecCopy(zz,yy); 83017ab2063SBarry Smith y = y + shift; /* shift for Fortran start by 1 indexing */ 83117ab2063SBarry Smith for ( i=0; i<m; i++ ) { 832416022c9SBarry Smith idx = a->j + a->i[i] + shift; 833416022c9SBarry Smith v = a->a + a->i[i] + shift; 834416022c9SBarry Smith n = a->i[i+1] - a->i[i]; 83517ab2063SBarry Smith alpha = x[i]; 83617ab2063SBarry Smith while (n-->0) {y[*idx++] += alpha * *v++;} 83717ab2063SBarry Smith } 83890f02eecSBarry Smith PLogFlops(2*a->nz); 839*3a40ed3dSBarry Smith PetscFunctionReturn(0); 84017ab2063SBarry Smith } 84117ab2063SBarry Smith 8425615d1e5SSatish Balay #undef __FUNC__ 8435615d1e5SSatish Balay #define __FUNC__ "MatMult_SeqAIJ" 84444cd7ae7SLois Curfman McInnes int MatMult_SeqAIJ(Mat A,Vec xx,Vec yy) 84517ab2063SBarry Smith { 846416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 84717ab2063SBarry Smith Scalar *x, *y, *v, sum; 8489ea0dfa2SSatish Balay int m = a->m, n, i, *idx, shift = a->indexshift,*ii,jrow,j; 84917ab2063SBarry Smith 850*3a40ed3dSBarry Smith PetscFunctionBegin; 85190f02eecSBarry Smith VecGetArray_Fast(xx,x); VecGetArray_Fast(yy,y); 85217ab2063SBarry Smith x = x + shift; /* shift for Fortran start by 1 indexing */ 853416022c9SBarry Smith idx = a->j; 8549b88f4a7SBarry Smith v = a->a + shift; /* shift for Fortran start by 1 indexing */ 855416022c9SBarry Smith ii = a->i; 8568d195f9aSBarry Smith #if defined(USE_FORTRAN_KERNELS) 8578d195f9aSBarry Smith fortranmultaij_(&m,x,ii,idx,v,y); 8588d195f9aSBarry Smith #else 85917ab2063SBarry Smith for ( i=0; i<m; i++ ) { 8609ea0dfa2SSatish Balay jrow = ii[i]; 8619ea0dfa2SSatish Balay n = ii[i+1] - jrow; 86217ab2063SBarry Smith sum = 0.0; 8639ea0dfa2SSatish Balay for ( j=0; j<n; j++) { 8649ea0dfa2SSatish Balay sum += v[jrow]*x[idx[jrow]]; jrow++; 8659ea0dfa2SSatish Balay } 86617ab2063SBarry Smith y[i] = sum; 86717ab2063SBarry Smith } 8688d195f9aSBarry Smith #endif 869416022c9SBarry Smith PLogFlops(2*a->nz - m); 870*3a40ed3dSBarry Smith PetscFunctionReturn(0); 87117ab2063SBarry Smith } 87217ab2063SBarry Smith 8735615d1e5SSatish Balay #undef __FUNC__ 8745615d1e5SSatish Balay #define __FUNC__ "MatMultAdd_SeqAIJ" 87544cd7ae7SLois Curfman McInnes int MatMultAdd_SeqAIJ(Mat A,Vec xx,Vec yy,Vec zz) 87617ab2063SBarry Smith { 877416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 87817ab2063SBarry Smith Scalar *x, *y, *z, *v, sum; 8799ea0dfa2SSatish Balay int m = a->m, n, i, *idx, shift = a->indexshift,*ii,jrow,j; 8809ea0dfa2SSatish Balay 88117ab2063SBarry Smith 882*3a40ed3dSBarry Smith PetscFunctionBegin; 88390f02eecSBarry Smith VecGetArray_Fast(xx,x); VecGetArray_Fast(yy,y); VecGetArray_Fast(zz,z); 88417ab2063SBarry Smith x = x + shift; /* shift for Fortran start by 1 indexing */ 885cddf8d76SBarry Smith idx = a->j; 8869b88f4a7SBarry Smith v = a->a + shift; /* shift for Fortran start by 1 indexing */ 887cddf8d76SBarry Smith ii = a->i; 88802ab625aSSatish Balay #if defined(USE_FORTRAN_KERNELS) 88902ab625aSSatish Balay fortranmultaddaij_(&m,x,ii,idx,v,y,z); 89002ab625aSSatish Balay #else 89117ab2063SBarry Smith for ( i=0; i<m; i++ ) { 8929ea0dfa2SSatish Balay jrow = ii[i]; 8939ea0dfa2SSatish Balay n = ii[i+1] - jrow; 89417ab2063SBarry Smith sum = y[i]; 8959ea0dfa2SSatish Balay for ( j=0; j<n; j++) { 8969ea0dfa2SSatish Balay sum += v[jrow]*x[idx[jrow]]; jrow++; 8979ea0dfa2SSatish Balay } 89817ab2063SBarry Smith z[i] = sum; 89917ab2063SBarry Smith } 90002ab625aSSatish Balay #endif 901416022c9SBarry Smith PLogFlops(2*a->nz); 902*3a40ed3dSBarry Smith PetscFunctionReturn(0); 90317ab2063SBarry Smith } 90417ab2063SBarry Smith 90517ab2063SBarry Smith /* 90617ab2063SBarry Smith Adds diagonal pointers to sparse matrix structure. 90717ab2063SBarry Smith */ 90817ab2063SBarry Smith 9095615d1e5SSatish Balay #undef __FUNC__ 9105615d1e5SSatish Balay #define __FUNC__ "MatMarkDiag_SeqAIJ" 911416022c9SBarry Smith int MatMarkDiag_SeqAIJ(Mat A) 91217ab2063SBarry Smith { 913416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 914416022c9SBarry Smith int i,j, *diag, m = a->m,shift = a->indexshift; 91517ab2063SBarry Smith 916*3a40ed3dSBarry Smith PetscFunctionBegin; 9170452661fSBarry Smith diag = (int *) PetscMalloc( (m+1)*sizeof(int)); CHKPTRQ(diag); 918416022c9SBarry Smith PLogObjectMemory(A,(m+1)*sizeof(int)); 919416022c9SBarry Smith for ( i=0; i<a->m; i++ ) { 920416022c9SBarry Smith for ( j=a->i[i]+shift; j<a->i[i+1]+shift; j++ ) { 921416022c9SBarry Smith if (a->j[j]+shift == i) { 92217ab2063SBarry Smith diag[i] = j - shift; 92317ab2063SBarry Smith break; 92417ab2063SBarry Smith } 92517ab2063SBarry Smith } 92617ab2063SBarry Smith } 927416022c9SBarry Smith a->diag = diag; 928*3a40ed3dSBarry Smith PetscFunctionReturn(0); 92917ab2063SBarry Smith } 93017ab2063SBarry Smith 9315615d1e5SSatish Balay #undef __FUNC__ 9325615d1e5SSatish Balay #define __FUNC__ "MatRelax_SeqAIJ" 93344cd7ae7SLois Curfman McInnes int MatRelax_SeqAIJ(Mat A,Vec bb,double omega,MatSORType flag, 93417ab2063SBarry Smith double fshift,int its,Vec xx) 93517ab2063SBarry Smith { 936416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 937416022c9SBarry Smith Scalar *x, *b, *bs, d, *xs, sum, *v = a->a,*t,scale,*ts, *xb; 938d5d45c9bSBarry Smith int ierr, *idx, *diag,n = a->n, m = a->m, i, shift = a->indexshift; 93917ab2063SBarry Smith 940*3a40ed3dSBarry Smith PetscFunctionBegin; 94190f02eecSBarry Smith VecGetArray_Fast(xx,x); VecGetArray_Fast(bb,b); 942d00d2cf4SBarry Smith if (!a->diag) {ierr = MatMarkDiag_SeqAIJ(A);CHKERRQ(ierr);} 943416022c9SBarry Smith diag = a->diag; 944416022c9SBarry Smith xs = x + shift; /* shifted by one for index start of a or a->j*/ 94517ab2063SBarry Smith if (flag == SOR_APPLY_UPPER) { 94617ab2063SBarry Smith /* apply ( U + D/omega) to the vector */ 94717ab2063SBarry Smith bs = b + shift; 94817ab2063SBarry Smith for ( i=0; i<m; i++ ) { 949416022c9SBarry Smith d = fshift + a->a[diag[i] + shift]; 950416022c9SBarry Smith n = a->i[i+1] - diag[i] - 1; 951416022c9SBarry Smith idx = a->j + diag[i] + (!shift); 952416022c9SBarry Smith v = a->a + diag[i] + (!shift); 95317ab2063SBarry Smith sum = b[i]*d/omega; 95417ab2063SBarry Smith SPARSEDENSEDOT(sum,bs,v,idx,n); 95517ab2063SBarry Smith x[i] = sum; 95617ab2063SBarry Smith } 957*3a40ed3dSBarry Smith PetscFunctionReturn(0); 95817ab2063SBarry Smith } 95917ab2063SBarry Smith if (flag == SOR_APPLY_LOWER) { 960e3372554SBarry Smith SETERRQ(1,0,"SOR_APPLY_LOWER is not done"); 961*3a40ed3dSBarry Smith } else if (flag & SOR_EISENSTAT) { 96217ab2063SBarry Smith /* Let A = L + U + D; where L is lower trianglar, 96317ab2063SBarry Smith U is upper triangular, E is diagonal; This routine applies 96417ab2063SBarry Smith 96517ab2063SBarry Smith (L + E)^{-1} A (U + E)^{-1} 96617ab2063SBarry Smith 96717ab2063SBarry Smith to a vector efficiently using Eisenstat's trick. This is for 96817ab2063SBarry Smith the case of SSOR preconditioner, so E is D/omega where omega 96917ab2063SBarry Smith is the relaxation factor. 97017ab2063SBarry Smith */ 9710452661fSBarry Smith t = (Scalar *) PetscMalloc( m*sizeof(Scalar) ); CHKPTRQ(t); 97217ab2063SBarry Smith scale = (2.0/omega) - 1.0; 97317ab2063SBarry Smith 97417ab2063SBarry Smith /* x = (E + U)^{-1} b */ 97517ab2063SBarry Smith for ( i=m-1; i>=0; i-- ) { 976416022c9SBarry Smith d = fshift + a->a[diag[i] + shift]; 977416022c9SBarry Smith n = a->i[i+1] - diag[i] - 1; 978416022c9SBarry Smith idx = a->j + diag[i] + (!shift); 979416022c9SBarry Smith v = a->a + diag[i] + (!shift); 98017ab2063SBarry Smith sum = b[i]; 98117ab2063SBarry Smith SPARSEDENSEMDOT(sum,xs,v,idx,n); 98217ab2063SBarry Smith x[i] = omega*(sum/d); 98317ab2063SBarry Smith } 98417ab2063SBarry Smith 98517ab2063SBarry Smith /* t = b - (2*E - D)x */ 986416022c9SBarry Smith v = a->a; 98717ab2063SBarry Smith for ( i=0; i<m; i++ ) { t[i] = b[i] - scale*(v[*diag++ + shift])*x[i]; } 98817ab2063SBarry Smith 98917ab2063SBarry Smith /* t = (E + L)^{-1}t */ 990416022c9SBarry Smith ts = t + shift; /* shifted by one for index start of a or a->j*/ 991416022c9SBarry Smith diag = a->diag; 99217ab2063SBarry Smith for ( i=0; i<m; i++ ) { 993416022c9SBarry Smith d = fshift + a->a[diag[i]+shift]; 994416022c9SBarry Smith n = diag[i] - a->i[i]; 995416022c9SBarry Smith idx = a->j + a->i[i] + shift; 996416022c9SBarry Smith v = a->a + a->i[i] + shift; 99717ab2063SBarry Smith sum = t[i]; 99817ab2063SBarry Smith SPARSEDENSEMDOT(sum,ts,v,idx,n); 99917ab2063SBarry Smith t[i] = omega*(sum/d); 100017ab2063SBarry Smith } 100117ab2063SBarry Smith 100217ab2063SBarry Smith /* x = x + t */ 100317ab2063SBarry Smith for ( i=0; i<m; i++ ) { x[i] += t[i]; } 10040452661fSBarry Smith PetscFree(t); 1005*3a40ed3dSBarry Smith PetscFunctionReturn(0); 100617ab2063SBarry Smith } 100717ab2063SBarry Smith if (flag & SOR_ZERO_INITIAL_GUESS) { 100817ab2063SBarry Smith if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP){ 100917ab2063SBarry Smith for ( i=0; i<m; i++ ) { 1010416022c9SBarry Smith d = fshift + a->a[diag[i]+shift]; 1011416022c9SBarry Smith n = diag[i] - a->i[i]; 1012416022c9SBarry Smith idx = a->j + a->i[i] + shift; 1013416022c9SBarry Smith v = a->a + a->i[i] + shift; 101417ab2063SBarry Smith sum = b[i]; 101517ab2063SBarry Smith SPARSEDENSEMDOT(sum,xs,v,idx,n); 101617ab2063SBarry Smith x[i] = omega*(sum/d); 101717ab2063SBarry Smith } 101817ab2063SBarry Smith xb = x; 1019*3a40ed3dSBarry Smith } else xb = b; 102017ab2063SBarry Smith if ((flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP) && 102117ab2063SBarry Smith (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP)) { 102217ab2063SBarry Smith for ( i=0; i<m; i++ ) { 1023416022c9SBarry Smith x[i] *= a->a[diag[i]+shift]; 102417ab2063SBarry Smith } 102517ab2063SBarry Smith } 102617ab2063SBarry Smith if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP){ 102717ab2063SBarry Smith for ( i=m-1; i>=0; i-- ) { 1028416022c9SBarry Smith d = fshift + a->a[diag[i] + shift]; 1029416022c9SBarry Smith n = a->i[i+1] - diag[i] - 1; 1030416022c9SBarry Smith idx = a->j + diag[i] + (!shift); 1031416022c9SBarry Smith v = a->a + diag[i] + (!shift); 103217ab2063SBarry Smith sum = xb[i]; 103317ab2063SBarry Smith SPARSEDENSEMDOT(sum,xs,v,idx,n); 103417ab2063SBarry Smith x[i] = omega*(sum/d); 103517ab2063SBarry Smith } 103617ab2063SBarry Smith } 103717ab2063SBarry Smith its--; 103817ab2063SBarry Smith } 103917ab2063SBarry Smith while (its--) { 104017ab2063SBarry Smith if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP){ 104117ab2063SBarry Smith for ( i=0; i<m; i++ ) { 1042416022c9SBarry Smith d = fshift + a->a[diag[i]+shift]; 1043416022c9SBarry Smith n = a->i[i+1] - a->i[i]; 1044416022c9SBarry Smith idx = a->j + a->i[i] + shift; 1045416022c9SBarry Smith v = a->a + a->i[i] + shift; 104617ab2063SBarry Smith sum = b[i]; 104717ab2063SBarry Smith SPARSEDENSEMDOT(sum,xs,v,idx,n); 10487e33a6baSBarry Smith x[i] = (1. - omega)*x[i] + omega*(sum + a->a[diag[i]+shift]*x[i])/d; 104917ab2063SBarry Smith } 105017ab2063SBarry Smith } 105117ab2063SBarry Smith if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP){ 105217ab2063SBarry Smith for ( i=m-1; i>=0; i-- ) { 1053416022c9SBarry Smith d = fshift + a->a[diag[i] + shift]; 1054416022c9SBarry Smith n = a->i[i+1] - a->i[i]; 1055416022c9SBarry Smith idx = a->j + a->i[i] + shift; 1056416022c9SBarry Smith v = a->a + a->i[i] + shift; 105717ab2063SBarry Smith sum = b[i]; 105817ab2063SBarry Smith SPARSEDENSEMDOT(sum,xs,v,idx,n); 10597e33a6baSBarry Smith x[i] = (1. - omega)*x[i] + omega*(sum + a->a[diag[i]+shift]*x[i])/d; 106017ab2063SBarry Smith } 106117ab2063SBarry Smith } 106217ab2063SBarry Smith } 1063*3a40ed3dSBarry Smith PetscFunctionReturn(0); 106417ab2063SBarry Smith } 106517ab2063SBarry Smith 10665615d1e5SSatish Balay #undef __FUNC__ 10675615d1e5SSatish Balay #define __FUNC__ "MatGetInfo_SeqAIJ" 10688f6be9afSLois Curfman McInnes int MatGetInfo_SeqAIJ(Mat A,MatInfoType flag,MatInfo *info) 106917ab2063SBarry Smith { 1070416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 10714e220ebcSLois Curfman McInnes 1072*3a40ed3dSBarry Smith PetscFunctionBegin; 10734e220ebcSLois Curfman McInnes info->rows_global = (double)a->m; 10744e220ebcSLois Curfman McInnes info->columns_global = (double)a->n; 10754e220ebcSLois Curfman McInnes info->rows_local = (double)a->m; 10764e220ebcSLois Curfman McInnes info->columns_local = (double)a->n; 10774e220ebcSLois Curfman McInnes info->block_size = 1.0; 10784e220ebcSLois Curfman McInnes info->nz_allocated = (double)a->maxnz; 10794e220ebcSLois Curfman McInnes info->nz_used = (double)a->nz; 10804e220ebcSLois Curfman McInnes info->nz_unneeded = (double)(a->maxnz - a->nz); 10814e220ebcSLois Curfman McInnes /* if (info->nz_unneeded != A->info.nz_unneeded) 10824e220ebcSLois Curfman McInnes printf("space descrepancy: maxnz-nz = %d, nz_unneeded = %d\n",(int)info->nz_unneeded,(int)A->info.nz_unneeded); */ 10834e220ebcSLois Curfman McInnes info->assemblies = (double)A->num_ass; 10844e220ebcSLois Curfman McInnes info->mallocs = (double)a->reallocs; 10854e220ebcSLois Curfman McInnes info->memory = A->mem; 10864e220ebcSLois Curfman McInnes if (A->factor) { 10874e220ebcSLois Curfman McInnes info->fill_ratio_given = A->info.fill_ratio_given; 10884e220ebcSLois Curfman McInnes info->fill_ratio_needed = A->info.fill_ratio_needed; 10894e220ebcSLois Curfman McInnes info->factor_mallocs = A->info.factor_mallocs; 10904e220ebcSLois Curfman McInnes } else { 10914e220ebcSLois Curfman McInnes info->fill_ratio_given = 0; 10924e220ebcSLois Curfman McInnes info->fill_ratio_needed = 0; 10934e220ebcSLois Curfman McInnes info->factor_mallocs = 0; 10944e220ebcSLois Curfman McInnes } 1095*3a40ed3dSBarry Smith PetscFunctionReturn(0); 109617ab2063SBarry Smith } 109717ab2063SBarry Smith 109817ab2063SBarry Smith extern int MatLUFactorSymbolic_SeqAIJ(Mat,IS,IS,double,Mat*); 109917ab2063SBarry Smith extern int MatLUFactorNumeric_SeqAIJ(Mat,Mat*); 110017ab2063SBarry Smith extern int MatLUFactor_SeqAIJ(Mat,IS,IS,double); 110117ab2063SBarry Smith extern int MatSolve_SeqAIJ(Mat,Vec,Vec); 110217ab2063SBarry Smith extern int MatSolveAdd_SeqAIJ(Mat,Vec,Vec,Vec); 110317ab2063SBarry Smith extern int MatSolveTrans_SeqAIJ(Mat,Vec,Vec); 110417ab2063SBarry Smith extern int MatSolveTransAdd_SeqAIJ(Mat,Vec,Vec,Vec); 110517ab2063SBarry Smith 11065615d1e5SSatish Balay #undef __FUNC__ 11075615d1e5SSatish Balay #define __FUNC__ "MatZeroRows_SeqAIJ" 11088f6be9afSLois Curfman McInnes int MatZeroRows_SeqAIJ(Mat A,IS is,Scalar *diag) 110917ab2063SBarry Smith { 1110416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 1111416022c9SBarry Smith int i,ierr,N, *rows,m = a->m - 1,shift = a->indexshift; 111217ab2063SBarry Smith 1113*3a40ed3dSBarry Smith PetscFunctionBegin; 111477c4ece6SBarry Smith ierr = ISGetSize(is,&N); CHKERRQ(ierr); 111517ab2063SBarry Smith ierr = ISGetIndices(is,&rows); CHKERRQ(ierr); 111617ab2063SBarry Smith if (diag) { 111717ab2063SBarry Smith for ( i=0; i<N; i++ ) { 1118e3372554SBarry Smith if (rows[i] < 0 || rows[i] > m) SETERRQ(1,0,"row out of range"); 1119416022c9SBarry Smith if (a->ilen[rows[i]] > 0) { /* in case row was completely empty */ 1120416022c9SBarry Smith a->ilen[rows[i]] = 1; 1121416022c9SBarry Smith a->a[a->i[rows[i]]+shift] = *diag; 1122416022c9SBarry Smith a->j[a->i[rows[i]]+shift] = rows[i]+shift; 1123*3a40ed3dSBarry Smith } else { 112417ab2063SBarry Smith ierr = MatSetValues_SeqAIJ(A,1,&rows[i],1,&rows[i],diag,INSERT_VALUES); 112517ab2063SBarry Smith CHKERRQ(ierr); 112617ab2063SBarry Smith } 112717ab2063SBarry Smith } 1128*3a40ed3dSBarry Smith } else { 112917ab2063SBarry Smith for ( i=0; i<N; i++ ) { 1130e3372554SBarry Smith if (rows[i] < 0 || rows[i] > m) SETERRQ(1,0,"row out of range"); 1131416022c9SBarry Smith a->ilen[rows[i]] = 0; 113217ab2063SBarry Smith } 113317ab2063SBarry Smith } 113417ab2063SBarry Smith ISRestoreIndices(is,&rows); 113543a90d84SBarry Smith ierr = MatAssemblyEnd_SeqAIJ(A,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr); 1136*3a40ed3dSBarry Smith PetscFunctionReturn(0); 113717ab2063SBarry Smith } 113817ab2063SBarry Smith 11395615d1e5SSatish Balay #undef __FUNC__ 11405615d1e5SSatish Balay #define __FUNC__ "MatGetSize_SeqAIJ" 11418f6be9afSLois Curfman McInnes int MatGetSize_SeqAIJ(Mat A,int *m,int *n) 114217ab2063SBarry Smith { 1143416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 1144*3a40ed3dSBarry Smith 1145*3a40ed3dSBarry Smith PetscFunctionBegin; 11460752156aSBarry Smith if (m) *m = a->m; 11470752156aSBarry Smith if (n) *n = a->n; 1148*3a40ed3dSBarry Smith PetscFunctionReturn(0); 114917ab2063SBarry Smith } 115017ab2063SBarry Smith 11515615d1e5SSatish Balay #undef __FUNC__ 11525615d1e5SSatish Balay #define __FUNC__ "MatGetOwnershipRange_SeqAIJ" 11538f6be9afSLois Curfman McInnes int MatGetOwnershipRange_SeqAIJ(Mat A,int *m,int *n) 115417ab2063SBarry Smith { 1155416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 1156*3a40ed3dSBarry Smith 1157*3a40ed3dSBarry Smith PetscFunctionBegin; 1158416022c9SBarry Smith *m = 0; *n = a->m; 1159*3a40ed3dSBarry Smith PetscFunctionReturn(0); 116017ab2063SBarry Smith } 1161026e39d0SSatish Balay 11625615d1e5SSatish Balay #undef __FUNC__ 11635615d1e5SSatish Balay #define __FUNC__ "MatGetRow_SeqAIJ" 11644e093b46SBarry Smith int MatGetRow_SeqAIJ(Mat A,int row,int *nz,int **idx,Scalar **v) 116517ab2063SBarry Smith { 1166416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 1167c456f294SBarry Smith int *itmp,i,shift = a->indexshift; 116817ab2063SBarry Smith 1169*3a40ed3dSBarry Smith PetscFunctionBegin; 1170e3372554SBarry Smith if (row < 0 || row >= a->m) SETERRQ(1,0,"Row out of range"); 117117ab2063SBarry Smith 1172416022c9SBarry Smith *nz = a->i[row+1] - a->i[row]; 1173416022c9SBarry Smith if (v) *v = a->a + a->i[row] + shift; 117417ab2063SBarry Smith if (idx) { 1175416022c9SBarry Smith itmp = a->j + a->i[row] + shift; 11764e093b46SBarry Smith if (*nz && shift) { 11770452661fSBarry Smith *idx = (int *) PetscMalloc( (*nz)*sizeof(int) ); CHKPTRQ(*idx); 117817ab2063SBarry Smith for ( i=0; i<(*nz); i++ ) {(*idx)[i] = itmp[i] + shift;} 11794e093b46SBarry Smith } else if (*nz) { 11804e093b46SBarry Smith *idx = itmp; 118117ab2063SBarry Smith } 118217ab2063SBarry Smith else *idx = 0; 118317ab2063SBarry Smith } 1184*3a40ed3dSBarry Smith PetscFunctionReturn(0); 118517ab2063SBarry Smith } 118617ab2063SBarry Smith 11875615d1e5SSatish Balay #undef __FUNC__ 11885615d1e5SSatish Balay #define __FUNC__ "MatRestoreRow_SeqAIJ" 11894e093b46SBarry Smith int MatRestoreRow_SeqAIJ(Mat A,int row,int *nz,int **idx,Scalar **v) 119017ab2063SBarry Smith { 11914e093b46SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 1192*3a40ed3dSBarry Smith 1193*3a40ed3dSBarry Smith PetscFunctionBegin; 11944e093b46SBarry Smith if (idx) {if (*idx && a->indexshift) PetscFree(*idx);} 1195*3a40ed3dSBarry Smith PetscFunctionReturn(0); 119617ab2063SBarry Smith } 119717ab2063SBarry Smith 11985615d1e5SSatish Balay #undef __FUNC__ 11995615d1e5SSatish Balay #define __FUNC__ "MatNorm_SeqAIJ" 12008f6be9afSLois Curfman McInnes int MatNorm_SeqAIJ(Mat A,NormType type,double *norm) 120117ab2063SBarry Smith { 1202416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 1203416022c9SBarry Smith Scalar *v = a->a; 120417ab2063SBarry Smith double sum = 0.0; 1205416022c9SBarry Smith int i, j,shift = a->indexshift; 120617ab2063SBarry Smith 1207*3a40ed3dSBarry Smith PetscFunctionBegin; 120817ab2063SBarry Smith if (type == NORM_FROBENIUS) { 1209416022c9SBarry Smith for (i=0; i<a->nz; i++ ) { 1210*3a40ed3dSBarry Smith #if defined(USE_PETSC_COMPLEX) 121117ab2063SBarry Smith sum += real(conj(*v)*(*v)); v++; 121217ab2063SBarry Smith #else 121317ab2063SBarry Smith sum += (*v)*(*v); v++; 121417ab2063SBarry Smith #endif 121517ab2063SBarry Smith } 121617ab2063SBarry Smith *norm = sqrt(sum); 1217*3a40ed3dSBarry Smith } else if (type == NORM_1) { 121817ab2063SBarry Smith double *tmp; 1219416022c9SBarry Smith int *jj = a->j; 122066963ce1SSatish Balay tmp = (double *) PetscMalloc( (a->n+1)*sizeof(double) ); CHKPTRQ(tmp); 1221cddf8d76SBarry Smith PetscMemzero(tmp,a->n*sizeof(double)); 122217ab2063SBarry Smith *norm = 0.0; 1223416022c9SBarry Smith for ( j=0; j<a->nz; j++ ) { 1224a2744918SBarry Smith tmp[*jj++ + shift] += PetscAbsScalar(*v); v++; 122517ab2063SBarry Smith } 1226416022c9SBarry Smith for ( j=0; j<a->n; j++ ) { 122717ab2063SBarry Smith if (tmp[j] > *norm) *norm = tmp[j]; 122817ab2063SBarry Smith } 12290452661fSBarry Smith PetscFree(tmp); 1230*3a40ed3dSBarry Smith } else if (type == NORM_INFINITY) { 123117ab2063SBarry Smith *norm = 0.0; 1232416022c9SBarry Smith for ( j=0; j<a->m; j++ ) { 1233416022c9SBarry Smith v = a->a + a->i[j] + shift; 123417ab2063SBarry Smith sum = 0.0; 1235416022c9SBarry Smith for ( i=0; i<a->i[j+1]-a->i[j]; i++ ) { 1236cddf8d76SBarry Smith sum += PetscAbsScalar(*v); v++; 123717ab2063SBarry Smith } 123817ab2063SBarry Smith if (sum > *norm) *norm = sum; 123917ab2063SBarry Smith } 1240*3a40ed3dSBarry Smith } else { 1241e3372554SBarry Smith SETERRQ(1,0,"No support for two norm yet"); 124217ab2063SBarry Smith } 1243*3a40ed3dSBarry Smith PetscFunctionReturn(0); 124417ab2063SBarry Smith } 124517ab2063SBarry Smith 12465615d1e5SSatish Balay #undef __FUNC__ 12475615d1e5SSatish Balay #define __FUNC__ "MatTranspose_SeqAIJ" 12488f6be9afSLois Curfman McInnes int MatTranspose_SeqAIJ(Mat A,Mat *B) 124917ab2063SBarry Smith { 1250416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 1251416022c9SBarry Smith Mat C; 1252416022c9SBarry Smith int i, ierr, *aj = a->j, *ai = a->i, m = a->m, len, *col; 1253416022c9SBarry Smith int shift = a->indexshift; 1254d5d45c9bSBarry Smith Scalar *array = a->a; 125517ab2063SBarry Smith 1256*3a40ed3dSBarry Smith PetscFunctionBegin; 1257*3a40ed3dSBarry Smith if (B == PETSC_NULL && m != a->n) SETERRQ(1,0,"Square matrix only for in-place"); 12580452661fSBarry Smith col = (int *) PetscMalloc((1+a->n)*sizeof(int)); CHKPTRQ(col); 1259cddf8d76SBarry Smith PetscMemzero(col,(1+a->n)*sizeof(int)); 126017ab2063SBarry Smith if (shift) { 126117ab2063SBarry Smith for ( i=0; i<ai[m]-1; i++ ) aj[i] -= 1; 126217ab2063SBarry Smith } 126317ab2063SBarry Smith for ( i=0; i<ai[m]+shift; i++ ) col[aj[i]] += 1; 1264416022c9SBarry Smith ierr = MatCreateSeqAIJ(A->comm,a->n,m,0,col,&C); CHKERRQ(ierr); 12650452661fSBarry Smith PetscFree(col); 126617ab2063SBarry Smith for ( i=0; i<m; i++ ) { 126717ab2063SBarry Smith len = ai[i+1]-ai[i]; 1268416022c9SBarry Smith ierr = MatSetValues(C,len,aj,1,&i,array,INSERT_VALUES); CHKERRQ(ierr); 126917ab2063SBarry Smith array += len; aj += len; 127017ab2063SBarry Smith } 127117ab2063SBarry Smith if (shift) { 127217ab2063SBarry Smith for ( i=0; i<ai[m]-1; i++ ) aj[i] += 1; 127317ab2063SBarry Smith } 127417ab2063SBarry Smith 12756d4a8577SBarry Smith ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr); 12766d4a8577SBarry Smith ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr); 127717ab2063SBarry Smith 12783638b69dSLois Curfman McInnes if (B != PETSC_NULL) { 1279416022c9SBarry Smith *B = C; 128017ab2063SBarry Smith } else { 1281416022c9SBarry Smith /* This isn't really an in-place transpose */ 12820452661fSBarry Smith PetscFree(a->a); 12830452661fSBarry Smith if (!a->singlemalloc) {PetscFree(a->i); PetscFree(a->j);} 12840452661fSBarry Smith if (a->diag) PetscFree(a->diag); 12850452661fSBarry Smith if (a->ilen) PetscFree(a->ilen); 12860452661fSBarry Smith if (a->imax) PetscFree(a->imax); 12870452661fSBarry Smith if (a->solve_work) PetscFree(a->solve_work); 12881073c447SSatish Balay if (a->inode.size) PetscFree(a->inode.size); 12890452661fSBarry Smith PetscFree(a); 1290f09e8eb9SSatish Balay PetscMemcpy(A,C,sizeof(struct _p_Mat)); 12910452661fSBarry Smith PetscHeaderDestroy(C); 129217ab2063SBarry Smith } 1293*3a40ed3dSBarry Smith PetscFunctionReturn(0); 129417ab2063SBarry Smith } 129517ab2063SBarry Smith 12965615d1e5SSatish Balay #undef __FUNC__ 12975615d1e5SSatish Balay #define __FUNC__ "MatDiagonalScale_SeqAIJ" 12988f6be9afSLois Curfman McInnes int MatDiagonalScale_SeqAIJ(Mat A,Vec ll,Vec rr) 129917ab2063SBarry Smith { 1300416022c9SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 130117ab2063SBarry Smith Scalar *l,*r,x,*v; 1302d5d45c9bSBarry Smith int i,j,m = a->m, n = a->n, M, nz = a->nz, *jj,shift = a->indexshift; 130317ab2063SBarry Smith 1304*3a40ed3dSBarry Smith PetscFunctionBegin; 130517ab2063SBarry Smith if (ll) { 13063ea7c6a1SSatish Balay /* The local size is used so that VecMPI can be passed to this routine 13073ea7c6a1SSatish Balay by MatDiagonalScale_MPIAIJ */ 13089b1297e1SSatish Balay VecGetLocalSize_Fast(ll,m); 1309e3372554SBarry Smith if (m != a->m) SETERRQ(1,0,"Left scaling vector wrong length"); 131090f02eecSBarry Smith VecGetArray_Fast(ll,l); 1311416022c9SBarry Smith v = a->a; 131217ab2063SBarry Smith for ( i=0; i<m; i++ ) { 131317ab2063SBarry Smith x = l[i]; 1314416022c9SBarry Smith M = a->i[i+1] - a->i[i]; 131517ab2063SBarry Smith for ( j=0; j<M; j++ ) { (*v++) *= x;} 131617ab2063SBarry Smith } 131744cd7ae7SLois Curfman McInnes PLogFlops(nz); 131817ab2063SBarry Smith } 131917ab2063SBarry Smith if (rr) { 13209b1297e1SSatish Balay VecGetLocalSize_Fast(rr,n); 1321e3372554SBarry Smith if (n != a->n) SETERRQ(1,0,"Right scaling vector wrong length"); 132290f02eecSBarry Smith VecGetArray_Fast(rr,r); 1323416022c9SBarry Smith v = a->a; jj = a->j; 132417ab2063SBarry Smith for ( i=0; i<nz; i++ ) { 132517ab2063SBarry Smith (*v++) *= r[*jj++ + shift]; 132617ab2063SBarry Smith } 132744cd7ae7SLois Curfman McInnes PLogFlops(nz); 132817ab2063SBarry Smith } 1329*3a40ed3dSBarry Smith PetscFunctionReturn(0); 133017ab2063SBarry Smith } 133117ab2063SBarry Smith 13325615d1e5SSatish Balay #undef __FUNC__ 13335615d1e5SSatish Balay #define __FUNC__ "MatGetSubMatrix_SeqAIJ" 13348f6be9afSLois Curfman McInnes int MatGetSubMatrix_SeqAIJ(Mat A,IS isrow,IS iscol,MatGetSubMatrixCall scall,Mat *B) 133517ab2063SBarry Smith { 1336db02288aSLois Curfman McInnes Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data,*c; 133702834360SBarry Smith int nznew, *smap, i, k, kstart, kend, ierr, oldcols = a->n,*lens; 133899141d43SSatish Balay int row,mat_i,*mat_j,tcol,first,step,*mat_ilen; 1339a2744918SBarry Smith register int sum,lensi; 134099141d43SSatish Balay int *irow, *icol, nrows, ncols, shift = a->indexshift,*ssmap; 134199141d43SSatish Balay int *starts,*j_new,*i_new,*aj = a->j, *ai = a->i,ii,*ailen = a->ilen; 134299141d43SSatish Balay Scalar *a_new,*mat_a; 1343416022c9SBarry Smith Mat C; 134417ab2063SBarry Smith 1345*3a40ed3dSBarry Smith PetscFunctionBegin; 1346b48a1e75SSatish Balay ierr = ISSorted(isrow,(PetscTruth*)&i); 1347e3372554SBarry Smith if (!i) SETERRQ(1,0,"ISrow is not sorted"); 134899141d43SSatish Balay ierr = ISSorted(iscol,(PetscTruth*)&i); 1349e3372554SBarry Smith if (!i) SETERRQ(1,0,"IScol is not sorted"); 135099141d43SSatish Balay 135117ab2063SBarry Smith ierr = ISGetIndices(isrow,&irow); CHKERRQ(ierr); 135217ab2063SBarry Smith ierr = ISGetSize(isrow,&nrows); CHKERRQ(ierr); 135317ab2063SBarry Smith ierr = ISGetSize(iscol,&ncols); CHKERRQ(ierr); 135417ab2063SBarry Smith 13557264ac53SSatish Balay if (ISStrideGetInfo(iscol,&first,&step) && step == 1) { /* no need to sort */ 135602834360SBarry Smith /* special case of contiguous rows */ 135757faeb66SBarry Smith lens = (int *) PetscMalloc((ncols+nrows+1)*sizeof(int)); CHKPTRQ(lens); 135802834360SBarry Smith starts = lens + ncols; 135902834360SBarry Smith /* loop over new rows determining lens and starting points */ 136002834360SBarry Smith for (i=0; i<nrows; i++) { 1361a2744918SBarry Smith kstart = ai[irow[i]]+shift; 1362a2744918SBarry Smith kend = kstart + ailen[irow[i]]; 136302834360SBarry Smith for ( k=kstart; k<kend; k++ ) { 1364d8ced48eSBarry Smith if (aj[k]+shift >= first) { 136502834360SBarry Smith starts[i] = k; 136602834360SBarry Smith break; 136702834360SBarry Smith } 136802834360SBarry Smith } 1369a2744918SBarry Smith sum = 0; 137002834360SBarry Smith while (k < kend) { 1371d8ced48eSBarry Smith if (aj[k++]+shift >= first+ncols) break; 1372a2744918SBarry Smith sum++; 137302834360SBarry Smith } 1374a2744918SBarry Smith lens[i] = sum; 137502834360SBarry Smith } 137602834360SBarry Smith /* create submatrix */ 1377cddf8d76SBarry Smith if (scall == MAT_REUSE_MATRIX) { 137808480c60SBarry Smith int n_cols,n_rows; 137908480c60SBarry Smith ierr = MatGetSize(*B,&n_rows,&n_cols); CHKERRQ(ierr); 1380e3372554SBarry Smith if (n_rows != nrows || n_cols != ncols) SETERRQ(1,0,""); 1381d8ced48eSBarry Smith ierr = MatZeroEntries(*B); CHKERRQ(ierr); 138208480c60SBarry Smith C = *B; 1383*3a40ed3dSBarry Smith } else { 138402834360SBarry Smith ierr = MatCreateSeqAIJ(A->comm,nrows,ncols,0,lens,&C);CHKERRQ(ierr); 138508480c60SBarry Smith } 1386db02288aSLois Curfman McInnes c = (Mat_SeqAIJ*) C->data; 1387db02288aSLois Curfman McInnes 138802834360SBarry Smith /* loop over rows inserting into submatrix */ 1389db02288aSLois Curfman McInnes a_new = c->a; 1390db02288aSLois Curfman McInnes j_new = c->j; 1391db02288aSLois Curfman McInnes i_new = c->i; 1392db02288aSLois Curfman McInnes i_new[0] = -shift; 139302834360SBarry Smith for (i=0; i<nrows; i++) { 1394a2744918SBarry Smith ii = starts[i]; 1395a2744918SBarry Smith lensi = lens[i]; 1396a2744918SBarry Smith for ( k=0; k<lensi; k++ ) { 1397a2744918SBarry Smith *j_new++ = aj[ii+k] - first; 139802834360SBarry Smith } 1399a2744918SBarry Smith PetscMemcpy(a_new,a->a + starts[i],lensi*sizeof(Scalar)); 1400a2744918SBarry Smith a_new += lensi; 1401a2744918SBarry Smith i_new[i+1] = i_new[i] + lensi; 1402a2744918SBarry Smith c->ilen[i] = lensi; 140302834360SBarry Smith } 14040452661fSBarry Smith PetscFree(lens); 1405*3a40ed3dSBarry Smith } else { 140602834360SBarry Smith ierr = ISGetIndices(iscol,&icol); CHKERRQ(ierr); 14070452661fSBarry Smith smap = (int *) PetscMalloc((1+oldcols)*sizeof(int)); CHKPTRQ(smap); 140802834360SBarry Smith ssmap = smap + shift; 140999141d43SSatish Balay lens = (int *) PetscMalloc((1+nrows)*sizeof(int)); CHKPTRQ(lens); 1410cddf8d76SBarry Smith PetscMemzero(smap,oldcols*sizeof(int)); 141117ab2063SBarry Smith for ( i=0; i<ncols; i++ ) smap[icol[i]] = i+1; 141202834360SBarry Smith /* determine lens of each row */ 141302834360SBarry Smith for (i=0; i<nrows; i++) { 1414d8ced48eSBarry Smith kstart = ai[irow[i]]+shift; 141502834360SBarry Smith kend = kstart + a->ilen[irow[i]]; 141602834360SBarry Smith lens[i] = 0; 141702834360SBarry Smith for ( k=kstart; k<kend; k++ ) { 1418d8ced48eSBarry Smith if (ssmap[aj[k]]) { 141902834360SBarry Smith lens[i]++; 142002834360SBarry Smith } 142102834360SBarry Smith } 142202834360SBarry Smith } 142317ab2063SBarry Smith /* Create and fill new matrix */ 1424a2744918SBarry Smith if (scall == MAT_REUSE_MATRIX) { 142599141d43SSatish Balay c = (Mat_SeqAIJ *)((*B)->data); 142699141d43SSatish Balay 1427e3372554SBarry Smith if (c->m != nrows || c->n != ncols) SETERRQ(1,0,""); 142899141d43SSatish Balay if (PetscMemcmp(c->ilen,lens, c->m *sizeof(int))) { 1429e3372554SBarry Smith SETERRQ(1,0,"Cannot reuse matrix. wrong no of nonzeros"); 143099141d43SSatish Balay } 143199141d43SSatish Balay PetscMemzero(c->ilen,c->m*sizeof(int)); 143208480c60SBarry Smith C = *B; 1433*3a40ed3dSBarry Smith } else { 143402834360SBarry Smith ierr = MatCreateSeqAIJ(A->comm,nrows,ncols,0,lens,&C);CHKERRQ(ierr); 143508480c60SBarry Smith } 143699141d43SSatish Balay c = (Mat_SeqAIJ *)(C->data); 143717ab2063SBarry Smith for (i=0; i<nrows; i++) { 143899141d43SSatish Balay row = irow[i]; 143917ab2063SBarry Smith nznew = 0; 144099141d43SSatish Balay kstart = ai[row]+shift; 144199141d43SSatish Balay kend = kstart + a->ilen[row]; 144299141d43SSatish Balay mat_i = c->i[i]+shift; 144399141d43SSatish Balay mat_j = c->j + mat_i; 144499141d43SSatish Balay mat_a = c->a + mat_i; 144599141d43SSatish Balay mat_ilen = c->ilen + i; 144617ab2063SBarry Smith for ( k=kstart; k<kend; k++ ) { 144799141d43SSatish Balay if ((tcol=ssmap[a->j[k]])) { 144899141d43SSatish Balay *mat_j++ = tcol - (!shift); 144999141d43SSatish Balay *mat_a++ = a->a[k]; 145099141d43SSatish Balay (*mat_ilen)++; 145199141d43SSatish Balay 145217ab2063SBarry Smith } 145317ab2063SBarry Smith } 145417ab2063SBarry Smith } 145502834360SBarry Smith /* Free work space */ 145602834360SBarry Smith ierr = ISRestoreIndices(iscol,&icol); CHKERRQ(ierr); 145799141d43SSatish Balay PetscFree(smap); PetscFree(lens); 145802834360SBarry Smith } 14596d4a8577SBarry Smith ierr = MatAssemblyBegin(C,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr); 14606d4a8577SBarry Smith ierr = MatAssemblyEnd(C,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr); 146117ab2063SBarry Smith 146217ab2063SBarry Smith ierr = ISRestoreIndices(isrow,&irow); CHKERRQ(ierr); 1463416022c9SBarry Smith *B = C; 1464*3a40ed3dSBarry Smith PetscFunctionReturn(0); 146517ab2063SBarry Smith } 146617ab2063SBarry Smith 1467a871dcd8SBarry Smith /* 146863b91edcSBarry Smith note: This can only work for identity for row and col. It would 146963b91edcSBarry Smith be good to check this and otherwise generate an error. 1470a871dcd8SBarry Smith */ 14715615d1e5SSatish Balay #undef __FUNC__ 14725615d1e5SSatish Balay #define __FUNC__ "MatILUFactor_SeqAIJ" 14738f6be9afSLois Curfman McInnes int MatILUFactor_SeqAIJ(Mat inA,IS row,IS col,double efill,int fill) 1474a871dcd8SBarry Smith { 147563b91edcSBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) inA->data; 147608480c60SBarry Smith int ierr; 147763b91edcSBarry Smith Mat outA; 147863b91edcSBarry Smith 1479*3a40ed3dSBarry Smith PetscFunctionBegin; 1480e3372554SBarry Smith if (fill != 0) SETERRQ(1,0,"Only fill=0 supported"); 1481a871dcd8SBarry Smith 148263b91edcSBarry Smith outA = inA; 148363b91edcSBarry Smith inA->factor = FACTOR_LU; 148463b91edcSBarry Smith a->row = row; 148563b91edcSBarry Smith a->col = col; 148663b91edcSBarry Smith 148794a9d846SBarry Smith if (!a->solve_work) { /* this matrix may have been factored before */ 14880452661fSBarry Smith a->solve_work = (Scalar *) PetscMalloc( (a->m+1)*sizeof(Scalar)); CHKPTRQ(a->solve_work); 148994a9d846SBarry Smith } 149063b91edcSBarry Smith 149108480c60SBarry Smith if (!a->diag) { 149208480c60SBarry Smith ierr = MatMarkDiag_SeqAIJ(inA); CHKERRQ(ierr); 149363b91edcSBarry Smith } 149463b91edcSBarry Smith ierr = MatLUFactorNumeric_SeqAIJ(inA,&outA); CHKERRQ(ierr); 1495*3a40ed3dSBarry Smith PetscFunctionReturn(0); 1496a871dcd8SBarry Smith } 1497a871dcd8SBarry Smith 1498f0b747eeSBarry Smith #include "pinclude/plapack.h" 14995615d1e5SSatish Balay #undef __FUNC__ 15005615d1e5SSatish Balay #define __FUNC__ "MatScale_SeqAIJ" 15018f6be9afSLois Curfman McInnes int MatScale_SeqAIJ(Scalar *alpha,Mat inA) 1502f0b747eeSBarry Smith { 1503f0b747eeSBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) inA->data; 1504f0b747eeSBarry Smith int one = 1; 1505*3a40ed3dSBarry Smith 1506*3a40ed3dSBarry Smith PetscFunctionBegin; 1507f0b747eeSBarry Smith BLscal_( &a->nz, alpha, a->a, &one ); 1508f0b747eeSBarry Smith PLogFlops(a->nz); 1509*3a40ed3dSBarry Smith PetscFunctionReturn(0); 1510f0b747eeSBarry Smith } 1511f0b747eeSBarry Smith 15125615d1e5SSatish Balay #undef __FUNC__ 15135615d1e5SSatish Balay #define __FUNC__ "MatGetSubMatrices_SeqAIJ" 15148f6be9afSLois Curfman McInnes int MatGetSubMatrices_SeqAIJ(Mat A,int n, IS *irow,IS *icol,MatGetSubMatrixCall scall, 1515cddf8d76SBarry Smith Mat **B) 1516cddf8d76SBarry Smith { 1517cddf8d76SBarry Smith int ierr,i; 1518cddf8d76SBarry Smith 1519*3a40ed3dSBarry Smith PetscFunctionBegin; 1520cddf8d76SBarry Smith if (scall == MAT_INITIAL_MATRIX) { 15210452661fSBarry Smith *B = (Mat *) PetscMalloc( (n+1)*sizeof(Mat) ); CHKPTRQ(*B); 1522cddf8d76SBarry Smith } 1523cddf8d76SBarry Smith 1524cddf8d76SBarry Smith for ( i=0; i<n; i++ ) { 1525905e6a2fSBarry Smith ierr = MatGetSubMatrix_SeqAIJ(A,irow[i],icol[i],scall,&(*B)[i]);CHKERRQ(ierr); 1526cddf8d76SBarry Smith } 1527*3a40ed3dSBarry Smith PetscFunctionReturn(0); 1528cddf8d76SBarry Smith } 1529cddf8d76SBarry Smith 15305615d1e5SSatish Balay #undef __FUNC__ 15315615d1e5SSatish Balay #define __FUNC__ "MatGetBlockSize_SeqAIJ" 15328f6be9afSLois Curfman McInnes int MatGetBlockSize_SeqAIJ(Mat A, int *bs) 15335a838052SSatish Balay { 15345a838052SSatish Balay *bs = 1; 1535*3a40ed3dSBarry Smith PetscFunctionReturn(0); 15365a838052SSatish Balay } 15375a838052SSatish Balay 15385615d1e5SSatish Balay #undef __FUNC__ 15395615d1e5SSatish Balay #define __FUNC__ "MatIncreaseOverlap_SeqAIJ" 15408f6be9afSLois Curfman McInnes int MatIncreaseOverlap_SeqAIJ(Mat A, int is_max, IS *is, int ov) 15414dcbc457SBarry Smith { 1542e4d965acSSatish Balay Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 154306763907SSatish Balay int shift, row, i,j,k,l,m,n, *idx,ierr, *nidx, isz, val; 15448a047759SSatish Balay int start, end, *ai, *aj; 1545bbd702dbSSatish Balay BT table; 1546bbd702dbSSatish Balay 1547*3a40ed3dSBarry Smith PetscFunctionBegin; 15488a047759SSatish Balay shift = a->indexshift; 1549e4d965acSSatish Balay m = a->m; 1550e4d965acSSatish Balay ai = a->i; 15518a047759SSatish Balay aj = a->j+shift; 15528a047759SSatish Balay 1553e3372554SBarry Smith if (ov < 0) SETERRQ(1,0,"illegal overlap value used"); 155406763907SSatish Balay 155506763907SSatish Balay nidx = (int *) PetscMalloc((m+1)*sizeof(int)); CHKPTRQ(nidx); 1556bbd702dbSSatish Balay ierr = BTCreate(m,table); CHKERRQ(ierr); 155706763907SSatish Balay 1558e4d965acSSatish Balay for ( i=0; i<is_max; i++ ) { 1559b97fc60eSLois Curfman McInnes /* Initialize the two local arrays */ 1560e4d965acSSatish Balay isz = 0; 1561bbd702dbSSatish Balay BTMemzero(m,table); 1562e4d965acSSatish Balay 1563e4d965acSSatish Balay /* Extract the indices, assume there can be duplicate entries */ 15644dcbc457SBarry Smith ierr = ISGetIndices(is[i],&idx); CHKERRQ(ierr); 156577c4ece6SBarry Smith ierr = ISGetSize(is[i],&n); CHKERRQ(ierr); 1566e4d965acSSatish Balay 1567dd097bc3SLois Curfman McInnes /* Enter these into the temp arrays. I.e., mark table[row], enter row into new index */ 1568e4d965acSSatish Balay for ( j=0; j<n ; ++j){ 1569bbd702dbSSatish Balay if(!BTLookupSet(table, idx[j])) { nidx[isz++] = idx[j];} 15704dcbc457SBarry Smith } 157106763907SSatish Balay ierr = ISRestoreIndices(is[i],&idx); CHKERRQ(ierr); 157206763907SSatish Balay ierr = ISDestroy(is[i]); CHKERRQ(ierr); 1573e4d965acSSatish Balay 157404a348a9SBarry Smith k = 0; 157504a348a9SBarry Smith for ( j=0; j<ov; j++){ /* for each overlap */ 157604a348a9SBarry Smith n = isz; 157706763907SSatish Balay for ( ; k<n ; k++){ /* do only those rows in nidx[k], which are not done yet */ 1578e4d965acSSatish Balay row = nidx[k]; 1579e4d965acSSatish Balay start = ai[row]; 1580e4d965acSSatish Balay end = ai[row+1]; 158104a348a9SBarry Smith for ( l = start; l<end ; l++){ 15828a047759SSatish Balay val = aj[l] + shift; 15832a8f2162SSatish Balay if (!BTLookupSet(table,val)) {nidx[isz++] = val;} 1584e4d965acSSatish Balay } 1585e4d965acSSatish Balay } 1586e4d965acSSatish Balay } 1587029af93fSBarry Smith ierr = ISCreateGeneral(PETSC_COMM_SELF, isz, nidx, (is+i)); CHKERRQ(ierr); 1588e4d965acSSatish Balay } 1589bbd702dbSSatish Balay BTDestroy(table); 159006763907SSatish Balay PetscFree(nidx); 1591*3a40ed3dSBarry Smith PetscFunctionReturn(0); 15924dcbc457SBarry Smith } 159317ab2063SBarry Smith 15940513a670SBarry Smith /* -------------------------------------------------------------- */ 15950513a670SBarry Smith #undef __FUNC__ 15960513a670SBarry Smith #define __FUNC__ "MatPermute_SeqAIJ" 15970513a670SBarry Smith int MatPermute_SeqAIJ(Mat A, IS rowp, IS colp, Mat *B) 15980513a670SBarry Smith { 15990513a670SBarry Smith Mat_SeqAIJ *a = (Mat_SeqAIJ *) A->data; 16000513a670SBarry Smith Scalar *vwork; 16010513a670SBarry Smith int i, ierr, nz, m = a->m, n = a->n, *cwork; 16020513a670SBarry Smith int *row,*col,*cnew,j,*lens; 160356cd22aeSBarry Smith IS icolp,irowp; 16040513a670SBarry Smith 1605*3a40ed3dSBarry Smith PetscFunctionBegin; 160656cd22aeSBarry Smith ierr = ISInvertPermutation(rowp,&irowp); CHKERRQ(ierr); 160756cd22aeSBarry Smith ierr = ISGetIndices(irowp,&row); CHKERRQ(ierr); 160856cd22aeSBarry Smith ierr = ISInvertPermutation(colp,&icolp); CHKERRQ(ierr); 160956cd22aeSBarry Smith ierr = ISGetIndices(icolp,&col); CHKERRQ(ierr); 16100513a670SBarry Smith 16110513a670SBarry Smith /* determine lengths of permuted rows */ 16120513a670SBarry Smith lens = (int *) PetscMalloc( (m+1)*sizeof(int) ); CHKPTRQ(lens); 16130513a670SBarry Smith for (i=0; i<m; i++ ) { 16140513a670SBarry Smith lens[row[i]] = a->i[i+1] - a->i[i]; 16150513a670SBarry Smith } 16160513a670SBarry Smith ierr = MatCreateSeqAIJ(A->comm,m,n,0,lens,B);CHKERRQ(ierr); 16170513a670SBarry Smith PetscFree(lens); 16180513a670SBarry Smith 16190513a670SBarry Smith cnew = (int *) PetscMalloc( n*sizeof(int) ); CHKPTRQ(cnew); 16200513a670SBarry Smith for (i=0; i<m; i++) { 16210513a670SBarry Smith ierr = MatGetRow(A,i,&nz,&cwork,&vwork); CHKERRQ(ierr); 16220513a670SBarry Smith for (j=0; j<nz; j++ ) { cnew[j] = col[cwork[j]];} 16230513a670SBarry Smith ierr = MatSetValues(*B,1,&row[i],nz,cnew,vwork,INSERT_VALUES); CHKERRQ(ierr); 16240513a670SBarry Smith ierr = MatRestoreRow(A,i,&nz,&cwork,&vwork); CHKERRQ(ierr); 16250513a670SBarry Smith } 16260513a670SBarry Smith PetscFree(cnew); 16270513a670SBarry Smith ierr = MatAssemblyBegin(*B,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr); 16280513a670SBarry Smith ierr = MatAssemblyEnd(*B,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr); 162956cd22aeSBarry Smith ierr = ISRestoreIndices(irowp,&row); CHKERRQ(ierr); 163056cd22aeSBarry Smith ierr = ISRestoreIndices(icolp,&col); CHKERRQ(ierr); 163156cd22aeSBarry Smith ierr = ISDestroy(irowp); CHKERRQ(ierr); 163256cd22aeSBarry Smith ierr = ISDestroy(icolp); CHKERRQ(ierr); 1633*3a40ed3dSBarry Smith PetscFunctionReturn(0); 16340513a670SBarry Smith } 16350513a670SBarry Smith 16365615d1e5SSatish Balay #undef __FUNC__ 16375615d1e5SSatish Balay #define __FUNC__ "MatPrintHelp_SeqAIJ" 1638682d7d0cSBarry Smith int MatPrintHelp_SeqAIJ(Mat A) 1639682d7d0cSBarry Smith { 1640682d7d0cSBarry Smith static int called = 0; 1641682d7d0cSBarry Smith MPI_Comm comm = A->comm; 1642682d7d0cSBarry Smith 1643*3a40ed3dSBarry Smith PetscFunctionBegin; 1644*3a40ed3dSBarry Smith if (called) {PetscFunctionReturn(0);} else called = 1; 164577c4ece6SBarry Smith PetscPrintf(comm," Options for MATSEQAIJ and MATMPIAIJ matrix formats (the defaults):\n"); 16460f665d81SLois Curfman McInnes PetscPrintf(comm," -mat_lu_pivotthreshold <threshold>: Set pivoting threshold\n"); 16470f665d81SLois Curfman McInnes PetscPrintf(comm," -mat_aij_oneindex: internal indices begin at 1 instead of the default 0.\n"); 16480f665d81SLois Curfman McInnes PetscPrintf(comm," -mat_aij_no_inode: Do not use inodes\n"); 16490f665d81SLois Curfman McInnes PetscPrintf(comm," -mat_aij_inode_limit <limit>: Set inode limit (max limit=5)\n"); 1650682d7d0cSBarry Smith #if defined(HAVE_ESSL) 16510f665d81SLois Curfman McInnes PetscPrintf(comm," -mat_aij_essl: Use IBM sparse LU factorization and solve.\n"); 1652682d7d0cSBarry Smith #endif 1653*3a40ed3dSBarry Smith PetscFunctionReturn(0); 1654682d7d0cSBarry Smith } 16558f6be9afSLois Curfman McInnes extern int MatEqual_SeqAIJ(Mat A,Mat B, PetscTruth* flg); 1656a93ec695SBarry Smith extern int MatFDColoringCreate_SeqAIJ(Mat,ISColoring,MatFDColoring); 1657a93ec695SBarry Smith extern int MatColoringPatch_SeqAIJ(Mat,int,int *,ISColoring *); 1658a93ec695SBarry Smith 1659682d7d0cSBarry Smith /* -------------------------------------------------------------------*/ 166017ab2063SBarry Smith static struct _MatOps MatOps = {MatSetValues_SeqAIJ, 166117ab2063SBarry Smith MatGetRow_SeqAIJ,MatRestoreRow_SeqAIJ, 1662416022c9SBarry Smith MatMult_SeqAIJ,MatMultAdd_SeqAIJ, 1663416022c9SBarry Smith MatMultTrans_SeqAIJ,MatMultTransAdd_SeqAIJ, 166417ab2063SBarry Smith MatSolve_SeqAIJ,MatSolveAdd_SeqAIJ, 166517ab2063SBarry Smith MatSolveTrans_SeqAIJ,MatSolveTransAdd_SeqAIJ, 166617ab2063SBarry Smith MatLUFactor_SeqAIJ,0, 166717ab2063SBarry Smith MatRelax_SeqAIJ, 166817ab2063SBarry Smith MatTranspose_SeqAIJ, 16697264ac53SSatish Balay MatGetInfo_SeqAIJ,MatEqual_SeqAIJ, 1670f0b747eeSBarry Smith MatGetDiagonal_SeqAIJ,MatDiagonalScale_SeqAIJ,MatNorm_SeqAIJ, 167117ab2063SBarry Smith 0,MatAssemblyEnd_SeqAIJ, 167217ab2063SBarry Smith MatCompress_SeqAIJ, 167317ab2063SBarry Smith MatSetOption_SeqAIJ,MatZeroEntries_SeqAIJ,MatZeroRows_SeqAIJ, 167417ab2063SBarry Smith MatLUFactorSymbolic_SeqAIJ,MatLUFactorNumeric_SeqAIJ,0,0, 167517ab2063SBarry Smith MatGetSize_SeqAIJ,MatGetSize_SeqAIJ,MatGetOwnershipRange_SeqAIJ, 167617ab2063SBarry Smith MatILUFactorSymbolic_SeqAIJ,0, 167794a9d846SBarry Smith 0,0, 16783d1612f7SBarry Smith MatConvertSameType_SeqAIJ,0,0, 1679cddf8d76SBarry Smith MatILUFactor_SeqAIJ,0,0, 16807eb43aa7SLois Curfman McInnes MatGetSubMatrices_SeqAIJ,MatIncreaseOverlap_SeqAIJ, 1681682d7d0cSBarry Smith MatGetValues_SeqAIJ,0, 1682f0b747eeSBarry Smith MatPrintHelp_SeqAIJ, 16835a838052SSatish Balay MatScale_SeqAIJ,0,0, 16846945ee14SBarry Smith MatILUDTFactor_SeqAIJ, 16856945ee14SBarry Smith MatGetBlockSize_SeqAIJ, 16863b2fbd54SBarry Smith MatGetRowIJ_SeqAIJ, 16873b2fbd54SBarry Smith MatRestoreRowIJ_SeqAIJ, 16883b2fbd54SBarry Smith MatGetColumnIJ_SeqAIJ, 1689a93ec695SBarry Smith MatRestoreColumnIJ_SeqAIJ, 1690a93ec695SBarry Smith MatFDColoringCreate_SeqAIJ, 16910513a670SBarry Smith MatColoringPatch_SeqAIJ, 16920513a670SBarry Smith 0, 16930513a670SBarry Smith MatPermute_SeqAIJ}; 169417ab2063SBarry Smith 169517ab2063SBarry Smith extern int MatUseSuperLU_SeqAIJ(Mat); 169617ab2063SBarry Smith extern int MatUseEssl_SeqAIJ(Mat); 169717ab2063SBarry Smith extern int MatUseDXML_SeqAIJ(Mat); 169817ab2063SBarry Smith 16995615d1e5SSatish Balay #undef __FUNC__ 17005615d1e5SSatish Balay #define __FUNC__ "MatCreateSeqAIJ" 170117ab2063SBarry Smith /*@C 1702682d7d0cSBarry Smith MatCreateSeqAIJ - Creates a sparse matrix in AIJ (compressed row) format 17030d15e28bSLois Curfman McInnes (the default parallel PETSc format). For good matrix assembly performance 17046e62573dSLois Curfman McInnes the user should preallocate the matrix storage by setting the parameter nz 17052bd5e0b2SLois Curfman McInnes (or the array nzz). By setting these parameters accurately, performance 17062bd5e0b2SLois Curfman McInnes during matrix assembly can be increased by more than a factor of 50. 170717ab2063SBarry Smith 170817ab2063SBarry Smith Input Parameters: 1709029af93fSBarry Smith . comm - MPI communicator, set to PETSC_COMM_SELF 171017ab2063SBarry Smith . m - number of rows 171117ab2063SBarry Smith . n - number of columns 171217ab2063SBarry Smith . nz - number of nonzeros per row (same for all rows) 17132bd5e0b2SLois Curfman McInnes . nzz - array containing the number of nonzeros in the various rows 17142bd5e0b2SLois Curfman McInnes (possibly different for each row) or PETSC_NULL 171517ab2063SBarry Smith 171617ab2063SBarry Smith Output Parameter: 1717416022c9SBarry Smith . A - the matrix 171817ab2063SBarry Smith 171917ab2063SBarry Smith Notes: 172017ab2063SBarry Smith The AIJ format (also called the Yale sparse matrix format or 172117ab2063SBarry Smith compressed row storage), is fully compatible with standard Fortran 77 17220002213bSLois Curfman McInnes storage. That is, the stored row and column indices can begin at 172344cd7ae7SLois Curfman McInnes either one (as in Fortran) or zero. See the users' manual for details. 172417ab2063SBarry Smith 172517ab2063SBarry Smith Specify the preallocated storage with either nz or nnz (not both). 1726a40aa06bSLois Curfman McInnes Set nz=PETSC_DEFAULT and nnz=PETSC_NULL for PETSc to control dynamic memory 17273d323bbdSBarry Smith allocation. For large problems you MUST preallocate memory or you 17286da5968aSLois Curfman McInnes will get TERRIBLE performance, see the users' manual chapter on matrices. 172917ab2063SBarry Smith 1730682d7d0cSBarry Smith By default, this format uses inodes (identical nodes) when possible, to 17314fca80b9SLois Curfman McInnes improve numerical efficiency of matrix-vector products and solves. We 1732682d7d0cSBarry Smith search for consecutive rows with the same nonzero structure, thereby 17336c7ebb05SLois Curfman McInnes reusing matrix information to achieve increased efficiency. 17346c7ebb05SLois Curfman McInnes 17356c7ebb05SLois Curfman McInnes Options Database Keys: 17366c7ebb05SLois Curfman McInnes $ -mat_aij_no_inode - Do not use inodes 17376e62573dSLois Curfman McInnes $ -mat_aij_inode_limit <limit> - Set inode limit. 17386e62573dSLois Curfman McInnes $ (max limit=5) 17396e62573dSLois Curfman McInnes $ -mat_aij_oneindex - Internally use indexing starting at 1 17406e62573dSLois Curfman McInnes $ rather than 0. Note: When calling MatSetValues(), 17416e62573dSLois Curfman McInnes $ the user still MUST index entries starting at 0! 174217ab2063SBarry Smith 174317ab2063SBarry Smith .seealso: MatCreate(), MatCreateMPIAIJ(), MatSetValues() 174417ab2063SBarry Smith @*/ 1745416022c9SBarry Smith int MatCreateSeqAIJ(MPI_Comm comm,int m,int n,int nz,int *nnz, Mat *A) 174617ab2063SBarry Smith { 1747416022c9SBarry Smith Mat B; 1748416022c9SBarry Smith Mat_SeqAIJ *b; 17496945ee14SBarry Smith int i, len, ierr, flg,size; 17506945ee14SBarry Smith 1751*3a40ed3dSBarry Smith PetscFunctionBegin; 17526945ee14SBarry Smith MPI_Comm_size(comm,&size); 1753e3372554SBarry Smith if (size > 1) SETERRQ(1,0,"Comm must be of size 1"); 1754d5d45c9bSBarry Smith 1755416022c9SBarry Smith *A = 0; 1756d4bb536fSBarry Smith PetscHeaderCreate(B,_p_Mat,MAT_COOKIE,MATSEQAIJ,comm,MatDestroy,MatView); 1757416022c9SBarry Smith PLogObjectCreate(B); 17580452661fSBarry Smith B->data = (void *) (b = PetscNew(Mat_SeqAIJ)); CHKPTRQ(b); 175944cd7ae7SLois Curfman McInnes PetscMemzero(b,sizeof(Mat_SeqAIJ)); 1760416022c9SBarry Smith PetscMemcpy(&B->ops,&MatOps,sizeof(struct _MatOps)); 1761416022c9SBarry Smith B->destroy = MatDestroy_SeqAIJ; 1762416022c9SBarry Smith B->view = MatView_SeqAIJ; 1763416022c9SBarry Smith B->factor = 0; 1764416022c9SBarry Smith B->lupivotthreshold = 1.0; 176590f02eecSBarry Smith B->mapping = 0; 17667a743949SBarry Smith ierr = OptionsGetDouble(PETSC_NULL,"-mat_lu_pivotthreshold",&B->lupivotthreshold, 176769957df2SSatish Balay &flg); CHKERRQ(ierr); 17687a743949SBarry Smith b->ilu_preserve_row_sums = PETSC_FALSE; 17697a743949SBarry Smith ierr = OptionsHasName(PETSC_NULL,"-pc_ilu_preserve_row_sums", 17707a743949SBarry Smith (int*) &b->ilu_preserve_row_sums); CHKERRQ(ierr); 1771416022c9SBarry Smith b->row = 0; 1772416022c9SBarry Smith b->col = 0; 1773416022c9SBarry Smith b->indexshift = 0; 1774b810aeb4SBarry Smith b->reallocs = 0; 177569957df2SSatish Balay ierr = OptionsHasName(PETSC_NULL,"-mat_aij_oneindex", &flg); CHKERRQ(ierr); 177669957df2SSatish Balay if (flg) b->indexshift = -1; 177717ab2063SBarry Smith 177844cd7ae7SLois Curfman McInnes b->m = m; B->m = m; B->M = m; 177944cd7ae7SLois Curfman McInnes b->n = n; B->n = n; B->N = n; 17800452661fSBarry Smith b->imax = (int *) PetscMalloc( (m+1)*sizeof(int) ); CHKPTRQ(b->imax); 1781b4fd4287SBarry Smith if (nnz == PETSC_NULL) { 17827b8455f0SLois Curfman McInnes if (nz == PETSC_DEFAULT) nz = 10; 17837b8455f0SLois Curfman McInnes else if (nz <= 0) nz = 1; 1784416022c9SBarry Smith for ( i=0; i<m; i++ ) b->imax[i] = nz; 178517ab2063SBarry Smith nz = nz*m; 1786*3a40ed3dSBarry Smith } else { 178717ab2063SBarry Smith nz = 0; 1788416022c9SBarry Smith for ( i=0; i<m; i++ ) {b->imax[i] = nnz[i]; nz += nnz[i];} 178917ab2063SBarry Smith } 179017ab2063SBarry Smith 179117ab2063SBarry Smith /* allocate the matrix space */ 1792416022c9SBarry Smith len = nz*(sizeof(int) + sizeof(Scalar)) + (b->m+1)*sizeof(int); 17930452661fSBarry Smith b->a = (Scalar *) PetscMalloc( len ); CHKPTRQ(b->a); 1794416022c9SBarry Smith b->j = (int *) (b->a + nz); 1795cddf8d76SBarry Smith PetscMemzero(b->j,nz*sizeof(int)); 1796416022c9SBarry Smith b->i = b->j + nz; 1797416022c9SBarry Smith b->singlemalloc = 1; 179817ab2063SBarry Smith 1799416022c9SBarry Smith b->i[0] = -b->indexshift; 180017ab2063SBarry Smith for (i=1; i<m+1; i++) { 1801416022c9SBarry Smith b->i[i] = b->i[i-1] + b->imax[i-1]; 180217ab2063SBarry Smith } 180317ab2063SBarry Smith 1804416022c9SBarry Smith /* b->ilen will count nonzeros in each row so far. */ 18050452661fSBarry Smith b->ilen = (int *) PetscMalloc((m+1)*sizeof(int)); 1806f09e8eb9SSatish Balay PLogObjectMemory(B,len+2*(m+1)*sizeof(int)+sizeof(struct _p_Mat)+sizeof(Mat_SeqAIJ)); 1807416022c9SBarry Smith for ( i=0; i<b->m; i++ ) { b->ilen[i] = 0;} 180817ab2063SBarry Smith 1809416022c9SBarry Smith b->nz = 0; 1810416022c9SBarry Smith b->maxnz = nz; 1811416022c9SBarry Smith b->sorted = 0; 1812416022c9SBarry Smith b->roworiented = 1; 1813416022c9SBarry Smith b->nonew = 0; 1814416022c9SBarry Smith b->diag = 0; 1815416022c9SBarry Smith b->solve_work = 0; 181671bd300dSLois Curfman McInnes b->spptr = 0; 1817754ec7b1SSatish Balay b->inode.node_count = 0; 1818754ec7b1SSatish Balay b->inode.size = 0; 18196c7ebb05SLois Curfman McInnes b->inode.limit = 5; 18206c7ebb05SLois Curfman McInnes b->inode.max_limit = 5; 18214e220ebcSLois Curfman McInnes B->info.nz_unneeded = (double)b->maxnz; 182217ab2063SBarry Smith 1823416022c9SBarry Smith *A = B; 18244e220ebcSLois Curfman McInnes 18254b14c69eSBarry Smith /* SuperLU is not currently supported through PETSc */ 18264b14c69eSBarry Smith #if defined(HAVE_SUPERLU) 182769957df2SSatish Balay ierr = OptionsHasName(PETSC_NULL,"-mat_aij_superlu", &flg); CHKERRQ(ierr); 182869957df2SSatish Balay if (flg) { ierr = MatUseSuperLU_SeqAIJ(B); CHKERRQ(ierr); } 18294b14c69eSBarry Smith #endif 183069957df2SSatish Balay ierr = OptionsHasName(PETSC_NULL,"-mat_aij_essl", &flg); CHKERRQ(ierr); 183169957df2SSatish Balay if (flg) { ierr = MatUseEssl_SeqAIJ(B); CHKERRQ(ierr); } 183269957df2SSatish Balay ierr = OptionsHasName(PETSC_NULL,"-mat_aij_dxml", &flg); CHKERRQ(ierr); 183369957df2SSatish Balay if (flg) { 1834e3372554SBarry Smith if (!b->indexshift) SETERRQ(1,0,"need -mat_aij_oneindex with -mat_aij_dxml"); 1835416022c9SBarry Smith ierr = MatUseDXML_SeqAIJ(B); CHKERRQ(ierr); 183617ab2063SBarry Smith } 183769957df2SSatish Balay ierr = OptionsHasName(PETSC_NULL,"-help", &flg); CHKERRQ(ierr); 183869957df2SSatish Balay if (flg) {ierr = MatPrintHelp(B); CHKERRQ(ierr); } 1839*3a40ed3dSBarry Smith PetscFunctionReturn(0); 184017ab2063SBarry Smith } 184117ab2063SBarry Smith 18425615d1e5SSatish Balay #undef __FUNC__ 18435615d1e5SSatish Balay #define __FUNC__ "MatConvertSameType_SeqAIJ" 18443d1612f7SBarry Smith int MatConvertSameType_SeqAIJ(Mat A,Mat *B,int cpvalues) 184517ab2063SBarry Smith { 1846416022c9SBarry Smith Mat C; 1847416022c9SBarry Smith Mat_SeqAIJ *c,*a = (Mat_SeqAIJ *) A->data; 184808480c60SBarry Smith int i,len, m = a->m,shift = a->indexshift; 184917ab2063SBarry Smith 1850*3a40ed3dSBarry Smith PetscFunctionBegin; 18514043dd9cSLois Curfman McInnes *B = 0; 1852d4bb536fSBarry Smith PetscHeaderCreate(C,_p_Mat,MAT_COOKIE,MATSEQAIJ,A->comm,MatDestroy,MatView); 1853416022c9SBarry Smith PLogObjectCreate(C); 18540452661fSBarry Smith C->data = (void *) (c = PetscNew(Mat_SeqAIJ)); CHKPTRQ(c); 185541c01911SSatish Balay PetscMemcpy(&C->ops,&A->ops,sizeof(struct _MatOps)); 1856416022c9SBarry Smith C->destroy = MatDestroy_SeqAIJ; 1857416022c9SBarry Smith C->view = MatView_SeqAIJ; 1858416022c9SBarry Smith C->factor = A->factor; 1859416022c9SBarry Smith c->row = 0; 1860416022c9SBarry Smith c->col = 0; 1861416022c9SBarry Smith c->indexshift = shift; 1862c456f294SBarry Smith C->assembled = PETSC_TRUE; 186317ab2063SBarry Smith 186444cd7ae7SLois Curfman McInnes c->m = C->m = a->m; 186544cd7ae7SLois Curfman McInnes c->n = C->n = a->n; 186644cd7ae7SLois Curfman McInnes C->M = a->m; 186744cd7ae7SLois Curfman McInnes C->N = a->n; 186817ab2063SBarry Smith 18690452661fSBarry Smith c->imax = (int *) PetscMalloc((m+1)*sizeof(int)); CHKPTRQ(c->imax); 18700452661fSBarry Smith c->ilen = (int *) PetscMalloc((m+1)*sizeof(int)); CHKPTRQ(c->ilen); 187117ab2063SBarry Smith for ( i=0; i<m; i++ ) { 1872416022c9SBarry Smith c->imax[i] = a->imax[i]; 1873416022c9SBarry Smith c->ilen[i] = a->ilen[i]; 187417ab2063SBarry Smith } 187517ab2063SBarry Smith 187617ab2063SBarry Smith /* allocate the matrix space */ 1877416022c9SBarry Smith c->singlemalloc = 1; 1878416022c9SBarry Smith len = (m+1)*sizeof(int)+(a->i[m])*(sizeof(Scalar)+sizeof(int)); 18790452661fSBarry Smith c->a = (Scalar *) PetscMalloc( len ); CHKPTRQ(c->a); 1880416022c9SBarry Smith c->j = (int *) (c->a + a->i[m] + shift); 1881416022c9SBarry Smith c->i = c->j + a->i[m] + shift; 1882416022c9SBarry Smith PetscMemcpy(c->i,a->i,(m+1)*sizeof(int)); 188317ab2063SBarry Smith if (m > 0) { 1884416022c9SBarry Smith PetscMemcpy(c->j,a->j,(a->i[m]+shift)*sizeof(int)); 188508480c60SBarry Smith if (cpvalues == COPY_VALUES) { 1886416022c9SBarry Smith PetscMemcpy(c->a,a->a,(a->i[m]+shift)*sizeof(Scalar)); 188717ab2063SBarry Smith } 188808480c60SBarry Smith } 188917ab2063SBarry Smith 1890f09e8eb9SSatish Balay PLogObjectMemory(C,len+2*(m+1)*sizeof(int)+sizeof(struct _p_Mat)+sizeof(Mat_SeqAIJ)); 1891416022c9SBarry Smith c->sorted = a->sorted; 1892416022c9SBarry Smith c->roworiented = a->roworiented; 1893416022c9SBarry Smith c->nonew = a->nonew; 18947a743949SBarry Smith c->ilu_preserve_row_sums = a->ilu_preserve_row_sums; 189517ab2063SBarry Smith 1896416022c9SBarry Smith if (a->diag) { 18970452661fSBarry Smith c->diag = (int *) PetscMalloc( (m+1)*sizeof(int) ); CHKPTRQ(c->diag); 1898416022c9SBarry Smith PLogObjectMemory(C,(m+1)*sizeof(int)); 189917ab2063SBarry Smith for ( i=0; i<m; i++ ) { 1900416022c9SBarry Smith c->diag[i] = a->diag[i]; 190117ab2063SBarry Smith } 1902*3a40ed3dSBarry Smith } else c->diag = 0; 19036c7ebb05SLois Curfman McInnes c->inode.limit = a->inode.limit; 19046c7ebb05SLois Curfman McInnes c->inode.max_limit = a->inode.max_limit; 1905754ec7b1SSatish Balay if (a->inode.size){ 1906daed632aSSatish Balay c->inode.size = (int *) PetscMalloc( (m+1)*sizeof(int) ); CHKPTRQ(c->inode.size); 1907754ec7b1SSatish Balay c->inode.node_count = a->inode.node_count; 1908daed632aSSatish Balay PetscMemcpy( c->inode.size, a->inode.size, (m+1)*sizeof(int)); 1909754ec7b1SSatish Balay } else { 1910754ec7b1SSatish Balay c->inode.size = 0; 1911754ec7b1SSatish Balay c->inode.node_count = 0; 1912754ec7b1SSatish Balay } 1913416022c9SBarry Smith c->nz = a->nz; 1914416022c9SBarry Smith c->maxnz = a->maxnz; 1915416022c9SBarry Smith c->solve_work = 0; 191676dd722bSSatish Balay c->spptr = 0; /* Dangerous -I'm throwing away a->spptr */ 1917754ec7b1SSatish Balay 1918416022c9SBarry Smith *B = C; 1919*3a40ed3dSBarry Smith PetscFunctionReturn(0); 192017ab2063SBarry Smith } 192117ab2063SBarry Smith 19225615d1e5SSatish Balay #undef __FUNC__ 19235615d1e5SSatish Balay #define __FUNC__ "MatLoad_SeqAIJ" 192419bcc07fSBarry Smith int MatLoad_SeqAIJ(Viewer viewer,MatType type,Mat *A) 192517ab2063SBarry Smith { 1926416022c9SBarry Smith Mat_SeqAIJ *a; 1927416022c9SBarry Smith Mat B; 192817699dbbSLois Curfman McInnes int i, nz, ierr, fd, header[4],size,*rowlengths = 0,M,N,shift; 1929bcd2baecSBarry Smith MPI_Comm comm; 193017ab2063SBarry Smith 1931*3a40ed3dSBarry Smith PetscFunctionBegin; 193219bcc07fSBarry Smith PetscObjectGetComm((PetscObject) viewer,&comm); 193317699dbbSLois Curfman McInnes MPI_Comm_size(comm,&size); 1934e3372554SBarry Smith if (size > 1) SETERRQ(1,0,"view must have one processor"); 193590ace30eSBarry Smith ierr = ViewerBinaryGetDescriptor(viewer,&fd); CHKERRQ(ierr); 19360752156aSBarry Smith ierr = PetscBinaryRead(fd,header,4,PETSC_INT); CHKERRQ(ierr); 1937e3372554SBarry Smith if (header[0] != MAT_COOKIE) SETERRQ(1,0,"not matrix object in file"); 193817ab2063SBarry Smith M = header[1]; N = header[2]; nz = header[3]; 193917ab2063SBarry Smith 194017ab2063SBarry Smith /* read in row lengths */ 19410452661fSBarry Smith rowlengths = (int*) PetscMalloc( M*sizeof(int) ); CHKPTRQ(rowlengths); 19420752156aSBarry Smith ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT); CHKERRQ(ierr); 194317ab2063SBarry Smith 194417ab2063SBarry Smith /* create our matrix */ 1945416022c9SBarry Smith ierr = MatCreateSeqAIJ(comm,M,N,0,rowlengths,A); CHKERRQ(ierr); 1946416022c9SBarry Smith B = *A; 1947416022c9SBarry Smith a = (Mat_SeqAIJ *) B->data; 1948416022c9SBarry Smith shift = a->indexshift; 194917ab2063SBarry Smith 195017ab2063SBarry Smith /* read in column indices and adjust for Fortran indexing*/ 19510752156aSBarry Smith ierr = PetscBinaryRead(fd,a->j,nz,PETSC_INT); CHKERRQ(ierr); 195217ab2063SBarry Smith if (shift) { 195317ab2063SBarry Smith for ( i=0; i<nz; i++ ) { 1954416022c9SBarry Smith a->j[i] += 1; 195517ab2063SBarry Smith } 195617ab2063SBarry Smith } 195717ab2063SBarry Smith 195817ab2063SBarry Smith /* read in nonzero values */ 19590752156aSBarry Smith ierr = PetscBinaryRead(fd,a->a,nz,PETSC_SCALAR); CHKERRQ(ierr); 196017ab2063SBarry Smith 196117ab2063SBarry Smith /* set matrix "i" values */ 1962416022c9SBarry Smith a->i[0] = -shift; 196317ab2063SBarry Smith for ( i=1; i<= M; i++ ) { 1964416022c9SBarry Smith a->i[i] = a->i[i-1] + rowlengths[i-1]; 1965416022c9SBarry Smith a->ilen[i-1] = rowlengths[i-1]; 196617ab2063SBarry Smith } 19670452661fSBarry Smith PetscFree(rowlengths); 196817ab2063SBarry Smith 19696d4a8577SBarry Smith ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr); 19706d4a8577SBarry Smith ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr); 1971*3a40ed3dSBarry Smith PetscFunctionReturn(0); 197217ab2063SBarry Smith } 197317ab2063SBarry Smith 19745615d1e5SSatish Balay #undef __FUNC__ 19755615d1e5SSatish Balay #define __FUNC__ "MatEqual_SeqAIJ" 19768f6be9afSLois Curfman McInnes int MatEqual_SeqAIJ(Mat A,Mat B, PetscTruth* flg) 19777264ac53SSatish Balay { 19787264ac53SSatish Balay Mat_SeqAIJ *a = (Mat_SeqAIJ *)A->data, *b = (Mat_SeqAIJ *)B->data; 19797264ac53SSatish Balay 1980*3a40ed3dSBarry Smith PetscFunctionBegin; 1981e3372554SBarry Smith if (B->type !=MATSEQAIJ)SETERRQ(1,0,"Matrices must be same type"); 19827264ac53SSatish Balay 19837264ac53SSatish Balay /* If the matrix dimensions are not equal, or no of nonzeros or shift */ 19847264ac53SSatish Balay if ((a->m != b->m ) || (a->n !=b->n) ||( a->nz != b->nz)|| 1985bcd2baecSBarry Smith (a->indexshift != b->indexshift)) { 1986*3a40ed3dSBarry Smith *flg = PETSC_FALSE; PetscFunctionReturn(0); 1987bcd2baecSBarry Smith } 19887264ac53SSatish Balay 19897264ac53SSatish Balay /* if the a->i are the same */ 19908108c231SLois Curfman McInnes if (PetscMemcmp(a->i,b->i,(a->m+1)*sizeof(int))) { 1991*3a40ed3dSBarry Smith *flg = PETSC_FALSE; PetscFunctionReturn(0); 19927264ac53SSatish Balay } 19937264ac53SSatish Balay 19947264ac53SSatish Balay /* if a->j are the same */ 1995bcd2baecSBarry Smith if (PetscMemcmp(a->j, b->j, (a->nz)*sizeof(int))) { 1996*3a40ed3dSBarry Smith *flg = PETSC_FALSE; PetscFunctionReturn(0); 1997bcd2baecSBarry Smith } 1998bcd2baecSBarry Smith 1999bcd2baecSBarry Smith /* if a->a are the same */ 200019bcc07fSBarry Smith if (PetscMemcmp(a->a, b->a, (a->nz)*sizeof(Scalar))) { 2001*3a40ed3dSBarry Smith *flg = PETSC_FALSE; PetscFunctionReturn(0); 20027264ac53SSatish Balay } 200377c4ece6SBarry Smith *flg = PETSC_TRUE; 2004*3a40ed3dSBarry Smith PetscFunctionReturn(0); 20057264ac53SSatish Balay 20067264ac53SSatish Balay } 2007