FormulaEvaluator.h

Go to the documentation of this file.

#ifndef CommonTools_Utils_FormulaEvaluator_h
#define CommonTools_Utils_FormulaEvaluator_h
// -*- C++ -*-
//
// Package:     CommonTools/Utils
// Class  :     FormulaEvaluator
//
//
// Original Author:  Christopher Jones
//         Created:  Wed, 23 Sep 2015 21:12:11 GMT
//
 
// system include files
#include <array>
#include <vector>
#include <memory>
#include <string>
 
// user include files
 
// forward declarations
namespace reco {
  namespace formula {
    struct ArrayAdaptor {
      ArrayAdaptor(double const* iStart, size_t iSize) : m_start(iStart), m_size(iSize) {}
      size_t size() const { return m_size; }
      bool empty() const { return m_size == 0; }
      double const* start() const { return m_start; }
 
    private:
      double const* m_start;
      size_t m_size;
    };
    inline double const* startingAddress(ArrayAdaptor const& iV) {
      if (iV.empty()) {
        return nullptr;
      }
      return iV.start();
    }
 
    class EvaluatorBase;
    inline double const* startingAddress(std::vector<double> const& iV) {
      if (iV.empty()) {
        return nullptr;
      }
      return &iV[0];
    }
 
    template <size_t t>
    inline double const* startingAddress(std::array<double, t> const& iV) {
      if (iV.empty()) {
        return nullptr;
      }
      return &iV[0];
    }
 
  }  // namespace formula
 
  class FormulaEvaluator {
  public:
    explicit FormulaEvaluator(std::string const& iFormula);
 
    // ---------- const member functions ---------------------
    template <typename V, typename P>
    double evaluate(V const& iVariables, P const& iParameters) const {
      if (m_nVariables > iVariables.size()) {
        throwWrongNumberOfVariables(iVariables.size());
      }
      if (m_nParameters > iParameters.size()) {
        throwWrongNumberOfParameters(iParameters.size());
      }
      return evaluate(formula::startingAddress(iVariables), formula::startingAddress(iParameters));
    }
 
    unsigned int numberOfParameters() const { return m_nParameters; }
    unsigned int numberOfVariables() const { return m_nVariables; }
 
    std::vector<std::string> abstractSyntaxTree() const;
 
  private:
    double evaluate(double const* iVariables, double const* iParameters) const;
 
    void throwWrongNumberOfVariables(size_t) const;
    void throwWrongNumberOfParameters(size_t) const;
 
    std::shared_ptr<formula::EvaluatorBase const> m_evaluator;
    unsigned int m_nVariables = 0;
    unsigned int m_nParameters = 0;
  };
}  // namespace reco
 
#endif