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