def __init__(self, parent=None):
super(MainWindow, self).__init__(parent)
self.setupUi(self)
# fix QtCreator issues
self.tabWidget.setCurrentIndex(0)
self.tableWidgetExif.verticalHeader().setVisible(True)
self.tableWidgetExif.verticalHeader().setResizeMode(QtGui.QHeaderView.Fixed)
# Hold references to current image
self.current_image_filename = None
self.current_image = None
self.current_image_qt = None
self.current_pixmap = None
self.current_image_rotate = 0 # How much (if any) to rotate the current image
self.current_image_exif = {}
# Hold references to thumbs
self.thumbs = []
self.scene = QtGui.QGraphicsScene()
self.graphicsView.setScene(self.scene)
self.resize(800, 600)
self.splitter.setSizes([600, 180])
self.listWidgetThumbs.setIconSize(QtCore.QSize(100, 100))
# Create thread and connect
self.rabbitmq_reader_thread = RabbitMQToQtSignalThread()
self.rabbitmq_reader_thread.start()
self.rabbitmq_keepalive_thread = RabbitMQKeepAliveThread()
self.rabbitmq_keepalive_thread.start()
# self.clear_exif()
self.update_exif(None)
self._connect_signals()
class MainWindow(QtGui.QMainWindow, Ui_MainWindow):
def __init__(self, parent=None):
super(MainWindow, self).__init__(parent)
self.setupUi(self)
# fix QtCreator issues
self.tabWidget.setCurrentIndex(0)
self.tableWidgetExif.verticalHeader().setVisible(True)
self.tableWidgetExif.verticalHeader().setResizeMode(QtGui.QHeaderView.Fixed)
# Hold references to current image
self.current_image_filename = None
self.current_image = None
self.current_image_qt = None
self.current_pixmap = None
self.current_image_rotate = 0 # How much (if any) to rotate the current image
self.current_image_exif = {}
# Hold references to thumbs
self.thumbs = []
self.scene = QtGui.QGraphicsScene()
self.graphicsView.setScene(self.scene)
self.resize(800, 600)
self.splitter.setSizes([600, 180])
self.listWidgetThumbs.setIconSize(QtCore.QSize(100, 100))
# Create thread and connect
self.rabbitmq_reader_thread = RabbitMQToQtSignalThread()
self.rabbitmq_reader_thread.start()
self.rabbitmq_keepalive_thread = RabbitMQKeepAliveThread()
self.rabbitmq_keepalive_thread.start()
# self.clear_exif()
self.update_exif(None)
self._connect_signals()
def resizeEvent(self, event):
"""Override `resizeEvent()`"""
self._resize_current_image()
def service_status_changed(self, json_msg):
logger.info("service_status_changed()")
msg = json.loads(json_msg)
if msg['origin'] == SERVICE_NAME_RABBITMQ:
checkbox_component = self.status_checkBox_rabbitmq
label_component = self.status_label_rabbitmq
elif msg['origin'] == SERVICE_NAME_EYEFISERVER2:
checkbox_component = self.status_checkBox_eyefiserver2
label_component = self.status_label_eyefiserver2
else:
logger.warn("Unknown origin of event: %s", msg['origin'])
return
if msg['extra'].get('new_status', None) == SERVICE_STATUS_UP:
checkbox_component.setChecked(True)
label_component.setText("<font color='green'><b>RUNNING</b></font>")
elif msg['extra'].get('new_status', None) == SERVICE_STATUS_DOWN:
checkbox_component.setChecked(False)
label_component.setText("<font color='red'><b>STOPPED</b></font>")
else:
logger.warn("Unknown 'new_status': %s", msg['extra'].get('new_status', ''))
def _connect_signals(self):
"""Connect the signals"""
self.connect(self.rabbitmq_reader_thread,
QtCore.SIGNAL("display_image(QString)"),
self.display_image)
self.connect(self.rabbitmq_reader_thread,
QtCore.SIGNAL("service_status_changed(QString)"),
self.service_status_changed)
self.connect(self.splitter,
QtCore.SIGNAL("splitterMoved(int, int)"),
self._resize_current_image)
self.listWidgetThumbs.currentItemChanged.connect(
self._show_image_from_thumbs)
def _show_image_from_thumbs(self, current, previous):
"""Shows and image from the thumbnails"""
img_list = self.thumbs[self.listWidgetThumbs.currentRow()]
self.display_image(img_list[0], add_to_thumb_list=False)
def _resize_current_image(self):
if self.current_image:
w, h = self.current_image.size
self.graphicsView.fitInView(QtCore.QRectF(0, 0, w, h), QtCore.Qt.KeepAspectRatio)
self.scene.update()
def clear_exif(self):
self.tableWidgetExif.clear()
self.tableWidgetExif.setColumnCount(1)
self.tableWidgetExif.setHorizontalHeaderLabels(["Value"])
# FIXME: clean with one call (don't know how)
while self.tableWidgetExif.rowCount():
self.tableWidgetExif.removeRow(0)
def update_exif(self, image_filename):
"""Updates the EXIF information, updates `image_rotate`"""
self.clear_exif()
self.current_image_rotate = 0
if image_filename:
self.current_image_exif = get_exif_tags(image_filename)
else:
self.current_image_exif = {}
exif_dict_to_show = get_tags_to_show(self.current_image_exif)
# exif_keys = exif_dict_to_show.keys()
# for row_num in range(0, len(exif_dict_to_show)):
row_num = 0
for _, tag_value in exif_dict_to_show.iteritems():
valueItem = QtGui.QTableWidgetItem()
valueItem.setTextAlignment(QtCore.Qt.AlignVCenter | QtCore.Qt.AlignLeft)
valueItem.setFlags(valueItem.flags() ^ QtCore.Qt.ItemIsEditable)
valueItem.setText(unicode(tag_value))
self.tableWidgetExif.insertRow(row_num)
self.tableWidgetExif.setItem(row_num, 0, valueItem)
row_num += 1
self.tableWidgetExif.setVerticalHeaderLabels(exif_dict_to_show.keys())
self.tableWidgetExif.resizeColumnsToContents()
self.current_image_rotate = how_much_rotate(self.current_image_exif) or 0
def add_current_image_to_thumb_list(self):
# if len(self.thumbs) > 4:
# self.thumbs.pop(0)
# self.listWidgetThumbs.takeItem(0)
# Create resized image
icon_image = self.current_image.copy()
icon_image.thumbnail((100, 100,), resample=Image.ANTIALIAS)
icon_image_qt = ImageQt.ImageQt(icon_image)
icon_qpixmap = QtGui.QPixmap.fromImage(icon_image_qt, QtCore.Qt.ImageConversionFlag.AutoColor)
icon = QtGui.QIcon(icon_qpixmap)
thumb_item = QtGui.QListWidgetItem()
thumb_item.setIcon(icon)
self.listWidgetThumbs.addItem(thumb_item)
# self.listWidgetThumbs.setCurrentRow(thumb_item)
# scrollbar = self.listWidgetThumbs.horizontalScrollBar()
# assert isinstance(scrollbar, QtGui.QScrollBar)
# scrollbar.value()
# scrollbar.heightForWidth()
self.thumbs.append(
(self.current_image_filename, icon_image, icon_image_qt, icon_qpixmap, icon)
)
def display_image(self, image_filename, add_to_thumb_list=True):
self.current_image_filename = image_filename
self.current_image = Image.open(self.current_image_filename)
self.scene.clear()
self.graphicsView.resetTransform()
self.graphicsView.resetMatrix()
# QtGui.QApplication.processEvents()
# Right now, calling `QtGui.QApplication.processEvents()` is BOGUS,
# since it allow the user to clic a thumb and change `self.current_imagexxx`,
# before the thumb is created!
self.statusBar.showMessage("Image: {0}".format(self.current_image_filename))
# QtGui.QApplication.processEvents()
self.current_image_qt = ImageQt.ImageQt(self.current_image)
# QtGui.QApplication.processEvents()
self.current_pixmap = QtGui.QPixmap.fromImage(self.current_image_qt, QtCore.Qt.ImageConversionFlag.AutoColor)
# QtGui.QApplication.processEvents()
self.scene.addPixmap(self.current_pixmap)
if self.current_image_rotate != 0:
self.graphicsView.rotate(self.current_image_rotate) # update_exif() updates `image_rotate`
self._resize_current_image()
# QtGui.QApplication.processEvents()
self.update_exif(self.current_image_filename)
# QtGui.QApplication.processEvents()
if add_to_thumb_list:
self.add_current_image_to_thumb_list()