File multipleparametergetter.h

File List > code_source > templat > include > TempLat > parameters > multipleparametergetter.h

Go to the documentation of this file

#ifndef TEMPLAT_PARAMETERS_MULTIPLEPARAMETERGETTER_H
#define TEMPLAT_PARAMETERS_MULTIPLEPARAMETERGETTER_H

/* This file is part of TempLat, available at https://cosmolattice.github.io/templat .
   Copyright 2021-2026 The TempLat authors, see AUTHORS.md.
   Released under the MIT license, see LICENSE.md. */

// File info: Main contributor(s): Adrien Florio, Year: 2019

#include "TempLat/parameters/parametergetter.h"
#include "TempLat/util/rangeiteration/tag.h"
#include <sstream>

namespace TempLat
{
  template <typename T> class MultipleParameterGetter
  {
  public:
    // Put public methods here. These should change very little over time.
    MultipleParameterGetter() = default;

    void push_back(const ParameterGetter<T> &p) { params.push_back(p); }
    void clear() { params.clear(); }
    size_t size() { return params.size(); }
    ParameterGetter<T> operator[](int i) { return params[i]; }
    const T &operator()(int i) { return params[i](); }

    template <int Q> auto getComp(Tag<Q> t) { return params[Q](); }

    std::string toString() const { return params[0].toString(); }

    friend std::ostream &operator<<(std::ostream &os, const MultipleParameterGetter<T> &mp)
    {
      os << mp.toString() << "(";

      std::stringstream stmp;

      for (auto p : mp.params) {
        stmp << p() << ",";
      }

      std::string tmp = stmp.str();

      tmp.pop_back();

      os << tmp << ")";

      return os;
    }

    operator std::vector<T>()
    {
      std::vector<T> res;
      for (size_t i = 0; i < params.size(); ++i) {
        res.push_back(params[i]());
      }
      return res;
    }

    template <int N> operator std::array<T, N>()
    {
      std::array<T, N> res;
      for (size_t i = 0; i < params.size(); ++i) {
        res[i] = params[i]();
      }
      return res;
    }

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

    std::vector<ParameterGetter<T>> params;
  };
} // namespace TempLat

#endif