如何在Python中高效向动画WEBP末尾追加新帧？

首先明确：无法实现纯O(1)的直接追加帧到现有WEBP动画。原因是WebP动画基于RIFF容器格式，其文件头部包含整个文件的总字节数，追加新帧后必须修改头部的这个数值；同时动画的帧信息是通过逐个ANMF块记录的，虽然帧数据可以追加，但头部的修改无法绕过，因此做不到完全的"末尾追加"式O(1)操作。

不过我们可以实现增量式的近似O(1)优化，避免每次重新编码所有旧帧，只处理新增的帧，将时间复杂度降到O(新增帧数)（当仅新增1帧时就是O(1)）。

优化思路 #

核心是缓存已编码的帧数据块，每次新增帧时只编码该帧并生成对应的WebP动画块，最后复用缓存的旧块组装完整动画，避免重复编码旧帧。

具体实现代码 #

import os
import io
from PIL import Image

FRAMES_DIR = "./frames"

# 缓存WebP动画的核心组件
_riff_prefix = None
_anim_chunk = None
_frame_chunks = []
_frame_width = 0
_frame_height = 0

def _init_animation(first_img):
    """初始化动画的全局元数据"""
    global _riff_prefix, _anim_chunk, _frame_width, _frame_height
    _frame_width, _frame_height = first_img.size
    # RIFF容器头部占位，后续更新总大小
    _riff_prefix = b"RIFF\x00\x00\x00\x00WEBP"
    # ANIM块：定义动画全局参数（无限循环、背景色、尺寸）
    anim_data = (b"\x01\x00"  # 版本
                 + b"\x00\x00"  # 无限循环
                 + b"\x00\x00\x00\x00"  # 透明背景
                 + _frame_width.to_bytes(2, byteorder='little')
                 + _frame_height.to_bytes(2, byteorder='little'))
    _anim_chunk = b"ANIM" + len(anim_data).to_bytes(4, byteorder='little') + anim_data

def _add_single_frame(img, duration=400):
    """编码单帧并生成对应的ANMF+VP8数据块"""
    global _frame_chunks
    # 编码当前帧为WebP格式，提取核心的VP8/VP8L数据块
    frame_byteio = io.BytesIO()
    img.save(frame_byteio, format='WEBP', lossless=False)
    frame_data = frame_byteio.getvalue()
    
    # 定位帧数据的VP8/VP8L块（跳过RIFF容器头）
    vp8_mark = frame_data.find(b"VP8 ")
    if vp8_mark == -1:
        vp8_mark = frame_data.find(b"VP8L")
    if vp8_mark == -1:
        raise ValueError("当前帧无法解析为WebP格式")
    vp8_chunk = frame_data[vp8_mark:]
    
    # 生成ANMF块：记录帧的时间戳、位置、尺寸等信息
    timestamp = len(_frame_chunks) * duration
    anmf_data = (timestamp.to_bytes(4, byteorder='little')
                 + b"\x00\x00\x00\x00"  # 帧偏移（全屏帧设为0）
                 + _frame_width.to_bytes(2, byteorder='little')
                 + _frame_height.to_bytes(2, byteorder='little')
                 + b"\xff\x00")  # 完全不透明、正常混合模式
    anmf_chunk = b"ANMF" + len(anmf_data).to_bytes(4, byteorder='little') + anmf_data
    
    _frame_chunks.append(anmf_chunk + vp8_chunk)

def get_webp():
    """生成包含所有帧的动画WebP，仅处理新增帧"""
    global _frame_chunks
    frame_paths = sorted(os.listdir(FRAMES_DIR))
    current_processed = len(_frame_chunks)
    
    # 处理新增的帧
    for idx in range(current_processed, len(frame_paths)):
        frame_path = os.path.join(FRAMES_DIR, frame_paths[idx])
        with Image.open(frame_path) as img:
            if idx == 0:
                _init_animation(img)
            _add_single_frame(img, duration=400)
    
    if not _anim_chunk:
        return io.BytesIO()
    
    # 计算总文件大小，更新RIFF头的大小字段
    total_size = 4 + 4 + len(_anim_chunk) + sum(len(chunk) for chunk in _frame_chunks)
    riff_header = _riff_prefix[:4] + total_size.to_bytes(4, byteorder='little') + _riff_prefix[8:]
    
    # 组装完整的WebP动画字节流
    webp_bytes = riff_header + _anim_chunk + b"".join(_frame_chunks)
    webp = io.BytesIO(webp_bytes)
    webp.seek(0)
    return webp

方案优势 #

1. 仅对新增帧进行编码和块生成，旧帧的编码结果完全复用，避免了原方案中每次重新编码所有帧的开销
2. 全程在内存中操作，无需落地临时文件
3. 每次调用仅处理新帧，当仅新增1帧时，时间复杂度接近O(1)

内容的提问来源于stack exchange，提问作者Egor