ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitHzz4l/Selection/src/Selection.cc
(Generate patch)

Comparing UserCode/MitHzz4l/Selection/src/Selection.cc (file contents):
Revision 1.12 by khahn, Sun Oct 23 10:53:01 2011 UTC vs.
Revision 1.13 by khahn, Sun Oct 30 15:12:30 2011 UTC

# Line 7 | Line 7
7   #include "HZZLikelihoodElectronSelection.h"
8   #include "HZZBDTElectronSelection.h"
9   #include "RunLumiRangeMap.h"
10 + #include "EfficiencyWeightsInterface.h"
11  
12   RunLumiRangeMap rlrm;
13  
14 < unsigned getGenChannel(mithep::TGenInfo * ginfo) {
15 <  int gchannel=-1;
16 <  if( abs(ginfo->id_1_a) == EGenType::kElectron && abs(ginfo->id_1_b) == EGenType::kElectron )  gchannel=0;
17 <  else if( abs(ginfo->id_1_a) == EGenType::kMuon && abs(ginfo->id_1_b) == EGenType::kMuon ) gchannel=1;
18 <  else if( (abs(ginfo->id_1_a) == EGenType::kElectron && abs(ginfo->id_1_b) == EGenType::kMuon) ||
19 <           (abs(ginfo->id_1_a) == EGenType::kMuon && abs(ginfo->id_1_b) == EGenType::kElectron) ) gchannel=2;
14 > // #include "TH2D.h"
15 > // extern TH2D * h_mu_eff_idiso_2011a;
16 > // extern TH2D * h_mu_eff_idiso_s11;
17 > // extern TH2D * h_mu_effdata_trigger_doublemu7_2011a;
18 > // extern TH2D * h_mu_effdata_trigger_doublemu13_8_leading_2011a;
19 > // extern TH2D * h_mu_effdata_trigger_doublemu13_8_trailing_2011a;
20 >
21 > // unsigned getGenChannel(mithep::TGenInfo * ginfo) {
22 > //   int gchannel=-1;
23 > //   if( abs(ginfo->id_1_a) == EGenType::kElectron && abs(ginfo->id_1_b) == EGenType::kElectron )  gchannel=0;
24 > //   else if( abs(ginfo->id_1_a) == EGenType::kMuon && abs(ginfo->id_1_b) == EGenType::kMuon ) gchannel=1;
25 > //   else if( (abs(ginfo->id_1_a) == EGenType::kElectron && abs(ginfo->id_1_b) == EGenType::kMuon) ||
26 > //         (abs(ginfo->id_1_a) == EGenType::kMuon && abs(ginfo->id_1_b) == EGenType::kElectron) ) gchannel=2;
27    
28 <  return gchannel;
29 < };
28 > //   return gchannel;
29 > // };
30  
23 –
31   void initRunLumiRangeMap() {
32    rlrm.AddJSONFile(std::string("./data/Cert_136033-149442_7TeV_Apr21ReReco_Collisions10_JSON.txt"));
33    //  rlrm.AddJSONFile(std::string("./data/Cert_160404-173244_7TeV_PromptReco_Collisions11_JSON_v2.txt"));
# Line 29 | Line 36 | void initRunLumiRangeMap() {
36    rlrm.AddJSONFile(std::string("./data/Cert_170249-172619_7TeV_ReReco5Aug_Collisions11_JSON.txt"));  
37   };
38  
39 < unsigned fails_HZZ4L_selection(ControlFlags &ctrl,           // input control
40 <                               mithep::TEventInfo *info,     // input event inof
41 <                               TClonesArray *electronArr,    // input electrons
42 <                               TClonesArray *muonArr,        // input muons
43 <                               double eventweight,           // weight
44 <                               TTree * passtuple ) {
39 > void initRunLumiRangeMap(RunLumiRangeMap &rlrm) {
40 >  cout << "adding JSONS ... "  << endl;
41 >  rlrm.AddJSONFile(std::string("./data/Cert_136033-149442_7TeV_Apr21ReReco_Collisions10_JSON.txt"));
42 >  //  rlrm.AddJSONFile(std::string("./data/Cert_160404-173244_7TeV_PromptReco_Collisions11_JSON_v2.txt"));
43 >  rlrm.AddJSONFile(std::string("./data/Cert_160404-178078_7TeV_PromptReco_Collisions11_JSON.txt"));
44 >  rlrm.AddJSONFile(std::string("./data/Cert_160404-163869_7TeV_May10ReReco_Collisions11_JSON_v3.txt"));  
45 >  rlrm.AddJSONFile(std::string("./data/Cert_170249-172619_7TeV_ReReco5Aug_Collisions11_JSON.txt"));  
46 > };
47  
48 <  fails_HZZ4L_selection( ctrl, NULL, info, electronArr, muonArr, eventweight, passtuple, NULL );
48 > // unsigned fails_HZZ4L_selection(ControlFlags &ctrl,           // input control
49 > //                             mithep::TEventInfo *info,     // input event inof
50 > //                             TClonesArray *electronArr,    // input electrons
51 > //                             TClonesArray *muonArr,        // input muons
52 > //                             double eventweight,           // weight
53 > //                             TTree * passtuple ) {
54  
55 < };
55 > //   fails_HZZ4L_selection( ctrl, info, electronArr, muonArr, eventweight, passtuple, NULL );
56  
57 + // };
58  
59  
45 – unsigned fails_HZZ4L_selection(ControlFlags &ctrl,           // input control
46 –                               mithep::TEventInfo *info,     // input event inof
47 –                               TClonesArray *electronArr,    // input electrons
48 –                               TClonesArray *muonArr,        // input muons
49 –                               double eventweight,           // weight
50 –                               LabVectors  *l ) {
60  
61 <  fails_HZZ4L_selection( ctrl, NULL, info, electronArr, muonArr, eventweight, NULL, l );
61 > // unsigned fails_HZZ4L_selection(ControlFlags &ctrl,           // input control
62 > //                             mithep::TEventInfo *info,     // input event inof
63 > //                             TClonesArray *electronArr,    // input electrons
64 > //                             TClonesArray *muonArr,        // input muons
65 > //                             double eventweight,           // weight
66 > //                             LabVectors  *l ) {
67  
68 < };
68 > //   fails_HZZ4L_selection( ctrl, info, electronArr, muonArr, eventweight, NULL, l );
69 >
70 > // };
71  
72  
73   unsigned fails_HZZ4L_selection(ControlFlags &ctrl,           // input control
# Line 62 | Line 78 | unsigned fails_HZZ4L_selection(ControlFl
78                                 double eventweight,           // weight
79                                 TTree * passtuple ) {
80  
81 <  fails_HZZ4L_selection( ctrl, ginfo, info, electronArr, muonArr, eventweight, passtuple, NULL );
81 >  fails_HZZ4L_selection( ctrl, info, electronArr, muonArr, eventweight, passtuple, NULL, NULL );
82  
83   };
84  
85 + unsigned fails_HZZ4L_selection(ControlFlags &ctrl,           // input control
86 +                               mithep::TEventInfo *info,     // input event info
87 +                               TClonesArray *electronArr,    // input electrons
88 +                               TClonesArray *muonArr,        // input muons
89 +                               double eventweight,           // weight
90 +                               TTree * passtuple,
91 +                               LabVectors * l ) {
92 +  fails_HZZ4L_selection( ctrl, info, electronArr, muonArr, eventweight, passtuple, l, NULL );
93 + }
94 +
95  
96   unsigned fails_HZZ4L_selection(ControlFlags &ctrl,           // input control
71 –                               mithep::TGenInfo *ginfo,     // input gen info
97                                 mithep::TEventInfo *info,     // input event info
98                                 TClonesArray *electronArr,    // input electrons
99                                 TClonesArray *muonArr,        // input muons
100                                 double eventweight,           // weight
101                                 TTree * passtuple,
102 <                               LabVectors * l) {       // output ntuple
102 >                               LabVectors * l=NULL,
103 >                               TClonesArray *jetArr=NULL ) {       // output ntuple
104  
105    unsigned evtfail = 0x0;
106    unsigned gchannel=0xdeaddead;
107  
108 <  if( ctrl.mc && ginfo != NULL ) {
109 <    gchannel = getGenChannel(ginfo);
110 <  }
108 > //   if( ctrl.mc && ginfo != NULL ) {
109 > //     gchannel = getGenChannel(ginfo);
110 > //   }
111  
112  
113    if( ctrl.debug ) {
# Line 110 | Line 136 | unsigned fails_HZZ4L_selection(ControlFl
136    //********************************************************
137    if( !ctrl.mc ) {
138      //  if( !(passHLT(info->triggerBits, info->runNum, channel) )  ) {
139 <    if( !(passHLT(info->triggerBits, info->runNum, 999) )  ) {
140 <      evtfail |= (1<<EVTFAIL_TRIGGER);
141 <      return evtfail;
142 <    }  
139 > //     if( !(passHLT(info->triggerBits, info->runNum, 999) )  ) {
140 > //       evtfail |= (1<<EVTFAIL_TRIGGER);
141 > //       return evtfail;
142 > //     }        
143    } else {
144      if( !(passHLTMC(info->triggerBits)) ) {
145        evtfail |= (1<<EVTFAIL_TRIGGER);
# Line 178 | Line 204 | unsigned fails_HZZ4L_selection(ControlFl
204    for(Int_t i=0; i<electronArr->GetEntries(); i++) {
205      const mithep::TElectron *ele = (mithep::TElectron*)((*electronArr)[i]);
206  
207 +    if( !(isEleFO(ctrl,ele) ) ) continue;
208 +    if( ele->pt < 7 ) continue; //move this to ID
209 +
210      Bool_t isMuonOverlap = kFALSE;
211      for (int k=0; k<lepvec.size(); ++k) {
212        TVector3 tmplep;
# Line 230 | Line 259 | unsigned fails_HZZ4L_selection(ControlFl
259        }
260        FAIL = failsBDT;
261      }
233 –    unsigned  failsSi=0;
234 –    if(ctrl.eleSele=="si") {
235 –      failsSi = failsSiMVAElectronSelection(ctrl, ele, 0.95, ctrl.kinematics);
236 –      FAIL = failsSi;
237 –    }
238 –
239 –
262  
263  
264      if( ctrl.debug ){
# Line 276 | Line 298 | unsigned fails_HZZ4L_selection(ControlFl
298    
299    sort( lepvec.begin(), lepvec.end(), SimpleLepton::lep_pt_sort );
300  
301 +  for( int i=0; i<lepvec.size(); i++ ) {
302 +    //
303 +    //  check for a matched btagged jet
304 +    //
305 +    lepvec[i].tche = -1;
306 +    lepvec[i].tchp = -1;
307 +    lepvec[i].csv = -1;
308 +    lepvec[i].csvMva = -1;
309 +
310 +    if( jetArr != NULL ) {
311 +      for(int k=0; k<jetArr->GetEntries(); k++) {
312 +        const mithep::TJet *jet = (mithep::TJet*)((*jetArr)[k]);      
313 +        TVector3 jvec;
314 +        jvec.SetPtEtaPhi(jet->pt, jet->eta, jet->phi);
315 +        if( jvec.DrEtaPhi( lepvec[i].vec.Vect()) < 0.2 ) {
316 +          lepvec[i].tche = jet->tche;
317 +          lepvec[i].tchp = jet->tchp;
318 +          lepvec[i].csv = jet->csv;
319 +          lepvec[i].csvMva = jet->csvMva;
320 +        }
321 +      }
322 +    }
323 +  }
324 +
325    int nmu=0, nele=0;
326    for( int i=0; i<lepvec.size(); i++ ) {
327      if( abs(lepvec[i].type) == 11 ) nele++;
# Line 692 | Line 738 | unsigned fails_HZZ4L_selection(ControlFl
738              (lepvec[Z1LeptonMinusIndex].type == 13 && lepvec[Z2LeptonMinusIndex].type == 11)) channel=2;
739          
740  
741 +        double w_offline=-1, werr_offline=0;
742 +        double w_online=-1, werr_online=0;
743 +
744 +        if( ctrl.mc ) {
745 +
746 +          vector< pair <double,double> > wlegs; // pair here is eff & err
747 +          vector< pair <float,float> > mvec;  // pair here is eta & pt
748 +          vector< pair <float,float> > evec;  // pair here is eta & pt
749 +
750 +          for( int k=0; k<lepvec.size(); k++ ) {
751 +            if( !(lepvec[k].is4l) ) continue;
752 +            if( abs(lepvec[k].type) == 13 ) {
753 +              mvec.push_back( std::pair<float,float> (fabs(lepvec[k].vec.Eta()), lepvec[k].vec.Pt()) );
754 +              wlegs.push_back( muonPerLegOfflineEfficiencyWeight( fabs(lepvec[k].vec.Eta()),
755 +                                                                  lepvec[k].vec.Pt() ) );
756 +            } else {
757 +              evec.push_back( std::pair<float,float> (fabs(lepvec[k].vec.Eta()), lepvec[k].vec.Pt()) );
758 +              wlegs.push_back( elePerLegOfflineEfficiencyWeight(  fabs(lepvec[k].vec.Eta()),
759 +                                                                 lepvec[k].vec.Pt() ) );
760 +                      //              wlegs.push_back( pair<double,double> (1.,0.) );
761 +            }
762 +          }
763  
764 +          pair<double,double> offpair = getOfflineEfficiencyWeight( wlegs );
765 +          w_offline    = offpair.first;
766 +          werr_offline = offpair.second;
767 +
768 +          pair<double,double> onpair  = getOnlineEfficiencyWeight( mvec, evec );
769 +          w_online    = onpair.first;
770 +          werr_online = onpair.second;
771 +        } // if mc
772 +
773 +
774 +        if( l != NULL ) {
775 +          l->vec4l  = ZZSystem;
776 +          l->vecz1  = Z1Candidate;
777 +          l->vecz2  = Z2Candidate;
778 +          l->vecl1p = lepvec[Z1LeptonPlusIndex].vec;
779 +          l->vecl1m = lepvec[Z1LeptonMinusIndex].vec;
780 +          l->vecl2p = lepvec[Z2LeptonPlusIndex].vec;
781 +          l->vecl2m = lepvec[Z2LeptonMinusIndex].vec;
782 +        }
783          if( passtuple != NULL ) {
784            unsigned run   = info->runNum;
785            unsigned evt   = info->evtNum;
# Line 705 | Line 792 | unsigned fails_HZZ4L_selection(ControlFl
792            float pt4l     = ZZSystem.Pt() ;
793            unsigned tZ1   = abs(lepvec[Z1LeptonMinusIndex].type);
794            unsigned tZ2   = abs(lepvec[Z2LeptonMinusIndex].type);
795 +          float l3tche   = lepvec[Z2LeptonMinusIndex].tche;
796 +          float l4tche   = lepvec[Z2LeptonPlusIndex].tche;
797 +          float l3tchp   = lepvec[Z2LeptonMinusIndex].tchp;
798 +          float l4tchp   = lepvec[Z2LeptonPlusIndex].tchp;
799 +          float l3csv    = lepvec[Z2LeptonMinusIndex].csv;
800 +          float l4csv    = lepvec[Z2LeptonPlusIndex].csv;
801 +          float l3csvMva = lepvec[Z2LeptonMinusIndex].csvMva;
802 +          float l4csvMva = lepvec[Z2LeptonPlusIndex].csvMva;
803 +
804 +
805            passtuple->SetBranchAddress("channel",  &channel);
806            passtuple->SetBranchAddress("run",  &run);
807            passtuple->SetBranchAddress("evt",  &evt);
# Line 716 | Line 813 | unsigned fails_HZZ4L_selection(ControlFl
813            passtuple->SetBranchAddress("m4l",  &m4l);
814            passtuple->SetBranchAddress("pt4l", &pt4l);
815            passtuple->SetBranchAddress("w",    &w);
816 +          if( ctrl.mc ) {
817 +            passtuple->SetBranchAddress("woff", &w_offline );
818 +            passtuple->SetBranchAddress("werroff", &werr_offline );
819 +            passtuple->SetBranchAddress("won", &w_online );
820 +            passtuple->SetBranchAddress("werron", &werr_online );
821 +          }
822 +          if( ctrl.btag ) {
823 +          passtuple->SetBranchAddress("l3tche",    &l3tche);
824 +          passtuple->SetBranchAddress("l4tche",    &l4tche);
825 +          passtuple->SetBranchAddress("l3tchp",    &l3tchp);
826 +          passtuple->SetBranchAddress("l4tchp",    &l4tchp);
827 +          passtuple->SetBranchAddress("l3csv",     &l3csv);
828 +          passtuple->SetBranchAddress("l4csv",     &l4csv);
829 +          passtuple->SetBranchAddress("l3csvMva",  &l3csvMva);
830 +          passtuple->SetBranchAddress("l4csvMva",  &l4csvMva);
831 +          }
832 +
833            if( ctrl.mc )           passtuple->SetBranchAddress("gchannel",  &gchannel);
834            passtuple->Fill( );
835 +
836          }
837  
838          if( ctrl.debug ) cout  << "run: " << info->runNum  

Diff Legend

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