Salome HOME
Creat\Edit stream operation.
[modules/hydro.git] / src / HYDROGUI / HYDROGUI_Operations.cxx
index 0378857cad988d630d31302ab8ca41267e7eeaeb..6e1bc51ca7a06c39aad7d5c114953e0bd794f183 100644 (file)
 #include "HYDROGUI_Module.h"
 #include "HYDROGUI_ObserveImageOp.h"
 #include "HYDROGUI_PolylineOp.h"
+#include "HYDROGUI_Poly3DOp.h"
+#include "HYDROGUI_ProfileOp.h"
 #include "HYDROGUI_RemoveImageRefsOp.h"
 #include "HYDROGUI_ShowHideOp.h"
-#include "HYDROData_SplitToZonesTool.h"
+#include "HYDROGUI_StreamOp.h"
 #include "HYDROGUI_TwoImagesOp.h"
 #include "HYDROGUI_UpdateFlags.h"
 #include "HYDROGUI_UpdateImageOp.h"
 #include "HYDROGUI_ImportGeomObjectOp.h"
 #include "HYDROGUI_ImportObstacleFromFileOp.h"
 #include "HYDROGUI_ExportCalculationOp.h"
+#include "HYDROGUI_ImportProfilesOp.h"
+#include "HYDROGUI_GeoreferencementOp.h"
+#include "HYDROGUI_SetColorOp.h"
 
-#include "HYDROData_Document.h"
-#include "HYDROData_Obstacle.h"
+#include <HYDROData_Document.h>
+#include <HYDROData_Obstacle.h>
+#include <HYDROData_SplitToZonesTool.h>
 
 #include <GeometryGUI.h>
 #include <GeometryGUI_Operations.h>
@@ -103,11 +109,23 @@ void HYDROGUI_Module::createActions()
   createAction( CreatePolylineId, "CREATE_POLYLINE" );
   createAction( EditPolylineId, "EDIT_POLYLINE" ); 
 
+  createAction( CreatePolyline3DId, "CREATE_POLYLINE_3D" );
+  createAction( EditPolyline3DId, "EDIT_POLYLINE_3D" ); 
+
+  createAction( CreateProfileId, "CREATE_PROFILE" );
+  createAction( ImportProfilesId, "IMPORT_PROFILES" );
+  createAction( EditProfileId, "EDIT_PROFILE" ); 
+  createAction( AllGeoreferencementId, "GEOREFERENCEMENT" ); 
+  createAction( SelectedGeoreferencementId, "GEOREFERENCEMENT" ); 
+  
   createAction( ImportBathymetryId, "IMPORT_BATHYMETRY", "", Qt::CTRL + Qt::Key_B );
 
   createAction( CreateImmersibleZoneId, "CREATE_IMMERSIBLE_ZONE" );
   createAction( EditImmersibleZoneId, "EDIT_IMMERSIBLE_ZONE" );
 
+  createAction( CreateStreamId, "CREATE_STREAM" );
+  createAction( EditStreamId, "EDIT_STREAM" );
+
   createAction( ImportObstacleFromFileId, "IMPORT_OBSTACLE_FROM_FILE" );
   createAction( ImportGeomObjectId, "IMPORT_GEOM_OBJECT" );
   createAction( CreateBoxId, "CREATE_BOX" );
@@ -126,7 +144,10 @@ void HYDROGUI_Module::createActions()
   createAction( SplitImageId, "SPLIT_IMAGE" );
   createAction( EditSplittedImageId, "EDIT_SPLITTED_IMAGE" );
 
-  createAction( DeleteId, "DELETE", "", Qt::Key_Delete );
+  createAction( DeleteId, "DELETE", "", Qt::Key_Delete, false,
+                SLOT( onDelete() ) );
+
+  createAction( SetColorId, "COLOR" );
 
   createAction( ShowId, "SHOW" );
   createAction( ShowOnlyId, "SHOW_ONLY" );
@@ -154,7 +175,15 @@ void HYDROGUI_Module::createMenus()
   createMenu( ImportImageId, aHydroId, -1, -1 );
   createMenu( ImportBathymetryId, aHydroId, -1, -1 );
   createMenu( CreatePolylineId, aHydroId, -1, -1 );
