implementing getting and reading track container

Reconstruction/VKalVrt/GNNVertexConstructor/CMakeLists.txt

@@ -11,7 +11,7 @@ atlas_add_component( GNNVertexConstructor
 atlas_add_library(   GNNVertexConstructorLib
                      src/*.cxx
                      PUBLIC_HEADERS GNNVertexConstructor 
-                     LINK_LIBRARIES AthenaBaseComps )
+                     LINK_LIBRARIES AthenaBaseComps xAODTracking )
                      
 # Install files from the package:
 atlas_install_python_modules( python/*.py POST_BUILD_CMD ${ATLAS_FLAKE8} )

Reconstruction/VKalVrt/GNNVertexConstructor/GNNVertexConstructor/GNNVertexConstructorAlg.h

@@ -6,6 +6,7 @@
 #include "GNNVertexConstructor/IGNNVertexConstructorInterface.h"
 #include "GaudiKernel/ToolHandle.h"
 
+#include "xAODTracking/TrackParticleContainer.h"
 
 namespace Rec {
 
@@ -20,10 +21,10 @@ namespace Rec {
 
     private:
       ToolHandle<Rec::IGNNVertexConstructorInterface> m_testTool;
-      
+      SG::ReadHandleKey<xAOD::TrackParticleContainer> m_inTrackKey{this, "InputTrackContainer", "InDetTrackParticles", "Input track particle container"};      
         
 
   };
 }
 
-#endif
\ No newline at end of file
+#endif

Reconstruction/VKalVrt/GNNVertexConstructor/GNNVertexConstructor/GNNVertexConstructorTool.h

@@ -4,6 +4,8 @@
 #include "AthenaBaseComps/AthAlgTool.h"
 #include "GNNVertexConstructor/IGNNVertexConstructorInterface.h"
 
+#include "xAODTracking/TrackParticleContainer.h"
+
 namespace Rec {
 
    class GNNVertexConstructorTool : public AthAlgTool, virtual public IGNNVertexConstructorInterface {
@@ -19,11 +21,11 @@ namespace Rec {
 
 
       
-      unsigned int addTwoNumbers( const unsigned int & NoOne,
-                                                     const unsigned int & NoTwo) const final;
+      unsigned int addTwoNumbers( const unsigned int & NoOne, const unsigned int & NoTwo) const final;
+      virtual StatusCode decorateTracks( const xAOD::TrackParticleContainer& trkCont ) const;
 
     private:
   };
 }
 
-#endif
\ No newline at end of file
+#endif

Reconstruction/VKalVrt/GNNVertexConstructor/GNNVertexConstructor/IGNNVertexConstructorInterface.h

@@ -4,6 +4,7 @@
 // Gaudi includes
 #include "AthenaBaseComps/AthAlgTool.h"
 
+#include "xAODTracking/TrackParticleContainer.h"
  
 //------------------------------------------------------------------------
 namespace Rec {
@@ -22,11 +23,11 @@ namespace Rec {
     
     
   */
-      virtual unsigned int addTwoNumbers( const unsigned int & NoOne,
-                                                     const unsigned int & NoTwo) const =0;
+      virtual unsigned int addTwoNumbers( const unsigned int & NoOne, const unsigned int & NoTwo) const =0;
+      virtual StatusCode decorateTracks( const xAOD::TrackParticleContainer& trkCont ) const = 0;
 
   };
 
 }  //end namespace
 
-#endif
\ No newline at end of file
+#endif

Reconstruction/VKalVrt/GNNVertexConstructor/src/GNNVertexConstructorAlg.cxx

