Salome HOME
Fix for the bug #255: VTK viewer is not updated after modification of objects.
[modules/hydro.git] / src / HYDROGUI / HYDROGUI_Operation.cxx
index 293364126d95066a638661e2324a424a52efac23..bdbd6fa05ffd1c86b4ea310a681dbf3f63b3ac1a 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 ),
+  myIsPrintErrorMessage( true )
 {
+  connect( this, SIGNAL( helpContextModule( const QString&, const QString&,
+                                            const QString& ) ),
+           theModule->application(), SLOT( onHelpContextModule( const QString&,
+                                            const QString&, const QString& ) ) );
 }
 
 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() ) );
+    connect( myPanel, SIGNAL( panelHelp() ), this, SLOT( onHelp() ) );
+  }
+  return myPanel;
+}
+
 SUIT_SelectionMgr* HYDROGUI_Operation::selectionMgr() const
 {
   return myModule->getApp()->selectionMgr();
@@ -56,71 +90,180 @@ 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;
+}
+
+bool HYDROGUI_Operation::processApply( int& theUpdateFlags,
+                                       QString& theErrorMsg )
+{
+  return false;
+}
+
+void HYDROGUI_Operation::processCancel()
+{
+}
+
+void HYDROGUI_Operation::startDocOperation()
+{
+  // Open transaction in the model document
+  if ( !doc()->IsOperation() )
+    doc()->StartOperation();
+}
+
+void HYDROGUI_Operation::abortDocOperation()
+{
+  // Abort transaction in the model document
+  if ( doc()->IsOperation() )
+    doc()->AbortOperation();
+}
+
+void HYDROGUI_Operation::commitDocOperation()
+{
+  // Commit transaction in the model document
+  if ( doc()->IsOperation() )
+    doc()->CommitOperation( HYDROGUI_Tool::ToExtString( getName() ) );
 }
 
 Handle_HYDROData_Document HYDROGUI_Operation::doc() const
 {
-  int aStudyId = myModule->application()->activeStudy()->id();
-  return HYDROData_Document::Document( aStudyId );
+  return HYDROData_Document::Document( myModule->getStudyId() );
 }
 
-void HYDROGUI_Operation::OnApply()
+void HYDROGUI_Operation::onApply()
 {
-  doc()->CommitOperation();
-  inputPanel()->hide();
+  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 ( ... )
+  {
+    aResult = false;
+  }
+  
+  QApplication::restoreOverrideCursor();
+
+  if ( aResult )
+  {
+    module()->update( anUpdateFlags );
+    commitDocOperation();
+    commit();
+  }
+  else
+  {
+    abortDocOperation();
+    printErrorMessage( anErrorMsg );
+    // If the operation has no input panel - do abort
+    if ( !inputPanel() ) {
+      abort();
+    }
+  }
 }
 
-void HYDROGUI_Operation::OnCancel()
+void HYDROGUI_Operation::setPrintErrorMessage( const bool theIsPrint )
 {
-  doc()->AbortOperation();
-  inputPanel()->hide();
+  myIsPrintErrorMessage = theIsPrint;
 }
 
-Handle_HYDROData_Object HYDROGUI_Operation::FindObjectByName( const QString& theName, int theKind ) const
+void HYDROGUI_Operation::printErrorMessage( const QString& theErrorMsg )
 {
-  HYDROData_Iterator anIt( doc(), theKind );
-  for( ; anIt.More(); anIt.Next() )
+  if ( myIsPrintErrorMessage )
   {
-    if( anIt.Current()->GetName() == theName )
-      return anIt.Current();
+    QString aMsg = tr( "INPUT_VALID_DATA" );
+    if( !theErrorMsg.isEmpty() )
+      aMsg.prepend( theErrorMsg + "\n" );
+    SUIT_MessageBox::critical( module()->getApp()->desktop(),
+                               tr( "INSUFFICIENT_INPUT_DATA" ),
+                               aMsg );
+
   }
-  return Handle_HYDROData_Object();
+  
+  myIsPrintErrorMessage = true;
 }
+
+void HYDROGUI_Operation::onCancel()
+{
+  processCancel();
+  abort();
+}
+
+void HYDROGUI_Operation::onHelp()
+{
+   emit helpContextModule( getHelpComponent(), getHelpFile(), getHelpContext() );
+}
+
+QString HYDROGUI_Operation::getHelpComponent() const
+{
+   return module()->moduleName();
+}
+
+QString HYDROGUI_Operation::getHelpFile() const
+{
+   QString aFileName = ((myName.isEmpty())? operationName() : myName);
+   return aFileName.replace(QRegExp("\\s"), "_").append(".html");
+}
+
+QString HYDROGUI_Operation::getHelpContext() const
+{
+   return QString();
+}
+
+