MainFrame.h

Go to the documentation of this file.

#pragma once
 
#include "Components/PanelDescriptor.h"
#include "Interfaces/ITreeNode.h"
#include "Utils/ProportionalLayout.h"
#include <map>
#include <string>
#include <wx/aui/framemanager.h>
#include <wx/checkbox.h>
#include <wx/listbox.h>
#include <wx/notebook.h>
#include <wx/scrolwin.h>
#include <wx/spinctrl.h>
#include <wx/splitter.h>
#include <wx/srchctrl.h>
#include <wx/treectrl.h>
#include <wx/wx.h>
 
// Forward declarations
class BottomPanel;
class LeftSidePanel;
class RightSidePanel;
class SimplePositioningFrame;
class PropertiesTab;
class ForgeNavigatorTab;
class MainPanel;
class BehaviorTreeScene;
class BlackboardScene;
 
namespace EmberCore {
class ITreeNode;
class Node;
class Blackboard;
class BehaviorTree;
class BehaviorTreeProject;
} // namespace EmberCore

class MainFrame : public wxFrame {
  public:
    MainFrame(const wxString &title, const wxPoint &pos, const wxSize &size, long style = wxDEFAULT_FRAME_STYLE);

    virtual ~MainFrame();
 
    // Helper method to log messages (moved to public so panels can access it)
    void LogMessage(const wxString &message);
 
    // Panel communication methods
    void OnHierarchySelectionChanged(EmberCore::ITreeNode *selectedNode);
    void OnSceneSelectionChanged(EmberCore::ITreeNode *selectedNode);
    void CenterOnNode(EmberCore::ITreeNode *node);
 
    void InvalidateCanvasLayout();
    void RefreshHierarchyTree();
 
    // Unified selection synchronization
    void SyncSelection(EmberCore::ITreeNode *selectedNode);
 
    // Get the shared behavior tree from the active scene
    std::shared_ptr<EmberCore::Node> GetSharedBehaviorTree() const;
 
    // Update properties tab reference (called when a new properties tab is created)
    void UpdatePropertiesTabReference(PropertiesTab *propertiesTab);
 
    // Navigator tab reference
    void UpdateNavigatorTabReference(ForgeNavigatorTab *navigatorTab);
    void OnNavigatorTabClosed();
 
    // Tab lifecycle notifications
    void OnPropertiesTabClosed();
 
    // AUI management methods
    wxAuiManager *GetAuiManager() { return m_auiManager; }
 
    // File loading methods (used by menu and drag-and-drop)
    void LoadXMLFile(const wxString &filePath, bool confirmOverride = false);
    void LoadXMLFileInNewScene(const wxString &filePath);
 
    // Open a tree in a separate scene tab
    void OpenTreeInNewScene(const std::string &treeId);
 
    bool IsActiveSceneBehaviorTree() const;
 
    // Project methods
    void NewProject();
    void OpenProject();
    void OpenProject(const wxString &projectPath);
    void CloseProject();
    void SaveProject();
    void ShowProjectSettings();
    bool HasActiveProject() const;
    std::shared_ptr<EmberCore::BehaviorTreeProject> GetActiveProject() const;
 
    // Preferences refresh (called by PreferencesDialog)
    void RefreshPreferences(); // Apply preference changes immediately
 
    // Update panel toggle button icons to match current panel visibility
    void UpdatePanelToggleButtons();
 
    // Count how many instances of a tab exist across all panels
    int CountTabGlobally(const wxString &tabName) const;
 
    // Canvas refresh (called when tree structure changes)
    void RefreshCanvasVisualization();
 
    // Tree navigation (used by ForgeNavigatorTab)
    void LoadTreeIntoScene(const std::string &treeId);
    void UpdateTreeSelector();
    const std::string &GetCurrentTreeId() const { return m_currentTreeId; }
    void NavigateToBlackboard(const std::string &bbId);
 
  private:
    void OnExit(wxCommandEvent &event);
    void OnAbout(wxCommandEvent &event);
    void OnLoadXML(wxCommandEvent &event);   // XML loading with libxml2
    void OnSaveXML(wxCommandEvent &event);   // XML saving
    void OnSaveAsXML(wxCommandEvent &event); // XML save as
 
