loadmodel.cpp File Reference

#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_Face.hxx>
#include <TopoDS_Compound.hxx>
#include <TopExp_Explorer.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 "plugins/3dapi/ifsg_all.h"

Go to the source code of this file.

Classes
struct	DATA

Macros
#define	MASK_OCE   wxT( "PLUGIN_OCE" )
#define	MASK_OCE_EXTRA   wxT( "PLUGIN_OCE_EXTRA" )
#define	USER_PREC   (0.14)
#define	USER_ANGLE   (0.52359878)

Typedefs
typedef std::map< Standard_Real, 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

Enumerations
enum	FormatType { FMT_NONE = 0, FMT_STEP, FMT_STPZ, FMT_IGES }

Functions
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_Color *color)
FormatType	fileType (const char *aFileName)
void	getTag (const TDF_Label &aLabel, std::string &aTag)
	Gets the absolute tag string for a given label in the form of ##:##:##:##. More...
static wxString	getLabelName (const TDF_Label &aLabel)
std::string	getShapeName (TopAbs_ShapeEnum aShape)
	Gets a string for a given TopAbs_ShapeEnum element. More...
static int	colorFloatToDecimal (float aVal)
static std::ostream &	operator<< (std::ostream &aOStream, const Quantity_ColorRGBA &aColor)
static void	printLabel (TDF_Label aLabel, Handle(XCAFDoc_ShapeTool) aShapeTool, Handle(XCAFDoc_ColorTool) aColorTool, const char *aPreMsg=nullptr)
	Gets a string for a given TopAbs_ShapeEnum element. More...
static void	dumpLabels (TDF_Label aLabel, Handle(XCAFDoc_ShapeTool) aShapeTool, Handle(XCAFDoc_ColorTool) aColorTool, int aDepth=0)
	Dumps a label and the entire tree underneath it. More...
bool	getColor (DATA &data, TDF_Label label, Quantity_Color &color)
void	addItems (SGNODE *parent, std::vector< SGNODE * > *lp)
bool	readIGES (Handle(TDocStd_Document) &m_doc, const char *fname)
bool	readSTEP (Handle(TDocStd_Document)&m_doc, const char *fname)
bool	readSTEPZ (Handle(TDocStd_Document)&m_doc, const char *aFileName)
SCENEGRAPH *	LoadModel (char const *filename)
bool	processShell (const TopoDS_Shape &shape, DATA &data, SGNODE *parent, std::vector< SGNODE * > *items, Quantity_Color *color)
bool	processSolid (const TopoDS_Shape &shape, DATA &data, SGNODE *parent, std::vector< SGNODE * > *items)

Macro Definition Documentation

MASK_OCE

#define MASK_OCE   wxT( "PLUGIN_OCE" )

Definition at line 87 of file loadmodel.cpp.

MASK_OCE_EXTRA

#define MASK_OCE_EXTRA   wxT( "PLUGIN_OCE_EXTRA" )

Definition at line 88 of file loadmodel.cpp.

USER_ANGLE

#define USER_ANGLE   (0.52359878)

Definition at line 97 of file loadmodel.cpp.

USER_PREC

#define USER_PREC   (0.14)

Definition at line 91 of file loadmodel.cpp.

Typedef Documentation

COLORMAP

typedef std::map< Standard_Real, SGNODE* > COLORMAP

Definition at line 99 of file loadmodel.cpp.

FACEMAP

typedef std::map< std::string, SGNODE* > FACEMAP

Definition at line 100 of file loadmodel.cpp.

NODEITEM

typedef std::pair< std::string, std::vector< SGNODE* > > NODEITEM

Definition at line 102 of file loadmodel.cpp.

NODEMAP

typedef std::map< std::string, std::vector< SGNODE* > > NODEMAP

Definition at line 101 of file loadmodel.cpp.

Enumeration Type Documentation

FormatType

enum FormatType

Enumerator
FMT_NONE
FMT_STEP
FMT_STPZ
FMT_IGES

Definition at line 269 of file loadmodel.cpp.

{
 FMT_NONE = 0,
 FMT_STEP,
 FMT_STPZ,
 FMT_IGES
};

Function Documentation

addItems()

void addItems	(	SGNODE *	parent,
		std::vector< SGNODE * > *	lp
	)

Definition at line 501 of file loadmodel.cpp.

{
 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;
 }
}

References S3D::AddSGNodeChild(), S3D::AddSGNodeRef(), and S3D::GetSGNodeParent().

Referenced by processSolid().

colorFloatToDecimal()

static int colorFloatToDecimal ( float  aVal )

