Python console has been added for HYDRO module (Bug #22).

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

diff --git a/src/HYDROGUI/HYDROGUI_DataModel.cxx b/src/HYDROGUI/HYDROGUI_DataModel.cxx
index e72d9488163a46cac9d7c3e38bcad4d7852a9836..d1fff0327b08f53aa0fa5cc948799ac3712acd6c 100644 (file)
--- a/src/HYDROGUI/HYDROGUI_DataModel.cxx
+++ b/src/HYDROGUI/HYDROGUI_DataModel.cxx
@@ -27,6 +27,7 @@
 #include "HYDROGUI_Tool.h"
 
 #include <HYDROData_Bathymetry.h>
+#include <HYDROData_Calculation.h>
 #include <HYDROData_Document.h>
 #include <HYDROData_Image.h>
 #include <HYDROData_Iterator.h>
@@ -248,6 +249,17 @@ void HYDROGUI_DataModel::update( const int theStudyId )
       createObject( aBathymetryRootObj, aBathymetryObj );
   }
 
+  LightApp_DataObject* aCalculRootObj = createObject( aRootObj, "CALCULATION CASES" );
+
+  anIterator = HYDROData_Iterator( aDocument, KIND_CALCULATION );
+  for( ; anIterator.More(); anIterator.Next() )
+  {
+    Handle(HYDROData_Calculation) aCalculObj =
+      Handle(HYDROData_Calculation)::DownCast( anIterator.Current() );
+    if( !aCalculObj.IsNull() )
+      createObject( aCalculRootObj, aCalculObj );
+  }
+
   LightApp_DataObject* aPolylineRootObj = createObject( aRootObj, "POLYLINES" );
 
   anIterator = HYDROData_Iterator( aDocument, KIND_POLYLINE );