#include <cctk_Schedule.h>

// Global variables
int ghext_handle = -1;

// Registered functions
void *SetupGH(tFleshConfig *fc, int convLevel, cGH *cctkGH);
int InitGH(cGH *cctkGH);
int ScheduleTraverseGH(cGH *cctkGH, const char *where);

int CallFunction(void *function, cFunctionData *attribute, void *data);
// Start driver

  CCTK_VINFO("Startup");
  // Output a startup message

  CCTK_RegisterBanner(banner.c_str());

  int ierr = CCTK_RegisterBanner(banner.c_str());
  assert(!ierr);
  // Register a GH extension
  ghext_handle = CCTK_RegisterGHExtension("AMReX");
  assert(ghext_handle >= 0);
  int iret = CCTK_RegisterGHExtensionSetupGH(ghext_handle, SetupGH);
  assert(iret);
  iret = CCTK_RegisterGHExtensionInitGH(ghext_handle, InitGH);
  assert(iret);
  iret = CCTK_RegisterGHExtensionScheduleTraverseGH(ghext_handle,
                                                    ScheduleTraverseGH);
  assert(iret);
  return 0;
}
// Set up GH extension
void *SetupGH(tFleshConfig *fc, int convLevel, cGH *cctkGH) {
  CCTK_VINFO("SetupGH");
  assert(fc);
  assert(convLevel == 0);
  assert(cctkGH);

  return ghext.get();
}

// Initialize GH extension
int InitGH(cGH *cctkGH) {
  CCTK_VINFO("InitGH");
  assert(cctkGH);

  return 0;

  return 0; // unused
}
// Traverse schedule
int ScheduleTraverseGH(cGH *cctkGH, const char *where) {
  CCTK_VINFO("ScheduleTraverseGH");
  int ierr = CCTK_ScheduleTraverse(where, cctkGH, CallFunction);
  assert(!ierr);
  return 0; // unused

// Call a scheduled function
int CallFunction(void *function, cFunctionData *attribute, void *data) {
  CCTK_VINFO("CallFunction");
  int didsync = CCTK_CallFunction(function, attribute, data);
  return didsync;
}
// Shut down driver

  CCTK_VINFO("Shutdown");
  int iret = CCTK_UnregisterGHExtension("AMReX");
  assert(iret == 0);