netlist_exporter_base.cpp

Go to the documentation of this file.

/*
 * This program source code file is part of KiCad, a free EDA CAD application.
 *
 * Copyright (C) 1992-2017 jp.charras at wanadoo.fr
 * Copyright (C) 2013-2017 SoftPLC Corporation, Dick Hollenbeck <[email protected]>
 * 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 <https://www.gnu.org/licenses/>.
 */
 
#include <netlist_exporter_base.h>
 
#include <string_utils.h>
 
#include <trace_helpers.h>
#include <connection_graph.h>
#include <kiface_ids.h>
#include <kiway.h>
#include <lib_id.h>
#include <sch_pin.h>
#include <sch_reference_list.h>
#include <sch_screen.h>
#include <sch_symbol.h>
#include <schematic.h>
 
 
// a "less than" test on two LIB_SYMBOLs (.m_name wxStrings)
bool LIB_SYMBOL_LESS_THAN::operator()( LIB_SYMBOL* const& libsymbol1,
                                       LIB_SYMBOL* const& libsymbol2 ) const
{
    // Use case specific GetName() wxString compare
    return libsymbol1->GetLibId() < libsymbol2->GetLibId();
}
 
 
wxString NETLIST_EXPORTER_BASE::MakeCommandLine( const wxString& aFormatString,
                                                 const wxString& aNetlistFile,
                                                 const wxString& aFinalFile,
                                                 const wxString& aProjectPath )
{
    // Expand format symbols in the command line:
    // %B => base filename of selected output file, minus path and extension.
    // %P => project directory name, without trailing '/' or '\'.
    // %I => full filename of the input file (the intermediate net file).
    // %O => complete filename and path (but without extension) of the user chosen output file.
 
    wxString   ret  = aFormatString;
    wxFileName in   = aNetlistFile;
    wxFileName out  = aFinalFile;
    wxString str_out  = out.GetFullPath();
 
    ret.Replace( "%P", aProjectPath, true );
    ret.Replace( "%B", out.GetName(), true );
    ret.Replace( "%I", in.GetFullPath(), true );
 
#ifdef __WINDOWS__
    // A ugly hack to run xsltproc that has a serious bug on Window since a long time:
    // the filename given after -o option (output filename) cannot use '\' in filename
    // so replace if by '/' if possible (I mean if the filename does not start by "\\"
    // that is a filename on a Windows server)
 
    if( !str_out.StartsWith( "\\\\" ) )
        str_out.Replace( "\\", "/" );
#endif
 
    ret.Replace( "%O", str_out, true );
 
    return ret;
}
 
 
SCH_SYMBOL* NETLIST_EXPORTER_BASE::findNextSymbol( EDA_ITEM* aItem,
                                                   const SCH_SHEET_PATH& aSheetPath )
{
    wxCHECK( aItem, nullptr );
 
    wxString ref;
 
    if( aItem->Type() != SCH_SYMBOL_T )
        return nullptr;
 
    // found next symbol
    SCH_SYMBOL* symbol = (SCH_SYMBOL*) aItem;
 
    // Power symbols and other symbols which have the reference starting with "#" are not
    // included in netlist (pseudo or virtual symbols)
    ref = symbol->GetRef( &aSheetPath );
 
    if( ref[0] == wxChar( '#' ) )
        return nullptr;
 
    SCH_SCREEN* screen = aSheetPath.LastScreen();
 
    wxCHECK( screen, nullptr );
 
    auto it = screen->GetLibSymbols().find( symbol->GetSchSymbolLibraryName() );
 
    if( it == screen->GetLibSymbols().end() )
        return nullptr;
 
    LIB_SYMBOL* libSymbol = it->second;
 
    // If symbol is a "multi parts per package" type
    if( libSymbol->GetUnitCount() > 1 )
    {
        // test if this reference has already been processed, and if so skip
        if( m_referencesAlreadyFound.Lookup( ref ) )
            return nullptr;
    }
 
    // record the usage of this library symbol entry.
    m_libParts.insert( libSymbol ); // rejects non-unique pointers
 
    return symbol;
}
 
 
std::vector<PIN_INFO> NETLIST_EXPORTER_BASE::CreatePinList( SCH_SYMBOL* aSymbol,
                                                            const SCH_SHEET_PATH& aSheetPath,
                                                            bool aKeepUnconnectedPins )
{
    std::vector<PIN_INFO> pins;
 
    if( !aSymbol )
        return pins;
 
    wxString ref( aSymbol->GetRef( &aSheetPath ) );
 
    // Power symbols and other symbols which have the reference starting with "#" are not
    // included in netlist (pseudo or virtual symbols)
    if( ( ref[0] == wxChar( '#' ) ) || aSymbol->IsPower() )
        return pins;
 
    // if( aSymbol->m_FlagControlMulti == 1 )
    //    continue;                                      /* yes */
    // removed because with multiple instances of one schematic (several sheets pointing to
    // 1 screen), this will be erroneously be toggled.
 
    if( !aSymbol->GetLibSymbolRef() )
        return pins;
 
    // If symbol is a "multi parts per package" type
    if( aSymbol->GetLibSymbolRef()->GetUnitCount() > 1 )
    {
        // Collect all pins for this reference designator by searching the entire design for
        // other parts with the same reference designator.
        findAllUnitsOfSymbol( aSymbol, aSheetPath, pins, aKeepUnconnectedPins );
    }
 
    else // GetUnitCount() <= 1 means one part per package
    {
        CONNECTION_GRAPH* graph = m_schematic->ConnectionGraph();
 
        for( const SCH_PIN* pin : aSymbol->GetPins( &aSheetPath ) )
        {
            if( SCH_CONNECTION* conn = pin->Connection( &aSheetPath ) )
            {
                const wxString& netName = conn->Name();
 
                if( !aKeepUnconnectedPins )     // Skip unconnected pins if requested
                {
                    CONNECTION_SUBGRAPH* sg = graph->FindSubgraphByName( netName, aSheetPath );
 
                    if( !sg || sg->GetNoConnect() || sg->GetItems().size() < 2 )
                        continue;
                }
 
                appendResolvedPins( pins, pin, aSheetPath, netName );
            }
        }
    }
 
    // Sort pins in m_SortedSymbolPinList by pin number
    std::sort( pins.begin(), pins.end(),
               []( const PIN_INFO& lhs, const PIN_INFO& rhs )
               {
                   return StrNumCmp( lhs.num, rhs.num, true ) < 0;
               } );
 
    // Remove duplicate Pins in m_SortedSymbolPinList
    eraseDuplicatePins( pins );
 
    // record the usage of this library symbol
    m_libParts.insert( aSymbol->GetLibSymbolRef().get() ); // rejects non-unique pointers
 
    return pins;
}
 
 
std::vector<wxString> NETLIST_EXPORTER_BASE::resolvePadNumbers( const SCH_PIN*        aPin,
                                                                const SCH_SHEET_PATH& aSheetPath ) const
{
    // Shared resolution kernel for all netlist paths (issue #2282) so they cannot drift.
    const wxString variantName = m_schematic ? m_schematic->GetCurrentVariant() : wxString();
 
    if( m_kiway )
    {
        if( const SCH_SYMBOL* symbol = dynamic_cast<const SCH_SYMBOL*>( aPin->GetParentSymbol() ) )
        {
            wxString fpText = symbol->GetFootprintFieldText( true, &aSheetPath, false, variantName );
            LIB_ID   fpId;
 
            if( !fpText.IsEmpty() && fpId.Parse( fpText, true ) < 0 )
            {
                const std::set<wxString>& pads = footprintPads( fpId.GetUniStringLibId() );
 
                if( !pads.empty() )
                {
                    SCH_PIN::PAD_RESOLUTION state = SCH_PIN::PAD_RESOLUTION::MAPPED;
                    wxString pad = aPin->GetEffectivePadNumber( aSheetPath, variantName, fpId, &pads, &state );
 
                    if( state == SCH_PIN::PAD_RESOLUTION::UNMAPPED )
                        return {};
 
                    return ExpandStackedPinNotation( pad );
                }
            }
        }
    }
 
    return ExpandStackedPinNotation( aPin->GetEffectivePadNumber( aSheetPath, variantName ) );
}
 
 
const std::set<wxString>& NETLIST_EXPORTER_BASE::footprintPads( const wxString& aFootprintId ) const
{
    auto it = m_footprintPadCache.find( aFootprintId );
 
    if( it != m_footprintPadCache.end() )
        return it->second;
 
    std::set<wxString>& pads = m_footprintPadCache[aFootprintId];
 
    if( m_kiway && !aFootprintId.IsEmpty() )
    {
        if( KIFACE* cvpcb = m_kiway->KiFACE( KIWAY::FACE_CVPCB ) )
        {
            typedef void ( *PAD_NUMBERS_FN_PTR )( const wxString&, PROJECT*, std::set<wxString>& );
 
            if( auto fetch = (PAD_NUMBERS_FN_PTR) cvpcb->IfaceOrAddress( KIFACE_FOOTPRINT_PAD_NUMBERS ) )
                fetch( aFootprintId, &m_schematic->Project(), pads );
        }
    }
 
    return pads;
}
 
 
void NETLIST_EXPORTER_BASE::appendResolvedPins( std::vector<PIN_INFO>& aPins, const SCH_PIN* aPin,
                                                const SCH_SHEET_PATH& aSheetPath, const wxString& aNetName )
{
    const wxString baseName = aPin->GetShownName();
 
    for( const wxString& padNum : resolvePadNumbers( aPin, aSheetPath ) )
    {
        wxString pinName = baseName.IsEmpty() ? padNum : baseName + wxT( "_" ) + padNum;
        aPins.emplace_back( padNum, aNetName, pinName, aPin->GetNumber() );
    }
}
 
 
void NETLIST_EXPORTER_BASE::eraseDuplicatePins( std::vector<PIN_INFO>& aPins )
{
    // Helper to check if a net name is auto-generated rather than user-assigned.
    // Auto-generated nets start with "unconnected-(" for NC pins or "Net-(" for unnamed nets.
    auto isAutoGeneratedNet = []( const wxString& aNetName ) -> bool
    {
        return aNetName.StartsWith( wxT( "unconnected-(" ) )
            || aNetName.StartsWith( wxT( "Net-(" ) );
    };
 
    for( unsigned ii = 0; ii < aPins.size(); ii++ )
    {
        if( aPins[ii].num.empty() )
            continue;
 
        // Search for duplicated pins and keep only one. Duplicate pin numbers can occur
        // for multi-unit symbols with shared pins, or for symbols with the "duplicate pin
        // numbers are jumpers" flag. In either case, all pins with the same number should
        // connect to the same net. When they don't (user error), we prefer user-assigned
        // nets over auto-generated ones to preserve user intent in the netlist.
        // Because the pin list is sorted by pin number, duplicates are consecutive.
        unsigned idxBest = ii;
 
        for( unsigned jj = ii + 1; jj < aPins.size(); jj++ )
        {
            if( aPins[jj].num.empty() )
                continue;
 
            if( aPins[idxBest].num != aPins[jj].num )
                break;
 
            // A genuine many-to-one mapping collision - two *different* symbol pins resolved to
            // the same pad on *different* nets - must survive so it stays visible and is flagged
            // by ERC (issue #2282).  Shared multi-unit pins and jumpers carry the same source pin
            // number, so they still collapse below as before.
            if( aPins[idxBest].netName != aPins[jj].netName && !aPins[idxBest].srcPin.IsEmpty()
                && !aPins[jj].srcPin.IsEmpty() && aPins[idxBest].srcPin != aPins[jj].srcPin )
            {
                continue;
            }
 
            // Check if jj has a better (user-assigned) net than the current best.
            // Prefer user-assigned nets over auto-generated "unconnected-(" or "Net-(" nets.
            bool bestIsAuto = isAutoGeneratedNet( aPins[idxBest].netName );
            bool jjIsAuto = isAutoGeneratedNet( aPins[jj].netName );
 
            if( bestIsAuto && !jjIsAuto )
            {
                // jj has a user-assigned net while best has auto-generated; switch to jj
                aPins[idxBest].num.clear();
                idxBest = jj;
            }
            else
            {
                aPins[jj].num.clear();
            }
        }
    }
}
 
 
void NETLIST_EXPORTER_BASE::findAllUnitsOfSymbol( SCH_SYMBOL* aSchSymbol,
                                                  const SCH_SHEET_PATH& aSheetPath,
                                                  std::vector<PIN_INFO>& aPins,
                                                  bool aKeepUnconnectedPins )
{
    wxString ref = aSchSymbol->GetRef( &aSheetPath );
    wxString ref2;
 
    CONNECTION_GRAPH* graph = m_schematic->ConnectionGraph();
 
    for( const SCH_SHEET_PATH& sheet : m_schematic->Hierarchy() )
    {
        for( SCH_ITEM* item : sheet.LastScreen()->Items().OfType( SCH_SYMBOL_T ) )
        {
            SCH_SYMBOL* comp2 = static_cast<SCH_SYMBOL*>( item );
 
            ref2 = comp2->GetRef( &sheet );
 
            if( ref2.CmpNoCase( ref ) != 0 )
                continue;
 
            for( const SCH_PIN* pin : comp2->GetPins( &sheet ) )
            {
                if( SCH_CONNECTION* conn = pin->Connection( &sheet ) )
                {
                    const wxString& netName = conn->Name();
 
                    if( !aKeepUnconnectedPins )     // Skip unconnected pins if requested
                    {
                        CONNECTION_SUBGRAPH* sg = graph->FindSubgraphByName( netName, sheet );
 
                        if( !sg || sg->GetNoConnect() || sg->GetItems().size() < 2 )
                            continue;
                    }
 
                    appendResolvedPins( aPins, pin, sheet, netName );
                }
            }
        }
    }
}