BansheeEngine/Source/EditorManaged/Windows/Scene/Gizmos/ReflectionProbeGizmo.cs

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

namespace bs.Editor
{
    /** @addtogroup Gizmos
     *  @{
     */

    /// <summary>
    /// Handles drawing of gizmos for the <see cref="ReflectionProbe"/> component.
    /// </summary>
    internal class ReflectionProbeGizmos
    {
        /// <summary>
        /// Draws the reflection probe shape.
        /// </summary>
        /// <param name="reflProbe">Reflection probe to draw gizmos for.</param>
        [DrawGizmo(DrawGizmoFlags.Selected)]
        private static void Draw(ReflectionProbe reflProbe)
        {
            Vector3 position = reflProbe.SceneObject.Position;

            Gizmos.Color = Color.Yellow;
            switch (reflProbe.Type)
            {
                case ReflectionProbeType.Box:
                    SceneObject so = reflProbe.SceneObject;

                    Gizmos.Transform = Matrix4.TRS(so.Position, so.Rotation, Vector3.One);

                    Vector3 scaledExtents = reflProbe.Extents * so.Scale;
                    Gizmos.DrawWireCube(Vector3.Zero, scaledExtents);
                    break;
                case ReflectionProbeType.Sphere:
                    Gizmos.DrawWireSphere(position, reflProbe.Radius);
                    break;
            }
        }

        /// <summary>
        /// Draws reflection probe icon in scene view.
        /// </summary>
        /// <param name="reflProbe">Reflection probe to draw the icon for.</param>
        [DrawGizmo(DrawGizmoFlags.NotSelected | DrawGizmoFlags.Pickable)]
        private static void DrawIcon(ReflectionProbe reflProbe)
        {
            Gizmos.DrawIcon(reflProbe.SceneObject.Position, 
                EditorBuiltin.GetSceneViewIcon(SceneViewIcon.ReflectionProbe), false);
        }
    }

    /** @} */
}