1 |
|
#include<iostream> |
2 |
|
#include<math.h> |
3 |
+ |
#include<string> |
4 |
|
|
5 |
+ |
#include "TSystem.h" |
6 |
|
#include "TFile.h" |
7 |
|
#include "TTree.h" |
8 |
|
#include "TChain.h" |
7 |
– |
#include "TString.h" |
9 |
|
#include "TH1F.h" |
10 |
|
|
11 |
|
#include "SusyEvent.h" |
13 |
|
|
14 |
|
class TreeWriter { |
15 |
|
public : |
16 |
< |
TreeWriter(TString inputName, TString outputName, int loggingVerbosity_); |
17 |
< |
TreeWriter(TChain* inputName, TString outputName, int loggingVerbosity_); |
18 |
< |
void Init( TString outputName, int loggingVerbosity_ ); |
16 |
> |
TreeWriter(std::string inputName, std::string outputName, int loggingVerbosity_); |
17 |
> |
TreeWriter(TChain* inputName, std::string outputName, int loggingVerbosity_); |
18 |
> |
void Init( std::string outputName, int loggingVerbosity_ ); |
19 |
|
virtual ~TreeWriter(); |
20 |
|
virtual void Loop(); |
21 |
|
|
23 |
|
void SetReportEvents(int nEvents) { reportEvery = nEvents; } |
24 |
|
void SetLoggingVerbosity(int logVerb) { loggingVerbosity = logVerb; } |
25 |
|
void SkimEvents(bool skim_){ skim = skim_; } |
26 |
< |
void PileUpWeightFile( string pileupFileName ); |
26 |
> |
void PileUpWeightFile( std::string pileupFileName ); |
27 |
|
|
28 |
|
TChain *inputTree; |
29 |
|
susy::Event *event; |
32 |
|
TTree *tree; |
33 |
|
TH1F *eventNumbers; |
34 |
|
|
34 |
– |
float getPtFromMatchedJet( susy::Photon, susy::Event ); |
35 |
– |
float deltaR( TLorentzVector, TLorentzVector ); |
36 |
– |
|
35 |
|
private: |
36 |
|
int processNEvents; // number of events to be processed |
37 |
|
int reportEvery; |