TruthVertexFixerAlg.h

Go to the documentation of this file.

// Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
#ifndef XAODTRUTHCNV_TRUTHVERTEXFIXERALG_H
#define XAODTRUTHCNV_TRUTHVERTEXFIXERALG_H
 
// Framework include(s).
#include "AnaAlgorithm/AnaReentrantAlgorithm.h"
#include "AsgDataHandles/ReadHandleKey.h"
#include "AsgDataHandles/WriteHandleKey.h"
#include "AsgTools/PropertyWrapper.h"
 
// EDM include(s).
#include "xAODTruth/TruthVertexContainer.h"
 
// System include(s).
#include <string>
#include <vector>
 
namespace xAODMaker {
 
class TruthVertexFixerAlg final : public EL::AnaReentrantAlgorithm {
 
 public:
  using EL::AnaReentrantAlgorithm::AnaReentrantAlgorithm;
 
  virtual StatusCode initialize() override;
  virtual StatusCode execute(const EventContext& ctx) const override;
 
 private:
  SG::ReadHandleKey<xAOD::TruthVertexContainer> m_inputContainerKey{
      this, "InputContainer", "InputFileTruthVertices",
      "Input TruthVertices container"};
 
  SG::WriteHandleKey<xAOD::TruthVertexContainer> m_outputContainerKey{
      this, "OutputContainer", "TruthVertices",
      "Output TruthVertices container"};
 
  Gaudi::Property<std::vector<std::string>> m_particleLinks{
      this, "ParticleLinks", {}, "Names of the truth particle links to fix"};
 
  Gaudi::Property<std::vector<std::string>> m_vertexLinks{
      this, "VertexLinks", {}, "Names of the truth vertex links to fix"};
 
  Gaudi::Property<std::string> m_linkPrefixToRemove{
      this, "LinkPrefixToRemove", "", "Prefix to remove from the link names"};
 
};  // class TruthVertexFixerAlg
 
}  // namespace xAODMaker
 
#endif  // XAODTRUTHCNV_TRUTHVERTEXFIXERALG_H