Minor fix.

[modules/hydro.git] / src / HYDROGUI / HYDROGUI_DataModel.cxx

diff --git a/src/HYDROGUI/HYDROGUI_DataModel.cxx b/src/HYDROGUI/HYDROGUI_DataModel.cxx
index 4108df409fe06fa37bd36136b5745bd72118b5dc..abcc62743d6b4bc4045b06f76bfa6453acc5ecaa 100644 (file)
--- a/src/HYDROGUI/HYDROGUI_DataModel.cxx
+++ b/src/HYDROGUI/HYDROGUI_DataModel.cxx
@@ -29,6 +29,7 @@
 #include <HYDROData_Document.h>
 #include <HYDROData_Image.h>
 #include <HYDROData_Iterator.h>
+#include <HYDROData_Polyline.h>
 
 #include <CAM_Application.h>
 #include <CAM_DataObject.h>
@@ -147,8 +148,7 @@ bool HYDROGUI_DataModel::close()
 
 bool HYDROGUI_DataModel::isModified() const
 {
-  int aStudyId = module()->application()->activeStudy()->id();
-  return HYDROData_Document::Document( aStudyId )->IsModified();
+  return getDocument()->IsModified();
 }
 
 bool HYDROGUI_DataModel::isSaved() const
@@ -197,11 +197,11 @@ void HYDROGUI_DataModel::update( const int theStudyId )
 
   LightApp_DataObject* aPolylineRootObj = createObject( aRootObj, "POLYLINES" );
 
-  HYDROData_Iterator aPolyIterator( aDocument, KIND_IMAGE );
+  HYDROData_Iterator aPolyIterator( aDocument, KIND_POLYLINE );
   for( ; aPolyIterator.More(); aPolyIterator.Next() )
   {
-    Handle(HYDROData_Image) aPolylineObj =
-      Handle(HYDROData_Image)::DownCast( anIterator.Current() );
+    Handle(HYDROData_Polyline) aPolylineObj =
+      Handle(HYDROData_Polyline)::DownCast( aPolyIterator.Current() );
     if( !aPolylineObj.IsNull() )
       createObject( aPolylineRootObj, aPolylineObj );
   }
@@ -255,8 +255,7 @@ void HYDROGUI_DataModel::updateModel()
 Handle(HYDROData_Object) HYDROGUI_DataModel::objectByEntry( const QString& theEntry,
                                                             const ObjectKind theObjectKind )
 {
-  const int aStudyId = module()->application()->activeStudy()->id();
-  Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( aStudyId );
+  Handle(HYDROData_Document) aDocument = getDocument();
   if( !aDocument.IsNull() )
   {
     HYDROData_Iterator anIterator( aDocument, theObjectKind );