# IoT Dashboard Tkinter (DataSource選択対応版)
# ---- すべてのセンサーが「Random / CSV / MQTT」から選んだデータソースのみ更新 ----

import tkinter as tk
from tkinter import ttk, filedialog
import random
import math
import pandas as pd
from matplotlib.figure import Figure
from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg
import threading
import paho.mqtt.client as mqtt

# -------------------------------------------------------------
# SensorTab: センサーごとのタブ（データソース選択に対応）
# -------------------------------------------------------------
class SensorTab(ttk.Frame):
    def __init__(self, parent, name, app):
        super().__init__(parent)
        self.name = name
        self.app = app

        self.data = []
        self.max_points = 50

        # データソース選択
        self.datasource = tk.StringVar(value="Random")

        # CSVデータ
        self.csv_data = None
        self.csv_index = 0

        # レイアウト
        self.build_ui()

        # 更新ループ開始
        self.after(1000, self.update_loop)

    # -----------------------------------------------------
    def build_ui(self):
        top = ttk.Frame(self)
        top.pack(fill=tk.X, pady=5)

        ttk.Label(top, text=f"Sensor: {self.name}").pack(side=tk.LEFT, padx=5)

        ttk.Label(top, text="Data Source:").pack(side=tk.LEFT)
        cmb = ttk.Combobox(top, textvariable=self.datasource,
                           values=["Random", "CSV", "MQTT"], width=10)
        cmb.pack(side=tk.LEFT, padx=5)

        # CSV読み込みボタン
        btn = ttk.Button(top, text="Load CSV", command=self.load_csv)
        btn.pack(side=tk.LEFT, padx=5)

        # 閾値設定
        self.threshold = tk.DoubleVar(value=80.0)
        ttk.Label(top, text="Threshold:").pack(side=tk.LEFT, padx=5)
        ttk.Entry(top, textvariable=self.threshold, width=6).pack(side=tk.LEFT)

        # グラフ
        fig = Figure(figsize=(5,2.5))
        self.ax = fig.add_subplot(111)
        self.ax.set_title(self.name)
        self.line, = self.ax.plot([], [])
        self.ax.grid(True)

        self.canvas = FigureCanvasTkAgg(fig, master=self)
        self.canvas.get_tk_widget().pack(fill=tk.BOTH, expand=True)

    # -----------------------------------------------------
    def load_csv(self):
        path = filedialog.askopenfilename(filetypes=[("CSV Files","*.csv")])
        if not path:
            return
        df = pd.read_csv(path)
        # 数値列を自動検出
        num_cols = df.select_dtypes(include=['number']).columns
        if len(num_cols) == 0:
            return
        self.csv_data = df[num_cols[0]].tolist()
        self.csv_index = 0
        self.datasource.set("CSV")

    # -----------------------------------------------------
    def update_loop(self):
        src = self.datasource.get()

        if src == "Random":
            value = random.uniform(0, 100)

        elif src == "CSV":
            if self.csv_data:
                value = self.csv_data[self.csv_index]
                self.csv_index = (self.csv_index + 1) % len(self.csv_data)
            else:
                value = None

        elif src == "MQTT":
            # MQTTは値をpush_valueで外部から受信するためここでは更新しない
            value = None

        else:
            value = None

        if value is not None:
            self.push_value(value)

        self.after(1000, self.update_loop)

    # -----------------------------------------------------
    def push_value(self, val):
        self.data.append(val)
        if len(self.data) > self.max_points:
            self.data = self.data[-self.max_points:]

        # グラフ更新
        self.line.set_data(range(len(self.data)), self.data)
        self.ax.set_xlim(0, self.max_points)
        self.ax.set_ylim(min(self.data)-5, max(self.data)+5)
        self.canvas.draw()

        # 閾値アラート
        if val > self.threshold.get():
            self.ax.set_facecolor("mistyrose")
        else:
            self.ax.set_facecolor("white")


# -------------------------------------------------------------
# IoT Dashboard Main App
# -------------------------------------------------------------
class IoTDashboardApp:
    def __init__(self, root):
        self.root = root
        self.root.title("IoT Dashboard - DataSource Switch Version")

        # Notebook
        self.nb = ttk.Notebook(root)
        self.nb.pack(fill=tk.BOTH, expand=True)

        self.sensors = []
        for i in range(1, 4):
            tab = SensorTab(self.nb, f"Sensor-{i}", self)
            self.sensors.append(tab)
            self.nb.add(tab, text=f"Sensor-{i}")

        # MQTT クライアント
        self.mqtt_client = None

        # MQTT接続ボタン
        btn = ttk.Button(root, text="Connect MQTT", command=self.setup_mqtt)
        btn.pack(pady=5)

    # -----------------------------------------------------
    def setup_mqtt(self, broker="localhost", port=1883):
        if self.mqtt_client:
            return

        self.mqtt_client = mqtt.Client()
        self.mqtt_client.on_connect = self._on_mqtt_connect
        self.mqtt_client.on_message = self._on_mqtt_message

        t = threading.Thread(target=lambda: self._mqtt_thread(broker, port), daemon=True)
        t.start()

    # -----------------------------------------------------
    def _mqtt_thread(self, broker, port):
        try:
            self.mqtt_client.connect(broker, port, 60)
            self.mqtt_client.loop_forever()
        except:
            pass

    # -----------------------------------------------------
    def _on_mqtt_connect(self, client, userdata, flags, rc):
        for s in self.sensors:
            topic = f"sensors/{s.name}"
            client.subscribe(topic)

    # -----------------------------------------------------
    def _on_mqtt_message(self, client, userdata, msg):
        try:
            sname = msg.topic.split("/")[1]
            value = float(msg.payload.decode("utf-8"))
        except:
            return

        # 対象センサーへ値を反映
        for s in self.sensors:
            if s.name == sname:
                self.root.after(0, lambda v=value, ss=s: ss.push_value(v))
                break


# -------------------------------------------------------------
if __name__ == "__main__":
    root = tk.Tk()
    app = IoTDashboardApp(root)
    root.mainloop()