Speex audio decoding

git-svn-id: svn://svn.mplayerhq.hu/mplayer/trunk@16917 b3059339-0415-0410-9bf9-f77b7e298cf2

1 files changed, 113 insertions, 0 deletions

diff --git a/libmpcodecs/ad_speex.c b/libmpcodecs/ad_speex.c
new file mode 100644
index 0000000000..284be899de
--- /dev/null
+++ b/libmpcodecs/ad_speex.c
@@ -0,0 +1,113 @@
+/**
+ * Speex decoder by Reimar Döffinger <Reimar.Doeffinger@stud.uni-karlsruhe.de>
+ * License: GPL
+ * This code may be be relicensed under the terms of the GNU LGPL when it
+ * becomes part of the FFmpeg project (ffmpeg.org)
+ */
+#include "config.h"
+#include <speex/speex.h>
+#include <speex/speex_stereo.h>
+#include <speex/speex_header.h>
+#include "ad_internal.h"
+
+static ad_info_t info = {
+  "Speex audio decoder",
+  "speex",
+  "Reimar Döffinger",
+  "",
+  ""
+};
+
+LIBAD_EXTERN(speex)
+
+typedef struct {
+  SpeexBits bits;
+  void *dec_context;
+  SpeexStereoState stereo;
+  SpeexHeader *hdr;
+} context_t;
+
+static int preinit(sh_audio_t *sh) {
+  return 1;
+}
+
+static int init(sh_audio_t *sh) {
+  context_t *ctx = (context_t *)calloc(1, sizeof(context_t));
+  const SpeexMode *spx_mode;
+  const SpeexStereoState st_st = SPEEX_STEREO_STATE_INIT; // hack
+  int mode;
+  if (!sh->wf || sh->wf->cbSize < 80) {
+    mp_msg(MSGT_DECAUDIO, MSGL_FATAL, "Missing extradata!\n");
+    return 0;
+  }
+  ctx->hdr = speex_packet_to_header((char *)&sh->wf[1], sh->wf->cbSize);
+  if (ctx->hdr->nb_channels != 1 && ctx->hdr->nb_channels != 2) {
+    mp_msg(MSGT_DECAUDIO, MSGL_WARN, "Invalid number of channels (%i), "
+            "assuming mono\n", ctx->hdr->nb_channels);
+    ctx->hdr->nb_channels = 1;
+  }
+  switch (ctx->hdr->mode) {
+    case 0:
+      spx_mode = &speex_nb_mode; break;
+    case 1:
+      spx_mode = &speex_wb_mode; break;
+    case 2:
+      spx_mode = &speex_uwb_mode; break;
+    default:
+      mp_msg(MSGT_DECAUDIO, MSGL_WARN, "Unknown speex mode (%i)\n", mode);
+      spx_mode = &speex_nb_mode;
+  }
+  ctx->dec_context = speex_decoder_init(spx_mode);
+  speex_bits_init(&ctx->bits);
+  memcpy(&ctx->stereo, &st_st, sizeof(ctx->stereo)); // hack part 2
+  sh->channels = ctx->hdr->nb_channels;
+  sh->samplerate = ctx->hdr->rate;
+  sh->samplesize = 2;
+  sh->sample_format = AF_FORMAT_S16_NE;
+  sh->context = ctx;
+  return 1;
+}
+
+static void uninit(sh_audio_t *sh) {
+  context_t *ctx = sh->context;
+  if (ctx) {
+    speex_bits_destroy(&ctx->bits);
+    speex_decoder_destroy(ctx->dec_context);
+    if (ctx->hdr)
+      speex_free(ctx->hdr);
+    free(ctx);
+  }
+  ctx = NULL;
+}
+
+static int decode_audio(sh_audio_t *sh, unsigned char *buf,
+                        int minlen, int maxlen) {
+  context_t *ctx = sh->context;
+  int len, framelen, framesamples;
+  char *packet;
+  int i, err;
+  speex_decoder_ctl(ctx->dec_context, SPEEX_GET_FRAME_SIZE, &framesamples);
+  framelen = framesamples * ctx->hdr->nb_channels * sizeof(short);
+  if (maxlen < ctx->hdr->frames_per_packet * framelen) {
+    mp_msg(MSGT_DECAUDIO, MSGL_V, "maxlen too small in decode_audio\n");
+    return -1;
+  }
+  len = ds_get_packet(sh->ds, (unsigned char **)&packet);
+  if (len <= 0) return -1;
+  speex_bits_read_from(&ctx->bits, packet, len);
+  i = ctx->hdr->frames_per_packet;
+  do {
+    err = speex_decode_int(ctx->dec_context, &ctx->bits, (short *)buf);
+    if (err == -2)
+      mp_msg(MSGT_DECAUDIO, MSGL_ERR, "Error decoding file.\n");
+    if (ctx->hdr->nb_channels == 2)
+      speex_decode_stereo_int((short *)buf, framesamples, &ctx->stereo);
+    buf = &buf[framelen];
+  } while (--i > 0);
+  return ctx->hdr->frames_per_packet * framelen;
+}
+
+static int control(sh_audio_t *sh, int cmd, void *arg, ...) {
+  return CONTROL_UNKNOWN;
+}
+