petsc/include/petscmat.h

*de7da479SBarry Smith/* $Id: mat.h,v 1.60 1995/09/07 22:37:32 bsmith Exp bsmith $ */
2eac72dbSBarry Smith/*
2eac72dbSBarry Smith     Include file for the matrix component of PETSc
2eac72dbSBarry Smith*/
1eb62cbbSBarry Smith#ifndef __MAT_PACKAGE
1eb62cbbSBarry Smith#define __MAT_PACKAGE
8ed539a5SBarry Smith#include "vec.h"
2eac72dbSBarry Smith
9cd28387SBarry Smith#define MAT_COOKIE         PETSC_COOKIE+5
b7e13cc3SBarry Smith#define MAT_SCATTER_COOKIE PETSC_COOKIE+15
f0479e8cSBarry Smith
2eac72dbSBarry Smithtypedef struct _Mat*           Mat;
2eac72dbSBarry Smithtypedef struct _MatScatterCtx* MatScatterCtx;
2eac72dbSBarry Smith
4af0de53SLois Curfman McInnestypedef enum { MATSAME=-1, MATDENSE, MATAIJ, MATMPIAIJ, MATSHELL, MATROW,
ee50ffe9SBarry Smith               MATMPIROW, MATMPIROW_BS, MATBDIAG, MATMPIBDIAG } MatType;
28988994SBarry Smith
*de7da479SBarry Smithextern int MatCreateSeqDense(MPI_Comm,int,int,Mat*);
*de7da479SBarry Smithextern int MatCreateSeqAIJ(MPI_Comm,int,int,int,int *,Mat*);
9717bf64SBarry Smithextern int MatCreateMPIAIJ(MPI_Comm,int,int,int,int,int,int*,int,int*,Mat*);
*de7da479SBarry Smithextern int MatCreateSeqRow(MPI_Comm,int,int,int,int *,Mat*);
d7692543SLois Curfman McInnesextern int MatCreateMPIRow(MPI_Comm,int,int,int,int,int,int*,int,int*,Mat*);
cfe223acSLois Curfman McInnesextern int MatCreateMPIRowbs(MPI_Comm,int,int,int,int*,void*,Mat*);
*de7da479SBarry Smithextern int MatCreateSeqBDiag(MPI_Comm,int,int,int,int,int*,
fdb36d4aSLois Curfman McInnes                                    Scalar**,Mat*);
f6f1fd9dSLois Curfman McInnesextern int MatCreateMPIBDiag(MPI_Comm,int,int,int,int,int,int*,
f6f1fd9dSLois Curfman McInnes                                    Scalar**,Mat*);
9717bf64SBarry Smith
6b5873e3SBarry Smithextern int MatShellCreate(MPI_Comm,int,int,void *,Mat*);
aabeff55SBarry Smithextern int MatShellSetMult(Mat,int (*)(void*,Vec,Vec));
b9fa9cd0SBarry Smithextern int MatShellSetDestroy(Mat,int (*)(void*));
aabeff55SBarry Smithextern int MatShellSetMultTrans(Mat,int (*)(void*,Vec,Vec));
f0479e8cSBarry Smithextern int MatShellSetMultTransAdd(Mat,int (*)(void*,Vec,Vec,Vec));
2eac72dbSBarry Smith
8ed539a5SBarry Smith/* ------------------------------------------------------------*/
8ed539a5SBarry Smithextern int  MatValidMatrix(Mat);
2eac72dbSBarry Smith
ee50ffe9SBarry Smithtypedef enum {FLUSH_ASSEMBLY=1,FINAL_ASSEMBLY=0} MatAssemblyType;
2eac72dbSBarry Smith
ee50ffe9SBarry Smithextern int MatSetValues(Mat,int,int*,int,int*,Scalar*,InsertMode);
ee50ffe9SBarry Smithextern int MatAssemblyBegin(Mat,MatAssemblyType);
ee50ffe9SBarry Smithextern int MatAssemblyEnd(Mat,MatAssemblyType);
ee50ffe9SBarry Smith
ee50ffe9SBarry Smithtypedef enum {ROW_ORIENTED=1,COLUMN_ORIENTED=2,ROWS_SORTED=4,
ee50ffe9SBarry Smith              COLUMNS_SORTED=8,NO_NEW_NONZERO_LOCATIONS=16,
69fb8b30SBarry Smith              YES_NEW_NONZERO_LOCATIONS=32,SYMMETRIC_MATRIX=64} MatOption;
ee50ffe9SBarry Smith
ee50ffe9SBarry Smithextern int MatSetOption(Mat,MatOption);
dbd7a890SLois Curfman McInnesextern int MatGetType(Mat,MatType*);
dae03382SLois Curfman McInnesextern int MatGetName(Mat,char**);
ee50ffe9SBarry Smithextern int MatGetValues(Mat,int,int*,int,int*,Scalar*);
8ed539a5SBarry Smithextern int MatGetRow(Mat,int,int *,int **,Scalar**);
8ed539a5SBarry Smithextern int MatRestoreRow(Mat,int,int *,int **,Scalar**);
8ed539a5SBarry Smithextern int MatGetCol(Mat,int,int *,int **,Scalar**);
8ed539a5SBarry Smithextern int MatRestoreCol(Mat,int,int *,int **,Scalar**);
64e87e97SBarry Smithextern int MatGetArray(Mat,Scalar **);
8ed539a5SBarry Smithextern int MatMult(Mat,Vec,Vec);
8ed539a5SBarry Smithextern int MatMultAdd(Mat,Vec,Vec,Vec);
8ed539a5SBarry Smithextern int MatMultTrans(Mat,Vec,Vec);
8ed539a5SBarry Smithextern int MatMultTransAdd(Mat,Vec,Vec,Vec);
2eac72dbSBarry Smith
83f0b094SBarry Smithtypedef enum {ORDER_NATURAL=0,ORDER_ND=1,ORDER_1WD=2,
464493b3SBarry Smith              ORDER_RCM=3,ORDER_QMD=4,ORDER_APPLICATION_1,
464493b3SBarry Smith              ORDER_APPLICATION_2} MatOrdering;
ee50ffe9SBarry Smith
ee50ffe9SBarry Smithextern int MatGetReordering(Mat,MatOrdering,IS*,IS*);
832a8ecbSBarry Smithextern int MatReorderForNonzeroDiagonal(Mat,double,IS,IS);
464493b3SBarry Smithextern int MatReorderingRegister(MatOrdering,char*,
464493b3SBarry Smith                     int (*)(int*,int*,int*,int*,int*));
464493b3SBarry Smithextern int MatReorderingRegisterAll();
464493b3SBarry Smithextern int MatReorderingRegisterDestroy();
2eac72dbSBarry Smith
49d8b64dSBarry Smithextern int MatLUFactor(Mat,IS,IS,double);
49d8b64dSBarry Smithextern int MatCholeskyFactor(Mat,IS,double);
64051cb3SBarry Smithextern int MatLUFactorSymbolic(Mat,IS,IS,double,Mat*);
64051cb3SBarry Smithextern int MatILUFactorSymbolic(Mat,IS,IS,double,int,Mat*);
64051cb3SBarry Smithextern int MatCholeskyFactorSymbolic(Mat,IS,double,Mat*);
64051cb3SBarry Smithextern int MatIncompleteCholeskyFactorSymbolic(Mat,IS,double,int,Mat*);
20563c6bSBarry Smithextern int MatLUFactorNumeric(Mat,Mat*);
20563c6bSBarry Smithextern int MatCholeskyFactorNumeric(Mat,Mat*);
8ed539a5SBarry Smith
8ed539a5SBarry Smithextern int MatSolve(Mat,Vec,Vec);
22169318SLois Curfman McInnesextern int MatForwardSolve(Mat,Vec,Vec);
22169318SLois Curfman McInnesextern int MatBackwardSolve(Mat,Vec,Vec);
8ed539a5SBarry Smithextern int MatSolveAdd(Mat,Vec,Vec,Vec);
28988994SBarry Smithextern int MatSolveTrans(Mat,Vec,Vec);
28988994SBarry Smithextern int MatSolveTransAdd(Mat,Vec,Vec,Vec);
8ed539a5SBarry Smith
ee50ffe9SBarry Smithtypedef enum {SOR_FORWARD_SWEEP=1,SOR_BACKWARD_SWEEP=2,SOR_SYMMETRIC_SWEEP=3,
ee50ffe9SBarry Smith              SOR_LOCAL_FORWARD_SWEEP=4,SOR_LOCAL_BACKWARD_SWEEP=8,
ee50ffe9SBarry Smith              SOR_LOCAL_SYMMETRIC_SWEEP=12,SOR_ZERO_INITIAL_GUESS=16,
ca9b4cbeSLois Curfman McInnes              SOR_EISENSTAT=32,SOR_APPLY_UPPER=64,SOR_APPLY_LOWER=128
ca9b4cbeSLois Curfman McInnes              } MatSORType;
ee50ffe9SBarry Smith
ca9b4cbeSLois Curfman McInnesextern int MatRelax(Mat,Vec,double,MatSORType,double,int,Vec);
8ed539a5SBarry Smith
ca9b4cbeSLois Curfman McInnesextern int MatConvert(Mat,MatType,Mat*);
8ed539a5SBarry Smithextern int MatView(Mat,Viewer);
5f52d6abSBarry Smithextern int MatLoad(Viewer,MatType,Mat*);
8d371f33SLois Curfman McInnes
8ed539a5SBarry Smith#include <stdio.h>
96c0e93cSLois Curfman McInnes
ee50ffe9SBarry Smithtypedef enum {MAT_LOCAL=1,MAT_GLOBAL_MAX=2,MAT_GLOBAL_SUM=3} MatInfoType;
96c0e93cSLois Curfman McInnes
ee50ffe9SBarry Smithextern int MatGetInfo(Mat,MatInfoType,int*,int*,int*);
8ed539a5SBarry Smithextern int MatGetDiagonal(Mat,Vec);
b7e13cc3SBarry Smithextern int MatTranspose(Mat,Mat*);
8ed539a5SBarry Smithextern int MatScale(Mat,Vec,Vec);
8ed539a5SBarry Smithextern int MatEqual(Mat,Mat);
b7e13cc3SBarry Smithextern int MatScatterBegin(Mat,Mat,InsertMode,MatScatterCtx);
b7e13cc3SBarry Smithextern int MatScatterEnd(Mat,Mat,InsertMode,MatScatterCtx);
b7e13cc3SBarry Smithextern int MatScatterCtxCreate(Mat,IS,IS,Mat,IS,IS,MatScatterCtx*);
b7e13cc3SBarry Smithextern int MatScatterCtxDestroy(MatScatterCtx);
2eac72dbSBarry Smith
ee50ffe9SBarry Smithtypedef enum {NORM_1=1,NORM_2=2,NORM_FROBENIUS=3,NORM_INFINITY=4} MatNormType;
ee50ffe9SBarry Smithextern int MatNorm(Mat,MatNormType,double *);
2eac72dbSBarry Smith
8ed539a5SBarry Smithextern int MatZeroEntries(Mat);
20563c6bSBarry Smithextern int MatZeroRows(Mat,IS,Scalar*);
20563c6bSBarry Smithextern int MatZeroColumns(Mat,IS,Scalar*);
8ed539a5SBarry Smith
8ed539a5SBarry Smithextern int MatDestroy(Mat);
8ed539a5SBarry Smith
1eb62cbbSBarry Smithextern int MatGetSize(Mat,int*,int*);
1eb62cbbSBarry Smithextern int MatGetLocalSize(Mat,int*,int*);
d6dfbf8fSBarry Smithextern int MatGetOwnershipRange(Mat,int*,int*);
1eb62cbbSBarry Smith
6469c4f9SBarry Smithextern int MatCreate(MPI_Comm,int,int,Mat*);
8ed539a5SBarry Smith
6eafb6e7SBarry Smithextern int MatGetSubMatrix(Mat,IS,IS,Mat*);
6eafb6e7SBarry Smithextern int MatGetSubMatrixInPlace(Mat,IS,IS);
0357936bSBarry Smith
0357936bSBarry Smithextern int MatAXPY(Scalar *,Mat,Mat);
c01c455dSBarry Smithextern int MatCompress(Mat);
c804fac0SBarry Smith
fa10aba9SLois Curfman McInnes/* Routines unique to particular data structures */
336446d4SLois Curfman McInnesextern int MatBDiagGetData(Mat,int*,int*,int**,int**,Scalar***);
fa10aba9SLois Curfman McInnes
2eac72dbSBarry Smith#endif
2eac72dbSBarry Smith
2eac72dbSBarry Smith
9d00d63dSBarry Smith