Raw DTS demuxer

CVS patchset: 7562
CVS date: 2005/05/26 22:09:23

4 files changed, 407 insertions, 3 deletions

diff --git a/src/demuxers/Makefile.am b/src/demuxers/Makefile.am
index 043efe6ca..fb0481fc7 100644
--- a/src/demuxers/Makefile.am
+++ b/src/demuxers/Makefile.am
@@ -122,7 +122,7 @@ xineplug_dmx_audio_la_SOURCES = group_audio.c demux_aud.c demux_aiff.c \
 				demux_realaudio.c demux_snd.c demux_voc.c \
 				demux_vox.c demux_wav.c demux_ac3.c id3.c \
 				demux_aac.c demux_mod.c demux_flac.c \
-				demux_mpc.c
+				demux_mpc.c demux_dts.c
 xineplug_dmx_audio_la_LIBADD = $(XINE_LIB) $(LIBMODPLUG_LIBS)
 xineplug_dmx_audio_la_LDFLAGS = -avoid-version -module @XINE_PLUGIN_MIN_SYMS@
 
diff --git a/src/demuxers/demux_dts.c b/src/demuxers/demux_dts.c
new file mode 100644
index 000000000..70c7e55d0
--- /dev/null
+++ b/src/demuxers/demux_dts.c
@@ -0,0 +1,398 @@
+/*
+ * Copyright (C) 2005 the xine project
+ *
+ * This file is part of xine, a free video player.
+ *
+ * xine 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.
+ *
+ * xine 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., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA
+ *
+ * Raw DTS Demuxer by James Stembridge (jstembridge@gmail.com)
+ *
+ * $Id: demux_dts.c,v 1.1 2005/05/26 22:09:23 jstembridge Exp $
+ */
+
+#ifdef HAVE_CONFIG_H
+#include "config.h"
+#endif
+
+#include <stdio.h>
+#include <fcntl.h>
+#include <unistd.h>
+#include <string.h>
+#include <stdlib.h>
+#include <ctype.h>
+
+#define LOG_MODULE "demux_dts"
+#define LOG_VERBOSE
+/*
+#define LOG
+*/
+
+#include "xine_internal.h"
+#include "xineutils.h"
+#include "demux.h"
+#include "buffer.h"
+#include "bswap.h"
+#include "group_audio.h"
+
+#define DATA_TAG 0x61746164
+
+typedef struct {
+  demux_plugin_t       demux_plugin;
+
+  xine_stream_t       *stream;
+  fifo_buffer_t       *video_fifo;
+  fifo_buffer_t       *audio_fifo;
+  input_plugin_t      *input;
+  int                  status;
+
+  int                  seek_flag;
+  int                  samples_per_frame;
+  int                  sample_rate;
+  int                  frame_size;
+
+  off_t                data_start;
+} demux_dts_t;
+
+typedef struct {
+  demux_class_t     demux_class;
+} demux_dts_class_t;
+
+static const int dts_sample_rates[] =
+{
+  0, 8000, 16000, 32000, 0, 0, 11025, 22050, 44100, 0, 0,
+  12000, 24000, 48000, 96000, 192000
+};
+
+static int open_dts_file(demux_dts_t *this) {
+  int i, offset = 0;
+  uint32_t syncword = 0;
+  uint8_t peak[MAX_PREVIEW_SIZE];
+
+  if (_x_demux_read_header(this->input, peak, MAX_PREVIEW_SIZE) != 
+      MAX_PREVIEW_SIZE)
+    return 0;
+
+  /* Check for wav header, as we'll handle DTS with a wav header shoved
+   * on the front for CD burning */
+  if ((peak[0]  == 'R') && (peak[1]  == 'I') && (peak[2]  == 'F') && 
+      (peak[3]  == 'F') && (peak[8]  == 'W') && (peak[9]  == 'A') &&
+      (peak[10] == 'V') && (peak[11] == 'E') && (peak[12] == 'f') &&
+      (peak[13] == 'm') && (peak[14] == 't') && (peak[15] == ' ')) {
+    /* Check this looks like a cd audio wav */
+    unsigned int audio_type;
+    xine_waveformatex *wave = (xine_waveformatex *) &peak[20];
+
+    _x_waveformatex_le2me(wave);
+    audio_type = _x_formattag_to_buf_audio(wave->wFormatTag);
+
+    if ((audio_type != BUF_AUDIO_LPCM_LE) || (wave->nChannels != 2) ||
+        (wave->nSamplesPerSec != 44100) || (wave->wBitsPerSample != 16))
+      return 0;
+
+    lprintf("looks like a cd audio wav file\n");
+
+    /* Find the data chunk */
+    offset = 20 + LE_32(&peak[16]);
+    while (offset < MAX_PREVIEW_SIZE-8) {
+      unsigned int chunk_tag = LE_32(&peak[offset]);
+      unsigned int chunk_size = LE_32(&peak[offset+4]);
+
+      if (chunk_tag == DATA_TAG) {
+        offset += 8;
+        lprintf("found the start of the data at offset %d\n", offset);
+        break;
+      } else
+        offset += chunk_size;
+    }
+  }
+
+  /* Look for a valid DTS syncword */
+  for (i=offset; i < MAX_PREVIEW_SIZE; i++) {
+    /* 14 bits and little endian bitstream */
+    if ((syncword == 0xff1f00e8) && 
+        ((peak[i] & 0xf0) == 0xf0) && (peak[i+1] == 0x07)) {
+      this->data_start = i-4;
+      lprintf("found DTS syncword at offset %d\n", i-4);
+      break;
+    }
+
+    syncword = (syncword << 8) | peak[i];
+  }
+
+  if (i < MAX_PREVIEW_SIZE-9) {
+    unsigned int nblks, fsize, sfreq;
+
+    /* 14 bits and little endian bitstream */
+    nblks = ((peak[this->data_start+4] & 0x07) << 4) |
+             ((peak[this->data_start+7] & 0x3c) >> 2);
+    fsize = (((peak[this->data_start+7] & 0x03) << 12) |
+              (peak[this->data_start+6] << 4) |
+             ((peak[this->data_start+9] & 0x3c) >> 2)) + 1;
+    sfreq = peak[this->data_start+8] & 0x0f;
+
+    if ((sfreq > sizeof(dts_sample_rates)/sizeof(int)) ||
+        (dts_sample_rates[sfreq] == 0))
+      return 0;
+
+    /* Big assumption - this is CBR data */
+    this->samples_per_frame = (nblks + 1) * 32;
+    this->frame_size = fsize * 8 / 14 * 2;
+    this->sample_rate = dts_sample_rates[sfreq];
+
+    lprintf("samples per frame: %d\n", this->samples_per_frame);
+    lprintf("frame size: %d\n", this->frame_size);
+    lprintf("sample rate: %d\n", this->sample_rate);
+
+    /* Seek to start of DTS data */
+    this->input->seek(this->input, this->data_start, SEEK_SET);
+
+    return 1;
+  }
+
+  return 0;
+}
+
+static int demux_dts_send_chunk (demux_plugin_t *this_gen) {
+  demux_dts_t *this = (demux_dts_t *) this_gen;
+
+  buf_element_t *buf = NULL;
+  off_t current_stream_pos;
+  int64_t audio_pts;
+  int frame_number;
+
+  current_stream_pos = this->input->get_current_pos(this->input) - 
+                       this->data_start;
+  frame_number = current_stream_pos / this->frame_size;
+
+  audio_pts = frame_number;
+  audio_pts *= 90000;
+  audio_pts *= this->samples_per_frame;
+  audio_pts /= this->sample_rate;
+
+  if (this->seek_flag) {
+    _x_demux_control_newpts(this->stream, audio_pts, BUF_FLAG_SEEK);
+    this->seek_flag = 0;
+  }
+
+  buf = this->audio_fifo->buffer_pool_alloc (this->audio_fifo);
+  buf->size = this->input->read(this->input, buf->content, this->frame_size);
+
+  if (buf->size == 0) {
+    buf->free_buffer(buf);
+    this->status = DEMUX_FINISHED;
+    return this->status;
+  }
+
+  buf->type = BUF_AUDIO_DTS;
+  if( this->input->get_length (this->input) )
+    buf->extra_info->input_normpos = (int)( (double) current_stream_pos * 
+        65535 / (this->input->get_length(this->input) - this->data_start) );
+  buf->extra_info->input_time = audio_pts / 90;
+  buf->pts = audio_pts;
+  buf->decoder_flags |= BUF_FLAG_FRAME_END;
+
+  this->audio_fifo->put (this->audio_fifo, buf);
+
+  return this->status;
+}
+
+static void demux_dts_send_headers(demux_plugin_t *this_gen) {
+  demux_dts_t *this = (demux_dts_t *) this_gen;
+  buf_element_t *buf;
+
+  this->video_fifo  = this->stream->video_fifo;
+  this->audio_fifo  = this->stream->audio_fifo;
+
+  this->status = DEMUX_OK;
+
+  /* load stream information */
+  _x_stream_info_set(this->stream, XINE_STREAM_INFO_HAS_VIDEO, 0);
+  _x_stream_info_set(this->stream, XINE_STREAM_INFO_HAS_AUDIO, 1);
+
+  /* send start buffers */
+  _x_demux_control_start(this->stream);
+
+  /* send init info to decoders */
+  if (this->audio_fifo) {
+    buf = this->audio_fifo->buffer_pool_alloc (this->audio_fifo);
+    buf->type = BUF_AUDIO_DTS;
+    buf->decoder_flags = BUF_FLAG_HEADER|BUF_FLAG_FRAME_END;
+    buf->size = 0;
+    this->audio_fifo->put (this->audio_fifo, buf);
+  }
+}
+
+static int demux_dts_get_stream_length (demux_plugin_t *this_gen) {
+  demux_dts_t *this = (demux_dts_t *) this_gen;
+  int stream_length = 0;
+
+  if (this->input->get_length(this->input)) {
+    stream_length = this->input->get_length(this->input) - this->data_start;
+    stream_length /= this->frame_size;
+    stream_length *= this->samples_per_frame;
+    stream_length /= this->sample_rate;
+    stream_length *= 1000;
+
+    lprintf("running time: %u\n", stream_length);
+  }
+
+  return stream_length;
+}
+
+static int demux_dts_seek (demux_plugin_t *this_gen,
+                           off_t start_pos, int start_time, int playing) {
+
+  demux_dts_t *this = (demux_dts_t *) this_gen;
+
+  this->seek_flag = 1;
+  this->status = DEMUX_OK;
+  _x_demux_flush_engine (this->stream);
+
+  /* if input is non-seekable, do not proceed with the rest of this
+   * seek function */
+  if (!INPUT_IS_SEEKABLE(this->input))
+    return this->status;
+
+  start_pos = (off_t) ( (double) start_pos / 65535 *
+      (this->input->get_length(this->input) - this->data_start) );
+
+  if (start_time) {
+    int length = demux_dts_get_stream_length (this_gen);
+    if (length != 0) {
+      start_pos = start_time * 
+          (this->input->get_length(this->input) - this->data_start) / length;
+    }
+  }
+
+  /* divide the requested offset integer-wise by the frame alignment and
+   * multiply by the frame alignment to determine the new starting block */
+  start_pos /= this->frame_size;
+  start_pos *= this->frame_size;
+  start_pos += this->data_start;
+  this->input->seek(this->input, start_pos, SEEK_SET);
+
+  return this->status;
+}
+
+static void demux_dts_dispose (demux_plugin_t *this_gen) {
+  demux_dts_t *this = (demux_dts_t *) this_gen;
+
+  free(this);
+}
+
+static int demux_dts_get_status (demux_plugin_t *this_gen) {
+  demux_dts_t *this = (demux_dts_t *) this_gen;
+
+  return this->status;
+}
+
+static uint32_t demux_dts_get_capabilities(demux_plugin_t *this_gen) {
+  return DEMUX_CAP_NOCAP;
+}
+
+static int demux_dts_get_optional_data(demux_plugin_t *this_gen,
+                                       void *data, int data_type) {
+  return DEMUX_OPTIONAL_UNSUPPORTED;
+}
+
+static demux_plugin_t *open_plugin (demux_class_t *class_gen, xine_stream_t *stream,
+                                   input_plugin_t *input) {
+
+  demux_dts_t   *this;
+
+  this         = xine_xmalloc (sizeof (demux_dts_t));
+  this->stream = stream;
+  this->input  = input;
+
+  this->demux_plugin.send_headers      = demux_dts_send_headers;
+  this->demux_plugin.send_chunk        = demux_dts_send_chunk;
+  this->demux_plugin.seek              = demux_dts_seek;
+  this->demux_plugin.dispose           = demux_dts_dispose;
+  this->demux_plugin.get_status        = demux_dts_get_status;
+  this->demux_plugin.get_stream_length = demux_dts_get_stream_length;
+  this->demux_plugin.get_capabilities  = demux_dts_get_capabilities;
+  this->demux_plugin.get_optional_data = demux_dts_get_optional_data;
+  this->demux_plugin.demux_class       = class_gen;
+
+  this->status = DEMUX_FINISHED;
+
+  switch (stream->content_detection_method) {
+
+  case METHOD_BY_EXTENSION: {
+    char *extensions, *mrl;
+
+    mrl = input->get_mrl (input);
+    extensions = class_gen->get_extensions (class_gen);
+
+    if (!_x_demux_check_extension (mrl, extensions)) {
+      free (this);
+      return NULL;
+    }
+  }
+  /* falling through is intended */
+
+  case METHOD_BY_CONTENT:
+  case METHOD_EXPLICIT:
+    if (!open_dts_file(this)) {
+      free (this);
+      return NULL;
+    }
+  break;
+
+  default:
+    free (this);
+    return NULL;
+  }
+
+  return &this->demux_plugin;
+}
+
+static char *get_description (demux_class_t *this_gen) {
+  return "Raw DTS file demux plugin";
+}
+
+static char *get_identifier (demux_class_t *this_gen) {
+  return "DTS";
+}
+
+static char *get_extensions (demux_class_t *this_gen) {
+  return "dts";
+}
+
+static char *get_mimetypes (demux_class_t *this_gen) {
+  return NULL;
+}
+
+static void class_dispose (demux_class_t *this_gen) {
+  demux_dts_class_t *this = (demux_dts_class_t *) this_gen;
+
+  free (this);
+}
+
+void *demux_dts_init_plugin (xine_t *xine, void *data) {
+  demux_dts_class_t     *this;
+
+  this = xine_xmalloc (sizeof (demux_dts_class_t));
+
+  this->demux_class.open_plugin     = open_plugin;
+  this->demux_class.get_description = get_description;
+  this->demux_class.get_identifier  = get_identifier;
+  this->demux_class.get_mimetypes   = get_mimetypes;
+  this->demux_class.get_extensions  = get_extensions;
+  this->demux_class.dispose         = class_dispose;
+
+  return this;
+}
+
diff --git a/src/demuxers/group_audio.c b/src/demuxers/group_audio.c
index dc47df990..af5a7ce41 100644
--- a/src/demuxers/group_audio.c
+++ b/src/demuxers/group_audio.c
@@ -19,7 +19,7 @@
  *
  * This file contains plugin entries for several demuxers used in games
  *
