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