edm::PreMixingPileupCopy Class Reference

#include <PreMixingPileupCopy.h>

Public Member Functions
void	addPileupInfo (PileUpEventPrincipal const &pep)
int	getBunchSpacing () const
const std::vector< PileupSummaryInfo > &	getPileupSummaryInfo () const
float	getTrueNumInteractions (PileUpEventPrincipal const &pep) const
	PreMixingPileupCopy (const edm::ParameterSet &ps, edm::ProducesCollector, edm::ConsumesCollector &&iC)
void	putPileupInfo (edm::Event &e)
	~PreMixingPileupCopy ()=default

Private Attributes
int	bsStorage_
edm::InputTag	bunchSpacingInputTag_
edm::InputTag	cfPlaybackInputTag_
CrossingFramePlaybackInfoNew	crossingFramePlaybackStorage_
bool	foundPlayback_
std::vector< std::vector< reco::GenParticle > >	genPUProtons_
std::vector< std::string >	genPUProtons_labels_
std::vector< edm::InputTag >	genPUProtonsInputTags_
edm::InputTag	pileupInfoInputTag_
std::vector< PileupSummaryInfo >	pileupSummaryStorage_

Detailed Description

Definition at line 34 of file PreMixingPileupCopy.h.

Constructor & Destructor Documentation

PreMixingPileupCopy()

PreMixingPileupCopy::PreMixingPileupCopy	(	const edm::ParameterSet &	ps,
		edm::ProducesCollector	producesCollector,
		edm::ConsumesCollector &&	iC
	)

Definition at line 12 of file PreMixingPileupCopy.cc.

References genPUProtonsInputTags_, edm::ProducesCollector::produces(), and makeGlobalPositionRcd_cfg::tag.

      : pileupInfoInputTag_(ps.getParameter<edm::InputTag>("PileupInfoInputTag")),
        bunchSpacingInputTag_(ps.getParameter<edm::InputTag>("BunchSpacingInputTag")),
        cfPlaybackInputTag_(ps.getParameter<edm::InputTag>("CFPlaybackInputTag")),
        genPUProtonsInputTags_(ps.getParameter<std::vector<edm::InputTag>>("GenPUProtonsInputTags")) {
    producesCollector.produces<std::vector<PileupSummaryInfo>>();
    producesCollector.produces<int>("bunchSpacing");
    producesCollector.produces<CrossingFramePlaybackInfoNew>();

    for (const auto& tag : genPUProtonsInputTags_) {
      producesCollector.produces<std::vector<reco::GenParticle>>(tag.label());
    }
  }

~PreMixingPileupCopy()

edm::PreMixingPileupCopy::~PreMixingPileupCopy ( )

default

Member Function Documentation

addPileupInfo()

void PreMixingPileupCopy::addPileupInfo

(

PileUpEventPrincipal const &

pep

)

Definition at line 41 of file PreMixingPileupCopy.cc.

References bsStorage_, bunchSpacingInputTag_, cfPlaybackInputTag_, crossingFramePlaybackStorage_, foundPlayback_, genPUProtons_, genPUProtons_labels_, genPUProtonsInputTags_, PileUpEventPrincipal::getByLabel(), edm::EventPrincipal::id(), edm::HandleBase::isValid(), LogDebug, pileupInfoInputTag_, pileupSummaryStorage_, PileUpEventPrincipal::principal(), and makeGlobalPositionRcd_cfg::tag.

Referenced by edm::PreMixingModule::pileWorker().

                                                                         {
    LogDebug("PreMixingPileupCopy") << "\n===============> adding pileup Info from event  " << pep.principal().id();

    // find PileupSummaryInfo, CFPlayback information, if it's there

    // Pileup info first
    edm::Handle<std::vector<PileupSummaryInfo>> pileupInfoHandle;
    pep.getByLabel(pileupInfoInputTag_, pileupInfoHandle);

    edm::Handle<int> bsHandle;
    pep.getByLabel(bunchSpacingInputTag_, bsHandle);

    if (pileupInfoHandle.isValid()) {
      pileupSummaryStorage_ = *pileupInfoHandle;
      LogDebug("PreMixingPileupCopy") << "PileupInfo Size: " << pileupSummaryStorage_.size();
    }
    bsStorage_ = bsHandle.isValid() ? *bsHandle : 10000;

    // Gen. PU protons
    edm::Handle<std::vector<reco::GenParticle>> genPUProtonsHandle;
    for (const auto& tag : genPUProtonsInputTags_) {
      pep.getByLabel(tag, genPUProtonsHandle);
      if (genPUProtonsHandle.isValid()) {
        genPUProtons_.push_back(*genPUProtonsHandle);
        genPUProtons_labels_.push_back(tag.label());
      } else {
        edm::LogWarning("PreMixingPileupCopy") << "Missing product with label: " << tag.label();
      }
    }

    // Playback
    edm::Handle<CrossingFramePlaybackInfoNew> playbackHandle;
    pep.getByLabel(cfPlaybackInputTag_, playbackHandle);
    foundPlayback_ = false;
    if (playbackHandle.isValid()) {
      crossingFramePlaybackStorage_ = *playbackHandle;
      foundPlayback_ = true;
    }
  }

getBunchSpacing()

int edm::PreMixingPileupCopy::getBunchSpacing ( ) const

