PollackPrograms/src/TCTau.h

#ifndef _TCTau_H
#define _TCTau_H

#include "TObject.h"
#include "TLorentzVector.h"
#include <vector>

class TCTau : public TLorentzVector {
        private:
                TVector3 _positionFromTauObject;
                TVector3 _positionFromLeadTrack;

                int tauCharge;
                int nChHad;
                int nGamma;
                int nNeutrHad;
                int decayMode;

                TLorentzVector leadChHadP4;
                TLorentzVector leadNeutrP4;

                float isoGammaEtSum;
                float isoChHadPtSum;


                // Discriminators - names as used in CMSSW
                // For the most part the discriminators are binary

                // HPS discriminators ---------
                //
                // against e,mu misidentification
                float hpsPFTauDiscriminationAgainstElectronLoose;        
                float hpsPFTauDiscriminationAgainstMuonLoose;        
                float hpsPFTauDiscriminationAgainstElectronMedium;        
                float hpsPFTauDiscriminationAgainstMuonMedium;        
                float hpsPFTauDiscriminationAgainstElectronTight;        
                float hpsPFTauDiscriminationAgainstMuonTight;        
                
                // by decay mode 
                float hpsPFTauDiscriminationByDecayModeFinding;

                // by charged isolation
                float hpsPFTauDiscriminationByVLooseIsolation;
                float hpsPFTauDiscriminationByLooseIsolation;        
                float hpsPFTauDiscriminationByMediumIsolation;        
                float hpsPFTauDiscriminationByTightIsolation;
                
                // with iso corrections
                float hpsPFTauDiscriminationByVLooseIsolationDBSumPtCorr;
                float hpsPFTauDiscriminationByLooseIsolationDBSumPtCorr;
                float hpsPFTauDiscriminationByMediumIsolationDBSumPtCorr;
                float hpsPFTauDiscriminationByTightIsolationDBSumPtCorr;

                // by combined isolation with iso corrections
                float hpsPFTauDiscriminationByVLooseCombinedIsolationDBSumPtCorr;
                float hpsPFTauDiscriminationByLooseCombinedIsolationDBSumPtCorr;
                float hpsPFTauDiscriminationByMediumCombinedIsolationDBSumPtCorr;
                float hpsPFTauDiscriminationByTightCombinedIsolationDBSumPtCorr;


        public:
                TCTau();
                virtual ~TCTau();

                TVector3 PositionFromTrack() const;
                TVector3 PositionFromTau() const;

                TVector2 P2() const;

                int DecayMode() const { return decayMode; }
                int Charge() const {return tauCharge; }
                int NChHad() const  {return nChHad; }
                int NGamma() const {return nGamma; }
                int NNeutrHad() const {return nNeutrHad; }

                TLorentzVector LeadChHadP4() const { return leadChHadP4; }
                TLorentzVector LeadNeutrP4() const { return leadNeutrP4; }

                float IsoGammaEtSum() { return isoGammaEtSum; }
                float IsoChHadPtSum() { return isoChHadPtSum; }


                void SetDecayMode(int d) {decayMode = d; } 
                void SetCharge(int c) { tauCharge = c; }
                void SetNChHad(int n) { nChHad = n; }
                void SetNGamma(int n) { nGamma = n; }
                void SetNNeutrHad(int n) { nNeutrHad = n; }

                void SetPositionFromTau(float x, float y, float z);
                void SetPositionFromTrack(float x, float y, float z);


                void SetLeadChHadP4(TLorentzVector p4) { leadChHadP4 = p4; }
                void SetLeadNeutrP4(TLorentzVector p4) { leadNeutrP4 = p4; }

                void SetLeadChHadP4(float x, float y, float z, float e) { leadChHadP4.SetPxPyPzE(x,y,z,e); }
                void SetLeadNeutrP4(float x, float y, float z, float e) { leadNeutrP4.SetPxPyPzE(x,y,z,e); }


                void SetIsoGammaEtSum(float i) { isoGammaEtSum = i; }
                void SetIsoChHadPtSum(float i) { isoChHadPtSum = i; }


                // Discriminators: for now only for HPS

