BehaviorTreeScene.cpp

Go to the documentation of this file.

#include "Scenes/BehaviorTreeScene.h"
#include "Adapters/DirectTreeAdapter.h"
#include "App/App.h"
#include "App/MainFrame.h"
#include "Components/XMLFileDropTarget.h"
#include "Utils/Logger.h"
 
BehaviorTreeScene::BehaviorTreeScene(wxWindow *parent, const wxString &title)
    : m_panel(nullptr), m_treeVisualization(nullptr), m_behaviorTreeRoot(nullptr), m_treeAdapter(nullptr),
      m_title(title), m_hasUnsavedChanges(false) {
    m_panel = new wxPanel(parent, wxID_ANY);
    m_panel->SetBackgroundColour(wxColour(45, 45, 50));
 
    MainFrame *mainFrame = dynamic_cast<MainFrame *>(wxGetApp().GetTopWindow());
    if (mainFrame) {
        m_panel->SetDropTarget(
            new EmberForge::XMLFileDropTarget([mainFrame](const wxString &filePath, bool loadInCurrentScene) {
                if (loadInCurrentScene) {
                    mainFrame->LoadXMLFile(filePath, true);
                } else {
                    mainFrame->LoadXMLFileInNewScene(filePath);
                }
            }));
    }
 
    CreateLayout();
}
 
BehaviorTreeScene::~BehaviorTreeScene() {
    if (m_treeVisualization) {
        m_treeVisualization->SetNodeSelectionCallback(nullptr);
        m_treeVisualization->SetVisibilityChangeCallback(nullptr);
        m_treeVisualization->SetTree(nullptr);
        m_treeVisualization = nullptr;
    }
    m_nodeSelectionCallback = nullptr;
}
 
void BehaviorTreeScene::CreateLayout() {
    wxBoxSizer *sizer = new wxBoxSizer(wxVERTICAL);
 
    m_treeVisualization = new EmberForge::ForgeTreeCanvas(m_panel);
 
    m_treeVisualization->SetNodeSelectionCallback([this](EmberCore::ITreeNode *node) {
        if (m_nodeSelectionCallback) {
            m_nodeSelectionCallback(node);
        }
    });
 
    if (m_treeAdapter) {
        m_treeVisualization->SetTree(m_treeAdapter);
    }
 
    sizer->Add(m_treeVisualization, 1, wxEXPAND | wxALL, 5);
 
    m_panel->SetSizer(sizer);
}
 
void BehaviorTreeScene::OnActivated() {
    if (m_treeVisualization) {
        m_treeVisualization->RefreshCanvas();
        m_treeVisualization->SetFocus();
    }
}
 
void BehaviorTreeScene::OnDeactivated() {}
 
void BehaviorTreeScene::Refresh() {
    if (m_treeVisualization) {
        m_treeVisualization->RefreshCanvas();
    }
}
 
bool BehaviorTreeScene::Save() {
    m_hasUnsavedChanges = false;
    return true;
}
 
void BehaviorTreeScene::SetBehaviorTree(std::shared_ptr<EmberCore::Node> root) {
    m_behaviorTreeRoot = root;
 
    if (m_behaviorTreeRoot) {
        auto behaviorTree = std::make_shared<EmberCore::BehaviorTree>("Scene Tree");
        behaviorTree->SetRootNode(m_behaviorTreeRoot);
        m_treeAdapter = std::make_shared<EmberCore::DirectTreeAdapter>(behaviorTree);
        LOG_INFO("Scene",
                 "Using DirectTreeAdapter with BehaviorTree wrapper: " + std::string(m_behaviorTreeRoot->GetName()));
    } else {
        m_treeAdapter = nullptr;
    }
 
    if (m_treeVisualization && m_treeAdapter) {
        m_treeVisualization->SetTree(m_treeAdapter);
        m_hasUnsavedChanges = true;
    }
 
    LOG_INFO("Scene", "Behavior tree updated with shared root node: " +
                          std::string(m_behaviorTreeRoot ? m_behaviorTreeRoot->GetName() : "null"));
}
 
void BehaviorTreeScene::SetBehaviorTree(std::shared_ptr<EmberCore::ITreeStructure> tree) {
    m_treeAdapter = tree;
    m_behaviorTreeRoot = nullptr;
 
    if (m_treeVisualization && m_treeAdapter) {
        m_treeVisualization->SetTree(m_treeAdapter);
        m_hasUnsavedChanges = true;
    }
 
    LOG_INFO("Scene", "Behavior tree updated with abstracted tree: " + std::string(tree ? tree->GetName() : "null"));
}
 
void BehaviorTreeScene::CreateSampleBehaviorTree() {
    auto uniqueTree = EmberCore::NodeFactory::CreateSampleTree();
    m_behaviorTreeRoot = std::shared_ptr<EmberCore::Node>(uniqueTree.release());
 
    if (m_behaviorTreeRoot) {
        auto behaviorTree = std::make_shared<EmberCore::BehaviorTree>("Sample Tree");
        behaviorTree->SetRootNode(m_behaviorTreeRoot);
        m_treeAdapter = std::make_shared<EmberCore::DirectTreeAdapter>(behaviorTree);
        LOG_INFO("Scene", "Using DirectTreeAdapter with BehaviorTree wrapper for sample tree: " +
                              std::string(m_behaviorTreeRoot->GetName()));
    }
 
    if (m_treeVisualization && m_treeAdapter) {
        m_treeVisualization->SetTree(m_treeAdapter);
    }
 
    m_hasUnsavedChanges = true;
 
    LOG_INFO("Scene", "Created sample behavior tree with " +
                          std::to_string(m_behaviorTreeRoot ? m_behaviorTreeRoot->GetSubtreeNodeCount() : 0) +
                          " nodes");
}
 
void BehaviorTreeScene::SetTitle(const wxString &title) { m_title = title; }
 
void BehaviorTreeScene::SetCanvasBackgroundColor(const wxColour &color) {
    if (m_panel) {
        m_panel->SetBackgroundColour(color);
        m_panel->Refresh();
    }
}