Salome HOME
6dc01fa3dc56900b9ac9b1b60b37e47c669d6605
[modules/shaper.git] / 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   // only viewer selection is processed
144   if (aSelection->getSelected(ModuleBase_ISelection::Viewer).size() == 0) {
145     // it seems that this code is not nesessary anymore. It leads to incorrect case:
146     // sketch operation start, click in any place in the viewer. The result is all nested
147     // entities are enabled(but the sketch plane is not selected yet). Any sketch operation
148     // can be started but will be incorrect on preview build before it uses the sketch unset plane.
149     /*foreach(QString aFeatureId, aIdList) {
150       foreach(QString aId, nestedCommands(aFeatureId)) {
151         setActionEnabled(aId, true);
152       }
153     }*/
154   } else { 
155     SessionPtr aMgr = ModelAPI_Session::get();
156     ModelAPI_ValidatorsFactory* aFactory = aMgr->validators();
157     foreach(QString aFeatureId, aIdList) {
158       foreach(QString aId, nestedCommands(aFeatureId)) {
159         ModelAPI_ValidatorsFactory::Validators aValidators;
160         aFactory->validators(aId.toStdString(), aValidators);
161         ModelAPI_ValidatorsFactory::Validators::iterator aValidatorIt = aValidators.begin();
162         for (; aValidatorIt != aValidators.end(); ++aValidatorIt) {
163           const ModuleBase_SelectionValidator* aSelValidator =
164               dynamic_cast<const ModuleBase_SelectionValidator*>(aFactory->validator(aValidatorIt->first));
165           if (!aSelValidator)
166             continue;
167           setActionEnabled(aId, aSelValidator->isValid(aSelection, aValidatorIt->second));
168         }
169       }
170     }
171   }
172 }
173
174 QKeySequence XGUI_ActionsMgr::registerShortcut(const QKeySequence& theKeySequence)
175 {
176   if (theKeySequence.isEmpty()) {
177     return QKeySequence();
178   }
179   if (myShortcuts.contains(theKeySequence)) {
180     QString aMessage = tr("Shortcut %1 is already defined. Ignore.");
181     aMessage = aMessage.arg(theKeySequence.toString());
182     Events_Error::send(aMessage.toStdString());
183     return QKeySequence();
184   }
185   myShortcuts.append(theKeySequence);
186   return theKeySequence;
187 }
188
189 QKeySequence XGUI_ActionsMgr::registerShortcut(const QString& theKeySequence)
190 {
191   if (theKeySequence.isEmpty()) {
192     return QKeySequence();
193   }
194   QKeySequence aResult(theKeySequence);
195   registerShortcut(aResult);
196   return aResult;
197 }
198
199 void XGUI_ActionsMgr::processEvent(const std::shared_ptr<Events_Message>& theMessage)
200 {
201   const Events_ID kResponseEvent =
202       Events_Loop::loop()->eventByName(EVENT_FEATURE_STATE_RESPONSE);
203   if (theMessage->eventID() == kResponseEvent) {
204     std::shared_ptr<ModelAPI_FeatureStateMessage> aStateMessage =
205         std::dynamic_pointer_cast<ModelAPI_FeatureStateMessage>(theMessage);
206     if (!aStateMessage.get())
207       return;
208     std::list<std::string> aFeaturesList = aStateMessage->features();
209     std::list<std::string>::iterator it = aFeaturesList.begin();
210     for( ; it != aFeaturesList.end(); ++it) {
211       QString anActionId = QString::fromStdString(*it);
212       bool theDefaultState = false;
213       if (myActions.contains(anActionId)) {
214         theDefaultState = myActions[anActionId]->isEnabled();
215       }
216       setActionEnabled(anActionId, aStateMessage->state(*it, theDefaultState));
217     }
218   } else if (theMessage.get()) {
219     #ifdef _DEBUG
220     std::cout << "XGUI_ActionsMgr::processEvent: unhandled message caught: " << std::endl
221               << theMessage->eventID().eventText() << std::endl;
222     #endif
223   }
224 }
225
226 QAction* XGUI_ActionsMgr::operationStateAction(OperationStateActionId theId, QObject* theParent)
227 {
228   QAction* aResult = NULL;
229   if (myOperationActions.contains(theId)) {
230     aResult = myOperationActions.value(theId);
231     if (theParent && aResult->parent() != theParent) {
232       aResult->setParent(theParent);
233     }
234   } else {
235     switch (theId) {
236       case Accept:
237       case AcceptAll:
238         aResult = new QAction(QIcon(":pictures/button_ok.png"), "", theParent);
239         // the default value is disabled, some modifications should be performed in order
240         // to update enable state of this action.
241         // AcceptAll is blinked when sketch is started from enabled to disabled.
242         aResult->setEnabled(false);
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       new ModelAPI_FeatureStateMessage(aStateRequestEventId, this));
361   aMsg->setFeature(anActiveFeature);
362   Events_Loop::loop()->send(aMsg, false);
363 }