EVA/gui/QDepthTape.py at master · PoliTOcean/EVA · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
from PyQt5.QtGui import QColor
from PyQt5.QtQml import QQmlComponent
from PyQt5.QtQuick import QQuickView, QQuickItem
from PyQt5.QtWidgets import QWidget
from PyQt5.QtCore import Qt, QUrl, pyqtProperty, pyqtSignal, pyqtSlot


class QDepthTape(QWidget):
    def __init__(self, maxDepth=5, parent=None) -> None:
        QWidget.__init__(self, parent)

        self.maxDepth = maxDepth
        self.tickList = []
        self.lastDepth = 0

        self.viewer = QQuickView()
        self.viewer.setSource(QUrl('resources/DepthTape.qml'))
        self.viewer.setResizeMode(QQuickView.SizeRootObjectToView)
        self.viewer.setColor(QColor(53, 53, 53))

        self.container = QWidget.createWindowContainer(self.viewer)
        self.container.setFixedWidth(70)
        self.container.setFocusPolicy(Qt.TabFocus)

        self.currentReadingComponent = QQmlComponent(
            self.viewer.engine(), QUrl("resources/DepthCurrentReading.qml"))

        self.currentReadingItem = self.currentReadingComponent.create()
        self.currentReadingItem.setParentItem(self.viewer.rootObject())

        self.__updateCurrentReadingPosition()
        self.currentReadingItem.setZ(5)

        for i in range(self.maxDepth + 1):
            tickComponent = QQmlComponent(
                self.viewer.engine(), QUrl("resources/DepthTick.qml"))

            tickItem = tickComponent.create()
            tickItem.setParentItem(self.viewer.rootObject())

            tickItem.text = str(i)
            tickItem.setX(15 - tickItem.property("textX"))
            tickItem.setY(i * 40 + self.viewer.height() /
                          2 - tickItem.height()/2)

            self.tickList.append(tickItem)

        self.viewer.heightChanged.connect(self.onResize)

        self.container.show()

    def show(self) -> None:
        return self.container.show()

    def onResize(self):
        self.__updateCurrentReadingPosition()
        self.__updateTicksPosition(self.lastDepth)

    @pyqtSlot(float)
    def updateDepth(self, depth: float):
        self.currentReadingItem.text = str(depth)
        self.__updateTicksPosition(depth)

        self.lastDepth = depth

    def __updateTicksPosition(self, depth: float):
        for i, tickItem in enumerate(self.tickList, start=0):
            tickItem.setX(15 - tickItem.property("textX"))
            tickItem.setY((i - depth) * 40 + self.viewer.height() /
                          2 - tickItem.height()/2)

    def __updateCurrentReadingPosition(self):
        self.currentReadingItem.setX(self.viewer.width() -
                                     self.currentReadingItem.width() + 2)
        self.currentReadingItem.setY(
            self.viewer.height()/2 - self.currentReadingItem.height()/2)


class QDepthTick(QQuickItem):
    textChanged = pyqtSignal(str)

    def __init__(self, parent=None):
        QQuickItem.__init__(self, parent)
        self._text = "0"

    @pyqtProperty(str, notify=textChanged)
    def text(self):
        return self._text

    @text.setter
    def text(self, text):
        self._text = text
        self.textChanged.emit(text)


class QDepthCurrentReading(QQuickItem):
    textChanged = pyqtSignal(str)

    def __init__(self, parent=None) -> None:
        QQuickItem.__init__(self, parent)

        self._text = "0.0m"

    @pyqtProperty(str, notify=textChanged)
    def text(self):
        return self._text

    @text.setter
    def text(self, text):
        self._text = text+"m"
        self.textChanged.emit(text)