2 * MicroDVD subtitle demuxer
3 * Copyright (c) 2010 Aurelien Jacobs <aurel@gnuage.org>
4 * Copyright (c) 2012 Clément Bœsch <u pkh me>
6 * This file is part of FFmpeg.
8 * FFmpeg is free software; you can redistribute it and/or
9 * modify it under the terms of the GNU Lesser General Public
10 * License as published by the Free Software Foundation; either
11 * version 2.1 of the License, or (at your option) any later version.
13 * FFmpeg is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 * Lesser General Public License for more details.
18 * You should have received a copy of the GNU Lesser General Public
19 * License along with FFmpeg; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
25 #include "subtitles.h"
26 #include "libavutil/intreadwrite.h"
27 #include "libavutil/opt.h"
29 #define MAX_LINESIZE 2048
34 FFDemuxSubtitlesQueue q
;
35 AVRational frame_rate
;
39 static int microdvd_probe(AVProbeData
*p
)
42 const uint8_t *ptr
= p
->buf
;
45 if (AV_RB24(ptr
) == 0xEFBBBF)
46 ptr
+= 3; /* skip UTF-8 BOM */
49 if (sscanf(ptr
, "{%*d}{}%c", &c
) != 1 &&
50 sscanf(ptr
, "{%*d}{%*d}%c", &c
) != 1 &&
51 sscanf(ptr
, "{DEFAULT}{}%c", &c
) != 1)
53 ptr
+= ff_subtitles_next_line(ptr
);
55 return AVPROBE_SCORE_MAX
;
58 static int64_t get_pts(const char *buf
)
63 if (sscanf(buf
, "{%d}{%c", &frame
, &c
) == 2)
65 return AV_NOPTS_VALUE
;
68 static int get_duration(const char *buf
)
70 int frame_start
, frame_end
;
72 if (sscanf(buf
, "{%d}{%d}", &frame_start
, &frame_end
) == 2)
73 return frame_end
- frame_start
;
77 static const char *bom
= "\xEF\xBB\xBF";
79 static int microdvd_read_header(AVFormatContext
*s
)
81 AVRational pts_info
= (AVRational
){ 2997, 125 }; /* default: 23.976 fps */
82 MicroDVDContext
*microdvd
= s
->priv_data
;
83 AVStream
*st
= avformat_new_stream(s
, NULL
);
85 char line_buf
[MAX_LINESIZE
];
89 return AVERROR(ENOMEM
);
91 while (!avio_feof(s
->pb
)) {
94 int64_t pos
= avio_tell(s
->pb
);
95 int len
= ff_get_line(s
->pb
, line_buf
, sizeof(line_buf
));
96 char *line
= line_buf
;
98 if (!strncmp(line
, bom
, 3))
104 line
[strcspn(line
, "\r\n")] = 0;
110 if ((sscanf(line
, "{%d}{}%6lf", &frame
, &fps
) == 2 ||
111 sscanf(line
, "{%d}{%*d}%6lf", &frame
, &fps
) == 2)
112 && frame
<= 1 && fps
> 3 && fps
< 100) {
113 pts_info
= av_d2q(fps
, 100000);
116 if (!st
->codec
->extradata
&& sscanf(line
, "{DEFAULT}{}%c", &c
) == 1) {
117 st
->codec
->extradata
= av_strdup(line
+ 11);
118 if (!st
->codec
->extradata
)
119 return AVERROR(ENOMEM
);
120 st
->codec
->extradata_size
= strlen(st
->codec
->extradata
) + 1;
124 #define SKIP_FRAME_ID \
125 p = strchr(p, '}'); \
127 av_log(s, AV_LOG_WARNING, "Invalid event \"%s\"" \
128 " at line %d\n", line, i); \
136 sub
= ff_subtitles_queue_insert(µdvd
->q
, p
, strlen(p
), 0);
138 return AVERROR(ENOMEM
);
140 sub
->pts
= get_pts(line
);
141 sub
->duration
= get_duration(line
);
143 ff_subtitles_queue_finalize(µdvd
->q
);
145 /* export the FPS info only if set in the file */
146 microdvd
->frame_rate
= pts_info
;
147 } else if (microdvd
->frame_rate
.num
) {
148 /* fallback on user specified frame rate */
149 pts_info
= microdvd
->frame_rate
;
151 avpriv_set_pts_info(st
, 64, pts_info
.den
, pts_info
.num
);
152 st
->codec
->codec_type
= AVMEDIA_TYPE_SUBTITLE
;
153 st
->codec
->codec_id
= AV_CODEC_ID_MICRODVD
;
157 static int microdvd_read_packet(AVFormatContext
*s
, AVPacket
*pkt
)
159 MicroDVDContext
*microdvd
= s
->priv_data
;
160 return ff_subtitles_queue_read_packet(µdvd
->q
, pkt
);
163 static int microdvd_read_seek(AVFormatContext
*s
, int stream_index
,
164 int64_t min_ts
, int64_t ts
, int64_t max_ts
, int flags
)
166 MicroDVDContext
*microdvd
= s
->priv_data
;
167 return ff_subtitles_queue_seek(µdvd
->q
, s
, stream_index
,
168 min_ts
, ts
, max_ts
, flags
);
171 static int microdvd_read_close(AVFormatContext
*s
)
173 MicroDVDContext
*microdvd
= s
->priv_data
;
174 ff_subtitles_queue_clean(µdvd
->q
);
179 #define OFFSET(x) offsetof(MicroDVDContext, x)
180 #define SD AV_OPT_FLAG_SUBTITLE_PARAM|AV_OPT_FLAG_DECODING_PARAM
181 static const AVOption microdvd_options
[] = {
182 { "subfps", "set the movie frame rate fallback", OFFSET(frame_rate
), AV_OPT_TYPE_RATIONAL
, {.dbl
=0}, 0, INT_MAX
, SD
},
186 static const AVClass microdvd_class
= {
187 .class_name
= "microdvddec",
188 .item_name
= av_default_item_name
,
189 .option
= microdvd_options
,
190 .version
= LIBAVUTIL_VERSION_INT
,
193 AVInputFormat ff_microdvd_demuxer
= {
195 .long_name
= NULL_IF_CONFIG_SMALL("MicroDVD subtitle format"),
196 .priv_data_size
= sizeof(MicroDVDContext
),
197 .read_probe
= microdvd_probe
,
198 .read_header
= microdvd_read_header
,
199 .read_packet
= microdvd_read_packet
,
200 .read_seek2
= microdvd_read_seek
,
201 .read_close
= microdvd_read_close
,
202 .priv_class
= µdvd_class
,