0023216: [CEA 1691] Import the compounds with their names from a STEP file

[modules/geom.git] / src / STEPPlugin / STEPPlugin_IImport.hxx

diff --git a/src/STEPPlugin/STEPPlugin_IImport.hxx b/src/STEPPlugin/STEPPlugin_IImport.hxx
index 3b63fdfdc4c185a6cde0822fd137deacf98ebe62..f7630e4ebb2fcb84d8e79d0fc953173a7dfdb35c 100644 (file)
--- a/src/STEPPlugin/STEPPlugin_IImport.hxx
+++ b/src/STEPPlugin/STEPPlugin_IImport.hxx
@@ -22,8 +22,9 @@
 
 #include "GEOM_Function.hxx"
 
-#define IMPORTSTEP_ARG_FILENAME     1
-#define IMPORTSTEP_ARG_IGNORE_UNITS 2
+#define IMPORTSTEP_ARG_FILENAME          1
+#define IMPORTSTEP_ARG_IGNORE_UNITS      2
+#define IMPORTSTEP_ARG_CREATE_ASSEMBLIES 3
 
 class STEPPlugin_IImport
 {
@@ -41,6 +42,12 @@ public:
   bool GetIsIgnoreUnits()
     { return bool( _func->GetInteger( IMPORTSTEP_ARG_IGNORE_UNITS ) ); }
 
+  void SetIsCreateAssemblies( bool IsCreateAssemblies )
+  { _func->SetInteger
+        ( IMPORTSTEP_ARG_CREATE_ASSEMBLIES, IsCreateAssemblies ? 1 : 0 ); }
+  bool GetIsCreateAssemblies()
+    { return ( _func->GetInteger( IMPORTSTEP_ARG_CREATE_ASSEMBLIES ) != 0 ); }
+
 private:
   Handle(GEOM_Function) _func;
 };