1 // Copyright (C) 2014-2019 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 "SketcherPrs_LengthDimension.h"
21 #include "SketcherPrs_Tools.h"
22 #include "SketcherPrs_DimensionStyleListener.h"
24 #include <SketchPlugin_Constraint.h>
25 #include <SketchPlugin_ConstraintLength.h>
26 #include <SketchPlugin_ConstraintDistance.h>
27 #include <SketchPlugin_ConstraintDistanceHorizontal.h>
28 #include <SketchPlugin_ConstraintDistanceVertical.h>
29 #include <SketchPlugin_Line.h>
30 #include <SketchPlugin_Point.h>
31 #include <SketchPlugin_Circle.h>
33 #include <GeomDataAPI_Point2D.h>
34 #include <GeomAPI_Pnt.h>
35 #include <GeomAPI_XYZ.h>
36 #include <GeomAPI_Pnt2d.h>
37 #include <GeomAPI_Lin2d.h>
39 #include <ModelAPI_Data.h>
40 #include <ModelAPI_AttributeDouble.h>
41 #include <ModelAPI_AttributeInteger.h>
43 #include <AIS_DisplaySpecialSymbol.hxx>
45 #define OCCT_28850_FIXED
47 /// Creates an aspect to be shown in length/radius dimension presentations
48 /// \return an instance of aspect
49 Handle(Prs3d_DimensionAspect) createDimensionAspect()
51 Handle(Prs3d_DimensionAspect) anAspect = new Prs3d_DimensionAspect();
52 anAspect->MakeArrows3d(false);
53 anAspect->MakeText3d(false);
54 anAspect->MakeTextShaded(false);
55 anAspect->MakeUnitsDisplayed(false);
56 anAspect->TextAspect()->SetHeight(SketcherPrs_Tools::getConfigTextHeight());
57 anAspect->ArrowAspect()->SetLength(SketcherPrs_Tools::getArrowSize());
62 /// Update variable aspect parameters (depending on viewer scale)
63 /// \param theDimAspect an aspect to be changed
64 /// \param theDimValue an arrow value
65 /// \param theTextSize an arrow value
66 void updateArrows(Handle(Prs3d_DimensionAspect) theDimAspect,
67 double theDimValue, double theTextSize, SketcherPrs_Tools::LocationType theLocationType)
69 if (theLocationType == SketcherPrs_Tools::LOCATION_AUTOMATIC) {
70 double anArrowLength = theDimAspect->ArrowAspect()->Length();
71 // This is not realy correct way to get viewer scale.
72 double aViewerScale = (double) SketcherPrs_Tools::getConfigArrowSize() / anArrowLength;
74 if(theTextSize > ((theDimValue - 3 * SketcherPrs_Tools::getArrowSize()) * aViewerScale)) {
75 theDimAspect->SetTextHorizontalPosition(Prs3d_DTHP_Left);
76 theDimAspect->SetArrowOrientation(Prs3d_DAO_External);
77 theDimAspect->SetExtensionSize(
78 (theTextSize / aViewerScale + SketcherPrs_Tools::getArrowSize()) / 2.0);
80 theDimAspect->SetTextHorizontalPosition(Prs3d_DTHP_Center);
81 theDimAspect->SetArrowOrientation(Prs3d_DAO_Internal);
84 else if (theLocationType == SketcherPrs_Tools::LOCATION_RIGHT ||
85 theLocationType == SketcherPrs_Tools::LOCATION_LEFT) {
86 theDimAspect->SetTextHorizontalPosition(
87 theLocationType == SketcherPrs_Tools::LOCATION_LEFT ? Prs3d_DTHP_Left : Prs3d_DTHP_Right);
88 theDimAspect->SetArrowOrientation(Prs3d_DAO_External);
90 double anArrowLength = theDimAspect->ArrowAspect()->Length();
91 // This is not realy correct way to get viewer scale.
92 double aViewerScale = (double) SketcherPrs_Tools::getConfigArrowSize() / anArrowLength;
93 theDimAspect->SetExtensionSize(
94 (theTextSize / aViewerScale + SketcherPrs_Tools::getArrowSize()) / 2.0);
96 theDimAspect->TextAspect()->SetFont(Config_PropManager::string("Visualization",
97 "dimension_font").c_str());
99 theDimAspect->SetArrowTailSize(theDimAspect->ArrowAspect()->Length());
100 // The value of vertical aligment is sometimes changed
101 theDimAspect->TextAspect()->SetVerticalJustification(Graphic3d_VTA_CENTER);
105 static const gp_Pnt MyDefStart(0,0,0);
106 static const gp_Pnt MyDefEnd(1,0,0);
107 static const gp_Pln MyDefPln(gp_Pnt(0,0,0), gp_Dir(0,0,1));
109 IMPLEMENT_STANDARD_RTTIEXT(SketcherPrs_LengthDimension, AIS_LengthDimension);
111 SketcherPrs_LengthDimension::SketcherPrs_LengthDimension(ModelAPI_Feature* theConstraint,
112 SketchPlugin_Sketch* theSketcher)
113 : AIS_LengthDimension(MyDefStart, MyDefEnd, MyDefPln),
114 myConstraint(theConstraint),
115 mySketcher(theSketcher),
116 myFirstPoint(MyDefStart),
117 mySecondPoint(MyDefEnd),
120 myValue(0., false, "")
122 SetDimensionAspect(createDimensionAspect());
123 myStyleListener = new SketcherPrs_DimensionStyleListener();
125 setDirection(theConstraint, plane());
128 SketcherPrs_LengthDimension::~SketcherPrs_LengthDimension()
130 delete myStyleListener;
133 bool SketcherPrs_LengthDimension::IsReadyToDisplay(ModelAPI_Feature* theConstraint,
134 const std::shared_ptr<GeomAPI_Ax3>& thePlane)
137 return readyToDisplay(theConstraint, thePlane, aPnt1, aPnt2);
140 void SketcherPrs_LengthDimension::Compute(
141 const Handle(PrsMgr_PresentationManager3d)& thePresentationManager,
142 const Handle(Prs3d_Presentation)& thePresentation,
143 const Standard_Integer theMode)
148 bool aReadyToDisplay = readyToDisplay(myConstraint, plane(), aPnt1, aPnt2);
149 if (aReadyToDisplay) {
150 myFirstPoint = aPnt1;
151 mySecondPoint = aPnt2;
153 myDistance = SketcherPrs_Tools::getFlyoutDistance(myConstraint);
154 myPlane = gp_Pln(plane()->impl<gp_Ax3>());
156 DataPtr aData = myConstraint->data();
157 AttributeDoublePtr anAttributeValue;
158 if (myConstraint->getKind() == SketchPlugin_ConstraintDistanceHorizontal::ID() ||
159 myConstraint->getKind() == SketchPlugin_ConstraintDistanceVertical::ID())
160 anAttributeValue = aData->real(SketchPlugin_ConstraintDistanceAlongDir::DISTANCE_VALUE_ID());
162 anAttributeValue = aData->real(SketchPlugin_Constraint::VALUE());
163 myValue.init(anAttributeValue);
165 setDirection(myConstraint, plane());
168 // compute flyout distance
169 SetFlyout(myDistance);
170 SetMeasuredGeometry(myFirstPoint, mySecondPoint, myPlane);
172 // Update variable aspect parameters (depending on viewer scale)
173 double aTextSize = 0.0;
174 GetValueString(aTextSize);
176 std::string aLocationAttribute;
177 std::string aConstraintKind = myConstraint->getKind();
178 if (aConstraintKind == SketchPlugin_ConstraintLength::ID())
179 aLocationAttribute = SketchPlugin_ConstraintLength::LOCATION_TYPE_ID();
180 else if (aConstraintKind == SketchPlugin_ConstraintDistance::ID())
181 aLocationAttribute = SketchPlugin_ConstraintDistance::LOCATION_TYPE_ID();
182 else if (aConstraintKind == SketchPlugin_ConstraintDistanceHorizontal::ID())
183 aLocationAttribute = SketchPlugin_ConstraintDistanceHorizontal::LOCATION_TYPE_ID();
184 else if (aConstraintKind == SketchPlugin_ConstraintDistanceVertical::ID())
185 aLocationAttribute = SketchPlugin_ConstraintDistanceVertical::LOCATION_TYPE_ID();
187 AttributeIntegerPtr aLocAttr = std::dynamic_pointer_cast<ModelAPI_AttributeInteger>
188 (myConstraint->data()->attribute(aLocationAttribute));
189 SketcherPrs_Tools::LocationType aLocationType = aLocAttr->isInitialized() ?
190 (SketcherPrs_Tools::LocationType)(aLocAttr->value()) : SketcherPrs_Tools::LOCATION_AUTOMATIC;
191 updateArrows(DimensionAspect(), GetValue(), aTextSize, aLocationType);
193 // Update text visualization: parameter value or parameter text
194 myStyleListener->updateDimensions(this, myValue);
196 AIS_LengthDimension::Compute(thePresentationManager, thePresentation, theMode);
198 if (!aReadyToDisplay)
199 SketcherPrs_Tools::sendEmptyPresentationError(myConstraint,
200 "An empty AIS presentation: SketcherPrs_LengthDimension");
203 bool SketcherPrs_LengthDimension::readyToDisplay(ModelAPI_Feature* theConstraint,
204 const std::shared_ptr<GeomAPI_Ax3>& thePlane,
205 gp_Pnt& thePnt1, gp_Pnt& thePnt2)
210 DataPtr aData = theConstraint->data();
211 if (theConstraint->getKind() == SketchPlugin_ConstraintLength::ID()) {
212 // The constraint is length
213 std::shared_ptr<ModelAPI_AttributeRefAttr> anAttr =
214 std::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>
215 (aData->attribute(SketchPlugin_Constraint::ENTITY_A()));
219 FeaturePtr aFeature = ModelAPI_Feature::feature(anAttr->object());
220 if (!aFeature || aFeature->getKind() != SketchPlugin_Line::ID())
223 // Get geometry of the object
224 DataPtr aLineData = aFeature->data();
225 std::shared_ptr<GeomDataAPI_Point2D> aStartPoint =
226 std::dynamic_pointer_cast<GeomDataAPI_Point2D>
227 (aLineData->attribute(SketchPlugin_Line::START_ID()));
228 std::shared_ptr<GeomDataAPI_Point2D> aEndPoint =
229 std::dynamic_pointer_cast<GeomDataAPI_Point2D>
230 (aLineData->attribute(SketchPlugin_Line::END_ID()));
231 thePnt1 = thePlane->to3D(aStartPoint->x(), aStartPoint->y())->impl<gp_Pnt>();
232 thePnt2 = thePlane->to3D(aEndPoint->x(), aEndPoint->y())->impl<gp_Pnt>();
235 } else if (theConstraint->getKind() == SketchPlugin_ConstraintDistance::ID() ||
236 theConstraint->getKind() == SketchPlugin_ConstraintDistanceHorizontal::ID() ||
237 theConstraint->getKind() == SketchPlugin_ConstraintDistanceVertical::ID()) {
238 // The constraint is distance
239 std::shared_ptr<GeomDataAPI_Point2D> aPoint_A = SketcherPrs_Tools::getFeaturePoint(
240 aData, SketchPlugin_Constraint::ENTITY_A(), thePlane);
241 std::shared_ptr<GeomDataAPI_Point2D> aPoint_B = SketcherPrs_Tools::getFeaturePoint(
242 aData, SketchPlugin_Constraint::ENTITY_B(), thePlane);
244 std::shared_ptr<GeomAPI_Pnt2d> aPnt_A;
245 std::shared_ptr<GeomAPI_Pnt2d> aPnt_B;
246 if (aPoint_A && aPoint_B) {
247 // Both objects are points
248 aPnt_A = aPoint_A->pnt();
249 aPnt_B = aPoint_B->pnt();
250 } else if (!aPoint_A && aPoint_B) {
251 // First object is line
252 FeaturePtr aLine = SketcherPrs_Tools::getFeatureLine(
253 aData, SketchPlugin_Constraint::ENTITY_A());
255 aPnt_B = aPoint_B->pnt();
256 aPnt_A = SketcherPrs_Tools::getProjectionPoint(aLine, aPnt_B);
258 } else if (aPoint_A && !aPoint_B) {
259 // Second object is line
260 FeaturePtr aLine = SketcherPrs_Tools::getFeatureLine(
261 aData, SketchPlugin_Constraint::ENTITY_B());
263 aPnt_A = aPoint_A->pnt();
264 aPnt_B = SketcherPrs_Tools::getProjectionPoint(aLine, aPnt_A);
267 if (!aPnt_A || !aPnt_B) // Objects not found
270 if (theConstraint->getKind() == SketchPlugin_ConstraintDistanceHorizontal::ID()) {
271 if (fabs(aPnt_A->x() - aPnt_B->x()) < Precision::Confusion())
274 else if (theConstraint->getKind() == SketchPlugin_ConstraintDistanceVertical::ID()) {
275 if (fabs(aPnt_A->y() - aPnt_B->y()) < Precision::Confusion())
279 // Get points from these object
280 std::shared_ptr<GeomAPI_Pnt> aPoint1 = thePlane->to3D(aPnt_A->x(), aPnt_A->y());
281 std::shared_ptr<GeomAPI_Pnt> aPoint2 = thePlane->to3D(aPnt_B->x(), aPnt_B->y());
282 thePnt1 = aPoint1->impl<gp_Pnt>();
283 thePnt2 = aPoint2->impl<gp_Pnt>();
291 void SketcherPrs_LengthDimension::ComputeSelection(const Handle(SelectMgr_Selection)& aSelection,
292 const Standard_Integer theMode)
294 // Map the application selection modes to standard ones
295 Standard_Integer aMode;
297 case 0: // we should use selection of all objects
300 case SketcherPrs_Tools::Sel_Dimension_All:
303 case SketcherPrs_Tools::Sel_Dimension_Line:
306 case SketcherPrs_Tools::Sel_Dimension_Text:
310 // there are own selection modes, so the others should be ignored
311 // otherwise, the text selection appears in the viewer
315 SetSelToleranceForText2d(SketcherPrs_Tools::getArrowSize()/5.);
316 AIS_LengthDimension::ComputeSelection(aSelection, aMode);
319 void SketcherPrs_LengthDimension::setDirection(ModelAPI_Feature* theConstraint,
320 const std::shared_ptr<GeomAPI_Ax3>& thePlane)
322 #ifdef OCCT_28850_FIXED
323 if (theConstraint->getKind() == SketchPlugin_ConstraintDistanceHorizontal::ID())
324 SetDirection(thePlane->dirX()->impl<gp_Dir>(), true);
325 else if (theConstraint->getKind() == SketchPlugin_ConstraintDistanceVertical::ID())
326 SetDirection(thePlane->dirY()->impl<gp_Dir>(), true);