0022616: [CEA 1038] Improve the quality of stl and vtk exports

[modules/geom.git] / src / GEOM / GEOM_BaseDriver.cxx

diff --git a/src/GEOM/GEOM_BaseDriver.cxx b/src/GEOM/GEOM_BaseDriver.cxx
index 2097da59306d017d74dcc558ad3d2f45beed8c4e..2ace1ce544e4ce1240bd0639c6153e450ff1ebdb 100644 (file)
--- a/src/GEOM/GEOM_BaseDriver.cxx
+++ b/src/GEOM/GEOM_BaseDriver.cxx
@@ -28,13 +28,29 @@
 
 #include "GEOM_Function.hxx"
 #include "GEOM_Object.hxx"
+#include "GEOM_Engine.hxx"
 
 #include <TColStd_HArray1OfInteger.hxx>
 #include <TDataStd_Name.hxx>
+#include <TDocStd_Owner.hxx>
 
 IMPLEMENT_STANDARD_HANDLE (GEOM_BaseDriver,TFunction_Driver);
 IMPLEMENT_STANDARD_RTTIEXT(GEOM_BaseDriver,TFunction_Driver);
 
+//================================================================================
+/*!
+ * Returns document id
+ */
+//================================================================================
+int GEOM_BaseDriver::GetDocID() const
+{
+  int docId = 0;
+  if (!Label().IsNull()) {
+    Handle(TDocStd_Document) aDoc = TDocStd_Owner::GetDocument(Label().Data());
+    docId = GEOM_Engine::GetEngine()->GetDocID(aDoc);
+  }
+  return docId;
+}
 
 //================================================================================
 /*!