#include <RecoJets/FFTJetProducers/plugins/FFTJetCorrectionProducer.cc>

Inheritance diagram for FFTJetCorrectionProducer:

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

	~FFTJetCorrectionProducer ()

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

bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)

virtual	~EDConsumerBase ()

Private Member Functions
template<typename Jet >
void	applyCorrections (edm::Event &iEvent, const edm::EventSetup &iSetup)

virtual void	beginJob () override

virtual void	endJob () override

template<typename Jet >
void	makeProduces (const std::string &alias, const std::string &tag)

template<typename Jet >
void	performPileupSubtraction (Jet &)

virtual void	produce (edm::Event &, const edm::EventSetup &) override

Private Attributes
unsigned long	eventCount

const edm::InputTag	inputLabel

const JetType	jetType

const std::string	outputLabel

const std::vector< std::string >	records

std::vector< int >	sequenceMasks

const bool	subtractPileup

const bool	subtractPileupAs4Vec

const bool	verbose

const bool	writeUncertainties

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

Description: producer for correcting jets created by FFTJetProducer

Implementation: [Notes on implementation]

Definition at line 92 of file FFTJetCorrectionProducer.cc.

Constructor & Destructor Documentation

FFTJetCorrectionProducer::FFTJetCorrectionProducer ( const edm::ParameterSet & ps )

explicit

Definition at line 360 of file FFTJetCorrectionProducer.cc.

References edm::ParameterSet::getUntrackedParameter(), jet_type_switch, makeProduces(), outputLabel, AlCaHLTBitMon_QueryRunRegistry::string, and writeUncertainties.

     : inputLabel(ps.getParameter<edm::InputTag>("src")),
       outputLabel(ps.getParameter<std::string>("outputLabel")),
       jetType(parseJetType(ps.getParameter<std::string>("jetType"))),
       records(ps.getParameter<std::vector<std::string> >("records")),
       writeUncertainties(ps.getParameter<bool>("writeUncertainties")),
       subtractPileup(ps.getParameter<bool>("subtractPileup")),
       subtractPileupAs4Vec(ps.getParameter<bool>("subtractPileupAs4Vec")),
       verbose(ps.getUntrackedParameter<bool>("verbose", false)),
       eventCount(0UL)
 {
     const std::string alias(ps.getUntrackedParameter<std::string>(
                                 "alias", outputLabel));
     jet_type_switch(makeProduces, alias, outputLabel);
 
     if (writeUncertainties)
         produces<std::vector<float> >(outputLabel).setBranchAlias(alias);
 }

FFTJetCorrectionProducer::~FFTJetCorrectionProducer ( )

Definition at line 380 of file FFTJetCorrectionProducer.cc.

381 {

382 }

Member Function Documentation

template<typename Jet >

