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