using System.IO;
using Microsoft.Xna.Framework;
using NIFLib.Enums;
using Color3 = Microsoft.Xna.Framework.Color;
using Color4 = Microsoft.Xna.Framework.Color;

namespace NIFLib
{
    public static class ReaderExtensions
    {
        /// <summary>
    	/// Read a <see cref="bool"/> from the stream depending on Nif Version
    	/// </summary>
    	/// <param name="reader">Reader</param>
    	/// <param name="version">Nif Object Version</param>
    	/// <returns>The <see cref="bool"/> from <see cref="int"/> or <see cref="byte"/></returns>
    	public static bool ReadBoolean(this BinaryReader reader, NifVersion version)
        {
            if (version < NifVersion.VER_4_1_0_1)
                return reader.ReadUInt32() != 0;

            return reader.ReadBoolean();
        }

        /// <summary>
        /// Reads a <see cref="float"/>[] from the stream
        /// </summary>
        /// <param name="reader">The <see cref="BinaryReader"/></param>
        /// <param name="length">The length.</param>
        /// <returns>The<see cref="float"/>[]</returns>
        public static float[] ReadFloatArray(this BinaryReader reader, int length)
        {
            var array = new float[length];
            for (var i = 0; i < array.Length; i++)
            {
                array[i] = reader.ReadSingle();
            }
            return array;
        }

        /// <summary>
        /// Reads a <see cref="uint"/>[] from the stream
        /// </summary>
        /// <param name="reader">The <see cref="BinaryReader"/></param>
        /// <param name="length">The length.</param>
        /// <returns>The<see cref="uint"/>[].</returns>
        public static uint[] ReadUInt32Array(this BinaryReader reader, int length)
        {
            var array = new uint[length];
            for (var i = 0; i < array.Length; i++)
            {
                array[i] = reader.ReadUInt32();
            }
            return array;
        }

        /// <summary>
        /// Reads a <see cref="ushort"/>[] from the stream
        /// </summary>
        /// <param name="reader">The <see cref="BinaryReader"/></param>
        /// <param name="length">The length.</param>
        /// <returns>The <see cref="ushort"/>[]</returns>
        public static ushort[] ReadUInt16Array(this BinaryReader reader, int length)
        {
            var array = new ushort[length];
            for (var i = 0; i < array.Length; i++)
            {
                array[i] = reader.ReadUInt16();
            }
            return array;
        }

        /// <summary>
        /// Reads a <see cref="Vector2"/> from the stream
        /// </summary>
        /// <param name="reader">The <see cref="BinaryReader"/></param>
        /// <returns>The <see cref="Vector2"/></returns>
        public static Vector2 ReadVector2(this BinaryReader reader)
        {
            return new Vector2(reader.ReadSingle(), reader.ReadSingle());
        }

        /// <summary>
        /// Reads a <see cref="Vector3"/> from the stream
        /// </summary>
        /// <param name="reader">The <see cref="BinaryReader"/></param>
        /// <returns>The <see cref="Vector3"/></returns>
        public static Vector3 ReadVector3(this BinaryReader reader)
        {
            return new Vector3(reader.ReadSingle(), reader.ReadSingle(), reader.ReadSingle());
        }

        /// <summary>
        /// Reads <see cref="Vector4"/> from the stream
        /// </summary>
        /// <param name="reader">The <see cref="BinaryReader"/></param>
        /// <returns>The <see cref="Vector4"/></returns>
        public static Vector4 ReadVector4(this BinaryReader reader)
        {
            return new Vector4(reader.ReadSingle(), reader.ReadSingle(), reader.ReadSingle(), reader.ReadSingle());
        }

        /// <summary>
        /// Reads a <see cref="Color3"/> from the stream
        /// </summary>
        /// <param name="reader">The <see cref="BinaryReader"/></param>
        /// <returns>The <see cref="Color3"/></returns>
        public static Color3 ReadColor3(this BinaryReader reader)
        {

            return new Color3(reader.ReadSingle(), reader.ReadSingle(), reader.ReadSingle());
        }

        /// <summary>
        /// Reads a <see cref="Color4"/> from the stream
        /// </summary>
        /// <param name="reader">The <see cref="BinaryReader"/></param>
        /// <returns>The <see cref="Color4"/></returns>
        public static Color4 ReadColor4(this BinaryReader reader)
        {
            return new Color4(reader.ReadSingle(), reader.ReadSingle(), reader.ReadSingle(), reader.ReadSingle());
        }

        /// <summary>
        /// Reads a <see cref="Color4"/> in byte mode from the stream
        /// </summary>
        /// <param name="reader">The <see cref="BinaryReader"/></param>
        /// <returns>The <see cref="Color4"/></returns>
        public static Color4 ReadColor4Byte(this BinaryReader reader)
        {
            return new Color4((float)reader.ReadByte() / 255f, (float)reader.ReadByte() / 255f, (float)reader.ReadByte() / 255f, (float)reader.ReadByte() / 255f);
        }

        /// <summary>
        /// Reads a <see cref="Matrix"/> from the stream
        /// </summary>
        /// <param name="reader">The <see cref="BinaryReader"/></param>
        /// <returns>The <see cref="Matrix"/></returns>
        public static Matrix ReadMatrix(this BinaryReader reader)
        {
            var identity = Matrix.Identity;
            for (var i = 0; i < 3; i++)
            {
                for (var j = 0; j < 3; j++)
                {
                    identity[j, i] = reader.ReadSingle();
                }
            }
            return identity;
        }
    }
}