                // HPS
                float GetHpsPFTauDiscriminationAgainstElectronLoose() {
                  return hpsPFTauDiscriminationAgainstElectronLoose;
                }
                float GetHpsPFTauDiscriminationAgainstMuonLoose() {
                  return hpsPFTauDiscriminationAgainstMuonLoose;
                }
                float GetHpsPFTauDiscriminationAgainstElectronMedium() {
                  return hpsPFTauDiscriminationAgainstElectronMedium;
                }
                float GetHpsPFTauDiscriminationAgainstMuonMedium() {
                  return hpsPFTauDiscriminationAgainstMuonMedium;
                }
                float GetHpsPFTauDiscriminationAgainstElectronTight() {
                  return hpsPFTauDiscriminationAgainstElectronTight;
                }
                float GetHpsPFTauDiscriminationAgainstMuonTight() {
                  return hpsPFTauDiscriminationAgainstMuonTight;
                }
                float GetHpsPFTauDiscriminationByDecayModeFinding() {
                  return hpsPFTauDiscriminationByDecayModeFinding;
                }
                float GetHpsPFTauDiscriminationByVLooseIsolation() {
                  return hpsPFTauDiscriminationByVLooseIsolation;
                }
                float GetHpsPFTauDiscriminationByLooseIsolation() {
                  return hpsPFTauDiscriminationByLooseIsolation;
                }
                float GetHpsPFTauDiscriminationByMediumIsolation() {
                  return hpsPFTauDiscriminationByMediumIsolation;
                }
                float GetHpsPFTauDiscriminationByTightIsolation() {
                  return hpsPFTauDiscriminationByTightIsolation;
                }
                

                float GetHpsPFTauDiscriminationByVLooseIsolationDBSumPtCorr() {
                  return hpsPFTauDiscriminationByVLooseIsolationDBSumPtCorr;
                }
                float GetHpsPFTauDiscriminationByLooseIsolationDBSumPtCorr() {
                  return hpsPFTauDiscriminationByLooseIsolationDBSumPtCorr;
                }
                float GetHpsPFTauDiscriminationByMediumIsolationDBSumPtCorr() {
                  return hpsPFTauDiscriminationByMediumIsolationDBSumPtCorr;
                }
                float GetHpsPFTauDiscriminationByTightIsolationDBSumPtCorr() {
                  return hpsPFTauDiscriminationByTightIsolationDBSumPtCorr;
                }


                float GetHpsPFTauDiscriminationByVLooseCombinedIsolationDBSumPtCorr() {
                  return hpsPFTauDiscriminationByVLooseCombinedIsolationDBSumPtCorr;
                }
                float GetHpsPFTauDiscriminationByLooseCombinedIsolationDBSumPtCorr() {
                  return hpsPFTauDiscriminationByLooseCombinedIsolationDBSumPtCorr;
                }
                float GetHpsPFTauDiscriminationByMediumCombinedIsolationDBSumPtCorr() {
                  return hpsPFTauDiscriminationByMediumCombinedIsolationDBSumPtCorr;
                }
                float GetHpsPFTauDiscriminationByTightCombinedIsolationDBSumPtCorr() {
                  return hpsPFTauDiscriminationByTightCombinedIsolationDBSumPtCorr;
                }


                // HPS discriminator setters
                void SetHpsPFTauDiscriminationAgainstElectronLoose(float f) {
                  hpsPFTauDiscriminationAgainstElectronLoose = f;
                }
                void SetHpsPFTauDiscriminationAgainstMuonLoose(float f) {
                  hpsPFTauDiscriminationAgainstMuonLoose = f;
                }
                void SetHpsPFTauDiscriminationAgainstElectronMedium(float f) {
                  hpsPFTauDiscriminationAgainstElectronMedium = f;
                }
                void SetHpsPFTauDiscriminationAgainstMuonMedium(float f) {
                  hpsPFTauDiscriminationAgainstMuonMedium = f;
                }
                void SetHpsPFTauDiscriminationAgainstElectronTight(float f) {
                  hpsPFTauDiscriminationAgainstElectronTight = f;
                }
                void SetHpsPFTauDiscriminationAgainstMuonTight(float f) {
                  hpsPFTauDiscriminationAgainstMuonTight = f;
                }
                void SetHpsPFTauDiscriminationByDecayModeFinding(float f) {
                  hpsPFTauDiscriminationByDecayModeFinding = f;
                }
                void SetHpsPFTauDiscriminationByVLooseIsolation(float f) {
                  hpsPFTauDiscriminationByVLooseIsolation = f;
                }
                void SetHpsPFTauDiscriminationByLooseIsolation(float f) {
                  hpsPFTauDiscriminationByLooseIsolation = f;
                }
                void SetHpsPFTauDiscriminationByMediumIsolation(float f) {
                  hpsPFTauDiscriminationByMediumIsolation = f;
                }
                void SetHpsPFTauDiscriminationByTightIsolation(float f) {
                  hpsPFTauDiscriminationByTightIsolation = f;
                }

