VHbbDataFormats/interface/VHbbCandidateTools.h

#ifndef VHBBCANDIDATETOOLS_H
#define VHBBCANDIDATETOOLS_H

#include "VHbbAnalysis/VHbbDataFormats/interface/VHbbCandidate.h"

#include <iostream>

struct CompareJetPtMuons {
  bool operator()( const VHbbEvent::MuonInfo& j1, const  VHbbEvent::MuonInfo& j2 ) const {
    return j1.p4.Pt() > j2.p4.Pt();
  }
};
struct CompareJetPtElectrons {
  bool operator()( const VHbbEvent::ElectronInfo& j1, const  VHbbEvent::ElectronInfo& j2 ) const {
    return j1.p4.Pt() > j2.p4.Pt();
  }
};


class VHbbCandidateTools {
 public:

 VHbbCandidateTools(bool verbose = false): verbose_(verbose){}

  float deltaPhi(float in2, float in1){
    float dphi = in2-in1;
    if ( dphi > M_PI ) {
      dphi -= 2.0*M_PI;
    } else if ( dphi <= -M_PI ) {
      dphi += 2.0*M_PI;
    }
    return dphi;
  }

  VHbbCandidate getHZmumuCandidate(const VHbbCandidate & in, bool & ok){
    if (verbose_){
      std::cout <<" getHZmumuCandidate input mu "<<in.V.muons.size()<<" e "<<in.V.electrons.size()<<std::endl;
    }
    ok = false;
    VHbbCandidate temp=in;

    //
    // change: allow for additional leptons; by definition 
    //
    if (temp.V.muons.size()<2) return in ;
    //    if (temp.V.electrons.size()!=0) return in ;
    std::vector<VHbbEvent::MuonInfo> muons_ = temp.V.muons;
    CompareJetPtMuons ptComparator;
    std::sort(muons_.begin(), muons_.end(), ptComparator);
    if (muons_[0].p4.Pt()<20 || muons_[1].p4.Pt()<20 ) return in;
    temp.V.p4 = muons_[0].p4+muons_[1].p4;
    std::vector<VHbbEvent::MuonInfo> muons2_;
    for (std::vector<VHbbEvent::MuonInfo>::const_iterator it = muons_.begin(); it!= muons_.end(); ++it){
      if (it->p4.Pt()>20) muons2_.push_back(*it);
    }
    temp.V.muons = muons2_;
  
    // the same for electrons
    std::vector<VHbbEvent::ElectronInfo> electrons_ = temp.V.electrons;
    CompareJetPtElectrons ptComparator2;
    std::sort(electrons_.begin(), electrons_.end(), ptComparator2);
    std::vector<VHbbEvent::ElectronInfo> electrons2_;
    for (std::vector<VHbbEvent::ElectronInfo>::const_iterator it = electrons_.begin(); it!= electrons_.end(); ++it){
      if (it->p4.Pt()>20) electrons2_.push_back(*it);
    }
    temp.V.electrons = electrons2_;
    
   //
    // consider all 
    //
    
    
    //    if (temp.V.Pt()<150 ) return in;
    //    if (temp.H.Pt()<150) return in;
    //    if (temp.H.firstJet().csv< 0.9) return in;
    //    if (temp.H.secondJet().csv<0.5) return in;
    //    if (deltaPhi(temp.V.Phi(),temp.H.Phi())<2.7) return in;
    //    if (temp.V.FourMomentum.Mass()<75 || temp.V.FourMomentum.Mass()>105) return in; 
    //    if (temp.additionalJets.size()>0) return in;
    //    if (std::Abs(deltaTheta) ????
    ok = true;
    return temp;
  }  
  VHbbCandidate getHZeeCandidate(const VHbbCandidate & in, bool & ok){
    if (verbose_){
      std::cout <<" getHZeeCandidate input mu "<<in.V.electrons.size()<<" e "<<in.V.muons.size()<<std::endl;
    }
    ok = false;
    VHbbCandidate temp=in;

    //
    // change: allow for additional leptons; by definition 
    //
    if (temp.V.electrons.size()<2) return in ;
    //    if (temp.V.electrons.size()!=0) return in ;
    std::vector<VHbbEvent::ElectronInfo> electrons_ = temp.V.electrons;
    CompareJetPtElectrons ptComparator;
    std::sort(electrons_.begin(), electrons_.end(), ptComparator);
    if (electrons_[0].p4.Pt()<20 || electrons_[1].p4.Pt()<20 ) return in;
    temp.V.p4 = electrons_[0].p4+electrons_[1].p4;
    std::vector<VHbbEvent::ElectronInfo> electrons2_;
    for (std::vector<VHbbEvent::ElectronInfo>::const_iterator it = electrons_.begin(); it!= electrons_.end(); ++it){
      if (it->p4.Pt()>20) electrons2_.push_back(*it);
    }
    temp.V.electrons = electrons2_;
  
    // the same for muonss
    std::vector<VHbbEvent::MuonInfo> muons_ = temp.V.muons;
    CompareJetPtMuons ptComparator2;
    std::sort(muons_.begin(), muons_.end(), ptComparator2);
    std::vector<VHbbEvent::MuonInfo> muons2_;
    for (std::vector<VHbbEvent::MuonInfo>::const_iterator it = muons_.begin(); it!= muons_.end(); ++it){
      if (it->p4.Pt()>20) muons2_.push_back(*it);
    }
    temp.V.muons = muons2_;
    
   ok = true;
    return temp;
  }  
  VHbbCandidate getHZnnCandidate(const VHbbCandidate & in, bool & ok){
    if (verbose_){
      std::cout <<" getHZnnCandidate input mu "<<in.V.muons.size()<<" e "<<in.V.electrons.size()<<" met "<<in.V.mets.size()<<std::endl;
    }
    
    ok = false;
    VHbbCandidate temp=in;
    if (temp.V.mets.size()!=1) return in;
    if (temp.V.muons.size()!=0) return in ;
    if (temp.V.electrons.size()!=0) return in ;
    
    temp.V.p4 = temp.V.mets[0].p4;
    if (verbose_) {
      std::cout <<" debug met "<< temp.V.mets[0].metSig << " " <<  temp.V.mets[0].sumEt<< std::endl;
    }   
    if (temp.V.mets[0].metSig<5) return in;
    if (temp.V.mets[0].sumEt<50) return in;
    //    if (temp.H.p4.Pt()<150)return in;
    //    if (temp.H.firstJet().csv< 0.9) return in;
    //    if (temp.H.secondJet().csv<0.5) return in;
    //    if (deltaPhi(temp.V.p4.Phi(),temp.H.p4.Phi())<2.95) return in;
    //    if (temp.V.electrons.size()>0 || temp.V.muons.size()>0 ) return in;
    //    if (temp.additionalJets.size()>0) return in;
    //    if (std::Abs(deltaTheta) ????
    
    ok = true;
    return temp;
  }

