mirror of
https://github.com/alicevision/Meshroom.git
synced 2025-05-01 03:07:46 +02:00
125 lines
2.8 KiB
Python
125 lines
2.8 KiB
Python
|
|
class CoreDictModel:
|
|
|
|
def __init__(self, keyAttrName, **kwargs):
|
|
self._objects = {}
|
|
self._keyAttrName = keyAttrName
|
|
|
|
def __len__(self):
|
|
return len(self._objects)
|
|
|
|
def __iter__(self):
|
|
""" Enables iteration over the list of objects. """
|
|
return iter(self._objects.values())
|
|
|
|
def keys(self):
|
|
return self._objects.keys()
|
|
|
|
@property
|
|
def objects(self):
|
|
return self._objects
|
|
|
|
# TODO: operator[]
|
|
def get(self, key):
|
|
"""
|
|
Raises a KeyError if key is not in the map.
|
|
:param key:
|
|
:return:
|
|
"""
|
|
return self._objects[key]
|
|
|
|
def add(self, obj):
|
|
key = getattr(obj, self._keyAttrName, None)
|
|
assert key is not None
|
|
assert key not in self._objects
|
|
self._objects[key] = obj
|
|
|
|
def pop(self, key):
|
|
assert key in self._objects
|
|
return self._objects.pop(key)
|
|
|
|
def remove(self, obj):
|
|
assert obj in self._objects.values()
|
|
return self._objects.pop(getattr(obj, self._keyAttrName))
|
|
|
|
def clear(self):
|
|
self._objects.clear()
|
|
|
|
def update(self, objects):
|
|
for obj in objects:
|
|
self.add(obj)
|
|
|
|
def reset(self, objects):
|
|
self.clear()
|
|
self.update(objects)
|
|
|
|
|
|
class CoreListModel:
|
|
def __init__(self, parent=None):
|
|
self._objects = []
|
|
|
|
def __iter__(self):
|
|
return self._objects.__iter__()
|
|
|
|
def at(self, idx):
|
|
return self._objects[idx]
|
|
|
|
def append(self, obj):
|
|
self._objects.append(obj)
|
|
|
|
def extend(self, iterable):
|
|
self._objects.extend(iterable)
|
|
|
|
def indexOf(self, obj):
|
|
self._objects.index(obj)
|
|
|
|
def removeAt(self, idx):
|
|
del self._objects[idx]
|
|
|
|
def remove(self, obj):
|
|
self._objects.remove(obj)
|
|
|
|
|
|
class CoreSignal:
|
|
""" Simple signal/callback implementation """
|
|
def __init__(self):
|
|
self._callbacks = set()
|
|
|
|
def emit(self, *args):
|
|
# TODO: check if we really need this in non-UI mode
|
|
# [cb(*args) for cb in self._callbacks]
|
|
pass
|
|
|
|
def connect(self, func):
|
|
self._callbacks.add(func)
|
|
|
|
|
|
def CoreSlot(*args, **kwargs):
|
|
def slot_decorator(func):
|
|
def func_wrapper(*f_args, **f_kwargs):
|
|
return func(*f_args, **f_kwargs)
|
|
return func_wrapper
|
|
return slot_decorator
|
|
|
|
|
|
class CoreProperty(property):
|
|
def __init__(self, ptype, fget=None, fset=None, **kwargs):
|
|
super(CoreProperty, self).__init__(fget, fset)
|
|
|
|
|
|
class CoreObject(object):
|
|
def __init__(self, parent=None, *args, **kwargs):
|
|
super(CoreObject, self).__init__()
|
|
self._parent = parent
|
|
|
|
def parent(self):
|
|
return self._parent
|
|
|
|
|
|
DictModel = CoreDictModel
|
|
ListModel = CoreListModel
|
|
Slot = CoreSlot
|
|
Signal = CoreSignal
|
|
Property = CoreProperty
|
|
BaseObject = CoreObject
|
|
Variant = object
|