junction_helpers.cpp

Go to the documentation of this file.

/*
 * This program source code file is part of KiCad, a free EDA CAD application.
 *
 * Copyright The 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 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 <https://www.gnu.org/licenses/>.
 */
 
#include "junction_helpers.h"
 
#include <sch_line.h>
#include <sch_screen.h>
#include <sch_junction.h>
#include <sch_item.h>
#include <trigo.h>
 
#include <unordered_set>
 
using namespace JUNCTION_HELPERS;
 
POINT_INFO JUNCTION_HELPERS::AnalyzePoint( const EE_RTREE& aItems, const VECTOR2I& aPosition,
                                           bool aBreakCrossings )
{
    enum layers
    {
        WIRES = 0,
        BUSES
    };
 
    POINT_INFO info{};
    info.hasBusEntry = false;
    info.hasExplicitJunctionDot = false;
    info.isJunction = false;
    info.hasBusEntryToMultipleWires = false;
    info.hasBusAtPoint = false;
 
    bool                         breakLines[2] = { false };
    std::unordered_set<int>      exitAngles[2];
    std::vector<const SCH_LINE*> midPointLines[2];
 
    EE_RTREE filtered;
    std::list<std::unique_ptr<SCH_LINE>> mergedLines;
 
    // Ignore items that are currently being moved or flagged to skip
    // and temporarily merge collinear wires before analyzing the point.
    for( SCH_ITEM* item : aItems.Overlapping( aPosition ) )
    {
        if( item->GetEditFlags() & ( SKIP_STRUCT | STRUCT_DELETED ) )
            continue;
 
        switch( item->Type() )
        {
        case  SCH_LINE_T:
        {
            SCH_LINE* line = static_cast<SCH_LINE*>( item );
 
            if( line->IsConnectable() )
                mergedLines.emplace_back( new SCH_LINE( *line ) );
 
            break;
        }
 
        case SCH_JUNCTION_T:
            if( item->HitTest( aPosition, -1 ) )
                info.hasExplicitJunctionDot = true;
 
            filtered.insert( item );
            break;
 
        case SCH_BUS_WIRE_ENTRY_T:
            info.hasBusEntry = true;
            filtered.insert( item );
            break;
 
        case SCH_SHEET_T:
        case SCH_SYMBOL_T:
        case SCH_LABEL_T:
        case SCH_HIER_LABEL_T:
        case SCH_GLOBAL_LABEL_T:
            filtered.insert( item );
            break;
 
        default:
            break;
        }
    }
 
    if( mergedLines.size() + filtered.size() < 2 )
        return info;
 
    // Skip collinear merging when enough distinct endpoints already meet here.
    // Merging would turn an N-way stub junction (e.g. LTspice four-wire cross:
    // two opposite stubs on each axis) into an unmarked mid-segment crossing
    // and hide the needed junction.
    std::unordered_set<int> preMergeWireExits;
    std::unordered_set<int> preMergeBusExits;
 
    for( const auto& line : mergedLines )
    {
        if( line->GetStartPoint() == line->GetEndPoint() )
            continue;
 
        if( !line->IsConnected( aPosition ) )
            continue;
 
        if( line->GetLayer() == LAYER_WIRE )
            preMergeWireExits.insert( line->GetAngleFrom( aPosition ) );
        else if( line->GetLayer() == LAYER_BUS )
            preMergeBusExits.insert( line->GetAngleFrom( aPosition ) );
    }
 
    const bool keepStubJunction = preMergeWireExits.size() >= 3 || preMergeBusExits.size() >= 3;
 
    // Merge collinear wire segments
    bool merged = false;
 
    do
    {
        if( info.hasExplicitJunctionDot || aBreakCrossings || keepStubJunction )
            break;
 
        merged = false;
 
        for( auto it_i = mergedLines.begin(); it_i != mergedLines.end() && !merged; ++it_i )
        {
            for( auto it_j = std::next( it_i ); it_j != mergedLines.end(); ++it_j )
            {
                if( auto* line = ( *it_i )->MergeOverlap( nullptr, it_j->get(), false ) )
                {
                    it_i->reset( line );
                    mergedLines.erase( it_j );
                    merged = true;
                    break;
                }
            }
        }
    } while( merged );
 
    for( const auto& line : mergedLines )
        filtered.insert( line.get() );
 
 
    // A pin at 90° still shouldn't match a line at 90° so just give pins unique numbers
    int uniqueAngle = 10000;
 
    for( const SCH_ITEM* item : filtered )
    {
        if( item->GetEditFlags() & STRUCT_DELETED )
            continue;
 
        switch( item->Type() )
        {
        case SCH_JUNCTION_T:
            if( item->HitTest( aPosition, -1 ) )
                info.hasExplicitJunctionDot = true;
 
            break;
 
        case SCH_LINE_T:
        {
            const SCH_LINE* line = static_cast<const SCH_LINE*>( item );
            int             layer;
 
            if( line->GetStartPoint() == line->GetEndPoint() )
                break;
            else if( line->GetLayer() == LAYER_WIRE )
                layer = WIRES;
            else if( line->GetLayer() == LAYER_BUS )
                layer = BUSES;
            else
                break;
 
            if( line->IsConnected( aPosition ) )
            {
                breakLines[layer] = true;
                exitAngles[layer].insert( line->GetAngleFrom( aPosition ) );
            }
            else if( line->HitTest( aPosition, -1 ) )
            {
                if( aBreakCrossings )
                    breakLines[layer] = true;
 
                // Defer any line midpoints until we know whether or not we're breaking them
                midPointLines[layer].push_back( line );
            }
 
            if( layer == BUSES && line->HitTest( aPosition, -1 ) )
                info.hasBusAtPoint = true;
        }
        break;
 
        case SCH_BUS_WIRE_ENTRY_T:
            if( item->IsConnected( aPosition ) )
            {
                breakLines[BUSES] = true;
                exitAngles[BUSES].insert( uniqueAngle++ );
                breakLines[WIRES] = true;
                exitAngles[WIRES].insert( uniqueAngle++ );
                info.hasBusEntry = true;
            }
 
            break;
 
        case SCH_SYMBOL_T:
        case SCH_SHEET_T:
            if( item->IsConnected( aPosition ) )
            {
                breakLines[WIRES] = true;
                exitAngles[WIRES].insert( uniqueAngle++ );
            }
 
            break;
 
        case SCH_LABEL_T:
            if( item->IsConnected( aPosition ) )
            {
                if( SCH_CONNECTION::IsBusLabel( static_cast<const SCH_LABEL*>( item )->GetText() ) )
                    breakLines[BUSES] = true;
                else
                    breakLines[WIRES] = true;
            }
 
            break;
 
        case SCH_HIER_LABEL_T:
        case SCH_GLOBAL_LABEL_T:
            if( item->IsConnected( aPosition ) )
                breakLines[WIRES] = true;
 
            break;
 
        default:
            break;
        }
    }
 
    for( int layer : { WIRES, BUSES } )
    {
        if( breakLines[layer] )
        {
            for( const SCH_LINE* line : midPointLines[layer] )
            {
                exitAngles[layer].insert( line->GetAngleFrom( aPosition ) );
                exitAngles[layer].insert( line->GetReverseAngleFrom( aPosition ) );
            }
        }
    }
 
    if( info.hasBusEntry )
    {
        // The bus entry and one wire is 2 wires, and the one entry is exactly one bus
        // Any more wires must be multiple wires, but any more buses means a wire
        // crossing at the bus entry root.
        info.hasBusEntryToMultipleWires = exitAngles[WIRES].size() > 2 && exitAngles[BUSES].size() == 1;
    }
 
    // Any three things of the same type is a junction of some sort
    info.isJunction = exitAngles[WIRES].size() >= 3 || exitAngles[BUSES].size() >= 3;
 
    return info;
}
 
 
std::vector<SCH_JUNCTION*> JUNCTION_HELPERS::PreviewJunctions( const SCH_SCREEN* aScreen,
                                                               const std::vector<SCH_ITEM*>& aItems )
{
    EE_RTREE combined;
    std::unordered_set<const SCH_ITEM*> previewSet( aItems.begin(), aItems.end() );
 
    // Existing items, skipping any that are also in aItems to avoid double-counting
    for( const SCH_ITEM* item : aScreen->Items() )
    {
        if( !item->IsConnectable() )
            continue;
 
        if( previewSet.count( item ) )
            continue;
 
        combined.insert( const_cast<SCH_ITEM*>( item ) );
    }
 
    // Temporary/preview items
    for( SCH_ITEM* item : aItems )
    {
        if( !item || !item->IsConnectable() )
            continue;
 
        combined.insert( item );
    }
 
    std::vector<VECTOR2I> connections = aScreen->GetConnections();
    std::vector<VECTOR2I> pts;
 
    for( SCH_ITEM* item : aItems )
    {
        if( !item || !item->IsConnectable() )
            continue;
 
        std::vector<VECTOR2I> new_pts = item->GetConnectionPoints();
        pts.insert( pts.end(), new_pts.begin(), new_pts.end() );
 
        if( item->Type() == SCH_LINE_T )
        {
            SCH_LINE* line = static_cast<SCH_LINE*>( item );
 
            for( const VECTOR2I& pt : connections )
            {
                if( IsPointOnSegment( line->GetStartPoint(), line->GetEndPoint(), pt ) )
                    pts.push_back( pt );
            }
        }
    }
 
    std::sort( pts.begin(), pts.end(),
               []( const VECTOR2I& a, const VECTOR2I& b )
               {
                   return a.x < b.x || ( a.x == b.x && a.y < b.y );
               } );
 
    pts.erase( std::unique( pts.begin(), pts.end() ), pts.end() );
 
    std::vector<SCH_JUNCTION*> jcts;
 
    for( const VECTOR2I& pt : pts )
    {
        POINT_INFO info = AnalyzePoint( combined, pt, false );
 
        if( info.isJunction && ( !info.hasBusEntry || info.hasBusEntryToMultipleWires ) )
        {
            SCH_JUNCTION* junction = new SCH_JUNCTION( pt );
 
            if( info.hasBusAtPoint )
                junction->SetLayer( LAYER_BUS_JUNCTION );
 
            jcts.push_back( junction );
        }
    }
 
    return jcts;
}