NtupleWriter/Objects/Jet.h

#ifndef Jet_H
#define Jet_H

#include "Particle.h"

/**
 *  @short jet class
 *  @author Thomas Peiffer
 */

class Jet : public Particle{

 public:

  Jet(){
     m_nTracks=0;
     m_jetArea=0;
     m_pileup=0;
     m_numberOfDaughters=0; 
     m_neutralEmEnergyFraction=0;
     m_neutralHadronEnergyFraction=0;
     m_chargedEmEnergyFraction=0;
     m_chargedHadronEnergyFraction=0;
     m_muonEnergyFraction=0;
     m_photonEnergyFraction=0;
     m_chargedMultiplicity=0;
     m_neutralMultiplicity=0;
     m_muonMultiplicity=0; 
     m_electronMultiplicity=0;
     m_photonMultiplicity=0;
     m_btag_simpleSecondaryVertexHighEff=0;
     m_btag_simpleSecondaryVertexHighPur=0;
     m_btag_combinedSecondaryVertex=0;
     m_btag_combinedSecondaryVertexMVA=0;
     m_btag_jetBProbability=0;
     m_btag_jetProbability=0;
     m_JEC_uncertainty=0;
     m_JEC_factor_raw=0;
     m_genjet_pt=0;
     m_genjet_eta=0;
     m_genjet_phi=0;
     m_genjet_energy=0;
     m_genparticles_indices.clear();
  };

  ~Jet(){
  };

  LorentzVector genjet_v4() const{
    LorentzVector v4;
    v4.SetPt(m_genjet_pt);
    v4.SetEta(m_genjet_eta);
    v4.SetPhi(m_genjet_phi);
    v4.SetE(m_genjet_energy);
    return v4;
  };

  int nTracks() const{return m_nTracks;}
  float jetArea() const{return m_jetArea;}
  float pileup() const{return m_pileup;}
  int numberOfDaughters() const{return m_numberOfDaughters;} 
  float neutralEmEnergyFraction() const{return m_neutralEmEnergyFraction;}
  float neutralHadronEnergyFraction() const{return m_neutralHadronEnergyFraction;}
  float chargedEmEnergyFraction() const{return m_chargedEmEnergyFraction;}
  float chargedHadronEnergyFraction() const{return m_chargedHadronEnergyFraction;}
  float muonEnergyFraction() const{return m_muonEnergyFraction;}
  float photonEnergyFraction() const{return m_photonEnergyFraction;}
  int chargedMultiplicity() const{return m_chargedMultiplicity;}
  int neutralMultiplicity() const{return m_neutralMultiplicity;}
  int muonMultiplicity() const{return m_muonMultiplicity;} 
  int electronMultiplicity() const{return m_electronMultiplicity;}
  int photonMultiplicity() const{return m_photonMultiplicity;}
  float btag_simpleSecondaryVertexHighEff() const{return m_btag_simpleSecondaryVertexHighEff;}
  float btag_simpleSecondaryVertexHighPur() const{return m_btag_simpleSecondaryVertexHighPur;}
  float btag_combinedSecondaryVertex() const{return m_btag_combinedSecondaryVertex;}
  float btag_combinedSecondaryVertexMVA() const{return m_btag_combinedSecondaryVertexMVA;}
  float btag_jetBProbability() const{return m_btag_jetBProbability;}
  float btag_jetProbability() const{return m_btag_jetProbability;}
  float JEC_uncertainty() const{return m_JEC_uncertainty;}
  float JEC_factor_raw() const{return m_JEC_factor_raw;}
  float genjet_pt() const{return m_genjet_pt;}
  float genjet_eta() const{return m_genjet_eta;}
  float genjet_phi() const{return m_genjet_phi;}
  float genjet_energy() const{return m_genjet_energy;}
  std::vector<unsigned int> genparticles_indices() const{return m_genparticles_indices;}

