services/evaluator/EvaluatorSvc.cc

0001 // SPDX-License-Identifier: LGPL-3.0-or-later
0002 // Copyright (C) 2024 Dmitry Kalinkin
0003
0004 #include <TInterpreter.h>
0005 #include <TInterpreterValue.h>
0006 #include <fmt/core.h>
0007 #include <memory>
0008 #include <sstream>
0009
0010 #include "EvaluatorSvc.h"
0011
0012 namespace eicrecon {
0013
0014 void EvaluatorSvc::init() {
0015   // This is needed to bypass condition in algorithms::LoggerMixin::report and
0016   // forward all messages to our instance of LogSvc/spdlog.
0017   level(algorithms::LogLevel::kTrace);
0018 }
0019
0020 std::function<double(const std::unordered_map<std::string, double>&)>
0021 EvaluatorSvc::_compile(const std::string& expr, std::vector<std::string> params) {
0022   std::lock_guard<std::mutex> guard(m_interpreter_mutex);
0023
0024   std::string func_name = fmt::format("_eicrecon_{}", m_function_id++);
0025   std::ostringstream sstr;
0026   sstr << "double " << func_name << "(double params[]){";
0027   for (unsigned int param_ix = 0; const auto& p : params) {
0028     sstr << "double " << p << " = params[" << (param_ix++) << "];";
0029   }
0030   sstr << "return " << expr << ";";
0031   sstr << "}";
0032
0033   TInterpreter* interp = TInterpreter::Instance();
0034   debug("Compiling {}", sstr.str());
0035   interp->ProcessLine(sstr.str().c_str());
0036   std::unique_ptr<TInterpreterValue> func_val{gInterpreter->MakeInterpreterValue()};
0037   interp->Evaluate(func_name.c_str(), *func_val);
0038   typedef double (*func_t)(double params[]);
0039   func_t func = ((func_t)(func_val->GetAsPointer()));
0040
0041   return [params, func](const std::unordered_map<std::string, double>& param_values) {
0042     std::vector<double> value_list;
0043     value_list.reserve(params.size());
0044     for (const auto& p : params) {
0045       value_list.push_back(param_values.at(p));
0046     }
0047     return func(value_list.data());
0048   };
0049 }
0050
0051 } // namespace eicrecon