began to simplify code - use modified API for stream reading

Originally committed as revision 548 to svn://svn.ffmpeg.org/ffmpeg/trunk
pull/126/head
Fabrice Bellard 23 years ago
parent c9a65ca8c3
commit 79fdaa4cc0
  1. 507
      ffmpeg.c

@ -71,7 +71,8 @@ static int nb_output_files = 0;
static AVStreamMap stream_maps[MAX_FILES]; static AVStreamMap stream_maps[MAX_FILES];
static int nb_stream_maps; static int nb_stream_maps;
static AVFormat *file_format; static AVInputFormat *file_iformat;
static AVOutputFormat *file_oformat;
static int frame_width = 160; static int frame_width = 160;
static int frame_height = 128; static int frame_height = 128;
static int frame_rate = 25 * FRAME_RATE_BASE; static int frame_rate = 25 * FRAME_RATE_BASE;
@ -151,6 +152,7 @@ typedef struct AVInputFile {
int buffer_size; /* current total buffer size */ int buffer_size; /* current total buffer size */
int buffer_size_max; /* buffer size at which we consider we can stop int buffer_size_max; /* buffer size at which we consider we can stop
buffering */ buffering */
int nb_streams; /* nb streams we are aware of */
} AVInputFile; } AVInputFile;
#ifndef CONFIG_WIN32 #ifndef CONFIG_WIN32
@ -224,12 +226,12 @@ static int read_key(void)
int read_ffserver_streams(AVFormatContext *s, const char *filename) int read_ffserver_streams(AVFormatContext *s, const char *filename)
{ {
int i; int i, err;
AVFormatContext *ic; AVFormatContext *ic;
ic = av_open_input_file(filename, NULL, FFM_PACKET_SIZE, NULL); err = av_open_input_file(&ic, filename, NULL, FFM_PACKET_SIZE, NULL);
if (!ic) if (err < 0)
return -EIO; return err;
/* copy stream format */ /* copy stream format */
s->nb_streams = ic->nb_streams; s->nb_streams = ic->nb_streams;
for(i=0;i<ic->nb_streams;i++) { for(i=0;i<ic->nb_streams;i++) {
@ -281,7 +283,7 @@ static void do_audio_out(AVFormatContext *s,
&ost->fifo.rptr) == 0) { &ost->fifo.rptr) == 0) {
ret = avcodec_encode_audio(enc, audio_out, sizeof(audio_out), ret = avcodec_encode_audio(enc, audio_out, sizeof(audio_out),
(short *)audio_buf); (short *)audio_buf);
s->format->write_packet(s, ost->index, audio_out, ret, 0); s->oformat->write_packet(s, ost->index, audio_out, ret, 0);
} }
} else { } else {
/* output a pcm frame */ /* output a pcm frame */
@ -298,7 +300,7 @@ static void do_audio_out(AVFormatContext *s,
} }
ret = avcodec_encode_audio(enc, audio_out, size_out, ret = avcodec_encode_audio(enc, audio_out, size_out,
(short *)buftmp); (short *)buftmp);
s->format->write_packet(s, ost->index, audio_out, ret, 0); s->oformat->write_packet(s, ost->index, audio_out, ret, 0);
} }
} }
@ -394,7 +396,7 @@ static void write_picture(AVFormatContext *s, int index, AVPicture *picture,
default: default:
return; return;
} }
s->format->write_packet(s, index, buf, size, 0); s->oformat->write_packet(s, index, buf, size, 0);
av_free(buf); av_free(buf);
} }
@ -417,7 +419,7 @@ static void do_video_out(AVFormatContext *s,
frame_number = ist->frame_number; frame_number = ist->frame_number;
dec_frame_rate = ist->st->r_frame_rate; dec_frame_rate = ist->st->r_frame_rate;
//fprintf(stderr, "\n%d", dec_frame_rate); // fprintf(stderr, "\n%d", dec_frame_rate);
/* first drop frame if needed */ /* first drop frame if needed */
n1 = ((INT64)frame_number * enc->frame_rate) / dec_frame_rate; n1 = ((INT64)frame_number * enc->frame_rate) / dec_frame_rate;
n2 = (((INT64)frame_number + 1) * enc->frame_rate) / dec_frame_rate; n2 = (((INT64)frame_number + 1) * enc->frame_rate) / dec_frame_rate;
@ -495,7 +497,7 @@ static void do_video_out(AVFormatContext *s,
video_buffer, sizeof(video_buffer), video_buffer, sizeof(video_buffer),
picture); picture);
//enc->frame_number = enc->real_pict_num; //enc->frame_number = enc->real_pict_num;
s->format->write_packet(s, ost->index, video_buffer, ret, 0); s->oformat->write_packet(s, ost->index, video_buffer, ret, 0);
*frame_size = ret; *frame_size = ret;
//fprintf(stderr,"\nFrame: %3d %3d size: %5d type: %d", //fprintf(stderr,"\nFrame: %3d %3d size: %5d type: %d",
// enc->frame_number-1, enc->real_pict_num, ret, // enc->frame_number-1, enc->real_pict_num, ret,
@ -565,32 +567,6 @@ static void do_video_stats(AVOutputStream *ost,
} }
static void hex_dump(UINT8 *buf, int size)
{
int len, i, j, c;
for(i=0;i<size;i+=16) {
len = size - i;
if (len > 16)
len = 16;
printf("%08x ", i);
for(j=0;j<16;j++) {
if (j < len)
printf(" %02x", buf[i+j]);
else
printf(" ");
}
printf(" ");
for(j=0;j<len;j++) {
c = buf[i+j];
if (c < ' ' || c > '~')
c = '.';
printf("%c", c);
}
printf("\n");
}
}
/* /*
* The following code is the main loop of the file converter * The following code is the main loop of the file converter
*/ */
@ -618,6 +594,7 @@ static int av_encode(AVFormatContext **output_files,
for(i=0;i<nb_input_files;i++) { for(i=0;i<nb_input_files;i++) {
is = input_files[i]; is = input_files[i];
file_table[i].ist_index = j; file_table[i].ist_index = j;
file_table[i].nb_streams = is->nb_streams;
j += is->nb_streams; j += is->nb_streams;
} }
nb_istreams = j; nb_istreams = j;
@ -875,7 +852,7 @@ static int av_encode(AVFormatContext **output_files,
/* open files and write file headers */ /* open files and write file headers */
for(i=0;i<nb_output_files;i++) { for(i=0;i<nb_output_files;i++) {
os = output_files[i]; os = output_files[i];
if (os->format->write_header(os) < 0) { if (av_write_header(os) < 0) {
fprintf(stderr, "Could not write header for output file #%d (incorrect codec paramters ?)\n", i); fprintf(stderr, "Could not write header for output file #%d (incorrect codec paramters ?)\n", i);
ret = -EINVAL; ret = -EINVAL;
goto fail; goto fail;
@ -949,6 +926,10 @@ static int av_encode(AVFormatContext **output_files,
} else { } else {
stream_no_data = 0; stream_no_data = 0;
} }
/* the following test is needed in case new streams appear
dynamically in stream : we ignore them */
if (pkt.stream_index >= file_table[file_index].nb_streams)
continue;
ist_index = file_table[file_index].ist_index + pkt.stream_index; ist_index = file_table[file_index].ist_index + pkt.stream_index;
ist = ist_table[ist_index]; ist = ist_table[ist_index];
if (ist->discard) { if (ist->discard) {
@ -960,7 +941,7 @@ static int av_encode(AVFormatContext **output_files,
if (do_hex_dump) { if (do_hex_dump) {
printf("stream #%d, size=%d:\n", pkt.stream_index, pkt.size); printf("stream #%d, size=%d:\n", pkt.stream_index, pkt.size);
hex_dump(pkt.data, pkt.size); av_hex_dump(pkt.data, pkt.size);
} }
// printf("read #%d.%d size=%d\n", ist->file_index, ist->index, pkt.size); // printf("read #%d.%d size=%d\n", ist->file_index, ist->index, pkt.size);
@ -1093,7 +1074,7 @@ static int av_encode(AVFormatContext **output_files,
} else { } else {
/* no reencoding needed : output the packet directly */ /* no reencoding needed : output the packet directly */
/* force the input stream PTS */ /* force the input stream PTS */
os->format->write_packet(os, ost->index, data_buf, data_size, pkt.pts); os->oformat->write_packet(os, ost->index, data_buf, data_size, pkt.pts);
} }
} }
} }
@ -1219,7 +1200,7 @@ static int av_encode(AVFormatContext **output_files,
/* write the trailer if needed and close file */ /* write the trailer if needed and close file */
for(i=0;i<nb_output_files;i++) { for(i=0;i<nb_output_files;i++) {
os = output_files[i]; os = output_files[i];
os->format->write_trailer(os); av_write_trailer(os);
} }
/* finished ! */ /* finished ! */
@ -1300,14 +1281,12 @@ void show_licence(void)
void opt_format(const char *arg) void opt_format(const char *arg)
{ {
AVFormat *f; file_iformat = av_find_input_format(arg);
f = first_format; file_oformat = guess_format(arg, NULL, NULL);
while (f != NULL && strcmp(f->name, arg) != 0) f = f->next; if (!file_iformat && !file_oformat) {
if (f == NULL) { fprintf(stderr, "Unknown input or output format: %s\n", arg);
fprintf(stderr, "Invalid format: %s\n", arg);
exit(1); exit(1);
} }
file_format = f;
} }
void opt_video_bitrate(const char *arg) void opt_video_bitrate(const char *arg)
@ -1514,243 +1493,25 @@ void opt_recording_time(const char *arg)
recording_time = parse_date(arg, 1); recording_time = parse_date(arg, 1);
} }
/* return the number of packet read to find the codec parameters */ void print_error(const char *filename, int err)
int find_codec_parameters(AVFormatContext *ic)
{
int val, i, count, ret, got_picture, size;
AVCodec *codec;
AVCodecContext *enc;
AVStream *st;
AVPacket *pkt;
AVPicture picture;
AVPacketList *pktl=NULL, **ppktl;
short samples[AVCODEC_MAX_AUDIO_FRAME_SIZE / 2];
UINT8 *ptr;
count = 0;
ppktl = &ic->packet_buffer;
for(;;) {
for(i=0;i<ic->nb_streams;i++) {
enc = &ic->streams[i]->codec;
switch(enc->codec_type) {
case CODEC_TYPE_AUDIO:
val = enc->sample_rate;
break;
case CODEC_TYPE_VIDEO:
val = enc->width;
break;
default:
val = 1;
break;
}
/* if no parameters supplied, then we should read it from
the stream */
if (val == 0)
break;
}
if (i == ic->nb_streams) {
ret = count;
break;
}
if (count == 0) {
/* open each codec */
for(i=0;i<ic->nb_streams;i++) {
st = ic->streams[i];
codec = avcodec_find_decoder(st->codec.codec_id);
if (codec == NULL) {
ret = -1;
goto the_end;
}
ret = avcodec_open(&st->codec, codec);
if (ret < 0)
goto the_end;
}
}
pktl = av_mallocz(sizeof(AVPacketList));
if (!pktl) {
ret = -1;
break;
}
/* add the packet in the buffered packet list */
*ppktl = pktl;
ppktl = &pktl->next;
pkt = &pktl->pkt;
if (ic->format->read_packet(ic, pkt) < 0) {
ret = -1;
break;
}
st = ic->streams[pkt->stream_index];
/* decode the data and update codec parameters */
ptr = pkt->data;
size = pkt->size;
while (size > 0) {
switch(st->codec.codec_type) {
case CODEC_TYPE_VIDEO:
ret = avcodec_decode_video(&st->codec, &picture, &got_picture, ptr, size);
if (st->codec.codec_id == CODEC_ID_MPEG1VIDEO) {
//mpegvid = pkt->stream_index;
//fps = st->codec.frame_rate;
}
break;
case CODEC_TYPE_AUDIO:
ret = avcodec_decode_audio(&st->codec, samples, &got_picture, ptr, size);
break;
default:
ret = -1;
break;
}
if (ret < 0) {
ret = -1;
goto the_end;
}
if (got_picture)
break;
ptr += ret;
size -= ret;
}
count++;
}
the_end:
if (count > 0) {
/* close each codec */
for(i=0;i<ic->nb_streams;i++) {
st = ic->streams[i];
avcodec_close(&st->codec);
}
}
return ret;
}
/* Returns the real frame rate of telecine streams */
int get_real_fps(AVFormatContext *ic, AVFormat *fmt, AVFormatParameters *ap, int stream_id)
{
int frame_num, r_frames, fps, rfps;
int ret, got_picture, size;
UINT8 *ptr;
AVStream *st;
AVPacket *pkt;
AVPicture picture;
AVPacketList *pktl=NULL, **ppktl=NULL;
AVCodec *codec;
AVFormatContext *fc;
frame_num = 0;
r_frames = 0;
fps = rfps = -1;
if (stream_id < 0 || stream_id >= ic->nb_streams)
return -1;
/* We must use another AVFormatContext, and open the
file again, since we do not have good seeking */
fc = av_mallocz(sizeof(AVFormatContext));
if (!fc)
goto the_end;
strcpy(fc->filename, ic->filename);
/* open file */
if (!(fmt->flags & AVFMT_NOFILE)) {
if (url_fopen(&fc->pb, fc->filename, URL_RDONLY) < 0) {
fprintf(stderr, "Could not open '%s'\n", fc->filename);
exit(1);
}
}
/* check format */
if (!fmt) {
goto the_end;
}
fc->format = fmt;
/* Read header */
ret = fc->format->read_header(fc, ap);
if (ret < 0) {
fprintf(stderr, "%s: Error while parsing header\n", fc->filename);
goto the_end;
}
/* Find and open codec */
st = fc->streams[stream_id];
codec = avcodec_find_decoder(st->codec.codec_id);
if (codec == NULL) {
goto the_end;
}
ret = avcodec_open(&st->codec, codec);
if (ret < 0)
goto the_end;
ppktl = &fc->packet_buffer;
if (stream_id > -1) {
/* check telecine MPEG video streams, we */
/* decode 40 frames to get the real fps */
while(1) {
pktl = av_mallocz(sizeof(AVPacketList));
if (!pktl) {
goto the_end;
break;
}
/* add the packet in the buffered packet list */
*ppktl = pktl;
ppktl = &pktl->next;
pkt = &pktl->pkt;
if (fc->format->read_packet(fc, pkt) < 0) {
goto the_end;
break;
}
st = fc->streams[pkt->stream_index];
/* decode the video */
ptr = pkt->data;
size = pkt->size;
while (size > 0) {
if (pkt->stream_index == stream_id) {
ret = avcodec_decode_video(&st->codec, &picture, &got_picture, ptr, size);
fps = st->codec.frame_rate;
if (ret < 0) {
goto the_end;
}
if (got_picture) {
frame_num++;
r_frames += st->codec.repeat_pict;
}
}
ptr += ret;
size -= ret;
}
if (frame_num > 39)
break;
}
rfps = (fps * frame_num) / (frame_num + (r_frames >> 1));
/* close codec */
avcodec_close(&st->codec);
}
the_end:
/* FIXME: leak in packet_buffer */
av_free(fc);
return rfps;
}
int filename_number_test(const char *filename)
{ {
char buf[1024]; switch(err) {
case AVERROR_NUMEXPECTED:
if (get_frame_filename(buf, sizeof(buf), filename, 1) < 0) {
fprintf(stderr, "%s: Incorrect image filename syntax.\n" fprintf(stderr, "%s: Incorrect image filename syntax.\n"
"Use '%%d' to specify the image number:\n" "Use '%%d' to specify the image number:\n"
" for img1.jpg, img2.jpg, ..., use 'img%%d.jpg';\n" " for img1.jpg, img2.jpg, ..., use 'img%%d.jpg';\n"
" for img001.jpg, img002.jpg, ..., use 'img%%03d.jpg'.\n", " for img001.jpg, img002.jpg, ..., use 'img%%03d.jpg'.\n",
filename); filename);
return -1; break;
} else { case AVERROR_INVALIDDATA:
return 0; fprintf(stderr, "%s: Error while parsing header\n", filename);
break;
case AVERROR_NOFMT:
fprintf(stderr, "%s: Unknown format\n", filename);
break;
default:
fprintf(stderr, "%s: Error while opening file\n", filename);
break;
} }
} }
@ -1758,79 +1519,26 @@ void opt_input_file(const char *filename)
{ {
AVFormatContext *ic; AVFormatContext *ic;
AVFormatParameters params, *ap = &params; AVFormatParameters params, *ap = &params;
URLFormat url_format;
AVFormat *fmt;
int err, i, ret, rfps; int err, i, ret, rfps;
ic = av_mallocz(sizeof(AVFormatContext));
strcpy(ic->filename, filename);
/* first format guess to know if we must open file */
fmt = file_format;
if (!fmt)
fmt = guess_format(NULL, filename, NULL);
if (fmt == NULL || !(fmt->flags & AVFMT_NOFILE)) {
/* open file */
if (url_fopen(&ic->pb, filename, URL_RDONLY) < 0) {
fprintf(stderr, "Could not open '%s'\n", filename);
exit(1);
}
/* find format and set default parameters */
fmt = file_format;
err = url_getformat(url_fileno(&ic->pb), &url_format);
if (err >= 0) {
if (!fmt)
fmt = guess_format(url_format.format_name, NULL, NULL);
ap->sample_rate = url_format.sample_rate;
ap->frame_rate = url_format.frame_rate;
ap->channels = url_format.channels;
ap->width = url_format.width;
ap->height = url_format.height;
ap->pix_fmt = url_format.pix_fmt;
} else {
if (!fmt)
fmt = guess_format(NULL, filename, NULL);
memset(ap, 0, sizeof(*ap));
}
} else {
memset(ap, 0, sizeof(*ap));
}
if (!fmt || !fmt->read_header) {
fprintf(stderr, "%s: Unknown file format\n", filename);
exit(1);
}
ic->format = fmt;
/* get default parameters from command line */ /* get default parameters from command line */
if (!ap->sample_rate) memset(ap, 0, sizeof(*ap));
ap->sample_rate = audio_sample_rate; ap->sample_rate = audio_sample_rate;
if (!ap->channels) ap->channels = audio_channels;
ap->channels = audio_channels; ap->frame_rate = frame_rate;
ap->width = frame_width;
if (!ap->frame_rate) ap->height = frame_height;
ap->frame_rate = frame_rate;
if (!ap->width) /* open the input file with generic libav function */
ap->width = frame_width; err = av_open_input_file(&ic, filename, file_iformat, 0, ap);
if (!ap->height)
ap->height = frame_height;
/* check filename in case of an image number is expected */
if (ic->format->flags & AVFMT_NEEDNUMBER) {
if (filename_number_test(ic->filename) < 0)
exit(1);
}
err = ic->format->read_header(ic, ap);
if (err < 0) { if (err < 0) {
fprintf(stderr, "%s: Error while parsing header\n", filename); print_error(filename, err);
exit(1); exit(1);
} }
/* If not enough info for the codecs, we decode the first frames /* If not enough info to get the stream parameters, we decode the
to get it. (used in mpeg case for example) */ first frames to get it. (used in mpeg case for example) */
ret = find_codec_parameters(ic); ret = av_find_stream_info(ic);
if (ret < 0) { if (ret < 0) {
fprintf(stderr, "%s: could not find codec parameters\n", filename); fprintf(stderr, "%s: could not find codec parameters\n", filename);
exit(1); exit(1);
@ -1848,18 +1556,12 @@ void opt_input_file(const char *filename)
case CODEC_TYPE_VIDEO: case CODEC_TYPE_VIDEO:
frame_height = enc->height; frame_height = enc->height;
frame_width = enc->width; frame_width = enc->width;
rfps = enc->frame_rate; rfps = ic->streams[i]->r_frame_rate;
if (enc->codec_id == CODEC_ID_MPEG1VIDEO) { if (enc->frame_rate != rfps) {
rfps = get_real_fps(ic, fmt, ap, i);
}
if (rfps > 0 && rfps != enc->frame_rate) {
frame_rate = rfps;
fprintf(stderr,"\nSeems that stream %d comes from film source: %2.2f->%2.2f\n", fprintf(stderr,"\nSeems that stream %d comes from film source: %2.2f->%2.2f\n",
i, (float)enc->frame_rate / FRAME_RATE_BASE, i, (float)enc->frame_rate / FRAME_RATE_BASE,
(float)rfps / FRAME_RATE_BASE); (float)rfps / FRAME_RATE_BASE);
} else }
frame_rate = enc->frame_rate;
ic->streams[i]->r_frame_rate = rfps;
break; break;
default: default:
abort(); abort();
@ -1870,7 +1572,8 @@ void opt_input_file(const char *filename)
/* dump the file content */ /* dump the file content */
dump_format(ic, nb_input_files, filename, 0); dump_format(ic, nb_input_files, filename, 0);
nb_input_files++; nb_input_files++;
file_format = NULL; file_iformat = NULL;
file_oformat = NULL;
} }
void check_audio_video_inputs(int *has_video_ptr, int *has_audio_ptr) void check_audio_video_inputs(int *has_video_ptr, int *has_audio_ptr)
@ -1912,15 +1615,18 @@ void opt_output_file(const char *filename)
oc = av_mallocz(sizeof(AVFormatContext)); oc = av_mallocz(sizeof(AVFormatContext));
if (!file_format) { if (!file_oformat) {
file_format = guess_format(NULL, filename, NULL); file_oformat = guess_format(NULL, filename, NULL);
if (!file_format) if (!file_oformat) {
file_format = &mpeg_mux_format; fprintf(stderr, "Unable for find a suitable output format for '%s'\n",
filename);
exit(1);
}
} }
oc->format = file_format; oc->oformat = file_oformat;
if (!strcmp(file_format->name, "ffm") && if (!strcmp(file_oformat->name, "ffm") &&
strstart(filename, "http:", NULL)) { strstart(filename, "http:", NULL)) {
/* special case for files sent to ffserver: we get the stream /* special case for files sent to ffserver: we get the stream
parameters from ffserver */ parameters from ffserver */
@ -1929,8 +1635,8 @@ void opt_output_file(const char *filename)
exit(1); exit(1);
} }
} else { } else {
use_video = file_format->video_codec != CODEC_ID_NONE; use_video = file_oformat->video_codec != CODEC_ID_NONE;
use_audio = file_format->audio_codec != CODEC_ID_NONE; use_audio = file_oformat->audio_codec != CODEC_ID_NONE;
/* disable if no corresponding type found and at least one /* disable if no corresponding type found and at least one
input file */ input file */
@ -1961,7 +1667,7 @@ void opt_output_file(const char *filename)
} }
video_enc = &st->codec; video_enc = &st->codec;
codec_id = file_format->video_codec; codec_id = file_oformat->video_codec;
if (video_codec_id != CODEC_ID_NONE) if (video_codec_id != CODEC_ID_NONE)
codec_id = video_codec_id; codec_id = video_codec_id;
@ -2016,8 +1722,7 @@ void opt_output_file(const char *filename)
video_enc->me_method = me_method; video_enc->me_method = me_method;
/* XXX: need to find a way to set codec parameters */ /* XXX: need to find a way to set codec parameters */
if (oc->format == &ppm_format || if (oc->oformat->flags & AVFMT_RGB24) {
oc->format == &ppmpipe_format) {
video_enc->pix_fmt = PIX_FMT_RGB24; video_enc->pix_fmt = PIX_FMT_RGB24;
} }
@ -2034,7 +1739,7 @@ void opt_output_file(const char *filename)
exit(1); exit(1);
} }
audio_enc = &st->codec; audio_enc = &st->codec;
codec_id = file_format->audio_codec; codec_id = file_oformat->audio_codec;
if (audio_codec_id != CODEC_ID_NONE) if (audio_codec_id != CODEC_ID_NONE)
codec_id = audio_codec_id; codec_id = audio_codec_id;
audio_enc->codec_id = codec_id; audio_enc->codec_id = codec_id;
@ -2060,13 +1765,13 @@ void opt_output_file(const char *filename)
} }
if (str_title) if (str_title)
nstrcpy(oc->title, sizeof(oc->title), str_title); pstrcpy(oc->title, sizeof(oc->title), str_title);
if (str_author) if (str_author)
nstrcpy(oc->author, sizeof(oc->author), str_author); pstrcpy(oc->author, sizeof(oc->author), str_author);
if (str_copyright) if (str_copyright)
nstrcpy(oc->copyright, sizeof(oc->copyright), str_copyright); pstrcpy(oc->copyright, sizeof(oc->copyright), str_copyright);
if (str_comment) if (str_comment)
nstrcpy(oc->comment, sizeof(oc->comment), str_comment); pstrcpy(oc->comment, sizeof(oc->comment), str_comment);
} }
output_files[nb_output_files] = oc; output_files[nb_output_files] = oc;
@ -2077,12 +1782,14 @@ void opt_output_file(const char *filename)
strcpy(oc->filename, filename); strcpy(oc->filename, filename);
/* check filename in case of an image number is expected */ /* check filename in case of an image number is expected */
if (oc->format->flags & AVFMT_NEEDNUMBER) { if (oc->oformat->flags & AVFMT_NEEDNUMBER) {
if (filename_number_test(oc->filename) < 0) if (filename_number_test(oc->filename) < 0) {
print_error(oc->filename, AVERROR_NUMEXPECTED);
exit(1); exit(1);
}
} }
if (!(oc->format->flags & AVFMT_NOFILE)) { if (!(oc->oformat->flags & AVFMT_NOFILE)) {
/* test if it already exists to avoid loosing precious files */ /* test if it already exists to avoid loosing precious files */
if (!file_overwrite && if (!file_overwrite &&
(strchr(filename, ':') == NULL || (strchr(filename, ':') == NULL ||
@ -2108,7 +1815,8 @@ void opt_output_file(const char *filename)
} }
/* reset some options */ /* reset some options */
file_format = NULL; file_oformat = NULL;
file_iformat = NULL;
audio_disable = 0; audio_disable = 0;
video_disable = 0; video_disable = 0;
audio_codec_id = CODEC_ID_NONE; audio_codec_id = CODEC_ID_NONE;
@ -2162,20 +1870,22 @@ void prepare_grab(void)
} }
if (has_video) { if (has_video) {
ic = av_open_input_file("", "video_grab_device", 0, ap); AVInputFormat *fmt1;
/* by now video grab has one stream */ fmt1 = av_find_input_format("video_grab_device");
ic->streams[0]->r_frame_rate = ap->frame_rate; if (av_open_input_file(&ic, "", fmt1, 0, ap) < 0) {
if (!ic) {
fprintf(stderr, "Could not open video grab device\n"); fprintf(stderr, "Could not open video grab device\n");
exit(1); exit(1);
} }
/* by now video grab has one stream */
ic->streams[0]->r_frame_rate = ap->frame_rate;
input_files[nb_input_files] = ic; input_files[nb_input_files] = ic;
dump_format(ic, nb_input_files, v4l_device, 0); dump_format(ic, nb_input_files, v4l_device, 0);
nb_input_files++; nb_input_files++;
} }
if (has_audio) { if (has_audio) {
ic = av_open_input_file("", "audio_device", 0, ap); AVInputFormat *fmt1;
if (!ic) { fmt1 = av_find_input_format("audio_device");
if (av_open_input_file(&ic, "", fmt1, 0, ap) < 0) {
fprintf(stderr, "Could not open audio grab device\n"); fprintf(stderr, "Could not open audio grab device\n");
exit(1); exit(1);
} }
@ -2198,15 +1908,14 @@ void prepare_grab(void)
/* open the necessary output devices for playing */ /* open the necessary output devices for playing */
void prepare_play(void) void prepare_play(void)
{ {
#ifndef __BEOS__ AVOutputFormat *ofmt;
file_format = guess_format("audio_device", NULL, NULL); ofmt = guess_format("audio_device", NULL, NULL);
if (!file_format) { if (!ofmt) {
fprintf(stderr, "Could not find audio device\n"); fprintf(stderr, "Could not find audio device\n");
exit(1); exit(1);
} }
opt_output_file(audio_device); opt_output_file(audio_device);
#endif
} }
@ -2225,24 +1934,32 @@ INT64 getutime(void)
} }
#endif #endif
extern int ffm_nopts;
void opt_bitexact(void)
{
avcodec_set_bit_exact();
/* disable generate of real time pts in ffm (need to be supressed anyway) */
ffm_nopts = 1;
}
void show_formats(void) void show_formats(void)
{ {
AVFormat *f; AVInputFormat *ifmt;
AVOutputFormat *ofmt;
URLProtocol *up; URLProtocol *up;
AVCodec *p; AVCodec *p;
const char **pp; const char **pp;
printf("File formats:\n"); printf("File formats:\n");
printf(" Encoding:"); printf(" Encoding:");
for(f = first_format; f != NULL; f = f->next) { for(ofmt = first_oformat; ofmt != NULL; ofmt = ofmt->next) {
if (f->write_header) printf(" %s", ofmt->name);
printf(" %s", f->name);
} }
printf("\n"); printf("\n");
printf(" Decoding:"); printf(" Decoding:");
for(f = first_format; f != NULL; f = f->next) { for(ifmt = first_iformat; ifmt != NULL; ifmt = ifmt->next) {
if (f->read_header) printf(" %s", ifmt->name);
printf(" %s", f->name);
} }
printf("\n"); printf("\n");
@ -2380,7 +2097,7 @@ const OptionDef options[] = {
"dump each input packet" }, "dump each input packet" },
{ "psnr", OPT_BOOL | OPT_EXPERT, {(void*)&do_psnr}, "calculate PSNR of compressed frames" }, { "psnr", OPT_BOOL | OPT_EXPERT, {(void*)&do_psnr}, "calculate PSNR of compressed frames" },
{ "vstats", OPT_BOOL | OPT_EXPERT, {(void*)&do_vstats}, "dump video coding statistics to file" }, { "vstats", OPT_BOOL | OPT_EXPERT, {(void*)&do_vstats}, "dump video coding statistics to file" },
{ "bitexact", OPT_EXPERT, {(void*)opt_bitexact}, "only use bit exact algorithms (for codec testing)" },
{ NULL, }, { NULL, },
}; };
@ -2468,13 +2185,11 @@ int main(int argc, char **argv)
/* close files */ /* close files */
for(i=0;i<nb_output_files;i++) { for(i=0;i<nb_output_files;i++) {
if (!(output_files[i]->format->flags & AVFMT_NOFILE)) if (!(output_files[i]->oformat->flags & AVFMT_NOFILE))
url_fclose(&output_files[i]->pb); url_fclose(&output_files[i]->pb);
} }
for(i=0;i<nb_input_files;i++) { for(i=0;i<nb_input_files;i++)
if (!(input_files[i]->format->flags & AVFMT_NOFILE)) av_close_input_file(input_files[i]);
url_fclose(&input_files[i]->pb);
}
return 0; return 0;
} }

Loading…
Cancel
Save