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 ca645914050cc0e2e43b44d96d0d82c044243928..4a1baf917f2cff0683977c17c43ec6dc5a2ba887 100644 (file)
--- a/src/HYDROGUI/HYDROGUI_DataModel.cxx
+++ b/src/HYDROGUI/HYDROGUI_DataModel.cxx
@@ -70,6 +70,11 @@
 #include <QApplication>
 #include <QDir>
 
+// #define DEB_GROUPS 1
+#ifdef DEB_GROUPS
+#include <HYDROData_ShapesGroup.h>
+#endif
+
 static HYDROData_SequenceOfObjects myCopyingObjects;
 
 HYDROGUI_DataModel::HYDROGUI_DataModel( CAM_Module* theModule )
@@ -721,6 +726,21 @@ void HYDROGUI_DataModel::buildObjectTree( SUIT_DataObject* theParent,
     Handle(HYDROData_DummyObject3D) anObject3D = aGeomObj->GetObject3D();
     if ( !anObject3D.IsNull() )
       createObject( aGuiObj, anObject3D, aGuiObj->entry(), false );
+
+#ifdef DEB_GROUPS
+    LightApp_DataObject* aGroupsSect = 
+      createObject( aGuiObj, tr( "OBJECT_GROUPS" ), aGuiObj->entry() );
+
+    HYDROData_SequenceOfObjects anObjGroups = aGeomObj->GetGroups();
+    HYDROData_SequenceOfObjects::Iterator anIter( anObjGroups );
+    for ( ; anIter.More(); anIter.Next() )
+    {
+      Handle(HYDROData_ShapesGroup) anObjGroup =
+        Handle(HYDROData_ShapesGroup)::DownCast( anIter.Value() );
+      if( !anObjGroup.IsNull() && !anObjGroup->IsRemoved() )
+        createObject( aGroupsSect, anObjGroup, aGuiObj->entry(), false );
+    }
+#endif
   }
 
   ObjectKind anObjectKind = aDataObj->GetKind();
@@ -755,6 +775,40 @@ void HYDROGUI_DataModel::buildObjectTree( SUIT_DataObject* theParent,
     if ( !aBathymetry.IsNull() && !aBathymetry->IsRemoved() )
       createObject( aBathSect, aBathymetry, aGuiObj->entry(), false );
   }
+  else if ( anObjectKind == KIND_POLYLINE )
+  {
+    Handle(HYDROData_Polyline3D) aPolyline3D =
+      Handle(HYDROData_Polyline3D)::DownCast( aDataObj );
+
+    LightApp_DataObject* aPolylineSect = 
+      createObject( aGuiObj, tr( "POLYLINE3D_POLYLINE" ), aGuiObj->entry() );
+
+    Handle(HYDROData_PolylineXY) aPolylineXY = aPolyline3D->GetPolylineXY();
+    if ( !aPolylineXY.IsNull() && !aPolylineXY->IsRemoved() )
+      createObject( aPolylineSect, aPolylineXY, aGuiObj->entry(), false );
+
+    LightApp_DataObject* aProfileSect = 
+      createObject( aGuiObj, tr( "POLYLINE3D_PROFILE" ), aGuiObj->entry() );
+
+    Handle(HYDROData_ProfileUZ) aProfileUZ = aPolyline3D->GetProfileUZ();
+    if ( aProfileUZ.IsNull() || aProfileUZ->IsRemoved() )
+      aProfileUZ = aPolyline3D->GetChildProfileUZ( false );
+
+    if ( !aProfileUZ.IsNull() && !aProfileUZ->IsRemoved() )
+    {
+      Handle(HYDROData_Profile) aProfile = 
+        Handle(HYDROData_Profile)::DownCast( aProfileUZ->GetFatherObject() );
+      if ( !aProfile.IsNull() && !aProfile->IsRemoved() )
+        createObject( aProfileSect, aProfile, aGuiObj->entry(), false );
+    }
+
+    LightApp_DataObject* aBathSect = 
+      createObject( aGuiObj, tr( "POLYLINE3D_BATHYMETRY" ), aGuiObj->entry() );
+
+    Handle(HYDROData_Bathymetry) aBathymetry = aPolyline3D->GetBathymetry();
+    if ( !aBathymetry.IsNull() && !aBathymetry->IsRemoved() )
+      createObject( aBathSect, aBathymetry, aGuiObj->entry(), false );
+  }
   else if ( anObjectKind == KIND_CALCULATION )
   {
     Handle(HYDROData_CalculationCase) aCaseObj =
@@ -795,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 )