Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- commit f4bd048d4121ce017c7a9427f647566083cccbd1
- Author: Eejya <singh.eejya@gmail.com>
- Date: Fri Oct 3 01:43:20 2014 +0530
- Added STL demuxer for supporting subtitles with .stl file format.
- diff --git a/libavformat/Makefile b/libavformat/Makefile
- index 86064ea..60e056d 100644
- --- a/libavformat/Makefile
- +++ b/libavformat/Makefile
- @@ -433,6 +433,7 @@ OBJS-$(CONFIG_VOBSUB_DEMUXER) += subtitles.o # mpeg demuxer is in the
- OBJS-$(CONFIG_VOC_DEMUXER) += vocdec.o voc.o
- OBJS-$(CONFIG_VOC_MUXER) += vocenc.o voc.o
- OBJS-$(CONFIG_VPLAYER_DEMUXER) += vplayerdec.o subtitles.o
- +OBJS-$(CONFIG_STL_DEMUXER) += stldec.o subtitles.o
- OBJS-$(CONFIG_VQF_DEMUXER) += vqf.o
- OBJS-$(CONFIG_W64_DEMUXER) += wavdec.o w64.o pcm.o
- OBJS-$(CONFIG_W64_MUXER) += wavenc.o w64.o
- diff --git a/libavformat/allformats.c b/libavformat/allformats.c
- index d54ed9b..c1a3a4e 100644
- --- a/libavformat/allformats.c
- +++ b/libavformat/allformats.c
- @@ -258,6 +258,7 @@ void av_register_all(void)
- REGISTER_DEMUXER (SBG, sbg);
- REGISTER_DEMUXER (SDP, sdp);
- REGISTER_DEMUXER (SDR2, sdr2);
- + REGISTER_DEMUXER (STL, stl);
- #if CONFIG_RTPDEC
- ff_register_rtp_dynamic_payload_handlers();
- ff_register_rdt_dynamic_payload_handlers();
- diff --git a/libavformat/stldec.c b/libavformat/stldec.c
- new file mode 100644
- index 0000000..a428de0
- --- /dev/null
- +++ b/libavformat/stldec.c
- @@ -0,0 +1,110 @@
- +#include "avformat.h"
- +#include "internal.h"
- +#include "subtitles.h"
- +#include "libavutil/intreadwrite.h"
- +typedef struct {
- + FFDemuxSubtitlesQueue q;
- +} STLContext;
- +
- +static int stl_probe(AVProbeData *p)
- +{
- + char c;
- + const unsigned char *ptr = p->buf;
- + while(*ptr=='\r' || *ptr=='\n' || *ptr=='$' || (ptr[0]=='/' && ptr[1]=='/'))
- + ptr+=ff_subtitles_next_line(ptr);
- +
- + if (sscanf(ptr, "%*d:%*d:%*d:%*d , %*d:%*d:%*d:%*d , %c", &c) == 1)
- + return AVPROBE_SCORE_MAX;
- + return 0;
- +
- +}
- +static int64_t get_pts(char **buf, int *duration)
- +{
- + int hh1, mm1, ss1, ms1;
- + int hh2, mm2, ss2, ms2;
- + int len;
- + if (sscanf(*buf, "%2d:%2d:%2d:%2d , %2d:%2d:%2d:%2d , %n",
- + &hh1, &mm1, &ss1, &ms1,
- + &hh2, &mm2, &ss2, &ms2, &len) >= 6) {
- +
- + int64_t start = (hh1*3600LL + mm1*60LL + ss1) * 100LL + ms1;
- + int64_t end = (hh2*3600LL + mm2*60LL + ss2) * 100LL + ms2;
- + *duration = end - start;
- + *buf+=len;
- + return start;
- + }
- + return AV_NOPTS_VALUE;
- +}
- +
- +static int stl_read_header(AVFormatContext *s)
- +{
- + STLContext *stl = s->priv_data;
- + AVStream *st = avformat_new_stream(s, NULL);
- +
- + if (!st)
- + return AVERROR(ENOMEM);
- + avpriv_set_pts_info(st, 64, 1, 100);
- + st->codec->codec_type = AVMEDIA_TYPE_SUBTITLE;
- + st->codec->codec_id = AV_CODEC_ID_TEXT;
- +
- + while (!avio_feof(s->pb)) {
- + char line[4096];
- + char *p = line;
- + const int64_t pos = avio_tell(s->pb);
- + int len = ff_get_line(s->pb, line, sizeof(line));
- + int64_t pts_start;
- + int duration;
- + if (!len)
- + break;
- +
- + line[strcspn(line, "\r\n")] = 0;
- +
- + pts_start = get_pts(&p , &duration);
- + if (pts_start != AV_NOPTS_VALUE) {
- + AVPacket *sub;
- +
- + sub = ff_subtitles_queue_insert(&stl->q, p, strlen(p), 0);
- + if (!sub)
- + return AVERROR(ENOMEM);
- + sub->pos = pos;
- + sub->pts = pts_start;
- + sub->duration = duration;
- + }
- + }
- +
- + ff_subtitles_queue_finalize(&stl->q);
- + return 0;
- +
- +}
- +static int stl_read_packet(AVFormatContext *s, AVPacket *pkt)
- +{
- + STLContext *stl = s->priv_data;
- + return ff_subtitles_queue_read_packet(&stl->q, pkt);
- +}
- +
- +static int stl_read_seek(AVFormatContext *s, int stream_index,
- + int64_t min_ts, int64_t ts, int64_t max_ts, int flags)
- +{
- + STLContext *stl = s->priv_data;
- + return ff_subtitles_queue_seek(&stl->q, s, stream_index,
- + min_ts, ts, max_ts, flags);
- +}
- +
- +static int stl_read_close(AVFormatContext *s)
- +{
- + STLContext *stl = s->priv_data;
- + ff_subtitles_queue_clean(&stl->q);
- + return 0;
- +}
- +
- +AVInputFormat ff_stl_demuxer = {
- + .name = "stl",
- + .long_name = NULL_IF_CONFIG_SMALL("STL subtitles"),
- + .priv_data_size = sizeof(STLContext),
- + .read_probe = stl_probe,
- + .read_header = stl_read_header,
- + .read_packet = stl_read_packet,
- + .read_seek2 = stl_read_seek,
- + .read_close = stl_read_close,
- + .extensions = "stl",
- +};
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement