import logging
from typing import TYPE_CHECKING, Literal

from psychopy.clock import MonotonicClock  # type: ignore
from psychopy.visual.rect import Rect  # type: ignore

if TYPE_CHECKING:
    from psychopy.visual.window import Window  # type: ignore


class ParallelPort:
    def send_trigger(self, trigger_code: int):
        logging.debug(f"Sending trigger code: {trigger_code}")


class Square(Rect):
    def __init__(self, win: Window, size: int, pos: tuple[int, int], color: tuple[int, int, int]):
        super().__init__(win, width=size, height=size, pos=pos, fillColor=color, lineColor=color)  # type: ignore

    def draw(self):
        super().draw()


class PhotoDiodeSquare(Square):
    def __init__(
        self,
        win: Window,
        size: int,
        corner: Literal["tl", "tr", "bl", "br"] = "br",
        color: tuple[int, int, int] = (1, 1, 1),
    ):
        if corner == "tl":
            pos = (-win.size[0] / 2 + size / 2, win.size[1] / 2 - size / 2)
        elif corner == "tr":
            pos = (win.size[0] / 2 - size / 2, win.size[1] / 2 - size / 2)
        elif corner == "bl":
            pos = (-win.size[0] / 2 + size / 2, -win.size[1] / 2 + size / 2)
        elif corner == "br":
            pos = (win.size[0] / 2 - size / 2, -win.size[1] / 2 + size / 2)
        super().__init__(win, size, pos, color)

    def draw(self):
        super().draw()


class ParallelPhotdiodeTrigger(PhotoDiodeSquare):
    """Class for sending a trigger signal a photodiode + simultaneous parallel port trigger."""

    def __init__(
        self,
        win: "Window",
        port: "ParallelPort | None",
        size: int = 100,
        corner: Literal["tl", "tr", "bl", "br"] = "br",
        color: tuple[int, int, int] = (1, 1, 1),
    ):
        if port is None:
            msg = "Parallel port is required to send triggers, perhaps you want to use PhotoDiodeSquare instead"
            raise ValueError(msg)
        self._port = port
        super().__init__(win, size, corner, color)

    def send_trigger(self, trigger_code: int, high_duration: float = 0.1, visual_duration: float | None = None):
        """Sends a trigger signal to the parallel port while drawing the photodiode square.

        NOTE: The trigger will be sent on the next flip. This means:
          - The trigger and the photodiode square are in sync.


        Args:
            trigger_code: The trigger code to send.
            high_duration: The number of frames to keep the trigger high.
            visual_duration: The number of frames to draw the photodiode square.
                If None, the square will be drawn for high_duration.
        """
        if visual_duration is None:
            visual_duration = high_duration
        self.draw()
        self.win.callOnFlip(self.start_schedule, trigger_code, high_duration, visual_duration)

    def start_schedule(self, trigger_code: int, high_duration: float, visual_duration: float):
        """Schedules a trigger signal to the parallel port.

        Args:
            trigger_code: The trigger code to send.
            high_duration: The number of frames to keep the trigger high.
            visual_duration: The number of frames to draw the photodiode square.
        """
        clock = MonotonicClock()
        logging.debug(f"Port trigger ({trigger_code}) set HIGH")
        self._port.send_trigger(trigger_code)
        self.draw()
        self.win.callOnFlip(self.schedule_low_trigger, trigger_code, high_duration, visual_duration, clock)

    def schedule_low_trigger(
        self, trigger_code: int, high_duration: float, visual_duration: float, clock: MonotonicClock
    ):
        """Schedules a low trigger signal to the parallel port.

        Args:
            trigger_code: The trigger code to send.
            high_duration: The number of frames to keep the trigger high.
            visual_duration: The number of frames to draw the photodiode square.
        """
        if high_duration > 0:
            if clock.getTime() >= high_duration:
                logging.debug(f"Port trigger ({trigger_code}) set LOW (0)")
                self._port.send_trigger(0)
                high_duration = 0

        if visual_duration > 0:
            if clock.getTime() < visual_duration:
                self.draw()
            else:
                logging.debug("Visual trigger hidden")
                visual_duration = 0

        if high_duration > 0 or visual_duration > 0:
            self.win.callOnFlip(self.schedule_low_trigger, trigger_code, high_duration, visual_duration, clock)