mirror of
https://github.com/quantum5/punyverse.git
synced 2025-04-24 13:11:57 -04:00
249 lines
7.1 KiB
Python
249 lines
7.1 KiB
Python
from __future__ import print_function
|
|
|
|
import itertools
|
|
import os.path
|
|
import struct
|
|
from ctypes import c_int, byref, c_uint
|
|
from io import BytesIO
|
|
|
|
import six
|
|
from pyglet import image
|
|
from pyglet.gl import *
|
|
from six.moves import zip, range
|
|
|
|
try:
|
|
from ._glgeom import bgr_to_rgb, flip_vertical
|
|
except ImportError:
|
|
import warnings
|
|
warnings.warn('Compile _glgeom.c, or double the start up time.')
|
|
|
|
def bgr_to_rgb(source, width, height, alpha=False):
|
|
length = len(source)
|
|
depth = length // (width * height)
|
|
depth2 = depth - alpha
|
|
result = bytearray(length)
|
|
row = width * depth
|
|
for y in range(height):
|
|
for x in range(width):
|
|
offset = y * row + x * depth
|
|
for i in range(depth2):
|
|
result[offset + i] = source[offset + depth2 - i - 1]
|
|
if alpha:
|
|
result[offset + depth2] = source[offset + depth2]
|
|
return six.binary_type(result)
|
|
|
|
|
|
def flip_vertical(source, width, height):
|
|
length = len(source)
|
|
row = length // height
|
|
result = bytearray(length)
|
|
for y1 in range(height):
|
|
y2 = height - y1 - 1
|
|
result[y1 * row:y1 * row + row] = source[y2 * row:y2 * row + row]
|
|
return six.binary_type(result)
|
|
|
|
__all__ = ['load_texture', 'load_clouds', 'load_image', 'get_best_texture']
|
|
|
|
id = 0
|
|
cache = {}
|
|
|
|
max_texture = None
|
|
power_of_two = None
|
|
bgra = False
|
|
|
|
|
|
def init():
|
|
global max_texture, power_of_two, bgra
|
|
|
|
if max_texture is None:
|
|
buf = c_int()
|
|
glGetIntegerv(GL_MAX_TEXTURE_SIZE, byref(buf))
|
|
max_texture = buf.value
|
|
bgra = gl_info.have_extension('GL_EXT_bgra')
|
|
|
|
if power_of_two is None:
|
|
power_of_two = gl_info.have_version(2) or gl_info.have_extension('GL_ARB_texture_non_power_of_two')
|
|
|
|
|
|
is_power2 = lambda num: num != 0 and ((num & (num - 1)) == 0)
|
|
|
|
|
|
def image_info(data):
|
|
data = six.binary_type(data)
|
|
size = len(data)
|
|
height = -1
|
|
width = -1
|
|
content_type = ''
|
|
|
|
# handle GIFs
|
|
if size >= 10 and data[:6] in (b'GIF87a', b'GIF89a'):
|
|
# Check to see if content_type is correct
|
|
content_type = 'image/gif'
|
|
w, h = struct.unpack('<HH', data[6:10])
|
|
width = int(w)
|
|
height = int(h)
|
|
|
|
# See PNG 2. Edition spec (http://www.w3.org/TR/PNG/)
|
|
# Bytes 0-7 are below, 4-byte chunk length, then 'IHDR'
|
|
# and finally the 4-byte width, height
|
|
elif size >= 24 and data.startswith(b'\211PNG\r\n\032\n') and data[12:16] == b'IHDR':
|
|
content_type = 'image/png'
|
|
w, h = struct.unpack('>LL', data[16:24])
|
|
width = int(w)
|
|
height = int(h)
|
|
|
|
# Maybe this is for an older PNG version.
|
|
elif size >= 16 and data.startswith(b'\211PNG\r\n\032\n'):
|
|
content_type = 'image/png'
|
|
w, h = struct.unpack('>LL', data[8:16])
|
|
width = int(w)
|
|
height = int(h)
|
|
|
|
# handle JPEGs
|
|
elif size >= 2 and data.startswith(b'\377\330'):
|
|
content_type = 'image/jpeg'
|
|
jpeg = BytesIO(data)
|
|
jpeg.read(2)
|
|
b = jpeg.read(1)
|
|
try:
|
|
while b and ord(b) != 0xDA:
|
|
while ord(b) != 0xFF:
|
|
b = jpeg.read(1)
|
|
while ord(b) == 0xFF:
|
|
b = jpeg.read(1)
|
|
if 0xC0 <= ord(b) <= 0xC3:
|
|
jpeg.read(3)
|
|
height, width = struct.unpack('>HH', jpeg.read(4))
|
|
break
|
|
else:
|
|
jpeg.read(int(struct.unpack('>H', jpeg.read(2))[0]) - 2)
|
|
b = jpeg.read(1)
|
|
except struct.error:
|
|
pass
|
|
except ValueError:
|
|
pass
|
|
|
|
return content_type, width, height
|
|
|
|
|
|
def check_size(width, height):
|
|
init()
|
|
if width > max_texture or height > max_texture:
|
|
print('too large')
|
|
raise ValueError('Texture too large')
|
|
elif not power_of_two:
|
|
if not is_power2(width) or not is_power2(height):
|
|
print('not power of two')
|
|
raise ValueError('Texture not power of two')
|
|
|
|
|
|
def load_image(file, path):
|
|
print('Loading image %s...' % file, end=' ')
|
|
|
|
try:
|
|
file = open(path, 'rb')
|
|
except IOError:
|
|
print('does not exist')
|
|
raise ValueError('Texture does not exist')
|
|
|
|
type, width, height = image_info(file.read(65536))
|
|
file.seek(0, 0)
|
|
if type:
|
|
check_size(width, height)
|
|
|
|
try:
|
|
raw = image.load(path, file=file)
|
|
except Exception:
|
|
print('cannot be loaded')
|
|
raise ValueError('cannot be loaded')
|
|
|
|
width, height = raw.width, raw.height
|
|
check_size(width, height)
|
|
print()
|
|
|
|
mode = GL_RGBA if 'A' in raw.format else GL_RGB
|
|
|
|
# Flip from BGR to RGB
|
|
if raw.format in ('BGR', 'BGRA'):
|
|
if bgra:
|
|
mode = GL_BGRA if 'A' in raw.format else GL_BGR
|
|
texture = raw.data
|
|
else:
|
|
texture = bgr_to_rgb(raw.data, width, height, 'A' in raw.format)
|
|
elif raw.format in ('RGB', 'RGBA'):
|
|
texture = raw.data
|
|
else:
|
|
texture = raw.get_data('RGBA', width * 4)
|
|
|
|
return path, width, height, len(raw.format), mode, flip_vertical(texture, width, height)
|
|
|
|
|
|
def load_texture(file):
|
|
if os.path.isabs(file):
|
|
path = file
|
|
file = os.path.basename(path)
|
|
else:
|
|
path = os.path.join(os.path.dirname(__file__), 'assets', 'textures', file)
|
|
|
|
if path in cache:
|
|
return cache[path]
|
|
|
|
path, width, height, depth, mode, texture = load_image(file, path)
|
|
|
|
buffer = c_uint()
|
|
glGenTextures(1, byref(buffer))
|
|
id = buffer.value
|
|
|
|
glBindTexture(GL_TEXTURE_2D, id)
|
|
|
|
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR)
|
|
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR_MIPMAP_LINEAR)
|
|
gluBuild2DMipmaps(GL_TEXTURE_2D, depth, width, height, mode, GL_UNSIGNED_BYTE, texture)
|
|
|
|
cache[path] = id
|
|
return id
|
|
|
|
|
|
def load_clouds(file):
|
|
if os.path.isabs(file):
|
|
path = file
|
|
file = os.path.basename(path)
|
|
else:
|
|
path = os.path.join(os.path.dirname(__file__), 'assets', 'textures', file)
|
|
|
|
if path in cache:
|
|
return cache[path]
|
|
|
|
path, width, height, depth, mode, texture = load_image(file, path)
|
|
|
|
buffer = c_uint()
|
|
glGenTextures(1, byref(buffer))
|
|
id = buffer.value
|
|
|
|
pixels = bytearray(len(texture) * 4)
|
|
white = b'\xff'[0]
|
|
pixels[:] = itertools.chain.from_iterable(zip(itertools.repeat(white), itertools.repeat(white),
|
|
itertools.repeat(white),
|
|
itertools.islice(texture, 0, None, depth)))
|
|
|
|
glBindTexture(GL_TEXTURE_2D, id)
|
|
|
|
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR)
|
|
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR)
|
|
glTexImage2D(GL_TEXTURE_2D, 0, GL_RGBA, width, height, 0, GL_RGBA, GL_UNSIGNED_BYTE, six.binary_type(pixels))
|
|
|
|
cache[path] = id
|
|
return id
|
|
|
|
|
|
def get_best_texture(info, loader=load_texture):
|
|
if isinstance(info, list):
|
|
for item in info:
|
|
try:
|
|
return loader(item)
|
|
except ValueError:
|
|
pass
|
|
else:
|
|
return loader(info)
|
|
raise ValueError('No texture found')
|