userspace.py

# ------------------------------------------------------------------
# userspace.py
# 
# Things related to user code execution
# ------------------------------------------------------------------

import time
import vec2d
import pygame
import keymap
import threading

# a 'wait()' function for user code - sleep in blocks to allow
# cancelling
wait_block = 1
def wait(t):
    loops = t // wait_block
    extra = t - float(loops * wait_block)

    while loops > 0:
        loops -= 1
        time.sleep(wait_block)
    if (extra):
        time.sleep(extra)

# an 'output()' function for user code - print to the GUI.
# TODO: Make this actually print to the GUI
def output(s):
    print s

# the space visible to user code
space = dict(
        Vec2d = vec2d.Vec2d,
        wait = wait,
        output = output,
        )

def run(code):
    exec(code, space, space)

# reset keybindings to defaults
def reset_keybindings():
    space['keybindings'] = {}

# run action associated with a keybinding
def do_key(keycode):
    if 'keybindings' not in space or type(space['keybindings']) is not dict:
        reset_keybindings()

    code = space['keybindings'].get(keymap.key_to_str[keycode])
    if code:
        from gui import Runner
        runner = Runner("keypress", code)
        runner.start()