MonitorFrame.h

Go to the documentation of this file.

#pragma once
 
#include <memory>
 
#include <wx/aui/aui.h>
#include <wx/wx.h>
 
#include "Interfaces/ITreeNode.h"
#include "Network/Builder/StateManager.h"
#include "Network/Builder/TreeBuilder.h"
#include "Network/Codec/FlatBufferCodec.h"
#include "Network/Codec/MessageFrame.h"
#include "Network/TCPServer.h"
#include "Utils/ProportionalLayout.h"
 
namespace Ember {
namespace Monitor {
 
class ConnectionPanel;
class MainPanel;
class RightSidePanel;
class MonitorNavigatorTab;
class PropertiesTab;

class MonitorFrame : public wxFrame {
  public:
    MonitorFrame(const wxString &title, const wxPoint &pos, const wxSize &size);
    ~MonitorFrame();
 
  private:
    void CreateMenuBar();
    void CreateLayout();
    void SetupCallbacks();
 
    void OnClose(wxCloseEvent &event);
    void OnQuit(wxCommandEvent &event);
    void OnAbout(wxCommandEvent &event);
    void OnPollTimer(wxTimerEvent &event);
    void OnFrameResize(wxSizeEvent &event);
 
    void OnClientConnect(const std::string &address);
    void OnClientDisconnect();
    void OnServerMessage(const Network::MessageFrame &frame);
 
    void HandleHandshake(const Network::MessageFrame &frame);
    void HandleTreeInit(const Network::MessageFrame &frame);
    void HandleTickUpdate(const Network::MessageFrame &frame);
    void HandleBlackboardUpdate(const Network::MessageFrame &frame);
    void HandleTreeReset(const Network::MessageFrame &frame);
    void HandleDisconnect(const Network::MessageFrame &frame);
 
    void OnNodeSelected(EmberCore::ITreeNode *node);
    void RefreshVisualization();
    void UpdateStatusBar();
 
    bool m_inNodeSelection = false;
 
    enum class ConnectionState { Idle, Connected, AwaitingTree, TreeReady };
 
    TCPServer m_server;
    Network::TreeBuilder m_treeBuilder;
    Network::FlatBufferCodec m_codec;
    std::shared_ptr<Network::StateManager> m_stateManager;
    std::shared_ptr<EmberCore::BehaviorTree> m_currentTree;
 
    std::string m_currentTreeId;
    ConnectionState m_connectionState = ConnectionState::Idle;
    bool m_stateChanged = false;
    Network::MessageFrame m_pendingTickUpdate;
    bool m_hasPendingTickUpdate = false;
 
    wxAuiManager *m_auiManager;
    ConnectionPanel *m_connectionPanel;
    MainPanel *m_mainPanel;
    RightSidePanel *m_rightPanel;
    MonitorNavigatorTab *m_navigatorTab;
    PropertiesTab *m_propertiesTab;
    wxStatusBar *m_statusBar;
 
    wxTimer m_pollTimer;
 
    EmberUI::ProportionalLayout m_layout;
    wxSize m_lastClientSize;
 
    wxDECLARE_EVENT_TABLE();
};
 
} // namespace Monitor
} // namespace Ember