def __init__(self, parent=None):
super(MainWindow, self).__init__()
# Set title
self.setWindowTitle('Cinema Desktop')
# Set up UI
self._mainWidget = QSplitter(Qt.Horizontal, self)
self.setCentralWidget(self._mainWidget)
self._displayWidget = QRenderView(self)
self._displayWidget.setRenderHints(QPainter.SmoothPixmapTransform)
self._displayWidget.setAlignment(Qt.AlignCenter)
self._displayWidget.setSizePolicy(QSizePolicy.Ignored,
QSizePolicy.Ignored)
self._parametersWidget = QWidget(self)
self._parametersWidget.setMinimumSize(QSize(200, 100))
self._parametersWidget.setSizePolicy(QSizePolicy.Preferred,
QSizePolicy.MinimumExpanding)
self._mainWidget.addWidget(self._displayWidget)
self._mainWidget.addWidget(self._parametersWidget)
layout = QVBoxLayout()
self._parametersWidget.setLayout(layout)
self.createMenus()
# Set up render view interactor
self._mouseInteractor = RenderViewMouseInteractor()
def __init__(self, parent=None):
super(MainWindow, self).__init__()
# Set title
self.setWindowTitle('Cinema Desktop')
# Set up UI
self._mainWidget = QSplitter(Qt.Horizontal, self)
self.setCentralWidget(self._mainWidget)
self._displayWidget = QRenderView(self)
self._displayWidget.setRenderHints(QPainter.SmoothPixmapTransform)
self._displayWidget.setAlignment(Qt.AlignCenter)
self._displayWidget.setSizePolicy(QSizePolicy.Ignored, QSizePolicy.Ignored)
self._parametersWidget = QWidget(self)
self._parametersWidget.setMinimumSize(QSize(200, 100))
self._parametersWidget.setSizePolicy(QSizePolicy.Preferred, QSizePolicy.MinimumExpanding)
self._mainWidget.addWidget(self._displayWidget)
self._mainWidget.addWidget(self._parametersWidget)
layout = QVBoxLayout()
self._parametersWidget.setLayout(layout)
self.createMenus()
# Set up render view interactor
self._mouseInteractor = RenderViewMouseInteractor()
class MainWindow(QMainWindow):
def __init__(self, parent=None):
super(MainWindow, self).__init__()
# Set title
self.setWindowTitle('Cinema Desktop')
# Set up UI
self._mainWidget = QSplitter(Qt.Horizontal, self)
self.setCentralWidget(self._mainWidget)
self._displayWidget = QRenderView(self)
self._displayWidget.setRenderHints(QPainter.SmoothPixmapTransform)
self._displayWidget.setAlignment(Qt.AlignCenter)
self._displayWidget.setSizePolicy(QSizePolicy.Ignored, QSizePolicy.Ignored)
self._parametersWidget = QWidget(self)
self._parametersWidget.setMinimumSize(QSize(200, 100))
self._parametersWidget.setSizePolicy(QSizePolicy.Preferred, QSizePolicy.MinimumExpanding)
self._mainWidget.addWidget(self._displayWidget)
self._mainWidget.addWidget(self._parametersWidget)
layout = QVBoxLayout()
self._parametersWidget.setLayout(layout)
self.createMenus()
# Set up render view interactor
self._mouseInteractor = RenderViewMouseInteractor()
# Create the menu bars
def createMenus(self):
# File menu
self._exitAction = QAction('E&xit', self, statusTip='Exit the application',
triggered=self.close)
self._fileToolBar = self.menuBar().addMenu('&File')
self._fileToolBar.addAction(self._exitAction)
# Set the store currently being displayed
def setStore(self, store):
self._store = store
self._initializeCurrentQuery()
# Disconnect all mouse signals in case the store has no phi or theta values
self._disconnectMouseSignals()
if ('phi' in store.parameter_list):
self._mouseInteractor.setPhiValues(store.parameter_list['phi']['values'])
if ('theta' in store.parameter_list):
self._mouseInteractor.setThetaValues(store.parameter_list['theta']['values'])
if ('phi' in store.parameter_list or 'theta' in store.parameter_list):
self._connectMouseSignals()
# Display the default image
doc = self._store.find(dict(self._currentQuery)).next()
self.displayDocument(doc)
self._createParameterUI()
# Disconnect mouse signals
def _disconnectMouseSignals(self):
try:
dw = self._displayWidget
dw.mousePressSignal.disconnect(self._initializeCamera)
dw.mousePressSignal.disconnect(self._mouseInteractor.onMousePress)
dw.mouseMoveSignal.disconnect(self._mouseInteractor.onMouseMove)
dw.mouseReleaseSignal.disconnect(self._mouseInteractor.onMouseRelease)
dw.mouseWheelSignal.disconnect(self._mouseInteractor.onMouseWheel)
# Update camera phi-theta if mouse is dragged
self._displayWidget.mouseMoveSignal.disconnect(self._updateCamera)
# Update camera if mouse wheel is moved
self._displayWidget.mouseWheelSignal.disconnect(self._updateCamera)
except:
# No big deal if we can't disconnect
pass
# Connect mouse signals
def _connectMouseSignals(self):
dw = self._displayWidget
dw.mousePressSignal.connect(self._initializeCamera)
dw.mousePressSignal.connect(self._mouseInteractor.onMousePress)
dw.mouseMoveSignal.connect(self._mouseInteractor.onMouseMove)
dw.mouseReleaseSignal.connect(self._mouseInteractor.onMouseRelease)
dw.mouseWheelSignal.connect(self._mouseInteractor.onMouseWheel)
# Update camera phi-theta if mouse is dragged
self._displayWidget.mouseMoveSignal.connect(self._updateCamera)
# Update camera if mouse wheel is moved
self._displayWidget.mouseWheelSignal.connect(self._updateCamera)
# Initializes image store query.
def _initializeCurrentQuery(self):
self._currentQuery = dict()
dd = self._store.parameter_list
for name, properties in dd.items():
self._currentQuery[name] = dd[name]['default']
# Create property UI
def _createParameterUI(self):
keys = sorted(self._store.parameter_list)
for name in keys:
properties = self._store.parameter_list[name]
if len(properties['values']) == 1:
#don't have widget if no choice possible
continue
labelValueWidget = QWidget(self)
labelValueWidget.setSizePolicy(QSizePolicy.MinimumExpanding, QSizePolicy.Fixed)
labelValueWidget.setLayout(QHBoxLayout())
labelValueWidget.layout().setContentsMargins(0, 0, 0, 0)
self._parametersWidget.layout().addWidget(labelValueWidget)
textLabel = QLabel(properties['label'], self)
labelValueWidget.layout().addWidget(textLabel)
valueLabel = QLabel('0', self)
valueLabel.setAlignment(Qt.AlignRight)
valueLabel.setObjectName(name + "ValueLabel")
labelValueWidget.layout().addWidget(valueLabel)
sliderControlsWidget = QWidget(self)
sliderControlsWidget.setSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.Fixed)
sliderControlsWidget.setLayout(QHBoxLayout())
sliderControlsWidget.layout().setContentsMargins(0, 0, 0, 0)
#sliderControlsWidget.setContentsMargins(0, 0, 0, 0)
self._parametersWidget.layout().addWidget(sliderControlsWidget)
flat = False
width = 25
skipBackwardIcon = self.style().standardIcon(QStyle.SP_MediaSkipBackward)
skipBackwardButton = QPushButton(skipBackwardIcon, '', self)
skipBackwardButton.setObjectName("SkipBackwardButton." + name)
skipBackwardButton.setFlat(flat)
skipBackwardButton.setMaximumWidth(width)
skipBackwardButton.clicked.connect(self.onSkipBackward)
sliderControlsWidget.layout().addWidget(skipBackwardButton)
seekBackwardIcon = self.style().standardIcon(QStyle.SP_MediaSeekBackward)
seekBackwardButton = QPushButton(seekBackwardIcon, '', self)
seekBackwardButton.setObjectName("SeekBackwardButton." + name)
seekBackwardButton.setFlat(flat)
seekBackwardButton.setMaximumWidth(width)
seekBackwardButton.clicked.connect(self.onSeekBackward)
sliderControlsWidget.layout().addWidget(seekBackwardButton)
slider = QSlider(Qt.Horizontal, self)
slider.setObjectName(name)
sliderControlsWidget.layout().addWidget(slider);
seekForwardIcon = self.style().standardIcon(QStyle.SP_MediaSeekForward)
seekForwardButton = QPushButton(seekForwardIcon, '', self)
seekForwardButton.setObjectName("SeekForwardButton." + name)
seekForwardButton.setFlat(flat)
seekForwardButton.setMaximumWidth(width)
seekForwardButton.clicked.connect(self.onSeekForward)
sliderControlsWidget.layout().addWidget(seekForwardButton)
skipForwardIcon = self.style().standardIcon(QStyle.SP_MediaSkipForward)
skipForwardButton = QPushButton(skipForwardIcon, '', self)
skipForwardButton.setObjectName("SkipForwardButton." + name)
skipForwardButton.setFlat(flat)
skipForwardButton.setMaximumWidth(width)
skipForwardButton.clicked.connect(self.onSkipForward)
sliderControlsWidget.layout().addWidget(skipForwardButton)
playIcon = self.style().standardIcon(QStyle.SP_MediaPlay)
playButton = QPushButton(playIcon, '', self)
playButton.setObjectName("PlayButton." + name)
playButton.setFlat(flat)
playButton.setMaximumWidth(width)
playButton.clicked.connect(self.onPlay)
sliderControlsWidget.layout().addWidget(playButton)
# Configure the slider
self.configureSlider(slider, properties)
self._updateSlider(properties['label'], properties['default'])
self._parametersWidget.layout().addStretch()
# Convenience function for setting up a slider
def configureSlider(self, slider, properties):
default = properties['default']
values = properties['values']
typeValue = properties['type']
label = properties['label']
slider.setMinimum(0)
slider.setMaximum(len(values)-1)
slider.setPageStep(1)
slider.valueChanged.connect(self.onSliderMoved)
# Respond to a slider movement
def onSliderMoved(self):
parameterName = self.sender().objectName()
sliderIndex = self.sender().value()
pl = self._store.parameter_list
parameterValue = pl[parameterName]['values'][sliderIndex]
self._currentQuery[parameterName] = parameterValue
# Update value label
valueLabel = self._parametersWidget.findChild(QLabel, parameterName + "ValueLabel")
valueLabel.setText(self._formatText(parameterValue))
self.render()
# Back up slider all the way to the left
def onSkipBackward(self):
parameterName = self.sender().objectName().replace("SkipBackwardButton.", "")
slider = self._parametersWidget.findChild(QSlider, parameterName)
slider.setValue(0)
# Back up slider one step to the left
def onSeekBackward(self):
parameterName = self.sender().objectName().replace("SeekBackwardButton.", "")
slider = self._parametersWidget.findChild(QSlider, parameterName)
slider.setValue(0 if slider.value() == 0 else slider.value() - 1)
# Forward slider one step to the right
def onSeekForward(self):
parameterName = self.sender().objectName().replace("SeekForwardButton.", "")
slider = self._parametersWidget.findChild(QSlider, parameterName)
maximum = slider.maximum()
slider.setValue(maximum if slider.value() == maximum else slider.value() + 1)
# Forward the slider all the way to the right
def onSkipForward(self):
parameterName = self.sender().objectName().replace("SkipForwardButton.", "")
slider = self._parametersWidget.findChild(QSlider, parameterName)
slider.setValue(slider.maximum())
# Play forward through the parameters
def onPlay(self):
parameterName = self.sender().objectName().replace("PlayButton.", "")
timer = QTimer(self)
timer.setObjectName("Timer." + parameterName)
timer.setInterval(200)
timer.timeout.connect(self.onPlayTimer)
timer.start()
def onPlayTimer(self):
parameterName = self.sender().objectName().replace("Timer.", "")
slider = self._parametersWidget.findChild(QSlider, parameterName)
maximum = slider.maximum()
if (slider.value() == slider.maximum()):
self.sender().stop()
else:
slider.setValue(maximum if slider.value() == maximum else slider.value() + 1)
# Format string from number
def _formatText(self, value):
try:
intValue = int(value)
return '{0}'.format(intValue)
except:
pass
try:
floatValue = float(value)
return '{0}'.format(floatValue)
except:
pass
# String
return value
# Update slider from value
def _updateSlider(self, parameterName, value):
pl = self._store.parameter_list
index = pl[parameterName]['values'].index(value)
slider = self._parametersWidget.findChild(QSlider, parameterName)
slider.setValue(index)
# Initialize the angles for the camera
def _initializeCamera(self):
self._mouseInteractor.setPhi(self._currentQuery['phi'])
self._mouseInteractor.setTheta(self._currentQuery['theta'])
# Update the camera angle
def _updateCamera(self):
# Set the camera settings if available
phi = self._mouseInteractor.getPhi()
theta = self._mouseInteractor.getTheta()
if ('phi' in self._currentQuery):
self._currentQuery['phi'] = phi
if ('theta' in self._currentQuery):
self._currentQuery['theta'] = theta
# Update the sliders for phi and theta
self._updateSlider('phi', phi)
self._updateSlider('theta', theta)
scale = self._mouseInteractor.getScale()
self._displayWidget.resetTransform()
self._displayWidget.scale(scale, scale)
self.render()
# Query the image store and display the retrieved image
def render(self):
# Retrieve image from data store with the current query. Only
# care about the first - there should be only one if we have
# correctly specified all the properties.
docs = [doc for doc in self._store.find(self._currentQuery)]
if (len(docs) > 0):
self.displayDocument(docs[0])
else:
self._displayWidget.setPixmap(None)
self._displayWidget.setAlignment(Qt.AlignCenter)
# Get the main widget
def mainWidget(self):
return self._mainWidget
# Given a document, read the data into an image that can be displayed in Qt
def displayDocument(self, doc):
#load it into PIL
imageparser = PIL.ImageFile.Parser()
imageparser.feed(doc.data)
pimg = imageparser.close()
imageString = pimg.convert('RGBA').tostring('raw', 'BGRA')
qimg = QImage(imageString, pimg.size[0], pimg.size[1], QImage.Format_ARGB32)
pix = QPixmap.fromImage(qimg)
# Try to resize the display widget
self._displayWidget.sizeHint = pix.size
self._displayWidget.setPixmap(pix)
class MainWindow(QMainWindow):
def __init__(self, parent=None):
super(MainWindow, self).__init__()
# Set title
self.setWindowTitle('Cinema Desktop')
# Set up UI
self._mainWidget = QSplitter(Qt.Horizontal, self)
self.setCentralWidget(self._mainWidget)
self._displayWidget = QRenderView(self)
self._displayWidget.setRenderHints(QPainter.SmoothPixmapTransform)
self._displayWidget.setAlignment(Qt.AlignCenter)
self._displayWidget.setSizePolicy(QSizePolicy.Ignored,
QSizePolicy.Ignored)
self._parametersWidget = QWidget(self)
self._parametersWidget.setMinimumSize(QSize(200, 100))
self._parametersWidget.setSizePolicy(QSizePolicy.Preferred,
QSizePolicy.MinimumExpanding)
self._mainWidget.addWidget(self._displayWidget)
self._mainWidget.addWidget(self._parametersWidget)
layout = QVBoxLayout()
self._parametersWidget.setLayout(layout)
self.createMenus()
# Set up render view interactor
self._mouseInteractor = RenderViewMouseInteractor()
# Create the menu bars
def createMenus(self):
# File menu
self._exitAction = QAction('E&xit',
self,
statusTip='Exit the application',
triggered=self.close)
self._fileToolBar = self.menuBar().addMenu('&File')
self._fileToolBar.addAction(self._exitAction)
# Set the store currently being displayed
def setStore(self, store):
self._store = store
self._initializeCurrentQuery()
# Disconnect all mouse signals in case the store has no phi or theta values
self._disconnectMouseSignals()
if ('phi' in store.parameter_list):
self._mouseInteractor.setPhiValues(
store.parameter_list['phi']['values'])
if ('theta' in store.parameter_list):
self._mouseInteractor.setThetaValues(
store.parameter_list['theta']['values'])
if ('phi' in store.parameter_list or 'theta' in store.parameter_list):
self._connectMouseSignals()
# Display the default image
doc = self._store.find(dict(self._currentQuery)).next()
self.displayDocument(doc)
self._createParameterUI()
# Disconnect mouse signals
def _disconnectMouseSignals(self):
try:
dw = self._displayWidget
dw.mousePressSignal.disconnect(self._initializeCamera)
dw.mousePressSignal.disconnect(self._mouseInteractor.onMousePress)
dw.mouseMoveSignal.disconnect(self._mouseInteractor.onMouseMove)
dw.mouseReleaseSignal.disconnect(
self._mouseInteractor.onMouseRelease)
dw.mouseWheelSignal.disconnect(self._mouseInteractor.onMouseWheel)
# Update camera phi-theta if mouse is dragged
self._displayWidget.mouseMoveSignal.disconnect(self._updateCamera)
# Update camera if mouse wheel is moved
self._displayWidget.mouseWheelSignal.disconnect(self._updateCamera)
except:
# No big deal if we can't disconnect
pass
# Connect mouse signals
def _connectMouseSignals(self):
dw = self._displayWidget
dw.mousePressSignal.connect(self._initializeCamera)
dw.mousePressSignal.connect(self._mouseInteractor.onMousePress)
dw.mouseMoveSignal.connect(self._mouseInteractor.onMouseMove)
dw.mouseReleaseSignal.connect(self._mouseInteractor.onMouseRelease)
dw.mouseWheelSignal.connect(self._mouseInteractor.onMouseWheel)
# Update camera phi-theta if mouse is dragged
self._displayWidget.mouseMoveSignal.connect(self._updateCamera)
# Update camera if mouse wheel is moved
self._displayWidget.mouseWheelSignal.connect(self._updateCamera)
# Initializes image store query.
def _initializeCurrentQuery(self):
self._currentQuery = dict()
dd = self._store.parameter_list
for name, properties in dd.items():
self._currentQuery[name] = dd[name]['default']
# Create property UI
def _createParameterUI(self):
keys = sorted(self._store.parameter_list)
for name in keys:
properties = self._store.parameter_list[name]
if len(properties['values']) == 1:
#don't have widget if no choice possible
continue
labelValueWidget = QWidget(self)
labelValueWidget.setSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.Fixed)
labelValueWidget.setLayout(QHBoxLayout())
labelValueWidget.layout().setContentsMargins(0, 0, 0, 0)
self._parametersWidget.layout().addWidget(labelValueWidget)
textLabel = QLabel(properties['label'], self)
labelValueWidget.layout().addWidget(textLabel)
valueLabel = QLabel('0', self)
valueLabel.setAlignment(Qt.AlignRight)
valueLabel.setObjectName(name + "ValueLabel")
labelValueWidget.layout().addWidget(valueLabel)
sliderControlsWidget = QWidget(self)
sliderControlsWidget.setSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.Fixed)
sliderControlsWidget.setLayout(QHBoxLayout())
sliderControlsWidget.layout().setContentsMargins(0, 0, 0, 0)
#sliderControlsWidget.setContentsMargins(0, 0, 0, 0)
self._parametersWidget.layout().addWidget(sliderControlsWidget)
flat = False
width = 25
skipBackwardIcon = self.style().standardIcon(
QStyle.SP_MediaSkipBackward)
skipBackwardButton = QPushButton(skipBackwardIcon, '', self)
skipBackwardButton.setObjectName("SkipBackwardButton." + name)
skipBackwardButton.setFlat(flat)
skipBackwardButton.setMaximumWidth(width)
skipBackwardButton.clicked.connect(self.onSkipBackward)
sliderControlsWidget.layout().addWidget(skipBackwardButton)
seekBackwardIcon = self.style().standardIcon(
QStyle.SP_MediaSeekBackward)
seekBackwardButton = QPushButton(seekBackwardIcon, '', self)
seekBackwardButton.setObjectName("SeekBackwardButton." + name)
seekBackwardButton.setFlat(flat)
seekBackwardButton.setMaximumWidth(width)
seekBackwardButton.clicked.connect(self.onSeekBackward)
sliderControlsWidget.layout().addWidget(seekBackwardButton)
slider = QSlider(Qt.Horizontal, self)
slider.setObjectName(name)
sliderControlsWidget.layout().addWidget(slider)
seekForwardIcon = self.style().standardIcon(
QStyle.SP_MediaSeekForward)
seekForwardButton = QPushButton(seekForwardIcon, '', self)
seekForwardButton.setObjectName("SeekForwardButton." + name)
seekForwardButton.setFlat(flat)
seekForwardButton.setMaximumWidth(width)
seekForwardButton.clicked.connect(self.onSeekForward)
sliderControlsWidget.layout().addWidget(seekForwardButton)
skipForwardIcon = self.style().standardIcon(
QStyle.SP_MediaSkipForward)
skipForwardButton = QPushButton(skipForwardIcon, '', self)
skipForwardButton.setObjectName("SkipForwardButton." + name)
skipForwardButton.setFlat(flat)
skipForwardButton.setMaximumWidth(width)
skipForwardButton.clicked.connect(self.onSkipForward)
sliderControlsWidget.layout().addWidget(skipForwardButton)
playIcon = self.style().standardIcon(QStyle.SP_MediaPlay)
playButton = QPushButton(playIcon, '', self)
playButton.setObjectName("PlayButton." + name)
playButton.setFlat(flat)
playButton.setMaximumWidth(width)
playButton.clicked.connect(self.onPlay)
sliderControlsWidget.layout().addWidget(playButton)
# Configure the slider
self.configureSlider(slider, properties)
self._updateSlider(properties['label'], properties['default'])
self._parametersWidget.layout().addStretch()
# Convenience function for setting up a slider
def configureSlider(self, slider, properties):
default = properties['default']
values = properties['values']
typeValue = properties['type']
label = properties['label']
slider.setMinimum(0)
slider.setMaximum(len(values) - 1)
slider.setPageStep(1)
slider.valueChanged.connect(self.onSliderMoved)
# Respond to a slider movement
def onSliderMoved(self):
parameterName = self.sender().objectName()
sliderIndex = self.sender().value()
pl = self._store.parameter_list
parameterValue = pl[parameterName]['values'][sliderIndex]
self._currentQuery[parameterName] = parameterValue
# Update value label
valueLabel = self._parametersWidget.findChild(
QLabel, parameterName + "ValueLabel")
valueLabel.setText(self._formatText(parameterValue))
self.render()
# Back up slider all the way to the left
def onSkipBackward(self):
parameterName = self.sender().objectName().replace(
"SkipBackwardButton.", "")
slider = self._parametersWidget.findChild(QSlider, parameterName)
slider.setValue(0)
# Back up slider one step to the left
def onSeekBackward(self):
parameterName = self.sender().objectName().replace(
"SeekBackwardButton.", "")
slider = self._parametersWidget.findChild(QSlider, parameterName)
slider.setValue(0 if slider.value() == 0 else slider.value() - 1)
# Forward slider one step to the right
def onSeekForward(self):
parameterName = self.sender().objectName().replace(
"SeekForwardButton.", "")
slider = self._parametersWidget.findChild(QSlider, parameterName)
maximum = slider.maximum()
slider.setValue(maximum if slider.value() ==
maximum else slider.value() + 1)
# Forward the slider all the way to the right
def onSkipForward(self):
parameterName = self.sender().objectName().replace(
"SkipForwardButton.", "")
slider = self._parametersWidget.findChild(QSlider, parameterName)
slider.setValue(slider.maximum())
# Play forward through the parameters
def onPlay(self):
parameterName = self.sender().objectName().replace("PlayButton.", "")
timer = QTimer(self)
timer.setObjectName("Timer." + parameterName)
timer.setInterval(200)
timer.timeout.connect(self.onPlayTimer)
timer.start()
def onPlayTimer(self):
parameterName = self.sender().objectName().replace("Timer.", "")
slider = self._parametersWidget.findChild(QSlider, parameterName)
maximum = slider.maximum()
if (slider.value() == slider.maximum()):
self.sender().stop()
else:
slider.setValue(maximum if slider.value() ==
maximum else slider.value() + 1)
# Format string from number
def _formatText(self, value):
try:
intValue = int(value)
return '{0}'.format(intValue)
except:
pass
try:
floatValue = float(value)
return '{0}'.format(floatValue)
except:
pass
# String
return value
# Update slider from value
def _updateSlider(self, parameterName, value):
pl = self._store.parameter_list
index = pl[parameterName]['values'].index(value)
slider = self._parametersWidget.findChild(QSlider, parameterName)
slider.setValue(index)
# Initialize the angles for the camera
def _initializeCamera(self):
self._mouseInteractor.setPhi(self._currentQuery['phi'])
self._mouseInteractor.setTheta(self._currentQuery['theta'])
# Update the camera angle
def _updateCamera(self):
# Set the camera settings if available
phi = self._mouseInteractor.getPhi()
theta = self._mouseInteractor.getTheta()
if ('phi' in self._currentQuery):
self._currentQuery['phi'] = phi
if ('theta' in self._currentQuery):
self._currentQuery['theta'] = theta
# Update the sliders for phi and theta
self._updateSlider('phi', phi)
self._updateSlider('theta', theta)
scale = self._mouseInteractor.getScale()
self._displayWidget.resetTransform()
self._displayWidget.scale(scale, scale)
self.render()
# Query the image store and display the retrieved image
def render(self):
# Retrieve image from data store with the current query. Only
# care about the first - there should be only one if we have
# correctly specified all the properties.
docs = [doc for doc in self._store.find(self._currentQuery)]
if (len(docs) > 0):
self.displayDocument(docs[0])
else:
self._displayWidget.setPixmap(None)
self._displayWidget.setAlignment(Qt.AlignCenter)
# Get the main widget
def mainWidget(self):
return self._mainWidget
# Given a document, read the data into an image that can be displayed in Qt
def displayDocument(self, doc):
#load it into PIL
imageparser = PIL.ImageFile.Parser()
imageparser.feed(doc.data)
pimg = imageparser.close()
imageString = pimg.convert('RGBA').tostring('raw', 'BGRA')
qimg = QImage(imageString, pimg.size[0], pimg.size[1],
QImage.Format_ARGB32)
pix = QPixmap.fromImage(qimg)
# Try to resize the display widget
self._displayWidget.sizeHint = pix.size
self._displayWidget.setPixmap(pix)
class MainWindow(QMainWindow):
def __init__(self, parent=None):
super(MainWindow, self).__init__()
# Set title
self.setWindowTitle('Cinema Desktop')
# Set up UI
self._mainWidget = QSplitter(Qt.Horizontal, self)
self.setCentralWidget(self._mainWidget)
self._displayWidget = QRenderView(self)
self._displayWidget.setRenderHints(QPainter.SmoothPixmapTransform)
self._displayWidget.setAlignment(Qt.AlignCenter)
self._displayWidget.setSizePolicy(QSizePolicy.Ignored, QSizePolicy.Ignored)
self._parametersWidget = QWidget(self)
self._parametersWidget.setMinimumSize(QSize(200, 100))
self._parametersWidget.setSizePolicy(QSizePolicy.Preferred, QSizePolicy.MinimumExpanding)
self._mainWidget.addWidget(self._displayWidget)
self._mainWidget.addWidget(self._parametersWidget)
layout = QVBoxLayout()
self._parametersWidget.setLayout(layout)
#keep track of widgets that depend on others for easy updating
self._dependent_widgets = {}
self.createMenus()
# Set up render view interactor
self._mouseInteractor = RenderViewMouseInteractor()
# Create the menu bars
def createMenus(self):
# File menu
self._exitAction = QAction('E&xit', self, statusTip='Exit the application',
triggered=self.close)
self._fileToolBar = self.menuBar().addMenu('&File')
self._fileToolBar.addAction(self._exitAction)
# Set the store currently being displayed
def setStore(self, store):
self._store = store
self._initializeCurrentQuery()
# Disconnect all mouse signals in case the store has no phi or theta values
self._disconnectMouseSignals()
if ('phi' in store.parameter_list):
self._mouseInteractor.setPhiValues(store.parameter_list['phi']['values'])
if ('theta' in store.parameter_list):
self._mouseInteractor.setThetaValues(store.parameter_list['theta']['values'])
if ('phi' in store.parameter_list or 'theta' in store.parameter_list):
self._connectMouseSignals()
# Display the default image
self.render()
# Make the GUI
self._createParameterUI()
# Disconnect mouse signals
def _disconnectMouseSignals(self):
try:
dw = self._displayWidget
dw.mousePressSignal.disconnect(self._initializeCamera)
dw.mousePressSignal.disconnect(self._mouseInteractor.onMousePress)
dw.mouseMoveSignal.disconnect(self._mouseInteractor.onMouseMove)
dw.mouseReleaseSignal.disconnect(self._mouseInteractor.onMouseRelease)
dw.mouseWheelSignal.disconnect(self._mouseInteractor.onMouseWheel)
# Update camera phi-theta if mouse is dragged
self._displayWidget.mouseMoveSignal.disconnect(self._updateCamera)
# Update camera if mouse wheel is moved
self._displayWidget.mouseWheelSignal.disconnect(self._updateCamera)
except:
# No big deal if we can't disconnect
pass
# Connect mouse signals
def _connectMouseSignals(self):
dw = self._displayWidget
dw.mousePressSignal.connect(self._initializeCamera)
dw.mousePressSignal.connect(self._mouseInteractor.onMousePress)
dw.mouseMoveSignal.connect(self._mouseInteractor.onMouseMove)
dw.mouseReleaseSignal.connect(self._mouseInteractor.onMouseRelease)
dw.mouseWheelSignal.connect(self._mouseInteractor.onMouseWheel)
# Update camera phi-theta if mouse is dragged
self._displayWidget.mouseMoveSignal.connect(self._updateCamera)
# Update camera if mouse wheel is moved
self._displayWidget.mouseWheelSignal.connect(self._updateCamera)
# Initializes image store query.
def _initializeCurrentQuery(self):
self._currentQuery = dict()
dd = self._store.parameter_list
for name, ignored in dd.items():
value = dd[name]['default']
s = set()
s.add(value)
self._currentQuery[name] = s
# Create a slider for a 'range' parameter
def _createRangeSlider(self, name, properties):
labelValueWidget = QWidget(self)
labelValueWidget.setSizePolicy(QSizePolicy.MinimumExpanding, QSizePolicy.Fixed)
labelValueWidget.setLayout(QHBoxLayout())
labelValueWidget.layout().setContentsMargins(0, 0, 0, 0)
self._parametersWidget.layout().addWidget(labelValueWidget)
textLabel = QLabel(properties['label'], self)
labelValueWidget.layout().addWidget(textLabel)
valueLabel = QLabel('0', self)
valueLabel.setAlignment(Qt.AlignRight)
valueLabel.setObjectName(name + "ValueLabel")
labelValueWidget.layout().addWidget(valueLabel)
controlsWidget = QWidget(self)
controlsWidget.setSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.Fixed)
controlsWidget.setLayout(QHBoxLayout())
controlsWidget.layout().setContentsMargins(0, 0, 0, 0)
#controlsWidget.setContentsMargins(0, 0, 0, 0)
self._parametersWidget.layout().addWidget(controlsWidget)
flat = False
width = 25
skipBackwardIcon = self.style().standardIcon(QStyle.SP_MediaSkipBackward)
skipBackwardButton = QPushButton(skipBackwardIcon, '', self)
skipBackwardButton.setObjectName("SkipBackwardButton." + name)
skipBackwardButton.setFlat(flat)
skipBackwardButton.setMaximumWidth(width)
skipBackwardButton.clicked.connect(self.onSkipBackward)
controlsWidget.layout().addWidget(skipBackwardButton)
seekBackwardIcon = self.style().standardIcon(QStyle.SP_MediaSeekBackward)
seekBackwardButton = QPushButton(seekBackwardIcon, '', self)
seekBackwardButton.setObjectName("SeekBackwardButton." + name)
seekBackwardButton.setFlat(flat)
seekBackwardButton.setMaximumWidth(width)
seekBackwardButton.clicked.connect(self.onSeekBackward)
controlsWidget.layout().addWidget(seekBackwardButton)
slider = QSlider(Qt.Horizontal, self)
slider.setObjectName(name)
controlsWidget.layout().addWidget(slider);
seekForwardIcon = self.style().standardIcon(QStyle.SP_MediaSeekForward)
seekForwardButton = QPushButton(seekForwardIcon, '', self)
seekForwardButton.setObjectName("SeekForwardButton." + name)
seekForwardButton.setFlat(flat)
seekForwardButton.setMaximumWidth(width)
seekForwardButton.clicked.connect(self.onSeekForward)
controlsWidget.layout().addWidget(seekForwardButton)
skipForwardIcon = self.style().standardIcon(QStyle.SP_MediaSkipForward)
skipForwardButton = QPushButton(skipForwardIcon, '', self)
skipForwardButton.setObjectName("SkipForwardButton." + name)
skipForwardButton.setFlat(flat)
skipForwardButton.setMaximumWidth(width)
skipForwardButton.clicked.connect(self.onSkipForward)
controlsWidget.layout().addWidget(skipForwardButton)
playIcon = self.style().standardIcon(QStyle.SP_MediaPlay)
playButton = QPushButton(playIcon, '', self)
playButton.setObjectName("PlayButton." + name)
playButton.setFlat(flat)
playButton.setMaximumWidth(width)
playButton.clicked.connect(self.onPlay)
controlsWidget.layout().addWidget(playButton)
# Configure the slider
default = properties['default']
values = [self._formatText(x) for x in properties['values']]
typeValue = properties['type']
label = properties['label']
slider.setMinimum(0)
slider.setMaximum(len(values)-1)
slider.setPageStep(1)
slider.valueChanged.connect(self.onSliderMoved)
self._updateSlider(properties['label'], properties['default'])
return controlsWidget
# Create a slider for a 'list' parameter
def _createListPulldown(self, name, properties):
labelValueWidget = QWidget(self)
labelValueWidget.setSizePolicy(QSizePolicy.MinimumExpanding, QSizePolicy.Fixed)
labelValueWidget.setLayout(QHBoxLayout())
labelValueWidget.layout().setContentsMargins(0, 0, 0, 0)
self._parametersWidget.layout().addWidget(labelValueWidget)
textLabel = QLabel(properties['label'], self)
labelValueWidget.layout().addWidget(textLabel)
controlsWidget = QWidget(self)
controlsWidget.setSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.Fixed)
controlsWidget.setLayout(QHBoxLayout())
controlsWidget.layout().setContentsMargins(0, 0, 0, 0)
#controlsWidget.setContentsMargins(0, 0, 0, 0)
self._parametersWidget.layout().addWidget(controlsWidget)
menu = QComboBox(self)
menu.setObjectName(name)
controlsWidget.layout().addWidget(menu);
found = -1
for idx in range(0,len(properties['values'])):
entry = properties['values'][idx]
if entry == properties['default']:
found = menu.count()
#skip depth images, which we don't render directly
if self._store.determine_type({name: entry}) == 'Z':
pass
else:
menu.addItem(str(entry))
menu.setCurrentIndex(found)
menu.currentIndexChanged.connect(self.onChosen)
return controlsWidget
# Create a slider for an 'option' parameter
def _createOptionCheckbox(self, name, properties):
labelValueWidget = QWidget(self)
labelValueWidget.setSizePolicy(QSizePolicy.MinimumExpanding, QSizePolicy.Fixed)
labelValueWidget.setLayout(QHBoxLayout())
labelValueWidget.layout().setContentsMargins(0, 0, 0, 0)
self._parametersWidget.layout().addWidget(labelValueWidget)
textLabel = QLabel(properties['label'], self)
labelValueWidget.layout().addWidget(textLabel)
controlsWidget = QWidget(self)
controlsWidget.setSizePolicy(QSizePolicy.MinimumExpanding,
QSizePolicy.Fixed)
controlsWidget.setLayout(QHBoxLayout())
controlsWidget.layout().setContentsMargins(0, 0, 0, 0)
#controlsWidget.setContentsMargins(0, 0, 0, 0)
self._parametersWidget.layout().addWidget(controlsWidget)
for entry in properties['values']:
cb = QCheckBox(str(entry), self)
cb.setObjectName(name)
cb.value = entry
cb.setText(self._formatText(entry))
if entry == properties['default']:
cb.setChecked(True)
cb.stateChanged.connect(self.onChecked)
controlsWidget.layout().addWidget(cb)
return controlsWidget
# Create property UI
def _createParameterUI(self):
keys = sorted(self._store.parameter_list)
dependencies = self._store.parameter_associations
#reorder for clarity
#these three are the most important
keys2 = []
for special in ['time','phi','theta']:
if special in keys:
keys.remove(special)
keys2.append(special)
#any other globals
for name in keys:
if (not self._store.isdepender(name) and not self._store.isdependee(name)):
keys.remove(name)
keys2.append(name)
#dependencies, depth first
for name in keys:
if (not self._store.isdepender(name)):
keys.remove(name)
keys2.append(name)
dependers = self._store.getdependers(name)
while len(dependers)>0:
dn = dependers.pop(0)
keys.remove(dn)
keys2.append(dn)
subdeps = self._store.getdependers(dn)
subdeps.extend(dependers)
dependers = subdeps
if len(keys)>0:
raise ValueError("Well that was unexpected")
keys = keys2
for name in keys:
properties = self._store.parameter_list[name]
widget = None
if len(properties['values']) == 1:
#don't have widget if no choice possible
continue
if properties['type'] == 'range':
widget = self._createRangeSlider(name, properties)
if (properties['type'] == 'list' or
('isfield' in properties and properties['isfield'] == 'yes')):
widget = self._createListPulldown(name, properties)
if properties['type'] == 'option':
widget = self._createOptionCheckbox(name, properties)
#if properties['type'] == 'hidden': #disabled for testing fields
#continue
if widget and name in dependencies:
# disable widgets that depend on settings of others
widget.setEnabled(False)
self._dependent_widgets[name] = widget
self._parametersWidget.layout().addStretch()
self._updateDependentWidgets()
def _dependencies_satisfied(self, name):
#translate the sets we use for parameters of the query
#then call the same named method on the store
#TODO: using sets for everything to do options, but probably better if the store supported it
params = []
values = []
for n,vs in self._currentQuery.iteritems():
params.append(n)
values.append(vs)
for element in itertools.product(*values):
q = dict(itertools.izip(params, element))
if self._store.dependencies_satisfied(name, q):
return True
return False
# Update enable state of all dependent widgets
def _updateDependentWidgets(self):
for name, widget in self._dependent_widgets.iteritems():
if self._dependencies_satisfied(name):
widget.setEnabled(True)
else:
widget.setEnabled(False)
# Respond to a slider movement
def onSliderMoved(self):
parameterName = self.sender().objectName()
sliderIndex = self.sender().value()
pl = self._store.parameter_list
value = pl[parameterName]['values'][sliderIndex]
s = set()
s.add(value)
self._currentQuery[parameterName] = s
# Update value label
valueLabel = self._parametersWidget.findChild(QLabel, parameterName + "ValueLabel")
valueLabel.setText(self._formatText(value))
self._updateDependentWidgets()
self.render()
# Respond to a combobox change
def onChosen(self, index):
parameterName = self.sender().objectName()
pl = self._store.parameter_list
value = self.sender().itemText(index) #can't use index directly since menu skips 'depth'
#value = pl[parameterName]['values'][index]
s = set()
s.add(value)
self._currentQuery[parameterName] = s
self._updateDependentWidgets()
self.render()
# Respond to a checkbox change
def onChecked(self, state):
parameterName = self.sender().objectName()
parameterValue = self.sender().value
currentValues = self._currentQuery[parameterName]
if state:
currentValues.add(parameterValue)
else:
if len(currentValues)>1:
currentValues.remove(parameterValue)
else:
self.sender().click() #must have at least one checked
self._currentQuery[parameterName] = currentValues
self._updateDependentWidgets()
self.render()
# Back up slider all the way to the left
def onSkipBackward(self):
parameterName = self.sender().objectName().replace("SkipBackwardButton.", "")
slider = self._parametersWidget.findChild(QSlider, parameterName)
slider.setValue(0)
# Back up slider one step to the left
def onSeekBackward(self):
parameterName = self.sender().objectName().replace("SeekBackwardButton.", "")
slider = self._parametersWidget.findChild(QSlider, parameterName)
slider.setValue(0 if slider.value() == 0 else slider.value() - 1)
# Forward slider one step to the right
def onSeekForward(self):
parameterName = self.sender().objectName().replace("SeekForwardButton.", "")
slider = self._parametersWidget.findChild(QSlider, parameterName)
maximum = slider.maximum()
slider.setValue(maximum if slider.value() == maximum else slider.value() + 1)
# Forward the slider all the way to the right
def onSkipForward(self):
parameterName = self.sender().objectName().replace("SkipForwardButton.", "")
slider = self._parametersWidget.findChild(QSlider, parameterName)
slider.setValue(slider.maximum())
# Play forward through the parameters
def onPlay(self):
parameterName = self.sender().objectName().replace("PlayButton.", "")
timer = QTimer(self)
timer.setObjectName("Timer." + parameterName)
timer.setInterval(200)
timer.timeout.connect(self.onPlayTimer)
timer.start()
def onPlayTimer(self):
parameterName = self.sender().objectName().replace("Timer.", "")
slider = self._parametersWidget.findChild(QSlider, parameterName)
maximum = slider.maximum()
if (slider.value() == slider.maximum()):
self.sender().stop()
else:
slider.setValue(maximum if slider.value() == maximum else slider.value() + 1)
# Format string from number
def _formatText(self, value):
if isinstance(value, int):
return '{0}'.format(value)
if isinstance(value, float):
return '{:2.4f}'.format(value)
# String
return value
# Update slider from value
def _updateSlider(self, parameterName, value):
pl = self._store.parameter_list
index = pl[parameterName]['values'].index(value)
slider = self._parametersWidget.findChild(QSlider, parameterName)
slider.setValue(index)
# Initialize the angles for the camera
def _initializeCamera(self):
self._mouseInteractor.setPhi(next(iter(self._currentQuery['phi'])))
self._mouseInteractor.setTheta(next(iter(self._currentQuery['theta'])))
# Update the camera angle
def _updateCamera(self):
# Set the camera settings if available
phi = self._mouseInteractor.getPhi()
theta = self._mouseInteractor.getTheta()
if ('phi' in self._currentQuery):
s = set()
s.add(phi)
self._currentQuery['phi'] = s
if ('theta' in self._currentQuery):
s = set()
s.add(theta)
self._currentQuery['theta'] = s
# Update the sliders for phi and theta
self._updateSlider('phi', phi)
self._updateSlider('theta', theta)
scale = self._mouseInteractor.getScale()
self._displayWidget.resetTransform()
self._displayWidget.scale(scale, scale)
self.render()
# Perform query requested of the UI
# retrieve documents that go into the result,
# display the retrieved image.
def render(self):
# translate GUI choices (self._currentQuery) into a set of queries
# that we need to render with
def _getfieldsfor(self, n):
param = self._store.parameter_list[n]
vals = param['values']
vals2 = []
#return currently selected color AND depth
#TODO: when we get more complicated GUI for color and shaders we'll return more
for v in vals:
if v in self._currentQuery[n]:
vals2.append(v)
else:
if 'types' in param:
idx = param['values'].index(v)
if param['types'][idx] == 'depth':
vals2.append(v)
return vals2
def _buildqueryfor(self, n, query):
if not self._store.dependencies_satisfied(n, query.dict):
return
if self._store.isfield(n):
colorcomponents = _getfieldsfor(self, n)
for c in colorcomponents:
img_type = self._store.determine_type({n:c})
query.addQuery(img_type, n, c)
layers.append(query)
return
values = self._currentQuery[n]
dependers = self._store.getdependers(n)
for v in values:
lquery = copy.deepcopy(query)
lquery.addToBaseQuery({n:v})
for d in dependers:
_buildqueryfor(self, d, lquery)
dd = self._store.parameter_list
#make query for static contents (e.g. current time and camera)
base_query = LayerSpec.LayerSpec()
potentials = []
for name in dd.keys():
if (not self._store.isdepender(name) and not self._store.islayer(name)):
values = self._currentQuery[name]
v = list(iter(values))[0] #no options in query, so only 1 result not many
base_query.addToBaseQuery({name:v})
else:
potentials.append(name)
#add to the above queries for all of the layers
#each layer query is composed of sequence of field queries
layers = []
hasLayer = False
for name in potentials:
if self._store.islayer(name) and not self._store.isdepender(name):
#name is a top level choice
hasLayer = True
_buildqueryfor(self, name, base_query) #recurse to find subchoices
if not hasLayer:
layers.append(base_query)
#send queries to the store to obtain images
for l in range(0,len(layers)):
layers[l].loadImages(self._store)
if len(layers) == 0:
self._displayWidget.setPixmap(None)
self._displayWidget.setAlignment(Qt.AlignCenter)
return
#render, by iterating through the layers, rendering each ontop and continuing
l0 = layers[0]
c0 = np.copy(l0.getColor1()) #TODO: apply frag shader to derive color from values
if hasLayer:
d0 = np.copy(l0.getDepth())
# composite in the rest of the layers, picking color of nearest pixel
for idx in range(1,len(layers)):
cnext = layers[idx].getColor1()
dnext = layers[idx].getDepth()
indxarray = np.where(dnext<d0)
c0[indxarray[0],indxarray[1],:] = cnext[indxarray[0],indxarray[1],:]
d0[indxarray[0],indxarray[1],:] = dnext[indxarray[0],indxarray[1],:]
# show the result
pimg = PIL.Image.fromarray(c0)
imageString = pimg.tostring('raw', 'RGB')
qimg = QImage(imageString, pimg.size[0], pimg.size[1], QImage.Format_RGB888)
pix = QPixmap.fromImage(qimg)
# Try to resize the display widget
self._displayWidget.sizeHint = pix.size
self._displayWidget.setPixmap(pix)
