AxibugEmuOnline/AxibugEmuOnline.Client/Assets/MyNes.Core/WaveRecorder.cs

using System;
using System.IO;
using System.Text;

namespace MyNes.Core;

public class WaveRecorder
{
	private string _fileName;

	private Stream STR;

	private bool _IsRecording;

	private int SIZE;

	private int NoOfSamples;

	private int _Time;

	private int TimeSamples;

	private short channels;

	private short bitsPerSample;

	private int Frequency;

	public int Time => _Time;

	public bool IsRecording => _IsRecording;

	public void Record(string FilePath, short channels, short bitsPerSample, int Frequency)
	{
		_fileName = FilePath;
		this.channels = channels;
		this.bitsPerSample = bitsPerSample;
		this.Frequency = Frequency;
		_Time = 0;
		STR = new FileStream(FilePath, FileMode.Create);
		ASCIIEncoding aSCIIEncoding = new ASCIIEncoding();
		STR.Write(aSCIIEncoding.GetBytes("RIFF"), 0, 4);
		STR.WriteByte(0);
		STR.WriteByte(0);
		STR.WriteByte(0);
		STR.WriteByte(0);
		STR.Write(aSCIIEncoding.GetBytes("WAVE"), 0, 4);
		STR.Write(aSCIIEncoding.GetBytes("fmt "), 0, 4);
		STR.WriteByte(16);
		STR.WriteByte(0);
		STR.WriteByte(0);
		STR.WriteByte(0);
		STR.WriteByte(1);
		STR.WriteByte(0);
		STR.WriteByte((byte)((uint)channels & 0xFFu));
		STR.WriteByte((byte)((channels & 0xFF00) >> 8));
		STR.WriteByte((byte)((uint)Frequency & 0xFFu));
		STR.WriteByte((byte)((Frequency & 0xFF00) >> 8));
		STR.WriteByte((byte)((Frequency & 0xFF0000) >> 16));
		STR.WriteByte((byte)((Frequency & 0xFF000000u) >> 24));
		int num = Frequency * channels * (bitsPerSample / 8);
		STR.WriteByte((byte)((uint)num & 0xFFu));
		STR.WriteByte((byte)((num & 0xFF00) >> 8));
		STR.WriteByte((byte)((num & 0xFF0000) >> 16));
		STR.WriteByte((byte)((num & 0xFF000000u) >> 24));
		short num2 = (short)(channels * (bitsPerSample / 8));
		STR.WriteByte((byte)((uint)num2 & 0xFFu));
		STR.WriteByte((byte)((num2 & 0xFF00) >> 8));
		STR.WriteByte((byte)((uint)bitsPerSample & 0xFFu));
		STR.WriteByte((byte)((bitsPerSample & 0xFF00) >> 8));
		STR.Write(aSCIIEncoding.GetBytes("data"), 0, 4);
		STR.WriteByte(0);
		STR.WriteByte(0);
		STR.WriteByte(0);
		STR.WriteByte(0);
		_IsRecording = true;
	}

	public void AddBuffer(ref byte[] buffer)
	{
		for (int i = 0; i < buffer.Length; i++)
		{
			switch (channels)
			{
			case 1:
				switch (bitsPerSample)
				{
				case 8:
					STR.WriteByte(buffer[i]);
					NoOfSamples++;
					TimeSamples++;
					if (TimeSamples >= Frequency)
					{
						_Time++;
						TimeSamples = 0;
					}
					break;
				case 16:
					STR.WriteByte(buffer[i]);
					i++;
					STR.WriteByte(buffer[i]);
					NoOfSamples++;
					TimeSamples++;
					if (TimeSamples >= Frequency)
					{
						_Time++;
						TimeSamples = 0;
					}
					break;
				case 32:
					STR.WriteByte(buffer[i]);
					i++;
					STR.WriteByte(buffer[i]);
					i++;
					STR.WriteByte(buffer[i]);
					i++;
					STR.WriteByte(buffer[i]);
					NoOfSamples++;
					TimeSamples++;
					if (TimeSamples >= Frequency)
					{
						_Time++;
						TimeSamples = 0;
					}
					break;
				}
				break;
			case 2:
				switch (bitsPerSample)
				{
				case 8:
					STR.WriteByte(buffer[i]);
					STR.WriteByte(buffer[i]);
					NoOfSamples++;
					TimeSamples++;
					if (TimeSamples >= Frequency)
					{
						_Time++;
						TimeSamples = 0;
					}
					break;
				case 16:
					STR.WriteByte(buffer[i]);
					STR.WriteByte(buffer[i]);
					i++;
					STR.WriteByte(buffer[i]);
					STR.WriteByte(buffer[i]);
					NoOfSamples++;
					TimeSamples++;
					if (TimeSamples >= Frequency)
					{
						_Time++;
						TimeSamples = 0;
					}
					break;
				case 32:
					STR.WriteByte(buffer[i]);
					i++;
					STR.WriteByte(buffer[i]);
					STR.WriteByte(buffer[i]);
					i++;
					STR.WriteByte(buffer[i]);
					STR.WriteByte(buffer[i]);
					i++;
					STR.WriteByte(buffer[i]);
					STR.WriteByte(buffer[i]);
					i++;
					STR.WriteByte(buffer[i]);
					NoOfSamples++;
					TimeSamples++;
					if (TimeSamples >= Frequency)
					{
						_Time++;
						TimeSamples = 0;
					}
					break;
				}
				break;
			}
		}
	}

