pat::LeptonUpdater< T > Class Template Reference

Inheritance diagram for pat::LeptonUpdater< T >:

Public Member Functions
	LeptonUpdater (const edm::ParameterSet &iConfig)
const std::vector< double > &	miniIsoParams (const T &lep) const
template<>
const std::vector< double > &	miniIsoParams (const pat::Electron &lep) const
void	produce (edm::StreamID, edm::Event &, edm::EventSetup const &) const override
void	readMiniIsoParams (const edm::ParameterSet &iConfig)
template<>
void	readMiniIsoParams (const edm::ParameterSet &iConfig)
void	recomputeMuonBasicSelectors (T &, const reco::Vertex &, const bool) const
template<>
void	recomputeMuonBasicSelectors (pat::Muon &lep, const reco::Vertex &pv, const bool do_hip_mitigation_2016) const
void	setDZ (T &lep, const reco::Vertex &pv) const
template<>
void	setDZ (pat::Electron &anElectron, const reco::Vertex &pv) const
template<>
void	setDZ (pat::Muon &aMuon, const reco::Vertex &pv) const
	~LeptonUpdater () override
Public Member Functions inherited from edm::global::EDProducer<>
	EDProducer ()=default
bool	hasAbilityToProduceInLumis () const final
bool	hasAbilityToProduceInRuns () const final
bool	wantsGlobalLuminosityBlocks () const final
bool	wantsGlobalRuns () const final
bool	wantsStreamLuminosityBlocks () const final
bool	wantsStreamRuns () const final
Public Member Functions inherited from edm::global::EDProducerBase
	EDProducerBase ()
ModuleDescription const &	moduleDescription () const
	~EDProducerBase () override
Public Member Functions inherited from edm::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)
std::vector< edm::ProductResolverIndex > const &	indiciesForPutProducts (BranchType iBranchType) const
	ProducerBase ()
std::vector< edm::ProductResolverIndex > const &	putTokenIndexToProductResolverIndex () const
void	registerProducts (ProducerBase *, ProductRegistry *, ModuleDescription const &)
std::function< void(BranchDescription const &)>	registrationCallback () const
	used by the fwk to register list of products More...
void	resolvePutIndicies (BranchType iBranchType, ModuleToResolverIndicies const &iIndicies, std::string const &moduleLabel)
	~ProducerBase () noexcept(false) override
Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const
void	convertCurrentProcessAlias (std::string const &processName)
	Convert "@currentProcess" in InputTag process names to the actual current process name. More...
	EDConsumerBase ()
	EDConsumerBase (EDConsumerBase const &)=delete
	EDConsumerBase (EDConsumerBase &&)=default
ProductResolverIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const
void	itemsMayGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const
void	itemsToGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const
std::vector< ProductResolverIndexAndSkipBit > const &	itemsToGetFrom (BranchType iType) const
void	labelsForToken (EDGetToken iToken, Labels &oLabels) const
void	modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const
EDConsumerBase const &	operator= (EDConsumerBase const &)=delete
EDConsumerBase &	operator= (EDConsumerBase &&)=default
bool	registeredToConsume (ProductResolverIndex, bool, BranchType) const
bool	registeredToConsumeMany (TypeID const &, BranchType) const
ProductResolverIndexAndSkipBit	uncheckedIndexFrom (EDGetToken) const
void	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)
virtual	~EDConsumerBase () noexcept(false)

Static Public Member Functions
static void	fillDescriptions (edm::ConfigurationDescriptions &descriptions)
Static Public Member Functions inherited from edm::global::EDProducerBase
static const std::string &	baseType ()
static void	fillDescriptions (ConfigurationDescriptions &descriptions)
static void	prevalidate (ConfigurationDescriptions &descriptions)

Private Attributes
edm::EDGetTokenT< reco::BeamSpot >	beamLineToken_
bool	computeMiniIso_
bool	fixDxySign_
std::vector< double >	miniIsoParams_ [2]
edm::EDGetTokenT< pat::PackedCandidateCollection >	pcToken_
bool	recomputeMuonBasicSelectors_
edm::EDGetTokenT< std::vector< T > >	src_
edm::EDGetTokenT< std::vector< reco::Vertex > >	vertices_

Additional Inherited Members
Public Types inherited from edm::global::EDProducerBase
typedef EDProducerBase	ModuleType
Public Types inherited from edm::ProducerBase
using	ModuleToResolverIndicies = std::unordered_multimap< std::string, std::tuple< edm::TypeID const *, const char *, edm::ProductResolverIndex >>
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList
Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels
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)

Detailed Description

template<typename T>
class pat::LeptonUpdater< T >

Definition at line 19 of file LeptonUpdater.cc.

Constructor & Destructor Documentation

template<typename T >

