using AssetRipper.SourceGenerated.Extensions.Enums.Shader.VertexFormat;
using AssetRipper.SourceGenerated.Subclasses.ChannelInfo;

namespace AssetRipper.SourceGenerated.Extensions;

/*
	/// <summary>
	/// Stream index
	/// BinaryData:[Stream0][Align][Stream1][Align]...
	/// </summary>
	byte Stream { get; set; }
	/// <summary>
	/// Offset inside stream
	/// Stream:[FirstVertex: VertexOffset,NormalOffset,TangentOffset...][SecondVertex: VertexOffset,NormalOffset,TangentOffset...]...
	/// </summary>
	byte Offset { get; set; }
	/// <summary>
	/// Data format: float, int, byte
	/// </summary>
	byte Format { get; set; }
	/// <summary>
	/// An unprocessed byte value containing the data dimension
	/// </summary>
	byte Dimension { get; set; }
 */

public static class ChannelInfoExtensions
{
	public static bool IsSet(this IChannelInfo channelInfo) => channelInfo.Dimension > 0;

	/// <summary>
	/// Data dimention: Vector3, Vector2, Vector1
	/// </summary>
	public static byte GetDataDimension(this IChannelInfo channelInfo)
	{
		return (byte)(channelInfo.Dimension & 0b00001111);
	}

	/// <summary>
	/// Data dimention: Vector3, Vector2, Vector1
	/// </summary>
	public static void SetDataDimension(this IChannelInfo channelInfo, byte value)
	{
		channelInfo.Dimension = (byte)(channelInfo.Dimension & 0b11110000 | value & 0b00001111);
	}

	public static byte GetStride(this IChannelInfo channelInfo, UnityVersion version)
	{
		return channelInfo.GetVertexFormat(version).CalculateStride(version, channelInfo.GetDataDimension());
	}

	public static VertexFormat GetVertexFormat(this IChannelInfo channelInfo, UnityVersion version)
	{
		if (VertexFormatExtensions.VertexFormat2019Relevant(version))
		{
			return ((VertexFormat2019)channelInfo.Format).ToVertexFormat();
		}
		else if (VertexFormatExtensions.VertexFormat2017Relevant(version))
		{
			return ((VertexFormat2017)channelInfo.Format).ToVertexFormat();
		}
		else
		{
			return ((VertexChannelFormat)channelInfo.Format).ToVertexFormat();
		}
	}
}