1*71aaadd7SBarry Smith /* $Id: mat.h,v 1.97 1996/03/07 01:43:07 bsmith 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 **); 5909883261SBarry Smith extern int MatRestoreArray(Mat,Scalar **); 607b80b807SBarry Smith 618ed539a5SBarry Smith extern int MatMult(Mat,Vec,Vec); 628ed539a5SBarry Smith extern int MatMultAdd(Mat,Vec,Vec,Vec); 638ed539a5SBarry Smith extern int MatMultTrans(Mat,Vec,Vec); 648ed539a5SBarry Smith extern int MatMultTransAdd(Mat,Vec,Vec,Vec); 652eac72dbSBarry Smith 667b80b807SBarry Smith extern int MatConvert(Mat,MatType,Mat*); 677b80b807SBarry Smith extern int MatCopy(Mat,Mat); 687b80b807SBarry Smith extern int MatView(Mat,Viewer); 697b80b807SBarry Smith extern int MatLoad(Viewer,MatType,Mat*); 707b80b807SBarry Smith 717b80b807SBarry Smith typedef enum {MAT_LOCAL=1,MAT_GLOBAL_MAX=2,MAT_GLOBAL_SUM=3} MatInfoType; 727b80b807SBarry Smith extern int MatGetInfo(Mat,MatInfoType,int*,int*,int*); 731e2fae70SLois Curfman McInnes extern int MatValidMatrix(Mat,int*); 747b80b807SBarry Smith extern int MatGetDiagonal(Mat,Vec); 75cd8486b4SBarry Smith extern int MatTranspose(Mat,Mat*); 76f4c33d5aSBarry Smith extern int MatDiagonalScale(Mat,Vec,Vec); 778b3e568eSBarry Smith extern int MatDiagonalShift(Mat,Vec); 789ea5d5aeSSatish Balay extern int MatEqual(Mat,Mat, int*); 797b80b807SBarry Smith 807b80b807SBarry Smith extern int MatNorm(Mat,NormType,double *); 817b80b807SBarry Smith extern int MatZeroEntries(Mat); 827b80b807SBarry Smith extern int MatZeroRows(Mat,IS,Scalar*); 837b80b807SBarry Smith extern int MatZeroColumns(Mat,IS,Scalar*); 847b80b807SBarry Smith 857b80b807SBarry Smith extern int MatGetSize(Mat,int*,int*); 867b80b807SBarry Smith extern int MatGetLocalSize(Mat,int*,int*); 877b80b807SBarry Smith extern int MatGetOwnershipRange(Mat,int*,int*); 887b80b807SBarry Smith 897b80b807SBarry Smith typedef enum {MAT_INITIAL_MATRIX, MAT_REUSE_MATRIX} MatGetSubMatrixCall; 907b80b807SBarry Smith extern int MatGetSubMatrix(Mat,IS,IS,MatGetSubMatrixCall,Mat*); 917b80b807SBarry Smith extern int MatGetSubMatrixInPlace(Mat,IS,IS); 927b80b807SBarry Smith extern int MatGetSubMatrices(Mat,int,IS *,IS *,MatGetSubMatrixCall,Mat **); 937b80b807SBarry Smith extern int MatIncreaseOverlap(Mat,int,IS *,int); 947b80b807SBarry Smith 957b80b807SBarry Smith extern int MatAXPY(Scalar *,Mat,Mat); 967b80b807SBarry Smith extern int MatCompress(Mat); 977b80b807SBarry Smith 98052efed2SBarry Smith extern int MatScale(Scalar *,Mat); 99052efed2SBarry Smith extern int MatShift(Scalar *,Mat); 100052efed2SBarry Smith 1017b80b807SBarry Smith /* Routines unique to particular data structures */ 1027b80b807SBarry Smith extern int MatBDiagGetData(Mat,int*,int*,int**,int**,Scalar***); 1037b80b807SBarry Smith 1047b80b807SBarry Smith /* 1057b80b807SBarry Smith These routines are not usually accessed directly, rather solving is 1067b80b807SBarry Smith done through the SLES, KSP and PC interfaces. 1077b80b807SBarry Smith */ 1087b80b807SBarry Smith 10983f0b094SBarry Smith typedef enum {ORDER_NATURAL=0,ORDER_ND=1,ORDER_1WD=2, 11009883261SBarry Smith ORDER_RCM=3,ORDER_QMD=4,ORDER_ROWLENGTH=5,ORDER_APPLICATION_1, 111464493b3SBarry Smith ORDER_APPLICATION_2} MatOrdering; 112ee50ffe9SBarry Smith extern int MatGetReordering(Mat,MatOrdering,IS*,IS*); 113d7b056bfSBarry Smith extern int MatGetReorderingTypeFromOptions(char *,MatOrdering*); 114832a8ecbSBarry Smith extern int MatReorderForNonzeroDiagonal(Mat,double,IS,IS); 115*71aaadd7SBarry Smith extern int MatReorderingRegister(MatOrdering *,char*,PetscTruth,int, 11609883261SBarry Smith int (*)(int*,int*,int*,int*,int*)); 117464493b3SBarry Smith extern int MatReorderingRegisterAll(); 118464493b3SBarry Smith extern int MatReorderingRegisterDestroy(); 11909883261SBarry Smith extern int MatReorderingGetName(MatOrdering,char **); 12009883261SBarry Smith extern PetscTruth MatReorderingRequiresSymmetric[]; 12109883261SBarry Smith extern int MatReorderingIndexShift[]; 1222eac72dbSBarry Smith 12349d8b64dSBarry Smith extern int MatLUFactor(Mat,IS,IS,double); 124d7b056bfSBarry Smith extern int MatILUFactor(Mat,IS,IS,double,int); 12549d8b64dSBarry Smith extern int MatCholeskyFactor(Mat,IS,double); 12664051cb3SBarry Smith extern int MatLUFactorSymbolic(Mat,IS,IS,double,Mat*); 12764051cb3SBarry Smith extern int MatILUFactorSymbolic(Mat,IS,IS,double,int,Mat*); 12864051cb3SBarry Smith extern int MatCholeskyFactorSymbolic(Mat,IS,double,Mat*); 12964051cb3SBarry Smith extern int MatIncompleteCholeskyFactorSymbolic(Mat,IS,double,int,Mat*); 13020563c6bSBarry Smith extern int MatLUFactorNumeric(Mat,Mat*); 13120563c6bSBarry Smith extern int MatCholeskyFactorNumeric(Mat,Mat*); 1328ed539a5SBarry Smith 1338ed539a5SBarry Smith extern int MatSolve(Mat,Vec,Vec); 13422169318SLois Curfman McInnes extern int MatForwardSolve(Mat,Vec,Vec); 13522169318SLois Curfman McInnes extern int MatBackwardSolve(Mat,Vec,Vec); 1368ed539a5SBarry Smith extern int MatSolveAdd(Mat,Vec,Vec,Vec); 13728988994SBarry Smith extern int MatSolveTrans(Mat,Vec,Vec); 13828988994SBarry Smith extern int MatSolveTransAdd(Mat,Vec,Vec,Vec); 1398ed539a5SBarry Smith 140ee50ffe9SBarry Smith typedef enum {SOR_FORWARD_SWEEP=1,SOR_BACKWARD_SWEEP=2,SOR_SYMMETRIC_SWEEP=3, 141ee50ffe9SBarry Smith SOR_LOCAL_FORWARD_SWEEP=4,SOR_LOCAL_BACKWARD_SWEEP=8, 142ee50ffe9SBarry Smith SOR_LOCAL_SYMMETRIC_SWEEP=12,SOR_ZERO_INITIAL_GUESS=16, 143ca9b4cbeSLois Curfman McInnes SOR_EISENSTAT=32,SOR_APPLY_UPPER=64,SOR_APPLY_LOWER=128 144ca9b4cbeSLois Curfman McInnes } MatSORType; 145ca9b4cbeSLois Curfman McInnes extern int MatRelax(Mat,Vec,double,MatSORType,double,int,Vec); 1468ed539a5SBarry Smith 147112a2221SBarry Smith typedef enum { MAT_GET_DIAGONAL=17 } MatOperation; 148112a2221SBarry Smith extern int MatHasOperation(Mat,MatOperation,PetscTruth*); 149112a2221SBarry Smith 150037ecd73SBarry Smith /* Not currently supported! 151037ecd73SBarry Smith #define MAT_SCATTER_COOKIE PETSC_COOKIE+15 152037ecd73SBarry Smith typedef struct _MatScatter* MatScatter; 153037ecd73SBarry Smith 154037ecd73SBarry Smith extern int MatScatterBegin(Mat,Mat,InsertMode,MatScatter); 155037ecd73SBarry Smith extern int MatScatterEnd(Mat,Mat,InsertMode,MatScatter); 156037ecd73SBarry Smith extern int MatScatterCreate(Mat,IS,IS,Mat,IS,IS,MatScatter*); 157037ecd73SBarry Smith extern int MatScatterDestroy(MatScatter); 158037ecd73SBarry Smith */ 1592eac72dbSBarry Smith 1602eac72dbSBarry Smith #endif 1612eac72dbSBarry Smith 1622eac72dbSBarry Smith 1639d00d63dSBarry Smith 164