Skip to content

Commit a82d29b

Browse files
committed
Add per-camera image getters in gRPC client
1 parent d1f3063 commit a82d29b

File tree

1 file changed

+27
-0
lines changed

1 file changed

+27
-0
lines changed

aegis_grpc/python_client/aegis_grpc_client/grpc_client.py

Lines changed: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -153,6 +153,33 @@ async def get_robot_state(self) -> dict[str, np.ndarray]:
153153
self.logger.error(f"GetRobotState failed: {e}")
154154
raise
155155

156+
async def get_camera_scene_image(self) -> np.ndarray:
157+
self._check_connected()
158+
try:
159+
vision = await self.read_stub.GetCameraSceneImage(Empty())
160+
return self._image_to_array(vision)
161+
except grpc.RpcError as e:
162+
self.logger.error(f"GetCameraSceneImage failed: {e}")
163+
raise
164+
165+
async def get_camera_right_image(self) -> np.ndarray:
166+
self._check_connected()
167+
try:
168+
vision = await self.read_stub.GetCameraRightImage(Empty())
169+
return self._image_to_array(vision)
170+
except grpc.RpcError as e:
171+
self.logger.error(f"GetCameraRightImage failed: {e}")
172+
raise
173+
174+
async def get_camera_left_image(self) -> np.ndarray:
175+
self._check_connected()
176+
try:
177+
vision = await self.read_stub.GetCameraLeftImage(Empty())
178+
return self._image_to_array(vision)
179+
except grpc.RpcError as e:
180+
self.logger.error(f"GetCameraLeftImage failed: {e}")
181+
raise
182+
156183
async def get_robot_vision(self) -> dict[str, np.ndarray]:
157184
self._check_connected()
158185
try:

0 commit comments

Comments
 (0)