ArrayAdaptor

FormulaEvaluator

Macros

Line Code
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100
#ifndef CommonTools_Utils_FormulaEvaluator_h
#define CommonTools_Utils_FormulaEvaluator_h
// -*- C++ -*-
//
// Package:     CommonTools/Utils
// Class  :     FormulaEvaluator
//
/**\class FormulaEvaluator FormulaEvaluator.h "CommonTools/Utils/interface/FormulaEvaluator.h"

 Description: [one line class summary]

 Usage:
    <usage>

*/
//
// 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