static

Definition at line 390 of file loadmodel.cpp.

{
 return aVal * 255;
}

Referenced by operator<<().

dumpLabels()

static void dumpLabels ( TDF_Label  aLabel, Handle(XCAFDoc_ShapeTool)  aShapeTool, Handle(XCAFDoc_ColorTool)  aColorTool, int  aDepth = 0  )

static

Dumps a label and the entire tree underneath it.

Parameters

aLabel	Label to convert
aShapeTool	Handle to shape tool being used
aColorTool	Handle to color tool being used
aDepth	Indentation level to offset labels (used recursively by dumpLabels)

Definition at line 469 of file loadmodel.cpp.

{
 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 );
}

References printLabel().

Referenced by LoadModel().

fileType()

FormatType fileType

(

const char *

aFileName

)

Definition at line 278 of file loadmodel.cpp.

{
 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;
}

References FMT_IGES, FMT_NONE, FMT_STEP, and FMT_STPZ.

Referenced by PANEL_FP_LIB_TABLE::browseLibrariesHandler(), GRAPHICS_IMPORT_MGR::GetPluginByExt(), SCH_EDIT_FRAME::importFile(), FOOTPRINT_EDIT_FRAME::ImportFootprint(), LoadModel(), PANEL_FP_LIB_TABLE::PANEL_FP_LIB_TABLE(), GERBVIEW_CONTROL::ReloadAllLayers(), and SYMBOL_EDIT_FRAME::saveLibrary().

getColor()

bool getColor	(	DATA &	data,
		TDF_Label	label,
		Quantity_Color &	color
	)

Definition at line 480 of file loadmodel.cpp.

{
 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;
}

References color, and DATA::GetColor().

Referenced by processSolid().

getLabelName()

static wxString getLabelName ( const TDF_Label &  aLabel )

static

Definition at line 349 of file loadmodel.cpp.

{
 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;
}

References name.

Referenced by printLabel().

getShapeName()

std::string getShapeName

(

TopAbs_ShapeEnum

aShape

)

Gets a string for a given TopAbs_ShapeEnum element.

Parameters

aShape

enum value to convert

Definition at line 372 of file loadmodel.cpp.

{
 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";
}

Referenced by printLabel().

getTag()

void getTag	(	const TDF_Label &	aLabel,
		std::string &	aTag
	)

Gets the absolute tag string for a given label in the form of ##:##:##:##.

Parameters

aLabel	is the label to get the string for
aTag	is the resulting tag string based by reference

Definition at line 325 of file loadmodel.cpp.

{
 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
}

References MASK_OCE.

Referenced by processFace(), processLabel(), and processSolid().

LoadModel()

SCENEGRAPH* LoadModel

(

char const *

filename

)

Definition at line 660 of file loadmodel.cpp.

{
 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;
}

References dumpLabels(), fileType(), FMT_IGES, FMT_STEP, FMT_STPZ, IFSG_NODE::GetRawPtr(), MASK_OCE, processLabel(), readIGES(), readSTEP(), readSTEPZ(), DATA::renderBoth, DATA::scene, and S3D::WriteVRML().

Referenced by Load().

operator<<()

static std::ostream& operator<< ( std::ostream &  aOStream, const Quantity_ColorRGBA &  aColor  )

inlinestatic

Definition at line 396 of file loadmodel.cpp.

{
 Quantity_Color rgb = aColor.GetRGB();

 return aOStream << "rgba(" << colorFloatToDecimal( rgb.Red() ) << ","
 << colorFloatToDecimal( rgb.Green() ) << ","
 << colorFloatToDecimal( rgb.Blue() ) << ","
 << colorFloatToDecimal( aColor.Alpha() )
 << ")";
}

References colorFloatToDecimal().

printLabel()

static void printLabel ( TDF_Label  aLabel, Handle(XCAFDoc_ShapeTool)  aShapeTool, Handle(XCAFDoc_ColorTool)  aColorTool, const char *  aPreMsg = nullptr  )

static

Gets a string for a given TopAbs_ShapeEnum element.

Parameters

aLabel	Label to convert
aShapeTool	Handle to shape tool being used
aColorTool	Handle to color tool being used
aPregMsg	Any prefixed message to insert (used for indentation in dump)

Definition at line 416 of file loadmodel.cpp.

{
 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() );
}

References getLabelName(), getShapeName(), and MASK_OCE.

Referenced by dumpLabels().

processFace()

bool processFace	(	const TopoDS_Face &	face,
		DATA &	data,
		SGNODE *	parent,
		std::vector< SGNODE * > *	items,
		Quantity_Color *	color
	)

