Enums.py

Go to the documentation of this file.

 
def IsicellEnum(name,values):
  class EnumItem(int):
    def __new__(cls, enum, name, value, *args, **kwargs):
      test =  super(cls, cls).__new__(cls,value)
      test.__doc__ = name+' : int'
      test.__enum = enum
      test.__name = name
      return test
    def __repr__(self):
      return f'<{self.__enum}.{self.__name}: {int(self)}>'
 
  def managed_attribute(name):
    @property
    def prop(self):
        return self._data[name]
    return prop
 
  class Enum(): 
    def __init__(self,name,values):
      self.__doc__ = name+' : Enum'+name
      self.__name__ = name
      self._data = {k:EnumItem(name,k,v) for v,k in values.items()}
      self._cast = {int(v):v for v in self._data.values()}
      self._dict = {k:v for v,k in values.items()}
      self._invert_dict = {v:k for v,k in values.items()}
      for v,k in values.items():
        setattr(self.__class__, k, managed_attribute(k))
    def __repr__(self):
        return f"<enum {self.__name__} : [{', '.join([f'{k}={int(v)}' for k,v in sorted(self._data.items(),key=lambda i:i[1])])}]>" 
    def to_dict(self):
      r"""to_dict(Enum self) -> dict"""
      return self._dict
    def to_invert_dict(self):
      r"""to_invert_dict(Enum self) -> dict"""
      return self._invert_dict
    def __call__(self, key):
      r"""__call__(Enum self, key) -> EnumItem"""
      if isinstance(key, int):
        return self._cast[key]
      elif isinstance(key, str):
        return self._data[key]
      else:
        raise KeyError(f"Key must be an int or str, got {type(key)}")
 
  return Enum(name,values)
 
/*isigen:insert@SWIGenumDeclarations'''
    tmpAllEnum = {'State':dict(enumerate(sorted([s["name"] for s in builderData["cell"]["States"]])))}
    tmpAllEnum.update({'Type':dict(enumerate((sorted(builderData["cellTypes"]) if len(builderData["cellTypes"])>0 else ['Common'])))})
    if "enums" in builderData:
        tmpAllEnum.update({upperFirst(name): dict(enumerate(sorted(listV))) for name, listV in builderData["enums"].items()})
    print(f'enumList = {tmpAllEnum}')
    '''*/
for enumName,enumValues in enumList.items():
  globals()[enumName] = IsicellEnum(enumName,enumValues)
del enumName,IsicellEnum,enumValues,enumList