def __init__(self, parent=None): QtGui.QWidget.__init__(self, parent) self.ui = Ui_mainDialog() self.ui.setupUi(self) self.child = None self.logs = None self.legends = None self.plotlogs = None self.title = None self.xlabel = None self.ylabel = None
class MainDialog(QtGui.QDialog): def __init__(self, parent=None): QtGui.QWidget.__init__(self, parent) self.ui = Ui_mainDialog() self.ui.setupUi(self) self.child = None self.logs = None self.legends = None self.plotlogs = None self.title = None self.xlabel = None self.ylabel = None def prepare(self): self.logs = [] for item in self.ui.filelistWidget.iterItems(): self.logs.append(LogData(item.text())) self.plotlogs = [] self.legends = [] for i, log in enumerate(self.logs): self.legends.append(str(i + 1)) self.plotlogs.append((log.idx, log.verr, None)) self.title = "Validation error per exemplar in %" self.xlabel = "Iteration" self.ylabel = "% error / exemplar" def plot(self): self.prepare() if self.child == None: self.child = PlotWindow(self) else: self.child.pwidget.axes.clear() self.child.plot_figure(self.plotlogs, self.title, self.xlabel, None, self.ylabel, None, self.legends) self.child.show() def moveUp(self): item = self.ui.filelistWidget.currentItem() row = self.ui.filelistWidget.row(item) if row == 0: return self.ui.filelistWidget.takeItem(row) self.ui.filelistWidget.insertItem(row - 1, item) self.ui.filelistWidget.setCurrentRow(row - 1) def moveDown(self): item = self.ui.filelistWidget.currentItem() row = self.ui.filelistWidget.row(item) if row == self.ui.filelistWidget.count() - 1: return self.ui.filelistWidget.takeItem(row) self.ui.filelistWidget.insertItem(row + 1, item) self.ui.filelistWidget.setCurrentRow(row + 1) def deleteSelected(self): for item in self.ui.filelistWidget.selectedItems(): self.ui.filelistWidget.takeItem(self.ui.filelistWidget.row(item)) def clearList(self): self.ui.filelistWidget.clear() def close_child(self): self.child = None def closeEvent(self, ce): self.close()