src/XGUI/XGUI_ContextMenuMgr.cpp

   1 #include "XGUI_ContextMenuMgr.h"
   2 #include "XGUI_Workshop.h"
   3 #include "XGUI_ObjectsBrowser.h"
   4 #include "XGUI_SelectionMgr.h"
   5 #include "XGUI_Displayer.h"
   6 #include "XGUI_MainWindow.h"
   7 #include "XGUI_ViewerProxy.h"
   8 #include "XGUI_Selection.h"
   9
  10 #include "PartSetPlugin_Part.h"
  11
  12 #include <ModelAPI_Data.h>
  13 #include <ModelAPI_AttributeDocRef.h>
  14 #include <ModelAPI_Object.h>
  15 #include <ModelAPI_ResultPart.h>
  16
  17 #include <QAction>
  18 #include <QContextMenuEvent>
  19 #include <QMenu>
  20 #include <QMdiArea>
  21
  22 XGUI_ContextMenuMgr::XGUI_ContextMenuMgr(XGUI_Workshop* theParent)
  23     : QObject(theParent),
  24       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.png"), tr("Show only"), this);
  50   addAction("SHOW_ONLY_CMD", aAction);
  51
  52   aAction = new QAction(QIcon(":pictures/eye_pencil_closed.png"), tr("Hide"), this);
  53   addAction("HIDE_CMD", aAction);
  54 }
  55
  56 void XGUI_ContextMenuMgr::addAction(const QString& theId, QAction* theAction)
  57 {
  58   if (myActions.contains(theId))
  59     qCritical("A command with Id = '%s' already defined!", qPrintable(theId));
  60   theAction->setData(theId);
  61   connect(theAction, SIGNAL(triggered(bool)), this, SLOT(onAction(bool)));
  62   myActions[theId] = theAction;
  63 }
  64
  65 QAction* XGUI_ContextMenuMgr::action(const QString& theId) const
  66 {
  67   if (myActions.contains(theId))
  68     return myActions[theId];
  69   return 0;
  70 }
  71
  72 QStringList XGUI_ContextMenuMgr::actionIds() const
  73 {
  74   return myActions.keys();
  75 }
  76
  77 void XGUI_ContextMenuMgr::onAction(bool isChecked)
  78 {
  79   QAction* aAction = static_cast<QAction*>(sender());
  80   emit actionTriggered(aAction->data().toString(), isChecked);
  81 }
  82
  83 void XGUI_ContextMenuMgr::updateCommandsStatus()
  84 {
  85 }
  86
  87 void XGUI_ContextMenuMgr::onContextMenuRequest(QContextMenuEvent* theEvent)
  88 {
  89   QMenu* aMenu = 0;
  90   if (sender() == myWorkshop->objectBrowser())
  91     aMenu = objectBrowserMenu();
  92   else if (sender() == myWorkshop->viewer()) {
  93     aMenu = viewerMenu();
  94   }
  95
  96   if (aMenu && (aMenu->actions().size() > 0)) {
  97     aMenu->exec(theEvent->globalPos());
  98     delete aMenu;
  99   }
 100 }
 101
 102 QMenu* XGUI_ContextMenuMgr::objectBrowserMenu() const
 103 {
 104   QMenu* aMenu = new QMenu();
 105   XGUI_SelectionMgr* aSelMgr = myWorkshop->selector();
 106   QList<ObjectPtr> aObjects = aSelMgr->selection()->selectedObjects();
 107   int aSelected = aObjects.size();
 108   if (aSelected > 0) {
 109     SessionPtr aMgr = ModelAPI_Session::get();
 110     XGUI_Displayer* aDisplayer = myWorkshop->displayer();
 111     bool hasResult = false;
 112     bool hasFeature = false;
 113     foreach(ObjectPtr aObj, aObjects)
 114     {
 115       FeaturePtr aFeature = boost::dynamic_pointer_cast<ModelAPI_Feature>(aObj);
 116       ResultPtr aResult = boost::dynamic_pointer_cast<ModelAPI_Result>(aObj);
 117       if (aResult)
 118         hasResult = true;
 119       if (aFeature)
 120         hasFeature = true;
 121       if (hasFeature && hasResult)
 122         break;
 123     }
 124     //Process Feature
 125     if (aSelected == 1) {
 126       ObjectPtr aObject = aObjects.first();
 127       if (aObject) {
 128         ResultPartPtr aPart = boost::dynamic_pointer_cast<ModelAPI_ResultPart>(aObject);
 129         if (aPart) {
 130           if (aMgr->activeDocument() == aPart->partDoc())
 131             aMenu->addAction(action("DEACTIVATE_PART_CMD"));
 132           else
 133             aMenu->addAction(action("ACTIVATE_PART_CMD"));
 134         } else if (hasFeature) {
 135           aMenu->addAction(action("EDIT_CMD"));
 136         } else {
 137           if (aDisplayer->isVisible(aObject))
 138             aMenu->addAction(action("HIDE_CMD"));
 139           else {
 140             aMenu->addAction(action("SHOW_CMD"));
 141           }
 142           aMenu->addAction(action("SHOW_ONLY_CMD"));
 143         }
 144       } else {  // If feature is 0 the it means that selected root object (document)
 145         if (aMgr->activeDocument() != aMgr->moduleDocument())
 146           aMenu->addAction(action("ACTIVATE_PART_CMD"));
 147       }
 148     } else {
 149       if (hasResult) {
 150         aMenu->addAction(action("SHOW_CMD"));
 151         aMenu->addAction(action("HIDE_CMD"));
 152         aMenu->addAction(action("SHOW_ONLY_CMD"));
 153       }
 154     }
 155     if (hasFeature)
 156       aMenu->addAction(action("DELETE_CMD"));
 157   }
 158   aMenu->addActions(myWorkshop->objectBrowser()->actions());
 159   if (aMenu->actions().size() > 0) {
 160     return aMenu;
 161   }
 162   delete aMenu;
 163   return 0;
 164 }
 165
 166 QMenu* XGUI_ContextMenuMgr::viewerMenu() const
 167 {
 168   QMenu* aMenu = new QMenu();
 169   addViewerItems(aMenu);
 170   if (aMenu->actions().size() > 0) {
 171     return aMenu;
 172   }
 173   delete aMenu;
 174   return 0;
 175 }
 176
 177 void XGUI_ContextMenuMgr::addViewerItems(QMenu* theMenu) const
 178 {
 179   XGUI_SelectionMgr* aSelMgr = myWorkshop->selector();
 180   QList<ObjectPtr> aObjects = aSelMgr->selection()->selectedObjects();
 181   if (aObjects.size() > 0) {
 182     //if (aObjects.size() == 1)
 183     //  theMenu->addAction(action("EDIT_CMD"));
 184     bool isVisible = false;
 185     foreach(ObjectPtr aObject, aObjects)
 186     {
 187       ResultPtr aRes = boost::dynamic_pointer_cast<ModelAPI_Result>(aObject);
 188       if (aRes && myWorkshop->displayer()->isVisible(aRes)) {
 189         isVisible = true;
 190         break;
 191       }
 192     }
 193     if (isVisible)
 194       theMenu->addAction(action("HIDE_CMD"));
 195     else
 196       theMenu->addAction(action("SHOW_CMD"));
 197     //theMenu->addAction(action("DELETE_CMD"));
 198   }
 199   if (!myWorkshop->isSalomeMode()) {
 200     QMdiArea* aMDI = myWorkshop->mainWindow()->mdiArea();
 201     if (aMDI->actions().size() > 0) {
 202       QMenu* aSubMenu = theMenu->addMenu(tr("Windows"));
 203       aSubMenu->addActions(aMDI->actions());
 204     }
 205   }
 206 }
 207
 208 void XGUI_ContextMenuMgr::connectObjectBrowser() const
 209 {
 210   connect(myWorkshop->objectBrowser(), SIGNAL(contextMenuRequested(QContextMenuEvent*)), this,
 211           SLOT(onContextMenuRequest(QContextMenuEvent*)));
 212 }
 213
 214 void XGUI_ContextMenuMgr::connectViewer() const
 215 {
 216   connect(myWorkshop->viewer(), SIGNAL(contextMenuRequested(QContextMenuEvent*)), this,
 217           SLOT(onContextMenuRequest(QContextMenuEvent*)));
 218 }
 219