/*
* FFmpeg main
- * Copyright (c) 2000,2001 Gerard Lantau
+ * Copyright (c) 2000, 2001, 2002 Fabrice Bellard
*
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
+ * This library 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 of the License, or (at your option) any later version.
*
- * This program is distributed in the hope that it will be useful,
+ * This library 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 General Public License for more details.
+ * 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 General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*/
-#include <stdlib.h>
-#include <stdio.h>
+#define HAVE_AV_CONFIG_H
+#include "avformat.h"
+#include "framehook.h"
+
+#ifndef CONFIG_WIN32
#include <unistd.h>
#include <fcntl.h>
#include <sys/ioctl.h>
-#include <errno.h>
#include <sys/time.h>
-#include <string.h>
-#include <sys/poll.h>
#include <termios.h>
+#include <sys/resource.h>
+#endif
+#ifdef CONFIG_OS2
+#include <sys/types.h>
+#include <sys/select.h>
+#include <stdlib.h>
+#endif
+#include <time.h>
#include <ctype.h>
-#include "avformat.h"
+#if !defined(INFINITY) && defined(HUGE_VAL)
+#define INFINITY HUGE_VAL
+#endif
+
+#define MAXINT64 int64_t_C(0x7fffffffffffffff)
typedef struct {
const char *name;
#define OPT_EXPERT 0x0004
#define OPT_STRING 0x0008
union {
- void (*func_arg)();
+ void (*func_arg)(const char *);
int *int_arg;
char **str_arg;
} u;
static AVStreamMap stream_maps[MAX_FILES];
static int nb_stream_maps;
-static AVFormat *file_format;
+static AVInputFormat *file_iformat;
+static AVOutputFormat *file_oformat;
+static AVImageFormat *image_format;
static int frame_width = 160;
static int frame_height = 128;
+static int frame_topBand = 0;
+static int frame_bottomBand = 0;
+static int frame_leftBand = 0;
+static int frame_rightBand = 0;
static int frame_rate = 25 * FRAME_RATE_BASE;
-static int video_bit_rate = 200000;
+extern int emulate_frame_rate;
+static int video_bit_rate = 200*1000;
+static int video_bit_rate_tolerance = 4000*1000;
static int video_qscale = 0;
+static int video_qmin = 2;
+static int video_qmax = 31;
+static int video_mb_qmin = 2;
+static int video_mb_qmax = 31;
+static int video_qdiff = 3;
+static float video_qblur = 0.5;
+static float video_qcomp = 0.5;
+#if 0 //experimental, (can be removed)
+static float video_rc_qsquish=1.0;
+static float video_rc_qmod_amp=0;
+static int video_rc_qmod_freq=0;
+#endif
+static char *video_rc_override_string=NULL;
+static char *video_rc_eq="tex^qComp";
+static int video_rc_buffer_size=0;
+static float video_rc_buffer_aggressivity=1.0;
+static int video_rc_max_rate=0;
+static int video_rc_min_rate=0;
+static float video_rc_initial_cplx=0;
+static float video_b_qfactor = 1.25;
+static float video_b_qoffset = 1.25;
+static float video_i_qfactor = -0.8;
+static float video_i_qoffset = 0.0;
+static int me_method = ME_EPZS;
static int video_disable = 0;
static int video_codec_id = CODEC_ID_NONE;
static int same_quality = 0;
+static int b_frames = 0;
+static int use_hq = 0;
+static int use_4mv = 0;
+static int do_deinterlace = 0;
+static int workaround_bugs = FF_BUG_AUTODETECT;
+static int error_resilience = 2;
+static int error_concealment = 3;
+static int dct_algo = 0;
+static int idct_algo = 0;
+static int use_part = 0;
+static int packet_size = 0;
+static int strict = 0;
+static int debug = 0;
static int gop_size = 12;
static int intra_only = 0;
static int audio_channels = 1;
static int audio_codec_id = CODEC_ID_NONE;
-static INT64 recording_time = 0;
+static int64_t recording_time = 0;
static int file_overwrite = 0;
static char *str_title = NULL;
static char *str_author = NULL;
static char *str_copyright = NULL;
static char *str_comment = NULL;
+static int do_benchmark = 0;
+static int do_hex_dump = 0;
+static int do_play = 0;
+static int do_psnr = 0;
+static int do_vstats = 0;
+static int do_pass = 0;
+static char *pass_logfilename = NULL;
+static int audio_stream_copy = 0;
+static int video_stream_copy = 0;
+
+static char *video_grab_format = "video4linux";
+static char *video_device = NULL;
+static int video_channel = 0;
+
+static char *audio_grab_format = "audio_device";
+static char *audio_device = NULL;
+
+#define DEFAULT_PASS_LOGFILENAME "ffmpeg2pass"
typedef struct AVOutputStream {
int file_index; /* file index */
int index; /* stream index in the output file */
int source_index; /* AVInputStream index */
AVStream *st; /* stream in the output file */
- int encoding_needed; /* true if encoding needed for this stream */
-
- int fifo_packet_rptr; /* read index in the corresponding
- avinputstream packet fifo */
+ int encoding_needed; /* true if encoding needed for this stream */
+ int frame_number;
+ /* input pts and corresponding output pts
+ for A/V sync */
+ double sync_ipts;
+ double sync_ipts_offset;
+ int64_t sync_opts;
/* video only */
- AVPicture pict_tmp; /* temporary image for resizing */
- int video_resample;
+ int video_resample; /* video_resample and video_crop are mutually exclusive */
+ AVPicture pict_tmp; /* temporary image for resampling */
ImgReSampleContext *img_resample_ctx; /* for image resampling */
+
+ int video_crop; /* video_resample and video_crop are mutually exclusive */
+ int topBand; /* cropping area sizes */
+ int leftBand;
/* audio only */
int audio_resample;
ReSampleContext *resample; /* for audio resampling */
FifoBuffer fifo; /* for compression: one audio fifo per codec */
+ FILE *logfile;
} AVOutputStream;
typedef struct AVInputStream {
AVStream *st;
int discard; /* true if stream data should be discarded */
int decoding_needed; /* true if the packets must be decoded in 'raw_fifo' */
- INT64 pts; /* current pts */
- int frame_number; /* current frame */
- INT64 sample_index; /* current sample */
+ int64_t sample_index; /* current sample */
+ int frame_decoded; /* true if a video or audio frame has been decoded */
} AVInputStream;
typedef struct AVInputFile {
int buffer_size; /* current total buffer size */
int buffer_size_max; /* buffer size at which we consider we can stop
buffering */
+ int nb_streams; /* nb streams we are aware of */
} AVInputFile;
+#ifndef CONFIG_WIN32
+
/* init terminal so that we can grab keys */
static struct termios oldtty;
tcsetattr (0, TCSANOW, &tty);
atexit(term_exit);
+#ifdef CONFIG_BEOS_NETSERVER
+ fcntl(0, F_SETFL, fcntl(0, F_GETFL) | O_NONBLOCK);
+#endif
}
/* read a key without blocking */
static int read_key(void)
{
- struct timeval tv;
- int n;
+ int n = 1;
unsigned char ch;
+#ifndef CONFIG_BEOS_NETSERVER
+ struct timeval tv;
fd_set rfds;
FD_ZERO(&rfds);
tv.tv_sec = 0;
tv.tv_usec = 0;
n = select(1, &rfds, NULL, NULL, &tv);
+#endif
if (n > 0) {
- if (read(0, &ch, 1) == 1)
+ n = read(0, &ch, 1);
+ if (n == 1)
return ch;
+
+ return n;
}
return -1;
}
-#define AUDIO_FIFO_SIZE 8192
-
-/* main loop for grabbing */
-int av_grab(AVFormatContext *s)
-{
- UINT8 audio_buf[AUDIO_FIFO_SIZE/2];
- UINT8 audio_buf1[AUDIO_FIFO_SIZE/2];
- UINT8 audio_out[AUDIO_FIFO_SIZE/2];
- UINT8 video_buffer[128*1024];
- char buf[256];
- short *samples;
- URLContext *audio_handle = NULL, *video_handle = NULL;
- int ret;
- AVCodecContext *enc, *first_video_enc = NULL;
- int frame_size, frame_bytes;
- int use_audio, use_video;
- int frame_rate, sample_rate, channels;
- int width, height, frame_number, i, pix_fmt = 0;
- AVOutputStream *ost_table[s->nb_streams], *ost;
- UINT8 *picture_in_buf = NULL, *picture_420p = NULL;
- int audio_fifo_size = 0, picture_size = 0;
- INT64 time_start;
-
- /* init output stream info */
- for(i=0;i<s->nb_streams;i++)
- ost_table[i] = NULL;
-
- /* output stream init */
- for(i=0;i<s->nb_streams;i++) {
- ost = av_mallocz(sizeof(AVOutputStream));
- if (!ost)
- goto fail;
- ost->index = i;
- ost->st = s->streams[i];
- ost_table[i] = ost;
- }
-
- use_audio = 0;
- use_video = 0;
- frame_rate = 0;
- sample_rate = 0;
- frame_size = 0;
- channels = 1;
- width = 0;
- height = 0;
- frame_number = 0;
-
- for(i=0;i<s->nb_streams;i++) {
- AVCodec *codec;
-
- ost = ost_table[i];
- enc = &ost->st->codec;
- codec = avcodec_find_encoder(enc->codec_id);
- if (!codec) {
- fprintf(stderr, "Unknown codec\n");
- return -1;
- }
- if (avcodec_open(enc, codec) < 0) {
- fprintf(stderr, "Incorrect encode parameters\n");
- return -1;
- }
- switch(enc->codec_type) {
- case CODEC_TYPE_AUDIO:
- use_audio = 1;
- if (enc->sample_rate > sample_rate)
- sample_rate = enc->sample_rate;
- if (enc->frame_size > frame_size)
- frame_size = enc->frame_size;
- if (enc->channels > channels)
- channels = enc->channels;
- break;
- case CODEC_TYPE_VIDEO:
- if (!first_video_enc)
- first_video_enc = enc;
- use_video = 1;
- if (enc->frame_rate > frame_rate)
- frame_rate = enc->frame_rate;
- if (enc->width > width)
- width = enc->width;
- if (enc->height > height)
- height = enc->height;
- break;
- }
- }
-
- /* audio */
- samples = NULL;
- if (use_audio) {
- snprintf(buf, sizeof(buf), "audio:%d,%d", sample_rate, channels);
- ret = url_open(&audio_handle, buf, URL_RDONLY);
- if (ret < 0) {
- fprintf(stderr, "Could not open audio device: disabling audio capture\n");
- use_audio = 0;
- } else {
- URLFormat f;
- /* read back exact grab parameters */
- if (url_getformat(audio_handle, &f) < 0) {
- fprintf(stderr, "could not read back video grab parameters\n");
- goto fail;
- }
- sample_rate = f.sample_rate;
- channels = f.channels;
- audio_fifo_size = ((AUDIO_FIFO_SIZE / 2) / audio_handle->packet_size) *
- audio_handle->packet_size;
- fprintf(stderr, "Audio sampling: %d Hz, %s\n",
- sample_rate, channels == 2 ? "stereo" : "mono");
- }
- }
-
- /* video */
- if (use_video) {
- snprintf(buf, sizeof(buf), "video:%d,%d,%f",
- width, height, (float)frame_rate / FRAME_RATE_BASE);
-
- ret = url_open(&video_handle, buf, URL_RDONLY);
- if (ret < 0) {
- fprintf(stderr,"Could not init video 4 linux capture: disabling video capture\n");
- use_video = 0;
- } else {
- URLFormat f;
- const char *pix_fmt_str;
- /* read back exact grab parameters */
- if (url_getformat(video_handle, &f) < 0) {
- fprintf(stderr, "could not read back video grab parameters\n");
- goto fail;
- }
- width = f.width;
- height = f.height;
- pix_fmt = f.pix_fmt;
- switch(pix_fmt) {
- case PIX_FMT_YUV420P:
- pix_fmt_str = "420P";
- break;
- case PIX_FMT_YUV422:
- pix_fmt_str = "422";
- break;
- case PIX_FMT_RGB24:
- pix_fmt_str = "RGB24";
- break;
- case PIX_FMT_BGR24:
- pix_fmt_str = "BGR24";
- break;
- default:
- pix_fmt_str = "???";
- break;
- }
- picture_size = video_handle->packet_size;
- picture_in_buf = malloc(picture_size);
- if (!picture_in_buf)
- goto fail;
- /* allocate a temporary picture if not grabbing in 420P format */
- if (pix_fmt != PIX_FMT_YUV420P) {
- picture_420p = malloc((width * height * 3) / 2);
- }
- fprintf(stderr, "Video sampling: %dx%d, %s format, %0.2f fps\n",
- width, height, pix_fmt_str, (float)frame_rate / FRAME_RATE_BASE);
- }
- }
-
- if (!use_video && !use_audio) {
- fprintf(stderr,"Could not open grab devices : exiting\n");
- exit(1);
- }
-
- /* init built in conversion functions */
- for(i=0;i<s->nb_streams;i++) {
- ost = ost_table[i];
- enc = &ost->st->codec;
- switch(enc->codec_type) {
- case CODEC_TYPE_AUDIO:
- ost->audio_resample = 0;
- if ((enc->channels != channels ||
- enc->sample_rate != sample_rate)) {
- ost->audio_resample = 1;
- ost->resample = audio_resample_init(enc->channels, channels,
- enc->sample_rate, sample_rate);
- }
- if (fifo_init(&ost->fifo, (2 * audio_fifo_size * enc->sample_rate) /
- sample_rate))
- goto fail;
- break;
- case CODEC_TYPE_VIDEO:
- ost->video_resample = 0;
- if (enc->width != width ||
- enc->height != height) {
- UINT8 *buf;
- ost->video_resample = 1;
- buf = malloc((enc->width * enc->height * 3) / 2);
- if (!buf)
- goto fail;
- ost->pict_tmp.data[0] = buf;
- ost->pict_tmp.data[1] = buf + enc->width * height;
- ost->pict_tmp.data[2] = ost->pict_tmp.data[1] + (enc->width * height) / 4;
- ost->pict_tmp.linesize[0] = enc->width;
- ost->pict_tmp.linesize[1] = enc->width / 2;
- ost->pict_tmp.linesize[2] = enc->width / 2;
- ost->img_resample_ctx = img_resample_init(
- ost->st->codec.width, ost->st->codec.height,
- width, height);
- }
- }
- }
-
- fprintf(stderr, "Press [q] to stop encoding\n");
-
- s->format->write_header(s);
- time_start = gettime();
- term_init();
-
- for(;;) {
- /* if 'q' pressed, exits */
- if (read_key() == 'q')
- break;
-
- /* read & compress audio frames */
- if (use_audio) {
- int ret, nb_samples, nb_samples_out;
- UINT8 *buftmp;
-
- for(;;) {
- ret = url_read(audio_handle, audio_buf, audio_fifo_size);
- if (ret <= 0)
- break;
- /* fill each codec fifo by doing the right sample
- rate conversion. This is not optimal because we
- do too much work, but it is easy to do */
- nb_samples = ret / (channels * 2);
- for(i=0;i<s->nb_streams;i++) {
- ost = ost_table[i];
- enc = &ost->st->codec;
- if (enc->codec_type == CODEC_TYPE_AUDIO) {
- /* rate & stereo convertion */
- if (!ost->audio_resample) {
- buftmp = audio_buf;
- nb_samples_out = nb_samples;
- } else {
- buftmp = audio_buf1;
- nb_samples_out = audio_resample(ost->resample,
- (short *)buftmp, (short *)audio_buf,
- nb_samples);
- }
- fifo_write(&ost->fifo, buftmp, nb_samples_out * enc->channels * 2,
- &ost->fifo.wptr);
- }
- }
-
- /* compress as many frame as possible with each audio codec */
- for(i=0;i<s->nb_streams;i++) {
- ost = ost_table[i];
- enc = &ost->st->codec;
- if (enc->codec_type == CODEC_TYPE_AUDIO) {
- frame_bytes = enc->frame_size * 2 * enc->channels;
-
- while (fifo_read(&ost->fifo, audio_buf,
- frame_bytes, &ost->fifo.rptr) == 0) {
- ret = avcodec_encode_audio(enc,
- audio_out, sizeof(audio_out),
- (short *)audio_buf);
- s->format->write_packet(s, ost->index, audio_out, ret);
- }
- }
- }
- }
- }
-
- if (use_video) {
- AVPicture *picture1;
- AVPicture picture;
- UINT8 *pict_buffer;
-
- ret = url_read(video_handle, picture_in_buf, picture_size);
- if (ret < 0)
- break;
- if (pix_fmt != PIX_FMT_YUV420P) {
- pict_buffer = picture_420p;
- img_convert_to_yuv420(pict_buffer, picture_in_buf, pix_fmt, width, height);
- } else {
- pict_buffer = picture_in_buf;
- }
- /* build a picture storage */
- picture.data[0] = pict_buffer;
- picture.data[1] = picture.data[0] + width * height;
- picture.data[2] = picture.data[1] + (width * height) / 4;
- picture.linesize[0] = width;
- picture.linesize[1] = width / 2;
- picture.linesize[2] = width / 2;
-
- for(i=0;i<s->nb_streams;i++) {
- ost = ost_table[i];
- enc = &ost->st->codec;
- if (enc->codec_type == CODEC_TYPE_VIDEO) {
- int n1, n2, nb;
-
- /* feed each codec with its requested frame rate */
- n1 = ((INT64)frame_number * enc->frame_rate) / frame_rate;
- n2 = (((INT64)frame_number + 1) * enc->frame_rate) / frame_rate;
- nb = n2 - n1;
- if (nb > 0) {
- /* resize the picture if needed */
- if (ost->video_resample) {
- picture1 = &ost->pict_tmp;
- img_resample(ost->img_resample_ctx,
- picture1, &picture);
- } else {
- picture1 = &picture;
- }
- ret = avcodec_encode_video(enc, video_buffer,
- sizeof(video_buffer),
- picture1);
- s->format->write_packet(s, ost->index, video_buffer, ret);
- }
- }
- }
- frame_number++;
- }
-
- /* write report */
- {
- char buf[1024];
- INT64 total_size;
- float ti, bitrate;
- static float last_ti;
- INT64 ti1;
-
- total_size = url_ftell(&s->pb);
- ti1 = gettime() - time_start;
- /* check elapsed time */
- if (recording_time && ti1 >= recording_time)
- break;
-
- ti = ti1 / 1000000.0;
- if (ti < 0.1)
- ti = 0.1;
- /* dispaly twice per second */
- if ((ti - last_ti) >= 0.5) {
- last_ti = ti;
- bitrate = (int)((total_size * 8) / ti / 1000.0);
-
- buf[0] = '\0';
- if (use_video) {
- sprintf(buf + strlen(buf), "frame=%5d fps=%4.1f q=%2d ",
- frame_number, (float)frame_number / ti, first_video_enc->quality);
- }
-
- sprintf(buf + strlen(buf), "size=%8LdkB time=%0.1f bitrate=%6.1fkbits/s",
- total_size / 1024, ti, bitrate);
- fprintf(stderr, "%s \r", buf);
- fflush(stderr);
- }
- }
- }
- term_exit();
-
- for(i=0;i<s->nb_streams;i++) {
- ost = ost_table[i];
- enc = &ost->st->codec;
- avcodec_close(enc);
- }
- s->format->write_trailer(s);
-
- if (audio_handle)
- url_close(audio_handle);
-
- if (video_handle)
- url_close(video_handle);
-
- /* write report */
- {
- float ti, bitrate;
- INT64 total_size;
+#else
- total_size = url_ftell(&s->pb);
-
- ti = (gettime() - time_start) / 1000000.0;
- if (ti < 0.1)
- ti = 0.1;
- bitrate = (int)((total_size * 8) / ti / 1000.0);
+/* no interactive support */
+static void term_exit(void)
+{
+}
- fprintf(stderr, "\033[K\nTotal time = %0.1f s, %Ld KBytes, %0.1f kbits/s\n",
- ti, total_size / 1024, bitrate);
- if (use_video) {
- fprintf(stderr, "Total frames = %d\n", frame_number);
- }
- }
+static void term_init(void)
+{
+}
- ret = 0;
- fail1:
- if (picture_in_buf)
- free(picture_in_buf);
- if (picture_420p)
- free(picture_420p);
- for(i=0;i<s->nb_streams;i++) {
- ost = ost_table[i];
- if (ost) {
- if (ost->fifo.buffer)
- fifo_free(&ost->fifo);
- if (ost->pict_tmp.data[0])
- free(ost->pict_tmp.data[0]);
- if (ost->video_resample)
- img_resample_close(ost->img_resample_ctx);
- if (ost->audio_resample)
- audio_resample_close(ost->resample);
- free(ost);
- }
- }
- return ret;
- fail:
- ret = -ENOMEM;
- goto fail1;
+static int read_key(void)
+{
+ return 0;
}
-int read_ffserver_streams(AVFormatContext *s, const char *filename)
+#endif
+
+static int read_ffserver_streams(AVFormatContext *s, const char *filename)
{
- int i;
+ int i, err;
AVFormatContext *ic;
- ic = av_open_input_file(filename, FFM_PACKET_SIZE);
- if (!ic)
- return -EIO;
+ err = av_open_input_file(&ic, filename, NULL, FFM_PACKET_SIZE, NULL);
+ if (err < 0)
+ return err;
/* copy stream format */
s->nb_streams = ic->nb_streams;
for(i=0;i<ic->nb_streams;i++) {
AVStream *st;
+
st = av_mallocz(sizeof(AVFormatContext));
memcpy(st, ic->streams[i], sizeof(AVStream));
s->streams[i] = st;
return 0;
}
-#define MAX_AUDIO_PACKET_SIZE 16384
+#define MAX_AUDIO_PACKET_SIZE (128 * 1024)
static void do_audio_out(AVFormatContext *s,
AVOutputStream *ost,
AVInputStream *ist,
unsigned char *buf, int size)
{
- UINT8 *buftmp;
- UINT8 audio_buf[2*MAX_AUDIO_PACKET_SIZE]; /* XXX: allocate it */
- UINT8 audio_out[MAX_AUDIO_PACKET_SIZE]; /* XXX: allocate it */
+ uint8_t *buftmp;
+ uint8_t audio_buf[2*MAX_AUDIO_PACKET_SIZE]; /* XXX: allocate it */
+ uint8_t audio_out[4*MAX_AUDIO_PACKET_SIZE]; /* XXX: allocate it - yep really WMA */
int size_out, frame_bytes, ret;
AVCodecContext *enc;
}
/* now encode as many frames as possible */
- if (enc->codec_id != CODEC_ID_PCM) {
+ if (enc->frame_size > 1) {
/* output resampled raw samples */
fifo_write(&ost->fifo, buftmp, size_out,
&ost->fifo.wptr);
while (fifo_read(&ost->fifo, audio_buf, frame_bytes,
&ost->fifo.rptr) == 0) {
- ret = avcodec_encode_audio(enc,
- audio_out, sizeof(audio_out), (short *)audio_buf);
- s->format->write_packet(s, ost->index, audio_out, ret);
+ ret = avcodec_encode_audio(enc, audio_out, sizeof(audio_out),
+ (short *)audio_buf);
+ av_write_frame(s, ost->index, audio_out, ret);
}
} else {
- /* XXX: handle endianness */
- s->format->write_packet(s, ost->index, buftmp, size_out);
+ /* output a pcm frame */
+ /* XXX: change encoding codec API to avoid this ? */
+ switch(enc->codec->id) {
+ case CODEC_ID_PCM_S16LE:
+ case CODEC_ID_PCM_S16BE:
+ case CODEC_ID_PCM_U16LE:
+ case CODEC_ID_PCM_U16BE:
+ break;
+ default:
+ size_out = size_out >> 1;
+ break;
+ }
+ ret = avcodec_encode_audio(enc, audio_out, size_out,
+ (short *)buftmp);
+ av_write_frame(s, ost->index, audio_out, ret);
}
}
/* write a picture to a raw mux */
-static void write_picture(AVFormatContext *s, int index, AVPicture *picture, int w, int h)
+static void write_picture(AVFormatContext *s, int index, AVPicture *picture,
+ int pix_fmt, int w, int h)
{
- UINT8 *buf, *src, *dest;
+ uint8_t *buf, *src, *dest;
int size, j, i;
+
/* XXX: not efficient, should add test if we can take
directly the AVPicture */
- size = (w * h) * 3 / 2;
- buf = malloc(size);
- dest = buf;
- for(i=0;i<3;i++) {
- if (i == 1) {
- w >>= 1;
- h >>= 1;
+ switch(pix_fmt) {
+ case PIX_FMT_YUV420P:
+ size = avpicture_get_size(pix_fmt, w, h);
+ buf = av_malloc(size);
+ if (!buf)
+ return;
+ dest = buf;
+ for(i=0;i<3;i++) {
+ if (i == 1) {
+ w >>= 1;
+ h >>= 1;
+ }
+ src = picture->data[i];
+ for(j=0;j<h;j++) {
+ memcpy(dest, src, w);
+ dest += w;
+ src += picture->linesize[i];
+ }
}
- src = picture->data[i];
+ break;
+ case PIX_FMT_YUV422P:
+ size = (w * h) * 2;
+ buf = av_malloc(size);
+ if (!buf)
+ return;
+ dest = buf;
+ for(i=0;i<3;i++) {
+ if (i == 1) {
+ w >>= 1;
+ }
+ src = picture->data[i];
+ for(j=0;j<h;j++) {
+ memcpy(dest, src, w);
+ dest += w;
+ src += picture->linesize[i];
+ }
+ }
+ break;
+ case PIX_FMT_YUV444P:
+ size = (w * h) * 3;
+ buf = av_malloc(size);
+ if (!buf)
+ return;
+ dest = buf;
+ for(i=0;i<3;i++) {
+ src = picture->data[i];
+ for(j=0;j<h;j++) {
+ memcpy(dest, src, w);
+ dest += w;
+ src += picture->linesize[i];
+ }
+ }
+ break;
+ case PIX_FMT_YUV422:
+ size = (w * h) * 2;
+ buf = av_malloc(size);
+ if (!buf)
+ return;
+ dest = buf;
+ src = picture->data[0];
+ for(j=0;j<h;j++) {
+ memcpy(dest, src, w * 2);
+ dest += w * 2;
+ src += picture->linesize[0];
+ }
+ break;
+ case PIX_FMT_RGB24:
+ case PIX_FMT_BGR24:
+ size = (w * h) * 3;
+ buf = av_malloc(size);
+ if (!buf)
+ return;
+ dest = buf;
+ src = picture->data[0];
for(j=0;j<h;j++) {
- memcpy(dest, src, w);
- dest += w;
- src += picture->linesize[i];
+ memcpy(dest, src, w * 3);
+ dest += w * 3;
+ src += picture->linesize[0];
+ }
+ break;
+ default:
+ return;
+ }
+ av_write_frame(s, index, buf, size);
+ av_free(buf);
+}
+
+static void pre_process_video_frame(AVInputStream *ist, AVPicture *picture, void **bufp)
+{
+ AVCodecContext *dec;
+ AVPicture *picture2;
+ AVPicture picture_tmp;
+ uint8_t *buf = 0;
+
+ dec = &ist->st->codec;
+
+ /* deinterlace : must be done before any resize */
+ if (do_deinterlace) {
+ int size;
+
+ /* create temporary picture */
+ size = avpicture_get_size(dec->pix_fmt, dec->width, dec->height);
+ buf = av_malloc(size);
+ if (!buf)
+ return;
+
+ picture2 = &picture_tmp;
+ avpicture_fill(picture2, buf, dec->pix_fmt, dec->width, dec->height);
+
+ if (avpicture_deinterlace(picture2, picture,
+ dec->pix_fmt, dec->width, dec->height) < 0) {
+ /* if error, do not deinterlace */
+ av_free(buf);
+ buf = NULL;
+ picture2 = picture;
}
+ } else {
+ picture2 = picture;
}
- s->format->write_packet(s, index, buf, size);
- free(buf);
+
+ frame_hook_process(picture2, dec->pix_fmt, dec->width, dec->height);
+
+ if (picture != picture2)
+ *picture = *picture2;
+ *bufp = buf;
}
+/* we begin to correct av delay at this threshold */
+#define AV_DELAY_MAX 0.100
static void do_video_out(AVFormatContext *s,
AVOutputStream *ost,
AVInputStream *ist,
- AVPicture *pict)
+ AVPicture *in_picture,
+ int *frame_size, AVOutputStream *audio_sync)
{
- int n1, n2, nb, i, ret, frame_number;
- AVPicture *picture;
- UINT8 video_buffer[128*1024];
- AVCodecContext *enc;
+ int nb_frames, i, ret;
+ AVPicture *final_picture, *formatted_picture;
+ AVPicture picture_format_temp, picture_crop_temp;
+ static uint8_t *video_buffer;
+ uint8_t *buf = NULL, *buf1 = NULL;
+ AVCodecContext *enc, *dec;
+
+#define VIDEO_BUFFER_SIZE (1024*1024)
enc = &ost->st->codec;
+ dec = &ist->st->codec;
+
+ /* by default, we output a single frame */
+ nb_frames = 1;
+
+ *frame_size = 0;
+
+ /* NOTE: the A/V sync is always done by considering the audio is
+ the master clock. It is suffisant for transcoding or playing,
+ but not for the general case */
+ if (audio_sync) {
+ /* compute the A-V delay and duplicate/remove frames if needed */
+ double adelta, vdelta, apts, vpts, av_delay;
+
+ if (audio_sync->sync_ipts != AV_NOPTS_VALUE &&
+ ost->sync_ipts != AV_NOPTS_VALUE) {
+
+ adelta = (double)(ost->st->pts.val - audio_sync->sync_opts) *
+ s->pts_num / s->pts_den;
+ apts = audio_sync->sync_ipts + adelta;
+
+ vdelta = (double)(ost->st->pts.val - ost->sync_opts) *
+ s->pts_num / s->pts_den;
+ vpts = ost->sync_ipts + vdelta;
+
+ av_delay = apts - vpts;
+ // printf("delay=%f\n", av_delay);
+ if (av_delay < -AV_DELAY_MAX)
+ nb_frames = 2;
+ else if (av_delay > AV_DELAY_MAX)
+ nb_frames = 0;
+ }
+ } else {
+ double vdelta;
+
+ if (ost->sync_ipts != AV_NOPTS_VALUE) {
+ vdelta = (double)(ost->st->pts.val) * s->pts_num / s->pts_den - (ost->sync_ipts - ost->sync_ipts_offset);
+ if (vdelta < 100 && vdelta > -100 && ost->sync_ipts_offset) {
+ if (vdelta < -AV_DELAY_MAX)
+ nb_frames = 2;
+ else if (vdelta > AV_DELAY_MAX)
+ nb_frames = 0;
+ } else {
+ ost->sync_ipts_offset -= vdelta;
+ if (!ost->sync_ipts_offset)
+ ost->sync_ipts_offset = 0.000001; /* one microsecond */
+ }
+
+#if defined(PJSG)
+ {
+ static char *action[] = { "drop frame", "copy frame", "dup frame" };
+ printf("Input PTS %12.6f, output PTS %12.6f: %s\n",
+ (double) ost->sync_ipts, (double) ost->st->pts.val * s->pts_num / s->pts_den,
+ action[nb_frames]);
+ }
+#endif
+ }
+ }
+ if (nb_frames <= 0)
+ return;
- frame_number = ist->frame_number;
- /* first drop frame if needed */
- n1 = ((INT64)frame_number * enc->frame_rate) / ist->st->codec.frame_rate;
- n2 = (((INT64)frame_number + 1) * enc->frame_rate) / ist->st->codec.frame_rate;
- nb = n2 - n1;
- if (nb <= 0)
+ if (!video_buffer)
+ video_buffer = av_malloc(VIDEO_BUFFER_SIZE);
+ if (!video_buffer)
return;
-
- if (ost->video_resample) {
- picture = &ost->pict_tmp;
- img_resample(ost->img_resample_ctx, picture, pict);
+
+ /* convert pixel format if needed */
+ if (enc->pix_fmt != dec->pix_fmt) {
+ int size;
+
+ /* create temporary picture */
+ size = avpicture_get_size(enc->pix_fmt, dec->width, dec->height);
+ buf = av_malloc(size);
+ if (!buf)
+ return;
+ formatted_picture = &picture_format_temp;
+ avpicture_fill(formatted_picture, buf, enc->pix_fmt, dec->width, dec->height);
+
+ if (img_convert(formatted_picture, enc->pix_fmt,
+ in_picture, dec->pix_fmt,
+ dec->width, dec->height) < 0) {
+ fprintf(stderr, "pixel format conversion not handled\n");
+ goto the_end;
+ }
} else {
- picture = pict;
+ formatted_picture = in_picture;
}
+ /* XXX: resampling could be done before raw format convertion in
+ some cases to go faster */
+ /* XXX: only works for YUV420P */
+ if (ost->video_resample) {
+ final_picture = &ost->pict_tmp;
+ img_resample(ost->img_resample_ctx, final_picture, formatted_picture);
+ } else if (ost->video_crop) {
+ picture_crop_temp.data[0] = formatted_picture->data[0] +
+ (ost->topBand * formatted_picture->linesize[0]) + ost->leftBand;
+
+ picture_crop_temp.data[1] = formatted_picture->data[1] +
+ ((ost->topBand >> 1) * formatted_picture->linesize[1]) +
+ (ost->leftBand >> 1);
+
+ picture_crop_temp.data[2] = formatted_picture->data[2] +
+ ((ost->topBand >> 1) * formatted_picture->linesize[2]) +
+ (ost->leftBand >> 1);
+
+ picture_crop_temp.linesize[0] = formatted_picture->linesize[0];
+ picture_crop_temp.linesize[1] = formatted_picture->linesize[1];
+ picture_crop_temp.linesize[2] = formatted_picture->linesize[2];
+ final_picture = &picture_crop_temp;
+ } else {
+ final_picture = formatted_picture;
+ }
/* duplicates frame if needed */
/* XXX: pb because no interleaving */
- for(i=0;i<nb;i++) {
+ for(i=0;i<nb_frames;i++) {
if (enc->codec_id != CODEC_ID_RAWVIDEO) {
+ AVFrame big_picture;
+
+ memset(&big_picture, 0, sizeof(AVFrame));
+ *(AVPicture*)&big_picture= *final_picture;
+
/* handles sameq here. This is not correct because it may
not be a global option */
if (same_quality) {
- ost->st->codec.quality = ist->st->codec.quality;
+ big_picture.quality = ist->st->quality;
+ }else
+ big_picture.quality = ost->st->quality;
+
+ ret = avcodec_encode_video(enc,
+ video_buffer, VIDEO_BUFFER_SIZE,
+ &big_picture);
+ //enc->frame_number = enc->real_pict_num;
+ av_write_frame(s, ost->index, video_buffer, ret);
+ *frame_size = ret;
+ //fprintf(stderr,"\nFrame: %3d %3d size: %5d type: %d",
+ // enc->frame_number-1, enc->real_pict_num, ret,
+ // enc->pict_type);
+ /* if two pass, output log */
+ if (ost->logfile && enc->stats_out) {
+ fprintf(ost->logfile, "%s", enc->stats_out);
}
- ret = avcodec_encode_video(&ost->st->codec,
- video_buffer, sizeof(video_buffer),
- picture);
- s->format->write_packet(s, ost->index, video_buffer, ret);
} else {
- write_picture(s, ost->index, picture, enc->width, enc->height);
+ if (s->oformat->flags & AVFMT_RAWPICTURE) {
+ /* raw pictures are written as AVPicture structure to
+ avoid any copies. We support temorarily the older
+ method. */
+ av_write_frame(s, ost->index,
+ (uint8_t *)final_picture, sizeof(AVPicture));
+ } else {
+ write_picture(s, ost->index, final_picture, enc->pix_fmt,
+ enc->width, enc->height);
+ }
}
+ ost->frame_number++;
}
+ the_end:
+ av_free(buf);
+ av_free(buf1);
+}
+
+static double psnr(double d){
+ if(d==0) return INFINITY;
+ return -10.0*log(d)/log(10.0);
}
-//#define HEX_DUMP
+static void do_video_stats(AVFormatContext *os, AVOutputStream *ost,
+ int frame_size)
+{
+ static FILE *fvstats=NULL;
+ static int64_t total_size = 0;
+ char filename[40];
+ time_t today2;
+ struct tm *today;
+ AVCodecContext *enc;
+ int frame_number;
+ int64_t ti;
+ double ti1, bitrate, avg_bitrate;
+
+ if (!fvstats) {
+ today2 = time(NULL);
+ today = localtime(&today2);
+ sprintf(filename, "vstats_%02d%02d%02d.log", today->tm_hour,
+ today->tm_min,
+ today->tm_sec);
+ fvstats = fopen(filename,"w");
+ if (!fvstats) {
+ perror("fopen");
+ exit(1);
+ }
+ }
+
+ ti = MAXINT64;
+ enc = &ost->st->codec;
+ total_size += frame_size;
+ if (enc->codec_type == CODEC_TYPE_VIDEO) {
+ frame_number = ost->frame_number;
+ fprintf(fvstats, "frame= %5d q= %2.1f ", frame_number, enc->coded_frame->quality);
+ if (enc->flags&CODEC_FLAG_PSNR)
+ fprintf(fvstats, "PSNR= %6.2f ", psnr(enc->coded_frame->error[0]/(enc->width*enc->height*255.0*255.0)));
+
+ fprintf(fvstats,"f_size= %6d ", frame_size);
+ /* compute pts value */
+ ti1 = (double)ost->st->pts.val * os->pts_num / os->pts_den;
+ if (ti1 < 0.01)
+ ti1 = 0.01;
+
+ bitrate = (double)(frame_size * 8) * enc->frame_rate / FRAME_RATE_BASE / 1000.0;
+ avg_bitrate = (double)(total_size * 8) / ti1 / 1000.0;
+ fprintf(fvstats, "s_size= %8.0fkB time= %0.3f br= %7.1fkbits/s avg_br= %7.1fkbits/s ",
+ (double)total_size / 1024, ti1, bitrate, avg_bitrate);
+ fprintf(fvstats,"type= %s\n", enc->coded_frame->key_frame == 1 ? "I" : "P");
+ }
+}
-#ifdef HEX_DUMP
-static void hex_dump(UINT8 *buf, int size)
+static void print_report(AVFormatContext **output_files,
+ AVOutputStream **ost_table, int nb_ostreams,
+ int is_last_report)
{
- int len, i, j, c;
+ char buf[1024];
+ AVOutputStream *ost;
+ AVFormatContext *oc, *os;
+ int64_t total_size;
+ AVCodecContext *enc;
+ int frame_number, vid, i;
+ double bitrate, ti1, pts;
+ static int64_t last_time = -1;
+
+ if (!is_last_report) {
+ int64_t cur_time;
+ /* display the report every 0.5 seconds */
+ cur_time = av_gettime();
+ if (last_time == -1) {
+ last_time = cur_time;
+ return;
+ }
+ if ((cur_time - last_time) < 500000)
+ return;
+ last_time = cur_time;
+ }
+
- 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(" ");
+ oc = output_files[0];
+
+ total_size = url_ftell(&oc->pb);
+
+ buf[0] = '\0';
+ ti1 = 1e10;
+ vid = 0;
+ for(i=0;i<nb_ostreams;i++) {
+ ost = ost_table[i];
+ os = output_files[ost->file_index];
+ enc = &ost->st->codec;
+ if (vid && enc->codec_type == CODEC_TYPE_VIDEO) {
+ sprintf(buf + strlen(buf), "q=%2.1f ",
+ enc->coded_frame->quality);
}
- printf(" ");
- for(j=0;j<len;j++) {
- c = buf[i+j];
- if (c < ' ' || c > '~')
- c = '.';
- printf("%c", c);
+ if (!vid && enc->codec_type == CODEC_TYPE_VIDEO) {
+ frame_number = ost->frame_number;
+ sprintf(buf + strlen(buf), "frame=%5d q=%2.1f ",
+ frame_number, enc->coded_frame ? enc->coded_frame->quality : 0);
+ if (enc->flags&CODEC_FLAG_PSNR)
+ sprintf(buf + strlen(buf), "PSNR= %6.2f ", psnr(enc->coded_frame->error[0]/(enc->width*enc->height*255.0*255.0)));
+ vid = 1;
}
- printf("\n");
+ /* compute min output value */
+ pts = (double)ost->st->pts.val * os->pts_num / os->pts_den;
+ if ((pts < ti1) && (pts > 0))
+ ti1 = pts;
+ }
+ if (ti1 < 0.01)
+ ti1 = 0.01;
+ bitrate = (double)(total_size * 8) / ti1 / 1000.0;
+
+ sprintf(buf + strlen(buf),
+ "size=%8.0fkB time=%0.1f bitrate=%6.1fkbits/s",
+ (double)total_size / 1024, ti1, bitrate);
+
+ fprintf(stderr, "%s ", buf);
+
+ if (is_last_report) {
+ fprintf(stderr, "\n");
+ } else {
+ fprintf(stderr, "\r");
+ fflush(stderr);
}
}
-#endif
/*
* The following code is the main loop of the file converter
int nb_input_files,
AVStreamMap *stream_maps, int nb_stream_maps)
{
- int ret, i, j, k, n, nb_istreams, nb_ostreams = 0;
+ int ret, i, j, k, n, nb_istreams = 0, nb_ostreams = 0, pts_set;
AVFormatContext *is, *os;
AVCodecContext *codec, *icodec;
AVOutputStream *ost, **ost_table = NULL;
AVInputStream *ist, **ist_table = NULL;
- INT64 min_pts, start_time;
- AVInputFile file_table[nb_input_files];
+ AVInputFile *file_table;
+ AVFormatContext *stream_no_data;
+ int key;
+
+ file_table= (AVInputFile*) av_mallocz(nb_input_files * sizeof(AVInputFile));
+ if (!file_table)
+ goto fail;
- memset(file_table, 0, sizeof(file_table));
-
/* 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)
- return -ENOMEM;
+ goto fail;
for(i=0;i<nb_istreams;i++) {
ist = av_mallocz(sizeof(AVInputStream));
}
}
- /* dump the stream mapping */
- fprintf(stderr, "Stream mapping:\n");
- for(i=0;i<nb_ostreams;i++) {
- ost = ost_table[i];
- fprintf(stderr, " Stream #%d.%d -> #%d.%d\n",
- ist_table[ost->source_index]->file_index,
- ist_table[ost->source_index]->index,
- ost->file_index,
- ost->index);
- }
-
/* for each output stream, we compute the right encoding parameters */
for(i=0;i<nb_ostreams;i++) {
ost = ost_table[i];
codec = &ost->st->codec;
icodec = &ist->st->codec;
- switch(codec->codec_type) {
- case CODEC_TYPE_AUDIO:
- /* check if same codec with same parameters. If so, no
- reencoding is needed */
- if (codec->codec_id == icodec->codec_id &&
- codec->bit_rate == icodec->bit_rate &&
- codec->sample_rate == icodec->sample_rate &&
- codec->channels == icodec->channels) {
- /* no reencoding */
- } else {
- if (fifo_init(&ost->fifo, 2 * MAX_AUDIO_PACKET_SIZE))
+ if (ost->st->stream_copy) {
+ /* if stream_copy is selected, no need to decode or encode */
+ codec->codec_id = icodec->codec_id;
+ codec->codec_type = icodec->codec_type;
+ codec->codec_tag = icodec->codec_tag;
+ codec->bit_rate = icodec->bit_rate;
+ switch(codec->codec_type) {
+ case CODEC_TYPE_AUDIO:
+ codec->sample_rate = icodec->sample_rate;
+ codec->channels = icodec->channels;
+ break;
+ case CODEC_TYPE_VIDEO:
+ codec->frame_rate = icodec->frame_rate;
+ codec->width = icodec->width;
+ codec->height = icodec->height;
+ break;
+ default:
+ av_abort();
+ }
+ } else {
+ switch(codec->codec_type) {
+ case CODEC_TYPE_AUDIO:
+ if (fifo_init(&ost->fifo, 2 * MAX_AUDIO_PACKET_SIZE))
goto fail;
-
+
if (codec->channels == icodec->channels &&
codec->sample_rate == icodec->sample_rate) {
ost->audio_resample = 0;
} else {
- ost->audio_resample = 1;
- ost->resample = audio_resample_init(codec->channels, icodec->channels,
+ if (codec->channels != icodec->channels &&
+ icodec->codec_id == CODEC_ID_AC3) {
+ /* Special case for 5:1 AC3 input */
+ /* and mono or stereo output */
+ /* Request specific number of channels */
+ icodec->channels = codec->channels;
+ if (codec->sample_rate == icodec->sample_rate)
+ ost->audio_resample = 0;
+ else {
+ ost->audio_resample = 1;
+ ost->resample = audio_resample_init(codec->channels, icodec->channels,
codec->sample_rate,
icodec->sample_rate);
+ }
+ /* Request specific number of channels */
+ icodec->channels = codec->channels;
+ } else {
+ ost->audio_resample = 1;
+ ost->resample = audio_resample_init(codec->channels, icodec->channels,
+ codec->sample_rate,
+ icodec->sample_rate);
+ }
}
ist->decoding_needed = 1;
ost->encoding_needed = 1;
- }
- break;
- case CODEC_TYPE_VIDEO:
- /* check if same codec with same parameters. If so, no
- reencoding is needed */
- if (codec->codec_id == icodec->codec_id &&
- codec->bit_rate == icodec->bit_rate &&
- codec->frame_rate == icodec->frame_rate &&
- codec->width == icodec->width &&
- codec->height == icodec->height) {
- /* no reencoding */
- } else {
+ break;
+ case CODEC_TYPE_VIDEO:
if (codec->width == icodec->width &&
- codec->height == icodec->height) {
+ codec->height == icodec->height &&
+ frame_topBand == 0 &&
+ frame_bottomBand == 0 &&
+ frame_leftBand == 0 &&
+ frame_rightBand == 0)
+ {
+ ost->video_resample = 0;
+ ost->video_crop = 0;
+ } else if ((codec->width == icodec->width -
+ (frame_leftBand + frame_rightBand)) &&
+ (codec->height == icodec->height -
+ (frame_topBand + frame_bottomBand)))
+ {
ost->video_resample = 0;
+ ost->video_crop = 1;
+ ost->topBand = frame_topBand;
+ ost->leftBand = frame_leftBand;
} else {
- UINT8 *buf;
+ uint8_t *buf;
ost->video_resample = 1;
- buf = malloc((codec->width * codec->height * 3) / 2);
+ ost->video_crop = 0; // cropping is handled as part of resample
+ buf = av_malloc((codec->width * codec->height * 3) / 2);
if (!buf)
goto fail;
ost->pict_tmp.data[0] = buf;
ost->pict_tmp.linesize[1] = codec->width / 2;
ost->pict_tmp.linesize[2] = codec->width / 2;
- ost->img_resample_ctx = img_resample_init(
+ ost->img_resample_ctx = img_resample_full_init(
ost->st->codec.width, ost->st->codec.height,
- ist->st->codec.width, ist->st->codec.height);
+ ist->st->codec.width, ist->st->codec.height,
+ frame_topBand, frame_bottomBand,
+ frame_leftBand, frame_rightBand);
}
ost->encoding_needed = 1;
ist->decoding_needed = 1;
+ break;
+ default:
+ av_abort();
+ }
+ /* two pass mode */
+ if (ost->encoding_needed &&
+ (codec->flags & (CODEC_FLAG_PASS1 | CODEC_FLAG_PASS2))) {
+ char logfilename[1024];
+ FILE *f;
+ int size;
+ char *logbuffer;
+
+ snprintf(logfilename, sizeof(logfilename), "%s-%d.log",
+ pass_logfilename ?
+ pass_logfilename : DEFAULT_PASS_LOGFILENAME, i);
+ if (codec->flags & CODEC_FLAG_PASS1) {
+ f = fopen(logfilename, "w");
+ if (!f) {
+ perror(logfilename);
+ exit(1);
+ }
+ ost->logfile = f;
+ } else {
+ /* read the log file */
+ f = fopen(logfilename, "r");
+ if (!f) {
+ perror(logfilename);
+ exit(1);
+ }
+ fseek(f, 0, SEEK_END);
+ size = ftell(f);
+ fseek(f, 0, SEEK_SET);
+ logbuffer = av_malloc(size + 1);
+ if (!logbuffer) {
+ fprintf(stderr, "Could not allocate log buffer\n");
+ exit(1);
+ }
+ fread(logbuffer, 1, size, f);
+ fclose(f);
+ logbuffer[size] = '\0';
+ codec->stats_in = logbuffer;
+ }
}
- break;
}
}
+ /* dump the file output parameters - cannot be done before in case
+ of stream copy */
+ for(i=0;i<nb_output_files;i++) {
+ dump_format(output_files[i], i, output_files[i]->filename, 1);
+ }
+
+ /* dump the stream mapping */
+ fprintf(stderr, "Stream mapping:\n");
+ for(i=0;i<nb_ostreams;i++) {
+ ost = ost_table[i];
+ fprintf(stderr, " Stream #%d.%d -> #%d.%d\n",
+ ist_table[ost->source_index]->file_index,
+ ist_table[ost->source_index]->index,
+ ost->file_index,
+ ost->index);
+ }
+
/* open each encoder */
for(i=0;i<nb_ostreams;i++) {
ost = ost_table[i];
AVCodec *codec;
codec = avcodec_find_decoder(ist->st->codec.codec_id);
if (!codec) {
- fprintf(stderr, "Unsupported codec for input stream #%d.%d\n",
- ist->file_index, ist->index);
+ fprintf(stderr, "Unsupported codec (id=%d) for input stream #%d.%d\n",
+ ist->st->codec.codec_id, ist->file_index, ist->index);
exit(1);
}
if (avcodec_open(&ist->st->codec, codec) < 0) {
ist->file_index, ist->index);
exit(1);
}
+ //if (ist->st->codec.codec_type == CODEC_TYPE_VIDEO)
+ // ist->st->codec.flags |= CODEC_FLAG_REPEAT_FIELD;
+ ist->frame_decoded = 1;
}
}
/* init pts */
for(i=0;i<nb_istreams;i++) {
ist = ist_table[i];
- ist->pts = 0;
- ist->frame_number = 0;
}
/* compute buffer size max (should use a complete heuristic) */
/* open files and write file headers */
for(i=0;i<nb_output_files;i++) {
os = output_files[i];
- os->format->write_header(os);
+ if (av_write_header(os) < 0) {
+ fprintf(stderr, "Could not write header for output file #%d (incorrect codec paramters ?)\n", i);
+ ret = -EINVAL;
+ goto fail;
+ }
+ }
+
+#ifndef CONFIG_WIN32
+ if (!do_play) {
+ fprintf(stderr, "Press [q] to stop encoding\n");
+ } else {
+ fprintf(stderr, "Press [q] to stop playing\n");
}
+#endif
+ term_init();
+
+ stream_no_data = 0;
+ key = -1;
- start_time = gettime();
- min_pts = 0;
for(;;) {
int file_index, ist_index;
AVPacket pkt;
- UINT8 *ptr;
+ uint8_t *ptr;
int len;
- UINT8 *data_buf;
+ uint8_t *data_buf;
int data_size, got_picture;
AVPicture picture;
short samples[AVCODEC_MAX_AUDIO_FRAME_SIZE / 2];
-
- /* select the input file with the smallest pts */
+ void *buffer_to_free;
+ double pts_min;
+
redo:
+ /* if 'q' pressed, exits */
+ if (key) {
+ /* read_key() returns 0 on EOF */
+ key = read_key();
+ if (key == 'q')
+ break;
+ }
+
+ /* select the stream that we must read now by looking at the
+ smallest output pts */
file_index = -1;
- min_pts = (1ULL << 63) - 1;
- for(i=0;i<nb_istreams;i++) {
- ist = ist_table[i];
- if (!ist->discard && !file_table[ist->file_index].eof_reached && ist->pts < min_pts) {
- min_pts = ist->pts;
+ pts_min = 1e10;
+ for(i=0;i<nb_ostreams;i++) {
+ double pts;
+ ost = ost_table[i];
+ os = output_files[ost->file_index];
+ ist = ist_table[ost->source_index];
+ pts = (double)ost->st->pts.val * os->pts_num / os->pts_den;
+ if (!file_table[ist->file_index].eof_reached &&
+ pts < pts_min) {
+ pts_min = pts;
file_index = ist->file_index;
}
}
/* if none, if is finished */
- if (file_index < 0)
+ if (file_index < 0) {
break;
+ }
+
/* finish if recording time exhausted */
- if (recording_time > 0 && min_pts >= recording_time)
+ if (recording_time > 0 && pts_min >= (recording_time / 1000000.0))
break;
+
/* read a packet from it and output it in the fifo */
-
is = input_files[file_index];
if (av_read_packet(is, &pkt) < 0) {
file_table[file_index].eof_reached = 1;
continue;
}
+ if (!pkt.size) {
+ stream_no_data = is;
+ } else {
+ stream_no_data = 0;
+ }
+ if (do_hex_dump) {
+ printf("stream #%d, size=%d:\n", pkt.stream_index, pkt.size);
+ av_hex_dump(pkt.data, pkt.size);
+ }
+ /* 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)
+ goto discard_packet;
ist_index = file_table[file_index].ist_index + pkt.stream_index;
ist = ist_table[ist_index];
- if (ist->discard) {
- continue;
- }
-
-#ifdef HEX_DUMP
- printf("stream #%d, size=%d:\n", pkt.stream_index, pkt.size);
- hex_dump(pkt.data, pkt.size);
-#endif
+ if (ist->discard)
+ goto discard_packet;
- // 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);
len = pkt.size;
ptr = pkt.data;
+ pts_set = 0;
while (len > 0) {
+ int64_t ipts;
+
+ ipts = AV_NOPTS_VALUE;
/* decode the packet if needed */
data_buf = NULL; /* fail safe */
data_size = 0;
if (ist->decoding_needed) {
+ /* NOTE1: we only take into account the PTS if a new
+ frame has begun (MPEG semantics) */
+ /* NOTE2: even if the fraction is not initialized,
+ av_frac_set can be used to set the integer part */
+ if (ist->frame_decoded &&
+ pkt.pts != AV_NOPTS_VALUE &&
+ !pts_set) {
+ ipts = pkt.pts;
+ ist->frame_decoded = 0;
+ pts_set = 1;
+ }
+
switch(ist->st->codec.codec_type) {
case CODEC_TYPE_AUDIO:
- if (ist->st->codec.codec_id == CODEC_ID_PCM) {
- /* no need to call a codec */
- data_buf = ptr;
- data_size = len;
- ret = len;
- } else {
- ret = avcodec_decode_audio(&ist->st->codec, samples, &data_size,
- ptr, len);
- if (ret < 0)
- goto fail_decode;
- if (data_size == 0) {
- /* no audio frame */
- ptr += ret;
- len -= ret;
- continue;
- }
- data_buf = (UINT8 *)samples;
+ /* XXX: could avoid copy if PCM 16 bits with same
+ endianness as CPU */
+ ret = avcodec_decode_audio(&ist->st->codec, samples, &data_size,
+ ptr, len);
+ if (ret < 0)
+ goto fail_decode;
+ /* Some bug in mpeg audio decoder gives */
+ /* data_size < 0, it seems they are overflows */
+ if (data_size <= 0) {
+ /* no audio frame */
+ ptr += ret;
+ len -= ret;
+ continue;
}
+ data_buf = (uint8_t *)samples;
break;
case CODEC_TYPE_VIDEO:
if (ist->st->codec.codec_id == CODEC_ID_RAWVIDEO) {
int size;
+
size = (ist->st->codec.width * ist->st->codec.height);
-
- picture.data[0] = ptr;
- picture.data[1] = picture.data[0] + size;
- picture.data[2] = picture.data[1] + size / 4;
- picture.linesize[0] = ist->st->codec.width;
- picture.linesize[1] = ist->st->codec.width / 2;
- picture.linesize[2] = ist->st->codec.width / 2;
+ avpicture_fill(&picture, ptr,
+ ist->st->codec.pix_fmt,
+ ist->st->codec.width,
+ ist->st->codec.height);
ret = len;
} else {
+ AVFrame big_picture;
+
data_size = (ist->st->codec.width * ist->st->codec.height * 3) / 2;
ret = avcodec_decode_video(&ist->st->codec,
- &picture, &got_picture, ptr, len);
+ &big_picture, &got_picture, ptr, len);
+ picture= *(AVPicture*)&big_picture;
+ ist->st->quality= big_picture.quality;
if (ret < 0) {
fail_decode:
fprintf(stderr, "Error while decoding stream #%d.%d\n",
len -= ret;
continue;
}
+
}
break;
default:
data_size = len;
ret = len;
}
- /* update pts */
- switch(ist->st->codec.codec_type) {
- case CODEC_TYPE_AUDIO:
- ist->pts = (INT64)1000000 * ist->sample_index / ist->st->codec.sample_rate;
- ist->sample_index += data_size / (2 * ist->st->codec.channels);
- break;
- case CODEC_TYPE_VIDEO:
- ist->frame_number++;
- ist->pts = ((INT64)ist->frame_number * 1000000 * FRAME_RATE_BASE) /
- ist->st->codec.frame_rate;
- break;
- }
ptr += ret;
len -= ret;
+ buffer_to_free = 0;
+ if (ist->st->codec.codec_type == CODEC_TYPE_VIDEO) {
+ pre_process_video_frame(ist, &picture, &buffer_to_free);
+ }
+
+ ist->frame_decoded = 1;
+
+#if 0
+ /* mpeg PTS deordering : if it is a P or I frame, the PTS
+ is the one of the next displayed one */
+ /* XXX: add mpeg4 too ? */
+ if (ist->st->codec.codec_id == CODEC_ID_MPEG1VIDEO) {
+ if (ist->st->codec.pict_type != B_TYPE) {
+ int64_t tmp;
+ tmp = ist->last_ip_pts;
+ ist->last_ip_pts = ist->frac_pts.val;
+ ist->frac_pts.val = tmp;
+ }
+ }
+#endif
/* transcode raw format, encode packets and output them */
-
+
for(i=0;i<nb_ostreams;i++) {
+ int frame_size;
+
ost = ost_table[i];
if (ost->source_index == ist_index) {
os = output_files[ost->file_index];
+ if (ipts != AV_NOPTS_VALUE) {
+#if 0
+ printf("%d: got pts=%f %f\n",
+ i, pkt.pts / 90000.0,
+ (ipts - ost->st->pts.val) / 90000.0);
+#endif
+ /* set the input output pts pairs */
+ ost->sync_ipts = (double)ipts * is->pts_num /
+ is->pts_den;
+ /* XXX: take into account the various fifos,
+ in particular for audio */
+ ost->sync_opts = ost->st->pts.val;
+ //printf("ipts=%lld sync_ipts=%f sync_opts=%lld pts.val=%lld pkt.pts=%lld\n", ipts, ost->sync_ipts, ost->sync_opts, ost->st->pts.val, pkt.pts);
+ } else {
+ //printf("pts.val=%lld\n", ost->st->pts.val);
+ ost->sync_ipts = AV_NOPTS_VALUE;
+ }
+
if (ost->encoding_needed) {
switch(ost->st->codec.codec_type) {
case CODEC_TYPE_AUDIO:
do_audio_out(os, ost, ist, data_buf, data_size);
break;
case CODEC_TYPE_VIDEO:
- do_video_out(os, ost, ist, &picture);
+ /* find an audio stream for synchro */
+ {
+ int i;
+ AVOutputStream *audio_sync, *ost1;
+ audio_sync = NULL;
+ for(i=0;i<nb_ostreams;i++) {
+ ost1 = ost_table[i];
+ if (ost1->file_index == ost->file_index &&
+ ost1->st->codec.codec_type == CODEC_TYPE_AUDIO) {
+ audio_sync = ost1;
+ break;
+ }
+ }
+
+ do_video_out(os, ost, ist, &picture, &frame_size, audio_sync);
+ if (do_vstats && frame_size)
+ do_video_stats(os, ost, frame_size);
+ }
break;
+ default:
+ av_abort();
}
} else {
+ AVFrame avframe;
+
/* no reencoding needed : output the packet directly */
- os->format->write_packet(os, ost->index, data_buf, data_size);
+ /* force the input stream PTS */
+
+ //XXX/FIXME set keyframe flag from demuxer (or optionally from decoder)
+ memset(&avframe, 0, sizeof(AVFrame));
+ ost->st->codec.coded_frame= &avframe;
+
+ av_write_frame(os, ost->index, data_buf, data_size);
+ ost->st->codec.frame_number++;
+ ost->frame_number++;
}
}
}
+ av_free(buffer_to_free);
+ ipts = AV_NOPTS_VALUE;
}
+ discard_packet:
av_free_packet(&pkt);
- /* dump report by using the first video and audio streams */
- {
- char buf[1024];
- AVFormatContext *oc;
- INT64 total_size, ti;
- AVCodecContext *enc;
- int frame_number, vid;
- double bitrate, ti1;
- static INT64 last_time;
-
- if ((min_pts - last_time) >= 500000) {
- last_time = min_pts;
-
- oc = output_files[0];
-
- total_size = url_ftell(&oc->pb);
-
- buf[0] = '\0';
- ti = (1ULL << 63) - 1;
- vid = 0;
- for(i=0;i<nb_ostreams;i++) {
- ost = ost_table[i];
- enc = &ost->st->codec;
- ist = ist_table[ost->source_index];
- if (!vid && enc->codec_type == CODEC_TYPE_VIDEO) {
- frame_number = ist->frame_number;
- sprintf(buf + strlen(buf), "frame=%5d q=%2d ",
- frame_number, enc->quality);
- vid = 1;
- }
- /* compute min pts value */
- if (!ist->discard && ist->pts < ti) {
- ti = ist->pts;
- }
- }
-
- ti1 = ti / 1000000.0;
- if (ti1 < 0.1)
- ti1 = 0.1;
- bitrate = (double)(total_size * 8) / ti1 / 1000.0;
-
- sprintf(buf + strlen(buf), "size=%8LdkB time=%0.1f bitrate=%6.1fkbits/s",
- total_size / 1024, ti1, bitrate);
-
- fprintf(stderr, "%s \r", buf);
- fflush(stderr);
- }
- }
+ /* dump report by using the output first video and audio streams */
+ print_report(output_files, ost_table, nb_ostreams, 0);
}
+ term_exit();
/* dump report by using the first video and audio streams */
- {
- char buf[1024];
- AVFormatContext *oc;
- INT64 total_size, ti;
- AVCodecContext *enc;
- int frame_number, vid;
- double bitrate, ti1;
-
- oc = output_files[0];
-
- total_size = url_ftell(&oc->pb);
-
- buf[0] = '\0';
- ti = (1ULL << 63) - 1;
- vid = 0;
- for(i=0;i<nb_ostreams;i++) {
- ost = ost_table[i];
- enc = &ost->st->codec;
- ist = ist_table[ost->source_index];
- if (!vid && enc->codec_type == CODEC_TYPE_VIDEO) {
- frame_number = ist->frame_number;
- sprintf(buf + strlen(buf), "frame=%5d q=%2d ",
- frame_number, enc->quality);
- vid = 1;
- }
- /* compute min pts value */
- if (!ist->discard && ist->pts < ti) {
- ti = ist->pts;
- }
- }
-
- ti1 = ti / 1000000.0;
- if (ti1 < 0.1)
- ti1 = 0.1;
- bitrate = (double)(total_size * 8) / ti1 / 1000.0;
-
- sprintf(buf + strlen(buf), "size=%8LdkB time=%0.1f bitrate=%6.1fkbits/s",
- total_size / 1024, ti1, bitrate);
-
- fprintf(stderr, "%s \n", buf);
- }
+ print_report(output_files, ost_table, nb_ostreams, 1);
+
/* close each encoder */
for(i=0;i<nb_ostreams;i++) {
ost = ost_table[i];
if (ost->encoding_needed) {
+ av_freep(&ost->st->codec.stats_in);
avcodec_close(&ost->st->codec);
}
}
/* write the trailer if needed and close file */
for(i=0;i<nb_output_files;i++) {
os = output_files[i];
- os->format->write_trailer(os);
+ av_write_trailer(os);
}
/* finished ! */
ret = 0;
fail1:
+ av_free(file_table);
+
if (ist_table) {
for(i=0;i<nb_istreams;i++) {
ist = ist_table[i];
- if (ist) {
- free(ist);
- }
+ av_free(ist);
}
- free(ist_table);
+ av_free(ist_table);
}
if (ost_table) {
for(i=0;i<nb_ostreams;i++) {
ost = ost_table[i];
if (ost) {
- if (ost->pict_tmp.data[0])
- free(ost->pict_tmp.data[0]);
+ if (ost->logfile) {
+ fclose(ost->logfile);
+ ost->logfile = NULL;
+ }
+ fifo_free(&ost->fifo); /* works even if fifo is not
+ initialized but set to zero */
+ av_free(ost->pict_tmp.data[0]);
if (ost->video_resample)
img_resample_close(ost->img_resample_ctx);
if (ost->audio_resample)
audio_resample_close(ost->resample);
- free(ost);
+ av_free(ost);
}
}
- free(ost_table);
+ av_free(ost_table);
}
return ret;
fail:
}
#endif
-void show_licence(void)
+static void show_licence(void)
{
printf(
"ffmpeg version " FFMPEG_VERSION "\n"
- "Copyright (c) 2000,2001 Gerard Lantau\n"
- "This program is free software; you can redistribute it and/or modify\n"
- "it under the terms of the GNU General Public License as published by\n"
- "the Free Software Foundation; either version 2 of the License, or\n"
- "(at your option) any later version.\n"
+ "Copyright (c) 2000, 2001, 2002 Fabrice Bellard\n"
+ "This library is free software; you can redistribute it and/or\n"
+ "modify it under the terms of the GNU Lesser General Public\n"
+ "License as published by the Free Software Foundation; either\n"
+ "version 2 of the License, or (at your option) any later version.\n"
"\n"
- "This program is distributed in the hope that it will be useful,\n"
+ "This library is distributed in the hope that it will be useful,\n"
"but WITHOUT ANY WARRANTY; without even the implied warranty of\n"
- "MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the\n"
- "GNU General Public License for more details.\n"
+ "MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU\n"
+ "Lesser General Public License for more details.\n"
"\n"
- "You should have received a copy of the GNU General Public License\n"
- "along with this program; if not, write to the Free Software\n"
- "Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.\n"
+ "You should have received a copy of the GNU Lesser General Public\n"
+ "License along with this library; if not, write to the Free Software\n"
+ "Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA\n"
);
exit(1);
}
-void opt_format(const char *arg)
+static void opt_image_format(const char *arg)
{
- AVFormat *f;
- f = first_format;
- while (f != NULL && strcmp(f->name, arg) != 0) f = f->next;
- if (f == NULL) {
- fprintf(stderr, "Invalid format: %s\n", arg);
+ AVImageFormat *f;
+
+ for(f = first_image_format; f != NULL; f = f->next) {
+ if (!strcmp(arg, f->name))
+ break;
+ }
+ if (!f) {
+ fprintf(stderr, "Unknown image format: '%s'\n", arg);
+ exit(1);
+ }
+ image_format = f;
+}
+
+static void opt_format(const char *arg)
+{
+ /* compatibility stuff for pgmyuv */
+ if (!strcmp(arg, "pgmyuv")) {
+ opt_image_format(arg);
+ arg = "image";
+ }
+
+ file_iformat = av_find_input_format(arg);
+ file_oformat = guess_format(arg, NULL, NULL);
+ if (!file_iformat && !file_oformat) {
+ fprintf(stderr, "Unknown input or output format: %s\n", arg);
exit(1);
}
- file_format = f;
}
-void opt_video_bitrate(const char *arg)
+static void opt_video_bitrate(const char *arg)
{
video_bit_rate = atoi(arg) * 1000;
}
-void opt_frame_rate(const char *arg)
+static void opt_video_bitrate_tolerance(const char *arg)
+{
+ video_bit_rate_tolerance = atoi(arg) * 1000;
+}
+
+static void opt_video_bitrate_max(const char *arg)
+{
+ video_rc_max_rate = atoi(arg) * 1000;
+}
+
+static void opt_video_bitrate_min(const char *arg)
+{
+ video_rc_min_rate = atoi(arg) * 1000;
+}
+
+static void opt_video_buffer_size(const char *arg)
+{
+ video_rc_buffer_size = atoi(arg) * 1000;
+}
+
+static void opt_video_rc_eq(char *arg)
+{
+ video_rc_eq = arg;
+}
+
+static void opt_video_rc_override_string(char *arg)
+{
+ video_rc_override_string = arg;
+}
+
+
+static void opt_workaround_bugs(const char *arg)
+{
+ workaround_bugs = atoi(arg);
+}
+
+static void opt_dct_algo(const char *arg)
+{
+ dct_algo = atoi(arg);
+}
+
+static void opt_idct_algo(const char *arg)
+{
+ idct_algo = atoi(arg);
+}
+
+
+static void opt_error_resilience(const char *arg)
+{
+ error_resilience = atoi(arg);
+}
+
+static void opt_error_concealment(const char *arg)
+{
+ error_concealment = atoi(arg);
+}
+
+static void opt_debug(const char *arg)
+{
+ debug = atoi(arg);
+}
+
+static void opt_frame_rate(const char *arg)
{
frame_rate = (int)(strtod(arg, 0) * FRAME_RATE_BASE);
}
-void opt_frame_size(const char *arg)
+
+static void opt_frame_crop_top(const char *arg)
+{
+ frame_topBand = atoi(arg);
+ if (frame_topBand < 0) {
+ fprintf(stderr, "Incorrect top crop size\n");
+ exit(1);
+ }
+ if ((frame_topBand % 2) != 0) {
+ fprintf(stderr, "Top crop size must be a multiple of 2\n");
+ exit(1);
+ }
+ if ((frame_topBand) >= frame_height){
+ fprintf(stderr, "Vertical crop dimensions are outside the range of the original image.\nRemember to crop first and scale second.\n");
+ exit(1);
+ }
+ frame_height -= frame_topBand;
+}
+
+static void opt_frame_crop_bottom(const char *arg)
+{
+ frame_bottomBand = atoi(arg);
+ if (frame_bottomBand < 0) {
+ fprintf(stderr, "Incorrect bottom crop size\n");
+ exit(1);
+ }
+ if ((frame_bottomBand % 2) != 0) {
+ fprintf(stderr, "Bottom crop size must be a multiple of 2\n");
+ exit(1);
+ }
+ if ((frame_bottomBand) >= frame_height){
+ fprintf(stderr, "Vertical crop dimensions are outside the range of the original image.\nRemember to crop first and scale second.\n");
+ exit(1);
+ }
+ frame_height -= frame_bottomBand;
+}
+
+static void opt_frame_crop_left(const char *arg)
+{
+ frame_leftBand = atoi(arg);
+ if (frame_leftBand < 0) {
+ fprintf(stderr, "Incorrect left crop size\n");
+ exit(1);
+ }
+ if ((frame_leftBand % 2) != 0) {
+ fprintf(stderr, "Left crop size must be a multiple of 2\n");
+ exit(1);
+ }
+ if ((frame_leftBand) >= frame_width){
+ fprintf(stderr, "Horizontal crop dimensions are outside the range of the original image.\nRemember to crop first and scale second.\n");
+ exit(1);
+ }
+ frame_width -= frame_leftBand;
+}
+
+static void opt_frame_crop_right(const char *arg)
+{
+ frame_rightBand = atoi(arg);
+ if (frame_rightBand < 0) {
+ fprintf(stderr, "Incorrect right crop size\n");
+ exit(1);
+ }
+ if ((frame_rightBand % 2) != 0) {
+ fprintf(stderr, "Right crop size must be a multiple of 2\n");
+ exit(1);
+ }
+ if ((frame_rightBand) >= frame_width){
+ fprintf(stderr, "Horizontal crop dimensions are outside the range of the original image.\nRemember to crop first and scale second.\n");
+ exit(1);
+ }
+ frame_width -= frame_rightBand;
+}
+
+static void opt_frame_size(const char *arg)
{
parse_image_size(&frame_width, &frame_height, arg);
if (frame_width <= 0 || frame_height <= 0) {
}
}
-void opt_gop_size(const char *arg)
+static void opt_gop_size(const char *arg)
+{
+ gop_size = atoi(arg);
+}
+
+static void opt_b_frames(const char *arg)
+{
+ b_frames = atoi(arg);
+ if (b_frames > FF_MAX_B_FRAMES) {
+ fprintf(stderr, "\nCannot have more than %d B frames, increase FF_MAX_B_FRAMES.\n", FF_MAX_B_FRAMES);
+ exit(1);
+ } else if (b_frames < 1) {
+ fprintf(stderr, "\nNumber of B frames must be higher than 0\n");
+ exit(1);
+ }
+}
+
+static void opt_qscale(const char *arg)
+{
+ video_qscale = atoi(arg);
+ if (video_qscale < 0 ||
+ video_qscale > 31) {
+ fprintf(stderr, "qscale must be >= 1 and <= 31\n");
+ exit(1);
+ }
+}
+
+static void opt_qmin(const char *arg)
+{
+ video_qmin = atoi(arg);
+ if (video_qmin < 0 ||
+ video_qmin > 31) {
+ fprintf(stderr, "qmin must be >= 1 and <= 31\n");
+ exit(1);
+ }
+}
+
+static void opt_qmax(const char *arg)
+{
+ video_qmax = atoi(arg);
+ if (video_qmax < 0 ||
+ video_qmax > 31) {
+ fprintf(stderr, "qmax must be >= 1 and <= 31\n");
+ exit(1);
+ }
+}
+
+static void opt_mb_qmin(const char *arg)
+{
+ video_mb_qmin = atoi(arg);
+ if (video_mb_qmin < 0 ||
+ video_mb_qmin > 31) {
+ fprintf(stderr, "qmin must be >= 1 and <= 31\n");
+ exit(1);
+ }
+}
+
+static void opt_mb_qmax(const char *arg)
+{
+ video_mb_qmax = atoi(arg);
+ if (video_mb_qmax < 0 ||
+ video_mb_qmax > 31) {
+ fprintf(stderr, "qmax must be >= 1 and <= 31\n");
+ exit(1);
+ }
+}
+
+static void opt_qdiff(const char *arg)
+{
+ video_qdiff = atoi(arg);
+ if (video_qdiff < 0 ||
+ video_qdiff > 31) {
+ fprintf(stderr, "qdiff must be >= 1 and <= 31\n");
+ exit(1);
+ }
+}
+
+static void opt_qblur(const char *arg)
+{
+ video_qblur = atof(arg);
+}
+
+static void opt_qcomp(const char *arg)
+{
+ video_qcomp = atof(arg);
+}
+
+static void opt_rc_initial_cplx(const char *arg)
+{
+ video_rc_initial_cplx = atof(arg);
+}
+static void opt_b_qfactor(const char *arg)
+{
+ video_b_qfactor = atof(arg);
+}
+static void opt_i_qfactor(const char *arg)
+{
+ video_i_qfactor = atof(arg);
+}
+static void opt_b_qoffset(const char *arg)
+{
+ video_b_qoffset = atof(arg);
+}
+static void opt_i_qoffset(const char *arg)
+{
+ video_i_qoffset = atof(arg);
+}
+
+static void opt_packet_size(const char *arg)
{
- gop_size = atoi(arg);
+ packet_size= atoi(arg);
}
-void opt_qscale(const char *arg)
+static void opt_strict(const char *arg)
{
- video_qscale = atoi(arg);
- if (video_qscale < 0 ||
- video_qscale > 31) {
- fprintf(stderr, "qscale must be >= 1 and <= 31\n");
- exit(1);
- }
+ strict= atoi(arg);
}
-
-void opt_audio_bitrate(const char *arg)
+static void opt_audio_bitrate(const char *arg)
{
audio_bit_rate = atoi(arg) * 1000;
}
-void opt_audio_rate(const char *arg)
+static void opt_audio_rate(const char *arg)
{
audio_sample_rate = atoi(arg);
}
-void opt_audio_channels(const char *arg)
+static void opt_audio_channels(const char *arg)
{
audio_channels = atoi(arg);
}
-void opt_video_device(const char *arg)
+static void opt_video_device(const char *arg)
{
- v4l_device = strdup(arg);
+ video_device = av_strdup(arg);
}
-void opt_audio_device(const char *arg)
+static void opt_video_channel(const char *arg)
{
- audio_device = strdup(arg);
+ video_channel = strtol(arg, NULL, 0);
}
-void opt_audio_codec(const char *arg)
+static void opt_audio_device(const char *arg)
+{
+ audio_device = av_strdup(arg);
+}
+
+static void opt_dv1394(const char *arg)
+{
+ video_grab_format = "dv1394";
+ audio_grab_format = NULL;
+}
+
+static void opt_audio_codec(const char *arg)
{
AVCodec *p;
- p = first_avcodec;
- while (p) {
- if (!strcmp(p->name, arg) && p->type == CODEC_TYPE_AUDIO)
- break;
- p = p->next;
+ if (!strcmp(arg, "copy")) {
+ audio_stream_copy = 1;
+ } else {
+ p = first_avcodec;
+ while (p) {
+ if (!strcmp(p->name, arg) && p->type == CODEC_TYPE_AUDIO)
+ break;
+ p = p->next;
+ }
+ if (p == NULL) {
+ fprintf(stderr, "Unknown audio codec '%s'\n", arg);
+ exit(1);
+ } else {
+ audio_codec_id = p->id;
+ }
+ }
+}
+
+static void add_frame_hooker(const char *arg)
+{
+ int argc = 0;
+ char *argv[64];
+ int i;
+ char *args = av_strdup(arg);
+
+ argv[0] = strtok(args, " ");
+ while (argc < 62 && (argv[++argc] = strtok(NULL, " "))) {
}
- if (p == NULL) {
- fprintf(stderr, "Unknown audio codec '%s'\n", arg);
+
+ i = frame_hook_add(argc, argv);
+
+ if (i != 0) {
+ fprintf(stderr, "Failed to add video hook function: %s\n", arg);
exit(1);
- } else {
- audio_codec_id = p->id;
}
}
"full",
"log",
"phods",
+ "epzs",
+ "x1",
NULL,
};
-void opt_motion_estimation(const char *arg)
+static void opt_motion_estimation(const char *arg)
{
const char **p;
p = motion_str;
break;
p++;
}
- motion_estimation_method = p - motion_str;
+ me_method = (p - motion_str) + 1;
}
-void opt_video_codec(const char *arg)
+static void opt_video_codec(const char *arg)
{
AVCodec *p;
- p = first_avcodec;
- while (p) {
- if (!strcmp(p->name, arg) && p->type == CODEC_TYPE_VIDEO)
- break;
- p = p->next;
- }
- if (p == NULL) {
- fprintf(stderr, "Unknown video codec '%s'\n", arg);
- exit(1);
+ if (!strcmp(arg, "copy")) {
+ video_stream_copy = 1;
} else {
- video_codec_id = p->id;
+ p = first_avcodec;
+ while (p) {
+ if (!strcmp(p->name, arg) && p->type == CODEC_TYPE_VIDEO)
+ break;
+ p = p->next;
+ }
+ if (p == NULL) {
+ fprintf(stderr, "Unknown video codec '%s'\n", arg);
+ exit(1);
+ } else {
+ video_codec_id = p->id;
+ }
}
}
-void opt_map(const char *arg)
+static void opt_map(const char *arg)
{
AVStreamMap *m;
const char *p;
m->file_index = strtol(arg, (char **)&p, 0);
if (*p)
p++;
- m->stream_index = strtol(arg, (char **)&p, 0);
+
+ m->stream_index = strtol(p, (char **)&p, 0);
}
-void opt_recording_time(const char *arg)
+static void opt_recording_time(const char *arg)
{
recording_time = parse_date(arg, 1);
}
-/* return the number of packet read to find the codec parameters */
-int find_codec_parameters(AVFormatContext *ic)
+static void print_error(const char *filename, int err)
{
- int val, i, count, ret, got_picture, size;
- AVCodec *codec;
- AVCodecContext *enc;
- AVStream *st;
- AVPacket *pkt;
- AVPicture picture;
- AVPacketList *pktl, **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;
- }
- avcodec_open(&st->codec, codec);
- }
- }
- 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);
- 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);
- }
+ switch(err) {
+ case AVERROR_NUMEXPECTED:
+ fprintf(stderr, "%s: Incorrect image filename syntax.\n"
+ "Use '%%d' to specify the image number:\n"
+ " for img1.jpg, img2.jpg, ..., use 'img%%d.jpg';\n"
+ " for img001.jpg, img002.jpg, ..., use 'img%%03d.jpg'.\n",
+ filename);
+ break;
+ case AVERROR_INVALIDDATA:
+ 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;
}
- return ret;
}
-
-void opt_input_file(const char *filename)
+static void opt_input_file(const char *filename)
{
AVFormatContext *ic;
AVFormatParameters params, *ap = ¶ms;
- URLFormat url_format;
- AVFormat *fmt;
- int err, i, ret;
-
- 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));
- }
+ int err, i, ret, rfps;
- if (!fmt || !fmt->read_header) {
- fprintf(stderr, "%s: Unknown file format\n", filename);
- exit(1);
- }
- ic->format = fmt;
+ if (!strcmp(filename, "-"))
+ filename = "pipe:";
/* get default parameters from command line */
- if (!ap->sample_rate)
- ap->sample_rate = audio_sample_rate;
- if (!ap->channels)
- ap->channels = audio_channels;
-
- if (!ap->frame_rate)
- ap->frame_rate = frame_rate;
- if (!ap->width)
- ap->width = frame_width;
- if (!ap->height)
- ap->height = frame_height;
-
- err = ic->format->read_header(ic, ap);
+ memset(ap, 0, sizeof(*ap));
+ ap->sample_rate = audio_sample_rate;
+ ap->channels = audio_channels;
+ ap->frame_rate = frame_rate;
+ ap->width = frame_width;
+ ap->height = frame_height;
+ ap->image_format = image_format;
+
+ /* open the input file with generic libav function */
+ err = av_open_input_file(&ic, filename, file_iformat, 0, ap);
if (err < 0) {
- fprintf(stderr, "%s: Error while parsing header\n", filename);
+ print_error(filename, err);
exit(1);
}
- /* If not enough info for the codecs, we decode the first frames
- to get it. (used in mpeg case for example) */
- ret = find_codec_parameters(ic);
+ /* If not enough info to get the stream parameters, we decode the
+ first frames to get it. (used in mpeg case for example) */
+ ret = av_find_stream_info(ic);
if (ret < 0) {
fprintf(stderr, "%s: could not find codec parameters\n", filename);
exit(1);
AVCodecContext *enc = &ic->streams[i]->codec;
switch(enc->codec_type) {
case CODEC_TYPE_AUDIO:
+ //fprintf(stderr, "\nInput Audio channels: %d", enc->channels);
audio_channels = enc->channels;
audio_sample_rate = enc->sample_rate;
break;
case CODEC_TYPE_VIDEO:
frame_height = enc->height;
frame_width = enc->width;
- frame_rate = enc->frame_rate;
+ rfps = ic->streams[i]->r_frame_rate;
+ enc->workaround_bugs = workaround_bugs;
+ enc->error_resilience = error_resilience;
+ enc->error_concealment = error_concealment;
+ enc->idct_algo= idct_algo;
+ enc->debug= debug;
+/* if(enc->codec->capabilities & CODEC_CAP_TRUNCATED)
+ enc->flags|= CODEC_FLAG_TRUNCATED; */
+ if(/*enc->codec_id==CODEC_ID_MPEG4 || */enc->codec_id==CODEC_ID_MPEG1VIDEO)
+ enc->flags|= CODEC_FLAG_TRUNCATED;
+
+ if (enc->frame_rate != rfps) {
+ fprintf(stderr,"\nSeems that stream %d comes from film source: %2.2f->%2.2f\n",
+ i, (float)enc->frame_rate / FRAME_RATE_BASE,
+ (float)rfps / FRAME_RATE_BASE);
+ }
+ /* update the current frame rate to match the stream frame rate */
+ frame_rate = rfps;
break;
+ default:
+ av_abort();
}
}
/* dump the file content */
dump_format(ic, nb_input_files, filename, 0);
nb_input_files++;
- file_format = NULL;
+ file_iformat = NULL;
+ file_oformat = NULL;
+ image_format = NULL;
+}
+
+static void check_audio_video_inputs(int *has_video_ptr, int *has_audio_ptr)
+{
+ int has_video, has_audio, i, j;
+ AVFormatContext *ic;
+
+ has_video = 0;
+ has_audio = 0;
+ for(j=0;j<nb_input_files;j++) {
+ ic = input_files[j];
+ for(i=0;i<ic->nb_streams;i++) {
+ AVCodecContext *enc = &ic->streams[i]->codec;
+ switch(enc->codec_type) {
+ case CODEC_TYPE_AUDIO:
+ has_audio = 1;
+ break;
+ case CODEC_TYPE_VIDEO:
+ has_video = 1;
+ break;
+ default:
+ av_abort();
+ }
+ }
+ }
+ *has_video_ptr = has_video;
+ *has_audio_ptr = has_audio;
}
-void opt_output_file(const char *filename)
+static void opt_output_file(const char *filename)
{
AVStream *st;
AVFormatContext *oc;
- int use_video, use_audio, nb_streams;
+ int use_video, use_audio, nb_streams, input_has_video, input_has_audio;
int codec_id;
+ AVFormatParameters params, *ap = ¶ms;
if (!strcmp(filename, "-"))
filename = "pipe:";
oc = av_mallocz(sizeof(AVFormatContext));
- if (!file_format) {
- file_format = guess_format(NULL, filename, NULL);
- if (!file_format)
- file_format = &mpeg_mux_format;
+ if (!file_oformat) {
+ file_oformat = guess_format(NULL, filename, NULL);
+ if (!file_oformat) {
+ 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)) {
/* special case for files sent to ffserver: we get the stream
parameters from ffserver */
exit(1);
}
} else {
- use_video = file_format->video_codec != CODEC_ID_NONE;
- use_audio = file_format->audio_codec != CODEC_ID_NONE;
-
+ use_video = file_oformat->video_codec != CODEC_ID_NONE;
+ use_audio = file_oformat->audio_codec != CODEC_ID_NONE;
+
+ /* disable if no corresponding type found and at least one
+ input file */
+ if (nb_input_files > 0) {
+ check_audio_video_inputs(&input_has_video, &input_has_audio);
+ if (!input_has_video)
+ use_video = 0;
+ if (!input_has_audio)
+ use_audio = 0;
+ }
+
+ /* manual disable */
if (audio_disable) {
use_audio = 0;
}
fprintf(stderr, "Could not alloc stream\n");
exit(1);
}
- video_enc = &st->codec;
+ avcodec_get_context_defaults(&st->codec);
- codec_id = file_format->video_codec;
- if (video_codec_id != CODEC_ID_NONE)
- codec_id = video_codec_id;
-
- video_enc->codec_id = codec_id;
- video_enc->codec_type = CODEC_TYPE_VIDEO;
+ video_enc = &st->codec;
+ if (video_stream_copy) {
+ st->stream_copy = 1;
+ video_enc->codec_type = CODEC_TYPE_VIDEO;
+ } else {
+ char *p;
+ int i;
+
+ codec_id = file_oformat->video_codec;
+ if (video_codec_id != CODEC_ID_NONE)
+ codec_id = video_codec_id;
+
+ video_enc->codec_id = codec_id;
+
+ video_enc->bit_rate = video_bit_rate;
+ video_enc->bit_rate_tolerance = video_bit_rate_tolerance;
+ video_enc->frame_rate = frame_rate;
+
+ video_enc->width = frame_width;
+ video_enc->height = frame_height;
+
+ if (!intra_only)
+ video_enc->gop_size = gop_size;
+ else
+ video_enc->gop_size = 0;
+ if (video_qscale || same_quality) {
+ video_enc->flags |= CODEC_FLAG_QSCALE;
+ st->quality = video_qscale;
+ }
- video_enc->bit_rate = video_bit_rate;
- video_enc->frame_rate = frame_rate;
+ if (use_hq) {
+ video_enc->flags |= CODEC_FLAG_HQ;
+ }
- video_enc->width = frame_width;
- video_enc->height = frame_height;
- if (!intra_only)
- video_enc->gop_size = gop_size;
- else
- video_enc->gop_size = 0;
- if (video_qscale || same_quality) {
- video_enc->flags |= CODEC_FLAG_QSCALE;
- video_enc->quality = video_qscale;
- }
+ if (use_4mv) {
+ video_enc->flags |= CODEC_FLAG_HQ;
+ video_enc->flags |= CODEC_FLAG_4MV;
+ }
+
+ if(use_part)
+ video_enc->flags |= CODEC_FLAG_PART;
+
+
+ if (b_frames) {
+ video_enc->max_b_frames = b_frames;
+ video_enc->b_frame_strategy = 0;
+ video_enc->b_quant_factor = 2.0;
+ }
+
+ video_enc->qmin = video_qmin;
+ video_enc->qmax = video_qmax;
+ video_enc->mb_qmin = video_mb_qmin;
+ video_enc->mb_qmax = video_mb_qmax;
+ video_enc->max_qdiff = video_qdiff;
+ video_enc->qblur = video_qblur;
+ video_enc->qcompress = video_qcomp;
+ video_enc->rc_eq = video_rc_eq;
+ video_enc->debug= debug;
+
+ p= video_rc_override_string;
+ for(i=0; p; i++){
+ int start, end, q;
+ int e=sscanf(p, "%d,%d,%d", &start, &end, &q);
+ if(e!=3){
+ fprintf(stderr, "error parsing rc_override\n");
+ exit(1);
+ }
+ video_enc->rc_override=
+ av_realloc(video_enc->rc_override,
+ sizeof(RcOverride)*(i+1));
+ video_enc->rc_override[i].start_frame= start;
+ video_enc->rc_override[i].end_frame = end;
+ if(q>0){
+ video_enc->rc_override[i].qscale= q;
+ video_enc->rc_override[i].quality_factor= 1.0;
+ }
+ else{
+ video_enc->rc_override[i].qscale= 0;
+ video_enc->rc_override[i].quality_factor= -q/100.0;
+ }
+ p= strchr(p, '/');
+ if(p) p++;
+ }
+ video_enc->rc_override_count=i;
+
+ video_enc->rc_max_rate = video_rc_max_rate;
+ video_enc->rc_min_rate = video_rc_min_rate;
+ video_enc->rc_buffer_size = video_rc_buffer_size;
+ video_enc->rc_buffer_aggressivity= video_rc_buffer_aggressivity;
+ video_enc->rc_initial_cplx= video_rc_initial_cplx;
+ video_enc->i_quant_factor = video_i_qfactor;
+ video_enc->b_quant_factor = video_b_qfactor;
+ video_enc->i_quant_offset = video_i_qoffset;
+ video_enc->b_quant_offset = video_b_qoffset;
+ video_enc->dct_algo = dct_algo;
+ video_enc->idct_algo = idct_algo;
+ video_enc->strict_std_compliance = strict;
+ if(packet_size){
+ video_enc->rtp_mode= 1;
+ video_enc->rtp_payload_size= packet_size;
+ }
+
+ if (do_psnr)
+ video_enc->flags|= CODEC_FLAG_PSNR;
+
+ video_enc->me_method = me_method;
+ /* two pass mode */
+ if (do_pass) {
+ if (do_pass == 1) {
+ video_enc->flags |= CODEC_FLAG_PASS1;
+ } else {
+ video_enc->flags |= CODEC_FLAG_PASS2;
+ }
+ }
+ }
oc->streams[nb_streams] = st;
nb_streams++;
}
fprintf(stderr, "Could not alloc stream\n");
exit(1);
}
+ avcodec_get_context_defaults(&st->codec);
+
audio_enc = &st->codec;
- codec_id = file_format->audio_codec;
- if (audio_codec_id != CODEC_ID_NONE)
- codec_id = audio_codec_id;
- audio_enc->codec_id = codec_id;
audio_enc->codec_type = CODEC_TYPE_AUDIO;
-
- audio_enc->bit_rate = audio_bit_rate;
- audio_enc->sample_rate = audio_sample_rate;
- audio_enc->channels = audio_channels;
+ if (audio_stream_copy) {
+ st->stream_copy = 1;
+ } else {
+ codec_id = file_oformat->audio_codec;
+ if (audio_codec_id != CODEC_ID_NONE)
+ codec_id = audio_codec_id;
+ audio_enc->codec_id = codec_id;
+
+ audio_enc->bit_rate = audio_bit_rate;
+ audio_enc->sample_rate = audio_sample_rate;
+ /* For audio codecs other than AC3 we limit */
+ /* the number of coded channels to stereo */
+ if (audio_channels > 2 && codec_id != CODEC_ID_AC3) {
+ audio_enc->channels = 2;
+ } else
+ audio_enc->channels = audio_channels;
+ }
oc->streams[nb_streams] = st;
nb_streams++;
}
oc->nb_streams = nb_streams;
if (!nb_streams) {
- fprintf(stderr, "No audio or video selected\n");
+ fprintf(stderr, "No audio or video streams available\n");
exit(1);
}
if (str_title)
- nstrcpy(oc->title, sizeof(oc->title), str_title);
+ pstrcpy(oc->title, sizeof(oc->title), str_title);
if (str_author)
- nstrcpy(oc->author, sizeof(oc->author), str_author);
+ pstrcpy(oc->author, sizeof(oc->author), str_author);
if (str_copyright)
- nstrcpy(oc->copyright, sizeof(oc->copyright), str_copyright);
+ pstrcpy(oc->copyright, sizeof(oc->copyright), str_copyright);
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;
- /* dump the file content */
- dump_format(oc, nb_output_files, filename, 1);
- nb_output_files++;
+ output_files[nb_output_files++] = oc;
strcpy(oc->filename, filename);
- if (!(oc->format->flags & AVFMT_NOFILE)) {
+
+ /* check filename in case of an image number is expected */
+ if (oc->oformat->flags & AVFMT_NEEDNUMBER) {
+ if (filename_number_test(oc->filename) < 0) {
+ print_error(oc->filename, AVERROR_NUMEXPECTED);
+ exit(1);
+ }
+ }
+
+ if (!(oc->oformat->flags & AVFMT_NOFILE)) {
/* test if it already exists to avoid loosing precious files */
if (!file_overwrite &&
(strchr(filename, ':') == NULL ||
}
}
+ memset(ap, 0, sizeof(*ap));
+ ap->image_format = image_format;
+ if (av_set_parameters(oc, ap) < 0) {
+ fprintf(stderr, "%s: Invalid encoding parameters\n",
+ oc->filename);
+ exit(1);
+ }
+
/* reset some options */
- file_format = NULL;
+ file_oformat = NULL;
+ file_iformat = NULL;
+ image_format = NULL;
audio_disable = 0;
video_disable = 0;
audio_codec_id = CODEC_ID_NONE;
video_codec_id = CODEC_ID_NONE;
+ audio_stream_copy = 0;
+ video_stream_copy = 0;
+}
+
+/* prepare dummy protocols for grab */
+static void prepare_grab(void)
+{
+ int has_video, has_audio, i, j;
+ AVFormatContext *oc;
+ AVFormatContext *ic;
+ AVFormatParameters vp1, *vp = &vp1;
+ AVFormatParameters ap1, *ap = &ap1;
+
+ /* see if audio/video inputs are needed */
+ has_video = 0;
+ has_audio = 0;
+ memset(ap, 0, sizeof(*ap));
+ memset(vp, 0, sizeof(*vp));
+ for(j=0;j<nb_output_files;j++) {
+ oc = output_files[j];
+ for(i=0;i<oc->nb_streams;i++) {
+ AVCodecContext *enc = &oc->streams[i]->codec;
+ switch(enc->codec_type) {
+ case CODEC_TYPE_AUDIO:
+ if (enc->sample_rate > ap->sample_rate)
+ ap->sample_rate = enc->sample_rate;
+ if (enc->channels > ap->channels)
+ ap->channels = enc->channels;
+ has_audio = 1;
+ break;
+ case CODEC_TYPE_VIDEO:
+ if (enc->width > vp->width)
+ vp->width = enc->width;
+ if (enc->height > vp->height)
+ vp->height = enc->height;
+ if (enc->frame_rate > vp->frame_rate)
+ vp->frame_rate = enc->frame_rate;
+ has_video = 1;
+ break;
+ default:
+ av_abort();
+ }
+ }
+ }
+
+ if (has_video == 0 && has_audio == 0) {
+ fprintf(stderr, "Output file must have at least one audio or video stream\n");
+ exit(1);
+ }
+
+ if (has_video) {
+ AVInputFormat *fmt1;
+ fmt1 = av_find_input_format(video_grab_format);
+ vp->device = video_device;
+ vp->channel = video_channel;
+ if (av_open_input_file(&ic, "", fmt1, 0, vp) < 0) {
+ fprintf(stderr, "Could not find video grab device\n");
+ exit(1);
+ }
+ /* by now video grab has one stream */
+ ic->streams[0]->r_frame_rate = vp->frame_rate;
+ input_files[nb_input_files] = ic;
+ dump_format(ic, nb_input_files, "", 0);
+ nb_input_files++;
+ }
+ if (has_audio && audio_grab_format) {
+ AVInputFormat *fmt1;
+ fmt1 = av_find_input_format(audio_grab_format);
+ ap->device = audio_device;
+ if (av_open_input_file(&ic, "", fmt1, 0, ap) < 0) {
+ fprintf(stderr, "Could not find audio grab device\n");
+ exit(1);
+ }
+ input_files[nb_input_files] = ic;
+ dump_format(ic, nb_input_files, "", 0);
+ nb_input_files++;
+ }
+}
+
+/* open the necessary output devices for playing */
+static void prepare_play(void)
+{
+ int has_video, has_audio;
+
+ check_audio_video_inputs(&has_video, &has_audio);
+
+ /* manual disable */
+ if (audio_disable) {
+ has_audio = 0;
+ }
+ if (video_disable) {
+ has_video = 0;
+ }
+
+ if (has_audio) {
+ file_oformat = guess_format("audio_device", NULL, NULL);
+ if (!file_oformat) {
+ fprintf(stderr, "Could not find audio device\n");
+ exit(1);
+ }
+ opt_output_file(audio_device?audio_device:"/dev/dsp");
+ }
+
+ if (has_video) {
+ file_oformat = guess_format("framebuffer_device", NULL, NULL);
+ if (!file_oformat) {
+ fprintf(stderr, "Could not find framebuffer device\n");
+ exit(1);
+ }
+ opt_output_file("");
+ }
+}
+
+/* same option as mencoder */
+static void opt_pass(const char *pass_str)
+{
+ int pass;
+ pass = atoi(pass_str);
+ if (pass != 1 && pass != 2) {
+ fprintf(stderr, "pass number can be only 1 or 2\n");
+ exit(1);
+ }
+ do_pass = pass;
+}
+
+#if defined(CONFIG_WIN32) || defined(CONFIG_OS2)
+static int64_t getutime(void)
+{
+ return av_gettime();
+}
+#else
+static int64_t getutime(void)
+{
+ struct rusage rusage;
+
+ getrusage(RUSAGE_SELF, &rusage);
+ return (rusage.ru_utime.tv_sec * 1000000LL) + rusage.ru_utime.tv_usec;
+}
+#endif
+
+extern int ffm_nopts;
+
+static 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)
+static void show_formats(void)
{
- AVFormat *f;
+ AVInputFormat *ifmt;
+ AVOutputFormat *ofmt;
+ AVImageFormat *image_fmt;
URLProtocol *up;
AVCodec *p;
const char **pp;
- printf("File formats:\n");
- printf(" Encoding:");
- for(f = first_format; f != NULL; f = f->next) {
- if (f->write_header)
- printf(" %s", f->name);
+ printf("Output audio/video file formats:");
+ for(ofmt = first_oformat; ofmt != NULL; ofmt = ofmt->next) {
+ printf(" %s", ofmt->name);
+ }
+ printf("\n");
+
+ printf("Input audio/video file formats:");
+ for(ifmt = first_iformat; ifmt != NULL; ifmt = ifmt->next) {
+ printf(" %s", ifmt->name);
}
printf("\n");
- printf(" Decoding:");
- for(f = first_format; f != NULL; f = f->next) {
- if (f->read_header)
- printf(" %s", f->name);
+
+ printf("Output image formats:");
+ for(image_fmt = first_image_format; image_fmt != NULL;
+ image_fmt = image_fmt->next) {
+ if (image_fmt->img_write)
+ printf(" %s", image_fmt->name);
+ }
+ printf("\n");
+
+ printf("Input image formats:");
+ for(image_fmt = first_image_format; image_fmt != NULL;
+ image_fmt = image_fmt->next) {
+ if (image_fmt->img_read)
+ printf(" %s", image_fmt->name);
}
printf("\n");
pp = motion_str;
while (*pp) {
printf(" %s", *pp);
- if ((pp - motion_str) == ME_ZERO)
+ if ((pp - motion_str + 1) == ME_ZERO)
printf("(fastest)");
- else if ((pp - motion_str) == ME_FULL)
+ else if ((pp - motion_str + 1) == ME_FULL)
printf("(slowest)");
- else if ((pp - motion_str) == ME_LOG)
+ else if ((pp - motion_str + 1) == ME_EPZS)
printf("(default)");
pp++;
}
void show_help(void)
{
+ const char *prog;
const OptionDef *po;
int i, expert;
+
+ prog = do_play ? "ffplay" : "ffmpeg";
- printf("ffmpeg version " FFMPEG_VERSION ", Copyright (c) 2000,2001 Gerard Lantau\n"
- "usage: ffmpeg [[options] -i input_file]... {[options] outfile}...\n"
- "Hyper fast MPEG1/MPEG4/H263/RV and AC3/MPEG audio encoder\n"
- "\n"
+ printf("%s version " FFMPEG_VERSION ", Copyright (c) 2000, 2001, 2002 Fabrice Bellard\n",
+ prog);
+
+ if (!do_play) {
+ printf("usage: ffmpeg [[options] -i input_file]... {[options] outfile}...\n"
+ "Hyper fast MPEG1/MPEG4/H263/RV and AC3/MPEG audio encoder\n");
+ } else {
+ printf("usage: ffplay [options] input_file...\n"
+ "Simple audio player\n");
+ }
+
+ printf("\n"
"Main options are:\n");
for(i=0;i<2;i++) {
if (i == 1)
}
const OptionDef options[] = {
- { "L", 0, {show_licence}, "show license" },
- { "h", 0, {show_help}, "show help" },
- { "formats", 0, {show_formats}, "show available formats, codecs, protocols, ..." },
- { "f", HAS_ARG, {opt_format}, "force format", "fmt" },
- { "i", HAS_ARG, {opt_input_file}, "input file name", "filename" },
- { "y", OPT_BOOL, {int_arg:&file_overwrite}, "overwrite output files" },
- { "map", HAS_ARG | OPT_EXPERT, {opt_map}, "set input stream mapping", "file:stream" },
- { "t", HAS_ARG, {opt_recording_time}, "set the recording time", "duration" },
- { "title", HAS_ARG | OPT_STRING, {str_arg: &str_title}, "set the title", "string" },
- { "author", HAS_ARG | OPT_STRING, {str_arg: &str_author}, "set the author", "string" },
- { "copyright", HAS_ARG | OPT_STRING, {str_arg: &str_copyright}, "set the copyright", "string" },
- { "comment", HAS_ARG | OPT_STRING, {str_arg: &str_comment}, "set the comment", "string" },
+ { "L", 0, {(void*)show_licence}, "show license" },
+ { "h", 0, {(void*)show_help}, "show help" },
+ { "formats", 0, {(void*)show_formats}, "show available formats, codecs, protocols, ..." },
+ { "f", HAS_ARG, {(void*)opt_format}, "force format", "fmt" },
+ { "img", HAS_ARG, {(void*)opt_image_format}, "force image format", "img_fmt" },
+ { "i", HAS_ARG, {(void*)opt_input_file}, "input file name", "filename" },
+ { "y", OPT_BOOL, {(void*)&file_overwrite}, "overwrite output files" },
+ { "map", HAS_ARG | OPT_EXPERT, {(void*)opt_map}, "set input stream mapping", "file:stream" },
+ { "t", HAS_ARG, {(void*)opt_recording_time}, "set the recording time", "duration" },
+ { "title", HAS_ARG | OPT_STRING, {(void*)&str_title}, "set the title", "string" },
+ { "author", HAS_ARG | OPT_STRING, {(void*)&str_author}, "set the author", "string" },
+ { "copyright", HAS_ARG | OPT_STRING, {(void*)&str_copyright}, "set the copyright", "string" },
+ { "comment", HAS_ARG | OPT_STRING, {(void*)&str_comment}, "set the comment", "string" },
+ { "pass", HAS_ARG, {(void*)&opt_pass}, "select the pass number (1 or 2)", "n" },
+ { "passlogfile", HAS_ARG | OPT_STRING, {(void*)&pass_logfilename}, "select two pass log file name", "file" },
/* video options */
- { "b", HAS_ARG, {opt_video_bitrate}, "set video bitrate (in kbit/s)", "bitrate" },
- { "r", HAS_ARG, {opt_frame_rate}, "set frame rate (in Hz)", "rate" },
- { "s", HAS_ARG, {opt_frame_size}, "set frame size (WxH or abbreviation)", "size" },
- { "g", HAS_ARG | OPT_EXPERT, {opt_gop_size}, "set the group of picture size", "gop_size" },
- { "intra", OPT_BOOL | OPT_EXPERT, {int_arg: &intra_only}, "use only intra frames"},
- { "vn", OPT_BOOL, {int_arg: &video_disable}, "disable video" },
- { "qscale", HAS_ARG | OPT_EXPERT, {opt_qscale}, "use fixed video quantiser scale (VBR)", "q" },
- { "vd", HAS_ARG | OPT_EXPERT, {opt_video_device}, "set video device", "device" },
- { "vcodec", HAS_ARG | OPT_EXPERT, {opt_video_codec}, "force video codec", "codec" },
- { "me", HAS_ARG | OPT_EXPERT, {opt_motion_estimation}, "set motion estimation method",
+ { "b", HAS_ARG, {(void*)opt_video_bitrate}, "set video bitrate (in kbit/s)", "bitrate" },
+ { "r", HAS_ARG, {(void*)opt_frame_rate}, "set frame rate (in Hz)", "rate" },
+ { "em_rate", OPT_BOOL|OPT_EXPERT, {(void*)&emulate_frame_rate}, "makes img reading happen at nominal frame rate" },
+ { "s", HAS_ARG, {(void*)opt_frame_size}, "set frame size (WxH or abbreviation)", "size" },
+ { "croptop", HAS_ARG, {(void*)opt_frame_crop_top}, "set top crop band size (in pixels)", "size" },
+ { "cropbottom", HAS_ARG, {(void*)opt_frame_crop_bottom}, "set bottom crop band size (in pixels)", "size" },
+ { "cropleft", HAS_ARG, {(void*)opt_frame_crop_left}, "set left crop band size (in pixels)", "size" },
+ { "cropright", HAS_ARG, {(void*)opt_frame_crop_right}, "set right crop band size (in pixels)", "size" },
+ { "g", HAS_ARG | OPT_EXPERT, {(void*)opt_gop_size}, "set the group of picture size", "gop_size" },
+ { "intra", OPT_BOOL | OPT_EXPERT, {(void*)&intra_only}, "use only intra frames"},
+ { "vn", OPT_BOOL, {(void*)&video_disable}, "disable video" },
+ { "qscale", HAS_ARG | OPT_EXPERT, {(void*)opt_qscale}, "use fixed video quantiser scale (VBR)", "q" },
+ { "qmin", HAS_ARG | OPT_EXPERT, {(void*)opt_qmin}, "min video quantiser scale (VBR)", "q" },
+ { "qmax", HAS_ARG | OPT_EXPERT, {(void*)opt_qmax}, "max video quantiser scale (VBR)", "q" },
+ { "mbqmin", HAS_ARG | OPT_EXPERT, {(void*)opt_mb_qmin}, "min macroblock quantiser scale (VBR)", "q" },
+ { "mbqmax", HAS_ARG | OPT_EXPERT, {(void*)opt_mb_qmax}, "max macroblock quantiser scale (VBR)", "q" },
+ { "qdiff", HAS_ARG | OPT_EXPERT, {(void*)opt_qdiff}, "max difference between the quantiser scale (VBR)", "q" },
+ { "qblur", HAS_ARG | OPT_EXPERT, {(void*)opt_qblur}, "video quantiser scale blur (VBR)", "blur" },
+ { "qcomp", HAS_ARG | OPT_EXPERT, {(void*)opt_qcomp}, "video quantiser scale compression (VBR)", "compression" },
+ { "rc_init_cplx", HAS_ARG | OPT_EXPERT, {(void*)opt_rc_initial_cplx}, "initial complexity for 1-pass encoding", "complexity" },
+ { "b_qfactor", HAS_ARG | OPT_EXPERT, {(void*)opt_b_qfactor}, "qp factor between p and b frames", "factor" },
+ { "i_qfactor", HAS_ARG | OPT_EXPERT, {(void*)opt_i_qfactor}, "qp factor between p and i frames", "factor" },
+ { "b_qoffset", HAS_ARG | OPT_EXPERT, {(void*)opt_b_qoffset}, "qp offset between p and b frames", "offset" },
+ { "i_qoffset", HAS_ARG | OPT_EXPERT, {(void*)opt_i_qoffset}, "qp offset between p and i frames", "offset" },
+// { "b_strategy", HAS_ARG | OPT_EXPERT, {(void*)opt_b_strategy}, "dynamic b frame selection strategy", "strategy" },
+ { "rc_eq", HAS_ARG | OPT_EXPERT, {(void*)opt_video_rc_eq}, "", "equation" },
+ { "rc_override", HAS_ARG | OPT_EXPERT, {(void*)opt_video_rc_override_string}, "Rate control override", "qualities for specific intervals" },
+ { "bt", HAS_ARG, {(void*)opt_video_bitrate_tolerance}, "set video bitrate tolerance (in kbit/s)", "tolerance" },
+ { "maxrate", HAS_ARG, {(void*)opt_video_bitrate_max}, "set max video bitrate tolerance (in kbit/s)", "bitrate" },
+ { "minrate", HAS_ARG, {(void*)opt_video_bitrate_min}, "set min video bitrate tolerance (in kbit/s)", "bitrate" },
+ { "bufsize", HAS_ARG, {(void*)opt_video_buffer_size}, "set ratecontrol buffere size (in kbit)", "size" },
+ { "vd", HAS_ARG | OPT_EXPERT, {(void*)opt_video_device}, "set video grab device", "device" },
+ { "vc", HAS_ARG | OPT_EXPERT, {(void*)opt_video_channel}, "set video grab channel (DV1394 only)", "channel" },
+ { "dv1394", OPT_EXPERT, {(void*)opt_dv1394}, "set DV1394 grab", "" },
+ { "vcodec", HAS_ARG | OPT_EXPERT, {(void*)opt_video_codec}, "force video codec ('copy' to copy stream)", "codec" },
+ { "me", HAS_ARG | OPT_EXPERT, {(void*)opt_motion_estimation}, "set motion estimation method",
"method" },
- { "sameq", OPT_BOOL, {int_arg: &same_quality},
+ { "dct_algo", HAS_ARG | OPT_EXPERT, {(void*)opt_dct_algo}, "set dct algo", "algo" },
+ { "idct_algo", HAS_ARG | OPT_EXPERT, {(void*)opt_idct_algo}, "set idct algo", "algo" },
+ { "er", HAS_ARG | OPT_EXPERT, {(void*)opt_error_resilience}, "set error resilience", "" },
+ { "ec", HAS_ARG | OPT_EXPERT, {(void*)opt_error_concealment}, "set error concealment", "" },
+ { "bf", HAS_ARG | OPT_EXPERT, {(void*)opt_b_frames}, "use 'frames' B frames (only MPEG-4)", "frames" },
+ { "hq", OPT_BOOL | OPT_EXPERT, {(void*)&use_hq}, "activate high quality settings" },
+ { "4mv", OPT_BOOL | OPT_EXPERT, {(void*)&use_4mv}, "use four motion vector by macroblock (only MPEG-4)" },
+ { "part", OPT_BOOL | OPT_EXPERT, {(void*)&use_part}, "use data partitioning (only MPEG-4)" },
+ { "bug", HAS_ARG | OPT_EXPERT, {(void*)opt_workaround_bugs}, "workaround not auto detected encoder bugs", "param" },
+ { "ps", HAS_ARG | OPT_EXPERT, {(void*)opt_packet_size}, "packet size", "size in bits" },
+ { "strict", HAS_ARG | OPT_EXPERT, {(void*)opt_strict}, "strictness", "how strictly to follow the standarts" },
+ { "sameq", OPT_BOOL, {(void*)&same_quality},
"use same video quality as source (implies VBR)" },
+ { "debug", HAS_ARG | OPT_EXPERT, {(void*)opt_debug}, "print specific debug info", "" },
/* audio options */
- { "ab", HAS_ARG, {opt_audio_bitrate}, "set audio bitrate (in kbit/s)", "bitrate", },
- { "ar", HAS_ARG, {opt_audio_rate}, "set audio sampling rate (in Hz)", "rate" },
- { "ac", HAS_ARG, {opt_audio_channels}, "set number of audio channels", "channels" },
- { "an", OPT_BOOL, {int_arg: &audio_disable}, "disable audio" },
- { "ad", HAS_ARG | OPT_EXPERT, {opt_audio_device}, "set audio device", "device" },
- { "acodec", HAS_ARG | OPT_EXPERT, {opt_audio_codec}, "force audio codec", "codec" },
-
+ { "ab", HAS_ARG, {(void*)opt_audio_bitrate}, "set audio bitrate (in kbit/s)", "bitrate", },
+ { "ar", HAS_ARG, {(void*)opt_audio_rate}, "set audio sampling rate (in Hz)", "rate" },
+ { "ac", HAS_ARG, {(void*)opt_audio_channels}, "set number of audio channels", "channels" },
+ { "an", OPT_BOOL, {(void*)&audio_disable}, "disable audio" },
+ { "ad", HAS_ARG | OPT_EXPERT, {(void*)opt_audio_device}, "set audio device", "device" },
+ { "acodec", HAS_ARG | OPT_EXPERT, {(void*)opt_audio_codec}, "force audio codec ('copy' to copy stream)", "codec" },
+ { "deinterlace", OPT_BOOL | OPT_EXPERT, {(void*)&do_deinterlace},
+ "deinterlace pictures" },
+ { "benchmark", OPT_BOOL | OPT_EXPERT, {(void*)&do_benchmark},
+ "add timings for benchmarking" },
+ { "hex", OPT_BOOL | OPT_EXPERT, {(void*)&do_hex_dump},
+ "dump each input packet" },
+ { "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" },
+ { "bitexact", OPT_EXPERT, {(void*)opt_bitexact}, "only use bit exact algorithms (for codec testing)" },
+ { "vhook", HAS_ARG | OPT_EXPERT, {(void*)add_frame_hooker}, "insert video processing module", "module name and parameters" },
{ NULL, },
};
int optindex, i;
const char *opt, *arg;
const OptionDef *po;
-
- register_all();
+ int64_t ti;
+
+ av_register_all();
+
+ /* detect if invoked as player */
+ i = strlen(argv[0]);
+ if (i >= 6 && !strcmp(argv[0] + i - 6, "ffplay"))
+ do_play = 1;
if (argc <= 1)
show_help();
-
+
+ /* parse options */
optindex = 1;
while (optindex < argc) {
opt = argv[optindex++];
exit(1);
}
arg = NULL;
- if (po->flags & HAS_ARG)
+ if (po->flags & HAS_ARG) {
arg = argv[optindex++];
+ if (!arg) {
+ fprintf(stderr, "%s: missing argument for option '%s'\n", argv[0], opt);
+ exit(1);
+ }
+ }
if (po->flags & OPT_STRING) {
char *str;
- str = strdup(arg);
+ str = av_strdup(arg);
*po->u.str_arg = str;
} else if (po->flags & OPT_BOOL) {
*po->u.int_arg = 1;
} else {
- po->u.func_arg(arg);
+ po->u.func_arg(arg);
}
} else {
- opt_output_file(opt);
+ if (!do_play) {
+ opt_output_file(opt);
+ } else {
+ opt_input_file(opt);
+ }
}
}
- if (nb_input_files == 0) {
- if (nb_output_files != 1) {
- fprintf(stderr, "Only one output file supported when grabbing\n");
+ if (!do_play) {
+ /* file converter / grab */
+ if (nb_output_files <= 0) {
+ fprintf(stderr, "Must supply at least one output file\n");
exit(1);
}
- av_grab(output_files[0]);
+
+ if (nb_input_files == 0) {
+ prepare_grab();
+ }
} else {
- if (nb_output_files <= 0) {
- fprintf(stderr, "Must supply at least one output file\n");
+ /* player */
+ if (nb_input_files <= 0) {
+ fprintf(stderr, "Must supply at least one input file\n");
exit(1);
}
- av_encode(output_files, nb_output_files, input_files, nb_input_files,
- stream_maps, nb_stream_maps);
+ prepare_play();
+ }
+
+ ti = getutime();
+ av_encode(output_files, nb_output_files, input_files, nb_input_files,
+ stream_maps, nb_stream_maps);
+ ti = getutime() - ti;
+ if (do_benchmark) {
+ printf("bench: utime=%0.3fs\n", ti / 1000000.0);
}
/* close files */
for(i=0;i<nb_output_files;i++) {
- if (!(output_files[i]->format->flags & AVFMT_NOFILE))
- url_fclose(&output_files[i]->pb);
- }
- for(i=0;i<nb_input_files;i++) {
- if (!(input_files[i]->format->flags & AVFMT_NOFILE))
- url_fclose(&input_files[i]->pb);
+ /* maybe av_close_output_file ??? */
+ AVFormatContext *s = output_files[i];
+ int j;
+ if (!(s->oformat->flags & AVFMT_NOFILE))
+ url_fclose(&s->pb);
+ for(j=0;j<s->nb_streams;j++)
+ av_free(s->streams[j]);
+ av_free(s);
}
+ for(i=0;i<nb_input_files;i++)
+ av_close_input_file(input_files[i]);
+
+ av_free_static();
+
+
+#ifdef POWERPC_TBL_PERFORMANCE_REPORT
+ extern void powerpc_display_perf_report(void);
+ powerpc_display_perf_report();
+#endif /* POWERPC_TBL_PERFORMANCE_REPORT */
return 0;
}