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