using UnityEngine.Rendering;


namespace UnityEngine.Experimental.Rendering.HDPipeline
{
    public struct ZonalHarmonicsL2
    {
        public float[] coeffs; // Must have the size of 3

        public static ZonalHarmonicsL2 GetHenyeyGreensteinPhaseFunction(float anisotropy)
        {
            float g = anisotropy;

            var zh = new ZonalHarmonicsL2();
            zh.coeffs = new float[3];

            zh.coeffs[0] = 0.5f * Mathf.Sqrt(1.0f / Mathf.PI);
            zh.coeffs[1] = 0.5f * Mathf.Sqrt(3.0f / Mathf.PI) * g;
            zh.coeffs[2] = 0.5f * Mathf.Sqrt(5.0f / Mathf.PI) * g * g;

            return zh;
        }

        public static ZonalHarmonicsL2 GetCornetteShanksPhaseFunction(float anisotropy)
        {
            float g = anisotropy;

            var zh = new ZonalHarmonicsL2();
            zh.coeffs = new float[3];

            zh.coeffs[0] = 0.282095f;
            zh.coeffs[1] = 0.293162f * g * (4.0f + (g * g)) / (2.0f + (g * g));
            zh.coeffs[2] = (0.126157f + 1.44179f * (g * g) + 0.324403f * (g * g) * (g * g)) / (2.0f + (g * g));

            return zh;
        }
    }

    public class SphericalHarmonicMath
    {
        // Ref: "Stupid Spherical Harmonics Tricks", p. 6.
        public static SphericalHarmonicsL2 Convolve(SphericalHarmonicsL2 sh, ZonalHarmonicsL2 zh)
        {
            for (int l = 0; l <= 2; l++)
            {
                float n = Mathf.Sqrt((4.0f * Mathf.PI) / (2 * l + 1));
                float k = zh.coeffs[l];
                float p = n * k;

                for (int m = -l; m <= l; m++)
                {
                    int i = l * (l + 1) + m;

                    for (int c = 0; c < 3; c++)
                    {
                        sh[c, i] *= p;
                    }
                }
            }

            return sh;
        }

        // Undoes coefficient rescaling due to the convolution with the clamped cosine kernel
        // to obtain the canonical values of SH.
        public static SphericalHarmonicsL2 UndoCosineRescaling(SphericalHarmonicsL2 sh)
        {
            const float c0 = 0.28209479177387814347f; // 1/2  * sqrt(1/Pi)
            const float c1 = 0.32573500793527994772f; // 1/3  * sqrt(3/Pi)
            const float c2 = 0.27313710764801976764f; // 1/8  * sqrt(15/Pi)
            const float c3 = 0.07884789131313000151f; // 1/16 * sqrt(5/Pi)
            const float c4 = 0.13656855382400988382f; // 1/16 * sqrt(15/Pi)

            // Compute the inverse of SphericalHarmonicsL2::kNormalizationConstants.
            // See SetSHEMapConstants() in "Stupid Spherical Harmonics Tricks".
            float[] invNormConsts = { 1 / c0, -1 / c1, 1 / c1, -1 / c1, 1 / c2, -1 / c2, 1 / c3, -1 / c2, 1 / c4 };

            for (int c = 0; c < 3; c++)
            {
                for (int i = 0; i < 9; i++)
                {
                    sh[c, i] *= invNormConsts[i];
                }
            }

            return sh;
        }

        // Premultiplies the SH with the polynomial coefficients of SH basis functions,
        // which avoids using any constants during SH evaluation.
        // The resulting evaluation takes the form:
        // (c_0 - c_6) + c_1 y + c_2 z + c_3 x + c_4 x y + c_5 y z + c_6 (3 z^2) + c_7 x z + c_8 (x^2 - y^2)
        public static SphericalHarmonicsL2 PremultiplyCoefficients(SphericalHarmonicsL2 sh)
        {
            const float k0 = 0.28209479177387814347f; // {0, 0} : 1/2 * sqrt(1/Pi)
            const float k1 = 0.48860251190291992159f; // {1, 0} : 1/2 * sqrt(3/Pi)
            const float k2 = 1.09254843059207907054f; // {2,-2} : 1/2 * sqrt(15/Pi)
            const float k3 = 0.31539156525252000603f; // {2, 0} : 1/4 * sqrt(5/Pi)
            const float k4 = 0.54627421529603953527f; // {2, 2} : 1/4 * sqrt(15/Pi)

            float[] ks = { k0, -k1, k1, -k1, k2, -k2, k3, -k2, k4 };

            for (int c = 0; c < 3; c++)
            {
                for (int i = 0; i < 9; i++)
                {
                    sh[c, i] *= ks[i];
                }
            }

            return sh;
        }

        public static SphericalHarmonicsL2 RescaleCoefficients(SphericalHarmonicsL2 sh, float scalar)
        {
            for (int c = 0; c < 3; c++)
            {
                for (int i = 0; i < 9; i++)
                {
                    sh[c, i] *= scalar;
                }
            }

            return sh;
        }

        // Packs coefficients so that we can use Peter-Pike Sloan's shader code.
        // Does not perform premultiplication with coefficients of SH basis functions.
        // See SetSHEMapConstants() in "Stupid Spherical Harmonics Tricks".
        public static Vector4[] PackCoefficients(SphericalHarmonicsL2 sh)
        {
            Vector4[] coeffs = new Vector4[7];

            // Constant + linear
            for (int c = 0; c < 3; c++)
            {
                coeffs[c].x = sh[c, 3];
                coeffs[c].y = sh[c, 1];
                coeffs[c].z = sh[c, 2];
                coeffs[c].w = sh[c, 0] - sh[c, 6];
            }

            // Quadratic (4/5)
            for (int c = 0; c < 3; c++)
            {
                coeffs[3 + c].x = sh[c, 4];
                coeffs[3 + c].y = sh[c, 5];
                coeffs[3 + c].z = sh[c, 6] * 3.0f;
                coeffs[3 + c].w = sh[c, 7];
            }

            // Quadratic (5)
            coeffs[6].x = sh[0, 8];
            coeffs[6].y = sh[1, 8];
            coeffs[6].z = sh[2, 8];
            coeffs[6].w = 1.0f;

            return coeffs;
        }
    }
}