src/XGUI/XGUI_Workshop.cpp

   1 #include "ModuleBase_IModule.h"
   2 #include "XGUI_Constants.h"
   3 #include "XGUI_Command.h"
   4 #include "XGUI_MainMenu.h"
   5 #include "XGUI_MainWindow.h"
   6 #include "XGUI_MenuGroupPanel.h"
   7 #include "XGUI_Tools.h"
   8 #include "XGUI_Workbench.h"
   9 #include "XGUI_Workshop.h"
  10 #include "XGUI_Viewer.h"
  11 #include "ModuleBase_WidgetFactory.h"
  12 #include "XGUI_SelectionMgr.h"
  13 #include "XGUI_Selection.h"
  14 #include "XGUI_ObjectsBrowser.h"
  15 #include "XGUI_Displayer.h"
  16 #include "XGUI_OperationMgr.h"
  17 #include "XGUI_SalomeConnector.h"
  18 #include "XGUI_SalomeViewer.h"
  19 #include "XGUI_ActionsMgr.h"
  20 #include "XGUI_ErrorDialog.h"
  21 #include "XGUI_ViewerProxy.h"
  22 #include "XGUI_PropertyPanel.h"
  23 #include "XGUI_ContextMenuMgr.h"
  24 #include "XGUI_ModuleConnector.h"
  25
  26 #include <ModelAPI_Events.h>
  27 #include <ModelAPI_PluginManager.h>
  28 #include <ModelAPI_Feature.h>
  29 #include <ModelAPI_Data.h>
  30 #include <ModelAPI_AttributeDocRef.h>
  31 #include <ModelAPI_Object.h>
  32 #include <ModelAPI_Validator.h>
  33 #include <ModelAPI_ResultPart.h>
  34
  35 #include <PartSetPlugin_Part.h>
  36
  37 #include <Events_Loop.h>
  38 #include <Events_Error.h>
  39
  40 #include <ModuleBase_Operation.h>
  41 #include <ModuleBase_Operation.h>
  42 #include <ModuleBase_OperationDescription.h>
  43 #include <ModuleBase_SelectionValidator.h>
  44
  45 #include <Config_Common.h>
  46 #include <Config_FeatureMessage.h>
  47 #include <Config_PointerMessage.h>
  48 #include <Config_ModuleReader.h>
  49
  50 #include <QApplication>
  51 #include <QFileDialog>
  52 #include <QMessageBox>
  53 #include <QMdiSubWindow>
  54 #include <QPushButton>
  55 #include <QDockWidget>
  56 #include <QLayout>
  57 #include <QTimer>
  58
  59 #ifdef _DEBUG
  60 #include <QDebug>
  61 #endif
  62
  63 #ifdef WIN32
  64 #include <windows.h>
  65 #else
  66 #include <dlfcn.h>
  67 #endif
  68
  69
  70 QMap<QString, QString> XGUI_Workshop::myIcons;
  71
  72 QString XGUI_Workshop::featureIcon(const std::string& theId)
  73 {
  74   QString aId(theId.c_str());
  75   if (myIcons.contains(aId))
  76     return myIcons[aId];
  77   return QString();
  78 }
  79
  80 XGUI_Workshop::XGUI_Workshop(XGUI_SalomeConnector* theConnector)
  81   : QObject(),
  82   myCurrentDir(QString()),
  83   myModule(NULL),
  84   mySalomeConnector(theConnector),
  85   myPropertyPanel(0),
  86   myObjectBrowser(0),
  87   myDisplayer(0)
  88 {
  89   myMainWindow = mySalomeConnector? 0 : new XGUI_MainWindow();
  90
  91   myDisplayer = new XGUI_Displayer(this);
  92
  93   mySelector = new XGUI_SelectionMgr(this);
  94   //connect(mySelector, SIGNAL(selectionChanged()), this, SLOT(updateModuleCommands()));
  95
  96   myOperationMgr = new XGUI_OperationMgr(this);
  97   myActionsMgr = new XGUI_ActionsMgr(this);
  98   myErrorDlg = new XGUI_ErrorDialog(myMainWindow);
  99   myContextMenuMgr = new XGUI_ContextMenuMgr(this);
 100   connect(myContextMenuMgr, SIGNAL(actionTriggered(const QString&, bool)),
 101           this, SLOT(onContextMenuCommand(const QString&, bool)));
 102
 103   myViewerProxy = new XGUI_ViewerProxy(this);
 104   connect(myViewerProxy, SIGNAL(selectionChanged()), this, SLOT(updateCommandsOnViewSelection()));
 105
 106   myModuleConnector = new XGUI_ModuleConnector(this);
 107
 108   connect(myOperationMgr, SIGNAL(operationStarted()), SLOT(onOperationStarted()));
 109   connect(myOperationMgr, SIGNAL(operationResumed()), SLOT(onOperationStarted()));
 110   connect(myOperationMgr, SIGNAL(operationStopped(ModuleBase_Operation*)), SLOT(onOperationStopped(ModuleBase_Operation*)));
 111   connect(myMainWindow, SIGNAL(exitKeySequence()), SLOT(onExit()));
 112   connect(myOperationMgr, SIGNAL(operationStarted()), myActionsMgr, SLOT(update()));
 113   connect(myOperationMgr, SIGNAL(operationStopped(ModuleBase_Operation*)), myActionsMgr, SLOT(update()));
 114   connect(this, SIGNAL(errorOccurred(const QString&)), myErrorDlg, SLOT(addError(const QString&)));
 115 }
 116
 117 //******************************************************
 118 XGUI_Workshop::~XGUI_Workshop(void)
 119 {
 120 }
 121
 122 //******************************************************
 123 void XGUI_Workshop::startApplication()
 124 {
 125   initMenu();
 126   //Initialize event listening
 127   Events_Loop* aLoop = Events_Loop::loop();
 128   aLoop->registerListener(this, Events_Error::errorID()); //!< Listening application errors.
 129   //TODO(sbh): Implement static method to extract event id [SEID]
 130   aLoop->registerListener(this, Events_Loop::eventByName(EVENT_FEATURE_LOADED));
 131   // TODO Is it good to use non standard event within workshop?
 132   aLoop->registerListener(this, Events_Loop::eventByName(EVENT_OPERATION_LAUNCHED));
 133   aLoop->registerListener(this, Events_Loop::eventByName(EVENT_OBJECT_UPDATED));
 134   aLoop->registerListener(this, Events_Loop::eventByName(EVENT_OBJECT_CREATED));
 135   aLoop->registerListener(this, Events_Loop::eventByName(EVENT_OBJECT_TO_REDISPLAY));
 136   aLoop->registerListener(this, Events_Loop::eventByName(EVENT_OBJECT_DELETED));
 137
 138   activateModule();
 139   if (myMainWindow) {
 140     myMainWindow->show();
 141     updateCommandStatus();
 142   }
 143   onNew();
 144 }
 145
 146 //******************************************************
 147 void XGUI_Workshop::initMenu()
 148 {
 149   myContextMenuMgr->createActions();
 150
 151   if (isSalomeMode()) {
 152     // Create only Undo, Redo commands
 153     QAction* aAction = salomeConnector()->addEditCommand("UNDO_CMD",
 154                                       tr("Undo"), tr("Undo last command"),
 155                                       QIcon(":pictures/undo.png"),
 156                                       QKeySequence::Undo, false);
 157     connect(aAction, SIGNAL(triggered(bool)), this, SLOT(onUndo()));
 158     aAction = salomeConnector()->addEditCommand("REDO_CMD",
 159                                       tr("Redo"), tr("Redo last command"),
 160                                       QIcon(":pictures/redo.png"),
 161                                       QKeySequence::Redo, false);
 162     connect(aAction, SIGNAL(triggered(bool)), this, SLOT(onRedo()));
 163     salomeConnector()->addEditMenuSeparator();
 164     return;
 165   }
 166   XGUI_Workbench* aPage = myMainWindow->menuObject()->generalPage();
 167
 168   // File commands group
 169   XGUI_MenuGroupPanel* aGroup = aPage->addGroup("Default");
 170
 171   XGUI_Command* aCommand;
 172
 173   aCommand = aGroup->addFeature("SAVE_CMD", tr("Save..."), tr("Save the document"),
 174                                 QIcon(":pictures/save.png"), QKeySequence::Save);
 175   aCommand->connectTo(this, SLOT(onSave()));
 176   //aCommand->disable();
 177
 178   aCommand = aGroup->addFeature("UNDO_CMD", tr("Undo"), tr("Undo last command"),
 179                                 QIcon(":pictures/undo.png"), QKeySequence::Undo);
 180   aCommand->connectTo(this, SLOT(onUndo()));
 181
 182   aCommand = aGroup->addFeature("REDO_CMD", tr("Redo"), tr("Redo last command"),
 183                                 QIcon(":pictures/redo.png"), QKeySequence::Redo);
 184   aCommand->connectTo(this, SLOT(onRedo()));
 185
 186   aCommand = aGroup->addFeature("REBUILD_CMD", tr("Rebuild"), tr("Rebuild data objects"),
 187                                 QIcon(":pictures/rebuild.png"));
 188
 189   aCommand = aGroup->addFeature("SAVEAS_CMD", tr("Save as..."), tr("Save the document into a file"),
 190                                 QIcon(":pictures/save.png"));
 191   aCommand->connectTo(this, SLOT(onSaveAs()));
 192   //aCommand->disable();
 193
 194   aCommand = aGroup->addFeature("OPEN_CMD", tr("Open..."), tr("Open a new document"),
 195                                 QIcon(":pictures/open.png"), QKeySequence::Open);
 196   aCommand->connectTo(this, SLOT(onOpen()));
 197
 198   //aCommand = aGroup->addFeature("NEW_CMD", tr("New"), tr("Create a new document"),
 199   //                              QIcon(":pictures/new.png"), QKeySequence::New);
 200   //aCommand->connectTo(this, SLOT(onNew()));
 201
 202   aCommand = aGroup->addFeature("EXIT_CMD", tr("Exit"), tr("Exit application"),
 203                                 QIcon(":pictures/close.png"), QKeySequence::Close);
 204   aCommand->connectTo(this, SLOT(onExit()));
 205   //FIXME: SBH's test action. Can be used for some GUI tests.
 206   //#ifdef _DEBUG
 207   //  aCommand = aGroup->addFeature("TEST_CMD", "Test!", "Private debug button",
 208   //                                QIcon(":pictures/close.png"));
 209   //  aCommand->connectTo(myActionsMgr, SLOT(update()));
 210   //#endif
 211 }
 212
 213 //******************************************************
 214 XGUI_Workbench* XGUI_Workshop::addWorkbench(const QString& theName)
 215 {
 216   XGUI_MainMenu* aMenuBar = myMainWindow->menuObject();
 217   return aMenuBar->addWorkbench(theName);
 218 }
 219
 220 //******************************************************
 221 void XGUI_Workshop::processEvent(const Events_Message* theMessage)
 222 {
 223   //A message to start feature creation received.
 224   static Events_ID aFeatureLoadedId = Events_Loop::loop()->eventByName(EVENT_FEATURE_LOADED);
 225   if (theMessage->eventID() == aFeatureLoadedId) {
 226     const Config_FeatureMessage* aFeatureMsg = dynamic_cast<const Config_FeatureMessage*>(theMessage);
 227     if(!aFeatureMsg->isInternal()) {
 228       addFeature(aFeatureMsg);
 229     }
 230     return;
 231   }
 232
 233   // Process creation of Part
 234   if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_OBJECT_CREATED)) {
 235     const ModelAPI_ObjectUpdatedMessage* aUpdMsg = dynamic_cast<const ModelAPI_ObjectUpdatedMessage*>(theMessage);
 236     onFeatureCreatedMsg(aUpdMsg);
 237     return;
 238   }
 239
 240   // Redisplay feature
 241   if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_OBJECT_TO_REDISPLAY)) {
 242     const ModelAPI_ObjectUpdatedMessage* aUpdMsg = dynamic_cast<const ModelAPI_ObjectUpdatedMessage*>(theMessage);
 243     onFeatureRedisplayMsg(aUpdMsg);
 244     return;
 245   }
 246
 247   //Update property panel on corresponding message. If there is no current operation (no
 248   //property panel), or received message has different feature to the current - do nothing.
 249   if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_OBJECT_UPDATED)) {
 250     const ModelAPI_ObjectUpdatedMessage* anUpdateMsg =
 251         dynamic_cast<const ModelAPI_ObjectUpdatedMessage*>(theMessage);
 252     onFeatureUpdatedMsg(anUpdateMsg);
 253     return;
 254   }
 255
 256   if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_OBJECT_DELETED)) {
 257     const ModelAPI_ObjectDeletedMessage* aDelMsg =
 258         dynamic_cast<const ModelAPI_ObjectDeletedMessage*>(theMessage);
 259     onObjectDeletedMsg(aDelMsg);
 260     return;
 261   }
 262
 263   //An operation passed by message. Start it, process and commit.
 264   if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_OPERATION_LAUNCHED)) {
 265     const Config_PointerMessage* aPartSetMsg = dynamic_cast<const Config_PointerMessage*>(theMessage);
 266     //myPropertyPanel->cleanContent();
 267     ModuleBase_Operation* anOperation = (ModuleBase_Operation*)aPartSetMsg->pointer();
 268
 269     if (myOperationMgr->startOperation(anOperation)) {
 270       myPropertyPanel->updateContentWidget(anOperation->feature());
 271       if (!anOperation->getDescription()->hasXmlRepresentation()) {
 272         anOperation->commit();
 273         updateCommandStatus();
 274       }
 275     }
 276     return;
 277   }
 278   //Show error dialog if error message received.
 279   const Events_Error* anAppError = dynamic_cast<const Events_Error*>(theMessage);
 280   if (anAppError) {
 281     emit errorOccurred(QString::fromLatin1(anAppError->description()));
 282   }
 283 }
 284
 285 //******************************************************
 286 void XGUI_Workshop::onFeatureUpdatedMsg(const ModelAPI_ObjectUpdatedMessage* theMsg)
 287 {
 288   std::set<ObjectPtr> aFeatures = theMsg->objects();
 289   if (myOperationMgr->hasOperation())
 290   {
 291     FeaturePtr aCurrentFeature = myOperationMgr->currentOperation()->feature();
 292     std::set<ObjectPtr>::const_iterator aIt;
 293     for (aIt = aFeatures.begin(); aIt != aFeatures.end(); ++aIt) {
 294       ObjectPtr aNewFeature = (*aIt);
 295       if(aNewFeature == aCurrentFeature) {
 296         myPropertyPanel->updateContentWidget(aCurrentFeature);
 297         break;
 298       }
 299     }
 300   }
 301 }
 302
 303 //******************************************************
 304 void XGUI_Workshop::onFeatureRedisplayMsg(const ModelAPI_ObjectUpdatedMessage* theMsg)
 305 {
 306   std::set<ObjectPtr> aObjects = theMsg->objects();
 307   std::set<ObjectPtr>::const_iterator aIt;
 308   for (aIt = aObjects.begin(); aIt != aObjects.end(); ++aIt) {
 309     ObjectPtr aObj = (*aIt);
 310     if (!aObj->data() )
 311       myDisplayer->erase(aObj, false);
 312     else {
 313       if (myDisplayer->isVisible(aObj)) // TODO VSV: Correction sketch drawing
 314         myDisplayer->display(aObj, false); // In order to update presentation
 315     }
 316   }
 317   myDisplayer->updateViewer();
 318 }
 319
 320 //******************************************************
 321 void XGUI_Workshop::onFeatureCreatedMsg(const ModelAPI_ObjectUpdatedMessage* theMsg)
 322 {
 323   std::set<ObjectPtr> aObjects = theMsg->objects();
 324
 325   std::set<ObjectPtr>::const_iterator aIt;
 326   bool aHasPart = false;
 327   bool isDisplayed = false;
 328   for (aIt = aObjects.begin(); aIt != aObjects.end(); ++aIt) {
 329     ResultPartPtr aPart = boost::dynamic_pointer_cast<ModelAPI_ResultPart>(*aIt);
 330     if (aPart) {
 331       aHasPart = true;
 332     // If a feature is created from the aplication's python console
 333     // it doesn't stored in the operation mgr and doesn't displayed
 334     } else if(myOperationMgr->hasOperation()) {
 335       ModuleBase_Operation* aOperation = myOperationMgr->currentOperation();
 336       if (aOperation->hasObject(*aIt)) { // Display only current operation results
 337         myDisplayer->display(*aIt, false);
 338         isDisplayed = true;
 339       }
 340     }
 341   }
 342   if (isDisplayed)
 343     myDisplayer->updateViewer();
 344   if (aHasPart) {
 345     //The created part will be created in Object Browser later and we have to activate it
 346     // only when it is created everywere
 347     QTimer::singleShot(50, this, SLOT(activateLastPart()));
 348   }
 349 }
 350
 351 //******************************************************
 352 void XGUI_Workshop::onObjectDeletedMsg(const ModelAPI_ObjectDeletedMessage* theMsg)
 353 {
 354   //std::set<ObjectPtr> aFeatures = theMsg->objects();
 355 }
 356
 357 //******************************************************
 358 void XGUI_Workshop::onOperationStarted()
 359 {
 360   ModuleBase_Operation* aOperation = myOperationMgr->currentOperation();
 361
 362   if(aOperation->getDescription()->hasXmlRepresentation()) { //!< No need for property panel
 363     connectWithOperation(aOperation);
 364
 365     showPropertyPanel();
 366     QString aXmlRepr = aOperation->getDescription()->xmlRepresentation();
 367     ModuleBase_WidgetFactory aFactory = ModuleBase_WidgetFactory(aXmlRepr.toStdString(), myModuleConnector);
 368
 369     myPropertyPanel->cleanContent();
 370     aFactory.createWidget(myPropertyPanel->contentWidget());
 371
 372     QList<ModuleBase_ModelWidget*> aWidgets = aFactory.getModelWidgets();
 373     QList<ModuleBase_ModelWidget*>::const_iterator anIt = aWidgets.begin(), aLast = aWidgets.end();
 374     ModuleBase_ModelWidget* aWidget;
 375     for (; anIt != aLast; anIt++) {
 376       aWidget = *anIt;
 377       //QObject::connect(aWidget, SIGNAL(valuesChanged()),  aOperation, SLOT(storeCustomValue()));
 378       QObject::connect(aWidget, SIGNAL(valuesChanged()),
 379                        this, SLOT(onWidgetValuesChanged()));
 380       // Init default values
 381       if (!aOperation->isEditOperation() && aWidget->hasDefaultValue()) {
 382         aWidget->storeValue(aOperation->feature());
 383       }
 384     }
 385
 386     myPropertyPanel->setModelWidgets(aWidgets);
 387     myPropertyPanel->setWindowTitle(aOperation->getDescription()->description());
 388   }
 389   updateCommandStatus();
 390 }
 391
 392 //******************************************************
 393 void XGUI_Workshop::onOperationStopped(ModuleBase_Operation* theOperation)
 394 {
 395   //!< No need for property panel
 396   updateCommandStatus();
 397   hidePropertyPanel();
 398   myPropertyPanel->cleanContent();
 399 }
 400
 401 /*
 402  *
 403  */
 404 void XGUI_Workshop::addFeature(const Config_FeatureMessage* theMessage)
 405 {
 406   if (!theMessage) {
 407 #ifdef _DEBUG
 408     qDebug() << "XGUI_Workshop::addFeature: NULL message.";
 409 #endif
 410     return;
 411   }
 412   // Remember features icons
 413   myIcons[QString::fromStdString(theMessage->id())] = QString::fromStdString(theMessage->icon());
 414
 415   //Find or create Workbench
 416   QString aWchName = QString::fromStdString(theMessage->workbenchId());
 417   QString aNestedFeatures = QString::fromStdString(theMessage->nestedFeatures());
 418   bool isUsePropPanel = theMessage->isUseInput();
 419   QString aId = QString::fromStdString(theMessage->id());
 420   if (isSalomeMode()) {
 421     QAction* aAction = salomeConnector()->addFeature(aWchName,
 422                               aId,
 423                               QString::fromStdString(theMessage->text()),
 424                               QString::fromStdString(theMessage->tooltip()),
 425                               QIcon(theMessage->icon().c_str()),
 426                               QKeySequence(), isUsePropPanel);
 427     salomeConnector()->setNestedActions(aId, aNestedFeatures.split(" "));
 428     myActionsMgr->addCommand(aAction);
 429     myModule->featureCreated(aAction);
 430   } else {
 431
 432     XGUI_MainMenu* aMenuBar = myMainWindow->menuObject();
 433     XGUI_Workbench* aPage = aMenuBar->findWorkbench(aWchName);
 434     if (!aPage) {
 435       aPage = addWorkbench(aWchName);
 436     }
 437     //Find or create Group
 438     QString aGroupName = QString::fromStdString(theMessage->groupId());
 439     XGUI_MenuGroupPanel* aGroup = aPage->findGroup(aGroupName);
 440     if (!aGroup) {
 441       aGroup = aPage->addGroup(aGroupName);
 442     }
 443     //Create feature...
 444     XGUI_Command* aCommand = aGroup->addFeature(aId,
 445                                                 QString::fromStdString(theMessage->text()),
 446                                                 QString::fromStdString(theMessage->tooltip()),
 447                                                 QIcon(theMessage->icon().c_str()),
 448                                                 QKeySequence(), isUsePropPanel);
 449     aCommand->setNestedCommands(aNestedFeatures.split(" ", QString::SkipEmptyParts));
 450     myActionsMgr->addCommand(aCommand);
 451     myModule->featureCreated(aCommand);
 452   }
 453 }
 454
 455 /*
 456  * Makes a signal/slot connections between Property Panel
 457  * and given operation. The given operation becomes a
 458  * current operation and previous operation if exists
 459  */
 460 void XGUI_Workshop::connectWithOperation(ModuleBase_Operation* theOperation)
 461 {
 462   QAction* aCommand = 0;
 463   if (isSalomeMode()) {
 464     aCommand = salomeConnector()->command(theOperation->getDescription()->operationId());
 465   } else {
 466     XGUI_MainMenu* aMenu = myMainWindow->menuObject();
 467     aCommand = aMenu->feature(theOperation->getDescription()->operationId());
 468   }
 469   //Abort operation on uncheck the command
 470   connect(aCommand, SIGNAL(triggered(bool)), theOperation, SLOT(setRunning(bool)));
 471 }
 472
 473 /*
 474  * Saves document with given name.
 475  */
 476 void XGUI_Workshop::saveDocument(QString theName)
 477 {
 478   QApplication::restoreOverrideCursor();
 479   PluginManagerPtr aMgr = ModelAPI_PluginManager::get();
 480   DocumentPtr aDoc = aMgr->rootDocument();
 481   aDoc->save(theName.toLatin1().constData());
 482   QApplication::restoreOverrideCursor();
 483 }
 484
 485 //******************************************************
 486 void XGUI_Workshop::onExit()
 487 {
 488   PluginManagerPtr aMgr = ModelAPI_PluginManager::get();
 489   DocumentPtr aDoc = aMgr->rootDocument();
 490   if(aDoc->isModified()) {
 491     int anAnswer = QMessageBox::question(
 492         myMainWindow, tr("Save current file"),
 493         tr("The document is modified, save before exit?"),
 494         QMessageBox::Save | QMessageBox::Discard | QMessageBox::Cancel, QMessageBox::Cancel);
 495     if(anAnswer == QMessageBox::Save) {
 496       bool saved = onSave();
 497       if(!saved) {
 498         return;
 499       }
 500     } else if (anAnswer == QMessageBox::Cancel) {
 501       return;
 502     }
 503   }
 504   qApp->exit();
 505 }
 506
 507 //******************************************************
 508 void XGUI_Workshop::onNew()
 509 {
 510   QApplication::setOverrideCursor(Qt::WaitCursor);
 511   if (objectBrowser() == 0) {
 512     createDockWidgets();
 513     mySelector->connectViewers();
 514   }
 515   myViewerProxy->connectToViewer();
 516   showObjectBrowser();
 517   if (!isSalomeMode()) {
 518     myMainWindow->showPythonConsole();
 519     QMdiSubWindow* aWnd = myMainWindow->viewer()->createView();
 520     aWnd->showMaximized();
 521     updateCommandStatus();
 522   }
 523   myContextMenuMgr->connectViewer();
 524   QApplication::restoreOverrideCursor();
 525 }
 526
 527 //******************************************************
 528 void XGUI_Workshop::onOpen()
 529 {
 530   //save current file before close if modified
 531   PluginManagerPtr aMgr = ModelAPI_PluginManager::get();
 532   DocumentPtr aDoc = aMgr->rootDocument();
 533   if(aDoc->isModified()) {
 534     //TODO(sbh): re-launch the app?
 535     int anAnswer = QMessageBox::question(
 536         myMainWindow, tr("Save current file"),
 537         tr("The document is modified, save before opening another?"),
 538         QMessageBox::Save | QMessageBox::Discard | QMessageBox::Cancel, QMessageBox::Cancel);
 539     if(anAnswer == QMessageBox::Save) {
 540       onSave();
 541     } else if (anAnswer == QMessageBox::Cancel) {
 542       return;
 543     }
 544     aDoc->close();
 545     myCurrentDir = "";
 546   }
 547
 548   //show file dialog, check if readable and open
 549   myCurrentDir = QFileDialog::getExistingDirectory(mainWindow());
 550   if(myCurrentDir.isEmpty())
 551     return;
 552   QFileInfo aFileInfo(myCurrentDir);
 553   if(!aFileInfo.exists() || !aFileInfo.isReadable()) {
 554     QMessageBox::critical(myMainWindow, tr("Warning"), tr("Unable to open the file."));
 555     myCurrentDir = "";
 556     return;
 557   }
 558   QApplication::setOverrideCursor(Qt::WaitCursor);
 559   aDoc->load(myCurrentDir.toLatin1().constData());
 560   updateCommandStatus();
 561   myObjectBrowser->rebuildDataTree();
 562   QApplication::restoreOverrideCursor();
 563 }
 564
 565 //******************************************************
 566 bool XGUI_Workshop::onSave()
 567 {
 568   if(myCurrentDir.isEmpty()) {
 569     return onSaveAs();
 570   }
 571   saveDocument(myCurrentDir);
 572   updateCommandStatus();
 573   return true;
 574 }
 575
 576 //******************************************************
 577 bool XGUI_Workshop::onSaveAs()
 578 {
 579   QFileDialog dialog(mainWindow());
 580   dialog.setWindowTitle(tr("Select directory to save files..."));
 581   dialog.setFileMode(QFileDialog::Directory);
 582   dialog.setFilter(tr("Folders (*)"));
 583   dialog.setOptions(QFileDialog::HideNameFilterDetails | QFileDialog::ShowDirsOnly);
 584   dialog.setViewMode(QFileDialog::Detail);
 585
 586   if(!dialog.exec()) {
 587     return false;
 588   }
 589   QString aTempDir = dialog.selectedFiles().first();
 590   QDir aDir(aTempDir);
 591   if(aDir.exists() && !aDir.entryInfoList(QDir::NoDotAndDotDot|QDir::AllEntries).isEmpty()) {
 592     int answer = QMessageBox::question(myMainWindow,
 593                                        //: Title of the dialog which asks user if he wants to save study in existing non-empty folder
 594                                        tr("Save"),
 595                                        tr("The folder already contains some files, save anyway?"),
 596                                        QMessageBox::Save|QMessageBox::Cancel);
 597     if(answer == QMessageBox::Cancel) {
 598       return false;
 599     }
 600   }
 601   myCurrentDir = aTempDir;
 602   return onSave();
 603 }
 604
 605 //******************************************************
 606 void XGUI_Workshop::onUndo()
 607 {
 608   objectBrowser()->treeView()->setCurrentIndex(QModelIndex());
 609   PluginManagerPtr aMgr = ModelAPI_PluginManager::get();
 610   DocumentPtr aDoc = aMgr->rootDocument();
 611   if (aDoc->isOperation())
 612     operationMgr()->abortOperation();
 613   aDoc->undo();
 614   updateCommandStatus();
 615 }
 616
 617 //******************************************************
 618 void XGUI_Workshop::onRedo()
 619 {
 620   objectBrowser()->treeView()->setCurrentIndex(QModelIndex());
 621   PluginManagerPtr aMgr = ModelAPI_PluginManager::get();
 622   DocumentPtr aDoc = aMgr->rootDocument();
 623   if (aDoc->isOperation())
 624     operationMgr()->abortOperation();
 625   aDoc->redo();
 626   updateCommandStatus();
 627 }
 628
 629 //******************************************************
 630 ModuleBase_IModule* XGUI_Workshop::loadModule(const QString& theModule)
 631 {
 632   QString libName =
 633       QString::fromStdString(library(theModule.toStdString()));
 634   if (libName.isEmpty()) {
 635     qWarning(
 636     qPrintable( tr( "Information about module \"%1\" doesn't exist." ).arg( theModule ) ));
 637     return 0;
 638   }
 639
 640   QString err;
 641   CREATE_FUNC crtInst = 0;
 642
 643 #ifdef WIN32
 644   HINSTANCE modLib = ::LoadLibrary((LPTSTR) qPrintable(libName));
 645   if (!modLib) {
 646     LPVOID lpMsgBuf;
 647     ::FormatMessage(
 648         FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS,
 649         0, ::GetLastError(), 0, (LPTSTR) & lpMsgBuf, 0, 0);
 650     QString aMsg((char*) &lpMsgBuf);
 651     err = QString("Failed to load  %1. %2").arg(libName).arg(aMsg);
 652     ::LocalFree(lpMsgBuf);
 653   } else {
 654     crtInst = (CREATE_FUNC) ::GetProcAddress(modLib, CREATE_MODULE);
 655     if (!crtInst) {
 656       LPVOID lpMsgBuf;
 657       ::FormatMessage(
 658           FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM
 659               | FORMAT_MESSAGE_IGNORE_INSERTS,
 660           0, ::GetLastError(), 0, (LPTSTR) & lpMsgBuf, 0, 0);
 661       QString aMsg((char*) &lpMsgBuf);
 662       err = QString("Failed to find  %1 function. %2").arg( CREATE_MODULE).arg(aMsg);
 663       ::LocalFree(lpMsgBuf);
 664     }
 665   }
 666 #else
 667   void* modLib = dlopen( libName.toLatin1(), RTLD_LAZY | RTLD_GLOBAL );
 668   if ( !modLib ) {
 669     err = QString( "Can not load library %1. %2" ).arg( libName ).arg( dlerror() );
 670   } else {
 671     crtInst = (CREATE_FUNC)dlsym( modLib, CREATE_MODULE );
 672     if ( !crtInst ) {
 673       err = QString( "Failed to find function %1. %2" ).arg( CREATE_MODULE ).arg( dlerror() );
 674     }
 675   }
 676 #endif
 677
 678   ModuleBase_IModule* aModule = crtInst ? crtInst(this) : 0;
 679
 680   if (!err.isEmpty()) {
 681     if (mainWindow()) {
 682       QMessageBox::warning(mainWindow(), tr("Error"), err);
 683     } else {
 684       qWarning( qPrintable( err ));
 685     }
 686   }
 687   return aModule;
 688 }
 689
 690 //******************************************************
 691 bool XGUI_Workshop::activateModule()
 692 {
 693   Config_ModuleReader aModuleReader;
 694   QString moduleName = QString::fromStdString(aModuleReader.getModuleName());
 695   myModule = loadModule(moduleName);
 696   if (!myModule)
 697     return false;
 698   myModule->createFeatures();
 699   myActionsMgr->update();
 700   return true;
 701 }
 702
 703 //******************************************************
 704 void XGUI_Workshop::updateCommandStatus()
 705 {
 706   QList<QAction*> aCommands;
 707   if (isSalomeMode()) { // update commands in SALOME mode
 708     aCommands = salomeConnector()->commandList();
 709   } else {
 710     XGUI_MainMenu* aMenuBar = myMainWindow->menuObject();
 711     foreach (XGUI_Command* aCmd, aMenuBar->features())
 712       aCommands.append(aCmd);
 713   }
 714   PluginManagerPtr aMgr = ModelAPI_PluginManager::get();
 715   if (aMgr->hasRootDocument()) {
 716     QAction* aUndoCmd;
 717     QAction* aRedoCmd;
 718     foreach(QAction* aCmd, aCommands) {
 719       QString aId = aCmd->data().toString();
 720       if (aId == "UNDO_CMD")
 721         aUndoCmd = aCmd;
 722       else if (aId == "REDO_CMD")
 723         aRedoCmd = aCmd;
 724       else // Enable all commands
 725         aCmd->setEnabled(true);
 726     }
 727     DocumentPtr aDoc = aMgr->rootDocument();
 728     aUndoCmd->setEnabled(aDoc->canUndo());
 729     aRedoCmd->setEnabled(aDoc->canRedo());
 730   } else {
 731     foreach(QAction* aCmd, aCommands) {
 732       QString aId = aCmd->data().toString();
 733       if (aId == "NEW_CMD")
 734         aCmd->setEnabled(true);
 735       else if (aId == "EXIT_CMD")
 736         aCmd->setEnabled(true);
 737       else
 738         aCmd->setEnabled(false);
 739     }
 740   }
 741   myActionsMgr->update();
 742 }
 743
 744 //******************************************************
 745 QList<QAction*> XGUI_Workshop::getModuleCommands() const
 746 {
 747   QList<QAction*> aCommands;
 748   if (isSalomeMode()) { // update commands in SALOME mode
 749     aCommands = salomeConnector()->commandList();
 750   } else {
 751     XGUI_MainMenu* aMenuBar = myMainWindow->menuObject();
 752     foreach (XGUI_Workbench* aWb, aMenuBar->workbenches()) {
 753       if (aWb != aMenuBar->generalPage()) {
 754         foreach(XGUI_Command* aCmd, aWb->features())
 755           aCommands.append(aCmd);
 756       }
 757     }
 758   }
 759   return aCommands;
 760 }
 761
 762 //******************************************************
 763 QDockWidget* XGUI_Workshop::createObjectBrowser(QWidget* theParent)
 764 {
 765   QDockWidget* aObjDock = new QDockWidget(theParent);
 766   aObjDock->setAllowedAreas(Qt::LeftDockWidgetArea | Qt::RightDockWidgetArea);
 767   aObjDock->setWindowTitle(tr("Object browser"));
 768   aObjDock->setStyleSheet("::title { position: relative; padding-left: 5px; text-align: left center }");
 769   myObjectBrowser = new XGUI_ObjectsBrowser(aObjDock);
 770   connect(myObjectBrowser, SIGNAL(activePartChanged(FeaturePtr)), this, SLOT(changeCurrentDocument(FeaturePtr)));
 771   aObjDock->setWidget(myObjectBrowser);
 772
 773   myContextMenuMgr->connectObjectBrowser();
 774   return aObjDock;
 775 }
 776
 777 //******************************************************
 778 /*
 779  * Creates dock widgets, places them in corresponding area
 780  * and tabifies if necessary.
 781  */
 782 void XGUI_Workshop::createDockWidgets()
 783 {
 784   QMainWindow* aDesktop = isSalomeMode()? salomeConnector()->desktop() :
 785                                           myMainWindow;
 786   QDockWidget* aObjDock = createObjectBrowser(aDesktop);
 787   aDesktop->addDockWidget(Qt::LeftDockWidgetArea, aObjDock);
 788   myPropertyPanel = new XGUI_PropertyPanel(aDesktop);
 789   aDesktop->addDockWidget(Qt::LeftDockWidgetArea, myPropertyPanel);
 790   hidePropertyPanel(); //<! Invisible by default
 791   hideObjectBrowser();
 792   aDesktop->tabifyDockWidget(aObjDock, myPropertyPanel);
 793
 794   QPushButton* aOkBtn = myPropertyPanel->findChild<QPushButton*>(XGUI::PROP_PANEL_OK);
 795   connect(aOkBtn, SIGNAL(clicked()), myOperationMgr, SLOT(onCommitOperation()));
 796   QPushButton* aCancelBtn = myPropertyPanel->findChild<QPushButton*>(XGUI::PROP_PANEL_CANCEL);
 797   connect(aCancelBtn, SIGNAL(clicked()), myOperationMgr, SLOT(onAbortOperation()));
 798
 799   connect(myPropertyPanel, SIGNAL(keyReleased(const std::string&, QKeyEvent*)),
 800           myOperationMgr, SLOT(onKeyReleased(const std::string&, QKeyEvent*)));
 801
 802   connect(myPropertyPanel, SIGNAL(widgetActivated(ModuleBase_ModelWidget*)),
 803           myOperationMgr, SLOT(onWidgetActivated(ModuleBase_ModelWidget*)));
 804   connect(myOperationMgr, SIGNAL(activateNextWidget(ModuleBase_ModelWidget*)),
 805           myPropertyPanel, SLOT(onActivateNextWidget(ModuleBase_ModelWidget*)));
 806 }
 807
 808 //******************************************************
 809 void XGUI_Workshop::showPropertyPanel()
 810 {
 811   QAction* aViewAct = myPropertyPanel->toggleViewAction();
 812   //<! Restore ability to close panel from the window's menu
 813   aViewAct->setEnabled(true);
 814   myPropertyPanel->show();
 815   myPropertyPanel->raise();
 816 }
 817
 818 //******************************************************
 819 void XGUI_Workshop::hidePropertyPanel()
 820 {
 821   QAction* aViewAct = myPropertyPanel->toggleViewAction();
 822   //<! Do not allow to show empty property panel
 823   aViewAct->setEnabled(false);
 824   myPropertyPanel->hide();
 825 }
 826
 827 //******************************************************
 828 void XGUI_Workshop::showObjectBrowser()
 829 {
 830   myObjectBrowser->parentWidget()->show();
 831 }
 832
 833 //******************************************************
 834 void XGUI_Workshop::hideObjectBrowser()
 835 {
 836   myObjectBrowser->parentWidget()->hide();
 837 }
 838
 839 //******************************************************
 840 void XGUI_Workshop::onFeatureTriggered()
 841 {
 842   QAction* aCmd = dynamic_cast<QAction*>(sender());
 843   if (aCmd) {
 844     QString aId = salomeConnector()->commandId(aCmd);
 845     if (!aId.isNull())
 846       myModule->launchOperation(aId);
 847   }
 848 }
 849
 850 //******************************************************
 851 void XGUI_Workshop::changeCurrentDocument(ObjectPtr theObj)
 852 {
 853   PluginManagerPtr aMgr = ModelAPI_PluginManager::get();
 854   if (theObj) {
 855     ResultPartPtr aPart = boost::dynamic_pointer_cast<ModelAPI_ResultPart>(theObj);
 856     if (aPart) {
 857       DocumentPtr aPartDoc = aPart->partDoc();
 858       if (aPartDoc) {
 859         aMgr->setCurrentDocument(aPartDoc);
 860         return;
 861       }
 862     }
 863   }
 864   aMgr->setCurrentDocument(aMgr->rootDocument());
 865 }
 866
 867 //******************************************************
 868 void XGUI_Workshop::salomeViewerSelectionChanged()
 869 {
 870   emit salomeViewerSelection();
 871 }
 872
 873
 874 //**************************************************************
 875 XGUI_SalomeViewer* XGUI_Workshop::salomeViewer() const
 876 {
 877   return mySalomeConnector->viewer();
 878 }
 879
 880 //**************************************************************
 881 void XGUI_Workshop::onContextMenuCommand(const QString& theId, bool isChecked)
 882 {
 883   QList<ObjectPtr> aObjects = mySelector->selection()->selectedObjects();
 884   if ((theId == "ACTIVATE_PART_CMD") && (aObjects.size() > 0)) {
 885     ResultPartPtr aPart = boost::dynamic_pointer_cast<ModelAPI_ResultPart>(aObjects.first());
 886     activatePart(aPart);
 887   } else if (theId == "DEACTIVATE_PART_CMD")
 888     activatePart(ResultPartPtr());
 889   else if (theId == "DELETE_CMD")
 890     deleteObjects(aObjects);
 891   else if (theId == "SHOW_CMD")
 892     showObjects(aObjects, true);
 893   else if (theId == "HIDE_CMD")
 894     showObjects(aObjects, false);
 895 }
 896
 897 //**************************************************************
 898 void XGUI_Workshop::onWidgetValuesChanged()
 899 {
 900   ModuleBase_Operation* anOperation = myOperationMgr->currentOperation();
 901   FeaturePtr aFeature = anOperation->feature();
 902
 903   ModuleBase_ModelWidget* aSenderWidget = dynamic_cast<ModuleBase_ModelWidget*>(sender());
 904   //if (aCustom)
 905   //  aCustom->storeValue(aFeature);
 906
 907   const QList<ModuleBase_ModelWidget*>& aWidgets = myPropertyPanel->modelWidgets();
 908   QList<ModuleBase_ModelWidget*>::const_iterator anIt = aWidgets.begin(), aLast = aWidgets.end();
 909   for (; anIt != aLast; anIt++) {
 910     ModuleBase_ModelWidget* aCustom = *anIt;
 911     if (aCustom && (/*!aCustom->isInitialized(aFeature) ||*/ aCustom == aSenderWidget)) {
 912       aCustom->storeValue(aFeature);
 913     }
 914   }
 915 }
 916
 917 //**************************************************************
 918 void XGUI_Workshop::activatePart(ResultPartPtr theFeature)
 919 {
 920   changeCurrentDocument(theFeature);
 921   myObjectBrowser->activatePart(theFeature);
 922 }
 923
 924 //**************************************************************
 925 void XGUI_Workshop::activateLastPart()
 926 {
 927   PluginManagerPtr aMgr = ModelAPI_PluginManager::get();
 928   DocumentPtr aDoc = aMgr->rootDocument();
 929   std::string aGrpName = ModelAPI_ResultPart::group();
 930   ObjectPtr aLastPart = aDoc->object(aGrpName, aDoc->size(aGrpName) - 1);
 931   ResultPartPtr aPart = boost::dynamic_pointer_cast<ModelAPI_ResultPart>(aLastPart);
 932   if (aPart)
 933     activatePart(aPart);
 934 }
 935
 936 //**************************************************************
 937 void XGUI_Workshop::deleteObjects(const QList<ObjectPtr>& theList)
 938 {
 939   QMainWindow* aDesktop = isSalomeMode()? salomeConnector()->desktop() : myMainWindow;
 940   QMessageBox::StandardButton aRes = QMessageBox::warning(aDesktop, tr("Delete features"),
 941                                                           tr("Seleted features will be deleted. Continue?"),
 942                                                           QMessageBox::No | QMessageBox::Yes, QMessageBox::No);
 943   // ToDo: definbe deleting method
 944   if (aRes == QMessageBox::Yes) {
 945     PluginManagerPtr aMgr = ModelAPI_PluginManager::get();
 946     aMgr->rootDocument()->startOperation();
 947     foreach (ObjectPtr aObj, theList) {
 948       ResultPartPtr aPart = boost::dynamic_pointer_cast<ModelAPI_ResultPart>(aObj);
 949       if (aPart) {
 950         DocumentPtr aDoc = aPart->document();
 951         if (aDoc == aMgr->currentDocument()) {
 952           aDoc->close();
 953         }
 954         //aMgr->rootDocument()->removeFeature(aPart->owner());
 955       } else {
 956         FeaturePtr aFeature = boost::dynamic_pointer_cast<ModelAPI_Feature>(aObj);
 957         if (aFeature)
 958           aObj->document()->removeFeature(aFeature);
 959       }
 960     }
 961     myDisplayer->updateViewer();
 962     aMgr->rootDocument()->finishOperation();
 963   }
 964 }
 965
 966 //**************************************************************
 967 void XGUI_Workshop::showObjects(const QList<ObjectPtr>& theList, bool isVisible)
 968 {
 969   foreach (ObjectPtr aObj, theList) {
 970     ResultPtr aRes = boost::dynamic_pointer_cast<ModelAPI_Result>(aObj);
 971     if (aRes) {
 972       if (isVisible) {
 973         myDisplayer->display(aRes, false);
 974       } else {
 975         myDisplayer->erase(aRes, false);
 976       }
 977     }
 978   }
 979   myDisplayer->updateViewer();
 980 }
 981
 982 //**************************************************************
 983 void XGUI_Workshop::updateCommandsOnViewSelection()
 984 {
 985   PluginManagerPtr aMgr = ModelAPI_PluginManager::get();
 986   ModelAPI_ValidatorsFactory* aFactory = aMgr->validators();
 987   XGUI_Selection* aSelection = mySelector->selection();
 988
 989   QList<QAction*> aActions = getModuleCommands();
 990   foreach(QAction* aAction, aActions) {
 991     QString aId = aAction->data().toString();
 992     const ModelAPI_Validator* aValidator = aFactory->validator(aId.toStdString());
 993     if (aValidator) {
 994       const ModuleBase_SelectionValidator* aSelValidator =
 995         dynamic_cast<const ModuleBase_SelectionValidator*>(aValidator);
 996       if (aSelValidator) {
 997         aAction->setEnabled(aSelValidator->isValid(aSelection));
 998       }
 999     }
1000   }
1001 }