diff options
author | Aurelien Jacobs <aurel@gnuage.org> | 2010-07-25 02:50:12 +0400 |
---|---|---|
committer | Aurelien Jacobs <aurel@gnuage.org> | 2010-07-25 02:50:12 +0400 |
commit | ba2c508d0c7192e623fe835b3dea9328fdc0f03b (patch) | |
tree | 85adc3c7429dc5eb66dbe4ea52d2d3641b7ed916 /libavformat | |
parent | e3f7bf774caad334b258f4494e77420a7c161b63 (diff) |
add SubRip muxer and demuxer
Originally committed as revision 24488 to svn://svn.ffmpeg.org/ffmpeg/trunk
Diffstat (limited to 'libavformat')
-rw-r--r-- | libavformat/Makefile | 2 | ||||
-rw-r--r-- | libavformat/allformats.c | 1 | ||||
-rw-r--r-- | libavformat/avformat.h | 2 | ||||
-rw-r--r-- | libavformat/raw.c | 12 | ||||
-rw-r--r-- | libavformat/srtdec.c | 102 |
5 files changed, 118 insertions, 1 deletions
diff --git a/libavformat/Makefile b/libavformat/Makefile index 50a7b3adef..c32a254878 100644 --- a/libavformat/Makefile +++ b/libavformat/Makefile @@ -242,6 +242,8 @@ OBJS-$(CONFIG_SOL_DEMUXER) += sol.o raw.o 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 --git a/libavformat/allformats.c b/libavformat/allformats.c index ce0e90d544..0b39e84adb 100644 --- a/libavformat/allformats.c +++ b/libavformat/allformats.c @@ -189,6 +189,7 @@ void av_register_all(void) 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 --git a/libavformat/avformat.h b/libavformat/avformat.h index 076b8c84eb..8ba2d81605 100644 --- a/libavformat/avformat.h +++ b/libavformat/avformat.h @@ -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 --git a/libavformat/raw.c b/libavformat/raw.c index 0c83c429ba..dc48c91a46 100644 --- a/libavformat/raw.c +++ b/libavformat/raw.c @@ -1082,6 +1082,18 @@ AVOutputFormat mlp_muxer = { }; #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 --git a/libavformat/srtdec.c b/libavformat/srtdec.c new file mode 100644 index 0000000000..104031a422 --- /dev/null +++ b/libavformat/srtdec.c @@ -0,0 +1,102 @@ +/* + * SubRip subtitle demuxer + * Copyright (c) 2010 Aurelien Jacobs <aurel@gnuage.org> + * + * 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-buffer<sizeof(buffer)-1); + + if (buffer[0] && !(res = av_new_packet(pkt, ptr-buffer))) { + memcpy(pkt->data, 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, +}; |