TRT_TrackExtensionToolCosmics.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_TrackExtensionToolCosmics
// (c) ATLAS Detector software
//  Concret implementation for base class ITRT_TrackExtensionTool
// Version 1.0 22/08/2005 I.Gavrilenko
 
#ifndef TRT_TrackExtensionToolCosmics_H
#define TRT_TrackExtensionToolCosmics_H
 
 
 
#include "GaudiKernel/ToolHandle.h"
#include "AthenaBaseComps/AthAlgTool.h"
#include "InDetRecToolInterfaces/ITRT_TrackExtensionTool.h"
#include "InDetPrepRawData/TRT_DriftCircleContainer.h"
#include "TrkEventUtils/EventDataBase.h"
#include "TrkExInterfaces/IExtrapolator.h"
 
#include "StoreGate/ReadHandleKey.h"
#include <iosfwd>
#include <vector>
 
class MsgStream;
class TRT_ID;
class AtlasDetectorID;
class PropDirection;
 
namespace Trk{
  class Surface;
}
 
namespace InDet {

 
  class TRT_TrackExtensionToolCosmics : 
 
    virtual public ITRT_TrackExtensionTool, public AthAlgTool
    {
      // Public methods:
      
    public:
      
      // Standard tool methods
 
      TRT_TrackExtensionToolCosmics
    (const std::string&,const std::string&,const IInterface*);
      virtual ~TRT_TrackExtensionToolCosmics();
      virtual StatusCode initialize() override;
      virtual StatusCode finalize  () override;

      // Main methods for track extension to TRT
 
      virtual std::vector<const Trk::MeasurementBase*>& extendTrack
         (const EventContext& ctx,
          const Trk::Track&,
          InDet::ITRT_TrackExtensionTool::IEventData &virt_event_data,
          InDet::TRT_DetElementLink_xk::TRT_DetElemUsedMap& used) const override;
 
      virtual std::vector<const Trk::MeasurementBase*>& extendTrack
         (const EventContext& ctx,
          const Trk::TrackParameters *,
          InDet::ITRT_TrackExtensionTool::IEventData &virt_event_data,
          InDet::TRT_DetElementLink_xk::TRT_DetElemUsedMap& used) const override;
 
      virtual Trk::TrackSegment* findSegment
     (const EventContext& ctx,
          const Trk::TrackParameters *,
          InDet::ITRT_TrackExtensionTool::IEventData &virt_event_data,
          InDet::TRT_DetElementLink_xk::TRT_DetElemUsedMap& used) const override;
 
      virtual Trk::Track* newTrack
         (const EventContext& ctx,
          const Trk::Track&,
          InDet::ITRT_TrackExtensionTool::IEventData &virt_event_data,
          InDet::TRT_DetElementLink_xk::TRT_DetElemUsedMap& used) const override;
 
      virtual std::unique_ptr<InDet::ITRT_TrackExtensionTool::IEventData> newEvent(const EventContext& ctx) const override;

      // Print internal tool parameters and status
 
      virtual MsgStream&    dump(MsgStream&    out) const override;
      virtual std::ostream& dump(std::ostream& out) const override;
 
    protected:
      
      // Protected Data
 
      ToolHandle<Trk::IExtrapolator>        m_extrapolator
        {this, "Extrapolator", "Trk::Extrapolator/InDetExtrapolator"};
      ToolHandle<Trk::IRIO_OnTrackCreator>  m_riontrackN
    {this, "RIOonTrackToolNoDr", "InDet::TRT_DriftCircleOnTrackNoDriftTimeTool/TRT_DriftCircleOnTrackNoDriftTimeTool"}; //
 
      int                              m_outputlevel{}    ; // Print level
      int                              m_nprint{}         ; // Kind of print
      DoubleProperty m_roadwidth{this, "RoadWidth", 10.,
    "Max width of the road"};
      DoubleProperty m_roadwidth_locz{this, "BoundaryLocZTolerance", 10.,
    "Max width of the road along the straw"};
      BooleanProperty m_searchNeighbour{this, "SearchNeighbour", false,
    "Also search neighbouring detector elements?"};
 
      SG::ReadHandleKey<TRT_DriftCircleContainer> m_trtname {this,"TRT_ClustersContainer","TRT_DriftCircles","RHK to retrieve TRT_DriftCircles"};
 
      class EventData;
      class EventData : public Trk::EventDataBase<EventData,InDet::ITRT_TrackExtensionTool::IEventData>
      {
         friend class TRT_TrackExtensionToolCosmics;
      public:
         EventData(const TRT_DriftCircleContainer *trtcontainer) : m_trtcontainer(trtcontainer) {}
 
         ~EventData() {
            delete m_trtcylinder;
            delete m_trtdiscA;
            delete m_trtdiscC;
         }
 
      protected:
         const TRT_DriftCircleContainer  *m_trtcontainer = nullptr;
         Trk::Surface                    *m_trtcylinder  = nullptr;
         Trk::Surface                    *m_trtdiscA     = nullptr;
         Trk::Surface                    *m_trtdiscC     = nullptr;
 
         std::vector<const Trk::MeasurementBase*>  m_measurement  ;
      };
 
 
      AtlasDetectorID*                       m_idHelper{}; //<! Detector ID helper
    
      const TRT_ID* m_trtid{} ;
 
      static MsgStream&    dumpConditions(MsgStream   & out) ;
      static MsgStream&    dumpEvent     (MsgStream   & out) ;
 
 
 
      void analyze_tpars(const std::vector<const Trk::TrackParameters* >* tpars,
                         InDet::TRT_TrackExtensionToolCosmics::EventData &event_data) const;
 
      Trk::Surface* findBoundarySurface(const Trk::TrackParameters& par,
                                        Trk::PropDirection dir,
                                        InDet::TRT_TrackExtensionToolCosmics::EventData &event_data) const;
 
      static Amg::Vector3D intersect(const Trk::Surface *surf,const Trk::Perigee *per);
 
    };
 
  MsgStream&    operator << (MsgStream&   ,const TRT_TrackExtensionToolCosmics&);
  std::ostream& operator << (std::ostream&,const TRT_TrackExtensionToolCosmics&); 
 
} // end of name space
 
#endif // TRT_TrackExtensionToolCosmics_H