InterestingTrackEcalDetIdProducer.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:    InterestingTrackEcalDetIdProducer
// Class:      InterestingTrackEcalDetIdProducer
// 
//
// Original Author:  
//         Created:  Wed Sep 22 17:02:51 CEST 2010
//
//


// system include files
#include <memory>

// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "Geometry/CaloEventSetup/interface/CaloTopologyRecord.h"
#include "Geometry/CaloTopology/interface/CaloTopology.h"
#include "Geometry/CaloTopology/interface/CaloSubdetectorTopology.h"

#include "DataFormats/DetId/interface/DetIdCollection.h"
#include "DataFormats/EcalRecHit/interface/EcalRecHitCollections.h"
#include "DataFormats/EcalDetId/interface/EBDetId.h"
#include "DataFormats/EcalDetId/interface/EEDetId.h"

#include "DataFormats/TrackReco/interface/Track.h"

#include "TrackingTools/TrackAssociator/interface/TrackDetectorAssociator.h"
#include "TrackingTools/TrackAssociator/interface/TrackAssociatorParameters.h"


//
// class declaration
//

class InterestingTrackEcalDetIdProducer : public edm::stream::EDProducer<> {
   public:
      explicit InterestingTrackEcalDetIdProducer(const edm::ParameterSet&);
      ~InterestingTrackEcalDetIdProducer();

   private:
      virtual void produce(edm::Event&, const edm::EventSetup&) override;
      void beginRun(edm::Run const&, const edm::EventSetup&) override;

      
      // ----------member data ---------------------------
      edm::EDGetTokenT<reco::TrackCollection> trackCollectionToken_;
      edm::InputTag trackCollection_;
      edm::ParameterSet trackAssociatorPS_;

      double minTrackPt_;

      const CaloTopology* caloTopology_;
      TrackDetectorAssociator trackAssociator_; 
      TrackAssociatorParameters trackAssociatorParameters_; 


};

//
// constants, enums and typedefs
//


//
// static data member definitions
//

//
// constructors and destructor
//
InterestingTrackEcalDetIdProducer::InterestingTrackEcalDetIdProducer(const edm::ParameterSet& iConfig) :

  trackAssociatorPS_ (iConfig.getParameter<edm::ParameterSet>("TrackAssociatorParameters")),
  minTrackPt_ (iConfig.getParameter<double>("MinTrackPt"))

{
  trackCollectionToken_=
      consumes<reco::TrackCollection> (iConfig.getParameter<edm::InputTag>("TrackCollection"));  
  trackAssociator_.useDefaultPropagator();
  edm::ConsumesCollector iC = consumesCollector();
  trackAssociatorParameters_.loadParameters(trackAssociatorPS_, iC);

  produces<DetIdCollection>(); 
}


InterestingTrackEcalDetIdProducer::~InterestingTrackEcalDetIdProducer()
{
 
   // do anything here that needs to be done at desctruction time
   // (e.g. close files, deallocate resources etc.)

}


//
// member functions
//

// ------------ method called to produce the data  ------------
void
InterestingTrackEcalDetIdProducer::produce(edm::Event& iEvent, const edm::EventSetup& iSetup)
{
   using namespace edm;

   auto interestingDetIdCollection = std::make_unique<DetIdCollection>();

   // Get tracks from event
   edm::Handle<reco::TrackCollection> tracks;
   iEvent.getByToken(trackCollectionToken_,tracks);

   // Loop over tracks
   for(reco::TrackCollection::const_iterator tkItr = tracks->begin(); tkItr != tracks->end(); ++tkItr)
   {
     if(tkItr->pt() < minTrackPt_)
       continue;

     TrackDetMatchInfo info = trackAssociator_.associate( iEvent, iSetup, 
         trackAssociator_.getFreeTrajectoryState(iSetup, *tkItr),
         trackAssociatorParameters_ );

     DetId centerId = info.findMaxDeposition(TrackDetMatchInfo::EcalRecHits);

     if(centerId.rawId()==0)
       continue;

     // Find 5x5 around max
     const CaloSubdetectorTopology* topology = caloTopology_->getSubdetectorTopology(DetId::Ecal,centerId.subdetId());
     const std::vector<DetId>& ids = topology->getWindow(centerId, 5, 5);
     for(std::vector<DetId>::const_iterator idItr = ids.begin(); idItr != ids.end(); ++idItr)
     {
       if(std::find(interestingDetIdCollection->begin(), interestingDetIdCollection->end(), *idItr)
           == interestingDetIdCollection->end())
         interestingDetIdCollection->push_back(*idItr);
     }

   }

   iEvent.put(std::move(interestingDetIdCollection));

}

void InterestingTrackEcalDetIdProducer::beginRun(edm::Run const& run, const edm::EventSetup & iSetup)  
{
  edm::ESHandle<CaloTopology> theCaloTopology;
  iSetup.get<CaloTopologyRecord>().get(theCaloTopology);
  caloTopology_ = &(*theCaloTopology); 
}

//define this as a plug-in
DEFINE_FWK_MODULE(InterestingTrackEcalDetIdProducer);