1 #include <petsc-private/fortranimpl.h> 2 #include <petscmat.h> 3 4 #if defined(PETSC_HAVE_FORTRAN_CAPS) 5 #define matmpibaijgetseqbaij_ MATMPIBAIJGETSEQBAIJ 6 #define matcreatebaij_ MATCREATEBAIJ 7 #define matmpibaijsetpreallocation_ MATMPIBAIJSETPREALLOCATION 8 #elif !defined(PETSC_HAVE_FORTRAN_UNDERSCORE) 9 #define matmpibaijgetseqbaij_ matmpibaijgetseqbaij 10 #define matcreatebaij_ matcreatebaij 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 const PetscInt *i; 19 *ierr = MatMPIBAIJGetSeqBAIJ(*A,Ad,Ao,&i);if (*ierr) return; 20 *iic = PetscIntAddressToFortran(ic,(PetscInt*)i); 21 } 22 23 void PETSC_STDCALL matcreatebaij_(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 = MatCreateBAIJ(MPI_Comm_f2c(*(MPI_Fint*)&*comm),*bs,*m,*n,*M,*N,*d_nz,d_nnz,*o_nz,o_nnz,newmat); 29 } 30 31 void PETSC_STDCALL matmpibaijsetpreallocation_(Mat *mat,PetscInt *bs,PetscInt *d_nz,PetscInt *d_nnz,PetscInt *o_nz,PetscInt *o_nnz,PetscErrorCode *ierr) 32 { 33 CHKFORTRANNULLINTEGER(d_nnz); 34 CHKFORTRANNULLINTEGER(o_nnz); 35 *ierr = MatMPIBAIJSetPreallocation(*mat,*bs,*d_nz,d_nnz,*o_nz,o_nnz); 36 } 37 38 EXTERN_C_END 39