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 double PartSet_FeatureArcPrs::distanceToPoint(FeaturePtr theFeature,
  98                                                double theX, double theY)
  99 {
 100   double aDelta = 0;
 101   if (!theFeature || theFeature->getKind() != getKind())
 102     return aDelta;
 103   boost::shared_ptr<GeomAPI_Pnt2d> aPoint2d(new GeomAPI_Pnt2d(theX, theY));
 104
 105
 106   boost::shared_ptr<ModelAPI_Data> aData = theFeature->data();
 107
 108   boost::shared_ptr<GeomDataAPI_Point2D> aPoint1 =
 109         boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(ARC_ATTR_CENTER));
 110   aDelta = aPoint1->pnt()->distance(aPoint2d);
 111
 112   boost::shared_ptr<GeomDataAPI_Point2D> aPoint2 =
 113         boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(ARC_ATTR_START));
 114   aDelta = qMin(aDelta, aPoint2->pnt()->distance(aPoint2d));
 115
 116   boost::shared_ptr<GeomDataAPI_Point2D> aPoint3 =
 117         boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(ARC_ATTR_END));
 118   aDelta = qMin(aDelta, aPoint3->pnt()->distance(aPoint2d));
 119
 120   return aDelta;
 121 }
 122
 123 boost::shared_ptr<GeomDataAPI_Point2D> PartSet_FeatureArcPrs::findPoint(FeaturePtr theFeature,
 124                                                                         double theX, double theY)
 125 {
 126   boost::shared_ptr<GeomDataAPI_Point2D> aPoint2D;
 127   if (!theFeature || theFeature->getKind() != getKind())
 128     return aPoint2D;
 129
 130   boost::shared_ptr<ModelAPI_Data> aData = theFeature->data();
 131   boost::shared_ptr<GeomDataAPI_Point2D> aPoint =
 132         boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(ARC_ATTR_CENTER));
 133   if (fabs(aPoint->x() - theX) < Precision::Confusion() &&
 134       fabs(aPoint->y() - theY) < Precision::Confusion()) {
 135     aPoint2D = aPoint;
 136   }
 137   if (!aPoint2D) {
 138     aPoint = boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(ARC_ATTR_START));
 139     if (fabs(aPoint->x() - theX) < Precision::Confusion() &&
 140         fabs(aPoint->y() - theY) < Precision::Confusion())
 141       aPoint2D = aPoint;
 142   }
 143   if (!aPoint2D) {
 144     aPoint = boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(ARC_ATTR_END));
 145     if (fabs(aPoint->x() - theX) < Precision::Confusion() &&
 146         fabs(aPoint->y() - theY) < Precision::Confusion())
 147       aPoint2D = aPoint;
 148   }
 149   return aPoint2D;
 150 }
 151
 152 void PartSet_FeatureArcPrs::projectPointOnFeature(FeaturePtr theFeature, FeaturePtr theSketch,
 153                                                   gp_Pnt& thePoint, Handle(V3d_View) theView,
 154                                                   double& theX, double& theY)
 155 {
 156   FeaturePtr aSketch = theSketch;
 157   if (aSketch) {
 158     double aX, anY;
 159     PartSet_Tools::convertTo2D(thePoint, aSketch, theView, aX, anY);
 160
 161     // circle origin point and radius
 162     boost::shared_ptr<ModelAPI_Data> aData = theFeature->data();
 163     boost::shared_ptr<GeomDataAPI_Point2D> aCenter = boost::dynamic_pointer_cast<GeomDataAPI_Point2D>
 164                                                               (aData->attribute(ARC_ATTR_CENTER));
 165     boost::shared_ptr<GeomDataAPI_Point2D> aStart = boost::dynamic_pointer_cast<GeomDataAPI_Point2D>
 166                                                               (aData->attribute(ARC_ATTR_START));
 167
 168     boost::shared_ptr<GeomAPI_Circ2d> aCirc(new GeomAPI_Circ2d(aCenter->pnt(), aStart->pnt()));
 169     boost::shared_ptr<GeomAPI_Pnt2d> aGeomPoint2d(new GeomAPI_Pnt2d(aX, anY));
 170     boost::shared_ptr<GeomAPI_Pnt2d> aPnt2d = aCirc->project(aGeomPoint2d);
 171     if (aPnt2d) {
 172       theX = aPnt2d->x();
 173       theY = aPnt2d->y();
 174     }
 175   }
 176 }
 177
 178 boost::shared_ptr<GeomDataAPI_Point2D> PartSet_FeatureArcPrs::featurePoint
 179                                                      (const PartSet_SelectionMode& theMode)
 180 {
 181   std::string aPointArg;
 182   switch (theMode)
 183   {
 184     case SM_FirstPoint:
 185       aPointArg = ARC_ATTR_CENTER;
 186       break;
 187     case SM_SecondPoint:
 188       aPointArg = ARC_ATTR_START;
 189       break;
 190     case SM_ThirdPoint:
 191       aPointArg = ARC_ATTR_END;
 192       break;
 193     default:
 194       break;
 195   }
 196   boost::shared_ptr<ModelAPI_Data> aData = feature()->data();
 197   boost::shared_ptr<GeomDataAPI_Point2D> aPoint = boost::dynamic_pointer_cast<GeomDataAPI_Point2D>
 198                                                               (aData->attribute(aPointArg));
 199   return aPoint;
 200 }
 201
 202 double PartSet_FeatureArcPrs::radius(FeaturePtr theFeature)
 203 {
 204   if (!theFeature)
 205     return 0;
 206
 207   boost::shared_ptr<ModelAPI_Data> aData = theFeature->data();
 208
 209   boost::shared_ptr<GeomDataAPI_Point2D> aCenterAttr =
 210     boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(ARC_ATTR_CENTER));
 211   boost::shared_ptr<GeomDataAPI_Point2D> aStartAttr =
 212     boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(ARC_ATTR_START));
 213
 214   return aCenterAttr->pnt()->distance(aStartAttr->pnt());
 215 }