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