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::ProducerBase &producer, 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 33 of file PreMixingPileupCopy.h.

Constructor & Destructor Documentation

PreMixingPileupCopy::PreMixingPileupCopy	(	const edm::ParameterSet &	ps,
		edm::ProducerBase &	producer,
		edm::ConsumesCollector &&	iC
	)

Definition at line 14 of file PreMixingPileupCopy.cc.

References genPUProtonsInputTags_, edm::ProductRegistryHelper::produces(), and GlobalPosition_Frontier_DevDB_cff::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"))
  {
    producer.produces<std::vector<PileupSummaryInfo> >();
    producer.produces<int>("bunchSpacing");
    producer.produces<CrossingFramePlaybackInfoNew>();

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

edm::PreMixingPileupCopy::~PreMixingPileupCopy ( )

default

Member Function Documentation

void PreMixingPileupCopy::addPileupInfo

(

PileUpEventPrincipal const &

pep

)

Definition at line 44 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 GlobalPosition_Frontier_DevDB_cff::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;
    }
  }

int edm::PreMixingPileupCopy::getBunchSpacing ( ) const

inline

Definition at line 42 of file PreMixingPileupCopy.h.

References MillePedeFileConverter_cfg::e.

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

{ return bsStorage_; }

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

inline

Definition at line 41 of file PreMixingPileupCopy.h.

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

{ return pileupSummaryStorage_; }

float PreMixingPileupCopy::getTrueNumInteractions

(

PileUpEventPrincipal const &

pep

)

const

Definition at line 29 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();
  }

void PreMixingPileupCopy::putPileupInfo

(

edm::Event &

e

)

Definition at line 86 of file PreMixingPileupCopy.cc.

References bsStorage_, crossingFramePlaybackStorage_, foundPlayback_, genPUProtons_, genPUProtons_labels_, training_settings::idx, eostools::move(), pileupSummaryStorage_, and edm::Event::put().

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

int edm::PreMixingPileupCopy::bsStorage_

private

Definition at line 55 of file PreMixingPileupCopy.h.

Referenced by addPileupInfo(), and putPileupInfo().

edm::InputTag edm::PreMixingPileupCopy::bunchSpacingInputTag_

private

Definition at line 47 of file PreMixingPileupCopy.h.

Referenced by addPileupInfo().

edm::InputTag edm::PreMixingPileupCopy::cfPlaybackInputTag_

private

Definition at line 48 of file PreMixingPileupCopy.h.

Referenced by addPileupInfo().

CrossingFramePlaybackInfoNew edm::PreMixingPileupCopy::crossingFramePlaybackStorage_

private

Definition at line 53 of file PreMixingPileupCopy.h.

Referenced by addPileupInfo(), and putPileupInfo().

bool edm::PreMixingPileupCopy::foundPlayback_

private

Definition at line 60 of file PreMixingPileupCopy.h.

Referenced by addPileupInfo(), and putPileupInfo().

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

private

Definition at line 58 of file PreMixingPileupCopy.h.

Referenced by addPileupInfo(), and putPileupInfo().

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

private

Definition at line 57 of file PreMixingPileupCopy.h.

Referenced by addPileupInfo(), and putPileupInfo().

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

private

Definition at line 50 of file PreMixingPileupCopy.h.

Referenced by addPileupInfo(), and PreMixingPileupCopy().

edm::InputTag edm::PreMixingPileupCopy::pileupInfoInputTag_

private

Definition at line 46 of file PreMixingPileupCopy.h.

Referenced by addPileupInfo(), and getTrueNumInteractions().

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

private

Definition at line 54 of file PreMixingPileupCopy.h.

Referenced by addPileupInfo(), and putPileupInfo().