1b8a66259SBarry Smith 22d40f771SBarry Smith #if !defined(__AIJ_H) 32d40f771SBarry Smith #define __AIJ_H 4da33ede1SBarry Smith #include "src/mat/matimpl.h" 52d40f771SBarry Smith 6ed2daf61SLois Curfman McInnes /* Info about i-nodes (identical nodes) */ 7957e4c30SSatish Balay typedef struct { 84c49b128SBarry Smith PetscTruth use; 997f1f81fSBarry Smith PetscInt node_count; /* number of inodes */ 1097f1f81fSBarry Smith PetscInt *size; /* size of each inode */ 1197f1f81fSBarry Smith PetscInt limit; /* inode limit */ 1297f1f81fSBarry Smith PetscInt max_limit; /* maximum supported inode limit */ 133a7fca6bSBarry Smith PetscTruth checked; /* if inodes have been checked for */ 1476dd722bSSatish Balay } Mat_SeqAIJ_Inode; 15957e4c30SSatish Balay 162d40f771SBarry Smith /* 17ec8511deSBarry Smith MATSEQAIJ format - Compressed row storage (also called Yale sparse matrix 18dfbc5765Svictorle format). The i[] and j[] arrays start at 0. For example, 19dfbc5765Svictorle j[i[k]+p] is the pth column in row k. Note that the diagonal 205768c4f9SLois Curfman McInnes matrix elements are stored with the rest of the nonzeros (not separately). 212d40f771SBarry Smith */ 22d35516d3SLois Curfman McInnes 23b8a66259SBarry Smith typedef struct { 247c922b88SBarry Smith PetscTruth sorted; /* if true, rows are sorted by increasing columns */ 257c922b88SBarry Smith PetscTruth roworiented; /* if true, row-oriented input, default */ 26*38baddfdSBarry Smith PetscInt nonew; /* 1 don't add new nonzeros, -1 generate error on new */ 277c922b88SBarry Smith PetscTruth singlemalloc; /* if true a, i, and j have been obtained with 28d35516d3SLois Curfman McInnes one big malloc */ 29329f5518SBarry Smith PetscTruth freedata; /* free the i,j,a data when the matrix is destroyed; true by default */ 3097f1f81fSBarry Smith PetscInt nz,maxnz; /* nonzeros, allocated nonzeros */ 3197f1f81fSBarry Smith PetscInt *diag; /* pointers to diagonal elements */ 3297f1f81fSBarry Smith PetscInt *i; /* pointer to beginning of each row */ 3397f1f81fSBarry Smith PetscInt *imax; /* maximum space allocated for each row */ 34*38baddfdSBarry Smith PetscInt *ilen; /* actual length of each row */ 3597f1f81fSBarry Smith PetscInt *j; /* column values: j + i[k] - 1 is start of row k */ 36ea709b57SSatish Balay PetscScalar *a; /* nonzero elements */ 3782bf6240SBarry Smith IS row,col,icol; /* index sets, used for reorderings */ 38ea709b57SSatish Balay PetscScalar *solve_work; /* work space used in MatSolve */ 3976dd722bSSatish Balay Mat_SeqAIJ_Inode inode; /* identical node informaton */ 4097f1f81fSBarry Smith PetscInt reallocs; /* number of mallocs done during MatSetValues() 417823673eSLois Curfman McInnes as more values are set than were prealloced */ 4297f1f81fSBarry Smith PetscInt rmax; /* max nonzeros in any row */ 4335aab85fSBarry Smith PetscTruth ilu_preserve_row_sums; 44b9b97703SBarry Smith PetscReal lu_dtcol; 45b9b97703SBarry Smith PetscReal lu_damping; 46f4cce38bSHong Zhang PetscReal lu_shift; /* Manteuffel shift switch, fraction */ 476cc28720Svictorle PetscReal lu_shift_fraction; 4887828ca2SBarry Smith PetscReal lu_zeropivot; 49ea709b57SSatish Balay PetscScalar *saved_values; /* location for stashing nonzero values of matrix */ 50ea709b57SSatish Balay PetscScalar *idiag,*ssor; /* inverse of diagonal entries; space for eisen */ 517c922b88SBarry Smith 527c922b88SBarry Smith PetscTruth keepzeroedrows; /* keeps matrix structure same in calls to MatZeroRows()*/ 53329f5518SBarry Smith PetscTruth ignorezeroentries; 543a7fca6bSBarry Smith ISColoring coloring; /* set with MatADSetColoring() used by MatADSetValues() */ 550968510aSHong Zhang Mat sbaijMat; /* mat in sbaij format */ 56a30b2313SHong Zhang 5797f1f81fSBarry Smith PetscInt *xtoy,*xtoyB; /* map nonzero pattern of X into Y's, used by MatAXPY() */ 58a30b2313SHong Zhang Mat XtoY; /* used by MatAXPY() */ 59ec8511deSBarry Smith } Mat_SeqAIJ; 60b8a66259SBarry Smith 61dfbe8321SBarry Smith EXTERN PetscErrorCode MatILUFactorSymbolic_SeqAIJ(Mat,IS,IS,MatFactorInfo*,Mat *); 62dfbe8321SBarry Smith EXTERN PetscErrorCode MatICCFactorSymbolic_SeqAIJ(Mat,IS,MatFactorInfo*,Mat *); 63dfbe8321SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ(Mat,IS,MatFactorInfo*,Mat*); 64dfbe8321SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ(Mat,Mat *); 65dfbe8321SBarry Smith EXTERN PetscErrorCode MatDuplicate_SeqAIJ(Mat,MatDuplicateOption,Mat*); 66dfbe8321SBarry Smith EXTERN PetscErrorCode MatMissingDiagonal_SeqAIJ(Mat); 67dfbe8321SBarry Smith EXTERN PetscErrorCode MatMarkDiagonal_SeqAIJ(Mat); 6808480c60SBarry Smith 69dfbe8321SBarry Smith EXTERN PetscErrorCode MatMult_SeqAIJ(Mat A,Vec,Vec); 70dfbe8321SBarry Smith EXTERN PetscErrorCode MatMultAdd_SeqAIJ(Mat A,Vec,Vec,Vec); 71dfbe8321SBarry Smith EXTERN PetscErrorCode MatMultTranspose_SeqAIJ(Mat A,Vec,Vec); 72dfbe8321SBarry Smith EXTERN PetscErrorCode MatMultTransposeAdd_SeqAIJ(Mat A,Vec,Vec,Vec); 73*38baddfdSBarry Smith EXTERN PetscErrorCode MatRelax_SeqAIJ(Mat,Vec,PetscReal,MatSORType,PetscReal,PetscInt,PetscInt,Vec); 7408480c60SBarry Smith 75dfbe8321SBarry Smith EXTERN PetscErrorCode MatSetColoring_SeqAIJ(Mat,ISColoring); 76dfbe8321SBarry Smith EXTERN PetscErrorCode MatSetValuesAdic_SeqAIJ(Mat,void*); 77*38baddfdSBarry Smith EXTERN PetscErrorCode MatSetValuesAdifor_SeqAIJ(Mat,PetscInt,void*); 783a7fca6bSBarry Smith 79*38baddfdSBarry Smith EXTERN PetscErrorCode MatGetSymbolicTranspose_SeqAIJ(Mat,PetscInt *[],PetscInt *[]); 80*38baddfdSBarry Smith EXTERN PetscErrorCode MatRestoreSymbolicTranspose_SeqAIJ(Mat,PetscInt *[],PetscInt *[]); 81*38baddfdSBarry Smith EXTERN PetscErrorCode MatToSymmetricIJ_SeqAIJ(PetscInt,PetscInt*,PetscInt*,PetscInt,PetscInt,PetscInt**,PetscInt**); 82dfbe8321SBarry Smith EXTERN PetscErrorCode Mat_AIJ_CheckInode(Mat,PetscTruth); 83dfbe8321SBarry Smith EXTERN PetscErrorCode MatLUFactorSymbolic_SeqAIJ(Mat,IS,IS,MatFactorInfo*,Mat*); 84dfbe8321SBarry Smith EXTERN PetscErrorCode MatLUFactorNumeric_SeqAIJ(Mat,Mat*); 85dfbe8321SBarry Smith EXTERN PetscErrorCode MatLUFactor_SeqAIJ(Mat,IS,IS,MatFactorInfo*); 86dfbe8321SBarry Smith EXTERN PetscErrorCode MatSolve_SeqAIJ(Mat,Vec,Vec); 87dfbe8321SBarry Smith EXTERN PetscErrorCode MatSolveAdd_SeqAIJ(Mat,Vec,Vec,Vec); 88dfbe8321SBarry Smith EXTERN PetscErrorCode MatSolveTranspose_SeqAIJ(Mat,Vec,Vec); 89dfbe8321SBarry Smith EXTERN PetscErrorCode MatSolveTransposeAdd_SeqAIJ(Mat,Vec,Vec,Vec); 90dfbe8321SBarry Smith EXTERN PetscErrorCode MatEqual_SeqAIJ(Mat A,Mat B,PetscTruth* flg); 91dfbe8321SBarry Smith EXTERN PetscErrorCode MatFDColoringCreate_SeqAIJ(Mat,ISColoring,MatFDColoring); 92dfbe8321SBarry Smith EXTERN PetscErrorCode MatILUDTFactor_SeqAIJ(Mat,MatFactorInfo*,IS,IS,Mat*); 93dfbe8321SBarry Smith EXTERN PetscErrorCode MatLoad_SeqAIJ(PetscViewer,const MatType,Mat*); 94dfbe8321SBarry Smith EXTERN PetscErrorCode RegisterApplyPtAPRoutines_Private(Mat); 95dfbe8321SBarry Smith EXTERN PetscErrorCode MatMatMult_SeqAIJ_SeqAIJ(Mat,Mat,MatReuse,PetscReal,Mat*); 96dfbe8321SBarry Smith EXTERN PetscErrorCode MatMatMultSymbolic_SeqAIJ_SeqAIJ(Mat,Mat,PetscReal,Mat*); 97dfbe8321SBarry Smith EXTERN PetscErrorCode MatMatMultNumeric_SeqAIJ_SeqAIJ(Mat,Mat,Mat); 98dfbe8321SBarry Smith EXTERN PetscErrorCode MatPtAP_SeqAIJ_SeqAIJ(Mat,Mat,MatReuse,PetscReal,Mat*); 99dfbe8321SBarry Smith EXTERN PetscErrorCode MatPtAPSymbolic_SeqAIJ_SeqAIJ(Mat,Mat,PetscReal,Mat*); 100dfbe8321SBarry Smith EXTERN PetscErrorCode MatPtAPNumeric_SeqAIJ_SeqAIJ(Mat,Mat,Mat); 101bc011b1eSHong Zhang EXTERN PetscErrorCode MatMatMultTranspose_SeqAIJ_SeqAIJ(Mat,Mat,MatReuse,PetscReal,Mat*); 102bc011b1eSHong Zhang EXTERN PetscErrorCode MatMatMultTransposeSymbolic_SeqAIJ_SeqAIJ(Mat,Mat,PetscReal,Mat*); 103bc011b1eSHong Zhang EXTERN PetscErrorCode MatMatMultTransposeNumeric_SeqAIJ_SeqAIJ(Mat,Mat,Mat); 104*38baddfdSBarry Smith EXTERN PetscErrorCode MatSetValues_SeqAIJ(Mat,PetscInt,const PetscInt[],PetscInt,const PetscInt[],const PetscScalar[],InsertMode); 105*38baddfdSBarry Smith EXTERN PetscErrorCode MatGetRow_SeqAIJ(Mat,PetscInt,PetscInt*,PetscInt**,PetscScalar**); 106*38baddfdSBarry Smith EXTERN PetscErrorCode MatRestoreRow_SeqAIJ(Mat,PetscInt,PetscInt*,PetscInt**,PetscScalar**); 10711d72411SSatish Balay EXTERN PetscErrorCode MatPrintHelp_SeqAIJ(Mat); 10811d72411SSatish Balay EXTERN PetscErrorCode MatAXPY_SeqAIJ(const PetscScalar[],Mat,Mat,MatStructure); 1099af31e4aSHong Zhang 11097304618SKris Buschelman EXTERN_C_BEGIN 111dfbe8321SBarry Smith EXTERN PetscErrorCode MatConvert_SeqAIJ_SeqSBAIJ(Mat,const MatType,Mat*); 112dfbe8321SBarry Smith EXTERN PetscErrorCode MatConvert_SeqAIJ_SeqBAIJ(Mat,const MatType,Mat*); 113dfbe8321SBarry Smith EXTERN PetscErrorCode MatReorderForNonzeroDiagonal_SeqAIJ(Mat,PetscReal,IS,IS); 114dfbe8321SBarry Smith EXTERN PetscErrorCode MatAdjustForInodes_SeqAIJ(Mat,IS*,IS*); 115*38baddfdSBarry Smith EXTERN PetscErrorCode MatSeqAIJGetInodeSizes_SeqAIJ(Mat,PetscInt*,PetscInt*[],PetscInt*); 11697304618SKris Buschelman EXTERN_C_END 11770f19b1fSKris Buschelman 1182d40f771SBarry Smith #endif 119