TRT_SegmentToTrackTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/


 
#ifndef TRT_SegmentToTrackTool_H
#define TRT_SegmentToTrackTool_H
 
#include "InDetRecToolInterfaces/ITRT_SegmentToTrackTool.h"
//
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ToolHandle.h"
//
#include "TrkFitterInterfaces/ITrackFitter.h"
#include "TrkToolInterfaces/IExtendedTrackSummaryTool.h"
#include "TrkToolInterfaces/IPRDtoTrackMapTool.h"
//Extrapolator tool
#include "TrkExInterfaces/IExtrapolator.h"
//Scoring tool
#include "TrkToolInterfaces/ITrackScoringTool.h"
 
#include "TrkEventPrimitives/TrackScore.h"
// MagField cache
#include "MagFieldConditions/AtlasFieldCacheCondObj.h"
#include "MagFieldElements/AtlasFieldCache.h"
 
#include <string>
 
class MsgStream;
class TRT_ID;
 
namespace Trk {
  class Track;
  class TrackSegment;
  class IExtrapolator;
  class ITrackScoringTool;
}
 
namespace InDet {
 
  //class TrackSegment;
 
  class TRT_SegmentToTrackTool final: 
  virtual public ITRT_SegmentToTrackTool,
    public AthAlgTool
    {
 
    public:
 
      //constructor
      TRT_SegmentToTrackTool(const std::string& type, 
                   const std::string& name, 
                   const IInterface* parent );
 
      virtual ~TRT_SegmentToTrackTool();
 
      virtual StatusCode initialize() override;
      virtual StatusCode finalize() override;
 
      virtual Trk::Track* segToTrack(const EventContext& ctx, const Trk::TrackSegment&) const override;
      virtual bool segIsUsed(const Trk::TrackSegment&,
                             const Trk::PRDtoTrackMap *prd_to_track_map) const override;
      virtual bool toLower(const Trk::TrackSegment&) const override;

      virtual void addNewTrack(Trk::Track*,
                               ITRT_SegmentToTrackTool::EventData &event_data) const override;
      virtual TrackCollection* resolveTracks(const Trk::PRDtoTrackMap *prd_to_track_map,
                                             ITRT_SegmentToTrackTool::EventData &event_data) const override;
 
      virtual MsgStream&    dump     (MsgStream&    out) const override;
      virtual std::ostream& dump(std::ostream& out) const override;
 
    private:
      BooleanProperty m_doRefit{this, "FinalRefit", false,
      "Do final careful refit of tracks"};
      static constexpr double m_fieldUnitConversion = 1000.;
      DoubleProperty m_sharedFrac{this, "MaxSharedHitsFraction", 0.3,
    "Maximum fraction of shared TRT drift circles"};
      BooleanProperty m_suppressHoleSearch{this, "SuppressHoleSearch", false,
    "Suppress hole search during the track summary creation"};
 
      ToolHandle<Trk::IExtrapolator> m_extrapolator
    {this, "Extrapolator", "Trk::Extrapolator/InDetExtrapolator"};
 
      ToolHandle<Trk::ITrackFitter> m_fitterTool{ this, "RefitterTool", "" };
      ToolHandle<Trk::IPRDtoTrackMapTool> m_assoTool{
        this,
        "AssociationTool",
        "InDet::InDetPRDtoTrackMapToolGangedPixels"
      };
      ToolHandle<Trk::IExtendedTrackSummaryTool> m_trackSummaryTool{
        this,
        "TrackSummaryTool",
        "InDetTrackSummaryToolNoHoleSearch"
      };
 
      ToolHandle<Trk::ITrackScoringTool> m_scoringTool
    {this, "ScoringTool", "Trk::TrackScoringTool/TrackScoringTool"};

      // Read handle for conditions object to get the field cache
      SG::ReadCondHandleKey<AtlasFieldCacheCondObj> m_fieldCacheCondObjInputKey {this, "AtlasFieldCacheCondObj", "fieldCondObj",
                                                                                 "Name of the Magnetic Field conditions object key"};

 

      const TRT_ID* m_trtId = nullptr;
 
      static MsgStream&    dumpevent      (MsgStream&    out) ;
      MsgStream&    dumpconditions (MsgStream&    out) const;
 
    };

   // Overload of << operator for MsgStream and  std::ostream
 
   MsgStream&    operator << (MsgStream&   ,const ITRT_SegmentToTrackTool&);
   std::ostream& operator << (std::ostream&,const ITRT_SegmentToTrackTool&);

   // Inline methods

   // Overload of << operator MsgStream
 
   inline MsgStream& operator    <<     (MsgStream& sl,const ITRT_SegmentToTrackTool& se)
   {
      return se.dump(sl);
   }

   // Overload of << operator std::ostream
 
   inline std::ostream& operator <<     (std::ostream& sl,const ITRT_SegmentToTrackTool& se)
   {
      return se.dump(sl);
   }
 
}
 
#endif //TRT_SegmentToTrackTool_H