pypog/tools/gridviewer/GridViewerCell.py

'''

    ** By Cro-Ki l@b, 2017 **
'''
from PyQt5.QtCore import QPointF
from PyQt5.QtGui import QPolygonF, QPen, QBrush, QColor, QFont
from PyQt5.QtWidgets import QGraphicsPolygonItem, QGraphicsItem, \
    QGraphicsSimpleTextItem

class GridViewerCell(QGraphicsPolygonItem):

    def __init__(self, gridViewer, x, y):
        super(GridViewerCell, self).__init__()
        self.gridViewer = gridViewer
        self.x = x
        self.y = y
        self.selected = False

    def generate(self, points, show_label=False):

        qpolygon = QPolygonF([QPointF(xp, yp) for xp, yp in points])

        self.setPolygon(qpolygon)

        pen = QPen()
        pen.setWidth(3)
        self.setPen(pen)

        self.setFlag(QGraphicsItem.ItemIsFocusable)

        self.label = QGraphicsSimpleTextItem("{}-{}".format(self.x, self.y), parent=self)
        self.label.setVisible(show_label)

        if len(points) == 6:
            k = 0.5 if (self.x % 2) != 0 else 0
            self.label.setPos(QPointF(((self.x * 0.866) + 0.2886) * 120, (self.y + k + 0.5) * 120))
        else:
            self.label.setPos(QPointF(self.x * 120, self.y * 120))

        font = QFont()
        font.setPointSize(20)
        self.label.setFont(font)

    def select(self):
        self.setBrush(QBrush(QColor(200, 0, 0, 100)))
        self.selected = True
        self.gridViewer.add_to_selection(self.x, self.y)

    def unselect(self):
        self.setBrush(QBrush())
        self.selected = False
        self.gridViewer.remove_from_selection(self.x, self.y)

    def mousePressEvent(self, *args, **kwargs):
        if self.selected:
            self.unselect()
        else:
            self.select()

    def show_label(self, visible):
        self.label.setVisible(visible)