using UnityEngine;

/// <summary>
/// RenderTextureの平均輝度を計算するユーティリティクラス
/// </summary>
public static class LuminanceUtility
{
    // Compute Shader用の参照(Inspector等で設定する場合は別途管理クラスが必要)
    private static ComputeShader luminanceShader;

    /// <summary>
    /// RenderTextureの平均輝度を計算します
    /// </summary>
    /// <param name="renderTexture">計算対象のRenderTexture</param>
    /// <returns>平均輝度値(0.0～1.0)。エラー時は-1.0を返す</returns>
    public static float CalculateAverageLuminance(RenderTexture renderTexture)
    {
        // 入力検証
        if (renderTexture == null || !renderTexture.IsCreated())
        {
            Debug.LogError("RenderTextureが無効です。");
            return -1.0f;
        }

        // Compute Shaderが利用可能な場合はGPU計算を使用
        if (SystemInfo.supportsComputeShaders && luminanceShader != null)
        {
            return CalculateAverageLuminanceGPU(renderTexture);
        }
        else
        {
            // フォールバック: CPU計算
            return CalculateAverageLuminanceCPU(renderTexture);
        }
    }

    /// <summary>
    /// Compute Shaderを使用した高速なGPU計算
    /// </summary>
    private static float CalculateAverageLuminanceGPU(RenderTexture renderTexture)
    {
        // Compute Shader実装の概要:
        // 1. RenderTextureを入力テクスチャとして設定
        // 2. 並列リダクションで輝度合計を計算
        // 3. ComputeBufferから結果を読み出し
        // 4. ピクセル数で割って平均を算出

        int kernelIndex = luminanceShader.FindKernel("CSMain");

        // バッファ準備
        int width = renderTexture.width;
        int height = renderTexture.height;
        ComputeBuffer resultBuffer = new ComputeBuffer(1, sizeof(float));

        // Compute Shader設定
        luminanceShader.SetTexture(kernelIndex, "InputTexture", renderTexture);
        luminanceShader.SetBuffer(kernelIndex, "ResultBuffer", resultBuffer);
        luminanceShader.SetInt("Width", width);
        luminanceShader.SetInt("Height", height);

        // 実行(スレッドグループ数は適切に計算)
        int threadGroupsX = Mathf.CeilToInt(width / 8.0f);
        int threadGroupsY = Mathf.CeilToInt(height / 8.0f);
        luminanceShader.Dispatch(kernelIndex, threadGroupsX, threadGroupsY, 1);

        // 結果取得
        float[] results = new float[1];
        resultBuffer.GetData(results);
        resultBuffer.Release();

        return results[0] / (width * height);
    }

    /// <summary>
    /// CPUベースの計算(フォールバック用)
    /// </summary>
    private static float CalculateAverageLuminanceCPU(RenderTexture renderTexture)
    {
        // 一時的なTexture2Dを作成してピクセルデータを読み出し
        RenderTexture.active = renderTexture;
        Texture2D tempTexture = new Texture2D(renderTexture.width, renderTexture.height, TextureFormat.RGBAFloat, false);
        tempTexture.ReadPixels(new Rect(0, 0, renderTexture.width, renderTexture.height), 0, 0);
        tempTexture.Apply();
        RenderTexture.active = null;

        // 全ピクセルの輝度を計算
        Color[] pixels = tempTexture.GetPixels();
        float totalLuminance = 0f;

        foreach (Color pixel in pixels)
        {
            // ITU-R BT.709 標準の輝度計算式
            // Y = 0.2126R + 0.7152G + 0.0722B
            float luminance = 0.2126f * pixel.r + 0.7152f * pixel.g + 0.0722f * pixel.b;
            totalLuminance += luminance;
        }

        // クリーンアップ
        Object.Destroy(tempTexture);

        // 平均輝度を計算
        return totalLuminance / pixels.Length;
    }

    /// <summary>
    /// Compute Shaderを設定(初期化時に呼び出し)
    /// </summary>
    public static void SetComputeShader(ComputeShader shader)
    {
        luminanceShader = shader;
    }
}