RooEvaluatorWrapper.h

Go to the documentation of this file.

/// \cond ROOFIT_INTERNAL
 
/*
 * Project: RooFit
 * Authors:
 *   Jonas Rembser, CERN 2023
 *
 * Copyright (c) 2023, CERN
 *
 * Redistribution and use in source and binary forms,
 * with or without modification, are permitted according to the terms
 * listed in LICENSE (http://roofit.sourceforge.net/license.txt)
 */
 
#ifndef RooFit_RooEvaluatorWrapper_h
#define RooFit_RooEvaluatorWrapper_h
 
#include <RooAbsData.h>
#include <RooFit/Evaluator.h>
#include <RooGlobalFunc.h>
#include <RooRealProxy.h>
#include <RooSetProxy.h>
 
#include <memory>
#include <stack>
 
class ChangeOperModeRAII;
class RooAbsArg;
class RooAbsCategory;
class RooAbsPdf;
 
namespace RooFit::Experimental {
 
class RooFuncWrapper;
 
class RooEvaluatorWrapper final : public RooAbsReal {
public:
   RooEvaluatorWrapper(RooAbsReal &topNode, RooAbsData *data, bool useGPU, std::string const &rangeName,
                       RooAbsPdf const *simPdf, bool takeGlobalObservablesFromData, int nWorkers = 1);
 
   RooEvaluatorWrapper(const RooEvaluatorWrapper &other, const char *name = nullptr);
 
   ~RooEvaluatorWrapper();
 
   TObject *clone(const char *newname) const override { return new RooEvaluatorWrapper(*this, newname); }
 
   double defaultErrorLevel() const override { return _topNode->defaultErrorLevel(); }
 
   bool getParameters(const RooArgSet *observables, RooArgSet &outputSet, bool stripDisconnected = true) const override;
 
   bool setData(RooAbsData &data, bool cloneData) override;
 
   double getValV(const RooArgSet *) const override { return evaluate(); }
 
   void applyWeightSquared(bool flag) override { _topNode->applyWeightSquared(flag); }
 
   bool hasGradient() const override;
   bool hasHessian() const override;
 
   void gradient(double *out) const override;
   void hessian(double *out) const override;
 
   void generateGradient();
   void generateHessian();
 
   void setUseGeneratedFunctionCode(bool);
   void writeDebugMacro(std::string const &) const;
 
   std::unique_ptr<ChangeOperModeRAII> setOperModes(RooAbsArg::OperMode opMode);
 
   RooFit::Evaluator &evaluator() const { return *_evaluator; }
 
   /// The RooFit object that this wrapper evaluates.
   RooAbsReal const &topNode() const { return *_topNode; }
 
protected:
   double evaluate() const override;
 
private:
   void createFuncWrapper();
 
   std::shared_ptr<RooFit::Evaluator> _evaluator;
   std::shared_ptr<RooFuncWrapper> _funcWrapper;
   RooRealProxy _topNode;
   RooAbsData *_data = nullptr;
   RooSetProxy _paramSet;
   std::string _rangeName;
   RooAbsPdf const *_pdf = nullptr;
   const bool _takeGlobalObservablesFromData;
   bool _useGeneratedFunctionCode = false;
   std::stack<std::vector<double>> _vectorBuffers; // used for preserving resources
   std::map<RooFit::Detail::DataKey, std::span<const double>> _dataSpans;
};
 
} // namespace RooFit::Experimental
 
#endif
 
/// \endcond