Version 2.1.4: Switch to GLAD, minor improvements to simulate.

PiperOrigin-RevId: 439277749
Change-Id: I741d1c499ae88753338fcc6d141a65e92e13023e
This commit is contained in:
Saran Tunyasuvunakool
2022-04-04 13:53:21 +01:00
parent 3c71720d83
commit 90dea1bdfb
23 changed files with 270 additions and 298 deletions
+3 -19
View File
@@ -55,7 +55,7 @@ include(FindOrFetch)
# ==================== MUJOCO LIBRARY ==========================================
if(NOT TARGET mujoco)
find_library(MUJOCO_LIBRARY mujoco mujoco.2.1.3 HINTS ${MUJOCO_LIBRARY_DIR} REQUIRED)
find_library(MUJOCO_LIBRARY mujoco mujoco.2.1.4 HINTS ${MUJOCO_LIBRARY_DIR} REQUIRED)
find_path(MUJOCO_INCLUDE mujoco.h HINTS ${MUJOCO_INCLUDE_DIR} REQUIRED)
message("MuJoCo is at ${MUJOCO_LIBRARY}")
message("MuJoCo headers are at ${MUJOCO_INCLUDE}")
@@ -152,7 +152,7 @@ else()
GIT_REPO
https://github.com/pybind/pybind11
GIT_TAG
08ea85b0ac0d4d46ac081ff24f481b41322d7159
a8f1a5567608f346bdba293b3d062a288ee16cd4
TARGETS
pybind11::pybind11_headers
EXCLUDE_FROM_ALL
@@ -257,12 +257,7 @@ macro(mujoco_pybind11_module name)
@rpath/$<TARGET_FILE_NAME:mujoco> -add_rpath @loader_path $<TARGET_FILE:${name}>
)
elseif(NOT WIN32)
add_custom_command(
TARGET ${name}
POST_BUILD
COMMAND patchelf --remove-needed $<TARGET_SONAME_FILE_NAME:mujoco>
$<TARGET_FILE_NAME:${name}>
)
set_target_properties(${name} PROPERTIES INSTALL_RPATH $ORIGIN BUILD_WITH_INSTALL_RPATH TRUE)
endif()
endmacro()
@@ -347,17 +342,6 @@ set(LIBRARIES_FOR_WHEEL
"$<TARGET_FILE:_structs>"
"$<TARGET_FILE:mujoco>"
)
if(NOT APPLE)
set(LIBRARIES_FOR_WHEEL ${LIBRARIES_FOR_WHEEL} $<TARGET_FILE:mujoco_nogl>)
endif()
if(NOT (APPLE OR WIN32))
set(LIBRARIES_FOR_WHEEL
${LIBRARIES_FOR_WHEEL}
"$<TARGET_FILE:glew>"
"$<TARGET_FILE:glewegl>"
"$<TARGET_FILE:glewosmesa>"
)
endif()
if(MUJOCO_PYTHON_MAKE_WHEEL)
add_custom_target(
+24 -49
View File
@@ -15,69 +15,44 @@
"""Python bindings for MuJoCo."""
import ctypes
import ctypes.util
import os
import platform
import subprocess
HEADERS_DIR = os.path.join(os.path.dirname(__file__), 'include')
_MUJOCO_GL_ENABLE = ('enable', 'enabled', 'on', 'true', '1' , '')
_MUJOCO_GL_DISABLE = ('disable', 'disabled', 'off', 'false', '0')
_MUJOCO_GL = os.environ.get('MUJOCO_GL', '').lower().strip()
_MUJOCO_GL_IS_VALID = True
_SYSTEM = platform.system()
if _SYSTEM == 'Linux':
libglew_name = None
if _MUJOCO_GL in _MUJOCO_GL_ENABLE + ('glfw', 'glx'):
libglew_name = 'libglew.so'
elif _MUJOCO_GL == 'egl':
libglew_name = 'libglewegl.so'
elif _MUJOCO_GL == 'osmesa':
libglew_name = 'libglewosmesa.so'
elif _MUJOCO_GL not in _MUJOCO_GL_DISABLE:
_MUJOCO_GL_IS_VALID = False
if libglew_name is not None:
ctypes.CDLL(os.path.join(os.path.dirname(__file__), libglew_name),
ctypes.RTLD_GLOBAL)
ctypes.CDLL(
os.path.join(os.path.dirname(__file__), 'libmujoco.so.2.1.3'),
ctypes.RTLD_GLOBAL)
else:
ctypes.CDLL(
os.path.join(os.path.dirname(__file__), 'libmujoco_nogl.so.2.1.3'),
ctypes.RTLD_GLOBAL)
elif _SYSTEM == 'Windows':
if _MUJOCO_GL in _MUJOCO_GL_ENABLE + ('glfw', 'wgl'):
ctypes.WinDLL(os.path.join(os.path.dirname(__file__), 'mujoco.dll'))
elif _MUJOCO_GL in _MUJOCO_GL_DISABLE:
ctypes.WinDLL(os.path.join(os.path.dirname(__file__), 'mujoco_nogl.dll'))
else:
_MUJOCO_GL_IS_VALID = False
if not _MUJOCO_GL_IS_VALID:
raise RuntimeError(
f'invalid value for environment variable MUJOCO_GL: {_MUJOCO_GL}')
if _SYSTEM == 'Windows':
ctypes.WinDLL(os.path.join(os.path.dirname(__file__), 'mujoco.dll'))
from mujoco._callbacks import *
from mujoco._constants import *
from mujoco._enums import *
from mujoco._errors import *
from mujoco._functions import *
from mujoco._render import *
from mujoco._structs import *
# pylint: disable=g-import-not-at-top
if _MUJOCO_GL not in _MUJOCO_GL_DISABLE:
from mujoco._render import *
if _SYSTEM != 'Linux':
from mujoco.glfw import GLContext
_MUJOCO_GL = os.environ.get('MUJOCO_GL', '').lower().strip()
if _MUJOCO_GL not in ('disable', 'disabled', 'off', 'false', '0'):
_VALID_MUJOCO_GL = ('enable', 'enabled', 'on', 'true', '1' , 'glfw', '')
if _SYSTEM == 'Linux':
_VALID_MUJOCO_GL += ('glx', 'egl', 'osmesa')
elif _SYSTEM == 'Windows':
_VALID_MUJOCO_GL += ('wgl',)
elif _SYSTEM == 'Darwin':
_VALID_MUJOCO_GL += ('cgl',)
if _MUJOCO_GL not in _VALID_MUJOCO_GL:
raise RuntimeError(
f'invalid value for environment variable MUJOCO_GL: {_MUJOCO_GL}')
if _SYSTEM == 'Linux' and _MUJOCO_GL == 'osmesa':
from mujoco.osmesa import GLContext
elif _SYSTEM == 'Linux' and _MUJOCO_GL == 'egl':
from mujoco.egl import GLContext
else:
_dl_handle = ctypes.CDLL(None)
if hasattr(_dl_handle, 'OSMesaCreateContextExt'):
from mujoco.osmesa import GLContext
elif hasattr(_dl_handle, 'eglCreateContext'):
from mujoco.egl import GLContext
else:
from mujoco.glfw import GLContext
from mujoco.glfw import GLContext
HEADERS_DIR = os.path.join(os.path.dirname(__file__), 'include')
__version__ = mj_versionString() # pylint: disable=undefined-variable
+66 -64
View File
@@ -665,7 +665,7 @@ Euler integrator, semi-implicit in velocity.
def test_enum_as_index(self):
x = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k']
self.assertEqual(x[mujoco.mjtFrame.mjNFRAME], 'h')
self.assertEqual(x[mujoco.mjtFrame.mjFRAME_WORLD], 'h')
self.assertEqual(
x[mujoco.mjtFrame.mjFRAME_GEOM:mujoco.mjtFrame.mjFRAME_CAMERA],
['c', 'd'])
@@ -673,83 +673,85 @@ Euler integrator, semi-implicit in velocity.
def test_enum_ops(self):
# Note: when modifying this test, make sure the enum value is an odd number
# so that the division tests are correctly exercised.
self.assertEqual(mujoco.mjtFrame.mjNFRAME, 7)
self.assertEqual(mujoco.mjtFrame.mjNFRAME, 7.0)
self.assertEqual(7, mujoco.mjtFrame.mjNFRAME)
self.assertEqual(7.0, mujoco.mjtFrame.mjNFRAME)
self.assertEqual(mujoco.mjtFrame.mjNFRAME, mujoco.mjtFrame.mjNFRAME)
self.assertNotEqual(mujoco.mjtFrame.mjNFRAME, mujoco.mjtFrame.mjFRAME_NONE)
self.assertEqual(mujoco.mjtFrame.mjFRAME_WORLD, 7)
self.assertEqual(mujoco.mjtFrame.mjFRAME_WORLD, 7.0)
self.assertEqual(7, mujoco.mjtFrame.mjFRAME_WORLD)
self.assertEqual(7.0, mujoco.mjtFrame.mjFRAME_WORLD)
self.assertEqual(mujoco.mjtFrame.mjFRAME_WORLD,
mujoco.mjtFrame.mjFRAME_WORLD)
self.assertNotEqual(mujoco.mjtFrame.mjFRAME_WORLD,
mujoco.mjtFrame.mjFRAME_NONE)
self.assertEqual(-mujoco.mjtFrame.mjNFRAME, -7)
self.assertIsInstance(-mujoco.mjtFrame.mjNFRAME, int)
self.assertEqual(-mujoco.mjtFrame.mjFRAME_WORLD, -7)
self.assertIsInstance(-mujoco.mjtFrame.mjFRAME_WORLD, int)
self.assertEqual(mujoco.mjtFrame.mjNFRAME + 1, 8)
self.assertIsInstance(mujoco.mjtFrame.mjNFRAME + 1, int)
self.assertEqual(2 + mujoco.mjtFrame.mjNFRAME, 9)
self.assertIsInstance(2 + mujoco.mjtFrame.mjNFRAME, int)
self.assertEqual(mujoco.mjtFrame.mjNFRAME + 1.75, 8.75)
self.assertEqual(2.75 + mujoco.mjtFrame.mjNFRAME, 9.75)
self.assertEqual(mujoco.mjtFrame.mjFRAME_WORLD + 1, 8)
self.assertIsInstance(mujoco.mjtFrame.mjFRAME_WORLD + 1, int)
self.assertEqual(2 + mujoco.mjtFrame.mjFRAME_WORLD, 9)
self.assertIsInstance(2 + mujoco.mjtFrame.mjFRAME_WORLD, int)
self.assertEqual(mujoco.mjtFrame.mjFRAME_WORLD + 1.75, 8.75)
self.assertEqual(2.75 + mujoco.mjtFrame.mjFRAME_WORLD, 9.75)
self.assertEqual(mujoco.mjtFrame.mjNFRAME - 2, 5)
self.assertIsInstance(mujoco.mjtFrame.mjNFRAME - 2, int)
self.assertEqual(8 - mujoco.mjtFrame.mjNFRAME, 1)
self.assertIsInstance(8 - mujoco.mjtFrame.mjNFRAME, int)
self.assertEqual(mujoco.mjtFrame.mjNFRAME - 2.25, 4.75)
self.assertEqual(8.25 - mujoco.mjtFrame.mjNFRAME, 1.25)
self.assertEqual(mujoco.mjtFrame.mjFRAME_WORLD - 2, 5)
self.assertIsInstance(mujoco.mjtFrame.mjFRAME_WORLD - 2, int)
self.assertEqual(8 - mujoco.mjtFrame.mjFRAME_WORLD, 1)
self.assertIsInstance(8 - mujoco.mjtFrame.mjFRAME_WORLD, int)
self.assertEqual(mujoco.mjtFrame.mjFRAME_WORLD - 2.25, 4.75)
self.assertEqual(8.25 - mujoco.mjtFrame.mjFRAME_WORLD, 1.25)
self.assertEqual(mujoco.mjtFrame.mjNFRAME * 3, 21)
self.assertIsInstance(mujoco.mjtFrame.mjNFRAME * 3, int)
self.assertEqual(3 * mujoco.mjtFrame.mjNFRAME, 21)
self.assertIsInstance(3 * mujoco.mjtFrame.mjNFRAME, int)
self.assertEqual(mujoco.mjtFrame.mjNFRAME * 3.5, 24.5)
self.assertEqual(3.5 * mujoco.mjtFrame.mjNFRAME, 24.5)
self.assertEqual(mujoco.mjtFrame.mjFRAME_WORLD * 3, 21)
self.assertIsInstance(mujoco.mjtFrame.mjFRAME_WORLD * 3, int)
self.assertEqual(3 * mujoco.mjtFrame.mjFRAME_WORLD, 21)
self.assertIsInstance(3 * mujoco.mjtFrame.mjFRAME_WORLD, int)
self.assertEqual(mujoco.mjtFrame.mjFRAME_WORLD * 3.5, 24.5)
self.assertEqual(3.5 * mujoco.mjtFrame.mjFRAME_WORLD, 24.5)
self.assertEqual(mujoco.mjtFrame.mjNFRAME / 2, 3.5)
self.assertEqual(17.5 / mujoco.mjtFrame.mjNFRAME, 2.5)
self.assertEqual(mujoco.mjtFrame.mjNFRAME // 2, 3)
self.assertIsInstance(mujoco.mjtFrame.mjNFRAME // 2, int)
self.assertEqual(-mujoco.mjtFrame.mjNFRAME // 2, -4)
self.assertIsInstance(-mujoco.mjtFrame.mjNFRAME // 2, int)
self.assertEqual(20 // mujoco.mjtFrame.mjNFRAME, 2)
self.assertIsInstance(20 // mujoco.mjtFrame.mjNFRAME, int)
self.assertEqual(-20 // mujoco.mjtFrame.mjNFRAME, -3)
self.assertIsInstance(-20 // mujoco.mjtFrame.mjNFRAME, int)
self.assertEqual(mujoco.mjtFrame.mjNFRAME // 2.0, 3)
self.assertIsInstance(mujoco.mjtFrame.mjNFRAME // 2.0, float)
self.assertEqual(-mujoco.mjtFrame.mjNFRAME // 2.0, -4)
self.assertIsInstance(-mujoco.mjtFrame.mjNFRAME // 2.0, float)
self.assertEqual(20.0 // mujoco.mjtFrame.mjNFRAME, 2)
self.assertIsInstance(20.0 // mujoco.mjtFrame.mjNFRAME, float)
self.assertEqual(-20 // mujoco.mjtFrame.mjNFRAME, -3)
self.assertIsInstance(-20.0 // mujoco.mjtFrame.mjNFRAME, float)
self.assertEqual(mujoco.mjtFrame.mjFRAME_WORLD / 2, 3.5)
self.assertEqual(17.5 / mujoco.mjtFrame.mjFRAME_WORLD, 2.5)
self.assertEqual(mujoco.mjtFrame.mjFRAME_WORLD // 2, 3)
self.assertIsInstance(mujoco.mjtFrame.mjFRAME_WORLD // 2, int)
self.assertEqual(-mujoco.mjtFrame.mjFRAME_WORLD // 2, -4)
self.assertIsInstance(-mujoco.mjtFrame.mjFRAME_WORLD // 2, int)
self.assertEqual(20 // mujoco.mjtFrame.mjFRAME_WORLD, 2)
self.assertIsInstance(20 // mujoco.mjtFrame.mjFRAME_WORLD, int)
self.assertEqual(-20 // mujoco.mjtFrame.mjFRAME_WORLD, -3)
self.assertIsInstance(-20 // mujoco.mjtFrame.mjFRAME_WORLD, int)
self.assertEqual(mujoco.mjtFrame.mjFRAME_WORLD // 2.0, 3)
self.assertIsInstance(mujoco.mjtFrame.mjFRAME_WORLD // 2.0, float)
self.assertEqual(-mujoco.mjtFrame.mjFRAME_WORLD // 2.0, -4)
self.assertIsInstance(-mujoco.mjtFrame.mjFRAME_WORLD // 2.0, float)
self.assertEqual(20.0 // mujoco.mjtFrame.mjFRAME_WORLD, 2)
self.assertIsInstance(20.0 // mujoco.mjtFrame.mjFRAME_WORLD, float)
self.assertEqual(-20 // mujoco.mjtFrame.mjFRAME_WORLD, -3)
self.assertIsInstance(-20.0 // mujoco.mjtFrame.mjFRAME_WORLD, float)
self.assertEqual(mujoco.mjtFrame.mjNFRAME % 4, 3)
self.assertIsInstance(mujoco.mjtFrame.mjNFRAME % 4, int)
self.assertEqual(-mujoco.mjtFrame.mjNFRAME % -4, -3)
self.assertIsInstance(-mujoco.mjtFrame.mjNFRAME % -4, int)
self.assertEqual(-mujoco.mjtFrame.mjNFRAME % 4, 1)
self.assertIsInstance(-mujoco.mjtFrame.mjNFRAME % 4, int)
self.assertEqual(mujoco.mjtFrame.mjNFRAME % -4, -1)
self.assertIsInstance(mujoco.mjtFrame.mjNFRAME % -4, int)
self.assertEqual(9 % mujoco.mjtFrame.mjNFRAME, 2)
self.assertIsInstance(9 % mujoco.mjtFrame.mjNFRAME, int)
self.assertEqual(-9 % -mujoco.mjtFrame.mjNFRAME, -2)
self.assertIsInstance(-9 % -mujoco.mjtFrame.mjNFRAME, int)
self.assertEqual(-9 % mujoco.mjtFrame.mjNFRAME, 5)
self.assertIsInstance(-9 % mujoco.mjtFrame.mjNFRAME, int)
self.assertEqual(9 % -mujoco.mjtFrame.mjNFRAME, -5)
self.assertIsInstance(9 % -mujoco.mjtFrame.mjNFRAME, int)
self.assertEqual(mujoco.mjtFrame.mjFRAME_WORLD % 4, 3)
self.assertIsInstance(mujoco.mjtFrame.mjFRAME_WORLD % 4, int)
self.assertEqual(-mujoco.mjtFrame.mjFRAME_WORLD % -4, -3)
self.assertIsInstance(-mujoco.mjtFrame.mjFRAME_WORLD % -4, int)
self.assertEqual(-mujoco.mjtFrame.mjFRAME_WORLD % 4, 1)
self.assertIsInstance(-mujoco.mjtFrame.mjFRAME_WORLD % 4, int)
self.assertEqual(mujoco.mjtFrame.mjFRAME_WORLD % -4, -1)
self.assertIsInstance(mujoco.mjtFrame.mjFRAME_WORLD % -4, int)
self.assertEqual(9 % mujoco.mjtFrame.mjFRAME_WORLD, 2)
self.assertIsInstance(9 % mujoco.mjtFrame.mjFRAME_WORLD, int)
self.assertEqual(-9 % -mujoco.mjtFrame.mjFRAME_WORLD, -2)
self.assertIsInstance(-9 % -mujoco.mjtFrame.mjFRAME_WORLD, int)
self.assertEqual(-9 % mujoco.mjtFrame.mjFRAME_WORLD, 5)
self.assertIsInstance(-9 % mujoco.mjtFrame.mjFRAME_WORLD, int)
self.assertEqual(9 % -mujoco.mjtFrame.mjFRAME_WORLD, -5)
self.assertIsInstance(9 % -mujoco.mjtFrame.mjFRAME_WORLD, int)
with self.assertRaises(ZeroDivisionError):
_ = mujoco.mjtFrame.mjNFRAME / 0
_ = mujoco.mjtFrame.mjFRAME_WORLD / 0
with self.assertRaises(ZeroDivisionError):
_ = 1 / mujoco.mjtFrame.mjFRAME_NONE
with self.assertRaises(ZeroDivisionError):
_ = mujoco.mjtFrame.mjNFRAME // 0
_ = mujoco.mjtFrame.mjFRAME_WORLD // 0
with self.assertRaises(ZeroDivisionError):
_ = 1 // mujoco.mjtFrame.mjFRAME_NONE
with self.assertRaises(ZeroDivisionError):
_ = mujoco.mjtFrame.mjNFRAME % 0
_ = mujoco.mjtFrame.mjFRAME_WORLD % 0
with self.assertRaises(ZeroDivisionError):
_ = 1 % mujoco.mjtFrame.mjFRAME_NONE
+1 -1
View File
@@ -206,7 +206,7 @@ PYBIND11_MODULE(_render, pymodule) {
X(nskin);
X(charHeight);
X(charHeightBig);
X(glewInitialized);
X(glInitialized);
X(windowAvailable);
X(windowSamples);
X(windowStereo);
+1 -1
View File
@@ -19,7 +19,7 @@ import mujoco
import numpy as np
@absltest.skipUnless(hasattr(mujoco, 'mjr_render'),
@absltest.skipUnless(hasattr(mujoco, 'GLContext'),
'MuJoCo rendering is disabled')
class MuJoCoRenderTest(absltest.TestCase):
+11 -13
View File
@@ -30,7 +30,7 @@ from setuptools import find_packages
from setuptools import setup
from setuptools.command import build_ext
__version__ = '2.1.3'
__version__ = '2.1.4'
MUJOCO_CMAKE = 'MUJOCO_CMAKE'
MUJOCO_CMAKE_ARGS = 'MUJOCO_CMAKE_ARGS'
@@ -54,20 +54,20 @@ def get_long_description():
def get_mujoco_lib_pattern():
if platform.system() == 'Windows':
return 'mujoco*.lib'
return 'mujoco.lib'
elif platform.system() == 'Darwin':
return 'libmujoco*.dylib'
return 'libmujoco.*.dylib'
else:
return 'libmujoco*.so*'
return 'libmujoco.so.*'
def get_external_lib_patterns():
if platform.system() == 'Windows':
return ['mujoco*.dll']
return ['mujoco.dll']
elif platform.system() == 'Darwin':
return ['libmujoco*.dylib']
return ['libmujoco.*.dylib']
else:
return ['libmujoco*.so*', 'libglew*.so']
return ['libmujoco.so.*']
def start_and_end(iterable):
@@ -290,12 +290,10 @@ setup(
find_data_files(
package_dir='mujoco',
patterns=[
'libmujoco*.dylib',
'libmujoco*.so*',
'mujoco*.dll',
'libglew*.so*',
'mujoco.h',
'mj*.h',
'libmujoco.*.dylib',
'libmujoco*.so.*',
'mujoco.dll',
'include/*.h',
]),
},
)