Source code for siriushla.widgets.matplotlib

"""Derive matplotlib FigureCanvas to resize labels in ZoomIn/ZoomOut."""
from qtpy.QtCore import QEvent
from qtpy.QtWidgets import QApplication, QSizePolicy, QWidget, QVBoxLayout
from matplotlib.backends.backend_qt5agg import \
    FigureCanvasQTAgg as FigureCanvas, NavigationToolbar2QT
from matplotlib.figure import Figure




class MatplotlibCanvas(FigureCanvas):
    """MatplotlibCanvas class."""

    def __init__(self, figure=None, parent=None):
        """."""
        figure = figure or Figure()
        super().__init__(figure)
        self.setParent(parent)
        self.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
        self.updateGeometry()



    def changeEvent(self, event):
        """Reimplement changeEvent to handle labels resizing."""
        if event.type() == QEvent.FontChange:
            self.app = QApplication.instance()
            fontsize = self.app.font().pointSize()

            axes = self.figure.get_axes()
            handles = []
            labels = []
            for axis in axes:
                axis.tick_params(labelsize=fontsize)
                xl = axis.get_xlabel()
                axis.set_xlabel(xl, {'size': fontsize})
                yl = axis.get_ylabel()
                axis.set_ylabel(yl, {'size': fontsize})
                title = axis.get_title()
                axis.set_title(title, {'size': fontsize})
                legh, legl = axis.get_legend_handles_labels()
                if legl:
                    handles.extend(legh)
                    labels.extend(legl)
            if labels:
                axes[0].legend(handles, labels, fontsize=fontsize)

            self.figure.canvas.draw()






class MatplotlibWidget(QWidget):
    """MatplotlibWidget class."""

    def __init__(self, parent=None, figure=None, toolbar_position='bottom'):
        """."""
        super().__init__(parent)
        self.setLayout(QVBoxLayout())
        self.figure = figure or Figure()
        self.canvas = MatplotlibCanvas(self.figure, self)
        if toolbar_position is None:
            self.layout().addWidget(self.canvas)
            self.navtool = None
            return
        self.navtool = NavigationToolbar2QT(self.canvas, self)
        self.navtool.setObjectName('toolbar')
        if toolbar_position.lower().startswith('bottom'):
            self.layout().addWidget(self.canvas)
            self.layout().addWidget(self.navtool)
        else:
            self.layout().addWidget(self.navtool)
            self.layout().addWidget(self.canvas)