2 TestConstraintPerpendicular.py
3 Unit test of SketchPlugin_ConstraintPerpendicular class
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");
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());
19 from GeomDataAPI import *
20 from ModelAPI import *
21 #=========================================================================
22 # Initialization of the test
23 #=========================================================================
25 __updated__ = "2014-09-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")
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)
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 #=========================================================================
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()
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()
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()
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 #=========================================================================
156 #=========================================================================