xref: /petsc/include/petscmat.h (revision 8b3e568e5a27c52bac4d391d6abc96331f935c8f)
1*8b3e568eSBarry Smith /* $Id: mat.h,v 1.93 1996/02/22 16:33:58 curfman Exp bsmith $ */
22eac72dbSBarry Smith /*
32eac72dbSBarry Smith      Include file for the matrix component of PETSc
42eac72dbSBarry Smith */
51eb62cbbSBarry Smith #ifndef __MAT_PACKAGE
61eb62cbbSBarry Smith #define __MAT_PACKAGE
78ed539a5SBarry Smith #include "vec.h"
82eac72dbSBarry Smith 
99cd28387SBarry Smith #define MAT_COOKIE         PETSC_COOKIE+5
10f0479e8cSBarry Smith 
112eac72dbSBarry Smith typedef struct _Mat*           Mat;
122eac72dbSBarry Smith 
134ac9ca07SLois Curfman McInnes typedef enum { MATSAME=-1, MATSEQDENSE, MATSEQAIJ, MATMPIAIJ, MATSHELL,
14c456f294SBarry Smith                MATMPIROWBS, MATSEQBDIAG, MATMPIBDIAG,
151f3bdca4SBarry Smith                MATMPIDENSE, MATSEQBAIJ } MatType;
1628988994SBarry Smith 
170e998a85SBarry Smith extern int MatCreate(MPI_Comm,int,int,Mat*);
18df6e0191SLois Curfman McInnes extern int MatCreateSeqDense(MPI_Comm,int,int,Scalar*,Mat*);
19d5bc1036SLois Curfman McInnes extern int MatCreateMPIDense(MPI_Comm,int,int,int,int,Scalar*,Mat*);
20de7da479SBarry Smith extern int MatCreateSeqAIJ(MPI_Comm,int,int,int,int*,Mat*);
219717bf64SBarry Smith extern int MatCreateMPIAIJ(MPI_Comm,int,int,int,int,int,int*,int,int*,Mat*);
22cfe223acSLois Curfman McInnes extern int MatCreateMPIRowbs(MPI_Comm,int,int,int,int*,void*,Mat*);
23f26ada1bSBarry Smith extern int MatCreateSeqBDiag(MPI_Comm,int,int,int,int,int*,Scalar**,Mat*);
24f26ada1bSBarry Smith extern int MatCreateMPIBDiag(MPI_Comm,int,int,int,int,int,int*,Scalar**,Mat*);
25df8a92d1SBarry Smith extern int MatCreateSeqBAIJ(MPI_Comm,int,int,int,int,int*,Mat*);
269717bf64SBarry Smith 
277b80b807SBarry Smith extern int MatDestroy(Mat);
287b80b807SBarry Smith 
29052efed2SBarry Smith extern int MatCreateShell(MPI_Comm,int,int,void *,Mat*);
3017116c71SBarry Smith extern int MatShellGetContext(Mat,void **);
31aabeff55SBarry Smith extern int MatShellSetMult(Mat,int (*)(void*,Vec,Vec));
32b9fa9cd0SBarry Smith extern int MatShellSetDestroy(Mat,int (*)(void*));
33aabeff55SBarry Smith extern int MatShellSetMultTrans(Mat,int (*)(void*,Vec,Vec));
34f0479e8cSBarry Smith extern int MatShellSetMultTransAdd(Mat,int (*)(void*,Vec,Vec,Vec));
352eac72dbSBarry Smith 
36ec0117caSBarry Smith extern int MatPrintHelp(Mat);
37ec0117caSBarry Smith 
388ed539a5SBarry Smith /* ------------------------------------------------------------*/
39ee50ffe9SBarry Smith extern int MatSetValues(Mat,int,int*,int,int*,Scalar*,InsertMode);
407b80b807SBarry Smith typedef enum {FLUSH_ASSEMBLY=1,FINAL_ASSEMBLY=0} MatAssemblyType;
41ee50ffe9SBarry Smith extern int MatAssemblyBegin(Mat,MatAssemblyType);
42ee50ffe9SBarry Smith extern int MatAssemblyEnd(Mat,MatAssemblyType);
43ee50ffe9SBarry Smith 
44ee50ffe9SBarry Smith typedef enum {ROW_ORIENTED=1,COLUMN_ORIENTED=2,ROWS_SORTED=4,
45ee50ffe9SBarry Smith               COLUMNS_SORTED=8,NO_NEW_NONZERO_LOCATIONS=16,
4602834360SBarry Smith               YES_NEW_NONZERO_LOCATIONS=32,SYMMETRIC_MATRIX=64,
47c0bbcb79SLois Curfman McInnes               STRUCTURALLY_SYMMETRIC_MATRIX,NO_NEW_DIAGONALS,
48429c0c78SLois Curfman McInnes               YES_NEW_DIAGONALS,INODE_LIMIT_1,INODE_LIMIT_2,
49429c0c78SLois Curfman McInnes               INODE_LIMIT_3,INODE_LIMIT_4,INODE_LIMIT_5} MatOption;
50ee50ffe9SBarry Smith extern int MatSetOption(Mat,MatOption);
514b0e389bSBarry Smith extern int MatGetType(Mat,MatType*,char**);
52e3726954SLois Curfman McInnes extern int MatGetFormatFromOptions(MPI_Comm,char*,MatType*,int*);
53ee50ffe9SBarry Smith extern int MatGetValues(Mat,int,int*,int,int*,Scalar*);
548ed539a5SBarry Smith extern int MatGetRow(Mat,int,int *,int **,Scalar**);
558ed539a5SBarry Smith extern int MatRestoreRow(Mat,int,int *,int **,Scalar**);
568ed539a5SBarry Smith extern int MatGetCol(Mat,int,int *,int **,Scalar**);
578ed539a5SBarry Smith extern int MatRestoreCol(Mat,int,int *,int **,Scalar**);
5864e87e97SBarry Smith extern int MatGetArray(Mat,Scalar **);
597b80b807SBarry Smith 
608ed539a5SBarry Smith extern int MatMult(Mat,Vec,Vec);
618ed539a5SBarry Smith extern int MatMultAdd(Mat,Vec,Vec,Vec);
628ed539a5SBarry Smith extern int MatMultTrans(Mat,Vec,Vec);
638ed539a5SBarry Smith extern int MatMultTransAdd(Mat,Vec,Vec,Vec);
642eac72dbSBarry Smith 
657b80b807SBarry Smith extern int MatConvert(Mat,MatType,Mat*);
667b80b807SBarry Smith extern int MatCopy(Mat,Mat);
677b80b807SBarry Smith extern int MatView(Mat,Viewer);
687b80b807SBarry Smith extern int MatLoad(Viewer,MatType,Mat*);
697b80b807SBarry Smith 
707b80b807SBarry Smith typedef enum {MAT_LOCAL=1,MAT_GLOBAL_MAX=2,MAT_GLOBAL_SUM=3} MatInfoType;
717b80b807SBarry Smith extern int MatGetInfo(Mat,MatInfoType,int*,int*,int*);
721e2fae70SLois Curfman McInnes extern int MatValidMatrix(Mat,int*);
737b80b807SBarry Smith extern int MatGetDiagonal(Mat,Vec);
74*8b3e568eSBarry Smith extern int MatTranspose(Mat,Mat*)
75*8b3e568eSBarry Smith extern int MatDiagonalScale(Mat,Vec,Vec)
76*8b3e568eSBarry Smith extern int MatDiagonalShift(Mat,Vec);
779ea5d5aeSSatish Balay extern int MatEqual(Mat,Mat, int*);
787b80b807SBarry Smith 
797b80b807SBarry Smith extern int MatNorm(Mat,NormType,double *);
807b80b807SBarry Smith extern int MatZeroEntries(Mat);
817b80b807SBarry Smith extern int MatZeroRows(Mat,IS,Scalar*);
827b80b807SBarry Smith extern int MatZeroColumns(Mat,IS,Scalar*);
837b80b807SBarry Smith 
847b80b807SBarry Smith extern int MatGetSize(Mat,int*,int*);
857b80b807SBarry Smith extern int MatGetLocalSize(Mat,int*,int*);
867b80b807SBarry Smith extern int MatGetOwnershipRange(Mat,int*,int*);
877b80b807SBarry Smith 
887b80b807SBarry Smith typedef enum {MAT_INITIAL_MATRIX, MAT_REUSE_MATRIX} MatGetSubMatrixCall;
897b80b807SBarry Smith extern int MatGetSubMatrix(Mat,IS,IS,MatGetSubMatrixCall,Mat*);
907b80b807SBarry Smith extern int MatGetSubMatrixInPlace(Mat,IS,IS);
917b80b807SBarry Smith extern int MatGetSubMatrices(Mat,int,IS *,IS *,MatGetSubMatrixCall,Mat **);
927b80b807SBarry Smith extern int MatIncreaseOverlap(Mat,int,IS *,int);
937b80b807SBarry Smith 
947b80b807SBarry Smith extern int MatAXPY(Scalar *,Mat,Mat);
957b80b807SBarry Smith extern int MatCompress(Mat);
967b80b807SBarry Smith 
97052efed2SBarry Smith extern int MatScale(Scalar *,Mat);
98052efed2SBarry Smith extern int MatShift(Scalar *,Mat);
99052efed2SBarry Smith 
1007b80b807SBarry Smith /* Routines unique to particular data structures */
1017b80b807SBarry Smith extern int MatBDiagGetData(Mat,int*,int*,int**,int**,Scalar***);
1027b80b807SBarry Smith 
1037b80b807SBarry Smith /*
1047b80b807SBarry Smith   These routines are not usually accessed directly, rather solving is
1057b80b807SBarry Smith   done through the SLES, KSP and PC interfaces.
1067b80b807SBarry Smith */
1077b80b807SBarry Smith 
10883f0b094SBarry Smith typedef enum {ORDER_NATURAL=0,ORDER_ND=1,ORDER_1WD=2,
109464493b3SBarry Smith               ORDER_RCM=3,ORDER_QMD=4,ORDER_APPLICATION_1,
110464493b3SBarry Smith               ORDER_APPLICATION_2} MatOrdering;
111ee50ffe9SBarry Smith extern int MatGetReordering(Mat,MatOrdering,IS*,IS*);
112d7b056bfSBarry Smith extern int MatGetReorderingTypeFromOptions(char *,MatOrdering*);
113832a8ecbSBarry Smith extern int MatReorderForNonzeroDiagonal(Mat,double,IS,IS);
114f26ada1bSBarry Smith extern int MatReorderingRegister(MatOrdering,char*,int (*)(int*,int*,int*,int*,int*));
115464493b3SBarry Smith extern int MatReorderingRegisterAll();
116464493b3SBarry Smith extern int MatReorderingRegisterDestroy();
1172eac72dbSBarry Smith 
11849d8b64dSBarry Smith extern int MatLUFactor(Mat,IS,IS,double);
119d7b056bfSBarry Smith extern int MatILUFactor(Mat,IS,IS,double,int);
12049d8b64dSBarry Smith extern int MatCholeskyFactor(Mat,IS,double);
12164051cb3SBarry Smith extern int MatLUFactorSymbolic(Mat,IS,IS,double,Mat*);
12264051cb3SBarry Smith extern int MatILUFactorSymbolic(Mat,IS,IS,double,int,Mat*);
12364051cb3SBarry Smith extern int MatCholeskyFactorSymbolic(Mat,IS,double,Mat*);
12464051cb3SBarry Smith extern int MatIncompleteCholeskyFactorSymbolic(Mat,IS,double,int,Mat*);
12520563c6bSBarry Smith extern int MatLUFactorNumeric(Mat,Mat*);
12620563c6bSBarry Smith extern int MatCholeskyFactorNumeric(Mat,Mat*);
1278ed539a5SBarry Smith 
1288ed539a5SBarry Smith extern int MatSolve(Mat,Vec,Vec);
12922169318SLois Curfman McInnes extern int MatForwardSolve(Mat,Vec,Vec);
13022169318SLois Curfman McInnes extern int MatBackwardSolve(Mat,Vec,Vec);
1318ed539a5SBarry Smith extern int MatSolveAdd(Mat,Vec,Vec,Vec);
13228988994SBarry Smith extern int MatSolveTrans(Mat,Vec,Vec);
13328988994SBarry Smith extern int MatSolveTransAdd(Mat,Vec,Vec,Vec);
1348ed539a5SBarry Smith 
135ee50ffe9SBarry Smith typedef enum {SOR_FORWARD_SWEEP=1,SOR_BACKWARD_SWEEP=2,SOR_SYMMETRIC_SWEEP=3,
136ee50ffe9SBarry Smith               SOR_LOCAL_FORWARD_SWEEP=4,SOR_LOCAL_BACKWARD_SWEEP=8,
137ee50ffe9SBarry Smith               SOR_LOCAL_SYMMETRIC_SWEEP=12,SOR_ZERO_INITIAL_GUESS=16,
138ca9b4cbeSLois Curfman McInnes               SOR_EISENSTAT=32,SOR_APPLY_UPPER=64,SOR_APPLY_LOWER=128
139ca9b4cbeSLois Curfman McInnes               } MatSORType;
140ca9b4cbeSLois Curfman McInnes extern int MatRelax(Mat,Vec,double,MatSORType,double,int,Vec);
1418ed539a5SBarry Smith 
142112a2221SBarry Smith typedef enum { MAT_GET_DIAGONAL=17 } MatOperation;
143112a2221SBarry Smith extern int MatHasOperation(Mat,MatOperation,PetscTruth*);
144112a2221SBarry Smith 
145037ecd73SBarry Smith /*  Not currently supported!
146037ecd73SBarry Smith #define MAT_SCATTER_COOKIE PETSC_COOKIE+15
147037ecd73SBarry Smith typedef struct _MatScatter* MatScatter;
148037ecd73SBarry Smith 
149037ecd73SBarry Smith extern int MatScatterBegin(Mat,Mat,InsertMode,MatScatter);
150037ecd73SBarry Smith extern int MatScatterEnd(Mat,Mat,InsertMode,MatScatter);
151037ecd73SBarry Smith extern int MatScatterCreate(Mat,IS,IS,Mat,IS,IS,MatScatter*);
152037ecd73SBarry Smith extern int MatScatterDestroy(MatScatter);
153037ecd73SBarry Smith */
1542eac72dbSBarry Smith 
1552eac72dbSBarry Smith #endif
1562eac72dbSBarry Smith 
1572eac72dbSBarry Smith 
1589d00d63dSBarry Smith 
159