Moved simulation loaders into their own subdirectory.

c_tools/mock/loaders/multidark_loader.cpp

@@ -0,0 +1,108 @@
+#include <cassert>
+#include <string>
+#include <fstream>
+#include <iostream>
+#include <CosmoTool/loadRamses.hpp>
+#include <CosmoTool/fortran.hpp>
+#include "simulation_loader.hpp"
+
+using namespace std;
+using namespace CosmoTool;
+
+class MultiDarkLoader: public SimulationLoader
+{
+protected:
+  SimuData *header;
+  string darkname;
+public:
+  MultiDarkLoader(const std::string& name, SimuData *h)
+    : darkname(name), header(h)
+  {
+  }
+
+  ~MultiDarkLoader()
+  {
+    delete header;
+  }
+
+  int num_files()
+  {
+    return 1;
+  }
+
+  SimuData *getHeader()
+  {
+    return header;
+  }
+
+  SimuData *loadFile(int id)
+  {
+    if (id != 0)
+      return 0;
+
+    ifstream fp(darkname.c_str());
+    SimuData *simu;
+
+    fp >> simu->BoxSize >> simu->Omega_M >> simu->Hubble >> simu->time >> simu->NumPart;
+    simu->time = 1./(1.+simu->time); // convert to scale factor
+    simu->TotalNumPart = simu->NumPart;
+    simu->Omega_Lambda = 1.0 - simu->Omega_M;
+
+    for (int k = 0; k < 3; k++)
+      simu->Pos[k] = new float[simu->NumPart];
+    simu->Vel[2] = new float[simu->NumPart];
+    simu->Id = new int[simu->NumPart];
+    long *uniqueID = new long[simu->NumPart];
+
+    simu->new_attribute("uniqueID", uniqueID, delete_adaptor<long>);
+
+    cout << "loading multidark particles" << endl;
+    long actualNumPart = 0;
+    for (int i = 0; i < simu->NumPart; i++) {
+
+      fp >> simu->Id[i] >> simu->Pos[0][i] >> simu->Pos[1][i]
+	 >> simu->Pos[2][i] >> simu->Vel[2][i];
+
+      uniqueID[i] = 1;
+      //uniqueID[i] = 1 * simu->Id[i];
+
+      if (simu->Id[i] == -99 && 
+          simu->Pos[0][i] == -99 && simu->Pos[1][i] == -99 && 
+          simu->Pos[2][i] == -99 && simu->Vel[2][i] == -99) {
+        break;
+      } else {
+        actualNumPart++;
+      }
+    }
+
+    simu->NumPart = actualNumPart;
+    simu->TotalNumPart = actualNumPart;
+    return simu;
+  }
+};
+
+SimulationLoader *multidarkLoader(const string& multidarkname)
+{
+  SimuData *header;
+  int actualNumPart;
+  ifstream fp(multidarkname.c_str());
+
+  cout << "opening multidark file " << multidarkname << endl;
+  if (!fp)
+    {
+      cout << "could not open file!" << endl;
+      return 0;
+    }
+
+  header = new SimuData();
+  fp >> header->BoxSize >> header->Omega_M >> header->Hubble >> header->time >> header->NumPart;
+  
+  header->time = 1./(1.+header->time); // convert to scale factor
+  header->TotalNumPart = header->NumPart;
+  header->Omega_Lambda = 1.0 - header->Omega_M;
+
+  return new MultiDarkLoader(multidarkname, header);
+}
+
+  
+