FFmpeg coverage


Directory: ../../../ffmpeg/
File: src/libavformat/microdvddec.c
Date: 2022-12-09 07:38:14
Exec Total Coverage
Lines: 73 85 85.9%
Functions: 6 7 85.7%
Branches: 40 56 71.4%

Line Branch Exec Source
1 /*
2 * MicroDVD subtitle demuxer
3 * Copyright (c) 2010 Aurelien Jacobs <aurel@gnuage.org>
4 * Copyright (c) 2012 Clément Bœsch <u pkh me>
5 *
6 * This file is part of FFmpeg.
7 *
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.
12 *
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.
17 *
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
21 */
22
23 #include "avformat.h"
24 #include "internal.h"
25 #include "subtitles.h"
26 #include "libavutil/intreadwrite.h"
27 #include "libavutil/opt.h"
28
29 #define MAX_LINESIZE 2048
30
31
32 typedef struct {
33 const AVClass *class;
34 FFDemuxSubtitlesQueue q;
35 AVRational frame_rate;
36 } MicroDVDContext;
37
38
39 6782 static int microdvd_probe(const AVProbeData *p)
40 {
41 unsigned char c;
42 6782 const uint8_t *ptr = p->buf;
43 int i;
44
45
2/2
✓ Branch 0 taken 9 times.
✓ Branch 1 taken 6773 times.
6782 if (AV_RB24(ptr) == 0xEFBBBF)
46 9 ptr += 3; /* skip UTF-8 BOM */
47
48
2/2
✓ Branch 0 taken 6788 times.
✓ Branch 1 taken 3 times.
6791 for (i=0; i<3; i++) {
49
2/2
✓ Branch 0 taken 6784 times.
✓ Branch 1 taken 4 times.
6788 if (sscanf(ptr, "{%*d}{}%c", &c) != 1 &&
50
2/2
✓ Branch 0 taken 6781 times.
✓ Branch 1 taken 3 times.
6784 sscanf(ptr, "{%*d}{%*d}%c", &c) != 1 &&
51
2/2
✓ Branch 0 taken 6779 times.
✓ Branch 1 taken 2 times.
6781 sscanf(ptr, "{DEFAULT}{}%c", &c) != 1)
52 6779 return 0;
53 9 ptr += ff_subtitles_next_line(ptr);
54 }
55 3 return AVPROBE_SCORE_MAX;
56 }
57
58 70 static int64_t get_pts(const char *buf)
59 {
60 int frame;
61 char c;
62
63
1/2
✓ Branch 0 taken 70 times.
✗ Branch 1 not taken.
70 if (sscanf(buf, "{%d}{%c", &frame, &c) == 2)
64 70 return frame;
65 return AV_NOPTS_VALUE;
66 }
67
68 70 static int64_t get_duration(const char *buf)
69 {
70 int frame_start, frame_end;
71
72
2/2
✓ Branch 0 taken 60 times.
✓ Branch 1 taken 10 times.
70 if (sscanf(buf, "{%d}{%d}", &frame_start, &frame_end) == 2)
73 60 return frame_end - (int64_t)frame_start;
74 10 return -1;
75 }
76
77 static const char *bom = "\xEF\xBB\xBF";
78
79 3 static int microdvd_read_header(AVFormatContext *s)
80 {
81 3 AVRational pts_info = (AVRational){ 2997, 125 }; /* default: 23.976 fps */
82 3 MicroDVDContext *microdvd = s->priv_data;
83 3 AVStream *st = avformat_new_stream(s, NULL);
84 3 int i = 0, ret;
85 char line_buf[MAX_LINESIZE];
86 3 int has_real_fps = 0;
87
88
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3 times.
3 if (!st)
89 return AVERROR(ENOMEM);
90
91
1/2
✓ Branch 1 taken 77 times.
✗ Branch 2 not taken.
77 while (!avio_feof(s->pb)) {
92 char *p;
93 AVPacket *sub;
94 77 int64_t pos = avio_tell(s->pb);
95 77 int len = ff_get_line(s->pb, line_buf, sizeof(line_buf));
96 77 char *line = line_buf;
97 int64_t pts;
98
99
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 77 times.
77 if (!strncmp(line, bom, 3))
100 line += 3;
101 77 p = line;
102
103
2/2
✓ Branch 0 taken 3 times.
✓ Branch 1 taken 74 times.
77 if (!len)
104 3 break;
105 74 line[strcspn(line, "\r\n")] = 0;
106
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 74 times.
74 if (!*p)
107 continue;
108
2/2
✓ Branch 0 taken 9 times.
✓ Branch 1 taken 65 times.
74 if (i++ < 3) {
109 int frame;
110 double fps;
111 char c;
112
113
2/2
✓ Branch 0 taken 7 times.
✓ Branch 1 taken 2 times.
9 if ((sscanf(line, "{%d}{}%6lf", &frame, &fps) == 2 ||
114
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 7 times.
7 sscanf(line, "{%d}{%*d}%6lf", &frame, &fps) == 2)
115
3/6
✓ Branch 0 taken 2 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 2 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 2 times.
✗ Branch 5 not taken.
2 && frame <= 1 && fps > 3 && fps < 100) {
116 2 pts_info = av_d2q(fps, 100000);
117 2 has_real_fps = 1;
118 4 continue;
119 }
120
4/4
✓ Branch 0 taken 5 times.
✓ Branch 1 taken 2 times.
✓ Branch 2 taken 2 times.
✓ Branch 3 taken 3 times.
7 if (!st->codecpar->extradata && sscanf(line, "{DEFAULT}{}%c", &c) == 1) {
121 2 int size = strlen(line + 11);
122 2 ret = ff_alloc_extradata(st->codecpar, size);
123
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
2 if (ret < 0)
124 return ret;
125 2 memcpy(st->codecpar->extradata, line + 11, size);
126 2 continue;
127 }
128 }
129 #define SKIP_FRAME_ID \
130 p = strchr(p, '}'); \
131 if (!p) { \
132 av_log(s, AV_LOG_WARNING, "Invalid event \"%s\"" \
133 " at line %d\n", line, i); \
134 continue; \
135 } \
136 p++
137
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 70 times.
70 SKIP_FRAME_ID;
138
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 70 times.
70 SKIP_FRAME_ID;
139
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 70 times.
70 if (!*p)
140 continue;
141 70 pts = get_pts(line);
142
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 70 times.
70 if (pts == AV_NOPTS_VALUE)
143 continue;
144 70 sub = ff_subtitles_queue_insert(&microdvd->q, p, strlen(p), 0);
145
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 70 times.
70 if (!sub)
146 return AVERROR(ENOMEM);
147 70 sub->pos = pos;
148 70 sub->pts = pts;
149 70 sub->duration = get_duration(line);
150 }
151 3 ff_subtitles_queue_finalize(s, &microdvd->q);
152
2/2
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 1 times.
3 if (has_real_fps) {
153 /* export the FPS info only if set in the file */
154 2 microdvd->frame_rate = pts_info;
155
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1 times.
1 } else if (microdvd->frame_rate.num) {
156 /* fallback on user specified frame rate */
157 pts_info = microdvd->frame_rate;
158 }
159 3 avpriv_set_pts_info(st, 64, pts_info.den, pts_info.num);
160 3 st->codecpar->codec_type = AVMEDIA_TYPE_SUBTITLE;
161 3 st->codecpar->codec_id = AV_CODEC_ID_MICRODVD;
162 3 return 0;
163 }
164
165 73 static int microdvd_read_packet(AVFormatContext *s, AVPacket *pkt)
166 {
167 73 MicroDVDContext *microdvd = s->priv_data;
168 73 return ff_subtitles_queue_read_packet(&microdvd->q, pkt);
169 }
170
171 static int microdvd_read_seek(AVFormatContext *s, int stream_index,
172 int64_t min_ts, int64_t ts, int64_t max_ts, int flags)
173 {
174 MicroDVDContext *microdvd = s->priv_data;
175 return ff_subtitles_queue_seek(&microdvd->q, s, stream_index,
176 min_ts, ts, max_ts, flags);
177 }
178
179 3 static int microdvd_read_close(AVFormatContext *s)
180 {
181 3 MicroDVDContext *microdvd = s->priv_data;
182 3 ff_subtitles_queue_clean(&microdvd->q);
183 3 return 0;
184 }
185
186
187 #define OFFSET(x) offsetof(MicroDVDContext, x)
188 #define SD AV_OPT_FLAG_SUBTITLE_PARAM|AV_OPT_FLAG_DECODING_PARAM
189 static const AVOption microdvd_options[] = {
190 { "subfps", "set the movie frame rate fallback", OFFSET(frame_rate), AV_OPT_TYPE_RATIONAL, {.dbl=0}, 0, INT_MAX, SD },
191 { NULL }
192 };
193
194 static const AVClass microdvd_class = {
195 .class_name = "microdvddec",
196 .item_name = av_default_item_name,
197 .option = microdvd_options,
198 .version = LIBAVUTIL_VERSION_INT,
199 };
200
201 const AVInputFormat ff_microdvd_demuxer = {
202 .name = "microdvd",
203 .long_name = NULL_IF_CONFIG_SMALL("MicroDVD subtitle format"),
204 .priv_data_size = sizeof(MicroDVDContext),
205 .flags_internal = FF_FMT_INIT_CLEANUP,
206 .read_probe = microdvd_probe,
207 .read_header = microdvd_read_header,
208 .read_packet = microdvd_read_packet,
209 .read_seek2 = microdvd_read_seek,
210 .read_close = microdvd_read_close,
211 .priv_class = &microdvd_class,
212 };
213