ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitHzz4l/Selection/src/applySelection.cc
Revision: 1.50
Committed: Fri Jun 22 03:43:08 2012 UTC (12 years, 10 months ago) by khahn
Content type: text/plain
Branch: MAIN
Changes since 1.49: +1 -1 lines
Log Message:
*** empty log message ***

File Contents

# User Rev Content
1 khahn 1.1 //
2     // System headers
3     //
4     #include <vector> // STL vector class
5     #include <iostream> // standard I/O
6     #include <iomanip> // functions to format standard I/O
7 khahn 1.13 #include <bitset>
8 khahn 1.19 #include <map>
9 khahn 1.1 using namespace std;
10    
11     //
12 khahn 1.13 // root headers
13 khahn 1.1 //
14 khahn 1.13 #include <TFile.h>
15     #include <TTree.h>
16     #include <TChain.h>
17     #include <TBranch.h>
18     #include <TClonesArray.h>
19 khahn 1.1
20 khahn 1.11
21 khahn 1.1 //
22 khahn 1.13 // TMVA
23     //
24     #include "TMVA/Reader.h"
25     #include "TMVA/Tools.h"
26     #include "TMVA/Config.h"
27     #include "TMVA/MethodBDT.h"
28    
29     //
30 khahn 1.1 // ntuple format headers
31     //
32 khahn 1.13 #include "EventHeader.h"
33     #include "Electron.h"
34     #include "Muon.h"
35     #include "Vertex.h"
36 dkralph 1.48 #include "PFMet.h"
37 khahn 1.13 #include "PFCandidate.h"
38     #include "PFCandidateCol.h"
39 khahn 1.16 #include "PileupInfoCol.h"
40     #include "PileupEnergyDensity.h"
41 khahn 1.23 #include "MCParticle.h"
42 khahn 1.13 #include "TriggerMask.h"
43     #include "TriggerTable.h"
44 khahn 1.41 #include "TriggerObject.h"
45     #include "TriggerObjectRel.h"
46 khahn 1.13 #include "Names.h"
47     #include "BaseMod.h"
48 dkralph 1.44 #include "TriggerObjectsTable.h"
49 khahn 1.43
50 khahn 1.1 //
51 khahn 1.13 // our headers
52 khahn 1.1 //
53     #include "ParseArgs.h"
54 khahn 1.13 #include "MuonSelection.h"
55     #include "ElectronSelection.h"
56     #include "IsolationSelection.h"
57 khahn 1.29 #include "ReferenceSelection.h"
58 khahn 1.14
59 khahn 1.13 #include "TriggerUtils.h"
60     #include "PassHLT.h"
61     #include "Angles.h"
62     #include "KinematicsStruct.h"
63     #include "InfoStruct.h"
64 khahn 1.28 #include "GenInfoStruct.h"
65 khahn 1.13 #include "WeightStruct.h"
66     #include "AngleTuple.h"
67 khahn 1.15 #include "FOTuple.h"
68 khahn 1.13
69 khahn 1.33 #include "RunLumiRangeMap.h"
70 dkralph 1.2 #include "SampleWeight.h"
71 khahn 1.13 #include "EfficiencyWeightsInterface.h"
72 dkralph 1.38 #include "SelectionFuncs.h"
73 khahn 1.1
74 khahn 1.16 #include "SimpleLepton.h"
75    
76 khahn 1.13 #ifndef CMSSW_BASE
77     #define CMSSW_BASE "../"
78     #endif
79 khahn 1.1
80 khahn 1.13 using namespace mithep;
81 khahn 1.11
82 khahn 1.16 //
83     // globals
84     //----------------------------------------------------------------------------
85 khahn 1.47 TH1D *hpu_2011, *hpu_2012;
86 khahn 1.16 RunLumiRangeMap rlrm;
87     vector<SimpleLepton> failingLeptons ; // for fake estimation
88     vector<SimpleLepton> passingLeptons; // for fake estimation
89 khahn 1.19 vector<unsigned> cutvec;
90     vector<vector<unsigned> > zcutvec;
91     vector<vector<unsigned> > zzcutvec;
92     map<unsigned,float> evtrhoMap;
93     vector<string> cutstrs;
94 khahn 1.21 bool passes_HLT_MC;
95 khahn 1.22 vector<bool> PFnoPUflag;;
96 khahn 1.11
97 khahn 1.16 //
98     // prototypes
99 khahn 1.13 //----------------------------------------------------------------------------
100 dkralph 1.48 void initRunLumiRangeMap(ControlFlags &ctrl);
101 khahn 1.13 //----------------------------------------------------------------------------
102 khahn 1.11
103 khahn 1.16
104     //
105     // MAIN
106     //----------------------------------------------------------------------------
107 khahn 1.1 int main(int argc, char** argv)
108 khahn 1.16 //----------------------------------------------------------------------------
109 khahn 1.1 {
110 khahn 1.19 cutstrs.push_back(string("skim0")); //0
111     cutstrs.push_back(string("skim1")); //1
112 anlevin 1.25 cutstrs.push_back(string("skim2")); //1
113 khahn 1.21 cutstrs.push_back(string("trigger")); //2
114 khahn 1.19 // -------------------------------------------------
115 khahn 1.21 cutstrs.push_back(string("Z candidate")); //3
116     cutstrs.push_back(string("good Z1")); //4
117 khahn 1.19 // -------------------------------------------------
118 khahn 1.21 cutstrs.push_back(string("4l")); //5
119     cutstrs.push_back(string("ZZ candidate")); //6
120     cutstrs.push_back(string("good Z2")); //7
121     cutstrs.push_back(string("ZZ 20/10")); //8
122     cutstrs.push_back(string("resonance")); //9
123     cutstrs.push_back(string("m4l>70")); //10
124 anlevin 1.25 cutstrs.push_back(string("m4l>100, mZ2 > 12")); //11
125 khahn 1.19
126    
127    
128    
129 khahn 1.13 string cmsswpath(CMSSW_BASE);
130     cmsswpath.append("/src");
131     std::bitset<1024> triggerBits;
132 khahn 1.1 vector<vector<string> > inputFiles;
133     inputFiles.push_back(vector<string>());
134 khahn 1.16 map<string,unsigned> entrymap; // number of unskimmed entries in each file
135 khahn 1.21 for( int i=0; i<cutstrs.size(); i++ ) cutvec.push_back(0);
136 khahn 1.19 for( int i=0; i<2; i++ ) {
137     zcutvec.push_back(vector<unsigned>());
138 khahn 1.21 for( int j=0; j<cutstrs.size(); j++ ) zcutvec[i].push_back(0);
139 khahn 1.19 }
140     for( int i=0; i<3; i++ ) {
141     zzcutvec.push_back(vector<unsigned>());
142 khahn 1.21 for( int j=0; j<cutstrs.size(); j++ ) zzcutvec[i].push_back(0);
143 khahn 1.19 }
144    
145 khahn 1.1
146     //
147     // args
148     //--------------------------------------------------------------------------------------------------------------
149     ControlFlags ctrl;
150     parse_args( argc, argv, ctrl );
151 khahn 1.13 if( ctrl.inputfiles.empty() &&ctrl.inputfile.empty() )
152     {
153     cerr << "usage: applySelection.exe <flags> " << endl;
154     cerr << "\tmandoatory flags : " << endl;
155     cerr << "\t--inputfiles | file containing a list of ntuples to run over" << endl;
156     cerr << "\t--inputfile | a file to run over" << endl;
157     cerr << "\t--outputfile | file to store selected evet" << endl;
158     return 1;
159     }
160 khahn 1.1 ctrl.dump();
161 khahn 1.13
162 khahn 1.1
163 dkralph 1.4
164 khahn 1.1 //
165     // File I/O
166     //--------------------------------------------------------------------------------------------------------------
167     TChain * chain = new TChain("Events");
168 khahn 1.13 TChain * hltchain = new TChain("HLT");
169    
170 khahn 1.1 string fname;
171 dkralph 1.44 unsigned total_unskimmed=0;
172 khahn 1.13 if( !ctrl.inputfiles.empty() ) {
173     ifstream f(ctrl.inputfiles.c_str());
174     while (f >> fname) {
175     if( !(strncmp( fname.c_str(), "#", 1 ) ) ) continue; // skip commented lines
176     cout << "adding inputfile : " << fname.c_str() << endl;
177     entrymap[string(fname.c_str())] = unskimmedEntries(fname.c_str());
178 khahn 1.16 cout << "unskimmed entries: " << entrymap[string(fname.c_str())] << endl;
179 dkralph 1.44 total_unskimmed += entrymap[string(fname.c_str())];
180 khahn 1.13 chain->AddFile(fname.c_str());
181     hltchain->AddFile(fname.c_str());
182     }
183     } else {
184     cout << "adding inputfile : " << ctrl.inputfile.c_str() << endl;
185 dkralph 1.44 unsigned unsk_ents = unskimmedEntries(ctrl.inputfile.c_str());
186     entrymap[string(ctrl.inputfile.c_str())] = unsk_ents;
187     cout << "unskimmed entries: " << unsk_ents << endl;
188     total_unskimmed += unsk_ents;
189 khahn 1.13 chain->AddFile(ctrl.inputfile.c_str());
190     hltchain->AddFile(ctrl.inputfile.c_str());
191 khahn 1.1 }
192 dkralph 1.44 // // write the total number of unskimmed events that went into making this output file to a text file
193     // writeEntries(ctrl,total_unskimmed);
194 khahn 1.1
195 dkralph 1.10 const char * ofname;
196     if( strcmp( ctrl.outputfile.c_str(), "") ) {
197     ofname = ctrl.outputfile.c_str();
198     } else {
199     ofname = "tmp.root";
200     }
201 khahn 1.13 // table of cross section values
202     string xspath = (cmsswpath+string("/MitPhysics/data/xs.dat"));
203     cout << "xspath: " << xspath.c_str() << endl;
204     SimpleTable xstab(xspath.c_str());
205 khahn 1.11
206 khahn 1.1
207 dkralph 1.44
208 khahn 1.1 //
209 khahn 1.13 // Setup
210 khahn 1.1 //--------------------------------------------------------------------------------------------------------------
211 khahn 1.13 Angles angles;
212     KinematicsStruct kinematics;
213     InfoStruct evtinfo;
214     WeightStruct weights;
215 khahn 1.28 GenInfoStruct geninfo;
216 khahn 1.13
217     AngleTuple nt( (const char*)ofname, (const char*)"zznt");
218     nt.makeAngleBranch(angles);
219     nt.makeKinematicsBranch(kinematics);
220     nt.makeInfoBranch(evtinfo);
221 khahn 1.17
222 khahn 1.33
223 khahn 1.17 FOTuple foTree( nt.getFile(), (const char*)"FOtree");
224 khahn 1.16 foTree.makeFailedLeptonBranch(failingLeptons);
225 khahn 1.33 foTree.makePassedLeptonBranch(passingLeptons);
226 khahn 1.13
227 khahn 1.28 TH1F * h_wf11_hpt;
228 khahn 1.13 if(ctrl.mc) {
229     nt.makeWeightBranch(weights);
230 khahn 1.35 if(ctrl.fillGen ) nt.makeGenInfoBranch(geninfo);
231 khahn 1.49 initEfficiencyWeights();
232 khahn 1.16 initPUWeights();
233 khahn 1.28
234 khahn 1.13 // Higgs only, pt reweighting
235     if( ctrl.mH <= 140 ) {
236     char wptname[256];
237     sprintf( wptname, "/data/blue/pharris/Flat/ntupler/root/weight_ptH_%i.root", ctrl.mH );
238     TFile * f_hpt = new TFile(wptname);
239     f_hpt->Print();
240     sprintf(wptname, "weight_hqt_fehipro_fit_%i", ctrl.mH);
241 khahn 1.28 h_wf11_hpt = (TH1F*)(f_hpt->FindObjectAny(wptname));
242 khahn 1.13 }
243 khahn 1.28
244 khahn 1.13 } else {
245 khahn 1.50 initRunLumiRangeMap(ctrl);
246 khahn 1.13 }
247    
248 khahn 1.14 // initMuonIDMVA();
249 khahn 1.33 initElectronIDMVAV1();
250 khahn 1.13 initTrigger();
251 khahn 1.19
252 khahn 1.16 //
253 khahn 1.13 // Setup tree I/O
254 khahn 1.16 //--------------------------------------------------------------------------------------------------------------
255 khahn 1.1 TFile *inputFile=0;
256     TTree *eventTree=0;
257 khahn 1.13 string currentFile("");
258    
259 dkralph 1.44 UInt_t fNMaxTriggers = 1024;
260 khahn 1.13 mithep::EventHeader *info = new mithep::EventHeader();
261 dkralph 1.48 mithep::Array<mithep::PFMet> *metArr = new mithep::Array<mithep::PFMet>();
262 khahn 1.13 mithep::Array<mithep::Electron> *electronArr = new mithep::Array<mithep::Electron>();
263     mithep::Array<mithep::Muon> *muonArr = new mithep::Array<mithep::Muon>();
264 khahn 1.28 mithep::Array<mithep::Vertex> *vtxArr = new mithep::Array<mithep::Vertex>();
265 khahn 1.13 mithep::Array<mithep::PFCandidate> *pfArr = new mithep::Array<mithep::PFCandidate>();
266 khahn 1.16 mithep::Array<mithep::PileupInfo> *puArr = new mithep::Array<mithep::PileupInfo>();
267     mithep::Array<mithep::PileupEnergyDensity> *puDArr = new mithep::Array<mithep::PileupEnergyDensity>();
268 khahn 1.13 mithep::Array<mithep::Track> *trkArr = new mithep::Array<mithep::Track>();
269 khahn 1.23 mithep::Array<mithep::MCParticle> *mcArr = new mithep::Array<mithep::MCParticle>();
270 khahn 1.28 mithep::MCEventInfo *mcEvtInfo = new mithep::MCEventInfo();
271 khahn 1.13 mithep::TriggerMask *trigMask = new mithep::TriggerMask();
272 dkralph 1.44 mithep::TriggerTable *hltTable = new mithep::TriggerTable(fNMaxTriggers);
273 khahn 1.13 vector<string> *hltTableStrings = new vector<string>();
274 khahn 1.41 vector<string> *hltLabelStrings = new vector<string>();
275     mithep::Array<mithep::TriggerObject> *hltObjArr = new mithep::Array<mithep::TriggerObject>();
276     mithep::Array<mithep::TriggerObjectRel> *hltRelsArr = new mithep::Array<mithep::TriggerObjectRel>();
277     std::vector<std::string> *hltTab = new vector<string>();
278 dkralph 1.44 std::vector<std::string> *hltLab = new vector<string>();
279     TriggerObjectsTable *fTrigObjs = new TriggerObjectsTable(hltTable,fNMaxTriggers);
280 khahn 1.41
281 khahn 1.13
282     TString fElectronName(Names::gkElectronBrn);
283     TString fMuonName(Names::gkMuonBrn);
284     TString fInfoName(Names::gkEvtHeaderBrn);
285 dkralph 1.48 TString fPFMetName("PFMet");
286 khahn 1.13 TString fPrimVtxName(Names::gkPVBrn);
287     TString fPFCandidateName(Names::gkPFCandidatesBrn);
288 khahn 1.16 TString fPileupInfoName(Names::gkPileupInfoBrn);
289 khahn 1.13 TString fPileupEnergyDensityName(Names::gkPileupEnergyDensityBrn);
290     TString fTrackName(Names::gkTrackBrn);
291 khahn 1.23 TString fMCParticleName(Names::gkMCPartBrn);
292 khahn 1.28 TString fMCEvtInfoName(Names::gkMCEvtInfoBrn);
293 khahn 1.13 TString fTriggerMaskName(Names::gkHltBitBrn);
294     TString fTriggerTableName(Names::gkHltTableBrn);
295 khahn 1.41 TString fTriggerLabelName(Names::gkHltLabelBrn);
296     TString fTriggerObjectName(Names::gkHltObjBrn);
297     TString fTriggerObjectRelsName(Form("HLTObjectsRelation"));
298    
299 khahn 1.43 // TString fHLTLabName(Names::gkHltLabelBrn);
300     // TString fHLTTabName(Names::gkHltTableBrn);
301     // TString fHLTTabNamePub(Form("%sFwk",fHLTTabName.Data()));
302     // TString fHLTLabNamePub(Form("%sFwk",fHLTLabName.Data()));
303     // TString fObjsNamePub(Form("%sFwk",fTriggerObjectName.Data()));
304 khahn 1.41
305    
306 dkralph 1.44
307 khahn 1.13 chain->SetBranchAddress(fInfoName, &info);
308 dkralph 1.48 chain->SetBranchAddress(fPFMetName, &metArr);
309 khahn 1.13 chain->SetBranchAddress(fElectronName, &electronArr);
310     chain->SetBranchAddress(fMuonName, &muonArr);
311     chain->SetBranchAddress(fPrimVtxName, &vtxArr);
312     chain->SetBranchAddress(fPFCandidateName, &pfArr);
313 khahn 1.23 if( ctrl.mc ) {
314     chain->SetBranchAddress(fPileupInfoName, &puArr);
315     chain->SetBranchAddress(fMCParticleName, &mcArr);
316 khahn 1.28 chain->SetBranchAddress(fMCEvtInfoName, &mcEvtInfo);
317 khahn 1.23 }
318 khahn 1.16 chain->SetBranchAddress(fPileupEnergyDensityName, &puDArr);
319 khahn 1.13 chain->SetBranchAddress(fTrackName, &trkArr);
320     chain->SetBranchAddress(fTriggerMaskName, &trigMask);
321 khahn 1.41 chain->SetBranchAddress(fTriggerObjectName, &hltObjArr);
322     chain->SetBranchAddress(fTriggerObjectRelsName, &hltRelsArr);
323 khahn 1.43 // chain->SetBranchAddress(fHLTTabNamePub, &hltTab);
324     // chain->SetBranchAddress(fHLTLabNamePub, &hltLab);
325 khahn 1.41
326 khahn 1.13 cout << "hlttable: " << fTriggerTableName << endl;
327     hltchain->SetBranchAddress(fTriggerTableName, &hltTableStrings);
328 khahn 1.41 hltchain->SetBranchAddress(fTriggerLabelName, &hltLabelStrings);
329 khahn 1.1
330 khahn 1.13 mithep::Vertex vtx; // best primary vertex in the event
331 khahn 1.1
332 khahn 1.13 // ginfo = NULL;
333     // if( ctrl.mc ) { chain->SetBranchAddress("Gen", &ginfo);}
334 khahn 1.1
335 khahn 1.9 int count=0, pass=0;
336 khahn 1.13 float passcorr=0., passcorr_errup=0., passcorr_errdown=0.;
337 khahn 1.11 float denom[3]={0.,0.,0.};
338     float numer[3]={0.,0.,0.};
339     float numercorr[3]={0.,0.,0.};
340    
341 khahn 1.13 // only 1 HLT table / file ???
342     hltchain->GetEntry(0);
343    
344     int imax = chain->GetEntries();
345 khahn 1.11 cout << "nEntries: " << imax << endl;
346    
347 khahn 1.16 //
348     // Loop !!!!!!!!!
349     //--------------------------------------------------------------------------------------------------------------
350 khahn 1.13 for(UInt_t ientry=0; ientry<imax; ientry++)
351     {
352     chain->GetEntry(ientry);
353     if(!(ientry%1000)) cerr << "entry: " << ientry << endl;
354 dkralph 1.44
355 khahn 1.41 if( ctrl.debug ) {
356     cout << "-----------------------------------------------------------------" << endl;
357     cout << "-----------------------------------------------------------------" << endl;
358     cout << "Run: " << info->RunNum()
359     << "\tEvt: " << info->EvtNum()
360     << "\tLumi: " << info->LumiSec()
361     << endl;
362     cout << "-----------------------------------------------------------------" << endl;
363     }
364    
365 khahn 1.13
366 khahn 1.28 string fname = string(chain->GetFile()->GetEndpointUrl()->GetFile());
367 khahn 1.49 if(ctrl.debug) cout << "era is " << ctrl.era << endl;
368     if( ctrl.era == 0 ) {
369     setEra( fname, ctrl );
370     if(ctrl.debug) cout << "era is now " << ctrl.era << endl;
371     }
372 khahn 1.41
373 khahn 1.22 //
374     // pfNoPU
375     //
376     //mithep::Array<PFCandidate> pfNoPileUp;
377     PFnoPUflag.clear();
378     int pfnopu_size = makePFnoPUArray( pfArr, PFnoPUflag, vtxArr );
379     assert(pfnopu_size == pfArr->GetEntries());
380 khahn 1.13
381 khahn 1.16 //
382     // data/MC
383     //
384 khahn 1.13 if(ctrl.mc) {
385 khahn 1.16 //
386 khahn 1.28 // gen info
387     //
388 khahn 1.31 if( ctrl.fillGen )
389 khahn 1.28 fillGenInfo( mcArr, mcEvtInfo, geninfo, ESampleType::kHZZ, ctrl);
390    
391     //
392 khahn 1.16 // xsec & PU weights
393     //
394 khahn 1.31 weights.w = getWeight(xstab,entrymap,chain)/ctrl.totalMC;
395 khahn 1.49 cout << "xsec weight: " << weights.w << endl;
396    
397     unsigned npu = getNPU(puArr, 0);
398     weights.npu = npu;
399     weights.npuw = getPUWeight(ctrl.era,npu);
400     cerr << "weights.npuw: " << weights.npuw << endl;
401    
402 khahn 1.28 //
403     // pt reweighting for Higgs < 140, F11
404     //
405     if( ctrl.fillGen ) {
406     if( (fname.find("f11-h115zz4l") != string::npos) ||
407     (fname.find("f11-h120zz4l") != string::npos) ||
408     (fname.find("f11-h130zz4l") != string::npos) ) {
409     weights.w *= h_wf11_hpt->GetBinContent(h_wf11_hpt->FindBin(geninfo.pt_zz));
410     }
411     }
412    
413 khahn 1.20 //
414     // trigger
415     //
416     if( string(chain->GetFile()->GetEndpointUrl()->GetFile()) != currentFile ) {
417     currentFile = string(chain->GetFile()->GetEndpointUrl()->GetFile());
418     hltchain->SetBranchAddress(fTriggerTableName, &hltTableStrings);
419 khahn 1.41 hltchain->SetBranchAddress(fTriggerLabelName, &hltLabelStrings);
420 khahn 1.20 hltchain->GetEntry(0);
421     hltTable->Clear();
422 anlevin 1.32 fillTriggerNames(hltTable, hltTableStrings );
423 khahn 1.20 }
424     if( ctrl.debug ) cout << "file is : " << currentFile << endl;
425 khahn 1.34 fillTriggerBits( hltTable, trigMask, triggerBits );
426 khahn 1.21 passes_HLT_MC = true; // passed to apply as extern global, just for sync
427 anlevin 1.27 if( !passHLTMC(triggerBits, info->RunNum(), info->EvtNum(), k2012_MC ) ) {
428 khahn 1.21 passes_HLT_MC = false;
429 khahn 1.20 }
430 khahn 1.41
431 dkralph 1.44 setHLTObjectRelations( hltObjArr, hltRelsArr, hltTableStrings, hltLabelStrings, fTrigObjs );
432 khahn 1.41
433 khahn 1.13 } else {
434 khahn 1.16 //
435     // JSON
436     //
437 khahn 1.40 if(!(ctrl.noJSON) ) {
438 khahn 1.36 RunLumiRangeMap::RunLumiPairType rl(info->RunNum(), info->LumiSec());
439     if( !(rlrm.HasRunLumi(rl)) ) {
440 khahn 1.41 if( ctrl.debug ) cout << "fails JSON" << endl;
441 khahn 1.36 continue;
442 khahn 1.40 }
443 khahn 1.16 }
444 dkralph 1.44
445 khahn 1.16 //
446     // trigger
447     //
448 khahn 1.13 if( string(chain->GetFile()->GetEndpointUrl()->GetFile()) != currentFile ) {
449     currentFile = string(chain->GetFile()->GetEndpointUrl()->GetFile());
450     hltchain->SetBranchAddress(fTriggerTableName, &hltTableStrings);
451 khahn 1.41 hltchain->SetBranchAddress(fTriggerLabelName, &hltLabelStrings);
452 khahn 1.13 hltchain->GetEntry(0);
453     hltTable->Clear();
454 anlevin 1.32 fillTriggerNames(hltTable, hltTableStrings );
455 khahn 1.13 }
456 khahn 1.41
457 dkralph 1.44 setHLTObjectRelations( hltObjArr, hltRelsArr, hltTableStrings, hltLabelStrings, fTrigObjs );
458 khahn 1.41
459 khahn 1.13 if( ctrl.debug ) cout << "file is : " << currentFile << endl;
460 khahn 1.21 /*
461 khahn 1.34 fillTriggerBits( hltTable, trigMask, triggerBits );
462 khahn 1.13 if( !passHLT(triggerBits, info->RunNum(), info->EvtNum() ) ) {
463     if( ctrl.debug ) cout << "\tfails trigger ... " << endl;
464     continue;
465     }
466 khahn 1.21 */
467 khahn 1.13 }
468 khahn 1.16
469 khahn 1.13 //
470     // lepton/kinematic selection ...
471     //
472 khahn 1.16 failingLeptons.clear();
473     passingLeptons.clear();
474 khahn 1.13 EventData ret4l =
475 khahn 1.21
476 khahn 1.23
477     // reference
478 khahn 1.29 apply_HZZ4L_reference_selection(ctrl, info,
479 khahn 1.33 vtxArr,
480     pfArr,
481     puDArr,
482     electronArr,
483     &electronReferencePreSelection,
484     &electronReferenceIDMVASelectionV1,
485     &electronReferenceIsoSelection,
486     muonArr,
487     &muonReferencePreSelection,
488     &muonIDPFSelection,
489     &muonReferenceIsoSelection);
490    
491 khahn 1.13 if( ctrl.debug ) cout << endl;
492 khahn 1.14
493 khahn 1.19
494     int Z1type=0, Z2type=0;
495     if( ret4l.status.selectionBits.to_ulong() >= PASS_ZCANDIDATE ) {
496     if( abs(ret4l.Z1leptons[0].type) == 11 ) Z1type = 11;
497     if( abs(ret4l.Z1leptons[0].type) == 13 ) Z1type = 13;
498     }
499     if( ret4l.status.selectionBits.to_ulong() >= PASS_ZZCANDIDATE ) {
500     if( abs(ret4l.Z2leptons[0].type) == 11 ) Z2type = 11;
501     if( abs(ret4l.Z2leptons[0].type) == 13 ) Z2type = 13;
502     }
503    
504 khahn 1.22 #ifdef SYNC
505 dkralph 1.48 if(ctrl.debug)
506     cout << "bits: " << ret4l.status.selectionBits << "\t"
507     << "Z1: " << Z1type << "\t"
508     << "Z2: " << Z2type << endl << endl;
509 khahn 1.22 #endif
510 khahn 1.13
511     if( ret4l.status.pass() ) {
512     fillAngles(ret4l,angles);
513     fillKinematics(ret4l,kinematics);
514 khahn 1.31 fillMassErrors(ret4l,muonArr,electronArr,kinematics);
515 dkralph 1.48 TLorentzVector pfmet; pfmet.SetPxPyPzE(metArr->At(0)->Mex(),metArr->At(0)->Mey(),0,0);
516     fillEventInfo( info, pfmet, evtinfo, ctrl.mc ? getNPU(puArr) : 0);
517 khahn 1.16 if( ctrl.mc) {
518 khahn 1.49 // if( std::string(ctrl.efftype) != std::string("WTF?")) {
519 khahn 1.46 setEffiencyWeights(ctrl.era, ret4l, weights);
520 khahn 1.49 // } else {
521     // weights.won = weights.woff = 1.;
522     // }
523 dkralph 1.44 if(ctrl.debug)
524     cout << "w: " << weights.w << "\t"
525 khahn 1.49 << "npuw: " << weights.npuw << "\t"
526 dkralph 1.44 << "won: " << weights.won << "\t"
527     << "woff: " << weights.woff << endl;
528 khahn 1.16 }
529 khahn 1.13
530 khahn 1.28
531 khahn 1.13 nt.Fill();
532    
533     cerr << "PASS:: "
534     << "\trun: " << evtinfo.run
535     << "\tevt: " << evtinfo.evt
536     << "\tlumi: " << evtinfo.lumi
537 khahn 1.45 << "\tchannel: " << kinematics.channel
538     << "\tm4l: " << kinematics.m4l
539     << "\tmZ1: " << kinematics.mZ1
540     << "\tmZ2: " << kinematics.mZ2
541 khahn 1.13 << endl;
542     pass++;
543     // if( pass > 3 ) break;
544 khahn 1.8
545 khahn 1.33 } else if( ret4l.status.selectionBits.test(PASS_GOODZ1) &&
546 khahn 1.36 ret4l.status.selectionBits.to_ulong() < (1UL<<PASS_4L)) { // save for fakes ...
547     if(ctrl.debug) {
548 khahn 1.18 cout << "failingLeptons : " << failingLeptons.size() << endl;
549     for( int f=0; f<failingLeptons.size(); f++ ) {
550     SimpleLepton l = failingLeptons[f];
551     cout << "f: " << f << "\t"
552     << "type: " << l.type << "\t"
553     << "pT: " << l.vec.Pt() << "\t"
554     << "eta: " << l.vec.Eta()
555     << endl;
556 khahn 1.36 }
557 khahn 1.16 }
558     foTree.Fill();
559 khahn 1.36 } else {
560 dkralph 1.48 if(ctrl.debug)
561     cout << "failing with some other code : " << ret4l.status.selectionBits << endl;
562 khahn 1.15 }
563 khahn 1.13 }
564 khahn 1.1
565 khahn 1.16
566     foTree.getFile()->cd();
567     foTree.getTree()->Write();
568 khahn 1.13 nt.WriteClose();
569 khahn 1.19
570 dkralph 1.48 if(ctrl.debug)
571     cout << endl << endl;
572 khahn 1.19 for( int i=0; i<cutvec.size(); i++ ) {
573     cout << "cut: " << i << "\t"
574     << "pass: " << cutvec[i];
575    
576 khahn 1.21 if( i>PASS_TRIGGER&&i<=PASS_GOODZ1 )
577 khahn 1.19 cout << "\t2e: " << zcutvec[0][i]
578     << "\t2m: " << zcutvec[1][i];
579    
580 khahn 1.21 if( i>PASS_ZCANDIDATE )
581 khahn 1.19 cout << "\t4e: " << zzcutvec[0][i]
582     << "\t4m: " << zzcutvec[1][i]
583     << "\t2e2m: " << zzcutvec[2][i];
584    
585     cout << "\t" << cutstrs[i] << endl;
586    
587     cout << endl;
588     }
589    
590 khahn 1.13 }
591 dkralph 1.10
592 khahn 1.16
593     //----------------------------------------------------------------------------
594 dkralph 1.48 void initRunLumiRangeMap(ControlFlags &ctrl)
595 khahn 1.16 //----------------------------------------------------------------------------
596     {
597     /*
598     rlrm.AddJSONFile(std::string("./data/Cert_136033-149442_7TeV_Apr21ReReco_Collisions10_JSON.txt"));
599     // rlrm.AddJSONFile(std::string("./data/Cert_160404-173244_7TeV_PromptReco_Collisions11_JSON_v2.txt"));
600     rlrm.AddJSONFile(std::string("./data/Cert_160404-178078_7TeV_PromptReco_Collisions11_JSON.txt"));
601     rlrm.AddJSONFile(std::string("./data/Cert_160404-163869_7TeV_May10ReReco_Collisions11_JSON_v3.txt"));
602     rlrm.AddJSONFile(std::string("./data/Cert_170249-172619_7TeV_ReReco5Aug_Collisions11_JSON.txt"));
603     // r11b
604     rlrm.AddJSONFile(std::string("./data/Cert_160404-180252_7TeV_PromptReco_Collisions11_JSON.txt"));
605     */
606 khahn 1.36
607     // 2012 only for now ...
608 dkralph 1.48 std::string jsonFile;
609     if(ctrl.jsonFile=="")
610     jsonFile = "./data/Cert_190456-194479_8TeV_PromptReco_Collisions12_JSON.txt";
611     else
612     jsonFile = ctrl.jsonFile;
613     rlrm.AddJSONFile(jsonFile);
614 khahn 1.16 };