src/PartSet/PartSet_FeatureArcPrs.cpp

   1 // File:        PartSet_FeaturePrs.h
   2 // Created:     04 Jun 2014
   3 // Author:      Natalia ERMOLAEVA
   4
   5 #include <PartSet_FeatureArcPrs.h>
   6 #include <PartSet_Tools.h>
   7
   8 #include <SketchPlugin_Feature.h>
   9 #include <SketchPlugin_Sketch.h>
  10 #include <SketchPlugin_Arc.h>
  11
  12 #include <GeomDataAPI_Point2D.h>
  13 #include <GeomAPI_Pnt2d.h>
  14 #include <GeomAPI_Circ2d.h>
  15
  16 #include <ModelAPI_Data.h>
  17 #include <ModelAPI_Document.h>
  18 #include <ModelAPI_AttributeRefAttr.h>
  19 #include <ModelAPI_AttributeRefList.h>
  20
  21 #include <V3d_View.hxx>
  22
  23 #include <Precision.hxx>
  24
  25 using namespace std;
  26
  27 PartSet_FeatureArcPrs::PartSet_FeatureArcPrs(FeaturePtr theSketch)
  28 : PartSet_FeaturePrs(theSketch)
  29 {
  30 }
  31
  32 std::string PartSet_FeatureArcPrs::getKind()
  33 {
  34   return SKETCH_ARC_KIND;
  35 }
  36
  37 PartSet_SelectionMode PartSet_FeatureArcPrs::setPoint(double theX, double theY,
  38                                                        const PartSet_SelectionMode& theMode)
  39 {
  40   PartSet_SelectionMode aMode = theMode;
  41   switch (theMode)
  42   {
  43     case SM_FirstPoint: {
  44       PartSet_Tools::setFeaturePoint(feature(), theX, theY, ARC_ATTR_CENTER);
  45       aMode = SM_SecondPoint;
  46     }
  47     break;
  48     case SM_SecondPoint: {
  49       PartSet_Tools::setFeaturePoint(feature(), theX, theY, ARC_ATTR_START);
  50       aMode = SM_ThirdPoint;
  51    }
  52    break;
  53    case SM_ThirdPoint: {
  54      PartSet_Tools::setFeaturePoint(feature(), theX, theY, ARC_ATTR_END);
  55      aMode = SM_DonePoint;
  56    }
  57     break;
  58     default:
  59       break;
  60   }
  61   return aMode;
  62 }
  63
  64 std::string PartSet_FeatureArcPrs::getAttribute(const PartSet_SelectionMode& theMode) const
  65 {
  66   std::string aAttribute;
  67   switch (theMode)
  68   {
  69     case SM_FirstPoint:
  70       aAttribute = ARC_ATTR_CENTER;
  71     break;
  72     case SM_SecondPoint:
  73       aAttribute = ARC_ATTR_START;
  74     break;
  75     case SM_ThirdPoint:
  76       aAttribute = ARC_ATTR_END;
  77     break;
  78     default:
  79     break;
  80   }
  81   return aAttribute;
  82 }
  83
  84 PartSet_SelectionMode PartSet_FeatureArcPrs::getNextMode(const std::string& theAttribute) const
  85 {
  86   PartSet_SelectionMode aMode;
  87
  88   if (theAttribute == ARC_ATTR_CENTER)
  89     aMode = SM_SecondPoint;
  90   else if (theAttribute == ARC_ATTR_START)
  91     aMode = SM_ThirdPoint;
  92   else if (theAttribute == ARC_ATTR_END)
  93     aMode = SM_DonePoint;
  94   return aMode;
  95 }
  96
  97 void PartSet_FeatureArcPrs::projectPointOnArc(gp_Pnt& thePoint, Handle(V3d_View) theView,
  98                                               double& theX, double& theY)
  99 {
 100   FeaturePtr aSketch = sketch();
 101   if (aSketch) {
 102     double aX, anY;
 103     PartSet_Tools::convertTo2D(thePoint, aSketch, theView, aX, anY);
 104
 105     // circle origin point and radius
 106     boost::shared_ptr<ModelAPI_Data> aData = feature()->data();
 107     boost::shared_ptr<GeomDataAPI_Point2D> aCenter = boost::dynamic_pointer_cast<GeomDataAPI_Point2D>
 108                                                               (aData->attribute(ARC_ATTR_CENTER));
 109     boost::shared_ptr<GeomDataAPI_Point2D> aStart = boost::dynamic_pointer_cast<GeomDataAPI_Point2D>
 110                                                               (aData->attribute(ARC_ATTR_START));
 111
 112     boost::shared_ptr<GeomAPI_Circ2d> aCirc(new GeomAPI_Circ2d(aCenter->pnt(), aStart->pnt()));
 113     boost::shared_ptr<GeomAPI_Pnt2d> aGeomPoint2d(new GeomAPI_Pnt2d(aX, anY));
 114     boost::shared_ptr<GeomAPI_Pnt2d> aPnt2d = aCirc->project(aGeomPoint2d);
 115     if (aPnt2d) {
 116       theX = aPnt2d->x();
 117       theY = aPnt2d->y();
 118     }
 119   }
 120 }
 121
 122 boost::shared_ptr<GeomDataAPI_Point2D> PartSet_FeatureArcPrs::featurePoint
 123                                                      (const PartSet_SelectionMode& theMode)
 124 {
 125   std::string aPointArg;
 126   switch (theMode)
 127   {
 128     case SM_FirstPoint:
 129       aPointArg = ARC_ATTR_CENTER;
 130       break;
 131     case SM_SecondPoint:
 132       aPointArg = ARC_ATTR_START;
 133       break;
 134     case SM_ThirdPoint:
 135       aPointArg = ARC_ATTR_END;
 136       break;
 137     default:
 138       break;
 139   }
 140   boost::shared_ptr<ModelAPI_Data> aData = feature()->data();
 141   boost::shared_ptr<GeomDataAPI_Point2D> aPoint = boost::dynamic_pointer_cast<GeomDataAPI_Point2D>
 142                                                               (aData->attribute(aPointArg));
 143   return aPoint;
 144 }