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