pat::LeptonUpdater< T >::LeptonUpdater ( const edm::ParameterSet &  iConfig )

inlineexplicit

Definition at line 23 of file LeptonUpdater.cc.

References pat::LeptonUpdater< T >::computeMiniIso_, edm::ParameterSet::getParameter(), pat::LeptonUpdater< T >::pcToken_, pat::LeptonUpdater< T >::readMiniIsoParams(), and pat::LeptonUpdater< T >::recomputeMuonBasicSelectors_.

                                                              :
            src_(consumes<std::vector<T>>(iConfig.getParameter<edm::InputTag>("src"))),
            vertices_(consumes<std::vector<reco::Vertex>>(iConfig.getParameter<edm::InputTag>("vertices"))),
        beamLineToken_(consumes<reco::BeamSpot>(iConfig.getParameter<edm::InputTag>("beamspot"))),
        computeMiniIso_(iConfig.getParameter<bool>("computeMiniIso")),
        fixDxySign_(iConfig.getParameter<bool>("fixDxySign"))
        {
            //for mini-isolation calculation
            if (computeMiniIso_) {
                readMiniIsoParams(iConfig);
                pcToken_ = consumes<pat::PackedCandidateCollection >(iConfig.getParameter<edm::InputTag>("pfCandsForMiniIso"));
            }
        recomputeMuonBasicSelectors_ = false;
            if (typeid(T) == typeid(pat::Muon)) recomputeMuonBasicSelectors_ = iConfig.getParameter<bool>("recomputeMuonBasicSelectors");
            produces<std::vector<T>>();
        }

template<typename T >

pat::LeptonUpdater< T >::~LeptonUpdater ( )

inlineoverride

Definition at line 40 of file LeptonUpdater.cc.

References pat::LeptonUpdater< T >::produce().

{}

Member Function Documentation

template<typename T >

static void pat::LeptonUpdater< T >::fillDescriptions ( edm::ConfigurationDescriptions &  descriptions )

inlinestatic

Definition at line 44 of file LeptonUpdater.cc.

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

                                                                                {
          edm::ParameterSetDescription desc;
          desc.add<edm::InputTag>("src")->setComment("Lepton collection");
          desc.add<edm::InputTag>("vertices")->setComment("Vertex collection");
      desc.add<edm::InputTag>("beamspot", edm::InputTag("offlineBeamSpot"))->setComment("Beam spot");
      desc.add<bool>("computeMiniIso", false)->setComment("Recompute miniIsolation");
      desc.add<bool>("fixDxySign", false)->setComment("Fix the IP sign");
          desc.addOptional<edm::InputTag>("pfCandsForMiniIso", edm::InputTag("packedPFCandidates"))->setComment("PackedCandidate collection used for miniIso");
          if (typeid(T) == typeid(pat::Muon)) {
            desc.add<bool>("recomputeMuonBasicSelectors",false)->setComment("Recompute basic cut-based muon selector flags");
            desc.addOptional<std::vector<double>>("miniIsoParams")->setComment("Parameters used for miniIso (as in PATMuonProducer)");
            descriptions.add("muonsUpdated", desc);
          } else if (typeid(T) == typeid(pat::Electron)) {
            desc.addOptional<std::vector<double>>("miniIsoParamsB")->setComment("Parameters used for miniIso in the barrel (as in PATElectronProducer)");
            desc.addOptional<std::vector<double>>("miniIsoParamsE")->setComment("Parameters used for miniIso in the endcap (as in PATElectronProducer)");
            descriptions.add("electronsUpdated", desc);
          }
      }

template<typename T >

const std::vector<double>& pat::LeptonUpdater< T >::miniIsoParams ( const T &  lep ) const

inline

Definition at line 69 of file LeptonUpdater.cc.

References pat::LeptonUpdater< T >::miniIsoParams_, and pat::LeptonUpdater< T >::recomputeMuonBasicSelectors().

Referenced by pat::LeptonUpdater< T >::produce().

{ return miniIsoParams_[0]; }

template<>

const std::vector< double > & pat::LeptonUpdater< pat::Electron >::miniIsoParams

(

const pat::Electron &

lep

)

const

Definition at line 107 of file LeptonUpdater.cc.

References reco::GsfElectron::isEE(), and pat::LeptonUpdater< T >::miniIsoParams_.

                                                                                                  { 
      return miniIsoParams_[lep.isEE()]; 
  }

template<typename T >

void pat::LeptonUpdater< T >::produce ( edm::StreamID  , edm::Event &  iEvent, edm::EventSetup const &    ) const

overridevirtual

Implements edm::global::EDProducerBase.

Definition at line 122 of file LeptonUpdater.cc.

