// Copyright (C) 2024 The Qt Company Ltd.
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR LGPL-3.0-only OR GPL-2.0-only OR GPL-3.0-only

#include "qffmpegavaudioformat_p.h"
#include "qaudioformat.h"
#include "qffmpegmediaformatinfo_p.h"

#include <QtCore/qdebug.h>

extern "C" {
#include <libavutil/opt.h>
}

QT_BEGIN_NAMESPACE

namespace QFFmpeg {

AVAudioFormat::AVAudioFormat(const AVCodecContext *context)
    : sampleFormat(context->sample_fmt), sampleRate(context->sample_rate)
{
#if QT_FFMPEG_HAS_AV_CHANNEL_LAYOUT
    channelLayout = context->ch_layout;
#else
    if (context->channel_layout) {
        channelLayoutMask = context->channel_layout;
    } else {
        const auto channelConfig =
                QAudioFormat::defaultChannelConfigForChannelCount(context->channels);
        channelLayoutMask = QFFmpegMediaFormatInfo::avChannelLayout(channelConfig);
    }
#endif
}

AVAudioFormat::AVAudioFormat(const AVCodecParameters *codecPar)
    : sampleFormat(AVSampleFormat(codecPar->format)), sampleRate(codecPar->sample_rate)
{
#if QT_FFMPEG_HAS_AV_CHANNEL_LAYOUT
    channelLayout = codecPar->ch_layout;
#else
    if (codecPar->channel_layout) {
        channelLayoutMask = codecPar->channel_layout;
    } else {
        const auto channelConfig =
                QAudioFormat::defaultChannelConfigForChannelCount(codecPar->channels);
        channelLayoutMask = QFFmpegMediaFormatInfo::avChannelLayout(channelConfig);
    }
#endif
}

AVAudioFormat::AVAudioFormat(const QAudioFormat &audioFormat)
    : sampleFormat(QFFmpegMediaFormatInfo::avSampleFormat(audioFormat.sampleFormat())),
      sampleRate(audioFormat.sampleRate())
{
    const auto channelConfig = audioFormat.channelConfig() == QAudioFormat::ChannelConfigUnknown
            ? QAudioFormat::defaultChannelConfigForChannelCount(audioFormat.channelCount())
            : audioFormat.channelConfig();

    const auto mask = QFFmpegMediaFormatInfo::avChannelLayout(channelConfig);

#if QT_FFMPEG_HAS_AV_CHANNEL_LAYOUT
    av_channel_layout_from_mask(&channelLayout, mask);
#else
    channelLayoutMask = mask;
#endif
}

bool operator==(const AVAudioFormat &lhs, const AVAudioFormat &rhs)
{
    return lhs.sampleFormat == rhs.sampleFormat && lhs.sampleRate == rhs.sampleRate &&
#if QT_FFMPEG_HAS_AV_CHANNEL_LAYOUT
            lhs.channelLayout == rhs.channelLayout
#else
            lhs.channelLayoutMask == rhs.channelLayoutMask
#endif
            ;
}

QDebug operator<<(QDebug dbg, const AVAudioFormat &format)
{
    dbg << '[';
    const char *sampleFormatName = av_get_sample_fmt_name(format.sampleFormat);
    dbg << "sample format:" << (sampleFormatName ? sampleFormatName : "unknown");
    dbg << ", sample rate:" << format.sampleRate;

#if QT_FFMPEG_HAS_AV_CHANNEL_LAYOUT
    dbg << ", channel layout:" << format.channelLayout;
#else
    dbg << "channel layout:" << Qt::bin << format.channelLayoutMask << Qt::dec;
#endif
    dbg << ']';
    return dbg;
}

} // namespace QFFmpeg

QT_END_NAMESPACE