PFMCTruthTreeProducer.cc

Go to the documentation of this file.

#include <iostream>
#include <sstream>
#include <istream>
#include <fstream>
#include <iomanip>
#include <string>
#include <cmath>
#include <functional>

#include "JetMETCorrections/MCJet/plugins/PFMCTruthTreeProducer.h"
#include "JetMETCorrections/MCJet/plugins/JetUtilMC.h"
#include "FWCore/Framework/interface/Event.h"
#include "DataFormats/Common/interface/Handle.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "Geometry/Records/interface/IdealGeometryRecord.h"
#include "DataFormats/JetReco/interface/PFJet.h"
#include "DataFormats/JetReco/interface/GenJet.h"
using namespace edm;
using namespace reco;
using namespace std;
//namespace cms
//{

PFMCTruthTreeProducer::PFMCTruthTreeProducer(edm::ParameterSet const& cfg)
{
  jets_          = consumes<PFJetCollection>(edm::InputTag(cfg.getParameter<std::string> ("jets")));
  genjets_       = consumes<GenJetCollection>(edm::InputTag(cfg.getParameter<std::string> ("genjets")));
  gen_       = consumes<GenEventInfoProduct>(edm::InputTag("generator"));
  histogramFile_ = cfg.getParameter<std::string> ("histogramFile");
}
void PFMCTruthTreeProducer::beginJob()
{
  file_          = new TFile(histogramFile_.c_str(),"RECREATE");
  mcTruthTree_   = new TTree("mcTruthTree","mcTruthTree");

  mcTruthTree_->Branch("ptJet",      &ptJet_,      "ptJet_/F");
  mcTruthTree_->Branch("ptGen",      &ptGen_,      "ptGen_/F");
  mcTruthTree_->Branch("ptHat",      &ptHat_,      "ptHat_/F");
  mcTruthTree_->Branch("chfJet",     &chfJet_,     "chfJet_/F");
  mcTruthTree_->Branch("nhfJet",     &nhfJet_,     "nhfJet_/F");
  mcTruthTree_->Branch("cemfJet",    &cemfJet_,    "cemfJet_/F");
  mcTruthTree_->Branch("nemfJet",    &nemfJet_,    "nemfJet_/F");
  mcTruthTree_->Branch("cmultiJet",  &cmultiJet_,  "cmultiJet_/I");
  mcTruthTree_->Branch("nmultiJet",  &nmultiJet_,  "nmultiJet_/I");
  mcTruthTree_->Branch("etaJet",     &etaJet_,    "etaJet_/F");
  mcTruthTree_->Branch("etaGen",     &etaGen_,     "etaGen_/F");
  mcTruthTree_->Branch("phiJet",     &phiJet_,     "phiJet_/F");
  mcTruthTree_->Branch("phiGen",     &phiGen_,     "phiGen_/F");
  mcTruthTree_->Branch("dR",         &dR_,         "dR_/F");
  mcTruthTree_->Branch("rank",       &rank_,       "rank_/I");
}
void PFMCTruthTreeProducer::endJob()
{
  if (file_ !=nullptr)
    {
      file_->cd();
      mcTruthTree_->Write();
    }
  file_ = nullptr;
}
void PFMCTruthTreeProducer::analyze(edm::Event const& event, edm::EventSetup const& iSetup)
{
  edm::Handle<GenJetCollection> genjets;
  edm::Handle<PFJetCollection> jets;
  edm::Handle<GenEventInfoProduct> hEventInfo;
  PFJetCollection::const_iterator i_jet,i_matched;
  GenJetCollection::const_iterator i_genjet;
  event.getByToken (genjets_,genjets);
  event.getByToken (jets_,jets);
  event.getByToken(gen_,hEventInfo);
  ptHat_ = hEventInfo->binningValues()[0];
  float rr;
  int njet(0);
  if (!jets->empty() && !genjets->empty())
    {
      for (i_genjet = genjets->begin(); i_genjet != genjets->end(); i_genjet++)
       {
         float rmin(99);
         for(i_jet = jets->begin();i_jet != jets->end(); i_jet++)
           {
         rr=radius(i_genjet,i_jet);
         if (rr<rmin)
               {
                 rmin = rr;
                 i_matched = i_jet;
               }
       }
         ptGen_     = i_genjet->pt();
         etaGen_    = i_genjet->eta();
         phiGen_    = i_genjet->phi();
         ptJet_     = i_matched->pt();
         etaJet_    = i_matched->eta();
         phiJet_    = i_matched->phi();
         chfJet_    = i_matched->chargedHadronEnergyFraction();
         nhfJet_    = i_matched->neutralHadronEnergyFraction();
         cemfJet_   = i_matched->chargedEmEnergyFraction();
         nemfJet_   = i_matched->neutralEmEnergyFraction();
         cmultiJet_ = i_matched->chargedMultiplicity();
         nmultiJet_ = i_matched->neutralMultiplicity();
         dR_        = rmin;
         rank_      = njet;
         mcTruthTree_->Fill();
         njet++;
       }
    }
}
PFMCTruthTreeProducer::~PFMCTruthTreeProducer()
{
  delete file_;
  delete mcTruthTree_;
}
//}