Belle II Software development
Manager.h
1/**************************************************************************
2 * basf2 (Belle II Analysis Software Framework) *
3 * Author: The Belle II Collaboration *
4 * *
5 * See git log for contributors and copyright holders. *
6 * This file is licensed under LGPL-3.0, see LICENSE.md. *
7 **************************************************************************/
8
9#pragma once
10
11#include <framework/logging/Logger.h>
12
13#include <string>
14#include <map>
15#include <vector>
16#include <functional>
17#include <memory>
18#include <variant>
19
20namespace Belle2 {
25 class Particle;
26 class ParticleList;
27
28 namespace Variable {
100 class Manager {
107
108 public:
110 typedef std::variant<double, int, bool> VarVariant;
112 typedef std::function<VarVariant(const Particle*)> FunctionPtr;
114 typedef std::function<VarVariant(const Particle*, const std::vector<double>&)> ParameterFunctionPtr;
116 typedef std::function<FunctionPtr(const std::vector<std::string>&)> MetaFunctionPtr;
119
122 c_double = 0,
123 c_int = 1,
124 c_bool = 2
125 };
126
128 struct VarBase {
129 std::string name;
130 std::string description;
131 std::string group;
133 std::string functionName;
135 VarBase(const std::string& n, const std::string& d, const std::string& g, const VariableDataType& v, const std::string& fName = "")
136 : name(n), description(d), group(g), variabletype(v), functionName(fName.empty() ? n : fName) { }
137
139 void extendDescriptionString(const std::string& d)
140 {
141 description.append(d);
142 }
143 };
144
146 struct Var : public VarBase {
149 Var(const std::string& n, FunctionPtr f, const std::string& d, const std::string& g = "",
150 const VariableDataType& v = VariableDataType::c_double, const std::string& fName = "")
151 : VarBase(n, d, g, v, fName), function(f) { }
152 };
153
155 struct ParameterVar : public VarBase {
158 ParameterVar(const std::string& n, ParameterFunctionPtr f, const std::string& d, const std::string& g = "",
159 const VariableDataType& v = VariableDataType::c_double, const std::string& fName = "")
160 : VarBase(n, d, g, v, fName), function(f) { }
161 };
162
164 struct MetaVar : public VarBase {
167 explicit MetaVar(const std::string& n, MetaFunctionPtr f, const std::string& d, const std::string& g = "",
168 const VariableDataType& v = VariableDataType::c_double, const std::string& fName = "")
169 : VarBase(n, d, g, v, fName), function(f) { }
170 };
171
173 static Manager& Instance();
174
179 const Var* getVariable(std::string name);
180
187 const Var* getVariable(const std::string& functionName, const std::vector<std::string>& functionArguments);
188
192 std::vector<const Variable::Manager::Var*> getVariables(const std::vector<std::string>& variables);
193
197 bool addAlias(const std::string& alias, const std::string& variable);
198
202 void printAliases();
203
207 void clearAliases();
208
212 std::string resolveAlias(const std::string& alias);
213
217 bool addCollection(const std::string& collection, const std::vector<std::string>& variables);
218
222 std::vector<std::string> getCollection(const std::string& collection);
223
227 std::vector<std::string> resolveCollections(const std::vector<std::string>& variables);
228
230 std::vector<const Variable::Manager::VarBase*> getVariables() const { return m_variablesInRegistrationOrder; }
231
232
234 void setVariableGroup(const std::string& groupName);
235
237 void registerVariable(const std::string& name, const Manager::FunctionPtr& f, const std::string& description,
238 const Manager::VariableDataType& v, const std::string& unit = "", const std::string& fName = "");
240 void registerVariable(const std::string& name, const Manager::ParameterFunctionPtr& f, const std::string& description,
241 const Manager::VariableDataType& v, const std::string& unit = "", const std::string& fName = "");
243 void registerVariable(const std::string& name, const Manager::MetaFunctionPtr& f, const std::string& description,
244 const Manager::VariableDataType& v, const std::string& fName = "");
246 void deprecateVariable(const std::string& name, bool make_fatal, const std::string& version, const std::string& description);
247
253 void checkDeprecatedVariable(const std::string& name);
254
261 double evaluate(const std::string& varName, const Particle* p);
262
267 std::vector<double> evaluateVariables(const std::vector<std::string>& varNames, const ParticleList* plist);
268
270 std::vector<std::string> getNames() const;
271
273 std::vector<std::string> getAliasNames() const;
274
276 void assertValidName(const std::string& name);
277
278 private:
279 Manager() {};
281 Manager(const Manager&);
282 ~Manager();
283
285 bool createVariable(const std::string& name);
286
291 bool createVariable(const std::string& fullname, const std::string& functionName,
292 const std::vector<std::string>& functionArguments);
293
295 std::string m_currentGroup;
296
298 std::vector<const VarBase*> m_variablesInRegistrationOrder;
299
301 std::map<std::string, std::string> m_alias;
303 std::map<std::string, std::vector<std::string>> m_collection;
305 std::map<std::string, std::shared_ptr<Var>> m_variables;
307 std::map<std::string, std::shared_ptr<ParameterVar>> m_parameter_variables;
309 std::map<std::string, std::shared_ptr<MetaVar>> m_meta_variables;
311 std::map<std::string, std::pair<bool, std::string>> m_deprecated;
312 };
313
315 class Proxy {
316 public:
318 Proxy(const std::string& name, Manager::FunctionPtr f, const std::string& description, Manager::VariableDataType v,
319 const std::string& unit = "", const std::string& fName = "")
320 {
321 Manager::Instance().registerVariable(name, f, description, v, unit, fName);
322 }
323
324 Proxy(const std::string& name, Manager::ParameterFunctionPtr f, const std::string& description, Manager::VariableDataType v,
325 const std::string& unit = "", const std::string& fName = "")
326 {
327 Manager::Instance().registerVariable(name, f, description, v, unit, fName);
328 }
329
330 Proxy(const std::string& name, Manager::MetaFunctionPtr f, const std::string& description, Manager::VariableDataType v,
331 const std::string& fName = "")
332 {
333 Manager::Instance().registerVariable(name, f, description, v, fName);
334 }
335 };
336
339 public:
341 explicit GroupProxy(const std::string& groupName)
342 {
344 }
345 };
346
349 public:
351 DeprecateProxy(const std::string& name, bool make_fatal, const std::string& version, const std::string& description)
352 {
353 Manager::Instance().deprecateVariable(name, make_fatal, version, description);
354 }
355 };
356
357 template<typename T>
358 std::function<T> make_function(T* t)
359 {
360 return { t };
361 }
362
363 template<typename T>
364 Variable::Manager::VariableDataType get_function_type(const std::string& name, T* t)
365 {
366 auto func = std::function{t};
367 using ReturnType = typename decltype(func)::result_type;
368 if (std::is_same_v<ReturnType, double>) {
369 return Variable::Manager::VariableDataType::c_double;
370 } else if (std::is_same_v<ReturnType, int>) {
371 return Variable::Manager::VariableDataType::c_int;
372 } else if (std::is_same_v<ReturnType, bool>) {
373 return Variable::Manager::VariableDataType::c_bool;
374 } else {
375 B2FATAL("Metavariables must be registered using the REGISTER_METAVARIABLE macro." << LogVar("Variablename", name));
376 }
377 }
378
382#define VARMANAGER_CONCATENATE_DETAIL(x, y) x##y
386#define VARMANAGER_CONCATENATE(x, y) VARMANAGER_CONCATENATE_DETAIL(x, y)
390#define VARMANAGER_MAKE_UNIQUE(x) VARMANAGER_CONCATENATE(x, __LINE__)
391
397#define REGISTER_VARIABLE_NO_UNIT(name, function, description) \
398 static Proxy VARMANAGER_MAKE_UNIQUE(_variableproxy)(std::string(name), \
399 Variable::make_function(function), std::string(description), \
400 Variable::get_function_type(name,function), std::string(""), std::string(#function));
401
407#define REGISTER_VARIABLE_WITH_UNIT(name, function, description, unit) \
408 static Proxy VARMANAGER_MAKE_UNIQUE(_variableproxy)(std::string(name), \
409 Variable::make_function(function), std::string(description), \
410 Variable::get_function_type(name,function), std::string(unit), std::string(#function));
411
417#define PICK_FIFTH_ARG(arg1, arg2, arg3, arg4, arg5, ...) arg5
418
424#define REGISTER_VARIABLE_MACRO_CHOOSER(...) PICK_FIFTH_ARG(__VA_ARGS__, REGISTER_VARIABLE_WITH_UNIT, REGISTER_VARIABLE_NO_UNIT, )
425
431#define REGISTER_VARIABLE(...) REGISTER_VARIABLE_MACRO_CHOOSER(__VA_ARGS__)(__VA_ARGS__)
432
438#define REGISTER_METAVARIABLE(name, function, description, variabledatatype) \
439 static Proxy VARMANAGER_MAKE_UNIQUE(_variableproxy)(std::string(name), \
440 Variable::make_function(function), std::string(description), \
441 Variable::Manager::VariableDataType(variabledatatype), std::string(#function));
442
448#define VARIABLE_GROUP(groupName) \
449 static GroupProxy VARMANAGER_MAKE_UNIQUE(_variablegroupproxy)(groupName);
450
451 }
456#define MAKE_DEPRECATED(name, make_fatal, version, description) \
457 static DeprecateProxy VARMANAGER_MAKE_UNIQUE(_deprecateproxy)(std::string(name), bool(make_fatal), std::string(version), std::string(description));
458
459}
ParticleList is a container class that stores a collection of Particle objects.
Class to store reconstructed particles.
Definition Particle.h:76
DeprecateProxy(const std::string &name, bool make_fatal, const std::string &version, const std::string &description)
constructor.
Definition Manager.h:351
GroupProxy(const std::string &groupName)
constructor.
Definition Manager.h:341
Global list of available variables.
Definition Manager.h:100
Particle Object
Typedef for the cut, that we use Particles as our base objects.
Definition Manager.h:118
std::vector< const VarBase * > m_variablesInRegistrationOrder
List of variables in registration order.
Definition Manager.h:298
std::vector< std::string > getAliasNames() const
Return a list of all variable alias names (in reverse order added).
Definition Manager.cc:470
std::function< VarVariant(const Particle *)> FunctionPtr
functions stored take a const Particle* and return VarVariant.
Definition Manager.h:112
std::vector< std::string > resolveCollections(const std::vector< std::string > &variables)
Resolve Collection Returns variable names corresponding to the given collection or if it is not a col...
Definition Manager.cc:180
Manager(const Manager &)
Copy constructor disabled (not defined).
std::string m_currentGroup
Group last set via VARIABLE_GROUP().
Definition Manager.h:295
const Var * getVariable(std::string name)
Get the variable belonging to the given key.
Definition Manager.cc:58
std::variant< double, int, bool > VarVariant
NOTE: the python interface is documented manually in analysis/doc/Variables.rst (because we use ROOT ...
Definition Manager.h:110
std::map< std::string, std::string > m_alias
List of registered aliases.
Definition Manager.h:301
std::map< std::string, std::shared_ptr< Var > > m_variables
List of registered variables.
Definition Manager.h:305
void deprecateVariable(const std::string &name, bool make_fatal, const std::string &version, const std::string &description)
Make a variable deprecated.
Definition Manager.cc:401
std::map< std::string, std::shared_ptr< ParameterVar > > m_parameter_variables
List of registered parameter variables.
Definition Manager.h:307
static Manager & Instance()
get singleton instance.
Definition Manager.cc:26
void printAliases()
Print existing aliases.
Definition Manager.cc:123
std::map< std::string, std::pair< bool, std::string > > m_deprecated
List of deprecated variables.
Definition Manager.h:311
bool createVariable(const std::string &name)
Creates and registers a concrete variable (Var) from a MetaVar, ParameterVar or numeric constant.
Definition Manager.cc:214
void assertValidName(const std::string &name)
Abort with B2FATAL if name is not a valid name for a variable.
Definition Manager.cc:198
void clearAliases()
Clear existing aliases.
Definition Manager.cc:118
std::function< FunctionPtr(const std::vector< std::string > &)> MetaFunctionPtr
meta functions stored take a const std::vector<std::string>& and return a FunctionPtr.
Definition Manager.h:116
std::vector< std::string > getNames() const
Return list of all variable names (in order registered).
Definition Manager.cc:461
std::map< std::string, std::vector< std::string > > m_collection
List of registered collections.
Definition Manager.h:303
std::vector< const Variable::Manager::VarBase * > getVariables() const
Return list of all variables (in order registered).
Definition Manager.h:230
void setVariableGroup(const std::string &groupName)
All variables registered after VARIABLE_GROUP(groupName) will be added to this group.
Definition Manager.cc:209
void registerVariable(const std::string &name, const Manager::FunctionPtr &f, const std::string &description, const Manager::VariableDataType &v, const std::string &unit="", const std::string &fName="")
Register a variable.
Definition Manager.cc:332
std::string resolveAlias(const std::string &alias)
Resolve alias Return original variable name.
Definition Manager.cc:140
std::vector< std::string > getCollection(const std::string &collection)
Get Collection Returns variable names corresponding to the given collection.
Definition Manager.cc:173
bool addAlias(const std::string &alias, const std::string &variable)
Add alias Return true if the alias was successfully added.
Definition Manager.cc:96
bool addCollection(const std::string &collection, const std::vector< std::string > &variables)
Add collection Return true if the collection was successfully added.
Definition Manager.cc:152
std::function< VarVariant(const Particle *, const std::vector< double > &)> ParameterFunctionPtr
parameter functions stored take a const Particle*, const std::vector<double>& and return VarVariant.
Definition Manager.h:114
VariableDataType
data type of variables
Definition Manager.h:121
void checkDeprecatedVariable(const std::string &name)
Check if a variable is deprecated.
Definition Manager.cc:444
double evaluate(const std::string &varName, const Particle *p)
Evaluate variable 'varName' on given Particle.
Definition Manager.cc:477
std::vector< double > evaluateVariables(const std::vector< std::string > &varNames, const ParticleList *plist)
Evaluate each variable in the vector 'varNames' on given ParticleList and return a flattened vector o...
Definition Manager.cc:493
std::map< std::string, std::shared_ptr< MetaVar > > m_meta_variables
List of registered meta variables.
Definition Manager.h:309
Proxy(const std::string &name, Manager::MetaFunctionPtr f, const std::string &description, Manager::VariableDataType v, const std::string &fName="")
constructor.
Definition Manager.h:330
Proxy(const std::string &name, Manager::FunctionPtr f, const std::string &description, Manager::VariableDataType v, const std::string &unit="", const std::string &fName="")
constructor.
Definition Manager.h:318
Proxy(const std::string &name, Manager::ParameterFunctionPtr f, const std::string &description, Manager::VariableDataType v, const std::string &unit="", const std::string &fName="")
constructor.
Definition Manager.h:324
Abstract base class for different kinds of events.
MetaFunctionPtr function
Pointer to function.
Definition Manager.h:165
MetaVar(const std::string &n, MetaFunctionPtr f, const std::string &d, const std::string &g="", const VariableDataType &v=VariableDataType::c_double, const std::string &fName="")
ctor
Definition Manager.h:167
ParameterVar(const std::string &n, ParameterFunctionPtr f, const std::string &d, const std::string &g="", const VariableDataType &v=VariableDataType::c_double, const std::string &fName="")
ctor
Definition Manager.h:158
ParameterFunctionPtr function
Pointer to function.
Definition Manager.h:156
VarBase(const std::string &n, const std::string &d, const std::string &g, const VariableDataType &v, const std::string &fName="")
ctor
Definition Manager.h:135
std::string functionName
Name of underlying function.
Definition Manager.h:133
std::string description
Description of what this function does.
Definition Manager.h:130
void extendDescriptionString(const std::string &d)
function to extend the description of the variable
Definition Manager.h:139
VariableDataType variabletype
data type of variable
Definition Manager.h:132
std::string group
Associated group.
Definition Manager.h:131
std::string name
Unique identifier of the function, used as key.
Definition Manager.h:129
Var(const std::string &n, FunctionPtr f, const std::string &d, const std::string &g="", const VariableDataType &v=VariableDataType::c_double, const std::string &fName="")
ctor
Definition Manager.h:149
FunctionPtr function
Pointer to function.
Definition Manager.h:147