|
|
- using UnityEditorInternal;
- using UnityEngine;
- using UnityEngine.PostProcessing;
-
- namespace UnityEditor.PostProcessing
- {
- public class VectorscopeMonitor : PostProcessingMonitor
- {
- static GUIContent s_MonitorTitle = new GUIContent("Vectorscope");
-
- ComputeShader m_ComputeShader;
- ComputeBuffer m_Buffer;
- Material m_Material;
- RenderTexture m_VectorscopeTexture;
- Rect m_MonitorAreaRect;
-
- public VectorscopeMonitor()
- {
- m_ComputeShader = EditorResources.Load<ComputeShader>("Monitors/VectorscopeCompute.compute");
- }
-
- public override void Dispose()
- {
- GraphicsUtils.Destroy(m_Material);
- GraphicsUtils.Destroy(m_VectorscopeTexture);
-
- if (m_Buffer != null)
- m_Buffer.Release();
-
- m_Material = null;
- m_VectorscopeTexture = null;
- m_Buffer = null;
- }
-
- public override bool IsSupported()
- {
- return m_ComputeShader != null && GraphicsUtils.supportsDX11;
- }
-
- public override GUIContent GetMonitorTitle()
- {
- return s_MonitorTitle;
- }
-
- public override void OnMonitorSettings()
- {
- EditorGUI.BeginChangeCheck();
-
- bool refreshOnPlay = m_MonitorSettings.refreshOnPlay;
- float exposure = m_MonitorSettings.vectorscopeExposure;
- bool showBackground = m_MonitorSettings.vectorscopeShowBackground;
-
- refreshOnPlay = GUILayout.Toggle(refreshOnPlay, new GUIContent(FxStyles.playIcon, "Keep refreshing the vectorscope in play mode; this may impact performances."), FxStyles.preButton);
- exposure = GUILayout.HorizontalSlider(exposure, 0.05f, 0.3f, FxStyles.preSlider, FxStyles.preSliderThumb, GUILayout.Width(40f));
- showBackground = GUILayout.Toggle(showBackground, new GUIContent(FxStyles.checkerIcon, "Show an YUV background in the vectorscope."), FxStyles.preButton);
-
- if (EditorGUI.EndChangeCheck())
- {
- Undo.RecordObject(m_BaseEditor.serializedObject.targetObject, "Vectorscope Settings Changed");
- m_MonitorSettings.refreshOnPlay = refreshOnPlay;
- m_MonitorSettings.vectorscopeExposure = exposure;
- m_MonitorSettings.vectorscopeShowBackground = showBackground;
- InternalEditorUtility.RepaintAllViews();
- }
- }
-
- public override void OnMonitorGUI(Rect r)
- {
- if (Event.current.type == EventType.Repaint)
- {
- // If m_MonitorAreaRect isn't set the preview was just opened so refresh the render to get the vectoscope data
- if (Mathf.Approximately(m_MonitorAreaRect.width, 0) && Mathf.Approximately(m_MonitorAreaRect.height, 0))
- InternalEditorUtility.RepaintAllViews();
-
- // Sizing
- float size = 0f;
-
- if (r.width < r.height)
- {
- size = m_VectorscopeTexture != null
- ? Mathf.Min(m_VectorscopeTexture.width, r.width - 35f)
- : r.width;
- }
- else
- {
- size = m_VectorscopeTexture != null
- ? Mathf.Min(m_VectorscopeTexture.height, r.height - 25f)
- : r.height;
- }
-
- m_MonitorAreaRect = new Rect(
- Mathf.Floor(r.x + r.width / 2f - size / 2f),
- Mathf.Floor(r.y + r.height / 2f - size / 2f - 5f),
- size, size
- );
-
- if (m_VectorscopeTexture != null)
- {
- m_Material.SetFloat("_Exposure", m_MonitorSettings.vectorscopeExposure);
-
- var oldActive = RenderTexture.active;
- Graphics.Blit(null, m_VectorscopeTexture, m_Material, m_MonitorSettings.vectorscopeShowBackground ? 0 : 1);
- RenderTexture.active = oldActive;
-
- Graphics.DrawTexture(m_MonitorAreaRect, m_VectorscopeTexture);
-
- var color = Color.white;
- const float kTickSize = 10f;
- const int kTickCount = 24;
-
- float radius = m_MonitorAreaRect.width / 2f;
- float midX = m_MonitorAreaRect.x + radius;
- float midY = m_MonitorAreaRect.y + radius;
- var center = new Vector2(midX, midY);
-
- // Cross
- color.a *= 0.5f;
- Handles.color = color;
- Handles.DrawLine(new Vector2(midX, m_MonitorAreaRect.y), new Vector2(midX, m_MonitorAreaRect.y + m_MonitorAreaRect.height));
- Handles.DrawLine(new Vector2(m_MonitorAreaRect.x, midY), new Vector2(m_MonitorAreaRect.x + m_MonitorAreaRect.width, midY));
-
- if (m_MonitorAreaRect.width > 100f)
- {
- color.a = 1f;
-
- // Ticks
- Handles.color = color;
- for (int i = 0; i < kTickCount; i++)
- {
- float a = (float)i / (float)kTickCount;
- float theta = a * (Mathf.PI * 2f);
- float tx = Mathf.Cos(theta + (Mathf.PI / 2f));
- float ty = Mathf.Sin(theta - (Mathf.PI / 2f));
- var innerVec = center + new Vector2(tx, ty) * (radius - kTickSize);
- var outerVec = center + new Vector2(tx, ty) * radius;
- Handles.DrawAAPolyLine(3f, innerVec, outerVec);
- }
-
- // Labels (where saturation reaches 75%)
- color.a = 1f;
- var oldColor = GUI.color;
- GUI.color = color * 2f;
-
- var point = new Vector2(-0.254f, -0.750f) * radius + center;
- var rect = new Rect(point.x - 10f, point.y - 10f, 20f, 20f);
- GUI.Label(rect, "[R]", FxStyles.tickStyleCenter);
-
- point = new Vector2(-0.497f, 0.629f) * radius + center;
- rect = new Rect(point.x - 10f, point.y - 10f, 20f, 20f);
- GUI.Label(rect, "[G]", FxStyles.tickStyleCenter);
-
- point = new Vector2(0.750f, 0.122f) * radius + center;
- rect = new Rect(point.x - 10f, point.y - 10f, 20f, 20f);
- GUI.Label(rect, "[B]", FxStyles.tickStyleCenter);
-
- point = new Vector2(-0.750f, -0.122f) * radius + center;
- rect = new Rect(point.x - 10f, point.y - 10f, 20f, 20f);
- GUI.Label(rect, "[Y]", FxStyles.tickStyleCenter);
-
- point = new Vector2(0.254f, 0.750f) * radius + center;
- rect = new Rect(point.x - 10f, point.y - 10f, 20f, 20f);
- GUI.Label(rect, "[C]", FxStyles.tickStyleCenter);
-
- point = new Vector2(0.497f, -0.629f) * radius + center;
- rect = new Rect(point.x - 10f, point.y - 10f, 20f, 20f);
- GUI.Label(rect, "[M]", FxStyles.tickStyleCenter);
- GUI.color = oldColor;
- }
- }
- }
- }
-
- public override void OnFrameData(RenderTexture source)
- {
- if (Application.isPlaying && !m_MonitorSettings.refreshOnPlay)
- return;
-
- if (Mathf.Approximately(m_MonitorAreaRect.width, 0) || Mathf.Approximately(m_MonitorAreaRect.height, 0))
- return;
-
- float ratio = (float)source.width / (float)source.height;
- int h = 384;
- int w = Mathf.FloorToInt(h * ratio);
-
- var rt = RenderTexture.GetTemporary(w, h, 0, source.format);
- Graphics.Blit(source, rt);
- ComputeVectorscope(rt);
- m_BaseEditor.Repaint();
- RenderTexture.ReleaseTemporary(rt);
- }
-
- void CreateBuffer(int width, int height)
- {
- m_Buffer = new ComputeBuffer(width * height, sizeof(uint));
- }
-
- void ComputeVectorscope(RenderTexture source)
- {
- if (m_Buffer == null)
- {
- CreateBuffer(source.width, source.height);
- }
- else if (m_Buffer.count != (source.width * source.height))
- {
- m_Buffer.Release();
- CreateBuffer(source.width, source.height);
- }
-
- var cs = m_ComputeShader;
-
- int kernel = cs.FindKernel("KVectorscopeClear");
- cs.SetBuffer(kernel, "_Vectorscope", m_Buffer);
- cs.SetVector("_Res", new Vector4(source.width, source.height, 0f, 0f));
- cs.Dispatch(kernel, Mathf.CeilToInt(source.width / 32f), Mathf.CeilToInt(source.height / 32f), 1);
-
- kernel = cs.FindKernel("KVectorscope");
- cs.SetBuffer(kernel, "_Vectorscope", m_Buffer);
- cs.SetTexture(kernel, "_Source", source);
- cs.SetInt("_IsLinear", GraphicsUtils.isLinearColorSpace ? 1 : 0);
- cs.SetVector("_Res", new Vector4(source.width, source.height, 0f, 0f));
- cs.Dispatch(kernel, Mathf.CeilToInt(source.width / 32f), Mathf.CeilToInt(source.height / 32f), 1);
-
- if (m_VectorscopeTexture == null || m_VectorscopeTexture.width != source.width || m_VectorscopeTexture.height != source.height)
- {
- GraphicsUtils.Destroy(m_VectorscopeTexture);
- m_VectorscopeTexture = new RenderTexture(source.width, source.height, 0, RenderTextureFormat.ARGB32, RenderTextureReadWrite.Linear)
- {
- hideFlags = HideFlags.DontSave,
- wrapMode = TextureWrapMode.Clamp,
- filterMode = FilterMode.Bilinear
- };
- }
-
- if (m_Material == null)
- m_Material = new Material(Shader.Find("Hidden/Post FX/Monitors/Vectorscope Render")) { hideFlags = HideFlags.DontSave };
-
- m_Material.SetBuffer("_Vectorscope", m_Buffer);
- m_Material.SetVector("_Size", new Vector2(m_VectorscopeTexture.width, m_VectorscopeTexture.height));
- }
- }
- }
|