Modify creation of curves: 1) using QDockWidget instead of QDialog; 2) selection...

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

diff --git a/src/HYDROGUI/HYDROGUI_Operations.cxx b/src/HYDROGUI/HYDROGUI_Operations.cxx
index 87428aa30f7ff69a7c24ac9bc3abb7b97376f648..72c51c9a2d295398b918fcdf6d1bf11aa49e494d 100644 (file)
--- a/src/HYDROGUI/HYDROGUI_Operations.cxx
+++ b/src/HYDROGUI/HYDROGUI_Operations.cxx
@@ -33,6 +33,7 @@
 #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"
@@ -136,6 +137,9 @@ void HYDROGUI_Module::createMenus()
   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()
@@ -169,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()
@@ -313,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 );
+}