action_toolbar.cpp

Go to the documentation of this file.

/*
 * This program source code file is part of KiCad, a free EDA CAD application.
 *
 * Copyright (C) 2019 CERN
 * Copyright The KiCad Developers, see CHANGELOG.txt for contributors.
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */
 
#include <algorithm>
#include <advanced_config.h>
#include <bitmaps.h>
#include <bitmap_store.h>
#include <eda_draw_frame.h>
#include <functional>
#include <kiplatform/ui.h>
#include <math/util.h>
#include <memory>
#include <pgm_base.h>
#include <settings/common_settings.h>
#include <trace_helpers.h>
#include <tool/action_toolbar.h>
#include <tool/tool_action.h>
#include <tool/tool_event.h>
#include <tool/tool_interactive.h>
#include <tool/tool_manager.h>
#include <tool/ui/toolbar_configuration.h>
#include <tool/ui/toolbar_context_menu_registry.h>
#include <widgets/bitmap_button.h>
#include <widgets/wx_aui_art_providers.h>
 
#include <wx/log.h>
#include <wx/popupwin.h>
#include <wx/renderer.h>
#include <wx/sizer.h>
#include <wx/dcclient.h>
#include <wx/settings.h>
 
#ifdef __WXMSW__
#include <windows.h>
#endif
 
// Needed to handle adding the plugins to the toolbar
// TODO (ISM): This should be better abstracted away from the toolbars
#include <api/api_plugin_manager.h>
 
 
ACTION_GROUP::ACTION_GROUP( const std::string_view& aName )
{
    m_name = aName;
    m_id   = ACTION_MANAGER::MakeActionId( m_name );
    m_defaultAction = nullptr;
}
 
 
ACTION_GROUP::ACTION_GROUP( const std::string_view& aName,
                            const std::vector<const TOOL_ACTION*>& aActions )
{
    m_name = aName;
    m_id   = ACTION_MANAGER::MakeActionId( m_name );
 
    SetActions( aActions );
}
 
 
void ACTION_GROUP::SetActions( const std::vector<const TOOL_ACTION*>& aActions )
{
    wxASSERT_MSG( aActions.size() > 0, wxS( "Action groups must have at least one action" ) );
 
    // The default action is just the first action in the vector
    m_actions       = aActions;
    m_defaultAction = m_actions[0];
}
 
