InDetUsedInFitTrackDecoratorTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 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/TrackParticleContainerFwd.h"
#include "xAODTracking/VertexContainerFwd.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:
 
 
    std::string m_vtxDecoName;
 
    std::string m_wgtDecoName;
 
    SG::ReadHandleKey<xAOD::TrackParticleContainer> m_trkContKey;
 
    SG::ReadHandleKey<xAOD::VertexContainer> m_vtxContKey;
 
 
  private:
 
 
    SG::WriteDecorHandleKey<xAOD::TrackParticleContainer> m_vtxDecoKey;
 
    SG::WriteDecorHandleKey<xAOD::TrackParticleContainer> m_wgtDecoKey;
 
 
  }; // class: InDetUsedInFitTrackDecoratorTool
} // namespace: InDet
 
#endif // end: INDETUSEDINFITTRACKDECORATORTOOL_INDETUSEDINFITTRACKDECORATORTOOL_H