Salome HOME
bf5be2e6b4813e8e45ad7488ba91ae32a69ffa0f
[modules/shaper.git] / src / XGUI / XGUI_Workshop.cpp
1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D -->
2
3 //#include "XGUI_Constants.h"
4 #include "XGUI_Tools.h"
5 #include "XGUI_Workshop.h"
6 #include "XGUI_SelectionMgr.h"
7 #include "XGUI_Selection.h"
8 #include "XGUI_ObjectsBrowser.h"
9 #include "XGUI_Displayer.h"
10 #include "XGUI_OperationMgr.h"
11 #include "XGUI_SalomeConnector.h"
12 #include "XGUI_ActionsMgr.h"
13 #include "XGUI_ErrorDialog.h"
14 #include "XGUI_ColorDialog.h"
15 #include "XGUI_ViewerProxy.h"
16 #include "XGUI_PropertyPanel.h"
17 #include "XGUI_ContextMenuMgr.h"
18 #include "XGUI_ModuleConnector.h"
19 #include "XGUI_WorkshopListener.h"
20
21 #include <XGUI_QtEvents.h>
22 #include <XGUI_HistoryMenu.h>
23 #include <XGUI_CustomPrs.h>
24
25 #include <AppElements_Workbench.h>
26 #include <AppElements_Viewer.h>
27 #include <AppElements_Command.h>
28 #include <AppElements_MainMenu.h>
29 #include <AppElements_MainWindow.h>
30 #include <AppElements_MenuGroupPanel.h>
31 #include <AppElements_Button.h>
32
33 #include <ModuleBase_IModule.h>
34 #include <ModuleBase_Preferences.h>
35
36 #include <ModelAPI_Events.h>
37 #include <ModelAPI_Session.h>
38 #include <ModelAPI_Feature.h>
39 #include <ModelAPI_Data.h>
40 #include <ModelAPI_AttributeDocRef.h>
41 #include <ModelAPI_Object.h>
42 #include <ModelAPI_Validator.h>
43 #include <ModelAPI_ResultGroup.h>
44 #include <ModelAPI_ResultConstruction.h>
45 #include <ModelAPI_ResultBody.h>
46 #include <ModelAPI_AttributeIntArray.h>
47 #include <ModelAPI_ResultParameter.h>
48
49 //#include <PartSetPlugin_Part.h>
50
51 #include <Events_Loop.h>
52 #include <Events_Error.h>
53 #include <Events_LongOp.h>
54
55 #include <ModuleBase_Operation.h>
56 #include <ModuleBase_OperationDescription.h>
57 #include <ModuleBase_SelectionValidator.h>
58 #include <ModuleBase_WidgetFactory.h>
59 #include <ModuleBase_Tools.h>
60 #include <ModuleBase_IViewer.h>
61 #include <ModuleBase_FilterFactory.h>
62 #include <ModuleBase_PageBase.h>
63 #include <ModuleBase_Tools.h>
64
65 #include <Config_Common.h>
66 #include <Config_FeatureMessage.h>
67 #include <Config_PointerMessage.h>
68 #include <Config_ModuleReader.h>
69 #include <Config_PropManager.h>
70 #include <Config_SelectionFilterMessage.h>
71
72 #include <SUIT_ResourceMgr.h>
73
74 #include <QApplication>
75 #include <QFileDialog>
76 #include <QMessageBox>
77 #include <QMdiSubWindow>
78 #include <QPushButton>
79 #include <QDockWidget>
80 #include <QLayout>
81 #include <QThread>
82 #include <QObject>
83 #include <QMenu>
84 #include <QToolButton>
85 #include <QAction>
86 #include <QDesktopWidget>
87
88 #ifdef _DEBUG
89 #include <QDebug>
90 #include <iostream>
91 #endif
92
93 #ifdef WIN32
94 #include <windows.h>
95 #else
96 #include <dlfcn.h>
97 #endif
98
99 //#define DEBUG_DELETE
100
101 XGUI_Workshop::XGUI_Workshop(XGUI_SalomeConnector* theConnector)
102     : QObject(),
103       myCurrentDir(QString()),
104       myModule(NULL),
105       mySalomeConnector(theConnector),
106       myPropertyPanel(0),
107       myObjectBrowser(0),
108       myDisplayer(0)
109 {
110   myMainWindow = mySalomeConnector ? 0 : new AppElements_MainWindow();
111
112   if (myMainWindow) {
113     SUIT_ResourceMgr* aResMgr = ModuleBase_Preferences::resourceMgr();
114     bool aCloc = aResMgr->booleanValue("language", "locale", true);
115     if (aCloc)
116       QLocale::setDefault( QLocale::c() );
117     else 
118       QLocale::setDefault( QLocale::system() );
119   }
120
121   myDisplayer = new XGUI_Displayer(this);
122
123   mySelector = new XGUI_SelectionMgr(this);
124   //connect(mySelector, SIGNAL(selectionChanged()), this, SLOT(updateModuleCommands()));
125
126   myOperationMgr = new XGUI_OperationMgr(this, 0);
127   myActionsMgr = new XGUI_ActionsMgr(this);
128   myErrorDlg = new XGUI_ErrorDialog(QApplication::desktop());
129   myContextMenuMgr = new XGUI_ContextMenuMgr(this);
130   connect(myContextMenuMgr, SIGNAL(actionTriggered(const QString&, bool)), this,
131           SLOT(onContextMenuCommand(const QString&, bool)));
132
133   myViewerProxy = new XGUI_ViewerProxy(this);
134   connect(myViewerProxy, SIGNAL(selectionChanged()),
135           myActionsMgr,  SLOT(updateOnViewSelection()));
136
137   myModuleConnector = new XGUI_ModuleConnector(this);
138
139   ModuleBase_IWorkshop* aWorkshop = moduleConnector();
140   myOperationMgr->setWorkshop(aWorkshop);
141
142   myEventsListener = new XGUI_WorkshopListener(aWorkshop);
143
144   connect(myOperationMgr, SIGNAL(operationStarted(ModuleBase_Operation*)), 
145           SLOT(onOperationStarted(ModuleBase_Operation*)));
146   connect(myOperationMgr, SIGNAL(operationResumed(ModuleBase_Operation*)),
147           SLOT(onOperationResumed(ModuleBase_Operation*)));
148   connect(myOperationMgr, SIGNAL(operationStopped(ModuleBase_Operation*)),
149           SLOT(onOperationStopped(ModuleBase_Operation*)));
150   connect(myOperationMgr, SIGNAL(operationCommitted(ModuleBase_Operation*)), 
151           SLOT(onOperationCommitted(ModuleBase_Operation*)));
152   connect(myOperationMgr, SIGNAL(operationAborted(ModuleBase_Operation*)), 
153           SLOT(onOperationAborted(ModuleBase_Operation*)));
154   if (myMainWindow)
155     connect(myMainWindow, SIGNAL(exitKeySequence()), SLOT(onExit()));
156   connect(this, SIGNAL(errorOccurred(const QString&)), myErrorDlg, SLOT(addError(const QString&)));
157   connect(myEventsListener, SIGNAL(errorOccurred(const QString&)),
158           myErrorDlg, SLOT(addError(const QString&)));
159
160   //Config_PropManager::registerProp("Visualization", "object_default_color", "Object color",
161   //                                 Config_Prop::Color, "225,225,225");
162
163   Config_PropManager::registerProp("Visualization", "result_body_color", "Body color",
164                                    Config_Prop::Color, ModelAPI_ResultBody::DEFAULT_COLOR());
165   Config_PropManager::registerProp("Visualization", "result_group_color", "Group color",
166                                    Config_Prop::Color, ModelAPI_ResultGroup::DEFAULT_COLOR());
167   Config_PropManager::registerProp("Visualization", "result_construction_color", "Construction color",
168                                    Config_Prop::Color, ModelAPI_ResultConstruction::DEFAULT_COLOR());
169   Config_PropManager::registerProp("Visualization", "result_part_color", "Part color",
170                                    Config_Prop::Color, ModelAPI_ResultPart::DEFAULT_COLOR());
171 }
172
173 //******************************************************
174 XGUI_Workshop::~XGUI_Workshop(void)
175 {
176   delete myDisplayer;
177 }
178
179 //******************************************************
180 void XGUI_Workshop::startApplication()
181 {
182   initMenu();
183
184   Config_PropManager::registerProp("Plugins", "default_path", "Default Path",
185                                    Config_Prop::Directory, "");
186
187   //Initialize event listening
188   myEventsListener->initializeEventListening();
189
190   registerValidators();
191
192   // Calling of  loadCustomProps before activating module is required
193   // by Config_PropManger to restore user-defined path to plugins
194   ModuleBase_Preferences::loadCustomProps();
195   createModule();
196   if (myMainWindow) {
197     myMainWindow->show();
198     updateCommandStatus();
199   }
200   
201   onNew();
202
203   emit applicationStarted();
204 }
205
206 void XGUI_Workshop::activateModule()
207 {
208   myModule->activateSelectionFilters();
209
210   connect(myDisplayer, SIGNAL(objectDisplayed(ObjectPtr, AISObjectPtr)),
211     myModule, SLOT(onObjectDisplayed(ObjectPtr, AISObjectPtr)));
212   connect(myDisplayer, SIGNAL(beforeObjectErase(ObjectPtr, AISObjectPtr)),
213     myModule, SLOT(onBeforeObjectErase(ObjectPtr, AISObjectPtr)));
214
215   myActionsMgr->update();
216
217 }
218
219 void XGUI_Workshop::deactivateModule()
220 {
221   myModule->deactivateSelectionFilters();
222
223   disconnect(myDisplayer, SIGNAL(objectDisplayed(ObjectPtr, AISObjectPtr)),
224     myModule, SLOT(onObjectDisplayed(ObjectPtr, AISObjectPtr)));
225   disconnect(myDisplayer, SIGNAL(beforeObjectErase(ObjectPtr, AISObjectPtr)),
226     myModule, SLOT(onBeforeObjectErase(ObjectPtr, AISObjectPtr)));
227 }
228
229 //******************************************************
230 void XGUI_Workshop::initMenu()
231 {
232   myContextMenuMgr->createActions();
233
234   if (isSalomeMode()) {
235     // Create only Undo, Redo commands
236     QAction* aAction = salomeConnector()->addDesktopCommand("UNDO_CMD", tr("Undo"),
237                                                          tr("Undo last command"),
238                                                          QIcon(":pictures/undo.png"),
239                                                          QKeySequence::Undo, false, "MEN_DESK_EDIT");
240     connect(aAction, SIGNAL(triggered(bool)), this, SLOT(onUndo()));
241     addHistoryMenu(aAction, SIGNAL(updateUndoHistory(const QList<ActionInfo>&)), SLOT(onUndo(int)));
242
243     aAction = salomeConnector()->addDesktopCommand("REDO_CMD", tr("Redo"), tr("Redo last command"),
244                                                 QIcon(":pictures/redo.png"), QKeySequence::Redo,
245                                                 false, "MEN_DESK_EDIT");
246     connect(aAction, SIGNAL(triggered(bool)), this, SLOT(onRedo()));
247     addHistoryMenu(aAction, SIGNAL(updateRedoHistory(const QList<ActionInfo>&)), SLOT(onRedo(int)));
248
249     salomeConnector()->addDesktopMenuSeparator("MEN_DESK_EDIT");
250     aAction = salomeConnector()->addDesktopCommand("REBUILD_CMD", tr("Rebuild"), tr("Rebuild data objects"),
251                                                 QIcon(":pictures/rebuild.png"), QKeySequence(),
252                                                 false, "MEN_DESK_EDIT");
253     connect(aAction, SIGNAL(triggered(bool)), this, SLOT(onRebuild()));
254     salomeConnector()->addDesktopMenuSeparator("MEN_DESK_EDIT");
255
256     aAction = salomeConnector()->addDesktopCommand("SAVEAS_CMD", tr("Export NewGeom..."), tr("Export the current document into a NewGeom file"),
257                                                 QIcon(), QKeySequence(),
258                                                 false, "MEN_DESK_FILE");
259     connect(aAction, SIGNAL(triggered(bool)), this, SLOT(onSaveAs()));
260
261     aAction = salomeConnector()->addDesktopCommand("OPEN_CMD", tr("Import NewGeom..."), tr("Import a NewGeom file"),
262                                                 QIcon(), QKeySequence(),
263                                                 false, "MEN_DESK_FILE");
264     connect(aAction, SIGNAL(triggered(bool)), this, SLOT(onOpen()));
265     salomeConnector()->addDesktopMenuSeparator("MEN_DESK_FILE");
266
267     return;
268   }
269   // File commands group
270   AppElements_MenuGroupPanel* aGroup = myMainWindow->menuObject()->generalPage();
271
272   AppElements_Command* aCommand;
273
274   aCommand = aGroup->addFeature("SAVE_CMD", tr("Save"), tr("Save the document"),
275                                 QIcon(":pictures/save.png"), QKeySequence::Save);
276   aCommand->connectTo(this, SLOT(onSave()));
277   //aCommand->disable();
278
279   QString aUndoId = "UNDO_CMD";
280   aCommand = aGroup->addFeature(aUndoId, tr("Undo"), tr("Undo last command"),
281                                 QIcon(":pictures/undo.png"), QKeySequence::Undo);
282   aCommand->connectTo(this, SLOT(onUndo()));
283   AppElements_Button* aUndoButton = qobject_cast<AppElements_Button*>(aGroup->widget(aUndoId));
284   addHistoryMenu(aUndoButton,
285                  SIGNAL(updateUndoHistory(const QList<ActionInfo>&)),
286                  SLOT(onUndo(int)));
287
288   QString aRedoId = "REDO_CMD";
289   aCommand = aGroup->addFeature(aRedoId, tr("Redo"), tr("Redo last command"),
290                                 QIcon(":pictures/redo.png"), QKeySequence::Redo);
291   aCommand->connectTo(this, SLOT(onRedo()));
292   AppElements_Button* aRedoButton = qobject_cast<AppElements_Button*>(aGroup->widget(aRedoId));
293   addHistoryMenu(aRedoButton,
294                  SIGNAL(updateRedoHistory(const QList<ActionInfo>&)),
295                  SLOT(onRedo(int)));
296
297   aCommand = aGroup->addFeature("REBUILD_CMD", tr("Rebuild"), tr("Rebuild data objects"),
298     QIcon(":pictures/rebuild.png"), QKeySequence());
299   aCommand->connectTo(this, SLOT(onRebuild()));
300
301   aCommand = aGroup->addFeature("SAVEAS_CMD", tr("Save as..."), tr("Save the document into a file"),
302                                 QIcon(":pictures/save.png"), QKeySequence());
303   aCommand->connectTo(this, SLOT(onSaveAs()));
304   //aCommand->disable();
305
306   aCommand = aGroup->addFeature("OPEN_CMD", tr("Open..."), tr("Open a new document"),
307                                 QIcon(":pictures/open.png"), QKeySequence::Open);
308   aCommand->connectTo(this, SLOT(onOpen()));
309
310   //aCommand = aGroup->addFeature("NEW_CMD", tr("New"), tr("Create a new document"),
311   //                              QIcon(":pictures/new.png"), QKeySequence::New);
312   //aCommand->connectTo(this, SLOT(onNew()));
313
314   aCommand = aGroup->addFeature("PREF_CMD", tr("Preferences"), tr("Edit preferences"),
315                                 QIcon(":pictures/preferences.png"), QKeySequence::Preferences);
316   aCommand->connectTo(this, SLOT(onPreferences()));
317
318   aCommand = aGroup->addFeature("EXIT_CMD", tr("Exit"), tr("Exit application"),
319                                 QIcon(":pictures/close.png"), QKeySequence::Close);
320   aCommand->connectTo(this, SLOT(onExit()));
321   //FIXME: SBH's test action. Can be used for some GUI tests.
322 //  #ifdef _DEBUG
323 //    aCommand = aGroup->addFeature("TEST_CMD", "Test!", "Private debug button",
324 //                                  QIcon(":pictures/close.png"), QKeySequence(), true);
325 //    aCommand->connectTo(myMainWindow, SLOT(dockPythonConsole()));
326 //  #endif
327 }
328
329 //******************************************************
330 AppElements_Workbench* XGUI_Workshop::addWorkbench(const QString& theName)
331 {
332   AppElements_MainMenu* aMenuBar = myMainWindow->menuObject();
333   return aMenuBar->addWorkbench(theName);
334 }
335
336 //******************************************************
337 QMainWindow* XGUI_Workshop::desktop() const
338 {
339   return isSalomeMode() ? salomeConnector()->desktop() : myMainWindow;
340 }
341
342 //******************************************************
343 void XGUI_Workshop::onStartWaiting()
344 {
345   if (Events_LongOp::isPerformed()) {
346     QApplication::setOverrideCursor(QCursor(Qt::WaitCursor));
347   }
348 }
349
350
351 //******************************************************
352 void XGUI_Workshop::deactivateActiveObject(const ObjectPtr& theObject, const bool theUpdateViewer)
353 {
354   if (!myModule->canActivateSelection(theObject)) {
355     if (myDisplayer->isActive(theObject))
356       myDisplayer->deactivate(theObject, theUpdateViewer);
357   }
358 }
359
360 //******************************************************
361 void XGUI_Workshop::onOperationStarted(ModuleBase_Operation* theOperation)
362 {
363   setNestedFeatures(theOperation);
364
365   if (theOperation->getDescription()->hasXmlRepresentation()) {  //!< No need for property panel
366     connectWithOperation(theOperation);
367     setPropertyPanel(theOperation);
368   }
369   updateCommandStatus();
370
371   myModule->onOperationStarted(theOperation);
372
373   // the objects of the current operation should be deactivated
374   QObjectPtrList anObjects;
375   FeaturePtr aFeature = theOperation->feature();
376   anObjects.append(aFeature);
377   std::list<ResultPtr> aResults = aFeature->results();
378   std::list<ResultPtr>::const_iterator aIt;
379   for (aIt = aResults.begin(); aIt != aResults.end(); ++aIt) {
380     anObjects.append(*aIt);
381   }
382   QObjectPtrList::const_iterator anIt = anObjects.begin(), aLast = anObjects.end();
383   for (; anIt != aLast; anIt++)
384     deactivateActiveObject(*anIt, false);
385   if (anObjects.size() > 0)
386     myDisplayer->updateViewer();
387 }
388
389 //******************************************************
390 void XGUI_Workshop::onOperationResumed(ModuleBase_Operation* theOperation)
391 {
392   setNestedFeatures(theOperation);
393
394   if (theOperation->getDescription()->hasXmlRepresentation()) {  //!< No need for property panel
395     // connectWithOperation(theOperation); already connected
396     setPropertyPanel(theOperation);
397   }
398   updateCommandStatus();
399
400   myModule->onOperationResumed(theOperation);
401 }
402
403
404 //******************************************************
405 void XGUI_Workshop::onOperationStopped(ModuleBase_Operation* theOperation)
406 {
407   ModuleBase_ISelection* aSel = mySelector->selection();
408   QObjectPtrList aObj = aSel->selectedPresentations();
409   //!< No need for property panel
410   updateCommandStatus();
411   hidePropertyPanel();
412   myPropertyPanel->cleanContent();
413
414   myModule->onOperationStopped(theOperation);
415
416   // the deactivated objects of the current operation should be activated back.
417   // They were deactivated on operation start or an object redisplay
418   QObjectPtrList anObjects;
419   FeaturePtr aFeature = theOperation->feature();
420   if (myDisplayer->isVisible(aFeature) && !myDisplayer->isActive(aFeature))
421     anObjects.append(aFeature);
422   std::list<ResultPtr> aResults = aFeature->results();
423   std::list<ResultPtr>::const_iterator aIt;
424   for (aIt = aResults.begin(); aIt != aResults.end(); ++aIt) {
425     ResultPtr anObject = *aIt;
426     if (myDisplayer->isVisible(anObject) && !myDisplayer->isActive(anObject)) {
427       anObjects.append(anObject);
428     }
429   }
430   QIntList aModes;
431   module()->activeSelectionModes(aModes);
432   myDisplayer->activateObjects(aModes, anObjects);
433 }
434
435
436 void XGUI_Workshop::onOperationCommitted(ModuleBase_Operation* theOperation)
437 {
438   myModule->onOperationCommitted(theOperation);
439 }
440
441 void XGUI_Workshop::onOperationAborted(ModuleBase_Operation* theOperation)
442 {
443   myModule->onOperationAborted(theOperation);
444 }
445
446 void XGUI_Workshop::setNestedFeatures(ModuleBase_Operation* theOperation)
447 {
448   if (this->isSalomeMode()) 
449     theOperation->setNestedFeatures(mySalomeConnector->nestedActions(theOperation->id()));
450   else 
451     theOperation->setNestedFeatures(myActionsMgr->nestedCommands(theOperation->id()));
452 }
453
454 void XGUI_Workshop::setPropertyPanel(ModuleBase_Operation* theOperation)
455 {
456   showPropertyPanel();
457   QString aXmlRepr = theOperation->getDescription()->xmlRepresentation();
458   ModuleBase_WidgetFactory aFactory = ModuleBase_WidgetFactory(aXmlRepr.toStdString(),
459                                                                 myModuleConnector);
460
461   myPropertyPanel->cleanContent();
462   aFactory.createWidget(myPropertyPanel->contentWidget());
463
464   QList<ModuleBase_ModelWidget*> aWidgets = aFactory.getModelWidgets();
465   foreach (ModuleBase_ModelWidget* aWidget, aWidgets) {
466     bool isStoreValue = !theOperation->isEditOperation() &&
467                         !aWidget->getDefaultValue().empty() &&
468                         !aWidget->isComputedDefault();
469     aWidget->setFeature(theOperation->feature(), isStoreValue);
470     aWidget->enableFocusProcessing();
471   }
472   
473   myPropertyPanel->setModelWidgets(aWidgets);
474   theOperation->setPropertyPanel(myPropertyPanel);
475
476   myModule->propertyPanelDefined(theOperation);
477
478   myPropertyPanel->setWindowTitle(theOperation->getDescription()->description());
479 }
480
481 /*
482  * Makes a signal/slot connections between Property Panel
483  * and given operation. The given operation becomes a
484  * current operation and previous operation if exists
485  */
486 void XGUI_Workshop::connectWithOperation(ModuleBase_Operation* theOperation)
487 {
488   QAction* aCommand = 0;
489   if (isSalomeMode()) {
490     aCommand = salomeConnector()->command(theOperation->getDescription()->operationId());
491   } else {
492     AppElements_MainMenu* aMenu = myMainWindow->menuObject();
493     FeaturePtr aFeature = theOperation->feature();
494     if(aFeature)
495       aCommand = aMenu->feature(QString::fromStdString(aFeature->getKind()));
496   }
497   //Abort operation on uncheck the command
498   if (aCommand) {
499     connect(aCommand, SIGNAL(triggered(bool)), theOperation, SLOT(setRunning(bool)));
500   }
501 }
502
503 /*
504  * Saves document with given name.
505  */
506 void XGUI_Workshop::saveDocument(const QString& theName, std::list<std::string>& theFileNames)
507 {
508   QApplication::restoreOverrideCursor();
509   SessionPtr aMgr = ModelAPI_Session::get();
510   aMgr->save(theName.toLatin1().constData(), theFileNames);
511   QApplication::restoreOverrideCursor();
512 }
513
514 bool XGUI_Workshop::isActiveOperationAborted()
515 {
516   return myOperationMgr->abortAllOperations();
517 }
518
519 //******************************************************
520 void XGUI_Workshop::onExit()
521 {
522   SessionPtr aMgr = ModelAPI_Session::get();
523   if (aMgr->isModified()) {
524     int anAnswer = QMessageBox::question(
525         myMainWindow, tr("Save current file"), tr("The document is modified, save before exit?"),
526         QMessageBox::Save | QMessageBox::Discard | QMessageBox::Cancel, QMessageBox::Cancel);
527     if (anAnswer == QMessageBox::Save) {
528       bool saved = onSave();
529       if (!saved) {
530         return;
531       }
532     } else if (anAnswer == QMessageBox::Cancel) {
533       return;
534     }
535   }
536   qApp->exit();
537 }
538
539 //******************************************************
540 void XGUI_Workshop::onNew()
541 {
542   QApplication::setOverrideCursor(Qt::WaitCursor);
543   if (objectBrowser() == 0) {
544     createDockWidgets();
545     mySelector->connectViewers();
546   }
547   myViewerProxy->connectToViewer();
548   showObjectBrowser();
549   if (!isSalomeMode()) {
550     myMainWindow->showPythonConsole();
551     QMdiSubWindow* aWnd = myMainWindow->viewer()->createView();
552     aWnd->showMaximized();
553     updateCommandStatus();
554   }
555   myContextMenuMgr->connectViewer();
556   QApplication::restoreOverrideCursor();
557 }
558
559 //******************************************************
560 void XGUI_Workshop::onOpen()
561 {
562   if(!isActiveOperationAborted())
563     return;
564   //save current file before close if modified
565   SessionPtr aSession = ModelAPI_Session::get();
566   if (aSession->isModified()) {
567     //TODO(sbh): re-launch the app?
568     int anAnswer = QMessageBox::question(
569         myMainWindow, tr("Save current file"),
570         tr("The document is modified, save before opening another?"),
571         QMessageBox::Save | QMessageBox::Discard | QMessageBox::Cancel, QMessageBox::Cancel);
572     if (anAnswer == QMessageBox::Save) {
573       onSave();
574     } else if (anAnswer == QMessageBox::Cancel) {
575       return;
576     }
577     myCurrentDir = "";
578   }
579
580   //show file dialog, check if readable and open
581   myCurrentDir = QFileDialog::getExistingDirectory(mainWindow(), tr("Select directory"));
582   if (myCurrentDir.isEmpty())
583     return;
584   QFileInfo aFileInfo(myCurrentDir);
585   if (!aFileInfo.exists() || !aFileInfo.isReadable()) {
586     QMessageBox::critical(myMainWindow, tr("Warning"), tr("Unable to open the file."));
587     myCurrentDir = "";
588     return;
589   }
590   QApplication::setOverrideCursor(Qt::WaitCursor);
591   aSession->closeAll();
592   aSession->load(myCurrentDir.toLatin1().constData());
593   myObjectBrowser->rebuildDataTree();
594   //displayAllResults();
595   updateCommandStatus();
596   QApplication::restoreOverrideCursor();
597 }
598
599 //******************************************************
600 bool XGUI_Workshop::onSave()
601 {
602   if(!isActiveOperationAborted())
603     return false;
604   if (myCurrentDir.isEmpty()) {
605     return onSaveAs();
606   }
607   std::list<std::string> aFiles;
608   saveDocument(myCurrentDir, aFiles);
609   updateCommandStatus();
610   if (!isSalomeMode())
611     myMainWindow->setModifiedState(false);
612   return true;
613 }
614
615 //******************************************************
616 bool XGUI_Workshop::onSaveAs()
617 {
618   if(!isActiveOperationAborted())
619     return false;
620   QFileDialog dialog(mainWindow());
621   dialog.setWindowTitle(tr("Select directory to save files..."));
622   dialog.setFileMode(QFileDialog::Directory);
623   dialog.setFilter(tr("Directories (*)"));
624   dialog.setOptions(QFileDialog::HideNameFilterDetails | QFileDialog::ShowDirsOnly);
625   dialog.setViewMode(QFileDialog::Detail);
626
627   if (!dialog.exec()) {
628     return false;
629   }
630
631   QString aTempDir = dialog.selectedFiles().first();
632   QDir aDir(aTempDir);
633   if (aDir.exists() && !aDir.entryInfoList(QDir::NoDotAndDotDot | QDir::AllEntries).isEmpty()) {
634     int answer = QMessageBox::question(
635         myMainWindow,
636         //: Title of the dialog which asks user if he wants to save study in existing non-empty folder
637         tr("Save"),
638         tr("The directory already contains some files, save anyway?"),
639         QMessageBox::Save | QMessageBox::Cancel);
640     if (answer == QMessageBox::Cancel) {
641       return false;
642     }
643   }
644   myCurrentDir = aTempDir;
645   if (!isSalomeMode()) {
646     myMainWindow->setCurrentDir(myCurrentDir, false);
647     myMainWindow->setModifiedState(false);
648   }
649   return onSave();
650 }
651
652 //******************************************************
653 void XGUI_Workshop::onUndo(int theTimes)
654 {
655   objectBrowser()->treeView()->setCurrentIndex(QModelIndex());
656   SessionPtr aMgr = ModelAPI_Session::get();
657   if (aMgr->isOperation())
658     operationMgr()->onAbortOperation();
659   for (int i = 0; i < theTimes; ++i) {
660     aMgr->undo();
661   }
662   updateCommandStatus();
663 }
664
665 //******************************************************
666 void XGUI_Workshop::onRedo(int theTimes)
667 {
668   // the viewer update should be blocked in order to avoid the features blinking. For the created
669   // feature a results are created, the flush of the created signal caused the viewer redisplay for
670   // each created result. After a redisplay signal is flushed. So, the viewer update is blocked until
671   // redo of all possible objects happens
672   bool isUpdateEnabled = myDisplayer->enableUpdateViewer(false);
673
674   objectBrowser()->treeView()->setCurrentIndex(QModelIndex());
675   SessionPtr aMgr = ModelAPI_Session::get();
676   if (aMgr->isOperation())
677     operationMgr()->onAbortOperation();
678   for (int i = 0; i < theTimes; ++i) {
679     aMgr->redo();
680   }
681   updateCommandStatus();
682
683   // unblock the viewer update functionality and make update on purpose
684   myDisplayer->enableUpdateViewer(isUpdateEnabled);
685   myDisplayer->updateViewer();
686 }
687
688 //******************************************************
689 void XGUI_Workshop::onRebuild()
690 {
691   SessionPtr aMgr = ModelAPI_Session::get();
692   bool aWasOperation = aMgr->isOperation(); // keep this value
693   if (!aWasOperation) {
694     aMgr->startOperation("Rebuild");
695   }
696   static const Events_ID aRebuildEvent = Events_Loop::loop()->eventByName("Rebuild");
697   Events_Loop::loop()->send(std::shared_ptr<Events_Message>(
698     new Events_Message(aRebuildEvent, this)));
699   if (!aWasOperation) {
700     aMgr->finishOperation();
701   }
702   updateCommandStatus();
703 }
704
705 //******************************************************
706 void XGUI_Workshop::onPreferences()
707 {
708   ModuleBase_Prefs aModif;
709   ModuleBase_Preferences::editPreferences(aModif);
710   if (aModif.size() > 0) {
711     QString aSection;
712     foreach (ModuleBase_Pref aPref, aModif)
713     {
714       aSection = aPref.first;
715       if (aSection == ModuleBase_Preferences::VIEWER_SECTION) {
716         if (!isSalomeMode())
717           myMainWindow->viewer()->updateFromResources();
718       } else if (aSection == ModuleBase_Preferences::MENU_SECTION) {
719         if (!isSalomeMode())
720           myMainWindow->menuObject()->updateFromResources();
721       }
722     }
723   }
724 }
725
726 //******************************************************
727 ModuleBase_IModule* XGUI_Workshop::loadModule(const QString& theModule)
728 {
729   QString libName = QString::fromStdString(library(theModule.toStdString()));
730   if (libName.isEmpty()) {
731     qWarning(qPrintable(tr("Information about module \"%1\" doesn't exist.").arg(theModule)));
732     return 0;
733   }
734
735   QString err;
736   CREATE_FUNC crtInst = 0;
737
738 #ifdef WIN32
739   HINSTANCE modLib = ::LoadLibrary((LPTSTR) qPrintable(libName));
740   if (!modLib) {
741     LPVOID lpMsgBuf;
742     ::FormatMessage(
743         FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS,
744         0, ::GetLastError(), 0, (LPTSTR) & lpMsgBuf, 0, 0);
745     QString aMsg((char*) &lpMsgBuf);
746     err = QString("Failed to load  %1. %2").arg(libName).arg(aMsg);
747     ::LocalFree(lpMsgBuf);
748   } else {
749     crtInst = (CREATE_FUNC) ::GetProcAddress(modLib, CREATE_MODULE);
750     if (!crtInst) {
751       LPVOID lpMsgBuf;
752       ::FormatMessage(
753           FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM
754               | FORMAT_MESSAGE_IGNORE_INSERTS,
755           0, ::GetLastError(), 0, (LPTSTR) & lpMsgBuf, 0, 0);
756       QString aMsg((char*) &lpMsgBuf);
757       err = QString("Failed to find  %1 function. %2").arg( CREATE_MODULE).arg(aMsg);
758       ::LocalFree(lpMsgBuf);
759     }
760   }
761 #else
762   void* modLib = dlopen( libName.toLatin1(), RTLD_LAZY | RTLD_GLOBAL );
763   if ( !modLib ) {
764     err = QString( "Can not load library %1. %2" ).arg( libName ).arg( dlerror() );
765   } else {
766     crtInst = (CREATE_FUNC)dlsym( modLib, CREATE_MODULE );
767     if ( !crtInst ) {
768       err = QString( "Failed to find function %1. %2" ).arg( CREATE_MODULE ).arg( dlerror() );
769     }
770   }
771 #endif
772
773   ModuleBase_IModule* aModule = crtInst ? crtInst(myModuleConnector) : 0;
774
775   if (!err.isEmpty()) {
776     if (mainWindow()) {
777       Events_Error::send(err.toStdString());
778     } else {
779       qWarning(qPrintable(err));
780     }
781   }
782   return aModule;
783 }
784
785 //******************************************************
786 bool XGUI_Workshop::createModule()
787 {
788   Config_ModuleReader aModuleReader;
789   QString moduleName = QString::fromStdString(aModuleReader.getModuleName());
790   myModule = loadModule(moduleName);
791   if (!myModule)
792     return false;
793
794   //connect(myDisplayer, SIGNAL(objectDisplayed(ObjectPtr, AISObjectPtr)),
795   //  myModule, SLOT(onObjectDisplayed(ObjectPtr, AISObjectPtr)));
796   //connect(myDisplayer, SIGNAL(beforeObjectErase(ObjectPtr, AISObjectPtr)),
797   //  myModule, SLOT(onBeforeObjectErase(ObjectPtr, AISObjectPtr)));
798
799   myModule->createFeatures();
800   //myActionsMgr->update();
801   return true;
802 }
803
804 //******************************************************
805 void XGUI_Workshop::updateCommandStatus()
806 {
807   QList<QAction*> aCommands;
808   if (isSalomeMode()) {  // update commands in SALOME mode
809     aCommands = salomeConnector()->commandList();
810   } else {
811     AppElements_MainMenu* aMenuBar = myMainWindow->menuObject();
812     foreach (AppElements_Command* aCmd, aMenuBar->features())
813       aCommands.append(aCmd);
814   }
815   SessionPtr aMgr = ModelAPI_Session::get();
816   if (aMgr->hasModuleDocument()) {
817     foreach(QAction* aCmd, aCommands) {
818       QString aId = aCmd->data().toString();
819       if (aId == "UNDO_CMD")
820         aCmd->setEnabled(myModule->canUndo());
821       else if (aId == "REDO_CMD")
822         aCmd->setEnabled(myModule->canRedo());
823       else
824         // Enable all commands
825         aCmd->setEnabled(true);
826     }
827     updateHistory();
828   } else {
829     foreach(QAction* aCmd, aCommands) {
830       QString aId = aCmd->data().toString();
831       if (aId == "NEW_CMD")
832         aCmd->setEnabled(true);
833       else if (aId == "EXIT_CMD")
834         aCmd->setEnabled(true);
835       else
836         aCmd->setEnabled(false);
837     }
838   }
839   myActionsMgr->update();
840   emit commandStatusUpdated();
841 }
842
843 void XGUI_Workshop::updateHistory()
844 {
845   std::list<std::string> aUndoList = ModelAPI_Session::get()->undoList();
846   QList<ActionInfo> aUndoRes = processHistoryList(aUndoList);
847   emit updateUndoHistory(aUndoRes);
848
849   std::list<std::string> aRedoList = ModelAPI_Session::get()->redoList();
850   QList<ActionInfo> aRedoRes = processHistoryList(aRedoList);
851   emit updateRedoHistory(aRedoRes);
852 }
853
854 //******************************************************
855 QDockWidget* XGUI_Workshop::createObjectBrowser(QWidget* theParent)
856 {
857   QDockWidget* aObjDock = new QDockWidget(theParent);
858   aObjDock->setAllowedAreas(Qt::LeftDockWidgetArea | Qt::RightDockWidgetArea | Qt::BottomDockWidgetArea);
859   aObjDock->setWindowTitle(tr("Object browser"));
860   aObjDock->setStyleSheet(
861       "::title { position: relative; padding-left: 5px; text-align: left center }");
862   myObjectBrowser = new XGUI_ObjectsBrowser(aObjDock);
863   myObjectBrowser->setDataModel(myModule->dataModel());
864   myModule->customizeObjectBrowser(myObjectBrowser);
865   aObjDock->setWidget(myObjectBrowser);
866
867   myContextMenuMgr->connectObjectBrowser();
868   return aObjDock;
869 }
870
871 //******************************************************
872 /*
873  * Creates dock widgets, places them in corresponding area
874  * and tabifies if necessary.
875  */
876 void XGUI_Workshop::createDockWidgets()
877 {
878   QMainWindow* aDesktop = isSalomeMode() ? salomeConnector()->desktop() : myMainWindow;
879   QDockWidget* aObjDock = createObjectBrowser(aDesktop);
880   aDesktop->addDockWidget(Qt::LeftDockWidgetArea, aObjDock);
881   myPropertyPanel = new XGUI_PropertyPanel(aDesktop);
882   myPropertyPanel->setupActions(myActionsMgr);
883   myPropertyPanel->setAllowedAreas(Qt::LeftDockWidgetArea | Qt::RightDockWidgetArea | Qt::BottomDockWidgetArea);
884   aDesktop->addDockWidget(Qt::LeftDockWidgetArea, myPropertyPanel);
885   hidePropertyPanel();  ///<! Invisible by default
886   hideObjectBrowser();
887   aDesktop->tabifyDockWidget(aObjDock, myPropertyPanel);
888   myPropertyPanel->installEventFilter(myOperationMgr);
889
890   QAction* aOkAct = myActionsMgr->operationStateAction(XGUI_ActionsMgr::Accept);
891   connect(aOkAct, SIGNAL(triggered()), myOperationMgr, SLOT(onCommitOperation()));
892   QAction* aCancelAct = myActionsMgr->operationStateAction(XGUI_ActionsMgr::Abort);
893   connect(aCancelAct, SIGNAL(triggered()), myOperationMgr, SLOT(onAbortOperation()));
894   connect(myPropertyPanel, SIGNAL(noMoreWidgets()), myModule, SLOT(onNoMoreWidgets()));
895   connect(myPropertyPanel, SIGNAL(keyReleased(QKeyEvent*)),
896           myOperationMgr,  SLOT(onKeyReleased(QKeyEvent*)));
897   connect(myOperationMgr,  SIGNAL(validationStateChanged(bool)),
898           aOkAct,          SLOT(setEnabled(bool)));
899   QAction* aAcceptAllAct = myActionsMgr->operationStateAction(XGUI_ActionsMgr::AcceptAll);
900   connect(myOperationMgr,  SIGNAL(nestedStateChanged(bool)),
901           aAcceptAllAct,   SLOT(setEnabled(bool)));
902
903 }
904
905 //******************************************************
906 void XGUI_Workshop::showPropertyPanel()
907 {
908   QAction* aViewAct = myPropertyPanel->toggleViewAction();
909   ///<! Restore ability to close panel from the window's menu
910   aViewAct->setEnabled(true);
911   myPropertyPanel->show();
912   myPropertyPanel->raise();
913 }
914
915 //******************************************************
916 void XGUI_Workshop::hidePropertyPanel()
917 {
918   QAction* aViewAct = myPropertyPanel->toggleViewAction();
919   ///<! Do not allow to show empty property panel
920   aViewAct->setEnabled(false);
921   myPropertyPanel->hide();
922 }
923
924 //******************************************************
925 void XGUI_Workshop::showObjectBrowser()
926 {
927   myObjectBrowser->parentWidget()->show();
928 }
929
930 //******************************************************
931 void XGUI_Workshop::hideObjectBrowser()
932 {
933   myObjectBrowser->parentWidget()->hide();
934 }
935
936 //******************************************************
937 //void XGUI_Workshop::onFeatureTriggered()
938 //{
939 //  QAction* aCmd = dynamic_cast<QAction*>(sender());
940 //  if (aCmd) {
941 //    QString aId = salomeConnector()->commandId(aCmd);
942 //    if (!aId.isNull())
943 //      myModule->launchOperation(aId);
944 //  }
945 //}
946
947 //******************************************************
948 void XGUI_Workshop::salomeViewerSelectionChanged()
949 {
950   emit salomeViewerSelection();
951 }
952
953 //**************************************************************
954 ModuleBase_IViewer* XGUI_Workshop::salomeViewer() const
955 {
956   return mySalomeConnector->viewer();
957 }
958
959 //**************************************************************
960 void XGUI_Workshop::onContextMenuCommand(const QString& theId, bool isChecked)
961 {
962   QObjectPtrList aObjects = mySelector->selection()->selectedObjects();
963   if (theId == "DELETE_CMD")
964     deleteObjects();
965   if (theId == "MOVE_CMD")
966     moveObjects();
967   else if (theId == "COLOR_CMD")
968     changeColor(aObjects);
969   else if (theId == "SHOW_CMD")
970     showObjects(aObjects, true);
971   else if (theId == "HIDE_CMD")
972     showObjects(aObjects, false);
973   else if (theId == "SHOW_ONLY_CMD")
974     showOnlyObjects(aObjects);
975   else if (theId == "SHADING_CMD")
976     setDisplayMode(aObjects, XGUI_Displayer::Shading);
977   else if (theId == "WIREFRAME_CMD")
978     setDisplayMode(aObjects, XGUI_Displayer::Wireframe);
979   else if (theId == "HIDEALL_CMD") {
980     QObjectPtrList aList = myDisplayer->displayedObjects();
981     foreach (ObjectPtr aObj, aList)
982       aObj->setDisplayed(false);
983     Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_TO_REDISPLAY));
984   }
985 }
986
987 //**************************************************************
988 void XGUI_Workshop::deleteObjects()
989 {
990   ModuleBase_IModule* aModule = module();
991   // 1. allow the module to delete objects, do nothing if it has succeed
992   if (aModule->deleteObjects()) {
993     updateCommandStatus();
994     return;
995   }
996
997   if (!isActiveOperationAborted())
998     return;
999   QObjectPtrList anObjects = mySelector->selection()->selectedObjects();
1000   bool hasResult = false;
1001   bool hasFeature = false;
1002   bool hasParameter = false;
1003   ModuleBase_Tools::checkObjects(anObjects, hasResult, hasFeature, hasParameter);
1004   if (!(hasFeature || hasParameter))
1005     return;
1006
1007   // 1. start operation
1008   QString aDescription = contextMenuMgr()->action("DELETE_CMD")->text();
1009   aDescription += tr(" %1");
1010   QStringList aObjectNames;
1011   foreach (ObjectPtr aObj, anObjects) {
1012     if (!aObj->data()->isValid())
1013       continue;
1014     aObjectNames << QString::fromStdString(aObj->data()->name());
1015   }
1016   aDescription = aDescription.arg(aObjectNames.join(", "));
1017
1018   SessionPtr aMgr = ModelAPI_Session::get();
1019   aMgr->startOperation(aDescription.toStdString());
1020   // 2. close the documents of the removed parts if the result part is in a list of selected objects
1021   // this is performed in the RemoveFeature of Part object.
1022   /*foreach (ObjectPtr aObj, anObjects)
1023   {
1024     ResultPartPtr aPart = std::dynamic_pointer_cast<ModelAPI_ResultPart>(aObj);
1025     if (aPart) {
1026       DocumentPtr aDoc = aObj->document();
1027       if (aDoc == aMgr->activeDocument()) {
1028         aDoc->close();
1029       }
1030     }
1031   }*/
1032   // 3. delete objects
1033   QMainWindow* aDesktop = isSalomeMode() ? salomeConnector()->desktop() : myMainWindow;
1034   std::set<FeaturePtr> anIgnoredFeatures;
1035   if (deleteFeatures(anObjects, anIgnoredFeatures, aDesktop, true)) {
1036     myDisplayer->updateViewer();
1037     aMgr->finishOperation();
1038     updateCommandStatus();
1039   }
1040   else {
1041     aMgr->abortOperation();
1042   }
1043 }
1044
1045 //**************************************************************
1046 void XGUI_Workshop::moveObjects()
1047 {
1048   if (!isActiveOperationAborted())
1049     return;
1050
1051   SessionPtr aMgr = ModelAPI_Session::get();
1052
1053   QString aDescription = contextMenuMgr()->action("MOVE_CMD")->text();
1054   aMgr->startOperation(aDescription.toStdString());
1055
1056   QObjectPtrList anObjects = mySelector->selection()->selectedObjects();
1057   DocumentPtr anActiveDocument = aMgr->activeDocument();
1058
1059   FeaturePtr aCurrentFeature = anActiveDocument->currentFeature(true);
1060   foreach (ObjectPtr aObj, anObjects) {
1061     FeaturePtr aFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(aObj);
1062     if (aFeature.get()) {
1063       anActiveDocument->moveFeature(aFeature, aCurrentFeature);
1064       aCurrentFeature = anActiveDocument->currentFeature(true);
1065     }
1066   }
1067   aMgr->finishOperation();
1068 }
1069
1070 //**************************************************************
1071 std::set<FeaturePtr> refFeatures(const ObjectPtr& theObject, bool checkAllDocuments = true)
1072 {
1073   std::set<FeaturePtr> aRefFeatures;
1074   FeaturePtr aFeature = ModelAPI_Feature::feature(theObject);
1075   if (!aFeature.get())
1076     return aRefFeatures;
1077
1078   DocumentPtr aFeatureDoc = aFeature->document();
1079   // 1. find references in the current document
1080   aFeatureDoc->refsToFeature(aFeature, aRefFeatures, false);
1081
1082   // 2. find references in all documents if the document of the feature is
1083   // "PartSet". Features of this document can be used in all other documents
1084   if (checkAllDocuments) {
1085     SessionPtr aMgr = ModelAPI_Session::get();
1086     DocumentPtr aModuleDoc = aMgr->moduleDocument();
1087     if (aFeatureDoc == aModuleDoc) {
1088       // the feature and results of the feature should be found in references
1089       std::list<ObjectPtr> aDeletedObjects;
1090       aDeletedObjects.push_back(aFeature);
1091       typedef std::list<std::shared_ptr<ModelAPI_Result> > ResultsList;
1092       const ResultsList& aResults = aFeature->results();
1093       ResultsList::const_iterator aRIter = aResults.begin();
1094       for (; aRIter != aResults.cend(); aRIter++) {
1095         ResultPtr aRes = *aRIter;
1096         if (aRes.get())
1097           aDeletedObjects.push_back(aRes);
1098       }
1099       // get all opened documents; found features in the documents;
1100       // get a list of objects where a feature refers;
1101       // search in these objects the deleted objects.
1102       std::list<DocumentPtr> anOpenedDocs = aMgr->allOpenedDocuments();
1103       std::list<DocumentPtr>::const_iterator anIt = anOpenedDocs.begin(),
1104                                               aLast = anOpenedDocs.end();
1105       std::list<std::pair<std::string, std::list<ObjectPtr> > > aRefs;
1106       for (; anIt != aLast; anIt++) {
1107         DocumentPtr aDocument = *anIt;
1108         if (aDocument == aFeatureDoc)
1109           continue; // this document has been already processed in 1.1
1110
1111         int aFeaturesCount = aDocument->size(ModelAPI_Feature::group());
1112         for (int aId = 0; aId < aFeaturesCount; aId++) {
1113           ObjectPtr anObject = aDocument->object(ModelAPI_Feature::group(), aId);
1114           FeaturePtr aFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(anObject);
1115           if (!aFeature.get())
1116             continue;
1117
1118           aRefs.clear();
1119           aFeature->data()->referencesToObjects(aRefs);
1120           std::list<std::pair<std::string, std::list<ObjectPtr> > >::iterator aRef = aRefs.begin();
1121           bool aHasReferenceToObjetc = false;
1122           for(; aRef != aRefs.end() && !aHasReferenceToObjetc; aRef++) {
1123             std::list<ObjectPtr>::iterator aRefObj = aRef->second.begin();
1124             for(; aRefObj != aRef->second.end() && !aHasReferenceToObjetc; aRefObj++) {
1125               std::list<ObjectPtr>::const_iterator aObjIt = aDeletedObjects.begin();
1126               for(; aObjIt != aDeletedObjects.end() && !aHasReferenceToObjetc; aObjIt++) {
1127                 aHasReferenceToObjetc = *aObjIt == *aRefObj;
1128               }
1129             }
1130           }
1131           if (aHasReferenceToObjetc)
1132             aRefFeatures.insert(aFeature);
1133         }
1134       }
1135     }
1136   }
1137   return aRefFeatures;
1138 }
1139
1140 //**************************************************************
1141 bool XGUI_Workshop::deleteFeatures(const QObjectPtrList& theList,
1142                                    const std::set<FeaturePtr>& theIgnoredFeatures,
1143                                    QWidget* theParent,
1144                                    const bool theAskAboutDeleteReferences)
1145 {
1146 #ifdef DEBUG_DELETE
1147   QStringList aDInfo;
1148   QObjectPtrList::const_iterator aDIt = theList.begin(), aDLast = theList.end();
1149   for (; aDIt != aDLast; ++aDIt) {
1150     aDInfo.append(ModuleBase_Tools::objectInfo((*aDIt)));
1151   }
1152   QString anInfoStr = aDInfo.join(", ");
1153   qDebug(QString("deleteFeatures: %1, %2").arg(theList.size()).arg(anInfoStr).toStdString().c_str());
1154 #endif
1155
1156   // 1. find all referenced features
1157   std::set<FeaturePtr> aRefFeatures;
1158   foreach (ObjectPtr aDeletedObj, theList) {
1159     std::set<FeaturePtr> aFeatures = refFeatures(aDeletedObj);
1160     aRefFeatures.insert(aFeatures.begin(), aFeatures.end());
1161   }
1162   // 2. warn about the references remove, break the delete operation if the user chose it
1163   if (theAskAboutDeleteReferences && !aRefFeatures.empty()) {
1164     QStringList aRefNames;
1165     std::set<FeaturePtr>::const_iterator anIt = aRefFeatures.begin(),
1166                                          aLast = aRefFeatures.end();
1167     for (; anIt != aLast; anIt++) {
1168       aRefNames.append((*anIt)->name().c_str());
1169     }
1170     QString aNames = aRefNames.join(", ");
1171
1172     QMessageBox::StandardButton aRes = QMessageBox::warning(
1173         theParent, tr("Delete features"),
1174         QString(tr("Selected features are used in the following features: %1.\
1175 These features will be deleted also. Would you like to continue?")).arg(aNames),
1176         QMessageBox::No | QMessageBox::Yes, QMessageBox::No);
1177     if (aRes != QMessageBox::Yes)
1178       return false;
1179   }
1180
1181   // 3. remove referenced features
1182   std::set<FeaturePtr>::const_iterator anIt = aRefFeatures.begin(),
1183                                        aLast = aRefFeatures.end();
1184 #ifdef DEBUG_DELETE
1185   QStringList anInfo;
1186 #endif
1187   for (; anIt != aLast; anIt++) {
1188     FeaturePtr aFeature = (*anIt);
1189     DocumentPtr aDoc = aFeature->document();
1190     if (theIgnoredFeatures.find(aFeature) == theIgnoredFeatures.end()) {
1191       aDoc->removeFeature(aFeature);
1192 #ifdef DEBUG_DELETE
1193       anInfo.append(ModuleBase_Tools::objectInfo(aFeature).toStdString().c_str());
1194 #endif
1195     }
1196   }
1197 #ifdef DEBUG_DELETE
1198   qDebug(QString("remove references:%1").arg(anInfo.join("; ")).toStdString().c_str());
1199   anInfo.clear();
1200 #endif
1201
1202   QString anId = QString::fromStdString("DELETE_CMD");
1203   QStringList anObjectGroups = contextMenuMgr()->actionObjectGroups(anId);
1204   // 4. remove the parameter features
1205   foreach (ObjectPtr aObj, theList) {
1206     // features and parameters can be removed here,
1207     // the results are removed only by a corresponded feature remove
1208     std::string aGroupName = aObj->groupName();
1209     if (!anObjectGroups.contains(aGroupName.c_str()))
1210       continue;
1211
1212     FeaturePtr aFeature = ModelAPI_Feature::feature(aObj);
1213     if (aFeature) {
1214       // TODO: to learn the workshop to delegate the Part object deletion to the PartSet module
1215       // part features are removed in the PartSet module. This condition should be moved there
1216       if (aFeature->getKind() == "Part")
1217         continue;
1218
1219       DocumentPtr aDoc = aObj->document();
1220       if (theIgnoredFeatures.find(aFeature) == theIgnoredFeatures.end()) {
1221 #ifdef DEBUG_DELETE
1222         QString anInfoStr = ModuleBase_Tools::objectInfo(aFeature);
1223         anInfo.append(anInfoStr);
1224         qDebug(QString("remove feature :%1").arg(anInfoStr).toStdString().c_str());
1225 #endif
1226         aDoc->removeFeature(aFeature);
1227       }
1228     }
1229   }
1230 #ifdef DEBUG_DELETE
1231   qDebug(QString("remove features:%1").arg(anInfo.join("; ")).toStdString().c_str());
1232 #endif
1233   return true;
1234 }
1235
1236 bool hasResults(QObjectPtrList theObjects, const std::set<std::string>& theTypes)
1237 {
1238   bool isFoundResultType = false;
1239   foreach(ObjectPtr anObj, theObjects)
1240   {
1241     ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_Result>(anObj);
1242     if (aResult.get() == NULL)
1243       continue;
1244
1245     isFoundResultType = theTypes.find(aResult->groupName()) != theTypes.end();
1246     if (isFoundResultType)
1247       break;
1248   }
1249   return isFoundResultType;
1250 }
1251
1252 //**************************************************************
1253 // Returns the list of features placed between theObject and the current feature
1254 // in the same document. Excludes theObject, includes the current feature.
1255 std::list<FeaturePtr> toCurrentFeatures(const ObjectPtr& theObject)
1256 {
1257   std::list<FeaturePtr> aResult;
1258   DocumentPtr aDocument = theObject->document();
1259   std::list<FeaturePtr> anAllFeatures = aDocument->allFeatures();
1260   // find the object iterator
1261   std::list<FeaturePtr>::iterator aObjectIt = std::find(anAllFeatures.begin(), anAllFeatures.end(), theObject);
1262   if (aObjectIt == anAllFeatures.end()) 
1263     return aResult;
1264   // find the current feature iterator
1265   std::list<FeaturePtr>::iterator aCurrentIt = std::find(anAllFeatures.begin(), anAllFeatures.end(), aDocument->currentFeature(true));
1266   if (aCurrentIt == anAllFeatures.end()) 
1267     return aResult;
1268   // check the right order
1269   if (std::distance(aObjectIt, anAllFeatures.end()) <= std::distance(aCurrentIt, anAllFeatures.end()))
1270     return aResult;
1271   // exclude the object
1272   std::advance(aObjectIt, 1);
1273   // include the current feature
1274   std::advance(aCurrentIt, 1);
1275   return std::list<FeaturePtr>(aObjectIt, aCurrentIt);
1276 }
1277
1278 bool XGUI_Workshop::canMoveFeature()
1279 {
1280   QObjectPtrList aObjects = mySelector->selection()->selectedObjects();
1281   foreach (ObjectPtr aObject, aObjects) {
1282     // 1. Get features placed between selected and current in the document 
1283     std::list<FeaturePtr> aFeaturesBetween = toCurrentFeatures(aObject);
1284     // if aFeaturesBetween is empty it means wrong order or aObject is the current feature
1285     if (aFeaturesBetween.empty())
1286       return false;
1287     std::set<FeaturePtr> aPlacedFeatures(aFeaturesBetween.begin(), aFeaturesBetween.end());
1288     // 2. Get all reference features to the selected object in the document 
1289     std::set<FeaturePtr> aRefFeatures = refFeatures(aObject, false);
1290     if (aRefFeatures.empty())
1291       continue;
1292     // 3. Find any placed features in all reference features
1293     std::set<FeaturePtr> aIntersectionFeatures;
1294     std::set_intersection(aRefFeatures.begin(), aRefFeatures.end(),
1295                           aPlacedFeatures.begin(), aPlacedFeatures.end(),
1296                           std::inserter(aIntersectionFeatures, aIntersectionFeatures.begin()));
1297     // 4. Return false if any reference feature is placed before curent feature
1298     if (!aIntersectionFeatures.empty())
1299       return false;
1300   }
1301   return true;
1302 }
1303
1304 //**************************************************************
1305 bool XGUI_Workshop::canChangeColor() const
1306 {
1307   QObjectPtrList aObjects = mySelector->selection()->selectedObjects();
1308
1309   std::set<std::string> aTypes;
1310   aTypes.insert(ModelAPI_ResultGroup::group());
1311   aTypes.insert(ModelAPI_ResultConstruction::group());
1312   aTypes.insert(ModelAPI_ResultBody::group());
1313   aTypes.insert(ModelAPI_ResultPart::group());
1314
1315   return hasResults(aObjects, aTypes);
1316 }
1317
1318 void setColor(ResultPtr theResult, std::vector<int>& theColor)
1319 {
1320   if (!theResult.get())
1321     return;
1322
1323   AttributeIntArrayPtr aColorAttr = theResult->data()->intArray(ModelAPI_Result::COLOR_ID());
1324   if (aColorAttr.get() != NULL) {
1325     if (!aColorAttr->size()) {
1326       aColorAttr->setSize(3);
1327     }
1328     aColorAttr->setValue(0, theColor[0]);
1329     aColorAttr->setValue(1, theColor[1]);
1330     aColorAttr->setValue(2, theColor[2]);
1331   }
1332 }
1333
1334 //**************************************************************
1335 void XGUI_Workshop::changeColor(const QObjectPtrList& theObjects)
1336 {
1337   AttributeIntArrayPtr aColorAttr;
1338   // 1. find the current color of the object. This is a color of AIS presentation
1339   // The objects are iterated until a first valid color is found 
1340   std::vector<int> aColor;
1341   foreach(ObjectPtr anObject, theObjects) {
1342     ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_Result>(anObject);
1343     if (aResult.get())
1344       XGUI_CustomPrs::getResultColor(aResult, aColor);
1345     else {
1346       // TODO: remove the obtaining a color from the AIS object
1347       // this does not happen never because:
1348       // 1. The color can be changed only on results
1349       // 2. The result can be not visualized in the viewer(e.g. Origin Construction)
1350       AISObjectPtr anAISObj = myDisplayer->getAISObject(anObject);
1351       if (anAISObj.get()) {
1352         aColor.resize(3);
1353         anAISObj->getColor(aColor[0], aColor[1], aColor[2]);
1354       }
1355     }
1356     if (!aColor.empty())
1357       break;
1358   }
1359   if (aColor.size() != 3)
1360     return;
1361
1362   // 2. show the dialog to change the value
1363   XGUI_ColorDialog* aDlg = new XGUI_ColorDialog(mainWindow());
1364   aDlg->setColor(aColor);
1365   aDlg->move(QCursor::pos());
1366   bool isDone = aDlg->exec() == QDialog::Accepted;
1367   if (!isDone)
1368     return;
1369
1370   bool isRandomColor = aDlg->isRandomColor();
1371
1372   // 3. abort the previous operation and start a new one
1373   SessionPtr aMgr = ModelAPI_Session::get();
1374   bool aWasOperation = aMgr->isOperation(); // keep this value
1375   if (!aWasOperation) {
1376     QString aDescription = contextMenuMgr()->action("COLOR_CMD")->text();
1377     aMgr->startOperation(aDescription.toStdString());
1378   }
1379
1380   // 4. set the value to all results
1381   foreach(ObjectPtr anObj, theObjects) {
1382     ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_Result>(anObj);
1383     if (aResult.get() != NULL) {
1384       std::vector<int> aColorResult = aDlg->getColor();
1385       setColor(aResult, aColorResult);
1386     }
1387   }
1388   if (!aWasOperation)
1389     aMgr->finishOperation();
1390   updateCommandStatus();
1391 }
1392
1393 //**************************************************************
1394 #define SET_DISPLAY_GROUP(aGroupName, aDisplay) \
1395 for (int i = 0; i < aDoc->size(aGroupName); i++) { \
1396   aDoc->object(aGroupName, i)->setDisplayed(aDisplay); \
1397 }
1398 void XGUI_Workshop::showObjects(const QObjectPtrList& theList, bool isVisible)
1399 {
1400   foreach (ObjectPtr aObj, theList) {
1401     /*
1402     ResultPartPtr aPartRes = std::dynamic_pointer_cast<ModelAPI_ResultPart>(aObj);
1403     if (aPartRes) {
1404       DocumentPtr aDoc = aPartRes->partDoc();
1405       SET_DISPLAY_GROUP(ModelAPI_ResultBody::group(), isVisible)
1406       SET_DISPLAY_GROUP(ModelAPI_ResultConstruction::group(), isVisible)
1407       SET_DISPLAY_GROUP(ModelAPI_ResultGroup::group(), isVisible)
1408     } else {
1409     */
1410       aObj->setDisplayed(isVisible);
1411     //}
1412   }
1413   Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_TO_REDISPLAY));
1414 }
1415
1416 //**************************************************************
1417 void XGUI_Workshop::showOnlyObjects(const QObjectPtrList& theList)
1418 {
1419   // Hide all displayed objects
1420   QObjectPtrList aList = myDisplayer->displayedObjects();
1421   foreach (ObjectPtr aObj, aList)
1422     aObj->setDisplayed(false);
1423
1424   // Show only objects from the list
1425   foreach (ObjectPtr aObj, theList) {
1426     /*
1427     ResultPartPtr aPartRes = std::dynamic_pointer_cast<ModelAPI_ResultPart>(aObj);
1428     if (aPartRes) {
1429       DocumentPtr aDoc = aPartRes->partDoc();
1430       SET_DISPLAY_GROUP(ModelAPI_ResultBody::group(), true)
1431       SET_DISPLAY_GROUP(ModelAPI_ResultConstruction::group(), true)
1432       SET_DISPLAY_GROUP(ModelAPI_ResultGroup::group(), true)
1433     } else {
1434     */
1435       aObj->setDisplayed(true);
1436     //}
1437   }
1438   Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_TO_REDISPLAY));
1439
1440 }
1441
1442
1443 //**************************************************************
1444 void XGUI_Workshop::registerValidators() const
1445 {
1446   SessionPtr aMgr = ModelAPI_Session::get();
1447   ModelAPI_ValidatorsFactory* aFactory = aMgr->validators();
1448 }
1449
1450 //**************************************************************
1451 /*void XGUI_Workshop::displayAllResults()
1452 {
1453   SessionPtr aMgr = ModelAPI_Session::get();
1454   DocumentPtr aRootDoc = aMgr->moduleDocument();
1455   displayDocumentResults(aRootDoc);
1456   for (int i = 0; i < aRootDoc->size(ModelAPI_ResultPart::group()); i++) {
1457     ObjectPtr aObject = aRootDoc->object(ModelAPI_ResultPart::group(), i);
1458     ResultPartPtr aPart = std::dynamic_pointer_cast<ModelAPI_ResultPart>(aObject);
1459     displayDocumentResults(aPart->partDoc());
1460   }
1461   myDisplayer->updateViewer();
1462 }*/
1463
1464 //**************************************************************
1465 void XGUI_Workshop::displayDocumentResults(DocumentPtr theDoc)
1466 {
1467   if (!theDoc)
1468     return;
1469   displayGroupResults(theDoc, ModelAPI_ResultConstruction::group());
1470   displayGroupResults(theDoc, ModelAPI_ResultBody::group());
1471 }
1472
1473 //**************************************************************
1474 void XGUI_Workshop::displayGroupResults(DocumentPtr theDoc, std::string theGroup)
1475 {
1476   for (int i = 0; i < theDoc->size(theGroup); i++) 
1477     theDoc->object(theGroup, i)->setDisplayed(true);
1478     //displayObject(theDoc->object(theGroup, i));
1479   Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_TO_REDISPLAY));
1480 }
1481
1482 //**************************************************************
1483 void XGUI_Workshop::setDisplayMode(const QObjectPtrList& theList, int theMode)
1484 {
1485   foreach(ObjectPtr aObj, theList) {
1486     myDisplayer->setDisplayMode(aObj, (XGUI_Displayer::DisplayMode)theMode, false);
1487   }
1488   if (theList.size() > 0)
1489     myDisplayer->updateViewer();
1490 }
1491
1492 //**************************************************************
1493 void XGUI_Workshop::closeDocument()
1494 {
1495   ModuleBase_Operation* anOperation = operationMgr()->currentOperation();
1496   while (anOperation) {
1497     anOperation->abort();
1498     anOperation = operationMgr()->currentOperation();
1499   }
1500   myDisplayer->closeLocalContexts();
1501   myDisplayer->eraseAll();
1502   objectBrowser()->clearContent();
1503
1504   SessionPtr aMgr = ModelAPI_Session::get();
1505   aMgr->closeAll();
1506 }
1507
1508 void XGUI_Workshop::addHistoryMenu(QObject* theObject, const char* theSignal, const char* theSlot)
1509 {
1510   XGUI_HistoryMenu* aMenu = NULL;
1511   if (isSalomeMode()) {
1512     QAction* anAction = qobject_cast<QAction*>(theObject);
1513     if (!anAction)
1514       return;
1515     aMenu = new XGUI_HistoryMenu(anAction);
1516   } else {
1517     QToolButton* aButton =  qobject_cast<QToolButton*>(theObject);
1518     aMenu = new XGUI_HistoryMenu(aButton);
1519   }
1520   connect(this, theSignal, aMenu, SLOT(setHistory(const QList<ActionInfo>&)));
1521   connect(aMenu, SIGNAL(actionSelected(int)), this, theSlot);
1522 }
1523
1524 QList<ActionInfo> XGUI_Workshop::processHistoryList(const std::list<std::string>& theList) const
1525 {
1526   QList<ActionInfo> aResult;
1527   std::list<std::string>::const_iterator it = theList.cbegin();
1528   for (; it != theList.cend(); it++) {
1529     QString anId = QString::fromStdString(*it);
1530     bool isEditing = anId.endsWith(ModuleBase_Operation::EditSuffix());
1531     if (isEditing) {
1532       anId.chop(ModuleBase_Operation::EditSuffix().size());
1533     }
1534     ActionInfo anInfo;
1535     QAction* aContextMenuAct = myContextMenuMgr->actionByName(anId);
1536     if (aContextMenuAct) {
1537       anInfo.initFrom(aContextMenuAct);
1538     } else {
1539       anInfo = myActionsMgr->actionInfoById(anId);
1540     }
1541     if (isEditing) {
1542       anInfo.text = anInfo.text.prepend("Modify ");
1543     }
1544     aResult << anInfo;
1545   }
1546   return aResult;
1547 }