1*b16c9716SHong Zhang /* $Id: sbaij.h,v 1.1 2000/06/21 15:47:04 balay Exp hzhang $ */ 249b5e25fSSatish Balay 349b5e25fSSatish Balay #include "src/mat/matimpl.h" 449b5e25fSSatish Balay 549b5e25fSSatish Balay #if !defined(__SBAIJ_H) 649b5e25fSSatish Balay #define __SBAIJ_H 749b5e25fSSatish Balay 849b5e25fSSatish Balay /* 949b5e25fSSatish Balay MATSEQSBAIJ format - Block compressed row storage. The i[] and j[] 1049b5e25fSSatish Balay arrays start at 0. 1149b5e25fSSatish Balay */ 1249b5e25fSSatish Balay 1349b5e25fSSatish Balay typedef struct { 1449b5e25fSSatish Balay PetscTruth sorted; /* if true, rows are sorted by increasing columns */ 1549b5e25fSSatish Balay PetscTruth roworiented; /* if true, row-oriented input, default */ 1649b5e25fSSatish Balay int nonew; /* 1 don't add new nonzeros, -1 generate error on new */ 1749b5e25fSSatish Balay PetscTruth singlemalloc; /* if true a, i, and j have been obtained with 1849b5e25fSSatish Balay one big malloc */ 1949b5e25fSSatish Balay int m,n; /* rows or columns */ 2049b5e25fSSatish Balay int bs,bs2; /* block size, square of block size */ 2149b5e25fSSatish Balay int mbs,nbs; /* rows/bs or columns/bs */ 2249b5e25fSSatish Balay int s_nz,s_maxnz; /* total diagonal and superdiagonal nonzero blocks, 2349b5e25fSSatish Balay total allocated diagonal and superdiagonal nonzero blocks*/ 2449b5e25fSSatish Balay int *diag; /* pointers to diagonal elements */ 2549b5e25fSSatish Balay int *i; /* pointer to beginning of each row */ 2649b5e25fSSatish Balay int *imax; /* maximum space allocated for each row */ 2749b5e25fSSatish Balay int *ilen; /* actual length of each row */ 2849b5e25fSSatish Balay int *j; /* column values: j + i[k] is start of row k */ 2949b5e25fSSatish Balay MatScalar *a; /* nonzero diagonal and subdiagonal elements */ 3049b5e25fSSatish Balay IS row,col,icol; /* index sets, used for reorderings */ 3149b5e25fSSatish Balay Scalar *solve_work; /* work space used in MatSolve */ 3249b5e25fSSatish Balay void *spptr; /* pointer for special library like SuperLU */ 3349b5e25fSSatish Balay int reallocs; /* number of mallocs done during MatSetValues() 3449b5e25fSSatish Balay as more values are set then were preallocated */ 3549b5e25fSSatish Balay Scalar *mult_work; /* work array for matrix vector product*/ 3649b5e25fSSatish Balay Scalar *saved_values; 3749b5e25fSSatish Balay 3849b5e25fSSatish Balay PetscTruth keepzeroedrows; /* if true, MatZeroRows() will not change nonzero structure */ 3949b5e25fSSatish Balay } Mat_SeqSBAIJ; 4049b5e25fSSatish Balay 4149b5e25fSSatish Balay extern int MatILUFactorSymbolic_SeqSBAIJ(Mat,IS,IS,MatILUInfo*,Mat *); 4249b5e25fSSatish Balay extern int MatConvert_SeqSBAIJ(Mat,MatType,Mat *); 4349b5e25fSSatish Balay extern int MatDuplicate_SeqSBAIJ(Mat,MatDuplicateOption,Mat*); 4449b5e25fSSatish Balay extern int MatMarkDiagonal_SeqSBAIJ(Mat); 4549b5e25fSSatish Balay 4649b5e25fSSatish Balay extern int MatSolveTranspose_SeqSBAIJ_1_NaturalOrdering(Mat,Vec,Vec); 4749b5e25fSSatish Balay extern int MatLUFactorNumeric_SeqSBAIJ_2_NaturalOrdering(Mat,Mat*); 4849b5e25fSSatish Balay extern int MatSolve_SeqSBAIJ_2_NaturalOrdering(Mat,Vec,Vec); 4949b5e25fSSatish Balay extern int MatSolveTranspose_SeqSBAIJ_2_NaturalOrdering(Mat,Vec,Vec); 5049b5e25fSSatish Balay extern int MatLUFactorNumeric_SeqSBAIJ_3_NaturalOrdering(Mat,Mat*); 5149b5e25fSSatish Balay extern int MatSolve_SeqSBAIJ_3_NaturalOrdering(Mat,Vec,Vec); 5249b5e25fSSatish Balay extern int MatSolveTranspose_SeqSBAIJ_3_NaturalOrdering(Mat,Vec,Vec); 5349b5e25fSSatish Balay extern int MatLUFactorNumeric_SeqSBAIJ_4_NaturalOrdering(Mat,Mat*); 5449b5e25fSSatish Balay extern int MatSolve_SeqSBAIJ_4_NaturalOrdering(Mat,Vec,Vec); 5549b5e25fSSatish Balay extern int MatSolveTranspose_SeqSBAIJ_4_NaturalOrdering(Mat,Vec,Vec); 5649b5e25fSSatish Balay extern int MatLUFactorNumeric_SeqSBAIJ_5_NaturalOrdering(Mat,Mat*); 5749b5e25fSSatish Balay extern int MatSolve_SeqSBAIJ_5_NaturalOrdering(Mat,Vec,Vec); 5849b5e25fSSatish Balay extern int MatSolveTranspose_SeqSBAIJ_5_NaturalOrdering(Mat,Vec,Vec); 5949b5e25fSSatish Balay extern int MatLUFactorNumeric_SeqSBAIJ_6_NaturalOrdering(Mat,Mat*); 6049b5e25fSSatish Balay extern int MatSolve_SeqSBAIJ_6_NaturalOrdering(Mat,Vec,Vec); 6149b5e25fSSatish Balay extern int MatSolveTranspose_SeqSBAIJ_6_NaturalOrdering(Mat,Vec,Vec); 6249b5e25fSSatish Balay extern int MatLUFactorNumeric_SeqSBAIJ_7_NaturalOrdering(Mat,Mat*); 6349b5e25fSSatish Balay extern int MatSolve_SeqSBAIJ_7_NaturalOrdering(Mat,Vec,Vec); 6449b5e25fSSatish Balay extern int MatSolveTranspose_SeqSBAIJ_7_NaturalOrdering(Mat,Vec,Vec); 6549b5e25fSSatish Balay 66*b16c9716SHong Zhang extern int MatReordering SeqSBAIJ(Mat,IS); 6749b5e25fSSatish Balay /* 6849b5e25fSSatish Balay extern int MatCreateSeqSBAIJ(MPI_Comm,int,int,int,int,int*,Mat*); 6949b5e25fSSatish Balay extern int MatSetValuesBlocked_SeqSBAIJ(Mat,int,int*,int,int*,Scalar*,InsertMode); 7049b5e25fSSatish Balay */ 7149b5e25fSSatish Balay #endif 72