int ACTION_GROUP::GetUIId() const
{
    return m_id + TOOL_ACTION::GetBaseUIId();
}
 
 
void ACTION_GROUP::SetDefaultAction( const TOOL_ACTION& aDefault )
{
    bool valid = std::any_of( m_actions.begin(), m_actions.end(),
                              [&]( const TOOL_ACTION* aAction ) -> bool
                              {
                                  // For some reason, we can't compare the actions directly
                                  return aAction->GetId() == aDefault.GetId();
                              } );
 
    wxASSERT_MSG( valid, wxS( "Action must be present in a group to be the default" ) );
 
    m_defaultAction = &aDefault;
}
 
 
#define PALETTE_BORDER FromDIP( 4 ) // The border around the palette buttons on all sides
#define BUTTON_BORDER  FromDIP( 1 ) // The border on the sides of the buttons that touch other buttons
 
 
ACTION_TOOLBAR_PALETTE::ACTION_TOOLBAR_PALETTE( wxWindow* aParent, bool aVertical ) :
        wxPopupTransientWindow( aParent, wxBORDER_NONE ),
        m_group( nullptr ),
        m_isVertical( aVertical ),
        m_panel( nullptr ),
        m_mainSizer( nullptr ),
        m_buttonSizer( nullptr )
{
    m_panel = new wxPanel( this, wxID_ANY );
    m_panel->SetBackgroundColour( wxSystemSettings::GetColour( wxSYS_COLOUR_WINDOW ) );
 
    // This sizer holds the buttons for the actions
    m_buttonSizer = new wxBoxSizer( aVertical ? wxVERTICAL : wxHORIZONTAL );
 
    // This sizer holds the other sizer, so that a consistent border is present on all sides
    m_mainSizer = new wxBoxSizer( aVertical ? wxVERTICAL : wxHORIZONTAL );
    m_mainSizer->Add( m_buttonSizer, wxSizerFlags().Border( wxALL, PALETTE_BORDER ) );
 
    m_panel->SetSizer( m_mainSizer );
 
    Connect( wxEVT_CHAR_HOOK, wxCharEventHandler( ACTION_TOOLBAR_PALETTE::onCharHook ), nullptr, this );
}
 
 
void ACTION_TOOLBAR_PALETTE::AddAction( const TOOL_ACTION& aAction )
{
    int            iconSize = Pgm().GetCommonSettings()->m_Appearance.toolbar_icon_size;
    wxBitmapBundle normalBmp = KiBitmapBundleDef( aAction.GetIcon(), iconSize );
    int            paddingDip = ( ToDIP( m_buttonSize.GetWidth() ) - iconSize ) / 2;
 
    BITMAP_BUTTON* button = new BITMAP_BUTTON( m_panel, aAction.GetUIId() );
 
    button->SetIsToolbarButton();
    button->SetBitmap( normalBmp );
    button->SetDisabledBitmap( KiDisabledBitmapBundleDef( aAction.GetIcon(), iconSize ) );
    button->SetPadding( paddingDip );
    button->SetToolTip( aAction.GetButtonTooltip() );
    button->AcceptDragInAsClick();
    button->SetBitmapCentered();
 
    m_buttons[aAction.GetUIId()] = button;
 
    if( m_isVertical )
        m_buttonSizer->Add( button, wxSizerFlags().Border( wxTOP | wxBOTTOM, BUTTON_BORDER ) );
    else
        m_buttonSizer->Add( button, wxSizerFlags().Border( wxLEFT | wxRIGHT, BUTTON_BORDER ) );
 
    m_buttonSizer->Layout();
}
 
 
void ACTION_TOOLBAR_PALETTE::EnableAction( const TOOL_ACTION& aAction, bool aEnable )
{
    auto it = m_buttons.find( aAction.GetUIId() );
 
    if( it != m_buttons.end() )
        it->second->Enable( aEnable );
}
 
 
void ACTION_TOOLBAR_PALETTE::CheckAction( const TOOL_ACTION& aAction, bool aCheck )
{
    auto it = m_buttons.find( aAction.GetUIId() );
 
    if( it != m_buttons.end() )
        it->second->Check( aCheck );
}
 
 
wxSize ACTION_TOOLBAR_PALETTE::GetFittedSize()
{
    m_mainSizer->Fit( m_panel );
    SetClientSize( m_panel->GetSize() );
 
    return m_panel->GetSize();
}
 
 
void ACTION_TOOLBAR_PALETTE::Popup( wxWindow* aFocus )
{
    GetFittedSize();
 
    wxPopupTransientWindow::Popup( aFocus );
}
 
 
void ACTION_TOOLBAR_PALETTE::onCharHook( wxKeyEvent& aEvent )
{
    // Allow the escape key to dismiss this popup
    if( aEvent.GetKeyCode() == WXK_ESCAPE )
        Dismiss();
    else
        aEvent.Skip();
}
 
 
#ifdef __WXMSW__
bool ACTION_TOOLBAR_PALETTE::MSWHandleMessage( WXLRESULT* aResult, WXUINT aMessage,
                                               WXWPARAM aWParam, WXLPARAM aLParam )
{
    // The Windows "Activate on hover" option (active window tracking) activates whatever
    // top-level window the pointer is over. As the pointer travels from the toolbar button
    // toward this palette it crosses the owner frame, which then steals activation and would
    // normally deactivate and dismiss this transient popup before the user can reach it.
    // Ignore that specific deactivation so the palette survives the trip. Dismissal by Escape,
    // by pressing a palette button, or by switching to any other window is unaffected because
    // those do not hand activation back to the owner frame.
    if( aMessage == WM_ACTIVATE && LOWORD( aWParam ) == WA_INACTIVE )
    {
        BOOL tracking = FALSE;
 
        if( ::SystemParametersInfo( SPI_GETACTIVEWINDOWTRACKING, 0, &tracking, 0 ) && tracking )
        {
            HWND      activated = reinterpret_cast<HWND>( aLParam );
            wxWindow* owner = MSWGetOwner();
 
            if( activated && owner && ::GetAncestor( activated, GA_ROOT ) == owner->GetHWND() )
                return wxPopupTransientWindowBase::MSWHandleMessage( aResult, aMessage, aWParam,
                                                                     aLParam );
        }
    }
 
    return wxPopupTransientWindow::MSWHandleMessage( aResult, aMessage, aWParam, aLParam );
}
#endif
 
 
ACTION_TOOLBAR::ACTION_TOOLBAR( EDA_BASE_FRAME* parent, wxWindowID id, const wxPoint& pos, const wxSize& size,
                                long style ) :
        wxAuiToolBar( parent, id, pos, size, style ),
        m_parent( parent ),
        m_paletteTimer( nullptr ),
        m_auiManager( nullptr ),
        m_toolManager( parent->GetToolManager() ),
        m_palette( nullptr )
{
    m_paletteTimer = new wxTimer( this );
 
    SetArtProvider( new WX_AUI_TOOLBAR_ART );
 
    Connect( wxEVT_COMMAND_TOOL_CLICKED, wxAuiToolBarEventHandler( ACTION_TOOLBAR::onToolEvent ), nullptr, this );
    Connect( wxEVT_AUITOOLBAR_RIGHT_CLICK, wxAuiToolBarEventHandler( ACTION_TOOLBAR::onRightClick ), nullptr, this );
    Connect( wxEVT_RIGHT_UP, wxMouseEventHandler( ACTION_TOOLBAR::onRightUp ), nullptr, this );
    Connect( wxEVT_AUITOOLBAR_BEGIN_DRAG, wxAuiToolBarEventHandler( ACTION_TOOLBAR::onItemDrag ), nullptr, this );
    Connect( wxEVT_LEFT_DOWN, wxMouseEventHandler( ACTION_TOOLBAR::onMouseClick ), nullptr, this );
    Connect( wxEVT_LEFT_UP, wxMouseEventHandler( ACTION_TOOLBAR::onMouseClick ), nullptr, this );
    Connect( m_paletteTimer->GetId(), wxEVT_TIMER, wxTimerEventHandler( ACTION_TOOLBAR::onTimerDone ), nullptr, this );
 
    Bind( wxEVT_SIZE,
          [&]( wxSizeEvent& aEvent )
          {
              CallAfter(
                      [&]()
                      {
                          SetOverflowVisible( !GetToolBarFits() );
                      } );
 
              aEvent.Skip();
          } );
 
    Bind( wxEVT_SYS_COLOUR_CHANGED, wxSysColourChangedEventHandler( ACTION_TOOLBAR::onThemeChanged ), this );
 
    Bind( wxEVT_DPI_CHANGED,
          [&]( wxDPIChangedEvent& aEvent )
          {
#ifdef __WXMSW__
              // Update values which are normally only initialized in wxAuiToolBar::Create.
              // FromDIP is no-op on backends other than wxMSW.
              SetToolPacking( FromDIP( 2 ) );
              SetToolBorderPadding( FromDIP( 3 ) );
 
              wxSize margin_lt = FromDIP( wxSize( 5, 5 ) );
              wxSize margin_rb = FromDIP( wxSize( 2, 2 ) );
              SetMargins( margin_lt.x, margin_lt.y, margin_rb.x, margin_rb.y );
 
              // Re-realize the toolbar to recalculate all item sizes with the new DPI.
              // This fixes excessive button padding when moving windows between displays
              // with different DPI scaling factors.
              if( GetToolCount() > 0 )
              {
                  UpdateControlWidths();
                  InvalidateBestSize();
                  KiRealize();
              }
#endif
 
              aEvent.Skip();
          } );
}
 
 
ACTION_TOOLBAR::~ACTION_TOOLBAR()
{
    Disconnect( wxEVT_COMMAND_TOOL_CLICKED, wxAuiToolBarEventHandler( ACTION_TOOLBAR::onToolEvent ), nullptr, this );
    Disconnect( wxEVT_AUITOOLBAR_RIGHT_CLICK, wxAuiToolBarEventHandler( ACTION_TOOLBAR::onRightClick ), nullptr, this );
    Disconnect( wxEVT_RIGHT_UP, wxMouseEventHandler( ACTION_TOOLBAR::onRightUp ), nullptr, this );
    Disconnect( wxEVT_AUITOOLBAR_BEGIN_DRAG, wxAuiToolBarEventHandler( ACTION_TOOLBAR::onItemDrag ), nullptr, this );
    Disconnect( wxEVT_LEFT_DOWN, wxMouseEventHandler( ACTION_TOOLBAR::onMouseClick ), nullptr, this );
    Disconnect( wxEVT_LEFT_UP, wxMouseEventHandler( ACTION_TOOLBAR::onMouseClick ), nullptr, this );
    Disconnect( m_paletteTimer->GetId(), wxEVT_TIMER, wxTimerEventHandler( ACTION_TOOLBAR::onTimerDone ), nullptr,
                this );
 
    Unbind( wxEVT_SYS_COLOUR_CHANGED, wxSysColourChangedEventHandler( ACTION_TOOLBAR::onThemeChanged ), this );
 
    delete m_paletteTimer;
 
    // Clear all the maps keeping track of our items on the toolbar
    m_toolMenus.clear();
    m_actionGroups.clear();
    m_toolCancellable.clear();
    m_toolKinds.clear();
    m_toolActions.clear();
}
 
 
std::list<ACTION_TOOLBAR_CONTROL*> ACTION_TOOLBAR::GetCustomControlList( FRAME_T aContext )
{
    std::list<ACTION_TOOLBAR_CONTROL*> controls;
 
    for( ACTION_TOOLBAR_CONTROL* control : GetAllCustomControls() )
    {
        if( control->SupportedFor( aContext ) )
            controls.push_back( control );
    }
 
    return controls;
}
 
 
void ACTION_TOOLBAR::ApplyConfiguration( const TOOLBAR_CONFIGURATION& aConfig )
{
    wxASSERT( GetParent() );
 
    // Keep each group's selection across the rebuild.
    std::map<std::string, std::string> currentGroupItems = m_groupSelections;
 
    // Remove existing tools
    ClearToolbar();
 
    std::vector<TOOLBAR_ITEM> items = aConfig.GetToolbarItems();
 
    // Add all the items to the toolbar
    for( auto& item : items )
    {
        switch( item.m_Type )
        {
        case TOOLBAR_ITEM_TYPE::SEPARATOR:
            AddScaledSeparator( GetParent() );
            break;
 
        case TOOLBAR_ITEM_TYPE::SPACER:
            AddSpacer( item.m_Size );
            break;
 
        case TOOLBAR_ITEM_TYPE::TB_GROUP:
        {
            // Add a group of items to the toolbar
            std::string                     groupName = item.m_GroupName.ToStdString();
            std::vector<const TOOL_ACTION*> tools;
            const TOOL_ACTION*              defaultTool = nullptr;
 
            for( TOOLBAR_ITEM& groupItem : item.m_GroupItems )
            {
                switch( groupItem.m_Type )
                {
                case TOOLBAR_ITEM_TYPE::SEPARATOR:
                case TOOLBAR_ITEM_TYPE::SPACER:
                case TOOLBAR_ITEM_TYPE::TB_GROUP:
                case TOOLBAR_ITEM_TYPE::CONTROL:
                    wxFAIL_MSG( wxT( "Unsupported group item type" ) );
                    continue;
 
                case TOOLBAR_ITEM_TYPE::TOOL:
                    TOOL_ACTION* grpAction = m_toolManager->GetActionManager()->FindAction( groupItem.m_ActionName );
 
                    if( !grpAction )
                    {
                        wxFAIL_MSG( wxString::Format( wxT( "Unable to find group tool %s" ), groupItem.m_ActionName ) );
                        continue;
                    }
 
                    tools.push_back( grpAction );
 
                    if( currentGroupItems[groupName] == groupItem.m_ActionName )
                        defaultTool = grpAction;
                }
            }
 
            // A group needs at least one action
            if( tools.empty() )
                continue;
 
            std::unique_ptr<ACTION_GROUP> group = std::make_unique<ACTION_GROUP>( groupName, tools );
 
            if( defaultTool )
                group->SetDefaultAction( *defaultTool );
 
            AddGroup( std::move( group ) );
 
            // Look up and attach context menu if one is registered for this group
            auto menuFactory = TOOLBAR_CONTEXT_MENU_REGISTRY::GetGroupMenuFactory( groupName );
 
            if( menuFactory && m_toolManager )
            {
                // Register the menu for each action in the group
                for( const TOOL_ACTION* grpAction : tools )
                    AddToolContextMenu( *grpAction, menuFactory( m_toolManager ) );
            }
 
            break;
        }
 
        case TOOLBAR_ITEM_TYPE::CONTROL:
        {
            // Add a custom control to the toolbar
            EDA_BASE_FRAME* frame = static_cast<EDA_BASE_FRAME*>( GetParent() );
            ACTION_TOOLBAR_CONTROL_FACTORY* factory = frame->GetCustomToolbarControlFactory( item.m_ControlName );
 
            if( !factory )
            {
                wxFAIL_MSG( wxString::Format( wxT( "Unable to find control factory for %s" ), item.m_ControlName ) );
                continue;
            }
 
            // The factory functions are responsible for adding the controls to the toolbar themselves
            (*factory)( this );
            break;
        }
 
        case TOOLBAR_ITEM_TYPE::TOOL:
        {
            TOOL_ACTION* action = m_toolManager->GetActionManager()->FindAction( item.m_ActionName );
 
            if( !action )
            {
                wxFAIL_MSG( wxString::Format( wxT( "Unable to find toolbar tool %s" ), item.m_ActionName ) );
                continue;
            }
 
            Add( *action );
 
            // Look up and attach context menu if one is registered for this action
            auto factory = TOOLBAR_CONTEXT_MENU_REGISTRY::GetMenuFactory( item.m_ActionName );
 
            if( factory && m_toolManager )
                AddToolContextMenu( *action, factory( m_toolManager ) );
 
            break;
        }
        }
    }
 
    // Apply the configuration
    KiRealize();
}
 
 
void ACTION_TOOLBAR::DoSetToolTipText( const wxString& aTip )
{
    // We use \t in short description to align accelerators in wxMenuItem
    // But they should be converted when displaying tooltips
    wxString tip = aTip;
    tip.Replace( "\t", "  " );
 
    wxAuiToolBar::DoSetToolTipText( tip );
}
 
 
void ACTION_TOOLBAR::Add( const TOOL_ACTION& aAction )
{
    wxASSERT_MSG( !aAction.CheckToolbarState( TOOLBAR_STATE::HIDDEN ),
                  wxString::Format( "Attempting to add hidden action %s to the toolbar", aAction.GetName() ) );
 
    bool isToggleEntry = aAction.CheckToolbarState( TOOLBAR_STATE::TOGGLE );
    bool isCancellable = aAction.CheckToolbarState( TOOLBAR_STATE::CANCEL );
 
    Add( aAction, isToggleEntry, isCancellable );
}
 
 
void ACTION_TOOLBAR::Add( const TOOL_ACTION& aAction, bool aIsToggleEntry, bool aIsCancellable )
{
    wxASSERT( GetParent() );
    wxASSERT_MSG( !( aIsCancellable && !aIsToggleEntry ),
                  wxS( "aIsCancellable requires aIsToggleEntry" ) );
 
    int toolId = aAction.GetUIId();
    int iconSize = Pgm().GetCommonSettings()->m_Appearance.toolbar_icon_size;
 
    AddTool( toolId, wxEmptyString,
             KiBitmapBundleDef( aAction.GetIcon(), iconSize ),
             KiDisabledBitmapBundleDef( aAction.GetIcon(), iconSize ),
             aIsToggleEntry ? wxITEM_CHECK : wxITEM_NORMAL,
             aAction.GetButtonTooltip(), wxEmptyString, nullptr );
 
    m_toolKinds[ toolId ]       = aIsToggleEntry;
    m_toolActions[ toolId ]     = &aAction;
    m_toolCancellable[ toolId ] = aIsCancellable;
}
 
 
void ACTION_TOOLBAR::AddButton( const TOOL_ACTION& aAction )
{
    int toolId = aAction.GetUIId();
    int iconSize = Pgm().GetCommonSettings()->m_Appearance.toolbar_icon_size;
 
    AddTool( toolId, wxEmptyString,
             KiBitmapBundleDef( aAction.GetIcon(), iconSize ),
             KiDisabledBitmapBundleDef( aAction.GetIcon(), iconSize ),
             wxITEM_NORMAL, aAction.GetButtonTooltip(), wxEmptyString, nullptr );
 
    m_toolKinds[ toolId ] = false;
    m_toolActions[ toolId ] = &aAction;
}
 
 
void ACTION_TOOLBAR::AddScaledSeparator( wxWindow* aWindow )
{
    int scale = KiIconScale( aWindow );
 
    if( scale > 4 )
        AddSpacer( 16 * ( scale - 4 ) / 4 );
 
    AddSeparator();
 
    if( scale > 4 )
        AddSpacer( 16 * ( scale - 4 ) / 4 );
}
 
 
void ACTION_TOOLBAR::Add( wxControl* aControl, const wxString& aLabel )
{
    wxASSERT( aControl );
    m_controlIDs.push_back( aControl->GetId() );
    AddControl( aControl, aLabel );
}
 
 
void ACTION_TOOLBAR::AddToolContextMenu( const TOOL_ACTION& aAction, std::unique_ptr<ACTION_MENU> aMenu )
{
    int toolId = aAction.GetUIId();
 
    m_toolMenus[toolId] = std::move( aMenu );
}
 
 
void ACTION_TOOLBAR::AddGroup( std::unique_ptr<ACTION_GROUP> aGroup )
{
    int                groupId = aGroup->GetUIId();
    const TOOL_ACTION* defaultAction = aGroup->GetDefaultAction();
    int                iconSize = Pgm().GetCommonSettings()->m_Appearance.toolbar_icon_size;
 
    wxASSERT( GetParent() );
    wxASSERT( defaultAction );
 
    // Turn this into a toggle entry if any one of the actions is a toggle entry
    bool isToggleEntry = false;
 
    for( const auto& act : aGroup->GetActions() )
        isToggleEntry |= act->CheckToolbarState( TOOLBAR_STATE::TOGGLE );
 
    m_toolKinds[ groupId ]    = isToggleEntry;
    m_toolActions[ groupId ]  = defaultAction;
    m_actionGroups[ groupId ] = std::move( aGroup );
 
    // Add the main toolbar item representing the group
    AddTool( groupId, wxEmptyString,
             KiBitmapBundleDef( defaultAction->GetIcon(), iconSize ),
             KiDisabledBitmapBundleDef( defaultAction->GetIcon(), iconSize ),
             isToggleEntry ? wxITEM_CHECK : wxITEM_NORMAL, wxEmptyString, wxEmptyString, nullptr );
 
    // Select the default action
    doSelectAction( m_actionGroups[ groupId ].get(), *defaultAction );
}
 
 
void ACTION_TOOLBAR::SelectAction( ACTION_GROUP* aGroup, const TOOL_ACTION& aAction )
{
    bool valid = std::any_of( aGroup->m_actions.begin(), aGroup->m_actions.end(),
                              [&]( const TOOL_ACTION* action2 ) -> bool
                              {
                                  // For some reason, we can't compare the actions directly
                                  return aAction.GetId() == action2->GetId();
                              } );
 
    if( valid )
        doSelectAction( aGroup, aAction );
}
 
 
void ACTION_TOOLBAR::SelectAction( const TOOL_ACTION& aAction )
{
    // Find the group that contains this action and select it
    for( auto& [id, groupPtr] : m_actionGroups )
    {
        ACTION_GROUP* group = groupPtr.get();
 
        bool inGroup = std::any_of( group->m_actions.begin(), group->m_actions.end(),
                                     [&]( const TOOL_ACTION* action2 )
                                     {
                                         return aAction.GetId() == action2->GetId();
                                     } );
 
        if( inGroup )
        {
            doSelectAction( group, aAction );
            break;
        }
    }
}
 
 
void ACTION_TOOLBAR::doSelectAction( ACTION_GROUP* aGroup, const TOOL_ACTION& aAction )
{
    wxASSERT( GetParent() );
 
    int groupId = aGroup->GetUIId();
 
    wxAuiToolBarItem* item   = FindTool( groupId );
 
    if( !item )
        return;
 
    int iconSize = Pgm().GetCommonSettings()->m_Appearance.toolbar_icon_size;
 
    // Update the item information
    item->SetShortHelp( aAction.GetButtonTooltip() );
    item->SetBitmap( KiBitmapBundleDef( aAction.GetIcon(), iconSize ) );
    item->SetDisabledBitmap( KiDisabledBitmapBundleDef( aAction.GetIcon(), iconSize ) );
 
    // Register a new handler with the new UI conditions
    if( m_toolManager )
    {
        m_toolManager->GetToolHolder()->UnregisterUIUpdateHandler( groupId );
 
        const ACTION_CONDITIONS* cond = m_toolManager->GetActionManager()->GetCondition( aAction );
 
        // Register the new UI condition to control this entry
        if( cond )
        {
            m_toolManager->GetToolHolder()->RegisterUIUpdateHandler( groupId, *cond );
        }
        else
        {
            wxLogTrace( kicadTraceToolStack, wxString::Format( "No UI condition for action %s",
                                                               aAction.GetName() ) );
        }
    }
 
    // Update the currently selected action
    m_toolActions[ groupId ] = &aAction;
    m_groupSelections[aGroup->GetName()] = aAction.GetName();
 
    Refresh();
}
 
 
void ACTION_TOOLBAR::UpdateControlWidths()
{
    for( int id : m_controlIDs )
        UpdateControlWidth( id );
}
 
 
void ACTION_TOOLBAR::UpdateControlWidth( int aID )
{
    wxAuiToolBarItem* item = FindTool( aID );
    wxASSERT_MSG( item, wxString::Format( "No toolbar item found for ID %d", aID ) );
 
    // The control on the toolbar is stored inside the window field of the item
    wxControl* control = dynamic_cast<wxControl*>( item->GetWindow() );
    wxASSERT_MSG( control, wxString::Format( "No control located in toolbar item with ID %d", aID ) );
 
    // Update the size the item has stored using the best size of the control
    control->InvalidateBestSize();
    wxSize bestSize = control->GetBestSize();
    item->SetMinSize( bestSize );
 
    // Update the sizer item sizes
    // This is a bit convoluted because there are actually 2 sizers that need to be updated:
    // 1. The main sizer that is used for the entire toolbar (this sizer item can be found in the
    // toolbar item)
    if( wxSizerItem* szrItem = item->GetSizerItem() )
        szrItem->SetMinSize( bestSize );
 
    // 2. The controls have a second sizer that allows for padding above/below the control with
    // stretch space, so we also need to update the sizer item for the control in that sizer with
    // the new size.  We let wx do the search for us, since SetItemMinSize is recursive and will
    // locate the control on that sizer.
    if( m_sizer )
    {
        m_sizer->SetItemMinSize( control, bestSize );
 
        // Now actually update the toolbar with the new sizes
        m_sizer->Layout();
    }
}
 
 
void ACTION_TOOLBAR::ClearToolbar()
{
    // A group made again gets a new id, so the frame would keep this one's handler for good
    if( m_toolManager )
    {
        for( const auto& [groupId, group] : m_actionGroups )
            m_toolManager->GetToolHolder()->UnregisterUIUpdateHandler( groupId );
    }
 
    // Clear all the maps keeping track of our items on the toolbar
    m_toolMenus.clear();
    m_actionGroups.clear();
    m_toolCancellable.clear();
    m_toolKinds.clear();
    m_toolActions.clear();
 
    for( int id : m_controlIDs )
    {
        m_parent->ClearToolbarControl( id );
        DestroyTool( id );
    }
 
    m_controlIDs.clear();
 
    // Remove the actual tools from the toolbar
    Clear();
}
 
 
void ACTION_TOOLBAR::SetToolBitmap( const TOOL_ACTION& aAction, const wxBitmapBundle& aBitmap )
{
    int toolId = aAction.GetUIId();
 
    // Set the disabled bitmap: we use the disabled bitmap version of aBitmap.
    wxAuiToolBarItem* tb_item = wxAuiToolBar::FindTool( toolId );
 
    if( !tb_item )
        return;
 
    wxBitmap bm = aBitmap.GetBitmapFor( this );
 
    tb_item->SetBitmap( aBitmap );
    tb_item->SetDisabledBitmap( bm.ConvertToDisabled( KIPLATFORM::UI::IsDarkTheme() ? 70 : 255 ) );
}
 
 
void ACTION_TOOLBAR::Toggle( const TOOL_ACTION& aAction, bool aState )
{
    int toolId = aAction.GetUIId();
 
    if( m_toolKinds[ toolId ] )
        ToggleTool( toolId, aState );
    else
        EnableTool( toolId, aState );
}
 
 
void ACTION_TOOLBAR::Toggle( const TOOL_ACTION& aAction, bool aEnabled, bool aChecked )
{
    int toolId = aAction.GetUIId();
 
    EnableTool( toolId, aEnabled );
    ToggleTool( toolId, aEnabled && aChecked );
}
 
 
void ACTION_TOOLBAR::onToolEvent( wxAuiToolBarEvent& aEvent )
{
    int            id   = aEvent.GetId();
    wxEventType    type = aEvent.GetEventType();
    OPT_TOOL_EVENT evt;
 
    bool handled = false;
 
    if( m_toolManager && type == wxEVT_COMMAND_TOOL_CLICKED )
    {
        const auto actionIt = m_toolActions.find( id );
        const auto cancelIt = m_toolCancellable.find( id );
        const auto groupIt  = m_actionGroups.find( id );
 
        // Determine if the tool is actually cancellable
        bool isCancellable = ( cancelIt != m_toolCancellable.end() ) ? cancelIt->second : false;
 
        // The toolbar item is toggled before the event is sent, so we check for it not being
        // toggled to see if it was toggled originally
        if( isCancellable && !GetToolToggled( id ) )
        {
            // Send a cancel event
            m_toolManager->CancelTool();
            handled = true;
        }
        else if( groupIt != m_actionGroups.end()
                 && std::none_of( groupIt->second->GetActions().begin(),
                                  groupIt->second->GetActions().end(),
                                  []( const TOOL_ACTION* a )
                                  {
                                      return a->IsActivation();
                                  } ) )
        {
            // For non-tool toggle groups (units, crosshair, line modes), cycle to the next
            // action on click. Tool groups (route track, etc.) fall through and just dispatch
            // the currently displayed action.
            ACTION_GROUP*                          group   = groupIt->second.get();
            const std::vector<const TOOL_ACTION*>& actions = group->GetActions();
            const TOOL_ACTION*                     current = actionIt->second;
 
            const TOOL_ACTION* next = actions[0];
 
            for( size_t i = 0; i < actions.size(); ++i )
            {
                if( actions[i]->GetId() == current->GetId() )
                {
                    next = actions[( i + 1 ) % actions.size()];
                    break;
                }
            }
 
            evt = next->MakeEvent();
            evt->SetHasPosition( false );
            m_toolManager->ProcessEvent( *evt );
            m_toolManager->GetToolHolder()->RefreshCanvas();
 
            doSelectAction( group, *next );
            handled = true;
        }
        else if( actionIt != m_toolActions.end() )
        {
            // Dispatch a tool event
            evt = actionIt->second->MakeEvent();
            evt->SetHasPosition( false );
            m_toolManager->ProcessEvent( *evt );
            m_toolManager->GetToolHolder()->RefreshCanvas();
            handled = true;
        }
    }
 
    // Skip the event if we don't handle it
    if( !handled )
        aEvent.Skip();
}
 
 
void ACTION_TOOLBAR::onRightClick( wxAuiToolBarEvent& aEvent )
{
    int toolId = aEvent.GetToolId();
 
    // This means the event was not on a button
    if( toolId == -1 )
        return;
 
    showContextMenu( toolId );
}
 
 
void ACTION_TOOLBAR::onRightUp( wxMouseEvent& aEvent )
{
    // wxAuiToolBar::OnRightDown() uses horizontal-only geometry to reserve its overflow
    // dead-zone, which on a vertical toolbar kills right-clicks over part of every button.
    // Hit-test the tool ourselves so the whole button works.
    wxAuiToolBarItem* item = FindToolByPosition( aEvent.GetX(), aEvent.GetY() );
 
    if( !item )
    {
        aEvent.Skip();
        return;
    }
 
    // Don't Skip(): suppress wx's own OnRightUp() so the menu is shown exactly once.
    showContextMenu( item->GetId() );
}
 
 
void ACTION_TOOLBAR::showContextMenu( int aToolId )
{
    // Ensure that the ID maps to a proper tool ID. If right-clicked on a group item, this is needed
    // to get the ID of the currently selected action, since the event's ID is that of the group.
    const auto actionIt = m_toolActions.find( aToolId );
 
    if( actionIt != m_toolActions.end() )
        aToolId = actionIt->second->GetUIId();
 
    // Find the menu for the action
    const auto menuIt = m_toolMenus.find( aToolId );
 
    if( menuIt == m_toolMenus.end() )
        return;
 
    // Update and show the menu
    std::unique_ptr<ACTION_MENU>& owningMenu = menuIt->second;
 
    // Get the actual menu pointer to show it
    ACTION_MENU* menu = owningMenu.get();
    SELECTION    dummySel;
 
    if( CONDITIONAL_MENU* condMenu = dynamic_cast<CONDITIONAL_MENU*>( menu ) )
        condMenu->Evaluate( dummySel );
 
    menu->UpdateAll();
    PopupMenu( menu );
 
    // Remove hovered item when the menu closes, otherwise it remains hovered even if the
    // mouse is not on the toolbar
    SetHoverItem( nullptr );
}
 
 
// The time (in milliseconds) between pressing the left mouse button and opening the palette
#define PALETTE_OPEN_DELAY 500
 
 
void ACTION_TOOLBAR::onMouseClick( wxMouseEvent& aEvent )
{
    wxAuiToolBarItem* item = FindToolByPosition( aEvent.GetX(), aEvent.GetY() );
 
    if( item )
    {
        // Ensure there is no active palette
        if( m_palette )
        {
            m_palette->Hide();
            m_palette->Destroy();
            m_palette = nullptr;
        }
 
        // Start the popup conditions if it is a left mouse click and the tool clicked is a group
        if( aEvent.LeftDown() && ( m_actionGroups.find( item->GetId() ) != m_actionGroups.end() ) )
            m_paletteTimer->StartOnce( PALETTE_OPEN_DELAY );
 
        // Clear the popup conditions if it is a left up, because that implies a click happened
        if( aEvent.LeftUp() )
            m_paletteTimer->Stop();
    }
 
    // Skip the event so wx can continue processing the mouse event
    aEvent.Skip();
}
 
 
void ACTION_TOOLBAR::onItemDrag( wxAuiToolBarEvent& aEvent )
{
    int toolId = aEvent.GetToolId();
 
    if( m_actionGroups.find( toolId ) != m_actionGroups.end() )
    {
        wxAuiToolBarItem* item = FindTool( toolId );
 
        // Use call after because opening the palette from a mouse handler
        // creates a weird mouse state that causes problems on OSX.
        CallAfter( &ACTION_TOOLBAR::popupPalette, item );
 
        // Don't skip this event since we are handling it
        return;
    }
 
    // Skip since we don't care about it
    aEvent.Skip();
}
 
 
void ACTION_TOOLBAR::onTimerDone( wxTimerEvent& aEvent )
{
    // We need to search for the tool using the client coordinates
    wxPoint mousePos = ScreenToClient( KIPLATFORM::UI::GetMousePosition() );
 
    wxAuiToolBarItem* item = FindToolByPosition( mousePos.x, mousePos.y );
 
    if( item )
        popupPalette( item );
}
 
 
void ACTION_TOOLBAR::onPaletteEvent( wxCommandEvent& aEvent )
{
    if( !m_palette )
        return;
 
    // Clear m_palette up front so a re-entrant dispatch (modal dialog pumping events)
    // hits the null guard above instead of double-destroying.
    ACTION_TOOLBAR_PALETTE* palette = m_palette;
    m_palette = nullptr;
 
    OPT_TOOL_EVENT evt;
    ACTION_GROUP*  group = palette->GetGroup();
 
    // Find the action corresponding to the button press
    auto actionIt = std::find_if( group->GetActions().begin(), group->GetActions().end(),
                                  [&]( const TOOL_ACTION* aAction )
                                  {
                                      return aAction->GetUIId() == aEvent.GetId();
                                  } );
 
    if( actionIt != group->GetActions().end() )
    {
        const TOOL_ACTION* action = *actionIt;
 
        // Dispatch a tool event
        evt = action->MakeEvent();
        evt->SetHasPosition( false );
        m_toolManager->ProcessEvent( *evt );
        m_toolManager->GetToolHolder()->RefreshCanvas();
 
        // Update the main toolbar item with the selected action
        doSelectAction( group, *action );
    }
 
    // Hide the palette
    palette->Hide();
    palette->Destroy();
}
 
 
void ACTION_TOOLBAR::popupPalette( wxAuiToolBarItem* aItem )
{
    // Clear all popup conditions
    m_paletteTimer->Stop();
 
    wxWindow* toolParent = dynamic_cast<wxWindow*>( m_toolManager->GetToolHolder() );
 
    wxCHECK( GetParent() && m_auiManager && toolParent, /* void */ );
 
    // Ensure the item we are using for the palette has a group associated with it.
    const auto it = m_actionGroups.find( aItem->GetId() );
 
    if( it == m_actionGroups.end() )
        return;
 
    ACTION_GROUP* group = it->second.get();
 
    wxAuiPaneInfo& pane = m_auiManager->GetPane( this );
 
    // We use the size of the toolbar items for our palette buttons
    wxRect toolRect = GetToolRect( aItem->GetId() );
 
    // The position for the palette window must be in screen coordinates
    wxPoint pos( ClientToScreen( toolRect.GetPosition() ) );
 
    // True for vertical buttons, false for horizontal
    bool dir = pane.dock_direction != wxAUI_DOCK_LEFT && pane.dock_direction != wxAUI_DOCK_RIGHT;
 
    m_palette = new ACTION_TOOLBAR_PALETTE( GetParent(), dir );
 
    m_palette->SetGroup( group );
    m_palette->SetButtonSize( toolRect );
    m_palette->Connect( wxEVT_BUTTON, wxCommandEventHandler( ACTION_TOOLBAR::onPaletteEvent ), nullptr, this );
 
 
    for( const TOOL_ACTION* action : group->m_actions )
    {
        wxUpdateUIEvent evt( action->GetUIId() );
 
        toolParent->ProcessWindowEvent( evt );
 
        m_palette->AddAction( *action );
 
        if( evt.GetSetEnabled() )
            m_palette->EnableAction( *action, evt.GetEnabled() );
    }
 
    wxSize paletteSize = m_palette->GetFittedSize();
 
    // Determine the position of the top left corner of the palette window
    switch( pane.dock_direction )
    {
        case wxAUI_DOCK_TOP:
            // Top toolbars need to shift the palette window down by the toolbar padding
            pos = ClientToScreen( toolRect.GetBottomLeft() );
            pos += wxPoint( -PALETTE_BORDER,            // Shift left to align the button edges
                            m_bottomPadding );          // Shift down to move away from the toolbar
            break;
 
        case wxAUI_DOCK_BOTTOM:
            // Bottom toolbars need to shift the palette window up by its height (all buttons +
            // border + toolbar padding)
            pos = ClientToScreen( toolRect.GetTopLeft() );
            pos += wxPoint( -PALETTE_BORDER, // Shift left to align the button
                            // Shift up by the entire length of the palette.
                            -( paletteSize.GetHeight() + m_topPadding ) );
            break;
 
        case wxAUI_DOCK_LEFT:
            // Left toolbars open the palette to the right, with a small extra nudge so it does not
            // paint over the toolbar icon that opened it.
            pos = ClientToScreen( toolRect.GetTopRight() );
            pos += wxPoint( m_rightPadding + PALETTE_BORDER, 0 );
            break;
 
        case wxAUI_DOCK_RIGHT:
            // Right toolbars need to shift the palette window left by its width (all buttons +
            // border + toolbar padding)
            pos = ClientToScreen( toolRect.GetTopLeft() );
 
            // Shift left by the palette length, with a small extra nudge so it does not paint over
            // the toolbar icon that opened it.
            pos += wxPoint( -( paletteSize.GetWidth() + m_leftPadding + PALETTE_BORDER ), 0 );
            break;
    }
 
    // Release the mouse to ensure the first click will be recognized in the palette
    if( HasCapture() )
        ReleaseMouse();
 
    m_palette->SetPosition( pos );
    m_palette->Popup();
 
    // Clear the mouse state on the toolbar because otherwise wxWidgets gets confused
    // and won't properly display any highlighted items after the palette is closed.
    // (This is the equivalent of calling the DoResetMouseState() private function)
    RefreshOverflowState();
    SetHoverItem( nullptr );
    SetPressedItem( nullptr );
 
    m_dragging   = false;
    m_tipItem    = nullptr;
    m_actionPos  = wxPoint( -1, -1 );
    m_actionItem = nullptr;
}
 
 
void ACTION_TOOLBAR::OnCustomRender(wxDC& aDc, const wxAuiToolBarItem& aItem, const wxRect& aRect )
{
    auto it = m_actionGroups.find( aItem.GetId() );
 
    if( it == m_actionGroups.end() )
        return;
 
    // Choose the color to draw the triangle
    wxColour clr;
 
    if( aItem.GetState() & wxAUI_BUTTON_STATE_DISABLED )
        clr = wxSystemSettings::GetColour( wxSYS_COLOUR_GRAYTEXT );
    else
        clr = wxSystemSettings::GetColour( wxSYS_COLOUR_BTNTEXT );
 
    // Must set both the pen (for the outline) and the brush (for the polygon fill)
    aDc.SetPen( wxPen( clr ) );
    aDc.SetBrush( wxBrush( clr ) );
 
    // Make the side length of the triangle approximately 1/5th of the bitmap
    int sideLength = KiROUND( aRect.height / 5.0 );
 
    // This will create a triangle with its point at the bottom right corner,
    // and its other two corners along the right and bottom sides
    wxPoint btmRight = aRect.GetBottomRight();
    wxPoint topCorner( btmRight.x,              btmRight.y - sideLength );
    wxPoint btmCorner( btmRight.x - sideLength, btmRight.y  );
 
    wxPointList points;
    points.Append( &btmRight );
    points.Append( &topCorner );
    points.Append( &btmCorner );
 
    aDc.DrawPolygon( &points );
}
 
 
bool ACTION_TOOLBAR::KiRealize()
{
#if wxCHECK_VERSION( 3, 3, 0 )
    return Realize();
#else
    wxClientDC dc( this );
 
    if( !dc.IsOk() )
        return false;
 
    // calculate hint sizes for both horizontal and vertical
    // in the order that leaves toolbar in correct final state
 
    // however, skip calculating alternate orientations if we don't need them due to window style
    bool retval = true;
 
    if( m_orientation == wxHORIZONTAL )
    {
        if( !( GetWindowStyle() & wxAUI_TB_HORIZONTAL ) )
        {
            m_vertHintSize = GetSize();
            retval         = RealizeHelper( dc, false );
        }
 
        if( retval && RealizeHelper( dc, true ) )
            m_horzHintSize = GetSize();
        else
            retval = false;
    }
    else
    {
        if( !( GetWindowStyle() & wxAUI_TB_VERTICAL ) )
        {
            m_horzHintSize = GetSize();
            retval         = RealizeHelper( dc, true );
        }
 
        if( retval && RealizeHelper( dc, false ) )
            m_vertHintSize = GetSize();
        else
            retval = false;
    }
 
    Refresh( false );
    return retval;
#endif
}
 
 
void ACTION_TOOLBAR::onThemeChanged( wxSysColourChangedEvent &aEvent )
{
    GetBitmapStore()->ThemeChanged();
    RefreshBitmaps();
 
    aEvent.Skip();
}
 
 
void ACTION_TOOLBAR::RefreshBitmaps()
{
    int iconSize = Pgm().GetCommonSettings()->m_Appearance.toolbar_icon_size;
 
    for( const std::pair<int, const TOOL_ACTION*> pair : m_toolActions )
    {
        wxAuiToolBarItem* tool = FindTool( pair.first );
 
        tool->SetBitmap( KiBitmapBundleDef( pair.second->GetIcon(), iconSize ) );
        tool->SetDisabledBitmap( KiDisabledBitmapBundleDef( pair.second->GetIcon(), iconSize ) );
    }
 
    Refresh();
}
 
