/*

 * Copyright (c) 2000, 2001, 2002 Fabrice Bellard

 *

 * This file is part of FFmpeg.

 *

 * FFmpeg is free software; you can redistribute it and/or

 * modify it under the terms of the GNU Lesser General Public

 * License as published by the Free Software Foundation; either

 * version 2.1 of the License, or (at your option) any later version.

 *

 * FFmpeg is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

 * Lesser General Public License for more details.

 *

 * You should have received a copy of the GNU Lesser General Public

 * License along with FFmpeg; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA

 */

#include "avformat.h"

#include "avformat_internal.h"

#include "avio_internal.h"

#include "demux.h"

#include "internal.h"

#include "libavcodec/avcodec.h"

#include "libavcodec/codec_par.h"

#include "libavutil/attributes.h"

#include "libavutil/avassert.h"

#include "libavutil/iamf.h"

#include "libavutil/internal.h"

#include "libavutil/intmath.h"

#include "libavutil/mem.h"

#include "libavutil/opt.h"

/**

 * @file

 * Options definition for AVFormatContext.

 */

FF_DISABLE_DEPRECATION_WARNINGS

#include "options_table.h"

FF_ENABLE_DEPRECATION_WARNINGS

static const char* format_to_name(void* ptr)

{

    AVFormatContext* fc = (AVFormatContext*) ptr;

    if (fc->name) return fc->name;

    else if(fc->iformat) return fc->iformat->name;

    else if(fc->oformat) return fc->oformat->name;

    else return fc->av_class->class_name;

}

static void *format_child_next(void *obj, void *prev)

{

    AVFormatContext *s = obj;

    if (!prev && s->priv_data &&

        ((s->iformat && s->iformat->priv_class) ||

          s->oformat && s->oformat->priv_class))

        return s->priv_data;

    if (s->pb && s->pb->av_class && prev != s->pb)

        return s->pb;

    return NULL;

}

enum {

    CHILD_CLASS_ITER_AVIO = 0,

    CHILD_CLASS_ITER_MUX,

    CHILD_CLASS_ITER_DEMUX,

    CHILD_CLASS_ITER_DONE,

};

#define ITER_STATE_SHIFT 16

static const AVClass *format_child_class_iterate(void **iter)

{

    // we use the low 16 bits of iter as the value to be passed to

    // av_(de)muxer_iterate()

    void *val = (void*)(((uintptr_t)*iter) & ((1 << ITER_STATE_SHIFT) - 1));

    unsigned int state = ((uintptr_t)*iter) >> ITER_STATE_SHIFT;

    const AVClass *ret = NULL;

    if (state == CHILD_CLASS_ITER_AVIO) {

        ret = &ff_avio_class;

        state++;

        goto finish;

    }

    if (state == CHILD_CLASS_ITER_MUX) {

        const AVOutputFormat *ofmt;

        while ((ofmt = av_muxer_iterate(&val))) {

            ret = ofmt->priv_class;

            if (ret)

                goto finish;

        }

        val = NULL;

        state++;

    }

    if (state == CHILD_CLASS_ITER_DEMUX) {

        const AVInputFormat *ifmt;

        while ((ifmt = av_demuxer_iterate(&val))) {

            ret = ifmt->priv_class;

            if (ret)

                goto finish;

        }

        val = NULL;

        state++;

    }

finish:

    // make sure none av_(de)muxer_iterate does not set the high bits of val

    av_assert0(!((uintptr_t)val >> ITER_STATE_SHIFT));

    *iter = (void*)((uintptr_t)val | (state << ITER_STATE_SHIFT));

    return ret;

}

static AVClassCategory get_category(void *ptr)

{

    AVFormatContext* s = ptr;

    if(s->iformat) return AV_CLASS_CATEGORY_DEMUXER;

    else           return AV_CLASS_CATEGORY_MUXER;

}

static const AVClass av_format_context_class = {

    .class_name     = "AVFormatContext",

    .item_name      = format_to_name,

    .option         = avformat_options,

    .version        = LIBAVUTIL_VERSION_INT,

    .child_next     = format_child_next,

    .child_class_iterate = format_child_class_iterate,

    .category       = AV_CLASS_CATEGORY_MUXER,

    .get_category   = get_category,

};

