97 lines
3.8 KiB
Python
97 lines
3.8 KiB
Python
from Lights.Lights import Light
|
|
from OpenGL.GL import *
|
|
from MatrixStuff.Transformations import *
|
|
|
|
class Spotlight(Light):
|
|
programId = {}
|
|
depthshaderId = -1
|
|
|
|
@classmethod
|
|
def getDepthProgram(cls, vertexshader=-1, geometryshader=-1):
|
|
if ((
|
|
vertexshader, geometryshader) not in cls.programId.keys() and vertexshader != -1 and geometryshader != -1):
|
|
fragment_shader_id = -1
|
|
if cls.depthshaderId == -1:
|
|
with open('./Lights/depthfragment.glsl', 'r') as f:
|
|
fragment_shader_string = f.read()
|
|
fragment_shader_id = glCreateShader(GL_FRAGMENT_SHADER)
|
|
glShaderSource(fragment_shader_id, fragment_shader_string)
|
|
glCompileShader(fragment_shader_id)
|
|
if glGetShaderiv(fragment_shader_id, GL_COMPILE_STATUS) != GL_TRUE:
|
|
raise RuntimeError(glGetShaderInfoLog(fragment_shader_id))
|
|
|
|
program_id = glCreateProgram()
|
|
glAttachShader(program_id, vertexshader)
|
|
glAttachShader(program_id, geometryshader)
|
|
glAttachShader(program_id, fragment_shader_id)
|
|
glLinkProgram(program_id)
|
|
cls.programId[(vertexshader, geometryshader)] = program_id
|
|
return program_id
|
|
else:
|
|
if (vertexshader, geometryshader) not in cls.programId.keys():
|
|
return -1
|
|
return cls.programId[(vertexshader, geometryshader)]
|
|
|
|
def __init__(self, target=[0, 0, 0], opening=45):
|
|
super(Spotlight, self).__init__()
|
|
self.target = target
|
|
self.opening = opening
|
|
|
|
@property
|
|
def lightColor(self):
|
|
return self._lightColor
|
|
|
|
@lightColor.setter
|
|
def lightColor(self, value):
|
|
self._lightColor = value
|
|
|
|
@property
|
|
def ModelviewProjectionMatrix(self):
|
|
return self._ModelviewProjectionMatrix
|
|
|
|
@ModelviewProjectionMatrix.setter
|
|
def ModelviewProjectionMatrix(self, value):
|
|
self._ModelviewProjectionMatrix = np.matrix(value)
|
|
|
|
@property
|
|
def pos(self):
|
|
return self._pos
|
|
|
|
@pos.setter
|
|
def pos(self, value):
|
|
self._pos = value
|
|
|
|
def prepareForDepthMapping(self):
|
|
new = False
|
|
if self.FramebufferId == -1:
|
|
self.FramebufferId = glGenFramebuffers(1)
|
|
new = True
|
|
glClearColor(1.0, 1.0, 1.0, 1.0)
|
|
glBindFramebuffer(GL_FRAMEBUFFER, self.FramebufferId)
|
|
glCullFace(GL_FRONT)
|
|
glViewport(0, 0, self.map_size, self.map_size)
|
|
if new:
|
|
if self.DepthBuffer == -1:
|
|
self.DepthBuffer = glGenTextures(1)
|
|
glBindTexture(GL_TEXTURE_2D, self.DepthBuffer)
|
|
glTexImage2D(GL_TEXTURE_2D, 0, GL_DEPTH_COMPONENT, self.map_size, self.map_size, 0, GL_DEPTH_COMPONENT, GL_FLOAT, None)
|
|
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_NEAREST)
|
|
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_NEAREST)
|
|
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_NEAREST)
|
|
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_CLAMP_TO_BORDER)
|
|
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_CLAMP_TO_BORDER)
|
|
glTexParameterfv(GL_TEXTURE_2D, GL_TEXTURE_BORDER_COLOR, np.array([0, 0, 0], dtype=np.float32))
|
|
|
|
glFramebufferTexture2D(GL_FRAMEBUFFER, GL_DEPTH_ATTACHMENT, GL_TEXTURE_2D, self.FramebufferId, 0)
|
|
|
|
DrawBuffers = [GL_NONE]
|
|
glDrawBuffers(DrawBuffers)
|
|
if (glCheckFramebufferStatus(GL_FRAMEBUFFER) != GL_FRAMEBUFFER_COMPLETE):
|
|
return False
|
|
|
|
def finishDepthMapping(self):
|
|
glCullFace(GL_BACK)
|
|
DrawBuffers = [GL_COLOR_ATTACHMENT0]
|
|
glDrawBuffers(DrawBuffers)
|
|
glClearColor(0.0, 0.0, 0.0, 1.0)
|
|
glBindFramebuffer(GL_FRAMEBUFFER, 0) |