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
21 TestParameterCreation.py
23 class ParametersPlugin_Parameter
24 static const std::string MY_PARAMETER_ID("Parameter");
25 static const std::string MY_VARIABLE_ID("variable");
26 static const std::string MY_EXPRESSION_ID("expression");
28 data()->addAttribute(ParametersPlugin_Parameter::VARIABLE_ID(),
29 ModelAPI_AttributeString::typeId());
30 data()->addAttribute(ParametersPlugin_Parameter::EXPRESSION_ID(),
31 ModelAPI_AttributeString::typeId());
33 class ModelAPI_ResultParameter
34 static const std::string MY_VALUE_ID("Value");
35 static const std::string MY_VALUE_ID("State");
38 #=========================================================================
39 # Initialization of the test
40 #=========================================================================
41 from GeomDataAPI import *
42 from ModelAPI import *
45 from salome.shaper import model
47 __updated__ = "2015-04-27"
49 #=========================================================================
50 # Create several parameters and a feature.
51 # 1. Basic parameter definition:
52 # x1 = 150.0, y1 = 200.0
53 # 2. Referencing between parameters:
54 # x2 = x1 + y1 + 100.0
55 # 3. Referencing in feature SketchCircle
56 # CircleCenter = (x1 + 10.0, x1 + 20.0), CircleRadius = x1
59 #=========================================================================
60 class TestParameterRename(unittest.TestCase):
62 self.aSession = ModelAPI_Session.get()
63 self.aDocument = self.aSession.moduleDocument()
64 self.createParameters()
68 assert(model.checkPythonDump())
69 self.aSession.closeAll()
71 def createParameters(self):
72 ltNames = ["x1", "y1", "x2"]
73 ltExpressions = ["150.", "200.", "x1 + y1 + 100.0"]
75 for name, expr in zip(ltNames, ltExpressions):
76 self.aSession.startOperation()
77 aParam = self.aDocument.addFeature("Parameter")
78 aParamName = aParam.string("variable")
79 aParamName.setValue(name)
80 aParamExpr = aParam.string("expression")
81 aParamExpr.setValue(expr)
82 self.dtParams[name] = aParam
83 self.aSession.finishOperation()
84 self.assertEqual(len(self.dtParams), len(ltNames))
86 aParam = self.dtParams["x2"]
87 aResultAttr = modelAPI_ResultParameter(aParam.firstResult())
88 self.assertEqual(aResultAttr.data().real("Value").value(), 450.)
90 def createFeature(self):
91 self.aSession.startOperation()
92 aSketchCommonFeature = self.aDocument.addFeature("Sketch")
93 aSketchFeature = featureToCompositeFeature(aSketchCommonFeature)
94 origin = geomDataAPI_Point(aSketchFeature.attribute("Origin"))
95 origin.setValue(0, 0, 0)
96 dirx = geomDataAPI_Dir(aSketchFeature.attribute("DirX"))
97 dirx.setValue(1, 0, 0)
98 norm = geomDataAPI_Dir(aSketchFeature.attribute("Norm"))
99 norm.setValue(0, 0, 1)
100 aSketchCircle = aSketchFeature.addFeature("SketchCircle")
101 anCircleCentr = geomDataAPI_Point2D(aSketchCircle.attribute("circle_center"))
102 aRadiusAttr = aSketchCircle.real("circle_radius")
103 anCircleCentr.setValue(10., 20.)
104 aRadiusAttr.setValue(10.)
105 self.aSession.finishOperation()
107 self.anCircleCentr = anCircleCentr
108 self.aRadiusAttr = aRadiusAttr
110 # constraints to fix circle position and radius
111 self.aSession.startOperation()
113 aDistanceConstraint1 = aSketchFeature.addFeature("SketchConstraintDistance")
114 refattrA = aDistanceConstraint1.refattr("ConstraintEntityA")
115 refattrA.setAttr(anCircleCentr)
116 refattrB = aDistanceConstraint1.refattr("ConstraintEntityB")
117 anOY = aSketchFeature.addFeature("SketchLine")
118 aStartPoint = geomDataAPI_Point2D(anOY.attribute("StartPoint"))
119 anEndPoint = geomDataAPI_Point2D(anOY.attribute("EndPoint"))
120 aStartPoint.setValue(0., 0.)
121 anEndPoint.setValue(0., 100.)
122 anOYRes = modelAPI_Result(self.aDocument.objectByName("Construction", "OY"))
123 anOY.selection("External").setValue(anOYRes, anOYRes.shape())
125 refattrB.setObject(modelAPI_ResultConstruction(anOY.firstResult()))
126 value = aDistanceConstraint1.real("ConstraintValue")
127 value.setText("x1 + 10.0")
128 aDistanceConstraint1.execute();
130 aDistanceConstraint2 = aSketchFeature.addFeature("SketchConstraintDistance")
131 refattrA = aDistanceConstraint2.refattr("ConstraintEntityA")
132 refattrA.setAttr(anCircleCentr)
133 refattrB = aDistanceConstraint2.refattr("ConstraintEntityB")
134 anOX = aSketchFeature.addFeature("SketchLine")
135 aStartPoint = geomDataAPI_Point2D(anOX.attribute("StartPoint"))
136 anEndPoint = geomDataAPI_Point2D(anOX.attribute("EndPoint"))
137 aStartPoint.setValue(0., 0.)
138 anEndPoint.setValue(100., 0.)
139 anOXRes = modelAPI_Result(self.aDocument.objectByName("Construction", "OX"))
140 anOX.selection("External").setValue(anOXRes, anOXRes.shape())
142 refattrB.setObject(modelAPI_ResultConstruction(anOX.firstResult()))
143 value = aDistanceConstraint2.real("ConstraintValue")
144 value.setText("x1 + 20.0")
145 aDistanceConstraint2.execute();
147 aRadiusConstraint = aSketchFeature.addFeature("SketchConstraintRadius")
148 refattrA = aRadiusConstraint.refattr("ConstraintEntityA")
149 refattrA.setObject(modelAPI_ResultConstruction(aSketchCircle.lastResult()))
150 aRadiusConstrAttr = aRadiusConstraint.real("ConstraintValue")
151 aRadiusConstrAttr.setText("x1")
152 aRadiusConstraint.execute()
153 self.aSession.finishOperation()
155 self.assertEqual(self.anCircleCentr.x(), 160.)
156 self.assertEqual(self.anCircleCentr.y(), 170.)
157 self.assertEqual(aRadiusAttr.value(), 150.)
159 def test_change_value(self):
160 aParam = self.dtParams["x1"]
161 aResultAttr = modelAPI_ResultParameter(aParam.firstResult())
162 self.aSession.startOperation()
163 aParam.string("expression").setValue("200.0")
164 self.aSession.finishOperation()
166 # Check value of the parameter
167 self.assertEqual(aResultAttr.data().real("Value").value(), 200.)
168 # Check value of references
169 aParam = self.dtParams["x2"]
170 aResultAttr = modelAPI_ResultParameter(aParam.firstResult())
171 self.assertEqual(aResultAttr.data().real("Value").value(), 500.)
172 # Check value in the feature
173 self.assertEqual(self.anCircleCentr.x(), 210.)
174 self.assertEqual(self.anCircleCentr.y(), 220.)
175 self.assertEqual(self.aRadiusAttr.value(), 200.)
177 if __name__ == "__main__":
178 test_program = unittest.main(exit=False)
179 assert test_program.result.wasSuccessful(), "Test failed"
180 #=========================================================================
182 #=========================================================================