JetMETCorrections/Modules/plugins/JetResolutionDemo.cc - master - githubmirror/cmssw

JetMETCorrections/Modules/plugins/JetResolutionDemo.cc
211 строк8 KB
Shahzad Malik Muzaffar
[RECONSTRUCTION] [LLVM21] Apply clang-tidy changes
26 ноя 2025, 17:10
26 ноя 2025, 17:107863ba0
#include <memory>
#include <iomanip>

#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/one/EDAnalyzer.h"

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "JetMETCorrections/Modules/interface/JetResolution.h"

#include "FWCore/ServiceRegistry/interface/Service.h"
#include "CommonTools/UtilAlgos/interface/TFileService.h"

#include <DataFormats/PatCandidates/interface/Jet.h>

#include <TH2.h>

//
// class declaration
//

class JetResolutionDemo : public edm::one::EDAnalyzer<edm::one::SharedResources> {
public:
  explicit JetResolutionDemo(const edm::ParameterSet&);
  ~JetResolutionDemo() override;

private:
  void analyze(const edm::Event&, const edm::EventSetup&) override;
  void endJob() override;

  edm::EDGetTokenT<std::vector<pat::Jet>> m_jets_token;
  edm::EDGetTokenT<double> m_rho_token;

  bool m_debug = false;
  bool m_use_conddb = false;

  std::string m_payload;
  std::string m_resolutions_file;
  std::string m_scale_factors_file;
  JME::JetResolution::Token m_resolutions_token;
  JME::JetResolutionScaleFactor::Token m_scale_factors_token;
  edm::Service<TFileService> fs;

  TH2* m_res_vs_eta = nullptr;
};
//
//----------- Class Implementation ------------------------------------------
//
//---------------------------------------------------------------------------
JetResolutionDemo::JetResolutionDemo(const edm::ParameterSet& iConfig) {
  m_jets_token = consumes<std::vector<pat::Jet>>(iConfig.getParameter<edm::InputTag>("jets"));
  m_rho_token = consumes<double>(iConfig.getParameter<edm::InputTag>("rho"));
  m_debug = iConfig.getUntrackedParameter<bool>("debug", false);
  m_use_conddb = iConfig.getUntrackedParameter<bool>("useCondDB", false);

  if (m_use_conddb) {
    m_payload = iConfig.getParameter<std::string>("payload");
    m_resolutions_token = esConsumes(edm::ESInputTag("", m_payload));
    m_scale_factors_token = esConsumes(edm::ESInputTag("", m_payload));
  } else {
    m_resolutions_file = iConfig.getParameter<edm::FileInPath>("resolutionsFile").fullPath();
    m_scale_factors_file = iConfig.getParameter<edm::FileInPath>("scaleFactorsFile").fullPath();
  }
  usesResource(TFileService::kSharedResource);
}
//---------------------------------------------------------------------------
JetResolutionDemo::~JetResolutionDemo() {}

