using System.IO;

namespace BizHawk
{
	public class InputRecorder : IController
	{
		private IController baseController;
		private BinaryWriter writer;

		public InputRecorder(IController baseController, BinaryWriter writer)
		{
			this.baseController = baseController;
			this.writer = writer;
		}

		public void CloseMovie()
		{
			writer.Close();
		}

		public ControllerDefinition Type
		{
			get { return baseController.Type; }
		}

		public bool this[string button]
		{
			get { return baseController[button]; }
		}

		public bool IsPressed(string button)
		{
			return baseController[button];
		}

		public float GetFloat(string name)
		{
			return baseController.GetFloat(name);
		}

		private int frame;

		public void UpdateControls(int frame)
		{
			if (this.frame != frame)
			{
				this.frame = frame;
				baseController.UpdateControls(frame);
				RecordFrame();
			}
		}

		private void RecordFrame()
		{
			int encodedValue = 0;
			for (int i = 0; i < Type.BoolButtons.Count; i++)
			{
				if (baseController[Type.BoolButtons[i]])
				{
					encodedValue |= (1 << i);
				}
			}
			writer.Seek(frame * 2, SeekOrigin.Begin);
			writer.Write((ushort)encodedValue);
		}
	}

	public class InputPlayback : IController
	{
		private ControllerDefinition def;
		private int[] input;
		private int frame;

		public InputPlayback(ControllerDefinition controllerDefinition, BinaryReader reader)
		{
			def = controllerDefinition;
			int numFrames = (int)(reader.BaseStream.Length / 2);
			input = new int[numFrames];
			for (int i = 0; i < numFrames; i++)
				input[i] = reader.ReadUInt16();
		}

		public ControllerDefinition Type
		{
			get { return def; }
		}

		public bool this[string button]
		{
			get { return IsPressed(button); }
		}

		public bool IsPressed(string button)
		{
			if (frame >= input.Length)
				return false;

			for (int i = 0; i < def.BoolButtons.Count; i++)
			{
				if (def.BoolButtons[i] == button)
				{
					return (input[frame] & (1 << i)) != 0;
				}
			}
			return false;
		}

		public void UpdateControls(int frame)
		{
			this.frame = frame;
		}

		public float GetFloat(string name)
		{
			throw new System.NotImplementedException();
		}

		public void UnpressButton(string name) { }
		public void ForceButton(string button) { }
		public void SetSticky(string button, bool sticky) { }
		public bool IsSticky(string button) { return false; }

		public bool MovieEnded { get { return frame >= input.Length; } }
	}
}