dts decoder using libdca

1 files changed, 647 insertions, 0 deletions

diff --git a/plugins/dca/dcadec.c b/plugins/dca/dcadec.c
new file mode 100644
index 00000000..ae21ec81
--- /dev/null
+++ b/plugins/dca/dcadec.c
@@ -0,0 +1,647 @@
+/*
+ * dcadec.c
+ * Copyright (C) 2004 Gildas Bazin <gbazin@videolan.org>
+ * Copyright (C) 2000-2003 Michel Lespinasse <walken@zoy.org>
+ * Copyright (C) 1999-2000 Aaron Holtzman <aholtzma@ess.engr.uvic.ca>
+ *
+ * This file is part of libdca, a free DTS Coherent Acoustics stream decoder.
+ * See http://www.videolan.org/developers/libdca.html for updates.
+ *
+ * libdca 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.
+ *
+ * libdca 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 this program; if not, write to the
+ * Free Software Foundation, Inc.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ */
+
+#include "config.h"
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <errno.h>
+#include <getopt.h>
+#include <math.h>
+#include <signal.h>
+#ifdef HAVE_IO_H
+#include <fcntl.h>
+#include <io.h>
+#endif
+#include <inttypes.h>
+
+#include "dca.h"
+#include "audio_out.h"
+#include "gettimeofday.h"
+
+#define BUFFER_SIZE 24576
+#define HEADER_SIZE 14
+static uint8_t buffer[BUFFER_SIZE];
+static FILE * in_file;
+static int demux_track = 0;
+static int demux_pid = 0;
+static int demux_pes = 0;
+static int disable_accel = 0;
+static int disable_dynrng = 0;
+static int disable_adjust = 0;
+static float gain = 1;
+static ao_open_t * output_open = NULL;
+static ao_instance_t * output;
+static dca_state_t * state;
+static int sigint = 0;
+
+#ifdef HAVE_GETTIMEOFDAY
+
+static void print_fps (int final);
+
+static RETSIGTYPE signal_handler (int sig)
+{
+    sigint = 1;
+    signal (sig, SIG_DFL);
+    return (RETSIGTYPE)0;
+}
+
+static void print_fps (int final)
+{
+    static uint32_t frame_counter = 0;
+    static struct timeval tv_beg, tv_start;
+    static int total_elapsed;
+    static int last_count = 0;
+    struct timeval tv_end;
+    float fps, tfps;
+    int frames, elapsed;
+
+    gettimeofday (&tv_end, NULL);
+
+    if (!frame_counter) {
+	tv_start = tv_beg = tv_end;
+	signal (SIGINT, signal_handler);
+    }
+
+    elapsed = (tv_end.tv_sec - tv_beg.tv_sec) * 100 +
+	(tv_end.tv_usec - tv_beg.tv_usec) / 10000;
+    total_elapsed = (tv_end.tv_sec - tv_start.tv_sec) * 100 +
+	(tv_end.tv_usec - tv_start.tv_usec) / 10000;
+
+    if (final) {
+	if (total_elapsed)
+	    tfps = frame_counter * 100.0 / total_elapsed;
+	else
+	    tfps = 0;
+
+	fprintf (stderr,"\n%d frames decoded in %.2f seconds (%.2f fps)\n",
+		 frame_counter, total_elapsed / 100.0, tfps);
+
+	return;
+    }
+
+    frame_counter++;
+
+    if (elapsed < 50)	/* only display every 0.50 seconds */
+	return;
+
+    tv_beg = tv_end;
+    frames = frame_counter - last_count;
+
+    fps = frames * 100.0 / elapsed;
+    tfps = frame_counter * 100.0 / total_elapsed;
+
+    fprintf (stderr, "%d frames in %.2f sec (%.2f fps), "
+	     "%d last %.2f sec (%.2f fps)\033[K\r", frame_counter,
+	     total_elapsed / 100.0, tfps, frames, elapsed / 100.0, fps);
+
+    last_count = frame_counter;
+}
+
+#else /* !HAVE_GETTIMEOFDAY */
+
+static void print_fps (int final)
+{
+}
+
+#endif
+
+static void print_usage (char ** argv)
+{
+    int i;
+    ao_driver_t * drivers;
+
+    fprintf (stderr, "usage: "
+	     "%s [-h] [-o <mode>] [-s [<track>]] [-t <pid>] [-c] [-r] [-a] \\\n"
+	     "\t\t[-g <gain>] <file>\n"
+	     "\t-h\tdisplay help and available audio output modes\n"
+	     "\t-s\tuse program stream demultiplexer, track 0-7 or 0x80-0x87\n"
+	     "\t-t\tuse transport stream demultiplexer, pid 0x10-0x1ffe\n"
+	     "\t-T\tuse transport stream PES demultiplexer\n"
+	     "\t-c\tuse c implementation, disables all accelerations\n"
+	     "\t-r\tdisable dynamic range compression\n"
+	     "\t-a\tdisable level adjustment based on output mode\n"
+	     "\t-g\tadd specified gain in decibels, -96.0 to +96.0\n"
+	     "\t-o\taudio output mode\n", argv[0]);
+
+    drivers = ao_drivers ();
+    for (i = 0; drivers[i].name; i++)
+	fprintf (stderr, "\t\t\t%s\n", drivers[i].name);
+
+    exit (1);
+}
+
+static void handle_args (int argc, char ** argv)
+{
+    int c;
+    ao_driver_t * drivers;
+    int i;
+    char * s;
+
+    drivers = ao_drivers ();
+    while ((c = getopt (argc, argv, "hs::t:Tcrag:o:")) != -1)
+	switch (c) {
+	case 'o':
+	    for (i = 0; drivers[i].name != NULL; i++)
+		if (strcmp (drivers[i].name, optarg) == 0)
+		    output_open = drivers[i].open;
+	    if (output_open == NULL) {
+		fprintf (stderr, "Invalid video driver: %s\n", optarg);
+		print_usage (argv);
+	    }
+	    break;
+
+	case 's':
+	    demux_track = 0x88;
+	    if (optarg != NULL) {
+		demux_track = strtol (optarg, &s, 0);
+		if (demux_track < 0x88)
+		    demux_track += 0x88;
+		if (demux_track < 0x88 || demux_track > 0x8f || *s) {
+		    fprintf (stderr, "Invalid track number: %s\n", optarg);
+		    print_usage (argv);
+		}
+	    }
+	    break;
+
+	case 't':
+	    demux_pid = strtol (optarg, &s, 0);
+	    if (demux_pid < 0x10 || demux_pid > 0x1ffe || *s) {
+		fprintf (stderr, "Invalid pid: %s\n", optarg);
+		print_usage (argv);
+	    }
+	    break;
+
+	case 'T':
+	    demux_pes = 1;
+	    break;
+
+	case 'c':
+	    disable_accel = 1;
+	    break;
+
+	case 'r':
+	    disable_dynrng = 1;
+	    break;
+
+	case 'a':
+	    disable_adjust = 1;
+	    break;
+
+	case 'g':
+	    gain = strtod (optarg, &s);
+	    if (gain < -96 || gain > 96 || *s) {
+		fprintf (stderr, "Invalid gain: %s\n", optarg);
+		print_usage (argv);
+	    }
+	    gain = pow (2, gain / 6);
+	    break;
+
+	default:
+	    print_usage (argv);
+	}
+
+    /* -o not specified, use a default driver */
+    if (output_open == NULL)
+	output_open = drivers[0].open;
+
+    if (optind < argc) {
+	in_file = fopen (argv[optind], "rb");
+	if (!in_file) {
+	    fprintf (stderr, "%s - could not open file %s\n", strerror (errno),
+		     argv[optind]);
+	    exit (1);
+	}
+    } else
+	in_file = stdin;
+}
+
+static void dca_decode_data (uint8_t * start, uint8_t * end)
+{
+    static uint8_t buf[BUFFER_SIZE];
+    static uint8_t * bufptr = buf;
+    static uint8_t * bufpos = buf + HEADER_SIZE;
+
+    /*
+     * sample_rate and flags are static because this routine could
+     * exit between the dca_syncinfo() and the ao_setup(), and we want
+     * to have the same values when we get back !
+     */
+
+    static int sample_rate;
+    static int frame_length;
+    static int flags;
+    int bit_rate;
+    int len;
+
+    while (1) {
+	len = end - start;
+	if (!len)
+	    break;
+	if (len > bufpos - bufptr)
+	    len = bufpos - bufptr;
+	memcpy (bufptr, start, len);
+	bufptr += len;
+	start += len;
+	if (bufptr == bufpos) {
+	    if (bufpos == buf + HEADER_SIZE) {
+		int length;
+
+		length = dca_syncinfo (state, buf, &flags, &sample_rate,
+                                       &bit_rate, &frame_length);
+		if (!length) {
+		    fprintf (stderr, "skip\n");
+		    for (bufptr = buf; bufptr < buf + HEADER_SIZE-1; bufptr++)
+			bufptr[0] = bufptr[1];
+		    continue;
+		}
+		bufpos = buf + length;
+	    } else {
+		level_t level;
+		sample_t bias;
+		int i;
+
+		if (output->setup (output, sample_rate, &flags, &level, &bias))
+		    goto error;
+		if (!disable_adjust)
+		    flags |= DCA_ADJUST_LEVEL;
+		level = (level_t) (level * gain);
+		if (dca_frame (state, buf, &flags, &level, bias))
+		    goto error;
+		if (disable_dynrng)
+		    dca_dynrng (state, NULL, NULL);
+		for (i = 0; i < dca_blocks_num (state); i++) {
+		    if (dca_block (state))
+		        goto error;
+		    if (output->play (output, flags, dca_samples (state)))
+			goto error;
+		}
+		bufptr = buf;
+		bufpos = buf + HEADER_SIZE;
+		print_fps (0);
+		continue;
+	    error:
+		fprintf (stderr, "error\n");
+		bufptr = buf;
+		bufpos = buf + HEADER_SIZE;
+	    }
+	}
+    }
+}
+
+#define DEMUX_PAYLOAD_START 1
+static int demux (uint8_t * buf, uint8_t * end, int flags)
+{
+    static int mpeg1_skip_table[16] = {
+	0, 0, 4, 9, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0
+    };
+
+    /*
+     * the demuxer keeps some state between calls:
+     * if "state" = DEMUX_HEADER, then "head_buf" contains the first
+     *     "bytes" bytes from some header.
+     * if "state" == DEMUX_DATA, then we need to copy "bytes" bytes
+     *     of ES data before the next header.
+     * if "state" == DEMUX_SKIP, then we need to skip "bytes" bytes
+     *     of data before the next header.
+     *
+     * NEEDBYTES makes sure we have the requested number of bytes for a
+     * header. If we dont, it copies what we have into head_buf and returns,
+     * so that when we come back with more data we finish decoding this header.
+     *
+     * DONEBYTES updates "buf" to point after the header we just parsed.
+     */
+
+#define DEMUX_HEADER 0
+#define DEMUX_DATA 1
+#define DEMUX_SKIP 2
+    static int state = DEMUX_SKIP;
+    static int state_bytes = 0;
+    static uint8_t head_buf[268];
+
+    uint8_t * header;
+    int bytes;
+    int len;
+
+#define NEEDBYTES(x)						\
+    do {							\
+	int missing;						\
+								\
+	missing = (x) - bytes;					\
+	if (missing > 0) {					\
+	    if (header == head_buf) {				\
+		if (missing <= end - buf) {			\
+		    memcpy (header + bytes, buf, missing);	\
+		    buf += missing;				\
+		    bytes = (x);				\
+		} else {					\
+		    memcpy (header + bytes, buf, end - buf);	\
+		    state_bytes = bytes + end - buf;		\
+		    return 0;					\
+		}						\
+	    } else {						\
+		memcpy (head_buf, header, bytes);		\
+		state = DEMUX_HEADER;				\
+		state_bytes = bytes;				\
+		return 0;					\
+	    }							\
+	}							\
+    } while (0)
+
+#define DONEBYTES(x)		\
+    do {			\
+	if (header != head_buf)	\
+	    buf = header + (x);	\
+    } while (0)
+
+    if (flags & DEMUX_PAYLOAD_START)
+	goto payload_start;
+    switch (state) {
+    case DEMUX_HEADER:
+	if (state_bytes > 0) {
+	    header = head_buf;
+	    bytes = state_bytes;
+	    goto continue_header;
+	}
+	break;
+    case DEMUX_DATA:
+	if (demux_pid || (state_bytes > end - buf)) {
+	    dca_decode_data (buf, end);
+	    state_bytes -= end - buf;
+	    return 0;
+	}
+	dca_decode_data (buf, buf + state_bytes);
+	buf += state_bytes;
+	break;
+    case DEMUX_SKIP:
+	if (demux_pid || (state_bytes > end - buf)) {
+	    state_bytes -= end - buf;
+	    return 0;
+	}
+	buf += state_bytes;
+	break;
+    }
+
+    while (1) {
+	if (demux_pid) {
+	    state = DEMUX_SKIP;
+	    return 0;
+	}
+    payload_start:
+	header = buf;
+	bytes = end - buf;
+    continue_header:
+	NEEDBYTES (4);
+	if (header[0] || header[1] || (header[2] != 1)) {
+	    if (demux_pid) {
+		state = DEMUX_SKIP;
+		return 0;
+	    } else if (header != head_buf) {
+		buf++;
+		goto payload_start;
+	    } else {
+		header[0] = header[1];
+		header[1] = header[2];
+		header[2] = header[3];
+		bytes = 3;
+		goto continue_header;
+	    }
+	}
+	if (demux_pid || demux_pes) {
+	    if (header[3] != 0xbd) {
+		fprintf (stderr, "bad stream id %x\n", header[3]);
+		exit (1);
+	    }
+	    NEEDBYTES (9);
+	    if ((header[6] & 0xc8) != 0x88) {	/* not mpeg2 */
+		fprintf (stderr, "bad multiplex - not mpeg2\n");
+		exit (1);
+	    }
+	    len = 9 + header[8];
+	    NEEDBYTES (len);
+	    DONEBYTES (len);
+	    bytes = 6 + (header[4] << 8) + header[5] - len;
+	    if (bytes > end - buf) {
+		dca_decode_data (buf, end);
+		state = DEMUX_DATA;
+		state_bytes = bytes - (end - buf);
+		return 0;
+	    } else if (bytes > 0) {
+		dca_decode_data (buf, buf + bytes);
+		buf += bytes;
+	    }
+	} else switch (header[3]) {
+	case 0xb9:	/* program end code */
+	    /* DONEBYTES (4); */
+	    /* break;         */
+	    return 1;
+	case 0xba:	/* pack header */
+	    NEEDBYTES (5);
+	    if ((header[4] & 0xc0) == 0x40) {	/* mpeg2 */
+		NEEDBYTES (14);
+		len = 14 + (header[13] & 7);
+		NEEDBYTES (len);
+		DONEBYTES (len);
+		/* header points to the mpeg2 pack header */
+	    } else if ((header[4] & 0xf0) == 0x20) {	/* mpeg1 */
+		NEEDBYTES (12);
+		DONEBYTES (12);
+		/* header points to the mpeg1 pack header */
+	    } else {
+		fprintf (stderr, "weird pack header\n");
+		DONEBYTES (5);
+	    }
+	    break;
+	case 0xbd:	/* private stream 1 */
+	    NEEDBYTES (7);
+	    if ((header[6] & 0xc0) == 0x80) {	/* mpeg2 */
+		NEEDBYTES (9);
+		len = 10 + header[8];
+		NEEDBYTES (len);
+		/* header points to the mpeg2 pes header */
+	    } else {	/* mpeg1 */
+		len = 7;
+		while ((header-1)[len] == 0xff) {
+		    len++;
+		    NEEDBYTES (len);
+		    if (len == 23) {
+			fprintf (stderr, "too much stuffing\n");
+			break;
+		    }
+		}
+		if (((header-1)[len] & 0xc0) == 0x40) {
+		    len += 2;
+		    NEEDBYTES (len);
+		}
+		len += mpeg1_skip_table[(header - 1)[len] >> 4] + 1;
+		NEEDBYTES (len);
+		/* header points to the mpeg1 pes header */
+	    }
+	    if ((header-1)[len] != demux_track) {
+		DONEBYTES (len);
+		bytes = 6 + (header[4] << 8) + header[5] - len;
+		if (bytes <= 0)
+		    continue;
+		goto skip;
+	    }
+	    len += 3;
+	    NEEDBYTES (len);
+	    DONEBYTES (len);
+	    bytes = 6 + (header[4] << 8) + header[5] - len;
+	    if (bytes > end - buf) {
+		dca_decode_data (buf, end);
+		state = DEMUX_DATA;
+		state_bytes = bytes - (end - buf);
+		return 0;
+	    } else if (bytes > 0) {
+		dca_decode_data (buf, buf + bytes);
+		buf += bytes;
+	    }
+	    break;
+	default:
+	    if (header[3] < 0xb9) {
+		fprintf (stderr,
+			 "looks like a video stream, not system stream\n");
+		exit (1);
+	    } else {
+		NEEDBYTES (6);
+		DONEBYTES (6);
+		bytes = (header[4] << 8) + header[5];
+	    skip:
+		if (bytes > end - buf) {
+		    state = DEMUX_SKIP;
+		    state_bytes = bytes - (end - buf);
+		    return 0;
+		}
+		buf += bytes;
+	    }
+	}
+    }
+}
+
+static void ps_loop (void)
+{
+    uint8_t * end;
+
+    do {
+	end = buffer + fread (buffer, 1, BUFFER_SIZE, in_file);
+	if (demux (buffer, end, 0))
+	    break;	/* hit program_end_code */
+    } while (end == buffer + BUFFER_SIZE && !sigint);
+}
+
+static void ts_loop (void)
+{
+    uint8_t * buf;
+    uint8_t * nextbuf;
+    uint8_t * data;
+    uint8_t * end;
+    int pid;
+
+    buf = buffer;
+    do {
+	end = buf + fread (buf, 1, buffer + BUFFER_SIZE - buf, in_file);
+	buf = buffer;
+	for (; (nextbuf = buf + 188) <= end; buf = nextbuf) {
+	    if (*buf != 0x47) {
+		fprintf (stderr, "bad sync byte\n");
+		nextbuf = buf + 1;
+		continue;
+	    }
+	    pid = ((buf[1] << 8) + buf[2]) & 0x1fff;
+	    if (pid != demux_pid)
+		continue;
+	    data = buf + 4;
+	    if (buf[3] & 0x20) {	/* buf contains an adaptation field */
+		data = buf + 5 + buf[4];
+		if (data > nextbuf)
+		    continue;
+	    }
+	    if (buf[3] & 0x10)
+		demux (data, nextbuf,
+		       (buf[1] & 0x40) ? DEMUX_PAYLOAD_START : 0);
+	}
+	if (end != buffer + BUFFER_SIZE)
+	    break;
+	memcpy (buffer, buf, end - buf);
+	buf = buffer + (end - buf);
+    } while (!sigint);
+}
+
+static void es_loop (void)
+{
+    int size;
+		
+    do {
+	size = fread (buffer, 1, BUFFER_SIZE, in_file);
+	dca_decode_data (buffer, buffer + size);
+    } while (size == BUFFER_SIZE && !sigint);
+}
+
+int main (int argc, char ** argv)
+{
+    uint32_t accel;
+
+#ifdef HAVE_IO_H
+    setmode (fileno (stdin), O_BINARY);
+    setmode (fileno (stdout), O_BINARY);
+#endif
+
+    fprintf (stderr, PACKAGE"-"VERSION
+	     " - by Gildas Bazin <gbazin@videolan.org>\n");
+
+    fprintf (stderr, "  based on the a52dec code from "
+	     "Michel Lespinasse <walken@zoy.org> and Aaron Holtzman\n");
+
+    handle_args (argc, argv);
+
+    accel = 0;
+
+    output = output_open ();
+    if (output == NULL) {
+	fprintf (stderr, "Can not open output\n");
+	return 1;
+    }
+
+    state = dca_init (accel);
+    if (state == NULL) {
+	fprintf (stderr, "DTS init failed\n");
+	return 1;
+    }
+
+    if (demux_pid)
+	ts_loop ();
+    else if (demux_track || demux_pes)
+	ps_loop ();
+    else
+	es_loop ();
+
+    dca_free (state);
+    print_fps (1);
+    if (output->close)
+	output->close (output);
+    return 0;
+}