#include <type_traits>

#include <io.hxx>

#include <cctk_Schedule.h>

#include <memory>

#include <type_traits>
#include <utility>

  CCTK_OverloadOutputGH(OutputGH);

#include <AMReX.hxx>
#include <io.hxx>
#include <cctk.h>
#include <cctk_Arguments.h>
#include <cctk_Parameters.h>
#include <AMReX.H>
#include <AMReX_PlotFileUtil.H>
#include <cctype>
#include <memory>
#include <regex>
#include <sstream>
#include <string>
#include <string>
#include <type_traits>
#include <utility>
namespace AMReX {
using namespace amrex;
using namespace std;
int OutputGH(const cGH *restrict cctkGH) {
  DECLARE_CCTK_ARGUMENTS;
  DECLARE_CCTK_PARAMETERS;
  int count_vars = 0;
  const int numgroups = CCTK_NumGroups();
  for (int gi = 0; gi < numgroups; ++gi) {
    auto &restrict groupdata = ghext->groupdata.at(gi);
    string groupname = unique_ptr<char>(CCTK_GroupName(gi)).get();
    groupname = regex_replace(groupname, regex("::"), "-");
    for (auto &c : groupname)
      c = tolower(c);
    ostringstream buf;
    buf << "wavetoy/" << groupname << "." << setw(6) << setfill('0')
        << cctk_iteration;
    string filename = buf.str();
    Vector<string> varnames(groupdata.numvars * groupdata.numtimelevels);
    for (int tl = 0; tl < groupdata.numtimelevels; ++tl) {
      for (int n = 0; n < groupdata.numvars; ++n) {
        ostringstream buf;
        buf << CCTK_VarName(groupdata.firstvarindex + n);
        for (int i = 0; i < tl; ++i)
          buf << "_p";
        varnames.at(tl * groupdata.numvars + n) = buf.str();
      }
    }
    // TODO: Output only current time level. This might require having one
    // mfab per time level
    // TODO: Output everything into a single file
    WriteSingleLevelPlotfile(filename, groupdata.mfab, varnames, ghext->geom,
                             cctk_time, cctk_iteration);
    count_vars += groupdata.numvars;
  }
  return count_vars;
}
} // namespace AMReX

#ifndef IO_HXX
#define IO_HXX
#include <cctk.h>
namespace AMReX {
int OutputGH(const cGH *cctkGH);
} // namespace AMReX
#endif // #ifndef IO_HXX

SRCS = driver.cxx schedule.cxx

SRCS = driver.cxx io.cxx schedule.cxx

REQUIRES AMReX

USES INCLUDE HEADER: AMReX.hxx

SCHEDULE WaveToyAMReX_Error AT analysis BEFORE WaveToyAMReX_Output

SCHEDULE WaveToyAMReX_Error AT analysis

SCHEDULE WaveToyAMReX_Output AT analysis
{
  LANG: C
  OPTIONS: level
} "Output wave equation state"

#include <AMReX.hxx>
using namespace AMReX;

#include <AMReX_PlotFileUtil.H>
using namespace amrex;

#include <cctype>

#include <memory>
#include <regex>
#include <sstream>
#include <string>
using namespace std;

using namespace std;

extern "C" void WaveToyAMReX_Output(CCTK_ARGUMENTS) {
  DECLARE_CCTK_ARGUMENTS;
  DECLARE_CCTK_PARAMETERS;
  // Output phi
  const int numgroups = CCTK_NumGroups();
  for (int gi = 0; gi < numgroups; ++gi) {
    auto &restrict groupdata = ghext->groupdata.at(gi);

    string groupname = unique_ptr<char>(CCTK_GroupName(gi)).get();
    groupname = regex_replace(groupname, regex("::"), "-");
    for (auto &c : groupname)
      c = tolower(c);
    ostringstream buf;
    buf << "wavetoy/" << groupname << "." << setw(6) << setfill('0')
        << cctk_iteration;
    string filename = buf.str();
    Vector<string> varnames(groupdata.numvars * groupdata.numtimelevels);
    for (int tl = 0; tl < groupdata.numtimelevels; ++tl) {
      for (int n = 0; n < groupdata.numvars; ++n) {
        ostringstream buf;
        buf << CCTK_VarName(groupdata.firstvarindex + n);
        for (int i = 0; i < tl; ++i)
          buf << "_p";
        varnames.at(tl * groupdata.numvars + n) = buf.str();
      }
    }
    WriteSingleLevelPlotfile(filename, groupdata.mfab, varnames, ghext->geom,
                             cctk_time, cctk_iteration);
  }
}