119 lines
4.3 KiB
Python
119 lines
4.3 KiB
Python
from wsgiref.validate import check_errors
|
|
|
|
from OpenGL.GL.ARB.vertex_array_object import glDeleteVertexArrays
|
|
from OpenGL.GL.framebufferobjects import glBindFramebuffer
|
|
from OpenGL.GLUT import *
|
|
import OpenGL.GLUT.freeglut
|
|
from OpenGL.GLU import *
|
|
from OpenGL.GL import *
|
|
|
|
import numpy as np
|
|
|
|
def check_error(message):
|
|
gl_error = glGetError()
|
|
if (gl_error != GL_NO_ERROR):
|
|
print("Error: " + message)
|
|
if (gluErrorString(gl_error)):
|
|
print(gluErrorString(gl_error))
|
|
else:
|
|
print(hex(gl_error))
|
|
return True
|
|
return False
|
|
|
|
class Structure:
|
|
def __init__(self):
|
|
self.Objects = {}
|
|
self.vais = {}
|
|
self.Matrix = np.identity(4,np.float32)
|
|
self.dirty = False
|
|
|
|
def addShape(self,program,shape):
|
|
if not program in self.Objects.keys():
|
|
self.Objects[program] = []
|
|
self.Objects[program].append(shape)
|
|
self.dirty = True
|
|
|
|
def removeShape(self,program,shape):
|
|
if program in self.Objects.keys():
|
|
self.Objects[program].remove(shape)
|
|
if len(self.Objects[program]) == 0:
|
|
self.Objects.pop(program)
|
|
self.dirty = True
|
|
|
|
def buildvertexArrays(self):
|
|
if self.dirty:
|
|
self.clearVertexArrays()
|
|
glEnableClientState(GL_VERTEX_ARRAY)
|
|
glEnableClientState(GL_TEXTURE_COORD_ARRAY)
|
|
glEnableClientState(GL_NORMAL_ARRAY)
|
|
glEnableClientState(GL_COLOR_ARRAY)
|
|
self.vais = {}
|
|
|
|
for key,objects in self.Objects.items():
|
|
tvai = GLuint(-1)
|
|
tpbi = -1
|
|
tcbi = -1
|
|
tsbi = -1
|
|
glGenVertexArrays(1, tvai)
|
|
glBindVertexArray(tvai)
|
|
|
|
vid = glGetAttribLocation(key, "in_position")
|
|
glEnableVertexAttribArray(vid)
|
|
|
|
tpbi = glGenBuffers(1)
|
|
glBindBuffer(GL_ARRAY_BUFFER, tpbi)
|
|
positions = []
|
|
for o in objects:
|
|
positions.append(o.pos[0])
|
|
positions.append(o.pos[1])
|
|
positions.append(o.pos[2])
|
|
glBufferData(GL_ARRAY_BUFFER, np.asarray(positions), GL_STATIC_DRAW)
|
|
glVertexAttribPointer(vid, 3, GL_FLOAT, GL_FALSE, 0, None)
|
|
check_error("Could not create position buffer")
|
|
|
|
colors = []
|
|
for o in objects:
|
|
colors.append(o.color[0])
|
|
colors.append(o.color[1])
|
|
colors.append(o.color[2])
|
|
tcbi = glGenBuffers(1)
|
|
glBindBuffer(GL_ARRAY_BUFFER, tcbi)
|
|
glBufferData(GL_ARRAY_BUFFER, np.asarray(colors), GL_STATIC_DRAW)
|
|
vc = glGetAttribLocation(key, "MyInColor")
|
|
if vc != -1:
|
|
glEnableVertexAttribArray(vc)
|
|
glVertexAttribPointer(vc, 3, GL_FLOAT, GL_FALSE, 0, None)
|
|
check_error("Could not create color buffer")
|
|
|
|
if hasattr(objects[0],'size'):
|
|
sizes = []
|
|
for o in objects:
|
|
sizes.append(o.size[0])
|
|
sizes.append(o.size[1])
|
|
sizes.append(o.size[2])
|
|
tsbi = glGenBuffers(1)
|
|
glBindBuffer(GL_ARRAY_BUFFER, tsbi)
|
|
glBufferData(GL_ARRAY_BUFFER, np.asarray(sizes), GL_STATIC_DRAW)
|
|
vs = glGetAttribLocation(key, "MyInSize")
|
|
if vs != -1:
|
|
glEnableVertexAttribArray(vs)
|
|
glVertexAttribPointer(vs, 3, GL_FLOAT, GL_FALSE, 0, None)
|
|
check_error("Could not create size buffer")
|
|
|
|
glBindVertexArray(0)
|
|
self.vais[key] = (tvai,tpbi,tcbi,tsbi)
|
|
self.dirty = False
|
|
|
|
def clearVertexArrays(self):
|
|
for key,(a,p,c,s) in self.vais.items():
|
|
if p != -1:
|
|
glDisableVertexAttribArray(p)
|
|
glDeleteBuffers(1,[p])
|
|
if c != -1:
|
|
glDisableVertexAttribArray(c)
|
|
glDeleteBuffers(1,[c])
|
|
if s != -1 and s != GLuint(-1):
|
|
glDisableVertexAttribArray(s)
|
|
glDeleteBuffers(1,[s])
|
|
glDeleteVertexArrays(1, a)
|
|
check_error("Could not destroy vertex array") |