Difficult-Rocket/libs/pyglet/gl/gl_info.py
2022-05-25 09:16:38 +08:00

234 lines
8.0 KiB
Python

# ----------------------------------------------------------------------------
# pyglet
# Copyright (c) 2006-2008 Alex Holkner
# Copyright (c) 2008-2022 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.
# ----------------------------------------------------------------------------
"""Information about version and extensions of current GL implementation.
Usage::
from pyglet.gl import gl_info
if gl_info.have_extension('GL_NV_register_combiners'):
# ...
If you are using more than one context, you can set up a separate GLInfo
object for each context. Call `set_active_context` after switching to the
context::
from pyglet.gl.gl_info import GLInfo
info = GLInfo()
info.set_active_context()
if info.have_version(4, 5):
# ...
"""
from ctypes import c_char_p, cast, c_int
import warnings
from pyglet.gl.gl import (GL_EXTENSIONS, GL_RENDERER, GL_VENDOR,
GL_VERSION, GL_MAJOR_VERSION, GL_MINOR_VERSION)
from pyglet.util import asstr
class GLInfo:
"""Information interface for a single GL context.
A default instance is created automatically when the first OpenGL context
is created. You can use the module functions as a convenience for
this default instance's methods.
If you are using more than one context, you must call `set_active_context`
when the context is active for this `GLInfo` instance.
"""
have_context = False
version = '0.0'
vendor = ''
renderer = ''
extensions = set()
_have_info = False
def set_active_context(self):
"""Store information for the currently active context.
This method is called automatically for the default context.
"""
from pyglet.gl.gl import GLint, glGetIntegerv, glGetString, glGetStringi, GL_NUM_EXTENSIONS
self.have_context = True
if not self._have_info:
self.vendor = asstr(cast(glGetString(GL_VENDOR), c_char_p).value)
self.renderer = asstr(cast(glGetString(GL_RENDERER), c_char_p).value)
major_version = c_int()
glGetIntegerv(GL_MAJOR_VERSION, major_version)
self.major_version = major_version.value
minor_version = c_int()
glGetIntegerv(GL_MINOR_VERSION, minor_version)
self.minor_version = minor_version.value
self.version = asstr(cast(glGetString(GL_VERSION), c_char_p).value)
# NOTE: The version string requirements for gles is a lot stricter
# so using this to rely on detecting the API is not too unreasonable
self.opengl_api = "gles" if "opengl es" in self.version.lower() else "gl"
num_extensions = GLint()
glGetIntegerv(GL_NUM_EXTENSIONS, num_extensions)
self.extensions = (asstr(cast(glGetStringi(GL_EXTENSIONS, i), c_char_p).value)
for i in range(num_extensions.value))
self.extensions = set(self.extensions)
self._have_info = True
def remove_active_context(self):
self.have_context = False
self._have_info = False
def have_extension(self, extension):
"""Determine if an OpenGL extension is available.
:Parameters:
`extension` : str
The name of the extension to test for, including its
``GL_`` prefix.
:return: True if the extension is provided by the driver.
:rtype: bool
"""
if not self.have_context:
warnings.warn('No GL context created yet.')
return extension in self.extensions
def get_extensions(self):
"""Get a list of available OpenGL extensions.
:return: a list of the available extensions.
:rtype: list of str
"""
if not self.have_context:
warnings.warn('No GL context created yet.')
return self.extensions
def get_version(self):
"""Get the current OpenGL version.
:return: The major and minor version as a tuple
:rtype: tuple
"""
if not self.have_context:
warnings.warn('No GL context created yet.')
return self.major_version, self.minor_version
def get_version_string(self):
"""Get the current OpenGL version string.
:return: The OpenGL version string
:rtype: str
"""
if not self.have_context:
warnings.warn('No GL context created yet.')
return self.version
def have_version(self, major, minor=0):
"""Determine if a version of OpenGL is supported.
:Parameters:
`major` : int
The major revision number (typically 1 or 2).
`minor` : int
The minor revision number.
:rtype: bool
:return: True if the requested or a later version is supported.
"""
if not self.have_context:
warnings.warn('No GL context created yet.')
if not self.major_version and not self.minor_version:
return False
return (self.major_version > major or
(self.major_version == major and self.minor_version >= minor) or
(self.major_version == major and self.minor_version == minor))
def get_renderer(self):
"""Determine the renderer string of the OpenGL context.
:rtype: str
"""
if not self.have_context:
warnings.warn('No GL context created yet.')
return self.renderer
def get_vendor(self):
"""Determine the vendor string of the OpenGL context.
:rtype: str
"""
if not self.have_context:
warnings.warn('No GL context created yet.')
return self.vendor
def get_opengl_api(self):
"""Determine the OpenGL API version.
Usually ``gl`` or ``gles``.
:rtype: str
"""
if not self.have_context:
warnings.warn('No GL context created yet.')
return self.opengl_api
# Single instance useful for apps with only a single context
# (or all contexts have the same GL driver, a common case).
_gl_info = GLInfo()
set_active_context = _gl_info.set_active_context
remove_active_context = _gl_info.remove_active_context
have_extension = _gl_info.have_extension
get_extensions = _gl_info.get_extensions
get_version = _gl_info.get_version
get_version_string = _gl_info.get_version_string
have_version = _gl_info.have_version
get_renderer = _gl_info.get_renderer
get_vendor = _gl_info.get_vendor
get_opengl_api = _gl_info.get_opengl_api
def have_context():
"""Determine if a default OpenGL context has been set yet.
:rtype: bool
"""
return _gl_info.have_context