plotting_test.py

# Mantid Repository : https://github.com/mantidproject/mantid
#
# Copyright © 2018 ISIS Rutherford Appleton Laboratory UKRI,
#     NScD Oak Ridge National Laboratory, European Spallation Source
#     & Institut Laue - Langevin
# SPDX - License - Identifier: GPL - 3.0 +
from __future__ import absolute_import, print_function

from PyQt4 import QtGui

import sys

from Muon.GUI.Common import message_box

from MultiPlotting.multi_plotting_widget import MultiPlotWidget
from MultiPlotting.multi_plotting_context import *
from MultiPlotting.label import Label

import mantid.simpleapi as mantid


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, spec_num=26)
        self.test.plot("test", ws, spec_num=21)
        self.test.plot("test", ws, spec_num=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, spec_num=1)
        self.test.plot("EXTRA", ws, spec_num=42)
        self.test.plot("moo", ws, spec_num=42)
        self.test.plot("baa", ws, spec_num=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, spec_num=self.n)


def setUpSubplot():
    ws = mantid.LoadMuonNexus("MUSR00062260", OutputWorkspace="ws")
    return ws


def qapp():
    if QtGui.QApplication.instance():
        _app = QtGui.QApplication.instance()
    else:
        _app = QtGui.QApplication(sys.argv)
    return _app


app = qapp()
try:
    window = plotTestGui()
    window.show()
    app.exec_()
except RuntimeError as error:
    message_box.warning(str(error))