src/XGUI/XGUI_ActionsMgr.cpp

   1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D -->
   2
   3 /*
   4  * XGUI_ActionsMgr.cpp
   5  */
   6
   7 #include <AppElements_Command.h>
   8
   9 #include <XGUI_ActionsMgr.h>
  10 #include <XGUI_Workshop.h>
  11 #include <XGUI_OperationMgr.h>
  12 #include <XGUI_SalomeConnector.h>
  13 #include <XGUI_Selection.h>
  14 #include <XGUI_SelectionMgr.h>
  15
  16 #include <Events_Loop.h>
  17 #include <Events_Error.h>
  18
  19 #include <ModelAPI_Session.h>
  20 #include <ModelAPI_Events.h>
  21 #include <ModelAPI_Validator.h>
  22 #include <ModuleBase_Operation.h>
  23 #include <ModuleBase_SelectionValidator.h>
  24
  25
  26 #include <QAction>
  27
  28 #ifdef _DEBUG
  29 #include <iostream>
  30 #include <QDebug>
  31 #endif
  32
  33 XGUI_ActionsMgr::XGUI_ActionsMgr(XGUI_Workshop* theParent)
  34     : QObject(theParent),
  35       myWorkshop(theParent),
  36       myOperationMgr(theParent->operationMgr())
  37 {
  38   // Default shortcuts
  39   myShortcuts << QKeySequence::Save;
  40   myShortcuts << QKeySequence::Undo;
  41   myShortcuts << QKeySequence::Redo;
  42   myShortcuts << QKeySequence::Open;
  43   myShortcuts << QKeySequence::Close;
  44
  45   //Initialize event listening
  46   Events_Loop* aLoop = Events_Loop::loop();
  47   static Events_ID aStateResponseEventId =
  48       Events_Loop::loop()->eventByName(EVENT_FEATURE_STATE_RESPONSE);
  49   aLoop->registerListener(this, aStateResponseEventId, NULL, true);
  50 }
  51
  52 XGUI_ActionsMgr::~XGUI_ActionsMgr()
  53 {
  54 }
  55
  56 void XGUI_ActionsMgr::addCommand(QAction* theCmd)
  57 {
  58   QString aId = theCmd->data().toString();
  59   if (aId.isEmpty()) {
  60     return;
  61   }
  62   myActions.insert(aId, theCmd);
  63   AppElements_Command* aXCmd = dynamic_cast<AppElements_Command*>(theCmd);
  64   if (aXCmd) {
  65     myNestedActions[aId] = aXCmd->nestedCommands();
  66   } else {
  67     XGUI_Workshop* aWorkshop = static_cast<XGUI_Workshop*>(parent());
  68     myNestedActions[aId] = aWorkshop->salomeConnector()->nestedActions(aId);
  69   }
  70 }
  71
  72 void XGUI_ActionsMgr::addNestedCommands(const QString& theId, const QStringList& theCommands)
  73 {
  74   myNestedActions[theId] = theCommands;
  75 }
  76
  77 QStringList XGUI_ActionsMgr::nestedCommands(const QString& theId) const
  78 {
  79   if (myNestedActions.contains(theId))
  80     return myNestedActions[theId];
  81   return QStringList();
  82 }
  83
  84 bool XGUI_ActionsMgr::isNested(const QString& theId) const
  85 {
  86   foreach(QString aId, myNestedActions.keys())
  87   {
  88     QStringList aList = myNestedActions[aId];
  89     if (aList.contains(theId))
  90       return true;
  91   }
  92   return false;
  93 }
  94
  95 void XGUI_ActionsMgr::update()
  96 {
  97   FeaturePtr anActiveFeature = FeaturePtr();
  98   if (myOperationMgr->hasOperation()) {
  99     ModuleBase_Operation* anOperation = myOperationMgr->currentOperation();
 100     anActiveFeature = anOperation->feature();
 101     if(anActiveFeature.get()) {
 102       setAllEnabled(false);
 103       QString aFeatureId = QString::fromStdString(anActiveFeature->getKind());
 104       setActionEnabled(aFeatureId, true);
 105     }
 106     setNestedStackEnabled(anOperation);
 107   } else {
 108     setAllEnabled(true);
 109     setNestedCommandsEnabled(false);
 110   }
 111   // TODO(SBH): Get defaults state of actions from XML and remove the following method
 112   updateByDocumentKind();
 113   updateCheckState();
 114   updateByPlugins(anActiveFeature);
 115 }
 116
 117 void XGUI_ActionsMgr::updateCheckState()
 118 {
 119   QString eachCommand = QString();
 120   foreach(eachCommand, myActions.keys()) {
 121     setActionChecked(eachCommand, false);
 122   }
 123   QStringList ltActiveCommands = myOperationMgr->operationList();
 124   foreach(eachCommand, ltActiveCommands) {
 125     setActionChecked(eachCommand, true);
 126   }
 127 }
 128
 129 void XGUI_ActionsMgr::updateOnViewSelection()
 130 {
 131   if (!myOperationMgr->hasOperation())
 132     return;
 133
 134   QStringList aIdList = myOperationMgr->operationList();
 135   //ModuleBase_Operation* anOperation = myOperationMgr->currentOperation();
 136   //FeaturePtr anActiveFeature = anOperation->feature();
 137   //if(!anActiveFeature.get())
 138   if (aIdList.isEmpty())
 139     return;
 140
 141   //QString aFeatureId = QString::fromStdString(anActiveFeature->getKind());
 142   XGUI_Selection* aSelection = myWorkshop->selector()->selection();
 143   if (aSelection->getSelected().size() == 0) {
 144     // it seems that this code is not nesessary anymore. It leads to incorrect case:
 145     // sketch operation start, click in any place in the viewer. The result is all nested
 146     // entities are enabled(but the sketch plane is not selected yet). Any sketch operation
 147     // can be started but will be incorrect on preview build before it uses the sketch unset plane.
 148     /*foreach(QString aFeatureId, aIdList) {
 149       foreach(QString aId, nestedCommands(aFeatureId)) {
 150         setActionEnabled(aId, true);
 151       }
 152     }*/
 153   } else {
 154     SessionPtr aMgr = ModelAPI_Session::get();
 155     ModelAPI_ValidatorsFactory* aFactory = aMgr->validators();
 156     foreach(QString aFeatureId, aIdList) {
 157       foreach(QString aId, nestedCommands(aFeatureId)) {
 158         std::list<ModelAPI_Validator*> aValidators;
 159         std::list<std::list<std::string> > anArguments;
 160         aFactory->validators(aId.toStdString(), aValidators, anArguments);
 161         std::list<ModelAPI_Validator*>::iterator aValidator = aValidators.begin();
 162         std::list<std::list<std::string> >::iterator aValidatorArgs = anArguments.begin();
 163         for (; aValidator != aValidators.end(); aValidator++, aValidatorArgs++) {
 164           if (!(*aValidator))
 165             continue;
 166           const ModuleBase_SelectionValidator* aSelValidator =
 167               dynamic_cast<const ModuleBase_SelectionValidator*>(*aValidator);
 168           if (!aSelValidator)
 169             continue;
 170           setActionEnabled(aId, aSelValidator->isValid(aSelection, *aValidatorArgs));
 171
 172         }
 173       }
 174     }
 175   }
 176 }
 177
 178 QKeySequence XGUI_ActionsMgr::registerShortcut(const QKeySequence& theKeySequence)
 179 {
 180   if (theKeySequence.isEmpty()) {
 181     return QKeySequence();
 182   }
 183   if (myShortcuts.contains(theKeySequence)) {
 184     QString aMessage = tr("Shortcut %1 is already defined. Ignore.");
 185     aMessage = aMessage.arg(theKeySequence.toString());
 186     Events_Error::send(aMessage.toStdString());
 187     return QKeySequence();
 188   }
 189   myShortcuts.append(theKeySequence);
 190   return theKeySequence;
 191 }
 192
 193 QKeySequence XGUI_ActionsMgr::registerShortcut(const QString& theKeySequence)
 194 {
 195   if (theKeySequence.isEmpty()) {
 196     return QKeySequence();
 197   }
 198   QKeySequence aResult(theKeySequence);
 199   registerShortcut(aResult);
 200   return aResult;
 201 }
 202
 203 void XGUI_ActionsMgr::processEvent(const std::shared_ptr<Events_Message>& theMessage)
 204 {
 205   const Events_ID kResponseEvent =
 206       Events_Loop::loop()->eventByName(EVENT_FEATURE_STATE_RESPONSE);
 207   if (theMessage->eventID() == kResponseEvent) {
 208     std::shared_ptr<ModelAPI_FeatureStateMessage> aStateMessage =
 209         std::dynamic_pointer_cast<ModelAPI_FeatureStateMessage>(theMessage);
 210     if (!aStateMessage.get())
 211       return;
 212     std::list<std::string> aFeaturesList = aStateMessage->features();
 213     std::list<std::string>::iterator it = aFeaturesList.begin();
 214     for( ; it != aFeaturesList.end(); ++it) {
 215       QString anActionId = QString::fromStdString(*it);
 216       bool theDefaultState = false;
 217       if (myActions.contains(anActionId)) {
 218         theDefaultState = myActions[anActionId]->isEnabled();
 219       }
 220       setActionEnabled(anActionId, aStateMessage->state(*it, theDefaultState));
 221     }
 222   } else if (theMessage.get()) {
 223     #ifdef _DEBUG
 224     std::cout << "XGUI_ActionsMgr::processEvent: unhandled message caught: " << std::endl
 225               << theMessage->eventID().eventText() << std::endl;
 226     #endif
 227   }
 228 }
 229
 230 QAction* XGUI_ActionsMgr::operationStateAction(OperationStateActionId theId, QObject* theParent)
 231 {
 232   QAction* aResult = NULL;
 233   if (myOperationActions.contains(theId)) {
 234     aResult = myOperationActions.value(theId);
 235     if (theParent && aResult->parent() != theParent) {
 236       aResult->setParent(theParent);
 237     }
 238   } else {
 239     switch (theId) {
 240       case Accept:
 241       case AcceptAll:
 242         aResult = new QAction(QIcon(":pictures/button_ok.png"), "", theParent);
 243         break;
 244       case Abort:
 245       case AbortAll: {
 246         aResult = new QAction(QIcon(":pictures/button_cancel.png"), "", theParent);
 247         if(theId == Abort) {
 248           aResult->setShortcut(QKeySequence(Qt::Key_Escape));
 249         }
 250       }
 251       break;
 252       case Help:
 253         aResult = new QAction(QIcon(":pictures/button_help.png"), "", theParent);
 254         break;
 255       default:
 256         break;
 257     }
 258     myOperationActions.insert(theId, aResult);
 259   }
 260   return aResult;
 261 }
 262
 263 ActionInfo XGUI_ActionsMgr::actionInfoById(const QString& theId)
 264 {
 265   ActionInfo aResult;
 266   if(myActions.contains(theId)) {
 267     aResult.initFrom(myActions.value(theId));
 268   } else {
 269    aResult.id = theId;
 270    aResult.text = theId;
 271   }
 272   return aResult;
 273 }
 274
 275 void XGUI_ActionsMgr::setAllEnabled(bool isEnabled)
 276 {
 277   foreach(QString eachAction, myActions.keys())
 278   {
 279     setActionEnabled(eachAction, isEnabled);
 280   }
 281 }
 282
 283
 284 //!
 285 void XGUI_ActionsMgr::setNestedCommandsEnabled(bool theEnabled, const QString& theParent)
 286 {
 287   QStringList ltNestedActions;
 288   if (theParent.isEmpty()) {  //Disable ALL nested
 289     foreach(QString eachParent, myNestedActions.keys()) {
 290       ltNestedActions << myNestedActions[eachParent];
 291     }
 292   } else {
 293     ltNestedActions << myNestedActions[theParent];
 294   }
 295   foreach(QString eachNested, ltNestedActions) {
 296     setActionEnabled(eachNested, theEnabled);
 297   }
 298 }
 299
 300 void XGUI_ActionsMgr::setNestedStackEnabled(ModuleBase_Operation* theOperation)
 301 {
 302   if(!theOperation || !theOperation->feature())
 303     return;
 304   FeaturePtr aFeature = theOperation->feature();
 305   QString aFeatureId = QString::fromStdString(aFeature->getKind());
 306   setActionEnabled(aFeatureId, true);
 307   setNestedCommandsEnabled(true, aFeatureId);
 308
 309   setNestedStackEnabled(myOperationMgr->previousOperation(theOperation));
 310 }
 311
 312 void XGUI_ActionsMgr::setActionChecked(const QString& theId, const bool theChecked)
 313 {
 314   if (myActions.contains(theId)) {
 315     QAction* anAction = myActions[theId];
 316     if (anAction->isCheckable()) {
 317       anAction->setChecked(theChecked);
 318     }
 319   }
 320 }
 321
 322 void XGUI_ActionsMgr::setActionEnabled(const QString& theId, const bool theEnabled)
 323 {
 324   if (myActions.contains(theId)) {
 325     myActions[theId]->setEnabled(theEnabled);
 326   }
 327 }
 328
 329 /*
 330  * Disables all actions which have the Document Kind different to
 331  * the current document's kind
 332  */
 333 void XGUI_ActionsMgr::updateByDocumentKind()
 334 {
 335   std::string aStdDocKind = ModelAPI_Session::get()->activeDocument()->kind();
 336   QString aDocKind = QString::fromStdString(aStdDocKind);
 337   XGUI_Workshop* aWorkshop = static_cast<XGUI_Workshop*>(parent());
 338   foreach(QAction* eachAction, myActions.values()) {
 339     AppElements_Command* aCmd = dynamic_cast<AppElements_Command*>(eachAction);
 340     QString aCmdDocKind;
 341     if(aCmd) {
 342       aCmdDocKind = aCmd->documentKind();
 343     } else {
 344       QString aId = eachAction->data().toString();
 345       if (!aId.isEmpty()) {
 346         aCmdDocKind = aWorkshop->salomeConnector()->documentKind(aId);
 347       }
 348     }
 349     if(!aCmdDocKind.isEmpty() && aCmdDocKind != aDocKind) {
 350       eachAction->setEnabled(false);
 351     }
 352   }
 353 }
 354
 355 void XGUI_ActionsMgr::updateByPlugins(FeaturePtr anActiveFeature)
 356 {
 357   static Events_ID aStateRequestEventId = Events_Loop::loop()->eventByName(
 358       EVENT_FEATURE_STATE_REQUEST);
 359   std::shared_ptr<ModelAPI_FeatureStateMessage> aMsg =
 360       std::make_shared<ModelAPI_FeatureStateMessage>(aStateRequestEventId, this);
 361   aMsg->setFeature(anActiveFeature);
 362   Events_Loop::loop()->send(aMsg, false);
 363 }