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{false}; 
 
  };

  class NavGraph {
 
    public:

      NavGraph() = default;

      ~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{0}; 
  };
 
}
 
#endif // TrigCompositeUtils_NavGraph_h