Merge branch 'master' into dvd_scale_test
[enigma2.git] / lib / dvb / subtitle.cpp
1 #include <stdio.h>
2 #include <stdlib.h>
3 #include <fcntl.h>
4 #include <memory.h>
5 #include <time.h>
6
7 #include <asm/types.h>
8 #include <lib/dvb/idemux.h>
9 #include <lib/dvb/subtitle.h>
10 #include <lib/base/smartptr.h>
11 #include <lib/base/eerror.h>
12 #include <lib/gdi/gpixmap.h>
13
14 void bitstream_init(bitstream *bit, const void *buffer, int size)
15 {
16         bit->data = (__u8*) buffer;
17         bit->size = size;
18         bit->avail = 8;
19         bit->consumed = 0;
20 }
21
22 int bitstream_get(bitstream *bit)
23 {
24         int val;
25         bit->avail -= bit->size;
26         val = ((*bit->data) >> bit->avail) & ((1<<bit->size) - 1);
27         if (!bit->avail)
28         {
29                 bit->data++;
30                 bit->consumed++;
31                 bit->avail = 8;
32         }
33         return val;
34 }
35
36 static int extract_pts(pts_t &pts, __u8 *pkt)
37 {
38         pkt += 7;
39         int flags = *pkt++;
40
41         pkt++; // header length
42
43         if (flags & 0x80) /* PTS present? */
44         {
45                         /* damn gcc bug */
46                 pts  = ((unsigned long long)(((pkt[0] >> 1) & 7))) << 30;
47                 pts |=   pkt[1] << 22;
48                 pts |=  (pkt[2]>>1) << 15;
49                 pts |=   pkt[3] << 7;
50                 pts |=  (pkt[5]>>1);
51
52                 return 0;
53         } else
54                 return -1;
55 }
56
57 void eDVBSubtitleParser::subtitle_process_line(subtitle_page *page, int object_id, int line, __u8 *data, int len)
58 {
59         subtitle_region *region = page->regions;
60 //      eDebug("line for %d:%d", page->page_id, object_id);
61         while (region)
62         {
63                 subtitle_region_object *object = region->region_objects;
64                 while (object)
65                 {
66                         if (object->object_id == object_id)
67                         {
68                                 int x = object->object_horizontal_position;
69                                 int y = object->object_vertical_position + line;
70                                 if (x + len > region->region_width)
71                                 {
72                                         //eDebug("[SUB] !!!! XCLIP %d + %d > %d", x, len, region->region_width);
73                                         len = region->region_width - x;
74                                 }
75                                 if (len < 0)
76                                         break;
77                                 if (y >= region->region_height)
78                                 {
79                                         //eDebug("[SUB] !!!! YCLIP %d >= %d", y, region->region_height);
80                                         break;
81                                 }
82 //                              //eDebug("inserting %d bytes (into region %d)", len, region->region_id);
83 //                              eDebug("put data to region_buffer %p", &(*region->region_buffer));
84                                 memcpy((__u8*)region->region_buffer->surface->data + region->region_width * y + x, data, len);
85                         }
86                         object = object->next;
87                 }
88                 region = region->next;
89         }
90 }
91
92 int eDVBSubtitleParser::subtitle_process_pixel_data(subtitle_page *page, int object_id, int *linenr, int *linep, __u8 *data)
93 {
94         int data_type = *data++;
95         static __u8 line[720];
96
97         bitstream bit;
98         bit.size=0;
99         switch (data_type)
100         {
101         case 0x10: // 2bit pixel data
102                 bitstream_init(&bit, data, 2);
103                 while (1)
104                 {
105                         int len=0, col=0;
106                         int code = bitstream_get(&bit);
107                         if (code)
108                         {
109                                 col = code;
110                                 len = 1;
111                         } else
112                         {
113                                 code = bitstream_get(&bit);
114                                 if (!code)
115                                 {
116                                         code = bitstream_get(&bit);
117                                         if (code == 1)
118                                         {
119                                                 col = 0;
120                                                 len = 2;
121                                         } else if (code == 2)
122                                         {
123                                                 len = bitstream_get(&bit) << 2;
124                                                 len |= bitstream_get(&bit);
125                                                 len += 12;
126                                                 col = bitstream_get(&bit);
127                                         } else if (code == 3)
128                                         {
129                                                 len = bitstream_get(&bit) << 6;
130                                                 len |= bitstream_get(&bit) << 4;
131                                                 len |= bitstream_get(&bit) << 2;
132                                                 len |= bitstream_get(&bit);
133                                                 len += 29;
134                                                 col = bitstream_get(&bit);
135                                         } else
136                                                 break;
137                                 } else if (code==1)
138                                 {
139                                         col = 0;
140                                         len = 1;
141                                 } else if (code&2)
142                                 {
143                                         if (code&1)
144                                                 len = 3 + 4 + bitstream_get(&bit);
145                                         else
146                                                 len = 3 + bitstream_get(&bit);
147                                         col = bitstream_get(&bit);
148                                 }
149                         }
150                         while (len && ((*linep) < 720))
151                         {
152                                 line[(*linep)++] = col;
153                                 len--;
154                         }
155                 }
156                 while (bit.avail != 8)
157                         bitstream_get(&bit);
158                 return bit.consumed + 1;
159         case 0x11: // 4bit pixel data
160                 bitstream_init(&bit, data, 4);
161                 while (1)
162                 {
163                         int len=0, col=0;
164                         int code = bitstream_get(&bit);
165                         if (code)
166                         {
167                                 col = code;
168                                 len = 1;
169                         } else
170                         {
171                                 code = bitstream_get(&bit);
172                                 if (!code)
173                                         break;
174                                 else if (code == 0xC)
175                                 {
176                                         col = 0;
177                                         len = 1;
178                                 } else if (code == 0xD)
179                                 {
180                                         col = 0;
181                                         len = 2;
182                                 } else if (code < 8)
183                                 {
184                                         col = 0;
185                                         len = (code & 7) + 2;
186                                 } else if ((code & 0xC) == 0x8)
187                                 {
188                                         col = bitstream_get(&bit);
189                                         len = (code & 3) + 4;
190                                 } else if (code == 0xE)
191                                 {
192                                         len = bitstream_get(&bit) + 9;
193                                         col = bitstream_get(&bit);
194                                 } else if (code == 0xF)
195                                 {
196                                         len  = bitstream_get(&bit) << 4;
197                                         len |= bitstream_get(&bit);
198                                         len += 25;
199                                         col  = bitstream_get(&bit);
200                                 }
201                         }
202                         while (len && ((*linep) < 720))
203                         {
204                                 line[(*linep)++] = col;
205                                 len--;
206                         }
207                 }
208                 while (bit.avail != 8)
209                         bitstream_get(&bit);
210                 return bit.consumed + 1;
211         case 0x12: // 8bit pixel data
212                 bitstream_init(&bit, data, 8);
213                 while(1)
214                 {
215                         int len=0, col=0;
216                         int code = bitstream_get(&bit);
217                         if (code)
218                         {
219                                 col = code;
220                                 len = 1;
221                         } else
222                         {
223                                 code = bitstream_get(&bit);
224                                 if ((code & 0x80) == 0x80)
225                                 {
226                                         len = code&0x7F;
227                                         col = bitstream_get(&bit);
228                                 } else if (code&0x7F)
229                                 {
230                                         len = code&0x7F;
231                                         col = 0;
232                                 } else
233                                         break;
234                         }
235                         while (len && ((*linep) < 720))
236                         {
237                                 line[(*linep)++] = col;
238                                 len--;
239                         }
240                 }
241                 return bit.consumed + 1;
242         case 0x20:  // ignore 2 -> 4bit map table
243                 bitstream_init(&bit, data, 4);
244                 for ( int i=0; i < 4; ++i )
245                         bitstream_get(&bit);
246                 break;
247         case 0x21:  // ignore 2 -> 8bit map table
248                 bitstream_init(&bit, data, 8);
249                 for ( int i=0; i < 4; ++i )
250                         bitstream_get(&bit);
251                 break;
252         case 0x22:  // ignore 4 -> 8bit map table
253                 bitstream_init(&bit, data, 8);
254                 for ( int i=0; i < 16; ++i )
255                         bitstream_get(&bit);
256                 break;
257         case 0xF0:
258                 subtitle_process_line(page, object_id, *linenr, line, *linep);
259 /*              {
260                         int i;
261                         for (i=0; i<720; ++i)
262                                 //eDebugNoNewLine("%d ", line[i]);
263                         //eDebug("");
264                 } */
265                 (*linenr)+=2; // interlaced
266                 *linep = 0;
267 //              //eDebug("[SUB] EOL");
268                 return 1;
269         default:
270                 eDebug("subtitle_process_pixel_data: invalid data_type %02x", data_type);
271                 return -1;
272         }
273         return 0;
274 }
275
276 int eDVBSubtitleParser::subtitle_process_segment(__u8 *segment)
277 {
278         int segment_type, page_id, segment_length, processed_length;
279         if (*segment++ !=  0x0F)
280         {
281                 eDebug("out of sync.");
282                 return -1;
283         }
284         segment_type = *segment++;
285         page_id  = *segment++ << 8;
286         page_id |= *segment++;
287         segment_length  = *segment++ << 8;
288         segment_length |= *segment++;
289         if (segment_type == 0xFF)
290                 return segment_length + 6;
291         if (page_id != m_composition_page_id && page_id != m_ancillary_page_id)
292                 return segment_length + 6;
293         //eDebug("have %d bytes of segment data", segment_length);
294
295         //eDebug("page_id %d, segtype %02x", page_id, segment_type);
296
297         subtitle_page *page, **ppage;
298
299         page = m_pages; ppage = &m_pages;
300
301         while (page)
302         {
303                 if (page->page_id == page_id)
304                         break;
305                 ppage = &page->next;
306                 page = page->next;
307         }
308
309         processed_length = 0;
310
311         switch (segment_type)
312         {
313         case 0x10: // page composition segment
314         {
315                 int page_time_out = *segment++; processed_length++;
316                 int page_version_number = *segment >> 4;
317                 int page_state = (*segment >> 2) & 0x3;
318                 //eDebug("pcs with %d bytes data (%d:%d:%d)", segment_length, page_id, page_version_number, page_state);
319                 segment++;
320                 processed_length++;
321
322                 //eDebug("page time out: %d", page_time_out);
323                 //eDebug("page_version_number: %d" ,page_version_number);
324                 //eDebug("page_state: %d", page_state);
325
326                 if (!page)
327                 {
328                         //eDebug("page not found");
329                         page = new subtitle_page;
330                         page->page_regions = 0;
331                         page->regions = 0;
332                         page->page_id = page_id;
333                         page->cluts = 0;
334                         page->next = 0;
335                         *ppage = page;
336                 } else
337                 {
338                         if (page->pcs_size != segment_length)
339                                 page->page_version_number = -1;
340                                 // if no update, just skip this data.
341                         if (page->page_version_number == page_version_number)
342                         {
343 //                              eDebug("skip data... ");
344                                 break;
345                         }
346                 }
347
348                 //eDebug("page updated: old: %d, new: %d", page->page_version_number, page_version_number);
349                         // when acquisition point or mode change: remove all displayed pages.
350                 if ((page_state == 1) || (page_state == 2))
351                 {
352                         while (page->page_regions)
353                         {
354                                 subtitle_page_region *p = page->page_regions->next;
355                                 //eDebug("delete page_region %d", page->page_regions->region_id);
356                                 delete page->page_regions;
357                                 page->page_regions = p;
358                         }
359                         while (page->regions)
360                         {
361                                 subtitle_region *p = page->regions->next;
362                                 //eDebug("delete region %d", page->regions->region_id);
363                                 while(page->regions->region_objects)
364                                 {
365                                         //eDebug("delete region object");
366                                         subtitle_region_object *ob = page->regions->region_objects->next;
367                                         delete page->regions->region_objects;
368                                         page->regions->region_objects = ob;
369                                 }
370                                 delete page->regions;
371                                 page->regions = p;
372                         }
373
374                 }
375
376                 //eDebug("new page.. (%d)", page_state);
377
378                 page->page_time_out = page_time_out;
379
380                 page->page_version_number = page_version_number;
381
382                 subtitle_page_region **r = &page->page_regions;
383
384                 //eDebug("%d  / %d data left", processed_length, segment_length);
385
386                         // go to last entry
387                 while (*r)
388                         r = &(*r)->next;
389
390                 if (processed_length == segment_length && !page->page_regions)
391                 {
392                         //eDebug("no regions in page.. clear screen!!");
393                         subtitle_redraw(page->page_id);
394                 }
395
396                 while (processed_length < segment_length)
397                 {
398                         subtitle_page_region *pr;
399
400                                 // append new entry to list
401                         pr = new subtitle_page_region;
402                         pr->next = 0;
403                         *r = pr;
404                         r = &pr->next;
405
406                         pr->region_id = *segment++; processed_length++;
407                         segment++; processed_length++;
408
409                         pr->region_horizontal_address  = *segment++ << 8;
410                         pr->region_horizontal_address |= *segment++;
411                         processed_length += 2;
412
413                         pr->region_vertical_address  = *segment++ << 8;
414                         pr->region_vertical_address |= *segment++;
415                         processed_length += 2;
416
417                         //eDebug("appended active region");
418                 }
419
420                 if (processed_length != segment_length)
421                         eDebug("%d != %d", processed_length, segment_length);
422                 break;
423         }
424         case 0x11: // region composition segment
425         {
426                 int region_id = *segment++; processed_length++;
427                 int region_version_number = *segment >> 4;
428                 int region_fill_flag = (*segment >> 3) & 1;
429                 segment++; processed_length++;
430
431                         // if we didn't yet received the pcs for this page, drop the region
432                 if (!page)
433                 {
434                         eDebug("ignoring region %x, since page %02x doesn't yet exist.", region_id, page_id);
435                         break;
436                 }
437
438                 subtitle_region *region, **pregion;
439
440                 region = page->regions; pregion = &page->regions;
441
442                 while (region)
443                 {
444                         fflush(stdout);
445                         if (region->region_id == region_id)
446                                 break;
447                         pregion = &region->next;
448                         region = region->next;
449                 }
450
451                 if (!region)
452                 {
453                         //eDebug("create region !!!!!!!!!!");
454                         *pregion = region = new subtitle_region;
455                         region->next = 0;
456                         region->committed = false;
457                 }
458                 else if (region->region_version_number != region_version_number)
459                 {
460                         subtitle_region_object *objects = region->region_objects;
461                         //eDebug("unequal version %p %p", objects, objects?objects->next:objects);
462                         while (objects)
463                         {
464                                 subtitle_region_object *n = objects->next;
465                                 delete objects;
466                                 objects = n;
467                         }
468                         if (region->region_buffer)
469                         {
470                                 //eDebug("no more need of region_buffer %p", &(*region->region_buffer));
471                                 region->region_buffer=0;
472                         }
473                         region->committed = false;
474                 }
475                 else
476                         break;
477
478                 //eDebug("region %d:%d update", page_id, region_id);
479
480                 region->region_id = region_id;
481                 region->region_version_number = region_version_number;
482
483                 region->region_width  = *segment++ << 8;
484                 region->region_width |= *segment++;
485                 processed_length += 2;
486
487                 region->region_height  = *segment++ << 8;
488                 region->region_height |= *segment++;
489                 processed_length += 2;
490
491                 region->region_buffer = new gPixmap(eSize(region->region_width, region->region_height), 8);
492                 //eDebug("new region_buffer %p", &(*region->region_buffer));
493
494                 int region_level_of_compatibility, region_depth;
495
496                 region_level_of_compatibility = (*segment >> 5) & 7;
497                 region_depth = (*segment++ >> 2) & 7;
498                 region->region_depth = (subtitle_region::depth) region_depth;
499                 processed_length++;
500
501                 int CLUT_id = *segment++; processed_length++;
502
503                 region->clut_id = CLUT_id;
504
505                 int region_8bit_pixel_code, region_4bit_pixel_code, region_2bit_pixel_code;
506                 region_8bit_pixel_code = *segment++; processed_length++;
507                 region_4bit_pixel_code = *segment >> 4;
508                 region_2bit_pixel_code = (*segment++ >> 2) & 3;
509                 processed_length++;
510
511                 if (!region_fill_flag)
512                 {
513                         region_2bit_pixel_code = region_4bit_pixel_code = region_8bit_pixel_code = 0;
514                         region_fill_flag = 1;
515                 }
516
517                 if (region_fill_flag)
518                 {
519                         //eDebug("region fill region_buffer %p", &(*region->region_buffer));
520                         if (region_depth == 1)
521                                 memset(region->region_buffer->surface->data, region_2bit_pixel_code, region->region_height * region->region_width);
522                         else if (region_depth == 2)
523                                 memset(region->region_buffer->surface->data, region_4bit_pixel_code, region->region_height * region->region_width);
524                         else if (region_depth == 3)
525                                 memset(region->region_buffer->surface->data, region_8bit_pixel_code, region->region_height * region->region_width);
526                         else
527                                 eDebug("!!!! invalid depth");
528                 }
529
530                 //eDebug("region %02x, version %d, %dx%d", region->region_id, region->region_version_number, region->region_width, region->region_height);
531
532                 region->region_objects = 0;
533                 subtitle_region_object **pobject = &region->region_objects;
534
535                 while (processed_length < segment_length)
536                 {
537                         subtitle_region_object *object;
538
539                         object = new subtitle_region_object;
540
541                         *pobject = object;
542                         object->next = 0;
543                         pobject = &object->next;
544
545                         object->object_id  = *segment++ << 8;
546                         object->object_id |= *segment++; processed_length += 2;
547
548                         object->object_type = *segment >> 6;
549                         object->object_provider_flag = (*segment >> 4) & 3;
550                         object->object_horizontal_position  = (*segment++ & 0xF) << 8;
551                         object->object_horizontal_position |= *segment++;
552                         processed_length += 2;
553
554                         object->object_vertical_position  = *segment++ << 8;
555                         object->object_vertical_position |= *segment++ ;
556                         processed_length += 2;
557
558                         if ((object->object_type == 1) || (object->object_type == 2))
559                         {
560                                 object->foreground_pixel_value = *segment++;
561                                 object->background_pixel_value = *segment++;
562                                 processed_length += 2;
563                         }
564                 }
565
566                 if (processed_length != segment_length)
567                         eDebug("too less data! (%d < %d)", segment_length, processed_length);
568
569                 break;
570         }
571         case 0x12: // CLUT definition segment
572         {
573                 int CLUT_id, CLUT_version_number;
574                 subtitle_clut *clut, **pclut;
575
576                 if (!page)
577                         break;
578
579                 //eDebug("CLUT: %02x", *segment);
580                 CLUT_id = *segment++;
581
582                 CLUT_version_number = *segment++ >> 4;
583                 processed_length += 2;
584
585                 //eDebug("page %d, CLUT %02x, version %d", page->page_id, CLUT_id, CLUT_version_number);
586
587                 clut = page->cluts; pclut = &page->cluts;
588
589                 while (clut)
590                 {
591                         if (clut->clut_id == CLUT_id)
592                                 break;
593                         pclut = &clut->next;
594                         clut = clut->next;
595                 }
596
597                 if (!clut)
598                 {
599                         *pclut = clut = new subtitle_clut;
600                         clut->next = 0;
601                         clut->clut_id = CLUT_id;
602                 }
603                 else if (clut->CLUT_version_number == CLUT_version_number)
604                         break;
605
606                 clut->CLUT_version_number=CLUT_version_number;
607
608                 memset(clut->entries_2bit, 0, sizeof(clut->entries_2bit));
609                 memset(clut->entries_4bit, 0, sizeof(clut->entries_4bit));
610                 memset(clut->entries_8bit, 0, sizeof(clut->entries_8bit));
611
612                 //eDebug("new clut");
613                 while (processed_length < segment_length)
614                 {
615                         int CLUT_entry_id, entry_CLUT_flag, full_range;
616                         int v_Y, v_Cr, v_Cb, v_T;
617
618                         CLUT_entry_id = *segment++;
619                         full_range = *segment & 1;
620                         entry_CLUT_flag = (*segment++ & 0xE0) >> 5;
621                         processed_length += 2;
622
623                         if (full_range)
624                         {
625                                 v_Y  = *segment++;
626                                 v_Cr = *segment++;
627                                 v_Cb = *segment++;
628                                 v_T  = *segment++;
629                                 processed_length += 4;
630                         } else
631                         {
632                                 v_Y   = *segment & 0xFC;
633                                 v_Cr  = (*segment++ & 3) << 6;
634                                 v_Cr |= (*segment & 0xC0) >> 2;
635                                 v_Cb  = (*segment & 0x3C) << 2;
636                                 v_T   = (*segment++ & 3) << 6;
637                                 processed_length += 2;
638                         }
639
640                         if (entry_CLUT_flag & 1) // 8bit
641                         {
642                                 ASSERT(CLUT_entry_id < 256);
643                                 clut->entries_8bit[CLUT_entry_id].Y = v_Y;
644                                 clut->entries_8bit[CLUT_entry_id].Cr = v_Cr;
645                                 clut->entries_8bit[CLUT_entry_id].Cb = v_Cb;
646                                 clut->entries_8bit[CLUT_entry_id].T = v_T;
647                                 clut->entries_8bit[CLUT_entry_id].valid = 1;
648                         }
649                         if (entry_CLUT_flag & 2) // 4bit
650                         {
651                                 ASSERT(CLUT_entry_id < 16);
652                                 clut->entries_4bit[CLUT_entry_id].Y = v_Y;
653                                 clut->entries_4bit[CLUT_entry_id].Cr = v_Cr;
654                                 clut->entries_4bit[CLUT_entry_id].Cb = v_Cb;
655                                 clut->entries_4bit[CLUT_entry_id].T = v_T;
656                                 clut->entries_4bit[CLUT_entry_id].valid = 1;
657                         }
658                         if (entry_CLUT_flag & 4) // 2bit
659                         {
660                                 ASSERT(CLUT_entry_id < 4);
661                                 clut->entries_2bit[CLUT_entry_id].Y = v_Y;
662                                 clut->entries_2bit[CLUT_entry_id].Cr = v_Cr;
663                                 clut->entries_2bit[CLUT_entry_id].Cb = v_Cb;
664                                 clut->entries_2bit[CLUT_entry_id].T = v_T;
665                                 clut->entries_2bit[CLUT_entry_id].valid = 1;
666                         }
667                         //eDebug("  %04x %02x %02x %02x %02x", CLUT_entry_id, v_Y, v_Cb, v_Cr, v_T);
668                 }
669                 break;
670         }
671         case 0x13: // object data segment
672         {
673                 int object_id, object_version_number, object_coding_method, non_modifying_color_flag;
674
675                 object_id  = *segment++ << 8;
676                 object_id |= *segment++;
677                 processed_length += 2;
678
679                 object_version_number = *segment >> 4;
680                 object_coding_method  = (*segment >> 2) & 3;
681                 non_modifying_color_flag = (*segment++ >> 1) & 1;
682                 processed_length++;
683
684                 //eDebug("object id %04x, version %d, object_coding_method %d (page_id %d)", object_id, object_version_number, object_coding_method, page_id);
685
686                 if (object_coding_method == 0)
687                 {
688                         int top_field_data_blocklength, bottom_field_data_blocklength;
689                         int i, line, linep;
690
691                         top_field_data_blocklength  = *segment++ << 8;
692                         top_field_data_blocklength |= *segment++;
693
694                         bottom_field_data_blocklength  = *segment++ << 8;
695                         bottom_field_data_blocklength |= *segment++;
696                         eDebug("%d / %d bytes", top_field_data_blocklength, bottom_field_data_blocklength);
697                         processed_length += 4;
698
699                         i = 0;
700                         line = 0;
701                         linep = 0;
702                         while (i < top_field_data_blocklength)
703                         {
704                                 int len;
705                                 len = subtitle_process_pixel_data(page, object_id, &line, &linep, segment);
706                                 if (len < 0)
707                                         return -1;
708                                 segment += len;
709                                 processed_length += len;
710                                 i += len;
711                         }
712
713                         line = 1;
714                         linep = 0;
715
716                         if (bottom_field_data_blocklength)
717                         {
718                                 i = 0;
719                                 while (i < bottom_field_data_blocklength)
720                                 {
721                                         int len;
722                                         len = subtitle_process_pixel_data(page, object_id, &line, &linep, segment);
723                                         if (len < 0)
724                                                 return -1;
725                                         segment += len;
726                                         processed_length += len;
727                                         i += len;
728                                 }
729                         }
730                         else if (top_field_data_blocklength)
731                                 eDebug("!!!! unimplemented: no bottom field! (%d : %d)", top_field_data_blocklength, bottom_field_data_blocklength);
732
733                         if ((top_field_data_blocklength + bottom_field_data_blocklength) & 1)
734                         {
735                                 segment++; processed_length++;
736                         }
737                 }
738                 else if (object_coding_method == 1)
739                         eDebug("---- object_coding_method 1 unsupported!");
740
741                 break;
742         }
743         case 0x80: // end of display set segment
744         {
745                 //eDebug("end of display set segment");
746                 subtitle_redraw_all();
747         }
748         case 0xFF: // stuffing
749                 break;
750         default:
751                 eDebug("unhandled segment type %02x", segment_type);
752         }
753
754         return segment_length + 6;
755 }
756
757 void eDVBSubtitleParser::subtitle_process_pes(__u8 *pkt, int len)
758 {
759         eDebugNoNewLine("subtitle_process_pes");
760         if (!extract_pts(m_show_time, pkt))
761         {
762                 eDebug(" %lld", m_show_time);
763                 pkt += 6; len -= 6;
764                 // skip PES header
765                 pkt++; len--;
766                 pkt++; len--;
767
768                 int hdr_len = *pkt++; len--;
769
770                 pkt+=hdr_len; len-=hdr_len;
771
772                 if (*pkt != 0x20)
773                 {
774                         //eDebug("data identifier is 0x%02x, but not 0x20", *pkt);
775                         return;
776                 }
777                 pkt++; len--; // data identifier
778                 *pkt++; len--; // stream id;
779
780                 if (len <= 0)
781                 {
782                         //eDebug("no data left (%d)", len);
783                         return;
784                 }
785
786                 while (len && *pkt == 0x0F)
787                 {
788                         int l = subtitle_process_segment(pkt);
789                         if (l < 0)
790                                 break;
791                         pkt += l;
792                         len -= l;
793                 }
794                 if (len && *pkt != 0xFF)
795                         eDebug("strange data at the end");
796         }
797         else
798                 eDebug("\ndvb subtitle packet without PTS.. ignore!!");
799 }
800
801 void eDVBSubtitleParser::subtitle_redraw_all()
802 {
803 #if 1
804         subtitle_page *page = m_pages;
805         while(page)
806         {
807                 subtitle_redraw(page->page_id);
808                 page = page->next;
809         }
810 #else
811         subtitle_page *page = m_pages;
812         eDebug("----------- end of display set");
813         eDebug("active pages:");
814         while (page)
815         {
816                 eDebug("  page_id %02x", page->page_id);
817                 eDebug("  page_version_number: %d", page->page_version_number);
818                 eDebug("  active regions:");
819                 {
820                         subtitle_page_region *region = page->page_regions;
821                         while (region)
822                         {
823                                 eDebug("    region_id: %04x", region->region_id);
824                                 eDebug("    region_horizontal_address: %d", region->region_horizontal_address);
825                                 eDebug("    region_vertical_address: %d", region->region_vertical_address);
826
827                                 region = region->next;
828                         }
829                 }
830
831                 subtitle_redraw(page->page_id);
832                 eDebug("defined regions:");
833                 subtitle_region *region = page->regions;
834                 while (region)
835                 {
836                         eDebug("  region_id %04x, version %d, %dx%d", region->region_id, region->region_version_number, region->region_width, region->region_height);
837
838                         subtitle_region_object *object = region->region_objects;
839                         while (object)
840                         {
841                                 eDebug("  object %02x, type %d, %d:%d", object->object_id, object->object_type, object->object_horizontal_position, object->object_vertical_position);
842                                 object = object->next;
843                         }
844                         region = region->next;
845                 }
846                 page = page->next;
847         }
848 #endif
849 }
850
851 void eDVBSubtitleParser::subtitle_reset()
852 {
853         while (subtitle_page *page = m_pages)
854         {
855                         /* free page regions */
856                 while (page->page_regions)
857                 {
858                         subtitle_page_region *p = page->page_regions->next;
859                         delete page->page_regions;
860                         page->page_regions = p;
861                 }
862                         /* free regions */
863                 while (page->regions)
864                 {
865                         subtitle_region *region = page->regions;
866
867                         while (region->region_objects)
868                         {
869                                 subtitle_region_object *obj = region->region_objects;
870                                 region->region_objects = obj->next;
871                                 delete obj;
872                         }
873
874                         if (region->region_buffer)
875                         {
876 //                              eDebug("no more need of region_buffer 2 %p", &(*region->region_buffer));
877                                 region->region_buffer=0;
878                         }
879
880                         page->regions = region->next;
881                         delete region;
882                 }
883
884                         /* free CLUTs */
885                 while (page->cluts)
886                 {
887                         subtitle_clut *clut = page->cluts;
888                         page->cluts = clut->next;
889                         delete clut;
890                 }
891
892                 m_pages = page->next;
893                 delete page;
894         }
895 }
896
897 void eDVBSubtitleParser::subtitle_redraw(int page_id)
898 {
899         subtitle_page *page = m_pages;
900
901 //      eDebug("displaying page id %d", page_id);
902
903         while (page)
904         {
905                 if (page->page_id == page_id)
906                         break;
907                 page = page->next;
908         }
909         if (!page)
910         {
911                 //eDebug("page not found");
912                 return;
913         }
914
915         //eDebug("iterating regions..");
916                 /* iterate all regions in this pcs */
917         subtitle_page_region *region = page->page_regions;
918
919         eDVBSubtitlePage Page;
920         Page.m_show_time = m_show_time;
921         for (; region; region=region->next)
922         {
923 //              eDebug("region %d", region->region_id);
924                         /* find corresponding region */
925                 subtitle_region *reg = page->regions;
926                 while (reg)
927                 {
928                         if (reg->region_id == region->region_id)
929                                 break;
930                         reg = reg->next;
931                 }
932                 if (reg)
933                 {
934                         if (reg->committed)
935                                 continue;
936 //                      eDebug("copy region %d to %d, %d", region->region_id, region->region_horizontal_address, region->region_vertical_address);
937
938                         int x0 = region->region_horizontal_address;
939                         int y0 = region->region_vertical_address;
940
941                         if ((x0 < 0) || (y0 < 0))
942                         {
943 //                              eDebug("x0 %d, y0 %d",
944 //                                      x0, y0);
945                                 continue;
946                         }
947
948                         /* find corresponding clut */
949                         subtitle_clut *clut = page->cluts;
950                         while (clut)
951                         {
952                         //eDebug("have %d, want %d", clut->clut_id, main_clut_id);
953                                 if (clut->clut_id == reg->clut_id)
954                                         break;
955                                 clut = clut->next;
956                         }
957
958                         int clut_size = reg->region_buffer->surface->clut.colors = reg->region_depth == subtitle_region::bpp2 ?
959                                 4 : reg->region_depth == subtitle_region::bpp4 ? 16 : 256;
960
961                         reg->region_buffer->surface->clut.data = new gRGB[clut_size];
962 //                      eDebug("create clut data for region_buffer %p", &(*reg->region_buffer));
963
964                         gRGB *palette = reg->region_buffer->surface->clut.data;
965
966                         subtitle_clut_entry *entries=0;
967                         switch(reg->region_depth)
968                         {
969                                 case subtitle_region::bpp2:
970 //                                      eDebug("2BPP");
971                                         entries = clut->entries_2bit;
972                                         memset(palette, 0, 4*sizeof(gRGB));
973                                         palette[0].a = 0xFF;
974                                         palette[2].r = palette[2].g = palette[2].b = 0xFF;
975                                         palette[3].r = palette[3].g = palette[3].b = 0x80;
976                                         break;
977                                 case subtitle_region::bpp4:
978 //                                      eDebug("4BPP");
979                                         entries = clut->entries_4bit;
980                                         memset(palette, 0, 16*sizeof(gRGB));
981                                         for (int i=0; i < 16; ++i)
982                                         {
983                                                 if (!i)
984                                                         palette[i].a = 0xFF;
985                                                 else if (i & 1)
986                                                 {
987                                                         if (i & 8)
988                                                                 palette[i].r = 0x80;
989                                                         if (i & 4)
990                                                                 palette[i].g = 0x80;
991                                                         if (i & 2)
992                                                                 palette[i].b = 0x80;
993                                                 }
994                                                 else
995                                                 {
996                                                         if (i & 8)
997                                                                 palette[i].r = 0xFF;
998                                                         if (i & 4)
999                                                                 palette[i].g = 0xFF;
1000                                                         if (i & 2)
1001                                                                 palette[i].b = 0xFF;
1002                                                 }
1003                                         }
1004                                         break;
1005                                 case subtitle_region::bpp8:
1006 //                                      eDebug("8BPP");
1007                                         entries = clut->entries_8bit;
1008                                         memset(palette, 0, 16*sizeof(gRGB));
1009                                         for (int i=0; i < 256; ++i)
1010                                         {
1011                                                 switch (i & 17)
1012                                                 {
1013                                                 case 0: // b1 == 0 && b5 == 0
1014                                                         if (!(i & 14)) // b2 == 0 && b3 == 0 && b4 == 0
1015                                                         {
1016                                                                 if (!(i & 224)) // b6 == 0 && b7 == 0 && b8 == 0
1017                                                                         palette[i].a = 0xFF;
1018                                                                 else
1019                                                                 {
1020                                                                         if (i & 128) // R = 100% x b8
1021                                                                                 palette[i].r = 0xFF;
1022                                                                         if (i & 64) // G = 100% x b7
1023                                                                                 palette[i].g = 0xFF;
1024                                                                         if (i & 32) // B = 100% x b6
1025                                                                                 palette[i].b = 0xFF;
1026                                                                         palette[i].a = 0xBF; // T = 75%
1027                                                                 }
1028                                                                 break;
1029                                                         }
1030                                                         // fallthrough !!
1031                                                 case 16: // b1 == 0 && b5 == 1
1032                                                         if (i & 128) // R = 33% x b8
1033                                                                 palette[i].r = 0x55;
1034                                                         if (i & 64) // G = 33% x b7
1035                                                                 palette[i].g = 0x55;
1036                                                         if (i & 32) // B = 33% x b6
1037                                                                 palette[i].b = 0x55;
1038                                                         if (i & 8) // R + 66,7% x b4
1039                                                                 palette[i].r += 0xAA;
1040                                                         if (i & 4) // G + 66,7% x b3
1041                                                                 palette[i].g += 0xAA;
1042                                                         if (i & 2) // B + 66,7% x b2
1043                                                                 palette[i].b += 0xAA;
1044                                                         if (i & 16) // needed for fall through from case 0!!
1045                                                                 palette[i].a = 0x80; // T = 50%
1046                                                         break;
1047                                                 case 1: // b1 == 1 && b5 == 0
1048                                                         palette[i].r =
1049                                                         palette[i].g =
1050                                                         palette[i].b = 0x80; // 50%
1051                                                         // fall through!!
1052                                                 case 17: // b1 == 1 && b5 == 1
1053                                                         if (i & 128) // R += 16.7% x b8
1054                                                                 palette[i].r += 0x2A;
1055                                                         if (i & 64) // G += 16.7% x b7
1056                                                                 palette[i].g += 0x2A;
1057                                                         if (i & 32) // B += 16.7% x b6
1058                                                                 palette[i].b += 0x2A;
1059                                                         if (i & 8) // R += 33% x b4
1060                                                                 palette[i].r += 0x55;
1061                                                         if (i & 4) // G += 33% x b3
1062                                                                 palette[i].g += 0x55;
1063                                                         if (i & 2) // B += 33% x b2
1064                                                                 palette[i].b += 0x55;
1065                                                         break;
1066                                                 }
1067                                         }
1068                                         break;
1069                         }
1070
1071                         if (clut)
1072                         {
1073                                 for (int i=0; i<clut_size; ++i)
1074                                 {
1075                                         if (entries[i].valid)
1076                                         {
1077                                                 int y = entries[i].Y,
1078                                                         cr = entries[i].Cr,
1079                                                         cb = entries[i].Cb;
1080                                                 if (y > 0)
1081                                                 {
1082                                                         y -= 16;
1083                                                         cr -= 128;
1084                                                         cb -= 128;
1085                                                         palette[i].r = MAX(MIN(((298 * y            + 460 * cr) / 256), 255), 0);
1086                                                         palette[i].g = MAX(MIN(((298 * y -  55 * cb - 137 * cr) / 256), 255), 0);
1087                                                         palette[i].b = MAX(MIN(((298 * y + 543 * cb           ) / 256), 255), 0);
1088                                                         palette[i].a = (entries[i].T) & 0xFF;
1089 //                                                      eDebug("override clut entry %d RGBA %02x%02x%02x%02x", i,
1090 //                                                              palette[i].r, palette[i].g, palette[i].b, palette[i].a);
1091                                                 }
1092                                                 else
1093                                                 {
1094 //                                                      eDebug("mist %d", i);
1095                                                         palette[i].r = 0;
1096                                                         palette[i].g = 0;
1097                                                         palette[i].b = 0;
1098                                                         palette[i].a = 0xFF;
1099                                                 }
1100                                         }
1101                                 }
1102                         }
1103
1104 //                      eDebug("commit region_buffer %p", &(*reg->region_buffer));
1105                         eDVBSubtitleRegion Region;
1106                         Region.m_pixmap = reg->region_buffer;
1107                         Region.m_position.setX(x0);
1108                         Region.m_position.setY(y0);
1109                         Page.m_regions.push_back(Region);
1110                         reg->committed = true;
1111                 }
1112                 else
1113                         eDebug("region not found");
1114         }
1115         m_new_subtitle_page(Page);
1116         Page.m_regions.clear();
1117 //      eDebug("page timeout is %d", page->page_time_out);
1118 //      Page.m_show_time += (page->page_time_out * 90000);
1119 //      m_new_subtitle_page(Page);
1120         //eDebug("schon gut.");
1121 }
1122
1123 DEFINE_REF(eDVBSubtitleParser);
1124
1125 eDVBSubtitleParser::eDVBSubtitleParser(iDVBDemux *demux)
1126         :m_pages(0)
1127 {
1128         setStreamID(0xBD);
1129
1130         if (demux->createPESReader(eApp, m_pes_reader))
1131                 eDebug("failed to create dvb subtitle PES reader!");
1132         else
1133                 m_pes_reader->connectRead(slot(*this, &eDVBSubtitleParser::processData), m_read_connection);
1134 }
1135
1136 eDVBSubtitleParser::~eDVBSubtitleParser()
1137 {
1138         subtitle_reset();
1139 }
1140
1141 int eDVBSubtitleParser::stop()
1142 {
1143         if (m_pes_reader)
1144         {
1145                 eDebug("disable dvb subtitles");
1146                 return m_pes_reader->stop();
1147         }
1148         return -1;
1149 }
1150
1151 int eDVBSubtitleParser::start(int pid, int composition_page_id, int ancillary_page_id)
1152 {
1153         if (m_pes_reader)
1154         {
1155                 eDebug("start dvb subtitles on pid 0x%04x with composition_page_id %d and ancillary_page_id %d",
1156                         pid, composition_page_id, ancillary_page_id);
1157                 m_composition_page_id = composition_page_id;
1158                 m_ancillary_page_id = ancillary_page_id;
1159                 return m_pes_reader->start(pid);
1160         }
1161         return -1;
1162 }
1163
1164 void eDVBSubtitleParser::connectNewPage(const Slot1<void, const eDVBSubtitlePage&> &slot, ePtr<eConnection> &connection)
1165 {
1166         connection = new eConnection(this, m_new_subtitle_page.connect(slot));
1167 }