+class FavoriteCategoriesWindow(object):
+
+ # @todo Add All, None, and Invert actions
+
+ def __init__(self, parent, app, source, hidden):
+ self._app = app
+ self._source = list(source)
+ self._hidden = hidden
+
+ self._categories = QtGui.QTreeWidget()
+ self._categories.setHeaderLabels(["Categories"])
+ self._categories.setHeaderHidden(True)
+ self._categories.setAlternatingRowColors(True)
+ self._categories.setSelectionBehavior(QtGui.QAbstractItemView.SelectRows)
+ self._categories.setSelectionMode(QtGui.QAbstractItemView.MultiSelection)
+ for catName in self._source:
+ twi = QtGui.QTreeWidgetItem(self._categories)
+ twi.setText(0, catName)
+ if catName not in self._hidden:
+ self._categories.setItemSelected(twi, True)
+ self._selection = self._categories.selectionModel()
+ self._selection.selectionChanged.connect(self._on_selection_changed)
+
+ self._layout = QtGui.QVBoxLayout()
+ self._layout.addWidget(self._categories)
+
+ centralWidget = QtGui.QWidget()
+ centralWidget.setLayout(self._layout)
+
+ self._window = QtGui.QMainWindow(parent)
+ self._window.setAttribute(QtCore.Qt.WA_DeleteOnClose, True)
+ if parent is not None:
+ self._window.setWindowModality(QtCore.Qt.WindowModal)
+ self._window.setWindowTitle("%s - Favorites" % constants.__pretty_app_name__)
+ self._window.setWindowIcon(QtGui.QIcon(self._app.appIconPath))
+ self._window.setCentralWidget(centralWidget)
+
+ self._closeWindowAction = QtGui.QAction(None)
+ self._closeWindowAction.setText("Window")
+ self._closeWindowAction.setShortcut(QtGui.QKeySequence("CTRL+w"))
+ self._closeWindowAction.triggered.connect(self._on_close_window)
+
+ fileMenu = self._window.menuBar().addMenu("&Units")
+ fileMenu.addAction(self._closeWindowAction)
+ fileMenu.addAction(self._app.quitAction)
+
+ viewMenu = self._window.menuBar().addMenu("&View")
+ viewMenu.addAction(self._app.fullscreenAction)
+
+ self._window.addAction(self._app.logAction)
+
+ self._window.show()
+
+ @property
+ def window(self):
+ return self._window
+
+ def close(self):
+ self._window.close()
+
+ def set_fullscreen(self, isFullscreen):
+ if isFullscreen:
+ self._window.showFullScreen()
+ else:
+ self._window.showNormal()
+
+ @misc_utils.log_exception(_moduleLogger)
+ def _on_selection_changed(self, selected, deselected):
+ self._hidden.clear()
+ selectedNames = set(
+ str(item.text(0))
+ for item in self._categories.selectedItems()
+ )
+ for name in self._source:
+ if name not in selectedNames:
+ self._hidden.add(name)
+
+ @misc_utils.log_exception(_moduleLogger)
+ def _on_close_window(self, checked = True):
+ self.close()
+
+