canvas map overview WIP

ui/basemenu.py

@@ -49,6 +49,12 @@ class BaseMenu(Menu):
             for unit_type in units:
                 purchase_row(unit_type, db.PRICES[unit_type])
 
+    def dismiss(self):
+        if sum([x for x in self.event.units.values()]) == 0:
+            self.game.units_delivery_remove(self.event)
+
+        super(BaseMenu, self).dismiss()
+
     def buy(self, unit_type):
         def action():
             price = db.PRICES[unit_type]

ui/mainmenu.py

@@ -4,23 +4,26 @@ from tkinter.ttk import *
 from ui.window import *
 from ui.eventmenu import *
 from ui.basemenu import *
+from ui.overviewcanvas import *
 
 from game.game import *
 
 
 class MainMenu(Menu):
+    basemenu = None  # type: BaseMenu
+
     def __init__(self, window: Window, parent, game: Game):
         super(MainMenu, self).__init__(window, parent, game)
 
-        self.image = PhotoImage(file="resources/caumap.gif")
-        map = Label(window.left_pane, image=self.image)
-        map.grid()
+        self.upd = OverviewCanvas(self.window.left_pane, self, game)
+        self.upd.update()
 
         self.frame = self.window.right_pane
         self.frame.grid_columnconfigure(0, weight=1)
 
     def display(self):
         self.window.clear_right_pane()
+        self.upd.update()
 
         row = 1
 
@@ -34,23 +37,14 @@ class MainMenu(Menu):
             Button(self.frame, text=text, command=self.start_event(event)).grid(row=row, sticky=N)
             row += 1
 
-        def cp_button(cp):
-            nonlocal row
-            title = "{}{}{}{}".format(
-                cp.name,
-                "^" * cp.base.total_planes,
-                "." * cp.base.total_armor,
-                "*" * cp.base.total_aa)
-            Button(self.frame, text=title, command=self.go_cp(cp)).grid(row=row, sticky=NW)
-            row += 1
-
-        Label(self.frame, text="Budget: {}m".format(self.game.budget)).grid(column=0, row=0, sticky=NW)
-        Button(self.frame, text="Pass turn", command=self.pass_turn).grid(column=1, row=0, sticky=NE)
-        row += 1
+        Button(self.frame, text="Pass turn", command=self.pass_turn).grid(column=0, row=0, sticky=NE)
+        Label(self.frame, text="Budget: {}m (+{}m)".format(self.game.budget, self.game.budget_reward_amount)).grid(column=0, row=0, sticky=NW)
+        Separator(self.frame, orient='horizontal').grid(row=row, sticky=EW); row += 1
 
         for event in self.game.events:
             if not event.informational:
                 continue
+
             label(str(event))
 
         for event in self.game.events:
@@ -59,21 +53,6 @@ class MainMenu(Menu):
 
             event_button(event, "{} {}".format(event.attacker.name != self.game.player and "!" or " ", event))
 
-        Separator(self.frame, orient='horizontal').grid(row=row, sticky=EW); row += 1
-        for cp in self.game.theater.player_points():
-            cp_button(cp)
-
-        Separator(self.frame, orient='horizontal').grid(row=row, sticky=EW); row += 1
-        for cp in self.game.theater.enemy_bases():
-            title = "[{}] {}{}{}{}".format(
-                int(cp.base.strength * 10),
-                cp.name,
-                "^" * cp.base.total_planes,
-                "." * cp.base.total_armor,
-                "*" * cp.base.total_aa)
-            Label(self.frame, text=title).grid(row=row, sticky=NE)
-            row += 1
-
     def pass_turn(self):
         self.game.pass_turn(no_action=True)
         self.display()
@@ -81,5 +60,10 @@ class MainMenu(Menu):
     def start_event(self, event) -> typing.Callable:
         return lambda: EventMenu(self.window, self, self.game, event).display()
 
