17.12.2013. Added Partition algorithm (draft version).

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

diff --git a/src/HYDROGUI/HYDROGUI_Poly3DOp.cxx b/src/HYDROGUI/HYDROGUI_Poly3DOp.cxx
index 74ea159aa4bf7d11278edda85f239a06092c746e..7a4bd951fb8a8848fdcfd4c9ba5e24b6836ff792 100644 (file)
--- a/src/HYDROGUI/HYDROGUI_Poly3DOp.cxx
+++ b/src/HYDROGUI/HYDROGUI_Poly3DOp.cxx
@@ -77,15 +77,21 @@ void HYDROGUI_Poly3DOp::startOperation()
   QString aPolyName, aProfileName, aBathName;
   if( myIsEdit && !myEditedObject.IsNull() )
   {
-    Handle(HYDROData_Entity) aProfile = myEditedObject->GetProfileUZ()->GetFatherObject();
-    if( !aProfile.IsNull() )
-      aProfileName = aProfile->GetName();
+    Handle(HYDROData_ProfileUZ) aProfileUZ = myEditedObject->GetProfileUZ();
+    if( !aProfileUZ.IsNull() )
+    {
+      Handle(HYDROData_Entity) aProfile = aProfileUZ->GetFatherObject();
+      if ( !aProfile.IsNull() )
+        aProfileName = aProfile->GetName();
+    }
 
     Handle(HYDROData_Entity) aPoly = myEditedObject->GetPolylineXY();
     if( !aPoly.IsNull() )
       aPolyName = aPoly->GetName();
 
-    //TODO: use bathymetry from data model
+    Handle(HYDROData_Bathymetry) aBathymetry = myEditedObject->GetBathymetry();
+    if( !aBathymetry.IsNull() )
+      aBathName = aBathymetry->GetName();
 
     aPanel->setSelectedObjects( aPolyName, aProfileName, aBathName );
   }
@@ -125,33 +131,20 @@ bool HYDROGUI_Poly3DOp::processApply( int& theUpdateFlags,
     }
   }
 
-  Handle(HYDROData_Entity) aProfileEnt =
-    HYDROGUI_Tool::FindObjectByName( module(), aProfileName, KIND_PROFILE ) ;
-  Handle(HYDROData_Entity) aPolyEnt =
-    HYDROGUI_Tool::FindObjectByName( module(), aPolyName, KIND_POLYLINEXY );
-  Handle(HYDROData_Entity) aBathEnt =
-    HYDROGUI_Tool::FindObjectByName( module(), aBathName, KIND_BATHYMETRY );
-
-  if( aPolyEnt.IsNull() || ( aPolyEnt.IsNull() && aBathEnt.IsNull() ) )
-    return false;
-
-  Handle(HYDROData_Profile) aProfile = Handle(HYDROData_Profile)::DownCast( aProfileEnt );
-  Handle(HYDROData_Bathymetry) aBath = Handle(HYDROData_Bathymetry)::DownCast( aBathEnt );
-  Handle(HYDROData_PolylineXY) aPolyline = Handle(HYDROData_PolylineXY)::DownCast( aPolyEnt );
-
-  if( aProfile.IsNull() || ( aPolyline.IsNull() && aBath.IsNull() ) )
-    return false;
+  Handle(HYDROData_Profile) aProfile = Handle(HYDROData_Profile)::DownCast( 
+    HYDROGUI_Tool::FindObjectByName( module(), aProfileName, KIND_PROFILE ) );
+  Handle(HYDROData_PolylineXY) aPolyline = Handle(HYDROData_PolylineXY)::DownCast( 
+    HYDROGUI_Tool::FindObjectByName( module(), aPolyName, KIND_POLYLINEXY ) );
+  Handle(HYDROData_Bathymetry) aBath = Handle(HYDROData_Bathymetry)::DownCast( 
+    HYDROGUI_Tool::FindObjectByName( module(), aBathName, KIND_BATHYMETRY ) );
 
-  Handle(HYDROData_ProfileUZ) aProfileUZ = aProfile->GetProfileUZ();
-  if( aProfileUZ.IsNull() )
+  if ( aPolyline.IsNull() || ( aProfile.IsNull() && aBath.IsNull() ) )
     return false;
 
   Handle(HYDROData_Polyline3D) aResult;
   if( myIsEdit )
   {
     aResult = myEditedObject;
-    aResult->RemoveProfileUZ();
-    aResult->RemovePolylineXY();
   }
   else
   {
@@ -162,9 +155,20 @@ bool HYDROGUI_Poly3DOp::processApply( int& theUpdateFlags,
     return false;
 
   aResult->SetName( aResultName );
-  aResult->SetProfileUZ( aProfileUZ );
-  aResult->SetPolylineXY( aPolyline );
-  //TODO: set bathymetry
+
+  aResult->SetPolylineXY( aPolyline, false );
+  if ( !aProfile.IsNull() )
+  {
+    Handle(HYDROData_ProfileUZ) aProfileUZ = aProfile->GetProfileUZ();
+    if( aProfileUZ.IsNull() )
+      return false;
+    
+    aResult->SetProfileUZ( aProfileUZ );
+  }
+  else
+  {
+    aResult->SetBathymetry( aBath );
+  }
 
   if( !myIsEdit )
   {