+  createMenu( CreatePolyline3DId, aHydroId, -1, -1 );
+
+  int aNewProfileId = createMenu( tr( "MEN_PROFILE" ), aHydroId, -1 );
+  createMenu( CreateProfileId, aNewProfileId, -1, -1 );
+  createMenu( ImportProfilesId, aNewProfileId, -1, -1 );
+  createMenu( AllGeoreferencementId, aNewProfileId, -1, -1 );
+
   createMenu( CreateImmersibleZoneId, aHydroId, -1, -1 );
+  createMenu( CreateStreamId, aHydroId, -1, -1 );
 
   int aNewObstacleId = createMenu( tr( "MEN_OBSTACLE" ), aHydroId, -1 );
   createMenu( ImportObstacleFromFileId, aNewObstacleId, -1, -1 );
@@ -229,6 +258,16 @@ void HYDROGUI_Module::onOperation()
     startOperation( anId );
 }
 
+void HYDROGUI_Module::onDelete()
+{
+  SUIT_Operation* anOp = application()->activeStudy()->activeOperation();
+  HYDROGUI_PolylineOp* aPolylineOp = dynamic_cast<HYDROGUI_PolylineOp*>( anOp );
+  if ( aPolylineOp && aPolylineOp->deleteEnabled() )
+    aPolylineOp->deleteSelected();
+  else
+    startOperation( DeleteId );
+}
+
 bool HYDROGUI_Module::onUndo( int theNumActions )
 {
   QApplication::setOverrideCursor( Qt::WaitCursor );
@@ -303,10 +342,27 @@ LightApp_Operation* HYDROGUI_Module::createOperation( const int theId ) const
   case RemoveImageRefsId:
     anOp = new HYDROGUI_RemoveImageRefsOp( aModule );
     break;
+  case CreatePolyline3DId:
+  case EditPolyline3DId:
+    anOp = new HYDROGUI_Poly3DOp( aModule, theId == EditPolyline3DId );
+    break;
   case CreatePolylineId:
   case EditPolylineId:
     anOp = new HYDROGUI_PolylineOp( aModule, theId == EditPolylineId );
     break;
+  case CreateProfileId:
+  case EditProfileId:
+    anOp = new HYDROGUI_ProfileOp( aModule, theId == EditProfileId );
+    break;
+  case ImportProfilesId:
+    anOp = new HYDROGUI_ImportProfilesOp( aModule ) ;
+    break;
+  case AllGeoreferencementId:
+    anOp = new HYDROGUI_GeoreferencementOp( aModule, HYDROGUI_GeoreferencementOp::All ) ;
+    break;
+  case SelectedGeoreferencementId:
+    anOp = new HYDROGUI_GeoreferencementOp( aModule, HYDROGUI_GeoreferencementOp::Selected ) ;
+    break;
   case ImportBathymetryId:
     anOp = new HYDROGUI_ImportBathymetryOp( aModule );
     break;
@@ -314,6 +370,10 @@ LightApp_Operation* HYDROGUI_Module::createOperation( const int theId ) const
   case EditImmersibleZoneId:
     anOp = new HYDROGUI_ImmersibleZoneOp( aModule, theId == EditImmersibleZoneId );
     break;
+  case CreateStreamId:
+  case EditStreamId:
+    anOp = new HYDROGUI_StreamOp( aModule, theId == EditStreamId );
+    break;
   case CreateCalculationId:
   case EditCalculationId:
     anOp = new HYDROGUI_CalculationOp( aModule, theId == EditCalculationId );
@@ -351,6 +411,9 @@ LightApp_Operation* HYDROGUI_Module::createOperation( const int theId ) const
   case DeleteId:
     anOp = new HYDROGUI_DeleteOp( aModule );
     break;
+  case SetColorId:
+    anOp = new HYDROGUI_SetColorOp( aModule );
+    break;
   case ShowId:
   case ShowOnlyId:
   case ShowAllId:
@@ -392,7 +455,7 @@ void HYDROGUI_Module::onExternalOperationFinished( const QString& theModuleName,
   
   // Start import GEOM object operation
   myGeomObjectsToImport = theEntryList;
-  startOperation ( ImportCreatedPrimitiveId );
+  startOperation( ImportCreatedPrimitiveId );
   myGeomObjectsToImport.clear();
 }