#include <PerfStat.h>
#include <QtCore/QLoggingCategory>
#include <AudioConstants.h>



#include "OpusDecoder.h"

static QLoggingCategory decoder("AthenaOpusDecoder");

static QString error_to_string(int error) {
    switch (error) {
    case OPUS_OK:
        return "OK";
    case OPUS_BAD_ARG:
        return "One or more invalid/out of range arguments.";
    case OPUS_BUFFER_TOO_SMALL:
        return "The mode struct passed is invalid.";
    case OPUS_INTERNAL_ERROR:
        return "An internal error was detected.";
    case OPUS_INVALID_PACKET:
        return "The compressed data passed is corrupted.";
    case OPUS_UNIMPLEMENTED:
        return "Invalid/unsupported request number.";
    case OPUS_INVALID_STATE:
        return "An encoder or decoder structure is invalid or already freed.";
    default:
        return QString("Unknown error code: %i").arg(error);
    }
}


AthenaOpusDecoder::AthenaOpusDecoder(int sampleRate, int numChannels)
{
    int error;

    _opus_sample_rate = sampleRate;
    _opus_num_channels = numChannels;

    _decoder = opus_decoder_create(sampleRate, numChannels, &error);

    if ( error != OPUS_OK ) {
        qCCritical(decoder) << "Failed to initialize Opus encoder: " << error_to_string(error);
        _decoder = nullptr;
        return;
    }


    qCDebug(decoder) << "Opus decoder initialized, sampleRate = " << sampleRate << "; numChannels = " << numChannels;
}

AthenaOpusDecoder::~AthenaOpusDecoder()
{
    if ( _decoder )
        opus_decoder_destroy(_decoder);

}

void AthenaOpusDecoder::decode(const QByteArray &encodedBuffer, QByteArray &decodedBuffer)
{
    assert(_decoder);
    PerformanceTimer perfTimer("AthenaOpusDecoder::decode");

    // The audio system encodes and decodes always in fixed size chunks
    int buffer_size = AudioConstants::NETWORK_FRAME_SAMPLES_PER_CHANNEL * static_cast<int>(sizeof(int16_t)) * _opus_num_channels;

    decodedBuffer.resize( buffer_size );
    int buffer_frames = decodedBuffer.size() / _opus_num_channels / static_cast<int>(sizeof( opus_int16 ));

    qCDebug(decoder) << "Opus decode: encodedBytes = " << encodedBuffer.length() << "; decodedBufferBytes = " << decodedBuffer.size() << "; frameCount = " << buffer_frames;
    int decoded_frames = opus_decode( _decoder, reinterpret_cast<const unsigned char*>(encodedBuffer.data()), encodedBuffer.length(), reinterpret_cast<opus_int16*>(decodedBuffer.data()), buffer_frames, 0 );

    if ( decoded_frames >= 0 ) {
        qCDebug(decoder) << "Decoded " << decoded_frames << " Opus frames, " << buffer_frames << " expected";

        if ( decoded_frames < buffer_frames ) {
            qCWarning(decoder) << "Opus decoder returned " << decoded_frames << ", but " << buffer_frames << " were expected!";

            int start = decoded_frames * static_cast<int>(sizeof(int16_t)) * _opus_num_channels;
            memset( &decodedBuffer.data()[start], 0, static_cast<size_t>(decodedBuffer.length() - start));
        } else if ( decoded_frames > buffer_frames ) {
            // This should never happen
            qCCritical(decoder) << "Opus decoder returned " << decoded_frames << ", but only " << buffer_frames << " were expected! Buffer overflow!?";
        }
    } else {
        qCCritical(decoder) << "Failed to decode audio: " << error_to_string(decoded_frames);
        decodedBuffer.fill('\0');
    }

}

void AthenaOpusDecoder::lostFrame(QByteArray &decodedBuffer)
{
    assert(_decoder);

    PerformanceTimer perfTimer("AthenaOpusDecoder::lostFrame");

    int buffer_size = AudioConstants::NETWORK_FRAME_SAMPLES_PER_CHANNEL * static_cast<int>(sizeof(int16_t)) * _opus_num_channels;
    decodedBuffer.resize( buffer_size );
    int buffer_frames = decodedBuffer.size() / _opus_num_channels / static_cast<int>(sizeof( opus_int16 ));

    int decoded_frames = opus_decode( _decoder, nullptr, 0, reinterpret_cast<opus_int16*>(decodedBuffer.data()), buffer_frames, 1 );

    if ( decoded_frames >= 0 ) {

        qCDebug(decoder) << "Produced " << decoded_frames << " opus frames from a lost frame, " << buffer_frames << " expected";

        if ( decoded_frames < buffer_frames ) {
            qCWarning(decoder) << "Opus decoder returned " << decoded_frames << ", but " << buffer_frames << " were expected!";

            int start = decoded_frames * static_cast<int>(sizeof(int16_t)) * _opus_num_channels;
            memset( &decodedBuffer.data()[start], 0, static_cast<size_t>(decodedBuffer.length() - start));
        } else if ( decoded_frames > buffer_frames ) {
            // This should never happen
            qCCritical(decoder) << "Opus decoder returned " << decoded_frames << ", but only " << buffer_frames << " were expected! Buffer overflow!?";
        }
    } else {
        qCCritical(decoder) << "Failed to decode lost frame: " << error_to_string(decoded_frames);
        decodedBuffer.fill('\0');
    }
}