DistortedPFCandProducer Class Reference

Inheritance diagram for DistortedPFCandProducer:

Public Member Functions
	DistortedPFCandProducer (const edm::ParameterSet &)
	~DistortedPFCandProducer () override
Public Member Functions inherited from edm::EDProducer
	EDProducer ()
SerialTaskQueue *	globalLuminosityBlocksQueue ()
SerialTaskQueue *	globalRunsQueue ()
ModuleDescription const &	moduleDescription () const
	~EDProducer () 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
std::vector< bool > const &	recordProvenanceList () 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)
TypeLabelList const &	typeLabelList () const
	used by the fwk to register the list of products of this module More...
	~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 &&)=default
	EDConsumerBase (EDConsumerBase const &)=delete
ESProxyIndex const *	esGetTokenIndices (edm::Transition iTrans) const
std::vector< ESProxyIndex > const &	esGetTokenIndicesVector (edm::Transition iTrans) const
std::vector< ESRecordIndex > const &	esGetTokenRecordIndicesVector (edm::Transition iTrans) const
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 &	operator= (EDConsumerBase &&)=default
EDConsumerBase const &	operator= (EDConsumerBase const &)=delete
bool	registeredToConsume (ProductResolverIndex, bool, BranchType) const
bool	registeredToConsumeMany (TypeID const &, BranchType) const
ProductResolverIndexAndSkipBit	uncheckedIndexFrom (EDGetToken) const
void	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)
void	updateLookup (eventsetup::ESRecordsToProxyIndices const &)
virtual	~EDConsumerBase () noexcept(false)

Private Member Functions
void	beginJob () override
void	endJob () override
void	produce (edm::Event &, const edm::EventSetup &) override

Private Attributes
std::vector< double >	efficiencyRatioOverMC_
std::vector< double >	etaBinEdges_
edm::EDGetTokenT< reco::GenParticleMatch >	genMatchMapToken_
edm::EDGetTokenT< edm::View< reco::Muon > >	muonToken_
edm::EDGetTokenT< edm::View< reco::PFCandidate > >	pfToken_
std::vector< double >	relativeShiftOnPt_
std::vector< double >	relativeUncertaintyOnPt_
std::vector< double >	shiftOnOneOverPt_
std::vector< double >	uncertaintyOnOneOverPt_

Additional Inherited Members
Public Types inherited from edm::EDProducer
typedef EDProducer	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
Static Public Member Functions inherited from edm::EDProducer
static const std::string &	baseType ()
static void	fillDescriptions (ConfigurationDescriptions &descriptions)
static void	prevalidate (ConfigurationDescriptions &descriptions)
static bool	wantsGlobalLuminosityBlocks ()
static bool	wantsGlobalRuns ()
static bool	wantsInputProcessBlocks ()
static bool	wantsProcessBlocks ()
static bool	wantsStreamLuminosityBlocks ()
static bool	wantsStreamRuns ()
Protected Member Functions inherited from edm::ProducerBase
template<class ProductType >
BranchAliasSetterT< ProductType >	produces ()
	declare what type of product will make and with which optional label More...
