virtualenv/FFmpeg
1
0
Fork 0
mirror of https://github.com/FFmpeg/FFmpeg.git synced 2024-12-23 12:43:46 +02:00
Code Issues Releases Activity

ffmpeg: simplify managing input files and streams

Browse Source 
Grow the file and stream list in opt_input_file() instead of creating it
all at once in transcode().  This is simpler and will be useful for
following commits.
This commit is contained in:
Anton Khirnov 2011-05-22 22:12:35 +02:00
parent d2bc4da15b
commit 07633154ad
1 changed files with 68 additions and 95 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

163
ffmpeg.c
View File

@ -105,11 +105,9 @@ static const OptionDef options[];
#define FFM_PACKET_SIZE 4096 //XXX a duplicate of the line in ffm.h #define FFM_PACKET_SIZE 4096 //XXX a duplicate of the line in ffm.h
static const char *last_asked_format = NULL; static const char *last_asked_format = NULL;
static AVFormatContext *input_files[MAX_FILES];
static int64_t input_files_ts_offset[MAX_FILES]; static int64_t input_files_ts_offset[MAX_FILES];
static double *input_files_ts_scale[MAX_FILES] = {NULL}; static double *input_files_ts_scale[MAX_FILES] = {NULL};
static AVCodec **input_codecs = NULL; static AVCodec **input_codecs = NULL;
static int nb_input_files = 0;
static int nb_input_codecs = 0; static int nb_input_codecs = 0;
static int nb_input_files_ts_scale[MAX_FILES] = {0}; static int nb_input_files_ts_scale[MAX_FILES] = {0};
@ -327,12 +325,18 @@ typedef struct AVInputStream {
} AVInputStream; } AVInputStream;
typedef struct AVInputFile { typedef struct AVInputFile {
AVFormatContext *ctx;
int eof_reached; /* true if eof reached */ int eof_reached; /* true if eof reached */
int ist_index; /* index of first stream in ist_table */ int ist_index; /* index of first stream in ist_table */
int buffer_size; /* current total buffer size */ int buffer_size; /* current total buffer size */
int nb_streams; /* nb streams we are aware of */ int nb_streams; /* nb streams we are aware of */
} AVInputFile; } AVInputFile;
static AVInputStream *input_streams = NULL;
static int nb_input_streams = 0;
static AVInputFile *input_files = NULL;
static int nb_input_files = 0;
#if CONFIG_AVFILTER #if CONFIG_AVFILTER
static int configure_video_filters(AVInputStream *ist, AVOutputStream *ost) static int configure_video_filters(AVInputStream *ist, AVOutputStream *ost)
@ -462,7 +466,7 @@ static int ffmpeg_exit(int ret)
av_free(output_streams_for_file[i]); av_free(output_streams_for_file[i]);
} }
for(i=0;i<nb_input_files;i++) { for(i=0;i<nb_input_files;i++) {
av_close_input_file(input_files[i]); av_close_input_file(input_files[i].ctx);
av_free(input_files_ts_scale[i]); av_free(input_files_ts_scale[i]);
} }
@ -479,6 +483,9 @@ static int ffmpeg_exit(int ret)
av_free(stream_maps); av_free(stream_maps);
av_free(meta_data_maps); av_free(meta_data_maps);
av_freep(&input_streams);
av_freep(&input_files);
av_free(video_codec_name); av_free(video_codec_name);
av_free(audio_codec_name); av_free(audio_codec_name);
av_free(subtitle_codec_name); av_free(subtitle_codec_name);
@ -1861,7 +1868,7 @@ static void print_sdp(AVFormatContext **avc, int n)
static int copy_chapters(int infile, int outfile) static int copy_chapters(int infile, int outfile)
{ {
AVFormatContext *is = input_files[infile]; AVFormatContext *is = input_files[infile].ctx;
AVFormatContext *os = output_files[outfile]; AVFormatContext *os = output_files[outfile];
int i; int i;
@ -1927,60 +1934,23 @@ static void parse_forced_key_frames(char *kf, AVOutputStream *ost,
*/ */
static int transcode(AVFormatContext **output_files, static int transcode(AVFormatContext **output_files,
int nb_output_files, int nb_output_files,
AVFormatContext **input_files, AVInputFile *input_files,
int nb_input_files, int nb_input_files,
AVStreamMap *stream_maps, int nb_stream_maps) AVStreamMap *stream_maps, int nb_stream_maps)
{ {
int ret = 0, i, j, k, n, nb_istreams = 0, nb_ostreams = 0; int ret = 0, i, j, k, n, nb_ostreams = 0;
AVFormatContext *is, *os; AVFormatContext *is, *os;
AVCodecContext *codec, *icodec; AVCodecContext *codec, *icodec;
AVOutputStream *ost, **ost_table = NULL; AVOutputStream *ost, **ost_table = NULL;
AVInputStream *ist, **ist_table = NULL; AVInputStream *ist;
AVInputFile *file_table;
char error[1024]; char error[1024];
int want_sdp = 1; int want_sdp = 1;
uint8_t no_packet[MAX_FILES]={0}; uint8_t no_packet[MAX_FILES]={0};
int no_packet_count=0; int no_packet_count=0;
file_table= av_mallocz(nb_input_files * sizeof(AVInputFile)); if (rate_emu)
if (!file_table) for (i = 0; i < nb_input_streams; i++)
goto fail; input_streams[i].start = av_gettime();
/* input stream init */
j = 0;
for(i=0;i<nb_input_files;i++) {
is = input_files[i];
file_table[i].ist_index = j;
file_table[i].nb_streams = is->nb_streams;
j += is->nb_streams;
}
nb_istreams = j;
ist_table = av_mallocz(nb_istreams * sizeof(AVInputStream *));
if (!ist_table)
goto fail;
for(i=0;i<nb_istreams;i++) {
ist = av_mallocz(sizeof(AVInputStream));
if (!ist)
goto fail;
ist_table[i] = ist;
}
j = 0;
for(i=0;i<nb_input_files;i++) {
is = input_files[i];
for(k=0;k<is->nb_streams;k++) {
ist = ist_table[j++];
ist->st = is->streams[k];
ist->file_index = i;
ist->discard = 1; /* the stream is discarded by default
(changed later) */
if (rate_emu) {
ist->start = av_gettime();
}
}
}
/* output stream init */ /* output stream init */
nb_ostreams = 0; nb_ostreams = 0;
@ -2006,7 +1976,7 @@ static int transcode(AVFormatContext **output_files,
int si = stream_maps[i].stream_index; int si = stream_maps[i].stream_index;
if (fi < 0 || fi > nb_input_files - 1 || if (fi < 0 || fi > nb_input_files - 1 ||
si < 0 || si > file_table[fi].nb_streams - 1) { si < 0 || si > input_files[fi].nb_streams - 1) {
fprintf(stderr,"Could not find input stream #%d.%d\n", fi, si); fprintf(stderr,"Could not find input stream #%d.%d\n", fi, si);
ret = AVERROR(EINVAL); ret = AVERROR(EINVAL);
goto fail; goto fail;
@ -2014,7 +1984,7 @@ static int transcode(AVFormatContext **output_files,
fi = stream_maps[i].sync_file_index; fi = stream_maps[i].sync_file_index;
si = stream_maps[i].sync_stream_index; si = stream_maps[i].sync_stream_index;
if (fi < 0 || fi > nb_input_files - 1 || if (fi < 0 || fi > nb_input_files - 1 ||
si < 0 || si > file_table[fi].nb_streams - 1) { si < 0 || si > input_files[fi].nb_streams - 1) {
fprintf(stderr,"Could not find sync stream #%d.%d\n", fi, si); fprintf(stderr,"Could not find sync stream #%d.%d\n", fi, si);
ret = AVERROR(EINVAL); ret = AVERROR(EINVAL);
goto fail; goto fail;
@ -2032,11 +2002,11 @@ static int transcode(AVFormatContext **output_files,
ost = ost_table[n] = output_streams_for_file[k][i]; ost = ost_table[n] = output_streams_for_file[k][i];
ost->st = os->streams[i]; ost->st = os->streams[i];
if (nb_stream_maps > 0) { if (nb_stream_maps > 0) {
ost->source_index = file_table[stream_maps[n].file_index].ist_index + ost->source_index = input_files[stream_maps[n].file_index].ist_index +
stream_maps[n].stream_index; stream_maps[n].stream_index;
/* Sanity check that the stream types match */ /* Sanity check that the stream types match */
if (ist_table[ost->source_index]->st->codec->codec_type != ost->st->codec->codec_type) { if (input_streams[ost->source_index].st->codec->codec_type != ost->st->codec->codec_type) {
int i= ost->file_index; int i= ost->file_index;
av_dump_format(output_files[i], i, output_files[i]->filename, 1); av_dump_format(output_files[i], i, output_files[i]->filename, 1);
fprintf(stderr, "Codec type mismatch for mapping #%d.%d -> #%d.%d\n", fprintf(stderr, "Codec type mismatch for mapping #%d.%d -> #%d.%d\n",
@ -2049,12 +2019,12 @@ static int transcode(AVFormatContext **output_files,
int best_nb_frames=-1; int best_nb_frames=-1;
/* get corresponding input stream index : we select the first one with the right type */ /* get corresponding input stream index : we select the first one with the right type */
found = 0; found = 0;
for(j=0;j<nb_istreams;j++) { for (j = 0; j < nb_input_streams; j++) {
int skip=0; int skip=0;
ist = ist_table[j]; ist = &input_streams[j];
if(opt_programid){ if(opt_programid){
int pi,si; int pi,si;
AVFormatContext *f= input_files[ ist->file_index ]; AVFormatContext *f = input_files[ist->file_index].ctx;
skip=1; skip=1;
for(pi=0; pi<f->nb_programs; pi++){ for(pi=0; pi<f->nb_programs; pi++){
AVProgram *p= f->programs[pi]; AVProgram *p= f->programs[pi];
@ -2078,8 +2048,8 @@ static int transcode(AVFormatContext **output_files,
if (!found) { if (!found) {
if(! opt_programid) { if(! opt_programid) {
/* try again and reuse existing stream */ /* try again and reuse existing stream */
for(j=0;j<nb_istreams;j++) { for (j = 0; j < nb_input_streams; j++) {
ist = ist_table[j]; ist = &input_streams[j];
if ( ist->st->codec->codec_type == ost->st->codec->codec_type if ( ist->st->codec->codec_type == ost->st->codec->codec_type
&& ist->st->discard != AVDISCARD_ALL) { && ist->st->discard != AVDISCARD_ALL) {
ost->source_index = j; ost->source_index = j;
@ -2096,10 +2066,10 @@ static int transcode(AVFormatContext **output_files,
} }
} }
} }
ist = ist_table[ost->source_index]; ist = &input_streams[ost->source_index];
ist->discard = 0; ist->discard = 0;
ost->sync_ist = (nb_stream_maps > 0) ? ost->sync_ist = (nb_stream_maps > 0) ?
ist_table[file_table[stream_maps[n].sync_file_index].ist_index + &input_streams[input_files[stream_maps[n].sync_file_index].ist_index +
stream_maps[n].sync_stream_index] : ist; stream_maps[n].sync_stream_index] : ist;
} }
} }
@ -2108,7 +2078,7 @@ static int transcode(AVFormatContext **output_files,
for(i=0;i<nb_ostreams;i++) { for(i=0;i<nb_ostreams;i++) {
ost = ost_table[i]; ost = ost_table[i];
os = output_files[ost->file_index]; os = output_files[ost->file_index];
ist = ist_table[ost->source_index]; ist = &input_streams[ost->source_index];
codec = ost->st->codec; codec = ost->st->codec;
icodec = ist->st->codec; icodec = ist->st->codec;
@ -2306,7 +2276,7 @@ static int transcode(AVFormatContext **output_files,
ost = ost_table[i]; ost = ost_table[i];
if (ost->encoding_needed) { if (ost->encoding_needed) {
AVCodec *codec = i < nb_output_codecs ? output_codecs[i] : NULL; AVCodec *codec = i < nb_output_codecs ? output_codecs[i] : NULL;
AVCodecContext *dec = ist_table[ost->source_index]->st->codec; AVCodecContext *dec = input_streams[ost->source_index].st->codec;
if (!codec) if (!codec)
codec = avcodec_find_encoder(ost->st->codec->codec_id); codec = avcodec_find_encoder(ost->st->codec->codec_id);
if (!codec) { if (!codec) {
@ -2335,8 +2305,8 @@ static int transcode(AVFormatContext **output_files,
} }
/* open each decoder */ /* open each decoder */
for(i=0;i<nb_istreams;i++) { for (i = 0; i < nb_input_streams; i++) {
ist = ist_table[i]; ist = &input_streams[i];
if (ist->decoding_needed) { if (ist->decoding_needed) {
AVCodec *codec = i < nb_input_codecs ? input_codecs[i] : NULL; AVCodec *codec = i < nb_input_codecs ? input_codecs[i] : NULL;
if (!codec) if (!codec)
@ -2370,9 +2340,9 @@ static int transcode(AVFormatContext **output_files,
} }
/* init pts */ /* init pts */
for(i=0;i<nb_istreams;i++) { for (i = 0; i < nb_input_streams; i++) {
AVStream *st; AVStream *st;
ist = ist_table[i]; ist = &input_streams[i];
st= ist->st; st= ist->st;
ist->pts = st->avg_frame_rate.num ? - st->codec->has_b_frames*AV_TIME_BASE / av_q2d(st->avg_frame_rate) : 0; ist->pts = st->avg_frame_rate.num ? - st->codec->has_b_frames*AV_TIME_BASE / av_q2d(st->avg_frame_rate) : 0;
ist->next_pts = AV_NOPTS_VALUE; ist->next_pts = AV_NOPTS_VALUE;
@ -2402,7 +2372,7 @@ static int transcode(AVFormatContext **output_files,
METADATA_CHECK_INDEX(in_file_index, nb_input_files, "input file") METADATA_CHECK_INDEX(in_file_index, nb_input_files, "input file")
files[0] = output_files[out_file_index]; files[0] = output_files[out_file_index];
files[1] = input_files[in_file_index]; files[1] = input_files[in_file_index].ctx;
for (j = 0; j < 2; j++) { for (j = 0; j < 2; j++) {
AVMetaDataMap *map = &meta_data_maps[i][j]; AVMetaDataMap *map = &meta_data_maps[i][j];
@ -2433,7 +2403,7 @@ static int transcode(AVFormatContext **output_files,
if (metadata_global_autocopy) { if (metadata_global_autocopy) {
for (i = 0; i < nb_output_files; i++) for (i = 0; i < nb_output_files; i++)
av_metadata_copy(&output_files[i]->metadata, input_files[0]->metadata, av_metadata_copy(&output_files[i]->metadata, input_files[0].ctx->metadata,
AV_METADATA_DONT_OVERWRITE); AV_METADATA_DONT_OVERWRITE);
} }
@ -2460,7 +2430,7 @@ static int transcode(AVFormatContext **output_files,
/* copy chapters from the first input file that has them*/ /* copy chapters from the first input file that has them*/
if (!nb_chapter_maps) if (!nb_chapter_maps)
for (i = 0; i < nb_input_files; i++) { for (i = 0; i < nb_input_files; i++) {
if (!input_files[i]->nb_chapters) if (!input_files[i].ctx->nb_chapters)
continue; continue;
for (j = 0; j < nb_output_files; j++) for (j = 0; j < nb_output_files; j++)
@ -2495,11 +2465,11 @@ static int transcode(AVFormatContext **output_files,
for(i=0;i<nb_ostreams;i++) { for(i=0;i<nb_ostreams;i++) {
ost = ost_table[i]; ost = ost_table[i];
fprintf(stderr, " Stream #%d.%d -> #%d.%d", fprintf(stderr, " Stream #%d.%d -> #%d.%d",
ist_table[ost->source_index]->file_index, input_streams[ost->source_index].file_index,
ist_table[ost->source_index]->st->index, input_streams[ost->source_index].st->index,
ost->file_index, ost->file_index,
ost->index); ost->index);
if (ost->sync_ist != ist_table[ost->source_index]) if (ost->sync_ist != &input_streams[ost->source_index])
fprintf(stderr, " [sync #%d.%d]", fprintf(stderr, " [sync #%d.%d]",
ost->sync_ist->file_index, ost->sync_ist->file_index,
ost->sync_ist->st->index); ost->sync_ist->st->index);
@ -2539,12 +2509,12 @@ static int transcode(AVFormatContext **output_files,
double ipts, opts; double ipts, opts;
ost = ost_table[i]; ost = ost_table[i];
os = output_files[ost->file_index]; os = output_files[ost->file_index];
ist = ist_table[ost->source_index]; ist = &input_streams[ost->source_index];
if(ist->is_past_recording_time || no_packet[ist->file_index]) if(ist->is_past_recording_time || no_packet[ist->file_index])
continue; continue;
opts = ost->st->pts.val * av_q2d(ost->st->time_base); opts = ost->st->pts.val * av_q2d(ost->st->time_base);
ipts = (double)ist->pts; ipts = (double)ist->pts;
if (!file_table[ist->file_index].eof_reached){ if (!input_files[ist->file_index].eof_reached){
if(ipts < ipts_min) { if(ipts < ipts_min) {
ipts_min = ipts; ipts_min = ipts;
if(input_sync ) file_index = ist->file_index; if(input_sync ) file_index = ist->file_index;
@ -2575,7 +2545,7 @@ static int transcode(AVFormatContext **output_files,
break; break;
/* read a frame from it and output it in the fifo */ /* read a frame from it and output it in the fifo */
is = input_files[file_index]; is = input_files[file_index].ctx;
ret= av_read_frame(is, &pkt); ret= av_read_frame(is, &pkt);
if(ret == AVERROR(EAGAIN)){ if(ret == AVERROR(EAGAIN)){
no_packet[file_index]=1; no_packet[file_index]=1;
@ -2583,7 +2553,7 @@ static int transcode(AVFormatContext **output_files,
continue; continue;
} }
if (ret < 0) { if (ret < 0) {
file_table[file_index].eof_reached = 1; input_files[file_index].eof_reached = 1;
if (opt_shortest) if (opt_shortest)
break; break;
else else
@ -2599,10 +2569,10 @@ static int transcode(AVFormatContext **output_files,
} }
/* the following test is needed in case new streams appear /* the following test is needed in case new streams appear
dynamically in stream : we ignore them */ dynamically in stream : we ignore them */
if (pkt.stream_index >= file_table[file_index].nb_streams) if (pkt.stream_index >= input_files[file_index].nb_streams)
goto discard_packet; goto discard_packet;
ist_index = file_table[file_index].ist_index + pkt.stream_index; ist_index = input_files[file_index].ist_index + pkt.stream_index;
ist = ist_table[ist_index]; ist = &input_streams[ist_index];
if (ist->discard) if (ist->discard)
goto discard_packet; goto discard_packet;
@ -2661,8 +2631,8 @@ static int transcode(AVFormatContext **output_files,
} }
/* at the end of stream, we must flush the decoder buffers */ /* at the end of stream, we must flush the decoder buffers */
for(i=0;i<nb_istreams;i++) { for (i = 0; i < nb_input_streams; i++) {
ist = ist_table[i]; ist = &input_streams[i];
if (ist->decoding_needed) { if (ist->decoding_needed) {
output_packet(ist, i, ost_table, nb_ostreams, NULL); output_packet(ist, i, ost_table, nb_ostreams, NULL);
} }
@ -2692,8 +2662,8 @@ static int transcode(AVFormatContext **output_files,
} }
/* close each decoder */ /* close each decoder */
for(i=0;i<nb_istreams;i++) { for (i = 0; i < nb_input_streams; i++) {
ist = ist_table[i]; ist = &input_streams[i];
if (ist->decoding_needed) { if (ist->decoding_needed) {
avcodec_close(ist->st->codec); avcodec_close(ist->st->codec);
} }
@ -2704,15 +2674,7 @@ static int transcode(AVFormatContext **output_files,
fail: fail:
av_freep(&bit_buffer); av_freep(&bit_buffer);
av_free(file_table);
if (ist_table) {
for(i=0;i<nb_istreams;i++) {
ist = ist_table[i];
av_free(ist);
}
av_free(ist_table);
}
if (ost_table) { if (ost_table) {
for(i=0;i<nb_ostreams;i++) { for(i=0;i<nb_ostreams;i++) {
ost = ost_table[i]; ost = ost_table[i];
@ -3275,8 +3237,17 @@ static void opt_input_file(const char *filename)
for(i=0;i<ic->nb_streams;i++) { for(i=0;i<ic->nb_streams;i++) {
AVStream *st = ic->streams[i]; AVStream *st = ic->streams[i];
AVCodecContext *dec = st->codec; AVCodecContext *dec = st->codec;
AVInputStream *ist;
dec->thread_count = thread_count; dec->thread_count = thread_count;
input_codecs = grow_array(input_codecs, sizeof(*input_codecs), &nb_input_codecs, nb_input_codecs + 1); input_codecs = grow_array(input_codecs, sizeof(*input_codecs), &nb_input_codecs, nb_input_codecs + 1);
input_streams = grow_array(input_streams, sizeof(*input_streams), &nb_input_streams, nb_input_streams + 1);
ist = &input_streams[nb_input_streams - 1];
ist->st = st;
ist->file_index = nb_input_files;
ist->discard = 1;
switch (dec->codec_type) { switch (dec->codec_type) {
case AVMEDIA_TYPE_AUDIO: case AVMEDIA_TYPE_AUDIO:
input_codecs[nb_input_codecs-1] = avcodec_find_decoder_by_name(audio_codec_name); input_codecs[nb_input_codecs-1] = avcodec_find_decoder_by_name(audio_codec_name);
@ -3343,13 +3314,15 @@ static void opt_input_file(const char *filename)
} }
} }
input_files[nb_input_files] = ic;
input_files_ts_offset[nb_input_files] = input_ts_offset - (copy_ts ? 0 : timestamp); input_files_ts_offset[nb_input_files] = input_ts_offset - (copy_ts ? 0 : timestamp);
/* dump the file content */ /* dump the file content */
if (verbose >= 0) if (verbose >= 0)
av_dump_format(ic, nb_input_files, filename, 0); av_dump_format(ic, nb_input_files, filename, 0);
nb_input_files++; input_files = grow_array(input_files, sizeof(*input_files), &nb_input_files, nb_input_files + 1);
input_files[nb_input_files - 1].ctx = ic;
input_files[nb_input_files - 1].ist_index = nb_input_streams - ic->nb_streams;
input_files[nb_input_files - 1].nb_streams = ic->nb_streams;
video_channel = 0; video_channel = 0;
@ -3374,7 +3347,7 @@ static void check_inputs(int *has_video_ptr,
has_data = 0; has_data = 0;
for(j=0;j<nb_input_files;j++) { for(j=0;j<nb_input_files;j++) {
ic = input_files[j]; ic = input_files[j].ctx;
for(i=0;i<ic->nb_streams;i++) { for(i=0;i<ic->nb_streams;i++) {
AVCodecContext *enc = ic->streams[i]->codec; AVCodecContext *enc = ic->streams[i]->codec;
switch(enc->codec_type) { switch(enc->codec_type) {
@ -4070,9 +4043,9 @@ static void opt_target(const char *arg)
/* Try to determine PAL/NTSC by peeking in the input files */ /* Try to determine PAL/NTSC by peeking in the input files */
if(nb_input_files) { if(nb_input_files) {
int i, j; int i, j;
for(j = 0; j < nb_input_files; j++) { for (j = 0; j < nb_input_files; j++) {
for(i = 0; i < input_files[j]->nb_streams; i++) { for (i = 0; i < input_files[j].ctx->nb_streams; i++) {
AVCodecContext *c = input_files[j]->streams[i]->codec; AVCodecContext *c = input_files[j].ctx->streams[i]->codec;
if(c->codec_type != AVMEDIA_TYPE_VIDEO) if(c->codec_type != AVMEDIA_TYPE_VIDEO)
continue; continue;
fr = c->time_base.den * 1000 / c->time_base.num; fr = c->time_base.den * 1000 / c->time_base.num;