使用Glut和OpenGL与Python捕获屏幕

时间:2015-08-17 13:55:32

标签: python opengl glut

我正在使用Python和OpenGL进行一些与物理系统相关的计算,然后显示它们并能够操纵旋转它的图像,翻译它,......

我不是一名专业程序员,我没有使用OpenGL的经验,而且我在尝试获得我想要的东西时遇到了一些问题。我希望能够将我得到的窗口保存为视频文件中的输出,并且这样做我已经看到使用glReadPixels捕获每个帧然后将它们全部放在一起的可能性。

我正在使用的东西看起来像下面的代码,我希望能够将帧保存到图像,但是,虽然我已经能够保存以使用glReadPixels将像素保存到数组,我不知道在哪里或如何调用我定义为captureScreen的函数。我该怎么做以获得我正在寻找的输出?

from OpenGL.GL import *
from OpenGL.GLU import *
from OpenGL.GLUT import *
from PIL import Image
from scipy import *
import numpy as np
import random as rnd

def captureScreen(file_):

    glPixelStorei(GL_PACK_ALIGNMENT, 1)
    data = glReadPixels(0, 0, 800, 800, GL_RGBA, GL_UNSIGNED_BYTE)
    image = Image.fromstring("RGBA", (800, 800), data)
    image.save(file_, 'png')

def main():  

     glutInit(sys.argv)
     glutInitDisplayMode (GLUT_DOUBLE | GLUT_RGB | GLUT_DEPTH)
     glutInitWindowPosition(0, 0)
     glutInitWindowSize(800, 800)

     glutCreateWindow ("Test")

     glEnable(GL_DEPTH_TEST)
     glDepthMask(GL_TRUE)
     glEnable(GL_BLEND)
     glEnable(GL_TEXTURE_2D)
     glEnable(GL_COLOR_MATERIAL)
     glBlendFunc(GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA)
     glPointSize( 6.0 )
     glLineWidth( 2.0 )
     glEnable(GL_POINT_SMOOTH)
     glEnable(GL_LINE_SMOOTH)
     glHint(GL_LINE_SMOOTH_HINT, GL_NICEST)
     glEnable(GL_POLYGON_SMOOTH)
     glHint(GL_POLYGON_SMOOTH_HINT, GL_NICEST)

     glEnableClientState(GL_VERTEX_ARRAY)
     glEnableClientState(GL_COLOR_ARRAY)

     glClearColor(0.0, 0.0, 0.0, 0.0)

     glutFullScreen()
     glutKeyboardFunc(keyboard)
     glutIdleFunc(dynamics)
     glutDisplayFunc(display)
     glutMouseFunc(mouse)


     glutMainLoop()


if __name__ == '__main__':

     main()

键盘,显示,动态和鼠标是先前定义的功能。

1 个答案:

答案 0 :(得分:2)

你可以在 glutSwapBuffers()之后调用glReadPixels():

glutSwapBuffers()
glReadBuffer(GL_FRONT)
glReadPixels(...)

之前

glReadBuffer(GL_BACK)
glReadPixels(...)
glutSwapBuffers()

所以应该从你的"显示"中调用captureScreen()。函数可能是调用glutSwapBuffers()

的函数