Selector implementation for OCC viewer.

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

diff --git a/src/HYDROGUI/HYDROGUI_DataModel.cxx b/src/HYDROGUI/HYDROGUI_DataModel.cxx
index 392dfef5bd9b6d1262e77ccb1097ef0e28c8366d..ba15923855300b568f22dfb160d1ca30602d7581 100644 (file)
--- a/src/HYDROGUI/HYDROGUI_DataModel.cxx
+++ b/src/HYDROGUI/HYDROGUI_DataModel.cxx
@@ -33,6 +33,8 @@
 #include <HYDROData_Iterator.h>
 #include <HYDROData_Polyline.h>
 #include <HYDROData_VisualState.h>
+#include <HYDROData_Region.h>
+#include <HYDROData_Zone.h>
 
 #include <CAM_Application.h>
 #include <CAM_DataObject.h>
@@ -227,17 +229,7 @@ void HYDROGUI_DataModel::update( const int theStudyId )
     Handle(HYDROData_Image) anImageObj =
       Handle(HYDROData_Image)::DownCast( anIterator.Current() );
     if( !anImageObj.IsNull() )
-    {
-      if( LightApp_DataObject* anImageDataObj = createObject( anImageRootObj, anImageObj ) )
-      {
-        for( int anIndex = 0, aNbRef = anImageObj->NbReferences(); anIndex < aNbRef; anIndex++ )
-        {
-          Handle(HYDROData_Object) aRefObj = anImageObj->Reference( anIndex );
-          if( !aRefObj.IsNull() && !aRefObj->IsRemoved() )
-            createObject( anImageDataObj, aRefObj, anImageDataObj->entry() );
-        }
-      }
-    }
+      createObject( anImageRootObj, anImageObj );
   }
 
   LightApp_DataObject* aBathymetryRootObj = createObject( aRootObj, partitionName( KIND_BATHYMETRY ) );
@@ -273,6 +265,17 @@ void HYDROGUI_DataModel::update( const int theStudyId )
       createObject( aPolylineRootObj, aPolylineObj );
   }
 
+  LightApp_DataObject* aZonesRootObj = createObject( aRootObj, partitionName( KIND_ZONE ) );
+
+  anIterator = HYDROData_Iterator( aDocument, KIND_ZONE );
+  for( ; anIterator.More(); anIterator.Next() )
+  {
+    Handle(HYDROData_Zone) aZoneObj =
+      Handle(HYDROData_Zone)::DownCast( anIterator.Current() );
+    if( !aZoneObj.IsNull() )
+      createObject( aZonesRootObj, aZoneObj );
+  }
+
   LightApp_DataObject* aVisualStateRootObj = createObject( aRootObj, partitionName( KIND_VISUAL_STATE ) );
 
   anIterator = HYDROData_Iterator( aDocument, KIND_VISUAL_STATE );
@@ -506,6 +509,7 @@ QString HYDROGUI_DataModel::partitionName( const ObjectKind theObjectKind )
     case KIND_VISUAL_STATE: return "VISUAL_STATES";
     case KIND_BATHYMETRY:   return "BATHYMETRIES";
     case KIND_CALCULATION:  return "CALCULATION_CASES";
+    case KIND_ZONE:         return "ZONES";
     default: break;
   }
   return QString();
@@ -517,17 +521,117 @@ Handle(HYDROData_Document) HYDROGUI_DataModel::getDocument() const
   return HYDROData_Document::Document( aStudyId );
 }
 
