global_edit_tool.cpp

Go to the documentation of this file.

/*
 * This program source code file is part of KiCad, a free EDA CAD application.
 *
 * Copyright The KiCad Developers, see AUTHORS.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 <zone_manager/dialog_zone_manager.h>
#include <footprint.h>
#include <pcb_track.h>
#include <pcb_generator.h>
#include <tool/tool_manager.h>
#include <tools/pcb_actions.h>
#include <tools/edit_tool.h>
#include <dialogs/dialog_exchange_footprints.h>
#include <dialogs/dialog_cleanup_tracks_and_vias.h>
#include <dialogs/dialog_global_edit_tracks_and_vias.h>
#include <dialogs/dialog_swap_layers.h>
#include <dialogs/dialog_unused_pad_layers.h>
#include <tools/global_edit_tool.h>
#include <dialogs/dialog_cleanup_graphics.h>
#include <dialogs/dialog_migrate_3d_models.h>
#include <board_design_settings.h>
#include <pcb_edit_frame.h>
 
 
GLOBAL_EDIT_TOOL::GLOBAL_EDIT_TOOL() :
        PCB_TOOL_BASE( "pcbnew.GlobalEdit" ),
        m_selectionTool( nullptr )
{
}
 
 
void GLOBAL_EDIT_TOOL::Reset( RESET_REASON aReason )
{
    if( aReason != RUN )
        m_commit = std::make_unique<BOARD_COMMIT>( this );
}
 
 
bool GLOBAL_EDIT_TOOL::Init()
{
    // Find the selection tool, so they can cooperate
    m_selectionTool = m_toolMgr->GetTool<PCB_SELECTION_TOOL>();
 
    return true;
}
 
 
int GLOBAL_EDIT_TOOL::ExchangeFootprints( const TOOL_EVENT& aEvent )
{
    PCB_EDIT_FRAME* editFrame = getEditFrame<PCB_EDIT_FRAME>();
    PCB_SELECTION&  selection = m_selectionTool->GetSelection();
    FOOTPRINT*      footprint = nullptr;
    bool            updateMode = false;
    bool            currentMode = false;
 
    if( aEvent.HasPosition() )
        selection = m_selectionTool->RequestSelection( EDIT_TOOL::FootprintFilter );
 
    if( !selection.Empty() )
        footprint = selection.FirstOfKind<FOOTPRINT>();
 
    if( aEvent.IsAction( &PCB_ACTIONS::updateFootprint ) )
    {
        updateMode = true;
        currentMode = true;
    }
    else if( aEvent.IsAction( &PCB_ACTIONS::updateFootprints ) )
    {
        updateMode = true;
        currentMode = selection.CountType( PCB_FOOTPRINT_T ) > 0;
    }
    else if( aEvent.IsAction( &PCB_ACTIONS::changeFootprint ) )
    {
        updateMode = false;
        currentMode = true;
    }
    else if( aEvent.IsAction( &PCB_ACTIONS::changeFootprints ) )
    {
        updateMode = false;
        currentMode = selection.CountType( PCB_FOOTPRINT_T ) > 0;
    }
    else
    {
        wxFAIL_MSG( wxT( "ExchangeFootprints: unexpected action" ) );
    }
 
    DIALOG_EXCHANGE_FOOTPRINTS dialog( editFrame, footprint, updateMode, currentMode );
    dialog.ShowQuasiModal();
 
    return 0;
}
 
 
bool GLOBAL_EDIT_TOOL::swapBoardItem( BOARD_ITEM* aItem, std::map<PCB_LAYER_ID, PCB_LAYER_ID>& aLayerMap )
{
    LSET originalLayers = aItem->GetLayerSet();
    LSET newLayers;
 
    for( PCB_LAYER_ID original : originalLayers )
    {
        if( aLayerMap.count( original ) )
            newLayers.set( aLayerMap[ original ] );
        else
            newLayers.set( original );
    }
 
    if( originalLayers != newLayers )
    {
        m_commit->Modify( aItem, nullptr, RECURSE_MODE::RECURSE );
        aItem->SetLayerSet( newLayers );
        frame()->GetCanvas()->GetView()->Update( aItem, KIGFX::GEOMETRY );
        return true;
    }
 
    return false;
}
 
 
int GLOBAL_EDIT_TOOL::SwapLayers( const TOOL_EVENT& aEvent )
{
    std::map<PCB_LAYER_ID, PCB_LAYER_ID> layerMap;
 
    DIALOG_SWAP_LAYERS dlg( frame(), layerMap );
 
    if( dlg.ShowModal() != wxID_OK )
        return 0;
 
    bool hasChanges = false;
 
    // Change tracks.
    for( PCB_TRACK* segm : frame()->GetBoard()->Tracks() )
    {
        if( dynamic_cast<PCB_GENERATOR*>( segm->GetParentGroup() ) )
            continue;
 
        if( segm->Type() == PCB_VIA_T )
        {
            PCB_VIA*     via = static_cast<PCB_VIA*>( segm );
            PCB_LAYER_ID top_layer, bottom_layer;
 
            if( via->GetViaType() == VIATYPE::THROUGH )
                continue;
 
            via->LayerPair( &top_layer, &bottom_layer );
 
            if( layerMap[bottom_layer] != bottom_layer || layerMap[top_layer] != top_layer )
            {
                m_commit->Modify( via );
                via->SetLayerPair( layerMap[top_layer], layerMap[bottom_layer] );
                frame()->GetCanvas()->GetView()->Update( via, KIGFX::GEOMETRY );
                hasChanges = true;
            }
        }
        else
        {
            hasChanges |= swapBoardItem( segm, layerMap );
        }
    }
 
    for( PCB_GENERATOR* generator : frame()->GetBoard()->Generators() )
        hasChanges |= swapBoardItem( generator, layerMap );
 
    for( BOARD_ITEM* zone : frame()->GetBoard()->Zones() )
        hasChanges |= swapBoardItem( zone, layerMap );
 
    for( BOARD_ITEM* drawing : frame()->GetBoard()->Drawings() )
        hasChanges |= swapBoardItem( drawing, layerMap );
 
    if( hasChanges )
    {
        frame()->OnModify();
        m_commit->Push( _( "Swap Layers" ) );
        frame()->GetCanvas()->Refresh();
    }
 
    return 0;
}
 
 
int GLOBAL_EDIT_TOOL::EditTracksAndVias( const TOOL_EVENT& aEvent )
{
    PCB_EDIT_FRAME* editFrame = getEditFrame<PCB_EDIT_FRAME>();
    DIALOG_GLOBAL_EDIT_TRACKS_AND_VIAS dlg( editFrame );
 
    dlg.ShowQuasiModal();       // QuasiModal required for NET_SELECTOR
    return 0;
}
 
 
int GLOBAL_EDIT_TOOL::CleanupTracksAndVias( const TOOL_EVENT& aEvent )
{
    PCB_EDIT_FRAME* editFrame = getEditFrame<PCB_EDIT_FRAME>();
    DIALOG_CLEANUP_TRACKS_AND_VIAS dlg( editFrame );
 
    dlg.ShowModal();
    return 0;
}
 
 
int GLOBAL_EDIT_TOOL::CleanupGraphics( const TOOL_EVENT& aEvent )
{
    PCB_EDIT_FRAME* editFrame = getEditFrame<PCB_EDIT_FRAME>();
    DIALOG_CLEANUP_GRAPHICS dlg( editFrame, false );
 
    dlg.ShowModal();
    return 0;
}
 
 
int GLOBAL_EDIT_TOOL::RemoveUnusedPads( const TOOL_EVENT& aEvent )
{
    PCB_EDIT_FRAME* editFrame = getEditFrame<PCB_EDIT_FRAME>();
    PCB_SELECTION&  selection = m_selectionTool->RequestSelection(
            []( const VECTOR2I& aPt, GENERAL_COLLECTOR& aCollector, PCB_SELECTION_TOOL* sTool )
            {
                sTool->FilterCollectorForHierarchy( aCollector, true );
            } );
    DIALOG_UNUSED_PAD_LAYERS dlg( editFrame, selection, *m_commit );
 
    dlg.ShowModal();
 
    return 0;
}
 
int GLOBAL_EDIT_TOOL::ZonesManager( const TOOL_EVENT& aEvent )
{
    PCB_EDIT_FRAME* editFrame = getEditFrame<PCB_EDIT_FRAME>();
    BOARD_COMMIT    commit( editFrame );
    BOARD*          board = editFrame->GetBoard();
 
    for( ZONE* zone : board->Zones() )
        commit.Modify( zone );
 
    DIALOG_ZONE_MANAGER dlg( editFrame );
 
    int dialogResult = dlg.ShowQuasiModal();
 
    if( dialogResult == wxID_OK && dlg.GetRepourOnClose() )
        dialogResult = ZONE_MANAGER_REPOUR;
 
    if( dialogResult == wxID_CANCEL )
    {
        // The dialog hides zones in the view when they are marked for deletion
        if( KIGFX::VIEW* view = editFrame->GetCanvas()->GetView() )
        {
            for( ZONE* zone : dlg.GetZonesToDelete() )
                view->Hide( zone, false );
        }
 
        editFrame->GetCanvas()->Refresh();
        return 0;
    }
 
    // Promote the pre-staged CHT_MODIFY entries for any zone the user marked for deletion
    if( KIGFX::VIEW* view = editFrame->GetCanvas()->GetView() )
    {
        for( ZONE* zone : dlg.GetZonesToDelete() )
        {
            view->Hide( zone, false );
 
            commit.Unmodify( zone, nullptr );
            commit.Remove( zone );
        }
    }
 
    // Ensure all zones are deselected before make any change in view, to avoid
    // dangling pointers in EDIT_POINT
    PCB_SELECTION_TOOL* selTool = editFrame->GetToolManager()->GetTool<PCB_SELECTION_TOOL>();
    selTool->ClearSelection();
 
    wxBusyCursor dummy;
 
    // Clear the zone bounding box cache before Push() updates the VIEW, otherwise
    // View->Update() will query stale cached values and the VIEW's R-Tree will be
    // indexed with incorrect bounding boxes, causing single-click zone selection to fail.
    board->IncrementTimeStamp();
 
    commit.Push( _( "Zone Manager" ), SKIP_CONNECTIVITY );
 
    board->BuildConnectivity();
 
    if( TOOL_MANAGER* manager = GetManager() )
        manager->PostEvent( EVENTS::ConnectivityChangedEvent );
 
    editFrame->GetCanvas()->RedrawRatsnest();
 
    if( dialogResult == ZONE_MANAGER_REPOUR )
    {
        if( TOOL_MANAGER* manager = GetManager() )
            manager->PostAction( PCB_ACTIONS::zoneFillAll );
    }
 
    return 0;
}
 
 
 
int GLOBAL_EDIT_TOOL::Migrate3DModels( const TOOL_EVENT& /* aEvent */ )
{
    PCB_EDIT_FRAME* editFrame = getEditFrame<PCB_EDIT_FRAME>();
 
    if( !editFrame )
        return 0;
 
    DIALOG_MIGRATE_3D_MODELS dlg( editFrame );
    dlg.ShowModal();
    return 0;
}
 
 
void GLOBAL_EDIT_TOOL::setTransitions()
{
    Go( &GLOBAL_EDIT_TOOL::ExchangeFootprints,   PCB_ACTIONS::updateFootprint.MakeEvent() );
    Go( &GLOBAL_EDIT_TOOL::ExchangeFootprints,   PCB_ACTIONS::updateFootprints.MakeEvent() );
    Go( &GLOBAL_EDIT_TOOL::ExchangeFootprints,   PCB_ACTIONS::changeFootprint.MakeEvent() );
    Go( &GLOBAL_EDIT_TOOL::ExchangeFootprints,   PCB_ACTIONS::changeFootprints.MakeEvent() );
    Go( &GLOBAL_EDIT_TOOL::Migrate3DModels,      PCB_ACTIONS::migrate3DModels.MakeEvent() );
 
    Go( &GLOBAL_EDIT_TOOL::SwapLayers,           PCB_ACTIONS::swapLayers.MakeEvent() );
 
    Go( &GLOBAL_EDIT_TOOL::EditTracksAndVias,    PCB_ACTIONS::editTracksAndVias.MakeEvent() );
    Go( &GLOBAL_EDIT_TOOL::EditTextAndGraphics,  PCB_ACTIONS::editTextAndGraphics.MakeEvent() );
    Go( &GLOBAL_EDIT_TOOL::EditTeardrops,        PCB_ACTIONS::editTeardrops.MakeEvent() );
    Go( &GLOBAL_EDIT_TOOL::GlobalDeletions,      PCB_ACTIONS::globalDeletions.MakeEvent() );
    Go( &GLOBAL_EDIT_TOOL::CleanupTracksAndVias, PCB_ACTIONS::cleanupTracksAndVias.MakeEvent() );
    Go( &GLOBAL_EDIT_TOOL::CleanupGraphics,      PCB_ACTIONS::cleanupGraphics.MakeEvent() );
    Go( &GLOBAL_EDIT_TOOL::RemoveUnusedPads,     PCB_ACTIONS::removeUnusedPads.MakeEvent() );
    Go( &GLOBAL_EDIT_TOOL::ZonesManager,         PCB_ACTIONS::zonesManager.MakeEvent() );
}