RecoTauTauTagInfoWorkaroundModifier.cc

Go to the documentation of this file.

/*
 * ===========================================================================
 *
 *       Filename:  RecoTauTagInfoWorkaroundModifer
 *
 *    Description:  Add the PFTauTagInfoRef back to PFTaus so things don't
 *                  break.
 *
 *         Author:  Evan K. Friis (UC Davis)
 *
 * ===========================================================================
 */

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

namespace reco { namespace tau {

class RecoTauTagInfoWorkaroundModifer : public RecoTauModifierPlugin {
  public:
  explicit RecoTauTagInfoWorkaroundModifer(const edm::ParameterSet &pset, edm::ConsumesCollector &&iC);
    ~RecoTauTagInfoWorkaroundModifer() override {}
    void operator()(PFTau&) const override;
    // Called by base class
    void beginEvent() override;
  private:
    edm::InputTag pfTauTagInfoSrc_;
    edm::Handle<PFTauTagInfoCollection> infos_;
  edm::EDGetTokenT<PFTauTagInfoCollection> pfTauTagInfo_token;
};

RecoTauTagInfoWorkaroundModifer::RecoTauTagInfoWorkaroundModifer(
   const edm::ParameterSet &pset, edm::ConsumesCollector &&iC):RecoTauModifierPlugin(pset,std::move(iC)) {
  pfTauTagInfoSrc_ = pset.getParameter<edm::InputTag>("pfTauTagInfoSrc");
  pfTauTagInfo_token = iC.consumes<PFTauTagInfoCollection>(pfTauTagInfoSrc_);
}

// Load our tau tag infos from the event
void RecoTauTagInfoWorkaroundModifer::beginEvent() {

  evt()->getByToken(pfTauTagInfo_token, infos_);
}

void RecoTauTagInfoWorkaroundModifer::operator()(PFTau& tau) const {
  // Find the PFTauTagInfo that comes from the same PFJet
  JetBaseRef tauJetRef = tau.jetRef();
  for(size_t iInfo = 0; iInfo < infos_->size(); ++iInfo) {
    // Get jet ref from tau tag info
    PFTauTagInfoRef infoRef = PFTauTagInfoRef(infos_, iInfo);
    JetBaseRef infoJetRef = infoRef->pfjetRef();
    // Check if they come from the same jet
    if (infoJetRef == tauJetRef) {
      // This tau "comes" from this PFJetRef
      tau.setpfTauTagInfoRef(infoRef);
      break;
    }
  }
}
}}  // end namespace reco::tau
#include "FWCore/Framework/interface/MakerMacros.h"
DEFINE_EDM_PLUGIN(RecoTauModifierPluginFactory,
    reco::tau::RecoTauTagInfoWorkaroundModifer,
    "RecoTauTagInfoWorkaroundModifer");