1 """Common tools for other modules.
2 Author: Sergey Pokhodenko
3 Copyright (C) 2014-20xx CEA/DEN, EDF R&D
13 # from .sketcher.sketch import Sketch
16 def convert_to_underscore(name):
17 """Convert CamelCase to underscore_case."""
18 s1 = re.sub('(.)([A-Z][a-z]+)', r'\1_\2', name)
19 return re.sub('([a-z0-9])([A-Z])', r'\1_\2', s1).lower()
22 def get_value(attribute):
23 """Return simple type value from ModelAPI attribute if possible.
25 ModelAPI_AttributeDouble -> double
26 ModelAPI_AttributeString -> str
28 if (isinstance(attribute, ModelAPI.ModelAPI_AttributeDouble) or
29 isinstance(attribute, ModelAPI.ModelAPI_AttributeString)):
30 return attribute.value()
37 """Class for storing selection."""
39 def __init__(self, *args):
40 """Initialize selection."""
42 self.args = (None, None)
45 if len(args) == 1 and isinstance(args[0], basestring):
49 assert(len(args) > 1 and len(args) < 4)
50 assert(isinstance(args[0], basestring) or
51 isinstance(args[0], ModelAPI.ModelAPI_Result))
52 if isinstance(args[0], basestring):
53 assert(isinstance(args[1], basestring))
54 elif isinstance(args[0], ModelAPI.ModelAPI_Result) or args[0] is None:
55 assert(isinstance(args[1], GeomAPI.GeomAPI_Shape))
59 def fill_attribute(attribute, value):
60 """Set value to attribure.
62 This function processes complex cases.
63 ModelAPI_AttributeSelectionList can accept string, result + shape, list of
64 strings and [result + shape]...
65 ModelAPI_AttributeDouble can accept float and string
67 if (isinstance(attribute, ModelAPI.ModelAPI_AttributeBoolean) or
68 isinstance(attribute, ModelAPI.ModelAPI_AttributeDocRef) or
69 isinstance(attribute, ModelAPI.ModelAPI_AttributeInteger) or
70 isinstance(attribute, ModelAPI.ModelAPI_AttributeReference) or
71 isinstance(attribute, ModelAPI.ModelAPI_AttributeString)
73 attribute.setValue(value)
75 elif isinstance(attribute, ModelAPI.ModelAPI_AttributeDouble):
76 if isinstance(value, basestring):
77 attribute.setText(value)
79 attribute.setValue(value)
81 elif isinstance(attribute, ModelAPI.ModelAPI_AttributeIntArray):
82 attrubute.setSize(len(value))
83 for i in range(len(value)):
84 attrubute.setValue(i, value[i])
86 elif isinstance(attribute, ModelAPI.ModelAPI_AttributeRefAttr):
87 assert(isinstance(value, ModelAPI.ModelAPI_Attribute) or
88 isinstance(value, ModelAPI.ModelAPI_Object))
89 if isinstance(value, ModelAPI.ModelAPI_Attribute):
90 attrubute.setAttr(value)
91 elif isinstance(value, ModelAPI.ModelAPI_Object):
92 attrubute.setObject(value)
94 elif isinstance(attribute, ModelAPI.ModelAPI_AttributeRefList):
99 assert(isinstance(value, collections.Iterable))
101 assert(isinstance(item, ModelAPI.ModelAPI_Object))
102 attribute.append(item)
104 elif isinstance(attribute, ModelAPI.ModelAPI_AttributeSelection):
106 attribute.setValue(None, None)
109 assert(isinstance(value, Selection))
110 attribute.setValue(*value.args)
112 elif isinstance(attribute, ModelAPI.ModelAPI_AttributeSelectionList):
117 assert(isinstance(value, collections.Iterable))
119 assert(isinstance(item, Selection))
120 attribute.append(*item.args)
122 elif (isinstance(attribute, GeomDataAPI.GeomDataAPI_Dir) or
123 isinstance(attribute, GeomDataAPI.GeomDataAPI_Point) or
124 isinstance(attribute, GeomDataAPI.GeomDataAPI_Point2D)
126 assert(isinstance(value, collections.Iterable))
127 attribute.setValue(*value)
130 raise AssertionError("Wrong attribute type: %s" % type(attribute))