kicad/pcbnew/exporters/step/step_pcb_model.cpp

1172 lines
36 KiB
C++
Raw Normal View History

2016-09-02 10:08:40 +00:00
/*
* This program source code file is part of KiCad, a free EDA CAD application.
*
* Copyright (C) 2022 Mark Roszko <mark.roszko@gmail.com>
2016-09-02 10:08:40 +00:00
* Copyright (C) 2016 Cirilo Bernardo <cirilo.bernardo@gmail.com>
* Copyright (C) 2016-2022 KiCad Developers, see AUTHORS.txt for contributors.
2016-09-02 10:08:40 +00:00
*
* 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 <algorithm>
#include <cmath>
#include <sstream>
#include <string>
#include <utility>
#include <wx/filename.h>
#include <wx/filefn.h>
#include <wx/stdpaths.h>
#include <wx/wfstream.h>
#include <wx/zipstrm.h>
#include <decompress.hpp>
2016-09-02 10:08:40 +00:00
#include <footprint.h>
#include <pad.h>
#include "step_pcb_model.h"
2017-03-09 02:40:41 +00:00
#include "streamwrapper.h"
2016-09-02 10:08:40 +00:00
#include <IGESCAFControl_Reader.hxx>
#include <IGESCAFControl_Writer.hxx>
#include <IGESControl_Controller.hxx>
#include <IGESData_GlobalSection.hxx>
#include <IGESData_IGESModel.hxx>
#include <Interface_Static.hxx>
#include <Quantity_Color.hxx>
#include <STEPCAFControl_Reader.hxx>
#include <STEPCAFControl_Writer.hxx>
#include <APIHeaderSection_MakeHeader.hxx>
#include <Standard_Version.hxx>
2016-09-02 10:08:40 +00:00
#include <TCollection_ExtendedString.hxx>
#include <TDataStd_Name.hxx>
#include <TDataStd_TreeNode.hxx>
2016-09-02 10:08:40 +00:00
#include <TDF_LabelSequence.hxx>
#include <TDF_ChildIterator.hxx>
#include <TopExp_Explorer.hxx>
#include <XCAFDoc.hxx>
2016-09-02 10:08:40 +00:00
#include <XCAFDoc_DocumentTool.hxx>
#include <XCAFDoc_ColorTool.hxx>
#include <BRep_Tool.hxx>
#include <BRepMesh_IncrementalMesh.hxx>
#include <BRepBuilderAPI.hxx>
2016-09-02 10:08:40 +00:00
#include <BRepBuilderAPI_MakeEdge.hxx>
#include <BRepBuilderAPI_Transform.hxx>
#include <BRepBuilderAPI_GTransform.hxx>
2016-09-02 10:08:40 +00:00
#include <BRepBuilderAPI_MakeFace.hxx>
#include <BRepPrimAPI_MakePrism.hxx>
#include <BRepPrimAPI_MakeCylinder.hxx>
#include <BRepAlgoAPI_Cut.hxx>
#include <TopoDS.hxx>
#include <TopoDS_Wire.hxx>
#include <TopoDS_Face.hxx>
#include <TopoDS_Compound.hxx>
#include <TopoDS_Builder.hxx>
#include <Standard_Failure.hxx>
2016-09-02 10:08:40 +00:00
#include <gp_Ax2.hxx>
#include <gp_Circ.hxx>
#include <gp_Dir.hxx>
#include <gp_Pnt.hxx>
#include <Geom_BezierCurve.hxx>
2016-09-02 10:08:40 +00:00
#include <macros.h>
static constexpr double USER_PREC = 1e-4;
static constexpr double USER_ANGLE_PREC = 1e-6;
2021-07-29 11:10:58 +00:00
2016-09-02 10:08:40 +00:00
// minimum PCB thickness in mm (2 microns assumes a very thin polyimide film)
static constexpr double THICKNESS_MIN = 0.002;
2021-07-29 11:10:58 +00:00
2016-09-02 10:08:40 +00:00
// default PCB thickness in mm
static constexpr double THICKNESS_DEFAULT = 1.6;
2021-07-29 11:10:58 +00:00
2016-09-02 10:08:40 +00:00
// nominal offset from the board
static constexpr double BOARD_OFFSET = 0.05;
2021-07-29 11:10:58 +00:00
2016-09-02 10:08:40 +00:00
// min. length**2 below which 2 points are considered coincident
static constexpr double MIN_LENGTH2 = STEPEXPORT_MIN_DISTANCE * STEPEXPORT_MIN_DISTANCE;
2016-09-02 10:08:40 +00:00
// supported file types
enum FormatType
{
FMT_NONE,
FMT_STEP,
FMT_STEPZ,
FMT_IGES,
FMT_EMN,
FMT_IDF,
FMT_WRL,
FMT_WRZ
2016-09-02 10:08:40 +00:00
};
FormatType fileType( const char* aFileName )
{
2017-03-09 02:40:41 +00:00
wxFileName lfile( wxString::FromUTF8Unchecked( aFileName ) );
2016-09-02 10:08:40 +00:00
if( !lfile.FileExists() )
{
wxString msg;
2022-02-08 21:47:43 +00:00
msg.Printf( wxT( " * fileType(): no such file: %s\n" ),
wxString::FromUTF8Unchecked( aFileName ) );
2016-09-02 10:08:40 +00:00
ReportMessage( msg );
2016-09-02 10:08:40 +00:00
return FMT_NONE;
}
wxString ext = lfile.GetExt().Lower();
2016-09-02 10:08:40 +00:00
2022-02-08 21:47:43 +00:00
if( ext == wxT( "wrl" ) )
return FMT_WRL;
2022-02-08 21:47:43 +00:00
if( ext == wxT( "wrz" ) )
return FMT_WRZ;
2022-02-08 21:47:43 +00:00
if( ext == wxT( "idf" ) )
2016-09-02 10:08:40 +00:00
return FMT_IDF; // component outline
2022-02-08 21:47:43 +00:00
if( ext == wxT( "emn" ) )
2016-09-02 10:08:40 +00:00
return FMT_EMN; // PCB assembly
2022-02-08 21:47:43 +00:00
if( ext == wxT( "stpz" ) || ext == wxT( "gz" ) )
return FMT_STEPZ;
2017-03-09 02:40:41 +00:00
OPEN_ISTREAM( ifile, aFileName );
2016-09-02 10:08:40 +00:00
2017-03-09 02:40:41 +00:00
if( ifile.fail() )
2016-09-02 10:08:40 +00:00
return FMT_NONE;
char iline[82];
memset( iline, 0, 82 );
ifile.getline( iline, 82 );
2017-03-09 02:40:41 +00:00
CLOSE_STREAM( ifile );
2016-09-02 10:08:40 +00:00
iline[81] = 0; // ensure NULL termination when string is too long
// check for STEP in Part 21 format
// (this can give false positives since Part 21 is not exclusively STEP)
if( !strncmp( iline, "ISO-10303-21;", 13 ) )
return FMT_STEP;
std::string fstr = iline;
// check for STEP in XML format
// (this can give both false positive and false negatives)
if( fstr.find( "urn:oid:1.0.10303." ) != std::string::npos )
return FMT_STEP;
// Note: this is a very simple test which can yield false positives; the only
// sure method for determining if a file *not* an IGES model is to attempt
// to load it.
if( iline[72] == 'S' && ( iline[80] == 0 || iline[80] == 13 || iline[80] == 10 ) )
return FMT_IGES;
return FMT_NONE;
}
STEP_PCB_MODEL::STEP_PCB_MODEL( const wxString& aPcbName )
2016-09-02 10:08:40 +00:00
{
m_app = XCAFApp_Application::GetApplication();
m_app->NewDocument( "MDTV-XCAF", m_doc );
m_assy = XCAFDoc_DocumentTool::ShapeTool ( m_doc->Main() );
m_assy_label = m_assy->NewShape();
m_hasPCB = false;
m_components = 0;
m_precision = USER_PREC;
m_angleprec = USER_ANGLE_PREC;
m_thickness = THICKNESS_DEFAULT;
m_minDistance2 = MIN_LENGTH2;
2016-09-02 10:08:40 +00:00
m_minx = 1.0e10; // absurdly large number; any valid PCB X value will be smaller
m_pcbName = aPcbName;
BRepBuilderAPI::Precision( STEPEXPORT_MIN_DISTANCE );
2022-11-14 08:02:40 +00:00
m_maxError = 5000; // 5 microns
2016-09-02 10:08:40 +00:00
}
STEP_PCB_MODEL::~STEP_PCB_MODEL()
2016-09-02 10:08:40 +00:00
{
m_doc->Close();
}
2021-07-29 11:10:58 +00:00
bool STEP_PCB_MODEL::AddPadHole( const PAD* aPad, const VECTOR2D& aOrigin )
2016-09-02 10:08:40 +00:00
{
if( NULL == aPad || !aPad->GetDrillSize().x )
2016-09-02 10:08:40 +00:00
return false;
VECTOR2I pos = aPad->GetPosition();
2016-09-02 10:08:40 +00:00
if( aPad->GetDrillShape() == PAD_DRILL_SHAPE_CIRCLE )
2022-07-06 21:32:52 +00:00
{
TopoDS_Shape s =
BRepPrimAPI_MakeCylinder( pcbIUScale.IUTomm( aPad->GetDrillSize().x ) * 0.5, m_thickness * 2.0 ).Shape();
2016-09-02 10:08:40 +00:00
gp_Trsf shift;
shift.SetTranslation( gp_Vec( pcbIUScale.IUTomm( pos.x - aOrigin.x ),
-pcbIUScale.IUTomm( pos.y - aOrigin.y ),
-m_thickness * 0.5 ) );
2016-09-02 10:08:40 +00:00
BRepBuilderAPI_Transform hole( s, shift );
m_cutouts.push_back( hole.Shape() );
return true;
}
// slotted hole
SHAPE_POLY_SET holeOutlines;
if( !aPad->TransformHoleToPolygon( holeOutlines, 0, m_maxError, ERROR_INSIDE ) )
{
return false;
}
2016-09-02 10:08:40 +00:00
TopoDS_Shape hole;
2016-09-02 10:08:40 +00:00
if( holeOutlines.OutlineCount() > 0 )
2016-09-02 10:08:40 +00:00
{
if( MakeShape( hole, holeOutlines.COutline( 0 ), m_thickness, aOrigin ) )
{
m_cutouts.push_back( hole );
}
2016-09-02 10:08:40 +00:00
}
else
2016-09-02 10:08:40 +00:00
{
return false;
2016-09-02 10:08:40 +00:00
}
return true;
2016-09-02 10:08:40 +00:00
}
bool STEP_PCB_MODEL::AddComponent( const std::string& aFileNameUTF8, const std::string& aRefDes,
bool aBottom, VECTOR2D aPosition, double aRotation, VECTOR3D aOffset,
VECTOR3D aOrientation, VECTOR3D aScale, bool aSubstituteModels )
2016-09-02 10:08:40 +00:00
{
if( aFileNameUTF8.empty() )
{
2022-02-08 21:47:43 +00:00
ReportMessage( wxString::Format( wxT( "No model defined for component %s.\n" ), aRefDes ) );
return false;
}
wxString fileName( wxString::FromUTF8( aFileNameUTF8.c_str() ) );
2022-02-08 21:47:43 +00:00
ReportMessage( wxString::Format( wxT( "Add component %s.\n" ), aRefDes ) );
2016-09-02 10:08:40 +00:00
// first retrieve a label
TDF_Label lmodel;
wxString errorMessage;
2016-09-02 10:08:40 +00:00
if( !getModelLabel( aFileNameUTF8, aScale, lmodel, aSubstituteModels, &errorMessage ) )
2016-09-02 10:08:40 +00:00
{
if( errorMessage.IsEmpty() )
2022-02-08 21:47:43 +00:00
ReportMessage( wxString::Format( wxT( "No model for filename '%s'.\n" ), fileName ) );
else
ReportMessage( errorMessage );
2016-09-02 10:08:40 +00:00
return false;
}
// calculate the Location transform
TopLoc_Location toploc;
if( !getModelLocation( aBottom, aPosition, aRotation, aOffset, aOrientation, toploc ) )
{
ReportMessage(
wxString::Format( wxT( "No location data for filename '%s'.\n" ), fileName ) );
2016-09-02 10:08:40 +00:00
return false;
}
// add the located sub-assembly
TDF_Label llabel = m_assy->AddComponent( m_assy_label, lmodel, toploc );
if( llabel.IsNull() )
{
2022-02-08 21:47:43 +00:00
ReportMessage( wxString::Format( wxT( "Could not add component with filename '%s'.\n" ),
fileName ) );
2016-09-02 10:08:40 +00:00
return false;
}
// attach the RefDes name
TCollection_ExtendedString refdes( aRefDes.c_str() );
TDataStd_Name::Set( llabel, refdes );
return true;
}
void STEP_PCB_MODEL::SetPCBThickness( double aThickness )
2016-09-02 10:08:40 +00:00
{
if( aThickness < 0.0 )
m_thickness = THICKNESS_DEFAULT;
else if( aThickness < THICKNESS_MIN )
m_thickness = THICKNESS_MIN;
else
m_thickness = aThickness;
}
void STEP_PCB_MODEL::SetBoardColor( double r, double g, double b )
{
m_boardColor[0] = r;
m_boardColor[1] = g;
m_boardColor[2] = b;
}
void STEP_PCB_MODEL::SetMinDistance( double aDistance )
{
2022-10-07 16:53:31 +00:00
// Ensure a minimal value (in mm)
aDistance = std::max( aDistance, STEPEXPORT_MIN_ACCEPTABLE_DISTANCE );
2022-10-07 16:53:31 +00:00
// m_minDistance2 keeps a squared distance value
m_minDistance2 = aDistance * aDistance;
BRepBuilderAPI::Precision( aDistance );
}
bool STEP_PCB_MODEL::isBoardOutlineValid()
{
return m_pcb_labels.size() > 0;
}
bool STEP_PCB_MODEL::MakeShape( TopoDS_Shape& aShape, const SHAPE_LINE_CHAIN& aChain,
double aThickness, const VECTOR2D& aOrigin )
2016-09-02 10:08:40 +00:00
{
if( !aShape.IsNull() )
return false; // there is already data in the shape object
2016-09-02 10:08:40 +00:00
if( !aChain.IsClosed() )
return false; // the loop is not closed
2016-09-02 10:08:40 +00:00
BRepBuilderAPI_MakeWire wire;
TopoDS_Edge edge;
bool success = true;
gp_Pnt start = gp_Pnt( pcbIUScale.IUTomm( aChain.CPoint( 0 ).x - aOrigin.x ),
-pcbIUScale.IUTomm( aChain.CPoint( 0 ).y - aOrigin.y ), 0.0 );
for( int j = 0; j < aChain.PointCount(); j++ )
2016-09-02 10:08:40 +00:00
{
int next = j+1;
gp_Pnt end;
if( next >= aChain.PointCount() )
2016-09-02 10:08:40 +00:00
{
end = gp_Pnt( pcbIUScale.IUTomm( aChain.CPoint( 0 ).x - aOrigin.x ),
-pcbIUScale.IUTomm( aChain.CPoint( 0 ).y - aOrigin.y ), 0.0 );
}
else
{
end = gp_Pnt( pcbIUScale.IUTomm( aChain.CPoint( next ).x - aOrigin.x ),
-pcbIUScale.IUTomm( aChain.CPoint( next ).y - aOrigin.y ), 0.0 );
}
2016-09-02 10:08:40 +00:00
// Do not export very small segments: they can create broken outlines
double seg_len = std::abs( end.X() - start.X()) + std::abs(start.Y() - end.Y() );
double min_len = 0.0001; // In mm, i.e. 0.1 micron
if( seg_len < min_len )
continue;
try
{
edge = BRepBuilderAPI_MakeEdge( start, end );
wire.Add( edge );
2016-09-02 10:08:40 +00:00
if( BRepBuilderAPI_WireDone != wire.Error() )
2016-09-02 10:08:40 +00:00
{
ReportMessage( wxT( "failed to add curve\n" ) );
return false;
2016-09-02 10:08:40 +00:00
}
}
catch( const Standard_Failure& e )
{
ReportMessage(
wxString::Format( wxT( "OCC exception: %s\n" ), e.GetMessageString() ) );
success = false;
}
2016-09-02 10:08:40 +00:00
if( !success )
{
ReportMessage( wxS( "failed to add edge\n" ) );
return false;
2016-09-02 10:08:40 +00:00
}
start = end;
}
2016-09-02 10:08:40 +00:00
BRepBuilderAPI_MakeFace face;
try
{
face = BRepBuilderAPI_MakeFace( wire );
}
catch( const Standard_Failure& e )
{
ReportMessage(
wxString::Format( wxT( "OCC exception: %s\n" ), e.GetMessageString() ) );
return false;
}
2016-09-02 10:08:40 +00:00
aShape = BRepPrimAPI_MakePrism( face, gp_Vec( 0, 0, aThickness ) );
2016-09-02 10:08:40 +00:00
if( aShape.IsNull() )
{
ReportMessage( wxT( "failed to create a prismatic shape\n" ) );
return false;
}
2016-09-02 10:08:40 +00:00
return true;
}
bool STEP_PCB_MODEL::CreatePCB( SHAPE_POLY_SET& aOutline, VECTOR2D aOrigin )
{
if( m_hasPCB )
{
if( !isBoardOutlineValid() )
return false;
2016-09-02 10:08:40 +00:00
return true;
2016-09-02 10:08:40 +00:00
}
m_hasPCB = true; // whether or not operations fail we note that CreatePCB has been invoked
// Support for more than one main outline (more than one board)
std::vector<TopoDS_Shape> board_outlines;
for( int cnt = 0; cnt < aOutline.OutlineCount(); cnt++ )
2016-09-02 10:08:40 +00:00
{
const SHAPE_LINE_CHAIN& outline = aOutline.COutline( cnt );
ReportMessage( wxString::Format( wxT( "Build board main outline %d with %d points.\n" ),
cnt+1, outline.PointCount() ) );
TopoDS_Shape curr_brd;
if( !MakeShape( curr_brd, outline, m_thickness, aOrigin ) )
2016-09-02 10:08:40 +00:00
{
// Error
ReportMessage( wxString::Format(
wxT( "OCC error adding main outline polygon %d with %d points.\n" ),
cnt+1, outline.PointCount() ) );
2016-09-02 10:08:40 +00:00
}
else
board_outlines.push_back( curr_brd );
// Generate board cutouts in current main outline:
if( aOutline.HoleCount( cnt ) > 0 )
{
ReportMessage( wxString::Format( wxT( "Add cutouts in outline %d (%d cutout(s)).\n" ),
cnt+1, aOutline.HoleCount( cnt ) ) );
}
for( int ii = 0; ii < aOutline.HoleCount( cnt ); ii++ )
2016-09-02 10:08:40 +00:00
{
const SHAPE_LINE_CHAIN& holeOutline = aOutline.Hole( cnt, ii );
2016-09-02 10:08:40 +00:00
TopoDS_Shape hole;
if( MakeShape( hole, holeOutline, m_thickness, aOrigin ) )
2016-09-02 10:08:40 +00:00
{
m_cutouts.push_back( hole );
}
}
}
2016-09-02 10:08:40 +00:00
// subtract cutouts (if any)
if( m_cutouts.size() )
2021-07-06 20:22:06 +00:00
{
ReportMessage( wxString::Format( wxT( "Build board cutouts and holes (%d hole(s)).\n" ),
2022-02-08 21:47:43 +00:00
(int) m_cutouts.size() ) );
TopTools_ListOfShape holelist;
2022-11-14 08:02:40 +00:00
for( TopoDS_Shape& hole : m_cutouts )
holelist.Append( hole );
// Remove holes for each board (usually there is only one board
for( TopoDS_Shape& board: board_outlines )
{
BRepAlgoAPI_Cut Cut;
TopTools_ListOfShape mainbrd;
mainbrd.Append( board );
Cut.SetArguments( mainbrd );
Cut.SetTools( holelist );
Cut.Build();
board = Cut.Shape();
}
}
2016-09-02 10:08:40 +00:00
// push the board to the data structure
2022-02-08 21:47:43 +00:00
ReportMessage( wxT( "\nGenerate board full shape.\n" ) );
2021-09-14 01:17:22 +00:00
// Dont expand the component or else coloring it gets hard
for( TopoDS_Shape& board: board_outlines )
{
m_pcb_labels.push_back( m_assy->AddComponent( m_assy_label, board, false ) );
2016-09-02 10:08:40 +00:00
if( m_pcb_labels.back().IsNull() )
return false;
}
2016-09-02 10:08:40 +00:00
2022-07-06 21:32:52 +00:00
// AddComponent adds a label that has a reference (not a parent/child relation) to the real
// label. We need to extract that real label to name it for the STEP output cleanly
// Why are we trying to name the bare board? Because CAD tools like SolidWorks do fun things
2022-07-06 21:32:52 +00:00
// like "deduplicate" imported STEPs by swapping STEP assembly components with already
// identically named assemblies. So we want to avoid having the PCB be generally defaulted
// to "Component" or "Assembly".
// color the PCB
Handle( XCAFDoc_ColorTool ) colorTool = XCAFDoc_DocumentTool::ColorTool( m_doc->Main() );
Quantity_Color color( m_boardColor[0], m_boardColor[1], m_boardColor[2], Quantity_TOC_RGB );
int pcbIdx = 1;
for( TDF_Label& pcb_label : m_pcb_labels )
{
colorTool->SetColor( pcb_label, color, XCAFDoc_ColorSurf );
2022-07-06 21:32:52 +00:00
Handle( TDataStd_TreeNode ) node;
if( pcb_label.FindAttribute( XCAFDoc::ShapeRefGUID(), node ) )
{
// Gives a name to each board object
TDF_Label label = node->Father()->Label();
if( !label.IsNull() )
{
wxString pcbName;
if( m_pcb_labels.size() == 1 )
pcbName = wxT( "PCB" );
else
pcbName = wxString::Format( wxT( "PCB%d" ), pcbIdx++ );
2016-09-02 10:08:40 +00:00
std::string pcbNameStdString( pcbName.ToUTF8() );
TCollection_ExtendedString partname( pcbNameStdString.c_str() );
TDataStd_Name::Set( label, partname );
}
}
2016-09-02 10:08:40 +00:00
TopExp_Explorer topex;
// color the PCB
topex.Init( m_assy->GetShape( pcb_label ), TopAbs_SOLID );
while( topex.More() )
{
colorTool->SetColor( topex.Current(), color, XCAFDoc_ColorSurf );
topex.Next();
}
2016-09-02 10:08:40 +00:00
}
#if( defined OCC_VERSION_HEX ) && ( OCC_VERSION_HEX > 0x070101 )
m_assy->UpdateAssemblies();
#endif
2022-07-06 21:32:52 +00:00
2016-09-02 10:08:40 +00:00
return true;
}
#ifdef SUPPORTS_IGES
// write the assembly model in IGES format
bool STEP_PCB_MODEL::WriteIGES( const wxString& aFileName )
2016-09-02 10:08:40 +00:00
{
if( !isBoardOutlineValid() )
2016-09-02 10:08:40 +00:00
{
2022-02-08 21:47:43 +00:00
ReportMessage( wxString::Format( wxT( "No valid PCB assembly; cannot create output file "
"'%s'.\n" ),
2021-07-06 20:22:06 +00:00
aFileName ) );
2016-09-02 10:08:40 +00:00
return false;
}
wxFileName fn( aFileName );
IGESControl_Controller::Init();
IGESCAFControl_Writer writer;
writer.SetColorMode( Standard_True );
writer.SetNameMode( Standard_True );
IGESData_GlobalSection header = writer.Model()->GlobalSection();
header.SetFileName( new TCollection_HAsciiString( fn.GetFullName().ToAscii() ) );
2016-09-02 10:08:40 +00:00
header.SetSendName( new TCollection_HAsciiString( "KiCad electronic assembly" ) );
2022-07-06 21:32:52 +00:00
header.SetAuthorName(
new TCollection_HAsciiString( Interface_Static::CVal( "write.iges.header.author" ) ) );
header.SetCompanyName(
new TCollection_HAsciiString( Interface_Static::CVal( "write.iges.header.company" ) ) );
2016-09-02 10:08:40 +00:00
writer.Model()->SetGlobalSection( header );
if( Standard_False == writer.Perform( m_doc, aFileName.c_str() ) )
return false;
return true;
}
#endif
bool STEP_PCB_MODEL::WriteSTEP( const wxString& aFileName )
2016-09-02 10:08:40 +00:00
{
if( !isBoardOutlineValid() )
2016-09-02 10:08:40 +00:00
{
2022-02-08 21:47:43 +00:00
ReportMessage( wxString::Format( wxT( "No valid PCB assembly; cannot create output file "
"'%s'.\n" ),
2021-07-06 20:22:06 +00:00
aFileName ) );
2016-09-02 10:08:40 +00:00
return false;
}
2022-07-06 21:32:52 +00:00
wxFileName fn( aFileName );
2016-09-02 10:08:40 +00:00
STEPCAFControl_Writer writer;
writer.SetColorMode( Standard_True );
writer.SetNameMode( Standard_True );
// This must be set before we "transfer" the document.
// Should default to kicad_pcb.general.title_block.title,
// but in the meantime, defaulting to the basename of the output
// target is still better than "open cascade step translter v..."
// UTF8 should be ok from ISO 10303-21:2016, but... older stuff? use boring ascii
2021-07-06 20:22:06 +00:00
if( !Interface_Static::SetCVal( "write.step.product.name", fn.GetName().ToAscii() ) )
2022-02-08 21:47:43 +00:00
ReportMessage( wxT( "Failed to set step product name, but will attempt to continue." ) );
2016-09-02 10:08:40 +00:00
if( Standard_False == writer.Transfer( m_doc, STEPControl_AsIs ) )
return false;
APIHeaderSection_MakeHeader hdr( writer.ChangeWriter().Model() );
2021-07-29 11:10:58 +00:00
// Note: use only Ascii7 chars, non Ascii7 chars (therefore UFT8 chars)
// are creating issues in the step file
hdr.SetName( new TCollection_HAsciiString( fn.GetFullName().ToAscii() ) );
2021-07-29 11:10:58 +00:00
2016-09-02 10:08:40 +00:00
// TODO: how to control and ensure consistency with IGES?
hdr.SetAuthorValue( 1, new TCollection_HAsciiString( "Pcbnew" ) );
hdr.SetOrganizationValue( 1, new TCollection_HAsciiString( "Kicad" ) );
2016-09-02 10:08:40 +00:00
hdr.SetOriginatingSystem( new TCollection_HAsciiString( "KiCad to STEP converter" ) );
hdr.SetDescriptionValue( 1, new TCollection_HAsciiString( "KiCad electronic assembly" ) );
bool success = true;
2021-07-29 11:10:58 +00:00
// Creates a temporary file with a ascii7 name, because writer does not know unicode filenames.
wxString currCWD = wxGetCwd();
2021-07-29 11:10:58 +00:00
wxString workCWD = fn.GetPath();
if( !workCWD.IsEmpty() )
wxSetWorkingDirectory( workCWD );
char tmpfname[] = "$tempfile$.step";
2016-09-02 10:08:40 +00:00
if( Standard_False == writer.Write( tmpfname ) )
success = false;
if( success )
{
if( !wxRenameFile( tmpfname, fn.GetFullName(), true ) )
{
2022-02-08 21:47:43 +00:00
ReportMessage( wxString::Format( wxT( "Cannot rename temporary file '%s' to '%s'.\n" ),
2021-07-06 20:22:06 +00:00
tmpfname,
fn.GetFullName() ) );
success = false;
}
}
wxSetWorkingDirectory( currCWD );
return success;
2016-09-02 10:08:40 +00:00
}
bool STEP_PCB_MODEL::getModelLabel( const std::string& aFileNameUTF8, VECTOR3D aScale, TDF_Label& aLabel,
bool aSubstituteModels, wxString* aErrorMessage )
2016-09-02 10:08:40 +00:00
{
std::string model_key = aFileNameUTF8 + "_" + std::to_string( aScale.x )
+ "_" + std::to_string( aScale.y ) + "_" + std::to_string( aScale.z );
MODEL_MAP::const_iterator mm = m_models.find( model_key );
2016-09-02 10:08:40 +00:00
if( mm != m_models.end() )
{
aLabel = mm->second;
return true;
}
aLabel.Nullify();
Handle( TDocStd_Document ) doc;
m_app->NewDocument( "MDTV-XCAF", doc );
wxString fileName( wxString::FromUTF8( aFileNameUTF8.c_str() ) );
FormatType modelFmt = fileType( aFileNameUTF8.c_str() );
2016-09-02 10:08:40 +00:00
switch( modelFmt )
{
2021-07-29 11:10:58 +00:00
case FMT_IGES:
if( !readIGES( doc, aFileNameUTF8.c_str() ) )
2021-07-29 11:10:58 +00:00
{
2022-02-08 21:47:43 +00:00
ReportMessage( wxString::Format( wxT( "readIGES() failed on filename '%s'.\n" ),
fileName ) );
2021-07-29 11:10:58 +00:00
return false;
}
break;
2016-09-02 10:08:40 +00:00
2021-07-29 11:10:58 +00:00
case FMT_STEP:
if( !readSTEP( doc, aFileNameUTF8.c_str() ) )
{
2022-02-08 21:47:43 +00:00
ReportMessage( wxString::Format( wxT( "readSTEP() failed on filename '%s'.\n" ),
fileName ) );
2021-07-29 11:10:58 +00:00
return false;
}
break;
2021-07-29 11:10:58 +00:00
case FMT_STEPZ:
{
// To export a compressed step file (.stpz or .stp.gz file), the best way is to
// decaompress it in a temporaty file and load this temporary file
wxFFileInputStream ifile( fileName );
wxFileName outFile( fileName );
2021-07-29 11:10:58 +00:00
outFile.SetPath( wxStandardPaths::Get().GetTempDir() );
2022-02-08 21:47:43 +00:00
outFile.SetExt( wxT( "step" ) );
2021-07-29 11:10:58 +00:00
wxFileOffset size = ifile.GetLength();
2021-07-29 11:10:58 +00:00
if( size == wxInvalidOffset )
{
2022-02-08 21:47:43 +00:00
ReportMessage( wxString::Format( wxT( "getModelLabel() failed on filename '%s'.\n" ),
fileName ) );
2021-07-29 11:10:58 +00:00
return false;
}
2021-07-29 11:10:58 +00:00
{
bool success = false;
wxFFileOutputStream ofile( outFile.GetFullPath() );
2021-07-29 11:10:58 +00:00
if( !ofile.IsOk() )
return false;
2021-07-29 11:10:58 +00:00
char* buffer = new char[size];
2021-07-29 11:10:58 +00:00
ifile.Read( buffer, size );
std::string expanded;
2021-07-29 11:10:58 +00:00
try
{
expanded = gzip::decompress( buffer, size );
success = true;
}
catch( ... )
{
2022-02-08 21:47:43 +00:00
ReportMessage( wxString::Format( wxT( "failed to decompress '%s'.\n" ),
fileName ) );
2021-07-29 11:10:58 +00:00
}
2021-07-29 11:10:58 +00:00
if( expanded.empty() )
{
ifile.Reset();
ifile.SeekI( 0 );
wxZipInputStream izipfile( ifile );
std::unique_ptr<wxZipEntry> zip_file( izipfile.GetNextEntry() );
if( zip_file && !zip_file->IsDir() && izipfile.CanRead() )
{
2021-07-29 11:10:58 +00:00
izipfile.Read( ofile );
success = true;
}
2021-07-29 11:10:58 +00:00
}
else
{
ofile.Write( expanded.data(), expanded.size() );
}
2021-07-29 11:10:58 +00:00
delete[] buffer;
ofile.Close();
if( success )
{
std::string altFileNameUTF8 = TO_UTF8( outFile.GetFullPath() );
success =
getModelLabel( altFileNameUTF8, VECTOR3D( 1.0, 1.0, 1.0 ), aLabel, false );
}
2021-07-29 11:10:58 +00:00
return success;
}
2021-07-29 11:10:58 +00:00
break;
}
case FMT_WRL:
case FMT_WRZ:
/* WRL files are preferred for internal rendering, due to superior material properties, etc.
* However they are not suitable for MCAD export.
*
* If a .wrl file is specified, attempt to locate a replacement file for it.
*
* If a valid replacement file is found, the label for THAT file will be associated with
* the .wrl file
*/
if( aSubstituteModels )
{
wxFileName wrlName( fileName );
2021-07-29 11:10:58 +00:00
wxString basePath = wrlName.GetPath();
wxString baseName = wrlName.GetName();
// List of alternate files to look for
// Given in order of preference
// (Break if match is found)
wxArrayString alts;
// Step files
2022-02-08 21:47:43 +00:00
alts.Add( wxT( "stp" ) );
alts.Add( wxT( "step" ) );
alts.Add( wxT( "STP" ) );
alts.Add( wxT( "STEP" ) );
alts.Add( wxT( "Stp" ) );
alts.Add( wxT( "Step" ) );
alts.Add( wxT( "stpz" ) );
alts.Add( wxT( "stpZ" ) );
alts.Add( wxT( "STPZ" ) );
alts.Add( wxT( "step.gz" ) );
alts.Add( wxT( "stp.gz" ) );
2021-07-29 11:10:58 +00:00
// IGES files
2022-02-08 21:47:43 +00:00
alts.Add( wxT( "iges" ) );
alts.Add( wxT( "IGES" ) );
alts.Add( wxT( "igs" ) );
alts.Add( wxT( "IGS" ) );
2021-07-29 11:10:58 +00:00
//TODO - Other alternative formats?
for( const auto& alt : alts )
{
2022-02-08 21:47:43 +00:00
wxFileName altFile( basePath, baseName + wxT( "." ) + alt );
2021-07-29 11:10:58 +00:00
if( altFile.IsOk() && altFile.FileExists() )
{
std::string altFileNameUTF8 = TO_UTF8( altFile.GetFullPath() );
2021-07-29 11:10:58 +00:00
// When substituting a STEP/IGS file for VRML, do not apply the VRML scaling
// to the new STEP model. This process of auto-substitution is janky as all
// heck so let's not mix up un-displayed scale factors with potentially
// mis-matched files. And hope that the user doesn't have multiples files
// named "model.wrl" and "model.stp" referring to different parts.
// TODO: Fix model handling in v7. Default models should only be STP.
// Have option to override this in DISPLAY.
if( getModelLabel( altFileNameUTF8, VECTOR3D( 1.0, 1.0, 1.0 ), aLabel, false ) )
{
2021-07-29 11:10:58 +00:00
return true;
}
}
}
2021-07-29 11:10:58 +00:00
return false; // No replacement model found
}
else // Substitution is not allowed
{
if( aErrorMessage )
2022-02-08 21:47:43 +00:00
aErrorMessage->Printf( wxT( "Cannot add a VRML model to a STEP file.\n" ) );
2021-07-29 11:10:58 +00:00
return false;
}
break;
2016-09-02 10:08:40 +00:00
// TODO: implement IDF and EMN converters
2021-07-29 11:10:58 +00:00
default:
return false;
2016-09-02 10:08:40 +00:00
}
aLabel = transferModel( doc, m_doc, aScale );
2016-09-02 10:08:40 +00:00
if( aLabel.IsNull() )
{
2022-02-08 21:47:43 +00:00
ReportMessage( wxString::Format( wxT( "Could not transfer model data from file '%s'.\n" ),
fileName ) );
2016-09-02 10:08:40 +00:00
return false;
}
// attach the PART NAME ( base filename: note that in principle
// different models may have the same base filename )
wxFileName afile( fileName );
2016-09-02 10:08:40 +00:00
std::string pname( afile.GetName().ToUTF8() );
TCollection_ExtendedString partname( pname.c_str() );
TDataStd_Name::Set( aLabel, partname );
m_models.insert( MODEL_DATUM( model_key, aLabel ) );
2016-09-02 10:08:40 +00:00
++m_components;
return true;
}
bool STEP_PCB_MODEL::getModelLocation( bool aBottom, VECTOR2D aPosition, double aRotation, VECTOR3D aOffset, VECTOR3D aOrientation,
TopLoc_Location& aLocation )
2016-09-02 10:08:40 +00:00
{
// Order of operations:
// a. aOrientation is applied -Z*-Y*-X
// b. aOffset is applied
// Top ? add thickness to the Z offset
// c. Bottom ? Rotate on X axis (in contrast to most ECAD which mirror on Y),
// then rotate on +Z
// Top ? rotate on -Z
// d. aPosition is applied
//
// Note: Y axis is inverted in KiCad
gp_Trsf lPos;
lPos.SetTranslation( gp_Vec( aPosition.x, -aPosition.y, 0.0 ) );
// Offset board thickness
aOffset.z += BOARD_OFFSET;
2016-09-02 10:08:40 +00:00
gp_Trsf lRot;
if( aBottom )
{
lRot.SetRotation( gp_Ax1( gp_Pnt( 0.0, 0.0, 0.0 ), gp_Dir( 0.0, 0.0, 1.0 ) ), aRotation );
lPos.Multiply( lRot );
lRot.SetRotation( gp_Ax1( gp_Pnt( 0.0, 0.0, 0.0 ), gp_Dir( 1.0, 0.0, 0.0 ) ), M_PI );
lPos.Multiply( lRot );
}
else
{
aOffset.z += m_thickness;
lRot.SetRotation( gp_Ax1( gp_Pnt( 0.0, 0.0, 0.0 ), gp_Dir( 0.0, 0.0, 1.0 ) ), aRotation );
lPos.Multiply( lRot );
}
gp_Trsf lOff;
lOff.SetTranslation( gp_Vec( aOffset.x, aOffset.y, aOffset.z ) );
lPos.Multiply( lOff );
gp_Trsf lOrient;
2021-09-04 17:05:30 +00:00
lOrient.SetRotation( gp_Ax1( gp_Pnt( 0.0, 0.0, 0.0 ), gp_Dir( 0.0, 0.0, 1.0 ) ),
-aOrientation.z );
2016-09-02 10:08:40 +00:00
lPos.Multiply( lOrient );
2021-09-04 17:05:30 +00:00
lOrient.SetRotation( gp_Ax1( gp_Pnt( 0.0, 0.0, 0.0 ), gp_Dir( 0.0, 1.0, 0.0 ) ),
-aOrientation.y );
2016-09-02 10:08:40 +00:00
lPos.Multiply( lOrient );
2021-09-04 17:05:30 +00:00
lOrient.SetRotation( gp_Ax1( gp_Pnt( 0.0, 0.0, 0.0 ), gp_Dir( 1.0, 0.0, 0.0 ) ),
-aOrientation.x );
2016-09-02 10:08:40 +00:00
lPos.Multiply( lOrient );
aLocation = TopLoc_Location( lPos );
return true;
}
bool STEP_PCB_MODEL::readIGES( Handle( TDocStd_Document )& doc, const char* fname )
2016-09-02 10:08:40 +00:00
{
IGESControl_Controller::Init();
IGESCAFControl_Reader reader;
IFSelect_ReturnStatus stat = reader.ReadFile( fname );
if( stat != IFSelect_RetDone )
return false;
// Enable user-defined shape precision
if( !Interface_Static::SetIVal( "read.precision.mode", 1 ) )
return false;
// Set the shape conversion precision to USER_PREC (default 0.0001 has too many triangles)
if( !Interface_Static::SetRVal( "read.precision.val", USER_PREC ) )
return false;
// set other translation options
2022-07-06 21:32:52 +00:00
reader.SetColorMode( true ); // use model colors
reader.SetNameMode( false ); // don't use IGES label names
reader.SetLayerMode( false ); // ignore LAYER data
2016-09-02 10:08:40 +00:00
2022-07-06 21:32:52 +00:00
if( !reader.Transfer( doc ) )
2016-09-02 10:08:40 +00:00
{
doc->Close();
return false;
}
// are there any shapes to translate?
if( reader.NbShapes() < 1 )
{
doc->Close();
return false;
}
return true;
}
bool STEP_PCB_MODEL::readSTEP( Handle( TDocStd_Document )& doc, const char* fname )
2016-09-02 10:08:40 +00:00
{
STEPCAFControl_Reader reader;
IFSelect_ReturnStatus stat = reader.ReadFile( fname );
if( stat != IFSelect_RetDone )
return false;
// Enable user-defined shape precision
if( !Interface_Static::SetIVal( "read.precision.mode", 1 ) )
return false;
// Set the shape conversion precision to USER_PREC (default 0.0001 has too many triangles)
if( !Interface_Static::SetRVal( "read.precision.val", USER_PREC ) )
return false;
// set other translation options
2022-07-06 21:32:52 +00:00
reader.SetColorMode( true ); // use model colors
reader.SetNameMode( false ); // don't use label names
reader.SetLayerMode( false ); // ignore LAYER data
2016-09-02 10:08:40 +00:00
2022-07-06 21:32:52 +00:00
if( !reader.Transfer( doc ) )
2016-09-02 10:08:40 +00:00
{
doc->Close();
return false;
}
// are there any shapes to translate?
if( reader.NbRootsForTransfer() < 1 )
{
doc->Close();
return false;
}
return true;
}
TDF_Label STEP_PCB_MODEL::transferModel( Handle( TDocStd_Document )& source,
Handle( TDocStd_Document )& dest, VECTOR3D aScale )
2016-09-02 10:08:40 +00:00
{
// transfer data from Source into a top level component of Dest
gp_GTrsf scale_transform;
scale_transform.SetVectorialPart( gp_Mat( aScale.x, 0, 0,
0, aScale.y, 0,
0, 0, aScale.z ) );
BRepBuilderAPI_GTransform brep( scale_transform );
2016-09-02 10:08:40 +00:00
// s_assy = shape tool for the source
2022-07-06 21:32:52 +00:00
Handle(XCAFDoc_ShapeTool) s_assy = XCAFDoc_DocumentTool::ShapeTool( source->Main() );
2016-09-02 10:08:40 +00:00
// retrieve all free shapes within the assembly
TDF_LabelSequence frshapes;
s_assy->GetFreeShapes( frshapes );
// d_assy = shape tool for the destination
2022-07-06 21:32:52 +00:00
Handle( XCAFDoc_ShapeTool ) d_assy = XCAFDoc_DocumentTool::ShapeTool ( dest->Main() );
2016-09-02 10:08:40 +00:00
// create a new shape within the destination and set the assembly tool to point to it
TDF_Label component = d_assy->NewShape();
int nshapes = frshapes.Length();
int id = 1;
Handle( XCAFDoc_ColorTool ) scolor = XCAFDoc_DocumentTool::ColorTool( source->Main() );
Handle( XCAFDoc_ColorTool ) dcolor = XCAFDoc_DocumentTool::ColorTool( dest->Main() );
TopExp_Explorer dtop;
TopExp_Explorer stop;
while( id <= nshapes )
{
2022-07-06 21:32:52 +00:00
TopoDS_Shape shape = s_assy->GetShape( frshapes.Value( id ) );
2016-09-02 10:08:40 +00:00
2022-07-06 21:32:52 +00:00
if( !shape.IsNull() )
2016-09-02 10:08:40 +00:00
{
TopoDS_Shape scaled_shape( shape );
if( aScale.x != 1.0 || aScale.y != 1.0 || aScale.z != 1.0 )
{
brep.Perform( shape, Standard_False );
if( brep.IsDone() )
{
scaled_shape = brep.Shape();
}
else
{
ReportMessage( wxT( " * transfertModel(): failed to scale model\n" ) );
scaled_shape = shape;
}
}
TDF_Label niulab = d_assy->AddComponent( component, scaled_shape, Standard_False );
2016-09-02 10:08:40 +00:00
// check for per-surface colors
stop.Init( shape, TopAbs_FACE );
dtop.Init( d_assy->GetShape( niulab ), TopAbs_FACE );
while( stop.More() && dtop.More() )
{
Quantity_Color face_color;
TDF_Label tl;
// give priority to the base shape's color
if( s_assy->FindShape( stop.Current(), tl ) )
{
if( scolor->GetColor( tl, XCAFDoc_ColorSurf, face_color )
|| scolor->GetColor( tl, XCAFDoc_ColorGen, face_color )
|| scolor->GetColor( tl, XCAFDoc_ColorCurv, face_color ) )
{
dcolor->SetColor( dtop.Current(), face_color, XCAFDoc_ColorSurf );
}
}
2022-07-06 21:32:52 +00:00
else if( scolor->GetColor( stop.Current(), XCAFDoc_ColorSurf, face_color )
|| scolor->GetColor( stop.Current(), XCAFDoc_ColorGen, face_color )
|| scolor->GetColor( stop.Current(), XCAFDoc_ColorCurv, face_color ) )
2016-09-02 10:08:40 +00:00
{
dcolor->SetColor( dtop.Current(), face_color, XCAFDoc_ColorSurf );
}
stop.Next();
dtop.Next();
}
// check for per-solid colors
stop.Init( shape, TopAbs_SOLID );
dtop.Init( d_assy->GetShape( niulab ), TopAbs_SOLID, TopAbs_FACE );
while( stop.More() && dtop.More() )
{
Quantity_Color face_color;
TDF_Label tl;
// give priority to the base shape's color
if( s_assy->FindShape( stop.Current(), tl ) )
{
if( scolor->GetColor( tl, XCAFDoc_ColorSurf, face_color )
|| scolor->GetColor( tl, XCAFDoc_ColorGen, face_color )
|| scolor->GetColor( tl, XCAFDoc_ColorCurv, face_color ) )
{
dcolor->SetColor( dtop.Current(), face_color, XCAFDoc_ColorGen );
}
}
2022-07-06 21:32:52 +00:00
else if( scolor->GetColor( stop.Current(), XCAFDoc_ColorSurf, face_color )
|| scolor->GetColor( stop.Current(), XCAFDoc_ColorGen, face_color )
|| scolor->GetColor( stop.Current(), XCAFDoc_ColorCurv, face_color ) )
2016-09-02 10:08:40 +00:00
{
dcolor->SetColor( dtop.Current(), face_color, XCAFDoc_ColorSurf );
}
stop.Next();
dtop.Next();
}
}
++id;
};
return component;
}