141 lines
6.2 KiB
Python
141 lines
6.2 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.
|
||
|
|
# ----------------------------------------------------------------------------
|
||
|
|
|
||
|
|
from ctypes import c_void_p, c_ubyte
|
||
|
|
|
||
|
|
from pyglet.image import ImageData, Animation, AnimationFrame
|
||
|
|
from pyglet.image.codecs import *
|
||
|
|
|
||
|
|
from pyglet.libs.darwin.cocoapy import cf, quartz, NSMakeRect
|
||
|
|
from pyglet.libs.darwin.cocoapy import cfnumber_to_number
|
||
|
|
from pyglet.libs.darwin.cocoapy import kCGImageAlphaPremultipliedLast
|
||
|
|
from pyglet.libs.darwin.cocoapy import kCGImagePropertyGIFDictionary
|
||
|
|
from pyglet.libs.darwin.cocoapy import kCGImagePropertyGIFDelayTime
|
||
|
|
|
||
|
|
|
||
|
|
class QuartzImageDecoder(ImageDecoder):
|
||
|
|
def get_file_extensions(self):
|
||
|
|
# Quartz can actually decode many more formats, but these are the most common.
|
||
|
|
return [ '.bmp', '.cur', '.gif', '.ico', '.jp2', '.jpg', '.jpeg',
|
||
|
|
'.pcx', '.png', '.tga', '.tif', '.tiff', '.xbm', '.xpm' ]
|
||
|
|
|
||
|
|
def get_animation_file_extensions(self):
|
||
|
|
return ['.gif']
|
||
|
|
|
||
|
|
def _get_pyglet_ImageData_from_source_at_index(self, sourceRef, index):
|
||
|
|
imageRef = c_void_p(quartz.CGImageSourceCreateImageAtIndex(sourceRef, index, None))
|
||
|
|
|
||
|
|
# Regardless of the internal format of the image (L, LA, RGB, RGBA, etc)
|
||
|
|
# we just automatically convert everything to an RGBA format.
|
||
|
|
format = 'RGBA'
|
||
|
|
rgbColorSpace = c_void_p(quartz.CGColorSpaceCreateDeviceRGB())
|
||
|
|
bitsPerComponent = 8
|
||
|
|
width = quartz.CGImageGetWidth(imageRef)
|
||
|
|
height = quartz.CGImageGetHeight(imageRef)
|
||
|
|
bytesPerRow = 4 * width
|
||
|
|
|
||
|
|
# Create a buffer to store the RGBA formatted data.
|
||
|
|
bufferSize = height * bytesPerRow
|
||
|
|
buffer = (c_ubyte * bufferSize)()
|
||
|
|
|
||
|
|
# Create a bitmap context for the RGBA formatted data.
|
||
|
|
# Note that premultiplied alpha is required:
|
||
|
|
# http://developer.apple.com/library/mac/#qa/qa1037/_index.html
|
||
|
|
bitmap = c_void_p(quartz.CGBitmapContextCreate(buffer,
|
||
|
|
width, height,
|
||
|
|
bitsPerComponent,
|
||
|
|
bytesPerRow,
|
||
|
|
rgbColorSpace,
|
||
|
|
kCGImageAlphaPremultipliedLast))
|
||
|
|
|
||
|
|
# Write the image data into the bitmap.
|
||
|
|
quartz.CGContextDrawImage(bitmap, NSMakeRect(0,0,width,height), imageRef)
|
||
|
|
|
||
|
|
quartz.CGImageRelease(imageRef)
|
||
|
|
quartz.CGContextRelease(bitmap)
|
||
|
|
quartz.CGColorSpaceRelease(rgbColorSpace)
|
||
|
|
|
||
|
|
pitch = bytesPerRow
|
||
|
|
return ImageData(width, height, format, buffer, -pitch)
|
||
|
|
|
||
|
|
def decode(self, file, filename):
|
||
|
|
file_bytes = file.read()
|
||
|
|
data = c_void_p(cf.CFDataCreate(None, file_bytes, len(file_bytes)))
|
||
|
|
# Second argument is an options dictionary. It might be a good idea to provide
|
||
|
|
# a value for kCGImageSourceTypeIdentifierHint here using filename extension.
|
||
|
|
sourceRef = c_void_p(quartz.CGImageSourceCreateWithData(data, None))
|
||
|
|
image = self._get_pyglet_ImageData_from_source_at_index(sourceRef, 0)
|
||
|
|
|
||
|
|
cf.CFRelease(data)
|
||
|
|
cf.CFRelease(sourceRef)
|
||
|
|
|
||
|
|
return image
|
||
|
|
|
||
|
|
def decode_animation(self, file, filename):
|
||
|
|
# If file is not an animated GIF, it will be loaded as a single-frame animation.
|
||
|
|
file_bytes = file.read()
|
||
|
|
data = c_void_p(cf.CFDataCreate(None, file_bytes, len(file_bytes)))
|
||
|
|
sourceRef = c_void_p(quartz.CGImageSourceCreateWithData(data, None))
|
||
|
|
|
||
|
|
# Get number of frames in the animation.
|
||
|
|
count = quartz.CGImageSourceGetCount(sourceRef)
|
||
|
|
|
||
|
|
frames = []
|
||
|
|
|
||
|
|
for index in range(count):
|
||
|
|
# Try to determine frame duration from GIF properties dictionary.
|
||
|
|
duration = 0.1 # default duration if none found
|
||
|
|
props = c_void_p(quartz.CGImageSourceCopyPropertiesAtIndex(sourceRef, index, None))
|
||
|
|
if cf.CFDictionaryContainsKey(props, kCGImagePropertyGIFDictionary):
|
||
|
|
gif_props = c_void_p(cf.CFDictionaryGetValue(props, kCGImagePropertyGIFDictionary))
|
||
|
|
if cf.CFDictionaryContainsKey(gif_props, kCGImagePropertyGIFDelayTime):
|
||
|
|
duration = cfnumber_to_number(c_void_p(cf.CFDictionaryGetValue(gif_props, kCGImagePropertyGIFDelayTime)))
|
||
|
|
|
||
|
|
cf.CFRelease(props)
|
||
|
|
image = self._get_pyglet_ImageData_from_source_at_index(sourceRef, index)
|
||
|
|
frames.append( AnimationFrame(image, duration) )
|
||
|
|
|
||
|
|
cf.CFRelease(data)
|
||
|
|
cf.CFRelease(sourceRef)
|
||
|
|
|
||
|
|
return Animation(frames)
|
||
|
|
|
||
|
|
|
||
|
|
def get_decoders():
|
||
|
|
return [ QuartzImageDecoder() ]
|
||
|
|
|
||
|
|
def get_encoders():
|
||
|
|
return []
|