#include <cmath>
#include <iostream>
#include <cstdlib>
#include "cic.hpp"
#include "loadGadget.hpp"
#include "miniargs.hpp"
#include "yorick.hpp"

using namespace std;
using namespace CosmoTool;

int main(int argc, char **argv)
{
  uint32_t res;
  char *fname;
  int id;

  MiniArgDesc desc[] = {
    { "SNAPSHOT", &fname, MINIARG_STRING },
    { "ID", &id, MINIARG_INT },
    { "RESOLUTION", &res, MINIARG_INT },
    { 0, 0, MINIARG_NULL }
  };

  if (!parseMiniArgs(argc, argv, desc))
    return 1;

  SimuData *p = loadGadgetMulti(fname, 0, 0);
  double L0 = p->BoxSize/1000;
  CICFilter filter(res, L0);

  delete p;

  try {
  for (int cpuid=0;;cpuid++) {
    p = loadGadgetMulti(fname, cpuid, NEED_POSITION);
    for (uint32_t i = 0; i < p->NumPart; i++)
    {
      CICParticles a;
      
      a.mass = 1.0;
      a.coords[0] = p->Pos[0][i]/1000;
      a.coords[1] = p->Pos[1][i]/1000;
      a.coords[2] = p->Pos[2][i]/1000;
      filter.putParticles(&a, 1);
    }
  
    delete p;
  }
  } catch (const NoSuchFileException& e) {}

  CICType *denField;
  uint32_t Ntot;
  filter.getDensityField(denField, Ntot);
  
  cout << "L0=" << L0 << endl;
  cout << "Saving density field" << endl;
  uint32_t dimList[] = { res, res, res};
  saveArray("densityField.nc", denField, dimList, 3);

  return 0;
}