CachedGraph.hpp

#pragma once

#include "Export.hpp"
#include "CachedNode.hpp"

namespace Showcase
{

  class SHOWCASE_API CachedGraph : public std::enable_shared_from_this<CachedGraph>
  {
  public:
    CachedGraph(const GraphNode& root, GraphPtr graph);
    ~CachedGraph();

    CachedNodePtr getNode(const NodeId& id) const;

    void startCaching();

  private:
    void initNode(const GraphNode& node, std::unique_lock<std::mutex>&& lock);

    void nodeAdded(const GraphNode& node, const GraphNode& parent);
    void nodeRemoved(const GraphNode& node, const GraphNode& parent);

    void addEdge(const NodeId& setOwner, const NodeId& toSet,
                 std::unordered_map<NodeId, std::set<NodeId>>& edges);
    void removeEdge(const NodeId& from, const NodeId& to);

    void collectNodes(const NodeId& node, std::set<NodeId>& visited);


    mutable std::mutex m_nodeMutex;
    std::unordered_map<NodeId, CachedNodePtr> m_nodes;

    std::mutex m_edgeMutex;
    std::unordered_map<NodeId, std::set<NodeId>> m_outgoingEdges;

    std::unordered_map<NodeId, std::set<NodeId>> m_incomingEdges;

    std::weak_ptr<Graph> m_graph;
  };
  typedef std::shared_ptr<CachedGraph> CachedGraphPtr;

}