Definition at line 1040 of file loadmodel.cpp.

{
 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_Color lcolor;

 // check for a face color; this has precedence over SOLID colors
 do
 {
 TDF_Label L;

 if( data.m_color->ShapeTool()->Search( face, L ) )
 {
 if( data.m_color->GetColor( L, XCAFDoc_ColorGen, lcolor )
 || data.m_color->GetColor( L, XCAFDoc_ColorCurv, lcolor )
 || data.m_color->GetColor( L, XCAFDoc_ColorSurf, lcolor ) )
 color = &lcolor;
 }
 } while( 0 );

 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.insert( std::pair< std::string, SGNODE* >( 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.insert( std::pair< std::string, SGNODE* >( id2, vshape2.GetRawPtr() ) );
 }

 return true;
}

References S3D::AddSGNodeChild(), S3D::AddSGNodeRef(), IFSG_FACESET::CalcNormals(), color, DATA::faces, DATA::GetColor(), DATA::GetFace(), IFSG_NODE::GetRawPtr(), S3D::GetSGNodeParent(), getTag(), DATA::hasSolid, DATA::renderBoth, IFSG_COORDS::SetCoordsList(), IFSG_INDEX::SetIndices(), IFSG_NODE::SetParent(), USER_ANGLE, and USER_PREC.

Referenced by processLabel(), and processShell().

processLabel()

bool processLabel	(	const TDF_Label &	aLabel,
		DATA &	aData,
		SGNODE *	aParent,
		std::vector< SGNODE * > *	aItems
	)

Definition at line 904 of file loadmodel.cpp.

{
 std::string labelTag;

 if( wxLog::IsAllowedTraceMask( MASK_OCE ) )
 {
 // can be expensive, guard it if we arent 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() )
 {
 processSolid( xp.Current(), aData, pptr, aItems );
 ret = true;
 }

 for( xp.Init( shape, TopAbs_SHELL, TopAbs_SOLID ); xp.More(); xp.Next() )
 {
 processShell( xp.Current(), aData, pptr, aItems, nullptr );
 ret = true;
 }
 }
 }
 break;

 case TopAbs_SOLID:
 if( processSolid( shape, aData, pptr, aItems ) )
 ret = true;

 break;

 case TopAbs_SHELL:
 if( processShell( shape, aData, pptr, aItems, nullptr ) )
 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;
}

References PNS::angle(), IFSG_NODE::GetRawPtr(), DATA::GetShape(), getTag(), DATA::hasSolid, MASK_OCE, processFace(), processShell(), processSolid(), IFSG_TRANSFORM::SetRotation(), and IFSG_TRANSFORM::SetTranslation().

Referenced by LoadModel().

processShell()

bool processShell	(	const TopoDS_Shape &	shape,
		DATA &	data,
		SGNODE *	parent,
		std::vector< SGNODE * > *	items,
		Quantity_Color *	color
	)

Definition at line 771 of file loadmodel.cpp.

{
 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;
}

References color, MASK_OCE, and processFace().

Referenced by processLabel(), and processSolid().

processSolid()

bool processSolid	(	const TopoDS_Shape &	shape,
		DATA &	data,
		SGNODE *	parent,
		std::vector< SGNODE * > *	items
	)

Definition at line 790 of file loadmodel.cpp.

{
 TDF_Label label;
 data.hasSolid = true;
 std::string partID;
 Quantity_Color col;
 Quantity_Color* 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 the solid
 std::vector< SGNODE* > itemList;

 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;
}

References addItems(), IFSG_NODE::Destroy(), getColor(), IFSG_NODE::GetRawPtr(), DATA::GetShape(), getTag(), DATA::hasSolid, MASK_OCE, and processShell().

Referenced by processLabel().

readIGES()

bool readIGES	(	Handle(TDocStd_Document) &	m_doc,
		const char *	fname
	)

Definition at line 524 of file loadmodel.cpp.

{
 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;
}

Referenced by LoadModel().

readSTEP()

bool readSTEP	(	Handle(TDocStd_Document)&	m_doc,
		const char *	fname
	)

Definition at line 553 of file loadmodel.cpp.

{
 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;
}

References MASK_OCE, and USER_PREC.

Referenced by LoadModel(), and readSTEPZ().

readSTEPZ()

bool readSTEPZ	(	Handle(TDocStd_Document)&	m_doc,
		const char *	aFileName
	)

Definition at line 590 of file loadmodel.cpp.

{
 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;
}

References readSTEP().

Referenced by LoadModel().