TrackingRecHitAlgorithm.cc

Go to the documentation of this file.

#include "FastSimulation/TrackingRecHitProducer/interface/TrackingRecHitAlgorithm.h"

#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "FWCore/Utilities/interface/RandomNumberGenerator.h"

#include "DataFormats/DetId/interface/DetId.h"
#include "DataFormats/TrackerRecHit2D/interface/FastSingleTrackerRecHit.h"

#include "SimDataFormats/TrackingHit/interface/PSimHit.h"

#include "FastSimulation/TrackingRecHitProducer/interface/TrackingRecHitProduct.h"

#include "Geometry/Records/interface/TrackerDigiGeometryRecord.h"
#include "Geometry/Records/interface/TrackerTopologyRcd.h"

#include "FWCore/Utilities/interface/Exception.h"

#include "FWCore/MessageLogger/interface/MessageLogger.h"

TrackingRecHitAlgorithm::TrackingRecHitAlgorithm(
    const std::string& name,
    const edm::ParameterSet& config,
    edm::ConsumesCollector& consumesCollector
):
    _name(name),
    _selectionString(config.getParameter<std::string>("select")),
    _trackerTopology(nullptr),
    _trackerGeometry(nullptr),
    _misalignedTrackerGeometry(nullptr),
    _randomEngine(nullptr)
{
}

const TrackerTopology& TrackingRecHitAlgorithm::getTrackerTopology() const
{
    if (!_trackerTopology)
    {
        throw cms::Exception("TrackingRecHitAlgorithm ") << _name <<": TrackerTopology not defined";
    }
    return *_trackerTopology;
}

const TrackerGeometry& TrackingRecHitAlgorithm::getTrackerGeometry() const
{
    if (!_trackerGeometry)
    {
        throw cms::Exception("TrackingRecHitAlgorithm ") << _name <<": TrackerGeometry not defined";
    }
    return *_trackerGeometry;
}

const TrackerGeometry& TrackingRecHitAlgorithm::getMisalignedGeometry() const
{
    if (!_misalignedTrackerGeometry)
    {
        throw cms::Exception("TrackingRecHitAlgorithm ") << _name <<": MisalignedGeometry not defined";
    }
    return *_misalignedTrackerGeometry;
}

const RandomEngineAndDistribution& TrackingRecHitAlgorithm::getRandomEngine() const
{
    if (!_randomEngine)
    {
        throw cms::Exception("TrackingRecHitAlgorithm ") << _name <<": RandomEngineAndDistribution not defined";
    }
    return *_randomEngine;
}

void TrackingRecHitAlgorithm::beginStream(const edm::StreamID& id)
{
  _randomEngine = std::make_shared<RandomEngineAndDistribution>(id);
}

void TrackingRecHitAlgorithm::beginEvent(edm::Event& event, const edm::EventSetup& eventSetup)
{
    edm::ESHandle<TrackerTopology> trackerTopologyHandle;
    edm::ESHandle<TrackerGeometry> trackerGeometryHandle;
    edm::ESHandle<TrackerGeometry> misalignedGeometryHandle;

    eventSetup.get<TrackerTopologyRcd>().get(trackerTopologyHandle);
    eventSetup.get<TrackerDigiGeometryRecord>().get(trackerGeometryHandle);
    eventSetup.get<TrackerDigiGeometryRecord>().get("MisAligned",misalignedGeometryHandle);

    _trackerTopology = trackerTopologyHandle.product();
    _trackerGeometry = trackerGeometryHandle.product();
    _misalignedTrackerGeometry = misalignedGeometryHandle.product();

}

TrackingRecHitProductPtr TrackingRecHitAlgorithm::process(TrackingRecHitProductPtr product) const
{
    return product;
}

void TrackingRecHitAlgorithm::endEvent(edm::Event& event, const edm::EventSetup& eventSetup)
{
    //set these to 0 -> ensures that beginEvent needs to be executed before accessing these pointers again
    _trackerGeometry=nullptr;
    _trackerTopology=nullptr;
    _misalignedTrackerGeometry=nullptr;
}

void TrackingRecHitAlgorithm::endStream()
{
    _randomEngine.reset();
}

TrackingRecHitAlgorithm::~TrackingRecHitAlgorithm()
{
}