	public void AddSample(int Sample)
	{
		if (!_IsRecording)
		{
			return;
		}
		switch (channels)
		{
		case 1:
			switch (bitsPerSample)
			{
			case 8:
				AddSample_mono_08(Sample);
				break;
			case 16:
				AddSample_mono_16(Sample);
				break;
			case 32:
				AddSample_mono_32(Sample);
				break;
			}
			break;
		case 2:
			switch (bitsPerSample)
			{
			case 8:
				AddSample_stereo_08(Sample);
				break;
			case 16:
				AddSample_stereo_16(Sample);
				break;
			case 32:
				AddSample_stereo_32(Sample);
				break;
			}
			break;
		}
		NoOfSamples++;
		TimeSamples++;
		if (TimeSamples >= Frequency)
		{
			_Time++;
			TimeSamples = 0;
		}
	}

	private void AddSample_mono_08(int Sample)
	{
		STR.WriteByte((byte)((uint)Sample & 0xFFu));
	}

	private void AddSample_mono_16(int Sample)
	{
		STR.WriteByte((byte)((Sample & 0xFF00) >> 8));
		STR.WriteByte((byte)((uint)Sample & 0xFFu));
	}

	private void AddSample_mono_32(int Sample)
	{
		STR.WriteByte((byte)((Sample & 0xFF000000u) >> 24));
		STR.WriteByte((byte)((Sample & 0xFF0000) >> 16));
		STR.WriteByte((byte)((Sample & 0xFF00) >> 8));
		STR.WriteByte((byte)((uint)Sample & 0xFFu));
	}

	private void AddSample_stereo_08(int Sample)
	{
		STR.WriteByte((byte)((uint)Sample & 0xFFu));
		STR.WriteByte((byte)((uint)Sample & 0xFFu));
	}

	private void AddSample_stereo_16(int Sample)
	{
		STR.WriteByte((byte)((Sample & 0xFF00) >> 8));
		STR.WriteByte((byte)((uint)Sample & 0xFFu));
		STR.WriteByte((byte)((Sample & 0xFF00) >> 8));
		STR.WriteByte((byte)((uint)Sample & 0xFFu));
	}

	private void AddSample_stereo_32(int Sample)
	{
		STR.WriteByte((byte)((Sample & 0xFF000000u) >> 24));
		STR.WriteByte((byte)((Sample & 0xFF0000) >> 16));
		STR.WriteByte((byte)((Sample & 0xFF00) >> 8));
		STR.WriteByte((byte)((uint)Sample & 0xFFu));
		STR.WriteByte((byte)((Sample & 0xFF000000u) >> 24));
		STR.WriteByte((byte)((Sample & 0xFF0000) >> 16));
		STR.WriteByte((byte)((Sample & 0xFF00) >> 8));
		STR.WriteByte((byte)((uint)Sample & 0xFFu));
	}

	public void Stop()
	{
		if (_IsRecording & (STR != null))
		{
			NoOfSamples *= channels * (bitsPerSample / 8);
			SIZE = NoOfSamples + 36;
			byte[] array = new byte[4];
			byte[] array2 = new byte[4];
			array = BitConverter.GetBytes(SIZE);
			if (!BitConverter.IsLittleEndian)
			{
				Array.Reverse(array);
			}
			array2 = BitConverter.GetBytes(NoOfSamples);
			if (!BitConverter.IsLittleEndian)
			{
				Array.Reverse(array2);
			}
			_IsRecording = false;
			STR.Position = 4L;
			STR.Write(array, 0, 4);
			STR.Position = 40L;
			STR.Write(array2, 0, 4);
			STR.Close();
			MyNesMain.VideoProvider.WriteInfoNotification("Sound file saved at " + Path.GetFileName(_fileName), instant: false);
		}
	}
}