FunctionsWrappers.h

Go to the documentation of this file.

/*
 *  CepGen: a central exclusive processes event generator
 *  Copyright (C) 2022  Laurent Forthomme
 *
 *  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, either version 3 of the License, or
 *  any later version.
 *
 *  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, see <http://www.gnu.org/licenses/>.
 */
 
#ifndef CepGen_Utils_FunctionsWrappers_h
#define CepGen_Utils_FunctionsWrappers_h
 
#include <functional>
 
#include "CepGen/Core/ParametersList.h"
 
namespace cepgen {
  namespace utils {
    class Function1D {
    public:
      explicit Function1D(const std::function<double(double)>& func) : func_(func) {}
      explicit Function1D(const std::function<double(double, const ParametersList&)>& func) : func_params_(func) {}
      explicit Function1D(const std::function<double(double, void*)>& func) : func_obj_(func) {}
 
      double operator()(double x, const ParametersList& params = ParametersList()) const {
        if (func_params_)
          return func_params_(x, params);
        return func_(x);
      }
      double operator()(double x, void* obj) const {
        if (func_obj_)
          return func_obj_(x, obj);
        return func_(x);
      }
      template <typename T>
      double operator()(double x, const T& obj) const {
        return func_obj_(x, (void*)&obj);
      }
 
      operator const std::function<double(double)>&() { return func_; }
      operator const std::function<double(double, void*)>&() { return func_obj_; }
 
    private:
      std::function<double(double)> func_{nullptr};
      std::function<double(double, const ParametersList&)> func_params_{nullptr};
      std::function<double(double, void*)> func_obj_{nullptr};
    };
  }  // namespace utils
}  // namespace cepgen
 
#endif