felix/photobooth
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Pose state moved to Frames

Browse Source
This commit is contained in:
Balthasar Reuter
2018-06-04 22:23:43 +02:00
parent b5a7ab2af3
commit b15c67c47e
4 changed files with 104 additions and 83 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
photobooth/gui/GuiState.py
View File

@@ -103,10 +103,25 @@ class GreeterState(GuiState):
class PoseState(GuiState):
def __init__(self, **kwargs):
def __init__(self, num_picture, **kwargs):
super().__init__(**kwargs)
self.num_picture = num_picture
@property
def num_picture(self):
return self._num_picture
@num_picture.setter
def num_picture(self, num_picture):
if not isinstance(num_picture, int):
raise ValueError('Picture number must be an integer')
self._num_picture = num_picture
class AssembleState(GuiState):

75
photobooth/gui/PyQt5Gui.py
View File

@@ -114,7 +114,8 @@ class PyQt5Gui(Gui):
elif isinstance(state, GreeterState):
global cfg
self._p.handleKeypressEvent = self.handleKeypressEventNoTrigger
self._p.setCentralWidget( PyQt5GreeterMessage(
# self._p.setCentralWidget( PyQt5GreeterMessage(
self._p.setCentralWidget( Frames.GreeterMessage(
cfg.getInt('Picture', 'num_x'), cfg.getInt('Picture', 'num_y') ) )
QtCore.QTimer.singleShot(cfg.getInt('Photobooth', 'greeter_time') * 1000, self.sendAck)
@@ -126,7 +127,9 @@ class PyQt5Gui(Gui):
self._p.centralWidget().update()
elif isinstance(state, PoseState):
self._p.setCentralWidget(PyQt5PoseMessage())
# self._p.setCentralWidget(PyQt5PoseMessage())
self._p.setCentralWidget(Frames.PoseMessage(state.num_picture,
cfg.getInt('Picture', 'num_x'), cfg.getInt('Picture', 'num_y')))
elif isinstance(state, AssembleState):
self._p.setCentralWidget(Frames.WaitMessage('Processing picture...'))
@@ -312,45 +315,6 @@ class PyQt5MainWindow(QtWidgets.QMainWindow):
class PyQt5GreeterMessage(QtWidgets.QFrame):
def __init__(self, num_x, num_y):
super().__init__()
self._num_x = num_x
self._num_y = num_y
self._title = 'Get ready!'
self._text = 'We will capture {} pictures!'.format(num_x * num_y)
self.initFrame()
def initFrame(self):
self.setStyleSheet('background-color: black; color: white;')
def paintEvent(self, event):
painter = QtGui.QPainter(self)
f = self.font()
f.setPixelSize(self.height() / 5)
painter.setFont(f)
rect = QtCore.QRect(0, self.height() * 1 / 5, self.width(), self.height() * 3 / 10)
painter.drawText(rect, QtCore.Qt.AlignCenter, self._title)
f.setPixelSize(self.height() / 8)
painter.setFont(f)
rect = QtCore.QRect(0, self.height() * 3 / 5, self.width(), self.height() * 3 / 10)
painter.drawText(rect, QtCore.Qt.AlignCenter, self._text)
painter.end()
class PyQt5CountdownMessage(QtWidgets.QFrame):
def __init__(self, time, action):
@@ -448,35 +412,6 @@ class PyQt5CountdownMessage(QtWidgets.QFrame):
self.update()
class PyQt5PoseMessage(QtWidgets.QFrame):
def __init__(self):
super().__init__()
self.initFrame()
def initFrame(self):
self.setStyleSheet('background-color: black; color: white;')
def paintEvent(self, event):
painter = QtGui.QPainter(self)
f = self.font()
f.setPixelSize(self.height() / 3)
painter.setFont(f)
painter.drawText(event.rect(), QtCore.Qt.AlignCenter, 'Pose!')
painter.end()
class PyQt5PictureMessage(QtWidgets.QFrame):
def __init__(self, picture):

71
photobooth/gui/Qt5Gui/Frames.py
View File

@@ -64,6 +64,77 @@ class IdleMessage(QtWidgets.QFrame):
painter.end()
class GreeterMessage(QtWidgets.QFrame):
def __init__(self, num_x, num_y):
super().__init__()
self._title = 'Get ready!'
if num_x * num_y > 1:
self._text = ('Capturing {} pictures...'.format(num_x * num_y))
else:
self._text = 'Starting the countdown...'
def _paintMessage(self, painter):
f = self.font()
f.setPixelSize(self.height() / 5)
painter.setFont(f)
rect = QtCore.QRect(0, self.height() * 1 / 5,
self.width(), self.height() * 3 / 10)
painter.drawText(rect, QtCore.Qt.AlignCenter, self._title)
f.setPixelSize(self.height() / 8)
painter.setFont(f)
rect = QtCore.QRect(0, self.height() * 3 / 5,
self.width(), self.height() * 3 / 10)
painter.drawText(rect, QtCore.Qt.AlignCenter, self._text)
def paintEvent(self, event):
painter = QtGui.QPainter(self)
self._paintMessage(painter)
painter.end()
class PoseMessage(QtWidgets.QFrame):
def __init__(self, num_picture, num_x, num_y):
super().__init__()
self._title = 'Pose!'
if num_x * num_y > 1:
self._text = 'Picture {} of {}...'.format(num_picture,
num_x * num_y)
else:
self._text = 'Taking a photo...'
def _paintMessage(self, painter):
f = self.font()
f.setPixelSize(self.height() / 5)
painter.setFont(f)
rect = QtCore.QRect(0, self.height() * 1 / 5,
self.width(), self.height() * 3 / 10)
painter.drawText(rect, QtCore.Qt.AlignCenter, self._title)
f.setPixelSize(self.height() / 8)
painter.setFont(f)
rect = QtCore.QRect(0, self.height() * 3 / 5,
self.width(), self.height() * 3 / 10)
painter.drawText(rect, QtCore.Qt.AlignCenter, self._text)
def paintEvent(self, event):
painter = QtGui.QPainter(self)
self._paintMessage(painter)
painter.end()
class WaitMessage(QtWidgets.QFrame):
# With spinning wait clock, inspired by
# https://wiki.python.org/moin/PyQt/A%20full%20widget%20waiting%20indicator