Salome HOME
Modify creation of curves: 1) using QDockWidget instead of QDialog; 2) selection...
[modules/hydro.git] / src / HYDROGUI / HYDROGUI_Operations.cxx
index 77d054542e69df5fba289ad4527f98bd9e15c5b5..72c51c9a2d295398b918fcdf6d1bf11aa49e494d 100644 (file)
 
 #include "HYDROGUI_Operations.h"
 
+#include "HYDROGUI_CopyPasteOp.h"
 #include "HYDROGUI_DataModel.h"
 #include "HYDROGUI_DeleteOp.h"
 #include "HYDROGUI_ExportImageOp.h"
 #include "HYDROGUI_ImportImageOp.h"
+#include "HYDROGUI_ImportBathymetryOp.h"
+#include "HYDROGUI_CalculationOp.h"
 #include "HYDROGUI_Module.h"
 #include "HYDROGUI_ObserveImageOp.h"
 #include "HYDROGUI_PolylineOp.h"
 #include "HYDROGUI_ShowHideOp.h"
+#include "HYDROGUI_SplitZonesTool.h"
 #include "HYDROGUI_TwoImagesOp.h"
 #include "HYDROGUI_UpdateFlags.h"
+#include "HYDROGUI_UpdateImageOp.h"
 #include "HYDROGUI_VisualStateOp.h"
+#include "HYDROGUI_ZoneOp.h"
 
 #include <CAM_Application.h>
 
@@ -68,16 +74,34 @@ void HYDROGUI_Module::createActions()
   createAction( SaveVisualStateId, "SAVE_VISUAL_STATE" );
   createAction( LoadVisualStateId, "LOAD_VISUAL_STATE" );
 
+  createAction( CopyId, "COPY", "", Qt::CTRL + Qt::Key_C );
+  createAction( PasteId, "PASTE", "", Qt::CTRL + Qt::Key_V );
+
   createAction( ImportImageId, "IMPORT_IMAGE", "", Qt::CTRL + Qt::Key_I );
   createAction( EditImportedImageId, "EDIT_IMPORTED_IMAGE" );
-  createAction( EditCompositeImageId, "EDIT_COMPOSITE_IMAGE" );
   createAction( ObserveImageId, "OBSERVE_IMAGE" );
   createAction( ExportImageId, "EXPORT_IMAGE" );
+  createAction( UpdateImageId, "UPDATE_IMAGE" );
+
   createAction( CreatePolylineId, "CREATE_POLYLINE" );
   createAction( EditPolylineId, "EDIT_POLYLINE" ); 
 
-  createAction( FuseId, "FUSE_IMAGES" );
-  createAction( CutId, "CUT_IMAGES" );
+  createAction( ImportBathymetryId, "IMPORT_BATHYMETRY", "", Qt::CTRL + Qt::Key_B );
+
+  createAction( CreateZoneId, "CREATE_ZONE" );
+  createAction( EditZoneId, "EDIT_ZONE" );
+
+  createAction( CreateCalculationId, "CREATE_CALCULATION" );
+  createAction( EditCalculationId, "EDIT_CALCULATION" );
+
+  createAction( FuseImagesId, "FUSE_IMAGES" );
+  createAction( EditFusedImageId, "EDIT_FUSED_IMAGE" );
+
+  createAction( CutImagesId, "CUT_IMAGES" );
+  createAction( EditCutImageId, "EDIT_CUT_IMAGE" );
+
+  createAction( SplitImageId, "SPLIT_IMAGE" );
+  createAction( EditSplittedImageId, "EDIT_SPLITTED_IMAGE" );
 
   createAction( DeleteId, "DELETE", "", Qt::Key_Delete );
 
@@ -98,13 +122,24 @@ void HYDROGUI_Module::createMenus()
   int anEditMenu = createMenu( tr( "MEN_DESK_EDIT" ), -1, -1, 5 );
   createMenu( UndoId, anEditMenu );
   createMenu( RedoId, anEditMenu );
+  createMenu( separator(), anEditMenu );
+  createMenu( CopyId, anEditMenu );
+  createMenu( PasteId, anEditMenu );
 
   int aHydroMenu = 6; // Edit menu id == 5, View menu id == 10
   int aHydroId = createMenu( tr( "MEN_DESK_HYDRO" ), -1, -1, aHydroMenu );
   createMenu( ImportImageId, aHydroId, -1, -1 );
