WriteParasolid.hpp File Reference

Go to the source code of this file.

Functions
Dtk_string	PswGetLastOutputFile ()
const Dtk_tab< char > &	PswGetLastStream ()
void	PswWriteDocument (Dtk_MainDocPtr inDocument)
Dtk_ErrorStatus	PswWriteEnd ()
Dtk_ErrorStatus	PswWriteInit (const Dtk_string &inInputFile, const Dtk_string &inOutputFolder)
Dtk_ErrorStatus	PswWriteInitMemory ()
void	WriteBodyForParasolid (Dtk_BodyPtr inBody, Dtk_tab< Dtk_ID > &outIds)

Function Documentation

PswGetLastOutputFile()

Dtk_string PswGetLastOutputFile

(

)

{
     return g_lastOutputFile;
}

PswGetLastStream()

const Dtk_tab<char>& PswGetLastStream

(

)

{
     return g_lastStream;
}

PswWriteDocument()

void PswWriteDocument

(

Dtk_MainDocPtr

inDocument

)

{
     g_prototypeCache.clear();
 
     Dtk_ComponentPtr rootComponent = inDocument->RootComponent();
     if( rootComponent.IsNULL() )
     {
          cout << "Warning: document has no root component." << endl;
          return;
     }
 
     // Always create a top-level Parasolid assembly to hold the entire document.
     Dtk_ID rootAssemblyId = 0;
     psw_CreateEmptyAssembly( rootAssemblyId );
 
     // Set the document/root name on the assembly.
     Dtk_string docName = rootComponent->Name();
     if( docName.len() > 0 )
     {
          Dtk_InfoPtr info = Dtk_Info::Create();
          info->SetName( docName );
          psw_CreateAttribut( rootAssemblyId, info );
     }
 
     // Recursively traverse the component tree and fill the root assembly.
     PswWriteComponent( rootComponent, rootAssemblyId );
 
     // Finalise the root assembly (must be called once all instances are added).
     psw_LinkAttribut( rootAssemblyId );
 
     g_prototypeCache.clear();
}

PswWriteEnd()

Dtk_ErrorStatus PswWriteEnd

(

)

{
     Dtk_tab<Dtk_ID> emptyIds; // Root assembly is implicit when the tab is empty.
     Dtk_ErrorStatus err = psw_EndFile( emptyIds );
     if( err != dtkNoError )
          cout << "Warning: psw_EndFile failed: " << dtkTypeError( err ).c_str() << endl;
     return err;
}

PswWriteInit()

Dtk_ErrorStatus PswWriteInit	(	const Dtk_string &	inInputFile,
		const Dtk_string &	inOutputFolder
	)

{
     g_prototypeCache.clear();
     g_lastStream.clear();
 
     Dtk_string outputFile;
     if( inOutputFolder.len() > 0 )
          outputFile = inOutputFolder + inInputFile.filename() + inInputFile.extension() + L"_psw.x_t";
     else
          outputFile = inInputFile + L"_psw.x_t";
 
     // Persist the output path so callers can retrieve it via PswGetLastOutputFile().
     g_lastOutputFile = outputFile;
 
     const Dtk_bool muteMode = DTK_TRUE; // Suppress the psw internal log file.
     Dtk_ErrorStatus err = psw_InitFile( outputFile, muteMode );
 
     if( err == dtkNoError )
          cout << "Creating Parasolid file: " << outputFile.c_str() << endl;
     else
          cout << "Warning: psw_InitFile failed: " << dtkTypeError( err ).c_str() << endl;
 
     return err;
}

PswWriteInitMemory()

Dtk_ErrorStatus PswWriteInitMemory

(

)

{
     g_prototypeCache.clear();
     g_lastStream.clear();
     g_lastOutputFile = Dtk_string();   // no output file in memory mode
 
     const Dtk_bool muteMode = DTK_TRUE;
     Dtk_ErrorStatus err = psw_InitFile( g_lastStream, muteMode );
 
     if( err == dtkNoError )
          cout << "Parasolid writer: using in-memory stream (no .x_t file)." << endl;
     else
          cout << "Warning: psw_InitFile (memory) failed: " << dtkTypeError( err ).c_str() << endl;
 
     return err;
}

WriteBodyForParasolid()

