SETSeedFinder.h

Go to the documentation of this file.

#ifndef MuonSeedGenerator_SETSeedFinder_h
#define MuonSeedGenerator_SETSeedFinder_h
 
#include "RecoMuon/MuonSeedGenerator/src/MuonSeedVFinder.h"
#include "RecoMuon/TrackingTools/interface/MuonServiceProxy.h"
#include "RecoMuon/MuonSeedGenerator/src/MuonSeedPtExtractor.h"
#include "RecoMuon/MuonSeedGenerator/src/SETFilter.h"
#include "CLHEP/Matrix/Vector.h"
#include "CLHEP/Vector/ThreeVector.h"
 
class SETSeedFinder : public MuonSeedVFinder {
public:
  typedef MuonTransientTrackingRecHit::MuonRecHitContainer MuonRecHitContainer;
 
  explicit SETSeedFinder(const edm::ParameterSet& pset);
  ~SETSeedFinder() override { delete thePtExtractor; }
  void setBField(const MagneticField* field) override {}
 
  void seeds(const MuonRecHitContainer& cluster, std::vector<TrajectorySeed>& result) override;
 
  void setServiceProxy(MuonServiceProxy* service) { theService = service; }
 
  std::vector<MuonRecHitContainer> sortByLayer(MuonRecHitContainer& cluster) const;
 
  //---- For protection against huge memory consumtion
  void limitCombinatorics(std::vector<MuonRecHitContainer>& MuonRecHitContainer_perLayer);
 
  std::vector<MuonRecHitContainer> findAllValidSets(
      const std::vector<MuonRecHitContainer>& MuonRecHitContainer_perLayer);
 
  std::pair<int, int> checkAngleDeviation(double dPhi_1, double dPhi_2) const;
 
  void validSetsPrePruning(std::vector<MuonRecHitContainer>& allValidSets);
 
  void pre_prune(MuonRecHitContainer& validSet) const;
 
  std::vector<SeedCandidate> fillSeedCandidates(std::vector<MuonRecHitContainer>& allValidSets);
 
  void estimateMomentum(const MuonRecHitContainer& validSet, CLHEP::Hep3Vector& momentum, int& charge) const;
 
  TrajectorySeed makeSeed(const TrajectoryStateOnSurface& tsos,
                          const TransientTrackingRecHit::ConstRecHitContainer& hits) const;
 
private:
  MuonServiceProxy* theService;
 
  bool apply_prePruning;
  bool useSegmentsInTrajectory;
};
 
#endif