1234 lines
34 KiB
C++
1234 lines
34 KiB
C++
/*
|
|
* This program source code file is part of KiCad, a free EDA CAD application.
|
|
*
|
|
* Copyright (C) 2016 Cirilo Bernardo <cirilo.bernardo@gmail.com>
|
|
* Copyright (C) 2020-2021 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
|
|
*
|
|
*
|
|
* Some code lifted from FreeCAD, copyright (c) 2018 Zheng, Lei (realthunder) under GPLv2
|
|
*/
|
|
|
|
#include <iostream>
|
|
#include <fstream>
|
|
#include <sstream>
|
|
#include <string>
|
|
#include <cstring>
|
|
#include <map>
|
|
#include <vector>
|
|
#include <wx/filename.h>
|
|
#include <wx/log.h>
|
|
#include <wx/stdpaths.h>
|
|
#include <wx/string.h>
|
|
#include <wx/utils.h>
|
|
#include <wx/wfstream.h>
|
|
#include <wx/zipstrm.h>
|
|
|
|
#include <decompress.hpp>
|
|
|
|
#include <TDocStd_Document.hxx>
|
|
#include <TopoDS.hxx>
|
|
#include <TopoDS_Shape.hxx>
|
|
#include <Quantity_Color.hxx>
|
|
#include <XCAFApp_Application.hxx>
|
|
|
|
#include <AIS_Shape.hxx>
|
|
|
|
#include <IGESControl_Reader.hxx>
|
|
#include <IGESCAFControl_Reader.hxx>
|
|
#include <Interface_Static.hxx>
|
|
|
|
#include <STEPControl_Reader.hxx>
|
|
#include <STEPCAFControl_Reader.hxx>
|
|
|
|
#include <XCAFDoc_DocumentTool.hxx>
|
|
#include <XCAFDoc_ColorTool.hxx>
|
|
#include <XCAFDoc_ShapeTool.hxx>
|
|
|
|
#include <BRep_Tool.hxx>
|
|
#include <BRepMesh_IncrementalMesh.hxx>
|
|
|
|
#include <TopoDS.hxx>
|
|
#include <TopoDS_Shape.hxx>
|
|
#include <TopoDS_Face.hxx>
|
|
#include <TopoDS_Compound.hxx>
|
|
#include <TopExp_Explorer.hxx>
|
|
|
|
#include <Quantity_Color.hxx>
|
|
#include <Poly_Triangulation.hxx>
|
|
#include <Poly_PolygonOnTriangulation.hxx>
|
|
#include <Precision.hxx>
|
|
|
|
#include <TDF_LabelSequence.hxx>
|
|
#include <TDF_ChildIterator.hxx>
|
|
#include <TDF_Tool.hxx>
|
|
#include <TDataStd_Name.hxx>
|
|
#include <Standard_Version.hxx>
|
|
|
|
#include "plugins/3dapi/ifsg_all.h"
|
|
|
|
|
|
// log mask for wxLogTrace
|
|
#define MASK_OCE wxT( "PLUGIN_OCE" )
|
|
#define MASK_OCE_EXTRA wxT( "PLUGIN_OCE_EXTRA" )
|
|
|
|
// precision for mesh creation; 0.07 should be good enough for ECAD viewing
|
|
#define USER_PREC (0.14)
|
|
|
|
// angular deflection for meshing
|
|
// 10 deg (36 faces per circle) = 0.17453293
|
|
// 20 deg (18 faces per circle) = 0.34906585
|
|
// 30 deg (12 faces per circle) = 0.52359878
|
|
#define USER_ANGLE (0.52359878)
|
|
|
|
typedef std::map<std::size_t, SGNODE*> COLORMAP;
|
|
typedef std::map<std::string, SGNODE*> FACEMAP;
|
|
typedef std::map<std::string, std::vector<SGNODE*>> NODEMAP;
|
|
typedef std::pair<std::string, std::vector<SGNODE*>> NODEITEM;
|
|
|
|
struct DATA;
|
|
|
|
bool processLabel( const TDF_Label& aLabel, DATA& aData, SGNODE* aParent,
|
|
std::vector< SGNODE* >* aItems );
|
|
|
|
|
|
bool processFace( const TopoDS_Face& face, DATA& data, SGNODE* parent,
|
|
std::vector< SGNODE* >* items, Quantity_ColorRGBA* color );
|
|
|
|
#if OCC_VERSION_HEX >= 0x070500
|
|
#define OCC_COLOR_SPACE Quantity_TOC_sRGB
|
|
#else
|
|
#define OCC_COLOR_SPACE Quantity_TOC_RGB
|
|
#endif
|
|
|
|
struct DATA
|
|
{
|
|
Handle( TDocStd_Document ) m_doc;
|
|
Handle( XCAFDoc_ColorTool ) m_color;
|
|
Handle( XCAFDoc_ShapeTool ) m_assy;
|
|
SGNODE* scene;
|
|
SGNODE* defaultColor;
|
|
Quantity_Color refColor;
|
|
NODEMAP shapes; // SGNODE lists representing a TopoDS_SOLID / COMPOUND
|
|
COLORMAP colors; // SGAPPEARANCE nodes
|
|
FACEMAP faces; // SGSHAPE items representing a TopoDS_FACE
|
|
bool renderBoth; // set TRUE if we're processing IGES
|
|
bool hasSolid; // set TRUE if there is no parent SOLID
|
|
|
|
DATA()
|
|
{
|
|
scene = nullptr;
|
|
defaultColor = nullptr;
|
|
refColor.SetValues( Quantity_NOC_BLACK );
|
|
renderBoth = false;
|
|
hasSolid = false;
|
|
}
|
|
|
|
~DATA()
|
|
{
|
|
// destroy any colors with no parent
|
|
if( !colors.empty() )
|
|
{
|
|
COLORMAP::iterator sC = colors.begin();
|
|
COLORMAP::iterator eC = colors.end();
|
|
|
|
while( sC != eC )
|
|
{
|
|
if( nullptr == S3D::GetSGNodeParent( sC->second ) )
|
|
S3D::DestroyNode( sC->second );
|
|
|
|
++sC;
|
|
}
|
|
|
|
colors.clear();
|
|
}
|
|
|
|
if( defaultColor && nullptr == S3D::GetSGNodeParent( defaultColor ) )
|
|
S3D::DestroyNode(defaultColor);
|
|
|
|
// destroy any faces with no parent
|
|
if( !faces.empty() )
|
|
{
|
|
FACEMAP::iterator sF = faces.begin();
|
|
FACEMAP::iterator eF = faces.end();
|
|
|
|
while( sF != eF )
|
|
{
|
|
if( nullptr == S3D::GetSGNodeParent( sF->second ) )
|
|
S3D::DestroyNode( sF->second );
|
|
|
|
++sF;
|
|
}
|
|
|
|
faces.clear();
|
|
}
|
|
|
|
// destroy any shapes with no parent
|
|
if( !shapes.empty() )
|
|
{
|
|
NODEMAP::iterator sS = shapes.begin();
|
|
NODEMAP::iterator eS = shapes.end();
|
|
|
|
while( sS != eS )
|
|
{
|
|
std::vector< SGNODE* >::iterator sV = sS->second.begin();
|
|
std::vector< SGNODE* >::iterator eV = sS->second.end();
|
|
|
|
while( sV != eV )
|
|
{
|
|
if( nullptr == S3D::GetSGNodeParent( *sV ) )
|
|
S3D::DestroyNode( *sV );
|
|
|
|
++sV;
|
|
}
|
|
|
|
sS->second.clear();
|
|
++sS;
|
|
}
|
|
|
|
shapes.clear();
|
|
}
|
|
|
|
if( scene )
|
|
S3D::DestroyNode(scene);
|
|
}
|
|
|
|
// find collection of tagged nodes
|
|
bool GetShape( const std::string& id, std::vector< SGNODE* >*& listPtr )
|
|
{
|
|
listPtr = nullptr;
|
|
NODEMAP::iterator item;
|
|
item = shapes.find( id );
|
|
|
|
if( item == shapes.end() )
|
|
return false;
|
|
|
|
listPtr = &item->second;
|
|
return true;
|
|
}
|
|
|
|
// find collection of tagged nodes
|
|
SGNODE* GetFace( const std::string& id )
|
|
{
|
|
FACEMAP::iterator item;
|
|
item = faces.find( id );
|
|
|
|
if( item == faces.end() )
|
|
return nullptr;
|
|
|
|
return item->second;
|
|
}
|
|
|
|
// return color if found; if not found, create SGAPPEARANCE
|
|
SGNODE* GetColor( Quantity_ColorRGBA* colorObj )
|
|
{
|
|
if( nullptr == colorObj )
|
|
{
|
|
if( defaultColor )
|
|
return defaultColor;
|
|
|
|
IFSG_APPEARANCE app( true );
|
|
app.SetShininess( 0.05f );
|
|
app.SetSpecular( 0.04f, 0.04f, 0.04f );
|
|
app.SetAmbient( 0.1f, 0.1f, 0.1f );
|
|
app.SetDiffuse( 0.6f, 0.6f, 0.6f );
|
|
|
|
defaultColor = app.GetRawPtr();
|
|
return defaultColor;
|
|
}
|
|
|
|
Quantity_Color colorRgb = colorObj->GetRGB();
|
|
|
|
Standard_Real r, g, b;
|
|
colorObj->GetRGB().Values( r, g, b, OCC_COLOR_SPACE );
|
|
|
|
std::size_t hash = std::hash<double>{}( colorRgb.Distance( refColor ) )
|
|
^ ( std::hash<float>{}( colorObj->Alpha() ) << 1 );
|
|
|
|
std::map<std::size_t, SGNODE*>::iterator item;
|
|
item = colors.find( hash );
|
|
|
|
if( item != colors.end() )
|
|
return item->second;
|
|
|
|
IFSG_APPEARANCE app( true );
|
|
app.SetShininess( 0.1f );
|
|
app.SetSpecular( 0.12f, 0.12f, 0.12f );
|
|
app.SetAmbient( 0.1f, 0.1f, 0.1f );
|
|
app.SetDiffuse( r, g, b );
|
|
app.SetTransparency( 1.0f - colorObj->Alpha() );
|
|
colors.emplace( hash, app.GetRawPtr() );
|
|
|
|
return app.GetRawPtr();
|
|
}
|
|
};
|
|
|
|
|
|
enum FormatType
|
|
{
|
|
FMT_NONE = 0,
|
|
FMT_STEP,
|
|
FMT_STPZ,
|
|
FMT_IGES
|
|
};
|
|
|
|
|
|
FormatType fileType( const char* aFileName )
|
|
{
|
|
wxFileName fname( wxString::FromUTF8Unchecked( aFileName ) );
|
|
wxFFileInputStream ifile( fname.GetFullPath() );
|
|
|
|
if( !ifile.IsOk() )
|
|
return FMT_NONE;
|
|
|
|
if( fname.GetExt().MakeUpper().EndsWith( wxT( "STPZ" ) )
|
|
|| fname.GetExt().MakeUpper().EndsWith( wxT( "GZ" ) ) )
|
|
{
|
|
return FMT_STPZ;
|
|
}
|
|
|
|
char iline[82];
|
|
memset( iline, 0, 82 );
|
|
ifile.Read( iline, 82 );
|
|
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;
|
|
}
|
|
|
|
|
|
/**
|
|
* Gets the absolute tag string for a given label in the form of ##:##:##:##
|
|
*
|
|
* @param aLabel is the label to get the string for
|
|
* @param aTag is the resulting tag string based by reference
|
|
*/
|
|
void getTag( const TDF_Label& aLabel, std::string& aTag )
|
|
{
|
|
std::ostringstream ostr;
|
|
|
|
if( aLabel.IsNull() )
|
|
{
|
|
wxLogTrace( MASK_OCE, wxT( "Null label passed to getTag" ) );
|
|
return;
|
|
}
|
|
|
|
TColStd_ListOfInteger tagList;
|
|
TDF_Tool::TagList( aLabel, tagList );
|
|
|
|
for( TColStd_ListOfInteger::Iterator it( tagList ); it.More(); it.Next() )
|
|
{
|
|
ostr << it.Value();
|
|
ostr << ":";
|
|
}
|
|
|
|
aTag = ostr.str();
|
|
aTag.pop_back(); // kill the last colon
|
|
}
|
|
|
|
|
|
static wxString getLabelName( const TDF_Label& aLabel )
|
|
{
|
|
wxString txt;
|
|
Handle( TDataStd_Name ) name;
|
|
if( !aLabel.IsNull() && aLabel.FindAttribute( TDataStd_Name::GetID(), name ) )
|
|
{
|
|
TCollection_ExtendedString extstr = name->Get();
|
|
char* str = new char[extstr.LengthOfCString() + 1];
|
|
extstr.ToUTF8CString( str );
|
|
|
|
txt = wxString::FromUTF8( str );
|
|
delete[] str;
|
|
txt = txt.Trim();
|
|
}
|
|
return txt;
|
|
}
|
|
|
|
|
|
/**
|
|
* Gets a string for a given TopAbs_ShapeEnum element
|
|
*
|
|
* @param aShape enum value to convert
|
|
*/
|
|
std::string getShapeName( TopAbs_ShapeEnum aShape )
|
|
{
|
|
switch( aShape )
|
|
{
|
|
case TopAbs_COMPOUND: return "COMPOUND";
|
|
case TopAbs_COMPSOLID: return "COMPSOLID";
|
|
case TopAbs_SOLID: return "SOLID";
|
|
case TopAbs_SHELL: return "SHELL";
|
|
case TopAbs_FACE: return "FACE";
|
|
case TopAbs_WIRE: return "WIRE";
|
|
case TopAbs_EDGE: return "EDGE";
|
|
case TopAbs_VERTEX: return "VERTEX";
|
|
case TopAbs_SHAPE: return "SHAPE";
|
|
}
|
|
|
|
return "UNKNOWN";
|
|
}
|
|
|
|
static int colorFloatToDecimal( float aVal )
|
|
{
|
|
return aVal * 255;
|
|
}
|
|
|
|
|
|
static inline std::ostream& operator<<( std::ostream& aOStream, const Quantity_ColorRGBA& aColor )
|
|
{
|
|
Quantity_Color rgb = aColor.GetRGB();
|
|
|
|
return aOStream << "rgba(" << colorFloatToDecimal( rgb.Red() ) << ","
|
|
<< colorFloatToDecimal( rgb.Green() ) << ","
|
|
<< colorFloatToDecimal( rgb.Blue() ) << ","
|
|
<< colorFloatToDecimal( aColor.Alpha() )
|
|
<< ")";
|
|
}
|
|
|
|
|
|
/**
|
|
* Gets a string for a given TopAbs_ShapeEnum element
|
|
*
|
|
* @param aLabel Label to convert
|
|
* @param aShapeTool Handle to shape tool being used
|
|
* @param aColorTool Handle to color tool being used
|
|
* @param aPregMsg Any prefixed message to insert (used for indentation in dump)
|
|
*/
|
|
static void printLabel( TDF_Label aLabel, Handle( XCAFDoc_ShapeTool ) aShapeTool,
|
|
Handle( XCAFDoc_ColorTool ) aColorTool, const char* aPreMsg = nullptr )
|
|
{
|
|
if( aLabel.IsNull() )
|
|
return;
|
|
|
|
if( !aPreMsg )
|
|
aPreMsg = "Label: ";
|
|
|
|
TCollection_AsciiString entry;
|
|
TDF_Tool::Entry( aLabel, entry );
|
|
std::ostringstream ss;
|
|
ss << aPreMsg << entry << ", " << getLabelName( aLabel )
|
|
<< ( aShapeTool->IsShape( aLabel ) ? ", shape" : "" )
|
|
<< ( aShapeTool->IsTopLevel( aLabel ) ? ", topLevel" : "" )
|
|
<< ( aShapeTool->IsFree( aLabel ) ? ", free" : "" )
|
|
<< ( aShapeTool->IsAssembly( aLabel ) ? ", assembly" : "" )
|
|
<< ( aShapeTool->IsSimpleShape( aLabel ) ? ", simple" : "" )
|
|
<< ( aShapeTool->IsCompound( aLabel ) ? ", compound" : "" )
|
|
<< ( aShapeTool->IsReference( aLabel ) ? ", reference" : "" )
|
|
<< ( aShapeTool->IsComponent( aLabel ) ? ", component" : "" )
|
|
<< ( aShapeTool->IsSubShape( aLabel ) ? ", subshape" : "" );
|
|
|
|
if( aShapeTool->IsSubShape( aLabel ) )
|
|
{
|
|
auto shape = aShapeTool->GetShape( aLabel );
|
|
if( !shape.IsNull() )
|
|
ss << ", " << getShapeName( shape.ShapeType() );
|
|
}
|
|
|
|
if( aShapeTool->IsShape( aLabel ) )
|
|
{
|
|
Quantity_ColorRGBA c;
|
|
if( aColorTool->GetColor( aLabel, XCAFDoc_ColorGen, c ) )
|
|
ss << ", gc: " << c;
|
|
if( aColorTool->GetColor( aLabel, XCAFDoc_ColorSurf, c ) )
|
|
ss << ", sc: " << c;
|
|
if( aColorTool->GetColor( aLabel, XCAFDoc_ColorCurv, c ) )
|
|
ss << ", cc: " << c;
|
|
}
|
|
|
|
wxLogTrace( MASK_OCE, ss.str().c_str() );
|
|
}
|
|
|
|
|
|
/**
|
|
* Dumps a label and the entire tree underneath it
|
|
*
|
|
* @param aLabel Label to convert
|
|
* @param aShapeTool Handle to shape tool being used
|
|
* @param aColorTool Handle to color tool being used
|
|
* @param aDepth Indentation level to offset labels (used recursively by dumpLabels)
|
|
*/
|
|
static void dumpLabels( TDF_Label aLabel, Handle( XCAFDoc_ShapeTool ) aShapeTool,
|
|
Handle( XCAFDoc_ColorTool ) aColorTool, int aDepth = 0 )
|
|
{
|
|
std::string indent( aDepth * 2, ' ' );
|
|
printLabel( aLabel, aShapeTool, aColorTool, indent.c_str() );
|
|
TDF_ChildIterator it;
|
|
for( it.Initialize( aLabel ); it.More(); it.Next() )
|
|
dumpLabels( it.Value(), aShapeTool, aColorTool, aDepth + 1 );
|
|
}
|
|
|
|
|
|
bool getColor( DATA& data, TDF_Label label, Quantity_ColorRGBA& color )
|
|
{
|
|
while( true )
|
|
{
|
|
if( data.m_color->GetColor( label, XCAFDoc_ColorGen, color ) )
|
|
return true;
|
|
else if( data.m_color->GetColor( label, XCAFDoc_ColorSurf, color ) )
|
|
return true;
|
|
else if( data.m_color->GetColor( label, XCAFDoc_ColorCurv, color ) )
|
|
return true;
|
|
|
|
label = label.Father();
|
|
|
|
if( label.IsNull() )
|
|
break;
|
|
};
|
|
|
|
return false;
|
|
}
|
|
|
|
|
|
void addItems( SGNODE* parent, std::vector< SGNODE* >* lp )
|
|
{
|
|
if( nullptr == lp )
|
|
return;
|
|
|
|
std::vector< SGNODE* >::iterator sL = lp->begin();
|
|
std::vector< SGNODE* >::iterator eL = lp->end();
|
|
SGNODE* item;
|
|
|
|
while( sL != eL )
|
|
{
|
|
item = *sL;
|
|
|
|
if( nullptr == S3D::GetSGNodeParent( item ) )
|
|
S3D::AddSGNodeChild( parent, item );
|
|
else
|
|
S3D::AddSGNodeRef( parent, item );
|
|
|
|
++sL;
|
|
}
|
|
}
|
|
|
|
|
|
bool readIGES( Handle( TDocStd_Document ) & m_doc, const char* fname )
|
|
{
|
|
IGESCAFControl_Reader reader;
|
|
IFSelect_ReturnStatus stat = reader.ReadFile( fname );
|
|
reader.PrintCheckLoad( Standard_False, IFSelect_ItemsByEntity );
|
|
|
|
if( stat != IFSelect_RetDone )
|
|
return false;
|
|
|
|
// Enable file-defined shape precision
|
|
if( !Interface_Static::SetIVal( "read.precision.mode", 0 ) )
|
|
return false;
|
|
|
|
// set other translation options
|
|
reader.SetColorMode(true); // use model colors
|
|
reader.SetNameMode(false); // don't use IGES label names
|
|
reader.SetLayerMode(false); // ignore LAYER data
|
|
|
|
if ( !reader.Transfer( m_doc ) )
|
|
return false;
|
|
|
|
// are there any shapes to translate?
|
|
if( reader.NbShapes() < 1 )
|
|
return false;
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
bool readSTEP( Handle(TDocStd_Document)& m_doc, const char* fname )
|
|
{
|
|
wxLogTrace( MASK_OCE, wxT( "Reading step file %s" ), fname );
|
|
|
|
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
|
|
reader.SetColorMode( true ); // use model colors
|
|
reader.SetNameMode( false ); // don't use label names
|
|
reader.SetLayerMode( false ); // ignore LAYER data
|
|
|
|
if ( !reader.Transfer( m_doc ) )
|
|
{
|
|
m_doc->Close();
|
|
return false;
|
|
}
|
|
|
|
// are there any shapes to translate?
|
|
if( reader.NbRootsForTransfer() < 1 )
|
|
return false;
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
bool readSTEPZ( Handle(TDocStd_Document)& m_doc, const char* aFileName )
|
|
{
|
|
wxFileName fname( wxString::FromUTF8Unchecked( aFileName ) );
|
|
wxFFileInputStream ifile( fname.GetFullPath() );
|
|
|
|
wxFileName outFile( fname );
|
|
|
|
outFile.SetPath( wxStandardPaths::Get().GetTempDir() );
|
|
outFile.SetExt( wxT( "STEP" ) );
|
|
|
|
wxFileOffset size = ifile.GetLength();
|
|
wxBusyCursor busycursor;
|
|
|
|
if( size == wxInvalidOffset )
|
|
return false;
|
|
|
|
{
|
|
bool success = false;
|
|
wxFFileOutputStream ofile( outFile.GetFullPath() );
|
|
|
|
if( !ofile.IsOk() )
|
|
return false;
|
|
|
|
char *buffer = new char[size];
|
|
|
|
ifile.Read( buffer, size);
|
|
std::string expanded;
|
|
|
|
try
|
|
{
|
|
expanded = gzip::decompress( buffer, size );
|
|
success = true;
|
|
}
|
|
catch(...)
|
|
{}
|
|
|
|
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() )
|
|
{
|
|
izipfile.Read( ofile );
|
|
success = true;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
ofile.Write( expanded.data(), expanded.size() );
|
|
}
|
|
|
|
delete[] buffer;
|
|
ofile.Close();
|
|
|
|
if( !success )
|
|
return false;
|
|
}
|
|
|
|
bool retval = readSTEP( m_doc, outFile.GetFullPath().mb_str() );
|
|
|
|
// Cleanup our temporary file
|
|
wxRemoveFile( outFile.GetFullPath() );
|
|
|
|
return retval;
|
|
}
|
|
|
|
|
|
SCENEGRAPH* LoadModel( char const* filename )
|
|
{
|
|
DATA data;
|
|
|
|
Handle(XCAFApp_Application) m_app = XCAFApp_Application::GetApplication();
|
|
m_app->NewDocument( "MDTV-XCAF", data.m_doc );
|
|
FormatType modelFmt = fileType( filename );
|
|
|
|
switch( modelFmt )
|
|
{
|
|
case FMT_IGES:
|
|
data.renderBoth = true;
|
|
|
|
if( !readIGES( data.m_doc, filename ) )
|
|
{
|
|
m_app->Close( data.m_doc );
|
|
return nullptr;
|
|
}
|
|
|
|
break;
|
|
|
|
case FMT_STEP:
|
|
if( !readSTEP( data.m_doc, filename ) )
|
|
{
|
|
m_app->Close( data.m_doc );
|
|
return nullptr;
|
|
}
|
|
|
|
break;
|
|
|
|
case FMT_STPZ:
|
|
if( !readSTEPZ( data.m_doc, filename ) )
|
|
{
|
|
m_app->Close( data.m_doc );
|
|
return nullptr;
|
|
}
|
|
|
|
break;
|
|
|
|
|
|
default:
|
|
m_app->Close( data.m_doc );
|
|
return nullptr;
|
|
break;
|
|
}
|
|
|
|
data.m_assy = XCAFDoc_DocumentTool::ShapeTool( data.m_doc->Main() );
|
|
data.m_color = XCAFDoc_DocumentTool::ColorTool( data.m_doc->Main() );
|
|
|
|
// Check if the log mask is enabled otherwise the dump routine may be expensive before the wxLog call
|
|
if( wxLog::IsAllowedTraceMask( MASK_OCE ) )
|
|
{
|
|
dumpLabels( data.m_doc->Main(), data.m_assy, data.m_color );
|
|
}
|
|
|
|
// retrieve all free shapes
|
|
TDF_LabelSequence frshapes;
|
|
data.m_assy->GetFreeShapes( frshapes );
|
|
|
|
bool ret = false;
|
|
|
|
// create the top level SG node
|
|
IFSG_TRANSFORM topNode( true );
|
|
data.scene = topNode.GetRawPtr();
|
|
|
|
for( Standard_Integer i = 1; i <= frshapes.Length(); i++ )
|
|
{
|
|
const TDF_Label& label = frshapes.Value( i );
|
|
|
|
if( data.m_color->IsVisible( label ) )
|
|
{
|
|
if( processLabel( label, data, data.scene, nullptr ) )
|
|
ret = true;
|
|
}
|
|
}
|
|
|
|
if( !ret )
|
|
{
|
|
m_app->Close( data.m_doc );
|
|
return nullptr;
|
|
}
|
|
|
|
SCENEGRAPH* scene = (SCENEGRAPH*)data.scene;
|
|
|
|
// DEBUG: WRITE OUT VRML2 FILE TO CONFIRM STRUCTURE
|
|
#if ( defined( DEBUG_OCE ) && DEBUG_OCE > 3 )
|
|
if( data.scene )
|
|
{
|
|
wxFileName fn( wxString::FromUTF8Unchecked( filename ) );
|
|
wxString output;
|
|
|
|
if( FMT_STEP == modelFmt )
|
|
output = wxT( "_step-" );
|
|
else
|
|
output = wxT( "_iges-" );
|
|
|
|
output.append( fn.GetName() );
|
|
output.append( wxT( ".wrl" ) );
|
|
S3D::WriteVRML( output.ToUTF8(), true, data.scene, true, true );
|
|
}
|
|
#endif
|
|
|
|
// set to NULL to prevent automatic destruction of the scene data
|
|
data.scene = nullptr;
|
|
|
|
m_app->Close( data.m_doc );
|
|
return scene;
|
|
}
|
|
|
|
|
|
bool processShell( const TopoDS_Shape& shape, DATA& data, SGNODE* parent,
|
|
std::vector<SGNODE*>* items, Quantity_ColorRGBA* color )
|
|
{
|
|
TopoDS_Iterator it;
|
|
bool ret = false;
|
|
|
|
wxLogTrace( MASK_OCE, wxT( "Processing shell" ) );
|
|
for( it.Initialize( shape, false, false ); it.More(); it.Next() )
|
|
{
|
|
const TopoDS_Face& face = TopoDS::Face( it.Value() );
|
|
|
|
if( processFace( face, data, parent, items, color ) )
|
|
ret = true;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
|
|
bool processSolidOrShell( const TopoDS_Shape& shape, DATA& data, SGNODE* parent,
|
|
std::vector< SGNODE* >* items )
|
|
{
|
|
TDF_Label label;
|
|
data.hasSolid = true;
|
|
std::string partID;
|
|
Quantity_ColorRGBA col;
|
|
Quantity_ColorRGBA* lcolor = nullptr;
|
|
|
|
wxLogTrace( MASK_OCE, wxT( "Processing solid" ) );
|
|
|
|
// Search the whole model first to make sure something exists (may or may not have color)
|
|
if( !data.m_assy->Search( shape, label ) )
|
|
{
|
|
static int i = 0;
|
|
std::ostringstream ostr;
|
|
ostr << "KMISC_" << i++;
|
|
partID = ostr.str();
|
|
}
|
|
else
|
|
{
|
|
bool found_color = false;
|
|
|
|
if( getColor( data, label, col ) )
|
|
{
|
|
found_color = true;
|
|
lcolor = &col;
|
|
}
|
|
|
|
// If the top-level label doesn't have the color information, search components
|
|
if( !found_color )
|
|
{
|
|
if( data.m_assy->Search( shape, label, Standard_False, Standard_True, Standard_True ) &&
|
|
getColor( data, label, col ) )
|
|
{
|
|
found_color = true;
|
|
lcolor = &col;
|
|
}
|
|
}
|
|
|
|
// If the components do not have color information, search all components without location
|
|
if( !found_color )
|
|
{
|
|
if( data.m_assy->Search( shape, label, Standard_False, Standard_False,
|
|
Standard_True ) &&
|
|
getColor( data, label, col ) )
|
|
{
|
|
found_color = true;
|
|
lcolor = &col;
|
|
}
|
|
}
|
|
|
|
// Our last chance to find the color looks for color as a subshape of top-level simple
|
|
// shapes.
|
|
if( !found_color )
|
|
{
|
|
if( data.m_assy->Search( shape, label, Standard_False, Standard_False,
|
|
Standard_False ) &&
|
|
getColor( data, label, col ) )
|
|
{
|
|
found_color = true;
|
|
lcolor = &col;
|
|
}
|
|
}
|
|
|
|
getTag( label, partID );
|
|
}
|
|
|
|
TopoDS_Iterator it;
|
|
IFSG_TRANSFORM childNode( parent );
|
|
SGNODE* pptr = childNode.GetRawPtr();
|
|
bool ret = false;
|
|
|
|
|
|
std::vector< SGNODE* >* component = nullptr;
|
|
|
|
if( !partID.empty() )
|
|
data.GetShape( partID, component );
|
|
|
|
if( component )
|
|
{
|
|
addItems( pptr, component );
|
|
|
|
if( nullptr != items )
|
|
items->push_back( pptr );
|
|
}
|
|
|
|
// instantiate resulting object
|
|
std::vector<SGNODE*> itemList;
|
|
|
|
TopAbs_ShapeEnum stype = shape.ShapeType();
|
|
if( stype == TopAbs_SHELL )
|
|
{
|
|
if( processShell( shape, data, pptr, &itemList, lcolor ) )
|
|
ret = true;
|
|
}
|
|
else
|
|
{
|
|
|
|
for( it.Initialize( shape, false, false ); it.More(); it.Next() )
|
|
{
|
|
const TopoDS_Shape& subShape = it.Value();
|
|
|
|
if( subShape.ShapeType() == TopAbs_SHELL )
|
|
{
|
|
if( processShell( subShape, data, pptr, &itemList, lcolor ) )
|
|
ret = true;
|
|
}
|
|
else
|
|
{
|
|
wxLogTrace( MASK_OCE, wxT( "Unsupported subshape in solid" ) );
|
|
}
|
|
}
|
|
}
|
|
|
|
if( !ret )
|
|
childNode.Destroy();
|
|
else if( nullptr != items )
|
|
items->push_back( pptr );
|
|
|
|
return ret;
|
|
}
|
|
|
|
|
|
bool processLabel( const TDF_Label& aLabel, DATA& aData, SGNODE* aParent,
|
|
std::vector<SGNODE*>* aItems )
|
|
{
|
|
std::string labelTag;
|
|
|
|
if( wxLog::IsAllowedTraceMask( MASK_OCE ) )
|
|
{
|
|
// can be expensive, guard it if we aren't logging
|
|
getTag( aLabel, labelTag );
|
|
}
|
|
|
|
wxLogTrace( MASK_OCE, wxT( "Processing label %s" ), labelTag );
|
|
|
|
TopoDS_Shape originalShape;
|
|
TDF_Label shapeLabel = aLabel;
|
|
|
|
if( !aData.m_assy->GetShape( shapeLabel, originalShape ) )
|
|
{
|
|
return false;
|
|
}
|
|
|
|
TopoDS_Shape shape = originalShape;
|
|
|
|
if( aData.m_assy->IsReference( aLabel ) )
|
|
{
|
|
wxLogTrace( MASK_OCE, wxT( "Label %s is ref, trying to pull up referred label" ),
|
|
labelTag );
|
|
|
|
if( !aData.m_assy->GetReferredShape( aLabel, shapeLabel ) )
|
|
{
|
|
return false;
|
|
}
|
|
|
|
labelTag = static_cast<int>( shapeLabel.Tag() );
|
|
// wxLogTrace( MASK_OCE, wxT( "Label %s referred" ), labelTag );
|
|
|
|
if( !aData.m_assy->GetShape( shapeLabel, shape ) )
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
|
|
// Now let's see if the original label has a location
|
|
// Labels can be used to place copies of other labels at a specific location
|
|
IFSG_TRANSFORM childNode( aParent );
|
|
SGNODE* pptr = childNode.GetRawPtr();
|
|
const TopLoc_Location& loc = originalShape.Location();
|
|
|
|
if( !loc.IsIdentity() )
|
|
{
|
|
wxLogTrace( MASK_OCE, wxT( "Label %d has location" ), static_cast<int>( aLabel.Tag() ) );
|
|
gp_Trsf T = loc.Transformation();
|
|
gp_XYZ coord = T.TranslationPart();
|
|
childNode.SetTranslation( SGPOINT( coord.X(), coord.Y(), coord.Z() ) );
|
|
wxLogTrace( MASK_OCE, wxT( "Translation %f, %f, %f" ), coord.X(), coord.Y(), coord.Z() );
|
|
gp_XYZ axis;
|
|
Standard_Real angle;
|
|
|
|
if( T.GetRotation( axis, angle ) )
|
|
{
|
|
childNode.SetRotation( SGVECTOR( axis.X(), axis.Y(), axis.Z() ), angle );
|
|
wxLogTrace( MASK_OCE, wxT( "Rotation %f, %f, %f, angle %f" ),
|
|
axis.X(), axis.Y(), axis.Z(), angle );
|
|
}
|
|
}
|
|
|
|
TopAbs_ShapeEnum stype = shape.ShapeType();
|
|
bool ret = false;
|
|
aData.hasSolid = false;
|
|
|
|
switch( stype )
|
|
{
|
|
case TopAbs_COMPOUND:
|
|
{
|
|
// assemblies will report a shape type of compound which isn't what we are after
|
|
// we will still process the children of assemblies but they should just be label references to the actual shapes
|
|
if( !aData.m_assy->IsAssembly( shapeLabel ) )
|
|
{
|
|
TopExp_Explorer xp;
|
|
|
|
for( xp.Init( shape, TopAbs_SOLID ); xp.More(); xp.Next() )
|
|
{
|
|
processSolidOrShell( xp.Current(), aData, pptr, aItems );
|
|
ret = true;
|
|
}
|
|
|
|
// Get all shells, avoid those that may be attached to solids
|
|
for( xp.Init( shape, TopAbs_SHELL, TopAbs_SOLID ); xp.More(); xp.Next() )
|
|
{
|
|
processSolidOrShell( xp.Current(), aData, pptr, aItems );
|
|
ret = true;
|
|
}
|
|
|
|
// Get all faces, avoid those that may be attached to shells
|
|
for( xp.Init( shape, TopAbs_FACE, TopAbs_SHELL ); xp.More(); xp.Next() )
|
|
{
|
|
const TopoDS_Face& face = TopoDS::Face( xp.Current() );
|
|
processFace( face, aData, pptr, aItems, nullptr );
|
|
ret = true;
|
|
}
|
|
}
|
|
}
|
|
break;
|
|
|
|
case TopAbs_SOLID:
|
|
if( processSolidOrShell( shape, aData, pptr, aItems ) )
|
|
ret = true;
|
|
|
|
break;
|
|
|
|
case TopAbs_SHELL:
|
|
if( processSolidOrShell( shape, aData, pptr, aItems ) )
|
|
ret = true;
|
|
|
|
break;
|
|
|
|
case TopAbs_FACE:
|
|
if( processFace( TopoDS::Face( shape ), aData, pptr, aItems, nullptr ) )
|
|
ret = true;
|
|
|
|
break;
|
|
|
|
default:
|
|
break;
|
|
}
|
|
|
|
if( nullptr != aItems )
|
|
aItems->push_back( pptr );
|
|
|
|
if( !aData.m_assy->IsSimpleShape( shapeLabel ) && shapeLabel.HasChild() )
|
|
{
|
|
wxLogTrace( MASK_OCE, wxT( "Label %s has children" ), labelTag );
|
|
TDF_ChildIterator it;
|
|
|
|
for( it.Initialize( shapeLabel ); it.More(); it.Next() )
|
|
{
|
|
if( processLabel( it.Value(), aData, pptr, aItems ) )
|
|
ret = true;
|
|
}
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
|
|
bool processFace( const TopoDS_Face& face, DATA& data, SGNODE* parent, std::vector<SGNODE*>* items,
|
|
Quantity_ColorRGBA* color )
|
|
{
|
|
if( Standard_True == face.IsNull() )
|
|
return false;
|
|
|
|
bool reverse = ( face.Orientation() == TopAbs_REVERSED );
|
|
SGNODE* ashape = nullptr;
|
|
std::string partID;
|
|
TDF_Label label;
|
|
|
|
bool useBothSides = false;
|
|
|
|
// for IGES renderBoth = TRUE; for STEP if a shell or face is not a descendant
|
|
// of a SOLID then hasSolid = false and we must render both sides
|
|
if( data.renderBoth || !data.hasSolid )
|
|
useBothSides = true;
|
|
|
|
if( data.m_assy->FindShape( face, label, Standard_False ) )
|
|
getTag( label, partID );
|
|
|
|
if( !partID.empty() )
|
|
ashape = data.GetFace( partID );
|
|
|
|
if( ashape )
|
|
{
|
|
if( nullptr == S3D::GetSGNodeParent( ashape ) )
|
|
S3D::AddSGNodeChild( parent, ashape );
|
|
else
|
|
S3D::AddSGNodeRef( parent, ashape );
|
|
|
|
if( nullptr != items )
|
|
items->push_back( ashape );
|
|
|
|
if( useBothSides )
|
|
{
|
|
std::string id2 = partID;
|
|
id2.append( "b" );
|
|
SGNODE* shapeB = data.GetFace( id2 );
|
|
|
|
if( nullptr == S3D::GetSGNodeParent( shapeB ) )
|
|
S3D::AddSGNodeChild( parent, shapeB );
|
|
else
|
|
S3D::AddSGNodeRef( parent, shapeB );
|
|
|
|
if( nullptr != items )
|
|
items->push_back( shapeB );
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
TopLoc_Location loc;
|
|
Standard_Boolean isTessellate (Standard_False);
|
|
Handle( Poly_Triangulation ) triangulation = BRep_Tool::Triangulation( face, loc );
|
|
|
|
if( triangulation.IsNull() || triangulation->Deflection() > USER_PREC + Precision::Confusion() )
|
|
isTessellate = Standard_True;
|
|
|
|
if( isTessellate )
|
|
{
|
|
BRepMesh_IncrementalMesh IM(face, USER_PREC, Standard_False, USER_ANGLE );
|
|
triangulation = BRep_Tool::Triangulation( face, loc );
|
|
}
|
|
|
|
if( triangulation.IsNull() == Standard_True )
|
|
return false;
|
|
|
|
Quantity_ColorRGBA lcolor;
|
|
|
|
// check for a face color; this has precedence over SOLID colors
|
|
if( data.m_color->GetColor( face, XCAFDoc_ColorGen, lcolor )
|
|
|| data.m_color->GetColor( face, XCAFDoc_ColorCurv, lcolor )
|
|
|| data.m_color->GetColor( face, XCAFDoc_ColorSurf, lcolor ) )
|
|
color = &lcolor;
|
|
|
|
SGNODE* ocolor = data.GetColor( color );
|
|
|
|
// create a SHAPE and attach the color and data,
|
|
// then attach the shape to the parent and return TRUE
|
|
IFSG_SHAPE vshape( true );
|
|
IFSG_FACESET vface( vshape );
|
|
IFSG_COORDS vcoords( vface );
|
|
IFSG_COORDINDEX coordIdx( vface );
|
|
|
|
if( nullptr == S3D::GetSGNodeParent( ocolor ) )
|
|
S3D::AddSGNodeChild( vshape.GetRawPtr(), ocolor );
|
|
else
|
|
S3D::AddSGNodeRef( vshape.GetRawPtr(), ocolor );
|
|
|
|
std::vector< SGPOINT > vertices;
|
|
std::vector< int > indices;
|
|
std::vector< int > indices2;
|
|
gp_Trsf tx;
|
|
|
|
for( int i = 1; i <= triangulation->NbNodes(); i++ )
|
|
{
|
|
gp_XYZ v( triangulation->Node(i).Coord() );
|
|
vertices.emplace_back( v.X(), v.Y(), v.Z() );
|
|
}
|
|
|
|
for( int i = 1; i <= triangulation->NbTriangles(); i++ )
|
|
{
|
|
int a, b, c;
|
|
triangulation->Triangle(i).Get(a, b, c);
|
|
a--;
|
|
|
|
if( reverse )
|
|
{
|
|
int tmp = b - 1;
|
|
b = c - 1;
|
|
c = tmp;
|
|
}
|
|
else
|
|
{
|
|
b--;
|
|
c--;
|
|
}
|
|
|
|
indices.push_back( a );
|
|
indices.push_back( b );
|
|
indices.push_back( c );
|
|
|
|
if( useBothSides )
|
|
{
|
|
indices2.push_back( b );
|
|
indices2.push_back( a );
|
|
indices2.push_back( c );
|
|
}
|
|
}
|
|
|
|
vcoords.SetCoordsList( vertices.size(), &vertices[0] );
|
|
coordIdx.SetIndices( indices.size(), &indices[0] );
|
|
vface.CalcNormals( nullptr );
|
|
vshape.SetParent( parent );
|
|
|
|
if( !partID.empty() )
|
|
data.faces.emplace( partID, vshape.GetRawPtr() );
|
|
|
|
// The outer surface of an IGES model is indeterminate so
|
|
// we must render both sides of a surface.
|
|
if( useBothSides )
|
|
{
|
|
std::string id2 = partID;
|
|
id2.append( "b" );
|
|
IFSG_SHAPE vshape2( true );
|
|
IFSG_FACESET vface2( vshape2 );
|
|
IFSG_COORDS vcoords2( vface2 );
|
|
IFSG_COORDINDEX coordIdx2( vface2 );
|
|
S3D::AddSGNodeRef( vshape2.GetRawPtr(), ocolor );
|
|
|
|
vcoords2.SetCoordsList( vertices.size(), &vertices[0] );
|
|
coordIdx2.SetIndices( indices2.size(), &indices2[0] );
|
|
vface2.CalcNormals( nullptr );
|
|
vshape2.SetParent( parent );
|
|
|
|
if( !partID.empty() )
|
|
data.faces.emplace( id2, vshape2.GetRawPtr() );
|
|
}
|
|
|
|
return true;
|
|
}
|