HYTOS

# coding: utf-8

""" This is engineering stream no text item module """

import os.path

import sys

import copy

try:

    from PyQt5.QtCore import Qt, QPointF, QRectF, pyqtSignal, QT_VERSION_STR, QRect

    from PyQt5.QtGui import QImage, QPixmap, QPainterPath, QPainter, QPolygonF, QBrush, QPen, QTransform, QFont, QColor

    from PyQt5.QtWidgets import QGraphicsView, QGraphicsScene, QFileDialog, QGraphicsItem, QAbstractGraphicsShapeItem, \

        QGraphicsTextItem

except ImportError:

    try:

        from PyQt4.QtCore import Qt, QRectF, pyqtSignal, QT_VERSION_STR, QRect

        from PyQt4.QtGui import QGraphicsView, QGraphicsScene, QImage, QPixmap, QPainterPath, QFileDialog, QFont

    except ImportError:

        raise ImportError("ImageViewerQt: Requires PyQt5 or PyQt4.")

from AppDocData import AppDocData, MessageType

from EngineeringAbstractItem import QEngineeringAbstractItem

class QEngineeringStreamNoTextItem(QGraphicsTextItem):

    """ This is engineering stream no text item class """

    def __init__(self, text, parent=None):

        import uuid

        QGraphicsTextItem.__init__(self, text, parent)

        doc = self.document()

        option = doc.defaultTextOption()

        option.setAlignment(Qt.AlignCenter)

        doc.setDefaultTextOption(option)

        self.type = 'STREAM NO'

        font = QFont('Arial', 15)

        font.setPointSizeF(4)

        self.setFont(font)

    def highlight(self, flag):

        if flag:

            self.setDefaultTextColor(QColor(QEngineeringAbstractItem.HOVER_COLOR))

        else:

            self.setDefaultTextColor(Qt.black)

        self.update()

    def paint(self, painter, option, widget):

        """ override paint method """

        rect = self.boundingRect()

        r = rect.width()*0.5 if rect.width() > rect.height() else rect.height()*0.5

        painter.drawPolygon(self.create_poly(4, r, 0))

        super(QEngineeringStreamNoTextItem, self).paint(painter, option, widget)

    def create_poly(self, n, r, s):

        import math

        rect = self.boundingRect()

        polygon = QPolygonF()

        w = 360 / n  # angle per step

        for i in range(n):  # add the points of polygon

            t = w * i + s

            x = r * math.cos(math.radians(t))

            y = r * math.sin(math.radians(t))

            polygon.append(QPointF(rect.width() * 0.5 + x, rect.height() * 0.5 + y))

        return polygon

    def set_font_size(self, size):

        """set font size"""

        _font = self.font()

        _font.setPointSizeF(float(size))

        self.setFont(_font)

    def mouseDoubleClickEvent(self, event):

        """call parent's mouseDoubleClickEvent"""

        self.parentItem().mouseDoubleClickEvent(event)