src/SketchPlugin/SketchPlugin_Projection.cpp

   1 // Copyright (C) 2014-2017  CEA/DEN, EDF R&D
   2 //
   3 // This library is free software; you can redistribute it and/or
   4 // modify it under the terms of the GNU Lesser General Public
   5 // License as published by the Free Software Foundation; either
   6 // version 2.1 of the License, or (at your option) any later version.
   7 //
   8 // This library is distributed in the hope that it will be useful,
   9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 // Lesser General Public License for more details.
  12 //
  13 // You should have received a copy of the GNU Lesser General Public
  14 // License along with this library; if not, write to the Free Software
  15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  16 //
  17 // See http://www.salome-platform.org/ or
  18 // email : webmaster.salome@opencascade.com<mailto:webmaster.salome@opencascade.com>
  19 //
  20
  21 #include <SketchPlugin_Projection.h>
  22
  23 #include <SketchPlugin_Arc.h>
  24 #include <SketchPlugin_Circle.h>
  25 #include <SketchPlugin_Line.h>
  26 #include <SketchPlugin_Point.h>
  27 #include <SketchPlugin_Sketch.h>
  28 #include <SketchPlugin_ConstraintRigid.h>
  29
  30 #include <ModelAPI_AttributeRefAttr.h>
  31 #include <ModelAPI_AttributeSelection.h>
  32 #include <ModelAPI_AttributeDouble.h>
  33 #include <ModelAPI_ResultConstruction.h>
  34 #include <ModelAPI_Session.h>
  35 #include <ModelAPI_Validator.h>
  36 #include <ModelAPI_Tools.h>
  37
  38 #include <GeomAPI_Circ.h>
  39 #include <GeomAPI_Edge.h>
  40 #include <GeomAPI_Lin.h>
  41 #include <GeomAPI_Pnt.h>
  42 #include <GeomAPI_Pnt2d.h>
  43 #include <GeomAPI_Vertex.h>
  44 #include <GeomAlgoAPI_EdgeBuilder.h>
  45 #include <GeomDataAPI_Point2D.h>
  46
  47 #include <cmath>
  48
  49 static const double tolerance = 1.e-7;
  50
  51 SketchPlugin_Projection::SketchPlugin_Projection()
  52     : SketchPlugin_SketchEntity(),
  53       myIsComputing(false)
  54 {
  55 }
  56
  57 void SketchPlugin_Projection::initDerivedClassAttributes()
  58 {
  59   data()->addAttribute(EXTERNAL_FEATURE_ID(), ModelAPI_AttributeSelection::typeId());
  60   data()->addAttribute(PROJECTED_FEATURE_ID(), ModelAPI_AttributeRefAttr::typeId());
  61   data()->attribute(PROJECTED_FEATURE_ID())->setIsArgument(false);
  62
  63   data()->addAttribute(EXTERNAL_ID(), ModelAPI_AttributeSelection::typeId());
  64   ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), EXTERNAL_ID());
  65
  66   data()->addAttribute(INCLUDE_INTO_RESULT(), ModelAPI_AttributeBoolean::typeId());
  67
  68   ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), AUXILIARY_ID());
  69 }
  70
  71 void SketchPlugin_Projection::execute()
  72 {
  73   AttributeRefAttrPtr aRefAttr = data()->refattr(PROJECTED_FEATURE_ID());
  74   if (!aRefAttr || !aRefAttr->isInitialized())
  75     return;
  76   FeaturePtr aProjection = ModelAPI_Feature::feature(aRefAttr->object());
  77
  78   if (!lastResult().get() && aProjection->lastResult().get()) {
  79     ResultConstructionPtr aConstr = document()->createConstruction(data());
  80     aConstr->setShape(aProjection->lastResult()->shape());
  81     aConstr->setIsInHistory(false);
  82     aConstr->setDisplayed(false);
  83     setResult(aConstr);
  84
  85     aProjection->selection(EXTERNAL_ID())->setValue(lastResult(), lastResult()->shape());
  86   }
  87
  88   // is sketch plane is changed (issue 1791), attribute of projection is not changed, but
  89   // projection must be fully recomputed
  90   computeProjection(EXTERNAL_FEATURE_ID());
  91 }
  92
  93 void SketchPlugin_Projection::attributeChanged(const std::string& theID)
  94 {
  95   if ((theID == EXTERNAL_FEATURE_ID() || theID == EXTERNAL_ID()) && !myIsComputing) {
  96     myIsComputing = true;
  97     computeProjection(theID);
  98     myIsComputing = false;
  99   }
 100 }
 101
 102 static bool isValidProjectionType(FeaturePtr theProjection,
 103                                   GeomEdgePtr theEdge,
 104                                   GeomVertexPtr theVertex)
 105 {
 106   if (theVertex && theProjection->getKind() == SketchPlugin_Point::ID())
 107     return true;
 108   if (theEdge) {
 109     if (theEdge->isLine() && theProjection->getKind() == SketchPlugin_Line::ID())
 110       return true;
 111     else if (theEdge->isCircle() && theProjection->getKind() == SketchPlugin_Circle::ID())
 112       return true;
 113     else if (theEdge->isArc() && theProjection->getKind() == SketchPlugin_Arc::ID())
 114       return true;
 115   }
 116   return false;
 117 }
 118
 119 void SketchPlugin_Projection::computeProjection(const std::string& theID)
 120 {
 121   AttributeSelectionPtr aExtFeature =
 122       std::dynamic_pointer_cast<ModelAPI_AttributeSelection>(attribute(EXTERNAL_FEATURE_ID()));
 123
 124   GeomShapePtr aShape;
 125   GeomEdgePtr anEdge;
 126   GeomVertexPtr aVertex;
 127   if (aExtFeature)
 128     aShape = aExtFeature->value();
 129   if (!aShape && aExtFeature->context())
 130     aShape = aExtFeature->context()->shape();
 131   if (aShape) {
 132     if (aShape->isEdge())
 133       anEdge = GeomEdgePtr(new GeomAPI_Edge(aShape));
 134     else if (aShape->isVertex())
 135       aVertex = GeomVertexPtr(new GeomAPI_Vertex(aShape));
 136   }
 137   if (!anEdge && !aVertex)
 138     return;
 139
 140   AttributeRefAttrPtr aRefAttr = data()->refattr(PROJECTED_FEATURE_ID());
 141   FeaturePtr aProjection;
 142   if (aRefAttr && aRefAttr->isInitialized())
 143     aProjection = ModelAPI_Feature::feature(aRefAttr->object());
 144
 145   // if the type of feature differs with already selected, remove it and create once again
 146   bool hasPrevProj = aProjection.get() != 0;
 147   if (hasPrevProj && !isValidProjectionType(aProjection, anEdge, aVertex)) {
 148     DocumentPtr aDoc = sketch()->document();
 149
 150     aRefAttr->setObject(data()->owner()); // to not remove of this remove reference to aProjection
 151     std::set<FeaturePtr> aFeaturesToBeRemoved;
 152     aFeaturesToBeRemoved.insert(aProjection);
 153     ModelAPI_Tools::removeFeaturesAndReferences(aFeaturesToBeRemoved);
 154     aProjection = FeaturePtr();
 155     aRefAttr->setObject(aProjection);
 156     hasPrevProj = false;
 157   }
 158
 159   std::shared_ptr<GeomAPI_Pln> aSketchPlane = sketch()->plane();
 160
 161   ResultConstructionPtr aResult =
 162       std::dynamic_pointer_cast<ModelAPI_ResultConstruction>(lastResult());
 163   if (aResult && aResult->shape() && theID == EXTERNAL_FEATURE_ID()) {
 164     aResult->setShape(std::shared_ptr<GeomAPI_Edge>());
 165     aProjection->selection(EXTERNAL_ID())->setValue(lastResult(), lastResult()->shape());
 166   }
 167
 168   if (aVertex) {
 169     std::shared_ptr<GeomAPI_Pnt> aPrjPnt = aSketchPlane->project(aVertex->point());
 170     std::shared_ptr<GeomAPI_Pnt2d> aPntInSketch = sketch()->to2D(aPrjPnt);
 171
 172     if (!hasPrevProj)
 173       aProjection = sketch()->addFeature(SketchPlugin_Point::ID());
 174
 175     // update coordinates of projection
 176     std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
 177         aProjection->attribute(SketchPlugin_Point::COORD_ID()))->setValue(aPntInSketch);
 178   }
 179   else if (anEdge->isLine()) {
 180     std::shared_ptr<GeomAPI_Pnt> aFirst = aSketchPlane->project(anEdge->firstPoint());
 181     std::shared_ptr<GeomAPI_Pnt> aLast = aSketchPlane->project(anEdge->lastPoint());
 182
 183     std::shared_ptr<GeomAPI_Pnt2d> aFirstInSketch = sketch()->to2D(aFirst);
 184     std::shared_ptr<GeomAPI_Pnt2d> aLastInSketch = sketch()->to2D(aLast);
 185     if (aFirstInSketch->distance(aLastInSketch) < tolerance)
 186       return; // line is semi-orthogonal to the sketch plane
 187
 188     if (!hasPrevProj)
 189       aProjection = sketch()->addFeature(SketchPlugin_Line::ID());
 190
 191     // update attributes of projection
 192     std::shared_ptr<GeomDataAPI_Point2D> aStartPnt = std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
 193         aProjection->attribute(SketchPlugin_Line::START_ID()));
 194     std::shared_ptr<GeomDataAPI_Point2D> aEndPnt = std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
 195         aProjection->attribute(SketchPlugin_Line::END_ID()));
 196     aStartPnt->setValue(aFirstInSketch);
 197     aEndPnt->setValue(aLastInSketch);
 198   }
 199   else if (anEdge->isCircle()) {
 200     std::shared_ptr<GeomAPI_Circ> aCircle = anEdge->circle();
 201     double aRadius = aCircle->radius();
 202
 203     std::shared_ptr<GeomAPI_Pnt> aCenter = aSketchPlane->project(aCircle->center());
 204     std::shared_ptr<GeomAPI_Pnt2d> aCenterInSketch = sketch()->to2D(aCenter);
 205
 206     if (!hasPrevProj)
 207       aProjection = sketch()->addFeature(SketchPlugin_Circle::ID());
 208
 209     // update attributes of projection
 210     std::shared_ptr<GeomDataAPI_Point2D> aCenterPnt =
 211       std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
 212         aProjection->attribute(SketchPlugin_Circle::CENTER_ID()));
 213     aCenterPnt->setValue(aCenterInSketch);
 214     aProjection->real(SketchPlugin_Circle::RADIUS_ID())->setValue(aRadius);
 215   }
 216   else if (anEdge->isArc()) {
 217     std::shared_ptr<GeomAPI_Pnt> aFirst = aSketchPlane->project(anEdge->firstPoint());
 218     std::shared_ptr<GeomAPI_Pnt> aLast = aSketchPlane->project(anEdge->lastPoint());
 219     std::shared_ptr<GeomAPI_Pnt2d> aFirstInSketch = sketch()->to2D(aFirst);
 220     std::shared_ptr<GeomAPI_Pnt2d> aLastInSketch = sketch()->to2D(aLast);
 221
 222     std::shared_ptr<GeomAPI_Circ> aCircle = anEdge->circle();
 223     std::shared_ptr<GeomAPI_Pnt> aCenter = aSketchPlane->project(aCircle->center());
 224     std::shared_ptr<GeomAPI_Pnt2d> aCenterInSketch = sketch()->to2D(aCenter);
 225
 226     bool isInversed = aCircle->normal()->dot(aSketchPlane->direction()) < 0.;
 227
 228     if (!hasPrevProj)
 229       aProjection = sketch()->addFeature(SketchPlugin_Arc::ID());
 230
 231     bool aWasBlocked = aProjection->data()->blockSendAttributeUpdated(true);
 232
 233     // update attributes of projection
 234     std::shared_ptr<GeomDataAPI_Point2D> aCenterPnt =
 235       std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
 236         aProjection->attribute(SketchPlugin_Arc::CENTER_ID()));
 237     std::shared_ptr<GeomDataAPI_Point2D> aStartPnt =
 238       std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
 239         aProjection->attribute(SketchPlugin_Arc::START_ID()));
 240     std::shared_ptr<GeomDataAPI_Point2D> aEndPnt =
 241       std::dynamic_pointer_cast<GeomDataAPI_Point2D>(
 242         aProjection->attribute(SketchPlugin_Arc::END_ID()));
 243     aStartPnt->setValue(aFirstInSketch);
 244     aEndPnt->setValue(aLastInSketch);
 245     aCenterPnt->setValue(aCenterInSketch);
 246     aProjection->boolean(SketchPlugin_Arc::REVERSED_ID())->setValue(isInversed);
 247
 248     aProjection->data()->blockSendAttributeUpdated(aWasBlocked, false);
 249   } else
 250     return;
 251
 252   aProjection->boolean(COPY_ID())->setValue(true);
 253   aProjection->execute();
 254   aRefAttr->setObject(aProjection);
 255
 256   if (theID == EXTERNAL_FEATURE_ID()) {
 257     selection(EXTERNAL_ID())->selectValue(aExtFeature);
 258
 259     if (aResult) {
 260       aResult->setShape(aProjection->lastResult()->shape());
 261       setResult(aResult);
 262       aProjection->selection(EXTERNAL_ID())->setValue(lastResult(), lastResult()->shape());
 263     }
 264   }
 265 }