[ViewVC] Diff of: cvsroot/UserCode/MitPhysics/Utils/src/RecoilTools.cc

Comparing UserCode/MitPhysics/Utils/src/RecoilTools.cc (file contents):
Revision 1.7 by pharris, Mon Apr 23 17:40:23 2012 UTC vs.
Revision 1.24 by pharris, Sat Jan 12 11:49:50 2013 UTC

#	Line 6 \| Line 6 \| ClassImp(mithep::RecoilTools)
6
7		using namespace mithep;
8
9	<	RecoilTools::RecoilTools(TString iJetLowPtMVAFile,TString iJetHighPtMVAFile,TString iCutFile) {
9	>	RecoilTools::RecoilTools(TString iJetLowPtMVAFile,TString iJetHighPtMVAFile,TString iCutFile,bool i42,JetIDMVA::MVAType iType) {
10		fJetIDMVA = new JetIDMVA();
11	<	fJetIDMVA->Initialize( JetIDMVA::kMET,iJetLowPtMVAFile,iJetHighPtMVAFile,JetIDMVA::kBaseline,iCutFile);
11	>	fJetIDMVA->Initialize( JetIDMVA::kMET,iJetLowPtMVAFile,iJetHighPtMVAFile,iType,iCutFile);
12	>	f42 = i42;
13		}
14		//--------------------------------------------------------------------------------------------------
15		RecoilTools::~RecoilTools() {
#	Line 26 \| Line 27 \| *bool RecoilTools::filter(const PFJet iJ**
27		if(pDR2 < 0.5) return false;
28		return true;
29		}
30	<
30	>	//--------------------------------------------------------------------------------------------------
31	>	bool RecoilTools::filter(const PFCandidate *iCand,Double_t iPhi1,Double_t iEta1,Double_t iPhi2,Double_t iEta2) {
32	>	double pDEta1 = iCand->Eta() - iEta1;
33	>	double pDPhi1 = fabs(iCand->Phi() - iPhi1); if(pDPhi1 > 2.TMath::Pi()-pDPhi1) pDPhi1 = 2.TMath::Pi()-pDPhi1;
34	>	double pDR1 = sqrt(pDEta1pDEta1 + pDPhi1pDPhi1);
35	>	if(pDR1 < 0.3) return false;
36	>	double pDEta2 = iCand->Eta() - iEta2;
37	>	double pDPhi2 = fabs(iCand->Phi() - iPhi2); if(pDPhi2 > 2.TMath::Pi()-pDPhi2) pDPhi2 = 2.TMath::Pi()-pDPhi2;
38	>	double pDR2 = sqrt(pDEta2pDEta2 + pDPhi2pDPhi2);
39	>	if(pDR2 < 0.3) return false;
40	>	return true;
41	>	}
42		//--------------------------------------------------------------------------------------------------
43		Met RecoilTools::pfRecoil(Double_t iVisPt,Double_t iVisPhi,Double_t iVisSumEt,
44	<	const PFMet *iMet) {
45	<	Met lPFMet(iMet->Px(),iMet->Py());
46	<	lPFMet.SetSumEt(iMet->SumEt());
44	>	const PFCandidateCol *iCands) {
45	>	double lSumEt = 0;
46	>	FourVectorM lVec(0,0,0,0);
47	>	for(UInt_t i0 = 0; i0 < iCands->GetEntries(); i0++) {
48	>	lVec -= iCands->At(i0)->Mom();
49	>	lSumEt += iCands->At(i0)->Pt();
50	>	}
51	>	Met lPFMet(lVec.Px(),lVec.Py());
52	>	lPFMet.SetSumEt(lSumEt);
53		lPFMet.SetMex (lPFMet.Mex()+iVisPt*cos(iVisPhi));
54		lPFMet.SetMey (lPFMet.Mey()+iVisPt*sin(iVisPhi));
55		lPFMet.SetSumEt(lPFMet.SumEt()-iVisSumEt);
56		return lPFMet;
57		}
58		//--------------------------------------------------------------------------------------------------
59	+	Met RecoilTools::pfRecoil(double iPhi1,double iEta1,double iPhi2,double iEta2,
60	+	const PFCandidateCol *iCands) {
61	+	double lSumEt = 0;
62	+	FourVectorM lVec(0,0,0,0);
63	+	for(UInt_t i0 = 0; i0 < iCands->GetEntries(); i0++) {
64	+	const PFCandidate *pfcand = iCands->At(i0);
65	+	if(!filter(pfcand,iPhi1,iEta1,iPhi2,iEta2)) continue;
66	+	lVec -= pfcand->Mom();
67	+	lSumEt += pfcand->Pt();
68	+	}
69	+	Met lPFMet(lVec.Px(),lVec.Py());
70	+	lPFMet.SetSumEt(lSumEt);
71	+	return lPFMet;
72	+
73	+	}
74	+	//--------------------------------------------------------------------------------------------------
75	+	void RecoilTools::addType1(FourVectorM &iVec,Double_t &iSumEt,
76	+	const PFJetCol iJets,FactorizedJetCorrector iJetCorrector,const PileupEnergyDensityCol *iPUEnergyDensity,
77	+	double iPhi1,double iEta1,double iPhi2,double iEta2) {
78	+	for(UInt_t i0 = 0; i0 < iJets->GetEntries(); i0++) {
79	+	const PFJet *pJet = iJets->At(i0);
80	+	if(!JetTools::passPFLooseId(pJet)) continue;
81	+	if(!filter(pJet,iPhi1,iEta1,iPhi2,iEta2)) continue;
82	+	if(fJetIDMVA->correctedPt(pJet,iJetCorrector,iPUEnergyDensity) < 10) continue;
83	+	double lPt = fJetIDMVA->correctedPt(pJet,iJetCorrector,iPUEnergyDensity,RhoUtilities::DEFAULT,100);
84	+	FourVectorM pVec(0,0,0,0);
85	+	pVec.SetPt(lPt); pVec.SetEta(pJet->Eta()); pVec.SetPhi(pJet->Phi()); pVec.SetM(pJet->Mass());
86	+	iVec -= pVec;
87	+	iSumEt += pVec.Pt();
88	+	}
89	+	}
90	+	//--------------------------------------------------------------------------------------------------
91	+	Met RecoilTools::pfRecoilType1(Double_t iVisPt,Double_t iVisPhi,Double_t iVisSumEt,
92	+	const PFCandidateCol iCands,const PFJetCol iJets,
93	+	FactorizedJetCorrector iJetCorrector,const PileupEnergyDensityCol iPUEnergyDensity,
94	+	double iPhi1,double iEta1,double iPhi2,double iEta2) {
95	+	double lSumEt = 0;
96	+	FourVectorM lVec(0,0,0,0);
97	+	for(UInt_t i0 = 0; i0 < iCands->GetEntries(); i0++) {
98	+	lVec -= iCands->At(i0)->Mom();
99	+	lSumEt += iCands->At(i0)->Pt();
100	+	}
101	+	addType1(lVec,lSumEt,iJets,iJetCorrector,iPUEnergyDensity,iPhi1,iEta1,iPhi2,iEta2);
102	+	Met lPFMet(lVec.Px(),lVec.Py());
103	+	lPFMet.SetSumEt(lSumEt);
104	+	lPFMet.SetMex (lPFMet.Mex()+iVisPt*cos(iVisPhi));
105	+	lPFMet.SetMey (lPFMet.Mey()+iVisPt*sin(iVisPhi));
106	+	lPFMet.SetSumEt(lPFMet.SumEt()-iVisSumEt);
107	+	return lPFMet;
108	+	}
109	+	//--------------------------------------------------------------------------------------------------
110	+	Met RecoilTools::pfCone(double iPhi1,double iEta1,
111	+	const PFCandidateCol iCands,const Vertex iVertex,bool iCharge,Double_t iDZCut) {
112	+	double lSumEt = 0;
113	+	FourVectorM lVec(0,0,0,0);
114	+	for(UInt_t i0 = 0; i0 < iCands->GetEntries(); i0++) {
115	+	const PFCandidate *pfcand = iCands->At(i0);
116	+	if(filter(pfcand,iPhi1,iEta1,100,100)) continue;
117	+	if(iCharge) {
118	+	double lDZ = -999;
119	+	if(pfcand->HasTrackerTrk()) lDZ = fabs(pfcand->TrackerTrk()->DzCorrected(*iVertex));//
120	+	else if(pfcand->HasGsfTrk()) lDZ = fabs(pfcand->GsfTrk() ->DzCorrected(*iVertex));
121	+	if( fabs(lDZ) > iDZCut) continue;
122	+	}
123	+	lVec += pfcand->Mom();
124	+	lSumEt += pfcand->Pt();
125	+	}
126	+	Met lPFMet(lVec.Px(),lVec.Py());
127	+	lPFMet.SetSumEt(lSumEt);
128	+	return lPFMet;
129	+	}
130	+	//--------------------------------------------------------------------------------------------------
131		Met RecoilTools::trackMet(const PFCandidateCol iCands,const Vertex iVertex,Double_t iDZCut) {
132		double trkMetx = 0;
133		double trkMety = 0;
134		double trkSumEt = 0;
135	<	for(UInt_t i=0; i<iCands->GetEntries(); ++i) {
136	<	const PFCandidate *pfcand = iCands->At(i);
137	<	if( (pfcand->HasTrackerTrk() && (fabs(pfcand->TrackerTrk()->DzCorrected(*iVertex))< iDZCut)) ) {
138	<	//(pfcand->HasGsfTrk() && (fabs(pfcand->GsfTrk()->DzCorrected(*iVertex)) < iDZCut)) ) {
135	>	for(UInt_t i0=0; i0<iCands->GetEntries(); i0++) {
136	>	const PFCandidate *pfcand = iCands->At(i0);
137	>	double lDZ = -999;
138	>	if(pfcand->HasTrackerTrk()) lDZ = fabs(pfcand->TrackerTrk()->DzCorrected(*iVertex));//
139	>	else if(pfcand->HasGsfTrk()) lDZ = fabs(pfcand->GsfTrk() ->DzCorrected(*iVertex));
140	>	if( fabs(lDZ) > iDZCut) continue;
141	>	trkMetx -= pfcand->Px();
142	>	trkMety -= pfcand->Py();
143	>	trkSumEt += pfcand->Pt();
144	>	}
145	>	Met lMet(trkMetx,trkMety);
146	>	lMet.SetSumEt(trkSumEt);
147	>	return lMet;
148	>	}
149	>	//--------------------------------------------------------------------------------------------------
150	>	Met RecoilTools::trackRecoil(double iPhi1,double iEta1,double iPhi2,double iEta2,const PFCandidateCol iCands,const Vertex iVertex,Double_t iDZCut) {
151	>	double trkMetx = 0;
152	>	double trkMety = 0;
153	>	double trkSumEt = 0;
154	>	for(UInt_t i0=0; i0<iCands->GetEntries(); i0++) {
155	>	const PFCandidate *pfcand = iCands->At(i0);
156	>	if(!filter(pfcand,iPhi1,iEta1,iPhi2,iEta2)) continue;
157	>	if( (pfcand->HasTrackerTrk() && (fabs(pfcand->TrackerTrk()->DzCorrected(*iVertex))< iDZCut)) \|\|
158	>	(pfcand->HasGsfTrk() && (fabs(pfcand->GsfTrk()->DzCorrected(*iVertex)) < iDZCut)) ) {
159		trkMetx -= pfcand->Px();
160		trkMety -= pfcand->Py();
161		trkSumEt += pfcand->Pt();
#	Line 55 \| Line 165 \| Met RecoilTools::trackMet(const PFCandid
165		lMet.SetSumEt(trkSumEt);
166		return lMet;
167		}
58	–
168		//--------------------------------------------------------------------------------------------------
169		//Compute the recoil => here this requires the vector sum of the visible components
170		//VisPt => Vector sum pT of the visible non-recoiling components
#	Line 74 \| Line 183 \| *void RecoilTools::addNeut(const PFJet i**
183		FactorizedJetCorrector iJetCorrector,const PileupEnergyDensityCol iPUEnergyDensity,
184		int iSign) {
185		FourVectorM lVec(0,0,0,0);
186	<	double lPt = fJetIDMVA->correctedPt(iJet,iJetCorrector,iPUEnergyDensity);
187	<	if(iJet->RawMom().Pt() < 10) lPt = TMath::Max(iJet->RawMom().Pt()-iJet->JetArea()*iPUEnergyDensity->At(0)->Rho(),0.);
188	<	lPt *= (iJet->NeutralEmEnergy()/iJet->E() + iJet->NeutralHadronEnergy()/iJet->E());
186	>	double lPt = fJetIDMVA->correctedPt(iJet,iJetCorrector,iPUEnergyDensity);
187	>	double lFrac = (iJet->NeutralEmEnergy()/iJet->RawMom().E() + iJet->NeutralHadronEnergy()/iJet->RawMom().E());
188	>	if(fabs(iJet->Eta()) > 2.5 && !f42) lFrac = 1.;
189	>	lPt *= lFrac;
190		lVec.SetPt(lPt); lVec.SetEta(iJet->Eta()); lVec.SetPhi(iJet->Phi()); lVec.SetM(iJet->Mass());
191		if(iSign > 0) iVec -= lVec;
192		if(iSign < 0) iVec += lVec;
193	<	iSumEt += lPt;
193	>	//iSumEt += lPt;
194		//=== Above was a bug in the training
195	<	//if(iSign > 0) iSumEt += lPt;
196	<	//if(iSign < 0) iSumEt -= lPt;
195	>	if(iSign > 0) iSumEt += lPt;
196	>	if(iSign < 0) iSumEt -= lPt;
197		}
198
199		//--------------------------------------------------------------------------------------------------
200		//Corrected Jets
201	<	void RecoilTools::addNeut(const PFJet *iJet,FourVectorM &iVec,Double_t &iSumEt,int iSign) {
201	>	void RecoilTools::addNeut(const PFJet *iJet,FourVectorM &iVec,Double_t &iSumEt,double iRho,int iSign) {
202		FourVectorM lVec(0,0,0,0);
203		double lPt = iJet->Pt();
204	<	if(iJet->RawMom().Pt() < 10) lPt = iJet->RawMom().Pt()*iJet->L1OffsetCorrectionScale();
204	>	//if(iJet->RawMom().Pt() < 10) lPt = TMath::Max(iJet->RawMom().Pt()-iJet->JetArea()*iRho,0.);//to be fixed
205	>	//if(iJet->RawMom().Pt() < 10) lPt = iJet->RawMom().Pt()*iJet->L1OffsetCorrectionScale();
206		lPt *= (iJet->NeutralEmEnergy()/iJet->RawMom().E() + iJet->NeutralHadronEnergy()/iJet->RawMom().E());
207		lVec.SetPt(lPt); lVec.SetEta(iJet->Eta()); lVec.SetPhi(iJet->Phi()); lVec.SetM(iJet->Mass());
208		if(iSign > 0) iVec -= lVec;
209		if(iSign < 0) iVec += lVec;
210	<	iSumEt += lPt;
210	>	//iSumEt += lPt;
211		//=== Above was a bug in the training
212	<	//if(iSign > 0) iSumEt += lPt;
213	<	//if(iSign < 0) iSumEt -= lPt;
212	>	if(iSign > 0) iSumEt += lPt;
213	>	if(iSign < 0) iSumEt -= lPt;
214		}
215
216		//--------------------------------------------------------------------------------------------------
#	Line 111 \| Line 222 \| Met RecoilTools::NoPUMet( const PFJetCol
222		FourVectorM lVec (0,0,0,0); double lSumEt = 0;
223		for(UInt_t i0 = 0; i0 < iCands->GetEntries(); i0++) {
224		const PFCandidate *pPF = iCands->At(i0);
225	<	const Track* pTrack = pPF->TrackerTrk();
226	<	if(pPF->GsfTrk()) pTrack = pPF->GsfTrk();
227	<	if(pTrack == 0 ) continue;
228	<	if( !((pPF->HasTrackerTrk() && (fabs(pPF->TrackerTrk()->DzCorrected(*iVertex))<iDZCut)) \|\|
229	<	(pPF->HasGsfTrk() && (fabs(pPF->GsfTrk()->DzCorrected(*iVertex)) <iDZCut)))) continue;
225	>	double lDZ = 999;
226	>	if(pPF->HasTrackerTrk()) lDZ = fabs(pPF->TrackerTrk()->DzCorrected(*iVertex));//
227	>	else if(pPF->HasGsfTrk()) lDZ = fabs(pPF->GsfTrk() ->DzCorrected(*iVertex));
228	>	if( fabs(lDZ) > iDZCut) continue;
229	>	lSumEt += pPF->Pt();
230		lVec -= pPF->Mom();
120	–	lSumEt += pPF->Pt();
231		}
232		int lNPass = 0;
233		for(UInt_t i0 = 0; i0 < iJets->GetEntries(); i0++) {
234		const PFJet *pJet = iJets->At(i0);
235	<	if(!filter(pJet,iPhi1,iEta1,iPhi2,iEta2)) continue; //Quick cleaning==> if not done already
235	>	if(!JetTools::passPFLooseId(pJet)) continue;
236	>	if(!filter(pJet,iPhi1,iEta1,iPhi2,iEta2)) continue;
237	>	if(!fJetIDMVA->passPt(pJet,iJetCorrector,iPileupEnergyDensity)) continue;
238		if(!fJetIDMVA->pass(pJet,iVertex,iVertices,iJetCorrector,iPileupEnergyDensity)) continue;
239	+	double pDPhi1 = fabs(pJet->Phi() - iPhi1); if(pDPhi1 > 2.TMath::Pi()-pDPhi1) pDPhi1 = 2.TMath::Pi()-pDPhi1;
240	+	double pDPhi2 = fabs(pJet->Phi() - iPhi2); if(pDPhi2 > 2.TMath::Pi()-pDPhi2) pDPhi2 = 2.TMath::Pi()-pDPhi2;
241		addNeut(pJet,lVec,lSumEt,iJetCorrector,iPileupEnergyDensity);
242		lNPass++;
243		}
#	Line 131 \| Line 245 \| Met RecoilTools::NoPUMet( const PFJetCol
245		lMet.SetSumEt( lSumEt);
246		return lMet;
247		}
248	<	//--------------------------------------------------------------------------------------------------
248	>	//-------------------------------------------------------------------------------------------------
249		//Corrected Jets
250		Met RecoilTools::NoPUMet( const PFJetCol *iJets,
251	<	const PFCandidateCol iCands,const Vertex iVertex,const VertexCol *iVertices,
251	>	const PFCandidateCol iCands,const Vertex iVertex,const VertexCol *iVertices,Double_t iRho,
252		Double_t iPhi1,Double_t iEta1,Double_t iPhi2,Double_t iEta2,Double_t iDZCut) {
253
254		FourVectorM lVec (0,0,0,0); double lSumEt = 0;
255		for(UInt_t i0 = 0; i0 < iCands->GetEntries(); i0++) {
256		const PFCandidate *pPF = iCands->At(i0);
257	<	const Track* pTrack = pPF->TrackerTrk();
258	<	if(pPF->GsfTrk()) pTrack = pPF->GsfTrk();
259	<	if(pTrack == 0 ) continue;
260	<	if( !((pPF->HasTrackerTrk() && (fabs(pPF->TrackerTrk()->DzCorrected(*iVertex))<iDZCut)) \|\|
147	<	(pPF->HasGsfTrk() && (fabs(pPF->GsfTrk()->DzCorrected(*iVertex)) <iDZCut)))) continue;
257	>	double lDZ = 999;
258	>	if(pPF->HasTrackerTrk()) lDZ = fabs(pPF->TrackerTrk()->DzCorrected(*iVertex));//
259	>	else if(pPF->HasGsfTrk()) lDZ = fabs(pPF->GsfTrk() ->DzCorrected(*iVertex));
260	>	if( fabs(lDZ) > iDZCut) continue;
261		lVec -= pPF->Mom();
262	<	lSumEt += pPF->Pt();
262	>	lSumEt += pPF->Pt();
263		}
264		for(UInt_t i0 = 0; i0 < iJets->GetEntries(); i0++) {
265		const PFJet *pJet = iJets->At(i0);
266	+	if(!filter(pJet,iPhi1,iEta1,iPhi2,iEta2)) continue;
267	+	if(!fJetIDMVA->passPt(pJet)) continue;
268		if(!fJetIDMVA->pass(pJet,iVertex,iVertices)) continue;
269	<	if(!filter(pJet,iPhi1,iEta1,iPhi2,iEta2)) continue; //Quick cleaning==> if not done already
155	<	addNeut(pJet,lVec,lSumEt);
269	>	addNeut(pJet,lVec,lSumEt,iRho);
270		}
271		Met lMet(lVec.Px(),lVec.Py());
272		lMet.SetSumEt(lSumEt);
273		return lMet;
274		}
275		//--------------------------------------------------------------------------------------------------
276	+	Met RecoilTools::NoPURecoil(Double_t iPhi1,Double_t iEta1,Double_t iPhi2,Double_t iEta2,
277	+	const PFJetCol *iJets,
278	+	const PFCandidateCol *iCands ,
279	+	const Vertex iVertex,const VertexCol iVertices,
280	+	FactorizedJetCorrector *iJetCorrector,
281	+	const PileupEnergyDensityCol *iPileupEnergyDensity,Double_t iDZCut) {
282	+
283	+	FourVectorM lVec (0,0,0,0); double lSumEt = 0;
284	+	for(UInt_t i0 = 0; i0 < iCands->GetEntries(); i0++) {
285	+	const PFCandidate *pPF = iCands->At(i0);
286	+	if(!filter(pPF,iPhi1,iEta1,iPhi2,iEta2)) continue;
287	+	double lDZ = 999;
288	+	if(pPF->HasTrackerTrk()) lDZ = fabs(pPF->TrackerTrk()->DzCorrected(*iVertex));//
289	+	else if(pPF->HasGsfTrk()) lDZ = fabs(pPF->GsfTrk() ->DzCorrected(*iVertex));
290	+	if( fabs(lDZ) > iDZCut) continue;
291	+	lSumEt += pPF->Pt();
292	+	lVec -= pPF->Mom();
293	+	}
294	+	int lNPass = 0;
295	+	for(UInt_t i0 = 0; i0 < iJets->GetEntries(); i0++) {
296	+	const PFJet *pJet = iJets->At(i0);
297	+	if(!filter(pJet,iPhi1,iEta1,iPhi2,iEta2)) continue;
298	+	if(!fJetIDMVA->passPt(pJet,iJetCorrector,iPileupEnergyDensity)) continue;
299	+	if(iJetCorrector != 0 && iPileupEnergyDensity != 0) if(!fJetIDMVA->pass(pJet,iVertex,iVertices,iJetCorrector,iPileupEnergyDensity)) continue;
300	+	if(iJetCorrector == 0 \|\| iPileupEnergyDensity == 0) if(!fJetIDMVA->pass(pJet,iVertex,iVertices)) continue;
301	+	if(iJetCorrector != 0 && iPileupEnergyDensity != 0) addNeut(pJet,lVec,lSumEt,iJetCorrector,iPileupEnergyDensity);
302	+	if(iJetCorrector == 0 \|\| iPileupEnergyDensity == 0) addNeut(pJet,lVec,lSumEt,1.);
303	+	lNPass++;
304	+	}
305	+	Met lMet(lVec.Px(),lVec.Py());
306	+	lMet.SetSumEt( lSumEt);
307	+	return lMet;
308	+	}
309	+
310	+	//--------------------------------------------------------------------------------------------------
311		Met RecoilTools::NoPURecoil(Double_t iVisPt,Double_t iVisPhi,Double_t iVisSumEt,
312		const PFJetCol iJets,FactorizedJetCorrector iJetCorrector,
313		const PileupEnergyDensityCol *iPileupEnergyDensity,
#	Line 177 \| Line 326 \| Met RecoilTools::NoPURecoil(Double_t iVi
326		//Corrected Jets
327		Met RecoilTools::NoPURecoil(Double_t iVisPt,Double_t iVisPhi,Double_t iVisSumEt,
328		const PFJetCol iJets,const PFCandidateCol iCands,
329	<	const Vertex iVertex,const VertexCol iVertices,
329	>	const Vertex iVertex,const VertexCol iVertices,Double_t iRho,
330		Double_t iPhi1,Double_t iEta1,Double_t iPhi2,Double_t iEta2,
331		Double_t iDZCut) {
332
333	<	Met lNoPUMet = NoPUMet(iJets,iCands,iVertex,iVertices,iPhi1,iEta1,iPhi2,iEta2,iDZCut);
333	>	Met lNoPUMet = NoPUMet(iJets,iCands,iVertex,iVertices,iRho,iPhi1,iEta1,iPhi2,iEta2,iDZCut);
334		lNoPUMet.SetMex (lNoPUMet.Mex()+iVisPt*cos(iVisPhi));
335		lNoPUMet.SetMey (lNoPUMet.Mey()+iVisPt*sin(iVisPhi));
336		lNoPUMet.SetSumEt(lNoPUMet.SumEt()-iVisSumEt);
#	Line 207 \| Line 356 \| Met RecoilTools::PUCMet( const PFJetCol
356		pPF->PFType() == PFCandidate::eHadronHF ))
357		{lVec -= pPF->Mom(); lSumEt += pPF->Pt();}
358		if(pTrack == 0 ) continue;
359	<	if( !((pPF->HasTrackerTrk() && (fabs(pPF->TrackerTrk()->DzCorrected(*iVertex))<iDZCut)) \|\|
360	<	(pPF->HasGsfTrk() && (fabs(pPF->GsfTrk()->DzCorrected(*iVertex)) <iDZCut)))) continue;
359	>	double lDZ = 999;
360	>	if(pPF->HasTrackerTrk()) lDZ = fabs(pPF->TrackerTrk()->DzCorrected(*iVertex));//
361	>	else if(pPF->HasGsfTrk()) lDZ = fabs(pPF->GsfTrk() ->DzCorrected(*iVertex));
362	>	if( fabs(lDZ) > iDZCut) continue;
363		lVec -= pPF->Mom();
364		lSumEt += pPF->Pt();
365		}
366		for(UInt_t i0 = 0; i0 < iJets->GetEntries(); i0++) {
367		const PFJet *pJet = iJets->At(i0);
368		if(!JetTools::passPFLooseId(pJet)) continue;
369	+	if(!fJetIDMVA->passPt(pJet,iJetCorrector,iPileupEnergyDensity)) continue;
370		if(fJetIDMVA->pass(pJet,iVertex,iVertices,iJetCorrector,iPileupEnergyDensity)) continue;
371	<	if(!filter(pJet,iPhi1,iEta1,iPhi2,iEta2)) continue; //Quick cleaning==> if not done already
371	>	if(!filter(pJet,iPhi1,iEta1,iPhi2,iEta2)) continue;
372		addNeut(pJet,lVec,lSumEt,iJetCorrector,iPileupEnergyDensity,-1);
373		}
374		Met lMet(lVec.Px(),lVec.Py());
#	Line 226 \| Line 378 \| Met RecoilTools::PUCMet( const PFJetCol
378		//--------------------------------------------------------------------------------------------------
379		//Corrected jets
380		Met RecoilTools::PUCMet( const PFJetCol iJets,const PFCandidateCol iCands,
381	<	const Vertex iVertex,const VertexCol iVertices,
381	>	const Vertex iVertex,const VertexCol iVertices,Double_t iRho,
382		Double_t iPhi1,Double_t iEta1,Double_t iPhi2,Double_t iEta2,
383		Double_t iDZCut) {
384
#	Line 242 \| Line 394 \| Met RecoilTools::PUCMet( const PFJetCol
394		pPF->PFType() == PFCandidate::eHadronHF ))
395		{lVec -= pPF->Mom(); lSumEt += pPF->Pt();}
396		if(pTrack == 0 ) continue;
397	<	if( !((pPF->HasTrackerTrk() && (fabs(pPF->TrackerTrk()->DzCorrected(*iVertex))<iDZCut)) \|\|
398	<	(pPF->HasGsfTrk() && (fabs(pPF->GsfTrk()->DzCorrected(*iVertex)) <iDZCut)))) continue;
397	>	double lDZ = 999;
398	>	if(pPF->HasTrackerTrk()) lDZ = fabs(pPF->TrackerTrk()->DzCorrected(*iVertex));//
399	>	else if(pPF->HasGsfTrk()) lDZ = fabs(pPF->GsfTrk() ->DzCorrected(*iVertex));
400	>	if( fabs(lDZ) > iDZCut) continue;
401		lVec -= pPF->Mom();
402		lSumEt += pPF->Pt();
403		}
404		for(UInt_t i0 = 0; i0 < iJets->GetEntries(); i0++) {
405		const PFJet *pJet = iJets->At(i0);
406		if(!JetTools::passPFLooseId(pJet)) continue;
407	+	if(!fJetIDMVA->passPt(pJet)) continue;
408		if(fJetIDMVA->pass(pJet,iVertex,iVertices)) continue;
409		if(!filter(pJet,iPhi1,iEta1,iPhi2,iEta2)) continue; //Quick cleaning==> if not done already
410	<	addNeut(pJet,lVec,lSumEt,-1);
410	>	addNeut(pJet,lVec,lSumEt,iRho,-1);
411	>	}
412	>	Met lMet(lVec.Px(),lVec.Py());
413	>	lMet.SetSumEt(lSumEt);
414	>	return lMet;
415	>	}
416	>	//--------------------------------------------------------------------------------------------------
417	>	Met RecoilTools::PUCRecoil( Double_t iPhi1,Double_t iEta1,Double_t iPhi2,Double_t iEta2,
418	>	const PFJetCol *iJets,
419	>	const PFCandidateCol *iCands ,
420	>	const Vertex iVertex,const VertexCol iVertices,
421	>	FactorizedJetCorrector *iJetCorrector,
422	>	const PileupEnergyDensityCol *iPileupEnergyDensity,Double_t iDZCut) {
423	>
424	>
425	>	FourVectorM lVec (0,0,0,0); double lSumEt = 0;
426	>	for(UInt_t i0 = 0; i0 < iCands->GetEntries(); i0++) {
427	>	const PFCandidate *pPF = iCands->At(i0);
428	>	if(!filter(pPF,iPhi1,iEta1,iPhi2,iEta2)) continue;
429	>	const Track* pTrack = pPF->TrackerTrk();
430	>	if(pPF->GsfTrk()) pTrack = pPF->GsfTrk();
431	>	if(pTrack == 0 &&
432	>	(pPF->PFType() == PFCandidate::eGamma \|\|
433	>	pPF->PFType() == PFCandidate::eEGammaHF \|\|
434	>	pPF->PFType() == PFCandidate::eNeutralHadron \|\|
435	>	pPF->PFType() == PFCandidate::eHadronHF ))
436	>	{lVec -= pPF->Mom(); lSumEt += pPF->Pt();}
437	>	if(pTrack == 0 ) continue;
438	>	double lDZ = 999;
439	>	if(pPF->HasTrackerTrk()) lDZ = fabs(pPF->TrackerTrk()->DzCorrected(*iVertex));//
440	>	else if(pPF->HasGsfTrk()) lDZ = fabs(pPF->GsfTrk() ->DzCorrected(*iVertex));
441	>	if( fabs(lDZ) > iDZCut) continue;
442	>	lVec -= pPF->Mom();
443	>	lSumEt += pPF->Pt();
444	>	}
445	>	for(UInt_t i0 = 0; i0 < iJets->GetEntries(); i0++) {
446	>	const PFJet *pJet = iJets->At(i0);
447	>	if(!filter(pJet,iPhi1,iEta1,iPhi2,iEta2)) continue; //Quick cleaning==> if not done already
448	>	if(!fJetIDMVA->passPt(pJet,iJetCorrector,iPileupEnergyDensity)) continue;
449	>	if(!JetTools::passPFLooseId(pJet)) continue;
450	>	if(iJetCorrector != 0 && iPileupEnergyDensity != 0) if(fJetIDMVA->pass(pJet,iVertex,iVertices,iJetCorrector,iPileupEnergyDensity)) continue;
451	>	if(iJetCorrector == 0 \|\| iPileupEnergyDensity == 0) if(fJetIDMVA->pass(pJet,iVertex,iVertices)) continue;
452	>	if(iJetCorrector != 0 && iPileupEnergyDensity != 0) addNeut(pJet,lVec,lSumEt,iJetCorrector,iPileupEnergyDensity,-1);
453	>	if(iJetCorrector == 0 \|\| iPileupEnergyDensity == 0) addNeut(pJet,lVec,lSumEt,1,-1);
454		}
455		Met lMet(lVec.Px(),lVec.Py());
456		lMet.SetSumEt(lSumEt);
#	Line 278 \| Line 476 \| Met RecoilTools::PUCRecoil(Double_t iVis
476		Met RecoilTools::PUCRecoil(Double_t iVisPt,Double_t iVisPhi,Double_t iVisSumEt,
477		const PFJetCol *iJets,
478		const PFCandidateCol iCands,const Vertex iVertex,
479	<	const VertexCol *iVertices,
479	>	const VertexCol *iVertices,Double_t iRho,
480		Double_t iPhi1,Double_t iEta1,Double_t iPhi2,Double_t iEta2,
481		Double_t iDZCut) {
482	<	Met lPUCMet = PUCMet(iJets,iCands,iVertex,iVertices,iPhi1,iEta1,iPhi2,iEta2,iDZCut);
482	>	Met lPUCMet = PUCMet(iJets,iCands,iVertex,iVertices,iRho,iPhi1,iEta1,iPhi2,iEta2,iDZCut);
483		lPUCMet.SetMex (lPUCMet.Mex()+iVisPt*cos(iVisPhi));
484		lPUCMet.SetMey (lPUCMet.Mey()+iVisPt*sin(iVisPhi));
485		lPUCMet.SetSumEt(lPUCMet.SumEt()-iVisSumEt);
#	Line 301 \| Line 499 \| Met RecoilTools::PUMet( const PFJetCol
499		const Track* pTrack = pPF->TrackerTrk();
500		if(pPF->GsfTrk()) pTrack = pPF->GsfTrk();
501		if(pTrack == 0 ) continue;
502	<	if( ((pPF->HasTrackerTrk() && (fabs(pPF->TrackerTrk()->DzCorrected(*iVertex))<iDZCut)) \|\|
503	<	(pPF->HasGsfTrk() && (fabs(pPF->GsfTrk()->DzCorrected(*iVertex)) <iDZCut)))) continue;
502	>	double lDZ = 999;
503	>	if(pPF->HasTrackerTrk()) lDZ = fabs(pPF->TrackerTrk()->DzCorrected(*iVertex));//
504	>	else if(pPF->HasGsfTrk()) lDZ = fabs(pPF->GsfTrk() ->DzCorrected(*iVertex));
505	>	if( fabs(lDZ) < iDZCut) continue;
506		lVec -= pPF->Mom();
507		lSumEt += pPF->Pt();
508		}
509		for(UInt_t i0 = 0; i0 < iJets->GetEntries(); i0++) {
510		const PFJet *pJet = iJets->At(i0);
511		if(!JetTools::passPFLooseId(pJet)) continue;
512	<	if(!filter(pJet,iPhi1,iEta1,iPhi2,iEta2)) continue; //Quick cleaning
512	>	if(!fJetIDMVA->passPt(pJet,iJetCorrector,iPileupEnergyDensity)) continue;
513	>	if(!filter(pJet,iPhi1,iEta1,iPhi2,iEta2)) continue;
514		if(fJetIDMVA->pass(pJet,iVertex,iVertices,iJetCorrector,iPileupEnergyDensity)) continue;
515		addNeut(pJet,lVec,lSumEt,iJetCorrector,iPileupEnergyDensity);
516		}
#	Line 320 \| Line 521 \| Met RecoilTools::PUMet( const PFJetCol
521		//--------------------------------------------------------------------------------------------------
522		//Corrected Jets
523		Met RecoilTools::PUMet( const PFJetCol *iJets,
524	<	const PFCandidateCol iCands,const Vertex iVertex,const VertexCol *iVertices,
524	>	const PFCandidateCol iCands,const Vertex iVertex,const VertexCol *iVertices,Double_t iRho,
525		Double_t iPhi1,Double_t iEta1,Double_t iPhi2,Double_t iEta2,
526		Double_t iDZCut) {
527
#	Line 330 \| Line 531 \| Met RecoilTools::PUMet( const PFJetCol
531		const Track* pTrack = pPF->TrackerTrk();
532		if(pPF->GsfTrk()) pTrack = pPF->GsfTrk();
533		if(pTrack == 0 ) continue;
534	<	if( ((pPF->HasTrackerTrk() && (fabs(pPF->TrackerTrk()->DzCorrected(*iVertex))<iDZCut)) \|\|
535	<	(pPF->HasGsfTrk() && (fabs(pPF->GsfTrk()->DzCorrected(*iVertex)) <iDZCut)))) continue;
534	>	double lDZ = 999;
535	>	if(pPF->HasTrackerTrk()) lDZ = fabs(pPF->TrackerTrk()->DzCorrected(*iVertex));//
536	>	else if(pPF->HasGsfTrk()) lDZ = fabs(pPF->GsfTrk() ->DzCorrected(*iVertex));
537	>	if( fabs(lDZ) < iDZCut) continue;
538		lVec -= pPF->Mom();
539		lSumEt += pPF->Pt();
540		}
541		for(UInt_t i0 = 0; i0 < iJets->GetEntries(); i0++) {
542		const PFJet *pJet = iJets->At(i0);
543		if(!JetTools::passPFLooseId(pJet)) continue;
544	+	if(!fJetIDMVA->passPt(pJet)) continue;
545		if(!filter(pJet,iPhi1,iEta1,iPhi2,iEta2)) continue; //Quick cleaning
546		if(fJetIDMVA->pass(pJet,iVertex,iVertices)) continue;
547	<	addNeut(pJet,lVec,lSumEt);
547	>	addNeut(pJet,lVec,lSumEt,iRho);
548	>	}
549	>	Met lMet(lVec.Px(),lVec.Py());
550	>	lMet.SetSumEt(lSumEt);
551	>	return lMet;
552	>	}
553	>	//--------------------------------------------------------------------------------------------------
554	>	Met RecoilTools::PUMet( Double_t iPhi1,Double_t iEta1,Double_t iPhi2,Double_t iEta2,
555	>	const PFJetCol *iJets,
556	>	const PFCandidateCol *iCands ,
557	>	const Vertex iVertex,const VertexCol iVertices,
558	>	FactorizedJetCorrector *iJetCorrector,
559	>	const PileupEnergyDensityCol *iPileupEnergyDensity,Double_t iDZCut) {
560	>
561	>	FourVectorM lVec (0,0,0,0); double lSumEt = 0;
562	>	for(UInt_t i0 = 0; i0 < iCands->GetEntries(); i0++) {
563	>	const PFCandidate *pPF = iCands->At(i0);
564	>	if(!filter(pPF,iPhi1,iEta1,iPhi2,iEta2)) continue;
565	>	const Track* pTrack = pPF->TrackerTrk();
566	>	if(pPF->GsfTrk()) pTrack = pPF->GsfTrk();
567	>	if(pTrack == 0 ) continue;
568	>	double lDZ = 999;
569	>	if(pPF->HasTrackerTrk()) lDZ = fabs(pPF->TrackerTrk()->DzCorrected(*iVertex));//
570	>	else if(pPF->HasGsfTrk()) lDZ = fabs(pPF->GsfTrk() ->DzCorrected(*iVertex));
571	>	if( fabs(lDZ) < iDZCut) continue;
572	>	lVec -= pPF->Mom();
573	>	lSumEt += pPF->Pt();
574	>	}
575	>	for(UInt_t i0 = 0; i0 < iJets->GetEntries(); i0++) {
576	>	const PFJet *pJet = iJets->At(i0);
577	>	if(!JetTools::passPFLooseId(pJet)) continue;
578	>	if(!fJetIDMVA->passPt(pJet,iJetCorrector,iPileupEnergyDensity)) continue;
579	>	if(!filter(pJet,iPhi1,iEta1,iPhi2,iEta2)) continue; //Quick cleaning
580	>	if(fJetIDMVA->pass(pJet,iVertex,iVertices,iJetCorrector,iPileupEnergyDensity)) continue;
581	>	addNeut(pJet,lVec,lSumEt,iJetCorrector,iPileupEnergyDensity);
582		}
583		Met lMet(lVec.Px(),lVec.Py());
584		lMet.SetSumEt(lSumEt);
#	Line 364 \| Line 602 \| Met RecoilTools::PURecoil(Double_t iVisP
602		//Corrected Jets
603		Met RecoilTools::PURecoil(Double_t iVisPt,Double_t iVisPhi,Double_t iVisSumEt,
604		const PFJetCol *iJets,
605	<	const PFCandidateCol iCands,const Vertex iVertex,const VertexCol *iVertices,
605	>	const PFCandidateCol iCands,const Vertex iVertex,const VertexCol *iVertices,Double_t iRho,
606		Double_t iPhi1,Double_t iEta1,Double_t iPhi2,Double_t iEta2,
607		Double_t iDZCut) {
608	<	Met lPUMet = PUMet(iJets,iCands,iVertex,iVertices,iPhi1,iEta1,iPhi2,iEta2,iDZCut);
608	>	Met lPUMet = PUMet(iJets,iCands,iVertex,iVertices,iRho,iPhi1,iEta1,iPhi2,iEta2,iDZCut);
609		lPUMet.SetMex (lPUMet.Mex()+iVisPt*cos(iVisPhi));
610		lPUMet.SetMey (lPUMet.Mey()+iVisPt*sin(iVisPhi));
611		lPUMet.SetSumEt(lPUMet.SumEt()-iVisSumEt);

Diff Legend

-–
+Removed lines
-+
+Added lines
-<
+Changed lines
->
+Changed lines

Comparing UserCode/MitPhysics/Utils/src/RecoilTools.cc (file contents): Revision 1.7 by pharris, Mon Apr 23 17:40:23 2012 UTC vs. Revision 1.24 by pharris, Sat Jan 12 11:49:50 2013 UTC

Diff Legend

Comparing UserCode/MitPhysics/Utils/src/RecoilTools.cc (file contents):
Revision 1.7 by pharris, Mon Apr 23 17:40:23 2012 UTC vs.
Revision 1.24 by pharris, Sat Jan 12 11:49:50 2013 UTC