]> SALOME platform Git repositories - modules/shaper.git/blob - src/SketchPlugin/SketchPlugin_ConstraintLength.cpp
Salome HOME
Add SketchPlugin document template.
[modules/shaper.git] / src / SketchPlugin / SketchPlugin_ConstraintLength.cpp
1 // Copyright (C) 2014-2017  CEA/DEN, EDF R&D
2 //
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.
7 //
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.
12 //
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
16 //
17 // See http://www.salome-platform.org/ or
18 // email : webmaster.salome@opencascade.com<mailto:webmaster.salome@opencascade.com>
19 //
20
21 #include "SketchPlugin_ConstraintLength.h"
22 #include <SketchPlugin_Line.h>
23
24 #include <GeomDataAPI_Point2D.h>
25
26 #include <SketcherPrs_Factory.h>
27 #include <SketcherPrs_Tools.h>
28
29 #include <ModelAPI_AttributeDouble.h>
30 #include <ModelAPI_AttributeInteger.h>
31 #include <ModelAPI_Data.h>
32 #include <ModelAPI_Result.h>
33 #include <ModelAPI_Session.h>
34 #include <ModelAPI_Validator.h>
35
36 #include <GeomDataAPI_Point2D.h>
37
38 #include <GeomAPI_Dir2d.h>
39 #include <GeomAPI_Lin2d.h>
40 #include <GeomAPI_Pnt2d.h>
41 #include <GeomAPI_XY.h>
42
43 #include <Config_PropManager.h>
44
45 #include <math.h>
46
47 const double tolerance = 1e-7;
48
49 SketchPlugin_ConstraintLength::SketchPlugin_ConstraintLength()
50 {
51   myFlyoutUpdate = false;
52 }
53
54 void SketchPlugin_ConstraintLength::initAttributes()
55 {
56   data()->addAttribute(SketchPlugin_Constraint::VALUE(), ModelAPI_AttributeDouble::typeId());
57   data()->addAttribute(SketchPlugin_Constraint::FLYOUT_VALUE_PNT(), GeomDataAPI_Point2D::typeId());
58   data()->addAttribute(SketchPlugin_Constraint::ENTITY_A(), ModelAPI_AttributeRefAttr::typeId());
59
60   data()->addAttribute(SketchPlugin_ConstraintLength::LOCATION_TYPE_ID(),
61                        ModelAPI_AttributeInteger::typeId());
62   ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), LOCATION_TYPE_ID());
63 }
64
65 void SketchPlugin_ConstraintLength::colorConfigInfo(std::string& theSection, std::string& theName,
66                                                     std::string& theDefault)
67 {
68   theSection = "Visualization";
69   theName = "sketch_dimension_color";
70   theDefault = SKETCH_DIMENSION_COLOR;
71 }
72
73 void SketchPlugin_ConstraintLength::execute()
74 {
75   std::shared_ptr<ModelAPI_AttributeRefAttr> aRef = std::dynamic_pointer_cast<
76       ModelAPI_AttributeRefAttr>(data()->attribute(SketchPlugin_Constraint::ENTITY_A()));
77   FeaturePtr aFeature = ModelAPI_Feature::feature(aRef->object());
78   if (aFeature) {
79     // set length value
80     std::shared_ptr<GeomDataAPI_Point2D> aPoint1 = std::dynamic_pointer_cast<
81         GeomDataAPI_Point2D>(aFeature->data()->attribute(SketchPlugin_Line::START_ID()));
82     std::shared_ptr<GeomDataAPI_Point2D> aPoint2 = std::dynamic_pointer_cast<
83         GeomDataAPI_Point2D>(aFeature->data()->attribute(SketchPlugin_Line::END_ID()));
84
85     double aLenght = aPoint1->pnt()->distance(aPoint2->pnt());
86
87     //std::shared_ptr<ModelAPI_AttributeDouble> aValueAttr = std::dynamic_pointer_cast<
88     //    ModelAPI_AttributeDouble>(data()->attribute(SketchPlugin_Constraint::VALUE()));
89     //if(!aValueAttr->isInitialized()) {
90     //  aValueAttr->setValue(aLenght);
91     //}
92
93     // the value should to be computed here, not in the getAISObject
94     // in order to change the model value
95     // inside the object transaction. This is important for creating a constraint by preselection.
96     // The display of the presentation in this case happens after the transaction commit
97     AttributePtr aFlyOutAttribute =
98       data()->attribute(SketchPlugin_Constraint::FLYOUT_VALUE_PNT());
99     if (!aFlyOutAttribute->isInitialized()) {
100       compute(SketchPlugin_Constraint::FLYOUT_VALUE_PNT());
101     }
102   }
103 }
104
105 bool SketchPlugin_ConstraintLength::compute(const std::string& theAttributeId)
106 {
107   if (theAttributeId != SketchPlugin_Constraint::FLYOUT_VALUE_PNT())
108     return false;
109
110   std::shared_ptr<GeomAPI_Pnt> aPoint1, aPoint2;
111   std::shared_ptr<GeomDataAPI_Point2D> aStartPoint, anEndPoint;
112   if (!getPoints(aPoint1, aPoint2, aStartPoint, anEndPoint))
113     return false;
114
115   std::shared_ptr<GeomDataAPI_Point2D> aFlyOutAttr = std::dynamic_pointer_cast<
116       GeomDataAPI_Point2D>(data()->attribute(theAttributeId));
117
118   std::shared_ptr<GeomAPI_Lin2d> aLine =
119     std::shared_ptr<GeomAPI_Lin2d>(new GeomAPI_Lin2d(aStartPoint->pnt(), anEndPoint->pnt()));
120   if (!aFlyOutAttr->isInitialized() ||
121       (fabs(aFlyOutAttr->x()) < tolerance && fabs(aFlyOutAttr->y()) < tolerance)) {
122     double aDist = aPoint1->distance(aPoint2)/5.;
123     std::shared_ptr<GeomAPI_Pnt2d> aFPnt = aLine->shiftedLocation(aDist);
124     aFlyOutAttr->setValue(aFPnt);
125   }
126
127   return true;
128 }
129
130 bool SketchPlugin_ConstraintLength::computeLenghtValue(double& theValue)
131 {
132   bool aResult = false;
133   std::shared_ptr<GeomAPI_Pnt> aPoint1, aPoint2;
134   std::shared_ptr<GeomDataAPI_Point2D> aStartPoint, anEndPoint;
135   if (getPoints(aPoint1, aPoint2, aStartPoint, anEndPoint)) {
136     theValue = aPoint1->distance(aPoint2);
137     aResult = true;
138   }
139   return aResult;
140 }
141
142 bool SketchPlugin_ConstraintLength::getPoints(
143   std::shared_ptr<GeomAPI_Pnt>& thePoint1, std::shared_ptr<GeomAPI_Pnt>& thePoint2,
144   std::shared_ptr<GeomDataAPI_Point2D>& theStartPoint,
145   std::shared_ptr<GeomDataAPI_Point2D>& theEndPoint)
146 {
147   if (!sketch())
148     return false;
149   std::shared_ptr<ModelAPI_AttributeRefAttr> anAttr = std::dynamic_pointer_cast<
150       ModelAPI_AttributeRefAttr>(data()->attribute(SketchPlugin_Constraint::ENTITY_A()));
151   if (!anAttr)
152     return false;
153   FeaturePtr aFeature = ModelAPI_Feature::feature(anAttr->object());
154   if (!aFeature || aFeature->getKind() != SketchPlugin_Line::ID())
155     return false;
156   DataPtr aData = aFeature->data();
157   theStartPoint = std::dynamic_pointer_cast<
158       GeomDataAPI_Point2D>(aData->attribute(SketchPlugin_Line::START_ID()));
159   theEndPoint = std::dynamic_pointer_cast<
160       GeomDataAPI_Point2D>(aData->attribute(SketchPlugin_Line::END_ID()));
161   thePoint1 = sketch()->to3D(theStartPoint->x(), theStartPoint->y());
162   thePoint2 = sketch()->to3D(theEndPoint->x(), theEndPoint->y());
163   return true;
164 }
165
166 AISObjectPtr SketchPlugin_ConstraintLength::getAISObject(AISObjectPtr thePrevious)
167 {
168   if (!sketch())
169     return thePrevious;
170
171   AISObjectPtr anAIS = SketcherPrs_Factory::lengthDimensionConstraint(this,
172     sketch()->coordinatePlane(), thePrevious);
173   return anAIS;
174 }
175
176 void SketchPlugin_ConstraintLength::attributeChanged(const std::string& theID) {
177   if (theID == SketchPlugin_Constraint::ENTITY_A())
178   {
179     std::shared_ptr<ModelAPI_AttributeDouble> aValueAttr = std::dynamic_pointer_cast<
180       ModelAPI_AttributeDouble>(attribute(SketchPlugin_Constraint::VALUE()));
181     if (!aValueAttr->isInitialized()) {
182       // only if it is not initialized, try to compute the current value
183       double aLength;
184       if (computeLenghtValue(aLength))
185         aValueAttr->setValue(aLength);
186     }
187   } else if (theID == SketchPlugin_Constraint::FLYOUT_VALUE_PNT() && !myFlyoutUpdate) {
188     // Recalculate flyout point in local coordinates of the line:
189     // the X coordinate is a length of projection of the flyout point on the line
190     // the Y coordinate is a distance from the point to the line
191     std::shared_ptr<GeomDataAPI_Point2D> aFlyoutAttr =
192         std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
193         attribute(SketchPlugin_Constraint::FLYOUT_VALUE_PNT()));
194     AttributeRefAttrPtr aLineAttr = std::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>(
195         attribute(SketchPlugin_Constraint::ENTITY_A()));
196     if (!aLineAttr || !aLineAttr->isObject())
197       return;
198     FeaturePtr aLine = ModelAPI_Feature::feature(aLineAttr->object());
199     if (!aLine || aLine->getKind() != SketchPlugin_Line::ID())
200       return;
201
202     std::shared_ptr<GeomAPI_Pnt2d> aFlyoutPnt = aFlyoutAttr->pnt();
203     std::shared_ptr<GeomAPI_XY> aStartPnt = std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
204         aLine->attribute(SketchPlugin_Line::START_ID()))->pnt()->xy();
205     std::shared_ptr<GeomAPI_XY> aEndPnt = std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
206         aLine->attribute(SketchPlugin_Line::END_ID()))->pnt()->xy();
207
208     myFlyoutUpdate = true;
209     std::shared_ptr<GeomAPI_Dir2d> aLineDir(new GeomAPI_Dir2d(aEndPnt->decreased(aStartPnt)));
210     std::shared_ptr<GeomAPI_XY> aFlyoutDir = aFlyoutPnt->xy()->decreased(aStartPnt);
211     double X = aFlyoutDir->dot(aLineDir->xy());
212     double Y = -aFlyoutDir->cross(aLineDir->xy());
213     aFlyoutAttr->setValue(X, Y);
214     myFlyoutUpdate = false;
215   }
216 }