window.py

import urwid

class MainWindow:
    ##(name, foreground, background, mono, foreground_high, background_high)
    palette = [
        ('body', '', '', '', '#fff', '#000'),
        ('footer', '', '', '', '#fff', '#000'),
        ('header', '', '', '', '#fff', '#000'),
        ('white', '', '', '', '#fff', '#000'),
        ('brown', '', '', '', '#d80', '#000'),
        ('yellow', '', '', '', '#ff0', '#000'),
        ('red', '', '', '', '#f00', '#000'),
        ('flash brown', '', '', 'bold', '#d80,standout', '#000'),
        ('flash yellow', '', '', 'bold', '#ff0,standout', '#000'),
        ('flash red', '', '', 'bold', '#f00,standout', '#000'),
        ('light white', '', '', 'bold', '#fff', '#000'),
        ('light blue', '', '', 'bold', '#0af', '#000'),
        ('white blue', '', '', '', '#adf', '#000'),
    ]

    def __init__(self, game):
        self.game = game

        self.game.initialize_player()

        self.header_text = urwid.Text(u"Top\nheader")
        self.header = urwid.AttrWrap(self.header_text, 'header')

        self.footer_text = urwid.Text(u"")
        self.footer = urwid.AttrWrap(self.footer_text, 'footer')

        # self.cell_txt = urwid.Text("")
        # self.pathogen_txt = urwid.Text("")
        # self.player_col = urwid.Filler(self.cell_txt, valign='top', height='pack')
        # self.pathogen_col = urwid.Filler(self.pathogen_txt, valign='top', height='pack')
        # self.body = urwid.Columns([urwid.LineBox(self.player_col), urwid.LineBox(self.pathogen_col)])

        self.area = GameArea()
        self.area.mainloop = self
        body = self.area.board_overview()

        self.view = MainFrame(header=self.header, body=body, footer=self.footer, focus_part='body')
        self.view.mainloop = self

        self.last_alarm = None

    def main(self):

        #self.loop = urwid.MainLoop(self.view, self.palette, input_filter=self.input_filter) #, pop_ups=True)
        self.loop = urwid.MainLoop(self.view, self.palette, pop_ups=True)
        self.loop.main_window = self
        #self.loop.mwin = self

        self.loop.screen.set_terminal_properties(colors=256)
        self.loop.screen.register_palette(self.palette)

        ### TODO: check that constantly adding alerts doesn't leak mem

        # 1-second game tick
        self.last_alarm = self.loop.set_alarm_in(0, callback=self.game_tick)
        # 10-second game tick
        #self.loop.set_alarm_in(0, callback=self.protein_gain)

        self.loop.run()

    def write_header(self):
        header_txt = "Ticks: %s\nProteins: %s\nLevel: %s\n" % (self.game.ticks, self.game.proteins, self.game.level)
        self.header_text.set_text(header_txt)

    def write_footer(self):
        #footer_txt = self.game.display_log()
        self.footer_text.set_text("")

    def draw_cells(self):
        cell_str = []
        for i in self.game.player_cells:
            cell_str.append((i.palette, i.symbol))
        if cell_str:
            self.area.cells.set_text(cell_str)

    def draw_pathogens(self):
        pathogen_str = []
        for i in self.game.pathogens:
            pathogen_str.append((i.palette, i.symbol))
        if pathogen_str:
            self.area.pathogens.set_text(pathogen_str)

    def draw_cells_desc(self):
        content = []
        for i in self.game.player_cells:
            content.append((i.palette, i.symbol))
            content.append(('body', ": %s (age: %s)" % (i.name, i.age)))
            content.append(('body', "[ATK/DEF: %s/%s]" % (i.base_attack, i.defense)))
            content.append(('body', '\n'))
        if content:
            self.area.cells_txt.set_text(content)

    def draw_pathogens_desc(self):
        content = []
        for i in self.game.pathogens:
            content.append((i.palette, i.symbol))
            content.append(('body', ": %s (age: %s)" % (i.name, i.age)))
            content.append(('body', "[ATK/DEF: %s/%s]" % (i.base_attack, i.defense)))
            content.append(('body', '\n'))
        if content:
            self.area.pathogens_txt.set_text(content)


    def game_tick(self, loop, user_data):
        # So we have a single alarm object at all times?
        self.loop.remove_alarm(self.last_alarm)

        # If the game is paused, skip these events
        if not self.game.paused:
            # Player attacks
            #self.game.player_attack(self)

            # Cell actions
            for i in self.game.player_cells:
                i.heartbeat(self)

            # Pathogens actions
            for i in self.game.pathogens:
                i.heartbeat(self)

            # Gain a protein
            self.protein_gain(1)

            self.game.check_level_state()
            self.game.age_all_cells()
            self.game.ticks += 1

        self.write_header()

        self.draw_cells()
        self.draw_cells_desc()
        self.draw_pathogens()
        self.draw_pathogens_desc()

        self.write_footer()

        self.last_alarm = self.loop.set_alarm_in(sec=1, callback=self.game_tick)

    def protein_gain(self, amount=0):
        self.game.proteins += amount

class MainFrame(urwid.Frame):

    active_window = None

    def keypress(self, size, key):
        #self.mainloop.game.log("key pressed: %s" % key)
        if key == 'q':
            raise urwid.ExitMainLoop()
        elif key == 'esc':
            raise urwid.ExitMainLoop()
        elif key == 'n':
            self.mainloop.game.spawn_cell()
            self.mainloop.write_header()
        elif key == 'd':
            self.purchase_pop_up()
        elif key == ' ':
            if self.mainloop.game.paused:
                self.mainloop.game.log("Game unpaused.")
                self.mainloop.game.paused = False
            else:
                self.mainloop.game.log("Game paused.")
                self.mainloop.game.paused = True
        elif key == 'up':
            #self.mainloop.game.log("key pressed: %s" % key)
            body = self.mainloop.area.lymphatic_view()
            self.mainloop.view.contents['body'] = (body, None)
            self.mainloop.loop.draw_screen()
        elif key == 'down':
            #self.mainloop.game.log("key pressed: %s" % key)
            body = self.mainloop.area.board_overview()
            self.mainloop.view.contents['body'] = (body, None)
            self.mainloop.loop.draw_screen()
        elif key == 'm':
            # Pause the game while in log view
            self.mainloop.game.paused = True
            self.mainloop.game.log("Current focus: %s" % self.get_focus_widgets())
            body = self.mainloop.area.log_view()
            self.mainloop.view.contents['body'] = (body, None)
            self.mainloop.loop.draw_screen()


            #self.lymph_window()

    #def lymph_window(self):
        # lymph_txt = urwid.Text("Here would be details about the lymphatic system, the rate of WBC generation and the like.")
        # box = urwid.LineBox(urwid.Filler(lymph_txt))
        # overlay = urwid.Overlay(
        #     top_w = box, 
        #     bottom_w = self,
        #     align = 'left',
        #     width = 40,
        #     valign = 'middle',
        #     height = 40,
        #     #left = 5,
        #     )
        # self.mainloop.loop.widget = overlay


    def purchase_pop_up(self):

        question = urwid.Text(("bold", "A pop-up!"), "center")
        prompt = urwid.LineBox(urwid.Filler(question))
        overlay = OptionOverlay(
            top_w = prompt, 
            bottom_w = self,
            align = 'left',
            width = 10,
            valign = 'middle',
            height = 6,
            left = 5,
            )
        self.mainloop.loop.widget = overlay


class OptionOverlay(urwid.Overlay):
    def selectable(self):
        return True

    def keypress(self, size, key):
        if key == 'q':
            raise urwid.ExitMainLoop()
        elif key == 'enter':
            self.bottom_w.mainloop.loop.widget = self.bottom_w

class GameArea:
    def __init__(self):
        self.board_overview()

    def selectable(self):
        return True

    def keypress(self, size, key):
        pass
        #self.mainloop.game.log("key pressed: %s" % key)

    def board_overview(self):
        self.cells = urwid.Text("")
        self.pathogens = urwid.Text("")
        self.cells_txt = urwid.Text("")
        self.pathogens_txt = urwid.Text("")

        self.cells_col = urwid.Pile([
            (10, urwid.LineBox(urwid.Filler(self.cells), title='White Blood Cells')),
            urwid.LineBox(urwid.Filler(self.cells_txt, valign='top'), title='Cell Details'),
        ])

        self.pathogens_col = urwid.Pile([
            (10, urwid.LineBox(urwid.Filler(self.pathogens), title='Pathogens')),
            urwid.LineBox(urwid.Filler(self.pathogens_txt, valign='top'), title='Pathogen Details'),
        ])

        body = urwid.AttrWrap(urwid.Columns([self.cells_col, self.pathogens_col]), 'body')
        return body

    def lymphatic_view(self):
        lymph_txt = urwid.Text("Here would be details about the lymphatic system, the rate of WBC generation and the like.")
        box = urwid.LineBox(urwid.Filler(lymph_txt))
        body = urwid.AttrWrap(box, 'body')
        return body

    def log_view(self):
        footer_txt = self.mainloop.game.display_log()
        footer = urwid.Text(footer_txt)
        box = LogBody(urwid.Filler(footer))
        body = urwid.AttrWrap(box, 'body')
        return body


# test class to override keypresses in the body of MainFrame for specific windows
class LogBody(urwid.LineBox):
    def selectable(self):
        return True

    def keypress(self, size, key):   
        if key == 'esc':
            raise urwid.ExitMainLoop()