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
#	User	Rev	Content
1	kiesel	1.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	kiesel	1.4	#include "TH1F.h"
9	kiesel	1.1
10			#include "SusyEvent.h"
11			#include "TreeObjects.h"
12
13			class TreeWriter {
14			public :
15	kiesel	1.3	TreeWriter(TString inputName, TString outputName, int loggingVerbosity_);
16	kiesel	1.5	TreeWriter(TChain* inputName, TString outputName, int loggingVerbosity_);
17			void Init( TString outputName, int loggingVerbosity_ );
18	kiesel	1.1	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	kiesel	1.2	void SkimEvents(bool skim_){ skim = skim_; }
25	kiesel	1.4	void PileUpWeightFile( string pileupFileName );
26	kiesel	1.1
27			TChain *inputTree;
28			susy::Event *event;
29
30			TFile *outFile;
31			TTree *tree;
32	kiesel	1.6	TH1F *eventNumbers;
33	kiesel	1.1
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	kiesel	1.4	// important dataset information
44			TH1F* pileupHisto;
45
46	kiesel	1.1	// variables which will be stored in the tree
47			std::vector<tree::Photon> photon;
48			std::vector<tree::Jet> jet;
49	kiesel	1.4	std::vector<tree::Particle> electron;
50			std::vector<tree::Particle> muon;
51
52	kiesel	1.1	float met;
53	kiesel	1.4	float met_phi;
54			float type1met;
55			float type1met_phi;
56	kiesel	1.1	float ht;
57			int nVertex;
58	kiesel	1.4	float pu_weight;
59	kiesel	1.1	};
60