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_Error.h>
17 #include <Events_Loop.h>
19 #include <SketchPlugin_SketchEntity.h>
21 #include <Config_PropManager.h>
23 #include <BRep_Builder.hxx>
24 #include <Prs3d_Drawer.hxx>
26 #include <Prs3d_PointAspect.hxx>
27 #include <TopoDS_Builder.hxx>
28 #include <SelectMgr_SelectionManager.hxx>
29 #include <StdPrs_WFDeflectionShape.hxx>
30 #include <StdSelect_BRepSelectionTool.hxx>
31 #include <Graphic3d_AspectLine3d.hxx>
32 #include <Prs3d_LineAspect.hxx>
37 #include <TopTools_IndexedMapOfShape.hxx>
41 //*******************************************************************************************
43 IMPLEMENT_STANDARD_HANDLE(PartSet_ResultSketchPrs, ViewerData_AISShape);
44 IMPLEMENT_STANDARD_RTTIEXT(PartSet_ResultSketchPrs, ViewerData_AISShape);
46 PartSet_ResultSketchPrs::PartSet_ResultSketchPrs(ResultPtr theResult)
47 : ViewerData_AISShape(TopoDS_Shape()), myResult(theResult)
49 std::shared_ptr<GeomAPI_Shape> aShapePtr = ModelAPI_Tools::shape(theResult);
51 TopoDS_Shape aShape = aShapePtr->impl<TopoDS_Shape>();
53 Handle(Prs3d_Drawer) aDrawer = Attributes();
54 if (aDrawer->HasOwnPointAspect())
55 aDrawer->PointAspect()->SetTypeOfMarker(Aspect_TOM_PLUS);
57 aDrawer->SetPointAspect(new Prs3d_PointAspect(Aspect_TOM_PLUS, Quantity_NOC_YELLOW, 1.));
59 // Activate individual repaintng if this is a part of compsolid
60 ResultCompSolidPtr aCompSolid = ModelAPI_Tools::compSolidOwner(myResult);
61 SetAutoHilight(aCompSolid.get() == NULL);
64 bool PartSet_ResultSketchPrs::isValidShapeType(const TopAbs_ShapeEnum& theBaseType,
65 const TopAbs_ShapeEnum& theCheckedType)
67 bool aValid = theBaseType == theCheckedType;
69 // currently this functionality is for all, as we have no separate wire selection mode
70 // lately it should be corrected to have the following check only for sketch presentations
71 aValid = theBaseType == TopAbs_FACE && theCheckedType == TopAbs_WIRE;
76 void PartSet_ResultSketchPrs::Compute(const Handle(PrsMgr_PresentationManager3d)& thePresentationManager,
77 const Handle(Prs3d_Presentation)& thePresentation,
78 const Standard_Integer theMode)
80 thePresentation->Clear();
82 TopoDS_Shape aResultShape;
83 TopoDS_Compound anAuxiliaryCompound;
84 NCollection_List<TopoDS_Shape> aFaceList;
85 fillShapes(aResultShape, anAuxiliaryCompound, mySketchFaceList);
87 bool aReadyToDisplay = !aResultShape.IsNull() || !anAuxiliaryCompound.IsNull();
89 if (aReadyToDisplay) {
90 if (!aResultShape.IsNull()) {
91 myOriginalShape = aResultShape;
92 if (!myOriginalShape.IsNull())
95 myAuxiliaryCompound = anAuxiliaryCompound;
98 setAuxiliaryPresentationStyle(false);
100 // change deviation coefficient to provide more precise circle
101 ModuleBase_Tools::setDefaultDeviationCoefficient(Shape(), Attributes());
102 AIS_Shape::Compute(thePresentationManager, thePresentation, theMode);
104 if (!myAuxiliaryCompound.IsNull()) {
105 setAuxiliaryPresentationStyle(true);
107 Handle(Prs3d_Drawer) aDrawer = Attributes();
108 StdPrs_WFDeflectionShape::Add(thePresentation, myAuxiliaryCompound, aDrawer);
111 if (!aReadyToDisplay) {
112 Events_Error::throwException("An empty AIS presentation: PartSet_ResultSketchPrs");
113 static const Events_ID anEvent = Events_Loop::eventByName(EVENT_EMPTY_AIS_PRESENTATION);
114 ModelAPI_EventCreator::get()->sendUpdated(myResult, anEvent);
119 void debugInfo(const TopoDS_Shape& theShape, const TopAbs_ShapeEnum theType)
121 TopTools_IndexedMapOfShape aSubShapes;
122 TopExp::MapShapes (theShape, theType, aSubShapes);
124 Standard_Boolean isComesFromDecomposition = !((aSubShapes.Extent() == 1) && (theShape == aSubShapes (1)));
125 int anExtent = aSubShapes.Extent();
126 for (Standard_Integer aShIndex = 1; aShIndex <= aSubShapes.Extent(); ++aShIndex)
128 const TopoDS_Shape& aSubShape = aSubShapes (aShIndex);
134 void PartSet_ResultSketchPrs::ComputeSelection(const Handle(SelectMgr_Selection)& aSelection,
135 const Standard_Integer aMode)
138 // In order to avoid using custom selection modes
141 if (aMode == AIS_Shape::SelectionMode(TopAbs_FACE)) {
143 const TopoDS_Shape& aShape = Shape();
144 debugInfo(aShape, TopAbs_VERTEX); // 24
145 debugInfo(aShape, TopAbs_EDGE); // 12
146 debugInfo(aShape, TopAbs_WIRE); // 0
147 debugInfo(aShape, TopAbs_FACE); // 0
149 BRep_Builder aBuilder;
150 TopoDS_Compound aComp;
151 aBuilder.MakeCompound(aComp);
152 aBuilder.Add(aComp, myOriginalShape);
154 for(NCollection_List<TopoDS_Shape>::Iterator anIt(mySketchFaceList); anIt.More(); anIt.Next()) {
155 const TopoDS_Shape& aFace = anIt.Value();
156 aBuilder.Add(aComp, aFace);
157 // for sketch presentation in the face mode wires should be selectable also
158 // accoring to #1343 Improvement of Extrusion and Revolution operations
159 appendShapeSelection(aSelection, aFace, TopAbs_WIRE);
162 debugInfo(aComp, TopAbs_VERTEX); // 24
163 debugInfo(aComp, TopAbs_EDGE); // 12
164 debugInfo(aComp, TopAbs_WIRE); // 4
165 debugInfo(aComp, TopAbs_FACE); // 2
169 Set(myOriginalShape);
171 // append auxiliary compound to selection of edges/vertices
172 if (aMode == AIS_Shape::SelectionMode(TopAbs_EDGE) ||
173 aMode == AIS_Shape::SelectionMode(TopAbs_VERTEX)) {
175 bool isVertex = aMode == AIS_Shape::SelectionMode(TopAbs_VERTEX);
176 appendShapeSelection(aSelection, myAuxiliaryCompound, isVertex ? TopAbs_VERTEX : TopAbs_EDGE);
179 AIS_Shape::ComputeSelection(aSelection, aMode);
182 void PartSet_ResultSketchPrs::appendShapeSelection(const Handle(SelectMgr_Selection)& theSelection,
183 const TopoDS_Shape& theShape,
184 const TopAbs_ShapeEnum& theTypeOfSelection)
186 // POP protection against crash in low layers
187 Standard_Real aDeflection = Prs3d::GetDeflection(theShape, myDrawer);
189 StdSelect_BRepSelectionTool::Load(theSelection,
194 myDrawer->HLRAngle(),
195 myDrawer->IsAutoTriangulation());
196 } catch ( Standard_Failure ) {
200 void PartSet_ResultSketchPrs::setAuxiliaryPresentationStyle(const bool isAuxiliary)
202 std::vector<int> aColor;
203 Standard_Real aWidth;
204 Standard_Integer aLineStyle;
207 aColor = Config_PropManager::color("Visualization", "result_construction_color",
208 ModelAPI_ResultConstruction::DEFAULT_COLOR());
209 aWidth = PartSet_Tools::getAISDefaultWidth();// default width value
210 aLineStyle = SketchPlugin_SketchEntity::SKETCH_LINE_STYLE();
213 aColor = Config_PropManager::color("Visualization", "sketch_auxiliary_color",
214 SKETCH_AUXILIARY_COLOR);
215 aWidth = SketchPlugin_SketchEntity::SKETCH_LINE_WIDTH_AUXILIARY();
216 aLineStyle = SketchPlugin_SketchEntity::SKETCH_LINE_STYLE_AUXILIARY();
219 Handle(Prs3d_Drawer) aDrawer = Attributes();
220 setColor(aDrawer, Quantity_Color(aColor[0] / 255., aColor[1] / 255., aColor[2] / 255.,
223 setWidth(aDrawer, aWidth);
225 Handle(Prs3d_LineAspect) aLineAspect;
227 Aspect_TypeOfLine aType = (Aspect_TypeOfLine)aLineStyle;
228 if (aDrawer->HasOwnLineAspect()) {
229 aLineAspect = aDrawer->LineAspect();
231 if (aDrawer->HasOwnWireAspect()) {
232 aLineAspect = aDrawer->WireAspect();
234 Quantity_Color aCurrentColor;
235 Aspect_TypeOfLine aPrevLineType;
236 Standard_Real aCurrentWidth;
237 aLineAspect->Aspect()->Values(aCurrentColor, aPrevLineType, aCurrentWidth);
238 bool isChangedLineType = aType != aPrevLineType;
239 if (isChangedLineType) {
240 aLineAspect->SetTypeOfLine(aType);
244 void PartSet_ResultSketchPrs::fillShapes(TopoDS_Shape& theResultShape,
245 TopoDS_Compound& theAuxiliaryCompound,
246 NCollection_List<TopoDS_Shape>& theFaceList)
248 //if (!aResultShape.IsNull() || !anAuxiliaryCompound.IsNull())
249 std::shared_ptr<GeomAPI_Shape> aShapePtr = ModelAPI_Tools::shape(myResult);
254 ResultConstructionPtr aConstruction =
255 std::dynamic_pointer_cast<ModelAPI_ResultConstruction>(myResult);
256 if (aConstruction.get()) {
257 int aFacesNum = aConstruction->facesNum();
258 for(int aFaceIndex = 0; aFaceIndex < aFacesNum; aFaceIndex++) {
259 std::shared_ptr<GeomAPI_Face> aFaceShape = aConstruction->face(aFaceIndex);
260 if (aFaceShape.get()) {
261 TopoDS_Shape aFace = (aFaceShape)->impl<TopoDS_Shape>();
262 theFaceList.Append(aFace);
266 theResultShape = aShapePtr->impl<TopoDS_Shape>();
268 /// find auxiliary shapes
269 FeaturePtr aResultFeature = ModelAPI_Feature::feature(myResult);
270 CompositeFeaturePtr aSketchFeature = std::dynamic_pointer_cast<ModelAPI_CompositeFeature>
272 std::list<ResultPtr> anAuxiliaryResults;
273 for (int i = 0; i < aSketchFeature->numberOfSubs(); i++) {
274 FeaturePtr aFeature = aSketchFeature->subFeature(i);
275 if (PartSet_Tools::isAuxiliarySketchEntity(aFeature)) {
276 std::list<ResultPtr> aResults = aFeature->results();
277 std::list<ResultPtr>::const_iterator aIt;
278 for (aIt = aResults.cbegin(); aIt != aResults.cend(); ++aIt) {
279 ResultPtr aResult = *aIt;
280 if (aResult.get() && aResult->shape().get())
281 anAuxiliaryResults.push_back(aResult);
285 /// append not-edges shapes, e.g. center of a circle, an arc, a point feature
286 const std::list<std::shared_ptr<ModelAPI_Result> >& aRes = aFeature->results();
287 std::list<std::shared_ptr<ModelAPI_Result> >::const_iterator aResIter = aRes.cbegin();
288 for (; aResIter != aRes.cend(); aResIter++) {
289 std::shared_ptr<ModelAPI_ResultConstruction> aConstr = std::dynamic_pointer_cast<
290 ModelAPI_ResultConstruction>(*aResIter);
292 std::shared_ptr<GeomAPI_Shape> aGeomShape = aConstr->shape();
293 if (aGeomShape.get()) {
294 const TopoDS_Shape& aShape = aGeomShape->impl<TopoDS_Shape>();
295 if (aShape.ShapeType() != TopAbs_EDGE)
296 anAuxiliaryResults.push_back(aConstr);
303 if (anAuxiliaryResults.size() > 0) {
304 BRep_Builder aBuilder;
305 //TopoDS_Compound aComp;
306 aBuilder.MakeCompound(theAuxiliaryCompound);
307 std::list<ResultPtr>::const_iterator anIt = anAuxiliaryResults.begin(),
308 aLast = anAuxiliaryResults.end();
309 for (; anIt != aLast; anIt++) {
310 ResultPtr aResult = *anIt;
312 GeomShapePtr aGeomShape = aResult->shape();
313 if (aGeomShape.get()) {
314 const TopoDS_Shape& aShape = aGeomShape->impl<TopoDS_Shape>();
315 if (!aShape.IsNull())
316 aBuilder.Add(theAuxiliaryCompound, aShape);