forked from M-Labs/nac3
58 lines
1.9 KiB
Python
58 lines
1.9 KiB
Python
|
class EmbeddingMap:
|
||
|
|
def __init__(self):
|
||
|
|
self.object_inverse_map = {}
|
||
|
|
self.object_map = {}
|
||
|
|
self.string_map = {}
|
||
|
|
self.string_reverse_map = {}
|
||
|
|
self.function_map = {}
|
||
|
|
|
||
|
|
# preallocate exception names
|
||
|
|
self.preallocate_runtime_exception_names(["RuntimeError",
|
||
|
|
"RTIOUnderflow",
|
||
|
|
"RTIOOverflow",
|
||
|
|
"RTIODestinationUnreachable",
|
||
|
|
"DMAError",
|
||
|
|
"I2CError",
|
||
|
|
"CacheError",
|
||
|
|
"SPIError",
|
||
|
|
"0:ZeroDivisionError",
|
||
|
|
"0:IndexError"])
|
||
|
|
|
||
|
|
def preallocate_runtime_exception_names(self, names):
|
||
|
|
for i, name in enumerate(names):
|
||
|
|
if ":" not in name:
|
||
|
|
name = "0:artiq.coredevice.exceptions." + name
|
||
|
|
exn_id = self.store_str(name)
|
||
|
|
assert exn_id == i
|
||
|
|
|
||
|
|
def store_function(self, key, fun):
|
||
|
|
self.function_map[key] = fun
|
||
|
|
return key
|
||
|
|
|
||
|
|
def store_object(self, obj):
|
||
|
|
obj_id = id(obj)
|
||
|
|
if obj_id in self.object_inverse_map:
|
||
|
|
return self.object_inverse_map[obj_id]
|
||
|
|
key = len(self.object_map)
|
||
|
|
self.object_map[key] = obj
|
||
|
|
self.object_inverse_map[obj_id] = key
|
||
|
|
return key
|
||
|
|
|
||
|
|
def store_str(self, s):
|
||
|
|
if s in self.string_reverse_map:
|
||
|
|
return self.string_reverse_map[s]
|
||
|
|
key = len(self.string_map)
|
||
|
|
self.string_map[key] = s
|
||
|
|
self.string_reverse_map[s] = key
|
||
|
|
return key
|
||
|
|
|
||
|
|
def retrieve_function(self, key):
|
||
|
|
return self.function_map[key]
|
||
|
|
|
||
|
|
def retrieve_object(self, key):
|
||
|
|
return self.object_map[key]
|
||
|
|
|
||
|
|
def retrieve_str(self, key):
|
||
|
|
return self.string_map[key]
|
||
|
|
|