src/SketcherPrs/SketcherPrs_LengthDimension.cpp

   1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
   2
   3 // File:        SketcherPrs_LengthDimension.cpp
   4 // Created:     27 March 2015
   5 // Author:      Vitaly SMETANNIKOV
   6
   7 #include "SketcherPrs_LengthDimension.h"
   8 #include "SketcherPrs_Tools.h"
   9
  10 #include <SketchPlugin_Constraint.h>
  11 #include <SketchPlugin_ConstraintLength.h>
  12 #include <SketchPlugin_ConstraintDistance.h>
  13 #include <SketchPlugin_Line.h>
  14 #include <SketchPlugin_Point.h>
  15 #include <SketchPlugin_Circle.h>
  16
  17 #include <GeomDataAPI_Point2D.h>
  18 #include <GeomAPI_Pnt.h>
  19 #include <GeomAPI_XYZ.h>
  20 #include <GeomAPI_Pnt2d.h>
  21 #include <GeomAPI_Lin2d.h>
  22
  23 #include <ModelAPI_Data.h>
  24
  25
  26 const int CONSTRAINT_TEXT_HEIGHT = 20;  /// the text height of the constraint
  27 const int CONSTRAINT_TEXT_SELECTION_TOLERANCE = 20;  /// the text selection tolerance
  28
  29
  30 static const gp_Pnt MyDefStart(0,0,0);
  31 static const gp_Pnt MyDefEnd(1,0,0);
  32 static const gp_Pln MyDefPln(gp_Pnt(0,0,0), gp_Dir(0,0,1));
  33
  34
  35
  36
  37 IMPLEMENT_STANDARD_HANDLE(SketcherPrs_LengthDimension, AIS_LengthDimension);
  38 IMPLEMENT_STANDARD_RTTIEXT(SketcherPrs_LengthDimension, AIS_LengthDimension);
  39
  40 SketcherPrs_LengthDimension::SketcherPrs_LengthDimension(SketchPlugin_Constraint* theConstraint,
  41                         const std::shared_ptr<GeomAPI_Ax3>& thePlane)
  42 : AIS_LengthDimension(MyDefStart, MyDefEnd, MyDefPln),
  43 myConstraint(theConstraint), myPlane(thePlane)
  44 {
  45   myAspect = new Prs3d_DimensionAspect();
  46   myAspect->MakeArrows3d(false);
  47   myAspect->MakeText3d(false);
  48   myAspect->MakeTextShaded(false);
  49   myAspect->MakeUnitsDisplayed(false);
  50
  51   SetDimensionAspect(myAspect);
  52   SetSelToleranceForText2d(CONSTRAINT_TEXT_SELECTION_TOLERANCE);
  53 }
  54
  55
  56 void SketcherPrs_LengthDimension::Compute(const Handle(PrsMgr_PresentationManager3d)& thePresentationManager,
  57                                  const Handle(Prs3d_Presentation)& thePresentation,
  58                                  const Standard_Integer theMode)
  59 {
  60   gp_Pnt aPnt1, aPnt2;
  61   if (!getPoints(aPnt1, aPnt2))
  62     return;
  63   DataPtr aData = myConstraint->data();
  64
  65   AttributePtr aFlyOutAttribute = aData->attribute(SketchPlugin_Constraint::FLYOUT_VALUE_PNT());
  66   if (!aFlyOutAttribute->isInitialized()) {
  67     return; // not possible to show length because points are not defined
  68   }
  69   std::shared_ptr<GeomDataAPI_Point2D> aFlyOutAttr =
  70     std::dynamic_pointer_cast<GeomDataAPI_Point2D>(aFlyOutAttribute);
  71   std::shared_ptr<GeomAPI_Pnt> aFlyoutPnt = myPlane->to3D(aFlyOutAttr->x(), aFlyOutAttr->y());
  72   gp_Pnt aFlyPnt = aFlyoutPnt->impl<gp_Pnt>();
  73
  74   double aDistance = aPnt1.Distance(aPnt2);
  75
  76   double aFlyout = 0;
  77   double aDist = 0.0;
  78   if (aDistance < Precision::Confusion())
  79     aDist = aPnt1.Distance(aFlyPnt);
  80   else {
  81     gp_Lin aLine(aPnt1, gp_Vec(aPnt1, aPnt2));
  82     aDist = aLine.Distance(aFlyPnt);
  83   }
  84
  85   gp_XYZ aLineDir = aPnt2.XYZ().Subtracted(aPnt1.XYZ());
  86   gp_XYZ aFOutDir = aFlyPnt.XYZ().Subtracted(aPnt1.XYZ());
  87   gp_XYZ aNorm = myPlane->norm()->xyz()->impl<gp_XYZ>();
  88   if (aLineDir.Crossed(aFOutDir).Dot(aNorm) < 0)
  89     aDist = -aDist;
  90   aFlyout = aDist;
  91
  92   SetFlyout(aFlyout);
  93   myAspect->TextAspect()->SetHeight(CONSTRAINT_TEXT_HEIGHT);
  94   myAspect->ArrowAspect()->SetLength(CONSTRAINT_TEXT_HEIGHT * 2);
  95
  96   SetMeasuredGeometry(aPnt1, aPnt2, myPlane->impl<gp_Ax3>());
  97   AIS_LengthDimension::Compute(thePresentationManager, thePresentation, theMode);
  98 }
  99
 100 bool SketcherPrs_LengthDimension::getPoints(gp_Pnt& thePnt1, gp_Pnt& thePnt2) const
 101 {
 102   DataPtr aData = myConstraint->data();
 103   if (myConstraint->getKind() == SketchPlugin_ConstraintLength::ID()) {
 104     std::shared_ptr<ModelAPI_AttributeRefAttr> anAttr =
 105       std::dynamic_pointer_cast<ModelAPI_AttributeRefAttr>
 106       (aData->attribute(SketchPlugin_Constraint::ENTITY_A()));
 107     if (!anAttr)
 108       return false;
 109
 110     FeaturePtr aFeature = ModelAPI_Feature::feature(anAttr->object());
 111     if (!aFeature || aFeature->getKind() != SketchPlugin_Line::ID())
 112       return false;
 113
 114     DataPtr aLineData = aFeature->data();
 115     std::shared_ptr<GeomDataAPI_Point2D> aStartPoint =
 116       std::dynamic_pointer_cast<GeomDataAPI_Point2D>
 117       (aLineData->attribute(SketchPlugin_Line::START_ID()));
 118     std::shared_ptr<GeomDataAPI_Point2D> aEndPoint =
 119       std::dynamic_pointer_cast<GeomDataAPI_Point2D>
 120       (aLineData->attribute(SketchPlugin_Line::END_ID()));
 121     thePnt1 = myPlane->to3D(aStartPoint->x(), aStartPoint->y())->impl<gp_Pnt>();
 122     thePnt2 = myPlane->to3D(aEndPoint->x(), aEndPoint->y())->impl<gp_Pnt>();
 123     return true;
 124
 125   } else if (myConstraint->getKind() == SketchPlugin_ConstraintDistance::ID()) {
 126     std::shared_ptr<GeomDataAPI_Point2D> aPoint_A = SketcherPrs_Tools::getFeaturePoint(
 127         aData, SketchPlugin_Constraint::ENTITY_A());
 128     std::shared_ptr<GeomDataAPI_Point2D> aPoint_B = SketcherPrs_Tools::getFeaturePoint(
 129         aData, SketchPlugin_Constraint::ENTITY_B());
 130
 131     std::shared_ptr<GeomAPI_Pnt2d> aPnt_A;
 132     std::shared_ptr<GeomAPI_Pnt2d> aPnt_B;
 133     if (aPoint_A && aPoint_B) {
 134       aPnt_A = aPoint_A->pnt();
 135       aPnt_B = aPoint_B->pnt();
 136     } else if (!aPoint_A && aPoint_B) {
 137       std::shared_ptr<SketchPlugin_Line> aLine = SketcherPrs_Tools::getFeatureLine(
 138           aData, SketchPlugin_Constraint::ENTITY_A());
 139       if (aLine) {
 140         aPnt_B = aPoint_B->pnt();
 141         aPnt_A = SketcherPrs_Tools::getProjectionPoint(aLine, aPnt_B);
 142       }
 143     } else if (aPoint_A && !aPoint_B) {
 144       std::shared_ptr<SketchPlugin_Line> aLine = SketcherPrs_Tools::getFeatureLine(
 145           aData, SketchPlugin_Constraint::ENTITY_B());
 146       if (aLine) {
 147         aPnt_A = aPoint_A->pnt();
 148         aPnt_B = SketcherPrs_Tools::getProjectionPoint(aLine, aPnt_A);
 149       }
 150     }
 151     if (!aPnt_A || !aPnt_B)
 152       return false;
 153
 154     std::shared_ptr<GeomAPI_Pnt> aPoint1 = myPlane->to3D(aPnt_A->x(), aPnt_A->y());
 155     std::shared_ptr<GeomAPI_Pnt> aPoint2 = myPlane->to3D(aPnt_B->x(), aPnt_B->y());
 156     thePnt1 = aPoint1->impl<gp_Pnt>();
 157     thePnt2 = aPoint2->impl<gp_Pnt>();
 158     return true;
 159   }
 160   return false;
 161 }