]> SALOME platform Git repositories - modules/shaper.git/blob - src/ParametersPlugin/Test/TestParameterRename.py
Salome HOME
Merge remote-tracking branch 'remotes/origin/HighLevelDump'
[modules/shaper.git] / src / ParametersPlugin / Test / TestParameterRename.py
1 """
2     TestParameterCreation.py
3     
4     class ParametersPlugin_Parameter
5     static const std::string MY_PARAMETER_ID("Parameter");
6     static const std::string MY_VARIABLE_ID("variable");
7     static const std::string MY_EXPRESSION_ID("expression");
8     
9     data()->addAttribute(ParametersPlugin_Parameter::VARIABLE_ID(),
10                          ModelAPI_AttributeString::typeId());
11     data()->addAttribute(ParametersPlugin_Parameter::EXPRESSION_ID(),
12                          ModelAPI_AttributeString::typeId());
13                          
14     class ModelAPI_ResultParameter
15     static const std::string MY_VALUE_ID("Value");
16     static const std::string MY_VALUE_ID("State");
17 """
18
19 #=========================================================================
20 # Initialization of the test
21 #=========================================================================
22 from GeomDataAPI import *
23 from ModelAPI import *
24 import math
25 import unittest
26 import model
27
28 __updated__ = "2015-04-27"
29
30 #=========================================================================
31 # Create several parameters and a feature.
32 # 1. Basic parameter definition:
33 #    x1 = 150.0, y1 = 200.0
34 # 2. Referencing between parameters:
35 #    x2 = x1 + y1 + 100.0
36 # 3. Referencing in feature SketchCircle
37 #    CircleCenter = (x1 + 10.0, x1 + 20.0), CircleRadius = x1
38 # 3. Check renaming
39 #    x1 -> a1
40 # 4. Check renaming to not unique name
41 #    x1 -> y1
42 #=========================================================================
43 class TestParameterRename(unittest.TestCase):
44     def setUp(self):
45         self.aSession = ModelAPI_Session.get()
46         self.aDocument = self.aSession.moduleDocument()
47         self.createParameters()
48         self.createFeature()
49
50     def tearDown(self):
51         assert(model.checkPythonDump())
52         self.aSession.closeAll()
53
54     def createParameters(self):
55         ltNames = ["x1", "y1", "x2"]
56         ltExpressions = ["150.", "200.", "x1 + y1 + 100.0"]
57         self.dtParams = {}
58         for name, expr in zip(ltNames, ltExpressions):
59             self.aSession.startOperation()
60             aParam = self.aDocument.addFeature("Parameter")
61             aParamName = aParam.string("variable")
62             aParamName.setValue(name)
63             aParamExpr = aParam.string("expression")
64             aParamExpr.setValue(expr)
65             self.dtParams[name] = aParam
66             self.aSession.finishOperation()
67         self.assertEqual(len(self.dtParams), len(ltNames))
68
69         aParam = self.dtParams["x2"]
70         aResultAttr = modelAPI_ResultParameter(aParam.firstResult())
71         self.assertEqual(aResultAttr.data().real("Value").value(), 450.)
72
73     def createFeature(self):
74         self.aSession.startOperation()
75         aSketchCommonFeature = self.aDocument.addFeature("Sketch")
76         aSketchFeature = featureToCompositeFeature(aSketchCommonFeature)
77         origin = geomDataAPI_Point(aSketchFeature.attribute("Origin"))
78         origin.setValue(0, 0, 0)
79         dirx = geomDataAPI_Dir(aSketchFeature.attribute("DirX"))
80         dirx.setValue(1, 0, 0)
81         norm = geomDataAPI_Dir(aSketchFeature.attribute("Norm"))
82         norm.setValue(0, 0, 1)
83         aSketchCircle = aSketchFeature.addFeature("SketchCircle")
84         anCircleCentr = geomDataAPI_Point2D(aSketchCircle.attribute("CircleCenter"))
85         aRadiusAttr = aSketchCircle.real("CircleRadius")
86         anCircleCentr.setValue(10., 20.)
87         aRadiusAttr.setValue(10.)
88         self.aSession.finishOperation()
89
90         self.anCircleCentr = anCircleCentr
91         self.aRadiusAttr = aRadiusAttr
92
93         # constraints to fix circle position and radius
94         self.aSession.startOperation()
95         # fix X coordinate
96         aDistanceConstraint1 = aSketchFeature.addFeature("SketchConstraintDistance")
97         refattrA = aDistanceConstraint1.refattr("ConstraintEntityA")
98         refattrA.setAttr(anCircleCentr)
99         refattrB = aDistanceConstraint1.refattr("ConstraintEntityB")
100         anOY = aSketchFeature.addFeature("SketchLine")
101         aStartPoint = geomDataAPI_Point2D(anOY.attribute("StartPoint"))
102         anEndPoint = geomDataAPI_Point2D(anOY.attribute("EndPoint"))
103         aStartPoint.setValue(0., 0.)
104         anEndPoint.setValue(0., 100.)
105         anOYRes = modelAPI_Result(self.aDocument.objectByName("Construction", "OY"))
106         anOY.selection("External").setValue(anOYRes, anOYRes.shape())
107         anOY.execute()
108         refattrB.setObject(modelAPI_ResultConstruction(anOY.firstResult()))
109         valueX = aDistanceConstraint1.real("ConstraintValue")
110         valueX.setText("x1 + 10.0")
111         aDistanceConstraint1.execute();
112         # fix Y coordinate
113         aDistanceConstraint2 = aSketchFeature.addFeature("SketchConstraintDistance")
114         refattrA = aDistanceConstraint2.refattr("ConstraintEntityA")
115         refattrA.setAttr(anCircleCentr)
116         refattrB = aDistanceConstraint2.refattr("ConstraintEntityB")
117         anOX = aSketchFeature.addFeature("SketchLine")
118         aStartPoint = geomDataAPI_Point2D(anOX.attribute("StartPoint"))
119         anEndPoint = geomDataAPI_Point2D(anOX.attribute("EndPoint"))
120         aStartPoint.setValue(0., 0.)
121         anEndPoint.setValue(100., 0.)
122         anOXRes = modelAPI_Result(self.aDocument.objectByName("Construction", "OX"))
123         anOX.selection("External").setValue(anOXRes, anOXRes.shape())
124         anOX.execute()
125         refattrB.setObject(modelAPI_ResultConstruction(anOX.firstResult()))
126         valueY = aDistanceConstraint2.real("ConstraintValue")
127         valueY.setText("x1 + 20.0")
128         aDistanceConstraint2.execute();
129         # fix radius
130         aRadiusConstraint = aSketchFeature.addFeature("SketchConstraintRadius")
131         refattrA = aRadiusConstraint.refattr("ConstraintEntityA")
132         refattrA.setObject(modelAPI_ResultConstruction(aSketchCircle.lastResult()))
133         aRadiusConstrAttr = aRadiusConstraint.real("ConstraintValue")
134         aRadiusConstrAttr.setText("x1")
135         aRadiusConstraint.execute()
136         self.aSession.finishOperation()
137
138         self.aCircleCenterX = valueX
139         self.aCircleCenterY = valueY
140         self.aCircleRadius = aRadiusConstrAttr
141
142         self.assertEqual(self.anCircleCentr.x(), 160.)
143         self.assertEqual(self.anCircleCentr.y(), 170.)
144         self.assertEqual(aRadiusAttr.value(), 150.)
145
146     def test_rename(self):
147         # Rename
148         aParam = self.dtParams["x1"]
149         aResultAttr = modelAPI_ResultParameter(aParam.firstResult())
150         self.aSession.startOperation()
151         aResultAttr.data().setName("a1")
152         self.aSession.finishOperation()
153
154         # Check rename in the parameter 
155         self.assertEqual(aParam.name(), "a1")
156         self.assertEqual(aParam.string("variable").value(), "a1")
157         self.assertEqual(aResultAttr.data().name(), "a1")
158         # Check rename in references
159         aParam = self.dtParams["x2"]
160         self.assertEqual(aParam.string("expression").value(), "a1 + y1 + 100.0")
161         # Check rename in the feature
162         self.assertEqual(self.aCircleCenterX.text(), "a1 + 10.0")
163         self.assertEqual(self.aCircleCenterY.text(), "a1 + 20.0")
164         self.assertEqual(self.aCircleRadius.text(), "a1")
165         # Check values
166         self.assertEqual(self.anCircleCentr.x(), 160.)
167         self.assertEqual(self.anCircleCentr.y(), 170.)
168         self.assertEqual(self.aRadiusAttr.value(), 150.)
169
170     def test_rename_not_unique(self):
171         # Rename to not unique name
172         aParam = self.dtParams["x1"]
173         aResultAttr = modelAPI_ResultParameter(aParam.firstResult())
174         self.aSession.startOperation()
175         aResultAttr.data().setName("y1")
176         self.aSession.finishOperation()
177         # Check rename in the parameter (Expected: not renamed) 
178         self.assertEqual(aParam.name(), "x1")
179         self.assertEqual(aParam.string("variable").value(), "x1")
180         self.assertEqual(aResultAttr.data().name(), "x1")
181         # Check rename in references (Expected: not renamed)
182         aParam = self.dtParams["x2"]
183         self.assertEqual(aParam.string("expression").value(), "x1 + y1 + 100.0")
184         # Check rename in the feature (Expected: not renamed)
185         self.assertEqual(self.aCircleCenterX.text(), "x1 + 10.0")
186         self.assertEqual(self.aCircleCenterY.text(), "x1 + 20.0")
187         self.assertEqual(self.aCircleRadius.text(), "x1")
188         # Check values
189         self.assertEqual(self.anCircleCentr.x(), 160.)
190         self.assertEqual(self.anCircleCentr.y(), 170.)
191         self.assertEqual(self.aRadiusAttr.value(), 150.)
192
193 if __name__ == '__main__':
194     unittest.main()
195 #=========================================================================
196 # End of test
197 #=========================================================================