185 lines
6.7 KiB
Python
185 lines
6.7 KiB
Python
|
# ----------------------------------------------------------------------------
|
||
|
|
# pyglet
|
||
|
|
# Copyright (c) 2006-2008 Alex Holkner
|
||
|
|
# Copyright (c) 2008-2020 pyglet contributors
|
||
|
|
# All rights reserved.
|
||
|
|
#
|
||
|
|
# Redistribution and use in source and binary forms, with or without
|
||
|
|
# modification, are permitted provided that the following conditions
|
||
|
|
# are met:
|
||
|
|
#
|
||
|
|
# * Redistributions of source code must retain the above copyright
|
||
|
|
# notice, this list of conditions and the following disclaimer.
|
||
|
|
# * Redistributions in binary form must reproduce the above copyright
|
||
|
|
# notice, this list of conditions and the following disclaimer in
|
||
|
|
# the documentation and/or other materials provided with the
|
||
|
|
# distribution.
|
||
|
|
# * Neither the name of pyglet nor the names of its
|
||
|
|
# contributors may be used to endorse or promote products
|
||
|
|
# derived from this software without specific prior written
|
||
|
|
# permission.
|
||
|
|
#
|
||
|
|
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
||
|
|
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
||
|
|
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
|
||
|
|
# FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
|
||
|
|
# COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
|
||
|
|
# INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
|
||
|
|
# BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
|
||
|
|
# LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
|
||
|
|
# CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
|
||
|
|
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
|
||
|
|
# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
||
|
|
# POSSIBILITY OF SUCH DAMAGE.
|
||
|
|
# ----------------------------------------------------------------------------
|
||
|
|
|
||
|
|
import warnings
|
||
|
|
from ctypes import *
|
||
|
|
|
||
|
|
from .base import Config, CanvasConfig, Context
|
||
|
|
from pyglet.canvas.headless import HeadlessCanvas
|
||
|
|
from pyglet.libs.egl import egl
|
||
|
|
from pyglet.libs.egl.egl import *
|
||
|
|
from pyglet import gl
|
||
|
|
|
||
|
|
|
||
|
|
_fake_gl_attributes = {
|
||
|
|
'double_buffer': 0,
|
||
|
|
'stereo': 0,
|
||
|
|
'aux_buffers': 0,
|
||
|
|
'accum_red_size': 0,
|
||
|
|
'accum_green_size': 0,
|
||
|
|
'accum_blue_size': 0,
|
||
|
|
'accum_alpha_size': 0
|
||
|
|
}
|
||
|
|
|
||
|
|
class HeadlessConfig(Config):
|
||
|
|
def match(self, canvas):
|
||
|
|
if not isinstance(canvas, HeadlessCanvas):
|
||
|
|
raise RuntimeError('Canvas must be an instance of HeadlessCanvas')
|
||
|
|
|
||
|
|
display_connection = canvas.display._display_connection
|
||
|
|
|
||
|
|
# Construct array of attributes
|
||
|
|
attrs = []
|
||
|
|
for name, value in self.get_gl_attributes():
|
||
|
|
if name == 'double_buffer':
|
||
|
|
continue
|
||
|
|
attr = HeadlessCanvasConfig.attribute_ids.get(name, None)
|
||
|
|
if attr and value is not None:
|
||
|
|
attrs.extend([attr, int(value)])
|
||
|
|
attrs.extend([EGL_SURFACE_TYPE, EGL_PBUFFER_BIT])
|
||
|
|
attrs.extend([EGL_RENDERABLE_TYPE, EGL_OPENGL_BIT])
|
||
|
|
attrs.extend([EGL_NONE])
|
||
|
|
attrs_list = (egl.EGLint * len(attrs))(*attrs)
|
||
|
|
|
||
|
|
num_config = egl.EGLint()
|
||
|
|
egl.eglChooseConfig(display_connection, attrs_list, None, 0, byref(num_config))
|
||
|
|
configs = (egl.EGLConfig * num_config.value)()
|
||
|
|
egl.eglChooseConfig(display_connection, attrs_list, configs,
|
||
|
|
num_config.value, byref(num_config))
|
||
|
|
|
||
|
|
result = [HeadlessCanvasConfig(canvas, c, self) for c in configs]
|
||
|
|
return result
|
||
|
|
|
||
|
|
|
||
|
|
class HeadlessCanvasConfig(CanvasConfig):
|
||
|
|
attribute_ids = {
|
||
|
|
'buffer_size': egl.EGL_BUFFER_SIZE,
|
||
|
|
'level': egl.EGL_LEVEL, # Not supported
|
||
|
|
'red_size': egl.EGL_RED_SIZE,
|
||
|
|
'green_size': egl.EGL_GREEN_SIZE,
|
||
|
|
'blue_size': egl.EGL_BLUE_SIZE,
|
||
|
|
'alpha_size': egl.EGL_ALPHA_SIZE,
|
||
|
|
'depth_size': egl.EGL_DEPTH_SIZE,
|
||
|
|
'stencil_size': egl.EGL_STENCIL_SIZE,
|
||
|
|
'sample_buffers': egl.EGL_SAMPLE_BUFFERS,
|
||
|
|
'samples': egl.EGL_SAMPLES,
|
||
|
|
}
|
||
|
|
|
||
|
|
def __init__(self, canvas, egl_config, config):
|
||
|
|
super(HeadlessCanvasConfig, self).__init__(canvas, config)
|
||
|
|
self._egl_config = egl_config
|
||
|
|
context_attribs = (EGL_CONTEXT_MAJOR_VERSION, config.major_version or 2,
|
||
|
|
EGL_CONTEXT_MINOR_VERSION, config.minor_version or 0,
|
||
|
|
EGL_CONTEXT_OPENGL_FORWARD_COMPATIBLE, config.forward_compatible or 0,
|
||
|
|
EGL_CONTEXT_OPENGL_DEBUG, config.debug or 0,
|
||
|
|
EGL_NONE)
|
||
|
|
self._context_attrib_array = (egl.EGLint * len(context_attribs))(*context_attribs)
|
||
|
|
|
||
|
|
for name, attr in self.attribute_ids.items():
|
||
|
|
value = egl.EGLint()
|
||
|
|
egl.eglGetConfigAttrib(canvas.display._display_connection, egl_config, attr, byref(value))
|
||
|
|
setattr(self, name, value.value)
|
||
|
|
|
||
|
|
for name, value in _fake_gl_attributes.items():
|
||
|
|
setattr(self, name, value)
|
||
|
|
|
||
|
|
def compatible(self, canvas):
|
||
|
|
# TODO check more
|
||
|
|
return isinstance(canvas, HeadlessCanvas)
|
||
|
|
|
||
|
|
def create_context(self, share):
|
||
|
|
return HeadlessContext(self, share)
|
||
|
|
|
||
|
|
|
||
|
|
class HeadlessContext(Context):
|
||
|
|
def __init__(self, config, share):
|
||
|
|
super(HeadlessContext, self).__init__(config, share)
|
||
|
|
|
||
|
|
self.display_connection = config.canvas.display._display_connection
|
||
|
|
|
||
|
|
self.egl_context = self._create_egl_context(share)
|
||
|
|
if not self.egl_context:
|
||
|
|
raise gl.ContextException('Could not create GL context')
|
||
|
|
|
||
|
|
def _create_egl_context(self, share):
|
||
|
|
if share:
|
||
|
|
share_context = share.egl_context
|
||
|
|
else:
|
||
|
|
share_context = None
|
||
|
|
|
||
|
|
egl.eglBindAPI(egl.EGL_OPENGL_API)
|
||
|
|
return egl.eglCreateContext(self.config.canvas.display._display_connection,
|
||
|
|
self.config._egl_config, share_context,
|
||
|
|
self.config._context_attrib_array)
|
||
|
|
|
||
|
|
def attach(self, canvas):
|
||
|
|
if canvas is self.canvas:
|
||
|
|
return
|
||
|
|
|
||
|
|
super(HeadlessContext, self).attach(canvas)
|
||
|
|
|
||
|
|
self.egl_surface = canvas.egl_surface
|
||
|
|
self.set_current()
|
||
|
|
|
||
|
|
def set_current(self):
|
||
|
|
egl.eglMakeCurrent(
|
||
|
|
self.display_connection, self.egl_surface, self.egl_surface, self.egl_context)
|
||
|
|
super(HeadlessContext, self).set_current()
|
||
|
|
|
||
|
|
def detach(self):
|
||
|
|
if not self.canvas:
|
||
|
|
return
|
||
|
|
|
||
|
|
self.set_current()
|
||
|
|
gl.glFlush() # needs to be in try/except?
|
||
|
|
|
||
|
|
super(HeadlessContext, self).detach()
|
||
|
|
|
||
|
|
egl.eglMakeCurrent(
|
||
|
|
self.display_connection, 0, 0, None)
|
||
|
|
self.egl_surface = None
|
||
|
|
|
||
|
|
def destroy(self):
|
||
|
|
super(HeadlessContext, self).destroy()
|
||
|
|
if self.egl_context:
|
||
|
|
egl.eglDestroyContext(self.display_connection, self.egl_context)
|
||
|
|
self.egl_context = None
|
||
|
|
|
||
|
|
def flip(self):
|
||
|
|
if not self.egl_surface:
|
||
|
|
return
|
||
|
|
|
||
|
|
egl.eglSwapBuffers(self.display_connection, self.egl_surface)
|