  void set_nTracks(int x){m_nTracks=x;}
  void set_jetArea(float x){m_jetArea=x;}
  void set_pileup(float x){m_pileup=x;}
  void set_numberOfDaughters(int x){m_numberOfDaughters=x;} 
  void set_neutralEmEnergyFraction(float x){m_neutralEmEnergyFraction=x;}
  void set_neutralHadronEnergyFraction(float x){m_neutralHadronEnergyFraction=x;}
  void set_chargedEmEnergyFraction(float x){m_chargedEmEnergyFraction=x;}
  void set_chargedHadronEnergyFraction(float x){m_chargedHadronEnergyFraction=x;}
  void set_muonEnergyFraction(float x){m_muonEnergyFraction=x;}
  void set_photonEnergyFraction(float x){m_photonEnergyFraction=x;}
  void set_chargedMultiplicity(int x){m_chargedMultiplicity=x;}
  void set_neutralMultiplicity(int x){m_neutralMultiplicity=x;}
  void set_muonMultiplicity(int x){m_muonMultiplicity=x;} 
  void set_electronMultiplicity(int x){m_electronMultiplicity=x;}
  void set_photonMultiplicity(int x){m_photonMultiplicity=x;}
  void set_btag_simpleSecondaryVertexHighEff(float x){m_btag_simpleSecondaryVertexHighEff=x;}
  void set_btag_simpleSecondaryVertexHighPur(float x){m_btag_simpleSecondaryVertexHighPur=x;}
  void set_btag_combinedSecondaryVertex(float x){m_btag_combinedSecondaryVertex=x;}
  void set_btag_combinedSecondaryVertexMVA(float x){m_btag_combinedSecondaryVertexMVA=x;}
  void set_btag_jetBProbability(float x){m_btag_jetBProbability=x;}
  void set_btag_jetProbability(float x){m_btag_jetProbability=x;}
  void set_JEC_uncertainty(float x){m_JEC_uncertainty=x;}
  void set_JEC_factor_raw(float x){m_JEC_factor_raw=x;}
  void set_genjet_pt(float x){m_genjet_pt=x;}
  void set_genjet_eta(float x){m_genjet_eta=x;}
  void set_genjet_phi(float x){m_genjet_phi=x;}
  void set_genjet_energy(float x){m_genjet_energy=x;}
  void add_genparticles_index(unsigned int x){m_genparticles_indices.push_back(x);}

  bool has_genjet() const{return m_genjet_pt>0;}

  bool pfID(){
    //pf ID has already been applied when using goodPatJets
    if(numberOfDaughters()>1 
       && neutralHadronEnergyFraction()<0.99
       && neutralEmEnergyFraction()<0.99){
      
      if(fabs(eta())>=2.4)
        return true;
      
      if(chargedEmEnergyFraction()<0.99
         && chargedHadronEnergyFraction()>0
         && chargedMultiplicity()>0)
        return true;   
    }
    return false;
  }

 private:
  
  int m_nTracks;
  float m_jetArea;
  float m_pileup;
  int m_numberOfDaughters; 
  float m_neutralEmEnergyFraction;
  float m_neutralHadronEnergyFraction;
  float m_chargedEmEnergyFraction;
  float m_chargedHadronEnergyFraction;
  float m_muonEnergyFraction;
  float m_photonEnergyFraction;
  int m_chargedMultiplicity;
  int m_neutralMultiplicity;
  int m_muonMultiplicity; 
  int m_electronMultiplicity;
  int m_photonMultiplicity;
  float m_btag_simpleSecondaryVertexHighEff;
  float m_btag_simpleSecondaryVertexHighPur;
  float m_btag_combinedSecondaryVertex;
  float m_btag_combinedSecondaryVertexMVA;
  float m_btag_jetBProbability;
  float m_btag_jetProbability;
  float m_JEC_uncertainty;
  float m_JEC_factor_raw;
  float m_genjet_pt;
  float m_genjet_eta;
  float m_genjet_phi;
  float m_genjet_energy;

  std::vector<unsigned int> m_genparticles_indices;

};

