mirror of
https://github.com/FFmpeg/FFmpeg.git
synced 2024-12-02 03:06:28 +02:00
3c5bceb751
Signed-off-by: James Almer <jamrial@gmail.com>
356 lines
11 KiB
C
356 lines
11 KiB
C
/*
|
|
* 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 "avio_internal.h"
|
|
#include "demux.h"
|
|
#include "internal.h"
|
|
|
|
#include "libavcodec/avcodec.h"
|
|
#include "libavcodec/codec_par.h"
|
|
|
|
#include "libavutil/avassert.h"
|
|
#include "libavutil/internal.h"
|
|
#include "libavutil/intmath.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->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);
|
|
}
|
|
|
|
#if FF_API_AVFORMAT_IO_CLOSE
|
|
void ff_format_io_close_default(AVFormatContext *s, AVIOContext *pb)
|
|
{
|
|
avio_close(pb);
|
|
}
|
|
#endif
|
|
|
|
static int io_close2_default(AVFormatContext *s, AVIOContext *pb)
|
|
{
|
|
return avio_close(pb);
|
|
}
|
|
|
|
AVFormatContext *avformat_alloc_context(void)
|
|
{
|
|
FFFormatContext *const si = av_mallocz(sizeof(*si));
|
|
AVFormatContext *s;
|
|
|
|
if (!si)
|
|
return NULL;
|
|
|
|
s = &si->pub;
|
|
s->av_class = &av_format_context_class;
|
|
s->io_open = io_open_default;
|
|
#if FF_API_AVFORMAT_IO_CLOSE
|
|
FF_DISABLE_DEPRECATION_WARNINGS
|
|
s->io_close = ff_format_io_close_default;
|
|
FF_ENABLE_DEPRECATION_WARNINGS
|
|
#endif
|
|
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;
|
|
}
|
|
|
|
#if FF_API_LAVF_SHORTEST
|
|
si->shortest_end = AV_NOPTS_VALUE;
|
|
#endif
|
|
|
|
return s;
|
|
}
|
|
|
|
enum AVDurationEstimationMethod av_fmt_ctx_get_duration_estimation_method(const AVFormatContext* ctx)
|
|
{
|
|
return ctx->duration_estimation_method;
|
|
}
|
|
|
|
const AVClass *avformat_get_class(void)
|
|
{
|
|
return &av_format_context_class;
|
|
}
|
|
|
|
static const AVOption stream_options[] = {
|
|
{ "disposition", NULL, offsetof(AVStream, 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" },
|
|
{ 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)
|
|
{
|
|
FFFormatContext *const si = ffformatcontext(s);
|
|
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->avctx = avcodec_alloc_context3(NULL);
|
|
if (!sti->avctx)
|
|
goto fail;
|
|
|
|
if (s->iformat) {
|
|
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 };
|
|
|
|
#if FF_API_AVSTREAM_SIDE_DATA
|
|
sti->inject_global_side_data = si->inject_global_side_data;
|
|
#endif
|
|
|
|
sti->need_context_update = 1;
|
|
|
|
s->streams[s->nb_streams++] = st;
|
|
return st;
|
|
fail:
|
|
ff_free_stream(&st);
|
|
return NULL;
|
|
}
|
|
|
|
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;
|
|
}
|