Spaces:
Runtime error
Runtime error
| 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, filename, file): | |
| if not file: | |
| file = open(filename, 'rb') | |
| 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, filename, file): | |
| if not file: | |
| file = open(filename, 'rb') | |
| # 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 [] | |