static int io_open_default(AVFormatContext *s, AVIOContext **pb,

                           const char *url, int flags, AVDictionary **options)

{

    int loglevel;

    if (!strcmp(url, s->url) ||

        s->iformat && !strcmp(s->iformat->name, "image2") ||

        s->oformat && !strcmp(s->oformat->name, "image2")

    ) {

        loglevel = AV_LOG_DEBUG;

    } else

        loglevel = AV_LOG_INFO;

    av_log(s, loglevel, "Opening \'%s\' for %s\n", url, flags & AVIO_FLAG_WRITE ? "writing" : "reading");

    return ffio_open_whitelist(pb, url, flags, &s->interrupt_callback, options, s->protocol_whitelist, s->protocol_blacklist);

}

static int io_close2_default(AVFormatContext *s, AVIOContext *pb)

{

    return avio_close(pb);

}

AVFormatContext *avformat_alloc_context(void)

{

    FormatContextInternal *fci;

    FFFormatContext *si;

    AVFormatContext *s;

    fci = av_mallocz(sizeof(*fci));

    if (!fci)

        return NULL;

    si = &fci->fc;

    s = &si->pub;

    s->av_class = &av_format_context_class;

    s->io_open  = io_open_default;

    s->io_close2= io_close2_default;

    av_opt_set_defaults(s);

    si->pkt = av_packet_alloc();

    si->parse_pkt = av_packet_alloc();

    if (!si->pkt || !si->parse_pkt) {

        avformat_free_context(s);

        return NULL;

    }

    return s;

}

const AVClass *avformat_get_class(void)

{

    return &av_format_context_class;

}

#define DISPOSITION_OPT(ctx)                                                                                                        \

    { "disposition", NULL, offsetof(ctx, disposition), AV_OPT_TYPE_FLAGS, { .i64 = 0 },                                             \

        .flags = AV_OPT_FLAG_ENCODING_PARAM, .unit = "disposition" },                                                               \

        { "default",            .type = AV_OPT_TYPE_CONST, { .i64 = AV_DISPOSITION_DEFAULT           },    .unit = "disposition" }, \

        { "dub",                .type = AV_OPT_TYPE_CONST, { .i64 = AV_DISPOSITION_DUB               },    .unit = "disposition" }, \

        { "original",           .type = AV_OPT_TYPE_CONST, { .i64 = AV_DISPOSITION_ORIGINAL          },    .unit = "disposition" }, \

        { "comment",            .type = AV_OPT_TYPE_CONST, { .i64 = AV_DISPOSITION_COMMENT           },    .unit = "disposition" }, \

        { "lyrics",             .type = AV_OPT_TYPE_CONST, { .i64 = AV_DISPOSITION_LYRICS            },    .unit = "disposition" }, \

        { "karaoke",            .type = AV_OPT_TYPE_CONST, { .i64 = AV_DISPOSITION_KARAOKE           },    .unit = "disposition" }, \

        { "forced",             .type = AV_OPT_TYPE_CONST, { .i64 = AV_DISPOSITION_FORCED            },    .unit = "disposition" }, \

        { "hearing_impaired",   .type = AV_OPT_TYPE_CONST, { .i64 = AV_DISPOSITION_HEARING_IMPAIRED  },    .unit = "disposition" }, \

        { "visual_impaired",    .type = AV_OPT_TYPE_CONST, { .i64 = AV_DISPOSITION_VISUAL_IMPAIRED   },    .unit = "disposition" }, \

        { "clean_effects",      .type = AV_OPT_TYPE_CONST, { .i64 = AV_DISPOSITION_CLEAN_EFFECTS     },    .unit = "disposition" }, \

        { "attached_pic",       .type = AV_OPT_TYPE_CONST, { .i64 = AV_DISPOSITION_ATTACHED_PIC      },    .unit = "disposition" }, \

        { "timed_thumbnails",   .type = AV_OPT_TYPE_CONST, { .i64 = AV_DISPOSITION_TIMED_THUMBNAILS  },    .unit = "disposition" }, \

        { "non_diegetic",       .type = AV_OPT_TYPE_CONST, { .i64 = AV_DISPOSITION_NON_DIEGETIC      },    .unit = "disposition" }, \

        { "captions",           .type = AV_OPT_TYPE_CONST, { .i64 = AV_DISPOSITION_CAPTIONS          },    .unit = "disposition" }, \

        { "descriptions",       .type = AV_OPT_TYPE_CONST, { .i64 = AV_DISPOSITION_DESCRIPTIONS      },    .unit = "disposition" }, \

        { "metadata",           .type = AV_OPT_TYPE_CONST, { .i64 = AV_DISPOSITION_METADATA          },    .unit = "disposition" }, \

        { "dependent",          .type = AV_OPT_TYPE_CONST, { .i64 = AV_DISPOSITION_DEPENDENT         },    .unit = "disposition" }, \

        { "still_image",        .type = AV_OPT_TYPE_CONST, { .i64 = AV_DISPOSITION_STILL_IMAGE       },    .unit = "disposition" }, \

        { "multilayer",         .type = AV_OPT_TYPE_CONST, { .i64 = AV_DISPOSITION_MULTILAYER        },    .unit = "disposition" }

