Source code for titania.QtGUI.base_tab

from abc import ABC, abstractmethod, ABCMeta
from PyQt5.QtWidgets import QWidget, QGridLayout, QVBoxLayout
from titania.common.titania_tab import TitaniaPlotTab
from titania.common.interface_tab import TitaniaTabInterface
from titania.data.data_core import TitaniaDataInterface, EmptyTitaniaData
from titania.panels.main_control_panel import EmptyControlPanel, ControlPanelInterface
from titania.plots.base_plot import PlotInterface, NavToolbarPlot
from titania.plots.line_plot import ToolbarLinePlot


[docs]class FinalMetaQtTab(type(QWidget), type(TitaniaTabInterface)):
    """
    This class helps to fix the warning of:
    'TypeError: metaclass conflict: the metaclass of a derived class must be a (non-strict) subclass of the metaclasses of all its bases'
    """
    pass


[docs]class QtTabInterface(TitaniaTabInterface, QWidget, metaclass=FinalMetaQtTab):
    def __init__(self, parent: QWidget=None):
        """
        Tab interface for Qt
        """
        QWidget.__init__(self)
        TitaniaTabInterface.__init__(self)
        self.parent = parent
        self.plot_panel_grid = None
        self.lineLayout = None
        self.grid_layout = None


[docs]class QtPlotTab(QtTabInterface, TitaniaPlotTab):

    def __init__(self, data: TitaniaDataInterface, parent: QWidget=None):
        """
        Tab interface for plotting and Qt
        """
        QtTabInterface.__init__(self, parent=parent)
        TitaniaPlotTab.__init__(self, data)


[docs]    def initiate(self):
        TitaniaPlotTab.initiate(self)



[docs]class QtBaseLayoutTab(QtPlotTab):
    def __init__(self, data: TitaniaDataInterface = None, parent=None):
        """
        Base tab with left panel layout
        """
        QtPlotTab.__init__(self, data=data, parent=parent)
        self.parent = parent
        self.plot_panel_grid = QGridLayout()
        self.lineLayout = QVBoxLayout()
        self.grid_layout = QGridLayout()

[docs]    def add_layout(self):
        self.grid_layout.addLayout(self.control_panel.get_control_panel(), 0, 0)
        self.grid_layout.addLayout(self.lineLayout, 0, 1)
        self.grid_layout.addLayout(self.plot_panel_grid, 0, 2, 2, 3)

[docs]    def add_plots_to_layout(self):
        if self.plot is not None:
            self.plot_panel_grid.addWidget(self.plot.get_as_plot_widget(), 0, 0, 1, 3)

[docs]    def set_layout(self):
        self.setLayout(self.grid_layout)

[docs]    def set_separator_line(self):
        container = QWidget(self)
        container.setStyleSheet("background-color:black;")
        container.setMinimumWidth(2)
        container.setMaximumWidth(2)
        container.showMaximized()
        self.lineLayout.addWidget(container)

[docs]    def construct(self):
        self.set_separator_line()
        self.add_layout()
        self.add_plots_to_layout()
        self.set_layout()

[docs]    def initiate(self):
        QtPlotTab.initiate(self)
        self.construct()

[docs]    def make_plot(self):
        return NavToolbarPlot(view=self)




[docs]class SimpleTab(QtBaseLayoutTab):
    def __init__(self, data=None, parent=None, control_panel=EmptyControlPanel):
        data = data if data is not None else EmptyTitaniaData()
        self.control_panel = control_panel
        QtBaseLayoutTab.__init__(self, data, parent)

[docs]    def get_title(self):
        return "Base Title"

[docs]    def create_control_panel(self):
        return self.control_panel(data=self.data, widget=self)

[docs]    def make_plot(self):
        return ToolbarLinePlot(view=self)

[docs]    def initiate(self):
        QtBaseLayoutTab.initiate(self)