1/*
2 * Copyright (c) 2012 Cl��ment B��sch
3 *
4 * This file is part of FFmpeg.
5 *
6 * FFmpeg is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
10 *
11 * FFmpeg is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14 * Lesser General Public License for more details.
15 *
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with FFmpeg; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19 */
20
21/**
22 * @file
23 * VPlayer subtitles format demuxer
24 */
25
26#include "avformat.h"
27#include "internal.h"
28#include "subtitles.h"
29
30typedef struct {
31    FFDemuxSubtitlesQueue q;
32} VPlayerContext;
33
34static int vplayer_probe(AVProbeData *p)
35{
36    char c;
37    const unsigned char *ptr = p->buf;
38
39    if (sscanf(ptr, "%*d:%*d:%*d.%*d%c", &c) == 1 && strchr(": =", c))
40        return AVPROBE_SCORE_MAX;
41    return 0;
42}
43
44static int64_t read_ts(char **line)
45{
46    char c;
47    int hh, mm, ss, ms, len;
48
49    if (sscanf(*line, "%d:%d:%d.%d%c%n",
50               &hh, &mm, &ss, &ms, &c, &len) >= 5) {
51        *line += len;
52        return (hh*3600LL + mm*60LL + ss) * 100LL + ms;
53    }
54    return AV_NOPTS_VALUE;
55}
56
57static int vplayer_read_header(AVFormatContext *s)
58{
59    VPlayerContext *vplayer = s->priv_data;
60    AVStream *st = avformat_new_stream(s, NULL);
61
62    if (!st)
63        return AVERROR(ENOMEM);
64    avpriv_set_pts_info(st, 64, 1, 100);
65    st->codec->codec_type = AVMEDIA_TYPE_SUBTITLE;
66    st->codec->codec_id   = AV_CODEC_ID_VPLAYER;
67
68    while (!url_feof(s->pb)) {
69        char line[4096];
70        char *p = line;
71        const int64_t pos = avio_tell(s->pb);
72        int len = ff_get_line(s->pb, line, sizeof(line));
73        int64_t pts_start;
74
75        if (!len)
76            break;
77
78        line[strcspn(line, "\r\n")] = 0;
79
80        pts_start = read_ts(&p);
81        if (pts_start != AV_NOPTS_VALUE) {
82            AVPacket *sub;
83
84            sub = ff_subtitles_queue_insert(&vplayer->q, p, strlen(p), 0);
85            if (!sub)
86                return AVERROR(ENOMEM);
87            sub->pos = pos;
88            sub->pts = pts_start;
89            sub->duration = -1;
90        }
91    }
92
93    ff_subtitles_queue_finalize(&vplayer->q);
94    return 0;
95}
96
97static int vplayer_read_packet(AVFormatContext *s, AVPacket *pkt)
98{
99    VPlayerContext *vplayer = s->priv_data;
100    return ff_subtitles_queue_read_packet(&vplayer->q, pkt);
101}
102
103static int vplayer_read_seek(AVFormatContext *s, int stream_index,
104                             int64_t min_ts, int64_t ts, int64_t max_ts, int flags)
105{
106    VPlayerContext *vplayer = s->priv_data;
107    return ff_subtitles_queue_seek(&vplayer->q, s, stream_index,
108                                   min_ts, ts, max_ts, flags);
109}
110
111static int vplayer_read_close(AVFormatContext *s)
112{
113    VPlayerContext *vplayer = s->priv_data;
114    ff_subtitles_queue_clean(&vplayer->q);
115    return 0;
116}
117
118AVInputFormat ff_vplayer_demuxer = {
119    .name           = "vplayer",
120    .long_name      = NULL_IF_CONFIG_SMALL("VPlayer subtitles"),
121    .priv_data_size = sizeof(VPlayerContext),
122    .read_probe     = vplayer_probe,
123    .read_header    = vplayer_read_header,
124    .read_packet    = vplayer_read_packet,
125    .read_seek2     = vplayer_read_seek,
126    .read_close     = vplayer_read_close,
127    .extensions     = "txt",
128};
129