static const AVOption stream_options[] = {

    DISPOSITION_OPT(AVStream),

    { "discard", NULL, offsetof(AVStream, discard), AV_OPT_TYPE_INT, { .i64 = AVDISCARD_DEFAULT }, INT_MIN, INT_MAX,

        .flags = AV_OPT_FLAG_DECODING_PARAM, .unit = "avdiscard" },

        { "none",               .type = AV_OPT_TYPE_CONST, {.i64 = AVDISCARD_NONE     }, .unit = "avdiscard" },

        { "default",            .type = AV_OPT_TYPE_CONST, {.i64 = AVDISCARD_DEFAULT  }, .unit = "avdiscard" },

        { "noref",              .type = AV_OPT_TYPE_CONST, {.i64 = AVDISCARD_NONREF   }, .unit = "avdiscard" },

        { "bidir",              .type = AV_OPT_TYPE_CONST, {.i64 = AVDISCARD_BIDIR    }, .unit = "avdiscard" },

        { "nointra",            .type = AV_OPT_TYPE_CONST, {.i64 = AVDISCARD_NONINTRA }, .unit = "avdiscard" },

        { "nokey",              .type = AV_OPT_TYPE_CONST, {.i64 = AVDISCARD_NONKEY   }, .unit = "avdiscard" },

        { "all",                .type = AV_OPT_TYPE_CONST, {.i64 = AVDISCARD_ALL      }, .unit = "avdiscard" },

    { NULL }

};

static const AVClass stream_class = {

    .class_name     = "AVStream",

    .item_name      = av_default_item_name,

    .version        = LIBAVUTIL_VERSION_INT,

    .option         = stream_options,

};

const AVClass *av_stream_get_class(void)

{

    return &stream_class;

}

AVStream *avformat_new_stream(AVFormatContext *s, const AVCodec *c)

