kicad/pcbnew/ratsnest/ratsnest_data.cpp

470 lines
12 KiB
C++
Raw Normal View History

2013-11-25 15:50:03 +00:00
/*
* This program source code file is part of KICAD, a free EDA CAD application.
*
2017-03-22 13:43:10 +00:00
* Copyright (C) 2013-2017 CERN
2021-11-30 14:19:39 +00:00
* Copyright (C) 2019-2021 KiCad Developers, see AUTHORS.txt for contributors.
*
2013-11-25 15:50:03 +00:00
* @author Maciej Suminski <maciej.suminski@cern.ch>
2017-03-22 13:43:10 +00:00
* @author Tomasz Wlostowski <tomasz.wlostowski@cern.ch>
2013-11-25 15:50:03 +00:00
*
* 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.cpp
* @brief Class that computes missing connections on a PCB.
*/
2017-03-22 13:43:10 +00:00
#ifdef PROFILE
#include <profile.h>
#endif
2013-11-25 15:50:03 +00:00
#include <ratsnest/ratsnest_data.h>
#include <functional>
using namespace std::placeholders;
2013-11-25 15:50:03 +00:00
#include <algorithm>
#include <cassert>
2013-11-25 15:50:03 +00:00
#include <limits>
#include <delaunator.hpp>
2013-11-25 15:50:03 +00:00
class disjoint_set
{
2017-06-23 11:56:28 +00:00
public:
disjoint_set( size_t size )
{
m_data.resize( size );
m_depth.resize( size, 0 );
for( size_t i = 0; i < size; i++ )
m_data[i] = i;
}
int find( int aVal )
{
int root = aVal;
2013-11-25 15:50:03 +00:00
while( m_data[root] != root )
root = m_data[root];
2013-11-25 15:50:03 +00:00
// Compress the path
while( m_data[aVal] != aVal )
{
auto& tmp = m_data[aVal];
aVal = tmp;
tmp = root;
}
return root;
}
2013-11-25 15:50:03 +00:00
bool unite( int aVal1, int aVal2 )
2013-11-25 15:50:03 +00:00
{
aVal1 = find( aVal1 );
aVal2 = find( aVal2 );
2013-11-25 15:50:03 +00:00
if( aVal1 != aVal2 )
2013-11-25 15:50:03 +00:00
{
if( m_depth[aVal1] < m_depth[aVal2] )
2014-03-05 13:57:14 +00:00
{
m_data[aVal1] = aVal2;
2013-11-25 15:50:03 +00:00
}
else
{
m_data[aVal2] = aVal1;
if( m_depth[aVal1] == m_depth[aVal2] )
m_depth[aVal1]++;
}
return true;
2013-11-25 15:50:03 +00:00
}
return false;
2013-11-25 15:50:03 +00:00
}
private:
std::vector<int> m_data;
std::vector<int> m_depth;
};
2021-11-30 14:19:39 +00:00
void RN_NET::kruskalMST( const std::vector<CN_EDGE> &aEdges )
{
disjoint_set dset( m_nodes.size() );
m_rnEdges.clear();
2020-06-23 02:18:46 +00:00
int i = 0;
2021-11-30 14:19:39 +00:00
for( const CN_ANCHOR_PTR& node : m_nodes )
2020-06-23 02:18:46 +00:00
node->SetTag( i++ );
2021-11-30 14:19:39 +00:00
for( const CN_EDGE& tmp : aEdges )
{
int u = tmp.GetSourceNode()->GetTag();
int v = tmp.GetTargetNode()->GetTag();
if( dset.unite( u, v ) )
{
if( tmp.GetWeight() > 0 )
m_rnEdges.push_back( tmp );
}
}
2013-11-25 15:50:03 +00:00
}
2017-06-23 11:56:28 +00:00
2017-03-22 13:43:10 +00:00
class RN_NET::TRIANGULATOR_STATE
2013-11-25 15:50:03 +00:00
{
2017-03-22 13:43:10 +00:00
private:
std::multiset<CN_ANCHOR_PTR, CN_PTR_CMP> m_allNodes;
// Checks if all nodes in aNodes lie on a single line. Requires the nodes to
// have unique coordinates!
bool areNodesColinear( const std::vector<CN_ANCHOR_PTR>& aNodes ) const
{
if ( aNodes.size() <= 2 )
return true;
const VECTOR2I p0( aNodes[0]->Pos() );
const VECTOR2I v0( aNodes[1]->Pos() - p0 );
for( unsigned i = 2; i < aNodes.size(); i++ )
{
const VECTOR2I v1 = aNodes[i]->Pos() - p0;
if( v0.Cross( v1 ) != 0 )
return false;
}
return true;
}
2017-03-22 13:43:10 +00:00
public:
void Clear()
2013-11-25 15:50:03 +00:00
{
2017-03-22 13:43:10 +00:00
m_allNodes.clear();
}
2017-03-22 13:43:10 +00:00
void AddNode( CN_ANCHOR_PTR aNode )
{
m_allNodes.insert( aNode );
2017-03-22 13:43:10 +00:00
}
void Triangulate( std::vector<CN_EDGE>& mstEdges)
2017-03-22 13:43:10 +00:00
{
2021-11-30 14:19:39 +00:00
std::vector<double> node_pts;
std::vector<CN_ANCHOR_PTR> anchors;
std::vector< std::vector<CN_ANCHOR_PTR> > anchorChains( m_allNodes.size() );
node_pts.reserve( 2 * m_allNodes.size() );
anchors.reserve( m_allNodes.size() );
CN_ANCHOR_PTR prev = nullptr;
2013-11-25 15:50:03 +00:00
2021-11-30 14:19:39 +00:00
for( const CN_ANCHOR_PTR& n : m_allNodes )
2017-03-22 13:43:10 +00:00
{
if( !prev || prev->Pos() != n->Pos() )
{
node_pts.push_back( n->Pos().x );
node_pts.push_back( n->Pos().y );
anchors.push_back( n );
prev = n;
2017-03-22 13:43:10 +00:00
}
anchorChains[anchors.size() - 1].push_back( n );
2017-03-22 13:43:10 +00:00
}
if( anchors.size() < 2 )
2017-03-22 13:43:10 +00:00
{
return;
2017-03-22 13:43:10 +00:00
}
else if( areNodesColinear( anchors ) )
{
// special case: all nodes are on the same line - there's no
// triangulation for such set. In this case, we sort along any coordinate
// and chain the nodes together.
for( size_t i = 0; i < anchors.size() - 1; i++ )
{
2021-11-30 14:19:39 +00:00
const CN_ANCHOR_PTR& src = anchors[i];
const CN_ANCHOR_PTR& dst = anchors[i + 1];
mstEdges.emplace_back( src, dst, src->Dist( *dst ) );
}
2017-03-22 13:43:10 +00:00
}
else
{
delaunator::Delaunator delaunator( node_pts );
auto& triangles = delaunator.triangles;
for( size_t i = 0; i < triangles.size(); i += 3 )
{
CN_ANCHOR_PTR src = anchors[triangles[i]];
CN_ANCHOR_PTR dst = anchors[triangles[i + 1]];
mstEdges.emplace_back( src, dst, src->Dist( *dst ) );
src = anchors[triangles[i + 1]];
dst = anchors[triangles[i + 2]];
mstEdges.emplace_back( src, dst, src->Dist( *dst ) );
src = anchors[triangles[i + 2]];
dst = anchors[triangles[i]];
mstEdges.emplace_back( src, dst, src->Dist( *dst ) );
}
for( size_t i = 0; i < delaunator.halfedges.size(); i++ )
{
if( delaunator.halfedges[i] == delaunator::INVALID_INDEX )
continue;
2021-11-30 14:19:39 +00:00
const CN_ANCHOR_PTR& src = anchors[triangles[i]];
const CN_ANCHOR_PTR& dst = anchors[triangles[delaunator.halfedges[i]]];
mstEdges.emplace_back( src, dst, src->Dist( *dst ) );
}
}
2013-11-25 15:50:03 +00:00
for( size_t i = 0; i < anchorChains.size(); i++ )
2017-03-22 13:43:10 +00:00
{
2021-11-30 14:19:39 +00:00
std::vector<CN_ANCHOR_PTR>& chain = anchorChains[i];
2013-11-25 15:50:03 +00:00
2017-03-22 13:43:10 +00:00
if( chain.size() < 2 )
continue;
2013-11-25 15:50:03 +00:00
2017-03-22 13:43:10 +00:00
std::sort( chain.begin(), chain.end(),
2021-11-30 14:19:39 +00:00
[] ( const CN_ANCHOR_PTR& a, const CN_ANCHOR_PTR& b )
{
return a->GetCluster().get() < b->GetCluster().get();
} );
2013-11-25 15:50:03 +00:00
2017-06-23 17:22:44 +00:00
for( unsigned int j = 1; j < chain.size(); j++ )
2017-03-22 13:43:10 +00:00
{
2021-11-30 14:19:39 +00:00
const CN_ANCHOR_PTR& prevNode = chain[j - 1];
const CN_ANCHOR_PTR& curNode = chain[j];
2017-03-22 13:43:10 +00:00
int weight = prevNode->GetCluster() != curNode->GetCluster() ? 1 : 0;
mstEdges.emplace_back( prevNode, curNode, weight );
2017-03-22 13:43:10 +00:00
}
}
}
2017-03-22 13:43:10 +00:00
};
2013-11-25 15:50:03 +00:00
RN_NET::RN_NET() : m_dirty( true )
2013-11-25 15:50:03 +00:00
{
2017-03-22 13:43:10 +00:00
m_triangulator.reset( new TRIANGULATOR_STATE );
2013-11-25 15:50:03 +00:00
}
2017-06-23 11:56:28 +00:00
2013-11-25 15:50:03 +00:00
void RN_NET::compute()
{
// Special cases do not need complicated algorithms (actually, it does not work well with
// the Delaunay triangulator)
2017-03-22 13:43:10 +00:00
if( m_nodes.size() <= 2 )
2013-11-25 15:50:03 +00:00
{
2017-03-22 13:43:10 +00:00
m_rnEdges.clear();
2013-11-25 15:50:03 +00:00
// Check if the only possible connection exists
2017-03-22 13:43:10 +00:00
if( m_boardEdges.size() == 0 && m_nodes.size() == 2 )
2013-11-25 15:50:03 +00:00
{
2017-03-22 13:43:10 +00:00
auto last = ++m_nodes.begin();
2013-11-25 15:50:03 +00:00
// There can be only one possible connection, but it is missing
2020-06-23 02:18:46 +00:00
CN_EDGE edge ( *m_nodes.begin(), *last );
2017-03-22 13:43:10 +00:00
edge.GetSourceNode()->SetTag( 0 );
edge.GetTargetNode()->SetTag( 1 );
m_rnEdges.push_back( edge );
}
else
{
2017-03-22 13:43:10 +00:00
// Set tags to m_nodes as connected
2021-11-30 14:19:39 +00:00
for( const CN_ANCHOR_PTR& node : m_nodes )
node->SetTag( 0 );
2013-11-25 15:50:03 +00:00
}
return;
}
2017-03-22 13:43:10 +00:00
m_triangulator->Clear();
2013-11-25 15:50:03 +00:00
2021-11-30 14:19:39 +00:00
for( const CN_ANCHOR_PTR& n : m_nodes )
2017-03-22 13:43:10 +00:00
m_triangulator->AddNode( n );
2013-11-25 15:50:03 +00:00
std::vector<CN_EDGE> triangEdges;
triangEdges.reserve( m_nodes.size() + m_boardEdges.size() );
2021-11-30 14:19:39 +00:00
#ifdef PROFILE
2017-03-22 13:43:10 +00:00
PROF_COUNTER cnt("triangulate");
2021-11-30 14:19:39 +00:00
#endif
m_triangulator->Triangulate( triangEdges );
2021-11-30 14:19:39 +00:00
#ifdef PROFILE
2017-03-22 13:43:10 +00:00
cnt.Show();
2021-11-30 14:19:39 +00:00
#endif
2021-11-30 14:19:39 +00:00
for( const CN_EDGE& e : m_boardEdges )
triangEdges.emplace_back( e );
std::sort( triangEdges.begin(), triangEdges.end() );
2013-11-25 15:50:03 +00:00
2017-03-22 13:43:10 +00:00
// Get the minimal spanning tree
#ifdef PROFILE
PROF_COUNTER cnt2("mst");
#endif
kruskalMST( triangEdges );
2017-03-22 13:43:10 +00:00
#ifdef PROFILE
cnt2.Show();
#endif
2013-11-25 15:50:03 +00:00
}
void RN_NET::Update()
{
compute();
m_dirty = false;
}
2017-03-22 13:43:10 +00:00
void RN_NET::Clear()
2013-11-25 15:50:03 +00:00
{
2017-03-22 13:43:10 +00:00
m_rnEdges.clear();
m_boardEdges.clear();
m_nodes.clear();
2013-11-25 15:50:03 +00:00
m_dirty = true;
}
2017-03-22 13:43:10 +00:00
void RN_NET::AddCluster( CN_CLUSTER_PTR aCluster )
{
2017-03-22 13:43:10 +00:00
CN_ANCHOR_PTR firstAnchor;
2021-11-30 14:19:39 +00:00
for( CN_ITEM* item : *aCluster )
2013-11-25 15:50:03 +00:00
{
2021-11-30 14:19:39 +00:00
bool isZone = dynamic_cast<CN_ZONE_LAYER*>( item );
std::vector<CN_ANCHOR_PTR>& anchors = item->Anchors();
unsigned int nAnchors = isZone ? 1 : anchors.size();
2013-11-25 15:50:03 +00:00
2017-06-23 11:56:28 +00:00
if( nAnchors > anchors.size() )
2017-03-22 13:43:10 +00:00
nAnchors = anchors.size();
2013-11-25 15:50:03 +00:00
2017-06-23 17:22:44 +00:00
for( unsigned int i = 0; i < nAnchors; i++ )
2013-11-25 15:50:03 +00:00
{
2017-03-22 13:43:10 +00:00
anchors[i]->SetCluster( aCluster );
2020-06-23 02:18:46 +00:00
m_nodes.insert( anchors[i] );
2013-11-25 15:50:03 +00:00
2017-03-22 13:43:10 +00:00
if( firstAnchor )
2013-11-25 15:50:03 +00:00
{
2017-03-22 13:43:10 +00:00
if( firstAnchor != anchors[i] )
m_boardEdges.emplace_back( firstAnchor, anchors[i], 0 );
2013-11-25 15:50:03 +00:00
}
2017-03-22 13:43:10 +00:00
else
{
2017-03-22 13:43:10 +00:00
firstAnchor = anchors[i];
}
}
}
}
2017-06-23 11:56:28 +00:00
2017-03-22 13:43:10 +00:00
bool RN_NET::NearestBicoloredPair( const RN_NET& aOtherNet, CN_ANCHOR_PTR& aNode1,
2021-11-30 14:19:39 +00:00
CN_ANCHOR_PTR& aNode2 ) const
2013-11-25 15:50:03 +00:00
{
2017-03-22 13:43:10 +00:00
bool rv = false;
2013-11-25 15:50:03 +00:00
2021-11-30 14:19:39 +00:00
SEG::ecoord distMax_sq = VECTOR2I::ECOORD_MAX;
2021-11-30 14:19:39 +00:00
auto verify =
[&]( const std::shared_ptr<CN_ANCHOR>& aTestNode1,
const std::shared_ptr<CN_ANCHOR>& aTestNode2 )
{
2021-11-30 14:19:39 +00:00
VECTOR2I diff = aTestNode1->Pos() - aTestNode2->Pos();
SEG::ecoord dist_sq = diff.SquaredEuclideanNorm();
if( dist_sq < distMax_sq )
{
rv = true;
distMax_sq = dist_sq;
aNode1 = aTestNode1;
aNode2 = aTestNode2;
}
};
/// Sweep-line algorithm to cut the number of comparisons to find the closest point
///
/// Step 1: The outer loop needs to be the subset (selected nodes) as it is a linear search
2021-11-30 14:19:39 +00:00
for( const std::shared_ptr<CN_ANCHOR>& nodeA : aOtherNet.m_nodes )
{
if( nodeA->GetNoLine() )
continue;
/// Step 2: O( log n ) search to identify a close element ordered by x
/// The fwd_it iterator will move forward through the elements while
/// the rev_it iterator will move backward through the same set
auto fwd_it = m_nodes.lower_bound( nodeA );
auto rev_it = std::make_reverse_iterator( fwd_it );
for( ; fwd_it != m_nodes.end(); ++fwd_it )
{
2021-01-08 00:26:32 +00:00
const std::shared_ptr<CN_ANCHOR>& nodeB = *fwd_it;
if( nodeB->GetNoLine() )
continue;
2021-11-30 14:19:39 +00:00
SEG::ecoord distX_sq = SEG::Square( nodeA->Pos().x - nodeB->Pos().x );
/// As soon as the x distance (primary sort) is larger than the smallest distance,
/// stop checking further elements
2021-11-30 14:19:39 +00:00
if( distX_sq > distMax_sq )
break;
verify( nodeA, nodeB );
}
/// Step 3: using the same starting point, check points backwards for closer points
for( ; rev_it != m_nodes.rend(); ++rev_it )
2013-11-25 15:50:03 +00:00
{
2021-01-08 00:26:32 +00:00
const std::shared_ptr<CN_ANCHOR>& nodeB = *rev_it;
if( nodeB->GetNoLine() )
continue;
2015-06-04 12:54:07 +00:00
2021-11-30 14:19:39 +00:00
SEG::ecoord distX_sq = SEG::Square( nodeA->Pos().x - nodeB->Pos().x );
2021-11-30 14:19:39 +00:00
if( distX_sq > distMax_sq )
break;
verify( nodeA, nodeB );
}
2013-11-25 15:50:03 +00:00
}
2017-03-22 13:43:10 +00:00
return rv;
2013-11-25 15:50:03 +00:00
}
void RN_NET::SetVisible( bool aEnabled )
{
2021-11-30 14:19:39 +00:00
for( CN_EDGE& edge : m_rnEdges )
2017-06-23 11:56:28 +00:00
edge.SetVisible( aEnabled );
}