PFRecoTauDiscriminationAgainstElectronMVA6 Class Reference

Inheritance diagram for PFRecoTauDiscriminationAgainstElectronMVA6:

Public Member Functions
void	beginEvent (const edm::Event &, const edm::EventSetup &) override
double	discriminate (const PFTauRef &) const override
void	endEvent (edm::Event &) override
	PFRecoTauDiscriminationAgainstElectronMVA6 (const edm::ParameterSet &cfg)
	~PFRecoTauDiscriminationAgainstElectronMVA6 () override
Public Member Functions inherited from TauDiscriminationProducerBase< TauType, TauDiscriminator >
virtual double	discriminate (const TauRef &tau) const =0
void	produce (edm::Event &, const edm::EventSetup &) override
	TauDiscriminationProducerBase (const edm::ParameterSet &iConfig)
	TauDiscriminationProducerBase ()
	~TauDiscriminationProducerBase () override
Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default
bool	hasAbilityToProduceInLumis () const final
bool	hasAbilityToProduceInRuns () const final

Static Public Member Functions
static void	fillDescriptions (edm::ConfigurationDescriptions &descriptions)
Static Public Member Functions inherited from TauDiscriminationProducerBase< TauType, TauDiscriminator >
static void	fillProducerDescriptions (edm::ParameterSetDescription &desc)
static std::string	getTauTypeString ()
	helper method to retrieve tau type name, e.g. to build correct cfi getter More...

Private Member Functions
bool	isInEcalCrack (double) const

Private Attributes
std::unique_ptr< PFTauDiscriminator >	category_output_
edm::Handle< reco::GsfElectronCollection >	gsfElectrons_
edm::EDGetTokenT< reco::GsfElectronCollection >	GsfElectrons_token
std::string	moduleLabel_
std::unique_ptr< AntiElectronIDMVA6 >	mva_
edm::InputTag	srcGsfElectrons_
edm::Handle< TauCollection >	taus_
int	verbosity_
bool	vetoEcalCracks_

Additional Inherited Members
Public Types inherited from TauDiscriminationProducerBase< TauType, TauDiscriminator >
typedef std::vector< TauType >	TauCollection
typedef edm::Ref< TauCollection >	TauRef
typedef edm::RefProd< TauCollection >	TauRefProd
Public Types inherited from edm::stream::EDProducer<>
typedef CacheContexts< T... >	CacheTypes
typedef CacheTypes::GlobalCache	GlobalCache
typedef AbilityChecker< T... >	HasAbility
typedef CacheTypes::LuminosityBlockCache	LuminosityBlockCache
typedef LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >	LuminosityBlockContext
typedef CacheTypes::LuminosityBlockSummaryCache	LuminosityBlockSummaryCache
typedef CacheTypes::RunCache	RunCache
typedef RunContextT< RunCache, GlobalCache >	RunContext
typedef CacheTypes::RunSummaryCache	RunSummaryCache
Protected Attributes inherited from TauDiscriminationProducerBase< TauType, TauDiscriminator >
std::string	moduleLabel_
double	prediscriminantFailValue_
edm::EDGetTokenT< TauCollection >	Tau_token
size_t	tauIndex_
edm::InputTag	TauProducer_

Detailed Description

Definition at line 30 of file PFRecoTauDiscriminationAgainstElectronMVA6.cc.

Constructor & Destructor Documentation

PFRecoTauDiscriminationAgainstElectronMVA6::PFRecoTauDiscriminationAgainstElectronMVA6 ( const edm::ParameterSet &  cfg )

inlineexplicit

Definition at line 33 of file PFRecoTauDiscriminationAgainstElectronMVA6.cc.

References looper::cfg, and edm::ParameterSet::getParameter().

    : PFTauDiscriminationProducerBase(cfg),
      mva_(),
      category_output_()
  {
    mva_ = std::make_unique<AntiElectronIDMVA6>(cfg);

    srcGsfElectrons_ = cfg.getParameter<edm::InputTag>("srcGsfElectrons");
    GsfElectrons_token = consumes<reco::GsfElectronCollection>(srcGsfElectrons_);
    vetoEcalCracks_ = cfg.getParameter<bool>("vetoEcalCracks");

    verbosity_ = cfg.getParameter<int>("verbosity");

    // add category index
    produces<PFTauDiscriminator>("category");
  }

