xref: /petsc/src/vec/is/utils/ftn-custom/zisltogf.c (revision b0dcfd164860a975c76f90dabf1036901aab1c4e)

#include <petsc/private/ftnimpl.h>
#include <petscis.h>
#include <petscviewer.h>

#if defined(PETSC_HAVE_FORTRAN_CAPS)
  #define islocaltoglobalmpnggetinfosize_ ISLOCALTOGLOBALMPNGGETINFOSIZE
  #define islocaltoglobalmappinggetinfo_  ISLOCALTOGLOBALMAPPINGGETINFO
#elif !defined(PETSC_HAVE_FORTRAN_UNDERSCORE)
  #define islocaltoglobalmpnggetinfosize_ islocaltoglobalmpnggetinfosize
  #define islocaltoglobalmappinggetinfo_  islocaltoglobalmappinggetinfo
#endif

static PetscInt  *sprocs, *snumprocs, **sindices;
static PetscBool  called;
PETSC_EXTERN void islocaltoglobalmpnggetinfosize_(ISLocalToGlobalMapping *mapping, PetscInt *size, PetscInt *maxnumprocs, PetscErrorCode *ierr)
{
  PetscInt i;
  if (called) {
    *ierr = PETSC_ERR_ARG_WRONGSTATE;
    return;
  }
  *ierr = ISLocalToGlobalMappingGetInfo(*mapping, size, &sprocs, &snumprocs, &sindices);
  if (*ierr) return;
  *maxnumprocs = 0;
  for (i = 0; i < *size; i++) *maxnumprocs = PetscMax(*maxnumprocs, snumprocs[i]);
  called = PETSC_TRUE;
}

PETSC_EXTERN void islocaltoglobalmappinggetinfo_(ISLocalToGlobalMapping *mapping, PetscInt *size, PetscInt *procs, PetscInt *numprocs, PetscInt *indices, PetscErrorCode *ierr)
{
  PetscInt i, j;
  if (!called) {
    *ierr = PETSC_ERR_ARG_WRONGSTATE;
    return;
  }
  *ierr = PetscArraycpy(procs, sprocs, *size);
  if (*ierr) return;
  *ierr = PetscArraycpy(numprocs, snumprocs, *size);
  if (*ierr) return;
  for (i = 0; i < *size; i++) {
    for (j = 0; j < numprocs[i]; j++) indices[i + (*size) * j] = sindices[i][j];
  }
  *ierr = ISLocalToGlobalMappingRestoreInfo(*mapping, size, &sprocs, &snumprocs, &sindices);
  if (*ierr) return;
  called = PETSC_FALSE;
}