src/PythonAPI/model/tools.py

   1 """Common tools for other modules.
   2 Author: Sergey Pokhodenko
   3 Copyright (C) 2014-20xx CEA/DEN, EDF R&D
   4 """
   5
   6 import re
   7 import collections
   8
   9 import ModelAPI
  10 import GeomAPI
  11 import GeomDataAPI
  12
  13 # from .sketcher.sketch import Sketch
  14
  15
  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()
  20
  21
  22 class Selection:
  23     """Class for selection.
  24
  25     Selection() -> empty selection
  26     Selection(name, type) -> selection initialized with arguments:
  27         - name -- topological name
  28         - type -- type of the object
  29     Selection(context, shape) -> selection initialized with arguments:
  30         - context -- ModelAPI_Result object
  31         - shape -- GeomAPI_Shape shape
  32     """
  33
  34     def __init__(self, *args):
  35         """x.__init__(...) initializes x; see x.__class__.__doc__ for signature"""
  36
  37         if not args:
  38             self.args = (None, None)
  39             return
  40
  41         if len(args) == 1 and isinstance(args[0], basestring):
  42             self.args = args
  43             return
  44
  45         assert(len(args) > 1 and len(args) < 4)
  46         assert(isinstance(args[0], basestring) or
  47                isinstance(args[0], ModelAPI.ModelAPI_Result))
  48         if isinstance(args[0], basestring):
  49             assert(isinstance(args[1], basestring))
  50         elif isinstance(args[0], ModelAPI.ModelAPI_Result) or args[0] is None:
  51             assert(isinstance(args[1], GeomAPI.GeomAPI_Shape))
  52         self.args = args
  53
  54
  55 def fill_attribute(attribute, value):
  56     """Set value to attribure.
  57
  58     This function processes complex cases.
  59     ModelAPI_AttributeSelectionList can accept string, result + shape, list of
  60     strings and [result + shape]...
  61     ModelAPI_AttributeDouble can accept float and string
  62     """
  63     if (isinstance(attribute, ModelAPI.ModelAPI_AttributeBoolean) or
  64         isinstance(attribute, ModelAPI.ModelAPI_AttributeDocRef) or
  65         isinstance(attribute, ModelAPI.ModelAPI_AttributeInteger) or
  66         isinstance(attribute, ModelAPI.ModelAPI_AttributeReference) or
  67         isinstance(attribute, ModelAPI.ModelAPI_AttributeString)
  68         ):
  69         attribute.setValue(value)
  70
  71     elif isinstance(attribute, ModelAPI.ModelAPI_AttributeDouble):
  72         if isinstance(value, basestring):
  73             attribute.setText(value)
  74         else:
  75             attribute.setValue(value)
  76
  77     elif isinstance(attribute, ModelAPI.ModelAPI_AttributeIntArray):
  78         attrubute.setSize(len(value))
  79         for i in range(len(value)):
  80             attrubute.setValue(i, value[i])
  81
  82     elif isinstance(attribute, ModelAPI.ModelAPI_AttributeRefAttr):
  83         assert(isinstance(value, ModelAPI.ModelAPI_Attribute) or
  84                isinstance(value, ModelAPI.ModelAPI_Object))
  85         if isinstance(value, ModelAPI.ModelAPI_Attribute):
  86             attrubute.setAttr(value)
  87         elif isinstance(value, ModelAPI.ModelAPI_Object):
  88             attrubute.setObject(value)
  89
  90     elif isinstance(attribute, ModelAPI.ModelAPI_AttributeRefList):
  91         attribute.clear()
  92         if not value:
  93             return
  94
  95         assert(isinstance(value, collections.Iterable))
  96         for item in value:
  97             assert(isinstance(item, ModelAPI.ModelAPI_Object))
  98             attribute.append(item)
  99
 100     elif isinstance(attribute, ModelAPI.ModelAPI_AttributeSelection):
 101         if value is None:
 102             attribute.setValue(None, None)
 103             return
 104
 105         assert(isinstance(value, Selection))
 106         attribute.setValue(*value.args)
 107
 108     elif isinstance(attribute, ModelAPI.ModelAPI_AttributeSelectionList):
 109         attribute.clear()
 110         if not value:
 111             return
 112
 113         assert(isinstance(value, collections.Iterable))
 114         for item in value:
 115             assert(isinstance(item, Selection))
 116             attribute.append(*item.args)
 117
 118     elif (isinstance(attribute, GeomDataAPI.GeomDataAPI_Dir) or
 119           isinstance(attribute, GeomDataAPI.GeomDataAPI_Point) or
 120           isinstance(attribute, GeomDataAPI.GeomDataAPI_Point2D)
 121           ):
 122         assert(isinstance(value, collections.Iterable))
 123         attribute.setValue(*value)
 124
 125     else:
 126         raise AssertionError("Wrong attribute type: %s" % type(attribute))