BMixingModule.h

Go to the documentation of this file.

#ifndef BMixingModule_h
#define BMixingModule_h

#include <vector>
#include <memory>

#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "Mixing/Base/interface/PileUp.h"
#include "FWCore/Framework/interface/ESWatcher.h"
#include "CondFormats/DataRecord/interface/MixingRcd.h"

namespace edm {
  namespace MixingCache {
    struct Config {
      Config(edm::ParameterSet const& pset, unsigned int maxNbSources);
      int bunchSpace_;
      int minBunch_;
      int maxBunch_;
      bool playback_;
      bool configFromDB_ = false;
      std::vector<std::string> sourceNames_;
      std::vector<std::shared_ptr<PileUpConfig>> inputConfigs_;
    };
  }  // namespace MixingCache

  class BMixingModule : public stream::EDProducer<GlobalCache<MixingCache::Config>> {
  public:
    explicit BMixingModule(const edm::ParameterSet& ps, MixingCache::Config const* globalConf);

    ~BMixingModule() override;

    void registerLateConsumes(eventsetup::ESRecordsToProductResolverIndices const&) override;

    void produce(edm::Event& e1, const edm::EventSetup& c) override;

    virtual void initializeEvent(const edm::Event& event, const edm::EventSetup& setup) {}

    // edm::Event is non-const because digitizers put their products into the Event.
    virtual void finalizeEvent(edm::Event& event, const edm::EventSetup& setup) {}

    void beginRun(const edm::Run& r, const edm::EventSetup& setup) override;
    void beginLuminosityBlock(const edm::LuminosityBlock& l, const edm::EventSetup& setup) override;

    void endRun(const edm::Run& r, const edm::EventSetup& setup) override;
    void endLuminosityBlock(const edm::LuminosityBlock& l, const edm::EventSetup& setup) override;

    static std::unique_ptr<MixingCache::Config> initializeGlobalCache(edm::ParameterSet const&);
    static void globalEndJob(MixingCache::Config*) {}

    // to be overloaded by dependent class
    virtual void reload(const edm::EventSetup& setup) {}

    // Should 'averageNumber' return 0 or 1 if there is no mixing? It is the average number of
    // *crossings*, including the hard scatter, or the average number of overlapping events?
    // We have guessed 'overlapping events'.
    double averageNumber() const { return inputSources_[0] ? inputSources_[0]->averageNumber() : 0.0; }
    // Should 'poisson' return 0 or 1 if there is no mixing? See also averageNumber above.
    bool poisson() const { return inputSources_[0] ? inputSources_[0]->poisson() : 0.0; }

    virtual void createnewEDProduct();
    virtual void checkSignal(const edm::Event& e);
    virtual void addSignals(const edm::Event& e, const edm::EventSetup& c) {}
    virtual void addPileups(
        const int bcr, EventPrincipal* ep, unsigned int eventId, unsigned int worker, const edm::EventSetup& c) {}
    virtual void setBcrOffset();
    virtual void setSourceOffset(const unsigned int s);
    virtual void put(edm::Event& e, const edm::EventSetup& c) {}
    virtual void doPileUp(edm::Event& e, const edm::EventSetup& c);

  protected:
    void setupPileUpEvent(const edm::EventSetup& setup);
    void dropUnwantedBranches(std::vector<std::string> const& wantedBranches);
    void beginStream(edm::StreamID) override;
    void endStream() override;
    //      std::string type_;
    int bunchSpace_;
    int vertexOffset_;
    bool checktof_;
    int minBunch_;
    int maxBunch_;
    bool const mixProdStep1_;
    bool const mixProdStep2_;

    bool readDB_;
    bool playback_;
    const static unsigned int maxNbSources_;
    bool doit_[4];  //FIXME
    std::vector<float> TrueNumInteractions_;

    unsigned int eventId_;

    // input, cosmics, beamhalo_plus, beamhalo_minus
    std::vector<std::shared_ptr<PileUp>> inputSources_;

    void update(edm::EventSetup const&);
    edm::ESWatcher<MixingRcd> parameterWatcher_;
  };

}  // namespace edm

#endif