1 #include <petsc/private/f90impl.h> 2 #include <petsc/private/sfimpl.h> 3 4 #if defined(PETSC_HAVE_FORTRAN_CAPS) 5 #define petscsfview_ PETSCSFVIEW 6 #define petscsfgetgraph_ PETSCSFGETGRAPH 7 #define petscsfbcastbegin_ PETSCSFBCASTBEGIN 8 #define petscsfbcastend_ PETSCSFBCASTEND 9 #define f90arraysfnodecreate_ F90ARRAYSFNODECREATE 10 #define petscsfviewfromoptions_ PETSCSFVIEWFROMOPTIONS 11 #define petscsfdestroy_ PETSCSFDESTROY 12 #define petscsfsetgraph_ PETSCSFSETGRAPH 13 #define petscsfgetleafranks_ PETSCSFGETLEAFRANKS 14 #define petscsfgetrootranks_ PETSCSFGETROOTRANKS 15 #elif !defined(PETSC_HAVE_FORTRAN_UNDERSCORE) 16 #define petscsfgetgraph_ petscsfgetgraph 17 #define petscsfview_ petscsfview 18 #define petscsfbcastbegin_ petscsfbcastbegin 19 #define petscsfbcastend_ petscsfbcastend 20 #define f90arraysfnodecreate_ f90arraysfnodecreate 21 #define petscsfviewfromoptions_ petscsfviewfromoptions 22 #define petscsfdestroy_ petscsfdestroy 23 #define petscsfsetgraph_ petscsfsetgraph 24 #define petscsfgetleafranks_ petscsfgetleafranks 25 #define petscsfgetrootranks_ petscsfgetrootranks 26 #endif 27 28 PETSC_EXTERN void f90arraysfnodecreate_(const PetscInt *,PetscInt *,void * PETSC_F90_2PTR_PROTO_NOVAR); 29 30 PETSC_EXTERN void petscsfsetgraph_(PetscSF *sf,PetscInt *nroots,PetscInt *nleaves, PetscInt *ilocal,PetscCopyMode *localmode, PetscSFNode *iremote,PetscCopyMode *remotemode, int *ierr) 31 { 32 if (ilocal == PETSC_NULL_INTEGER_Fortran) ilocal = NULL; 33 *ierr = PetscSFSetGraph(*sf,*nroots,*nleaves,ilocal,*localmode,iremote,*remotemode); 34 } 35 36 PETSC_EXTERN void petscsfview_(PetscSF *sf, PetscViewer *vin, PetscErrorCode *ierr) 37 { 38 PetscViewer v; 39 40 PetscPatchDefaultViewers_Fortran(vin, v); 41 *ierr = PetscSFView(*sf, v); 42 } 43 44 PETSC_EXTERN void petscsfgetgraph_(PetscSF *sf,PetscInt *nroots,PetscInt *nleaves, F90Array1d *ailocal, F90Array1d *airemote, PetscErrorCode *ierr PETSC_F90_2PTR_PROTO(pilocal) PETSC_F90_2PTR_PROTO(piremote)) 45 { 46 const PetscInt *ilocal; 47 const PetscSFNode *iremote; 48 PetscInt nl; 49 50 *ierr = PetscSFGetGraph(*sf,nroots,nleaves,&ilocal,&iremote);if (*ierr) return; 51 nl = *nleaves; 52 if (!ilocal) nl = 0; 53 *ierr = F90Array1dCreate((void*)ilocal,MPIU_INT,1,nl, ailocal PETSC_F90_2PTR_PARAM(pilocal)); 54 /* this creates a memory leak */ 55 f90arraysfnodecreate_((PetscInt*)iremote,nleaves, airemote PETSC_F90_2PTR_PARAM(piremote)); 56 } 57 58 PETSC_EXTERN void petscsfgetleafranks_(PetscSF *sf, PetscInt *niranks, F90Array1d *airanks, F90Array1d *aioffset, F90Array1d *airootloc, PetscErrorCode *ierr PETSC_F90_2PTR_PROTO(piranks) PETSC_F90_2PTR_PROTO(pioffset) PETSC_F90_2PTR_PROTO(pirootloc)) 59 { 60 const PetscMPIInt *iranks = NULL; 61 const PetscInt *ioffset = NULL; 62 const PetscInt *irootloc = NULL; 63 64 *ierr = PetscSFGetLeafRanks(*sf, niranks, &iranks, &ioffset, &irootloc);if (*ierr) return; 65 *ierr = F90Array1dCreate((void*)irootloc, MPIU_INT, 1, ioffset[*niranks], airootloc PETSC_F90_2PTR_PARAM(pirootloc));if (*ierr) return; 66 *ierr = F90Array1dCreate((void*)iranks, MPI_INT, 1, *niranks, airanks PETSC_F90_2PTR_PARAM(piranks));if (*ierr) return; 67 *ierr = F90Array1dCreate((void*)ioffset, MPIU_INT, 1, *niranks+1, aioffset PETSC_F90_2PTR_PARAM(pioffset));if (*ierr) return; 68 } 69 70 PETSC_EXTERN void petscsfgetrootranks_(PetscSF *sf, PetscInt *nranks, F90Array1d *aranks, F90Array1d *aroffset, F90Array1d *armine, F90Array1d *arremote, 71 PetscErrorCode *ierr PETSC_F90_2PTR_PROTO(pranks) PETSC_F90_2PTR_PROTO(proffset) PETSC_F90_2PTR_PROTO(prmine) PETSC_F90_2PTR_PROTO(prremote)) 72 { 73 const PetscMPIInt *ranks = NULL; 74 const PetscInt *roffset = NULL; 75 const PetscInt *rmine = NULL; 76 const PetscInt *rremote = NULL; 77 78 *ierr = PetscSFGetRootRanks(*sf, nranks, &ranks, &roffset, &rmine, &rremote);if (*ierr) return; 79 *ierr = F90Array1dCreate((void*)ranks, MPI_INT, 1, *nranks, aranks PETSC_F90_2PTR_PARAM(pranks));if (*ierr) return; 80 *ierr = F90Array1dCreate((void*)roffset, MPIU_INT, 1, *nranks+1, aroffset PETSC_F90_2PTR_PARAM(proffset));if (*ierr) return; 81 *ierr = F90Array1dCreate((void*)rmine, MPIU_INT, 1, roffset[*nranks], armine PETSC_F90_2PTR_PARAM(prmine));if (*ierr) return; 82 *ierr = F90Array1dCreate((void*)rremote, MPIU_INT, 1, roffset[*nranks], arremote PETSC_F90_2PTR_PARAM(prremote));if (*ierr) return; 83 } 84 85 #if defined(PETSC_HAVE_F90_ASSUMED_TYPE_NOT_PTR) 86 PETSC_EXTERN void petscsfbcastbegin_(PetscSF *sf, MPI_Fint *unit, const void *rptr, void *lptr, MPI_Fint *op, PetscErrorCode *ierr) 87 { 88 MPI_Datatype dtype; 89 MPI_Op cop = MPI_Op_f2c(*op); 90 91 *ierr = PetscMPIFortranDatatypeToC(*unit,&dtype);if (*ierr) return; 92 *ierr = PetscSFBcastBegin(*sf, dtype, rptr, lptr, cop); 93 } 94 95 PETSC_EXTERN void petscsfbcastend_(PetscSF *sf, MPI_Fint *unit, const void *rptr, void *lptr, MPI_Fint *op, PetscErrorCode *ierr) 96 { 97 MPI_Datatype dtype; 98 MPI_Op cop = MPI_Op_f2c(*op); 99 100 *ierr = PetscMPIFortranDatatypeToC(*unit,&dtype);if (*ierr) return; 101 *ierr = PetscSFBcastEnd(*sf, dtype, rptr, lptr, cop); 102 } 103 104 #else 105 106 PETSC_EXTERN void petscsfbcastbegin_(PetscSF *sf, MPI_Fint *unit,F90Array1d *rptr, F90Array1d *lptr, MPI_Fint *op, PetscErrorCode *ierr PETSC_F90_2PTR_PROTO(rptrd) PETSC_F90_2PTR_PROTO(lptrd)) 107 { 108 MPI_Datatype dtype; 109 const void *rootdata; 110 void *leafdata; 111 MPI_Op cop = MPI_Op_f2c(*op); 112 113 *ierr = PetscMPIFortranDatatypeToC(*unit,&dtype);if (*ierr) return; 114 *ierr = F90Array1dAccess(rptr, dtype, (void**) &rootdata PETSC_F90_2PTR_PARAM(rptrd));if (*ierr) return; 115 *ierr = F90Array1dAccess(lptr, dtype, (void**) &leafdata PETSC_F90_2PTR_PARAM(lptrd));if (*ierr) return; 116 *ierr = PetscSFBcastBegin(*sf, dtype, rootdata, leafdata, cop); 117 } 118 119 PETSC_EXTERN void petscsfbcastend_(PetscSF *sf, MPI_Fint *unit,F90Array1d *rptr, F90Array1d *lptr, MPI_Fint *op, PetscErrorCode *ierr PETSC_F90_2PTR_PROTO(rptrd) PETSC_F90_2PTR_PROTO(lptrd)) 120 { 121 MPI_Datatype dtype; 122 const void *rootdata; 123 void *leafdata; 124 MPI_Op cop = MPI_Op_f2c(*op); 125 126 *ierr = PetscMPIFortranDatatypeToC(*unit,&dtype);if (*ierr) return; 127 *ierr = F90Array1dAccess(rptr, dtype, (void**) &rootdata PETSC_F90_2PTR_PARAM(rptrd));if (*ierr) return; 128 *ierr = F90Array1dAccess(lptr, dtype, (void**) &leafdata PETSC_F90_2PTR_PARAM(lptrd));if (*ierr) return; 129 *ierr = PetscSFBcastEnd(*sf, dtype, rootdata, leafdata, cop); 130 } 131 PETSC_EXTERN void petscsfviewfromoptions_(PetscSF *ao,PetscObject obj,char* type,PetscErrorCode *ierr,PETSC_FORTRAN_CHARLEN_T len) 132 { 133 char *t; 134 135 FIXCHAR(type,len,t); 136 CHKFORTRANNULLOBJECT(obj); 137 *ierr = PetscSFViewFromOptions(*ao,obj,t);if (*ierr) return; 138 FREECHAR(type,t); 139 } 140 141 PETSC_EXTERN void petscsfdestroy_(PetscSF *x,int *ierr) 142 { 143 PETSC_FORTRAN_OBJECT_F_DESTROYED_TO_C_NULL(x); 144 *ierr = PetscSFDestroy(x); if (*ierr) return; 145 PETSC_FORTRAN_OBJECT_C_NULL_TO_F_DESTROYED(x); 146 } 147 148 #endif 149