Register classes

Classes to store information about registers and interact with them

Signed-off-by: Daniel Garcia <[email protected]>

Author: demonpo

tbot_contrib/register/register.py

@@ -0,0 +1,37 @@
+import typing
+
+
+class Register:
+    def __init__(self, name: str, address: int, width: int):
+        self.name = name
+        self.address = address
+        assert width in [8, 16, 32, 64], f"Unsupported register width: {width!r}"
+        self.width = width
+
+
+class RegisterMap:
+    def __init__(self, registers_list: typing.List[Register] = []) -> None:
+        self.registers_dict = {}
+        if len(registers_list) > 0:
+            for register in registers_list:
+                self.registers_dict[register.name] = register
+
+    def add_register(self, register: Register) -> None:
+        if register.name in self.registers_dict:
+            raise Exception(
+                f"There is already a register with the name: {register.name}"
+            )
+        self.registers_dict[register.name] = register
+
+    def remove_register(self, name: str) -> Register:
+        if name in self.registers_dict:
+            return self.registers_dict.pop(name)
+        raise Exception(f"There is not a register with the name: {name}")
+
+    def get_register(self, name: str) -> Register:
+        if name in self.registers_dict:
+            return self.registers_dict[name]
+        raise Exception(f"There is not a register with the name: {name}")
+
+    def get_all_registers(self) -> typing.List[Register]:
+        return list(self.registers_dict.values())