HcalTBDigiProducer.h

Go to the documentation of this file.

#ifndef SimCalorimetry_HcalTestBeam_HcalTBDigiProducer_h
#define SimCalorimetry_HcalTestBeam_HcalTBDigiProducer_h

#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "SimCalorimetry/HcalSimAlgos/interface/HcalDigitizerTraits.h"
#include "SimDataFormats/CaloHit/interface/PCaloHitContainer.h"
#include "SimCalorimetry/CaloSimAlgos/interface/CaloTDigitizer.h"
#include "SimCalorimetry/HcalTestBeam/interface/HcalTBSimParameterMap.h"
#include "SimCalorimetry/HcalSimAlgos/interface/HcalShape.h"
#include "SimCalorimetry/HcalSimAlgos/interface/HcalElectronicsSim.h"
#include "SimCalorimetry/HcalSimAlgos/interface/HBHEHitFilter.h"
#include "SimCalorimetry/HcalSimAlgos/interface/HOHitFilter.h"
#include "SimCalorimetry/CaloSimAlgos/interface/CaloHitResponse.h"
#include "SimCalorimetry/HcalSimAlgos/interface/HcalAmplifier.h"
#include "SimCalorimetry/HcalSimAlgos/interface/HcalCoderFactory.h"
#include "SimCalorimetry/HcalSimAlgos/interface/HcalHitCorrection.h"
#include "SimGeneral/MixingModule/interface/DigiAccumulatorMixMod.h"

#include<vector>
#include<string>

class PEcalTBInfo;

namespace edm {
  class StreamID;
}

namespace CLHEP {
  class HepRandomEngine;
}

class HcalTBDigiProducer : public DigiAccumulatorMixMod {
public:

  explicit HcalTBDigiProducer(const edm::ParameterSet& ps, edm::stream::EDProducerBase& mixMod, edm::ConsumesCollector& iC);
  virtual ~HcalTBDigiProducer();

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

private:
  void accumulateCaloHits(edm::Handle<std::vector<PCaloHit> > const& hits, int bunchCrossing, CLHEP::HepRandomEngine*);

  void sortHits(const edm::PCaloHitContainer & hits);
  void checkGeometry(const edm::EventSetup& eventSetup);
  void updateGeometry();

  void setPhaseShift(const DetId & detId);

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

private:

  typedef CaloTDigitizer<HBHEDigitizerTraits> HBHEDigitizer;
  typedef CaloTDigitizer<HODigitizerTraits>   HODigitizer;

  HcalTBSimParameterMap * theParameterMap;
  CaloVShape * theHcalShape;
  CaloVShape * theHcalIntegratedShape;

  CaloHitResponse * theHBHEResponse;
  CaloHitResponse * theHOResponse;

  HcalAmplifier * theAmplifier;
  HcalCoderFactory * theCoderFactory;
  HcalElectronicsSim * theElectronicsSim;

  HBHEHitFilter theHBHEHitFilter;
  HOHitFilter   theHOHitFilter;

  HcalHitCorrection * theHitCorrection;

  HBHEDigitizer * theHBHEDigitizer;
  HODigitizer   * theHODigitizer;
 
  const CaloGeometry * theGeometry;
  std::vector<DetId>   hbheCells;
  std::vector<DetId>   hoCells;

  std::vector<PCaloHit> theHBHEHits, theHOHits;

  std::string ecalTBInfoLabel;
  double thisPhaseShift;

  bool   doPhaseShift;
  double tunePhaseShift;

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

#endif