mirror of
https://bitbucket.org/cosmicvoids/vide_public.git
synced 2025-07-04 23:31:12 +00:00
Added back Ramses loader
This commit is contained in:
parent
745b073ed6
commit
1a190a6be0
2 changed files with 80 additions and 1 deletions
79
c_tools/mock/ramses_loader.cpp
Normal file
79
c_tools/mock/ramses_loader.cpp
Normal 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);
|
||||||
|
}
|
||||||
|
|
|
@ -43,7 +43,7 @@ void delete_adaptor(void *ptr)
|
||||||
SimulationLoader *gadgetLoader(const std::string& snapshot, double Mpc_unitLength, int flags);
|
SimulationLoader *gadgetLoader(const std::string& snapshot, double Mpc_unitLength, int flags);
|
||||||
SimulationLoader *flashLoader(const std::string& snapshot, int flags);
|
SimulationLoader *flashLoader(const std::string& snapshot, int flags);
|
||||||
SimulationLoader *multidarkLoader(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
|
#endif
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue