forked from UCATLAS/xAODAnaHelpers
/
TreeAlgo.h
68 lines (54 loc) · 2.08 KB
/
TreeAlgo.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
#ifndef xAODAnaHelpers_TreeAlgo_H
#define xAODAnaHelpers_TreeAlgo_H
#include "TTree.h"
#include <xAODAnaHelpers/HelpTreeBase.h>
// algorithm wrapper
#include "xAODAnaHelpers/Algorithm.h"
class TreeAlgo : public xAH::Algorithm
{
// put your configuration variables here as public variables.
// that way they can be set directly from CINT and python.
public:
// choose whether the tree gets saved in the same directory as output histograms
bool m_outHistDir;
// holds bools that control which branches are filled
std::string m_evtDetailStr;
std::string m_trigDetailStr;
std::string m_jetTrigDetailStr;
std::string m_muDetailStr;
std::string m_elDetailStr;
std::string m_jetDetailStr;
std::string m_fatJetDetailStr;
std::string m_tauDetailStr;
std::string m_METDetailStr;
std::string m_evtContainerName;
std::string m_muContainerName;
std::string m_elContainerName;
std::string m_jetContainerName;
std::string m_fatJetContainerName;
std::string m_tauContainerName;
std::string m_METContainerName;
bool m_DC14;
private:
HelpTreeBase* m_helpTree; //!
public:
// this is a standard constructor
TreeAlgo (); //!
// these are the functions inherited from Algorithm
virtual EL::StatusCode setupJob (EL::Job& job); //!
virtual EL::StatusCode fileExecute (); //!
virtual EL::StatusCode treeInitialize (); //!
virtual EL::StatusCode changeInput (bool firstFile); //!
virtual EL::StatusCode initialize (); //!
virtual EL::StatusCode execute (); //!
virtual EL::StatusCode postExecute (); //!
virtual EL::StatusCode finalize (); //!
virtual EL::StatusCode treeFinalize (); //!
// these are the functions not inherited from Algorithm
virtual EL::StatusCode configure (); //!
/// @cond
// this is needed to distribute the algorithm to the workers
ClassDef(TreeAlgo, 1); //!
/// @endcond
};
#endif