                void SetHpsPFTauDiscriminationByVLooseIsolationDBSumPtCorr(float f) {
                  hpsPFTauDiscriminationByVLooseCombinedIsolationDBSumPtCorr = f;
                }
                void SetHpsPFTauDiscriminationByLooseIsolationDBSumPtCorr(float f) {
                  hpsPFTauDiscriminationByLooseIsolationDBSumPtCorr = f;
                }
                void SetHpsPFTauDiscriminationByMediumIsolationDBSumPtCorr(float f) {
                  hpsPFTauDiscriminationByMediumIsolationDBSumPtCorr = f;
                }
                void SetHpsPFTauDiscriminationByTightIsolationDBSumPtCorr(float f) {
                  hpsPFTauDiscriminationByTightIsolationDBSumPtCorr = f;
                }


                void SetHpsPFTauDiscriminationByVLooseCombinedIsolationDBSumPtCorr(float f) {
                  hpsPFTauDiscriminationByVLooseCombinedIsolationDBSumPtCorr = f;
                }
                void SetHpsPFTauDiscriminationByLooseCombinedIsolationDBSumPtCorr(float f) {
                  hpsPFTauDiscriminationByLooseCombinedIsolationDBSumPtCorr = f;
                }
                void SetHpsPFTauDiscriminationByMediumCombinedIsolationDBSumPtCorr(float f) {
                  hpsPFTauDiscriminationByMediumCombinedIsolationDBSumPtCorr = f;
                }
                void SetHpsPFTauDiscriminationByTightCombinedIsolationDBSumPtCorr(float f) {
                  hpsPFTauDiscriminationByTightCombinedIsolationDBSumPtCorr = f;
                }


                ClassDef(TCTau, 5);

};

