# HG changeset patch # User aurel # Date 1280011812 0 # Node ID 2b1ab404f1cf1a6343ead29e1cee2be094a9795d # Parent 43fe2984f310595e0f72bdff311f7f21925cf4cf add SubRip muxer and demuxer diff -r 43fe2984f310 -r 2b1ab404f1cf Makefile --- a/Makefile Sat Jul 24 12:58:28 2010 +0000 +++ b/Makefile Sat Jul 24 22:50:12 2010 +0000 @@ -242,6 +242,8 @@ OBJS-$(CONFIG_SOX_DEMUXER) += soxdec.o raw.o OBJS-$(CONFIG_SOX_MUXER) += soxenc.o OBJS-$(CONFIG_SPDIF_MUXER) += spdif.o +OBJS-$(CONFIG_SRT_DEMUXER) += srtdec.o +OBJS-$(CONFIG_SRT_MUXER) += raw.o OBJS-$(CONFIG_STR_DEMUXER) += psxstr.o OBJS-$(CONFIG_SWF_DEMUXER) += swfdec.o OBJS-$(CONFIG_SWF_MUXER) += swfenc.o diff -r 43fe2984f310 -r 2b1ab404f1cf allformats.c --- a/allformats.c Sat Jul 24 12:58:28 2010 +0000 +++ b/allformats.c Sat Jul 24 22:50:12 2010 +0000 @@ -189,6 +189,7 @@ REGISTER_DEMUXER (SOL, sol); REGISTER_MUXDEMUX (SOX, sox); REGISTER_MUXER (SPDIF, spdif); + REGISTER_MUXDEMUX (SRT, srt); REGISTER_DEMUXER (STR, str); REGISTER_MUXDEMUX (SWF, swf); REGISTER_MUXER (TG2, tg2); diff -r 43fe2984f310 -r 2b1ab404f1cf avformat.h --- a/avformat.h Sat Jul 24 12:58:28 2010 +0000 +++ b/avformat.h Sat Jul 24 22:50:12 2010 +0000 @@ -22,7 +22,7 @@ #define AVFORMAT_AVFORMAT_H #define LIBAVFORMAT_VERSION_MAJOR 52 -#define LIBAVFORMAT_VERSION_MINOR 76 +#define LIBAVFORMAT_VERSION_MINOR 77 #define LIBAVFORMAT_VERSION_MICRO 0 #define LIBAVFORMAT_VERSION_INT AV_VERSION_INT(LIBAVFORMAT_VERSION_MAJOR, \ diff -r 43fe2984f310 -r 2b1ab404f1cf raw.c --- a/raw.c Sat Jul 24 12:58:28 2010 +0000 +++ b/raw.c Sat Jul 24 22:50:12 2010 +0000 @@ -1082,6 +1082,18 @@ }; #endif +#if CONFIG_SRT_MUXER +AVOutputFormat srt_muxer = { + .name = "srt", + .long_name = NULL_IF_CONFIG_SMALL("SubRip subtitle format"), + .mime_type = "application/x-subrip", + .extensions = "srt", + .write_packet = raw_write_packet, + .flags = AVFMT_NOTIMESTAMPS, + .subtitle_codec = CODEC_ID_SRT, +}; +#endif + #if CONFIG_TRUEHD_DEMUXER AVInputFormat truehd_demuxer = { "truehd", diff -r 43fe2984f310 -r 2b1ab404f1cf srtdec.c --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/srtdec.c Sat Jul 24 22:50:12 2010 +0000 @@ -0,0 +1,102 @@ +/* + * SubRip subtitle demuxer + * Copyright (c) 2010 Aurelien Jacobs + * + * 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 "internal.h" +#include "libavutil/intreadwrite.h" + +static int srt_probe(AVProbeData *p) +{ + unsigned char *ptr = p->buf; + int i, v, num = 0; + + if (AV_RB24(ptr) == 0xEFBBBF) + ptr += 3; /* skip UTF-8 BOM */ + + for (i=0; i<2; i++) { + if (num == i && sscanf(ptr, "%*d:%*2d:%*2d%*1[,.]%*3d --> %*d:%*2d:%*2d%*1[,.]%3d", &v) == 1) + return AVPROBE_SCORE_MAX; + num = atoi(ptr); + ptr += strcspn(ptr, "\n") + 1; + } + return 0; +} + +static int srt_read_header(AVFormatContext *s, AVFormatParameters *ap) +{ + AVStream *st = av_new_stream(s, 0); + if (!st) + return -1; + av_set_pts_info(st, 64, 1, 1000); + st->codec->codec_type = AVMEDIA_TYPE_SUBTITLE; + st->codec->codec_id = CODEC_ID_SRT; + return 0; +} + +static int64_t get_pts(const char *buf) +{ + int i, v, hour, min, sec, hsec; + + for (i=0; i<2; i++) { + if (sscanf(buf, "%d:%2d:%2d%*1[,.]%3d --> %*d:%*2d:%*2d%*1[,.]%3d", + &hour, &min, &sec, &hsec, &v) == 5) { + min += 60*hour; + sec += 60*min; + return sec*1000+hsec; + } + buf += strcspn(buf, "\n") + 1; + } + return AV_NOPTS_VALUE; +} + +static inline int is_eol(char c) +{ + return c == '\r' || c == '\n'; +} + +static int srt_read_packet(AVFormatContext *s, AVPacket *pkt) +{ + char buffer[2048], *ptr = buffer, *ptr2; + int64_t pos = url_ftell(s->pb); + int res = AVERROR_EOF; + + do { + ptr2 = ptr; + ptr += ff_get_line(s->pb, ptr, sizeof(buffer)+buffer-ptr); + } while (!is_eol(*ptr2) && !url_feof(s->pb) && ptr-bufferdata, buffer, pkt->size); + pkt->flags |= AV_PKT_FLAG_KEY; + pkt->pos = pos; + pkt->pts = pkt->dts = get_pts(pkt->data); + } + return res; +} + +AVInputFormat srt_demuxer = { + .name = "srt", + .long_name = NULL_IF_CONFIG_SMALL("SubRip subtitle format"), + .read_probe = srt_probe, + .read_header = srt_read_header, + .read_packet = srt_read_packet, + .flags = AVFMT_GENERIC_INDEX, +};