kicad-source/qa/tests/eeschema/test_update_symbol_connectivity.cpp

/*
 * 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 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, see <http://www.gnu.org/licenses/>.
 */

#include <qa_utils/wx_utils/unit_test_utils.h>

#include <connection_graph.h>
#include <lib_symbol.h>
#include <pin_type.h>
#include <sch_pin.h>
#include <sch_symbol.h>
#include <sch_sheet.h>
#include <sch_screen.h>

#include <algorithm>
#include <set>


void boost_test_update_symbol_connectivity()
{
    // Create root sheet and screen
    SCH_SCREEN* screen = new SCH_SCREEN( nullptr );
    SCH_SHEET*  sheet = new SCH_SHEET( nullptr, VECTOR2I( 0, 0 ), VECTOR2I( 1000, 1000 ) );
    sheet->SetScreen( screen );

    SCH_SHEET_PATH sheetPath;
    sheetPath.push_back( sheet );

    // Build library symbol
    LIB_SYMBOL lib( "TEST", nullptr );
    lib.SetGlobalPower();
    lib.SetDuplicatePinNumbersAreJumpers( true );
    lib.JumperPinGroups().push_back( { wxString( "3" ), wxString( "4" ) } );

    auto make_pin = [&]( const wxString& num, const wxString& name, ELECTRICAL_PINTYPE type, const VECTOR2I& pos )
    {
        SCH_PIN* pin = new SCH_PIN( &lib );
        pin->SetNumber( num );
        pin->SetName( name );
        pin->SetType( type );
        pin->SetPosition( pos );
        lib.AddDrawItem( pin );
    };

    make_pin( "1", "VCC", ELECTRICAL_PINTYPE::PT_POWER_IN, VECTOR2I( 0, 0 ) );
    make_pin( "2", "A", ELECTRICAL_PINTYPE::PT_INPUT, VECTOR2I( 1000, 0 ) );
    make_pin( "2", "B", ELECTRICAL_PINTYPE::PT_INPUT, VECTOR2I( 2000, 0 ) );
    make_pin( "3", "C", ELECTRICAL_PINTYPE::PT_INPUT, VECTOR2I( 3000, 0 ) );
    make_pin( "4", "D", ELECTRICAL_PINTYPE::PT_INPUT, VECTOR2I( 4000, 0 ) );

    // Create schematic symbol instance
    SCH_SYMBOL symbol( lib, lib.GetLibId(), &sheetPath, 0, 0, VECTOR2I( 0, 0 ) );
    symbol.SetValueFieldText( "VCC" );
    symbol.UpdatePins();

    std::vector<SCH_PIN*> pins = symbol.GetPins( &sheetPath );
    SCH_PIN*              powerPin = nullptr;
    SCH_PIN*              dupA = nullptr;
    SCH_PIN*              dupB = nullptr;
    SCH_PIN*              jumpC = nullptr;
    SCH_PIN*              jumpD = nullptr;

    for( SCH_PIN* pin : pins )
    {
        if( pin->GetNumber() == "1" )
            powerPin = pin;
        else if( pin->GetNumber() == "2" )
        {
            if( !dupA )
                dupA = pin;
            else
                dupB = pin;
        }
        else if( pin->GetNumber() == "3" )
            jumpC = pin;
        else if( pin->GetNumber() == "4" )
            jumpD = pin;
    }

    CONNECTION_GRAPH                           graph;
    std::map<VECTOR2I, std::vector<SCH_ITEM*>> connection_map;
    graph.updateSymbolConnectivity( sheetPath, &symbol, connection_map );

    // Global power pin captured
    BOOST_REQUIRE_EQUAL( graph.m_global_power_pins.size(), 1 );
    BOOST_CHECK( graph.m_global_power_pins[0].second == powerPin );
    BOOST_CHECK_EQUAL( powerPin->Connection( &sheetPath )->Name( true ), "VCC" );

    // Duplicate pin numbers link together
    const SCH_ITEM_VEC& dupAConn = dupA->ConnectedItems( sheetPath );
    BOOST_CHECK( std::find( dupAConn.begin(), dupAConn.end(), dupB ) != dupAConn.end() );
    const SCH_ITEM_VEC& dupBConn = dupB->ConnectedItems( sheetPath );
    BOOST_CHECK( std::find( dupBConn.begin(), dupBConn.end(), dupA ) != dupBConn.end() );

    // Jumper group links pins
    const SCH_ITEM_VEC& jumpCConn = jumpC->ConnectedItems( sheetPath );
    BOOST_CHECK( std::find( jumpCConn.begin(), jumpCConn.end(), jumpD ) != jumpCConn.end() );
    const SCH_ITEM_VEC& jumpDConn = jumpD->ConnectedItems( sheetPath );
    BOOST_CHECK( std::find( jumpDConn.begin(), jumpDConn.end(), jumpC ) != jumpDConn.end() );

    // Connection map contains all pins
    std::set<SCH_PIN*> mapPins;
    for( const auto& [pos, items] : connection_map )
        for( SCH_ITEM* item : items )
            mapPins.insert( static_cast<SCH_PIN*>( item ) );
    BOOST_CHECK_EQUAL( mapPins.size(), 5 );

    // Item list contains all pins
    BOOST_CHECK_EQUAL( graph.m_items.size(), 5 );
    for( SCH_PIN* pin : { powerPin, dupA, dupB, jumpC, jumpD } )
    {
        BOOST_CHECK( std::find( graph.m_items.begin(), graph.m_items.end(), pin )
                     != graph.m_items.end() );
    }

    delete sheet; // deletes screen
}

BOOST_AUTO_TEST_SUITE( UpdateSymbolConnectivity )
BOOST_AUTO_TEST_CASE( SymbolConnectivityLinksPins )
{
    boost_test_update_symbol_connectivity();
}
BOOST_AUTO_TEST_SUITE_END()