SimGeneral/MixingModule/plugins/MixingWorker.h - master - githubmirror/cmssw

SimGeneral/MixingModule/plugins/MixingWorker.h
217 строк7 KB
Suchandra
code related feedback from release managers implemented
05 май 2026, 20:37
05 май 2026, 20:379a0beda
#ifndef MixingWorker_h
#define MixingWorker_h

/** \class MixingWorker
 *
 * MixingWorker is an auxiliary class for the MixingModule
 *
 * \author Ursula Berthon, LLR Palaiseau
 *
 * \version   1st Version JMarch 2008

 *
 ************************************************************/

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventPrincipal.h"
#include "DataFormats/Common/interface/Wrapper.h"
#include "DataFormats/Common/interface/Handle.h"

#include "FWCore/Framework/interface/EventSetup.h"

#include "SimDataFormats/CrossingFrame/interface/CrossingFrame.h"
#include "SimDataFormats/CrossingFrame/interface/PCrossingFrame.h"
#include "SimDataFormats/TrackingHit/interface/PSimHitContainer.h"
#include "SimDataFormats/Track/interface/SimTrackFwd.h"
#include "SimDataFormats/Vertex/interface/SimVertexFwd.h"
#include "FWCore/Utilities/interface/InputTag.h"

#include <memory>
#include <vector>
#include <string>
#include <typeinfo>
#include "MixingWorkerBase.h"

namespace edm {
  template <class T>
  class MixingWorker : public MixingWorkerBase {
  public:
    /** standard constructor*/
    explicit MixingWorker()
        : minBunch_(-5),
          maxBunch_(3),
          bunchSpace_(75),
          subdet_(std::string(" ")),
          label_(std::string(" ")),
          labelCF_(std::string(" ")),
          maxNbSources_(5),
          makePCrossingFrame_(false),
          tag_(),
          tagSignal_(),
          allTags_(),
          crFrame_(nullptr) {}

    /*Normal constructor*/
    MixingWorker(int minBunch,
                 int maxBunch,
                 int bunchSpace,
                 std::string subdet,
                 std::string label,
                 std::string labelCF,
                 int maxNbSources,
                 InputTag &tag,
                 InputTag &tagCF,
                 bool makePCrossingFrame = false)
        : MixingWorkerBase(),
          minBunch_(minBunch),
          maxBunch_(maxBunch),
          bunchSpace_(bunchSpace),
          subdet_(subdet),
          label_(label),
          labelCF_(labelCF),
          maxNbSources_(maxNbSources),
          makePCrossingFrame_(makePCrossingFrame),
          tag_(tag),
          tagSignal_(tagCF),
          allTags_(),
          crFrame_(nullptr) {}

    /*constructor for HepMCproduct case*/
    MixingWorker(int minBunch,
                 int maxBunch,
                 int bunchSpace,
                 std::string subdet,
                 std::string label,
                 std::string labelCF,
                 int maxNbSources,
                 InputTag &tag,
                 InputTag &tagCF,
                 std::vector<InputTag> const &tags)
        : MixingWorkerBase(),
          minBunch_(minBunch),
          maxBunch_(maxBunch),
          bunchSpace_(bunchSpace),
          subdet_(subdet),
          label_(label),
          labelCF_(labelCF),
          maxNbSources_(maxNbSources),
          makePCrossingFrame_(false),
          tag_(tag),
          tagSignal_(tagCF),
          allTags_(tags),
          crFrame_(nullptr) {}

    /**Default destructor*/
    ~MixingWorker() override { ; }

  public:
    void reload(int minBunch, int maxBunch, int bunchSpace) override {
      minBunch_ = minBunch;
      maxBunch_ = maxBunch;
      bunchSpace_ = bunchSpace;
    }

    bool checkSignal(const edm::Event &e) override {
      bool got;
      InputTag t;
      edm::Handle<std::vector<T> > result_t;
      got = e.getByLabel(tag_, result_t);
      t = InputTag(tag_.label(), tag_.instance());

      if (got)
        LogInfo("MixingModule") << " Will create a CrossingFrame for " << typeid(T).name()
                                << " with InputTag= " << t.encode();
      return got;
    }

