DataMixingModule.h

Go to the documentation of this file.

#ifndef SimGeneral_DataMixingModule_DataMixingModule_h
#define SimGeneral_DataMixingModule_DataMixingModule_h

#include "Mixing/Base/interface/BMixingModule.h"

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/EventPrincipal.h"

#include "DataFormats/Provenance/interface/ProductID.h"
#include "DataFormats/Common/interface/Handle.h"

#include "SimDataFormats/PileupSummaryInfo/interface/PileupSummaryInfo.h"

#include "SimGeneral/DataMixingModule/plugins/DataMixingEMWorker.h"
#include "SimGeneral/DataMixingModule/plugins/DataMixingHcalWorker.h"
#include "SimGeneral/DataMixingModule/plugins/DataMixingEMDigiWorker.h"
#include "SimGeneral/DataMixingModule/plugins/DataMixingEcalDigiWorkerProd.h"
#include "SimGeneral/DataMixingModule/plugins/DataMixingHcalDigiWorker.h"
#include "SimGeneral/DataMixingModule/plugins/DataMixingHcalDigiWorkerProd.h"
#include "SimGeneral/DataMixingModule/plugins/DataMixingMuonWorker.h"
#include "SimGeneral/DataMixingModule/plugins/DataMixingSiStripWorker.h"
#include "SimGeneral/DataMixingModule/plugins/DataMixingSiStripMCDigiWorker.h"
#include "SimGeneral/DataMixingModule/plugins/DataMixingSiStripRawWorker.h"
#include "SimGeneral/DataMixingModule/plugins/DataMixingSiPixelWorker.h"
#include "SimGeneral/DataMixingModule/plugins/DataMixingSiPixelMCDigiWorker.h"
#include "SimGeneral/DataMixingModule/plugins/DataMixingTrackingParticleWorker.h"
#include "SimGeneral/DataMixingModule/plugins/DataMixingPileupCopy.h"

#include <map>
#include <vector>
#include <string>

class DigiAccumulatorMixMod;

namespace edm {

  class ModuleCallingContext;

  class DataMixingModule : public BMixingModule
    {
    public:

      explicit DataMixingModule(const edm::ParameterSet& ps);

      virtual ~DataMixingModule();

      // copies, with EventSetup
      virtual void checkSignal(const edm::Event &e) {}
      virtual void createnewEDProduct() {}
      virtual void addSignals(const edm::Event &e, const edm::EventSetup& ES); 
      virtual void doPileUp(edm::Event &e,const edm::EventSetup& ES) override;
      virtual void put(edm::Event &e,const edm::EventSetup& ES) ;

      virtual void initializeEvent(edm::Event const& e, edm::EventSetup const& eventSetup);
      void beginRun(edm::Run const& run, edm::EventSetup const& eventSetup);
      void pileWorker(const edm::EventPrincipal&, int bcr, int EventId,const edm::EventSetup& ES, ModuleCallingContext const*);
      //virtual void beginJob();
      //virtual void endJob();
      virtual void beginLuminosityBlock(LuminosityBlock const& l1, EventSetup const& c) override;
      virtual void endLuminosityBlock(LuminosityBlock const& l1, EventSetup const& c) override;
      virtual void endRun(const edm::Run& r, const edm::EventSetup& setup) override;



    private:
      // data specifiers

      // Ecal
      //output:
      std::string EBRecHitCollectionDM_; // secondary name to be given to EB collection of hits
      std::string EERecHitCollectionDM_; // secondary name to be given to EE collection of hits
      std::string ESRecHitCollectionDM_; // secondary name to be given to EE collection of hits
      //Digis:
      //output
      std::string EBDigiCollectionDM_; // secondary name to be given to EB collection of hits
      std::string EEDigiCollectionDM_; // secondary name to be given to EE collection of hits
      std::string ESDigiCollectionDM_; // secondary name to be given to EE collection of hits

      // Hcal
      // Rechits:
      // output:
      std::string HBHERecHitCollectionDM_; // secondary name to be given to HBHE collection of hits
      std::string HORecHitCollectionDM_  ; // secondary name to be given to HO collection of hits
      std::string HFRecHitCollectionDM_  ; // secondary name to be given to HF collection of hits
      std::string ZDCRecHitCollectionDM_ ; // secondary name to be given to ZDC collection of hits
      // Digis:
      // output:
      std::string HBHEDigiCollectionDM_; // secondary name to be given to HBHE collection of hits
      std::string HODigiCollectionDM_  ; // secondary name to be given to HO collection of hits
      std::string HFDigiCollectionDM_  ; // secondary name to be given to HF collection of hits
      std::string ZDCDigiCollectionDM_ ; // secondary name to be given to ZDC collection of hits

