gui.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "scrolled": false
   },
   "outputs": [],
   "source": [
    "from ipywidgets import IntSlider, VBox, HBox, Dropdown, Output, IntText, Layout, Widget, Button, HTML\n",
    "from IPython.display import display, clear_output\n",
    "from colors import Colors\n",
    "from state import State, get_event_info\n",
    "from string_utils import Dashboard, print_resources, pretty_print_number, pretty_print_time, print_plan\n",
    "from time import sleep\n",
    "from strategy import Strategy\n",
    "from threading import Thread\n",
    "\n",
    "#hack to prevent output widget from becoming a scrolling output\n",
    "display(HTML(\"\"\"\n",
    "    <style>\n",
    "       .jupyter-widgets-output-area .output_scroll {\n",
    "            height: 100% !important;\n",
    "            border-radius: unset !important;\n",
    "            -webkit-box-shadow: unset !important;\n",
    "            box-shadow: unset !important;\n",
    "        }\n",
    "    </style>\n",
    "\"\"\"))\n",
    "\n",
    "MAX_LEVEL = 10\n",
    "MAX_CHAMPIONS = 8\n",
    "MAX_RESOURCES = 4\n",
    "\n",
    "class ClearOutput(Output):\n",
    "    \"\"\"\n",
    "    The context manager for output widgets can be a bit wonky.\n",
    "    This extension provides a clean mechanism for replicating\n",
    "    the following pattern:\n",
    "    \n",
    "    with out:\n",
    "        clear_output(wait = True)\n",
    "        print(text)\n",
    "        \n",
    "    which appears to be thread-safe\n",
    "    \"\"\"\n",
    "    \n",
    "    def clear_print(self, text):\n",
    "        self.outputs = ({'output_type': 'stream', 'name': 'stdout', 'text': text},)\n",
    "\n",
    "class NoWith:\n",
    "    \"\"\"\n",
    "    I don't know... trying to create a with block that does nothing special\n",
    "    \"\"\"\n",
    "    __enter__ = lambda *args: None\n",
    "    __exit__ = lambda *args: None\n",
    "\n",
    "class Solver(Thread):\n",
    "    def __init__(self, strategy, context=None, *args, **kwargs):\n",
    "        \"\"\"\n",
    "        strategy is a Strategy object that will be solved\n",
    "        context is the output context to be used, if None it will use stdout\n",
    "        the dashboard object will print to the output widget if supplied as context\n",
    "        \"\"\"\n",
    "        super().__init__()\n",
    "        self.strategy = strategy\n",
    "        self.strategy.stop = False\n",
    "        self.dashboard = Dashboard()\n",
    "        self.dashboard.set_state(self.strategy.state)\n",
    "        if context is not None:\n",
    "            self.dashboard.set_context(context)\n",
    "            self.context = context\n",
    "        self.args = args\n",
    "        self.kwargs = kwargs\n",
    "        \n",
    "    def interrupt(self):\n",
    "        \"\"\"\n",
    "        Call this to set a flag to stop the program\n",
    "        \"\"\"\n",
    "        self.strategy.stop = True\n",
    "        \n",
    "    def run(self):\n",
    "        self.strategy.plan(\n",
    "            dashboard = self.dashboard, \n",
    "            mode = 'hybrid', \n",
    "            swap_levels = 1,\n",
    "            *self.args, \n",
    "            **self.kwargs\n",
    "        )\n",
    "        self.dashboard.show(force = True, include_urgency = True)\n",
    "        \n",
    "        \n",
    "class Time(HBox):\n",
    "    def __init__(self):\n",
    "        self.display = ClearOutput(layout = Layout(width = '20em'))\n",
    "        style = {'description_width' : '6em'}\n",
    "        layout = Layout(width = 'max-content')\n",
    "        self.time = 3*24*3600\n",
    "        self.days = Dropdown(description = 'days', options = range(4), index = 3, style = style, layout = layout)\n",
    "        self.hours = Dropdown(description = 'hours', options = range(24), index = 0, style = style, layout = layout)\n",
    "        self.minutes = Dropdown(description='minutes', options = range(60), index = 0, style = style, layout = layout)\n",
    "        self.seconds = Dropdown(description='seconds', options = range(60), index = 0, style = style, layout = layout)\n",
    "        super().__init__([self.display, self.days, self.hours, self.minutes, self.seconds])\n",
    "        for widget in [self.days, self.hours, self.minutes, self.seconds]:\n",
    "            widget.observe(self.update)\n",
    "        self.update()\n",
    "        \n",
    "    def update(self, event = None):\n",
    "        self.time = (\n",
    "            self.days.value * 24 * 3600 + \n",
    "            self.hours.value * 3600 + \n",
    "            self.minutes.value * 60 + \n",
    "            self.seconds.value\n",
    "        )\n",
    "        self.display.clear_print(pretty_print_time(self.time))\n",
    "        \n",
    "        \n",
    "        \n",
    "class Resource (VBox):\n",
    "    def __init__(self, color, *args, **kwargs):\n",
    "        self.name = ClearOutput()\n",
    "        self.resource = IntText(continuous_update = False, layout = Layout(width = 'auto'))\n",
    "        self.resource.observe(self.on_resource_change)\n",
    "        self.pretty = ClearOutput()\n",
    "        self.color = color\n",
    "        super().__init__([self.name, self.resource, self.pretty], *args, **kwargs)\n",
    "        \n",
    "    def on_resource_change(self, change):\n",
    "        self.pretty.clear_print(Colors.color(pretty_print_number(self.resource.value), self.color))\n",
    "        \n",
    "\n",
    "class Resources(HBox):\n",
    "    COLORS = ['blue', 'green', 'red', 'gold', 'bold']\n",
    "    def __init__(self):\n",
    "        layout = Layout(width = '100%')\n",
    "        event_info = get_event_info()\n",
    "        self.resources = []\n",
    "        for i in range(5):\n",
    "            self.resources.append(Resource(self.COLORS[i], layout = layout))\n",
    "        super().__init__(self.resources)\n",
    "                \n",
    "class Upgrade(HBox):\n",
    "    def __init__(self, info):\n",
    "        style = {'description_width' : '12em'}\n",
    "        layout = Layout(width = '36em')\n",
    "        self.slider = IntSlider(value = 0, min = 0, max = MAX_LEVEL, style = style, layout = layout)\n",
    "        self.slider.observe(self.update)\n",
    "        self.cost = ClearOutput()\n",
    "        self.info = info\n",
    "        self.callback_f = lambda : None\n",
    "        super().__init__([self.slider, self.cost])\n",
    "        self.update()\n",
    "        \n",
    "    def set_callback(self, f):\n",
    "        self.callback_f = f\n",
    "        \n",
    "    def callback(self):\n",
    "        self.callback_f()\n",
    "        #allow higher level structure to receive updates from self\n",
    "        \n",
    "    def update(self, event = None):\n",
    "        self.slider.description = self.info['name']\n",
    "        self.slider.max = self.info['max_level']\n",
    "        \n",
    "        if self.slider.value == self.info['max_level']:\n",
    "            self.cost.clear_print(Colors.color('MAXED', 'rainbow'))\n",
    "        else:\n",
    "            self.cost.clear_print(print_resources(self.info['upgrade_costs'][self.slider.value]))\n",
    "        self.callback()\n",
    "        \n",
    "class Gem(Upgrade):\n",
    "    def __init__(self):\n",
    "        super().__init__(None)\n",
    "        \n",
    "    def update(self, event = None):\n",
    "        cost = [100, 250, 500, 750, 1000] #i think? don't really care\n",
    "        self.slider.description = \"gem production boost\"\n",
    "        self.slider.max = 5\n",
    "        if self.slider.value == 5:\n",
    "            self.cost.clear_print(Colors.color('MAXED', 'rainbow'))\n",
    "        else:\n",
    "            self.cost.clear_print(Colors.color(f\"{cost[self.slider.value]} gems\", 'rainbow'))\n",
    "        self.callback()\n",
    "\n",
    "class RMD(Upgrade):\n",
    "    def __init__(self):\n",
    "        super().__init__(None)\n",
    "        \n",
    "    def update(self, event = None):\n",
    "        cost = [0] #i think? don't really care\n",
    "        self.slider.description = \"real money boost\"\n",
    "        self.slider.max = 1\n",
    "        if self.slider.value == 1:\n",
    "            self.cost.clear_print(Colors.color('MAXED', 'rainbow'))\n",
    "        else:\n",
    "            self.cost.clear_print(Colors.color(f\"$$$$$\", 'rainbow'))\n",
    "        self.callback()\n",
    "                 \n",
    "    \n",
    "class GUI(VBox):\n",
    "    EVENTS = ['other_tower', 'sid_in_space', 'happy_time', 'aqua_adventure', 'lost_in_time', 'through_the_portal']\n",
    "    def __init__(self):\n",
    "        self.thread = None\n",
    "        self.state = State()\n",
    "        self.dropdown = Dropdown(options = self.EVENTS, index = 0, description = 'event')\n",
    "        self.dropdown.observe(self.change_event)\n",
    "        self.create_plan_button = Button(description = 'create_plan')\n",
    "        self.create_plan_button.on_click(self.execute)\n",
    "        self.interrupt_button = Button(description = 'interrupt')\n",
    "        self.interrupt_button.on_click(self.interrupt)\n",
    "        #create custom thread object with interrupt\n",
    "        self.strategy_output = ClearOutput(layout = Layout(height = '5000px'))\n",
    "        super().__init__([])\n",
    "        self.build()\n",
    "        \n",
    "    def change_event(self, event):\n",
    "        self.state.set_event_by_name(self.dropdown.value)\n",
    "        self.build()\n",
    "    \n",
    "    def sync(self):\n",
    "        resources = ['blue', 'green', 'red', 'gold', 'damage']\n",
    "        event_info = get_event_info()\n",
    "        self.state.time = self.time.time\n",
    "        self.state.gem_level = self.gem_level.slider.value\n",
    "        self.state.boost_level = self.boost_level.slider.value\n",
    "        self.state.damage_level = self.damage.slider.value\n",
    "        self.state.speed_level = self.speed.slider.value\n",
    "        if event_info['has_speed2']:\n",
    "            self.state.speed2_level = self.speed2.slider.value\n",
    "            \n",
    "        #setting structs and arrays of a non cimported class is tricky\n",
    "        #need to set them atomically, and not by setting their components\n",
    "        self.state.resources = {\n",
    "            resources[i] : self.current_resources.resources[i].resource.value for i in range(5)\n",
    "        }\n",
    "        self.state.champion_levels = [\n",
    "            self.champions[i].slider.value for i in range(event_info['n_champions'])\n",
    "        ] + [0] * (MAX_CHAMPIONS - event_info['n_champions'])\n",
    "        self.state.resource_levels = [\n",
    "            self.resources[i].slider.value for i in range(event_info['n_resources'])\n",
    "        ] + [0] * (MAX_RESOURCES - event_info['n_resources'])\n",
    "        self.state.update_resources_per_second()\n",
    "        \n",
    "    def interrupt(self, event = None):\n",
    "        if self.thread is not None:\n",
    "            self.thread.interrupt()\n",
    "            self.thread = None\n",
    "    \n",
    "    def execute(self, event = None):\n",
    "        self.sync()\n",
    "        if self.thread is not None:\n",
    "            self.thread.interrupt()\n",
    "            \n",
    "        self.strategy = Strategy(state = self.state)\n",
    "        self.thread = Solver(self.strategy, self.strategy_output)\n",
    "        self.thread.start()\n",
    "        \n",
    "    def build(self):\n",
    "        \"\"\"called on init or when the event is changed\"\"\"\n",
    "        # these names do not appear in the event_info currently\n",
    "        DAMAGE_NAME = ['Damage', 'Distance', 'Hearts', 'Depth', 'Time Dust', 'Damage']\n",
    "        self.interrupt()\n",
    "        event_info = get_event_info()\n",
    "        self.current_resources = Resources()\n",
    "        for i in range(event_info['n_resources']):\n",
    "            self.current_resources.resources[i].name.clear_print(\n",
    "                Colors.color(event_info['resources'][i]['name'].decode(), self.current_resources.COLORS[i])\n",
    "            )\n",
    "            self.current_resources.resources[i].layout = Layout(width = '100%', display = 'block')\n",
    "        for j in range(i+1, 4):\n",
    "            self.current_resources.resources[j].layout = Layout(display = 'none')\n",
    "        self.current_resources.resources[-1].name.clear_print(Colors.color(DAMAGE_NAME[self.dropdown.index], 'bold'))\n",
    "        self.time = Time()\n",
    "        self.gem_level = Gem()\n",
    "        self.boost_level = RMD()\n",
    "        self.champions = [Upgrade(event_info['champions'][i]) for i in range(event_info['n_champions'])]\n",
    "        self.resources = [Upgrade(event_info['resources'][i]) for i in range(event_info['n_resources'])]\n",
    "        self.damage = Upgrade(event_info['damage'])\n",
    "        self.speed = Upgrade(event_info['speed'])\n",
    "        self.speed2 = Upgrade(event_info['speed2'])if event_info['has_speed2'] else HBox()\n",
    "        self.children = (\n",
    "            self.dropdown,\n",
    "            self.time,\n",
    "            self.current_resources, \n",
    "            *self.champions,\n",
    "            self.gem_level,\n",
    "            self.boost_level,\n",
    "            *self.resources,\n",
    "            self.speed,\n",
    "            self.damage,\n",
    "            self.speed2,\n",
    "            HBox([self.create_plan_button, self.interrupt_button]),\n",
    "            self.strategy_output\n",
    "        )\n",
    "        \n",
    "State.set_event_by_name('other_tower')\n",
    "\n",
    "gui = GUI()\n",
    "gui"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.8.6"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}