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::projectPointOnFeature(FeaturePtr theFeature, FeaturePtr theSketch,
  98                                                   gp_Pnt& thePoint, Handle(V3d_View) theView,
  99                                                   double& theX, double& theY)
 100 {
 101   FeaturePtr aSketch = theSketch;
 102   if (aSketch) {
 103     double aX, anY;
 104     PartSet_Tools::convertTo2D(thePoint, aSketch, theView, aX, anY);
 105
 106     // circle origin point and radius
 107     boost::shared_ptr<ModelAPI_Data> aData = theFeature->data();
 108     boost::shared_ptr<GeomDataAPI_Point2D> aCenter = boost::dynamic_pointer_cast<GeomDataAPI_Point2D>
 109                                                               (aData->attribute(ARC_ATTR_CENTER));
 110     boost::shared_ptr<GeomDataAPI_Point2D> aStart = boost::dynamic_pointer_cast<GeomDataAPI_Point2D>
 111                                                               (aData->attribute(ARC_ATTR_START));
 112
 113     boost::shared_ptr<GeomAPI_Circ2d> aCirc(new GeomAPI_Circ2d(aCenter->pnt(), aStart->pnt()));
 114     boost::shared_ptr<GeomAPI_Pnt2d> aGeomPoint2d(new GeomAPI_Pnt2d(aX, anY));
 115     boost::shared_ptr<GeomAPI_Pnt2d> aPnt2d = aCirc->project(aGeomPoint2d);
 116     if (aPnt2d) {
 117       theX = aPnt2d->x();
 118       theY = aPnt2d->y();
 119     }
 120   }
 121 }
 122
 123 boost::shared_ptr<GeomDataAPI_Point2D> PartSet_FeatureArcPrs::featurePoint
 124                                                      (const PartSet_SelectionMode& theMode)
 125 {
 126   std::string aPointArg;
 127   switch (theMode)
 128   {
 129     case SM_FirstPoint:
 130       aPointArg = ARC_ATTR_CENTER;
 131       break;
 132     case SM_SecondPoint:
 133       aPointArg = ARC_ATTR_START;
 134       break;
 135     case SM_ThirdPoint:
 136       aPointArg = ARC_ATTR_END;
 137       break;
 138     default:
 139       break;
 140   }
 141   boost::shared_ptr<ModelAPI_Data> aData = feature()->data();
 142   boost::shared_ptr<GeomDataAPI_Point2D> aPoint = boost::dynamic_pointer_cast<GeomDataAPI_Point2D>
 143                                                               (aData->attribute(aPointArg));
 144   return aPoint;
 145 }
 146
 147 double PartSet_FeatureArcPrs::radius(FeaturePtr theFeature)
 148 {
 149   if (!theFeature)
 150     return 0;
 151
 152   boost::shared_ptr<ModelAPI_Data> aData = theFeature->data();
 153
 154   boost::shared_ptr<GeomDataAPI_Point2D> aCenterAttr =
 155     boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(ARC_ATTR_CENTER));
 156   boost::shared_ptr<GeomDataAPI_Point2D> aStartAttr =
 157     boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(ARC_ATTR_START));
 158
 159   return aCenterAttr->pnt()->distance(aStartAttr->pnt());
 160 }