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                 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);
 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 }