RecoTauDecayModeTruthMatchPlugin.cc

Go to the documentation of this file.

/*
 * RecoTauDecayModeTruthMatchPlugin
 *
 * Author: Evan K. Friis, UC Davis
 *
 * Implements a RecoTauCleaner plugin that returns the difference
 * between the reconstructed decay mode and true decay mode index.
 *
 * By requiring the return value to be zero one can select reco taus
 * that have the decay mode correctly reconstructed.
 *
 */

#include "RecoTauTag/RecoTau/interface/RecoTauBuilderPlugins.h"

#include "DataFormats/TauReco/interface/PFTau.h"
#include "DataFormats/TauReco/interface/PFTauFwd.h"
#include "DataFormats/JetReco/interface/GenJet.h"
#include "DataFormats/JetReco/interface/GenJetCollection.h"
#include "DataFormats/Common/interface/Association.h"

#include "RecoTauTag/RecoTau/interface/PFTauDecayModeTools.h"

namespace tautools {

class RecoTauDecayModeTruthMatchPlugin : public reco::tau::RecoTauCleanerPlugin
{
  public:
  explicit RecoTauDecayModeTruthMatchPlugin(const edm::ParameterSet& pset, edm::ConsumesCollector &&iC);
    ~RecoTauDecayModeTruthMatchPlugin() override {}
    double operator()(const reco::PFTauRef&) const override;
    void beginEvent() override;

  private:
    edm::InputTag matchingSrc_;
    typedef edm::Association<reco::GenJetCollection> GenJetAssociation;
    edm::Handle<GenJetAssociation> genTauMatch_;
};

// ctor
RecoTauDecayModeTruthMatchPlugin::RecoTauDecayModeTruthMatchPlugin(
                                   const edm::ParameterSet& pset, edm::ConsumesCollector &&iC): RecoTauCleanerPlugin(pset,std::move(iC)),
  matchingSrc_(pset.getParameter<edm::InputTag>("matching")) {}

// Called by base class at the beginning of each event
void RecoTauDecayModeTruthMatchPlugin::beginEvent() {
  // Load the matching information
  evt()->getByLabel(matchingSrc_, genTauMatch_);
}

// Determine a number giving the quality of the input tau.  Lower numbers are
// better - zero indicates that the reco decay mode matches the truth.
double RecoTauDecayModeTruthMatchPlugin::operator()(const reco::PFTauRef& tau)
  const {
  GenJetAssociation::reference_type truth = (*genTauMatch_)[tau];
  // Check if the matching exists, if not return +infinity
  if (truth.isNull())
    return std::numeric_limits<double>::infinity();
  // Get the difference in decay mode.  The closer to zero, the more the decay
  // mode is matched.
  return std::abs(
      reco::tau::getDecayMode(truth.get()) - tau->decayMode());
}

} // end tautools namespace

#include "FWCore/Framework/interface/MakerMacros.h"
DEFINE_EDM_PLUGIN(RecoTauCleanerPluginFactory, tautools::RecoTauDecayModeTruthMatchPlugin, "RecoTauDecayModeTruthMatchPlugin");