Inheritance diagram for DistortedMuonProducer:

Public Member Functions
	DistortedMuonProducer (const edm::ParameterSet &)

	~DistortedMuonProducer () 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_

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 DistortedMuonProducer.cc.

Constructor & Destructor Documentation

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

explicit

Definition at line 48 of file DistortedMuonProducer.cc.

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

                                                                         {
   // What is being produced
   produces<std::vector<reco::Muon> >();
 
   // 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")));
 
   // 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: DistortedMuonProducer : Size of some parameters do not match the EtaBinEdges vector!!";
   }
 }

DistortedMuonProducer::~DistortedMuonProducer ( )

override

Definition at line 122 of file DistortedMuonProducer.cc.

122 {}

Member Function Documentation

void DistortedMuonProducer::beginJob ( void )

overrideprivatevirtual

Reimplemented from edm::EDProducer.

Definition at line 125 of file DistortedMuonProducer.cc.

125 {}

void DistortedMuonProducer::endJob ( void )

overrideprivatevirtual

Reimplemented from edm::EDProducer.

Definition at line 128 of file DistortedMuonProducer.cc.

128 {}

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

overrideprivate

Definition at line 131 of file DistortedMuonProducer.cc.

References reco::Muon::clone(), DEFINE_FWK_MODULE, efficiencyRatioOverMC_, reco::LeafCandidate::eta(), muonRecoAnalyzer_cfi::etaBin, etaBinEdges_, genMatchMapToken_, edm::Event::getByToken(), mps_fire::i, edm::Ref< C, T, F >::isNull(), edm::EventBase::isRealData(), dqmiolumiharvest::j, LogTrace, reco::LeafCandidate::mass(), eostools::move(), amptDefaultParameters_cff::mu, pdwgLeptonRecoSkim_cfi::muonCollection, muonToken_, LaserClient_cfi::nbins, MuonAlignmentFromReference_cff::newmuons, reco::LeafCandidate::phi(), reco::LeafCandidate::pt(), edm::Event::put(), relativeShiftOnPt_, relativeUncertaintyOnPt_, reco::LeafCandidate::setP4(), shiftOnOneOverPt_, 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;
   }
 
   unsigned int muonCollectionSize = muonCollection->size();
 
   std::unique_ptr<reco::MuonCollection> newmuons(new reco::MuonCollection);
 
   for (unsigned int i = 0; i < muonCollectionSize; i++) {
     edm::RefToBase<reco::Muon> mu = muonCollection->refAt(i);
 
     double ptgen = mu->pt();
     double etagen = mu->eta();
     reco::GenParticleRef gen = (*genMatchMap)[mu];
     if (!gen.isNull()) {
       ptgen = gen->pt();
       etagen = gen->eta();
       LogTrace("") << ">>> Muon-GenParticle match found; ptmu= " << mu->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;
       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();
 
     // New muon
     double ptmu = mu->pt();
     ptmu += ptgen * (shift1 * ptgen + shift2 + sigma1 * rndg1 * ptgen + sigma2 * rndg2);
     reco::Muon* newmu = mu->clone();
     newmu->setP4(reco::Particle::PolarLorentzVector(ptmu, mu->eta(), mu->phi(), mu->mass()));
     newmuons->push_back(*newmu);
   }
 
   ev.put(std::move(newmuons));
 }