EcalDigiProducer.h

Go to the documentation of this file.

#ifndef SimCalorimetry_EcalSimProducers_EcalDigiProducer_h
#define SimCalorimetry_EcalSimProducers_EcalDigiProducer_h

#include "SimCalorimetry/EcalSimAlgos/interface/APDShape.h"
#include "SimCalorimetry/EcalSimAlgos/interface/EBShape.h"
#include "SimCalorimetry/EcalSimAlgos/interface/EEShape.h"
#include "SimCalorimetry/EcalSimAlgos/interface/ESShape.h"
#include "DataFormats/Math/interface/Error.h"
#include "SimGeneral/NoiseGenerators/interface/CorrelatedNoisifier.h"
#include "SimCalorimetry/EcalSimAlgos/interface/EcalCorrelatedNoiseMatrix.h"
#include "SimCalorimetry/EcalSimAlgos/interface/ESElectronicsSim.h"

#include "SimCalorimetry/CaloSimAlgos/interface/CaloTDigitizer.h"
#include "SimCalorimetry/EcalSimAlgos/interface/EcalTDigitizer.h"
#include "SimCalorimetry/EcalSimAlgos/interface/EcalDigitizerTraits.h"
#include "SimGeneral/MixingModule/interface/DigiAccumulatorMixMod.h"


#include <vector>

typedef EcalTDigitizer<EBDigitizerTraits> EBDigitizer  ;
typedef EcalTDigitizer<EEDigitizerTraits> EEDigitizer  ;
typedef CaloTDigitizer<ESOldDigitizerTraits> ESOldDigitizer  ;

class ESDigitizer ;

class APDSimParameters ;
class EBHitResponse ;
class EEHitResponse ;
class ESHitResponse ;
class CaloHitResponse ;
class EcalSimParameterMap ;
class EcalCoder ;
class EcalElectronicsSim ;
class ESElectronicsSim ;
class ESElectronicsSimFast ;
class EcalBaseSignalGenerator;
class CaloGeometry ;
class EBDigiCollection ;
class EEDigiCollection ;
class ESDigiCollection ;
class PileUpEventPrincipal ;

namespace edm {
  class ConsumesCollector;
  namespace one {
    class EDProducerBase;
  }
  class Event;
  class EventSetup;
  template<typename T> class Handle;
  class ParameterSet;
  class StreamID;
}

namespace CLHEP {
  class HepRandomEngine;
}

class EcalDigiProducer : public DigiAccumulatorMixMod {
   public:

      EcalDigiProducer( const edm::ParameterSet& params , edm::one::EDProducerBase& mixMod, edm::ConsumesCollector& iC);
      EcalDigiProducer( const edm::ParameterSet& params , edm::ConsumesCollector& iC);
      virtual ~EcalDigiProducer();

      virtual void initializeEvent(edm::Event const& e, edm::EventSetup const& c);
      virtual void accumulate(edm::Event const& e, edm::EventSetup const& c);
      virtual void accumulate(PileUpEventPrincipal const& e, edm::EventSetup const& c, edm::StreamID const&) override;
      virtual void finalizeEvent(edm::Event& e, edm::EventSetup const& c);
      virtual void beginLuminosityBlock(edm::LuminosityBlock const& lumi, edm::EventSetup const& setup) override;

      void setEBNoiseSignalGenerator(EcalBaseSignalGenerator * noiseGenerator);
      void setEENoiseSignalGenerator(EcalBaseSignalGenerator * noiseGenerator);
      void setESNoiseSignalGenerator(EcalBaseSignalGenerator * noiseGenerator);

   private:

      virtual void cacheEBDigis( const EBDigiCollection* ebDigiPtr ) const { }
      virtual void cacheEEDigis( const EEDigiCollection* eeDigiPtr ) const { }

      typedef edm::Handle<std::vector<PCaloHit> > HitsHandle;
      void accumulateCaloHits(HitsHandle const& ebHandle, HitsHandle const& eeHandle, HitsHandle const& esHandle, int bunchCrossing, CLHEP::HepRandomEngine*);

      void checkGeometry(const edm::EventSetup& eventSetup) ;

      void updateGeometry() ;

      void checkCalibrations(const edm::Event& event, const edm::EventSetup& eventSetup) ;

      CLHEP::HepRandomEngine* randomEngine(edm::StreamID const& streamID);

      const APDShape m_APDShape ;
      const EBShape  m_EBShape  ;
      const EEShape  m_EEShape  ;
      ESShape        m_ESShape  ; // no const because gain must be set

      const std::string m_EBdigiCollection ;
      const std::string m_EEdigiCollection ;
      const std::string m_ESdigiCollection ;
      const std::string m_hitsProducerTag  ;

      bool  m_useLCcorrection;

      const bool m_apdSeparateDigi ;

      const double m_EBs25notCont ;
      const double m_EEs25notCont ;

      const unsigned int         m_readoutFrameSize ;
   protected:
      const EcalSimParameterMap* m_ParameterMap  ;
   private:
      const std::string          m_apdDigiTag    ;
      const APDSimParameters*    m_apdParameters ;

      EBHitResponse* m_APDResponse ;
   protected:
      EBHitResponse* m_EBResponse ;
      EEHitResponse* m_EEResponse ;
   private:
      ESHitResponse* m_ESResponse ;
      CaloHitResponse* m_ESOldResponse ;

      const bool m_addESNoise ;
      const bool m_PreMix1 ;
      const bool m_PreMix2 ;

      const bool m_doFastES   ;

      ESElectronicsSim*     m_ESElectronicsSim     ;
      ESOldDigitizer*       m_ESOldDigitizer       ;
      ESElectronicsSimFast* m_ESElectronicsSimFast ;
      ESDigitizer*          m_ESDigitizer          ;

      EBDigitizer*          m_APDDigitizer ;
      EBDigitizer*          m_BarrelDigitizer ;
      EEDigitizer*          m_EndcapDigitizer ;

      EcalElectronicsSim*   m_ElectronicsSim ;
      EcalCoder*            m_Coder ;

      EcalElectronicsSim*   m_APDElectronicsSim ;
      EcalCoder*            m_APDCoder ;

      const CaloGeometry*   m_Geometry ;

      CorrelatedNoisifier<EcalCorrMatrix>* m_EBCorrNoise[3] ;
      CorrelatedNoisifier<EcalCorrMatrix>* m_EECorrNoise[3] ;

      std::vector<CLHEP::HepRandomEngine*> randomEngines_;
};

#endif