/* * This program source code file is part of KICAD, a free EDA CAD application. * * Copyright (C) 2013-2015 CERN * @author Maciej Suminski * * 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 */ /** * @file ratsnest_data.h * @brief Class that computes missing connections on a PCB. */ #ifndef RATSNEST_DATA_H #define RATSNEST_DATA_H #include #include #include #include #include #include class BOARD; class BOARD_ITEM; class BOARD_CONNECTED_ITEM; class MODULE; class D_PAD; class VIA; class TRACK; class ZONE_CONTAINER; class SHAPE_POLY_SET; ///> Types of items that are handled by the class enum RN_ITEM_TYPE { RN_PADS = 0x01, RN_VIAS = 0x02, RN_TRACKS = 0x04, RN_ZONES = 0x08, RN_ALL = 0xFF }; // Preserve KiCad coding style policy typedef hed::NODE RN_NODE; typedef hed::NODE_PTR RN_NODE_PTR; typedef hed::EDGE RN_EDGE; typedef hed::EDGE_PTR RN_EDGE_PTR; typedef hed::EDGE_MST RN_EDGE_MST; typedef hed::TRIANGULATION TRIANGULATOR; typedef boost::shared_ptr RN_EDGE_MST_PTR; bool operator==( const RN_NODE_PTR& aFirst, const RN_NODE_PTR& aSecond ); bool operator!=( const RN_NODE_PTR& aFirst, const RN_NODE_PTR& aSecond ); struct RN_NODE_OR_FILTER; struct RN_NODE_AND_FILTER; ///> General interface for filtering out nodes in search functions. struct RN_NODE_FILTER : public std::unary_function { virtual ~RN_NODE_FILTER() {} virtual bool operator()( const RN_NODE_PTR& aNode ) const { return true; // By default everything passes } friend RN_NODE_AND_FILTER operator&&( const RN_NODE_FILTER& aFilter1, const RN_NODE_FILTER& aFilter2 ); friend RN_NODE_OR_FILTER operator||( const RN_NODE_FILTER& aFilter1, const RN_NODE_FILTER& aFilter2 ); }; RN_NODE_AND_FILTER operator&&( const RN_NODE_FILTER& aFilter1, const RN_NODE_FILTER& aFilter2 ); RN_NODE_OR_FILTER operator||( const RN_NODE_FILTER& aFilter1, const RN_NODE_FILTER& aFilter2 ); ///> Filters out nodes that have the flag set. struct WITHOUT_FLAG : public RN_NODE_FILTER { bool operator()( const RN_NODE_PTR& aNode ) const { return !aNode->GetFlag(); } }; ///> Filters out nodes with a specific tag struct DIFFERENT_TAG : public RN_NODE_FILTER { DIFFERENT_TAG( int aTag ) : m_tag( aTag ) {} bool operator()( const RN_NODE_PTR& aNode ) const { return aNode->GetTag() != m_tag; } private: int m_tag; }; struct RN_NODE_AND_FILTER : public RN_NODE_FILTER { RN_NODE_AND_FILTER( const RN_NODE_FILTER& aFilter1, const RN_NODE_FILTER& aFilter2 ) : m_filter1( aFilter1 ), m_filter2( aFilter2 ) {} bool operator()( const RN_NODE_PTR& aNode ) const { return m_filter1( aNode ) && m_filter2( aNode ); } private: const RN_NODE_FILTER& m_filter1; const RN_NODE_FILTER& m_filter2; }; struct RN_NODE_OR_FILTER : public RN_NODE_FILTER { RN_NODE_OR_FILTER( const RN_NODE_FILTER& aFilter1, const RN_NODE_FILTER& aFilter2 ) : m_filter1( aFilter1 ), m_filter2( aFilter2 ) {} bool operator()( const RN_NODE_PTR& aNode ) const { return m_filter1( aNode ) || m_filter2( aNode ); } private: const RN_NODE_FILTER& m_filter1; const RN_NODE_FILTER& m_filter2; }; ///> Functor comparing if two nodes are equal by their coordinates. It is required to make set of ///> shared pointers work properly. struct RN_NODE_COMPARE : std::binary_function { bool operator()( const RN_NODE_PTR& aNode1, const RN_NODE_PTR& aNode2 ) const { return aNode1 == aNode2; } }; ///> Functor calculating hash for a given node. It is required to make set of shared pointers ///> work properly. struct RN_NODE_HASH : std::unary_function { std::size_t operator()( const RN_NODE_PTR& aNode ) const { std::size_t hash = 2166136261u; hash ^= aNode->GetX(); hash *= 16777619; hash ^= aNode->GetY(); return hash; } }; /** * Class RN_LINKS * Manages data describing nodes and connections for a given net. */ class RN_LINKS { public: // Helper typedefs typedef boost::unordered_set RN_NODE_SET; typedef std::list RN_EDGE_LIST; /** * Function AddNode() * Adds a node with given coordinates and returns pointer to the newly added node. If the node * existed before, only appropriate pointer is returned. * @param aX is the x coordinate of a node. * @param aY is the y coordinate of a node. * @return Pointer to the node with given coordinates. */ const RN_NODE_PTR& AddNode( int aX, int aY ); /** * Function RemoveNode() * Removes a node described by a given node pointer. * @param aNode is a pointer to node to be removed. * @return True if node was removed, false if there were other references, so it was kept. */ bool RemoveNode( const RN_NODE_PTR& aNode ); /** * Function GetNodes() * Returns the set of currently used nodes. * @return The set of currently used nodes. */ const RN_NODE_SET& GetNodes() const { return m_nodes; } /** * Function AddConnection() * Adds a connection between two nodes and of given distance. Edges with distance equal 0 are * considered to be existing connections. Distance different than 0 means that the connection * is missing. * @param aNode1 is the origin node of a new connection. * @param aNode2 is the end node of a new connection. * @param aDistance is the distance of the connection (0 means that nodes are actually * connected, >0 means a missing connection). */ RN_EDGE_MST_PTR AddConnection( const RN_NODE_PTR& aNode1, const RN_NODE_PTR& aNode2, unsigned int aDistance = 0 ); /** * Function RemoveConnection() * Removes a connection described by a given edge pointer. * @param aEdge is a pointer to edge to be removed. */ void RemoveConnection( const RN_EDGE_PTR& aEdge ) { m_edges.remove( aEdge ); } /** * Function GetConnections() * Returns the list of edges that currently connect nodes. * @return the list of edges that currently connect nodes. */ const RN_EDGE_LIST& GetConnections() const { return m_edges; } protected: ///> Set of nodes that are expected to be connected together (vias, tracks, pads). RN_NODE_SET m_nodes; ///> List of edges that currently connect nodes. RN_EDGE_LIST m_edges; }; /** * Class RN_POLY * Describes a single subpolygon (ZONE_CONTAINER is supposed to contain one or more of those) and * performs fast point-inside-polygon test. */ class RN_POLY { public: RN_POLY( const SHAPE_POLY_SET* aParent, int aSubpolygonIndex, RN_LINKS& aConnections, const BOX2I& aBBox ); /** * Function GetNode() * Returns node representing a polygon (it has the same coordinates as the first point of its * bounding polyline. */ inline const RN_NODE_PTR& GetNode() const { return m_node; } inline RN_NODE_PTR& GetNode() { return m_node; } /** * Function HitTest() * Tests if selected node is located within polygon boundaries. * @param aNode is a node to be checked. * @return True is the node is located within polygon boundaries. */ bool HitTest( const RN_NODE_PTR& aNode ) const; private: ///> Index of the outline in the parent polygon set int m_subpolygonIndex; ///> Bounding box of the polygon. BOX2I m_bbox; ///> Polygon set containing the geometry const SHAPE_POLY_SET* m_parentPolyset; ///> Node representing a polygon (it has the same coordinates as the first point of its ///> bounding polyline. RN_NODE_PTR m_node; friend bool sortArea( const RN_POLY& aP1, const RN_POLY& aP2 ); }; /** * Class RN_NET * Describes ratsnest for a single net. */ class RN_NET { public: ///> Default constructor. RN_NET() : m_dirty( true ), m_visible( true ) {} /** * Function SetVisible() * Sets state of the visibility flag. * @param aEnabled is new state. True if ratsnest for a given net is meant to be displayed, * false otherwise. */ void SetVisible( bool aEnabled ) { m_visible = aEnabled; } /** * Function IsVisible() * Returns the visibility flag state. * @return True if ratsnest for given net is set as visible, false otherwise, */ bool IsVisible() const { return m_visible; } /** * Function MarkDirty() * Marks ratsnest for given net as 'dirty', i.e. requiring recomputation. */ void MarkDirty() { m_dirty = true; } /** * Function IsDirty() * Returns state of the 'dirty' flag, indicating that ratsnest for a given net is invalid * and requires an update. * @return True if ratsnest requires recomputation, false otherwise. */ bool IsDirty() const { return m_dirty; } /** * Function GetUnconnected() * Returns pointer to a vector of edges that makes ratsnest for a given net. * @return Pointer to a vector of edges that makes ratsnest for a given net. */ const std::vector* GetUnconnected() const { return m_rnEdges.get(); } /** * Function Update() * Recomputes ratsnest for a net. */ void Update(); /** * Function AddItem() * Adds an appropriate node associated with selected pad, so it is * taken into account during ratsnest computations. * @param aPad is a pad for which node is added. */ void AddItem( const D_PAD* aPad ); /** * Function AddItem() * Adds an appropriate node associated with selected via, so it is * taken into account during ratsnest computations. * @param aVia is a via for which node is added. */ void AddItem( const VIA* aVia ); /** * Function AddItem() * Adds appropriate nodes and edges associated with selected track, so they are * taken into account during ratsnest computations. * @param aTrack is a track for which nodes and edges are added. */ void AddItem( const TRACK* aTrack ); /** * Function AddItem() * Processes zone to split it into subpolygons and adds appropriate nodes for them, so they are * taken into account during ratsnest computations. * @param aZone is a zone to be processed. */ void AddItem( const ZONE_CONTAINER* aZone ); /** * Function RemoveItem() * Removes all nodes and edges associated with selected pad, so they are not * taken into account during ratsnest computations anymore. * @param aPad is a pad for which nodes and edges are removed. */ void RemoveItem( const D_PAD* aPad ); /** * Function RemoveItem() * Removes all nodes and edges associated with selected via, so they are not * taken into account during ratsnest computations anymore. * @param aVia is a via for which nodes and edges are removed. */ void RemoveItem( const VIA* aVia ); /** * Function RemoveItem() * Removes all nodes and edges associated with selected track, so they are not * taken into account during ratsnest computations anymore. * @param aTrack is a track for which nodes and edges are removed. */ void RemoveItem( const TRACK* aTrack ); /** * Function RemoveItem() * Removes all nodes and edges associated with selected zone, so they are not * taken into account during ratsnest computations anymore. * @param aZone is a zone for which nodes and edges are removed. */ void RemoveItem( const ZONE_CONTAINER* aZone ); /** * Function GetNodes() * Returns list of nodes that are associated with a given item. * @param aItem is an item for which the list is generated. * @return List of associated nodes. */ std::list GetNodes( const BOARD_CONNECTED_ITEM* aItem ) const; /** * Function GetAllItems() * 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& aOutput, RN_ITEM_TYPE aType = RN_ALL ) const; /** * Function GetClosestNode() * Returns a single node that lies in the shortest distance from a specific node. * @param aNode is the node for which the closest node is searched. */ const RN_NODE_PTR GetClosestNode( const RN_NODE_PTR& aNode ) const; /** * Function GetClosestNode() * Returns a single node that lies in the shortest distance from a specific node and meets * selected filter criterion.. * @param aNode is the node for which the closest node is searched. * @param aFilter is a functor that filters nodes. */ const RN_NODE_PTR GetClosestNode( const RN_NODE_PTR& aNode, const RN_NODE_FILTER& aFilter ) const; /** * Function GetClosestNodes() * Returns list of nodes sorted by the distance from a specific node. * @param aNode is the node for which the closest nodes are searched. * @param aNumber is asked number of returned nodes. If it is negative then all nodes that * belong to the same net are returned. If asked number is greater than number of possible * nodes then the size of list is limited to number of possible nodes. */ std::list GetClosestNodes( const RN_NODE_PTR& aNode, int aNumber = -1 ) const; /** * Function GetClosestNodes() * Returns filtered list of nodes sorted by the distance from a specific node. * @param aNode is the node for which the closest nodes are searched. * @param aFilter is a functor that filters nodes. * @param aNumber is asked number of returned nodes. If it is negative then all nodes that * belong to the same net are returned. If asked number is greater than number of possible * nodes then the size of list is limited to number of possible nodes. */ std::list GetClosestNodes( const RN_NODE_PTR& aNode, const RN_NODE_FILTER& aFilter, int aNumber = -1 ) const; /** * Function AddSimple() * Changes drawing mode for an item to simple (i.e. one ratsnest line per node). * @param aNode is a node that changes its drawing mode. */ void AddSimple( const BOARD_CONNECTED_ITEM* aItem ); /** * Function AddBlockedNode() * Specifies a node as not suitable as a ratsnest line target (i.e. ratsnest lines will not * target the node). The status is cleared after calling ClearSimple(). * @param aNode is the node that is not going to be used as a ratsnest line target. */ inline void AddBlockedNode( RN_NODE_PTR& aNode ) { m_blockedNodes.insert( aNode ); aNode->SetFlag( true ); } /** * Function GetSimpleNodes() * Returns list of nodes for which ratsnest is drawn in simple mode (i.e. one * ratsnest line per node). * @return list of nodes for which ratsnest is drawn in simple mode. */ inline const boost::unordered_set& GetSimpleNodes() const { return m_simpleNodes; } /** * Function ClearSimple() * Removes all nodes and edges that are used for displaying ratsnest in simple mode. */ void ClearSimple(); /** * Function GetConnectedItems() * Adds items that are connected together to a list. * @param aItem is the reference item to find other connected items. * @param aOutput is the list that will contain found items. * @param aTypes allows to filter by item types. */ void GetConnectedItems( const BOARD_CONNECTED_ITEM* aItem, std::list& aOutput, RN_ITEM_TYPE aTypes = RN_ALL ) const; protected: ///> Validates edge, i.e. modifies source and target nodes for an edge ///> to make sure that they are not ones with the flag set. void validateEdge( RN_EDGE_MST_PTR& aEdge ); ///> Removes a link between a node and a parent, ///> and clears linked edges if it was the last parent. void removeNode( RN_NODE_PTR& aNode, const BOARD_CONNECTED_ITEM* aParent ); ///> Removes a link between an edge and a parent, ///> and clears its node data if it was the last parent. void removeEdge( RN_EDGE_MST_PTR& aEdge, const BOARD_CONNECTED_ITEM* aParent ); ///> Removes all ratsnest edges for a given node. void clearNode( const RN_NODE_PTR& aNode ); ///> Adds appropriate edges for nodes that are connected by zones. void processZones(); ///> Adds additional edges to account for connections made by items located in pads areas. void processPads(); ///> Recomputes ratsnset from scratch. void compute(); ////> Stores information about connections for a given net. RN_LINKS m_links; ///> Vector of edges that makes ratsnest for a given net. boost::shared_ptr< std::vector > m_rnEdges; ///> List of nodes which will not be used as ratsnest target nodes. boost::unordered_set m_blockedNodes; ///> Nodes to be displayed using the simplified ratsnest algorithm. boost::unordered_set m_simpleNodes; ///> Flag indicating necessity of recalculation of ratsnest for a net. bool m_dirty; ///> Structure to hold ratsnest data for ZONE_CONTAINER objects. typedef struct { ///> Subpolygons belonging to a zone std::deque m_Polygons; ///> Connections to other nodes std::deque m_Edges; } RN_ZONE_DATA; ///> Structureo to hold ratsnest data for D_PAD objects. typedef struct { ///> Node representing the pad. RN_NODE_PTR m_Node; ///> Helper nodes that make for connections to items located in the pad area. std::deque m_Edges; } RN_PAD_DATA; ///> Helper typedefs typedef boost::unordered_map PAD_NODE_MAP; typedef boost::unordered_map VIA_NODE_MAP; typedef boost::unordered_map TRACK_EDGE_MAP; typedef boost::unordered_map ZONE_DATA_MAP; ///> Map that associates nodes in the ratsnest model to respective nodes. PAD_NODE_MAP m_pads; ///> Map that associates nodes in the ratsnest model to respective vias. VIA_NODE_MAP m_vias; ///> Map that associates edges in the ratsnest model to respective tracks. TRACK_EDGE_MAP m_tracks; ///> Map that associates groups of subpolygons in the ratsnest model to respective zones. ZONE_DATA_MAP m_zones; ///> Visibility flag. bool m_visible; }; /** * Class RN_DATA * * Stores information about unconnected items for a board. */ class RN_DATA { public: /** * Default constructor * @param aBoard is the board to be processed in order to look for unconnected items. */ RN_DATA( const BOARD* aBoard ) : m_board( aBoard ) {} /** * Function Add() * Adds an item to the ratsnest data. * @param aItem is an item to be added. */ void Add( const BOARD_ITEM* aItem ); /** * Function Remove() * Removes an item from the ratsnest data. * @param aItem is an item to be updated. */ void Remove( const BOARD_ITEM* aItem ); /** * Function Update() * Updates the ratsnest data for an item. * @param aItem is an item to be updated. */ void Update( const BOARD_ITEM* aItem ); /** * Function AddSimple() * Sets an item to be drawn in simple mode (i.e. one line per node, instead of full ratsnest). * It is used for drawing quick, temporary ratsnest, eg. while moving an item. * @param aItem is an item to be drawn in simple node. */ void AddSimple( const BOARD_ITEM* aItem ); /** * Function AddBlocked() * Specifies an item as not suitable as a ratsnest line target (i.e. ratsnest lines will not * target its node(s)). The status is cleared after calling ClearSimple(). * @param aItem is the item of which node(s) are not going to be used as a ratsnest line target. */ void AddBlocked( const BOARD_ITEM* aItem ); /** * Function ClearSimple() * Clears the list of nodes for which ratsnest is drawn in simple mode (one line per node). */ void ClearSimple() { BOOST_FOREACH( RN_NET& net, m_nets ) net.ClearSimple(); } /** * Function ProcessBoard() * Prepares data for computing (computes a list of current nodes and connections). It is * required to run only once after loading a board. */ void ProcessBoard(); /** * Function Recalculate() * Recomputes ratsnest for selected net number or all nets that need updating. * @param aNet is a net number. If it is negative, all nets that need updating are recomputed. */ void Recalculate( int aNet = -1 ); /** * Function GetNetCount() * Returns the number of nets handled by the ratsnest. * @return Number of the nets. */ int GetNetCount() const { return m_nets.size(); } /** * Function GetNet() * Returns ratsnest grouped by net numbers. * @param aNetCode is the net code. * @return Ratsnest data for a specified net. */ RN_NET& GetNet( int aNetCode ) { assert( aNetCode > 0 ); // ratsnest does not handle the unconnected net return m_nets[aNetCode]; } /** * Function GetConnectedItems() * Adds items that are connected together to a list. * @param aItem is the reference item to find other connected items. * @param aOutput is the list that will contain found items. * @param aTypes allows to filter by item types. */ void GetConnectedItems( const BOARD_CONNECTED_ITEM* aItem, std::list& aOutput, 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& aOutput, RN_ITEM_TYPE aTypes = RN_ALL ) const; /** * Function AreConnected() * Checks if two items are connected with copper. * @param aThis is the first item. * @param aOther is the second item. * @return True if they are connected, false otherwise. */ bool AreConnected( const BOARD_CONNECTED_ITEM* aItem, const BOARD_CONNECTED_ITEM* aOther ); /** * Function GetUnconnectedCount() * Returns the number of missing connections. * @return Number of missing connections. */ int GetUnconnectedCount() const; protected: /** * Function updateNet() * Recomputes ratsnest for a single net. * @param aNetCode is the net number to be recomputed. */ void updateNet( int aNetCode ); ///> Board to be processed. const BOARD* m_board; ///> Stores information about ratsnest grouped by net numbers. std::vector m_nets; }; #endif /* RATSNEST_DATA_H */