class plotTestGui(QtGui.QMainWindow): def __init__(self, parent=None): super(plotTestGui, self).__init__(parent) self._context = PlottingContext() self.test = MultiPlotWidget(self._context, self) ws = setUpSubplot() self.test.add_subplot("test", 221) self.test.add_subplot("bob", 222) self.test.add_subplot("moo", 223) self.test.add_subplot("baa", 224) self.test.plot("test", ws, specNum=26) self.test.plot("test", ws, specNum=21) self.test.plot("test", ws, specNum=22) # defines position of label dummy = Label("dummy", 10.1, False, 0.9, True, rotation=-90) dummy2 = Label("protected", 5.1, False, 0.9, True, rotation=-90, protected=True) dummy3 = Label("just annotate", 14.1, False, 0.9, True) # defines position of line # need to add methods to add just a label # need to add_vline with a name and if protected but no # annotation self.test.add_vline_and_annotate("test", 10, dummy) self.test.add_vline_and_annotate("test", 5, dummy2) self.test.add_annotate("bob", dummy3) self.test.add_vline("bob", 1.2, "just a line") self.test.plot("bob", ws, specNum=1) self.test.plot("moo", ws, specNum=42) self.test.plot("baa", ws, specNum=2) self.test.set_all_values() self.setCentralWidget(self.test) self.setWindowTitle("plot test")
class plotTestGui(QtGui.QMainWindow): def __init__(self, parent=None): super(plotTestGui, self).__init__(parent) self._context = PlottingContext() self.test = MultiPlotWidget(self._context, self) ws = setUpSubplot() self.test.add_subplot("test") self.test.add_subplot("bob") self.test.add_subplot("moo") self.test.add_subplot("baa") self.test.add_subplot("EXTRA") self.test.plot("test", ws, specNum=26) self.test.plot("test", ws, specNum=21) self.test.plot("test", ws, specNum=22) # defines position of label dummy = Label("dummy", 10.1, False, 0.9, True, rotation=-90) dummy2 = Label( "protected", 5.1, False, 0.9, True, rotation=-90, protected=True) dummy3 = Label("just annotate", 14.1, False, 0.9, True) # defines position of line # need to add methods to add just a label # need to add_vline with a name and if protected but no # annotation self.test.add_vline_and_annotate("test", 10, dummy) self.test.add_vline_and_annotate("test", 5, dummy2) self.test.add_annotate("bob", dummy3) self.test.add_vline("bob", 1.2, "just a line") self.test.plot("bob", ws, specNum=1) self.test.plot("EXTRA", ws, specNum=42) self.test.plot("moo", ws, specNum=42) self.test.plot("baa", ws, specNum=2) self.test.set_all_values() self.test.connectCloseSignal(self.close) # add button for adding more plots self.n = 0 self.ws = ws self.btn = QtGui.QPushButton("add plot") self.btn.clicked.connect(self.add) self.grid = QtGui.QSplitter(QtCore.Qt.Vertical) self.grid.addWidget(self.test) self.grid.addWidget(self.btn) self.setCentralWidget(self.grid) self.setWindowTitle("plot test") def add(self): self.n += 1 self.test.add_subplot(str(self.n)) self.test.plot(str(self.n), self.ws, specNum=self.n)