25#include <unordered_map>
29bool startsWith(std::string_view str, std::string_view prefix)
31 return str.size() >= prefix.size() && 0 == str.compare(0, prefix.size(), prefix);
37namespace Experimental {
75 throw std::runtime_error(
"You requested the result of a vector observable outside a loop scope for it!");
139 for (
unsigned i = 0; i <
_indent; ++i)
163 std::string idx =
"loopIdx" + std::to_string(
loopLevel);
165 std::vector<TNamed const *> vars;
171 vars.push_back(it.first);
172 _nodeNames[it.first] =
"obs[" + std::to_string(it.second) +
" + " + idx +
"]";
178 std::size_t numEntries = 1;
179 for (
auto &it : vars) {
181 if (
n > 1 && numEntries > 1 &&
n != numEntries) {
182 throw std::runtime_error(
"Trying to loop over variables with different sizes!");
184 numEntries = std::max(
n, numEntries);
188 addToCodeBody(in,
"for(int " + idx +
" = 0; " + idx +
" < " + std::to_string(numEntries) +
"; " + idx +
"++) {\n");
190 return std::make_unique<LoopScope>(*
this, std::move(vars));
198 for (
auto const &ptr :
scope.vars()) {
254 for (
const auto arg : in) {
269 unsigned int n =
arr.size();
272 for (
unsigned int i = 0; i <
n; i++) {
275 return "xlArr + " +
offset;
280 std::ostringstream os;
284 const std::string
info =
"// Begin -- " +
_fn;
291 const std::string
info =
"// End -- " + _fn +
"\n";
292 _ctx.addToCodeBody(_arg,
info);
343 gInterpreter->Declare(
"#include <RooFit/CodegenImpl.h>\n");
353 <<
"constexpr double inf = std::numeric_limits<double>::infinity();\n"
359 errorMsg <<
"Function " <<
funcName <<
" could not be compiled. See above for details. Full code dumped to file "
367 throw std::runtime_error(
errorMsg.str().c_str());
380namespace Experimental {
382template <class Arg_t, int P>
383auto FUNC_NAME(Arg_t &arg, CodegenContext &ctx, Prio<P> p)
385 if constexpr (std::is_same<Prio<P>, PrioLowest>::value) {
386 return FUNC_NAME(arg, ctx);
388 return FUNC_NAME(arg, ctx, p.next());
392template <class Arg_t>
393struct Caller_FUNC_NAME {
395 static auto call(RooAbsArg &arg, CodegenContext &ctx)
397 return FUNC_NAME(static_cast<Arg_t &>(arg), ctx, PrioHighest{});
401} // namespace Experimental
424 static std::unordered_map<TClass *, Func>
dispatchMap;
429 func = found->second;
432 std::stringstream
cmd;
433 cmd <<
"&RooFit::Experimental::Caller_codegenImpl<" << tclass->
GetName() <<
">::call;";
434 func =
reinterpret_cast<Func
>(
gInterpreter->ProcessLine(
cmd.str().c_str()));
438 return func(arg, ctx);
bool startsWith(std::string_view str, std::string_view prefix)
const char Option_t
Option string (const char)
ROOT::Detail::TRangeCast< T, true > TRangeDynCast
TRangeDynCast is an adapter class that allows the typed iteration through a TCollection.
Option_t Option_t TPoint TPoint const char GetTextMagnitude GetFillStyle GetLineColor GetLineWidth GetMarkerStyle GetTextAlign GetTextColor GetTextSize void char Point_t Rectangle_t WindowAttributes_t Float_t Float_t Float_t Int_t Int_t UInt_t UInt_t Rectangle_t Int_t Int_t Window_t TString Int_t GCValues_t GetPrimarySelectionOwner GetDisplay GetScreen GetColormap GetNativeEvent const char const char dpyName wid window const char font_name cursor keysym reg const char only_if_exist regb h Point_t winding char text const char depth char const char Int_t count const char ColorStruct_t color const char Pixmap_t Pixmap_t PictureAttributes_t attr const char char ret_data h unsigned char height h offset
Option_t Option_t TPoint TPoint const char GetTextMagnitude GetFillStyle GetLineColor GetLineWidth GetMarkerStyle GetTextAlign GetTextColor GetTextSize void value
const_iterator end() const
Common abstract base class for objects that represent a value and a "shape" in RooFit.
bool dependsOn(const RooAbsCollection &serverList, const RooAbsArg *ignoreArg=nullptr, bool valueOnly=false) const
Test whether we depend on (ie, are served by) any object in the specified collection.
TClass * IsA() const override
const TNamed * namePtr() const
De-duplicated pointer to this object's name.
Int_t defaultPrintContents(Option_t *opt) const override
Define default contents to print.
virtual bool isReducerNode() const
Abstract container object that can hold multiple RooAbsArg objects.
RooFit::UniqueId< RooAbsCollection > const & uniqueId() const
Returns a unique ID that is different for every instantiated RooAbsCollection.
A class to manage loop scopes using the RAII technique.
A class to maintain the context for squashing of RooFit models into code.
std::unordered_map< RooFit::UniqueId< RooAbsCollection >::Value_t, std::string > _listNames
A map to keep track of list names as assigned by addResult.
void addToGlobalScope(std::string const &str)
Adds the given string to the string block that will be emitted at the top of the squashed function.
std::string const & getResult(RooAbsArg const &arg)
Gets the result for the given node using the node name.
std::string getTmpVarName() const
Get a unique variable name to be used in the generated code.
void addResult(RooAbsArg const *key, std::string const &value)
A function to save an expression that includes/depends on the result of the input node.
auto const & outputSizes() const
void addToCodeBody(RooAbsArg const *klass, std::string const &in)
Adds the input string to the squashed code body.
std::unique_ptr< LoopScope > beginLoop(RooAbsArg const *in)
Create a RAII scope for iterating over vector observables.
void collectFunction(std::string const &name)
Register a function that is only know to the interpreter to the context.
std::vector< double > _xlArr
void addVecObs(const char *key, int idx)
Since the squashed code represents all observables as a single flattened array, it is important to ke...
std::unordered_map< const TNamed *, int > _vecObsIndices
A map to keep track of the observable indices if they are non scalar.
int observableIndexOf(const RooAbsArg &arg) const
std::map< RooFit::Detail::DataKey, std::size_t > _nodeOutputSizes
Map of node output sizes.
std::string buildFunction(RooAbsArg const &arg, std::map< RooFit::Detail::DataKey, std::size_t > const &outputSizes={})
Assemble and return the final code with the return expression and global statements.
void endLoop(LoopScope const &scope)
std::vector< std::string > _collectedFunctions
bool isScopeIndependent(RooAbsArg const *in) const
std::vector< std::string > _code
The code layered by lexical scopes used as a stack.
unsigned _indent
The indentation level for pretty-printing.
std::unordered_map< const TNamed *, std::string > _nodeNames
Map of node names to their result strings.
std::size_t outputSize(RooFit::Detail::DataKey key) const
Figure out the output size of a node.
ScopeRAII OutputScopeRangeComment(RooAbsArg const *arg)
std::string buildArg(RooAbsCollection const &x)
Function to save a RooListProxy as an array in the squashed code.
int _tmpVarIdx
Index to get unique names for temporary variables.
static const TNamed * known(const char *stringPtr)
If the name is already known, return its TNamed pointer. Otherwise return 0 (don't register the name)...
virtual StyleOption defaultPrintStyle(Option_t *opt) const
virtual void printStream(std::ostream &os, Int_t contents, StyleOption style, TString indent="") const
Print description of object on ostream, printing contents set by contents integer,...
TClass instances represent classes, structs and namespaces in the ROOT type system.
The TNamed class is the base class for all named ROOT classes.
const char * GetName() const override
Returns name of object.
void replaceAll(std::string &inOut, std::string_view what, std::string_view with)
void declareDispatcherCode(std::string const &funcName)
void codegen(RooAbsArg &arg, CodegenContext &ctx)
The namespace RooFit contains mostly switches that change the behaviour of functions of PDFs (or othe...
ScopeRAII(RooAbsArg const *arg, CodegenContext &ctx)
constexpr Value_t value() const
Return numerical value of ID.