kiesel/TreeWriter/treeWriter.h

#include<iostream>
#include<math.h>

#include "TFile.h"
#include "TTree.h"
#include "TChain.h"
#include "TString.h"
#include "TH1F.h"

#include "SusyEvent.h"
#include "TreeObjects.h"

class TreeWriter {
        public :
                TreeWriter(TString inputName, TString outputName, int loggingVerbosity_);
                TreeWriter(TChain* inputName, TString outputName, int loggingVerbosity_);
                void Init( TString outputName, int loggingVerbosity_ );
                virtual ~TreeWriter();
                virtual void Loop();

                void SetProcessNEvents(int nEvents) { processNEvents = nEvents; }
                void SetReportEvents(int nEvents) { reportEvery = nEvents; }
                void SetLoggingVerbosity(int logVerb) { loggingVerbosity = logVerb; }
                void SkimEvents(bool skim_){ skim = skim_; }
                void PileUpWeightFile( string pileupFileName );

                TChain *inputTree;
                susy::Event *event;

                TFile *outFile;
                TTree *tree;
                TH1F *eventNumbers;

                float getPtFromMatchedJet( susy::Photon, susy::Event );
                float deltaR( TLorentzVector, TLorentzVector );

        private:
                int processNEvents; // number of events to be processed
                int reportEvery;
                int loggingVerbosity;
                bool skim;

                // important dataset information
                TH1F* pileupHisto;

                // variables which will be stored in the tree
                std::vector<tree::Photon> photon;
                std::vector<tree::Jet> jet;
                std::vector<tree::Particle> electron;
                std::vector<tree::Particle> muon;

                float met;
                float met_phi;
                float type1met;
                float type1met_phi;
                float ht;
                int nVertex;
                float pu_weight;
};

Revision:	1.6
Committed:	Wed Apr 17 09:57:52 2013 UTC (12 years ago) by kiesel
Content type:	text/plain
Branch:	MAIN
Changes since 1.5:	+1 -0 lines
Log Message:	vertex information for electron selection less buggy In the state right now, code still aborts when returning d0. I hope this can be fixed with cint->Make+cpp
#	Content
1	#include<iostream>
2	#include<math.h>
3
4	#include "TFile.h"
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, 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 SkimEvents(bool skim_){ skim = skim_; }
25	void PileUpWeightFile( string pileupFileName );
26
27	TChain *inputTree;
28	susy::Event *event;
29
30	TFile *outFile;
31	TTree *tree;
32	TH1F *eventNumbers;
33
34	float getPtFromMatchedJet( susy::Photon, susy::Event );
35	float deltaR( TLorentzVector, TLorentzVector );
36
37	private:
38	int processNEvents; // number of events to be processed
39	int reportEvery;
40	int loggingVerbosity;
41	bool skim;
42
43	// important dataset information
44	TH1F* pileupHisto;
45
46	// variables which will be stored in the tree
47	std::vector<tree::Photon> photon;
48	std::vector<tree::Jet> jet;
49	std::vector<tree::Particle> electron;
50	std::vector<tree::Particle> muon;
51
52	float met;
53	float met_phi;
54	float type1met;
55	float type1met_phi;
56	float ht;
57	int nVertex;
58	float pu_weight;
59	};
60