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