xref: /petsc/src/mat/impls/baij/seq/baij.h (revision 4110d75be2962da0a7dbdbf03d2053864a006c7d)
1455d6844SKris Buschelman /* $Id: baij.h,v 1.27 2000/10/24 20:25:52 bsmith Exp buschelm $ */
22593348eSBarry Smith 
370f55243SBarry Smith #include "src/mat/matimpl.h"
42593348eSBarry Smith 
535aab85fSBarry Smith #if !defined(__BAIJ_H)
635aab85fSBarry Smith #define __BAIJ_H
72593348eSBarry Smith 
82593348eSBarry Smith /*
935aab85fSBarry Smith   MATSEQBAIJ format - Block compressed row storage. The i[] and j[]
10e24b481bSBarry Smith   arrays start at 0.
112593348eSBarry Smith */
122593348eSBarry Smith 
132593348eSBarry Smith typedef struct {
14c4992f7dSBarry Smith   PetscTruth       sorted;       /* if true, rows are sorted by increasing columns */
15c4992f7dSBarry Smith   PetscTruth       roworiented;  /* if true, row-oriented input, default */
163369ce9aSBarry Smith   int              nonew;        /* 1 don't add new nonzeros, -1 generate error on new */
17c4992f7dSBarry Smith   PetscTruth       singlemalloc; /* if true a, i, and j have been obtained with
182593348eSBarry Smith                                         one big malloc */
192a90d852SSatish Balay   int              bs,bs2;       /* block size, square of block size */
2035aab85fSBarry Smith   int              mbs,nbs;      /* rows/bs, columns/bs */
212593348eSBarry Smith   int              nz,maxnz;     /* nonzeros, allocated nonzeros */
222593348eSBarry Smith   int              *diag;        /* pointers to diagonal elements */
232593348eSBarry Smith   int              *i;           /* pointer to beginning of each row */
242593348eSBarry Smith   int              *imax;        /* maximum space allocated for each row */
252593348eSBarry Smith   int              *ilen;        /* actual length of each row */
262593348eSBarry Smith   int              *j;           /* column values: j + i[k] - 1 is start of row k */
273f1db9ecSBarry Smith   MatScalar        *a;           /* nonzero elements */
28e51c0b9cSSatish Balay   IS               row,col,icol; /* index sets, used for reorderings */
292593348eSBarry Smith   Scalar           *solve_work;  /* work space used in MatSolve */
302593348eSBarry Smith   void             *spptr;       /* pointer for special library like SuperLU */
312593348eSBarry Smith   int              reallocs;     /* number of mallocs done during MatSetValues()
3247b4a8eaSLois Curfman McInnes                                     as more values are set then were preallocated */
334eeb42bcSBarry Smith   Scalar           *mult_work;   /* work array for matrix vector product*/
347fc3c18eSBarry Smith   Scalar           *saved_values;
35c4992f7dSBarry Smith 
36c4992f7dSBarry Smith   PetscTruth       keepzeroedrows; /* if true, MatZeroRows() will not change nonzero structure */
37a377f70aSBarry Smith 
38a377f70aSBarry Smith #if defined(PETSC_USE_MAT_SINGLE)
39e7592fafSBarry Smith   int              setvalueslen;
40455d6844SKris Buschelman   int              singleprecisionsolves;
41a377f70aSBarry Smith   MatScalar        *setvaluescopy; /* area double precision values in MatSetValuesXXX() are copied
42a377f70aSBarry Smith                                       before calling MatSetValuesXXX_SeqBAIJ_MatScalar() */
43a377f70aSBarry Smith #endif
4435aab85fSBarry Smith } Mat_SeqBAIJ;
452593348eSBarry Smith 
46ca44d042SBarry Smith EXTERN int MatILUFactorSymbolic_SeqBAIJ(Mat,IS,IS,MatILUInfo*,Mat *);
47ca44d042SBarry Smith EXTERN int MatDuplicate_SeqBAIJ(Mat,MatDuplicateOption,Mat*);
48ca44d042SBarry Smith EXTERN int MatMarkDiagonal_SeqBAIJ(Mat);
49be3590efSBarry Smith 
50e37c484bSBarry Smith EXTERN int MatSolve_SeqBAIJ_1_NaturalOrdering(Mat,Vec,Vec);
51ca44d042SBarry Smith EXTERN int MatSolveTranspose_SeqBAIJ_1_NaturalOrdering(Mat,Vec,Vec);
52ca44d042SBarry Smith EXTERN int MatLUFactorNumeric_SeqBAIJ_2_NaturalOrdering(Mat,Mat*);
53ca44d042SBarry Smith EXTERN int MatSolve_SeqBAIJ_2_NaturalOrdering(Mat,Vec,Vec);
54ca44d042SBarry Smith EXTERN int MatSolveTranspose_SeqBAIJ_2_NaturalOrdering(Mat,Vec,Vec);
55ca44d042SBarry Smith EXTERN int MatLUFactorNumeric_SeqBAIJ_3_NaturalOrdering(Mat,Mat*);
56ca44d042SBarry Smith EXTERN int MatSolve_SeqBAIJ_3_NaturalOrdering(Mat,Vec,Vec);
57ca44d042SBarry Smith EXTERN int MatSolveTranspose_SeqBAIJ_3_NaturalOrdering(Mat,Vec,Vec);
58ca44d042SBarry Smith EXTERN int MatLUFactorNumeric_SeqBAIJ_4_NaturalOrdering(Mat,Mat*);
59ca44d042SBarry Smith EXTERN int MatSolve_SeqBAIJ_4_NaturalOrdering(Mat,Vec,Vec);
60*4110d75bSKris Buschelman #ifdef PETSC_HAVE_SSE
61*4110d75bSKris Buschelman EXTERN int MatLUFactorNumeric_SeqBAIJ_4_NaturalOrdering_SSE(Mat,Mat*);
62*4110d75bSKris Buschelman EXTERN int MatSolve_SeqBAIJ_4_NaturalOrdering_SSE_Demotion(Mat,Vec,Vec);
63*4110d75bSKris Buschelman #endif
64ca44d042SBarry Smith EXTERN int MatSolveTranspose_SeqBAIJ_4_NaturalOrdering(Mat,Vec,Vec);
65ca44d042SBarry Smith EXTERN int MatLUFactorNumeric_SeqBAIJ_5_NaturalOrdering(Mat,Mat*);
66ca44d042SBarry Smith EXTERN int MatSolve_SeqBAIJ_5_NaturalOrdering(Mat,Vec,Vec);
67ca44d042SBarry Smith EXTERN int MatSolveTranspose_SeqBAIJ_5_NaturalOrdering(Mat,Vec,Vec);
68ca44d042SBarry Smith EXTERN int MatLUFactorNumeric_SeqBAIJ_6_NaturalOrdering(Mat,Mat*);
69ca44d042SBarry Smith EXTERN int MatSolve_SeqBAIJ_6_NaturalOrdering(Mat,Vec,Vec);
70ca44d042SBarry Smith EXTERN int MatSolveTranspose_SeqBAIJ_6_NaturalOrdering(Mat,Vec,Vec);
71ca44d042SBarry Smith EXTERN int MatLUFactorNumeric_SeqBAIJ_7_NaturalOrdering(Mat,Mat*);
72ca44d042SBarry Smith EXTERN int MatSolve_SeqBAIJ_7_NaturalOrdering(Mat,Vec,Vec);
73ca44d042SBarry Smith EXTERN int MatSolveTranspose_SeqBAIJ_7_NaturalOrdering(Mat,Vec,Vec);
742593348eSBarry Smith 
752593348eSBarry Smith #endif
76