245 lines
		
	
	
		
			9.8 KiB
		
	
	
	
		
			C#
		
	
	
	
			
		
		
	
	
			245 lines
		
	
	
		
			9.8 KiB
		
	
	
	
		
			C#
		
	
	
	
| //using System.Collections.Generic;
 | |
| //using System.Reflection;
 | |
| //using UnityEngine;
 | |
| //using UnityEngine.PostProcessing;
 | |
| 
 | |
| //namespace UnityEditor.PostProcessing
 | |
| //{
 | |
| //    [CustomPropertyDrawer(typeof(TrackballGroupAttribute))]
 | |
| //    sealed class TrackballGroupDrawer : PropertyDrawer
 | |
| //    {
 | |
| //        static Material s_Material;
 | |
| 
 | |
| //        const int k_MinWheelSize = 80;
 | |
| //        const int k_MaxWheelSize = 256;
 | |
| 
 | |
| //        bool m_ResetState;
 | |
| 
 | |
| //        // Cached trackball computation methods (for speed reasons)
 | |
| //        static Dictionary<string, MethodInfo> m_TrackballMethods = new Dictionary<string, MethodInfo>();
 | |
| 
 | |
| //        internal static int m_Size
 | |
| //        {
 | |
| //            get
 | |
| //            {
 | |
| //                int size = Mathf.FloorToInt(EditorGUIUtility.currentViewWidth / 3f) - 18;
 | |
| //                size = Mathf.Clamp(size, k_MinWheelSize, k_MaxWheelSize);
 | |
| //                return size;
 | |
| //            }
 | |
| //        }
 | |
| 
 | |
| //        public override void OnGUI(Rect position, SerializedProperty property, GUIContent label)
 | |
| //        {
 | |
| //            if (s_Material == null)
 | |
| //                s_Material = new Material(Shader.Find("Hidden/Post FX/UI/Trackball")) { hideFlags = HideFlags.HideAndDontSave };
 | |
| 
 | |
| //            position = new Rect(position.x, position.y, position.width / 3f, position.height);
 | |
| //            int size = m_Size;
 | |
| //            position.x += 5f;
 | |
| 
 | |
| //            var enumerator = property.GetEnumerator();
 | |
| //            while (enumerator.MoveNext())
 | |
| //            {
 | |
| //                var prop = enumerator.Current as SerializedProperty;
 | |
| //                if (prop == null || prop.propertyType != SerializedPropertyType.Color)
 | |
| //                    continue;
 | |
| 
 | |
| //                OnWheelGUI(position, size, prop.Copy());
 | |
| //                position.x += position.width;
 | |
| //            }
 | |
| //        }
 | |
| 
 | |
| //        void OnWheelGUI(Rect position, int size, SerializedProperty property)
 | |
| //        {
 | |
| //            if (Event.current.type == EventType.Layout)
 | |
| //                return;
 | |
| 
 | |
| //            var value = property.colorValue;
 | |
| //            float offset = value.a;
 | |
| 
 | |
| //            var wheelDrawArea = position;
 | |
| //            wheelDrawArea.height = size;
 | |
| 
 | |
| //            if (wheelDrawArea.width > wheelDrawArea.height)
 | |
| //            {
 | |
| //                wheelDrawArea.x += (wheelDrawArea.width - wheelDrawArea.height) / 2.0f;
 | |
| //                wheelDrawArea.width = position.height;
 | |
| //            }
 | |
| 
 | |
| //            wheelDrawArea.width = wheelDrawArea.height;
 | |
| 
 | |
| //            float hsize = size / 2f;
 | |
| //            float radius = 0.38f * size;
 | |
| //            Vector3 hsv;
 | |
| //            Color.RGBToHSV(value, out hsv.x, out hsv.y, out hsv.z);
 | |
| 
 | |
| //            if (Event.current.type == EventType.Repaint)
 | |
| //            {
 | |
| //                float scale = EditorGUIUtility.pixelsPerPoint;
 | |
| 
 | |
| //                // Wheel texture
 | |
| //                var oldRT = RenderTexture.active;
 | |
| //                var rt = RenderTexture.GetTemporary((int)(size * scale), (int)(size * scale), 0, RenderTextureFormat.ARGB32, RenderTextureReadWrite.Linear);
 | |
| //                s_Material.SetFloat("_Offset", offset);
 | |
