PFRecoTauDiscriminationAgainstElectronMVA5 Class Reference

Inheritance diagram for PFRecoTauDiscriminationAgainstElectronMVA5:

Public Member Functions
void	beginEvent (const edm::Event &, const edm::EventSetup &)
double	discriminate (const PFTauRef &) const
void	endEvent (edm::Event &)
	PFRecoTauDiscriminationAgainstElectronMVA5 (const edm::ParameterSet &cfg)
	~PFRecoTauDiscriminationAgainstElectronMVA5 ()
Public Member Functions inherited from TauDiscriminationProducerBase< TauType, TauDiscriminator >
virtual double	discriminate (const TauRef &tau) const =0
void	produce (edm::Event &, const edm::EventSetup &)
	TauDiscriminationProducerBase (const edm::ParameterSet &iConfig)
	TauDiscriminationProducerBase ()
virtual	~TauDiscriminationProducerBase ()
Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default
Public Member Functions inherited from edm::stream::EDProducerBase
	EDProducerBase ()
ModuleDescription const &	moduleDescription () const
virtual	~EDProducerBase ()
Public Member Functions inherited from edm::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)
	ProducerBase ()
void	registerProducts (ProducerBase *, ProductRegistry *, ModuleDescription const &)
std::function< void(BranchDescription const &)>	registrationCallback () const
	used by the fwk to register list of products More...
virtual	~ProducerBase ()
Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const
	EDConsumerBase ()
ProductHolderIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const
void	itemsMayGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const
void	itemsToGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const
std::vector < ProductHolderIndexAndSkipBit > const &	itemsToGetFromEvent () const
void	labelsForToken (EDGetToken iToken, Labels &oLabels) const
void	modulesDependentUpon (const std::string &iProcessName, std::vector< const char * > &oModuleLabels) const
void	modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const
bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const
bool	registeredToConsumeMany (TypeID const &, BranchType) const
void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)
virtual	~EDConsumerBase ()

Private Member Functions
bool	isInEcalCrack (double) const

Private Attributes
std::auto_ptr< PFTauDiscriminator >	category_output_
edm::Handle < reco::GsfElectronCollection >	gsfElectrons_
edm::EDGetTokenT < reco::GsfElectronCollection >	GsfElectrons_token
std::string	moduleLabel_
AntiElectronIDMVA5 *	mva_
float *	mvaInput_
edm::InputTag	srcGsfElectrons_
edm::Handle< TauCollection >	taus_
int	verbosity_

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
Public Types inherited from edm::stream::EDProducerBase
typedef EDProducerAdaptorBase	ModuleType
Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList
Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels
Static Public Member Functions inherited from edm::stream::EDProducerBase
static const std::string &	baseType ()
static void	fillDescriptions (ConfigurationDescriptions &descriptions)
static void	prevalidate (ConfigurationDescriptions &descriptions)
Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)
EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)
template<BranchType B>
EDGetToken	consumes (TypeToGet const &id, edm::InputTag const &tag)
ConsumesCollector	consumesCollector ()
	Use a ConsumesCollector to gather consumes information from helper functions. More...
template<typename ProductType , BranchType B = InEvent>
void	consumesMany ()
void	consumesMany (const TypeToGet &id)
template<BranchType B>
void	consumesMany (const TypeToGet &id)
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)
template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)
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 29 of file PFRecoTauDiscriminationAgainstElectronMVA5.cc.

Constructor & Destructor Documentation

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

inlineexplicit

Definition at line 32 of file PFRecoTauDiscriminationAgainstElectronMVA5.cc.

