pcb_io_easyedapro_v3.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 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
 */
 
#include "pcb_io_easyedapro_v3.h"
 
#include <io/easyedapro/easyedapro_import_utils.h>
#include <io/easyedapro/easyedapro_v3_parser.h>
 
#include <pcb_io/easyedapro/pcb_io_easyedapro_v3_parser.h>
#include <pcb_io/easyedapro/pcb_io_easyedapro_parser.h>
 
#include <board.h>
#include <font/fontconfig.h>
#include <footprint.h>
#include <ki_exception.h>
#include <progress_reporter.h>
#include <reporter.h>
 
#include <set>
 
#include <wx/filename.h>
#include <wx/log.h>
 
 
PCB_IO_EASYEDAPRO_V3::PCB_IO_EASYEDAPRO_V3() :
        PCB_IO( wxS( "EasyEDA (JLCEDA) Pro v3" ) )
{
}
 
 
PCB_IO_EASYEDAPRO_V3::~PCB_IO_EASYEDAPRO_V3() = default;
 
 
const EASYEDAPRO::V3_DOC_PARSER& PCB_IO_EASYEDAPRO_V3::getCachedLibraryParser( const wxString& aLibraryPath ) const
{
    long long timestamp = GetLibraryTimestamp( aLibraryPath );
 
    if( !m_cachedLibraryParser || m_cachedLibraryPath != aLibraryPath || m_cachedLibraryTimestamp != timestamp )
    {
        m_cachedLibraryParser = std::make_unique<EASYEDAPRO::V3_DOC_PARSER>( aLibraryPath );
        m_cachedLibraryParser->LoadLibrary();
        m_cachedLibraryPath = aLibraryPath;
        m_cachedLibraryTimestamp = timestamp;
    }
 
    return *m_cachedLibraryParser;
}
 
 
std::vector<std::pair<wxString, wxString>>
PCB_IO_EASYEDAPRO_V3::EnumerateProjectBoards( const wxString& aFileName ) const
{
    EASYEDAPRO::V3_DOC_PARSER parser( aFileName );
    parser.Load();
 
    const nlohmann::json  sourceProject = EASYEDAPRO::BuildV3ProjectIndexFromRawDocs( parser, false );
    const nlohmann::json& pcbsIndex = sourceProject.at( "pcbs" );
 
    std::vector<std::pair<wxString, wxString>> result;
    result.reserve( parser.GetRawDocs( wxS( "PCB" ) ).size() );
 
    for( const auto& [uuid, doc] : parser.GetRawDocs( wxS( "PCB" ) ) )
    {
        wxString title = uuid;
        auto     pcbIt = pcbsIndex.find( std::string( uuid.ToUTF8() ) );
 
        if( pcbIt != pcbsIndex.end() )
            title = EASYEDAPRO::V3GetString( pcbIt.value(), "title", title );
 
        result.emplace_back( uuid, title );
    }
 
    return result;
}
 
 
bool PCB_IO_EASYEDAPRO_V3::CanReadBoard( const wxString& aFileName ) const
{
    return EASYEDAPRO::V3_DOC_PARSER::IsV3Archive( aFileName );
}
 
 
bool PCB_IO_EASYEDAPRO_V3::CanReadLibrary( const wxString& aFileName ) const
{
    if( !PCB_IO::CanReadLibrary( aFileName ) )
        return false;
 
    return EASYEDAPRO::V3_DOC_PARSER::IsV3Library( aFileName, wxS( "FOOTPRINT" ) );
}
 
 
long long PCB_IO_EASYEDAPRO_V3::GetLibraryTimestamp( const wxString& aLibraryPath ) const
{
    wxFileName fileName( aLibraryPath );
 
    if( !fileName.FileExists() )
        return 0;
 
    wxDateTime modified = fileName.GetModificationTime();
 
    if( !modified.IsValid() )
        return 0;
 
    return modified.GetTicks();
}
 
 
void PCB_IO_EASYEDAPRO_V3::FootprintEnumerate( wxArrayString& aFootprintNames, const wxString& aLibraryPath,
                                               bool aBestEfforts, const std::map<std::string, UTF8>* aProperties )
{
    const EASYEDAPRO::V3_DOC_PARSER& v3 = getCachedLibraryParser( aLibraryPath );
 
    std::map<wxString, wxString> footprintMap =
            EASYEDAPRO::BuildV3LibraryItemMap( v3, "footprints", wxS( "FOOTPRINT" ) );
 
    for( const auto& [name, uuid] : footprintMap )
        aFootprintNames.Add( name );
}
 
 
std::vector<FOOTPRINT*> PCB_IO_EASYEDAPRO_V3::GetImportedCachedLibraryFootprints()
{
    std::vector<FOOTPRINT*> result;
 
    result.reserve( m_importedLibFootprints.size() );
 
    for( const std::unique_ptr<FOOTPRINT>& footprint : m_importedLibFootprints )
        result.push_back( static_cast<FOOTPRINT*>( footprint->Clone() ) );
 
    return result;
}
 
 
std::unique_ptr<FOOTPRINT> PCB_IO_EASYEDAPRO_V3::FootprintLoad( const wxString& aLibraryPath,
                                                                const wxString& aFootprintName, bool aKeepUUID,
                                                                const std::map<std::string, UTF8>* aProperties )
{
    FONTCONFIG_REPORTER_SCOPE fontconfigScope( nullptr );
 
    const EASYEDAPRO::V3_DOC_PARSER& v3 = getCachedLibraryParser( aLibraryPath );
 
    std::map<wxString, wxString> footprintMap =
            EASYEDAPRO::BuildV3LibraryItemMap( v3, "footprints", wxS( "FOOTPRINT" ) );
 
    auto fpIt = footprintMap.find( aFootprintName );
 
    if( fpIt == footprintMap.end() )
        return nullptr;
 
    const EASYEDAPRO::V3_DOC_RAW* rawDoc = v3.FindRawDoc( wxS( "FOOTPRINT" ), fpIt->second );
 
    if( !rawDoc )
        return nullptr;
 
    const nlohmann::json& index = v3.GetLibraryIndex();
 
    PCB_IO_EASYEDAPRO_V3_PARSER parser( nullptr, nullptr );
    std::unique_ptr<FOOTPRINT>  footprint;
 
    try
    {
        footprint = parser.ParseFootprint( EASYEDAPRO::BuildV3BlobMap( v3 ), *rawDoc );
    }
    catch( nlohmann::json::exception& e )
    {
        THROW_IO_ERRORF( _( "Cannot load footprint '%s' from '%s': %s" ), aFootprintName, aLibraryPath, e.what() );
    }
 
    if( !footprint )
        THROW_IO_ERRORF( _( "Cannot load footprint '%s' from '%s'" ), aFootprintName, aLibraryPath );
 
    // Library footprints are keyed by footprint geometry; pick any matching device's 3D model.
    if( index.contains( "devices" ) && index.at( "devices" ).is_object() )
    {
        for( const auto& [devUuid, deviceJson] : index.at( "devices" ).items() )
        {
            if( !deviceJson.is_object() || !deviceJson.contains( "attributes" )
                || !deviceJson.at( "attributes" ).is_object() )
            {
                continue;
            }
 
            const nlohmann::json& attrs = deviceJson.at( "attributes" );
 
            if( !attrs.contains( "Footprint" )
                || EASYEDAPRO::V3JsonToString( attrs.at( "Footprint" ) ) != fpIt->second )
            {
                continue;
            }
 
            wxString modelUuid;
            wxString modelTitle;
            wxString modelTransform;
 
            if( attrs.contains( "3D Model" ) )
                modelUuid = EASYEDAPRO::V3JsonToString( attrs.at( "3D Model" ) );
 
            if( attrs.contains( "3D Model Title" ) )
                modelTitle = EASYEDAPRO::V3JsonToString( attrs.at( "3D Model Title" ) ).Trim();
            else
                modelTitle = modelUuid;
 
            if( attrs.contains( "3D Model Transform" ) )
                modelTransform = EASYEDAPRO::V3JsonToString( attrs.at( "3D Model Transform" ) );
 
            PCB_IO_EASYEDAPRO_PARSER modelParser( nullptr, nullptr );
            modelParser.FillFootprintModelInfo( footprint.get(), modelUuid, modelTitle, modelTransform );
 
            break;
        }
    }
 
    footprint->SetFPID( EASYEDAPRO::ToKiCadLibID( wxEmptyString, aFootprintName ) );
    footprint->Reference().SetVisible( true );
    footprint->Value().SetText( aFootprintName );
    footprint->Value().SetVisible( true );
    footprint->AutoPositionFields();
 
    return footprint;
}
 
 
void PCB_IO_EASYEDAPRO_V3::loadBoard( const wxString& aFileName, BOARD& aBoard, bool aIsNewLoad,
                                      const std::map<std::string, UTF8>* aProperties, PROJECT* aProject )
{
    m_props = aProperties;
    m_importedDesignRules.clear();
 
    m_board = &aBoard;
 
    FONTCONFIG_REPORTER_SCOPE fontconfigScope( &LOAD_INFO_REPORTER::GetInstance() );
 
    if( m_progressReporter )
    {
        m_progressReporter->Report( wxString::Format( _( "Loading %s..." ), aFileName ) );
 
        if( !m_progressReporter->KeepRefreshing() )
            THROW_IO_CANCELLED();
    }
 
    EASYEDAPRO::V3_DOC_PARSER adapter( aFileName );
    adapter.Load();
 
    nlohmann::json project = EASYEDAPRO::BuildV3ProjectIndexFromRawDocs( adapter );
 
    wxString pcbToLoad;
 
    if( m_props && m_props->contains( "pcb_id" ) )
    {
        pcbToLoad = wxString::FromUTF8( m_props->at( "pcb_id" ) );
    }
    else
    {
        std::map<wxString, nlohmann::json> prjPcbs = project.at( "pcbs" );
 
        if( prjPcbs.size() == 1 )
        {
            pcbToLoad = prjPcbs.begin()->first;
        }
        else if( m_choose_project_handler )
        {
            std::vector<IMPORT_PROJECT_DESC> chosen =
                    m_choose_project_handler( EASYEDAPRO::ProjectToSelectorDialog( project, true, false ) );
 
            if( !chosen.empty() )
                pcbToLoad = chosen[0].PCBId;
        }
        else if( !prjPcbs.empty() )
        {
            pcbToLoad = prjPcbs.begin()->first;
        }
    }
 
    if( pcbToLoad.empty() )
        THROW_IO_ERROR( _( "No PCB was found in the project to import." ) );
 
    PCB_IO_EASYEDAPRO_V3_PARSER parser( nullptr, nullptr );
 
    std::map<wxString, EASYEDAPRO::BLOB> blobs = EASYEDAPRO::BuildV3BlobMap( adapter );
 
    wxFileName sourceName( aFileName );
    wxString   fpLibName = EASYEDAPRO::ShortenLibName( sourceName.GetName() );
 
    // PCB components reference Footprint (geometry) and Device (BOM / 3D attrs) separately, so
    // definitions are keyed by footprint package title (e.g. C0402)
    std::map<wxString, std::unique_ptr<FOOTPRINT>> footprints;
    std::set<wxString>                             usedLibNames;
 
    for( const auto& [uuid, rawDoc] : adapter.GetRawDocs( wxS( "FOOTPRINT" ) ) )
    {
        std::unique_ptr<FOOTPRINT> footprint( parser.ParseFootprint( blobs, rawDoc ) );
 
        if( !footprint )
            continue;
 
        std::string           uuidKey = std::string( uuid.ToUTF8() );
        const nlohmann::json& fpMetas = project.at( "footprints" );
        wxString              fpTitle = uuid;
 
        if( fpMetas.contains( uuidKey ) )
            fpTitle = EASYEDAPRO::GetV3LibraryItemTitle( fpMetas.at( uuidKey ), uuid );
 
        wxString itemName = EASYEDAPRO::MakeUniqueLibName( usedLibNames, fpTitle, uuid );
        footprint->SetFPID( EASYEDAPRO::ToKiCadLibID( fpLibName, itemName ) );
        footprints.emplace( uuid, std::move( footprint ) );
    }
 
    const EASYEDAPRO::V3_DOC_RAW* pcbRawDoc = adapter.FindRawDoc( wxS( "PCB" ), pcbToLoad );
 
    if( !pcbRawDoc )
        THROW_IO_ERRORF( _( "PCB document '%s' not found in '%s'" ), pcbToLoad, aFileName );
 
    std::multimap<wxString, EASYEDAPRO::POURED> poured; // Empty - v3 parser extracts from raw doc
 
    parser.ParseBoard( m_board, project, footprints, blobs, poured, *pcbRawDoc, fpLibName );
    m_importedDesignRules = parser.GenerateSafeSpacingRules();
 
 
    // Loading a board must not write to disk, so the project library is left to the reconciler
    m_importedLibFootprints.clear();
 
    for( auto& [uuid, footprint] : footprints )
    {
        if( footprint )
            m_importedLibFootprints.emplace_back( std::move( footprint ) );
    }
 
    if( adapter.GetSkippedCount() > 0 )
    {
        Report( wxString::Format( _( "EasyEDA (JLCEDA) Pro v3 import skipped %d unsupported object(s)." ),
                                  adapter.GetSkippedCount() ) , RPT_SEVERITY_WARNING );
    }
}