PFJetMonitor.h

Go to the documentation of this file.

#ifndef DQMOffline_PFTau_PFJetMonitor_h
#define DQMOffline_PFTau_PFJetMonitor_h

#include "DQMOffline/PFTau/interface/Benchmark.h"
#include "DQMOffline/PFTau/interface/CandidateBenchmark.h"
#include "DQMOffline/PFTau/interface/MatchCandidateBenchmark.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "DataFormats/JetReco/interface/BasicJetCollection.h"

#include <vector>

#include <TH1.h>  //needed by the deltaR->Fill() call

class PFJetMonitor : public Benchmark {
public:
  PFJetMonitor(float dRMax = 0.3, bool matchCharge = true, Benchmark::Mode mode = Benchmark::DEFAULT);

  ~PFJetMonitor() override;

  void setParameters(float dRMax,
                     bool matchCharge,
                     Benchmark::Mode mode,
                     float ptmin,
                     float ptmax,
                     float etamin,
                     float etamax,
                     float phimin,
                     float phimax,
                     bool fracHistoFlag = true);

  void setParameters(float dRMax,
                     bool onlyTwoJets,
                     bool matchCharge,
                     Benchmark::Mode mode,
                     float ptmin,
                     float ptmax,
                     float etamin,
                     float etamax,
                     float phimin,
                     float phimax,
                     bool fracHistoFlag = true);

  void setParameters(const edm::ParameterSet &parameterSet);

  void setDirectory(TDirectory *dir) override;

  void setup(DQMStore::IBooker &b);
  void setup(DQMStore::IBooker &b, const edm::ParameterSet &parameterSet);

  template <class T, class C>
  void fill(const T &jetCollection, const C &matchedJetCollection, float &minVal, float &maxVal);

  template <class T, class C>
  void fill(const T &candidateCollection,
            const C &matchedCandCollection,
            float &minVal,
            float &maxVal,
            float &jetpT,
            const edm::ParameterSet &parameterSet);

  void fillOne(const reco::Jet &jet, const reco::Jet &matchedJet);

protected:
  CandidateBenchmark candBench_;
  MatchCandidateBenchmark matchCandBench_;

  TH2F *delta_frac_VS_frac_muon_;
  TH2F *delta_frac_VS_frac_photon_;
  TH2F *delta_frac_VS_frac_electron_;
  TH2F *delta_frac_VS_frac_charged_hadron_;
  TH2F *delta_frac_VS_frac_neutral_hadron_;

  TH1F *deltaR_;
  float dRMax_;
  bool onlyTwoJets_;
  bool matchCharge_;
  bool createPFractionHistos_;
  bool histogramBooked_;
};

#include "DQMOffline/PFTau/interface/Matchers.h"
template <class T, class C>
void PFJetMonitor::fill(const T &jetCollection, const C &matchedJetCollection, float &minVal, float &maxVal) {
  std::vector<int> matchIndices;
  PFB::match(jetCollection, matchedJetCollection, matchIndices, matchCharge_, dRMax_);

  for (unsigned int i = 0; i < (jetCollection).size(); i++) {
    const reco::Jet &jet = jetCollection[i];

    if (!isInRange(jet.pt(), jet.eta(), jet.phi()))
      continue;

    int iMatch = matchIndices[i];
    assert(iMatch < static_cast<int>(matchedJetCollection.size()));

    if (iMatch != -1) {
      const reco::Candidate &matchedJet = matchedJetCollection[iMatch];
      if (!isInRange(matchedJet.pt(), matchedJet.eta(), matchedJet.phi()))
        continue;
      float ptRes = (jet.pt() - matchedJet.pt()) / matchedJet.pt();

      if (ptRes > maxVal)
        maxVal = ptRes;
      if (ptRes < minVal)
        minVal = ptRes;

      candBench_.fillOne(jet);
      matchCandBench_.fillOne(jet, matchedJetCollection[iMatch]);
      if (createPFractionHistos_ && histogramBooked_)
        fillOne(jet, matchedJetCollection[iMatch]);
    }
  }
}

template <class T, class C>
void PFJetMonitor::fill(const T &jetCollection,
                        const C &matchedJetCollection,
                        float &minVal,
                        float &maxVal,
                        float &jetpT,
                        const edm::ParameterSet &parameterSet) {
  std::vector<int> matchIndices;
  PFB::match(jetCollection, matchedJetCollection, matchIndices, matchCharge_, dRMax_);
  // now matchIndices[i] stores the j-th closest matched jet

  for (unsigned i = 0; i < jetCollection.size(); ++i) {
    // Count the number of jets with a larger energy = pT
    unsigned int highJets = 0;
    for (unsigned j = 0; j < jetCollection.size(); ++j) {
      if (j != i && jetCollection[j].pt() > jetCollection[i].pt())
        highJets++;
    }
    if (onlyTwoJets_ && highJets > 1)
      continue;

    const reco::Jet &jet = jetCollection[i];

    if (!isInRange(jet.pt(), jet.eta(), jet.phi()))
      continue;

    int iMatch = matchIndices[i];
    assert(iMatch < static_cast<int>(matchedJetCollection.size()));

    if (iMatch != -1) {
      const reco::Candidate &matchedJet = matchedJetCollection[iMatch];
      if (!isInRange(matchedJet.pt(), matchedJet.eta(), matchedJet.phi()))
        continue;

      float ptRes = (jet.pt() - matchedJet.pt()) / matchedJet.pt();

      jetpT = jet.pt();
      if (ptRes > maxVal)
        maxVal = ptRes;
      if (ptRes < minVal)
        minVal = ptRes;

      candBench_.fillOne(jet);  // fill pt eta phi and charge histos for MATCHED candidate jet
      matchCandBench_.fillOne(jet,
                              matchedJetCollection[iMatch],
                              parameterSet);  // fill delta_x_VS_y histos for matched couple
      if (createPFractionHistos_ && histogramBooked_)
        fillOne(jet,
                matchedJetCollection[iMatch]);  // book and fill delta_frac_VS_frac
                                                // histos for matched couple
    }

    for (unsigned j = 0; j < matchedJetCollection.size(); ++j)  // for DeltaR spectrum
      if (deltaR_)
        deltaR_->Fill(reco::deltaR(jetCollection[i], matchedJetCollection[j]));
  }  // end loop on jetCollection
}
#endif