from PyQt5 import QtWidgets, QtGui, QtCore

from widgets.flash_bootloader_widget import FlashBootloaderWidget
from widgets.data_identifier_widget import DataIdentifierWidget
from widgets.diagnostic_trouble_code_widget import DiagnosticTroubleCodeWidget


class MainWidget(QtWidgets.QWidget):

    def __init__(self):
        super().__init__()
        self.initUI()
        # self.move_center()

    def initUI(self):

        self.setFont(QtGui.QFont('Segoe UI'))
        self.setWindowTitle('Flash Bootloader')

        flash_bootloader_widget = FlashBootloaderWidget(self)
        data_identifier_widget = DataIdentifierWidget(self)
        diagnostic_trouble_code_widget = DiagnosticTroubleCodeWidget(self)

        tab_widget = QtWidgets.QTabWidget(self)
        tab_widget.setObjectName('tab_widget')
        tab_widget.addTab(flash_bootloader_widget, 'Flash Bootloader')
        tab_widget.addTab(data_identifier_widget, 'Data Identifier')
        tab_widget.addTab(diagnostic_trouble_code_widget,
                          'Diagnostic Trouble Code')

        hb0 = QtWidgets.QHBoxLayout()
        hb0.addWidget(tab_widget)

        vb = QtWidgets.QVBoxLayout()
        vb.addLayout(hb0)

        self.setLayout(vb)

        QtCore.QMetaObject.connectSlotsByName(self)

    def move_center(self):

        w = QtWidgets.QApplication.desktop().width()
        h = QtWidgets.QApplication.desktop().height()
        ww = self.width()
        hh = self.height()
        self.move((w - ww) / 2, (h - hh) / 2)

    @QtCore.pyqtSlot(int)
    def on_tab_widget_currentChanged(self, index):
        if index == 0:
            self.setWindowTitle('Flash Bootloader')
        elif index == 1:
            self.setWindowTitle('Data Identifier')
        elif index == 2:
            self.setWindowTitle('Diagnostic Trouble Code')