Mercurial > mplayer.hg
changeset 32459:1a605463f62b
Move vobsub.[ch] and unrar_exec.[ch] to the sub directory.
author | cigaes |
---|---|
date | Wed, 27 Oct 2010 16:58:23 +0000 |
parents | ab8fb38f28b1 |
children | d80bbc5868de |
files | Makefile TOOLS/subrip.c cfg-mplayer.h command.c libmpdemux/demux_mkv.c mencoder.c mpcommon.c mplayer.c sub/spudec.c sub/unrar_exec.c sub/unrar_exec.h sub/vobsub.c sub/vobsub.h unrar_exec.c unrar_exec.h vobsub.c vobsub.h |
diffstat | 17 files changed, 1800 insertions(+), 1799 deletions(-) [+] |
line wrap: on
line diff
--- a/Makefile Wed Oct 27 16:48:21 2010 +0000 +++ b/Makefile Wed Oct 27 16:58:23 2010 +0000 @@ -278,7 +278,7 @@ SRCS_COMMON-$(TV_V4L1) += stream/tvi_v4l.c stream/audio_in.c SRCS_COMMON-$(TV_V4L2) += stream/tvi_v4l2.c stream/audio_in.c -SRCS_COMMON-$(UNRAR_EXEC) += unrar_exec.c +SRCS_COMMON-$(UNRAR_EXEC) += sub/unrar_exec.c SRCS_COMMON-$(VCD) += stream/stream_vcd.c SRCS_COMMON-$(VORBIS) += libmpcodecs/ad_libvorbis.c \ libmpdemux/demux_ogg.c @@ -347,7 +347,6 @@ playtree.c \ playtreeparser.c \ subopt-helper.c \ - vobsub.c \ libaf/af.c \ libaf/af_center.c \ libaf/af_channels.c \ @@ -522,6 +521,7 @@ sub/spudec.c \ sub/sub_cc.c \ sub/subreader.c \ + sub/vobsub.c \ $(SRCS_COMMON-yes) @@ -1059,8 +1059,9 @@ TOOLS/bmovl-test$(EXESUF): -lSDL_image -TOOLS/subrip$(EXESUF): vobsub.o sub/spudec.o unrar_exec.o libvo/aclib.o \ - ffmpeg/libswscale/libswscale.a ffmpeg/libavutil/libavutil.a $(TEST_OBJS) +TOOLS/subrip$(EXESUF): sub/vobsub.o sub/spudec.o sub/unrar_exec.o \ + libvo/aclib.o ffmpeg/libswscale/libswscale.a ffmpeg/libavutil/libavutil.a \ + $(TEST_OBJS) TOOLS/vfw2menc$(EXESUF): -lwinmm -lole32
--- a/TOOLS/subrip.c Wed Oct 27 16:48:21 2010 +0000 +++ b/TOOLS/subrip.c Wed Oct 27 16:58:23 2010 +0000 @@ -31,7 +31,7 @@ #include <sys/types.h> #include <sys/wait.h> #include "libvo/video_out.h" -#include "vobsub.h" +#include "sub/vobsub.h" #include "sub/spudec.h" /* linking hacks */
--- a/cfg-mplayer.h Wed Oct 27 16:48:21 2010 +0000 +++ b/cfg-mplayer.h Wed Oct 27 16:58:23 2010 +0000 @@ -35,7 +35,7 @@ #include "libvo/vo_fbdev.h" #include "libvo/vo_zr.h" #include "mp_fifo.h" -#include "unrar_exec.h" +#include "sub/unrar_exec.h" const m_option_t vd_conf[]={
--- a/command.c Wed Oct 27 16:48:21 2010 +0000 +++ b/command.c Wed Oct 27 16:58:23 2010 +0000 @@ -44,7 +44,7 @@ #include "mixer.h" #include "libmpcodecs/dec_video.h" #include "libmpcodecs/dec_teletext.h" -#include "vobsub.h" +#include "sub/vobsub.h" #include "sub/spudec.h" #include "path.h" #include "stream/tv.h"
--- a/libmpdemux/demux_mkv.c Wed Oct 27 16:48:21 2010 +0000 +++ b/libmpdemux/demux_mkv.c Wed Oct 27 16:58:23 2010 +0000 @@ -39,7 +39,7 @@ #include "mp_msg.h" #include "help_mp.h" -#include "vobsub.h" +#include "sub/vobsub.h" #include "sub/subreader.h" #include "libvo/sub.h"
--- a/mencoder.c Wed Oct 27 16:48:21 2010 +0000 +++ b/mencoder.c Wed Oct 27 16:58:23 2010 +0000 @@ -95,7 +95,7 @@ #include "parser-mecmd.h" #include "path.h" #include "sub/spudec.h" -#include "vobsub.h" +#include "sub/vobsub.h" #include "eosd.h" #include "mencoder.h"
--- a/mpcommon.c Wed Oct 27 16:48:21 2010 +0000 +++ b/mpcommon.c Wed Oct 27 16:58:23 2010 +0000 @@ -36,7 +36,7 @@ #include "mp_msg.h" #include "sub/spudec.h" #include "version.h" -#include "vobsub.h" +#include "sub/vobsub.h" #include "sub/av_sub.h" #include "libmpcodecs/dec_teletext.h" #include "libavutil/intreadwrite.h"
--- a/mplayer.c Wed Oct 27 16:48:21 2010 +0000 +++ b/mplayer.c Wed Oct 27 16:58:23 2010 +0000 @@ -122,7 +122,7 @@ #include "playtreeparser.h" #include "sub/spudec.h" #include "sub/subreader.h" -#include "vobsub.h" +#include "sub/vobsub.h" #include "eosd.h" #include "osdep/getch2.h" #include "osdep/timer.h"
--- a/sub/spudec.c Wed Oct 27 16:48:21 2010 +0000 +++ b/sub/spudec.c Wed Oct 27 16:58:23 2010 +0000 @@ -40,7 +40,7 @@ #include "libvo/sub.h" #include "libvo/video_out.h" #include "sub/spudec.h" -#include "vobsub.h" +#include "sub/vobsub.h" #include "libavutil/avutil.h" #include "libavutil/intreadwrite.h" #include "libswscale/swscale.h"
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/sub/unrar_exec.c Wed Oct 27 16:58:23 2010 +0000 @@ -0,0 +1,235 @@ +/* + * List files and extract file from rars by using external executable unrar. + * + * Copyright (C) 2005 Jindrich Makovicka <makovick gmail com> + * Copyright (C) 2007 Ulion <ulion2002 gmail com> + * + * This file is part of MPlayer. + * + * MPlayer is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * MPlayer is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License along + * with MPlayer; if not, write to the Free Software Foundation, Inc., + * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. + */ + +#include <sys/wait.h> +#include <unistd.h> +#include <fcntl.h> +#include <stdlib.h> +#include <stdio.h> +#include <string.h> +#include <errno.h> +#include <locale.h> +#include "sub/unrar_exec.h" + +#include "mp_msg.h" + +#define UNRAR_LIST 1 +#define UNRAR_EXTRACT 2 + +char* unrar_executable = NULL; + +static FILE* launch_pipe(pid_t *apid, const char *executable, int action, + const char *archive, const char *filename) +{ + if (!executable || access(executable, R_OK | X_OK)) return NULL; + if (access(archive, R_OK)) return NULL; + { + int mypipe[2]; + pid_t pid; + + if (pipe(mypipe)) { + mp_msg(MSGT_GLOBAL, MSGL_ERR, "UnRAR: Cannot create pipe.\n"); + return NULL; + } + + pid = fork(); + if (pid == 0) { + /* This is the child process. Execute the unrar executable. */ + close(mypipe[0]); + // Close MPlayer's stdin, stdout and stderr so the unrar binary + // can not mess them up. + // TODO: Close all other files except the pipe. + close(0); close(1); close(2); + // Assign new stdin, stdout and stderr and check they actually got the + // right descriptors. + if (open("/dev/null", O_RDONLY) != 0 || dup(mypipe[1]) != 1 + || open("/dev/null", O_WRONLY) != 2) + _exit(EXIT_FAILURE); + if (action == UNRAR_LIST) + execl(executable, executable, "v", archive, NULL); + else if (action == UNRAR_EXTRACT) + execl(executable, executable, "p", "-inul", "-p-", + archive,filename,NULL); + mp_msg(MSGT_GLOBAL, MSGL_ERR, "UnRAR: Cannot execute %s\n", executable); + _exit(EXIT_FAILURE); + } + if (pid < 0) { + /* The fork failed. Report failure. */ + mp_msg(MSGT_GLOBAL, MSGL_ERR, "UnRAR: Fork failed\n"); + return NULL; + } + /* This is the parent process. Prepare the pipe stream. */ + close(mypipe[1]); + *apid = pid; + if (action == UNRAR_LIST) + mp_msg(MSGT_GLOBAL, MSGL_V, + "UnRAR: call unrar with command line: %s v %s\n", + executable, archive); + else if (action == UNRAR_EXTRACT) + mp_msg(MSGT_GLOBAL, MSGL_V, + "UnRAR: call unrar with command line: %s p -inul -p- %s %s\n", + executable, archive, filename); + return fdopen(mypipe[0], "r"); + } +} + +#define ALLOC_INCR 1 * 1024 * 1024 +int unrar_exec_get(unsigned char **output, unsigned long *size, + const char *filename, const char *rarfile) +{ + int bufsize = ALLOC_INCR, bytesread; + pid_t pid; + int status = 0; + FILE *rar_pipe; + + rar_pipe=launch_pipe(&pid,unrar_executable,UNRAR_EXTRACT,rarfile,filename); + if (!rar_pipe) return 0; + + *size = 0; + + *output = malloc(bufsize); + + while (*output) { + bytesread=fread(*output+*size, 1, bufsize-*size, rar_pipe); + if (bytesread <= 0) + break; + *size += bytesread; + if (*size == bufsize) { + char *p; + bufsize += ALLOC_INCR; + p = realloc(*output, bufsize); + if (!p) + free(*output); + *output = p; + } + } + fclose(rar_pipe); + pid = waitpid(pid, &status, 0); + if (!*output || !*size || (pid == -1 && errno != ECHILD) || + (pid > 0 && status)) { + free(*output); + *output = NULL; + *size = 0; + return 0; + } + if (bufsize > *size) { + char *p = realloc(*output, *size); + if (p) + *output = p; + } + mp_msg(MSGT_GLOBAL, MSGL_V, "UnRAR: got file %s len %lu\n", filename,*size); + return 1; +} + +#define PARSE_NAME 0 +#define PARSE_PROPS 1 + +int unrar_exec_list(const char *rarfile, ArchiveList_struct **list) +{ + char buf[1024], fname[1024]; + char *p; + pid_t pid; + int status = 0, file_num = -1, ignore_next_line = 0, state = PARSE_NAME; + FILE *rar_pipe; + ArchiveList_struct *alist = NULL, *current = NULL, *new; + + rar_pipe = launch_pipe(&pid, unrar_executable, UNRAR_LIST, rarfile, NULL); + if (!rar_pipe) return -1; + while (fgets(buf, sizeof(buf), rar_pipe)) { + int packsize, unpsize, ratio, day, month, year, hour, min; + int llen = strlen(buf); + // If read nothing, we got a file_num -1. + if (file_num == -1) + file_num = 0; + if (buf[llen-1] != '\n') + // The line is too long, ignore it. + ignore_next_line = 2; + if (ignore_next_line) { + --ignore_next_line; + state = PARSE_NAME; + continue; + } + // Trim the line. + while (llen > 0 && strchr(" \t\n\r\v\f", buf[llen-1])) + --llen; + buf[llen] = '\0'; + p = buf; + while (*p && strchr(" \t\n\r\v\f", *p)) + ++p; + if (!*p) { + state = PARSE_NAME; + continue; + } + + if (state == PARSE_PROPS && sscanf(p, "%d %d %d%% %d-%d-%d %d:%d", + &unpsize, &packsize, &ratio, &day, + &month, &year, &hour, &min) == 8) { + new = calloc(1, sizeof(ArchiveList_struct)); + if (!new) { + file_num = -1; + break; + } + if (!current) + alist = new; + else + current->next = new; + current = new; + current->item.Name = strdup(fname); + state = PARSE_NAME; + if (!current->item.Name) { + file_num = -1; + break; + } + current->item.PackSize = packsize; + current->item.UnpSize = unpsize; + ++file_num; + continue; + } + strcpy(fname, p); + state = PARSE_PROPS; + } + fclose(rar_pipe); + pid = waitpid(pid, &status, 0); + if (file_num < 0 || (pid == -1 && errno != ECHILD) || + (pid > 0 && status)) { + unrar_exec_freelist(alist); + return -1; + } + if (!alist) + return -1; + *list = alist; + mp_msg(MSGT_GLOBAL, MSGL_V, "UnRAR: list got %d files\n", file_num); + return file_num; +} + +void unrar_exec_freelist(ArchiveList_struct *list) +{ + ArchiveList_struct* tmp; + + while (list) { + tmp = list->next; + free(list->item.Name); + free(list); + list = tmp; + } +}
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/sub/unrar_exec.h Wed Oct 27 16:58:23 2010 +0000 @@ -0,0 +1,54 @@ +/* + * List files and extract file from rars by using external executable unrar. + * + * Copyright (C) 2005 Jindrich Makovicka <makovick gmail com> + * Copyright (C) 2007 Ulion <ulion2002 gmail com> + * + * This file is part of MPlayer. + * + * MPlayer is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * MPlayer is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License along + * with MPlayer; if not, write to the Free Software Foundation, Inc., + * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. + */ + +#ifndef MPLAYER_UNRAR_EXEC_H +#define MPLAYER_UNRAR_EXEC_H + +struct RAR_archive_entry +{ + char *Name; + unsigned long PackSize; + unsigned long UnpSize; + unsigned long FileCRC; + unsigned long FileTime; + unsigned char UnpVer; + unsigned char Method; + unsigned long FileAttr; +}; + +typedef struct archivelist +{ + struct RAR_archive_entry item; + struct archivelist *next; +} ArchiveList_struct; + +extern char* unrar_executable; + +int unrar_exec_get(unsigned char **output, unsigned long *size, + const char *filename, const char *rarfile); + +int unrar_exec_list(const char *rarfile, ArchiveList_struct **list); + +void unrar_exec_freelist(ArchiveList_struct *list); + +#endif /* MPLAYER_UNRAR_EXEC_H */
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/sub/vobsub.c Wed Oct 27 16:58:23 2010 +0000 @@ -0,0 +1,1451 @@ +/* + * Some code freely inspired from VobSub <URL:http://vobsub.edensrising.com>, + * with kind permission from Gabest <gabest@freemail.hu> + * + * This file is part of MPlayer. + * + * MPlayer is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * MPlayer is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License along + * with MPlayer; if not, write to the Free Software Foundation, Inc., + * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. + */ + +#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 "mpcommon.h" +#include "sub/vobsub.h" +#include "sub/spudec.h" +#include "mp_msg.h" +#include "sub/unrar_exec.h" +#include "libavutil/common.h" + +// 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 + **********************************************************************/ +#ifdef CONFIG_UNRAR_EXEC +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; + /* unrar_exec 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 = unrar_exec_get(&stream->data, &stream->size, 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; + num_files = unrar_exec_list(rar_filename, &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)) { + rc = unrar_exec_get(&stream->data, &stream->size, + lp->item.Name, rar_filename); + if (rc) + break; + } + } + } + unrar_exec_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; + int padding_was_here; + int merge; +} 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->padding_was_here = 1; + res->merge = 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(); + if (!mpeg->padding_was_here) + mpeg->merge = 1; + break; + case 0xbd: /* packet */ + if (rar_read(buf, 2, 1, mpeg->stream) != 1) + return -1; + mpeg->padding_was_here = 0; + 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; + mpeg->padding_was_here = 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]; + int delay; + unsigned int have_palette; + unsigned int orig_frame_width, orig_frame_height; + unsigned int origin_x, origin_y; + /* 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_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 formula. + // Here's reversed formula 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_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_one_line(vobsub_t *vob, rar_stream_t *fd, + unsigned char **extradata, + unsigned int *extradata_len) +{ + 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 || line_size > 1000000 || + *extradata_len+line_size > 10000000) { + break; + } + + *extradata = realloc(*extradata, *extradata_len+line_size+1); + memcpy(*extradata+*extradata_len, line, line_size); + *extradata_len += line_size; + (*extradata)[*extradata_len] = 0; + + 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("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 { + 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 = 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) +{ + unsigned char *extradata = NULL; + unsigned int extradata_len = 0; + vobsub_t *vob = calloc(1, sizeof(vobsub_t)); + if (spu) + *spu = NULL; + if (vobsubid == -2) + vobsubid = vobsub_id; + if (vob) { + char *buf; + 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, &extradata, &extradata_len) >= 0) + /* NOOP */ ; + rar_close(fd); + } + if (spu) + *spu = spudec_new_scaled(vob->palette, vob->orig_frame_width, vob->orig_frame_height, extradata, extradata_len); + if (extradata) + free(extradata); + + /* 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; + if (mpg->merge && queue->current_index > 0) { + packet_t *last = &queue->packets[queue->current_index - 1]; + pkt->pts100 = last->pts100; + } + mpg->merge = 0; + /* 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 = 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 = vobhandle; + return vob->spu_valid_streams_size; +} + +char *vobsub_get_id(void *vobhandle, unsigned int index) +{ + vobsub_t *vob = 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; +} + +int vobsub_set_from_lang(void *vobhandle, unsigned char * lang) +{ + int i; + vobsub_t *vob= 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; +} + +/// make sure we seek to the first packet of packets having same pts values. +static void vobsub_queue_reseek(packet_queue_t *queue, unsigned int pts100) +{ + int reseek_count = 0; + unsigned int lastpts = 0; + + if (queue->current_index > 0 + && (queue->packets[queue->current_index].pts100 == UINT_MAX + || queue->packets[queue->current_index].pts100 > pts100)) { + // possible pts seek previous, try to check it. + int i = 1; + while (queue->current_index >= i + && queue->packets[queue->current_index-i].pts100 == UINT_MAX) + ++i; + if (queue->current_index >= i + && queue->packets[queue->current_index-i].pts100 > pts100) + // pts seek previous confirmed, reseek from beginning + queue->current_index = 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; + } +} + +int vobsub_get_packet(void *vobhandle, float pts, void** data, int* timestamp) +{ + vobsub_t *vob = 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; + + vobsub_queue_reseek(queue, pts100); + + 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 = 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 = vobhandle; + packet_queue_t * queue; + int seek_pts100 = 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; + vobsub_queue_reseek(queue, seek_pts100); + } +} + +void vobsub_reset(void *vobhandle) +{ + vobsub_t *vob = 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 %s\n" + "# See <URL:http://www.mplayerhq.hu/> for more information about MPlayer\n" + "# See <URL:http://wiki.multimedia.cx/index.php?title=VOBsub> for more information about Vobsub\n" + "#\n" + "size: %ux%u\n", + mplayer_version, 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 = 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 = 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); + } +}
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/sub/vobsub.h Wed Oct 27 16:58:23 2010 +0000 @@ -0,0 +1,47 @@ +/* + * This file is part of MPlayer. + * + * MPlayer is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * MPlayer is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License along + * with MPlayer; if not, write to the Free Software Foundation, Inc., + * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. + */ + +#ifndef MPLAYER_VOBSUB_H +#define MPLAYER_VOBSUB_H + +void *vobsub_open(const char *subname, const char *const ifo, const int force, void** spu); +void vobsub_reset(void *vob); +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); +int vobsub_get_packet(void *vobhandle, float pts,void** data, int* timestamp); +int vobsub_get_next_packet(void *vobhandle, void** data, int* timestamp); +void vobsub_close(void *this); +unsigned int vobsub_get_indexes_count(void * /* vobhandle */); +char *vobsub_get_id(void * /* vobhandle */, unsigned int /* index */); + +/// Get vobsub id by its index in the valid streams. +int vobsub_get_id_by_index(void *vobhandle, unsigned int index); +/// Get index in the valid streams by vobsub id. +int vobsub_get_index_by_id(void *vobhandle, int id); + +/// Convert palette value in idx file to yuv. +unsigned int vobsub_palette_to_yuv(unsigned int pal); +/// Convert rgb value to yuv. +unsigned int vobsub_rgb_to_yuv(unsigned int rgb); + +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); +void vobsub_out_output(void *me, const unsigned char *packet, int len, double pts); +void vobsub_out_close(void *me); +int vobsub_set_from_lang(void *vobhandle, unsigned char * lang); +void vobsub_seek(void * vobhandle, float pts); + +#endif /* MPLAYER_VOBSUB_H */
--- a/unrar_exec.c Wed Oct 27 16:48:21 2010 +0000 +++ /dev/null Thu Jan 01 00:00:00 1970 +0000 @@ -1,235 +0,0 @@ -/* - * List files and extract file from rars by using external executable unrar. - * - * Copyright (C) 2005 Jindrich Makovicka <makovick gmail com> - * Copyright (C) 2007 Ulion <ulion2002 gmail com> - * - * This file is part of MPlayer. - * - * MPlayer is free software; you can redistribute it and/or modify - * it under the terms of the GNU General Public License as published by - * the Free Software Foundation; either version 2 of the License, or - * (at your option) any later version. - * - * MPlayer is distributed in the hope that it will be useful, - * but WITHOUT ANY WARRANTY; without even the implied warranty of - * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the - * GNU General Public License for more details. - * - * You should have received a copy of the GNU General Public License along - * with MPlayer; if not, write to the Free Software Foundation, Inc., - * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. - */ - -#include <sys/wait.h> -#include <unistd.h> -#include <fcntl.h> -#include <stdlib.h> -#include <stdio.h> -#include <string.h> -#include <errno.h> -#include <locale.h> -#include "unrar_exec.h" - -#include "mp_msg.h" - -#define UNRAR_LIST 1 -#define UNRAR_EXTRACT 2 - -char* unrar_executable = NULL; - -static FILE* launch_pipe(pid_t *apid, const char *executable, int action, - const char *archive, const char *filename) -{ - if (!executable || access(executable, R_OK | X_OK)) return NULL; - if (access(archive, R_OK)) return NULL; - { - int mypipe[2]; - pid_t pid; - - if (pipe(mypipe)) { - mp_msg(MSGT_GLOBAL, MSGL_ERR, "UnRAR: Cannot create pipe.\n"); - return NULL; - } - - pid = fork(); - if (pid == 0) { - /* This is the child process. Execute the unrar executable. */ - close(mypipe[0]); - // Close MPlayer's stdin, stdout and stderr so the unrar binary - // can not mess them up. - // TODO: Close all other files except the pipe. - close(0); close(1); close(2); - // Assign new stdin, stdout and stderr and check they actually got the - // right descriptors. - if (open("/dev/null", O_RDONLY) != 0 || dup(mypipe[1]) != 1 - || open("/dev/null", O_WRONLY) != 2) - _exit(EXIT_FAILURE); - if (action == UNRAR_LIST) - execl(executable, executable, "v", archive, NULL); - else if (action == UNRAR_EXTRACT) - execl(executable, executable, "p", "-inul", "-p-", - archive,filename,NULL); - mp_msg(MSGT_GLOBAL, MSGL_ERR, "UnRAR: Cannot execute %s\n", executable); - _exit(EXIT_FAILURE); - } - if (pid < 0) { - /* The fork failed. Report failure. */ - mp_msg(MSGT_GLOBAL, MSGL_ERR, "UnRAR: Fork failed\n"); - return NULL; - } - /* This is the parent process. Prepare the pipe stream. */ - close(mypipe[1]); - *apid = pid; - if (action == UNRAR_LIST) - mp_msg(MSGT_GLOBAL, MSGL_V, - "UnRAR: call unrar with command line: %s v %s\n", - executable, archive); - else if (action == UNRAR_EXTRACT) - mp_msg(MSGT_GLOBAL, MSGL_V, - "UnRAR: call unrar with command line: %s p -inul -p- %s %s\n", - executable, archive, filename); - return fdopen(mypipe[0], "r"); - } -} - -#define ALLOC_INCR 1 * 1024 * 1024 -int unrar_exec_get(unsigned char **output, unsigned long *size, - const char *filename, const char *rarfile) -{ - int bufsize = ALLOC_INCR, bytesread; - pid_t pid; - int status = 0; - FILE *rar_pipe; - - rar_pipe=launch_pipe(&pid,unrar_executable,UNRAR_EXTRACT,rarfile,filename); - if (!rar_pipe) return 0; - - *size = 0; - - *output = malloc(bufsize); - - while (*output) { - bytesread=fread(*output+*size, 1, bufsize-*size, rar_pipe); - if (bytesread <= 0) - break; - *size += bytesread; - if (*size == bufsize) { - char *p; - bufsize += ALLOC_INCR; - p = realloc(*output, bufsize); - if (!p) - free(*output); - *output = p; - } - } - fclose(rar_pipe); - pid = waitpid(pid, &status, 0); - if (!*output || !*size || (pid == -1 && errno != ECHILD) || - (pid > 0 && status)) { - free(*output); - *output = NULL; - *size = 0; - return 0; - } - if (bufsize > *size) { - char *p = realloc(*output, *size); - if (p) - *output = p; - } - mp_msg(MSGT_GLOBAL, MSGL_V, "UnRAR: got file %s len %lu\n", filename,*size); - return 1; -} - -#define PARSE_NAME 0 -#define PARSE_PROPS 1 - -int unrar_exec_list(const char *rarfile, ArchiveList_struct **list) -{ - char buf[1024], fname[1024]; - char *p; - pid_t pid; - int status = 0, file_num = -1, ignore_next_line = 0, state = PARSE_NAME; - FILE *rar_pipe; - ArchiveList_struct *alist = NULL, *current = NULL, *new; - - rar_pipe = launch_pipe(&pid, unrar_executable, UNRAR_LIST, rarfile, NULL); - if (!rar_pipe) return -1; - while (fgets(buf, sizeof(buf), rar_pipe)) { - int packsize, unpsize, ratio, day, month, year, hour, min; - int llen = strlen(buf); - // If read nothing, we got a file_num -1. - if (file_num == -1) - file_num = 0; - if (buf[llen-1] != '\n') - // The line is too long, ignore it. - ignore_next_line = 2; - if (ignore_next_line) { - --ignore_next_line; - state = PARSE_NAME; - continue; - } - // Trim the line. - while (llen > 0 && strchr(" \t\n\r\v\f", buf[llen-1])) - --llen; - buf[llen] = '\0'; - p = buf; - while (*p && strchr(" \t\n\r\v\f", *p)) - ++p; - if (!*p) { - state = PARSE_NAME; - continue; - } - - if (state == PARSE_PROPS && sscanf(p, "%d %d %d%% %d-%d-%d %d:%d", - &unpsize, &packsize, &ratio, &day, - &month, &year, &hour, &min) == 8) { - new = calloc(1, sizeof(ArchiveList_struct)); - if (!new) { - file_num = -1; - break; - } - if (!current) - alist = new; - else - current->next = new; - current = new; - current->item.Name = strdup(fname); - state = PARSE_NAME; - if (!current->item.Name) { - file_num = -1; - break; - } - current->item.PackSize = packsize; - current->item.UnpSize = unpsize; - ++file_num; - continue; - } - strcpy(fname, p); - state = PARSE_PROPS; - } - fclose(rar_pipe); - pid = waitpid(pid, &status, 0); - if (file_num < 0 || (pid == -1 && errno != ECHILD) || - (pid > 0 && status)) { - unrar_exec_freelist(alist); - return -1; - } - if (!alist) - return -1; - *list = alist; - mp_msg(MSGT_GLOBAL, MSGL_V, "UnRAR: list got %d files\n", file_num); - return file_num; -} - -void unrar_exec_freelist(ArchiveList_struct *list) -{ - ArchiveList_struct* tmp; - - while (list) { - tmp = list->next; - free(list->item.Name); - free(list); - list = tmp; - } -}
--- a/unrar_exec.h Wed Oct 27 16:48:21 2010 +0000 +++ /dev/null Thu Jan 01 00:00:00 1970 +0000 @@ -1,54 +0,0 @@ -/* - * List files and extract file from rars by using external executable unrar. - * - * Copyright (C) 2005 Jindrich Makovicka <makovick gmail com> - * Copyright (C) 2007 Ulion <ulion2002 gmail com> - * - * This file is part of MPlayer. - * - * MPlayer is free software; you can redistribute it and/or modify - * it under the terms of the GNU General Public License as published by - * the Free Software Foundation; either version 2 of the License, or - * (at your option) any later version. - * - * MPlayer is distributed in the hope that it will be useful, - * but WITHOUT ANY WARRANTY; without even the implied warranty of - * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the - * GNU General Public License for more details. - * - * You should have received a copy of the GNU General Public License along - * with MPlayer; if not, write to the Free Software Foundation, Inc., - * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. - */ - -#ifndef MPLAYER_UNRAR_EXEC_H -#define MPLAYER_UNRAR_EXEC_H - -struct RAR_archive_entry -{ - char *Name; - unsigned long PackSize; - unsigned long UnpSize; - unsigned long FileCRC; - unsigned long FileTime; - unsigned char UnpVer; - unsigned char Method; - unsigned long FileAttr; -}; - -typedef struct archivelist -{ - struct RAR_archive_entry item; - struct archivelist *next; -} ArchiveList_struct; - -extern char* unrar_executable; - -int unrar_exec_get(unsigned char **output, unsigned long *size, - const char *filename, const char *rarfile); - -int unrar_exec_list(const char *rarfile, ArchiveList_struct **list); - -void unrar_exec_freelist(ArchiveList_struct *list); - -#endif /* MPLAYER_UNRAR_EXEC_H */
--- a/vobsub.c Wed Oct 27 16:48:21 2010 +0000 +++ /dev/null Thu Jan 01 00:00:00 1970 +0000 @@ -1,1451 +0,0 @@ -/* - * Some code freely inspired from VobSub <URL:http://vobsub.edensrising.com>, - * with kind permission from Gabest <gabest@freemail.hu> - * - * This file is part of MPlayer. - * - * MPlayer is free software; you can redistribute it and/or modify - * it under the terms of the GNU General Public License as published by - * the Free Software Foundation; either version 2 of the License, or - * (at your option) any later version. - * - * MPlayer is distributed in the hope that it will be useful, - * but WITHOUT ANY WARRANTY; without even the implied warranty of - * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the - * GNU General Public License for more details. - * - * You should have received a copy of the GNU General Public License along - * with MPlayer; if not, write to the Free Software Foundation, Inc., - * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. - */ - -#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 "mpcommon.h" -#include "vobsub.h" -#include "sub/spudec.h" -#include "mp_msg.h" -#include "unrar_exec.h" -#include "libavutil/common.h" - -// 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 - **********************************************************************/ -#ifdef CONFIG_UNRAR_EXEC -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; - /* unrar_exec 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 = unrar_exec_get(&stream->data, &stream->size, 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; - num_files = unrar_exec_list(rar_filename, &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)) { - rc = unrar_exec_get(&stream->data, &stream->size, - lp->item.Name, rar_filename); - if (rc) - break; - } - } - } - unrar_exec_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; - int padding_was_here; - int merge; -} 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->padding_was_here = 1; - res->merge = 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(); - if (!mpeg->padding_was_here) - mpeg->merge = 1; - break; - case 0xbd: /* packet */ - if (rar_read(buf, 2, 1, mpeg->stream) != 1) - return -1; - mpeg->padding_was_here = 0; - 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; - mpeg->padding_was_here = 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]; - int delay; - unsigned int have_palette; - unsigned int orig_frame_width, orig_frame_height; - unsigned int origin_x, origin_y; - /* 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_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 formula. - // Here's reversed formula 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_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_one_line(vobsub_t *vob, rar_stream_t *fd, - unsigned char **extradata, - unsigned int *extradata_len) -{ - 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 || line_size > 1000000 || - *extradata_len+line_size > 10000000) { - break; - } - - *extradata = realloc(*extradata, *extradata_len+line_size+1); - memcpy(*extradata+*extradata_len, line, line_size); - *extradata_len += line_size; - (*extradata)[*extradata_len] = 0; - - 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("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 { - 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 = 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) -{ - unsigned char *extradata = NULL; - unsigned int extradata_len = 0; - vobsub_t *vob = calloc(1, sizeof(vobsub_t)); - if (spu) - *spu = NULL; - if (vobsubid == -2) - vobsubid = vobsub_id; - if (vob) { - char *buf; - 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, &extradata, &extradata_len) >= 0) - /* NOOP */ ; - rar_close(fd); - } - if (spu) - *spu = spudec_new_scaled(vob->palette, vob->orig_frame_width, vob->orig_frame_height, extradata, extradata_len); - if (extradata) - free(extradata); - - /* 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; - if (mpg->merge && queue->current_index > 0) { - packet_t *last = &queue->packets[queue->current_index - 1]; - pkt->pts100 = last->pts100; - } - mpg->merge = 0; - /* 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 = 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 = vobhandle; - return vob->spu_valid_streams_size; -} - -char *vobsub_get_id(void *vobhandle, unsigned int index) -{ - vobsub_t *vob = 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; -} - -int vobsub_set_from_lang(void *vobhandle, unsigned char * lang) -{ - int i; - vobsub_t *vob= 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; -} - -/// make sure we seek to the first packet of packets having same pts values. -static void vobsub_queue_reseek(packet_queue_t *queue, unsigned int pts100) -{ - int reseek_count = 0; - unsigned int lastpts = 0; - - if (queue->current_index > 0 - && (queue->packets[queue->current_index].pts100 == UINT_MAX - || queue->packets[queue->current_index].pts100 > pts100)) { - // possible pts seek previous, try to check it. - int i = 1; - while (queue->current_index >= i - && queue->packets[queue->current_index-i].pts100 == UINT_MAX) - ++i; - if (queue->current_index >= i - && queue->packets[queue->current_index-i].pts100 > pts100) - // pts seek previous confirmed, reseek from beginning - queue->current_index = 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; - } -} - -int vobsub_get_packet(void *vobhandle, float pts, void** data, int* timestamp) -{ - vobsub_t *vob = 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; - - vobsub_queue_reseek(queue, pts100); - - 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 = 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 = vobhandle; - packet_queue_t * queue; - int seek_pts100 = 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; - vobsub_queue_reseek(queue, seek_pts100); - } -} - -void vobsub_reset(void *vobhandle) -{ - vobsub_t *vob = 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 %s\n" - "# See <URL:http://www.mplayerhq.hu/> for more information about MPlayer\n" - "# See <URL:http://wiki.multimedia.cx/index.php?title=VOBsub> for more information about Vobsub\n" - "#\n" - "size: %ux%u\n", - mplayer_version, 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 = 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 = 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); - } -}
--- a/vobsub.h Wed Oct 27 16:48:21 2010 +0000 +++ /dev/null Thu Jan 01 00:00:00 1970 +0000 @@ -1,47 +0,0 @@ -/* - * This file is part of MPlayer. - * - * MPlayer is free software; you can redistribute it and/or modify - * it under the terms of the GNU General Public License as published by - * the Free Software Foundation; either version 2 of the License, or - * (at your option) any later version. - * - * MPlayer is distributed in the hope that it will be useful, - * but WITHOUT ANY WARRANTY; without even the implied warranty of - * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the - * GNU General Public License for more details. - * - * You should have received a copy of the GNU General Public License along - * with MPlayer; if not, write to the Free Software Foundation, Inc., - * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. - */ - -#ifndef MPLAYER_VOBSUB_H -#define MPLAYER_VOBSUB_H - -void *vobsub_open(const char *subname, const char *const ifo, const int force, void** spu); -void vobsub_reset(void *vob); -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); -int vobsub_get_packet(void *vobhandle, float pts,void** data, int* timestamp); -int vobsub_get_next_packet(void *vobhandle, void** data, int* timestamp); -void vobsub_close(void *this); -unsigned int vobsub_get_indexes_count(void * /* vobhandle */); -char *vobsub_get_id(void * /* vobhandle */, unsigned int /* index */); - -/// Get vobsub id by its index in the valid streams. -int vobsub_get_id_by_index(void *vobhandle, unsigned int index); -/// Get index in the valid streams by vobsub id. -int vobsub_get_index_by_id(void *vobhandle, int id); - -/// Convert palette value in idx file to yuv. -unsigned int vobsub_palette_to_yuv(unsigned int pal); -/// Convert rgb value to yuv. -unsigned int vobsub_rgb_to_yuv(unsigned int rgb); - -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); -void vobsub_out_output(void *me, const unsigned char *packet, int len, double pts); -void vobsub_out_close(void *me); -int vobsub_set_from_lang(void *vobhandle, unsigned char * lang); -void vobsub_seek(void * vobhandle, float pts); - -#endif /* MPLAYER_VOBSUB_H */