Salome HOME
Image positioning by two points.
[modules/hydro.git] / src / HYDROGUI / HYDROGUI_Operation.cxx
index 293364126d95066a638661e2324a424a52efac23..52fbcd1739a140a21570c64536dbfa19d6a2e97e 100644 (file)
 //
 
 #include "HYDROGUI_Operation.h"
-#include "HYDROGUI_Module.h"
+
 #include "HYDROGUI_InputPanel.h"
+#include "HYDROGUI_Module.h"
+#include "HYDROGUI_Tool.h"
 
 #include <HYDROData_Document.h>
 #include <HYDROData_Iterator.h>
 #include <LightApp_SelectionMgr.h>
 
 #include <SUIT_Desktop.h>
+#include <SUIT_MessageBox.h>
 #include <SUIT_Study.h>
 
+#include <QApplication>
+
 HYDROGUI_Operation::HYDROGUI_Operation( HYDROGUI_Module* theModule )
-: LightApp_Operation(), myModule( theModule ), myPanel( 0 )
+: LightApp_Operation(),
+  myModule( theModule ),
+  myPanel( 0 )
 {
 }
 
@@ -42,6 +49,27 @@ HYDROGUI_Operation::~HYDROGUI_Operation()
 {
 }
 
+void HYDROGUI_Operation::setName( const QString& theName )
+{
+  myName = theName;
+}
+
+const QString& HYDROGUI_Operation::getName() const
+{
+  return myName;
+}
+
+HYDROGUI_InputPanel* HYDROGUI_Operation::inputPanel() const
+{
+  if( !myPanel )
+  {
+    ( ( HYDROGUI_Operation* )this )->myPanel = createInputPanel();
+    connect( myPanel, SIGNAL( panelApply() ),  this, SLOT( onApply() ) );
+    connect( myPanel, SIGNAL( panelCancel() ), this, SLOT( onCancel() ) );
+  }
+  return myPanel;
+}
+
 SUIT_SelectionMgr* HYDROGUI_Operation::selectionMgr() const
 {
   return myModule->getApp()->selectionMgr();
@@ -56,71 +84,133 @@ void HYDROGUI_Operation::startOperation()
 {
   LightApp_Operation::startOperation();
 
-  doc()->StartOperation();
-
   if( inputPanel() )
   {
-    inputPanel()->show();
     myModule->getApp()->desktop()->addDockWidget( Qt::RightDockWidgetArea, inputPanel() );
+    inputPanel()->show();
   }
 }
 
 void HYDROGUI_Operation::abortOperation()
 {
-  doc()->AbortOperation();
-
   LightApp_Operation::abortOperation();
-
-  if( inputPanel() )
-    inputPanel()->hide();
+  closeInputPanel();
 }
 
 void HYDROGUI_Operation::commitOperation()
 {
-  doc()->CommitOperation();
-
   LightApp_Operation::commitOperation();
+  closeInputPanel();
+}
 
-  if( inputPanel() )
-    inputPanel()->hide();
+void HYDROGUI_Operation::setDialogActive( const bool active )
+{
+  LightApp_Operation::setDialogActive( active );
+  if( myPanel )
+  {
+    if( active )
+    {
+      myPanel->show();
+    }
+  }
 }
 
-HYDROGUI_InputPanel* HYDROGUI_Operation::inputPanel() const
+HYDROGUI_InputPanel* HYDROGUI_Operation::createInputPanel() const
 {
-  if( !myPanel )
+  return NULL;
+}
+
+void HYDROGUI_Operation::closeInputPanel()
+{
+  if( myPanel )
   {
-    ( ( HYDROGUI_Operation* )this )->myPanel = createInputPanel();
-    connect( myPanel, SIGNAL( panelApply() ),  this, SLOT( OnApply() ) );
-    connect( myPanel, SIGNAL( panelCancel() ), this, SLOT( OnCancel() ) );
+    myModule->getApp()->desktop()->removeDockWidget( myPanel );
+    delete myPanel;
+    myPanel = 0;
   }
-  return myPanel;
 }
 
-Handle_HYDROData_Document HYDROGUI_Operation::doc() const
+bool HYDROGUI_Operation::processApply( int& theUpdateFlags,
+                                       QString& theErrorMsg )
 {
-  int aStudyId = myModule->application()->activeStudy()->id();
-  return HYDROData_Document::Document( aStudyId );
+  return false;
 }
 
-void HYDROGUI_Operation::OnApply()
+void HYDROGUI_Operation::processCancel()
 {
-  doc()->CommitOperation();
-  inputPanel()->hide();
 }
 
-void HYDROGUI_Operation::OnCancel()
+void HYDROGUI_Operation::startDocOperation()
 {
+  // Open transaction in the model document
+  doc()->StartOperation();
+}
+
+void HYDROGUI_Operation::abortDocOperation()
+{
+  // Abort transaction in the model document
   doc()->AbortOperation();
-  inputPanel()->hide();
 }
 
-Handle_HYDROData_Object HYDROGUI_Operation::FindObjectByName( const QString& theName, int theKind ) const
+void HYDROGUI_Operation::commitDocOperation()
+{
+  // Commit transaction in the model document
+  doc()->CommitOperation( HYDROGUI_Tool::ToExtString( getName() ) );
+}
+
+Handle_HYDROData_Document HYDROGUI_Operation::doc() const
 {
-  HYDROData_Iterator anIt( doc(), theKind );
-  for( ; anIt.More(); anIt.Next() )
+  return HYDROData_Document::Document( myModule->getStudyId() );
+}
+
+void HYDROGUI_Operation::onApply()
+{
+  QApplication::setOverrideCursor( Qt::WaitCursor );
+
+  startDocOperation();
+
+  int anUpdateFlags = 0;
+  QString anErrorMsg;
+
+  bool aResult = false;
+  
+  try
+  {
+    aResult = processApply( anUpdateFlags, anErrorMsg );
+  }
+  catch ( Standard_Failure )
+  {
+    Handle(Standard_Failure) aFailure = Standard_Failure::Caught();
+    anErrorMsg = aFailure->GetMessageString();
+    aResult = false;
+  }
+  catch ( ... )
   {
-    if( anIt.Current()->GetName() == theName )
-      return anIt.Current();
+    aResult = false;
   }
-  return Handle_HYDROData_Object();
+  
+  QApplication::restoreOverrideCursor();
+
+  if ( aResult )
+  {
+    module()->update( anUpdateFlags );
+    commitDocOperation();
+    commit();
+  }
+  else
+  {
+    abortDocOperation();
+    QString aMsg = tr( "INPUT_VALID_DATA" );
+    if( !anErrorMsg.isEmpty() )
+      aMsg.prepend( anErrorMsg + "\n" );
+    SUIT_MessageBox::critical( module()->getApp()->desktop(),
+                               tr( "INSUFFICIENT_INPUT_DATA" ),
+                               aMsg ); 
+  }
+}
+
+void HYDROGUI_Operation::onCancel()
+{
+  processCancel();
+  abort();
 }