PollackPrograms/src/TCTau.h

#ifndef _TCTau_H
#define _TCTau_H

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

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

                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;

                int DecayMode() const { return decayMode; }
                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 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.2
Committed:	Wed Feb 20 21:39:44 2013 UTC (12 years, 2 months ago) by bpollack
Content type:	text/plain
Branch:	MAIN
CVS Tags:	HEAD
Changes since 1.1:	+2 -15 lines
Log Message:	commiting moriond analysis package
#	User	Rev	Content
1	bpollack	1.1	#ifndef _TCTau_H
2			#define _TCTau_H
3
4			#include "TObject.h"
5			#include "TLorentzVector.h"
6	bpollack	1.2	#include "TCPhysObject.h"
7	bpollack	1.1	#include <vector>
8
9	bpollack	1.2	class TCTau : public TCPhysObject {
10	bpollack	1.1	private:
11			TVector3 _positionFromTauObject;
12			TVector3 _positionFromLeadTrack;
13
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			public:
62			TCTau();
63			virtual ~TCTau();
64
65			TVector3 PositionFromTrack() const;
66			TVector3 PositionFromTau() const;
67
68			int DecayMode() const { return decayMode; }
69			int NChHad() const {return nChHad; }
70			int NGamma() const {return nGamma; }
71			int NNeutrHad() const {return nNeutrHad; }
72
73			TLorentzVector LeadChHadP4() const { return leadChHadP4; }
74			TLorentzVector LeadNeutrP4() const { return leadNeutrP4; }
75
76			float IsoGammaEtSum() { return isoGammaEtSum; }
77			float IsoChHadPtSum() { return isoChHadPtSum; }
78
79			void SetDecayMode(int d) {decayMode = d; }
80			void SetNChHad(int n) { nChHad = n; }
81			void SetNGamma(int n) { nGamma = n; }
82			void SetNNeutrHad(int n) { nNeutrHad = n; }
83
84			void SetPositionFromTau(float x, float y, float z);
85			void SetPositionFromTrack(float x, float y, float z);
86
87			void SetLeadChHadP4(TLorentzVector p4) { leadChHadP4 = p4; }
88			void SetLeadNeutrP4(TLorentzVector p4) { leadNeutrP4 = p4; }
89
90			void SetLeadChHadP4(float x, float y, float z, float e) { leadChHadP4.SetPxPyPzE(x,y,z,e); }
91			void SetLeadNeutrP4(float x, float y, float z, float e) { leadNeutrP4.SetPxPyPzE(x,y,z,e); }
92
93			void SetIsoGammaEtSum(float i) { isoGammaEtSum = i; }
94			void SetIsoChHadPtSum(float i) { isoChHadPtSum = i; }
95
96
97			// Discriminators: for now only for HPS
98
99			// HPS
100			float GetHpsPFTauDiscriminationAgainstElectronLoose() {
101			return hpsPFTauDiscriminationAgainstElectronLoose;
102			}
103			float GetHpsPFTauDiscriminationAgainstMuonLoose() {
104			return hpsPFTauDiscriminationAgainstMuonLoose;
105			}
106			float GetHpsPFTauDiscriminationAgainstElectronMedium() {
107			return hpsPFTauDiscriminationAgainstElectronMedium;
108			}
109			float GetHpsPFTauDiscriminationAgainstMuonMedium() {
110			return hpsPFTauDiscriminationAgainstMuonMedium;
111			}
112			float GetHpsPFTauDiscriminationAgainstElectronTight() {
113			return hpsPFTauDiscriminationAgainstElectronTight;
114			}
115			float GetHpsPFTauDiscriminationAgainstMuonTight() {
116			return hpsPFTauDiscriminationAgainstMuonTight;
117			}
118			float GetHpsPFTauDiscriminationByDecayModeFinding() {
119			return hpsPFTauDiscriminationByDecayModeFinding;
120			}
121			float GetHpsPFTauDiscriminationByVLooseIsolation() {
122			return hpsPFTauDiscriminationByVLooseIsolation;
123			}
124			float GetHpsPFTauDiscriminationByLooseIsolation() {
125			return hpsPFTauDiscriminationByLooseIsolation;
126			}
127			float GetHpsPFTauDiscriminationByMediumIsolation() {
128			return hpsPFTauDiscriminationByMediumIsolation;
129			}
130			float GetHpsPFTauDiscriminationByTightIsolation() {
131			return hpsPFTauDiscriminationByTightIsolation;
132			}
133
134
135
136			float GetHpsPFTauDiscriminationByVLooseIsolationDBSumPtCorr() {
137			return hpsPFTauDiscriminationByVLooseIsolationDBSumPtCorr;
138			}
139			float GetHpsPFTauDiscriminationByLooseIsolationDBSumPtCorr() {
140			return hpsPFTauDiscriminationByLooseIsolationDBSumPtCorr;
141			}
142			float GetHpsPFTauDiscriminationByMediumIsolationDBSumPtCorr() {
143			return hpsPFTauDiscriminationByMediumIsolationDBSumPtCorr;
144			}
145			float GetHpsPFTauDiscriminationByTightIsolationDBSumPtCorr() {
146			return hpsPFTauDiscriminationByTightIsolationDBSumPtCorr;
147			}
148
149
150			float GetHpsPFTauDiscriminationByVLooseCombinedIsolationDBSumPtCorr() {
151			return hpsPFTauDiscriminationByVLooseCombinedIsolationDBSumPtCorr;
152			}
153			float GetHpsPFTauDiscriminationByLooseCombinedIsolationDBSumPtCorr() {
154			return hpsPFTauDiscriminationByLooseCombinedIsolationDBSumPtCorr;
155			}
156			float GetHpsPFTauDiscriminationByMediumCombinedIsolationDBSumPtCorr() {
157			return hpsPFTauDiscriminationByMediumCombinedIsolationDBSumPtCorr;
158			}
159			float GetHpsPFTauDiscriminationByTightCombinedIsolationDBSumPtCorr() {
160			return hpsPFTauDiscriminationByTightCombinedIsolationDBSumPtCorr;
161			}
162
163
164
165			// HPS discriminator setters
166			void SetHpsPFTauDiscriminationAgainstElectronLoose(float f) {
167			hpsPFTauDiscriminationAgainstElectronLoose = f;
168			}
169			void SetHpsPFTauDiscriminationAgainstMuonLoose(float f) {
170			hpsPFTauDiscriminationAgainstMuonLoose = f;
171			}
172			void SetHpsPFTauDiscriminationAgainstElectronMedium(float f) {
173			hpsPFTauDiscriminationAgainstElectronMedium = f;
174			}
175			void SetHpsPFTauDiscriminationAgainstMuonMedium(float f) {
176			hpsPFTauDiscriminationAgainstMuonMedium = f;
177			}
178			void SetHpsPFTauDiscriminationAgainstElectronTight(float f) {
179			hpsPFTauDiscriminationAgainstElectronTight = f;
180			}
181			void SetHpsPFTauDiscriminationAgainstMuonTight(float f) {
182			hpsPFTauDiscriminationAgainstMuonTight = f;
183			}
184			void SetHpsPFTauDiscriminationByDecayModeFinding(float f) {
185			hpsPFTauDiscriminationByDecayModeFinding = f;
186			}
187			void SetHpsPFTauDiscriminationByVLooseIsolation(float f) {
188			hpsPFTauDiscriminationByVLooseIsolation = f;
189			}
190			void SetHpsPFTauDiscriminationByLooseIsolation(float f) {
191			hpsPFTauDiscriminationByLooseIsolation = f;
192			}
193			void SetHpsPFTauDiscriminationByMediumIsolation(float f) {
194			hpsPFTauDiscriminationByMediumIsolation = f;
195			}
196			void SetHpsPFTauDiscriminationByTightIsolation(float f) {
197			hpsPFTauDiscriminationByTightIsolation = f;
198			}
199
200			void SetHpsPFTauDiscriminationByVLooseIsolationDBSumPtCorr(float f) {
201			hpsPFTauDiscriminationByVLooseCombinedIsolationDBSumPtCorr = f;
202			}
203			void SetHpsPFTauDiscriminationByLooseIsolationDBSumPtCorr(float f) {
204			hpsPFTauDiscriminationByLooseIsolationDBSumPtCorr = f;
205			}
206			void SetHpsPFTauDiscriminationByMediumIsolationDBSumPtCorr(float f) {
207			hpsPFTauDiscriminationByMediumIsolationDBSumPtCorr = f;
208			}
209			void SetHpsPFTauDiscriminationByTightIsolationDBSumPtCorr(float f) {
210			hpsPFTauDiscriminationByTightIsolationDBSumPtCorr = f;
211			}
212
213
214			void SetHpsPFTauDiscriminationByVLooseCombinedIsolationDBSumPtCorr(float f) {
215			hpsPFTauDiscriminationByVLooseCombinedIsolationDBSumPtCorr = f;
216			}
217			void SetHpsPFTauDiscriminationByLooseCombinedIsolationDBSumPtCorr(float f) {
218			hpsPFTauDiscriminationByLooseCombinedIsolationDBSumPtCorr = f;
219			}
220			void SetHpsPFTauDiscriminationByMediumCombinedIsolationDBSumPtCorr(float f) {
221			hpsPFTauDiscriminationByMediumCombinedIsolationDBSumPtCorr = f;
222			}
223			void SetHpsPFTauDiscriminationByTightCombinedIsolationDBSumPtCorr(float f) {
224			hpsPFTauDiscriminationByTightCombinedIsolationDBSumPtCorr = f;
225			}
226
227
228			ClassDef(TCTau, 5);
229
230			};
231
232			#endif