using System;
using System.Runtime.InteropServices;
using System.Text;
using Unity.Profiling.Editor;
using UnityEditor;
using UnityEngine.UIElements;
using UnityEngine.AdaptivePerformance;
using System.Reflection;
using UnityEngine;
using UnityEditor.AdaptivePerformance.UI.Editor;

internal class AdaptivePerformanceDetailsViewController : ProfilerModuleViewController
{
    const string k_UxmlResourceName = "Packages/com.unity.adaptiveperformance/Editor/Profiler/AdaptivePerformanceDetailsView.uxml";
    const string k_UxmlResourceNameScaler = "Packages/com.unity.adaptiveperformance/Editor/Profiler/AdaptivePerformanceScalerElement.uxml";

    VisualElement m_view;
    Label m_DetailsViewLabel;
    VisualElement m_Scalers;
    UsageDial m_UsageDial;
    Label m_BottleneckLabel;
    VisualElement m_BottleneckIcon;
    Label m_UsageDialLabel;
    StyleColor m_appliedScalerColor = new StyleColor(new Color(0.09f, 0.69f, 0.3f, 1f));
    StyleColor m_unappliedScalerColor = new StyleColor(new Color(0.09f, 0.3f, 0.69f, 1f));
    StyleColor m_inactiveColor = new StyleColor(new Color(0.29f, 0.69f, 0.3f, 0.3f));
    Length m_scalerOffset = new Length(-200, LengthUnit.Pixel);
    Length m_midDistance = new Length(100, LengthUnit.Pixel);
    StyleRotate m_scalerRotate = new StyleRotate(new Rotate(180));
    static readonly StyleColor k_Green = new StyleColor(new Color32(136, 176, 49, byte.MaxValue));
    static readonly StyleColor k_Yellow = new StyleColor(new Color32(221, 124, 69, byte.MaxValue));
    static readonly StyleColor k_Red = new StyleColor(new Color32(219, 89, 81, byte.MaxValue));

    public AdaptivePerformanceDetailsViewController(ProfilerWindow profilerWindow) : base(profilerWindow) {}

    protected override VisualElement CreateView()
    {
        var apDetailsView = AssetDatabase.LoadAssetAtPath<VisualTreeAsset>(k_UxmlResourceName);
        m_view = apDetailsView.Instantiate();
        m_DetailsViewLabel = m_view.Q<Label>("ap-details-view-label");
        m_Scalers = m_view.Q<VisualElement>("ap-scalers");
        m_BottleneckLabel = m_view.Q<Label>("ap-details-view-bottleneck-icon-label");
        m_BottleneckIcon = m_view.Q<VisualElement>("ap-details-view-bottleneck-icon");

        m_UsageDial = m_view.Q<UsageDial>();
        m_UsageDialLabel = m_view.Q<Label>("ap-details-view-thermal-label");
        if (m_UsageDial != null)
        {
            m_UsageDial.ShowLabel = false;
            m_UsageDial?.SetThresholds(40, 75);
        }

        // Create settings for each one of the scalers
        Type ti = typeof(AdaptivePerformanceScaler);
        var scalerTree = AssetDatabase.LoadAssetAtPath<VisualTreeAsset>(k_UxmlResourceNameScaler);
        foreach (Assembly asm in AppDomain.CurrentDomain.GetAssemblies())
        {
            foreach (Type t in asm.GetTypes())
            {
                if (ti.IsAssignableFrom(t) && !t.IsAbstract)
                {
                    var container = scalerTree.CloneTree();
                    var viewName = container.Q<Label>("ap-scaler-element-label");
                    var barFill = container.Q<VisualElement>("ap-scaler-element-bar-fill");
                    var maxLabel = container.Q<Label>("ap-scaler-element-max-label");
                    var currentLabel = container.Q<Label>("ap-scaler-element-level-label");

                    viewName.text = $"{t.Name.Replace("Adaptive", "")}";
                    viewName.name = $"{t.Name}-element-label";
                    barFill.name = $"{t.Name}-element-bar-fill";
                    maxLabel.name = $"{t.Name}-element-max-label";
                    currentLabel.name = $"{t.Name}-element-current-label";
                    currentLabel.style.bottom = m_midDistance;

                    m_Scalers.Add(container);
                }
            }
        }

        ReloadData(ProfilerWindow.selectedFrameIndex);
        ProfilerWindow.SelectedFrameIndexChanged += OnSelectedFrameIndexChanged;

        return m_view;
    }

