Salome HOME
Merge branch 'master' of newgeom:newgeom.git
[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 "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->display(aObj, false);
312     else {
313       myDisplayer->erase(aObj, false);
314     }
315   }
316   myDisplayer->updateViewer();
317 }
318
319 //******************************************************
320 void XGUI_Workshop::onFeatureCreatedMsg(const ModelAPI_ObjectUpdatedMessage* theMsg)
321 {
322   std::set<ObjectPtr> aObjects = theMsg->objects();
323
324   std::set<ObjectPtr>::const_iterator aIt;
325   bool aHasPart = false;
326   bool isDisplayed = false;
327   for (aIt = aObjects.begin(); aIt != aObjects.end(); ++aIt) {
328      ResultPartPtr aPart = boost::dynamic_pointer_cast<ModelAPI_ResultPart>(*aIt);
329     if (aPart) {
330       aHasPart = true;
331       //break;
332     } else {
333       myDisplayer->display(*aIt, false);
334       isDisplayed = true;
335     }
336   }
337   if (isDisplayed)
338     myDisplayer->updateViewer();
339   if (aHasPart) {
340     //The created part will be created in Object Browser later and we have to activate it
341     // only when it is created everywere
342     QTimer::singleShot(50, this, SLOT(activateLastPart()));
343   }
344 }
345
346 //******************************************************
347 void XGUI_Workshop::onObjectDeletedMsg(const ModelAPI_ObjectDeletedMessage* theMsg)
348 {
349   //std::set<ObjectPtr> aFeatures = theMsg->objects();
350 }
351  
352 //******************************************************
353 void XGUI_Workshop::onOperationStarted()
354 {
355   ModuleBase_Operation* aOperation = myOperationMgr->currentOperation();
356
357   if(aOperation->getDescription()->hasXmlRepresentation()) { //!< No need for property panel
358     connectWithOperation(aOperation);
359
360     showPropertyPanel();
361     QString aXmlRepr = aOperation->getDescription()->xmlRepresentation();
362     ModuleBase_WidgetFactory aFactory = ModuleBase_WidgetFactory(aXmlRepr.toStdString(), myModuleConnector);
363
364     myPropertyPanel->cleanContent();
365     aFactory.createWidget(myPropertyPanel->contentWidget());
366     
367     QList<ModuleBase_ModelWidget*> aWidgets = aFactory.getModelWidgets();
368     QList<ModuleBase_ModelWidget*>::const_iterator anIt = aWidgets.begin(), aLast = aWidgets.end();
369     ModuleBase_ModelWidget* aWidget;
370     for (; anIt != aLast; anIt++) {
371       aWidget = *anIt;
372       //QObject::connect(aWidget, SIGNAL(valuesChanged()),  aOperation, SLOT(storeCustomValue()));
373       QObject::connect(aWidget, SIGNAL(valuesChanged()),
374                        this, SLOT(onWidgetValuesChanged()));
375       // Init default values
376       if (!aOperation->isEditOperation() && aWidget->hasDefaultValue()) {
377         aWidget->storeValue(aOperation->feature());
378       }
379     }
380
381     myPropertyPanel->setModelWidgets(aWidgets);
382     myPropertyPanel->setWindowTitle(aOperation->getDescription()->description());
383   }
384   updateCommandStatus();
385 }
386
387 //******************************************************
388 void XGUI_Workshop::onOperationStopped(ModuleBase_Operation* theOperation)
389 {
390   //!< No need for property panel
391   updateCommandStatus();
392   hidePropertyPanel();
393   myPropertyPanel->cleanContent();
394 }
395
396 /*
397  *
398  */
399 void XGUI_Workshop::addFeature(const Config_FeatureMessage* theMessage)
400 {
401   if (!theMessage) {
402 #ifdef _DEBUG
403     qDebug() << "XGUI_Workshop::addFeature: NULL message.";
404 #endif
405     return;
406   }
407   // Remember features icons
408   myIcons[QString::fromStdString(theMessage->id())] = QString::fromStdString(theMessage->icon());
409
410   //Find or create Workbench
411   QString aWchName = QString::fromStdString(theMessage->workbenchId());
412   QString aNestedFeatures = QString::fromStdString(theMessage->nestedFeatures());
413   bool isUsePropPanel = theMessage->isUseInput();
414   QString aId = QString::fromStdString(theMessage->id());
415   if (isSalomeMode()) {
416     QAction* aAction = salomeConnector()->addFeature(aWchName,
417                               aId,
418                               QString::fromStdString(theMessage->text()),
419                               QString::fromStdString(theMessage->tooltip()),
420                               QIcon(theMessage->icon().c_str()),
421                               QKeySequence(), isUsePropPanel);
422     salomeConnector()->setNestedActions(aId, aNestedFeatures.split(" "));
423     myActionsMgr->addCommand(aAction);
424     myModule->featureCreated(aAction);
425   } else {
426
427     XGUI_MainMenu* aMenuBar = myMainWindow->menuObject();
428     XGUI_Workbench* aPage = aMenuBar->findWorkbench(aWchName);
429     if (!aPage) {
430       aPage = addWorkbench(aWchName);
431     }
432     //Find or create Group
433     QString aGroupName = QString::fromStdString(theMessage->groupId());
434     XGUI_MenuGroupPanel* aGroup = aPage->findGroup(aGroupName);
435     if (!aGroup) {
436       aGroup = aPage->addGroup(aGroupName);
437     }
438     //Create feature...
439     XGUI_Command* aCommand = aGroup->addFeature(aId,
440                                                 QString::fromStdString(theMessage->text()),
441                                                 QString::fromStdString(theMessage->tooltip()),
442                                                 QIcon(theMessage->icon().c_str()),
443                                                 QKeySequence(), isUsePropPanel);
444     aCommand->setNestedCommands(aNestedFeatures.split(" ", QString::SkipEmptyParts));
445     myActionsMgr->addCommand(aCommand);
446     myModule->featureCreated(aCommand);
447   }
448 }
449
450 /*
451  * Makes a signal/slot connections between Property Panel
452  * and given operation. The given operation becomes a
453  * current operation and previous operation if exists
454  */
455 void XGUI_Workshop::connectWithOperation(ModuleBase_Operation* theOperation)
456 {
457   QAction* aCommand = 0;
458   if (isSalomeMode()) {
459     aCommand = salomeConnector()->command(theOperation->getDescription()->operationId());
460   } else {
461     XGUI_MainMenu* aMenu = myMainWindow->menuObject();
462     aCommand = aMenu->feature(theOperation->getDescription()->operationId());
463   }
464   //Abort operation on uncheck the command
465   connect(aCommand, SIGNAL(triggered(bool)), theOperation, SLOT(setRunning(bool)));
466 }
467
468 /*
469  * Saves document with given name.
470  */
471 void XGUI_Workshop::saveDocument(QString theName)
472 {
473   QApplication::restoreOverrideCursor();
474   PluginManagerPtr aMgr = ModelAPI_PluginManager::get();
475   DocumentPtr aDoc = aMgr->rootDocument();
476   aDoc->save(theName.toLatin1().constData());
477   QApplication::restoreOverrideCursor();
478 }
479
480 //******************************************************
481 void XGUI_Workshop::onExit()
482 {
483   PluginManagerPtr aMgr = ModelAPI_PluginManager::get();
484   DocumentPtr aDoc = aMgr->rootDocument();
485   if(aDoc->isModified()) {
486     int anAnswer = QMessageBox::question(
487         myMainWindow, tr("Save current file"),
488         tr("The document is modified, save before exit?"),
489         QMessageBox::Save | QMessageBox::Discard | QMessageBox::Cancel, QMessageBox::Cancel);
490     if(anAnswer == QMessageBox::Save) {
491       bool saved = onSave();
492       if(!saved) {
493         return;
494       }
495     } else if (anAnswer == QMessageBox::Cancel) {
496       return;
497     }
498   }
499   qApp->exit();
500 }
501
502 //******************************************************
503 void XGUI_Workshop::onNew()
504 {
505   QApplication::setOverrideCursor(Qt::WaitCursor);
506   if (objectBrowser() == 0) {
507     createDockWidgets();
508     mySelector->connectViewers();
509   }
510   myViewerProxy->connectToViewer();
511   showObjectBrowser();
512   if (!isSalomeMode()) {
513     myMainWindow->showPythonConsole();
514     QMdiSubWindow* aWnd = myMainWindow->viewer()->createView();
515     aWnd->showMaximized();
516     updateCommandStatus();
517   }
518   myContextMenuMgr->connectViewer();
519   QApplication::restoreOverrideCursor();
520 }
521
522 //******************************************************
523 void XGUI_Workshop::onOpen()
524 {
525   //save current file before close if modified
526   PluginManagerPtr aMgr = ModelAPI_PluginManager::get();
527   DocumentPtr aDoc = aMgr->rootDocument();
528   if(aDoc->isModified()) {
529     //TODO(sbh): re-launch the app?
530     int anAnswer = QMessageBox::question(
531         myMainWindow, tr("Save current file"),
532         tr("The document is modified, save before opening another?"),
533         QMessageBox::Save | QMessageBox::Discard | QMessageBox::Cancel, QMessageBox::Cancel);
534     if(anAnswer == QMessageBox::Save) {
535       onSave();
536     } else if (anAnswer == QMessageBox::Cancel) {
537       return;
538     }
539     aDoc->close();
540     myCurrentDir = "";
541   }
542
543   //show file dialog, check if readable and open
544   myCurrentDir = QFileDialog::getExistingDirectory(mainWindow());
545   if(myCurrentDir.isEmpty())
546     return;
547   QFileInfo aFileInfo(myCurrentDir);
548   if(!aFileInfo.exists() || !aFileInfo.isReadable()) {
549     QMessageBox::critical(myMainWindow, tr("Warning"), tr("Unable to open the file."));
550     myCurrentDir = "";
551     return;
552   }
553   QApplication::setOverrideCursor(Qt::WaitCursor);
554   aDoc->load(myCurrentDir.toLatin1().constData());
555   updateCommandStatus();
556   myObjectBrowser->rebuildDataTree();
557   QApplication::restoreOverrideCursor();
558 }
559
560 //******************************************************
561 bool XGUI_Workshop::onSave()
562 {
563   if(myCurrentDir.isEmpty()) {
564     return onSaveAs();
565   }
566   saveDocument(myCurrentDir);
567   updateCommandStatus();
568   return true;
569 }
570
571 //******************************************************
572 bool XGUI_Workshop::onSaveAs()
573 {
574   QFileDialog dialog(mainWindow());
575   dialog.setWindowTitle(tr("Select directory to save files..."));
576   dialog.setFileMode(QFileDialog::Directory);
577   dialog.setFilter(tr("Folders (*)"));
578   dialog.setOptions(QFileDialog::HideNameFilterDetails | QFileDialog::ShowDirsOnly);
579   dialog.setViewMode(QFileDialog::Detail);
580
581   if(!dialog.exec()) {
582     return false;
583   }
584   QString aTempDir = dialog.selectedFiles().first();
585   QDir aDir(aTempDir);
586   if(aDir.exists() && !aDir.entryInfoList(QDir::NoDotAndDotDot|QDir::AllEntries).isEmpty()) {
587     int answer = QMessageBox::question(myMainWindow,
588                                        //: Title of the dialog which asks user if he wants to save study in existing non-empty folder
589                                        tr("Save"),
590                                        tr("The folder already contains some files, save anyway?"),
591                                        QMessageBox::Save|QMessageBox::Cancel);
592     if(answer == QMessageBox::Cancel) {
593       return false;
594     }
595   }
596   myCurrentDir = aTempDir;
597   return onSave();
598 }
599
600 //******************************************************
601 void XGUI_Workshop::onUndo()
602 {
603   objectBrowser()->treeView()->setCurrentIndex(QModelIndex());
604   PluginManagerPtr aMgr = ModelAPI_PluginManager::get();
605   DocumentPtr aDoc = aMgr->rootDocument();
606   if (aDoc->isOperation())
607     operationMgr()->abortOperation();
608   aDoc->undo();
609   updateCommandStatus();
610 }
611
612 //******************************************************
613 void XGUI_Workshop::onRedo()
614 {
615   objectBrowser()->treeView()->setCurrentIndex(QModelIndex());
616   PluginManagerPtr aMgr = ModelAPI_PluginManager::get();
617   DocumentPtr aDoc = aMgr->rootDocument();
618   if (aDoc->isOperation())
619     operationMgr()->abortOperation();
620   aDoc->redo();
621   updateCommandStatus();
622 }
623
624 //******************************************************
625 ModuleBase_IModule* XGUI_Workshop::loadModule(const QString& theModule)
626 {
627   QString libName =
628       QString::fromStdString(library(theModule.toStdString()));
629   if (libName.isEmpty()) {
630     qWarning(
631     qPrintable( tr( "Information about module \"%1\" doesn't exist." ).arg( theModule ) ));
632     return 0;
633   }
634
635   QString err;
636   CREATE_FUNC crtInst = 0;
637
638 #ifdef WIN32
639   HINSTANCE modLib = ::LoadLibrary((LPTSTR) qPrintable(libName));
640   if (!modLib) {
641     LPVOID lpMsgBuf;
642     ::FormatMessage(
643         FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS,
644         0, ::GetLastError(), 0, (LPTSTR) & lpMsgBuf, 0, 0);
645     QString aMsg((char*) &lpMsgBuf);
646     err = QString("Failed to load  %1. %2").arg(libName).arg(aMsg);
647     ::LocalFree(lpMsgBuf);
648   } else {
649     crtInst = (CREATE_FUNC) ::GetProcAddress(modLib, CREATE_MODULE);
650     if (!crtInst) {
651       LPVOID lpMsgBuf;
652       ::FormatMessage(
653           FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM
654               | FORMAT_MESSAGE_IGNORE_INSERTS,
655           0, ::GetLastError(), 0, (LPTSTR) & lpMsgBuf, 0, 0);
656       QString aMsg((char*) &lpMsgBuf);
657       err = QString("Failed to find  %1 function. %2").arg( CREATE_MODULE).arg(aMsg);
658       ::LocalFree(lpMsgBuf);
659     }
660   }
661 #else
662   void* modLib = dlopen( libName.toLatin1(), RTLD_LAZY | RTLD_GLOBAL );
663   if ( !modLib ) {
664     err = QString( "Can not load library %1. %2" ).arg( libName ).arg( dlerror() );
665   } else {
666     crtInst = (CREATE_FUNC)dlsym( modLib, CREATE_MODULE );
667     if ( !crtInst ) {
668       err = QString( "Failed to find function %1. %2" ).arg( CREATE_MODULE ).arg( dlerror() );
669     }
670   }
671 #endif
672
673   ModuleBase_IModule* aModule = crtInst ? crtInst(this) : 0;
674
675   if (!err.isEmpty()) {
676     if (mainWindow()) {
677       QMessageBox::warning(mainWindow(), tr("Error"), err);
678     } else {
679       qWarning( qPrintable( err ));
680     }
681   }
682   return aModule;
683 }
684
685 //******************************************************
686 bool XGUI_Workshop::activateModule()
687 {
688   Config_ModuleReader aModuleReader;
689   QString moduleName = QString::fromStdString(aModuleReader.getModuleName());
690   myModule = loadModule(moduleName);
691   if (!myModule)
692     return false;
693   myModule->createFeatures();
694   myActionsMgr->update();
695   return true;
696 }
697
698 //******************************************************
699 void XGUI_Workshop::updateCommandStatus()
700 {
701   QList<QAction*> aCommands;
702   if (isSalomeMode()) { // update commands in SALOME mode
703     aCommands = salomeConnector()->commandList();
704   } else {
705     XGUI_MainMenu* aMenuBar = myMainWindow->menuObject();
706     foreach (XGUI_Command* aCmd, aMenuBar->features())
707       aCommands.append(aCmd);
708   }
709   PluginManagerPtr aMgr = ModelAPI_PluginManager::get();
710   if (aMgr->hasRootDocument()) {
711     QAction* aUndoCmd;
712     QAction* aRedoCmd;
713     foreach(QAction* aCmd, aCommands) {
714       QString aId = aCmd->data().toString();
715       if (aId == "UNDO_CMD")
716         aUndoCmd = aCmd;
717       else if (aId == "REDO_CMD")
718         aRedoCmd = aCmd;
719       else // Enable all commands
720         aCmd->setEnabled(true);
721     }
722     DocumentPtr aDoc = aMgr->rootDocument();
723     aUndoCmd->setEnabled(aDoc->canUndo());
724     aRedoCmd->setEnabled(aDoc->canRedo());
725   } else {
726     foreach(QAction* aCmd, aCommands) {
727       QString aId = aCmd->data().toString();
728       if (aId == "NEW_CMD")
729         aCmd->setEnabled(true);
730       else if (aId == "EXIT_CMD")
731         aCmd->setEnabled(true);
732       else 
733         aCmd->setEnabled(false);
734     }
735   }
736   myActionsMgr->update();
737 }
738
739 //******************************************************
740 QList<QAction*> XGUI_Workshop::getModuleCommands() const
741 {
742   QList<QAction*> aCommands;
743   if (isSalomeMode()) { // update commands in SALOME mode
744     aCommands = salomeConnector()->commandList();
745   } else {
746     XGUI_MainMenu* aMenuBar = myMainWindow->menuObject();
747     foreach (XGUI_Workbench* aWb, aMenuBar->workbenches()) {
748       if (aWb != aMenuBar->generalPage()) {
749         foreach(XGUI_Command* aCmd, aWb->features())
750           aCommands.append(aCmd);
751       }
752     }
753   }
754   return aCommands;
755 }
756
757 //******************************************************
758 QDockWidget* XGUI_Workshop::createObjectBrowser(QWidget* theParent)
759 {
760   QDockWidget* aObjDock = new QDockWidget(theParent);
761   aObjDock->setAllowedAreas(Qt::LeftDockWidgetArea | Qt::RightDockWidgetArea);
762   aObjDock->setWindowTitle(tr("Object browser"));
763   aObjDock->setStyleSheet("::title { position: relative; padding-left: 5px; text-align: left center }");
764   myObjectBrowser = new XGUI_ObjectsBrowser(aObjDock);
765   connect(myObjectBrowser, SIGNAL(activePartChanged(FeaturePtr)), this, SLOT(changeCurrentDocument(FeaturePtr)));
766   aObjDock->setWidget(myObjectBrowser);
767
768   myContextMenuMgr->connectObjectBrowser();
769   return aObjDock;
770 }
771
772 //******************************************************
773 /*
774  * Creates dock widgets, places them in corresponding area
775  * and tabifies if necessary.
776  */
777 void XGUI_Workshop::createDockWidgets()
778 {
779   QMainWindow* aDesktop = isSalomeMode()? salomeConnector()->desktop() :
780                                           myMainWindow;
781   QDockWidget* aObjDock = createObjectBrowser(aDesktop);
782   aDesktop->addDockWidget(Qt::LeftDockWidgetArea, aObjDock);
783   myPropertyPanel = new XGUI_PropertyPanel(aDesktop);
784   aDesktop->addDockWidget(Qt::LeftDockWidgetArea, myPropertyPanel);
785   hidePropertyPanel(); //<! Invisible by default
786   hideObjectBrowser();
787   aDesktop->tabifyDockWidget(aObjDock, myPropertyPanel);
788
789   QPushButton* aOkBtn = myPropertyPanel->findChild<QPushButton*>(XGUI::PROP_PANEL_OK);
790   connect(aOkBtn, SIGNAL(clicked()), myOperationMgr, SLOT(onCommitOperation()));
791   QPushButton* aCancelBtn = myPropertyPanel->findChild<QPushButton*>(XGUI::PROP_PANEL_CANCEL);
792   connect(aCancelBtn, SIGNAL(clicked()), myOperationMgr, SLOT(onAbortOperation()));
793
794   connect(myPropertyPanel, SIGNAL(keyReleased(const std::string&, QKeyEvent*)),
795           myOperationMgr, SLOT(onKeyReleased(const std::string&, QKeyEvent*)));
796
797   connect(myPropertyPanel, SIGNAL(widgetActivated(ModuleBase_ModelWidget*)),
798           myOperationMgr, SLOT(onWidgetActivated(ModuleBase_ModelWidget*)));
799   connect(myOperationMgr, SIGNAL(activateNextWidget(ModuleBase_ModelWidget*)),
800           myPropertyPanel, SLOT(onActivateNextWidget(ModuleBase_ModelWidget*)));
801 }
802
803 //******************************************************
804 void XGUI_Workshop::showPropertyPanel()
805 {
806   QAction* aViewAct = myPropertyPanel->toggleViewAction();
807   //<! Restore ability to close panel from the window's menu
808   aViewAct->setEnabled(true);
809   myPropertyPanel->show();
810   myPropertyPanel->raise();
811 }
812
813 //******************************************************
814 void XGUI_Workshop::hidePropertyPanel()
815 {
816   QAction* aViewAct = myPropertyPanel->toggleViewAction();
817   //<! Do not allow to show empty property panel
818   aViewAct->setEnabled(false);
819   myPropertyPanel->hide();
820 }
821
822 //******************************************************
823 void XGUI_Workshop::showObjectBrowser()
824 {
825   myObjectBrowser->parentWidget()->show();
826 }
827
828 //******************************************************
829 void XGUI_Workshop::hideObjectBrowser()
830 {
831   myObjectBrowser->parentWidget()->hide();
832 }
833
834 //******************************************************
835 void XGUI_Workshop::onFeatureTriggered()
836 {
837   QAction* aCmd = dynamic_cast<QAction*>(sender());
838   if (aCmd) {
839     QString aId = salomeConnector()->commandId(aCmd);
840     if (!aId.isNull())
841       myModule->launchOperation(aId);
842   }
843 }
844
845 //******************************************************
846 void XGUI_Workshop::changeCurrentDocument(ObjectPtr theObj)
847 {
848   PluginManagerPtr aMgr = ModelAPI_PluginManager::get();
849   if (theObj) {
850     ResultPartPtr aPart = boost::dynamic_pointer_cast<ModelAPI_ResultPart>(theObj);
851     if (aPart) {
852       DocumentPtr aPartDoc = aPart->partDoc();
853       if (aPartDoc) {
854         aMgr->setCurrentDocument(aPartDoc);
855         return;
856       }
857     }
858   } 
859   aMgr->setCurrentDocument(aMgr->rootDocument());
860 }
861
862 //******************************************************
863 void XGUI_Workshop::salomeViewerSelectionChanged()
864 {
865   emit salomeViewerSelection();
866 }
867
868
869 //**************************************************************
870 XGUI_SalomeViewer* XGUI_Workshop::salomeViewer() const 
871
872   return mySalomeConnector->viewer(); 
873 }
874
875 //**************************************************************
876 void XGUI_Workshop::onContextMenuCommand(const QString& theId, bool isChecked)
877 {
878   QList<ObjectPtr> aObjects = mySelector->selection()->selectedObjects();
879   if ((theId == "ACTIVATE_PART_CMD") && (aObjects.size() > 0)) {
880     ResultPartPtr aPart = boost::dynamic_pointer_cast<ModelAPI_ResultPart>(aObjects.first());
881     activatePart(aPart);
882   } else if (theId == "DEACTIVATE_PART_CMD") 
883     activatePart(ResultPartPtr());
884   else if (theId == "DELETE_CMD")
885     deleteObjects(aObjects);
886   else if (theId == "SHOW_CMD")
887     showObjects(aObjects, true);
888   else if (theId == "HIDE_CMD")
889     showObjects(aObjects, false);
890 }
891
892 //**************************************************************
893 void XGUI_Workshop::onWidgetValuesChanged()
894 {
895   ModuleBase_Operation* anOperation = myOperationMgr->currentOperation();
896   FeaturePtr aFeature = anOperation->feature();
897
898   ModuleBase_ModelWidget* aSenderWidget = dynamic_cast<ModuleBase_ModelWidget*>(sender());
899   //if (aCustom)
900   //  aCustom->storeValue(aFeature);
901
902   const QList<ModuleBase_ModelWidget*>& aWidgets = myPropertyPanel->modelWidgets();
903   QList<ModuleBase_ModelWidget*>::const_iterator anIt = aWidgets.begin(), aLast = aWidgets.end();
904   for (; anIt != aLast; anIt++) {
905     ModuleBase_ModelWidget* aCustom = *anIt;
906     if (aCustom && (/*!aCustom->isInitialized(aFeature) ||*/ aCustom == aSenderWidget)) {
907       aCustom->storeValue(aFeature);
908     }
909   }
910 }
911
912 //**************************************************************
913 void XGUI_Workshop::activatePart(ResultPartPtr theFeature)
914 {
915   changeCurrentDocument(theFeature);
916   myObjectBrowser->activatePart(theFeature);
917 }
918
919 //**************************************************************
920 void XGUI_Workshop::activateLastPart()
921 {
922   PluginManagerPtr aMgr = ModelAPI_PluginManager::get();
923   DocumentPtr aDoc = aMgr->rootDocument();
924   std::string aGrpName = ModelAPI_ResultPart::group();
925   ObjectPtr aLastPart = aDoc->object(aGrpName, aDoc->size(aGrpName) - 1);
926   ResultPartPtr aPart = boost::dynamic_pointer_cast<ModelAPI_ResultPart>(aLastPart);
927   if (aPart)
928     activatePart(aPart);
929 }
930
931 //**************************************************************
932 void XGUI_Workshop::deleteObjects(const QList<ObjectPtr>& theList)
933 {
934   QMainWindow* aDesktop = isSalomeMode()? salomeConnector()->desktop() : myMainWindow;
935   QMessageBox::StandardButton aRes = QMessageBox::warning(aDesktop, tr("Delete features"), 
936                                                           tr("Seleted features will be deleted. Continue?"), 
937                                                           QMessageBox::No | QMessageBox::Yes, QMessageBox::No);
938   // ToDo: definbe deleting method
939   if (aRes == QMessageBox::Yes) {
940     PluginManagerPtr aMgr = ModelAPI_PluginManager::get();
941     aMgr->rootDocument()->startOperation();
942     foreach (ObjectPtr aObj, theList) {
943       ResultPartPtr aPart = boost::dynamic_pointer_cast<ModelAPI_ResultPart>(aObj);
944       if (aPart) {
945         DocumentPtr aDoc = aPart->document();
946         if (aDoc == aMgr->currentDocument()) {
947           aDoc->close();
948         }
949         //aMgr->rootDocument()->removeFeature(aPart->owner());
950       } else {
951         FeaturePtr aFeature = boost::dynamic_pointer_cast<ModelAPI_Feature>(aObj);
952         if (aFeature)
953           aObj->document()->removeFeature(aFeature);
954       }
955     }
956     myDisplayer->updateViewer();
957     aMgr->rootDocument()->finishOperation();
958   }
959 }
960
961 //**************************************************************
962 void XGUI_Workshop::showObjects(const QList<ObjectPtr>& theList, bool isVisible)
963 {
964   foreach (ObjectPtr aObj, theList) {
965     ResultPtr aRes = boost::dynamic_pointer_cast<ModelAPI_Result>(aObj);
966     if (aRes) {
967       if (isVisible) {
968         myDisplayer->display(aRes, false);
969       } else {
970         myDisplayer->erase(aRes, false);
971       }
972     }
973   }
974   myDisplayer->updateViewer();
975 }
976
977 //**************************************************************
978 void XGUI_Workshop::updateCommandsOnViewSelection()
979 {
980   PluginManagerPtr aMgr = ModelAPI_PluginManager::get();
981   ModelAPI_ValidatorsFactory* aFactory = aMgr->validators();
982   XGUI_Selection* aSelection = mySelector->selection();
983
984   QList<QAction*> aActions = getModuleCommands();
985   foreach(QAction* aAction, aActions) {
986     QString aId = aAction->data().toString();
987     const ModelAPI_Validator* aValidator = aFactory->validator(aId.toStdString());
988     if (aValidator) {
989       const ModuleBase_SelectionValidator* aSelValidator = 
990         dynamic_cast<const ModuleBase_SelectionValidator*>(aValidator);
991       if (aSelValidator) {
992         aAction->setEnabled(aSelValidator->isValid(aSelection));
993       }
994     }
995   }
996 }