main.py

# ------------------------------------------------------------------
# main.py
# 
# Initialise stuff, handle events, main loop
# ------------------------------------------------------------------

import pygame
import pygtk
import gtk
import threading
import os
import sys
import traceback

from vec2d import Vec2d
import shared
import userspace
import gui
import levelmanager
import objects

background_color = (0,0,0) #black
frame_rate = 30.0
savedata_filename = os.path.join('userdata', 'savefile')

shared.stop_game = False

initial_help_text = """
# 
# Welcome to thegame!
# 
# Select a level. If you're new to thegame, you might want to start with
# 'Basics 1'. Have fun!
# 
"""

def init():
    # set WM_CLASS under X
    os.environ['SDL_VIDEO_X11_WMCLASS'] = "thegame"
    gtk.gdk.set_program_class("thegame")

    # initialise gtk
    gtk.threads_init()
    shared.gtkwin = gtk.Window(gtk.WINDOW_TOPLEVEL)

    # window sizes/positions
    gtkscreen = shared.gtkwin.get_screen()
    dim = Vec2d(gtkscreen.get_width(), 
            gtkscreen.get_height())

    pygame_pos = dim * 0.01
    pygame_dim = Vec2d(800, 640)
    gtk_dim = Vec2d(640, 480)
    gtk_pos = dim - gtk_dim - Vec2d(70, 70)

    shared.gtkwin.move(int(gtk_pos.x), int(gtk_pos.y))
    shared.gtkwin.resize(int(gtk_dim.x), int(gtk_dim.y))

    shared.dim = pygame_dim
    userspace.space['window_dim'] = pygame_dim
    
    # initialise pygame
    pygame.init()
    global clock
    clock = pygame.time.Clock()
    os.environ['SDL_VIDEO_WINDOW_POS'] = "%d,%d" % (pygame_pos.x, pygame_pos.y)
    shared.canvas = pygame.display.set_mode((int(pygame_dim.x), 
        int(pygame_dim.y)))

    # set window titles
    shared.gtkwin.show()
    shared.gtkwin.set_title("thegame - code editor")
    pygame.display.set_caption("thegame", "thegame")

    # set keybindings
    userspace.reset_keybindings()

    # make the code editor gui
    shared.gui = gui.Gui()

    # load levels and restore saved level data (if exists)
    shared.levelmgr = levelmanager.LevelManager()

    if not os.path.exists('userdata'):
        os.makedirs('userdata')

    try:
        f = open(savedata_filename, 'r')
        shared.levelmgr.load_level_data(f)
    except IOError:
        pass

    # set initial help text
    shared.gui.help_page.set_text(initial_help_text)

    # start the game!
    shared.levelmgr.start()

def handle_events():
    # pygame events
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            return False

        shared.levelmgr.event(event)

        if event.type == pygame.KEYDOWN:
            # some temporary key binds
            if event.key == pygame.K_ESCAPE and shared.levelmgr.current_level != 0:
                shared.levelmgr.request_goto_level(0)
            elif event.key == pygame.K_F2:
                shared.levelmgr.request_next_level()

            userspace.do_key(event.key)

    return True

def loop():
    # run gtk main loop in separate thread
    class GtkLoop(threading.Thread):
        def run(self):
            gtk.main()
    global gtk_thread
    gtk_thread = GtkLoop()
    gtk_thread.start()

    # game loop
    while not shared.stop_game:
        elapsed = 1 / frame_rate

        # events
        if not handle_events():
            break

        # action!
        # TODO: fix elapsed time handling!
        shared.levelmgr.step(elapsed)
        map(lambda o: o.step(elapsed), objects.world)
        shared.gui.step(elapsed)

        # handle requests
        objects.handle_requests()
        shared.levelmgr.handle_requests()
        
        # draw pretty things
        shared.canvas.fill(background_color)
        map(lambda o: o.draw(), objects.world)
        pygame.display.update()

        # not so fast
        clock.tick(frame_rate)

def end():
    # stop the game and save level data
    shared.levelmgr.stop()

    f = open(savedata_filename, 'w')
    shared.levelmgr.save_level_data(f)

    # stop the gui
    shared.gui.quit()
    gtk.threads_enter()
    gtk.main_quit()
    gtk.threads_leave()
    gtk_thread.join()

    # stop pygame
    pygame.quit()

# play the game!
if __name__ == '__main__':
    try:
        init()
        loop()
    except Exception, e:
        tb = sys.exc_info()[2]
        traceback.print_exception(e.__class__, e, tb)
    end()