Mercurial > mplayer.hg
view vobsub.c @ 25349:b257a1967539
cosmetics: Move public function declarations together.
author | diego |
---|---|
date | Wed, 12 Dec 2007 08:45:05 +0000 |
parents | 078bdfd44751 |
children | f95cd1391ea0 |
line wrap: on
line source
/* * Some code freely inspired from VobSub <URL:http://vobsub.edensrising.com>, * with kind permission from Gabest <gabest@freemail.hu> */ #include <ctype.h> #include <errno.h> #include <limits.h> #include <stddef.h> #include <stdio.h> #include <stdlib.h> #include <string.h> #include <fcntl.h> #include <unistd.h> #include <sys/stat.h> #include <sys/types.h> #include "config.h" #include "version.h" #include "vobsub.h" #include "spudec.h" #include "mp_msg.h" #ifdef USE_UNRARLIB #include "unrarlib.h" #endif #include "libavutil/common.h" extern int vobsub_id; // Record the original -vobsubid set by commandline, since vobsub_id will be // overridden if slang match any of vobsub streams. static int vobsubid = -2; /********************************************************************** * RAR stream handling * The RAR file must have the same basename as the file to open * See <URL:http://www.unrarlib.org/> **********************************************************************/ #ifdef USE_UNRARLIB typedef struct { FILE *file; unsigned char *data; unsigned long size; unsigned long pos; } rar_stream_t; static rar_stream_t * rar_open(const char *const filename, const char *const mode) { rar_stream_t *stream; /* unrarlib can only read */ if (strcmp("r", mode) && strcmp("rb", mode)) { errno = EINVAL; return NULL; } stream = malloc(sizeof(rar_stream_t)); if (stream == NULL) return NULL; /* first try normal access */ stream->file = fopen(filename, mode); if (stream->file == NULL) { char *rar_filename; const char *p; int rc; /* Guess the RAR archive filename */ rar_filename = NULL; p = strrchr(filename, '.'); if (p) { ptrdiff_t l = p - filename; rar_filename = malloc(l + 5); if (rar_filename == NULL) { free(stream); return NULL; } strncpy(rar_filename, filename, l); strcpy(rar_filename + l, ".rar"); } else { rar_filename = malloc(strlen(filename) + 5); if (rar_filename == NULL) { free(stream); return NULL; } strcpy(rar_filename, filename); strcat(rar_filename, ".rar"); } /* get rid of the path if there is any */ if ((p = strrchr(filename, '/')) == NULL) { p = filename; } else { p++; } rc = urarlib_get(&stream->data, &stream->size, (char*) p, rar_filename, ""); if (!rc) { /* There is no matching filename in the archive. However, sometimes * the files we are looking for have been given arbitrary names in the archive. * Let's look for a file with an exact match in the extension only. */ int i, num_files, name_len; ArchiveList_struct *list, *lp; /* the cast in the next line is a hack to overcome a design flaw (IMHO) in unrarlib */ num_files = urarlib_list (rar_filename, (ArchiveList_struct *)&list); if (num_files > 0) { char *demanded_ext; demanded_ext = strrchr (p, '.'); if (demanded_ext) { int demanded_ext_len = strlen (demanded_ext); for (i=0, lp=list; i<num_files; i++, lp=lp->next) { name_len = strlen (lp->item.Name); if (name_len >= demanded_ext_len && !strcasecmp (lp->item.Name + name_len - demanded_ext_len, demanded_ext)) { if ((rc = urarlib_get(&stream->data, &stream->size, lp->item.Name, rar_filename, ""))) { break; } } } } urarlib_freelist (list); } if (!rc) { free(rar_filename); free(stream); return NULL; } } free(rar_filename); stream->pos = 0; } return stream; } static int rar_close(rar_stream_t *stream) { if (stream->file) return fclose(stream->file); free(stream->data); return 0; } static int rar_eof(rar_stream_t *stream) { if (stream->file) return feof(stream->file); return stream->pos >= stream->size; } static long rar_tell(rar_stream_t *stream) { if (stream->file) return ftell(stream->file); return stream->pos; } static int rar_seek(rar_stream_t *stream, long offset, int whence) { if (stream->file) return fseek(stream->file, offset, whence); switch (whence) { case SEEK_SET: if (offset < 0) { errno = EINVAL; return -1; } stream->pos = offset; break; case SEEK_CUR: if (offset < 0 && stream->pos < (unsigned long) -offset) { errno = EINVAL; return -1; } stream->pos += offset; break; case SEEK_END: if (offset < 0 && stream->size < (unsigned long) -offset) { errno = EINVAL; return -1; } stream->pos = stream->size + offset; break; default: errno = EINVAL; return -1; } return 0; } static int rar_getc(rar_stream_t *stream) { if (stream->file) return getc(stream->file); if (rar_eof(stream)) return EOF; return stream->data[stream->pos++]; } static size_t rar_read(void *ptr, size_t size, size_t nmemb, rar_stream_t *stream) { size_t res; unsigned long remain; if (stream->file) return fread(ptr, size, nmemb, stream->file); if (rar_eof(stream)) return 0; res = size * nmemb; remain = stream->size - stream->pos; if (res > remain) res = remain / size * size; memcpy(ptr, stream->data + stream->pos, res); stream->pos += res; res /= size; return res; } #else typedef FILE rar_stream_t; #define rar_open fopen #define rar_close fclose #define rar_eof feof #define rar_tell ftell #define rar_seek fseek #define rar_getc getc #define rar_read fread #endif /**********************************************************************/ static ssize_t vobsub_getline(char **lineptr, size_t *n, rar_stream_t *stream) { size_t res = 0; int c; if (*lineptr == NULL) { *lineptr = malloc(4096); if (*lineptr) *n = 4096; } else if (*n == 0) { char *tmp = realloc(*lineptr, 4096); if (tmp) { *lineptr = tmp; *n = 4096; } } if (*lineptr == NULL || *n == 0) return -1; for (c = rar_getc(stream); c != EOF; c = rar_getc(stream)) { if (res + 1 >= *n) { char *tmp = realloc(*lineptr, *n * 2); if (tmp == NULL) return -1; *lineptr = tmp; *n *= 2; } (*lineptr)[res++] = c; if (c == '\n') { (*lineptr)[res] = 0; return res; } } if (res == 0) return -1; (*lineptr)[res] = 0; return res; } /********************************************************************** * MPEG parsing **********************************************************************/ typedef struct { rar_stream_t *stream; unsigned int pts; int aid; unsigned char *packet; unsigned int packet_reserve; unsigned int packet_size; } mpeg_t; static mpeg_t * mpeg_open(const char *filename) { mpeg_t *res = malloc(sizeof(mpeg_t)); int err = res == NULL; if (!err) { res->pts = 0; res->aid = -1; res->packet = NULL; res->packet_size = 0; res->packet_reserve = 0; res->stream = rar_open(filename, "rb"); err = res->stream == NULL; if (err) perror("fopen Vobsub file failed"); if (err) free(res); } return err ? NULL : res; } static void mpeg_free(mpeg_t *mpeg) { if (mpeg->packet) free(mpeg->packet); if (mpeg->stream) rar_close(mpeg->stream); free(mpeg); } static int mpeg_eof(mpeg_t *mpeg) { return rar_eof(mpeg->stream); } static off_t mpeg_tell(mpeg_t *mpeg) { return rar_tell(mpeg->stream); } static int mpeg_run(mpeg_t *mpeg) { unsigned int len, idx, version; int c; /* Goto start of a packet, it starts with 0x000001?? */ const unsigned char wanted[] = { 0, 0, 1 }; unsigned char buf[5]; mpeg->aid = -1; mpeg->packet_size = 0; if (rar_read(buf, 4, 1, mpeg->stream) != 1) return -1; while (memcmp(buf, wanted, sizeof(wanted)) != 0) { c = rar_getc(mpeg->stream); if (c < 0) return -1; memmove(buf, buf + 1, 3); buf[3] = c; } switch (buf[3]) { case 0xb9: /* System End Code */ break; case 0xba: /* Packet start code */ c = rar_getc(mpeg->stream); if (c < 0) return -1; if ((c & 0xc0) == 0x40) version = 4; else if ((c & 0xf0) == 0x20) version = 2; else { mp_msg(MSGT_VOBSUB,MSGL_ERR, "VobSub: Unsupported MPEG version: 0x%02x\n", c); return -1; } if (version == 4) { if (rar_seek(mpeg->stream, 9, SEEK_CUR)) return -1; } else if (version == 2) { if (rar_seek(mpeg->stream, 7, SEEK_CUR)) return -1; } else abort(); break; case 0xbd: /* packet */ if (rar_read(buf, 2, 1, mpeg->stream) != 1) return -1; len = buf[0] << 8 | buf[1]; idx = mpeg_tell(mpeg); c = rar_getc(mpeg->stream); if (c < 0) return -1; if ((c & 0xC0) == 0x40) { /* skip STD scale & size */ if (rar_getc(mpeg->stream) < 0) return -1; c = rar_getc(mpeg->stream); if (c < 0) return -1; } if ((c & 0xf0) == 0x20) { /* System-1 stream timestamp */ /* Do we need this? */ abort(); } else if ((c & 0xf0) == 0x30) { /* Do we need this? */ abort(); } else if ((c & 0xc0) == 0x80) { /* System-2 (.VOB) stream */ unsigned int pts_flags, hdrlen, dataidx; c = rar_getc(mpeg->stream); if (c < 0) return -1; pts_flags = c; c = rar_getc(mpeg->stream); if (c < 0) return -1; hdrlen = c; dataidx = mpeg_tell(mpeg) + hdrlen; if (dataidx > idx + len) { mp_msg(MSGT_VOBSUB,MSGL_ERR, "Invalid header length: %d (total length: %d, idx: %d, dataidx: %d)\n", hdrlen, len, idx, dataidx); return -1; } if ((pts_flags & 0xc0) == 0x80) { if (rar_read(buf, 5, 1, mpeg->stream) != 1) return -1; if (!(((buf[0] & 0xf0) == 0x20) && (buf[0] & 1) && (buf[2] & 1) && (buf[4] & 1))) { mp_msg(MSGT_VOBSUB,MSGL_ERR, "vobsub PTS error: 0x%02x %02x%02x %02x%02x \n", buf[0], buf[1], buf[2], buf[3], buf[4]); mpeg->pts = 0; } else mpeg->pts = ((buf[0] & 0x0e) << 29 | buf[1] << 22 | (buf[2] & 0xfe) << 14 | buf[3] << 7 | (buf[4] >> 1)); } else /* if ((pts_flags & 0xc0) == 0xc0) */ { /* what's this? */ /* abort(); */ } rar_seek(mpeg->stream, dataidx, SEEK_SET); mpeg->aid = rar_getc(mpeg->stream); if (mpeg->aid < 0) { mp_msg(MSGT_VOBSUB,MSGL_ERR, "Bogus aid %d\n", mpeg->aid); return -1; } mpeg->packet_size = len - ((unsigned int) mpeg_tell(mpeg) - idx); if (mpeg->packet_reserve < mpeg->packet_size) { if (mpeg->packet) free(mpeg->packet); mpeg->packet = malloc(mpeg->packet_size); if (mpeg->packet) mpeg->packet_reserve = mpeg->packet_size; } if (mpeg->packet == NULL) { mp_msg(MSGT_VOBSUB,MSGL_FATAL,"malloc failure"); mpeg->packet_reserve = 0; mpeg->packet_size = 0; return -1; } if (rar_read(mpeg->packet, mpeg->packet_size, 1, mpeg->stream) != 1) { mp_msg(MSGT_VOBSUB,MSGL_ERR,"fread failure"); mpeg->packet_size = 0; return -1; } idx = len; } break; case 0xbe: /* Padding */ if (rar_read(buf, 2, 1, mpeg->stream) != 1) return -1; len = buf[0] << 8 | buf[1]; if (len > 0 && rar_seek(mpeg->stream, len, SEEK_CUR)) return -1; break; default: if (0xc0 <= buf[3] && buf[3] < 0xf0) { /* MPEG audio or video */ if (rar_read(buf, 2, 1, mpeg->stream) != 1) return -1; len = buf[0] << 8 | buf[1]; if (len > 0 && rar_seek(mpeg->stream, len, SEEK_CUR)) return -1; } else { mp_msg(MSGT_VOBSUB,MSGL_ERR,"unknown header 0x%02X%02X%02X%02X\n", buf[0], buf[1], buf[2], buf[3]); return -1; } } return 0; } /********************************************************************** * Packet queue **********************************************************************/ typedef struct { unsigned int pts100; off_t filepos; unsigned int size; unsigned char *data; } packet_t; typedef struct { char *id; packet_t *packets; unsigned int packets_reserve; unsigned int packets_size; unsigned int current_index; } packet_queue_t; static void packet_construct(packet_t *pkt) { pkt->pts100 = 0; pkt->filepos = 0; pkt->size = 0; pkt->data = NULL; } static void packet_destroy(packet_t *pkt) { if (pkt->data) free(pkt->data); } static void packet_queue_construct(packet_queue_t *queue) { queue->id = NULL; queue->packets = NULL; queue->packets_reserve = 0; queue->packets_size = 0; queue->current_index = 0; } static void packet_queue_destroy(packet_queue_t *queue) { if (queue->packets) { while (queue->packets_size--) packet_destroy(queue->packets + queue->packets_size); free(queue->packets); } return; } /* Make sure there is enough room for needed_size packets in the packet queue. */ static int packet_queue_ensure(packet_queue_t *queue, unsigned int needed_size) { if (queue->packets_reserve < needed_size) { if (queue->packets) { packet_t *tmp = realloc(queue->packets, 2 * queue->packets_reserve * sizeof(packet_t)); if (tmp == NULL) { mp_msg(MSGT_VOBSUB,MSGL_FATAL,"realloc failure"); return -1; } queue->packets = tmp; queue->packets_reserve *= 2; } else { queue->packets = malloc(sizeof(packet_t)); if (queue->packets == NULL) { mp_msg(MSGT_VOBSUB,MSGL_FATAL,"malloc failure"); return -1; } queue->packets_reserve = 1; } } return 0; } /* add one more packet */ static int packet_queue_grow(packet_queue_t *queue) { if (packet_queue_ensure(queue, queue->packets_size + 1) < 0) return -1; packet_construct(queue->packets + queue->packets_size); ++queue->packets_size; return 0; } /* insert a new packet, duplicating pts from the current one */ static int packet_queue_insert(packet_queue_t *queue) { packet_t *pkts; if (packet_queue_ensure(queue, queue->packets_size + 1) < 0) return -1; /* XXX packet_size does not reflect the real thing here, it will be updated a bit later */ memmove(queue->packets + queue->current_index + 2, queue->packets + queue->current_index + 1, sizeof(packet_t) * (queue->packets_size - queue->current_index - 1)); pkts = queue->packets + queue->current_index; ++queue->packets_size; ++queue->current_index; packet_construct(pkts + 1); pkts[1].pts100 = pkts[0].pts100; pkts[1].filepos = pkts[0].filepos; return 0; } /********************************************************************** * Vobsub **********************************************************************/ typedef struct { unsigned int palette[16]; unsigned int cuspal[4]; int delay; unsigned int custom; unsigned int have_palette; unsigned int orig_frame_width, orig_frame_height; unsigned int origin_x, origin_y; unsigned int forced_subs; /* index */ packet_queue_t *spu_streams; unsigned int spu_streams_size; unsigned int spu_streams_current; unsigned int spu_valid_streams_size; } vobsub_t; /* Make sure that the spu stream idx exists. */ static int vobsub_ensure_spu_stream(vobsub_t *vob, unsigned int index) { if (index >= vob->spu_streams_size) { /* This is a new stream */ if (vob->spu_streams) { packet_queue_t *tmp = realloc(vob->spu_streams, (index + 1) * sizeof(packet_queue_t)); if (tmp == NULL) { mp_msg(MSGT_VOBSUB,MSGL_ERR,"vobsub_ensure_spu_stream: realloc failure"); return -1; } vob->spu_streams = tmp; } else { vob->spu_streams = malloc((index + 1) * sizeof(packet_queue_t)); if (vob->spu_streams == NULL) { mp_msg(MSGT_VOBSUB,MSGL_ERR,"vobsub_ensure_spu_stream: malloc failure"); return -1; } } while (vob->spu_streams_size <= index) { packet_queue_construct(vob->spu_streams + vob->spu_streams_size); ++vob->spu_streams_size; } } return 0; } static int vobsub_add_id(vobsub_t *vob, const char *id, size_t idlen, const unsigned int index) { if (vobsub_ensure_spu_stream(vob, index) < 0) return -1; if (id && idlen) { if (vob->spu_streams[index].id) free(vob->spu_streams[index].id); vob->spu_streams[index].id = malloc(idlen + 1); if (vob->spu_streams[index].id == NULL) { mp_msg(MSGT_VOBSUB,MSGL_FATAL,"vobsub_add_id: malloc failure"); return -1; } vob->spu_streams[index].id[idlen] = 0; memcpy(vob->spu_streams[index].id, id, idlen); } vob->spu_streams_current = index; mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_VOBSUB_ID=%d\n", index); if (id && idlen) mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_VSID_%d_LANG=%s\n", index, vob->spu_streams[index].id); mp_msg(MSGT_VOBSUB,MSGL_V,"[vobsub] subtitle (vobsubid): %d language %s\n", index, vob->spu_streams[index].id); return 0; } static int vobsub_add_timestamp(vobsub_t *vob, off_t filepos, int ms) { packet_queue_t *queue; packet_t *pkt; if (vob->spu_streams == 0) { mp_msg(MSGT_VOBSUB,MSGL_WARN,"[vobsub] warning, binning some index entries. Check your index file\n"); return -1; } queue = vob->spu_streams + vob->spu_streams_current; if (packet_queue_grow(queue) >= 0) { pkt = queue->packets + (queue->packets_size - 1); pkt->filepos = filepos; pkt->pts100 = ms < 0 ? UINT_MAX : (unsigned int)ms * 90; return 0; } return -1; } static int vobsub_parse_id(vobsub_t *vob, const char *line) { // id: xx, index: n size_t idlen; const char *p, *q; p = line; while (isspace(*p)) ++p; q = p; while (isalpha(*q)) ++q; idlen = q - p; if (idlen == 0) return -1; ++q; while (isspace(*q)) ++q; if (strncmp("index:", q, 6)) return -1; q += 6; while (isspace(*q)) ++q; if (!isdigit(*q)) return -1; return vobsub_add_id(vob, p, idlen, atoi(q)); } static int vobsub_parse_timestamp(vobsub_t *vob, const char *line) { // timestamp: HH:MM:SS.mmm, filepos: 0nnnnnnnnn const char *p; int h, m, s, ms; off_t filepos; while (isspace(*line)) ++line; p = line; while (isdigit(*p)) ++p; if (p - line != 2) return -1; h = atoi(line); if (*p != ':') return -1; line = ++p; while (isdigit(*p)) ++p; if (p - line != 2) return -1; m = atoi(line); if (*p != ':') return -1; line = ++p; while (isdigit(*p)) ++p; if (p - line != 2) return -1; s = atoi(line); if (*p != ':') return -1; line = ++p; while (isdigit(*p)) ++p; if (p - line != 3) return -1; ms = atoi(line); if (*p != ',') return -1; line = p + 1; while (isspace(*line)) ++line; if (strncmp("filepos:", line, 8)) return -1; line += 8; while (isspace(*line)) ++line; if (! isxdigit(*line)) return -1; filepos = strtol(line, NULL, 16); return vobsub_add_timestamp(vob, filepos, vob->delay + ms + 1000 * (s + 60 * (m + 60 * h))); } static int vobsub_parse_size(vobsub_t *vob, const char *line) { // size: WWWxHHH char *p; while (isspace(*line)) ++line; if (!isdigit(*line)) return -1; vob->orig_frame_width = strtoul(line, &p, 10); if (*p != 'x') return -1; ++p; vob->orig_frame_height = strtoul(p, NULL, 10); return 0; } static int vobsub_parse_origin(vobsub_t *vob, const char *line) { // org: X,Y char *p; while (isspace(*line)) ++line; if (!isdigit(*line)) return -1; vob->origin_x = strtoul(line, &p, 10); if (*p != ',') return -1; ++p; vob->origin_y = strtoul(p, NULL, 10); return 0; } unsigned int vobsub_palette_to_yuv(unsigned int pal) { int r, g, b, y, u, v; // Palette in idx file is not rgb value, it was calculated by wrong forumla. // Here's reversed forumla of the one used to generate palette in idx file. r = pal >> 16 & 0xff; g = pal >> 8 & 0xff; b = pal & 0xff; y = av_clip_uint8( 0.1494 * r + 0.6061 * g + 0.2445 * b); u = av_clip_uint8( 0.6066 * r - 0.4322 * g - 0.1744 * b + 128); v = av_clip_uint8(-0.08435 * r - 0.3422 * g + 0.4266 * b + 128); y = y * 219 / 255 + 16; return y << 16 | u << 8 | v; } unsigned int vobsub_rgb_to_yuv(unsigned int rgb) { int r, g, b, y, u, v; r = rgb >> 16 & 0xff; g = rgb >> 8 & 0xff; b = rgb & 0xff; y = ( 0.299 * r + 0.587 * g + 0.114 * b) * 219 / 255 + 16.5; u = (-0.16874 * r - 0.33126 * g + 0.5 * b) * 224 / 255 + 128.5; v = ( 0.5 * r - 0.41869 * g - 0.08131 * b) * 224 / 255 + 128.5; return y << 16 | u << 8 | v; } static int vobsub_parse_palette(vobsub_t *vob, const char *line) { // palette: XXXXXX, XXXXXX, XXXXXX, XXXXXX, XXXXXX, XXXXXX, XXXXXX, XXXXXX, XXXXXX, XXXXXX, XXXXXX, XXXXXX, XXXXXX, XXXXXX, XXXXXX, XXXXXX unsigned int n; n = 0; while (1) { const char *p; int tmp; while (isspace(*line)) ++line; p = line; while (isxdigit(*p)) ++p; if (p - line != 6) return -1; tmp = strtoul(line, NULL, 16); vob->palette[n++] = vobsub_palette_to_yuv(tmp); if (n == 16) break; if (*p == ',') ++p; line = p; } vob->have_palette = 1; return 0; } static int vobsub_parse_custom(vobsub_t *vob, const char *line) { //custom colors: OFF/ON(0/1) if ((strncmp("ON", line + 15, 2) == 0)||strncmp("1", line + 15, 1) == 0) vob->custom=1; else if ((strncmp("OFF", line + 15, 3) == 0)||strncmp("0", line + 15, 1) == 0) vob->custom=0; else return -1; return 0; } static int vobsub_parse_cuspal(vobsub_t *vob, const char *line) { //colors: XXXXXX, XXXXXX, XXXXXX, XXXXXX unsigned int n, tmp; n = 0; line += 40; while(1){ const char *p; while (isspace(*line)) ++line; p=line; while (isxdigit(*p)) ++p; if (p - line !=6) return -1; tmp = strtoul(line, NULL, 16); vob->cuspal[n++] = vobsub_rgb_to_yuv(tmp); if (n==4) break; if(*p == ',') ++p; line = p; } return 0; } /* don't know how to use tridx */ static int vobsub_parse_tridx(const char *line) { //tridx: XXXX int tridx; tridx = strtoul((line + 26), NULL, 16); tridx = ((tridx&0x1000)>>12) | ((tridx&0x100)>>7) | ((tridx&0x10)>>2) | ((tridx&1)<<3); return tridx; } static int vobsub_parse_delay(vobsub_t *vob, const char *line) { int h, m, s, ms; int forward = 1; if (*(line + 7) == '+'){ forward = 1; line++; } else if (*(line + 7) == '-'){ forward = -1; line++; } mp_msg(MSGT_SPUDEC,MSGL_V, "forward=%d", forward); h = atoi(line + 7); mp_msg(MSGT_VOBSUB,MSGL_V, "h=%d," ,h); m = atoi(line + 10); mp_msg(MSGT_VOBSUB,MSGL_V, "m=%d,", m); s = atoi(line + 13); mp_msg(MSGT_VOBSUB,MSGL_V, "s=%d,", s); ms = atoi(line + 16); mp_msg(MSGT_VOBSUB,MSGL_V, "ms=%d", ms); vob->delay = (ms + 1000 * (s + 60 * (m + 60 * h))) * forward; return 0; } static int vobsub_set_lang(const char *line) { if (vobsub_id == -1) vobsub_id = atoi(line + 8); return 0; } static int vobsub_parse_forced_subs(vobsub_t *vob, const char *line) { const char *p; p = line; while (isspace(*p)) ++p; if (strncasecmp("on",p,2) == 0){ vob->forced_subs=~0; return 0; } else if (strncasecmp("off",p,3) == 0){ vob->forced_subs=0; return 0; } return -1; } static int vobsub_parse_one_line(vobsub_t *vob, rar_stream_t *fd) { ssize_t line_size; int res = -1; size_t line_reserve = 0; char *line = NULL; do { line_size = vobsub_getline(&line, &line_reserve, fd); if (line_size < 0) { break; } if (*line == 0 || *line == '\r' || *line == '\n' || *line == '#') continue; else if (strncmp("langidx:", line, 8) == 0) res = vobsub_set_lang(line); else if (strncmp("delay:", line, 6) == 0) res = vobsub_parse_delay(vob, line); else if (strncmp("id:", line, 3) == 0) res = vobsub_parse_id(vob, line + 3); else if (strncmp("palette:", line, 8) == 0) res = vobsub_parse_palette(vob, line + 8); else if (strncmp("size:", line, 5) == 0) res = vobsub_parse_size(vob, line + 5); else if (strncmp("org:", line, 4) == 0) res = vobsub_parse_origin(vob, line + 4); else if (strncmp("timestamp:", line, 10) == 0) res = vobsub_parse_timestamp(vob, line + 10); else if (strncmp("custom colors:", line, 14) == 0) //custom colors: ON/OFF, tridx: XXXX, colors: XXXXXX, XXXXXX, XXXXXX,XXXXXX res = vobsub_parse_cuspal(vob, line) + vobsub_parse_tridx(line) + vobsub_parse_custom(vob, line); else if (strncmp("forced subs:", line, 12) == 0) res = vobsub_parse_forced_subs(vob, line + 12); else { mp_msg(MSGT_VOBSUB,MSGL_V, "vobsub: ignoring %s", line); continue; } if (res < 0) mp_msg(MSGT_VOBSUB,MSGL_ERR, "ERROR in %s", line); break; } while (1); if (line) free(line); return res; } int vobsub_parse_ifo(void* this, const char *const name, unsigned int *palette, unsigned int *width, unsigned int *height, int force, int sid, char *langid) { vobsub_t *vob = (vobsub_t*)this; int res = -1; rar_stream_t *fd = rar_open(name, "rb"); if (fd == NULL) { if (force) mp_msg(MSGT_VOBSUB,MSGL_WARN, "VobSub: Can't open IFO file\n"); } else { // parse IFO header unsigned char block[0x800]; const char *const ifo_magic = "DVDVIDEO-VTS"; if (rar_read(block, sizeof(block), 1, fd) != 1) { if (force) mp_msg(MSGT_VOBSUB,MSGL_ERR, "VobSub: Can't read IFO header\n"); } else if (memcmp(block, ifo_magic, strlen(ifo_magic) + 1)) mp_msg(MSGT_VOBSUB,MSGL_ERR, "VobSub: Bad magic in IFO header\n"); else { unsigned long pgci_sector = block[0xcc] << 24 | block[0xcd] << 16 | block[0xce] << 8 | block[0xcf]; int standard = (block[0x200] & 0x30) >> 4; int resolution = (block[0x201] & 0x0c) >> 2; *height = standard ? 576 : 480; *width = 0; switch (resolution) { case 0x0: *width = 720; break; case 0x1: *width = 704; break; case 0x2: *width = 352; break; case 0x3: *width = 352; *height /= 2; break; default: mp_msg(MSGT_VOBSUB,MSGL_WARN,"Vobsub: Unknown resolution %d \n", resolution); } if (langid && 0 <= sid && sid < 32) { unsigned char *tmp = block + 0x256 + sid * 6 + 2; langid[0] = tmp[0]; langid[1] = tmp[1]; langid[2] = 0; } if (rar_seek(fd, pgci_sector * sizeof(block), SEEK_SET) || rar_read(block, sizeof(block), 1, fd) != 1) mp_msg(MSGT_VOBSUB,MSGL_ERR, "VobSub: Can't read IFO PGCI\n"); else { unsigned long idx; unsigned long pgc_offset = block[0xc] << 24 | block[0xd] << 16 | block[0xe] << 8 | block[0xf]; for (idx = 0; idx < 16; ++idx) { unsigned char *p = block + pgc_offset + 0xa4 + 4 * idx; palette[idx] = p[0] << 24 | p[1] << 16 | p[2] << 8 | p[3]; } if(vob) vob->have_palette = 1; res = 0; } } rar_close(fd); } return res; } void * vobsub_open(const char *const name,const char *const ifo,const int force,void** spu) { vobsub_t *vob = malloc(sizeof(vobsub_t)); if(spu) *spu = NULL; if (vobsubid == -2) vobsubid = vobsub_id; if (vob) { char *buf; vob->custom = 0; vob->have_palette = 0; vob->orig_frame_width = 0; vob->orig_frame_height = 0; vob->spu_streams = NULL; vob->spu_streams_size = 0; vob->spu_streams_current = 0; vob->spu_valid_streams_size = 0; vob->delay = 0; vob->forced_subs=0; buf = malloc(strlen(name) + 5); if (buf) { rar_stream_t *fd; mpeg_t *mpg; /* read in the info file */ if(!ifo) { strcpy(buf, name); strcat(buf, ".ifo"); vobsub_parse_ifo(vob,buf, vob->palette, &vob->orig_frame_width, &vob->orig_frame_height, force, -1, NULL); } else vobsub_parse_ifo(vob,ifo, vob->palette, &vob->orig_frame_width, &vob->orig_frame_height, force, -1, NULL); /* read in the index */ strcpy(buf, name); strcat(buf, ".idx"); fd = rar_open(buf, "rb"); if (fd == NULL) { if(force) mp_msg(MSGT_VOBSUB,MSGL_ERR,"VobSub: Can't open IDX file\n"); else { free(buf); free(vob); return NULL; } } else { while (vobsub_parse_one_line(vob, fd) >= 0) /* NOOP */ ; rar_close(fd); } /* if no palette in .idx then use custom colors */ if ((vob->custom == 0)&&(vob->have_palette!=1)) vob->custom = 1; if (spu && vob->orig_frame_width && vob->orig_frame_height) *spu = spudec_new_scaled_vobsub(vob->palette, vob->cuspal, vob->custom, vob->orig_frame_width, vob->orig_frame_height); /* read the indexed mpeg_stream */ strcpy(buf, name); strcat(buf, ".sub"); mpg = mpeg_open(buf); if (mpg == NULL) { if(force) mp_msg(MSGT_VOBSUB,MSGL_ERR,"VobSub: Can't open SUB file\n"); else { free(buf); free(vob); return NULL; } } else { long last_pts_diff = 0; while (!mpeg_eof(mpg)) { off_t pos = mpeg_tell(mpg); if (mpeg_run(mpg) < 0) { if (!mpeg_eof(mpg)) mp_msg(MSGT_VOBSUB,MSGL_ERR,"VobSub: mpeg_run error\n"); break; } if (mpg->packet_size) { if ((mpg->aid & 0xe0) == 0x20) { unsigned int sid = mpg->aid & 0x1f; if (vobsub_ensure_spu_stream(vob, sid) >= 0) { packet_queue_t *queue = vob->spu_streams + sid; /* get the packet to fill */ if (queue->packets_size == 0 && packet_queue_grow(queue) < 0) abort(); while (queue->current_index + 1 < queue->packets_size && queue->packets[queue->current_index + 1].filepos <= pos) ++queue->current_index; if (queue->current_index < queue->packets_size) { packet_t *pkt; if (queue->packets[queue->current_index].data) { /* insert a new packet and fix the PTS ! */ packet_queue_insert(queue); queue->packets[queue->current_index].pts100 = mpg->pts + last_pts_diff; } pkt = queue->packets + queue->current_index; if (pkt->pts100 != UINT_MAX) { if (queue->packets_size > 1) last_pts_diff = pkt->pts100 - mpg->pts; else pkt->pts100 = mpg->pts; /* FIXME: should not use mpg_sub internal informations, make a copy */ pkt->data = mpg->packet; pkt->size = mpg->packet_size; mpg->packet = NULL; mpg->packet_reserve = 0; mpg->packet_size = 0; } } } else mp_msg(MSGT_VOBSUB,MSGL_WARN, "don't know what to do with subtitle #%u\n", sid); } } } vob->spu_streams_current = vob->spu_streams_size; while (vob->spu_streams_current-- > 0) { vob->spu_streams[vob->spu_streams_current].current_index = 0; if (vobsubid == vob->spu_streams_current || vob->spu_streams[vob->spu_streams_current].packets_size > 0) ++vob->spu_valid_streams_size; } mpeg_free(mpg); } free(buf); } } return vob; } void vobsub_close(void *this) { vobsub_t *vob = (vobsub_t *)this; if (vob->spu_streams) { while (vob->spu_streams_size--) packet_queue_destroy(vob->spu_streams + vob->spu_streams_size); free(vob->spu_streams); } free(vob); } unsigned int vobsub_get_indexes_count(void *vobhandle) { vobsub_t *vob = (vobsub_t *) vobhandle; return vob->spu_valid_streams_size; } char * vobsub_get_id(void *vobhandle, unsigned int index) { vobsub_t *vob = (vobsub_t *) vobhandle; return (index < vob->spu_streams_size) ? vob->spu_streams[index].id : NULL; } int vobsub_get_id_by_index(void *vobhandle, unsigned int index) { vobsub_t *vob = vobhandle; int i, j; if (vob == NULL) return -1; for (i = 0, j = 0; i < vob->spu_streams_size; ++i) if (i == vobsubid || vob->spu_streams[i].packets_size > 0) { if (j == index) return i; ++j; } return -1; } int vobsub_get_index_by_id(void *vobhandle, int id) { vobsub_t *vob = vobhandle; int i, j; if (vob == NULL || id < 0 || id >= vob->spu_streams_size) return -1; if (id != vobsubid && !vob->spu_streams[id].packets_size) return -1; for (i = 0, j = 0; i < id; ++i) if (i == vobsubid || vob->spu_streams[i].packets_size > 0) ++j; return j; } unsigned int vobsub_get_forced_subs_flag(void const * const vobhandle) { if (vobhandle) return ((vobsub_t*) vobhandle)->forced_subs; else return 0; } int vobsub_set_from_lang(void *vobhandle, unsigned char * lang) { int i; vobsub_t *vob= (vobsub_t *) vobhandle; while(lang && strlen(lang) >= 2){ for(i=0; i < vob->spu_streams_size; i++) if (vob->spu_streams[i].id) if ((strncmp(vob->spu_streams[i].id, lang, 2)==0)){ vobsub_id=i; mp_msg(MSGT_VOBSUB, MSGL_INFO, "Selected VOBSUB language: %d language: %s\n", i, vob->spu_streams[i].id); return 0; } lang+=2;while (lang[0]==',' || lang[0]==' ') ++lang; } mp_msg(MSGT_VOBSUB, MSGL_WARN, "No matching VOBSUB language found!\n"); return -1; } int vobsub_get_packet(void *vobhandle, float pts,void** data, int* timestamp) { vobsub_t *vob = (vobsub_t *)vobhandle; unsigned int pts100 = 90000 * pts; if (vob->spu_streams && 0 <= vobsub_id && (unsigned) vobsub_id < vob->spu_streams_size) { packet_queue_t *queue = vob->spu_streams + vobsub_id; int reseek_count = 0; unsigned int lastpts = 0; while (queue->current_index < queue->packets_size && queue->packets[queue->current_index].pts100 <= pts100) { lastpts = queue->packets[queue->current_index].pts100; ++queue->current_index; ++reseek_count; } while (reseek_count--) { --queue->current_index; if (queue->packets[queue->current_index-1].pts100 != UINT_MAX && queue->packets[queue->current_index-1].pts100 != lastpts) break; } while (queue->current_index < queue->packets_size) { packet_t *pkt = queue->packets + queue->current_index; if (pkt->pts100 != UINT_MAX) if (pkt->pts100 <= pts100) { ++queue->current_index; *data = pkt->data; *timestamp = pkt->pts100; return pkt->size; } else break; else ++queue->current_index; } } return -1; } int vobsub_get_next_packet(void *vobhandle, void** data, int* timestamp) { vobsub_t *vob = (vobsub_t *)vobhandle; if (vob->spu_streams && 0 <= vobsub_id && (unsigned) vobsub_id < vob->spu_streams_size) { packet_queue_t *queue = vob->spu_streams + vobsub_id; if (queue->current_index < queue->packets_size) { packet_t *pkt = queue->packets + queue->current_index; ++queue->current_index; *data = pkt->data; *timestamp = pkt->pts100; return pkt->size; } } return -1; } void vobsub_seek(void * vobhandle, float pts) { vobsub_t * vob = (vobsub_t *)vobhandle; packet_queue_t * queue; int seek_pts100 = (int)pts * 90000; if (vob->spu_streams && 0 <= vobsub_id && (unsigned) vobsub_id < vob->spu_streams_size) { /* do not seek if we don't know the id */ if (vobsub_get_id(vob, vobsub_id) == NULL) return; queue = vob->spu_streams + vobsub_id; queue->current_index = 0; while (queue->current_index < queue->packets_size && (queue->packets + queue->current_index)->pts100 < seek_pts100) ++queue->current_index; if (queue->current_index > 0) --queue->current_index; } } void vobsub_reset(void *vobhandle) { vobsub_t *vob = (vobsub_t *)vobhandle; if (vob->spu_streams) { unsigned int n = vob->spu_streams_size; while (n-- > 0) vob->spu_streams[n].current_index = 0; } } /********************************************************************** * Vobsub output **********************************************************************/ typedef struct { FILE *fsub; FILE *fidx; unsigned int aid; } vobsub_out_t; static void create_idx(vobsub_out_t *me, const unsigned int *palette, unsigned int orig_width, unsigned int orig_height) { int i; fprintf(me->fidx, "# VobSub index file, v7 (do not modify this line!)\n" "#\n" "# Generated by MPlayer " VERSION "\n" "# See <URL:http://www.mplayerhq.hu/> for more information about MPlayer\n" "# See <URL:http://vobsub.edensrising.com/> for more information about Vobsub\n" "#\n" "size: %ux%u\n", orig_width, orig_height); if (palette) { fputs("palette:", me->fidx); for (i = 0; i < 16; ++i) { const double y = palette[i] >> 16 & 0xff, u = (palette[i] >> 8 & 0xff) - 128.0, v = (palette[i] & 0xff) - 128.0; if (i) putc(',', me->fidx); fprintf(me->fidx, " %02x%02x%02x", av_clip_uint8(y + 1.4022 * u), av_clip_uint8(y - 0.3456 * u - 0.7145 * v), av_clip_uint8(y + 1.7710 * v)); } putc('\n', me->fidx); } fprintf(me->fidx,"# ON: displays only forced subtitles, OFF: shows everything\n" "forced subs: OFF\n"); } void * vobsub_out_open(const char *basename, const unsigned int *palette, unsigned int orig_width, unsigned int orig_height, const char *id, unsigned int index) { vobsub_out_t *result = NULL; char *filename; filename = malloc(strlen(basename) + 5); if (filename) { result = malloc(sizeof(vobsub_out_t)); if (result) { result->aid = index; strcpy(filename, basename); strcat(filename, ".sub"); result->fsub = fopen(filename, "ab"); if (result->fsub == NULL) perror("Error: vobsub_out_open subtitle file open failed"); strcpy(filename, basename); strcat(filename, ".idx"); result->fidx = fopen(filename, "ab"); if (result->fidx) { if (ftell(result->fidx) == 0){ create_idx(result, palette, orig_width, orig_height); /* Make the selected language the default language */ fprintf(result->fidx, "\n# Language index in use\nlangidx: %u\n", index); } fprintf(result->fidx, "\nid: %s, index: %u\n", id ? id : "xx", index); /* So that we can check the file now */ fflush(result->fidx); } else perror("Error: vobsub_out_open index file open failed"); free(filename); } } return result; } void vobsub_out_close(void *me) { vobsub_out_t *vob = (vobsub_out_t*)me; if (vob->fidx) fclose(vob->fidx); if (vob->fsub) fclose(vob->fsub); free(vob); } void vobsub_out_output(void *me, const unsigned char *packet, int len, double pts) { static double last_pts; static int last_pts_set = 0; vobsub_out_t *vob = (vobsub_out_t*)me; if (vob->fsub) { /* Windows' Vobsub require that every packet is exactly 2kB long */ unsigned char buffer[2048]; unsigned char *p; int remain = 2048; /* Do not output twice a line with the same timestamp, this breaks Windows' Vobsub */ if (vob->fidx && (!last_pts_set || last_pts != pts)) { static unsigned int last_h = 9999, last_m = 9999, last_s = 9999, last_ms = 9999; unsigned int h, m, ms; double s; s = pts; h = s / 3600; s -= h * 3600; m = s / 60; s -= m * 60; ms = (s - (unsigned int) s) * 1000; if (ms >= 1000) /* prevent overfolws or bad float stuff */ ms = 0; if (h != last_h || m != last_m || (unsigned int) s != last_s || ms != last_ms) { fprintf(vob->fidx, "timestamp: %02u:%02u:%02u:%03u, filepos: %09lx\n", h, m, (unsigned int) s, ms, ftell(vob->fsub)); last_h = h; last_m = m; last_s = (unsigned int) s; last_ms = ms; } } last_pts = pts; last_pts_set = 1; /* Packet start code: Windows' Vobsub needs this */ p = buffer; *p++ = 0; /* 0x00 */ *p++ = 0; *p++ = 1; *p++ = 0xba; *p++ = 0x40; memset(p, 0, 9); p += 9; { /* Packet */ static unsigned char last_pts[5] = { 0, 0, 0, 0, 0}; unsigned char now_pts[5]; int pts_len, pad_len, datalen = len; pts *= 90000; now_pts[0] = 0x21 | (((unsigned long)pts >> 29) & 0x0e); now_pts[1] = ((unsigned long)pts >> 22) & 0xff; now_pts[2] = 0x01 | (((unsigned long)pts >> 14) & 0xfe); now_pts[3] = ((unsigned long)pts >> 7) & 0xff; now_pts[4] = 0x01 | (((unsigned long)pts << 1) & 0xfe); pts_len = memcmp(last_pts, now_pts, sizeof(now_pts)) ? sizeof(now_pts) : 0; memcpy(last_pts, now_pts, sizeof(now_pts)); datalen += 3; /* Version, PTS_flags, pts_len */ datalen += pts_len; datalen += 1; /* AID */ pad_len = 2048 - (p - buffer) - 4 /* MPEG ID */ - 2 /* payload len */ - datalen; /* XXX - Go figure what should go here! In any case the packet has to be completly filled. If I can fill it with padding (0x000001be) latter I'll do that. But if there is only room for 6 bytes then I can not write a padding packet. So I add some padding in the PTS field. This looks like a dirty kludge. Oh well... */ if (pad_len < 0) { /* Packet is too big. Let's try ommiting the PTS field */ datalen -= pts_len; pts_len = 0; pad_len = 0; } else if (pad_len > 6) pad_len = 0; datalen += pad_len; *p++ = 0; /* 0x0e */ *p++ = 0; *p++ = 1; *p++ = 0xbd; *p++ = (datalen >> 8) & 0xff; /* length of payload */ *p++ = datalen & 0xff; *p++ = 0x80; /* System-2 (.VOB) stream */ *p++ = pts_len ? 0x80 : 0x00; /* pts_flags */ *p++ = pts_len + pad_len; memcpy(p, now_pts, pts_len); p += pts_len; memset(p, 0, pad_len); p += pad_len; } *p++ = 0x20 | vob->aid; /* aid */ if (fwrite(buffer, p - buffer, 1, vob->fsub) != 1 || fwrite(packet, len, 1, vob->fsub) != 1) perror("ERROR: vobsub write failed"); else remain -= p - buffer + len; /* Padding */ if (remain >= 6) { p = buffer; *p++ = 0x00; *p++ = 0x00; *p++ = 0x01; *p++ = 0xbe; *p++ = (remain - 6) >> 8; *p++ = (remain - 6) & 0xff; /* for better compression, blank this */ memset(buffer + 6, 0, remain - (p - buffer)); if (fwrite(buffer, remain, 1, vob->fsub) != 1) perror("ERROR: vobsub padding write failed"); } else if (remain > 0) { /* I don't know what to output. But anyway the block needs to be 2KB big */ memset(buffer, 0, remain); if (fwrite(buffer, remain, 1, vob->fsub) != 1) perror("ERROR: vobsub blank padding write failed"); } else if (remain < 0) fprintf(stderr, "\nERROR: wrong thing happenned...\n" " I wrote a %i data bytes spu packet and that's too long\n", len); } }