template<typename ProductType , BranchType B>
BranchAliasSetterT< ProductType >	produces ()
template<typename ProductType , Transition B>
BranchAliasSetterT< ProductType >	produces ()
BranchAliasSetter	produces (const TypeID &id, std::string instanceName=std::string(), bool recordProvenance=true)
template<BranchType B>
BranchAliasSetter	produces (const TypeID &id, std::string instanceName=std::string(), bool recordProvenance=true)
template<Transition B>
BranchAliasSetter	produces (const TypeID &id, std::string instanceName=std::string(), bool recordProvenance=true)
template<typename ProductType , Transition B>
BranchAliasSetterT< ProductType >	produces (std::string instanceName)
template<class ProductType >
BranchAliasSetterT< ProductType >	produces (std::string instanceName)
template<typename ProductType , BranchType B>
BranchAliasSetterT< ProductType >	produces (std::string instanceName)
ProducesCollector	producesCollector ()
Protected Member Functions inherited from edm::EDConsumerBase
EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)
template<BranchType B = InEvent>
EDConsumerBaseAdaptor< B >	consumes (edm::InputTag tag) noexcept
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 ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes ()
template<Transition Tr = Transition::Event>
constexpr auto	esConsumes () noexcept
template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes (ESInputTag const &tag)
template<Transition Tr = Transition::Event>
auto	esConsumes (ESInputTag tag) noexcept
template<Transition Tr = Transition::Event>
ESGetTokenGeneric	esConsumes (eventsetup::EventSetupRecordKey const &iRecord, eventsetup::DataKey const &iKey)
	Used with EventSetupRecord::doGet. More...
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)
template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)

Detailed Description

Definition at line 17 of file DistortedPFCandProducer.cc.

Constructor & Destructor Documentation

DistortedPFCandProducer()

DistortedPFCandProducer::DistortedPFCandProducer ( const edm::ParameterSet &  pset )

explicit

Definition at line 51 of file DistortedPFCandProducer.cc.

                                                                            {
  // What is being produced
  produces<std::vector<reco::PFCandidate> >();
 
  // Input products
  muonToken_ =
      consumes<edm::View<reco::Muon> >(pset.getUntrackedParameter<edm::InputTag>("MuonTag", edm::InputTag("muons")));
  genMatchMapToken_ = consumes<reco::GenParticleMatch>(
      pset.getUntrackedParameter<edm::InputTag>("GenMatchMapTag", edm::InputTag("genMatchMap")));
  pfToken_ = consumes<edm::View<reco::PFCandidate> >(
      pset.getUntrackedParameter<edm::InputTag>("PFTag", edm::InputTag("particleFlow")));
 
  // Eta edges
  std::vector<double> defEtaEdges;
  defEtaEdges.push_back(-999999.);
  defEtaEdges.push_back(999999.);
  etaBinEdges_ = pset.getUntrackedParameter<std::vector<double> >("EtaBinEdges", defEtaEdges);
  unsigned int ninputs_expected = etaBinEdges_.size() - 1;
 
  // Distortions in muon momentum
  std::vector<double> defDistortion;
  defDistortion.push_back(0.);
 
  shiftOnOneOverPt_ =
      pset.getUntrackedParameter<std::vector<double> >("ShiftOnOneOverPt", defDistortion);  // in [1/GeV]
  if (shiftOnOneOverPt_.size() == 1 && ninputs_expected > 1) {
    for (unsigned int i = 1; i < ninputs_expected; i++) {
      shiftOnOneOverPt_.push_back(shiftOnOneOverPt_[0]);
    }
  }
 
  relativeShiftOnPt_ = pset.getUntrackedParameter<std::vector<double> >("RelativeShiftOnPt", defDistortion);
  if (relativeShiftOnPt_.size() == 1 && ninputs_expected > 1) {
    for (unsigned int i = 1; i < ninputs_expected; i++) {
      relativeShiftOnPt_.push_back(relativeShiftOnPt_[0]);
    }
  }
 
  uncertaintyOnOneOverPt_ =
      pset.getUntrackedParameter<std::vector<double> >("UncertaintyOnOneOverPt", defDistortion);  // in [1/GeV]
  if (uncertaintyOnOneOverPt_.size() == 1 && ninputs_expected > 1) {
    for (unsigned int i = 1; i < ninputs_expected; i++) {
      uncertaintyOnOneOverPt_.push_back(uncertaintyOnOneOverPt_[0]);
    }
  }
 
  relativeUncertaintyOnPt_ = pset.getUntrackedParameter<std::vector<double> >("RelativeUncertaintyOnPt", defDistortion);
  if (relativeUncertaintyOnPt_.size() == 1 && ninputs_expected > 1) {
    for (unsigned int i = 1; i < ninputs_expected; i++) {
      relativeUncertaintyOnPt_.push_back(relativeUncertaintyOnPt_[0]);
    }
  }
 
  // Data/MC efficiency ratios
  std::vector<double> defEfficiencyRatio;
  defEfficiencyRatio.push_back(1.);
  efficiencyRatioOverMC_ =
      pset.getUntrackedParameter<std::vector<double> >("EfficiencyRatioOverMC", defEfficiencyRatio);
  if (efficiencyRatioOverMC_.size() == 1 && ninputs_expected > 1) {
    for (unsigned int i = 1; i < ninputs_expected; i++) {
      efficiencyRatioOverMC_.push_back(efficiencyRatioOverMC_[0]);
    }
  }
 
  // Send a warning if there are inconsistencies in vector sizes !!
  bool effWrong = efficiencyRatioOverMC_.size() != ninputs_expected;
  bool momWrong = shiftOnOneOverPt_.size() != ninputs_expected || relativeShiftOnPt_.size() != ninputs_expected ||
                  uncertaintyOnOneOverPt_.size() != ninputs_expected ||
                  relativeUncertaintyOnPt_.size() != ninputs_expected;
  if (effWrong and momWrong) {
    edm::LogError("")
        << "WARNING: DistortedPFCandProducer : Size of some parameters do not match the EtaBinEdges vector!!";
  }
}

