mirror of https://github.com/FFmpeg/FFmpeg.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
1098 lines
28 KiB
1098 lines
28 KiB
/* |
|
* Various utilities for ffmpeg system |
|
* Copyright (c) 2000, 2001, 2002 Gerard Lantau |
|
* |
|
* 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 program 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. |
|
* |
|
* 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. |
|
*/ |
|
#include "avformat.h" |
|
#include "tick.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 |
|
|
|
AVInputFormat *first_iformat; |
|
AVOutputFormat *first_oformat; |
|
|
|
void av_register_input_format(AVInputFormat *format) |
|
{ |
|
AVInputFormat **p; |
|
p = &first_iformat; |
|
while (*p != NULL) p = &(*p)->next; |
|
*p = format; |
|
format->next = NULL; |
|
} |
|
|
|
void av_register_output_format(AVOutputFormat *format) |
|
{ |
|
AVOutputFormat **p; |
|
p = &first_oformat; |
|
while (*p != NULL) p = &(*p)->next; |
|
*p = format; |
|
format->next = NULL; |
|
} |
|
|
|
int match_ext(const char *filename, const char *extensions) |
|
{ |
|
const char *ext, *p; |
|
char ext1[32], *q; |
|
|
|
ext = strrchr(filename, '.'); |
|
if (ext) { |
|
ext++; |
|
p = extensions; |
|
for(;;) { |
|
q = ext1; |
|
while (*p != '\0' && *p != ',') |
|
*q++ = *p++; |
|
*q = '\0'; |
|
if (!strcasecmp(ext1, ext)) |
|
return 1; |
|
if (*p == '\0') |
|
break; |
|
p++; |
|
} |
|
} |
|
return 0; |
|
} |
|
|
|
AVOutputFormat *guess_format(const char *short_name, const char *filename, |
|
const char *mime_type) |
|
{ |
|
AVOutputFormat *fmt, *fmt_found; |
|
int score_max, score; |
|
|
|
/* find the proper file type */ |
|
fmt_found = NULL; |
|
score_max = 0; |
|
fmt = first_oformat; |
|
while (fmt != NULL) { |
|
score = 0; |
|
if (fmt->name && short_name && !strcmp(fmt->name, short_name)) |
|
score += 100; |
|
if (fmt->mime_type && mime_type && !strcmp(fmt->mime_type, mime_type)) |
|
score += 10; |
|
if (filename && fmt->extensions && |
|
match_ext(filename, fmt->extensions)) { |
|
score += 5; |
|
} |
|
if (score > score_max) { |
|
score_max = score; |
|
fmt_found = fmt; |
|
} |
|
fmt = fmt->next; |
|
} |
|
return fmt_found; |
|
} |
|
|
|
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; |
|
} |
|
|
|
|
|
/** |
|
* 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. |
|
* |
|
* @param str input string |
|
* @param val prefix to test |
|
* @param ptr updated after the prefix in str in there is a match |
|
* @return TRUE if there is a match |
|
*/ |
|
int strstart(const char *str, const char *val, const char **ptr) |
|
{ |
|
const char *p, *q; |
|
p = str; |
|
q = val; |
|
while (*q != '\0') { |
|
if (*p != *q) |
|
return 0; |
|
p++; |
|
q++; |
|
} |
|
if (ptr) |
|
*ptr = p; |
|
return 1; |
|
} |
|
|
|
/** |
|
* Copy the string str to buf. If str length is bigger than buf_size - |
|
* 1 then it is clamped to buf_size - 1. |
|
* NOTE: this function does what strncpy should have done to be |
|
* useful. NEVER use strncpy. |
|
* |
|
* @param buf destination buffer |
|
* @param buf_size size of destination buffer |
|
* @param str source string |
|
*/ |
|
void pstrcpy(char *buf, int buf_size, const char *str) |
|
{ |
|
int c; |
|
char *q = buf; |
|
|
|
for(;;) { |
|
c = *str++; |
|
if (c == 0 || q >= buf + buf_size - 1) |
|
break; |
|
*q++ = c; |
|
} |
|
*q = '\0'; |
|
} |
|
|
|
void register_all(void) |
|
{ |
|
avcodec_init(); |
|
avcodec_register_all(); |
|
|
|
mpegps_init(); |
|
mpegts_init(); |
|
crc_init(); |
|
img_init(); |
|
raw_init(); |
|
rm_init(); |
|
asf_init(); |
|
avienc_init(); |
|
avidec_init(); |
|
wav_init(); |
|
swf_init(); |
|
au_init(); |
|
gif_init(); |
|
mov_init(); |
|
jpeg_init(); |
|
|
|
#ifndef CONFIG_WIN32 |
|
ffm_init(); |
|
#endif |
|
#ifdef CONFIG_GRAB |
|
video_grab_init(); |
|
audio_init(); |
|
#endif |
|
/* file protocols */ |
|
register_protocol(&file_protocol); |
|
register_protocol(&pipe_protocol); |
|
#ifndef CONFIG_WIN32 |
|
register_protocol(&udp_protocol); |
|
register_protocol(&http_protocol); |
|
#endif |
|
} |
|
|
|
/* 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 = av_malloc(size); |
|
if (!pkt->data) |
|
return AVERROR_NOMEM; |
|
pkt->size = size; |
|
/* sane state */ |
|
pkt->pts = 0; |
|
pkt->stream_index = 0; |
|
pkt->flags = 0; |
|
return 0; |
|
} |
|
|
|
/** |
|
* Free a packet |
|
* |
|
* @param pkt packet to free |
|
*/ |
|
void av_free_packet(AVPacket *pkt) |
|
{ |
|
av_freep(&pkt->data); |
|
/* fail safe */ |
|
pkt->size = 0; |
|
} |
|
|
|
/* fifo handling */ |
|
|
|
int fifo_init(FifoBuffer *f, int size) |
|
{ |
|
f->buffer = av_malloc(size); |
|
if (!f->buffer) |
|
return -1; |
|
f->end = f->buffer + size; |
|
f->wptr = f->rptr = f->buffer; |
|
return 0; |
|
} |
|
|
|
void fifo_free(FifoBuffer *f) |
|
{ |
|
av_free(f->buffer); |
|
} |
|
|
|
int fifo_size(FifoBuffer *f, UINT8 *rptr) |
|
{ |
|
int size; |
|
|
|
if (f->wptr >= rptr) { |
|
size = f->wptr - rptr; |
|
} else { |
|
size = (f->end - rptr) + (f->wptr - f->buffer); |
|
} |
|
return size; |
|
} |
|
|
|
/* get data from the fifo (return -1 if not enough data) */ |
|
int fifo_read(FifoBuffer *f, UINT8 *buf, int buf_size, UINT8 **rptr_ptr) |
|
{ |
|
UINT8 *rptr = *rptr_ptr; |
|
int size, len; |
|
|
|
if (f->wptr >= rptr) { |
|
size = f->wptr - rptr; |
|
} else { |
|
size = (f->end - rptr) + (f->wptr - f->buffer); |
|
} |
|
|
|
if (size < buf_size) |
|
return -1; |
|
while (buf_size > 0) { |
|
len = f->end - rptr; |
|
if (len > buf_size) |
|
len = buf_size; |
|
memcpy(buf, rptr, len); |
|
buf += len; |
|
rptr += len; |
|
if (rptr >= f->end) |
|
rptr = f->buffer; |
|
buf_size -= len; |
|
} |
|
*rptr_ptr = rptr; |
|
return 0; |
|
} |
|
|
|
void fifo_write(FifoBuffer *f, UINT8 *buf, int size, UINT8 **wptr_ptr) |
|
{ |
|
int len; |
|
UINT8 *wptr; |
|
wptr = *wptr_ptr; |
|
while (size > 0) { |
|
len = f->end - wptr; |
|
if (len > size) |
|
len = size; |
|
memcpy(wptr, buf, len); |
|
wptr += len; |
|
if (wptr >= f->end) |
|
wptr = f->buffer; |
|
buf += len; |
|
size -= len; |
|
} |
|
*wptr_ptr = wptr; |
|
} |
|
|
|
int filename_number_test(const char *filename) |
|
{ |
|
char buf[1024]; |
|
return get_frame_filename(buf, sizeof(buf), filename, 1); |
|
} |
|
|
|
/* guess file format */ |
|
static AVInputFormat *probe_input_format(AVProbeData *pd, int is_opened) |
|
{ |
|
AVInputFormat *fmt1, *fmt; |
|
int score, score_max; |
|
|
|
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->extensions) { |
|
if (match_ext(pd->filename, fmt1->extensions)) { |
|
score = 50; |
|
} |
|
} else if (fmt1->read_probe) { |
|
score = fmt1->read_probe(pd); |
|
} |
|
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) |
|
{ |
|
AVFormatContext *ic = NULL; |
|
int err; |
|
char buf[PROBE_BUF_SIZE]; |
|
AVProbeData probe_data, *pd = &probe_data; |
|
|
|
ic = av_mallocz(sizeof(AVFormatContext)); |
|
if (!ic) { |
|
err = AVERROR_NOMEM; |
|
goto fail; |
|
} |
|
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 = probe_input_format(pd, 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); |
|
} |
|
/* 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 = probe_input_format(pd, 1); |
|
} |
|
|
|
/* if still no format found, error */ |
|
if (!fmt) { |
|
err = AVERROR_NOFMT; |
|
goto fail; |
|
} |
|
|
|
ic->iformat = fmt; |
|
|
|
/* allocate private data */ |
|
ic->priv_data = av_mallocz(fmt->priv_data_size); |
|
if (!ic->priv_data) { |
|
err = AVERROR_NOMEM; |
|
goto fail; |
|
} |
|
|
|
/* 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) { |
|
av_free(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; |
|
|
|
pktl = s->packet_buffer; |
|
if (pktl) { |
|
/* read packet from packet buffer, if there is data */ |
|
*pkt = pktl->pkt; |
|
s->packet_buffer = pktl->next; |
|
av_free(pktl); |
|
return 0; |
|
} else { |
|
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) { |
|
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->iformat->read_close) |
|
s->iformat->read_close(s); |
|
for(i=0;i<s->nb_streams;i++) { |
|
av_free(s->streams[i]); |
|
} |
|
if (s->packet_buffer) { |
|
AVPacketList *p, *p1; |
|
p = s->packet_buffer; |
|
while (p != NULL) { |
|
p1 = p->next; |
|
av_free_packet(&p->pkt); |
|
av_free(p); |
|
p = p1; |
|
} |
|
s->packet_buffer = NULL; |
|
} |
|
if (!(s->iformat->flags & AVFMT_NOFILE)) { |
|
url_fclose(&s->pb); |
|
} |
|
av_free(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 */ |
|
|
|
/** |
|
* 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) |
|
{ |
|
s->priv_data = av_mallocz(s->oformat->priv_data_size); |
|
if (!s->priv_data) |
|
return AVERROR_NOMEM; |
|
return s->oformat->write_header(s); |
|
} |
|
|
|
/** |
|
* write a packet to an output media file |
|
* |
|
* @param s media file handle |
|
* @param pkt packet to write |
|
* @param force_pts XXX: need to suppress that |
|
*/ |
|
int av_write_packet(AVFormatContext *s, AVPacket *pkt, int force_pts) |
|
{ |
|
/* XXX: currently, an emulation because internal API must change */ |
|
return s->oformat->write_packet(s, pkt->stream_index, pkt->data, pkt->size, force_pts); |
|
} |
|
|
|
/** |
|
* 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 */ |
|
|
|
void dump_format(AVFormatContext *ic, |
|
int index, |
|
const char *url, |
|
int is_output) |
|
{ |
|
int i, flags; |
|
char buf[256]; |
|
|
|
fprintf(stderr, "%s #%d, %s, %s '%s':\n", |
|
is_output ? "Output" : "Input", |
|
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", 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); |
|
} |
|
} |
|
|
|
typedef struct { |
|
const char *str; |
|
int width, height; |
|
} SizeEntry; |
|
|
|
static SizeEntry sizes[] = { |
|
{ "sqcif", 128, 96 }, |
|
{ "qcif", 176, 144 }, |
|
{ "cif", 352, 288 }, |
|
{ "4cif", 704, 576 }, |
|
}; |
|
|
|
int parse_image_size(int *width_ptr, int *height_ptr, const char *str) |
|
{ |
|
int i; |
|
int n = sizeof(sizes) / sizeof(SizeEntry); |
|
const char *p; |
|
int frame_width = 0, frame_height = 0; |
|
|
|
for(i=0;i<n;i++) { |
|
if (!strcmp(sizes[i].str, str)) { |
|
frame_width = sizes[i].width; |
|
frame_height = sizes[i].height; |
|
break; |
|
} |
|
} |
|
if (i == n) { |
|
p = str; |
|
frame_width = strtol(p, (char **)&p, 10); |
|
if (*p) |
|
p++; |
|
frame_height = strtol(p, (char **)&p, 10); |
|
} |
|
if (frame_width <= 0 || frame_height <= 0) |
|
return -1; |
|
*width_ptr = frame_width; |
|
*height_ptr = frame_height; |
|
return 0; |
|
} |
|
|
|
INT64 gettime(void) |
|
{ |
|
#ifdef CONFIG_WIN32 |
|
struct _timeb tb; |
|
_ftime(&tb); |
|
return ((INT64)tb.time * INT64_C(1000) + (INT64)tb.millitm) * INT64_C(1000); |
|
#else |
|
struct timeval tv; |
|
gettimeofday(&tv,NULL); |
|
return (INT64)tv.tv_sec * 1000000 + tv.tv_usec; |
|
#endif |
|
} |
|
|
|
/* syntax: [YYYY-MM-DD ][[HH:]MM:]SS[.m...] . Return the date in micro seconds since 1970 */ |
|
INT64 parse_date(const char *datestr, int duration) |
|
{ |
|
const char *p; |
|
INT64 t; |
|
int sec; |
|
|
|
p = datestr; |
|
if (!duration) { |
|
static const UINT8 months[12] = { 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31 }; |
|
int year, month, day, i; |
|
|
|
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++; |
|
} |
|
} |
|
for(i=0;i<month;i++) |
|
day += months[i]; |
|
} else { |
|
day = (time(NULL) / (3600 * 24)); |
|
} |
|
t = day * (3600 * 24); |
|
} else { |
|
t = 0; |
|
} |
|
|
|
sec = 0; |
|
for(;;) { |
|
int val; |
|
val = strtol(p, (char **)&p, 10); |
|
sec = sec * 60 + val; |
|
if (*p != ':') |
|
break; |
|
p++; |
|
} |
|
t = (t + sec) * 1000000; |
|
if (*p == '.') { |
|
int val, n; |
|
p++; |
|
n = strlen(p); |
|
if (n > 6) |
|
n = 6; |
|
val = strtol(p, NULL, 10); |
|
while (n < 6) { |
|
val = val * 10; |
|
n++; |
|
} |
|
t += val; |
|
} |
|
return t; |
|
} |
|
|
|
/* syntax: '?tag1=val1&tag2=val2...'. No URL decoding is done. Return |
|
1 if found */ |
|
int find_info_tag(char *arg, int arg_size, const char *tag1, const char *info) |
|
{ |
|
const char *p; |
|
char tag[128], *q; |
|
|
|
p = info; |
|
if (*p == '?') |
|
p++; |
|
for(;;) { |
|
q = tag; |
|
while (*p != '\0' && *p != '=' && *p != '&') { |
|
if ((q - tag) < sizeof(tag) - 1) |
|
*q++ = *p; |
|
p++; |
|
} |
|
*q = '\0'; |
|
q = arg; |
|
if (*p == '=') { |
|
p++; |
|
while (*p != '&' && *p != '\0') { |
|
if ((q - arg) < arg_size - 1) |
|
*q++ = *p; |
|
p++; |
|
} |
|
*q = '\0'; |
|
} |
|
if (!strcmp(tag, tag1)) |
|
return 1; |
|
if (*p != '&') |
|
break; |
|
p++; |
|
} |
|
return 0; |
|
} |
|
|
|
/* Return in 'buf' the path with '%d' replaced by number. Also handles |
|
the '%0nd' format where 'n' is the total number of digits and |
|
'%%'. Return 0 if OK, and -1 if format error */ |
|
int get_frame_filename(char *buf, int buf_size, |
|
const char *path, int number) |
|
{ |
|
const char *p; |
|
char *q, buf1[20]; |
|
int nd, len, c, percentd_found; |
|
|
|
q = buf; |
|
p = path; |
|
percentd_found = 0; |
|
for(;;) { |
|
c = *p++; |
|
if (c == '\0') |
|
break; |
|
if (c == '%') { |
|
nd = 0; |
|
while (*p >= '0' && *p <= '9') { |
|
nd = nd * 10 + *p++ - '0'; |
|
} |
|
c = *p++; |
|
switch(c) { |
|
case '%': |
|
goto addchar; |
|
case 'd': |
|
if (percentd_found) |
|
goto fail; |
|
percentd_found = 1; |
|
snprintf(buf1, sizeof(buf1), "%0*d", nd, number); |
|
len = strlen(buf1); |
|
if ((q - buf + len) > buf_size - 1) |
|
goto fail; |
|
memcpy(q, buf1, len); |
|
q += len; |
|
break; |
|
default: |
|
goto fail; |
|
} |
|
} else { |
|
addchar: |
|
if ((q - buf) < buf_size - 1) |
|
*q++ = c; |
|
} |
|
} |
|
if (!percentd_found) |
|
goto fail; |
|
*q = '\0'; |
|
return 0; |
|
fail: |
|
*q = '\0'; |
|
return -1; |
|
} |
|
|
|
static int gcd(INT64 a, INT64 b) |
|
{ |
|
INT64 c; |
|
|
|
while (1) { |
|
c = a % b; |
|
if (c == 0) |
|
return b; |
|
a = b; |
|
b = c; |
|
} |
|
} |
|
|
|
void ticker_init(Ticker *tick, INT64 inrate, INT64 outrate) |
|
{ |
|
int g; |
|
|
|
g = gcd(inrate, outrate); |
|
inrate /= g; |
|
outrate /= g; |
|
|
|
tick->value = -outrate/2; |
|
|
|
tick->inrate = inrate; |
|
tick->outrate = outrate; |
|
tick->div = tick->outrate / tick->inrate; |
|
tick->mod = tick->outrate % tick->inrate; |
|
} |
|
|
|
/** |
|
* |
|
* 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"); |
|
} |
|
} |
|
|
|
|