xref: /petsc/include/petscmat.h (revision 64e87e97507246f9229332c68ca2aa6070f62830)
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 
89cd28387SBarry Smith #define MAT_COOKIE PETSC_COOKIE+5
9f0479e8cSBarry Smith 
102eac72dbSBarry Smith typedef struct _Mat*           Mat;
112eac72dbSBarry Smith typedef struct _MatScatterCtx* MatScatterCtx;
122eac72dbSBarry Smith 
132eac72dbSBarry Smith 
149cd28387SBarry Smith #define MATDENSESEQ  0
159cd28387SBarry Smith #define MATAIJSEQ    1
169cd28387SBarry Smith #define MATAIJMPI    2
179cd28387SBarry Smith #define MATSHELL     3
18d7692543SLois Curfman McInnes #define MATROWSEQ    4
19d7692543SLois Curfman McInnes #define MATROWMPI    5
2028988994SBarry Smith 
218ed539a5SBarry Smith extern int MatCreateSequentialDense(int,int,Mat*);
22aabeff55SBarry Smith extern int MatCreateSequentialAIJ(int,int,int,int *,Mat*);
239717bf64SBarry Smith extern int MatCreateMPIAIJ(MPI_Comm,int,int,int,int,int,int*,int,int*,Mat*);
24d7692543SLois Curfman McInnes extern int MatCreateSequentialRow(int,int,int,int *,Mat*);
25d7692543SLois Curfman McInnes extern int MatCreateMPIRow(MPI_Comm,int,int,int,int,int,int*,int,int*,Mat*);
269717bf64SBarry Smith 
27aabeff55SBarry Smith extern int MatShellCreate(int,int,void *,Mat*);
28aabeff55SBarry Smith extern int MatShellSetMult(Mat,int (*)(void*,Vec,Vec));
29aabeff55SBarry Smith extern int MatShellSetMultTrans(Mat,int (*)(void*,Vec,Vec));
30f0479e8cSBarry Smith extern int MatShellSetMultTransAdd(Mat,int (*)(void*,Vec,Vec,Vec));
312eac72dbSBarry Smith 
328ed539a5SBarry Smith /* ------------------------------------------------------------*/
338ed539a5SBarry Smith extern int  MatValidMatrix(Mat);
342eac72dbSBarry Smith 
3520563c6bSBarry Smith extern int MatSetValues(Mat,int,int*,int,int*,Scalar*,InsertMode);
368ed539a5SBarry Smith extern int MatBeginAssembly(Mat);
378ed539a5SBarry Smith extern int MatEndAssembly(Mat);
3820563c6bSBarry Smith extern int MatSetOption(Mat,int);
398ed539a5SBarry Smith #define ROW_ORIENTED              1
408ed539a5SBarry Smith #define COLUMN_ORIENTED           2
4120563c6bSBarry Smith #define ROWS_SORTED               4
4220563c6bSBarry Smith #define COLUMNS_SORTED            8
4320563c6bSBarry Smith #define NO_NEW_NONZERO_LOCATIONS  16
4420563c6bSBarry Smith #define YES_NEW_NONZERO_LOCATIONS 32
452eac72dbSBarry Smith 
468ed539a5SBarry Smith extern int MatGetValues(Mat,Scalar*,int,int*,int,int*);
478ed539a5SBarry Smith extern int MatGetRow(Mat,int,int *,int **,Scalar**);
488ed539a5SBarry Smith extern int MatRestoreRow(Mat,int,int *,int **,Scalar**);
498ed539a5SBarry Smith extern int MatGetCol(Mat,int,int *,int **,Scalar**);
508ed539a5SBarry Smith extern int MatRestoreCol(Mat,int,int *,int **,Scalar**);
51*64e87e97SBarry Smith extern int MatGetArray(Mat,Scalar **);
528ed539a5SBarry Smith extern int MatMult(Mat,Vec,Vec);
538ed539a5SBarry Smith extern int MatMultAdd(Mat,Vec,Vec,Vec);
548ed539a5SBarry Smith extern int MatMultTrans(Mat,Vec,Vec);
558ed539a5SBarry Smith extern int MatMultTransAdd(Mat,Vec,Vec,Vec);
562eac72dbSBarry Smith 
578ed539a5SBarry Smith #define ORDER_NATURAL 0
588ed539a5SBarry Smith #define ORDER_ND      1
598ed539a5SBarry Smith #define ORDER_1WD     2
608ed539a5SBarry Smith #define ORDER_RCM     3
618ed539a5SBarry Smith #define ORDER_QMD     4
628ed539a5SBarry Smith extern int MatGetReordering(Mat,int,IS*,IS*);
632eac72dbSBarry Smith 
648ed539a5SBarry Smith extern int MatLUFactor(Mat,IS,IS);
658ed539a5SBarry Smith extern int MatCholeskyFactor(Mat,IS);
668ed539a5SBarry Smith extern int MatLUFactorSymbolic(Mat,IS,IS,Mat*);
67da3a660dSBarry Smith extern int MatILUFactorSymbolic(Mat,IS,IS,int,Mat*);
688ed539a5SBarry Smith extern int MatCholeskyFactorSymbolic(Mat,IS,Mat*);
69da3a660dSBarry Smith extern int MatIncompleteCholeskyFactorSymbolic(Mat,IS,int,Mat*);
7020563c6bSBarry Smith extern int MatLUFactorNumeric(Mat,Mat*);
7120563c6bSBarry Smith extern int MatCholeskyFactorNumeric(Mat,Mat*);
728ed539a5SBarry Smith 
738ed539a5SBarry Smith extern int MatSolve(Mat,Vec,Vec);
748ed539a5SBarry Smith extern int MatSolveAdd(Mat,Vec,Vec,Vec);
7528988994SBarry Smith extern int MatSolveTrans(Mat,Vec,Vec);
7628988994SBarry Smith extern int MatSolveTransAdd(Mat,Vec,Vec,Vec);
778ed539a5SBarry Smith 
788ed539a5SBarry Smith #define SOR_FORWARD_SWEEP            1
798ed539a5SBarry Smith #define SOR_BACKWARD_SWEEP           2
808ed539a5SBarry Smith #define SOR_SYMMETRIC_SWEEP          3
81d6dfbf8fSBarry Smith #define SOR_LOCAL_FORWARD_SWEEP      4
82d6dfbf8fSBarry Smith #define SOR_LOCAL_BACKWARD_SWEEP     8
83d6dfbf8fSBarry Smith #define SOR_LOCAL_SYMMETRIC_SWEEP    12
84d6dfbf8fSBarry Smith #define SOR_ZERO_INITIAL_GUESS       16
85da3a660dSBarry Smith #define SOR_EISENSTAT                32
869cd28387SBarry Smith #define SOR_APPLY_UPPER              64
879cd28387SBarry Smith #define SOR_APPLY_LOWER              128
88d6dfbf8fSBarry Smith extern int MatRelax(Mat,Vec,double,int,double,int,Vec);
898ed539a5SBarry Smith 
908ed539a5SBarry Smith extern int MatCopy(Mat,Mat*);
918ed539a5SBarry Smith extern int MatView(Mat,Viewer);
928ed539a5SBarry Smith #include <stdio.h>
938ed539a5SBarry Smith extern int MatPrintMatlab(Mat,FILE*,char *);
948ed539a5SBarry Smith extern int MatNonZeros(Mat,int*);
958ed539a5SBarry Smith extern int MatMemoryUsed(Mat,int*);
968ed539a5SBarry Smith extern int MatGetDiagonal(Mat,Vec);
978ed539a5SBarry Smith extern int MatTranspose(Mat);
988ed539a5SBarry Smith extern int MatScale(Mat,Vec,Vec);
998ed539a5SBarry Smith extern int MatShrink(Mat,int,int*,int,int*);
1008ed539a5SBarry Smith extern int MatEqual(Mat,Mat);
10120563c6bSBarry Smith extern int MatScatterBegin(Mat,IS,IS,Mat,IS,IS,InsertMode,MatScatterCtx*);
10220563c6bSBarry Smith extern int MatScatterEnd(Mat,IS,IS,Mat,IS,IS,InsertMode,MatScatterCtx*);
1038ed539a5SBarry Smith extern int MatReOrder(Mat,IS,IS);
1042eac72dbSBarry Smith 
1052eac72dbSBarry Smith #define NORM_1         1
1062eac72dbSBarry Smith #define NORM_2         2
1072eac72dbSBarry Smith #define NORM_FROBENIUS 3
1082eac72dbSBarry Smith #define NORM_INFINITY  4
1098ed539a5SBarry Smith extern int MatNorm(Mat,int,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 MatCompress(Mat);
1168ed539a5SBarry Smith extern int MatDestroy(Mat);
1178ed539a5SBarry Smith 
1181eb62cbbSBarry Smith extern int MatGetSize(Mat,int*,int*);
1191eb62cbbSBarry Smith extern int MatGetLocalSize(Mat,int*,int*);
120d6dfbf8fSBarry Smith extern int MatGetOwnershipRange(Mat,int*,int*);
1211eb62cbbSBarry Smith 
1228ed539a5SBarry Smith extern int MatCreateInitialMatrix(int,int,Mat*);
1238ed539a5SBarry Smith 
1248ed539a5SBarry Smith 
1258ed539a5SBarry Smith 
1261eb62cbbSBarry Smith 
1272eac72dbSBarry Smith #endif
1282eac72dbSBarry Smith 
1292eac72dbSBarry Smith 
130