Mercurial > mplayer.hg
view vobsub.c @ 5390:386f6fcc92d5
libavcodec 2pass broke mencoder if divx4encore was not available. fixed.
author | rfelker |
---|---|
date | Fri, 29 Mar 2002 07:20:44 +0000 |
parents | 3af2729c5c87 |
children | 71372b7125cf |
line wrap: on
line source
/* * Some code freely inspired from VobSub <URL:http://vobsub.edensrising.com>, * with kind permission from Gabest <gabest@freemail.hu> */ /* #define HAVE_GETLINE */ #include <ctype.h> #include <errno.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 "stream.h" #include "vobsub.h" #include "spudec.h" #include "mp_msg.h" extern int vobsub_id; extern int verbose; #ifdef HAVE_GETLINE extern ssize_t getline(char **, size_t *, FILE *); #else /* FIXME This should go into a general purpose library or even a separate file. */ static ssize_t getline (char **lineptr, size_t *n, FILE *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 = fgetc(stream); c != EOF; c = fgetc(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; } #endif /********************************************************************** * MPEG parsing **********************************************************************/ typedef struct { stream_t *stream; unsigned int pts; int aid; unsigned char *packet; unsigned int packet_reserve; unsigned int packet_size; } mpeg_t; static mpeg_t * mpeg_open(const char *filename) { mpeg_t *res = malloc(sizeof(mpeg_t)); int err = res == NULL; if (!err) { int fd; res->pts = 0; res->aid = -1; res->packet = NULL; res->packet_size = 0; res->packet_reserve = 0; fd = open(filename, O_RDONLY); err = fd < 0; if (!err) { res->stream = new_stream(fd, STREAMTYPE_FILE); err = res->stream == NULL; if (err) close(fd); } if (err) free(res); } return err ? NULL : res; } static void mpeg_free(mpeg_t *mpeg) { int fd; if (mpeg->packet) free(mpeg->packet); fd = mpeg->stream->fd; free_stream(mpeg->stream); close(fd); free(mpeg); } static int mpeg_eof(mpeg_t *mpeg) { return stream_eof(mpeg->stream); } static off_t mpeg_tell(mpeg_t *mpeg) { return stream_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 (stream_read(mpeg->stream, buf, 4) != 4) return -1; while (memcmp(buf, wanted, sizeof(wanted)) != 0) { c = stream_read_char(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 = stream_read_char(mpeg->stream); if (c < 0) return -1; if ((c & 0xc0) == 0x40) version = 4; else if ((c & 0xf0) == 0x20) version = 2; else { fprintf(stderr, "Unsupported MPEG version: 0x%02x", c); return -1; } if (version == 4) { if (!stream_skip(mpeg->stream, 9)) return -1; } else if (version == 2) { if (!stream_skip(mpeg->stream, 7)) return -1; } else abort(); break; case 0xbd: /* packet */ if (stream_read(mpeg->stream, buf, 2) != 2) return -1; len = buf[0] << 8 | buf[1]; idx = mpeg_tell(mpeg); c = stream_read_char(mpeg->stream); if (c < 0) return -1; if ((c & 0xC0) == 0x40) { /* skip STD scale & size */ if (stream_read_char(mpeg->stream) < 0) return -1; c = stream_read_char(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 = stream_read_char(mpeg->stream); if (c < 0) return -1; pts_flags = c; c = stream_read_char(mpeg->stream); if (c < 0) return -1; hdrlen = c; dataidx = mpeg_tell(mpeg) + hdrlen; if (dataidx > idx + len) { fprintf(stderr, "Invalid header length: %d (total length: %d, idx: %d, dataidx: %d)\n", hdrlen, len, idx, dataidx); return -1; } if ((pts_flags & 0xc0) == 0x80) { if (stream_read(mpeg->stream, buf, 5) != 5) return -1; if (!(((buf[0] & 0xf0) == 0x20) && (buf[0] & 1) && (buf[2] & 1) && (buf[4] & 1))) { fprintf(stderr, "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)) / 900; } else /* if ((pts_flags & 0xc0) == 0xc0) */ { /* what's this? */ /* abort(); */ } stream_seek(mpeg->stream, dataidx); mpeg->aid = stream_read_char(mpeg->stream); if (mpeg->aid < 0) { fprintf(stderr, "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) { perror("malloc failure"); mpeg->packet_reserve = 0; mpeg->packet_size = 0; return -1; } if (stream_read(mpeg->stream, mpeg->packet, mpeg->packet_size) != mpeg->packet_size) { perror("stream_read failure"); mpeg->packet_size = 0; return -1; } idx = len; } break; case 0xbe: /* Padding */ if (stream_read(mpeg->stream, buf, 2) != 2) return -1; len = buf[0] << 8 | buf[1]; if (len > 0 && !stream_skip(mpeg->stream, len)) return -1; break; default: if (0xc0 <= buf[3] && buf[3] < 0xf0) { /* MPEG audio or video */ if (stream_read(mpeg->stream, buf, 2) != 2) return -1; len = buf[0] << 8 | buf[1]; if (len > 0 && !stream_skip(mpeg->stream, len)) return -1; } else { fprintf(stderr, "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) { perror("realloc failure"); return -1; } queue->packets = tmp; queue->packets_reserve *= 2; } else { queue->packets = malloc(sizeof(packet_t)); if (queue->packets == NULL) { perror("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; } /********************************************************************** * Vosub **********************************************************************/ typedef struct { void *spudec; unsigned int palette[16]; unsigned int orig_frame_width, orig_frame_height; /* index */ packet_queue_t *spu_streams; unsigned int spu_streams_size; unsigned int spu_streams_current; } 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) { perror("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) { perror("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) { perror("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; if (verbose) fprintf(stderr, "[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, unsigned int ms) { packet_queue_t *queue; packet_t *pkt; if (vob->spu_streams == 0) { fprintf(stderr, "[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 / 10; 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, ms + 1000 * (s + 60 * (m + 60 * h))); } static int vobsub_parse_size(vobsub_t *vob, const char *line) { // size: WWWxHHH char *p; while (isspace(*line)) ++line; if (!isdigit(*line)) return -1; vob->orig_frame_width = strtoul(line, &p, 10); if (*p != 'x') return -1; ++p; vob->orig_frame_height = strtoul(p, NULL, 10); return 0; } static int vobsub_parse_palette(vobsub_t *vob, const char *line) { // palette: XXXXXX, XXXXXX, XXXXXX, XXXXXX, XXXXXX, XXXXXX, XXXXXX, XXXXXX, XXXXXX, XXXXXX, XXXXXX, XXXXXX, XXXXXX, XXXXXX, XXXXXX, XXXXXX unsigned int n; n = 0; while (1) { const char *p; while (isspace(*line)) ++line; p = line; while (isxdigit(*p)) ++p; if (p - line != 6) return -1; vob->palette[n++] = strtoul(line, NULL, 16); if (n == 16) break; if (*p == ',') ++p; line = p; } return 0; } static int vobsub_parse_one_line(vobsub_t *vob, FILE *fd) { ssize_t line_size; int res = -1; do { int line_reserve = 0; char *line = NULL; line_size = getline(&line, &line_reserve, fd); if (line_size < 0) { if (line) free(line); break; } if (*line == 0 || *line == '\r' || *line == '\n' || *line == '#') continue; else if (strncmp("id:", line, 3) == 0) res = vobsub_parse_id(vob, line + 3); else if (strncmp("palette:", line, 8) == 0) res = vobsub_parse_palette(vob, line + 8); else if (strncmp("size:", line, 5) == 0) res = vobsub_parse_size(vob, line + 5); else if (strncmp("timestamp:", line, 10) == 0) res = vobsub_parse_timestamp(vob, line + 10); else { if (verbose) fprintf(stderr, "vobsub: ignoring %s", line); continue; } if (res < 0) fprintf(stderr, "ERROR in %s", line); break; } while (1); return res; } int vobsub_parse_ifo(const char *const name, unsigned int *palette, unsigned int *width, unsigned int *height, int force) { int res = -1; FILE *fd = fopen(name, "rb"); if (fd == NULL) perror("Can't open IFO file"); else { // parse IFO header unsigned char block[0x800]; const char *const ifo_magic = "DVDVIDEO-VTS"; if (fread(block, sizeof(block), 1, fd) != 1) { if (force) perror("Can't read IFO header"); } else if (memcmp(block, ifo_magic, strlen(ifo_magic) + 1)) fprintf(stderr, "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: fprintf(stderr, "Unknown resolution %d \n", resolution); } if (fseek(fd, pgci_sector * sizeof(block), SEEK_SET) || fread(block, sizeof(block), 1, fd) != 1) perror("Can't read IFO PGCI"); 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]; } res = 0; } } fclose(fd); } return res; } void * vobsub_open(const char *const name, const int force) { vobsub_t *vob = malloc(sizeof(vobsub_t)); if (vob) { char *buf; vob->spudec = NULL; vob->orig_frame_width = 0; vob->orig_frame_height = 0; vob->spu_streams = NULL; vob->spu_streams_size = 0; vob->spu_streams_current = 0; buf = malloc((strlen(name) + 5) * sizeof(char)); if (buf) { FILE *fd; mpeg_t *mpg; /* read in the info file */ strcpy(buf, name); strcat(buf, ".ifo"); vobsub_parse_ifo(buf, vob->palette, &vob->orig_frame_width, &vob->orig_frame_height, force); /* read in the index */ strcpy(buf, name); strcat(buf, ".idx"); fd = fopen(buf, "rb"); if (fd == NULL) { if(force) perror("VobSub: Can't open IDX file"); else return NULL; } else { while (vobsub_parse_one_line(vob, fd) >= 0) /* NOOP */ ; fclose(fd); } if (vob->orig_frame_width && vob->orig_frame_height) vob->spudec = spudec_new_scaled(vob->palette, vob->orig_frame_width, vob->orig_frame_height); /* read the indexed mpeg_stream */ strcpy(buf, name); strcat(buf, ".sub"); mpg = mpeg_open(buf); if (mpg == NULL) { if(force) perror("VobSub: Can't open SUB file"); } 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)) perror("mpeg_run error"); 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; last_pts_diff = pkt->pts100 - mpg->pts; /* FIXME: should not use mpg_sub internal informations, make a copy */ pkt->data = mpg->packet; pkt->size = mpg->packet_size; mpg->packet = NULL; mpg->packet_reserve = 0; mpg->packet_size = 0; } } else fprintf(stderr, "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; mpeg_free(mpg); } free(buf); } } return vob; } void vobsub_close(void *this) { vobsub_t *vob = (vobsub_t *)this; if (vob->spudec) spudec_free(vob->spudec); 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); } void vobsub_draw(void *this, int dxs, int dys, void (*draw_alpha)(int x0,int y0, int w,int h, unsigned char* src, unsigned char *srca, int stride)) { vobsub_t *vob = (vobsub_t *)this; if (vob->spudec) spudec_draw_scaled(vob->spudec, dxs, dys, draw_alpha); } void vobsub_process(void *vobhandle, float pts) { vobsub_t *vob = (vobsub_t *)vobhandle; unsigned int pts100 = 100 * pts; if (vob->spudec) { spudec_heartbeat(vob->spudec, pts100); if (vob->spu_streams && 0 <= vobsub_id && (unsigned) vobsub_id < vob->spu_streams_size) { packet_queue_t *queue = vob->spu_streams + vobsub_id; while (queue->current_index < queue->packets_size) { packet_t *pkt = queue->packets + queue->current_index; if (pkt->pts100 <= pts100) { spudec_assemble(vob->spudec, pkt->data, pkt->size, pkt->pts100); ++queue->current_index; } else break; } } } } void vobsub_reset(void *vobhandle) { vobsub_t *vob = (vobsub_t *)vobhandle; if (vob->spu_streams) { unsigned int n = vob->spu_streams_size; while (n-- > 0) vob->spu_streams[n].current_index = 0; } if (vob->spudec) spudec_reset(vob->spudec); }