PFCandidateManager.h

Go to the documentation of this file.

#ifndef RecoParticleFlow_Benchmark_BenchmarkManager_h
#define RecoParticleFlow_Benchmark_BenchmarkManager_h
 
#include "DQMOffline/PFTau/interface/Benchmark.h"
#include "DQMOffline/PFTau/interface/CandidateBenchmark.h"
#include "DQMOffline/PFTau/interface/MatchCandidateBenchmark.h"
#include "DQMOffline/PFTau/interface/PFCandidateBenchmark.h"
 
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidate.h"
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidateFwd.h"
 
#include <vector>
 
class PFCandidateManager : public Benchmark {
public:
  PFCandidateManager(float dRMax = 0.3, bool matchCharge = true, Benchmark::Mode mode = Benchmark::DEFAULT)
      : Benchmark(mode),
        candBench_(mode),
        pfCandBench_(mode),
        matchCandBench_(mode),
        dRMax_(dRMax),
        matchCharge_(matchCharge) {}
 
  ~PFCandidateManager() override;
 
  void setParameters(float dRMax = 0.3, bool matchCharge = true, Benchmark::Mode mode = Benchmark::DEFAULT);
 
  void setDirectory(TDirectory *dir) override;
 
  void setup(DQMStore::IBooker &b);
 
  template <class C>
  void fill(const reco::PFCandidateCollection &candCollection, const C &matchedCandCollection);
 
protected:
  CandidateBenchmark candBench_;
  PFCandidateBenchmark pfCandBench_;
  MatchCandidateBenchmark matchCandBench_;
 
  float dRMax_;
  bool matchCharge_;
};
 
#include "DQMOffline/PFTau/interface/Matchers.h"
 
template <class C>
void PFCandidateManager::fill(const reco::PFCandidateCollection &candCollection, const C &matchCandCollection) {
  std::vector<int> matchIndices;
  PFB::match(candCollection, matchCandCollection, matchIndices, matchCharge_, dRMax_);
 
  for (unsigned int i = 0; i < candCollection.size(); i++) {
    const reco::PFCandidate &cand = candCollection[i];
 
    if (!isInRange(cand.pt(), cand.eta(), cand.phi()))
      continue;
 
    int iMatch = matchIndices[i];
 
    assert(iMatch < static_cast<int>(matchCandCollection.size()));
 
    // COLIN how to handle efficiency plots?
 
    // filling the histograms in CandidateBenchmark only in case
    // of a matching.
    if (iMatch != -1) {
      candBench_.fillOne(cand);
      pfCandBench_.fillOne(cand);
      matchCandBench_.fillOne(cand, matchCandCollection[iMatch]);
    }
  }
}
 
#endif