def load_project_list(self):
"""
Populate the project selection list.
"""
self.list_projects.clear()
for path in self.projects_dict:
project = self.projects_dict[path]
name = project.name
item = QListWidgetItem(name)
if getattr(project, 'icon', None):
icon = self.api.load_icon(path, project)
else:
icon = qta.icon('fa.cog')
item.setIcon(icon)
item.project = project
item.path = path
if project.commands:
item.setFlags(Qt.ItemIsSelectable | Qt.ItemIsUserCheckable
| Qt.ItemIsEnabled)
if project.is_app:
item.setCheckState(Qt.Checked)
else:
item.setCheckState(Qt.Unchecked)
self.list_projects.addItem(item)
self.toogle_dev_tools(visible=self.dev_tool_visible)
for i in range(self.list_projects.count()):
item = self.list_projects.item(i)
item.setSizeHint(QSize(item.sizeHint().width(), self._height()))
def make_items(self, uri_file_list):
self.clear()
for filename, is_label in uri_file_list:
item = QListWidgetItem(filename)
item.setFlags(Qt.ItemIsEnabled | Qt.ItemIsSelectable)
item.setCheckState(Qt.Checked if is_label else Qt.Unchecked)
self.addItem(item)
def _on_data_change(self, *event):
components = getattr(type(self.state), 'component').get_choices(self.state)
self.ui.list_component.clear()
for component in components:
item = QListWidgetItem(component.label)
item.setCheckState(Qt.Checked)
self.ui.list_component.addItem(item)
def load_file_list(self):
self.file_list_widget.clear()
for image_dir in self.file_dirs:
if not QFile.exists(image_dir):
continue
item = QListWidgetItem(image_dir)
item.setFlags(Qt.ItemIsEnabled | Qt.ItemIsSelectable)
label_dir = os.path.splitext(image_dir)[0] + '.json'
if QFile.exists(label_dir):
item.setCheckState(Qt.Checked)
else:
item.setCheckState(Qt.Unchecked)
self.file_list_widget.addItem(item)
def _on_data_change(self, *event):
components = getattr(type(self.state), 'component').get_choices(self.state)
self.ui.list_component.clear()
for component in components:
if isinstance(component, ChoiceSeparator):
item = QListWidgetItem(str(component))
item.setFlags(item.flags() & ~Qt.ItemIsSelectable)
item.setForeground(Qt.gray)
else:
item = QListWidgetItem(component.label)
item.setCheckState(Qt.Checked)
self.ui.list_component.addItem(item)
def modelChanged(self):
self._list.clear()
items = self._model.getList()
for item in items:
list_item = QListWidgetItem(item)
list_item.setFlags(list_item.flags() | Qt.ItemIsUserCheckable)
if self._model.isValueSelected(item):
list_item.setCheckState(Qt.Checked)
else:
list_item.setCheckState(Qt.Unchecked)
self._list.addItem(list_item)
self.filterList(self._search_box.filter())
def _on_data_change(self, *event):
components = getattr(type(self.state),
'component').get_choices(self.state)
self.ui.list_component.clear()
for component in components:
if isinstance(component, ChoiceSeparator):
item = QListWidgetItem(str(component))
item.setFlags(item.flags() & ~Qt.ItemIsSelectable)
item.setForeground(Qt.gray)
else:
item = QListWidgetItem(component.label)
item.setCheckState(Qt.Checked)
self.ui.list_component.addItem(item)
def set_options(self, generators, icons):
self.build_name_edit.clear()
self.generators_list.clear()
self.icon_combobox.clear()
for generator in generators:
item = QListWidgetItem(generator, self.generators_list)
item.setFlags(Qt.ItemIsEnabled | Qt.ItemIsUserCheckable)
item.setCheckState(Qt.Checked)
self.generators_list.addItem(item)
for icon in icons:
name = icon.get("name")
pix = QPixmap()
pix.loadFromData(base64.b64decode(icon.get("ico")))
ico = QIcon()
ico.addPixmap(pix)
self.icon_combobox.addItem(ico, name)
self.spinner.stop()
def __init__(self, export_dict, viewer, parent=None):
super().__init__(parent=parent)
self.setWindowTitle("Export")
self.export_dict = export_dict
self.viewer = viewer()
self.list_view = SearchableListWidget()
self.check_state = np.zeros(len(export_dict), dtype=bool)
self.check_state[...] = True
for el in sorted(export_dict.keys()):
item = QListWidgetItem(el)
# noinspection PyTypeChecker
item.setFlags(item.flags() | Qt.ItemIsUserCheckable)
item.setCheckState(Qt.Checked)
self.list_view.addItem(item)
self.checked_num = len(export_dict)
self.export_btn = QPushButton("Export")
self.cancel_btn = QPushButton("Cancel")
self.check_btn = QPushButton("Check all")
self.uncheck_btn = QPushButton("Uncheck all")
self.cancel_btn.clicked.connect(self.close)
self.export_btn.clicked.connect(self.accept)
self.check_btn.clicked.connect(self.check_all)
self.uncheck_btn.clicked.connect(self.uncheck_all)
self.list_view.itemSelectionChanged.connect(self.preview)
self.list_view.itemChanged.connect(self.checked_change)
layout = QVBoxLayout()
info_layout = QHBoxLayout()
info_layout.addWidget(self.list_view)
info_layout.addWidget(self.viewer)
layout.addLayout(info_layout)
btn_layout = QHBoxLayout()
btn_layout.addWidget(self.check_btn)
btn_layout.addWidget(self.uncheck_btn)
btn_layout.addStretch()
btn_layout.addWidget(self.export_btn)
btn_layout.addWidget(self.cancel_btn)
layout.addLayout(btn_layout)
self.setLayout(layout)
def _on_data_change(self, *event):
components = getattr(type(self.state),
'component').get_choices(self.state)
self.ui.list_component.clear()
for (component, k) in zip(components, np.arange(0, len(components))):
if isinstance(component, ChoiceSeparator):
item = QListWidgetItem(str(component))
item.setFlags(item.flags() & ~Qt.ItemIsSelectable)
item.setForeground(Qt.gray)
else:
item = QListWidgetItem(component.label)
if self.checked_dictionary[self.state.data.label][k] == True:
item.setCheckState(Qt.Checked)
else:
item.setCheckState(Qt.Unchecked)
self.ui.list_component.addItem(item)
def _set_list_preview_items(self, *, items=None):
"""
Set items of the list for selecting channels in preview tab. If the list of items is
not specified, then it is loaded from the respective data structure.
Parameters
----------
items: list(str)
The list of items. The list may be cleared by calling
`self._set_cbox_channel_items(items=[])`
"""
self.list_preview.itemChanged.disconnect(
self.list_preview_item_changed)
self.list_preview.clear()
if items is None:
items = list(self.gpc.get_file_channel_list())
for s in items:
wi = QListWidgetItem(s, self.list_preview)
wi.setFlags(wi.flags() | Qt.ItemIsUserCheckable)
wi.setFlags(wi.flags() & ~Qt.ItemIsSelectable)
wi.setCheckState(Qt.Unchecked)
# Adjust height so that it fits all the elements
adjust_qlistwidget_height(self.list_preview,
other_widgets=[self.group_preview, self])
# This will cause the preview data to be plotted (the plot is expected to be hidden,
# since no channels were selected). Here we select the first channel in the list.
for n, item in enumerate(items):
state = Qt.Checked if self.gpc.is_dset_item_selected_for_preview(
item) else Qt.Unchecked
self.list_preview.item(n).setCheckState(state)
self.list_preview.itemChanged.connect(self.list_preview_item_changed)
def _create_item(self, path):
"""Helper to create a new list item."""
item = QListWidgetItem(path)
item.setIcon(ima.icon('DirClosedIcon'))
if path in self.read_only_path:
item.setFlags(Qt.NoItemFlags | Qt.ItemIsUserCheckable)
item.setCheckState(Qt.Checked)
elif path in self.not_active_path:
item.setFlags(item.flags() | Qt.ItemIsUserCheckable)
item.setCheckState(Qt.Unchecked)
else:
item.setFlags(item.flags() | Qt.ItemIsUserCheckable)
item.setCheckState(Qt.Checked)
return item
def update_list(self):
"""Update path list"""
self.listwidget.clear()
for name in self.pathlist + self.ro_pathlist:
item = QListWidgetItem(name)
item.setIcon(ima.icon('DirClosedIcon'))
if name in self.ro_pathlist:
item.setFlags(Qt.NoItemFlags | Qt.ItemIsUserCheckable)
item.setCheckState(Qt.Checked)
elif name in self.not_active_pathlist:
item.setFlags(item.flags() | Qt.ItemIsUserCheckable)
item.setCheckState(Qt.Unchecked)
else:
item.setFlags(item.flags() | Qt.ItemIsUserCheckable)
item.setCheckState(Qt.Checked)
self.listwidget.addItem(item)
self.refresh()
def update_list(self):
"""Update path list"""
self.listwidget.clear()
for name in self.pathlist+self.ro_pathlist:
item = QListWidgetItem(name)
item.setIcon(ima.icon('DirClosedIcon'))
if name in self.ro_pathlist:
item.setFlags(Qt.NoItemFlags | Qt.ItemIsUserCheckable)
item.setCheckState(Qt.Checked)
elif name in self.not_active_pathlist:
item.setFlags(item.flags() | Qt.ItemIsUserCheckable)
item.setCheckState(Qt.Unchecked)
else:
item.setFlags(item.flags() | Qt.ItemIsUserCheckable)
item.setCheckState(Qt.Checked)
self.listwidget.addItem(item)
self.refresh()
