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