def set_aspect(self, aspect):
if not aspect:
aspect = self.win_width / self.win_height
self.aspect = aspect
if self.cache_win != (self.win_width, self.win_height):
self.aspect_cache = {}
self.cache_win = (self.win_width, self.win_height)
if aspect in self.aspect_cache:
self.viewmatrix = self.aspect_cache[aspect]
return
display_aspect = self.win_width / self.win_height
if display_aspect > self.aspect:
self.width = int(round(self.aspect * self.win_height))
self.height = self.win_height
else:
self.width = self.win_width
self.height = int(round(self.win_width / self.aspect))
off_x = int((self.win_width - self.width) / 2)
off_y = int((self.win_height - self.height) / 2)
matrix = Matrix()
matrix.translate(-1.0, -1.0)
matrix.scale(2.0/self.win_width, 2.0/self.win_height)
matrix.translate(off_x, off_y, 0)
matrix.scale(self.width, self.width, 1)
self.viewmatrix = matrix
self.aspect_cache[self.aspect] = matrix
def set_aspect(self, aspect):
if not aspect:
aspect = self.win_width / self.win_height
self.aspect = aspect
if self.cache_win != (self.win_width, self.win_height):
self.aspect_cache = {}
self.cache_win = (self.win_width, self.win_height)
if aspect in self.aspect_cache:
self.viewmatrix, self.width, self.height = self.aspect_cache[aspect]
return
display_aspect = self.win_width / self.win_height
if display_aspect > self.aspect:
self.width = int(round(self.aspect * self.win_height))
self.height = self.win_height
else:
self.width = self.win_width
self.height = int(round(self.win_width / self.aspect))
off_x = int((self.win_width - self.width) / 2)
off_y = int((self.win_height - self.height) / 2)
matrix = Matrix()
matrix.translate(-1.0, -1.0)
matrix.scale(2.0/self.win_width, 2.0/self.win_height)
matrix.translate(off_x, off_y, 0)
matrix.scale(self.width, self.width, 1)
self.viewmatrix = matrix
self.aspect_cache[self.aspect] = matrix, self.width, self.height
class TextureRenderer(SolidRenderer):
UNIFORMS = ["tex", "color", "transform", "textransform"]
VS = vs_texture
FS = fs_texture
def __init__(self, display):
SolidRenderer.__init__(self, display)
self.tmatrix = Matrix()
def setup(self):
SolidRenderer.setup(self)
gl.glUniform1i(self.l_tex, 0)
def draw(self, pos, size, tpos, tsize, color):
self.activate()
self.tmatrix.reset()
self.tmatrix.translate(*tpos)
self.tmatrix.scale(*tsize)
gl.glUniformMatrix4fv(self.l_textransform, 1, False,
self.tmatrix.m.transpose())
SolidRenderer.draw(self, pos, size, color)
class TextureRenderer(SolidRenderer):
UNIFORMS = [
"tex", "color", "transform", "textransform"
]
VS = vs_texture
FS = fs_texture
def __init__(self, display):
SolidRenderer.__init__(self, display)
self.tmatrix = Matrix()
def setup(self):
SolidRenderer.setup(self)
gl.glUniform1i(self.l_tex, 0)
def draw(self, pos, size, tpos, tsize, color):
self.activate()
self.tmatrix.reset()
self.tmatrix.translate(*tpos)
self.tmatrix.scale(*tsize)
gl.glUniformMatrix4fv(self.l_textransform, 1, False, self.tmatrix.m)
SolidRenderer.draw(self, pos, size, color)
class BaseDisplay(object):
BLACK = (0.0, 0.0, 0.0, 1.0)
TRANSPARENT = (0.0, 0.0, 0.0, 0.0)
def __init__(self, width=640, height=480, fullscreen=False, aspect=None):
self.kbd_handler = None
self.win_width = self.width = width
self.win_height = self.height = height
self.matrix = Matrix()
self.viewmatrix = Matrix()
self.set_aspect(aspect)
self.clear_color = self.BLACK
self.cleanup = None
self.fullscreen = fullscreen
def toggle_fullscreen(self):
pass
def set_aspect(self, aspect):
if aspect is None:
aspect = self.win_width / self.win_height
self.aspect = aspect
display_aspect = self.win_width / self.win_height
if self.aspect:
if display_aspect > self.aspect:
self.width = int(round(self.aspect * self.win_height))
self.height = self.win_height
else:
self.width = self.win_width
self.height = int(round(self.win_width / self.aspect))
else:
self.width = self.win_width
self.height = self.win_height
off_x = int((self.win_width - self.width) / 2)
off_y = int((self.win_height - self.height) / 2)
self.viewmatrix.reset()
self.viewmatrix.translate(-1.0, -1.0)
self.viewmatrix.scale(2.0 / self.win_width, 2.0 / self.win_height)
self.viewmatrix.translate(off_x, off_y, 0)
self.viewmatrix.scale(self.width, self.width, 1)
def commit_matrix(self, uniform):
m = self.viewmatrix.m * self.matrix.m
self.gl.glUniformMatrix4fv(uniform, 1, False, m.transpose())
def set_render_gen(self, gen):
self.frames = gen()
def set_keyboard_handler(self, f):
self.kbd_handler = f
def _initialize(self):
# Set up common GL state
self.gl.glActiveTexture(self.gl.GL_TEXTURE0)
self.gl.glBlendFunc(self.gl.GL_ONE, self.gl.GL_ONE_MINUS_SRC_ALPHA)
self.gl.glEnable(self.gl.GL_BLEND)
def _render(self):
self.matrix.reset()
self.gl.glClearColor(*self.clear_color)
self.gl.glClear(self.gl.GL_COLOR_BUFFER_BIT
| self.gl.GL_DEPTH_BUFFER_BIT)
next(self.frames)
graphics.get_renderer().cleanup()
def main_loop(self):
while True:
try:
self.st = self.lt = time.time()
self._render()
except StopIteration:
break
self.swap_buffers()
def log(self, msg):
t = time.time()
print("[%5.02f %5.02f] %s" % ((t - self.st) * 1000,
(t - self.lt) * 1000, msg))
self.lt = t
def round_coord(self, c):
return int(round(c * self.width)) / self.width
@property
def top(self):
return self.height / self.width