File spectrumsaver.h

File List > code_source > cosmolattice > include > CosmoInterface > measurements > measurementsIO > spectrumsaver.h

Go to the documentation of this file

#ifndef COSMOINTERFACE_MEASUREMENTS_SPECTRUMSAVER_H
#define COSMOINTERFACE_MEASUREMENTS_SPECTRUMSAVER_H

/* This file is part of CosmoLattice, available at www.cosmolattice.net .
   Copyright Daniel G. Figueroa, Adrien Florio, Francisco Torrenti and Wessel Valkenburg.
   Released under the MIT license, see LICENSE.md. */

// File info: Main contributor(s): Daniel G. Figueroa, Adrien Florio, Francisco Torrenti,  Year: 2019

#include "CosmoInterface/measurements/measurementsIO/std/spectrumsaverstd.h"
#include "TempLat/lattice/measuringtools/projectionhelpers/unbinnedradialprojectionresult.h" //Added so that the code compiles without HDF5, even if not used here
#ifdef HAVE_HDF5
#include "CosmoInterface/measurements/measurementsIO/hdf5/spectrumsaverhdf5.h"
#include "CosmoInterface/measurements/measurementsIO/hdf5/unbinnedspectrumsaverhdf5.h"
#endif

namespace TempLat
{
  template <typename T> class SpectrumSaver
  {
  public:
    // Put public methods here. These should change very little over time.

    // @label:spectrumsaver_constructor
    template <size_t NDim>
    SpectrumSaver(FilesManager<NDim> &fm, std::string fn, bool amIRoot, bool appendMode, const RunParameters<T> &rPar,
                  bool dontCreate = false):
                  useHDF5(fm.getUseHDF5Spectra()),
                  unbinnedSpectra(rPar.unbinnedSpectra)

    {
      if (! dontCreate) {
        if (unbinnedSpectra) {
#ifdef HAVE_HDF5
          unbinnedSaverHDF5 = std::make_shared<UnbinnedSpectrumSaverHDF5<T>>(fm, fn, amIRoot, appendMode, rPar);
#else
          throw(UseHDF5ButNotCompiled(
              "Unbinned spectra needs HDF5, but compiled without HDF5 option."));
#endif
        }
        else if (useHDF5) {
#ifdef HAVE_HDF5
          saverHDF5 = std::make_shared<SpectrumSaverHDF5<T>>(fm, fn, amIRoot, appendMode, rPar);
#else
          throw(UseHDF5ButNotCompiled(
              "Call to use HDF5 for the measurementsIO output, but compiled without HDF5 option."));
#endif
        } else {
          saverStd = std::make_shared<SpectrumSaverStd<T>>(fm, fn, amIRoot, appendMode, rPar);
        }
      }
    }
    // @endlabel

    template <size_t NDim>
    SpectrumSaver(FilesManager<NDim> &fm, const Field<T, NDim> &fld, bool amIRoot, bool appendMode,
                  const RunParameters<T> &rPar, bool dontCreate = false) :
        useHDF5(fm.getUseHDF5Spectra()),
        unbinnedSpectra(rPar.unbinnedSpectra)
    {

      if (! dontCreate) {
        if(unbinnedSpectra) {
#ifdef HAVE_HDF5
          unbinnedSaverHDF5 = std::make_shared<UnbinnedSpectrumSaverHDF5<T>>(fm, fld, amIRoot, appendMode, rPar);
#else
          throw(UseHDF5ButNotCompiled(
            "Unbinned spectra needs HDF5, but compiled without HDF5 option."));
#endif
        }
        else if (useHDF5) {
#ifdef HAVE_HDF5
          saverHDF5 = std::make_shared<SpectrumSaverHDF5<T>>(fm, fld, amIRoot, appendMode, rPar);
#else
          throw(
              UseHDF5ButNotCompiled("Call to use HDF5 for the measurementsIO output, but compiled without HDF5 option."));
#endif
        } else {
          saverStd = std::make_shared<SpectrumSaverStd<T>>(fm, fld, amIRoot, appendMode, rPar);
        }
      }
    }


    template<template<typename> typename Spectrum, template<typename> typename... Spectra>
    void saveV(bool lastMeas, T& t, std::vector<std::shared_ptr<Spectrum<T>>>& v, Spectrum<T> spectrum1, Spectra<T>... spectra){
      v.push_back(std::make_shared<Spectrum<T>>(spectrum1));
      saveV(lastMeas, t, v, spectra...);
    }

    template<template<typename> typename Spectrum>
    void saveV(bool lastMeas, T& t, std::vector<std::shared_ptr<Spectrum<T>>>& v){
      saveSpectra(lastMeas, t, v);
    }

    template<template<typename> typename Spectrum, template<typename> typename... Spectra>
    void save(bool lastMeas, T& t, Spectrum<T> spectrum1, Spectra<T>... spectra){
      std::vector<std::shared_ptr<Spectrum<T>>> v = {std::make_shared<Spectrum<T>>(spectrum1)};
      saveV(lastMeas, t, v, spectra...);
    }

    template<template<typename> typename Spectrum, template<typename> typename... Spectra>
    void save(T& t, Spectrum<T> spectrum1, Spectra<T>... spectra){
      std::vector<std::shared_ptr<Spectrum<T>>> v = {std::make_shared<Spectrum<T>>(spectrum1)};
      saveV(false, t, v, spectra...);
    }

    template<template<typename> typename Spectrum>
    void save(bool lastMeas, T& t, Spectrum<T> spectrum1){
      std::vector<std::shared_ptr<Spectrum<T>>> v = {std::make_shared<Spectrum<T>>(spectrum1)};
      saveSpectra(lastMeas, t, v);
    }

    template<template<typename> typename Spectrum>
    void save(T& t, Spectrum<T> spectrum1){
      std::vector<std::shared_ptr<Spectrum<T>>> v = {std::make_shared<Spectrum<T>>(spectrum1)};
      saveSpectra(false, t, v);
    }

    void saveSpectra(bool lastMeas, T& t, std::vector<std::shared_ptr<UnbinnedRadialProjectionResult<T>>> spectra) {
#ifdef HAVE_HDF5
      unbinnedSaverHDF5->save(lastMeas, spectra, t);
#else
      throw (UseHDF5ButNotCompiled(
        "Unbinned spectra needs HDF5, but compiled without HDF5 option."));
#endif
    }

    // @label:spectrumsaver_save
    void saveSpectra(bool lastMeas, T& t, std::vector<std::shared_ptr<RadialProjectionResult<T>>> spectra){
      if(useHDF5) {
#ifdef HAVE_HDF5
        saverHDF5->save(lastMeas,spectra, t);
#else
        throw (UseHDF5ButNotCompiled(
          "Call to use HDF5 for the measurementsIO output, but compiled without HDF5 option."));
#endif
      }
      else {
        saverStd->save(spectra, t);
      }
    }
    // @endlabel

  private:
    /* Put all member variables and private methods here. These may change arbitrarily. */

    bool useHDF5, unbinnedSpectra;;
    std::shared_ptr<SpectrumSaverStd<T>> saverStd;
#ifdef HAVE_HDF5
    std::shared_ptr<SpectrumSaverHDF5<T>> saverHDF5;
    std::shared_ptr<UnbinnedSpectrumSaverHDF5<T>> unbinnedSaverHDF5;
#endif
  };

} // namespace TempLat

#endif