-LightApp_DataObject* HYDROGUI_DataModel::createObject( SUIT_DataObject* theParent,
+LightApp_DataObject* HYDROGUI_DataModel::createObject( SUIT_DataObject*         theParent,
                                                        Handle(HYDROData_Object) theModelObject,
-                                                       const QString& theParentEntry )
+                                                       const QString&           theParentEntry,
+                                                       const bool               theIsBuildTree )
 {
-  return new HYDROGUI_DataObject( theParent, theModelObject, theParentEntry );
+  HYDROGUI_DataObject* aResObj = 
+    new HYDROGUI_DataObject( theParent, theModelObject, theParentEntry );
+  
+  if ( theIsBuildTree )
+  {
+    buildObjectTree( theParent, aResObj, theParentEntry );
+  }
+
+  return aResObj;
 }
 
 LightApp_DataObject* HYDROGUI_DataModel::createObject( SUIT_DataObject* theParent,
-                                                       const QString& theName )
+                                                       const QString&   theName,
+                                                       const QString&   theParentEntry )
+{
+  return new HYDROGUI_NamedObject( theParent, theName, theParentEntry );
+}
+
+void HYDROGUI_DataModel::buildObjectTree( SUIT_DataObject* theParent,
+                                          SUIT_DataObject* theObject,
+                                          const QString&   theParentEntry )
 {
-  return new HYDROGUI_NamedObject( theParent, theName );
+  HYDROGUI_DataObject* aGuiObj = dynamic_cast<HYDROGUI_DataObject*>( theObject );
+  if ( !aGuiObj )
+    return;
+
+  Handle(HYDROData_Object) aDataObj = aGuiObj->modelObject();
+  if ( aDataObj.IsNull() )
+    return;
+
+  ObjectKind anObjectKind = aDataObj->GetKind();
+
+  if ( anObjectKind == KIND_IMAGE )
+  {
+    Handle(HYDROData_Image) anImageObj =
+      Handle(HYDROData_Image)::DownCast( aDataObj );
+    for ( int anIndex = 0, aNbRef = anImageObj->NbReferences(); anIndex < aNbRef; anIndex++ )
+    {
+      Handle(HYDROData_Object) aRefObj = anImageObj->Reference( anIndex );
+      if ( !aRefObj.IsNull() && !aRefObj->IsRemoved() )
+        createObject( aGuiObj, aRefObj, aGuiObj->entry(), false );
+    }
+  }
+  else if ( anObjectKind == KIND_ZONE )
+  {
+    Handle(HYDROData_Zone) aZoneObj =
+      Handle(HYDROData_Zone)::DownCast( aDataObj );
+
+    LightApp_DataObject* aPolylineSect = 
+      createObject( aGuiObj, tr( "ZONE_POLYLINE" ), aGuiObj->entry() );
+
+    Handle(HYDROData_Polyline) aPolyline = aZoneObj->GetPolyline();
+    if ( !aPolyline.IsNull() && !aPolyline->IsRemoved() )
+      createObject( aPolylineSect, aPolyline, aGuiObj->entry(), false );
+
+    LightApp_DataObject* aBathsSect = 
+      createObject( aGuiObj, tr( "ZONE_BATHYMETRIES" ), aGuiObj->entry() );
+
+    HYDROData_SequenceOfObjects aZoneBaths = aZoneObj->GetBathymetries();
+    HYDROData_SequenceOfObjects::Iterator aBathsIter( aZoneBaths );
+    for ( ; aBathsIter.More(); aBathsIter.Next() )
+    {
+      Handle(HYDROData_Bathymetry) aRefBath =
+        Handle(HYDROData_Bathymetry)::DownCast( aBathsIter.Value() );
+      if( !aRefBath.IsNull() && !aRefBath->IsRemoved() )
+        createObject( aBathsSect, aRefBath, aGuiObj->entry(), false );
+    }
+  }
+  else if ( anObjectKind == KIND_CALCULATION )
+  {
+    Handle(HYDROData_Calculation) aCaseObj =
+      Handle(HYDROData_Calculation)::DownCast( aDataObj );
+
+    LightApp_DataObject* aPolylineSect = 
+      createObject( aGuiObj, tr( "CASE_BND_POLYLINE" ), aGuiObj->entry() );
+
+    Handle(HYDROData_Polyline) aPolyline = aCaseObj->GetBoundaryPolyline();
+    if ( !aPolyline.IsNull() && !aPolyline->IsRemoved() )
+      createObject( aPolylineSect, aPolyline, aGuiObj->entry(), false );
+
+    LightApp_DataObject* aRefZonesSect = 
+      createObject( aGuiObj, tr( "CASE_REFERENCE_ZONES" ), aGuiObj->entry() );
+
+    HYDROData_SequenceOfObjects aRefZones = aCaseObj->GetZones();
+    HYDROData_SequenceOfObjects::Iterator anIter( aRefZones );
+    for ( ; anIter.More(); anIter.Next() )
+    {
+      Handle(HYDROData_Zone) aRefZone =
+        Handle(HYDROData_Zone)::DownCast( anIter.Value() );
+      if( !aRefZone.IsNull() && !aRefZone->IsRemoved() )
+        createObject( aRefZonesSect, aRefZone, aGuiObj->entry(), false );
+    }
+
+    LightApp_DataObject* aCaseRegionsSect = 
+      createObject( aGuiObj, tr( "CASE_REGIONS" ), aGuiObj->entry() );
+
+    HYDROData_SequenceOfObjects aCaseRegions = aCaseObj->GetRegions();
+    anIter.Init( aCaseRegions );
+    for ( ; anIter.More(); anIter.Next() )
+    {
+      Handle(HYDROData_Region) aCaseRegion =
+        Handle(HYDROData_Region)::DownCast( anIter.Value() );
+      if( !aCaseRegion.IsNull() && !aCaseRegion->IsRemoved() )
+        createObject( aCaseRegionsSect, aCaseRegion, aGuiObj->entry(), false );
+    }
+  }
 }
 
 void HYDROGUI_DataModel::removeChild( SUIT_DataObject* theParent,