Page MenuHomeHEPForge
Files F10664408

OutputVarUtils.hh
No OneTemporary
Actions

OutputVarUtils.hh
View Options

#ifndef _OUTPUTVARUTILS_HH_
#define _OUTPUTVARUTILS_HH_
#include <TTree.h>
#include "JetCollection.hh"
#include "JetMomentMap.hh"
#include "TreeBranchUtils.hh"
#include "CustomMessage.hh"
///////////////////////////////////////////////////////
/// JetVar classes hold variables for a jet collection
/// in order to save them in a ROOT ntuple.
/// They create TTree branches for these variables
///
/// Output variables can be simple type, pure C arrays or STL vectors.
/// Supported base types are float, double and int.
/// Current limitation : when using pure C arrays, constituents number must NOT exceed 10000
///
/// A 'factory' OutputVarFactory is provided to create those complex JetVar classes
/// from simple input parameters (ex "vector" and "float"). It allows to hide template
/// complexity.
///////////////////////////////////////////////////////////
namespace SpartyJet {
/// Define the naming of output variables.
class JetVar_Names {
public:
JetVar_Names();
std::string eta;
std::string phi;
std::string p_T;
std::string e;
std::string mass;
std::string ind;
std::string numC;
};
///////////////////////////////////////////////////////////////
/// \class IGenericJetVar is an interface for all JetVar classes
///////////////////////////////////////////////////////////////
class IGenericJetVar {
public:
IGenericJetVar() {};
IGenericJetVar(std::string name, JetMomentMap *mmap=0) {};
virtual ~IGenericJetVar(){};
/// Create branches for each variable in the given tree
virtual void schedule(TTree* tree) = 0;
/// Associate data from jet lists to arrays to be dumped in TTree
virtual void set_data(JetCollection &theJets) = 0;
/// set the names of output variables
virtual void set_var_names(JetVar_Names &vn) = 0;
/// when ROOT output variables are C arrays, one needs the max size of the array
/// and the actual size. the following function give this info.
virtual BranchWrap_simple<int> * get_input_refN() = 0;
virtual size_t get_size_max() = 0;
virtual std::string name() = 0;
static bool do_phiconversion;
};
///////////////////////////////////////////////////////////////
/// \class GenericJetVar
/// \brief holds variables corresponding to jet moments
///
/// It is associated to a JetCollection.
/// It is responsible for creating TBranch and transfering data from JetCollection to TTree
/// it holds abstracted branches (see TreeBranchUtils.hh) for every moment associated to the collection
/// and for the number of jets in the collection (N).
///
/// The form of the output variables are given by the template parameters
/// which are expected to be abstracted branches (see TreeBranchUtils.hh).
///////////////////////////////////////////////////////////////
template< class BRANCH0D, class BRANCH1D, class BRANCH2D>
class GenericJetVar : public IGenericJetVar{
public:
GenericJetVar(): N(0){};
GenericJetVar(std::string name, JetMomentMap *mmap=0) : m_prefix(name), m_mmap(mmap),N("") {m_log.set_name(name+"_output");};
virtual ~GenericJetVar(){}
/// Create branches for each variable in the given tree
virtual void schedule(TTree* tree) ;
/// Associate data from jet lists to arrays
virtual void set_data(JetCollection &theJets);
virtual void set_var_names(JetVar_Names &vn) {};
/// return pointer to jet size in current event (useful when BRANCH1D is not a vector)
BranchWrap_simple<int> * get_input_refN() {return & N;};
/// return max size of arrays (useful when BRANCH1D is not a vector)
size_t get_size_max(){return m_Nmax;}
virtual std::string name() {return m_prefix;}
protected:
std::string m_prefix;
JetMomentMap *m_mmap;
size_t m_Nmax; /// The maximum number of jets. derived classes redefine it according to their Branch type
std::vector<BRANCH1D> m_moment_vars;
std::vector<BRANCH2D> m_momentarray_vars;
std::vector<BRANCH0D> m_event_vars;
std::vector<BRANCH1D> m_eventarray_vars;
BranchWrap_simple<int> N; /// The number of jets in the collection
Message m_log;
template<class BRANCH>
void set_name_register(BRANCH &b, std::string name, TTree *t){
std::string fname = m_prefix+"_" + name ;
m_log << " scheduling "<< fname << std::endl;
b.set_name(fname);
b.registerBranch(t);
}
/// All branch-arrays are resize to N (usefull only when BRANCHXD are vectors)
virtual void resize_vars(JetCollection &theJets);
};
///////////////////////////////////////////////////////////////
/// \class JetVar
/// \brief specialization of GenericJetVar with kinematic variables.
///
/// Just adds kinematic variables of jets
/// Same template parameter as GenericJetVar
///////////////////////////////////////////////////////////////
template< class BRANCH0D, class BRANCH1D, class BRANCH2D>
class JetVar : public GenericJetVar<BRANCH0D, BRANCH1D, BRANCH2D>{
public:
JetVar(){};
JetVar(std::string name, JetMomentMap *mmap=0) : GenericJetVar<BRANCH0D,BRANCH1D,BRANCH2D>(name,mmap) {}
virtual ~JetVar(){}
virtual void set_var_names(std::string eta,std::string phi,std::string pt,std::string e,std::string mass, std::string ind, std::string numc );
virtual void set_var_names(JetVar_Names &vn){var_names = vn;}
virtual void schedule(TTree* tree);
virtual void set_data(JetCollection &theJets);
protected :
/// All branch-arrays are resize to N (usefull only when BRANCHXD are vectors)
virtual void resize_vars(JetCollection &theJets);
JetVar_Names var_names;
BRANCH1D eta;
BRANCH1D phi;
BRANCH1D p_T;
BRANCH1D e;
BRANCH1D mass;
};
///////////////////////////////////////////////////////////////
/// \class JetVarIndex
///
/// \brief specialization of JetVar adding variables to hold jet constituents index
///
/// adds a new template parameter : the abstracted branch type describing indices.
///////////////////////////////////////////////////////////////
template< class BRANCH0D, class BRANCH1D, class BRANCH2D, class BRANCH1Dind>
class JetVarIndex : public JetVar<BRANCH0D, BRANCH1D, BRANCH2D>{
public:
JetVarIndex(std::string name,JetMomentMap *mmap=0) ;
virtual ~JetVarIndex(){}
virtual void schedule(TTree* tree);
virtual void set_data(JetCollection &theJets);
void set_input_refN(BranchWrap_simple<int> *jv){m_input_refN = jv;}
void set_input_refMax(size_t max){m_input_sizeMax = max;}
int get_index_max(){return index.size_max();}
protected :
/// All branch-arrays are resize to N (usefull only when BRANCHXD are vectors)
virtual void resize_vars(JetCollection &theJets);
BranchWrap_simple<int> * m_input_refN; // Branch holding the number of jets. Mandatory when using array in BRANCH1D
size_t m_input_sizeMax;
BRANCH1Dind index;
BRANCH1D num_constit;
};
///////////////////////////////////////////////////////////////
/// \class OutputVarFactory
///
/// \brief convenience class : create function JetVar classes from simple parameters
///
///////////////////////////////////////////////////////////////
class OutputVarFactory {
public:
OutputVarFactory() : array_type("array"), base_type("float") {};
std::string array_type ;
std::string base_type ;
/// Build a JetVar class. if inputvar != NULL then build JetVarIndex else JetVar.
IGenericJetVar * get_jet_var(std::string name, JetMomentMap * map, IGenericJetVar * inputvar = 0, bool onlymoment = false);
/// Build a JetVar class for input jet (no index, no moment, size 10000 in case of array)
IGenericJetVar * get_inputjet_var(std::string name, JetMomentMap *map= NULL);
protected:
// a short cut for hopefully better readability
template<class JVI>
IGenericJetVar* prepare_input(JVI* jvar, IGenericJetVar * inputvar){
jvar->set_input_refN(inputvar->get_input_refN() );
return jvar;
}
template <int NMAX>
IGenericJetVar * _get_jet_var(std::string name, JetMomentMap * map, IGenericJetVar * inputvar = 0, bool onlymoment = false);
};
}
#ifndef G__DICTIONARY
// prevent dict generation by rootcint for this part of header
#include "OutputVarUtils.icc"
#endif
#endif

File Metadata

Mime Type
text/x-c
Expires
Thu, Apr 24, 6:38 AM (1 d, 19 h)
Storage Engine
blob
Storage Format
Raw Data
Storage Handle
4838461
Default Alt Text
OutputVarUtils.hh (8 KB)

Event Timeline