using NAudio.Wave;
using NAudio.Extras;
using NAudio.Wave.SampleProviders;

public class AudioPlaybackEngine : IDisposable
{
    public static readonly AudioPlaybackEngine Instance = new AudioPlaybackEngine(44100, 2);

    private readonly IWavePlayer outputDevice;
    private readonly MixingSampleProvider mixer;
    public int SampleRate { get; }

    private AudioPlaybackEngine(int sampleRate = 44100, int channelCount = 2)
    {
        SampleRate = sampleRate;
        outputDevice = new WaveOutEvent();
        
        mixer = new MixingSampleProvider(WaveFormat.CreateIeeeFloatWaveFormat(sampleRate, channelCount));
        mixer.ReadFully = true;

        outputDevice.Init(mixer);
        outputDevice.Play();
    }

    private ISampleProvider ConvertToRightChannelCount(ISampleProvider input)
    {
        if (input.WaveFormat.Channels == mixer.WaveFormat.Channels)
        {
            return input;
        }
        if (input.WaveFormat.Channels == 1 && mixer.WaveFormat.Channels == 2)
        {
            return new MonoToStereoSampleProvider(input);
        }
        throw new NotImplementedException("Not yet implemented this channel count conversion");
    }

    public void PlaySound(string fileName)
    {
        var input = new AudioFileReader(fileName);
        AddMixerInput(new WdlResamplingSampleProvider(ConvertToRightChannelCount(new AutoDisposeFileReader(input)), SampleRate));
    }

    public void PlaySound(NetworkWavSound sound)
    {
        AddMixerInput(new CachedWavProvider(sound));
    }

    public ISampleProvider ConvertSound(IWaveProvider provider) {
        ISampleProvider converted = null;
        if (provider.WaveFormat.Encoding == WaveFormatEncoding.Pcm) {
            if (provider.WaveFormat.BitsPerSample == 8) {
                converted = new Pcm8BitToSampleProvider(provider);
            } else if (provider.WaveFormat.BitsPerSample == 16) {
                converted = new Pcm16BitToSampleProvider(provider);
            } else if (provider.WaveFormat.BitsPerSample == 24) {
                converted = new Pcm24BitToSampleProvider(provider);
            } else if (provider.WaveFormat.BitsPerSample == 32) {
                converted = new Pcm32BitToSampleProvider(provider);
            }
        } else if (provider.WaveFormat.Encoding == WaveFormatEncoding.IeeeFloat) {
            if (provider.WaveFormat.BitsPerSample == 64) {
                converted = new WaveToSampleProvider64(provider);
            } else {
                converted = new WaveToSampleProvider(provider);
            }
        } else {
            throw new ArgumentException("Unsupported source encoding while adding to mixer.");
        }
        return ConvertToRightChannelCount(converted);
    }

    public void AddMixerInput(ISampleProvider input)
    {
        mixer.AddMixerInput(input);
    }

    public void AddMixerInput(IWaveProvider input)
    {
        mixer.AddMixerInput(input);
    }

    public void RemoveMixerInput(ISampleProvider sound) {
        mixer.RemoveMixerInput(sound);
    }

    public void AddOnMixerInputEnded(EventHandler<SampleProviderEventArgs> e) {
        mixer.MixerInputEnded += e;
    }

    public void Dispose() {
        outputDevice.Dispose();
    }
}