vobsub.c

   1 /*
   2  * Some code freely inspired from VobSub <URL:http://vobsub.edensrising.com>,
   3  * with kind permission from Gabest <gabest@freemail.hu>
   4  */
   5 #include <ctype.h>
   6 #include <errno.h>
   7 #include <limits.h>
   8 #include <stddef.h>
   9 #include <stdio.h>
  10 #include <stdlib.h>
  11 #include <string.h>
  12 #include <fcntl.h>
  13 #include <unistd.h>
  14 #include <sys/stat.h>
  15 #include <sys/types.h>
  16
  17 #include "config.h"
  18 #include "version.h"
  19
  20 #include "vobsub.h"
  21 #include "spudec.h"
  22 #include "mp_msg.h"
  23 #ifdef USE_UNRAR_EXEC
  24 #include "unrar_exec.h"
  25 #elif defined(USE_UNRARLIB)
  26 #include "unrarlib.h"
  27 #endif
  28 #include "libavutil/common.h"
  29
  30 extern int vobsub_id;
  31 // Record the original -vobsubid set by commandline, since vobsub_id will be
  32 // overridden if slang match any of vobsub streams.
  33 static int vobsubid = -2;
  34
  35 /**********************************************************************
  36  * RAR stream handling
  37  * The RAR file must have the same basename as the file to open
  38  * See <URL:http://www.unrarlib.org/>
  39  **********************************************************************/
  40 #if defined(USE_UNRARLIB) || defined(USE_UNRAR_EXEC)
  41 typedef struct {
  42     FILE *file;
  43     unsigned char *data;
  44     unsigned long size;
  45     unsigned long pos;
  46 } rar_stream_t;
  47 static rar_stream_t *
  48 rar_open(const char *const filename, const char *const mode)
  49 {
  50     rar_stream_t *stream;
  51     /* unrarlib can only read */
  52     if (strcmp("r", mode) && strcmp("rb", mode)) {
  53         errno = EINVAL;
  54         return NULL;
  55     }
  56     stream = malloc(sizeof(rar_stream_t));
  57     if (stream == NULL)
  58         return NULL;
  59     /* first try normal access */
  60     stream->file = fopen(filename, mode);
  61     if (stream->file == NULL) {
  62         char *rar_filename;
  63         const char *p;
  64         int rc = 0;
  65         /* Guess the RAR archive filename */
  66         rar_filename = NULL;
  67         p = strrchr(filename, '.');
  68         if (p) {
  69             ptrdiff_t l = p - filename;
  70             rar_filename = malloc(l + 5);
  71             if (rar_filename == NULL) {
  72                 free(stream);
  73                 return NULL;
  74             }
  75             strncpy(rar_filename, filename, l);
  76             strcpy(rar_filename + l, ".rar");
  77         }
  78         else {
  79             rar_filename = malloc(strlen(filename) + 5);
  80             if (rar_filename == NULL) {
  81                 free(stream);
  82                 return NULL;
  83             }
  84             strcpy(rar_filename, filename);
  85             strcat(rar_filename, ".rar");
  86         }
  87         /* get rid of the path if there is any */
  88         if ((p = strrchr(filename, '/')) == NULL) {
  89                 p = filename;
  90         } else {
  91                 p++;
  92         }
  93 #ifdef USE_UNRAR_EXEC
  94         rc = unrar_exec_get(&stream->data, &stream->size, p, rar_filename);
  95 #endif
  96 #ifdef USE_UNRARLIB
  97         if (!rc)
  98         rc = urarlib_get(&stream->data, &stream->size, (char*) p, rar_filename, "");
  99 #endif
 100         if (!rc) {
 101             /* There is no matching filename in the archive. However, sometimes
 102              * the files we are looking for have been given arbitrary names in the archive.
 103              * Let's look for a file with an exact match in the extension only. */
 104             int i, num_files = -1, name_len;
 105             ArchiveList_struct *list, *lp;
 106 #ifdef USE_UNRARLIB
 107             /* the cast in the next line is a hack to overcome a design flaw (IMHO) in unrarlib */
 108             num_files = urarlib_list (rar_filename, (ArchiveList_struct *)&list);
 109 #endif
 110 #ifdef USE_UNRAR_EXEC
 111             if (num_files <= 0)
 112             num_files = unrar_exec_list(rar_filename, &list);
 113 #endif
 114             if (num_files > 0) {
 115                 char *demanded_ext;
 116                 demanded_ext = strrchr (p, '.');
 117                 if (demanded_ext) {
 118                     int demanded_ext_len = strlen (demanded_ext);
 119                     for (i=0, lp=list; i<num_files; i++, lp=lp->next) {
 120                         name_len = strlen (lp->item.Name);
 121                         if (name_len >= demanded_ext_len && !strcasecmp (lp->item.Name + name_len - demanded_ext_len, demanded_ext)) {
 122 #ifdef USE_UNRAR_EXEC
 123                             rc = unrar_exec_get(&stream->data, &stream->size,
 124                                                 lp->item.Name, rar_filename);
 125                             if (rc) break;
 126 #endif
 127 #ifdef USE_UNRARLIB
 128                             rc = urarlib_get(&stream->data, &stream->size,
 129                                              lp->item.Name, rar_filename, "");
 130                             if (rc) break;
 131 #endif
 132                         }
 133                     }
 134                 }
 135 #ifdef USE_UNRARLIB
 136                 urarlib_freelist (list);
 137 #else
 138                 unrar_exec_freelist(list);
 139 #endif
 140             }
 141             if (!rc) {
 142                 free(rar_filename);
 143                 free(stream);
 144                 return NULL;
 145             }
 146         }
 147
 148         free(rar_filename);
 149         stream->pos = 0;
 150     }
 151     return stream;
 152 }
 153
 154 static int
 155 rar_close(rar_stream_t *stream)
 156 {
 157     if (stream->file)
 158         return fclose(stream->file);
 159     free(stream->data);
 160     return 0;
 161 }
 162
 163 static int
 164 rar_eof(rar_stream_t *stream)
 165 {
 166     if (stream->file)
 167         return feof(stream->file);
 168     return stream->pos >= stream->size;
 169 }
 170
 171 static long
 172 rar_tell(rar_stream_t *stream)
 173 {
 174     if (stream->file)
 175         return ftell(stream->file);
 176     return stream->pos;
 177 }
 178
 179 static int
 180 rar_seek(rar_stream_t *stream, long offset, int whence)
 181 {
 182     if (stream->file)
 183         return fseek(stream->file, offset, whence);
 184     switch (whence) {
 185     case SEEK_SET:
 186         if (offset < 0) {
 187             errno = EINVAL;
 188             return -1;
 189         }
 190         stream->pos = offset;
 191         break;
 192     case SEEK_CUR:
 193         if (offset < 0 && stream->pos < (unsigned long) -offset) {
 194             errno = EINVAL;
 195             return -1;
 196         }
 197         stream->pos += offset;
 198         break;
 199     case SEEK_END:
 200         if (offset < 0 && stream->size < (unsigned long) -offset) {
 201             errno = EINVAL;
 202             return -1;
 203         }
 204         stream->pos = stream->size + offset;
 205         break;
 206     default:
 207         errno = EINVAL;
 208         return -1;
 209     }
 210     return 0;
 211 }
 212
 213 static int
 214 rar_getc(rar_stream_t *stream)
 215 {
 216     if (stream->file)
 217         return getc(stream->file);
 218     if (rar_eof(stream))
 219         return EOF;
 220     return stream->data[stream->pos++];
 221 }
 222
 223 static size_t
 224 rar_read(void *ptr, size_t size, size_t nmemb, rar_stream_t *stream)
 225 {
 226     size_t res;
 227     unsigned long remain;
 228     if (stream->file)
 229         return fread(ptr, size, nmemb, stream->file);
 230     if (rar_eof(stream))
 231         return 0;
 232     res = size * nmemb;
 233     remain = stream->size - stream->pos;
 234     if (res > remain)
 235         res = remain / size * size;
 236     memcpy(ptr, stream->data + stream->pos, res);
 237     stream->pos += res;
 238     res /= size;
 239     return res;
 240 }
 241
 242 #else
 243 typedef FILE rar_stream_t;
 244 #define rar_open        fopen
 245 #define rar_close       fclose
 246 #define rar_eof         feof
 247 #define rar_tell        ftell
 248 #define rar_seek        fseek
 249 #define rar_getc        getc
 250 #define rar_read        fread
 251 #endif
 252
 253 /**********************************************************************/
 254
 255 static ssize_t
 256 vobsub_getline(char **lineptr, size_t *n, rar_stream_t *stream)
 257 {
 258     size_t res = 0;
 259     int c;
 260     if (*lineptr == NULL) {
 261         *lineptr = malloc(4096);
 262         if (*lineptr)
 263             *n = 4096;
 264     }
 265     else if (*n == 0) {
 266         char *tmp = realloc(*lineptr, 4096);
 267         if (tmp) {
 268             *lineptr = tmp;
 269             *n = 4096;
 270         }
 271     }
 272     if (*lineptr == NULL || *n == 0)
 273         return -1;
 274
 275     for (c = rar_getc(stream); c != EOF; c = rar_getc(stream)) {
 276         if (res + 1 >= *n) {
 277             char *tmp = realloc(*lineptr, *n * 2);
 278             if (tmp == NULL)
 279                 return -1;
 280             *lineptr = tmp;
 281             *n *= 2;
 282         }
 283         (*lineptr)[res++] = c;
 284         if (c == '\n') {
 285             (*lineptr)[res] = 0;
 286             return res;
 287         }
 288     }
 289     if (res == 0)
 290         return -1;
 291     (*lineptr)[res] = 0;
 292     return res;
 293 }
 294
 295 /**********************************************************************
 296  * MPEG parsing
 297  **********************************************************************/
 298
 299 typedef struct {
 300     rar_stream_t *stream;
 301     unsigned int pts;
 302     int aid;
 303     unsigned char *packet;
 304     unsigned int packet_reserve;
 305     unsigned int packet_size;
 306 } mpeg_t;
 307
 308 static mpeg_t *
 309 mpeg_open(const char *filename)
 310 {
 311     mpeg_t *res = malloc(sizeof(mpeg_t));
 312     int err = res == NULL;
 313     if (!err) {
 314         res->pts = 0;
 315         res->aid = -1;
 316         res->packet = NULL;
 317         res->packet_size = 0;
 318         res->packet_reserve = 0;
 319         res->stream = rar_open(filename, "rb");
 320         err = res->stream == NULL;
 321         if (err)
 322             perror("fopen Vobsub file failed");
 323         if (err)
 324             free(res);
 325     }
 326     return err ? NULL : res;
 327 }
 328
 329 static void
 330 mpeg_free(mpeg_t *mpeg)
 331 {
 332     if (mpeg->packet)
 333         free(mpeg->packet);
 334     if (mpeg->stream)
 335         rar_close(mpeg->stream);
 336     free(mpeg);
 337 }
 338
 339 static int
 340 mpeg_eof(mpeg_t *mpeg)
 341 {
 342     return rar_eof(mpeg->stream);
 343 }
 344
 345 static off_t
 346 mpeg_tell(mpeg_t *mpeg)
 347 {
 348     return rar_tell(mpeg->stream);
 349 }
 350
 351 static int
 352 mpeg_run(mpeg_t *mpeg)
 353 {
 354     unsigned int len, idx, version;
 355     int c;
 356     /* Goto start of a packet, it starts with 0x000001?? */
 357     const unsigned char wanted[] = { 0, 0, 1 };
 358     unsigned char buf[5];
 359
 360     mpeg->aid = -1;
 361     mpeg->packet_size = 0;
 362     if (rar_read(buf, 4, 1, mpeg->stream) != 1)
 363         return -1;
 364     while (memcmp(buf, wanted, sizeof(wanted)) != 0) {
 365         c = rar_getc(mpeg->stream);
 366         if (c < 0)
 367             return -1;
 368         memmove(buf, buf + 1, 3);
 369         buf[3] = c;
 370     }
 371     switch (buf[3]) {
 372     case 0xb9:                  /* System End Code */
 373         break;
 374     case 0xba:                  /* Packet start code */
 375         c = rar_getc(mpeg->stream);
 376         if (c < 0)
 377             return -1;
 378         if ((c & 0xc0) == 0x40)
 379             version = 4;
 380         else if ((c & 0xf0) == 0x20)
 381             version = 2;
 382         else {
 383             mp_msg(MSGT_VOBSUB,MSGL_ERR, "VobSub: Unsupported MPEG version: 0x%02x\n", c);
 384             return -1;
 385         }
 386         if (version == 4) {
 387             if (rar_seek(mpeg->stream, 9, SEEK_CUR))
 388                 return -1;
 389         }
 390         else if (version == 2) {
 391             if (rar_seek(mpeg->stream, 7, SEEK_CUR))
 392                 return -1;
 393         }
 394         else
 395             abort();
 396         break;
 397     case 0xbd:                  /* packet */
 398         if (rar_read(buf, 2, 1, mpeg->stream) != 1)
 399             return -1;
 400         len = buf[0] << 8 | buf[1];
 401         idx = mpeg_tell(mpeg);
 402         c = rar_getc(mpeg->stream);
 403         if (c < 0)
 404             return -1;
 405         if ((c & 0xC0) == 0x40) { /* skip STD scale & size */
 406             if (rar_getc(mpeg->stream) < 0)
 407                 return -1;
 408             c = rar_getc(mpeg->stream);
 409             if (c < 0)
 410                 return -1;
 411         }
 412         if ((c & 0xf0) == 0x20) { /* System-1 stream timestamp */
 413             /* Do we need this? */
 414             abort();
 415         }
 416         else if ((c & 0xf0) == 0x30) {
 417             /* Do we need this? */
 418             abort();
 419         }
 420         else if ((c & 0xc0) == 0x80) { /* System-2 (.VOB) stream */
 421             unsigned int pts_flags, hdrlen, dataidx;
 422             c = rar_getc(mpeg->stream);
 423             if (c < 0)
 424                 return -1;
 425             pts_flags = c;
 426             c = rar_getc(mpeg->stream);
 427             if (c < 0)
 428                 return -1;
 429             hdrlen = c;
 430             dataidx = mpeg_tell(mpeg) + hdrlen;
 431             if (dataidx > idx + len) {
 432                 mp_msg(MSGT_VOBSUB,MSGL_ERR, "Invalid header length: %d (total length: %d, idx: %d, dataidx: %d)\n",
 433                         hdrlen, len, idx, dataidx);
 434                 return -1;
 435             }
 436             if ((pts_flags & 0xc0) == 0x80) {
 437                 if (rar_read(buf, 5, 1, mpeg->stream) != 1)
 438                     return -1;
 439                 if (!(((buf[0] & 0xf0) == 0x20) && (buf[0] & 1) && (buf[2] & 1) &&  (buf[4] & 1))) {
 440                     mp_msg(MSGT_VOBSUB,MSGL_ERR, "vobsub PTS error: 0x%02x %02x%02x %02x%02x \n",
 441                             buf[0], buf[1], buf[2], buf[3], buf[4]);
 442                     mpeg->pts = 0;
 443                 }
 444                 else
 445                     mpeg->pts = ((buf[0] & 0x0e) << 29 | buf[1] << 22 | (buf[2] & 0xfe) << 14
 446                         | buf[3] << 7 | (buf[4] >> 1));
 447             }
 448             else /* if ((pts_flags & 0xc0) == 0xc0) */ {
 449                 /* what's this? */
 450                 /* abort(); */
 451             }
 452             rar_seek(mpeg->stream, dataidx, SEEK_SET);
 453             mpeg->aid = rar_getc(mpeg->stream);
 454             if (mpeg->aid < 0) {
 455                 mp_msg(MSGT_VOBSUB,MSGL_ERR, "Bogus aid %d\n", mpeg->aid);
 456                 return -1;
 457             }
 458             mpeg->packet_size = len - ((unsigned int) mpeg_tell(mpeg) - idx);
 459             if (mpeg->packet_reserve < mpeg->packet_size) {
 460                 if (mpeg->packet)
 461                     free(mpeg->packet);
 462                 mpeg->packet = malloc(mpeg->packet_size);
 463                 if (mpeg->packet)
 464                     mpeg->packet_reserve = mpeg->packet_size;
 465             }
 466             if (mpeg->packet == NULL) {
 467                 mp_msg(MSGT_VOBSUB,MSGL_FATAL,"malloc failure");
 468                 mpeg->packet_reserve = 0;
 469                 mpeg->packet_size = 0;
 470                 return -1;
 471             }
 472             if (rar_read(mpeg->packet, mpeg->packet_size, 1, mpeg->stream) != 1) {
 473                 mp_msg(MSGT_VOBSUB,MSGL_ERR,"fread failure");
 474                 mpeg->packet_size = 0;
 475                 return -1;
 476             }
 477             idx = len;
 478         }
 479         break;
 480     case 0xbe:                  /* Padding */
 481         if (rar_read(buf, 2, 1, mpeg->stream) != 1)
 482             return -1;
 483         len = buf[0] << 8 | buf[1];
 484         if (len > 0 && rar_seek(mpeg->stream, len, SEEK_CUR))
 485             return -1;
 486         break;
 487     default:
 488         if (0xc0 <= buf[3] && buf[3] < 0xf0) {
 489             /* MPEG audio or video */
 490             if (rar_read(buf, 2, 1, mpeg->stream) != 1)
 491                 return -1;
 492             len = buf[0] << 8 | buf[1];
 493             if (len > 0 && rar_seek(mpeg->stream, len, SEEK_CUR))
 494                 return -1;
 495
 496         }
 497         else {
 498             mp_msg(MSGT_VOBSUB,MSGL_ERR,"unknown header 0x%02X%02X%02X%02X\n",
 499                     buf[0], buf[1], buf[2], buf[3]);
 500             return -1;
 501         }
 502     }
 503     return 0;
 504 }
 505
 506 /**********************************************************************
 507  * Packet queue
 508  **********************************************************************/
 509
 510 typedef struct {
 511     unsigned int pts100;
 512     off_t filepos;
 513     unsigned int size;
 514     unsigned char *data;
 515 } packet_t;
 516
 517 typedef struct {
 518     char *id;
 519     packet_t *packets;
 520     unsigned int packets_reserve;
 521     unsigned int packets_size;
 522     unsigned int current_index;
 523 } packet_queue_t;
 524
 525 static void
 526 packet_construct(packet_t *pkt)
 527 {
 528     pkt->pts100 = 0;
 529     pkt->filepos = 0;
 530     pkt->size = 0;
 531     pkt->data = NULL;
 532 }
 533
 534 static void
 535 packet_destroy(packet_t *pkt)
 536 {
 537     if (pkt->data)
 538         free(pkt->data);
 539 }
 540
 541 static void
 542 packet_queue_construct(packet_queue_t *queue)
 543 {
 544     queue->id = NULL;
 545     queue->packets = NULL;
 546     queue->packets_reserve = 0;
 547     queue->packets_size = 0;
 548     queue->current_index = 0;
 549 }
 550
 551 static void
 552 packet_queue_destroy(packet_queue_t *queue)
 553 {
 554     if (queue->packets) {
 555         while (queue->packets_size--)
 556             packet_destroy(queue->packets + queue->packets_size);
 557         free(queue->packets);
 558     }
 559     return;
 560 }
 561
 562 /* Make sure there is enough room for needed_size packets in the
 563    packet queue. */
 564 static int
 565 packet_queue_ensure(packet_queue_t *queue, unsigned int needed_size)
 566 {
 567     if (queue->packets_reserve < needed_size) {
 568         if (queue->packets) {
 569             packet_t *tmp = realloc(queue->packets, 2 * queue->packets_reserve * sizeof(packet_t));
 570             if (tmp == NULL) {
 571                 mp_msg(MSGT_VOBSUB,MSGL_FATAL,"realloc failure");
 572                 return -1;
 573             }
 574             queue->packets = tmp;
 575             queue->packets_reserve *= 2;
 576         }
 577         else {
 578             queue->packets = malloc(sizeof(packet_t));
 579             if (queue->packets == NULL) {
 580                 mp_msg(MSGT_VOBSUB,MSGL_FATAL,"malloc failure");
 581                 return -1;
 582             }
 583             queue->packets_reserve = 1;
 584         }
 585     }
 586     return 0;
 587 }
 588
 589 /* add one more packet */
 590 static int
 591 packet_queue_grow(packet_queue_t *queue)
 592 {
 593     if (packet_queue_ensure(queue, queue->packets_size + 1) < 0)
 594         return -1;
 595     packet_construct(queue->packets + queue->packets_size);
 596     ++queue->packets_size;
 597     return 0;
 598 }
 599
 600 /* insert a new packet, duplicating pts from the current one */
 601 static int
 602 packet_queue_insert(packet_queue_t *queue)
 603 {
 604     packet_t *pkts;
 605     if (packet_queue_ensure(queue, queue->packets_size + 1) < 0)
 606         return -1;
 607     /* XXX packet_size does not reflect the real thing here, it will be updated a bit later */
 608     memmove(queue->packets + queue->current_index + 2,
 609             queue->packets + queue->current_index + 1,
 610             sizeof(packet_t) * (queue->packets_size - queue->current_index - 1));
 611     pkts = queue->packets + queue->current_index;
 612     ++queue->packets_size;
 613     ++queue->current_index;
 614     packet_construct(pkts + 1);
 615     pkts[1].pts100 = pkts[0].pts100;
 616     pkts[1].filepos = pkts[0].filepos;
 617     return 0;
 618 }
 619
 620 /**********************************************************************
 621  * Vobsub
 622  **********************************************************************/
 623
 624 typedef struct {
 625     unsigned int palette[16];
 626     unsigned int cuspal[4];
 627     int delay;
 628     unsigned int custom;
 629     unsigned int have_palette;
 630     unsigned int orig_frame_width, orig_frame_height;
 631     unsigned int origin_x, origin_y;
 632     unsigned int forced_subs;
 633     /* index */
 634     packet_queue_t *spu_streams;
 635     unsigned int spu_streams_size;
 636     unsigned int spu_streams_current;
 637     unsigned int spu_valid_streams_size;
 638 } vobsub_t;
 639
 640 /* Make sure that the spu stream idx exists. */
 641 static int
 642 vobsub_ensure_spu_stream(vobsub_t *vob, unsigned int index)
 643 {
 644     if (index >= vob->spu_streams_size) {
 645         /* This is a new stream */
 646         if (vob->spu_streams) {
 647             packet_queue_t *tmp = realloc(vob->spu_streams, (index + 1) * sizeof(packet_queue_t));
 648             if (tmp == NULL) {
 649                 mp_msg(MSGT_VOBSUB,MSGL_ERR,"vobsub_ensure_spu_stream: realloc failure");
 650                 return -1;
 651             }
 652             vob->spu_streams = tmp;
 653         }
 654         else {
 655             vob->spu_streams = malloc((index + 1) * sizeof(packet_queue_t));
 656             if (vob->spu_streams == NULL) {
 657                 mp_msg(MSGT_VOBSUB,MSGL_ERR,"vobsub_ensure_spu_stream: malloc failure");
 658                 return -1;
 659             }
 660         }
 661         while (vob->spu_streams_size <= index) {
 662             packet_queue_construct(vob->spu_streams + vob->spu_streams_size);
 663             ++vob->spu_streams_size;
 664         }
 665     }
 666     return 0;
 667 }
 668
 669 static int
 670 vobsub_add_id(vobsub_t *vob, const char *id, size_t idlen, const unsigned int index)
 671 {
 672     if (vobsub_ensure_spu_stream(vob, index) < 0)
 673         return -1;
 674     if (id && idlen) {
 675         if (vob->spu_streams[index].id)
 676             free(vob->spu_streams[index].id);
 677         vob->spu_streams[index].id = malloc(idlen + 1);
 678         if (vob->spu_streams[index].id == NULL) {
 679             mp_msg(MSGT_VOBSUB,MSGL_FATAL,"vobsub_add_id: malloc failure");
 680             return -1;
 681         }
 682         vob->spu_streams[index].id[idlen] = 0;
 683         memcpy(vob->spu_streams[index].id, id, idlen);
 684     }
 685     vob->spu_streams_current = index;
 686     mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_VOBSUB_ID=%d\n", index);
 687     if (id && idlen)
 688         mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_VSID_%d_LANG=%s\n", index, vob->spu_streams[index].id);
 689     mp_msg(MSGT_VOBSUB,MSGL_V,"[vobsub] subtitle (vobsubid): %d language %s\n",
 690                 index, vob->spu_streams[index].id);
 691     return 0;
 692 }
 693
 694 static int
 695 vobsub_add_timestamp(vobsub_t *vob, off_t filepos, int ms)
 696 {
 697     packet_queue_t *queue;
 698     packet_t *pkt;
 699     if (vob->spu_streams == 0) {
 700         mp_msg(MSGT_VOBSUB,MSGL_WARN,"[vobsub] warning, binning some index entries.  Check your index file\n");
 701         return -1;
 702     }
 703     queue = vob->spu_streams + vob->spu_streams_current;
 704     if (packet_queue_grow(queue) >= 0) {
 705         pkt = queue->packets + (queue->packets_size - 1);
 706         pkt->filepos = filepos;
 707         pkt->pts100 = ms < 0 ? UINT_MAX : (unsigned int)ms * 90;
 708         return 0;
 709     }
 710     return -1;
 711 }
 712
 713 static int
 714 vobsub_parse_id(vobsub_t *vob, const char *line)
 715 {
 716     // id: xx, index: n
 717     size_t idlen;
 718     const char *p, *q;
 719     p  = line;
 720     while (isspace(*p))
 721         ++p;
 722     q = p;
 723     while (isalpha(*q))
 724         ++q;
 725     idlen = q - p;
 726     if (idlen == 0)
 727         return -1;
 728     ++q;
 729     while (isspace(*q))
 730         ++q;
 731     if (strncmp("index:", q, 6))
 732         return -1;
 733     q += 6;
 734     while (isspace(*q))
 735         ++q;
 736     if (!isdigit(*q))
 737         return -1;
 738     return vobsub_add_id(vob, p, idlen, atoi(q));
 739 }
 740
 741 static int
 742 vobsub_parse_timestamp(vobsub_t *vob, const char *line)
 743 {
 744     // timestamp: HH:MM:SS.mmm, filepos: 0nnnnnnnnn
 745     const char *p;
 746     int h, m, s, ms;
 747     off_t filepos;
 748     while (isspace(*line))
 749         ++line;
 750     p = line;
 751     while (isdigit(*p))
 752         ++p;
 753     if (p - line != 2)
 754         return -1;
 755     h = atoi(line);
 756     if (*p != ':')
 757         return -1;
 758     line = ++p;
 759     while (isdigit(*p))
 760         ++p;
 761     if (p - line != 2)
 762         return -1;
 763     m = atoi(line);
 764     if (*p != ':')
 765         return -1;
 766     line = ++p;
 767     while (isdigit(*p))
 768         ++p;
 769     if (p - line != 2)
 770         return -1;
 771     s = atoi(line);
 772     if (*p != ':')
 773         return -1;
 774     line = ++p;
 775     while (isdigit(*p))
 776         ++p;
 777     if (p - line != 3)
 778         return -1;
 779     ms = atoi(line);
 780     if (*p != ',')
 781         return -1;
 782     line = p + 1;
 783     while (isspace(*line))
 784         ++line;
 785     if (strncmp("filepos:", line, 8))
 786         return -1;
 787     line += 8;
 788     while (isspace(*line))
 789         ++line;
 790     if (! isxdigit(*line))
 791         return -1;
 792     filepos = strtol(line, NULL, 16);
 793     return vobsub_add_timestamp(vob, filepos, vob->delay + ms + 1000 * (s + 60 * (m + 60 * h)));
 794 }
 795
 796 static int
 797 vobsub_parse_size(vobsub_t *vob, const char *line)
 798 {
 799     // size: WWWxHHH
 800     char *p;
 801     while (isspace(*line))
 802         ++line;
 803     if (!isdigit(*line))
 804         return -1;
 805     vob->orig_frame_width = strtoul(line, &p, 10);
 806     if (*p != 'x')
 807         return -1;
 808     ++p;
 809     vob->orig_frame_height = strtoul(p, NULL, 10);
 810     return 0;
 811 }
 812
 813 static int
 814 vobsub_parse_origin(vobsub_t *vob, const char *line)
 815 {
 816     // org: X,Y
 817     char *p;
 818     while (isspace(*line))
 819         ++line;
 820     if (!isdigit(*line))
 821         return -1;
 822     vob->origin_x = strtoul(line, &p, 10);
 823     if (*p != ',')
 824         return -1;
 825     ++p;
 826     vob->origin_y = strtoul(p, NULL, 10);
 827     return 0;
 828 }
 829
 830 unsigned int vobsub_palette_to_yuv(unsigned int pal)
 831 {
 832     int r, g, b, y, u, v;
 833     // Palette in idx file is not rgb value, it was calculated by wrong forumla.
 834     // Here's reversed forumla of the one used to generate palette in idx file.
 835     r = pal >> 16 & 0xff;
 836     g = pal >> 8 & 0xff;
 837     b = pal & 0xff;
 838     y = av_clip_uint8( 0.1494  * r + 0.6061 * g + 0.2445 * b);
 839     u = av_clip_uint8( 0.6066  * r - 0.4322 * g - 0.1744 * b + 128);
 840     v = av_clip_uint8(-0.08435 * r - 0.3422 * g + 0.4266 * b + 128);
 841     y = y * 219 / 255 + 16;
 842     return y << 16 | u << 8 | v;
 843 }
 844
 845 unsigned int vobsub_rgb_to_yuv(unsigned int rgb)
 846 {
 847     int r, g, b, y, u, v;
 848     r = rgb >> 16 & 0xff;
 849     g = rgb >> 8 & 0xff;
 850     b = rgb & 0xff;
 851     y = ( 0.299   * r + 0.587   * g + 0.114   * b) * 219 / 255 + 16.5;
 852     u = (-0.16874 * r - 0.33126 * g + 0.5     * b) * 224 / 255 + 128.5;
 853     v = ( 0.5     * r - 0.41869 * g - 0.08131 * b) * 224 / 255 + 128.5;
 854     return y << 16 | u << 8 | v;
 855 }
 856
 857 static int
 858 vobsub_parse_palette(vobsub_t *vob, const char *line)
 859 {
 860     // palette: XXXXXX, XXXXXX, XXXXXX, XXXXXX, XXXXXX, XXXXXX, XXXXXX, XXXXXX, XXXXXX, XXXXXX, XXXXXX, XXXXXX, XXXXXX, XXXXXX, XXXXXX, XXXXXX
 861     unsigned int n;
 862     n = 0;
 863     while (1) {
 864         const char *p;
 865         int tmp;
 866         while (isspace(*line))
 867             ++line;
 868         p = line;
 869         while (isxdigit(*p))
 870             ++p;
 871         if (p - line != 6)
 872             return -1;
 873         tmp = strtoul(line, NULL, 16);
 874         vob->palette[n++] = vobsub_palette_to_yuv(tmp);
 875         if (n == 16)
 876             break;
 877         if (*p == ',')
 878             ++p;
 879         line = p;
 880     }
 881     vob->have_palette = 1;
 882     return 0;
 883 }
 884
 885 static int
 886 vobsub_parse_custom(vobsub_t *vob, const char *line)
 887 {
 888     //custom colors: OFF/ON(0/1)
 889     if ((strncmp("ON", line + 15, 2) == 0)||strncmp("1", line + 15, 1) == 0)
 890         vob->custom=1;
 891     else if ((strncmp("OFF", line + 15, 3) == 0)||strncmp("0", line + 15, 1) == 0)
 892         vob->custom=0;
 893     else
 894         return -1;
 895     return 0;
 896 }
 897
 898 static int
 899 vobsub_parse_cuspal(vobsub_t *vob, const char *line)
 900 {
 901     //colors: XXXXXX, XXXXXX, XXXXXX, XXXXXX
 902     unsigned int n, tmp;
 903     n = 0;
 904     line += 40;
 905     while(1){
 906         const char *p;
 907         while (isspace(*line))
 908             ++line;
 909         p=line;
 910         while (isxdigit(*p))
 911             ++p;
 912         if (p - line !=6)
 913             return -1;
 914         tmp = strtoul(line, NULL, 16);
 915         vob->cuspal[n++] = vobsub_rgb_to_yuv(tmp);
 916         if (n==4)
 917             break;
 918         if(*p == ',')
 919             ++p;
 920         line = p;
 921     }
 922     return 0;
 923 }
 924
 925 /* don't know how to use tridx */
 926 static int
 927 vobsub_parse_tridx(const char *line)
 928 {
 929     //tridx: XXXX
 930     int tridx;
 931     tridx = strtoul((line + 26), NULL, 16);
 932     tridx = ((tridx&0x1000)>>12) | ((tridx&0x100)>>7) | ((tridx&0x10)>>2) | ((tridx&1)<<3);
 933     return tridx;
 934 }
 935
 936 static int
 937 vobsub_parse_delay(vobsub_t *vob, const char *line)
 938 {
 939     int h, m, s, ms;
 940     int forward = 1;
 941     if (*(line + 7) == '+'){
 942         forward = 1;
 943         line++;
 944     }
 945     else if (*(line + 7) == '-'){
 946         forward = -1;
 947         line++;
 948     }
 949     mp_msg(MSGT_SPUDEC,MSGL_V, "forward=%d", forward);
 950     h = atoi(line + 7);
 951     mp_msg(MSGT_VOBSUB,MSGL_V, "h=%d," ,h);
 952     m = atoi(line + 10);
 953     mp_msg(MSGT_VOBSUB,MSGL_V, "m=%d,", m);
 954     s = atoi(line + 13);
 955     mp_msg(MSGT_VOBSUB,MSGL_V, "s=%d,", s);
 956     ms = atoi(line + 16);
 957     mp_msg(MSGT_VOBSUB,MSGL_V, "ms=%d", ms);
 958     vob->delay = (ms + 1000 * (s + 60 * (m + 60 * h))) * forward;
 959     return 0;
 960 }
 961
 962 static int
 963 vobsub_set_lang(const char *line)
 964 {
 965     if (vobsub_id == -1)
 966         vobsub_id = atoi(line + 8);
 967     return 0;
 968 }
 969
 970 static int
 971 vobsub_parse_forced_subs(vobsub_t *vob, const char *line)
 972 {
 973     const char *p;
 974
 975     p  = line;
 976     while (isspace(*p))
 977         ++p;
 978
 979     if (strncasecmp("on",p,2) == 0){
 980             vob->forced_subs=~0;
 981             return 0;
 982     } else if (strncasecmp("off",p,3) == 0){
 983             vob->forced_subs=0;
 984             return 0;
 985     }
 986
 987     return -1;
 988 }
 989
 990 static int
 991 vobsub_parse_one_line(vobsub_t *vob, rar_stream_t *fd)
 992 {
 993     ssize_t line_size;
 994     int res = -1;
 995         size_t line_reserve = 0;
 996         char *line = NULL;
 997     do {
 998         line_size = vobsub_getline(&line, &line_reserve, fd);
 999         if (line_size < 0) {
1000             break;
1001         }
1002         if (*line == 0 || *line == '\r' || *line == '\n' || *line == '#')
1003             continue;
1004         else if (strncmp("langidx:", line, 8) == 0)
1005             res = vobsub_set_lang(line);
1006         else if (strncmp("delay:", line, 6) == 0)
1007             res = vobsub_parse_delay(vob, line);
1008         else if (strncmp("id:", line, 3) == 0)
1009             res = vobsub_parse_id(vob, line + 3);
1010         else if (strncmp("palette:", line, 8) == 0)
1011             res = vobsub_parse_palette(vob, line + 8);
1012         else if (strncmp("size:", line, 5) == 0)
1013             res = vobsub_parse_size(vob, line + 5);
1014         else if (strncmp("org:", line, 4) == 0)
1015             res = vobsub_parse_origin(vob, line + 4);
1016         else if (strncmp("timestamp:", line, 10) == 0)
1017             res = vobsub_parse_timestamp(vob, line + 10);
1018         else if (strncmp("custom colors:", line, 14) == 0)
1019             //custom colors: ON/OFF, tridx: XXXX, colors: XXXXXX, XXXXXX, XXXXXX,XXXXXX
1020             res = vobsub_parse_cuspal(vob, line) + vobsub_parse_tridx(line) + vobsub_parse_custom(vob, line);
1021         else if (strncmp("forced subs:", line, 12) == 0)
1022                 res = vobsub_parse_forced_subs(vob, line + 12);
1023         else {
1024             mp_msg(MSGT_VOBSUB,MSGL_V, "vobsub: ignoring %s", line);
1025             continue;
1026         }
1027         if (res < 0)
1028             mp_msg(MSGT_VOBSUB,MSGL_ERR,  "ERROR in %s", line);
1029         break;
1030     } while (1);
1031     if (line)
1032       free(line);
1033     return res;
1034 }
1035
1036 int
1037 vobsub_parse_ifo(void* this, const char *const name, unsigned int *palette, unsigned int *width, unsigned int *height, int force,
1038                  int sid, char *langid)
1039 {
1040     vobsub_t *vob = (vobsub_t*)this;
1041     int res = -1;
1042     rar_stream_t *fd = rar_open(name, "rb");
1043     if (fd == NULL) {
1044         if (force)
1045             mp_msg(MSGT_VOBSUB,MSGL_WARN, "VobSub: Can't open IFO file\n");
1046     } else {
1047         // parse IFO header
1048         unsigned char block[0x800];
1049         const char *const ifo_magic = "DVDVIDEO-VTS";
1050         if (rar_read(block, sizeof(block), 1, fd) != 1) {
1051             if (force)
1052                 mp_msg(MSGT_VOBSUB,MSGL_ERR, "VobSub: Can't read IFO header\n");
1053         } else if (memcmp(block, ifo_magic, strlen(ifo_magic) + 1))
1054             mp_msg(MSGT_VOBSUB,MSGL_ERR, "VobSub: Bad magic in IFO header\n");
1055         else {
1056             unsigned long pgci_sector = block[0xcc] << 24 | block[0xcd] << 16
1057                 | block[0xce] << 8 | block[0xcf];
1058             int standard = (block[0x200] & 0x30) >> 4;
1059             int resolution = (block[0x201] & 0x0c) >> 2;
1060             *height = standard ? 576 : 480;
1061             *width = 0;
1062             switch (resolution) {
1063             case 0x0:
1064                 *width = 720;
1065                 break;
1066             case 0x1:
1067                 *width = 704;
1068                 break;
1069             case 0x2:
1070                 *width = 352;
1071                 break;
1072             case 0x3:
1073                 *width = 352;
1074                 *height /= 2;
1075                 break;
1076             default:
1077                 mp_msg(MSGT_VOBSUB,MSGL_WARN,"Vobsub: Unknown resolution %d \n", resolution);
1078             }
1079             if (langid && 0 <= sid && sid < 32) {
1080                 unsigned char *tmp = block + 0x256 + sid * 6 + 2;
1081                 langid[0] = tmp[0];
1082                 langid[1] = tmp[1];
1083                 langid[2] = 0;
1084             }
1085             if (rar_seek(fd, pgci_sector * sizeof(block), SEEK_SET)
1086                 || rar_read(block, sizeof(block), 1, fd) != 1)
1087                 mp_msg(MSGT_VOBSUB,MSGL_ERR, "VobSub: Can't read IFO PGCI\n");
1088             else {
1089                 unsigned long idx;
1090                 unsigned long pgc_offset = block[0xc] << 24 | block[0xd] << 16
1091                     | block[0xe] << 8 | block[0xf];
1092                 for (idx = 0; idx < 16; ++idx) {
1093                     unsigned char *p = block + pgc_offset + 0xa4 + 4 * idx;
1094                     palette[idx] = p[0] << 24 | p[1] << 16 | p[2] << 8 | p[3];
1095                 }
1096                 if(vob)
1097                   vob->have_palette = 1;
1098                 res = 0;
1099             }
1100         }
1101         rar_close(fd);
1102     }
1103     return res;
1104 }
1105
1106 void *
1107 vobsub_open(const char *const name,const char *const ifo,const int force,void** spu)
1108 {
1109     vobsub_t *vob = malloc(sizeof(vobsub_t));
1110     if(spu)
1111       *spu = NULL;
1112     if (vobsubid == -2)
1113       vobsubid = vobsub_id;
1114     if (vob) {
1115         char *buf;
1116         vob->custom = 0;
1117         vob->have_palette = 0;
1118         vob->orig_frame_width = 0;
1119         vob->orig_frame_height = 0;
1120         vob->spu_streams = NULL;
1121         vob->spu_streams_size = 0;
1122         vob->spu_streams_current = 0;
1123         vob->spu_valid_streams_size = 0;
1124         vob->delay = 0;
1125         vob->forced_subs=0;
1126         buf = malloc(strlen(name) + 5);
1127         if (buf) {
1128             rar_stream_t *fd;
1129             mpeg_t *mpg;
1130             /* read in the info file */
1131             if(!ifo) {
1132               strcpy(buf, name);
1133               strcat(buf, ".ifo");
1134               vobsub_parse_ifo(vob,buf, vob->palette, &vob->orig_frame_width, &vob->orig_frame_height, force, -1, NULL);
1135             } else
1136               vobsub_parse_ifo(vob,ifo, vob->palette, &vob->orig_frame_width, &vob->orig_frame_height, force, -1, NULL);
1137             /* read in the index */
1138             strcpy(buf, name);
1139             strcat(buf, ".idx");
1140             fd = rar_open(buf, "rb");
1141             if (fd == NULL) {
1142                 if(force)
1143                   mp_msg(MSGT_VOBSUB,MSGL_ERR,"VobSub: Can't open IDX file\n");
1144                 else {
1145                   free(buf);
1146                   free(vob);
1147                   return NULL;
1148                 }
1149             } else {
1150                 while (vobsub_parse_one_line(vob, fd) >= 0)
1151                     /* NOOP */ ;
1152                 rar_close(fd);
1153             }
1154             /* if no palette in .idx then use custom colors */
1155             if ((vob->custom == 0)&&(vob->have_palette!=1))
1156                 vob->custom = 1;
1157             if (spu && vob->orig_frame_width && vob->orig_frame_height)
1158               *spu = spudec_new_scaled_vobsub(vob->palette, vob->cuspal, vob->custom, vob->orig_frame_width, vob->orig_frame_height);
1159
1160             /* read the indexed mpeg_stream */
1161             strcpy(buf, name);
1162             strcat(buf, ".sub");
1163             mpg = mpeg_open(buf);
1164             if (mpg == NULL) {
1165               if(force)
1166                 mp_msg(MSGT_VOBSUB,MSGL_ERR,"VobSub: Can't open SUB file\n");
1167               else {
1168
1169                 free(buf);
1170                 free(vob);
1171                 return NULL;
1172               }
1173             } else {
1174                 long last_pts_diff = 0;
1175                 while (!mpeg_eof(mpg)) {
1176                     off_t pos = mpeg_tell(mpg);
1177                     if (mpeg_run(mpg) < 0) {
1178                         if (!mpeg_eof(mpg))
1179                             mp_msg(MSGT_VOBSUB,MSGL_ERR,"VobSub: mpeg_run error\n");
1180                         break;
1181                     }
1182                     if (mpg->packet_size) {
1183                         if ((mpg->aid & 0xe0) == 0x20) {
1184                             unsigned int sid = mpg->aid & 0x1f;
1185                             if (vobsub_ensure_spu_stream(vob, sid) >= 0)  {
1186                                 packet_queue_t *queue = vob->spu_streams + sid;
1187                                 /* get the packet to fill */
1188                                 if (queue->packets_size == 0 && packet_queue_grow(queue)  < 0)
1189                                   abort();
1190                                 while (queue->current_index + 1 < queue->packets_size
1191                                        && queue->packets[queue->current_index + 1].filepos <= pos)
1192                                     ++queue->current_index;
1193                                 if (queue->current_index < queue->packets_size) {
1194                                     packet_t *pkt;
1195                                     if (queue->packets[queue->current_index].data) {
1196                                         /* insert a new packet and fix the PTS ! */
1197                                         packet_queue_insert(queue);
1198                                         queue->packets[queue->current_index].pts100 =
1199                                             mpg->pts + last_pts_diff;
1200                                     }
1201                                     pkt = queue->packets + queue->current_index;
1202                                     if (pkt->pts100 != UINT_MAX) {
1203                                     if (queue->packets_size > 1)
1204                                         last_pts_diff = pkt->pts100 - mpg->pts;
1205                                     else
1206                                         pkt->pts100 = mpg->pts;
1207                                     /* FIXME: should not use mpg_sub internal informations, make a copy */
1208                                     pkt->data = mpg->packet;
1209                                     pkt->size = mpg->packet_size;
1210                                     mpg->packet = NULL;
1211                                     mpg->packet_reserve = 0;
1212                                     mpg->packet_size = 0;
1213                                     }
1214                                 }
1215                             }
1216                             else
1217                                 mp_msg(MSGT_VOBSUB,MSGL_WARN, "don't know what to do with subtitle #%u\n", sid);
1218                         }
1219                     }
1220                 }
1221                 vob->spu_streams_current = vob->spu_streams_size;
1222                 while (vob->spu_streams_current-- > 0) {
1223                     vob->spu_streams[vob->spu_streams_current].current_index = 0;
1224                     if (vobsubid == vob->spu_streams_current ||
1225                             vob->spu_streams[vob->spu_streams_current].packets_size > 0)
1226                         ++vob->spu_valid_streams_size;
1227                 }
1228                 mpeg_free(mpg);
1229             }
1230             free(buf);
1231         }
1232     }
1233     return vob;
1234 }
1235
1236 void
1237 vobsub_close(void *this)
1238 {
1239     vobsub_t *vob = (vobsub_t *)this;
1240     if (vob->spu_streams) {
1241         while (vob->spu_streams_size--)
1242             packet_queue_destroy(vob->spu_streams + vob->spu_streams_size);
1243         free(vob->spu_streams);
1244     }
1245     free(vob);
1246 }
1247
1248 unsigned int
1249 vobsub_get_indexes_count(void *vobhandle)
1250 {
1251     vobsub_t *vob = (vobsub_t *) vobhandle;
1252     return vob->spu_valid_streams_size;
1253 }
1254
1255 char *
1256 vobsub_get_id(void *vobhandle, unsigned int index)
1257 {
1258     vobsub_t *vob = (vobsub_t *) vobhandle;
1259     return (index < vob->spu_streams_size) ? vob->spu_streams[index].id : NULL;
1260 }
1261
1262 int vobsub_get_id_by_index(void *vobhandle, unsigned int index)
1263 {
1264     vobsub_t *vob = vobhandle;
1265     int i, j;
1266     if (vob == NULL)
1267         return -1;
1268     for (i = 0, j = 0; i < vob->spu_streams_size; ++i)
1269         if (i == vobsubid || vob->spu_streams[i].packets_size > 0) {
1270             if (j == index)
1271                 return i;
1272             ++j;
1273         }
1274     return -1;
1275 }
1276
1277 int vobsub_get_index_by_id(void *vobhandle, int id)
1278 {
1279     vobsub_t *vob = vobhandle;
1280     int i, j;
1281     if (vob == NULL || id < 0 || id >= vob->spu_streams_size)
1282         return -1;
1283     if (id != vobsubid && !vob->spu_streams[id].packets_size)
1284         return -1;
1285     for (i = 0, j = 0; i < id; ++i)
1286         if (i == vobsubid || vob->spu_streams[i].packets_size > 0)
1287             ++j;
1288     return j;
1289 }
1290
1291 unsigned int
1292 vobsub_get_forced_subs_flag(void const * const vobhandle)
1293 {
1294   if (vobhandle)
1295     return ((vobsub_t*) vobhandle)->forced_subs;
1296   else
1297     return 0;
1298 }
1299
1300 int
1301 vobsub_set_from_lang(void *vobhandle, unsigned char * lang)
1302 {
1303     int i;
1304     vobsub_t *vob= (vobsub_t *) vobhandle;
1305     while(lang && strlen(lang) >= 2){
1306       for(i=0; i < vob->spu_streams_size; i++)
1307         if (vob->spu_streams[i].id)
1308           if ((strncmp(vob->spu_streams[i].id, lang, 2)==0)){
1309             vobsub_id=i;
1310             mp_msg(MSGT_VOBSUB, MSGL_INFO, "Selected VOBSUB language: %d language: %s\n", i, vob->spu_streams[i].id);
1311             return 0;
1312           }
1313       lang+=2;while (lang[0]==',' || lang[0]==' ') ++lang;
1314     }
1315     mp_msg(MSGT_VOBSUB, MSGL_WARN, "No matching VOBSUB language found!\n");
1316     return -1;
1317 }
1318
1319 int
1320 vobsub_get_packet(void *vobhandle, float pts,void** data, int* timestamp) {
1321   vobsub_t *vob = (vobsub_t *)vobhandle;
1322   unsigned int pts100 = 90000 * pts;
1323   if (vob->spu_streams && 0 <= vobsub_id && (unsigned) vobsub_id < vob->spu_streams_size) {
1324     packet_queue_t *queue = vob->spu_streams + vobsub_id;
1325
1326     int reseek_count = 0;
1327     unsigned int lastpts = 0;
1328     while (queue->current_index < queue->packets_size
1329             && queue->packets[queue->current_index].pts100 <= pts100) {
1330       lastpts = queue->packets[queue->current_index].pts100;
1331       ++queue->current_index;
1332       ++reseek_count;
1333     }
1334     while (reseek_count--) {
1335       --queue->current_index;
1336       if (queue->packets[queue->current_index-1].pts100 != UINT_MAX &&
1337           queue->packets[queue->current_index-1].pts100 != lastpts)
1338         break;
1339     }
1340
1341     while (queue->current_index < queue->packets_size) {
1342       packet_t *pkt = queue->packets + queue->current_index;
1343       if (pkt->pts100 != UINT_MAX)
1344       if (pkt->pts100 <= pts100) {
1345         ++queue->current_index;
1346         *data = pkt->data;
1347         *timestamp = pkt->pts100;
1348         return pkt->size;
1349       } else break;
1350       else
1351           ++queue->current_index;
1352     }
1353   }
1354   return -1;
1355 }
1356
1357 int
1358 vobsub_get_next_packet(void *vobhandle, void** data, int* timestamp)
1359 {
1360   vobsub_t *vob = (vobsub_t *)vobhandle;
1361   if (vob->spu_streams && 0 <= vobsub_id && (unsigned) vobsub_id < vob->spu_streams_size) {
1362     packet_queue_t *queue = vob->spu_streams + vobsub_id;
1363     if (queue->current_index < queue->packets_size) {
1364       packet_t *pkt = queue->packets + queue->current_index;
1365       ++queue->current_index;
1366       *data = pkt->data;
1367       *timestamp = pkt->pts100;
1368       return pkt->size;
1369     }
1370   }
1371   return -1;
1372 }
1373
1374 void vobsub_seek(void * vobhandle, float pts)
1375 {
1376   vobsub_t * vob = (vobsub_t *)vobhandle;
1377   packet_queue_t * queue;
1378   int seek_pts100 = (int)pts * 90000;
1379
1380   if (vob->spu_streams && 0 <= vobsub_id && (unsigned) vobsub_id < vob->spu_streams_size) {
1381     /* do not seek if we don't know the id */
1382     if (vobsub_get_id(vob, vobsub_id) == NULL)
1383             return;
1384     queue = vob->spu_streams + vobsub_id;
1385     queue->current_index = 0;
1386     while (queue->current_index < queue->packets_size
1387             && (queue->packets + queue->current_index)->pts100 < seek_pts100)
1388       ++queue->current_index;
1389     if (queue->current_index > 0)
1390       --queue->current_index;
1391   }
1392 }
1393
1394 void
1395 vobsub_reset(void *vobhandle)
1396 {
1397     vobsub_t *vob = (vobsub_t *)vobhandle;
1398     if (vob->spu_streams) {
1399         unsigned int n = vob->spu_streams_size;
1400         while (n-- > 0)
1401             vob->spu_streams[n].current_index = 0;
1402     }
1403 }
1404
1405 /**********************************************************************
1406  * Vobsub output
1407  **********************************************************************/
1408
1409 typedef struct {
1410     FILE *fsub;
1411     FILE *fidx;
1412     unsigned int aid;
1413 } vobsub_out_t;
1414
1415 static void
1416 create_idx(vobsub_out_t *me, const unsigned int *palette, unsigned int orig_width, unsigned int orig_height)
1417 {
1418     int i;
1419     fprintf(me->fidx,
1420             "# VobSub index file, v7 (do not modify this line!)\n"
1421             "#\n"
1422             "# Generated by MPlayer " VERSION "\n"
1423             "# See <URL:http://www.mplayerhq.hu/> for more information about MPlayer\n"
1424             "# See <URL:http://vobsub.edensrising.com/> for more information about Vobsub\n"
1425             "#\n"
1426             "size: %ux%u\n",
1427             orig_width, orig_height);
1428     if (palette) {
1429         fputs("palette:", me->fidx);
1430         for (i = 0; i < 16; ++i) {
1431             const double y = palette[i] >> 16 & 0xff,
1432                 u = (palette[i] >> 8 & 0xff) - 128.0,
1433                 v = (palette[i] & 0xff) - 128.0;
1434             if (i)
1435                 putc(',', me->fidx);
1436             fprintf(me->fidx, " %02x%02x%02x",
1437                     av_clip_uint8(y + 1.4022 * u),
1438                     av_clip_uint8(y - 0.3456 * u - 0.7145 * v),
1439                     av_clip_uint8(y + 1.7710 * v));
1440         }
1441         putc('\n', me->fidx);
1442     }
1443
1444     fprintf(me->fidx,"# ON: displays only forced subtitles, OFF: shows everything\n"
1445             "forced subs: OFF\n");
1446 }
1447
1448 void *
1449 vobsub_out_open(const char *basename, const unsigned int *palette,
1450                 unsigned int orig_width, unsigned int orig_height,
1451                 const char *id, unsigned int index)
1452 {
1453     vobsub_out_t *result = NULL;
1454     char *filename;
1455     filename = malloc(strlen(basename) + 5);
1456     if (filename) {
1457         result = malloc(sizeof(vobsub_out_t));
1458         if (result) {
1459             result->aid = index;
1460             strcpy(filename, basename);
1461             strcat(filename, ".sub");
1462             result->fsub = fopen(filename, "ab");
1463             if (result->fsub == NULL)
1464                 perror("Error: vobsub_out_open subtitle file open failed");
1465             strcpy(filename, basename);
1466             strcat(filename, ".idx");
1467             result->fidx = fopen(filename, "ab");
1468             if (result->fidx) {
1469                 if (ftell(result->fidx) == 0){
1470                     create_idx(result, palette, orig_width, orig_height);
1471                     /* Make the selected language the default language */
1472                     fprintf(result->fidx, "\n# Language index in use\nlangidx: %u\n", index);
1473                 }
1474                 fprintf(result->fidx, "\nid: %s, index: %u\n", id ? id : "xx", index);
1475                 /* So that we can check the file now */
1476                 fflush(result->fidx);
1477             }
1478             else
1479                 perror("Error: vobsub_out_open index file open failed");
1480             free(filename);
1481         }
1482     }
1483     return result;
1484 }
1485
1486 void
1487 vobsub_out_close(void *me)
1488 {
1489     vobsub_out_t *vob = (vobsub_out_t*)me;
1490     if (vob->fidx)
1491         fclose(vob->fidx);
1492     if (vob->fsub)
1493         fclose(vob->fsub);
1494     free(vob);
1495 }
1496
1497 void
1498 vobsub_out_output(void *me, const unsigned char *packet, int len, double pts)
1499 {
1500     static double last_pts;
1501     static int last_pts_set = 0;
1502     vobsub_out_t *vob = (vobsub_out_t*)me;
1503     if (vob->fsub) {
1504         /*  Windows' Vobsub require that every packet is exactly 2kB long */
1505         unsigned char buffer[2048];
1506         unsigned char *p;
1507         int remain = 2048;
1508         /* Do not output twice a line with the same timestamp, this
1509            breaks Windows' Vobsub */
1510         if (vob->fidx && (!last_pts_set || last_pts != pts)) {
1511             static unsigned int last_h = 9999, last_m = 9999, last_s = 9999, last_ms = 9999;
1512             unsigned int h, m, ms;
1513             double s;
1514             s = pts;
1515             h = s / 3600;
1516             s -= h * 3600;
1517             m = s / 60;
1518             s -= m * 60;
1519             ms = (s - (unsigned int) s) * 1000;
1520             if (ms >= 1000)     /* prevent overfolws or bad float stuff */
1521                 ms = 0;
1522             if (h != last_h || m != last_m || (unsigned int) s != last_s || ms != last_ms) {
1523                 fprintf(vob->fidx, "timestamp: %02u:%02u:%02u:%03u, filepos: %09lx\n",
1524                         h, m, (unsigned int) s, ms, ftell(vob->fsub));
1525                 last_h = h;
1526                 last_m = m;
1527                 last_s = (unsigned int) s;
1528                 last_ms = ms;
1529             }
1530         }
1531         last_pts = pts;
1532         last_pts_set = 1;
1533
1534         /* Packet start code: Windows' Vobsub needs this */
1535         p = buffer;
1536         *p++ = 0;               /* 0x00 */
1537         *p++ = 0;
1538         *p++ = 1;
1539         *p++ = 0xba;
1540         *p++ = 0x40;
1541         memset(p, 0, 9);
1542         p += 9;
1543         {   /* Packet */
1544             static unsigned char last_pts[5] = { 0, 0, 0, 0, 0};
1545             unsigned char now_pts[5];
1546             int pts_len, pad_len, datalen = len;
1547             pts *= 90000;
1548             now_pts[0] = 0x21 | (((unsigned long)pts >> 29) & 0x0e);
1549             now_pts[1] = ((unsigned long)pts >> 22) & 0xff;
1550             now_pts[2] = 0x01 | (((unsigned long)pts >> 14) & 0xfe);
1551             now_pts[3] = ((unsigned long)pts >> 7) & 0xff;
1552             now_pts[4] = 0x01 | (((unsigned long)pts << 1) & 0xfe);
1553             pts_len = memcmp(last_pts, now_pts, sizeof(now_pts)) ? sizeof(now_pts) : 0;
1554             memcpy(last_pts, now_pts, sizeof(now_pts));
1555
1556             datalen += 3;       /* Version, PTS_flags, pts_len */
1557             datalen += pts_len;
1558             datalen += 1;       /* AID */
1559             pad_len = 2048 - (p - buffer) - 4 /* MPEG ID */ - 2 /* payload len */ - datalen;
1560             /* XXX - Go figure what should go here!  In any case the
1561                packet has to be completly filled.  If I can fill it
1562                with padding (0x000001be) latter I'll do that.  But if
1563                there is only room for 6 bytes then I can not write a
1564                padding packet.  So I add some padding in the PTS
1565                field.  This looks like a dirty kludge.  Oh well... */
1566             if (pad_len < 0) {
1567                 /* Packet is too big.  Let's try ommiting the PTS field */
1568                 datalen -= pts_len;
1569                 pts_len = 0;
1570                 pad_len = 0;
1571             }
1572             else if (pad_len > 6)
1573                 pad_len = 0;
1574             datalen += pad_len;
1575
1576             *p++ = 0;           /* 0x0e */
1577             *p++ = 0;
1578             *p++ = 1;
1579             *p++ = 0xbd;
1580
1581             *p++ = (datalen >> 8) & 0xff; /* length of payload */
1582             *p++ = datalen & 0xff;
1583             *p++ = 0x80;                /* System-2 (.VOB) stream */
1584             *p++ = pts_len ? 0x80 : 0x00; /* pts_flags */
1585             *p++ = pts_len + pad_len;
1586             memcpy(p, now_pts, pts_len);
1587             p += pts_len;
1588             memset(p, 0, pad_len);
1589             p += pad_len;
1590         }
1591         *p++ = 0x20 |  vob->aid; /* aid */
1592         if (fwrite(buffer, p - buffer, 1, vob->fsub) != 1
1593             || fwrite(packet, len, 1, vob->fsub) != 1)
1594             perror("ERROR: vobsub write failed");
1595         else
1596             remain -= p - buffer + len;
1597
1598         /* Padding */
1599         if (remain >= 6) {
1600             p = buffer;
1601             *p++ = 0x00;
1602             *p++ = 0x00;
1603             *p++ = 0x01;
1604             *p++ = 0xbe;
1605             *p++ = (remain - 6) >> 8;
1606             *p++ = (remain - 6) & 0xff;
1607             /* for better compression, blank this */
1608             memset(buffer + 6, 0, remain - (p - buffer));
1609             if (fwrite(buffer, remain, 1, vob->fsub) != 1)
1610                 perror("ERROR: vobsub padding write failed");
1611         }
1612         else if (remain > 0) {
1613             /* I don't know what to output.  But anyway the block
1614                needs to be 2KB big */
1615             memset(buffer, 0, remain);
1616             if (fwrite(buffer, remain, 1, vob->fsub) != 1)
1617                 perror("ERROR: vobsub blank padding write failed");
1618         }
1619         else if (remain < 0)
1620             fprintf(stderr,
1621                     "\nERROR: wrong thing happenned...\n"
1622                     "  I wrote a %i data bytes spu packet and that's too long\n", len);
1623     }
1624 }