    protected override void Dispose(bool disposing)
    {
        if (!disposing)
            return;

        ProfilerWindow.SelectedFrameIndexChanged -= OnSelectedFrameIndexChanged;
        base.Dispose(disposing);
    }

    void OnSelectedFrameIndexChanged(long selectedFrameIndex)
    {
        ReloadData(selectedFrameIndex);
    }

    void ReloadData(long selectedFrameIndex)
    {
        var selectedFrameIndexInt32 = Convert.ToInt32(selectedFrameIndex);
        using (var frameData = UnityEditorInternal.ProfilerDriver.GetRawFrameDataView(selectedFrameIndexInt32, 0))
        {
            if (frameData == null || !frameData.valid)
            {
                m_DetailsViewLabel.text = "No Adaptive Performance Frame Data.";
                return;
            }

            var thermalWarningLevel = (WarningLevel)ExtractAdaptivePerformanceCounterValueInt(frameData, "Thermal Warning Level");
            var bottleneck = (PerformanceBottleneck)ExtractAdaptivePerformanceCounterValueInt(frameData, "Bottleneck");
            m_DetailsViewLabel.text = $"CPU frametime: {ExtractAdaptivePerformanceCounterValueFloat(frameData, "CPU frametime") / 1000000.0f} ms \t\t" +
                $"Average CPU frametime: {ExtractAdaptivePerformanceCounterValueFloat(frameData, "CPU avg frametime") / 1000000.0f} ms \n" +
                $"GPU frametime: {ExtractAdaptivePerformanceCounterValueFloat(frameData, "GPU frametime") / 1000000.0f} ms \t\t" +
                $"Average GPU frametime: {ExtractAdaptivePerformanceCounterValueFloat(frameData, "GPU avg frametime") / 1000000.0f} ms \n" +
                $"Frametime: {ExtractAdaptivePerformanceCounterValueFloat(frameData, "Frametime") / 1000000.0f} ms \t\t\t" +
                $"Average frametime: {ExtractAdaptivePerformanceCounterValueFloat(frameData, "Avg frametime") / 1000000.0f} ms \n" +
                $"\n" +
                $"CPU performance level: {ExtractAdaptivePerformanceCounterValueInt(frameData, "CPU performance level")} \n" +
                $"GPU performance level: {ExtractAdaptivePerformanceCounterValueInt(frameData, "GPU performance level")} \n" +
                $"\n" +
                $"Temperature Level: {ExtractAdaptivePerformanceCounterValueFloat(frameData, "Temperature Level")} \n" +
                $"Temperature Trend: {ExtractAdaptivePerformanceCounterValueFloat(frameData, "Temperature Trend")} \n" +
                $"\n" +
                $"Thermal Warning Level: {thermalWarningLevel} \n" +
                $"Bottleneck: {bottleneck} \n";

            if (m_BottleneckLabel != null && m_BottleneckIcon != null)
            {
                if (bottleneck == PerformanceBottleneck.CPU)
                {
                    m_BottleneckIcon.style.backgroundColor = k_Red;
                    m_BottleneckLabel.text = "CPU";
                }
                else if (bottleneck == PerformanceBottleneck.GPU)
                {
                    m_BottleneckIcon.style.backgroundColor = k_Red;
                    m_BottleneckLabel.text = "GPU";
                }
                else if (bottleneck == PerformanceBottleneck.TargetFrameRate)
                {
                    m_BottleneckIcon.style.backgroundColor = k_Yellow;
                    m_BottleneckLabel.text = "Target Framerate";
                }
                else
                {
                    m_BottleneckIcon.style.backgroundColor = k_Yellow;
                    m_BottleneckLabel.text = "Unknown";
                }
            }

            if (m_UsageDial != null)
            {
                if (thermalWarningLevel == WarningLevel.NoWarning)
                {
                    m_UsageDial.Value = 25;
                    m_UsageDialLabel.text = "No Warning";
                }
                else if (thermalWarningLevel == WarningLevel.ThrottlingImminent)
                {
                    m_UsageDial.Value = 60;
                    m_UsageDialLabel.text = "Throttling Imminent";
                }
                else if (thermalWarningLevel >= WarningLevel.Throttling)
                {
                    m_UsageDial.Value = 90;
                    m_UsageDialLabel.text = "Throttling";
                }
            }

            var returnVal = new AdaptivePerformanceProfilerStats.ScalerInfo[] {};
            var scalerInfos = GetScalerFromProfilerStream(selectedFrameIndexInt32);
            unsafe
            {
                foreach (var scalerInfo in scalerInfos)
                {
                    const int arraySize = 320;
                    var arr = new byte[arraySize];
                    Marshal.Copy((IntPtr)scalerInfo.scalerName, arr, 0, arraySize);
                    var scalerName = Encoding.ASCII.GetString(arr).Replace(" ", "");
                    scalerName = scalerName.Replace("\0", "");
                    var viewName = m_Scalers.Q<Label>($"{scalerName}-element-label");
                    var barFill = m_Scalers.Q<VisualElement>($"{scalerName}-element-bar-fill");
                    var maxLabel = m_Scalers.Q<Label>($"{scalerName}-element-max-label");
                    var currentLabel = m_Scalers.Q<Label>($"{scalerName}-element-current-label");

                    if (currentLabel == null || maxLabel == null || barFill == null || viewName == null)
                        continue;

                    currentLabel.text = $"{scalerInfo.currentLevel}";
                    if (scalerInfo.enabled == 0)
                    {
                        barFill.style.backgroundColor = m_inactiveColor;
                    }
                    else
                    {
                        if (scalerInfo.applied == 1)
                            barFill.style.backgroundColor = m_appliedScalerColor;
                        else
                            barFill.style.backgroundColor = m_unappliedScalerColor;
                    }
                    var height = new Length((((float)scalerInfo.currentLevel / (float)scalerInfo.maxLevel)) * 100.0f, LengthUnit.Percent);
                    barFill.style.height = height;
                    barFill.style.bottom = m_scalerOffset;
                    barFill.style.rotate = m_scalerRotate;
                    maxLabel.text = $"{scalerInfo.maxLevel}";
                }
            }
        }
    }

    static int ExtractAdaptivePerformanceCounterValueInt(UnityEditor.Profiling.FrameDataView frameData, string counterName)
    {
        if (frameData == null || counterName.Length == 0)
            return 0;

        var counterMarkerId = frameData.GetMarkerId(counterName);
        return frameData.GetCounterValueAsInt(counterMarkerId);
    }

    static float ExtractAdaptivePerformanceCounterValueFloat(UnityEditor.Profiling.FrameDataView frameData, string counterName)
    {
        if (frameData == null || counterName.Length == 0)
            return 0;

        var counterMarkerId = frameData.GetMarkerId(counterName);
        return frameData.GetCounterValueAsFloat(counterMarkerId);
    }

    public static AdaptivePerformanceProfilerStats.ScalerInfo[] GetScalerFromProfilerStream(int frame)
    {
        using (var frameData = UnityEditorInternal.ProfilerDriver.GetRawFrameDataView(frame, 0))
        {
            var returnVal = new AdaptivePerformanceProfilerStats.ScalerInfo[] {};
            if (frameData != null)
            {
                var clientInfos =
                    frameData.GetFrameMetaData<AdaptivePerformanceProfilerStats.ScalerInfo>(AdaptivePerformanceProfilerStats.kAdaptivePerformanceProfilerModuleGuid, AdaptivePerformanceProfilerStats.kScalerDataTag);
                if (clientInfos.Length != 0)
                {
                    returnVal = clientInfos.ToArray();
                }
            }
            return returnVal;
        }
    }
}