Inheritance diagram for DistortedPFCandProducer:

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

	~DistortedPFCandProducer ()

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

ModuleDescription const &	moduleDescription () const

virtual	~EDProducer ()

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

void	registerProducts (ProducerBase , ProductRegistry , ModuleDescription const &)

std::function< void(BranchDescription const &)>	registrationCallback () const
	used by the fwk to register list of products More...

virtual	~ProducerBase ()

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

ProductHolderIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const

void	itemsMayGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const

void	itemsToGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const

std::vector < ProductHolderIndexAndSkipBit > const &	itemsToGetFromEvent () const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

void	modulesDependentUpon (const std::string &iProcessName, std::vector< const char * > &oModuleLabels) const

bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)

virtual	~EDConsumerBase ()

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_

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
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList

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::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

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

Definition at line 113 of file DistortedPFCandProducer.cc.

113 {

114 }

Member Function Documentation

void DistortedPFCandProducer::beginJob ( void )

overrideprivatevirtual

Reimplemented from edm::EDProducer.

Definition at line 117 of file DistortedPFCandProducer.cc.

117 {

118 }

void DistortedPFCandProducer::endJob ( void )

overrideprivatevirtual

Reimplemented from edm::EDProducer.

Definition at line 121 of file DistortedPFCandProducer.cc.

121 {

122 }

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

overrideprivatevirtual

Implements edm::EDProducer.

Definition at line 125 of file DistortedPFCandProducer.cc.

References reco::PFCandidate::clone(), efficiencyRatioOverMC_, reco::LeafCandidate::eta(), etaBinEdges_, edm::false, relval_steps::gen(), genMatchMapToken_, edm::Event::getByToken(), 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(), j, edm::RefToBase< T >::key(), edm::Ref< C, T, F >::key(), LogTrace, reco::LeafCandidate::mass(), RPCpg::mu, reco::PFCandidate::mu, reco::PFCandidate::muonRef(), muonToken_, pileupCalc::nbins, MuonAlignmentFromReference_cff::newmuons, reco::PFCandidate::particleId(), 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::auto_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(newmuons);
 
 
 }