spacepaste

Paste details

reply | raw

  1.  
  2. import ctypes
  3. import OpenGL.GL as gl
  4. from sdl2 import * #everything has a SDL_ prefix. ick!
  5. import numpy as np
  7. W,H=640,480
  8. FPS=60
  9. frame_minticks=1000/60
  11. # === SDL/OpenGL init ===
  13. SDL_Init(SDL_INIT_VIDEO)
  15. SDL_GL_SetAttribute(SDL_GL_CONTEXT_MAJOR_VERSION,3);
  16. SDL_GL_SetAttribute(SDL_GL_CONTEXT_MINOR_VERSION,3);
  17. SDL_GL_SetAttribute(SDL_GL_DOUBLEBUFFER,1);
  18. SDL_GL_SetAttribute(SDL_GL_DEPTH_SIZE,24);
  20. window=SDL_CreateWindow("OpenGL test",100,100,W,H,SDL_WINDOW_OPENGL|SDL_WINDOW_SHOWN)
  22. context=SDL_GL_CreateContext(window)
  23. SDL_GL_SetSwapInterval(0)
  25. gl.glClearColor(1.0,1.0,1.0,1.0)
  26. gl.glClear(gl.GL_COLOR_BUFFER_BIT|gl.GL_DEPTH_BUFFER_BIT)
  27. SDL_GL_SwapWindow(window)
  29. # === shader compilation ===
  31. shader_program=gl.glCreateProgram()
  32. shader_vertex=gl.glCreateShader(gl.GL_VERTEX_SHADER)
  33. shader_fragment=gl.glCreateShader(gl.GL_FRAGMENT_SHADER)
  35. gl.glShaderSource(shader_vertex, """
  36. #version 330
  37. layout(location=0) in vec3 position;
  38. layout(location=1) in vec4 color;
  39. out vec4 v_color;
  40. void main()
  41. {
  42. gl_Position=vec4(position,1.0);
  43. v_color=color;
  44. }
  45. """)
  46. gl.glShaderSource(shader_fragment, """
  47. #version 330
  48. in vec4 v_color;
  49. out vec4 f_color;
  50. void main()
  51. {
  52. f_color=v_color;
  53. }
  54. """)
  56. gl.glCompileShader(shader_vertex)
  57. print "=== vertex shader log ==="
  58. if gl.glGetShaderiv(shader_vertex,gl.GL_COMPILE_STATUS):
  59. print "OK"
  60. else:
  61. log=gl.glGetShaderInfoLog(shader_vertex)
  62. print log
  64. gl.glCompileShader(shader_fragment)
  65. print "=== fragment shader log ==="
  66. if gl.glGetShaderiv(shader_fragment,gl.GL_COMPILE_STATUS):
  67. print "OK"
  68. else:
  69. log=gl.glGetShaderInfoLog(shader_fragment)
  70. print log
  72. print "=== end shader logs ==="
  74. gl.glAttachShader(shader_program,shader_vertex)
  75. gl.glAttachShader(shader_program,shader_fragment)
  76. gl.glLinkProgram(shader_program)
  78. gl.glUseProgram(shader_program)
  80. # === buffer creation ===
  82. vertex_data=np.zeros(4,dtype=[("position",np.float32,3),("color",np.float32,4)])
  83. vertex_data[1]=((-1,-1,1),(1,0,0,1))
  84. vertex_data[0]=(( 1,-1,1),(0,1,0,1))
  85. vertex_data[2]=(( 1, 1,1),(0,0,1,1))
  86. vertex_data[3]=((-1, 1,1),(1,1,0,1))
  88. vertex_buffer=gl.glGenBuffers(1)
  89. gl.glBindBuffer(gl.GL_ARRAY_BUFFER,vertex_buffer)
  90. gl.glBufferData(gl.GL_ARRAY_BUFFER,vertex_data.nbytes,vertex_data,gl.GL_DYNAMIC_DRAW)
  92. # === buffer binding ===
  94. stride=vertex_data.strides[0]
  96. offset=ctypes.c_void_p(0)
  97. loc=gl.glGetAttribLocation(shader_program,"position")
  98. gl.glEnableVertexAttribArray(loc)
  99. gl.glBindBuffer(gl.GL_ARRAY_BUFFER,vertex_buffer)
  100. gl.glVertexAttribPointer(loc,3,gl.GL_FLOAT,False,stride,offset)
  102. offset=ctypes.c_void_p(vertex_data.dtype["position"].itemsize)
  103. loc=gl.glGetAttribLocation(shader_program,"color")
  104. gl.glEnableVertexAttribArray(loc)
  105. gl.glBindBuffer(gl.GL_ARRAY_BUFFER,vertex_buffer)
  106. gl.glVertexAttribPointer(loc,4,gl.GL_FLOAT,False,stride,offset)
  108. # === main loop ===
  110. event=SDL_Event()
  111. while True:
  112. frame_startticks=SDL_GetTicks()
  114. while SDL_PollEvent(ctypes.byref(event))!=0:
  115. if event.type==SDL_QUIT:
  116. SDL_DestroyWindow(window)
  117. SDL_Quit()
  118. raise SystemExit("clean exit")
  120. gl.glClear(gl.GL_COLOR_BUFFER_BIT|gl.GL_DEPTH_BUFFER_BIT)
  122. gl.glDrawArrays(gl.GL_TRIANGLE_STRIP, 0, 4)
  124. SDL_GL_SwapWindow(window)
  126. frame_endticks=SDL_GetTicks()
  127. frame_spentticks=frame_endticks-frame_startticks
  128. if(frame_spentticks<frame_minticks):
  129. SDL_Delay(frame_minticks-frame_spentticks)
  130.  
Powered by spacepaste. Paste removal.