RInferenceUtils.hxx

Go to the documentation of this file.

#ifndef TMVA_RINFERENCEUTILS
#define TMVA_RINFERENCEUTILS

#include "ROOT/RIntegerSequence.hxx" // std::index_sequence
#include <utility> // std::forward

namespace TMVA {
namespace Experimental {

namespace Internal {

/// Compute helper
template <typename I, typename T, typename F>
class ComputeHelper;

template <std::size_t... N, typename T, typename F>
class ComputeHelper<std::index_sequence<N...>, T, F> {
   template <std::size_t Idx>
   using AlwaysT = T;
   F fFunc;

public:
   ComputeHelper(F &&f) : fFunc(std::forward<F>(f)) {}
   auto operator()(AlwaysT<N>... args) -> decltype(fFunc.Compute({args...})) { return fFunc.Compute({args...}); }
};

} // namespace Internal

/// Helper to pass TMVA model to RDataFrame.Define nodes
template <std::size_t N, typename T, typename F>
auto Compute(F &&f) -> Internal::ComputeHelper<std::make_index_sequence<N>, T, F>
{
   return Internal::ComputeHelper<std::make_index_sequence<N>, T, F>(std::forward<F>(f));
}

} // namespace Experimental
} // namespace TMVA

#endif // TMVA_RINFERENCEUTILS