Salome HOME
Create Extrusion Cut
[modules/shaper.git] / src / ModuleBase / ModuleBase_Operation.cpp
1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
2
3 /*
4  * ModuleBase_Operation.cpp
5  *
6  *  Created on: Apr 2, 2014
7  *      Author: sbh
8  */
9
10 #include "ModuleBase_Operation.h"
11
12 #include "ModuleBase_OperationDescription.h"
13 #include "ModuleBase_ModelWidget.h"
14 #include "ModuleBase_ViewerPrs.h"
15 #include "ModuleBase_IPropertyPanel.h"
16 #include "ModuleBase_ISelection.h"
17 #include "ModuleBase_IViewer.h"
18
19 #include <ModelAPI_AttributeDouble.h>
20 #include <ModelAPI_Document.h>
21 #include <ModelAPI_Feature.h>
22 #include <ModelAPI_Data.h>
23 #include <ModelAPI_Document.h>
24 #include <ModelAPI_Events.h>
25 #include <ModelAPI_Result.h>
26 #include <ModelAPI_Object.h>
27 #include <ModelAPI_Validator.h>
28 #include <ModelAPI_Session.h>
29
30 #include <GeomAPI_Pnt2d.h>
31
32 #include <Events_Loop.h>
33
34 #include <QTimer>
35
36 #ifdef _DEBUG
37 #include <QDebug>
38 #endif
39
40 ModuleBase_Operation::ModuleBase_Operation(const QString& theId, QObject* theParent)
41     : QObject(theParent),
42       myIsEditing(false),
43       myIsModified(false),
44       myPropertyPanel(NULL)
45 {
46   myDescription = new ModuleBase_OperationDescription(theId);
47 }
48
49 ModuleBase_Operation::~ModuleBase_Operation()
50 {
51   delete myDescription;
52   clearPreselection();
53 }
54
55 QString ModuleBase_Operation::id() const
56 {
57   return getDescription()->operationId();
58 }
59
60 FeaturePtr ModuleBase_Operation::feature() const
61 {
62   return myFeature;
63 }
64
65 bool ModuleBase_Operation::isValid() const
66 {
67   if (!myFeature || !myFeature->data()->isValid())
68     return true; // rename operation
69   if (myFeature->isAction())
70     return true;
71   //Get validators for the Id
72   SessionPtr aMgr = ModelAPI_Session::get();
73   ModelAPI_ValidatorsFactory* aFactory = aMgr->validators();
74   bool aValid = aFactory->validate(myFeature);
75
76   // the feature exec state should be checked in order to do not apply features, which result can not
77   // be built. E.g. extrusion on sketch, where the "to" is a perpendicular plane to the sketch
78   bool isDone = ( myFeature->data()->execState() == ModelAPI_StateDone
79                || myFeature->data()->execState() == ModelAPI_StateMustBeUpdated );
80
81   return aValid && isDone;
82 }
83
84
85 bool ModuleBase_Operation::canBeCommitted() const
86 {
87   return isValid();
88 }
89
90 FeaturePtr ModuleBase_Operation::createFeature(const bool theFlushMessage)
91 {
92   if (myParentFeature.get()) {
93     myFeature = myParentFeature->addFeature(getDescription()->operationId().toStdString());
94   } else {
95     std::shared_ptr<ModelAPI_Document> aDoc = ModelAPI_Session::get()->activeDocument();
96     myFeature = aDoc->addFeature(getDescription()->operationId().toStdString());
97   }
98   if (myFeature) {  // TODO: generate an error if feature was not created
99     myIsModified = true;
100     // Model update should call "execute" of a feature.
101     //myFeature->execute();
102     // Init default values
103     /*QList<ModuleBase_ModelWidget*> aWidgets = getDescription()->modelWidgets();
104      QList<ModuleBase_ModelWidget*>::const_iterator anIt = aWidgets.begin(), aLast = aWidgets.end();
105      for (; anIt != aLast; anIt++) {
106      (*anIt)->storeValue(aFeature);
107      }*/
108   }
109
110   if (theFlushMessage)
111     Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_CREATED));
112   return myFeature;
113 }
114
115 void ModuleBase_Operation::setFeature(FeaturePtr theFeature)
116 {
117   myFeature = theFeature;
118   myIsEditing = true;
119 }
120
121 bool ModuleBase_Operation::hasObject(ObjectPtr theObj) const
122 {
123   FeaturePtr aFeature = feature();
124   if (aFeature) {
125     if (aFeature == theObj)
126       return true;
127     std::list<ResultPtr> aResults = aFeature->results();
128     std::list<ResultPtr>::const_iterator aIt;
129     for (aIt = aResults.cbegin(); aIt != aResults.cend(); ++aIt) {
130       if (theObj == (*aIt))
131         return true;
132     }
133   }
134   return false;
135 }
136
137 void ModuleBase_Operation::start()
138 {
139   QString anId = getDescription()->operationId();
140   if (myIsEditing) {
141     anId = anId.append(EditSuffix());
142   }
143   ModelAPI_Session::get()->startOperation(anId.toStdString());
144
145   if (!myIsEditing) {
146     FeaturePtr aFeature = createFeature();
147     // if the feature is not created, there is no sense to start the operation
148     if (aFeature.get() == NULL) {
149       // it is necessary to abor the operation in the session and emit the aborted signal
150       // in order to update commands status in the workshop, to be exact the feature action
151       // to be unchecked
152       abort();
153       return;
154     }
155   }
156   /// Set current feature and remeber old current feature
157   if (myIsEditing) {
158     SessionPtr aMgr = ModelAPI_Session::get();
159     DocumentPtr aDoc = aMgr->activeDocument();
160     myCurrentFeature = aDoc->currentFeature(true);
161     aDoc->setCurrentFeature(feature(), false);
162   }
163
164   startOperation();
165   emit started();
166
167 }
168
169 void ModuleBase_Operation::postpone()
170 {
171   postponeOperation();
172   emit postponed();
173 }
174
175 void ModuleBase_Operation::resume()
176 {
177   resumeOperation();
178   emit resumed();
179 }
180
181 void ModuleBase_Operation::abort()
182 {
183   // the viewer update should be blocked in order to avoid the features blinking before they are
184   // hidden
185   std::shared_ptr<Events_Message> aMsg = std::shared_ptr<Events_Message>(
186       new Events_Message(Events_Loop::eventByName(EVENT_UPDATE_VIEWER_BLOCKED)));
187   Events_Loop::loop()->send(aMsg);
188
189   SessionPtr aMgr = ModelAPI_Session::get();
190   if (myIsEditing) {
191     DocumentPtr aDoc = aMgr->activeDocument();
192     bool aIsOp = aMgr->isOperation();
193     if (!aIsOp)
194       aMgr->startOperation();
195     aDoc->setCurrentFeature(myCurrentFeature, true);
196     if (!aIsOp)
197       aMgr->finishOperation();
198     myCurrentFeature = FeaturePtr();
199   }
200   abortOperation();
201   emit aborted();
202
203   stopOperation();
204   // is is necessary to deactivate current widgets before the model operation is aborted
205   // because abort removes the feature and activated filters should not check it
206   propertyPanel()->cleanContent();
207
208   aMgr->abortOperation();
209   emit stopped();
210   // the viewer update should be unblocked in order to avoid the features blinking before they are
211   // hidden
212   aMsg = std::shared_ptr<Events_Message>(
213                 new Events_Message(Events_Loop::eventByName(EVENT_UPDATE_VIEWER_UNBLOCKED)));
214
215   Events_Loop::loop()->send(aMsg);
216 }
217
218 bool ModuleBase_Operation::commit()
219 {
220   if (canBeCommitted()) {
221     SessionPtr aMgr = ModelAPI_Session::get();
222     /// Set current feature and remeber old current feature
223     if (myIsEditing) {
224       DocumentPtr aDoc = aMgr->activeDocument();
225       bool aIsOp = aMgr->isOperation();
226       if (!aIsOp)
227         aMgr->startOperation();
228       aDoc->setCurrentFeature(myCurrentFeature, true);
229       if (!aIsOp)
230         aMgr->finishOperation();
231       myCurrentFeature = FeaturePtr();
232     }
233     commitOperation();
234     // check whether there are modifications performed during the current operation
235     // in the model
236     // in case if there are no modifications, do not increase the undo/redo stack
237     if (aMgr->isModified())
238       aMgr->finishOperation();
239     else
240       aMgr->abortOperation();
241
242     stopOperation();
243     emit stopped();
244     emit committed();
245
246     afterCommitOperation();
247     return true;
248   }
249   return false;
250 }
251
252 void ModuleBase_Operation::setRunning(bool theState)
253 {
254   emit triggered(theState);
255 }
256
257 //TODO: nds stabilization hotfix
258 void ModuleBase_Operation::commitOperation()
259 {
260   if(!myPropertyPanel) {
261     return;
262   }
263   ModuleBase_ModelWidget* aWidget = myPropertyPanel->activeWidget();
264   if (aWidget)
265     aWidget->disconnectSignals();
266 }
267
268 void ModuleBase_Operation::activateByPreselection()
269 {
270   if (!myPropertyPanel || myPreSelection.empty()) {
271     myPropertyPanel->activateNextWidget(NULL);
272     return;
273   }
274   const QList<ModuleBase_ModelWidget*>& aWidgets = myPropertyPanel->modelWidgets();
275   if (aWidgets.empty()) {
276     myPropertyPanel->activateNextWidget(NULL);
277     return;
278   }
279   
280   ModuleBase_ModelWidget* aWgt, *aFilledWgt = 0;
281   QList<ModuleBase_ModelWidget*>::const_iterator aWIt;
282   bool isSet = false;
283   // 1. apply the selection to controls
284   int aCurrentPosition = 0;
285   for (aWIt = aWidgets.constBegin(); aWIt != aWidgets.constEnd(); ++aWIt) {
286     aWgt = (*aWIt);
287     if (!aWgt->canSetValue())
288       continue;
289
290     if (!aWgt->setSelection(myPreSelection, aCurrentPosition/*aValue*/)) {
291       isSet = false;
292       break;
293     } else {
294       isSet = true;
295       aFilledWgt = aWgt;
296     }
297   }
298   // 2. ignore not obligatory widgets
299   /*for (; aWIt != aWidgets.constEnd(); ++aWIt) {
300     aWgt = (*aWIt);
301     if (aWgt && aWgt->isObligatory())
302       continue;
303     aFilledWgt = aWgt;
304   }*/
305
306   // 3. a signal should be emitted before the next widget activation
307   // because, the activation of the next widget will give a focus to the widget. As a result
308   // the value of the widget is initialized. And commit may happens until the value is entered.
309   if (aFilledWgt)
310     emit activatedByPreselection();
311
312   // 4. activate the next obligatory widget
313   myPropertyPanel->activateNextWidget(aFilledWgt);
314 }
315
316 void ModuleBase_Operation::setParentFeature(CompositeFeaturePtr theParent)
317 {
318   myParentFeature = theParent;
319 }
320
321 CompositeFeaturePtr ModuleBase_Operation::parentFeature() const
322 {
323   return myParentFeature;
324 }
325
326 void ModuleBase_Operation::initSelection(ModuleBase_ISelection* theSelection,
327                                          ModuleBase_IViewer* theViewer)
328 {
329   clearPreselection();
330
331   QList<ModuleBase_ViewerPrs> aPreSelected;
332   // Check that the selected result are not results of operation feature
333   FeaturePtr aFeature = feature();
334   if (aFeature) {
335     QList<ModuleBase_ViewerPrs> aSelected = theSelection->getSelected();
336
337     std::list<ResultPtr> aResults = aFeature->results();
338     QObjectPtrList aResList;
339     std::list<ResultPtr>::const_iterator aIt;
340     for (aIt = aResults.begin(); aIt != aResults.end(); ++aIt)
341       aResList.append(*aIt);
342
343     foreach (ModuleBase_ViewerPrs aPrs, aSelected) {
344       if ((!aResList.contains(aPrs.object())) && (aPrs.object() != aFeature))
345         aPreSelected.append(aPrs);
346     }
347   } else
348     aPreSelected = theSelection->getSelected();
349
350   // convert the selection values to the values, which are set to the operation widgets
351
352   //Handle(V3d_View) aView = theViewer->activeView();
353   //foreach (ModuleBase_ViewerPrs aPrs, aPreSelected) {
354   //  ModuleBase_WidgetValueFeature* aValue = new ModuleBase_WidgetValueFeature();
355   //  aValue->setObject(aPrs.object());
356
357   //  double aX, anY;
358   //  if (getViewerPoint(aPrs, theViewer, aX, anY))
359   //    aValue->setPoint(std::shared_ptr<GeomAPI_Pnt2d>(new GeomAPI_Pnt2d(aX, anY)));
360   //  myPreSelection.append(aValue);
361   //}
362   myPreSelection = aPreSelected;
363 }
364
365 //void ModuleBase_Operation::onWidgetActivated(ModuleBase_ModelWidget* theWidget)
366 //{
367 //  //activateByPreselection();
368 //  //if (theWidget && myPropertyPanel) {
369 //  //  myPropertyPanel->activateNextWidget();
370 //  ////  //emit activateNextWidget(myActiveWidget);
371 //  //}
372 //}
373
374 //bool ModuleBase_Operation::setWidgetValue(ObjectPtr theFeature, double theX, double theY)
375 //{
376 //  ModuleBase_ModelWidget* aActiveWgt = myPropertyPanel->activeWidget();
377 //  if (!aActiveWgt)
378 //    return false;
379 //  ModuleBase_WidgetValueFeature* aValue = new ModuleBase_WidgetValueFeature();
380 //  aValue->setObject(theFeature);
381 //  aValue->setPoint(std::shared_ptr<GeomAPI_Pnt2d>(new GeomAPI_Pnt2d(theX, theY)));
382 //  bool isApplyed = aActiveWgt->setValue(aValue);
383 //
384 //  delete aValue;
385 //  myIsModified = (myIsModified || isApplyed);
386 //  return isApplyed;
387 //}
388
389 bool ModuleBase_Operation::getViewerPoint(ModuleBase_ViewerPrs thePrs,
390                                                ModuleBase_IViewer* theViewer,
391                                                double& theX, double& theY)
392 {
393   return false;
394 }
395
396 void ModuleBase_Operation::clearPreselection()
397 {
398   myPreSelection.clear();
399 }
400
401 void ModuleBase_Operation::setPropertyPanel(ModuleBase_IPropertyPanel* theProp) 
402
403   myPropertyPanel = theProp; 
404   myPropertyPanel->setEditingMode(isEditOperation());
405
406   // Do not activate widgets by default if the current operation is editing operation
407   // Because we don't know which widget is going to be edited. 
408   if (!isEditOperation())
409     activateByPreselection();
410 }
411
412 bool ModuleBase_Operation::isGranted(QString theId) const
413 {
414   return myNestedFeatures.contains(theId);
415 }