  VHbbCandidate getHWmunCandidate(const VHbbCandidate & in, bool & ok){
    ok = false;
    VHbbCandidate temp=in;
    // require a muon and no electrons
    if (temp.V.muons.size()!=1) return in ;
    if (temp.V.electrons.size()!=0) return in ;
    if (temp.V.mets.size()<1) return in ;
    //
    /*pT(W) > 150 GeV (pt(W) computed using lepton px,y and PF MET x and y components)
      pT(H) > 150 GeV
      best btag, CSV > 0.90
      second-best btag, CSV > 0.50
      Dphi(W,H) > 2.95
      no additional isolated leptons (pT > 15 GeV)
      
      same lepton definition as in the physics objects section 
      
      No additional ak5PFjets (pT > 30 GeV; |eta| < 2.4)
      MET>35. for the electron BDT analysis
      |cos(theta * )| (TBO)
      color flow pull angle (TBO)
      We don't cut on the transverse mass (for boosted objects cutting on the transverse mass introduces an inefficiency due to the angle between the MET and the lepton being close to 0.) 
    */
    
    temp.V.p4 = temp.V.muons[0].p4+temp.V.mets[0].p4;

    ok=true;
    return temp;
  }

  VHbbCandidate getHWenCandidate(const VHbbCandidate & in, bool & ok){
    ok = false;
    VHbbCandidate temp=in;
    if (temp.V.electrons.size()!=1) return in ;
    if (temp.V.muons.size()!=0) return in ;
    if (temp.V.mets.size()<1) return in ;
    //
    /*pT(W) > 150 GeV (pt(W) computed using lepton px,y and PF MET x and y components)
      pT(H) > 150 GeV
      best btag, CSV > 0.90
      second-best btag, CSV > 0.50
      Dphi(W,H) > 2.95
      no additional isolated leptons (pT > 15 GeV)
      
      same lepton definition as in the physics objects section 
      
      No additional ak5PFjets (pT > 30 GeV; |eta| < 2.4)
      MET>35. for the electron BDT analysis
      |cos(theta * )| (TBO)
      color flow pull angle (TBO)
      We don't cut on the transverse mass (for boosted objects cutting on the transverse mass introduces an inefficiency due to the angle between the MET and the lepton being close to 0.) 
    */
    
    ok=true;

    temp.V.p4 = temp.V.electrons[0].p4+temp.V.mets[0].p4;
    return temp;
  }

