1 // Copyright (C) 2014-2019 CEA/DEN, EDF R&D
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.
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.
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
17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
20 #include "PartSet_WidgetFeaturePointSelector.h"
21 #include "PartSet_Tools.h"
22 #include "PartSet_ExternalObjectsMgr.h"
24 #include <Config_WidgetAPI.h>
26 #include <Events_Loop.h>
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>
34 #include <ModuleBase_ISelection.h>
35 #include <ModuleBase_ViewerPrs.h>
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>
43 #include <ModuleBase_IViewWindow.h>
44 #include <ModuleBase_IWorkshop.h>
45 #include <ModuleBase_IModule.h>
47 #include <SketchPlugin_Point.h>
49 #include <XGUI_Tools.h>
50 #include <XGUI_Workshop.h>
51 #include <XGUI_Displayer.h>
52 #include <XGUI_ViewerProxy.h>
55 #include <QMouseEvent>
57 #define HIGHLIGHT_STAYS_PROBLEM
58 #ifdef HIGHLIGHT_STAYS_PROBLEM
59 #include <Quantity_Color.hxx>
60 #define SKETCH_ENTITY_COLOR "225,0,0"
63 PartSet_WidgetFeaturePointSelector::PartSet_WidgetFeaturePointSelector(QWidget* theParent,
64 ModuleBase_IWorkshop* theWorkshop,
65 const Config_WidgetAPI* theData)
66 : ModuleBase_WidgetShapeSelector(theParent, theWorkshop, theData)
68 std::string anAttributes = theData->getProperty("selection_attributes");
69 QStringList anAttributesList =
70 QString(anAttributes.c_str()).split(' ', QString::SkipEmptyParts);
72 myHasPreview = anAttributesList.size() >= 4;
74 mySelectedObjectAttribute = anAttributesList[0].toStdString();
75 mySelectedPointAttribute = anAttributesList[1].toStdString();
77 myPreviewObjectAttribute = anAttributesList[2].toStdString();
78 myPreviewPointAttribute = anAttributesList[3].toStdString();
80 myExternalObjectMgr = new PartSet_ExternalObjectsMgr(theData->getProperty("use_external"),
81 theData->getProperty("can_create_external"), true);
84 PartSet_WidgetFeaturePointSelector::~PartSet_WidgetFeaturePointSelector()
86 delete myExternalObjectMgr;
89 //********************************************************************
90 bool PartSet_WidgetFeaturePointSelector::isValidSelection(
91 const std::shared_ptr<ModuleBase_ViewerPrs>& theValue)
93 return ModuleBase_WidgetShapeSelector::isValidSelection(theValue);
97 //********************************************************************
98 void PartSet_WidgetFeaturePointSelector::activateCustom()
100 ModuleBase_WidgetShapeSelector::activateCustom();
102 myWorkshop->module()->activateCustomPrs(myFeature,
103 ModuleBase_IModule::CustomizeHighlightedObjects, true);
106 //********************************************************************
107 void PartSet_WidgetFeaturePointSelector::deactivate()
109 ModuleBase_WidgetShapeSelector::deactivate();
112 //********************************************************************
113 void PartSet_WidgetFeaturePointSelector::mouseMoved(ModuleBase_IViewWindow* theWindow,
114 QMouseEvent* theEvent)
116 ModuleBase_ISelection* aSelect = myWorkshop->selection();
117 QList<ModuleBase_ViewerPrsPtr> aHighlighted = aSelect->getHighlighted();
119 ModuleBase_ViewerPrsPtr aPrs = !aHighlighted.empty() ? aHighlighted.first()
120 : ModuleBase_ViewerPrsPtr();
121 myPreviewPoint = PartSet_Tools::getPnt2d(theEvent, theWindow, mySketch);
123 if (aPrs.get() && aPrs->object().get())
124 myPreviewObject = aPrs->object();
126 myPreviewObject = ObjectPtr();
128 updateObject(feature());
129 Events_Loop::loop()->flush(Events_Loop::eventByName(EVENT_OBJECT_TO_REDISPLAY));
133 //********************************************************************
134 void PartSet_WidgetFeaturePointSelector::mouseReleased(ModuleBase_IViewWindow* theWindow,
135 QMouseEvent* theEvent)
137 // the contex menu release by the right button should not be processed by this widget
138 if (theEvent->button() != Qt::LeftButton)
141 ModuleBase_ISelection* aSelection = myWorkshop->selection();
142 QList<ModuleBase_ViewerPrsPtr> aSelected =
143 aSelection->getSelected(ModuleBase_ISelection::Viewer);
145 ModuleBase_ViewerPrsPtr aPrs =
146 !aSelected.empty() ? aSelected.first() : ModuleBase_ViewerPrsPtr();
147 if (aPrs.get() && aPrs->object().get()) {
148 myPreviewObject = aSelection->getResult(aPrs);
149 GeomShapePtr aShape = aSelection->getShape(aPrs);
150 myExternalObjectMgr->getGeomSelection(aPrs, myPreviewObject, aShape,
151 myWorkshop, sketch(), true);
153 myPreviewPoint = PartSet_Tools::getPnt2d(theEvent, theWindow, mySketch);
155 ObjectPtr aPreviewObject;
156 GeomPnt2dPtr aPreviewPoint;
158 std::shared_ptr<ModelAPI_AttributeReference> aRefPreviewAttr =
159 std::dynamic_pointer_cast<ModelAPI_AttributeReference>(
160 feature()->data()->attribute(myPreviewObjectAttribute));
161 aPreviewObject = aRefPreviewAttr->value();
163 std::shared_ptr<GeomDataAPI_Point2D> aPointPreviewAttr =
164 std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
165 feature()->data()->attribute(myPreviewPointAttribute));
166 aPreviewPoint = aPointPreviewAttr->pnt();
169 aPreviewObject = myPreviewObject;
170 aPreviewPoint = myPreviewPoint;
173 // do not move focus from the current widget if the object is not highlighted/selected
174 if (!aPreviewObject.get())
177 // Do not use non-sketcher objects
178 if (!sketch()->isSub(aPreviewObject))
181 // set parameters of preview into parameters of selection in the feature
182 std::shared_ptr<GeomDataAPI_Point2D> aPointSelectedAttr =
183 std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
184 feature()->data()->attribute(mySelectedPointAttribute));
185 aPointSelectedAttr->setValue(aPreviewPoint);
187 AttributeReferencePtr aRefSelectedAttr = feature()->reference(mySelectedObjectAttribute);
188 if (aRefSelectedAttr)
189 aRefSelectedAttr->setValue(aPreviewObject);
191 AttributeRefAttrPtr aRefAttrSelectedAttr = feature()->refattr(mySelectedObjectAttribute);
192 if (aRefAttrSelectedAttr)
193 aRefAttrSelectedAttr->setObject(aPreviewObject);
196 updateObject(feature());
198 // we need to deselect base feature for better visibility of selected feature
199 XGUI_Tools::workshop(myWorkshop)->displayer()->clearSelected(false);
201 // focusOutWidget should be the last functionality in the method because after this emit,
202 // the widget may be deleted and members of this class are deleted (e.g. myWorkshop)
203 emit focusOutWidget(this);
206 //********************************************************************
207 bool PartSet_WidgetFeaturePointSelector::fillFeature()
210 std::shared_ptr<ModelAPI_AttributeReference> aRef =
211 std::dynamic_pointer_cast<ModelAPI_AttributeReference>(
212 feature()->data()->attribute(myPreviewObjectAttribute));
213 aRef->setValue(myPreviewObject);
214 if (myPreviewPoint.get()) {
215 std::shared_ptr<GeomDataAPI_Point2D> anAttributePoint =
216 std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
217 feature()->data()->attribute(myPreviewPointAttribute));
218 anAttributePoint->setValue(myPreviewPoint);
222 // Do not use non-sketcher objects
223 if (!sketch()->isSub(myPreviewObject))
226 // set parameters of preview into parameters of selection in the feature
227 if (myPreviewPoint.get()) {
228 std::shared_ptr<GeomDataAPI_Point2D> aPointSelectedAttr =
229 std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
230 feature()->data()->attribute(mySelectedPointAttribute));
231 aPointSelectedAttr->setValue(myPreviewPoint);
233 AttributeReferencePtr aRefSelectedAttr = feature()->reference(mySelectedObjectAttribute);
234 if (aRefSelectedAttr)
235 aRefSelectedAttr->setValue(myPreviewObject);
237 AttributeRefAttrPtr aRefAttrSelectedAttr = feature()->refattr(mySelectedObjectAttribute);
238 if (aRefAttrSelectedAttr)
239 aRefAttrSelectedAttr->setObject(myPreviewObject);
242 // redisplay AIS presentation in viewer
243 #ifndef HIGHLIGHT_STAYS_PROBLEM
244 // an attempt to clear highlighted item in the viewer: but of OCCT
245 XGUI_Tools::workshop(myWorkshop)->displayer()->clearSelected(true);
250 //********************************************************************
251 QList<ModuleBase_ViewerPrsPtr> PartSet_WidgetFeaturePointSelector::getAttributeSelection() const
253 return QList<ModuleBase_ViewerPrsPtr>();
256 //********************************************************************
257 bool PartSet_WidgetFeaturePointSelector::setSelectionCustom(const ModuleBase_ViewerPrsPtr& thePrs)
259 if (!thePrs.get() || !thePrs->object().get())
262 ModuleBase_ISelection* aSelection = myWorkshop->selection();
263 myPreviewObject = aSelection->getResult(thePrs);
264 GeomShapePtr aShape = aSelection->getShape(thePrs);
265 myExternalObjectMgr->getGeomSelection(thePrs, myPreviewObject, aShape, myWorkshop,
267 return fillFeature();
270 //********************************************************************
271 void PartSet_WidgetFeaturePointSelector::setPreSelection(
272 const ModuleBase_ViewerPrsPtr& thePreSelected,
273 ModuleBase_IViewWindow* theWnd,
274 QMouseEvent* theEvent)
276 // the method is empty because firstly by starging of the feature there is no selection of
277 // sub-segments in the viewer, secondly preselection of restart operation is processed by
278 // special reentrant message sent by the feature
281 //********************************************************************
282 void PartSet_WidgetFeaturePointSelector::getGeomSelection(const ModuleBase_ViewerPrsPtr& thePrs,
283 ObjectPtr& theObject, GeomShapePtr& theShape)
285 ModuleBase_WidgetShapeSelector::getGeomSelection(thePrs, theObject, theShape);
287 myExternalObjectMgr->getGeomSelection(thePrs, theObject, theShape,
288 myWorkshop, sketch(), myIsInValidate);
289 myPreviewObject = theObject;
292 //********************************************************************
293 void PartSet_WidgetFeaturePointSelector::restoreAttributeValue(const AttributePtr& theAttribute,
296 ModuleBase_WidgetShapeSelector::restoreAttributeValue(theAttribute, theValid);
297 myExternalObjectMgr->removeExternal(sketch(), myFeature, myWorkshop, true);
298 myPreviewObject = ObjectPtr();