ui_offscreen_units.py

from collections import defaultdict

from PyQt5 import QtWidgets

from aoc_object_consts import GROUPS
from ui_icon_offscreen_unit import IconOffscreenUnit


class OverlayOffscreenUnits(QtWidgets.QWidget):
    def __init__(self, name, parent):
        super(OverlayOffscreenUnits, self).__init__(parent)
        self.parent = parent
        self.get_icons = lambda: {}
        # Sets the widget to be on the whole gamescreen. 
        self.icons = {}
        self.update_position()

    def update_position(self):
        self.setGeometry(0, 32, self.parent.width(), self.parent.height() - 32 - 175)

    def set_movable(self, *arg):
        # This widget is not movable!
        pass

    def update(self):
        for icon in self.icons:
            self.icons[icon].delete = True
        new_icons = self.get_icons()

        # Need to transform the screen into game tiles
        width = (self.width()) / 90  # Transform the width to the ratio. We got squares now
        height = (self.height()) / 45  # UI height contains a bar (32px) and bottom UI (175px)

        # print(width, height, width/height)
        c_w = width / 2
        c_h = height / 2
        # Some weird math. I have no theoretical idea how it works. But hey, it works!
        gx, gy = self.parent.game.screen_position
        c_a = gx + c_w + gy + c_w
        c_b = -gx - c_h + gy - c_h + 2
        c_c = gx - c_w + gy - c_w
        c_d = -gx + c_h + gy + c_h + 1
        for game_obj in new_icons:
            x, y = game_obj.position
            xy = x + y
            xy_diff = y - x
            if c_a > xy and c_b < xy_diff and c_c < xy and c_d > xy_diff:
                # print("ON SCREEN")
                pass
            else:
                group = defaultdict(str, GROUPS)[game_obj.group]
                if game_obj not in self.icons:
                    self.icons[game_obj] = IconOffscreenUnit(self, game_obj)
                    self.icons[game_obj].bottom_text = group
                    self.icons[game_obj].redraw()
                elif group != self.icons[game_obj].bottom_text:
                    self.icons[game_obj].bottom_text = group
                    self.icons[game_obj].redraw()

                # print("NOT ON SCREEN")
                tx, ty = x - gx, y - gy
                k = abs(tx / ty) / 2 if abs(tx) < abs(ty) else 1 - abs(ty / tx) / 2  # might divide by zero
                if tx < 0 and ty < 0:
                    # print("LEFT", k)
                    self.icons[game_obj].update_left(k)
                elif tx < 0 and ty > 0:
                    k = 1 - k
                    self.icons[game_obj].update_bottom(k)
                elif tx > 0 and ty > 0:
                    k = 1 - k
                    # print("RIGHT", k)
                    self.icons[game_obj].update_right(k)
                else:  # tx > 0
                    # print("TOP", k)
                    self.icons[game_obj].update_top(k)

                self.icons[game_obj].delete = False
        # stupid way how to get rid of not selcted units
        remove_list = []
        for icon in self.icons:
            if self.icons[icon].delete:
                remove_list += [icon]
        if remove_list:
            # Remove the stuff..
            for i in remove_list:
                self.icons[i].deleteLater()
                del self.icons[i]
        # print(self.icons)


if __name__ == '__main__':
    pass