HardScatterVertexDecorator.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef DERIVATIONFRAMEWORK_HARDSCATTERVERTEXDECORATOR_H
#define DERIVATIONFRAMEWORK_HARDSCATTERVERTEXDECORATOR_H
 
// Framework include(s):
#include "AsgTools/PropertyWrapper.h"
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/WriteDecorHandleKey.h"
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ToolHandle.h"
 
// EDM include(s):
#include "xAODTracking/VertexContainerFwd.h"
#include "xAODEventInfo/EventInfo.h"
 
// Tool include(s):
#include "DerivationFrameworkInterfaces/IAugmentationTool.h"
#include "InDetRecToolInterfaces/IInDetHardScatterSelectionTool.h"
 
namespace DerivationFramework {
 
  class HardScatterVertexDecorator : public extends<AthAlgTool, IAugmentationTool>
  {
    // Public methods:
  public:

 
    using base_class::base_class;



    virtual StatusCode initialize() override final;

    virtual StatusCode addBranches(const EventContext& ctx) const override final;


    // Private data:
  private:


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

    ToolHandle<InDet::IInDetHardScatterSelectionTool> m_vtxSelectTool{this, "HardScatterSelectionTool", "",
                                                                      "IInDetHardScatterSelectionTool for selecting the hardscatter vertex" };

    SG::ReadHandleKey<xAOD::EventInfo> m_evtInfoKey {this, "EventInfo", "EventInfo", "EventInfo key"};

    SG::WriteDecorHandleKey<xAOD::EventInfo> m_evtDecoKey{this, "HardScatterDecoName", m_evtInfoKey, "hardScatterVertexLink",
                                              "Name of the hardscatter vertex decoration (applied to xAOD::EventInfo)"};

 
  }; // end: class HardScatterVertexDecorator
} // end: namespace DerivationFramework
 
#endif // end: DERIVATIONFRAMEWORK_HARDSCATTERVERTEXDECORATOR_H