1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
3 // File: PartSet_ResultSketchPrs.h
4 // Created: 25 March 2015
5 // Author: Natalia ERMOLAEVA
7 #include "PartSet_ResultSketchPrs.h"
8 #include "PartSet_Tools.h"
9 #include "ModuleBase_Tools.h"
11 #include <ModelAPI_Events.h>
12 #include <ModelAPI_Tools.h>
13 #include <ModelAPI_ResultConstruction.h>
14 #include <GeomAPI_PlanarEdges.h>
16 #include <Events_InfoMessage.h>
17 #include <Events_Loop.h>
19 #include <SketchPlugin_SketchEntity.h>
21 #include <SketcherPrs_Tools.h>
23 #include <Config_PropManager.h>
25 #include <BRep_Builder.hxx>
26 #include <Prs3d_Drawer.hxx>
28 #include <Prs3d_PointAspect.hxx>
29 #include <TopoDS_Builder.hxx>
30 #include <SelectMgr_SelectionManager.hxx>
31 #include <StdPrs_WFDeflectionShape.hxx>
32 #include <StdSelect_BRepSelectionTool.hxx>
33 #include <Graphic3d_AspectLine3d.hxx>
34 #include <Prs3d_LineAspect.hxx>
35 #include <Graphic3d_AspectMarker3d.hxx>
40 #include <TopTools_IndexedMapOfShape.hxx>
44 //*******************************************************************************************
46 IMPLEMENT_STANDARD_HANDLE(PartSet_ResultSketchPrs, ViewerData_AISShape);
47 IMPLEMENT_STANDARD_RTTIEXT(PartSet_ResultSketchPrs, ViewerData_AISShape);
49 PartSet_ResultSketchPrs::PartSet_ResultSketchPrs(ResultPtr theResult)
50 : ViewerData_AISShape(TopoDS_Shape()), myResult(theResult)
52 std::shared_ptr<GeomAPI_Shape> aShapePtr = ModelAPI_Tools::shape(theResult);
54 TopoDS_Shape aShape = aShapePtr->impl<TopoDS_Shape>();
56 Handle(Prs3d_Drawer) aDrawer = Attributes();
57 if (aDrawer->HasOwnPointAspect())
58 aDrawer->PointAspect()->SetTypeOfMarker(Aspect_TOM_PLUS);
60 aDrawer->SetPointAspect(new Prs3d_PointAspect(Aspect_TOM_PLUS, Quantity_NOC_YELLOW, 1.));
62 // Activate individual repaintng if this is a part of compsolid
63 ResultCompSolidPtr aCompSolid = ModelAPI_Tools::compSolidOwner(myResult);
64 SetAutoHilight(aCompSolid.get() == NULL);
66 ModuleBase_Tools::setPointBallHighlighting(this);
69 void PartSet_ResultSketchPrs::Compute(
70 const Handle(PrsMgr_PresentationManager3d)& thePresentationManager,
71 const Handle(Prs3d_Presentation)& thePresentation,
72 const Standard_Integer theMode)
74 thePresentation->Clear();
76 TopoDS_Shape aResultShape;
77 TopoDS_Compound anAuxiliaryCompound;
78 NCollection_List<TopoDS_Shape> aFaceList;
79 fillShapes(aResultShape, anAuxiliaryCompound, mySketchFaceList);
81 bool aReadyToDisplay = !aResultShape.IsNull() || !anAuxiliaryCompound.IsNull();
83 if (aReadyToDisplay) {
84 if (!aResultShape.IsNull()) {
85 myOriginalShape = aResultShape;
86 if (!myOriginalShape.IsNull())
89 myAuxiliaryCompound = anAuxiliaryCompound;
92 setAuxiliaryPresentationStyle(false);
94 // change deviation coefficient to provide more precise circle
95 //ModuleBase_Tools::setDefaultDeviationCoefficient(myResult, Attributes());
96 AIS_Shape::Compute(thePresentationManager, thePresentation, theMode);
98 if (!myAuxiliaryCompound.IsNull()) {
99 setAuxiliaryPresentationStyle(true);
101 Handle(Prs3d_Drawer) aDrawer = Attributes();
102 StdPrs_WFDeflectionShape::Add(thePresentation, myAuxiliaryCompound, aDrawer);
105 if (!aReadyToDisplay) {
106 Events_InfoMessage("PartSet_ResultSketchPrs",
107 "An empty AIS presentation: PartSet_ResultSketchPrs").send();
108 static const Events_ID anEvent = Events_Loop::eventByName(EVENT_EMPTY_AIS_PRESENTATION);
109 ModelAPI_EventCreator::get()->sendUpdated(myResult, anEvent);
114 void debugInfo(const TopoDS_Shape& theShape, const TopAbs_ShapeEnum theType)
116 TopTools_IndexedMapOfShape aSubShapes;
117 TopExp::MapShapes (theShape, theType, aSubShapes);
120 isComesFromDecomposition = !((aSubShapes.Extent() == 1) && (theShape == aSubShapes (1)));
121 int anExtent = aSubShapes.Extent();
122 for (Standard_Integer aShIndex = 1; aShIndex <= aSubShapes.Extent(); ++aShIndex)
124 const TopoDS_Shape& aSubShape = aSubShapes (aShIndex);
130 void PartSet_ResultSketchPrs::ComputeSelection(const Handle(SelectMgr_Selection)& aSelection,
131 const Standard_Integer theMode)
136 aMode != SketcherPrs_Tools::Sel_Sketch_Face &&
137 aMode != SketcherPrs_Tools::Sel_Sketch_Wire)
138 // In order to avoid using custom selection modes
141 bool aShapeIsChanged = false;
142 double aPrevDeviation = Attributes()->DeviationCoefficient();
143 if (aMode == SketcherPrs_Tools::Sel_Sketch_Face ||
144 aMode == SketcherPrs_Tools::Sel_Sketch_Wire) {
145 aMode = (aMode == SketcherPrs_Tools::Sel_Sketch_Face) ? AIS_Shape::SelectionMode(TopAbs_FACE)
146 : AIS_Shape::SelectionMode(TopAbs_WIRE);
148 const TopoDS_Shape& aShape = Shape();
149 debugInfo(aShape, TopAbs_VERTEX); // 24
150 debugInfo(aShape, TopAbs_EDGE); // 12
151 debugInfo(aShape, TopAbs_WIRE); // 0
152 debugInfo(aShape, TopAbs_FACE); // 0
154 BRep_Builder aBuilder;
155 TopoDS_Compound aComp;
156 aBuilder.MakeCompound(aComp);
157 aBuilder.Add(aComp, myOriginalShape);
159 for(NCollection_List<TopoDS_Shape>::Iterator anIt(mySketchFaceList); anIt.More(); anIt.Next()) {
160 const TopoDS_Shape& aFace = anIt.Value();
161 aBuilder.Add(aComp, aFace);
164 debugInfo(aComp, TopAbs_VERTEX); // 24
165 debugInfo(aComp, TopAbs_EDGE); // 12
166 debugInfo(aComp, TopAbs_WIRE); // 4
167 debugInfo(aComp, TopAbs_FACE); // 2
170 double aBodyDefDeflection = Config_PropManager::real("Visualization", "body_deflection",
171 ModelAPI_ResultBody::DEFAULT_DEFLECTION());
172 Attributes()->SetDeviationCoefficient(aBodyDefDeflection);
173 aShapeIsChanged = true;
176 Set(myOriginalShape);
178 // append auxiliary compound to selection of edges/vertices
179 if (aMode == AIS_Shape::SelectionMode(TopAbs_EDGE) ||
180 aMode == AIS_Shape::SelectionMode(TopAbs_VERTEX)) {
182 bool isVertex = aMode == AIS_Shape::SelectionMode(TopAbs_VERTEX);
183 appendShapeSelection(aSelection, myAuxiliaryCompound, isVertex ? TopAbs_VERTEX : TopAbs_EDGE);
186 AIS_Shape::ComputeSelection(aSelection, aMode);
188 if (aShapeIsChanged) {
189 Attributes()->SetDeviationCoefficient(aPrevDeviation);
190 Set(myOriginalShape);
194 void PartSet_ResultSketchPrs::appendShapeSelection(const Handle(SelectMgr_Selection)& theSelection,
195 const TopoDS_Shape& theShape,
196 const TopAbs_ShapeEnum& theTypeOfSelection)
198 // POP protection against crash in low layers
199 Standard_Real aDeflection = Prs3d::GetDeflection(theShape, myDrawer);
201 StdSelect_BRepSelectionTool::Load(theSelection,
206 myDrawer->HLRAngle(),
207 myDrawer->IsAutoTriangulation());
208 } catch ( Standard_Failure ) {
212 void PartSet_ResultSketchPrs::setAuxiliaryPresentationStyle(const bool isAuxiliary)
214 Standard_Integer aLineStyle = isAuxiliary?
215 SketchPlugin_SketchEntity::SKETCH_LINE_STYLE_AUXILIARY() :
216 SketchPlugin_SketchEntity::SKETCH_LINE_STYLE();
218 Handle(Prs3d_Drawer) aDrawer = Attributes();
221 Handle(Prs3d_LineAspect) aLineAspect;
223 Aspect_TypeOfLine aType = (Aspect_TypeOfLine)aLineStyle;
224 if (aDrawer->HasOwnLineAspect()) {
225 aLineAspect = aDrawer->LineAspect();
227 if (aDrawer->HasOwnWireAspect()) {
228 aLineAspect = aDrawer->WireAspect();
230 Quantity_Color aCurrentColor;
231 Aspect_TypeOfLine aPrevLineType;
232 Standard_Real aCurrentWidth;
233 aLineAspect->Aspect()->Values(aCurrentColor, aPrevLineType, aCurrentWidth);
234 bool isChangedLineType = aType != aPrevLineType;
235 if (isChangedLineType) {
236 aLineAspect->SetTypeOfLine(aType);
240 void PartSet_ResultSketchPrs::fillShapes(TopoDS_Shape& theResultShape,
241 TopoDS_Compound& theAuxiliaryCompound,
242 NCollection_List<TopoDS_Shape>& theFaceList)
244 //if (!aResultShape.IsNull() || !anAuxiliaryCompound.IsNull())
245 std::shared_ptr<GeomAPI_Shape> aShapePtr = ModelAPI_Tools::shape(myResult);
250 ResultConstructionPtr aConstruction =
251 std::dynamic_pointer_cast<ModelAPI_ResultConstruction>(myResult);
252 if (aConstruction.get()) {
253 int aFacesNum = aConstruction->facesNum();
254 for(int aFaceIndex = 0; aFaceIndex < aFacesNum; aFaceIndex++) {
255 std::shared_ptr<GeomAPI_Face> aFaceShape = aConstruction->face(aFaceIndex);
256 if (aFaceShape.get()) {
257 TopoDS_Shape aFace = (aFaceShape)->impl<TopoDS_Shape>();
258 theFaceList.Append(aFace);
262 theResultShape = aShapePtr->impl<TopoDS_Shape>();
264 /// find auxiliary shapes
265 FeaturePtr aResultFeature = ModelAPI_Feature::feature(myResult);
266 CompositeFeaturePtr aSketchFeature = std::dynamic_pointer_cast<ModelAPI_CompositeFeature>
268 std::list<ResultPtr> anAuxiliaryResults;
269 for (int i = 0; i < aSketchFeature->numberOfSubs(); i++) {
270 FeaturePtr aFeature = aSketchFeature->subFeature(i);
271 if (PartSet_Tools::isAuxiliarySketchEntity(aFeature)) {
272 std::list<ResultPtr> aResults = aFeature->results();
273 std::list<ResultPtr>::const_iterator aIt;
274 for (aIt = aResults.cbegin(); aIt != aResults.cend(); ++aIt) {
275 ResultPtr aResult = *aIt;
276 if (aResult.get() && aResult->shape().get())
277 anAuxiliaryResults.push_back(aResult);
281 /// append not-edges shapes, e.g. center of a circle, an arc, a point feature
282 const std::list<std::shared_ptr<ModelAPI_Result> >& aRes = aFeature->results();
283 std::list<std::shared_ptr<ModelAPI_Result> >::const_iterator aResIter = aRes.cbegin();
284 for (; aResIter != aRes.cend(); aResIter++) {
285 std::shared_ptr<ModelAPI_ResultConstruction> aConstr = std::dynamic_pointer_cast<
286 ModelAPI_ResultConstruction>(*aResIter);
288 std::shared_ptr<GeomAPI_Shape> aGeomShape = aConstr->shape();
289 if (aGeomShape.get()) {
290 const TopoDS_Shape& aShape = aGeomShape->impl<TopoDS_Shape>();
291 if (aShape.ShapeType() != TopAbs_EDGE)
292 anAuxiliaryResults.push_back(aConstr);
299 if (anAuxiliaryResults.size() > 0) {
300 BRep_Builder aBuilder;
301 //TopoDS_Compound aComp;
302 aBuilder.MakeCompound(theAuxiliaryCompound);
303 std::list<ResultPtr>::const_iterator anIt = anAuxiliaryResults.begin(),
304 aLast = anAuxiliaryResults.end();
305 for (; anIt != aLast; anIt++) {
306 ResultPtr aResult = *anIt;
308 GeomShapePtr aGeomShape = aResult->shape();
309 if (aGeomShape.get()) {
310 const TopoDS_Shape& aShape = aGeomShape->impl<TopoDS_Shape>();
311 if (!aShape.IsNull())
312 aBuilder.Add(theAuxiliaryCompound, aShape);