Add copy methods, fix state imports, pass lock
Add copy() to PosArray, DegArray and RadArray to allow easy shallow copies. Change RobotState imports in idle.py and walking.py to use explicit 'from RobotState.RobotState import RobotState' to avoid import issues. In main.py create a threading.Lock and pass it into the controller thread args to prepare for synchronized access in controller_loop.
This commit is contained in:
@@ -19,6 +19,8 @@ class PosArray:
|
||||
def __getitem__(self, key): return self.data[key]
|
||||
def __iter__(self): return iter(self.data)
|
||||
def __repr__(self): return f"PosArray(\n{self.data}\n)"
|
||||
def copy(self):
|
||||
return PosArray(self.data.copy())
|
||||
|
||||
|
||||
@dataclass
|
||||
@@ -37,6 +39,8 @@ class DegArray:
|
||||
def __getitem__(self, key): return self.data[key]
|
||||
def __iter__(self): return iter(self.data)
|
||||
def __repr__(self): return f"PosArray(\n{self.data}\n)"
|
||||
def copy(self):
|
||||
return DegArray(self.data.copy())
|
||||
|
||||
@dataclass
|
||||
class RadArray:
|
||||
@@ -53,6 +57,8 @@ class RadArray:
|
||||
def __getitem__(self, key): return self.data[key]
|
||||
def __iter__(self): return iter(self.data)
|
||||
def __repr__(self): return f"PosArray(\n{self.data}\n)"
|
||||
def copy(self):
|
||||
return RadArray(self.data.copy())
|
||||
|
||||
@dataclass
|
||||
class RobotCommand:
|
||||
|
||||
Reference in New Issue
Block a user