1 2 #if !defined(__AIJ_H) 3 #define __AIJ_H 4 #include "src/mat/matimpl.h" 5 6 /* Info about i-nodes (identical nodes) */ 7 typedef struct { 8 PetscTruth use; 9 int node_count; /* number of inodes */ 10 int *size; /* size of each inode */ 11 int limit; /* inode limit */ 12 int max_limit; /* maximum supported inode limit */ 13 PetscTruth checked; /* if inodes have been checked for */ 14 } Mat_SeqAIJ_Inode; 15 16 /* 17 MATSEQAIJ format - Compressed row storage (also called Yale sparse matrix 18 format). The i[] and j[] arrays start at 0. For example, 19 j[i[k]+p] is the pth column in row k. Note that the diagonal 20 matrix elements are stored with the rest of the nonzeros (not separately). 21 */ 22 23 typedef struct { 24 PetscTruth sorted; /* if true, rows are sorted by increasing columns */ 25 PetscTruth roworiented; /* if true, row-oriented input, default */ 26 int nonew; /* 1 don't add new nonzeros, -1 generate error on new */ 27 PetscTruth singlemalloc; /* if true a, i, and j have been obtained with 28 one big malloc */ 29 PetscTruth freedata; /* free the i,j,a data when the matrix is destroyed; true by default */ 30 int nz,maxnz; /* nonzeros, allocated nonzeros */ 31 int *diag; /* pointers to diagonal elements */ 32 int *i; /* pointer to beginning of each row */ 33 int *imax; /* maximum space allocated for each row */ 34 int *ilen; /* actual length of each row */ 35 int *j; /* column values: j + i[k] - 1 is start of row k */ 36 PetscScalar *a; /* nonzero elements */ 37 IS row,col,icol; /* index sets, used for reorderings */ 38 PetscScalar *solve_work; /* work space used in MatSolve */ 39 Mat_SeqAIJ_Inode inode; /* identical node informaton */ 40 int reallocs; /* number of mallocs done during MatSetValues() 41 as more values are set than were prealloced */ 42 int rmax; /* max nonzeros in any row */ 43 PetscTruth ilu_preserve_row_sums; 44 PetscReal lu_dtcol; 45 PetscReal lu_damping; 46 PetscReal lu_shift; /* Manteuffel shift switch, fraction */ 47 PetscReal lu_shift_fraction; 48 PetscReal lu_zeropivot; 49 PetscScalar *saved_values; /* location for stashing nonzero values of matrix */ 50 PetscScalar *idiag,*ssor; /* inverse of diagonal entries; space for eisen */ 51 52 PetscTruth keepzeroedrows; /* keeps matrix structure same in calls to MatZeroRows()*/ 53 PetscTruth ignorezeroentries; 54 ISColoring coloring; /* set with MatADSetColoring() used by MatADSetValues() */ 55 Mat sbaijMat; /* mat in sbaij format */ 56 57 int *xtoy,*xtoyB; /* map nonzero pattern of X into Y's, used by MatAXPY() */ 58 Mat XtoY; /* used by MatAXPY() */ 59 } Mat_SeqAIJ; 60 61 EXTERN int MatILUFactorSymbolic_SeqAIJ(Mat,IS,IS,MatFactorInfo*,Mat *); 62 EXTERN int MatICCFactorSymbolic_SeqAIJ(Mat,IS,MatFactorInfo*,Mat *); 63 EXTERN int MatCholeskyFactorSymbolic_SeqAIJ(Mat,IS,MatFactorInfo*,Mat*); 64 EXTERN int MatCholeskyFactorNumeric_SeqAIJ(Mat,Mat *); 65 EXTERN int MatDuplicate_SeqAIJ(Mat,MatDuplicateOption,Mat*); 66 EXTERN int MatMissingDiagonal_SeqAIJ(Mat); 67 EXTERN int MatMarkDiagonal_SeqAIJ(Mat); 68 69 EXTERN int MatMult_SeqAIJ(Mat A,Vec,Vec); 70 EXTERN int MatMultAdd_SeqAIJ(Mat A,Vec,Vec,Vec); 71 EXTERN int MatMultTranspose_SeqAIJ(Mat A,Vec,Vec); 72 EXTERN int MatMultTransposeAdd_SeqAIJ(Mat A,Vec,Vec,Vec); 73 EXTERN int MatRelax_SeqAIJ(Mat,Vec,PetscReal,MatSORType,PetscReal,int,int,Vec); 74 75 EXTERN int MatSetColoring_SeqAIJ(Mat,ISColoring); 76 EXTERN int MatSetValuesAdic_SeqAIJ(Mat,void*); 77 EXTERN int MatSetValuesAdifor_SeqAIJ(Mat,int,void*); 78 79 EXTERN int MatGetSymbolicTranspose_SeqAIJ(Mat,int *[],int *[]); 80 EXTERN int MatRestoreSymbolicTranspose_SeqAIJ(Mat,int *[],int *[]); 81 EXTERN int MatToSymmetricIJ_SeqAIJ(int,int*,int*,int,int,int**,int**); 82 EXTERN int Mat_AIJ_CheckInode(Mat,PetscTruth); 83 EXTERN int MatLUFactorSymbolic_SeqAIJ(Mat,IS,IS,MatFactorInfo*,Mat*); 84 EXTERN int MatLUFactorNumeric_SeqAIJ(Mat,Mat*); 85 EXTERN int MatLUFactor_SeqAIJ(Mat,IS,IS,MatFactorInfo*); 86 EXTERN int MatSolve_SeqAIJ(Mat,Vec,Vec); 87 EXTERN int MatSolveAdd_SeqAIJ(Mat,Vec,Vec,Vec); 88 EXTERN int MatSolveTranspose_SeqAIJ(Mat,Vec,Vec); 89 EXTERN int MatSolveTransposeAdd_SeqAIJ(Mat,Vec,Vec,Vec); 90 EXTERN int MatEqual_SeqAIJ(Mat A,Mat B,PetscTruth* flg); 91 EXTERN int MatFDColoringCreate_SeqAIJ(Mat,ISColoring,MatFDColoring); 92 EXTERN int MatILUDTFactor_SeqAIJ(Mat,MatFactorInfo*,IS,IS,Mat*); 93 EXTERN int MatLoad_SeqAIJ(PetscViewer,const MatType,Mat*); 94 EXTERN int RegisterApplyPtAPRoutines_Private(Mat); 95 EXTERN int MatMatMult_SeqAIJ_SeqAIJ(Mat,Mat,MatReuse,PetscReal,Mat*); 96 EXTERN int MatMatMultSymbolic_SeqAIJ_SeqAIJ(Mat,Mat,PetscReal,Mat*); 97 EXTERN int MatMatMultNumeric_SeqAIJ_SeqAIJ(Mat,Mat,Mat); 98 EXTERN int MatPtAP_SeqAIJ_SeqAIJ(Mat,Mat,MatReuse,PetscReal,Mat*); 99 EXTERN int MatPtAPSymbolic_SeqAIJ_SeqAIJ(Mat,Mat,PetscReal,Mat*); 100 EXTERN int MatPtAPNumeric_SeqAIJ_SeqAIJ(Mat,Mat,Mat); 101 102 EXTERN_C_BEGIN 103 EXTERN int MatConvert_SeqAIJ_SeqSBAIJ(Mat,const MatType,Mat*); 104 EXTERN int MatConvert_SeqAIJ_SeqBAIJ(Mat,const MatType,Mat*); 105 EXTERN int MatReorderForNonzeroDiagonal_SeqAIJ(Mat,PetscReal,IS,IS); 106 EXTERN int MatAdjustForInodes_SeqAIJ(Mat,IS*,IS*); 107 EXTERN int MatSeqAIJGetInodeSizes_SeqAIJ(Mat,int*,int*[],int*); 108 EXTERN_C_END 109 110 #endif 111