References edm::ParameterSet::exists(), and edm::ParameterSet::getParameter().

    : PFTauDiscriminationProducerBase(cfg),
      mva_(0),
      category_output_(0)
  {
    mva_ = new AntiElectronIDMVA5(cfg);

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

    verbosity_ = ( cfg.exists("verbosity") ) ?
      cfg.getParameter<int>("verbosity") : 0;

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

PFRecoTauDiscriminationAgainstElectronMVA5::~PFRecoTauDiscriminationAgainstElectronMVA5 ( )

inline

Definition at line 55 of file PFRecoTauDiscriminationAgainstElectronMVA5.cc.

  {
    delete mva_;
  }

Member Function Documentation

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

virtual

Reimplemented from TauDiscriminationProducerBase< TauType, TauDiscriminator >.

Definition at line 78 of file PFRecoTauDiscriminationAgainstElectronMVA5.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 PFRecoTauDiscriminationAgainstElectronMVA5::discriminate

(

const PFTauRef &

thePFTauRef

)

const

Definition at line 88 of file PFRecoTauDiscriminationAgainstElectronMVA5.cc.

References Abs(), python.rootplot.argparse::category, reco::deltaR(), reco::TrackBase::eta(), edm::Ref< C, T, F >::isNonnull(), Min(), and reco::TrackBase::pt().

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

  float deltaRDummy = 9.9;

  float tauEtaAtEcalEntrance = -99.;
  float sumEtaTimesEnergy = 0.;
  float sumEnergy = 0.;
  const std::vector<reco::PFCandidatePtr>& signalPFCands = thePFTauRef->signalPFCands();
  for ( std::vector<reco::PFCandidatePtr>::const_iterator pfCandidate = signalPFCands.begin();
    pfCandidate != signalPFCands.end(); ++pfCandidate ) {
    sumEtaTimesEnergy += ((*pfCandidate)->positionAtECALEntrance().eta()*(*pfCandidate)->energy());
    sumEnergy += (*pfCandidate)->energy();
  }
  if ( sumEnergy > 0. ) {
    tauEtaAtEcalEntrance = sumEtaTimesEnergy/sumEnergy;
  }

  float leadChargedPFCandEtaAtEcalEntrance = -99.;
  float leadChargedPFCandPt = -99.;
  for ( std::vector<reco::PFCandidatePtr>::const_iterator pfCandidate = signalPFCands.begin();
    pfCandidate != signalPFCands.end(); ++pfCandidate ) {
    const reco::Track* track = 0;
    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).leadPFChargedHadrCand().isNonnull()) {
    for ( reco::GsfElectronCollection::const_iterator theGsfElectron = gsfElectrons_->begin();
      theGsfElectron != gsfElectrons_->end(); ++theGsfElectron ) {
      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);
      size_t numSignalPFGammaCands = thePFTauRef->signalPFGammaCands().size();
      bool hasGsfTrack = thePFTauRef->leadPFChargedHadrCand()->gsfTrackRef().isNonnull();
        
      if ( isInEcalCrack(tauEtaAtEcalEntrance) || isInEcalCrack(leadChargedPFCandEtaAtEcalEntrance) ) {
        // add category index
        category_output_->setValue(tauIndex_, category);
        // return MVA output value
        return -99;
      }

      if ( TMath::Abs(tauEtaAtEcalEntrance) < 1.479 ) { // Barrel
        if        ( numSignalPFGammaCands == 0 && !hasGsfTrack ) {
          category = 4.;
        } else if ( numSignalPFGammaCands == 0 &&  hasGsfTrack ) {
          category = 5.;
        } else if ( numSignalPFGammaCands >= 1 && !hasGsfTrack ) {
          category = 6.;
        } else if ( numSignalPFGammaCands >= 1 &&  hasGsfTrack ) {
          category = 7.;
        }
      } else { // Endcap
        if        ( numSignalPFGammaCands == 0 && !hasGsfTrack ) {
          category = 12.;
        } else if ( numSignalPFGammaCands == 0 &&  hasGsfTrack ) {
          category = 13.;
        } else if ( numSignalPFGammaCands >= 1 && !hasGsfTrack ) {
          category = 14.;
        } else if ( numSignalPFGammaCands >= 1 &&  hasGsfTrack ) {
          category = 15.;
        }
      }

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

    if ( !isGsfElectronMatched ) {
      mvaValue = mva_->MVAValue(*thePFTauRef);
      size_t numSignalPFGammaCands = thePFTauRef->signalPFGammaCands().size();
      bool hasGsfTrack = thePFTauRef->leadPFChargedHadrCand()->gsfTrackRef().isNonnull();
      
      if ( isInEcalCrack(tauEtaAtEcalEntrance) || isInEcalCrack(leadChargedPFCandEtaAtEcalEntrance) ) {
    // add category index
    category_output_->setValue(tauIndex_, category);
    // return MVA output value
    return -99;
      }
      
      if ( TMath::Abs(tauEtaAtEcalEntrance) < 1.479 ) { // Barrel
    if        ( numSignalPFGammaCands == 0 && !hasGsfTrack ) {
      category = 0.;
    } else if ( numSignalPFGammaCands == 0 &&  hasGsfTrack ) {
      category = 1.;
    } else if ( numSignalPFGammaCands >= 1 && !hasGsfTrack ) {
      category = 2.;
    } else if ( numSignalPFGammaCands >= 1 &&  hasGsfTrack ) {
      category = 3.;
    }
      } else { // Endcap
    if        ( numSignalPFGammaCands == 0 && !hasGsfTrack ) {
      category = 8.;
    } else if ( numSignalPFGammaCands == 0 &&  hasGsfTrack ) {
      category = 9.;
    } else if ( numSignalPFGammaCands >= 1 && !hasGsfTrack ) {
      category = 10.;
    } else if ( numSignalPFGammaCands >= 1 &&  hasGsfTrack ) {
      category = 11.;
    }
      }
    }
  }

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

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

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

