New tool to convert a gadget to a HDF5 file
This commit is contained in:
parent
47255ea25a
commit
468f5222b8
@ -37,6 +37,10 @@ if (HDF5_FOUND)
|
|||||||
|
|
||||||
add_executable(testHDF5 testHDF5.cpp)
|
add_executable(testHDF5 testHDF5.cpp)
|
||||||
target_link_libraries(testHDF5 ${tolink})
|
target_link_libraries(testHDF5 ${tolink})
|
||||||
|
|
||||||
|
add_executable(gadgetToArray gadgetToArray.cpp)
|
||||||
|
target_link_libraries(gadgetToArray ${tolink})
|
||||||
|
|
||||||
endif (HDF5_FOUND)
|
endif (HDF5_FOUND)
|
||||||
|
|
||||||
|
|
||||||
|
54
sample/gadgetToArray.cpp
Normal file
54
sample/gadgetToArray.cpp
Normal file
@ -0,0 +1,54 @@
|
|||||||
|
#include <cmath>
|
||||||
|
#include <iostream>
|
||||||
|
#include <cstdlib>
|
||||||
|
#include "cic.hpp"
|
||||||
|
#include "loadGadget.hpp"
|
||||||
|
#include "miniargs.hpp"
|
||||||
|
#include <H5Cpp.h>
|
||||||
|
#include "hdf5_array.hpp"
|
||||||
|
|
||||||
|
using namespace CosmoTool;
|
||||||
|
using namespace std;
|
||||||
|
|
||||||
|
int main(int argc, char **argv)
|
||||||
|
{
|
||||||
|
typedef boost::multi_array<float, 2> array_type;
|
||||||
|
uint32_t res;
|
||||||
|
char *fname;
|
||||||
|
int id;
|
||||||
|
|
||||||
|
MiniArgDesc desc[] = {
|
||||||
|
{ "SNAPSHOT", &fname, MINIARG_STRING },
|
||||||
|
{ 0, 0, MINIARG_NULL }
|
||||||
|
};
|
||||||
|
|
||||||
|
if (!parseMiniArgs(argc, argv, desc))
|
||||||
|
return 1;
|
||||||
|
|
||||||
|
H5::H5File f("density.h5", H5F_ACC_TRUNC);
|
||||||
|
|
||||||
|
|
||||||
|
SimuData *p = loadGadgetMulti(fname, 0, 0);
|
||||||
|
double L0 = p->BoxSize/1000;
|
||||||
|
array_type parts(boost::extents[p->TotalNumPart][3]);
|
||||||
|
uint64_t q = 0;
|
||||||
|
|
||||||
|
try {
|
||||||
|
for (int cpuid=0;;cpuid++) {
|
||||||
|
p = loadGadgetMulti(fname, cpuid, NEED_POSITION);
|
||||||
|
for (uint32_t i = 0; i < p->NumPart; i++)
|
||||||
|
{
|
||||||
|
parts[q][0] = p->Pos[0][i]/1000;
|
||||||
|
parts[q][1] = p->Pos[1][i]/1000;
|
||||||
|
parts[q][2] = p->Pos[2][i]/1000;
|
||||||
|
q++;
|
||||||
|
}
|
||||||
|
|
||||||
|
delete p;
|
||||||
|
}
|
||||||
|
} catch (const NoSuchFileException& e) {}
|
||||||
|
|
||||||
|
hdf5_write_array(f, "particles", parts);
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
Loading…
Reference in New Issue
Block a user