Update flag for calculation case during polyline changing.

[modules/hydro.git] / src / HYDROData / HYDROData_Document.h

diff --git a/src/HYDROData/HYDROData_Document.h b/src/HYDROData/HYDROData_Document.h
index 28ac1f4133ccafc2cf9042fc7f9f79704a17e42d..13bdffb007a9cdbe653e3d4065be3e46bf4ecf19 100644 (file)
--- a/src/HYDROData/HYDROData_Document.h
+++ b/src/HYDROData/HYDROData_Document.h
@@ -2,7 +2,7 @@
 #define HYDROData_Document_HeaderFile
 
 #include <HYDROData.h>
-#include <HYDROData_Object.h>
+#include <HYDROData_Entity.h>
 
 #include <TDocStd_Document.hxx>
 
@@ -41,6 +41,10 @@ public:
   //! Returns the existing document or creates new if it is not exist
   HYDRODATA_EXPORT static Handle(HYDROData_Document) Document(const int theStudyID);
 
+  //! Returns the document by object
+  HYDRODATA_EXPORT static Handle(HYDROData_Document) Document( 
+    const TDF_Label& theObjectLabel );
+
   //! Returns true if data model contains document for this study
   HYDRODATA_EXPORT static bool HasDocument(const int theStudyID);
 
@@ -106,7 +110,7 @@ public:
   //! Creates and locates in the document a new object
   //! \param theKind kind of the created object, can not be UNKNOWN
   //! \returns the created object
-  HYDRODATA_EXPORT Handle_HYDROData_Object CreateObject(const ObjectKind theKind);
+  HYDRODATA_EXPORT Handle(HYDROData_Entity) CreateObject(const ObjectKind theKind);
 
 protected: