def create(self, width, height):
self.width = width
self.height = height
self.depthbuffer = CubeDepthbuffer()
self.framebuffer = Framebuffer()
self.depthbuffer.create(width, height)
self.framebuffer.create(self.depthbuffer)
def testFramebuffer(self):
self.framebuffer = Framebuffer()
self.framebuffer.bind()
depthbuffer = Depthbuffer()
depthbuffer.create(100, 100)
depthbuffer.attach()
self.framebuffer.checkComplete()
self.assertEqual(self.framebuffer.getId(), 1)
def create(self, width, height):
self.width = width
self.height = height
self.depthbuffer = Depthbuffer()
self.framebuffer = Framebuffer()
self.framebuffer.bind()
self.depthbuffer.create(width, height)
self.depthbuffer.attach()
self.framebuffer.checkComplete()
class Shadow:
def __init__(self, nearPlane, farPlane):
self.nearPlane = nearPlane
self.farPlane = farPlane
def create(self, width, height):
self.width = width
self.height = height
self.depthbuffer = CubeDepthbuffer()
self.framebuffer = Framebuffer()
self.depthbuffer.create(width, height)
self.framebuffer.create(self.depthbuffer)
def castShadow(self, depthProgram, lightPos):
self.lightPos = lightPos
depthProgram.use()
depthProgram.setFloat('farPlane', self.farPlane)
depthProgram.setVec3('lightPos', self.lightPos)
transforms = self.__getTransforms()
for i in range(6):
depthProgram.setMat4(f'shadowMatrices[{i}]', transforms[i])
glViewport(0, 0, self.width, self.height)
self.framebuffer.bind()
glClear(GL_DEPTH_BUFFER_BIT)
def endCastShadow(self, program):
self.framebuffer.unbind()
program.use()
program.setFloat('farPlane', self.farPlane)
program.setInt('depthMap', 0)
glActiveTexture(GL_TEXTURE0)
self.depthbuffer.bind()
def __getTransforms(self):
st = []
pos = self.lightPos
proj = glm.perspective(glm.radians(90), 1, self.nearPlane, self.farPlane)
st.append(proj * glm.lookAt(pos, pos + glm.vec3(1, 0, 0), glm.vec3(0, -1, 0)))
st.append(proj * glm.lookAt(pos, pos + glm.vec3(-1, 0, 0), glm.vec3(0, -1, 0)))
st.append(proj * glm.lookAt(pos, pos + glm.vec3(0, 1, 0), glm.vec3(0, 0, 1)))
st.append(proj * glm.lookAt(pos, pos + glm.vec3(0, -1, 0), glm.vec3(0, 0, -1)))
st.append(proj * glm.lookAt(pos, pos + glm.vec3(0, 0, 1), glm.vec3(0, -1, 0)))
st.append(proj * glm.lookAt(pos, pos + glm.vec3(0, 0, -1), glm.vec3(0, -1, 0)))
return st
def delete(self):
self.framebuffer.delete()
self.depthbuffer.delete()
class Shadow:
def __init__(self, lightPos, nearPlane, farPlane):
self.updateMatrix(lightPos, nearPlane, farPlane)
def updateMatrix(self, lightPos, nearPlane, farPlane):
lightProjection = glm.ortho(-10, 10, -10, 10, nearPlane, farPlane)
lightView = glm.lookAt(lightPos, glm.vec3(0), glm.vec3(0, 1, 0))
self.lightSpaceMatrix = lightProjection * lightView
def create(self, width, height):
self.width = width
self.height = height
self.depthbuffer = Depthbuffer()
self.framebuffer = Framebuffer()
self.framebuffer.bind()
self.depthbuffer.create(width, height)
self.depthbuffer.attach()
self.framebuffer.checkComplete()
def castShadow(self, depthProgram):
glDisable(GL_CULL_FACE)
depthProgram.use()
depthProgram.setMat4('lightSpaceMatrix', self.lightSpaceMatrix)
glViewport(0, 0, self.width, self.height)
self.framebuffer.bind()
glClear(GL_DEPTH_BUFFER_BIT)
def endCastShadow(self, program):
self.framebuffer.unbind()
program.use()
program.setMat4('lightSpaceMatrix', self.lightSpaceMatrix)
program.setInt('shadowMap', 10)
glActiveTexture(GL_TEXTURE10)
self.depthbuffer.bind()
glEnable(GL_CULL_FACE)
def delete(self):
self.framebuffer.delete()
self.depthbuffer.delete()
def __createMultisampleFBO(self):
self.__hdrFBO_MS = Framebuffer()
self.__hdrFBO_MS.bind()
self.__colorBuffersMS = glGenTextures(2)
for i in range(2):
glBindTexture(GL_TEXTURE_2D_MULTISAMPLE, self.__colorBuffersMS[i])
glTexImage2DMultisample(GL_TEXTURE_2D_MULTISAMPLE,
config['sampling_level'], GL_RGB16F,
self.width, self.height, GL_TRUE)
glFramebufferTexture2D(GL_FRAMEBUFFER, GL_COLOR_ATTACHMENT0 + i,
GL_TEXTURE_2D_MULTISAMPLE,
self.__colorBuffersMS[i], 0)
self.__rboDepthMS = glGenRenderbuffers(1)
glBindRenderbuffer(GL_RENDERBUFFER, self.__rboDepthMS)
glRenderbufferStorageMultisample(GL_RENDERBUFFER,
config['sampling_level'],
GL_DEPTH_COMPONENT, self.width,
self.height)
glFramebufferRenderbuffer(GL_FRAMEBUFFER, GL_DEPTH_ATTACHMENT,
GL_RENDERBUFFER, self.__rboDepthMS)
glDrawBuffers(2, [GL_COLOR_ATTACHMENT0, GL_COLOR_ATTACHMENT1])
self.__hdrFBO_MS.checkComplete()
def __createFBO(self):
self.hdrFBO = Framebuffer()
self.hdrFBO.bind()
self.colorBuffers = glGenTextures(2)
for i in range(2):
glBindTexture(GL_TEXTURE_2D, self.colorBuffers[i])
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_CLAMP_TO_EDGE)
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_CLAMP_TO_EDGE)
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR)
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR)
glTexImage2D(GL_TEXTURE_2D, 0, GL_RGB16F, self.width, self.height,
0, GL_RGB, GL_FLOAT, None)
glFramebufferTexture2D(GL_FRAMEBUFFER, GL_COLOR_ATTACHMENT0 + i,
GL_TEXTURE_2D, self.colorBuffers[i], 0)
self.rboDepth = glGenRenderbuffers(1)
glBindRenderbuffer(GL_RENDERBUFFER, self.rboDepth)
glRenderbufferStorage(GL_RENDERBUFFER, GL_DEPTH_COMPONENT, self.width,
self.height)
glFramebufferRenderbuffer(GL_FRAMEBUFFER, GL_DEPTH_ATTACHMENT,
GL_RENDERBUFFER, self.rboDepth)
glDrawBuffers(2, [GL_COLOR_ATTACHMENT0, GL_COLOR_ATTACHMENT1])
self.hdrFBO.checkComplete()
class FramebufferTest(unittest.TestCase):
@classmethod
def setUpClass(cls):
cls.window = helper.initAndGetWindow()
@classmethod
def tearDownClass(cls):
glfw.terminate()
def testFramebuffer(self):
self.framebuffer = Framebuffer()
self.framebuffer.bind()
depthbuffer = Depthbuffer()
depthbuffer.create(100, 100)
depthbuffer.attach()
self.framebuffer.checkComplete()
self.assertEqual(self.framebuffer.getId(), 1)
def testFramebuffer(self):
self.framebuffer = Framebuffer()
self.framebuffer.bind()
self.assertRaises(RuntimeError, self.framebuffer.checkComplete)
class HDRbuffer:
def create(self, width, height):
self.width = width
self.height = height
self.__createFBO()
self.__createMultisampleFBO()
def __createFBO(self):
self.hdrFBO = Framebuffer()
self.hdrFBO.bind()
self.colorBuffers = glGenTextures(2)
for i in range(2):
glBindTexture(GL_TEXTURE_2D, self.colorBuffers[i])
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_CLAMP_TO_EDGE)
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_CLAMP_TO_EDGE)
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR)
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR)
glTexImage2D(GL_TEXTURE_2D, 0, GL_RGB16F, self.width, self.height,
0, GL_RGB, GL_FLOAT, None)
glFramebufferTexture2D(GL_FRAMEBUFFER, GL_COLOR_ATTACHMENT0 + i,
GL_TEXTURE_2D, self.colorBuffers[i], 0)
self.rboDepth = glGenRenderbuffers(1)
glBindRenderbuffer(GL_RENDERBUFFER, self.rboDepth)
glRenderbufferStorage(GL_RENDERBUFFER, GL_DEPTH_COMPONENT, self.width,
self.height)
glFramebufferRenderbuffer(GL_FRAMEBUFFER, GL_DEPTH_ATTACHMENT,
GL_RENDERBUFFER, self.rboDepth)
glDrawBuffers(2, [GL_COLOR_ATTACHMENT0, GL_COLOR_ATTACHMENT1])
self.hdrFBO.checkComplete()
def __createMultisampleFBO(self):
self.__hdrFBO_MS = Framebuffer()
self.__hdrFBO_MS.bind()
self.__colorBuffersMS = glGenTextures(2)
for i in range(2):
glBindTexture(GL_TEXTURE_2D_MULTISAMPLE, self.__colorBuffersMS[i])
glTexImage2DMultisample(GL_TEXTURE_2D_MULTISAMPLE,
config['sampling_level'], GL_RGB16F,
self.width, self.height, GL_TRUE)
glFramebufferTexture2D(GL_FRAMEBUFFER, GL_COLOR_ATTACHMENT0 + i,
GL_TEXTURE_2D_MULTISAMPLE,
self.__colorBuffersMS[i], 0)
self.__rboDepthMS = glGenRenderbuffers(1)
glBindRenderbuffer(GL_RENDERBUFFER, self.__rboDepthMS)
glRenderbufferStorageMultisample(GL_RENDERBUFFER,
config['sampling_level'],
GL_DEPTH_COMPONENT, self.width,
self.height)
glFramebufferRenderbuffer(GL_FRAMEBUFFER, GL_DEPTH_ATTACHMENT,
GL_RENDERBUFFER, self.__rboDepthMS)
glDrawBuffers(2, [GL_COLOR_ATTACHMENT0, GL_COLOR_ATTACHMENT1])
self.__hdrFBO_MS.checkComplete()
def bind(self):
self.__hdrFBO_MS.bind()
def finalize(self):
glBindFramebuffer(GL_READ_FRAMEBUFFER, self.__hdrFBO_MS.getId())
glBindFramebuffer(GL_DRAW_FRAMEBUFFER, self.hdrFBO.getId())
for i in range(2):
glReadBuffer(GL_COLOR_ATTACHMENT0 + i)
glDrawBuffer(GL_COLOR_ATTACHMENT0 + i)
glBlitFramebuffer(0, 0, self.width, self.height, 0, 0, self.width,
self.height, GL_COLOR_BUFFER_BIT, GL_NEAREST)
glBindFramebuffer(GL_READ_FRAMEBUFFER, 0)
glBindFramebuffer(GL_DRAW_FRAMEBUFFER, 0)
glBindFramebuffer(GL_FRAMEBUFFER, 0)
def unbind(self):
self.__hdrFBO_MS.unbind()
def __del__(self):
self.delete()
def delete(self):
self.hdrFBO.delete()
self.__hdrFBO_MS.delete()
try:
glDeleteRenderbuffers(1, self.rboDepth)
glDeleteRenderbuffers(1, self.rboDepthMS)
glDeleteTextures(2, self.colorBuffers)
glDeleteTextures(2, self.__colorBuffersMS)
self.colorBuffers, self.__colorBuffersMS = 0, 0
self.rboDepth, self.rboDepthMS = 0, 0
except (NullFunctionError, TypeError):
pass
def testFramebuffer(self):
self.framebuffer = Framebuffer()
self.framebuffer.bind()
self.assertRaises(RuntimeError, self.framebuffer.checkComplete)