PFRecoTauDiscriminationAgainstElectronMVA6::~PFRecoTauDiscriminationAgainstElectronMVA6 ( )

inlineoverride

Definition at line 56 of file PFRecoTauDiscriminationAgainstElectronMVA6.cc.

References fillDescriptions().

{}

Member Function Documentation

void PFRecoTauDiscriminationAgainstElectronMVA6::beginEvent ( const edm::Event &  evt, const edm::EventSetup &  es  )

overridevirtual

Reimplemented from TauDiscriminationProducerBase< TauType, TauDiscriminator >.

Definition at line 78 of file PFRecoTauDiscriminationAgainstElectronMVA6.cc.

References edm::Event::getByToken().

{
  mva_->beginEvent(evt, es);

  evt.getByToken(Tau_token, taus_);
  category_output_.reset(new PFTauDiscriminator(TauRefProd(taus_)));

  evt.getByToken(GsfElectrons_token, gsfElectrons_);
}

double PFRecoTauDiscriminationAgainstElectronMVA6::discriminate ( const PFTauRef &  thePFTauRef ) const

override

Definition at line 88 of file PFRecoTauDiscriminationAgainstElectronMVA6.cc.

References funct::abs(), taus_updatedMVAIds_cff::category, reco::deltaR(), reco::TrackBase::eta(), reco::PFCandidate::gsfTrackRef(), edm::Ptr< T >::isNonnull(), edm::Ref< C, T, F >::isNonnull(), SiStripPI::max, min(), boostedElectronIsolation_cff::pfGamma, reco::TrackBase::pt(), and HiIsolationCommonParameters_cff::track.

