1 # Copyright (C) 2014-2019 CEA/DEN, EDF R&D
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.
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.
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
17 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
24 from GeomDataAPI import *
25 from ModelAPI import *
31 #=========================================================================
33 #=========================================================================
34 def mirrorDiagonal(theSketch, allLines):
36 for aLine in allLines:
37 aStartPoint = geomDataAPI_Point2D(aLine.attribute("StartPoint"))
38 anEndPoint = geomDataAPI_Point2D(aLine.attribute("EndPoint"))
40 aNewLine = aSketchFeature.addFeature("SketchLine")
41 aNewStartPoint = geomDataAPI_Point2D(aNewLine.attribute("StartPoint"))
42 aNewEndPoint = geomDataAPI_Point2D(aNewLine.attribute("EndPoint"))
43 aNewStartPoint.setValue(aStartPoint.y(), aStartPoint.x())
44 aNewEndPoint.setValue(anEndPoint.y(), anEndPoint.x())
45 result.append(aNewLine)
46 allLines.extend(result)
49 def mirrorX(theSketch, allLines):
51 for aLine in allLines:
52 aStartPoint = geomDataAPI_Point2D(aLine.attribute("StartPoint"))
53 anEndPoint = geomDataAPI_Point2D(aLine.attribute("EndPoint"))
55 aNewLine = aSketchFeature.addFeature("SketchLine")
56 aNewStartPoint = geomDataAPI_Point2D(aNewLine.attribute("StartPoint"))
57 aNewEndPoint = geomDataAPI_Point2D(aNewLine.attribute("EndPoint"))
58 aNewStartPoint.setValue(aStartPoint.x() * -1., aStartPoint.y())
59 aNewEndPoint.setValue(anEndPoint.x() * -1., anEndPoint.y())
60 result.append(aNewLine)
61 allLines.extend(result)
64 def mirrorY(theSketch, allLines):
66 for aLine in allLines:
67 aStartPoint = geomDataAPI_Point2D(aLine.attribute("StartPoint"))
68 anEndPoint = geomDataAPI_Point2D(aLine.attribute("EndPoint"))
70 aNewLine = aSketchFeature.addFeature("SketchLine")
71 aNewStartPoint = geomDataAPI_Point2D(aNewLine.attribute("StartPoint"))
72 aNewEndPoint = geomDataAPI_Point2D(aNewLine.attribute("EndPoint"))
73 aNewStartPoint.setValue(aStartPoint.x(), aStartPoint.y() * -1.)
74 aNewEndPoint.setValue(anEndPoint.x(), anEndPoint.y() * -1.)
75 result.append(aNewLine)
76 allLines.extend(result)
79 def initContour(theNumLines):
82 deltaY = random.uniform(20, 25)
83 for i in range(1, theNumLines):
84 rangeXMax = (prevPoint[1] + deltaY) * (30. / i if i % 2 != 0 else 2)
85 newX = random.uniform(prevPoint[1] + deltaY,
86 max(rangeXMax, prevPoint[1] + deltaY + 25.))
87 newPoint = (round(newX, 4), round(prevPoint[1] + deltaY, 4))
88 result.append((prevPoint, newPoint))
91 result.append((prevPoint, (prevPoint[1], prevPoint[1])))
95 def makeLinesCoincident(theSketch, allLines):
96 allPoints = collections.defaultdict(list)
97 for aLine in allLines:
98 aStartPoint = geomDataAPI_Point2D(aLine.attribute("StartPoint"))
99 anEndPoint = geomDataAPI_Point2D(aLine.attribute("EndPoint"))
100 aStartPointValue = (aStartPoint.x(), aStartPoint.y())
101 anEndPointValue = (anEndPoint.x(), anEndPoint.y())
102 allPoints[aStartPointValue].append(aStartPoint)
103 allPoints[anEndPointValue].append(anEndPoint)
104 for keyPoint, valuePoints in allPoints.items():
105 if len(valuePoints) != 2:
106 print("Strange point: ", keyPoint, "has in coincidence: ", len(valuePoints))
108 aConstraint = theSketch.addFeature("SketchConstraintCoincidence")
109 aConstraint.refattr("ConstraintEntityA").setAttr(valuePoints[0])
110 aConstraint.refattr("ConstraintEntityB").setAttr(valuePoints[1])
113 #=========================================================================
114 # Initialization of the test
115 #=========================================================================
117 __updated__ = "2014-11-27"
119 aSession = ModelAPI_Session.get()
120 aDocument = aSession.moduleDocument()
121 #=========================================================================
122 # Creation of a sketch
123 #=========================================================================
124 aSession.startOperation()
125 aSketchCommonFeature = aDocument.addFeature("Sketch")
126 aSketchFeature = featureToCompositeFeature(aSketchCommonFeature)
127 origin = geomDataAPI_Point(aSketchFeature.attribute("Origin"))
128 origin.setValue(0, 0, 0)
129 dirx = geomDataAPI_Dir(aSketchFeature.attribute("DirX"))
130 dirx.setValue(1, 0, 0)
131 norm = geomDataAPI_Dir(aSketchFeature.attribute("Norm"))
132 norm.setValue(0, 0, 1)
133 #=========================================================================
134 # 1 With the predefined contour or a new random generated in X0Y
135 # 2 Make complete contour using the diagonal, X and Y mirror symmetry
136 # 3 Link all lines in contour with coincidence
137 #=========================================================================
138 initialContour = [((35, 0), (134.8224, 22.0494)), ((134.8224, 22.0494), (71.987, 44.0988)),
139 ((71.987, 44.0988), (205.1448, 66.1482)), ((205.1448, 66.1482), (127.862, 88.1976)),
140 ((127.862, 88.1976), (416.861, 110.247)), ((416.861, 110.247), (233.0475, 132.2964)),
141 ((233.0475, 132.2964), (251.1518, 154.3458)), ((251.1518, 154.3458), (351.0993, 176.3952)),
142 ((351.0993, 176.3952), (432.3397, 198.4446)), ((432.3397, 198.4446), (223.3246, 220.494)),
143 ((223.3246, 220.494), (617.4069, 242.5434)), ((617.4069, 242.5434), (479.6524, 264.5928)),
144 ((479.6524, 264.5928), (453.5778, 286.6422)), ((453.5778, 286.6422), (541.3463, 308.6916)),
145 ((541.3463, 308.6916), (564.5509, 330.741)), ((564.5509, 330.741), (636.9284, 352.7904)),
146 ((636.9284, 352.7904), (383.5946, 374.8398)), ((383.5946, 374.8398), (403.3764, 396.8892)),
147 ((403.3764, 396.8892), (463.9447, 418.9386)), ((463.9447, 418.9386), (669.1751, 440.988)),
148 ((669.1751, 440.988), (602.2044, 463.0374)), ((602.2044, 463.0374), (942.0456, 485.0868)),
149 ((942.0456, 485.0868), (526.574, 507.1362)), ((526.574, 507.1362), (826.3619, 529.1856)),
150 ((826.3619, 529.1856), (576.9488, 551.235)), ((576.9488, 551.235), (624.5595, 573.2844)),
151 ((624.5595, 573.2844), (648.7146, 595.3338)), ((648.7146, 595.3338), (1194.6944, 617.3832)),
152 ((1194.6944, 617.3832), (646.6597, 639.4326)), ((646.6597, 639.4326), (839.8201, 661.482)),
153 ((839.8201, 661.482), (690.7487, 683.5314)), ((690.7487, 683.5314), (1350.2538, 705.5808)),
154 ((1350.2538, 705.5808), (731.0722, 727.6302)), ((731.0722, 727.6302), (1324.0992, 749.6796)),
155 ((1324.0992, 749.6796), (790.4873, 771.729)), ((790.4873, 771.729), (813.9883, 793.7784)),
156 ((813.9883, 793.7784), (828.9997, 815.8278)), ((828.9997, 815.8278), (1321.9798, 837.8772)),
157 ((1321.9798, 837.8772), (872.1503, 859.9266)), ((872.1503, 859.9266), (859.9266, 859.9266))]
160 # initialContour = initContour(40)
162 for begin, end in initialContour:
163 aNewLine = aSketchFeature.addFeature("SketchLine")
164 aStartPoint = geomDataAPI_Point2D(aNewLine.attribute("StartPoint"))
165 anEndPoint = geomDataAPI_Point2D(aNewLine.attribute("EndPoint"))
166 aStartPoint.setValue(begin[0], begin[1])
167 anEndPoint.setValue(end[0], end[1])
168 allLines.append(aNewLine)
169 mirrorDiagonal(aSketchFeature, allLines)
170 mirrorX(aSketchFeature, allLines)
171 mirrorY(aSketchFeature, allLines)
172 makeLinesCoincident(aSketchFeature, allLines)
173 aSession.finishOperation()
174 #=========================================================================
176 #=========================================================================
178 from salome.shaper import model
179 assert(model.checkPythonDump())