InDetUsedInFitTrackDecoratorTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 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 override;
    virtual void decorate() const override;
 
 
    // Private data:
   private:
 
 
    StringProperty m_vtxDecoName
      {this, "AMVFVerticesDecoName", "TTVA_AMVFVertices",
    "Name of the per-track AMVF vertices decoration"};
 
    StringProperty m_wgtDecoName
      {this, "AMVFWeightsDecoName", "TTVA_AMVFWeights",
    "Name of the per-track AMVF weights decoration"};
 
    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;
 
    SG::WriteDecorHandleKey<xAOD::TrackParticleContainer> m_wgtDecoKey;
 
 
  }; // class: InDetUsedInFitTrackDecoratorTool
} // namespace: InDet
 
#endif // end: INDETUSEDINFITTRACKDECORATORTOOL_INDETUSEDINFITTRACKDECORATORTOOL_H