?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
libcdio
[mplayer.git] / vobsub.c
blobb6d06106f305271b680653f5f7e773837842747a
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 }
MPlayer development branch