pns_component_dragger.cpp

Go to the documentation of this file.

/*
 * KiRouter - a push-and-(sometimes-)shove PCB router
 *
 * Copyright (C) 2013-2020 CERN
 * Copyright (C) 2021 KiCad Developers, see AUTHORS.txt for contributors.
 * Author: Tomasz Wlostowski <[email protected]>
 *
 * 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 3 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 <http://www.gnu.org/licenses/>.
 */
 
#include <memory>
 
#include "pns_arc.h"
#include "pns_line.h"
#include "pns_solid.h"
#include "pns_router.h"
 
#include "pns_component_dragger.h"
#include "pns_debug_decorator.h"
 
namespace PNS
{
 
COMPONENT_DRAGGER::COMPONENT_DRAGGER( ROUTER* aRouter ) : DRAG_ALGO( aRouter )
{
 // ensure all variables are initialized
 m_dragStatus = false;
 m_currentNode = nullptr;
}
 
 
COMPONENT_DRAGGER::~COMPONENT_DRAGGER()
{
}
 
 
bool COMPONENT_DRAGGER::Start( const VECTOR2I& aP, ITEM_SET& aPrimitives )
{
 assert( m_world );
 
 m_currentNode = nullptr;
 m_initialDraggedItems = aPrimitives;
 m_p0 = aP;
 
 std::unordered_set<LINKED_ITEM*> seenItems;
 
 auto addLinked =
 [&]( SOLID* aSolid, JOINT* aJoint, LINKED_ITEM* aItem, VECTOR2I aOffset = {} )
 {
 if( seenItems.count( aItem ) )
 return;
 
 seenItems.insert( aItem );
 
 // Segments that go directly between two linked pads are special-cased
 VECTOR2I otherEnd = ( aJoint->Pos() == aItem->Anchor( 0 ) ) ?
 aItem->Anchor( 1 ) : aItem->Anchor( 0 );
 JOINT* otherJoint = m_world->FindJoint( otherEnd, aItem->Layer(), aItem->Net() );
 
 if( otherJoint && otherJoint->LinkCount( ITEM::SOLID_T ) )
 {
 for( const ITEM_SET::ENTRY& otherItem : otherJoint->LinkList() )
 {
 if( aPrimitives.Contains( otherItem.item ) )
 {
 m_fixedItems.insert( aItem );
 return;
 }
 }
 }
 
 int segIndex;
 DRAGGED_CONNECTION cn;
 
 cn.origLine = m_world->AssembleLine( aItem, &segIndex );
 cn.attachedPad = aSolid;
 cn.offset = aOffset;
 
 // Lines that go directly between two linked pads are also special-cased
 const SHAPE_LINE_CHAIN& line = cn.origLine.CLine();
 JOINT* jA = m_world->FindJoint( line.CPoint( 0 ), aItem->Layer(), aItem->Net() );
 JOINT* jB = m_world->FindJoint( line.CPoint( -1 ), aItem->Layer(), aItem->Net() );
 
 wxASSERT( jA == aJoint || jB == aJoint );
 JOINT* jSearch = ( jA == aJoint ) ? jB : jA;
 
 if( jSearch && jSearch->LinkCount( ITEM::SOLID_T ) )
 {
 for( const ITEM_SET::ENTRY& otherItem : jSearch->LinkList() )
 {
 if( aPrimitives.Contains( otherItem.item ) )
 {
 for( ITEM* item : cn.origLine.Links() )
 m_fixedItems.insert( item );
 
 return;
 }
 }
 }
 
 m_conns.push_back( cn );
 };
 
 for( auto item : aPrimitives.Items() )
 {
 if( item.item->Kind() != ITEM::SOLID_T )
 continue;
 
 SOLID* solid = static_cast<SOLID*>( item.item );
 
 m_solids.insert( solid );
 
 if( !item.item->IsRoutable() )
 continue;
 
 JOINT* jt = m_world->FindJoint( solid->Pos(), solid );
 
 for( auto link : jt->LinkList() )
 {
 if( link.item->OfKind( ITEM::SEGMENT_T | ITEM::ARC_T ) )
 addLinked( solid, jt, static_cast<LINKED_ITEM*>( link.item ) );
 }
 
 std::vector<JOINT*> extraJoints;
 
 m_world->QueryJoints( solid->Hull().BBox(), extraJoints, solid->Layers(),
 ITEM::SEGMENT_T | ITEM::ARC_T );
 
 for( JOINT* extraJoint : extraJoints )
 {
 if( extraJoint->Net() == jt->Net() && extraJoint->LinkCount() == 1 )
 {
 LINKED_ITEM* li = static_cast<LINKED_ITEM*>( extraJoint->LinkList()[0].item );
 
 if( li->Collide( solid, nullptr, m_world ) )
 addLinked( solid, extraJoint, li, extraJoint->Pos() - solid->Pos() );
 }
 }
 }
 
 return true;
}
 
 
bool COMPONENT_DRAGGER::Drag( const VECTOR2I& aP )
{
 assert( m_world );
 
 m_world->KillChildren();
 m_currentNode = m_world->Branch();
 
 for( const ITEM_SET::ENTRY& item : m_initialDraggedItems.Items() )
 m_currentNode->Remove( item );
 
 m_draggedItems.Clear();
 
 for( SOLID* s : m_solids )
 {
 VECTOR2I p_next = aP - m_p0 + s->Pos();
 std::unique_ptr<SOLID> snew( static_cast<SOLID*>( s->Clone() ) );
 snew->SetPos( p_next );
 
 m_draggedItems.Add( snew.get() );
 m_currentNode->Add( std::move( snew ) );
 
 if( !s->IsRoutable() )
 continue;
 
 for( DRAGGED_CONNECTION& l : m_conns )
 {
 if( l.attachedPad == s )
 {
 l.p_orig = s->Pos() + l.offset;
 l.p_next = p_next + l.offset;
 }
 }
 }
 
 for( ITEM* item : m_fixedItems )
 {
 m_currentNode->Remove( item );
 
 switch( item->Kind() )
 {
 case ITEM::SEGMENT_T:
 {
 SEGMENT* s = static_cast<SEGMENT*>( item );
 std::unique_ptr<SEGMENT> s_new( s->Clone() );
 
 SEG orig = s->Seg();
 s_new->SetEnds( aP - m_p0 + orig.A, aP - m_p0 + orig.B );
 
 m_draggedItems.Add( s_new.get() );
 m_currentNode->Add( std::move( s_new ) );
 
 break;
 }
 
 case ITEM::ARC_T:
 {
 ARC* a = static_cast<ARC*>( item );
 std::unique_ptr<ARC> a_new( a->Clone() );
 
 SHAPE_ARC& arc = a_new->Arc();
 arc.Move( aP - m_p0 );
 
 m_draggedItems.Add( a_new.get() );
 m_currentNode->Add( std::move( a_new ) );
 break;
 }
 
 default:
 wxFAIL_MSG( wxT( "Unexpected item type in COMPONENT_DRAGGER::m_fixedItems" ) );
 }
 }
 
 for( COMPONENT_DRAGGER::DRAGGED_CONNECTION& cn : m_conns )
 {
 LINE l_new( cn.origLine );
 l_new.Unmark();
 l_new.ClearLinks();
 l_new.DragCorner( cn.p_next, cn.origLine.CLine().Find( cn.p_orig ) );
 
 PNS_DBG( Dbg(), AddItem, &l_new, BLUE, 0, wxT( "cdrag-new-fanout" ) );
 m_draggedItems.Add( l_new );
 
 LINE l_orig( cn.origLine );
 m_currentNode->Remove( l_orig );
 m_currentNode->Add( l_new );
 }
 
 return true;
}
 
 
bool COMPONENT_DRAGGER::FixRoute()
{
 NODE* node = CurrentNode();
 
 if( node )
 {
 if( Settings().AllowDRCViolations() || !node->CheckColliding( m_draggedItems ) )
 {
 Router()->CommitRouting( node );
 return true;
 }
 }
 
 return false;
}
 
 
NODE* COMPONENT_DRAGGER::CurrentNode() const
{
 return m_currentNode ? m_currentNode : m_world;
}
 
 
const ITEM_SET COMPONENT_DRAGGER::Traces()
{
 return m_draggedItems;
}
 
}; // namespace PNS