chenlin/Mujoco_WASM
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

run pyink and isort

Browse Source
This commit is contained in:
andrew
2025-03-16 17:44:24 -04:00
parent f02fdfe2d1
commit 86fd31b9b8
1 changed files with 15 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/mujoco/viewer.py
+13 -5
View File
@@ -25,10 +25,12 @@ import threading
import time import time
from typing import Callable, List, Optional, Tuple, Union from typing import Callable, List, Optional, Tuple, Union
import weakref import weakref
import glfw import glfw
import numpy as np
import mujoco import mujoco
from mujoco import _simulate from mujoco import _simulate
import numpy as np
if not glfw._glfw: # pylint: disable=protected-access if not glfw._glfw: # pylint: disable=protected-access
raise RuntimeError('GLFW dynamic library handle is not available') raise RuntimeError('GLFW dynamic library handle is not available')
@@ -114,7 +116,9 @@ class Handle:
return sim.viewport return sim.viewport
return None return None
def set_figures(self, viewports_figures: List[Tuple[mujoco.MjrRect, mujoco.MjvFigure]]): def set_figures(
self, viewports_figures: List[Tuple[mujoco.MjrRect, mujoco.MjvFigure]]
):
sim = self._sim() sim = self._sim()
if sim is not None: if sim is not None:
sim.set_figures(viewports_figures) sim.set_figures(viewports_figures)
@@ -125,7 +129,7 @@ class Handle:
sim.clear_figures() sim.clear_figures()
def overlay_text(self, overlay_texts: List[Tuple[int, int, str, str]]): def overlay_text(self, overlay_texts: List[Tuple[int, int, str, str]]):
""" Overlay text on the viewer. """Overlay text on the viewer.
Args: Args:
overlay_texts: List of tuples of (font, gridpos, text1, text2) overlay_texts: List of tuples of (font, gridpos, text1, text2)
@@ -144,11 +148,15 @@ class Handle:
if sim is not None: if sim is not None:
sim.clear_overlay_text() sim.clear_overlay_text()
def set_images(self, viewports_images: List[Tuple[mujoco.MjrRect, np.ndarray]]): def set_images(
self, viewports_images: List[Tuple[mujoco.MjrRect, np.ndarray]]
):
sim = self._sim() sim = self._sim()
if sim is not None: if sim is not None:
# Nearest neighbor resize # Nearest neighbor resize
resize = lambda a, s: a[(np.arange(s[0]) * a.shape[0]) // s[0]][:, (np.arange(s[1]) * a.shape[1]) // s[1]] resize = lambda a, s: a[(np.arange(s[0]) * a.shape[0]) // s[0]][
:, (np.arange(s[1]) * a.shape[1]) // s[1]
]
resized_viewports_images = [] resized_viewports_images = []
for viewport, image in viewports_images: for viewport, image in viewports_images:
targ_shape = (viewport.height, viewport.width) targ_shape = (viewport.height, viewport.width)