{

    FFStream *sti;

    AVStream *st;

    AVStream **streams;

    if (s->nb_streams >= s->max_streams) {

        av_log(s, AV_LOG_ERROR, "Number of streams exceeds max_streams parameter"

               " (%d), see the documentation if you wish to increase it\n",

               s->max_streams);

        return NULL;

    }

    streams = av_realloc_array(s->streams, s->nb_streams + 1, sizeof(*streams));

    if (!streams)

        return NULL;

    s->streams = streams;

    sti = av_mallocz(sizeof(*sti));

    if (!sti)

        return NULL;

    st = &sti->pub;

    st->av_class = &stream_class;

    st->codecpar = avcodec_parameters_alloc();

    if (!st->codecpar)

        goto fail;

    sti->fmtctx = s;

    sti->parse_pkt = av_packet_alloc();

    if (!sti->parse_pkt)

        goto fail;

    if (s->iformat) {

        sti->avctx = avcodec_alloc_context3(NULL);

        if (!sti->avctx)

            goto fail;

        sti->info = av_mallocz(sizeof(*sti->info));

        if (!sti->info)

            goto fail;

#if FF_API_R_FRAME_RATE

        sti->info->last_dts      = AV_NOPTS_VALUE;

#endif

        sti->info->fps_first_dts = AV_NOPTS_VALUE;

        sti->info->fps_last_dts  = AV_NOPTS_VALUE;

        /* default pts setting is MPEG-like */

        avpriv_set_pts_info(st, 33, 1, 90000);

        /* we set the current DTS to 0 so that formats without any timestamps

         * but durations get some timestamps, formats with some unknown

         * timestamps have their first few packets buffered and the

         * timestamps corrected before they are returned to the user */

        sti->cur_dts = RELATIVE_TS_BASE;

    } else {

        sti->cur_dts = AV_NOPTS_VALUE;

    }

    st->index      = s->nb_streams;

    st->start_time = AV_NOPTS_VALUE;

    st->duration   = AV_NOPTS_VALUE;

    sti->first_dts     = AV_NOPTS_VALUE;

    sti->probe_packets = s->max_probe_packets;

    sti->pts_wrap_reference = AV_NOPTS_VALUE;

    sti->pts_wrap_behavior  = AV_PTS_WRAP_IGNORE;

    sti->last_IP_pts = AV_NOPTS_VALUE;

    sti->last_dts_for_order_check = AV_NOPTS_VALUE;

    for (int i = 0; i < MAX_REORDER_DELAY + 1; i++)

        sti->pts_buffer[i] = AV_NOPTS_VALUE;

    st->sample_aspect_ratio = (AVRational) { 0, 1 };

    sti->need_context_update = 1;

    s->streams[s->nb_streams++] = st;

    return st;

fail:

    ff_free_stream(&st);

    return NULL;

}

#define FLAGS AV_OPT_FLAG_ENCODING_PARAM | AV_OPT_FLAG_VIDEO_PARAM

#define OFFSET(x) offsetof(AVStreamGroupTileGrid, x)

static const AVOption tile_grid_options[] = {

    { "grid_size", "size of the output canvas", OFFSET(coded_width),

        AV_OPT_TYPE_IMAGE_SIZE, { .str = NULL }, 0, INT_MAX, FLAGS },

    { "output_size", "size of valid pixels in output image meant for presentation", OFFSET(width),

        AV_OPT_TYPE_IMAGE_SIZE, { .str = NULL }, 0, INT_MAX, FLAGS },

    { "background_color", "set a background color for unused pixels",

        OFFSET(background), AV_OPT_TYPE_COLOR, { .str = "black"}, 0, 0, FLAGS },

    { "horizontal_offset", NULL, OFFSET(horizontal_offset), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, INT_MAX, FLAGS },

    { "vertical_offset",   NULL, OFFSET(vertical_offset),   AV_OPT_TYPE_INT, { .i64 = 0 }, 0, INT_MAX, FLAGS },

    { NULL },

};

#undef OFFSET

static const AVClass tile_grid_class = {

    .class_name = "AVStreamGroupTileGrid",

    .version    = LIBAVUTIL_VERSION_INT,

    .option     = tile_grid_options,

};

#define OFFSET(x) offsetof(AVStreamGroupTREF, x)

static const AVOption tref_options[] = {

    { "metadata_index", "Index of the data stream within the group", OFFSET(metadata_index),

        AV_OPT_TYPE_UINT, { .i64 = 0 }, 0, UINT_MAX, FLAGS },

    { NULL },

};

#undef OFFSET

static const AVClass tref_class = {

    .class_name = "AVStreamGroupTREF",

    .version    = LIBAVUTIL_VERSION_INT,

    .option     = tref_options,

};

#if FF_API_LCEVC_STRUCT

FF_DISABLE_DEPRECATION_WARNINGS

#endif

#define OFFSET(x) offsetof(AVStreamGroupLayeredVideo, x)

