cosmotool/sample/gadgetToDensity.cpp

63 lines
1.3 KiB
C++
Raw Normal View History

#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;
}