Skim_HighPtTrackEcalDetIdProducer.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:    HighPtTrackEcalDetIdProducer
// Class:      HighPtTrackEcalDetIdProducer
//
/*\class HighPtTrackEcalDetIdProducer HighPtTrackEcalDetIdProducer.cc

 Description: [one line class summary]

 Implementation:
     [Notes on implementation]
*/
//
// Original Author:  Jie Chen
//         Created:  Mon Apr 12 16:41:46 CDT 2010
//
//

// system include files
#include <memory>

// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/EDProducer.h"

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"

#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/Common/interface/Handle.h"
#include "FWCore/Framework/interface/ESHandle.h"

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

#include "DataFormats/EcalRecHit/interface/EcalRecHit.h"
#include "DataFormats/EcalRecHit/interface/EcalRecHitCollections.h"

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

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

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

//
// class declaration
//

class HighPtTrackEcalDetIdProducer : public edm::EDProducer {
public:
  explicit HighPtTrackEcalDetIdProducer(const edm::ParameterSet&);
  ~HighPtTrackEcalDetIdProducer() override;
  void beginRun(const edm::Run&, const edm::EventSetup&) override;
  void produce(edm::Event&, const edm::EventSetup&) override;

private:
  edm::EDGetTokenT<reco::TrackCollection> inputCollectionToken_;
  const CaloTopology* caloTopology_;
  TrackDetectorAssociator trackAssociator_;
  TrackAssociatorParameters parameters_;
  double ptcut_;
  // ----------member data ---------------------------
};

//
// constants, enums and typedefs
//

//
// static data member definitions
//

//
// constructors and destructor
//
HighPtTrackEcalDetIdProducer::HighPtTrackEcalDetIdProducer(const edm::ParameterSet& iConfig) {
  inputCollectionToken_ = consumes<reco::TrackCollection>(iConfig.getParameter<edm::InputTag>("inputCollection"));
  ptcut_ = iConfig.getParameter<double>("TrackPt");

  produces<DetIdCollection>();
  // TrackAssociator parameters
  edm::ParameterSet parameters = iConfig.getParameter<edm::ParameterSet>("TrackAssociatorParameters");
  edm::ConsumesCollector iC = consumesCollector();
  parameters_.loadParameters(parameters, iC);
  trackAssociator_.useDefaultPropagator();
}

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

//
// member functions
//

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

// ------------ method called to produce the data  ------------
void HighPtTrackEcalDetIdProducer::produce(edm::Event& iEvent, const edm::EventSetup& iSetup) {
  using namespace edm;
  using reco::TrackCollection;
  //   if(!iSetup) continue;
  Handle<TrackCollection> tkTracks;
  iEvent.getByToken(inputCollectionToken_, tkTracks);
  std::unique_ptr<DetIdCollection> interestingDetIdCollection(new DetIdCollection());
  for (TrackCollection::const_iterator itTrack = tkTracks->begin(); itTrack != tkTracks->end(); ++itTrack) {
    if (itTrack->pt() > ptcut_) {
      TrackDetMatchInfo info =
          trackAssociator_.associate(iEvent, iSetup, *itTrack, parameters_, TrackDetectorAssociator::InsideOut);
      if (info.crossedEcalIds.empty())
        break;

      if (!info.crossedEcalIds.empty()) {
        DetId centerId = info.crossedEcalIds.front();

        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 id = ids.begin(); id != ids.end(); ++id)
          if (std::find(interestingDetIdCollection->begin(), interestingDetIdCollection->end(), *id) ==
              interestingDetIdCollection->end())
            interestingDetIdCollection->push_back(*id);
      }
    }
  }
  iEvent.put(std::move(interestingDetIdCollection));
}
//define this as a plug-in
DEFINE_FWK_MODULE(HighPtTrackEcalDetIdProducer);