EcalTBDigiProducer.cc

Go to the documentation of this file.

#include <memory>

#include "DataFormats/Common/interface/Handle.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "Geometry/CaloGeometry/interface/CaloGeometry.h"
#include "SimCalorimetry/EcalSimAlgos/interface/EBHitResponse.h"
#include "SimCalorimetry/EcalSimAlgos/interface/EEHitResponse.h"
#include "SimCalorimetry/EcalSimAlgos/interface/EcalSimParameterMap.h"
#include "SimCalorimetry/EcalTestBeam/interface/EcalTBDigiProducer.h"
#include "SimDataFormats/EcalTestBeam/interface/PEcalTBInfo.h"

EcalTBDigiProducer::EcalTBDigiProducer(const edm::ParameterSet &params,
                                       edm::ProducesCollector producesCollector,
                                       edm::ConsumesCollector &iC)
    : EcalDigiProducer(params, producesCollector, iC), m_geometryToken(iC.esConsumes()) {
  std::string const instance("simEcalUnsuppressedDigis");
  m_EBdigiFinalTag = params.getParameter<std::string>("EBdigiFinalCollection");
  m_EBdigiTempTag = params.getParameter<std::string>("EBdigiCollection");

  producesCollector.produces<EBDigiCollection>(instance + m_EBdigiFinalTag);  // after selective readout
  producesCollector.produces<EcalTBTDCRawInfo>(instance);

  const bool syncPhase(params.getParameter<bool>("syncPhase"));

  // possible phase shift for asynchronous trigger (e.g. test-beam)

  m_doPhaseShift = !syncPhase;
  m_thisPhaseShift = 1.;

  typedef std::vector<edm::ParameterSet> Parameters;
  Parameters ranges = params.getParameter<Parameters>("tdcRanges");
  for (Parameters::iterator itRanges = ranges.begin(); itRanges != ranges.end(); ++itRanges) {
    EcalTBTDCRecInfoAlgo::EcalTBTDCRanges aRange;
    aRange.runRanges.first = itRanges->getParameter<int>("startRun");
    aRange.runRanges.second = itRanges->getParameter<int>("endRun");
    aRange.tdcMin = itRanges->getParameter<std::vector<double>>("tdcMin");
    aRange.tdcMax = itRanges->getParameter<std::vector<double>>("tdcMax");
    m_tdcRanges.push_back(std::move(aRange));
  }

  m_use2004OffsetConvention = params.getUntrackedParameter<bool>("use2004OffsetConvention", false);

  m_ecalTBInfoLabel = params.getUntrackedParameter<std::string>("EcalTBInfoLabel", "SimEcalTBG4Object");

  m_doReadout = params.getParameter<bool>("doReadout");

  m_theTBReadout = new EcalTBReadout(m_ecalTBInfoLabel);

  m_tunePhaseShift = params.getParameter<double>("tunePhaseShift");

  if (m_doPhaseShift) {
    iC.consumes<PEcalTBInfo>(
        edm::InputTag(params.getUntrackedParameter<std::string>("EcalTBInfoLabel", "SimEcalTBG4Object")));
  }
}

EcalTBDigiProducer::~EcalTBDigiProducer() {}

void EcalTBDigiProducer::initializeEvent(edm::Event const &event, edm::EventSetup const &eventSetup) {
  std::cout << "====****Entering EcalTBDigiProducer produce()" << std::endl;
  const std::vector<DetId> &theBarrelDets = eventSetup.getData(m_geometryToken).getValidDetIds(DetId::Ecal, EcalBarrel);

  m_theTBReadout->setDetIds(theBarrelDets);

  m_TDCproduct = std::make_unique<EcalTBTDCRawInfo>(1);
  if (m_doPhaseShift) {
    edm::Handle<PEcalTBInfo> theEcalTBInfo;
    event.getByLabel(m_ecalTBInfoLabel, theEcalTBInfo);
    m_thisPhaseShift = theEcalTBInfo->phaseShift();

    DetId detId(DetId::Ecal, 1);
    setPhaseShift(detId);

    fillTBTDCRawInfo(*m_TDCproduct);  // fill the TDC info in the event
  }
  EcalDigiProducer::initializeEvent(event, eventSetup);
}

void EcalTBDigiProducer::finalizeEvent(edm::Event &event, const edm::EventSetup &eventSetup) {
  m_ebDigis = std::make_unique<EBDigiCollection>();

  EcalDigiProducer::finalizeEvent(event, eventSetup);

  const EBDigiCollection *barrelResult(&*m_ebDigis);

  std::unique_ptr<EBDigiCollection> barrelReadout(new EBDigiCollection());
  if (m_doReadout) {
    m_theTBReadout->performReadout(event, m_theTTmap, *barrelResult, *barrelReadout);
  } else {
    *barrelReadout = *barrelResult;
  }

  std::cout << "===**** EcalTBDigiProducer: number of barrel digis = " << barrelReadout->size() << std::endl;

  std::string const instance("simEcalUnsuppressedDigis");
  event.put(std::move(barrelReadout), instance + m_EBdigiFinalTag);
  event.put(std::move(m_TDCproduct), instance);

  m_ebDigis.reset();  // release memory
  m_eeDigis.reset();  // release memory
}

void EcalTBDigiProducer::setPhaseShift(const DetId &detId) {
  const CaloSimParameters &parameters(EcalDigiProducer::m_ParameterMap->simParameters(detId));

  if (!parameters.syncPhase()) {
    const int myDet(detId.subdetId());

    LogDebug("EcalDigi") << "Setting the phase shift " << m_thisPhaseShift << " and the offset " << m_tunePhaseShift
                         << " for the subdetector " << myDet;

    if (myDet == 1) {
      double passPhaseShift(m_thisPhaseShift + m_tunePhaseShift);
      if (m_use2004OffsetConvention)
        passPhaseShift = 1. - passPhaseShift;
      EcalDigiProducer::m_EBResponse->setPhaseShift(passPhaseShift);
      EcalDigiProducer::m_EEResponse->setPhaseShift(passPhaseShift);
    }
  }
}

void EcalTBDigiProducer::fillTBTDCRawInfo(EcalTBTDCRawInfo &theTBTDCRawInfo) {
  const unsigned int thisChannel(1);

  const unsigned int thisCount((unsigned int)(m_thisPhaseShift * (m_tdcRanges[0].tdcMax[0] - m_tdcRanges[0].tdcMin[0]) +
                                              m_tdcRanges[0].tdcMin[0]));

  EcalTBTDCSample theTBTDCSample(thisChannel, thisCount);

  const unsigned int sampleIndex(0);
  theTBTDCRawInfo.setSample(sampleIndex, theTBTDCSample);

  LogDebug("EcalDigi") << theTBTDCSample << "\n" << theTBTDCRawInfo;
}

void EcalTBDigiProducer::cacheEBDigis(const EBDigiCollection *ebDigiPtr) const {
  m_ebDigis = std::make_unique<EBDigiCollection>();
  *m_ebDigis = *ebDigiPtr;
}

void EcalTBDigiProducer::cacheEEDigis(const EEDigiCollection *eeDigiPtr) const {
  std::cout << "===**** EcalTBDigiProducer: number of endcap digis = " << eeDigiPtr->size() << std::endl;
}