KaptureControlGui/KCG/base/plotWidget.py

"""
This Module implements the Plot Windows used in KCG
"""
import pyqtgraph as pg
pg.setConfigOption('useOpenGl', True)
pg.setConfigOption('background', 'w')
pg.setConfigOption('foreground', 'k')
import numpy as np
from PyQt4 import QtCore, QtGui
from groupedelements import live_plot_windows
from backend import io
from backend import board
from backend.board import available_boards
import kcgwidget as kcgw
from .. import config

tr = kcgw.tr

LIVE = 1
FILE = 2
ERROR = 99
NO_DATA = 98


class Enum():
    """
    Simple Enum Class (as this is not supported by clean python)
    """
    def __init__(self, *args):
        self.idx = 0
        for i in args:
            setattr(self, i, self.idx)
            self.idx += 1

plotList = [tr("Label", "Heatmap"), tr("Label", "FFT"), tr("Label", "Trains"), tr("Label", "Combined"), tr("Label", "Compare"), tr("Label", "Follow")]
PlotType = Enum(*[str(i) for i in plotList])

# gradient = {'mode': 'rgb',
#             'ticks': [(0.0, (0.37281, 0.11883, 3.53583, 255)),
#                       (0.0390625, (4.9401150000000005, 3.858915, 22.635585, 255)),
#                       (0.078125, (15.6417, 9.330449999999999, 45.29871, 255)),
#                       (0.1171875, (29.74728, 12.13137, 69.44185499999999, 255)),
#                       (0.15625, (46.774395000000005, 10.283895, 90.51760499999999, 255)),
#                       (0.1953125, (64.1631, 9.614775, 102.86139, 255)),
#                       (0.234375, (80.65191, 13.63995, 108.40458, 255)),
#                       (0.2734375, (96.64525499999999, 19.444515, 110.343345, 255)),
#                       (0.3125, (112.507785, 25.33119, 110.05646999999999, 255)),
#                       (0.3515625, (128.390715, 31.001625, 107.95578, 255)),
#                       (0.390625, (144.29277, 36.609585, 104.10579, 255)),
#                       (0.4296875, (160.100985, 42.476625, 98.50038, 255)),
#                       (0.46875, (175.606515, 49.020945, 91.188765, 255)),
#                       (0.5078125, (190.517385, 56.70639, 82.32827999999999, 255)),
#                       (0.546875, (204.477105, 65.96187, 72.190245, 255)),
#                       (0.5859375, (217.10292, 77.07629999999999, 61.10718, 255)),
#                       (0.625, (228.047775, 90.11674500000001, 49.36392, 255)),
#                       (0.6640625, (237.05922, 104.927145, 37.068585, 255)),
#                       (0.703125, (243.99726, 121.21578, 24.147225, 255)),
#                       (0.7421875, (248.797635, 138.66849, 11.122589999999999, 255)),
#                       (0.78125, (251.41776000000002, 157.01625, 6.52596, 255)),
#                       (0.8203125, (251.80332, 176.04333, 20.397450000000003, 255)),
#                       (0.859375, (249.90816, 195.543435, 42.420015, 255)),
#                       (0.8984375, (245.92047, 215.18124, 69.714705, 255)),
#                       (0.9375, (241.63647, 233.936745, 104.719575, 255)),
#                       (0.9765625, (245.26206, 248.86062, 145.840875, 255)),
#                       (0.99609375, (252.03231, 254.58282, 164.45562, 255))]}

gradient = {'mode': 'rgb',
            'ticks': [(0.0, (0.37281, 0.11883, 3.53583, 255)),
                      (0.1171875, (29.74728, 12.13137, 69.44185499999999, 255)),
                      (0.234375, (80.65191, 13.63995, 108.40458, 255)),
                      (0.3515625, (128.390715, 31.001625, 107.95578, 255)),
                      (0.46875, (175.606515, 49.020945, 91.188765, 255)),
                      (0.5859375, (217.10292, 77.07629999999999, 61.10718, 255)),
                      (0.703125, (243.99726, 121.21578, 24.147225, 255)),
                      (0.8203125, (251.80332, 176.04333, 20.397450000000003, 255)),
                      (0.9375, (241.63647, 233.936745, 104.719575, 255)),
                      (0.99609375, (252.03231, 254.58282, 164.45562, 255))]}