References efficiencyRatioOverMC_, etaBinEdges_, genMatchMapToken_, mps_fire::i, HLT_FULL_cff::InputTag, muonToken_, pfToken_, muonDTDigis_cfi::pset, relativeShiftOnPt_, relativeUncertaintyOnPt_, shiftOnOneOverPt_, and uncertaintyOnOneOverPt_.

~DistortedPFCandProducer()

DistortedPFCandProducer::~DistortedPFCandProducer ( )

override

Definition at line 127 of file DistortedPFCandProducer.cc.

{}

Member Function Documentation

beginJob()

void DistortedPFCandProducer::beginJob ( void  )

overrideprivatevirtual

Reimplemented from edm::EDProducer.

Definition at line 130 of file DistortedPFCandProducer.cc.

{}

endJob()

void DistortedPFCandProducer::endJob ( void  )

overrideprivatevirtual

Reimplemented from edm::EDProducer.

Definition at line 133 of file DistortedPFCandProducer.cc.

{}

produce()

void DistortedPFCandProducer::produce ( edm::Event &  ev, const edm::EventSetup &  iSetup  )

overrideprivatevirtual

Implements edm::EDProducer.

Definition at line 136 of file DistortedPFCandProducer.cc.

                                                                               {
  if (ev.isRealData())
    return;
 
  // Muon collection
  edm::Handle<edm::View<reco::Muon> > muonCollection;
  if (!ev.getByToken(muonToken_, muonCollection)) {
    edm::LogError("") << ">>> Muon collection does not exist !!!";
    return;
  }
 
  edm::Handle<reco::GenParticleMatch> genMatchMap;
  if (!ev.getByToken(genMatchMapToken_, genMatchMap)) {
    edm::LogError("") << ">>> Muon-GenParticle match map does not exist !!!";
    return;
  }
 
  // Get PFCandidate collection
  edm::Handle<edm::View<reco::PFCandidate> > pfCollection;
  if (!ev.getByToken(pfToken_, pfCollection)) {
    edm::LogError("") << ">>> PFCandidate collection does not exist !!!";
    return;
  }
 
  unsigned int muonCollectionSize = muonCollection->size();
  unsigned int pfCollectionSize = pfCollection->size();
 
  if (pfCollectionSize < 1)
    return;
 
  // Ask for PfMuon consistency
  bool pfMuonFound = false;
 
  std::unique_ptr<reco::PFCandidateCollection> newmuons(new reco::PFCandidateCollection);
 
  // Loop on all PF candidates
  for (unsigned int j = 0; j < pfCollectionSize; j++) {
    edm::RefToBase<reco::PFCandidate> pf = pfCollection->refAt(j);
 
    // New PF muon
    double ptmu = pf->pt();
 
    for (unsigned int i = 0; i < muonCollectionSize; i++) {
      edm::RefToBase<reco::Muon> mu = muonCollection->refAt(i);
 
      // Check the muon is in the PF collection
      if (pf->particleId() == reco::PFCandidate::mu) {
        reco::MuonRef muref = pf->muonRef();
        if (muref.isNonnull()) {
          if (muref.key() == mu.key()) {
            if (mu->isStandAloneMuon() && ptmu == muref->standAloneMuon()->pt() &&
                ((!mu->isGlobalMuon() || (mu->isGlobalMuon() && ptmu != muref->combinedMuon()->pt())) &&
                 (!mu->isTrackerMuon() || (mu->isTrackerMuon() && ptmu != mu->track()->pt())))) {
              pfMuonFound = false;
            } else if (!mu->isTrackerMuon()) {
              pfMuonFound = false;
            } else {
              pfMuonFound = true;
            }
          } else {
            pfMuonFound = false;
          }
        }
      }
 
      // do nothing if StandAlone muon
      //const reco::Track& track = *trackRef;
 
      if (!pfMuonFound)
        continue;
 
      double ptgen = pf->pt();
      double etagen = pf->eta();
 
      reco::GenParticleRef gen = (*genMatchMap)[mu];
      if (!gen.isNull()) {
        ptgen = gen->pt();
        etagen = gen->eta();
        LogTrace("") << ">>> Muon-GenParticle match found; ptmu= " << pf->pt() << ", ptgen= " << ptgen;
      } else {
        LogTrace("") << ">>> MUON-GENPARTICLE MATCH NOT FOUND!!!";
      }
 
      // Initialize parameters
      double effRatio = 0.;
      double shift1 = 0.;
      double shift2 = 0.;
      double sigma1 = 0.;
      double sigma2 = 0.;
 
      // Find out which eta bin should be used
      unsigned int nbins = etaBinEdges_.size() - 1;
      unsigned int etaBin = nbins;
      if (etagen > etaBinEdges_[0] && etagen < etaBinEdges_[nbins]) {
        for (unsigned int j = 1; j <= nbins; ++j) {
          if (etagen > etaBinEdges_[j])
            continue;
          etaBin = j - 1;
          break;
        }
      }
      if (etaBin < nbins) {
        LogTrace("") << ">>> etaBin: " << etaBin << ", for etagen =" << etagen;
      } else {
        // Muon is rejected if outside the considered eta range
        LogTrace("") << ">>> Muon outside eta range: reject it; etagen = " << etagen;
        pfMuonFound = false;
        continue;
      }
 
      if (!pfMuonFound)
        continue;
 
      // Set shifts
      shift1 = shiftOnOneOverPt_[etaBin];
      shift2 = relativeShiftOnPt_[etaBin];
      LogTrace("") << "\tshiftOnOneOverPt= " << shift1 * 100 << " [%]";
      LogTrace("") << "\trelativeShiftOnPt= " << shift2 * 100 << " [%]";
 
      // Set resolutions
      sigma1 = uncertaintyOnOneOverPt_[etaBin];
      sigma2 = relativeUncertaintyOnPt_[etaBin];
      LogTrace("") << "\tuncertaintyOnOneOverPt= " << sigma1 << " [1/GeV]";
      LogTrace("") << "\trelativeUncertaintyOnPt= " << sigma2 * 100 << " [%]";
 
      // Set efficiency ratio
      effRatio = efficiencyRatioOverMC_[etaBin];
      LogTrace("") << "\tefficiencyRatioOverMC= " << effRatio;
 
      // Reject muons according to efficiency ratio
      double rndf = CLHEP::RandFlat::shoot();
      if (rndf > effRatio)
        continue;
 
      // Gaussian Random numbers for smearing
      double rndg1 = CLHEP::RandGauss::shoot();
      double rndg2 = CLHEP::RandGauss::shoot();
 
      // change here the pt of the candidate, if it is a muon
 
      ptmu += ptgen * (shift1 * ptgen + shift2 + sigma1 * rndg1 * ptgen + sigma2 * rndg2);
      pfMuonFound = false;
    }
 
    reco::PFCandidate* newmu = pf->clone();
    newmu->setP4(reco::Particle::PolarLorentzVector(ptmu, pf->eta(), pf->phi(), pf->mass()));
 
    newmuons->push_back(*newmu);
  }
 
  ev.put(std::move(newmuons));
}

