def reset_model(self):
self.current_model_index = QtCore.QModelIndex()
self.model.clear()
self.model.setColumnCount(2)
self.model.setHeaderData(0, QtCore.Qt.Orientation.Horizontal, 'Image')
self.model.setHeaderData(1, QtCore.Qt.Orientation.Horizontal, 'Count')
self.treeView.setExpandsOnDoubleClick(False)
self.treeView.header().setStretchLastSection(False)
self.treeView.header().setSectionResizeMode(
0, QtWidgets.QHeaderView.ResizeMode.Stretch)
self.treeView.setTextElideMode(QtCore.Qt.TextElideMode.ElideMiddle)
def __init__(self, canvas, parent=None):
QtWidgets.QWidget.__init__(self, parent)
self.setupUi(self)
self.canvas = canvas
self.pushButtonAddClass.clicked.connect(self.add_class)
self.pushButtonRemoveClass.clicked.connect(self.remove_class)
self.pushButtonImport.clicked.connect(self.import_metadata)
self.pushButtonSave.clicked.connect(self.save)
self.pushButtonLoadPoints.clicked.connect(self.load)
self.pushButtonReset.clicked.connect(self.reset)
self.pushButtonExport.clicked.connect(self.export)
self.pushButtonExport.setIcon(QtGui.QIcon('icons:export.svg'))
self.pushButtonReset.setIcon(QtGui.QIcon('icons:reset.svg'))
self.pushButtonImport.setIcon(QtGui.QIcon('icons:import.svg'))
self.pushButtonSave.setIcon(QtGui.QIcon('icons:save.svg'))
self.pushButtonLoadPoints.setIcon(QtGui.QIcon('icons:load.svg'))
self.pushButtonRemoveClass.setIcon(QtGui.QIcon('icons:delete.svg'))
self.pushButtonAddClass.setIcon(QtGui.QIcon('icons:add.svg'))
self.tableWidgetClasses.verticalHeader().setVisible(False)
self.tableWidgetClasses.horizontalHeader().setMinimumSectionSize(1)
self.tableWidgetClasses.horizontalHeader().setStretchLastSection(False)
self.tableWidgetClasses.horizontalHeader().setSectionResizeMode(
0, QtWidgets.QHeaderView.ResizeMode.Stretch)
self.tableWidgetClasses.setColumnWidth(1, 30)
self.tableWidgetClasses.cellClicked.connect(self.cell_clicked)
self.tableWidgetClasses.cellChanged.connect(self.cell_changed)
self.tableWidgetClasses.selectionModel().selectionChanged.connect(
self.selection_changed)
self.checkBoxDisplayPoints.toggled.connect(self.display_points)
self.checkBoxDisplayGrid.toggled.connect(self.display_grid)
self.canvas.image_loaded.connect(self.image_loaded)
self.canvas.update_point_count.connect(self.update_point_count)
self.canvas.points_loaded.connect(self.points_loaded)
self.canvas.metadata_imported.connect(self.display_count_tree)
self.model = QtGui.QStandardItemModel()
self.current_model_index = QtCore.QModelIndex()
self.treeView.setModel(self.model)
self.reset_model()
self.treeView.doubleClicked.connect(self.select_model_item)
self.previous_file_name = None # used for quick save
self.spinBoxPointRadius.valueChanged.connect(
self.canvas.set_point_radius)
self.spinBoxGrid.valueChanged.connect(self.canvas.set_grid_size)
icon = QtGui.QPixmap(20, 20)
icon.fill(QtCore.Qt.GlobalColor.yellow)
self.labelPointColor.setPixmap(icon)
self.labelPointColor.mousePressEvent = self.change_active_point_color
icon = QtGui.QPixmap(20, 20)
icon.fill(QtCore.Qt.GlobalColor.white)
self.labelGridColor.setPixmap(icon)
self.labelGridColor.mousePressEvent = self.change_grid_color
self.checkBoxImageFields.clicked.connect(self.hide_custom_fields.emit)
