|
|
/*
* This program source code file is part of KiCad, a free EDA CAD application. * * Copyright (C) 2019 CERN * Copyright (C) 2019 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, you may find one here: * http://www.gnu.org/licenses/old-licenses/gpl-2.0.html
* or you may search the http://www.gnu.org website for the version 2 license,
* or you may write to the Free Software Foundation, Inc., * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA */
#include <tool/tool_manager.h>
#include <tools/ee_grid_helper.h>
#include <tools/ee_selection_tool.h>
#include <tools/sch_line_wire_bus_tool.h>
#include <ee_actions.h>
#include <bitmaps.h>
#include <base_struct.h>
#include <sch_item.h>
#include <sch_component.h>
#include <sch_sheet.h>
#include <sch_view.h>
#include <sch_line.h>
#include <sch_edit_frame.h>
#include <eeschema_id.h>
#include <pgm_base.h>
#include <settings/settings_manager.h>
#include "sch_move_tool.h"
// For adding to or removing from selections
#define QUIET_MODE true
SCH_MOVE_TOOL::SCH_MOVE_TOOL() : EE_TOOL_BASE<SCH_EDIT_FRAME>( "eeschema.InteractiveMove" ), m_moveInProgress( false ), m_isDragOperation( false ), m_moveOffset( 0, 0 ){}
bool SCH_MOVE_TOOL::Init(){ EE_TOOL_BASE::Init();
auto moveCondition = [] ( const SELECTION& aSel ) { if( aSel.Empty() ) return false;
if( SCH_LINE_WIRE_BUS_TOOL::IsDrawingLineWireOrBus( aSel ) ) return false;
return true; };
// Add move actions to the selection tool menu
//
CONDITIONAL_MENU& selToolMenu = m_selectionTool->GetToolMenu().GetMenu();
selToolMenu.AddItem( EE_ACTIONS::move, moveCondition, 150 ); selToolMenu.AddItem( EE_ACTIONS::drag, moveCondition, 150 ); selToolMenu.AddItem( EE_ACTIONS::alignToGrid, moveCondition, 150 );
return true;}
static const KICAD_T movableItems[] ={ SCH_MARKER_T, SCH_JUNCTION_T, SCH_NO_CONNECT_T, SCH_BUS_BUS_ENTRY_T, SCH_BUS_WIRE_ENTRY_T, SCH_LINE_T, SCH_BITMAP_T, SCH_TEXT_T, SCH_LABEL_T, SCH_GLOBAL_LABEL_T, SCH_HIER_LABEL_T, SCH_FIELD_T, SCH_COMPONENT_T, SCH_SHEET_PIN_T, SCH_SHEET_T, EOT};
/* TODO - Tom/Jeff
- add preferences option "Move origin: always cursor / item origin" - add preferences option "Default drag action: drag items / move" - add preferences option "Drag always selects" */
int SCH_MOVE_TOOL::Main( const TOOL_EVENT& aEvent ){ EESCHEMA_SETTINGS* cfg = Pgm().GetSettingsManager().GetAppSettings<EESCHEMA_SETTINGS>(); KIGFX::VIEW_CONTROLS* controls = getViewControls(); EE_GRID_HELPER grid( m_toolMgr );
m_anchorPos.reset();
if( aEvent.IsAction( &EE_ACTIONS::move ) ) m_isDragOperation = false; else if( aEvent.IsAction( &EE_ACTIONS::drag ) ) m_isDragOperation = true; else if( aEvent.IsAction( &EE_ACTIONS::moveActivate ) ) m_isDragOperation = !cfg->m_Input.drag_is_move; else return 0;
if( m_moveInProgress ) { auto sel = m_selectionTool->GetSelection().Front();
if( sel && !sel->IsNew() ) { // User must have switched from move to drag or vice-versa. Reset the selected
// items so we can start again with the current m_isDragOperation.
m_frame->RollbackSchematicFromUndo(); m_selectionTool->RemoveItemsFromSel( &m_dragAdditions, QUIET_MODE ); m_anchorPos = m_cursor - m_moveOffset; m_moveInProgress = false; controls->SetAutoPan( false );
// And give it a kick so it doesn't have to wait for the first mouse movement to
// refresh.
m_toolMgr->RunAction( EE_ACTIONS::restartMove ); }
return 0; }
// Be sure that there is at least one item that we can move. If there's no selection try
// looking for the stuff under mouse cursor (i.e. Kicad old-style hover selection).
EE_SELECTION& selection = m_selectionTool->RequestSelection( movableItems ); bool unselect = selection.IsHover();
// Keep an original copy of the starting points for cleanup after the move
std::vector<DANGLING_END_ITEM> internalPoints;
if( selection.Empty() ) return 0;
Activate(); controls->ShowCursor( true );
bool restore_state = false; bool chain_commands = false; TOOL_EVENT* evt = const_cast<TOOL_EVENT*>( &aEvent ); VECTOR2I prevPos;
std::string tool = aEvent.GetCommandStr().get(); m_frame->PushTool( tool ); m_cursor = controls->GetCursorPosition();
// Main loop: keep receiving events
do { m_frame->GetCanvas()->SetCurrentCursor( wxCURSOR_ARROW );
if( evt->IsAction( &EE_ACTIONS::moveActivate ) || evt->IsAction( &EE_ACTIONS::restartMove ) || evt->IsAction( &EE_ACTIONS::move ) || evt->IsAction( &EE_ACTIONS::drag ) || evt->IsMotion() || evt->IsDrag( BUT_LEFT ) || evt->IsAction( &ACTIONS::refreshPreview ) ) { if( !m_moveInProgress ) // Prepare to start moving/dragging
{ SCH_ITEM* sch_item = (SCH_ITEM*) selection.Front(); bool appendUndo = sch_item && sch_item->IsNew(); bool placingNewItems = sch_item && sch_item->IsNew();
//------------------------------------------------------------------------
// Setup a drag or a move
//
m_dragAdditions.clear(); m_specialCaseLabels.clear(); internalPoints.clear();
for( auto it : m_frame->GetScreen()->Items() ) { it->ClearFlags( TEMP_SELECTED );
if( !it->IsSelected() ) it->ClearFlags( STARTPOINT | ENDPOINT );
if( !selection.IsHover() && it->IsSelected() ) it->SetFlags( STARTPOINT | ENDPOINT ); }
if( m_isDragOperation ) { // Add connections to the selection for a drag.
//
for( EDA_ITEM* item : selection ) { std::vector<wxPoint> connections;
if( item->Type() == SCH_LINE_T ) static_cast<SCH_LINE*>( item )->GetSelectedPoints( connections ); else connections = static_cast<SCH_ITEM*>( item )->GetConnectionPoints();
for( wxPoint point : connections ) getConnectedDragItems( static_cast<SCH_ITEM*>( item ), point, m_dragAdditions ); }
m_selectionTool->AddItemsToSel( &m_dragAdditions, QUIET_MODE ); } else { // Mark the edges of the block with dangling flags for a move.
for( EDA_ITEM* item : selection ) static_cast<SCH_ITEM*>( item )->GetEndPoints( internalPoints );
for( EDA_ITEM* item : selection ) static_cast<SCH_ITEM*>( item )->UpdateDanglingState( internalPoints ); } // Generic setup
//
for( EDA_ITEM* item : selection ) { if( item->IsNew() ) { if( item->HasFlag( TEMP_SELECTED ) && m_isDragOperation ) { // Item was added in getConnectedDragItems
saveCopyInUndoList( (SCH_ITEM*) item, UNDO_REDO::NEWITEM, appendUndo ); appendUndo = true; } else { // Item was added in a previous command (and saved to undo by
// that command)
} } else if( item->GetParent() && item->GetParent()->IsSelected() ) { // Item will be (or has been) saved to undo by parent
} else { saveCopyInUndoList( (SCH_ITEM*) item, UNDO_REDO::CHANGED, appendUndo ); appendUndo = true; }
SCH_ITEM* schItem = (SCH_ITEM*) item; schItem->SetStoredPos( schItem->GetPosition() ); }
// Set up the starting position and move/drag offset
//
m_cursor = controls->GetCursorPosition();
if( evt->IsAction( &EE_ACTIONS::restartMove ) ) { wxASSERT_MSG( m_anchorPos, "Should be already set from previous cmd" ); } else if( placingNewItems ) { m_anchorPos = selection.GetReferencePoint(); }
if( m_anchorPos ) { VECTOR2I delta = m_cursor - (*m_anchorPos);
// Drag items to the current cursor position
for( EDA_ITEM* item : selection ) { // Don't double move pins, fields, etc.
if( item->GetParent() && item->GetParent()->IsSelected() ) continue;
moveItem( item, delta ); updateView( item ); }
m_anchorPos = m_cursor; } // For some items, moving the cursor to anchor is not good (for instance large
// hierarchical sheets or components can have the anchor outside the view)
else if( selection.Size() == 1 && !sch_item->IsMovableFromAnchorPoint() ) { m_cursor = getViewControls()->GetCursorPosition( true ); m_anchorPos = m_cursor; } else { if( m_frame->GetMoveWarpsCursor() ) { // User wants to warp the mouse
std::vector<SCH_ITEM*> items;
for( EDA_ITEM* item : selection ) items.push_back( static_cast<SCH_ITEM*>( item ) );
m_cursor = grid.BestDragOrigin( m_cursor, items ); } else { // User does not want to warp the mouse
m_cursor = getViewControls()->GetCursorPosition( true ); } }
controls->SetCursorPosition( m_cursor, false ); m_toolMgr->PostEvent( EVENTS::SelectedItemsModified );
prevPos = m_cursor; controls->SetAutoPan( true ); m_moveInProgress = true; }
//------------------------------------------------------------------------
// Follow the mouse
//
m_cursor = controls->GetCursorPosition(); VECTOR2I delta( m_cursor - prevPos ); m_anchorPos = m_cursor;
m_moveOffset += delta; prevPos = m_cursor;
for( EDA_ITEM* item : selection ) { // Don't double move pins, fields, etc.
if( item->GetParent() && item->GetParent()->IsSelected() ) continue;
moveItem( item, delta ); updateView( item ); }
m_toolMgr->PostEvent( EVENTS::SelectedItemsMoved ); } //------------------------------------------------------------------------
// Handle cancel
//
else if( evt->IsCancelInteractive() ) { if( m_moveInProgress ) { evt->SetPassEvent( false ); restore_state = true; }
break; } //------------------------------------------------------------------------
// Handle TOOL_ACTION special cases
//
else if( evt->Action() == TA_UNDO_REDO_PRE ) { unselect = true; break; } else if( evt->IsAction( &ACTIONS::doDelete ) ) { // Exit on a remove operation; there is no further processing for removed items.
break; } else if( evt->IsAction( &ACTIONS::duplicate ) ) { if( selection.Front()->IsNew() ) { // This doesn't really make sense; we'll just end up dragging a stack of
// objects so Duplicate() is going to ignore this and we'll just carry on.
continue; }
// Move original back and exit. The duplicate will run in its own loop.
restore_state = true; unselect = false; chain_commands = true; break; } else if( evt->Action() == TA_CHOICE_MENU_CHOICE ) { if( evt->GetCommandId().get() >= ID_POPUP_SCH_SELECT_UNIT_CMP && evt->GetCommandId().get() <= ID_POPUP_SCH_SELECT_UNIT_CMP_MAX ) { SCH_COMPONENT* component = dynamic_cast<SCH_COMPONENT*>( selection.Front() ); int unit = evt->GetCommandId().get() - ID_POPUP_SCH_SELECT_UNIT_CMP;
if( component ) { m_frame->SelectUnit( component, unit ); m_toolMgr->RunAction( ACTIONS::refreshPreview ); } } } //------------------------------------------------------------------------
// Handle context menu
//
else if( evt->IsClick( BUT_RIGHT ) ) { m_menu.ShowContextMenu( m_selectionTool->GetSelection() ); } //------------------------------------------------------------------------
// Handle drop
//
else if( evt->IsMouseUp( BUT_LEFT ) || evt->IsClick( BUT_LEFT ) ) { break; // Finish
} else evt->SetPassEvent();
controls->SetAutoPan( m_moveInProgress );
} while( ( evt = Wait() ) ); //Should be assignment not equality test
controls->ForceCursorPosition( false ); controls->ShowCursor( false ); controls->SetAutoPan( false );
if( !chain_commands ) m_moveOffset = { 0, 0 };
m_anchorPos.reset();
for( EDA_ITEM* item : selection ) item->ClearEditFlags();
if( restore_state ) { m_selectionTool->RemoveItemsFromSel( &m_dragAdditions, QUIET_MODE ); m_frame->RollbackSchematicFromUndo(); } else { // Moving items changes the RTree box bounds.
for( auto item : selection ) { switch( item->Type() ) { // Moving sheet pins does not change the BBox.
case SCH_SHEET_PIN_T: break;
// Moving fields should update the associated component
case SCH_FIELD_T: if( item->GetParent() ) m_frame->GetScreen()->Update( static_cast<SCH_ITEM*>( item->GetParent() ) );
break;
default: m_frame->GetScreen()->Update( static_cast<SCH_ITEM*>( item ) ); } }
// If we move items away from a junction, we _may_ want to add a junction there
// to denote the state.
for( auto it : internalPoints ) { if( m_frame->GetScreen()->IsJunctionNeeded( it.GetPosition(), true ) ) m_frame->AddJunction( m_frame->GetScreen(), it.GetPosition(), true, false ); }
m_toolMgr->RunAction( EE_ACTIONS::addNeededJunctions, true, &selection ); m_selectionTool->RemoveItemsFromSel( &m_dragAdditions, QUIET_MODE );
m_frame->SchematicCleanUp(); m_frame->TestDanglingEnds();
m_frame->OnModify(); }
if( unselect ) m_toolMgr->RunAction( EE_ACTIONS::clearSelection, true ); else m_selectionTool->RebuildSelection(); // Schematic cleanup might have merged lines, etc.
m_dragAdditions.clear(); m_moveInProgress = false; m_frame->PopTool( tool ); return 0;}
void SCH_MOVE_TOOL::getConnectedDragItems( SCH_ITEM* aOriginalItem, wxPoint aPoint, EDA_ITEMS& aList ){ EE_RTREE& items = m_frame->GetScreen()->Items();
for( SCH_ITEM *test : items.Overlapping( aOriginalItem->GetBoundingBox() ) ) { if( test == aOriginalItem || test->IsSelected() || !test->CanConnect( aOriginalItem ) ) continue;
KICAD_T testType = test->Type();
switch( testType ) { case SCH_LINE_T: { // Select the connected end of wires/bus connections.
SCH_LINE* testLine = static_cast<SCH_LINE*>( test ); wxPoint otherEnd;
if( testLine->GetStartPoint() == aPoint ) { if( !testLine->HasFlag( TEMP_SELECTED ) ) aList.push_back( testLine );
testLine->SetFlags( STARTPOINT | TEMP_SELECTED ); otherEnd = testLine->GetEndPoint(); } else if( testLine->GetEndPoint() == aPoint ) { if( !testLine->HasFlag( TEMP_SELECTED ) ) aList.push_back( testLine );
testLine->SetFlags( ENDPOINT | TEMP_SELECTED ); otherEnd = testLine->GetStartPoint(); } else { break; }
// Since only one end is going to move, the movement vector of any labels attached
// to it is scaled by the proportion of the line length the label is from the moving
// end.
for( SCH_ITEM* item : m_frame->GetScreen()->Items().OfType( SCH_LABEL_T ) ) { if( item->IsSelected() ) continue; // These will be moved on their own because they're selected
if( item->CanConnect( testLine ) && testLine->HitTest( item->GetPosition(), 1 ) ) { SCH_TEXT* label = static_cast<SCH_TEXT*>( item );
if( !label->HasFlag( TEMP_SELECTED ) ) aList.push_back( label );
SPECIAL_CASE_LABEL_INFO info; info.attachedLine = testLine; info.originalLabelPos = label->GetPosition(); m_specialCaseLabels[label] = info; } }
break; }
case SCH_SHEET_T: // Dragging a sheet just because it's connected to something else feels a bit like
// the tail wagging the dog, but this could be moved down to the next case.
break;
case SCH_COMPONENT_T: if( test->IsConnected( aPoint ) ) { // Add a new wire between the component and the selected item so the selected
// item can be dragged.
SCH_LINE* newWire = new SCH_LINE( aPoint, LAYER_WIRE ); newWire->SetFlags( IS_NEW ); m_frame->AddToScreen( newWire, m_frame->GetScreen() );
newWire->SetFlags( TEMP_SELECTED | STARTPOINT ); aList.push_back( newWire ); } break;
case SCH_NO_CONNECT_T: case SCH_JUNCTION_T: // Select no-connects and junctions that are connected to items being moved.
if( !test->HasFlag( TEMP_SELECTED ) && test->IsConnected( aPoint ) ) { aList.push_back( test ); test->SetFlags( TEMP_SELECTED ); }
break;
case SCH_LABEL_T: case SCH_GLOBAL_LABEL_T: case SCH_HIER_LABEL_T: case SCH_BUS_WIRE_ENTRY_T: case SCH_BUS_BUS_ENTRY_T: // Performance optimization:
if( test->HasFlag( TEMP_SELECTED ) ) break;
// Select labels and bus entries that are connected to a wire being moved.
if( aOriginalItem->Type() == SCH_LINE_T ) { std::vector<wxPoint> connections = test->GetConnectionPoints();
for( wxPoint& point : connections ) { if( aOriginalItem->HitTest( point, 1 ) ) { test->SetFlags( TEMP_SELECTED ); aList.push_back( test );
// A bus entry needs its wire & label as well
if( testType == SCH_BUS_WIRE_ENTRY_T || testType == SCH_BUS_BUS_ENTRY_T ) { std::vector<wxPoint> ends = test->GetConnectionPoints(); wxPoint otherEnd;
if( ends[0] == point ) otherEnd = ends[1]; else otherEnd = ends[0];
getConnectedDragItems( test, otherEnd, aList ); } break; } } }
break;
default: break; } }}
void SCH_MOVE_TOOL::moveItem( EDA_ITEM* aItem, const VECTOR2I& aDelta ){ switch( aItem->Type() ) { case SCH_LINE_T: { SCH_LINE* line = static_cast<SCH_LINE*>( aItem );
if( aItem->HasFlag( STARTPOINT ) ) line->MoveStart( (wxPoint) aDelta );
if( aItem->HasFlag( ENDPOINT ) ) line->MoveEnd( (wxPoint) aDelta );
} break;
case SCH_PIN_T: case SCH_FIELD_T: { SCH_ITEM* parent = (SCH_ITEM*) aItem->GetParent(); wxPoint delta( aDelta );
if( parent && parent->Type() == SCH_COMPONENT_T ) { SCH_COMPONENT* component = (SCH_COMPONENT*) aItem->GetParent(); TRANSFORM transform = component->GetTransform().InverseTransform();
delta = transform.TransformCoordinate( delta ); }
static_cast<SCH_ITEM*>( aItem )->Move( delta );
// If we're moving a field with respect to its parent then it's no longer auto-placed
if( aItem->Type() == SCH_FIELD_T && parent && !parent->IsSelected() ) parent->ClearFieldsAutoplaced();
break; } case SCH_SHEET_PIN_T: { SCH_SHEET_PIN* pin = (SCH_SHEET_PIN*) aItem; pin->SetStoredPos( pin->GetStoredPos() + (wxPoint) aDelta ); pin->ConstrainOnEdge( pin->GetStoredPos() ); break; } case SCH_LABEL_T: { SCH_TEXT* label = static_cast<SCH_TEXT*>( aItem );
if( m_specialCaseLabels.count( label ) ) { SPECIAL_CASE_LABEL_INFO info = m_specialCaseLabels[ label ]; SEG currentLine( info.attachedLine->GetStartPoint(), info.attachedLine->GetEndPoint() ); label->SetPosition( (wxPoint) currentLine.NearestPoint( info.originalLabelPos ) ); } else label->Move( (wxPoint) aDelta );
break; } default: static_cast<SCH_ITEM*>( aItem )->Move( (wxPoint) aDelta ); break; }
getView()->Hide( aItem, false ); aItem->SetFlags( IS_MOVED );}
int SCH_MOVE_TOOL::AlignElements( const TOOL_EVENT& aEvent ){ EE_GRID_HELPER grid( m_toolMgr); EE_SELECTION& selection = m_selectionTool->RequestSelection( movableItems ); bool append_undo = false;
for( EDA_ITEM* item : selection ) { if( item->Type() == SCH_LINE_T ) { SCH_LINE* line = static_cast<SCH_LINE*>( item ); std::vector<int> flags{ STARTPOINT, ENDPOINT }; std::vector<wxPoint> pts{ line->GetStartPoint(), line->GetEndPoint() };
for( int ii = 0; ii < 2; ++ii ) { EDA_ITEMS drag_items{ item }; line->ClearFlags(); line->SetFlags( flags[ii] ); getConnectedDragItems( line, pts[ii], drag_items ); std::set<EDA_ITEM*> unique_items( drag_items.begin(), drag_items.end() );
VECTOR2I gridpt = grid.AlignGrid( pts[ii] ) - pts[ii];
if( gridpt != VECTOR2I( 0, 0 ) ) { for( auto dritem : unique_items ) { if( dritem->GetParent() && dritem->GetParent()->IsSelected() ) continue;
saveCopyInUndoList( dritem, UNDO_REDO::CHANGED, append_undo ); append_undo = true;
moveItem( dritem, gridpt ); updateView( dritem ); } }
} } else { std::vector<wxPoint> connections; EDA_ITEMS drag_items{ item }; connections = static_cast<SCH_ITEM*>( item )->GetConnectionPoints();
for( wxPoint point : connections ) getConnectedDragItems( static_cast<SCH_ITEM*>( item ), point, drag_items );
std::map<VECTOR2I, int> shifts; VECTOR2I most_common( 0, 0 ); int max_count = 0;
for( auto& conn : connections ) { VECTOR2I gridpt = grid.AlignGrid( conn ) - conn;
shifts[gridpt]++;
if( shifts[gridpt] > max_count ) { most_common = gridpt; max_count = shifts[most_common]; } }
if( most_common != VECTOR2I( 0, 0 ) ) { for( auto dritem : drag_items ) { if( dritem->GetParent() && dritem->GetParent()->IsSelected() ) continue;
saveCopyInUndoList( dritem, UNDO_REDO::CHANGED, append_undo ); append_undo = true;
moveItem( dritem, most_common ); updateView( dritem ); } } } }
m_toolMgr->PostEvent( EVENTS::SelectedItemsMoved ); m_toolMgr->RunAction( EE_ACTIONS::addNeededJunctions, true, &selection );
m_frame->SchematicCleanUp(); m_frame->TestDanglingEnds();
m_frame->OnModify(); return 0;}
void SCH_MOVE_TOOL::setTransitions(){ Go( &SCH_MOVE_TOOL::Main, EE_ACTIONS::moveActivate.MakeEvent() ); Go( &SCH_MOVE_TOOL::Main, EE_ACTIONS::move.MakeEvent() ); Go( &SCH_MOVE_TOOL::Main, EE_ACTIONS::drag.MakeEvent() ); Go( &SCH_MOVE_TOOL::AlignElements, EE_ACTIONS::alignToGrid.MakeEvent() );}
|