void FFTJetCorrectionProducer::applyCorrections	(	edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

private

Definition at line 170 of file FFTJetCorrectionProducer.cc.

References alignCSCRings::corr, gather_cfg::cout, edm::hlt::Exception, f, reco::FFTJet< Real >::f_recoScale(), reco::FFTJet< Real >::f_status(), reco::FFTJet< Real >::f_vec(), edm::Event::getByLabel(), i, dtTPAnalyzer_cfg::inputLabel, instance, makeTFileFromDB::isMC, edm::EventBase::isRealData(), j, fwrapper::jets, testEve_cfg::level, PILEUP_CALCULATION_MASK, PILEUP_SUBTRACTION_MASK_ANY, RecoTauCleanerPlugins::pt, edm::Event::put(), alignCSCRings::s, python.multivaluedict::sort(), and mathSSE::sqrt().

Referenced by produce().

 {
     using reco::FFTJet;
 
     // Various useful typedefs
     typedef reco::FFTAnyJet<Jet> MyJet;
     typedef std::vector<MyJet> MyCollection;
     typedef typename FFTJetCorrectorSequenceTypemap<MyJet>::loader Loader;
     typedef typename Loader::data_type CorrectorSequence;
     typedef typename CorrectorSequence::result_type CorrectorResult;
 
     // Load the jet corrector sequences
     const unsigned nRecords = records.size();
     std::vector<edm::ESHandle<CorrectorSequence> > handles(nRecords);
     for (unsigned irec=0; irec<nRecords; ++irec)
         Loader::instance().load(iSetup, records[irec], handles[irec]);
 
     // Figure out which correction levels we are applying
     // and create masks which will indicate this
     sequenceMasks.clear();
     sequenceMasks.reserve(nRecords);
 
     int totalMask = 0;
     for (unsigned irec=0; irec<nRecords; ++irec)
     {
         int levelMask = 0;
         const unsigned nLevels = handles[irec]->nLevels();
         for (unsigned i=0; i<nLevels; ++i)
         {
             const unsigned lev = (*handles[irec])[i].level();
  
             // Not tracking "level 0" corrections in the status word.
             // Level 0 is basically reserved for uncertainty calculations.
             if (lev)
             {
                 const int mask = (1 << lev);
                 if (totalMask & mask)
                     throw cms::Exception("FFTJetBadConfig")
                         << "Error in FFTJetCorrectionProducer::applyCorrections:"
                         << " jet correction at level " << lev
                         << " is applied more than once\n";
                 totalMask |= mask;
                 levelMask |= mask;
             }
         }
         sequenceMasks.push_back(levelMask << 12);
     }
     totalMask = (totalMask << 12);
 
     // Is this data or MC?
     const bool isMC = !iEvent.isRealData();
 
     // Load the jet collection
     edm::Handle<MyCollection> jets;
     iEvent.getByLabel(inputLabel, jets);
 
     // Create the output collection
     const unsigned nJets = jets->size();
     std::auto_ptr<MyCollection> coll(new MyCollection());
     coll->reserve(nJets);
 
     // Cycle over jets and apply the corrector sequences
     bool sorted = true;
     double previousPt = DBL_MAX;
     for (unsigned ijet=0; ijet<nJets; ++ijet)
     {
         const MyJet& j((*jets)[ijet]);
 
         // Check that this jet has not been corrected yet
         const int initialStatus = j.getFFTSpecific().f_status();
         if (initialStatus & totalMask)
             throw cms::Exception("FFTJetBadConfig")
                 << "Error in FFTJetCorrectionProducer::applyCorrections: "
                 << "this jet collection is already corrected for some or all "
                 << "of the specified levels\n";
 
         MyJet corJ(j);
 
         if (verbose)
         {
             const reco::FFTJet<float>& fj = corJ.getFFTSpecific();
             std::cout << "++++ Evt " << eventCount << " jet " << ijet
                       << ": pt = " << corJ.pt()
                       << ", eta = " << fj.f_vec().eta()
                       << ", R = " << fj.f_recoScale()
                       << ", s = 0x" << std::hex << fj.f_status() << std::dec
                       << std::endl;
         }
 
         // Check if we need to subtract pileup first.
         // Pileup subtraction is not part of the corrector sequence
         // itself because 4-vector subtraction does not map well
         // into multiplication of 4-vectors by a scale factor.
         if (subtractPileup)
         {
             if (initialStatus & PILEUP_SUBTRACTION_MASK_ANY)
                 throw cms::Exception("FFTJetBadConfig")
                     << "Error in FFTJetCorrectionProducer::applyCorrections: "
                     << "this jet collection is already pileup-subtracted\n";
             if (!(initialStatus & PILEUP_CALCULATION_MASK))
                 throw cms::Exception("FFTJetBadConfig")
                     << "Error in FFTJetCorrectionProducer::applyCorrections: "
                     << "pileup was not calculated for this jet collection\n";
             performPileupSubtraction(corJ);
 
             if (verbose)
             {
                 const reco::FFTJet<float>& fj = corJ.getFFTSpecific();
                 std::cout << "     Pileup subtract"
                           << ": pt = " << corJ.pt()
                           << ", eta = " << fj.f_vec().eta()
                           << ", R = " << fj.f_recoScale()
                           << ", s = 0x" << std::hex << fj.f_status() << std::dec
                           << std::endl;
             }
         }
 
         // Apply all jet correction sequences
         double sigmaSquared = 0.0;
         for (unsigned irec=0; irec<nRecords; ++irec)
         {
             const CorrectorResult& corr = handles[irec]->correct(corJ, isMC);
 
             // Update the 4-vector
             FFTJet<float>& fftJet(const_cast<FFTJet<float>&>(corJ.getFFTSpecific()));
             corJ.setP4(corr.vec());
             fftJet.setFourVec(corr.vec());
 
             // Update the jet correction status
             fftJet.setStatus(fftJet.f_status() | sequenceMasks[irec]);
 
             // Update the (systematic) uncertainty
             const double s = corr.sigma();
             sigmaSquared += s*s;
         }
 
         // There is no place for uncertainty in the jet structure.
         // However, there is the unused pileup field (FFTJet maintains
         // the pileup separately as a 4-vector). Use this unused field
         // to store the uncertainty. This hack is needed because
         // subsequent sequence sorting by Pt can change the jet ordering.
         if (writeUncertainties)
             corJ.setPileup(sqrt(sigmaSquared));
 
         coll->push_back(corJ);
 
         // Check whether the sequence remains sorted by pt
         const double pt = corJ.pt();
         if (pt > previousPt)
             sorted = false;
         previousPt = pt;
 
         if (verbose)
         {
             const reco::FFTJet<float>& fj = corJ.getFFTSpecific();
             std::cout << "     Fully corrected"
                       << ": pt = " << corJ.pt()
                       << ", eta = " << fj.f_vec().eta()
                       << ", R = " << fj.f_recoScale()
                       << ", s = 0x" << std::hex << fj.f_status() << std::dec
                       << std::endl;
         }
     }
 
     if (!sorted)
         std::sort(coll->begin(), coll->end(), LocalSortByPt());
 
     // Create the uncertainty sequence
     if (writeUncertainties)
     {
         std::auto_ptr<std::vector<float> > unc(new std::vector<float>());
         unc->reserve(nJets);
         for (unsigned ijet=0; ijet<nJets; ++ijet)
         {
             MyJet& j((*coll)[ijet]);
             unc->push_back(j.pileup());
             j.setPileup(0.f);
         }
         iEvent.put(unc, outputLabel);
     }
 
     iEvent.put(coll, outputLabel);
     ++eventCount;
 }

void FFTJetCorrectionProducer::beginJob ( void )

overrideprivatevirtual

Reimplemented from edm::EDProducer.

Definition at line 393 of file FFTJetCorrectionProducer.cc.

394 {

395 }

void FFTJetCorrectionProducer::endJob ( void )

overrideprivatevirtual

Reimplemented from edm::EDProducer.

Definition at line 398 of file FFTJetCorrectionProducer.cc.

399 {

400 }

template<typename T >

void FFTJetCorrectionProducer::makeProduces	(	const std::string &	alias,
		const std::string &	tag
	)

private

Definition at line 143 of file FFTJetCorrectionProducer.cc.

References GlobalPosition_Frontier_DevDB_cff::tag.

Referenced by FFTJetCorrectionProducer().

 {
     produces<std::vector<reco::FFTAnyJet<T> > >(tag).setBranchAlias(alias);
 }

template<typename Jet >

void FFTJetCorrectionProducer::performPileupSubtraction ( Jet & jet )

private

Definition at line 151 of file FFTJetCorrectionProducer.cc.

References fftjetcms::adjustForPileup(), PILEUP_SUBTRACTION_MASK_4VEC, PILEUP_SUBTRACTION_MASK_PT, and ntuplemaker::status.

 {
     using reco::FFTJet;
 
     FFTJet<float>& fftJet(const_cast<FFTJet<float>&>(jet.getFFTSpecific()));
     const math::XYZTLorentzVector& new4vec = adjustForPileup(
         fftJet.f_vec(), fftJet.f_pileup(), subtractPileupAs4Vec);
     fftJet.setFourVec(new4vec);
     int status = fftJet.f_status();
     if (subtractPileupAs4Vec)
         status |= PILEUP_SUBTRACTION_MASK_4VEC;
     else
         status |= PILEUP_SUBTRACTION_MASK_PT;
     fftJet.setStatus(status);
     jet.setP4(new4vec);
 }

void FFTJetCorrectionProducer::produce	(	edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

overrideprivatevirtual

Implements edm::EDProducer.

Definition at line 386 of file FFTJetCorrectionProducer.cc.

References applyCorrections(), and jet_type_switch.

 {
     jet_type_switch(applyCorrections, iEvent, iSetup);
 }

Member Data Documentation

unsigned long FFTJetCorrectionProducer::eventCount

private

Definition at line 138 of file FFTJetCorrectionProducer.cc.

const edm::InputTag FFTJetCorrectionProducer::inputLabel

private

Definition at line 113 of file FFTJetCorrectionProducer.cc.

const JetType FFTJetCorrectionProducer::jetType

private

Definition at line 119 of file FFTJetCorrectionProducer.cc.

const std::string FFTJetCorrectionProducer::outputLabel

private

Definition at line 116 of file FFTJetCorrectionProducer.cc.

Referenced by FFTJetCorrectionProducer().

const std::vector<std::string> FFTJetCorrectionProducer::records

private

Definition at line 122 of file FFTJetCorrectionProducer.cc.

std::vector<int> FFTJetCorrectionProducer::sequenceMasks

private

Definition at line 135 of file FFTJetCorrectionProducer.cc.

const bool FFTJetCorrectionProducer::subtractPileup

private

Definition at line 128 of file FFTJetCorrectionProducer.cc.

const bool FFTJetCorrectionProducer::subtractPileupAs4Vec

private

Definition at line 129 of file FFTJetCorrectionProducer.cc.

const bool FFTJetCorrectionProducer::verbose

private

Definition at line 132 of file FFTJetCorrectionProducer.cc.

const bool FFTJetCorrectionProducer::writeUncertainties

private

Definition at line 125 of file FFTJetCorrectionProducer.cc.

Referenced by FFTJetCorrectionProducer().

Public Member Functions

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation