view.py

# Mantid Repository : https://github.com/mantidproject/mantid
#
# Copyright © 2019 ISIS Rutherford Appleton Laboratory UKRI,
#     NScD Oak Ridge National Laboratory, European Spallation Source
#     & Institut Laue - Langevin
# SPDX - License - Identifier: GPL - 3.0 +
#  This file is part of the mantid workbench.

from __future__ import (absolute_import, unicode_literals)

from qtpy.QtCore import Qt
from qtpy.QtWidgets import QWidget

from mantidqt.widgets.plotconfigdialog.colorselector import ColorSelector
from mantidqt.utils.qt import load_ui


class ErrorbarsTabWidgetView(QWidget):

    def __init__(self, parent=None):
        super(ErrorbarsTabWidgetView, self).__init__(parent=parent)

        self.ui = load_ui(__file__,
                          'curves_tab_errorbars_tab.ui',
                          baseinstance=self)
        self.color_selector_widget = ColorSelector(parent=self)
        self.layout.replaceWidget(self.color_dummy_widget,
                                  self.color_selector_widget)
        self.setAttribute(Qt.WA_DeleteOnClose, True)

        # Disable errorbar fields when they are hidden
        self.hide_errorbars_tickbox.stateChanged.connect(
            lambda: self.set_fields_enabled(not self.get_hide())
        )

        self.multiple_curves = False

    def get_hide(self):
        return self.hide_errorbars_tickbox.checkState() == Qt.Checked

    def set_hide(self, state):
        if state:
            self.hide_errorbars_tickbox.setCheckState(Qt.Checked)
        else:
            self.hide_errorbars_tickbox.setCheckState(Qt.Unchecked)

    def get_width(self):
        return self.width_spin_box.value()

    def set_width(self, width):
        self.width_spin_box.setValue(width)

    def get_capsize(self):
        return self.capsize_spin_box.value()

    def set_capsize(self, size):
        self.capsize_spin_box.setValue(size)

    def get_cap_thickness(self):
        return self.cap_thickness_spin_box.value()

    def set_cap_thickness(self, thickness):
        self.cap_thickness_spin_box.setValue(thickness)

    def get_error_every(self):
        return self.error_every_spin_box.value()

    def set_error_every(self, error_every):
        self.error_every_spin_box.setValue(error_every)

    def get_color(self):
        return self.color_selector_widget.get_color()

    def set_color(self, color):
        self.color_selector_widget.set_color(color)

    def set_fields_enabled(self, enable):
        """Set field entry boxes to enabled/disabled"""
        self.width_spin_box.setEnabled(enable)
        self.capsize_spin_box.setEnabled(enable)
        self.cap_thickness_spin_box.setEnabled(enable)
        self.error_every_spin_box.setEnabled(enable)
        self.color_selector_widget.setEnabled(enable)

        if self.multiple_curves:
            self.apply_to_all_button.setEnabled(enable)

    def update_fields(self, curve_props):
        self.set_hide(curve_props.hide_errors)
        self.set_width(curve_props.elinewidth)
        self.set_capsize(curve_props.capsize)
        self.set_cap_thickness(curve_props.capthick)
        self.set_error_every(curve_props.errorevery)
        self.set_color(curve_props.ecolor)

    def set_plot_all_enabled(self, enable):
        self.apply_to_all_button.setEnabled(enable)