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