xref: /petsc/src/sys/utils/ftn-custom/zsortsof.c (revision 4e8208cbcbc709572b8abe32f33c78b69c819375)

#include <petsc/private/ftnimpl.h>
#include <petscsys.h>

#if defined(PETSC_HAVE_FORTRAN_CAPS)
  #define petsctimsort_          PETSCTIMSORT
  #define petsctimsortwitharray_ PETSCTIMSORTWITHARRAY
#elif !defined(PETSC_HAVE_FORTRAN_UNDERSCORE)
  #define petsctimsort_          petsctimsort
  #define petsctimsortwitharray_ petsctimsortwitharray
#endif

struct fc_c {
  void (*fcmp)(const void *a, const void *b, void *c, int *res);
  void *fctx;
} fc_c;

int cmp_via_fortran(const void *a, const void *b, PetscCtx ctx)
{
  int          result;
  struct fc_c *fc = (struct fc_c *)ctx;
  fc->fcmp(a, b, fc->fctx, &result);
  return result;
}

PETSC_EXTERN void petsctimsort_(PetscInt *n, void *arr, size_t *size, void (*cmp)(const void *, const void *, void *, int *), PetscCtx ctx, PetscErrorCode *ierr)
{
  struct fc_c fc = {cmp, ctx};
  *ierr          = PetscTimSort(*n, arr, *size, cmp_via_fortran, &fc);
}

PETSC_EXTERN void petsctimsortwitharray_(PetscInt *n, void *arr, size_t *asize, void *barr, size_t *bsize, void (*cmp)(const void *, const void *, void *, int *), PetscCtx ctx, PetscErrorCode *ierr)
{
  struct fc_c fc = {cmp, ctx};
  *ierr          = PetscTimSortWithArray(*n, arr, *asize, barr, *bsize, cmp_via_fortran, &fc);
}