| 5 |
|
#include "TTree.h" |
| 6 |
|
#include "TChain.h" |
| 7 |
|
#include "TString.h" |
| 8 |
+ |
#include "TH1F.h" |
| 9 |
|
|
| 10 |
|
#include "SusyEvent.h" |
| 11 |
|
#include "TreeObjects.h" |
| 12 |
|
|
| 13 |
|
class TreeWriter { |
| 14 |
|
public : |
| 15 |
< |
TreeWriter(TString inputName, TString outputName ); |
| 15 |
> |
TreeWriter(TString inputName, TString outputName, int loggingVerbosity_); |
| 16 |
> |
TreeWriter(TChain* inputName, TString outputName, int loggingVerbosity_); |
| 17 |
> |
void Init( TString outputName, int loggingVerbosity_ ); |
| 18 |
|
virtual ~TreeWriter(); |
| 19 |
|
virtual void Loop(); |
| 20 |
|
|
| 21 |
|
void SetProcessNEvents(int nEvents) { processNEvents = nEvents; } |
| 22 |
|
void SetReportEvents(int nEvents) { reportEvery = nEvents; } |
| 23 |
|
void SetLoggingVerbosity(int logVerb) { loggingVerbosity = logVerb; } |
| 24 |
< |
void DontSkimEvents(){ skim = false; } |
| 24 |
> |
void SkimEvents(bool skim_){ skim = skim_; } |
| 25 |
> |
void PileUpWeightFile( string pileupFileName ); |
| 26 |
|
|
| 27 |
|
TChain *inputTree; |
| 28 |
|
susy::Event *event; |
| 39 |
|
int loggingVerbosity; |
| 40 |
|
bool skim; |
| 41 |
|
|
| 42 |
+ |
// important dataset information |
| 43 |
+ |
TH1F* pileupHisto; |
| 44 |
+ |
|
| 45 |
|
// variables which will be stored in the tree |
| 46 |
|
std::vector<tree::Photon> photon; |
| 47 |
|
std::vector<tree::Jet> jet; |
| 48 |
+ |
std::vector<tree::Particle> electron; |
| 49 |
+ |
std::vector<tree::Particle> muon; |
| 50 |
+ |
|
| 51 |
|
float met; |
| 52 |
+ |
float met_phi; |
| 53 |
+ |
float type1met; |
| 54 |
+ |
float type1met_phi; |
| 55 |
|
float ht; |
| 56 |
|
int nVertex; |
| 57 |
< |
int nElectron; |
| 57 |
> |
float pu_weight; |
| 58 |
|
}; |
| 59 |
|
|
