如何将pygame场景保存为jpeg?

问题描述

我需要执行以下操作。我使用 OBJFileLoader 加载了一个 obj 模型并将其渲染到一个窗口。如何将渲染的场景保存为图片? 是否可以在不显示窗口的情况下执行此操作?

解决方法

您可以保存一个 pygame.Surface 对象,对象为与带有 pygame.image.save() 的屏幕关联的Surface

这会将您的 Surface 保存为 BMP、TGA、PNG 或 JPEG 图像。

screen = pygame.display.set_mode((w,h))

# [...]

pygame.image.save(screen,"screenshot.jpg")

但是,这不适用于 pygame.OPENGL 表面。在更新显示之前(在 pygame.display.flip()pygame.display.update() 之前),您必须使用 glReadPixels 读取帧缓冲区。使用 pygame.image.fromstring() 从缓冲区创建新的表面。最后,将表面保存到一个文件中:

screen = pygame.display.set_mode((w,h),pygame.DOUBLEBUF | pygame.OPENGL)

# [...]

size = screen.get_size()
buffer = glReadPixels(0,*size,GL_RGBA,GL_UNSIGNED_BYTE)
pygame.display.flip()

screen_surf = pygame.image.fromstring(buffer,size,"RGBA")
pygame.image.save(screen_surf,"screenshot.jpg")