/*
 * Common controls for the toolbar
 */
ACTION_TOOLBAR_CONTROL ACTION_TOOLBAR_CONTROLS::gridSelect( "control.GridSelector",
                                                            _( "Grid selector" ),
                                                            _( "Grid Selection box" ),
                                                            { FRAME_SCH,
                                                              FRAME_SCH_SYMBOL_EDITOR,
                                                              FRAME_SCH_VIEWER,
                                                              FRAME_PCB_EDITOR,
                                                              FRAME_FOOTPRINT_EDITOR,
                                                              FRAME_FOOTPRINT_VIEWER,
                                                              FRAME_GERBER,
                                                              FRAME_PL_EDITOR } );
 
 
ACTION_TOOLBAR_CONTROL ACTION_TOOLBAR_CONTROLS::zoomSelect( "control.ZoomSelector",
                                                            _( "Zoom selector" ),
                                                            _( "Zoom Selection box" ),
                                                            { FRAME_SCH,
                                                              FRAME_SCH_SYMBOL_EDITOR,
                                                              FRAME_SCH_VIEWER,
                                                              FRAME_PCB_EDITOR,
                                                              FRAME_PCB_DISPLAY3D,
                                                              FRAME_FOOTPRINT_EDITOR,
                                                              FRAME_FOOTPRINT_VIEWER,
                                                              FRAME_GERBER,
                                                              FRAME_PL_EDITOR } );
 
