Actual source code: pythonmat.c
1: #include <petsc/private/matimpl.h>
3: /*@C
4: MatPythonSetType - Initialize a Mat object implemented in Python.
6: Collective on Mat
8: Input Parameters:
9: + mat - the matrix (Mat) object.
10: - pyname - full dotted Python name [package].module[.{class|function}]
12: Options Database Key:
13: . -mat_python_type <pyname> - python class
15: Level: intermediate
17: .seealso: MatCreate(), MatSetType(), MATPYTHON, PetscPythonInitialize()
18: @*/
19: PetscErrorCode MatPythonSetType(Mat mat,const char pyname[])
20: {
23: PetscTryMethod(mat,"MatPythonSetType_C",(Mat, const char[]),(mat,pyname));
24: return 0;
25: }
27: /*@C
28: MatPythonCreate - Create a Mat object implemented in Python.
30: Collective on Mat
32: Input Parameters:
33: + comm - MPI communicator
34: . m - number of local rows (or PETSC_DECIDE to have calculated if M is given)
35: . n - number of local columns (or PETSC_DECIDE to have calculated if N is given)
36: . M - number of global rows (or PETSC_DECIDE to have calculated if m is given)
37: . N - number of global columns (or PETSC_DECIDE to have calculated if n is given)
38: - pyname - full dotted Python name [package].module[.{class|function}]
40: Output Parameter:
41: . A - the matrix
43: Level: intermediate
45: .seealso: MATPYTHON, MatPythonSetType(), PetscPythonInitialize()
47: @*/
48: PetscErrorCode MatPythonCreate(MPI_Comm comm,PetscInt m,PetscInt n,PetscInt M,PetscInt N,const char pyname[],Mat *A)
49: {
52: MatCreate(comm,A);
53: MatSetSizes(*A,m,n,M,N);
54: MatSetType(*A,MATPYTHON);
55: MatPythonSetType(*A,pyname);
56: MatBindToCPU(*A,PETSC_FALSE);
57: return 0;
58: }