/*
 * Copyright 2010 Stephan Aßmus <superstippi@gmx.de>. All Rights Reserved.
 * Distributed under the terms of the MIT License.
 */


#include "CodecTable.h"

extern "C" {
        #include "avcodec.h"
        #include "avformat.h"
}


struct AVInputFamily gAVInputFamilies[] = {
        { B_AIFF_FORMAT_FAMILY, "aiff" },
        { B_AVI_FORMAT_FAMILY, "avi" },
        { B_MPEG_FORMAT_FAMILY, "mpeg" },
        { B_QUICKTIME_FORMAT_FAMILY, "mov" },
        { B_ANY_FORMAT_FAMILY, NULL}
};

static const int32 sMaxFormatCount = 1024;
media_format gAVCodecFormats[sMaxFormatCount];


status_t
build_decoder_formats(media_format** _formats, size_t* _count)
{
        BMediaFormats mediaFormats;
        if (mediaFormats.InitCheck() != B_OK)
                return B_ERROR;

        int32 index = 0;
        void* cookie = NULL;
        const AVCodec* codec = NULL;
        while ((codec = av_codec_iterate(&cookie)) != NULL) {
                if (index >= sMaxFormatCount) {
                        fprintf(stderr, "Maximum format count reached for auto-generated "
                                "AVCodec to media_format mapping, but there are still more "
                                "AVCodecs compiled into libavcodec!\n");
                        break;
                }
                media_format format;
                // Determine media type
                switch (codec->type) {
                        case AVMEDIA_TYPE_VIDEO:
                                format.type = B_MEDIA_ENCODED_VIDEO;
                                break;
                        case AVMEDIA_TYPE_AUDIO:
                                format.type = B_MEDIA_ENCODED_AUDIO;
                                break;
                        default:
                                // ignore this AVCodec
                                continue;
                }

                media_format_description description;
                memset(description._reserved_, 0, sizeof(description._reserved_));
                memset(description.u._reserved_, 0, sizeof(description.u._reserved_));

                // Hard-code everything to B_MISC_FORMAT_FAMILY to ease matching
                // later on.
                description.family = B_MISC_FORMAT_FAMILY;
                description.u.misc.file_format = 'ffmp';
                description.u.misc.codec = codec->id;

                format.require_flags = 0;
                format.deny_flags = B_MEDIA_MAUI_UNDEFINED_FLAGS;

                if (mediaFormats.MakeFormatFor(&description, 1, &format) != B_OK)
                        return B_ERROR;

                gAVCodecFormats[index] = format;

                index++;
        }

        *_formats = gAVCodecFormats;
        *_count = index;

        return B_OK;
}