[ViewVC] Diff of: cvsroot/UserCode/HbbAnalysis/plugins/MCTreeMaker.cc

Comparing UserCode/HbbAnalysis/plugins/MCTreeMaker.cc (file contents):
Revision 1.1 by amagnan, Wed Apr 18 12:56:47 2012 UTC vs.
Revision 1.2 by amagnan, Wed Jun 6 13:42:42 2012 UTC

+#include "DataFormats/HepMCCandidate/interface/GenParticle.h"
+#include "DataFormats/GeometryVector/interface/GlobalPoint.h"
+#include "DataFormats/JetReco/interface/GenJet.h"
-+
+#include "DataFormats/JetReco/interface/Jet.h"
+#include "SimDataFormats/GeneratorProducts/interface/GenRunInfoProduct.h"
+#include "SimDataFormats/GeneratorProducts/interface/HepMCProduct.h"
+#include "SimDataFormats/GeneratorProducts/interface/LHEEventProduct.h"
+  genParticleSrc_(pset.getParameter<edm::InputTag>("GenParticles")),
+  genJetSrc_(pset.getParameter<edm::InputTag>("GenJets")),
+  partonJetSrc_(pset.getParameter<edm::InputTag>("PartonJets")),
-+
+  pfJetSrc_(pset.getParameter<edm::InputTag>("PFJets")),
+  status1ToKeep_(pset.getParameter<std::vector<std::string> >("Status1ToKeep")),
+  status2ToKeep_(pset.getParameter<std::vector<std::string> >("Status2ToKeep")),
+  status3ToKeep_(pset.getParameter<std::vector<std::string> >("Status3ToKeep")),
+  //if (doGen_) HbbParticles(lGenParticles,event_->particles());
+  HbbParticles(lGenParticles,event_->particles(), event_->p4total());
-+
+  edm::Handle<std::vector<pat::Jet> > lPfJetCollection;
-+
-+
+  try {
-+
+    aEvt.getByLabel(pfJetSrc_,lPfJetCollection);
-+
+    if (!lPfJetCollection.isValid()){
-+
+      edm::LogInfo("ERROR")<< "Error! Can't get pfJets by label. ";
-+
+    }
-+
+    if (debug_) std::cout << "** pfJetcollection = " << lPfJetCollection->size() << " elements." << std::endl;
-+
+  } catch(cms::Exception& e)  {
-+
+    std::cout << "AMM: Collection " << pfJetSrc_  << " not available! Exception : " << e.what() << ". " << std::endl;
-+
+  }
-+
-+
+  //std::cout << "Processing PF jets:" << std::endl;
-+
+  HbbJets(lPfJetCollection,lGenParticles,event_->pfjets());
-+
+  tree_->Fill();
+  firstEvent = false;
+    }//loop on particles
-<
+  if (debug_>1){
->
+  if (debug_>2){
+    for (unsigned int imc(0); imc < aVec.size(); ++imc){
+      HbbAnalysis::GenParticle & p = aVec.at(imc);
+      p.Print();
+    }//loop on particles
-<
+  if (debug_>1){
->
+  if (debug_>2){
+    for (unsigned int imc(0); imc < aVec.size(); ++imc){
+      //HbbAnalysis::GenParticle & p = aVec.at(imc);
+      //p.print();
-+
-+
+void MCTreeMaker::HbbJets(const edm::Handle<std::vector<pat::Jet> > & aCol,
-+
+                           const edm::Handle<reco::GenParticleCollection> & aGenParticles,
-+
+                           std::vector<HbbAnalysis::Jet> & aVec)
-+
+{//HbbJets
-+
-+
+  if (aCol.isValid()){//valid
-+
+    if (aCol->size() > 0) {//non empty
-+
-+
+      unsigned int iEle = 0;
-+
+      for (std::vector<pat::Jet>::const_iterator iter = aCol->begin();
-+
+           iter != aCol->end();
-+
+           ++iter,++iEle)
-+
+        {//loop on element
-+
+          if (debug_ > 1) std::cout << "**** Ele #" << iEle << ", pT,eta,phi =  " << (*iter).pt() << " " << (*iter).eta() << " " << (*iter).phi() << std::endl;
-+
-+
+          //----------------------------------------------------------------------
-+
+          HbbAnalysis::FourMomentum lFourVec;
-+
+          //----------------------------------------------------------------------
-+
+          lFourVec.e = (*iter).energy();
-+
+          lFourVec.px = (*iter).px();
-+
+          lFourVec.py = (*iter).py();
-+
+          lFourVec.pz = (*iter).pz();
-+
+          //----------------------------------------------------------------------
-+
+          HbbAnalysis::JetVars lJet;
-+
+          //----------------------------------------------------------------------
-+
+          lJet.partonFlavour = (*iter).partonFlavour();
-+
+          lJet.nAssociatedTracks = ((*iter).associatedTracks()).size();
-+
+          lJet.etaMean = (*iter).etaPhiStatistics().etaMean;
-+
+          lJet.phiMean = (*iter).etaPhiStatistics().phiMean;
-+
+          lJet.etaEtaMoment = (*iter).etaPhiStatistics().etaEtaMoment;
-+
+          lJet.phiPhiMoment = (*iter).etaPhiStatistics().phiPhiMoment;
-+
+          lJet.etaPhiMoment = (*iter).etaPhiStatistics().etaPhiMoment;
-+
+          //Construct a GenParticle if processing MC and a match has been found by PAT
-+
+          if ((*iter).genParton()){
-+
+            HbbAnalysis::FourMomentum lGenFourVec;
-+
+            HbbAnalysis::GenVars lGen;
-+
+            lGenFourVec.px = (*iter).genParton()->px();
-+
+            lGenFourVec.py = (*iter).genParton()->py();
-+
+            lGenFourVec.pz = (*iter).genParton()->pz();
-+
+            lGenFourVec.e = (*iter).genParton()->energy();
-+
+            lGen.charge = (*iter).genParton()->charge();
-+
+            lGen.pdgId = (*iter).genParton()->pdgId();
-+
+            lGen.status = (*iter).genParton()->status();
-+
+            lGen.vertex.vx = (*iter).genParton()->vx();
-+
+            lGen.vertex.vy = (*iter).genParton()->vy();
-+
+            lGen.vertex.vz = (*iter).genParton()->vz();
-+
+            lJet.genMatches.push_back(HbbAnalysis::GenParticle(lGenFourVec, lGen));
-+
+          }
-+
+          //Construct a GenJet if processing MC and a match has been found by PAT
-+
+          if ((*iter).genJet()){
-+
+            HbbAnalysis::FourMomentum lGenJetFourVec;
-+
+            HbbAnalysis::GenJetVars lGenJet;
-+
+            lGenJetFourVec.px = (*iter).genJet()->px();
-+
+            lGenJetFourVec.py = (*iter).genJet()->py();
-+
+            lGenJetFourVec.pz = (*iter).genJet()->pz();
-+
+            lGenJetFourVec.e = (*iter).genJet()->energy();
-+
+            lGenJet.flavour = 0;  //Flavour unknown in this context (will be filled for parton jets)
-+
+            lGenJet.mass = (*iter).genJet()->mass();
-+
+            lGenJet.nConstituents = (*iter).genJet()->getGenConstituents().size();
-+
+            lJet.genJetMatches.push_back(HbbAnalysis::GenJet(lGenJetFourVec, lGenJet));
-+
+          }
-+
-+
+          HbbAnalysis::JetECorVars lEnergyCorrections;
-+
+          if ((*iter).jecSetsAvailable()){
-+
+            std::vector<std::string> lJECnames = (*iter).availableJECSets();
-+
+            for (unsigned int iSet(0); iSet < lJECnames.size(); ++iSet){
-+
+              if (debug_>1) std::cout << " -- JEC SET " << iSet << " " << lJECnames[iSet] << std::endl;
-+
+              std::vector<std::string> lJEClevels = (*iter).availableJECLevels(iSet);
-+
+              for (unsigned int iLev(0); iLev<lJEClevels.size(); ++iLev){
-+
+                if (debug_>1) std::cout << " ---- JEC level " << iLev << " " << lJEClevels[iLev]
-+
+                                      << " , factor = " << (*iter).jecFactor(iLev,pat::JetCorrFactors::NONE,iSet)
-+
+                                      << std::endl;
-+
+                lEnergyCorrections.Levels[(lJEClevels[iLev])] = (*iter).jecFactor(iLev,pat::JetCorrFactors::NONE,iSet);
-+
+              }
-+
+            }
-+
+          }
-+
-+
+          HbbAnalysis::JetBtagVars lBtag;
-+
+          if (debug_>1) {
-+
+            const std::vector<std::pair<std::string, float> > & lDiscris = (*iter).getPairDiscri();
-+
+            std::cout << " -- btaggers: " << lDiscris.size() << " elements"
-+
+                      << std::endl;
-+
+            for (unsigned int iD(0); iD<lDiscris.size(); ++iD){
-+
+              std::cout << "---- " << iD << " " << lDiscris[iD].first << std::endl;
-+
+            }
-+
+          }
-+
-+
+          lBtag.cSV = (*iter).bDiscriminator("combinedSecondaryVertexBJetTags");
-+
+          lBtag.cSVMVA = (*iter).bDiscriminator("combinedSecondaryVertexMVABJetTags");
-+
+          //lBtag.iPMVA = (*iter).bDiscriminator("impactParameterMVABJetTags");
-+
+          lBtag.bProba = (*iter).bDiscriminator("jetBProbabilityBJetTags");
-+
+          lBtag.probability = (*iter).bDiscriminator("jetProbabilityBJetTags");
-+
+          lBtag.sSVHP = (*iter).bDiscriminator("simpleSecondaryVertexHighPurBJetTags");
-+
+          lBtag.sSVHE = (*iter).bDiscriminator("simpleSecondaryVertexHighEffBJetTags");
-+
+          //lBtag.softElectronByPt = ((*iter).bDiscriminator("softElectronByPtBJetTags") < -100) ? -100 : (*iter).bDiscriminator("softElectronByPtBJetTags");
-+
+          //lBtag.softElectronByIP3d = ((*iter).bDiscriminator("softElectronByIP3dBJetTags") < -100) ? -100 : (*iter).bDiscriminator("softElectronByIP3dBJetTags");
-+
+          lBtag.softMuon = ((*iter).bDiscriminator("softMuonBJetTags") < -100) ? -100 : (*iter).bDiscriminator("softMuonBJetTags");
-+
+          lBtag.softMuonByPt = ((*iter).bDiscriminator("softMuonByPtBJetTags") < -100) ? -100 : (*iter).bDiscriminator("softMuonByPtBJetTags");
-+
+          lBtag.softMuonByIP3d = ((*iter).bDiscriminator("softMuonByIP3dBJetTags") < -100) ? -100 : (*iter).bDiscriminator("softMuonByIP3dBJetTags");
-+
+          lBtag.tCHE = (*iter).bDiscriminator("trackCountingHighEffBJetTags");
-+
+          lBtag.tCHP = (*iter).bDiscriminator("trackCountingHighPurBJetTags");
-+
-+
+          //std::cout << " -- New values of b-discri for jet : " << iEle << std::endl
-+
+          //        << " ---- softElecs: " << lBtag.softElectronByPt << " " << lBtag.softElectronByIP3d << std::endl
-+
+          //        << " ---- softMus: " << lBtag.softMuon << " " << lBtag.softMuonByPt << " " << lBtag.softMuonByIP3d << std::endl;
-+
-+
+          if (debug_>1) {
-+
+            std::cout << " -- Filling tag info variables : "
-+
+                      << (*iter).hasTagInfo("SecondaryVertex")
-+
+                      << std::endl;
-+
+          }
-+
-+
+          std::vector<HbbAnalysis::JetSecVertexVars> jetSecVertexVars;
-+
-+
+          for (unsigned iVtx = 0; iVtx < (*iter).tagInfoSecondaryVertex()->nVertices(); ++iVtx){
-+
+            HbbAnalysis::JetSecVertexVars jetSecVertexVarsTmp;
-+
+            math::XYZTLorentzVectorD vec = (*iter).tagInfoSecondaryVertex()->secondaryVertex(iVtx).p4();
-+
+            jetSecVertexVarsTmp.fourVec.e  = vec.E();
-+
+            jetSecVertexVarsTmp.fourVec.px = vec.Px();
-+
+            jetSecVertexVarsTmp.fourVec.py = vec.Py();
-+
+            jetSecVertexVarsTmp.fourVec.pz = vec.Pz();
-+
+            jetSecVertexVarsTmp.vertex.vx = (*iter).tagInfoSecondaryVertex()->secondaryVertex(iVtx).x();
-+
+            jetSecVertexVarsTmp.vertex.vy = (*iter).tagInfoSecondaryVertex()->secondaryVertex(iVtx).y();
-+
+            jetSecVertexVarsTmp.vertex.vz = (*iter).tagInfoSecondaryVertex()->secondaryVertex(iVtx).z();
-+
+            jetSecVertexVars.push_back(jetSecVertexVarsTmp);
-+
+          }
-+
-+
-+
+          bool isPF = (*iter).isPFJet();
-+
+          if (debug_>1) {
-+
+            std::cout << " -- Filling JPTPF jet variables " << std::endl;
-+
+          }
-+
-+
+          HbbAnalysis::JPTPFJetVars lJPTPF;
-+
+          if (isPF){
-+
+            lJPTPF.chargedEmEnergy = (*iter).chargedEmEnergy();
-+
+            lJPTPF.neutralEmEnergy = (*iter).neutralEmEnergy();
-+
+            lJPTPF.muonMultiplicity = (*iter).muonMultiplicity();
-+
+            lJPTPF.chargedHadronEnergy = (*iter).chargedHadronEnergy();
-+
+            lJPTPF.neutralHadronEnergy = (*iter).neutralHadronEnergy();
-+
+            lJPTPF.chargedMultiplicity = (*iter).chargedMultiplicity();
-+
+            lJPTPF.chargedHadronEnergyFraction = (*iter).chargedHadronEnergyFraction();
-+
+            lJPTPF.neutralHadronEnergyFraction = (*iter).neutralHadronEnergyFraction();
-+
+            lJPTPF.chargedEmEnergyFraction = (*iter).chargedEmEnergyFraction();
-+
+            lJPTPF.neutralEmEnergyFraction = (*iter).neutralEmEnergyFraction();
-+
+          }
-+
+          if (debug_>1) {
-+
+            std::cout << " -- Filling PF jet variables " << std::endl;
-+
+          }
-+
+          if (isPF) {
-+
+            HbbAnalysis::PFJetVars lPF;
-+
+            lPF.chargedMuEnergy = (*iter).chargedMuEnergy();
-+
+            lPF.chargedMuEnergyFraction = (*iter).chargedMuEnergyFraction();
-+
+            lPF.neutralMultiplicity = (*iter).neutralMultiplicity();
-+
+            lPF.numberOfDaughters = (*iter).numberOfDaughters();
-+
+            lPF.HFHadronEnergy = (*iter).HFHadronEnergy();
-+
-+
+            HbbAnalysis::Jet lObj(lFourVec,lJet,lPF,lJPTPF,lBtag,lEnergyCorrections);
-+
+            lObj.secVertexVars(jetSecVertexVars);
-+
+            aVec.push_back(lObj);
-+
+          }
-+
-+
+          if (debug_>1) {
-+
+            std::cout << " -- Element added. " << std::endl;
-+
+          }
-+
-+
+        }//loop on element
-+
+    }//non empty
-+
+  }//valid
-+
-+
+}//HbbJets
-+
-+
+#include "FWCore/Framework/interface/MakerMacros.h"
+DEFINE_FWK_MODULE(MCTreeMaker);

Diff Legend

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

Comparing UserCode/HbbAnalysis/plugins/MCTreeMaker.cc (file contents): Revision 1.1 by amagnan, Wed Apr 18 12:56:47 2012 UTC vs. Revision 1.2 by amagnan, Wed Jun 6 13:42:42 2012 UTC

Diff Legend

Comparing UserCode/HbbAnalysis/plugins/MCTreeMaker.cc (file contents):
Revision 1.1 by amagnan, Wed Apr 18 12:56:47 2012 UTC vs.
Revision 1.2 by amagnan, Wed Jun 6 13:42:42 2012 UTC