#include "FEDataStructures.h"
#include <mpi.h>

#ifdef USE_CATALYST
#include "FEAdaptor.h"
#include <CPythonAdaptorAPI.h>
#include <stdio.h>
#include <string.h>
#endif

// Example of a C adaptor for a simulation code
// where the simulation code has a fixed topology
// grid. We treat the grid as an unstructured
// grid even though in the example provided it
// would be best described as a vtkImageData.
// Also, the points are stored in an inconsistent
// manner with respect to the velocity vector.
// This is purposefully done to demonstrate
// the different approaches for getting data
// into Catalyst. In this example we use
// some of the API in CPythonAdaptorAPI.h
// to assist in setting the problem up.
// CFullExample does essentially the same
// thing but without using the existing
// helper API. Note that through configuration
// that the driver can be run without linking
// to Catalyst.

int main(int argc, char* argv[])
{
  MPI_Init(&argc, &argv);
  Grid grid = (Grid) { .NumberOfPoints = 0, .Points = 0, .NumberOfCells = 0, .Cells = 0};
  unsigned int numPoints[3] = {70, 60, 44};
  double spacing[3] = {1, 1.1, 1.3};
  InitializeGrid(&grid, numPoints, spacing);
  Attributes attributes;
  InitializeAttributes(&attributes, &grid);

#ifdef USE_CATALYST
  int fileNameLength = 0;
  if(argc < 2)
    {
    printf("Must pass in a Catalyst script\n");
    MPI_Finalize();
    return 1;
    }
  fileNameLength = strlen(argv[1]);
  coprocessorinitializewithpython(argv[1], &fileNameLength);
  int i;
  for(i=2;i<argc;i++)
    {
    // Add in any other Python script pipelines that are passed in.
    fileNameLength = strlen(argv[i]);
    coprocessoraddpythonscript(argv[i], &fileNameLength);
    }
#endif
  unsigned int numberOfTimeSteps = 100;
  unsigned int timeStep;
  for(timeStep=0;timeStep<numberOfTimeSteps;timeStep++)
    {
    // use a time step length of 0.1
    double time = timeStep * 0.1;
    UpdateFields(&attributes, time);
#ifdef USE_CATALYST
    int lastTimeStep = 0;
    if(timeStep == numberOfTimeSteps-1)
      {
      lastTimeStep = 1;
      }
    CatalystCoProcess(grid.NumberOfPoints, grid.Points, grid.NumberOfCells, grid.Cells,
                      attributes.Velocity, attributes.Pressure, time, timeStep, lastTimeStep);
#endif
    }

#ifdef USE_CATALYST
  CatalystFinalize();
  coprocessorfinalize();
#endif
  MPI_Finalize();

  return 0;
}
