1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
3 // File: SketcherPrs_Angle.cpp
4 // Created: 20 August 2015
5 // Author: Vitaly SMETANNIKOV
7 #include "SketcherPrs_Angle.h"
8 #include "SketcherPrs_Tools.h"
10 #include <SketchPlugin_ConstraintAngle.h>
11 #include <SketchPlugin_Constraint.h>
13 #include <GeomAPI_Edge.h>
14 #include <GeomAPI_Lin.h>
15 #include <GeomDataAPI_Point2D.h>
17 #include <ModelAPI_AttributeRefAttr.h>
18 #include <ModelAPI_AttributeDouble.h>
21 #include <BRep_Tool.hxx>
23 #include <Events_Error.h>
25 #define PI 3.1415926535897932
27 IMPLEMENT_STANDARD_HANDLE(SketcherPrs_Angle, AIS_AngleDimension);
28 IMPLEMENT_STANDARD_RTTIEXT(SketcherPrs_Angle, AIS_AngleDimension);
30 SketcherPrs_Angle::SketcherPrs_Angle(ModelAPI_Feature* theConstraint,
31 const std::shared_ptr<GeomAPI_Ax3>& thePlane)
32 : AIS_AngleDimension(gp_Pnt(0,0,0), gp_Pnt(1,0,0), gp_Pnt(0,1,0)), myConstraint(theConstraint), myPlane(thePlane)
34 myAspect = new Prs3d_DimensionAspect();
35 myAspect->MakeArrows3d(false);
36 myAspect->MakeText3d(false);
37 myAspect->MakeTextShaded(false);
38 myAspect->MakeUnitsDisplayed(false);
39 myAspect->TextAspect()->SetHeight(SketcherPrs_Tools::getDefaultTextHeight());
40 myAspect->ArrowAspect()->SetLength(SketcherPrs_Tools::getArrowSize());
42 SetDimensionAspect(myAspect);
43 SetSelToleranceForText2d(SketcherPrs_Tools::getDefaultTextHeight());
46 bool SketcherPrs_Angle::IsReadyToDisplay(ModelAPI_Feature* theConstraint,
47 const std::shared_ptr<GeomAPI_Ax3>&/* thePlane*/)
49 bool aReadyToDisplay = false;
51 DataPtr aData = theConstraint->data();
54 std::shared_ptr<GeomDataAPI_Point2D> aFlyoutAttr =
55 std::dynamic_pointer_cast<GeomDataAPI_Point2D>
56 (aData->attribute(SketchPlugin_Constraint::FLYOUT_VALUE_PNT()));
57 if (!aFlyoutAttr->isInitialized())
58 return aReadyToDisplay; // can not create a good presentation
60 AttributeRefAttrPtr anAttr1 = aData->refattr(SketchPlugin_Constraint::ENTITY_A());
61 if (!anAttr1->isInitialized())
62 return aReadyToDisplay;
64 AttributeRefAttrPtr anAttr2 = aData->refattr(SketchPlugin_Constraint::ENTITY_B());
65 if (!anAttr2->isInitialized())
66 return aReadyToDisplay;
69 ObjectPtr aObj1 = anAttr1->object();
70 ObjectPtr aObj2 = anAttr2->object();
72 std::shared_ptr<GeomAPI_Shape> aShape1 = SketcherPrs_Tools::getShape(aObj1);
73 std::shared_ptr<GeomAPI_Shape> aShape2 = SketcherPrs_Tools::getShape(aObj2);
75 aReadyToDisplay = aShape1.get() && aShape2.get();
76 return aReadyToDisplay;
79 void SketcherPrs_Angle::Compute(const Handle(PrsMgr_PresentationManager3d)& thePresentationManager,
80 const Handle(Prs3d_Presentation)& thePresentation,
81 const Standard_Integer theMode)
83 DataPtr aData = myConstraint->data();
85 if (!IsReadyToDisplay(myConstraint, myPlane)) {
86 Events_Error::throwException("An empty AIS presentation: SketcherPrs_Angle");
87 return; // can not create a good presentation
91 std::shared_ptr<GeomDataAPI_Point2D> aFlyoutAttr =
92 std::dynamic_pointer_cast<GeomDataAPI_Point2D>
93 (aData->attribute(SketchPlugin_Constraint::FLYOUT_VALUE_PNT()));
94 std::shared_ptr<GeomAPI_Pnt> aFlyoutPnt = myPlane->to3D(aFlyoutAttr->x(), aFlyoutAttr->y());
96 AttributeRefAttrPtr anAttr1 = aData->refattr(SketchPlugin_Constraint::ENTITY_A());
97 AttributeRefAttrPtr anAttr2 = aData->refattr(SketchPlugin_Constraint::ENTITY_B());
100 ObjectPtr aObj1 = anAttr1->object();
101 ObjectPtr aObj2 = anAttr2->object();
103 std::shared_ptr<GeomAPI_Shape> aShape1 = SketcherPrs_Tools::getShape(aObj1);
104 std::shared_ptr<GeomAPI_Shape> aShape2 = SketcherPrs_Tools::getShape(aObj2);
106 TopoDS_Shape aTEdge1 = aShape1->impl<TopoDS_Shape>();
107 TopoDS_Shape aTEdge2 = aShape2->impl<TopoDS_Shape>();
109 TopoDS_Edge aEdge1 = TopoDS::Edge(aTEdge1);
110 TopoDS_Edge aEdge2 = TopoDS::Edge(aTEdge2);
111 SetMeasuredGeometry(aEdge1, aEdge2);
113 const gp_Pnt& aCenter = CenterPoint();
114 const gp_Pnt& aFirst = FirstPoint();
115 const gp_Pnt& aSecond = SecondPoint();
117 gp_Dir aBisector((aFirst.XYZ() + aSecond.XYZ()) * 0.5 - aCenter.XYZ());
119 gp_Pnt aFlyPnt(aFlyoutPnt->x(), aFlyoutPnt->y(), aFlyoutPnt->z());
120 gp_XYZ aFlyDir = aFlyPnt.XYZ() - aCenter.XYZ();
121 double aDist = aFlyDir.Dot(aBisector.XYZ());
124 // Angle value is in degrees
125 AttributeDoublePtr aVal = aData->real(SketchPlugin_Constraint::VALUE());
126 SetCustomValue(aVal->value() * PI / 180.0);
128 myAspect->SetExtensionSize(myAspect->ArrowAspect()->Length());
129 myAspect->SetArrowTailSize(myAspect->ArrowAspect()->Length());
131 AttributeDoublePtr aValue = myConstraint->data()->real(SketchPlugin_Constraint::VALUE());
132 SketcherPrs_Tools::setDisplaySpecialSymbol(this, aValue->usedParameters().size() > 0);
134 AIS_AngleDimension::Compute(thePresentationManager, thePresentation, theMode);
137 void SketcherPrs_Angle::ComputeSelection(const Handle(SelectMgr_Selection)& aSelection,
138 const Standard_Integer theMode)
140 Standard_Integer aMode;
142 case 0: // we should use selection of all objects
145 case SketcherPrs_Tools::Sel_Dimension_All:
148 case SketcherPrs_Tools::Sel_Dimension_Line:
151 case SketcherPrs_Tools::Sel_Dimension_Text:
155 // there are own selection modes, so the others should be ignored
156 // otherwise, the text selection appears in the viewer
160 AIS_AngleDimension::ComputeSelection(aSelection, aMode);