ACTION_TOOLBAR_CONTROL ACTION_TOOLBAR_CONTROLS::ipcScripting( "control.IPCPlugin",
                                                              _( "IPC/Scripting plugins" ),
                                                              _( "Region to hold the IPC/Scripting action buttons" ),
                                                              { FRAME_SCH,
                                                                FRAME_SCH_SYMBOL_EDITOR,
                                                                FRAME_PCB_EDITOR,
                                                                FRAME_FOOTPRINT_EDITOR } );
 
ACTION_TOOLBAR_CONTROL ACTION_TOOLBAR_CONTROLS::layerSelector( "control.LayerSelector",
                                                               _( "Layer selector" ),
                                                               _( "Control to select the layer" ),
                                                               { FRAME_PCB_EDITOR,
                                                                 FRAME_FOOTPRINT_EDITOR,
                                                                 FRAME_FOOTPRINT_VIEWER,
                                                                 FRAME_GERBER } );
 
ACTION_TOOLBAR_CONTROL ACTION_TOOLBAR_CONTROLS::unitSelector( "control.UnitSelector",
                                                              _( "Symbol unit selector" ),
                                                              _( "Displays the current unit" ),
                                                              { FRAME_SCH_SYMBOL_EDITOR,
                                                                FRAME_SCH_VIEWER } );
 
ACTION_TOOLBAR_CONTROL ACTION_TOOLBAR_CONTROLS::bodyStyleSelector( "control.BodyStyleSelector",
                                                                   _( "Symbol body style selector" ),
                                                                   _( "Displays the current body style" ),
                                                                   { FRAME_SCH_SYMBOL_EDITOR,
                                                                     FRAME_SCH_VIEWER } );
 
ACTION_TOOLBAR_CONTROL ACTION_TOOLBAR_CONTROLS::overrideLocks( "control.OverrideLocks",
                                                               _( "Override locks" ),
                                                               _( "Allow moving of locked items with the mouse" ),
                                                               { FRAME_PCB_EDITOR, FRAME_SCH } );