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;
}  // namespace edm
 
//              ---------------------
//              -- 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