//---------------------------------------------------------------------------
void JetResolutionDemo::analyze(const edm::Event& iEvent, const edm::EventSetup& iSetup) {
  edm::Handle<std::vector<pat::Jet>> jets;
  iEvent.getByToken(m_jets_token, jets);

  edm::Handle<double> rho;
  iEvent.getByToken(m_rho_token, rho);

  // Access jet resolution and scale factor from the condition database
  // or from text files
  JME::JetResolution resolution;
  JME::JetResolutionScaleFactor res_sf;

  // Two differents way to create a class instance
  if (m_use_conddb) {
    // First way, using the get() static method
    resolution = JME::JetResolution::get(iSetup, m_resolutions_token);
    res_sf = JME::JetResolutionScaleFactor::get(iSetup, m_scale_factors_token);
  } else {
    // Second way, using the constructor
    resolution = JME::JetResolution(m_resolutions_file);
    res_sf = JME::JetResolutionScaleFactor(m_scale_factors_file);
  }

  if (m_debug) {
    // Dump resolution
    resolution.dump();
  }

  if (!m_res_vs_eta) {
    // Advanced usage. Create the histogram by retriving the eta binning directly from the JetResolutionObject. This suppose you need that the parametrization only depends on eta.

    // Get the list of bins of this object
    const std::vector<JME::Binning>& bins = resolution.getResolutionObject()->getDefinition().getBins();

    // Check that the first bin is eta
    if ((!bins.empty()) && (bins[0] == JME::Binning::JetEta)) {
      const std::vector<JME::JetResolutionObject::Record> records = resolution.getResolutionObject()->getRecords();
      // Get all records from the object. Each record correspond to a different binning and different parameters

      std::vector<float> etas;
      for (const auto& record : records) {
        if (etas.empty()) {
          etas.push_back(record.getBinsRange()[0].min);
          etas.push_back(record.getBinsRange()[0].max);
        } else {
          etas.push_back(record.getBinsRange()[0].max);
        }
      }

      std::vector<float> res;
      res.reserve(40);
      for (std::size_t i = 0; i < 40; i++) {
        res.push_back(i * 0.005);
      }

      m_res_vs_eta = fs->make<TH2F>("res_vs_eta", "res_vs_eta", etas.size() - 1, &etas[0], res.size() - 1, &res[0]);
    }
  }

  for (const auto& jet : *jets) {
    if (m_debug) {
      std::cout << "New jet; pt=" << jet.pt() << "  eta=" << jet.eta() << "  phi=" << jet.phi()
                << "  e=" << jet.energy() << "  rho=" << *rho << std::endl;
    }

    // Get resolution for this jet
    // The method to use is getResolution(const JME::JetParameters& parameters) from a JetResolution object

    // You *must* now in advance which variables are needed for getting the resolution. For the moment, only pt and eta are needed, but this will
    // probably change in the futur when PU dependency is added. Please keep an eye on the twiki page
    //     https://twiki.cern.ch/twiki/bin/view/CMSPublic/WorkBookJetEnergyResolution
    // to stay up-to-date. All currently supported parameters (ie, 'set' functions) are available here:
    //     https://twiki.cern.ch/twiki/bin/view/CMSPublic/WorkBookJetEnergyResolution#List_of_supported_parameters

    // Three way to create a JetParameters object

    // First, using 'set' functions
    JME::JetParameters parameters_1;
    parameters_1.setJetPt(jet.pt());
    parameters_1.setJetEta(jet.eta());
    parameters_1.setRho(*rho);

    // You can also chain calls

    JME::JetParameters parameters_2;
    parameters_2.setJetPt(jet.pt()).setJetEta(jet.eta()).setRho(*rho);

    // Second, using the set() function
    JME::JetParameters parameters_3;
    parameters_3.set(JME::Binning::JetPt, jet.pt());
    parameters_3.set({JME::Binning::JetEta, jet.eta()});
    parameters_3.set({JME::Binning::Rho, *rho});

    // Or

    JME::JetParameters parameters_4;
    parameters_4.set(JME::Binning::JetPt, jet.pt()).set(JME::Binning::JetEta, jet.eta()).set(JME::Binning::Rho, *rho);

    // Third, using a initializer_list
    JME::JetParameters parameters_5 = {
        {JME::Binning::JetPt, jet.pt()}, {JME::Binning::JetEta, jet.eta()}, {JME::Binning::Rho, *rho}};

    // Now, get the resolution

    float r = resolution.getResolution(parameters_1);
    if (m_debug) {
      std::cout << "Resolution with parameters_1: " << r << std::endl;
      std::cout << "Resolution with parameters_2: " << resolution.getResolution(parameters_2) << std::endl;
      std::cout << "Resolution with parameters_3: " << resolution.getResolution(parameters_3) << std::endl;
      std::cout << "Resolution with parameters_4: " << resolution.getResolution(parameters_4) << std::endl;
      std::cout << "Resolution with parameters_5: " << resolution.getResolution(parameters_5) << std::endl;

      // You can also use a shortcut to get the resolution
      float r2 = resolution.getResolution(
          {{JME::Binning::JetPt, jet.pt()}, {JME::Binning::JetEta, jet.eta()}, {JME::Binning::Rho, *rho}});
      std::cout << "Resolution using shortcut   : " << r2 << std::endl;
    }

    m_res_vs_eta->Fill(jet.eta(), r);

    // We do the same thing to access the scale factors
    float sf = res_sf.getScaleFactor({{JME::Binning::JetPt, jet.pt()}, {JME::Binning::JetEta, jet.eta()}});

    // Access up and down variation of the scale factor
    float sf_up =
        res_sf.getScaleFactor({{JME::Binning::JetPt, jet.pt()}, {JME::Binning::JetEta, jet.eta()}}, Variation::UP);
    float sf_down =
        res_sf.getScaleFactor({{JME::Binning::JetPt, jet.pt()}, {JME::Binning::JetEta, jet.eta()}}, Variation::DOWN);

    if (m_debug) {
      std::cout << "Scale factors (Nominal / Up / Down) : " << sf << " / " << sf_up << " / " << sf_down << std::endl;
    }
  }
}
//---------------------------------------------------------------------------
void JetResolutionDemo::endJob() {}
//---------------------------------------------------------------------------
//define this as a plug-in
DEFINE_FWK_MODULE(JetResolutionDemo);