static const AVOption layered_video_options[] = {

#if FF_API_LCEVC_STRUCT

    { "lcevc_index", "Index of the LCEVC stream within the group", OFFSET(lcevc_index),

        AV_OPT_TYPE_INT, { .i64 = 0 }, 0, INT_MAX, FLAGS | AV_OPT_FLAG_DEPRECATED },

#endif

    { "el_index", "Index of the enhancement layer stream within the group", OFFSET(el_index),

        AV_OPT_TYPE_INT, { .i64 = 0 }, 0, INT_MAX, FLAGS },

    { "video_size", "size of the final layered video presentation", OFFSET(width),

        AV_OPT_TYPE_IMAGE_SIZE, { .str = NULL }, 0, INT_MAX, FLAGS },

    { NULL },

};

#if FF_API_LCEVC_STRUCT

FF_ENABLE_DEPRECATION_WARNINGS

#endif

#undef OFFSET

static const AVClass layered_video_class = {

    .class_name = "AVStreamGroupLayeredVideo",

    .version    = LIBAVUTIL_VERSION_INT,

    .option     = layered_video_options,

};

static void *stream_group_child_next(void *obj, void *prev)

{

    AVStreamGroup *stg = obj;

    if (!prev) {

        switch(stg->type) {

        case AV_STREAM_GROUP_PARAMS_IAMF_AUDIO_ELEMENT:

            return stg->params.iamf_audio_element;

        case AV_STREAM_GROUP_PARAMS_IAMF_MIX_PRESENTATION:

            return stg->params.iamf_mix_presentation;

        case AV_STREAM_GROUP_PARAMS_TILE_GRID:

            return stg->params.tile_grid;

        case AV_STREAM_GROUP_PARAMS_TREF:

            return stg->params.tref;

        case AV_STREAM_GROUP_PARAMS_LCEVC:

        case AV_STREAM_GROUP_PARAMS_DOLBY_VISION:

            return stg->params.layered_video;

        default:

            break;

        }

    }

    return NULL;

}

#undef FLAGS

static const AVClass *stream_group_child_iterate(void **opaque)

{

    uintptr_t i = (uintptr_t)*opaque;

    const AVClass *ret = NULL;

    switch(i) {

    case AV_STREAM_GROUP_PARAMS_NONE:

        i++;

        av_fallthrough;

    case AV_STREAM_GROUP_PARAMS_IAMF_AUDIO_ELEMENT:

        ret = av_iamf_audio_element_get_class();

        break;

    case AV_STREAM_GROUP_PARAMS_IAMF_MIX_PRESENTATION:

        ret = av_iamf_mix_presentation_get_class();

        break;

    case AV_STREAM_GROUP_PARAMS_TILE_GRID:

        ret = &tile_grid_class;

        break;

    case AV_STREAM_GROUP_PARAMS_TREF:

        ret = &tref_class;

        break;

    case AV_STREAM_GROUP_PARAMS_LCEVC:

    case AV_STREAM_GROUP_PARAMS_DOLBY_VISION:

        ret = &layered_video_class;

        break;

    default:

        break;

    }

    if (ret)

        *opaque = (void*)(i + 1);

    return ret;

}

static const AVOption stream_group_options[] = {

    DISPOSITION_OPT(AVStreamGroup),

    {"id", "Set group id", offsetof(AVStreamGroup, id), AV_OPT_TYPE_INT64, {.i64 = 0}, 0, INT64_MAX, AV_OPT_FLAG_ENCODING_PARAM },

    { NULL }

};

static const AVClass stream_group_class = {

    .class_name     = "AVStreamGroup",

    .item_name      = av_default_item_name,

    .version        = LIBAVUTIL_VERSION_INT,

    .option         = stream_group_options,

    .child_next     = stream_group_child_next,

    .child_class_iterate = stream_group_child_iterate,

};

const AVClass *av_stream_group_get_class(void)

{

    return &stream_group_class;

}

AVStreamGroup *avformat_stream_group_create(AVFormatContext *s,

                                            enum AVStreamGroupParamsType type,

                                            AVDictionary **options)

