]> SALOME platform Git repositories - modules/shaper.git/blob - src/SketchPlugin/Test/TestConstraintTangent.py
Salome HOME
Merge remote-tracking branch 'remotes/origin/BR_PlaneGCS' into CodeCleanup
[modules/shaper.git] / src / SketchPlugin / Test / TestConstraintTangent.py
1 """
2     TestConstraintTangent.py
3     Unit test of SketchPlugin_ConstraintTangent class
4         
5     SketchPlugin_ConstraintTangent
6         static const std::string MY_CONSTRAINT_TANGENT_ID("SketchConstraintTangent");
7         data()->addAttribute(SketchPlugin_Constraint::ENTITY_A(), ModelAPI_AttributeRefAttr::typeId());
8         data()->addAttribute(SketchPlugin_Constraint::ENTITY_B(), ModelAPI_AttributeRefAttr::typeId());
9
10 """
11 from GeomDataAPI import *
12 from GeomAPI import *
13 from ModelAPI import *
14 import math
15 #=========================================================================
16 # Initialization of the test
17 #=========================================================================
18
19 __updated__ = "2015-03-17"
20
21 aSession = ModelAPI_Session.get()
22 aDocument = aSession.moduleDocument()
23 #=========================================================================
24 # Creation of a sketch
25 #=========================================================================
26 aSession.startOperation()
27 aSketchCommonFeature = aDocument.addFeature("Sketch")
28 aSketchFeature = featureToCompositeFeature(aSketchCommonFeature)
29 origin = geomDataAPI_Point(aSketchFeature.attribute("Origin"))
30 origin.setValue(0, 0, 0)
31 dirx = geomDataAPI_Dir(aSketchFeature.attribute("DirX"))
32 dirx.setValue(1, 0, 0)
33 norm = geomDataAPI_Dir(aSketchFeature.attribute("Norm"))
34 norm.setValue(0, 0, 1)
35 aSession.finishOperation()
36 #=========================================================================
37 # TEST 1. Arc-line tangency
38 #=========================================================================
39 # Creation of an arc and two lines
40 #=========================================================================
41 # Arc
42 aSession.startOperation()
43 aSketchArc1 = aSketchFeature.addFeature("SketchArc")
44 anArcCentr = geomDataAPI_Point2D(aSketchArc1.attribute("ArcCenter"))
45 anArcCentr.setValue(10., 10.)
46 anArcStartPoint = geomDataAPI_Point2D(aSketchArc1.attribute("ArcStartPoint"))
47 anArcStartPoint.setValue(0., 50.)
48 anArcEndPoint = geomDataAPI_Point2D(aSketchArc1.attribute("ArcEndPoint"))
49 anArcEndPoint.setValue(50., 0.)
50 aSession.finishOperation()
51 # Line 1
52 aSession.startOperation()
53 aSketchLine1 = aSketchFeature.addFeature("SketchLine")
54 aLine1StartPoint = geomDataAPI_Point2D(aSketchLine1.attribute("StartPoint"))
55 aLine1EndPoint = geomDataAPI_Point2D(aSketchLine1.attribute("EndPoint"))
56 aLine1StartPoint.setValue(0., 50.)
57 aLine1EndPoint.setValue(0., 100.)
58 aSession.finishOperation()
59 # Line 2
60 aSession.startOperation()
61 aSketchLine2 = aSketchFeature.addFeature("SketchLine")
62 aLine2StartPoint = geomDataAPI_Point2D(aSketchLine2.attribute("StartPoint"))
63 aLine2EndPoint = geomDataAPI_Point2D(aSketchLine2.attribute("EndPoint"))
64 aLine2StartPoint.setValue(50., 0.)
65 aLine2EndPoint.setValue(100., 0.)
66 aSession.finishOperation()
67 #=========================================================================
68 # Link arc points and lines points by the coincidence constraint
69 #=========================================================================
70 aSession.startOperation()
71 aConstraint = aSketchFeature.addFeature("SketchConstraintCoincidence")
72 reflistA = aConstraint.refattr("ConstraintEntityA")
73 reflistB = aConstraint.refattr("ConstraintEntityB")
74 reflistA.setAttr(anArcStartPoint)
75 reflistB.setAttr(aLine1StartPoint)
76 aConstraint.execute()
77 aSession.finishOperation()
78 aSession.startOperation()
79 aConstraint = aSketchFeature.addFeature("SketchConstraintCoincidence")
80 reflistA = aConstraint.refattr("ConstraintEntityA")
81 reflistB = aConstraint.refattr("ConstraintEntityB")
82 reflistA.setAttr(anArcEndPoint)
83 reflistB.setAttr(aLine2StartPoint)
84 aConstraint.execute()
85 aSession.finishOperation()
86 #=========================================================================
87 # Add tangency constraint and check correctness
88 #=========================================================================
89 aSession.startOperation()
90 aTangency = aSketchFeature.addFeature("SketchConstraintTangent")
91 aRefObjectA = aTangency.refattr("ConstraintEntityA")
92 aRefObjectB = aTangency.refattr("ConstraintEntityB")
93 anObjectA = modelAPI_ResultConstruction(aSketchArc1.lastResult())
94 anObjectB = modelAPI_ResultConstruction(aSketchLine1.firstResult())
95 assert (anObjectA is not None)
96 assert (anObjectB is not None)
97 aRefObjectA.setObject(anObjectA)
98 aRefObjectB.setObject(anObjectB)
99 aTangency.execute()
100 aSession.finishOperation()
101 anArcVecX = anArcStartPoint.x() - anArcCentr.x()
102 anArcVecY = anArcStartPoint.y() - anArcCentr.y()
103 aLen = math.sqrt(anArcVecX**2 + anArcVecY**2)
104 aLineVecX = aLine1EndPoint.x() - aLine1StartPoint.x()
105 aLineVecY = aLine1EndPoint.y() - aLine1StartPoint.y()
106 aLen = aLen * math.sqrt(aLineVecX**2 + aLineVecY**2)
107 aDot = anArcVecX * aLineVecX + anArcVecY * aLineVecY
108 assert math.fabs(aDot) <= 2.e-6 * aLen, "Observed dot product: {0}".format(aDot)
109 #=========================================================================
110 # Add tangency constraint for arc and second line and check correctness
111 #=========================================================================
112 aSession.startOperation()
113 aTangency = aSketchFeature.addFeature("SketchConstraintTangent")
114 aRefObjectA = aTangency.refattr("ConstraintEntityA")
115 aRefObjectB = aTangency.refattr("ConstraintEntityB")
116 anObjectA = modelAPI_ResultConstruction(aSketchArc1.lastResult())
117 anObjectB = modelAPI_ResultConstruction(aSketchLine2.firstResult())
118 assert (anObjectA is not None)
119 assert (anObjectB is not None)
120 aRefObjectA.setObject(anObjectA)
121 aRefObjectB.setObject(anObjectB)
122 aTangency.execute()
123 aSession.finishOperation()
124 anArcVecX = anArcEndPoint.x() - anArcCentr.x()
125 anArcVecY = anArcEndPoint.y() - anArcCentr.y()
126 aLen = math.sqrt(anArcVecX**2 + anArcVecY**2)
127 aLineVecX = aLine2EndPoint.x() - aLine2StartPoint.x()
128 aLineVecY = aLine2EndPoint.y() - aLine2StartPoint.y()
129 aLen = aLen * math.sqrt(aLineVecX**2 + aLineVecY**2)
130 aDot = anArcVecX * aLineVecX + anArcVecY * aLineVecY
131 assert math.fabs(aDot) <= 2.e-6 * aLen, "Observed dot product: {0}".format(aDot)
132
133 #=========================================================================
134 # TEST 2. Arc-arc tangency
135 #=========================================================================
136 # Creation of arcs
137 #=========================================================================
138 # Arc 1
139 aSession.startOperation()
140 aSketchArc1 = aSketchFeature.addFeature("SketchArc")
141 anArc1Centr = geomDataAPI_Point2D(aSketchArc1.attribute("ArcCenter"))
142 anArc1Centr.setValue(10., 10.)
143 anArc1StartPoint = geomDataAPI_Point2D(aSketchArc1.attribute("ArcStartPoint"))
144 anArc1StartPoint.setValue(50., 0.)
145 anArc1EndPoint = geomDataAPI_Point2D(aSketchArc1.attribute("ArcEndPoint"))
146 anArc1EndPoint.setValue(0., 50.)
147 aSession.finishOperation()
148 # Arc 2
149 aSession.startOperation()
150 aSketchArc2 = aSketchFeature.addFeature("SketchArc")
151 anArc2Centr = geomDataAPI_Point2D(aSketchArc2.attribute("ArcCenter"))
152 anArc2Centr.setValue(-10., 10.)
153 anArc2StartPoint = geomDataAPI_Point2D(aSketchArc2.attribute("ArcStartPoint"))
154 anArc2StartPoint.setValue(0., 50.)
155 anArc2EndPoint = geomDataAPI_Point2D(aSketchArc2.attribute("ArcEndPoint"))
156 anArc2EndPoint.setValue(-50., 0.)
157 aSession.finishOperation()
158 #=========================================================================
159 # Link points of arcs by the coincidence constraint
160 #=========================================================================
161 aSession.startOperation()
162 aConstraint = aSketchFeature.addFeature("SketchConstraintCoincidence")
163 reflistA = aConstraint.refattr("ConstraintEntityA")
164 reflistB = aConstraint.refattr("ConstraintEntityB")
165 reflistA.setAttr(anArc1EndPoint)
166 reflistB.setAttr(anArc2StartPoint)
167 aConstraint.execute()
168 aSession.finishOperation()
169 #=========================================================================
170 # Add tangency constraint and check correctness
171 #=========================================================================
172 aSession.startOperation()
173 aTangency = aSketchFeature.addFeature("SketchConstraintTangent")
174 aRefObjectA = aTangency.refattr("ConstraintEntityA")
175 aRefObjectB = aTangency.refattr("ConstraintEntityB")
176 anObjectA = modelAPI_ResultConstruction(aSketchArc1.lastResult())
177 anObjectB = modelAPI_ResultConstruction(aSketchArc2.lastResult())
178 assert (anObjectA is not None)
179 assert (anObjectB is not None)
180 aRefObjectA.setObject(anObjectA)
181 aRefObjectB.setObject(anObjectB)
182 aTangency.execute()
183 aSession.finishOperation()
184 anArc1VecX = anArc1EndPoint.x() - anArc1Centr.x()
185 anArc1VecY = anArc1EndPoint.y() - anArc1Centr.y()
186 aLen = math.sqrt(anArc1VecX**2 + anArc1VecY**2)
187 anArc2VecX = anArc2StartPoint.x() - anArc2Centr.x()
188 anArc2VecY = anArc2StartPoint.y() - anArc2Centr.y()
189 aLen = aLen * math.sqrt(anArc2VecX**2 + anArc2VecY**2)
190 aCross = anArc1VecX * anArc2VecY - anArc1VecY * anArc2VecX
191 assert math.fabs(aCross) <= 2.e-6 * aLen, "Observed cross product: {0}".format(aCross)
192
193 #=========================================================================
194 # TEST 3. Tangency between non-connected objects should be wrong
195 #=========================================================================
196 # Store data
197 aLine2StartPointPrev = (aLine2StartPoint.x(), aLine2StartPoint.y())
198 aLine2EndPointPrev = (aLine2EndPoint.x(), aLine2EndPoint.y())
199 anArc2CenterPrev = (anArc2Centr.x(), anArc2Centr.y())
200 anArc2StartPointPrev = (anArc2StartPoint.x(), anArc2StartPoint.y())
201 anArc2EndPointPrev = (anArc2EndPoint.x(), anArc2EndPoint.y())
202 #=========================================================================
203 # Add tangency between arc2 and line2
204 #=========================================================================
205 aSession.startOperation()
206 aTangency = aSketchFeature.addFeature("SketchConstraintTangent")
207 aRefObjectA = aTangency.refattr("ConstraintEntityA")
208 aRefObjectB = aTangency.refattr("ConstraintEntityB")
209 anObjectA = modelAPI_ResultConstruction(aSketchArc2.lastResult())
210 anObjectB = modelAPI_ResultConstruction(aSketchLine2.firstResult())
211 assert (anObjectA is not None)
212 assert (anObjectB is not None)
213 aRefObjectA.setObject(anObjectA)
214 aRefObjectB.setObject(anObjectB)
215 aTangency.execute()
216 aSession.finishOperation()
217 # Check that nothing is changed
218 aLine2StartPointNew = (aLine2StartPoint.x(), aLine2StartPoint.y())
219 aLine2EndPointNew = (aLine2EndPoint.x(), aLine2EndPoint.y())
220 anArc2CenterNew = (anArc2Centr.x(), anArc2Centr.y())
221 anArc2StartPointNew = (anArc2StartPoint.x(), anArc2StartPoint.y())
222 anArc2EndPointNew = (anArc2EndPoint.x(), anArc2EndPoint.y())
223 assert(aLine2StartPointNew == aLine2StartPointPrev)
224 assert(aLine2EndPointNew == aLine2EndPointPrev)
225 assert(anArc2CenterNew == anArc2CenterPrev)
226 assert(anArc2StartPointNew == anArc2StartPointPrev)
227 assert(anArc2EndPointNew == anArc2EndPointPrev)
228 #=========================================================================
229 # End of test
230 #=========================================================================