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