60 lines
1.4 KiB
C++
60 lines
1.4 KiB
C++
|
#include <assert.h>
|
||
|
|
#include <stdio.h>
|
||
|
|
#include <stdlib.h>
|
||
|
|
#include "load_data.hpp"
|
||
|
|
#include "loadGadget.hpp"
|
||
|
|
#include "fortran.hpp"
|
||
|
|
|
||
|
|
using namespace CosmoTool;
|
||
|
|
|
||
|
|
PurePositionData *CosmoTool::loadGadgetPosition(const char *fname)
|
||
|
|
{
|
||
|
|
PurePositionData *data;
|
||
|
|
int p, n;
|
||
|
|
UnformattedRead f(fname);
|
||
|
|
GadgetHeader h;
|
||
|
|
|
||
|
|
data = new PurePositionData;
|
||
|
|
f.beginCheckpoint();
|
||
|
|
for (int i = 0; i < 6; i++)
|
||
|
|
h.npart[i] = f.readInt32();
|
||
|
|
for (int i = 0; i < 6; i++)
|
||
|
|
h.mass[i] = f.readReal64();
|
||
|
|
h.time = f.readReal64();
|
||
|
|
h.redshift = f.readReal64();
|
||
|
|
h.flag_sfr = f.readInt32();
|
||
|
|
h.flag_feedback = f.readInt32();
|
||
|
|
for (int i = 0; i < 6; i++)
|
||
|
|
h.npartTotal[i] = f.readInt32();
|
||
|
|
h.flag_cooling = f.readInt32();
|
||
|
|
h.num_files = f.readInt32();
|
||
|
|
h.BoxSize = f.readReal64();
|
||
|
|
h.Omega0 = f.readReal64();
|
||
|
|
h.OmegaLambda = f.readReal64();
|
||
|
|
h.HubbleParam = f.readReal64();
|
||
|
|
f.endCheckpoint(true);
|
||
|
|
|
||
|
|
data->NumPart = 0;
|
||
|
|
for(int k=0; k<5; k++)
|
||
|
|
data->NumPart += h.npart[k];
|
||
|
|
|
||
|
|
data->pos = new FCoordinates[data->NumPart];
|
||
|
|
|
||
|
|
f.beginCheckpoint();
|
||
|
|
for(int k = 0, p = 0; k < 5; k++) {
|
||
|
|
for(int n = 0; n < h.npart[k]; n++) {
|
||
|
|
data->pos[p][0] = f.readReal32();
|
||
|
|
data->pos[p][1] = f.readReal32();
|
||
|
|
data->pos[p][2] = f.readReal32();
|
||
|
|
p++;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
f.endCheckpoint();
|
||
|
|
|
||
|
|
// Skip velocities
|
||
|
|
f.skip((long)data->NumPart*3+2*4);
|
||
|
|
// Skip ids
|
||
|
|
|
||
|
|
return data;
|
||
|
|
}
|