NavGraph.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TrigCompositeUtils_NavGraph_h
#define TrigCompositeUtils_NavGraph_h
 
#include "AthContainers/AuxElement.h"
#include "xAODTrigger/TrigCompositeContainer.h"
#include "xAODTrigger/TrigCompositeAuxContainer.h"
 
 
namespace TrigCompositeUtils {
 
  class NavGraphNode {
    public:
      NavGraphNode(const Decision* me);
 
      ~NavGraphNode() = default;
 
      NavGraphNode(const NavGraphNode& obj) = delete;
 
      NavGraphNode(NavGraphNode&& obj) = delete;
      bool linksTo(NavGraphNode* to);
 
      void dropLinks(NavGraphNode* node);
 
      void keep();
 
      void resetKeep() ;
 
      bool getKeep() const;
 
      const Decision* node() const;
 
      const std::vector<NavGraphNode*>& seeds() const;
 
      const std::vector<NavGraphNode*>& children() const;
 
    private:
 
      static bool addIfNotDuplicate(std::vector<NavGraphNode*>& container, NavGraphNode* toAdd);
 
      const Decision* m_decisionObject; 
      std::vector<NavGraphNode*> m_filteredSeeds; 
      std::vector<NavGraphNode*> m_filteredChildren; 
      bool m_keepFlag; 
 
  };
 
  class NavGraph {
 
    public:
 
      NavGraph();
 
      ~NavGraph() = default;
 
      NavGraph(const NavGraph& obj) = delete;
 
      NavGraph(NavGraph&& obj) = delete;
 
      void addNode(const Decision* node, const Decision* comingFrom = nullptr);
 
      const std::vector<NavGraphNode*>& finalNodes() const;
 
      std::vector<NavGraphNode*> allNodes();
 
      size_t nodes() const;
 
      size_t edges() const;
 
      std::vector<const Decision*> thin();
 
      void printAllPaths(MsgStream& log, MSG::Level msgLevel = MSG::VERBOSE) const;
 
    private:
 
      void rewireNodeForRemoval(NavGraphNode& toBeDeleted);
 
      void recursivePrintNavPath(const NavGraphNode& nav, size_t level, MsgStream& log, MSG::Level msgLevel) const;
 
      std::map<const Decision*, size_t> m_nodePositionMap; 
      std::vector<std::unique_ptr<NavGraphNode>> m_nodes; 
      std::vector<NavGraphNode*> m_finalNodes; 
      size_t m_edges; 
  };
 
}
 
#endif // TrigCompositeUtils_NavGraph_h