DirectTreeAdapter.h

Go to the documentation of this file.

#pragma once

 
#include "Adapters/NodeAdapter.h"
#include "Core/BehaviorTree.h"
#include "Interfaces/ITreeStructure.h"
#include <memory>
#include <unordered_map>
 
namespace EmberCore {

class DirectTreeAdapter : public ITreeStructure {
  public:
    explicit DirectTreeAdapter(std::shared_ptr<BehaviorTree> tree);
    virtual ~DirectTreeAdapter() = default;
 
    // Tree structure access
    ITreeNode *GetRootNode() const override;
 
    // Tree properties
    const EmberCore::String &GetName() const override;
    const EmberCore::String &GetDescription() const override;
 
    // Tree statistics
    size_t GetNodeCount() const override;
    size_t GetMaxDepth() const override;
 
    // Node search operations
    ITreeNode *FindNodeById(size_t id) const override;
    ITreeNode *FindNodeByName(const EmberCore::String &name) const override;
    std::vector<ITreeNode *> FindNodesByType(ITreeNode::NodeType type) const override;
 
    // Tree traversal
    void TraverseNodes(std::function<void(ITreeNode *)> visitor) override;
    void TraverseNodes(std::function<void(const ITreeNode *)> visitor) const override;
 
    // Tree validation
    bool IsValid() const override;
    bool HasCycles() const override;
 
    // Utility operations
    std::vector<ITreeNode *> GetAllNodes() override;
    std::vector<const ITreeNode *> GetAllNodes() const override;
 
    // Tree metadata for visualization
    EmberCore::String GetTreeStructure() const override;
    void PrintTree() const override;
 
    // Access to the wrapped tree (for direct access when needed)
    std::shared_ptr<BehaviorTree> GetWrappedTree() const { return behavior_tree_; }
    std::shared_ptr<BehaviorTree> GetTree() const { return behavior_tree_; }
 
  private:
    std::shared_ptr<BehaviorTree> behavior_tree_;
 
    // Cache for root node adapter to maintain consistent pointer identity
    mutable std::unique_ptr<NodeAdapter> root_adapter_;
 
    // Node adapter cache for consistent pointer identity
    mutable std::unordered_map<Node *, std::unique_ptr<NodeAdapter>> node_adapter_cache_;
 
    // Helper methods
    ITreeNode::NodeType ConvertNodeType(Node::Type type) const;
    Node::Type ConvertNodeType(ITreeNode::NodeType type) const;
 
    // Ensure adapters are created and cached
    void EnsureRootAdapter() const;
    NodeAdapter *GetOrCreateNodeAdapter(Node *node) const;
};
 
} // namespace EmberCore