import numpy as np
import sounddevice as sd
import matplotlib.pyplot as plt
import matplotlib.animation as animation
import japanize_matplotlib

SAMPLE_RATE = 44100
BLOCK_SIZE = 1024
HISTORY_LEN = 100
THRESHOLD = 0.15
COOLDOWN_FRAMES = 15
GREEN_BACK = "#00FF00"


class StreamingCounter:
    """マイクの音量を監視し、配信向けの見た目でカウントを表示するクラス"""

    def __init__(self, fig, ax, threshold=THRESHOLD, cooldown_frames=COOLDOWN_FRAMES,
                 history_len=HISTORY_LEN):
        self.threshold = threshold
        self.cooldown_frames = cooldown_frames
        self.cooldown = 0
        self.count = 0

        self.history_len = history_len
        self.volume_history = [0.0] * history_len
        self.current_volume = 0.0

        fig.patch.set_facecolor(GREEN_BACK)
        ax.set_facecolor(GREEN_BACK)

        self.line, = ax.plot(self.volume_history, color="white", linewidth=2)
        ax.axhline(threshold, color="black", linestyle="--", label="閾値")
        ax.set_ylim(0, 1.0)
        ax.set_title("Step6: 配信向けグリーンバック演出")
        ax.legend(loc="upper right")

        self.count_text = ax.text(
            0.02, 0.92, "", transform=ax.transAxes,
            fontsize=16, color="white", fontweight="bold"
        )

    def audio_callback(self, indata, frames, time, status):
        self.current_volume = float(np.sqrt(np.mean(indata ** 2)))

    def update(self, frame):
        self.volume_history.append(self.current_volume)
        self.volume_history.pop(0)
        self.line.set_ydata(self.volume_history)

        if self.cooldown > 0:
            self.cooldown -= 1
        elif self.current_volume > self.threshold:
            self.count += 1
            self.cooldown = self.cooldown_frames

        if self.current_volume > self.threshold:
            self.line.set_color("red")
        else:
            self.line.set_color("white")

        self.count_text.set_text(f"カウント: {self.count}")
        return self.line, self.count_text


fig, ax = plt.subplots()
counter = StreamingCounter(fig, ax)

stream = sd.InputStream(
    channels=1, samplerate=SAMPLE_RATE, blocksize=BLOCK_SIZE,
    callback=counter.audio_callback,
)

with stream:
    ani = animation.FuncAnimation(fig, counter.update, interval=50)
    plt.show()

GREEN_BACK = "#00FF00"

fig.patch.set_facecolor(GREEN_BACK)  # ウィンドウ全体(外枠)
ax.set_facecolor(GREEN_BACK)         # グラフ描画エリア

if self.current_volume > self.threshold:
    self.line.set_color("red")
else:
    self.line.set_color("white")