Salome HOME
Merge branch 'master' of newgeom:newgeom.git
[modules/shaper.git] / src / XGUI / XGUI_ContextMenuMgr.cpp
1
2 #include "XGUI_ContextMenuMgr.h"
3 #include "XGUI_Workshop.h"
4 #include "XGUI_ObjectsBrowser.h"
5 #include "XGUI_SelectionMgr.h"
6 #include "XGUI_Displayer.h"
7 #include "XGUI_MainWindow.h"
8 #include "XGUI_ViewerProxy.h"
9 #include "XGUI_Selection.h"
10
11 #include "PartSetPlugin_Part.h"
12
13 #include <ModelAPI_Data.h>
14 #include <ModelAPI_AttributeDocRef.h>
15 #include <ModelAPI_Object.h>
16 #include <ModelAPI_ResultPart.h>
17
18 #include <QAction>
19 #include <QContextMenuEvent>
20 #include <QMenu>
21 #include <QMdiArea>
22
23 XGUI_ContextMenuMgr::XGUI_ContextMenuMgr(XGUI_Workshop* theParent) :
24 QObject(theParent), myWorkshop(theParent)
25 {
26 }
27
28 XGUI_ContextMenuMgr::~XGUI_ContextMenuMgr()
29 {
30 }
31
32 void XGUI_ContextMenuMgr::createActions()
33 {
34   QAction* aAction = new QAction(QIcon(":pictures/edit.png"), tr("Edit..."), this);
35   addAction("EDIT_CMD", aAction);
36
37   aAction = new QAction(QIcon(":pictures/activate.png"), tr("Activate"), this);
38   addAction("ACTIVATE_PART_CMD", aAction);
39
40   aAction = new QAction(QIcon(":pictures/assembly.png"), tr("Deactivate"), this);
41   addAction("DEACTIVATE_PART_CMD", aAction);
42
43   aAction = new QAction(QIcon(":pictures/delete.png"), tr("Delete"), this);
44   addAction("DELETE_CMD", aAction);
45
46   aAction = new QAction(QIcon(":pictures/eye_pencil.png"), tr("Show"), this);
47   addAction("SHOW_CMD", aAction);
48
49   aAction = new QAction(QIcon(":pictures/eye_pencil_closed.png"), tr("Hide"), this);
50   addAction("HIDE_CMD", aAction);
51 }
52
53 void XGUI_ContextMenuMgr::addAction(const QString& theId, QAction* theAction)
54 {
55   if (myActions.contains(theId))
56     qCritical("A command with Id = '%s' already defined!", qPrintable(theId));
57   theAction->setData(theId);
58   connect(theAction, SIGNAL(triggered(bool)), this, SLOT(onAction(bool)));
59   myActions[theId] = theAction;
60 }
61
62 QAction* XGUI_ContextMenuMgr::action(const QString& theId) const
63 {
64   if (myActions.contains(theId))
65     return myActions[theId];
66   return 0;
67 }
68
69 QStringList XGUI_ContextMenuMgr::actionIds() const
70 {
71   return myActions.keys();
72 }
73
74 void XGUI_ContextMenuMgr::onAction(bool isChecked)
75 {
76   QAction* aAction = static_cast<QAction*>(sender());
77   emit actionTriggered(aAction->data().toString(), isChecked);
78 }
79
80 void XGUI_ContextMenuMgr::updateCommandsStatus()
81 {
82 }
83
84 void XGUI_ContextMenuMgr::onContextMenuRequest(QContextMenuEvent* theEvent)
85 {
86   QMenu* aMenu = 0;
87   if (sender() == myWorkshop->objectBrowser())
88     aMenu = objectBrowserMenu();
89   else if (sender() == myWorkshop->viewer()) {
90     aMenu = viewerMenu();
91   }
92   
93   if (aMenu && (aMenu->actions().size() > 0)) {
94     aMenu->exec(theEvent->globalPos());
95     delete aMenu;
96   }
97 }
98
99 QMenu* XGUI_ContextMenuMgr::objectBrowserMenu() const
100 {
101   QMenu* aMenu = new QMenu();
102   XGUI_SelectionMgr* aSelMgr = myWorkshop->selector();
103   QList<ObjectPtr> aObjects = aSelMgr->selection()->selectedObjects();
104   if (aObjects.size() == 1) {
105     PluginManagerPtr aMgr = ModelAPI_PluginManager::get();
106     ObjectPtr aObject = aObjects.first();
107     //Process Feature
108     if (aObject) {
109       ResultPartPtr aPart = boost::dynamic_pointer_cast<ModelAPI_ResultPart>(aObject);
110       if (aPart) {
111         DocumentPtr aFeaDoc = aPart->document();
112         if (aMgr->currentDocument() == aFeaDoc)
113           aMenu->addAction(action("DEACTIVATE_PART_CMD"));
114         else 
115           aMenu->addAction(action("ACTIVATE_PART_CMD"));
116       } else {
117         ResultPtr aResult = boost::dynamic_pointer_cast<ModelAPI_Result>(aObject);
118         if (aResult) {
119           aMenu->addAction(action("EDIT_CMD"));
120
121           XGUI_Displayer* aDisplayer = myWorkshop->displayer();
122           if (aDisplayer->isVisible(aResult))
123             aMenu->addAction(action("HIDE_CMD"));
124           else
125             aMenu->addAction(action("SHOW_CMD"));
126         }
127       }
128       aMenu->addAction(action("DELETE_CMD"));
129       aMenu->addSeparator();
130
131     // Process Root object (document)
132     } else { // If feature is 0 the it means that selected root object (document)
133       if (aMgr->currentDocument() != aMgr->rootDocument()) {
134         aMenu->addAction(action("ACTIVATE_PART_CMD"));
135       }
136     }
137   }
138   aMenu->addActions(myWorkshop->objectBrowser()->actions());
139   if (aMenu->actions().size() > 0) {
140     return aMenu;
141   }
142   delete aMenu;
143   return 0;
144 }
145
146 QMenu* XGUI_ContextMenuMgr::viewerMenu() const
147 {
148   QMenu* aMenu = new QMenu();
149   addViewerItems(aMenu);
150   if (aMenu->actions().size() > 0) {
151     return aMenu;
152   }
153   delete aMenu;
154   return 0;
155 }
156
157 void XGUI_ContextMenuMgr::addViewerItems(QMenu* theMenu) const
158 {
159   XGUI_SelectionMgr* aSelMgr = myWorkshop->selector();
160   QList<ObjectPtr> aObjects = aSelMgr->selection()->selectedObjects();
161   if (aObjects.size() > 0) {
162     if (aObjects.size() == 1)
163       theMenu->addAction(action("EDIT_CMD"));
164     bool isVisible = false;
165     foreach(ObjectPtr aObject, aObjects) {
166       ResultPtr aRes = boost::dynamic_pointer_cast<ModelAPI_Result>(aObject);
167       if (aRes && myWorkshop->displayer()->isVisible(aRes)) {
168         isVisible = true;
169         break;
170       }
171     }
172     if (isVisible)
173       theMenu->addAction(action("HIDE_CMD"));
174     else 
175       theMenu->addAction(action("SHOW_CMD"));
176     theMenu->addAction(action("DELETE_CMD"));
177   }
178   if (!myWorkshop->isSalomeMode()) {
179     QMdiArea* aMDI = myWorkshop->mainWindow()->mdiArea();
180     if (aMDI->actions().size() > 0) {
181       QMenu* aSubMenu = theMenu->addMenu(tr("Windows"));
182       aSubMenu->addActions(aMDI->actions());
183     }
184   }
185 }
186
187 void XGUI_ContextMenuMgr::connectObjectBrowser() const
188 {
189   connect(myWorkshop->objectBrowser(), SIGNAL(contextMenuRequested(QContextMenuEvent*)), 
190     this, SLOT(onContextMenuRequest(QContextMenuEvent*)));
191 }
192
193 void XGUI_ContextMenuMgr::connectViewer() const
194 {
195   connect(myWorkshop->viewer(), SIGNAL(contextMenuRequested(QContextMenuEvent*)), 
196     this, SLOT(onContextMenuRequest(QContextMenuEvent*)));
197 }
198