{
  double mvaValue = 1.;
  double category = -1.;
  bool isGsfElectronMatched = false;

  float deltaRDummy = 9.9;

  const float ECALBarrelEndcapEtaBorder = 1.479;
  float tauEtaAtEcalEntrance = -99.;
  float sumEtaTimesEnergy = 0.;
  float sumEnergy = 0.;
  for ( const auto & pfCandidate : thePFTauRef->signalPFCands() ) {
    sumEtaTimesEnergy += (pfCandidate->positionAtECALEntrance().eta()*pfCandidate->energy());
    sumEnergy += pfCandidate->energy();
  }
  if ( sumEnergy > 0. ) {
    tauEtaAtEcalEntrance = sumEtaTimesEnergy/sumEnergy;
  }

  float leadChargedPFCandEtaAtEcalEntrance = -99.;
  float leadChargedPFCandPt = -99.;
  for ( const auto & pfCandidate : thePFTauRef->signalPFCands() ) {
    const reco::Track* track = nullptr;
    if ( pfCandidate->trackRef().isNonnull() ) track = pfCandidate->trackRef().get();
    else if ( pfCandidate->muonRef().isNonnull() && pfCandidate->muonRef()->innerTrack().isNonnull()  ) track = pfCandidate->muonRef()->innerTrack().get();
    else if ( pfCandidate->muonRef().isNonnull() && pfCandidate->muonRef()->globalTrack().isNonnull() ) track = pfCandidate->muonRef()->globalTrack().get();
    else if ( pfCandidate->muonRef().isNonnull() && pfCandidate->muonRef()->outerTrack().isNonnull()  ) track = pfCandidate->muonRef()->outerTrack().get();
    else if ( pfCandidate->gsfTrackRef().isNonnull() ) track = pfCandidate->gsfTrackRef().get();
    if ( track ) {
      if ( track->pt() > leadChargedPFCandPt ) {
    leadChargedPFCandEtaAtEcalEntrance = pfCandidate->positionAtECALEntrance().eta();
    leadChargedPFCandPt = track->pt();
      }
    }
  }

  if( (*thePFTauRef).leadChargedHadrCand().isNonnull()) {

    int numSignalGammaCandsInSigCone = 0;
    const std::vector<reco::CandidatePtr>& signalGammaCands = thePFTauRef->signalGammaCands();
    
    for ( const auto & pfGamma : signalGammaCands ) {
            
      double dR = deltaR(pfGamma->p4(), thePFTauRef->leadChargedHadrCand()->p4());
      double signalrad = std::max(0.05, std::min(0.10, 3.0/std::max(1.0, thePFTauRef->pt())));
            
      // pfGammas inside the tau signal cone
      if (dR < signalrad) {
        numSignalGammaCandsInSigCone += 1;
      }
    }
    
    // loop over the electrons
    for ( const auto & theGsfElectron : *gsfElectrons_ ) {
      if ( theGsfElectron.pt() > 10. ) { // CV: only take electrons above some minimal energy/Pt into account...
    double deltaREleTau = deltaR(theGsfElectron.p4(), thePFTauRef->p4());
    deltaRDummy = deltaREleTau;
    if ( deltaREleTau < 0.3 ) {
      double mva_match = mva_->MVAValue(*thePFTauRef, theGsfElectron);
      const reco::PFCandidatePtr& lpfch = thePFTauRef->leadPFChargedHadrCand();
      bool hasGsfTrack = false;
      if (lpfch.isNonnull()) {
        hasGsfTrack = lpfch->gsfTrackRef().isNonnull();
      }
      if ( !hasGsfTrack )
            hasGsfTrack = theGsfElectron.gsfTrack().isNonnull();

      if ( vetoEcalCracks_ && (isInEcalCrack(tauEtaAtEcalEntrance) || isInEcalCrack(leadChargedPFCandEtaAtEcalEntrance)) ) {
        // add category index
        category_output_->setValue(tauIndex_, category);
        // return MVA output value
        return -99;
      }

      if ( std::abs(tauEtaAtEcalEntrance) < ECALBarrelEndcapEtaBorder ) { // Barrel
        if ( numSignalGammaCandsInSigCone == 0 && hasGsfTrack ) {
          category = 5.;
        }
        else if ( numSignalGammaCandsInSigCone >= 1 && hasGsfTrack ) {
          category = 7.;
        }
      } else { // Endcap
        if ( numSignalGammaCandsInSigCone == 0 && hasGsfTrack ) {
          category = 13.;
        }
        else if ( numSignalGammaCandsInSigCone >= 1 && hasGsfTrack ) {
          category = 15.;
        }
      }

      mvaValue = std::min(mvaValue, mva_match);
      isGsfElectronMatched = true;
    } // deltaR < 0.3
      } // electron pt > 10
    } // end of loop over electrons

    if ( !isGsfElectronMatched ) {
      mvaValue = mva_->MVAValue(*thePFTauRef);
      const reco::PFCandidatePtr& lpfch = thePFTauRef->leadPFChargedHadrCand();
      bool hasGsfTrack = false;
      if (lpfch.isNonnull()) {
    hasGsfTrack = lpfch->gsfTrackRef().isNonnull();
      }
      
      if ( vetoEcalCracks_ && (isInEcalCrack(tauEtaAtEcalEntrance) || isInEcalCrack(leadChargedPFCandEtaAtEcalEntrance)) ) {
    // add category index
    category_output_->setValue(tauIndex_, category);
    // return MVA output value
    return -99;
      }
      
      if ( std::abs(tauEtaAtEcalEntrance) < ECALBarrelEndcapEtaBorder ) { // Barrel
    if ( numSignalGammaCandsInSigCone == 0 && !hasGsfTrack ) {
      category = 0.;
    }
    else if ( numSignalGammaCandsInSigCone >= 1 && !hasGsfTrack ) {
      category = 2.;
    }
      } else { // Endcap
    if ( numSignalGammaCandsInSigCone == 0 && !hasGsfTrack ) {
      category = 8.;
    }
    else if ( numSignalGammaCandsInSigCone >= 1 && !hasGsfTrack ) {
      category = 10.;
    }
      }
    }
  }

  if ( verbosity_ ) {
    edm::LogPrint("PFTauAgainstEleMVA6") << "<PFRecoTauDiscriminationAgainstElectronMVA6::discriminate>:" ;
    edm::LogPrint("PFTauAgainstEleMVA6") << " tau: Pt = " << thePFTauRef->pt() << ", eta = " << thePFTauRef->eta() << ", phi = " << thePFTauRef->phi();
    edm::LogPrint("PFTauAgainstEleMVA6") << " deltaREleTau = " << deltaRDummy << ", isGsfElectronMatched = " << isGsfElectronMatched;
    edm::LogPrint("PFTauAgainstEleMVA6") << " #Prongs = " << thePFTauRef->signalChargedHadrCands().size();
    edm::LogPrint("PFTauAgainstEleMVA6") << " MVA = " << mvaValue << ", category = " << category;
  }

  // add category index
  category_output_->setValue(tauIndex_, category);
  // return MVA output value
  return mvaValue;
}

