]> SALOME platform Git repositories - modules/shaper.git/blob - src/SketchPlugin/SketchPlugin_Arc.cpp
Salome HOME
Initial version of redesign of working with results
[modules/shaper.git] / src / SketchPlugin / SketchPlugin_Arc.cpp
1 // File:        SketchPlugin_Arc.cpp
2 // Created:     26 Apr 2014
3 // Author:      Artem ZHIDKOV
4
5 #include "SketchPlugin_Arc.h"
6 #include "SketchPlugin_Sketch.h"
7 #include <ModelAPI_Data.h>
8 #include <ModelAPI_ResultConstruction.h>
9
10 #include <GeomAPI_Circ2d.h>
11 #include <GeomAPI_Pnt2d.h>
12 #include <GeomDataAPI_Point2D.h>
13 #include <GeomDataAPI_Dir.h>
14 #include <GeomAlgoAPI_PointBuilder.h>
15 #include <GeomAlgoAPI_EdgeBuilder.h>
16 #include <GeomAlgoAPI_CompoundBuilder.h>
17
18 const double tolerance = 1e-7;
19
20 SketchPlugin_Arc::SketchPlugin_Arc()
21   : SketchPlugin_Feature()
22 {
23 }
24
25 void SketchPlugin_Arc::initAttributes()
26 {
27   data()->addAttribute(ARC_ATTR_CENTER, GeomDataAPI_Point2D::type());
28   data()->addAttribute(ARC_ATTR_START,  GeomDataAPI_Point2D::type());
29   data()->addAttribute(ARC_ATTR_END,    GeomDataAPI_Point2D::type());
30 }
31
32 void SketchPlugin_Arc::execute() 
33 {
34   SketchPlugin_Sketch* aSketch = sketch();
35   if (aSketch) {
36     std::list<boost::shared_ptr<GeomAPI_Shape> > aShapes;
37
38     // compute a circle point in 3D view
39     boost::shared_ptr<GeomDataAPI_Point2D> aCenterAttr = 
40       boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(data()->attribute(ARC_ATTR_CENTER));
41       // compute the arc start point
42       boost::shared_ptr<GeomDataAPI_Point2D> aStartAttr = 
43         boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(data()->attribute(ARC_ATTR_START));
44       if (aCenterAttr->isInitialized() && aStartAttr->isInitialized()) {
45       boost::shared_ptr<GeomAPI_Pnt> aCenter(aSketch->to3D(aCenterAttr->x(), aCenterAttr->y()));
46       // make a visible point
47       boost::shared_ptr<GeomAPI_Shape> aCenterPointShape = GeomAlgoAPI_PointBuilder::point(aCenter);
48       aShapes.push_back(aCenterPointShape);
49
50       // make a visible circle
51       boost::shared_ptr<GeomDataAPI_Dir> aNDir = 
52         boost::dynamic_pointer_cast<GeomDataAPI_Dir>(aSketch->data()->attribute(SKETCH_ATTR_NORM));
53       bool aHasPlane = aNDir && !(aNDir->x() == 0 && aNDir->y() == 0 && aNDir->z() == 0);
54       if (aHasPlane) {
55         boost::shared_ptr<GeomAPI_Dir> aNormal = aNDir->dir();
56         boost::shared_ptr<GeomAPI_Pnt> aStartPoint(aSketch->to3D(aStartAttr->x(), aStartAttr->y()));
57
58         // compute and change the arc end point
59         boost::shared_ptr<GeomDataAPI_Point2D> anEndAttr = 
60           boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(data()->attribute(ARC_ATTR_END));
61         if (anEndAttr->isInitialized())
62         {
63           boost::shared_ptr<GeomAPI_Circ2d> aCircleForArc(
64             new GeomAPI_Circ2d(aCenterAttr->pnt(), aStartAttr->pnt()));
65           boost::shared_ptr<GeomAPI_Pnt2d> aProjection = aCircleForArc->project(anEndAttr->pnt());
66           if (aProjection && anEndAttr->pnt()->distance(aProjection) > tolerance)
67             anEndAttr->setValue(aProjection);
68         }
69         boost::shared_ptr<GeomAPI_Pnt> aEndPoint(aSketch->to3D(anEndAttr->x(), anEndAttr->y()));
70
71         boost::shared_ptr<GeomAPI_Shape> aCircleShape = 
72                    GeomAlgoAPI_EdgeBuilder::lineCircleArc(aCenter, aStartPoint, aEndPoint, aNormal);
73         if (aCircleShape)
74           aShapes.push_back(aCircleShape);
75       }
76       boost::shared_ptr<GeomAPI_Shape> aCompound = GeomAlgoAPI_CompoundBuilder::compound(aShapes);
77       // store the result
78       boost::shared_ptr<ModelAPI_ResultConstruction> aConstr = document()->createConstruction();
79       aConstr->setShape(aCompound);
80       results().push_back(aConstr);
81     }
82   }
83 }
84
85 void SketchPlugin_Arc::move(double theDeltaX, double theDeltaY)
86 {
87   boost::shared_ptr<ModelAPI_Data> aData = data();
88   if (!aData->isValid())
89     return;
90
91   boost::shared_ptr<GeomDataAPI_Point2D> aPoint1 =
92         boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(ARC_ATTR_CENTER));
93   aPoint1->setValue(aPoint1->x() + theDeltaX, aPoint1->y() + theDeltaY);
94
95   boost::shared_ptr<GeomDataAPI_Point2D> aPoint2 =
96         boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(ARC_ATTR_START));
97   aPoint2->setValue(aPoint2->x() + theDeltaX, aPoint2->y() + theDeltaY);
98
99   boost::shared_ptr<GeomDataAPI_Point2D> aPoint3 =
100         boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(ARC_ATTR_END));
101   aPoint3->setValue(aPoint3->x() + theDeltaX, aPoint3->y() + theDeltaY);
102 }
103
104 double SketchPlugin_Arc::distanceToPoint(const boost::shared_ptr<GeomAPI_Pnt2d>& thePoint)
105 {
106   double aDelta = 0;
107   boost::shared_ptr<ModelAPI_Data> aData = data();
108
109   boost::shared_ptr<GeomDataAPI_Point2D> aPoint1 =
110         boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(ARC_ATTR_CENTER));
111   aDelta = aPoint1->pnt()->distance(thePoint);
112
113   boost::shared_ptr<GeomDataAPI_Point2D> aPoint2 =
114         boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(ARC_ATTR_START));
115   double aDistance = aPoint2->pnt()->distance(thePoint);
116   if (aDelta < aDistance)
117     aDelta = aDistance;
118
119   boost::shared_ptr<GeomDataAPI_Point2D> aPoint3 =
120         boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(ARC_ATTR_END));
121   aDistance = aPoint3->pnt()->distance(thePoint);
122   if (aDelta < aDistance)
123     aDelta = aDistance;
124
125   return aDelta;
126 }