RecoMuon/TrackerSeedGenerator/interface/TrackerSeedCleaner.h - master - githubmirror/cmssw

RecoMuon/TrackerSeedGenerator/interface/TrackerSeedCleaner.h
83 строки3 KB
Marino Missiroli
speed up shared-hits cleaning in TrackerSeedCleaner
22 дек 2024, 20:04
22 дек 2024, 20:04faa3212
#ifndef RecoMuon_TrackerSeedCleaner_H
#define RecoMuon_TrackerSeedCleaner_H

/** \class TrackerSeedCleaner
 *  Seeds Cleaner based on direction
    \author A. Grelli -  Purdue University, Pavia University
 */

#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 "TrackingTools/Records/interface/TransientRecHitRecord.h"
#include "RecoTracker/TkTrackingRegions/interface/RectangularEtaPhiTrackingRegion.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;
  /// constructor
  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);
    theTTRHBuilderToken = iC.esConsumes(edm::ESInputTag("", builderName_));
  }

  ///intizialization
  virtual void init(const MuonServiceProxy* service);

  /// destructor
  virtual ~TrackerSeedCleaner() {}
  /// the cleaner
  virtual void clean(const reco::TrackRef&, const RectangularEtaPhiTrackingRegion& region, tkSeeds&);
  /// setEvent
  virtual void setEvent(const edm::Event&);

  tkSeeds nonRedundantSeeds(tkSeeds const&) const;

private:
  bool seedIsNotRedundant(std::vector<TrajectorySeed> const& seeds,
                          TrajectorySeed const& s1,
                          std::vector<uint> const& tripletsIdx) const;

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

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

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

#endif