References funct::abs(), pat::LeptonUpdater< T >::beamLineToken_, ecalDrivenElectronSeedsParameters_cff::beamSpot, pat::LeptonUpdater< T >::computeMiniIso_, pat::LeptonUpdater< T >::fixDxySign_, edm::Event::getByToken(), pat::getMiniPFIsolation(), mps_fire::i, edm::HandleBase::isValid(), pat::LeptonUpdater< T >::miniIsoParams(), eostools::move(), gen::n, MillePedeFileConverter_cfg::out, pat::LeptonUpdater< T >::pcToken_, reco::Vertex::position(), edm::Handle< T >::product(), edm::Event::put(), MetAnalyzer::pv(), pat::LeptonUpdater< T >::recomputeMuonBasicSelectors(), pat::LeptonUpdater< T >::recomputeMuonBasicSelectors_, edm::Event::run(), pat::LeptonUpdater< T >::setDZ(), TrackRefitter_38T_cff::src, pat::LeptonUpdater< T >::src_, electrons_cff::vertices, and pat::LeptonUpdater< T >::vertices_.

Referenced by pat::LeptonUpdater< T >::~LeptonUpdater().

                                                                                           {
    edm::Handle<std::vector<T>> src;
    iEvent.getByToken(src_, src);

    edm::Handle<std::vector<reco::Vertex>> vertices;
    iEvent.getByToken(vertices_, vertices);
    const reco::Vertex & pv = vertices->front();

    edm::Handle<pat::PackedCandidateCollection> pc;
    if(computeMiniIso_) iEvent.getByToken(pcToken_, pc);

    edm::Handle<reco::BeamSpot> beamSpotHandle;
    iEvent.getByToken(beamLineToken_, beamSpotHandle);
    reco::BeamSpot beamSpot;
    bool beamSpotIsValid = false;
    if(beamSpotHandle.isValid()) {
      beamSpot = *beamSpotHandle;
      beamSpotIsValid = true;
    } else {
      edm::LogError("DataNotAvailable") << "No beam spot available  \n";
    }

    std::unique_ptr<std::vector<T>> out(new std::vector<T>(*src));

    const bool do_hip_mitigation_2016 = recomputeMuonBasicSelectors_ && (272728 <= iEvent.run() && iEvent.run() <= 278808);

    for (unsigned int i = 0, n = src->size(); i < n; ++i) {
        T & lep = (*out)[i];
        setDZ(lep, pv);
        if (computeMiniIso_) {
            const auto & params = miniIsoParams(lep);
            pat::PFIsolation miniiso = pat::getMiniPFIsolation(pc.product(), lep.p4(),
                                                               params[0], params[1], params[2],
                                                               params[3], params[4], params[5],
                                                               params[6], params[7], params[8]);
            lep.setMiniPFIsolation(miniiso);
        }
    if (recomputeMuonBasicSelectors_) recomputeMuonBasicSelectors(lep,pv,do_hip_mitigation_2016);
    //Fixing the sign of impact parameters
    if (fixDxySign_) {
      float signPV = 1.;
      float signBS = 1.;
      if (beamSpotIsValid) {
        signBS = copysign(1., lep.bestTrack()->dxy(beamSpot));
      }
      signPV = copysign(1., lep.bestTrack()->dxy(pv.position()));
      lep.setDB(abs(lep.dB(T::PV2D)) * signPV, lep.edB(T::PV2D), T::PV2D);
      lep.setDB(abs(lep.dB(T::BS2D)) * signBS, lep.edB(T::BS2D), T::BS2D);
    }
    }

    iEvent.put(std::move(out));
}

template<typename T >

void pat::LeptonUpdater< T >::readMiniIsoParams ( const edm::ParameterSet &  iConfig )

inline

Definition at line 65 of file LeptonUpdater.cc.

References edm::ParameterSet::getParameter(), pat::LeptonUpdater< T >::miniIsoParams_, and findQualityFiles::size.

Referenced by pat::LeptonUpdater< T >::LeptonUpdater().

                                                              { 
          miniIsoParams_[0] = iConfig.getParameter<std::vector<double> >("miniIsoParams");
          if(miniIsoParams_[0].size() != 9) throw cms::Exception("ParameterError", "miniIsoParams must have exactly 9 elements.\n");
      }

template<>

void pat::LeptonUpdater< pat::Electron >::readMiniIsoParams

(

const edm::ParameterSet &

iConfig

)

Definition at line 100 of file LeptonUpdater.cc.

References edm::ParameterSet::getParameter(), pat::LeptonUpdater< T >::miniIsoParams_, and findQualityFiles::size.

                                                                                    {
      miniIsoParams_[0] = iConfig.getParameter<std::vector<double> >("miniIsoParamsB");
      miniIsoParams_[1] = iConfig.getParameter<std::vector<double> >("miniIsoParamsE");
      if(miniIsoParams_[0].size() != 9) throw cms::Exception("ParameterError", "miniIsoParamsB must have exactly 9 elements.\n");
      if(miniIsoParams_[1].size() != 9) throw cms::Exception("ParameterError", "miniIsoParamsE must have exactly 9 elements.\n");
  }