 public: 
  bool verbose_;
  
 public:
  float getDeltaTheta( const VHbbEvent::SimpleJet & j1, const VHbbEvent::SimpleJet & j2 ) const {
    double deltaTheta = 1e10;
    TLorentzVector pi(0,0,0,0);
    TLorentzVector v_j1 = j1.chargedTracksFourMomentum;
    TLorentzVector v_j2 = j2.chargedTracksFourMomentum;
    
    if( v_j2.Mag() == 0 
        || v_j1.Mag() == 0 )
      return deltaTheta = 1e10;
    
    //use j1 to calculate the pull vector
    TVector2 t = j1.tVector;
    
    if( t.Mod() == 0 )
      return deltaTheta = 1e10;
    
    Double_t dphi =  v_j2.Phi()- v_j1.Phi();
    if ( dphi > M_PI ) {
      dphi -= 2.0*M_PI;
    } else if ( dphi <= -M_PI ) {
      dphi += 2.0*M_PI;
    }
    Double_t deltaeta = v_j2.Rapidity() - v_j1.Rapidity();
    TVector2 BBdir( deltaeta, dphi );
    
    deltaTheta = t.DeltaPhi(BBdir);
    
    return deltaTheta;
    
  }
  
  float getHelicity(const VHbbEvent::SimpleJet& j, TVector3 boost) const {
    double hel = 1e10;
    TLorentzVector jet = j.p4;
    jet.Boost( -boost );
    hel = TMath::Cos( jet.Vect().Angle( boost ) );
    return hel;
  }


};


#endif


