mirror of https://github.com/FFmpeg/FFmpeg.git
Baptiste Coudurier
17 years ago
parent
96e2fbf27b
commit
5118cae694
9 changed files with 242 additions and 3 deletions
@ -0,0 +1,227 @@ |
||||
/*
|
||||
* Ogg muxer |
||||
* Copyright (c) 2007 Baptiste Coudurier <baptiste dot coudurier at free dot fr> |
||||
* |
||||
* This file is part of FFmpeg. |
||||
* |
||||
* FFmpeg 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.1 of the License, or (at your option) any later version. |
||||
* |
||||
* FFmpeg 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 |
||||
* Lesser General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU Lesser General Public |
||||
* License along with FFmpeg; if not, write to the Free Software |
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
||||
*/ |
||||
|
||||
#include "avformat.h" |
||||
#include "crc.h" |
||||
#include "xiph.h" |
||||
#include "bytestream.h" |
||||
|
||||
typedef struct { |
||||
int64_t duration; |
||||
unsigned page_counter; |
||||
uint8_t *header[3]; |
||||
int header_len[3]; |
||||
/** for theora granule */ |
||||
int kfgshift; |
||||
int64_t last_kf_pts; |
||||
int vrev; |
||||
} OGGStreamContext; |
||||
|
||||
static void ogg_update_checksum(AVFormatContext *s, offset_t crc_offset) |
||||
{ |
||||
offset_t pos = url_ftell(&s->pb); |
||||
uint32_t checksum = get_checksum(&s->pb); |
||||
url_fseek(&s->pb, crc_offset, SEEK_SET); |
||||
put_be32(&s->pb, checksum); |
||||
url_fseek(&s->pb, pos, SEEK_SET); |
||||
} |
||||
|
||||
static int ogg_write_page(AVFormatContext *s, const uint8_t *data, int size, |
||||
int64_t granule, int stream_index, int flags) |
||||
{ |
||||
OGGStreamContext *oggstream = s->streams[stream_index]->priv_data; |
||||
offset_t crc_offset; |
||||
int page_segments, i; |
||||
|
||||
size = FFMIN(size, 255*255); |
||||
page_segments = FFMIN((size/255)+!!size, 255); |
||||
|
||||
init_checksum(&s->pb, ff_crc04C11DB7_update, 0); |
||||
put_tag(&s->pb, "OggS"); |
||||
put_byte(&s->pb, 0); |
||||
put_byte(&s->pb, flags); |
||||
put_le64(&s->pb, granule); |
||||
put_le32(&s->pb, stream_index); |
||||
put_le32(&s->pb, oggstream->page_counter++); |
||||
crc_offset = url_ftell(&s->pb); |
||||
put_le32(&s->pb, 0); // crc
|
||||
put_byte(&s->pb, page_segments); |
||||
for (i = 0; i < page_segments-1; i++) |
||||
put_byte(&s->pb, 255); |
||||
if (size) { |
||||
put_byte(&s->pb, size - (page_segments-1)*255); |
||||
put_buffer(&s->pb, data, size); |
||||
} |
||||
ogg_update_checksum(s, crc_offset); |
||||
put_flush_packet(&s->pb); |
||||
return size; |
||||
} |
||||
|
||||
static int ogg_build_flac_headers(const uint8_t *extradata, int extradata_size, |
||||
OGGStreamContext *oggstream, int bitexact) |
||||
{ |
||||
const char *vendor = bitexact ? "ffmpeg" : LIBAVFORMAT_IDENT; |
||||
uint8_t *p; |
||||
if (extradata_size != 34) |
||||
return -1; |
||||
oggstream->header_len[0] = 79; |
||||
oggstream->header[0] = av_mallocz(79); // per ogg flac specs
|
||||
p = oggstream->header[0]; |
||||
bytestream_put_byte(&p, 0x7F); |
||||
bytestream_put_buffer(&p, "FLAC", 4); |
||||
bytestream_put_byte(&p, 1); // major version
|
||||
bytestream_put_byte(&p, 0); // minor version
|
||||
bytestream_put_be16(&p, 1); // headers packets without this one
|
||||
bytestream_put_buffer(&p, "fLaC", 4); |
||||
bytestream_put_byte(&p, 0x00); // streaminfo
|
||||
bytestream_put_be24(&p, 34); |
||||
bytestream_put_buffer(&p, extradata, 34); |
||||
oggstream->header_len[1] = 1+3+4+strlen(vendor)+4; |
||||
oggstream->header[1] = av_mallocz(oggstream->header_len[1]); |
||||
p = oggstream->header[1]; |
||||
bytestream_put_byte(&p, 0x84); // last metadata block and vorbis comment
|
||||
bytestream_put_be24(&p, oggstream->header_len[1] - 4); |
||||
bytestream_put_le32(&p, strlen(vendor)); |
||||
bytestream_put_buffer(&p, vendor, strlen(vendor)); |
||||
bytestream_put_le32(&p, 0); // user comment list length
|
||||
return 0; |
||||
} |
||||
|
||||
static int ogg_write_header(AVFormatContext *s) |
||||
{ |
||||
OGGStreamContext *oggstream; |
||||
int i, j; |
||||
for (i = 0; i < s->nb_streams; i++) { |
||||
AVStream *st = s->streams[i]; |
||||
if (st->codec->codec_type == CODEC_TYPE_AUDIO) |
||||
av_set_pts_info(st, 64, 1, st->codec->sample_rate); |
||||
else if (st->codec->codec_type == CODEC_TYPE_VIDEO) |
||||
av_set_pts_info(st, 64, st->codec->time_base.num, st->codec->time_base.den); |
||||
if (st->codec->codec_id != CODEC_ID_VORBIS && |
||||
st->codec->codec_id != CODEC_ID_THEORA && |
||||
st->codec->codec_id != CODEC_ID_FLAC) { |
||||
av_log(s, AV_LOG_ERROR, "Unsupported codec id in stream %d\n", i); |
||||
return -1; |
||||
} |
||||
|
||||
if (!st->codec->extradata || !st->codec->extradata_size) { |
||||
av_log(s, AV_LOG_ERROR, "No extradata present\n"); |
||||
return -1; |
||||
} |
||||
oggstream = av_mallocz(sizeof(*oggstream)); |
||||
st->priv_data = oggstream; |
||||
if (st->codec->codec_id == CODEC_ID_FLAC) { |
||||
if (ogg_build_flac_headers(st->codec->extradata, st->codec->extradata_size, |
||||
oggstream, st->codec->flags & CODEC_FLAG_BITEXACT) < 0) { |
||||
av_log(s, AV_LOG_ERROR, "Extradata corrupted\n"); |
||||
av_freep(&st->priv_data); |
||||
} |
||||
} else { |
||||
if (ff_split_xiph_headers(st->codec->extradata, st->codec->extradata_size, |
||||
st->codec->codec_id == CODEC_ID_VORBIS ? 30 : 42, |
||||
oggstream->header, oggstream->header_len) < 0) { |
||||
av_log(s, AV_LOG_ERROR, "Extradata corrupted\n"); |
||||
av_freep(&st->priv_data); |
||||
return -1; |
||||
} |
||||
if (st->codec->codec_id == CODEC_ID_THEORA) { |
||||
/** KFGSHIFT is the width of the less significant section of the granule position
|
||||
The less significant section is the frame count since the last keyframe */ |
||||
oggstream->kfgshift = ((oggstream->header[0][40]&3)<<3)|(oggstream->header[0][41]>>5); |
||||
oggstream->vrev = oggstream->header[0][9]; |
||||
av_log(s, AV_LOG_DEBUG, "theora kfgshift %d, vrev %d\n", |
||||
oggstream->kfgshift, oggstream->vrev); |
||||
} |
||||
} |
||||
} |
||||
for (i = 0; i < 3; i++) { |
||||
for (j = 0; j < s->nb_streams; j++) { |
||||
AVStream *st = s->streams[j]; |
||||
OGGStreamContext *oggstream = st->priv_data; |
||||
if (oggstream && oggstream->header_len[i]) { |
||||
ogg_write_page(s, oggstream->header[i], oggstream->header_len[i], |
||||
0, st->index, i ? 0 : 2); // bos
|
||||
} |
||||
} |
||||
} |
||||
return 0; |
||||
} |
||||
|
||||
static int ogg_write_packet(AVFormatContext *s, AVPacket *pkt) |
||||
{ |
||||
AVStream *st = s->streams[pkt->stream_index]; |
||||
OGGStreamContext *oggstream = st->priv_data; |
||||
uint8_t *ptr = pkt->data; |
||||
int ret, size = pkt->size; |
||||
int64_t granule; |
||||
|
||||
if (st->codec->codec_id == CODEC_ID_THEORA) { |
||||
int64_t pts = oggstream->vrev < 1 ? pkt->pts : pkt->pts + pkt->duration; |
||||
int pframe_count; |
||||
if (pkt->flags & PKT_FLAG_KEY) |
||||
oggstream->last_kf_pts = pts; |
||||
pframe_count = pts - oggstream->last_kf_pts; |
||||
// prevent frame count from overflow if key frame flag is not set
|
||||
if (pframe_count >= (1<<oggstream->kfgshift)) { |
||||
oggstream->last_kf_pts += pframe_count; |
||||
pframe_count = 0; |
||||
} |
||||
granule = (oggstream->last_kf_pts<<oggstream->kfgshift) | pframe_count; |
||||
} else |
||||
granule = pkt->pts + pkt->duration; |
||||
oggstream->duration = granule; |
||||
do { |
||||
ret = ogg_write_page(s, ptr, size, granule, pkt->stream_index, ptr != pkt->data); |
||||
ptr += ret; size -= ret; |
||||
} while (size > 0 || ret == 255*255); // need to output a last nil page
|
||||
|
||||
return 0; |
||||
} |
||||
|
||||
|
||||
static int ogg_write_trailer(AVFormatContext *s) |
||||
{ |
||||
int i; |
||||
for (i = 0; i < s->nb_streams; i++) { |
||||
AVStream *st = s->streams[i]; |
||||
OGGStreamContext *oggstream = st->priv_data; |
||||
ogg_write_page(s, NULL, 0, oggstream->duration, i, 4); // eos
|
||||
if (st->codec->codec_id == CODEC_ID_FLAC) { |
||||
av_free(oggstream->header[0]); |
||||
av_free(oggstream->header[1]); |
||||
} |
||||
av_freep(&st->priv_data); |
||||
} |
||||
return 0; |
||||
} |
||||
|
||||
AVOutputFormat ogg_muxer = { |
||||
"ogg", |
||||
"Ogg format", |
||||
"application/ogg", |
||||
"ogg", |
||||
0, |
||||
CODEC_ID_FLAC, |
||||
CODEC_ID_THEORA, |
||||
ogg_write_header, |
||||
ogg_write_packet, |
||||
ogg_write_trailer, |
||||
}; |
||||
Loading…
Reference in new issue