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 <Standard_Version.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)
#define	OCC_COLOR_SPACE   Quantity_TOC_RGB

Typedefs
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

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_ColorRGBA *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_ColorRGBA &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_ColorRGBA *color)
bool	processSolidOrShell (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 88 of file loadmodel.cpp.

MASK_OCE_EXTRA

#define MASK_OCE_EXTRA   wxT( "PLUGIN_OCE_EXTRA" )

Definition at line 89 of file loadmodel.cpp.

OCC_COLOR_SPACE

#define OCC_COLOR_SPACE   Quantity_TOC_RGB

Definition at line 117 of file loadmodel.cpp.

USER_ANGLE

#define USER_ANGLE   (0.52359878)

Definition at line 98 of file loadmodel.cpp.

USER_PREC

#define USER_PREC   (0.14)

Definition at line 92 of file loadmodel.cpp.

Typedef Documentation

COLORMAP

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

Definition at line 100 of file loadmodel.cpp.

FACEMAP

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

Definition at line 101 of file loadmodel.cpp.

NODEITEM

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

Definition at line 103 of file loadmodel.cpp.

NODEMAP

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

Definition at line 102 of file loadmodel.cpp.

Enumeration Type Documentation

FormatType

enum FormatType

Enumerator
FMT_NONE
FMT_STEP
FMT_STPZ
FMT_IGES

Definition at line 283 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 515 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 processSolidOrShell().

colorFloatToDecimal()

static int colorFloatToDecimal ( float  aVal )

static

Definition at line 404 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 483 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 dumpLabels(), and printLabel().

Referenced by dumpLabels(), and LoadModel().

fileType()

FormatType fileType

(

const char *

aFileName

)

Definition at line 292 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 LoadModel().

getColor()

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

Definition at line 494 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 processSolidOrShell().

getLabelName()

static wxString getLabelName ( const TDF_Label &  aLabel )

static

Definition at line 363 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 386 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 339 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 processSolidOrShell().

LoadModel()

SCENEGRAPH * LoadModel

(

char const *

filename

)

Definition at line 674 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 410 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 430 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_ColorRGBA *	color
	)

Definition at line 1072 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_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;
}

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 927 of file loadmodel.cpp.

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

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

Referenced by LoadModel(), and processLabel().

processShell()

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

Definition at line 784 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 processSolidOrShell().

processSolidOrShell()

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

Definition at line 803 of file loadmodel.cpp.

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

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 538 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 567 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 604 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().