template<typename T >

void pat::LeptonUpdater< T >::recomputeMuonBasicSelectors	(	T &	lep,
		const reco::Vertex &	pv,
		const bool	do_hip_mitigation_2016
	)		const

Definition at line 112 of file LeptonUpdater.cc.

Referenced by pat::LeptonUpdater< T >::miniIsoParams(), and pat::LeptonUpdater< T >::produce().

{}

template<>

void pat::LeptonUpdater< pat::Muon >::recomputeMuonBasicSelectors	(	pat::Muon &	lep,
		const reco::Vertex &	pv,
		const bool	do_hip_mitigation_2016
	)		const

Definition at line 115 of file LeptonUpdater.cc.

References muon::setCutBasedSelectorFlags().

                                                                                                                                        {
    muon::setCutBasedSelectorFlags(lep, &pv, do_hip_mitigation_2016);
  }

template<typename T >

void pat::LeptonUpdater< T >::setDZ ( T &  lep, const reco::Vertex &  pv  ) const

inline

Definition at line 63 of file LeptonUpdater.cc.

Referenced by pat::LeptonUpdater< T >::produce().

{}

template<>

void pat::LeptonUpdater< pat::Electron >::setDZ	(	pat::Electron &	anElectron,
		const reco::Vertex &	pv
	)		const

Definition at line 88 of file LeptonUpdater.cc.

References pat::Electron::gsfTrack(), reco::Vertex::position(), pat::Electron::PVDZ, pat::Electron::setDB(), HiIsolationCommonParameters_cff::track, and reco::Vertex::zError().

                                                                                              {
      auto track = anElectron.gsfTrack();
      anElectron.setDB( track->dz(pv.position()), std::hypot(track->dzError(), pv.zError()), pat::Electron::PVDZ );
  }

template<>

void pat::LeptonUpdater< pat::Muon >::setDZ	(	pat::Muon &	aMuon,
		const reco::Vertex &	pv
	)		const

Definition at line 94 of file LeptonUpdater.cc.

References pat::Muon::muonBestTrack(), reco::Vertex::position(), pat::Muon::PVDZ, pat::Muon::setDB(), HiIsolationCommonParameters_cff::track, and reco::Vertex::zError().

                                                                                 {
      auto track = aMuon.muonBestTrack();
      aMuon.setDB( track->dz(pv.position()), std::hypot(track->dzError(), pv.zError()), pat::Muon::PVDZ );
  }

Member Data Documentation

template<typename T >

edm::EDGetTokenT<reco::BeamSpot> pat::LeptonUpdater< T >::beamLineToken_

private

Definition at line 77 of file LeptonUpdater.cc.

Referenced by pat::LeptonUpdater< T >::produce().

template<typename T >

bool pat::LeptonUpdater< T >::computeMiniIso_

private

Definition at line 78 of file LeptonUpdater.cc.

Referenced by pat::LeptonUpdater< T >::LeptonUpdater(), and pat::LeptonUpdater< T >::produce().

template<typename T >

bool pat::LeptonUpdater< T >::fixDxySign_

private

Definition at line 79 of file LeptonUpdater.cc.

Referenced by pat::LeptonUpdater< T >::produce().

template<typename T >

std::vector<double> pat::LeptonUpdater< T >::miniIsoParams_[2]

private

Definition at line 81 of file LeptonUpdater.cc.

Referenced by pat::LeptonUpdater< T >::miniIsoParams(), and pat::LeptonUpdater< T >::readMiniIsoParams().

template<typename T >

edm::EDGetTokenT<pat::PackedCandidateCollection> pat::LeptonUpdater< T >::pcToken_

private

Definition at line 82 of file LeptonUpdater.cc.

Referenced by pat::LeptonUpdater< T >::LeptonUpdater(), and pat::LeptonUpdater< T >::produce().

template<typename T >

bool pat::LeptonUpdater< T >::recomputeMuonBasicSelectors_

private

Definition at line 80 of file LeptonUpdater.cc.

Referenced by pat::LeptonUpdater< T >::LeptonUpdater(), and pat::LeptonUpdater< T >::produce().

template<typename T >

edm::EDGetTokenT<std::vector<T> > pat::LeptonUpdater< T >::src_

private

Definition at line 75 of file LeptonUpdater.cc.

Referenced by pat::LeptonUpdater< T >::produce().

template<typename T >

edm::EDGetTokenT<std::vector<reco::Vertex> > pat::LeptonUpdater< T >::vertices_

private

Definition at line 76 of file LeptonUpdater.cc.

Referenced by pat::LeptonUpdater< T >::produce().