DTCombinatorialExtendedPatternReco.h

Go to the documentation of this file.

#ifndef DTSegment_DTCombinatorialExtendedPatternReco_h
#define DTSegment_DTCombinatorialExtendedPatternReco_h
 
/* Base Class Headers */
#include "RecoLocalMuon/DTSegment/src/DTRecSegment2DBaseAlgo.h"
 
/* Collaborating Class Declarations */
namespace edm {
  class ParameterSet;
  class EventSetup;
  //  class ESHandle;
}  // namespace edm
class DTSegmentUpdator;
class DTSegmentCleaner;
class DTHitPairForFit;
class DTSegmentCand;
class DTSegmentExtendedCand;
#include "DataFormats/DTRecHit/interface/DTSLRecCluster.h"
 
/* C++ Headers */
#include <vector>
#include <deque>
#include <utility>
 
#include "Geometry/DTGeometry/interface/DTGeometry.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "RecoLocalMuon/DTSegment/src/DTSegmentCand.h"
 
/* ====================================================================== */
 
/* Class DTCombinatorialExtendedPatternReco Interface */
 
class DTCombinatorialExtendedPatternReco : private DTRecSegment2DBaseAlgo {
public:
  DTCombinatorialExtendedPatternReco(const edm::ParameterSet& pset);
 
  ~DTCombinatorialExtendedPatternReco() override;
 
  /* Operations */
 
  edm::OwnVector<DTSLRecSegment2D> reconstruct(const DTSuperLayer* sl,
                                               const std::vector<DTRecHit1DPair>& hits) override;
 
  std::string algoName() const override { return theAlgoName; }
 
  void setES(const edm::EventSetup& setup) override;
 
  // pass clusters to algo
  void setClusters(const std::vector<DTSLRecCluster>& clusters);
 
protected:
private:
  // create the DTHitPairForFit from the pairs for easy use
  std::vector<std::shared_ptr<DTHitPairForFit>> initHits(const DTSuperLayer* sl,
                                                         const std::vector<DTRecHit1DPair>& hits);
 
  // search for candidate, starting from pairs of hits in different layers
  std::vector<DTSegmentCand*> buildSegments(const DTSuperLayer* sl,
                                            const std::vector<std::shared_ptr<DTHitPairForFit>>& hits);
 
  // find all the hits compatible with the candidate
  std::vector<DTSegmentCand::AssPoint> findCompatibleHits(const LocalPoint& pos,
                                                          const LocalVector& dir,
                                                          const std::vector<std::shared_ptr<DTHitPairForFit>>& hits);
 
  // build segments from hits collection
  DTSegmentExtendedCand* buildBestSegment(std::vector<DTSegmentCand::AssPoint>& assHits, const DTSuperLayer* sl);
 
  bool checkDoubleCandidates(std::vector<DTSegmentCand*>& segs, DTSegmentCand* seg);
 
  void buildPointsCollection(std::vector<DTSegmentCand::AssPoint>& points,
                             std::deque<std::shared_ptr<DTHitPairForFit>>& pointsNoLR,
                             std::vector<DTSegmentCand*>& candidates,
                             const DTSuperLayer* sl);
 
  std::vector<DTSegmentExtendedCand*> extendCandidates(std::vector<DTSegmentCand*>& candidates, const DTSuperLayer* sl);
 
  bool closeSL(const DTSuperLayerId& id1, const DTSuperLayerId& id2);
 
private:
  std::string theAlgoName;
  unsigned int theMaxAllowedHits;
  double theAlphaMaxTheta;
  double theAlphaMaxPhi;
  bool debug;
  bool usePairs;
  DTSegmentUpdator* theUpdator;  // the updator and fitter
  DTSegmentCleaner* theCleaner;  // the cleaner
 
  edm::ESHandle<DTGeometry> theDTGeometry;  // the DT geometry
 
private:
  std::vector<std::vector<int>> theTriedPattern;
  std::vector<DTSLRecCluster> theClusters;
};
#endif  // DTSegment_DTCombinatorialExtendedPatternReco_h