class CustomGradientEditorItem(pg.GradientEditorItem):
    """
    A Gradient Editor Item to insert a perception linear gradient
    """
    def __init__(self, **kwargs):
        pg.GradientEditorItem.__init__(self, **kwargs)
        self.customGradients = {}

    def addGrad(self, name, grad_dic):
        """
        Add a gradient to the list of gradients in the gui
        :param name: the name of the gradient
        :param grad_dic: the dictionary containing the gradient data
        """
        if name not in self.customGradients:
            self.customGradients[name] = grad_dic
            px = QtGui.QPixmap(100, 15)
            p = QtGui.QPainter(px)
            self.restoreState(grad_dic)
            length_backup = self.length
            self.length = 100
            grad = self.getGradient()
            self.length = length_backup
            brush = QtGui.QBrush(grad)
            p.fillRect(QtCore.QRect(0, 0, 100, 15), brush)
            p.end()
            label = QtGui.QLabel()
            label.setPixmap(px)
            label.setContentsMargins(1, 1, 1, 1)
            act = QtGui.QWidgetAction(self)
            act.setDefaultWidget(label)
            act.triggered.connect(self.contextMenuClicked)
            act.name = name
            if len(self.customGradients) > 1:
                self.menu.insertAction(self.menu.actions()[0], act)
            else:
                sep = self.menu.insertSeparator(self.menu.actions()[0])
                self.menu.insertAction(sep, act)

        self.restoreState(grad_dic)

    def restoreState(self, state):
        """
        Reimplemented of pyqtgraph.GradientEditorItem.restoreState to work with our custom perception linear gradient.
        :param state: the state to restore to
        """
        self.setColorMode(state['mode'])
        for t in list(self.ticks.keys()):
            self.removeTick(t, finish=False)
        self.bottomTick = [None, 10]
        self.topmostTick = [None, -1]
        for t in state['ticks']:
            c = QtGui.QColor(*t[1])
            tick = self.addTick(t[0], c, finish=False)
            if t[0] < self.bottomTick[1]:
                self.bottomTick = [tick, t[0]]
            if t[0] > self.topmostTick[1]:
                self.topmostTick = [tick, t[0]]
        self.tickPosInPercent = self.ticks.copy()
        for t in self.ticks:
            if t is self.topmostTick[0] or t is self.bottomTick[0]:
                continue
            t.hide()
            t.movable = False
        # self.bottomTick[1] *= self.length
        # self.topmostTick[1] *= self.length
        self.updateGradient()
        self.sigGradientChangeFinished.emit(self)

    def loadPreset(self, name):
        """
        Reimplemented of pyqtgraph.GradientEditorItem.loadPreset to work with our custom perception linear gradient.
        :param name: the name of the preset to load
        """
        if name in self.customGradients:
            self.restoreState(self.customGradients[name])
        else:
            super(CustomGradientEditorItem, self).loadPreset(name)

    def tickMoved(self, tick, pos):
        """
        Reimplemented of pyqtgraph.GradientEditorItem.tickMoved to work with our custom perception linear gradient,
        which has a lot of steps which would create a lot of ticks in the gradient legend. This removes all but 2
        ticks and aligns all the internal ticks accordingly.
        :param tick: the tick to move
        :param pos: the position to move to
        """
        if tick is self.bottomTick[0]:
            # self.bottomTick[1] = pos.x()
            pg.TickSliderItem.tickMoved(self, tick, pos)
        elif tick is self.topmostTick[0]:
            # self.topmostTick[1] = pos.x()
            pg.TickSliderItem.tickMoved(self, tick, pos)
        if tick in [self.bottomTick[0], self.topmostTick[0]]:  # if bottom or topmost tick was moved
            if pos.x() < self.length and pos.x() > 0:  # if tick is not at top or bottom
                newUnit = (self.topmostTick[0].pos().x() - self.bottomTick[0].pos().x())  # create new virtual unit length
                for t in self.ticks:  # for every tick
                    if t not in [self.bottomTick[0], self.topmostTick[0]]:  # if tick is not bottom or topmost tick
                        pos = t.pos()
                        new_x = self.bottomTick[0].pos().x() + self.tickPosInPercent[t] * newUnit
                        pos.setX(new_x)
                        t.setPos(pos)
                        pg.TickSliderItem.tickMoved(self, t, pos)
        self.updateGradient()


