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
 165 void PartSet_Module::onOperationStopped(ModuleBase_Operation* theOperation)
 166 {
 167   if (theOperation->id().toStdString() == SketchPlugin_Sketch::ID()) {
 168     DataPtr aData = myCurrentSketch->data();
 169     if ((!aData) || (!aData->isValid())) {
 170       // The sketch was aborted
 171       myCurrentSketch = CompositeFeaturePtr();
 172       return;
 173     }
 174     // Hide all sketcher sub-Objects
 175     XGUI_ModuleConnector* aConnector = dynamic_cast<XGUI_ModuleConnector*>(workshop());
 176     XGUI_Displayer* aDisplayer = aConnector->workshop()->displayer();
 177     for (int i = 0; i < myCurrentSketch->numberOfSubs(); i++) {
 178       FeaturePtr aFeature = myCurrentSketch->subFeature(i);
 179       std::list<ResultPtr> aResults = aFeature->results();
 180       std::list<ResultPtr>::const_iterator aIt;
 181       for (aIt = aResults.begin(); aIt != aResults.end(); ++aIt) {
 182         aDisplayer->erase((*aIt), false);
 183       }
 184       aDisplayer->erase(aFeature, false);
 185     }
 186     // Display sketcher result
 187     std::list<ResultPtr> aResults = myCurrentSketch->results();
 188     std::list<ResultPtr>::const_iterator aIt;
 189     for (aIt = aResults.begin(); aIt != aResults.end(); ++aIt) {
 190       aDisplayer->display((*aIt), false);
 191     }
 192     aDisplayer->display(myCurrentSketch);
 193
 194     myCurrentSketch = CompositeFeaturePtr();
 195   }
 196 }
 197
 198
 199
 200 void PartSet_Module::propertyPanelDefined(ModuleBase_Operation* theOperation)
 201 {
 202   if ((theOperation->id() == myLastOperationId) && myLastFeature) {
 203     ModuleBase_ModelWidget* aWgt = theOperation->propertyPanel()->activeWidget();
 204     if (theOperation->id().toStdString() == SketchPlugin_Line::ID()) {
 205       PartSet_WidgetPoint2D* aPnt2dWgt = dynamic_cast<PartSet_WidgetPoint2D*>(aWgt);
 206       if (aPnt2dWgt) {
 207         std::shared_ptr<ModelAPI_Data> aData = myLastFeature->data();
 208         std::shared_ptr<GeomDataAPI_Point2D> aPoint =
 209           std::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(SketchPlugin_Line::END_ID()));
 210         if (aPoint) {
 211           aPnt2dWgt->setPoint(aPoint->x(), aPoint->y());
 212           PartSet_Tools::setConstraints(myCurrentSketch, theOperation->feature(),
 213             aWgt->attributeID(), aPoint->x(), aPoint->y());
 214           theOperation->propertyPanel()->activateNextWidget(aPnt2dWgt);
 215         }
 216       }
 217     }
 218   }
 219 }
 220
 221
 222 void PartSet_Module::onSelectionChanged()
 223 {
 224 }
 225
 226 void PartSet_Module::onMousePressed(ModuleBase_IViewWindow* theWnd, QMouseEvent* theEvent)
 227 {
 228   if (!(theEvent->buttons() & Qt::LeftButton))
 229     return;
 230   ModuleBase_Operation* aOperation = myWorkshop->currentOperation();
 231   // Use only for sketch operations
 232   if (aOperation && myCurrentSketch) {
 233     if (!PartSet_Tools::sketchPlane(myCurrentSketch))
 234       return;
 235
 236     bool isSketcher = (aOperation->id().toStdString() == SketchPlugin_Sketch::ID());
 237     bool isSketchOpe = sketchOperationIdList().contains(aOperation->id());
 238
 239     // Avoid non-sketch operations
 240     if ((!isSketchOpe) && (!isSketcher))
 241       return;
 242
 243     bool isEditing = aOperation->isEditOperation();
 244
 245     // Ignore creation sketch operation
 246     if ((!isSketcher) && (!isEditing))
 247       return;
 248
 249     // Remember highlighted objects for editing
 250     ModuleBase_ISelection* aSelect = myWorkshop->selection();
 251     QList<ModuleBase_ViewerPrs> aObjects = aSelect->getHighlighted();
 252      myEditingFeatures.clear();
 253     if (aObjects.size() > 0) {
 254       foreach(ModuleBase_ViewerPrs aPrs, aObjects) {
 255         FeaturePtr aFeature = ModelAPI_Feature::feature(aObjects.first().object());
 256         if (aFeature)
 257           myEditingFeatures.append(aFeature);
 258       }
 259     }
 260     // If nothing highlighted - return
 261     if (myEditingFeatures.size() == 0)
 262       return;
 263
 264     if (isSketcher) {
 265       CompositeFeaturePtr aSketch =
 266         std::dynamic_pointer_cast<ModelAPI_CompositeFeature>(aOperation->feature());
 267       if (!PartSet_Tools::sketchPlane(aSketch))
 268         return;
 269
 270       //myCurrentSketch = aOperation->feature();
 271       myIsDragging = true;
 272       get2dPoint(theWnd, theEvent, myCurX, myCurY);
 273       myDragDone = false;
 274       myWorkshop->viewer()->enableSelection(false);
 275
 276       launchEditing();
 277
 278     } else if (isSketchOpe && isEditing) {
 279       aOperation->abort();
 280
 281       myIsDragging = true;
 282       get2dPoint(theWnd, theEvent, myCurX, myCurY);
 283       myDragDone = false;
 284       myWorkshop->viewer()->enableSelection(false);
 285
 286       // This is necessary in order to finalize previous operation
 287       QApplication::processEvents();
 288       launchEditing();
 289     }
 290   }
 291 }
 292
 293
 294 void PartSet_Module::get2dPoint(ModuleBase_IViewWindow* theWnd, QMouseEvent* theEvent,
 295                                 double& theX, double& theY)
 296 {
 297   Handle(V3d_View) aView = theWnd->v3dView();
 298   gp_Pnt aPoint = PartSet_Tools::convertClickToPoint(theEvent->pos(), aView);
 299   PartSet_Tools::convertTo2D(aPoint, myCurrentSketch, aView, theX, theY);
 300 }
 301
 302
 303 void PartSet_Module::launchEditing()
 304 {
 305   if (myEditingFeatures.size() == 1) {
 306     FeaturePtr aFeature = myEditingFeatures.first();
 307     std::shared_ptr<SketchPlugin_Feature> aSPFeature =
 308               std::dynamic_pointer_cast<SketchPlugin_Feature>(aFeature);
 309     if (aSPFeature) {
 310       editFeature(aSPFeature);
 311     }
 312   }
 313 }
 314
 315 /// Returns new instance of operation object (used in createOperation for customization)
 316 ModuleBase_Operation* PartSet_Module::getNewOperation(const std::string& theFeatureId)
 317 {
 318   if (theFeatureId == PartSet_OperationSketch::Type()) {
 319     return new PartSet_OperationSketch(theFeatureId.c_str(), this);
 320   }
 321   return ModuleBase_IModule::getNewOperation(theFeatureId);
 322 }
 323
 324
 325 void PartSet_Module::onMouseReleased(ModuleBase_IViewWindow* theWnd, QMouseEvent* theEvent)
 326 {
 327   if (myIsDragging) {
 328     myIsDragging = false;
 329     myWorkshop->viewer()->enableSelection(true);
 330     if (myDragDone)
 331       myWorkshop->currentOperation()->commit();
 332   }
 333 }
 334
 335
 336 void PartSet_Module::onMouseMoved(ModuleBase_IViewWindow* theWnd, QMouseEvent* theEvent)
 337 {
 338   if (myIsDragging) {
 339     ModuleBase_Operation* aOperation = myWorkshop->currentOperation();
 340     if (aOperation->id().toStdString() == SketchPlugin_Sketch::ID())
 341       return; // No edit operation activated
 342
 343     static Events_ID anEvent = Events_Loop::eventByName(EVENT_OBJECT_MOVED);
 344     Handle(V3d_View) aView = theWnd->v3dView();
 345     gp_Pnt aPoint = PartSet_Tools::convertClickToPoint(theEvent->pos(), aView);
 346     double aX, aY;
 347     PartSet_Tools::convertTo2D(aPoint, myCurrentSketch, aView, aX, aY);
 348     double dX =  aX - myCurX;
 349     double dY =  aY - myCurY;
 350
 351     std::shared_ptr<SketchPlugin_Feature> aSketchFeature =
 352       std::dynamic_pointer_cast<SketchPlugin_Feature>(aOperation->feature());
 353     if (aSketchFeature) {
 354       aSketchFeature->move(dX, dY);
 355       ModelAPI_EventCreator::get()->sendUpdated(aSketchFeature, anEvent);
 356       Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_UPDATED));
 357     }/* else { // Alternative case for moving
 358       ModuleBase_IPropertyPanel* aPanel = aOperation->propertyPanel();
 359       QList<ModuleBase_ModelWidget*> aWidgets = aPanel->modelWidgets();
 360       foreach(ModuleBase_ModelWidget* aWgt, aWidgets) {
 361         PartSet_WidgetPoint2D* aWgt2d = dynamic_cast<PartSet_WidgetPoint2D*>(aWgt);
 362         if (aWgt2d) {
 363           aWgt2d->setPoint(aWgt2d->x() + dX, aWgt2d->y() + dY);
 364         }
 365       }
 366     }*/
 367     myDragDone = true;
 368     myCurX = aX;
 369     myCurY = aY;
 370   }
 371 }
 372
 373
 374 QStringList PartSet_Module::sketchOperationIdList() const
 375 {
 376   QStringList aIds;
 377   aIds << SketchPlugin_Line::ID().c_str();
 378   aIds << SketchPlugin_Point::ID().c_str();
 379   aIds << SketchPlugin_Arc::ID().c_str();
 380   aIds << SketchPlugin_Circle::ID().c_str();
 381   aIds << SketchPlugin_ConstraintLength::ID().c_str();
 382   aIds << SketchPlugin_ConstraintDistance::ID().c_str();
 383   aIds << SketchPlugin_ConstraintRigid::ID().c_str();
 384   aIds << SketchPlugin_ConstraintRadius::ID().c_str();
 385   aIds << SketchPlugin_ConstraintPerpendicular::ID().c_str();
 386   aIds << SketchPlugin_ConstraintParallel::ID().c_str();
 387   return aIds;
 388 }
 389
 390 void PartSet_Module::onVertexSelected(ObjectPtr theObject, const TopoDS_Shape& theShape)
 391 {
 392   ModuleBase_Operation* aOperation = myWorkshop->currentOperation();
 393   if (aOperation->id().toStdString() == SketchPlugin_Line::ID()) {
 394     /// If last line finished on vertex the lines creation sequence has to be break
 395     ModuleBase_IPropertyPanel* aPanel = aOperation->propertyPanel();
 396     const QList<ModuleBase_ModelWidget*>& aWidgets = aPanel->modelWidgets();
 397     if (aWidgets.last() == aPanel->activeWidget()) {
 398       breakOperationSequence();
 399     }
 400   }
 401 }
 402
 403 QWidget* PartSet_Module::createWidgetByType(const std::string& theType, QWidget* theParent,
 404                                             Config_WidgetAPI* theWidgetApi, std::string theParentId,
 405                                             QList<ModuleBase_ModelWidget*>& theModelWidgets)
 406 {
 407   XGUI_ModuleConnector* aConnector = dynamic_cast<XGUI_ModuleConnector*>(workshop());
 408   XGUI_Workshop* aWorkshop = aConnector->workshop();
 409   if (theType == "sketch-start-label") {
 410     PartSet_WidgetSketchLabel* aWgt = new PartSet_WidgetSketchLabel(theParent, theWidgetApi, theParentId);
 411     aWgt->setWorkshop(aWorkshop);
 412     theModelWidgets.append(aWgt);
 413     return aWgt->getControl();
 414
 415   } else if (theType == "sketch-2dpoint_selector") {
 416     PartSet_WidgetPoint2D* aWgt = new PartSet_WidgetPoint2D(theParent, theWidgetApi, theParentId);
 417     aWgt->setWorkshop(aWorkshop);
 418     aWgt->setSketch(myCurrentSketch);
 419
 420     connect(aWgt, SIGNAL(vertexSelected(ObjectPtr, const TopoDS_Shape&)),
 421       this, SLOT(onVertexSelected(ObjectPtr, const TopoDS_Shape&)));
 422
 423     theModelWidgets.append(aWgt);
 424     return aWgt->getControl();
 425
 426   } if (theType == "point2ddistance") {
 427     PartSet_WidgetPoint2dDistance* aWgt = new PartSet_WidgetPoint2dDistance(theParent, theWidgetApi, theParentId);
 428     aWgt->setWorkshop(aWorkshop);
 429     aWgt->setSketch(myCurrentSketch);
 430
 431     theModelWidgets.append(aWgt);
 432     return aWgt->getControl();
 433
 434   } if (theType == "sketch_shape_selector") {
 435     PartSet_WidgetShapeSelector* aWgt =
 436       new PartSet_WidgetShapeSelector(theParent, workshop(), theWidgetApi, theParentId);
 437     aWgt->setSketcher(myCurrentSketch);
 438
 439     theModelWidgets.append(aWgt);
 440     return aWgt->getControl();
 441
 442   }else
 443     return 0;
 444 }