src/BuildPlugin/Test/TestFilling_ByEdges.py

   1 # Copyright (C) 2017-2023  CEA, EDF
   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 # Initialization of the test
  21 from ModelAPI import *
  22 from GeomDataAPI import *
  23 from GeomAlgoAPI import *
  24 from GeomAPI import *
  25
  26 # Get document
  27 aSession = ModelAPI_Session.get()
  28 aDocument = aSession.moduleDocument()
  29
  30 # Create a part
  31 aSession.startOperation()
  32 aPartFeature = aDocument.addFeature("Part")
  33 aSession.finishOperation()
  34 aPartResult = modelAPI_ResultPart(aPartFeature.firstResult())
  35 aPart = aPartResult.partDoc()
  36
  37 # Create first edge in a sketch
  38 aSession.startOperation()
  39 aSketch1 = featureToCompositeFeature(aPart.addFeature("Sketch"))
  40 anOrigin = geomDataAPI_Point(aSketch1.attribute("Origin"))
  41 anOrigin.setValue(0, 0, 0)
  42 aDirX = geomDataAPI_Dir(aSketch1.attribute("DirX"))
  43 aDirX.setValue(1, 0, 0)
  44 aNorm = geomDataAPI_Dir(aSketch1.attribute("Norm"))
  45 aNorm.setValue(0, 0, 1)
  46 # an arc
  47 anArc = aSketch1.addFeature("SketchArc")
  48 anArcCenter = geomDataAPI_Point2D(anArc.attribute("center_point"))
  49 anArcCenter.setValue(1, 1)
  50 anArcStartPoint = geomDataAPI_Point2D(anArc.attribute("start_point"))
  51 anArcStartPoint.setValue(20, 5)
  52 anArcEndPoint = geomDataAPI_Point2D(anArc.attribute("end_point"))
  53 anArcEndPoint.setValue(5, 20)
  54 aSession.finishOperation()
  55 aSketch1Result = aSketch1.firstResult()
  56
  57 # Create second edge in another sketch
  58 aSession.startOperation()
  59 aSketch2 = featureToCompositeFeature(aPart.addFeature("Sketch"))
  60 anOrigin = geomDataAPI_Point(aSketch2.attribute("Origin"))
  61 anOrigin.setValue(0, 0, 0)
  62 aDirX = geomDataAPI_Dir(aSketch2.attribute("DirX"))
  63 aDirX.setValue(1, 0, 0)
  64 aNorm = geomDataAPI_Dir(aSketch2.attribute("Norm"))
  65 aNorm.setValue(0, 0.7071067811865475, 0.7071067811865475)
  66 # a line
  67 aLine = aSketch2.addFeature("SketchLine")
  68 aLineStartPoint = geomDataAPI_Point2D(aLine.attribute("StartPoint"))
  69 aLineStartPoint.setValue(0, 0)
  70 aLineEndPoint = geomDataAPI_Point2D(aLine.attribute("EndPoint"))
  71 aLineEndPoint.setValue(20, 20)
  72 aSession.finishOperation()
  73 aSketch2Result = aSketch2.firstResult()
  74 # an edge
  75 aSession.startOperation()
  76 anEdge = aPart.addFeature("Edge")
  77 aBaseObjectsList = anEdge.selectionList("base_objects")
  78 aBaseObjectsList.append(aSketch2Result, aLine.lastResult().shape())
  79 aSession.finishOperation()
  80
  81 # Create filling
  82 aSession.startOperation()
  83 aFillingFeature = aPart.addFeature("Filling")
  84 aFillingFeature.string("advanced_options").setValue("")
  85 aBaseObjectsList = aFillingFeature.selectionList("base_objects")
  86 aBaseObjectsList.append(aSketch1Result, anArc.lastResult().shape())
  87 aSession.finishOperation()
  88
  89 # =============================================================================
  90 # Test 1. Filling on one edge is failed (no result is built)
  91 # =============================================================================
  92 assert(len(aFillingFeature.results()) == 0)
  93
  94 # =============================================================================
  95 # Test 2. Add another edge, filling should be completed
  96 # =============================================================================
  97 aSession.startOperation()
  98 aBaseObjectsList.append(anEdge.lastResult(), None)
  99 aSession.finishOperation()
 100 assert(len(aFillingFeature.results()) > 0)
 101
 102 # =============================================================================
 103 # Test 3. Change parameters one-by-one and check validity of result
 104 # =============================================================================
 105 aSession.startOperation()
 106 aFillingFeature.string("advanced_options").setValue("true")
 107 aSession.finishOperation()
 108 orientations = ["auto_correct", "curve_info", "edge_orient"]
 109 tolerances = [0.0001, 0.001]
 110 for ori in orientations:
 111   for minDeg in range(2, 4):
 112     for maxDeg in range(5, 7):
 113       for nbIter in range(0, 3, 2):
 114         for tol2d in tolerances:
 115           for tol3d in tolerances:
 116             for approx in [False, True]:
 117               aSession.startOperation()
 118               aFillingFeature.string("orientation").setValue(ori)
 119               aFillingFeature.integer("min_degree").setValue(minDeg)
 120               aFillingFeature.integer("max_degree").setValue(maxDeg)
 121               aFillingFeature.integer("nb_iter").setValue(nbIter)
 122               aFillingFeature.real("tol_2d").setValue(tol2d)
 123               aFillingFeature.real("tol_3d").setValue(tol3d)
 124               aFillingFeature.boolean("approximation").setValue(approx)
 125               aSession.finishOperation()
 126               assert(len(aFillingFeature.results()) > 0), "Filling feature failed with parameters:\n  orientation={}\n  min deg={}\n  max deg={}\n  nb iter={}\n  tol 2d={}\n  tol 3d={}\n  approximation={}".format(ori, minDeg, maxDeg, nbIter, tol2d, tol3d, approx)
 127
 128 # =============================================================================
 129 # Test 4. Discard parameters to default and add one more edge
 130 # =============================================================================
 131 aSession.startOperation()
 132 aFillingFeature.string("advanced_options").setValue("")
 133 aSession.finishOperation()
 134 # new arc
 135 aSession.startOperation()
 136 anArc2 = aSketch1.addFeature("SketchArc")
 137 anArc2Center = geomDataAPI_Point2D(anArc2.attribute("center_point"))
 138 anArc2Center.setValue(0, -5)
 139 anArc2StartPoint = geomDataAPI_Point2D(anArc2.attribute("start_point"))
 140 anArc2StartPoint.setValue(-20, -5)
 141 anArc2EndPoint = geomDataAPI_Point2D(anArc2.attribute("end_point"))
 142 anArc2EndPoint.setValue(20, -5)
 143 aSession.finishOperation()
 144 aSketch1Result = aSketch1.firstResult()
 145 # update filling
 146 aSession.startOperation()
 147 aPart.setCurrentFeature(aFillingFeature, True)
 148 aBaseObjectsList.append(aSketch1Result, anArc2.lastResult().shape())
 149 aSession.finishOperation()
 150 assert(len(aFillingFeature.results()) > 0)
 151
 152 from salome.shaper import model
 153 assert(model.checkPythonDump())