InDetUsedInFitTrackDecoratorTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef INDETUSEDINFITTRACKDECORATORTOOL_INDETUSEDINFITTRACKDECORATORTOOL_H
#define INDETUSEDINFITTRACKDECORATORTOOL_INDETUSEDINFITTRACKDECORATORTOOL_H
 
// Framework include(s):
#include "AsgTools/AsgTool.h"
#include "AsgDataHandles/ReadHandleKey.h"
#include "AsgDataHandles/WriteDecorHandleKey.h"
 
// EDM include(s):
#include "xAODTracking/TrackParticleContainer.h"
#include "xAODTracking/VertexContainer.h"
 
// Tool include(s)
#include "InDetRecToolInterfaces/IInDetUsedInFitTrackDecoratorTool.h"
 
// STL includes:
#include <string>

 
namespace InDet {

  class InDetUsedInFitTrackDecoratorTool
    : public asg::AsgTool,
      public virtual IInDetUsedInFitTrackDecoratorTool
  {
 
    ASG_TOOL_CLASS(InDetUsedInFitTrackDecoratorTool, IInDetUsedInFitTrackDecoratorTool)
 
    
    // Public methods:
    public:


    InDetUsedInFitTrackDecoratorTool(const std::string& name);

    ~InDetUsedInFitTrackDecoratorTool();



    virtual StatusCode initialize() override;

    virtual StatusCode finalize() override;



    virtual void decorate(const xAOD::TrackParticleContainer* trkCont, const xAOD::VertexContainer* vtxCont, const EventContext& ctx) const override;
    virtual void decorate(const EventContext& ctx) const override;


    // Private data:
  private:


    SG::ReadHandleKey<xAOD::TrackParticleContainer> m_trkContKey
      {this, "TrackContainer", "InDetTrackParticles",
       "Name of the track particle container"};

    SG::ReadHandleKey<xAOD::VertexContainer> m_vtxContKey
      {this, "VertexContainer", "PrimaryVertices",
       "Name of the primary vertex container"};

 
  private:


    SG::WriteDecorHandleKey<xAOD::TrackParticleContainer> m_vtxDecoKey
    { this, "AMVFVerticesDecoName", m_trkContKey, "TTVA_AMVFVertices",
        "Name of the per-track AMVF vertices decoration"};

    SG::WriteDecorHandleKey<xAOD::TrackParticleContainer> m_wgtDecoKey
    { this, "AMVFWeightsDecoName", m_trkContKey, "TTVA_AMVFWeights",
        "Name of the per-track AMVF weights decoration"};

 
  }; // class: InDetUsedInFitTrackDecoratorTool
} // namespace: InDet
 
#endif // end: INDETUSEDINFITTRACKDECORATORTOOL_INDETUSEDINFITTRACKDECORATORTOOL_H