"Select net/connection" (GAL).

pcbnew/ratsnest_data.cpp

@@ -730,6 +730,34 @@ std::list<RN_NODE_PTR> RN_NET::GetNodes( const BOARD_CONNECTED_ITEM* aItem ) con
 }
 
 
+void RN_NET::GetAllItems( std::list<BOARD_CONNECTED_ITEM*>& aOutput, RN_ITEM_TYPE aType ) const
+{
+    if( aType & RN_PADS )
+    {
+        BOOST_FOREACH( const BOARD_CONNECTED_ITEM* aItem, m_pads | boost::adaptors::map_keys )
+            aOutput.push_back( const_cast<BOARD_CONNECTED_ITEM*>( aItem ) );
+    }
+
+    if( aType & RN_VIAS )
+    {
+        BOOST_FOREACH( const BOARD_CONNECTED_ITEM* aItem, m_vias | boost::adaptors::map_keys )
+            aOutput.push_back( const_cast<BOARD_CONNECTED_ITEM*>( aItem ) );
+    }
+
+    if( aType & RN_TRACKS )
+    {
+        BOOST_FOREACH( const BOARD_CONNECTED_ITEM* aItem, m_tracks | boost::adaptors::map_keys )
+            aOutput.push_back( const_cast<BOARD_CONNECTED_ITEM*>( aItem ) );
+    }
+
+    if( aType & RN_ZONES )
+    {
+        BOOST_FOREACH( const BOARD_CONNECTED_ITEM* aItem, m_zoneConnections | boost::adaptors::map_keys )
+            aOutput.push_back( const_cast<BOARD_CONNECTED_ITEM*>( aItem ) );
+    }
+}
+
+
 void RN_NET::ClearSimple()
 {
     BOOST_FOREACH( const RN_NODE_PTR& node, m_simpleNodes )
@@ -745,7 +773,7 @@ void RN_NET::ClearSimple()
 
 void RN_NET::GetConnectedItems( const BOARD_CONNECTED_ITEM* aItem,
                                 std::list<BOARD_CONNECTED_ITEM*>& aOutput,
-                                RN_ITEM_TYPES aTypes ) const
+                                RN_ITEM_TYPE aTypes ) const
 {
     std::list<RN_NODE_PTR> nodes = GetNodes( aItem );
     assert( !nodes.empty() );
@@ -872,7 +900,7 @@ void RN_DATA::AddSimple( const VECTOR2I& aPosition, int aNetCode )
 
 void RN_DATA::GetConnectedItems( const BOARD_CONNECTED_ITEM* aItem,
                                  std::list<BOARD_CONNECTED_ITEM*>& aOutput,
-                                 RN_ITEM_TYPES aTypes ) const
+                                 RN_ITEM_TYPE aTypes ) const
 {
     int net = aItem->GetNetCode();
 
@@ -885,6 +913,18 @@ void RN_DATA::GetConnectedItems( const BOARD_CONNECTED_ITEM* aItem,
 }
 
 
+void RN_DATA::GetNetItems( int aNetCode, std::list<BOARD_CONNECTED_ITEM*>& aOutput,
+                           RN_ITEM_TYPE aTypes ) const
+{
+    if( aNetCode < 1 )
+        return;
+
+    assert( aNetCode < (int) m_nets.size() );
+
+    m_nets[aNetCode].GetAllItems( aOutput, aTypes );
+}
+
+
 bool RN_DATA::AreConnected( const BOARD_CONNECTED_ITEM* aItem, const BOARD_CONNECTED_ITEM* aOther )
 {
     int net1 = aItem->GetNetCode();

pcbnew/ratsnest_data.h

@@ -50,7 +50,7 @@ class ZONE_CONTAINER;
 class CPolyPt;
 
 ///> Types of items that are handled by the class
-enum RN_ITEM_TYPES
+enum RN_ITEM_TYPE
 {
     RN_PADS    = 0x01,
     RN_VIAS    = 0x02,
@@ -399,6 +399,14 @@ public:
      */
     std::list<RN_NODE_PTR> GetNodes( const BOARD_CONNECTED_ITEM* aItem ) const;
 
+    /**
+     * Function GetAllNodes()
+     * Adds all stored items to a list.
+     * @param aOutput is the list that will have items added.
+     * @param aType determines the type of added items.
+     */
+    void GetAllItems( std::list<BOARD_CONNECTED_ITEM*>& aOutput, RN_ITEM_TYPE aType = RN_ALL ) const;
+
     /**
      * Function GetClosestNode()
      * Returns a single node that lies in the shortest distance from a specific node.
@@ -487,7 +495,7 @@ public:
      */
     void GetConnectedItems( const BOARD_CONNECTED_ITEM* aItem,
                             std::list<BOARD_CONNECTED_ITEM*>& aOutput,
-                            RN_ITEM_TYPES aTypes = RN_ALL) const;
+                            RN_ITEM_TYPE aTypes = RN_ALL ) const;
 
 protected:
     ///> Validates edge, i.e. modifies source and target nodes for an edge
@@ -524,6 +532,7 @@ protected:
     typedef boost::unordered_map<const TRACK*, RN_EDGE_MST_PTR> TRACK_EDGE_MAP;
     typedef boost::unordered_map<const ZONE_CONTAINER*, std::deque<RN_POLY> > ZONE_POLY_MAP;
     typedef boost::unordered_map<const ZONE_CONTAINER*, std::deque<RN_EDGE_MST_PTR> > ZONE_EDGE_MAP;
+
     ///> Map that associates nodes in the ratsnest model to respective nodes.
     PAD_NODE_MAP m_pads;
 
@@ -661,7 +670,17 @@ public:
      */
     void GetConnectedItems( const BOARD_CONNECTED_ITEM* aItem,
                             std::list<BOARD_CONNECTED_ITEM*>& aOutput,
-                            RN_ITEM_TYPES aTypes = RN_ALL ) const;
+                            RN_ITEM_TYPE aTypes = RN_ALL ) const;
+
+    /**
+     * Function GetNetItems()
+     * Adds all items that belong to a certain net to a list.
+     * @param aNetCode is the net code.
+     * @param aOutput is the list that will have items added.
+     * @param aTypes allows to filter by item types.
+     */
+    void GetNetItems( int aNetCode, std::list<BOARD_CONNECTED_ITEM*>& aOutput,
+                            RN_ITEM_TYPE aTypes = RN_ALL ) const;
 
     /**
      * Function AreConnected()

pcbnew/tools/common_actions.cpp

@@ -48,6 +48,14 @@ TOOL_ACTION COMMON_ACTIONS::selectionClear( "pcbnew.InteractiveSelection.Clear",
         AS_GLOBAL, 0,
         "", "" );    // No description, it is not supposed to be shown anywhere
 
+TOOL_ACTION COMMON_ACTIONS::selectConnection( "pcbnew.InteractiveSelection.SelectConnection",
+        AS_GLOBAL, 0,
+        "copper connection", "Selects whole copper connection." );
+
+TOOL_ACTION COMMON_ACTIONS::selectNet( "pcbnew.InteractiveSelection.SelectNet",
+        AS_GLOBAL, 0,
+        "whole net", "Selects all tracks & vias belonging to the same net." );
+
 TOOL_ACTION COMMON_ACTIONS::find( "pcbnew.InteractiveSelection.Find",
         AS_GLOBAL, 0,                                    // it is handled by wxWidgets hotkey system
         "Find an item", "Searches the document for an item" );

pcbnew/tools/common_actions.h

@@ -56,6 +56,12 @@ public:
     /// Unselects an item (specified as the event parameter).
     static TOOL_ACTION unselectItem;
 
+    /// Selects whole copper connection.
+    static TOOL_ACTION selectConnection;
+
+    /// Selects all connections belonging to a single net.
+    static TOOL_ACTION selectNet;
+
     // Edit Tool
     /// Activation of the edit tool
     static TOOL_ACTION editActivate;

pcbnew/tools/selection_conditions.cpp

@@ -34,6 +34,23 @@ bool SELECTION_CONDITIONS::NotEmpty( const SELECTION& aSelection )
 }
 
 
+bool SELECTION_CONDITIONS::OnlyConnectedItems( const SELECTION& aSelection )
+{
+    if( aSelection.Empty() )
+        return false;
+
+    for( int i = 0; i < aSelection.Size(); ++i )
+    {
+        KICAD_T type = aSelection.Item<EDA_ITEM>( i )->Type();
+
+        if( type != PCB_PAD_T && type != PCB_VIA_T && type != PCB_TRACE_T && type != PCB_ZONE_T )
+            return false;
+    }
+
+    return true;
+}
+
+
 SELECTION_CONDITION SELECTION_CONDITIONS::HasType( KICAD_T aType )
 {
     return boost::bind( &SELECTION_CONDITIONS::hasTypeFunc, _1, aType );

pcbnew/tools/selection_conditions.h

@@ -65,6 +65,14 @@ public:
      */
     static bool NotEmpty( const SELECTION& aSelection );
 
+    /**
+     * Function OnlyConnectedItems
+     * Tests if selection contains exclusively connected items (pads, tracks, vias, zones).
+     * @param aSelection is the selection to be tested.
+     * @return True if there are only connected items connected.
+     */
+    static bool OnlyConnectedItems( const SELECTION& aSelection );
+
     /**
      * Function HasType
      * Creates functor that tests if among the selected items there is at least one of a given type.

pcbnew/tools/selection_tool.cpp

@@ -28,7 +28,6 @@
 #include <boost/bind.hpp>
 #include <boost/function.hpp>
 
-
 #include <class_board.h>
 #include <class_board_item.h>
 #include <class_track.h>
@@ -48,12 +47,23 @@
 
 #include <tool/tool_event.h>
 #include <tool/tool_manager.h>
+#include <ratsnest_data.h>
 
 #include "selection_tool.h"
 #include "selection_area.h"
 #include "bright_box.h"
 #include "common_actions.h"
 
+class SELECT_MENU: public CONTEXT_MENU
+{
+public:
+    SELECT_MENU()
+    {
+        Add( COMMON_ACTIONS::selectConnection );
+        Add( COMMON_ACTIONS::selectNet );
+    }
+};
+
 SELECTION_TOOL::SELECTION_TOOL() :
         TOOL_INTERACTIVE( "pcbnew.InteractiveSelection" ),
         m_frame( NULL ), m_additive( false ), m_multiple( false ),
@@ -61,6 +71,10 @@ SELECTION_TOOL::SELECTION_TOOL() :
 {
     m_selArea = new SELECTION_AREA;
     m_selection.group = new KIGFX::VIEW_GROUP;
+
+    AddSubMenu( new SELECT_MENU, "Select...",
+            (SELECTION_CONDITION) SELECTION_CONDITIONS::OnlyConnectedItems &&
+            SELECTION_CONDITIONS::Count( 1 ) );
 }
 
 
@@ -207,6 +221,16 @@ int SELECTION_TOOL::Main( const TOOL_EVENT& aEvent )
         {
             clearSelection();
         }
+
+        else if( evt->IsAction( &COMMON_ACTIONS::selectConnection ) )
+        {
+            selectConnection( *evt );
+        }
+
+        else if( evt->IsAction( &COMMON_ACTIONS::selectNet ) )
+        {
+            selectNet( *evt );
+        }
     }
 
     // This tool is supposed to be active forever
@@ -408,8 +432,11 @@ void SELECTION_TOOL::setTransitions()
     Go( &SELECTION_TOOL::ClearSelection, COMMON_ACTIONS::selectionClear.MakeEvent() );
     Go( &SELECTION_TOOL::SelectItem, COMMON_ACTIONS::selectItem.MakeEvent() );
     Go( &SELECTION_TOOL::UnselectItem, COMMON_ACTIONS::unselectItem.MakeEvent() );
+    Go( &SELECTION_TOOL::SelectItem, COMMON_ACTIONS::selectItem.MakeEvent() );
     Go( &SELECTION_TOOL::find, COMMON_ACTIONS::find.MakeEvent() );
     Go( &SELECTION_TOOL::findMove, COMMON_ACTIONS::findMove.MakeEvent() );
+    Go( &SELECTION_TOOL::selectConnection, COMMON_ACTIONS::selectConnection.MakeEvent() );
+    Go( &SELECTION_TOOL::selectNet, COMMON_ACTIONS::selectNet.MakeEvent() );
 }
 
 
@@ -512,6 +539,58 @@ int SELECTION_TOOL::UnselectItem( const TOOL_EVENT& aEvent )
     return 0;
 }
 
+
+int SELECTION_TOOL::selectConnection( const TOOL_EVENT& aEvent )
+{
+    std::list<BOARD_CONNECTED_ITEM*> itemsList;
+    RN_DATA* ratsnest = getModel<BOARD>()->GetRatsnest();
+    BOARD_CONNECTED_ITEM* item = m_selection.Item<BOARD_CONNECTED_ITEM>( 0 );
+
+    clearSelection();
+    ratsnest->GetConnectedItems( item, itemsList, (RN_ITEM_TYPE)( RN_TRACKS | RN_VIAS ) );
+
+    BOOST_FOREACH( BOARD_CONNECTED_ITEM* i, itemsList )
+        select( i );
+
+    // Inform other potentially interested tools
+    if( itemsList.size() > 0 )
+    {
+        TOOL_EVENT selectEvent( SelectedEvent );
+        m_toolMgr->ProcessEvent( selectEvent );
+    }
+
+    setTransitions();
+
+    return 0;
+}
+
+
+int SELECTION_TOOL::selectNet( const TOOL_EVENT& aEvent )
+{
+    std::list<BOARD_CONNECTED_ITEM*> itemsList;
+    RN_DATA* ratsnest = getModel<BOARD>()->GetRatsnest();
+    BOARD_CONNECTED_ITEM* item = m_selection.Item<BOARD_CONNECTED_ITEM>( 0 );
+    int netCode = item->GetNetCode();
+
+    clearSelection();
+    ratsnest->GetNetItems( netCode, itemsList, (RN_ITEM_TYPE)( RN_TRACKS | RN_VIAS ) );
+
+    BOOST_FOREACH( BOARD_CONNECTED_ITEM* i, itemsList )
+        select( i );
+
+    // Inform other potentially interested tools
+    if( itemsList.size() > 0 )
+    {
+        TOOL_EVENT selectEvent( SelectedEvent );
+        m_toolMgr->ProcessEvent( selectEvent );
+    }
+
+    setTransitions();
+
+    return 0;
+}
+
+
 void SELECTION_TOOL::findCallback( BOARD_ITEM* aItem )
 {
     clearSelection();

pcbnew/tools/selection_tool.h

@@ -206,6 +206,12 @@ private:
      */
     bool selectMultiple();
 
+    ///> Selects a continuous copper connection.
+    int selectConnection( const TOOL_EVENT& aEvent );
+
+    ///> Selects all copper connections belonging to a single net.
+    int selectNet( const TOOL_EVENT& aEvent );
+
     ///> Find dialog callback.
     void findCallback( BOARD_ITEM* aItem );