Inheritance diagram for DistortedMuonProducer:

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

	~DistortedMuonProducer ()

Public Member Functions inherited from edm::EDProducer
	EDProducer ()

ModuleDescription const &	moduleDescription () const

virtual	~EDProducer ()

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

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, std::unordered_multimap< std::string, edm::ProductResolverIndex > const &iIndicies, std::string const &moduleLabel)

virtual	~ProducerBase () noexcept(false)

Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const

	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 &	itemsToGetFromEvent () 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

void	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)

virtual	~EDConsumerBase () noexcept(false)

Private Member Functions
virtual void	beginJob () override

virtual void	endJob () override

virtual 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
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)

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

Definition at line 17 of file DistortedMuonProducer.cc.

Constructor & Destructor Documentation

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

explicit

Definition at line 49 of file DistortedMuonProducer.cc.

References efficiencyRatioOverMC_, etaBinEdges_, genMatchMapToken_, edm::ParameterSet::getUntrackedParameter(), mps_fire::i, 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 ( )

Definition at line 110 of file DistortedMuonProducer.cc.

110 {

111 }

Member Function Documentation

void DistortedMuonProducer::beginJob ( void )

overrideprivatevirtual

Reimplemented from edm::EDProducer.

Definition at line 114 of file DistortedMuonProducer.cc.

114 {

115 }

void DistortedMuonProducer::endJob ( void )

overrideprivatevirtual

Reimplemented from edm::EDProducer.

Definition at line 118 of file DistortedMuonProducer.cc.

118 {

119 }

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

overrideprivatevirtual

Definition at line 122 of file DistortedMuonProducer.cc.

References reco::Muon::clone(), DEFINE_FWK_MODULE, efficiencyRatioOverMC_, reco::LeafCandidate::eta(), conversionPostprocessing_cfi::etaBin, etaBinEdges_, genMatchMapToken_, edm::Event::getByToken(), mps_fire::i, edm::Ref< C, T, F >::isNull(), edm::EventBase::isRealData(), LogTrace, reco::LeafCandidate::mass(), eostools::move(), RPCpg::mu, HiRecoMuon_cff::muonCollection, muonToken_, pileupCalc::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));
 }