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
|
#!/usr/bin/env python
#Author velociraptor Genjix <aphidia@hotmail.com>
from PySide.QtGui import *
from PySide.QtCore import *
class LightWidget(QWidget):
def __init__(self, colour):
super(LightWidget, self).__init__()
self.colour = colour
self.onVal = False
def isOn(self):
return self.onVal
def setOn(self, on):
if self.onVal == on:
return
self.onVal = on
self.update()
@Slot()
def turnOff(self):
self.setOn(False)
@Slot()
def turnOn(self):
self.setOn(True)
def paintEvent(self, e):
if not self.onVal:
return
painter = QPainter(self)
painter.setRenderHint(QPainter.Antialiasing)
painter.setBrush(self.colour)
painter.drawEllipse(0, 0, self.width(), self.height())
on = Property(bool, isOn, setOn)
class TrafficLightWidget(QWidget):
def __init__(self):
super(TrafficLightWidget, self).__init__()
vbox = QVBoxLayout(self)
self.redLight = LightWidget(Qt.red)
vbox.addWidget(self.redLight)
self.yellowLight = LightWidget(Qt.yellow)
vbox.addWidget(self.yellowLight)
self.greenLight = LightWidget(Qt.green)
vbox.addWidget(self.greenLight)
pal = QPalette()
pal.setColor(QPalette.Background, Qt.black)
self.setPalette(pal)
self.setAutoFillBackground(True)
def createLightState(light, duration, parent=None):
lightState = QState(parent)
timer = QTimer(lightState)
timer.setInterval(duration)
timer.setSingleShot(True)
timing = QState(lightState)
timing.entered.connect(light.turnOn)
timing.entered.connect(timer.start)
timing.exited.connect(light.turnOff)
done = QFinalState(lightState)
timing.addTransition(timer, SIGNAL('timeout()'), done)
lightState.setInitialState(timing)
return lightState
class TrafficLight(QWidget):
def __init__(self):
super(TrafficLight, self).__init__()
vbox = QVBoxLayout(self)
widget = TrafficLightWidget()
vbox.addWidget(widget)
vbox.setContentsMargins(0, 0, 0, 0)
machine = QStateMachine(self)
redGoingYellow = createLightState(widget.redLight, 1000)
redGoingYellow.setObjectName('redGoingYellow')
yellowGoingGreen = createLightState(widget.redLight, 1000)
yellowGoingGreen.setObjectName('redGoingYellow')
redGoingYellow.addTransition(redGoingYellow, SIGNAL('finished()'), yellowGoingGreen)
greenGoingYellow = createLightState(widget.yellowLight, 3000)
greenGoingYellow.setObjectName('redGoingYellow')
yellowGoingGreen.addTransition(yellowGoingGreen, SIGNAL('finished()'), greenGoingYellow)
yellowGoingRed = createLightState(widget.greenLight, 1000)
yellowGoingRed.setObjectName('redGoingYellow')
greenGoingYellow.addTransition(greenGoingYellow, SIGNAL('finished()'), yellowGoingRed)
yellowGoingRed.addTransition(yellowGoingRed, SIGNAL('finished()'), redGoingYellow)
machine.addState(redGoingYellow)
machine.addState(yellowGoingGreen)
machine.addState(greenGoingYellow)
machine.addState(yellowGoingRed)
machine.setInitialState(redGoingYellow)
machine.start()
if __name__ == '__main__':
import sys
app = QApplication(sys.argv)
widget = TrafficLight()
widget.resize(110, 300)
widget.show()
sys.exit(app.exec_())
|