from PySide2.QtCore import QPoint, QRect, QPointF
from PySide2.QtGui import QPainter
from PySide2.QtWidgets import QGraphicsRectItem

import qt_ui.uiconstants as CONST
from game import db
from theater import TheaterGroundObject, ControlPoint


class QMapGroundObject(QGraphicsRectItem):

    def __init__(self, parent, x: float, y: float, w: float, h: float, cp: ControlPoint, model: TheaterGroundObject):
        super(QMapGroundObject, self).__init__(x, y, w, h)
        self.model = model
        self.cp = cp
        self.parent = parent
        self.setAcceptHoverEvents(True)
        self.setZValue(2)

        if len(self.model.groups) > 0:
            units = {}
            for g in self.model.groups:
                for u in g.units:
                    if u.type in units.keys():
                        units[u.type] = units[u.type]+1
                    else:
                        units[u.type] = 1
            tooltip = ""
            for unit in units.keys():
                tooltip = tooltip + str(unit) + "x" + str(units[unit]) + "\n"
            self.setToolTip(tooltip[:-1])
        else:
            self.setToolTip(cp.name + "'s " + self.model.category)


    def paint(self, painter, option, widget=None):
        #super(QMapControlPoint, self).paint(painter, option, widget)
        if self.parent.get_display_rule("go"):
            painter.save()
            if not self.model.is_dead and not self.cp.captured:
                painter.drawPixmap(option.rect, CONST.ICONS[self.model.category])
            elif not self.model.is_dead and self.model.category == "aa":
                painter.drawPixmap(option.rect, CONST.ICONS["aa_blue"])
            else:
                painter.drawPixmap(option.rect, CONST.ICONS["cleared"])
            painter.restore()