- * $Id: group_audio.c,v 1.17 2005/02/06 15:26:19 tmattern Exp $
+ * $Id: group_audio.c,v 1.18 2005/05/26 22:09:23 jstembridge Exp $
  */
 
 #ifdef HAVE_CONFIG_H
@@ -55,6 +55,10 @@ demuxer_info_t demux_info_cdda = {
   10                       /* priority */
 };
 
+demuxer_info_t demux_info_dts = {
+  0                        /* priority */
+};
+
 demuxer_info_t demux_info_flac = {
   10                       /* priority */
 };
@@ -104,6 +108,7 @@ plugin_info_t xine_plugin_info[] = {
   { PLUGIN_DEMUX, 26, "aud",       XINE_VERSION_CODE, &demux_info_aud,       demux_aud_init_plugin },
   { PLUGIN_DEMUX, 26, "aiff",      XINE_VERSION_CODE, &demux_info_aiff,      demux_aiff_init_plugin },
   { PLUGIN_DEMUX, 26, "cdda",      XINE_VERSION_CODE, &demux_info_cdda,      demux_cdda_init_plugin },
+  { PLUGIN_DEMUX, 26, "dts",       XINE_VERSION_CODE, &demux_info_dts,       demux_dts_init_plugin },
   { PLUGIN_DEMUX, 26, "flac",      XINE_VERSION_CODE, &demux_info_flac,      demux_flac_init_plugin },
   { PLUGIN_DEMUX, 26, "mp3",       XINE_VERSION_CODE, &demux_info_mpgaudio,  demux_mpgaudio_init_class },
   { PLUGIN_DEMUX, 26, "mpc",       XINE_VERSION_CODE, &demux_info_mpc,       demux_mpc_init_plugin },
diff --git a/src/demuxers/group_audio.h b/src/demuxers/group_audio.h
index 7badc9339..115787a5e 100644
--- a/src/demuxers/group_audio.h
+++ b/src/demuxers/group_audio.h
@@ -17,7 +17,7 @@
  * along with this program; if not, write to the Free Software
  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA
  *
- * $Id: group_audio.h,v 1.6 2005/01/14 15:24:23 jstembridge Exp $
+ * $Id: group_audio.h,v 1.7 2005/05/26 22:09:23 jstembridge Exp $
  */
 
 #ifndef HAVE_GROUP_AUDIO_H
@@ -30,6 +30,7 @@ void *demux_ac3_init_plugin (xine_t *xine, void *data);
 void *demux_aud_init_plugin (xine_t *xine, void *data);
 void *demux_aiff_init_plugin (xine_t *xine, void *data);
 void *demux_cdda_init_plugin (xine_t *xine, void *data);
+void *demux_dts_init_plugin (xine_t *xine, void *data);
 void *demux_flac_init_plugin (xine_t *xine, void *data);
 void *demux_mpgaudio_init_class (xine_t *xine, void *data);
 void *demux_mpc_init_plugin (xine_t *xine, void *data);