1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
3 // File: PartSet_WidgetSubShapeSelector.cpp
4 // Created: 21 Jul 2016
5 // Author: Natalia ERMOLAEVA
7 #include "PartSet_WidgetSubShapeSelector.h"
8 #include "PartSet_Tools.h"
10 #include <ModuleBase_ISelection.h>
11 #include <ModuleBase_ViewerPrs.h>
13 #include <ModelAPI_Feature.h>
14 #include <ModelAPI_Tools.h>
15 #include <GeomDataAPI_Point2D.h>
17 #include <GeomDataAPI_Point.h>
18 #include <GeomAPI_Edge.h>
19 #include <GeomAPI_Pnt2d.h>
21 #include <GeomAlgoAPI_ShapeTools.h>
22 #include <ModelGeomAlgo_Point2D.h>
24 #include <ModelGeomAlgo_Point2D.h>
26 #include <SketchPlugin_ConstraintCoincidence.h>
27 #include <SketchPlugin_Constraint.h>
28 #include <SketchPlugin_Point.h>
30 #include <ModuleBase_IViewWindow.h>
31 #include <ModuleBase_IWorkshop.h>
32 #include <ModuleBase_IModule.h>
34 #include <Config_WidgetAPI.h>
36 #include <XGUI_Tools.h>
37 #include <XGUI_Workshop.h>
38 #include <XGUI_Displayer.h>
41 #include <QMouseEvent>
43 PartSet_WidgetSubShapeSelector::PartSet_WidgetSubShapeSelector(QWidget* theParent,
44 ModuleBase_IWorkshop* theWorkshop,
45 const Config_WidgetAPI* theData)
46 : ModuleBase_WidgetShapeSelector(theParent, theWorkshop, theData)
48 myCurrentSubShape = std::shared_ptr<ModuleBase_ViewerPrs>(new ModuleBase_ViewerPrs());
51 PartSet_WidgetSubShapeSelector::~PartSet_WidgetSubShapeSelector()
53 myCashedShapes.clear();
56 //********************************************************************
57 void PartSet_WidgetSubShapeSelector::activateCustom()
59 ModuleBase_WidgetShapeSelector::activateCustom();
61 myWorkshop->module()->activateCustomPrs(myFeature,
62 ModuleBase_IModule::CustomizeHighlightedObjects, true);
65 //********************************************************************
66 void PartSet_WidgetSubShapeSelector::deactivate()
68 ModuleBase_WidgetShapeSelector::deactivate();
70 myWorkshop->module()->deactivateCustomPrs(ModuleBase_IModule::CustomizeHighlightedObjects, true);
73 //********************************************************************
74 void PartSet_WidgetSubShapeSelector::mouseMoved(ModuleBase_IViewWindow* theWindow,
75 QMouseEvent* theEvent)
77 ModuleBase_ISelection* aSelect = myWorkshop->selection();
78 QList<ModuleBase_ViewerPrsPtr> aHighlighted = aSelect->getHighlighted();
80 if (!aHighlighted.empty()) {
81 ModuleBase_ViewerPrsPtr aPrs = aHighlighted.first();
82 if (aPrs.get() && aPrs->object().get()) {
83 ObjectPtr anObject = aPrs->object();
84 if (myCashedShapes.find(anObject) == myCashedShapes.end())
85 fillObjectShapes(anObject);
86 const std::set<GeomShapePtr>& aShapes = myCashedShapes[anObject];
87 if (!aShapes.empty()) {
88 gp_Pnt aPnt = PartSet_Tools::convertClickToPoint(theEvent->pos(), theWindow->v3dView());
90 Handle(V3d_View) aView = theWindow->v3dView();
91 PartSet_Tools::convertTo2D(aPnt, mySketch, aView, aX, anY);
92 std::shared_ptr<GeomAPI_Pnt> aPoint = PartSet_Tools::convertTo3D(aX, anY, mySketch);
94 std::set<GeomShapePtr>::const_iterator anIt = aShapes.begin(), aLast = aShapes.end();
95 for (; anIt != aLast; anIt++) {
96 GeomShapePtr aBaseShape = *anIt;
97 std::shared_ptr<GeomAPI_Pnt> aProjectedPoint;
98 if (ModelGeomAlgo_Point2D::isPointOnEdge(aBaseShape, aPoint, aProjectedPoint)) {
99 XGUI_Tools::workshop(myWorkshop)->displayer()->clearSelected(false);
100 if (myCurrentSubShape->object() != anObject ||
101 myCurrentSubShape->shape() != aBaseShape) {
102 myCurrentSubShape->setObject(anObject);
103 myCurrentSubShape->setShape(aBaseShape);
105 ModuleBase_IModule* aModule = myWorkshop->module();
108 if (!aModule->isCustomPrsActivated(ModuleBase_IModule::CustomizeHighlightedObjects))
109 aModule->activateCustomPrs(myFeature,
110 ModuleBase_IModule::CustomizeHighlightedObjects, true);
111 aModule->customizeObject(myFeature,
112 ModuleBase_IModule::CustomizeHighlightedObjects, true);
115 XGUI_Tools::workshop(myWorkshop)->displayer()->updateViewer();;
124 //********************************************************************
125 void PartSet_WidgetSubShapeSelector::getGeomSelection(const ModuleBase_ViewerPrsPtr& thePrs,
126 ObjectPtr& theObject,
127 GeomShapePtr& theShape)
129 ModuleBase_ISelection* aSelection = myWorkshop->selection();
130 theObject = aSelection->getResult(thePrs);
131 if (!theObject.get() && myCurrentSubShape->object())
132 theObject = myCurrentSubShape->object();
135 //********************************************************************
136 QList<ModuleBase_ViewerPrsPtr> PartSet_WidgetSubShapeSelector::getAttributeSelection() const
138 return QList<ModuleBase_ViewerPrsPtr>();
142 //********************************************************************
143 bool PartSet_WidgetSubShapeSelector::setSelection(
144 QList<std::shared_ptr<ModuleBase_ViewerPrs>>& theValues,
145 const bool theToValidate)
147 ObjectPtr aBaseObject = myCurrentSubShape->object();
148 GeomShapePtr aBaseShape = myCurrentSubShape->shape();
149 bool aResult = aBaseObject.get() && aBaseShape.get();
150 // firstly set the selection to the attribute
152 QList<ModuleBase_ViewerPrsPtr> aValues;
153 aValues.append(myCurrentSubShape);
154 aResult = ModuleBase_WidgetShapeSelector::setSelection(aValues, theToValidate);
156 // secondly fill additional attributes
159 if (aBaseShape->shapeType() == GeomAPI_Shape::EDGE) {
160 std::shared_ptr<GeomAPI_Edge> anEdge(new GeomAPI_Edge(aBaseShape));
162 std::shared_ptr<GeomAPI_Pnt> aFirstPnt = anEdge->firstPoint();
163 std::shared_ptr<GeomAPI_Pnt> aLastPnt = anEdge->lastPoint();
165 std::shared_ptr<GeomDataAPI_Point2D> aFirstPointAttr, aLastPointAttr;
166 /// find the points in feature attributes
167 FeaturePtr aBaseFeature = ModelAPI_Feature::feature(aBaseObject);
168 std::list<AttributePtr> a2DPointAttributes = aBaseFeature->data()->attributes(
169 GeomDataAPI_Point2D::typeId());
170 std::list<AttributePtr>::const_iterator anIt = a2DPointAttributes.begin(),
171 aLast = a2DPointAttributes.end();
172 for (; anIt != aLast; anIt++) {
173 std::shared_ptr<GeomDataAPI_Point2D> anAttributePoint =
174 std::dynamic_pointer_cast<GeomDataAPI_Point2D>(*anIt);
175 std::shared_ptr<GeomAPI_Pnt2d> aPoint2D = anAttributePoint->pnt();
176 std::shared_ptr<GeomAPI_Pnt> aPoint3D = PartSet_Tools::convertTo3D(aPoint2D->x(),
177 aPoint2D->y(), mySketch);
178 if (aFirstPnt->isEqual(aPoint3D))
179 aFirstPointAttr = anAttributePoint;
180 else if (aLastPnt->isEqual(aPoint3D))
181 aLastPointAttr = anAttributePoint;
184 /// find the points in coincident features
185 PntToAttributesMap aRefAttributes = myCashedReferences[aBaseObject];
186 PntToAttributesMap::const_iterator
187 aRIt = aRefAttributes.begin(), aRLast = aRefAttributes.end();
188 for (; aRIt != aRLast; aRIt++) {
189 std::shared_ptr<GeomDataAPI_Point2D> anAttribute = aRIt->first;
190 std::shared_ptr<GeomAPI_Pnt> aPoint = aRIt->second;
191 if (!aFirstPointAttr.get() && aFirstPnt->isEqual(aPoint))
192 aFirstPointAttr = anAttribute;
193 if (!aLastPointAttr.get() && aLastPnt->isEqual(aPoint))
194 aLastPointAttr = anAttribute;
195 if (aFirstPointAttr.get() && aLastPointAttr.get())
198 if (!aFirstPointAttr.get() || !aLastPointAttr)
201 FeaturePtr aFeature = feature();
202 AttributeRefAttrPtr anAPointAttr = std::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(
203 aFeature->attribute(SketchPlugin_Constraint::ENTITY_A()));
204 AttributeRefAttrPtr aBPointAttr = std::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(
205 aFeature->attribute(SketchPlugin_Constraint::ENTITY_B()));
206 anAPointAttr->setAttr(aFirstPointAttr);
207 aBPointAttr->setAttr(aLastPointAttr);
215 //********************************************************************
216 void PartSet_WidgetSubShapeSelector::getHighlighted(
217 QList<std::shared_ptr<ModuleBase_ViewerPrs>>& theValues)
219 if (myCurrentSubShape.get() && myCurrentSubShape->object().get())
220 theValues.append(myCurrentSubShape);
223 //********************************************************************
224 void PartSet_WidgetSubShapeSelector::fillObjectShapes(const ObjectPtr& theObject)
226 std::set<std::shared_ptr<GeomAPI_Shape> > aShapes;
227 std::map<std::shared_ptr<GeomDataAPI_Point2D>, std::shared_ptr<GeomAPI_Pnt> > aPointToAttributes;
228 std::set<std::shared_ptr<GeomDataAPI_Point2D> > aRefAttributes;
230 FeaturePtr aFeature = ModelAPI_Feature::feature(theObject);
231 std::set<GeomShapePtr> anEdgeShapes;
233 ModelAPI_Tools::shapesOfType(aFeature, GeomAPI_Shape::EDGE, anEdgeShapes);
234 if (!anEdgeShapes.empty()) {
235 GeomShapePtr aFeatureShape = *anEdgeShapes.begin();
237 // coincidences to the feature
238 ModelGeomAlgo_Point2D::getPointsOfReference(aFeature, SketchPlugin_ConstraintCoincidence::ID(),
239 aRefAttributes, SketchPlugin_Point::ID(), SketchPlugin_Point::COORD_ID());
240 // layed on feature coincidences to divide it on several shapes
241 CompositeFeaturePtr aSketch = sketch();
242 std::shared_ptr<ModelAPI_Data> aData = aSketch->data();
243 std::shared_ptr<GeomDataAPI_Point> aC = std::dynamic_pointer_cast<GeomDataAPI_Point>(
244 aData->attribute(SketchPlugin_Sketch::ORIGIN_ID()));
245 std::shared_ptr<GeomDataAPI_Dir> aX = std::dynamic_pointer_cast<GeomDataAPI_Dir>(
246 aData->attribute(SketchPlugin_Sketch::DIRX_ID()));
247 std::shared_ptr<GeomDataAPI_Dir> aNorm = std::dynamic_pointer_cast<GeomDataAPI_Dir>(
248 aData->attribute(SketchPlugin_Sketch::NORM_ID()));
249 std::shared_ptr<GeomAPI_Dir> aY(new GeomAPI_Dir(aNorm->dir()->cross(aX->dir())));
250 std::list<std::shared_ptr<GeomAPI_Pnt> > aPoints;
251 ModelGeomAlgo_Point2D::getPointsInsideShape(aFeatureShape, aRefAttributes, aC->pnt(),
252 aX->dir(), aY, aPoints, aPointToAttributes);
254 GeomAlgoAPI_ShapeTools::splitShape(aFeatureShape, aPoints, aShapes);
256 myCashedShapes[theObject] = aShapes;
257 myCashedReferences[theObject] = aPointToAttributes;