vide_public/c_tools/libzobov/particleInfo.cpp

/*+
    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;
}