-    def go_cp(self, cp: ControlPoint) -> typing.Callable:
-        return lambda: BaseMenu(self.window, self, self.game, cp).display()
+    def go_cp(self, cp: ControlPoint):
+        if self.basemenu:
+            self.basemenu.dismiss()
+            self.basemenu = None
+
+        self.basemenu = BaseMenu(self.window, self, self.game, cp)
+        self.basemenu.display()

ui/overviewcanvas.py

@@ -0,0 +1,88 @@
+from tkinter import *
+from tkinter.ttk import *
+
+from ui.window import *
+
+from game.game import *
+
+
+class OverviewCanvas:
+    mainmenu = None  # type: ui.mainmenu.MainMenu
+
+    def __init__(self, frame: Frame, parent, game: Game):
+        self.canvas = Canvas(frame, width=616, height=350)
+        self.canvas.grid(column=0, row=0, sticky=NSEW)
+        self.image = PhotoImage(file="resources/caumap.gif")
+        self.parent = parent
+
+        self.game = game
+
+    def cp_coordinates(self, cp: ControlPoint) -> (int, int):
+        point_a = (-317948.32727306, 635639.37385346)
+        point_a_img = 282.5, 319
+
+        point_b = (-355692.3067714, 617269.96285781)
+        point_b_img = 269, 352
+
+        x_dist = point_a_img[0] - point_b_img[0]
+        lon_dist = point_a[1] - point_b[1]
+
+        y_dist = point_a_img[1] - point_b_img[1]
+        lat_dist = point_b[0] - point_a[0]
+
+        x_scale = float(x_dist) / float(lon_dist)
+        y_scale = float(y_dist) / float(lat_dist)
+
+        # ---
+        x_offset = cp.position.x - point_a[0]
+        y_offset = cp.position.y - point_a[1]
+
+        return point_b_img[1] + y_offset * y_scale, point_a_img[0] - x_offset * x_scale
+
+    def create_cp_title(self, coords, cp: ControlPoint):
+        title = cp.name
+        font = ("Helvetica", 13)
+
+        self.canvas.create_text(coords[0]+1, coords[1]+1, text=title, fill='white', font=font)
+        self.canvas.create_text(coords[0], coords[1], text=title, font=font)
+
+    def update(self):
+        self.canvas.delete(ALL)
+        self.canvas.create_image((self.image.width()/2, self.image.height()/2), image=self.image)
+
+        for cp in self.game.theater.controlpoints:
+            coords = self.cp_coordinates(cp)
+            for connected_cp in cp.connected_points:
+                connected_coords = self.cp_coordinates(connected_cp)
+                if connected_cp.captured != cp.captured:
+                    color = "red"
+                elif connected_cp.captured and cp.captured:
+                    color = "blue"
+                else:
+                    color = "black"
+
+                self.canvas.create_line((coords[0], coords[1], connected_coords[0], connected_coords[1]), width=2, fill=color)
+
+        for cp in self.game.theater.controlpoints:
+            coords = self.cp_coordinates(cp)
+            arc_size = 18 * math.pow(cp.importance, 1)
+            extent = max(cp.base.strength * 180, 10)
+            start = (180 - extent) / 2
+            color = cp.captured and 'blue' or 'red'
+
+            cp_id = self.canvas.create_arc((coords[0] - arc_size/2, coords[1] - arc_size/2),
+                                           (coords[0]+arc_size/2, coords[1]+arc_size/2),
+                                           fill=color,
+                                           style=PIESLICE,
+                                           start=start,
+                                           extent=extent)
+            self.canvas.tag_bind(cp_id, "<Button-1>", self.display(cp))
+            self.create_cp_title((coords[0] + arc_size/2, coords[1] + arc_size/2), cp)
+            self.canvas.create_text(coords[0], coords[1] - arc_size / 1.5, text="8/4/2", font=("Helvetica", 10))
+
+    def display(self, cp: ControlPoint):
+        def action(_):
+            return self.parent.go_cp(cp)
+
+        return action
+