diff options
Diffstat (limited to 'plugins/motion_capture/device.py')
| -rw-r--r-- | plugins/motion_capture/device.py | 17 |
1 files changed, 16 insertions, 1 deletions
diff --git a/plugins/motion_capture/device.py b/plugins/motion_capture/device.py index 8eb1591..58f89ce 100644 --- a/plugins/motion_capture/device.py +++ b/plugins/motion_capture/device.py @@ -24,10 +24,13 @@ from devices.base_device import ( class CameraDevice(BaseDevice): + DEVICE_TYPE = "camera" + def __init__(self, device_id: str = "cam_0", camera_index: int = 0, simulate: bool = False, - resolution=None): + resolution=None, + fps=None): name = "Camera (Sim)" if simulate else f"Camera {camera_index}" super().__init__(DeviceInfo( device_id = device_id, @@ -57,6 +60,7 @@ class CameraDevice(BaseDevice): self._camera_index = camera_index self._simulate = simulate self._resolution = resolution + self._fps = fps self._tracker: Optional[Any] = None self._tracking_win = None @@ -68,6 +72,7 @@ class CameraDevice(BaseDevice): self._tracker.start( -1 if self._simulate else self._camera_index, resolution=self._resolution, + fps=self._fps, ) if self._tracker.simulated: self.status = DeviceStatus.SIMULATED @@ -93,6 +98,16 @@ class CameraDevice(BaseDevice): x, y = pos return {"x_pos": float(x), "y_pos": float(y)} + def get_save_config(self) -> dict: + return { + "device_type": self.DEVICE_TYPE, + "device_id": self.info.device_id, + "camera_index": self._camera_index, + "simulate": self._simulate, + "resolution": list(self._resolution) if self._resolution else None, + "fps": self._fps, + } + def write_channel(self, channel_id: str, value: Any) -> bool: return False |