BansheeEngine/Source/Scripting/MBansheeEditor/Windows/Scene/Gizmos/LightGizmos.cs

//********************************** Banshee Engine (www.banshee3d.com) **************************************************//
//**************** Copyright (c) 2016 Marko Pintera ([email protected]). All rights reserved. **********************//
using BansheeEngine;

namespace BansheeEditor
{
    /** @addtogroup Gizmos
     *  @{
     */

    /// <summary>
    /// Handles drawing of gizmos for the <see cref="Light"/> component.
    /// </summary>
    internal class LightGizmos
    {
        /// <summary>
        /// Method called by the runtime when gizmos are meant to be drawn.
        /// </summary>
        /// <param name="light">Light to draw gizmos for.</param>
        [DrawGizmo(DrawGizmoFlags.Selected)]
        private static void Draw(Light light)
        {
            Vector3 position = light.SceneObject.Position;

            Gizmos.Color = Color.Yellow;
            switch (light.Type)
            {
                case LightType.Directional:
                {
                    Vector3 right = light.SceneObject.Rotation.Rotate(Vector3.XAxis);
                    Vector3 up = light.SceneObject.Rotation.Rotate(Vector3.YAxis);
                    Vector3 forward = light.SceneObject.Forward;

                    Vector3 topLeft = position - right + up;
                    Vector3 topRight = position + right + up;
                    Vector3 botLeft = position - right - up;
                    Vector3 botRight = position + right - up;

                    Gizmos.DrawLine(topLeft, topRight);
                    Gizmos.DrawLine(topRight, botRight);
                    Gizmos.DrawLine(botRight, botLeft);
                    Gizmos.DrawLine(botLeft, topLeft);

                    Gizmos.DrawLine(topLeft, topLeft + forward*5.0f);
                    Gizmos.DrawLine(topRight, topRight + forward*5.0f);
                    Gizmos.DrawLine(botRight, botRight + forward*5.0f);
                    Gizmos.DrawLine(botLeft, botLeft + forward*5.0f);
                }
                    break;
                case LightType.Radial:
                    Gizmos.DrawWireSphere(position, light.AttenuationRadius);

                    if (light.SourceRadius > 0.0f)
                    {
                        Gizmos.Color = light.Color;
                        Gizmos.DrawSphere(position, light.SourceRadius);
                    }

                    break;
                case LightType.Spot:
                {
                    Vector3 right = light.SceneObject.Rotation.Rotate(Vector3.XAxis);
                    Vector3 up = light.SceneObject.Rotation.Rotate(Vector3.YAxis);
                    Vector3 forward = light.SceneObject.Forward;

                    float discRadius = light.AttenuationRadius * MathEx.Tan(light.SpotAngle * 0.5f);

                    Gizmos.DrawLine(position, position + forward * light.AttenuationRadius + up * discRadius);
                    Gizmos.DrawLine(position, position + forward * light.AttenuationRadius - up * discRadius);
                    Gizmos.DrawLine(position, position + forward * light.AttenuationRadius + right * discRadius);
                    Gizmos.DrawLine(position, position + forward * light.AttenuationRadius - right * discRadius);

                    float falloffDiscRadius = light.AttenuationRadius * MathEx.Tan(light.SpotAngleFalloff * 0.5f);

                    Gizmos.DrawWireDisc(position + forward * light.AttenuationRadius, forward, discRadius);
                    Gizmos.DrawWireDisc(position + forward * light.AttenuationRadius, forward, falloffDiscRadius);

                    if (light.SourceRadius > 0.0f)
                    {
                        Gizmos.Color = light.Color;
                        Gizmos.DrawDisc(position, forward, light.SourceRadius);
                    }
                }
                    break;
            }
        }
    }

    /** @} */
}