"""
devices/analog_input.py

Analog Input device module — reads voltage/current channels.
Includes a simulation mode (no hardware required) for development/demo.
"""

import math
import random
import time
from typing import Any, Dict

from PyQt6.QtWidgets import (
    QWidget, QVBoxLayout, QHBoxLayout, QLabel, QComboBox,
    QDoubleSpinBox, QGroupBox, QCheckBox, QSpinBox, QFormLayout
)
from PyQt6.QtCore import Qt

from devices.base_device import BaseDevice, ChannelConfig, DeviceInfo, DeviceStatus


CHANNEL_COLORS = ["#00d4ff", "#ff6b35", "#7fff6e", "#ffcc00", "#c77dff", "#ff4d6d", "#4cc9f0", "#f72585"]


class AnalogInputDevice(BaseDevice):
    """
    Analog voltage/current input module.

    Supports up to 16 channels. In simulation mode, generates
    realistic waveforms (sine, ramp, noise) for each channel.
    Real hardware: override read_channels() with your SDK calls.
    """

    DEVICE_TYPE = "analog_input"
    ICON = "〜"

    def __init__(self, device_id: str = "ai_0", num_channels: int = 4,
                 simulate: bool = True, sample_rate_hz: float = 10.0):
        channels = [
            ChannelConfig(
                channel_id=f"ch{i}",
                name=f"AI {i}",
                unit="V",
                min_value=-10.0,
                max_value=10.0,
                alarm_low=-8.0,
                alarm_high=8.0,
                color=CHANNEL_COLORS[i % len(CHANNEL_COLORS)],
            )
            for i in range(num_channels)
        ]
        info = DeviceInfo(
            device_id=device_id,
            name="Analog Input",
            device_type=self.DEVICE_TYPE,
            description="Multi-channel analog voltage/current input",
            manufacturer="Generic",
            model="AI-16",
            icon=self.ICON,
            channels=channels,
        )
        super().__init__(info)
        self.simulate = simulate
        self.sample_rate_hz = sample_rate_hz
        self._start_time = 0.0
        # Sim parameters per channel
        self._sim_params = [
            {"freq": 0.5 + i * 0.3, "amp": 5.0, "offset": 0.0, "noise": 0.05, "mode": "sine"}
            for i in range(num_channels)
        ]

    # ------------------------------------------------------------------ #
    #  BaseDevice interface                                                #
    # ------------------------------------------------------------------ #

    def connect(self) -> bool:
        if self.simulate:
            self._start_time = time.time()
            self.status = DeviceStatus.SIMULATED
            return True
        # TODO: Replace with real hardware SDK init
        # e.g. import nidaqmx; self._task = nidaqmx.Task(); ...
        self.status = DeviceStatus.ERROR
        return False

    def disconnect(self) -> None:
        self.status = DeviceStatus.DISCONNECTED

    def read_channels(self) -> Dict[str, float]:
        if self.simulate:
            return self._simulate_read()
        # TODO: Replace with real hardware read
        return {}

    def write_channel(self, channel_id: str, value: Any) -> bool:
        # Analog inputs don't support write — subclass for AO
        return False

    def get_config_widget(self) -> QWidget:
        return AnalogInputConfigWidget(self)

    # ------------------------------------------------------------------ #
    #  Simulation                                                          #
    # ------------------------------------------------------------------ #

    def _simulate_read(self) -> Dict[str, float]:
        t = time.time() - self._start_time
        result = {}
        for i, ch in enumerate(self.info.channels):
            if not ch.enabled:
                continue
            p = self._sim_params[i]
            if p["mode"] == "sine":
                val = p["amp"] * math.sin(2 * math.pi * p["freq"] * t) + p["offset"]
            elif p["mode"] == "ramp":
                period = 1.0 / max(p["freq"], 0.01)
                val = p["amp"] * ((t % period) / period) * 2 - p["amp"] + p["offset"]
            elif p["mode"] == "square":
                val = p["amp"] * math.copysign(1, math.sin(2 * math.pi * p["freq"] * t)) + p["offset"]
            else:
                val = p["offset"]
            val += random.gauss(0, p["noise"] * p["amp"])
            val = max(ch.min_value, min(ch.max_value, val))
            result[ch.channel_id] = round(val, 4)
        return result


# ------------------------------------------------------------------ #
#  Config Widget                                                       #
# ------------------------------------------------------------------ #

class AnalogInputConfigWidget(QWidget):
    def __init__(self, device: AnalogInputDevice):
        super().__init__()
        self.device = device
        self._build_ui()

    def _build_ui(self):
        layout = QVBoxLayout(self)
        layout.setContentsMargins(0, 0, 0, 0)

        # Global settings
        global_group = QGroupBox("Device Settings")
        form = QFormLayout(global_group)

        self.sim_check = QCheckBox("Simulation Mode")
        self.sim_check.setChecked(self.device.simulate)
        form.addRow(self.sim_check)

        self.rate_spin = QDoubleSpinBox()
        self.rate_spin.setRange(0.1, 1000.0)
        self.rate_spin.setValue(self.device.sample_rate_hz)
        self.rate_spin.setSuffix(" Hz")
        form.addRow("Sample Rate:", self.rate_spin)

        layout.addWidget(global_group)

        # Per-channel
        ch_group = QGroupBox("Channel Configuration")
        ch_layout = QVBoxLayout(ch_group)

        for i, ch in enumerate(self.device.info.channels):
            row = QHBoxLayout()
            en = QCheckBox(ch.name)
            en.setChecked(ch.enabled)
            row.addWidget(en)

            mode_cb = QComboBox()
            mode_cb.addItems(["sine", "ramp", "square", "dc"])
            mode_cb.setCurrentText(self.device._sim_params[i]["mode"])
            row.addWidget(QLabel("Mode:"))
            row.addWidget(mode_cb)

            freq_sp = QDoubleSpinBox()
            freq_sp.setRange(0.01, 100.0)
            freq_sp.setValue(self.device._sim_params[i]["freq"])
            freq_sp.setSuffix(" Hz")
            row.addWidget(QLabel("Freq:"))
            row.addWidget(freq_sp)

            ch_layout.addLayout(row)

        layout.addWidget(ch_group)
        layout.addStretch()