Salome HOME
Merge with Dev_1.5.0
[modules/shaper.git] / src / SketcherPrs / SketcherPrs_Radius.cpp
1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
2
3 // File:        SketcherPrs_Radius.cpp
4 // Created:     26 March 2015
5 // Author:      Vitaly SMETANNIKOV
6
7 #include "SketcherPrs_Radius.h"
8 #include "SketcherPrs_Tools.h"
9
10 #include <SketchPlugin_ConstraintRadius.h>
11 #include <SketchPlugin_Constraint.h>
12 #include <SketchPlugin_Circle.h>
13 #include <SketchPlugin_Arc.h>
14
15 #include <GeomDataAPI_Point2D.h>
16 #include <GeomAPI_Pnt2d.h>
17 #include <GeomAPI_Circ.h>
18 #include <GeomAPI_XYZ.h>
19 #include <ModelAPI_AttributeDouble.h>
20
21 #include <gp_Circ.hxx>
22
23 static const gp_Circ MyDefCirc(gp_Ax2(gp_Pnt(0,0,0), gp_Dir(0,0,1)), 1);
24
25 IMPLEMENT_STANDARD_HANDLE(SketcherPrs_Radius, AIS_RadiusDimension);
26 IMPLEMENT_STANDARD_RTTIEXT(SketcherPrs_Radius, AIS_RadiusDimension);
27
28 SketcherPrs_Radius::SketcherPrs_Radius(ModelAPI_Feature* theConstraint, 
29                                        const std::shared_ptr<GeomAPI_Ax3>& thePlane)
30 : AIS_RadiusDimension(MyDefCirc), myConstraint(theConstraint), myPlane(thePlane)
31 {
32   // Set default values of the presentation
33   myAspect = new Prs3d_DimensionAspect();
34   myAspect->MakeArrows3d(false);
35   myAspect->MakeText3d(false);
36   myAspect->MakeTextShaded(false);
37   myAspect->MakeUnitsDisplayed(false);
38   myAspect->TextAspect()->SetHeight(SketcherPrs_Tools::getDefaultTextHeight());
39   myAspect->ArrowAspect()->SetLength(SketcherPrs_Tools::getArrowSize());
40   
41   SetDimensionAspect(myAspect);
42   SetSelToleranceForText2d(SketcherPrs_Tools::getDefaultTextHeight());
43 }
44
45 void SketcherPrs_Radius::Compute(const Handle(PrsMgr_PresentationManager3d)& thePresentationManager,
46                                  const Handle(Prs3d_Presentation)& thePresentation, 
47                                  const Standard_Integer theMode)
48 {
49   DataPtr aData = myConstraint->data();
50
51   // Flyout point
52   std::shared_ptr<GeomDataAPI_Point2D> aFlyoutAttr = 
53     std::dynamic_pointer_cast<GeomDataAPI_Point2D>
54     (aData->attribute(SketchPlugin_Constraint::FLYOUT_VALUE_PNT()));
55   if (!aFlyoutAttr->isInitialized())
56     return; // can not create a good presentation
57
58   // Get circle
59   std::shared_ptr<ModelAPI_AttributeRefAttr> anAttr = 
60     std::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>
61     (aData->attribute(SketchPlugin_Constraint::ENTITY_A()));
62   if (!anAttr) return;
63
64   std::shared_ptr<ModelAPI_Feature> aCyrcFeature = ModelAPI_Feature::feature(anAttr->object());
65   double aRadius = 1;
66   std::shared_ptr<GeomDataAPI_Point2D> aCenterAttr;
67   // it is possible that circle result becomes zero, in this case the presentation should disappear
68   // for example, it happens when circle radius is set to zero
69   if (!aCyrcFeature)
70     return;
71   if (aCyrcFeature->getKind() == SketchPlugin_Circle::ID()) { // circle
72     aCenterAttr = std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
73         aCyrcFeature->data()->attribute(SketchPlugin_Circle::CENTER_ID()));
74     AttributeDoublePtr aCircRadius = 
75       std::dynamic_pointer_cast<ModelAPI_AttributeDouble>(
76       aCyrcFeature->data()->attribute(SketchPlugin_Circle::RADIUS_ID()));
77     aRadius = aCircRadius->value();
78   } else { // arc
79     aCenterAttr = std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
80         aCyrcFeature->data()->attribute(SketchPlugin_Arc::CENTER_ID()));
81     std::shared_ptr<GeomDataAPI_Point2D> aStartAttr = 
82       std::dynamic_pointer_cast<GeomDataAPI_Point2D>
83       (aCyrcFeature->data()->attribute(SketchPlugin_Arc::START_ID()));
84     aRadius = aCenterAttr->pnt()->distance(aStartAttr->pnt());
85   }
86   std::shared_ptr<GeomAPI_Pnt> aCenter = myPlane->to3D(aCenterAttr->x(), aCenterAttr->y());
87   std::shared_ptr<GeomAPI_Dir> aNormal = myPlane->normal();
88
89   GeomAPI_Circ aCircle(aCenter, aNormal, aRadius);
90     
91   std::shared_ptr<GeomAPI_Pnt> anAnchor = SketcherPrs_Tools::getAnchorPoint(myConstraint, myPlane);
92
93   gp_Circ aCirc = aCircle.impl<gp_Circ>();
94   gp_Pnt anAncorPnt = anAnchor->impl<gp_Pnt>();
95   // anchor point should not coincide to the location point of the circle
96   // OCCT does not process this case.
97   if (anAncorPnt.Distance(aCirc.Location()) < 1e-7)
98     return;
99   SetMeasuredGeometry(aCirc, anAncorPnt);
100   SetCustomValue(aRadius);
101
102   myAspect->SetExtensionSize(myAspect->ArrowAspect()->Length());
103   myAspect->SetArrowTailSize(myAspect->ArrowAspect()->Length());
104   // The value of vertical aligment is sometimes changed
105   myAspect->TextAspect()->SetVerticalJustification(Graphic3d_VTA_CENTER);
106
107   AIS_RadiusDimension::Compute(thePresentationManager, thePresentation, theMode);
108 }
109
110 void SketcherPrs_Radius::ComputeSelection(const Handle(SelectMgr_Selection)& aSelection,
111                                                    const Standard_Integer theMode)
112 {
113   // Map the application selection modes to standard ones
114   Standard_Integer aMode;
115   switch (theMode) {
116   case 0: // we should use selection of all objects
117     aMode = 0;
118     break;
119   case SketcherPrs_Tools::Sel_Dimension_All:
120     aMode = 0;
121     break;
122   case SketcherPrs_Tools::Sel_Dimension_Line:
123     aMode = 1;
124     break;
125   case SketcherPrs_Tools::Sel_Dimension_Text:
126     aMode = 2;
127     break;
128   default: {
129     // there are own selection modes, so the others should be ignored
130     // otherwise, the text selection appears in the viewer
131     return; 
132   }
133   }
134   AIS_RadiusDimension::ComputeSelection(aSelection, aMode);
135 }