class SpectrogramColorLegendItem(pg.GraphicsWidget):
    """
    The Item used as Legend for Heatmap and FFT Plot
    """
    gradientChanged = QtCore.pyqtSignal()
    def __init__(self, img=None):
        """
        Initialise the Legend
        :param img: (ImageItem) the image item this is the legend for
        :return: -
        """
        super(SpectrogramColorLegendItem, self).__init__()
        self.layout = QtGui.QGraphicsGridLayout()
        self.setLayout(self.layout)
        self.img = img
        # self.gei = pg.GradientEditorItem(orientation='right')
        self.gei = CustomGradientEditorItem(orientation='right', allowAdd=False)
        self.legend_axis = pg.AxisItem(orientation='left')
        # self.addItem(self.legend_axis)
        self.layout.addItem(self.legend_axis, 0, 0, alignment=QtCore.Qt.AlignVCenter)
        # self.addItem(self.gei)
        self.layout.addItem(self.gei, 0, 1)
        self.gei.sigGradientChanged.connect(self.gradient_changed)
        # self.gei.loadPreset('spectrum')
        self.gei.addGrad('inferno', gradient)
        # self.setFixedWidth(80)
        self.image_changed()

    def gradient_changed(self):
        """
        Proxy function for the gradient_cnahged event of the GradientEditorItem
        :return:
        """
        self.gradientChanged.emit()

    def set_image(self, img):
        """
        Set the Image this is the legend for (only needed if not already done upon initialisation)
        :param img: (ImageItem) The image item to use
        :return: -
        """
        self.img = img
        self.image_changed(True)

    def update_axis(self):
        """
        Update the axis of this legend
        :return: -
        """
        if self.img.getLevels():
            # min, max = self.img.getLevels()
            min, max = (np.min(self.img.image), np.max(self.img.image))
            self.legend_axis.setRange(min, max)

    def reset_gradient(self):
        """
        Reset the gradient to the preset "spectrum"
        :return: -
        """
        # self.gei.loadPreset('spectrum')
        self.gei.loadPreset('inferno')
        # self.gei.restoreState(gradient)

    def image_changed(self, reset_gradient=True):
        """
        Call this to adjust the legend when the image has changed
        :param reset_gradient: (bool) whether to reset the gradient or not
        :return:
        """
        if self.img:
            self.update_axis()
            if reset_gradient:
                self.reset_gradient()

    def getLookupTable(self, nPts, alpha=None):
        """
        Get the look up table of the imageitem
        :param nPts: (int) number of points the lookup table is defined on
        :param alpha: ??
        :return: -
        """
        return self.gei.getLookupTable(nPts, alpha=alpha)

    def resizeEvent(self, event):
        """
        Handle resizing of the window
        """
        self.legend_axis.setHeight(self.gei.length)