Revision:	1.7
Committed:	Mon Jul 25 15:55:56 2011 UTC (13 years, 9 months ago) by tboccali
Content type:	text/plain
Branch:	MAIN
CVS Tags:	HBB_EDMNtupleV1_ProcV2, Jul28th2011, Jul26th2011
Changes since 1.6:	+70 -24 lines
Log Message:	allow for more leptons when having the Z
#	Content
1	#ifndef VHBBCANDIDATETOOLS_H
2	#define VHBBCANDIDATETOOLS_H
3
4	#include "VHbbAnalysis/VHbbDataFormats/interface/VHbbCandidate.h"
5
6	#include <iostream>
7
8	struct CompareJetPtMuons {
9	bool operator()( const VHbbEvent::MuonInfo& j1, const VHbbEvent::MuonInfo& j2 ) const {
10	return j1.p4.Pt() > j2.p4.Pt();
11	}
12	};
13	struct CompareJetPtElectrons {
14	bool operator()( const VHbbEvent::ElectronInfo& j1, const VHbbEvent::ElectronInfo& j2 ) const {
15	return j1.p4.Pt() > j2.p4.Pt();
16	}
17	};
18
19
20
21	class VHbbCandidateTools {
22	public:
23
24	VHbbCandidateTools(bool verbose = false): verbose_(verbose){}
25
26	float deltaPhi(float in2, float in1){
27	float dphi = in2-in1;
28	if ( dphi > M_PI ) {
29	dphi -= 2.0*M_PI;
30	} else if ( dphi <= -M_PI ) {
31	dphi += 2.0*M_PI;
32	}
33	return dphi;
34	}
35
36	VHbbCandidate getHZmumuCandidate(const VHbbCandidate & in, bool & ok){
37	if (verbose_){
38	std::cout <<" getHZmumuCandidate input mu "<<in.V.muons.size()<<" e "<<in.V.electrons.size()<<std::endl;
39	}
40	ok = false;
41	VHbbCandidate temp=in;
42
43	//
44	// change: allow for additional leptons; by definition
45	//
46	if (temp.V.muons.size()<2) return in ;
47	// if (temp.V.electrons.size()!=0) return in ;
48	std::vector<VHbbEvent::MuonInfo> muons_ = temp.V.muons;
49	CompareJetPtMuons ptComparator;
50	std::sort(muons_.begin(), muons_.end(), ptComparator);
51	if (muons_[0].p4.Pt()<20 \|\| muons_[1].p4.Pt()<20 ) return in;
52	temp.V.p4 = muons_[0].p4+muons_[1].p4;
53	std::vector<VHbbEvent::MuonInfo> muons2_;
54	for (std::vector<VHbbEvent::MuonInfo>::const_iterator it = muons_.begin(); it!= muons_.end(); ++it){
55	if (it->p4.Pt()>20) muons2_.push_back(*it);
56	}
57	temp.V.muons = muons2_;
58
59	// the same for electrons
60	std::vector<VHbbEvent::ElectronInfo> electrons_ = temp.V.electrons;
61	CompareJetPtElectrons ptComparator2;
62	std::sort(electrons_.begin(), electrons_.end(), ptComparator2);
63	std::vector<VHbbEvent::ElectronInfo> electrons2_;
64	for (std::vector<VHbbEvent::ElectronInfo>::const_iterator it = electrons_.begin(); it!= electrons_.end(); ++it){
65	if (it->p4.Pt()>20) electrons2_.push_back(*it);
66	}
67	temp.V.electrons = electrons2_;
68
69	//
70	// consider all
71	//
72
73
74	// if (temp.V.Pt()<150 ) return in;
75	// if (temp.H.Pt()<150) return in;
76	// if (temp.H.firstJet().csv< 0.9) return in;
77	// if (temp.H.secondJet().csv<0.5) return in;
78	// if (deltaPhi(temp.V.Phi(),temp.H.Phi())<2.7) return in;
79	// if (temp.V.FourMomentum.Mass()<75 \|\| temp.V.FourMomentum.Mass()>105) return in;
80	// if (temp.additionalJets.size()>0) return in;
81	// if (std::Abs(deltaTheta) ????
82	ok = true;
83	return temp;
84	}
85	VHbbCandidate getHZeeCandidate(const VHbbCandidate & in, bool & ok){
86	if (verbose_){
87	std::cout <<" getHZeeCandidate input mu "<<in.V.electrons.size()<<" e "<<in.V.muons.size()<<std::endl;
88	}
89	ok = false;
90	VHbbCandidate temp=in;
91
92	//
93	// change: allow for additional leptons; by definition
94	//
95	if (temp.V.electrons.size()<2) return in ;
96	// if (temp.V.electrons.size()!=0) return in ;
97	std::vector<VHbbEvent::ElectronInfo> electrons_ = temp.V.electrons;
98	CompareJetPtElectrons ptComparator;
99	std::sort(electrons_.begin(), electrons_.end(), ptComparator);
100	if (electrons_[0].p4.Pt()<20 \|\| electrons_[1].p4.Pt()<20 ) return in;
101	temp.V.p4 = electrons_[0].p4+electrons_[1].p4;
102	std::vector<VHbbEvent::ElectronInfo> electrons2_;
103	for (std::vector<VHbbEvent::ElectronInfo>::const_iterator it = electrons_.begin(); it!= electrons_.end(); ++it){
104	if (it->p4.Pt()>20) electrons2_.push_back(*it);
105	}
106	temp.V.electrons = electrons2_;
107
108	// the same for muonss
109	std::vector<VHbbEvent::MuonInfo> muons_ = temp.V.muons;
110	CompareJetPtMuons ptComparator2;
111	std::sort(muons_.begin(), muons_.end(), ptComparator2);
112	std::vector<VHbbEvent::MuonInfo> muons2_;
113	for (std::vector<VHbbEvent::MuonInfo>::const_iterator it = muons_.begin(); it!= muons_.end(); ++it){
114	if (it->p4.Pt()>20) muons2_.push_back(*it);
115	}
116	temp.V.muons = muons2_;
117
118	ok = true;
119	return temp;
120	}
121	VHbbCandidate getHZnnCandidate(const VHbbCandidate & in, bool & ok){
122	if (verbose_){
123	std::cout <<" getHZnnCandidate input mu "<<in.V.muons.size()<<" e "<<in.V.electrons.size()<<" met "<<in.V.mets.size()<<std::endl;
124	}
125
126	ok = false;
127	VHbbCandidate temp=in;
128	if (temp.V.mets.size()!=1) return in;
129	if (temp.V.muons.size()!=0) return in ;
130	if (temp.V.electrons.size()!=0) return in ;
131
132	temp.V.p4 = temp.V.mets[0].p4;
133	if (verbose_) {
134	std::cout <<" debug met "<< temp.V.mets[0].metSig << " " << temp.V.mets[0].sumEt<< std::endl;
135	}
136	if (temp.V.mets[0].metSig<5) return in;
137	if (temp.V.mets[0].sumEt<50) return in;
138	// if (temp.H.p4.Pt()<150)return in;
139	// if (temp.H.firstJet().csv< 0.9) return in;
140	// if (temp.H.secondJet().csv<0.5) return in;
141	// if (deltaPhi(temp.V.p4.Phi(),temp.H.p4.Phi())<2.95) return in;
142	// if (temp.V.electrons.size()>0 \|\| temp.V.muons.size()>0 ) return in;
143	// if (temp.additionalJets.size()>0) return in;
144	// if (std::Abs(deltaTheta) ????
145
146	ok = true;
147	return temp;
148	}
149
150	VHbbCandidate getHWmunCandidate(const VHbbCandidate & in, bool & ok){
151	ok = false;
152	VHbbCandidate temp=in;
153	// require a muon and no electrons
154	if (temp.V.muons.size()!=1) return in ;
155	if (temp.V.electrons.size()!=0) return in ;
156	if (temp.V.mets.size()<1) return in ;
157	//
158	/*pT(W) > 150 GeV (pt(W) computed using lepton px,y and PF MET x and y components)
159	pT(H) > 150 GeV
160	best btag, CSV > 0.90
161	second-best btag, CSV > 0.50
162	Dphi(W,H) > 2.95
163	no additional isolated leptons (pT > 15 GeV)
164
165	same lepton definition as in the physics objects section
166
167	No additional ak5PFjets (pT > 30 GeV; \|eta\| < 2.4)
168	MET>35. for the electron BDT analysis
169	\|cos(theta * )\| (TBO)
170	color flow pull angle (TBO)
171	We don't cut on the transverse mass (for boosted objects cutting on the transverse mass introduces an inefficiency due to the angle between the MET and the lepton being close to 0.)
172	*/
173
174	temp.V.p4 = temp.V.muons[0].p4+temp.V.mets[0].p4;
175
176	ok=true;
177	return temp;
178	}
179
180	VHbbCandidate getHWenCandidate(const VHbbCandidate & in, bool & ok){
181	ok = false;
182	VHbbCandidate temp=in;
183	if (temp.V.electrons.size()!=1) return in ;
184	if (temp.V.muons.size()!=0) return in ;
185	if (temp.V.mets.size()<1) return in ;
186	//
187	/*pT(W) > 150 GeV (pt(W) computed using lepton px,y and PF MET x and y components)
188	pT(H) > 150 GeV
189	best btag, CSV > 0.90
190	second-best btag, CSV > 0.50
191	Dphi(W,H) > 2.95
192	no additional isolated leptons (pT > 15 GeV)
193
194	same lepton definition as in the physics objects section
195
196	No additional ak5PFjets (pT > 30 GeV; \|eta\| < 2.4)
197	MET>35. for the electron BDT analysis
198	\|cos(theta * )\| (TBO)
199	color flow pull angle (TBO)
200	We don't cut on the transverse mass (for boosted objects cutting on the transverse mass introduces an inefficiency due to the angle between the MET and the lepton being close to 0.)
201	*/
202
203	ok=true;
204
205	temp.V.p4 = temp.V.electrons[0].p4+temp.V.mets[0].p4;
206	return temp;
207	}
208
209	public:
210	bool verbose_;
211
212	public:
213	float getDeltaTheta( const VHbbEvent::SimpleJet & j1, const VHbbEvent::SimpleJet & j2 ) const {
214	double deltaTheta = 1e10;
215	TLorentzVector pi(0,0,0,0);
216	TLorentzVector v_j1 = j1.chargedTracksFourMomentum;
217	TLorentzVector v_j2 = j2.chargedTracksFourMomentum;
218
219	if( v_j2.Mag() == 0
220	\|\| v_j1.Mag() == 0 )
221	return deltaTheta = 1e10;
222
223	//use j1 to calculate the pull vector
224	TVector2 t = j1.tVector;
225
226	if( t.Mod() == 0 )
227	return deltaTheta = 1e10;
228
229	Double_t dphi = v_j2.Phi()- v_j1.Phi();
230	if ( dphi > M_PI ) {
231	dphi -= 2.0*M_PI;
232	} else if ( dphi <= -M_PI ) {
233	dphi += 2.0*M_PI;
234	}
235	Double_t deltaeta = v_j2.Rapidity() - v_j1.Rapidity();
236	TVector2 BBdir( deltaeta, dphi );
237
238	deltaTheta = t.DeltaPhi(BBdir);
239
240	return deltaTheta;
241
242	}
243
244	float getHelicity(const VHbbEvent::SimpleJet& j, TVector3 boost) const {
245	double hel = 1e10;
246	TLorentzVector jet = j.p4;
247	jet.Boost( -boost );
248	hel = TMath::Cos( jet.Vect().Angle( boost ) );
249	return hel;
250	}
251
252
253	};
254
255
256
257	#endif
258
259
260
261
262
263