1 2 #if !defined(__SBAIJ_H) 3 #define __SBAIJ_H 4 #include <petsc-private/matimpl.h> 5 #include <../src/mat/impls/baij/seq/baij.h> 6 7 /* 8 MATSEQSBAIJ format - Block compressed row storage. The i[] and j[] 9 arrays start at 0. 10 */ 11 12 typedef struct { 13 SEQAIJHEADER(MatScalar); 14 SEQBAIJHEADER; 15 PetscInt *inew; /* pointer to beginning of each row of reordered matrix */ 16 PetscInt *jnew; /* column values: jnew + i[k] is start of row k */ 17 MatScalar *anew; /* nonzero diagonal and superdiagonal elements of reordered matrix */ 18 PetscScalar *solves_work; /* work space used in MatSolves */ 19 PetscInt solves_work_n;/* size of solves_work */ 20 PetscInt *a2anew; /* map used for symm permutation */ 21 PetscBool permute; /* if true, a non-trivial permutation is used for factorization */ 22 PetscBool ignore_ltriangular; /* if true, ignore the lower triangular values inserted by users */ 23 PetscBool getrow_utriangular; /* if true, MatGetRow_SeqSBAIJ() is enabled to get the upper part of the row */ 24 Mat_SeqAIJ_Inode inode; 25 unsigned short *jshort; 26 PetscBool free_jshort; 27 } Mat_SeqSBAIJ; 28 29 EXTERN_C_BEGIN 30 extern PetscErrorCode MatSeqSBAIJSetPreallocation_SeqSBAIJ(Mat,PetscInt,PetscInt,PetscInt*); 31 EXTERN_C_END 32 extern PetscErrorCode MatCholeskyFactorSymbolic_SeqSBAIJ(Mat,Mat,IS,const MatFactorInfo*); 33 extern PetscErrorCode MatCholeskyFactorSymbolic_SeqSBAIJ_inplace(Mat,Mat,IS,const MatFactorInfo*); 34 extern PetscErrorCode MatCholeskyFactor_SeqSBAIJ(Mat,IS,const MatFactorInfo*); 35 extern PetscErrorCode MatICCFactorSymbolic_SeqSBAIJ(Mat,Mat,IS,const MatFactorInfo*); 36 extern PetscErrorCode MatICCFactorSymbolic_SeqSBAIJ_inplace(Mat,Mat,IS,const MatFactorInfo*); 37 extern PetscErrorCode MatDuplicate_SeqSBAIJ(Mat,MatDuplicateOption,Mat*); 38 extern PetscErrorCode MatMarkDiagonal_SeqSBAIJ(Mat); 39 extern PetscErrorCode MatIncreaseOverlap_SeqSBAIJ(Mat,PetscInt,IS[],PetscInt); 40 extern PetscErrorCode MatGetSubMatrix_SeqSBAIJ(Mat,IS,IS,MatReuse,Mat*); 41 extern PetscErrorCode MatGetSubMatrices_SeqSBAIJ(Mat,PetscInt,const IS[],const IS[],MatReuse,Mat*[]); 42 extern PetscErrorCode MatScale_SeqSBAIJ(Mat,PetscScalar); 43 extern PetscErrorCode MatNorm_SeqSBAIJ(Mat,NormType,PetscReal *); 44 extern PetscErrorCode MatEqual_SeqSBAIJ(Mat,Mat,PetscBool *); 45 extern PetscErrorCode MatGetDiagonal_SeqSBAIJ(Mat,Vec); 46 extern PetscErrorCode MatDiagonalScale_SeqSBAIJ(Mat,Vec,Vec); 47 extern PetscErrorCode MatGetInfo_SeqSBAIJ(Mat,MatInfoType,MatInfo *); 48 extern PetscErrorCode MatZeroEntries_SeqSBAIJ(Mat); 49 extern PetscErrorCode MatGetRowMaxAbs_SeqSBAIJ(Mat,Vec,PetscInt[]); 50 extern PetscErrorCode MatGetInertia_SeqSBAIJ(Mat,PetscInt*,PetscInt*,PetscInt*); 51 extern PetscErrorCode MatDestroy_SeqSBAIJ(Mat); 52 extern PetscErrorCode MatView_SeqSBAIJ(Mat,PetscViewer); 53 54 extern PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering(Mat,Mat,const MatFactorInfo*); 55 extern PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering_inplace(Mat,Mat,const MatFactorInfo*); 56 extern PetscErrorCode MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace(Mat,Vec,Vec); 57 extern PetscErrorCode MatSolve_SeqSBAIJ_1_NaturalOrdering(Mat,Vec,Vec); 58 59 extern PetscErrorCode MatForwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace(Mat,Vec,Vec); 60 extern PetscErrorCode MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace(Mat,Vec,Vec); 61 extern PetscErrorCode MatForwardSolve_SeqSBAIJ_1_inplace(Mat,Vec,Vec); 62 extern PetscErrorCode MatBackwardSolve_SeqSBAIJ_1_inplace(Mat,Vec,Vec); 63 64 extern PetscErrorCode MatForwardSolve_SeqSBAIJ_1_NaturalOrdering(Mat,Vec,Vec); 65 extern PetscErrorCode MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering(Mat,Vec,Vec); 66 extern PetscErrorCode MatForwardSolve_SeqSBAIJ_1(Mat,Vec,Vec); 67 extern PetscErrorCode MatBackwardSolve_SeqSBAIJ_1(Mat,Vec,Vec); 68 69 extern PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_2_NaturalOrdering(Mat,Mat,const MatFactorInfo*); 70 extern PetscErrorCode MatSolve_SeqSBAIJ_2_NaturalOrdering_inplace(Mat,Vec,Vec); 71 extern PetscErrorCode MatForwardSolve_SeqSBAIJ_2_NaturalOrdering_inplace(Mat,Vec,Vec); 72 extern PetscErrorCode MatBackwardSolve_SeqSBAIJ_2_NaturalOrdering_inplace(Mat,Vec,Vec); 73 74 extern PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering(Mat,Mat,const MatFactorInfo*); 75 extern PetscErrorCode MatSolve_SeqSBAIJ_3_NaturalOrdering_inplace(Mat,Vec,Vec); 76 extern PetscErrorCode MatForwardSolve_SeqSBAIJ_3_NaturalOrdering_inplace(Mat,Vec,Vec); 77 extern PetscErrorCode MatBackwardSolve_SeqSBAIJ_3_NaturalOrdering_inplace(Mat,Vec,Vec); 78 79 extern PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering(Mat,Mat,const MatFactorInfo*); 80 extern PetscErrorCode MatSolve_SeqSBAIJ_4_NaturalOrdering_inplace(Mat,Vec,Vec); 81 extern PetscErrorCode MatForwardSolve_SeqSBAIJ_4_NaturalOrdering_inplace(Mat,Vec,Vec); 82 extern PetscErrorCode MatBackwardSolve_SeqSBAIJ_4_NaturalOrdering_inplace(Mat,Vec,Vec); 83 84 extern PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering(Mat,Mat,const MatFactorInfo*); 85 extern PetscErrorCode MatSolve_SeqSBAIJ_5_NaturalOrdering_inplace(Mat,Vec,Vec); 86 extern PetscErrorCode MatForwardSolve_SeqSBAIJ_5_NaturalOrdering_inplace(Mat,Vec,Vec); 87 extern PetscErrorCode MatBackwardSolve_SeqSBAIJ_5_NaturalOrdering_inplace(Mat,Vec,Vec); 88 89 extern PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering(Mat,Mat,const MatFactorInfo*); 90 extern PetscErrorCode MatSolve_SeqSBAIJ_6_NaturalOrdering_inplace(Mat,Vec,Vec); 91 extern PetscErrorCode MatForwardSolve_SeqSBAIJ_6_NaturalOrdering_inplace(Mat,Vec,Vec); 92 extern PetscErrorCode MatBackwardSolve_SeqSBAIJ_6_NaturalOrdering_inplace(Mat,Vec,Vec); 93 94 extern PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering(Mat,Mat,const MatFactorInfo*); 95 extern PetscErrorCode MatSolve_SeqSBAIJ_7_NaturalOrdering_inplace(Mat,Vec,Vec); 96 extern PetscErrorCode MatForwardSolve_SeqSBAIJ_7_NaturalOrdering_inplace(Mat,Vec,Vec); 97 extern PetscErrorCode MatBackwardSolve_SeqSBAIJ_7_NaturalOrdering_inplace(Mat,Vec,Vec); 98 99 extern PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering(Mat,Mat,const MatFactorInfo*); 100 extern PetscErrorCode MatSolve_SeqSBAIJ_N_NaturalOrdering_inplace(Mat,Vec,Vec); 101 extern PetscErrorCode MatForwardSolve_SeqSBAIJ_N_NaturalOrdering_inplace(Mat,Vec,Vec); 102 extern PetscErrorCode MatBackwardSolve_SeqSBAIJ_N_NaturalOrdering_inplace(Mat,Vec,Vec); 103 extern PetscErrorCode MatForwardSolve_SeqSBAIJ_N_inplace(Mat,Vec,Vec); 104 extern PetscErrorCode MatBackwardSolve_SeqSBAIJ_N_inplace(Mat,Vec,Vec); 105 106 extern PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_N(Mat,Mat,const MatFactorInfo*); 107 extern PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_1_inplace(Mat,Mat,const MatFactorInfo*); 108 extern PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_2(Mat,Mat,const MatFactorInfo*); 109 extern PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_3(Mat,Mat,const MatFactorInfo*); 110 extern PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_4(Mat,Mat,const MatFactorInfo*); 111 extern PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_5(Mat,Mat,const MatFactorInfo*); 112 extern PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_6(Mat,Mat,const MatFactorInfo*); 113 extern PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_7(Mat,Mat,const MatFactorInfo*); 114 115 extern PetscErrorCode MatSolve_SeqSBAIJ_N_inplace(Mat,Vec,Vec); 116 extern PetscErrorCode MatSolve_SeqSBAIJ_1_inplace(Mat,Vec,Vec); 117 extern PetscErrorCode MatSolve_SeqSBAIJ_1(Mat,Vec,Vec); 118 extern PetscErrorCode MatSolve_SeqSBAIJ_2_inplace(Mat,Vec,Vec); 119 extern PetscErrorCode MatSolve_SeqSBAIJ_3_inplace(Mat,Vec,Vec); 120 extern PetscErrorCode MatSolve_SeqSBAIJ_4_inplace(Mat,Vec,Vec); 121 extern PetscErrorCode MatSolve_SeqSBAIJ_5_inplace(Mat,Vec,Vec); 122 extern PetscErrorCode MatSolve_SeqSBAIJ_6_inplace(Mat,Vec,Vec); 123 extern PetscErrorCode MatSolve_SeqSBAIJ_7_inplace(Mat,Vec,Vec); 124 125 extern PetscErrorCode MatSolves_SeqSBAIJ_1_inplace(Mat,Vecs,Vecs); 126 extern PetscErrorCode MatSolves_SeqSBAIJ_1(Mat,Vecs,Vecs); 127 128 extern PetscErrorCode MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace(Mat,Vec,Vec); 129 extern PetscErrorCode MatSolve_SeqSBAIJ_1_NaturalOrdering(Mat,Vec,Vec); 130 extern PetscErrorCode MatSolve_SeqSBAIJ_2_NaturalOrdering_inplace(Mat,Vec,Vec); 131 extern PetscErrorCode MatSolve_SeqSBAIJ_3_NaturalOrdering_inplace(Mat,Vec,Vec); 132 extern PetscErrorCode MatSolve_SeqSBAIJ_4_NaturalOrdering_inplace(Mat,Vec,Vec); 133 extern PetscErrorCode MatSolve_SeqSBAIJ_5_NaturalOrdering_inplace(Mat,Vec,Vec); 134 extern PetscErrorCode MatSolve_SeqSBAIJ_6_NaturalOrdering_inplace(Mat,Vec,Vec); 135 extern PetscErrorCode MatSolve_SeqSBAIJ_7_NaturalOrdering_inplace(Mat,Vec,Vec); 136 extern PetscErrorCode MatSolve_SeqSBAIJ_N_NaturalOrdering_inplace(Mat,Vec,Vec); 137 138 extern PetscErrorCode MatMult_SeqSBAIJ_1(Mat,Vec,Vec); 139 extern PetscErrorCode MatMult_SeqSBAIJ_2(Mat,Vec,Vec); 140 extern PetscErrorCode MatMult_SeqSBAIJ_3(Mat,Vec,Vec); 141 extern PetscErrorCode MatMult_SeqSBAIJ_4(Mat,Vec,Vec); 142 extern PetscErrorCode MatMult_SeqSBAIJ_5(Mat,Vec,Vec); 143 extern PetscErrorCode MatMult_SeqSBAIJ_6(Mat,Vec,Vec); 144 extern PetscErrorCode MatMult_SeqSBAIJ_7(Mat,Vec,Vec); 145 extern PetscErrorCode MatMult_SeqSBAIJ_N(Mat,Vec,Vec); 146 147 extern PetscErrorCode MatMult_SeqSBAIJ_1_Hermitian(Mat,Vec,Vec); 148 149 extern PetscErrorCode MatMultAdd_SeqSBAIJ_1(Mat,Vec,Vec,Vec); 150 extern PetscErrorCode MatMultAdd_SeqSBAIJ_2(Mat,Vec,Vec,Vec); 151 extern PetscErrorCode MatMultAdd_SeqSBAIJ_3(Mat,Vec,Vec,Vec); 152 extern PetscErrorCode MatMultAdd_SeqSBAIJ_4(Mat,Vec,Vec,Vec); 153 extern PetscErrorCode MatMultAdd_SeqSBAIJ_5(Mat,Vec,Vec,Vec); 154 extern PetscErrorCode MatMultAdd_SeqSBAIJ_6(Mat,Vec,Vec,Vec); 155 extern PetscErrorCode MatMultAdd_SeqSBAIJ_7(Mat,Vec,Vec,Vec); 156 extern PetscErrorCode MatMultAdd_SeqSBAIJ_N(Mat,Vec,Vec,Vec); 157 158 extern PetscErrorCode MatSOR_SeqSBAIJ(Mat,Vec,PetscReal,MatSORType,PetscReal,PetscInt,PetscInt,Vec); 159 extern PetscErrorCode MatLoad_SeqSBAIJ(Mat,PetscViewer); 160 extern PetscErrorCode MatSeqSBAIJSetNumericFactorization_inplace(Mat,PetscBool); 161 162 #endif 163