import os

from tkinter import *
from tkinter.ttk import *

from ui.window import *

from game.game import *
from gen.conflictgen import Conflict
from theater.conflicttheater import *


class OverviewCanvas:
    mainmenu = None  # type: ui.mainmenu.MainMenu

    def __init__(self, frame: Frame, parent, game: Game):
        self.parent = parent
        self.game = game

        self.image = PhotoImage(file=os.path.join("resources", game.theater.overview_image))
        self.canvas = Canvas(frame, width=self.image.width(), height=self.image.height())
        self.canvas.grid(column=0, row=0, sticky=NSEW)

    def transform_point(self, p: Point) -> (int, int):
        point_a = list(self.game.theater.reference_points.keys())[0]
        point_a_img = self.game.theater.reference_points[point_a]

        point_b = list(self.game.theater.reference_points.keys())[1]
        point_b_img = self.game.theater.reference_points[point_b]

        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 = p.x - point_a[0]
        y_offset = p.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", 10)

        id = self.canvas.create_text(coords[0]+1, coords[1]+1, text=title, fill='white', font=font)
        self.canvas.tag_bind(id, "<Button-1>", self.display(cp))
        id = self.canvas.create_text(coords[0], coords[1], text=title, font=font)
        self.canvas.tag_bind(id, "<Button-1>", self.display(cp))

    def _player_color(self):
        return self.game.player == "USA" and "blue" or "red"

    def _enemy_color(self):
        return self.game.player == "USA" and "red" or "blue"

    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.transform_point(cp.position)
            for connected_cp in cp.connected_points:
                connected_coords = self.transform_point(connected_cp.position)
                if connected_cp.captured != cp.captured:
                    color = self._enemy_color()
                elif connected_cp.captured and cp.captured:
                    color = self._player_color()
                else:
                    color = "black"

                self.canvas.create_line((coords[0], coords[1], connected_coords[0], connected_coords[1]), width=2, fill=color)

                if cp.captured and not connected_cp.captured and Conflict.has_frontline_between(cp, connected_cp):
                    frontline_pos, heading, distance = Conflict.frontline_vector(cp, connected_cp, self.game.theater)
                    start_coords, end_coords = self.transform_point(frontline_pos), self.transform_point(frontline_pos.point_from_heading(heading, distance))
                    self.canvas.create_line((*start_coords, *end_coords), width=2, fill=color)

        for cp in self.game.theater.controlpoints:
            coords = self.transform_point(cp.position)
            arc_size = 16 * math.pow(cp.importance, 1)
            extent = max(cp.base.strength * 180, 10)
            start = (180 - extent) / 2

            if cp.captured:
                color = self._player_color()
            else:
                color = self._enemy_color()

            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)

            """
            #For debugging purposes
            
            for r in cp.radials:
                p = self.transform_point(cp.position.point_from_heading(r, 20000))
                self.canvas.create_text(p[0], p[1], text="{}".format(r))
            continue
            """

            self.canvas.tag_bind(cp_id, "<Button-1>", self.display(cp))
            self.create_cp_title((coords[0] + arc_size/4, coords[1] + arc_size/4), cp)

            units_title = "{}/{}/{}".format(cp.base.total_planes, cp.base.total_armor, cp.base.total_aa)
            self.canvas.create_text(coords[0], coords[1] - arc_size / 1.5, text=units_title, font=("Helvetica", 10))

    def display(self, cp: ControlPoint):
        def action(_):
            return self.parent.go_cp(cp)

        return action