#endif
Revision:	1.5
Committed:	Thu Jun 28 16:04:11 2012 UTC (12 years, 10 months ago) by peiffer
Content type:	text/plain
Branch:	MAIN
CVS Tags:	Jan-17-2013-v2, Jan-17-2013-v1, Jan-16-2012-v1, Jan-09-2012-v2, Jan-09-2012-v1, Dec-26-2012-v1, Dec-20-2012-v1, Dec-17-2012-v1, Dec-05-2012-v1, Nov-30-2012-v3, Nov-30-2012-v2, Nov-30-2012-v1
Changes since 1.4:	+17 -0 lines
Log Message:	moved lepton and jet id
#	Content
1	#ifndef Jet_H
2	#define Jet_H
3
4	#include "Particle.h"
5
6	/**
7	* @short jet class
8	* @author Thomas Peiffer
9	*/
10
11	class Jet : public Particle{
12
13	public:
14
15	Jet(){
16	m_nTracks=0;
17	m_jetArea=0;
18	m_pileup=0;
19	m_numberOfDaughters=0;
20	m_neutralEmEnergyFraction=0;
21	m_neutralHadronEnergyFraction=0;
22	m_chargedEmEnergyFraction=0;
23	m_chargedHadronEnergyFraction=0;
24	m_muonEnergyFraction=0;
25	m_photonEnergyFraction=0;
26	m_chargedMultiplicity=0;
27	m_neutralMultiplicity=0;
28	m_muonMultiplicity=0;
29	m_electronMultiplicity=0;
30	m_photonMultiplicity=0;
31	m_btag_simpleSecondaryVertexHighEff=0;
32	m_btag_simpleSecondaryVertexHighPur=0;
33	m_btag_combinedSecondaryVertex=0;
34	m_btag_combinedSecondaryVertexMVA=0;
35	m_btag_jetBProbability=0;
36	m_btag_jetProbability=0;
37	m_JEC_uncertainty=0;
38	m_JEC_factor_raw=0;
39	m_genjet_pt=0;
40	m_genjet_eta=0;
41	m_genjet_phi=0;
42	m_genjet_energy=0;
43	m_genparticles_indices.clear();
44	};
45
46	~Jet(){
47	};
48
49	LorentzVector genjet_v4() const{
50	LorentzVector v4;
51	v4.SetPt(m_genjet_pt);
52	v4.SetEta(m_genjet_eta);
53	v4.SetPhi(m_genjet_phi);
54	v4.SetE(m_genjet_energy);
55	return v4;
56	};
57
58	int nTracks() const{return m_nTracks;}
59	float jetArea() const{return m_jetArea;}
60	float pileup() const{return m_pileup;}
61	int numberOfDaughters() const{return m_numberOfDaughters;}
62	float neutralEmEnergyFraction() const{return m_neutralEmEnergyFraction;}
63	float neutralHadronEnergyFraction() const{return m_neutralHadronEnergyFraction;}
64	float chargedEmEnergyFraction() const{return m_chargedEmEnergyFraction;}
65	float chargedHadronEnergyFraction() const{return m_chargedHadronEnergyFraction;}
66	float muonEnergyFraction() const{return m_muonEnergyFraction;}
67	float photonEnergyFraction() const{return m_photonEnergyFraction;}
68	int chargedMultiplicity() const{return m_chargedMultiplicity;}
69	int neutralMultiplicity() const{return m_neutralMultiplicity;}
70	int muonMultiplicity() const{return m_muonMultiplicity;}
71	int electronMultiplicity() const{return m_electronMultiplicity;}
72	int photonMultiplicity() const{return m_photonMultiplicity;}
73	float btag_simpleSecondaryVertexHighEff() const{return m_btag_simpleSecondaryVertexHighEff;}
74	float btag_simpleSecondaryVertexHighPur() const{return m_btag_simpleSecondaryVertexHighPur;}
75	float btag_combinedSecondaryVertex() const{return m_btag_combinedSecondaryVertex;}
76	float btag_combinedSecondaryVertexMVA() const{return m_btag_combinedSecondaryVertexMVA;}
77	float btag_jetBProbability() const{return m_btag_jetBProbability;}
78	float btag_jetProbability() const{return m_btag_jetProbability;}
79	float JEC_uncertainty() const{return m_JEC_uncertainty;}
80	float JEC_factor_raw() const{return m_JEC_factor_raw;}
81	float genjet_pt() const{return m_genjet_pt;}
82	float genjet_eta() const{return m_genjet_eta;}
83	float genjet_phi() const{return m_genjet_phi;}
84	float genjet_energy() const{return m_genjet_energy;}
85	std::vector<unsigned int> genparticles_indices() const{return m_genparticles_indices;}
86
87	void set_nTracks(int x){m_nTracks=x;}
88	void set_jetArea(float x){m_jetArea=x;}
89	void set_pileup(float x){m_pileup=x;}
90	void set_numberOfDaughters(int x){m_numberOfDaughters=x;}
91	void set_neutralEmEnergyFraction(float x){m_neutralEmEnergyFraction=x;}
92	void set_neutralHadronEnergyFraction(float x){m_neutralHadronEnergyFraction=x;}
93	void set_chargedEmEnergyFraction(float x){m_chargedEmEnergyFraction=x;}
94	void set_chargedHadronEnergyFraction(float x){m_chargedHadronEnergyFraction=x;}
95	void set_muonEnergyFraction(float x){m_muonEnergyFraction=x;}
96	void set_photonEnergyFraction(float x){m_photonEnergyFraction=x;}
97	void set_chargedMultiplicity(int x){m_chargedMultiplicity=x;}
98	void set_neutralMultiplicity(int x){m_neutralMultiplicity=x;}
99	void set_muonMultiplicity(int x){m_muonMultiplicity=x;}
100	void set_electronMultiplicity(int x){m_electronMultiplicity=x;}
101	void set_photonMultiplicity(int x){m_photonMultiplicity=x;}
102	void set_btag_simpleSecondaryVertexHighEff(float x){m_btag_simpleSecondaryVertexHighEff=x;}
103	void set_btag_simpleSecondaryVertexHighPur(float x){m_btag_simpleSecondaryVertexHighPur=x;}
104	void set_btag_combinedSecondaryVertex(float x){m_btag_combinedSecondaryVertex=x;}
105	void set_btag_combinedSecondaryVertexMVA(float x){m_btag_combinedSecondaryVertexMVA=x;}
106	void set_btag_jetBProbability(float x){m_btag_jetBProbability=x;}
107	void set_btag_jetProbability(float x){m_btag_jetProbability=x;}
108	void set_JEC_uncertainty(float x){m_JEC_uncertainty=x;}
109	void set_JEC_factor_raw(float x){m_JEC_factor_raw=x;}
110	void set_genjet_pt(float x){m_genjet_pt=x;}
111	void set_genjet_eta(float x){m_genjet_eta=x;}
112	void set_genjet_phi(float x){m_genjet_phi=x;}
113	void set_genjet_energy(float x){m_genjet_energy=x;}
114	void add_genparticles_index(unsigned int x){m_genparticles_indices.push_back(x);}
115
116	bool has_genjet() const{return m_genjet_pt>0;}
117
118	bool pfID(){
119	//pf ID has already been applied when using goodPatJets
120	if(numberOfDaughters()>1
121	&& neutralHadronEnergyFraction()<0.99
122	&& neutralEmEnergyFraction()<0.99){
123
124	if(fabs(eta())>=2.4)
125	return true;
126
127	if(chargedEmEnergyFraction()<0.99
128	&& chargedHadronEnergyFraction()>0
129	&& chargedMultiplicity()>0)
130	return true;
131	}
132	return false;
133	}
134
135	private:
136
137	int m_nTracks;
138	float m_jetArea;
139	float m_pileup;
140	int m_numberOfDaughters;
141	float m_neutralEmEnergyFraction;
142	float m_neutralHadronEnergyFraction;
143	float m_chargedEmEnergyFraction;
144	float m_chargedHadronEnergyFraction;
145	float m_muonEnergyFraction;
146	float m_photonEnergyFraction;
147	int m_chargedMultiplicity;
148	int m_neutralMultiplicity;
149	int m_muonMultiplicity;
150	int m_electronMultiplicity;
151	int m_photonMultiplicity;
152	float m_btag_simpleSecondaryVertexHighEff;
153	float m_btag_simpleSecondaryVertexHighPur;
154	float m_btag_combinedSecondaryVertex;
155	float m_btag_combinedSecondaryVertexMVA;
156	float m_btag_jetBProbability;
157	float m_btag_jetProbability;
158	float m_JEC_uncertainty;
159	float m_JEC_factor_raw;
160	float m_genjet_pt;
161	float m_genjet_eta;
162	float m_genjet_phi;
163	float m_genjet_energy;
164
165	std::vector<unsigned int> m_genparticles_indices;
166
167	};
168
169	#endif