#endif
Revision:	1.1
Committed:	Tue Jul 24 20:24:58 2012 UTC (12 years, 9 months ago) by bpollack
Content type:	text/plain
Branch:	MAIN
Log Message:	adding a cleaned-up analyzer for 2012
#	User	Rev	Content
1	bpollack	1.1	#ifndef _TCTau_H
2			#define _TCTau_H
3
4			#include "TObject.h"
5			#include "TLorentzVector.h"
6			#include <vector>
7
8			class TCTau : public TLorentzVector {
9			private:
10			TVector3 _positionFromTauObject;
11			TVector3 _positionFromLeadTrack;
12
13			int tauCharge;
14			int nChHad;
15			int nGamma;
16			int nNeutrHad;
17			int decayMode;
18
19			TLorentzVector leadChHadP4;
20			TLorentzVector leadNeutrP4;
21
22			float isoGammaEtSum;
23			float isoChHadPtSum;
24
25
26			// Discriminators - names as used in CMSSW
27			// For the most part the discriminators are binary
28
29			// HPS discriminators ---------
30			//
31			// against e,mu misidentification
32			float hpsPFTauDiscriminationAgainstElectronLoose;
33			float hpsPFTauDiscriminationAgainstMuonLoose;
34			float hpsPFTauDiscriminationAgainstElectronMedium;
35			float hpsPFTauDiscriminationAgainstMuonMedium;
36			float hpsPFTauDiscriminationAgainstElectronTight;
37			float hpsPFTauDiscriminationAgainstMuonTight;
38
39			// by decay mode
40			float hpsPFTauDiscriminationByDecayModeFinding;
41
42			// by charged isolation
43			float hpsPFTauDiscriminationByVLooseIsolation;
44			float hpsPFTauDiscriminationByLooseIsolation;
45			float hpsPFTauDiscriminationByMediumIsolation;
46			float hpsPFTauDiscriminationByTightIsolation;
47
48			// with iso corrections
49			float hpsPFTauDiscriminationByVLooseIsolationDBSumPtCorr;
50			float hpsPFTauDiscriminationByLooseIsolationDBSumPtCorr;
51			float hpsPFTauDiscriminationByMediumIsolationDBSumPtCorr;
52			float hpsPFTauDiscriminationByTightIsolationDBSumPtCorr;
53
54			// by combined isolation with iso corrections
55			float hpsPFTauDiscriminationByVLooseCombinedIsolationDBSumPtCorr;
56			float hpsPFTauDiscriminationByLooseCombinedIsolationDBSumPtCorr;
57			float hpsPFTauDiscriminationByMediumCombinedIsolationDBSumPtCorr;
58			float hpsPFTauDiscriminationByTightCombinedIsolationDBSumPtCorr;
59
60
61
62
63
64
65
66			public:
67			TCTau();
68			virtual ~TCTau();
69
70			TVector3 PositionFromTrack() const;
71			TVector3 PositionFromTau() const;
72
73			TVector2 P2() const;
74
75			int DecayMode() const { return decayMode; }
76			int Charge() const {return tauCharge; }
77			int NChHad() const {return nChHad; }
78			int NGamma() const {return nGamma; }
79			int NNeutrHad() const {return nNeutrHad; }
80
81			TLorentzVector LeadChHadP4() const { return leadChHadP4; }
82			TLorentzVector LeadNeutrP4() const { return leadNeutrP4; }
83
84			float IsoGammaEtSum() { return isoGammaEtSum; }
85			float IsoChHadPtSum() { return isoChHadPtSum; }
86
87
88
89			void SetDecayMode(int d) {decayMode = d; }
90			void SetCharge(int c) { tauCharge = c; }
91			void SetNChHad(int n) { nChHad = n; }
92			void SetNGamma(int n) { nGamma = n; }
93			void SetNNeutrHad(int n) { nNeutrHad = n; }
94
95			void SetPositionFromTau(float x, float y, float z);
96			void SetPositionFromTrack(float x, float y, float z);
97
98
99			void SetLeadChHadP4(TLorentzVector p4) { leadChHadP4 = p4; }
100			void SetLeadNeutrP4(TLorentzVector p4) { leadNeutrP4 = p4; }
101
102			void SetLeadChHadP4(float x, float y, float z, float e) { leadChHadP4.SetPxPyPzE(x,y,z,e); }
103			void SetLeadNeutrP4(float x, float y, float z, float e) { leadNeutrP4.SetPxPyPzE(x,y,z,e); }
104
105
106			void SetIsoGammaEtSum(float i) { isoGammaEtSum = i; }
107			void SetIsoChHadPtSum(float i) { isoChHadPtSum = i; }
108
109
110			// Discriminators: for now only for HPS
111
112			// HPS
113			float GetHpsPFTauDiscriminationAgainstElectronLoose() {
114			return hpsPFTauDiscriminationAgainstElectronLoose;
115			}
116			float GetHpsPFTauDiscriminationAgainstMuonLoose() {
117			return hpsPFTauDiscriminationAgainstMuonLoose;
118			}
119			float GetHpsPFTauDiscriminationAgainstElectronMedium() {
120			return hpsPFTauDiscriminationAgainstElectronMedium;
121			}
122			float GetHpsPFTauDiscriminationAgainstMuonMedium() {
123			return hpsPFTauDiscriminationAgainstMuonMedium;
124			}
125			float GetHpsPFTauDiscriminationAgainstElectronTight() {
126			return hpsPFTauDiscriminationAgainstElectronTight;
127			}
128			float GetHpsPFTauDiscriminationAgainstMuonTight() {
129			return hpsPFTauDiscriminationAgainstMuonTight;
130			}
131			float GetHpsPFTauDiscriminationByDecayModeFinding() {
132			return hpsPFTauDiscriminationByDecayModeFinding;
133			}
134			float GetHpsPFTauDiscriminationByVLooseIsolation() {
135			return hpsPFTauDiscriminationByVLooseIsolation;
136			}
137			float GetHpsPFTauDiscriminationByLooseIsolation() {
138			return hpsPFTauDiscriminationByLooseIsolation;
139			}
140			float GetHpsPFTauDiscriminationByMediumIsolation() {
141			return hpsPFTauDiscriminationByMediumIsolation;
142			}
143			float GetHpsPFTauDiscriminationByTightIsolation() {
144			return hpsPFTauDiscriminationByTightIsolation;
145			}
146
147
148
149			float GetHpsPFTauDiscriminationByVLooseIsolationDBSumPtCorr() {
150			return hpsPFTauDiscriminationByVLooseIsolationDBSumPtCorr;
151			}
152			float GetHpsPFTauDiscriminationByLooseIsolationDBSumPtCorr() {
153			return hpsPFTauDiscriminationByLooseIsolationDBSumPtCorr;
154			}
155			float GetHpsPFTauDiscriminationByMediumIsolationDBSumPtCorr() {
156			return hpsPFTauDiscriminationByMediumIsolationDBSumPtCorr;
157			}
158			float GetHpsPFTauDiscriminationByTightIsolationDBSumPtCorr() {
159			return hpsPFTauDiscriminationByTightIsolationDBSumPtCorr;
160			}
161
162
163			float GetHpsPFTauDiscriminationByVLooseCombinedIsolationDBSumPtCorr() {
164			return hpsPFTauDiscriminationByVLooseCombinedIsolationDBSumPtCorr;
165			}
166			float GetHpsPFTauDiscriminationByLooseCombinedIsolationDBSumPtCorr() {
167			return hpsPFTauDiscriminationByLooseCombinedIsolationDBSumPtCorr;
168			}
169			float GetHpsPFTauDiscriminationByMediumCombinedIsolationDBSumPtCorr() {
170			return hpsPFTauDiscriminationByMediumCombinedIsolationDBSumPtCorr;
171			}
172			float GetHpsPFTauDiscriminationByTightCombinedIsolationDBSumPtCorr() {
173			return hpsPFTauDiscriminationByTightCombinedIsolationDBSumPtCorr;
174			}
175
176
177
178			// HPS discriminator setters
179			void SetHpsPFTauDiscriminationAgainstElectronLoose(float f) {
180			hpsPFTauDiscriminationAgainstElectronLoose = f;
181			}
182			void SetHpsPFTauDiscriminationAgainstMuonLoose(float f) {
183			hpsPFTauDiscriminationAgainstMuonLoose = f;
184			}
185			void SetHpsPFTauDiscriminationAgainstElectronMedium(float f) {
186			hpsPFTauDiscriminationAgainstElectronMedium = f;
187			}
188			void SetHpsPFTauDiscriminationAgainstMuonMedium(float f) {
189			hpsPFTauDiscriminationAgainstMuonMedium = f;
190			}
191			void SetHpsPFTauDiscriminationAgainstElectronTight(float f) {
192			hpsPFTauDiscriminationAgainstElectronTight = f;
193			}
194			void SetHpsPFTauDiscriminationAgainstMuonTight(float f) {
195			hpsPFTauDiscriminationAgainstMuonTight = f;
196			}
197			void SetHpsPFTauDiscriminationByDecayModeFinding(float f) {
198			hpsPFTauDiscriminationByDecayModeFinding = f;
199			}
200			void SetHpsPFTauDiscriminationByVLooseIsolation(float f) {
201			hpsPFTauDiscriminationByVLooseIsolation = f;
202			}
203			void SetHpsPFTauDiscriminationByLooseIsolation(float f) {
204			hpsPFTauDiscriminationByLooseIsolation = f;
205			}
206			void SetHpsPFTauDiscriminationByMediumIsolation(float f) {
207			hpsPFTauDiscriminationByMediumIsolation = f;
208			}
209			void SetHpsPFTauDiscriminationByTightIsolation(float f) {
210			hpsPFTauDiscriminationByTightIsolation = f;
211			}
212
213			void SetHpsPFTauDiscriminationByVLooseIsolationDBSumPtCorr(float f) {
214			hpsPFTauDiscriminationByVLooseCombinedIsolationDBSumPtCorr = f;
215			}
216			void SetHpsPFTauDiscriminationByLooseIsolationDBSumPtCorr(float f) {
217			hpsPFTauDiscriminationByLooseIsolationDBSumPtCorr = f;
218			}
219			void SetHpsPFTauDiscriminationByMediumIsolationDBSumPtCorr(float f) {
220			hpsPFTauDiscriminationByMediumIsolationDBSumPtCorr = f;
221			}
222			void SetHpsPFTauDiscriminationByTightIsolationDBSumPtCorr(float f) {
223			hpsPFTauDiscriminationByTightIsolationDBSumPtCorr = f;
224			}
225
226
227			void SetHpsPFTauDiscriminationByVLooseCombinedIsolationDBSumPtCorr(float f) {
228			hpsPFTauDiscriminationByVLooseCombinedIsolationDBSumPtCorr = f;
229			}
230			void SetHpsPFTauDiscriminationByLooseCombinedIsolationDBSumPtCorr(float f) {
231			hpsPFTauDiscriminationByLooseCombinedIsolationDBSumPtCorr = f;
232			}
233			void SetHpsPFTauDiscriminationByMediumCombinedIsolationDBSumPtCorr(float f) {
234			hpsPFTauDiscriminationByMediumCombinedIsolationDBSumPtCorr = f;
235			}
236			void SetHpsPFTauDiscriminationByTightCombinedIsolationDBSumPtCorr(float f) {
237			hpsPFTauDiscriminationByTightCombinedIsolationDBSumPtCorr = f;
238			}
239
240
241			ClassDef(TCTau, 5);
242
243			};
244
245			#endif