main.py

#!/usr/bin/python2
import collections
import sys
import pygame
import pygame.locals
import cairo
import numpy
import math
import Image

import graphics

GAME_NAME = 'A Hobo\'s Quest'

class Window(object):
    def __init__(self, name = GAME_NAME, size=(800,600)):
        # initialize pygame
        pygame.init()
        self._size = size
        self._window = pygame.display.set_mode(size)
        pygame.display.set_caption(name)

        # data to talk between pygame and cairo
        self._data = numpy.empty(size[0] * size[1] * 4, dtype=numpy.int8)

        # initialize cairo
        self._surface = cairo.ImageSurface.create_for_data(self._data, cairo.FORMAT_ARGB32, size[0], size[1], size[0] * 4)
        

    def run(self, callback):
        # test rendering

        while True:
            ctx = cairo.Context(self._surface)
            ctx.set_source_rgb(0, 0, 0)
            ctx.paint()
            if not callback(ctx, self._size):
                pygame.quit()
                return
            img = Image.frombuffer('RGBA', (self._surface.get_width(),
                                            self._surface.get_height()),
                                   self._surface.get_data(), 'raw', 'BGRA', 0, 1)
            img = img.tostring('raw', 'RGBA', 0, 1)
            psurf = pygame.image.frombuffer(img, self._size, 'RGBA')
            self._window.blit(psurf, (0, 0))
            pygame.display.flip()

if __name__ == '__main__':
    from stateframe import FrameUpdate, stack
    from menuframe import MenuFrame
    from boardframe import BoardFrame

    def InitGame():
        def SetCaffeine(vis):
            graphics.CAFFEINE = vis
        submenu_list = collections.OrderedDict()
        submenu_list['CAFFEINE'] = lambda: SetCaffeine(True)
        submenu_list['Sane'] = lambda: SetCaffeine(False)
        
        main_menu_list = {'Start Game':(lambda : BoardFrame()), 'Menu Skin': submenu_list, 'Exit': (lambda : exit(0))}
        mainMenu = MenuFrame(main_menu_list, GAME_NAME, fontSizeTitle=50, fontSize=30, displayItems = 4)
        stack.append(mainMenu)

    
    sprite = graphics.Sprite('test')
    win = Window()
    InitGame()        #Places the main menu in the stack
    win.run(FrameUpdate)