From f142d87b93fc246cc12c799a7a9e1785985c9b65 Mon Sep 17 00:00:00 2001
From: Guilhem Lavaux <guilhem@mithril.(none)>
Date: Tue, 27 Apr 2010 13:38:00 +0200
Subject: [PATCH] New function to load gadget in any possible way

---
 src/loadGadget.cpp | 101 +++++++++++++++++++++++++++++++++++++++++++++
 src/loadGadget.hpp |   3 ++
 2 files changed, 104 insertions(+)

diff --git a/src/loadGadget.cpp b/src/loadGadget.cpp
index a8a3e34..4cb3d7d 100644
--- a/src/loadGadget.cpp
+++ b/src/loadGadget.cpp
@@ -57,3 +57,104 @@ PurePositionData *CosmoTool::loadGadgetPosition(const char *fname)
 
   return data;
 }
+
+
+
+
+SimuData *CosmoTool::loadGadgetMulti(const char *fname, int id, int loadflags)
+{
+  SimuData *data;
+  int p, n;
+  UnformattedRead f(fname);
+  GadgetHeader h;
+
+  data = new SimuData;
+
+  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);
+
+  long NumPart = 0;
+  for(int k=0; k<5; k++)
+    NumPart += h.npart[k];
+  data->NumPart = NumPart;
+
+  if (loadflags & NEED_POSITION) {
+      
+    for (int i = 0; i < 3; i++)
+      data->Pos[i] = new float[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();
+    
+  } else {
+    // Skip positions
+    f.skip(NumPart * 3 + 2*4);
+  }
+  
+  if (loadflags & NEED_VELOCITY) {
+    for (int i = 0; i < 3; i++)
+      data->Vel[i] = new float[data->NumPart];
+    
+    f.beginCheckpoint();
+    for(int k = 0, p = 0; k < 5; k++) {
+      for(int n = 0; n < h.npart[k]; n++) {
+	data->Vel[p][0] = f.readReal32();
+	data->Vel[p][1] = f.readReal32();
+	data->Vel[p][2] = f.readReal32();
+	p++;
+      }
+    }
+    f.endCheckpoint();
+    
+    // TODO: FIX THE UNITS OF THESE FUNKY VELOCITIES !!!
+  } else {
+    // Skip velocities
+    f.skip(NumPart*3+2*4);
+  }
+
+  // Skip ids
+  if (loadflags & NEED_GADGET_ID) {
+    f.beginCheckpoint();
+    for(int k = 0, p = 0; k < 6; k++)
+      {
+	for(int n = 0; n < h.npart[k]; n++)
+	  {
+	    data->Id[p] = f.readInt32();
+	    p++;
+	  }
+      }
+    f.endCheckpoint();
+  } else {
+    f.skip(2*4);
+    for (int k = 0; k < 6; k++)
+      f.skip(h.npart[k]);
+  }
+
+  return data;
+}
+
+
diff --git a/src/loadGadget.hpp b/src/loadGadget.hpp
index aa01281..59a46d4 100644
--- a/src/loadGadget.hpp
+++ b/src/loadGadget.hpp
@@ -2,10 +2,13 @@
 #define __COSMO_LOAD_GADGET_HPP
 
 #include "load_data.hpp"
+#include "loadSimu.hpp"
 
 namespace CosmoTool {
   
   PurePositionData *loadGadgetPosition(const char *fname);
+
+  SimuData *loadGadgetMulti(const char *fname, int id, int flags);
   
 };