/*
* Various utilities for ffmpeg system
- * 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 "avformat.h"
+#include <ctype.h>
#ifndef CONFIG_WIN32
#include <unistd.h>
#include <fcntl.h>
#include <sys/time.h>
-#include <time.h>
#else
#define strcasecmp _stricmp
#include <sys/types.h>
#include <sys/timeb.h>
#endif
+#include <time.h>
+
+#ifndef HAVE_STRPTIME
+#include "strptime.h"
+#endif
+
+AVInputFormat *first_iformat;
+AVOutputFormat *first_oformat;
-AVFormat *first_format;
+void av_register_input_format(AVInputFormat *format)
+{
+ AVInputFormat **p;
+ p = &first_iformat;
+ while (*p != NULL) p = &(*p)->next;
+ *p = format;
+ format->next = NULL;
+}
-void register_avformat(AVFormat *format)
+void av_register_output_format(AVOutputFormat *format)
{
- AVFormat **p;
- p = &first_format;
+ AVOutputFormat **p;
+ p = &first_oformat;
while (*p != NULL) p = &(*p)->next;
*p = format;
format->next = NULL;
return 0;
}
-AVFormat *guess_format(const char *short_name, const char *filename, const char *mime_type)
+AVOutputFormat *guess_format(const char *short_name, const char *filename,
+ const char *mime_type)
{
- AVFormat *fmt, *fmt_found;
+ AVOutputFormat *fmt, *fmt_found;
int score_max, score;
/* find the proper file type */
fmt_found = NULL;
score_max = 0;
- fmt = first_format;
+ fmt = first_oformat;
while (fmt != NULL) {
score = 0;
if (fmt->name && short_name && !strcmp(fmt->name, short_name))
return fmt_found;
}
-/* return TRUE if val is a prefix of str. If it returns TRUE, ptr is
- set to the next character in 'str' after the prefix */
-int strstart(const char *str, const char *val, const char **ptr)
+AVOutputFormat *guess_stream_format(const char *short_name, const char *filename,
+ const char *mime_type)
{
- const char *p, *q;
- p = str;
- q = val;
- while (*q != '\0') {
- if (*p != *q)
- return 0;
- p++;
- q++;
- }
- if (ptr)
- *ptr = p;
- return 1;
-}
+ AVOutputFormat *fmt = guess_format(short_name, filename, mime_type);
-void nstrcpy(char *buf, int buf_size, const char *str)
-{
- int c;
- char *q = buf;
+ if (fmt) {
+ AVOutputFormat *stream_fmt;
+ char stream_format_name[64];
- for(;;) {
- c = *str++;
- if (c == 0 || q >= buf + buf_size - 1)
- break;
- *q++ = c;
+ snprintf(stream_format_name, sizeof(stream_format_name), "%s_stream", fmt->name);
+ stream_fmt = guess_format(stream_format_name, NULL, NULL);
+
+ if (stream_fmt)
+ fmt = stream_fmt;
}
- *q = '\0';
+
+ return fmt;
}
-void register_all(void)
-{
- avcodec_init();
- avcodec_register_all();
-
- register_avformat(&mp2_format);
- register_avformat(&ac3_format);
- register_avformat(&mpeg_mux_format);
- register_avformat(&mpeg1video_format);
- register_avformat(&mjpeg_format);
- register_avformat(&h263_format);
- register_avformat(&rm_format);
- register_avformat(&asf_format);
- register_avformat(&avi_format);
- register_avformat(&mpjpeg_format);
- register_avformat(&jpeg_format);
- register_avformat(&single_jpeg_format);
- register_avformat(&swf_format);
- register_avformat(&wav_format);
- register_avformat(&pcm_format);
- register_avformat(&rawvideo_format);
-#ifndef CONFIG_WIN32
- register_avformat(&ffm_format);
-#endif
- register_avformat(&pgm_format);
- register_avformat(&ppm_format);
- register_avformat(&pgmyuv_format);
- register_avformat(&imgyuv_format);
- register_avformat(&pgmpipe_format);
- register_avformat(&pgmyuvpipe_format);
- register_avformat(&ppmpipe_format);
-
- register_protocol(&file_protocol);
- register_protocol(&pipe_protocol);
-#ifdef CONFIG_GRAB
- register_protocol(&audio_protocol);
- register_protocol(&video_protocol);
-#endif
-#ifndef CONFIG_WIN32
- register_protocol(&udp_protocol);
- register_protocol(&http_protocol);
-#endif
+AVInputFormat *av_find_input_format(const char *short_name)
+{
+ AVInputFormat *fmt;
+ for(fmt = first_iformat; fmt != NULL; fmt = fmt->next) {
+ if (!strcmp(fmt->name, short_name))
+ return fmt;
+ }
+ return NULL;
}
/* memory handling */
+/**
+ * Allocate the payload of a packet and intialized its fields to default values.
+ *
+ * @param pkt packet
+ * @param size wanted payload size
+ * @return 0 if OK. AVERROR_xxx otherwise.
+ */
int av_new_packet(AVPacket *pkt, int size)
{
- pkt->data = malloc(size);
+ int i;
+ pkt->data = av_malloc(size + FF_INPUT_BUFFER_PADDING_SIZE);
if (!pkt->data)
- return -ENOMEM;
+ return AVERROR_NOMEM;
pkt->size = size;
/* sane state */
- pkt->pts = 0;
+ pkt->pts = AV_NOPTS_VALUE;
pkt->stream_index = 0;
pkt->flags = 0;
+
+ for(i=0; i<FF_INPUT_BUFFER_PADDING_SIZE; i++)
+ pkt->data[size+i]= 0;
+
return 0;
}
+/**
+ * Free a packet
+ *
+ * @param pkt packet to free
+ */
void av_free_packet(AVPacket *pkt)
{
- free(pkt->data);
+ av_freep(&pkt->data);
/* fail safe */
- pkt->data = NULL;
pkt->size = 0;
}
int fifo_init(FifoBuffer *f, int size)
{
- f->buffer = malloc(size);
+ f->buffer = av_malloc(size);
if (!f->buffer)
return -1;
f->end = f->buffer + size;
void fifo_free(FifoBuffer *f)
{
- free(f->buffer);
+ av_free(f->buffer);
}
int fifo_size(FifoBuffer *f, UINT8 *rptr)
*wptr_ptr = wptr;
}
-/* media file handling */
+int filename_number_test(const char *filename)
+{
+ char buf[1024];
+ return get_frame_filename(buf, sizeof(buf), filename, 1);
+}
+
+/* guess file format */
+AVInputFormat *av_probe_input_format(AVProbeData *pd, int is_opened)
+{
+ AVInputFormat *fmt1, *fmt;
+ int score, score_max;
-AVFormatContext *av_open_input_file(const char *filename, int buf_size)
+ fmt = NULL;
+ score_max = 0;
+ for(fmt1 = first_iformat; fmt1 != NULL; fmt1 = fmt1->next) {
+ if (!is_opened && !(fmt1->flags & AVFMT_NOFILE))
+ continue;
+ score = 0;
+ if (fmt1->read_probe) {
+ score = fmt1->read_probe(pd);
+ } else if (fmt1->extensions) {
+ if (match_ext(pd->filename, fmt1->extensions)) {
+ score = 50;
+ }
+ }
+ if (score > score_max) {
+ score_max = score;
+ fmt = fmt1;
+ }
+ }
+ return fmt;
+}
+
+/************************************************************/
+/* input media file */
+
+#define PROBE_BUF_SIZE 2048
+
+/**
+ * Open a media file as input. The codec are not opened. Only the file
+ * header (if present) is read.
+ *
+ * @param ic_ptr the opened media file handle is put here
+ * @param filename filename to open.
+ * @param fmt if non NULL, force the file format to use
+ * @param buf_size optional buffer size (zero if default is OK)
+ * @param ap additionnal parameters needed when opening the file (NULL if default)
+ * @return 0 if OK. AVERROR_xxx otherwise.
+ */
+int av_open_input_file(AVFormatContext **ic_ptr, const char *filename,
+ AVInputFormat *fmt,
+ int buf_size,
+ AVFormatParameters *ap)
{
- AVFormatParameters params, *ap;
- AVFormat *fmt;
AVFormatContext *ic = NULL;
- URLFormat url_format;
int err;
+ char buf[PROBE_BUF_SIZE];
+ AVProbeData probe_data, *pd = &probe_data;
ic = av_mallocz(sizeof(AVFormatContext));
- if (!ic)
- goto fail;
- if (url_fopen(&ic->pb, filename, URL_RDONLY) < 0)
+ if (!ic) {
+ err = AVERROR_NOMEM;
goto fail;
-
- if (buf_size > 0) {
- url_setbufsize(&ic->pb, buf_size);
- }
-
- /* find format */
- err = url_getformat(url_fileno(&ic->pb), &url_format);
- if (err >= 0) {
- fmt = guess_format(url_format.format_name, NULL, NULL);
- ap = ¶ms;
- 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 {
- fmt = guess_format(NULL, filename, NULL);
- ap = NULL;
}
- if (!fmt || !fmt->read_header) {
- return NULL;
+ pstrcpy(ic->filename, sizeof(ic->filename), filename);
+ pd->filename = ic->filename;
+ pd->buf = buf;
+ pd->buf_size = 0;
+
+ if (!fmt) {
+ /* guess format if no file can be opened */
+ fmt = av_probe_input_format(pd, 0);
}
- ic->format = fmt;
- err = ic->format->read_header(ic, ap);
- if (err < 0) {
+ /* if no file needed do not try to open one */
+ if (!fmt || !(fmt->flags & AVFMT_NOFILE)) {
+ if (url_fopen(&ic->pb, filename, URL_RDONLY) < 0) {
+ err = AVERROR_IO;
+ goto fail;
+ }
+ if (buf_size > 0) {
+ url_setbufsize(&ic->pb, buf_size);
+ }
+ if (!fmt) {
+ /* read probe data */
+ pd->buf_size = get_buffer(&ic->pb, buf, PROBE_BUF_SIZE);
+ url_fseek(&ic->pb, 0, SEEK_SET);
+ }
+ }
+
+ /* guess file format */
+ if (!fmt) {
+ fmt = av_probe_input_format(pd, 1);
+ }
+
+ /* if still no format found, error */
+ if (!fmt) {
+ err = AVERROR_NOFMT;
+ goto fail;
+ }
+
+ /* XXX: suppress this hack for redirectors */
+ if (fmt == &redir_demux) {
+ err = redir_open(ic_ptr, &ic->pb);
url_fclose(&ic->pb);
+ av_free(ic);
+ return err;
+ }
+
+ ic->iformat = fmt;
+
+ /* allocate private data */
+ ic->priv_data = av_mallocz(fmt->priv_data_size);
+ if (!ic->priv_data) {
+ err = AVERROR_NOMEM;
goto fail;
}
-
- return ic;
+ /* default pts settings is MPEG like */
+ av_set_pts_info(ic, 33, 1, 90000);
+
+ /* check filename in case of an image number is expected */
+ if (ic->iformat->flags & AVFMT_NEEDNUMBER) {
+ if (filename_number_test(ic->filename) < 0) {
+ err = AVERROR_NUMEXPECTED;
+ goto fail1;
+ }
+ }
+
+ err = ic->iformat->read_header(ic, ap);
+ if (err < 0)
+ goto fail1;
+ *ic_ptr = ic;
+ return 0;
+ fail1:
+ if (!(fmt->flags & AVFMT_NOFILE)) {
+ url_fclose(&ic->pb);
+ }
fail:
- if (ic)
- free(ic);
- return NULL;
+ if (ic) {
+ av_freep(&ic->priv_data);
+ }
+ av_free(ic);
+ *ic_ptr = NULL;
+ return err;
}
+/**
+ * Read a packet from a media file
+ * @param s media file handle
+ * @param pkt is filled
+ * @return 0 if OK. AVERROR_xxx if error.
+ */
int av_read_packet(AVFormatContext *s, AVPacket *pkt)
{
AVPacketList *pktl;
/* read packet from packet buffer, if there is data */
*pkt = pktl->pkt;
s->packet_buffer = pktl->next;
- free(pktl);
+ av_free(pktl);
return 0;
} else {
- return s->format->read_packet(s, pkt);
+ return s->iformat->read_packet(s, pkt);
}
}
+/* state for codec information */
+#define CSTATE_NOTFOUND 0
+#define CSTATE_DECODING 1
+#define CSTATE_FOUND 2
+
+static int has_codec_parameters(AVCodecContext *enc)
+{
+ int val;
+ 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;
+ }
+ return (val != 0);
+}
+
+/**
+ * Read the beginning of a media file to get stream information. This
+ * is useful for file formats with no headers such as MPEG. This
+ * function also compute the real frame rate in case of mpeg2 repeat
+ * frame mode.
+ *
+ * @param ic media file handle
+ * @return >=0 if OK. AVERROR_xxx if error.
+ */
+int av_find_stream_info(AVFormatContext *ic)
+{
+ int i, count, ret, got_picture, size, read_size;
+ AVCodec *codec;
+ AVStream *st;
+ AVPacket *pkt;
+ AVPicture picture;
+ AVPacketList *pktl=NULL, **ppktl;
+ short samples[AVCODEC_MAX_AUDIO_FRAME_SIZE / 2];
+ UINT8 *ptr;
+ int min_read_size, max_read_size;
+
+ /* typical mpeg ts rate is 40 Mbits. DVD rate is about 10
+ Mbits. We read at most 0.1 second of file to find all streams */
+
+ /* XXX: base it on stream bitrate when possible */
+ if (ic->iformat == &mpegts_demux) {
+ /* maximum number of bytes we accept to read to find all the streams
+ in a file */
+ min_read_size = 3000000;
+ } else {
+ min_read_size = 125000;
+ }
+ /* max read size is 2 seconds of video max */
+ max_read_size = min_read_size * 20;
+
+ /* set initial codec state */
+ for(i=0;i<ic->nb_streams;i++) {
+ st = ic->streams[i];
+ if (has_codec_parameters(&st->codec))
+ st->codec_info_state = CSTATE_FOUND;
+ else
+ st->codec_info_state = CSTATE_NOTFOUND;
+ st->codec_info_nb_repeat_frames = 0;
+ st->codec_info_nb_real_frames = 0;
+ }
+
+ count = 0;
+ read_size = 0;
+ ppktl = &ic->packet_buffer;
+ for(;;) {
+ /* check if one codec still needs to be handled */
+ for(i=0;i<ic->nb_streams;i++) {
+ st = ic->streams[i];
+ if (st->codec_info_state != CSTATE_FOUND)
+ break;
+ }
+ if (i == ic->nb_streams) {
+ /* NOTE: if the format has no header, then we need to read
+ some packets to get most of the streams, so we cannot
+ stop here */
+ if (!(ic->iformat->flags & AVFMT_NOHEADER) ||
+ read_size >= min_read_size) {
+ /* if we found the info for all the codecs, we can stop */
+ ret = count;
+ break;
+ }
+ } else {
+ /* we did not get all the codec info, but we read too much data */
+ if (read_size >= max_read_size) {
+ ret = count;
+ break;
+ }
+ }
+
+ pktl = av_mallocz(sizeof(AVPacketList));
+ if (!pktl) {
+ ret = AVERROR_NOMEM;
+ break;
+ }
+
+ /* add the packet in the buffered packet list */
+ *ppktl = pktl;
+ ppktl = &pktl->next;
+
+ /* NOTE: a new stream can be added there if no header in file
+ (AVFMT_NOHEADER) */
+ pkt = &pktl->pkt;
+ if (ic->iformat->read_packet(ic, pkt) < 0) {
+ /* EOF or error */
+ ret = -1; /* we could not have all the codec parameters before EOF */
+ if ((ic->iformat->flags & AVFMT_NOHEADER) &&
+ i == ic->nb_streams)
+ ret = 0;
+ break;
+ }
+ read_size += pkt->size;
+
+ /* open new codecs */
+ for(i=0;i<ic->nb_streams;i++) {
+ st = ic->streams[i];
+ if (st->codec_info_state == CSTATE_NOTFOUND) {
+ /* set to found in case of error */
+ st->codec_info_state = CSTATE_FOUND;
+ codec = avcodec_find_decoder(st->codec.codec_id);
+ if (codec) {
+ if(codec->capabilities & CODEC_CAP_TRUNCATED)
+ st->codec.flags |= CODEC_FLAG_TRUNCATED;
+
+ ret = avcodec_open(&st->codec, codec);
+ if (ret >= 0)
+ st->codec_info_state = CSTATE_DECODING;
+ }
+ }
+ }
+
+ st = ic->streams[pkt->stream_index];
+ if (st->codec_info_state == CSTATE_DECODING) {
+ /* 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) {
+ /* if error, simply ignore because another packet
+ may be OK */
+ break;
+ }
+ if (got_picture) {
+ /* we got the parameters - now we can stop
+ examining this stream */
+ /* XXX: add a codec info so that we can decide if
+ the codec can repeat frames */
+ if (st->codec.codec_id == CODEC_ID_MPEG1VIDEO &&
+ ic->iformat != &mpegts_demux &&
+ st->codec.sub_id == 2) {
+ /* for mpeg2 video, we want to know the real
+ frame rate, so we decode 40 frames. In mpeg
+ TS case we do not do it because it would be
+ too long */
+ st->codec_info_nb_real_frames++;
+ st->codec_info_nb_repeat_frames += st->codec.repeat_pict;
+#if 0
+ /* XXX: testing */
+ if ((st->codec_info_nb_real_frames % 24) == 23) {
+ st->codec_info_nb_repeat_frames += 2;
+ }
+#endif
+ /* stop after 40 frames */
+ if (st->codec_info_nb_real_frames >= 40) {
+ st->r_frame_rate = (st->codec.frame_rate *
+ st->codec_info_nb_real_frames) /
+ (st->codec_info_nb_real_frames +
+ (st->codec_info_nb_repeat_frames >> 1));
+ goto close_codec;
+ }
+ } else {
+ close_codec:
+ st->codec_info_state = CSTATE_FOUND;
+ avcodec_close(&st->codec);
+ break;
+ }
+ }
+ ptr += ret;
+ size -= ret;
+ }
+ }
+ count++;
+ }
+
+ /* close each codec if there are opened */
+ for(i=0;i<ic->nb_streams;i++) {
+ st = ic->streams[i];
+ if (st->codec_info_state == CSTATE_DECODING)
+ avcodec_close(&st->codec);
+ }
+
+ /* set real frame rate info */
+ for(i=0;i<ic->nb_streams;i++) {
+ st = ic->streams[i];
+ if (st->codec.codec_type == CODEC_TYPE_VIDEO) {
+ if (!st->r_frame_rate)
+ st->r_frame_rate = st->codec.frame_rate;
+ }
+ }
+
+ return ret;
+}
+
+/**
+ * Close a media file (but not its codecs)
+ *
+ * @param s media file handle
+ */
void av_close_input_file(AVFormatContext *s)
{
int i;
- if (s->format->read_close)
- s->format->read_close(s);
+ if (s->iformat->read_close)
+ s->iformat->read_close(s);
for(i=0;i<s->nb_streams;i++) {
- free(s->streams[i]);
+ av_free(s->streams[i]);
}
if (s->packet_buffer) {
AVPacketList *p, *p1;
while (p != NULL) {
p1 = p->next;
av_free_packet(&p->pkt);
- free(p);
+ av_free(p);
p = p1;
}
s->packet_buffer = NULL;
}
- url_fclose(&s->pb);
- free(s);
+ if (!(s->iformat->flags & AVFMT_NOFILE)) {
+ url_fclose(&s->pb);
+ }
+ av_freep(&s->priv_data);
+ av_free(s);
+}
+
+/**
+ * Add a new stream to a media file. Can only be called in the
+ * read_header function. If the flag AVFMT_NOHEADER is in the format
+ * description, then new streams can be added in read_packet too.
+ *
+ *
+ * @param s media file handle
+ * @param id file format dependent stream id
+ */
+AVStream *av_new_stream(AVFormatContext *s, int id)
+{
+ AVStream *st;
+
+ if (s->nb_streams >= MAX_STREAMS)
+ return NULL;
+
+ st = av_mallocz(sizeof(AVStream));
+ if (!st)
+ return NULL;
+ st->index = s->nb_streams;
+ st->id = id;
+ s->streams[s->nb_streams++] = st;
+ return st;
}
+/************************************************************/
+/* output media file */
-int av_write_packet(AVFormatContext *s, AVPacket *pkt)
+/**
+ * allocate the stream private data and write the stream header to an
+ * output media file
+ *
+ * @param s media file handle
+ * @return 0 if OK. AVERROR_xxx if error.
+ */
+int av_write_header(AVFormatContext *s)
{
- /* XXX: currently, an emulation because internal API must change */
- return s->format->write_packet(s, pkt->stream_index, pkt->data, pkt->size);
+ int ret, i;
+ AVStream *st;
+
+ s->priv_data = av_mallocz(s->oformat->priv_data_size);
+ if (!s->priv_data)
+ return AVERROR_NOMEM;
+ /* default pts settings is MPEG like */
+ av_set_pts_info(s, 33, 1, 90000);
+ ret = s->oformat->write_header(s);
+ if (ret < 0)
+ return ret;
+
+ /* init PTS generation */
+ for(i=0;i<s->nb_streams;i++) {
+ st = s->streams[i];
+
+ switch (st->codec.codec_type) {
+ case CODEC_TYPE_AUDIO:
+ av_frac_init(&st->pts, 0, 0,
+ (INT64)s->pts_num * st->codec.sample_rate);
+ break;
+ case CODEC_TYPE_VIDEO:
+ av_frac_init(&st->pts, 0, 0,
+ (INT64)s->pts_num * st->codec.frame_rate);
+ break;
+ default:
+ break;
+ }
+ }
+ return 0;
+}
+
+/**
+ * Write a packet to an output media file. The packet shall contain
+ * one audio or video frame.
+ *
+ * @param s media file handle
+ * @param stream_index stream index
+ * @param buf buffer containing the frame data
+ * @param size size of buffer
+ * @return < 0 if error, = 0 if OK, 1 if end of stream wanted.
+ */
+int av_write_frame(AVFormatContext *s, int stream_index, const uint8_t *buf,
+ int size)
+{
+ AVStream *st;
+ INT64 pts_mask;
+ int ret, frame_size;
+
+ st = s->streams[stream_index];
+ pts_mask = (1LL << s->pts_wrap_bits) - 1;
+ ret = s->oformat->write_packet(s, stream_index, (uint8_t *)buf, size,
+ st->pts.val & pts_mask);
+ if (ret < 0)
+ return ret;
+
+ /* update pts */
+ switch (st->codec.codec_type) {
+ case CODEC_TYPE_AUDIO:
+ if (st->codec.frame_size <= 1) {
+ frame_size = size / st->codec.channels;
+ /* specific hack for pcm codecs because no frame size is provided */
+ switch(st->codec.codec_id) {
+ case CODEC_ID_PCM_S16LE:
+ case CODEC_ID_PCM_S16BE:
+ case CODEC_ID_PCM_U16LE:
+ case CODEC_ID_PCM_U16BE:
+ frame_size >>= 1;
+ break;
+ default:
+ break;
+ }
+ } else {
+ frame_size = st->codec.frame_size;
+ }
+ av_frac_add(&st->pts,
+ (INT64)s->pts_den * frame_size);
+ break;
+ case CODEC_TYPE_VIDEO:
+ av_frac_add(&st->pts,
+ (INT64)s->pts_den * FRAME_RATE_BASE);
+ break;
+ default:
+ break;
+ }
+ return ret;
+}
+
+/**
+ * write the stream trailer to an output media file and and free the
+ * file private data.
+ *
+ * @param s media file handle
+ * @return 0 if OK. AVERROR_xxx if error. */
+int av_write_trailer(AVFormatContext *s)
+{
+ int ret;
+ ret = s->oformat->write_trailer(s);
+ av_freep(&s->priv_data);
+ return ret;
}
/* "user interface" functions */
const char *url,
int is_output)
{
- int i;
+ int i, flags;
char buf[256];
fprintf(stderr, "%s #%d, %s, %s '%s':\n",
is_output ? "Output" : "Input",
- index, ic->format->name,
+ index,
+ is_output ? ic->oformat->name : ic->iformat->name,
is_output ? "to" : "from", url);
for(i=0;i<ic->nb_streams;i++) {
AVStream *st = ic->streams[i];
avcodec_string(buf, sizeof(buf), &st->codec, is_output);
- fprintf(stderr, " Stream #%d.%d: %s\n", index, i, buf);
+ fprintf(stderr, " Stream #%d.%d", index, i);
+ /* the pid is an important information, so we display it */
+ /* XXX: add a generic system */
+ if (is_output)
+ flags = ic->oformat->flags;
+ else
+ flags = ic->iformat->flags;
+ if (flags & AVFMT_SHOW_IDS) {
+ fprintf(stderr, "[0x%x]", st->id);
+ }
+ fprintf(stderr, ": %s\n", buf);
}
}
return 0;
}
-INT64 gettime(void)
+INT64 av_gettime(void)
{
#ifdef CONFIG_WIN32
struct _timeb tb;
#endif
}
-/* syntax: [YYYY-MM-DD ][[HH:]MM:]SS[.m...] . Return the date in micro seconds since 1970 */
+static time_t mktimegm(struct tm *tm)
+{
+ time_t t;
+
+ int y = tm->tm_year + 1900, m = tm->tm_mon + 1, d = tm->tm_mday;
+
+ if (m < 3) {
+ m += 12;
+ y--;
+ }
+
+ t = 86400 *
+ (d + (153 * m - 457) / 5 + 365 * y + y / 4 - y / 100 + y / 400 - 719469);
+
+ t += 3600 * tm->tm_hour + 60 * tm->tm_min + tm->tm_sec;
+
+ return t;
+}
+
+/* Syntax:
+ * - If not a duration:
+ * [{YYYY-MM-DD|YYYYMMDD}]{T| }{HH[:MM[:SS[.m...]]][Z]|HH[MM[SS[.m...]]][Z]}
+ * Time is localtime unless Z is suffixed to the end. In this case GMT
+ * Return the date in micro seconds since 1970
+ * - If duration:
+ * HH[:MM[:SS[.m...]]]
+ * S+[.m...]
+ */
INT64 parse_date(const char *datestr, int duration)
{
const char *p;
INT64 t;
- int sec;
+ struct tm dt;
+ int i;
+ static const char *date_fmt[] = {
+ "%Y-%m-%d",
+ "%Y%m%d",
+ };
+ static const char *time_fmt[] = {
+ "%H:%M:%S",
+ "%H%M%S",
+ };
+ const char *q;
+ int is_utc, len;
+ char lastch;
+ time_t now = time(0);
+
+ len = strlen(datestr);
+ if (len > 0)
+ lastch = datestr[len - 1];
+ else
+ lastch = '\0';
+ is_utc = (lastch == 'z' || lastch == 'Z');
+
+ memset(&dt, 0, sizeof(dt));
p = datestr;
+ q = NULL;
if (!duration) {
- static const UINT8 months[12] = { 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31 };
- int year, month, day, i;
+ for (i = 0; i < sizeof(date_fmt) / sizeof(date_fmt[0]); i++) {
+ q = strptime(p, date_fmt[i], &dt);
+ if (q) {
+ break;
+ }
+ }
- if (strlen(p) >= 5 && p[4] == '-') {
-
- year = strtol(p, (char **)&p, 10);
- if (*p)
- p++;
- month = strtol(p, (char **)&p, 10) - 1;
- if (*p)
- p++;
- day = strtol(p, (char **)&p, 10) - 1;
- if (*p)
- p++;
- day += (year - 1970) * 365;
- /* if >= March, take February of current year into account too */
- if (month >= 2)
- year++;
- for(i=1970;i<year;i++) {
- if ((i % 100) == 0) {
- if ((i % 400) == 0) day++;
- } else if ((i % 4) == 0) {
- day++;
- }
+ if (!q) {
+ if (is_utc) {
+ dt = *gmtime(&now);
+ } else {
+ dt = *localtime(&now);
}
- for(i=0;i<month;i++)
- day += months[i];
+ dt.tm_hour = dt.tm_min = dt.tm_sec = 0;
} else {
- day = (time(NULL) / (3600 * 24));
+ p = q;
+ }
+
+ if (*p == 'T' || *p == 't' || *p == ' ')
+ p++;
+
+ for (i = 0; i < sizeof(time_fmt) / sizeof(time_fmt[0]); i++) {
+ q = strptime(p, time_fmt[i], &dt);
+ if (q) {
+ break;
+ }
}
- t = day * (3600 * 24);
} else {
- t = 0;
+ q = strptime(p, time_fmt[0], &dt);
+ if (!q) {
+ dt.tm_sec = strtol(p, (char **)&q, 10);
+ dt.tm_min = 0;
+ dt.tm_hour = 0;
+ }
}
-
- sec = 0;
- for(;;) {
- int val;
- val = strtol(p, (char **)&p, 10);
- sec = sec * 60 + val;
- if (*p != ':')
- break;
- p++;
+
+ /* Now we have all the fields that we can get */
+ if (!q) {
+ if (duration)
+ return 0;
+ else
+ return now * INT64_C(1000000);
+ }
+
+ if (duration) {
+ t = dt.tm_hour * 3600 + dt.tm_min * 60 + dt.tm_sec;
+ } else {
+ dt.tm_isdst = -1; /* unknown */
+ if (is_utc) {
+ t = mktimegm(&dt);
+ } else {
+ t = mktime(&dt);
+ }
}
- t = (t + sec) * 1000000;
- if (*p == '.') {
+
+ t *= 1000000;
+
+ if (*q == '.') {
int val, n;
- p++;
- n = strlen(p);
- if (n > 6)
- n = 6;
- val = strtol(p, NULL, 10);
- while (n < 6) {
- val = val * 10;
- n++;
+ q++;
+ for (val = 0, n = 100000; n >= 1; n /= 10, q++) {
+ if (!isdigit(*q))
+ break;
+ val += n * (*q - '0');
}
t += val;
}
return t;
}
-/* syntax: '?tag1=val1&tag2=val2...'. No URL decoding is done. Return
+/* syntax: '?tag1=val1&tag2=val2...'. Little URL decoding is done. Return
1 if found */
int find_info_tag(char *arg, int arg_size, const char *tag1, const char *info)
{
if (*p == '=') {
p++;
while (*p != '&' && *p != '\0') {
- if ((q - arg) < arg_size - 1)
- *q++ = *p;
+ if ((q - arg) < arg_size - 1) {
+ if (*p == '+')
+ *q++ = ' ';
+ else
+ *q++ = *p;
+ }
p++;
}
*q = '\0';
return 1;
if (*p != '&')
break;
+ p++;
}
return 0;
}
return -1;
}
+/**
+ *
+ * Print on stdout a nice hexa dump of a buffer
+ * @param buf buffer
+ * @param size buffer size
+ */
+void av_hex_dump(UINT8 *buf, int size)
+{
+ int len, i, j, c;
+ for(i=0;i<size;i+=16) {
+ len = size - i;
+ if (len > 16)
+ len = 16;
+ printf("%08x ", i);
+ for(j=0;j<16;j++) {
+ if (j < len)
+ printf(" %02x", buf[i+j]);
+ else
+ printf(" ");
+ }
+ printf(" ");
+ for(j=0;j<len;j++) {
+ c = buf[i+j];
+ if (c < ' ' || c > '~')
+ c = '.';
+ printf("%c", c);
+ }
+ printf("\n");
+ }
+}
+
+void url_split(char *proto, int proto_size,
+ char *hostname, int hostname_size,
+ int *port_ptr,
+ char *path, int path_size,
+ const char *url)
+{
+ const char *p;
+ char *q;
+ int port;
+
+ port = -1;
+
+ p = url;
+ q = proto;
+ while (*p != ':' && *p != '\0') {
+ if ((q - proto) < proto_size - 1)
+ *q++ = *p;
+ p++;
+ }
+ if (proto_size > 0)
+ *q = '\0';
+ if (*p == '\0') {
+ if (proto_size > 0)
+ proto[0] = '\0';
+ if (hostname_size > 0)
+ hostname[0] = '\0';
+ p = url;
+ } else {
+ p++;
+ if (*p == '/')
+ p++;
+ if (*p == '/')
+ p++;
+ q = hostname;
+ while (*p != ':' && *p != '/' && *p != '?' && *p != '\0') {
+ if ((q - hostname) < hostname_size - 1)
+ *q++ = *p;
+ p++;
+ }
+ if (hostname_size > 0)
+ *q = '\0';
+ if (*p == ':') {
+ p++;
+ port = strtoul(p, (char **)&p, 10);
+ }
+ }
+ if (port_ptr)
+ *port_ptr = port;
+ pstrcpy(path, path_size, p);
+}
+
+/**
+ * Set the pts for a given stream
+ * @param s stream
+ * @param pts_wrap_bits number of bits effectively used by the pts
+ * (used for wrap control, 33 is the value for MPEG)
+ * @param pts_num numerator to convert to seconds (MPEG: 1)
+ * @param pts_den denominator to convert to seconds (MPEG: 90000)
+ */
+void av_set_pts_info(AVFormatContext *s, int pts_wrap_bits,
+ int pts_num, int pts_den)
+{
+ s->pts_wrap_bits = pts_wrap_bits;
+ s->pts_num = pts_num;
+ s->pts_den = pts_den;
+}
+
+/* fraction handling */
+
+/**
+ * f = val + (num / den) + 0.5. 'num' is normalized so that it is such
+ * as 0 <= num < den.
+ *
+ * @param f fractional number
+ * @param val integer value
+ * @param num must be >= 0
+ * @param den must be >= 1
+ */
+void av_frac_init(AVFrac *f, INT64 val, INT64 num, INT64 den)
+{
+ num += (den >> 1);
+ if (num >= den) {
+ val += num / den;
+ num = num % den;
+ }
+ f->val = val;
+ f->num = num;
+ f->den = den;
+}
+
+/* set f to (val + 0.5) */
+void av_frac_set(AVFrac *f, INT64 val)
+{
+ f->val = val;
+ f->num = f->den >> 1;
+}
+
+/**
+ * Fractionnal addition to f: f = f + (incr / f->den)
+ *
+ * @param f fractional number
+ * @param incr increment, can be positive or negative
+ */
+void av_frac_add(AVFrac *f, INT64 incr)
+{
+ INT64 num, den;
+
+ num = f->num + incr;
+ den = f->den;
+ if (num < 0) {
+ f->val += num / den;
+ num = num % den;
+ if (num < 0) {
+ num += den;
+ f->val--;
+ }
+ } else if (num >= den) {
+ f->val += num / den;
+ num = num % den;
+ }
+ f->num = num;
+}