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

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

ESProxyIndex const *	esGetTokenIndices (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 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)

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	wantsStreamLuminosityBlocks ()

static bool	wantsStreamRuns ()

Protected Member Functions inherited from edm::ProducerBase
ProducesCollector	producesCollector ()

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 ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes ()

template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes (ESInputTag const &tag)

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

Definition at line 17 of file DistortedPFCandProducer.cc.

Constructor & Destructor Documentation

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

explicit

Definition at line 51 of file DistortedPFCandProducer.cc.

References efficiencyRatioOverMC_, etaBinEdges_, genMatchMapToken_, edm::ParameterSet::getUntrackedParameter(), mps_fire::i, HLT_2018_cff::InputTag, muonToken_, pfToken_, relativeShiftOnPt_, relativeUncertaintyOnPt_, shiftOnOneOverPt_, and uncertaintyOnOneOverPt_.

                                                                             {
   // 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!!";
   }
 }

DistortedPFCandProducer::~DistortedPFCandProducer ( )

override

Definition at line 127 of file DistortedPFCandProducer.cc.

127 {}

Member Function Documentation

void DistortedPFCandProducer::beginJob ( void )

overrideprivatevirtual

Reimplemented from edm::EDProducer.

Definition at line 130 of file DistortedPFCandProducer.cc.

130 {}

void DistortedPFCandProducer::endJob ( void )

overrideprivatevirtual

Reimplemented from edm::EDProducer.

Definition at line 133 of file DistortedPFCandProducer.cc.

133 {}

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

overrideprivate

Definition at line 136 of file DistortedPFCandProducer.cc.

References reco::PFCandidate::clone(), DEFINE_FWK_MODULE, efficiencyRatioOverMC_, reco::LeafCandidate::eta(), muonRecoAnalyzer_cfi::etaBin, etaBinEdges_, genMatchMapToken_, edm::Event::getByToken(), mps_fire::i, reco::Muon::isGlobalMuon(), edm::Ref< C, T, F >::isNonnull(), edm::Ref< C, T, F >::isNull(), edm::EventBase::isRealData(), reco::Muon::isStandAloneMuon(), reco::Muon::isTrackerMuon(), dqmiolumiharvest::j, edm::RefToBase< T >::key(), edm::Ref< C, T, F >::key(), LogTrace, reco::LeafCandidate::mass(), eostools::move(), amptDefaultParameters_cff::mu, reco::PFCandidate::mu, pdwgLeptonRecoSkim_cfi::muonCollection, reco::PFCandidate::muonRef(), muonToken_, LaserClient_cfi::nbins, MuonAlignmentFromReference_cff::newmuons, reco::PFCandidate::particleId(), packedPFCandidateRefMixer_cfi::pf, pfToken_, reco::LeafCandidate::phi(), reco::LeafCandidate::pt(), edm::Event::put(), relativeShiftOnPt_, relativeUncertaintyOnPt_, reco::LeafCandidate::setP4(), shiftOnOneOverPt_, reco::Muon::track(), and uncertaintyOnOneOverPt_.

Referenced by JSONExport.JsonExport::export(), HTMLExport.HTMLExport::export(), and HTMLExport.HTMLExportStatic::export().

                                                                                {
   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));
 }