TruthNavigationDecorator.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef DERIVATIONFRAMEWORK_TRUTHNAVIGATIONDECORATOR_H
#define DERIVATIONFRAMEWORK_TRUTHNAVIGATIONDECORATOR_H
 
// Interface classes
#include "AthenaBaseComps/AthAlgTool.h"
#include "DerivationFrameworkInterfaces/IAugmentationTool.h"
 
// Handles
#include "StoreGate/ReadHandleKeyArray.h"
#include "StoreGate/ReadHandleKey.h" 
#include "StoreGate/WriteDecorHandleKeyArray.h"
 
// EDM includes -- typedefs, so can't just be forward declared
#include "xAODTruth/TruthParticleContainer.h"
#include "xAODTruth/TruthEventContainer.h"
 
// STL includes
#include <string>
#include <vector>
 
namespace DerivationFramework {
 
  class TruthNavigationDecorator : public AthAlgTool, public IAugmentationTool {
    public: 
      TruthNavigationDecorator(const std::string& t, const std::string& n, const IInterface* p);
      ~TruthNavigationDecorator();
      StatusCode initialize();
      virtual StatusCode addBranches() const;
 
    private:
      SG::ReadHandleKeyArray<xAOD::TruthParticleContainer> m_inputKeys
         {this, "InputCollections", {}, "Input truth particle collection keys"};
      SG::ReadHandleKey<xAOD::TruthEventContainer> m_truthEventKey
         {this, "TruthEventKey", "TruthEvents", "SG key for the TruthEvent container"};
      SG::WriteDecorHandleKeyArray<xAOD::TruthParticleContainer, std::vector<ElementLink<xAOD::TruthParticleContainer> >> m_parentLinksDecorKeys
         {this, "DoNotSet_parentDecorKeys", {}, "WriteHandleKeyArray - set internally but must be property"};
      SG::WriteDecorHandleKeyArray<xAOD::TruthParticleContainer, std::vector<ElementLink<xAOD::TruthParticleContainer> >> m_childLinksDecorKeys
         {this, "DoNotSet_childDecorKeys", {}, "WriteHandleKeyArray - set internally but must be property"};
      void find_parents( const xAOD::TruthParticle* part ,
                         std::vector<ElementLink<xAOD::TruthParticleContainer> >& parents ,
                         std::map<int,ElementLink<xAOD::TruthParticleContainer> >& linkMap ,
                         std::vector<int>& seen_particles ) const;
      void find_children( const xAOD::TruthParticle* part ,
                          std::vector<ElementLink<xAOD::TruthParticleContainer> >& parents ,
                          std::map<int,ElementLink<xAOD::TruthParticleContainer> >& linkMap ,
                          std::vector<int>& seen_particles ) const;
  }; 
}
 
#endif // DERIVATIONFRAMEWORK_TRUTHNAVIGATIONDECORATOR_H