xref: /petsc/src/mat/impls/baij/mpi/ftn-custom/zmpibaijf.c (revision 09f3b4e5628a00a1eaf17d80982cfbcc515cc9c1)
1 #include "zpetsc.h"
2 #include "petscmat.h"
3 
4 #if defined(PETSC_HAVE_FORTRAN_CAPS)
5 #define matmpibaijgetseqbaij_            MATMPIBAIJGETSEQBAIJ
6 #define matcreatempibaij_                MATCREATEMPIBAIJ
7 #define matmpibaijsetpreallocation_      MATMPIBAIJSETPREALLOCATION
8 #elif !defined(PETSC_HAVE_FORTRAN_UNDERSCORE)
9 #define matmpibaijgetseqbaij_            matmpibaijgetseqbaij
10 #define matcreatempibaij_                matcreatempibaij
11 #define matmpibaijsetpreallocation_      matmpibaijsetpreallocation
12 #endif
13 
14 EXTERN_C_BEGIN
15 
16 void PETSC_STDCALL matmpibaijgetseqbaij_(Mat *A,Mat *Ad,Mat *Ao,PetscInt *ic,size_t *iic,PetscErrorCode *ierr)
17 {
18   PetscInt *i;
19   *ierr = MatMPIBAIJGetSeqBAIJ(*A,Ad,Ao,&i);if (*ierr) return;
20   *iic  = PetscIntAddressToFortran(ic,i);
21 }
22 
23 void PETSC_STDCALL matcreatempibaij_(MPI_Comm *comm,PetscInt *bs,PetscInt *m,PetscInt *n,PetscInt *M,PetscInt *N,
24          PetscInt *d_nz,PetscInt *d_nnz,PetscInt *o_nz,PetscInt *o_nnz,Mat *newmat,PetscErrorCode *ierr)
25 {
26   CHKFORTRANNULLINTEGER(d_nnz);
27   CHKFORTRANNULLINTEGER(o_nnz);
28   *ierr = MatCreateMPIBAIJ((MPI_Comm)PetscToPointerComm(*comm),
29                              *bs,*m,*n,*M,*N,*d_nz,d_nnz,*o_nz,o_nnz,newmat);
30 }
31 
32 void PETSC_STDCALL matmpibaijsetpreallocation_(Mat *mat,PetscInt *bs,PetscInt *d_nz,PetscInt *d_nnz,PetscInt *o_nz,PetscInt *o_nnz,PetscErrorCode *ierr)
33 {
34   CHKFORTRANNULLINTEGER(d_nnz);
35   CHKFORTRANNULLINTEGER(o_nnz);
36   *ierr = MatMPIBAIJSetPreallocation(*mat,*bs,*d_nz,d_nnz,*o_nz,o_nnz);
37 }
38 
39 EXTERN_C_END
40