CrowdControl/Assets/Feel/MMTools/Editor/MMCamera/MMAspectRatioSafeZonesEditor.cs

using System.Collections;
using System.Collections.Generic;
using UnityEngine;
using UnityEditor;

namespace MoreMountains.Tools
{
	/// <summary>
	/// Custom editor for the MMScreenSafeZones component
	/// </summary>
	[CustomEditor(typeof(MMAspectRatioSafeZones), true)]
	[CanEditMultipleObjects]
	public class MMAspectRatioSafeZonesEditor : Editor
	{
		static MMAspectRatioSafeZones safeZones;

		/// <summary>
		/// On enable, registers to the OnSceneGUI hook
		/// </summary>
		void OnEnable()
		{
			SceneView.duringSceneGui -= OnSceneGUI;
			safeZones = (MMAspectRatioSafeZones)target;
			SceneView.duringSceneGui += OnSceneGUI;            
		}

		/// <summary>
		/// OnSceneGUI, draws center and ratios
		/// </summary>
		/// <param name="sceneView"></param>
		private static void OnSceneGUI(SceneView sceneView)
		{
			DrawFrameCenter(sceneView);
			DrawRatios(sceneView);
		}

		/// <summary>
		/// Draws a rectangle for each ratio
		/// </summary>
		/// <param name="sceneView"></param>
		private static void DrawRatios(SceneView sceneView)
		{
			if (!safeZones.DrawRatios)
			{
				return;
			}

			Vector3 center = sceneView.pivot;

			float width = sceneView.position.width;
			float height = sceneView.position.height;

			Vector3 bottomLeft = new Vector3(center.x - width / 2f, center.y - height / 2f, 0f);
			Vector3 topRight = new Vector3(center.x + width / 2f, center.y + height / 2f, 0f);
            
			Vector3 topLeft = bottomLeft;
			topLeft.y = topRight.y;
			Vector3 bottomRight = topRight;
			bottomRight.y = bottomLeft.y;

			float size = safeZones.CameraSize;
            
			// dotted lines
			float spacing = 2f;
			Color dottedLineColor = Color.white;
			dottedLineColor.a = 0.4f;
			Handles.color = dottedLineColor;
			// top
			Handles.DrawDottedLine(new Vector3(topLeft.x, center.y + size, 0f), new Vector3(topRight.x, center.y + size, 0f), spacing);
			// bottom
			Handles.DrawDottedLine(new Vector3(topLeft.x, center.y - size, 0f), new Vector3(topRight.x, center.y - size, 0f), spacing);

			foreach (Ratio ratio in safeZones.Ratios)
			{
				if (ratio.DrawRatio)
				{
					float aspectRatio = ratio.Size.x / ratio.Size.y;

					Handles.color = ratio.RatioColor;

					// aspect ratio positions
					Vector3 ratioTopLeft =       new Vector3(center.x - size * aspectRatio, center.y + size, 0f);
					Vector3 ratioTopRight =      new Vector3(center.x + size * aspectRatio, center.y + size, 0f);
					Vector3 ratioBottomLeft =    new Vector3(center.x - size * aspectRatio, center.y - size, 0f);
					Vector3 ratioBottomRight =   new Vector3(center.x + size * aspectRatio, center.y - size, 0f);
					Vector3 ratioLabelPosition = ratioBottomLeft + 0.1f * Vector3.down + 0.1f * Vector3.right;

					// draws a label under the rectangle
					GUIStyle style = new GUIStyle();
					style.normal.textColor = ratio.RatioColor;
					style.fontSize = 8;
					Handles.Label(ratioLabelPosition, ratio.Size.x + ":" + ratio.Size.y, style);

					// draws a rectangle around the aspect ratio
					Vector3[] verts = new Vector3[] { ratioTopLeft, ratioTopRight, ratioBottomRight, ratioBottomLeft };
					Handles.DrawSolidRectangleWithOutline(verts, new Color(0, 0, 0, 0), ratio.RatioColor);

					// draws the dead zone of that ratio
					Color zoneColor = ratio.RatioColor;
					zoneColor.a = zoneColor.a * safeZones.UnsafeZonesOpacity;

					// top rectangle
					verts = new Vector3[] { topLeft, topRight, new Vector3(topLeft.x, ratioTopLeft.y, 0f), new Vector3(topRight.x, ratioTopRight.y, 0f) };
					Handles.DrawSolidRectangleWithOutline(verts, zoneColor, new Color(0, 0, 0, 0));

					// bottom rectangle
					verts = new Vector3[] { bottomLeft, new Vector3(topLeft.x, ratioBottomLeft.y, 0f), new Vector3(topRight.x, ratioBottomRight.y, 0f), bottomRight };
					Handles.DrawSolidRectangleWithOutline(verts, zoneColor, new Color(0, 0, 0, 0));

					// left rectangle
					verts = new Vector3[] { new Vector3(topLeft.x, ratioTopLeft.y, 0f), ratioTopLeft, ratioBottomLeft, new Vector3(bottomLeft.x, ratioBottomLeft.y, 0f) };
					Handles.DrawSolidRectangleWithOutline(verts, zoneColor, new Color(0, 0, 0, 0));

					// right rectangle
					verts = new Vector3[] { new Vector3(topRight.x, ratioTopRight.y, 0f), new Vector3(bottomRight.x, ratioBottomRight.y, 0f), ratioBottomRight, ratioTopRight};
					Handles.DrawSolidRectangleWithOutline(verts, zoneColor, new Color(0, 0, 0, 0));

					// dotted line left 
					Handles.DrawDottedLine(new Vector3(ratioBottomLeft.x, topLeft.y, 0f), new Vector3(ratioTopLeft.x, bottomLeft.y, 0f), spacing);
					// dotted line right 
					Handles.DrawDottedLine(new Vector3(ratioBottomRight.x, topLeft.y, 0f), new Vector3(ratioBottomRight.x, bottomLeft.y, 0f), spacing);
				}
			}
		}