{

    AVStreamGroup **stream_groups;

    AVStreamGroup *stg;

    FFStreamGroup *stgi;

    stream_groups = av_realloc_array(s->stream_groups, s->nb_stream_groups + 1,

                                     sizeof(*stream_groups));

    if (!stream_groups)

        return NULL;

    s->stream_groups = stream_groups;

    stgi = av_mallocz(sizeof(*stgi));

    if (!stgi)

        return NULL;

    stg = &stgi->pub;

    stg->av_class = &stream_group_class;

    av_opt_set_defaults(stg);

    stg->type = type;

    switch (type) {

    case AV_STREAM_GROUP_PARAMS_IAMF_AUDIO_ELEMENT:

        stg->params.iamf_audio_element = av_iamf_audio_element_alloc();

        if (!stg->params.iamf_audio_element)

            goto fail;

        break;

    case AV_STREAM_GROUP_PARAMS_IAMF_MIX_PRESENTATION:

        stg->params.iamf_mix_presentation = av_iamf_mix_presentation_alloc();

        if (!stg->params.iamf_mix_presentation)

            goto fail;

        break;

    case AV_STREAM_GROUP_PARAMS_TILE_GRID:

        stg->params.tile_grid = av_mallocz(sizeof(*stg->params.tile_grid));

        if (!stg->params.tile_grid)

            goto fail;

        stg->params.tile_grid->av_class = &tile_grid_class;

        av_opt_set_defaults(stg->params.tile_grid);

        break;

    case AV_STREAM_GROUP_PARAMS_TREF:

        stg->params.tref = av_mallocz(sizeof(*stg->params.tref));

        if (!stg->params.tref)

            goto fail;

        stg->params.tref->av_class = &tref_class;

        av_opt_set_defaults(stg->params.tref);

        break;

    case AV_STREAM_GROUP_PARAMS_LCEVC:

    case AV_STREAM_GROUP_PARAMS_DOLBY_VISION:

        stg->params.layered_video = av_mallocz(sizeof(*stg->params.layered_video));

        if (!stg->params.layered_video)

            goto fail;

        stg->params.layered_video->av_class = &layered_video_class;

        av_opt_set_defaults(stg->params.layered_video);

        break;

    default:

        goto fail;

    }

    if (options) {

        if (av_opt_set_dict2(stg, options, AV_OPT_SEARCH_CHILDREN))

            goto fail;

    }

    stgi->fmtctx = s;

    stg->index   = s->nb_stream_groups;

    s->stream_groups[s->nb_stream_groups++] = stg;

    return stg;

fail:

    ff_free_stream_group(&stg);

    return NULL;

}

static int stream_group_add_stream(AVStreamGroup *stg, AVStream *st)

{

    AVStream **streams = av_realloc_array(stg->streams, stg->nb_streams + 1,

                                          sizeof(*stg->streams));

    if (!streams)

        return AVERROR(ENOMEM);

    stg->streams = streams;

    stg->streams[stg->nb_streams++] = st;

    return 0;

}

int avformat_stream_group_add_stream(AVStreamGroup *stg, AVStream *st)

{

    const FFStreamGroup *stgi = cffstreamgroup(stg);

    const FFStream *sti = cffstream(st);

    if (stgi->fmtctx != sti->fmtctx)

        return AVERROR(EINVAL);

    for (int i = 0; i < stg->nb_streams; i++)

        if (stg->streams[i]->index == st->index)

            return AVERROR(EEXIST);

    return stream_group_add_stream(stg, st);

}

static int option_is_disposition(const AVOption *opt)

{

    return opt->type == AV_OPT_TYPE_CONST &&

           opt->unit && !strcmp(opt->unit, "disposition");

}

int av_disposition_from_string(const char *disp)

{

    for (const AVOption *opt = stream_options; opt->name; opt++)

        if (option_is_disposition(opt) && !strcmp(disp, opt->name))

            return opt->default_val.i64;

    return AVERROR(EINVAL);

}

const char *av_disposition_to_string(int disposition)

{

    int val;

    if (disposition <= 0)

        return NULL;

    val = 1 << ff_ctz(disposition);

    for (const AVOption *opt = stream_options; opt->name; opt++)

        if (option_is_disposition(opt) && opt->default_val.i64 == val)

            return opt->name;

    return NULL;

}