ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/HbbAnalysis/plugins/HbbTreeMaker.cc
(Generate patch)

Comparing UserCode/HbbAnalysis/plugins/HbbTreeMaker.cc (file contents):
Revision 1.3 by amagnan, Tue Oct 13 12:14:05 2009 UTC vs.
Revision 1.12 by amagnan, Fri Mar 26 17:52:50 2010 UTC

# Line 6 | Line 6
6   #include "DataFormats/HepMCCandidate/interface/GenParticle.h"
7   #include "DataFormats/MuonReco/interface/Muon.h"
8   #include "DataFormats/MuonReco/interface/MuonFwd.h"
9 + #include "DataFormats/MuonReco/interface/MuonSelectors.h"
10   #include "DataFormats/VertexReco/interface/Vertex.h"
11   #include "DataFormats/HLTReco/interface/TriggerEvent.h"
12   #include "DataFormats/TrackReco/interface/TrackFwd.h"
# Line 18 | Line 19
19   #include "DataFormats/PatCandidates/interface/MET.h"
20  
21   #include "FWCore/ServiceRegistry/interface/Service.h"
22 < #include "PhysicsTools/UtilAlgos/interface/TFileService.h"
22 > #include "CommonTools/UtilAlgos/interface/TFileService.h"
23  
24  
25   #include "UserCode/HbbAnalysis/interface/Electron.hh"
# Line 27 | Line 28
28   #include "UserCode/HbbAnalysis/interface/Jet.hh"
29   #include "UserCode/HbbAnalysis/interface/Met.hh"
30   #include "UserCode/HbbAnalysis/interface/Trigger.hh"
31 + #include "UserCode/HbbAnalysis/interface/Vertex.hh"
32  
33   #include "UserCode/HbbAnalysis/plugins/HbbTreeMaker.hh"
34  
# Line 34 | Line 36 | using namespace HbbAnalysis;
36  
37   HbbTreeMaker::HbbTreeMaker(const edm::ParameterSet & pset):
38    debug_(pset.getParameter<int>("DEBUG")),
39 +  processData_(false),
40    flavour_(pset.getParameter<unsigned int>("JetFlavour")),
41    doGen_(pset.getParameter<bool>("DOGEN")),
42    genParticleSrc_(pset.getParameter<edm::InputTag>("GenParticles")),
# Line 67 | Line 70 | HbbTreeMaker::~HbbTreeMaker(){//destruct
70  
71  
72  
73 < void HbbTreeMaker::beginJob(const edm::EventSetup&){//beginJob
73 > void HbbTreeMaker::beginJob(){//beginJob
74  
75  
76    edm::Service<TFileService> lFileService;
# Line 81 | Line 84 | void HbbTreeMaker::beginJob(const edm::E
84    if (debug_) std::cout << "Initialising JetFlavour : " << std::endl;
85  
86    lDir = lFileService->mkdir("JetFlavours");
84
87    jetFlav_.Initialise(lDir, debug_, flavour_);
88  
89   }//beginJob
90  
91   void HbbTreeMaker::endJob(){//endJob
92 <  jetFlav_.printSummary();
92 >  if (!processData_) jetFlav_.printSummary();
93  
94    //tree_->Write();
95    //delete tree_;
# Line 105 | Line 107 | void HbbTreeMaker::analyze(const edm::Ev
107  
108    event_->Clear();
109    event_->event(aEvt.id().event());
110 +  processData_ = aEvt.isRealData();
111 +
112 +  event_->run(aEvt.run());
113 +  event_->lumiBlock(aEvt.luminosityBlock());
114 +  event_->isRealData(aEvt.isRealData());
115  
116    edm::Handle<reco::GenParticleCollection> lGenParticles;
117    try {
118      aEvt.getByLabel(genParticleSrc_,lGenParticles);  
119      if (debug_) std::cout << "** ngenParticles = " << lGenParticles->size() << std::endl;
120    } catch(cms::Exception& e)  {
121 <    std::cout << "AMM: Collection genParticles not available! Exception : " << e.what() << ". " << std::endl;
121 >    if (!processData_) std::cout << "AMM: Collection genParticles not available! Exception : " << e.what() << ". " << std::endl;
122    }
123  
124 <  if (doGen_) HbbParticles(lGenParticles,event_->particles());
124 >  if (doGen_ && !processData_) HbbParticles(lGenParticles,event_->particles());
125  
126 <  unsigned int lNPartons = jetFlav_.fillPartons(lGenParticles);
126 >  unsigned int lNPartons = 0;
127 >  if (!processData_) lNPartons = jetFlav_.fillPartons(lGenParticles);
128    
129 <  if (debug_) std::cout << "--- Number of partons = " << lNPartons << "." << std::endl;
129 >  if (debug_ && !processData_) std::cout << "--- Number of partons = " << lNPartons << "." << std::endl;
130  
131  
132    edm::Handle<std::vector<reco::Vertex> > lRecoVertices;
# Line 132 | Line 140 | void HbbTreeMaker::analyze(const edm::Ev
140      std::cout << "AMM: Collection " << vertexSrc_  << " not available! Exception : " << e.what() << ". " << std::endl;
141    }
142  
143 +  HbbVertices(lRecoVertices,event_->vertices());
144 +
145    edm::Handle<std::vector<pat::Electron> > lElectronCollection;
146    
147    try {
# Line 162 | Line 172 | void HbbTreeMaker::analyze(const edm::Ev
172    HbbMuons(lMuonCollection,lRecoVertices,event_->muons());
173  
174  
175 <  edm::Handle<std::vector<pat::Tau> > lTauCollection;
175 >  if (!( caloTauSrc_.label()=="" && caloTauSrc_.instance()=="" )) {
176 >    edm::Handle<std::vector<pat::Tau> > lTauCollection;
177    
178 <  try {
179 <    aEvt.getByLabel(caloTauSrc_,lTauCollection);
180 <    if (!lTauCollection.isValid()){
181 <      edm::LogInfo("ERROR")<< "Error! Can't get caloTau by label. ";
182 <    }
183 <    if (debug_) std::cout << "** caloTaucollection = " << lTauCollection->size() << " elements." << std::endl;
184 <  } catch(cms::Exception& e)  {
185 <    std::cout << "AMM: Collection " << caloTauSrc_  << " not available! Exception : " << e.what() << ". " << std::endl;
186 <  }
178 >    try {
179 >      aEvt.getByLabel(caloTauSrc_,lTauCollection);
180 >      if (!lTauCollection.isValid()){
181 >        edm::LogInfo("ERROR")<< "Error! Can't get caloTau by label. ";
182 >      }
183 >      if (debug_) std::cout << "** caloTaucollection = " << lTauCollection->size() << " elements." << std::endl;
184 >    } catch(cms::Exception& e)  {
185 >      std::cout << "AMM: Collection " << caloTauSrc_  << " not available! Exception : " << e.what() << ". " << std::endl;
186 >    }
187  
188 <  HbbTaus(lTauCollection,lRecoVertices,event_->caloTaus());
188 >    HbbTaus(lTauCollection,lRecoVertices,event_->caloTaus());
189 >  }
190  
191    edm::Handle<std::vector<pat::Tau> > lPFTauCollection;
192    
# Line 202 | Line 214 | void HbbTreeMaker::analyze(const edm::Ev
214      std::cout << "AMM: Collection " << caloJetSrc_  << " not available! Exception : " << e.what() << ". " << std::endl;
215    }
216  
217 +  //std::cout << "Processing calo jets:" << std::endl;
218    HbbJets(lCaloJetCollection,jetFlav_,lGenParticles,event_->caloJets());
219    
220    edm::Handle<std::vector<pat::Jet> > lJptJetCollection;
# Line 216 | Line 229 | void HbbTreeMaker::analyze(const edm::Ev
229      std::cout << "AMM: Collection " << jptJetSrc_  << " not available! Exception : " << e.what() << ". " << std::endl;
230    }
231  
232 +  //std::cout << "Processing JPT jets:" << std::endl;
233    HbbJets(lJptJetCollection,jetFlav_,lGenParticles,event_->jptJets());
234    
235    edm::Handle<std::vector<pat::Jet> > lPfJetCollection;
# Line 230 | Line 244 | void HbbTreeMaker::analyze(const edm::Ev
244      std::cout << "AMM: Collection " << pfJetSrc_  << " not available! Exception : " << e.what() << ". " << std::endl;
245    }
246  
247 +  //std::cout << "Processing PF jets:" << std::endl;
248    HbbJets(lPfJetCollection,jetFlav_,lGenParticles,event_->pfJets());
249    
250    edm::Handle<std::vector<pat::MET> > lCaloMetCol;
# Line 312 | Line 327 | void HbbTreeMaker::HbbElectrons(const ed
327            //if (debug_ > 1) std::cout << "**** Ele #" << iEle << ", pT,eta,phi =  " << (*iter).pt() << " " << (*iter).eta() << " " << (*iter).phi() << std::endl;
328  
329            HbbAnalysis::GenVars lGen;
330 <          if ((*iter).genLepton()){
330 >          if (!processData_ && (*iter).genLepton()){
331              lGen.valid = true;
332              lGen.E = (*iter).genLepton()->energy();
333              lGen.pT = (*iter).genLepton()->pt();
# Line 370 | Line 385 | void HbbTreeMaker::HbbElectrons(const ed
385            lID.hOverE = (*iter).hadronicOverEm();
386            lID.deltaPhiIn = (*iter).deltaPhiSuperClusterTrackAtVtx();
387            lID.deltaEtaIn = (*iter).deltaEtaSuperClusterTrackAtVtx();
388 +          lID.ecalDrivenSeed = (*iter).ecalDrivenSeed();
389 +          lID.trackerDrivenSeed = (*iter).trackerDrivenSeed();
390  
391            HbbAnalysis::Electron lObj(lGen,lReco,lSC,lIso,lID);
392            aVec.push_back(lObj);
# Line 397 | Line 414 | void HbbTreeMaker::HbbMuons(const edm::H
414            const reco::Muon* recoMuon = dynamic_cast<const reco::Muon*>((*iter).originalObject());
415  
416            HbbAnalysis::GenVars lGen;
417 <          if ((*iter).genLepton()){
417 >          if (!processData_ && (*iter).genLepton()){
418              lGen.valid = true;
419              lGen.E = (*iter).genLepton()->energy();
420              lGen.pT = (*iter).genLepton()->pt();
# Line 479 | Line 496 | void HbbTreeMaker::HbbMuons(const edm::H
496            else if (recoMuon->isCaloMuon()) lID.type = 4;
497            else lID.type = 0;
498  
499 <          if (recoMuon->isGood(reco::Muon::AllGlobalMuons)) lID.ids.push_back(1);
500 <          if (recoMuon->isGood(reco::Muon::AllStandAloneMuons)) lID.ids.push_back(2);
501 <          if (recoMuon->isGood(reco::Muon::AllTrackerMuons)) lID.ids.push_back(3);
502 <          if (recoMuon->isGood(reco::Muon::TrackerMuonArbitrated)) lID.ids.push_back(4);
503 <          if (recoMuon->isGood(reco::Muon::AllArbitrated)) lID.ids.push_back(5);
504 <          if (recoMuon->isGood(reco::Muon::GlobalMuonPromptTight)) lID.ids.push_back(6);
505 <          if (recoMuon->isGood(reco::Muon::TMLastStationLoose)) lID.ids.push_back(7);
506 <          if (recoMuon->isGood(reco::Muon::TMLastStationTight)) lID.ids.push_back(8);
507 <          if (recoMuon->isGood(reco::Muon::TM2DCompatibilityLoose)) lID.ids.push_back(9);
508 <          if (recoMuon->isGood(reco::Muon::TM2DCompatibilityTight)) lID.ids.push_back(10);
509 <          if (recoMuon->isGood(reco::Muon::TMOneStationLoose)) lID.ids.push_back(11);
510 <          if (recoMuon->isGood(reco::Muon::TMOneStationTight)) lID.ids.push_back(12);
511 <          if (recoMuon->isGood(reco::Muon::TMLastStationOptimizedLowPtLoose)) lID.ids.push_back(13);
512 <          if (recoMuon->isGood(reco::Muon::TMLastStationOptimizedLowPtTight)) lID.ids.push_back(14);
499 >          if (muon::isGoodMuon(*recoMuon,muon::AllGlobalMuons)) lID.ids.push_back(1);
500 >          if (muon::isGoodMuon(*recoMuon,muon::AllStandAloneMuons)) lID.ids.push_back(2);
501 >          if (muon::isGoodMuon(*recoMuon,muon::AllTrackerMuons)) lID.ids.push_back(3);
502 >          if (muon::isGoodMuon(*recoMuon,muon::TrackerMuonArbitrated)) lID.ids.push_back(4);
503 >          if (muon::isGoodMuon(*recoMuon,muon::AllArbitrated)) lID.ids.push_back(5);
504 >          if (muon::isGoodMuon(*recoMuon,muon::GlobalMuonPromptTight)) lID.ids.push_back(6);
505 >          if (muon::isGoodMuon(*recoMuon,muon::TMLastStationLoose)) lID.ids.push_back(7);
506 >          if (muon::isGoodMuon(*recoMuon,muon::TMLastStationTight)) lID.ids.push_back(8);
507 >          if (muon::isGoodMuon(*recoMuon,muon::TM2DCompatibilityLoose)) lID.ids.push_back(9);
508 >          if (muon::isGoodMuon(*recoMuon,muon::TM2DCompatibilityTight)) lID.ids.push_back(10);
509 >          if (muon::isGoodMuon(*recoMuon,muon::TMOneStationLoose)) lID.ids.push_back(11);
510 >          if (muon::isGoodMuon(*recoMuon,muon::TMOneStationTight)) lID.ids.push_back(12);
511 >          if (muon::isGoodMuon(*recoMuon,muon::TMLastStationOptimizedLowPtLoose)) lID.ids.push_back(13);
512 >          if (muon::isGoodMuon(*recoMuon,muon::TMLastStationOptimizedLowPtTight)) lID.ids.push_back(14);
513 >          if (muon::isGoodMuon(*recoMuon,muon::GMTkChiCompatibility)) lID.ids.push_back(15);
514 >          if (muon::isGoodMuon(*recoMuon,muon::GMStaChiCompatibility)) lID.ids.push_back(16);
515 >          if (muon::isGoodMuon(*recoMuon,muon::GMTkKinkTight)) lID.ids.push_back(17);
516 >          if (muon::isGoodMuon(*recoMuon,muon::TMLastStationAngLoose)) lID.ids.push_back(18);
517 >          if (muon::isGoodMuon(*recoMuon,muon::TMLastStationAngTight)) lID.ids.push_back(19);
518 >          if (muon::isGoodMuon(*recoMuon,muon::TMOneStationAngLoose)) lID.ids.push_back(20);
519 >          if (muon::isGoodMuon(*recoMuon,muon::TMOneStationAngTight)) lID.ids.push_back(21);
520 >          if (muon::isGoodMuon(*recoMuon,muon::TMLastStationOptimizedBarrelLowPtLoose)) lID.ids.push_back(22);
521 >          if (muon::isGoodMuon(*recoMuon,muon::TMLastStationOptimizedBarrelLowPtTight)) lID.ids.push_back(23);
522  
523            lID.caloCompat = recoMuon->caloCompatibility();
524 <          lID.segCompat = recoMuon->segmentCompatibility();
524 >          lID.segCompat = muon::segmentCompatibility(*recoMuon);
525            lID.nChambers = recoMuon->numberOfChambers();
526            lID.nMatchesLoose = recoMuon->numberOfMatches(reco::Muon::NoArbitration);
527            lID.nMatchesMedium = recoMuon->numberOfMatches(reco::Muon::SegmentArbitration);
# Line 526 | Line 552 | void HbbTreeMaker::HbbTaus(const edm::Ha
552            //if (debug_ > 1) std::cout << "**** Ele #" << iEle << ", pT,eta,phi =  " << (*iter).pt() << " " << (*iter).eta() << " " << (*iter).phi() << std::endl;
553  
554            HbbAnalysis::GenVars lGen;
555 <          if ((*iter).genLepton()){
555 >          if (!processData_ && (*iter).genLepton()){
556              lGen.valid = true;
557              lGen.E = (*iter).genLepton()->energy();
558              lGen.pT = (*iter).genLepton()->pt();
# Line 557 | Line 583 | void HbbTreeMaker::HbbTaus(const edm::Ha
583  
584  
585            HbbAnalysis::GenVars lGenJet;
586 <          if ((*iter).genJet()){
586 >          if (!processData_ && (*iter).genJet()){
587              lGenJet.valid = true;
588 +            lGenJet.E = (*iter).genJet()->energy();
589              lGenJet.pT = (*iter).genJet()->pt();
590              lGenJet.eta = (*iter).genJet()->eta();
591              lGenJet.phi = (*iter).genJet()->phi();
# Line 572 | Line 599 | void HbbTreeMaker::HbbTaus(const edm::Ha
599            }
600            else {
601              lGenJet.valid = false;
602 +            lGenJet.E = 0;
603              lGenJet.pT = 0;
604              lGenJet.eta = 0;
605              lGenJet.phi = 0;
# Line 673 | Line 701 | void HbbTreeMaker::HbbTaus(const edm::Ha
701            const std::vector<std::pair<std::string,float> > &  lIDs = (*iter).tauIDs();
702            bool lPrint = false;
703            //a few warning if additional discriminants are found:
704 <          if ((isPF && lIDs.size() != 7) ||
704 >          if ((isPF && lIDs.size() != 16) ||
705                (isCalo && lIDs.size() != 3))
706              lPrint = true;
707  
708            if (lPrint) {
709 <            std::cout << "!!!!!!! Discriminants changed, please update histograms !!!!!!!" << std::endl;
709 >            std::cout << "!!!!!!! Discriminants changed, please update tree !!!!!!!" << std::endl;
710              std::cout << "--- isCaloTau = " << isCalo << ", isPFTau = " << isPF << std::endl;
711              std::cout << "------ ID names = " << std::endl;
712            }
713  
714  
687          HbbAnalysis::PFTauIDVars lpfId;
715            HbbAnalysis::CaloTauIDVars lcaloId;
716 +          lcaloId.byIsolation = 0;
717 +          lcaloId.byLeadingTrackFinding = 0;
718 +          lcaloId.byLeadingTrackPtCut = 0;
719 +
720 +          HbbAnalysis::PFTauIDVars lpfId;
721 +          lpfId.byLeadingTrackFinding = 0;
722 +          lpfId.byLeadingTrackPtCut = 0;
723 +          lpfId.byTrackIsolation = 0;
724 +          lpfId.byECALIsolation = 0;
725 +          lpfId.byIsolation = 0;
726 +          lpfId.againstElectron = 0;
727 +          lpfId.againstMuon = 0;
728 +          lpfId.byIsolationUsingLeadingPion = 0;
729 +          lpfId.byTaNC = 0;
730 +          lpfId.byTaNCfrHalfPercent = 0;
731 +          lpfId.byTaNCfrOnePercent = 0;
732 +          lpfId.byTaNCfrQuarterPercent = 0;
733 +          lpfId.byTaNCfrTenthPercent = 0;
734 +          lpfId.ecalIsolationUsingLeadingPion = 0;
735 +          lpfId.leadingPionPtCut = 0;
736 +          lpfId.trackIsolationUsingLeadingPion = 0;
737 +
738 +
739  
740            for (unsigned int id(0); id<lIDs.size(); id++){
741  
# Line 698 | Line 748 | void HbbTreeMaker::HbbTaus(const edm::Ha
748              if (isPF) {
749                if (lName.find("leadingTrackFinding") != lName.npos) lpfId.byLeadingTrackFinding = lDiscri;
750                if (lName.find("leadingTrackPtCut") != lName.npos) lpfId.byLeadingTrackPtCut = lDiscri;
751 <              if (lName.find("trackIsolation") != lName.npos) lpfId.byTrackIsolation = lDiscri;
752 <              if (lName.find("ecalIsolation") != lName.npos) lpfId.byECALIsolation = lDiscri;
753 <              if (lName.find("byIsolation") != lName.npos) lpfId.byIsolation = lDiscri;
751 >              if (lName.find("trackIsolationUsingLeadingPion") != lName.npos) lpfId.trackIsolationUsingLeadingPion = lDiscri;
752 >              if (lName.find("trackIsolation") != lName.npos &&
753 >                  lName.find("trackIsolationUsingLeadingPion") == lName.npos) lpfId.byTrackIsolation = lDiscri;
754 >              if (lName.find("ecalIsolationUsingLeadingPion") != lName.npos) lpfId.ecalIsolationUsingLeadingPion = lDiscri;
755 >              if (lName.find("ecalIsolation") != lName.npos &&
756 >                  lName.find("ecalIsolationUsingLeadingPion") == lName.npos) lpfId.byECALIsolation = lDiscri;
757 >              if (lName.find("byIsolationUsingLeadingPion") != lName.npos) lpfId.byIsolationUsingLeadingPion = lDiscri;
758 >              if (lName.find("byIsolation") != lName.npos &&
759 >                  lName.find("byIsolationUsingLeadingPion") == lName.npos) lpfId.byIsolation = lDiscri;
760                if (lName.find("againstElectron") != lName.npos) lpfId.againstElectron = lDiscri;
761                if (lName.find("againstMuon") != lName.npos) lpfId.againstMuon = lDiscri;
762 +              if (lName.find("byTaNCfrHalfPercent") != lName.npos) lpfId.byTaNCfrHalfPercent = lDiscri;
763 +              if (lName.find("byTaNCfrOnePercent") != lName.npos) lpfId.byTaNCfrOnePercent = lDiscri;
764 +              if (lName.find("byTaNCfrQuarterPercent") != lName.npos) lpfId.byTaNCfrQuarterPercent = lDiscri;
765 +              if (lName.find("byTaNCfrTenthPercent") != lName.npos) lpfId.byTaNCfrTenthPercent = lDiscri;
766 +              if (lName.find("byTaNC") != lName.npos &&
767 +                  lName.find("byTaNCfr") == lName.npos) lpfId.byTaNC = lDiscri;
768 +              if (lName.find("leadingPionPtCut") != lName.npos) lpfId.leadingPionPtCut = lDiscri;
769              }
770              if (isCalo){
771                if (lName.find("byIsolation") != lName.npos) lcaloId.byIsolation = lDiscri;
# Line 751 | Line 814 | void HbbTreeMaker::HbbTaus(const edm::Ha
814              HbbAnalysis::PFTauCandVars lNeutr;
815              lNeutr.nSigCands = (*iter).signalPFNeutrHadrCands().size();
816              lNeutr.nIsoCands = (*iter).isolationPFNeutrHadrCands().size();
817 <            lNeutr.isolationPtSum = 0;
817 >            lNeutr.isolationPtSum = (*iter).neutralHadronIso();
818  
819              HbbAnalysis::PFTauCandVars lGamma;
820              lGamma.nSigCands = (*iter).signalPFGammaCands().size();
# Line 805 | Line 868 | void HbbTreeMaker::HbbJets(const edm::Ha
868            //if (debug_ > 1) std::cout << "**** Ele #" << iEle << ", pT,eta,phi =  " << (*iter).pt() << " " << (*iter).eta() << " " << (*iter).phi() << std::endl;
869  
870            HbbAnalysis::GenVars lGen;
871 <          if ((*iter).genParton()){
871 >          if (!processData_ && (*iter).genParton()){
872              lGen.valid = true;
873              lGen.E = (*iter).genParton()->energy();
874              lGen.pT = (*iter).genParton()->pt();
# Line 836 | Line 899 | void HbbTreeMaker::HbbJets(const edm::Ha
899  
900  
901            HbbAnalysis::GenVars lGenJet;
902 <          if ((*iter).genJet()){
902 >          if (!processData_ && (*iter).genJet()){
903              lGenJet.valid = true;
904 +            lGenJet.E = (*iter).genJet()->energy();
905              lGenJet.pT = (*iter).genJet()->pt();
906              lGenJet.eta = (*iter).genJet()->eta();
907              lGenJet.phi = (*iter).genJet()->phi();
# Line 851 | Line 915 | void HbbTreeMaker::HbbJets(const edm::Ha
915            }
916            else {
917              lGenJet.valid = false;
918 +            lGenJet.E = 0;
919              lGenJet.pT = 0;
920              lGenJet.eta = 0;
921              lGenJet.phi = 0;
# Line 874 | Line 939 | void HbbTreeMaker::HbbJets(const edm::Ha
939            lReco.vz = (*iter).vz();
940  
941            unsigned int lFlav = 0;
942 <          if (aJetFlav.nPartons()) {
942 >          if (!processData_ && aJetFlav.nPartons()) {
943              lFlav = aJetFlav.partonMatchingGenJet((*iter),aGenParticles,0.4).second;
944            }
945  
# Line 883 | Line 948 | void HbbTreeMaker::HbbJets(const edm::Ha
948            lCommon.partonFlavour = (*iter).partonFlavour();
949            lCommon.nAssociatedTracks = ((*iter).associatedTracks()).size();
950            lCommon.rawpT = (*iter).pt();
951 <          if ((*iter).hasJetCorrFactors())
952 <            lCommon.rawpT = (*iter).pt()/((*iter).jetCorrFactors().scaleDefault());
951 >          if ((*iter).hasCorrFactors())
952 >            lCommon.rawpT = (*iter).pt()/((*iter).corrFactor((*iter).corrStep()));
953 >
954 >          lCommon.etaMean = (*iter).etaPhiStatistics().etaMean;
955 >          lCommon.phiMean = (*iter).etaPhiStatistics().phiMean;
956 >          lCommon.etaEtaMoment = (*iter).etaPhiStatistics().etaEtaMoment;
957 >          lCommon.phiPhiMoment = (*iter).etaPhiStatistics().phiPhiMoment;
958 >          lCommon.etaPhiMoment = (*iter).etaPhiStatistics().etaPhiMoment;
959 >          
960            //lCommon.rawEta = (*iter).;
961            //lCommon.rawPhi = (*iter).;
962            //    p_hasJetCorrFactors->Fill(aJet.hasJetCorrFactors());
# Line 896 | Line 968 | void HbbTreeMaker::HbbJets(const edm::Ha
968            lBtag.bProba = (*iter).bDiscriminator("jetBProbabilityBJetTags");
969            lBtag.probability = (*iter).bDiscriminator("jetProbabilityBJetTags");
970            lBtag.sSV = (*iter).bDiscriminator("simpleSecondaryVertexBJetTags");
971 <          lBtag.softElectron = (*iter).bDiscriminator("softElectronBJetTags");
972 <          lBtag.softMuon = (*iter).bDiscriminator("softMuonBJetTags");
973 <          lBtag.softMuonNoIP = (*iter).bDiscriminator("softMuonNoIPBJetTags");
971 >          lBtag.softElectronByPt = ((*iter).bDiscriminator("softElectronByPtBJetTags") < -1000) ? -1000 : (*iter).bDiscriminator("softElectronByPtBJetTags");
972 >          lBtag.softElectronByIP3d = ((*iter).bDiscriminator("softElectronByIP3dBJetTags") < -1000) ? -1000 : (*iter).bDiscriminator("softElectronByIP3dBJetTags");
973 >          lBtag.softMuon = ((*iter).bDiscriminator("softMuonBJetTags") < -1000) ? -1000 : (*iter).bDiscriminator("softMuonBJetTags");
974 >          lBtag.softMuonByPt = ((*iter).bDiscriminator("softMuonByPtBJetTags") < -1000) ? -1000 : (*iter).bDiscriminator("softMuonByPtBJetTags");
975 >          lBtag.softMuonByIP3d = ((*iter).bDiscriminator("softMuonByIP3dBJetTags") < -1000) ? -1000 : (*iter).bDiscriminator("softMuonByIP3dBJetTags");
976 >
977 >          //std::cout << " -- New values of b-discri for jet : " << iEle << std::endl
978 >          //        << " ---- softElecs: " << lBtag.softElectronByPt << " " << lBtag.softElectronByIP3d << std::endl
979 >          //        << " ---- softMus: " << lBtag.softMuon << " " << lBtag.softMuonByPt << " " << lBtag.softMuonByIP3d << std::endl;
980 >
981 >
982            lBtag.tCHE = (*iter).bDiscriminator("trackCountingHighEffBJetTags");
983            lBtag.tCHP = (*iter).bDiscriminator("trackCountingHighPurBJetTags");
984 <          
984 >
985 >
986            bool isCalo = (*iter).isCaloJet();
987            bool isPF = (*iter).isPFJet();
988  
989            assert (isCalo == !isPF);
990            if (isCalo) {
991 +            HbbAnalysis::JetIDVars lId;
992 +            lId.fHPD = (*iter).jetID().fHPD;
993 +            lId.fRBX = (*iter).jetID().fRBX;
994 +            lId.n90Hits = (*iter).jetID().n90Hits;
995 +            
996              HbbAnalysis::CaloJetVars lCalo;
997              lCalo.maxEInEmTowers = (*iter).maxEInEmTowers();
998              lCalo.maxEInHadTowers = (*iter).maxEInHadTowers();
# Line 923 | Line 1009 | void HbbTreeMaker::HbbJets(const edm::Ha
1009              lCalo.n90 = (*iter).n90();
1010              lCalo.n60 = (*iter).n60();
1011  
1012 <            HbbAnalysis::Jet lObj(lGen,lGenJet,lReco,lCommon,lCalo,lBtag);
1012 >            HbbAnalysis::Jet lObj(lGen,lGenJet,lReco,lCommon,lCalo,lBtag,lId);
1013              aVec.push_back(lObj);
1014            
1015            }
# Line 949 | Line 1035 | void HbbTreeMaker::HbbJets(const edm::Ha
1035            
1036            }
1037  
1038 +
1039 +
1040            iEle++;
1041          }//loop on element
1042      }//non empty
# Line 974 | Line 1062 | void HbbTreeMaker::HbbMet(const edm::Han
1062  
1063    const reco::GenMET *lGenMET = lMet.genMET();
1064  
1065 <  if (lGenMET){
1065 >  if (!processData_ && lGenMET){
1066      lGen.mET = lGenMET->pt();
1067      lGen.mEx = lGenMET->px();
1068      lGen.mEy = lGenMET->py();
# Line 1086 | Line 1174 | void HbbTreeMaker::HbbParticles(const ed
1174  
1175  
1176   }//genparticles
1177 +
1178 +
1179 +
1180 + void HbbTreeMaker::HbbVertices(const edm::Handle<std::vector<reco::Vertex> > & aCol,
1181 +                               std::vector<HbbAnalysis::Vertex> & aVec)
1182 + {
1183 +
1184 +  for (std::vector<reco::Vertex>::const_iterator iter = aCol->begin();
1185 +       iter != aCol->end();
1186 +       iter++)
1187 +    {
1188 +
1189 +      if (!((*iter).isValid()) || (*iter).isFake()) continue;
1190 +
1191 +      HbbAnalysis::VertexVars lVtx;
1192 +
1193 +      if ((*iter).tracksSize() > 0) {
1194 +        for (std::vector<reco::TrackBaseRef>::const_iterator lTrk = (*iter).tracks_begin();
1195 +             lTrk != (*iter).tracks_end();
1196 +             lTrk++) {
1197 +          lVtx.trackWeights.push_back((*iter).trackWeight(*lTrk));
1198 +        }
1199 +      }
1200 +
1201 +      if (lVtx.trackWeights.size() != (*iter).tracksSize()) {
1202 +        std::cout<< " -- Problem with tracks, size is not as expected ! "
1203 +                 << std::endl
1204 +                 << " --- Size of recoVertex tracks = " << (*iter).tracksSize()
1205 +                 << std::endl
1206 +                 << " --- Size of trackWeights = " << lVtx.trackWeights.size()
1207 +                 << std::endl;
1208 +      }
1209 +
1210 +      lVtx.chi2 = (*iter).chi2();
1211 +      lVtx.ndof = (*iter).ndof();
1212 +      lVtx.x = (*iter).x();
1213 +      lVtx.y = (*iter).y();
1214 +      lVtx.z = (*iter).z();
1215 +      lVtx.xError = (*iter).xError();
1216 +      lVtx.yError = (*iter).yError();
1217 +      lVtx.zError = (*iter).zError();
1218 +      lVtx.cov01 = (*iter).covariance(0,1);
1219 +      lVtx.cov02 = (*iter).covariance(0,2);
1220 +      lVtx.cov12 = (*iter).covariance(1,2);
1221 +
1222 +      HbbAnalysis::Vertex lObj(lVtx);
1223 +      aVec.push_back(lObj);
1224 +    }
1225 +
1226 + }
1227 +
1228 + #include "FWCore/Framework/interface/MakerMacros.h"
1229 + DEFINE_FWK_MODULE(HbbTreeMaker);
1230 +
1231 +

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines