using System;
using System.Collections.Generic;
using System.Reflection;
using UnityEditor.Callbacks;
using UnityEngine;
using UnityEngine.Experimental.Rendering;
using UnityEngine.Experimental.Rendering.HDPipeline;
using Object = UnityEngine.Object;

namespace UnityEditor.Experimental.Rendering.HDPipeline
{
    abstract class HDProbeEditor : Editor
    {
        static Dictionary<HDProbe, HDProbeUI> s_StateMap = new Dictionary<HDProbe, HDProbeUI>();

        internal static bool TryGetUIStateFor(HDProbe p, out HDProbeUI r)
        {
            return s_StateMap.TryGetValue(p, out r);
        }

        internal abstract HDProbe GetTarget(Object editorTarget);

        protected SerializedHDProbe m_SerializedHDProbe;
        HDProbeUI m_UIState;
        HDProbeUI[] m_UIHandleState;
        protected HDProbe[] m_TypedTargets;

        protected virtual void OnEnable()
        {
            if(m_UIState == null)
            {
                m_UIState = HDProbeUI.CreateFor(this);
            }
            m_UIState.Reset(m_SerializedHDProbe, Repaint);

            m_TypedTargets = new HDProbe[targets.Length];
            m_UIHandleState = new HDProbeUI[m_TypedTargets.Length];
            for (var i = 0; i < m_TypedTargets.Length; i++)
            {
                m_TypedTargets[i] = GetTarget(targets[i]);
                m_UIHandleState[i] = HDProbeUI.CreateFor(m_TypedTargets[i]);
                m_UIHandleState[i].Reset(m_SerializedHDProbe, null);

                s_StateMap[m_TypedTargets[i]] = m_UIHandleState[i];
            }
        }

        protected virtual void OnDisable()
        {
            for (var i = 0; i < m_TypedTargets.Length; i++)
                s_StateMap.Remove(m_TypedTargets[i]);
        }

        protected abstract void Draw(HDProbeUI s, SerializedHDProbe serialized, Editor owner);

        public override void OnInspectorGUI()
        {
            var s = m_UIState;
            var d = m_SerializedHDProbe;
            var o = this;

            s.Update();
            d.Update();

            Draw(s, d, o);

            d.Apply();
        }

        protected virtual void OnSceneGUI()
        {
            for (var i = 0; i < m_TypedTargets.Length; i++)
            {
                m_UIHandleState[i].Update();
                m_UIHandleState[i].influenceVolume.showInfluenceHandles = m_UIState.influenceVolume.isSectionExpandedShape.target;
                m_UIHandleState[i].showCaptureHandles = m_UIState.isSectionExpandedCaptureSettings.target;
                HDProbeUI.DrawHandles(m_UIHandleState[i], m_TypedTargets[i], this);
            }

            m_UIState.DoShortcutKey(this);
        }


    }
}