virtual

Reimplemented from TauDiscriminationProducerBase< TauType, TauDiscriminator >.

Definition at line 227 of file PFRecoTauDiscriminationAgainstElectronMVA5.cc.

References edm::Event::put().

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

bool PFRecoTauDiscriminationAgainstElectronMVA5::isInEcalCrack ( double  eta ) const

private

Definition at line 234 of file PFRecoTauDiscriminationAgainstElectronMVA5.cc.

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

Member Data Documentation

std::auto_ptr<PFTauDiscriminator> PFRecoTauDiscriminationAgainstElectronMVA5::category_output_

private

Definition at line 73 of file PFRecoTauDiscriminationAgainstElectronMVA5.cc.

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

private

Definition at line 70 of file PFRecoTauDiscriminationAgainstElectronMVA5.cc.

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

private

Definition at line 69 of file PFRecoTauDiscriminationAgainstElectronMVA5.cc.

std::string PFRecoTauDiscriminationAgainstElectronMVA5::moduleLabel_

private

Definition at line 63 of file PFRecoTauDiscriminationAgainstElectronMVA5.cc.

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

AntiElectronIDMVA5* PFRecoTauDiscriminationAgainstElectronMVA5::mva_

private

Definition at line 65 of file PFRecoTauDiscriminationAgainstElectronMVA5.cc.

float* PFRecoTauDiscriminationAgainstElectronMVA5::mvaInput_

private

Definition at line 66 of file PFRecoTauDiscriminationAgainstElectronMVA5.cc.

edm::InputTag PFRecoTauDiscriminationAgainstElectronMVA5::srcGsfElectrons_

private

Definition at line 68 of file PFRecoTauDiscriminationAgainstElectronMVA5.cc.

edm::Handle<TauCollection> PFRecoTauDiscriminationAgainstElectronMVA5::taus_

private

Definition at line 71 of file PFRecoTauDiscriminationAgainstElectronMVA5.cc.

int PFRecoTauDiscriminationAgainstElectronMVA5::verbosity_

private

Definition at line 75 of file PFRecoTauDiscriminationAgainstElectronMVA5.cc.