Back to home page

Project CMSSW displayed by LXR

 
 

    


File indexing completed on 2021-11-05 00:08:04

0001 #include "FWCore/Framework/interface/Frameworkfwd.h"
0002 #include "FWCore/Framework/interface/global/EDProducer.h"
0003 
0004 #include "FWCore/Framework/interface/Event.h"
0005 #include "FWCore/Framework/interface/MakerMacros.h"
0006 
0007 #include "FWCore/ParameterSet/interface/ParameterSet.h"
0008 #include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
0009 #include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
0010 #include "FWCore/Utilities/interface/EDGetToken.h"
0011 #include "FWCore/MessageLogger/interface/MessageLogger.h"
0012 
0013 #include "RecoPixelVertexing/PixelTrackFitting/interface/PixelTrackFilter.h"
0014 #include "RecoHI/HiTracking/interface/HIPixelTrackFilter.h"
0015 #include "DataFormats/SiPixelCluster/interface/SiPixelClusterShapeCache.h"
0016 #include "Geometry/Records/interface/TrackerDigiGeometryRecord.h"
0017 #include "RecoTracker/Record/interface/CkfComponentsRecord.h"
0018 
0019 class HIPixelTrackFilterProducer : public edm::global::EDProducer<> {
0020 public:
0021   explicit HIPixelTrackFilterProducer(const edm::ParameterSet& iConfig);
0022   ~HIPixelTrackFilterProducer() override;
0023 
0024   static void fillDescriptions(edm::ConfigurationDescriptions& descriptions);
0025 
0026 private:
0027   void produce(edm::StreamID, edm::Event& iEvent, const edm::EventSetup& iSetup) const override;
0028 
0029   edm::EDGetTokenT<SiPixelClusterShapeCache> theClusterShapeCacheToken;
0030   edm::EDGetTokenT<reco::VertexCollection> theVertexCollectionToken;
0031   edm::ESGetToken<TrackerGeometry, TrackerDigiGeometryRecord> theTrackerToken;
0032   edm::ESGetToken<ClusterShapeHitFilter, CkfComponentsRecord> theShapeToken;
0033   edm::ESGetToken<TrackerTopology, TrackerTopologyRcd> theTopoToken;
0034   double theTIPMax, theNSigmaTipMaxTolerance;
0035   double theLIPMax, theNSigmaLipMaxTolerance;
0036   double theChi2Max, thePtMin, thePtMax;
0037   bool useClusterShape;
0038 };
0039 
0040 HIPixelTrackFilterProducer::HIPixelTrackFilterProducer(const edm::ParameterSet& iConfig)
0041     : theClusterShapeCacheToken(
0042           consumes<SiPixelClusterShapeCache>(iConfig.getParameter<edm::InputTag>("clusterShapeCacheSrc"))),
0043       theVertexCollectionToken(
0044           consumes<reco::VertexCollection>(iConfig.getParameter<edm::InputTag>("VertexCollection"))),
0045       theTrackerToken(esConsumes()),
0046       theShapeToken(esConsumes(edm::ESInputTag("", "ClusterShapeHitFilter"))),
0047       theTopoToken(esConsumes()),
0048       theTIPMax(iConfig.getParameter<double>("tipMax")),
0049       theNSigmaTipMaxTolerance(iConfig.getParameter<double>("nSigmaTipMaxTolerance")),
0050       theLIPMax(iConfig.getParameter<double>("lipMax")),
0051       theNSigmaLipMaxTolerance(iConfig.getParameter<double>("nSigmaLipMaxTolerance")),
0052       theChi2Max(iConfig.getParameter<double>("chi2")),
0053       thePtMin(iConfig.getParameter<double>("ptMin")),
0054       thePtMax(iConfig.getParameter<double>("ptMax")),
0055       useClusterShape(iConfig.getParameter<bool>("useClusterShape")) {
0056   produces<PixelTrackFilter>();
0057 }
0058 
0059 void HIPixelTrackFilterProducer::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
0060   edm::ParameterSetDescription desc;
0061 
0062   desc.add<edm::InputTag>("clusterShapeCacheSrc", edm::InputTag("siPixelClusterShapeCache"));
0063   desc.add<edm::InputTag>("VertexCollection", edm::InputTag("hiSelectedPixelVertex"));
0064   desc.add<double>("ptMin", 1.5);
0065   desc.add<double>("ptMax", 999999.);
0066   desc.add<double>("tipMax", 0);
0067   desc.add<double>("nSigmaTipMaxTolerance", 6.0);
0068   desc.add<double>("lipMax", 0.3);
0069   desc.add<double>("nSigmaLipMaxTolerance", 0);
0070   desc.add<double>("chi2", 1000);
0071   desc.add<bool>("useClusterShape", false);
0072 
0073   descriptions.add("hiPixelTrackFilter", desc);
0074 }
0075 
0076 HIPixelTrackFilterProducer::~HIPixelTrackFilterProducer() {}
0077 
0078 void HIPixelTrackFilterProducer::produce(edm::StreamID, edm::Event& iEvent, const edm::EventSetup& iSetup) const {
0079   edm::Handle<SiPixelClusterShapeCache> cache;
0080   iEvent.getByToken(theClusterShapeCacheToken, cache);
0081 
0082   edm::Handle<reco::VertexCollection> vc;
0083   iEvent.getByToken(theVertexCollectionToken, vc);
0084   const reco::VertexCollection* vertices = vc.product();
0085 
0086   if (!vertices->empty()) {
0087     edm::LogInfo("HeavyIonVertexing") << "[HIPixelTrackFilterProducer] Pixel track selection based on best vertex"
0088                                       << "\n   vz = " << vertices->begin()->z()
0089                                       << "\n   vz sigma = " << vertices->begin()->zError();
0090   } else {
0091     edm::EDConsumerBase::Labels labels;
0092     labelsForToken(theVertexCollectionToken, labels);
0093 
0094     edm::LogError("HeavyIonVertexing")  // this can be made a warning when operator() is fixed
0095         << "No vertex found in collection '" << labels.module << "'";
0096   }
0097 
0098   auto impl = std::make_unique<HIPixelTrackFilter>(cache.product(),
0099                                                    thePtMin,
0100                                                    thePtMax,
0101                                                    &iSetup.getData(theTrackerToken),
0102                                                    &iSetup.getData(theShapeToken),
0103                                                    &iSetup.getData(theTopoToken),
0104                                                    vertices,
0105                                                    theTIPMax,
0106                                                    theNSigmaTipMaxTolerance,
0107                                                    theLIPMax,
0108                                                    theNSigmaLipMaxTolerance,
0109                                                    theChi2Max,
0110                                                    useClusterShape);
0111   auto prod = std::make_unique<PixelTrackFilter>(std::move(impl));
0112   iEvent.put(std::move(prod));
0113 }
0114 
0115 DEFINE_FWK_MODULE(HIPixelTrackFilterProducer);