1 // Copyright (C) 2014-2022 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_ResultSketchPrs.h"
21 #include "PartSet_Tools.h"
22 #include "ModuleBase_Tools.h"
24 #include <ModelAPI_Events.h>
25 #include <ModelAPI_Tools.h>
26 #include <ModelAPI_ResultConstruction.h>
27 #include <ModelAPI_ResultBody.h>
28 #include <GeomAPI_PlanarEdges.h>
30 #include <Events_InfoMessage.h>
31 #include <Events_Loop.h>
33 #include <SketchPlugin_SketchEntity.h>
35 #include <SketcherPrs_Tools.h>
37 #include <Config_PropManager.h>
39 #include <BRep_Builder.hxx>
40 #include <Prs3d_Drawer.hxx>
42 #include <Prs3d_PointAspect.hxx>
43 #include <TopoDS_Builder.hxx>
44 #include <SelectMgr_SelectionManager.hxx>
45 #include <StdPrs_WFShape.hxx>
46 #include <StdSelect_BRepSelectionTool.hxx>
47 #include <Graphic3d_AspectLine3d.hxx>
48 #include <Prs3d_LineAspect.hxx>
49 #include <Graphic3d_AspectMarker3d.hxx>
54 #include <TopTools_IndexedMapOfShape.hxx>
58 #include <Standard_Version.hxx>
60 #if OCC_VERSION_HEX > 0x070400
61 #include <StdPrs_ToolTriangulatedShape.hxx>
64 //*******************************************************************************************
66 IMPLEMENT_STANDARD_RTTIEXT(PartSet_ResultSketchPrs, ViewerData_AISShape);
68 PartSet_ResultSketchPrs::PartSet_ResultSketchPrs(ResultPtr theResult)
69 : ViewerData_AISShape(TopoDS_Shape()), myResult(theResult)
71 std::shared_ptr<GeomAPI_Shape> aShapePtr = ModelAPI_Tools::shape(theResult);
73 TopoDS_Shape aShape = aShapePtr->impl<TopoDS_Shape>();
75 Handle(Prs3d_Drawer) aDrawer = Attributes();
76 if (aDrawer->HasOwnPointAspect())
77 aDrawer->PointAspect()->SetTypeOfMarker(Aspect_TOM_PLUS);
79 aDrawer->SetPointAspect(new Prs3d_PointAspect(Aspect_TOM_PLUS, Quantity_NOC_YELLOW, 1.));
80 //aDrawer->SetTypeOfDeflection(Aspect_TOD_ABSOLUTE);
82 // Activate individual repaintng if this is a part of compsolid
83 ResultBodyPtr anOwner = ModelAPI_Tools::bodyOwner(myResult);
84 SetAutoHilight(anOwner.get() == NULL);
86 ModuleBase_Tools::setPointBallHighlighting(this);
88 // change deviation coefficient to provide more precise circle
89 ModuleBase_Tools::setDefaultDeviationCoefficient(Shape(), DynamicHilightAttributes());
90 ModuleBase_Tools::setDefaultDeviationCoefficient(Shape(), Attributes());
93 void PartSet_ResultSketchPrs::Compute(
94 const Handle(PrsMgr_PresentationManager3d)& thePresentationManager,
95 const Handle(Prs3d_Presentation)& thePresentation,
96 const Standard_Integer theMode)
98 thePresentation->Clear();
100 TopoDS_Shape aResultShape;
101 TopoDS_Compound anAuxiliaryCompound;
102 NCollection_List<TopoDS_Shape> aFaceList;
103 fillShapes(aResultShape, anAuxiliaryCompound, mySketchFaceList);
105 bool aReadyToDisplay = !aResultShape.IsNull() || !anAuxiliaryCompound.IsNull();
107 if (aReadyToDisplay) {
108 if (!aResultShape.IsNull()) {
109 myOriginalShape = aResultShape;
110 if (!myOriginalShape.IsNull())
111 Set(myOriginalShape);
113 myAuxiliaryCompound = anAuxiliaryCompound;
116 setAuxiliaryPresentationStyle(false);
119 AIS_Shape::Compute(thePresentationManager, thePresentation, theMode);
124 if (!myAuxiliaryCompound.IsNull()) {
125 setAuxiliaryPresentationStyle(true);
127 Handle(Prs3d_Drawer) aDrawer = Attributes();
128 StdPrs_WFShape::Add(thePresentation, myAuxiliaryCompound, aDrawer);
131 if (!aReadyToDisplay) {
132 Events_InfoMessage("PartSet_ResultSketchPrs",
133 "An empty AIS presentation: PartSet_ResultSketchPrs").send();
134 static const Events_ID anEvent = Events_Loop::eventByName(EVENT_EMPTY_AIS_PRESENTATION);
135 ModelAPI_EventCreator::get()->sendUpdated(myResult, anEvent);
140 void debugInfo(const TopoDS_Shape& theShape, const TopAbs_ShapeEnum theType)
142 TopTools_IndexedMapOfShape aSubShapes;
143 TopExp::MapShapes (theShape, theType, aSubShapes);
146 isComesFromDecomposition = !((aSubShapes.Extent() == 1) && (theShape == aSubShapes (1)));
147 int anExtent = aSubShapes.Extent();
148 for (Standard_Integer aShIndex = 1; aShIndex <= aSubShapes.Extent(); ++aShIndex)
150 const TopoDS_Shape& aSubShape = aSubShapes (aShIndex);
156 void PartSet_ResultSketchPrs::ComputeSelection(const Handle(SelectMgr_Selection)& aSelection,
157 const Standard_Integer theMode)
162 aMode != SketcherPrs_Tools::Sel_Sketch_Face &&
163 aMode != SketcherPrs_Tools::Sel_Sketch_Wire)
164 // In order to avoid using custom selection modes
167 bool aShapeIsChanged = false;
168 double aPrevDeviation = Attributes()->DeviationCoefficient();
169 if (aMode == SketcherPrs_Tools::Sel_Sketch_Face ||
170 aMode == SketcherPrs_Tools::Sel_Sketch_Wire) {
171 aMode = (aMode == SketcherPrs_Tools::Sel_Sketch_Face) ? AIS_Shape::SelectionMode(TopAbs_FACE)
172 : AIS_Shape::SelectionMode(TopAbs_WIRE);
174 const TopoDS_Shape& aShape = Shape();
175 debugInfo(aShape, TopAbs_VERTEX); // 24
176 debugInfo(aShape, TopAbs_EDGE); // 12
177 debugInfo(aShape, TopAbs_WIRE); // 0
178 debugInfo(aShape, TopAbs_FACE); // 0
180 BRep_Builder aBuilder;
181 TopoDS_Compound aComp;
182 aBuilder.MakeCompound(aComp);
183 aBuilder.Add(aComp, myOriginalShape);
185 for(NCollection_List<TopoDS_Shape>::Iterator anIt(mySketchFaceList); anIt.More(); anIt.Next()) {
186 const TopoDS_Shape& aFace = anIt.Value();
187 aBuilder.Add(aComp, aFace);
190 debugInfo(aComp, TopAbs_VERTEX); // 24
191 debugInfo(aComp, TopAbs_EDGE); // 12
192 debugInfo(aComp, TopAbs_WIRE); // 4
193 debugInfo(aComp, TopAbs_FACE); // 2
196 double aBodyDefDeflection = Config_PropManager::real("Visualization", "body_deflection");
197 Attributes()->SetDeviationCoefficient(aBodyDefDeflection);
198 aShapeIsChanged = true;
201 Set(myOriginalShape);
203 // append auxiliary compound to selection of edges/vertices
204 if (aMode == AIS_Shape::SelectionMode(TopAbs_EDGE) ||
205 aMode == AIS_Shape::SelectionMode(TopAbs_VERTEX)) {
207 bool isVertex = aMode == AIS_Shape::SelectionMode(TopAbs_VERTEX);
208 appendShapeSelection(aSelection, myAuxiliaryCompound, isVertex ? TopAbs_VERTEX : TopAbs_EDGE);
211 AIS_Shape::ComputeSelection(aSelection, aMode);
213 if (aShapeIsChanged) {
214 Attributes()->SetDeviationCoefficient(aPrevDeviation);
215 Set(myOriginalShape);
219 void PartSet_ResultSketchPrs::appendShapeSelection(const Handle(SelectMgr_Selection)& theSelection,
220 const TopoDS_Shape& theShape,
221 const TopAbs_ShapeEnum& theTypeOfSelection)
223 // POP protection against crash in low layers
224 #if OCC_VERSION_HEX > 0x070400
225 Standard_Real aDeflection = StdPrs_ToolTriangulatedShape::GetDeflection(theShape, myDrawer);
227 Standard_Real aDeflection = Prs3d::GetDeflection(theShape, myDrawer);
230 StdSelect_BRepSelectionTool::Load(theSelection,
235 myDrawer->DeviationAngle(),
236 myDrawer->IsAutoTriangulation());
237 } catch ( Standard_Failure ) {
241 void PartSet_ResultSketchPrs::setAuxiliaryPresentationStyle(const bool isAuxiliary)
243 Standard_Integer aLineStyle = isAuxiliary?
244 SketchPlugin_SketchEntity::SKETCH_LINE_STYLE_AUXILIARY() :
245 SketchPlugin_SketchEntity::SKETCH_LINE_STYLE();
247 Handle(Prs3d_Drawer) aDrawer = Attributes();
250 Handle(Prs3d_LineAspect) aLineAspect;
252 Aspect_TypeOfLine aType = (Aspect_TypeOfLine)aLineStyle;
253 if (aDrawer->HasOwnLineAspect()) {
254 aLineAspect = aDrawer->LineAspect();
256 if (aDrawer->HasOwnWireAspect()) {
257 aLineAspect = aDrawer->WireAspect();
259 Quantity_Color aCurrentColor = aLineAspect->Aspect()->Color();
260 Aspect_TypeOfLine aPrevLineType = aLineAspect->Aspect()->Type();
261 Standard_Real aCurrentWidth = aLineAspect->Aspect()->Width();
262 bool isChangedLineType = aType != aPrevLineType;
263 if (isChangedLineType) {
264 Handle(Prs3d_LineAspect) aAspect = new Prs3d_LineAspect(aCurrentColor, aType, aCurrentWidth);
265 aDrawer->SetLineAspect(aAspect);
266 aDrawer->SetWireAspect(aAspect);
267 SetAttributes(aDrawer);
271 void PartSet_ResultSketchPrs::fillShapes(TopoDS_Shape& theResultShape,
272 TopoDS_Compound& theAuxiliaryCompound,
273 NCollection_List<TopoDS_Shape>& theFaceList)
275 //if (!aResultShape.IsNull() || !anAuxiliaryCompound.IsNull())
276 std::shared_ptr<GeomAPI_Shape> aShapePtr = ModelAPI_Tools::shape(myResult);
281 ResultConstructionPtr aConstruction =
282 std::dynamic_pointer_cast<ModelAPI_ResultConstruction>(myResult);
283 if (aConstruction.get()) {
284 int aFacesNum = aConstruction->facesNum();
285 for(int aFaceIndex = 0; aFaceIndex < aFacesNum; aFaceIndex++) {
286 std::shared_ptr<GeomAPI_Face> aFaceShape = aConstruction->face(aFaceIndex);
287 if (aFaceShape.get()) {
288 TopoDS_Shape aFace = (aFaceShape)->impl<TopoDS_Shape>();
289 theFaceList.Append(aFace);
293 theResultShape = aShapePtr->impl<TopoDS_Shape>();
295 /// find auxiliary shapes
296 FeaturePtr aResultFeature = ModelAPI_Feature::feature(myResult);
297 CompositeFeaturePtr aSketchFeature = std::dynamic_pointer_cast<ModelAPI_CompositeFeature>
299 std::list<ResultPtr> anAuxiliaryResults;
300 int aNumberOfSubs = aSketchFeature->numberOfSubs();
301 for (int i = 0; i < aNumberOfSubs; i++) {
302 FeaturePtr aFeature = aSketchFeature->subFeature(i);
303 if (PartSet_Tools::isAuxiliarySketchEntity(aFeature)) {
304 std::list<ResultPtr> aResults = aFeature->results();
305 std::list<ResultPtr>::const_iterator aIt;
306 for (aIt = aResults.cbegin(); aIt != aResults.cend(); ++aIt) {
307 ResultPtr aResult = *aIt;
308 if (aResult.get() && aResult->shape().get())
309 anAuxiliaryResults.push_back(aResult);
312 else if (PartSet_Tools::isIncludeIntoSketchResult(aFeature)) {
313 // Append not-edges shapes, e.g. center of a circle, an arc, a point feature.
314 // Issue #2535: do not show elements produced by Projection or Intersection features,
315 // which should not be included into the sketch result.
316 const std::list<std::shared_ptr<ModelAPI_Result> >& aRes = aFeature->results();
317 std::list<std::shared_ptr<ModelAPI_Result> >::const_iterator aResIter = aRes.cbegin();
318 for (; aResIter != aRes.cend(); aResIter++) {
319 std::shared_ptr<ModelAPI_ResultConstruction> aConstr = std::dynamic_pointer_cast<
320 ModelAPI_ResultConstruction>(*aResIter);
322 std::shared_ptr<GeomAPI_Shape> aGeomShape = aConstr->shape();
323 if (aGeomShape.get()) {
324 const TopoDS_Shape& aShape = aGeomShape->impl<TopoDS_Shape>();
325 if (aShape.ShapeType() != TopAbs_EDGE)
326 anAuxiliaryResults.push_back(aConstr);
333 if (anAuxiliaryResults.size() > 0) {
334 BRep_Builder aBuilder;
335 //TopoDS_Compound aComp;
336 aBuilder.MakeCompound(theAuxiliaryCompound);
337 std::list<ResultPtr>::const_iterator anIt = anAuxiliaryResults.begin(),
338 aLast = anAuxiliaryResults.end();
339 for (; anIt != aLast; anIt++) {
340 ResultPtr aResult = *anIt;
342 GeomShapePtr aGeomShape = aResult->shape();
343 if (aGeomShape.get()) {
344 const TopoDS_Shape& aShape = aGeomShape->impl<TopoDS_Shape>();
345 if (!aShape.IsNull())
346 aBuilder.Add(theAuxiliaryCompound, aShape);