Salome HOME
Result attributes validators created
[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 #include <ModuleBase_ResultValidators.h>
45
46 #include <Config_Common.h>
47 #include <Config_FeatureMessage.h>
48 #include <Config_PointerMessage.h>
49 #include <Config_ModuleReader.h>
50
51 #include <QApplication>
52 #include <QFileDialog>
53 #include <QMessageBox>
54 #include <QMdiSubWindow>
55 #include <QPushButton>
56 #include <QDockWidget>
57 #include <QLayout>
58 #include <QTimer>
59
60 #ifdef _DEBUG
61 #include <QDebug>
62 #endif
63
64 #ifdef WIN32
65 #include <windows.h>
66 #else
67 #include <dlfcn.h>
68 #endif
69
70
71 QMap<QString, QString> XGUI_Workshop::myIcons;
72
73 QString XGUI_Workshop::featureIcon(const std::string& theId)
74 {
75   QString aId(theId.c_str());
76   if (myIcons.contains(aId))
77     return myIcons[aId];
78   return QString();
79 }
80
81 XGUI_Workshop::XGUI_Workshop(XGUI_SalomeConnector* theConnector)
82   : QObject(),
83   myCurrentDir(QString()),
84   myModule(NULL),
85   mySalomeConnector(theConnector),
86   myPropertyPanel(0),
87   myObjectBrowser(0),
88   myDisplayer(0)
89 {
90   myMainWindow = mySalomeConnector? 0 : new XGUI_MainWindow();
91
92   myDisplayer = new XGUI_Displayer(this);
93
94   mySelector = new XGUI_SelectionMgr(this);
95   //connect(mySelector, SIGNAL(selectionChanged()), this, SLOT(updateModuleCommands()));
96
97   myOperationMgr = new XGUI_OperationMgr(this);
98   myActionsMgr = new XGUI_ActionsMgr(this);
99   myErrorDlg = new XGUI_ErrorDialog(myMainWindow);
100   myContextMenuMgr = new XGUI_ContextMenuMgr(this);
101   connect(myContextMenuMgr, SIGNAL(actionTriggered(const QString&, bool)), 
102           this, SLOT(onContextMenuCommand(const QString&, bool)));
103
104   myViewerProxy = new XGUI_ViewerProxy(this);
105   connect(myViewerProxy, SIGNAL(selectionChanged()), this, SLOT(updateCommandsOnViewSelection()));
106   
107   myModuleConnector = new XGUI_ModuleConnector(this);
108
109   connect(myOperationMgr, SIGNAL(operationStarted()), SLOT(onOperationStarted()));
110   connect(myOperationMgr, SIGNAL(operationResumed()), SLOT(onOperationStarted()));
111   connect(myOperationMgr, SIGNAL(operationStopped(ModuleBase_Operation*)), SLOT(onOperationStopped(ModuleBase_Operation*)));
112   connect(myMainWindow, SIGNAL(exitKeySequence()), SLOT(onExit()));
113   connect(myOperationMgr, SIGNAL(operationStarted()), myActionsMgr, SLOT(update()));
114   connect(myOperationMgr, SIGNAL(operationStopped(ModuleBase_Operation*)), myActionsMgr, SLOT(update()));
115   connect(this, SIGNAL(errorOccurred(const QString&)), myErrorDlg, SLOT(addError(const QString&)));
116 }
117
118 //******************************************************
119 XGUI_Workshop::~XGUI_Workshop(void)
120 {
121 }
122
123 //******************************************************
124 void XGUI_Workshop::startApplication()
125 {
126   initMenu();
127   //Initialize event listening
128   Events_Loop* aLoop = Events_Loop::loop();
129   aLoop->registerListener(this, Events_Error::errorID()); //!< Listening application errors.
130   //TODO(sbh): Implement static method to extract event id [SEID]
131   aLoop->registerListener(this, Events_Loop::eventByName(EVENT_FEATURE_LOADED));
132   // TODO Is it good to use non standard event within workshop?
133   aLoop->registerListener(this, Events_Loop::eventByName(EVENT_OPERATION_LAUNCHED));
134   aLoop->registerListener(this, Events_Loop::eventByName(EVENT_OBJECT_UPDATED));
135   aLoop->registerListener(this, Events_Loop::eventByName(EVENT_OBJECT_CREATED));
136   aLoop->registerListener(this, Events_Loop::eventByName(EVENT_OBJECT_TO_REDISPLAY));
137   aLoop->registerListener(this, Events_Loop::eventByName(EVENT_OBJECT_DELETED));
138
139   registerValidators();
140   activateModule();
141   if (myMainWindow) {
142     myMainWindow->show();
143     updateCommandStatus();
144   }
145   onNew();
146 }
147
148 //******************************************************
149 void XGUI_Workshop::initMenu()
150 {
151   myContextMenuMgr->createActions();
152
153   if (isSalomeMode()) {
154     // Create only Undo, Redo commands
155     QAction* aAction = salomeConnector()->addEditCommand("UNDO_CMD", 
156                                       tr("Undo"), tr("Undo last command"),
157                                       QIcon(":pictures/undo.png"), 
158                                       QKeySequence::Undo, false);
159     connect(aAction, SIGNAL(triggered(bool)), this, SLOT(onUndo()));
160     aAction = salomeConnector()->addEditCommand("REDO_CMD", 
161                                       tr("Redo"), tr("Redo last command"),
162                                       QIcon(":pictures/redo.png"), 
163                                       QKeySequence::Redo, false);
164     connect(aAction, SIGNAL(triggered(bool)), this, SLOT(onRedo()));
165     salomeConnector()->addEditMenuSeparator();
166     return;
167   }
168   XGUI_Workbench* aPage = myMainWindow->menuObject()->generalPage();
169
170   // File commands group
171   XGUI_MenuGroupPanel* aGroup = aPage->addGroup("Default");
172
173   XGUI_Command* aCommand;
174
175   aCommand = aGroup->addFeature("SAVE_CMD", tr("Save..."), tr("Save the document"),
176                                 QIcon(":pictures/save.png"), QKeySequence::Save);
177   aCommand->connectTo(this, SLOT(onSave()));
178   //aCommand->disable();
179
180   aCommand = aGroup->addFeature("UNDO_CMD", tr("Undo"), tr("Undo last command"),
181                                 QIcon(":pictures/undo.png"), QKeySequence::Undo);
182   aCommand->connectTo(this, SLOT(onUndo()));
183
184   aCommand = aGroup->addFeature("REDO_CMD", tr("Redo"), tr("Redo last command"),
185                                 QIcon(":pictures/redo.png"), QKeySequence::Redo);
186   aCommand->connectTo(this, SLOT(onRedo()));
187
188   aCommand = aGroup->addFeature("REBUILD_CMD", tr("Rebuild"), tr("Rebuild data objects"),
189                                 QIcon(":pictures/rebuild.png"));
190
191   aCommand = aGroup->addFeature("SAVEAS_CMD", tr("Save as..."), tr("Save the document into a file"),
192                                 QIcon(":pictures/save.png"));
193   aCommand->connectTo(this, SLOT(onSaveAs()));
194   //aCommand->disable();
195
196   aCommand = aGroup->addFeature("OPEN_CMD", tr("Open..."), tr("Open a new document"),
197                                 QIcon(":pictures/open.png"), QKeySequence::Open);
198   aCommand->connectTo(this, SLOT(onOpen()));
199
200   //aCommand = aGroup->addFeature("NEW_CMD", tr("New"), tr("Create a new document"),
201   //                              QIcon(":pictures/new.png"), QKeySequence::New);
202   //aCommand->connectTo(this, SLOT(onNew()));
203
204   aCommand = aGroup->addFeature("EXIT_CMD", tr("Exit"), tr("Exit application"),
205                                 QIcon(":pictures/close.png"), QKeySequence::Close);
206   aCommand->connectTo(this, SLOT(onExit()));
207   //FIXME: SBH's test action. Can be used for some GUI tests.
208   //#ifdef _DEBUG
209   //  aCommand = aGroup->addFeature("TEST_CMD", "Test!", "Private debug button",
210   //                                QIcon(":pictures/close.png"));
211   //  aCommand->connectTo(myActionsMgr, SLOT(update()));
212   //#endif
213 }
214
215 //******************************************************
216 XGUI_Workbench* XGUI_Workshop::addWorkbench(const QString& theName)
217 {
218   XGUI_MainMenu* aMenuBar = myMainWindow->menuObject();
219   return aMenuBar->addWorkbench(theName);
220 }
221
222 //******************************************************
223 void XGUI_Workshop::processEvent(const Events_Message* theMessage)
224 {
225   //A message to start feature creation received.
226   static Events_ID aFeatureLoadedId = Events_Loop::loop()->eventByName(EVENT_FEATURE_LOADED);
227   if (theMessage->eventID() == aFeatureLoadedId) {
228     const Config_FeatureMessage* aFeatureMsg = dynamic_cast<const Config_FeatureMessage*>(theMessage);
229     if(!aFeatureMsg->isInternal()) {
230       addFeature(aFeatureMsg);
231     }
232     return;
233   }
234
235   // Process creation of Part
236   if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_OBJECT_CREATED)) {
237     const ModelAPI_ObjectUpdatedMessage* aUpdMsg = dynamic_cast<const ModelAPI_ObjectUpdatedMessage*>(theMessage);
238     onFeatureCreatedMsg(aUpdMsg);
239     return;
240   }
241
242   // Redisplay feature
243   if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_OBJECT_TO_REDISPLAY)) {
244     const ModelAPI_ObjectUpdatedMessage* aUpdMsg = dynamic_cast<const ModelAPI_ObjectUpdatedMessage*>(theMessage);
245     onFeatureRedisplayMsg(aUpdMsg);
246     return;
247   }
248
249   //Update property panel on corresponding message. If there is no current operation (no
250   //property panel), or received message has different feature to the current - do nothing.
251   if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_OBJECT_UPDATED)) {
252     const ModelAPI_ObjectUpdatedMessage* anUpdateMsg =
253         dynamic_cast<const ModelAPI_ObjectUpdatedMessage*>(theMessage);
254     onFeatureUpdatedMsg(anUpdateMsg);
255     return;
256   }
257
258   if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_OBJECT_DELETED)) {
259     const ModelAPI_ObjectDeletedMessage* aDelMsg =
260         dynamic_cast<const ModelAPI_ObjectDeletedMessage*>(theMessage);
261     onObjectDeletedMsg(aDelMsg);
262     return;
263   }
264
265   //An operation passed by message. Start it, process and commit.
266   if (theMessage->eventID() == Events_Loop::loop()->eventByName(EVENT_OPERATION_LAUNCHED)) {
267     const Config_PointerMessage* aPartSetMsg = dynamic_cast<const Config_PointerMessage*>(theMessage);
268     //myPropertyPanel->cleanContent();
269     ModuleBase_Operation* anOperation = (ModuleBase_Operation*)aPartSetMsg->pointer();
270
271     if (myOperationMgr->startOperation(anOperation)) {
272       myPropertyPanel->updateContentWidget(anOperation->feature());
273       if (!anOperation->getDescription()->hasXmlRepresentation()) {
274         anOperation->commit();
275         updateCommandStatus();
276       }
277     }
278     return;
279   }
280   //Show error dialog if error message received.
281   const Events_Error* anAppError = dynamic_cast<const Events_Error*>(theMessage);
282   if (anAppError) {
283     emit errorOccurred(QString::fromLatin1(anAppError->description()));
284   }
285 }
286
287 //******************************************************
288 void XGUI_Workshop::onFeatureUpdatedMsg(const ModelAPI_ObjectUpdatedMessage* theMsg)
289 {
290   std::set<ObjectPtr> aFeatures = theMsg->objects();
291   if (myOperationMgr->hasOperation())
292   {
293     FeaturePtr aCurrentFeature = myOperationMgr->currentOperation()->feature();
294     std::set<ObjectPtr>::const_iterator aIt;
295     for (aIt = aFeatures.begin(); aIt != aFeatures.end(); ++aIt) {
296       ObjectPtr aNewFeature = (*aIt);
297       if(aNewFeature == aCurrentFeature) {
298         myPropertyPanel->updateContentWidget(aCurrentFeature);
299         break;
300       } 
301     }
302   }
303 }
304
305 //******************************************************
306 void XGUI_Workshop::onFeatureRedisplayMsg(const ModelAPI_ObjectUpdatedMessage* theMsg)
307 {
308   std::set<ObjectPtr> aObjects = theMsg->objects();
309   std::set<ObjectPtr>::const_iterator aIt;
310   for (aIt = aObjects.begin(); aIt != aObjects.end(); ++aIt) {
311     ObjectPtr aObj = (*aIt);
312     if (!aObj->data() )
313       myDisplayer->erase(aObj, false);
314     else {
315       if (myDisplayer->isVisible(aObj)) // TODO VSV: Correction sketch drawing
316         myDisplayer->display(aObj, false); // In order to update presentation
317     }
318   }
319   myDisplayer->updateViewer();
320 }
321
322 //******************************************************
323 void XGUI_Workshop::onFeatureCreatedMsg(const ModelAPI_ObjectUpdatedMessage* theMsg)
324 {
325   std::set<ObjectPtr> aObjects = theMsg->objects();
326
327   std::set<ObjectPtr>::const_iterator aIt;
328   bool aHasPart = false;
329   bool isDisplayed = false;
330   for (aIt = aObjects.begin(); aIt != aObjects.end(); ++aIt) {
331     ResultPartPtr aPart = boost::dynamic_pointer_cast<ModelAPI_ResultPart>(*aIt);
332     if (aPart) {
333       aHasPart = true;
334     } else {
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 }
1002
1003
1004 //**************************************************************
1005 void XGUI_Workshop::registerValidators() const
1006 {
1007   PluginManagerPtr aMgr = ModelAPI_PluginManager::get();
1008   ModelAPI_ValidatorsFactory* aFactory = aMgr->validators();
1009
1010   aFactory->registerValidator("ModuleBase_ResulPointValidator", new ModuleBase_ResulPointValidator);
1011   aFactory->registerValidator("ModuleBase_ResulLineValidator", new ModuleBase_ResulLineValidator);
1012   aFactory->registerValidator("ModuleBase_ResulArcValidator", new ModuleBase_ResulArcValidator);
1013 }