Skip to content

Commit e78fc81

Browse files
committed
Added window width/height optional parameters
1 parent 81e4793 commit e78fc81

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

mujoco_viewer/mujoco_viewer.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88

99

1010
class MujocoViewer:
11-
def __init__(self, model, data):
11+
def __init__(self, model, data, width=None, height=None):
1212
self.model = model
1313
self.data = data
1414

@@ -39,7 +39,13 @@ def __init__(self, model, data):
3939

4040
# glfw init
4141
glfw.init()
42-
width, height = glfw.get_video_mode(glfw.get_primary_monitor()).size
42+
43+
if not width:
44+
width, _ = glfw.get_video_mode(glfw.get_primary_monitor()).size
45+
46+
if not height:
47+
_, height = glfw.get_video_mode(glfw.get_primary_monitor()).size
48+
4349
self.window = glfw.create_window(
4450
width, height, "mujoco-python-viewer", None, None)
4551
glfw.make_context_current(self.window)

0 commit comments

Comments
 (0)