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"
12 #include <AppElements_MainWindow.h>
14 //#include "PartSetPlugin_Part.h"
16 #include <ModelAPI_Data.h>
17 #include <ModelAPI_AttributeDocRef.h>
18 #include <ModelAPI_Object.h>
19 #include <ModelAPI_Session.h>
20 #include <ModelAPI_ResultGroup.h>
21 #include <ModelAPI_ResultParameter.h>
22 #include <ModelAPI_ResultConstruction.h>
23 #include <ModelAPI_ResultBody.h>
25 #include <ModuleBase_IModule.h>
26 #include <ModuleBase_Tools.h>
29 #include <QContextMenuEvent>
34 XGUI_ContextMenuMgr::XGUI_ContextMenuMgr(XGUI_Workshop* theParent)
36 myWorkshop(theParent),
41 XGUI_ContextMenuMgr::~XGUI_ContextMenuMgr()
45 void XGUI_ContextMenuMgr::createActions()
47 QAction* aAction = new QAction(QIcon(":pictures/delete.png"), tr("Delete"), this);
48 QMainWindow* aDesktop = myWorkshop->mainWindow();
50 aDesktop = myWorkshop->salomeConnector()->desktop();
51 aDesktop->addAction(aAction);
53 addAction("DELETE_CMD", aAction);
54 aAction->setShortcut(Qt::Key_Delete);
55 aAction->setShortcutContext(Qt::ApplicationShortcut);
57 aAction = new QAction(QIcon(":pictures/move.png"), tr("Move..."), this);
58 addAction("MOVE_CMD", aAction);
60 aAction = new QAction(QIcon(":pictures/color.png"), tr("Color..."), this);
61 addAction("COLOR_CMD", aAction);
63 aAction = new QAction(QIcon(":pictures/eye_pencil.png"), tr("Show"), this);
64 addAction("SHOW_CMD", aAction);
66 aAction = new QAction(QIcon(":pictures/eye_pencil.png"), tr("Show only"), this);
67 addAction("SHOW_ONLY_CMD", aAction);
69 aAction = new QAction(QIcon(":pictures/eye_pencil_closed.png"), tr("Hide"), this);
70 addAction("HIDE_CMD", aAction);
72 aAction = new QAction(QIcon(":pictures/eye_pencil_closed.png"), tr("Hide all"), this);
73 addAction("HIDEALL_CMD", aAction);
75 aAction = new QAction(QIcon(":pictures/shading.png"), tr("Shading"), this);
76 addAction("SHADING_CMD", aAction);
78 aAction = new QAction(QIcon(":pictures/wireframe.png"), tr("Wireframe"), this);
79 addAction("WIREFRAME_CMD", aAction);
81 mySeparator = new QAction(this);
82 mySeparator->setSeparator(true);
85 buildObjBrowserMenu();
89 void XGUI_ContextMenuMgr::addAction(const QString& theId, QAction* theAction)
91 if (myActions.contains(theId))
92 qCritical("A command with Id = '%s' already defined!", qPrintable(theId));
93 theAction->setData(theId);
94 connect(theAction, SIGNAL(triggered(bool)), this, SLOT(onAction(bool)));
95 myActions[theId] = theAction;
98 QAction* XGUI_ContextMenuMgr::action(const QString& theId) const
100 if (myActions.contains(theId))
101 return myActions[theId];
105 QAction* XGUI_ContextMenuMgr::actionByName(const QString& theName) const
107 foreach(QAction* eachAction, myActions) {
108 if (eachAction->text() == theName) {
115 QStringList XGUI_ContextMenuMgr::actionIds() const
117 return myActions.keys();
120 void XGUI_ContextMenuMgr::onAction(bool isChecked)
122 QAction* aAction = static_cast<QAction*>(sender());
123 emit actionTriggered(aAction->data().toString(), isChecked);
126 void XGUI_ContextMenuMgr::updateCommandsStatus()
130 void XGUI_ContextMenuMgr::onContextMenuRequest(QContextMenuEvent* theEvent)
132 QMenu* aMenu = new QMenu();
133 if (sender() == myWorkshop->objectBrowser()) {
134 updateObjectBrowserMenu();
135 addObjBrowserMenu(aMenu);
136 } else if (sender() == myWorkshop->viewer()) {
138 addViewerMenu(aMenu);
141 if (aMenu && (aMenu->actions().size() > 0)) {
142 // it is possible that some objects should do something before and after the popup menu exec
143 // e.g. a sketch manager changes an internal flag on this signals in order to do not hide
145 emit beforeContextMenu();
146 aMenu->exec(theEvent->globalPos());
147 emit afterContextMenu();
152 void XGUI_ContextMenuMgr::updateObjectBrowserMenu()
154 foreach(QAction* aAction, myActions)
155 aAction->setEnabled(false);
157 XGUI_SelectionMgr* aSelMgr = myWorkshop->selector();
158 QObjectPtrList aObjects = aSelMgr->selection()->selectedObjects();
159 int aSelected = aObjects.size();
161 SessionPtr aMgr = ModelAPI_Session::get();
162 XGUI_Displayer* aDisplayer = myWorkshop->displayer();
163 bool hasResult = false;
164 bool hasFeature = false;
165 bool hasParameter = false;
166 bool hasSubFeature = false;
167 ModuleBase_Tools::checkObjects(aObjects, hasResult, hasFeature, hasParameter, hasSubFeature);
170 if (aSelected == 1) {
171 ObjectPtr aObject = aObjects.first();
174 if (aObject->isDisplayed()) {
175 if (aDisplayer->canBeShaded(aObject)) {
176 action("WIREFRAME_CMD")->setEnabled(true);
177 action("SHADING_CMD")->setEnabled(true);
179 action("HIDE_CMD")->setEnabled(true);
180 } else if (hasResult && (!hasParameter)) {
181 action("SHOW_CMD")->setEnabled(true);
184 if (!(hasParameter || hasFeature))
185 action("SHOW_ONLY_CMD")->setEnabled(true);
187 else if (hasFeature && myWorkshop->canMoveFeature())
188 action("MOVE_CMD")->setEnabled(true);
191 if (hasResult && (!hasParameter)) {
192 action("SHOW_CMD")->setEnabled(true);
193 action("HIDE_CMD")->setEnabled(true);
194 action("SHOW_ONLY_CMD")->setEnabled(true);
195 action("SHADING_CMD")->setEnabled(true);
196 action("WIREFRAME_CMD")->setEnabled(true);
199 if (!hasSubFeature) {
200 if (hasFeature || hasParameter)
201 action("DELETE_CMD")->setEnabled(true);
204 if (myWorkshop->canChangeColor())
205 action("COLOR_CMD")->setEnabled(true);
207 ModuleBase_IModule* aModule = myWorkshop->module();
209 aModule->updateObjectBrowserMenu(myActions);
212 void XGUI_ContextMenuMgr::updateViewerMenu()
214 foreach(QAction* aAction, myActions)
215 aAction->setEnabled(false);
217 XGUI_SelectionMgr* aSelMgr = myWorkshop->selector();
218 QList<ModuleBase_ViewerPrs> aPrsList = aSelMgr->selection()->getSelected(ModuleBase_ISelection::Viewer);
219 if (aPrsList.size() > 0) {
220 bool isVisible = false;
221 bool isShading = false;
222 bool canBeShaded = false;
224 foreach(ModuleBase_ViewerPrs aPrs, aPrsList) {
225 aObject = aPrs.object();
226 ResultPtr aRes = std::dynamic_pointer_cast<ModelAPI_Result>(aObject);
227 if (aRes && aRes->isDisplayed()) {
229 canBeShaded = myWorkshop->displayer()->canBeShaded(aObject);
230 isShading = (myWorkshop->displayer()->displayMode(aObject) == XGUI_Displayer::Shading);
236 action("WIREFRAME_CMD")->setEnabled(true);
237 action("SHADING_CMD")->setEnabled(true);
239 action("SHOW_ONLY_CMD")->setEnabled(true);
240 action("HIDE_CMD")->setEnabled(true);
242 action("SHOW_CMD")->setEnabled(true);
244 if (myWorkshop->displayer()->objectsCount() > 0)
245 action("HIDEALL_CMD")->setEnabled(true);
246 if (myWorkshop->canChangeColor())
247 action("COLOR_CMD")->setEnabled(true);
249 action("DELETE_CMD")->setEnabled(true);
251 ModuleBase_IModule* aModule = myWorkshop->module();
253 aModule->updateViewerMenu(myActions);
256 void XGUI_ContextMenuMgr::connectObjectBrowser()
258 connect(myWorkshop->objectBrowser(), SIGNAL(contextMenuRequested(QContextMenuEvent*)), this,
259 SLOT(onContextMenuRequest(QContextMenuEvent*)));
262 void XGUI_ContextMenuMgr::connectViewer()
264 connect(myWorkshop->viewer(), SIGNAL(contextMenuRequested(QContextMenuEvent*)), this,
265 SLOT(onContextMenuRequest(QContextMenuEvent*)));
269 void XGUI_ContextMenuMgr::buildObjBrowserMenu()
271 QAction* aSeparator = new QAction(this);
272 aSeparator->setSeparator(true);
276 // Result construction menu
277 aList.append(action("SHOW_CMD"));
278 aList.append(action("HIDE_CMD"));
279 aList.append(action("SHOW_ONLY_CMD"));
280 aList.append(action("COLOR_CMD"));
281 myObjBrowserMenus[ModelAPI_ResultConstruction::group()] = aList;
282 //-------------------------------------
285 aList.append(action("WIREFRAME_CMD"));
286 aList.append(action("SHADING_CMD"));
287 aList.append(action("COLOR_CMD"));
288 aList.append(mySeparator);
289 aList.append(action("SHOW_CMD"));
290 aList.append(action("HIDE_CMD"));
291 aList.append(action("SHOW_ONLY_CMD"));
292 myObjBrowserMenus[ModelAPI_ResultBody::group()] = aList;
294 myObjBrowserMenus[ModelAPI_ResultGroup::group()] = aList;
296 myObjBrowserMenus[ModelAPI_ResultPart::group()] = aList;
297 //-------------------------------------
300 aList.append(action("DELETE_CMD"));
301 aList.append(action("MOVE_CMD"));
302 myObjBrowserMenus[ModelAPI_Feature::group()] = aList;
305 aList.append(action("DELETE_CMD"));
306 myObjBrowserMenus[ModelAPI_ResultParameter::group()] = aList;
307 //-------------------------------------
310 void XGUI_ContextMenuMgr::buildViewerMenu()
313 // Result construction menu
314 aList.append(action("HIDE_CMD"));
315 aList.append(action("SHOW_ONLY_CMD"));
316 aList.append(action("HIDEALL_CMD"));
317 aList.append(action("COLOR_CMD"));
318 myViewerMenu[ModelAPI_ResultConstruction::group()] = aList;
320 myViewerMenu[ModelAPI_ResultPart::group()] = aList;
321 //-------------------------------------
324 aList.append(action("WIREFRAME_CMD"));
325 aList.append(action("SHADING_CMD"));
326 aList.append(action("COLOR_CMD"));
327 aList.append(mySeparator);
328 aList.append(action("HIDE_CMD"));
329 aList.append(action("SHOW_ONLY_CMD"));
330 aList.append(action("HIDEALL_CMD"));
331 myViewerMenu[ModelAPI_ResultBody::group()] = aList;
333 myViewerMenu[ModelAPI_ResultGroup::group()] = aList;
334 //-------------------------------------
339 void XGUI_ContextMenuMgr::addObjBrowserMenu(QMenu* theMenu) const
341 XGUI_SelectionMgr* aSelMgr = myWorkshop->selector();
342 QObjectPtrList aObjects = aSelMgr->selection()->selectedObjects();
343 int aSelected = aObjects.size();
344 QActionsList aActions;
345 if (aSelected == 1) {
346 ObjectPtr aObject = aObjects.first();
347 std::string aName = aObject->groupName();
348 if (myObjBrowserMenus.contains(aName))
349 aActions = myObjBrowserMenus[aName];
350 } else if (aSelected > 1) {
351 aActions.append(action("SHADING_CMD"));
352 aActions.append(action("WIREFRAME_CMD"));
353 aActions.append(mySeparator);
354 aActions.append(action("SHOW_CMD"));
355 aActions.append(action("HIDE_CMD"));
356 aActions.append(action("SHOW_ONLY_CMD"));
357 aActions.append(mySeparator);
358 aActions.append(action("DELETE_CMD"));
359 //aActions.append(action("MOVE_CMD"));
361 theMenu->addActions(aActions);
363 ModuleBase_IModule* aModule = myWorkshop->module();
365 theMenu->addSeparator();
366 aModule->addObjectBrowserMenu(theMenu);
368 theMenu->addSeparator();
369 theMenu->addActions(myWorkshop->objectBrowser()->actions());
372 void XGUI_ContextMenuMgr::addViewerMenu(QMenu* theMenu) const
374 ModuleBase_IModule* aModule = myWorkshop->module();
376 if (aModule->addViewerMenu(theMenu, myActions))
377 theMenu->addSeparator();
379 XGUI_SelectionMgr* aSelMgr = myWorkshop->selector();
380 QList<ModuleBase_ViewerPrs> aPrsList = aSelMgr->selection()->getSelected(ModuleBase_ISelection::Viewer);
381 int aSelected = aPrsList.size();
382 QActionsList aActions;
383 if (aSelected == 1) {
384 ObjectPtr aObject = aPrsList.first().object();
385 std::string aName = aObject->groupName();
386 if (myViewerMenu.contains(aName))
387 aActions = myViewerMenu[aName];
388 } else if (aSelected > 1) {
389 aActions.append(action("HIDE_CMD"));
391 theMenu->addActions(aActions);
393 if (!myWorkshop->isSalomeMode()) {
394 theMenu->addSeparator();
395 QMdiArea* aMDI = myWorkshop->mainWindow()->mdiArea();
396 if (aMDI->actions().size() > 0) {
397 QMenu* aSubMenu = theMenu->addMenu(tr("Windows"));
398 aSubMenu->addActions(aMDI->actions());
403 QStringList XGUI_ContextMenuMgr::actionObjectGroups(const QString& theName)
407 QMap<std::string, QActionsList>::const_iterator anIt = myObjBrowserMenus.begin(),
408 aLast = myObjBrowserMenus.end();
409 for (; anIt != aLast; anIt++) {
410 QString aGroupName(anIt.key().c_str());
411 if (aGroups.contains(aGroupName))
413 QActionsList anActions = anIt.value();
414 QActionsList::const_iterator anAIt = anActions.begin(), anALast = anActions.end();
416 for (; anAIt != anALast && !aFound; anAIt++)
417 aFound = (*anAIt)->data().toString() == theName;
419 aGroups.append(aGroupName);