xref: /petsc/src/dm/impls/composite/ftn-custom/zfddaf.c (revision 8cc058d9cd56c1ccb3be12a47760ddfc446aaffc)
147c6ae99SBarry Smith 
2b45d2f2cSJed Brown #include <petsc-private/fortranimpl.h>
3c6db04a5SJed Brown #include <petscdmcomposite.h>
447c6ae99SBarry Smith 
547c6ae99SBarry Smith #if defined(PETSC_HAVE_FORTRAN_CAPS)
647c6ae99SBarry Smith #define dmcompositegetentries1_      DMCOMPOSITEGETENTRIES1
747c6ae99SBarry Smith #define dmcompositegetentries2_      DMCOMPOSITEGETENTRIES2
847c6ae99SBarry Smith #define dmcompositegetentries3_      DMCOMPOSITEGETENTRIES3
947c6ae99SBarry Smith #define dmcompositegetentries4_      DMCOMPOSITEGETENTRIES4
1047c6ae99SBarry Smith #define dmcompositegetentries5_      DMCOMPOSITEGETENTRIES5
1147c6ae99SBarry Smith #define dmcompositecreate_           DMCOMPOSITECREATE
1247c6ae99SBarry Smith #define dmcompositeadddm_            DMCOMPOSITEADDDM
1347c6ae99SBarry Smith #define dmcompositedestroy_          DMCOMPOSITEDESTROY
1447c6ae99SBarry Smith #define dmcompositegetaccess4_       DMCOMPOSITEGETACCESS4
1547c6ae99SBarry Smith #define dmcompositescatter4_         DMCOMPOSITESCATTER4
1647c6ae99SBarry Smith #define dmcompositerestoreaccess4_   DMCOMPOSITERESTOREACCESS4
1747c6ae99SBarry Smith #define dmcompositegetlocalvectors4_ DMCOMPOSITEGETLOCALVECTORS4
1847c6ae99SBarry Smith #define dmcompositerestorelocalvectors4_ DMCOMPOSITERESTORELOCALVECTORS4
1973e31fe2SJed Brown #define dmcompositegetglobaliss_     DMCOMPOSITEGETGLOBALISS
2073e31fe2SJed Brown #define dmcompositegetlocaliss_      DMCOMPOSITEGETLOCALISS
21f73e5cebSJed Brown #define dmcompositegetaccessarray_   DMCOMPOSITEGETACCESSARRAY
22f73e5cebSJed Brown #define dmcompositerestoreaccessarray_   DMCOMPOSITERESTOREACCESSARRAY
2347c6ae99SBarry Smith #elif !defined(PETSC_HAVE_FORTRAN_UNDERSCORE)
2447c6ae99SBarry Smith #define dmcompositegetentries1_      dmcompositegetentries1
2547c6ae99SBarry Smith #define dmcompositegetentries2_      dmcompositegetentries2
2647c6ae99SBarry Smith #define dmcompositegetentries3_      dmcompositegetentries3
2747c6ae99SBarry Smith #define dmcompositegetentries4_      dmcompositegetentries4
2847c6ae99SBarry Smith #define dmcompositegetentries5_      dmcompositegetentries5
2947c6ae99SBarry Smith #define dmcompositecreate_           dmcompositecreate
3047c6ae99SBarry Smith #define dmcompositeadddm_            dmcompositeadddm
3147c6ae99SBarry Smith #define dmcompositedestroy_          dmcompositedestroy
3247c6ae99SBarry Smith #define dmcompositegetaccess4_       dmcompositegetaccess4
3347c6ae99SBarry Smith #define dmcompositescatter4_         dmcompositescatter4
3447c6ae99SBarry Smith #define dmcompositerestoreaccess4_   dmcompositerestoreaccess4
3547c6ae99SBarry Smith #define dmcompositegetlocalvectors4_ dmcompositegetlocalvectors4
3647c6ae99SBarry Smith #define dmcompositerestorelocalvectors4_ dmcompositerestorelocalvectors4
3773e31fe2SJed Brown #define dmcompositegetglobaliss_     dmcompositegetglobaliss
3873e31fe2SJed Brown #define dmcompositegetlocaliss_      dmcompositegetlocaliss
39f73e5cebSJed Brown #define dmcompositegetaccessarray_   dmcompositegetaccessarray
40f73e5cebSJed Brown #define dmcompositerestoreaccessarray_   dmcompositerestoreaccessarray
4147c6ae99SBarry Smith #endif
4247c6ae99SBarry Smith 
43*8cc058d9SJed Brown PETSC_EXTERN void PETSC_STDCALL dmcompositegetentries1_(DM *dm,DM *da1,PetscErrorCode *ierr)
4447c6ae99SBarry Smith {
4547c6ae99SBarry Smith   *ierr = DMCompositeGetEntries(*dm,da1);
4647c6ae99SBarry Smith }
4747c6ae99SBarry Smith 
48*8cc058d9SJed Brown PETSC_EXTERN void PETSC_STDCALL dmcompositegetentries2_(DM *dm,DM *da1,DM *da2,PetscErrorCode *ierr)
4947c6ae99SBarry Smith {
5047c6ae99SBarry Smith   *ierr = DMCompositeGetEntries(*dm,da1,da2);
5147c6ae99SBarry Smith }
5247c6ae99SBarry Smith 
53*8cc058d9SJed Brown PETSC_EXTERN void PETSC_STDCALL dmcompositegetentries3_(DM *dm,DM *da1,DM *da2,DM *da3,PetscErrorCode *ierr)
5447c6ae99SBarry Smith {
5547c6ae99SBarry Smith   *ierr = DMCompositeGetEntries(*dm,da1,da2,da3);
5647c6ae99SBarry Smith }
5747c6ae99SBarry Smith 
58*8cc058d9SJed Brown PETSC_EXTERN void PETSC_STDCALL dmcompositegetentries4_(DM *dm,DM *da1,DM *da2,DM *da3,DM *da4,PetscErrorCode *ierr)
5947c6ae99SBarry Smith {
6047c6ae99SBarry Smith   *ierr = DMCompositeGetEntries(*dm,da1,da2,da3,da4);
6147c6ae99SBarry Smith }
6247c6ae99SBarry Smith 
63*8cc058d9SJed Brown PETSC_EXTERN void PETSC_STDCALL dmcompositegetentries5_(DM *dm,DM *da1,DM *da2,DM *da3,DM *da4,DM *da5,PetscErrorCode *ierr)
6447c6ae99SBarry Smith {
6547c6ae99SBarry Smith   *ierr = DMCompositeGetEntries(*dm,da1,da2,da3,da4,da5);
6647c6ae99SBarry Smith }
6747c6ae99SBarry Smith 
68*8cc058d9SJed Brown PETSC_EXTERN void PETSC_STDCALL dmcompositecreate_(MPI_Fint * comm,DM *A, int *ierr)
69c30958fdSKarl Rupp {
7047c6ae99SBarry Smith   *ierr = DMCompositeCreate(MPI_Comm_f2c(*(comm)),A);
7147c6ae99SBarry Smith }
7247c6ae99SBarry Smith 
73*8cc058d9SJed Brown PETSC_EXTERN void PETSC_STDCALL dmcompositeadddm_(DM *dm,DM *da,PetscErrorCode *ierr)
7447c6ae99SBarry Smith {
7547c6ae99SBarry Smith   *ierr = DMCompositeAddDM(*dm,*da);
7647c6ae99SBarry Smith }
7747c6ae99SBarry Smith 
78*8cc058d9SJed Brown PETSC_EXTERN void PETSC_STDCALL dmcompositedestroy_(DM *dm,PetscErrorCode *ierr)
7947c6ae99SBarry Smith {
80fcfd50ebSBarry Smith   *ierr = DMDestroy(dm);
8147c6ae99SBarry Smith }
8247c6ae99SBarry Smith 
83*8cc058d9SJed Brown PETSC_EXTERN void PETSC_STDCALL dmcompositegetaccess4_(DM *dm,Vec *v,void **v1,void **p1,void **v2,void **p2,PetscErrorCode *ierr)
8447c6ae99SBarry Smith {
8547c6ae99SBarry Smith   Vec *vv1 = (Vec*)v1,*vv2 = (Vec*)v2;
8647c6ae99SBarry Smith   *ierr = DMCompositeGetAccess(*dm,*v,vv1,(PetscScalar*)p1,vv2,(PetscScalar*)p2);
8747c6ae99SBarry Smith }
8847c6ae99SBarry Smith 
89*8cc058d9SJed Brown PETSC_EXTERN void PETSC_STDCALL dmcompositescatter4_(DM *dm,Vec *v,void *v1,void *p1,void *v2,void *p2,PetscErrorCode *ierr)
9047c6ae99SBarry Smith {
9147c6ae99SBarry Smith   Vec *vv1 = (Vec*)v1,*vv2 = (Vec*)v2;
9247c6ae99SBarry Smith   *ierr = DMCompositeScatter(*dm,*v,*vv1,(PetscScalar*)p1,*vv2,(PetscScalar*)p2);
9347c6ae99SBarry Smith }
9447c6ae99SBarry Smith 
95*8cc058d9SJed Brown PETSC_EXTERN void PETSC_STDCALL dmcompositerestoreaccess4_(DM *dm,Vec *v,void **v1,void **p1,void **v2,void **p2,PetscErrorCode *ierr)
9647c6ae99SBarry Smith {
9747c6ae99SBarry Smith   *ierr = DMCompositeRestoreAccess(*dm,*v,(Vec*)v1,0,(Vec*)v2,0);
9847c6ae99SBarry Smith }
9947c6ae99SBarry Smith 
100*8cc058d9SJed Brown PETSC_EXTERN void PETSC_STDCALL dmcompositegetlocalvectors4_(DM *dm,void **v1,void **p1,void **v2,void **p2,PetscErrorCode *ierr)
10147c6ae99SBarry Smith {
10247c6ae99SBarry Smith   Vec *vv1 = (Vec*)v1,*vv2 = (Vec*)v2;
10347c6ae99SBarry Smith   *ierr = DMCompositeGetLocalVectors(*dm,vv1,(PetscScalar*)p1,vv2,(PetscScalar*)p2);
10447c6ae99SBarry Smith }
10547c6ae99SBarry Smith 
106*8cc058d9SJed Brown PETSC_EXTERN void PETSC_STDCALL dmcompositerestorelocalvectors4_(DM *dm,void **v1,void **p1,void **v2,void **p2,PetscErrorCode *ierr)
10747c6ae99SBarry Smith {
10847c6ae99SBarry Smith   Vec *vv1 = (Vec*)v1,*vv2 = (Vec*)v2;
10947c6ae99SBarry Smith   *ierr = DMCompositeRestoreLocalVectors(*dm,vv1,(PetscScalar*)p1,vv2,(PetscScalar*)p2);
11047c6ae99SBarry Smith }
11147c6ae99SBarry Smith 
112*8cc058d9SJed Brown PETSC_EXTERN void PETSC_STDCALL dmcompositegetglobaliss_(DM *dm,IS *iss,PetscErrorCode *ierr)
11373e31fe2SJed Brown {
11473e31fe2SJed Brown   IS      *ais;
11573e31fe2SJed Brown   PetscInt i,ndm;
11673e31fe2SJed Brown   *ierr = DMCompositeGetGlobalISs(*dm,&ais); if (*ierr) return;
11773e31fe2SJed Brown   *ierr = DMCompositeGetNumberDM(*dm,&ndm); if (*ierr) return;
11873e31fe2SJed Brown   for (i=0; i<ndm; i++) iss[i] = ais[i];
11973e31fe2SJed Brown   *ierr = PetscFree(ais);
12073e31fe2SJed Brown }
12173e31fe2SJed Brown 
122*8cc058d9SJed Brown PETSC_EXTERN void PETSC_STDCALL dmcompositegetlocaliss_(DM *dm,IS *iss,PetscErrorCode *ierr)
12373e31fe2SJed Brown {
12473e31fe2SJed Brown   IS      *ais;
12573e31fe2SJed Brown   PetscInt i,ndm;
12673e31fe2SJed Brown   *ierr = DMCompositeGetLocalISs(*dm,&ais); if (*ierr) return;
12773e31fe2SJed Brown   *ierr = DMCompositeGetNumberDM(*dm,&ndm); if (*ierr) return;
12873e31fe2SJed Brown   for (i=0; i<ndm; i++) iss[i] = ais[i];
12973e31fe2SJed Brown   *ierr = PetscFree(ais);
13073e31fe2SJed Brown }
131f73e5cebSJed Brown 
132*8cc058d9SJed Brown PETSC_EXTERN void PETSC_STDCALL dmcompositegetaccessarray_(DM *dm,Vec *gvec,PetscInt *n,const PetscInt *wanted,Vec *vecs,PetscErrorCode *ierr)
133f73e5cebSJed Brown {
134f73e5cebSJed Brown   CHKFORTRANNULLINTEGER(wanted);
135f73e5cebSJed Brown   *ierr = DMCompositeGetAccessArray(*dm,*gvec,*n,wanted,vecs);
136f73e5cebSJed Brown }
137f73e5cebSJed Brown 
138*8cc058d9SJed Brown PETSC_EXTERN void PETSC_STDCALL dmcompositerestoreaccessarray_(DM *dm,Vec *gvec,PetscInt *n,const PetscInt *wanted,Vec *vecs,PetscErrorCode *ierr)
139f73e5cebSJed Brown {
140f73e5cebSJed Brown   CHKFORTRANNULLINTEGER(wanted);
141f73e5cebSJed Brown   *ierr = DMCompositeRestoreAccessArray(*dm,*gvec,*n,wanted,vecs);
142f73e5cebSJed Brown }
143