from PySide6.QtCore import Qt
from PySide6.QtWidgets import QMainWindow, QTabWidget, QVBoxLayout

from bedit_gui.ui.generated.ui_main_window import Ui_MainWindow
from bedit_gui.views.equation_editor_widget import EquationEditorWidget
from bedit_gui.views.graph_editor_widget import GraphEditorWidget


class MainWindow(QMainWindow):
    def __init__(self) -> None:
        super().__init__()

        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)

        central_layout = QVBoxLayout(self.ui.centralwidget)
        central_layout.setContentsMargins(0, 0, 0, 0)
        self.equation_editor = EquationEditorWidget(self.ui.centralwidget)
        self.equation_editor.hide()
        central_layout.addWidget(self.equation_editor)
        self.graph_editor = GraphEditorWidget(self.ui.centralwidget)
        self.graph_editor.hide()
        central_layout.addWidget(self.graph_editor)

        self.setTabPosition(Qt.AllDockWidgetAreas, QTabWidget.North)
        self.setCorner(Qt.Corner.BottomLeftCorner, Qt.DockWidgetArea.LeftDockWidgetArea)
        self.setCorner(Qt.Corner.BottomRightCorner, Qt.DockWidgetArea.RightDockWidgetArea)