| //                s_Material.SetFloat("_DisabledState", GUI.enabled ? 1f : 0.5f);
 | |
| //                s_Material.SetVector("_Resolution", new Vector2(size * scale, size * scale / 2f));
 | |
| //                Graphics.Blit(null, rt, s_Material, EditorGUIUtility.isProSkin ? 0 : 1);
 | |
| //                RenderTexture.active = oldRT;
 | |
| 
 | |
| //                GUI.DrawTexture(wheelDrawArea, rt);
 | |
| //                RenderTexture.ReleaseTemporary(rt);
 | |
| 
 | |
| //                // Thumb
 | |
| //                var thumbPos = Vector2.zero;
 | |
| //                float theta = hsv.x * (Mathf.PI * 2f);
 | |
| //                float len = hsv.y * radius;
 | |
| //                thumbPos.x = Mathf.Cos(theta + (Mathf.PI / 2f));
 | |
| //                thumbPos.y = Mathf.Sin(theta - (Mathf.PI / 2f));
 | |
| //                thumbPos *= len;
 | |
| //                var thumbSize = FxStyles.wheelThumbSize;
 | |
| //                var thumbSizeH = thumbSize / 2f;
 | |
| //                FxStyles.wheelThumb.Draw(new Rect(wheelDrawArea.x + hsize + thumbPos.x - thumbSizeH.x, wheelDrawArea.y + hsize + thumbPos.y - thumbSizeH.y, thumbSize.x, thumbSize.y), false, false, false, false);
 | |
| //            }
 | |
| 
 | |
| //            var bounds = wheelDrawArea;
 | |
| //            bounds.x += hsize - radius;
 | |
| //            bounds.y += hsize - radius;
 | |
| //            bounds.width = bounds.height = radius * 2f;
 | |
| //            hsv = GetInput(bounds, hsv, radius);
 | |
| //            value = Color.HSVToRGB(hsv.x, hsv.y, 1f);
 | |
| //            value.a = offset;
 | |
| 
 | |
| //            // Luminosity booster
 | |
| //            position = wheelDrawArea;
 | |
| //            float oldX = position.x;
 | |
| //            float oldW = position.width;
 | |
| //            position.y += position.height + 4f;
 | |
| //            position.x += (position.width - (position.width * 0.75f)) / 2f;
 | |
| //            position.width = position.width * 0.75f;
 | |
| //            position.height = EditorGUIUtility.singleLineHeight;
 | |
| //            value.a = GUI.HorizontalSlider(position, value.a, -1f, 1f);
 | |
| 
 | |
| //            // Advanced controls
 | |
| //            var data = Vector3.zero;
 | |
| 
 | |
| //            if (TryGetDisplayValue(value, property, out data))
 | |
| //            {
 | |
| //                position.x = oldX;
 | |
| //                position.y += position.height;
 | |
| //                position.width = oldW / 3f;
 | |
| 
 | |
| //                using (new EditorGUI.DisabledGroupScope(true))
 | |
| //                {
 | |
| //                    GUI.Label(position, data.x.ToString("F2"), EditorStyles.centeredGreyMiniLabel);
 | |
| //                    position.x += position.width;
 | |
| //                    GUI.Label(position, data.y.ToString("F2"), EditorStyles.centeredGreyMiniLabel);
 | |
| //                    position.x += position.width;
 | |
| //                    GUI.Label(position, data.z.ToString("F2"), EditorStyles.centeredGreyMiniLabel);
 | |
| //                    position.x += position.width;
 | |
| //                }
 | |
| //            }
 | |
| 
 | |
| //            // Title
 | |
| //            position.x = oldX;
 | |
| //            position.y += position.height;
 | |
| //            position.width = oldW;
 | |
| //            GUI.Label(position, property.displayName, EditorStyles.centeredGreyMiniLabel);
 | |
| 
 | |
| //            if (m_ResetState)
 | |
| //            {
 | |
| //                value = Color.clear;
 | |
| //                m_ResetState = false;
 | |
| //            }
 | |
| 
 | |
| //            property.colorValue = value;
 | |
| //        }
 | |
| 
 | |
| //        bool TryGetDisplayValue(Color color, SerializedProperty property, out Vector3 output)
 | |
| //        {
 | |
| //            output = Vector3.zero;
 | |
| //            MethodInfo method;
 | |
| 
 | |