void PFRecoTauDiscriminationAgainstElectronMVA6::endEvent ( edm::Event &  evt )

overridevirtual

Reimplemented from TauDiscriminationProducerBase< TauType, TauDiscriminator >.

Definition at line 236 of file PFRecoTauDiscriminationAgainstElectronMVA6.cc.

References eostools::move(), and edm::Event::put().

{
  // add all category indices to event
  evt.put(std::move(category_output_), "category");
}

void PFRecoTauDiscriminationAgainstElectronMVA6::fillDescriptions ( edm::ConfigurationDescriptions &  descriptions )

static

Definition at line 250 of file PFRecoTauDiscriminationAgainstElectronMVA6.cc.

References edm::ConfigurationDescriptions::add(), edm::ParameterSetDescription::add(), edm::ParameterSetDescription::addOptional(), DEFINE_FWK_MODULE, and AlCaHLTBitMon_QueryRunRegistry::string.

                                                                                                       {
  // pfRecoTauDiscriminationAgainstElectronMVA6
  edm::ParameterSetDescription desc;
  desc.add<double>("minMVANoEleMatchWOgWOgsfBL", 0.0);
  desc.add<edm::InputTag>("PFTauProducer", edm::InputTag("pfTauProducer"));
  desc.add<double>("minMVANoEleMatchWgWOgsfBL", 0.0);
  desc.add<std::string>("mvaName_wGwGSF_EC", "gbr_wGwGSF_EC");
  desc.add<double>("minMVAWgWgsfBL", 0.0);
  desc.add<std::string>("mvaName_woGwGSF_EC", "gbr_woGwGSF_EC");
  desc.add<double>("minMVAWOgWgsfEC", 0.0);
  desc.add<std::string>("mvaName_wGwGSF_BL", "gbr_wGwGSF_BL");
  desc.add<std::string>("mvaName_woGwGSF_BL", "gbr_woGwGSF_BL");
  desc.add<bool>("returnMVA", true);
  desc.add<bool>("loadMVAfromDB", true);
  {
    edm::ParameterSetDescription pset_Prediscriminants;
    pset_Prediscriminants.add<std::string>("BooleanOperator", "and");
    {
      edm::ParameterSetDescription psd1;
      psd1.add<double>("cut");
      psd1.add<edm::InputTag>("Producer");
      pset_Prediscriminants.addOptional<edm::ParameterSetDescription>("leadTrack", psd1);
    }
    {
      // encountered this at
      // RecoTauTag/Configuration/python/HPSPFTaus_cff.py
      edm::ParameterSetDescription psd1;
      psd1.add<double>("cut");
      psd1.add<edm::InputTag>("Producer");
      pset_Prediscriminants.addOptional<edm::ParameterSetDescription>("decayMode", psd1);
    }
    desc.add<edm::ParameterSetDescription>("Prediscriminants", pset_Prediscriminants);
  }
  desc.add<std::string>("mvaName_NoEleMatch_woGwoGSF_BL", "gbr_NoEleMatch_woGwoGSF_BL");
  desc.add<bool>("vetoEcalCracks", true);
  desc.add<bool>("usePhiAtEcalEntranceExtrapolation", false);
  desc.add<std::string>("mvaName_NoEleMatch_wGwoGSF_BL", "gbr_NoEleMatch_wGwoGSF_BL");
  desc.add<double>("minMVANoEleMatchWOgWOgsfEC", 0.0);
  desc.add<double>("minMVAWOgWgsfBL", 0.0);
  desc.add<double>("minMVAWgWgsfEC", 0.0);
  desc.add<int>("verbosity", 0);
  desc.add<std::string>("mvaName_NoEleMatch_wGwoGSF_EC", "gbr_NoEleMatch_wGwoGSF_EC");
  desc.add<std::string>("method", "BDTG");
  desc.add<edm::InputTag>("srcGsfElectrons", edm::InputTag("gedGsfElectrons"));
  desc.add<std::string>("mvaName_NoEleMatch_woGwoGSF_EC", "gbr_NoEleMatch_woGwoGSF_EC");
  desc.add<double>("minMVANoEleMatchWgWOgsfEC", 0.0);
  descriptions.add("pfRecoTauDiscriminationAgainstElectronMVA6", desc);
}

