punyverse/punyverse/loader.py

from __future__ import print_function

import os
import sys
import time

import pyglet
from pyglet.gl import *
from six.moves import zip_longest

from punyverse.world import World


class glContext(object):
    def __init__(self, context):
        self.new_context = context

    def __enter__(self):
        self.old_context = get_current_context()
        self.new_context.set_current()

    def __exit__(self, exc_type, exc_val, exc_tb):
        self.old_context.set_current()


class glSection(object):
    def __init__(self, type):
        self.type = type

    def __enter__(self):
        glBegin(self.type)

    def __exit__(self, exc_type, exc_val, exc_tb):
        glEnd()


def progress_bar(x, y, width, height, filled):
    glPushAttrib(GL_ENABLE_BIT)
    glDisable(GL_TEXTURE_2D)
    glDisable(GL_BLEND)
    x1 = x
    x2 = x + width
    y1 = y
    y2 = y - height
    y3 = 0.65 * y1 + 0.35 * y2
    y4 = 0.25 * y1 + 0.75 * y2

    glColor3f(0.6, 0.6, 0.6)
    with glSection(GL_LINE_LOOP):
        glVertex2f(x1, y1)
        glVertex2f(x1, y2)
        glVertex2f(x2, y2)
        glVertex2f(x2, y1)

    x1 += 1
    y1 -= 1
    x2 = x + width * filled - 1

    with glSection(GL_TRIANGLE_STRIP):
        glColor3f(0.81, 1, 0.82)
        glVertex2f(x1, y1)
        glVertex2f(x2, y1)
        glColor3f(0, 0.83, 0.16)
        glVertex2f(x1, y3)
        glVertex2f(x2, y3)
        glVertex2f(x1, y4)
        glVertex2f(x2, y4)
        glColor3f(0.37, 0.92, 0.43)
        glVertex2f(x1, y2)
        glVertex2f(x2, y2)
    glPopAttrib()


def get_context_info(context):
    info = ['  %-22s %s' % (key + ':', value)
            for key, value in context.config.get_gl_attributes()]
    info = ['%-30s %-30s' % group for group in
            zip_longest(info[::2], info[1::2], fillvalue='')]

    with glContext(context):
        return '\n'.join([
            'Graphics Vendor:   ' + gl_info.get_vendor(),
            'Graphics Version:  ' + gl_info.get_version(),
            'Graphics Renderer: ' + gl_info.get_renderer(),
        ]) + '\n\n' + 'OpenGL configuration:\n' + '\n'.join(info)


class LoaderWindow(pyglet.window.Window):
    MONOSPACE = ('Consolas', 'Droid Sans Mono', 'Courier', 'Courier New', 'Dejavu Sans Mono')

    def __init__(self, *args, **kwargs):
        super(LoaderWindow, self).__init__(*args, **kwargs)

        # work around pyglet bug: decoding font names as utf-8 instead of mbcs when using EnumFontsA.
        stderr = sys.stderr
        sys.stderr = open(os.devnull, 'w')
        pyglet.font.have_font(self.MONOSPACE[0])
        sys.stderr = stderr

        self.loading_phase = pyglet.text.Label(
            font_name=self.MONOSPACE, font_size=20, x=10, y=self.height - 50,
            color=(255, 255, 255, 255), width=self.width - 20, align='center',
            multiline=True, text='Punyverse is starting...'
        )
        self.loading_label = pyglet.text.Label(
            font_name=self.MONOSPACE, font_size=16, x=10, y=self.height - 120,
            color=(255, 255, 255, 255), width=self.width - 20, align='center',
            multiline=True
        )
        self.info_label = pyglet.text.Label(
            font_name=self.MONOSPACE, font_size=13, x=10, y=self.height - 220,
            color=(255, 255, 255, 255), width=self.width - 20,
            multiline=True
        )
        self.progress = 0

        self._main_context = None

    def set_main_context(self, context):
        self._main_context = context
        self.info_label.text = get_context_info(context)
        print(self.info_label.text)

    def _load_callback(self, phase, message, progress):
        print(message)
        with glContext(self.context):
            self.loading_phase.text = phase
            self.loading_label.text = message
            self.progress = progress

            self.on_draw()
            self.flip()
            self.dispatch_events()

    def load(self):
        start = time.clock()
        with glContext(self._main_context):
            world = World('world.json', self._load_callback)
        print('Loaded in %s seconds.' % (time.clock() - start))
        return world

    def on_draw(self):
        glClear(GL_COLOR_BUFFER_BIT)
        glLoadIdentity()
        self.loading_phase.draw()
        self.loading_label.draw()
        progress_bar(10, self.height - 140, self.width - 20, 50, self.progress)
        self.info_label.draw()

    def main_is_initializing(self):
        self._load_callback('Loading main window...', '', 0)


class LoaderConsole(object):
    def __init__(self):
        from ctypes import windll
        self._own_console = False
        if windll.kernel32.AllocConsole():
            self._own_console = True
            self._output = open('CONOUT$', 'w')
        else:
            self._output = sys.stdout
        self._main_context = None

    def _load_callback(self, phase, message, progress):
        print(message, file=self._output)

    def load(self):
        start = time.clock()
        with glContext(self._main_context):
            world = World('world.json', self._load_callback)
        print('Loaded in %s seconds.' % (time.clock() - start), file=self._output)
        return world

    def set_main_context(self, context):
        self._main_context = context
        print(get_context_info(context), file=self._output)
        print('=' * 79, file=self._output)
        print("You cannot see the normal loading screen because you are using Intel integrated graphics.",
              file=self._output)
        print('Please attempt to set python to execute with your dedicated graphics, if available.', file=self._output)
        print('=' * 79, file=self._output)

    def close(self):
        if self._own_console:
            self._output.close()
            from ctypes import windll
            windll.kernel32.FreeConsole()