如何加载动画GIF并获取Pygame中的所有单独帧?

时间:2015-04-10 22:06:39

标签: python image pygame gif

首先,对不起,如果这是重复的话。我发现的答案似乎无关紧要,但也许我用错误的关键词进行搜索。我想做的是拍摄动画GIF并将其拆分为帧列表。基本上,这样的事情:

frames = []
for frame in split_animated_gif("some_animated_gif.gif"):
    frames.append(frame)

其中split_animated_gif按顺序返回一个曲面列表,每个曲面都是GIF的一个帧。谢谢你的帮助。

编辑:在进行了一些窥探后,我找到了一段代码,在我的pygame中成功显示了一个动画GIF。它可以在https://github.com/piantado/kelpy/blob/master/kelpy/GIFImage.py找到。不过,我非常感谢你的帮助。

3 个答案:

答案 0 :(得分:2)

Pygame本身不支持GIF动画,这在文档中有说明。所以你必须

1)当你加载你的精灵或图像时,使用其他一些代码/库来分割gif。你可以通过谷歌搜索" python split gif"或者其他的东西。例如。 Python: Converting GIF frames to PNG

2)如果你自己创建了gif,只需逐帧导出你的精灵

无论哪种方式,您都必须手动制作动画部分。您可以通过Google搜索和#34; pygame animation"找到大量的教程。例如。 Animated sprite from few images

答案 1 :(得分:1)

此代码使用PIL / pillow库来实现。

from PIL import Image

def split_animated_gif(gif_file_path):
    ret = []
    gif = Image.open(gif_file_path)
    for frame_index in range(gif.n_frames):
        gif.seek(frame_index)
        frame_rgba = gif.convert("RGBA")
        pygame_image = pygame.image.fromstring(
            frame_rgba.tobytes(), frame_rgba.size, frame_rgba.mode
        )
        ret.append(pygame_image)
    return ret

答案 2 :(得分:0)

PyGame模块和pygame.image模块只能处理非动画的GIFs
但是在PyGame主页上引入了GIFImage库:

该库将GIF动画回放添加到pygame。


另一种选择是使用库将GIF逐帧加载到列表。

例如,使用Pillow库(pip install Pillow)。

编写一个函数,该函数可以将PIL图像转换为pygame.Surface,并使用PIL库逐帧加载GIF:
(另请参见Extracting The Frames Of An Animated GIF Using PillowPIL and pygame.image

from PIL import Image, ImageSequence
def pilImageToSurface(pilImage):
    mode, size, data = pilImage.mode, pilImage.size, pilImage.tobytes()
    return pygame.image.fromstring(data, size, mode).convert_alpha()

def loadGIF(filename):
    pilImage = Image.open(filename)
    frames = []
    if pilImage.format == 'GIF' and pilImage.is_animated:
        for frame in ImageSequence.Iterator(pilImage):
            pygameImage = pilImageToSurface(frame.convert('RGBA'))
            frames.append(pygameImage)
    else:
        frames.append(pilImageToSurface(pilImage))
    return frames

或使用OpenCV / opencv-python库和VideoCapture

编写一个可以将cv2 pygame.image图像转换为pygame.Surface的函数,并使用该库逐帧加载GIF:
(另请参见Read GIF files in PythonHow do I convert an OpenCV (cv2) image (BGR and BGRA) to a pygame.Surface object

import cv2
def cv2ImageToSurface(cv2Image):
    size = cv2Image.shape[1::-1]
    format = 'RGBA' if cv2Image.shape[2] == 4 else 'RGB'
    cv2Image[:, :, [0, 2]] = cv2Image[:, :, [2, 0]]
    surface = pygame.image.frombuffer(cv2Image.flatten(), size, format)
    return surface.convert_alpha() if format == 'RGBA' else surface.convert()

def loadGIF(filename):
    gif = cv2.VideoCapture(filename)
    frames = []
    while True:
        ret, cv2Image = gif.read()
        if not ret:
            break
        pygameImage = cv2ImageToSurface(cv2Image)
        frames.append(pygameImage)
    return frames

请参见简单的gif动画查看器示例:

import pygame
import cv2

def cv2ImageToSurface(cv2Image):
    size = cv2Image.shape[1::-1]
    format = 'RGBA' if cv2Image.shape[2] == 4 else 'RGB'
    cv2Image[:, :, [0, 2]] = cv2Image[:, :, [2, 0]]
    surface = pygame.image.frombuffer(cv2Image.flatten(), size, format)
    return surface.convert_alpha() if format == 'RGBA' else surface.convert()

def loadGIF(filename):
    gif = cv2.VideoCapture(filename)
    frames = []
    while True:
        ret, cv2Image = gif.read()
        if not ret:
            break
        pygameImage = cv2ImageToSurface(cv2Image)
        frames.append(pygameImage)
    return frames

pygame.init()
window = pygame.display.set_mode((500, 500))
clock = pygame.time.Clock()

gifFrameList = loadGIF(r"rubber_ball.gif")
currentFrame = 0

run = True
while run:
    clock.tick(20)
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            run = False

    window.fill(0)

    rect = gifFrameList[currentFrame].get_rect(center = (250, 250))
    window.blit(gifFrameList[currentFrame], rect)
    currentFrame = (currentFrame + 1) % len(gifFrameList)

    pygame.display.flip()