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_;
      std::vector<std::string> sourceNames_;
      std::vector<std::shared_ptr<PileUpConfig>> inputConfigs_;
    };
  }

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

      ~BMixingModule() 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_;
  };

}//edm

#endif