SiStripDigitizer.h

Go to the documentation of this file.

#ifndef SiStripDigitizer_h
#define SiStripDigitizer_h

#include <map>
#include <memory>
#include <string>
#include <vector>
#include <bitset>
#include "CalibFormats/SiStripObjects/interface/SiStripGain.h"
#include "CalibFormats/SiStripObjects/interface/SiStripDetCabling.h"
#include "CalibTracker/Records/interface/SiStripDependentRecords.h"
#include "CondFormats/DataRecord/interface/SiStripCondDataRecords.h"
#include "CondFormats/SiStripObjects/interface/SiStripApvSimulationParameters.h"
#include "CondFormats/SiStripObjects/interface/SiStripNoises.h"
#include "CondFormats/SiStripObjects/interface/SiStripPedestals.h"
#include "CondFormats/SiStripObjects/interface/SiStripThreshold.h"
#include "FWCore/Framework/interface/ProducesCollector.h"
#include "FWCore/Utilities/interface/ESGetToken.h"
#include "Geometry/Records/interface/TrackerDigiGeometryRecord.h"
#include "Geometry/Records/interface/TrackerTopologyRcd.h"
#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"
#include "SimDataFormats/PileupSummaryInfo/interface/PileupMixingContent.h"
#include "SimGeneral/MixingModule/interface/DigiAccumulatorMixMod.h"

class TrackerTopology;

namespace CLHEP {
  class HepRandomEngine;
}

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

class MagneticField;
class PileUpEventPrincipal;
class PSimHit;
class SiStripDigitizerAlgorithm;
class StripGeomDetUnit;
class TrackerGeometry;

class SiStripDigitizer : public DigiAccumulatorMixMod {
public:
  explicit SiStripDigitizer(const edm::ParameterSet& conf, edm::ProducesCollector, edm::ConsumesCollector& iC);

  ~SiStripDigitizer() 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;

  void StorePileupInformation(std::vector<int>& numInteractionList,
                              std::vector<int>& bunchCrossingList,
                              std::vector<float>& TrueInteractionList,
                              std::vector<edm::EventID>& eventInfoList,
                              int bunchSpacing) override {
    PileupInfo_ = std::make_unique<PileupMixingContent>(
        numInteractionList, bunchCrossingList, TrueInteractionList, eventInfoList, bunchSpacing);
  }

  PileupMixingContent* getEventPileupInfo() override { return PileupInfo_.get(); }

private:
  void accumulateStripHits(edm::Handle<std::vector<PSimHit>>,
                           const TrackerTopology* tTopo,
                           size_t globalSimHitIndex,
                           const unsigned int tofBin);

  typedef std::vector<std::string> vstring;
  typedef std::map<unsigned int, std::vector<std::pair<const PSimHit*, int>>, std::less<unsigned int>> simhit_map;
  typedef simhit_map::iterator simhit_map_iterator;

  const std::string hitsProducer;
  const vstring trackerContainers;
  const std::string ZSDigi;
  const std::string SCDigi;
  const std::string VRDigi;
  const std::string PRDigi;
  const bool useConfFromDB;
  const bool zeroSuppression;
  const bool makeDigiSimLinks_;
  const bool includeAPVSimulation_;
  const double fracOfEventsToSimAPV_;
  const edm::ESGetToken<TrackerTopology, TrackerTopologyRcd> tTopoToken_;
  const edm::ESGetToken<TrackerGeometry, TrackerDigiGeometryRecord> pDDToken_;
  const edm::ESGetToken<MagneticField, IdealMagneticFieldRecord> pSetupToken_;
  const edm::ESGetToken<SiStripGain, SiStripGainSimRcd> gainToken_;
  const edm::ESGetToken<SiStripNoises, SiStripNoisesRcd> noiseToken_;
  const edm::ESGetToken<SiStripThreshold, SiStripThresholdRcd> thresholdToken_;
  const edm::ESGetToken<SiStripPedestals, SiStripPedestalsRcd> pedestalToken_;
  edm::ESGetToken<SiStripDetCabling, SiStripDetCablingRcd> detCablingToken_;
  edm::ESGetToken<SiStripApvSimulationParameters, SiStripApvSimulationParametersRcd> apvSimulationParametersToken_;


  std::map<std::string, size_t> crossingSimHitIndexOffset_;

  std::unique_ptr<SiStripDigitizerAlgorithm> theDigiAlgo;
  std::map<uint32_t, std::vector<int>> theDetIdList;
  const TrackerGeometry* pDD = nullptr;
  const MagneticField* pSetup = nullptr;
  std::map<unsigned int, StripGeomDetUnit const*> detectorUnits;
  CLHEP::HepRandomEngine* randomEngine_ = nullptr;
  std::vector<std::pair<int, std::bitset<6>>> theAffectedAPVvector;

  std::unique_ptr<PileupMixingContent> PileupInfo_;
};

#endif