Salome HOME
Values of constraint (length, distance, radius) now initialized on reference definition
[modules/shaper.git] / src / SketchPlugin / SketchPlugin_ConstraintDistance.cpp
1 // File:    SketchPlugin_ConstraintDistance.cpp
2 // Created: 23 May 2014
3 // Author:  Artem ZHIDKOV
4
5 #include "SketchPlugin_ConstraintDistance.h"
6 #include <SketchPlugin_Point.h>
7 #include <SketchPlugin_Circle.h>
8 #include <SketchPlugin_Line.h>
9
10 #include <GeomAPI_Lin2d.h>
11 #include <GeomAPI_Pnt2d.h>
12 #include <GeomDataAPI_Point2D.h>
13
14 #include <ModelAPI_AttributeDouble.h>
15 #include <ModelAPI_Data.h>
16
17 #include <Config_PropManager.h>
18
19 SketchPlugin_ConstraintDistance::SketchPlugin_ConstraintDistance()
20 {
21 }
22
23 //*************************************************************************************
24 void SketchPlugin_ConstraintDistance::initAttributes()
25 {
26   data()->addAttribute(SketchPlugin_Constraint::VALUE(), ModelAPI_AttributeDouble::type());
27   data()->addAttribute(SketchPlugin_Constraint::FLYOUT_VALUE_PNT(), GeomDataAPI_Point2D::type());
28   data()->addAttribute(SketchPlugin_Constraint::ENTITY_A(), ModelAPI_AttributeRefAttr::type());
29   data()->addAttribute(SketchPlugin_Constraint::ENTITY_B(), ModelAPI_AttributeRefAttr::type());
30 }
31
32 //*************************************************************************************
33 void SketchPlugin_ConstraintDistance::execute()
34 {
35   std::shared_ptr<ModelAPI_Data> aData = data();
36   AttributeDoublePtr anAttrValue = std::dynamic_pointer_cast<ModelAPI_AttributeDouble>(
37       aData->attribute(SketchPlugin_Constraint::VALUE()));
38
39   if(anAttrValue->isInitialized())
40     return;
41
42   double aDistance = calculateCurrentDistance();
43   if(aDistance >= 0) {
44     anAttrValue->setValue(aDistance);
45   }
46 }
47
48 //*************************************************************************************
49 AISObjectPtr SketchPlugin_ConstraintDistance::getAISObject(AISObjectPtr thePrevious)
50 {
51   if (!sketch())
52     return thePrevious;
53
54   std::shared_ptr<GeomAPI_Pln> aPlane = sketch()->plane();
55
56   DataPtr aData = data();
57   std::shared_ptr<GeomDataAPI_Point2D> aPoint_A = getFeaturePoint(
58       aData, SketchPlugin_Constraint::ENTITY_A());
59   std::shared_ptr<GeomDataAPI_Point2D> aPoint_B = getFeaturePoint(
60       aData, SketchPlugin_Constraint::ENTITY_B());
61
62   std::shared_ptr<GeomAPI_Pnt2d> aPnt_A;
63   std::shared_ptr<GeomAPI_Pnt2d> aPnt_B;
64
65   if (aPoint_A && aPoint_B) {
66     aPnt_A = aPoint_A->pnt();
67     aPnt_B = aPoint_B->pnt();
68   } else if (!aPoint_A && aPoint_B) {
69     std::shared_ptr<SketchPlugin_Line> aLine = getFeatureLine(
70         aData, SketchPlugin_Constraint::ENTITY_A());
71     if (aLine) {
72       aPnt_B = aPoint_B->pnt();
73       aPnt_A = getProjectionPoint(aLine, aPnt_B);
74     }
75   } else if (aPoint_A && !aPoint_B) {
76     std::shared_ptr<SketchPlugin_Line> aLine = getFeatureLine(
77         aData, SketchPlugin_Constraint::ENTITY_B());
78     if (aLine) {
79       aPnt_A = aPoint_A->pnt();
80       aPnt_B = getProjectionPoint(aLine, aPnt_A);
81     }
82   }
83   if (!aPnt_A || !aPnt_B)
84     return AISObjectPtr();
85
86   std::shared_ptr<GeomDataAPI_Point2D> aFlyOutAttr = std::dynamic_pointer_cast<
87       GeomDataAPI_Point2D>(aData->attribute(SketchPlugin_Constraint::FLYOUT_VALUE_PNT()));
88
89   std::shared_ptr<GeomAPI_Pnt> aPoint1 = sketch()->to3D(aPnt_A->x(), aPnt_A->y());
90   std::shared_ptr<GeomAPI_Pnt> aPoint2 = sketch()->to3D(aPnt_B->x(), aPnt_B->y());
91   std::shared_ptr<GeomAPI_Pnt> aFlyoutPnt = std::shared_ptr<GeomAPI_Pnt>();
92   if(aFlyOutAttr->isInitialized()) {
93     aFlyoutPnt = sketch()->to3D(aFlyOutAttr->x(), aFlyOutAttr->y());
94   } else {
95     std::shared_ptr<GeomAPI_Lin2d> aLine = std::shared_ptr<GeomAPI_Lin2d>(new GeomAPI_Lin2d(aPnt_A, aPnt_B));
96     double aDist = aPoint1->distance(aPoint2)/5.;
97     std::shared_ptr<GeomAPI_Pnt2d> aFPnt = aLine->shiftedLocation(aDist);
98     aFlyOutAttr->setValue(aFPnt);
99     aFlyoutPnt = sketch()->to3D(aFPnt->x(), aFPnt->y());
100   }
101   // value calculation
102   std::shared_ptr<ModelAPI_AttributeDouble> aValueAttr = std::dynamic_pointer_cast<
103       ModelAPI_AttributeDouble>(aData->attribute(SketchPlugin_Constraint::VALUE()));
104   double aValue = 0;
105   // Issue #196: checking the positivity of the distance constraint
106   // there is a validator for a distance constraint, that the value should be positive
107   // in case if an invalid value is set, the current distance value is shown
108   if (aValueAttr->isInitialized())
109     aValue = aValueAttr->value();
110
111   AISObjectPtr anAIS = thePrevious;
112   if (!anAIS)
113     anAIS = AISObjectPtr(new GeomAPI_AISObject);
114   anAIS->createDistance(aPoint1, aPoint2, aFlyoutPnt, aPlane, aValue);
115
116   // Set color from preferences
117   std::vector<int> aRGB = Config_PropManager::color("Visualization", "distance_color",
118                                                     DISTANCE_COLOR);
119   anAIS->setColor(aRGB[0], aRGB[1], aRGB[2]);
120   return anAIS;
121 }
122
123 //*************************************************************************************
124 void SketchPlugin_ConstraintDistance::move(double theDeltaX, double theDeltaY)
125 {
126   std::shared_ptr<ModelAPI_Data> aData = data();
127   if (!aData->isValid())
128     return;
129
130   std::shared_ptr<GeomDataAPI_Point2D> aPoint = std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
131       aData->attribute(SketchPlugin_Constraint::FLYOUT_VALUE_PNT()));
132   aPoint->move(theDeltaX, theDeltaY);
133 }
134
135 double SketchPlugin_ConstraintDistance::calculateCurrentDistance() const
136 {
137   double aDistance = -1.;
138
139   std::shared_ptr<ModelAPI_Data> aData = data();
140   std::shared_ptr<GeomDataAPI_Point2D> aPointA =
141     getFeaturePoint(aData, SketchPlugin_Constraint::ENTITY_A());
142   std::shared_ptr<GeomDataAPI_Point2D> aPointB =
143       getFeaturePoint(aData, SketchPlugin_Constraint::ENTITY_B());
144
145   if (aPointA && aPointB) {  // both points
146     aDistance = aPointA->pnt()->distance(aPointB->pnt());
147   } else {
148     if (!aPointA && aPointB) {  //Line and point
149       std::shared_ptr<SketchPlugin_Line> aLine =
150           getFeatureLine(aData, SketchPlugin_Constraint::ENTITY_A());
151       if (aLine) {
152         aDistance = aLine->distanceToPoint(aPointB->pnt());
153       }
154     } else if (aPointA && !aPointB) {  // Point and line
155       std::shared_ptr<SketchPlugin_Line> aLine =
156           getFeatureLine(aData, SketchPlugin_Constraint::ENTITY_B());
157       if (aLine) {
158         aDistance = aLine->distanceToPoint(aPointA->pnt());
159       }
160     }
161   }
162   return aDistance;
163 }
164
165 void SketchPlugin_ConstraintDistance::attributeChanged(const std::string& theID) {
166   if (theID == SketchPlugin_Constraint::ENTITY_A() || 
167       theID == SketchPlugin_Constraint::ENTITY_B()) 
168   {
169     std::shared_ptr<ModelAPI_AttributeDouble> aValueAttr = std::dynamic_pointer_cast<
170         ModelAPI_AttributeDouble>(data()->attribute(SketchPlugin_Constraint::VALUE()));
171     if (!aValueAttr->isInitialized()) { // only if it is not initialized, try to compute the current value
172       double aDistance = calculateCurrentDistance();
173       if (aDistance > 0) { // set as value the length of updated references
174         aValueAttr->setValue(aDistance);
175       }
176     }
177   }
178 }
179
180 //*************************************************************************************
181 std::shared_ptr<GeomDataAPI_Point2D> getFeaturePoint(DataPtr theData,
182                                                        const std::string& theAttribute)
183 {
184   std::shared_ptr<GeomDataAPI_Point2D> aPointAttr;
185
186   if (!theData)
187     return aPointAttr;
188
189   FeaturePtr aFeature;
190   std::shared_ptr<ModelAPI_AttributeRefAttr> anAttr = std::dynamic_pointer_cast<
191       ModelAPI_AttributeRefAttr>(theData->attribute(theAttribute));
192   if (anAttr)
193     aFeature = ModelAPI_Feature::feature(anAttr->object());
194
195   if (aFeature && aFeature->getKind() == SketchPlugin_Point::ID())
196     aPointAttr = std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
197         aFeature->data()->attribute(SketchPlugin_Point::COORD_ID()));
198   else if (aFeature && aFeature->getKind() == SketchPlugin_Circle::ID())
199     aPointAttr = std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
200         aFeature->data()->attribute(SketchPlugin_Circle::CENTER_ID()));
201   else if (anAttr->attr()) {
202     aPointAttr = std::dynamic_pointer_cast<GeomDataAPI_Point2D>(anAttr->attr());
203   }
204   return aPointAttr;
205 }
206
207 //*************************************************************************************
208 std::shared_ptr<SketchPlugin_Line> getFeatureLine(DataPtr theData,
209                                                     const std::string& theAttribute)
210 {
211   std::shared_ptr<SketchPlugin_Line> aLine;
212   if (!theData)
213     return aLine;
214
215   std::shared_ptr<ModelAPI_AttributeRefAttr> anAttr = std::dynamic_pointer_cast<
216       ModelAPI_AttributeRefAttr>(theData->attribute(theAttribute));
217   if (anAttr) {
218     FeaturePtr aFeature = ModelAPI_Feature::feature(anAttr->object());
219     if (aFeature && aFeature->getKind() == SketchPlugin_Line::ID()) {
220       aLine = std::dynamic_pointer_cast<SketchPlugin_Line>(aFeature);
221     }
222   }
223   return aLine;
224 }
225
226 //*************************************************************************************
227 std::shared_ptr<GeomAPI_Pnt2d> getProjectionPoint(
228     const std::shared_ptr<SketchPlugin_Line>& theLine,
229     const std::shared_ptr<GeomAPI_Pnt2d>& thePoint)
230 {
231   std::shared_ptr<ModelAPI_Data> aData = theLine->data();
232   std::shared_ptr<GeomDataAPI_Point2D> aPoint1 = std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
233       aData->attribute(SketchPlugin_Line::START_ID()));
234   std::shared_ptr<GeomDataAPI_Point2D> aPoint2 = std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
235       aData->attribute(SketchPlugin_Line::END_ID()));
236
237   GeomAPI_Lin2d aLin2d(aPoint1->x(), aPoint1->y(), aPoint2->x(), aPoint2->y());
238   return aLin2d.project(thePoint);
239 }