| //            if (!m_TrackballMethods.TryGetValue(property.name, out method))
 | |
| //            {
 | |
| //                var field = ReflectionUtils.GetFieldInfoFromPath(property.serializedObject.targetObject, property.propertyPath);
 | |
| 
 | |
| //                if (!field.IsDefined(typeof(TrackballAttribute), false))
 | |
| //                    return false;
 | |
| 
 | |
| //                var attr = (TrackballAttribute)field.GetCustomAttributes(typeof(TrackballAttribute), false)[0];
 | |
| //                const BindingFlags flags = BindingFlags.Public | BindingFlags.NonPublic | BindingFlags.Instance | BindingFlags.Static;
 | |
| //                method = typeof(ColorGradingComponent).GetMethod(attr.method, flags);
 | |
| //                m_TrackballMethods.Add(property.name, method);
 | |
| //            }
 | |
| 
 | |
| //            if (method == null)
 | |
| //                return false;
 | |
| 
 | |
| //            output = (Vector3)method.Invoke(property.serializedObject.targetObject, new object[] { color });
 | |
| //            return true;
 | |
| //        }
 | |
| 
 | |
| //        static readonly int k_ThumbHash = "colorWheelThumb".GetHashCode();
 | |
| 
 | |
| //        Vector3 GetInput(Rect bounds, Vector3 hsv, float radius)
 | |
| //        {
 | |
| //            var e = Event.current;
 | |
| //            var id = GUIUtility.GetControlID(k_ThumbHash, FocusType.Passive, bounds);
 | |
| 
 | |
| //            var mousePos = e.mousePosition;
 | |
| //            var relativePos = mousePos - new Vector2(bounds.x, bounds.y);
 | |
| 
 | |
| //            if (e.type == EventType.MouseDown && GUIUtility.hotControl == 0 && bounds.Contains(mousePos))
 | |
| //            {
 | |
| //                if (e.button == 0)
 | |
| //                {
 | |
| //                    var center = new Vector2(bounds.x + radius, bounds.y + radius);
 | |
| //                    float dist = Vector2.Distance(center, mousePos);
 | |
| 
 | |
| //                    if (dist <= radius)
 | |
| //                    {
 | |
| //                        e.Use();
 | |
| //                        GetWheelHueSaturation(relativePos.x, relativePos.y, radius, out hsv.x, out hsv.y);
 | |
| //                        GUIUtility.hotControl = id;
 | |
| //                        GUI.changed = true;
 | |
| //                    }
 | |
| //                }
 | |
| //                else if (e.button == 1)
 | |
| //                {
 | |
| //                    e.Use();
 | |
| //                    GUI.changed = true;
 | |
| //                    m_ResetState = true;
 | |
| //                }
 | |
| //            }
 | |
| //            else if (e.type == EventType.MouseDrag && e.button == 0 && GUIUtility.hotControl == id)
 | |
| //            {
 | |
| //                e.Use();
 | |
| //                GUI.changed = true;
 | |
| //                GetWheelHueSaturation(relativePos.x, relativePos.y, radius, out hsv.x, out hsv.y);
 | |
| //            }
 | |
| //            else if (e.rawType == EventType.MouseUp && e.button == 0 && GUIUtility.hotControl == id)
 | |
| //            {
 | |
| //                e.Use();
 | |
| //                GUIUtility.hotControl = 0;
 | |
| //            }
 | |
| 
 | |
| //            return hsv;
 | |
| //        }
 | |
| 
 | |
| //        void GetWheelHueSaturation(float x, float y, float radius, out float hue, out float saturation)
 | |
| //        {
 | |
| //            float dx = (x - radius) / radius;
 | |
| //            float dy = (y - radius) / radius;
 | |
| //            float d = Mathf.Sqrt(dx * dx + dy * dy);
 | |
| //            hue = Mathf.Atan2(dx, -dy);
 | |
| //            hue = 1f - ((hue > 0) ? hue : (Mathf.PI * 2f) + hue) / (Mathf.PI * 2f);
 | |
| //            saturation = Mathf.Clamp01(d);
 | |
| //        }
 | |
| 
 | |
| //        public override float GetPropertyHeight(SerializedProperty property, GUIContent label)
 | |
| //        {
 | |
| //            return m_Size + 4f * 2f + EditorGUIUtility.singleLineHeight * 3f;
 | |
| //        }
 | |
| //    }
 | |
| //}
 |