a ij@sdZddlZddlZddlZejdZes8dejd<nedkrLedddl m Z ddl m Z dd Zdae jd e jd e jd e jd e jd e jd e je je je je je je jfZGd d d ZdS)HAn EGL context for headless accelerated OpenGL rendering on GPU devices.NPYOPENGL_PLATFORMZeglzCannot use EGL rendering platform. The PYOPENGL_PLATFORM environment variable is set to {!r} (should be either unset or 'egl').)egl_ext)errorc Cst}tjdd}|dur$|}nPt|}d|krDt|ksdntdt|dd|d|||d}|D]t}ttj |d}|tj krxt tj krxzt |dd}WntjyYqx0|tjkrxt tj krx|Sqxtj S)z8Creates an initialized EGL display directly on a device.ZMUJOCO_EGL_DEVICE_IDNrzOThe MUJOCO_EGL_DEVICE_ID environment variable must be an integer between 0 and z (inclusive), got .)EGLZeglQueryDevicesEXTosenvirongetintlen RuntimeErrorZeglGetPlatformDisplayEXTZEGL_PLATFORM_DEVICE_EXTEGL_NO_DISPLAYZ eglGetErrorZ EGL_SUCCESSZ eglInitializerZGLErrorZEGL_TRUE)Z all_devicesZselected_device candidatesZ device_idxZdevicedisplayZ initializedrS/home/ghrups/robot-bench/.venv/lib64/python3.9/site-packages/mujoco/egl/__init__.py%create_initialized_egl_device_display#s4  rc@s0eZdZdZddZddZddZdd Zd S) GLContextrcCs~~t}d}tj|}ttdurRtattjkrDtdt tj tt tt ||||jdkr|tdt ttjtt|dtjd|_|jstddS)NrzCannot initialize a EGL device display. This likely means that your EGL driver does not support the PLATFORM_DEVICE extension, which is required for creating a headless rendering context.zVEGL failed to find a framebuffer configuration that matches the desired attributes: {}rzCannot create an EGL context.)ctypesZc_longrZ EGLConfigeglReleaseThread EGL_DISPLAYrr ImportErroratexitregisterZ eglTerminateZeglChooseConfigEGL_ATTRIBUTESvaluerformatZ eglBindAPIZEGL_OPENGL_APIZeglCreateContextEGL_NO_CONTEXT_context)self max_width max_heightZ num_configsZ config_sizeZconfigsrrr__init__Us>    zGLContext.__init__cCs"tttjtj|jstddS)Nz'Failed to make the EGL context current.)reglMakeCurrentrEGL_NO_SURFACEr"rr#rrr make_currentwszGLContext.make_currentcCsV|jrLt}|r6|jj|jkr6tttjtjtjtt|jt d|_dS)z-Frees resources associated with this context.N) r"rZeglGetCurrentContextaddressr'rr(r!ZeglDestroyContextr)r#Zcurrent_contextrrrfree}s zGLContext.freecCs |dS)N)r,r)rrr__del__szGLContext.__del__N)__name__ __module__ __qualname____doc__r&r*r,r-rrrrrRs " r)r1rrr r r rlowerrZ mujoco.eglrrZOpenGLrrrZ EGL_RED_SIZEZEGL_GREEN_SIZEZ EGL_BLUE_SIZEZEGL_ALPHA_SIZEZEGL_DEPTH_SIZEZEGL_STENCIL_SIZEZEGL_COLOR_BUFFER_TYPEZEGL_RGB_BUFFERZEGL_SURFACE_TYPEZEGL_PBUFFER_BITZEGL_RENDERABLE_TYPEZEGL_OPENGL_BITZEGL_NONErrrrrrs4