EvaluatorSvc.h

Go to the documentation of this file.

// SPDX-License-Identifier: LGPL-3.0-or-later
// Copyright (C) 2024 Dmitry Kalinkin
 
#pragma once
 
#include <algorithms/logger.h>
#include <cstddef>
#include <exception>
#include <functional>
#include <mutex>
#include <stdexcept>
#include <string>
#include <unordered_map>
#include <utility>
#include <vector>
 
namespace eicrecon {
 
class EvaluatorSvc : public algorithms::LoggedService<EvaluatorSvc> {
public:
  void init();
 
  template <class... Args>
  std::function<double(Args...)>
  compile(const std::string& expr,
          std::function<std::unordered_map<std::string, double>(Args...)> transform) {
    try {
      // trivial function if string is representation of float
      std::size_t pos;
      float expr_value = std::stof(expr, &pos);
      if (pos < expr.size()) {
        throw std::invalid_argument("unparsed trailing characters");
      }
      return [expr_value](Args... /* args */) { return expr_value; };
    } catch (std::exception& e) {
      // more complicated function otherwise
      std::vector<std::string> params;
      // Call transform with default values to detect parameter names
      for (auto& p : transform(Args{}...)) {
        params.push_back(p.first);
      }
      auto compiled_expr = _compile(expr, params);
      return [compiled_expr, transform](Args... args) {
        return compiled_expr(transform(std::forward<Args>(args)...));
      };
    }
  };
 
  std::function<double(const std::unordered_map<std::string, double>&)>
  _compile(const std::string& expr, const std::vector<std::string>& params);
 
private:
  unsigned int m_function_id = 0;
  std::mutex m_interpreter_mutex;
 
  ALGORITHMS_DEFINE_LOGGED_SERVICE(EvaluatorSvc);
};
 
} // namespace eicrecon