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