116 lines
2.5 KiB
Python
116 lines
2.5 KiB
Python
from typing import Optional
|
|
|
|
import tcod
|
|
|
|
from actions import BaseAction, QuitAction, BumpAction, WaitAction
|
|
from engine import Engine
|
|
|
|
#input options
|
|
MOVE_KEYS = {
|
|
#arrow keys
|
|
tcod.event.KeySym.UP: (0, -1),
|
|
tcod.event.KeySym.DOWN: (0, 1),
|
|
tcod.event.KeySym.LEFT: (-1, 0),
|
|
tcod.event.KeySym.RIGHT: (1, 0),
|
|
|
|
tcod.event.KeySym.HOME: (-1, -1),
|
|
tcod.event.KeySym.END: (-1, 1),
|
|
tcod.event.KeySym.PAGEUP: (1, -1),
|
|
tcod.event.KeySym.PAGEDOWN: (1, 1),
|
|
|
|
#numpad keys
|
|
tcod.event.KeySym.KP_1: (-1, 1),
|
|
tcod.event.KeySym.KP_2: (0, 1),
|
|
tcod.event.KeySym.KP_3: (1, 1),
|
|
tcod.event.KeySym.KP_4: (-1, 0),
|
|
|
|
tcod.event.KeySym.KP_6: (1, 0),
|
|
tcod.event.KeySym.KP_7: (-1, -1),
|
|
tcod.event.KeySym.KP_8: (0, -1),
|
|
tcod.event.KeySym.KP_9: (1, -1),
|
|
|
|
#vi key mapping
|
|
tcod.event.KeySym.h: (-1, 0),
|
|
tcod.event.KeySym.j: (0, 1),
|
|
tcod.event.KeySym.k: (0, -1),
|
|
tcod.event.KeySym.l: (1, 0),
|
|
|
|
tcod.event.KeySym.y: (-1, -1),
|
|
tcod.event.KeySym.u: (1, -1),
|
|
tcod.event.KeySym.b: (-1, 1),
|
|
tcod.event.KeySym.n: (1, 1),
|
|
}
|
|
|
|
WAIT_KEYS = {
|
|
tcod.event.KeySym.PERIOD,
|
|
tcod.event.KeySym.KP_5,
|
|
tcod.event.KeySym.CLEAR,
|
|
}
|
|
|
|
#event handler is one part of the engine
|
|
class EventHandler(tcod.event.EventDispatch[BaseAction]):
|
|
def __init__(self, engine: Engine):
|
|
super().__init__()
|
|
self.engine = engine
|
|
|
|
#callbacks
|
|
def ev_quit(self, event: tcod.event.Quit) -> Optional[BaseAction]:
|
|
return QuitAction()
|
|
|
|
def handle_events(self) -> bool:
|
|
raise NotImplementedError()
|
|
|
|
|
|
class InGameEventHandler(EventHandler):
|
|
def handle_events(self) -> bool:
|
|
result = False
|
|
|
|
for event in tcod.event.wait():
|
|
action = self.dispatch(event)
|
|
|
|
if action is None:
|
|
continue
|
|
|
|
result |= action.apply() #entity references the engine
|
|
|
|
return result
|
|
|
|
def ev_keydown(self, event: tcod.event.KeyDown) -> Optional[BaseAction]:
|
|
key = event.sym #SDL stuff, neat.
|
|
|
|
player = self.engine.player
|
|
|
|
#player input
|
|
if key == tcod.event.KeySym.ESCAPE:
|
|
return QuitAction()
|
|
|
|
if key in MOVE_KEYS:
|
|
xdir, ydir = MOVE_KEYS[key]
|
|
return BumpAction(player, xdir = xdir, ydir = ydir)
|
|
|
|
if key in WAIT_KEYS:
|
|
return WaitAction(player)
|
|
|
|
|
|
class GameOverEventHandler(EventHandler):
|
|
def handle_events(self) -> bool:
|
|
result = False
|
|
|
|
for event in tcod.event.wait():
|
|
action = self.dispatch(event)
|
|
|
|
if action is None:
|
|
continue
|
|
|
|
result |= action.apply()
|
|
|
|
return result
|
|
|
|
def ev_keydown(self, event: tcod.event.KeyDown) -> Optional[BaseAction]:
|
|
key = event.sym #SDL stuff, neat.
|
|
|
|
#player input
|
|
if key == tcod.event.KeySym.ESCAPE:
|
|
return QuitAction()
|
|
|
|
return None |