Saran Tunyasuvunakool
4e67fe79f3
According to `man dyld`, the default value for DYLD_FALLBACK_LIBRARY_PATH changed in late 2023. In newer binaries, if DYLD_FALLBACK_LIBRARY_PATH is not set, dyld will use /usr/local/lib and /usr/lib as fallback. If it is set, these locations will not be used as lookup paths unless explicitly added. Fixes #2781 PiperOrigin-RevId: 791897707 Change-Id: I00d7e80bfbd5e5d7dbf2f601eaad4d9b0e1030f3
101 lines
3.9 KiB
Python
101 lines
3.9 KiB
Python
#!/usr/bin/env python
|
|
# Copyright 2023 DeepMind Technologies Limited
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
# ==============================================================================
|
|
"""Python interpreter trampoline for macOS to support non-block Cocoa GUI.
|
|
|
|
This script executes a native binary that runs the CPython interpreter entry
|
|
point in a separate thread, thus leaving the macOS main thread free for Cocoa
|
|
GUI calls without blocking the user's Python script. In other words, Python's
|
|
idea of the "main thread" is different from the thread that holds the
|
|
com.apple.main-thread DispatchQueue.
|
|
"""
|
|
import ctypes
|
|
import importlib.util
|
|
import os
|
|
import platform
|
|
import re
|
|
import subprocess
|
|
import sys
|
|
|
|
if platform.system() != 'Darwin':
|
|
raise RuntimeError('This script only works on macOS')
|
|
|
|
_NSGetExecutablePath = getattr(ctypes.CDLL(None), '_NSGetExecutablePath')
|
|
|
|
|
|
def get_executable_path():
|
|
c_path_size = ctypes.c_int32(0)
|
|
_NSGetExecutablePath(None, ctypes.byref(c_path_size))
|
|
c_path = (ctypes.c_char * c_path_size.value)()
|
|
_NSGetExecutablePath(ctypes.byref(c_path), ctypes.byref(c_path_size))
|
|
return c_path.value.decode()
|
|
|
|
|
|
def main(argv):
|
|
module_dir = os.path.dirname(importlib.util.find_spec('mujoco').origin)
|
|
os.environ['MJPYTHON_BIN'] = os.path.join(
|
|
module_dir, 'MuJoCo_(mjpython).app/Contents/MacOS/mjpython')
|
|
|
|
# Conda doesn't create a separate shared library for Python.
|
|
# We instead use the Python binary itself, which can be dlopened just as well.
|
|
libpython_path = get_executable_path()
|
|
os.environ['MJPYTHON_LIBPYTHON'] = libpython_path
|
|
|
|
# In some installations (e.g. CommandLineTools), the Python interpreter loads
|
|
# dylibs from @executable_path-relative paths. This will not resolve
|
|
# correctly since @executable_path will be the directory containing the
|
|
# mjpython binary when we execve. We therefore preemptively resolve all
|
|
# @executable_path-relative paths now and add them to
|
|
# DYLD_FALLBACK_LIBRARY_PATH.
|
|
libpython_dir = os.path.dirname(libpython_path)
|
|
dyld_fallback_paths = []
|
|
pattern = re.compile(r'@executable_path/(.+) \(offset \d+\)\Z')
|
|
otool_out = subprocess.run(
|
|
['otool', '-l', libpython_path],
|
|
capture_output=True,
|
|
check=True,
|
|
).stdout.decode()
|
|
for line in otool_out.split('\n'):
|
|
m = pattern.search(line)
|
|
if m is not None:
|
|
new_path = os.path.dirname(os.path.join(libpython_dir, m.group(1)))
|
|
if new_path not in dyld_fallback_paths:
|
|
dyld_fallback_paths.insert(0, new_path)
|
|
|
|
# According to `man dyld`, the default value for DYLD_FALLBACK_LIBRARY_PATH
|
|
# changed in late 2023. In newer binaries, if DYLD_FALLBACK_LIBRARY_PATH is
|
|
# not set, dyld will use /usr/local/lib and /usr/lib as fallback. If it is
|
|
# set, these locations will not be used as lookup paths unless explicitly
|
|
# added.
|
|
if dyld_fallback_paths:
|
|
if 'DYLD_FALLBACK_LIBRARY_PATH' in os.environ:
|
|
old_paths = os.environ['DYLD_FALLBACK_LIBRARY_PATH']
|
|
# Empty string splits to [''] but we don't want to append ''.
|
|
if old_paths:
|
|
dyld_fallback_paths.extend(old_paths.split(':'))
|
|
else:
|
|
dyld_fallback_paths.extend(('/usr/local/lib', '/usr/lib'))
|
|
os.environ['DYLD_FALLBACK_LIBRARY_PATH'] = ':'.join(dyld_fallback_paths)
|
|
|
|
# argv[0] is currently the path to this script.
|
|
# Replace it with sys.executable to preserve e.g. virtualenv path.
|
|
argv[0] = sys.executable
|
|
|
|
os.execve(os.environ['MJPYTHON_BIN'], argv, os.environ)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main(sys.argv)
|