xref: /petsc/src/mat/impls/aij/seq/aij.h (revision f349cc8147fd8be7fc2bb8bc6274b020a4c1628f)
1b8a66259SBarry Smith 
22d40f771SBarry Smith #if !defined(__AIJ_H)
32d40f771SBarry Smith #define __AIJ_H
471c2f376SKris Buschelman #include "src/mat/impls/csr/inode/inode.h"
52d40f771SBarry Smith /*
6ec8511deSBarry Smith   MATSEQAIJ format - Compressed row storage (also called Yale sparse matrix
7dfbc5765Svictorle   format).  The i[] and j[] arrays start at 0. For example,
8dfbc5765Svictorle   j[i[k]+p] is the pth column in row k.  Note that the diagonal
95768c4f9SLois Curfman McInnes   matrix elements are stored with the rest of the nonzeros (not separately).
102d40f771SBarry Smith */
11d35516d3SLois Curfman McInnes 
12b8a66259SBarry Smith typedef struct {
1371c2f376SKris Buschelman   MAT_INODE_HEADER;
147c922b88SBarry Smith   PetscTruth       sorted;           /* if true, rows are sorted by increasing columns */
157c922b88SBarry Smith   PetscTruth       roworiented;      /* if true, row-oriented input, default */
1638baddfdSBarry Smith   PetscInt         nonew;            /* 1 don't add new nonzeros, -1 generate error on new */
177c922b88SBarry Smith   PetscTruth       singlemalloc;     /* if true a, i, and j have been obtained with
18d35516d3SLois Curfman McInnes                                           one big malloc */
19329f5518SBarry Smith   PetscTruth       freedata;         /* free the i,j,a data when the matrix is destroyed; true by default */
2071c2f376SKris Buschelman   PetscInt         maxnz;            /* allocated nonzeros */
2197f1f81fSBarry Smith   PetscInt         *imax;            /* maximum space allocated for each row */
2238baddfdSBarry Smith   PetscInt         *ilen;            /* actual length of each row */
2397f1f81fSBarry Smith   PetscInt         reallocs;         /* number of mallocs done during MatSetValues()
247823673eSLois Curfman McInnes                                         as more values are set than were prealloced */
2597f1f81fSBarry Smith   PetscInt         rmax;             /* max nonzeros in any row */
26ea709b57SSatish Balay   PetscScalar      *saved_values;    /* location for stashing nonzero values of matrix */
27ea709b57SSatish Balay   PetscScalar      *idiag,*ssor;     /* inverse of diagonal entries; space for eisen */
287c922b88SBarry Smith 
297c922b88SBarry Smith   PetscTruth       keepzeroedrows;   /* keeps matrix structure same in calls to MatZeroRows()*/
30329f5518SBarry Smith   PetscTruth       ignorezeroentries;
313a7fca6bSBarry Smith   ISColoring       coloring;         /* set with MatADSetColoring() used by MatADSetValues() */
32a30b2313SHong Zhang 
3397f1f81fSBarry Smith   PetscInt         *xtoy,*xtoyB;     /* map nonzero pattern of X into Y's, used by MatAXPY() */
34a30b2313SHong Zhang   Mat              XtoY;             /* used by MatAXPY() */
3573e7a558SHong Zhang   Mat_CompressedRow compressedrow;   /* use compressed row format */
36ec8511deSBarry Smith } Mat_SeqAIJ;
37b8a66259SBarry Smith 
38*f349cc81SSatish Balay EXTERN_C_BEGIN
39ab93d7beSBarry Smith EXTERN PetscErrorCode MatSeqAIJSetPreallocation_SeqAIJ(Mat,PetscInt,PetscInt*);
40*f349cc81SSatish Balay EXTERN_C_END
41dfbe8321SBarry Smith EXTERN PetscErrorCode MatILUFactorSymbolic_SeqAIJ(Mat,IS,IS,MatFactorInfo*,Mat *);
42dfbe8321SBarry Smith EXTERN PetscErrorCode MatICCFactorSymbolic_SeqAIJ(Mat,IS,MatFactorInfo*,Mat *);
43dfbe8321SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorSymbolic_SeqAIJ(Mat,IS,MatFactorInfo*,Mat*);
44af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqAIJ(Mat,MatFactorInfo*,Mat*);
45dfbe8321SBarry Smith EXTERN PetscErrorCode MatDuplicate_SeqAIJ(Mat,MatDuplicateOption,Mat*);
46dfbe8321SBarry Smith EXTERN PetscErrorCode MatMissingDiagonal_SeqAIJ(Mat);
47dfbe8321SBarry Smith EXTERN PetscErrorCode MatMarkDiagonal_SeqAIJ(Mat);
4808480c60SBarry Smith 
49dfbe8321SBarry Smith EXTERN PetscErrorCode MatMult_SeqAIJ(Mat A,Vec,Vec);
50dfbe8321SBarry Smith EXTERN PetscErrorCode MatMultAdd_SeqAIJ(Mat A,Vec,Vec,Vec);
51dfbe8321SBarry Smith EXTERN PetscErrorCode MatMultTranspose_SeqAIJ(Mat A,Vec,Vec);
52dfbe8321SBarry Smith EXTERN PetscErrorCode MatMultTransposeAdd_SeqAIJ(Mat A,Vec,Vec,Vec);
5338baddfdSBarry Smith EXTERN PetscErrorCode MatRelax_SeqAIJ(Mat,Vec,PetscReal,MatSORType,PetscReal,PetscInt,PetscInt,Vec);
5408480c60SBarry Smith 
55dfbe8321SBarry Smith EXTERN PetscErrorCode MatSetColoring_SeqAIJ(Mat,ISColoring);
56dfbe8321SBarry Smith EXTERN PetscErrorCode MatSetValuesAdic_SeqAIJ(Mat,void*);
5738baddfdSBarry Smith EXTERN PetscErrorCode MatSetValuesAdifor_SeqAIJ(Mat,PetscInt,void*);
583a7fca6bSBarry Smith 
5938baddfdSBarry Smith EXTERN PetscErrorCode MatGetSymbolicTranspose_SeqAIJ(Mat,PetscInt *[],PetscInt *[]);
600390132cSHong Zhang EXTERN PetscErrorCode MatGetSymbolicTransposeReduced_SeqAIJ(Mat,PetscInt,PetscInt,PetscInt *[],PetscInt *[]);
6138baddfdSBarry Smith EXTERN PetscErrorCode MatRestoreSymbolicTranspose_SeqAIJ(Mat,PetscInt *[],PetscInt *[]);
6238baddfdSBarry Smith EXTERN PetscErrorCode MatToSymmetricIJ_SeqAIJ(PetscInt,PetscInt*,PetscInt*,PetscInt,PetscInt,PetscInt**,PetscInt**);
63dfbe8321SBarry Smith EXTERN PetscErrorCode MatLUFactorSymbolic_SeqAIJ(Mat,IS,IS,MatFactorInfo*,Mat*);
64af281ebdSHong Zhang EXTERN PetscErrorCode MatLUFactorNumeric_SeqAIJ(Mat,MatFactorInfo*,Mat*);
65dfbe8321SBarry Smith EXTERN PetscErrorCode MatLUFactor_SeqAIJ(Mat,IS,IS,MatFactorInfo*);
66dfbe8321SBarry Smith EXTERN PetscErrorCode MatSolve_SeqAIJ(Mat,Vec,Vec);
67dfbe8321SBarry Smith EXTERN PetscErrorCode MatSolveAdd_SeqAIJ(Mat,Vec,Vec,Vec);
68dfbe8321SBarry Smith EXTERN PetscErrorCode MatSolveTranspose_SeqAIJ(Mat,Vec,Vec);
69dfbe8321SBarry Smith EXTERN PetscErrorCode MatSolveTransposeAdd_SeqAIJ(Mat,Vec,Vec,Vec);
70dfbe8321SBarry Smith EXTERN PetscErrorCode MatEqual_SeqAIJ(Mat A,Mat B,PetscTruth* flg);
71dfbe8321SBarry Smith EXTERN PetscErrorCode MatFDColoringCreate_SeqAIJ(Mat,ISColoring,MatFDColoring);
727529efd4SKris Buschelman EXTERN PetscErrorCode MatILUDTFactor_SeqAIJ(Mat,IS,IS,MatFactorInfo*,Mat*);
73dfbe8321SBarry Smith EXTERN PetscErrorCode MatLoad_SeqAIJ(PetscViewer,const MatType,Mat*);
74dfbe8321SBarry Smith EXTERN PetscErrorCode RegisterApplyPtAPRoutines_Private(Mat);
75dfbe8321SBarry Smith EXTERN PetscErrorCode MatMatMult_SeqAIJ_SeqAIJ(Mat,Mat,MatReuse,PetscReal,Mat*);
76dfbe8321SBarry Smith EXTERN PetscErrorCode MatMatMultSymbolic_SeqAIJ_SeqAIJ(Mat,Mat,PetscReal,Mat*);
77dfbe8321SBarry Smith EXTERN PetscErrorCode MatMatMultNumeric_SeqAIJ_SeqAIJ(Mat,Mat,Mat);
787ba1a0bfSKris Buschelman EXTERN PetscErrorCode MatPtAPSymbolic_SeqAIJ(Mat,Mat,PetscReal,Mat*);
797ba1a0bfSKris Buschelman EXTERN PetscErrorCode MatPtAPNumeric_SeqAIJ(Mat,Mat,Mat);
80dfbe8321SBarry Smith EXTERN PetscErrorCode MatPtAPSymbolic_SeqAIJ_SeqAIJ(Mat,Mat,PetscReal,Mat*);
81dfbe8321SBarry Smith EXTERN PetscErrorCode MatPtAPNumeric_SeqAIJ_SeqAIJ(Mat,Mat,Mat);
82bc011b1eSHong Zhang EXTERN PetscErrorCode MatMatMultTranspose_SeqAIJ_SeqAIJ(Mat,Mat,MatReuse,PetscReal,Mat*);
83bc011b1eSHong Zhang EXTERN PetscErrorCode MatMatMultTransposeSymbolic_SeqAIJ_SeqAIJ(Mat,Mat,PetscReal,Mat*);
84bc011b1eSHong Zhang EXTERN PetscErrorCode MatMatMultTransposeNumeric_SeqAIJ_SeqAIJ(Mat,Mat,Mat);
8538baddfdSBarry Smith EXTERN PetscErrorCode MatSetValues_SeqAIJ(Mat,PetscInt,const PetscInt[],PetscInt,const PetscInt[],const PetscScalar[],InsertMode);
8638baddfdSBarry Smith EXTERN PetscErrorCode MatGetRow_SeqAIJ(Mat,PetscInt,PetscInt*,PetscInt**,PetscScalar**);
8738baddfdSBarry Smith EXTERN PetscErrorCode MatRestoreRow_SeqAIJ(Mat,PetscInt,PetscInt*,PetscInt**,PetscScalar**);
8811d72411SSatish Balay EXTERN PetscErrorCode MatPrintHelp_SeqAIJ(Mat);
8911d72411SSatish Balay EXTERN PetscErrorCode MatAXPY_SeqAIJ(const PetscScalar[],Mat,Mat,MatStructure);
909af31e4aSHong Zhang 
9197304618SKris Buschelman EXTERN_C_BEGIN
92be1d678aSKris Buschelman EXTERN PetscErrorCode PETSCMAT_DLLEXPORT MatConvert_SeqAIJ_SeqSBAIJ(Mat,const MatType,MatReuse,Mat*);
93be1d678aSKris Buschelman EXTERN PetscErrorCode PETSCMAT_DLLEXPORT MatConvert_SeqAIJ_SeqBAIJ(Mat,const MatType,MatReuse,Mat*);
94be1d678aSKris Buschelman EXTERN PetscErrorCode PETSCMAT_DLLEXPORT MatReorderForNonzeroDiagonal_SeqAIJ(Mat,PetscReal,IS,IS);
9597304618SKris Buschelman EXTERN_C_END
9670f19b1fSKris Buschelman 
972d40f771SBarry Smith #endif
98