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 const std::shared_ptr<GeomAPI_Ax3>& thePlane)
113 : AIS_LengthDimension(MyDefStart, MyDefEnd, MyDefPln),
114 myConstraint(theConstraint),
115 mySketcherPlane(thePlane),
116 myFirstPoint(MyDefStart),
117 mySecondPoint(MyDefEnd),
120 myValue(0., false, "")
122 SetDimensionAspect(createDimensionAspect());
123 myStyleListener = new SketcherPrs_DimensionStyleListener();
125 #ifdef OCCT_28850_FIXED
126 if (theConstraint->getKind() == SketchPlugin_ConstraintDistanceHorizontal::ID())
127 SetDirection(mySketcherPlane->dirX()->impl<gp_Dir>(), true);
128 else if (theConstraint->getKind() == SketchPlugin_ConstraintDistanceVertical::ID())
129 SetDirection(mySketcherPlane->dirY()->impl<gp_Dir>(), true);
133 SketcherPrs_LengthDimension::~SketcherPrs_LengthDimension()
135 delete myStyleListener;
138 bool SketcherPrs_LengthDimension::IsReadyToDisplay(ModelAPI_Feature* theConstraint,
139 const std::shared_ptr<GeomAPI_Ax3>& thePlane)
142 return readyToDisplay(theConstraint, thePlane, aPnt1, aPnt2);
145 void SketcherPrs_LengthDimension::Compute(
146 const Handle(PrsMgr_PresentationManager3d)& thePresentationManager,
147 const Handle(Prs3d_Presentation)& thePresentation,
148 const Standard_Integer theMode)
151 bool aReadyToDisplay = readyToDisplay(myConstraint, mySketcherPlane, aPnt1, aPnt2);
152 if (aReadyToDisplay) {
153 myFirstPoint = aPnt1;
154 mySecondPoint = aPnt2;
156 myDistance = SketcherPrs_Tools::getFlyoutDistance(myConstraint);
157 myPlane = gp_Pln(mySketcherPlane->impl<gp_Ax3>());
159 DataPtr aData = myConstraint->data();
160 AttributeDoublePtr anAttributeValue;
161 if (myConstraint->getKind() == SketchPlugin_ConstraintDistanceHorizontal::ID() ||
162 myConstraint->getKind() == SketchPlugin_ConstraintDistanceVertical::ID())
163 anAttributeValue = aData->real(SketchPlugin_ConstraintDistanceAlongDir::DISTANCE_VALUE_ID());
165 anAttributeValue = aData->real(SketchPlugin_Constraint::VALUE());
166 myValue.init(anAttributeValue);
169 // compute flyout distance
170 SetFlyout(myDistance);
171 SetMeasuredGeometry(myFirstPoint, mySecondPoint, myPlane);
173 // Update variable aspect parameters (depending on viewer scale)
174 double aTextSize = 0.0;
175 GetValueString(aTextSize);
177 std::string aLocationAttribute;
178 std::string aConstraintKind = myConstraint->getKind();
179 if (aConstraintKind == SketchPlugin_ConstraintLength::ID())
180 aLocationAttribute = SketchPlugin_ConstraintLength::LOCATION_TYPE_ID();
181 else if (aConstraintKind == SketchPlugin_ConstraintDistance::ID())
182 aLocationAttribute = SketchPlugin_ConstraintDistance::LOCATION_TYPE_ID();
183 else if (aConstraintKind == SketchPlugin_ConstraintDistanceHorizontal::ID())
184 aLocationAttribute = SketchPlugin_ConstraintDistanceHorizontal::LOCATION_TYPE_ID();
185 else if (aConstraintKind == SketchPlugin_ConstraintDistanceVertical::ID())
186 aLocationAttribute = SketchPlugin_ConstraintDistanceVertical::LOCATION_TYPE_ID();
188 AttributeIntegerPtr aLocAttr = std::dynamic_pointer_cast<ModelAPI_AttributeInteger>
189 (myConstraint->data()->attribute(aLocationAttribute));
190 SketcherPrs_Tools::LocationType aLocationType = aLocAttr->isInitialized() ?
191 (SketcherPrs_Tools::LocationType)(aLocAttr->value()) : SketcherPrs_Tools::LOCATION_AUTOMATIC;
192 updateArrows(DimensionAspect(), GetValue(), aTextSize, aLocationType);
194 // Update text visualization: parameter value or parameter text
195 myStyleListener->updateDimensions(this, myValue);
197 AIS_LengthDimension::Compute(thePresentationManager, thePresentation, theMode);
199 if (!aReadyToDisplay)
200 SketcherPrs_Tools::sendEmptyPresentationError(myConstraint,
201 "An empty AIS presentation: SketcherPrs_LengthDimension");
204 bool SketcherPrs_LengthDimension::readyToDisplay(ModelAPI_Feature* theConstraint,
205 const std::shared_ptr<GeomAPI_Ax3>& thePlane,
206 gp_Pnt& thePnt1, gp_Pnt& thePnt2)
208 DataPtr aData = theConstraint->data();
209 if (theConstraint->getKind() == SketchPlugin_ConstraintLength::ID()) {
210 // The constraint is length
211 std::shared_ptr<ModelAPI_AttributeRefAttr> anAttr =
212 std::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>
213 (aData->attribute(SketchPlugin_Constraint::ENTITY_A()));
217 FeaturePtr aFeature = ModelAPI_Feature::feature(anAttr->object());
218 if (!aFeature || aFeature->getKind() != SketchPlugin_Line::ID())
221 // Get geometry of the object
222 DataPtr aLineData = aFeature->data();
223 std::shared_ptr<GeomDataAPI_Point2D> aStartPoint =
224 std::dynamic_pointer_cast<GeomDataAPI_Point2D>
225 (aLineData->attribute(SketchPlugin_Line::START_ID()));
226 std::shared_ptr<GeomDataAPI_Point2D> aEndPoint =
227 std::dynamic_pointer_cast<GeomDataAPI_Point2D>
228 (aLineData->attribute(SketchPlugin_Line::END_ID()));
229 thePnt1 = thePlane->to3D(aStartPoint->x(), aStartPoint->y())->impl<gp_Pnt>();
230 thePnt2 = thePlane->to3D(aEndPoint->x(), aEndPoint->y())->impl<gp_Pnt>();
233 } else if (theConstraint->getKind() == SketchPlugin_ConstraintDistance::ID() ||
234 theConstraint->getKind() == SketchPlugin_ConstraintDistanceHorizontal::ID() ||
235 theConstraint->getKind() == SketchPlugin_ConstraintDistanceVertical::ID()) {
236 // The constraint is distance
237 std::shared_ptr<GeomDataAPI_Point2D> aPoint_A = SketcherPrs_Tools::getFeaturePoint(
238 aData, SketchPlugin_Constraint::ENTITY_A(), thePlane);
239 std::shared_ptr<GeomDataAPI_Point2D> aPoint_B = SketcherPrs_Tools::getFeaturePoint(
240 aData, SketchPlugin_Constraint::ENTITY_B(), thePlane);
242 std::shared_ptr<GeomAPI_Pnt2d> aPnt_A;
243 std::shared_ptr<GeomAPI_Pnt2d> aPnt_B;
244 if (aPoint_A && aPoint_B) {
245 // Both objects are points
246 aPnt_A = aPoint_A->pnt();
247 aPnt_B = aPoint_B->pnt();
248 } else if (!aPoint_A && aPoint_B) {
249 // First object is line
250 FeaturePtr aLine = SketcherPrs_Tools::getFeatureLine(
251 aData, SketchPlugin_Constraint::ENTITY_A());
253 aPnt_B = aPoint_B->pnt();
254 aPnt_A = SketcherPrs_Tools::getProjectionPoint(aLine, aPnt_B);
256 } else if (aPoint_A && !aPoint_B) {
257 // Second object is line
258 FeaturePtr aLine = SketcherPrs_Tools::getFeatureLine(
259 aData, SketchPlugin_Constraint::ENTITY_B());
261 aPnt_A = aPoint_A->pnt();
262 aPnt_B = SketcherPrs_Tools::getProjectionPoint(aLine, aPnt_A);
265 if (!aPnt_A || !aPnt_B) // Objects not found
268 if (theConstraint->getKind() == SketchPlugin_ConstraintDistanceHorizontal::ID()) {
269 if (fabs(aPnt_A->x() - aPnt_B->x()) < Precision::Confusion())
272 else if (theConstraint->getKind() == SketchPlugin_ConstraintDistanceVertical::ID()) {
273 if (fabs(aPnt_A->y() - aPnt_B->y()) < Precision::Confusion())
277 // Get points from these object
278 std::shared_ptr<GeomAPI_Pnt> aPoint1 = thePlane->to3D(aPnt_A->x(), aPnt_A->y());
279 std::shared_ptr<GeomAPI_Pnt> aPoint2 = thePlane->to3D(aPnt_B->x(), aPnt_B->y());
280 thePnt1 = aPoint1->impl<gp_Pnt>();
281 thePnt2 = aPoint2->impl<gp_Pnt>();
289 void SketcherPrs_LengthDimension::ComputeSelection(const Handle(SelectMgr_Selection)& aSelection,
290 const Standard_Integer theMode)
292 // Map the application selection modes to standard ones
293 Standard_Integer aMode;
295 case 0: // we should use selection of all objects
298 case SketcherPrs_Tools::Sel_Dimension_All:
301 case SketcherPrs_Tools::Sel_Dimension_Line:
304 case SketcherPrs_Tools::Sel_Dimension_Text:
308 // there are own selection modes, so the others should be ignored
309 // otherwise, the text selection appears in the viewer
313 SetSelToleranceForText2d(SketcherPrs_Tools::getArrowSize()/5.);
314 AIS_LengthDimension::ComputeSelection(aSelection, aMode);