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

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

diff --git a/src/HYDROGUI/HYDROGUI_CalculationOp.cxx b/src/HYDROGUI/HYDROGUI_CalculationOp.cxx
index 8e3e404f574236a7a5820a9283430fbc1974b728..707a308655bc1726839d9f5aa9a70db953a4e33b 100644 (file)
--- a/src/HYDROGUI/HYDROGUI_CalculationOp.cxx
+++ b/src/HYDROGUI/HYDROGUI_CalculationOp.cxx
@@ -93,12 +93,16 @@ void HYDROGUI_CalculationOp::startOperation()
     aPolylineObj = Handle(HYDROData_Polyline)::DownCast( anIter.Current() );
 
     if ( !aPolylineObj.IsNull() && aPolylineObj->IsClosed() )
-    {
-      aPolylineName = aPolylineObj->GetName();
-      if ( !aPolylineName.isEmpty() )
-      {
-        aList.append( aPolylineName );
-        anEntryList.append( HYDROGUI_DataObject::dataObjectEntry( aPolylineObj ) );
+    { 
+      // Check the polyline shape
+      TopoDS_Shape aPolylineShape = aPolylineObj->GetTopShape();
+      if ( !aPolylineShape.IsNull() && aPolylineShape.ShapeType() == TopAbs_WIRE ) {
+        aPolylineName = aPolylineObj->GetName();
+        if ( !aPolylineName.isEmpty() )
+        {
+          aList.append( aPolylineName );
+          anEntryList.append( HYDROGUI_DataObject::dataObjectEntry( aPolylineObj ) );
+        }
       }
     }
   }
@@ -154,7 +158,9 @@ void HYDROGUI_CalculationOp::getNamesAndEntries( const HYDROData_SequenceOfObjec
     anEntity = anIter.Value();
     if ( !anEntity.IsNull() )
     {
-      anObject = Handle(HYDROData_Object)::DownCast( anEntity );
+      // Temporary solution will be revised later
+      //anObject = Handle(HYDROData_Object)::DownCast( anEntity );
+      anObject = Handle(HYDROData_ImmersibleZone)::DownCast( anEntity );
       if ( !anObject.IsNull() )
       {
         theNames.append( anObject->GetName() );