b6c5b7189dd1fe7400b6cb01329bd1084f6c7cd1
[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_buffer->surface->stride * 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                 page->state = page_state;
349
350                 //eDebug("page updated: old: %d, new: %d", page->page_version_number, page_version_number);
351                         // when acquisition point or mode change: remove all displayed pages.
352                 if ((page_state == 1) || (page_state == 2))
353                 {
354                         while (page->page_regions)
355                         {
356                                 subtitle_page_region *p = page->page_regions->next;
357                                 //eDebug("delete page_region %d", page->page_regions->region_id);
358                                 delete page->page_regions;
359                                 page->page_regions = p;
360                         }
361                         while (page->regions)
362                         {
363                                 subtitle_region *p = page->regions->next;
364                                 //eDebug("delete region %d", page->regions->region_id);
365                                 while(page->regions->region_objects)
366                                 {
367                                         //eDebug("delete region object");
368                                         subtitle_region_object *ob = page->regions->region_objects->next;
369                                         delete page->regions->region_objects;
370                                         page->regions->region_objects = ob;
371                                 }
372                                 delete page->regions;
373                                 page->regions = p;
374                         }
375
376                 }
377
378                 //eDebug("new page.. (%d)", page_state);
379
380                 page->page_time_out = page_time_out;
381
382                 page->page_version_number = page_version_number;
383
384                 subtitle_page_region **r = &page->page_regions;
385
386                 //eDebug("%d  / %d data left", processed_length, segment_length);
387
388                         // go to last entry
389                 while (*r)
390                         r = &(*r)->next;
391
392                 if (processed_length == segment_length && !page->page_regions)
393                 {
394                         //eDebug("no regions in page.. clear screen!!");
395                         subtitle_redraw(page->page_id);
396                 }
397
398                 while (processed_length < segment_length)
399                 {
400                         subtitle_page_region *pr;
401
402                                 // append new entry to list
403                         pr = new subtitle_page_region;
404                         pr->next = 0;
405                         *r = pr;
406                         r = &pr->next;
407
408                         pr->region_id = *segment++; processed_length++;
409                         segment++; processed_length++;
410
411                         pr->region_horizontal_address  = *segment++ << 8;
412                         pr->region_horizontal_address |= *segment++;
413                         processed_length += 2;
414
415                         pr->region_vertical_address  = *segment++ << 8;
416                         pr->region_vertical_address |= *segment++;
417                         processed_length += 2;
418
419                         //eDebug("appended active region");
420                 }
421
422                 if (processed_length != segment_length)
423                         eDebug("%d != %d", processed_length, segment_length);
424                 break;
425         }
426         case 0x11: // region composition segment
427         {
428                 int region_id = *segment++; processed_length++;
429                 int region_version_number = *segment >> 4;
430                 int region_fill_flag = (*segment >> 3) & 1;
431                 segment++; processed_length++;
432
433                         // if we didn't yet received the pcs for this page, drop the region
434                 if (!page)
435                 {
436                         eDebug("ignoring region %x, since page %02x doesn't yet exist.", region_id, page_id);
437                         break;
438                 }
439
440                 subtitle_region *region, **pregion;
441
442                 region = page->regions; pregion = &page->regions;
443
444                 while (region)
445                 {
446                         fflush(stdout);
447                         if (region->region_id == region_id)
448                                 break;
449                         pregion = &region->next;
450                         region = region->next;
451                 }
452
453                 if (!region)
454                 {
455                         //eDebug("create region !!!!!!!!!!");
456                         *pregion = region = new subtitle_region;
457                         region->next = 0;
458                         region->committed = false;
459                 }
460                 else if (region->region_version_number != region_version_number)
461                 {
462                         subtitle_region_object *objects = region->region_objects;
463                         //eDebug("unequal version %p %p", objects, objects?objects->next:objects);
464                         while (objects)
465                         {
466                                 subtitle_region_object *n = objects->next;
467                                 delete objects;
468                                 objects = n;
469                         }
470                         if (region->region_buffer)
471                         {
472                                 //eDebug("no more need of region_buffer %p", &(*region->region_buffer));
473                                 region->region_buffer=0;
474                         }
475                         region->committed = false;
476                 }
477                 else
478                         break;
479
480                 //eDebug("region %d:%d update", page_id, region_id);
481
482                 region->region_id = region_id;
483                 region->region_version_number = region_version_number;
484
485                 region->region_width  = *segment++ << 8;
486                 region->region_width |= *segment++;
487                 processed_length += 2;
488
489                 region->region_height  = *segment++ << 8;
490                 region->region_height |= *segment++;
491                 processed_length += 2;
492
493                 region->region_buffer = new gPixmap(eSize(region->region_width, region->region_height), 8, 1);
494                 //eDebug("new region_buffer %p", &(*region->region_buffer));
495
496                 int region_level_of_compatibility, region_depth;
497
498                 region_level_of_compatibility = (*segment >> 5) & 7;
499                 region_depth = (*segment++ >> 2) & 7;
500                 region->region_depth = (subtitle_region::depth) region_depth;
501                 processed_length++;
502
503                 int CLUT_id = *segment++; processed_length++;
504
505                 region->clut_id = CLUT_id;
506
507                 int region_8bit_pixel_code, region_4bit_pixel_code, region_2bit_pixel_code;
508                 region_8bit_pixel_code = *segment++; processed_length++;
509                 region_4bit_pixel_code = *segment >> 4;
510                 region_2bit_pixel_code = (*segment++ >> 2) & 3;
511                 processed_length++;
512
513                 if (!region_fill_flag)
514                 {
515                         region_2bit_pixel_code = region_4bit_pixel_code = region_8bit_pixel_code = 0;
516                         region_fill_flag = 1;
517                 }
518
519                 if (region_fill_flag)
520                 {
521                         //eDebug("region fill region_buffer %p", &(*region->region_buffer));
522                         if (region_depth == 1)
523                                 memset(region->region_buffer->surface->data, region_2bit_pixel_code, region->region_height * region->region_width);
524                         else if (region_depth == 2)
525                                 memset(region->region_buffer->surface->data, region_4bit_pixel_code, region->region_height * region->region_width);
526                         else if (region_depth == 3)
527                                 memset(region->region_buffer->surface->data, region_8bit_pixel_code, region->region_height * region->region_width);
528                         else
529                                 eDebug("!!!! invalid depth");
530                 }
531
532                 //eDebug("region %02x, version %d, %dx%d", region->region_id, region->region_version_number, region->region_width, region->region_height);
533
534                 region->region_objects = 0;
535                 subtitle_region_object **pobject = &region->region_objects;
536
537                 while (processed_length < segment_length)
538                 {
539                         subtitle_region_object *object;
540
541                         object = new subtitle_region_object;
542
543                         *pobject = object;
544                         object->next = 0;
545                         pobject = &object->next;
546
547                         object->object_id  = *segment++ << 8;
548                         object->object_id |= *segment++; processed_length += 2;
549
550                         object->object_type = *segment >> 6;
551                         object->object_provider_flag = (*segment >> 4) & 3;
552                         object->object_horizontal_position  = (*segment++ & 0xF) << 8;
553                         object->object_horizontal_position |= *segment++;
554                         processed_length += 2;
555
556                         object->object_vertical_position  = *segment++ << 8;
557                         object->object_vertical_position |= *segment++ ;
558                         processed_length += 2;
559
560                         if ((object->object_type == 1) || (object->object_type == 2))
561                         {
562                                 object->foreground_pixel_value = *segment++;
563                                 object->background_pixel_value = *segment++;
564                                 processed_length += 2;
565                         }
566                 }
567
568                 if (processed_length != segment_length)
569                         eDebug("too less data! (%d < %d)", segment_length, processed_length);
570
571                 break;
572         }
573         case 0x12: // CLUT definition segment
574         {
575                 int CLUT_id, CLUT_version_number;
576                 subtitle_clut *clut, **pclut;
577
578                 if (!page)
579                         break;
580
581                 //eDebug("CLUT: %02x", *segment);
582                 CLUT_id = *segment++;
583
584                 CLUT_version_number = *segment++ >> 4;
585                 processed_length += 2;
586
587                 //eDebug("page %d, CLUT %02x, version %d", page->page_id, CLUT_id, CLUT_version_number);
588
589                 clut = page->cluts; pclut = &page->cluts;
590
591                 while (clut)
592                 {
593                         if (clut->clut_id == CLUT_id)
594                                 break;
595                         pclut = &clut->next;
596                         clut = clut->next;
597                 }
598
599                 if (!clut)
600                 {
601                         *pclut = clut = new subtitle_clut;
602                         clut->next = 0;
603                         clut->clut_id = CLUT_id;
604                 }
605                 else if (clut->CLUT_version_number == CLUT_version_number)
606                         break;
607
608                 clut->CLUT_version_number=CLUT_version_number;
609
610                 memset(clut->entries_2bit, 0, sizeof(clut->entries_2bit));
611                 memset(clut->entries_4bit, 0, sizeof(clut->entries_4bit));
612                 memset(clut->entries_8bit, 0, sizeof(clut->entries_8bit));
613
614                 //eDebug("new clut");
615                 while (processed_length < segment_length)
616                 {
617                         int CLUT_entry_id, entry_CLUT_flag, full_range;
618                         int v_Y, v_Cr, v_Cb, v_T;
619
620                         CLUT_entry_id = *segment++;
621                         full_range = *segment & 1;
622                         entry_CLUT_flag = (*segment++ & 0xE0) >> 5;
623                         processed_length += 2;
624
625                         if (full_range)
626                         {
627                                 v_Y  = *segment++;
628                                 v_Cr = *segment++;
629                                 v_Cb = *segment++;
630                                 v_T  = *segment++;
631                                 processed_length += 4;
632                         } else
633                         {
634                                 v_Y   = *segment & 0xFC;
635                                 v_Cr  = (*segment++ & 3) << 6;
636                                 v_Cr |= (*segment & 0xC0) >> 2;
637                                 v_Cb  = (*segment & 0x3C) << 2;
638                                 v_T   = (*segment++ & 3) << 6;
639                                 processed_length += 2;
640                         }
641
642                         if (entry_CLUT_flag & 1) // 8bit
643                         {
644                                 ASSERT(CLUT_entry_id < 256);
645                                 clut->entries_8bit[CLUT_entry_id].Y = v_Y;
646                                 clut->entries_8bit[CLUT_entry_id].Cr = v_Cr;
647                                 clut->entries_8bit[CLUT_entry_id].Cb = v_Cb;
648                                 clut->entries_8bit[CLUT_entry_id].T = v_T;
649                                 clut->entries_8bit[CLUT_entry_id].valid = 1;
650                         }
651                         if (entry_CLUT_flag & 2) // 4bit
652                         {
653                                 ASSERT(CLUT_entry_id < 16);
654                                 clut->entries_4bit[CLUT_entry_id].Y = v_Y;
655                                 clut->entries_4bit[CLUT_entry_id].Cr = v_Cr;
656                                 clut->entries_4bit[CLUT_entry_id].Cb = v_Cb;
657                                 clut->entries_4bit[CLUT_entry_id].T = v_T;
658                                 clut->entries_4bit[CLUT_entry_id].valid = 1;
659                         }
660                         if (entry_CLUT_flag & 4) // 2bit
661                         {
662                                 ASSERT(CLUT_entry_id < 4);
663                                 clut->entries_2bit[CLUT_entry_id].Y = v_Y;
664                                 clut->entries_2bit[CLUT_entry_id].Cr = v_Cr;
665                                 clut->entries_2bit[CLUT_entry_id].Cb = v_Cb;
666                                 clut->entries_2bit[CLUT_entry_id].T = v_T;
667                                 clut->entries_2bit[CLUT_entry_id].valid = 1;
668                         }
669                         //eDebug("  %04x %02x %02x %02x %02x", CLUT_entry_id, v_Y, v_Cb, v_Cr, v_T);
670                 }
671                 break;
672         }
673         case 0x13: // object data segment
674         {
675                 int object_id, object_version_number, object_coding_method, non_modifying_color_flag;
676
677                 object_id  = *segment++ << 8;
678                 object_id |= *segment++;
679                 processed_length += 2;
680
681                 object_version_number = *segment >> 4;
682                 object_coding_method  = (*segment >> 2) & 3;
683                 non_modifying_color_flag = (*segment++ >> 1) & 1;
684                 processed_length++;
685
686                 //eDebug("object id %04x, version %d, object_coding_method %d (page_id %d)", object_id, object_version_number, object_coding_method, page_id);
687
688                 if (object_coding_method == 0)
689                 {
690                         int top_field_data_blocklength, bottom_field_data_blocklength;
691                         int i, line, linep;
692
693                         top_field_data_blocklength  = *segment++ << 8;
694                         top_field_data_blocklength |= *segment++;
695
696                         bottom_field_data_blocklength  = *segment++ << 8;
697                         bottom_field_data_blocklength |= *segment++;
698                         eDebug("%d / %d bytes", top_field_data_blocklength, bottom_field_data_blocklength);
699                         processed_length += 4;
700
701                         i = 0;
702                         line = 0;
703                         linep = 0;
704                         while (i < top_field_data_blocklength)
705                         {
706                                 int len;
707                                 len = subtitle_process_pixel_data(page, object_id, &line, &linep, segment);
708                                 if (len < 0)
709                                         return -1;
710                                 segment += len;
711                                 processed_length += len;
712                                 i += len;
713                         }
714
715                         line = 1;
716                         linep = 0;
717
718                         if (bottom_field_data_blocklength)
719                         {
720                                 i = 0;
721                                 while (i < bottom_field_data_blocklength)
722                                 {
723                                         int len;
724                                         len = subtitle_process_pixel_data(page, object_id, &line, &linep, segment);
725                                         if (len < 0)
726                                                 return -1;
727                                         segment += len;
728                                         processed_length += len;
729                                         i += len;
730                                 }
731                         }
732                         else if (top_field_data_blocklength)
733                                 eDebug("!!!! unimplemented: no bottom field! (%d : %d)", top_field_data_blocklength, bottom_field_data_blocklength);
734
735                         if ((top_field_data_blocklength + bottom_field_data_blocklength) & 1)
736                         {
737                                 segment++; processed_length++;
738                         }
739                 }
740                 else if (object_coding_method == 1)
741                         eDebug("---- object_coding_method 1 unsupported!");
742
743                 break;
744         }
745         case 0x80: // end of display set segment
746         {
747                 //eDebug("end of display set segment");
748                 subtitle_redraw_all();
749         }
750         case 0xFF: // stuffing
751                 break;
752         default:
753                 eDebug("unhandled segment type %02x", segment_type);
754         }
755
756         return segment_length + 6;
757 }
758
759 void eDVBSubtitleParser::subtitle_process_pes(__u8 *pkt, int len)
760 {
761         eDebugNoNewLine("subtitle_process_pes");
762         if (!extract_pts(m_show_time, pkt))
763         {
764                 eDebug(" %lld", m_show_time);
765                 pkt += 6; len -= 6;
766                 // skip PES header
767                 pkt++; len--;
768                 pkt++; len--;
769
770                 int hdr_len = *pkt++; len--;
771
772                 pkt+=hdr_len; len-=hdr_len;
773
774                 if (*pkt != 0x20)
775                 {
776                         //eDebug("data identifier is 0x%02x, but not 0x20", *pkt);
777                         return;
778                 }
779                 pkt++; len--; // data identifier
780                 *pkt++; len--; // stream id;
781
782                 if (len <= 0)
783                 {
784                         //eDebug("no data left (%d)", len);
785                         return;
786                 }
787
788                 while (len && *pkt == 0x0F)
789                 {
790                         int l = subtitle_process_segment(pkt);
791                         if (l < 0)
792                                 break;
793                         pkt += l;
794                         len -= l;
795                 }
796                 if (len && *pkt != 0xFF)
797                         eDebug("strange data at the end");
798         }
799         else
800                 eDebug("\ndvb subtitle packet without PTS.. ignore!!");
801 }
802
803 void eDVBSubtitleParser::subtitle_redraw_all()
804 {
805 #if 1
806         subtitle_page *page = m_pages;
807         while(page)
808         {
809                 if (page->state != 0)
810                         subtitle_redraw(page->page_id);
811                 page = page->next;
812         }
813 #else
814         subtitle_page *page = m_pages;
815         eDebug("----------- end of display set");
816         eDebug("active pages:");
817         while (page)
818         {
819                 eDebug("  page_id %02x", page->page_id);
820                 eDebug("  page_version_number: %d", page->page_version_number);
821                 eDebug("  active regions:");
822                 {
823                         subtitle_page_region *region = page->page_regions;
824                         while (region)
825                         {
826                                 eDebug("    region_id: %04x", region->region_id);
827                                 eDebug("    region_horizontal_address: %d", region->region_horizontal_address);
828                                 eDebug("    region_vertical_address: %d", region->region_vertical_address);
829
830                                 region = region->next;
831                         }
832                 }
833
834                 subtitle_redraw(page->page_id);
835                 eDebug("defined regions:");
836                 subtitle_region *region = page->regions;
837                 while (region)
838                 {
839                         eDebug("  region_id %04x, version %d, %dx%d", region->region_id, region->region_version_number, region->region_width, region->region_height);
840
841                         subtitle_region_object *object = region->region_objects;
842                         while (object)
843                         {
844                                 eDebug("  object %02x, type %d, %d:%d", object->object_id, object->object_type, object->object_horizontal_position, object->object_vertical_position);
845                                 object = object->next;
846                         }
847                         region = region->next;
848                 }
849                 page = page->next;
850         }
851 #endif
852 }
853
854 void eDVBSubtitleParser::subtitle_reset()
855 {
856         while (subtitle_page *page = m_pages)
857         {
858                         /* free page regions */
859                 while (page->page_regions)
860                 {
861                         subtitle_page_region *p = page->page_regions->next;
862                         delete page->page_regions;
863                         page->page_regions = p;
864                 }
865                         /* free regions */
866                 while (page->regions)
867                 {
868                         subtitle_region *region = page->regions;
869
870                         while (region->region_objects)
871                         {
872                                 subtitle_region_object *obj = region->region_objects;
873                                 region->region_objects = obj->next;
874                                 delete obj;
875                         }
876
877                         if (region->region_buffer)
878                         {
879 //                              eDebug("no more need of region_buffer 2 %p", &(*region->region_buffer));
880                                 region->region_buffer=0;
881                         }
882
883                         page->regions = region->next;
884                         delete region;
885                 }
886
887                         /* free CLUTs */
888                 while (page->cluts)
889                 {
890                         subtitle_clut *clut = page->cluts;
891                         page->cluts = clut->next;
892                         delete clut;
893                 }
894
895                 m_pages = page->next;
896                 delete page;
897         }
898 }
899
900 void eDVBSubtitleParser::subtitle_redraw(int page_id)
901 {
902         subtitle_page *page = m_pages;
903
904 //      eDebug("displaying page id %d", page_id);
905
906         while (page)
907         {
908                 if (page->page_id == page_id)
909                         break;
910                 page = page->next;
911         }
912         if (!page)
913         {
914                 //eDebug("page not found");
915                 return;
916         }
917
918         //eDebug("iterating regions..");
919                 /* iterate all regions in this pcs */
920         subtitle_page_region *region = page->page_regions;
921
922         eDVBSubtitlePage Page;
923         Page.m_show_time = m_show_time;
924         for (; region; region=region->next)
925         {
926 //              eDebug("region %d", region->region_id);
927                         /* find corresponding region */
928                 subtitle_region *reg = page->regions;
929                 while (reg)
930                 {
931                         if (reg->region_id == region->region_id)
932                                 break;
933                         reg = reg->next;
934                 }
935                 if (reg)
936                 {
937                         if (reg->committed)
938                                 continue;
939 //                      eDebug("copy region %d to %d, %d", region->region_id, region->region_horizontal_address, region->region_vertical_address);
940
941                         int x0 = region->region_horizontal_address;
942                         int y0 = region->region_vertical_address;
943
944                         if ((x0 < 0) || (y0 < 0))
945                         {
946 //                              eDebug("x0 %d, y0 %d",
947 //                                      x0, y0);
948                                 continue;
949                         }
950
951                         /* find corresponding clut */
952                         subtitle_clut *clut = page->cluts;
953                         while (clut)
954                         {
955                         //eDebug("have %d, want %d", clut->clut_id, main_clut_id);
956                                 if (clut->clut_id == reg->clut_id)
957                                         break;
958                                 clut = clut->next;
959                         }
960
961                         int clut_size = reg->region_buffer->surface->clut.colors = reg->region_depth == subtitle_region::bpp2 ?
962                                 4 : reg->region_depth == subtitle_region::bpp4 ? 16 : 256;
963
964                         reg->region_buffer->surface->clut.data = new gRGB[clut_size];
965 //                      eDebug("create clut data for region_buffer %p", &(*reg->region_buffer));
966
967                         gRGB *palette = reg->region_buffer->surface->clut.data;
968
969                         subtitle_clut_entry *entries=0;
970                         switch(reg->region_depth)
971                         {
972                                 case subtitle_region::bpp2:
973 //                                      eDebug("2BPP");
974                                         entries = clut->entries_2bit;
975                                         memset(palette, 0, 4*sizeof(gRGB));
976                                         palette[0].a = 0xFF;
977                                         palette[2].r = palette[2].g = palette[2].b = 0xFF;
978                                         palette[3].r = palette[3].g = palette[3].b = 0x80;
979                                         break;
980                                 case subtitle_region::bpp4:
981 //                                      eDebug("4BPP");
982                                         entries = clut->entries_4bit;
983                                         memset(palette, 0, 16*sizeof(gRGB));
984                                         for (int i=0; i < 16; ++i)
985                                         {
986                                                 if (!i)
987                                                         palette[i].a = 0xFF;
988                                                 else if (i & 1)
989                                                 {
990                                                         if (i & 8)
991                                                                 palette[i].r = 0x80;
992                                                         if (i & 4)
993                                                                 palette[i].g = 0x80;
994                                                         if (i & 2)
995                                                                 palette[i].b = 0x80;
996                                                 }
997                                                 else
998                                                 {
999                                                         if (i & 8)
1000                                                                 palette[i].r = 0xFF;
1001                                                         if (i & 4)
1002                                                                 palette[i].g = 0xFF;
1003                                                         if (i & 2)
1004                                                                 palette[i].b = 0xFF;
1005                                                 }
1006                                         }
1007                                         break;
1008                                 case subtitle_region::bpp8:
1009 //                                      eDebug("8BPP");
1010                                         entries = clut->entries_8bit;
1011                                         memset(palette, 0, 16*sizeof(gRGB));
1012                                         for (int i=0; i < 256; ++i)
1013                                         {
1014                                                 switch (i & 17)
1015                                                 {
1016                                                 case 0: // b1 == 0 && b5 == 0
1017                                                         if (!(i & 14)) // b2 == 0 && b3 == 0 && b4 == 0
1018                                                         {
1019                                                                 if (!(i & 224)) // b6 == 0 && b7 == 0 && b8 == 0
1020                                                                         palette[i].a = 0xFF;
1021                                                                 else
1022                                                                 {
1023                                                                         if (i & 128) // R = 100% x b8
1024                                                                                 palette[i].r = 0xFF;
1025                                                                         if (i & 64) // G = 100% x b7
1026                                                                                 palette[i].g = 0xFF;
1027                                                                         if (i & 32) // B = 100% x b6
1028                                                                                 palette[i].b = 0xFF;
1029                                                                         palette[i].a = 0xBF; // T = 75%
1030                                                                 }
1031                                                                 break;
1032                                                         }
1033                                                         // fallthrough !!
1034                                                 case 16: // b1 == 0 && b5 == 1
1035                                                         if (i & 128) // R = 33% x b8
1036                                                                 palette[i].r = 0x55;
1037                                                         if (i & 64) // G = 33% x b7
1038                                                                 palette[i].g = 0x55;
1039                                                         if (i & 32) // B = 33% x b6
1040                                                                 palette[i].b = 0x55;
1041                                                         if (i & 8) // R + 66,7% x b4
1042                                                                 palette[i].r += 0xAA;
1043                                                         if (i & 4) // G + 66,7% x b3
1044                                                                 palette[i].g += 0xAA;
1045                                                         if (i & 2) // B + 66,7% x b2
1046                                                                 palette[i].b += 0xAA;
1047                                                         if (i & 16) // needed for fall through from case 0!!
1048                                                                 palette[i].a = 0x80; // T = 50%
1049                                                         break;
1050                                                 case 1: // b1 == 1 && b5 == 0
1051                                                         palette[i].r =
1052                                                         palette[i].g =
1053                                                         palette[i].b = 0x80; // 50%
1054                                                         // fall through!!
1055                                                 case 17: // b1 == 1 && b5 == 1
1056                                                         if (i & 128) // R += 16.7% x b8
1057                                                                 palette[i].r += 0x2A;
1058                                                         if (i & 64) // G += 16.7% x b7
1059                                                                 palette[i].g += 0x2A;
1060                                                         if (i & 32) // B += 16.7% x b6
1061                                                                 palette[i].b += 0x2A;
1062                                                         if (i & 8) // R += 33% x b4
1063                                                                 palette[i].r += 0x55;
1064                                                         if (i & 4) // G += 33% x b3
1065                                                                 palette[i].g += 0x55;
1066                                                         if (i & 2) // B += 33% x b2
1067                                                                 palette[i].b += 0x55;
1068                                                         break;
1069                                                 }
1070                                         }
1071                                         break;
1072                         }
1073
1074                         if (clut)
1075                         {
1076                                 for (int i=0; i<clut_size; ++i)
1077                                 {
1078                                         if (entries[i].valid)
1079                                         {
1080                                                 int y = entries[i].Y,
1081                                                         cr = entries[i].Cr,
1082                                                         cb = entries[i].Cb;
1083                                                 if (y > 0)
1084                                                 {
1085                                                         y -= 16;
1086                                                         cr -= 128;
1087                                                         cb -= 128;
1088                                                         palette[i].r = MAX(MIN(((298 * y            + 460 * cr) / 256), 255), 0);
1089                                                         palette[i].g = MAX(MIN(((298 * y -  55 * cb - 137 * cr) / 256), 255), 0);
1090                                                         palette[i].b = MAX(MIN(((298 * y + 543 * cb           ) / 256), 255), 0);
1091                                                         palette[i].a = (entries[i].T) & 0xFF;
1092 //                                                      eDebug("override clut entry %d RGBA %02x%02x%02x%02x", i,
1093 //                                                              palette[i].r, palette[i].g, palette[i].b, palette[i].a);
1094                                                 }
1095                                                 else
1096                                                 {
1097 //                                                      eDebug("mist %d", i);
1098                                                         palette[i].r = 0;
1099                                                         palette[i].g = 0;
1100                                                         palette[i].b = 0;
1101                                                         palette[i].a = 0xFF;
1102                                                 }
1103                                         }
1104                                 }
1105                         }
1106
1107 //                      eDebug("commit region_buffer %p", &(*reg->region_buffer));
1108                         eDVBSubtitleRegion Region;
1109                         Region.m_pixmap = reg->region_buffer;
1110                         Region.m_position.setX(x0);
1111                         Region.m_position.setY(y0);
1112                         Page.m_regions.push_back(Region);
1113                         reg->committed = true;
1114                 }
1115                 else
1116                         eDebug("region not found");
1117         }
1118         m_new_subtitle_page(Page);
1119         Page.m_regions.clear();
1120 //      eDebug("page timeout is %d", page->page_time_out);
1121 //      Page.m_show_time += (page->page_time_out * 90000);
1122 //      m_new_subtitle_page(Page);
1123         //eDebug("schon gut.");
1124 }
1125
1126 DEFINE_REF(eDVBSubtitleParser);
1127
1128 eDVBSubtitleParser::eDVBSubtitleParser(iDVBDemux *demux)
1129         :m_pages(0)
1130 {
1131         setStreamID(0xBD);
1132
1133         if (demux->createPESReader(eApp, m_pes_reader))
1134                 eDebug("failed to create dvb subtitle PES reader!");
1135         else
1136                 m_pes_reader->connectRead(slot(*this, &eDVBSubtitleParser::processData), m_read_connection);
1137 }
1138
1139 eDVBSubtitleParser::~eDVBSubtitleParser()
1140 {
1141         subtitle_reset();
1142 }
1143
1144 int eDVBSubtitleParser::stop()
1145 {
1146         if (m_pes_reader)
1147         {
1148                 eDebug("disable dvb subtitles");
1149                 return m_pes_reader->stop();
1150         }
1151         return -1;
1152 }
1153
1154 int eDVBSubtitleParser::start(int pid, int composition_page_id, int ancillary_page_id)
1155 {
1156         if (m_pes_reader)
1157         {
1158                 eDebug("start dvb subtitles on pid 0x%04x with composition_page_id %d and ancillary_page_id %d",
1159                         pid, composition_page_id, ancillary_page_id);
1160                 m_composition_page_id = composition_page_id;
1161                 m_ancillary_page_id = ancillary_page_id;
1162                 return m_pes_reader->start(pid);
1163         }
1164         return -1;
1165 }
1166
1167 void eDVBSubtitleParser::connectNewPage(const Slot1<void, const eDVBSubtitlePage&> &slot, ePtr<eConnection> &connection)
1168 {
1169         connection = new eConnection(this, m_new_subtitle_page.connect(slot));
1170 }