Feature #86: The hierarchy in the Object Browser (T 19).

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

diff --git a/src/HYDROGUI/HYDROGUI_DataModel.cxx b/src/HYDROGUI/HYDROGUI_DataModel.cxx
index 05259c16e82dbe655945a87c6e067d754d6a644a..9432c01c02f2cd996db52aad98d688073acce22c 100644 (file)
--- a/src/HYDROGUI/HYDROGUI_DataModel.cxx
+++ b/src/HYDROGUI/HYDROGUI_DataModel.cxx
@@ -35,10 +35,14 @@
 #include <HYDROData_ImmersibleZone.h>
 #include <HYDROData_Iterator.h>
 #include <HYDROData_Polyline.h>
+#include <HYDROData_Profile.h>
 #include <HYDROData_VisualState.h>
 #include <HYDROData_Region.h>
 #include <HYDROData_Zone.h>
 #include <HYDROData_Obstacle.h>
+#include <HYDROData_Channel.h>
+#include <HYDROData_Digue.h>
+#include <HYDROData_River.h>
 
 #include <CAM_Application.h>
 #include <CAM_DataObject.h>
@@ -225,81 +229,151 @@ void HYDROGUI_DataModel::update( const int theStudyId )
   if( aDocument.IsNull() )
     return;
 
-  LightApp_DataObject* anImageRootObj = createObject( aRootObj, tr( partitionName( KIND_IMAGE ).toAscii() ) );
+  // Create root objects:
 
-  HYDROData_Iterator anIterator( aDocument, KIND_IMAGE );
-  for( ; anIterator.More(); anIterator.Next() )
-  {
-    Handle(HYDROData_Image) anImageObj =
-      Handle(HYDROData_Image)::DownCast( anIterator.Current() );
-    if( !anImageObj.IsNull() )
-      createObject( anImageRootObj, anImageObj );
-  }
+  // IMAGES
+  LightApp_DataObject* anImageRootObj = createObject( aRootObj, tr( partitionName( KIND_IMAGE ).toAscii() ) );
 
+  // BATHYMETRY
   LightApp_DataObject* aBathymetryRootObj = createObject( aRootObj, tr( partitionName( KIND_BATHYMETRY ).toAscii() ) );
 
-  anIterator = HYDROData_Iterator( aDocument, KIND_BATHYMETRY );
-  for( ; anIterator.More(); anIterator.Next() )
-  {
-    Handle(HYDROData_Bathymetry) aBathymetryObj =
-      Handle(HYDROData_Bathymetry)::DownCast( anIterator.Current() );
-    if( !aBathymetryObj.IsNull() )
-      createObject( aBathymetryRootObj, aBathymetryObj );
-  }
+  // ARTIFICIAL OBJECTS
+  LightApp_DataObject* anArtificialObjectsRootObj = createObject( aRootObj, tr( partitionName( KIND_ARTIFICIAL_OBJECT ).toAscii() ) );
 
+  // NATURAL OBJECTS
+  LightApp_DataObject* aNaturalObjectsRootObj = createObject( aRootObj, tr( partitionName( KIND_NATURAL_OBJECT ).toAscii() ) );
+
+  // OBSTACLES
+  LightApp_DataObject* anObstaclesRootObj = createObject( aRootObj, tr( partitionName( KIND_OBSTACLE ).toAscii() ) );
+
+  // CALCULATION CASES
+  LightApp_DataObject* aCalculRootObj = createObject( aRootObj, tr( partitionName( KIND_CALCULATION ).toAscii() ) );
+
+  // POLYLINES
   LightApp_DataObject* aPolylineRootObj = createObject( aRootObj, tr( partitionName( KIND_POLYLINE ).toAscii() ) );
 
-  anIterator = HYDROData_Iterator( aDocument, KIND_POLYLINE );
-  for( ; anIterator.More(); anIterator.Next() )
-  {
-    Handle(HYDROData_Polyline) aPolylineObj =
-      Handle(HYDROData_Polyline)::DownCast( anIterator.Current() );
-    if( !aPolylineObj.IsNull() )
-      createObject( aPolylineRootObj, aPolylineObj );
-  }
+  // PROFILES
+  LightApp_DataObject* aProfileRootObj = createObject( aRootObj, tr( partitionName( KIND_PROFILE ).toAscii() ) );
 
