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