Mercurial > touhou
view pytouhou/ui/texture.pyx @ 402:88e2a2485b2b
Add a version attribute to ANM0 and latest discovered instruction.
author | Emmanuel Gil Peyrot <linkmauve@linkmauve.fr> |
---|---|
date | Sat, 16 Feb 2013 19:31:29 +0100 |
parents | 9e2cbb2c2c64 |
children | 3a7b36324611 |
line wrap: on
line source
# -*- encoding: utf-8 -*- ## ## Copyright (C) 2011 Thibaut Girka <thib@sitedethib.com> ## ## This program is free software; you can redistribute it and/or modify ## it under the terms of the GNU General Public License as published ## by the Free Software Foundation; version 3 only. ## ## This program is distributed in the hope that it will be useful, ## but WITHOUT ANY WARRANTY; without even the implied warranty of ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the ## GNU General Public License for more details. ## from libc.stdlib cimport malloc, free import pyglet from pyglet.gl import (glTexParameteri, GL_TEXTURE_MIN_FILTER, GL_TEXTURE_MAG_FILTER, GL_LINEAR, GL_BGRA, GL_RGBA, GL_RGB, GL_LUMINANCE, GL_UNSIGNED_BYTE, GL_UNSIGNED_SHORT_5_6_5, GL_UNSIGNED_SHORT_4_4_4_4_REV, glGenTextures, glBindTexture, glTexImage2D, GL_TEXTURE_2D) from ctypes import c_uint, byref import os from pytouhou.formats.thtx import Texture #TODO: perhaps define that elsewhere? cdef class TextureManager: def __init__(self, loader=None): self.loader = loader self.textures = {} def __getitem__(self, key): if not key in self.textures: self.textures[key] = self.load_texture(key) return self.textures[key] def preload(self, anm_wrapper): for anm in anm_wrapper: anm.texture = self.load_png_texture(anm.first_name, anm.secondary_name) def load_png_texture(self, first_name, secondary_name): cdef char *image, *alpha, *new_data cdef unsigned int i, width, height, pixels image_file = pyglet.image.load(first_name, file=self.loader.get_file(os.path.basename(first_name))) width, height = image_file.width, image_file.height image_data = image_file.get_data('RGB', width * 3) if secondary_name: alpha_file = pyglet.image.load(secondary_name, file=self.loader.get_file(os.path.basename(secondary_name))) assert (image_file.width, image_file.height) == (alpha_file.width, image_file.height) pixels = width * height alpha_data = alpha_file.get_data('RGB', width * 3) image = <char *>image_data alpha = <char *>alpha_data # TODO: further optimizations new_data = <char *>malloc(pixels * 4) for i in range(pixels): new_data[i*4] = image[i*3] new_data[i*4+1] = image[i*3+1] new_data[i*4+2] = image[i*3+2] new_data[i*4+3] = alpha[i*3] data = new_data[:(pixels * 4)] free(new_data) return Texture(width, height, -4, data) return Texture(width, height, -3, image_data) def load_texture(self, key): if not isinstance(key, Texture): first_name, secondary_name = key key = self.load_png_texture(first_name, secondary_name) if key.fmt == 1: format_ = GL_BGRA type_ = GL_UNSIGNED_BYTE composants = GL_RGBA elif key.fmt == 3: format_ = GL_RGB type_ = GL_UNSIGNED_SHORT_5_6_5 composants = GL_RGB elif key.fmt == 5: format_ = GL_BGRA type_ = GL_UNSIGNED_SHORT_4_4_4_4_REV composants = GL_RGBA elif key.fmt == 7: format_ = GL_LUMINANCE type_ = GL_UNSIGNED_BYTE composants = GL_LUMINANCE elif key.fmt == -3: #XXX: non-standard, remove it! format_ = GL_RGB type_ = GL_UNSIGNED_BYTE composants = GL_RGB elif key.fmt == -4: #XXX: non-standard format_ = GL_RGBA type_ = GL_UNSIGNED_BYTE composants = GL_RGBA else: raise Exception('Unknown texture type') id_ = c_uint() glGenTextures(1, byref(id_)) glBindTexture(GL_TEXTURE_2D, id_.value) glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR) glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR) glTexImage2D(GL_TEXTURE_2D, 0, composants, key.width, key.height, 0, format_, type_, key.data) return id_.value