src/PartSet/PartSet_OperationSketch.cpp

   1 // File:        PartSet_OperationSketch.h
   2 // Created:     20 Apr 2014
   3 // Author:      Natalia ERMOLAEVA
   4
   5 #include <PartSet_OperationSketch.h>
   6
   7 #include <PartSet_OperationEditLine.h>
   8 #include <PartSet_Tools.h>
   9
  10 #include <SketchPlugin_Sketch.h>
  11
  12 #include <ModelAPI_Data.h>
  13 #include <ModelAPI_AttributeDouble.h>
  14 #include <ModelAPI_AttributeRefList.h>
  15
  16 #include <GeomAlgoAPI_FaceBuilder.h>
  17 #include <GeomDataAPI_Point.h>
  18 #include <GeomDataAPI_Dir.h>
  19
  20 #include <XGUI_ViewerPrs.h>
  21
  22 #include <AIS_Shape.hxx>
  23 #include <AIS_ListOfInteractive.hxx>
  24 #include <V3d_View.hxx>
  25
  26 #ifdef _DEBUG
  27 #include <QDebug>
  28 #endif
  29
  30 #include <QMouseEvent>
  31
  32 using namespace std;
  33
  34 PartSet_OperationSketch::PartSet_OperationSketch(const QString& theId,
  35                                                      QObject* theParent)
  36 : PartSet_OperationSketchBase(theId, theParent), myIsEditMode(false)
  37 {
  38 }
  39
  40 PartSet_OperationSketch::~PartSet_OperationSketch()
  41 {
  42 }
  43
  44 std::list<int> PartSet_OperationSketch::getSelectionModes(boost::shared_ptr<ModelAPI_Feature> theFeature) const
  45 {
  46   std::list<int> aModes;
  47   if (!myIsEditMode)
  48     aModes.push_back(TopAbs_FACE);
  49   else {
  50     aModes.push_back(TopAbs_VERTEX);
  51     aModes.push_back(TopAbs_EDGE);
  52   }
  53   return aModes;
  54 }
  55
  56 void PartSet_OperationSketch::mousePressed(QMouseEvent* theEvent, Handle_V3d_View theView,
  57                                            const std::list<XGUI_ViewerPrs>& theSelected)
  58 {
  59   myFeatures = theSelected;
  60 }
  61
  62 void PartSet_OperationSketch::mouseReleased(QMouseEvent* theEvent, Handle_V3d_View theView,
  63                                             const std::list<XGUI_ViewerPrs>& theSelected)
  64 {
  65   if (theSelected.empty())
  66     return;
  67
  68   if (!myIsEditMode) {
  69     XGUI_ViewerPrs aPrs = theSelected.front();
  70     const TopoDS_Shape& aShape = aPrs.shape();
  71     if (!aShape.IsNull()) {
  72       setSketchPlane(aShape);
  73       myIsEditMode = true;
  74     }
  75   }
  76   else {
  77     if (theSelected.size() == 1) {
  78       boost::shared_ptr<ModelAPI_Feature> aFeature = theSelected.front().feature();
  79       if (aFeature)
  80         emit launchOperation(PartSet_OperationEditLine::Type(), aFeature);
  81     }
  82   }
  83   myFeatures.clear();
  84 }
  85
  86 void PartSet_OperationSketch::mouseMoved(QMouseEvent* theEvent, Handle(V3d_View) theView)
  87 {
  88   if (!myIsEditMode || !(theEvent->buttons() &  Qt::LeftButton) || myFeatures.empty())
  89     return;
  90
  91   if (myFeatures.size() != 1) {
  92     boost::shared_ptr<ModelAPI_Feature> aFeature = PartSet_Tools::NearestFeature(theEvent->pos(),
  93                                                                 theView, feature(), myFeatures);
  94     if (aFeature)
  95       emit launchOperation(PartSet_OperationEditLine::Type(), aFeature);
  96   }
  97 }
  98
  99 std::map<boost::shared_ptr<ModelAPI_Feature>, boost::shared_ptr<GeomAPI_Shape> >
 100                                                         PartSet_OperationSketch::preview() const
 101 {
 102   std::map<boost::shared_ptr<ModelAPI_Feature>, boost::shared_ptr<GeomAPI_Shape> > aPreviewMap;
 103
 104   boost::shared_ptr<SketchPlugin_Feature> aFeature;
 105
 106   boost::shared_ptr<ModelAPI_Data> aData = feature()->data();
 107   boost::shared_ptr<ModelAPI_AttributeRefList> aRefList =
 108         boost::dynamic_pointer_cast<ModelAPI_AttributeRefList>(aData->attribute(SKETCH_ATTR_FEATURES));
 109
 110   std::list<boost::shared_ptr<ModelAPI_Feature> > aFeatures = aRefList->list();
 111   std::list<boost::shared_ptr<ModelAPI_Feature> >::const_iterator anIt = aFeatures.begin(),
 112                                                                   aLast = aFeatures.end();
 113   for (; anIt != aLast; anIt++) {
 114     aFeature = boost::dynamic_pointer_cast<SketchPlugin_Feature>(*anIt);
 115     boost::shared_ptr<GeomAPI_Shape> aPreview = aFeature->preview();
 116     if (aPreview)
 117       aPreviewMap[aFeature] = aPreview;
 118   }
 119   return aPreviewMap;
 120 }
 121
 122 void PartSet_OperationSketch::setSketchPlane(const TopoDS_Shape& theShape)
 123 {
 124   if (theShape.IsNull())
 125     return;
 126
 127   // get selected shape
 128   boost::shared_ptr<GeomAPI_Shape> aGShape(new GeomAPI_Shape);
 129   aGShape->setImpl(new TopoDS_Shape(theShape));
 130
 131   // get plane parameters
 132   boost::shared_ptr<GeomAPI_Pln> aPlane = GeomAlgoAPI_FaceBuilder::plane(aGShape);
 133
 134   // set plane parameters to feature
 135   boost::shared_ptr<ModelAPI_Data> aData = feature()->data();
 136   double anA, aB, aC, aD;
 137   aPlane->coefficients(anA, aB, aC, aD);
 138
 139   boost::shared_ptr<ModelAPI_AttributeDouble> anAttr;
 140   /*
 141   aData->real(SKETCH_ATTR_PLANE_A)->setValue(anA);
 142   aData->real(SKETCH_ATTR_PLANE_B)->setValue(aB);
 143   aData->real(SKETCH_ATTR_PLANE_C)->setValue(aC);
 144   aData->real(SKETCH_ATTR_PLANE_D)->setValue(aD);
 145   */
 146   // temporary solution for main planes only
 147   boost::shared_ptr<GeomDataAPI_Point> anOrigin =
 148     boost::dynamic_pointer_cast<GeomDataAPI_Point>(aData->attribute(SKETCH_ATTR_ORIGIN));
 149   anOrigin->setValue(0, 0, 0);
 150   boost::shared_ptr<GeomDataAPI_Dir> aNormal =
 151     boost::dynamic_pointer_cast<GeomDataAPI_Dir>(aData->attribute(SKETCH_ATTR_NORM));
 152   aNormal->setValue(anA, aB, aC);
 153   boost::shared_ptr<GeomDataAPI_Dir> aDirX =
 154     boost::dynamic_pointer_cast<GeomDataAPI_Dir>(aData->attribute(SKETCH_ATTR_DIRX));
 155   aDirX->setValue(aB, aC, anA);
 156   boost::shared_ptr<GeomDataAPI_Dir> aDirY =
 157     boost::dynamic_pointer_cast<GeomDataAPI_Dir>(aData->attribute(SKETCH_ATTR_DIRY));
 158   aDirY->setValue(aC, anA, aB);
 159   boost::shared_ptr<GeomAPI_Dir> aDir = aPlane->direction();
 160   emit planeSelected(aDir->x(), aDir->y(), aDir->z());
 161 }