CastorDigiProducer.h

Go to the documentation of this file.

#ifndef CastorDigiProducer_h
#define CastorDigiProducer_h

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ProducerBase.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "SimCalorimetry/CaloSimAlgos/interface/CaloHitResponse.h"
#include "SimCalorimetry/CaloSimAlgos/interface/CaloTDigitizer.h"
#include "SimCalorimetry/CastorSim/src/CastorAmplifier.h"
#include "SimCalorimetry/CastorSim/src/CastorCoderFactory.h"
#include "SimCalorimetry/CastorSim/src/CastorDigitizerTraits.h"
#include "SimCalorimetry/CastorSim/src/CastorElectronicsSim.h"
#include "SimCalorimetry/CastorSim/src/CastorHitCorrection.h"
#include "SimCalorimetry/CastorSim/src/CastorHitFilter.h"
#include "SimCalorimetry/CastorSim/src/CastorShape.h"
#include "SimCalorimetry/CastorSim/src/CastorSimParameterMap.h"
#include "SimDataFormats/CaloHit/interface/PCaloHitContainer.h"
#include "SimGeneral/MixingModule/interface/DigiAccumulatorMixMod.h"

#include <vector>

namespace edm {
  class StreamID;
  class ConsumesCollector;
}  // namespace edm

namespace CLHEP {
  class HepRandomEngine;
}

class PCaloHit;
class PileUpEventPrincipal;

class CastorDigiProducer : public DigiAccumulatorMixMod {
public:
  explicit CastorDigiProducer(const edm::ParameterSet &ps, edm::ProducerBase &mixMod, edm::ConsumesCollector &iC);
  ~CastorDigiProducer() override;

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

private:
  void accumulateCaloHits(std::vector<PCaloHit> const &, int bunchCrossing);

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

  edm::InputTag theHitsProducerTag;

  typedef CaloTDigitizer<CastorDigitizerTraits> CastorDigitizer;

  CastorSimParameterMap *theParameterMap;
  CaloVShape *theCastorShape;
  CaloVShape *theCastorIntegratedShape;

  CaloHitResponse *theCastorResponse;

  CastorAmplifier *theAmplifier;
  CastorCoderFactory *theCoderFactory;
  CastorElectronicsSim *theElectronicsSim;

  CastorHitFilter theCastorHitFilter;

  CastorHitCorrection *theHitCorrection;

  CastorDigitizer *theCastorDigitizer;

  std::vector<PCaloHit> theCastorHits;

  CLHEP::HepRandomEngine *randomEngine_ = nullptr;
};

#endif