from PyQt6.QtWidgets import QWidget, QVBoxLayout, QLabel, QScrollArea, QTableView
from PyQt6.QtCore import QAbstractTableModel, QModelIndex, Qt

import numpy as np
from cpdctrl.led_script import LedScript

import time
import datetime
class TimeLeft(QWidget):
    def __init__(self, parent=None):
        super().__init__(parent)
        self.t_end = None
        self.setLayout(QVBoxLayout())
        self.layout().addWidget(QLabel("Time left"))
        self.w_time_left = QLabel("N.A.")
        self.layout().addWidget(self.w_time_left)
        self.w_end_time = QLabel("N.A.")
        self.layout().addWidget(self.w_end_time)

    def set_end_time(self, t_end: float):
        """
        Set the end time
        Parameters
        ----------
        t_end
            The end time in seconds since epoch
        """
        self.t_end = t_end
        self.w_end_time.setText(
            datetime.datetime.fromtimestamp(t_end).strftime("%Y-%m-%d %H:%M:%S")
        )

    def update_time(self, t_now: float):
        """
        Update the time left display
        Parameters
        ----------
        t_now
           The current time in seconds since epoch
        """
        if self.t_end is None: raise RuntimeError("Update called before end time was set")
        # convert to format <days>d, <hours>h, <minutes>m, <seconds>s
        print(self.t_end, t_now, self.t_end - t_now)
        sign = 1 if self.t_end > t_now else -1
        time_left = abs(int(self.t_end - t_now))
        days = sign * int(time_left / (24 * 3600))
        time_left %= (24 * 3600)
        hours = sign * int(time_left / 3600)
        time_left %= 3600
        minutes = sign * int(time_left / 60)
        time_left %= 60
        seconds = sign * int(time_left)
        self.w_time_left.setText(f"{days}d, {hours:02}h, {minutes:02}m, {seconds:02}s")



class LedScriptTableModel(QAbstractTableModel):
    """
    A table model for the led script.

    It only allows editing values that are in the future
    """
    def __init__(self, led_script: LedScript, parent=None):
        super().__init__(parent)
        self.led_script = led_script
        self.indices = [i[0] for i in LedScript.ARRAY_DTYPE]
        self.dt = 0

    def rowCount(self, parent=None):
        return self.led_script.script[self.indices[0]].shape[0]

    def columnCount(self, parent=None):
        return len(self.indices)

    def data(self, index: QModelIndex, role: int):
        if role == Qt.ItemDataRole.DisplayRole:
            return str(self.led_script.script[self.indices[index.column()]][index.row()])

    def headerData(self, section: int, orientation: Qt.Orientation, role: int):
        if role == Qt.ItemDataRole.DisplayRole:
            if orientation == Qt.Orientation.Horizontal:
                return self.indices[section]
            else:
                return str(section)

    def setData(self, index: QModelIndex, value, role: int):
        if role == Qt.ItemDataRole.EditRole:
            self.led_script.script[self.indices[index.column()]][index.row()] = value
            return True
        return False

    def addRows(self, rowCount: int, parent: QModelIndex):
        self.beginInsertRows(parent, self.rowCount(), self.rowCount() + rowCount - 1)
        for i in self.indices:
            np.append(self.led_script.script[i], np.zeros((rowCount, self.led_script.script[i].shape[1])), axis=0)

        self.endInsertRows()
    def removeRows(self, row: int, count: int, parent: QModelIndex):
        self.beginRemoveRows(parent, row, row + count - 1)
        rows = [i for i in range(row, row+count)]
        for i in self.indices:
            np.delete(self.led_script.script[i], rows, axis=0)
        self.endRemoveRows()

    def flags(self, index: QModelIndex):
        flags = Qt.ItemFlag.ItemIsSelectable
        if index.row() >= self.led_script.get_current_index(self.dt):
            flags |= Qt.ItemFlag.ItemIsEnabled
            if index.column() in [0, 2]:
                flags |= Qt.ItemFlag.ItemIsEditable
        return flags




class LedScriptViewer(QWidget):
    def __init__(self, led_script: LedScript, parent=None):
        super().__init__(parent)
        self.led_script = None
        self.model = LedScriptTableModel(led_script, self)

        self.l_vbox = QVBoxLayout()
        self.setLayout(self.l_vbox)

        self.l_vbox.addWidget(QLabel("You may edit the dt and led values of future rows."))

        self.w_scroll = QScrollArea()
        self.l_vbox.addWidget(self.w_scroll)

        self.w_table = QTableView(self)
        self.w_table.setModel(self.model)
        self.w_table.show()
        self.w_scroll.setWidget(self.w_table)

        self.w_scroll.setWidgetResizable(True)
        self.w_scroll.setFixedHeight(200)

        self.w_time_left = TimeLeft(self)
        self.l_vbox.addWidget(self.w_time_left)
        self.l_vbox.addStretch(1)

    def set_script(self, led_script: LedScript):
        self.model = LedScriptTableModel(led_script, self)
        self.w_table.setModel(self.model)
        self.w_table.show()

    def set_relative_time(self, t: float):
        pass
        raise NotImplementedError()

    def update_time(self, t_now: float):
        self.w_time_left.update_time(t_now)