Phase2TrackerRecHits.cc

Go to the documentation of this file.

#include "FWCore/Framework/interface/global/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/PluginManager/interface/ModuleDef.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/Framework/interface/ESHandle.h"

#include "Geometry/CommonDetUnit/interface/GeomDet.h"
#include "Geometry/Records/interface/TrackerDigiGeometryRecord.h"
#include "Geometry/TrackerGeometryBuilder/interface/TrackerGeometry.h"
#include "Geometry/TrackerGeometryBuilder/interface/PixelGeomDetUnit.h"

#include "DataFormats/Common/interface/DetSetVector.h"
#include "DataFormats/Phase2TrackerCluster/interface/Phase2TrackerCluster1D.h"
#include "DataFormats/TrackerRecHit2D/interface/Phase2TrackerRecHit1D.h"
#include "DataFormats/TrackerCommon/interface/TrackerTopology.h"
#include "DataFormats/DetId/interface/DetId.h"

#include "RecoLocalTracker/Records/interface/TkStripCPERecord.h"
#include "RecoLocalTracker/Phase2TrackerRecHits/interface/Phase2StripCPE.h"

#include <vector>
#include <string>

class Phase2TrackerRecHits : public edm::global::EDProducer<> {

        public:

            explicit Phase2TrackerRecHits(const edm::ParameterSet& conf);
            ~Phase2TrackerRecHits() override {};
            void produce(edm::StreamID sid, edm::Event& event, const edm::EventSetup& eventSetup) const     final;

        private:

            edm::EDGetTokenT< Phase2TrackerCluster1DCollectionNew > token_;
            edm::ESInputTag cpeTag_;

};

Phase2TrackerRecHits::Phase2TrackerRecHits(edm::ParameterSet const& conf) : 
  token_(consumes< Phase2TrackerCluster1DCollectionNew >(conf.getParameter<edm::InputTag>("src"))),
  cpeTag_(conf.getParameter<edm::ESInputTag>("Phase2StripCPE")) {
    produces<Phase2TrackerRecHit1DCollectionNew>();
}

void Phase2TrackerRecHits::produce(edm::StreamID sid, edm::Event& event, const edm::EventSetup& eventSetup) const {

  // Get the Clusters
  edm::Handle< Phase2TrackerCluster1DCollectionNew > clusters;
  event.getByToken(token_, clusters);

  // load the cpe via the eventsetup
  edm::ESHandle<ClusterParameterEstimator<Phase2TrackerCluster1D> > cpe;
  eventSetup.get<TkStripCPERecord>().get(cpeTag_, cpe);

  // Get the geometry
  edm::ESHandle< TrackerGeometry > geomHandle;
  eventSetup.get< TrackerDigiGeometryRecord >().get(geomHandle);
  const TrackerGeometry* tkGeom(&(*geomHandle));

  // Global container for the RecHits of each module
  auto outputRecHits = std::make_unique<Phase2TrackerRecHit1DCollectionNew>();

  // Loop over clusters
  for (const auto &clusterDetSet : *clusters) { 
  
    DetId detId(clusterDetSet.detId());

    // Geometry
    const GeomDetUnit * geomDetUnit(tkGeom->idToDetUnit(detId));

    // Container for the clusters that will be produced for this modules
    Phase2TrackerRecHit1DCollectionNew::FastFiller rechits(*outputRecHits, clusterDetSet.detId());

    for (const auto &clusterRef : clusterDetSet) { 
      ClusterParameterEstimator< Phase2TrackerCluster1D >::LocalValues lv = cpe->localParameters(clusterRef, *geomDetUnit);

      // Create a persistent edm::Ref to the cluster
      edm::Ref< Phase2TrackerCluster1DCollectionNew, Phase2TrackerCluster1D > cluster = edmNew::makeRefTo(clusters, &clusterRef);

      // Make a RecHit and add it to the DetSet
      Phase2TrackerRecHit1D hit(lv.first, lv.second, *geomDetUnit, cluster);

      rechits.push_back(hit);
    }
  }

  outputRecHits->shrink_to_fit();
  event.put(std::move(outputRecHits));

}

DEFINE_FWK_MODULE(Phase2TrackerRecHits);