    // Project menu event handlers
    void OnNewProject(wxCommandEvent &event);
    void OnOpenProject(wxCommandEvent &event);
    void OnCloseProject(wxCommandEvent &event);
    void OnSaveProject(wxCommandEvent &event);
    void OnProjectSettings(wxCommandEvent &event);
 
    // View menu event handlers
    void OnToggleMaximize(wxCommandEvent &event);
    void OnNextScene(wxCommandEvent &event);
    void OnPreviousScene(wxCommandEvent &event);
    void OnResetZoom(wxCommandEvent &event);
    void OnResetUI(wxCommandEvent &event);
 
    // Settings menu event handlers
    void OnPreferences(wxCommandEvent &event);
    void OnParserConfig(wxCommandEvent &event);
 
    // Tool button event handlers
    void OnEditorTool(wxCommandEvent &event);
    void OnMonitorTool(wxCommandEvent &event);
    void OnLogReplayTool(wxCommandEvent &event);
 
    // Panel toggle event handlers
    void OnToggleLeftPanel(wxCommandEvent &event);
    void OnToggleBottomPanel(wxCommandEvent &event);
    void OnToggleRightPanel(wxCommandEvent &event);
 
    // Performance monitoring event handlers
    void OnPaint(wxPaintEvent &event);
    void OnIdle(wxIdleEvent &event);
    void OnFrameResize(wxSizeEvent &event);
 
    // Setup methods
    void CreateMenuAndStatusBar();
    void CreatePanelToggleToolbar();
    void InitializeConfigurationMenu();
    void CreatePanelLayout();
    void RefreshMenuHotkeys();
 
    // Save helper method
    void DoSaveXML(const wxString &filePath);
 
    void CreateProjectOverviewScene();
 
    // Member variables
    wxAuiManager *m_auiManager; // Unified AUI manager for all panels
    LeftSidePanel *m_leftPanel;
    RightSidePanel *m_rightPanel;
    BottomPanel *m_bottomPanel;
    PropertiesTab *m_propertiesTab;
    ForgeNavigatorTab *m_navigatorTab;
    MainPanel *m_scenePanel;
    wxToolBar *m_panelToggleToolbar; // Toolbar for panel visibility toggles
 
    // File tracking
    wxString m_currentFilePath;
    bool m_treeModified = false;
 
    // Project tracking
    std::shared_ptr<EmberCore::BehaviorTreeProject> m_activeProject;
    std::map<std::string, std::shared_ptr<EmberCore::BehaviorTree>> m_projectTrees;
    std::map<std::string, std::shared_ptr<EmberCore::Blackboard>> m_projectBlackboards;
    std::map<std::string, std::vector<std::string>> m_blackboardIncludesMap;
    BlackboardScene *m_blackboardScene = nullptr;
    std::string m_currentTreeId;
    std::string m_mainTreeName;
 
    // Proportional panel layout
    EmberUI::ProportionalLayout m_layout;
    wxSize m_lastClientSize;
 
    // Bidirectional synchronization control
    bool m_isUpdatingFromHierarchy = false;
    bool m_isUpdatingFromScene = false;
 
    wxDECLARE_EVENT_TABLE();
};
 
enum {
    ID_LoadXML = 1,                   // ID for XML loading (libxml2)
    ID_SaveXML = wxID_HIGHEST + 50,   // ID for XML saving
    ID_SaveAsXML = wxID_HIGHEST + 51, // ID for XML save as
 
    // Project menu IDs
    ID_NewProject = wxID_HIGHEST + 60,
    ID_OpenProject,
    ID_CloseProject,
    ID_SaveProject,
    ID_ProjectSettings,
 
    // View menu IDs
    ID_ToggleMaximize,
    ID_NextScene,
    ID_PreviousScene,
    ID_ResetZoom,
    ID_ResetUI,
 
    // Settings menu IDs
    ID_Preferences,
    ID_ParserConfig,
 
    // Tool IDs - Updated to new 3-button layout
    ID_EditorTool,
    ID_MonitorTool,
    ID_LogReplayTool,
 
    // Panel toggle IDs
    ID_ToggleLeftPanel,
    ID_ToggleBottomPanel,
    ID_ToggleRightPanel
};