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 __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")
def _gen_label(self, name, x_value_in, element=None):
if element is None:
return
# check x value is a float
try:
x_value = float(x_value_in)
except:
return
if name not in self.element_lines[element]:
self.element_lines[element].append(name)
# make sure the names are strings and x values are numbers
return Label(str(name),
x_value,
False,
offset,
True,
rotation=-90,
protected=True)