xref: /petsc/include/petscmat.h (revision dae03382a1045a5f522b2689cebebda3f56b9289)
1*dae03382SLois Curfman McInnes /* $Id: mat.h,v 1.43 1995/07/09 23:17:55 bsmith Exp curfman $ */
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
10b7e13cc3SBarry Smith #define MAT_SCATTER_COOKIE PETSC_COOKIE+15
11f0479e8cSBarry Smith 
122eac72dbSBarry Smith typedef struct _Mat*           Mat;
132eac72dbSBarry Smith typedef struct _MatScatterCtx* MatScatterCtx;
142eac72dbSBarry Smith 
154af0de53SLois Curfman McInnes typedef enum { MATSAME=-1, MATDENSE, MATAIJ, MATMPIAIJ, MATSHELL, MATROW,
16ee50ffe9SBarry Smith                MATMPIROW, MATMPIROW_BS, MATBDIAG, MATMPIBDIAG } MatType;
1728988994SBarry Smith 
186b5873e3SBarry Smith extern int MatCreateSequentialDense(MPI_Comm,int,int,Mat*);
196b5873e3SBarry Smith extern int MatCreateSequentialAIJ(MPI_Comm,int,int,int,int *,Mat*);
209717bf64SBarry Smith extern int MatCreateMPIAIJ(MPI_Comm,int,int,int,int,int,int*,int,int*,Mat*);
216b5873e3SBarry Smith extern int MatCreateSequentialRow(MPI_Comm,int,int,int,int *,Mat*);
22d7692543SLois Curfman McInnes extern int MatCreateMPIRow(MPI_Comm,int,int,int,int,int,int*,int,int*,Mat*);
23cfe223acSLois Curfman McInnes extern int MatCreateMPIRowbs(MPI_Comm,int,int,int,int*,void*,Mat*);
24fdb36d4aSLois Curfman McInnes extern int MatCreateSequentialBDiag(MPI_Comm,int,int,int,int,int*,
25fdb36d4aSLois Curfman McInnes                                     Scalar**,Mat*);
26f6f1fd9dSLois Curfman McInnes extern int MatCreateMPIBDiag(MPI_Comm,int,int,int,int,int,int*,
27f6f1fd9dSLois Curfman McInnes                                     Scalar**,Mat*);
289717bf64SBarry Smith 
296b5873e3SBarry Smith extern int MatShellCreate(MPI_Comm,int,int,void *,Mat*);
30aabeff55SBarry Smith extern int MatShellSetMult(Mat,int (*)(void*,Vec,Vec));
31aabeff55SBarry Smith extern int MatShellSetMultTrans(Mat,int (*)(void*,Vec,Vec));
32f0479e8cSBarry Smith extern int MatShellSetMultTransAdd(Mat,int (*)(void*,Vec,Vec,Vec));
332eac72dbSBarry Smith 
348ed539a5SBarry Smith /* ------------------------------------------------------------*/
358ed539a5SBarry Smith extern int  MatValidMatrix(Mat);
362eac72dbSBarry Smith 
37ee50ffe9SBarry Smith typedef enum {FLUSH_ASSEMBLY=1,FINAL_ASSEMBLY=0} MatAssemblyType;
382eac72dbSBarry Smith 
39ee50ffe9SBarry Smith extern int MatSetValues(Mat,int,int*,int,int*,Scalar*,InsertMode);
40ee50ffe9SBarry Smith extern int MatAssemblyBegin(Mat,MatAssemblyType);
41ee50ffe9SBarry Smith extern int MatAssemblyEnd(Mat,MatAssemblyType);
42ee50ffe9SBarry Smith 
43ee50ffe9SBarry Smith typedef enum {ROW_ORIENTED=1,COLUMN_ORIENTED=2,ROWS_SORTED=4,
44ee50ffe9SBarry Smith               COLUMNS_SORTED=8,NO_NEW_NONZERO_LOCATIONS=16,
45ca9b4cbeSLois Curfman McInnes               YES_NEW_NONZERO_LOCATIONS=32} MatOption;
46ee50ffe9SBarry Smith 
47dbd7a890SLois Curfman McInnes extern int MatGetBDiagData(Mat,int*,int*,int**,Scalar***);
48dbd7a890SLois Curfman McInnes 
49ee50ffe9SBarry Smith extern int MatSetOption(Mat,MatOption);
50ee50ffe9SBarry Smith 
51dbd7a890SLois Curfman McInnes extern int MatGetType(Mat,MatType*);
52*dae03382SLois Curfman McInnes extern int MatGetName(Mat,char**);
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 **);
598ed539a5SBarry Smith extern int MatMult(Mat,Vec,Vec);
608ed539a5SBarry Smith extern int MatMultAdd(Mat,Vec,Vec,Vec);
618ed539a5SBarry Smith extern int MatMultTrans(Mat,Vec,Vec);
628ed539a5SBarry Smith extern int MatMultTransAdd(Mat,Vec,Vec,Vec);
632eac72dbSBarry Smith 
641f8ca976SLois Curfman McInnes typedef enum {ORDER_DATABASE=-1,ORDER_NATURAL=0,ORDER_ND=1,ORDER_1WD=2,
651f8ca976SLois Curfman McInnes               ORDER_RCM=3,ORDER_QMD=4} MatOrdering;
66ee50ffe9SBarry Smith 
67ee50ffe9SBarry Smith extern int MatGetReordering(Mat,MatOrdering,IS*,IS*);
68832a8ecbSBarry Smith extern int MatReorderForNonzeroDiagonal(Mat,double,IS,IS);
692eac72dbSBarry Smith 
7049d8b64dSBarry Smith extern int MatLUFactor(Mat,IS,IS,double);
7149d8b64dSBarry Smith extern int MatCholeskyFactor(Mat,IS,double);
7264051cb3SBarry Smith extern int MatLUFactorSymbolic(Mat,IS,IS,double,Mat*);
7364051cb3SBarry Smith extern int MatILUFactorSymbolic(Mat,IS,IS,double,int,Mat*);
7464051cb3SBarry Smith extern int MatCholeskyFactorSymbolic(Mat,IS,double,Mat*);
7564051cb3SBarry Smith extern int MatIncompleteCholeskyFactorSymbolic(Mat,IS,double,int,Mat*);
7620563c6bSBarry Smith extern int MatLUFactorNumeric(Mat,Mat*);
7720563c6bSBarry Smith extern int MatCholeskyFactorNumeric(Mat,Mat*);
788ed539a5SBarry Smith 
798ed539a5SBarry Smith extern int MatSolve(Mat,Vec,Vec);
808ed539a5SBarry Smith extern int MatSolveAdd(Mat,Vec,Vec,Vec);
8128988994SBarry Smith extern int MatSolveTrans(Mat,Vec,Vec);
8228988994SBarry Smith extern int MatSolveTransAdd(Mat,Vec,Vec,Vec);
838ed539a5SBarry Smith 
84ee50ffe9SBarry Smith typedef enum {SOR_FORWARD_SWEEP=1,SOR_BACKWARD_SWEEP=2,SOR_SYMMETRIC_SWEEP=3,
85ee50ffe9SBarry Smith               SOR_LOCAL_FORWARD_SWEEP=4,SOR_LOCAL_BACKWARD_SWEEP=8,
86ee50ffe9SBarry Smith               SOR_LOCAL_SYMMETRIC_SWEEP=12,SOR_ZERO_INITIAL_GUESS=16,
87ca9b4cbeSLois Curfman McInnes               SOR_EISENSTAT=32,SOR_APPLY_UPPER=64,SOR_APPLY_LOWER=128
88ca9b4cbeSLois Curfman McInnes               } MatSORType;
89ee50ffe9SBarry Smith 
90ca9b4cbeSLois Curfman McInnes extern int MatRelax(Mat,Vec,double,MatSORType,double,int,Vec);
918ed539a5SBarry Smith 
92ca9b4cbeSLois Curfman McInnes extern int MatConvert(Mat,MatType,Mat*);
938ed539a5SBarry Smith extern int MatView(Mat,Viewer);
948ed539a5SBarry Smith #include <stdio.h>
9596c0e93cSLois Curfman McInnes 
96ee50ffe9SBarry Smith typedef enum {MAT_LOCAL=1,MAT_GLOBAL_MAX=2,MAT_GLOBAL_SUM=3} MatInfoType;
9796c0e93cSLois Curfman McInnes 
98ee50ffe9SBarry Smith extern int MatGetInfo(Mat,MatInfoType,int*,int*,int*);
998ed539a5SBarry Smith extern int MatGetDiagonal(Mat,Vec);
100b7e13cc3SBarry Smith extern int MatTranspose(Mat,Mat*);
1018ed539a5SBarry Smith extern int MatScale(Mat,Vec,Vec);
1028ed539a5SBarry Smith extern int MatEqual(Mat,Mat);
103b7e13cc3SBarry Smith extern int MatScatterBegin(Mat,Mat,InsertMode,MatScatterCtx);
104b7e13cc3SBarry Smith extern int MatScatterEnd(Mat,Mat,InsertMode,MatScatterCtx);
105b7e13cc3SBarry Smith extern int MatScatterCtxCreate(Mat,IS,IS,Mat,IS,IS,MatScatterCtx*);
106b7e13cc3SBarry Smith extern int MatScatterCtxDestroy(MatScatterCtx);
1072eac72dbSBarry Smith 
108ee50ffe9SBarry Smith typedef enum {NORM_1=1,NORM_2=2,NORM_FROBENIUS=3,NORM_INFINITY=4} MatNormType;
109ee50ffe9SBarry Smith extern int MatNorm(Mat,MatNormType,double *);
1102eac72dbSBarry Smith 
1118ed539a5SBarry Smith extern int MatZeroEntries(Mat);
11220563c6bSBarry Smith extern int MatZeroRows(Mat,IS,Scalar*);
11320563c6bSBarry Smith extern int MatZeroColumns(Mat,IS,Scalar*);
1148ed539a5SBarry Smith 
1158ed539a5SBarry Smith extern int MatDestroy(Mat);
1168ed539a5SBarry Smith 
1171eb62cbbSBarry Smith extern int MatGetSize(Mat,int*,int*);
1181eb62cbbSBarry Smith extern int MatGetLocalSize(Mat,int*,int*);
119d6dfbf8fSBarry Smith extern int MatGetOwnershipRange(Mat,int*,int*);
1201eb62cbbSBarry Smith 
1216469c4f9SBarry Smith extern int MatCreate(MPI_Comm,int,int,Mat*);
1228ed539a5SBarry Smith 
1236eafb6e7SBarry Smith extern int MatGetSubMatrix(Mat,IS,IS,Mat*);
1246eafb6e7SBarry Smith extern int MatGetSubMatrixInPlace(Mat,IS,IS);
1250357936bSBarry Smith 
1260357936bSBarry Smith extern int MatAXPY(Scalar *,Mat,Mat);
127c804fac0SBarry Smith 
1282eac72dbSBarry Smith #endif
1292eac72dbSBarry Smith 
1302eac72dbSBarry Smith 
1319d00d63dSBarry Smith 
132