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;
|
}
|
}
|
}
|
