From f775fa01a867be37ea145c358777a81790b7fd71 Mon Sep 17 00:00:00 2001
From: Savannah Rose Shively <sshively@lxplus751.cern.ch>
Date: Wed, 28 Sep 2022 15:52:55 +0200
Subject: [PATCH] add extrapolator tool code

---
 .../SingleTrackExtrapolation/src/SingleTrackExtrapolation.cxx | 1 +
 .../SingleTrackExtrapolation/src/SingleTrackExtrapolation.h   | 4 ++++
 2 files changed, 5 insertions(+)

diff --git a/Tracking/Vertexing/SingleTrackExtrapolation/src/SingleTrackExtrapolation.cxx b/Tracking/Vertexing/SingleTrackExtrapolation/src/SingleTrackExtrapolation.cxx
index e5aaba86..496ca0f4 100644
--- a/Tracking/Vertexing/SingleTrackExtrapolation/src/SingleTrackExtrapolation.cxx
+++ b/Tracking/Vertexing/SingleTrackExtrapolation/src/SingleTrackExtrapolation.cxx
@@ -22,6 +22,7 @@ SingleTrackExtrapolation::SingleTrackExtrapolation(const std::string &name, ISvc
 StatusCode SingleTrackExtrapolation::initialize() {
     ATH_CHECK( m_trackCollection.initialize() );
     ATH_CHECK( m_mcEventKey.initialize() );
+    ATH_CHECK(m_extrapolationTool.retrieve());
     //ATH_CHECK(detStore()->retrieve(m_idHelper, "FaserSCT_ID"));
 
     m_tree = new TTree("trackCounts", "Track Counts");
diff --git a/Tracking/Vertexing/SingleTrackExtrapolation/src/SingleTrackExtrapolation.h b/Tracking/Vertexing/SingleTrackExtrapolation/src/SingleTrackExtrapolation.h
index 843a5b0b..4239700f 100644
--- a/Tracking/Vertexing/SingleTrackExtrapolation/src/SingleTrackExtrapolation.h
+++ b/Tracking/Vertexing/SingleTrackExtrapolation/src/SingleTrackExtrapolation.h
@@ -21,6 +21,8 @@
 #include <TH1.h>
 #include <TProfile.h>
 
+#include "FaserActsGeometryInterfaces/IFaserActsExtrapolationTool.h"
+
 
 class FaserSCT_ID;
 
@@ -48,6 +50,8 @@ class SingleTrackExtrapolation : public AthReentrantAlgorithm, AthHistogramming
         SG::ReadHandleKey<TrackCollection> m_trackCollection {this, "TrackCollection", "CKFTrackCollection", "Input track collection name" };
         SG::ReadHandleKey<McEventCollection> m_mcEventKey { this, "McEventCollection", "TruthEvent" };
 
+        ToolHandle<IFaserActsExtrapolationTool> m_extrapolationTool { this, "ExtrapolationTool", "FaserActsExtrapolationTool" };  
+
         mutable TTree* m_tree;
         ServiceHandle<ITHistSvc> m_histSvc;
 
-- 
GitLab