refs #430: incorrect coordinates in dump polyline

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

diff --git a/src/HYDROGUI/HYDROGUI_ZLevelsOp.cxx b/src/HYDROGUI/HYDROGUI_ZLevelsOp.cxx
index 7d12d15a7a0885f8e3ad4d4df800eb7ec1cadb2a..3f7c1eaf732e55ca72ae2f5dbaee49911b2edf11 100644 (file)
--- a/src/HYDROGUI/HYDROGUI_ZLevelsOp.cxx
+++ b/src/HYDROGUI/HYDROGUI_ZLevelsOp.cxx
@@ -27,6 +27,7 @@
 #include "HYDROGUI_Module.h"
 #include "HYDROGUI_Tool.h"
 #include "HYDROGUI_UpdateFlags.h"
+#include "HYDROGUI_ListSelector.h"
 
 #include <HYDROData_Entity.h>
 
@@ -35,28 +36,103 @@
 
 #include <SUIT_Desktop.h>
 
+/**
+  Constructor.
+  @param theModule the module
+*/
 HYDROGUI_ZLevelsOp::HYDROGUI_ZLevelsOp( HYDROGUI_Module* theModule )
 : HYDROGUI_Operation( theModule ),
-  myZLevelsDlg( NULL )
+  myDlg( NULL )
 {
   setName( tr( "SET_Z_LEVELS" ) );
 }
 
+/**
+  Destructor.
+*/
 HYDROGUI_ZLevelsOp::~HYDROGUI_ZLevelsOp()
 {
 }
 
+/**
+*/
 void HYDROGUI_ZLevelsOp::startOperation()
 {
   HYDROGUI_Operation::startOperation();
 
-  // TODO
-  abort();
+  // Prepare the list of objects
+  HYDROGUI_ListModel::Object2VisibleList anObject2VisibleList;
+
+  // get the document
+  Handle(HYDROData_Document) aDoc = doc();
+  if( !aDoc.IsNull() ) {
+    // get active OCC view id
+    size_t anActiveOCCViewId = HYDROGUI_Tool::GetActiveOCCViewId( module() );
+
+    // get objects list
+    HYDROData_SequenceOfObjects aSeqOfObjects = aDoc->GetObjectsLayerOrder( Standard_True );
+    HYDROData_SequenceOfObjects::Iterator anIter( aSeqOfObjects );
+    for ( ; anIter.More(); anIter.Next() ) {
+      Handle(HYDROData_Entity) anObject = anIter.Value();
+      if ( !anObject.IsNull() ) {
+        bool isVisible = module()->isObjectVisible( anActiveOCCViewId, anObject );
+        anObject2VisibleList << HYDROGUI_ListModel::Object2Visible( anObject, isVisible );
+      }
+    }
+  }
+
+  // Show the dialog
+  if ( !myDlg ) {
+    myDlg = new HYDROGUI_ZLevelsDlg( module()->getApp()->desktop(), module() );
+    connect( myDlg, SIGNAL( applyOrder() ), this, SLOT( onApply() ) );
+    connect( myDlg, SIGNAL( rejected() ), this, SLOT( onCancel() ) );
+  }
+  myDlg->setObjects( anObject2VisibleList );
+
+  myDlg->exec();
 }
 
+/**
+*/
 bool HYDROGUI_ZLevelsOp::processApply( int& theUpdateFlags,
-                                       QString& theErrorMsg )
+                                       QString& theErrorMsg,
+                                       QStringList& theBrowseObjectsEntries )
 {
-  // TODO
-  return false;
+  bool aRes = false;
+
+  if ( myDlg ) {
+    Handle(HYDROData_Document) aDoc = doc();
+    if( !aDoc.IsNull() ) {
+      HYDROGUI_ListModel::ObjectList anObjects = myDlg->getObjects();
+      HYDROData_SequenceOfObjects anOrderedObjects;
+      foreach ( const Handle(HYDROData_Entity) anObject, anObjects ) {
+        anOrderedObjects.Append( anObject );
+      }
+
+      aDoc->SetObjectsLayerOrder( anOrderedObjects );
+
+      theUpdateFlags = UF_Model | UF_OCCViewer;
+      aRes = true;
+    }
+  }
+
+  return aRes;
+}
+
+/**
+*/
+bool HYDROGUI_ZLevelsOp::isGranted() const
+{
+  return true;
 }
+
+/**
+*/
+void HYDROGUI_ZLevelsOp::processCancel()
+{
+  // Delete the dialog
+  if ( myDlg ) {
+    delete myDlg;
+    myDlg = 0;
+  }
+}
\ No newline at end of file