void WriteBodyForParasolid	(	Dtk_BodyPtr	inBody,
		Dtk_tab< Dtk_ID > &	outIds
	)

{
 
     Dtk_tab<Dtk_ID> bodyIds;
 
     DtkErrorStatus errf, err;
 
     Dtk_InfoPtr infBody = inBody->info();
     if( infBody.IsNULL() )
     {
          infBody = Dtk_Info::Create();
     }
     // Add custom attributes to the body and its faces, edges, and vertices
     // Custom attributes are stored in the Parasolid file and can be retrieved later
     // For example, you can use the following code to retrieve the custom attributes from a Parasolid file:
     infBody->AddCustomerDefinedAttribute( L"my_body_id", inBody->get_info()->GetId() );
     // You can also add custom attributes as string
     infBody->AddCustomerDefinedAttribute( L"my_feature", Dtk_string( L"created_from_feature" ) );
 
     for( size_t i = 0; i < inBody->GetNumTopologicalEntities( DTK_TYPE_UNKNOWN ); i++ )
     {
          Dtk_FacePtr face = Dtk_FacePtr::DtkDynamicCast( inBody->GetPtr( i ) );
          if( face.IsNotNULL() && face->get_info().IsNotNULL() )
          {
               face->get_info()->AddCustomerDefinedAttribute( L"my_face_id", face->get_info()->GetId() );
          }
          Dtk_EdgePtr edge = Dtk_EdgePtr::DtkDynamicCast( inBody->GetPtr( i ) );
          if( edge.IsNotNULL() && edge->get_info().IsNotNULL() )
          {
               edge->get_info()->AddCustomerDefinedAttribute( L"my_edge_id", edge->get_info()->GetId() );
          }
          Dtk_VertexPtr vertex = Dtk_VertexPtr::DtkDynamicCast( inBody->GetPtr( i ) );
          if( vertex.IsNotNULL() && vertex->get_info().IsNotNULL() )
          {
               vertex->get_info()->AddCustomerDefinedAttribute( L"my_vertex_id", vertex->get_info()->GetId() );
          }
     }
 
 
 
     // Initialize the Parasolid file in memory 
     Dtk_tab<char> outParasolidStream;
     errf = psw_InitFile( outParasolidStream, 0 );
 
     // Initialize the Parasolid file on disk (uncomment the following lines to write to disk)
     // The filename is constructed using the body name and ID to ensure uniqueness
     //Dtk_string myFilename = infBody->GetName() +"_";
     //myFilename.add_int( infBody->GetId() );
     //myFilename += "_myxbfile.x_b";
     //errf = psw_InitFile( myFilename ,1,270);
 
     // Write the body to the Parasolid file or memory stream
     err = psw_WriteBody( inBody, bodyIds );
 
     // End the Parasolid file and write the body IDs to the output file or memory stream
     errf = psw_EndFile( bodyIds );
 
     // The following code is commented out because it requires the Parasolid kernel headers and libraries to be linked
     // It allows to read the Parasolid file from memory and transmit it to an existing Parasolid session
 
     //// #include "Include_pskernel\parasolid_kernel.h"
     //// PK_PART_receive_o_t recvOpts;
     //// PK_PART_receive_o_m( recvOpts );
     //// int nParts = 0;
     //// PK_PART_t* parts = nullptr;
 
     //// PK_MEMORY_block_t block;
     //// block.next = nullptr;
     //// block.n_bytes = static_cast< size_t >( outParasolidStream.size() );
     //// block.bytes = &outParasolidStream.GetArray()[ 0 ];
 
     ////// Load stream into Parasolid memory and receive the parts (bodies) from it
     //// PK_ERROR_code_t pkErr = PK_PART_receive_b( block, &recvOpts, &nParts, &parts);
 
     ////// Control by dumping a file
     ////PK_PART_transmit_o_t transmit_option;
     ////PK_PART_transmit_o_m( transmit_option );
     ////pkErr = PK_PART_transmit( nParts, parts, "xmtControlFile", &transmit_option );
 
 
     if( err == dtkNoError )
     {
          for( Dtk_Size_t i = 0; i < bodyIds.size(); ++i )
               outIds.push_back( bodyIds[ i ] );
     }
     else
     {
          cout << "Warning: psw_WriteBody failed: " << dtkTypeError( err ).c_str() << endl;
     }
}