1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D -->
3 // File: SketchPlugin_Circle.cpp
4 // Created: 26 May 2014
5 // Author: Artem ZHIDKOV
7 #include "SketchPlugin_Circle.h"
8 #include "SketchPlugin_Sketch.h"
9 #include <ModelAPI_Data.h>
10 #include <ModelAPI_ResultConstruction.h>
11 #include <ModelAPI_AttributeSelection.h>
12 #include <ModelAPI_Validator.h>
13 #include <ModelAPI_AttributeDouble.h>
14 #include <ModelAPI_AttributeString.h>
15 #include <ModelAPI_Session.h>
17 #include <GeomAPI_Pnt2d.h>
18 #include <GeomAPI_Circ.h>
19 #include <GeomAPI_Circ2d.h>
20 #include <GeomAPI_XY.h>
21 #include <GeomDataAPI_Point2D.h>
22 #include <GeomDataAPI_Dir.h>
23 #include <GeomAlgoAPI_PointBuilder.h>
24 #include <GeomAlgoAPI_EdgeBuilder.h>
25 #include <GeomAlgoAPI_CompoundBuilder.h>
28 static const std::string& CIRCLE_TYPE()
30 static const std::string TYPE("CircleType");
33 static const std::string& CIRCLE_TYPE_CENTER_AND_RADIUS()
35 static const std::string TYPE("CenterRadius");
38 static const std::string& CIRCLE_TYPE_THREE_POINTS()
40 static const std::string TYPE("ThreePoints");
44 static const std::string& FIRST_POINT_ID()
46 static const std::string FIRST_PNT("FirstPoint");
49 static const std::string& SECOND_POINT_ID()
51 static const std::string SECOND_PNT("SecondPoint");
54 static const std::string& THIRD_POINT_ID()
56 static const std::string THIRD_PNT("ThirdPoint");
59 static const std::string& POINT_ID(int theIndex)
62 case 1: return FIRST_POINT_ID();
63 case 2: return SECOND_POINT_ID();
64 case 3: return THIRD_POINT_ID();
67 static const std::string DUMMY;
73 SketchPlugin_Circle::SketchPlugin_Circle()
74 : SketchPlugin_SketchEntity()
78 void SketchPlugin_Circle::initDerivedClassAttributes()
80 data()->addAttribute(CENTER_ID(), GeomDataAPI_Point2D::typeId());
81 data()->addAttribute(RADIUS_ID(), ModelAPI_AttributeDouble::typeId());
82 data()->addAttribute(EXTERNAL_ID(), ModelAPI_AttributeSelection::typeId());
83 ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), EXTERNAL_ID());
85 data()->addAttribute(CIRCLE_TYPE(), ModelAPI_AttributeString::typeId());
86 data()->addAttribute(FIRST_POINT_ID(), GeomDataAPI_Point2D::typeId());
87 data()->addAttribute(SECOND_POINT_ID(), GeomDataAPI_Point2D::typeId());
88 data()->addAttribute(THIRD_POINT_ID(), GeomDataAPI_Point2D::typeId());
89 std::dynamic_pointer_cast<ModelAPI_AttributeString>(
90 data()->attribute(CIRCLE_TYPE()))->setValue(CIRCLE_TYPE_CENTER_AND_RADIUS());
93 void SketchPlugin_Circle::execute()
95 SketchPlugin_Sketch* aSketch = sketch();
97 std::list<std::shared_ptr<GeomAPI_Shape> > aShapes;
99 // compute a circle point in 3D view
100 std::shared_ptr<GeomDataAPI_Point2D> aCenterAttr = std::dynamic_pointer_cast<
101 GeomDataAPI_Point2D>(data()->attribute(CENTER_ID()));
102 AttributeDoublePtr aRadiusAttr =
103 std::dynamic_pointer_cast<ModelAPI_AttributeDouble>(data()->attribute(RADIUS_ID()));
104 if (aCenterAttr->isInitialized() && aRadiusAttr->isInitialized()) {
105 std::shared_ptr<GeomAPI_Pnt> aCenter(aSketch->to3D(aCenterAttr->x(), aCenterAttr->y()));
106 //std::cout<<"Execute circle "<<aCenter->x()<<" "<<aCenter->y()<<" "<<aCenter->z()<<std::endl;
107 // make a visible point
108 SketchPlugin_Sketch::createPoint2DResult(this, sketch(), CENTER_ID(), 0);
110 // make a visible circle
111 std::shared_ptr<GeomDataAPI_Dir> aNDir = std::dynamic_pointer_cast<GeomDataAPI_Dir>(
112 aSketch->data()->attribute(SketchPlugin_Sketch::NORM_ID()));
113 std::shared_ptr<GeomAPI_Dir> aNormal(new GeomAPI_Dir(aNDir->x(), aNDir->y(), aNDir->z()));
114 // compute the circle radius
115 double aRadius = aRadiusAttr->value();
117 std::shared_ptr<GeomAPI_Shape> aCircleShape = GeomAlgoAPI_EdgeBuilder::lineCircle(
118 aCenter, aNormal, aRadius);
119 aShapes.push_back(aCircleShape);
120 std::shared_ptr<ModelAPI_ResultConstruction> aConstr2 = document()->createConstruction(
122 aConstr2->setShape(aCircleShape);
123 aConstr2->setIsInHistory(false);
124 setResult(aConstr2, 1);
129 AISObjectPtr SketchPlugin_Circle::getAISObject(AISObjectPtr thePrevious)
131 SketchPlugin_Sketch* aSketch = sketch();
132 if (aSketch && !isFeatureValid()) {
133 // compute a circle point in 3D view
134 std::shared_ptr<GeomDataAPI_Point2D> aCenterAttr = std::dynamic_pointer_cast<
135 GeomDataAPI_Point2D>(data()->attribute(CENTER_ID()));
136 AttributeDoublePtr aRadiusAttr =
137 std::dynamic_pointer_cast<ModelAPI_AttributeDouble>(attribute(RADIUS_ID()));
138 if (aCenterAttr->isInitialized() && aRadiusAttr->isInitialized()) {
139 std::shared_ptr<GeomAPI_Pnt> aCenter(aSketch->to3D(aCenterAttr->x(), aCenterAttr->y()));
141 // make a visible circle
142 std::shared_ptr<GeomDataAPI_Dir> aNDir = std::dynamic_pointer_cast<GeomDataAPI_Dir>(
143 aSketch->data()->attribute(SketchPlugin_Sketch::NORM_ID()));
144 std::shared_ptr<GeomAPI_Dir> aNormal = aNDir->dir();
146 double aRadius = aRadiusAttr->value();
147 std::shared_ptr<GeomAPI_Shape> aCircleShape = GeomAlgoAPI_EdgeBuilder::lineCircle(
148 aCenter, aNormal, aRadius);
149 if (aCircleShape && aRadius != 0) {
150 std::list<std::shared_ptr<GeomAPI_Shape> > aShapes;
151 // make a visible point
152 std::shared_ptr<GeomAPI_Shape> aCenterPointShape = GeomAlgoAPI_PointBuilder::point(aCenter);
153 aShapes.push_back(aCenterPointShape);
154 aShapes.push_back(aCircleShape);
156 std::shared_ptr<GeomAPI_Shape> aCompound = GeomAlgoAPI_CompoundBuilder::compound(aShapes);
157 AISObjectPtr anAIS = thePrevious;
159 anAIS = AISObjectPtr(new GeomAPI_AISObject);
160 anAIS->createShape(aCompound);
166 return AISObjectPtr();
169 bool SketchPlugin_Circle::isFeatureValid()
171 std::shared_ptr<GeomDataAPI_Point2D> aCenter =
172 std::dynamic_pointer_cast<GeomDataAPI_Point2D>(attribute(CENTER_ID()));
173 std::shared_ptr<GeomDataAPI_Point2D> aFirstPnt =
174 std::dynamic_pointer_cast<GeomDataAPI_Point2D>(attribute(FIRST_POINT_ID()));
175 std::shared_ptr<GeomDataAPI_Point2D> aSecondPnt =
176 std::dynamic_pointer_cast<GeomDataAPI_Point2D>(attribute(SECOND_POINT_ID()));
177 std::shared_ptr<GeomDataAPI_Point2D> aThirdPnt =
178 std::dynamic_pointer_cast<GeomDataAPI_Point2D>(attribute(THIRD_POINT_ID()));
180 return aCenter->isInitialized() && aFirstPnt->isInitialized() &&
181 aSecondPnt->isInitialized() && aThirdPnt->isInitialized();
184 void SketchPlugin_Circle::move(double theDeltaX, double theDeltaY)
186 std::shared_ptr<ModelAPI_Data> aData = data();
187 if (!aData->isValid())
190 std::shared_ptr<GeomDataAPI_Point2D> aPoint = std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
191 aData->attribute(CENTER_ID()));
192 aPoint->move(theDeltaX, theDeltaY);
194 aPoint = std::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(FIRST_POINT_ID()));
195 aPoint->move(theDeltaX, theDeltaY);
196 aPoint = std::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(SECOND_POINT_ID()));
197 aPoint->move(theDeltaX, theDeltaY);
198 aPoint = std::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(THIRD_POINT_ID()));
199 aPoint->move(theDeltaX, theDeltaY);
202 bool SketchPlugin_Circle::isFixed() {
203 return data()->selection(EXTERNAL_ID())->context().get() != NULL;
206 void SketchPlugin_Circle::attributeChanged(const std::string& theID) {
207 // the second condition for unability to move external segments anywhere
208 if (theID == EXTERNAL_ID() || isFixed()) {
209 std::shared_ptr<GeomAPI_Shape> aSelection = data()->selection(EXTERNAL_ID())->value();
210 // update arguments due to the selection value
211 if (aSelection && !aSelection->isNull() && aSelection->isEdge()) {
212 std::shared_ptr<GeomAPI_Edge> anEdge( new GeomAPI_Edge(aSelection));
213 std::shared_ptr<GeomAPI_Circ> aCirc = anEdge->circle();
214 std::shared_ptr<GeomDataAPI_Point2D> aCenterAttr =
215 std::dynamic_pointer_cast<GeomDataAPI_Point2D>(attribute(CENTER_ID()));
216 aCenterAttr->setValue(sketch()->to2D(aCirc->center()));
217 real(RADIUS_ID())->setValue(aCirc->radius());
220 else if (theID == CENTER_ID() || theID == RADIUS_ID()) {
221 std::string aType = std::dynamic_pointer_cast<ModelAPI_AttributeString>(
222 data()->attribute(CIRCLE_TYPE()))->value();
223 if (aType == CIRCLE_TYPE_THREE_POINTS())
226 std::shared_ptr<GeomDataAPI_Point2D> aCenterAttr =
227 std::dynamic_pointer_cast<GeomDataAPI_Point2D>(attribute(CENTER_ID()));
228 if (!aCenterAttr->isInitialized())
230 AttributeDoublePtr aRadiusAttr =
231 std::dynamic_pointer_cast<ModelAPI_AttributeDouble>(attribute(RADIUS_ID()));
232 if (!aRadiusAttr->isInitialized())
235 // check the execute() was called and the shape was built
239 data()->blockSendAttributeUpdated(true);
240 std::shared_ptr<GeomDataAPI_Point2D> aFirstPnt =
241 std::dynamic_pointer_cast<GeomDataAPI_Point2D>(attribute(FIRST_POINT_ID()));
242 std::shared_ptr<GeomDataAPI_Point2D> aSecondPnt =
243 std::dynamic_pointer_cast<GeomDataAPI_Point2D>(attribute(SECOND_POINT_ID()));
244 std::shared_ptr<GeomDataAPI_Point2D> aThirdPnt =
245 std::dynamic_pointer_cast<GeomDataAPI_Point2D>(attribute(THIRD_POINT_ID()));
246 double aRadius = aRadiusAttr->value();
247 aFirstPnt->setValue(aCenterAttr->x() + aRadius, aCenterAttr->y());
248 aSecondPnt->setValue(aCenterAttr->x(), aCenterAttr->y() + aRadius);
249 aThirdPnt->setValue(aCenterAttr->x() - aRadius, aCenterAttr->y());
250 data()->blockSendAttributeUpdated(false);
252 else if (theID == FIRST_POINT_ID() || theID == SECOND_POINT_ID() || theID == THIRD_POINT_ID()) {
253 std::string aType = std::dynamic_pointer_cast<ModelAPI_AttributeString>(
254 data()->attribute(CIRCLE_TYPE()))->value();
255 if (aType == CIRCLE_TYPE_CENTER_AND_RADIUS())
258 data()->blockSendAttributeUpdated(true);
260 std::shared_ptr<GeomAPI_Pnt2d> aPoints[3];
261 int aNbInitialized = 0;
262 for (int i = 1; i <= 3; ++i) {
263 std::shared_ptr<GeomDataAPI_Point2D> aCurPnt =
264 std::dynamic_pointer_cast<GeomDataAPI_Point2D>(attribute(POINT_ID(i)));
265 if (aCurPnt->isInitialized())
266 aPoints[aNbInitialized++] = aCurPnt->pnt();
269 std::shared_ptr<GeomDataAPI_Point2D> aCenterAttr = std::dynamic_pointer_cast<
270 GeomDataAPI_Point2D>(data()->attribute(CENTER_ID()));
271 AttributeDoublePtr aRadiusAttr =
272 std::dynamic_pointer_cast<ModelAPI_AttributeDouble>(data()->attribute(RADIUS_ID()));
274 if (aNbInitialized == 1)
275 aCenterAttr->setValue(aPoints[0]->x(), aPoints[0]->y());
276 else if (aNbInitialized == 2) {
277 std::shared_ptr<GeomAPI_XY> aCoord =
278 aPoints[0]->xy()->added(aPoints[1]->xy())->multiplied(0.5);
279 double aRadius = aPoints[0]->distance(aPoints[1]) * 0.5;
280 aCenterAttr->setValue(aCoord->x(), aCoord->y());
281 aRadiusAttr->setValue(aRadius);
283 std::shared_ptr<GeomAPI_Circ2d> aCircle(
284 new GeomAPI_Circ2d(aPoints[0], aPoints[1], aPoints[2]));
286 std::shared_ptr<GeomAPI_Pnt2d> aCenter = aCircle->center();
288 double aRadius = aCircle->radius();
289 aCenterAttr->setValue(aCenter->x(), aCenter->y());
290 aRadiusAttr->setValue(aRadius);
294 data()->blockSendAttributeUpdated(false);