    void createnewEDProduct() override {
      crFrame_ = std::make_unique<CrossingFrame<T> >(minBunch_, maxBunch_, bunchSpace_, subdet_, maxNbSources_);
    }

    void addSignals(const edm::Event &e) override {
      edm::Handle<std::vector<T> > result_t;
      bool got = e.getByLabel(tag_, result_t);
      if (got) {
        LogDebug("MixingModule") << " adding " << result_t.product()->size() << " signal objects for "
                                 << typeid(T).name() << " with " << tag_;
        crFrame_->addSignals(result_t.product(), e.id());
      } else {
        LogInfo("MixingModule") << "!!!!!!! Did not get any signal data for " << typeid(T).name() << ", with " << tag_;
      }
    }

    void addPileups(const EventPrincipal &ep, ModuleCallingContext const *, unsigned int eventNr) override;

    void setBcrOffset() override { crFrame_->setBcrOffset(); }
    void setSourceOffset(const unsigned int s) override { crFrame_->setSourceOffset(s); }

    void setTof() override;

    void put(edm::Event &e) override {
      if (makePCrossingFrame_) {
        e.put(std::make_unique<PCrossingFrame<T> >(*crFrame_), label_);
      }
      e.put(std::move(crFrame_), label_);
      LogDebug("MixingModule") << " CF was put for type " << typeid(T).name() << " with " << label_;
    }

    // When using mixed secondary source
    // Copy the data from the PCrossingFrame to the CrossingFrame
    virtual void copyPCrossingFrame(const PCrossingFrame<T> *PCF);
    InputTag getInputTag() const override { return tag_; }

  private:
    int minBunch_;
    int maxBunch_;
    int bunchSpace_;
    std::string const subdet_;
    std::string const label_;
    std::string const labelCF_;
    unsigned int const maxNbSources_;
    bool const makePCrossingFrame_;
    InputTag tag_;
    InputTag tagSignal_;
    std::vector<InputTag> allTags_;  // for HepMCProduct

    std::unique_ptr<CrossingFrame<T> > crFrame_;
  };

  template <typename T>
  void MixingWorker<T>::addPileups(const EventPrincipal &ep, ModuleCallingContext const *mcc, unsigned int eventNr) {
    std::shared_ptr<Wrapper<std::vector<T> > const> shPtr = getProductByTag<std::vector<T> >(ep, tag_, mcc);
    if (shPtr) {
      LogDebug("MixingModule") << shPtr->product()->size() << "  pileup objects  added, eventNr " << eventNr;
      crFrame_->setPileupPtr(shPtr);
      crFrame_->addPileups(*shPtr->product());
    }
  }
  //=============== template specializations ====================================================================================

  template <>
  void MixingWorker<HepMCProduct>::addPileups(const EventPrincipal &ep,
                                              ModuleCallingContext const *,
                                              unsigned int eventNr);

  template <class T>
  void MixingWorker<T>::setTof() {
    ;
  }

  template <class T>
  void MixingWorker<T>::copyPCrossingFrame(const PCrossingFrame<T> *PCF) {
    crFrame_->setBunchRange(PCF->getBunchRange());
    crFrame_->setBunchSpace(PCF->getBunchSpace());
    crFrame_->setMaxNbSources(PCF->getMaxNbSources());
    crFrame_->setSubDet(PCF->getSubDet());
    crFrame_->setPileupOffsetsBcr(PCF->getPileupOffsetsBcr());
    crFrame_->setPileupOffsetsSource(PCF->getPileupOffsetsSource());
    crFrame_->setPileups(PCF->getPileups());

    // For playback option
    crFrame_->setPileupFileNr(PCF->getPileupFileNr());
    crFrame_->setIdFirstPileup(PCF->getIdFirstPileup());
  }

}  // namespace edm

#endif