src/PartSet/PartSet_Module.cpp

   1 #include "PartSet_Module.h"
   2 #include <PartSet_OperationSketch.h>
   3 #include <PartSet_WidgetSketchLabel.h>
   4 #include <PartSet_Validators.h>
   5 #include <PartSet_Tools.h>
   6 #include <PartSet_WidgetPoint2D.h>
   7 #include <PartSet_WidgetPoint2dDistance.h>
   8 #include <PartSet_WidgetShapeSelector.h>
   9
  10 #include <ModuleBase_Operation.h>
  11 #include <ModuleBase_IViewer.h>
  12 #include <ModuleBase_IViewWindow.h>
  13 #include <ModuleBase_IPropertyPanel.h>
  14
  15 #include <ModelAPI_Object.h>
  16 #include <ModelAPI_Events.h>
  17 #include <ModelAPI_Validator.h>
  18 #include <ModelAPI_Data.h>
  19 #include <ModelAPI_Session.h>
  20
  21 #include <GeomDataAPI_Point2D.h>
  22 #include <GeomDataAPI_Point.h>
  23 #include <GeomDataAPI_Dir.h>
  24
  25 #include <XGUI_MainWindow.h>
  26 #include <XGUI_Displayer.h>
  27 #include <XGUI_Viewer.h>
  28 #include <XGUI_Workshop.h>
  29 #include <XGUI_OperationMgr.h>
  30 #include <XGUI_ViewPort.h>
  31 #include <XGUI_ActionsMgr.h>
  32 #include <XGUI_ViewerProxy.h>
  33 #include <XGUI_ContextMenuMgr.h>
  34 #include <XGUI_PropertyPanel.h>
  35 #include <XGUI_ModuleConnector.h>
  36 #include <XGUI_Tools.h>
  37
  38 #include <SketchPlugin_Line.h>
  39 #include <SketchPlugin_Sketch.h>
  40 #include <SketchPlugin_Point.h>
  41 #include <SketchPlugin_Arc.h>
  42 #include <SketchPlugin_Circle.h>
  43 #include <SketchPlugin_ConstraintLength.h>
  44 #include <SketchPlugin_ConstraintDistance.h>
  45 #include <SketchPlugin_ConstraintParallel.h>
  46 #include <SketchPlugin_ConstraintPerpendicular.h>
  47 #include <SketchPlugin_ConstraintRadius.h>
  48 #include <SketchPlugin_ConstraintRigid.h>
  49
  50 #include <Events_Loop.h>
  51 #include <StdSelect_TypeOfFace.hxx>
  52
  53 #include <QObject>
  54 #include <QMouseEvent>
  55 #include <QString>
  56 #include <QTimer>
  57 #include <QApplication>
  58
  59 #include <GeomAlgoAPI_FaceBuilder.h>
  60 #include <GeomDataAPI_Dir.h>
  61
  62 #ifdef _DEBUG
  63 #include <QDebug>
  64 #endif
  65
  66
  67 /*!Create and return new instance of XGUI_Module*/
  68 extern "C" PARTSET_EXPORT ModuleBase_IModule* createModule(ModuleBase_IWorkshop* theWshop)
  69 {
  70   return new PartSet_Module(theWshop);
  71 }
  72
  73 PartSet_Module::PartSet_Module(ModuleBase_IWorkshop* theWshop)
  74   : ModuleBase_IModule(theWshop),
  75   myIsDragging(false), myRestartingMode(true), myDragDone(false)
  76 {
  77   //myWorkshop = dynamic_cast<XGUI_Workshop*>(theWshop);
  78   ModuleBase_IViewer* aViewer = aViewer = theWshop->viewer();
  79   connect(aViewer, SIGNAL(mousePress(ModuleBase_IViewWindow*, QMouseEvent*)),
  80           this, SLOT(onMousePressed(ModuleBase_IViewWindow*, QMouseEvent*)));
  81
  82   connect(aViewer, SIGNAL(mouseRelease(ModuleBase_IViewWindow*, QMouseEvent*)),
  83           this, SLOT(onMouseReleased(ModuleBase_IViewWindow*, QMouseEvent*)));
  84
  85   connect(aViewer, SIGNAL(mouseMove(ModuleBase_IViewWindow*, QMouseEvent*)),
  86           this, SLOT(onMouseMoved(ModuleBase_IViewWindow*, QMouseEvent*)));
  87
  88 }
  89
  90 PartSet_Module::~PartSet_Module()
  91 {
  92 }
  93
  94 void PartSet_Module::registerValidators()
  95 {
  96   //Registering of validators
  97   SessionPtr aMgr = ModelAPI_Session::get();
  98   ModelAPI_ValidatorsFactory* aFactory = aMgr->validators();
  99   aFactory->registerValidator("PartSet_DistanceValidator", new PartSet_DistanceValidator);
 100   aFactory->registerValidator("PartSet_LengthValidator", new PartSet_LengthValidator);
 101   aFactory->registerValidator("PartSet_PerpendicularValidator", new PartSet_PerpendicularValidator);
 102   aFactory->registerValidator("PartSet_ParallelValidator", new PartSet_ParallelValidator);
 103   aFactory->registerValidator("PartSet_RadiusValidator", new PartSet_RadiusValidator);
 104 }
 105
 106
 107 void PartSet_Module::onOperationComitted(ModuleBase_Operation* theOperation)
 108 {
 109   if (theOperation->isEditOperation())
 110     return;
 111   /// Restart sketcher operations automatically
 112   FeaturePtr aFeature = theOperation->feature();
 113   std::shared_ptr<SketchPlugin_Feature> aSPFeature =
 114             std::dynamic_pointer_cast<SketchPlugin_Feature>(aFeature);
 115   if (aSPFeature && myRestartingMode) {
 116     myLastOperationId = theOperation->id();
 117     myLastFeature = theOperation->feature();
 118     launchOperation(myLastOperationId);
 119   } else {
 120     breakOperationSequence();
 121   }
 122 }
 123
 124 void PartSet_Module::breakOperationSequence()
 125 {
 126   myLastOperationId = "";
 127   myLastFeature = FeaturePtr();
 128   myRestartingMode = false;
 129
 130 }
 131
 132 void PartSet_Module::onOperationAborted(ModuleBase_Operation* theOperation)
 133 {
 134   breakOperationSequence();
 135 }
 136
 137 void PartSet_Module::onOperationStarted(ModuleBase_Operation* theOperation)
 138 {
 139   myRestartingMode = true;
 140   if (theOperation->id().toStdString() == SketchPlugin_Sketch::ID()) {
 141     // Display all sketcher sub-Objects
 142     myCurrentSketch = std::dynamic_pointer_cast<ModelAPI_CompositeFeature>(theOperation->feature());
 143     XGUI_ModuleConnector* aConnector = dynamic_cast<XGUI_ModuleConnector*>(workshop());
 144     XGUI_Displayer* aDisplayer = aConnector->workshop()->displayer();
 145
 146     for (int i = 0; i < myCurrentSketch->numberOfSubs(); i++) {
 147       FeaturePtr aFeature = myCurrentSketch->subFeature(i);
 148       std::list<ResultPtr> aResults = aFeature->results();
 149       std::list<ResultPtr>::const_iterator aIt;
 150       for (aIt = aResults.begin(); aIt != aResults.end(); ++aIt) {
 151         aDisplayer->display((*aIt), false);
 152       }
 153       aDisplayer->display(aFeature);
 154     }
 155     // Hide sketcher result
 156     std::list<ResultPtr> aResults = myCurrentSketch->results();
 157     std::list<ResultPtr>::const_iterator aIt;
 158     for (aIt = aResults.begin(); aIt != aResults.end(); ++aIt) {
 159       aDisplayer->erase((*aIt), false);
 160     }
 161     aDisplayer->erase(myCurrentSketch);
 162
 163
 164     if (myPlaneFilter.IsNull())
 165       myPlaneFilter = new ModuleBase_ShapeInPlaneFilter();
 166     myWorkshop->viewer()->addSelectionFilter(myPlaneFilter);
 167     if (theOperation->isEditOperation()) {
 168       // If it is editing of sketch then it means that plane is already defined
 169       std::shared_ptr<GeomAPI_Pln> aPln = PartSet_Tools::sketchPlane(myCurrentSketch);
 170       myPlaneFilter->setPlane(aPln->impl<gp_Pln>());
 171     }
 172   }
 173 }
 174
 175 void PartSet_Module::onOperationStopped(ModuleBase_Operation* theOperation)
 176 {
 177   if (theOperation->id().toStdString() == SketchPlugin_Sketch::ID()) {
 178     DataPtr aData = myCurrentSketch->data();
 179     if ((!aData) || (!aData->isValid())) {
 180       // The sketch was aborted
 181       myCurrentSketch = CompositeFeaturePtr();
 182       return;
 183     }
 184     // Hide all sketcher sub-Objects
 185     XGUI_ModuleConnector* aConnector = dynamic_cast<XGUI_ModuleConnector*>(workshop());
 186     XGUI_Displayer* aDisplayer = aConnector->workshop()->displayer();
 187     for (int i = 0; i < myCurrentSketch->numberOfSubs(); i++) {
 188       FeaturePtr aFeature = myCurrentSketch->subFeature(i);
 189       std::list<ResultPtr> aResults = aFeature->results();
 190       std::list<ResultPtr>::const_iterator aIt;
 191       for (aIt = aResults.begin(); aIt != aResults.end(); ++aIt) {
 192         aDisplayer->erase((*aIt), false);
 193       }
 194       aDisplayer->erase(aFeature, false);
 195     }
 196     // Display sketcher result
 197     std::list<ResultPtr> aResults = myCurrentSketch->results();
 198     std::list<ResultPtr>::const_iterator aIt;
 199     for (aIt = aResults.begin(); aIt != aResults.end(); ++aIt) {
 200       aDisplayer->display((*aIt), false);
 201     }
 202     aDisplayer->display(myCurrentSketch);
 203
 204     myCurrentSketch = CompositeFeaturePtr();
 205     myWorkshop->viewer()->removeSelectionFilter(myPlaneFilter);
 206   }
 207 }
 208
 209 void PartSet_Module::onPlaneSelected(const std::shared_ptr<GeomAPI_Pln>& thePln)
 210 {
 211   myPlaneFilter->setPlane(thePln->impl<gp_Pln>());
 212 }
 213
 214
 215 void PartSet_Module::propertyPanelDefined(ModuleBase_Operation* theOperation)
 216 {
 217   if ((theOperation->id() == myLastOperationId) && myLastFeature) {
 218     ModuleBase_ModelWidget* aWgt = theOperation->propertyPanel()->activeWidget();
 219     if (theOperation->id().toStdString() == SketchPlugin_Line::ID()) {
 220       PartSet_WidgetPoint2D* aPnt2dWgt = dynamic_cast<PartSet_WidgetPoint2D*>(aWgt);
 221       if (aPnt2dWgt) {
 222         std::shared_ptr<ModelAPI_Data> aData = myLastFeature->data();
 223         std::shared_ptr<GeomDataAPI_Point2D> aPoint =
 224           std::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(SketchPlugin_Line::END_ID()));
 225         if (aPoint) {
 226           aPnt2dWgt->setPoint(aPoint->x(), aPoint->y());
 227           PartSet_Tools::setConstraints(myCurrentSketch, theOperation->feature(),
 228             aWgt->attributeID(), aPoint->x(), aPoint->y());
 229           theOperation->propertyPanel()->activateNextWidget(aPnt2dWgt);
 230         }
 231       }
 232     }
 233   }
 234 }
 235
 236
 237 void PartSet_Module::onSelectionChanged()
 238 {
 239 }
 240
 241 void PartSet_Module::onMousePressed(ModuleBase_IViewWindow* theWnd, QMouseEvent* theEvent)
 242 {
 243   if (!(theEvent->buttons() & Qt::LeftButton))
 244     return;
 245   ModuleBase_Operation* aOperation = myWorkshop->currentOperation();
 246   // Use only for sketch operations
 247   if (aOperation && myCurrentSketch) {
 248     if (!PartSet_Tools::sketchPlane(myCurrentSketch))
 249       return;
 250
 251     bool isSketcher = (aOperation->id().toStdString() == SketchPlugin_Sketch::ID());
 252     bool isSketchOpe = sketchOperationIdList().contains(aOperation->id());
 253
 254     // Avoid non-sketch operations
 255     if ((!isSketchOpe) && (!isSketcher))
 256       return;
 257
 258     bool isEditing = aOperation->isEditOperation();
 259
 260     // Ignore creation sketch operation
 261     if ((!isSketcher) && (!isEditing))
 262       return;
 263
 264     // Remember highlighted objects for editing
 265     ModuleBase_ISelection* aSelect = myWorkshop->selection();
 266     QList<ModuleBase_ViewerPrs> aObjects = aSelect->getHighlighted();
 267      myEditingFeatures.clear();
 268     if (aObjects.size() > 0) {
 269       foreach(ModuleBase_ViewerPrs aPrs, aObjects) {
 270         FeaturePtr aFeature = ModelAPI_Feature::feature(aObjects.first().object());
 271         if (aFeature)
 272           myEditingFeatures.append(aFeature);
 273       }
 274     }
 275     // If nothing highlighted - return
 276     if (myEditingFeatures.size() == 0)
 277       return;
 278
 279     if (isSketcher) {
 280       CompositeFeaturePtr aSketch =
 281         std::dynamic_pointer_cast<ModelAPI_CompositeFeature>(aOperation->feature());
 282       if (!PartSet_Tools::sketchPlane(aSketch))
 283         return;
 284
 285       //myCurrentSketch = aOperation->feature();
 286       myIsDragging = true;
 287       get2dPoint(theWnd, theEvent, myCurX, myCurY);
 288       myDragDone = false;
 289       myWorkshop->viewer()->enableSelection(false);
 290
 291       launchEditing();
 292
 293     } else if (isSketchOpe && isEditing) {
 294       aOperation->abort();
 295
 296       myIsDragging = true;
 297       get2dPoint(theWnd, theEvent, myCurX, myCurY);
 298       myDragDone = false;
 299       myWorkshop->viewer()->enableSelection(false);
 300
 301       // This is necessary in order to finalize previous operation
 302       QApplication::processEvents();
 303       launchEditing();
 304     }
 305   }
 306 }
 307
 308
 309 void PartSet_Module::get2dPoint(ModuleBase_IViewWindow* theWnd, QMouseEvent* theEvent,
 310                                 double& theX, double& theY)
 311 {
 312   Handle(V3d_View) aView = theWnd->v3dView();
 313   gp_Pnt aPoint = PartSet_Tools::convertClickToPoint(theEvent->pos(), aView);
 314   PartSet_Tools::convertTo2D(aPoint, myCurrentSketch, aView, theX, theY);
 315 }
 316
 317
 318 void PartSet_Module::launchEditing()
 319 {
 320   if (myEditingFeatures.size() == 1) {
 321     FeaturePtr aFeature = myEditingFeatures.first();
 322     std::shared_ptr<SketchPlugin_Feature> aSPFeature =
 323               std::dynamic_pointer_cast<SketchPlugin_Feature>(aFeature);
 324     if (aSPFeature) {
 325       editFeature(aSPFeature);
 326     }
 327   }
 328 }
 329
 330 /// Returns new instance of operation object (used in createOperation for customization)
 331 ModuleBase_Operation* PartSet_Module::getNewOperation(const std::string& theFeatureId)
 332 {
 333   if (theFeatureId == PartSet_OperationSketch::Type()) {
 334     return new PartSet_OperationSketch(theFeatureId.c_str(), this);
 335   }
 336   return ModuleBase_IModule::getNewOperation(theFeatureId);
 337 }
 338
 339
 340 void PartSet_Module::onMouseReleased(ModuleBase_IViewWindow* theWnd, QMouseEvent* theEvent)
 341 {
 342   if (myIsDragging) {
 343     myIsDragging = false;
 344     myWorkshop->viewer()->enableSelection(true);
 345     if (myDragDone)
 346       myWorkshop->currentOperation()->commit();
 347   }
 348 }
 349
 350
 351 void PartSet_Module::onMouseMoved(ModuleBase_IViewWindow* theWnd, QMouseEvent* theEvent)
 352 {
 353   if (myIsDragging) {
 354     ModuleBase_Operation* aOperation = myWorkshop->currentOperation();
 355     if (aOperation->id().toStdString() == SketchPlugin_Sketch::ID())
 356       return; // No edit operation activated
 357
 358     static Events_ID anEvent = Events_Loop::eventByName(EVENT_OBJECT_MOVED);
 359     Handle(V3d_View) aView = theWnd->v3dView();
 360     gp_Pnt aPoint = PartSet_Tools::convertClickToPoint(theEvent->pos(), aView);
 361     double aX, aY;
 362     PartSet_Tools::convertTo2D(aPoint, myCurrentSketch, aView, aX, aY);
 363     double dX =  aX - myCurX;
 364     double dY =  aY - myCurY;
 365
 366     std::shared_ptr<SketchPlugin_Feature> aSketchFeature =
 367       std::dynamic_pointer_cast<SketchPlugin_Feature>(aOperation->feature());
 368     if (aSketchFeature) {
 369       aSketchFeature->move(dX, dY);
 370       ModelAPI_EventCreator::get()->sendUpdated(aSketchFeature, anEvent);
 371       Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_UPDATED));
 372     }/* else { // Alternative case for moving
 373       ModuleBase_IPropertyPanel* aPanel = aOperation->propertyPanel();
 374       QList<ModuleBase_ModelWidget*> aWidgets = aPanel->modelWidgets();
 375       foreach(ModuleBase_ModelWidget* aWgt, aWidgets) {
 376         PartSet_WidgetPoint2D* aWgt2d = dynamic_cast<PartSet_WidgetPoint2D*>(aWgt);
 377         if (aWgt2d) {
 378           aWgt2d->setPoint(aWgt2d->x() + dX, aWgt2d->y() + dY);
 379         }
 380       }
 381     }*/
 382     myDragDone = true;
 383     myCurX = aX;
 384     myCurY = aY;
 385   }
 386 }
 387
 388
 389 QStringList PartSet_Module::sketchOperationIdList() const
 390 {
 391   QStringList aIds;
 392   aIds << SketchPlugin_Line::ID().c_str();
 393   aIds << SketchPlugin_Point::ID().c_str();
 394   aIds << SketchPlugin_Arc::ID().c_str();
 395   aIds << SketchPlugin_Circle::ID().c_str();
 396   aIds << SketchPlugin_ConstraintLength::ID().c_str();
 397   aIds << SketchPlugin_ConstraintDistance::ID().c_str();
 398   aIds << SketchPlugin_ConstraintRigid::ID().c_str();
 399   aIds << SketchPlugin_ConstraintRadius::ID().c_str();
 400   aIds << SketchPlugin_ConstraintPerpendicular::ID().c_str();
 401   aIds << SketchPlugin_ConstraintParallel::ID().c_str();
 402   return aIds;
 403 }
 404
 405 void PartSet_Module::onVertexSelected(ObjectPtr theObject, const TopoDS_Shape& theShape)
 406 {
 407   ModuleBase_Operation* aOperation = myWorkshop->currentOperation();
 408   if (aOperation->id().toStdString() == SketchPlugin_Line::ID()) {
 409     /// If last line finished on vertex the lines creation sequence has to be break
 410     ModuleBase_IPropertyPanel* aPanel = aOperation->propertyPanel();
 411     const QList<ModuleBase_ModelWidget*>& aWidgets = aPanel->modelWidgets();
 412     if (aWidgets.last() == aPanel->activeWidget()) {
 413       breakOperationSequence();
 414     }
 415   }
 416 }
 417
 418 QWidget* PartSet_Module::createWidgetByType(const std::string& theType, QWidget* theParent,
 419                                             Config_WidgetAPI* theWidgetApi, std::string theParentId,
 420                                             QList<ModuleBase_ModelWidget*>& theModelWidgets)
 421 {
 422   XGUI_ModuleConnector* aConnector = dynamic_cast<XGUI_ModuleConnector*>(workshop());
 423   XGUI_Workshop* aWorkshop = aConnector->workshop();
 424   if (theType == "sketch-start-label") {
 425     PartSet_WidgetSketchLabel* aWgt = new PartSet_WidgetSketchLabel(theParent, theWidgetApi, theParentId);
 426     aWgt->setWorkshop(aWorkshop);
 427     connect(aWgt, SIGNAL(planeSelected(const std::shared_ptr<GeomAPI_Pln>&)),
 428       this, SLOT(onPlaneSelected(const std::shared_ptr<GeomAPI_Pln>&)));
 429     theModelWidgets.append(aWgt);
 430     return aWgt->getControl();
 431
 432   } else if (theType == "sketch-2dpoint_selector") {
 433     PartSet_WidgetPoint2D* aWgt = new PartSet_WidgetPoint2D(theParent, theWidgetApi, theParentId);
 434     aWgt->setWorkshop(aWorkshop);
 435     aWgt->setSketch(myCurrentSketch);
 436
 437     connect(aWgt, SIGNAL(vertexSelected(ObjectPtr, const TopoDS_Shape&)),
 438       this, SLOT(onVertexSelected(ObjectPtr, const TopoDS_Shape&)));
 439
 440     theModelWidgets.append(aWgt);
 441     return aWgt->getControl();
 442
 443   } if (theType == "point2ddistance") {
 444     PartSet_WidgetPoint2dDistance* aWgt = new PartSet_WidgetPoint2dDistance(theParent, theWidgetApi, theParentId);
 445     aWgt->setWorkshop(aWorkshop);
 446     aWgt->setSketch(myCurrentSketch);
 447
 448     theModelWidgets.append(aWgt);
 449     return aWgt->getControl();
 450
 451   } if (theType == "sketch_shape_selector") {
 452     PartSet_WidgetShapeSelector* aWgt =
 453       new PartSet_WidgetShapeSelector(theParent, workshop(), theWidgetApi, theParentId);
 454     aWgt->setSketcher(myCurrentSketch);
 455
 456     theModelWidgets.append(aWgt);
 457     return aWgt->getControl();
 458
 459   }else
 460     return 0;
 461 }