DataMixingHcalDigiWorkerProd.h

Go to the documentation of this file.

#ifndef DataMixingHcalDigiWorkerProd_h
#define SimDataMixingHcalDigiWorkerProd_h

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

#include "CalibFormats/CaloObjects/interface/CaloSamples.h"
#include "DataFormats/Provenance/interface/ProductID.h"
#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/HcalDigi/interface/HcalDigiCollections.h"
#include "DataFormats/HcalDigi/interface/HBHEDataFrame.h"
#include "DataFormats/HcalDigi/interface/HODataFrame.h"
#include "DataFormats/HcalDigi/interface/HFDataFrame.h"
#include "SimCalorimetry/HcalSimProducers/interface/HcalDigiProducer.h"
#include "SimCalorimetry/HcalSimAlgos/interface/HcalSignalGenerator.h"
#include "SimGeneral/DataMixingModule/plugins/HcalNoiseStorage.h"

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


namespace edm
{
  class ConsumesCollector;
  class ModuleCallingContext;

  class DataMixingHcalDigiWorkerProd
    {
    public:

      explicit DataMixingHcalDigiWorkerProd(const edm::ParameterSet& ps, edm::ConsumesCollector&& iC);

      virtual ~DataMixingHcalDigiWorkerProd();

      void putHcal(edm::Event &e,const edm::EventSetup& ES);
      void addHcalSignals(const edm::Event &e,const edm::EventSetup& ES); 
      void addHcalPileups(const int bcr, const edm::EventPrincipal*,unsigned int EventId,
                          const edm::EventSetup& ES, edm::ModuleCallingContext const*);

    // set tokens for data access
    void setHBHEAccess( edm::EDGetTokenT<HBHEDigitizerTraits::DigiCollection> tok) { tok_hbhe_ = tok; }
    void setHOAccess( edm::EDGetTokenT<HODigitizerTraits::DigiCollection> tok) { tok_ho_ = tok; }
    void setHFAccess( edm::EDGetTokenT<HFDigitizerTraits::DigiCollection> tok) { tok_hf_ = tok; }
    void setZDCAccess( edm::EDGetTokenT<ZDCDigitizerTraits::DigiCollection> tok) { tok_zdc_ = tok; }
    void beginRun(const edm::Run& run, const edm::EventSetup& ES);
    void initializeEvent(const edm::Event &e, const edm::EventSetup& ES);

    private:
      // data specifiers

      // Hcal
      //      edm::InputTag HBHEdigiCollectionSig_; // secondary name given to collection of digis
      // edm::InputTag HOdigiCollectionSig_  ; // secondary name given to collection of digis
      //edm::InputTag HFdigiCollectionSig_  ; // secondary name given to collection of digis
      //edm::InputTag ZDCdigiCollectionSig_ ; // secondary name given to collection of digis
      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
      std::string HBHEDigiCollectionDM_; // secondary name to be given to collection of digis
      std::string HODigiCollectionDM_  ; // secondary name to be given to collection of digis
      std::string HFDigiCollectionDM_  ; // secondary name to be given to collection of digis
      std::string ZDCDigiCollectionDM_ ; // secondary name to be given to collection of digis

      edm::EDGetTokenT<HBHEDigitizerTraits::DigiCollection> tok_hbhe_;
      edm::EDGetTokenT<HODigitizerTraits::DigiCollection> tok_ho_;
      edm::EDGetTokenT<HFDigitizerTraits::DigiCollection> tok_hf_;
      edm::EDGetTokenT<ZDCDigitizerTraits::DigiCollection> tok_zdc_;
  

      HcalDigiProducer* myHcalDigitizer_;
      HBHESignalGenerator theHBHESignalGenerator;
      HOSignalGenerator theHOSignalGenerator;
      HFSignalGenerator theHFSignalGenerator;
      ZDCSignalGenerator theZDCSignalGenerator;

      std::string label_;

    };
}//edm

#endif