bool PFRecoTauDiscriminationAgainstElectronMVA6::isInEcalCrack ( double  eta ) const

private

Definition at line 243 of file PFRecoTauDiscriminationAgainstElectronMVA6.cc.

{
  double absEta = fabs(eta);
  return (absEta > 1.460 && absEta < 1.558);
}

Member Data Documentation

std::unique_ptr<PFTauDiscriminator> PFRecoTauDiscriminationAgainstElectronMVA6::category_output_

private

Definition at line 71 of file PFRecoTauDiscriminationAgainstElectronMVA6.cc.

edm::Handle<reco::GsfElectronCollection> PFRecoTauDiscriminationAgainstElectronMVA6::gsfElectrons_

private

Definition at line 68 of file PFRecoTauDiscriminationAgainstElectronMVA6.cc.

edm::EDGetTokenT<reco::GsfElectronCollection> PFRecoTauDiscriminationAgainstElectronMVA6::GsfElectrons_token

private

Definition at line 67 of file PFRecoTauDiscriminationAgainstElectronMVA6.cc.

std::string PFRecoTauDiscriminationAgainstElectronMVA6::moduleLabel_

private

Definition at line 63 of file PFRecoTauDiscriminationAgainstElectronMVA6.cc.

Referenced by ExternalGeneratorFilter.ExternalGeneratorFilter::insertInto(), Modules.SwitchProducer::insertInto(), Mixins._TypedParameterizable::insertInto(), Modules.ESSource::nameInProcessDesc_(), and Modules.ESProducer::nameInProcessDesc_().

std::unique_ptr<AntiElectronIDMVA6> PFRecoTauDiscriminationAgainstElectronMVA6::mva_

private

Definition at line 64 of file PFRecoTauDiscriminationAgainstElectronMVA6.cc.

edm::InputTag PFRecoTauDiscriminationAgainstElectronMVA6::srcGsfElectrons_

private

Definition at line 66 of file PFRecoTauDiscriminationAgainstElectronMVA6.cc.

edm::Handle<TauCollection> PFRecoTauDiscriminationAgainstElectronMVA6::taus_

private

Definition at line 69 of file PFRecoTauDiscriminationAgainstElectronMVA6.cc.

int PFRecoTauDiscriminationAgainstElectronMVA6::verbosity_

private

Definition at line 75 of file PFRecoTauDiscriminationAgainstElectronMVA6.cc.

bool PFRecoTauDiscriminationAgainstElectronMVA6::vetoEcalCracks_

private

Definition at line 73 of file PFRecoTauDiscriminationAgainstElectronMVA6.cc.