TrackerSeedCleaner.h

Go to the documentation of this file.

#ifndef RecoMuon_TrackerSeedCleaner_H
#define RecoMuon_TrackerSeedCleaner_H

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

#include "DataFormats/TrajectorySeed/interface/TrajectorySeedCollection.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "TrackingTools/TransientTrackingRecHit/interface/TransientTrackingRecHitBuilder.h"
#include "RecoTracker/TransientTrackingRecHit/interface/TkTransientTrackingRecHitBuilder.h"
#include "RecoTracker/TkTrackingRegions/interface/RectangularEtaPhiTrackingRegion.h"
#include "RecoMuon/TrackerSeedGenerator/interface/RedundantSeedCleaner.h"
#include "DataFormats/BeamSpot/interface/BeamSpot.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"

class MuonServiceProxy;
class TSGFromL2Muon;
class MuonTrackingRegionBuilder;

namespace edm {class ParameterSet; class Event; class EventSetup;}

//              ---------------------
//              -- Class Interface --
//              ---------------------

class TrackerSeedCleaner{

public:

  typedef std::vector<TrajectorySeed> tkSeeds;
  TrackerSeedCleaner(const edm::ParameterSet& pset,edm::ConsumesCollector& iC) : theProxyService(nullptr),theEvent(nullptr) {
                   builderName_ = pset.getParameter<std::string>("TTRHBuilder");
                   theBeamSpotTag = pset.getParameter<edm::InputTag>("beamSpot");
                   useDirection_Cleaner = pset.getParameter<bool>("directionCleaner");
                   usePt_Cleaner = pset.getParameter<bool>("ptCleaner");
                   cleanBySharedHits = pset.getParameter<bool>("cleanerFromSharedHits");
           beamspotToken_ = iC.consumes<reco::BeamSpot>(theBeamSpotTag);
  }

  virtual void init(const MuonServiceProxy *service);
  
  virtual ~TrackerSeedCleaner() {}
  virtual void  clean(const reco::TrackRef& , const RectangularEtaPhiTrackingRegion& region, tkSeeds&);  
  virtual void setEvent(const edm::Event&);

private:

  const MuonServiceProxy * theProxyService;
  const edm::Event * theEvent;

  edm::InputTag theBeamSpotTag; //beam spot
  edm::Handle<reco::BeamSpot> bsHandle_;
  edm::EDGetTokenT<reco::BeamSpot> beamspotToken_;
  RedundantSeedCleaner * theRedundantCleaner;

  std::string builderName_;
  edm::ESHandle<TransientTrackingRecHitBuilder> theTTRHBuilder;
  bool useDirection_Cleaner,usePt_Cleaner,cleanBySharedHits; 
};

#endif