References efficiencyRatioOverMC_, etaBin(), etaBinEdges_, ev, genMatchMapToken_, mps_fire::i, edm::Ref< C, T, F >::isNonnull(), dqmiolumiharvest::j, edm::Ref< C, T, F >::key(), LogTrace, eostools::move(), amptDefaultParameters_cff::mu, reco::PFCandidate::mu, pdwgLeptonRecoSkim_cfi::muonCollection, muonToken_, LaserClient_cfi::nbins, MuonAlignmentFromReference_cff::newmuons, packedPFCandidateRefMixer_cfi::pf, pfToken_, relativeShiftOnPt_, relativeUncertaintyOnPt_, reco::LeafCandidate::setP4(), shiftOnOneOverPt_, and uncertaintyOnOneOverPt_.

Member Data Documentation

efficiencyRatioOverMC_

std::vector<double> DistortedPFCandProducer::efficiencyRatioOverMC_

private

Definition at line 37 of file DistortedPFCandProducer.cc.

Referenced by DistortedPFCandProducer(), and produce().

etaBinEdges_

std::vector<double> DistortedPFCandProducer::etaBinEdges_

private

Definition at line 30 of file DistortedPFCandProducer.cc.

Referenced by DistortedPFCandProducer(), and produce().

genMatchMapToken_