@@ -2,36 +2,53 @@
 
 namespace Rec {
 
-GNNVertexConstructorAlg::GNNVertexConstructorAlg(const std::string& name, ISvcLocator* pSvcLocator) : AthReentrantAlgorithm( name, pSvcLocator ),
-m_testTool("Rec::GNNVertexConstructorTool/testTool", this)
-{
-declareProperty("TestTool",m_testTool, "The test Tool");
-}
+  GNNVertexConstructorAlg::GNNVertexConstructorAlg(const std::string& name, ISvcLocator* pSvcLocator) 
+    : AthReentrantAlgorithm( name, pSvcLocator ),
+    m_testTool("Rec::GNNVertexConstructorTool/testTool", this){
+
+      declareProperty("TestTool",m_testTool, "The test Tool");
+      
+  }
 
-StatusCode GNNVertexConstructorAlg::initialize()
-{
+  StatusCode GNNVertexConstructorAlg::initialize(){
   
-  ATH_MSG_DEBUG("devAlg: In devAlg::initialize()");
-  ATH_CHECK( m_testTool.retrieve() );
-  return StatusCode::SUCCESS;
-}
+    ATH_MSG_DEBUG("devAlg: In devAlg::initialize()");
+    ATH_CHECK( m_testTool.retrieve() );
+    
+    ATH_CHECK(m_inTrackKey.initialize());
+
+    return StatusCode::SUCCESS;
 
-StatusCode GNNVertexConstructorAlg::finalize()
-{
-  ATH_MSG_DEBUG("devAlg: In devAlg::finalize()");
+  }
+
+  StatusCode GNNVertexConstructorAlg::finalize(){
+
+    ATH_MSG_DEBUG("devAlg: In devAlg::finalize()");
   
-  return StatusCode::SUCCESS;
-}
+    return StatusCode::SUCCESS;
+  }
+
+  StatusCode GNNVertexConstructorAlg::execute(const EventContext &ctx) const{
 
-StatusCode GNNVertexConstructorAlg::execute(const EventContext &ctx) const
-{
-  ATH_MSG_DEBUG("devAlg: In devAlg::execute()");
+    ATH_MSG_DEBUG("devAlg: In devAlg::execute()");
   
   
-  unsigned int sum = m_testTool -> addTwoNumbers(4, 9);
+    unsigned int sum = m_testTool -> addTwoNumbers(4, 9);
   
-  ATH_MSG_DEBUG("SUM is = " << sum );
-  return StatusCode::SUCCESS;
-}
+    ATH_MSG_DEBUG("SUM is = " << sum );
+
+
+    SG::ReadHandle<xAOD::TrackParticleContainer> inTracks(m_inTrackKey, ctx);
+    if (!inTracks.isValid()) {
+      if (!inTracks.isValid()) ATH_MSG_ERROR("TrackParticle container not found: " << m_inTrackKey.key());
+      return StatusCode::FAILURE;
+    }
+
 
-}
\ No newline at end of file
+    ATH_CHECK(m_testTool->decorateTracks(*inTracks));
+
+    return StatusCode::SUCCESS;
+
+  }
+
+}

Reconstruction/VKalVrt/GNNVertexConstructor/src/GNNVertexConstructorTool.cxx

@@ -5,47 +5,49 @@
 
 namespace Rec {
     
-    GNNVertexConstructorTool::GNNVertexConstructorTool(const std::string& type, const std::string& name, const IInterface* parent):
-    AthAlgTool(type,name,parent)
-    {
-//
-// Declare additional interface
-//
-    declareInterface< IGNNVertexConstructorInterface >(this);
-    ATH_MSG_DEBUG("GNNVertexConstructorTool constructor called");
+    GNNVertexConstructorTool::GNNVertexConstructorTool(const std::string& type, const std::string& name, const IInterface* parent)
+    : AthAlgTool(type,name,parent){
+
+      declareInterface< IGNNVertexConstructorInterface >(this);
+      ATH_MSG_DEBUG("GNNVertexConstructorTool constructor called");
     }   
      /* Destructor */
-    GNNVertexConstructorTool::~GNNVertexConstructorTool()
-    {
-    ATH_MSG_DEBUG("GNNVertexConstructorTool destructor called");
+    
+    GNNVertexConstructorTool::~GNNVertexConstructorTool(){
+
+      ATH_MSG_DEBUG("GNNVertexConstructorTool destructor called");
     }
 
 
 
-    StatusCode GNNVertexConstructorTool::initialize()
-    {
+    StatusCode GNNVertexConstructorTool::initialize(){
 
-    ATH_MSG_DEBUG("GNNVertexConstructor Tool in initialize()");
+      ATH_MSG_DEBUG("GNNVertexConstructor Tool in initialize()");
 
-    return StatusCode::SUCCESS;
+      return StatusCode::SUCCESS;
     }
     
-    StatusCode GNNVertexConstructorTool::finalize()
-    {
+    StatusCode GNNVertexConstructorTool::finalize(){
 
-    ATH_MSG_DEBUG("GNNVertexConstructor Tool in finalize()");
+      ATH_MSG_DEBUG("GNNVertexConstructor Tool in finalize()");
     
-    return StatusCode::SUCCESS;
+      return StatusCode::SUCCESS;
     }
 
 
-    unsigned int GNNVertexConstructorTool::addTwoNumbers( const unsigned int & NoOne,
-                                                     const unsigned int & NoTwo) const
-    
-    {
-    unsigned int sum=NoOne+NoTwo;
-    return sum;
+    unsigned int GNNVertexConstructorTool::addTwoNumbers( const unsigned int & NoOne, const unsigned int & NoTwo) const {
+      unsigned int sum=NoOne+NoTwo;
+      return sum;
     }
 
+    StatusCode GNNVertexConstructorTool::decorateTracks( const xAOD::TrackParticleContainer& trkCont ) const {
+
+      ATH_MSG_DEBUG("GNNVertexConstructor Tool decorating tracks");
+
+      int sum = trkCont.size();
+      ATH_MSG_DEBUG("Size is = " << sum );      
+
+      return StatusCode::SUCCESS;
+    }
 
 }  // end Rec namespace