ui.py

from pygame.locals import *
from constants import mainconfig
import pad, pygame

(PASS, CANCEL, UP, DOWN, LEFT, RIGHT, START, SELECT, CONFIRM,
 PGUP, PGDN, FULLSCREEN, SORT, CLEAR) = range(14)

QUIT = CANCEL
SCREENSHOT = SORT

non_player = [CANCEL, PASS]

# these defaults are INSANE and make the UI for the game far too complicated
# pad_defaults = {
  # pad.QUIT: CANCEL,
  # pad.UP: UP,
  # pad.DOWN: DOWN,
  # pad.LEFT: LEFT,
  # pad.RIGHT: RIGHT,
  # pad.START: START,
  # pad.UPRIGHT: CONFIRM,
  # pad.DOWNRIGHT: PGDN,
  # pad.UPLEFT: CANCEL,
  # pad.DOWNLEFT: PGUP,
  # pad.SELECT: SELECT,
  # pad.SCREENSHOT: SCREENSHOT,
  # }
# setup saner defaults
pad_defaults = {
  pad.UP: UP,
  pad.DOWN: DOWN,
  pad.LEFT: LEFT,
  pad.RIGHT: RIGHT,
  pad.START: CONFIRM,
  pad.SELECT: CANCEL
}

if mainconfig["centerconfirm"]: pad_defaults[pad.CENTER] = CONFIRM

key_defaults = {
  100 * K_f: FULLSCREEN,
  100 * K_BACKSPACE: SORT,
  100 * K_TAB: SELECT,
  100 * K_s: SORT,
  100 * K_INSERT: CANCEL,
  100 * K_F11: FULLSCREEN,
  100 * K_RETURN: CONFIRM,
  100 * K_PAGEDOWN: PGDN,
  100 * K_PAGEUP: PGUP,
  }

# This class wraps pad and maps pad events to UI events.
class UI(object):
  def __init__(self, handler):
    self.handler = handler
    self.events = {}
    self.states = {}
    self.last_press = ((None, None), None)
    self.last_repeat = 0
    self.merge_events(-2, key_defaults)
    self.merge_events(0, pad_defaults)
    self.merge_events(1, pad_defaults)

    # So we don't crash adjusting nonsensical states
    for i in non_player: self.add_event(i, -1, i)

  def add_event(self, key, pid, event):
    self.events[key] = event
    self.states[(pid, event)] = False

  def merge_events(self, pid, events):
    for key, event in events.items():
      self.add_event(key, pid, event)

  def poll(self):
    pid, ev = self.handler.poll()

    nev = self.events.get(abs(ev), PASS)

    if ev < 0:
      self.states[(pid, nev)] = False
      nev = -nev
    elif ev != PASS:
      self.states[(pid, nev)] = True
      self.last_press = ((pid, nev), pygame.time.get_ticks())
    elif (nev == PASS and self.last_press[0][1] and
          self.last_press[1] + 500 < pygame.time.get_ticks() and
          self.last_repeat + 30 < pygame.time.get_ticks() and
          self.states[self.last_press[0]] == True):
      self.last_repeat = pygame.time.get_ticks()
      return self.last_press[0]

    return (pid, nev)

  def forget_repeat(self):
    self.last_press = ((None, None), None)

  def wait(self, delay = 20):
    ev = (-1, PASS)
    while ev[1] == PASS:
      ev = self.poll()
      pygame.time.wait(delay)
    return ev

  def empty(self):
    ev = (0, QUIT)
    while ev[1] != PASS: ev = self.poll()

  def clear(self):
    self.empty()
    for k in self.states: self.states[k] = False

ui = UI(pad.pad)