ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitHzz4l/Selection/src/applySelection.cc
Revision: 1.73
Committed: Tue Feb 5 09:53:47 2013 UTC (12 years, 3 months ago) by anlevin
Content type: text/plain
Branch: MAIN
Changes since 1.72: +1 -1 lines
Log Message:
added Moriond 2013 electron scale/smearing corrrection tag

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 dkralph 1.60 #include <TFile.h>
15     #include <TRegexp.h>
16 khahn 1.13 #include <TTree.h>
17     #include <TChain.h>
18     #include <TBranch.h>
19     #include <TClonesArray.h>
20 khahn 1.1
21 khahn 1.11
22 khahn 1.1 //
23 khahn 1.13 // TMVA
24     //
25     #include "TMVA/Reader.h"
26     #include "TMVA/Tools.h"
27     #include "TMVA/Config.h"
28     #include "TMVA/MethodBDT.h"
29    
30 dkralph 1.60 #include "FactorizedJetCorrector.h"
31     #include "JetCorrectorParameters.h"
32    
33 khahn 1.13 //
34 khahn 1.1 // ntuple format headers
35     //
36 khahn 1.13 #include "EventHeader.h"
37     #include "Electron.h"
38     #include "Muon.h"
39     #include "Vertex.h"
40 dkralph 1.48 #include "PFMet.h"
41 dkralph 1.60 #include "PFJetCol.h"
42 khahn 1.13 #include "PFCandidate.h"
43     #include "PFCandidateCol.h"
44 khahn 1.16 #include "PileupInfoCol.h"
45     #include "PileupEnergyDensity.h"
46 khahn 1.23 #include "MCParticle.h"
47 khahn 1.13 #include "TriggerMask.h"
48     #include "TriggerTable.h"
49 khahn 1.41 #include "TriggerObject.h"
50     #include "TriggerObjectRel.h"
51 khahn 1.13 #include "Names.h"
52     #include "BaseMod.h"
53 dkralph 1.44 #include "TriggerObjectsTable.h"
54 dkralph 1.60 #include "JetIDMVA.h"
55 khahn 1.43
56 khahn 1.1 //
57 khahn 1.13 // our headers
58 khahn 1.1 //
59     #include "ParseArgs.h"
60 khahn 1.13 #include "MuonSelection.h"
61     #include "ElectronSelection.h"
62 anlevin 1.56 #include "ElectronMomentumCorrection.h"
63 dkralph 1.60 #include "JetSelection.h"
64 khahn 1.13 #include "IsolationSelection.h"
65 khahn 1.29 #include "ReferenceSelection.h"
66 dkralph 1.60 #include "MuonMomentumCorrection.h"
67 khahn 1.14
68 khahn 1.13 #include "TriggerUtils.h"
69     #include "PassHLT.h"
70     #include "KinematicsStruct.h"
71     #include "InfoStruct.h"
72 khahn 1.28 #include "GenInfoStruct.h"
73 khahn 1.13 #include "WeightStruct.h"
74     #include "AngleTuple.h"
75 khahn 1.15 #include "FOTuple.h"
76 khahn 1.13
77 khahn 1.33 #include "RunLumiRangeMap.h"
78 dkralph 1.2 #include "SampleWeight.h"
79 khahn 1.13 #include "EfficiencyWeightsInterface.h"
80 dkralph 1.38 #include "SelectionFuncs.h"
81 khahn 1.1
82 khahn 1.16 #include "SimpleLepton.h"
83    
84 anlevin 1.72 #include "Cintex/Cintex.h"
85    
86 khahn 1.13 #ifndef CMSSW_BASE
87     #define CMSSW_BASE "../"
88     #endif
89 khahn 1.1
90 khahn 1.13 using namespace mithep;
91 khahn 1.11
92 khahn 1.16 //
93     // globals
94     //----------------------------------------------------------------------------
95 khahn 1.47 TH1D *hpu_2011, *hpu_2012;
96 khahn 1.16 RunLumiRangeMap rlrm;
97     vector<SimpleLepton> failingLeptons ; // for fake estimation
98     vector<SimpleLepton> passingLeptons; // for fake estimation
99 khahn 1.19 vector<unsigned> cutvec;
100     vector<vector<unsigned> > zcutvec;
101     vector<vector<unsigned> > zzcutvec;
102     map<unsigned,float> evtrhoMap;
103 dkralph 1.62 TrigInfo ti;
104 khahn 1.19 vector<string> cutstrs;
105 dkralph 1.60 bool passes_HLT;
106 khahn 1.22 vector<bool> PFnoPUflag;;
107 anlevin 1.57 map<TString, map<TString,int>* > counts;
108 dkralph 1.60 ElectronMomentumCorrection electron_momentum_correction;
109     MuCorr *muCorr;
110 dkralph 1.62 void addDummyZ2Lepotons(EventData &ret4l);
111     void setNtupleVals(ControlFlags &ctrl, EventData &ret4l, InfoStruct &evtinfo, Angles &angles, KinematicsStruct &kine,
112 anlevin 1.70 JetInfoStruct &ji, WeightStruct &weights, EventHeader *info, Array<Muon> *muonArr, Array<Electron> *electronArr, Array<PFJet> *jetArr, Array<PFMet> *metArr,
113 dkralph 1.62 vector<SimpleLepton> &simpleLeptonJets, FactorizedJetCorrector *fJetCorrector, Array<PileupInfo> *puArr,
114     Array<PileupEnergyDensity> *puDArr, JetIDMVA *fJetIDMVA, Array<Vertex> *vtxArr, bitset<1024> &triggerBits,
115     TriggerTable *hltTable, Array<TriggerObject> *hltObjArr, TriggerObjectsTable *fTrigObjs, AngleTuple *nt);
116 dkralph 1.60 //----------------------------------------------------------------------------
117     void initRunLumiRangeMap(ControlFlags &ctrl)
118     //----------------------------------------------------------------------------
119     {
120     if(ctrl.jsonFile==""){
121     cout << "WARNING: running with empty jsonFile, so setting noJSON to true" << endl;
122     ctrl.noJSON = true;
123     }
124     rlrm.AddJSONFile(string(ctrl.jsonFile));
125     }
126 khahn 1.11
127 khahn 1.16 //
128     // prototypes
129 khahn 1.13 //----------------------------------------------------------------------------
130 khahn 1.11
131 khahn 1.16 //
132     // MAIN
133     //----------------------------------------------------------------------------
134 khahn 1.1 int main(int argc, char** argv)
135 khahn 1.16 //----------------------------------------------------------------------------
136 khahn 1.1 {
137 anlevin 1.72
138 anlevin 1.57 vector<TString> cuts;
139 dkralph 1.60 cuts.push_back("start"); // NOTE: is incremented in ReferenceSelection, so if you turn on JSONS be careful
140 anlevin 1.57 cuts.push_back("trigger");
141     cuts.push_back("sfOsHiPt");
142     cuts.push_back("4<mZ2<120");
143     cuts.push_back("pt>20,10");
144     cuts.push_back("mll>4");
145     cuts.push_back("m4l>70");
146 dkralph 1.60 cuts.push_back("mZ2>12");
147     cuts.push_back("m4l>100");
148 anlevin 1.68 cuts.push_back("m4l > 100 && >= 1 jets");
149     cuts.push_back("m4l > 100 && 2 jets");
150     cuts.push_back("m4l > 100 && fisher > 0.4");
151 anlevin 1.57
152     for(unsigned icut=0; icut<cuts.size(); icut++)
153     counts[cuts[icut]] = new map<TString,int>;
154    
155     map<TString,map<TString,int>* >::iterator cit;
156     for(cit=counts.begin(); cit!=counts.end(); cit++) {
157     map<TString,int> *cts = (*cit).second;
158     (*cts)["all"] = int(0);
159     (*cts)["4e"] = int(0);
160     (*cts)["4m"] = int(0);
161     (*cts)["2e2m"] = int(0);
162     }
163 dkralph 1.60
164 khahn 1.13 string cmsswpath(CMSSW_BASE);
165     cmsswpath.append("/src");
166 dkralph 1.60 bitset<1024> triggerBits;
167 khahn 1.1 vector<vector<string> > inputFiles;
168     inputFiles.push_back(vector<string>());
169 khahn 1.16 map<string,unsigned> entrymap; // number of unskimmed entries in each file
170 khahn 1.1
171     //
172     // args
173     //--------------------------------------------------------------------------------------------------------------
174     ControlFlags ctrl;
175     parse_args( argc, argv, ctrl );
176     ctrl.dump();
177 dkralph 1.60 cout << "\n\nNOTE: now setting era with arguments... double check this doesnt mess something else up\n\n" << endl;
178     assert(ctrl.doFSR);
179 dkralph 1.4
180 khahn 1.1 //
181     // File I/O
182     //--------------------------------------------------------------------------------------------------------------
183     TChain * chain = new TChain("Events");
184 khahn 1.13 TChain * hltchain = new TChain("HLT");
185 dkralph 1.53 TChain * runchain = new TChain(Names::gkRunTreeName);
186 khahn 1.13
187 khahn 1.1 string fname;
188 dkralph 1.44 unsigned total_unskimmed=0;
189 khahn 1.13 if( !ctrl.inputfiles.empty() ) {
190     ifstream f(ctrl.inputfiles.c_str());
191 dkralph 1.60 while (f >> fname) {
192 khahn 1.13 if( !(strncmp( fname.c_str(), "#", 1 ) ) ) continue; // skip commented lines
193 dkralph 1.60 if(fname.find("/store/") != string::npos) fname = "root://eoscms//"+fname;
194 khahn 1.13 cout << "adding inputfile : " << fname.c_str() << endl;
195     entrymap[string(fname.c_str())] = unskimmedEntries(fname.c_str());
196 dkralph 1.62 cerr << "unskimmed entries: " << entrymap[string(fname.c_str())] << endl;
197 dkralph 1.44 total_unskimmed += entrymap[string(fname.c_str())];
198 khahn 1.13 chain->AddFile(fname.c_str());
199     hltchain->AddFile(fname.c_str());
200 dkralph 1.53 runchain->AddFile(fname.c_str());
201 khahn 1.13 }
202     } else {
203 dkralph 1.60 if(ctrl.inputfile.find("/store/") != string::npos) ctrl.inputfile = "root://eoscms//"+ctrl.inputfile;
204 khahn 1.13 cout << "adding inputfile : " << ctrl.inputfile.c_str() << endl;
205 dkralph 1.44 unsigned unsk_ents = unskimmedEntries(ctrl.inputfile.c_str());
206     entrymap[string(ctrl.inputfile.c_str())] = unsk_ents;
207 dkralph 1.62 cerr << "unskimmed entries: " << unsk_ents << endl;
208 dkralph 1.44 total_unskimmed += unsk_ents;
209 khahn 1.13 chain->AddFile(ctrl.inputfile.c_str());
210     hltchain->AddFile(ctrl.inputfile.c_str());
211 dkralph 1.53 runchain->AddFile(ctrl.inputfile.c_str());
212 khahn 1.1 }
213 dkralph 1.60 // write the total number of unskimmed events that went into making this output file to a text file
214 dkralph 1.53 writeEntries(ctrl,total_unskimmed);
215 dkralph 1.60 ctrl.totalMC = total_unskimmed;
216 khahn 1.1
217 dkralph 1.10 const char * ofname;
218     if( strcmp( ctrl.outputfile.c_str(), "") ) {
219     ofname = ctrl.outputfile.c_str();
220     } else {
221     ofname = "tmp.root";
222     }
223 khahn 1.13 // table of cross section values
224     string xspath = (cmsswpath+string("/MitPhysics/data/xs.dat"));
225     cout << "xspath: " << xspath.c_str() << endl;
226     SimpleTable xstab(xspath.c_str());
227 dkralph 1.62 if(ctrl.mc) {
228 anlevin 1.66 if(ctrl.use_xs_weights) getWeight(xstab,entrymap,chain); // test *now* whether we've got the xs in here, so it fails before it creates the output
229 dkralph 1.62 }
230 khahn 1.11
231 anlevin 1.72 //root will write to this temporary file instead of to memory and the selection will run faster
232    
233 khahn 1.1 //
234 khahn 1.13 // Setup
235 khahn 1.1 //--------------------------------------------------------------------------------------------------------------
236 khahn 1.13 Angles angles;
237 anlevin 1.57 KinematicsStruct kine;
238 khahn 1.13 InfoStruct evtinfo;
239     WeightStruct weights;
240 khahn 1.28 GenInfoStruct geninfo;
241 dkralph 1.60 JetInfoStruct ji;
242     vector<SimpleLepton> simpleLeptonJets;
243 khahn 1.13
244     AngleTuple nt( (const char*)ofname, (const char*)"zznt");
245     nt.makeAngleBranch(angles);
246 anlevin 1.57 nt.makeKinematicsBranch(kine);
247 khahn 1.13 nt.makeInfoBranch(evtinfo);
248 dkralph 1.67 nt.makeJetInfoBranch(ji);
249 anlevin 1.70 //nt.makeJetAngleBranches("DataStruct/data/jetAngleVars.txt");
250 dkralph 1.62
251     FOTuple *foTree=0;
252     if(ctrl.fakeScheme != "none") {
253     foTree = new FOTuple( nt.getFile(), (const char*)"FOtree");
254     foTree->makeInfoBranch(evtinfo);
255     foTree->makeFailedLeptonBranch(failingLeptons);
256     foTree->makePassedLeptonBranch(passingLeptons);
257     }
258 khahn 1.17
259 dkralph 1.60 int genSampleType(0);
260 khahn 1.13 if(ctrl.mc) {
261     nt.makeWeightBranch(weights);
262 dkralph 1.60 if(ctrl.fillGen) {
263     genSampleType = getGenSampleType(ofname);
264     if(genSampleType == -1) {
265     cout << "\ngenSampleType returned as -1, setting fillGen to false" << endl;
266     ctrl.fillGen = false;
267     } else
268     nt.makeGenInfoBranch(geninfo);
269 khahn 1.13 }
270 anlevin 1.66 if(ctrl.use_eff_weights) initEfficiencyWeights();
271 khahn 1.13 } else {
272 dkralph 1.60 if(!(ctrl.noJSON) )
273     initRunLumiRangeMap(ctrl);
274 khahn 1.13 }
275    
276 khahn 1.33 initElectronIDMVAV1();
277 khahn 1.13 initTrigger();
278 dkralph 1.53 initAnalysisTriggers(ti);
279 anlevin 1.70 initMela(ctrl);
280     initMassErrors();
281 dkralph 1.62 ti.dump();
282    
283 anlevin 1.72 TFile tmp_file("tmp_file.root");
284    
285     if(ctrl.do_eregression) {
286 dkralph 1.62 string regPath("EGamma/EGammaAnalysisTools/data/eleEnergyRegWeights_V1.root");
287     // if(TString(getenv("HOSTNAME")).Contains(TRegexp("t3[bd][te][cs][hk]")))
288 anlevin 1.72 regPath = cmsswpath + "/" + regPath;
289 dkralph 1.62 electron_momentum_correction.initialize_regression(regPath);
290 anlevin 1.72 }
291    
292     if(ctrl.do_escale) {
293 dkralph 1.60 if(ctrl.mc) {
294 anlevin 1.61 if(ctrl.era == 2011) ctrl.smearing_scaling_dataset = "Fall11";
295 dkralph 1.60 else ctrl.smearing_scaling_dataset = "Summer12_DR53X_HCP2012";
296     } else {
297 anlevin 1.61 if(ctrl.era == 2011) ctrl.smearing_scaling_dataset = "Jan16ReReco";
298 anlevin 1.73 else ctrl.smearing_scaling_dataset = "Moriond2013";
299 dkralph 1.60 }
300     }
301     if(ctrl.correct_muon_momentum) {
302     muCorr = new MuCorr;
303     muCorr->corr2011 = new rochcor;
304     muCorr->corr2012 = new rochcor2012;
305     }
306    
307     vector<TString> fJetCorrParsv;
308     vector<JetCorrectorParameters> correctionParameters;
309     FactorizedJetCorrector *fJetCorrector;
310     JetIDMVA *fJetIDMVA;
311 dkralph 1.67 initJets(ctrl, cmsswpath, fJetCorrParsv, correctionParameters, fJetCorrector, fJetIDMVA);
312 khahn 1.19
313 khahn 1.16 //
314 khahn 1.13 // Setup tree I/O
315 khahn 1.16 //--------------------------------------------------------------------------------------------------------------
316 khahn 1.1 TFile *inputFile=0;
317     TTree *eventTree=0;
318 khahn 1.13 string currentFile("");
319    
320 dkralph 1.60 UInt_t fNMaxTriggers = TRIGGER_BIG_NUMBER;
321     EventHeader *info = new EventHeader();
322     Array<PFMet> *metArr = new Array<PFMet>();
323     Array<Electron> *electronArr = new Array<Electron>();
324     Array<Muon> *muonArr = new Array<Muon>();
325     Array<Vertex> *vtxArr = new Array<Vertex>();
326     Array<PFCandidate> *pfArr = new Array<PFCandidate>();
327     Array<PFJet> *jetArr = new Array<PFJet>();
328     Array<PileupInfo> *puArr = new Array<PileupInfo>();
329     Array<PileupEnergyDensity> *puDArr = new Array<PileupEnergyDensity>();
330     Array<Track> *trkArr = new Array<Track>();
331     Array<MCParticle> *mcArr = new Array<MCParticle>();
332     MCEventInfo *mcEvtInfo = new MCEventInfo();
333     TriggerMask *trigMask = new TriggerMask();
334     TriggerTable *hltTable = new TriggerTable(fNMaxTriggers);
335     vector<string> *hltTableStrings = new vector<string>();
336     vector<string> *hltLabelStrings = new vector<string>();
337     Array<TriggerObject> *hltObjArr = new Array<TriggerObject>();
338     Array<TriggerObjectRel> *hltRelsArr = new Array<TriggerObjectRel>();
339     TriggerObjectsTable *fTrigObjs = new TriggerObjectsTable(hltTable,fNMaxTriggers);
340     RunInfo *runInfo = new RunInfo();
341 dkralph 1.44
342 dkralph 1.60 chain->SetBranchAddress(Names::gkEvtHeaderBrn, &info);
343     chain->SetBranchAddress("PFMet", &metArr);
344     chain->SetBranchAddress(Names::gkElectronBrn, &electronArr);
345     chain->SetBranchAddress(Names::gkMuonBrn, &muonArr);
346     chain->SetBranchAddress(Names::gkPVBrn, &vtxArr);
347     chain->SetBranchAddress(Names::gkPFCandidatesBrn, &pfArr);
348     chain->SetBranchAddress(Names::gkPFJetBrn, &jetArr);
349 khahn 1.23 if( ctrl.mc ) {
350 dkralph 1.60 chain->SetBranchAddress(Names::gkPileupInfoBrn, &puArr);
351     chain->SetBranchAddress(Names::gkMCPartBrn, &mcArr);
352     chain->SetBranchAddress(Names::gkMCEvtInfoBrn, &mcEvtInfo);
353 khahn 1.23 }
354 dkralph 1.60 chain->SetBranchAddress(Names::gkPileupEnergyDensityBrn, &puDArr);
355     chain->SetBranchAddress(Names::gkTrackBrn, &trkArr);
356     chain->SetBranchAddress(Names::gkHltBitBrn, &trigMask);
357     chain->SetBranchAddress(Names::gkHltObjBrn, &hltObjArr);
358     chain->SetBranchAddress("HLTObjectsRelation", &hltRelsArr);
359     hltchain->SetBranchAddress(Names::gkHltTableBrn, &hltTableStrings);
360     hltchain->SetBranchAddress(Names::gkHltLabelBrn, &hltLabelStrings);
361 khahn 1.1
362 dkralph 1.53 runchain->SetBranchAddress(Names::gkRunInfoBrn, &runInfo);
363    
364 dkralph 1.60 Vertex vtx; // best primary vertex in the event
365 khahn 1.11
366 khahn 1.13 hltchain->GetEntry(0);
367    
368 dkralph 1.60 int imax = (ctrl.n_events_to_process < 0) ? chain->GetEntries() : ctrl.n_events_to_process;
369     cerr << "processing: " << imax << " ( / " << chain->GetEntries() << " )" << endl;
370 anlevin 1.54
371 dkralph 1.60 if(ctrl.mc && ctrl.debugOffMc) // necessary for batch mode -- we need debug output for, data but size is too large to store MC output
372     ctrl.debug = false;
373 khahn 1.11
374 dkralph 1.53 int lastHltEntry=-2;
375 dkralph 1.62 for(UInt_t ientry=0; ientry<imax; ientry++) {
376     chain->GetEntry(ientry);
377     if(!(ientry%1000)) cerr << "entry: " << ientry << " ( / " << imax << " )" << endl;
378    
379 anlevin 1.71 //if(info->EvtNum() != 217502611) continue;
380    
381 dkralph 1.62 increment(counts,"start");
382     if( ctrl.debug ) {
383     cout << "-----------------------------------------------------------------" << endl;
384     cout << "-----------------------------------------------------------------" << endl;
385     cout << "Run: " << info->RunNum()
386     << "\tEvt: " << info->EvtNum()
387     << "\tLumi: " << info->LumiSec()
388     << endl;
389     cerr << "Run: " << info->RunNum()
390     << "\tEvt: " << info->EvtNum()
391     << "\tLumi: " << info->LumiSec()
392     << endl;
393     cout << "-----------------------------------------------------------------" << endl;
394     }
395 khahn 1.41
396 dkralph 1.62 runchain->GetEvent(info->RunEntry());
397     hltchain->GetEntry(runInfo->HltEntry());
398     hltTable->Clear();
399     fillTriggerNames(hltTable, hltTableStrings );
400     // if(ctrl.debug && (lastHltEntry != runInfo->HltEntry())) hltTable->Print();
401     lastHltEntry = runInfo->HltEntry();
402     fillTriggerBits( hltTable, trigMask, triggerBits );
403     setHLTObjectRelations( hltObjArr, hltRelsArr, hltTableStrings, hltLabelStrings, fTrigObjs );
404     // printTriggerObjs( hltTable, fTrigObjs);
405 dkralph 1.53
406 dkralph 1.62 string fname = string(chain->GetFile()->GetEndpointUrl()->GetFile());
407     if(ctrl.debug) cout << "era is " << ctrl.era << endl;
408     if( ctrl.era == 0 ) {
409     setEra( fname, ctrl );
410     if(ctrl.debug) cout << "era is now " << ctrl.era << endl;
411     }
412 khahn 1.41
413 dkralph 1.62 // pfNoPU
414     PFnoPUflag.clear();
415     int pfnopu_size = makePFnoPUArray( pfArr, PFnoPUflag, vtxArr );
416     assert(pfnopu_size == pfArr->GetEntries());
417    
418     // trigger
419     if( string(chain->GetFile()->GetEndpointUrl()->GetFile()) != currentFile ) {
420     currentFile = string(chain->GetFile()->GetEndpointUrl()->GetFile());
421     }
422     if( ctrl.debug ) cout << "file is : " << currentFile << endl;
423     passes_HLT = passHLT(ctrl, triggerBits, ti, info->RunNum(), hltTable, fTrigObjs); // NOTE: run value is not used if it's MC
424 dkralph 1.67
425     // data/MC
426     if(ctrl.mc) {
427     if(ctrl.fillGen) fillGenInfo( mcArr, mcEvtInfo, geninfo, genSampleType, ctrl);
428     // NOTE: xs weight and npuw are reset in merge.cc
429     if(ctrl.use_xs_weights)
430     weights.w = getWeight(xstab,entrymap,chain)/ctrl.totalMC;
431     else
432     weights.w = 1;
433     weights.npu = getNPU(puArr, 0);
434     weights.npuw = getPUWeight(ctrl.era, fname, weights.npu);
435     } else {
436     // JSON
437     if(!(ctrl.noJSON) ) {
438     RunLumiRangeMap::RunLumiPairType rl(info->RunNum(), info->LumiSec());
439     if( !(rlrm.HasRunLumi(rl)) ) {
440     if( ctrl.debug ) cout << "fails JSON" << endl;
441     continue;
442 khahn 1.16 }
443 khahn 1.13 }
444 dkralph 1.67 }
445 khahn 1.16
446 dkralph 1.62 //
447     // lepton/kinematic selection ...
448     //
449     failingLeptons.clear();
450     passingLeptons.clear();
451     EventData ret4l = apply_HZZ4L_reference_selection(ctrl, info,
452     vtxArr,
453     pfArr,
454     puDArr,
455     electronArr,
456     &electronReferencePreSelection,
457     &electronReferenceIDMVASelectionV1, // not used anymore! (Blame the damn regression implementation, not me!)
458     &electronReferenceIsoSelection,
459     muonArr,
460     &muonReferencePreSelection,
461     &muonIDPFSelection,
462     &muonReferenceIsoSelection);
463 khahn 1.33
464 dkralph 1.62 if( ctrl.debug ) cout << endl;
465 khahn 1.14
466 dkralph 1.62 if(ctrl.fakeScheme == "none") { // fill angletuple if event passes full selection
467     if( ret4l.status.pass() ) {
468 anlevin 1.70 setNtupleVals(ctrl, ret4l, evtinfo, angles, kine, ji, weights, info, muonArr, electronArr, jetArr, metArr, simpleLeptonJets, fJetCorrector, puArr, puDArr, fJetIDMVA, vtxArr, triggerBits, hltTable, hltObjArr, fTrigObjs, &nt);
469 dkralph 1.62 nt.Fill();
470 anlevin 1.68
471     int theZ1type = kine.l1type;
472     int theZ2type = kine.l3type;
473    
474     if(ctrl.debug) cout << "simpleLeptonJets.size() = " << simpleLeptonJets.size() << endl;
475    
476     if(kine.m4l > 100){
477     if(simpleLeptonJets.size() >= 1) {
478     increment(counts,"m4l > 100 && >= 1 jets",theZ1type,theZ2type);
479     if(simpleLeptonJets.size() == 2) {
480     increment(counts,"m4l > 100 && 2 jets",theZ1type,theZ2type);
481     SimpleLepton j1 = simpleLeptonJets[0];
482     SimpleLepton j2 = simpleLeptonJets[1];
483     double deta = abs(j1.vec.Eta()-j2.vec.Eta());
484     double mjj = (j1.vec + j2.vec).M();
485     double fisher = 0.09407 * deta +0.00041581*mjj;
486     if(fisher > 0.4) increment(counts, "m4l > 100 && fisher > 0.4",theZ1type,theZ2type);
487     }
488     }
489     }
490 dkralph 1.69
491 anlevin 1.68
492 dkralph 1.62 } else {
493     if(ctrl.debug) cout << "failing with code : " << ret4l.status.selectionBits << endl;
494     }
495     } else { // fill FOtuple and angletuple if event has a good Z and one or two extra leptons
496     assert(ctrl.fakeScheme=="ZPlusF" || ctrl.fakeScheme=="ZPlusFF");
497     if(ret4l.status.selectionBits.test(PASS_GOODZ1) ) {
498     unsigned nleps = passingLeptons.size() + failingLeptons.size();
499     if( (nleps>=3 && ctrl.fakeScheme=="ZPlusF" ) || (nleps>=4 && ctrl.fakeScheme=="ZPlusFF" ) ) {
500     addDummyZ2Lepotons(ret4l);
501 anlevin 1.70 setNtupleVals(ctrl, ret4l, evtinfo, angles, kine, ji, weights, info, muonArr, electronArr, jetArr, metArr, simpleLeptonJets, fJetCorrector, puArr, puDArr, fJetIDMVA, vtxArr, triggerBits, hltTable, hltObjArr, fTrigObjs, &nt);
502 dkralph 1.62 nt.Fill();
503     foTree->Fill();
504 dkralph 1.60 }
505 dkralph 1.62 } else {
506     if(ctrl.debug) cout << "failing PASS_GOODZ1" << endl;
507 khahn 1.15 }
508 dkralph 1.53 }
509 anlevin 1.72
510     hltTable->Delete();
511    
512 dkralph 1.62 }
513 khahn 1.1
514 dkralph 1.62 if(ctrl.fakeScheme != "none") {
515     foTree->getFile()->cd();
516     foTree->getTree()->Write();
517     }
518 khahn 1.13 nt.WriteClose();
519 khahn 1.19
520 anlevin 1.57 for(unsigned icut=0; icut<cuts.size(); icut++) {
521 dkralph 1.60 map<TString,int> thisCount(*(counts[cuts[icut]]));
522 anlevin 1.68 cerr << setw(25) << cuts[icut] << "\t" << setw(9) << thisCount["all"] << " ("
523 dkralph 1.60 << setw(9) << thisCount["4e"] << setw(9) << thisCount["4m"] << setw(9) << thisCount["2e2m"] << ")" << endl;
524 anlevin 1.57 }
525 anlevin 1.72
526     tmp_file.Close();
527    
528 dkralph 1.62 }
529     //----------------------------------------------------------------------------------------
530     void addDummyZ2Lepotons(EventData &ret4l)
531     // for fake ntuples, we don't know which are the Z2 leptons yet, but need to fill something so we don't seg fault
532     {
533     SimpleLepton dum1,dum2;
534     dum1.vec.SetPtEtaPhiM(1,2,3,4);
535     dum2.vec.SetPtEtaPhiM(5,6,7,8);
536     ret4l.Z2leptons.push_back(dum1);
537     ret4l.Z2leptons.push_back(dum2);
538     }
539     //----------------------------------------------------------------------------------------
540     void setNtupleVals(ControlFlags &ctrl, EventData &ret4l, InfoStruct &evtinfo, Angles &angles, KinematicsStruct &kine,
541 anlevin 1.70 JetInfoStruct &ji, WeightStruct &weights, EventHeader *info, Array<Muon> *muonArr, Array<Electron> *electronArr, Array<PFJet> *jetArr, Array<PFMet> *metArr,
542 dkralph 1.62 vector<SimpleLepton> &simpleLeptonJets, FactorizedJetCorrector *fJetCorrector, Array<PileupInfo> *puArr,
543     Array<PileupEnergyDensity> *puDArr, JetIDMVA *fJetIDMVA, Array<Vertex> *vtxArr, bitset<1024> &triggerBits,
544     TriggerTable *hltTable, Array<TriggerObject> *hltObjArr, TriggerObjectsTable *fTrigObjs, AngleTuple *nt)
545     {
546     fillKinematics(ret4l,kine);
547 anlevin 1.70
548     //the last two arguments are usePt and useY respectively
549     fillMela(kine,0,0);
550    
551 anlevin 1.71 fillMassErrors(ctrl,ret4l,muonArr,electronArr,kine);
552 anlevin 1.70
553 dkralph 1.62 TLorentzVector pfmet; pfmet.SetPxPyPzE(metArr->At(0)->Mex(),metArr->At(0)->Mey(),0,0);
554     fillEventInfo( info, pfmet, evtinfo, ctrl.mc ? getNPU(puArr) : 0, vtxArr->GetEntries());
555 dkralph 1.67 fillJets(ctrl, jetArr, simpleLeptonJets, fJetCorrector, puDArr, fJetIDMVA, vtxArr, ret4l);
556     fillJetInfo(simpleLeptonJets, ji, ctrl);
557 anlevin 1.70 //fillJetAngleBranches(ji, ret4l, nt, ctrl);
558 dkralph 1.62 evtinfo.status = 0;
559     evtinfo.status = setHltBits(ti, triggerBits);
560    
561     if(ctrl.fakeScheme == "none") {
562     fillAngles(ret4l,angles);
563    
564     if( ctrl.mc) {
565 anlevin 1.66 if(ctrl.use_eff_weights) setEfficiencyWeights(ctrl, ctrl.era, ret4l, triggerBits, hltTable, hltObjArr, fTrigObjs, weights);
566 dkralph 1.62 if(ctrl.debug)
567     cout << "w: " << weights.w << "\t" << "npuw: " << weights.npuw << "\t" << "won: " << weights.won << "\t" << "woff: " << weights.woff << endl;
568     }
569    
570     if(ctrl.debug) {
571     stringstream ss;
572     ss << "PASS:: " << "\trun: " << evtinfo.run << "\tevt: " << evtinfo.evt << "\tlumi: " << evtinfo.lumi << "\tchannel: "
573     << kine.channel << "\tm4l: " << kine.m4l << "\tmZ1: " << kine.mZ1 << "\tmZ2: " << kine.mZ2 << endl;
574     cerr << ss.str();
575     cout << ss.str();
576     }
577     } else {
578     // fillAngles(ret4l,angles); // at this point we don't know which four leptons will constitute the ZZ system, so doesn't make sense to fill the angles
579     }
580     }