TRT_SeededTrackFinder.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 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                                                */
  BooleanProperty m_doRefit
    {this, "FinalRefit", false, "Do a final careful refit of tracks"};
  BooleanProperty m_doExtension
    {this, "TrtExtension", false, "Find the TRT extension of the track"};
  BooleanProperty m_rejectShortExten
    {this, "RejectShortExtension", false, "Reject short extensions"};
  BooleanProperty m_doStat
    {this, "FinalStatistics", false, "Statistics of final tracks"};
  BooleanProperty m_saveTRT
    {this, "OutputSegments", true, "Save stand-alone TRT segments"};
  IntegerProperty m_MaxSegNum
    {this, "MaxNumberSegments", 5000,
     "Maximum number of segments to be handled"};
  UnsignedIntegerProperty m_minTRTonSegment
    {this, "MinTRTonSegment", 10, "Minimum Number of TRT Hits on segment"};
  UnsignedIntegerProperty m_minTRTonly
    {this, "MinTRTonly", 15, "Minimum number of TRT hits on TRT only"};
 
  ToolHandle<ITRT_SeededTrackFinder> m_trackmaker
    {this, "TrackTool", "InDet::TRT_SeededTrackFinderTool",
     "Back tracking tool"};
  ToolHandle<Trk::ITrackFitter> m_fitterTool
    {this, "RefitterTool", "Trk::GlobalChi2Fitter/InDetTrackFitter",
     "Track refit tool"};
  ToolHandle<ITRT_TrackExtensionTool> m_trtExtension
    {this, "TrackExtensionTool", "InDet::TRT_TrackExtensionTool_xk",
     "TRT track extension tool "};
 
  SG::ReadHandleKey<Trk::SegmentCollection> m_SegmentsKey
    {this, "InputSegmentsLocation", "TRTSegments", "TRT segments to use"};
  SG::WriteHandleKey<TrackCollection> m_outTracksKey
    {this, "OutputTracksLocation", "TRTSeededTracks",
     "Output track collection"};
 
  SG::ReadHandleKey<Trk::PRDtoTrackMap> m_prdToTrackMap
    {this, "PRDtoTrackMap", "" };
  ToolHandle<Trk::IExtendedTrackSummaryTool> m_trackSummaryTool
    {this, "TrackSummaryTool", "InDetTrackSummaryToolNoHoleSearch"};
 
  ToolHandle<Trk::IExtrapolator> m_extrapolator{this, "Extrapolator", ""};
  SG::ReadCondHandleKey<InDet::BeamSpotData> m_beamSpotKey
    {this, "BeamSpotKey", "BeamSpotData", "SG key for beam spot"};
 
  BooleanProperty m_SiExtensionCuts
    {this, "SiExtensionCuts", false, "enable cuts after Si segment finding"};
  DoubleProperty m_minPt{this, "minPt", 500., "minimal Pt cut"};
  DoubleProperty m_maxEta{this, "maxEta", 2.7, "maximal Eta cut"};
  DoubleProperty m_maxRPhiImp
    {this, "maxRPhiImp", 10., "maximal RPhi impact parameter cut"};
  DoubleProperty m_maxZImp
    {this, "maxZImp", 250., "maximal z impact parameter cut"};
 
  BooleanProperty m_caloSeededRoI{this, "CaloSeededRoI", false};
  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"};
 
  FloatProperty m_deltaEta
    {this, "dEtaCaloRoI", 0.1, "delta Eta used for RoI creation"};
  FloatProperty m_deltaPhi
    {this, "dPhiCaloRoI", 0.25, "delta Phi used for RoI creation"};
  FloatProperty m_deltaZ
    {this, "dZCaloRoI", 300., "delta Z used for RoI creation"};

  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