-  LightApp_DataObject* aZonesRootObj = createObject( aRootObj, tr( partitionName( KIND_IMMERSIBLE_ZONE ).toAscii() ) );
+  // VISUAL STATES
+  LightApp_DataObject* aVisualStateRootObj = createObject( aRootObj, tr( partitionName( KIND_VISUAL_STATE ).toAscii() ) );
 
-  anIterator = HYDROData_Iterator( aDocument, KIND_IMMERSIBLE_ZONE );
-  for( ; anIterator.More(); anIterator.Next() )
-  {
-    Handle(HYDROData_ImmersibleZone) aZoneObj =
-      Handle(HYDROData_ImmersibleZone)::DownCast( anIterator.Current() );
-    if( !aZoneObj.IsNull() )
-      createObject( aZonesRootObj, aZoneObj );
-  }
+  HYDROData_Iterator anIterator( aDocument, KIND_UNKNOWN );
+  for( ; anIterator.More(); anIterator.Next() ) {
+    Handle(HYDROData_Entity) anObj = anIterator.Current();
 
-  LightApp_DataObject* anObstaclesRootObj = createObject( aRootObj, tr( partitionName( KIND_OBSTACLE ).toAscii() ) );
+    switch ( anObj->GetKind() ) {
+      case KIND_IMAGE:
+      {
+        Handle(HYDROData_Image) anImageObj =
+          Handle(HYDROData_Image)::DownCast( anObj );
+        if( !anImageObj.IsNull() ) {
+          createObject( anImageRootObj, anImageObj );
+        }
 
-  anIterator = HYDROData_Iterator( aDocument, KIND_OBSTACLE );
-  for( ; anIterator.More(); anIterator.Next() )
-  {
-    Handle(HYDROData_Obstacle) anObstacleObj =
-      Handle(HYDROData_Obstacle)::DownCast( anIterator.Current() );
-    if( !anObstacleObj.IsNull() )
-      createObject( anObstaclesRootObj, anObstacleObj );
-  }
+        break;
+      }
+      case KIND_BATHYMETRY:
+      {
+        Handle(HYDROData_Bathymetry) aBathymetryObj =
+          Handle(HYDROData_Bathymetry)::DownCast( anObj );
+        if( !aBathymetryObj.IsNull() ) {
+          createObject( aBathymetryRootObj, aBathymetryObj );
+        }
 
-  LightApp_DataObject* aCalculRootObj = createObject( aRootObj, tr( partitionName( KIND_CALCULATION ).toAscii() ) );
+        break;
+      }
+      case KIND_CHANNEL:
+      {
+        Handle(HYDROData_Channel) aChannelObj =
+          Handle(HYDROData_Channel)::DownCast( anObj );
+        if( !aChannelObj.IsNull() ) {
+          createObject( anArtificialObjectsRootObj, aChannelObj );
+        }
 
-  anIterator = HYDROData_Iterator( aDocument, KIND_CALCULATION );
-  for( ; anIterator.More(); anIterator.Next() )
-  {
-    Handle(HYDROData_CalculationCase) aCalculObj =
-      Handle(HYDROData_CalculationCase)::DownCast( anIterator.Current() );
-    if( !aCalculObj.IsNull() )
-      createObject( aCalculRootObj, aCalculObj );
-  }
+        break;
+      }
+      case KIND_DIGUE:
+      {
+        Handle(HYDROData_Digue) aDigueObj =
+          Handle(HYDROData_Digue)::DownCast( anObj );
+        if( !aDigueObj.IsNull() ) {
+          createObject( anArtificialObjectsRootObj, aDigueObj );
+        }
 
-  LightApp_DataObject* aVisualStateRootObj = createObject( aRootObj, tr( partitionName( KIND_VISUAL_STATE ).toAscii() ) );
+        break;
+      }
+      case KIND_IMMERSIBLE_ZONE:
+      {
+        Handle(HYDROData_ImmersibleZone) anImmersibleZoneObj =
+          Handle(HYDROData_ImmersibleZone)::DownCast( anObj );
+        if( !anImmersibleZoneObj.IsNull() ) {
+          createObject( aNaturalObjectsRootObj, anImmersibleZoneObj );
+        }
 
-  anIterator = HYDROData_Iterator( aDocument, KIND_VISUAL_STATE );
-  for( ; anIterator.More(); anIterator.Next() )
-  {
-    Handle(HYDROData_VisualState) aVisualStateObj =
-      Handle(HYDROData_VisualState)::DownCast( anIterator.Current() );
-    if( !aVisualStateObj.IsNull() )
-      createObject( aVisualStateRootObj, aVisualStateObj );
+        break;
+      }
+      case KIND_RIVER:
+      {
+        Handle(HYDROData_River) aRiverObj =
+          Handle(HYDROData_River)::DownCast( anObj );
+        if( !aRiverObj.IsNull() ) {
+          createObject( aNaturalObjectsRootObj, aRiverObj );
+        }
+
+        break;
+      }
+      case KIND_OBSTACLE:
+      {
+        Handle(HYDROData_Obstacle) anObstacleObj =
+          Handle(HYDROData_Obstacle)::DownCast( anObj );
+        if( !anObstacleObj.IsNull() ) {
+          createObject( anObstaclesRootObj, anObstacleObj );
+        }
+
+        break;
+      }
+      case KIND_CALCULATION:
+      {
+        Handle(HYDROData_CalculationCase) aCalculObj =
+          Handle(HYDROData_CalculationCase)::DownCast( anObj );
+        if( !aCalculObj.IsNull() ) {
+          createObject( aCalculRootObj, aCalculObj );
+        }
+
+        break;
+      }
+      case KIND_POLYLINE:
+      {
+        Handle(HYDROData_Polyline) aPolylineObj =
+          Handle(HYDROData_Polyline)::DownCast( anObj );
+        if( !aPolylineObj.IsNull() ) {
+          createObject( aPolylineRootObj, aPolylineObj );
+        }
+
+        break;
+      }
+      case KIND_PROFILE:
+      {
+        Handle(HYDROData_Profile) aProfileObj =
+          Handle(HYDROData_Profile)::DownCast( anObj );
+        if( !aProfileObj.IsNull() ) {
+          createObject( aProfileRootObj, aProfileObj );
+        }
+
+        break;
+      }
+      case KIND_VISUAL_STATE:
+      {
+        Handle(HYDROData_VisualState) aVisualStateObj =
+          Handle(HYDROData_VisualState)::DownCast( anObj );
+        if( !aVisualStateObj.IsNull() ) {
+          createObject( aVisualStateRootObj, aVisualStateObj );
+        }
+
+        break;
+      }
+    }
   }
 
   if( SUIT_DataBrowser* anObjectBrowser = anApp->objectBrowser() )
@@ -448,6 +522,7 @@ bool HYDROGUI_DataModel::canCopy()
   ObjectKind aKind = anObject->GetKind();
   if( aKind == KIND_IMAGE ||
       aKind == KIND_POLYLINE ||
+      aKind == KIND_PROFILE ||
       aKind == KIND_CALCULATION )
     return true;
 
@@ -519,13 +594,15 @@ QString HYDROGUI_DataModel::partitionName( const ObjectKind theObjectKind )
 {
   switch( theObjectKind )
   {
-    case KIND_IMAGE:           return "IMAGES";
-    case KIND_POLYLINE:        return "POLYLINES";
-    case KIND_VISUAL_STATE:    return "VISUAL_STATES";
-    case KIND_BATHYMETRY:      return "BATHYMETRIES";
-    case KIND_CALCULATION:     return "CALCULATION_CASES";
-    case KIND_IMMERSIBLE_ZONE: return "IMMERSIBLE_ZONES";
-    case KIND_OBSTACLE:        return "OBSTACLES";
+    case KIND_IMAGE:             return "IMAGES";
+    case KIND_POLYLINE:          return "POLYLINES";
+    case KIND_PROFILE:           return "PROFILES";
+    case KIND_VISUAL_STATE:      return "VISUAL_STATES";
+    case KIND_BATHYMETRY:        return "BATHYMETRIES";
+    case KIND_CALCULATION:       return "CALCULATION_CASES";
+    case KIND_OBSTACLE:          return "OBSTACLES";
+    case KIND_ARTIFICIAL_OBJECT: return "ARTIFICIAL_OBJECTS";
+    case KIND_NATURAL_OBJECT:    return "NATURAL_OBJECTS";
     default: break;
   }
   return QString();