+  createMenu( ImportBathymetryId, aHydroId, -1, -1 );
   createMenu( CreatePolylineId, aHydroId, -1, -1 );
-  createMenu( FuseId, aHydroId, -1, -1 );
-  createMenu( CutId, aHydroId, -1, -1 );
+  createMenu( CreateZoneId, aHydroId, -1, -1 );
+  createMenu( CreateCalculationId, aHydroId, -1, -1 );
+  createMenu( separator(), aHydroId );
+  createMenu( FuseImagesId, aHydroId, -1, -1 );
+  createMenu( CutImagesId, aHydroId, -1, -1 );
+  createMenu( SplitImageId, aHydroId, -1, -1 );
+
+  createMenu( separator(), aHydroId );
+  createMenu( TestSplitId, aHydroId, -1, -1 );
 }
 
 void HYDROGUI_Module::createPopups()
@@ -113,7 +148,7 @@ void HYDROGUI_Module::createPopups()
 
 void HYDROGUI_Module::createToolbars()
 {
-  int aToolBar = createTool( tr( "HYDRO_TOOLBAR" ) );
+  int aToolBar = createTool( tr( "MEN_DESK_HYDRO" ) );
   createTool( UndoId, aToolBar );
   createTool( RedoId, aToolBar );
 }
@@ -138,6 +173,10 @@ void HYDROGUI_Module::createUndoRedoActions()
 
   connect( anEditUndo, SIGNAL( triggered( int ) ), this, SLOT( onUndo( int ) ) );
   connect( anEditRedo, SIGNAL( triggered( int ) ), this, SLOT( onRedo( int ) ) );
+
+  QAction* aTestSplit = new QAction( "Test split", application()->desktop() );
+  registerAction( TestSplitId, aTestSplit );
+  connect( aTestSplit, SIGNAL( triggered() ), this, SLOT( onTestSplit() ) );
 }
 
 void HYDROGUI_Module::updateUndoRedoControls()
@@ -221,6 +260,10 @@ LightApp_Operation* HYDROGUI_Module::createOperation( const int theId ) const
   case LoadVisualStateId:
     anOp = new HYDROGUI_VisualStateOp( aModule, theId == LoadVisualStateId );
     break;
+  case CopyId:
+  case PasteId:
+    anOp = new HYDROGUI_CopyPasteOp( aModule, theId == PasteId );
+    break;
   case ImportImageId:
   case EditImportedImageId:
     anOp = new HYDROGUI_ImportImageOp( aModule, theId == EditImportedImageId );
@@ -231,18 +274,35 @@ LightApp_Operation* HYDROGUI_Module::createOperation( const int theId ) const
   case ExportImageId:
     anOp = new HYDROGUI_ExportImageOp( aModule );
     break;
+  case UpdateImageId:
+    anOp = new HYDROGUI_UpdateImageOp( aModule );
+    break;
   case CreatePolylineId:
   case EditPolylineId:
     anOp = new HYDROGUI_PolylineOp( aModule, theId == EditPolylineId );
     break;
-  case FuseId:
-    anOp = new HYDROGUI_TwoImagesOp( aModule, HYDROGUI_TwoImagesOp::Fuse );
+  case ImportBathymetryId:
+    anOp = new HYDROGUI_ImportBathymetryOp( aModule );
     break;
-  case CutId:
-    anOp = new HYDROGUI_TwoImagesOp( aModule, HYDROGUI_TwoImagesOp::Cut );
+  case CreateZoneId:
+  case EditZoneId:
+    anOp = new HYDROGUI_ZoneOp( aModule, theId == EditZoneId );
     break;
-  case EditCompositeImageId:
-    anOp = new HYDROGUI_TwoImagesOp( aModule, HYDROGUI_TwoImagesOp::Edit );
+  case CreateCalculationId:
+  case EditCalculationId:
+    anOp = new HYDROGUI_CalculationOp( aModule, theId == EditCalculationId );
+    break;
+  case FuseImagesId:
+  case EditFusedImageId:
+    anOp = new HYDROGUI_TwoImagesOp( aModule, HYDROGUI_TwoImagesOp::Fuse, theId == EditFusedImageId );
+    break;
+  case CutImagesId:
+  case EditCutImageId:
+    anOp = new HYDROGUI_TwoImagesOp( aModule, HYDROGUI_TwoImagesOp::Cut, theId == EditCutImageId );
+    break;
+  case SplitImageId:
+  case EditSplittedImageId:
+    anOp = new HYDROGUI_TwoImagesOp( aModule, HYDROGUI_TwoImagesOp::Split, theId == EditSplittedImageId );
     break;
   case DeleteId:
     anOp = new HYDROGUI_DeleteOp( aModule );
