First implementation of automatic mode of regions creation in the calculation case...

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

diff --git a/src/HYDROGUI/HYDROGUI_Poly3DOp.cxx b/src/HYDROGUI/HYDROGUI_Poly3DOp.cxx
index 9be0f7513dfcd06cf28f0061f4490a7ebf4b5269..5c8ff2f2a0606a5e0a0912ad13562c6de37b83c8 100644 (file)
--- a/src/HYDROGUI/HYDROGUI_Poly3DOp.cxx
+++ b/src/HYDROGUI/HYDROGUI_Poly3DOp.cxx
@@ -23,6 +23,7 @@
 #include "HYDROGUI_Poly3DOp.h"
 
 #include "HYDROGUI_Module.h"
+#include <HYDROGUI_DataObject.h>
 #include "HYDROGUI_Tool.h"
 #include "HYDROGUI_Poly3DDlg.h"
 #include "HYDROGUI_UpdateFlags.h"
@@ -31,6 +32,7 @@
 #include <HYDROData_Polyline3D.h>
 #include <HYDROData_Profile.h>
 #include <HYDROData_PolylineXY.h>
+#include <HYDROData_Bathymetry.h>
 
 #include <HYDROData_OperationsFactory.h>
 
@@ -69,20 +71,30 @@ void HYDROGUI_Poly3DOp::startOperation()
   }
   else
   {
-    aPoly3DName = HYDROGUI_Tool::GenerateObjectName( module(), tr( "POLYLINE_3D" ) );
+    aPoly3DName = HYDROGUI_Tool::GenerateObjectName( module(), tr( "DEFAULT_POLYLINE_3D_NAME" ) );
   }
   aPanel->setResultName( aPoly3DName );
 
-  QString aSelectedName1, aSelectedName2;
+  QString aPolyName, aProfileName, aBathName;
   if( myIsEdit && !myEditedObject.IsNull() )
   {
-    Handle(HYDROData_Entity) anObject1 = myEditedObject->GetProfileUZ()->GetFatherObject();
-    if( !anObject1.IsNull() )
-      aSelectedName1 = anObject1->GetName();
-    Handle(HYDROData_Entity) anObject2 = myEditedObject->GetPolylineXY();
-    if( !anObject2.IsNull() )
-      aSelectedName2 = anObject2->GetName();
-    aPanel->setSelectedObjects( aSelectedName1, aSelectedName2 );
+    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();
+
+    Handle(HYDROData_IAltitudeObject) anAltitudeObj = myEditedObject->GetAltitudeObject();
+    if( !anAltitudeObj.IsNull() )
+      aBathName = anAltitudeObj->GetName();
+
+    aPanel->setSelectedObjects( aPolyName, aProfileName, aBathName );
   }
   else if( !myIsEdit )
   {
@@ -97,7 +109,8 @@ void HYDROGUI_Poly3DOp::startOperation()
 }
 
 bool HYDROGUI_Poly3DOp::processApply( int& theUpdateFlags,
-                                         QString& theErrorMsg )
+                                      QString& theErrorMsg,
+                                      QStringList& theBrowseObjectsEntries )
 {
   HYDROGUI_Poly3DDlg* aPanel = dynamic_cast<HYDROGUI_Poly3DDlg*>( inputPanel() );
 
@@ -105,8 +118,8 @@ bool HYDROGUI_Poly3DOp::processApply( int& theUpdateFlags,
   if( aResultName.isEmpty() )
     return false;
 
-  QString aSelectedName1, aSelectedName2;
-  if( !aPanel->getSelectedObjects( aSelectedName1, aSelectedName2 ) )
+  QString aPolyName, aProfileName, aBathName;
+  if( !aPanel->getSelectedObjects( aPolyName, aProfileName, aBathName ) )
     return false;
 
   if( !myIsEdit || ( !myEditedObject.IsNull() && myEditedObject->GetName() != aResultName ) )
@@ -120,52 +133,63 @@ bool HYDROGUI_Poly3DOp::processApply( int& theUpdateFlags,
     }
   }
 
-  Handle(HYDROData_Entity) anObject1 =
-    HYDROGUI_Tool::FindObjectByName( module(), aSelectedName1, KIND_PROFILE ) ;
-  Handle(HYDROData_Entity) anObject2 =
-    HYDROGUI_Tool::FindObjectByName( module(), aSelectedName2, KIND_POLYLINEXY );
-  if( anObject1.IsNull() || anObject2.IsNull() )
-    return false;
-
-  Handle(HYDROData_Profile) aProfile = Handle(HYDROData_Profile)::DownCast( anObject1 );
-  Handle(HYDROData_PolylineXY) aPolyline = Handle(HYDROData_PolylineXY)::DownCast( anObject2 );
+  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 ) );
 
-  if( aProfile.IsNull() || aPolyline.IsNull() )
-    return false;
-
-  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
   {
     aResult = Handle(HYDROData_Polyline3D)::DownCast( doc()->CreateObject( KIND_POLYLINE ) );
+    QString anEntry = HYDROGUI_DataObject::dataObjectEntry( aResult );
+    theBrowseObjectsEntries.append( anEntry );
   }
 
   if( aResult.IsNull() )
     return false;
 
   aResult->SetName( aResultName );
-  aResult->SetProfileUZ( aProfileUZ );
-  aResult->SetPolylineXY( aPolyline );
+
+  aResult->SetPolylineXY( aPolyline, false );
+  if ( !aProfile.IsNull() )
+  {
+    Handle(HYDROData_ProfileUZ) aProfileUZ = aProfile->GetProfileUZ();
+    if( aProfileUZ.IsNull() )
+      return false;
+    
+    aResult->SetProfileUZ( aProfileUZ );
+  }
+  else
+  {
+    aResult->SetAltitudeObject( aBath );
+  }
+
+  if( !myIsEdit )
+  {
+    aResult->SetBorderColor( HYDROData_Polyline3D::DefaultBorderColor() );
+  }
 
   aResult->Update();
 
   if( !myIsEdit )
   {
-    size_t aViewId = HYDROGUI_Tool::GetActiveGraphicsViewId( module() );
-    module()->setObjectVisible( aViewId, anObject1, false );
-    module()->setObjectVisible( aViewId, anObject2, false );
+    size_t aViewId = HYDROGUI_Tool::GetActiveOCCViewId( module() );
+    module()->setObjectVisible( aViewId, aPolyline, false );
+    module()->setObjectVisible( aViewId, aProfile, false );
     module()->setObjectVisible( aViewId, aResult, true );
   }
+  module()->setIsToUpdate( aResult );
 
-  theUpdateFlags = UF_Model | UF_Viewer | UF_GV_Forced | UF_OCCViewer | UF_OCC_Forced;
+  theUpdateFlags = UF_Model | UF_Viewer | UF_GV_Forced | UF_OCCViewer | UF_OCC_Forced | UF_VTKViewer;
   return true;
 }