173f4d377SMatthew Knepley /* $Id: sbaij.h,v 1.21 2001/08/07 03:03:01 balay Exp $ */ 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 bs,bs2; /* block size, square of block size */ 2049b5e25fSSatish Balay int mbs,nbs; /* rows/bs or columns/bs */ 2149b5e25fSSatish Balay int s_nz,s_maxnz; /* total diagonal and superdiagonal nonzero blocks, 2249b5e25fSSatish Balay total allocated diagonal and superdiagonal nonzero blocks*/ 2349b5e25fSSatish Balay int *diag; /* pointers to diagonal elements */ 24a6175056SHong Zhang int *i; /* pointer to beginning of each row */ 25a6175056SHong Zhang int *inew; /* pointer to beginning of each row of reordered matrix */ 2649b5e25fSSatish Balay int *imax; /* maximum space allocated for each row */ 2749b5e25fSSatish Balay int *ilen; /* actual length of each row */ 28a6175056SHong Zhang int *j; /* column values: j + i[k] is start of row k of reordered matrix */ 29a6175056SHong Zhang int *jnew; /* column values: jnew + i[k] is start of row k */ 30a6175056SHong Zhang MatScalar *a; /* nonzero diagonal and superdiagonal elements */ 31a6175056SHong Zhang MatScalar *anew; /* nonzero diagonal and superdiagonal elements of reordered matrix */ 32392d98d6SHong Zhang IS row,icol; /* index sets, used for reorderings */ 33ea709b57SSatish Balay PetscScalar *solve_work; /* work space used in MatSolve */ 34d59c15a7SBarry Smith PetscScalar *solves_work; /* work space used in MatSolves */ 35d59c15a7SBarry Smith int solves_work_n;/* size of solves_work */ 3649b5e25fSSatish Balay int reallocs; /* number of mallocs done during MatSetValues() 3749b5e25fSSatish Balay as more values are set then were preallocated */ 38ea709b57SSatish Balay PetscScalar *mult_work; /* work array for matrix vector product*/ 39ea709b57SSatish Balay PetscScalar *saved_values; 4049b5e25fSSatish Balay 4149b5e25fSSatish Balay PetscTruth keepzeroedrows; /* if true, MatZeroRows() will not change nonzero structure */ 42a58ccc28SHong Zhang int *a2anew; /* map used for symm permutation */ 43d36791b2SHong Zhang PetscTruth permute; /* if true, a non-trivial permutation is used for factorization */ 44bcd9e38bSBarry Smith PetscTruth pivotinblocks; /* pivot inside factorization of each diagonal block */ 45b00f7748SHong Zhang 46b45a75daSHong Zhang /* carry MatFactorInfo from symbolic factor to numeric factor */ 47b45a75daSHong Zhang int factor_levels; 48b45a75daSHong Zhang PetscReal factor_damping; 49f4cce38bSHong Zhang PetscReal factor_shift; 50b45a75daSHong Zhang PetscReal factor_zeropivot; 51c4319e64SHong Zhang 52c4319e64SHong Zhang int *xtoy,*xtoyB; /* map nonzero pattern of X into Y's, used by MatAXPY() */ 53c4319e64SHong Zhang Mat XtoY; /* used by MatAXPY() */ 5449b5e25fSSatish Balay } Mat_SeqSBAIJ; 5549b5e25fSSatish Balay 5697304618SKris Buschelman EXTERN int MatICCFactorSymbolic_SeqSBAIJ(Mat,IS,MatFactorInfo*,Mat *); 5797304618SKris Buschelman EXTERN int MatDuplicate_SeqSBAIJ(Mat,MatDuplicateOption,Mat*); 5897304618SKris Buschelman EXTERN int MatMarkDiagonal_SeqSBAIJ(Mat); 5949b5e25fSSatish Balay 6097304618SKris Buschelman EXTERN int MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering(Mat,Mat*); 6197304618SKris Buschelman EXTERN int MatSolve_SeqSBAIJ_1_NaturalOrdering(Mat,Vec,Vec); 6297304618SKris Buschelman EXTERN int MatSolveTranspose_SeqSBAIJ_1_NaturalOrdering(Mat,Vec,Vec); 63266135f8SHong Zhang 6497304618SKris Buschelman EXTERN int MatCholeskyFactorNumeric_SeqSBAIJ_2_NaturalOrdering(Mat,Mat*); 6597304618SKris Buschelman EXTERN int MatSolve_SeqSBAIJ_2_NaturalOrdering(Mat,Vec,Vec); 6697304618SKris Buschelman EXTERN int MatSolveTranspose_SeqSBAIJ_2_NaturalOrdering(Mat,Vec,Vec); 67266135f8SHong Zhang 6897304618SKris Buschelman EXTERN int MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering(Mat,Mat*); 6997304618SKris Buschelman EXTERN int MatSolve_SeqSBAIJ_3_NaturalOrdering(Mat,Vec,Vec); 7097304618SKris Buschelman EXTERN int MatSolveTranspose_SeqSBAIJ_3_NaturalOrdering(Mat,Vec,Vec); 71266135f8SHong Zhang 7297304618SKris Buschelman EXTERN int MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering(Mat,Mat*); 7397304618SKris Buschelman EXTERN int MatSolve_SeqSBAIJ_4_NaturalOrdering(Mat,Vec,Vec); 7497304618SKris Buschelman EXTERN int MatSolveTranspose_SeqSBAIJ_4_NaturalOrdering(Mat,Vec,Vec); 75266135f8SHong Zhang 7697304618SKris Buschelman EXTERN int MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering(Mat,Mat*); 7797304618SKris Buschelman EXTERN int MatSolve_SeqSBAIJ_5_NaturalOrdering(Mat,Vec,Vec); 7897304618SKris Buschelman EXTERN int MatSolveTranspose_SeqSBAIJ_5_NaturalOrdering(Mat,Vec,Vec); 79266135f8SHong Zhang 8097304618SKris Buschelman EXTERN int MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering(Mat,Mat*); 8197304618SKris Buschelman EXTERN int MatSolve_SeqSBAIJ_6_NaturalOrdering(Mat,Vec,Vec); 8297304618SKris Buschelman EXTERN int MatSolveTranspose_SeqSBAIJ_6_NaturalOrdering(Mat,Vec,Vec); 83266135f8SHong Zhang 8497304618SKris Buschelman EXTERN int MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering(Mat,Mat*); 8597304618SKris Buschelman EXTERN int MatSolve_SeqSBAIJ_7_NaturalOrdering(Mat,Vec,Vec); 8697304618SKris Buschelman EXTERN int MatSolveTranspose_SeqSBAIJ_7_NaturalOrdering(Mat,Vec,Vec); 8749b5e25fSSatish Balay 8897304618SKris Buschelman EXTERN int MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering(Mat,Mat*); 8997304618SKris Buschelman EXTERN int MatSolve_SeqSBAIJ_N_NaturalOrdering(Mat,Vec,Vec); 90671cb588SHong Zhang 9197304618SKris Buschelman EXTERN int MatRelax_SeqSBAIJ(Mat,Vec,PetscReal,MatSORType,PetscReal,int,int,Vec); 92*f248c16bSBarry Smith EXTERN int MatLoad_SeqSBAIJ(PetscViewer,const MatType,Mat*); 93d06b337dSHong Zhang 9449b5e25fSSatish Balay #endif 95