Skim_ReduceHcalRecHitCollectionProducer.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:    ReduceHcalRecHitCollectionProducer
// Class:      ReduceHcalRecHitCollectionProducer
//
/*\class ReduceHcalRecHitCollectionProducer ReduceHcalRecHitCollectionProducer.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/Utilities/interface/InputTag.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"


#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/Framework/interface/EventSetup.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 "DataFormats/HcalRecHit/interface/HcalRecHitCollections.h"
#include "DataFormats/HcalDetId/interface/HcalSubdetector.h"
#include "DataFormats/HcalDetId/interface/HcalDetId.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"

#include <iostream>

//
// class declaration
//

class ReduceHcalRecHitCollectionProducer : public edm::EDProducer {
   public:
      explicit ReduceHcalRecHitCollectionProducer(const edm::ParameterSet&);
      ~ReduceHcalRecHitCollectionProducer() override;
      void produce(edm::Event&, const edm::EventSetup&) override;
   private:
      edm::EDGetTokenT<HBHERecHitCollection> recHitsToken_;
      std::string reducedHitsCollection_;
      edm::EDGetTokenT<reco::TrackCollection> inputCollectionToken_;
      TrackDetectorAssociator trackAssociator_;
      TrackAssociatorParameters parameters_;
      double  ptcut_;
      // ----------member data ---------------------------
};

//
// constants, enums and typedefs
//


//
// static data member definitions
//

//
// constructors and destructor
//
ReduceHcalRecHitCollectionProducer::ReduceHcalRecHitCollectionProducer(const edm::ParameterSet& iConfig)
{
  recHitsToken_ = consumes<HBHERecHitCollection>(iConfig.getParameter< edm::InputTag > ("recHitsLabel"));

  reducedHitsCollection_ = iConfig.getParameter<std::string>("reducedHitsCollection");

   //register your products
  produces< HBHERecHitCollection > (reducedHitsCollection_) ;

    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();

}


ReduceHcalRecHitCollectionProducer::~ReduceHcalRecHitCollectionProducer()
{

   // 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
ReduceHcalRecHitCollectionProducer::produce(edm::Event& iEvent, const edm::EventSetup& iSetup)
{
   using namespace edm;

   using namespace std;

   using reco::TrackCollection;

   Handle<HBHERecHitCollection> recHitsHandle;
   iEvent.getByToken(recHitsToken_,recHitsHandle);
   if( !recHitsHandle.isValid() )
     {
       edm::LogError("ReduceHcalRecHitCollectionProducer") << "RecHit collection not found";
       return;
     }

   //Create empty output collections
   std::unique_ptr< HBHERecHitCollection > miniRecHitCollection (new HBHERecHitCollection) ;

//loop through tracks.
   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.crossedHcalIds.empty()){
             //loop through hits in the cone
             for(std::vector<const HBHERecHit*>::const_iterator hit = info.hcalRecHits.begin();
                 hit != info.hcalRecHits.end(); ++hit)
             {
                DetId hitid=(*hit)->id();
                HBHERecHitCollection::const_iterator iRecHit = recHitsHandle->find(hitid);
                if ( (iRecHit != recHitsHandle->end()) && (miniRecHitCollection->find(hitid) == miniRecHitCollection->end()) )
                   miniRecHitCollection->push_back(*iRecHit);
             }


          }
        }
   }

   iEvent.put(std::move(miniRecHitCollection),reducedHitsCollection_ );



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