src/PartSet/PartSet_WidgetFeaturePointSelector.cpp

   1 // Copyright (C) 2014-2019  CEA/DEN, EDF R&D
   2 //
   3 // This library is free software; you can redistribute it and/or
   4 // modify it under the terms of the GNU Lesser General Public
   5 // License as published by the Free Software Foundation; either
   6 // version 2.1 of the License, or (at your option) any later version.
   7 //
   8 // This library is distributed in the hope that it will be useful,
   9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 // Lesser General Public License for more details.
  12 //
  13 // You should have received a copy of the GNU Lesser General Public
  14 // License along with this library; if not, write to the Free Software
  15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  16 //
  17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  18 //
  19
  20 #include "PartSet_WidgetFeaturePointSelector.h"
  21 #include "PartSet_Tools.h"
  22 #include "PartSet_ExternalObjectsMgr.h"
  23
  24 #include <Config_WidgetAPI.h>
  25
  26 #include <Events_Loop.h>
  27
  28 #include <GeomDataAPI_Point2D.h>
  29 #include <GeomDataAPI_Point.h>
  30 #include <GeomAPI_Edge.h>
  31 #include <GeomAPI_Pnt2d.h>
  32 #include <GeomAlgoAPI_ShapeTools.h>
  33
  34 #include <ModuleBase_ISelection.h>
  35 #include <ModuleBase_ViewerPrs.h>
  36
  37 #include <ModelAPI_AttributeRefAttr.h>
  38 #include <ModelAPI_AttributeReference.h>
  39 #include <ModelAPI_Events.h>
  40 #include <ModelAPI_Feature.h>
  41 #include <ModelAPI_Tools.h>
  42
  43 #include <ModuleBase_IViewWindow.h>
  44 #include <ModuleBase_IWorkshop.h>
  45 #include <ModuleBase_IModule.h>
  46
  47 #include <SketchPlugin_Point.h>
  48
  49 #include <XGUI_Tools.h>
  50 #include <XGUI_Workshop.h>
  51 #include <XGUI_Displayer.h>
  52 #include <XGUI_ViewerProxy.h>
  53
  54 #include <QWidget>
  55 #include <QMouseEvent>
  56
  57 #define HIGHLIGHT_STAYS_PROBLEM
  58 #ifdef HIGHLIGHT_STAYS_PROBLEM
  59 #include <Quantity_Color.hxx>
  60 #define SKETCH_ENTITY_COLOR "225,0,0"
  61 #endif
  62
  63 PartSet_WidgetFeaturePointSelector::PartSet_WidgetFeaturePointSelector(QWidget* theParent,
  64                                                          ModuleBase_IWorkshop* theWorkshop,
  65                                                          const Config_WidgetAPI* theData)
  66 : ModuleBase_WidgetShapeSelector(theParent, theWorkshop, theData)
  67 {
  68   std::string anAttributes = theData->getProperty("selection_attributes");
  69   QStringList anAttributesList = QString(anAttributes.c_str()).split(' ', QString::SkipEmptyParts);
  70
  71   myHasPreview = anAttributesList.size() >= 4;
  72
  73   mySelectedObjectAttribute = anAttributesList[0].toStdString();
  74   mySelectedPointAttribute = anAttributesList[1].toStdString();
  75   if (myHasPreview) {
  76     myPreviewObjectAttribute = anAttributesList[2].toStdString();
  77     myPreviewPointAttribute = anAttributesList[3].toStdString();
  78   }
  79   myExternalObjectMgr = new PartSet_ExternalObjectsMgr(theData->getProperty("use_external"),
  80     theData->getProperty("can_create_external"), true);
  81 }
  82
  83 PartSet_WidgetFeaturePointSelector::~PartSet_WidgetFeaturePointSelector()
  84 {
  85   delete myExternalObjectMgr;
  86 }
  87
  88 //********************************************************************
  89 bool PartSet_WidgetFeaturePointSelector::isValidSelection(
  90                                         const std::shared_ptr<ModuleBase_ViewerPrs>& theValue)
  91 {
  92   return ModuleBase_WidgetShapeSelector::isValidSelection(theValue);
  93   //return true;
  94 }
  95
  96 //********************************************************************
  97 void PartSet_WidgetFeaturePointSelector::activateCustom()
  98 {
  99   ModuleBase_WidgetShapeSelector::activateCustom();
 100
 101   myWorkshop->module()->activateCustomPrs(myFeature,
 102                             ModuleBase_IModule::CustomizeHighlightedObjects, true);
 103 }
 104
 105 //********************************************************************
 106 void PartSet_WidgetFeaturePointSelector::deactivate()
 107 {
 108   ModuleBase_WidgetShapeSelector::deactivate();
 109 }
 110
 111 //********************************************************************
 112 void PartSet_WidgetFeaturePointSelector::mouseMoved(ModuleBase_IViewWindow* theWindow,
 113                                                     QMouseEvent* theEvent)
 114 {
 115   ModuleBase_ISelection* aSelect = myWorkshop->selection();
 116   QList<ModuleBase_ViewerPrsPtr> aHighlighted = aSelect->getHighlighted();
 117
 118   ModuleBase_ViewerPrsPtr aPrs = !aHighlighted.empty() ? aHighlighted.first()
 119                                                        : ModuleBase_ViewerPrsPtr();
 120   myPreviewPoint = PartSet_Tools::getPnt2d(theEvent, theWindow, mySketch);
 121   if (myHasPreview) {
 122     if (aPrs.get() && aPrs->object().get())
 123       myPreviewObject = aPrs->object();
 124     else
 125       myPreviewObject = ObjectPtr();
 126     fillFeature();
 127     updateObject(feature());
 128     Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_TO_REDISPLAY));
 129   }
 130 }
 131
 132 //********************************************************************
 133 void PartSet_WidgetFeaturePointSelector::mouseReleased(ModuleBase_IViewWindow* theWindow,
 134                                                        QMouseEvent* theEvent)
 135 {
 136   // the contex menu release by the right button should not be processed by this widget
 137   if (theEvent->button() != Qt::LeftButton)
 138     return;
 139
 140   ModuleBase_ISelection* aSelection = myWorkshop->selection();
 141   QList<ModuleBase_ViewerPrsPtr> aSelected = aSelection->getSelected(ModuleBase_ISelection::Viewer);
 142
 143   ModuleBase_ViewerPrsPtr aPrs =
 144     !aSelected.empty() ? aSelected.first() : ModuleBase_ViewerPrsPtr();
 145   if (aPrs.get() && aPrs->object().get()) {
 146     myPreviewObject = aSelection->getResult(aPrs);
 147     GeomShapePtr aShape = aSelection->getShape(aPrs);
 148     myExternalObjectMgr->getGeomSelection(aPrs, myPreviewObject, aShape,
 149       myWorkshop, sketch(), true);
 150   }
 151   myPreviewPoint = PartSet_Tools::getPnt2d(theEvent, theWindow, mySketch);
 152
 153   ObjectPtr aPreviewObject;
 154   GeomPnt2dPtr aPreviewPoint;
 155   if (myHasPreview) {
 156     std::shared_ptr<ModelAPI_AttributeReference> aRefPreviewAttr =
 157                             std::dynamic_pointer_cast<ModelAPI_AttributeReference>(
 158                             feature()->data()->attribute(myPreviewObjectAttribute));
 159     aPreviewObject = aRefPreviewAttr->value();
 160
 161     std::shared_ptr<GeomDataAPI_Point2D> aPointPreviewAttr =
 162                             std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
 163                             feature()->data()->attribute(myPreviewPointAttribute));
 164     aPreviewPoint = aPointPreviewAttr->pnt();
 165   }
 166   else {
 167     aPreviewObject = myPreviewObject;
 168     aPreviewPoint = myPreviewPoint;
 169   }
 170
 171   // do not move focus from the current widget if the object is not highlighted/selected
 172   if (!aPreviewObject.get())
 173     return;
 174
 175   // Do not use non-sketcher objects
 176   if (!sketch()->isSub(aPreviewObject))
 177     return;
 178
 179   // set parameters of preview into parameters of selection in the feature
 180   std::shared_ptr<GeomDataAPI_Point2D> aPointSelectedAttr =
 181                           std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
 182                           feature()->data()->attribute(mySelectedPointAttribute));
 183   aPointSelectedAttr->setValue(aPreviewPoint);
 184
 185   AttributeReferencePtr aRefSelectedAttr = feature()->reference(mySelectedObjectAttribute);
 186   if (aRefSelectedAttr)
 187     aRefSelectedAttr->setValue(aPreviewObject);
 188   else {
 189     AttributeRefAttrPtr aRefAttrSelectedAttr = feature()->refattr(mySelectedObjectAttribute);
 190     if (aRefAttrSelectedAttr)
 191       aRefAttrSelectedAttr->setObject(aPreviewObject);
 192   }
 193
 194   updateObject(feature());
 195
 196   // we need to deselect base feature for better visibility of selected feature
 197   XGUI_Tools::workshop(myWorkshop)->displayer()->clearSelected(false);
 198
 199   // focusOutWidget should be the last functionality in the method because after this emit,
 200   // the widget may be deleted and members of this class are deleted (e.g. myWorkshop)
 201   emit focusOutWidget(this);
 202 }
 203
 204 //********************************************************************
 205 bool PartSet_WidgetFeaturePointSelector::fillFeature()
 206 {
 207   if (myHasPreview) {
 208     std::shared_ptr<ModelAPI_AttributeReference> aRef =
 209                             std::dynamic_pointer_cast<ModelAPI_AttributeReference>(
 210                             feature()->data()->attribute(myPreviewObjectAttribute));
 211     aRef->setValue(myPreviewObject);
 212     if (myPreviewPoint.get()) {
 213       std::shared_ptr<GeomDataAPI_Point2D> anAttributePoint =
 214         std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
 215           feature()->data()->attribute(myPreviewPointAttribute));
 216       anAttributePoint->setValue(myPreviewPoint);
 217     }
 218   }
 219   else {
 220     // Do not use non-sketcher objects
 221     if (!sketch()->isSub(myPreviewObject))
 222       return false;
 223
 224     // set parameters of preview into parameters of selection in the feature
 225     if (myPreviewPoint.get()) {
 226       std::shared_ptr<GeomDataAPI_Point2D> aPointSelectedAttr =
 227         std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
 228           feature()->data()->attribute(mySelectedPointAttribute));
 229       aPointSelectedAttr->setValue(myPreviewPoint);
 230     }
 231     AttributeReferencePtr aRefSelectedAttr = feature()->reference(mySelectedObjectAttribute);
 232     if (aRefSelectedAttr)
 233       aRefSelectedAttr->setValue(myPreviewObject);
 234     else {
 235       AttributeRefAttrPtr aRefAttrSelectedAttr = feature()->refattr(mySelectedObjectAttribute);
 236       if (aRefAttrSelectedAttr)
 237         aRefAttrSelectedAttr->setObject(myPreviewObject);
 238     }
 239   }
 240   // redisplay AIS presentation in viewer
 241 #ifndef HIGHLIGHT_STAYS_PROBLEM
 242   // an attempt to clear highlighted item in the viewer: but of OCCT
 243   XGUI_Tools::workshop(myWorkshop)->displayer()->clearSelected(true);
 244 #endif
 245   return true;
 246 }
 247
 248 //********************************************************************
 249 QList<ModuleBase_ViewerPrsPtr> PartSet_WidgetFeaturePointSelector::getAttributeSelection() const
 250 {
 251   return QList<ModuleBase_ViewerPrsPtr>();
 252 }
 253
 254 //********************************************************************
 255 bool PartSet_WidgetFeaturePointSelector::setSelectionCustom(const ModuleBase_ViewerPrsPtr& thePrs)
 256 {
 257   if (!thePrs.get() || !thePrs->object().get())
 258     return false;
 259
 260   ModuleBase_ISelection* aSelection = myWorkshop->selection();
 261   myPreviewObject = aSelection->getResult(thePrs);
 262   GeomShapePtr aShape = aSelection->getShape(thePrs);
 263   myExternalObjectMgr->getGeomSelection(thePrs, myPreviewObject, aShape, myWorkshop,
 264     sketch(), true);
 265   return fillFeature();
 266 }
 267
 268 //********************************************************************
 269 void PartSet_WidgetFeaturePointSelector::setPreSelection(
 270                                   const ModuleBase_ViewerPrsPtr& thePreSelected,
 271                                   ModuleBase_IViewWindow* theWnd,
 272                                   QMouseEvent* theEvent)
 273 {
 274   // the method is empty because firstly by starging of the feature there is no selection of
 275   // sub-segments in the viewer, secondly preselection of restart operation is processed by
 276   // special reentrant message sent by the feature
 277 }
 278
 279 //********************************************************************
 280 void PartSet_WidgetFeaturePointSelector::getGeomSelection(const ModuleBase_ViewerPrsPtr& thePrs,
 281   ObjectPtr& theObject, GeomShapePtr& theShape)
 282 {
 283   ModuleBase_WidgetShapeSelector::getGeomSelection(thePrs, theObject, theShape);
 284
 285   myExternalObjectMgr->getGeomSelection(thePrs, theObject, theShape,
 286     myWorkshop, sketch(), myIsInValidate);
 287   myPreviewObject = theObject;
 288 }
 289
 290 //********************************************************************
 291 void PartSet_WidgetFeaturePointSelector::restoreAttributeValue(const AttributePtr& theAttribute,
 292   const bool theValid)
 293 {
 294   ModuleBase_WidgetShapeSelector::restoreAttributeValue(theAttribute, theValid);
 295   myExternalObjectMgr->removeExternal(sketch(), myFeature, myWorkshop, true);
 296   myPreviewObject = ObjectPtr();
 297 }