MonitorTreeCanvas.cpp

Go to the documentation of this file.

#include "Visualization/MonitorTreeCanvas.h"
#include "Visualization/StatusOverlay.h"
 
namespace Ember {
namespace Monitor {
 
MonitorTreeCanvas::MonitorTreeCanvas(wxWindow *parent, wxWindowID id) : EmberUI::TreeCanvas(parent, id) {
    ApplyBaseConfig();
}
 
void MonitorTreeCanvas::ApplyBaseConfig() {
    auto &cfg = GetConfig();
 
    cfg.background_color = wxColour(0x2A, 0x2A, 0x2A);
    cfg.grid_color = wxColour(0x3E, 0x3E, 0x3E);
    cfg.grid_size = 20;
 
    cfg.node_color = wxColour(0x4B, 0x4B, 0x4B);
    cfg.border_color = wxColour(0x78, 0x78, 0x78);
    cfg.text_color = wxColour(0xB4, 0xB4, 0xB4);
    cfg.selected_border_color = wxColour(0x30, 0x5C, 0xDE);
    cfg.hovered_border_color = wxColour(0x90, 0xEE, 0x90);
    cfg.connection_color = wxColour(0xFF, 0xFF, 0xFF);
 
    cfg.action_type_color = wxColour(0x4A, 0x90, 0xD9);
    cfg.control_type_color = wxColour(0x5C, 0xB8, 0x5C);
    cfg.condition_type_color = wxColour(0xF0, 0xAD, 0x4E);
    cfg.decorator_type_color = wxColour(0x9B, 0x59, 0xB6);
    cfg.behavior_tree_type_color = wxColour(0xD9, 0x53, 0x4F);
 
    cfg.highlight_path_to_selected = true;
    cfg.path_highlight_color = wxColour(0x90, 0xEE, 0x90);
 
    cfg.enable_viewport_culling = true;
    cfg.viewport_culling_margin = 1500;
 
    cfg.show_navigation_info = false;
 
    SetShowGrid(true);
    SetShowMinimap(true);
    SetShowBreadcrumb(true);
    SetShowOverlayInfo(false);
}
 
void MonitorTreeCanvas::SetConnectionInfo(bool connected, int tickCount) {
    m_connected = connected;
    m_tickCount = tickCount;
}
 
wxColour MonitorTreeCanvas::GetNodeFillColor(EmberCore::ITreeNode *node, bool selected, bool hovered) {
    auto *statusProvider = GetStatusProvider();
    if (statusProvider) {
        int status = statusProvider->GetNodeStatus(static_cast<int64_t>(node->GetId()));
        if (status != 0)
            return EmberUI::StatusColors::GetFillColor(status);
    }
 
    wxColour base = GetConfig().node_color;
    if (selected) {
        return wxColour(std::min(255, base.Red() + 20), std::min(255, base.Green() + 20),
                        std::min(255, base.Blue() + 20));
    }
    if (hovered) {
        return wxColour(std::min(255, base.Red() + 10), std::min(255, base.Green() + 10),
                        std::min(255, base.Blue() + 10));
    }
    return base;
}
 
wxColour MonitorTreeCanvas::GetNodeBorderColor(EmberCore::ITreeNode *node, bool selected, bool hovered) {
    int64_t nodeId = static_cast<int64_t>(node->GetId());
    auto *statusProvider = GetStatusProvider();
 
    if (selected)
        return GetConfig().selected_border_color;
    if (hovered)
        return GetConfig().hovered_border_color;
 
    if (statusProvider) {
        if (statusProvider->IsNodeInExecutionPath(nodeId))
            return EmberUI::StatusColors::GetExecutionPathColor();
 
        int status = statusProvider->GetNodeStatus(nodeId);
        if (status != 0)
            return EmberUI::StatusColors::GetBorderColor(status);
    }
 
    return GetConfig().border_color;
}
 
wxColour MonitorTreeCanvas::GetNodeTextColor(EmberCore::ITreeNode *node, bool, bool) {
    auto *statusProvider = GetStatusProvider();
    if (statusProvider) {
        int status = statusProvider->GetNodeStatus(static_cast<int64_t>(node->GetId()));
        if (status != 0)
            return EmberUI::StatusColors::GetTextColor(status);
    }
    return GetConfig().text_color;
}
 
void MonitorTreeCanvas::OnBeforePaintOverlays(wxDC &dc) {
    dc.SetUserScale(1.0, 1.0);
    dc.SetDeviceOrigin(0, 0);
 
    wxSize panelSize = GetSize();
    int y = 10;
    int rightX = panelSize.x - 10;
 
    wxFont infoFont(9, wxFONTFAMILY_TELETYPE, wxFONTSTYLE_NORMAL, wxFONTWEIGHT_NORMAL);
    dc.SetFont(infoFont);
 
    wxString connText = m_connected ? wxString::Format("Connected | Ticks: %d", m_tickCount) : "Disconnected";
    wxColour connColor = m_connected ? wxColour(100, 200, 100) : wxColour(200, 100, 100);
    dc.SetTextForeground(connColor);
    wxSize connExtent = dc.GetTextExtent(connText);
    dc.DrawText(connText, rightX - connExtent.x, y);
    y += connExtent.y + 4;
 
    dc.SetTextForeground(wxColour(160, 160, 160));
    wxPoint viewOfs = GetViewOffset();
    wxString zoomText = wxString::Format("Zoom: %.0f%% | Pan: (%d, %d)", GetZoom() * 100, viewOfs.x, viewOfs.y);
    if (IsInFocusMode())
        zoomText += " | FOCUS MODE";
    wxSize zoomExtent = dc.GetTextExtent(zoomText);
    dc.DrawText(zoomText, rightX - zoomExtent.x, y);
    y += zoomExtent.y + 4;
 
    auto *selectedNode = GetSelectedNode();
    if (selectedNode) {
        wxString selText = "Selected: " + selectedNode->GetName() + " [" + selectedNode->GetTypeString() + "]";
 
        auto *statusProvider = GetStatusProvider();
        if (statusProvider) {
            int status = statusProvider->GetNodeStatus(static_cast<int64_t>(selectedNode->GetId()));
            wxString statusNames[] = {"Idle", "Running", "Success", "Failure", "Halted"};
            if (status >= 0 && status <= 4)
                selText += " | " + statusNames[status];
        }
 
        dc.SetTextForeground(wxColour(200, 200, 200));
        wxSize selExtent = dc.GetTextExtent(selText);
        dc.DrawText(selText, rightX - selExtent.x, y);
        y += selExtent.y + 4;
    }
 
    auto tree = GetTree();
    if (tree && tree->HasRootNode()) {
        dc.SetTextForeground(wxColour(130, 130, 130));
        wxString statsText = wxString::Format("Nodes: %lu", static_cast<unsigned long>(tree->GetNodeCount()));
        wxSize statsExtent = dc.GetTextExtent(statsText);
        dc.DrawText(statsText, rightX - statsExtent.x, y);
    }
}
 
} // namespace Monitor
} // namespace Ember