xref: /petsc/src/mat/impls/sbaij/seq/sbaij.h (revision 5f44c854a1643e5474f3e58a6ba8329779ac3923) 
149b5e25fSSatish Balay 
249b5e25fSSatish Balay #if !defined(__SBAIJ_H)
349b5e25fSSatish Balay #define __SBAIJ_H
47c4f633dSBarry Smith #include "private/matimpl.h"
57c4f633dSBarry Smith #include "../src/mat/impls/baij/seq/baij.h"
649b5e25fSSatish Balay 
749b5e25fSSatish Balay /*
849b5e25fSSatish Balay   MATSEQSBAIJ format - Block compressed row storage. The i[] and j[]
949b5e25fSSatish Balay   arrays start at 0.
1049b5e25fSSatish Balay */
1149b5e25fSSatish Balay 
1249b5e25fSSatish Balay typedef struct {
13dd6ea824SBarry Smith   SEQAIJHEADER(MatScalar);
14e6b907acSBarry Smith   SEQBAIJHEADER;
1513f74950SBarry Smith   PetscInt         *inew;        /* pointer to beginning of each row of reordered matrix */
1613f74950SBarry Smith   PetscInt         *jnew;        /* column values: jnew + i[k] is start of row k */
17a6175056SHong Zhang   MatScalar        *anew;        /* nonzero diagonal and superdiagonal elements of reordered matrix */
18d59c15a7SBarry Smith   PetscScalar      *solves_work; /* work space used in MatSolves */
1913f74950SBarry Smith   PetscInt         solves_work_n;/* size of solves_work */
20e2ee2a47SBarry Smith   PetscScalar      *relax_work;
2113f74950SBarry Smith   PetscInt         *a2anew;        /* map used for symm permutation */
22d36791b2SHong Zhang   PetscTruth       permute;        /* if true, a non-trivial permutation is used for factorization */
23941593c8SHong Zhang   PetscTruth       ignore_ltriangular; /* if true, ignore the lower triangular values inserted by users */
24f5edf698SHong Zhang   PetscTruth       getrow_utriangular; /* if true, MatGetRow_SeqSBAIJ() is enabled to get the upper part of the row */
250210e607SBarry Smith   Mat_Inode        inode;
2638702af4SBarry Smith   unsigned short   *jshort;
2749b5e25fSSatish Balay } Mat_SeqSBAIJ;
2849b5e25fSSatish Balay 
29f349cc81SSatish Balay EXTERN_C_BEGIN
30ab93d7beSBarry Smith EXTERN PetscErrorCode MatSeqSBAIJSetPreallocation_SeqSBAIJ(Mat,PetscInt,PetscInt,PetscInt*);
31f349cc81SSatish Balay EXTERN_C_END
320481f469SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorSymbolic_SeqSBAIJ(Mat,Mat,IS,const MatFactorInfo*);
330481f469SBarry Smith EXTERN PetscErrorCode MatCholeskyFactor_SeqSBAIJ(Mat,IS,const MatFactorInfo*);
340481f469SBarry Smith EXTERN PetscErrorCode MatICCFactorSymbolic_SeqSBAIJ(Mat,Mat,IS,const MatFactorInfo*);
35dfbe8321SBarry Smith EXTERN PetscErrorCode MatDuplicate_SeqSBAIJ(Mat,MatDuplicateOption,Mat*);
36dfbe8321SBarry Smith EXTERN PetscErrorCode MatMarkDiagonal_SeqSBAIJ(Mat);
379495ac64SHong Zhang EXTERN PetscErrorCode MatIncreaseOverlap_SeqSBAIJ(Mat,PetscInt,IS[],PetscInt);
384aa3045dSJed Brown EXTERN PetscErrorCode MatGetSubMatrix_SeqSBAIJ(Mat,IS,IS,MatReuse,Mat*);
399495ac64SHong Zhang EXTERN PetscErrorCode MatGetSubMatrices_SeqSBAIJ(Mat,PetscInt,const IS[],const IS[],MatReuse,Mat*[]);
409495ac64SHong Zhang EXTERN PetscErrorCode MatScale_SeqSBAIJ(Mat,PetscScalar);
419495ac64SHong Zhang EXTERN PetscErrorCode MatNorm_SeqSBAIJ(Mat,NormType,PetscReal *);
429495ac64SHong Zhang EXTERN PetscErrorCode MatEqual_SeqSBAIJ(Mat,Mat,PetscTruth*);
439495ac64SHong Zhang EXTERN PetscErrorCode MatGetDiagonal_SeqSBAIJ(Mat,Vec);
449495ac64SHong Zhang EXTERN PetscErrorCode MatDiagonalScale_SeqSBAIJ(Mat,Vec,Vec);
459495ac64SHong Zhang EXTERN PetscErrorCode MatGetInfo_SeqSBAIJ(Mat,MatInfoType,MatInfo *);
469495ac64SHong Zhang EXTERN PetscErrorCode MatZeroEntries_SeqSBAIJ(Mat);
47985db425SBarry Smith EXTERN PetscErrorCode MatGetRowMaxAbs_SeqSBAIJ(Mat,Vec,PetscInt[]);
489495ac64SHong Zhang EXTERN PetscErrorCode MatGetInertia_SeqSBAIJ(Mat,PetscInt*,PetscInt*,PetscInt*);
4949b5e25fSSatish Balay 
500481f469SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering(Mat,Mat,const MatFactorInfo*);
51dfbe8321SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_1_NaturalOrdering(Mat,Vec,Vec);
52*5f44c854SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_1_NaturalOrdering_newdatastruct(Mat,Vec,Vec);
534078b935SHong Zhang EXTERN PetscErrorCode MatForwardSolve_SeqSBAIJ_1_NaturalOrdering(Mat,Vec,Vec);
544078b935SHong Zhang EXTERN PetscErrorCode MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering(Mat,Vec,Vec);
559495ac64SHong Zhang EXTERN PetscErrorCode MatForwardSolve_SeqSBAIJ_1(Mat,Vec,Vec);
569495ac64SHong Zhang EXTERN PetscErrorCode MatBackwardSolve_SeqSBAIJ_1(Mat,Vec,Vec);
57266135f8SHong Zhang 
580481f469SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_2_NaturalOrdering(Mat,Mat,const MatFactorInfo*);
59dfbe8321SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_2_NaturalOrdering(Mat,Vec,Vec);
60759322afSHong Zhang EXTERN PetscErrorCode MatForwardSolve_SeqSBAIJ_2_NaturalOrdering(Mat,Vec,Vec);
61759322afSHong Zhang EXTERN PetscErrorCode MatBackwardSolve_SeqSBAIJ_2_NaturalOrdering(Mat,Vec,Vec);
62266135f8SHong Zhang 
630481f469SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering(Mat,Mat,const MatFactorInfo*);
64dfbe8321SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_3_NaturalOrdering(Mat,Vec,Vec);
65759322afSHong Zhang EXTERN PetscErrorCode MatForwardSolve_SeqSBAIJ_3_NaturalOrdering(Mat,Vec,Vec);
66759322afSHong Zhang EXTERN PetscErrorCode MatBackwardSolve_SeqSBAIJ_3_NaturalOrdering(Mat,Vec,Vec);
67266135f8SHong Zhang 
680481f469SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering(Mat,Mat,const MatFactorInfo*);
69dfbe8321SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_4_NaturalOrdering(Mat,Vec,Vec);
70759322afSHong Zhang EXTERN PetscErrorCode MatForwardSolve_SeqSBAIJ_4_NaturalOrdering(Mat,Vec,Vec);
71759322afSHong Zhang EXTERN PetscErrorCode MatBackwardSolve_SeqSBAIJ_4_NaturalOrdering(Mat,Vec,Vec);
72266135f8SHong Zhang 
730481f469SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering(Mat,Mat,const MatFactorInfo*);
74dfbe8321SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_5_NaturalOrdering(Mat,Vec,Vec);
75759322afSHong Zhang EXTERN PetscErrorCode MatForwardSolve_SeqSBAIJ_5_NaturalOrdering(Mat,Vec,Vec);
76759322afSHong Zhang EXTERN PetscErrorCode MatBackwardSolve_SeqSBAIJ_5_NaturalOrdering(Mat,Vec,Vec);
77266135f8SHong Zhang 
780481f469SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering(Mat,Mat,const MatFactorInfo*);
79dfbe8321SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_6_NaturalOrdering(Mat,Vec,Vec);
80759322afSHong Zhang EXTERN PetscErrorCode MatForwardSolve_SeqSBAIJ_6_NaturalOrdering(Mat,Vec,Vec);
81759322afSHong Zhang EXTERN PetscErrorCode MatBackwardSolve_SeqSBAIJ_6_NaturalOrdering(Mat,Vec,Vec);
82266135f8SHong Zhang 
830481f469SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering(Mat,Mat,const MatFactorInfo*);
84dfbe8321SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_7_NaturalOrdering(Mat,Vec,Vec);
85759322afSHong Zhang EXTERN PetscErrorCode MatForwardSolve_SeqSBAIJ_7_NaturalOrdering(Mat,Vec,Vec);
86759322afSHong Zhang EXTERN PetscErrorCode MatBackwardSolve_SeqSBAIJ_7_NaturalOrdering(Mat,Vec,Vec);
8749b5e25fSSatish Balay 
880481f469SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering(Mat,Mat,const MatFactorInfo*);
89dfbe8321SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_N_NaturalOrdering(Mat,Vec,Vec);
90759322afSHong Zhang EXTERN PetscErrorCode MatForwardSolve_SeqSBAIJ_N_NaturalOrdering(Mat,Vec,Vec);
91759322afSHong Zhang EXTERN PetscErrorCode MatBackwardSolve_SeqSBAIJ_N_NaturalOrdering(Mat,Vec,Vec);
929495ac64SHong Zhang EXTERN PetscErrorCode MatForwardSolve_SeqSBAIJ_N(Mat,Vec,Vec);
939495ac64SHong Zhang EXTERN PetscErrorCode MatBackwardSolve_SeqSBAIJ_N(Mat,Vec,Vec);
949495ac64SHong Zhang 
950481f469SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_N(Mat,Mat,const MatFactorInfo*);
960481f469SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_1(Mat,Mat,const MatFactorInfo*);
970481f469SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_2(Mat,Mat,const MatFactorInfo*);
980481f469SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_3(Mat,Mat,const MatFactorInfo*);
990481f469SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_4(Mat,Mat,const MatFactorInfo*);
1000481f469SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_5(Mat,Mat,const MatFactorInfo*);
1010481f469SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_6(Mat,Mat,const MatFactorInfo*);
1020481f469SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_7(Mat,Mat,const MatFactorInfo*);
1039495ac64SHong Zhang 
1049495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_N(Mat,Vec,Vec);
1059495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_1(Mat,Vec,Vec);
1069495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_2(Mat,Vec,Vec);
1079495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_3(Mat,Vec,Vec);
1089495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_4(Mat,Vec,Vec);
1099495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_5(Mat,Vec,Vec);
1109495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_6(Mat,Vec,Vec);
1119495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_7(Mat,Vec,Vec);
1129495ac64SHong Zhang 
1139495ac64SHong Zhang EXTERN PetscErrorCode MatSolves_SeqSBAIJ_1(Mat,Vecs,Vecs);
1149495ac64SHong Zhang 
1159495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_1_NaturalOrdering(Mat,Vec,Vec);
1169495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_2_NaturalOrdering(Mat,Vec,Vec);
1179495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_3_NaturalOrdering(Mat,Vec,Vec);
1189495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_4_NaturalOrdering(Mat,Vec,Vec);
1199495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_5_NaturalOrdering(Mat,Vec,Vec);
1209495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_6_NaturalOrdering(Mat,Vec,Vec);
1219495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_7_NaturalOrdering(Mat,Vec,Vec);
1229495ac64SHong Zhang EXTERN PetscErrorCode MatSolve_SeqSBAIJ_N_NaturalOrdering(Mat,Vec,Vec);
1239495ac64SHong Zhang 
1249495ac64SHong Zhang EXTERN PetscErrorCode MatMult_SeqSBAIJ_1(Mat,Vec,Vec);
1259495ac64SHong Zhang EXTERN PetscErrorCode MatMult_SeqSBAIJ_2(Mat,Vec,Vec);
1269495ac64SHong Zhang EXTERN PetscErrorCode MatMult_SeqSBAIJ_3(Mat,Vec,Vec);
1279495ac64SHong Zhang EXTERN PetscErrorCode MatMult_SeqSBAIJ_4(Mat,Vec,Vec);
1289495ac64SHong Zhang EXTERN PetscErrorCode MatMult_SeqSBAIJ_5(Mat,Vec,Vec);
1299495ac64SHong Zhang EXTERN PetscErrorCode MatMult_SeqSBAIJ_6(Mat,Vec,Vec);
1309495ac64SHong Zhang EXTERN PetscErrorCode MatMult_SeqSBAIJ_7(Mat,Vec,Vec);
1319495ac64SHong Zhang EXTERN PetscErrorCode MatMult_SeqSBAIJ_N(Mat,Vec,Vec);
1329495ac64SHong Zhang 
1339495ac64SHong Zhang EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_1(Mat,Vec,Vec,Vec);
1349495ac64SHong Zhang EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_2(Mat,Vec,Vec,Vec);
1359495ac64SHong Zhang EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_3(Mat,Vec,Vec,Vec);
1369495ac64SHong Zhang EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_4(Mat,Vec,Vec,Vec);
1379495ac64SHong Zhang EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_5(Mat,Vec,Vec,Vec);
1389495ac64SHong Zhang EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_6(Mat,Vec,Vec,Vec);
1399495ac64SHong Zhang EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_7(Mat,Vec,Vec,Vec);
1409495ac64SHong Zhang EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_N(Mat,Vec,Vec,Vec);
141671cb588SHong Zhang 
14213f74950SBarry Smith EXTERN PetscErrorCode MatRelax_SeqSBAIJ(Mat,Vec,PetscReal,MatSORType,PetscReal,PetscInt,PetscInt,Vec);
143a313700dSBarry Smith EXTERN PetscErrorCode MatLoad_SeqSBAIJ(PetscViewer, const MatType,Mat*);
144d06b337dSHong Zhang 
14549b5e25fSSatish Balay #endif
146