      // Muons
      // output:
      std::string DTDigiCollectionDM_;       // secondary name to be given to new DT digis
      std::string RPCDigiCollectionDM_;      // secondary name to be given to new RPC digis
      std::string CSCStripDigiCollectionDM_; // secondary name given to new collection of CSC Strip digis
      std::string CSCWireDigiCollectionDM_;  // secondary name given to new collection of CSC wire digis
      std::string CSCComparatorDigiCollectionDM_; // "     "                              CSC Comparator digis

      // SiStrips
      std::string SiStripDigiCollectionDM_  ; // secondary name to be given to new SiStrip digis

      // SiPixels
      std::string PixelDigiCollectionDM_  ; // secondary name to be given to new SiPixel digis

      // merge tracker digis or tracks?
      bool MergeTrackerDigis_;

      // Submodules to handle the individual detectors

      DataMixingPileupCopy  *PUWorker_;

      DataMixingEMWorker *EMWorker_ ;
      DataMixingEMDigiWorker *EMDigiWorker_ ;
      DataMixingEcalDigiWorkerProd *EcalDigiWorkerProd_ ;
      bool MergeEMDigis_;

      // Hcal 
      
      DataMixingHcalWorker *HcalWorker_ ;
      DataMixingHcalDigiWorker *HcalDigiWorker_ ;
      DataMixingHcalDigiWorkerProd *HcalDigiWorkerProd_ ;

     // tokens needed to DataMixingHcalDigiWorkerProd
      edm::InputTag EBPileInputTag_; // InputTag for Pileup Digis collection  
      edm::InputTag EEPileInputTag_  ; // InputTag for Pileup Digis collection
      edm::InputTag ESPileInputTag_  ; // InputTag for Pileup Digis collection
      edm::InputTag HBHEPileInputTag_; // InputTag for Pileup Digis collection  
      edm::InputTag HOPileInputTag_  ; // InputTag for Pileup Digis collection
      edm::InputTag HFPileInputTag_  ; // InputTag for Pileup Digis collection
      edm::InputTag ZDCPileInputTag_ ; // InputTag for Pileup Digis collection
      edm::EDGetTokenT<HBHEDigitizerTraits::DigiCollection> tok_hbhe_;
      edm::EDGetTokenT<HODigitizerTraits::DigiCollection> tok_ho_;
      edm::EDGetTokenT<HFDigitizerTraits::DigiCollection> tok_hf_;
      edm::EDGetTokenT<ZDCDigitizerTraits::DigiCollection> tok_zdc_;
      edm::EDGetTokenT<EBDigitizerTraits::DigiCollection> tok_eb_;
      edm::EDGetTokenT<EEDigitizerTraits::DigiCollection> tok_ee_;
      edm::EDGetTokenT<ESDigitizerTraits::DigiCollection> tok_es_;

      bool MergeHcalDigis_;
      bool MergeHcalDigisProd_;

      bool MergePileup_;
      bool AddedPileup_;

      // Muons

      DataMixingMuonWorker *MuonWorker_ ;

      // Si-Strips

      DataMixingSiStripWorker *SiStripWorker_ ;
      DataMixingSiStripMCDigiWorker *SiStripMCDigiWorker_ ;
      DataMixingSiStripRawWorker *SiStripRawWorker_ ;
      bool useSiStripRawDigi_;
      bool addMCDigiNoise_;
      std::string siStripRawDigiSource_;

      // Pixels

      DataMixingSiPixelMCDigiWorker *SiPixelMCDigiWorker_ ;
      DataMixingSiPixelWorker *SiPixelWorker_ ;

      // Tracks

      DigiAccumulatorMixMod * GeneralTrackWorker_;


      // Validation

      DataMixingTrackingParticleWorker * TrackingParticleWorker_ ;

      virtual void getSubdetectorNames();  

      // internally used information : subdetectors present in input
      std::vector<std::string> Subdetectors_;

      //      unsigned int eventId_; //=0 for signal, from 1-n for pileup events

      std::string label_;

    };
}//edm

#endif