Difficult-Rocket/libs/pyglet/image/__init__.py

2155 lines
74 KiB
Python
Raw Normal View History

2021-04-16 23:21:06 +08:00
# ----------------------------------------------------------------------------
# pyglet
# Copyright (c) 2006-2008 Alex Holkner
2022-04-30 13:56:57 +08:00
# Copyright (c) 2008-2022 pyglet contributors
2021-04-16 23:21:06 +08:00
# 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.
# ----------------------------------------------------------------------------
"""Image load, capture and high-level texture functions.
Only basic functionality is described here; for full reference see the
accompanying documentation.
To load an image::
from pyglet import image
pic = image.load('picture.png')
The supported image file types include PNG, BMP, GIF, JPG, and many more,
somewhat depending on the operating system. To load an image from a file-like
object instead of a filename::
pic = image.load('hint.jpg', file=fileobj)
The hint helps the module locate an appropriate decoder to use based on the
file extension. It is optional.
Once loaded, images can be used directly by most other modules of pyglet. All
images have a width and height you can access::
width, height = pic.width, pic.height
You can extract a region of an image (this keeps the original image intact;
the memory is shared efficiently)::
subimage = pic.get_region(x, y, width, height)
Remember that y-coordinates are always increasing upwards.
Drawing images
--------------
To draw an image at some point on the screen::
pic.blit(x, y, z)
This assumes an appropriate view transform and projection have been applied.
Some images have an intrinsic "anchor point": this is the point which will be
aligned to the ``x`` and ``y`` coordinates when the image is drawn. By
default the anchor point is the lower-left corner of the image. You can use
the anchor point to center an image at a given point, for example::
pic.anchor_x = pic.width // 2
pic.anchor_y = pic.height // 2
pic.blit(x, y, z)
Texture access
--------------
If you are using OpenGL directly, you can access the image as a texture::
texture = pic.get_texture()
(This is the most efficient way to obtain a texture; some images are
immediately loaded as textures, whereas others go through an intermediate
form). To use a texture with pyglet.gl::
from pyglet.gl import *
glEnable(texture.target) # typically target is GL_TEXTURE_2D
glBindTexture(texture.target, texture.id)
# ... draw with the texture
Pixel access
------------
To access raw pixel data of an image::
rawimage = pic.get_image_data()
(If the image has just been loaded this will be a very quick operation;
however if the image is a texture a relatively expensive readback operation
will occur). The pixels can be accessed as a string::
format = 'RGBA'
pitch = rawimage.width * len(format)
pixels = rawimage.get_data(format, pitch)
"format" strings consist of characters that give the byte order of each color
component. For example, if rawimage.format is 'RGBA', there are four color
components: red, green, blue and alpha, in that order. Other common format
strings are 'RGB', 'LA' (luminance, alpha) and 'I' (intensity).
The "pitch" of an image is the number of bytes in a row (this may validly be
more than the number required to make up the width of the image, it is common
to see this for word alignment). If "pitch" is negative the rows of the image
are ordered from top to bottom, otherwise they are ordered from bottom to top.
Retrieving data with the format and pitch given in `ImageData.format` and
`ImageData.pitch` avoids the need for data conversion (assuming you can make
use of the data in this arbitrary format).
"""
import re
import weakref
from ctypes import *
from io import open, BytesIO
import pyglet
2021-04-16 23:21:06 +08:00
from pyglet.gl import *
from pyglet.gl import gl_info
from pyglet.util import asbytes
from .codecs import ImageEncodeException, ImageDecodeException
2022-04-08 23:07:41 +08:00
from .codecs import registry as _codec_registry
from .codecs import add_default_codecs as _add_default_codecs
2021-04-16 23:21:06 +08:00
from .animation import Animation, AnimationFrame
from .buffer import *
2021-04-16 23:21:06 +08:00
from . import atlas
class ImageException(Exception):
pass
def load(filename, file=None, decoder=None):
"""Load an image from a file.
:note: You can make no assumptions about the return type; usually it will
be ImageData or CompressedImageData, but decoders are free to return
any subclass of AbstractImage.
:Parameters:
`filename` : str
Used to guess the image format, and to load the file if `file` is
unspecified.
`file` : file-like object or None
Source of image data in any supported format.
2021-04-16 23:21:06 +08:00
`decoder` : ImageDecoder or None
If unspecified, all decoders that are registered for the filename
extension are tried. If none succeed, the exception from the
first decoder is raised.
:rtype: AbstractImage
"""
2022-04-08 23:07:41 +08:00
if decoder:
return decoder.decode(filename, file)
2021-04-16 23:21:06 +08:00
else:
2022-04-08 23:07:41 +08:00
return _codec_registry.decode(filename, file)
2021-04-16 23:21:06 +08:00
def load_animation(filename, file=None, decoder=None):
"""Load an animation from a file.
Currently, the only supported format is GIF.
:Parameters:
`filename` : str
Used to guess the animation format, and to load the file if `file`
is unspecified.
`file` : file-like object or None
File object containing the animation stream.
`decoder` : ImageDecoder or None
If unspecified, all decoders that are registered for the filename
extension are tried. If none succeed, the exception from the
first decoder is raised.
:rtype: Animation
"""
if decoder:
2022-04-08 23:07:41 +08:00
return decoder.decode_animation(filename, file)
2021-04-16 23:21:06 +08:00
else:
2022-04-08 23:07:41 +08:00
return _codec_registry.decode_animation(filename, file)
2021-04-16 23:21:06 +08:00
def create(width, height, pattern=None):
"""Create an image optionally filled with the given pattern.
:note: You can make no assumptions about the return type; usually it will
be ImageData or CompressedImageData, but patterns are free to return
any subclass of AbstractImage.
:Parameters:
`width` : int
Width of image to create
`height` : int
Height of image to create
`pattern` : ImagePattern or None
Pattern to fill image with. If unspecified, the image will
initially be transparent.
:rtype: AbstractImage
"""
if not pattern:
pattern = SolidColorImagePattern()
return pattern.create_image(width, height)
def get_max_texture_size():
"""Query the maximum texture size available"""
size = c_int()
glGetIntegerv(GL_MAX_TEXTURE_SIZE, size)
return size.value
def get_max_array_texture_layers():
"""Query the maximum TextureArray depth"""
max_layers = c_int()
glGetIntegerv(GL_MAX_ARRAY_TEXTURE_LAYERS, max_layers)
return max_layers.value
2021-04-16 23:21:06 +08:00
def _color_as_bytes(color):
if len(color) != 4:
raise TypeError("color is expected to have 4 components")
return bytes(color)
class ImagePattern:
"""Abstract image creation class."""
def create_image(self, width, height):
"""Create an image of the given size.
:Parameters:
`width` : int
Width of image to create
`height` : int
Height of image to create
2021-04-16 23:21:06 +08:00
:rtype: AbstractImage
"""
raise NotImplementedError('abstract')
class SolidColorImagePattern(ImagePattern):
"""Creates an image filled with a solid color."""
def __init__(self, color=(0, 0, 0, 0)):
"""Create a solid image pattern with the given color.
:Parameters:
`color` : (int, int, int, int)
4-tuple of ints in range [0,255] giving RGBA components of
color to fill with.
"""
self.color = _color_as_bytes(color)
def create_image(self, width, height):
data = self.color * width * height
return ImageData(width, height, 'RGBA', data)
class CheckerImagePattern(ImagePattern):
"""Create an image with a tileable checker image.
"""
def __init__(self, color1=(150, 150, 150, 255), color2=(200, 200, 200, 255)):
"""Initialise with the given colors.
:Parameters:
`color1` : (int, int, int, int)
4-tuple of ints in range [0,255] giving RGBA components of
color to fill with. This color appears in the top-left and
bottom-right corners of the image.
`color2` : (int, int, int, int)
4-tuple of ints in range [0,255] giving RGBA components of
color to fill with. This color appears in the top-right and
bottom-left corners of the image.
"""
self.color1 = _color_as_bytes(color1)
self.color2 = _color_as_bytes(color2)
def create_image(self, width, height):
hw = width // 2
hh = height // 2
row1 = self.color1 * hw + self.color2 * hw
row2 = self.color2 * hw + self.color1 * hw
data = row1 * hh + row2 * hh
return ImageData(width, height, 'RGBA', data)
class AbstractImage:
"""Abstract class representing an image.
:Parameters:
`width` : int
Width of image
`height` : int
Height of image
`anchor_x` : int
X coordinate of anchor, relative to left edge of image data
`anchor_y` : int
Y coordinate of anchor, relative to bottom edge of image data
"""
anchor_x = 0
anchor_y = 0
def __init__(self, width, height):
self.width = width
self.height = height
def __repr__(self):
return "{}(size={}x{})".format(self.__class__.__name__, self.width, self.height)
2021-04-16 23:21:06 +08:00
def get_image_data(self):
"""Get an ImageData view of this image.
2021-04-16 23:21:06 +08:00
Changes to the returned instance may or may not be reflected in this
image.
:rtype: :py:class:`~pyglet.image.ImageData`
.. versionadded:: 1.1
"""
raise ImageException('Cannot retrieve image data for %r' % self)
def get_texture(self, rectangle=False):
"""A :py:class:`~pyglet.image.Texture` view of this image.
2021-04-16 23:21:06 +08:00
2022-03-05 23:10:18 +08:00
The Texture target will be (``GL_TEXTURE_RECTANGLE``) if the `rectangle`
parameter is ``True``, otherwise it will be (``GL_TEXTURE_2D``).
2021-04-16 23:21:06 +08:00
Changes to the returned instance may or may not be reflected in this
image.
:Parameters:
`rectangle` : bool
True if the texture can be created as a rectangle.
.. versionadded:: 1.1.4.
:rtype: :py:class:`~pyglet.image.Texture`
.. versionadded:: 1.1
"""
raise ImageException('Cannot retrieve texture for %r' % self)
def get_mipmapped_texture(self):
"""Retrieve a :py:class:`~pyglet.image.Texture` instance with all mipmap levels filled in.
:rtype: :py:class:`~pyglet.image.Texture`
.. versionadded:: 1.1
"""
raise ImageException('Cannot retrieve mipmapped texture for %r' % self)
def get_region(self, x, y, width, height):
"""Retrieve a rectangular region of this image.
:Parameters:
`x` : int
Left edge of region.
`y` : int
Bottom edge of region.
`width` : int
Width of region.
`height` : int
Height of region.
:rtype: AbstractImage
"""
raise ImageException('Cannot get region for %r' % self)
def save(self, filename=None, file=None, encoder=None):
"""Save this image to a file.
:Parameters:
`filename` : str
Used to set the image file format, and to open the output file
if `file` is unspecified.
`file` : file-like object or None
File to write image data to.
`encoder` : ImageEncoder or None
If unspecified, all encoders matching the filename extension
are tried. If all fail, the exception from the first one
attempted is raised.
"""
if not file:
file = open(filename, 'wb')
if encoder:
2022-04-08 23:07:41 +08:00
encoder.encode(self, filename, file)
2021-04-16 23:21:06 +08:00
else:
first_exception = None
2022-04-08 23:07:41 +08:00
for encoder in _codec_registry.get_encoders(filename):
2021-04-16 23:21:06 +08:00
try:
2022-04-08 23:07:41 +08:00
return encoder.encode(self, filename, file)
2021-04-16 23:21:06 +08:00
except ImageEncodeException as e:
first_exception = first_exception or e
file.seek(0)
if not first_exception:
raise ImageEncodeException('No image encoders are available')
2021-04-16 23:21:06 +08:00
raise first_exception
def blit(self, x, y, z=0):
"""Draw this image to the active framebuffers.
The image will be drawn with the lower-left corner at
2021-04-16 23:21:06 +08:00
(``x -`` `anchor_x`, ``y -`` `anchor_y`, ``z``).
"""
raise ImageException('Cannot blit %r.' % self)
def blit_into(self, source, x, y, z):
"""Draw `source` on this image.
`source` will be copied into this image such that its anchor point
is aligned with the `x` and `y` parameters. If this image is a 3D
texture, the `z` coordinate gives the image slice to copy into.
2021-04-16 23:21:06 +08:00
Note that if `source` is larger than this image (or the positioning
would cause the copy to go out of bounds) then you must pass a
region of `source` to this method, typically using get_region().
"""
raise ImageException('Cannot blit images onto %r.' % self)
def blit_to_texture(self, target, level, x, y, z=0):
"""Draw this image on the currently bound texture at `target`.
2021-04-16 23:21:06 +08:00
This image is copied into the texture such that this image's anchor
point is aligned with the given `x` and `y` coordinates of the
destination texture. If the currently bound texture is a 3D texture,
the `z` coordinate gives the image slice to blit into.
"""
raise ImageException('Cannot blit %r to a texture.' % self)
class AbstractImageSequence:
"""Abstract sequence of images.
The sequence is useful for storing image animations or slices of a volume.
For efficient access, use the `texture_sequence` member. The class
also implements the sequence interface (`__len__`, `__getitem__`,
`__setitem__`).
"""
def get_texture_sequence(self):
"""Get a TextureSequence.
:rtype: `TextureSequence`
.. versionadded:: 1.1
"""
raise NotImplementedError('abstract')
def get_animation(self, period, loop=True):
"""Create an animation over this image sequence for the given constant
framerate.
:Parameters
`period` : float
Number of seconds to display each frame.
`loop` : bool
If True, the animation will loop continuously.
:rtype: :py:class:`~pyglet.image.Animation`
.. versionadded:: 1.1
"""
return Animation.from_image_sequence(self, period, loop)
def __getitem__(self, slice):
"""Retrieve a (list of) image.
2021-04-16 23:21:06 +08:00
:rtype: AbstractImage
"""
raise NotImplementedError('abstract')
def __setitem__(self, slice, image):
"""Replace one or more images in the sequence.
2021-04-16 23:21:06 +08:00
:Parameters:
`image` : `~pyglet.image.AbstractImage`
The replacement image. The actual instance may not be used,
depending on this implementation.
"""
raise NotImplementedError('abstract')
def __len__(self):
raise NotImplementedError('abstract')
def __iter__(self):
"""Iterate over the images in sequence.
:rtype: Iterator
.. versionadded:: 1.1
"""
raise NotImplementedError('abstract')
class TextureSequence(AbstractImageSequence):
"""Interface for a sequence of textures.
Typical implementations store multiple :py:class:`~pyglet.image.TextureRegion` s within one
:py:class:`~pyglet.image.Texture` so as to minimise state changes.
"""
def get_texture_sequence(self):
return self
class UniformTextureSequence(TextureSequence):
"""Interface for a sequence of textures, each with the same dimensions.
:Parameters:
`item_width` : int
Width of each texture in the sequence.
`item_height` : int
Height of each texture in the sequence.
2021-04-16 23:21:06 +08:00
"""
def _get_item_width(self):
raise NotImplementedError('abstract')
def _get_item_height(self):
raise NotImplementedError('abstract')
@property
def item_width(self):
return self._get_item_width()
@property
def item_height(self):
return self._get_item_height()
class ImageData(AbstractImage):
"""An image represented as a string of unsigned bytes.
:Parameters:
`data` : str
Pixel data, encoded according to `format` and `pitch`.
`format` : str
The format string to use when reading or writing `data`.
`pitch` : int
Number of bytes per row. Negative values indicate a top-to-bottom
arrangement.
"""
_swap1_pattern = re.compile(asbytes('(.)'), re.DOTALL)
_swap2_pattern = re.compile(asbytes('(.)(.)'), re.DOTALL)
_swap3_pattern = re.compile(asbytes('(.)(.)(.)'), re.DOTALL)
_swap4_pattern = re.compile(asbytes('(.)(.)(.)(.)'), re.DOTALL)
_current_texture = None
_current_mipmap_texture = None
def __init__(self, width, height, fmt, data, pitch=None):
2021-04-16 23:21:06 +08:00
"""Initialise image data.
:Parameters:
`width` : int
Width of image data
`height` : int
Height of image data
`fmt` : str
2021-04-16 23:21:06 +08:00
A valid format string, such as 'RGB', 'RGBA', 'ARGB', etc.
`data` : sequence
String or array/list of bytes giving the decoded data.
`pitch` : int or None
If specified, the number of bytes per row. Negative values
indicate a top-to-bottom arrangement. Defaults to
2021-04-16 23:21:06 +08:00
``width * len(format)``.
"""
super().__init__(width, height)
2021-04-16 23:21:06 +08:00
self._current_format = self._desired_format = fmt.upper()
2021-04-16 23:21:06 +08:00
self._current_data = data
self.pitch = pitch or width * len(fmt)
self._current_pitch = self.pitch
2021-04-16 23:21:06 +08:00
self.mipmap_images = []
def __getstate__(self):
return {
'width': self.width,
'height': self.height,
'_current_data': self.get_data(self._current_format, self._current_pitch),
'_current_format': self._current_format,
'_desired_format': self._desired_format,
'_current_pitch': self._current_pitch,
'pitch': self.pitch,
'mipmap_images': self.mipmap_images
}
def get_image_data(self):
return self
@property
def format(self):
"""Format string of the data. Read-write.
2021-04-16 23:21:06 +08:00
:type: str
"""
return self._desired_format
@format.setter
def format(self, fmt):
self._desired_format = fmt.upper()
self._current_texture = None
def get_data(self, fmt=None, pitch=None):
"""Get the byte data of the image.
:Parameters:
`fmt` : str
Format string of the return data.
`pitch` : int
Number of bytes per row. Negative values indicate a
top-to-bottom arrangement.
.. versionadded:: 1.1
:rtype: sequence of bytes, or str
"""
fmt = fmt or self._desired_format
pitch = pitch or self._current_pitch
if fmt == self._current_format and pitch == self._current_pitch:
return self._current_data
return self._convert(fmt, pitch)
def set_data(self, fmt, pitch, data):
"""Set the byte data of the image.
:Parameters:
`fmt` : str
Format string of the return data.
`pitch` : int
Number of bytes per row. Negative values indicate a
top-to-bottom arrangement.
`data` : str or sequence of bytes
Image data.
.. versionadded:: 1.1
"""
self._current_format = fmt
self._current_pitch = pitch
self._current_data = data
self._current_texture = None
self._current_mipmap_texture = None
def set_mipmap_image(self, level, image):
"""Set a mipmap image for a particular level.
The mipmap image will be applied to textures obtained via
`get_mipmapped_texture`.
:Parameters:
`level` : int
Mipmap level to set image at, must be >= 1.
`image` : AbstractImage
Image to set. Must have correct dimensions for that mipmap
level (i.e., width >> level, height >> level)
"""
if level == 0:
raise ImageException('Cannot set mipmap image at level 0 (it is this image)')
2021-04-16 23:21:06 +08:00
# Check dimensions of mipmap
width, height = self.width, self.height
for i in range(level):
width >>= 1
height >>= 1
if width != image.width or height != image.height:
raise ImageException('Mipmap image has wrong dimensions for level %d' % level)
2021-04-16 23:21:06 +08:00
# Extend mipmap_images list to required level
self.mipmap_images += [None] * (level - len(self.mipmap_images))
self.mipmap_images[level - 1] = image
def create_texture(self, cls, rectangle=False):
2021-04-16 23:21:06 +08:00
"""Create a texture containing this image.
:Parameters:
`cls` : class (subclass of Texture)
Class to construct.
`rectangle` : bool
``True`` if a rectangle can be created; see
`AbstractImage.get_texture`.
.. versionadded:: 1.1
:rtype: cls or cls.region_class
"""
target = GL_TEXTURE_RECTANGLE if rectangle else GL_TEXTURE_2D
internalformat = self._get_internalformat(self._desired_format)
texture = cls.create(self.width, self.height, target, internalformat)
2021-04-16 23:21:06 +08:00
if self.anchor_x or self.anchor_y:
texture.anchor_x = self.anchor_x
texture.anchor_y = self.anchor_y
self.blit_to_texture(texture.target, texture.level, self.anchor_x, self.anchor_y, 0, None)
2021-04-16 23:21:06 +08:00
return texture
def get_texture(self, rectangle=False):
if not self._current_texture or (not self._current_texture.target == GL_TEXTURE_RECTANGLE and rectangle):
self._current_texture = self.create_texture(Texture, rectangle)
2021-04-16 23:21:06 +08:00
return self._current_texture
def get_mipmapped_texture(self):
"""Return a Texture with mipmaps.
2021-04-16 23:21:06 +08:00
If :py:class:`~pyglet.image.set_mipmap_Image` has been called with at least one image, the set
of images defined will be used. Otherwise, mipmaps will be
automatically generated.
:rtype: :py:class:`~pyglet.image.Texture`
.. versionadded:: 1.1
"""
if self._current_mipmap_texture:
return self._current_mipmap_texture
texture = Texture.create(self.width, self.height, GL_TEXTURE_2D, None)
2021-04-16 23:21:06 +08:00
if self.anchor_x or self.anchor_y:
texture.anchor_x = self.anchor_x
texture.anchor_y = self.anchor_y
internalformat = self._get_internalformat(self.format)
glBindTexture(texture.target, texture.id)
glTexParameteri(texture.target, GL_TEXTURE_MIN_FILTER, GL_LINEAR_MIPMAP_LINEAR)
if self.mipmap_images:
self.blit_to_texture(texture.target, texture.level, self.anchor_x, self.anchor_y, 0, internalformat)
2021-04-16 23:21:06 +08:00
level = 0
for image in self.mipmap_images:
level += 1
if image:
image.blit_to_texture(texture.target, level, self.anchor_x, self.anchor_y, 0, internalformat)
2021-04-16 23:21:06 +08:00
# TODO: should set base and max mipmap level if some mipmaps are missing.
else:
glGenerateMipmap(texture.target)
self.blit_to_texture(texture.target, texture.level, self.anchor_x, self.anchor_y, 0, internalformat)
2021-04-16 23:21:06 +08:00
self._current_mipmap_texture = texture
return texture
def get_region(self, x, y, width, height):
"""Retrieve a rectangular region of this image data.
:Parameters:
`x` : int
Left edge of region.
`y` : int
Bottom edge of region.
`width` : int
Width of region.
`height` : int
Height of region.
:rtype: ImageDataRegion
"""
return ImageDataRegion(x, y, width, height, self)
def blit(self, x, y, z=0, width=None, height=None):
self.get_texture().blit(x, y, z, width, height)
def blit_to_texture(self, target, level, x, y, z, internalformat=None):
"""Draw this image to to the currently bound texture at `target`.
This image's anchor point will be aligned to the given `x` and `y`
coordinates. If the currently bound texture is a 3D texture, the `z`
parameter gives the image slice to blit into.
If `internalformat` is specified, glTexImage is used to initialise
the texture; otherwise, glTexSubImage is used to update a region.
"""
x -= self.anchor_x
y -= self.anchor_y
data_format = self.format
data_pitch = abs(self._current_pitch)
# Determine pixel format from format string
fmt, gl_type = self._get_gl_format_and_type(data_format)
if fmt is None:
# Need to convert data to a standard form
data_format = {
1: 'R',
2: 'RG',
3: 'RGB',
4: 'RGBA'}.get(len(data_format))
fmt, gl_type = self._get_gl_format_and_type(data_format)
# Get data in required format (hopefully will be the same format it's already
# in, unless that's an obscure format, upside-down or the driver is old).
2021-04-16 23:21:06 +08:00
data = self._convert(data_format, data_pitch)
if data_pitch & 0x1:
align = 1
2021-04-16 23:21:06 +08:00
elif data_pitch & 0x2:
align = 2
2021-04-16 23:21:06 +08:00
else:
align = 4
2021-04-16 23:21:06 +08:00
row_length = data_pitch // len(data_format)
glPixelStorei(GL_UNPACK_ALIGNMENT, align)
2021-04-16 23:21:06 +08:00
glPixelStorei(GL_UNPACK_ROW_LENGTH, row_length)
self._apply_region_unpack()
if target == GL_TEXTURE_3D or target == GL_TEXTURE_2D_ARRAY:
2021-04-16 23:21:06 +08:00
assert not internalformat
glTexSubImage3D(target, level,
x, y, z,
self.width, self.height, 1,
fmt, gl_type,
2021-04-16 23:21:06 +08:00
data)
elif internalformat:
glTexImage2D(target, level,
internalformat,
self.width, self.height,
0,
fmt, gl_type,
2021-04-16 23:21:06 +08:00
data)
else:
glTexSubImage2D(target, level,
x, y,
self.width, self.height,
fmt, gl_type,
2021-04-16 23:21:06 +08:00
data)
# Unset GL_UNPACK_ROW_LENGTH:
glPixelStorei(GL_UNPACK_ROW_LENGTH, 0)
# Flush image upload before data get GC'd:
2021-04-16 23:21:06 +08:00
glFlush()
def _apply_region_unpack(self):
pass
def _convert(self, fmt, pitch):
2021-04-16 23:21:06 +08:00
"""Return data in the desired format; does not alter this instance's
current format or pitch.
"""
if fmt == self._current_format and pitch == self._current_pitch:
2021-04-16 23:21:06 +08:00
if type(self._current_data) is str:
return asbytes(self._current_data)
return self._current_data
self._ensure_bytes()
2021-04-16 23:21:06 +08:00
data = self._current_data
current_pitch = self._current_pitch
current_format = self._current_format
sign_pitch = current_pitch // abs(current_pitch)
if fmt != self._current_format:
# Create replacement string, e.g. r'\4\1\2\3' to convert RGBA to ARGB
2021-04-16 23:21:06 +08:00
repl = asbytes('')
for c in fmt:
2021-04-16 23:21:06 +08:00
try:
idx = current_format.index(c) + 1
except ValueError:
idx = 1
repl += asbytes(r'\%d' % idx)
if len(current_format) == 1:
swap_pattern = self._swap1_pattern
elif len(current_format) == 2:
swap_pattern = self._swap2_pattern
elif len(current_format) == 3:
swap_pattern = self._swap3_pattern
elif len(current_format) == 4:
swap_pattern = self._swap4_pattern
else:
raise ImageException('Current image format is wider than 32 bits.')
2021-04-16 23:21:06 +08:00
packed_pitch = self.width * len(current_format)
if abs(self._current_pitch) != packed_pitch:
# Pitch is wider than pixel data, need to go row-by-row.
new_pitch = abs(self._current_pitch)
rows = [data[i:i+new_pitch] for i in range(0, len(data), new_pitch)]
rows = [swap_pattern.sub(repl, r[:packed_pitch]) for r in rows]
data = asbytes('').join(rows)
else:
# Rows are tightly packed, apply regex over whole image.
data = swap_pattern.sub(repl, data)
# After conversion, rows will always be tightly packed
current_pitch = sign_pitch * (len(fmt) * self.width)
2021-04-16 23:21:06 +08:00
if pitch != current_pitch:
diff = abs(current_pitch) - abs(pitch)
if diff > 0:
# New pitch is shorter than old pitch, chop bytes off each row
new_pitch = abs(pitch)
rows = [data[i:i+new_pitch-diff] for i in range(0, len(data), new_pitch)]
data = asbytes('').join(rows)
2021-04-16 23:21:06 +08:00
elif diff < 0:
# New pitch is longer than old pitch, add '0' bytes to each row
new_pitch = abs(current_pitch)
padding = asbytes(1) * -diff
rows = [data[i:i+new_pitch] + padding for i in range(0, len(data), new_pitch)]
data = asbytes('').join(rows)
2021-04-16 23:21:06 +08:00
if current_pitch * pitch < 0:
# Pitch differs in sign, swap row order
new_pitch = abs(pitch)
rows = [data[i:i+new_pitch] for i in range(0, len(data), new_pitch)]
rows.reverse()
data = asbytes('').join(rows)
2021-04-16 23:21:06 +08:00
return asbytes(data)
def _ensure_bytes(self):
2021-04-16 23:21:06 +08:00
if type(self._current_data) is not bytes:
self._current_data = asbytes(self._current_data)
2021-04-16 23:21:06 +08:00
@staticmethod
def _get_gl_format_and_type(fmt):
if fmt == 'R':
2021-04-16 23:21:06 +08:00
return GL_RED, GL_UNSIGNED_BYTE
elif fmt == 'RG':
return GL_RG, GL_UNSIGNED_BYTE
elif fmt == 'RGB':
2021-04-16 23:21:06 +08:00
return GL_RGB, GL_UNSIGNED_BYTE
elif fmt == 'BGR':
2021-04-16 23:21:06 +08:00
return GL_BGR, GL_UNSIGNED_BYTE
elif fmt == 'RGBA':
return GL_RGBA, GL_UNSIGNED_BYTE
elif fmt == 'BGRA':
2021-04-16 23:21:06 +08:00
return GL_BGRA, GL_UNSIGNED_BYTE
elif fmt == 'L':
return GL_LUMINANCE, GL_UNSIGNED_BYTE
elif fmt == 'A':
return GL_ALPHA, GL_UNSIGNED_BYTE
2021-04-16 23:21:06 +08:00
return None, None
@staticmethod
def _get_internalformat(fmt):
if fmt == 'R':
return GL_RED
elif fmt == 'RG':
return GL_RG
elif fmt == 'RGB':
2021-04-16 23:21:06 +08:00
return GL_RGB
elif fmt == 'RGBA':
return GL_RGBA
elif fmt == 'D':
return GL_DEPTH_COMPONENT
elif fmt == 'DS':
return GL_DEPTH_STENCIL
elif fmt == 'L':
2021-04-16 23:21:06 +08:00
return GL_LUMINANCE
elif fmt == 'A':
return GL_ALPHA
2021-04-16 23:21:06 +08:00
return GL_RGBA
class ImageDataRegion(ImageData):
def __init__(self, x, y, width, height, image_data):
super().__init__(width, height,
image_data._current_format,
image_data._current_data,
image_data._current_pitch)
2021-04-16 23:21:06 +08:00
self.x = x
self.y = y
def __getstate__(self):
return {
'width': self.width,
'height': self.height,
'_current_data': self.get_data(self._current_format, self._current_pitch),
2021-04-16 23:21:06 +08:00
'_current_format': self._current_format,
'_desired_format': self._desired_format,
'_current_pitch': self._current_pitch,
'pitch': self.pitch,
'mipmap_images': self.mipmap_images,
'x': self.x,
'y': self.y
}
def get_data(self, fmt=None, pitch=None):
x1 = len(self._current_format) * self.x
x2 = len(self._current_format) * (self.x + self.width)
self._ensure_bytes()
2021-04-16 23:21:06 +08:00
data = self._convert(self._current_format, abs(self._current_pitch))
new_pitch = abs(self._current_pitch)
rows = [data[i:i+new_pitch] for i in range(0, len(data), new_pitch)]
rows = [row[x1:x2] for row in rows[self.y:self.y + self.height]]
self._current_data = b''.join(rows)
self._current_pitch = self.width * len(self._current_format)
self._current_texture = None
self.x = 0
self.y = 0
fmt = fmt or self._desired_format
pitch = pitch or self._current_pitch
return super().get_data(fmt, pitch)
2021-04-16 23:21:06 +08:00
def set_data(self, fmt, pitch, data):
self.x = 0
self.y = 0
super().set_data(fmt, pitch, data)
2021-04-16 23:21:06 +08:00
def _apply_region_unpack(self):
glPixelStorei(GL_UNPACK_SKIP_PIXELS, self.x)
glPixelStorei(GL_UNPACK_SKIP_ROWS, self.y)
def get_region(self, x, y, width, height):
x += self.x
y += self.y
return super().get_region(x, y, width, height)
2021-04-16 23:21:06 +08:00
class CompressedImageData(AbstractImage):
"""Image representing some compressed data suitable for direct uploading
to driver.
"""
_current_texture = None
_current_mipmap_texture = None
def __init__(self, width, height, gl_format, data, extension=None, decoder=None):
"""Construct a CompressedImageData with the given compressed data.
:Parameters:
`width` : int
Width of image
`height` : int
Height of image
`gl_format` : int
GL constant giving format of compressed data; for example,
``GL_COMPRESSED_RGBA_S3TC_DXT5_EXT``.
`data` : sequence
String or array/list of bytes giving compressed image data.
`extension` : str or None
If specified, gives the name of a GL extension to check for
before creating a texture.
`decoder` : function(data, width, height) -> AbstractImage
A function to decode the compressed data, to be used if the
required extension is not present.
2021-04-16 23:21:06 +08:00
"""
super().__init__(width, height)
2021-04-16 23:21:06 +08:00
self.data = data
self.gl_format = gl_format
self.extension = extension
self.decoder = decoder
self.mipmap_data = []
def set_mipmap_data(self, level, data):
"""Set data for a mipmap level.
Supplied data gives a compressed image for the given mipmap level.
The image must be of the correct dimensions for the level
2021-04-16 23:21:06 +08:00
(i.e., width >> level, height >> level); but this is not checked. If
any mipmap levels are specified, they are used; otherwise, mipmaps for
`mipmapped_texture` are generated automatically.
:Parameters:
`level` : int
Level of mipmap image to set.
`data` : sequence
String or array/list of bytes giving compressed image data.
Data must be in same format as specified in constructor.
"""
# Extend mipmap_data list to required level
self.mipmap_data += [None] * (level - len(self.mipmap_data))
self.mipmap_data[level - 1] = data
def _have_extension(self):
return self.extension is None or gl_info.have_extension(self.extension)
def _verify_driver_supported(self):
"""Assert that the extension required for this image data is
supported.
Raises `ImageException` if not.
"""
if not self._have_extension():
raise ImageException('%s is required to decode %r' % (self.extension, self))
def get_texture(self, rectangle=False):
if rectangle:
2021-04-16 23:21:06 +08:00
raise ImageException('Compressed texture rectangles not supported')
if self._current_texture:
return self._current_texture
texture = Texture.create(self.width, self.height, GL_TEXTURE_2D, None)
2021-04-16 23:21:06 +08:00
if self.anchor_x or self.anchor_y:
texture.anchor_x = self.anchor_x
texture.anchor_y = self.anchor_y
glBindTexture(texture.target, texture.id)
glTexParameteri(texture.target, GL_TEXTURE_MIN_FILTER, texture.min_filter)
glTexParameteri(texture.target, GL_TEXTURE_MAG_FILTER, texture.mag_filter)
2021-04-16 23:21:06 +08:00
if self._have_extension():
glCompressedTexImage2D(texture.target, texture.level,
self.gl_format,
self.width, self.height, 0,
len(self.data), self.data)
else:
image = self.decoder(self.data, self.width, self.height)
texture = image.get_texture()
assert texture.width == self.width
assert texture.height == self.height
glFlush()
self._current_texture = texture
return texture
def get_mipmapped_texture(self):
if self._current_mipmap_texture:
return self._current_mipmap_texture
if not self._have_extension():
# TODO mip-mapped software decoded compressed textures. For now,
# just return a non-mipmapped texture.
return self.get_texture()
texture = Texture.create(self.width, self.height, GL_TEXTURE_2D, None)
2021-04-16 23:21:06 +08:00
if self.anchor_x or self.anchor_y:
texture.anchor_x = self.anchor_x
texture.anchor_y = self.anchor_y
glBindTexture(texture.target, texture.id)
glTexParameteri(texture.target, GL_TEXTURE_MIN_FILTER, GL_LINEAR_MIPMAP_LINEAR)
if not self.mipmap_data:
glGenerateMipmap(texture.target)
2021-04-16 23:21:06 +08:00
glCompressedTexImage2D(texture.target, texture.level,
self.gl_format,
self.width, self.height, 0,
len(self.data), self.data)
2021-04-16 23:21:06 +08:00
width, height = self.width, self.height
level = 0
for data in self.mipmap_data:
width >>= 1
height >>= 1
level += 1
glCompressedTexImage2D(texture.target, level, self.gl_format, width, height, 0, len(data), data)
2021-04-16 23:21:06 +08:00
glFlush()
self._current_mipmap_texture = texture
return texture
def blit_to_texture(self, target, level, x, y, z):
self._verify_driver_supported()
if target == GL_TEXTURE_3D:
glCompressedTexSubImage3D(target, level,
x - self.anchor_x, y - self.anchor_y, z,
self.width, self.height, 1,
self.gl_format,
len(self.data), self.data)
2021-04-16 23:21:06 +08:00
else:
glCompressedTexSubImage2D(target, level,
x - self.anchor_x, y - self.anchor_y,
self.width, self.height,
self.gl_format,
len(self.data), self.data)
2021-04-16 23:21:06 +08:00
class Texture(AbstractImage):
"""An image loaded into video memory that can be efficiently drawn
to the framebuffer.
Typically you will get an instance of Texture by accessing the `texture`
member of any other AbstractImage.
:Parameters:
`region_class` : class (subclass of TextureRegion)
Class to use when constructing regions of this texture.
`tex_coords` : tuple
12-tuple of float, named (u1, v1, r1, u2, v2, r2, ...). u, v, r
give the 3D texture coordinates for vertices 1-4. The vertices
are specified in the order bottom-left, bottom-right, top-right
and top-left.
`target` : int
The GL texture target (e.g., ``GL_TEXTURE_2D``).
`level` : int
The mipmap level of this texture.
"""
region_class = None # Set to TextureRegion after it's defined
tex_coords = (0, 0, 0, 1, 0, 0, 1, 1, 0, 0, 1, 0)
2021-04-16 23:21:06 +08:00
tex_coords_order = (0, 1, 2, 3)
colors = (0, 0, 0, 0) * 4
2021-04-16 23:21:06 +08:00
level = 0
images = 1
x = y = z = 0
default_min_filter = GL_LINEAR
default_mag_filter = GL_LINEAR
def __init__(self, width, height, target, tex_id):
super().__init__(width, height)
2021-04-16 23:21:06 +08:00
self.target = target
self.id = tex_id
self._context = pyglet.gl.current_context
2021-04-16 23:21:06 +08:00
def __del__(self):
try:
self._context.delete_texture(self.id)
except Exception:
2021-04-16 23:21:06 +08:00
pass
@classmethod
def create(cls, width, height, target=GL_TEXTURE_2D, internalformat=GL_RGBA, min_filter=None, mag_filter=None):
"""Create a Texture
2021-04-16 23:21:06 +08:00
Create a Texture with the specified dimentions, target and format.
On return, the texture will be bound.
2021-04-16 23:21:06 +08:00
:Parameters:
`width` : int
Width of texture in pixels.
2021-04-16 23:21:06 +08:00
`height` : int
Height of texture in pixels.
2021-04-16 23:21:06 +08:00
`target` : int
GL constant giving texture target to use, typically ``GL_TEXTURE_2D``.
2021-04-16 23:21:06 +08:00
`internalformat` : int
GL constant giving internal format of texture; for example, ``GL_RGBA``.
If ``None``, the texture will be created but not initialized.
2021-04-16 23:21:06 +08:00
`min_filter` : int
The minifaction filter used for this texture, commonly ``GL_LINEAR`` or ``GL_NEAREST``
`mag_filter` : int
The magnification filter used for this texture, commonly ``GL_LINEAR`` or ``GL_NEAREST``
:rtype: :py:class:`~pyglet.image.Texture`
"""
min_filter = min_filter or cls.default_min_filter
mag_filter = mag_filter or cls.default_mag_filter
tex_id = GLuint()
glGenTextures(1, byref(tex_id))
glBindTexture(target, tex_id.value)
2021-04-16 23:21:06 +08:00
glTexParameteri(target, GL_TEXTURE_MIN_FILTER, min_filter)
glTexParameteri(target, GL_TEXTURE_MAG_FILTER, mag_filter)
if internalformat is not None:
blank = (GLubyte * (width * height * 4))()
glTexImage2D(target, 0,
internalformat,
width, height,
0,
GL_RGBA, GL_UNSIGNED_BYTE,
blank)
glFlush()
texture = cls(width, height, target, tex_id.value)
2021-04-16 23:21:06 +08:00
texture.min_filter = min_filter
texture.mag_filter = mag_filter
if target is GL_TEXTURE_RECTANGLE:
texture.tex_coords = (0, 0, 0, width, 0, 0, width, height, 0, 0, height, 0)
else:
texture.tex_coords = cls.tex_coords
2021-04-16 23:21:06 +08:00
return texture
def get_image_data(self, z=0):
"""Get the image data of this texture.
Changes to the returned instance will not be reflected in this
texture.
:Parameters:
`z` : int
For 3D textures, the image slice to retrieve.
:rtype: :py:class:`~pyglet.image.ImageData`
"""
glBindTexture(self.target, self.id)
# Always extract complete RGBA data. Could check internalformat
# to only extract used channels. XXX
fmt = 'RGBA'
2021-04-16 23:21:06 +08:00
gl_format = GL_RGBA
glPixelStorei(GL_PACK_ALIGNMENT, 1)
buf = (GLubyte * (self.width * self.height * self.images * len(fmt)))()
glGetTexImage(self.target, self.level, gl_format, GL_UNSIGNED_BYTE, buf)
2021-04-16 23:21:06 +08:00
data = ImageData(self.width, self.height, fmt, buf)
2021-04-16 23:21:06 +08:00
if self.images > 1:
data = data.get_region(0, z * self.height, self.width, self.height)
return data
def get_texture(self, rectangle=False):
if rectangle and not self.target == GL_TEXTURE_RECTANGLE:
raise ImageException('Texture is not a rectangle, it must be created as a rectangle.')
2021-04-16 23:21:06 +08:00
return self
# no implementation of blit_to_texture yet
2021-04-16 23:21:06 +08:00
def blit(self, x, y, z=0, width=None, height=None):
x1 = x - self.anchor_x
y1 = y - self.anchor_y
x2 = x1 + (width is None and self.width or width)
y2 = y1 + (height is None and self.height or height)
vertices = x1, y1, z, x2, y1, z, x2, y2, z, x1, y2, z
glActiveTexture(GL_TEXTURE0)
2021-04-16 23:21:06 +08:00
glBindTexture(self.target, self.id)
pyglet.graphics.draw_indexed(4, GL_TRIANGLES, [0, 1, 2, 0, 2, 3],
2022-03-05 23:10:18 +08:00
position=('f', vertices),
tex_coords=('f', self.tex_coords),
colors=('Bn', self.colors))
glBindTexture(self.target, 0)
2021-04-16 23:21:06 +08:00
def blit_into(self, source, x, y, z):
glBindTexture(self.target, self.id)
source.blit_to_texture(self.target, self.level, x, y, z)
def get_region(self, x, y, width, height):
return self.region_class(x, y, 0, width, height, self)
def get_transform(self, flip_x=False, flip_y=False, rotate=0):
"""Create a copy of this image applying a simple transformation.
The transformation is applied to the texture coordinates only;
:py:meth:`~pyglet.image.ImageData.get_image_data` will return the untransformed data. The
transformation is applied around the anchor point.
:Parameters:
`flip_x` : bool
If True, the returned image will be flipped horizontally.
`flip_y` : bool
If True, the returned image will be flipped vertically.
`rotate` : int
Degrees of clockwise rotation of the returned image. Only
2021-04-16 23:21:06 +08:00
90-degree increments are supported.
:rtype: :py:class:`~pyglet.image.TextureRegion`
"""
transform = self.get_region(0, 0, self.width, self.height)
bl, br, tr, tl = 0, 1, 2, 3
transform.anchor_x = self.anchor_x
transform.anchor_y = self.anchor_y
if flip_x:
bl, br, tl, tr = br, bl, tr, tl
transform.anchor_x = self.width - self.anchor_x
if flip_y:
bl, br, tl, tr = tl, tr, bl, br
transform.anchor_y = self.height - self.anchor_y
rotate %= 360
if rotate < 0:
rotate += 360
if rotate == 0:
pass
elif rotate == 90:
bl, br, tr, tl = br, tr, tl, bl
transform.anchor_x, transform.anchor_y = transform.anchor_y, transform.width - transform.anchor_x
2021-04-16 23:21:06 +08:00
elif rotate == 180:
bl, br, tr, tl = tr, tl, bl, br
transform.anchor_x = transform.width - transform.anchor_x
transform.anchor_y = transform.height - transform.anchor_y
elif rotate == 270:
bl, br, tr, tl = tl, bl, br, tr
transform.anchor_x, transform.anchor_y = transform.height - transform.anchor_y, transform.anchor_x
2021-04-16 23:21:06 +08:00
else:
assert False, 'Only 90 degree rotations are supported.'
if rotate in (90, 270):
transform.width, transform.height = transform.height, transform.width
transform._set_tex_coords_order(bl, br, tr, tl)
return transform
def _set_tex_coords_order(self, bl, br, tr, tl):
tex_coords = (self.tex_coords[:3],
self.tex_coords[3:6],
self.tex_coords[6:9],
self.tex_coords[9:])
self.tex_coords = tex_coords[bl] + tex_coords[br] + tex_coords[tr] + tex_coords[tl]
order = self.tex_coords_order
self.tex_coords_order = (order[bl], order[br], order[tr], order[tl])
def __repr__(self):
return "{}(id={}, size={}x{})".format(self.__class__.__name__, self.id, self.width, self.height)
2021-04-16 23:21:06 +08:00
class TextureRegion(Texture):
"""A rectangular region of a texture, presented as if it were a separate texture.
2021-04-16 23:21:06 +08:00
"""
def __init__(self, x, y, z, width, height, owner):
super().__init__(width, height, owner.target, owner.id)
2021-04-16 23:21:06 +08:00
self.x = x
self.y = y
self.z = z
self.owner = owner
owner_u1 = owner.tex_coords[0]
owner_v1 = owner.tex_coords[1]
owner_u2 = owner.tex_coords[3]
owner_v2 = owner.tex_coords[7]
scale_u = owner_u2 - owner_u1
scale_v = owner_v2 - owner_v1
u1 = x / owner.width * scale_u + owner_u1
v1 = y / owner.height * scale_v + owner_v1
u2 = (x + width) / owner.width * scale_u + owner_u1
v2 = (y + height) / owner.height * scale_v + owner_v1
r = z / owner.images + owner.tex_coords[2]
self.tex_coords = (u1, v1, r, u2, v1, r, u2, v2, r, u1, v2, r)
def get_image_data(self):
image_data = self.owner.get_image_data(self.z)
return image_data.get_region(self.x, self.y, self.width, self.height)
def get_region(self, x, y, width, height):
x += self.x
y += self.y
region = self.region_class(x, y, self.z, width, height, self.owner)
region._set_tex_coords_order(*self.tex_coords_order)
return region
def blit_into(self, source, x, y, z):
self.owner.blit_into(source, x + self.x, y + self.y, z + self.z)
def __repr__(self):
return "{}(id={}, size={}x{}, owner={}x{})".format(self.__class__.__name__, self.id, self.width, self.height,
self.owner.width, self.owner.height)
2021-04-16 23:21:06 +08:00
def __del__(self):
# only the owner Texture should handle deletion
pass
Texture.region_class = TextureRegion
class Texture3D(Texture, UniformTextureSequence):
"""A texture with more than one image slice.
Use `create_for_images` or `create_for_image_grid` classmethod to
construct.
"""
item_width = 0
item_height = 0
items = ()
@classmethod
def create_for_images(cls, images, internalformat=GL_RGBA):
item_width = images[0].width
item_height = images[0].height
for image in images:
if image.width != item_width or image.height != item_height:
raise ImageException('Images do not have same dimensions.')
depth = len(images)
texture = cls.create(item_width, item_height, GL_TEXTURE_3D, None)
2021-04-16 23:21:06 +08:00
if images[0].anchor_x or images[0].anchor_y:
texture.anchor_x = images[0].anchor_x
texture.anchor_y = images[0].anchor_y
texture.images = depth
blank = (GLubyte * (texture.width * texture.height * texture.images))()
glBindTexture(texture.target, texture.id)
glTexImage3D(texture.target, texture.level,
internalformat,
texture.width, texture.height, texture.images, 0,
GL_ALPHA, GL_UNSIGNED_BYTE,
blank)
items = []
for i, image in enumerate(images):
item = cls.region_class(0, 0, i, item_width, item_height, texture)
items.append(item)
image.blit_to_texture(texture.target, texture.level, image.anchor_x, image.anchor_y, i)
2021-04-16 23:21:06 +08:00
glFlush()
texture.items = items
texture.item_width = item_width
texture.item_height = item_height
return texture
@classmethod
def create_for_image_grid(cls, grid, internalformat=GL_RGBA):
return cls.create_for_images(grid[:], internalformat)
def __len__(self):
return len(self.items)
def __getitem__(self, index):
return self.items[index]
def __setitem__(self, index, value):
if type(index) is slice:
for item, image in zip(self[index], value):
image.blit_to_texture(self.target, self.level, image.anchor_x, image.anchor_y, item.z)
else:
value.blit_to_texture(self.target, self.level, value.anchor_x, value.anchor_y, self[index].z)
def __iter__(self):
return iter(self.items)
class TextureArrayRegion(TextureRegion):
"""A region of a TextureArray, presented as if it were a separate texture.
"""
def __init__(self, x, y, z, width, height, owner):
super().__init__(width, height, owner.target, owner.id)
self.x = x
self.y = y
self.z = z
self.owner = owner
owner_u1 = owner.tex_coords[0]
owner_v1 = owner.tex_coords[1]
owner_u2 = owner.tex_coords[3]
owner_v2 = owner.tex_coords[7]
scale_u = owner_u2 - owner_u1
scale_v = owner_v2 - owner_v1
u1 = x / owner.width * scale_u + owner_u1
v1 = y / owner.height * scale_v + owner_v1
u2 = (x + width) / owner.width * scale_u + owner_u1
v2 = (y + height) / owner.height * scale_v + owner_v1
z = float(z)
self.tex_coords = (u1, v1, z, u2, v1, z, u2, v2, z, u1, v2, z)
def __repr__(self):
return "{}(id={}, size={}x{}, layer={})".format(self.__class__.__name__, self.id, self.width, self.height, self.z)
class TextureArray(Texture, UniformTextureSequence):
allow_smaller_pack = True
@classmethod
def create(cls, width, height, internalformat=GL_RGBA, min_filter=None, mag_filter=None, max_depth=256):
"""Create an empty TextureArray.
You may specify the maximum depth, or layers, the Texture Array should have. This defaults
to 256, but will be hardware and driver dependent.
:Parameters:
`width` : int
Width of the texture.
`height` : int
Height of the texture.
`internalformat` : int
GL constant giving the internal format of the texture array; for example, ``GL_RGBA``.
`min_filter` : int
The minifaction filter used for this texture array, commonly ``GL_LINEAR`` or ``GL_NEAREST``
`mag_filter` : int
The magnification filter used for this texture array, commonly ``GL_LINEAR`` or ``GL_NEAREST``
`max_depth` : int
The number of layers in the texture array.
:rtype: :py:class:`~pyglet.image.TextureArray`
.. versionadded:: 2.0
"""
min_filter = min_filter or cls.default_min_filter
mag_filter = mag_filter or cls.default_mag_filter
max_depth_limit = get_max_array_texture_layers()
assert max_depth <= max_depth_limit, "TextureArray max_depth supported is {}.".format(max_depth_limit)
tex_id = GLuint()
glGenTextures(1, byref(tex_id))
glBindTexture(GL_TEXTURE_2D_ARRAY, tex_id.value)
glTexParameteri(GL_TEXTURE_2D_ARRAY, GL_TEXTURE_MIN_FILTER, min_filter)
glTexParameteri(GL_TEXTURE_2D_ARRAY, GL_TEXTURE_MAG_FILTER, mag_filter)
glTexImage3D(GL_TEXTURE_2D_ARRAY, 0,
internalformat,
width, height, max_depth,
0,
internalformat, GL_UNSIGNED_BYTE,
0)
glFlush()
texture = cls(width, height, GL_TEXTURE_2D_ARRAY, tex_id.value)
texture.items = [] # No items on creation
texture.max_depth = max_depth
texture.min_filter = min_filter
texture.mag_filter = mag_filter
return texture
def _verify_size(self, image):
if image.width > self.width or image.height > self.height:
raise ImageException('Image ({0}x{1}) exceeds the size of the TextureArray ({2}x{3})'.format(
image.width, image.height, self.width, self.height))
def allocate(self, *images):
if len(self.items) + len(images) > self.max_depth:
raise Exception("The amount of images being added exceeds the depth of this TextureArray.")
textures = []
start_length = len(self.items)
for i, image in enumerate(images):
self._verify_size(image)
item = self.region_class(0, 0, start_length + i, image.width, image.height, self)
self.items.append(item)
image.blit_to_texture(self.target, self.level, image.anchor_x, image.anchor_y, start_length + i)
glFlush()
return self.items[start_length:]
@classmethod
def create_for_image_grid(cls, grid, internalformat=GL_RGBA):
texture_array = cls.create(grid[0].width, grid[0].height, internalformat, max_depth=len(grid))
texture_array.allocate(*grid[:])
return texture_array
def __len__(self):
return len(self.items)
def __getitem__(self, index):
return self.items[index]
def __setitem__(self, index, value):
if type(index) is slice:
for old_item, image in zip(self[index], value):
self._verify_size(image)
item = self.region_class(0, 0, old_item.z, image.width, image.height, self)
image.blit_to_texture(self.target, self.level, image.anchor_x, image.anchor_y, old_item.z)
self.items[old_item.z] = item
2021-04-16 23:21:06 +08:00
else:
self._verify_size(value)
item = self.region_class(0, 0, index, value.width, value.height, self)
value.blit_to_texture(self.target, self.level, value.anchor_x, value.anchor_y, index)
self.items[index] = item
2021-04-16 23:21:06 +08:00
def __iter__(self):
return iter(self.items)
TextureArray.region_class = TextureArrayRegion
TextureArrayRegion.region_class = TextureArrayRegion
2021-04-16 23:21:06 +08:00
class TileableTexture(Texture):
"""A texture that can be tiled efficiently.
2022-03-05 23:10:18 +08:00
Use :py:class:`~pyglet.image.create_for_image` classmethod to construct.
2021-04-16 23:21:06 +08:00
"""
def get_region(self, x, y, width, height):
raise ImageException('Cannot get region of %r' % self)
def blit_tiled(self, x, y, z, width, height):
"""Blit this texture tiled over the given area.
2021-04-16 23:21:06 +08:00
The image will be tiled with the bottom-left corner of the destination
rectangle aligned with the anchor point of this texture.
"""
u1 = self.anchor_x / self.width
v1 = self.anchor_y / self.height
u2 = u1 + width / self.width
v2 = v1 + height / self.height
w, h = width, height
t = self.tex_coords
vertices = (x, y, z,
x + w, y, z,
x + w, y + h, z,
x, y + h, z)
tex_coords = (u1, v1, t[2],
u2, v1, t[5],
u2, v2, t[8],
u1, v2, t[11],)
glActiveTexture(GL_TEXTURE0)
2021-04-16 23:21:06 +08:00
glBindTexture(self.target, self.id)
pyglet.graphics.draw_indexed(4, GL_TRIANGLES, [0, 1, 2, 0, 2, 3],
2022-03-05 23:10:18 +08:00
position=('f', vertices),
tex_coords=('f', tex_coords))
glBindTexture(self.target, 0)
2021-04-16 23:21:06 +08:00
@classmethod
def create_for_image(cls, image):
image = image.get_image_data()
return image.create_texture(cls)
class DepthTexture(Texture):
"""A texture with depth samples (typically 24-bit)."""
def blit_into(self, source, x, y, z):
glBindTexture(self.target, self.id)
source.blit_to_texture(self.level, x, y, z)
class ImageGrid(AbstractImage, AbstractImageSequence):
"""An imaginary grid placed over an image allowing easy access to
regular regions of that image.
2021-04-16 23:21:06 +08:00
The grid can be accessed either as a complete image, or as a sequence
of images. The most useful applications are to access the grid
as a :py:class:`~pyglet.image.TextureGrid`::
2021-04-16 23:21:06 +08:00
image_grid = ImageGrid(...)
texture_grid = image_grid.get_texture_sequence()
2021-04-16 23:21:06 +08:00
or as a :py:class:`~pyglet.image.Texture3D`::
2021-04-16 23:21:06 +08:00
image_grid = ImageGrid(...)
texture_3d = Texture3D.create_for_image_grid(image_grid)
2021-04-16 23:21:06 +08:00
"""
_items = ()
_texture_grid = None
2021-04-16 23:21:06 +08:00
def __init__(self, image, rows, columns, item_width=None, item_height=None, row_padding=0, column_padding=0):
"""Construct a grid for the given image.
2021-04-16 23:21:06 +08:00
You can specify parameters for the grid, for example setting
the padding between cells. Grids are always aligned to the
bottom-left corner of the image.
:Parameters:
`image` : AbstractImage
Image over which to construct the grid.
`rows` : int
Number of rows in the grid.
`columns` : int
Number of columns in the grid.
`item_width` : int
Width of each column. If unspecified, is calculated such
that the entire image width is used.
`item_height` : int
Height of each row. If unspecified, is calculated such that
the entire image height is used.
`row_padding` : int
Pixels separating adjacent rows. The padding is only
inserted between rows, not at the edges of the grid.
`column_padding` : int
Pixels separating adjacent columns. The padding is only
inserted between columns, not at the edges of the grid.
2021-04-16 23:21:06 +08:00
"""
super().__init__(image.width, image.height)
self.image = image
self.rows = rows
self.columns = columns
self.item_width = item_width or (image.width - column_padding * (columns - 1)) // columns
self.item_height = item_height or (image.height - row_padding * (rows - 1)) // rows
self.row_padding = row_padding
self.column_padding = column_padding
2021-04-16 23:21:06 +08:00
def get_texture(self, rectangle=False):
return self.image.get_texture(rectangle)
2021-04-16 23:21:06 +08:00
def get_image_data(self):
return self.image.get_image_data()
def get_texture_sequence(self):
if not self._texture_grid:
self._texture_grid = TextureGrid(self)
return self._texture_grid
def __len__(self):
return self.rows * self.columns
def _update_items(self):
if not self._items:
self._items = []
y = 0
for row in range(self.rows):
x = 0
for col in range(self.columns):
self._items.append(self.image.get_region(x, y, self.item_width, self.item_height))
2021-04-16 23:21:06 +08:00
x += self.item_width + self.column_padding
y += self.item_height + self.row_padding
def __getitem__(self, index):
self._update_items()
if type(index) is tuple:
row, column = index
assert 0 <= row < self.rows and 0 <= column < self.columns
2021-04-16 23:21:06 +08:00
return self._items[row * self.columns + column]
else:
return self._items[index]
def __iter__(self):
self._update_items()
return iter(self._items)
class TextureGrid(TextureRegion, UniformTextureSequence):
"""A texture containing a regular grid of texture regions.
To construct, create an :py:class:`~pyglet.image.ImageGrid` first::
image_grid = ImageGrid(...)
texture_grid = TextureGrid(image_grid)
The texture grid can be accessed as a single texture, or as a sequence
of :py:class:`~pyglet.image.TextureRegion`. When accessing as a sequence, you can specify
integer indexes, in which the images are arranged in rows from the
bottom-left to the top-right::
# assume the texture_grid is 3x3:
current_texture = texture_grid[3] # get the middle-left image
You can also specify tuples in the sequence methods, which are addressed
as ``row, column``::
# equivalent to the previous example:
current_texture = texture_grid[1, 0]
When using tuples in a slice, the returned sequence is over the
rectangular region defined by the slice::
# returns center, center-right, center-top, top-right images in that
# order:
images = texture_grid[(1,1):]
# equivalent to
images = texture_grid[(1,1):(3,3)]
"""
items = ()
rows = 1
columns = 1
item_width = 0
item_height = 0
def __init__(self, grid):
image = grid.get_texture()
if isinstance(image, TextureRegion):
owner = image.owner
else:
owner = image
super().__init__(image.x, image.y, image.z, image.width, image.height, owner)
2021-04-16 23:21:06 +08:00
items = []
y = 0
for row in range(grid.rows):
x = 0
for col in range(grid.columns):
items.append(self.get_region(x, y, grid.item_width, grid.item_height))
2021-04-16 23:21:06 +08:00
x += grid.item_width + grid.column_padding
y += grid.item_height + grid.row_padding
self.items = items
self.rows = grid.rows
self.columns = grid.columns
self.item_width = grid.item_width
self.item_height = grid.item_height
def get(self, row, column):
return self[(row, column)]
def __getitem__(self, index):
if type(index) is slice:
if type(index.start) is not tuple and type(index.stop) is not tuple:
return self.items[index]
else:
row1 = 0
col1 = 0
row2 = self.rows
col2 = self.columns
if type(index.start) is tuple:
row1, col1 = index.start
elif type(index.start) is int:
row1 = index.start // self.columns
col1 = index.start % self.columns
assert 0 <= row1 < self.rows and 0 <= col1 < self.columns
2021-04-16 23:21:06 +08:00
if type(index.stop) is tuple:
row2, col2 = index.stop
elif type(index.stop) is int:
row2 = index.stop // self.columns
col2 = index.stop % self.columns
assert 0 <= row2 <= self.rows and 0 <= col2 <= self.columns
2021-04-16 23:21:06 +08:00
result = []
i = row1 * self.columns
for row in range(row1, row2):
result += self.items[i + col1:i + col2]
i += self.columns
return result
else:
if type(index) is tuple:
row, column = index
assert 0 <= row < self.rows and 0 <= column < self.columns
2021-04-16 23:21:06 +08:00
return self.items[row * self.columns + column]
elif type(index) is int:
return self.items[index]
def __setitem__(self, index, value):
if type(index) is slice:
for region, image in zip(self[index], value):
if image.width != self.item_width or image.height != self.item_height:
raise ImageException('Image has incorrect dimensions')
image.blit_into(region, image.anchor_x, image.anchor_y, 0)
else:
image = value
if image.width != self.item_width or image.height != self.item_height:
raise ImageException('Image has incorrect dimensions')
image.blit_into(self[index], image.anchor_x, image.anchor_y, 0)
def __len__(self):
return len(self.items)
def __iter__(self):
return iter(self.items)
# Initialise default codecs
2022-04-08 23:07:41 +08:00
_add_default_codecs()
# Default Framebuffer classes:
###############################################################
class BufferManager:
"""Manages the set of framebuffers for a context.
Use :py:func:`~pyglet.image.get_buffer_manager` to obtain the instance of this class for the
current context.
"""
def __init__(self):
self.color_buffer = None
self.depth_buffer = None
stencil_bits = GLint()
glGetFramebufferAttachmentParameteriv(
GL_DRAW_FRAMEBUFFER,
GL_STENCIL,
GL_FRAMEBUFFER_ATTACHMENT_STENCIL_SIZE,
stencil_bits,
)
self.free_stencil_bits = list(range(stencil_bits.value))
self.refs = []
def get_viewport(self):
"""Get the current OpenGL viewport dimensions.
:rtype: 4-tuple of float.
:return: Left, top, right and bottom dimensions.
"""
viewport = (GLint * 4)()
glGetIntegerv(GL_VIEWPORT, viewport)
return viewport
def get_color_buffer(self):
"""Get the color buffer.
:rtype: :py:class:`~pyglet.image.ColorBufferImage`
"""
viewport = self.get_viewport()
viewport_width = viewport[2]
viewport_height = viewport[3]
if (not self.color_buffer or
viewport_width != self.color_buffer.width or
viewport_height != self.color_buffer.height):
self.color_buffer = ColorBufferImage(*viewport)
return self.color_buffer
def get_depth_buffer(self):
"""Get the depth buffer.
:rtype: :py:class:`~pyglet.image.DepthBufferImage`
"""
viewport = self.get_viewport()
viewport_width = viewport[2]
viewport_height = viewport[3]
if (not self.depth_buffer or
viewport_width != self.depth_buffer.width or
viewport_height != self.depth_buffer.height):
self.depth_buffer = DepthBufferImage(*viewport)
return self.depth_buffer
def get_buffer_mask(self):
"""Get a free bitmask buffer.
A bitmask buffer is a buffer referencing a single bit in the stencil
buffer. If no bits are free, `ImageException` is raised. Bits are
released when the bitmask buffer is garbage collected.
:rtype: :py:class:`~pyglet.image.BufferImageMask`
"""
if not self.free_stencil_bits:
raise ImageException('No free stencil bits are available.')
stencil_bit = self.free_stencil_bits.pop(0)
x, y, width, height = self.get_viewport()
buffer = BufferImageMask(x, y, width, height)
buffer.stencil_bit = stencil_bit
def release_buffer(ref, self=self):
self.free_stencil_bits.insert(0, stencil_bit)
self.refs.append(weakref.ref(buffer, release_buffer))
return buffer
def get_buffer_manager():
"""Get the buffer manager for the current OpenGL context.
:rtype: :py:class:`~pyglet.image.BufferManager`
"""
context = pyglet.gl.current_context
if not hasattr(context, 'image_buffer_manager'):
context.image_buffer_manager = BufferManager()
return context.image_buffer_manager
class BufferImage(AbstractImage):
"""An abstract framebuffer.
"""
#: The OpenGL read and write target for this buffer.
gl_buffer = GL_BACK
#: The OpenGL format constant for image data.
gl_format = 0
#: The format string used for image data.
format = ''
owner = None
# TODO: enable methods
def __init__(self, x, y, width, height):
super().__init__(width, height)
self.x = x
self.y = y
self.width = width
self.height = height
def get_image_data(self):
buf = (GLubyte * (len(self.format) * self.width * self.height))()
x = self.x
y = self.y
if self.owner:
x += self.owner.x
y += self.owner.y
glReadBuffer(self.gl_buffer)
glPixelStorei(GL_PACK_ALIGNMENT, 1)
glReadPixels(x, y, self.width, self.height, self.gl_format, GL_UNSIGNED_BYTE, buf)
return ImageData(self.width, self.height, self.format, buf)
def get_region(self, x, y, width, height):
if self.owner:
return self.owner.get_region(x + self.x, y + self.y, width, height)
region = self.__class__(x + self.x, y + self.y, width, height)
region.gl_buffer = self.gl_buffer
region.owner = self
return region
class ColorBufferImage(BufferImage):
"""A color framebuffer.
This class is used to wrap the primary color buffer (i.e., the back
buffer)
"""
gl_format = GL_RGBA
format = 'RGBA'
def get_texture(self, rectangle=False):
target = GL_TEXTURE_RECTANGLE if rectangle else GL_TEXTURE_2D
texture = Texture.create(self.width, self.height, target, GL_RGBA)
self.blit_to_texture(texture.target, texture.level, self.anchor_x, self.anchor_y, 0)
return texture
def blit_to_texture(self, target, level, x, y, z):
glReadBuffer(self.gl_buffer)
glCopyTexSubImage2D(target, level, x-self.anchor_x, y-self.anchor_y, self.x, self.y, self.width, self.height)
class DepthBufferImage(BufferImage):
"""The depth buffer.
"""
gl_format = GL_DEPTH_COMPONENT
format = 'L'
def get_texture(self, rectangle=False):
assert rectangle is False, 'Depth textures cannot be rectangular'
texture = DepthTexture.create(self.width, self.height, GL_TEXTURE_2D, None)
if self.anchor_x or self.anchor_y:
texture.anchor_x = self.anchor_x
texture.anchor_y = self.anchor_y
glReadBuffer(self.gl_buffer)
glCopyTexImage2D(texture.target, 0,
GL_DEPTH_COMPONENT,
self.x, self.y, self.width, self.height,
0)
return texture
def blit_to_texture(self, target, level, x, y, z):
glReadBuffer(self.gl_buffer)
glCopyTexSubImage2D(target, level, x-self.anchor_x, y-self.anchor_y, self.x, self.y, self.width, self.height)
class BufferImageMask(BufferImage):
"""A single bit of the stencil buffer.
"""
gl_format = GL_STENCIL_INDEX
format = 'L'
# TODO mask methods