xref: /petsc/include/petscmat.h (revision f0479e8c2158d05ab5c06def5bd82110ef658075)
12eac72dbSBarry Smith /*
22eac72dbSBarry Smith      Include file for the matrix component of PETSc
32eac72dbSBarry Smith */
41eb62cbbSBarry Smith #ifndef __MAT_PACKAGE
51eb62cbbSBarry Smith #define __MAT_PACKAGE
68ed539a5SBarry Smith #include "vec.h"
72eac72dbSBarry Smith 
8*f0479e8cSBarry Smith #define MAT_COOKIE 0x404040
9*f0479e8cSBarry Smith 
102eac72dbSBarry Smith typedef struct _Mat*           Mat;
112eac72dbSBarry Smith typedef struct _MatScatterCtx* MatScatterCtx;
122eac72dbSBarry Smith 
132eac72dbSBarry Smith 
148ed539a5SBarry Smith extern int MatCreateSequentialDense(int,int,Mat*);
15aabeff55SBarry Smith extern int MatCreateSequentialAIJ(int,int,int,int *,Mat*);
169717bf64SBarry Smith extern int MatCreateMPIAIJ(MPI_Comm,int,int,int,int,int,int*,int,int*,Mat*);
179717bf64SBarry Smith 
18aabeff55SBarry Smith extern int MatShellCreate(int,int,void *,Mat*);
19aabeff55SBarry Smith extern int MatShellSetMult(Mat,int (*)(void*,Vec,Vec));
20aabeff55SBarry Smith extern int MatShellSetMultTrans(Mat,int (*)(void*,Vec,Vec));
21*f0479e8cSBarry Smith extern int MatShellSetMultTransAdd(Mat,int (*)(void*,Vec,Vec,Vec));
222eac72dbSBarry Smith 
238ed539a5SBarry Smith /* ------------------------------------------------------------*/
248ed539a5SBarry Smith extern int  MatValidMatrix(Mat);
252eac72dbSBarry Smith 
2620563c6bSBarry Smith extern int MatSetValues(Mat,int,int*,int,int*,Scalar*,InsertMode);
278ed539a5SBarry Smith extern int MatBeginAssembly(Mat);
288ed539a5SBarry Smith extern int MatEndAssembly(Mat);
2920563c6bSBarry Smith extern int MatSetOption(Mat,int);
308ed539a5SBarry Smith #define ROW_ORIENTED              1
318ed539a5SBarry Smith #define COLUMN_ORIENTED           2
3220563c6bSBarry Smith #define ROWS_SORTED               4
3320563c6bSBarry Smith #define COLUMNS_SORTED            8
3420563c6bSBarry Smith #define NO_NEW_NONZERO_LOCATIONS  16
3520563c6bSBarry Smith #define YES_NEW_NONZERO_LOCATIONS 32
362eac72dbSBarry Smith 
378ed539a5SBarry Smith extern int MatGetValues(Mat,Scalar*,int,int*,int,int*);
388ed539a5SBarry Smith extern int MatGetRow(Mat,int,int *,int **,Scalar**);
398ed539a5SBarry Smith extern int MatRestoreRow(Mat,int,int *,int **,Scalar**);
408ed539a5SBarry Smith extern int MatGetCol(Mat,int,int *,int **,Scalar**);
418ed539a5SBarry Smith extern int MatRestoreCol(Mat,int,int *,int **,Scalar**);
422eac72dbSBarry Smith 
438ed539a5SBarry Smith extern int MatMult(Mat,Vec,Vec);
448ed539a5SBarry Smith extern int MatMultAdd(Mat,Vec,Vec,Vec);
458ed539a5SBarry Smith extern int MatMultTrans(Mat,Vec,Vec);
468ed539a5SBarry Smith extern int MatMultTransAdd(Mat,Vec,Vec,Vec);
472eac72dbSBarry Smith 
488ed539a5SBarry Smith #define ORDER_NATURAL 0
498ed539a5SBarry Smith #define ORDER_ND      1
508ed539a5SBarry Smith #define ORDER_1WD     2
518ed539a5SBarry Smith #define ORDER_RCM     3
528ed539a5SBarry Smith #define ORDER_QMD     4
538ed539a5SBarry Smith extern int MatGetReordering(Mat,int,IS*,IS*);
542eac72dbSBarry Smith 
558ed539a5SBarry Smith extern int MatLUFactor(Mat,IS,IS);
568ed539a5SBarry Smith extern int MatCholeskyFactor(Mat,IS);
578ed539a5SBarry Smith extern int MatLUFactorSymbolic(Mat,IS,IS,Mat*);
588ed539a5SBarry Smith extern int MatCholeskyFactorSymbolic(Mat,IS,Mat*);
5920563c6bSBarry Smith extern int MatLUFactorNumeric(Mat,Mat*);
6020563c6bSBarry Smith extern int MatCholeskyFactorNumeric(Mat,Mat*);
618ed539a5SBarry Smith 
628ed539a5SBarry Smith extern int MatSolve(Mat,Vec,Vec);
638ed539a5SBarry Smith extern int MatSolveAdd(Mat,Vec,Vec,Vec);
648ed539a5SBarry Smith extern int MatSolveTran(Mat,Vec,Vec);
658ed539a5SBarry Smith extern int MatSolveTranAdd(Mat,Vec,Vec,Vec);
668ed539a5SBarry Smith 
678ed539a5SBarry Smith #define SOR_FORWARD_SWEEP      1
688ed539a5SBarry Smith #define SOR_BACKWARD_SWEEP     2
698ed539a5SBarry Smith #define SOR_SYMMETRIC_SWEEP    3
708ed539a5SBarry Smith #define SOR_ZERO_INITIAL_GUESS 4
718ed539a5SBarry Smith extern int MatRelax(Mat,Vec,double,int,IS,int,Vec);
728ed539a5SBarry Smith 
738ed539a5SBarry Smith extern int MatCopy(Mat,Mat*);
748ed539a5SBarry Smith extern int MatView(Mat,Viewer);
758ed539a5SBarry Smith #include <stdio.h>
768ed539a5SBarry Smith extern int MatPrintMatlab(Mat,FILE*,char *);
778ed539a5SBarry Smith extern int MatNonZeros(Mat,int*);
788ed539a5SBarry Smith extern int MatMemoryUsed(Mat,int*);
798ed539a5SBarry Smith extern int MatGetDiagonal(Mat,Vec);
808ed539a5SBarry Smith extern int MatTranspose(Mat);
818ed539a5SBarry Smith extern int MatScale(Mat,Vec,Vec);
828ed539a5SBarry Smith extern int MatShrink(Mat,int,int*,int,int*);
838ed539a5SBarry Smith extern int MatEqual(Mat,Mat);
8420563c6bSBarry Smith extern int MatScatterBegin(Mat,IS,IS,Mat,IS,IS,InsertMode,MatScatterCtx*);
8520563c6bSBarry Smith extern int MatScatterEnd(Mat,IS,IS,Mat,IS,IS,InsertMode,MatScatterCtx*);
868ed539a5SBarry Smith extern int MatReOrder(Mat,IS,IS);
872eac72dbSBarry Smith 
882eac72dbSBarry Smith #define NORM_1         1
892eac72dbSBarry Smith #define NORM_2         2
902eac72dbSBarry Smith #define NORM_FROBENIUS 3
912eac72dbSBarry Smith #define NORM_INFINITY  4
928ed539a5SBarry Smith extern int MatNorm(Mat,int,double *);
932eac72dbSBarry Smith 
948ed539a5SBarry Smith extern int MatZeroEntries(Mat);
9520563c6bSBarry Smith extern int MatZeroRows(Mat,IS,Scalar*);
9620563c6bSBarry Smith extern int MatZeroColumns(Mat,IS,Scalar*);
978ed539a5SBarry Smith 
988ed539a5SBarry Smith extern int MatCompress(Mat);
998ed539a5SBarry Smith extern int MatDestroy(Mat);
1008ed539a5SBarry Smith 
1011eb62cbbSBarry Smith extern int MatGetSize(Mat,int*,int*);
1021eb62cbbSBarry Smith extern int MatGetLocalSize(Mat,int*,int*);
1031eb62cbbSBarry Smith 
1048ed539a5SBarry Smith extern int MatCreateInitialMatrix(int,int,Mat*);
1058ed539a5SBarry Smith 
1068ed539a5SBarry Smith 
1078ed539a5SBarry Smith 
1081eb62cbbSBarry Smith 
1092eac72dbSBarry Smith #endif
1102eac72dbSBarry Smith 
1112eac72dbSBarry Smith 
112