class SubPlotWidget(pg.GraphicsLayoutWidget):
    """
    The Widget actually containing the plots and images
    """
    def __init__(self, dType=FILE):
        super(SubPlotWidget, self).__init__()

        self.dType = dType

        self._type_changed = False
        # self.setStyleSheet("border: 5px solid black; margin: 10px;")
        self.plotType = 1
        self.plotItem = pg.PlotItem()
        self.img = pg.ImageItem()
        self.plotItem2 = pg.PlotItem()
        self.img2 = pg.ImageItem()
        self.plotItem2.addItem(self.img2)
        self.setFrameStyle(self.StyledPanel | self.Sunken)

        self.gradient_legend = SpectrogramColorLegendItem(self.img)
        def changelut():
            """
            Handle changing of the lookup table
            """
            self.img.setLookupTable(self.gradient_legend.getLookupTable(512))
            self.img2.setLookupTable(self.gradient_legend.getLookupTable(512))
        self.gradient_legend.gradientChanged.connect(changelut)

        self.addItem(self.plotItem)
        self.plotItem.addItem(self.img)
        self.plotItemPlotScatter = pg.PlotDataItem(pen=None, symbolPen=pg.mkPen(50, 50, 255), symbolSize=3, pxMode=True, symbolBrush=pg.mkBrush(50, 50, 255), downsample=200)
        self.plotItemPlotScatter1 = pg.PlotDataItem(pen=None, symbolPen=pg.mkPen(255, 50, 50), symbolSize=3, pxMode=True, symbolBrush=pg.mkBrush(255, 50, 50), downsample=200)
        # self.plotItemPlotScatter2 = pg.PlotDataItem(pen=None, symbolPen=pg.mkPen(50, 255, 50), symbolSize=3, pxMode=True, symbolBrush=pg.mkBrush(50, 255, 50), downsample=200)
        self.plotItemPlot = pg.PlotDataItem()  # pen="#000000")
        self.plotItem.addItem(self.plotItemPlot)
        self.plotItem.addItem(self.plotItemPlotScatter)
        self.plotItem.addItem(self.plotItemPlotScatter1)
        # self.plotItem.addItem(self.plotItemPlotScatter2)

        self.plotItem.vb.origAutoRange = self.plotItem.vb.autoRange

        self.error_label = pg.LabelItem("Error during plot.", color=(200, 50, 50))
        self.error_label.hide()
        self.error_label.scale(1, -0.6)
        self.error_label.translate(0, -30)
        self.plotItem.addItem(self.error_label)

        self.no_data_label = pg.LabelItem("No data to plot.", color=(200, 50, 50))
        self.no_data_label.hide()
        self.no_data_label.scale(1, -0.6)
        self.no_data_label.translate(0, -30)
        self.plotItem.addItem(self.no_data_label)

    def _enableCustomAutoRange(self, data):
        """
        Enable custom auto range in this plot
        :param data: the data to autorange
        :return:
        """
        def newAutoRange(*args, **kwargs):
            ''' function to handle the new autorange '''
            if len(data) == 0:
                return
            bounds = [np.min(data), np.max(data)]
            self.plotItem.vb.setRange(xRange=[0, len(data)],
                yRange=[bounds[0]-0.1*(bounds[1]-bounds[0])-1, bounds[1]+0.1*(bounds[1]-bounds[0])+1], update=True)
            self.plotItem.update()
        self.plotItem.vb.autoRange = newAutoRange
        self.plotItem.autoBtn.clicked.disconnect()
        self.plotItem.autoBtn.clicked.connect(newAutoRange)

    def _disableCustomAutoRange(self):
        """
        Disable the custom autorange and reset it to default
        :return: -
        """
        self.plotItem.vb.autoRange = self.plotItem.vb.origAutoRange
        self.plotItem.autoBtn.clicked.disconnect()
        self.plotItem.autoBtn.clicked.connect(self.plotItem.autoBtnClicked)

    @QtCore.pyqtSlot(np.ndarray, tuple, tuple)
    def plot(self, data, xvalueborders=None, yvalueborders=None, autorange=True):
        """
        Plot Data. The plot type depends on the type property
        :param data: (dataset.DataSet) data to plot
        :param xvalueborders: (touple) the borders for the xvalues
        :param yvalueborders: (touple) the borders for the yvalues
        :param autorange: (bool) whether to perform a autorange or not
        :return: -
        """
        self.error_label.hide()
        self.no_data_label.hide()
        self.plotItemPlot.clear()
        self.plotItemPlotScatter.clear()
        self.plotItemPlotScatter1.clear()
        # self.plotItemPlotScatter2.clear()
        self.plotItem.resetTransform()
        self.plotItem.getAxis('bottom').setScale()
        self.img.show()
        if self._type_changed:
            self.plotItem.autoBtnClicked()
        try:
            self.removeItem(self.plotItem2)
        except Exception as e:
            if "Could not determine index of item" in str(e):
                pass
            else:
                raise

        if self.plotType == PlotType.FFT or self.plotType == PlotType.Heatmap:
            self.addItem(self.gradient_legend)
            self.gradient_legend.show()

            self.img.setImage(data, scale=[1, 1/1000.])
            self.img.resetTransform()
            self.img.setAutoDownsample(True)
            if autorange:
                self.gradient_legend.image_changed()

            if xvalueborders:
                self.plotItem.getAxis('bottom').setScale((xvalueborders[1]-xvalueborders[0])/float(data.shape[0]))
                self.img.translate(xvalueborders[0]/self.plotItem.getAxis('bottom').scale, 0)

            else:
                self.plotItem.getAxis('bottom').setScale()

            if yvalueborders:
                self.img.translate(0, yvalueborders[0])
                self.plotItem.getAxis('left').setScale(yvalueborders[1]/float(data.shape[1]))

            self._disableCustomAutoRange()
            self.plotItem.setClipToView(True)

        if self.plotType == PlotType.Compare:
            self.addItem(self.plotItem2)
            self.plotItem2.vb.linkView(self.plotItem2.vb.XAxis, self.plotItem.vb)
            self.plotItem2.vb.linkView(self.plotItem2.vb.YAxis, self.plotItem.vb)
            self.addItem(self.gradient_legend)
            self.gradient_legend.show()

            self.img.setImage(data[0], scale=[1, 1/1000.])
            self.img.show()
            self.img.resetTransform()
            self.img2.setImage(data[1], scale=[1, 1/1000.])
            self.img2.show()
            self.img2.resetTransform()
            self.img.setAutoDownsample(True)
            self.img2.setAutoDownsample(True)
            if autorange:
                self.gradient_legend.image_changed()
                pass

            if xvalueborders:
                self.plotItem.getAxis('bottom').setScale((xvalueborders[1]-xvalueborders[0])/float(data[0].shape[0]))
                self.plotItem2.getAxis('bottom').setScale((xvalueborders[1]-xvalueborders[0])/float(data[1].shape[0]))
                self.img.translate(xvalueborders[0]/self.plotItem.getAxis('bottom').scale, 0)
                self.img2.translate(xvalueborders[0]/self.plotItem.getAxis('bottom').scale, 0)

            else:
                self.plotItem.getAxis('bottom').setScale()
                self.plotItem2.getAxis('bottom').setScale()

            if yvalueborders:
                self.img.translate(0, yvalueborders[0])
                self.img2.translate(0, yvalueborders[0])
                self.plotItem.getAxis('left').setScale(yvalueborders[1]/float(data[0].shape[1]))
                self.plotItem2.getAxis('left').setScale(yvalueborders[1]/float(data[1].shape[1]))

            self._disableCustomAutoRange()
            self.plotItem.setClipToView(True)

        if self.plotType == PlotType.Trains or self.plotType == PlotType.Follow:
            self.img.clear()
            self.plotItemPlot.show()
            self.gradient_legend.hide()
            try:
                self.removeItem(self.gradient_legend)
                pass
            except Exception, e:
                if not "Could not determine index of item" in str(e):
                    raise
            self.plotItemPlotScatter.clear()
            self.plotItemPlotScatter1.clear()
            self.plotItemPlot.setData(data)

            self._enableCustomAutoRange(data)

        if self.plotType == PlotType.Combined:
            self.img.clear()
            self.plotItemPlotScatter.show()
            self.plotItemPlotScatter1.show()
            self.gradient_legend.hide()
            try:
                self.removeItem(self.gradient_legend)
                pass
            except Exception, e:
                if not "Could not determine index of item" in str(e):
                    raise
            self.plotItemPlot.clear()
            self.plotItemPlotScatter.setData(data[0].transpose())
            self.plotItemPlotScatter1.setData(data[1].transpose())

            self._enableCustomAutoRange(data[0][1])

            self.plotItem.setClipToView(False)  # NOTE: otherwise only a very small portion of data is visible :wonder:

        if self.plotType == ERROR or self.plotType == NO_DATA:
            self.img.hide()
            self.gradient_legend.hide()
            self.img2.hide()
            self.plotItemPlot.hide()
            self.plotItemPlotScatter.hide()
            self.plotItemPlotScatter1.hide()
            if self.plotType == ERROR:
                self.error_label.show()
            else:
                self.no_data_label.show()

        if autorange:
            self.plotItem.getViewBox().autoRange()
            self.plotItem.getViewBox().autoRange()

        self.labelAxes()
        if self.plotType in [PlotType.FFT, PlotType.Heatmap, PlotType.Compare]:
            self.gradient_legend.update_axis()

    def labelAxes(self):
        """
        Add Labels to the axis depending on the self.plotType property.
        :return: -
        """
        if self.plotType == PlotType.Heatmap:
            self.plotItem.setLabel('bottom', 'Turn', '')
            self.plotItem.setLabel('left', 'Bunch Position', '')
        elif self.plotType == PlotType.FFT:
            self.plotItem.setLabel('left', 'Bunch Position', '')
            self.plotItem.setLabel('bottom', 'Frequency', 'Hz')
        elif self.plotType == PlotType.Trains or self.plotType == PlotType.Follow:
            self.plotItem.setLabel('left', '', '')
            self.plotItem.setLabel('bottom', 'Sample Point', '')
        elif self.plotType == PlotType.Combined:
            self.plotItem.setLabel('left', '', '')
            self.plotItem.setLabel('bottom', '', '')
        if self.plotType == PlotType.Compare:
            self.plotItem.setLabel('bottom', 'Turn', '')
            self.plotItem.setLabel('left', 'Bunch Position', '')
            self.plotItem2.setLabel('bottom', 'Turn', '')
            self.plotItem2.setLabel('left', 'Bunch Position', '')

    def changeType(self, type):
        """
        Change the plot Type
        :param type: (int) the new type
        :return: -
        """
        if type != self.plotType:
            self._type_changed = True
        self.plotType = type
        # if type in [PlotType.FFT, PlotType.Compare, PlotType.Heatmap]:  # this will generate an error
        #     self.gradient_legend.image_changed()

