renamed c_tools to c_src and moved zobov to that directory

2025-07-05 07:41:11 +00:00 · 2024-05-22 15:54:36 -04:00 · 2024-05-22 15:54:36 -04:00 · 3203c24ec5
commit 3203c24ec5
parent 50ad2d6d73
48 changed files with 3 additions and 4 deletions
--- a/c_src/libzobov/particleInfo.cpp
+++ b/c_src/libzobov/particleInfo.cpp
@ -0,0 +1,137 @@
+/*+
+    VIDE -- Void IDentification and Examination -- ./c_tools/libzobov/particleInfo.cpp
+    Copyright (C) 2010-2014 Guilhem Lavaux
+    Copyright (C) 2011-2014 P. M. Sutter
+
+    This program is free software; you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation; version 2 of the License.
+
+    This program is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License along
+    with this program; if not, write to the Free Software Foundation, Inc.,
+    51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+*/
+
+#include "particleInfo.hpp"
+
+#include <CosmoTool/fortran.hpp>
+#include <cstdlib>
+#include <netcdf>
+
+using namespace std;
+using namespace CosmoTool;
+using namespace netCDF;
+
+template <bool failure>
+double retrieve_attr_safe_double(NcFile& f, const char* name, double defval) {
+  NcGroupAtt a = f.getAtt(name);
+  if (a.isNull()) {
+    if (failure) abort();
+    return defval;
+  }
+  if (a.getAttLength() != 1) {
+    abort();
+  }
+  double x;
+  a.getValues(&x);
+  return x;
+}
+
+template <bool failure>
+int retrieve_attr_safe_int(NcFile& f, const char* name, int defval) {
+  NcGroupAtt a = f.getAtt(name);
+  if (a.isNull()) {
+    if (failure) abort();
+    return defval;
+  }
+  if (a.getAttLength() != 1) {
+    abort();
+  }
+  int x;
+  a.getValues(&x);
+  return x;
+}
+
+bool loadParticleInfo(ParticleInfo& info, const std::string& particles,
+                      const std::string& extra_info) {
+  int numpart;
+  int isObservation;
+
+  try {
+    NcFile f_info(extra_info, NcFile::read);
+
+    info.ranges[0][0] =
+        retrieve_attr_safe_double<true>(f_info, "range_x_min", 0);
+    info.ranges[0][1] =
+        retrieve_attr_safe_double<true>(f_info, "range_x_max", 0);
+    info.ranges[1][0] =
+        retrieve_attr_safe_double<true>(f_info, "range_y_min", 0);
+    info.ranges[1][1] =
+        retrieve_attr_safe_double<true>(f_info, "range_y_max", 0);
+    info.ranges[2][0] =
+        retrieve_attr_safe_double<true>(f_info, "range_z_min", 0);
+    info.ranges[2][1] =
+        retrieve_attr_safe_double<true>(f_info, "range_z_max", 0);
+    info.mask_index = retrieve_attr_safe_int<true>(f_info, "mask_index", 0);
+    isObservation = retrieve_attr_safe_int<false>(f_info, "is_observation", 0);
+
+    for (int i = 0; i < 3; i++)
+      info.length[i] = info.ranges[i][1] - info.ranges[i][0];
+
+    try {
+      UnformattedRead f(particles);
+
+      float mul, offset;
+
+      f.beginCheckpoint();
+      numpart = f.readInt32();
+      f.endCheckpoint();
+
+      info.particles.resize(numpart);
+
+      offset = info.ranges[0][0];
+      // TEST PMS NON-COBIC BOXES
+      // mul = 1.0;
+      mul = info.ranges[0][1] - info.ranges[0][0];
+      f.beginCheckpoint();
+      for (int i = 0; i < numpart; i++)
+        info.particles[i].x = mul * f.readReal32();
+      f.endCheckpoint();
+
+      offset = info.ranges[1][0];
+      // mul = 1.0;
+      mul = info.ranges[1][1] - info.ranges[1][0];
+      f.beginCheckpoint();
+      for (int i = 0; i < numpart; i++)
+        info.particles[i].y = mul * f.readReal32();
+      f.endCheckpoint();
+
+      offset = info.ranges[2][0];
+      // mul = 1.0;
+      mul = info.ranges[2][1] - info.ranges[2][0];
+      f.beginCheckpoint();
+      for (int i = 0; i < numpart; i++)
+        info.particles[i].z = mul * f.readReal32();
+      f.endCheckpoint();
+
+      if (!isObservation) {
+        for (int i = 0; i < numpart; i++) {
+          info.particles[i].x += info.ranges[0][0];
+          info.particles[i].y += info.ranges[1][0];
+          info.particles[i].z += info.ranges[2][0];
+        }
+      }
+    } catch (NoSuchFileException const& e) {
+      return false;
+    }
+
+  } catch (exceptions::NcCantRead const&) {
+    return false;
+  }
+  return true;
+}