src/PartSet/PartSet_OperationSketchLine.cpp

   1 // File:        PartSet_OperationSketchLine.h
   2 // Created:     20 Apr 2014
   3 // Author:      Natalia ERMOLAEVA
   4
   5 #include <PartSet_OperationSketchLine.h>
   6
   7 #include <SketchPlugin_Feature.h>
   8 #include <GeomDataAPI_Point2D.h>
   9 #include <GeomDataAPI_Point.h>
  10 #include <GeomDataAPI_Dir.h>
  11 #include <ModelAPI_Data.h>
  12 #include <ModelAPI_Document.h>
  13
  14 #include <SketchPlugin_Sketch.h>
  15 #include <SketchPlugin_Line.h>
  16
  17 #ifdef _DEBUG
  18 #include <QDebug>
  19 #endif
  20
  21 using namespace std;
  22
  23 PartSet_OperationSketchLine::PartSet_OperationSketchLine(const QString& theId,
  24                                                   QObject* theParent,
  25                                               boost::shared_ptr<ModelAPI_Feature> theFeature)
  26 : PartSet_OperationSketchBase(theId, theParent), mySketch(theFeature),
  27   myPointSelectionMode(SM_FirstPoint)
  28 {
  29 }
  30
  31 PartSet_OperationSketchLine::~PartSet_OperationSketchLine()
  32 {
  33 }
  34
  35 bool PartSet_OperationSketchLine::isGranted() const
  36 {
  37   return true;
  38 }
  39
  40 int PartSet_OperationSketchLine::getSelectionMode(boost::shared_ptr<ModelAPI_Feature> theFeature) const
  41 {
  42   int aMode = 0;
  43   if (theFeature != feature())
  44     aMode = TopAbs_VERTEX;
  45   return aMode;
  46 }
  47
  48 void PartSet_OperationSketchLine::mouseReleased(const gp_Pnt& thePoint)
  49 {
  50   switch (myPointSelectionMode)
  51   {
  52     case SM_FirstPoint: {
  53       setLinePoint(thePoint, LINE_ATTR_START);
  54       myPointSelectionMode = SM_SecondPoint;
  55     }
  56     break;
  57     case SM_SecondPoint: {
  58       setLinePoint(thePoint, LINE_ATTR_END);
  59       myPointSelectionMode = SM_None;
  60     }
  61     break;
  62     case SM_None: {
  63
  64     }
  65     break;
  66     default:
  67       break;
  68   }
  69 }
  70
  71 void PartSet_OperationSketchLine::mouseMoved(const gp_Pnt& thePoint)
  72 {
  73   switch (myPointSelectionMode)
  74   {
  75     case SM_SecondPoint:
  76       setLinePoint(thePoint, LINE_ATTR_END);
  77       break;
  78     case SM_None: {
  79       boost::shared_ptr<ModelAPI_Feature> aPrevFeature = feature();
  80       // stop the last operation
  81       commitOperation();
  82       document()->finishOperation();
  83       //emit changeSelectionMode(aPrevFeature, TopAbs_VERTEX);
  84       // start a new operation
  85       document()->startOperation();
  86       startOperation();
  87       // use the last point of the previous feature as the first of the new one
  88       setLinePoint(aPrevFeature, LINE_ATTR_END, LINE_ATTR_START);
  89       myPointSelectionMode = SM_SecondPoint;
  90
  91       emit featureConstructed(aPrevFeature);
  92     }
  93     break;
  94     default:
  95       break;
  96   }
  97 }
  98
  99 void PartSet_OperationSketchLine::keyReleased(const int theKey)
 100 {
 101   switch (theKey) {
 102     case Qt::Key_Escape:
 103       abort();
 104       break;
 105     case Qt::Key_Enter:
 106       //myPointSelectionMode = myPointSelectionMode;
 107       break;
 108     default:
 109       break;
 110   }
 111 }
 112
 113 void PartSet_OperationSketchLine::startOperation()
 114 {
 115   PartSet_OperationSketchBase::startOperation();
 116
 117   if (mySketch) {
 118     boost::shared_ptr<SketchPlugin_Feature> aFeature =
 119                            boost::dynamic_pointer_cast<SketchPlugin_Feature>(mySketch);
 120
 121     aFeature->addSub(feature());
 122   }
 123   myPointSelectionMode = SM_FirstPoint;
 124 }
 125
 126 void PartSet_OperationSketchLine::stopOperation()
 127 {
 128   PartSet_OperationSketchBase::stopOperation();
 129   myPointSelectionMode = SM_None;
 130 }
 131
 132 void PartSet_OperationSketchLine::setLinePoint(const gp_Pnt& thePoint,
 133                                                const std::string& theAttribute)
 134 {
 135   boost::shared_ptr<ModelAPI_Data> aData = feature()->data();
 136   boost::shared_ptr<GeomDataAPI_Point2D> aPoint =
 137         boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(theAttribute));
 138
 139   double aX = 0;
 140   double anY = 0;
 141   convertTo2D(thePoint, aX, anY);
 142   aPoint->setValue(aX, anY);
 143 }
 144
 145 void PartSet_OperationSketchLine::setLinePoint(boost::shared_ptr<ModelAPI_Feature> theSourceFeature,
 146                                                const std::string& theSourceAttribute,
 147                                                const std::string& theAttribute)
 148 {
 149   boost::shared_ptr<ModelAPI_Data> aData = theSourceFeature->data();
 150   boost::shared_ptr<GeomDataAPI_Point2D> aPoint =
 151         boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(theSourceAttribute));
 152   double aX = aPoint->x();
 153   double anY = aPoint->y();
 154
 155   aData = feature()->data();
 156   aPoint = boost::dynamic_pointer_cast<GeomDataAPI_Point2D>(aData->attribute(theAttribute));
 157   aPoint->setValue(aX, anY);
 158 }
 159
 160 void PartSet_OperationSketchLine::convertTo2D(const gp_Pnt& thePoint, double& theX, double& theY)
 161 {
 162   if (!mySketch)
 163     return;
 164
 165   boost::shared_ptr<ModelAPI_AttributeDouble> anAttr;
 166   boost::shared_ptr<ModelAPI_Data> aData = mySketch->data();
 167
 168   boost::shared_ptr<GeomDataAPI_Point> anOrigin =
 169     boost::dynamic_pointer_cast<GeomDataAPI_Point>(aData->attribute(SKETCH_ATTR_ORIGIN));
 170
 171   boost::shared_ptr<GeomDataAPI_Dir> aX =
 172     boost::dynamic_pointer_cast<GeomDataAPI_Dir>(aData->attribute(SKETCH_ATTR_DIRX));
 173   boost::shared_ptr<GeomDataAPI_Dir> anY =
 174     boost::dynamic_pointer_cast<GeomDataAPI_Dir>(aData->attribute(SKETCH_ATTR_DIRY));
 175
 176   gp_Pnt aVec(thePoint.X() - anOrigin->x(), thePoint.Y() - anOrigin->y(), thePoint.Z() - anOrigin->z());
 177   theX = aVec.X() * aX->x() + aVec.Y() * aX->y() + aVec.Z() * aX->z();
 178   theY = aVec.X() * anY->x() + aVec.Y() * anY->y() + aVec.Z() * anY->z();
 179 }