TRT_SeededTrackFinder.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
/**********************************************************************************
   Header file for class TRT_SeededTrackFinder
  (c) ATLAS Detector software
  Algorithm for Trk::Track production in SCT and Pixels
  Version 1.0: 04/12/2006
  Authors    : Thomas Koffas, Markus Elsing
  email      : Thomas.Koffas@cern.ch
**********************************************************************************/
 
#ifndef TRT_SeededTrackFinder_H
#define TRT_SeededTrackFinder_H
 
#include "AthenaBaseComps/AthReentrantAlgorithm.h"
#include "GaudiKernel/ServiceHandle.h"
#include "GaudiKernel/ToolHandle.h"
 
#include "TrkTrack/TrackCollection.h"
 
#include "TrkSegment/SegmentCollection.h"
 
#include "TrkFitterInterfaces/ITrackFitter.h"
 
#include "InDetRecToolInterfaces/ITRT_TrackExtensionTool.h"
 
#include "BeamSpotConditionsData/BeamSpotData.h"
#include "InDetRecToolInterfaces/ITRT_SeededTrackFinder.h"
#include "TrkEventUtils/PRDtoTrackMap.h"
#include "TrkExInterfaces/IExtrapolator.h"
#include "TrkToolInterfaces/IExtendedTrackSummaryTool.h"
 
#include "CxxUtils/checker_macros.h"
 
#include "IRegionSelector/IRegSelTool.h"
#include "TrkCaloClusterROI/ROIPhiRZContainer.h"
 
class MsgStream;
 
namespace InDet {
 
class TRT_SeededTrackFinder : public AthReentrantAlgorithm
{
 
  // Public methods:
 
public:
 
 
  TRT_SeededTrackFinder(const std::string& name, ISvcLocator* pSvcLocator);
  virtual ~TRT_SeededTrackFinder() {}
  StatusCode initialize() override;
  StatusCode execute(const EventContext& ctx) const override;
  StatusCode finalize() override;
 
 
 
  MsgStream& dump(MsgStream& out) const;
  std::ostream& dump(std::ostream& out) const;
 
protected:
 
  /* Protected data                                                */
  bool m_doRefit;     
  bool m_doExtension; 
  bool m_rejectShortExten; 
  bool m_doStat;        
  bool m_saveTRT;       
  int m_MaxSegNum;      
  unsigned int m_minTRTonSegment; 
  unsigned int m_minTRTonly;      
  ToolHandle<ITRT_SeededTrackFinder> m_trackmaker; 
  ToolHandle<Trk::ITrackFitter> m_fitterTool;      
  ToolHandle<ITRT_TrackExtensionTool> m_trtExtension{
    this,
    "TrackExtensionTool",
    "InDet::TRT_TrackExtensionTool_xk",
    "TRT track extension tool "
  };
 
  SG::ReadHandleKey<Trk::SegmentCollection>
    m_SegmentsKey; 
  SG::WriteHandleKey<TrackCollection> m_outTracksKey;
 
  SG::ReadHandleKey<Trk::PRDtoTrackMap> m_prdToTrackMap{ this,
                                                         "PRDtoTrackMap",
                                                         "" };
  ToolHandle<Trk::IExtendedTrackSummaryTool> m_trackSummaryTool{
    this,
    "TrackSummaryTool",
    "InDetTrackSummaryToolNoHoleSearch"
  };
 
  ToolHandle<Trk::IExtrapolator> m_extrapolator; 
  SG::ReadCondHandleKey<InDet::BeamSpotData> m_beamSpotKey{
    this,
    "BeamSpotKey",
    "BeamSpotData",
    "SG key for beam spot"
  };
  bool m_SiExtensionCuts; 
  double m_minPt;         
  double m_maxEta;        
  double m_maxRPhiImp;    
  double m_maxZImp;       
 
  bool m_caloSeededRoI;
  SG::ReadHandleKey<ROIPhiRZContainer> m_caloClusterROIKey{
    this,
    "EMROIPhiRZContainer",
    "",
    "Name of the calo cluster ROIs in Phi,R,Z parameterization"
  };
 
  ToolHandle<IRegSelTool> m_regionSelector{
    this,
    "RegSelTool",
    "RegSelTool/RegSel_SCT",
    "Region selector service instance"
  };
 
  float m_deltaEta;  
  float m_deltaPhi;  
  float m_deltaZ;    
 
  struct Stat_t
  {
    enum ECounter
    {
      kNTrtSeg,      
      kNTrtFailSel,  
      kNTrtSegGood,  
      kNTrtLimit,    
      kNTrtNoSiExt,  
      kNExtCut,      
      kNBckTrkTrt,   
      kNTrtExtCalls, 
      kNTrtExt,      
      kNTrtExtBad,   
      kNTrtExtFail,  
      kNBckTrkSi, 
      kNBckTrk, 
      kNCounter
    };
    std::array<int, kNCounter> m_counter{};
 
    Stat_t& operator+=(const Stat_t& a)
    {
      for (unsigned int i = 0; i < a.m_counter.size(); ++i) {
        m_counter[i] += a.m_counter[i];
      }
      return *this;
    }
  };
 
  mutable std::mutex m_statMutex ATLAS_THREAD_SAFE;
  mutable Stat_t m_totalStat ATLAS_THREAD_SAFE;
 
 
 
  Trk::Track* mergeSegments(const Trk::Track&, const Trk::TrackSegment&) const;
 
  Trk::Track* mergeExtension(const Trk::Track&,
                             std::vector<const Trk::MeasurementBase*>&) const;
 
  Trk::Track* segToTrack(const EventContext&, const Trk::TrackSegment&) const;
 
  void Analyze(TrackCollection*) const;
 
  MsgStream& dumptools(MsgStream& out) const;
  MsgStream& dumpevent(MsgStream& out,
                       const InDet::TRT_SeededTrackFinder::Stat_t& stat) const;
};
}
#endif // TRT_SeededTrackFinder_H