如何使用Pickle分组转储含图片的对象？解决无法序列化pygame.surface.Surface对象问题

这个问题很常见——pygame的Surface对象是和系统底层图形资源绑定的，pickle没办法序列化这类依赖外部资源的对象，因为它本质上是指向GPU内存的引用，不是纯Python可序列化的数据结构。要解决这个问题，我们需要只序列化必要的可存储数据，在加载时重新创建Surface对象，而不是直接序列化整个包含Surface的实体类或精灵组。

核心思路 #

不要在实体类中存储Surface对象，而是存储图片的路径、缩放尺寸等可以被pickle序列化的信息。当需要使用时（比如加载后），再根据这些信息重新加载图片并创建Surface。

解决方案：自定义Pickle的序列化/反序列化过程 #

最优雅的方式是给你的entity类添加__getstate__和__setstate__方法，让pickle知道哪些属性需要保存，以及如何在加载时恢复Surface对象。

修改后的完整代码 #

import pygame
import pickle
pygame.init()

# 创建实体类，添加pickle定制方法
class entity(pygame.sprite.Sprite):
    def __init__(self, img_path, x, y, identity, scale=(50,50)):
        pygame.sprite.Sprite.__init__(self)
        self.img_path = img_path  # 存储图片路径而非Surface
        self.scale = scale
        self.x = x
        self.y = y
        self.identity = identity
        # 初始化时创建Surface
        self._load_image()

    def _load_image(self):
        """内部方法：根据路径和缩放尺寸创建Surface"""
        self.img = pygame.image.load(self.img_path)
        self.img = pygame.transform.scale(self.img, self.scale)

    def draw(self, surface):
        surface.blit(self.img, (self.x, self.y))

    def __getstate__(self):
        """定义pickle时需要保存的属性"""
        state = self.__dict__.copy()
        # 移除不可序列化的Surface对象
        del state['img']
        return state

    def __setstate__(self, state):
        """定义pickle加载后如何恢复对象"""
        self.__dict__.update(state)
        # 重新创建Surface对象
        self._load_image()

# 定义精灵组
entity_group = pygame.sprite.Group()

# 屏幕参数
screen_height = 500
screen_width = 500

# 注意：现在传入的是图片路径，而非已加载的Surface
entity_0 = entity("assets/entity.png", 0, 0, "hello!")
entity_1 = entity("assets/entity 2.png", 50, 50, "hola!")
entity_2 = entity("assets/entity.png", 100, 100, "ciao!")

# 添加到精灵组
entity_group.add(entity_0, entity_1, entity_2)

# 保存精灵组到.dat文件
with open(f"entities.dat", "wb") as file:
    pickle.dump(entity_group, file)

# 清空现有组
entity_group.empty()

# 从文件加载精灵组
with open(f"entities.dat", "rb") as file:
    # 注意：pickle.load直接返回精灵组，不需要用add（因为精灵组本身被序列化了）
    entity_group = pickle.load(file)

run = True
while run:
    screen.fill((255, 55, 50))
    # 绘制实体并打印标识
    for ent in entity_group:
        ent.draw(screen)
        print(ent.identity)
    # 事件循环
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            run = False
    pygame.display.update()

pygame.quit()

关键改动说明 #

1. 实体类初始化修改：不再接收已加载的Surface，而是接收图片路径和缩放尺寸，内部通过_load_image方法创建Surface。
2. __getstate__方法：复制对象的属性字典，删除不可序列化的img（Surface对象），只保存可存储的路径、位置、标识等数据。
3. __setstate__方法：加载时先恢复保存的属性，再调用_load_image重新创建Surface对象。
4. 加载逻辑修正：之前的entity_group.add(pickle.load(file))是错误的，因为pickle.load返回的是整个精灵组，直接赋值给entity_group即可。

另一种简化方案：手动保存/加载必要数据 #

如果你不想修改类的结构，也可以手动提取精灵组中每个实体的可序列化数据，保存为列表，加载时再重新创建实体对象：

# 保存时
save_data = []
for ent in entity_group:
    save_data.append({
        'img_path': ent.img_path,  # 假设你已经把img换成了img_path
        'x': ent.x,
        'y': ent.y,
        'identity': ent.identity,
        'scale': (50,50)
    })
with open("entities.dat", "wb") as f:
    pickle.dump(save_data, f)

# 加载时
with open("entities.dat", "rb") as f:
    save_data = pickle.load(f)
for data in save_data:
    ent = entity(data['img_path'], data['x'], data['y'], data['identity'], data['scale'])
    entity_group.add(ent)

这种方式更直观，但代码量稍大，适合简单场景。

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