edm::EDGetTokenT<reco::GenParticleMatch> DistortedPFCandProducer::genMatchMapToken_

private

Definition at line 28 of file DistortedPFCandProducer.cc.

Referenced by DistortedPFCandProducer(), and produce().

muonToken_

edm::EDGetTokenT<edm::View<reco::Muon> > DistortedPFCandProducer::muonToken_

private

Definition at line 27 of file DistortedPFCandProducer.cc.

Referenced by DistortedPFCandProducer(), and produce().

pfToken_

edm::EDGetTokenT<edm::View<reco::PFCandidate> > DistortedPFCandProducer::pfToken_

private

Definition at line 29 of file DistortedPFCandProducer.cc.

Referenced by DistortedPFCandProducer(), and produce().

relativeShiftOnPt_

std::vector<double> DistortedPFCandProducer::relativeShiftOnPt_

private

Definition at line 33 of file DistortedPFCandProducer.cc.

Referenced by DistortedPFCandProducer(), and produce().

relativeUncertaintyOnPt_

std::vector<double> DistortedPFCandProducer::relativeUncertaintyOnPt_

private

Definition at line 35 of file DistortedPFCandProducer.cc.

Referenced by DistortedPFCandProducer(), and produce().

shiftOnOneOverPt_

std::vector<double> DistortedPFCandProducer::shiftOnOneOverPt_

private

Definition at line 32 of file DistortedPFCandProducer.cc.

Referenced by DistortedPFCandProducer(), and produce().

uncertaintyOnOneOverPt_

std::vector<double> DistortedPFCandProducer::uncertaintyOnOneOverPt_

private

Definition at line 34 of file DistortedPFCandProducer.cc.

Referenced by DistortedPFCandProducer(), and produce().