xref: /petsc/src/mat/impls/baij/mpi/ftn-custom/zmpibaijf.c (revision 19caf8f3c08b1f0ca9f5469bde385c134aa76c82)
1af0996ceSBarry Smith #include <petsc/private/fortranimpl.h>
2c6db04a5SJed Brown #include <petscmat.h>
3f4e70085SSatish Balay 
4f4e70085SSatish Balay #if defined(PETSC_HAVE_FORTRAN_CAPS)
5f4e70085SSatish Balay #define matmpibaijgetseqbaij_            MATMPIBAIJGETSEQBAIJ
669b1f4b7SBarry Smith #define matcreatebaij_                   MATCREATEBAIJ
7f4e70085SSatish Balay #define matmpibaijsetpreallocation_      MATMPIBAIJSETPREALLOCATION
8f4e70085SSatish Balay #elif !defined(PETSC_HAVE_FORTRAN_UNDERSCORE)
9f4e70085SSatish Balay #define matmpibaijgetseqbaij_            matmpibaijgetseqbaij
1069b1f4b7SBarry Smith #define matcreatebaij_                   matcreatebaij
11f4e70085SSatish Balay #define matmpibaijsetpreallocation_      matmpibaijsetpreallocation
12f4e70085SSatish Balay #endif
13f4e70085SSatish Balay 
14*19caf8f3SSatish Balay PETSC_EXTERN void matmpibaijgetseqbaij_(Mat *A,Mat *Ad,Mat *Ao,PetscInt *ic,size_t *iic,PetscErrorCode *ierr)
15f4e70085SSatish Balay {
161a83f524SJed Brown   const PetscInt *i;
17f4e70085SSatish Balay   *ierr = MatMPIBAIJGetSeqBAIJ(*A,Ad,Ao,&i);if (*ierr) return;
181a83f524SJed Brown   *iic  = PetscIntAddressToFortran(ic,(PetscInt*)i);
19f4e70085SSatish Balay }
20f4e70085SSatish Balay 
21*19caf8f3SSatish Balay PETSC_EXTERN void matcreatebaij_(MPI_Comm *comm,PetscInt *bs,PetscInt *m,PetscInt *n,PetscInt *M,PetscInt *N,
22f4e70085SSatish Balay          PetscInt *d_nz,PetscInt *d_nnz,PetscInt *o_nz,PetscInt *o_nnz,Mat *newmat,PetscErrorCode *ierr)
23f4e70085SSatish Balay {
24f4e70085SSatish Balay   CHKFORTRANNULLINTEGER(d_nnz);
25f4e70085SSatish Balay   CHKFORTRANNULLINTEGER(o_nnz);
2669b1f4b7SBarry Smith   *ierr = MatCreateBAIJ(MPI_Comm_f2c(*(MPI_Fint*)&*comm),*bs,*m,*n,*M,*N,*d_nz,d_nnz,*o_nz,o_nnz,newmat);
27f4e70085SSatish Balay }
28f4e70085SSatish Balay 
29*19caf8f3SSatish Balay PETSC_EXTERN void matmpibaijsetpreallocation_(Mat *mat,PetscInt *bs,PetscInt *d_nz,PetscInt *d_nnz,PetscInt *o_nz,PetscInt *o_nnz,PetscErrorCode *ierr)
30f4e70085SSatish Balay {
31f4e70085SSatish Balay   CHKFORTRANNULLINTEGER(d_nnz);
32f4e70085SSatish Balay   CHKFORTRANNULLINTEGER(o_nnz);
33f4e70085SSatish Balay   *ierr = MatMPIBAIJSetPreallocation(*mat,*bs,*d_nz,d_nnz,*o_nz,o_nnz);
34f4e70085SSatish Balay }
35f4e70085SSatish Balay 
36