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 1419caf8f3SSatish 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; 17*5975b3b6SBarry Smith *ierr = MatMPIBAIJGetSeqBAIJ(*A, Ad, Ao, &i); 18*5975b3b6SBarry Smith if (*ierr) return; 191a83f524SJed Brown *iic = PetscIntAddressToFortran(ic, (PetscInt *)i); 20f4e70085SSatish Balay } 21f4e70085SSatish Balay 22*5975b3b6SBarry Smith PETSC_EXTERN void matcreatebaij_(MPI_Comm *comm, PetscInt *bs, PetscInt *m, PetscInt *n, PetscInt *M, PetscInt *N, 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 2919caf8f3SSatish 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 } 35