5 from plomrogue.parser import ArgError, Parser
6 from plomrogue.commands import cmd_PLAYER_ID, cmd_THING_HEALTH
7 from plomrogue.game import GameBase
8 from plomrogue.mapping import Map, MapGeometryHex, YX
9 from plomrogue.io import PlomSocket
10 from plomrogue.things import ThingBase
17 def y_cut(self, map_lines, center_y, view_height):
18 map_height = len(map_lines)
19 if map_height > view_height and center_y > view_height / 2:
20 if center_y > map_height - view_height / 2:
21 map_lines[:] = map_lines[map_height - view_height:]
23 start = center_y - int(view_height / 2) - 1
24 map_lines[:] = map_lines[start:start + view_height]
26 def x_cut(self, map_lines, center_x, view_width, map_width):
27 if map_width > view_width and center_x > view_width / 2:
28 if center_x > map_width - view_width / 2:
29 cut_start = map_width - view_width
32 cut_start = center_x - int(view_width / 2)
33 cut_end = cut_start + view_width
34 map_lines[:] = [line[cut_start:cut_end] for line in map_lines]
36 def format_to_view(self, map_cells, center, size):
38 def map_cells_to_lines(map_cells):
40 if self.start_indented:
41 map_view_chars += ['0']
44 for cell in map_cells:
46 map_view_chars += [cell, ' ']
48 map_view_chars += [cell[0], cell[1]]
51 map_view_chars += ['\n']
54 if y % 2 == int(not self.start_indented):
55 map_view_chars += ['0']
56 if y % 2 == int(not self.start_indented):
57 map_view_chars = map_view_chars[:-1]
58 map_view_chars = map_view_chars[:-1]
59 return ''.join(map_view_chars).split('\n')
61 map_lines = map_cells_to_lines(map_cells)
62 self.y_cut(map_lines, center[1].y, size.y)
63 map_width = self.size.x * 2 + 1
64 self.x_cut(map_lines, center[1].x * 2, size.x, map_width)
68 def cmd_LAST_PLAYER_TASK_RESULT(game, msg):
71 cmd_LAST_PLAYER_TASK_RESULT.argtypes = 'string'
74 def cmd_TURN_FINISHED(game, n):
75 """Do nothing. (This may be extended later.)"""
77 cmd_TURN_FINISHED.argtypes = 'int:nonneg'
80 def cmd_TURN(game, n):
81 """Set game.turn to n, empty game.things."""
85 game.pickable_items[:] = []
86 cmd_TURN.argtypes = 'int:nonneg'
89 def cmd_VISIBLE_MAP(game, size, indent_first_line):
90 game.new_map(size, indent_first_line)
91 cmd_VISIBLE_MAP.argtypes = 'yx_tuple:pos bool'
94 def cmd_VISIBLE_MAP_LINE(game, y, terrain_line):
95 game.map_.set_line(y, terrain_line)
96 cmd_VISIBLE_MAP_LINE.argtypes = 'int:nonneg string'
99 def cmd_GAME_STATE_COMPLETE(game):
100 game.tui.to_update['turn'] = True
101 game.tui.to_update['map'] = True
102 game.tui.to_update['inventory'] = True
105 def cmd_THING_TYPE(game, i, type_):
106 t = game.get_thing(i)
108 cmd_THING_TYPE.argtypes = 'int:nonneg string'
111 def cmd_THING_POS(game, i, yx):
112 t = game.get_thing(i)
113 t.position = YX(0,0), yx
114 cmd_THING_POS.argtypes = 'int:nonneg yx_tuple:nonneg'
117 def cmd_PLAYER_INVENTORY(game, ids):
118 game.player_inventory[:] = ids # TODO: test whether valid IDs
119 game.tui.to_update['inventory'] = True
120 cmd_PLAYER_INVENTORY.argtypes = 'seq:int:nonneg'
123 def cmd_PICKABLE_ITEMS(game, ids):
124 game.pickable_items[:] = ids
125 game.tui.to_update['pickable_items'] = True
126 cmd_PICKABLE_ITEMS.argtypes = 'seq:int:nonneg'
129 class Game(GameBase):
131 def __init__(self, *args, **kwargs):
132 super().__init__(*args, **kwargs)
133 self.map_ = ClientMap() # we need an empty default map cause we draw
134 self.offset = YX(0,0) # the map widget even before we get a real one
135 self.player_inventory = []
137 self.pickable_items = []
138 self.parser = Parser(self)
139 self.map_geometry = MapGeometryHex()
140 self.thing_type = ThingBase
141 self.commands = {'LAST_PLAYER_TASK_RESULT': cmd_LAST_PLAYER_TASK_RESULT,
142 'TURN_FINISHED': cmd_TURN_FINISHED,
144 'VISIBLE_MAP_LINE': cmd_VISIBLE_MAP_LINE,
145 'PLAYER_ID': cmd_PLAYER_ID,
146 'PLAYER_INVENTORY': cmd_PLAYER_INVENTORY,
147 'GAME_STATE_COMPLETE': cmd_GAME_STATE_COMPLETE,
148 'VISIBLE_MAP': cmd_VISIBLE_MAP,
149 'PICKABLE_ITEMS': cmd_PICKABLE_ITEMS,
150 'THING_TYPE': cmd_THING_TYPE,
151 'THING_HEALTH': cmd_THING_HEALTH,
152 'THING_POS': cmd_THING_POS}
157 def new_map(self, size, indent_first_line):
158 self.map_ = ClientMap(size, start_indented=indent_first_line)
162 return self.get_thing(self.player_id)
164 def get_command(self, command_name):
165 from functools import partial
166 if command_name in self.commands:
167 f = partial(self.commands[command_name], self)
168 if hasattr(self.commands[command_name], 'argtypes'):
169 f.argtypes = self.commands[command_name].argtypes
173 def get_string_options(self, string_option_type):
176 def handle_input(self, msg):
182 command, args = self.parser.parse(msg)
184 self.log('UNHANDLED INPUT: ' + msg)
187 except ArgError as e:
188 self.log('ARGUMENT ERROR: ' + msg + '\n' + str(e))
191 """Prefix msg plus newline to self.log_text."""
192 self.log_text = msg + '\n' + self.log_text
193 self.tui.to_update['log'] = True
195 def symbol_for_type(self, type_):
199 elif type_ == 'monster':
201 elif type_ == 'food':
206 ASCII_printable = ' !"#$%&\'\(\)*+,-./0123456789:;<=>?@ABCDEFGHIJKLMNOPQRSTUVWX'\
207 'YZ[\\]^_\`abcdefghijklmnopqrstuvwxyz{|}~'
210 def recv_loop(plom_socket, game, q):
211 for msg in plom_socket.recv():
217 def __init__(self, tui, start, size, check_updates=[], visible=True):
218 self.check_updates = check_updates
221 self.win = curses.newwin(1, 1, self.start.y, self.start.x)
222 self.size_def = size # store for re-calling .size on SIGWINCH
224 self.do_update = True
225 self.visible = visible
230 return YX(*self.win.getmaxyx())
233 def size(self, size):
234 """Set window size. Size be y,x tuple. If y or x None, use legal max."""
235 n_lines, n_cols = size
236 getmaxyx = YX(*self.tui.stdscr.getmaxyx())
238 n_lines = getmaxyx.y - self.start.y
240 n_cols = getmaxyx.x - self.start.x
241 self.win.resize(n_lines, n_cols)
244 getmaxyx = YX(*self.win.getmaxyx())
245 return getmaxyx.y * getmaxyx.x
247 def safe_write(self, foo):
249 def to_chars_with_attrs(part):
250 attr = curses.A_NORMAL
252 if not type(part) == str:
253 part_string = part[0]
255 return [(char, attr) for char in part_string]
257 chars_with_attrs = []
258 if type(foo) == str or (len(foo) == 2 and type(foo[1]) == int):
259 chars_with_attrs += to_chars_with_attrs(foo)
262 chars_with_attrs += to_chars_with_attrs(part)
264 if len(chars_with_attrs) < len(self):
265 for char_with_attr in chars_with_attrs:
266 self.win.addstr(char_with_attr[0], char_with_attr[1])
267 else: # workaround to <https://stackoverflow.com/q/7063128>
268 cut = chars_with_attrs[:len(self) - 1]
269 last_char_with_attr = chars_with_attrs[len(self) - 1]
270 self.win.addstr(self.size.y - 1, self.size.x - 2,
271 last_char_with_attr[0], last_char_with_attr[1])
272 self.win.insstr(self.size.y - 1, self.size.x - 2, ' ')
274 for char_with_attr in cut:
275 self.win.addstr(char_with_attr[0], char_with_attr[1])
277 def ensure_freshness(self, do_refresh=False):
281 for key in self.check_updates:
282 if key in self.tui.to_update and self.tui.to_update[key]:
290 for child in self.children:
291 did_refresh = child.ensure_freshness(do_refresh) | did_refresh
295 class EditWidget(Widget):
298 self.safe_write((''.join(self.tui.to_send), curses.color_pair(1)))
301 class TextLinesWidget(Widget):
304 lines = self.get_text_lines()
305 line_width = self.size.x
308 to_pad = line_width - (len(line) % line_width)
309 if to_pad == line_width:
311 to_join += [line + ' '*to_pad]
312 self.safe_write((''.join(to_join), curses.color_pair(3)))
315 class LogWidget(TextLinesWidget):
317 def get_text_lines(self):
318 return self.tui.game.log_text.split('\n')
321 class DescriptorWidget(TextLinesWidget):
323 def get_text_lines(self):
325 pos_i = self.tui.game.map_.\
326 get_position_index(self.tui.examiner_position[1])
327 terrain = self.tui.game.map_.terrain[pos_i]
329 for t in self.tui.game.things_at_pos(self.tui.examiner_position):
334 class PopUpWidget(Widget):
337 self.safe_write(self.tui.popup_text)
339 def reconfigure(self):
340 size = (1, len(self.tui.popup_text))
343 getmaxyx = YX(*self.tui.stdscr.getmaxyx())
344 offset_y = int(getmaxyx.y / 2 - size.y / 2)
345 offset_x = int(getmaxyx.x / 2 - size.x / 2)
346 self.start = YX(offset_y, offset_x)
347 self.win.mvwin(self.start.y, self.start.x)
350 class ItemsSelectorWidget(Widget):
352 def __init__(self, headline, selection, *args, **kwargs):
353 super().__init__(*args, **kwargs)
354 self.headline = headline
355 self.selection = selection
357 def ensure_freshness(self, *args, **kwargs):
358 # We only update pointer on non-empty selection so that the zero-ing
359 # of the selection at TURN_FINISHED etc. before pulling in a new
360 # state does not destroy any memory of previous item pointer positions.
361 if len(self.selection) > 0 and\
362 len(self.selection) < self.tui.item_pointer + 1 and\
363 self.tui.item_pointer > 0:
364 self.tui.item_pointer = max(0, len(self.selection) - 1)
365 self.tui.to_update[self.check_updates[0]] = True
366 super().ensure_freshness(*args, **kwargs)
369 lines = [self.headline]
371 for id_ in self.selection:
372 pointer = '*' if counter == self.tui.item_pointer else ' '
373 t = self.tui.game.get_thing(id_)
374 lines += ['%s %s' % (pointer, t.type_)]
376 line_width = self.size.x
379 to_pad = line_width - (len(line) % line_width)
380 if to_pad == line_width:
382 to_join += [line + ' '*to_pad]
383 self.safe_write((''.join(to_join), curses.color_pair(3)))
386 class MapWidget(Widget):
390 def annotated_terrain():
391 terrain_as_list = list(self.tui.game.map_.terrain[:])
392 for t in self.tui.game.things:
393 if t.id_ in self.tui.game.player_inventory:
395 pos_i = self.tui.game.map_.get_position_index(t.position[1])
396 symbol = self.tui.game.symbol_for_type(t.type_)
397 if terrain_as_list[pos_i][0] in {'f', '@', 'm'}:
398 old_symbol = terrain_as_list[pos_i][0]
399 if old_symbol in {'@', 'm'}:
401 terrain_as_list[pos_i] = (symbol, '+')
403 terrain_as_list[pos_i] = symbol
404 if self.tui.examiner_mode:
405 pos_i = self.tui.game.map_.\
406 get_position_index(self.tui.examiner_position[1])
407 terrain_as_list[pos_i] = (terrain_as_list[pos_i][0], '?')
408 return terrain_as_list
410 def pad_or_cut_x(lines):
411 line_width = self.size.x
412 for y in range(len(lines)):
414 if line_width > len(line):
415 to_pad = line_width - (len(line) % line_width)
416 lines[y] = line + '0' * to_pad
418 lines[y] = line[:line_width]
421 if len(lines) < self.size.y:
422 to_pad = self.size.y - len(lines)
423 lines += to_pad * ['0' * self.size.x]
425 def lines_to_colored_chars(lines):
426 chars_with_attrs = []
427 for c in ''.join(lines):
429 chars_with_attrs += [(c, curses.color_pair(1))]
431 chars_with_attrs += [(c, curses.color_pair(4))]
433 chars_with_attrs += [(c, curses.color_pair(2))]
434 elif c in {'x', 'X', '#'}:
435 chars_with_attrs += [(c, curses.color_pair(3))]
437 chars_with_attrs += [(c, curses.color_pair(5))]
439 chars_with_attrs += [c]
440 return chars_with_attrs
442 if self.tui.game.map_.terrain == '':
445 self.safe_write(''.join(lines))
448 annotated_terrain = annotated_terrain()
449 center = self.tui.game.player.position
450 if self.tui.examiner_mode:
451 center = self.tui.examiner_position
452 lines = self.tui.game.map_.\
453 format_to_view(annotated_terrain, center, self.size)
456 self.safe_write(lines_to_colored_chars(lines))
459 class TurnWidget(Widget):
462 self.safe_write((str(self.tui.game.turn), curses.color_pair(2)))
465 class HealthWidget(Widget):
468 if hasattr(self.tui.game.player, 'health'):
469 self.safe_write((str(self.tui.game.player.health),
470 curses.color_pair(2)))
473 class TextLineWidget(Widget):
475 def __init__(self, text_line, *args, **kwargs):
476 self.text_line = text_line
477 super().__init__(*args, **kwargs)
480 self.safe_write(self.text_line)
485 def __init__(self, plom_socket, game, q):
486 self.socket = plom_socket
490 self.parser = Parser(self.game)
492 self.item_pointer = 0
493 self.examiner_position = (YX(0,0), YX(0, 0))
494 self.examiner_mode = False
495 self.popup_text = 'Hi bob'
497 self.draw_popup_if_visible = True
498 curses.wrapper(self.loop)
500 def loop(self, stdscr):
502 def setup_screen(stdscr):
504 self.stdscr.refresh() # will be called by getkey else, clearing screen
505 self.stdscr.timeout(10)
507 def switch_widgets(widget_1, widget_2):
508 widget_1.visible = False
509 widget_2.visible = True
510 trigger = widget_2.check_updates[0]
511 self.to_update[trigger] = True
513 def selectables_menu(key, widget, selectables, f):
515 switch_widgets(widget, map_widget)
517 self.item_pointer += 1
518 elif key == 'k' and self.item_pointer > 0:
519 self.item_pointer -= 1
520 elif not f(key, selectables):
522 trigger = widget.check_updates[0]
523 self.to_update[trigger] = True
525 def pickup_menu(key):
527 def f(key, selectables):
528 if key == 'p' and len(selectables) > 0:
529 id_ = selectables[self.item_pointer]
530 self.socket.send('TASK:PICKUP %s' % id_)
531 self.socket.send('GET_PICKABLE_ITEMS')
536 selectables_menu(key, pickable_items_widget,
537 self.game.pickable_items, f)
539 def inventory_menu(key):
541 def f(key, selectables):
542 if key == 'd' and len(selectables) > 0:
543 id_ = selectables[self.item_pointer]
544 self.socket.send('TASK:DROP %s' % id_)
545 elif key == 'e' and len(selectables) > 0:
546 id_ = selectables[self.item_pointer]
547 self.socket.send('TASK:EAT %s' % id_)
552 selectables_menu(key, inventory_widget,
553 self.game.player_inventory, f)
555 def move_examiner(direction):
556 start_pos = self.examiner_position
557 new_examine_pos = self.game.map_geometry.move(start_pos, direction,
559 self.game.map_.start_indented)
560 if new_examine_pos[0] == (0,0):
561 self.examiner_position = new_examine_pos
562 self.to_update['map'] = True
564 def switch_to_pick_or_drop(target_widget):
565 self.item_pointer = 0
566 switch_widgets(map_widget, target_widget)
567 if self.examiner_mode:
568 self.examiner_mode = False
569 switch_widgets(descriptor_widget, log_widget)
571 def toggle_examiner_mode():
572 if self.examiner_mode:
573 self.examiner_mode = False
574 switch_widgets(descriptor_widget, log_widget)
576 self.examiner_mode = True
577 self.examiner_position = self.game.player.position
578 switch_widgets(log_widget, descriptor_widget)
579 self.to_update['map'] = True
582 if popup_widget.visible:
583 popup_widget.visible = False
584 for w in top_widgets:
585 w.ensure_freshness(True)
587 self.to_update['popup'] = True
588 popup_widget.visible = True
589 popup_widget.reconfigure()
590 self.draw_popup_if_visible = True
592 def try_write_keys():
593 if len(key) == 1 and key in ASCII_printable and \
594 len(self.to_send) < len(edit_line_widget):
595 self.to_send += [key]
596 self.to_update['edit'] = True
597 elif key == 'KEY_BACKSPACE':
598 self.to_send[:] = self.to_send[:-1]
599 self.to_update['edit'] = True
600 elif key == '\n': # Return key
601 self.socket.send(''.join(self.to_send))
603 self.to_update['edit'] = True
605 def try_examiner_keys():
607 move_examiner('UPLEFT')
609 move_examiner('UPRIGHT')
611 move_examiner('LEFT')
613 move_examiner('RIGHT')
615 move_examiner('DOWNLEFT')
617 move_examiner('DOWNRIGHT')
619 def try_player_move_keys():
621 self.socket.send('TASK:MOVE UPLEFT')
623 self.socket.send('TASK:MOVE UPRIGHT')
625 self.socket.send('TASK:MOVE LEFT')
627 self.socket.send('TASK:MOVE RIGHT')
629 self.socket.send('TASK:MOVE DOWNLEFT')
631 self.socket.send('TASK:MOVE DOWNRIGHT')
634 curses.init_pair(1, curses.COLOR_BLACK, curses.COLOR_RED)
635 curses.init_pair(2, curses.COLOR_BLACK, curses.COLOR_GREEN)
636 curses.init_pair(3, curses.COLOR_BLACK, curses.COLOR_BLUE)
637 curses.init_pair(4, curses.COLOR_BLACK, curses.COLOR_YELLOW)
638 curses.init_pair(5, curses.COLOR_BLACK, curses.COLOR_WHITE)
640 # Basic curses initialization work.
642 curses.curs_set(False) # hide cursor
645 # With screen initialized, set up widgets with their curses windows.
646 edit_widget = TextLineWidget('SEND:', self, YX(0, 0), YX(1, 20))
647 edit_line_widget = EditWidget(self, YX(0, 6), YX(1, 14), ['edit'])
648 edit_widget.children += [edit_line_widget]
649 turn_widget = TextLineWidget('TURN:', self, YX(2, 0), YX(1, 20))
650 turn_widget.children += [TurnWidget(self, YX(2, 6), YX(1, 14), ['turn'])]
651 health_widget = TextLineWidget('HEALTH:', self, YX(3, 0), YX(1, 20))
652 health_widget.children += [HealthWidget(self, YX(3, 8), YX(1, 12), ['turn'])]
653 log_widget = LogWidget(self, YX(5, 0), YX(None, 20), ['log'])
654 descriptor_widget = DescriptorWidget(self, YX(5, 0), YX(None, 20),
656 map_widget = MapWidget(self, YX(0, 21), YX(None, None), ['map'])
657 inventory_widget = ItemsSelectorWidget('INVENTORY:',
658 self.game.player_inventory,
659 self, YX(0, 21), YX(None, None),
660 ['inventory'], False)
661 pickable_items_widget = ItemsSelectorWidget('PICKABLE:',
662 self.game.pickable_items,
667 top_widgets = [edit_widget, turn_widget, health_widget, log_widget,
668 descriptor_widget, map_widget, inventory_widget,
669 pickable_items_widget]
670 popup_widget = PopUpWidget(self, YX(0, 0), YX(1, 1), visible=False)
672 # Ensure initial window state before loop starts.
673 for w in top_widgets:
674 w.ensure_freshness(True)
675 self.socket.send('GET_GAMESTATE')
680 for w in top_widgets:
681 if w.ensure_freshness():
682 self.draw_popup_if_visible = True
683 if popup_widget.visible and self.draw_popup_if_visible:
684 popup_widget.ensure_freshness(True)
685 self.draw_popup_if_visible = False
686 for k in self.to_update.keys():
687 self.to_update[k] = False
689 # Handle input from server.
692 command = self.queue.get(block=False)
695 self.game.handle_input(command)
697 # Handle keys (and resize event read as key).
699 key = self.stdscr.getkey()
700 if key == 'KEY_RESIZE':
702 setup_screen(curses.initscr())
703 for w in top_widgets:
705 w.ensure_freshness(True)
706 elif key == '\t': # Tabulator key.
707 write_mode = False if write_mode else True
712 elif map_widget.visible:
714 toggle_examiner_mode()
716 self.socket.send('GET_PICKABLE_ITEMS')
717 switch_to_pick_or_drop(pickable_items_widget)
719 switch_to_pick_or_drop(inventory_widget)
720 elif self.examiner_mode:
723 try_player_move_keys()
724 elif pickable_items_widget.visible:
726 elif inventory_widget.visible:
731 # Quit when server recommends it.
732 if self.game.do_quit:
736 s = socket.create_connection(('127.0.0.1', 5000))
737 plom_socket = PlomSocket(s)
740 t = threading.Thread(target=recv_loop, args=(plom_socket, game, q))
742 TUI(plom_socket, game, q)