1*329f5518SBarry Smith /* $Id: aij.h,v 1.36 1999/11/24 21:53:47 bsmith Exp bsmith $ */ 2b8a66259SBarry Smith 370f55243SBarry Smith #include "src/mat/matimpl.h" 4b8a66259SBarry Smith 52d40f771SBarry Smith #if !defined(__AIJ_H) 62d40f771SBarry Smith #define __AIJ_H 72d40f771SBarry Smith 8ed2daf61SLois Curfman McInnes /* Info about i-nodes (identical nodes) */ 9957e4c30SSatish Balay typedef struct { 10ed2daf61SLois Curfman McInnes int node_count; /* number of inodes */ 11ed2daf61SLois Curfman McInnes int *size; /* size of each inode */ 12ed2daf61SLois Curfman McInnes int limit; /* inode limit */ 13ed2daf61SLois Curfman McInnes int max_limit; /* maximum supported inode limit */ 1476dd722bSSatish Balay } Mat_SeqAIJ_Inode; 15957e4c30SSatish Balay 162d40f771SBarry Smith /* 17ec8511deSBarry Smith MATSEQAIJ format - Compressed row storage (also called Yale sparse matrix 18d35516d3SLois Curfman McInnes format), compatible with Fortran. The i[] and j[] arrays start at 1, 19d5d45c9bSBarry Smith or 0, depending on the value of shift. For example, in Fortran 205768c4f9SLois Curfman McInnes j[i[k]+p+shift] is the pth column in row k. Note that the diagonal 215768c4f9SLois Curfman McInnes matrix elements are stored with the rest of the nonzeros (not separately). 222d40f771SBarry Smith */ 23d35516d3SLois Curfman McInnes 24b8a66259SBarry Smith typedef struct { 257c922b88SBarry Smith PetscTruth sorted; /* if true, rows are sorted by increasing columns */ 267c922b88SBarry Smith PetscTruth roworiented; /* if true, row-oriented input, default */ 273369ce9aSBarry Smith int nonew; /* 1 don't add new nonzeros, -1 generate error on new */ 287c922b88SBarry Smith PetscTruth singlemalloc; /* if true a, i, and j have been obtained with 29d35516d3SLois Curfman McInnes one big malloc */ 30*329f5518SBarry Smith PetscTruth freedata; /* free the i,j,a data when the matrix is destroyed; true by default */ 311a1c3055SLois Curfman McInnes int m,n; /* rows, columns */ 3280b4ade8SLois Curfman McInnes int nz,maxnz; /* nonzeros, allocated nonzeros */ 33d35516d3SLois Curfman McInnes int *diag; /* pointers to diagonal elements */ 34d35516d3SLois Curfman McInnes int *i; /* pointer to beginning of each row */ 35d35516d3SLois Curfman McInnes int *imax; /* maximum space allocated for each row */ 36d35516d3SLois Curfman McInnes int *ilen; /* actual length of each row */ 37d35516d3SLois Curfman McInnes int *j; /* column values: j + i[k] - 1 is start of row k */ 38d35516d3SLois Curfman McInnes Scalar *a; /* nonzero elements */ 3982bf6240SBarry Smith IS row,col,icol; /* index sets, used for reorderings */ 40f6fb444dSLois Curfman McInnes Scalar *solve_work; /* work space used in MatSolve */ 41dbb450caSBarry Smith void *spptr; /* pointer for special library like SuperLU */ 42dbb450caSBarry Smith int indexshift; /* zero or -one for C or Fortran indexing */ 4376dd722bSSatish Balay Mat_SeqAIJ_Inode inode; /* identical node informaton */ 444fc8c0a5SBarry Smith int reallocs; /* number of mallocs done during MatSetValues() 457823673eSLois Curfman McInnes as more values are set than were prealloced */ 46*329f5518SBarry Smith int rmax; /* max nonzeros in any row */ 4735aab85fSBarry Smith PetscTruth ilu_preserve_row_sums; 482e8a6d31SBarry Smith Scalar *saved_values; /* location for stashing nonzero values of matrix */ 4966d5ce97SBarry Smith Scalar *idiag,*ssor; /* inverse of diagonal entries; space for eisen */ 507c922b88SBarry Smith 517c922b88SBarry Smith PetscTruth keepzeroedrows; /* keeps matrix structure same in calls to MatZeroRows()*/ 52*329f5518SBarry Smith PetscTruth ignorezeroentries; 53ec8511deSBarry Smith } Mat_SeqAIJ; 54b8a66259SBarry Smith 555ef9f2a5SBarry Smith extern int MatILUFactorSymbolic_SeqAIJ(Mat,IS,IS,MatILUInfo*,Mat *); 5608480c60SBarry Smith extern int MatConvert_SeqAIJ(Mat,MatType,Mat *); 572e8a6d31SBarry Smith extern int MatDuplicate_SeqAIJ(Mat,MatDuplicateOption,Mat*); 587c922b88SBarry Smith extern int MatMarkDiagonal_SeqAIJ(Mat); 5908480c60SBarry Smith 6018476b09SLois Curfman McInnes extern int MatMult_SeqAIJ(Mat A,Vec,Vec); 6118476b09SLois Curfman McInnes extern int MatMultAdd_SeqAIJ(Mat A,Vec,Vec,Vec); 627c922b88SBarry Smith extern int MatMultTranspose_SeqAIJ(Mat A,Vec,Vec); 637c922b88SBarry Smith extern int MatMultTransposeAdd_SeqAIJ(Mat A,Vec,Vec,Vec); 6418476b09SLois Curfman McInnes extern int MatRelax_SeqAIJ(Mat,Vec,double,MatSORType,double,int,Vec); 6508480c60SBarry Smith 662d40f771SBarry Smith #endif 67