arfoundation-samples/Assets/Scripts/LightEstimationUI.cs

using UnityEngine;
using UnityEngine.UI;
using System.Text;

using UnityEngine.Rendering;

/// <summary>
/// A simple UI controller to display light estimation information.
/// </summary>
[RequireComponent(typeof(LightEstimation))]
public class LightEstimationUI : MonoBehaviour
{
    [Tooltip("The UI Text element used to display the estimated brightness in the physical environment.")]
    [SerializeField]
    Text m_BrightnessText;

    /// <summary>
    /// The UI Text element used to display the estimated brightness value.
    /// </summary>
    public Text brightnessText
    {
        get { return m_BrightnessText; }
        set { m_BrightnessText = brightnessText; }
    }

    [Tooltip("The UI Text element used to display the estimated color temperature in the physical environment.")]
    [SerializeField]
    Text m_ColorTemperatureText;

    /// <summary>
    /// The UI Text element used to display the estimated color temperature in the scene.
    /// </summary>
    public Text colorTemperatureText
    {
        get { return m_ColorTemperatureText; }
        set { m_ColorTemperatureText = value; }
    }

    [Tooltip("The UI Text element used to display the estimated color correction value for the physical environment.")]
    [SerializeField]
    Text m_ColorCorrectionText;

    [Tooltip("The UI Text element used to display the estimated direction of the main light for the physical environment.")]
    [SerializeField]
    Text m_MainLightDirectionText;
    public Text mainLightDirectionText
    {
        get => m_MainLightDirectionText;
        set => m_MainLightDirectionText = value;
    }

    [Tooltip("The UI Text element used to display the estimated intensity in lumens of the main light for the physical environment.")]
    [SerializeField]
    Text m_MainLightIntensityLumens;
    public Text mainLightIntensityLumens
    {
        get => m_MainLightIntensityLumens;
        set => m_MainLightIntensityLumens = value;
    }

    [Tooltip("The UI Text element used to display the estimated color of the main light for the physical environment.")]
    [SerializeField]
    Text m_MainLightColor;
    public Text mainLightColorText
    {
        get => m_MainLightColor;
        set => m_MainLightColor = value;
    }

    [Tooltip("The UI Text element used to display the estimated spherical harmonics coefficients for the physical environment.")]
    [SerializeField]
    Text m_SphericalHarmonicsText;
    public Text ambientSphericalHarmonicsText
    {
        get => m_SphericalHarmonicsText;
        set => m_SphericalHarmonicsText = value;
    }
    StringBuilder m_SphericalHarmonicsStringBuilder = new StringBuilder("");

    /// <summary>
    /// The UI Text element used to display the estimated color correction value for the scene.
    /// </summary>
    public Text colorCorrectionText
    {
        get { return m_ColorCorrectionText; }
        set { m_ColorCorrectionText = value; }
    }

    void Awake()
    {
        m_LightEstimation = GetComponent<LightEstimation>();
    }

    void Update()
    {
        SetUIValue(m_LightEstimation.brightness, brightnessText);
        SetUIValue(m_LightEstimation.colorTemperature, colorTemperatureText);
        SetUIValue(m_LightEstimation.colorCorrection, colorCorrectionText);
        SetUIValue(m_LightEstimation.mainLightDirection, mainLightDirectionText);
        SetUIValue(m_LightEstimation.mainLightColor, mainLightColorText);
        SetUIValue(m_LightEstimation.mainLightIntensityLumens, mainLightIntensityLumens);
        SetSphericalHarmonicsUIValue(m_LightEstimation.sphericalHarmonics, ambientSphericalHarmonicsText);
    }

    void SetSphericalHarmonicsUIValue(SphericalHarmonicsL2? maybeAmbientSphericalHarmonics, Text text)
    {
        if (text != null)
        {
            if (maybeAmbientSphericalHarmonics.HasValue)
            {
                m_SphericalHarmonicsStringBuilder.Clear();
                for (int i = 0; i < 3; ++i)
                {
                    if (i == 0)
                        m_SphericalHarmonicsStringBuilder.Append("R:[");
                    else if (i == 1)
                        m_SphericalHarmonicsStringBuilder.Append("G:[");
                    else
                        m_SphericalHarmonicsStringBuilder.Append("B:[");

                    for (int j = 0; j < 9; ++j)
                    {
                        m_SphericalHarmonicsStringBuilder.Append(j != 8 ? $"{maybeAmbientSphericalHarmonics.Value[i, j]}, " : $"{maybeAmbientSphericalHarmonics.Value[i, j]}]\n");
                    }
                }
                text.text = m_SphericalHarmonicsStringBuilder.ToString();
            }
            else
            {
                text.text = k_UnavailableText;
            }
        }
    }

    void SetUIValue<T>(T? displayValue, Text text) where T : struct
    {
        if (text != null)
            text.text = displayValue.HasValue ? displayValue.Value.ToString(): k_UnavailableText;
    }

    const string k_UnavailableText = "Unavailable";

    LightEstimation m_LightEstimation;
}