FFmpeg coverage


Directory: ../../../ffmpeg/
File: src/libavcodec/dvbsubdec.c
Date: 2024-04-23 16:28:37
Exec Total Coverage
Lines: 475 838 56.7%
Functions: 19 23 82.6%
Branches: 267 587 45.5%

Line Branch Exec Source
1 /*
2 * DVB subtitle decoding
3 * Copyright (c) 2005 Ian Caulfield
4 *
5 * This file is part of FFmpeg.
6 *
7 * FFmpeg is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
11 *
12 * FFmpeg is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
16 *
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with FFmpeg; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20 */
21
22 #include "avcodec.h"
23 #include "get_bits.h"
24 #include "bytestream.h"
25 #include "codec_internal.h"
26 #include "decode.h"
27 #include "libavutil/colorspace.h"
28 #include "libavutil/imgutils.h"
29 #include "libavutil/mem.h"
30 #include "libavutil/opt.h"
31 #include "libavutil/thread.h"
32
33 #define DVBSUB_PAGE_SEGMENT 0x10
34 #define DVBSUB_REGION_SEGMENT 0x11
35 #define DVBSUB_CLUT_SEGMENT 0x12
36 #define DVBSUB_OBJECT_SEGMENT 0x13
37 #define DVBSUB_DISPLAYDEFINITION_SEGMENT 0x14
38 #define DVBSUB_DISPLAY_SEGMENT 0x80
39
40 #define cm (ff_crop_tab + MAX_NEG_CROP)
41
42 #define RGBA(r,g,b,a) (((unsigned)(a) << 24) | ((r) << 16) | ((g) << 8) | (b))
43
44 typedef struct DVBSubCLUT {
45 int id;
46 int version;
47
48 uint32_t clut4[4];
49 uint32_t clut16[16];
50 uint32_t clut256[256];
51
52 struct DVBSubCLUT *next;
53 } DVBSubCLUT;
54
55 static DVBSubCLUT default_clut;
56
57 typedef struct DVBSubObjectDisplay {
58 int object_id;
59 int region_id;
60
61 int x_pos;
62 int y_pos;
63
64 int fgcolor;
65 int bgcolor;
66
67 struct DVBSubObjectDisplay *region_list_next;
68 struct DVBSubObjectDisplay *object_list_next;
69 } DVBSubObjectDisplay;
70
71 typedef struct DVBSubObject {
72 int id;
73 int version;
74
75 int type;
76
77 DVBSubObjectDisplay *display_list;
78
79 struct DVBSubObject *next;
80 } DVBSubObject;
81
82 typedef struct DVBSubRegionDisplay {
83 int region_id;
84
85 int x_pos;
86 int y_pos;
87
88 struct DVBSubRegionDisplay *next;
89 } DVBSubRegionDisplay;
90
91 typedef struct DVBSubRegion {
92 int id;
93 int version;
94
95 int width;
96 int height;
97 int depth;
98
99 int clut;
100 int bgcolor;
101
102 uint8_t computed_clut[4*256];
103 int has_computed_clut;
104
105 uint8_t *pbuf;
106 int buf_size;
107 int dirty;
108
109 DVBSubObjectDisplay *display_list;
110
111 struct DVBSubRegion *next;
112 } DVBSubRegion;
113
114 typedef struct DVBSubDisplayDefinition {
115 int version;
116
117 int x;
118 int y;
119 int width;
120 int height;
121 } DVBSubDisplayDefinition;
122
123 typedef struct DVBSubContext {
124 AVClass *class;
125 int composition_id;
126 int ancillary_id;
127
128 int version;
129 int time_out;
130 int compute_edt; /**< if 1 end display time calculated using pts
131 if 0 (Default) calculated using time out */
132 int compute_clut;
133 int clut_count2[257][256];
134 int substream;
135 int64_t prev_start;
136 DVBSubRegion *region_list;
137 DVBSubCLUT *clut_list;
138 DVBSubObject *object_list;
139
140 DVBSubRegionDisplay *display_list;
141 DVBSubDisplayDefinition *display_definition;
142 } DVBSubContext;
143
144
145 126 static DVBSubObject* get_object(DVBSubContext *ctx, int object_id)
146 {
147 126 DVBSubObject *ptr = ctx->object_list;
148
149
4/4
✓ Branch 0 taken 120 times.
✓ Branch 1 taken 42 times.
✓ Branch 2 taken 36 times.
✓ Branch 3 taken 84 times.
162 while (ptr && ptr->id != object_id) {
150 36 ptr = ptr->next;
151 }
152
153 126 return ptr;
154 }
155
156 79 static DVBSubCLUT* get_clut(DVBSubContext *ctx, int clut_id)
157 {
158 79 DVBSubCLUT *ptr = ctx->clut_list;
159
160
4/4
✓ Branch 0 taken 102 times.
✓ Branch 1 taken 13 times.
✓ Branch 2 taken 36 times.
✓ Branch 3 taken 66 times.
115 while (ptr && ptr->id != clut_id) {
161 36 ptr = ptr->next;
162 }
163
164 79 return ptr;
165 }
166
167 273 static DVBSubRegion* get_region(DVBSubContext *ctx, int region_id)
168 {
169 273 DVBSubRegion *ptr = ctx->region_list;
170
171
4/4
✓ Branch 0 taken 878 times.
✓ Branch 1 taken 28 times.
✓ Branch 2 taken 633 times.
✓ Branch 3 taken 245 times.
906 while (ptr && ptr->id != region_id) {
172 633 ptr = ptr->next;
173 }
174
175 273 return ptr;
176 }
177
178 85 static void delete_region_display_list(DVBSubContext *ctx, DVBSubRegion *region)
179 {
180 DVBSubObject *object, *obj2, **obj2_ptr;
181 DVBSubObjectDisplay *display, *obj_disp, **obj_disp_ptr;
182
183
2/2
✓ Branch 0 taken 42 times.
✓ Branch 1 taken 85 times.
127 while (region->display_list) {
184 42 display = region->display_list;
185
186 42 object = get_object(ctx, display->object_id);
187
188
1/2
✓ Branch 0 taken 42 times.
✗ Branch 1 not taken.
42 if (object) {
189 42 obj_disp_ptr = &object->display_list;
190 42 obj_disp = *obj_disp_ptr;
191
192
2/4
✓ Branch 0 taken 42 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 42 times.
42 while (obj_disp && obj_disp != display) {
193 obj_disp_ptr = &obj_disp->object_list_next;
194 obj_disp = *obj_disp_ptr;
195 }
196
197
1/2
✓ Branch 0 taken 42 times.
✗ Branch 1 not taken.
42 if (obj_disp) {
198 42 *obj_disp_ptr = obj_disp->object_list_next;
199
200
1/2
✓ Branch 0 taken 42 times.
✗ Branch 1 not taken.
42 if (!object->display_list) {
201 42 obj2_ptr = &ctx->object_list;
202 42 obj2 = *obj2_ptr;
203
204
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 42 times.
42 while (obj2 != object) {
205 av_assert0(obj2);
206 obj2_ptr = &obj2->next;
207 obj2 = *obj2_ptr;
208 }
209
210 42 *obj2_ptr = obj2->next;
211
212 42 av_freep(&obj2);
213 }
214 }
215 }
216
217 42 region->display_list = display->region_list_next;
218
219 42 av_freep(&display);
220 }
221
222 85 }
223
224 15 static void delete_cluts(DVBSubContext *ctx)
225 {
226
2/2
✓ Branch 0 taken 13 times.
✓ Branch 1 taken 15 times.
28 while (ctx->clut_list) {
227 13 DVBSubCLUT *clut = ctx->clut_list;
228
229 13 ctx->clut_list = clut->next;
230
231 13 av_freep(&clut);
232 }
233 15 }
234
235 15 static void delete_objects(DVBSubContext *ctx)
236 {
237
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 15 times.
15 while (ctx->object_list) {
238 DVBSubObject *object = ctx->object_list;
239
240 ctx->object_list = object->next;
241
242 av_freep(&object);
243 }
244 15 }
245
246 15 static void delete_regions(DVBSubContext *ctx)
247 {
248
2/2
✓ Branch 0 taken 28 times.
✓ Branch 1 taken 15 times.
43 while (ctx->region_list) {
249 28 DVBSubRegion *region = ctx->region_list;
250
251 28 ctx->region_list = region->next;
252
253 28 delete_region_display_list(ctx, region);
254
255 28 av_freep(&region->pbuf);
256 28 av_freep(&region);
257 }
258 15 }
259
260 4 static av_cold void init_default_clut(void)
261 {
262 4 int i, r, g, b, a = 0;
263
264 4 default_clut.id = -1;
265 4 default_clut.next = NULL;
266
267 4 default_clut.clut4[0] = RGBA( 0, 0, 0, 0);
268 4 default_clut.clut4[1] = RGBA(255, 255, 255, 255);
269 4 default_clut.clut4[2] = RGBA( 0, 0, 0, 255);
270 4 default_clut.clut4[3] = RGBA(127, 127, 127, 255);
271
272 4 default_clut.clut16[0] = RGBA( 0, 0, 0, 0);
273
2/2
✓ Branch 0 taken 60 times.
✓ Branch 1 taken 4 times.
64 for (i = 1; i < 16; i++) {
274
2/2
✓ Branch 0 taken 28 times.
✓ Branch 1 taken 32 times.
60 if (i < 8) {
275
2/2
✓ Branch 0 taken 16 times.
✓ Branch 1 taken 12 times.
28 r = (i & 1) ? 255 : 0;
276
2/2
✓ Branch 0 taken 16 times.
✓ Branch 1 taken 12 times.
28 g = (i & 2) ? 255 : 0;
277
2/2
✓ Branch 0 taken 16 times.
✓ Branch 1 taken 12 times.
28 b = (i & 4) ? 255 : 0;
278 } else {
279
2/2
✓ Branch 0 taken 16 times.
✓ Branch 1 taken 16 times.
32 r = (i & 1) ? 127 : 0;
280
2/2
✓ Branch 0 taken 16 times.
✓ Branch 1 taken 16 times.
32 g = (i & 2) ? 127 : 0;
281
2/2
✓ Branch 0 taken 16 times.
✓ Branch 1 taken 16 times.
32 b = (i & 4) ? 127 : 0;
282 }
283 60 default_clut.clut16[i] = RGBA(r, g, b, 255);
284 }
285
286 4 default_clut.clut256[0] = RGBA( 0, 0, 0, 0);
287
2/2
✓ Branch 0 taken 1020 times.
✓ Branch 1 taken 4 times.
1024 for (i = 1; i < 256; i++) {
288
2/2
✓ Branch 0 taken 28 times.
✓ Branch 1 taken 992 times.
1020 if (i < 8) {
289
2/2
✓ Branch 0 taken 16 times.
✓ Branch 1 taken 12 times.
28 r = (i & 1) ? 255 : 0;
290
2/2
✓ Branch 0 taken 16 times.
✓ Branch 1 taken 12 times.
28 g = (i & 2) ? 255 : 0;
291
2/2
✓ Branch 0 taken 16 times.
✓ Branch 1 taken 12 times.
28 b = (i & 4) ? 255 : 0;
292 28 a = 63;
293 } else {
294
4/5
✓ Branch 0 taken 224 times.
✓ Branch 1 taken 256 times.
✓ Branch 2 taken 256 times.
✓ Branch 3 taken 256 times.
✗ Branch 4 not taken.
992 switch (i & 0x88) {
295 224 case 0x00:
296
4/4
✓ Branch 0 taken 112 times.
✓ Branch 1 taken 112 times.
✓ Branch 2 taken 128 times.
✓ Branch 3 taken 96 times.
224 r = ((i & 1) ? 85 : 0) + ((i & 0x10) ? 170 : 0);
297
4/4
✓ Branch 0 taken 112 times.
✓ Branch 1 taken 112 times.
✓ Branch 2 taken 128 times.
✓ Branch 3 taken 96 times.
224 g = ((i & 2) ? 85 : 0) + ((i & 0x20) ? 170 : 0);
298
4/4
✓ Branch 0 taken 112 times.
✓ Branch 1 taken 112 times.
✓ Branch 2 taken 128 times.
✓ Branch 3 taken 96 times.
224 b = ((i & 4) ? 85 : 0) + ((i & 0x40) ? 170 : 0);
299 224 a = 255;
300 224 break;
301 256 case 0x08:
302
4/4
✓ Branch 0 taken 128 times.
✓ Branch 1 taken 128 times.
✓ Branch 2 taken 128 times.
✓ Branch 3 taken 128 times.
256 r = ((i & 1) ? 85 : 0) + ((i & 0x10) ? 170 : 0);
303
4/4
✓ Branch 0 taken 128 times.
✓ Branch 1 taken 128 times.
✓ Branch 2 taken 128 times.
✓ Branch 3 taken 128 times.
256 g = ((i & 2) ? 85 : 0) + ((i & 0x20) ? 170 : 0);
304
4/4
✓ Branch 0 taken 128 times.
✓ Branch 1 taken 128 times.
✓ Branch 2 taken 128 times.
✓ Branch 3 taken 128 times.
256 b = ((i & 4) ? 85 : 0) + ((i & 0x40) ? 170 : 0);
305 256 a = 127;
306 256 break;
307 256 case 0x80:
308
4/4
✓ Branch 0 taken 128 times.
✓ Branch 1 taken 128 times.
✓ Branch 2 taken 128 times.
✓ Branch 3 taken 128 times.
256 r = 127 + ((i & 1) ? 43 : 0) + ((i & 0x10) ? 85 : 0);
309
4/4
✓ Branch 0 taken 128 times.
✓ Branch 1 taken 128 times.
✓ Branch 2 taken 128 times.
✓ Branch 3 taken 128 times.
256 g = 127 + ((i & 2) ? 43 : 0) + ((i & 0x20) ? 85 : 0);
310
4/4
✓ Branch 0 taken 128 times.
✓ Branch 1 taken 128 times.
✓ Branch 2 taken 128 times.
✓ Branch 3 taken 128 times.
256 b = 127 + ((i & 4) ? 43 : 0) + ((i & 0x40) ? 85 : 0);
311 256 a = 255;
312 256 break;
313 256 case 0x88:
314
4/4
✓ Branch 0 taken 128 times.
✓ Branch 1 taken 128 times.
✓ Branch 2 taken 128 times.
✓ Branch 3 taken 128 times.
256 r = ((i & 1) ? 43 : 0) + ((i & 0x10) ? 85 : 0);
315
4/4
✓ Branch 0 taken 128 times.
✓ Branch 1 taken 128 times.
✓ Branch 2 taken 128 times.
✓ Branch 3 taken 128 times.
256 g = ((i & 2) ? 43 : 0) + ((i & 0x20) ? 85 : 0);
316
4/4
✓ Branch 0 taken 128 times.
✓ Branch 1 taken 128 times.
✓ Branch 2 taken 128 times.
✓ Branch 3 taken 128 times.
256 b = ((i & 4) ? 43 : 0) + ((i & 0x40) ? 85 : 0);
317 256 a = 255;
318 256 break;
319 }
320 }
321 1020 default_clut.clut256[i] = RGBA(r, g, b, a);
322 }
323 4 }
324
325 8 static av_cold int dvbsub_init_decoder(AVCodecContext *avctx)
326 {
327 static AVOnce init_static_once = AV_ONCE_INIT;
328 8 DVBSubContext *ctx = avctx->priv_data;
329
330
1/2
✓ Branch 0 taken 8 times.
✗ Branch 1 not taken.
8 if (ctx->substream < 0) {
331 8 ctx->composition_id = -1;
332 8 ctx->ancillary_id = -1;
333 } else if (!avctx->extradata || (avctx->extradata_size < 4) || ((avctx->extradata_size % 5 != 0) && (avctx->extradata_size != 4))) {
334 av_log(avctx, AV_LOG_WARNING, "Invalid DVB subtitles stream extradata!\n");
335 ctx->composition_id = -1;
336 ctx->ancillary_id = -1;
337 } else {
338 if (avctx->extradata_size > 5*ctx->substream + 2) {
339 ctx->composition_id = AV_RB16(avctx->extradata + 5*ctx->substream);
340 ctx->ancillary_id = AV_RB16(avctx->extradata + 5*ctx->substream + 2);
341 } else {
342 av_log(avctx, AV_LOG_WARNING, "Selected DVB subtitles sub-stream %d is not available\n", ctx->substream);
343 ctx->composition_id = AV_RB16(avctx->extradata);
344 ctx->ancillary_id = AV_RB16(avctx->extradata + 2);
345 }
346 }
347
348 8 ctx->version = -1;
349 8 ctx->prev_start = AV_NOPTS_VALUE;
350
351 8 ff_thread_once(&init_static_once, init_default_clut);
352
353 8 return 0;
354 }
355
356 8 static av_cold int dvbsub_close_decoder(AVCodecContext *avctx)
357 {
358 8 DVBSubContext *ctx = avctx->priv_data;
359 DVBSubRegionDisplay *display;
360
361 8 delete_regions(ctx);
362
363 8 delete_objects(ctx);
364
365 8 delete_cluts(ctx);
366
367 8 av_freep(&ctx->display_definition);
368
369
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 8 times.
8 while (ctx->display_list) {
370 display = ctx->display_list;
371 ctx->display_list = display->next;
372
373 av_freep(&display);
374 }
375
376 8 return 0;
377 }
378
379 static int dvbsub_read_2bit_string(AVCodecContext *avctx,
380 uint8_t *destbuf, int dbuf_len,
381 const uint8_t **srcbuf, int buf_size,
382 int non_mod, uint8_t *map_table, int x_pos)
383 {
384 GetBitContext gb;
385
386 int bits;
387 int run_length;
388 int pixels_read = x_pos;
389
390 init_get_bits(&gb, *srcbuf, buf_size << 3);
391
392 destbuf += x_pos;
393
394 while (get_bits_count(&gb) < buf_size << 3 && pixels_read < dbuf_len) {
395 bits = get_bits(&gb, 2);
396
397 if (bits) {
398 if (non_mod != 1 || bits != 1) {
399 if (map_table)
400 *destbuf++ = map_table[bits];
401 else
402 *destbuf++ = bits;
403 }
404 pixels_read++;
405 } else {
406 bits = get_bits1(&gb);
407 if (bits == 1) {
408 run_length = get_bits(&gb, 3) + 3;
409 bits = get_bits(&gb, 2);
410
411 if (non_mod == 1 && bits == 1)
412 pixels_read += run_length;
413 else {
414 if (map_table)
415 bits = map_table[bits];
416 while (run_length-- > 0 && pixels_read < dbuf_len) {
417 *destbuf++ = bits;
418 pixels_read++;
419 }
420 }
421 } else {
422 bits = get_bits1(&gb);
423 if (bits == 0) {
424 bits = get_bits(&gb, 2);
425 if (bits == 2) {
426 run_length = get_bits(&gb, 4) + 12;
427 bits = get_bits(&gb, 2);
428
429 if (non_mod == 1 && bits == 1)
430 pixels_read += run_length;
431 else {
432 if (map_table)
433 bits = map_table[bits];
434 while (run_length-- > 0 && pixels_read < dbuf_len) {
435 *destbuf++ = bits;
436 pixels_read++;
437 }
438 }
439 } else if (bits == 3) {
440 run_length = get_bits(&gb, 8) + 29;
441 bits = get_bits(&gb, 2);
442
443 if (non_mod == 1 && bits == 1)
444 pixels_read += run_length;
445 else {
446 if (map_table)
447 bits = map_table[bits];
448 while (run_length-- > 0 && pixels_read < dbuf_len) {
449 *destbuf++ = bits;
450 pixels_read++;
451 }
452 }
453 } else if (bits == 1) {
454 if (map_table)
455 bits = map_table[0];
456 else
457 bits = 0;
458 run_length = 2;
459 while (run_length-- > 0 && pixels_read < dbuf_len) {
460 *destbuf++ = bits;
461 pixels_read++;
462 }
463 } else {
464 (*srcbuf) += (get_bits_count(&gb) + 7) >> 3;
465 return pixels_read;
466 }
467 } else {
468 if (map_table)
469 bits = map_table[0];
470 else
471 bits = 0;
472 *destbuf++ = bits;
473 pixels_read++;
474 }
475 }
476 }
477 }
478
479 if (get_bits(&gb, 6))
480 av_log(avctx, AV_LOG_ERROR, "line overflow\n");
481
482 (*srcbuf) += (get_bits_count(&gb) + 7) >> 3;
483
484 return pixels_read;
485 }
486
487 1512 static int dvbsub_read_4bit_string(AVCodecContext *avctx, uint8_t *destbuf, int dbuf_len,
488 const uint8_t **srcbuf, int buf_size,
489 int non_mod, uint8_t *map_table, int x_pos)
490 {
491 GetBitContext gb;
492
493 int bits;
494 int run_length;
495 1512 int pixels_read = x_pos;
496
497 1512 init_get_bits(&gb, *srcbuf, buf_size << 3);
498
499 1512 destbuf += x_pos;
500
501
2/4
✓ Branch 1 taken 70104 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 70104 times.
✗ Branch 4 not taken.
70104 while (get_bits_count(&gb) < buf_size << 3 && pixels_read < dbuf_len) {
502 70104 bits = get_bits(&gb, 4);
503
504
2/2
✓ Branch 0 taken 62235 times.
✓ Branch 1 taken 7869 times.
70104 if (bits) {
505
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 62235 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
62235 if (non_mod != 1 || bits != 1) {
506
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 62235 times.
62235 if (map_table)
507 *destbuf++ = map_table[bits];
508 else
509 62235 *destbuf++ = bits;
510 }
511 62235 pixels_read++;
512 } else {
513 7869 bits = get_bits1(&gb);
514
2/2
✓ Branch 0 taken 1512 times.
✓ Branch 1 taken 6357 times.
7869 if (bits == 0) {
515 1512 run_length = get_bits(&gb, 3);
516
517
1/2
✓ Branch 0 taken 1512 times.
✗ Branch 1 not taken.
1512 if (run_length == 0) {
518 1512 (*srcbuf) += (get_bits_count(&gb) + 7) >> 3;
519 1512 return pixels_read;
520 }
521
522 run_length += 2;
523
524 if (map_table)
525 bits = map_table[0];
526 else
527 bits = 0;
528
529 while (run_length-- > 0 && pixels_read < dbuf_len) {
530 *destbuf++ = bits;
531 pixels_read++;
532 }
533 } else {
534 6357 bits = get_bits1(&gb);
535
2/2
✓ Branch 0 taken 3538 times.
✓ Branch 1 taken 2819 times.
6357 if (bits == 0) {
536 3538 run_length = get_bits(&gb, 2) + 4;
537 3538 bits = get_bits(&gb, 4);
538
539
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 3538 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
3538 if (non_mod == 1 && bits == 1)
540 pixels_read += run_length;
541 else {
542
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 3538 times.
3538 if (map_table)
543 bits = map_table[bits];
544
3/4
✓ Branch 0 taken 17337 times.
✓ Branch 1 taken 3538 times.
✓ Branch 2 taken 17337 times.
✗ Branch 3 not taken.
20875 while (run_length-- > 0 && pixels_read < dbuf_len) {
545 17337 *destbuf++ = bits;
546 17337 pixels_read++;
547 }
548 }
549 } else {
550 2819 bits = get_bits(&gb, 2);
551
2/2
✓ Branch 0 taken 1741 times.
✓ Branch 1 taken 1078 times.
2819 if (bits == 2) {
552 1741 run_length = get_bits(&gb, 4) + 9;
553 1741 bits = get_bits(&gb, 4);
554
555
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 1741 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
1741 if (non_mod == 1 && bits == 1)
556 pixels_read += run_length;
557 else {
558
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1741 times.
1741 if (map_table)
559 bits = map_table[bits];
560
3/4
✓ Branch 0 taken 21366 times.
✓ Branch 1 taken 1741 times.
✓ Branch 2 taken 21366 times.
✗ Branch 3 not taken.
23107 while (run_length-- > 0 && pixels_read < dbuf_len) {
561 21366 *destbuf++ = bits;
562 21366 pixels_read++;
563 }
564 }
565
1/2
✓ Branch 0 taken 1078 times.
✗ Branch 1 not taken.
1078 } else if (bits == 3) {
566 1078 run_length = get_bits(&gb, 8) + 25;
567 1078 bits = get_bits(&gb, 4);
568
569
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 1078 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
1078 if (non_mod == 1 && bits == 1)
570 pixels_read += run_length;
571 else {
572
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1078 times.
1078 if (map_table)
573 bits = map_table[bits];
574
3/4
✓ Branch 0 taken 90942 times.
✓ Branch 1 taken 1078 times.
✓ Branch 2 taken 90942 times.
✗ Branch 3 not taken.
92020 while (run_length-- > 0 && pixels_read < dbuf_len) {
575 90942 *destbuf++ = bits;
576 90942 pixels_read++;
577 }
578 }
579 } else if (bits == 1) {
580 if (map_table)
581 bits = map_table[0];
582 else
583 bits = 0;
584 run_length = 2;
585 while (run_length-- > 0 && pixels_read < dbuf_len) {
586 *destbuf++ = bits;
587 pixels_read++;
588 }
589 } else {
590 if (map_table)
591 bits = map_table[0];
592 else
593 bits = 0;
594 *destbuf++ = bits;
595 pixels_read ++;
596 }
597 }
598 }
599 }
600 }
601
602 if (get_bits(&gb, 8))
603 av_log(avctx, AV_LOG_ERROR, "line overflow\n");
604
605 (*srcbuf) += (get_bits_count(&gb) + 7) >> 3;
606
607 return pixels_read;
608 }
609
610 static int dvbsub_read_8bit_string(AVCodecContext *avctx,
611 uint8_t *destbuf, int dbuf_len,
612 const uint8_t **srcbuf, int buf_size,
613 int non_mod, uint8_t *map_table, int x_pos)
614 {
615 const uint8_t *sbuf_end = (*srcbuf) + buf_size;
616 int bits;
617 int run_length;
618 int pixels_read = x_pos;
619
620 destbuf += x_pos;
621
622 while (*srcbuf < sbuf_end && pixels_read < dbuf_len) {
623 bits = *(*srcbuf)++;
624
625 if (bits) {
626 if (non_mod != 1 || bits != 1) {
627 if (map_table)
628 *destbuf++ = map_table[bits];
629 else
630 *destbuf++ = bits;
631 }
632 pixels_read++;
633 } else {
634 bits = *(*srcbuf)++;
635 run_length = bits & 0x7f;
636 if ((bits & 0x80) == 0) {
637 if (run_length == 0) {
638 return pixels_read;
639 }
640
641 bits = 0;
642 } else {
643 bits = *(*srcbuf)++;
644 }
645 if (non_mod == 1 && bits == 1)
646 pixels_read += run_length;
647 else {
648 if (map_table)
649 bits = map_table[bits];
650 while (run_length-- > 0 && pixels_read < dbuf_len) {
651 *destbuf++ = bits;
652 pixels_read++;
653 }
654 }
655 }
656 }
657
658 if (*(*srcbuf)++)
659 av_log(avctx, AV_LOG_ERROR, "line overflow\n");
660
661 return pixels_read;
662 }
663
664 static void compute_default_clut(DVBSubContext *ctx, uint8_t *clut, AVSubtitleRect *rect, int w, int h)
665 {
666 uint8_t list[256] = {0};
667 uint8_t list_inv[256];
668 int counttab[256] = {0};
669 int (*counttab2)[256] = ctx->clut_count2;
670 int count, i, x, y;
671 ptrdiff_t stride = rect->linesize[0];
672
673 memset(ctx->clut_count2, 0 , sizeof(ctx->clut_count2));
674
675 #define V(x,y) rect->data[0][(x) + (y)*stride]
676 for (y = 0; y<h; y++) {
677 for (x = 0; x<w; x++) {
678 int v = V(x,y) + 1;
679 int vl = x ? V(x-1,y) + 1 : 0;
680 int vr = x+1<w ? V(x+1,y) + 1 : 0;
681 int vt = y ? V(x,y-1) + 1 : 0;
682 int vb = y+1<h ? V(x,y+1) + 1 : 0;
683 counttab[v-1] += !!((v!=vl) + (v!=vr) + (v!=vt) + (v!=vb));
684 counttab2[vl][v-1] ++;
685 counttab2[vr][v-1] ++;
686 counttab2[vt][v-1] ++;
687 counttab2[vb][v-1] ++;
688 }
689 }
690 #define L(x,y) list[d[(x) + (y)*stride]]
691
692 for (i = 0; i<256; i++) {
693 counttab2[i+1][i] = 0;
694 }
695 for (i = 0; i<256; i++) {
696 int bestscore = 0;
697 int bestv = 0;
698
699 for (x = 0; x < 256; x++) {
700 int scorev = 0;
701 if (list[x])
702 continue;
703 scorev += counttab2[0][x];
704 for (y = 0; y < 256; y++) {
705 scorev += list[y] * counttab2[y+1][x];
706 }
707
708 if (scorev) {
709 int score = 1024LL*scorev / counttab[x];
710 if (score > bestscore) {
711 bestscore = score;
712 bestv = x;
713 }
714 }
715 }
716 if (!bestscore)
717 break;
718 list [ bestv ] = 1;
719 list_inv[ i ] = bestv;
720 }
721
722 count = FFMAX(i - 1, 1);
723 for (i--; i >= 0; i--) {
724 int v = i * 255 / count;
725 AV_WN32(clut + 4*list_inv[i], RGBA(v/2,v,v/2,v));
726 }
727 }
728
729
730 46 static int save_subtitle_set(AVCodecContext *avctx, AVSubtitle *sub, int *got_output)
731 {
732 46 DVBSubContext *ctx = avctx->priv_data;
733 DVBSubRegionDisplay *display;
734 46 DVBSubDisplayDefinition *display_def = ctx->display_definition;
735 DVBSubRegion *region;
736 AVSubtitleRect *rect;
737 const DVBSubCLUT *clut;
738 const uint32_t *clut_table;
739 int i;
740 46 int offset_x=0, offset_y=0;
741 46 int ret = 0;
742
743
744
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 46 times.
46 if (display_def) {
745 offset_x = display_def->x;
746 offset_y = display_def->y;
747 }
748
749 /* Not touching AVSubtitles again*/
750
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 46 times.
46 if (sub->num_rects) {
751 avpriv_request_sample(ctx, "Different Version of Segment asked Twice");
752 return AVERROR_PATCHWELCOME;
753 }
754
2/2
✓ Branch 0 taken 66 times.
✓ Branch 1 taken 46 times.
112 for (display = ctx->display_list; display; display = display->next) {
755 66 region = get_region(ctx, display->region_id);
756
2/4
✓ Branch 0 taken 66 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 66 times.
✗ Branch 3 not taken.
66 if (region && region->dirty)
757 66 sub->num_rects++;
758 }
759
760
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 46 times.
46 if (ctx->compute_edt == 0) {
761 sub->end_display_time = ctx->time_out * 1000;
762 *got_output = 1;
763
2/2
✓ Branch 0 taken 45 times.
✓ Branch 1 taken 1 times.
46 } else if (ctx->prev_start != AV_NOPTS_VALUE) {
764 45 sub->end_display_time = av_rescale_q((sub->pts - ctx->prev_start ), AV_TIME_BASE_Q, (AVRational){ 1, 1000 }) - 1;
765 45 *got_output = 1;
766 }
767
2/2
✓ Branch 0 taken 36 times.
✓ Branch 1 taken 10 times.
46 if (sub->num_rects > 0) {
768
769 36 sub->rects = av_calloc(sub->num_rects, sizeof(*sub->rects));
770
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 36 times.
36 if (!sub->rects) {
771 ret = AVERROR(ENOMEM);
772 goto fail;
773 }
774
775
2/2
✓ Branch 0 taken 66 times.
✓ Branch 1 taken 36 times.
102 for (i = 0; i < sub->num_rects; i++) {
776 66 sub->rects[i] = av_mallocz(sizeof(*sub->rects[i]));
777
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 66 times.
66 if (!sub->rects[i]) {
778 ret = AVERROR(ENOMEM);
779 goto fail;
780 }
781 }
782
783 36 i = 0;
784
785
2/2
✓ Branch 0 taken 66 times.
✓ Branch 1 taken 36 times.
102 for (display = ctx->display_list; display; display = display->next) {
786 66 region = get_region(ctx, display->region_id);
787
788
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 66 times.
66 if (!region)
789 continue;
790
791
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 66 times.
66 if (!region->dirty)
792 continue;
793
794 66 rect = sub->rects[i];
795 66 rect->x = display->x_pos + offset_x;
796 66 rect->y = display->y_pos + offset_y;
797 66 rect->w = region->width;
798 66 rect->h = region->height;
799 66 rect->nb_colors = (1 << region->depth);
800 66 rect->type = SUBTITLE_BITMAP;
801 66 rect->linesize[0] = region->width;
802
803 66 clut = get_clut(ctx, region->clut);
804
805
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 66 times.
66 if (!clut)
806 clut = &default_clut;
807
808
1/3
✗ Branch 0 not taken.
✗ Branch 1 not taken.
✓ Branch 2 taken 66 times.
66 switch (region->depth) {
809 case 2:
810 clut_table = clut->clut4;
811 break;
812 case 8:
813 clut_table = clut->clut256;
814 break;
815 66 case 4:
816 default:
817 66 clut_table = clut->clut16;
818 66 break;
819 }
820
821 66 rect->data[1] = av_mallocz(AVPALETTE_SIZE);
822
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 66 times.
66 if (!rect->data[1]) {
823 ret = AVERROR(ENOMEM);
824 goto fail;
825 }
826 66 memcpy(rect->data[1], clut_table, (1 << region->depth) * sizeof(*clut_table));
827
828 66 rect->data[0] = av_memdup(region->pbuf, region->buf_size);
829
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 66 times.
66 if (!rect->data[0]) {
830 ret = AVERROR(ENOMEM);
831 goto fail;
832 }
833
834
2/6
✗ Branch 0 not taken.
✓ Branch 1 taken 66 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 66 times.
66 if ((clut == &default_clut && ctx->compute_clut < 0) || ctx->compute_clut == 1) {
835 if (!region->has_computed_clut) {
836 compute_default_clut(ctx, region->computed_clut, rect, rect->w, rect->h);
837 region->has_computed_clut = 1;
838 }
839
840 memcpy(rect->data[1], region->computed_clut, sizeof(region->computed_clut));
841 }
842
843 66 i++;
844 }
845 }
846
847 46 return 0;
848 fail:
849 if (sub->rects) {
850 for (i=0; i < sub->num_rects; i++) {
851 rect = sub->rects[i];
852 if (rect) {
853 av_freep(&rect->data[0]);
854 av_freep(&rect->data[1]);
855 }
856 av_freep(&sub->rects[i]);
857 }
858 av_freep(&sub->rects);
859 }
860 sub->num_rects = 0;
861 return ret;
862 }
863
864 84 static void dvbsub_parse_pixel_data_block(AVCodecContext *avctx, DVBSubObjectDisplay *display,
865 const uint8_t *buf, int buf_size, int top_bottom, int non_mod)
866 {
867 84 DVBSubContext *ctx = avctx->priv_data;
868
869 84 DVBSubRegion *region = get_region(ctx, display->region_id);
870 84 const uint8_t *buf_end = buf + buf_size;
871 uint8_t *pbuf;
872 int x_pos, y_pos;
873 int i;
874
875 84 uint8_t map2to4[] = { 0x0, 0x7, 0x8, 0xf};
876 84 uint8_t map2to8[] = {0x00, 0x77, 0x88, 0xff};
877 84 uint8_t map4to8[] = {0x00, 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77,
878 0x88, 0x99, 0xaa, 0xbb, 0xcc, 0xdd, 0xee, 0xff};
879 uint8_t *map_table;
880
881 #if 0
882 ff_dlog(avctx, "DVB pixel block size %d, %s field:\n", buf_size,
883 top_bottom ? "bottom" : "top");
884
885 for (i = 0; i < buf_size; i++) {
886 if (i % 16 == 0)
887 ff_dlog(avctx, "0x%8p: ", buf+i);
888
889 ff_dlog(avctx, "%02x ", buf[i]);
890 if (i % 16 == 15)
891 ff_dlog(avctx, "\n");
892 }
893
894 if (i % 16)
895 ff_dlog(avctx, "\n");
896 #endif
897
898
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 84 times.
84 if (!region)
899 return;
900
901 84 pbuf = region->pbuf;
902 84 region->dirty = 1;
903
904 84 x_pos = display->x_pos;
905 84 y_pos = display->y_pos;
906
907 84 y_pos += top_bottom;
908
909
2/2
✓ Branch 0 taken 3024 times.
✓ Branch 1 taken 84 times.
3108 while (buf < buf_end) {
910
4/6
✓ Branch 0 taken 1512 times.
✓ Branch 1 taken 1512 times.
✓ Branch 2 taken 1512 times.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✓ Branch 5 taken 3024 times.
3024 if ((*buf!=0xf0 && x_pos >= region->width) || y_pos >= region->height) {
911 av_log(avctx, AV_LOG_ERROR, "Invalid object location! %d-%d %d-%d %02x\n", x_pos, region->width, y_pos, region->height, *buf);
912 return;
913 }
914
915
2/8
✗ Branch 0 not taken.
✓ Branch 1 taken 1512 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
✓ Branch 6 taken 1512 times.
✗ Branch 7 not taken.
3024 switch (*buf++) {
916 case 0x10:
917 if (region->depth == 8)
918 map_table = map2to8;
919 else if (region->depth == 4)
920 map_table = map2to4;
921 else
922 map_table = NULL;
923
924 x_pos = dvbsub_read_2bit_string(avctx, pbuf + (y_pos * region->width),
925 region->width, &buf, buf_end - buf,
926 non_mod, map_table, x_pos);
927 break;
928 1512 case 0x11:
929
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1512 times.
1512 if (region->depth < 4) {
930 av_log(avctx, AV_LOG_ERROR, "4-bit pixel string in %d-bit region!\n", region->depth);
931 return;
932 }
933
934
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 1512 times.
1512 if (region->depth == 8)
935 map_table = map4to8;
936 else
937 1512 map_table = NULL;
938
939 1512 x_pos = dvbsub_read_4bit_string(avctx, pbuf + (y_pos * region->width),
940 1512 region->width, &buf, buf_end - buf,
941 non_mod, map_table, x_pos);
942 1512 break;
943 case 0x12:
944 if (region->depth < 8) {
945 av_log(avctx, AV_LOG_ERROR, "8-bit pixel string in %d-bit region!\n", region->depth);
946 return;
947 }
948
949 x_pos = dvbsub_read_8bit_string(avctx, pbuf + (y_pos * region->width),
950 region->width, &buf, buf_end - buf,
951 non_mod, NULL, x_pos);
952 break;
953
954 case 0x20:
955 map2to4[0] = (*buf) >> 4;
956 map2to4[1] = (*buf++) & 0xf;
957 map2to4[2] = (*buf) >> 4;
958 map2to4[3] = (*buf++) & 0xf;
959 break;
960 case 0x21:
961 for (i = 0; i < 4; i++)
962 map2to8[i] = *buf++;
963 break;
964 case 0x22:
965 for (i = 0; i < 16; i++)
966 map4to8[i] = *buf++;
967 break;
968
969 1512 case 0xf0:
970 1512 x_pos = display->x_pos;
971 1512 y_pos += 2;
972 1512 break;
973 default:
974 av_log(avctx, AV_LOG_INFO, "Unknown/unsupported pixel block 0x%x\n", *(buf-1));
975 }
976 }
977
978
1/2
✓ Branch 0 taken 84 times.
✗ Branch 1 not taken.
84 if (ctx->compute_clut != -2)
979 84 region->has_computed_clut = 0;
980 }
981
982 42 static int dvbsub_parse_object_segment(AVCodecContext *avctx,
983 const uint8_t *buf, int buf_size)
984 {
985 42 DVBSubContext *ctx = avctx->priv_data;
986
987 42 const uint8_t *buf_end = buf + buf_size;
988 int object_id;
989 DVBSubObject *object;
990 DVBSubObjectDisplay *display;
991 int top_field_len, bottom_field_len;
992
993 int coding_method, non_modifying_color;
994
995 42 object_id = AV_RB16(buf);
996 42 buf += 2;
997
998 42 object = get_object(ctx, object_id);
999
1000
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 42 times.
42 if (!object)
1001 return AVERROR_INVALIDDATA;
1002
1003 42 coding_method = ((*buf) >> 2) & 3;
1004 42 non_modifying_color = ((*buf++) >> 1) & 1;
1005
1006
1/2
✓ Branch 0 taken 42 times.
✗ Branch 1 not taken.
42 if (coding_method == 0) {
1007 42 top_field_len = AV_RB16(buf);
1008 42 buf += 2;
1009 42 bottom_field_len = AV_RB16(buf);
1010 42 buf += 2;
1011
1012
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 42 times.
42 if (buf + top_field_len + bottom_field_len > buf_end) {
1013 av_log(avctx, AV_LOG_ERROR, "Field data size %d+%d too large\n", top_field_len, bottom_field_len);
1014 return AVERROR_INVALIDDATA;
1015 }
1016
1017
2/2
✓ Branch 0 taken 42 times.
✓ Branch 1 taken 42 times.
84 for (display = object->display_list; display; display = display->object_list_next) {
1018 42 const uint8_t *block = buf;
1019 42 int bfl = bottom_field_len;
1020
1021 42 dvbsub_parse_pixel_data_block(avctx, display, block, top_field_len, 0,
1022 non_modifying_color);
1023
1024
1/2
✓ Branch 0 taken 42 times.
✗ Branch 1 not taken.
42 if (bottom_field_len > 0)
1025 42 block = buf + top_field_len;
1026 else
1027 bfl = top_field_len;
1028
1029 42 dvbsub_parse_pixel_data_block(avctx, display, block, bfl, 1,
1030 non_modifying_color);
1031 }
1032 } else if (coding_method == 1) {
1033 avpriv_report_missing_feature(avctx, "coded as a string of characters");
1034 return AVERROR_PATCHWELCOME;
1035 } else if (coding_method == 2) {
1036 avpriv_report_missing_feature(avctx, "progressive coding of pixels");
1037 return AVERROR_PATCHWELCOME;
1038 } else {
1039 av_log(avctx, AV_LOG_ERROR, "Unknown object coding %d\n", coding_method);
1040 return AVERROR_INVALIDDATA;
1041 }
1042
1043 42 return 0;
1044 }
1045
1046 13 static int dvbsub_parse_clut_segment(AVCodecContext *avctx,
1047 const uint8_t *buf, int buf_size)
1048 {
1049 13 DVBSubContext *ctx = avctx->priv_data;
1050
1051 13 const uint8_t *buf_end = buf + buf_size;
1052 int i, clut_id;
1053 int version;
1054 DVBSubCLUT *clut;
1055 int entry_id, depth , full_range;
1056 int y, cr, cb, alpha;
1057 int r, g, b, r_add, g_add, b_add;
1058
1059 ff_dlog(avctx, "DVB clut packet:\n");
1060
1061
2/2
✓ Branch 0 taken 728 times.
✓ Branch 1 taken 13 times.
741 for (i=0; i < buf_size; i++) {
1062 ff_dlog(avctx, "%02x ", buf[i]);
1063 728 if (i % 16 == 15)
1064 ff_dlog(avctx, "\n");
1065 }
1066
1067 13 if (i % 16)
1068 ff_dlog(avctx, "\n");
1069
1070 13 clut_id = *buf++;
1071 13 version = ((*buf)>>4)&15;
1072 13 buf += 1;
1073
1074 13 clut = get_clut(ctx, clut_id);
1075
1076
1/2
✓ Branch 0 taken 13 times.
✗ Branch 1 not taken.
13 if (!clut) {
1077 13 clut = av_memdup(&default_clut, sizeof(*clut));
1078
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 13 times.
13 if (!clut)
1079 return AVERROR(ENOMEM);
1080
1081 13 clut->id = clut_id;
1082 13 clut->version = -1;
1083
1084 13 clut->next = ctx->clut_list;
1085 13 ctx->clut_list = clut;
1086 }
1087
1088
1/2
✓ Branch 0 taken 13 times.
✗ Branch 1 not taken.
13 if (clut->version != version) {
1089
1090 13 clut->version = version;
1091
1092
2/2
✓ Branch 0 taken 117 times.
✓ Branch 1 taken 13 times.
130 while (buf + 4 < buf_end) {
1093 117 entry_id = *buf++;
1094
1095 117 depth = (*buf) & 0xe0;
1096
1097
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 117 times.
117 if (depth == 0) {
1098 av_log(avctx, AV_LOG_ERROR, "Invalid clut depth 0x%x!\n", *buf);
1099 }
1100
1101 117 full_range = (*buf++) & 1;
1102
1103
1/2
✓ Branch 0 taken 117 times.
✗ Branch 1 not taken.
117 if (full_range) {
1104 117 y = *buf++;
1105 117 cr = *buf++;
1106 117 cb = *buf++;
1107 117 alpha = *buf++;
1108 } else {
1109 y = buf[0] & 0xfc;
1110 cr = (((buf[0] & 3) << 2) | ((buf[1] >> 6) & 3)) << 4;
1111 cb = (buf[1] << 2) & 0xf0;
1112 alpha = (buf[1] << 6) & 0xc0;
1113
1114 buf += 2;
1115 }
1116
1117
2/2
✓ Branch 0 taken 13 times.
✓ Branch 1 taken 104 times.
117 if (y == 0)
1118 13 alpha = 0xff;
1119
1120 117 YUV_TO_RGB1_CCIR(cb, cr);
1121 117 YUV_TO_RGB2_CCIR(r, g, b, y);
1122
1123 ff_dlog(avctx, "clut %d := (%d,%d,%d,%d)\n", entry_id, r, g, b, alpha);
1124
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 117 times.
117 if (!!(depth & 0x80) + !!(depth & 0x40) + !!(depth & 0x20) > 1) {
1125 ff_dlog(avctx, "More than one bit level marked: %x\n", depth);
1126 if (avctx->strict_std_compliance > FF_COMPLIANCE_NORMAL)
1127 return AVERROR_INVALIDDATA;
1128 }
1129
1130
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 117 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
117 if (depth & 0x80 && entry_id < 4)
1131 clut->clut4[entry_id] = RGBA(r,g,b,255 - alpha);
1132
2/4
✓ Branch 0 taken 117 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 117 times.
✗ Branch 3 not taken.
117 else if (depth & 0x40 && entry_id < 16)
1133 117 clut->clut16[entry_id] = RGBA(r,g,b,255 - alpha);
1134 else if (depth & 0x20)
1135 clut->clut256[entry_id] = RGBA(r,g,b,255 - alpha);
1136 }
1137 }
1138
1139 13 return 0;
1140 }
1141
1142
1143 57 static int dvbsub_parse_region_segment(AVCodecContext *avctx,
1144 const uint8_t *buf, int buf_size)
1145 {
1146 57 DVBSubContext *ctx = avctx->priv_data;
1147
1148 57 const uint8_t *buf_end = buf + buf_size;
1149 int region_id, object_id;
1150 int av_unused version;
1151 DVBSubRegion *region;
1152 DVBSubObject *object;
1153 DVBSubObjectDisplay *display;
1154 int fill;
1155 int ret;
1156
1157
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 57 times.
57 if (buf_size < 10)
1158 return AVERROR_INVALIDDATA;
1159
1160 57 region_id = *buf++;
1161
1162 57 region = get_region(ctx, region_id);
1163
1164
2/2
✓ Branch 0 taken 28 times.
✓ Branch 1 taken 29 times.
57 if (!region) {
1165 28 region = av_mallocz(sizeof(*region));
1166
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 28 times.
28 if (!region)
1167 return AVERROR(ENOMEM);
1168
1169 28 region->id = region_id;
1170 28 region->version = -1;
1171
1172 28 region->next = ctx->region_list;
1173 28 ctx->region_list = region;
1174 }
1175
1176 57 version = ((*buf)>>4) & 15;
1177 57 fill = ((*buf++) >> 3) & 1;
1178
1179 57 region->width = AV_RB16(buf);
1180 57 buf += 2;
1181 57 region->height = AV_RB16(buf);
1182 57 buf += 2;
1183
1184 57 ret = av_image_check_size2(region->width, region->height, avctx->max_pixels, AV_PIX_FMT_PAL8, 0, avctx);
1185
2/4
✓ Branch 0 taken 57 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 57 times.
57 if (ret >= 0 && region->width * region->height * 2 > 320 * 1024 * 8) {
1186 ret = AVERROR_INVALIDDATA;
1187 av_log(avctx, AV_LOG_ERROR, "Pixel buffer memory constraint violated\n");
1188 }
1189
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 57 times.
57 if (ret < 0) {
1190 region->width= region->height= 0;
1191 return ret;
1192 }
1193
1194
2/2
✓ Branch 0 taken 28 times.
✓ Branch 1 taken 29 times.
57 if (region->width * region->height != region->buf_size) {
1195 28 av_free(region->pbuf);
1196
1197 28 region->buf_size = region->width * region->height;
1198
1199 28 region->pbuf = av_malloc(region->buf_size);
1200
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 28 times.
28 if (!region->pbuf) {
1201 region->buf_size =
1202 region->width =
1203 region->height = 0;
1204 return AVERROR(ENOMEM);
1205 }
1206
1207 28 fill = 1;
1208 28 region->dirty = 0;
1209 }
1210
1211 57 region->depth = 1 << (((*buf++) >> 2) & 7);
1212
2/4
✓ Branch 0 taken 57 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 57 times.
57 if (region->depth < 2 || region->depth > 8) {
1213 av_log(avctx, AV_LOG_ERROR, "region depth %d is invalid\n", region->depth);
1214 region->depth= 4;
1215 }
1216 57 region->clut = *buf++;
1217
1218
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 57 times.
57 if (region->depth == 8) {
1219 region->bgcolor = *buf++;
1220 buf += 1;
1221 } else {
1222 57 buf += 1;
1223
1224
1/2
✓ Branch 0 taken 57 times.
✗ Branch 1 not taken.
57 if (region->depth == 4)
1225 57 region->bgcolor = (((*buf++) >> 4) & 15);
1226 else
1227 region->bgcolor = (((*buf++) >> 2) & 3);
1228 }
1229
1230 ff_dlog(avctx, "Region %d, (%dx%d)\n", region_id, region->width, region->height);
1231
1232
2/2
✓ Branch 0 taken 28 times.
✓ Branch 1 taken 29 times.
57 if (fill) {
1233 28 memset(region->pbuf, region->bgcolor, region->buf_size);
1234 ff_dlog(avctx, "Fill region (%d)\n", region->bgcolor);
1235 }
1236
1237 57 delete_region_display_list(ctx, region);
1238
1239
2/2
✓ Branch 0 taken 42 times.
✓ Branch 1 taken 57 times.
99 while (buf + 5 < buf_end) {
1240 42 object_id = AV_RB16(buf);
1241 42 buf += 2;
1242
1243 42 object = get_object(ctx, object_id);
1244
1245
1/2
✓ Branch 0 taken 42 times.
✗ Branch 1 not taken.
42 if (!object) {
1246 42 object = av_mallocz(sizeof(*object));
1247
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 42 times.
42 if (!object)
1248 return AVERROR(ENOMEM);
1249
1250 42 object->id = object_id;
1251 42 object->next = ctx->object_list;
1252 42 ctx->object_list = object;
1253 }
1254
1255 42 object->type = (*buf) >> 6;
1256
1257 42 display = av_mallocz(sizeof(*display));
1258
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 42 times.
42 if (!display)
1259 return AVERROR(ENOMEM);
1260
1261 42 display->object_id = object_id;
1262 42 display->region_id = region_id;
1263
1264 42 display->x_pos = AV_RB16(buf) & 0xfff;
1265 42 buf += 2;
1266 42 display->y_pos = AV_RB16(buf) & 0xfff;
1267 42 buf += 2;
1268
1269
1/2
✓ Branch 0 taken 42 times.
✗ Branch 1 not taken.
42 if (display->x_pos >= region->width ||
1270
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 42 times.
42 display->y_pos >= region->height) {
1271 av_log(avctx, AV_LOG_ERROR, "Object outside region\n");
1272 av_free(display);
1273 return AVERROR_INVALIDDATA;
1274 }
1275
1276
2/6
✓ Branch 0 taken 42 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 42 times.
✗ Branch 4 not taken.
✗ Branch 5 not taken.
42 if ((object->type == 1 || object->type == 2) && buf+1 < buf_end) {
1277 display->fgcolor = *buf++;
1278 display->bgcolor = *buf++;
1279 }
1280
1281 42 display->region_list_next = region->display_list;
1282 42 region->display_list = display;
1283
1284 42 display->object_list_next = object->display_list;
1285 42 object->display_list = display;
1286 }
1287
1288 57 return 0;
1289 }
1290
1291 46 static int dvbsub_parse_page_segment(AVCodecContext *avctx,
1292 const uint8_t *buf, int buf_size, AVSubtitle *sub, int *got_output)
1293 {
1294 46 DVBSubContext *ctx = avctx->priv_data;
1295 DVBSubRegionDisplay *display;
1296 DVBSubRegionDisplay *tmp_display_list, **tmp_ptr;
1297
1298 46 const uint8_t *buf_end = buf + buf_size;
1299 int region_id;
1300 int page_state;
1301 int timeout;
1302 int version;
1303
1304
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 46 times.
46 if (buf_size < 1)
1305 return AVERROR_INVALIDDATA;
1306
1307 46 timeout = *buf++;
1308 46 version = ((*buf)>>4) & 15;
1309 46 page_state = ((*buf++) >> 2) & 3;
1310
1311
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 46 times.
46 if (ctx->version == version) {
1312 return 0;
1313 }
1314
1315 46 ctx->time_out = timeout;
1316 46 ctx->version = version;
1317
1318 ff_dlog(avctx, "Page time out %ds, state %d\n", ctx->time_out, page_state);
1319
1320
1/2
✓ Branch 0 taken 46 times.
✗ Branch 1 not taken.
46 if (ctx->compute_edt == 1)
1321 46 save_subtitle_set(avctx, sub, got_output);
1322
1323
3/4
✓ Branch 0 taken 39 times.
✓ Branch 1 taken 7 times.
✗ Branch 2 not taken.
✓ Branch 3 taken 39 times.
46 if (page_state == 1 || page_state == 2) {
1324 7 delete_regions(ctx);
1325 7 delete_objects(ctx);
1326 7 delete_cluts(ctx);
1327 }
1328
1329 46 tmp_display_list = ctx->display_list;
1330 46 ctx->display_list = NULL;
1331
1332
2/2
✓ Branch 0 taken 66 times.
✓ Branch 1 taken 46 times.
112 while (buf + 5 < buf_end) {
1333 66 region_id = *buf++;
1334 66 buf += 1;
1335
1336 66 display = ctx->display_list;
1337
3/4
✓ Branch 0 taken 30 times.
✓ Branch 1 taken 66 times.
✓ Branch 2 taken 30 times.
✗ Branch 3 not taken.
96 while (display && display->region_id != region_id) {
1338 30 display = display->next;
1339 }
1340
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 66 times.
66 if (display) {
1341 av_log(avctx, AV_LOG_ERROR, "duplicate region\n");
1342 break;
1343 }
1344
1345 66 display = tmp_display_list;
1346 66 tmp_ptr = &tmp_display_list;
1347
1348
4/4
✓ Branch 0 taken 93 times.
✓ Branch 1 taken 2 times.
✓ Branch 2 taken 29 times.
✓ Branch 3 taken 64 times.
95 while (display && display->region_id != region_id) {
1349 29 tmp_ptr = &display->next;
1350 29 display = display->next;
1351 }
1352
1353
2/2
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 64 times.
66 if (!display) {
1354 2 display = av_mallocz(sizeof(*display));
1355
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 2 times.
2 if (!display)
1356 return AVERROR(ENOMEM);
1357 }
1358
1359 66 display->region_id = region_id;
1360
1361 66 display->x_pos = AV_RB16(buf);
1362 66 buf += 2;
1363 66 display->y_pos = AV_RB16(buf);
1364 66 buf += 2;
1365
1366 66 *tmp_ptr = display->next;
1367
1368 66 display->next = ctx->display_list;
1369 66 ctx->display_list = display;
1370
1371 ff_dlog(avctx, "Region %d, (%d,%d)\n", region_id, display->x_pos, display->y_pos);
1372 }
1373
1374
2/2
✓ Branch 0 taken 2 times.
✓ Branch 1 taken 46 times.
48 while (tmp_display_list) {
1375 2 display = tmp_display_list;
1376
1377 2 tmp_display_list = display->next;
1378
1379 2 av_freep(&display);
1380 }
1381
1382 46 return 0;
1383 }
1384
1385 static int dvbsub_parse_display_definition_segment(AVCodecContext *avctx,
1386 const uint8_t *buf,
1387 int buf_size)
1388 {
1389 DVBSubContext *ctx = avctx->priv_data;
1390 DVBSubDisplayDefinition *display_def = ctx->display_definition;
1391 int dds_version, info_byte;
1392
1393 if (buf_size < 5)
1394 return AVERROR_INVALIDDATA;
1395
1396 info_byte = bytestream_get_byte(&buf);
1397 dds_version = info_byte >> 4;
1398 if (display_def && display_def->version == dds_version)
1399 return 0; // already have this display definition version
1400
1401 if (!display_def) {
1402 display_def = av_mallocz(sizeof(*display_def));
1403 if (!display_def)
1404 return AVERROR(ENOMEM);
1405 ctx->display_definition = display_def;
1406 }
1407
1408 display_def->version = dds_version;
1409 display_def->x = 0;
1410 display_def->y = 0;
1411 display_def->width = bytestream_get_be16(&buf) + 1;
1412 display_def->height = bytestream_get_be16(&buf) + 1;
1413 if (!avctx->width || !avctx->height) {
1414 int ret = ff_set_dimensions(avctx, display_def->width, display_def->height);
1415 if (ret < 0)
1416 return ret;
1417 }
1418
1419 if (info_byte & 1<<3) { // display_window_flag
1420 if (buf_size < 13)
1421 return AVERROR_INVALIDDATA;
1422
1423 display_def->x = bytestream_get_be16(&buf);
1424 display_def->width = bytestream_get_be16(&buf) - display_def->x + 1;
1425 display_def->y = bytestream_get_be16(&buf);
1426 display_def->height = bytestream_get_be16(&buf) - display_def->y + 1;
1427 }
1428
1429 return 0;
1430 }
1431
1432 46 static int dvbsub_display_end_segment(AVCodecContext *avctx, const uint8_t *buf,
1433 int buf_size, AVSubtitle *sub,int *got_output)
1434 {
1435 46 DVBSubContext *ctx = avctx->priv_data;
1436
1437
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 46 times.
46 if (ctx->compute_edt == 0)
1438 save_subtitle_set(avctx, sub, got_output);
1439 46 return 0;
1440 }
1441
1442 46 static int dvbsub_decode(AVCodecContext *avctx, AVSubtitle *sub,
1443 int *got_sub_ptr, const AVPacket *avpkt)
1444 {
1445 46 const uint8_t *buf = avpkt->data;
1446 46 int buf_size = avpkt->size;
1447 46 DVBSubContext *ctx = avctx->priv_data;
1448 const uint8_t *p, *p_end;
1449 int segment_type;
1450 int page_id;
1451 int segment_length;
1452 int i;
1453 46 int ret = 0;
1454 46 int got_segment = 0;
1455 46 int got_dds = 0;
1456
1457 ff_dlog(avctx, "DVB sub packet:\n");
1458
1459
2/2
✓ Branch 0 taken 51158 times.
✓ Branch 1 taken 46 times.
51204 for (i=0; i < buf_size; i++) {
1460 ff_dlog(avctx, "%02x ", buf[i]);
1461 51158 if (i % 16 == 15)
1462 ff_dlog(avctx, "\n");
1463 }
1464
1465 46 if (i % 16)
1466 ff_dlog(avctx, "\n");
1467
1468
2/4
✓ Branch 0 taken 46 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 46 times.
46 if (buf_size <= 6 || *buf != 0x0f) {
1469 ff_dlog(avctx, "incomplete or broken packet");
1470 return AVERROR_INVALIDDATA;
1471 }
1472
1473 46 p = buf;
1474 46 p_end = buf + buf_size;
1475
1476
3/4
✓ Branch 0 taken 204 times.
✓ Branch 1 taken 46 times.
✓ Branch 2 taken 204 times.
✗ Branch 3 not taken.
250 while (p_end - p >= 6 && *p == 0x0f) {
1477 204 p += 1;
1478 204 segment_type = *p++;
1479 204 page_id = AV_RB16(p);
1480 204 p += 2;
1481 204 segment_length = AV_RB16(p);
1482 204 p += 2;
1483
1484
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 204 times.
204 if (avctx->debug & FF_DEBUG_STARTCODE) {
1485 av_log(avctx, AV_LOG_DEBUG, "segment_type:%d page_id:%d segment_length:%d\n", segment_type, page_id, segment_length);
1486 }
1487
1488
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 204 times.
204 if (p_end - p < segment_length) {
1489 ff_dlog(avctx, "incomplete or broken packet");
1490 ret = -1;
1491 goto end;
1492 }
1493
1494
2/4
✓ Branch 0 taken 204 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 204 times.
✗ Branch 3 not taken.
204 if (page_id == ctx->composition_id || page_id == ctx->ancillary_id ||
1495
1/4
✗ Branch 0 not taken.
✓ Branch 1 taken 204 times.
✗ Branch 2 not taken.
✗ Branch 3 not taken.
204 ctx->composition_id == -1 || ctx->ancillary_id == -1) {
1496 204 int ret = 0;
1497
5/7
✓ Branch 0 taken 46 times.
✓ Branch 1 taken 57 times.
✓ Branch 2 taken 13 times.
✓ Branch 3 taken 42 times.
✗ Branch 4 not taken.
✓ Branch 5 taken 46 times.
✗ Branch 6 not taken.
204 switch (segment_type) {
1498 46 case DVBSUB_PAGE_SEGMENT:
1499 46 ret = dvbsub_parse_page_segment(avctx, p, segment_length, sub, got_sub_ptr);
1500 46 got_segment |= 1;
1501 46 break;
1502 57 case DVBSUB_REGION_SEGMENT:
1503 57 ret = dvbsub_parse_region_segment(avctx, p, segment_length);
1504 57 got_segment |= 2;
1505 57 break;
1506 13 case DVBSUB_CLUT_SEGMENT:
1507 13 ret = dvbsub_parse_clut_segment(avctx, p, segment_length);
1508
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 13 times.
13 if (ret < 0) goto end;
1509 13 got_segment |= 4;
1510 13 break;
1511 42 case DVBSUB_OBJECT_SEGMENT:
1512 42 ret = dvbsub_parse_object_segment(avctx, p, segment_length);
1513 42 got_segment |= 8;
1514 42 break;
1515 case DVBSUB_DISPLAYDEFINITION_SEGMENT:
1516 ret = dvbsub_parse_display_definition_segment(avctx, p,
1517 segment_length);
1518 got_dds = 1;
1519 break;
1520 46 case DVBSUB_DISPLAY_SEGMENT:
1521 46 ret = dvbsub_display_end_segment(avctx, p, segment_length, sub, got_sub_ptr);
1522
6/8
✓ Branch 0 taken 7 times.
✓ Branch 1 taken 39 times.
✓ Branch 2 taken 7 times.
✗ Branch 3 not taken.
✓ Branch 4 taken 1 times.
✓ Branch 5 taken 6 times.
✓ Branch 6 taken 1 times.
✗ Branch 7 not taken.
46 if (got_segment == 15 && !got_dds && !avctx->width && !avctx->height) {
1523 // Default from ETSI EN 300 743 V1.3.1 (7.2.1)
1524 1 avctx->width = 720;
1525 1 avctx->height = 576;
1526 }
1527 46 got_segment |= 16;
1528 46 break;
1529 default:
1530 ff_dlog(avctx, "Subtitling segment type 0x%x, page id %d, length %d\n",
1531 segment_type, page_id, segment_length);
1532 break;
1533 }
1534
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 204 times.
204 if (ret < 0)
1535 goto end;
1536 }
1537
1538 204 p += segment_length;
1539 }
1540 // Some streams do not send a display segment but if we have all the other
1541 // segments then we need no further data.
1542
1/2
✓ Branch 0 taken 46 times.
✗ Branch 1 not taken.
46 if (got_segment == 15) {
1543 av_log(avctx, AV_LOG_DEBUG, "Missing display_end_segment, emulating\n");
1544 dvbsub_display_end_segment(avctx, p, 0, sub, got_sub_ptr);
1545 }
1546
1547 46 end:
1548
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 46 times.
46 if (ret < 0) {
1549 return ret;
1550 } else {
1551
1/2
✓ Branch 0 taken 46 times.
✗ Branch 1 not taken.
46 if (ctx->compute_edt == 1)
1552 46 FFSWAP(int64_t, ctx->prev_start, sub->pts);
1553 }
1554
1555 46 return p - buf;
1556 }
1557
1558 #define DS AV_OPT_FLAG_DECODING_PARAM | AV_OPT_FLAG_SUBTITLE_PARAM
1559 #define OFFSET(x) offsetof(DVBSubContext, x)
1560 static const AVOption options[] = {
1561 {"compute_edt", "compute end of time using pts or timeout", OFFSET(compute_edt), AV_OPT_TYPE_BOOL, {.i64 = 0}, 0, 1, DS},
1562 {"compute_clut", "compute clut when not available(-1) or only once (-2) or always(1) or never(0)", OFFSET(compute_clut), AV_OPT_TYPE_BOOL, {.i64 = -1}, -2, 1, DS},
1563 {"dvb_substream", "", OFFSET(substream), AV_OPT_TYPE_INT, {.i64 = -1}, -1, 63, DS},
1564 {NULL}
1565 };
1566 static const AVClass dvbsubdec_class = {
1567 .class_name = "DVB Sub Decoder",
1568 .item_name = av_default_item_name,
1569 .option = options,
1570 .version = LIBAVUTIL_VERSION_INT,
1571 };
1572
1573 const FFCodec ff_dvbsub_decoder = {
1574 .p.name = "dvbsub",
1575 CODEC_LONG_NAME("DVB subtitles"),
1576 .p.type = AVMEDIA_TYPE_SUBTITLE,
1577 .p.id = AV_CODEC_ID_DVB_SUBTITLE,
1578 .priv_data_size = sizeof(DVBSubContext),
1579 .init = dvbsub_init_decoder,
1580 .close = dvbsub_close_decoder,
1581 FF_CODEC_DECODE_SUB_CB(dvbsub_decode),
1582 .p.priv_class = &dvbsubdec_class,
1583 };
1584