xref: /petsc/include/petscmat.h (revision 8a1243692a7577dbaf7e3932d63d3490e8f07c26)
1*8a124369SBarry Smith /* $Id: petscmat.h,v 1.222 2001/07/11 15:46:57 bsmith Exp bsmith $ */
22eac72dbSBarry Smith /*
32eac72dbSBarry Smith      Include file for the matrix component of PETSc
42eac72dbSBarry Smith */
50a835dfdSSatish Balay #ifndef __PETSCMAT_H
60a835dfdSSatish Balay #define __PETSCMAT_H
70a835dfdSSatish Balay #include "petscvec.h"
82eac72dbSBarry Smith 
99cd28387SBarry Smith #define MAT_COOKIE         PETSC_COOKIE+5
10f0479e8cSBarry Smith 
11d9274352SBarry Smith /*S
12d9274352SBarry Smith      Mat - Abstract PETSc matrix object
132eac72dbSBarry Smith 
14d91e6319SBarry Smith    Level: beginner
15d91e6319SBarry Smith 
16d9274352SBarry Smith   Concepts: matrix; linear operator
17d9274352SBarry Smith 
18d9274352SBarry Smith .seealso:  MatCreate(), MatType, MatSetType()
19d9274352SBarry Smith S*/
20d9274352SBarry Smith typedef struct _p_Mat*           Mat;
21d9274352SBarry Smith 
22d9274352SBarry Smith /*E
23d9274352SBarry Smith     MatType - String with the name of a PETSc matrix or the creation function
24d9274352SBarry Smith        with an optional dynamic library name, for example
25d9274352SBarry Smith        http://www.mcs.anl.gov/petsc/lib.a:mymatcreate()
26d9274352SBarry Smith 
27d9274352SBarry Smith    Level: beginner
28d9274352SBarry Smith 
29d9274352SBarry Smith .seealso: MatSetType(), Mat
30d91e6319SBarry Smith E*/
31273d9f13SBarry Smith #define MATSAME     "same"
32273d9f13SBarry Smith #define MATSEQMAIJ  "seqmaij"
33273d9f13SBarry Smith #define MATMPIMAIJ  "mpimaij"
34273d9f13SBarry Smith #define MATIS       "is"
35273d9f13SBarry Smith #define MATMPIROWBS "mpirowbs"
36273d9f13SBarry Smith #define MATSEQDENSE "seqdense"
37273d9f13SBarry Smith #define MATSEQAIJ   "seqaij"
38273d9f13SBarry Smith #define MATMPIAIJ   "mpiaij"
39273d9f13SBarry Smith #define MATSHELL    "shell"
40273d9f13SBarry Smith #define MATSEQBDIAG "seqbdiag"
41273d9f13SBarry Smith #define MATMPIBDIAG "mpibdiag"
42273d9f13SBarry Smith #define MATMPIDENSE "mpidense"
43273d9f13SBarry Smith #define MATSEQBAIJ  "seqbaij"
44273d9f13SBarry Smith #define MATMPIBAIJ  "mpibaij"
45273d9f13SBarry Smith #define MATMPIADJ   "mpiadj"
46273d9f13SBarry Smith #define MATSEQSBAIJ "seqsbaij"
47273d9f13SBarry Smith #define MATMPISBAIJ "mpisbaij"
48cebc7f6cSBarry Smith #define MATDAAD     "daad"
49cebc7f6cSBarry Smith #define MATMFFD     "mffd"
50273d9f13SBarry Smith typedef char* MatType;
51d91e6319SBarry Smith 
52273d9f13SBarry Smith EXTERN int MatCreate(MPI_Comm,int,int,int,int,Mat*);
53273d9f13SBarry Smith EXTERN int MatSetType(Mat,MatType);
54273d9f13SBarry Smith EXTERN int MatSetFromOptions(Mat);
55273d9f13SBarry Smith EXTERN int MatSetUpPreallocation(Mat);
56273d9f13SBarry Smith EXTERN int MatRegisterAll(char*);
57273d9f13SBarry Smith EXTERN int MatRegister(char*,char*,char*,int(*)(Mat));
58273d9f13SBarry Smith #if defined(PETSC_USE_DYNAMIC_LIBRARIES)
59273d9f13SBarry Smith #define MatRegisterDynamic(a,b,c,d) MatRegister(a,b,c,0)
60273d9f13SBarry Smith #else
61273d9f13SBarry Smith #define MatRegisterDynamic(a,b,c,d) MatRegister(a,b,c,d)
62273d9f13SBarry Smith #endif
63273d9f13SBarry Smith extern PetscTruth MatRegisterAllCalled;
64b0a32e0cSBarry Smith extern PetscFList MatList;
6528988994SBarry Smith 
66ca44d042SBarry Smith EXTERN int MatCreate(MPI_Comm,int,int,int,int,Mat*);
67ca44d042SBarry Smith EXTERN int MatCreateSeqDense(MPI_Comm,int,int,Scalar*,Mat*);
68ca44d042SBarry Smith EXTERN int MatCreateMPIDense(MPI_Comm,int,int,int,int,Scalar*,Mat*);
69ca44d042SBarry Smith EXTERN int MatCreateSeqAIJ(MPI_Comm,int,int,int,int*,Mat*);
70ca44d042SBarry Smith EXTERN int MatCreateMPIAIJ(MPI_Comm,int,int,int,int,int,int*,int,int*,Mat*);
71c4f061fbSSatish Balay EXTERN int MatCreateMPIRowbs(MPI_Comm,int,int,int,int*,Mat*);
72ca44d042SBarry Smith EXTERN int MatCreateSeqBDiag(MPI_Comm,int,int,int,int,int*,Scalar**,Mat*);
73ca44d042SBarry Smith EXTERN int MatCreateMPIBDiag(MPI_Comm,int,int,int,int,int,int*,Scalar**,Mat*);
74ca44d042SBarry Smith EXTERN int MatCreateSeqBAIJ(MPI_Comm,int,int,int,int,int*,Mat*);
75ca44d042SBarry Smith EXTERN int MatCreateMPIBAIJ(MPI_Comm,int,int,int,int,int,int,int*,int,int*,Mat*);
76ca44d042SBarry Smith EXTERN int MatCreateMPIAdj(MPI_Comm,int,int,int*,int*,int *,Mat*);
77ca44d042SBarry Smith EXTERN int MatCreateSeqSBAIJ(MPI_Comm,int,int,int,int,int*,Mat*);
78ca44d042SBarry Smith EXTERN int MatCreateMPISBAIJ(MPI_Comm,int,int,int,int,int,int,int*,int,int*,Mat*);
79ca44d042SBarry Smith EXTERN int MatCreateShell(MPI_Comm,int,int,int,int,void *,Mat*);
803a7fca6bSBarry Smith EXTERN int MatCreateAdic(MPI_Comm,int,int,int,int,int,void (*)(void),Mat*);
81435da068SBarry Smith EXTERN int MatDestroy(Mat);
8221c89e3eSBarry Smith 
83ca44d042SBarry Smith EXTERN int MatPrintHelp(Mat);
84*8a124369SBarry Smith EXTERN int MatGetPetscMaps(Mat,PetscMap*,PetscMap*);
85ec0117caSBarry Smith 
868ed539a5SBarry Smith /* ------------------------------------------------------------*/
87ca44d042SBarry Smith EXTERN int MatSetValues(Mat,int,int*,int,int*,Scalar*,InsertMode);
88ca44d042SBarry Smith EXTERN int MatSetValuesBlocked(Mat,int,int*,int,int*,Scalar*,InsertMode);
8984cb2905SBarry Smith 
902ef4de8bSBarry Smith /*S
912ef4de8bSBarry Smith      MatStencil - Data structure (C struct) for storing information about a single row or
922ef4de8bSBarry Smith         column of a matrix as index on an associated grid.
932ef4de8bSBarry Smith 
942ef4de8bSBarry Smith    Level: beginner
952ef4de8bSBarry Smith 
962ef4de8bSBarry Smith   Concepts: matrix; linear operator
972ef4de8bSBarry Smith 
982ef4de8bSBarry Smith .seealso:  MatSetValuesStencil(), MatSetStencil()
992ef4de8bSBarry Smith S*/
100435da068SBarry Smith typedef struct {
101435da068SBarry Smith   int k,j,i,c;
102435da068SBarry Smith } MatStencil;
1032ef4de8bSBarry Smith 
104435da068SBarry Smith EXTERN int MatSetValuesStencil(Mat,int,MatStencil*,int,MatStencil*,Scalar*,InsertMode);
105435da068SBarry Smith EXTERN int MatSetValuesBlockedStencil(Mat,int,MatStencil*,int,MatStencil*,Scalar*,InsertMode);
106435da068SBarry Smith EXTERN int MatSetStencil(Mat,int,int*,int*,int);
107435da068SBarry Smith 
1083a7fca6bSBarry Smith EXTERN int MatSetColoring(Mat,ISColoring);
1093a7fca6bSBarry Smith EXTERN int MatSetValuesAdic(Mat,void*);
1103a7fca6bSBarry Smith EXTERN int MatSetValuesAdifor(Mat,int,void*);
1113a7fca6bSBarry Smith 
112d91e6319SBarry Smith /*E
113d91e6319SBarry Smith     MatAssemblyType - Indicates if the matrix is now to be used, or if you plan
114d91e6319SBarry Smith      to continue to add values to it
115d91e6319SBarry Smith 
116d91e6319SBarry Smith     Level: beginner
117d91e6319SBarry Smith 
118d91e6319SBarry Smith .seealso: MatAssemblyBegin(), MatAssemblyEnd()
119d91e6319SBarry Smith E*/
1206d4a8577SBarry Smith typedef enum {MAT_FLUSH_ASSEMBLY=1,MAT_FINAL_ASSEMBLY=0} MatAssemblyType;
121ca44d042SBarry Smith EXTERN int MatAssemblyBegin(Mat,MatAssemblyType);
122ca44d042SBarry Smith EXTERN int MatAssemblyEnd(Mat,MatAssemblyType);
123ca44d042SBarry Smith EXTERN int MatAssembled(Mat,PetscTruth*);
1244f9c727eSBarry Smith 
125b951964fSBarry Smith #define MatSetValue(v,i,j,va,mode) \
126d91e6319SBarry Smith 0; {int _ierr,_row = i,_col = j; Scalar _va = va; \
127b951964fSBarry Smith   _ierr = MatSetValues(v,1,&_row,1,&_col,&_va,mode);CHKERRQ(_ierr); \
128b951964fSBarry Smith }
129ea06a074SBarry Smith #define MatGetValue(v,i,j,va) \
130d91e6319SBarry Smith 0; {int _ierr,_row = i,_col = j; \
131ea06a074SBarry Smith   _ierr = MatGetValues(v,1,&_row,1,&_col,&va);CHKERRQ(_ierr); \
132ea06a074SBarry Smith }
133d91e6319SBarry Smith #define MatSetValueLocal(v,i,j,va,mode) \
134d91e6319SBarry Smith 0; {int _ierr,_row = i,_col = j; Scalar _va = va; \
135d91e6319SBarry Smith   _ierr = MatSetValuesLocal(v,1,&_row,1,&_col,&_va,mode);CHKERRQ(_ierr); \
136d91e6319SBarry Smith }
137d91e6319SBarry Smith /*E
138d91e6319SBarry Smith     MatOption - Options that may be set for a matrix and its behavior or storage
139d91e6319SBarry Smith 
140d91e6319SBarry Smith     Level: beginner
141d91e6319SBarry Smith 
1420a835dfdSSatish Balay    Any additions/changes here MUST also be made in include/finclude/petscmat.h
143d91e6319SBarry Smith 
144d91e6319SBarry Smith .seealso: MatSetOption()
145d91e6319SBarry Smith E*/
1466d4a8577SBarry Smith typedef enum {MAT_ROW_ORIENTED=1,MAT_COLUMN_ORIENTED=2,MAT_ROWS_SORTED=4,
1476d4a8577SBarry Smith               MAT_COLUMNS_SORTED=8,MAT_NO_NEW_NONZERO_LOCATIONS=16,
1486d4a8577SBarry Smith               MAT_YES_NEW_NONZERO_LOCATIONS=32,MAT_SYMMETRIC=64,
1496ca9ecd3SBarry Smith               MAT_STRUCTURALLY_SYMMETRIC=65,MAT_NO_NEW_DIAGONALS=66,
1506ca9ecd3SBarry Smith               MAT_YES_NEW_DIAGONALS=67,MAT_INODE_LIMIT_1=68,MAT_INODE_LIMIT_2=69,
1516ca9ecd3SBarry Smith               MAT_INODE_LIMIT_3=70,MAT_INODE_LIMIT_4=71,MAT_INODE_LIMIT_5=72,
1526ca9ecd3SBarry Smith               MAT_IGNORE_OFF_PROC_ENTRIES=73,MAT_ROWS_UNSORTED=74,
1534787f768SSatish Balay               MAT_COLUMNS_UNSORTED=75,MAT_NEW_NONZERO_LOCATION_ERR=76,
1547c922b88SBarry Smith               MAT_NEW_NONZERO_ALLOCATION_ERR=77,MAT_USE_HASH_TABLE=78,
1552bad1931SBarry Smith               MAT_KEEP_ZEROED_ROWS=79,MAT_IGNORE_ZERO_ENTRIES=80,MAT_USE_INODES=81,
1568b43beb6SKris Buschelman               MAT_DO_NOT_USE_INODES=82,MAT_USE_SINGLE_PRECISION_SOLVES=83} MatOption;
157ca44d042SBarry Smith EXTERN int MatSetOption(Mat,MatOption);
158273d9f13SBarry Smith EXTERN int MatGetType(Mat,MatType*);
15984cb2905SBarry Smith 
160ca44d042SBarry Smith EXTERN int MatGetValues(Mat,int,int*,int,int*,Scalar*);
161ca44d042SBarry Smith EXTERN int MatGetRow(Mat,int,int *,int **,Scalar**);
162ca44d042SBarry Smith EXTERN int MatRestoreRow(Mat,int,int *,int **,Scalar**);
163ca44d042SBarry Smith EXTERN int MatGetColumn(Mat,int,int *,int **,Scalar**);
164ca44d042SBarry Smith EXTERN int MatRestoreColumn(Mat,int,int *,int **,Scalar**);
165ca44d042SBarry Smith EXTERN int MatGetColumnVector(Mat,Vec,int);
166ca44d042SBarry Smith EXTERN int MatGetArray(Mat,Scalar **);
167ca44d042SBarry Smith EXTERN int MatRestoreArray(Mat,Scalar **);
168ca44d042SBarry Smith EXTERN int MatGetBlockSize(Mat,int *);
1697b80b807SBarry Smith 
170ca44d042SBarry Smith EXTERN int MatMult(Mat,Vec,Vec);
171ca44d042SBarry Smith EXTERN int MatMultAdd(Mat,Vec,Vec,Vec);
172ca44d042SBarry Smith EXTERN int MatMultTranspose(Mat,Vec,Vec);
173ca44d042SBarry Smith EXTERN int MatMultTransposeAdd(Mat,Vec,Vec,Vec);
1742eac72dbSBarry Smith 
175d91e6319SBarry Smith /*E
176d91e6319SBarry Smith     MatDuplicateOption - Indicates if a duplicated sparse matrix should have
177d91e6319SBarry Smith   its numerical values copied over or just its nonzero structure.
178d91e6319SBarry Smith 
179d91e6319SBarry Smith     Level: beginner
180d91e6319SBarry Smith 
181d91e6319SBarry Smith    Any additions/changes here MUST also be made in include/finclude/petscmat.h
182d91e6319SBarry Smith 
183d91e6319SBarry Smith .seealso: MatDuplicate()
184d91e6319SBarry Smith E*/
1852e8a6d31SBarry Smith typedef enum {MAT_DO_NOT_COPY_VALUES,MAT_COPY_VALUES} MatDuplicateOption;
1862e8a6d31SBarry Smith 
187273d9f13SBarry Smith EXTERN int MatConvertRegister(char*,char*,char*,int (*)(Mat,MatType,Mat*));
188273d9f13SBarry Smith #if defined(PETSC_USE_DYNAMIC_LIBRARIES)
189273d9f13SBarry Smith #define MatConvertRegisterDynamic(a,b,c,d) MatConvertRegister(a,b,c,0)
190273d9f13SBarry Smith #else
191273d9f13SBarry Smith #define MatConvertRegisterDynamic(a,b,c,d) MatConvertRegister(a,b,c,d)
192273d9f13SBarry Smith #endif
193273d9f13SBarry Smith EXTERN int        MatConvertRegisterAll(char*);
194273d9f13SBarry Smith EXTERN int        MatConvertRegisterDestroy(void);
195273d9f13SBarry Smith extern PetscTruth MatConvertRegisterAllCalled;
196b0a32e0cSBarry Smith extern PetscFList MatConvertList;
197ca44d042SBarry Smith EXTERN int        MatConvert(Mat,MatType,Mat*);
198ca44d042SBarry Smith EXTERN int        MatDuplicate(Mat,MatDuplicateOption,Mat*);
19994a9d846SBarry Smith 
200d91e6319SBarry Smith /*E
201d91e6319SBarry Smith     MatStructure - Indicates if the matrix has the same nonzero structure
202d91e6319SBarry Smith 
203d91e6319SBarry Smith     Level: beginner
204d91e6319SBarry Smith 
205d91e6319SBarry Smith    Any additions/changes here MUST also be made in include/finclude/petscmat.h
206d91e6319SBarry Smith 
207d91e6319SBarry Smith .seealso: MatCopy(), SLESSetOperators(), PCSetOperators()
208d91e6319SBarry Smith E*/
209cb5b572fSBarry Smith typedef enum {SAME_NONZERO_PATTERN,DIFFERENT_NONZERO_PATTERN,SAME_PRECONDITIONER} MatStructure;
210cb5b572fSBarry Smith 
211ca44d042SBarry Smith EXTERN int MatCopy(Mat,Mat,MatStructure);
212b0a32e0cSBarry Smith EXTERN int MatView(Mat,PetscViewer);
213273d9f13SBarry Smith 
214b0a32e0cSBarry Smith EXTERN int MatLoadRegister(char*,char*,char*,int (*)(PetscViewer,MatType,Mat*));
215273d9f13SBarry Smith #if defined(PETSC_USE_DYNAMIC_LIBRARIES)
216273d9f13SBarry Smith #define MatLoadRegisterDynamic(a,b,c,d) MatLoadRegister(a,b,c,0)
217273d9f13SBarry Smith #else
218273d9f13SBarry Smith #define MatLoadRegisterDynamic(a,b,c,d) MatLoadRegister(a,b,c,d)
219273d9f13SBarry Smith #endif
220273d9f13SBarry Smith EXTERN int        MatLoadRegisterAll(char*);
221273d9f13SBarry Smith EXTERN int        MatLoadRegisterDestroy(void);
222273d9f13SBarry Smith extern PetscTruth MatLoadRegisterAllCalled;
223b0a32e0cSBarry Smith extern PetscFList MatLoadList;
224b0a32e0cSBarry Smith EXTERN int        MatLoad(PetscViewer,MatType,Mat*);
2257b80b807SBarry Smith 
226ca44d042SBarry Smith EXTERN int MatGetRowIJ(Mat,int,PetscTruth,int*,int **,int **,PetscTruth *);
227ca44d042SBarry Smith EXTERN int MatRestoreRowIJ(Mat,int,PetscTruth,int *,int **,int **,PetscTruth *);
228ca44d042SBarry Smith EXTERN int MatGetColumnIJ(Mat,int,PetscTruth,int*,int **,int **,PetscTruth *);
229ca44d042SBarry Smith EXTERN int MatRestoreColumnIJ(Mat,int,PetscTruth,int *,int **,int **,PetscTruth *);
230d4fbbf0eSBarry Smith 
231d91e6319SBarry Smith /*S
232d91e6319SBarry Smith      MatInfo - Context of matrix information, used with MatGetInfo()
233d91e6319SBarry Smith 
234d91e6319SBarry Smith    In Fortran this is simply a double precision array of dimension MAT_INFO_SIZE
235d91e6319SBarry Smith 
236d91e6319SBarry Smith    Level: intermediate
237d91e6319SBarry Smith 
238d91e6319SBarry Smith   Concepts: matrix^nonzero information
239d91e6319SBarry Smith 
240d9274352SBarry Smith .seealso:  MatGetInfo(), MatInfoType
241d91e6319SBarry Smith S*/
2424e220ebcSLois Curfman McInnes typedef struct {
243b0a32e0cSBarry Smith   PetscLogDouble rows_global,columns_global;         /* number of global rows and columns */
244b0a32e0cSBarry Smith   PetscLogDouble rows_local,columns_local;           /* number of local rows and columns */
245b0a32e0cSBarry Smith   PetscLogDouble block_size;                         /* block size */
246b0a32e0cSBarry Smith   PetscLogDouble nz_allocated,nz_used,nz_unneeded;   /* number of nonzeros */
247b0a32e0cSBarry Smith   PetscLogDouble memory;                             /* memory allocated */
248b0a32e0cSBarry Smith   PetscLogDouble assemblies;                         /* number of matrix assemblies called */
249b0a32e0cSBarry Smith   PetscLogDouble mallocs;                            /* number of mallocs during MatSetValues() */
250b0a32e0cSBarry Smith   PetscLogDouble fill_ratio_given,fill_ratio_needed; /* fill ratio for LU/ILU */
251b0a32e0cSBarry Smith   PetscLogDouble factor_mallocs;                     /* number of mallocs during factorization */
2524e220ebcSLois Curfman McInnes } MatInfo;
2534e220ebcSLois Curfman McInnes 
254d9274352SBarry Smith /*E
255d9274352SBarry Smith     MatInfoType - Indicates if you want information about the local part of the matrix,
256d9274352SBarry Smith      the entire parallel matrix or the maximum over all the local parts.
257d9274352SBarry Smith 
258d9274352SBarry Smith     Level: beginner
259d9274352SBarry Smith 
260d9274352SBarry Smith    Any additions/changes here MUST also be made in include/finclude/petscmat.h
261d9274352SBarry Smith 
262d9274352SBarry Smith .seealso: MatGetInfo(), MatInfo
263d9274352SBarry Smith E*/
2647b80b807SBarry Smith typedef enum {MAT_LOCAL=1,MAT_GLOBAL_MAX=2,MAT_GLOBAL_SUM=3} MatInfoType;
265ca44d042SBarry Smith EXTERN int MatGetInfo(Mat,MatInfoType,MatInfo*);
266ca44d042SBarry Smith EXTERN int MatValid(Mat,PetscTruth*);
267ca44d042SBarry Smith EXTERN int MatGetDiagonal(Mat,Vec);
268273d9f13SBarry Smith EXTERN int MatGetRowMax(Mat,Vec);
269ca44d042SBarry Smith EXTERN int MatTranspose(Mat,Mat*);
270ca44d042SBarry Smith EXTERN int MatPermute(Mat,IS,IS,Mat *);
271ca44d042SBarry Smith EXTERN int MatDiagonalScale(Mat,Vec,Vec);
27206ef90c2SBarry Smith EXTERN int MatDiagonalSet(Mat,Vec,InsertMode);
273ca44d042SBarry Smith EXTERN int MatEqual(Mat,Mat,PetscTruth*);
2747b80b807SBarry Smith 
275ca44d042SBarry Smith EXTERN int MatNorm(Mat,NormType,double *);
276ca44d042SBarry Smith EXTERN int MatZeroEntries(Mat);
277ca44d042SBarry Smith EXTERN int MatZeroRows(Mat,IS,Scalar*);
278ca44d042SBarry Smith EXTERN int MatZeroColumns(Mat,IS,Scalar*);
2797b80b807SBarry Smith 
280ca44d042SBarry Smith EXTERN int MatUseScaledForm(Mat,PetscTruth);
281ca44d042SBarry Smith EXTERN int MatScaleSystem(Mat,Vec,Vec);
282ca44d042SBarry Smith EXTERN int MatUnScaleSystem(Mat,Vec,Vec);
2835ef9f2a5SBarry Smith 
284ca44d042SBarry Smith EXTERN int MatGetSize(Mat,int*,int*);
285ca44d042SBarry Smith EXTERN int MatGetLocalSize(Mat,int*,int*);
286ca44d042SBarry Smith EXTERN int MatGetOwnershipRange(Mat,int*,int*);
2877b80b807SBarry Smith 
288d91e6319SBarry Smith /*E
289d91e6319SBarry Smith     MatReuse - Indicates if matrices obtained from a previous call to MatGetSubMatrices()
290d91e6319SBarry Smith      or MatGetSubMatrix() are to be reused to store the new matrix values.
291d91e6319SBarry Smith 
292d91e6319SBarry Smith     Level: beginner
293d91e6319SBarry Smith 
294d91e6319SBarry Smith    Any additions/changes here MUST also be made in include/finclude/petscmat.h
295d91e6319SBarry Smith 
296d91e6319SBarry Smith .seealso: MatGetSubMatrices(), MatGetSubMatrix(), MatDestroyMatrices()
297d91e6319SBarry Smith E*/
2987b2a1423SBarry Smith typedef enum {MAT_INITIAL_MATRIX,MAT_REUSE_MATRIX} MatReuse;
299ca44d042SBarry Smith EXTERN int MatGetSubMatrices(Mat,int,IS *,IS *,MatReuse,Mat **);
300ca44d042SBarry Smith EXTERN int MatDestroyMatrices(int,Mat **);
301ca44d042SBarry Smith EXTERN int MatGetSubMatrix(Mat,IS,IS,int,MatReuse,Mat *);
3028efafbd8SBarry Smith 
303ca44d042SBarry Smith EXTERN int MatIncreaseOverlap(Mat,int,IS *,int);
3047b80b807SBarry Smith 
305ca44d042SBarry Smith EXTERN int MatAXPY(Scalar *,Mat,Mat);
306ca44d042SBarry Smith EXTERN int MatAYPX(Scalar *,Mat,Mat);
307ca44d042SBarry Smith EXTERN int MatCompress(Mat);
3087b80b807SBarry Smith 
309ca44d042SBarry Smith EXTERN int MatScale(Scalar *,Mat);
310ca44d042SBarry Smith EXTERN int MatShift(Scalar *,Mat);
311052efed2SBarry Smith 
312ca44d042SBarry Smith EXTERN int MatSetLocalToGlobalMapping(Mat,ISLocalToGlobalMapping);
313ca44d042SBarry Smith EXTERN int MatSetLocalToGlobalMappingBlock(Mat,ISLocalToGlobalMapping);
314ca44d042SBarry Smith EXTERN int MatZeroRowsLocal(Mat,IS,Scalar*);
315ca44d042SBarry Smith EXTERN int MatSetValuesLocal(Mat,int,int*,int,int*,Scalar*,InsertMode);
316ca44d042SBarry Smith EXTERN int MatSetValuesBlockedLocal(Mat,int,int*,int,int*,Scalar*,InsertMode);
31790f02eecSBarry Smith 
318ca44d042SBarry Smith EXTERN int MatSetStashInitialSize(Mat,int,int);
319649db694SBarry Smith 
320ca44d042SBarry Smith EXTERN int MatInterpolateAdd(Mat,Vec,Vec,Vec);
321ca44d042SBarry Smith EXTERN int MatInterpolate(Mat,Vec,Vec);
322ca44d042SBarry Smith EXTERN int MatRestrict(Mat,Vec,Vec);
3237c922b88SBarry Smith 
3247c922b88SBarry Smith /*
325c4f061fbSSatish Balay       These three (or four) macros MUST be used together. The third one closes the open { of the first one
3267c922b88SBarry Smith */
327c4f061fbSSatish Balay #define MatPreallocateInitialize(comm,nrows,ncols,dnz,onz) 0; \
3287c922b88SBarry Smith { \
3297c922b88SBarry Smith   int __ierr,__tmp = (nrows),__ctmp = (ncols),__rstart,__start,__end; \
33082502324SSatish Balay   __ierr = PetscMalloc(2*__tmp*sizeof(int),&dnz);CHKERRQ(__ierr);onz = dnz + __tmp;\
3317c922b88SBarry Smith   __ierr = PetscMemzero(dnz,2*__tmp*sizeof(int));CHKERRQ(__ierr);\
3327c922b88SBarry Smith   __ierr = MPI_Scan(&__ctmp,&__end,1,MPI_INT,MPI_SUM,comm);CHKERRQ(__ierr); __start = __end - __ctmp;\
3337c922b88SBarry Smith   __ierr = MPI_Scan(&__tmp,&__rstart,1,MPI_INT,MPI_SUM,comm);CHKERRQ(__ierr); __rstart = __rstart - __tmp;
3347c922b88SBarry Smith 
335c4f061fbSSatish Balay #define MatPreallocateSetLocal(map,nrows,rows,ncols,cols,dnz,onz) 0;\
336c4f061fbSSatish Balay {\
337c4f061fbSSatish Balay   int __l;\
338c4f061fbSSatish Balay   __ierr = ISLocalToGlobalMappingApply(map,nrows,rows,rows);CHKERRQ(__ierr);\
339c4f061fbSSatish Balay   __ierr = ISLocalToGlobalMappingApply(map,ncols,cols,cols);CHKERRQ(__ierr);\
340c4f061fbSSatish Balay   for (__l=0;__l<nrows;__l++) {\
341435da068SBarry Smith     __ierr = MatPreallocateSet((rows)[__l],ncols,cols,dnz,onz);CHKERRQ(__ierr);\
342c4f061fbSSatish Balay   }\
343c4f061fbSSatish Balay }
344c4f061fbSSatish Balay 
345c4f061fbSSatish Balay #define MatPreallocateSet(row,nc,cols,dnz,onz) 0;\
3467c922b88SBarry Smith { int __i; \
3477c922b88SBarry Smith   for (__i=0; __i<nc; __i++) {\
3487c922b88SBarry Smith     if (cols[__i] < __start || cols[__i] >= __end) onz[row - __rstart]++; \
3497c922b88SBarry Smith   }\
3507c922b88SBarry Smith   dnz[row - __rstart] = nc - onz[row - __rstart];\
3517c922b88SBarry Smith }
3527c922b88SBarry Smith 
353c4f061fbSSatish Balay #define MatPreallocateFinalize(dnz,onz) 0;__ierr = PetscFree(dnz);CHKERRQ(__ierr);}
3547c922b88SBarry Smith 
3557b80b807SBarry Smith /* Routines unique to particular data structures */
356435da068SBarry Smith EXTERN int MatShellGetContext(Mat,void **);
357435da068SBarry Smith 
358ca44d042SBarry Smith EXTERN int MatBDiagGetData(Mat,int*,int*,int**,int**,Scalar***);
359ca44d042SBarry Smith EXTERN int MatSeqAIJSetColumnIndices(Mat,int *);
360ca44d042SBarry Smith EXTERN int MatSeqBAIJSetColumnIndices(Mat,int *);
361ca44d042SBarry Smith EXTERN int MatCreateSeqAIJWithArrays(MPI_Comm,int,int,int*,int*,Scalar *,Mat*);
3627b80b807SBarry Smith 
363273d9f13SBarry Smith EXTERN int MatSeqBAIJSetPreallocation(Mat,int,int,int*);
364273d9f13SBarry Smith EXTERN int MatSeqSBAIJSetPreallocation(Mat,int,int,int*);
365273d9f13SBarry Smith EXTERN int MatSeqAIJSetPreallocation(Mat,int,int*);
366273d9f13SBarry Smith EXTERN int MatSeqDensePreallocation(Mat,Scalar*);
367273d9f13SBarry Smith EXTERN int MatSeqBDiagSetPreallocation(Mat,int,int,int*,Scalar**);
368273d9f13SBarry Smith EXTERN int MatSeqDenseSetPreallocation(Mat,Scalar*);
369273d9f13SBarry Smith 
370273d9f13SBarry Smith EXTERN int MatMPIBAIJSetPreallocation(Mat,int,int,int*,int,int*);
371273d9f13SBarry Smith EXTERN int MatMPISBAIJSetPreallocation(Mat,int,int,int*,int,int*);
372273d9f13SBarry Smith EXTERN int MatMPIAIJSetPreallocation(Mat,int,int*,int,int*);
373273d9f13SBarry Smith EXTERN int MatMPIDensePreallocation(Mat,Scalar*);
374273d9f13SBarry Smith EXTERN int MatMPIBDiagSetPreallocation(Mat,int,int,int*,Scalar**);
375273d9f13SBarry Smith EXTERN int MatMPIAdjSetPreallocation(Mat,int*,int*,int*);
376273d9f13SBarry Smith EXTERN int MatMPIDenseSetPreallocation(Mat,Scalar*);
377273d9f13SBarry Smith EXTERN int MatMPIRowbsSetPreallocation(Mat,int,int*);
378435da068SBarry Smith EXTERN int MatMPIAIJGetSeqAIJ(Mat,Mat*,Mat*,int**);
379435da068SBarry Smith EXTERN int MatMPIBAIJGetSeqBAIJ(Mat,Mat*,Mat*,int**);
3803a7fca6bSBarry Smith EXTERN int MatAdicSetLocalFunction(Mat,void (*)(void));
381273d9f13SBarry Smith 
382ca44d042SBarry Smith EXTERN int MatStoreValues(Mat);
383ca44d042SBarry Smith EXTERN int MatRetrieveValues(Mat);
3842e8a6d31SBarry Smith 
3853a7fca6bSBarry Smith EXTERN int MatDAADSetCtx(Mat,void*);
3863a7fca6bSBarry Smith 
3877b80b807SBarry Smith /*
3887b80b807SBarry Smith   These routines are not usually accessed directly, rather solving is
3897b80b807SBarry Smith   done through the SLES, KSP and PC interfaces.
3907b80b807SBarry Smith */
3917b80b807SBarry Smith 
392d9274352SBarry Smith /*E
393d9274352SBarry Smith     MatOrderingType - String with the name of a PETSc matrix ordering or the creation function
394d9274352SBarry Smith        with an optional dynamic library name, for example
395d9274352SBarry Smith        http://www.mcs.anl.gov/petsc/lib.a:orderingcreate()
396d9274352SBarry Smith 
397d9274352SBarry Smith    Level: beginner
398d9274352SBarry Smith 
399d9274352SBarry Smith .seealso: MatGetOrdering()
400d9274352SBarry Smith E*/
401b12f92e5SBarry Smith typedef char* MatOrderingType;
402b12f92e5SBarry Smith #define MATORDERING_NATURAL   "natural"
403b12f92e5SBarry Smith #define MATORDERING_ND        "nd"
404b12f92e5SBarry Smith #define MATORDERING_1WD       "1wd"
405b12f92e5SBarry Smith #define MATORDERING_RCM       "rcm"
406b12f92e5SBarry Smith #define MATORDERING_QMD       "qmd"
407b12f92e5SBarry Smith #define MATORDERING_ROWLENGTH "rowlength"
40862152c8bSBarry Smith #define MATORDERING_DSC_ND    "dsc_nd"
40962152c8bSBarry Smith #define MATORDERING_DSC_MMD   "dsc_mmd"
41062152c8bSBarry Smith #define MATORDERING_DSC_MDF   "dsc_mdf"
411b12f92e5SBarry Smith 
412ca44d042SBarry Smith EXTERN int MatGetOrdering(Mat,MatOrderingType,IS*,IS*);
413ca44d042SBarry Smith EXTERN int MatOrderingRegister(char*,char*,char*,int(*)(Mat,MatOrderingType,IS*,IS*));
414aa482453SBarry Smith #if defined(PETSC_USE_DYNAMIC_LIBRARIES)
415f1af5d2fSBarry Smith #define MatOrderingRegisterDynamic(a,b,c,d) MatOrderingRegister(a,b,c,0)
416b12f92e5SBarry Smith #else
417f1af5d2fSBarry Smith #define MatOrderingRegisterDynamic(a,b,c,d) MatOrderingRegister(a,b,c,d)
418b12f92e5SBarry Smith #endif
419ca44d042SBarry Smith EXTERN int        MatOrderingRegisterDestroy(void);
420ca44d042SBarry Smith EXTERN int        MatOrderingRegisterAll(char*);
4212bad1931SBarry Smith extern PetscTruth MatOrderingRegisterAllCalled;
422b0a32e0cSBarry Smith extern PetscFList      MatOrderingList;
423d4fbbf0eSBarry Smith 
424ca44d042SBarry Smith EXTERN int MatReorderForNonzeroDiagonal(Mat,double,IS,IS);
425a2ce50c7SBarry Smith 
426ca44d042SBarry Smith EXTERN int MatCholeskyFactor(Mat,IS,double);
427ca44d042SBarry Smith EXTERN int MatCholeskyFactorSymbolic(Mat,IS,double,Mat*);
428ca44d042SBarry Smith EXTERN int MatCholeskyFactorNumeric(Mat,Mat*);
429a2ce50c7SBarry Smith 
430d91e6319SBarry Smith /*S
431d91e6319SBarry Smith    MatILUInfo - Data based into the matrix ILU factorization routines
4325ef9f2a5SBarry Smith 
433d91e6319SBarry Smith    In Fortran these are simply double precision arrays of size MAT_ILUINFO_SIZE
43414822f30SBarry Smith 
435d91e6319SBarry Smith    Notes: These are not usually directly used by users, instead use the PC type of ILU
436d91e6319SBarry Smith           All entries are double precision.
437d91e6319SBarry Smith 
438d91e6319SBarry Smith    Level: developer
439d91e6319SBarry Smith 
440d91e6319SBarry Smith .seealso: MatILUFactorSymbolic(), MatILUFactor(), MatLUInfo, MatCholeskyInfo
441d91e6319SBarry Smith 
442d91e6319SBarry Smith S*/
4435ef9f2a5SBarry Smith typedef struct {
4445ef9f2a5SBarry Smith   double     levels;         /* ILU(levels) */
4455ef9f2a5SBarry Smith   double     fill;           /* expected fill; nonzeros in factored matrix/nonzeros in original matrix*/
4465ef9f2a5SBarry Smith   double     diagonal_fill;  /* force diagonal to fill in if initially not filled */
44736db0b34SBarry Smith   double     dt;             /* drop tolerance */
44836db0b34SBarry Smith   double     dtcol;          /* tolerance for pivoting */
44936db0b34SBarry Smith   double     dtcount;        /* maximum nonzeros to be allowed per row */
450d91e6319SBarry Smith   double     damping;        /* scaling of identity added to matrix to prevent zero pivots */
451d91e6319SBarry Smith   double     damp;           /* if is 1.0 and factorization fails, damp until successful */
4525ef9f2a5SBarry Smith } MatILUInfo;
4535ef9f2a5SBarry Smith 
454d91e6319SBarry Smith /*S
455d91e6319SBarry Smith    MatLUInfo - Data based into the matrix LU factorization routines
456d91e6319SBarry Smith 
457d91e6319SBarry Smith    In Fortran these are simply double precision arrays of size MAT_LUINFO_SIZE
458d91e6319SBarry Smith 
459d91e6319SBarry Smith    Notes: These are not usually directly used by users, instead use the PC type of LU
460d91e6319SBarry Smith           All entries are double precision.
461d91e6319SBarry Smith 
462d91e6319SBarry Smith    Level: developer
463d91e6319SBarry Smith 
464d91e6319SBarry Smith .seealso: MatLUFactorSymbolic(), MatILUInfo, MatCholeskyInfo
465d91e6319SBarry Smith 
466d91e6319SBarry Smith S*/
46714822f30SBarry Smith typedef struct {
46814822f30SBarry Smith   double     fill;    /* expected fill; nonzeros in factored matrix/nonzeros in original matrix */
46914822f30SBarry Smith   double     dtcol;   /* tolerance for pivoting; pivot if off_diagonal*dtcol > diagonal */
470d91e6319SBarry Smith   double     damping; /* scaling of identity added to matrix to prevent zero pivots */
471d91e6319SBarry Smith   double     damp;    /* if this is 1.0 and factorization fails, damp until successful */
47214822f30SBarry Smith } MatLUInfo;
47314822f30SBarry Smith 
474d91e6319SBarry Smith /*S
475d91e6319SBarry Smith    MatCholeskyInfo - Data based into the matrix Cholesky factorization routines
476d91e6319SBarry Smith 
477d91e6319SBarry Smith    In Fortran these are simply double precision arrays of size MAT_CHOLESKYINFO_SIZE
478d91e6319SBarry Smith 
479d91e6319SBarry Smith    Notes: These are not usually directly used by users, instead use the PC type of Cholesky
480d91e6319SBarry Smith           All entries are double precision.
481d91e6319SBarry Smith 
482d91e6319SBarry Smith    Level: developer
483d91e6319SBarry Smith 
484d91e6319SBarry Smith .seealso: MatCholeskyFactorSymbolic(), MatLUInfo, MatILUInfo
485d91e6319SBarry Smith 
486d91e6319SBarry Smith S*/
487ffa6d0a5SLois Curfman McInnes typedef struct {
488ffa6d0a5SLois Curfman McInnes   double     fill;    /* expected fill; nonzeros in factored matrix/nonzeros in original matrix */
489d91e6319SBarry Smith   double     damping; /* scaling of identity added to matrix to prevent zero pivots */
490d91e6319SBarry Smith   double     damp;    /* if this is 1.0 and factorization fails, damp until successful */
491ffa6d0a5SLois Curfman McInnes } MatCholeskyInfo;
492ffa6d0a5SLois Curfman McInnes 
49314822f30SBarry Smith EXTERN int MatLUFactor(Mat,IS,IS,MatLUInfo*);
494ca44d042SBarry Smith EXTERN int MatILUFactor(Mat,IS,IS,MatILUInfo*);
49514822f30SBarry Smith EXTERN int MatLUFactorSymbolic(Mat,IS,IS,MatLUInfo*,Mat*);
496ca44d042SBarry Smith EXTERN int MatILUFactorSymbolic(Mat,IS,IS,MatILUInfo*,Mat*);
4974d101231SSatish Balay EXTERN int MatICCFactorSymbolic(Mat,IS,double,int,Mat*);
4984d101231SSatish Balay EXTERN int MatICCFactor(Mat,IS,double,int);
499ca44d042SBarry Smith EXTERN int MatLUFactorNumeric(Mat,Mat*);
500ca44d042SBarry Smith EXTERN int MatILUDTFactor(Mat,MatILUInfo*,IS,IS,Mat *);
501a2ce50c7SBarry Smith 
502ca44d042SBarry Smith EXTERN int MatSolve(Mat,Vec,Vec);
503ca44d042SBarry Smith EXTERN int MatForwardSolve(Mat,Vec,Vec);
504ca44d042SBarry Smith EXTERN int MatBackwardSolve(Mat,Vec,Vec);
505ca44d042SBarry Smith EXTERN int MatSolveAdd(Mat,Vec,Vec,Vec);
506ca44d042SBarry Smith EXTERN int MatSolveTranspose(Mat,Vec,Vec);
507ca44d042SBarry Smith EXTERN int MatSolveTransposeAdd(Mat,Vec,Vec,Vec);
5088ed539a5SBarry Smith 
509ca44d042SBarry Smith EXTERN int MatSetUnfactored(Mat);
510bb5a7306SBarry Smith 
511bb1eb677SSatish Balay /*  MatSORType may be bitwise ORd together, so do not change the numbers */
512d91e6319SBarry Smith /*E
513d91e6319SBarry Smith     MatSORType - What type of (S)SOR to perform
514bb1eb677SSatish Balay 
515d91e6319SBarry Smith     Level: beginner
516d91e6319SBarry Smith 
517d9274352SBarry Smith    May be bitwise ORd together
518d9274352SBarry Smith 
519d91e6319SBarry Smith    Any additions/changes here MUST also be made in include/finclude/petscmat.h
520d91e6319SBarry Smith 
521d91e6319SBarry Smith .seealso: MatRelax()
522d91e6319SBarry Smith E*/
523ee50ffe9SBarry Smith typedef enum {SOR_FORWARD_SWEEP=1,SOR_BACKWARD_SWEEP=2,SOR_SYMMETRIC_SWEEP=3,
524ee50ffe9SBarry Smith               SOR_LOCAL_FORWARD_SWEEP=4,SOR_LOCAL_BACKWARD_SWEEP=8,
525ee50ffe9SBarry Smith               SOR_LOCAL_SYMMETRIC_SWEEP=12,SOR_ZERO_INITIAL_GUESS=16,
52684cb2905SBarry Smith               SOR_EISENSTAT=32,SOR_APPLY_UPPER=64,SOR_APPLY_LOWER=128} MatSORType;
527ca44d042SBarry Smith EXTERN int MatRelax(Mat,Vec,double,MatSORType,double,int,Vec);
5288ed539a5SBarry Smith 
529d4fbbf0eSBarry Smith /*
530639f9d9dSBarry Smith     These routines are for efficiently computing Jacobians via finite differences.
531639f9d9dSBarry Smith */
532b12f92e5SBarry Smith 
533d9274352SBarry Smith /*E
534d9274352SBarry Smith     MatColoringType - String with the name of a PETSc matrix coloring or the creation function
535d9274352SBarry Smith        with an optional dynamic library name, for example
536d9274352SBarry Smith        http://www.mcs.anl.gov/petsc/lib.a:coloringcreate()
537d9274352SBarry Smith 
538d9274352SBarry Smith    Level: beginner
539d9274352SBarry Smith 
540d9274352SBarry Smith .seealso: MatGetColoring()
541d9274352SBarry Smith E*/
542b12f92e5SBarry Smith typedef char* MatColoringType;
543b12f92e5SBarry Smith #define MATCOLORING_NATURAL "natural"
544b12f92e5SBarry Smith #define MATCOLORING_SL      "sl"
545b12f92e5SBarry Smith #define MATCOLORING_LF      "lf"
546b12f92e5SBarry Smith #define MATCOLORING_ID      "id"
547b12f92e5SBarry Smith 
548ca44d042SBarry Smith EXTERN int MatGetColoring(Mat,MatColoringType,ISColoring*);
549ca44d042SBarry Smith EXTERN int MatColoringRegister(char*,char*,char*,int(*)(Mat,MatColoringType,ISColoring *));
550aa482453SBarry Smith #if defined(PETSC_USE_DYNAMIC_LIBRARIES)
551f1af5d2fSBarry Smith #define MatColoringRegisterDynamic(a,b,c,d) MatColoringRegister(a,b,c,0)
552b12f92e5SBarry Smith #else
553f1af5d2fSBarry Smith #define MatColoringRegisterDynamic(a,b,c,d) MatColoringRegister(a,b,c,d)
554b12f92e5SBarry Smith #endif
555ca44d042SBarry Smith EXTERN int        MatColoringRegisterAll(char *);
5562bad1931SBarry Smith extern PetscTruth MatColoringRegisterAllCalled;
557ca44d042SBarry Smith EXTERN int        MatColoringRegisterDestroy(void);
558b9617806SBarry Smith EXTERN int        MatColoringPatch(Mat,int,int,int *,ISColoring*);
559639f9d9dSBarry Smith 
5601a0a18cdSSatish Balay #define MAT_FDCOLORING_COOKIE PETSC_COOKIE + 23
561d9274352SBarry Smith /*S
562d9274352SBarry Smith      MatFDColoring - Object for computing a sparse Jacobian via finite differences
563d9274352SBarry Smith         and coloring
564639f9d9dSBarry Smith 
565d9274352SBarry Smith    Level: beginner
566d9274352SBarry Smith 
567d9274352SBarry Smith   Concepts: coloring, sparse Jacobian, finite differences
568d9274352SBarry Smith 
569d9274352SBarry Smith .seealso:  MatFDColoringCreate()
570d9274352SBarry Smith S*/
571e2a1c21fSSatish Balay typedef struct _p_MatFDColoring *MatFDColoring;
572639f9d9dSBarry Smith 
573ca44d042SBarry Smith EXTERN int MatFDColoringCreate(Mat,ISColoring,MatFDColoring *);
574ca44d042SBarry Smith EXTERN int MatFDColoringDestroy(MatFDColoring);
575b0a32e0cSBarry Smith EXTERN int MatFDColoringView(MatFDColoring,PetscViewer);
576ca44d042SBarry Smith EXTERN int MatFDColoringSetFunction(MatFDColoring,int (*)(void),void*);
577ca44d042SBarry Smith EXTERN int MatFDColoringSetParameters(MatFDColoring,double,double);
578ca44d042SBarry Smith EXTERN int MatFDColoringSetFrequency(MatFDColoring,int);
579ca44d042SBarry Smith EXTERN int MatFDColoringGetFrequency(MatFDColoring,int*);
580ca44d042SBarry Smith EXTERN int MatFDColoringSetFromOptions(MatFDColoring);
581ca44d042SBarry Smith EXTERN int MatFDColoringApply(Mat,MatFDColoring,Vec,MatStructure*,void *);
582ca44d042SBarry Smith EXTERN int MatFDColoringApplyTS(Mat,MatFDColoring,double,Vec,MatStructure*,void *);
58362152c8bSBarry Smith EXTERN int MatFDColoringSetRecompute(MatFDColoring);
5843a7fca6bSBarry Smith EXTERN int MatFDColoringSetF(MatFDColoring,Vec);
585639f9d9dSBarry Smith 
586639f9d9dSBarry Smith /*
5870752156aSBarry Smith     These routines are for partitioning matrices: currently used only
5883eda8832SBarry Smith   for adjacency matrix, MatCreateMPIAdj().
5890752156aSBarry Smith */
59091e9ee9fSBarry Smith #define MATPARTITIONING_COOKIE PETSC_COOKIE + 25
591ca161407SBarry Smith 
592d9274352SBarry Smith /*S
593d9274352SBarry Smith      MatPartitioning - Object for managing the partitioning of a matrix or graph
594d9274352SBarry Smith 
595d9274352SBarry Smith    Level: beginner
596d9274352SBarry Smith 
597d9274352SBarry Smith   Concepts: partitioning
598d9274352SBarry Smith 
599d9274352SBarry Smith .seealso:  MatParitioningCreate(), MatPartitioningType
600d9274352SBarry Smith S*/
60191e9ee9fSBarry Smith typedef struct _p_MatPartitioning *MatPartitioning;
602d9274352SBarry Smith 
603d9274352SBarry Smith /*E
604d9274352SBarry Smith     MatPartitioningType - String with the name of a PETSc matrix partitioing or the creation function
605d9274352SBarry Smith        with an optional dynamic library name, for example
606d9274352SBarry Smith        http://www.mcs.anl.gov/petsc/lib.a:partitioningcreate()
607d9274352SBarry Smith 
608d9274352SBarry Smith    Level: beginner
609d9274352SBarry Smith 
610d9274352SBarry Smith .seealso: MatPartitioingCreate(), MatPartitioning
611d9274352SBarry Smith E*/
6122aabb6bbSBarry Smith typedef char* MatPartitioningType;
6132aabb6bbSBarry Smith #define MATPARTITIONING_CURRENT  "current"
6142aabb6bbSBarry Smith #define MATPARTITIONING_PARMETIS "parmetis"
615ca161407SBarry Smith 
616ca44d042SBarry Smith EXTERN int MatPartitioningCreate(MPI_Comm,MatPartitioning*);
617ca44d042SBarry Smith EXTERN int MatPartitioningSetType(MatPartitioning,MatPartitioningType);
618ca44d042SBarry Smith EXTERN int MatPartitioningSetAdjacency(MatPartitioning,Mat);
619ca44d042SBarry Smith EXTERN int MatPartitioningSetVertexWeights(MatPartitioning,int*);
620ca44d042SBarry Smith EXTERN int MatPartitioningApply(MatPartitioning,IS*);
621ca44d042SBarry Smith EXTERN int MatPartitioningDestroy(MatPartitioning);
6222aabb6bbSBarry Smith 
623ca44d042SBarry Smith EXTERN int MatPartitioningRegister(char*,char*,char*,int(*)(MatPartitioning));
624aa482453SBarry Smith #if defined(PETSC_USE_DYNAMIC_LIBRARIES)
625f1af5d2fSBarry Smith #define MatPartitioningRegisterDynamic(a,b,c,d) MatPartitioningRegister(a,b,c,0)
6262aabb6bbSBarry Smith #else
627f1af5d2fSBarry Smith #define MatPartitioningRegisterDynamic(a,b,c,d) MatPartitioningRegister(a,b,c,d)
6282aabb6bbSBarry Smith #endif
6292aabb6bbSBarry Smith 
630ca44d042SBarry Smith EXTERN int        MatPartitioningRegisterAll(char *);
6312bad1931SBarry Smith extern PetscTruth MatPartitioningRegisterAllCalled;
632ca44d042SBarry Smith EXTERN int        MatPartitioningRegisterDestroy(void);
6332bad1931SBarry Smith 
634b0a32e0cSBarry Smith EXTERN int MatPartitioningView(MatPartitioning,PetscViewer);
635ca44d042SBarry Smith EXTERN int MatPartitioningSetFromOptions(MatPartitioning);
636ca44d042SBarry Smith EXTERN int MatPartitioningGetType(MatPartitioning,MatPartitioningType*);
637ca161407SBarry Smith 
638ca44d042SBarry Smith EXTERN int MatPartitioningParmetisSetCoarseSequential(MatPartitioning);
6390752156aSBarry Smith 
6400752156aSBarry Smith /*
6410a835dfdSSatish Balay     If you add entries here you must also add them to finclude/petscmat.h
642d4fbbf0eSBarry Smith */
6431c1c02c0SLois Curfman McInnes typedef enum { MATOP_SET_VALUES=0,
6441c1c02c0SLois Curfman McInnes                MATOP_GET_ROW=1,
6451c1c02c0SLois Curfman McInnes                MATOP_RESTORE_ROW=2,
6461c1c02c0SLois Curfman McInnes                MATOP_MULT=3,
6471c1c02c0SLois Curfman McInnes                MATOP_MULT_ADD=4,
6487c922b88SBarry Smith                MATOP_MULT_TRANSPOSE=5,
6497c922b88SBarry Smith                MATOP_MULT_TRANSPOSE_ADD=6,
6501c1c02c0SLois Curfman McInnes                MATOP_SOLVE=7,
6511c1c02c0SLois Curfman McInnes                MATOP_SOLVE_ADD=8,
6527c922b88SBarry Smith                MATOP_SOLVE_TRANSPOSE=9,
6537c922b88SBarry Smith                MATOP_SOLVE_TRANSPOSE_ADD=10,
6541c1c02c0SLois Curfman McInnes                MATOP_LUFACTOR=11,
6551c1c02c0SLois Curfman McInnes                MATOP_CHOLESKYFACTOR=12,
6561c1c02c0SLois Curfman McInnes                MATOP_RELAX=13,
6571c1c02c0SLois Curfman McInnes                MATOP_TRANSPOSE=14,
6581c1c02c0SLois Curfman McInnes                MATOP_GETINFO=15,
6591c1c02c0SLois Curfman McInnes                MATOP_EQUAL=16,
6601c1c02c0SLois Curfman McInnes                MATOP_GET_DIAGONAL=17,
6611c1c02c0SLois Curfman McInnes                MATOP_DIAGONAL_SCALE=18,
6621c1c02c0SLois Curfman McInnes                MATOP_NORM=19,
6631c1c02c0SLois Curfman McInnes                MATOP_ASSEMBLY_BEGIN=20,
6641c1c02c0SLois Curfman McInnes                MATOP_ASSEMBLY_END=21,
6651c1c02c0SLois Curfman McInnes                MATOP_COMPRESS=22,
6661c1c02c0SLois Curfman McInnes                MATOP_SET_OPTION=23,
6671c1c02c0SLois Curfman McInnes                MATOP_ZERO_ENTRIES=24,
6681c1c02c0SLois Curfman McInnes                MATOP_ZERO_ROWS=25,
6691c1c02c0SLois Curfman McInnes                MATOP_LUFACTOR_SYMBOLIC=26,
6701c1c02c0SLois Curfman McInnes                MATOP_LUFACTOR_NUMERIC=27,
6711c1c02c0SLois Curfman McInnes                MATOP_CHOLESKY_FACTOR_SYMBOLIC=28,
6721c1c02c0SLois Curfman McInnes                MATOP_CHOLESKY_FACTOR_NUMERIC=29,
6731c1c02c0SLois Curfman McInnes                MATOP_GET_SIZE=30,
6741c1c02c0SLois Curfman McInnes                MATOP_GET_LOCAL_SIZE=31,
6751c1c02c0SLois Curfman McInnes                MATOP_GET_OWNERSHIP_RANGE=32,
6761c1c02c0SLois Curfman McInnes                MATOP_ILUFACTOR_SYMBOLIC=33,
6774d101231SSatish Balay                MATOP_ICCFACTOR_SYMBOLIC=34,
6781c1c02c0SLois Curfman McInnes                MATOP_GET_ARRAY=35,
6791c1c02c0SLois Curfman McInnes                MATOP_RESTORE_ARRAY=36,
6807bf97ca4SSatish Balay 
681005c665bSBarry Smith                MATOP_CONVERT_SAME_TYPE=37,
682005c665bSBarry Smith                MATOP_FORWARD_SOLVE=38,
683005c665bSBarry Smith                MATOP_BACKWARD_SOLVE=39,
684005c665bSBarry Smith                MATOP_ILUFACTOR=40,
6854d101231SSatish Balay                MATOP_ICCFACTOR=41,
686005c665bSBarry Smith                MATOP_AXPY=42,
687005c665bSBarry Smith                MATOP_GET_SUBMATRICES=43,
688005c665bSBarry Smith                MATOP_INCREASE_OVERLAP=44,
689005c665bSBarry Smith                MATOP_GET_VALUES=45,
690005c665bSBarry Smith                MATOP_COPY=46,
691005c665bSBarry Smith                MATOP_PRINT_HELP=47,
692005c665bSBarry Smith                MATOP_SCALE=48,
693005c665bSBarry Smith                MATOP_SHIFT=49,
694005c665bSBarry Smith                MATOP_DIAGONAL_SHIFT=50,
695005c665bSBarry Smith                MATOP_ILUDT_FACTOR=51,
696005c665bSBarry Smith                MATOP_GET_BLOCK_SIZE=52,
697005c665bSBarry Smith                MATOP_GET_ROW_IJ=53,
698005c665bSBarry Smith                MATOP_RESTORE_ROW_IJ=54,
699005c665bSBarry Smith                MATOP_GET_COLUMN_IJ=55,
700005c665bSBarry Smith                MATOP_RESTORE_COLUMN_IJ=56,
701005c665bSBarry Smith                MATOP_FDCOLORING_CREATE=57,
702005c665bSBarry Smith                MATOP_COLORING_PATCH=58,
703005c665bSBarry Smith                MATOP_SET_UNFACTORED=59,
704005c665bSBarry Smith                MATOP_PERMUTE=60,
705005c665bSBarry Smith                MATOP_SET_VALUES_BLOCKED=61,
7061c1c02c0SLois Curfman McInnes                MATOP_DESTROY=250,
7071c1c02c0SLois Curfman McInnes                MATOP_VIEW=251
708fae171e0SBarry Smith              } MatOperation;
709ca44d042SBarry Smith EXTERN int MatHasOperation(Mat,MatOperation,PetscTruth*);
71037bd1cefSSatish Balay EXTERN int MatShellSetOperation(Mat,MatOperation,void(*)());
71137bd1cefSSatish Balay EXTERN int MatShellGetOperation(Mat,MatOperation,void(**)());
712273d9f13SBarry Smith EXTERN int MatShellSetContext(Mat,void*);
713112a2221SBarry Smith 
71490ace30eSBarry Smith /*
71590ace30eSBarry Smith    Codes for matrices stored on disk. By default they are
71690ace30eSBarry Smith  stored in a universal format. By changing the format with
717fb9695e5SSatish Balay  PetscViewerSetFormat(viewer,PETSC_VIEWER_BINARY_NATIVE); the matrices will
71890ace30eSBarry Smith  be stored in a way natural for the matrix, for example dense matrices
71990ace30eSBarry Smith  would be stored as dense. Matrices stored this way may only be
72090ace30eSBarry Smith  read into matrices of the same time.
72190ace30eSBarry Smith */
72290ace30eSBarry Smith #define MATRIX_BINARY_FORMAT_DENSE -1
72390ace30eSBarry Smith 
7243f1d51d7SBarry Smith /*
7253f1d51d7SBarry Smith      New matrix classes not yet distributed
7263f1d51d7SBarry Smith */
7273f1d51d7SBarry Smith /*
7283f1d51d7SBarry Smith     MatAIJIndices is a data structure for storing the nonzero location information
7293f1d51d7SBarry Smith   for sparse matrices. Several matrices with identical nonzero structure can share
7303f1d51d7SBarry Smith   the same MatAIJIndices.
7313f1d51d7SBarry Smith */
732e2a1c21fSSatish Balay typedef struct _p_MatAIJIndices* MatAIJIndices;
7333f1d51d7SBarry Smith 
734ca44d042SBarry Smith EXTERN int MatCreateAIJIndices(int,int,int*,int*,PetscTruth,MatAIJIndices*);
735ca44d042SBarry Smith EXTERN int MatCreateAIJIndicesEmpty(int,int,int*,PetscTruth,MatAIJIndices*);
736ca44d042SBarry Smith EXTERN int MatAttachAIJIndices(MatAIJIndices,MatAIJIndices*);
737ca44d042SBarry Smith EXTERN int MatDestroyAIJIndices(MatAIJIndices);
738ca44d042SBarry Smith EXTERN int MatCopyAIJIndices(MatAIJIndices,MatAIJIndices*);
739ca44d042SBarry Smith EXTERN int MatValidateAIJIndices(int,MatAIJIndices);
740ca44d042SBarry Smith EXTERN int MatShiftAIJIndices(MatAIJIndices);
741ca44d042SBarry Smith EXTERN int MatShrinkAIJIndices(MatAIJIndices);
742ca44d042SBarry Smith EXTERN int MatTransposeAIJIndices(MatAIJIndices,MatAIJIndices*);
7433f1d51d7SBarry Smith 
744ca44d042SBarry Smith EXTERN int MatCreateSeqCSN(MPI_Comm,int,int,int*,int,Mat*);
745ca44d042SBarry Smith EXTERN int MatCreateSeqCSN_Single(MPI_Comm,int,int,int*,int,Mat*);
746ca44d042SBarry Smith EXTERN int MatCreateSeqCSNWithPrecision(MPI_Comm,int,int,int*,int,ScalarPrecision,Mat*);
7473f1d51d7SBarry Smith 
748ca44d042SBarry Smith EXTERN int MatCreateSeqCSNIndices(MPI_Comm,MatAIJIndices,int,Mat *);
749ca44d042SBarry Smith EXTERN int MatCreateSeqCSNIndices_Single(MPI_Comm,MatAIJIndices,int,Mat *);
750ca44d042SBarry Smith EXTERN int MatCreateSeqCSNIndicesWithPrecision(MPI_Comm,MatAIJIndices,int,ScalarPrecision,Mat *);
7513f1d51d7SBarry Smith 
7526d053be9SSatish Balay EXTERN int MatMPIBAIJSetHashTableFactor(Mat,PetscReal);
753ca44d042SBarry Smith EXTERN int MatSeqAIJGetInodeSizes(Mat,int *,int *[],int *);
75408918a0eSSatish Balay EXTERN int MatMPIRowbsGetColor(Mat,ISColoring *);
755860d1616SSatish Balay 
756d9274352SBarry Smith /*S
757d9274352SBarry Smith      MatNullSpace - Object that removes a null space from a vector, i.e.
758d9274352SBarry Smith          orthogonalizes the vector to a subsapce
759d9274352SBarry Smith 
760d9274352SBarry Smith    Level: beginner
761d9274352SBarry Smith 
762d9274352SBarry Smith   Concepts: matrix; linear operator, null space
763d9274352SBarry Smith 
764d9274352SBarry Smith .seealso:  MatNullSpaceCreate()
765d9274352SBarry Smith S*/
76674637425SBarry Smith typedef struct _p_MatNullSpace* MatNullSpace;
767d9274352SBarry Smith 
76874637425SBarry Smith #define MATNULLSPACE_COOKIE    PETSC_COOKIE+17
76974637425SBarry Smith 
77074637425SBarry Smith EXTERN int MatNullSpaceCreate(MPI_Comm,int,int,Vec *,MatNullSpace*);
77174637425SBarry Smith EXTERN int MatNullSpaceDestroy(MatNullSpace);
77274637425SBarry Smith EXTERN int MatNullSpaceRemove(MatNullSpace,Vec,Vec*);
77374637425SBarry Smith EXTERN int MatNullSpaceAttach(Mat,MatNullSpace);
77474637425SBarry Smith EXTERN int MatNullSpaceTest(MatNullSpace,Mat);
77574637425SBarry Smith 
776273d9f13SBarry Smith EXTERN int MatReorderingSeqSBAIJ(Mat A,IS isp);
777273d9f13SBarry Smith EXTERN int MatMPISBAIJSetHashTableFactor(Mat,PetscReal);
778273d9f13SBarry Smith EXTERN int MatSeqSBAIJSetColumnIndices(Mat,int *);
779273d9f13SBarry Smith 
7803f1d51d7SBarry Smith 
781f069c275SSatish Balay EXTERN int MatCreateMAIJ(Mat,int,Mat*);
782c4f061fbSSatish Balay EXTERN int MatMAIJRedimension(Mat,int,Mat*);
783c4f061fbSSatish Balay EXTERN int MatMAIJGetAIJ(Mat,Mat*);
784c4f061fbSSatish Balay 
785273d9f13SBarry Smith EXTERN int MatMPIAdjSetValues(Mat,int*,int*,int*);
786f069c275SSatish Balay 
787b0a32e0cSBarry Smith EXTERN int MatComputeExplicitOperator(Mat,Mat*);
788b0a32e0cSBarry Smith 
7892eac72dbSBarry Smith #endif
7902eac72dbSBarry Smith 
7912eac72dbSBarry Smith 
7929d00d63dSBarry Smith 
793