BehaviorTreeScene.h

Go to the documentation of this file.

#pragma once
 
#include "Adapters/DirectTreeAdapter.h"
#include "Core/Node.h"
#include "Interfaces/IScene.h"
#include "Interfaces/ITreeNode.h"
#include "Interfaces/ITreeStructure.h"
#include "Scenes/ForgeTreeCanvas.h"
#include <functional>
#include <memory>
#include <string>
#include <vector>
#include <wx/wx.h>

class BehaviorTreeScene : public IScene {
  public:
    // Callback type for node selection events
    using NodeSelectionCallback = std::function<void(EmberCore::ITreeNode *)>;

    BehaviorTreeScene(wxWindow *parent, const wxString &title = "Behavior Tree");
 
    virtual ~BehaviorTreeScene();
 
    // IScene interface implementation
    wxPanel *GetPanel() override { return m_panel; }
    wxString GetTitle() const override { return m_title; }
    wxString GetSceneType() const override { return "BehaviorTree"; }
    void OnActivated() override;
    void OnDeactivated() override;
    void Refresh() override;
    bool HasUnsavedChanges() const override { return m_hasUnsavedChanges; }
    bool Save() override;
 
    // Behavior tree specific methods
    void SetBehaviorTree(std::shared_ptr<EmberCore::Node> root);

    void SetBehaviorTree(std::shared_ptr<EmberCore::ITreeStructure> tree);

    EmberCore::Node *GetBehaviorTree() const { return m_behaviorTreeRoot.get(); }

    EmberCore::ITreeStructure *GetAbstractedTree() const { return m_treeAdapter.get(); }
 
    std::shared_ptr<EmberCore::ITreeStructure> GetTreeAdapter() const { return m_treeAdapter; }
    EmberForge::ForgeTreeCanvas *GetTreeVisualization() const { return m_treeVisualization; }

    void CreateSampleBehaviorTree();

    void SetTitle(const wxString &title);

    void SetNodeSelectionCallback(NodeSelectionCallback callback) { m_nodeSelectionCallback = callback; }

    void SetCanvasBackgroundColor(const wxColour &color);
 
    // Non-copyable
    BehaviorTreeScene(const BehaviorTreeScene &) = delete;
    BehaviorTreeScene &operator=(const BehaviorTreeScene &) = delete;
 
  private:
    void CreateLayout();
 
    // UI components
    wxPanel *m_panel;
    EmberForge::ForgeTreeCanvas *m_treeVisualization;
 
    // Data
    std::shared_ptr<EmberCore::Node> m_behaviorTreeRoot;      // Legacy storage for backward compatibility
    std::shared_ptr<EmberCore::ITreeStructure> m_treeAdapter; // Adapter for abstraction layer
    wxString m_title;
    bool m_hasUnsavedChanges;
 
    // Callbacks
    NodeSelectionCallback m_nodeSelectionCallback;
};