TCPTreeAdapter.h

Go to the documentation of this file.

#pragma once
 
#include <memory>
#include <unordered_map>
 
#include "Adapters/NodeAdapter.h"
#include "Core/BehaviorTree.h"
#include "Interfaces/ITreeStructure.h"
#include "Network/Builder/StateManager.h"
 
namespace Ember {
namespace Network {
 
class TCPTreeAdapter : public EmberCore::ITreeStructure {
  public:
    using NodeStatus = Protocol::NodeStatus;
 
    TCPTreeAdapter(std::shared_ptr<EmberCore::BehaviorTree> tree, std::shared_ptr<StateManager> state_manager);
    ~TCPTreeAdapter() override = default;
 
    EmberCore::ITreeNode *GetRootNode() const override;
    bool HasRootNode() const override { return m_tree && m_tree->HasRootNode(); }
 
    const EmberCore::String &GetName() const override;
    const EmberCore::String &GetDescription() const override;
 
    size_t GetNodeCount() const override;
    size_t GetMaxDepth() const override;
 
    EmberCore::ITreeNode *FindNodeById(size_t id) const override;
    EmberCore::ITreeNode *FindNodeByName(const EmberCore::String &name) const override;
    std::vector<EmberCore::ITreeNode *> FindNodesByType(EmberCore::ITreeNode::NodeType type) const override;
 
    void TraverseNodes(std::function<void(EmberCore::ITreeNode *)> visitor) override;
    void TraverseNodes(std::function<void(const EmberCore::ITreeNode *)> visitor) const override;
 
    bool IsValid() const override;
    bool HasCycles() const override;
 
    std::vector<EmberCore::ITreeNode *> GetAllNodes() override;
    std::vector<const EmberCore::ITreeNode *> GetAllNodes() const override;
 
    EmberCore::String GetTreeStructure() const override;
    void PrintTree() const override;
 
    NodeStatus GetNodeStatus(size_t node_id) const;
    bool IsNodeInExecutionPath(size_t node_id) const;
    int64_t GetCurrentTick() const;
 
    std::shared_ptr<StateManager> GetStateManager() { return m_stateManager; }
    std::shared_ptr<const StateManager> GetStateManager() const { return m_stateManager; }
 
    std::shared_ptr<EmberCore::BehaviorTree> GetBehaviorTree() { return m_tree; }
    std::shared_ptr<const EmberCore::BehaviorTree> GetBehaviorTree() const { return m_tree; }
 
  private:
    EmberCore::NodeAdapter *GetOrCreateAdapter(EmberCore::Node *node) const;
    void BuildAdapterCache() const;
 
    std::shared_ptr<EmberCore::BehaviorTree> m_tree;
    std::shared_ptr<StateManager> m_stateManager;
 
    mutable std::unordered_map<size_t, std::unique_ptr<EmberCore::NodeAdapter>> m_adapters;
    mutable EmberCore::NodeAdapter *m_rootAdapter = nullptr;
    mutable bool m_adapterCacheBuilt = false;
};
 
} // namespace Network
} // namespace Ember