guilhem_lavaux/vide_public
0
0
Fork 0
mirror of https://bitbucket.org/cosmicvoids/vide_public.git synced 2025-07-04 15:21:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Added back Ramses loader

Browse source
This commit is contained in:
Guilhem Lavaux 2012-11-23 14:42:57 -06:00
parent 745b073ed6
commit 1a190a6be0
2 changed files with 80 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

79
c_tools/mock/ramses_loader.cpp Normal file
View file

@ -0,0 +1,79 @@
#include <cassert>
#include <string>
#include <CosmoTool/loadRamses.hpp>
#include <CosmoTool/fortran.hpp>
#include "simulation_loader.hpp"
using namespace std;
using namespace CosmoTool;
class RamsesLoader: public SimulationLoader
{
private:
int load_flags;
int _num_files;
int baseid;
bool double_precision;
SimuData *ramses_header;
string snapshot_name;
public:
RamsesLoader(const string& basename, int baseid, bool dp, SimuData *header, int flags, int _num)
: snapshot_name(basename), load_flags(flags), _num_files(_num), double_precision(dp),
ramses_header(header)
{
}
~RamsesLoader()
{
delete ramses_header;
}
SimuData *getHeader() {
return ramses_header;
}
int num_files() {
return _num_files;
}
SimuData *loadFile(int id) {
SimuData *d;
if (id >= _num_files)
return 0;
d = loadRamsesSimu(snapshot_name.c_str(), baseid, id, double_precision, load_flags);
assert(d != 0);
long *uniqueID = new long[d->NumPart];
for (long i = 0; i < d->NumPart; i++)
{
uniqueID[i] = d->Id[i];
}
d->new_attribute("uniqueID", uniqueID, delete_adaptor<long>);
applyTransformations(d);
return d;
}
};
SimulationLoader *ramsesLoader(const std::string& snapshot, int baseid, bool double_precision, int flags)
{
SimuData *d, *header;
int num_files = 0;
header = loadRamsesSimu(snapshot.c_str(), baseid, 0, double_precision, 0);
if (header == 0)
return 0;
while ((d = loadRamsesSimu(snapshot.c_str(), baseid, num_files, double_precision, 0)) != 0)
{
num_files++;
delete d;
}
return new RamsesLoader(snapshot, baseid, double_precision, header, flags, num_files);
}

2
c_tools/mock/simulation_loader.hpp
View file

@ -43,7 +43,7 @@ void delete_adaptor(void *ptr)
SimulationLoader *gadgetLoader(const std::string& snapshot, double Mpc_unitLength, int flags);
SimulationLoader *flashLoader(const std::string& snapshot, int flags);
SimulationLoader *multidarkLoader(const std::string& snapshot, int flags);
SimulationLoader *ramsesLoader(const std::string& snapshot, int flags);
SimulationLoader *ramsesLoader(const std::string& snapshot, int baseid, bool double_precision, int flags);
#endif