RobustToolbox/Robust.Client/Utility/DirExt.cs

using System;
using Robust.Client.Graphics;
using Robust.Shared.Maths;
using RSIDirection = Robust.Client.Graphics.RSI.State.Direction;

namespace Robust.Client.Utility
{
    public static class DirExt
    {
        public static Direction Convert(this RSIDirection dir)
        {
            switch (dir)
            {
                case RSIDirection.South:
                    return Direction.South;

                case RSIDirection.North:
                    return Direction.North;

                case RSIDirection.East:
                    return Direction.East;

                case RSIDirection.West:
                    return Direction.West;

                case RSIDirection.SouthEast:
                    return Direction.SouthEast;

                case RSIDirection.SouthWest:
                    return Direction.SouthWest;

                case RSIDirection.NorthEast:
                    return Direction.NorthEast;

                case RSIDirection.NorthWest:
                    return Direction.NorthWest;
            }

            throw new ArgumentException($"Unknown RSI dir: {dir}.", nameof(dir));
        }

        /// <summary>
        /// 'Rounds' a diagonal direction down to a cardinal direction
        /// </summary>
        /// <param name="dir">The direction to round</param>
        /// <returns><paramref name="dir"/> if it's a cardinal direction, otherwise either north or
        /// south.</returns>
        public static RSIDirection RoundToCardinal(this RSIDirection dir)
        {
            switch (dir)
            {
                case RSIDirection.NorthEast:
                case RSIDirection.NorthWest:
                    return RSIDirection.North;

                case RSIDirection.SouthEast:
                case RSIDirection.SouthWest:
                    return RSIDirection.South;

                default:
                    return dir;
            }
        }

        public static RSIDirection Convert(this Direction dir, RSI.State.DirectionType type)
        {
            // Round down to a four-way direction if appropriate
            if (type != RSI.State.DirectionType.Dir8)
            {
                return dir.Convert(RSI.State.DirectionType.Dir8).RoundToCardinal();
            }

            switch (dir)
            {
                case Direction.North:
                    return RSIDirection.North;

                case Direction.South:
                    return RSIDirection.South;

                case Direction.East:
                    return RSIDirection.East;

                case Direction.West:
                    return RSIDirection.West;

                case Direction.SouthEast:
                    return RSIDirection.SouthEast;

                case Direction.SouthWest:
                    return RSIDirection.SouthWest;

                case Direction.NorthEast:
                    return RSIDirection.NorthEast;

                case Direction.NorthWest:
                    return RSIDirection.NorthWest;
            }

            throw new ArgumentException($"Unknown dir: {dir}.", nameof(dir));
        }

        public static Direction TurnCw(this Direction dir)
        {
            return (Direction)(((int)dir - 1) % 8);
        }

        public static Direction TurnCcw(this Direction dir)
        {
            return (Direction)(((int)dir + 1) % 8);
        }
    }
}