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)
  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   return aFactory->validate(myFeature);
  75 }
  76
  77
  78 bool ModuleBase_Operation::canBeCommitted() const
  79 {
  80   return isValid();
  81 }
  82
  83 FeaturePtr ModuleBase_Operation::createFeature(const bool theFlushMessage)
  84 {
  85   if (myParentFeature.get()) {
  86     myFeature = myParentFeature->addFeature(getDescription()->operationId().toStdString());
  87   } else {
  88     std::shared_ptr<ModelAPI_Document> aDoc = ModelAPI_Session::get()->activeDocument();
  89     myFeature = aDoc->addFeature(getDescription()->operationId().toStdString());
  90   }
  91   if (myFeature) {  // TODO: generate an error if feature was not created
  92     myIsModified = true;
  93     // Model update should call "execute" of a feature.
  94     //myFeature->execute();
  95     // Init default values
  96     /*QList<ModuleBase_ModelWidget*> aWidgets = getDescription()->modelWidgets();
  97      QList<ModuleBase_ModelWidget*>::const_iterator anIt = aWidgets.begin(), aLast = aWidgets.end();
  98      for (; anIt != aLast; anIt++) {
  99      (*anIt)->storeValue(aFeature);
 100      }*/
 101   }
 102
 103   if (theFlushMessage)
 104     Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_CREATED));
 105   return myFeature;
 106 }
 107
 108 void ModuleBase_Operation::setFeature(FeaturePtr theFeature)
 109 {
 110   myFeature = theFeature;
 111   myIsEditing = true;
 112 }
 113
 114 bool ModuleBase_Operation::hasObject(ObjectPtr theObj) const
 115 {
 116   FeaturePtr aFeature = feature();
 117   if (aFeature) {
 118     if (aFeature == theObj)
 119       return true;
 120     std::list<ResultPtr> aResults = aFeature->results();
 121     std::list<ResultPtr>::const_iterator aIt;
 122     for (aIt = aResults.cbegin(); aIt != aResults.cend(); ++aIt) {
 123       if (theObj == (*aIt))
 124         return true;
 125     }
 126   }
 127   return false;
 128 }
 129
 130 void ModuleBase_Operation::start()
 131 {
 132   QString anId = getDescription()->operationId();
 133   if (myIsEditing) {
 134     anId = anId.append(EditSuffix());
 135   }
 136   ModelAPI_Session::get()->startOperation(anId.toStdString());
 137
 138   if (!myIsEditing) {
 139     FeaturePtr aFeature = createFeature();
 140     // if the feature is not created, there is no sense to start the operation
 141     if (aFeature.get() == NULL) {
 142       // it is necessary to abor the operation in the session and emit the aborted signal
 143       // in order to update commands status in the workshop, to be exact the feature action
 144       // to be unchecked
 145       abort();
 146       return;
 147     }
 148   }
 149   /// Set current feature and remeber old current feature
 150   if (myIsEditing) {
 151     SessionPtr aMgr = ModelAPI_Session::get();
 152     DocumentPtr aDoc = aMgr->activeDocument();
 153     myCurrentFeature = aDoc->currentFeature(true);
 154     aDoc->setCurrentFeature(feature(), false);
 155   }
 156
 157   startOperation();
 158   emit started();
 159
 160 }
 161
 162 void ModuleBase_Operation::postpone()
 163 {
 164   postponeOperation();
 165   emit postponed();
 166 }
 167
 168 void ModuleBase_Operation::resume()
 169 {
 170   resumeOperation();
 171   emit resumed();
 172 }
 173
 174 void ModuleBase_Operation::abort()
 175 {
 176   if (myIsEditing) {
 177     SessionPtr aMgr = ModelAPI_Session::get();
 178     DocumentPtr aDoc = aMgr->activeDocument();
 179     aDoc->setCurrentFeature(myCurrentFeature, true);
 180     myCurrentFeature = FeaturePtr();
 181   }
 182   abortOperation();
 183   emit aborted();
 184
 185   stopOperation();
 186
 187   ModelAPI_Session::get()->abortOperation();
 188   emit stopped();
 189 }
 190
 191 bool ModuleBase_Operation::commit()
 192 {
 193   if (canBeCommitted()) {
 194     SessionPtr aMgr = ModelAPI_Session::get();
 195     /// Set current feature and remeber old current feature
 196     if (myIsEditing) {
 197       DocumentPtr aDoc = aMgr->activeDocument();
 198       bool aIsOp = aMgr->isOperation();
 199       if (!aIsOp)
 200         aMgr->startOperation();
 201       aDoc->setCurrentFeature(myCurrentFeature, true);
 202       if (!aIsOp)
 203         aMgr->finishOperation();
 204       myCurrentFeature = FeaturePtr();
 205     }
 206     commitOperation();
 207     // check whether there are modifications performed during the current operation
 208     // in the model
 209     // in case if there are no modifications, do not increase the undo/redo stack
 210     if (aMgr->isModified())
 211       aMgr->finishOperation();
 212     else
 213       aMgr->abortOperation();
 214
 215     stopOperation();
 216     emit stopped();
 217     emit committed();
 218
 219     afterCommitOperation();
 220     return true;
 221   }
 222   return false;
 223 }
 224
 225 void ModuleBase_Operation::setRunning(bool theState)
 226 {
 227   emit triggered(theState);
 228 }
 229
 230 void ModuleBase_Operation::activateByPreselection()
 231 {
 232   if (!myPropertyPanel || myPreSelection.empty()) {
 233     myPropertyPanel->activateNextWidget(NULL);
 234     return;
 235   }
 236   const QList<ModuleBase_ModelWidget*>& aWidgets = myPropertyPanel->modelWidgets();
 237   if (aWidgets.empty()) {
 238     myPropertyPanel->activateNextWidget(NULL);
 239     return;
 240   }
 241
 242   ModuleBase_ModelWidget* aWgt, *aFilledWgt = 0;
 243   QList<ModuleBase_ModelWidget*>::const_iterator aWIt;
 244   bool isSet = false;
 245   // 1. apply the selection to controls
 246   int aCurrentPosition = 0;
 247   for (aWIt = aWidgets.constBegin(); aWIt != aWidgets.constEnd(); ++aWIt) {
 248     aWgt = (*aWIt);
 249     if (!aWgt->canSetValue())
 250       continue;
 251
 252     if (!aWgt->setSelection(myPreSelection, aCurrentPosition/*aValue*/)) {
 253       isSet = false;
 254       break;
 255     } else {
 256       isSet = true;
 257       aFilledWgt = aWgt;
 258     }
 259   }
 260   // 2. ignore not obligatory widgets
 261   /*for (; aWIt != aWidgets.constEnd(); ++aWIt) {
 262     aWgt = (*aWIt);
 263     if (aWgt && aWgt->isObligatory())
 264       continue;
 265     aFilledWgt = aWgt;
 266   }*/
 267
 268   // 3. activate the next obligatory widget
 269   myPropertyPanel->activateNextWidget(aFilledWgt);
 270   if (aFilledWgt)
 271     emit activatedByPreselection();
 272
 273 }
 274
 275 void ModuleBase_Operation::setParentFeature(CompositeFeaturePtr theParent)
 276 {
 277   myParentFeature = theParent;
 278 }
 279
 280 CompositeFeaturePtr ModuleBase_Operation::parentFeature() const
 281 {
 282   return myParentFeature;
 283 }
 284
 285 void ModuleBase_Operation::initSelection(ModuleBase_ISelection* theSelection,
 286                                          ModuleBase_IViewer* theViewer)
 287 {
 288   clearPreselection();
 289
 290   QList<ModuleBase_ViewerPrs> aPreSelected;
 291   // Check that the selected result are not results of operation feature
 292   FeaturePtr aFeature = feature();
 293   if (aFeature) {
 294     QList<ModuleBase_ViewerPrs> aSelected = theSelection->getSelected();
 295
 296     std::list<ResultPtr> aResults = aFeature->results();
 297     QObjectPtrList aResList;
 298     std::list<ResultPtr>::const_iterator aIt;
 299     for (aIt = aResults.begin(); aIt != aResults.end(); ++aIt)
 300       aResList.append(*aIt);
 301
 302     foreach (ModuleBase_ViewerPrs aPrs, aSelected) {
 303       if ((!aResList.contains(aPrs.object())) && (aPrs.object() != aFeature))
 304         aPreSelected.append(aPrs);
 305     }
 306   } else
 307     aPreSelected = theSelection->getSelected();
 308
 309   // convert the selection values to the values, which are set to the operation widgets
 310
 311   //Handle(V3d_View) aView = theViewer->activeView();
 312   //foreach (ModuleBase_ViewerPrs aPrs, aPreSelected) {
 313   //  ModuleBase_WidgetValueFeature* aValue = new ModuleBase_WidgetValueFeature();
 314   //  aValue->setObject(aPrs.object());
 315
 316   //  double aX, anY;
 317   //  if (getViewerPoint(aPrs, theViewer, aX, anY))
 318   //    aValue->setPoint(std::shared_ptr<GeomAPI_Pnt2d>(new GeomAPI_Pnt2d(aX, anY)));
 319   //  myPreSelection.append(aValue);
 320   //}
 321   myPreSelection = aPreSelected;
 322 }
 323
 324 //void ModuleBase_Operation::onWidgetActivated(ModuleBase_ModelWidget* theWidget)
 325 //{
 326 //  //activateByPreselection();
 327 //  //if (theWidget && myPropertyPanel) {
 328 //  //  myPropertyPanel->activateNextWidget();
 329 //  ////  //emit activateNextWidget(myActiveWidget);
 330 //  //}
 331 //}
 332
 333 //bool ModuleBase_Operation::setWidgetValue(ObjectPtr theFeature, double theX, double theY)
 334 //{
 335 //  ModuleBase_ModelWidget* aActiveWgt = myPropertyPanel->activeWidget();
 336 //  if (!aActiveWgt)
 337 //    return false;
 338 //  ModuleBase_WidgetValueFeature* aValue = new ModuleBase_WidgetValueFeature();
 339 //  aValue->setObject(theFeature);
 340 //  aValue->setPoint(std::shared_ptr<GeomAPI_Pnt2d>(new GeomAPI_Pnt2d(theX, theY)));
 341 //  bool isApplyed = aActiveWgt->setValue(aValue);
 342 //
 343 //  delete aValue;
 344 //  myIsModified = (myIsModified || isApplyed);
 345 //  return isApplyed;
 346 //}
 347
 348 bool ModuleBase_Operation::getViewerPoint(ModuleBase_ViewerPrs thePrs,
 349                                                ModuleBase_IViewer* theViewer,
 350                                                double& theX, double& theY)
 351 {
 352   return false;
 353 }
 354
 355 void ModuleBase_Operation::clearPreselection()
 356 {
 357   myPreSelection.clear();
 358 }
 359
 360 void ModuleBase_Operation::setPropertyPanel(ModuleBase_IPropertyPanel* theProp)
 361 {
 362   myPropertyPanel = theProp;
 363   myPropertyPanel->setEditingMode(isEditOperation());
 364
 365   // Do not activate widgets by default if the current operation is editing operation
 366   // Because we don't know which widget is going to be edited.
 367   if (!isEditOperation())
 368     activateByPreselection();
 369 }
 370
 371 bool ModuleBase_Operation::isGranted(QString theId) const
 372 {
 373   return myNestedFeatures.contains(theId);
 374 }