CustomAuiNotebook.cpp

Go to the documentation of this file.

#include "Utils/CustomAuiNotebook.h"
#include "Utils/Logger.h"
#include "Utils/ResourcePath.h"
#include <wx/dcbuffer.h>
 
namespace EmberForge {
 
// Event IDs for the buttons
enum { ID_ADD_TAB_BUTTON = wxID_HIGHEST + 5000, ID_MENU_BUTTON };
 
wxBEGIN_EVENT_TABLE(CustomAuiNotebook, wxAuiNotebook) EVT_SIZE(CustomAuiNotebook::OnSize) wxEND_EVENT_TABLE()
 
    CustomAuiNotebook::CustomAuiNotebook(wxWindow *parent, wxWindowID id, const wxPoint &pos, const wxSize &size,
                                         long style)
    : wxAuiNotebook(parent, id, pos, size, style), m_addButton(nullptr), m_menuButton(nullptr), m_buttonsEnabled(true) {
    // Create and set our custom tab art
    m_customTabArt = new CustomTabArt();
    SetArtProvider(m_customTabArt);
 
    // Load button icons and create the buttons
    LoadButtonIcons();
    CreateButtons();
}
 
void CustomAuiNotebook::LoadButtonIcons() {
    // Load plus button icons
    wxString plusIconPath = ResourcePath::GetDir("buttons/plus");
    m_plusNormal = wxBitmap(plusIconPath + "Plus_Normal_20.png", wxBITMAP_TYPE_PNG);
    m_plusHovered = wxBitmap(plusIconPath + "Plus_Hovered_20.png", wxBITMAP_TYPE_PNG);
    m_plusPressed = wxBitmap(plusIconPath + "Plus_Pressed_20.png", wxBITMAP_TYPE_PNG);
 
    // Load menu button icons
    wxString menuIconPath = ResourcePath::GetDir("buttons/menu");
    m_menuNormal = wxBitmap(menuIconPath + "Menu_Normal_20.png", wxBITMAP_TYPE_PNG);
    m_menuHovered = wxBitmap(menuIconPath + "Menu_Hovered_20.png", wxBITMAP_TYPE_PNG);
    m_menuPressed = wxBitmap(menuIconPath + "Menu_Pressed_20.png", wxBITMAP_TYPE_PNG);
}
 
void CustomAuiNotebook::CreateButtons() {
    // Get the parent (SidePanel) to create buttons as its children
    // This allows buttons to be positioned over the tab bar area
    wxWindow *parentPanel = GetParent();
    if (!parentPanel) {
        LOG_ERROR("CustomAuiNotebook", "Cannot create buttons - no parent panel");
        return;
    }
 
    // Create add tab button as child of parent panel
    m_addButton = new wxBitmapButton(parentPanel, ID_ADD_TAB_BUTTON, m_plusNormal, wxDefaultPosition,
                                     wxSize(BUTTON_SIZE, BUTTON_SIZE), wxBORDER_NONE);
    m_addButton->SetBitmapMargins(2, 2);
    if (m_plusHovered.IsOk()) {
        m_addButton->SetBitmapHover(m_plusHovered);
    }
    if (m_plusPressed.IsOk()) {
        m_addButton->SetBitmapPressed(m_plusPressed);
    }
    m_addButton->SetToolTip("Add a new tab");
 
    // Bind button click event
    m_addButton->Bind(wxEVT_BUTTON, &CustomAuiNotebook::OnAddButtonClicked, this);
 
    // Create menu button as child of parent panel
    m_menuButton = new wxBitmapButton(parentPanel, ID_MENU_BUTTON, m_menuNormal, wxDefaultPosition,
                                      wxSize(BUTTON_SIZE, BUTTON_SIZE), wxBORDER_NONE);
    m_menuButton->SetBitmapMargins(2, 2);
    if (m_menuHovered.IsOk()) {
        m_menuButton->SetBitmapHover(m_menuHovered);
    }
    if (m_menuPressed.IsOk()) {
        m_menuButton->SetBitmapPressed(m_menuPressed);
    }
    m_menuButton->SetToolTip("Panel options");
 
    // Bind button click event
    m_menuButton->Bind(wxEVT_BUTTON, &CustomAuiNotebook::OnMenuButtonClicked, this);
 
    // Position the buttons initially
    UpdateButtonPositions();
}
 
void CustomAuiNotebook::UpdateButtonPositions() {
    if (!m_addButton || !m_menuButton) {
        return;
    }
 
    // Get the notebook's position relative to its parent (SidePanel)
    wxPoint notebookPos = GetPosition();
    wxSize notebookSize = GetSize();
 
    // Get the tab bar height
    int tabHeight = GetTabCtrlHeight();
    if (tabHeight <= 0) {
        tabHeight = 25; // Default fallback
    }
 
    // Calculate vertical center of the tab bar (relative to parent's coordinate space)
    int buttonY = notebookPos.y + (tabHeight - BUTTON_SIZE) / 2;
    if (buttonY < notebookPos.y + 2)
        buttonY = notebookPos.y + 2;
 
    // Position menu button at the right edge of the notebook
    int menuX = notebookPos.x + notebookSize.GetWidth() - BUTTON_MARGIN - BUTTON_SIZE;
    m_menuButton->SetPosition(wxPoint(menuX, buttonY));
 
    // Position add button to the left of menu button
    int addX = menuX - BUTTON_SPACING - BUTTON_SIZE;
    m_addButton->SetPosition(wxPoint(addX, buttonY));
 
    // Make sure buttons are visible
    m_addButton->Show(m_buttonsEnabled);
    m_menuButton->Show(m_buttonsEnabled);
 
    // Raise buttons to ensure they're on top
    m_addButton->Raise();
    m_menuButton->Raise();
}
 
void CustomAuiNotebook::SetTabBarButtonsEnabled(bool enabled) {
    m_buttonsEnabled = enabled;
    if (m_addButton) {
        m_addButton->Show(enabled);
    }
    if (m_menuButton) {
        m_menuButton->Show(enabled);
    }
    Refresh();
}
 
void CustomAuiNotebook::OnSize(wxSizeEvent &event) {
    event.Skip();
    // Update button positions when the notebook is resized
    CallAfter(&CustomAuiNotebook::UpdateButtonPositions);
}
 
void CustomAuiNotebook::OnAddButtonClicked(wxCommandEvent &event) {
    // Send the custom event to the parent (SidePanel)
    wxCommandEvent addEvent(EVT_TAB_ART_ADD_BUTTON);
    addEvent.SetEventObject(this);
    GetParent()->GetEventHandler()->ProcessEvent(addEvent);
}
 
void CustomAuiNotebook::OnMenuButtonClicked(wxCommandEvent &event) {
    // Send the custom event to the parent (SidePanel)
    wxCommandEvent menuEvent(EVT_TAB_ART_MENU_BUTTON);
    menuEvent.SetEventObject(this);
    GetParent()->GetEventHandler()->ProcessEvent(menuEvent);
}
 
} // namespace EmberForge