class PlotWidget(kcgw.KCGWidgets):
    """
    The container Class holding various buttons and controls and the actual plots as SubPlotWidgets instance
    """
    close_signal = QtCore.pyqtSignal()
    change_type_signal = QtCore.pyqtSignal(int, int, str)

    def __init__(self, board_id, parent=None, name=None, unique_id=None, type=None, datatype=None, prefix=None, fName=None, data=None):
        """
        Initialise the Plot Widgt
        :param parent: (QWidget) the parent widget
        :param name: (str) name of this widget
        :param unique_id: (int) unique id of this widget
        :param type: (int) type of this widget
        :param datatype: (int) datatype (LIVE or FILE)
        :param prefix: (str) prefix for text in the listview in LeftBar
        :param fName: (str) the filename (this is only used if datatype is FILE)
        :param data: (dataset.DataSet) the data to be plotted
        :return: -
        """
        super(PlotWidget, self).__init__()

        self.board_id = board_id

        if name != None:
            self.theName = name
        if unique_id != None:
            self.theId = unique_id
        if type != None:
            self.theType = type
            self._old_type = type
        if datatype != None:
            self.theDataType = datatype
        if prefix != None:
            self.thePrefix = prefix
        if self.theDataType == FILE:
            self.fName = fName
        if self.theDataType == FILE:
            self.data = data
        else:
            self.data = None

        self.close_silent = False
        self.parent = parent
        self.adc = 1
        self.secadc = 2
        self._single_adc_checked = False
        self.initUI()
        self.changePlotType(self.theType)  # initially mark the correct button
        self.setWindowTitle(self.theName + " - " + plotList[type] + " - " + str(self.thePrefix))
        if self.theDataType == FILE:
            self.plot(type)
        else:
            live_plot_windows.addWindow(board_id, self)
            if board.get_board_status(board_id).last_file is not None:
                self.data = io.read_from_file(board.get_board_status(board_id).last_file)
                self.plot(type)

    def initUI(self):
        """
        Initialise the UI
        :return: -
        """
        self.plot_widget = SubPlotWidget(dType=self.theDataType)
        self.layout = QtGui.QVBoxLayout()
        self.plot_buttons_layout = QtGui.QHBoxLayout()

        self.heatmap_button = self.createButton(text=tr("Button", "Heatmap"), connect=lambda: self.plot(type=PlotType.Heatmap))
        self.fft_button = self.createButton(text=tr("Button", "FFT"), connect=lambda: self.plot(type=PlotType.FFT))
        self.trains_button = self.createButton(text=tr("Button", "Trains"), connect=lambda: self.plot(type=PlotType.Trains))
        self.combined_button = self.createButton(text=tr("Button", "Combined"), connect=lambda: self.plot(type=PlotType.Combined))
        self.compare_button = self.createButton(text=tr("Button", "Compare"), connect=lambda: self.plot(type=PlotType.Compare))
        self.follow_button = self.createButton(text=tr("Button", "Follow"), connect=lambda: self.plot(type=PlotType.Follow))
        self.type_buttons = {PlotType.Heatmap:self.heatmap_button, PlotType.FFT:self.fft_button,
                             PlotType.Trains:self.trains_button, PlotType.Combined:self.combined_button,
                             PlotType.Compare:self.compare_button, PlotType.Follow:self.follow_button}
        self.defaultButtonStyleSheet = self.heatmap_button.styleSheet()

        self.adc1 = self.createCheckbox(text="ADC 1", connect=self.change_adc)
        self.adc1.setChecked(True)
        self.adc2 = self.createCheckbox(text="ADC 2", connect=self.change_adc)
        self.adc3 = self.createCheckbox(text="ADC 3", connect=self.change_adc)
        self.adc4 = self.createCheckbox(text="ADC 4", connect=self.change_adc)

        self.plot_buttons_layout.addWidget(self.heatmap_button)
        self.plot_buttons_layout.addWidget(self.fft_button)
        self.plot_buttons_layout.addWidget(self.trains_button)
        self.plot_buttons_layout.addWidget(self.combined_button)
        self.plot_buttons_layout.addWidget(self.compare_button)
        self.plot_buttons_layout.addWidget(self.follow_button)

        self.adc_checkbox_layout = QtGui.QHBoxLayout()
        self.adc_checkbox_layout.addWidget(self.adc1)
        self.adc_checkbox_layout.addWidget(self.adc2)
        self.adc_checkbox_layout.addWidget(self.adc3)
        self.adc_checkbox_layout.addWidget(self.adc4)

        self.groupWidget = QtGui.QWidget()
        self.groupWidget.setLayout(self.adc_checkbox_layout)
        self.group = QtGui.QButtonGroup()
        self.group.addButton(self.adc1, 1)
        self.group.addButton(self.adc2, 2)
        self.group.addButton(self.adc3, 3)
        self.group.addButton(self.adc4, 4)
        self.group.setExclusive(True)

        self.compare_heading_left = self.createLabel("ADC 1")
        self.compare_heading_left.setAlignment(QtCore.Qt.AlignCenter)
        self.compare_heading_right = self.createLabel("ADC 2")
        self.compare_heading_right.setAlignment(QtCore.Qt.AlignCenter)
        self.compare_heading = QtGui.QWidget()
        self.compare_heading_layout = QtGui.QHBoxLayout()
        self.compare_heading_layout.addWidget(self.compare_heading_left)
        self.compare_heading_layout.addWidget(self.compare_heading_right)
        self.compare_heading.setLayout(self.compare_heading_layout)
        self.compare_heading.hide()
        self.adc1Compare = self.createCheckbox(text="ADC 1", connect=lambda: self.change_adc_compare(who=1))
        self.adc2Compare = self.createCheckbox(text="ADC 2", connect=lambda: self.change_adc_compare(who=2))
        self.adc3Compare = self.createCheckbox(text="ADC 3", connect=lambda: self.change_adc_compare(who=3))
        self.adc4Compare = self.createCheckbox(text="ADC 4", connect=lambda: self.change_adc_compare(who=4))
        self.adc1Compare.setChecked(True)
        self.adc2Compare.setChecked(True)

        self.adc_checkbox_compare_layout = QtGui.QHBoxLayout()
        self.adc_checkbox_compare_layout.addWidget(self.adc1Compare)
        self.adc_checkbox_compare_layout.addWidget(self.adc2Compare)
        self.adc_checkbox_compare_layout.addWidget(self.adc3Compare)
        self.adc_checkbox_compare_layout.addWidget(self.adc4Compare)

        self.groupWidgetCompare = QtGui.QWidget()
        self.groupWidgetCompare.setLayout(self.adc_checkbox_compare_layout)
        self.groupCompare = QtGui.QButtonGroup()
        self.groupCompare.setExclusive(False)
        self.groupCompare.addButton(self.adc1Compare, 1)
        self.groupCompare.addButton(self.adc2Compare, 2)
        self.groupCompare.addButton(self.adc3Compare, 3)
        self.groupCompare.addButton(self.adc4Compare, 4)
        self.groupWidgetCompare.hide()

        self.from_to_layout = QtGui.QHBoxLayout()
        self.bucket_part = QtGui.QWidget()
        self.bucket_layout = QtGui.QHBoxLayout()
        self.bucket_layout.addWidget(self.createLabel(tr("Label", "Bucket:")))
        self.bucket_spinbox = self.createSpinbox(0, config.bunches_per_turn-1, interval=1, connect=lambda: self.plot(self.theType))
        self.bucket_layout.addWidget(self.bucket_spinbox)
        self.bucket_part.setLayout(self.bucket_layout)
        self.from_spinbox = self.createSpinbox(0, 100000000, interval=100, connect=lambda: self.plot(self.theType))
        self.to_spinbox = self.createSpinbox(0, 100000000, start_value=1000, interval=100, connect=lambda: self.plot(self.theType))
        self.from_to_layout.addStretch()
        self.from_to_layout.addWidget(self.bucket_part)
        self.from_to_layout.addWidget(self.createLabel(tr("Label", "From:")))
        self.from_to_layout.addWidget(self.from_spinbox)
        self.from_to_layout.addWidget(self.createLabel(tr("Label", "To:")))
        self.from_to_layout.addWidget(self.to_spinbox)

        self.layout.addLayout(self.plot_buttons_layout)
        self.layout.addWidget(self.compare_heading)
        self.layout.addWidget(self.plot_widget)
        self.layout.addWidget(self.groupWidget)
        self.layout.addWidget(self.groupWidgetCompare)
        self.layout.addLayout(self.from_to_layout)
        self.setLayout(self.layout)

    def change_adc(self):
        """
        Change the adc for which data is plotted
        :return: -
        """
        self.adc = self.group.checkedId()
        self.change_identifier_text()
        self.plot(self.theType)

    def change_adc_compare(self, who):
        """
        Change the adcs displayed in a compare plot
        :return:
        """
        if not self._single_adc_checked:
            self.adc1Compare.setChecked(False)
            self.adc2Compare.setChecked(False)
            self.adc3Compare.setChecked(False)
            self.adc4Compare.setChecked(False)
            getattr(self, 'adc'+str(who)+'Compare').setChecked(True)
            self._single_adc_checked = True
        else:
            checked = [i.isChecked() for i in self.groupCompare.buttons()]
            self._single_adc_checked = False
            self.adc = np.where(np.array(checked) == True)[0][0] + 1  # +1 because adcs are 1 based and indices 0
            self.secadc = np.where(np.array(checked) == True)[0][1] + 1  # +1 because adcs are 1 based and indices 0
            self.compare_heading_left.setText("ADC "+str(self.adc))
            self.compare_heading_right.setText("ADC "+str(self.secadc))
            self.change_identifier_text()
            if self.theType in [ERROR, NO_DATA]:
                self.plot(self._old_type)
            else:
                self.plot(self.theType)

    def disable_buttons(self, b_bool):
        """
        Disable the buttons on this widget
        This is not used at the moment
        :param b_bool: (bool) disable(False) or enable(True)
        :return: -
        """
        self.heatmap_button.setDisabled(b_bool)
        self.fft_button.setDisabled(b_bool)
        self.trains_button.setDisabled(b_bool)
        self.combined_button.setDisabled(b_bool)

    def changePlotType(self, type):
        """
        Change the plot type to the given type.
        :param type: the new type
        """
        if type not in [ERROR, NO_DATA]:
            self._old_type = type
        self.theType = type
        self.plot_widget.changeType(type)
        for btype, button in self.type_buttons.iteritems():
            style = ""
            if btype == type:
                style += "QPushButton {background-color:lightgreen;}"
            else:
                style += self.defaultButtonStyleSheet
            style += "QPushButton:focus{background-color: lightgrey!important; border-color: lightblue;}"  # Note: this does not work
            button.setStyleSheet(style)
        if self.theType == PlotType.Follow:
            self.bucket_part.show()
        else:
            self.bucket_part.hide()
    def plot(self, type=None):
        """
        Wrapper function to call the correct plot function depending on type
        :param type: (int) the plot type
        :return: -
        """
        try:
            if type == None:
                if self.theType in [ERROR, NO_DATA]:
                    type = self._old_type
                else:
                    type = self.theType
            self.do_plot(type)
        except Exception:
            if self.data is None or len(self.data.array) == 0:
                self.changePlotType(NO_DATA)
            else:
                self.changePlotType(ERROR)
            self.plot_widget.plot(None)

    def plot_live(self, type=None, data=None):
        """
        Function to call when livedata is to be plotted
        :param type: (int) plot type
        :param data: (dataset.DataSet) data to plot
        :return: -
        """
        self.data = data
        try:
            if type == None:
                if self.theType in [ERROR, NO_DATA]:
                    type = self._old_type
                else:
                    type = self.theType
            if type == PlotType.Trains or type == PlotType.Combined or type == PlotType.Follow:
                self.do_plot(type, autorange=True)
            else:
                self.do_plot(type, autorange=False)
        except Exception:
            if len(self.data.array) == 0:
                self.changePlotType(NO_DATA)
            else:
                self.changePlotType(ERROR)
            self.plot_widget.plot(None)

    def do_plot(self, type, autorange=True):
        """
        Actually perform a plot (this calls SubPlotWidget.plot)
        :param type: (int) plot type
        :param autorange: (bool) whether to perform a autorange upon plot
        :return: -
        """
        if self.theType != type:
            if type == PlotType.Combined:
                self.groupWidget.hide()
            elif self.theType == PlotType.Combined:
                self.groupWidget.show()

            if type == PlotType.Compare:
                self.groupWidget.hide()
                self.groupWidgetCompare.show()
                self.compare_heading.show()
            else:
                self.groupWidget.show()
                self.groupWidgetCompare.hide()
                self.compare_heading.hide()

            self.changePlotType(type)
            self.setWindowTitle(self.theName + " - " + plotList[type] + " - " + str(self.thePrefix))
            self.change_identifier_text()
            if self.theType == PlotType.Compare:
                self.change_identifier_text()

        if self.theType == PlotType.Compare:
            self.groupWidget.hide()
            self.groupWidgetCompare.show()
            self.compare_heading.show()
        else:
            self.groupWidgetCompare.hide()
            self.compare_heading.hide()

        f = self.from_spinbox.value()
        t = self.to_spinbox.value() if self.to_spinbox.value() > f else f+1
        b = self.bucket_spinbox.value()
        if type == PlotType.FFT:
            self.plot_widget.plot(np.abs(self.data.fft(adc=self.adc, frm=f, to=t+1, drop_first_bin=True)).transpose(),
                                  autorange=autorange,
                                  xvalueborders=[self.data.fft_freq_dist(), self.data.fft_max_freq()])
        if type == PlotType.Heatmap:
            self.plot_widget.plot(self.data.heatmap(adc=self.adc, frm=f, to=t).transpose(), autorange=autorange)
        if type == PlotType.Compare:
            self.plot_widget.plot([self.data.heatmap(adc=self.adc, frm=f, to=t).transpose(),self.data.heatmap(adc=self.secadc, frm=f, to=t).transpose()], autorange=autorange)
        if type == PlotType.Trains:
            self.plot_widget.plot(self.data.train(adc=self.adc, frm=f, to=t), autorange=autorange)
        if type == PlotType.Follow:
            self.plot_widget.plot(self.data.follow(adc=self.adc, frm=f, to=t, bunch=b), autorange=autorange)
        if type == PlotType.Combined:
            self.plot_widget.plot(self.data.combined(frm=f, to=t), autorange=autorange)

    def change_identifier_text(self):
        """
        Change the text that identifies the plot in the left bar
        :return:
        """
        if self.theType is PlotType.Compare:
            the_text = str(self.adc)+"+"+str(self.secadc)
        else:
            the_text = str(self.adc)
        if self.theType in [ERROR, NO_DATA]:
            type = self._old_type
        else:
            type = self.theType
        if self.theDataType is LIVE:
            self.change_type_signal.emit(self.theId, type,
                                         the_text+" B: "+available_boards.get_board_name_from_id(self.board_id))
        else:
            self.change_type_signal.emit(self.theId, type, the_text)

    def closeEvent(self, event):
        """
        Event Handler to handle the event of closing this window
        :param event: QEvent
        :return: -
        """
        if not self.close_silent:
            if not self.parent.remove_plot(self.theId, silent=self.close_silent):
                event.ignore()
                return
        del self.data
        del self.plot_widget
        super(PlotWidget, self).closeEvent(event)
        if self.theDataType == LIVE:
            live_plot_windows.removeWindow(self.board_id, self)
        self.close_signal.emit()
        del self