@@ -261,3 +321,100 @@ LightApp_Operation* HYDROGUI_Module::createOperation( const int theId ) const
 
   return anOp;
 }
+
+//-----------------------------------------------------------------------------
+// Test splitting
+//-----------------------------------------------------------------------------
+#include <HYDROData_Iterator.h>
+#include <HYDROGUI_Tool.h>
+void HYDROGUI_Module::onTestSplit()
+{
+  Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( getStudyId() );
+
+  aDocument->StartOperation();
+
+  HYDROData_SequenceOfObjects aZoneList;
+  HYDROData_Iterator anIterator( aDocument, KIND_ZONE );
+  for( ; anIterator.More(); anIterator.Next() )
+    aZoneList.Append( anIterator.Current() );
+
+  HYDROGUI_SplitZonesTool::SplitDataList aSplitDataList =
+    HYDROGUI_SplitZonesTool::SplitZones( aZoneList, NULL );
+
+  int anIndex = 0;
+  HYDROGUI_SplitZonesTool::SplitDataListIterator anIter( aSplitDataList );
+  while( anIter.hasNext() )
+  {
+    const HYDROGUI_SplitZonesTool::SplitData& aSplitData = anIter.next();
+    const QPainterPath& aPath = aSplitData.Path;
+    const QStringList& aZoneNames = aSplitData.ZoneNames;
+
+    Handle(HYDROData_Polyline) aPolyline =
+      Handle(HYDROData_Polyline)::DownCast( aDocument->CreateObject( KIND_POLYLINE ) );
+    if( !aPolyline.IsNull() )
+    {
+      aPolyline->SetName( QString( "SplitPolyline_%1" ).arg( ++anIndex ) );
+      aPolyline->setDimension( 2 );
+
+      QList<PolylineSection> aPolylineData;
+      for( int i = 0, n = aPath.elementCount(); i < n; i++ )
+      {
+        const QPainterPath::Element anElement = aPath.elementAt( i );
+        switch( anElement.type )
+        {
+          case QPainterPath::MoveToElement:
+            aPolylineData.append( PolylineSection() );
+            break;
+          case QPainterPath::LineToElement:
+            if( !aPolylineData.isEmpty() )
+            {
+              PolylineSection& aSection = aPolylineData.last();
+              aSection.myCoords << anElement.x;
+              aSection.myCoords << anElement.y;
+            }
+            break;
+          case QPainterPath::CurveToElement: // currently not supported
+          default:
+            break;
+        }
+      }
+      aPolyline->setPolylineData( aPolylineData );
+    }
+
+    Handle(HYDROData_Zone) aZone =
+      Handle(HYDROData_Zone)::DownCast( aDocument->CreateObject( KIND_ZONE ) );
+    if( !aZone.IsNull() )
+    {
+      aZone->SetName( QString( "SplitZone_%1" ).arg( anIndex ) );
+      aZone->SetPolyline( aPolyline );
+
+      aZone->SetBorderColor( Qt::black );
+
+      int aCounter = 0;
+      int aR = 0, aG = 0, aB = 0;
+      QStringListIterator aZoneNameIter( aZoneNames );
+      while( aZoneNameIter.hasNext() )
+      {
+        const QString& aZoneName = aZoneNameIter.next();
+        Handle(HYDROData_Zone) aRefZone = Handle(HYDROData_Zone)::DownCast(
+          HYDROGUI_Tool::FindObjectByName( this, aZoneName, KIND_ZONE ) );
+        if( !aRefZone.IsNull() )
+        {
+          QColor aRefColor = aRefZone->GetFillingColor();
+          aR += aRefColor.red();
+          aG += aRefColor.green();
+          aB += aRefColor.blue();
+          aCounter++;
+        }
+      }
+      if( aCounter > 0 )
+      {
+        QColor aFillingColor( aR / aCounter, aG / aCounter, aB / aCounter );
+        aZone->SetFillingColor( aFillingColor );
+      }
+    }
+  }
+
+  aDocument->CommitOperation();
+  update( UF_Model );
+}