2023-09-06 11:58:39 +00:00
|
|
|
/*
|
|
|
|
* This program source code file is part of KiCad, a free EDA CAD application.
|
|
|
|
*
|
|
|
|
* Copyright (C) 2023 Alex Shvartzkop <dudesuchamazing@gmail.com>
|
|
|
|
* Copyright (C) 2023 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 "easyedapro_import_utils.h"
|
2023-10-30 06:34:45 +00:00
|
|
|
#include "easyedapro_parser.h"
|
|
|
|
|
2023-12-19 17:39:26 +00:00
|
|
|
#include <io/common/plugin_common_choose_project.h>
|
2023-09-06 11:58:39 +00:00
|
|
|
|
2023-09-08 02:29:40 +00:00
|
|
|
#include <ki_exception.h>
|
2023-09-06 11:58:39 +00:00
|
|
|
#include <string_utils.h>
|
|
|
|
#include <nlohmann/json.hpp>
|
2023-10-30 06:34:45 +00:00
|
|
|
#include <core/json_serializers.h>
|
2023-09-06 11:58:39 +00:00
|
|
|
|
|
|
|
#include <wx/log.h>
|
|
|
|
#include <wx/stream.h>
|
|
|
|
#include <wx/zipstrm.h>
|
|
|
|
#include <wx/wfstream.h>
|
|
|
|
#include <wx/mstream.h>
|
|
|
|
#include <wx/txtstrm.h>
|
|
|
|
|
|
|
|
|
|
|
|
wxString EASYEDAPRO::ShortenLibName( wxString aProjectName )
|
|
|
|
{
|
|
|
|
wxString shortenedName = aProjectName;
|
|
|
|
shortenedName.Replace( wxS( "ProProject_" ), wxS( "" ) );
|
|
|
|
shortenedName.Replace( wxS( "ProDocument_" ), wxS( "" ) );
|
|
|
|
shortenedName = shortenedName.substr( 0, 10 );
|
|
|
|
|
|
|
|
return LIB_ID::FixIllegalChars( shortenedName + wxS( "-easyedapro" ), true );
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
LIB_ID EASYEDAPRO::ToKiCadLibID( const wxString& aLibName, const wxString& aLibReference )
|
|
|
|
{
|
|
|
|
wxString libName = LIB_ID::FixIllegalChars( aLibName, true );
|
|
|
|
wxString libReference = EscapeString( aLibReference, CTX_LIBID );
|
|
|
|
|
|
|
|
wxString key = !aLibName.empty() ? ( aLibName + ':' + libReference ) : libReference;
|
|
|
|
|
|
|
|
LIB_ID libId;
|
|
|
|
libId.Parse( key, true );
|
|
|
|
|
|
|
|
return libId;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2023-10-30 06:34:45 +00:00
|
|
|
std::vector<IMPORT_PROJECT_DESC>
|
|
|
|
EASYEDAPRO::ProjectToSelectorDialog( const nlohmann::json& aProject, bool aPcbOnly, bool aSchOnly )
|
|
|
|
{
|
|
|
|
std::vector<IMPORT_PROJECT_DESC> result;
|
|
|
|
|
|
|
|
std::map<wxString, EASYEDAPRO::PRJ_SCHEMATIC> prjSchematics = aProject.at( "schematics" );
|
|
|
|
std::map<wxString, EASYEDAPRO::PRJ_BOARD> prjBoards = aProject.at( "boards" );
|
|
|
|
std::map<wxString, wxString> prjPcbNames = aProject.at( "pcbs" );
|
|
|
|
|
|
|
|
for( const auto& [prjName, board] : prjBoards )
|
|
|
|
{
|
|
|
|
IMPORT_PROJECT_DESC desc;
|
|
|
|
desc.ComboName = desc.ComboId = prjName;
|
|
|
|
desc.PCBId = board.pcb;
|
|
|
|
desc.SchematicId = board.schematic;
|
|
|
|
|
|
|
|
auto pcbNameIt = prjPcbNames.find( desc.PCBId );
|
|
|
|
if( pcbNameIt != prjPcbNames.end() )
|
|
|
|
{
|
|
|
|
desc.PCBName = pcbNameIt->second;
|
|
|
|
|
|
|
|
if( desc.PCBName.empty() )
|
|
|
|
desc.PCBName = pcbNameIt->first;
|
|
|
|
|
|
|
|
prjPcbNames.erase( pcbNameIt );
|
|
|
|
}
|
|
|
|
|
|
|
|
auto schIt = prjSchematics.find( desc.SchematicId );
|
|
|
|
if( schIt != prjSchematics.end() )
|
|
|
|
{
|
|
|
|
desc.SchematicName = schIt->second.name;
|
|
|
|
|
|
|
|
if( desc.SchematicName.empty() )
|
|
|
|
desc.SchematicName = schIt->first;
|
|
|
|
|
|
|
|
prjSchematics.erase( schIt );
|
|
|
|
}
|
|
|
|
|
|
|
|
result.emplace_back( desc );
|
|
|
|
}
|
|
|
|
|
|
|
|
if( !aSchOnly )
|
|
|
|
{
|
|
|
|
for( const auto& [pcbId, pcbName] : prjPcbNames )
|
|
|
|
{
|
|
|
|
IMPORT_PROJECT_DESC desc;
|
|
|
|
desc.PCBId = pcbId;
|
|
|
|
desc.PCBName = pcbName;
|
|
|
|
|
|
|
|
if( desc.PCBName.empty() )
|
|
|
|
desc.PCBName = pcbId;
|
|
|
|
|
|
|
|
result.emplace_back( desc );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if( !aPcbOnly )
|
|
|
|
{
|
|
|
|
for( const auto& [schId, schData] : prjSchematics )
|
|
|
|
{
|
|
|
|
IMPORT_PROJECT_DESC desc;
|
|
|
|
desc.SchematicId = schId;
|
|
|
|
desc.SchematicName = schData.name;
|
|
|
|
|
|
|
|
if( desc.SchematicName.empty() )
|
|
|
|
desc.SchematicName = schId;
|
|
|
|
|
|
|
|
result.emplace_back( desc );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2023-12-25 04:00:11 +00:00
|
|
|
nlohmann::json EASYEDAPRO::ReadProjectOrDeviceFile( const wxString& aZipFileName )
|
2023-09-06 11:58:39 +00:00
|
|
|
{
|
|
|
|
std::shared_ptr<wxZipEntry> entry;
|
|
|
|
wxFFileInputStream in( aZipFileName );
|
|
|
|
wxZipInputStream zip( in );
|
|
|
|
|
|
|
|
while( entry.reset( zip.GetNextEntry() ), entry.get() != NULL )
|
|
|
|
{
|
|
|
|
wxString name = entry->GetName();
|
|
|
|
|
2024-03-03 18:20:14 +00:00
|
|
|
try
|
2023-09-06 11:58:39 +00:00
|
|
|
{
|
2024-03-03 18:20:14 +00:00
|
|
|
if( name == wxS( "project.json" ) || name == wxS( "device.json" ) )
|
|
|
|
{
|
|
|
|
wxMemoryOutputStream memos;
|
|
|
|
memos << zip;
|
|
|
|
wxStreamBuffer* buf = memos.GetOutputStreamBuffer();
|
2023-09-06 11:58:39 +00:00
|
|
|
|
2024-03-03 18:20:14 +00:00
|
|
|
wxString str =
|
|
|
|
wxString::FromUTF8( (char*) buf->GetBufferStart(), buf->GetBufferSize() );
|
|
|
|
|
|
|
|
return nlohmann::json::parse( str );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
catch( nlohmann::json::exception& e )
|
|
|
|
{
|
|
|
|
THROW_IO_ERROR(
|
|
|
|
wxString::Format( _( "JSON error reading '%s': %s" ), name, e.what() ) );
|
|
|
|
}
|
|
|
|
catch( std::exception& e )
|
|
|
|
{
|
|
|
|
THROW_IO_ERROR( wxString::Format( _( "Error reading '%s': %s" ), name, e.what() ) );
|
2023-09-06 11:58:39 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2023-12-25 04:00:11 +00:00
|
|
|
THROW_IO_ERROR( wxString::Format(
|
|
|
|
_( "'%s' does not appear to be a valid EasyEDA (JLCEDA) Pro "
|
|
|
|
"project or library file. Cannot find project.json or device.json." ),
|
|
|
|
aZipFileName ) );
|
2023-09-06 11:58:39 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void EASYEDAPRO::IterateZipFiles(
|
|
|
|
const wxString& aFileName,
|
|
|
|
std::function<bool( const wxString&, const wxString&, wxInputStream& )> aCallback )
|
|
|
|
{
|
|
|
|
std::shared_ptr<wxZipEntry> entry;
|
|
|
|
wxFFileInputStream in( aFileName );
|
|
|
|
wxZipInputStream zip( in );
|
|
|
|
|
|
|
|
if( !zip.IsOk() )
|
|
|
|
{
|
|
|
|
THROW_IO_ERROR( wxString::Format( _( "Cannot read ZIP archive '%s'" ), aFileName ) );
|
|
|
|
}
|
|
|
|
|
|
|
|
while( entry.reset( zip.GetNextEntry() ), entry.get() != NULL )
|
|
|
|
{
|
|
|
|
wxString name = entry->GetName();
|
|
|
|
wxString baseName = name.AfterLast( '\\' ).AfterLast( '/' ).BeforeFirst( '.' );
|
|
|
|
|
|
|
|
try
|
|
|
|
{
|
|
|
|
if( aCallback( name, baseName, zip ) )
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
catch( nlohmann::json::exception& e )
|
|
|
|
{
|
|
|
|
THROW_IO_ERROR(
|
|
|
|
wxString::Format( _( "JSON error reading '%s': %s" ), name, e.what() ) );
|
|
|
|
}
|
|
|
|
catch( std::exception& e )
|
|
|
|
{
|
|
|
|
THROW_IO_ERROR( wxString::Format( _( "Error reading '%s': %s" ), name, e.what() ) );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
std::vector<nlohmann::json> EASYEDAPRO::ParseJsonLines( wxInputStream& aInput,
|
|
|
|
const wxString& aSource )
|
|
|
|
{
|
|
|
|
wxTextInputStream txt( aInput, wxS( " " ), wxConvUTF8 );
|
|
|
|
|
|
|
|
int currentLine = 1;
|
|
|
|
|
|
|
|
std::vector<nlohmann::json> lines;
|
|
|
|
while( aInput.CanRead() )
|
|
|
|
{
|
|
|
|
try
|
|
|
|
{
|
|
|
|
nlohmann::json js = nlohmann::json::parse( txt.ReadLine() );
|
|
|
|
lines.emplace_back( js );
|
|
|
|
}
|
|
|
|
catch( nlohmann::json::exception& e )
|
|
|
|
{
|
|
|
|
wxLogWarning( wxString::Format( _( "Cannot parse JSON line %d in '%s': %s" ),
|
|
|
|
currentLine, aSource, e.what() ) );
|
|
|
|
}
|
|
|
|
|
|
|
|
currentLine++;
|
|
|
|
}
|
|
|
|
|
|
|
|
return lines;
|
|
|
|
}
|