inline

Definition at line 43 of file PreMixingPileupCopy.h.

References bsStorage_.

Referenced by edm::PreMixingModule::put().

{ return bsStorage_; }

getPileupSummaryInfo()

const std::vector<PileupSummaryInfo>& edm::PreMixingPileupCopy::getPileupSummaryInfo ( ) const

inline

Definition at line 42 of file PreMixingPileupCopy.h.

References pileupSummaryStorage_.

Referenced by edm::PreMixingModule::put().

{ return pileupSummaryStorage_; }

getTrueNumInteractions()

float PreMixingPileupCopy::getTrueNumInteractions

(

PileUpEventPrincipal const &

pep

)

const

Definition at line 28 of file PreMixingPileupCopy.cc.

References Exception, PileUpEventPrincipal::getByLabel(), pileupInfoInputTag_, and alignCSCRings::s.

Referenced by edm::PreMixingModule::pileWorker().

                                                                                         {
    edm::Handle<std::vector<PileupSummaryInfo>> pileupInfoHandle;
    pep.getByLabel(pileupInfoInputTag_, pileupInfoHandle);

    auto it = std::find_if(
        pileupInfoHandle->begin(), pileupInfoHandle->end(), [](const auto& s) { return s.getBunchCrossing() == 0; });
    if (it == pileupInfoHandle->end()) {
      throw cms::Exception("LogicError") << "Did not find PileupSummaryInfo in bunch crossing 0";
    }

    return it->getTrueNumInteractions();
  }

putPileupInfo()

void PreMixingPileupCopy::putPileupInfo

(

edm::Event &

e

)

Definition at line 81 of file PreMixingPileupCopy.cc.

References bsStorage_, crossingFramePlaybackStorage_, MillePedeFileConverter_cfg::e, foundPlayback_, genPUProtons_, genPUProtons_labels_, heavyIonCSV_trainingSettings::idx, eostools::move(), and pileupSummaryStorage_.

Referenced by edm::PreMixingModule::put().

                                                     {
    if (foundPlayback_) {
      e.put(std::make_unique<CrossingFramePlaybackInfoNew>(std::move(crossingFramePlaybackStorage_)));
    }
    e.put(std::make_unique<std::vector<PileupSummaryInfo>>(std::move(pileupSummaryStorage_)));
    e.put(std::make_unique<int>(bsStorage_), "bunchSpacing");

    // Gen. PU protons
    for (size_t idx = 0; idx < genPUProtons_.size(); ++idx) {
      e.put(std::make_unique<std::vector<reco::GenParticle>>(std::move(genPUProtons_[idx])), genPUProtons_labels_[idx]);
    }

    // clear local storage after this event
    pileupSummaryStorage_.clear();
    genPUProtons_.clear();
    genPUProtons_labels_.clear();
  }

Member Data Documentation

bsStorage_

int edm::PreMixingPileupCopy::bsStorage_

private

Definition at line 56 of file PreMixingPileupCopy.h.

Referenced by addPileupInfo(), getBunchSpacing(), and putPileupInfo().

bunchSpacingInputTag_

edm::InputTag edm::PreMixingPileupCopy::bunchSpacingInputTag_

private

Definition at line 48 of file PreMixingPileupCopy.h.

Referenced by addPileupInfo().

cfPlaybackInputTag_

edm::InputTag edm::PreMixingPileupCopy::cfPlaybackInputTag_

private

Definition at line 49 of file PreMixingPileupCopy.h.

Referenced by addPileupInfo().

crossingFramePlaybackStorage_

CrossingFramePlaybackInfoNew edm::PreMixingPileupCopy::crossingFramePlaybackStorage_

private

Definition at line 54 of file PreMixingPileupCopy.h.

Referenced by addPileupInfo(), and putPileupInfo().

foundPlayback_

bool edm::PreMixingPileupCopy::foundPlayback_

private

Definition at line 61 of file PreMixingPileupCopy.h.

Referenced by addPileupInfo(), and putPileupInfo().

genPUProtons_

std::vector<std::vector<reco::GenParticle> > edm::PreMixingPileupCopy::genPUProtons_

private

Definition at line 59 of file PreMixingPileupCopy.h.

Referenced by addPileupInfo(), and putPileupInfo().

genPUProtons_labels_

std::vector<std::string> edm::PreMixingPileupCopy::genPUProtons_labels_

private

Definition at line 58 of file PreMixingPileupCopy.h.

Referenced by addPileupInfo(), and putPileupInfo().

genPUProtonsInputTags_

std::vector<edm::InputTag> edm::PreMixingPileupCopy::genPUProtonsInputTags_

private

Definition at line 51 of file PreMixingPileupCopy.h.

Referenced by addPileupInfo(), and PreMixingPileupCopy().

pileupInfoInputTag_

edm::InputTag edm::PreMixingPileupCopy::pileupInfoInputTag_

private

Definition at line 47 of file PreMixingPileupCopy.h.

Referenced by addPileupInfo(), and getTrueNumInteractions().

pileupSummaryStorage_

std::vector<PileupSummaryInfo> edm::PreMixingPileupCopy::pileupSummaryStorage_

private

Definition at line 55 of file PreMixingPileupCopy.h.

Referenced by addPileupInfo(), getPileupSummaryInfo(), and putPileupInfo().