src/SketchPlugin/Test/TestTrimEllipse.py

   1 # Copyright (C) 2019-2021  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 email : webmaster.salome@opencascade.com
  18 #
  19
  20 from salome.shaper import model
  21 from salome.shaper import geom
  22 import math
  23
  24 from ModelAPI import *
  25 from SketchAPI import *
  26
  27 CENTER = geom.Pnt2d(10, 10)
  28 MAJOR_RADIUS = 50
  29 MINOR_RADIUS = 30
  30
  31 DOF = 11
  32 NB_LINES = 3
  33 NB_ELLIPSES = 1
  34 NB_ELLIPTIC_ARCS = 0
  35 NB_COINCIDENCES = 6
  36 NB_EQUALS = 0
  37
  38 TOLERANCE = 1.e-6
  39
  40 def checkFeaturesQuantity(theSketch):
  41     model.testNbSubFeatures(theSketch, "SketchLine", NB_LINES)
  42     model.testNbSubFeatures(theSketch, "SketchEllipse", NB_ELLIPSES)
  43     model.testNbSubFeatures(theSketch, "SketchEllipticArc", NB_ELLIPTIC_ARCS)
  44     model.testNbSubFeatures(theSketch, "SketchConstraintCoincidence", NB_COINCIDENCES)
  45     model.testNbSubFeatures(theSketch, "SketchConstraintEqual", NB_EQUALS)
  46     assert(model.dof(theSketch) == DOF)
  47
  48 def checkEllipticArcs(theSketch):
  49     for aSub in theSketch.features().list():
  50         aFeature = ModelAPI_Feature.feature(aSub)
  51         if aFeature is not None and aFeature.getKind() == "SketchEllipticArc":
  52             assertEllipticArc(SketchAPI_EllipticArc(aFeature))
  53
  54 def assertEllipticArc(theArc):
  55     assertPoints(theArc.center(), CENTER)
  56     assertPoints(theArc.majorAxisPositive(), geom.Pnt2d(CENTER.x() + MAJOR_RADIUS, CENTER.y()))
  57     assertPoints(theArc.minorAxisPositive(), geom.Pnt2d(CENTER.x(), CENTER.y() + MINOR_RADIUS))
  58
  59 def assertPoints(thePoint1, thePoint2):
  60   assert(math.fabs(thePoint1.x() - thePoint2.x()) < TOLERANCE), "{} != {}".format(thePoint1.x(), thePoint2.x())
  61   assert(math.fabs(thePoint1.y() - thePoint2.y()) < TOLERANCE), "{} != {}".format(thePoint1.y(), thePoint2.y())
  62
  63
  64 model.begin()
  65 partSet = model.moduleDocument()
  66 Part_1 = model.addPart(partSet)
  67 Part_1_doc = Part_1.document()
  68 Sketch_1 = model.addSketch(Part_1_doc, model.defaultPlane("XOY"))
  69 SketchEllipse_1 = Sketch_1.addEllipse(CENTER.x(), CENTER.y(), CENTER.x() + math.sqrt(MAJOR_RADIUS**2 - MINOR_RADIUS**2), CENTER.y(), MINOR_RADIUS)
  70 SketchLine_1 = Sketch_1.addLine(-16.74176451428603, -15.34869012470842, -16.85909682653373, 35.30399198463829)
  71 SketchConstraintCoincidence_1 = Sketch_1.setCoincident(SketchLine_1.startPoint(), SketchEllipse_1.result())
  72 SketchConstraintCoincidence_2 = Sketch_1.setCoincident(SketchLine_1.endPoint(), SketchEllipse_1.result())
  73 SketchLine_2 = Sketch_1.addLine(-16.85909682653373, 35.30399198463829, 20.9032928583277, -19.27802168426675)
  74 SketchConstraintCoincidence_3 = Sketch_1.setCoincident(SketchLine_1.endPoint(), SketchLine_2.startPoint())
  75 SketchConstraintCoincidence_4 = Sketch_1.setCoincident(SketchLine_2.startPoint(), SketchEllipse_1.result())
  76 SketchConstraintCoincidence_5 = Sketch_1.setCoincident(SketchLine_2.endPoint(), SketchEllipse_1.result())
  77 SketchLine_3 = Sketch_1.addLine(34.69765676551338, 36.08465583643841, 35.0422024535432, -17.96612629290852)
  78 SketchConstraintCoincidence_6 = Sketch_1.setCoincident(SketchLine_3.startPoint(), SketchEllipse_1.result())
  79 model.do()
  80
  81 checkFeaturesQuantity(Sketch_1)
  82 checkEllipticArcs(Sketch_1)
  83
  84 # trim the ellipse
  85 SketchSplit = Sketch_1.addTrim(SketchEllipse_1, geom.Pnt2d(CENTER.x() + MAJOR_RADIUS, CENTER.y()))
  86 model.do()
  87 NB_ELLIPSES -= 1
  88 NB_ELLIPTIC_ARCS += 1
  89 NB_COINCIDENCES += 1
  90
  91 checkFeaturesQuantity(Sketch_1)
  92 checkEllipticArcs(Sketch_1)
  93
  94 # trim the middle arc of ellipse
  95 EllipticArc = SketchAPI_EllipticArc(model.lastSubFeature(Sketch_1, "SketchEllipticArc"))
  96 ANGLE = -math.pi/2 - math.pi/10
  97 SketchSplit = Sketch_1.addTrim(EllipticArc, geom.Pnt2d(CENTER.x() + MAJOR_RADIUS * math.cos(ANGLE), CENTER.y() + MINOR_RADIUS * math.sin(ANGLE)))
  98 model.do()
  99 NB_ELLIPTIC_ARCS += 1
 100 NB_COINCIDENCES += 1
 101 NB_EQUALS += 1
 102 DOF += 1
 103
 104 checkFeaturesQuantity(Sketch_1)
 105 checkEllipticArcs(Sketch_1)
 106
 107 # trim the boundary arc of ellipse
 108 SketchSplit = Sketch_1.addTrim(EllipticArc, geom.Pnt2d(CENTER.x() - MAJOR_RADIUS, CENTER.y()))
 109 model.do()
 110 NB_COINCIDENCES -= 1
 111 DOF += 1
 112
 113 checkFeaturesQuantity(Sketch_1)
 114 checkEllipticArcs(Sketch_1)
 115
 116 model.end()
 117
 118 assert(model.checkPythonDump())