Create goups for stream.

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

diff --git a/src/HYDROGUI/HYDROGUI_DataModel.cxx b/src/HYDROGUI/HYDROGUI_DataModel.cxx
index 084fab39315e0a7b4468ff2b3ef3768ab9a512ab..4a1baf917f2cff0683977c17c43ec6dc5a2ba887 100644 (file)
--- a/src/HYDROGUI/HYDROGUI_DataModel.cxx
+++ b/src/HYDROGUI/HYDROGUI_DataModel.cxx
@@ -72,7 +72,7 @@
 
 // #define DEB_GROUPS 1
 #ifdef DEB_GROUPS
-#include <HYDROData_EdgesGroup.h>
+#include <HYDROData_ShapesGroup.h>
 #endif
 
 static HYDROData_SequenceOfObjects myCopyingObjects;
@@ -735,8 +735,8 @@ void HYDROGUI_DataModel::buildObjectTree( SUIT_DataObject* theParent,
     HYDROData_SequenceOfObjects::Iterator anIter( anObjGroups );
     for ( ; anIter.More(); anIter.Next() )
     {
-      Handle(HYDROData_EdgesGroup) anObjGroup =
-        Handle(HYDROData_EdgesGroup)::DownCast( anIter.Value() );
+      Handle(HYDROData_ShapesGroup) anObjGroup =
+        Handle(HYDROData_ShapesGroup)::DownCast( anIter.Value() );
       if( !anObjGroup.IsNull() && !anObjGroup->IsRemoved() )
         createObject( aGroupsSect, anObjGroup, aGuiObj->entry(), false );
     }
@@ -849,6 +849,17 @@ void HYDROGUI_DataModel::buildObjectTree( SUIT_DataObject* theParent,
 
     aGuiObj->setIsValid( aProfileObj->IsValid() );
   }
+  else if ( anObjectKind == KIND_CHANNEL || anObjectKind == KIND_DIGUE )
+  {
+    Handle(HYDROData_Channel) aChannelObj =
+      Handle(HYDROData_Channel)::DownCast( aDataObj );
+
+    Handle(HYDROData_Profile) aProfile = aChannelObj->GetProfile();
+    Handle(HYDROData_Polyline3D) aGuideLine = aChannelObj->GetGuideLine();
+
+    createObject( aGuiObj, aProfile, aGuiObj->entry(), false );
+    createObject( aGuiObj, aGuideLine, aGuiObj->entry(), false );
+  }
 }
 
 void HYDROGUI_DataModel::buildCaseTree( SUIT_DataObject* theParent, Handle(HYDROData_CalculationCase) theCase )