GEMSegmentAlgorithm.h

Go to the documentation of this file.

#ifndef GEMRecHit_GEMSegmentAlgorithm_h
#define GEMRecHit_GEMSegmentAlgorithm_h

#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "RecoLocalMuon/GEMSegment/plugins/GEMSegmentAlgorithmBase.h"
#include "DataFormats/GEMRecHit/interface/GEMRecHit.h"
#include "FWCore/Utilities/interface/Exception.h"
#include <deque>
#include <vector>

class MuonSegFit;

class GEMSegmentAlgorithm : public GEMSegmentAlgorithmBase {
public:
  typedef std::vector<const GEMRecHit*> EnsembleHitContainer;
  typedef std::vector<EnsembleHitContainer> ProtoSegments;

  explicit GEMSegmentAlgorithm(const edm::ParameterSet& ps);
  ~GEMSegmentAlgorithm() override;

  std::vector<GEMSegment> run(const GEMEnsemble& ensemble, const EnsembleHitContainer& rechits) override;

private:

  //  Build groups of rechits that are separated in x and y to save time on the segment finding
  ProtoSegments clusterHits(const GEMEnsemble& ensemble, const EnsembleHitContainer& rechits);

  // Build groups of rechits that are separated in strip numbers and Z to save time on the segment finding
  ProtoSegments chainHits(const GEMEnsemble& ensemble, const EnsembleHitContainer& rechits);

  bool isGoodToMerge(const GEMEnsemble& ensemble,
                     const EnsembleHitContainer& newChain,
                     const EnsembleHitContainer& oldChain);

  // Build track segments in this chamber (this is where the actual segment-building algorithm hides.)
  void buildSegments(const GEMEnsemble& ensemble,
                     const EnsembleHitContainer& rechits,
                     std::vector<GEMSegment>& gemsegs);

  // Member variables
  const std::string myName;

  // input from .cfi file
  bool debug;
  unsigned int minHitsPerSegment;
  bool preClustering;
  double dXclusBoxMax;
  double dYclusBoxMax;
  bool preClustering_useChaining;
  double dPhiChainBoxMax;
  double dEtaChainBoxMax;
  int maxRecHitsInCluster;
  bool clusterOnlySameBXRecHits;

  EnsembleHitContainer proto_segment;
  GEMDetId theChamberId;

  static constexpr float running_max = std::numeric_limits<float>::max();
  std::unique_ptr<MuonSegFit> sfit_;
};

#endif