Salome HOME
Merge branch 'Dev_1.1.0' of newgeom:newgeom.git into Dev_1.1.0
[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   XGUI_Selection* aSelection = myWorkshop->selector()->selection();
132   if (aSelection->getSelected().size() == 0 || !myOperationMgr->hasOperation())
133     return;
134   ModuleBase_Operation* anOperation = myOperationMgr->currentOperation();
135   FeaturePtr anActiveFeature = anOperation->feature();
136   if(!anActiveFeature.get())
137     return;
138   QString aFeatureId = QString::fromStdString(anActiveFeature->getKind());
139
140   SessionPtr aMgr = ModelAPI_Session::get();
141   ModelAPI_ValidatorsFactory* aFactory = aMgr->validators();
142   foreach(QString aId, nestedCommands(aFeatureId)) {
143     std::list<ModelAPI_Validator*> aValidators;
144     std::list<std::list<std::string> > anArguments;
145     if (!anArguments.empty()) {
146       std::list<std::string> firstArg = anArguments.front();
147     }
148     aFactory->validators(aId.toStdString(), aValidators, anArguments);
149     std::list<ModelAPI_Validator*>::iterator aValidator = aValidators.begin();
150     std::list<std::list<std::string> >::iterator aValidatorArgs = anArguments.begin();
151     for (; aValidator != aValidators.end(); aValidator++, aValidatorArgs++) {
152       if (!(*aValidator))
153         continue;
154       const ModuleBase_SelectionValidator* aSelValidator =
155           dynamic_cast<const ModuleBase_SelectionValidator*>(*aValidator);
156       if (!aSelValidator)
157         continue;
158       setActionEnabled(aId, aSelValidator->isValid(aSelection, *aValidatorArgs));
159
160     }
161   }
162 }
163
164 QKeySequence XGUI_ActionsMgr::registerShortcut(const QKeySequence& theKeySequence)
165 {
166   if (theKeySequence.isEmpty()) {
167     return QKeySequence();
168   }
169   if (myShortcuts.contains(theKeySequence)) {
170     QString aMessage = tr("Shortcut %1 is already defined. Ignore.");
171     aMessage = aMessage.arg(theKeySequence.toString());
172     Events_Error::send(aMessage.toStdString());
173     return QKeySequence();
174   }
175   myShortcuts.append(theKeySequence);
176   return theKeySequence;
177 }
178
179 QKeySequence XGUI_ActionsMgr::registerShortcut(const QString& theKeySequence)
180 {
181   if (theKeySequence.isEmpty()) {
182     return QKeySequence();
183   }
184   QKeySequence aResult(theKeySequence);
185   registerShortcut(aResult);
186   return aResult;
187 }
188
189 void XGUI_ActionsMgr::processEvent(const std::shared_ptr<Events_Message>& theMessage)
190 {
191   const Events_ID kResponseEvent =
192       Events_Loop::loop()->eventByName(EVENT_FEATURE_STATE_RESPONSE);
193   if (theMessage->eventID() == kResponseEvent) {
194     std::shared_ptr<ModelAPI_FeatureStateMessage> aStateMessage =
195         std::dynamic_pointer_cast<ModelAPI_FeatureStateMessage>(theMessage);
196     if (!aStateMessage.get())
197       return;
198     std::list<std::string> aFeaturesList = aStateMessage->features();
199     std::list<std::string>::iterator it = aFeaturesList.begin();
200     for( ; it != aFeaturesList.end(); ++it) {
201       QString anActionId = QString::fromStdString(*it);
202       bool theDefaultState = false;
203       if (myActions.contains(anActionId)) {
204         theDefaultState = myActions[anActionId]->isEnabled();
205       }
206       setActionEnabled(anActionId, aStateMessage->state(*it, theDefaultState));
207     }
208   } else if (theMessage.get()) {
209     #ifdef _DEBUG
210     std::cout << "XGUI_ActionsMgr::processEvent: unhandled message caught: " << std::endl
211               << theMessage->eventID().eventText() << std::endl;
212     #endif
213   }
214 }
215
216 void XGUI_ActionsMgr::setAllEnabled(bool isEnabled)
217 {
218   foreach(QString eachAction, myActions.keys())
219   {
220     setActionEnabled(eachAction, isEnabled);
221   }
222 }
223
224
225 //!
226 void XGUI_ActionsMgr::setNestedCommandsEnabled(bool theEnabled, const QString& theParent)
227 {
228   QStringList ltNestedActions;
229   if (theParent.isEmpty()) {  //Disable ALL nested
230     foreach(QString eachParent, myNestedActions.keys()) {
231       ltNestedActions << myNestedActions[eachParent];
232     }
233   } else {
234     ltNestedActions << myNestedActions[theParent];
235   }
236   foreach(QString eachNested, ltNestedActions) {
237     setActionEnabled(eachNested, theEnabled);
238   }
239 }
240
241 void XGUI_ActionsMgr::setNestedStackEnabled(ModuleBase_Operation* theOperation)
242 {
243   if(!theOperation || !theOperation->feature())
244     return;
245   FeaturePtr aFeature = theOperation->feature();
246   QString aFeatureId = QString::fromStdString(aFeature->getKind());
247   setNestedCommandsEnabled(true, aFeatureId);
248
249   setNestedStackEnabled(myOperationMgr->previousOperation(theOperation));
250 }
251
252 void XGUI_ActionsMgr::setActionChecked(const QString& theId, const bool theChecked)
253 {
254   if (myActions.contains(theId)) {
255     QAction* anAction = myActions[theId];
256     if (anAction->isCheckable()) {
257       anAction->setChecked(theChecked);
258     }
259   }
260 }
261
262 void XGUI_ActionsMgr::setActionEnabled(const QString& theId, const bool theEnabled)
263 {
264   if (myActions.contains(theId)) {
265     myActions[theId]->setEnabled(theEnabled);
266   }
267 }
268
269 /*
270  * Disables all actions which have the Document Kind different to
271  * the current document's kind
272  */
273 void XGUI_ActionsMgr::updateByDocumentKind()
274 {
275   std::string aStdDocKind = ModelAPI_Session::get()->activeDocument()->kind();
276   QString aDocKind = QString::fromStdString(aStdDocKind);
277   XGUI_Workshop* aWorkshop = static_cast<XGUI_Workshop*>(parent());
278   foreach(QAction* eachAction, myActions.values()) {
279     AppElements_Command* aCmd = dynamic_cast<AppElements_Command*>(eachAction);
280     QString aCmdDocKind;
281     if(aCmd) {
282       aCmdDocKind = aCmd->documentKind();
283     } else {
284       QString aId = eachAction->data().toString();
285       if (!aId.isEmpty()) {
286         aCmdDocKind = aWorkshop->salomeConnector()->documentKind(aId);
287       }
288     }
289     if(!aCmdDocKind.isEmpty() && aCmdDocKind != aDocKind) {
290       eachAction->setEnabled(false);
291     }
292   }
293 }
294
295 void XGUI_ActionsMgr::updateByPlugins(FeaturePtr anActiveFeature)
296 {
297   static Events_ID aStateRequestEventId = Events_Loop::loop()->eventByName(
298       EVENT_FEATURE_STATE_REQUEST);
299   std::shared_ptr<ModelAPI_FeatureStateMessage> aMsg =
300       std::make_shared<ModelAPI_FeatureStateMessage>(aStateRequestEventId, this);
301   aMsg->setFeature(anActiveFeature);
302   Events_Loop::loop()->send(aMsg, false);
303 }