1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D -->
3 #include "XGUI_ContextMenuMgr.h"
4 #include "XGUI_Workshop.h"
5 #include "XGUI_ObjectsBrowser.h"
6 #include "XGUI_SelectionMgr.h"
7 #include "XGUI_Displayer.h"
8 #include "XGUI_ViewerProxy.h"
9 #include "XGUI_Selection.h"
10 #include "XGUI_SalomeConnector.h"
11 #include "XGUI_DataModel.h"
12 #include "XGUI_OperationMgr.h"
13 #include "XGUI_Tools.h"
16 #include <AppElements_MainWindow.h>
19 //#include "PartSetPlugin_Part.h"
21 #include <ModelAPI_Data.h>
22 #include <ModelAPI_AttributeDocRef.h>
23 #include <ModelAPI_Object.h>
24 #include <ModelAPI_Session.h>
25 #include <ModelAPI_ResultGroup.h>
26 #include <ModelAPI_ResultParameter.h>
27 #include <ModelAPI_ResultConstruction.h>
28 #include <ModelAPI_ResultBody.h>
29 #include <ModelAPI_Tools.h>
31 #include <ModuleBase_IModule.h>
32 #include <ModuleBase_Tools.h>
33 #include <ModuleBase_OperationAction.h>
34 #include <ModuleBase_ViewerPrs.h>
37 #include <QActionGroup>
38 #include <QContextMenuEvent>
41 #include <QMainWindow>
44 XGUI_ContextMenuMgr::XGUI_ContextMenuMgr(XGUI_Workshop* theParent)
46 myWorkshop(theParent),
51 XGUI_ContextMenuMgr::~XGUI_ContextMenuMgr()
55 void XGUI_ContextMenuMgr::createActions()
58 QMainWindow* aDesktop = myWorkshop->salomeConnector()->desktop();
60 QMainWindow* aDesktop = myWorkshop->mainWindow();
63 QAction* aAction = new QAction(QIcon(":pictures/delete.png"), tr("Delete"), this);
64 aDesktop->addAction(aAction);
66 addAction("DELETE_CMD", aAction);
67 aAction->setShortcutContext(Qt::ApplicationShortcut);
69 aAction = new QAction(QIcon(":pictures/rename_edit.png"), tr("Rename"), this);
70 addAction("RENAME_CMD", aAction);
71 connect(aAction, SIGNAL(triggered(bool)), this, SLOT(onRename()));
73 aAction = new QAction(QIcon(":pictures/move.png"), XGUI_Workshop::MOVE_TO_END_COMMAND, this);
74 addAction("MOVE_CMD", aAction);
76 aAction = new QAction(QIcon(":pictures/clean_history.png"), tr("Clean history"), this);
77 addAction("CLEAN_HISTORY_CMD", aAction);
79 aAction = new QAction(QIcon(":pictures/color.png"), tr("Color..."), this);
80 addAction("COLOR_CMD", aAction);
82 aAction = new QAction(QIcon(":pictures/eye_pencil.png"), tr("Show"), this);
83 addAction("SHOW_CMD", aAction);
85 aAction = new QAction(QIcon(":pictures/eye_pencil.png"), tr("Show only"), this);
86 addAction("SHOW_ONLY_CMD", aAction);
88 aAction = new QAction(QIcon(":pictures/eye_pencil_closed.png"), tr("Hide"), this);
89 addAction("HIDE_CMD", aAction);
91 aAction = new QAction(QIcon(":pictures/eye_pencil_closed.png"), tr("Hide all"), this);
92 addAction("HIDEALL_CMD", aAction);
94 aAction = new QAction(QIcon(":pictures/shading.png"), tr("Shading"), this);
95 addAction("SHADING_CMD", aAction);
97 aAction = new QAction(QIcon(":pictures/wireframe.png"), tr("Wireframe"), this);
98 addAction("WIREFRAME_CMD", aAction);
100 mySeparator = new QAction(this);
101 mySeparator->setSeparator(true);
103 mySelectActions = new QActionGroup(this);
104 mySelectActions->setExclusive(true);
106 aAction = new QAction(QIcon(":pictures/vertex.png"), tr("Vertices"), this);
107 aAction->setCheckable(true);
108 addAction("SELECT_VERTEX_CMD", aAction);
109 mySelectActions->addAction(aAction);
111 aAction = new QAction(QIcon(":pictures/edge.png"), tr("Edges"), this);
112 aAction->setCheckable(true);
113 addAction("SELECT_EDGE_CMD", aAction);
114 mySelectActions->addAction(aAction);
116 aAction = new QAction(QIcon(":pictures/face.png"), tr("Faces"), this);
117 aAction->setCheckable(true);
118 addAction("SELECT_FACE_CMD", aAction);
119 mySelectActions->addAction(aAction);
121 aAction = new QAction(QIcon(":pictures/result.png"), tr("Result"), this);
122 aAction->setCheckable(true);
123 addAction("SELECT_RESULT_CMD", aAction);
124 mySelectActions->addAction(aAction);
126 aAction->setChecked(true);
128 aAction = new QAction(QIcon(":pictures/find_result.png"), tr("Select results"), this);
129 addAction("SHOW_RESULTS_CMD", aAction);
131 buildObjBrowserMenu();
135 void XGUI_ContextMenuMgr::addAction(const QString& theId, QAction* theAction)
137 if (myActions.contains(theId))
138 qCritical("A command with Id = '%s' already defined!", qPrintable(theId));
139 theAction->setData(theId);
140 connect(theAction, SIGNAL(triggered(bool)), this, SLOT(onAction(bool)));
141 myActions[theId] = theAction;
144 QAction* XGUI_ContextMenuMgr::action(const QString& theId) const
146 if (myActions.contains(theId))
147 return myActions[theId];
151 QAction* XGUI_ContextMenuMgr::actionByName(const QString& theName) const
153 foreach(QAction* eachAction, myActions) {
154 if (eachAction->text() == theName) {
161 QStringList XGUI_ContextMenuMgr::actionIds() const
163 return myActions.keys();
166 void XGUI_ContextMenuMgr::onAction(bool isChecked)
168 QAction* aAction = static_cast<QAction*>(sender());
169 emit actionTriggered(aAction->data().toString(), isChecked);
172 void XGUI_ContextMenuMgr::updateCommandsStatus()
176 void XGUI_ContextMenuMgr::onContextMenuRequest(QContextMenuEvent* theEvent)
178 QMenu* aMenu = new QMenu();
179 if (sender() == myWorkshop->objectBrowser()) {
180 updateObjectBrowserMenu();
181 addObjBrowserMenu(aMenu);
182 } else if (sender() == myWorkshop->viewer()) {
184 addViewerMenu(aMenu);
187 if (aMenu && (aMenu->actions().size() > 0)) {
188 // it is possible that some objects should do something before and after the popup menu exec
189 // e.g. a sketch manager changes an internal flag on this signals in order to do not hide
191 emit beforeContextMenu();
192 aMenu->exec(theEvent->globalPos());
193 emit afterContextMenu();
198 void XGUI_ContextMenuMgr::updateObjectBrowserMenu()
200 foreach(QAction* aAction, myActions)
201 aAction->setEnabled(false);
203 XGUI_SelectionMgr* aSelMgr = myWorkshop->selector();
204 QObjectPtrList aObjects = aSelMgr->selection()->selectedObjects();
205 int aSelected = aObjects.size();
207 SessionPtr aMgr = ModelAPI_Session::get();
208 XGUI_Displayer* aDisplayer = myWorkshop->displayer();
209 bool hasResult = false;
210 bool hasFeature = false;
211 bool hasParameter = false;
212 bool hasCompositeOwner = false;
213 ModuleBase_Tools::checkObjects(aObjects, hasResult, hasFeature, hasParameter,
216 if (aSelected == 1) {
217 ObjectPtr aObject = aObjects.first();
219 if (hasResult && myWorkshop->canBeShaded(aObject)) {
220 XGUI_Displayer::DisplayMode aMode = aDisplayer->displayMode(aObject);
221 if (aMode != XGUI_Displayer::NoMode) {
222 action("WIREFRAME_CMD")->setEnabled(aMode == XGUI_Displayer::Shading);
223 action("SHADING_CMD")->setEnabled(aMode == XGUI_Displayer::Wireframe);
225 action("WIREFRAME_CMD")->setEnabled(true);
226 action("SHADING_CMD")->setEnabled(true);
230 bool aHasSubResults = ModelAPI_Tools::hasSubResults(
231 std::dynamic_pointer_cast<ModelAPI_Result>(aObject));
232 if (aHasSubResults) {
233 action("HIDE_CMD")->setEnabled(true);
234 action("SHOW_CMD")->setEnabled(true);
237 if (aObject->isDisplayed()) {
238 action("HIDE_CMD")->setEnabled(true);
239 } else if (hasResult && (!hasParameter)) {
240 action("SHOW_CMD")->setEnabled(true);
243 if (!(hasParameter || hasFeature))
244 action("SHOW_ONLY_CMD")->setEnabled(true);
246 else if (hasFeature && myWorkshop->canMoveFeature())
247 action("MOVE_CMD")->setEnabled(true);
249 if( aMgr->activeDocument() == aObject->document() )
251 action("RENAME_CMD")->setEnabled(true);
252 action("DELETE_CMD")->setEnabled(!hasCompositeOwner);
253 action("CLEAN_HISTORY_CMD")->setEnabled(!hasCompositeOwner &&
254 (hasFeature || hasParameter));
258 // parameter is commented because the actions are not in the list of result parameter actions
259 if (hasResult /*&& (!hasParameter)*/) {
260 action("SHOW_CMD")->setEnabled(true);
261 action("HIDE_CMD")->setEnabled(true);
262 action("SHOW_ONLY_CMD")->setEnabled(true);
263 action("SHADING_CMD")->setEnabled(true);
264 action("WIREFRAME_CMD")->setEnabled(true);
267 bool allActive = true;
268 foreach( ObjectPtr aObject, aObjects )
269 if( aMgr->activeDocument() != aObject->document() ) {
273 if (!hasCompositeOwner && allActive ) {
274 if (hasFeature || hasParameter)
275 action("DELETE_CMD")->setEnabled(true);
277 if (!hasCompositeOwner && allActive && (hasFeature|| hasParameter))
278 action("CLEAN_HISTORY_CMD")->setEnabled(true);
280 action("SHOW_RESULTS_CMD")->setEnabled(hasFeature);
283 // Show/Hide command has to be disabled for objects from non active document
284 bool aDeactivate = false;
285 foreach (ObjectPtr aObj, aObjects) {
286 if (!aObj->document()->isActive()) {
287 if ((aObj->document() != ModelAPI_Session::get()->moduleDocument()) ||
288 aObj->groupName() == ModelAPI_ResultPart::group()) {
295 // If at leas a one objec can not be edited then Show/Hide has to be disabled
296 action("SHOW_CMD")->setEnabled(false);
297 action("HIDE_CMD")->setEnabled(false);
298 action("SHOW_ONLY_CMD")->setEnabled(false);
301 if (myWorkshop->canChangeColor())
302 action("COLOR_CMD")->setEnabled(true);
304 ModuleBase_IModule* aModule = myWorkshop->module();
306 aModule->updateObjectBrowserMenu(myActions);
309 void XGUI_ContextMenuMgr::updateViewerMenu()
311 foreach(QAction* aAction, myActions)
312 aAction->setEnabled(false);
314 XGUI_SelectionMgr* aSelMgr = myWorkshop->selector();
315 XGUI_Displayer* aDisplayer = myWorkshop->displayer();
316 QList<ModuleBase_ViewerPrsPtr> aPrsList = aSelMgr->selection()->getSelected(ModuleBase_ISelection::Viewer);
317 if (aPrsList.size() > 0) {
318 bool isVisible = false;
319 bool isShading = false;
320 bool canBeShaded = false;
322 foreach(ModuleBase_ViewerPrsPtr aPrs, aPrsList) {
323 aObject = aPrs->object();
324 ResultPtr aRes = std::dynamic_pointer_cast<ModelAPI_Result>(aObject);
325 if (aRes && aRes->isDisplayed()) {
327 canBeShaded = myWorkshop->displayer()->canBeShaded(aObject);
328 isShading = (myWorkshop->displayer()->displayMode(aObject) == XGUI_Displayer::Shading);
334 XGUI_Displayer::DisplayMode aMode = aDisplayer->displayMode(aObject);
335 if (aMode != XGUI_Displayer::NoMode) {
336 action("WIREFRAME_CMD")->setEnabled(aMode == XGUI_Displayer::Shading);
337 action("SHADING_CMD")->setEnabled(aMode == XGUI_Displayer::Wireframe);
339 action("WIREFRAME_CMD")->setEnabled(true);
340 action("SHADING_CMD")->setEnabled(true);
343 action("SHOW_ONLY_CMD")->setEnabled(true);
344 action("HIDE_CMD")->setEnabled(true);
346 action("SHOW_CMD")->setEnabled(true);
348 if (myWorkshop->displayer()->objectsCount() > 0)
349 action("HIDEALL_CMD")->setEnabled(true);
351 // Update selection menu
352 QIntList aModes = aDisplayer->activeSelectionModes();
353 if (aModes.count() <= 1) {
354 action("SELECT_VERTEX_CMD")->setEnabled(true);
355 action("SELECT_EDGE_CMD")->setEnabled(true);
356 action("SELECT_FACE_CMD")->setEnabled(true);
357 action("SELECT_RESULT_CMD")->setEnabled(true);
358 if (aModes.count() == 1) {
359 switch (aModes.first()) {
361 action("SELECT_VERTEX_CMD")->setChecked(true);
364 action("SELECT_EDGE_CMD")->setChecked(true);
367 action("SELECT_FACE_CMD")->setChecked(true);
370 action("SELECT_RESULT_CMD")->setChecked(true);
373 action("SELECT_RESULT_CMD")->setChecked(true);
376 ModuleBase_IModule* aModule = myWorkshop->module();
378 aModule->updateViewerMenu(myActions);
380 if (myWorkshop->canChangeColor())
381 action("COLOR_CMD")->setEnabled(true);
383 action("DELETE_CMD")->setEnabled(true);
386 void XGUI_ContextMenuMgr::connectObjectBrowser()
388 connect(myWorkshop->objectBrowser(), SIGNAL(contextMenuRequested(QContextMenuEvent*)), this,
389 SLOT(onContextMenuRequest(QContextMenuEvent*)));
392 void XGUI_ContextMenuMgr::connectViewer()
394 connect(myWorkshop->viewer(), SIGNAL(contextMenuRequested(QContextMenuEvent*)), this,
395 SLOT(onContextMenuRequest(QContextMenuEvent*)));
399 void XGUI_ContextMenuMgr::buildObjBrowserMenu()
401 QAction* aSeparator = new QAction(this);
402 aSeparator->setSeparator(true);
406 // Result construction menu
407 aList.append(action("SHOW_CMD"));
408 aList.append(action("HIDE_CMD"));
409 aList.append(action("SHOW_ONLY_CMD"));
410 aList.append(mySeparator);
411 aList.append(action("RENAME_CMD"));
412 aList.append(action("COLOR_CMD"));
413 myObjBrowserMenus[ModelAPI_ResultConstruction::group()] = aList;
415 //-------------------------------------
418 aList.append(action("WIREFRAME_CMD"));
419 aList.append(action("SHADING_CMD"));
420 aList.append(mySeparator); // this separator is not shown as this action is added after show only
421 // qt list container contains only one instance of the same action
422 aList.append(action("SHOW_CMD"));
423 aList.append(action("HIDE_CMD"));
424 aList.append(action("SHOW_ONLY_CMD"));
425 aList.append(mySeparator);
426 aList.append(action("RENAME_CMD"));
427 aList.append(action("COLOR_CMD"));
428 myObjBrowserMenus[ModelAPI_ResultBody::group()] = aList;
430 myObjBrowserMenus[ModelAPI_ResultGroup::group()] = aList;
432 myObjBrowserMenus[ModelAPI_ResultPart::group()] = aList;
433 //-------------------------------------
436 aList.append(action("SHOW_RESULTS_CMD"));
437 aList.append(mySeparator);
438 aList.append(action("RENAME_CMD"));
439 aList.append(action("MOVE_CMD"));
440 aList.append(mySeparator);
441 aList.append(action("CLEAN_HISTORY_CMD"));
442 aList.append(action("DELETE_CMD"));
443 myObjBrowserMenus[ModelAPI_Feature::group()] = aList;
446 aList.append(action("RENAME_CMD"));
447 aList.append(mySeparator);
448 aList.append(action("CLEAN_HISTORY_CMD"));
449 aList.append(action("DELETE_CMD"));
450 myObjBrowserMenus[ModelAPI_ResultParameter::group()] = aList;
451 //-------------------------------------
454 void XGUI_ContextMenuMgr::buildViewerMenu()
457 // Result construction menu
458 aList.append(action("HIDE_CMD"));
459 aList.append(action("SHOW_ONLY_CMD"));
460 aList.append(mySeparator);
461 aList.append(action("COLOR_CMD"));
462 myViewerMenu[ModelAPI_ResultConstruction::group()] = aList;
464 myViewerMenu[ModelAPI_ResultPart::group()] = aList;
465 //-------------------------------------
468 aList.append(action("WIREFRAME_CMD"));
469 aList.append(action("SHADING_CMD"));
470 aList.append(mySeparator);
471 aList.append(action("HIDE_CMD"));
472 aList.append(action("SHOW_ONLY_CMD"));
473 aList.append(mySeparator);
474 aList.append(action("COLOR_CMD"));
475 myViewerMenu[ModelAPI_ResultBody::group()] = aList;
477 myViewerMenu[ModelAPI_ResultGroup::group()] = aList;
478 //-------------------------------------
482 void XGUI_ContextMenuMgr::addObjBrowserMenu(QMenu* theMenu) const
484 ModuleBase_IModule* aModule = myWorkshop->module();
486 theMenu->addSeparator();
487 aModule->addObjectBrowserMenu(theMenu);
490 XGUI_SelectionMgr* aSelMgr = myWorkshop->selector();
491 QObjectPtrList aObjects = aSelMgr->selection()->selectedObjects();
492 int aSelected = aObjects.size();
493 QActionsList aActions;
494 if (aSelected == 1) {
495 ObjectPtr aObject = aObjects.first();
496 std::string aName = aObject->groupName();
497 if (myObjBrowserMenus.contains(aName))
498 aActions = myObjBrowserMenus[aName];
499 } else if (aSelected > 1) {
500 aActions.append(action("WIREFRAME_CMD"));
501 aActions.append(action("SHADING_CMD"));
502 aActions.append(mySeparator);
503 aActions.append(action("SHOW_CMD"));
504 aActions.append(action("HIDE_CMD"));
505 aActions.append(action("SHOW_ONLY_CMD"));
506 aActions.append(mySeparator);
507 //aActions.append(action("MOVE_CMD"));
508 aActions.append(action("COLOR_CMD"));
509 aActions.append(action("CLEAN_HISTORY_CMD"));
510 aActions.append(action("DELETE_CMD"));
512 theMenu->addActions(aActions);
514 theMenu->addSeparator();
515 theMenu->addActions(myWorkshop->objectBrowser()->actions());
518 void XGUI_ContextMenuMgr::addViewerMenu(QMenu* theMenu) const
520 XGUI_SelectionMgr* aSelMgr = myWorkshop->selector();
521 QList<ModuleBase_ViewerPrsPtr> aPrsList = aSelMgr->selection()->getSelected(ModuleBase_ISelection::Viewer);
522 int aSelected = aPrsList.size();
523 QActionsList aActions;
525 // Create selection menu
526 XGUI_OperationMgr* aOpMgr = myWorkshop->operationMgr();
528 myWorkshop->module()->activeSelectionModes(aModes);
529 if ((!aOpMgr->hasOperation()) && aModes.isEmpty()) {
530 QMenu* aSelMenu = new QMenu(tr("Selection mode"), theMenu);
531 aSelMenu->addAction(action("SELECT_VERTEX_CMD"));
532 aSelMenu->addAction(action("SELECT_EDGE_CMD"));
533 aSelMenu->addAction(action("SELECT_FACE_CMD"));
534 aSelMenu->addAction(action("SELECT_RESULT_CMD"));
535 theMenu->addMenu(aSelMenu);
536 theMenu->addSeparator();
538 if (aSelected == 1) {
539 ObjectPtr aObject = aPrsList.first()->object();
540 if (aObject.get() != NULL) {
541 std::string aName = aObject->groupName();
542 if (myViewerMenu.contains(aName))
543 aActions = myViewerMenu[aName];
545 } else if (aSelected > 1) {
546 aActions.append(action("HIDE_CMD"));
548 // hide all is shown always even if selection in the viewer is empty
549 aActions.append(action("HIDEALL_CMD"));
550 aActions.append(action("COLOR_CMD"));
552 theMenu->addActions(aActions);
554 QMap<int, QAction*> aMenuActions;
555 ModuleBase_IModule* aModule = myWorkshop->module();
557 if (aModule->addViewerMenu(myActions, theMenu, aMenuActions))
558 theMenu->addSeparator();
561 // insert the module menu items on specific positions in the popup menu: some actions should be
562 // in the begin of the list, Delete action should be the last by #1343 issue
563 QList<QAction*> anActions = theMenu->actions();
564 int anActionsSize = anActions.size();
565 QAction* aFirstAction = anActions[0];
566 QMap<int, QAction*>::const_iterator anIt = aMenuActions.begin(), aLast = aMenuActions.end();
567 for (; anIt != aLast; anIt++) {
568 if (anIt.key() > anActionsSize)
569 theMenu->addAction(anIt.value());
571 theMenu->insertAction(aFirstAction, *anIt);
575 theMenu->addSeparator();
576 QMdiArea* aMDI = myWorkshop->mainWindow()->mdiArea();
577 if (aMDI->actions().size() > 0) {
578 QMenu* aSubMenu = theMenu->addMenu(tr("Windows"));
579 aSubMenu->addActions(aMDI->actions());
584 QStringList XGUI_ContextMenuMgr::actionObjectGroups(const QString& theName)
588 QMap<std::string, QActionsList>::const_iterator anIt = myObjBrowserMenus.begin(),
589 aLast = myObjBrowserMenus.end();
590 for (; anIt != aLast; anIt++) {
591 QString aGroupName(anIt.key().c_str());
592 if (aGroups.contains(aGroupName))
594 QActionsList anActions = anIt.value();
595 QActionsList::const_iterator anAIt = anActions.begin(), anALast = anActions.end();
597 for (; anAIt != anALast && !aFound; anAIt++)
598 aFound = (*anAIt)->data().toString() == theName;
600 aGroups.append(aGroupName);
605 void XGUI_ContextMenuMgr::onRename()
607 QObjectPtrList anObjects = myWorkshop->selector()->selection()->selectedObjects();
608 if (!myWorkshop->abortAllOperations())
610 // restore selection in case if dialog box was shown
611 myWorkshop->objectBrowser()->setObjectsSelected(anObjects);
612 myWorkshop->objectBrowser()->onEditItem();