		/// <summary>
		/// Draws a crosshair at the center
		/// </summary>
		/// <param name="sceneView"></param>
		private static void DrawFrameCenter(SceneView sceneView)
		{
			if (!safeZones.DrawCenterCrosshair)
			{
				return;
			}

			Vector3 center = sceneView.pivot;
			float crossHairSize = safeZones.CenterCrosshairSize;

			float reticleSize = crossHairSize / 10f;

			Handles.color = safeZones.CenterCrosshairColor;

			Vector3 crosshairTopLeft = new Vector3(center.x - crossHairSize / 2f, center.y + crossHairSize / 2f, 0f);
			Vector3 crosshairTopRight = new Vector3(center.x + crossHairSize / 2f, center.y + crossHairSize / 2f, 0f);
			Vector3 crosshairBottomLeft = new Vector3(center.x - crossHairSize / 2f, center.y - crossHairSize / 2f, 0f);
			Vector3 crosshairBottomRight = new Vector3(center.x + crossHairSize / 2f, center.y - crossHairSize / 2f, 0f);

			// cross
			Handles.DrawLine(new Vector3(center.x, center.y + crossHairSize / 2f, 0f), new Vector3(center.x, center.y - crossHairSize / 2f, 0f));
			Handles.DrawLine(new Vector3(center.x - crossHairSize / 2f, center.y, 0f), new Vector3(center.x + crossHairSize / 2f, center.y, 0f));

			// top left
			Handles.DrawLine(crosshairTopLeft, new Vector3(crosshairTopLeft.x + reticleSize, crosshairTopLeft.y, 0f));
			Handles.DrawLine(crosshairTopLeft, new Vector3(crosshairTopLeft.x, crosshairTopLeft.y - reticleSize, 0f));
			// top right
			Handles.DrawLine(crosshairTopRight, new Vector3(crosshairTopRight.x - reticleSize, crosshairTopRight.y, 0f));
			Handles.DrawLine(crosshairTopRight, new Vector3(crosshairTopRight.x, crosshairTopRight.y - reticleSize, 0f));
			// bottom left
			Handles.DrawLine(crosshairBottomLeft, new Vector3(crosshairBottomLeft.x + reticleSize, crosshairBottomLeft.y, 0f));
			Handles.DrawLine(crosshairBottomLeft, new Vector3(crosshairBottomLeft.x, crosshairBottomLeft.y + reticleSize, 0f));
			// bottom right
			Handles.DrawLine(crosshairBottomRight, new Vector3(crosshairBottomRight.x - reticleSize, crosshairBottomRight.y, 0f));
			Handles.DrawLine(crosshairBottomRight, new Vector3(crosshairBottomRight.x, crosshairBottomRight.y + reticleSize, 0f));
		}
	}
}