#include "AudioVoice.hpp"
#include "AudioVoiceEngine.hpp"
#include "logvisor/logvisor.hpp"

namespace boo
{
static logvisor::Module Log("boo::AudioVoice");

AudioVoice::AudioVoice(BaseAudioVoiceEngine& root, IAudioMix& parent,
                       IAudioVoiceCallback* cb, bool dynamicRate)
: m_root(root), m_parent(parent), m_cb(cb), m_dynamicRate(dynamicRate) {}

AudioVoice::~AudioVoice()
{
    unbindVoice();
    soxr_delete(m_src);
}

void AudioVoice::_setPitchRatio(double ratio, bool slew)
{
    if (m_dynamicRate)
    {
        soxr_error_t err = soxr_set_io_ratio(m_src, ratio * m_sampleRateIn / m_sampleRateOut, slew ? m_root.m_5msFrames : 0);
        if (err)
        {
            Log.report(logvisor::Fatal, "unable to set resampler rate: %s", soxr_strerror(err));
            m_setPitchRatio = false;
            return;
        }
    }
    m_setPitchRatio = false;
}

void AudioVoice::_midUpdate()
{
    if (m_resetSampleRate)
        _resetSampleRate(m_deferredSampleRate);
    if (m_setPitchRatio)
        _setPitchRatio(m_pitchRatio, m_slew);
}

void AudioVoice::setPitchRatio(double ratio, bool slew)
{
    m_setPitchRatio = true;
    m_pitchRatio = ratio;
    m_slew = slew;
}

void AudioVoice::resetSampleRate(double sampleRate)
{
    m_resetSampleRate = true;
    m_deferredSampleRate = sampleRate;
}

void AudioVoice::start()
{
    m_running = true;
}

void AudioVoice::stop()
{
    m_running = false;
}

void AudioVoice::unbindVoice()
{
    if (m_bound)
    {
        m_parent._unbindFrom(m_parentIt);
        m_bound = false;
    }
}

AudioVoiceMono::AudioVoiceMono(BaseAudioVoiceEngine& root, IAudioMix& parent, IAudioVoiceCallback* cb,
                               double sampleRate, bool dynamicRate)
: AudioVoice(root, parent, cb, dynamicRate)
{
    _resetSampleRate(sampleRate);
}

void AudioVoiceMono::_resetSampleRate(double sampleRate)
{
    soxr_delete(m_src);

    double rateOut = m_parent.mixInfo().m_sampleRate;
    soxr_datatype_t formatOut = m_parent.mixInfo().m_sampleFormat;
    soxr_io_spec_t ioSpec = soxr_io_spec(SOXR_INT16_I, formatOut);
    soxr_quality_spec_t qSpec = soxr_quality_spec(SOXR_20_BITQ, m_dynamicRate ? SOXR_VR : 0);

    soxr_error_t err;
    m_src = soxr_create(sampleRate, rateOut, 1,
                        &err, &ioSpec, &qSpec, nullptr);

    if (err)
    {
        Log.report(logvisor::Fatal, "unable to create soxr resampler: %s", soxr_strerror(err));
        m_resetSampleRate = false;
        return;
    }

    m_sampleRateIn = sampleRate;
    m_sampleRateOut = rateOut;
    soxr_set_input_fn(m_src, soxr_input_fn_t(SRCCallback), this, 0);
    _setPitchRatio(m_pitchRatio, false);
    m_resetSampleRate = false;

    m_silentOut = true;
    switch (formatOut)
    {
    case SOXR_INT16_I:
    {
        std::vector<int16_t>& scratch16 = m_root.m_scratch16;
        if (scratch16.size() < m_root.m_5msFrames)
            scratch16.resize(m_root.m_5msFrames);
        soxr_output(m_src, scratch16.data(), m_root.m_5msFrames);
        break;
    }
    case SOXR_INT32_I:
    {
        std::vector<int32_t>& scratch32 = m_root.m_scratch32;
        if (scratch32.size() < m_root.m_5msFrames)
            scratch32.resize(m_root.m_5msFrames);
        soxr_output(m_src, scratch32.data(), m_root.m_5msFrames);
        break;
    }
    case SOXR_FLOAT32_I:
    {
        std::vector<float>& scratchFlt = m_root.m_scratchFlt;
        if (scratchFlt.size() < m_root.m_5msFrames)
            scratchFlt.resize(m_root.m_5msFrames);
        soxr_output(m_src, scratchFlt.data(), m_root.m_5msFrames);
        break;
    }
    default: break;
    }
    m_silentOut = false;
}

size_t AudioVoiceMono::SRCCallback(AudioVoiceMono* ctx, int16_t** data, size_t frames)
{
    std::vector<int16_t>& scratchIn = ctx->m_root.m_scratchIn;
    if (scratchIn.size() < frames)
        scratchIn.resize(frames);
    *data = scratchIn.data();
    if (ctx->m_silentOut)
    {
        memset(*data, 0, frames * 2);
        return frames;
    }
    else
        return ctx->m_cb->supplyAudio(*ctx, frames, scratchIn.data());
}

size_t AudioVoiceMono::pumpAndMix(const AudioVoiceEngineMixInfo& mixInfo,
                                  size_t frames, int16_t* buf, int16_t* rbuf)
{
    std::vector<int16_t>& scratch16 = m_root.m_scratch16;
    if (scratch16.size() < frames)
        scratch16.resize(frames);

    size_t oDone = soxr_output(m_src, scratch16.data(), frames);
    _midUpdate();

    if (oDone)
    {
        m_matrix.mixMonoSampleData(mixInfo, scratch16.data(), buf, oDone);
        if (rbuf)
            m_subMatrix.mixMonoSampleData(mixInfo, scratch16.data(), rbuf, oDone);
    }

    return oDone;
}

size_t AudioVoiceMono::pumpAndMix(const AudioVoiceEngineMixInfo& mixInfo,
                                  size_t frames, int32_t* buf, int32_t* rbuf)
{
    std::vector<int32_t>& scratch32 = m_root.m_scratch32;
    if (scratch32.size() < frames)
        scratch32.resize(frames);

    size_t oDone = soxr_output(m_src, scratch32.data(), frames);
    _midUpdate();

    if (oDone)
    {
        m_matrix.mixMonoSampleData(mixInfo, scratch32.data(), buf, oDone);
        if (rbuf)
            m_subMatrix.mixMonoSampleData(mixInfo, scratch32.data(), rbuf, oDone);
    }

    return oDone;
}

size_t AudioVoiceMono::pumpAndMix(const AudioVoiceEngineMixInfo& mixInfo,
                                  size_t frames, float* buf, float* rbuf)
{
    std::vector<float>& scratchFlt = m_root.m_scratchFlt;
    if (scratchFlt.size() < frames)
        scratchFlt.resize(frames);

    size_t oDone = soxr_output(m_src, scratchFlt.data(), frames);
    _midUpdate();

    if (oDone)
    {
        m_matrix.mixMonoSampleData(mixInfo, scratchFlt.data(), buf, oDone);
        if (rbuf)
            m_subMatrix.mixMonoSampleData(mixInfo, scratchFlt.data(), rbuf, oDone);
    }

    return oDone;
}

void AudioVoiceMono::setDefaultMatrixCoefficients()
{
    m_matrix.setDefaultMatrixCoefficients(m_parent.mixInfo().m_channels);
    float zero[8] = {};
    m_subMatrix.setMatrixCoefficients(zero);
}

void AudioVoiceMono::setMonoMatrixCoefficients(const float coefs[8], bool slew)
{
    m_matrix.setMatrixCoefficients(coefs, slew ? m_root.m_5msFrames : 0);
}

void AudioVoiceMono::setStereoMatrixCoefficients(const float coefs[8][2], bool slew)
{
    float newCoefs[8] =
    {
        coefs[0][0],
        coefs[1][0],
        coefs[2][0],
        coefs[3][0],
        coefs[4][0],
        coefs[5][0],
        coefs[6][0],
        coefs[7][0]
    };
    m_matrix.setMatrixCoefficients(newCoefs, slew ? m_root.m_5msFrames : 0);
}

void AudioVoiceMono::setMonoSubmixMatrixCoefficients(const float coefs[8], bool slew)
{
    m_subMatrix.setMatrixCoefficients(coefs, slew ? m_root.m_5msFrames : 0);
}

void AudioVoiceMono::setStereoSubmixMatrixCoefficients(const float coefs[8][2], bool slew)
{
    float newCoefs[8] =
    {
        coefs[0][0],
        coefs[1][0],
        coefs[2][0],
        coefs[3][0],
        coefs[4][0],
        coefs[5][0],
        coefs[6][0],
        coefs[7][0]
    };
    m_subMatrix.setMatrixCoefficients(newCoefs, slew ? m_root.m_5msFrames : 0);
}

AudioVoiceStereo::AudioVoiceStereo(BaseAudioVoiceEngine& root, IAudioMix& parent, IAudioVoiceCallback* cb,
                                   double sampleRate, bool dynamicRate)
: AudioVoice(root, parent, cb, dynamicRate)
{
    _resetSampleRate(sampleRate);
}

void AudioVoiceStereo::_resetSampleRate(double sampleRate)
{
    soxr_delete(m_src);

    double rateOut = m_parent.mixInfo().m_sampleRate;
    soxr_datatype_t formatOut = m_parent.mixInfo().m_sampleFormat;
    soxr_io_spec_t ioSpec = soxr_io_spec(SOXR_INT16_I, formatOut);
    soxr_quality_spec_t qSpec = soxr_quality_spec(SOXR_20_BITQ, m_dynamicRate ? SOXR_VR : 0);

    soxr_error_t err;
    m_src = soxr_create(sampleRate, rateOut, 2,
                        &err, &ioSpec, &qSpec, nullptr);

    if (!m_src)
    {
        Log.report(logvisor::Fatal, "unable to create soxr resampler: %s", soxr_strerror(err));
        m_resetSampleRate = false;
        return;
    }

    m_sampleRateIn = sampleRate;
    m_sampleRateOut = rateOut;
    soxr_set_input_fn(m_src, soxr_input_fn_t(SRCCallback), this, 0);
    _setPitchRatio(m_pitchRatio, false);
    m_resetSampleRate = false;

    m_silentOut = true;
    switch (formatOut)
    {
    case SOXR_INT16_I:
    {
        std::vector<int16_t>& scratch16 = m_root.m_scratch16;
        if (scratch16.size() < m_root.m_5msFrames * 2)
            scratch16.resize(m_root.m_5msFrames * 2);
        soxr_output(m_src, scratch16.data(), m_root.m_5msFrames);
        break;
    }
    case SOXR_INT32_I:
    {
        std::vector<int32_t>& scratch32 = m_root.m_scratch32;
        if (scratch32.size() < m_root.m_5msFrames * 2)
            scratch32.resize(m_root.m_5msFrames * 2);
        soxr_output(m_src, scratch32.data(), m_root.m_5msFrames);
        break;
    }
    case SOXR_FLOAT32_I:
    {
        std::vector<float>& scratchFlt = m_root.m_scratchFlt;
        if (scratchFlt.size() < m_root.m_5msFrames * 2)
            scratchFlt.resize(m_root.m_5msFrames * 2);
        soxr_output(m_src, scratchFlt.data(), m_root.m_5msFrames);
        break;
    }
    default: break;
    }
    m_silentOut = false;
}

size_t AudioVoiceStereo::SRCCallback(AudioVoiceStereo* ctx, int16_t** data, size_t frames)
{
    std::vector<int16_t>& scratchIn = ctx->m_root.m_scratchIn;
    size_t samples = frames * 2;
    if (scratchIn.size() < samples)
        scratchIn.resize(samples);
    *data = scratchIn.data();
    if (ctx->m_silentOut)
    {
        memset(*data, 0, samples * 2);
        return frames;
    }
    else
        return ctx->m_cb->supplyAudio(*ctx, frames, scratchIn.data());
}

size_t AudioVoiceStereo::pumpAndMix(const AudioVoiceEngineMixInfo& mixInfo,
                                    size_t frames, int16_t* buf, int16_t* rbuf)
{
    std::vector<int16_t>& scratch16 = m_root.m_scratch16;
    size_t samples = frames * 2;
    if (scratch16.size() < samples)
        scratch16.resize(samples);

    size_t oDone = soxr_output(m_src, scratch16.data(), frames);
    _midUpdate();

    if (oDone)
    {
        m_matrix.mixStereoSampleData(mixInfo, scratch16.data(), buf, oDone);
        if (rbuf)
            m_subMatrix.mixStereoSampleData(mixInfo, scratch16.data(), rbuf, oDone);
    }

    return oDone;
}

size_t AudioVoiceStereo::pumpAndMix(const AudioVoiceEngineMixInfo& mixInfo,
                                    size_t frames, int32_t* buf, int32_t* rbuf)
{
    std::vector<int32_t>& scratch32 = m_root.m_scratch32;
    size_t samples = frames * 2;
    if (scratch32.size() < samples)
        scratch32.resize(samples);

    size_t oDone = soxr_output(m_src, scratch32.data(), frames);
    _midUpdate();

    if (oDone)
    {
        m_matrix.mixStereoSampleData(mixInfo, scratch32.data(), buf, oDone);
        if (rbuf)
            m_subMatrix.mixStereoSampleData(mixInfo, scratch32.data(), rbuf, oDone);
    }

    return oDone;
}

size_t AudioVoiceStereo::pumpAndMix(const AudioVoiceEngineMixInfo& mixInfo,
                                    size_t frames, float* buf, float* rbuf)
{
    std::vector<float>& scratchFlt = m_root.m_scratchFlt;
    size_t samples = frames * 2;
    if (scratchFlt.size() < samples)
        scratchFlt.resize(samples);

    size_t oDone = soxr_output(m_src, scratchFlt.data(), frames);
    _midUpdate();

    if (oDone)
    {
        m_matrix.mixStereoSampleData(mixInfo, scratchFlt.data(), buf, oDone);
        if (rbuf)
            m_subMatrix.mixStereoSampleData(mixInfo, scratchFlt.data(), rbuf, oDone);
    }

    return oDone;
}

void AudioVoiceStereo::setDefaultMatrixCoefficients()
{
    m_matrix.setDefaultMatrixCoefficients(m_parent.mixInfo().m_channels);
    float zero[8][2] = {{}};
    m_subMatrix.setMatrixCoefficients(zero);
}

void AudioVoiceStereo::setMonoMatrixCoefficients(const float coefs[8], bool slew)
{
    float newCoefs[8][2] =
    {
        {coefs[0], coefs[0]},
        {coefs[1], coefs[1]},
        {coefs[2], coefs[2]},
        {coefs[3], coefs[3]},
        {coefs[4], coefs[4]},
        {coefs[5], coefs[5]},
        {coefs[6], coefs[6]},
        {coefs[7], coefs[7]}
    };
    m_matrix.setMatrixCoefficients(newCoefs, slew ? m_root.m_5msFrames : 0);
}

void AudioVoiceStereo::setStereoMatrixCoefficients(const float coefs[8][2], bool slew)
{
    m_matrix.setMatrixCoefficients(coefs, slew ? m_root.m_5msFrames : 0);
}

void AudioVoiceStereo::setMonoSubmixMatrixCoefficients(const float coefs[8], bool slew)
{
    float newCoefs[8][2] =
    {
        {coefs[0], coefs[0]},
        {coefs[1], coefs[1]},
        {coefs[2], coefs[2]},
        {coefs[3], coefs[3]},
        {coefs[4], coefs[4]},
        {coefs[5], coefs[5]},
        {coefs[6], coefs[6]},
        {coefs[7], coefs[7]}
    };
    m_subMatrix.setMatrixCoefficients(newCoefs, slew ? m_root.m_5msFrames : 0);
}

void AudioVoiceStereo::setStereoSubmixMatrixCoefficients(const float coefs[8][2], bool slew)
{
    m_subMatrix.setMatrixCoefficients(coefs, slew ? m_root.m_5msFrames : 0);
}

}