]> SALOME platform Git repositories - modules/shaper.git/blob - src/SketchPlugin/Test/TestConstraintPerpendicular.py
Salome HOME
Merge branch 'master' of newgeom:newgeom.git
[modules/shaper.git] / src / SketchPlugin / Test / TestConstraintPerpendicular.py
1 """
2     TestConstraintPerpendicular.py
3     Unit test of SketchPlugin_ConstraintPerpendicular class
4     
5     SketchPlugin_Constraint
6         static const std::string MY_CONSTRAINT_VALUE("ConstraintValue");
7         static const std::string MY_FLYOUT_VALUE_PNT("ConstraintFlyoutValuePnt");
8         static const std::string MY_ENTITY_A("ConstraintEntityA");
9         static const std::string MY_ENTITY_B("ConstraintEntityB");
10         static const std::string MY_ENTITY_C("ConstraintEntityC");
11         static const std::string MY_ENTITY_D("ConstraintEntityD");
12         
13     SketchPlugin_ConstraintPerpendicular
14         static const std::string MY_CONSTRAINT_PERPENDICULAR_ID("SketchConstraintPerpendicular");
15         data()->addAttribute(SketchPlugin_Constraint::ENTITY_A(), ModelAPI_AttributeRefAttr::type());
16         data()->addAttribute(SketchPlugin_Constraint::ENTITY_B(), ModelAPI_AttributeRefAttr::type());
17         data()->addAttribute(SketchPlugin_Constraint::FLYOUT_VALUE_PNT(), GeomDataAPI_Point2D::type()); 
18 """
19 from GeomDataAPI import *
20 from ModelAPI import *
21 #=========================================================================
22 # Initialization of the test
23 #=========================================================================
24
25 __updated__ = "2014-09-26"
26
27 aSession = ModelAPI_Session.get()
28 aDocument = aSession.moduleDocument()
29 #=========================================================================
30 # Creation of a sketch
31 #=========================================================================
32 aSession.startOperation()
33 aSketchFeature = aDocument.addFeature("Sketch")
34 aSketchFeatureData = aSketchFeature.data()
35 origin = geomDataAPI_Point(aSketchFeatureData.attribute("Origin"))
36 origin.setValue(0, 0, 0)
37 dirx = geomDataAPI_Dir(aSketchFeatureData.attribute("DirX"))
38 dirx.setValue(1, 0, 0)
39 diry = geomDataAPI_Dir(aSketchFeatureData.attribute("DirY"))
40 diry.setValue(0, 1, 0)
41 norm = geomDataAPI_Dir(aSketchFeatureData.attribute("Norm"))
42 norm.setValue(0, 0, 1)
43 aSession.finishOperation()
44 #=========================================================================
45 # Create two lines which are already perpendicular
46 #=========================================================================
47 aSession.startOperation()
48 aSketchReflist = aSketchFeatureData.reflist("Features")
49 # line A
50 aSketchLineA = aDocument.addFeature("SketchLine")
51 aSketchReflist.append(aSketchLineA)
52 aLineAStartPoint = geomDataAPI_Point2D(
53     aSketchLineA.data().attribute("StartPoint"))
54 aLineAEndPoint = geomDataAPI_Point2D(
55     aSketchLineA.data().attribute("EndPoint"))
56 aSketchLineB = aDocument.addFeature("SketchLine")
57 aLineAStartPoint.setValue(0., 25)
58 aLineAEndPoint.setValue(85., 25)
59 # line B
60 aSketchReflist.append(aSketchLineB)
61 aLineBStartPoint = geomDataAPI_Point2D(
62     aSketchLineB.data().attribute("StartPoint"))
63 aLineBEndPoint = geomDataAPI_Point2D(
64     aSketchLineB.data().attribute("EndPoint"))
65 aLineBStartPoint.setValue(25., 40.)
66 aLineBEndPoint.setValue(25., 125.)
67 aSession.finishOperation()
68 #=========================================================================
69 # Make a constraint to keep the length of the line constant
70 # to prevent perpendicular constraint collapsing line to point
71 #=========================================================================
72
73 for eachFeature in [aSketchLineA, aSketchLineB]:
74     aSession.startOperation()
75     aLengthConstraint = aDocument.addFeature("SketchConstraintLength")
76     aSketchReflist.append(aLengthConstraint)
77     aLengthConstraintData = aLengthConstraint.data()
78     refattrA = aLengthConstraintData.refattr("ConstraintEntityA")
79     aResultA = modelAPI_ResultConstruction(eachFeature.firstResult())
80     assert (aResultA is not None)
81     refattrA.setObject(aResultA)
82     aLengthConstraint.execute()
83     aSession.finishOperation()
84
85 # Coordinates of lines should not be changed after this constraint
86 assert (aLineAStartPoint.x() == 0)
87 assert (aLineAStartPoint.y() == 25)
88 assert (aLineAEndPoint.x() == 85)
89 assert (aLineAEndPoint.y() == 25)
90 assert (aLineBStartPoint.x() == 25)
91 assert (aLineBStartPoint.y() == 40)
92 assert (aLineBEndPoint.x() == 25)
93 assert (aLineBEndPoint.y() == 125)
94 #=========================================================================
95 # Link lines with perpendicular constraint
96 #=========================================================================
97 aSession.startOperation()
98 aPerpendicularConstraint = aDocument.addFeature(
99     "SketchConstraintPerpendicular")
100 aSketchReflist.append(aPerpendicularConstraint)
101 aConstraintData = aPerpendicularConstraint.data()
102 refattrA = aConstraintData.refattr("ConstraintEntityA")
103 refattrB = aConstraintData.refattr("ConstraintEntityB")
104 # aResultA is already defined for the length constraint
105 aResultB = modelAPI_ResultConstruction(aSketchLineB.firstResult())
106 assert (aResultB is not None)
107 refattrA.setObject(aResultA)
108 refattrB.setObject(aResultB)
109 aPerpendicularConstraint.execute()
110 aSession.finishOperation()
111 #=========================================================================
112 # Check values and move one constrainted object
113 #=========================================================================
114 # print "Check values and move one constrainted object"
115 # print "assert (aLineAStartPoint.x() == %d)" % aLineAStartPoint.x()
116 # print "assert (aLineAStartPoint.y() == %d)" % aLineAStartPoint.y()
117 # print "assert (aLineAEndPoint.x() == %d)" % aLineAEndPoint.x()
118 # print "assert (aLineAEndPoint.y() == %d)" % aLineAEndPoint.y()
119 # print "assert (aLineBStartPoint.x() == %d)" % aLineBStartPoint.x()
120 # print "assert (aLineBStartPoint.y() == %d)" % aLineBStartPoint.y()
121 # print "assert (aLineBEndPoint.x() == %d)" % aLineBEndPoint.x()
122 # print "assert (aLineBEndPoint.y() == %d)" % aLineBEndPoint.y()
123 deltaX = deltaY = 5.
124 # move line without rotation,
125 # check that reference's line points are not changed also
126 aLineAStartPoint.setValue(aLineAStartPoint.x() + deltaX,
127                           aLineAStartPoint.y() + deltaY)
128 aLineAEndPoint.setValue(aLineAEndPoint.x() + deltaX,
129                         aLineAEndPoint.y() + deltaY)
130 # print "move line without rotation"
131 # print "assert (aLineAStartPoint.x() == %d)" % aLineAStartPoint.x()
132 # print "assert (aLineAStartPoint.y() == %d)" % aLineAStartPoint.y()
133 # print "assert (aLineAEndPoint.x() == %d)" % aLineAEndPoint.x()
134 # print "assert (aLineAEndPoint.y() == %d)" % aLineAEndPoint.y()
135 # print "assert (aLineBStartPoint.x() == %d)" % aLineBStartPoint.x()
136 # print "assert (aLineBStartPoint.y() == %d)" % aLineBStartPoint.y()
137 # print "assert (aLineBEndPoint.x() == %d)" % aLineBEndPoint.x()
138 # print "assert (aLineBEndPoint.y() == %d)" % aLineBEndPoint.y()
139
140 # rotate line, check that reference's line points are moved also
141 aLineAStartPoint.setValue(aLineAStartPoint.x() + deltaX,
142                           aLineAStartPoint.y() + deltaY)
143 aLineAEndPoint.setValue(aLineAEndPoint.x() - deltaX,
144                         aLineAEndPoint.y() - deltaY)
145 # print "Rotate line, check that reference's line points are moved also"
146 # print "assert (aLineAStartPoint.x() == %d)" % aLineAStartPoint.x()
147 # print "assert (aLineAStartPoint.y() == %d)" % aLineAStartPoint.y()
148 # print "assert (aLineAEndPoint.x() == %d)" % aLineAEndPoint.x()
149 # print "assert (aLineAEndPoint.y() == %d)" % aLineAEndPoint.y()
150 # print "assert (aLineBStartPoint.x() == %d)" % aLineBStartPoint.x()
151 # print "assert (aLineBStartPoint.y() == %d)" % aLineBStartPoint.y()
152 # print "assert (aLineBEndPoint.x() == %d)" % aLineBEndPoint.x()
153 # print "assert (aLineBEndPoint.y() == %d)" % aLineBEndPoint.y()
154 #=========================================================================
155 # End of test
156 #=========================================================================